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

ttnn_visualizer/app.py

@@ -25,6 +25,7 @@ from ttnn_visualizer.exceptions import (
 )
 from ttnn_visualizer.instances import create_instance_from_local_paths
 from ttnn_visualizer.settings import Config, DefaultConfig
+from ttnn_visualizer.utils import create_path_resolver
 from werkzeug.debug import DebuggedApplication
 from werkzeug.middleware.proxy_fix import ProxyFix
 
@@ -78,6 +79,7 @@ def create_app(settings_override=None):
             js_config = {
                 "SERVER_MODE": app.config["SERVER_MODE"],
                 "BASE_PATH": app.config["BASE_PATH"],
+                "TT_METAL_HOME": app.config["TT_METAL_HOME"],
             }
             js = f"window.TTNN_VISUALIZER_CONFIG = {json.dumps(js_config)};"
 
@@ -192,6 +194,9 @@ def parse_args():
     parser.add_argument(
         "--performance-path", help="Specify a performance path", default=None
     )
+    parser.add_argument(
+        "--tt-metal-home", help="Specify a TT-Metal home path", default=None
+    )
     return parser.parse_args()
 
 
@@ -220,10 +225,40 @@ def main():
 
         instance_id = session.instance_id
 
+    if args.tt_metal_home:
+        config.TT_METAL_HOME = args.tt_metal_home
+
+    # Display mode information
+    app = create_app()
+    with app.app_context():
+        resolver = create_path_resolver(app)
+        mode_info = resolver.get_mode_info()
+
+        if mode_info["mode"] == "tt_metal":
+            print(
+                "🚀 TT-METAL MODE: Working directly with tt-metal generated directory"
+            )
+            print(f"   TT_METAL_HOME: {mode_info['tt_metal_home']}")
+            print(f"   Profiler reports: {mode_info['profiler_base']}")
+            print(f"   Performance reports: {mode_info['performance_base']}")
+
+            # Validate setup
+            is_valid, message = resolver.validate_tt_metal_setup()
+            if is_valid:
+                print(f"   ✓ {message}")
+            else:
+                print(f"   ⚠️  Warning: {message}")
+        else:
+            print(
+                "📁 UPLOAD/SYNC MODE: Using local data directory for uploaded/synced reports"
+            )
+            print(f"   Local directory: {mode_info['local_dir']}")
+            print(f"   Remote directory: {mode_info['remote_dir']}")
+
     # Check if DEBUG environment variable is set
     debug_mode = os.environ.get("DEBUG", "false").lower() == "true"
     if config.PRINT_ENV:
-        print("ENVIRONMENT:")
+        print("\nENVIRONMENT:")
         for key, value in config.to_dict().items():
             print(f"{key}={value}")
 

ttnn_visualizer/csv_queries.py

@@ -1,6 +1,7 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2025 Tenstorrent Inc.
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
+
 import csv
 import json
 import os

ttnn_visualizer/file_uploads.py

@@ -1,6 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 #
-# SPDX-FileCopyrightText: © 2025 Tenstorrent Inc.
+# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
 import logging
 import os

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/settings.py

@@ -42,6 +42,7 @@ class DefaultConfig(object):
     APPLICATION_DIR = os.path.abspath(os.path.join(__file__, "..", os.pardir))
     APP_DATA_DIRECTORY = os.getenv("APP_DATA_DIRECTORY", APPLICATION_DIR)
     STATIC_ASSETS_DIR = Path(APPLICATION_DIR).joinpath("ttnn_visualizer", "static")
+    TT_METAL_HOME = os.getenv("TT_METAL_HOME", None)
     SEND_FILE_MAX_AGE_DEFAULT = 0
 
     LAUNCH_BROWSER_ON_START = str_to_bool(os.getenv("LAUNCH_BROWSER_ON_START", "true"))

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-esBqnTg5.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-BANm1CMY.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};