ttnn-visualizer 0.45.0__py3-none-any.whl → 0.47.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/decorators.py

@@ -90,13 +90,13 @@ def remote_exception_handler(func):
             current_app.logger.error(f"File not found: {str(err)}")
             raise RemoteConnectionException(
                 status=ConnectionTestStates.FAILED,
-                message=f"Unable to open path {connection.path}: {str(err)}",
+                message=f"Unable to open path: {str(err)}",
             )
         except NoProjectsException as err:
             current_app.logger.error(f"No projects: {str(err)}")
             raise RemoteConnectionException(
                 status=ConnectionTestStates.FAILED,
-                message=f"No projects found at remote location: {connection.path}",
+                message=f"No projects found at remote location: {str(err)}",
             )
         except NoValidConnectionsError as err:
             current_app.logger.warning(
@@ -124,7 +124,7 @@ def remote_exception_handler(func):
                 status=ConnectionTestStates.FAILED, message=message
             )
         except IOError as err:
-            message = f"Error opening remote folder {connection.path}: {str(err)}"
+            message = f"Error opening remote folder: {str(err)}"
             if "Name or service not known" in str(err):
                 message = f"Unable to connect to {connection.host} - check hostname"
             raise RemoteConnectionException(

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

@@ -4,7 +4,6 @@
 
 import json
 import logging
-import re
 import subprocess
 import time
 from pathlib import Path
@@ -16,9 +15,7 @@ from flask import current_app
 from ttnn_visualizer.decorators import remote_exception_handler
 from ttnn_visualizer.enums import ConnectionTestStates
 from ttnn_visualizer.exceptions import (
-    AuthenticationException,
     NoProjectsException,
-    NoValidConnectionsError,
     RemoteConnectionException,
     SSHException,
 )
@@ -623,14 +620,30 @@ def read_remote_file(
 
 @remote_exception_handler
 def check_remote_path_for_reports(remote_connection):
-    """Check the remote path for config files."""
-    remote_config_paths = find_folders_by_files(
+    remote_profiler_paths = find_folders_by_files(
         remote_connection, remote_connection.profilerPath, [TEST_CONFIG_FILE]
     )
-    if not remote_config_paths:
+
+    remote_performance_paths = find_folders_by_files(
+        remote_connection, remote_connection.performancePath, [TEST_PROFILER_FILE]
+    )
+
+    errors = []
+    if not remote_profiler_paths and remote_connection.profilerPath:
+        errors.append(
+            f"No matching profiler projects found: {remote_connection.profilerPath}"
+        )
+    if not remote_performance_paths and remote_connection.performancePath:
+        errors.append(
+            f"No matching performance projects found: {remote_connection.performancePath}"
+        )
+
+    if errors:
         raise NoProjectsException(
-            message="No projects found at path", status=ConnectionTestStates.FAILED
+            message="; ".join(errors),
+            status=ConnectionTestStates.FAILED,
         )
+
     return True
 
 
@@ -648,8 +661,10 @@ def check_remote_path_exists(remote_connection: RemoteConnection, path_key: str)
             # Directory does not exist or is inaccessible
             if path_key == "performancePath":
                 message = "Performance directory does not exist or cannot be accessed"
-            else:
+            if path_key == "profilerPath":
                 message = "Profiler directory does not exist or cannot be accessed"
+            else:
+                message = f"Remote path '{path}' does not exist or cannot be accessed"
 
             logger.error(message)
             raise RemoteConnectionException(

ttnn_visualizer/static/assets/allPaths-OR2-IW-_.js

@@ -0,0 +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};

ttnn_visualizer/static/assets/allPathsLoader-xRXweacG.js

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