From 6a88ee5315b4ce5b023370c1e55e48bf9f2b6f67 Mon Sep 17 00:00:00 2001
From: Dmitrii Agibov <dmitrii.agibov@arm.com>
Date: Fri, 18 Nov 2022 17:21:09 +0000
Subject: Rename modules

- Rename module "mlia.devices" into "mlia.target"
- Rename module "mlia.target.ethosu" into "mlia.target.ethos_u"
- Rename module "mlia.target.cortexa" into "mlia.target.cortex_a"
- Rename and update tests

Change-Id: I6dca7c8646d881f739fb6b5914d1cc7e45e63dc2
---
 src/mlia/target/ethos_u/config.py | 90 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 90 insertions(+)
 create mode 100644 src/mlia/target/ethos_u/config.py

(limited to 'src/mlia/target/ethos_u/config.py')

diff --git a/src/mlia/target/ethos_u/config.py b/src/mlia/target/ethos_u/config.py
new file mode 100644
index 0000000..8d8f481
--- /dev/null
+++ b/src/mlia/target/ethos_u/config.py
@@ -0,0 +1,90 @@
+# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""Ethos-U configuration."""
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+from mlia.backend.vela.compiler import resolve_compiler_config
+from mlia.backend.vela.compiler import VelaCompilerOptions
+from mlia.target.config import IPConfiguration
+from mlia.utils.filesystem import get_profile
+from mlia.utils.filesystem import get_vela_config
+
+
+logger = logging.getLogger(__name__)
+
+
+class EthosUConfiguration(IPConfiguration):
+    """Ethos-U configuration."""
+
+    def __init__(self, target_profile: str) -> None:
+        """Init Ethos-U target configuration."""
+        target_data = get_profile(target_profile)
+        _check_target_data_complete(target_data)
+
+        target = target_data["target"]
+        super().__init__(target)
+
+        mac = target_data["mac"]
+        _check_device_options_valid(target, mac)
+
+        self.mac = mac
+        self.compiler_options = VelaCompilerOptions(
+            system_config=target_data["system_config"],
+            memory_mode=target_data["memory_mode"],
+            config_files=str(get_vela_config()),
+            accelerator_config=f"{self.target}-{mac}",  # type: ignore
+        )
+
+    @property
+    def resolved_compiler_config(self) -> dict[str, Any]:
+        """Resolve compiler configuration."""
+        return resolve_compiler_config(self.compiler_options)
+
+    def __str__(self) -> str:
+        """Return string representation."""
+        return (
+            f"Ethos-U target={self.target} "
+            f"mac={self.mac} "
+            f"compiler_options={self.compiler_options}"
+        )
+
+    def __repr__(self) -> str:
+        """Return string representation."""
+        return f"<Ethos-U configuration target={self.target}>"
+
+
+def get_target(target_profile: str) -> EthosUConfiguration:
+    """Get target instance based on provided params."""
+    if not target_profile:
+        raise Exception("No target profile given")
+
+    return EthosUConfiguration(target_profile)
+
+
+def _check_target_data_complete(target_data: dict[str, Any]) -> None:
+    """Check if profile contains all needed data."""
+    mandatory_keys = {"target", "mac", "system_config", "memory_mode"}
+    missing_keys = sorted(mandatory_keys - target_data.keys())
+
+    if missing_keys:
+        raise Exception(f"Mandatory fields missing from target profile: {missing_keys}")
+
+
+def _check_device_options_valid(target: str, mac: int) -> None:
+    """Check if mac is valid for selected device."""
+    target_mac_ranges = {
+        "ethos-u55": [32, 64, 128, 256],
+        "ethos-u65": [256, 512],
+    }
+
+    if target not in target_mac_ranges:
+        raise Exception(f"Unsupported target: {target}")
+
+    target_mac_range = target_mac_ranges[target]
+    if mac not in target_mac_range:
+        raise Exception(
+            f"Mac value for selected device should be in {target_mac_range}"
+        )
-- 
cgit v1.2.1