aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/devices/cortexa/operators.py
diff options
context:
space:
mode:
authorRaul Farkas <raul.farkas@arm.com>2022-10-12 10:48:35 +0100
committerRaul Farkas <raul.farkas@arm.com>2022-10-18 14:29:23 +0100
commit4fa21325ec498adbf467876c2413c054d0e85c5b (patch)
tree892d68467bc6666e9daf74ab4d141810dcec1ac6 /src/mlia/devices/cortexa/operators.py
parent89c1f4bafb51dbbed705b6960810d90825318b13 (diff)
downloadmlia-4fa21325ec498adbf467876c2413c054d0e85c5b.tar.gz
MLIA-409 Create new Cortex-A device skeleton
* Add Cortex-A device skeleton * Add unit tests for the Cortex-A device skeleton * Update profiles.json by adding the new "cortex-a" profile * Add new cortex-a factory to the get_advisor method in api.py * Disable performance and optimization commands for the cortex-a profile. * Update trademarks section in README.md * Update pyproject.toml to not run similarity check in imports Change-Id: I2e228aaada1e2d3c5cc329d70572b51962ff517f
Diffstat (limited to 'src/mlia/devices/cortexa/operators.py')
-rw-r--r--src/mlia/devices/cortexa/operators.py29
1 files changed, 29 insertions, 0 deletions
diff --git a/src/mlia/devices/cortexa/operators.py b/src/mlia/devices/cortexa/operators.py
new file mode 100644
index 0000000..6a314b7
--- /dev/null
+++ b/src/mlia/devices/cortexa/operators.py
@@ -0,0 +1,29 @@
+# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""Cortex-A tools module."""
+from __future__ import annotations
+
+from dataclasses import dataclass
+from pathlib import Path
+
+
+@dataclass
+class Operator:
+ """Cortex-A compatibility information of the operator."""
+
+ name: str
+ location: str
+ is_cortex_a_compatible: bool
+
+
+@dataclass
+class CortexACompatibilityInfo:
+ """Model's operators."""
+
+ cortex_a_compatible: bool
+ operators: list[Operator]
+
+
+def get_cortex_a_compatibility_info(model_path: Path) -> CortexACompatibilityInfo:
+ """Return list of model's operators."""
+ raise NotImplementedError()