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

ttnn_visualizer/csv_queries.py

@@ -14,7 +14,7 @@ import pandas as pd
 import zstd
 from tt_perf_report import perf_report
 from ttnn_visualizer.exceptions import DataFormatError
-from ttnn_visualizer.models import Instance, RemoteConnection
+from ttnn_visualizer.models import Instance
 
 
 class LocalCSVQueryRunner:
@@ -433,18 +433,25 @@ class OpsPerformanceReportQueries:
         raw_csv = OpsPerformanceQueries.get_raw_csv(instance)
         csv_file = StringIO(raw_csv)
         csv_output_file = tempfile.mktemp(suffix=".csv")
-        perf_report.generate_perf_report(
-            csv_file,
-            cls.DEFAULT_SIGNPOST,
-            cls.DEFAULT_IGNORE_SIGNPOSTS,
-            cls.DEFAULT_MIN_PERCENTAGE,
-            cls.DEFAULT_ID_RANGE,
-            csv_output_file,
-            cls.DEFAULT_NO_ADVICE,
-            cls.DEFAULT_TRACING_MODE,
-            True,
-            True,
-        )
+
+        try:
+            perf_report.generate_perf_report(
+                csv_file,
+                cls.DEFAULT_SIGNPOST,
+                cls.DEFAULT_IGNORE_SIGNPOSTS,
+                cls.DEFAULT_MIN_PERCENTAGE,
+                cls.DEFAULT_ID_RANGE,
+                csv_output_file,
+                cls.DEFAULT_NO_ADVICE,
+                cls.DEFAULT_TRACING_MODE,
+                True,
+                True,
+                True,
+                True,
+                False,
+            )
+        except Exception as e:
+            raise DataFormatError(f"Error generating performance report: {e}") from e
 
         ops_perf_results = []
         ops_perf_results_reader = csv.DictReader(StringIO(raw_csv))

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/static/assets/allPaths-FfZB-oxi.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-A5WDVH6-.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-icaZ0D6C.js

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