aboutsummaryrefslogtreecommitdiff
path: root/tests/test_target_cortex_a_reporters.py
diff options
context:
space:
mode:
authorAnnie Tallund <annie.tallund@arm.com>2023-02-06 14:56:31 +0100
committerBenjamin Klimczak <benjamin.klimczak@arm.com>2023-02-13 16:20:41 +0000
commit49dfd03617ed28a4fcbd847dd958c27442c84556 (patch)
tree1fe14ba3980dd6be15f866e97b67582831344eb9 /tests/test_target_cortex_a_reporters.py
parente3bef50932abdc2e32fa5636fb7a496b149e72d9 (diff)
downloadmlia-49dfd03617ed28a4fcbd847dd958c27442c84556.tar.gz
MLIA-709 Update compatibility data for Cortex-A0.6.0-rc.1
- Add operator compatibility data for Cortex-A via for ArmNN TensorFlow Lite delegate 22.11 - Extend the Cortex-A target profile to include the version of the ArmNN TensorFlow Lite delegate to be used. - Some re-factoring work to support multiple versions and the new target profile parameter. Change-Id: Iae91bb0757ea3909be975af68b34d0ca2be47c43
Diffstat (limited to 'tests/test_target_cortex_a_reporters.py')
-rw-r--r--tests/test_target_cortex_a_reporters.py19
1 files changed, 11 insertions, 8 deletions
diff --git a/tests/test_target_cortex_a_reporters.py b/tests/test_target_cortex_a_reporters.py
index 6866396..7ed0996 100644
--- a/tests/test_target_cortex_a_reporters.py
+++ b/tests/test_target_cortex_a_reporters.py
@@ -11,6 +11,7 @@ from mlia.nn.tensorflow.tflite_compat import TFLiteCompatibilityInfo
from mlia.nn.tensorflow.tflite_compat import TFLiteCompatibilityStatus
from mlia.nn.tensorflow.tflite_graph import TFL_ACTIVATION_FUNCTION
from mlia.target.cortex_a.config import CortexAConfiguration
+from mlia.target.cortex_a.operators import CortexACompatibilityInfo
from mlia.target.cortex_a.operators import Operator
from mlia.target.cortex_a.reporters import cortex_a_formatters
from mlia.target.cortex_a.reporters import report_target
@@ -27,14 +28,16 @@ def test_report_target() -> None:
(
[Advice(["Sample", "Advice"])],
TFLiteCompatibilityInfo(status=TFLiteCompatibilityStatus.COMPATIBLE),
- [
- Operator(
- name="Test",
- location="loc",
- support_type=Operator.SupportType.OP_NOT_SUPPORTED,
- activation_func=TFL_ACTIVATION_FUNCTION.NONE,
- )
- ],
+ CortexACompatibilityInfo(
+ [
+ Operator(
+ name="Test",
+ location="loc",
+ activation_func=TFL_ACTIVATION_FUNCTION.NONE,
+ )
+ ],
+ CortexAConfiguration.load_profile("cortex-a").armnn_tflite_delegate_version,
+ ),
),
)
def test_cortex_a_formatters(data: Any) -> None: