aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/target
diff options
context:
space:
mode:
authorAnnie Tallund <annie.tallund@arm.com>2022-12-14 15:55:19 +0100
committerBenjamin Klimczak <benjamin.klimczak@arm.com>2023-02-08 15:17:12 +0000
commit09ecc5c8acb758e8def33155feb746a34dd7b560 (patch)
tree65c39a7f7929b745b9c5a31ab48bb4c6e97cb3ea /src/mlia/target
parent6fbcffb4ee039438a409fbc92e38fa5d1d118833 (diff)
downloadmlia-09ecc5c8acb758e8def33155feb746a34dd7b560.tar.gz
MLIA-590 Support path to custom target profiles
- Start using TOML format for target profile - Add support for loading custom target profile files Change-Id: I6be019d4341e93115440ccdbdb6dafdc1c85b966
Diffstat (limited to 'src/mlia/target')
-rw-r--r--src/mlia/target/cortex_a/advisor.py6
-rw-r--r--src/mlia/target/ethos_u/advisor.py4
-rw-r--r--src/mlia/target/tosa/advisor.py6
3 files changed, 10 insertions, 6 deletions
diff --git a/src/mlia/target/cortex_a/advisor.py b/src/mlia/target/cortex_a/advisor.py
index 1249d93..52af592 100644
--- a/src/mlia/target/cortex_a/advisor.py
+++ b/src/mlia/target/cortex_a/advisor.py
@@ -64,7 +64,7 @@ class CortexAInferenceAdvisor(DefaultInferenceAdvisor):
def configure_and_get_cortexa_advisor(
context: ExecutionContext,
- target_profile: str,
+ target_profile: str | Path,
model: str | Path,
**_extra_args: Any,
) -> InferenceAdvisor:
@@ -78,7 +78,9 @@ def configure_and_get_cortexa_advisor(
return CortexAInferenceAdvisor()
-def _get_config_parameters(model: str | Path, target_profile: str) -> dict[str, Any]:
+def _get_config_parameters(
+ model: str | Path, target_profile: str | Path
+) -> dict[str, Any]:
"""Get configuration parameters for the advisor."""
advisor_parameters: dict[str, Any] = {
"cortex_a_inference_advisor": {
diff --git a/src/mlia/target/ethos_u/advisor.py b/src/mlia/target/ethos_u/advisor.py
index ce4e0fc..937e91c 100644
--- a/src/mlia/target/ethos_u/advisor.py
+++ b/src/mlia/target/ethos_u/advisor.py
@@ -123,7 +123,7 @@ class EthosUInferenceAdvisor(DefaultInferenceAdvisor):
def configure_and_get_ethosu_advisor(
context: ExecutionContext,
- target_profile: str,
+ target_profile: str | Path,
model: str | Path,
**extra_args: Any,
) -> InferenceAdvisor:
@@ -155,7 +155,7 @@ _DEFAULT_OPTIMIZATION_TARGETS = [
def _get_config_parameters(
model: str | Path,
- target_profile: str,
+ target_profile: str | Path,
**extra_args: Any,
) -> dict[str, Any]:
"""Get configuration parameters for the advisor."""
diff --git a/src/mlia/target/tosa/advisor.py b/src/mlia/target/tosa/advisor.py
index b60e824..e8aad53 100644
--- a/src/mlia/target/tosa/advisor.py
+++ b/src/mlia/target/tosa/advisor.py
@@ -78,7 +78,7 @@ class TOSAInferenceAdvisor(DefaultInferenceAdvisor):
def configure_and_get_tosa_advisor(
context: ExecutionContext,
- target_profile: str,
+ target_profile: str | Path,
model: str | Path,
**_extra_args: Any,
) -> InferenceAdvisor:
@@ -92,7 +92,9 @@ def configure_and_get_tosa_advisor(
return TOSAInferenceAdvisor()
-def _get_config_parameters(model: str | Path, target_profile: str) -> dict[str, Any]:
+def _get_config_parameters(
+ model: str | Path, target_profile: str | Path
+) -> dict[str, Any]:
"""Get configuration parameters for the advisor."""
advisor_parameters: dict[str, Any] = {
"tosa_inference_advisor": {