From dcd0bd31985c27e1d07333351b26cf8ad12ad1fd Mon Sep 17 00:00:00 2001
From: Benjamin Klimczak <benjamin.klimczak@arm.com>
Date: Wed, 14 Dec 2022 11:20:11 +0000
Subject: MLIA-589 Create an API to get target information

Change-Id: Ieeaa9188ea1e29e2ccaad7475d457bce71e3140d
---
 src/mlia/cli/main.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

(limited to 'src/mlia/cli/main.py')

diff --git a/src/mlia/cli/main.py b/src/mlia/cli/main.py
index 98fdb63..ac60308 100644
--- a/src/mlia/cli/main.py
+++ b/src/mlia/cli/main.py
@@ -12,6 +12,7 @@ from pathlib import Path
 
 from mlia import __version__
 from mlia.backend.errors import BackendUnavailableError
+from mlia.backend.registry import registry as backend_registry
 from mlia.cli.commands import all_tests
 from mlia.cli.commands import backend_install
 from mlia.cli.commands import backend_list
@@ -36,6 +37,7 @@ from mlia.cli.options import add_tflite_model_options
 from mlia.core.context import ExecutionContext
 from mlia.core.errors import ConfigurationError
 from mlia.core.errors import InternalError
+from mlia.target.registry import registry as target_registry
 
 
 logger = logging.getLogger(__name__)
@@ -46,11 +48,10 @@ ML Inference Advisor {__version__}
 Help the design and optimization of neural network models for efficient inference on a target CPU and NPU
 
 Supported targets:
+{target_registry}
 
- - Cortex-A  <op compatibility>
- - Ethos-U55 <op compatibility, perf estimation, model opt>
- - Ethos-U65 <op compatibility, perf estimation, model opt>
- - TOSA      <op compatibility>
+Supported backends:
+{backend_registry}
 
 """.strip()
 
-- 
cgit v1.2.1