diff options
Diffstat (limited to 'src/mlia/target/cortex_a/advisor.py')
| -rw-r--r-- | src/mlia/target/cortex_a/advisor.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/mlia/target/cortex_a/advisor.py b/src/mlia/target/cortex_a/advisor.py index a093784..5c077fb 100644 --- a/src/mlia/target/cortex_a/advisor.py +++ b/src/mlia/target/cortex_a/advisor.py @@ -36,11 +36,13 @@ class CortexAInferenceAdvisor(DefaultInferenceAdvisor): def get_collectors(self, context: Context) -> list[DataCollector]: """Return list of the data collectors.""" model = self.get_model(context) + target_profile = self.get_target_profile(context) + target_config = cast(CortexAConfiguration, profile(target_profile)) collectors: list[DataCollector] = [] if context.category_enabled(AdviceCategory.COMPATIBILITY): - collectors.append(CortexAOperatorCompatibility(model)) + collectors.append(CortexAOperatorCompatibility(model, target_config)) return collectors |