ttnn-visualizer 0.27.1__py3-none-any.whl → 0.28.0__py3-none-any.whl

ttnn_visualizer/serializers.py

@@ -1,14 +1,15 @@
 import dataclasses
 from collections import defaultdict
 from typing import List
+
+from ttnn_visualizer.models import BufferType, Operation, TensorComparisonRecord
+
+
 # SPDX-License-Identifier: Apache-2.0
 #
 # SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC
 
 
-from ttnn_visualizer.models import BufferType, Operation, TensorComparisonRecord
-
-
 def serialize_operations(
     inputs,
     operation_arguments,
@@ -39,7 +40,6 @@ def serialize_operations(
 
     results = []
     for operation in operations:
-
         inputs = inputs_dict[operation.operation_id]
         outputs = outputs_dict[operation.operation_id]
         arguments = [a.to_dict() for a in arguments_dict[operation.operation_id]]
@@ -207,7 +207,6 @@ def serialize_devices(devices):
 
 
 def serialize_operations_buffers(operations, buffers):
-
     buffer_dict = defaultdict(list)
     for b in buffers:
         buffer_dict[b.operation_id].append(b)
@@ -219,6 +218,15 @@ def serialize_operations_buffers(operations, buffers):
     return results
 
 
+def serialize_buffer(buffer):
+    return {
+        "buffer_type": buffer.buffer_type,
+        "device_id": buffer.device_id,
+        "size": buffer.max_size_per_bank,
+        "address": buffer.address,
+    }
+
+
 def serialize_tensors(
     tensors, producers_consumers, local_comparisons, global_comparisons
 ):

ttnn_visualizer/settings.py

@@ -19,7 +19,7 @@ class DefaultConfig(object):
 
     # Path Settings
     REPORT_DATA_DIRECTORY = Path(__file__).parent.absolute().joinpath("data")
-    VERSION = "0.26.0"
+    DB_VERSION = "0.26.0"  # App version when DB schema last changed
     LOCAL_DATA_DIRECTORY = Path(REPORT_DATA_DIRECTORY).joinpath("local")
     REMOTE_DATA_DIRECTORY = Path(REPORT_DATA_DIRECTORY).joinpath("remote")
     APPLICATION_DIR = os.path.abspath(os.path.join(__file__, "..", os.pardir))
@@ -37,7 +37,7 @@ class DefaultConfig(object):
 
     # SQL Alchemy Settings
     SQLALCHEMY_DATABASE_URI = (
-        f"sqlite:///{os.path.join(APPLICATION_DIR, f'ttnn_{VERSION}.db')}"
+        f"sqlite:///{os.path.join(APPLICATION_DIR, f'ttnn_{DB_VERSION}.db')}"
     )
     SQLALCHEMY_ENGINE_OPTIONS = {
         "pool_size": 10,  # Adjust pool size as needed (default is 5)

ttnn_visualizer/static/assets/{allPaths-C0lxNW4g.js → allPaths-DGdUWICT.js}

@@ -1 +1 @@
-import{I as n}from"./index-BVMreIQm.js";import{I as e}from"./index-Do7YB6C4.js";import{p as r,I as s}from"./index-BVO5CCjN.js";function I(o,t){var a=r(o);return t===s.STANDARD?n[a]:e[a]}function p(o){return r(o)}export{n as IconSvgPaths16,e as IconSvgPaths20,I as getIconPaths,p as iconNameToPathsRecordKey};
+import{I as n}from"./index-BVMreIQm.js";import{I as e}from"./index-Do7YB6C4.js";import{p as r,I as s}from"./index-CYZZ70nJ.js";function I(o,t){var a=r(o);return t===s.STANDARD?n[a]:e[a]}function p(o){return r(o)}export{n as IconSvgPaths16,e as IconSvgPaths20,I as getIconPaths,p as iconNameToPathsRecordKey};

ttnn_visualizer/static/assets/{allPathsLoader-DNFRLKF6.js → allPathsLoader-BSYo-ZTH.js}

@@ -1,2 +1,2 @@
-const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-C0lxNW4g.js","assets/index-BVMreIQm.js","assets/index-Do7YB6C4.js","assets/index-BVO5CCjN.js","assets/index-BTfoVg9a.css"])))=>i.map(i=>d[i]);
-import{_ as o,a as n,b as i}from"./index-BVO5CCjN.js";var _=function(e,a){return o(void 0,void 0,void 0,function(){var t;return n(this,function(r){switch(r.label){case 0:return[4,i(()=>import("./allPaths-C0lxNW4g.js"),__vite__mapDeps([0,1,2,3,4]))];case 1:return t=r.sent().getIconPaths,[2,t(e,a)]}})})};export{_ as allPathsLoader};
+const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-DGdUWICT.js","assets/index-BVMreIQm.js","assets/index-Do7YB6C4.js","assets/index-CYZZ70nJ.js","assets/index-DWVfq2z5.css"])))=>i.map(i=>d[i]);
+import{_ as o,a as n,b as i}from"./index-CYZZ70nJ.js";var _=function(e,a){return o(void 0,void 0,void 0,function(){var t;return n(this,function(r){switch(r.label){case 0:return[4,i(()=>import("./allPaths-DGdUWICT.js"),__vite__mapDeps([0,1,2,3,4]))];case 1:return t=r.sent().getIconPaths,[2,t(e,a)]}})})};export{_ as allPathsLoader};