ttnn-visualizer 0.47.0__py3-none-any.whl → 0.49.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/pytest_plugin.py

@@ -0,0 +1,93 @@
+# SPDX-License-Identifier: Apache-2.0
+#
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
+import json
+import os
+import urllib
+
+try:
+    import ttnn
+    from loguru import logger
+    from tt_metal.tools.profiler.common import PROFILER_OUTPUT_DIR
+except ImportError:
+    raise Exception("TT-Metal environment not found")
+
+
+def pytest_runtest_makereport(item, call):
+    if call.when != "call":
+        return
+
+    webhook_url = os.environ.get("TTNN_VISUALIZER_WEBHOOK_URL", "").strip()
+
+    if not webhook_url:
+        webhook_url = "http://localhost:8000/api/notify"
+
+    status_str = _get_test_status_from_call(call)
+    _notify_visualizer_webhook(status_str, webhook_url)
+
+
+def _notify_visualizer_webhook(status_str: str, webhook_url: str) -> None:
+    """POST JSON to the visualizer webhook endpoint."""
+    payload = {
+        "report_name": _get_ttnn_report_name(),
+        "profiler_path": _get_profiler_path(),
+        "performance_path": _get_performance_path(),
+        "status": status_str,
+    }
+
+    data = json.dumps(payload).encode("utf-8")
+    headers = {"Content-Type": "application/json"}
+    try:
+        request = urllib.request.Request(
+            webhook_url, data=data, headers=headers, method="POST"
+        )
+        logger.info(f"Posting report to TTNN-Visualizer: {payload}")
+        urllib.request.urlopen(request, timeout=2)
+    except urllib.error.HTTPError as error:
+        logger.error(
+            f"Error posting report to TTNN-Visualizer: {error.status} {error.reason}"
+        )
+    except Exception as err:
+        logger.error(err)
+
+
+def _get_test_status_from_call(call) -> str:
+    """Determine test status from the call object."""
+    if call.excinfo is None:
+        return "PASS"
+    elif call.excinfo[0] == AssertionError:
+        return "FAIL"
+    else:
+        return "ERROR"
+
+
+def _get_ttnn_report_name() -> str:
+    name = ttnn.CONFIG.report_name
+    return str(name) if name else "Not Available"
+
+
+def _get_profiler_path() -> str:
+    tt_metal_home = os.getenv("TT_METAL_HOME", "")
+    return f"{tt_metal_home}/{ttnn.CONFIG.report_path}"
+
+
+def _get_performance_path() -> str:
+    output_dir = PROFILER_OUTPUT_DIR
+
+    if os.getenv("TT_METAL_DEVICE_PROFILER", None) != "1":
+        return None
+
+    try:
+        directories = [item for item in output_dir.iterdir() if item.is_dir()]
+
+        if not directories:
+            return None
+
+        # Find the newest directory by modification time
+        newest_dir = max(directories, key=lambda d: d.stat().st_mtime)
+        return str(newest_dir)
+
+    except (OSError, ValueError):
+        return None
+
+    return str(output_dir)

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-G_CNx_x1.js

@@ -0,0 +1 @@
+import{I as s}from"./index-CnPrfHYh.js";import{I as r}from"./index-Cnc1EkDo.js";import{p as n,I as c}from"./index-DVrPLQJ7.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-s_Yfmxfp.js

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