ttnn-visualizer 0.27.3__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/sessions.py

@@ -271,17 +271,17 @@ def create_random_instance_id():
 
 
 def create_instance_from_local_paths(report_path, profiler_path):
-    _report_path = Path(report_path) if report_path else None
-    _profiler_path = Path(profiler_path) if profiler_path else None
+    _report_path = Path(report_path)
+    _profiler_path = Path(profiler_path)
 
-    if _report_path and (not _report_path.exists() or not _report_path.is_dir()):
+    if not _report_path.exists() or not _report_path.is_dir():
         raise InvalidReportPath()
 
-    if _profiler_path and (not _profiler_path.exists() or not _profiler_path.is_dir()):
+    if not _profiler_path.exists() or not _profiler_path.is_dir():
         raise InvalidProfilerPath()
 
-    report_name = _report_path.parts[-1] if _report_path and len(_report_path.parts) > 2 else ""
-    profile_name = _profiler_path.parts[-1] if _profiler_path and len(_profiler_path.parts) > 2 else ""
+    report_name = _report_path.parts[-1] if len(_report_path.parts) > 2 else ""
+    profile_name = _profiler_path.parts[-1] if len(_profiler_path.parts) > 2 else ""
     session_data = InstanceTable(
         instance_id=create_random_instance_id(),
         active_report={
@@ -289,8 +289,8 @@ def create_instance_from_local_paths(report_path, profiler_path):
             "profile_name": profile_name,
             "npe_name": None,
         },
-        report_path=f"{_report_path}/db.sqlite" if report_path else None,
-        profiler_path=profiler_path if profiler_path else None,
+        report_path=f"{report_path}/db.sqlite",
+        profiler_path=profiler_path,
         remote_connection=None,
         remote_folder=None,
         remote_profile_folder=None,

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-DhoJy5jM.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-BME2kYRR.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-B4SALZ6P.js → allPathsLoader-BSYo-ZTH.js}

@@ -1,2 +1,2 @@
-const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-DhoJy5jM.js","assets/index-BVMreIQm.js","assets/index-Do7YB6C4.js","assets/index-BME2kYRR.js","assets/index-BTfoVg9a.css"])))=>i.map(i=>d[i]);
-import{_ as o,a as n,b as i}from"./index-BME2kYRR.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-DhoJy5jM.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};