ttnn-visualizer 0.52.1__py3-none-any.whl → 0.53.0__py3-none-any.whl

ttnn_visualizer/app.py

@@ -265,6 +265,8 @@ def main():
 
     gunicorn_args = [
         "gunicorn",
+        "-t",
+        config.GUNICORN_TIMEOUT,
         "-k",
         config.GUNICORN_WORKER_CLASS,
         "-w",

ttnn_visualizer/csv_queries.py

@@ -437,6 +437,10 @@ class OpsPerformanceReportQueries:
     DEFAULT_ID_RANGE = None
     DEFAULT_NO_ADVICE = False
     DEFAULT_TRACING_MODE = False
+    DEFAULT_RAW_OP_CODES = True
+    DEFAULT_NO_HOST_OPS = False
+    DEFAULT_NO_STACKED_REPORT = False
+    DEFAULT_NO_STACK_BY_IN0 = True
 
     @classmethod
     def generate_report(cls, instance):
@@ -457,10 +461,10 @@ class OpsPerformanceReportQueries:
                 csv_output_file,
                 cls.DEFAULT_NO_ADVICE,
                 cls.DEFAULT_TRACING_MODE,
-                True,
-                True,
-                False,
-                True,  # no_stack_by_in0 - need to ask what this is
+                cls.DEFAULT_RAW_OP_CODES,
+                cls.DEFAULT_NO_HOST_OPS,
+                cls.DEFAULT_NO_STACKED_REPORT,
+                cls.DEFAULT_NO_STACK_BY_IN0,
                 csv_stacked_output_file,
             )
         except Exception as e:

ttnn_visualizer/serializers.py

@@ -190,14 +190,25 @@ def serialize_operation(
 
 
 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 +217,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

@@ -68,6 +68,7 @@ class DefaultConfig(object):
     # 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")
     DEV_SERVER_PORT = "5173"

ttnn_visualizer/static/assets/{allPaths-DOFQJGIv.js → allPaths-BlkeJGrc.js}

@@ -1 +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-WayH7MCF.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-CnPrfHYh.js";import{I as r}from"./index-Cnc1EkDo.js";import{p as n,I as c}from"./index-ASyuPW18.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--1bCnHgW.js

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