aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/target
diff options
context:
space:
mode:
Diffstat (limited to 'src/mlia/target')
-rw-r--r--src/mlia/target/config.py4
-rw-r--r--src/mlia/target/cortex_a/config.py10
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."""