diff options
Diffstat (limited to 'src/mlia/target/tosa/advisor.py')
-rw-r--r-- | src/mlia/target/tosa/advisor.py | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/src/mlia/target/tosa/advisor.py b/src/mlia/target/tosa/advisor.py index 4851113..0da44db 100644 --- a/src/mlia/target/tosa/advisor.py +++ b/src/mlia/target/tosa/advisor.py @@ -16,12 +16,16 @@ from mlia.core.context import ExecutionContext from mlia.core.data_analysis import DataAnalyzer from mlia.core.data_collection import DataCollector from mlia.core.events import Event +from mlia.core.metadata import MLIAMetadata +from mlia.core.metadata import ModelMetadata from mlia.target.tosa.advice_generation import TOSAAdviceProducer from mlia.target.tosa.config import TOSAConfiguration from mlia.target.tosa.data_analysis import TOSADataAnalyzer from mlia.target.tosa.data_collection import TOSAOperatorCompatibility from mlia.target.tosa.events import TOSAAdvisorStartedEvent from mlia.target.tosa.handlers import TOSAEventHandler +from mlia.target.tosa.metadata import TOSAMetadata +from mlia.target.tosa.reporters import MetadataDisplay class TOSAInferenceAdvisor(DefaultInferenceAdvisor): @@ -61,7 +65,15 @@ class TOSAInferenceAdvisor(DefaultInferenceAdvisor): target_profile = self.get_target_profile(context) return [ - TOSAAdvisorStartedEvent(model, TOSAConfiguration(target_profile)), + TOSAAdvisorStartedEvent( + model, + TOSAConfiguration(target_profile), + MetadataDisplay( + TOSAMetadata("tosa-checker"), + MLIAMetadata("mlia"), + ModelMetadata(model), + ), + ) ] |