diff options
Diffstat (limited to 'src/mlia/target')
-rw-r--r-- | src/mlia/target/config.py | 4 | ||||
-rw-r--r-- | src/mlia/target/cortex_a/config.py | 10 |
2 files changed, 8 insertions, 6 deletions
diff --git a/src/mlia/target/config.py b/src/mlia/target/config.py index 8ccdad8..8492086 100644 --- a/src/mlia/target/config.py +++ b/src/mlia/target/config.py @@ -85,9 +85,11 @@ T = TypeVar("T", bound="TargetProfile") class TargetProfile(ABC): """Base class for target profiles.""" - def __init__(self, target: str) -> None: + def __init__(self, target: str, backend_config: dict | None = None) -> None: """Init TargetProfile instance with the target name.""" self.target = target + # Load backend config(s) to be handled by the backend(s) later. + self.backend_config = {} if backend_config is None else backend_config @classmethod def load(cls: type[T], path: str | Path) -> T: diff --git a/src/mlia/target/cortex_a/config.py b/src/mlia/target/cortex_a/config.py index f91031e..4f33f3d 100644 --- a/src/mlia/target/cortex_a/config.py +++ b/src/mlia/target/cortex_a/config.py @@ -15,12 +15,12 @@ class CortexAConfiguration(TargetProfile): def __init__(self, **kwargs: Any) -> None: """Init Cortex-A target configuration.""" target = kwargs["target"] - super().__init__(target) + backend_config = kwargs.get("backend") + super().__init__(target, backend_config) - self.backend_config = kwargs.get("backend") - self.armnn_tflite_delegate_version = kwargs["backend"]["armnn-tflite-delegate"][ - "version" - ] + self.armnn_tflite_delegate_version = self.backend_config[ + "armnn-tflite-delegate" + ]["version"] def verify(self) -> None: """Check the parameters.""" |