ttnn-visualizer 0.47.0__py3-none-any.whl → 0.48.0__py3-none-any.whl

ttnn_visualizer/models.py

@@ -78,6 +78,7 @@ class Buffer(SerializeableDataclass):
     address: int
     max_size_per_bank: int
     buffer_type: BufferType
+    buffer_layout: Optional[int] = None
 
 
 @dataclasses.dataclass

ttnn_visualizer/queries.py

@@ -146,7 +146,7 @@ class DatabaseQueries:
     ) -> Generator[Buffer, None, None]:
         rows = self._query_table("buffers", filters)
         for row in rows:
-            yield Buffer(*row)
+            yield Buffer(*row[:6])
 
     def query_stack_traces(
         self, filters: Optional[Dict[str, Any]] = None

ttnn_visualizer/sockets.py

@@ -17,6 +17,7 @@ logger = getLogger(__name__)
 
 class Messages(object):
     FILE_TRANSFER_PROGRESS = "fileTransferProgress"
+    REPORT_GENERATED = "reportGenerated"
 
 
 class FileStatus(Enum):
@@ -27,6 +28,12 @@ class FileStatus(Enum):
     STARTED = "STARTED"
 
 
+class ExitStatus(Enum):
+    PASS = "PASS"
+    FAIL = "FAIL"
+    ERROR = "ERROR"
+
+
 @dataclass
 class FileProgress(SerializeableDataclass):
     current_file_name: str
@@ -40,6 +47,16 @@ class FileProgress(SerializeableDataclass):
         self.percent_of_current = round(self.percent_of_current, 2)
 
 
+@dataclass
+class ReportGenerated(SerializeableDataclass):
+    report_name: str
+    profiler_path: str | None = None
+    performance_path: str | None = None
+    exit_status: ExitStatus | None = None
+    message_type: str = "report_generated"
+    timestamp: str = field(default_factory=lambda: datetime.utcnow().isoformat())
+
+
 # For tracking connected clients subscriber ID
 tab_clients = {}
 
@@ -77,6 +94,20 @@ def emit_file_status(progress: FileProgress, instance_id=None):
         debounce_timer.start()
 
 
+def emit_report_generated(report_generated: ReportGenerated):
+    """Emit a report update notification to all connected clients."""
+    try:
+        if socketio is not None and hasattr(socketio, "emit"):
+            data = report_generated.to_dict()
+            socketio.emit(Messages.REPORT_GENERATED, data)
+            logger.info(
+                f"Report update notification sent: {report_generated.report_name}"
+            )
+    except NameError:
+        logger.warning("SocketIO not available - skipping report update notification")
+        pass  # Can silently pass since we know the NameError is from sockets being disabled
+
+
 def register_handlers(socketio_instance):
     global socketio
     socketio = socketio_instance

ttnn_visualizer/static/assets/{allPaths-OR2-IW-_.js → allPaths-DyhSRH--.js}

@@ -1 +1 @@
-import{I as s}from"./index-03c8d4Gh.js";import{I as r}from"./index-PKNBViIU.js";import{p as n,I as c}from"./index-CmA3KkTi.js";function p(t,a){const o=n(t);return a===c.STANDARD?s[o]:r[o]}export{s as IconSvgPaths16,r as IconSvgPaths20,p as getIconPaths};
+import{I as s}from"./index-03c8d4Gh.js";import{I as r}from"./index-PKNBViIU.js";import{p as n,I as c}from"./index-T8PVeTUn.js";function p(t,a){const o=n(t);return a===c.STANDARD?s[o]:r[o]}export{s as IconSvgPaths16,r as IconSvgPaths20,p as getIconPaths};

ttnn_visualizer/static/assets/allPathsLoader-DXd_rN1a.js

@@ -0,0 +1,2 @@
+const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-DyhSRH--.js","assets/index-03c8d4Gh.js","assets/index-PKNBViIU.js","assets/index-T8PVeTUn.js","assets/index-CZ0Uonoz.css"])))=>i.map(i=>d[i]);
+import{_ as e}from"./index-T8PVeTUn.js";const s=async(t,a)=>{const{getIconPaths:o}=await e(async()=>{const{getIconPaths:r}=await import("./allPaths-DyhSRH--.js");return{getIconPaths:r}},__vite__mapDeps([0,1,2,3,4]));return o(t,a)};export{s as allPathsLoader};