diff options
| author | Annie Tallund <annie.tallund@arm.com> | 2022-12-14 15:55:19 +0100 |
|---|---|---|
| committer | Benjamin Klimczak <benjamin.klimczak@arm.com> | 2023-02-08 15:17:12 +0000 |
| commit | 09ecc5c8acb758e8def33155feb746a34dd7b560 (patch) | |
| tree | 65c39a7f7929b745b9c5a31ab48bb4c6e97cb3ea /src/mlia/target/tosa/advisor.py | |
| parent | 6fbcffb4ee039438a409fbc92e38fa5d1d118833 (diff) | |
| download | mlia-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/tosa/advisor.py')
| -rw-r--r-- | src/mlia/target/tosa/advisor.py | 6 |
1 files changed, 4 insertions, 2 deletions
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": { |