aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/devices/tosa/events.py
diff options
context:
space:
mode:
authorDmitrii Agibov <dmitrii.agibov@arm.com>2022-07-21 14:06:50 +0100
committerBenjamin Klimczak <benjamin.klimczak@arm.com>2022-08-19 10:23:23 +0100
commit664d8c55609253e68d153a91514c8fefa00557b1 (patch)
tree4b2a0ecaf30e9151d6b971a24fa6c6104884896f /src/mlia/devices/tosa/events.py
parenta8ee1aee3e674c78a77801d1bf2256881ab6b4b9 (diff)
downloadmlia-664d8c55609253e68d153a91514c8fefa00557b1.tar.gz
MLIA-549 Integrate TOSA checker into MLIA
- Add new module for TOSA - Add advisor workflow components - Use TOSA checker for getting operators compatibility information Change-Id: I769e5e2a84e15779658f0895b4a347384def63bf
Diffstat (limited to 'src/mlia/devices/tosa/events.py')
-rw-r--r--src/mlia/devices/tosa/events.py24
1 files changed, 24 insertions, 0 deletions
diff --git a/src/mlia/devices/tosa/events.py b/src/mlia/devices/tosa/events.py
new file mode 100644
index 0000000..ceaba57
--- /dev/null
+++ b/src/mlia/devices/tosa/events.py
@@ -0,0 +1,24 @@
+# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""TOSA advisor events."""
+from dataclasses import dataclass
+from pathlib import Path
+
+from mlia.core.events import Event
+from mlia.core.events import EventDispatcher
+from mlia.devices.tosa.config import TOSAConfiguration
+
+
+@dataclass
+class TOSAAdvisorStartedEvent(Event):
+ """Event with TOSA advisor parameters."""
+
+ model: Path
+ device: TOSAConfiguration
+
+
+class TOSAAdvisorEventHandler(EventDispatcher):
+ """Event handler for the TOSA inference advisor."""
+
+ def on_tosa_advisor_started(self, event: TOSAAdvisorStartedEvent) -> None:
+ """Handle TOSAAdvisorStartedEvent event."""