aboutsummaryrefslogtreecommitdiff
path: root/tests/test_target_config.py
diff options
context:
space:
mode:
authorNathan Bailey <nathan.bailey@arm.com>2024-02-15 14:50:58 +0000
committerNathan Bailey <nathan.bailey@arm.com>2024-03-14 15:45:40 +0000
commit0b552d2ae47da4fb9c16d2a59d6ebe12c8307771 (patch)
tree09b40b939acbe0bcf02dcc77a7ed7ce4aba94322 /tests/test_target_config.py
parent09b272be6e88d84a30cb89fb71f3fc3c64d20d2e (diff)
downloadmlia-0b552d2ae47da4fb9c16d2a59d6ebe12c8307771.tar.gz
feat: Enable rewrite parameterisation
Enables user to provide a toml or default profile to change training settings for rewrite optimization Resolves: MLIA-1004 Signed-off-by: Nathan Bailey <nathan.bailey@arm.com> Change-Id: I3bf9f44b9a2062fb71ef36eb32c9a69edcc48061
Diffstat (limited to 'tests/test_target_config.py')
-rw-r--r--tests/test_target_config.py16
1 files changed, 8 insertions, 8 deletions
diff --git a/tests/test_target_config.py b/tests/test_target_config.py
index 8055af0..56e9f11 100644
--- a/tests/test_target_config.py
+++ b/tests/test_target_config.py
@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates.
+# SPDX-FileCopyrightText: Copyright 2022-2024, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Tests for the backend config module."""
from __future__ import annotations
@@ -10,9 +10,9 @@ from mlia.backend.config import BackendType
from mlia.backend.config import System
from mlia.core.common import AdviceCategory
from mlia.target.config import BUILTIN_SUPPORTED_PROFILE_NAMES
-from mlia.target.config import get_builtin_profile_path
from mlia.target.config import get_builtin_supported_profile_names
-from mlia.target.config import is_builtin_profile
+from mlia.target.config import get_builtin_target_profile_path
+from mlia.target.config import is_builtin_target_profile
from mlia.target.config import load_profile
from mlia.target.config import TargetInfo
from mlia.target.config import TargetProfile
@@ -33,23 +33,23 @@ def test_builtin_supported_profile_names() -> None:
"tosa",
]
for profile_name in BUILTIN_SUPPORTED_PROFILE_NAMES:
- assert is_builtin_profile(profile_name)
- profile_file = get_builtin_profile_path(profile_name)
+ assert is_builtin_target_profile(profile_name)
+ profile_file = get_builtin_target_profile_path(profile_name)
assert profile_file.is_file()
def test_builtin_profile_files() -> None:
"""Test function 'get_bulitin_profile_file'."""
- profile_file = get_builtin_profile_path("cortex-a")
+ profile_file = get_builtin_target_profile_path("cortex-a")
assert profile_file.is_file()
- profile_file = get_builtin_profile_path("UNKNOWN_FILE_THAT_DOES_NOT_EXIST")
+ profile_file = get_builtin_target_profile_path("UNKNOWN_FILE_THAT_DOES_NOT_EXIST")
assert not profile_file.exists()
def test_load_profile() -> None:
"""Test getting profile data."""
- profile_file = get_builtin_profile_path("ethos-u55-256")
+ profile_file = get_builtin_target_profile_path("ethos-u55-256")
assert load_profile(profile_file) == {
"target": "ethos-u55",
"mac": 256,