aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/target/cortex_a/config.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/mlia/target/cortex_a/config.py')
-rw-r--r--src/mlia/target/cortex_a/config.py15
1 files changed, 15 insertions, 0 deletions
diff --git a/src/mlia/target/cortex_a/config.py b/src/mlia/target/cortex_a/config.py
index fd39e0a..f91031e 100644
--- a/src/mlia/target/cortex_a/config.py
+++ b/src/mlia/target/cortex_a/config.py
@@ -5,6 +5,7 @@ from __future__ import annotations
from typing import Any
+from mlia.backend.armnn_tflite_delegate.compat import ARMNN_TFLITE_DELEGATE
from mlia.target.config import TargetProfile
@@ -16,8 +17,22 @@ class CortexAConfiguration(TargetProfile):
target = kwargs["target"]
super().__init__(target)
+ self.backend_config = kwargs.get("backend")
+ self.armnn_tflite_delegate_version = kwargs["backend"]["armnn-tflite-delegate"][
+ "version"
+ ]
+
def verify(self) -> None:
"""Check the parameters."""
super().verify()
if self.target != "cortex-a":
raise ValueError(f"Wrong target {self.target} for Cortex-A configuration.")
+
+ if not self.armnn_tflite_delegate_version:
+ raise ValueError("No version for ArmNN TensorFlow Lite delegate specified.")
+ if self.armnn_tflite_delegate_version not in ARMNN_TFLITE_DELEGATE["ops"]:
+ raise ValueError(
+ f"Version '{self.armnn_tflite_delegate_version}' of "
+ f"backend {ARMNN_TFLITE_DELEGATE['backend']} is not supported. "
+ f"Choose from: {', '.join(ARMNN_TFLITE_DELEGATE['ops'])}"
+ )