aboutsummaryrefslogtreecommitdiff
path: root/tests/test_backend_registry.py
diff options
context:
space:
mode:
authorBenjamin Klimczak <benjamin.klimczak@arm.com>2023-02-10 13:12:57 +0000
committerBenjamin Klimczak <benjamin.klimczak@arm.com>2023-02-10 14:56:43 +0000
commit718277eaece76902c4950f18d428907b39a18ef1 (patch)
tree6fda1b8f8c2c0151f3697a2e9f31c2f5551cc2ff /tests/test_backend_registry.py
parentfa1fad9332e2912f12a44a1b07716ee434174308 (diff)
downloadmlia-718277eaece76902c4950f18d428907b39a18ef1.tar.gz
MLIA-769 Add "pretty names" for targets / backends
- Provide "pretty names" to print information for targets and backends. - Use 'target_config' instead of 'target' if a target profile is used. - Fix minor issue in output regarding the output directory. Change-Id: Ib38231f30b4d609a0d1e8f9c52b2fb547c69cb6a
Diffstat (limited to 'tests/test_backend_registry.py')
-rw-r--r--tests/test_backend_registry.py16
1 files changed, 8 insertions, 8 deletions
diff --git a/tests/test_backend_registry.py b/tests/test_backend_registry.py
index e3e2da2..cc05632 100644
--- a/tests/test_backend_registry.py
+++ b/tests/test_backend_registry.py
@@ -19,19 +19,19 @@ from mlia.core.common import AdviceCategory
("backend", "advices", "systems", "type_"),
(
(
- "ArmNNTFLiteDelegate",
+ "armnn-tflite-delegate",
[AdviceCategory.COMPATIBILITY],
None,
BackendType.BUILTIN,
),
(
- "Corstone-300",
+ "corstone-300",
[AdviceCategory.PERFORMANCE, AdviceCategory.OPTIMIZATION],
[System.LINUX_AMD64],
BackendType.CUSTOM,
),
(
- "Corstone-310",
+ "corstone-310",
[AdviceCategory.PERFORMANCE, AdviceCategory.OPTIMIZATION],
[System.LINUX_AMD64],
BackendType.CUSTOM,
@@ -43,7 +43,7 @@ from mlia.core.common import AdviceCategory
BackendType.WHEEL,
),
(
- "Vela",
+ "vela",
[
AdviceCategory.COMPATIBILITY,
AdviceCategory.PERFORMANCE,
@@ -86,11 +86,11 @@ def test_backend_registry(
def test_get_supported_backends() -> None:
"""Test function get_supported_backends."""
assert get_supported_backends() == [
- "ArmNNTFLiteDelegate",
- "Corstone-300",
- "Corstone-310",
- "Vela",
+ "armnn-tflite-delegate",
+ "corstone-300",
+ "corstone-310",
"tosa-checker",
+ "vela",
]