ttnn-visualizer 0.69.0__py3-none-any.whl → 0.70.0__py3-none-any.whl

ttnn_visualizer/csv_queries.py

@@ -476,6 +476,7 @@ class OpsPerformanceReportQueries:
         stack_by_in0 = kwargs.get("stack_by_in0", cls.DEFAULT_NO_STACK_BY_IN0)
         no_host_ops = kwargs.get("hide_host_ops", cls.DEFAULT_NO_HOST_OPS)
         merge_devices = kwargs.get("merge_devices", cls.DEFAULT_MERGE_DEVICES)
+        tracing_mode = kwargs.get("tracing_mode", cls.DEFAULT_TRACING_MODE)
 
         if start_signpost or end_signpost:
             ignore_signposts = False
@@ -495,7 +496,7 @@ class OpsPerformanceReportQueries:
                 cls.DEFAULT_ID_RANGE,
                 csv_output_file,
                 cls.DEFAULT_NO_ADVICE,
-                cls.DEFAULT_TRACING_MODE,
+                tracing_mode,
                 cls.DEFAULT_RAW_OP_CODES,
                 no_host_ops,
                 cls.DEFAULT_NO_STACKED_REPORT,

ttnn_visualizer/models.py

@@ -50,7 +50,6 @@ class Device(SerializeableDataclass):
     address_at_first_l1_cb_buffer: int
     num_banks_per_storage_core: int
     num_compute_cores: int
-    num_storage_cores: int
     total_l1_memory: int
     total_l1_for_tensors: int
     total_l1_for_interleaved_buffers: int

ttnn_visualizer/queries.py

@@ -2,6 +2,7 @@
 #
 # SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
 
+import dataclasses
 import sqlite3
 from pathlib import Path
 from typing import Any, Dict, Generator, List, Optional, Union
@@ -88,14 +89,24 @@ class DatabaseQueries:
 
         return bool(rows)
 
+    def _get_table_columns(self, table_name: str) -> List[str]:
+        """
+        Gets the list of column names for a table.
+        """
+        query = f"PRAGMA table_info({table_name})"
+        rows = self.query_runner.execute_query(query)
+        return [row[1] for row in rows]  # row[1] is the column name
+
     def _query_table(
         self,
         table_name: str,
         filters: Optional[Dict[str, Union[Any, List[Any]]]] = None,
         additional_conditions: Optional[str] = None,
         additional_params: Optional[List[Any]] = None,
+        columns: Optional[List[str]] = None,
     ) -> List[Any]:
-        query = f"SELECT * FROM {table_name} WHERE 1=1"
+        columns_str = ", ".join(columns) if columns else "*"
+        query = f"SELECT {columns_str} FROM {table_name} WHERE 1=1"
         params = []
 
         if filters:
@@ -285,7 +296,19 @@ class DatabaseQueries:
     def query_devices(
         self, filters: Optional[Dict[str, Any]] = None
     ) -> Generator[Device, None, None]:
-        rows = self._query_table("devices", filters)
+        # Get the expected Device model field names in order
+        device_fields = [field.name for field in dataclasses.fields(Device)]
+
+        # Get all columns from the devices table
+        all_columns = self._get_table_columns("devices")
+
+        # Filter out num_storage_cores if it exists (for backwards compatibility)
+        # and ensure columns are in the order expected by the Device model
+        available_columns = set(all_columns) - {"num_storage_cores"}
+        columns = [col for col in device_fields if col in available_columns]
+
+        rows = self._query_table("devices", filters, columns=columns)
+
         for row in rows:
             yield Device(*row)
 

ttnn_visualizer/sftp_operations.py

@@ -7,7 +7,6 @@ import logging
 import subprocess
 import time
 from pathlib import Path
-from stat import S_ISDIR
 from threading import Thread
 from typing import List, Optional
 

ttnn_visualizer/static/assets/allPaths-DrpaDRaJ.js

@@ -0,0 +1 @@
+import{I as s}from"./index-BmDjQHI0.js";import{I as r}from"./index-CzNKtOwn.js";import{p as n,I as c}from"./index-SMzJpwbG.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-C78guN8s.js

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