diff options
Diffstat (limited to 'src/mlia/target/ethos_u/advisor.py')
-rw-r--r-- | src/mlia/target/ethos_u/advisor.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/mlia/target/ethos_u/advisor.py b/src/mlia/target/ethos_u/advisor.py index b5932d0..edcfcfc 100644 --- a/src/mlia/target/ethos_u/advisor.py +++ b/src/mlia/target/ethos_u/advisor.py @@ -109,7 +109,9 @@ class EthosUInferenceAdvisor(DefaultInferenceAdvisor): def _get_target_config(self, context: Context) -> EthosUConfiguration: """Get target configuration.""" target_profile = self.get_target_profile(context) - return cast(EthosUConfiguration, profile(target_profile)) + target_config = cast(EthosUConfiguration, profile(target_profile)) + target_config.compiler_options.output_dir = context.output_dir # type: ignore + return target_config def _get_optimization_settings(self, context: Context) -> list[list[dict]]: """Get optimization settings.""" |