ttnn-visualizer 0.49.0__py3-none-any.whl → 0.64.0__py3-none-any.whl

ttnn_visualizer/serializers.py

@@ -19,6 +19,7 @@ def serialize_operations(
     devices,
     producers_consumers,
     device_operations,
+    error_records=None,
 ):
     tensors_dict = {t.tensor_id: t for t in tensors}
     device_operations_dict = {
@@ -29,6 +30,11 @@ def serialize_operations(
 
     stack_traces_dict = {st.operation_id: st.stack_trace for st in stack_traces}
 
+    errors_dict = {}
+    if error_records:
+        for error in error_records:
+            errors_dict[error.operation_id] = error.to_nested_dict()
+
     arguments_dict = defaultdict(list)
     for argument in operation_arguments:
         arguments_dict[argument.operation_id].append(argument)
@@ -49,6 +55,8 @@ def serialize_operations(
         )
         id = operation_data.pop("operation_id", None)
 
+        error_data = errors_dict.get(operation.operation_id)
+
         results.append(
             {
                 **operation_data,
@@ -58,6 +66,7 @@ def serialize_operations(
                 "arguments": arguments,
                 "inputs": inputs,
                 "outputs": outputs,
+                "error": error_data,
             }
         )
     return results
@@ -144,6 +153,7 @@ def serialize_operation(
     devices,
     producers_consumers,
     device_operations,
+    error_record=None,
 ):
     tensors_dict = {t.tensor_id: t for t in tensors}
     comparisons = comparisons_by_tensor_id(
@@ -176,6 +186,9 @@ def serialize_operation(
             device_operations_data = do.captured_graph
             break
 
+    # Convert error record to nested dict if it exists (excludes operation_id and operation_name)
+    error_data = error_record.to_nested_dict() if error_record else None
+
     return {
         **operation_data,
         "id": id,
@@ -186,18 +199,30 @@ def serialize_operation(
         "arguments": arguments_data,
         "inputs": inputs_data or [],
         "outputs": outputs_data or [],
+        "error": error_data,
     }
 
 
 def serialize_operation_buffers(operation: Operation, operation_buffers):
-    buffer_data = [b.to_dict() for b in operation_buffers]
-    for b in buffer_data:
-        b.pop("operation_id")
-        b.update({"size": b.pop("max_size_per_bank")})
+    buffer_data = []
+    for b in operation_buffers:
+        buffer_dict = {
+            "device_id": b.device_id,
+            "address": b.address,
+            "buffer_type": (
+                b.buffer_type.value
+                if hasattr(b.buffer_type, "value")
+                else b.buffer_type
+            ),
+            "buffer_layout": b.buffer_layout,
+            "size": b.max_size_per_bank,
+        }
+        buffer_data.append(buffer_dict)
+
     return {
         "id": operation.operation_id,
         "name": operation.name,
-        "buffers": list(buffer_data),
+        "buffers": buffer_data,
     }
 
 
@@ -206,14 +231,33 @@ def serialize_devices(devices):
 
 
 def serialize_operations_buffers(operations, buffers):
-    buffer_dict = defaultdict(list)
+    # Pre-serialize all buffers once using optimized method with defaultdict
+    serialized_buffers = defaultdict(list)
     for b in buffers:
-        buffer_dict[b.operation_id].append(b)
+        buffer_dict = {
+            "device_id": b.device_id,
+            "address": b.address,
+            "buffer_type": (
+                b.buffer_type.value
+                if hasattr(b.buffer_type, "value")
+                else b.buffer_type
+            ),
+            "buffer_layout": b.buffer_layout,
+            "size": b.max_size_per_bank,
+        }
+        serialized_buffers[b.operation_id].append(buffer_dict)
 
     results = []
     for operation in operations:
-        operation_buffers = buffer_dict.get(operation.operation_id, [])
-        results.append(serialize_operation_buffers(operation, operation_buffers))
+        operation_buffers = serialized_buffers[operation.operation_id]
+        results.append(
+            {
+                "id": operation.operation_id,
+                "name": operation.name,
+                "buffers": operation_buffers,
+            }
+        )
+
     return results
 
 

ttnn_visualizer/settings.py

@@ -6,7 +6,12 @@ import os
 from pathlib import Path
 
 from dotenv import load_dotenv
-from ttnn_visualizer.utils import str_to_bool
+from sqlalchemy.pool import NullPool
+from ttnn_visualizer.utils import (
+    get_app_data_directory,
+    is_running_in_container,
+    str_to_bool,
+)
 
 load_dotenv()
 
@@ -40,9 +45,13 @@ class DefaultConfig(object):
     PERFORMANCE_DIRECTORY_NAME = "performance-reports"
     NPE_DIRECTORY_NAME = "npe-reports"
     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)
+    APP_DATA_DIRECTORY = os.getenv(
+        "APP_DATA_DIRECTORY",
+        get_app_data_directory(TT_METAL_HOME, APPLICATION_DIR),
+    )
+
+    STATIC_ASSETS_DIR = Path(APPLICATION_DIR).joinpath("ttnn_visualizer", "static")
     SEND_FILE_MAX_AGE_DEFAULT = 0
 
     LAUNCH_BROWSER_ON_START = str_to_bool(os.getenv("LAUNCH_BROWSER_ON_START", "true"))
@@ -55,21 +64,26 @@ class DefaultConfig(object):
     USE_WEBSOCKETS = str_to_bool(os.getenv("USE_WEBSOCKETS", "true"))
 
     # SQL Alchemy Settings
-    SQLALCHEMY_DATABASE_URI = (
-        f"sqlite:///{os.path.join(APP_DATA_DIRECTORY, f'ttnn_{DB_VERSION}.db')}"
-    )
+    # Build database path - use absolute path to avoid any ambiguity
+    _db_file_path = str(Path(APP_DATA_DIRECTORY) / f"ttnn_{DB_VERSION}.db")
+    SQLALCHEMY_DATABASE_URI = f"sqlite:///{_db_file_path}"
     SQLALCHEMY_ENGINE_OPTIONS = {
-        "pool_size": 10,  # Adjust pool size as needed (default is 5)
-        "max_overflow": 20,  # Allow overflow of the pool size if necessary
-        "pool_timeout": 30,  # Timeout in seconds before giving up on getting a connection
+        # SQLite-specific settings for multi-process/worker environments
+        # NullPool: Each worker gets its own connection, avoiding file locking issues
+        # This is critical for gunicorn's multi-worker mode with SQLite
+        "poolclass": NullPool,
+        "connect_args": {
+            "check_same_thread": False,  # Allow SQLite to be used across threads in gevent
+        },
     }
     SQLALCHEMY_TRACK_MODIFICATIONS = False
 
     # Gunicorn settings
     GUNICORN_WORKER_CLASS = os.getenv("GUNICORN_WORKER_CLASS", "gevent")
     GUNICORN_WORKERS = os.getenv("GUNICORN_WORKERS", "1")
+    GUNICORN_TIMEOUT = os.getenv("GUNICORN_TIMEOUT", "60")
     PORT = os.getenv("PORT", "8000")
-    HOST = os.getenv("HOST", "localhost")
+    HOST = os.getenv("HOST", "0.0.0.0" if is_running_in_container() else "localhost")
     DEV_SERVER_PORT = "5173"
     DEV_SERVER_HOST = "localhost"
 

ttnn_visualizer/ssh_client.py

@@ -210,9 +210,7 @@ class SSHClient:
                     detail=raw_error,
                 )
 
-    def read_file(
-        self, remote_path: Union[str, Path], timeout: int = 30
-    ) -> Optional[bytes]:
+    def read_file(self, remote_path: str, timeout: int = 30) -> Optional[bytes]:
         """
         Read a remote file using SSH cat command.
 
@@ -229,7 +227,6 @@ class SSHClient:
             return result.encode("utf-8")
         except SSHException as e:
             if "No such file" in str(e) or "cannot open" in str(e):
-                logger.error(f"File not found or cannot be read: {path}")
                 return None
             raise
 

ttnn_visualizer/static/assets/allPaths-DWjqav_8.js

@@ -0,0 +1 @@
+import{I as s}from"./index-voJy5fZe.js";import{I as r}from"./index-BZITDwoa.js";import{p as n,I as c}from"./index-DDrUX09k.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-B0eRT9aL.js

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