ttnn-visualizer 0.33.1__py3-none-any.whl → 0.34.0__py3-none-any.whl

ttnn_visualizer/csv_queries.py

@@ -482,6 +482,8 @@ class OpsPerformanceQueries:
 
         # Find the latest file with the correct prefix
         perf_files = list(
+            performance_path.glob(f"{OpsPerformanceQueries.PERF_RESULTS_PREFIX}.csv")
+        ) + list(
             performance_path.glob(f"{OpsPerformanceQueries.PERF_RESULTS_PREFIX}_*.csv")
         )
         if not perf_files:

ttnn_visualizer/file_uploads.py

@@ -4,11 +4,12 @@
 
 from pathlib import Path
 import logging
+import re
 
 logger = logging.getLogger(__name__)
 
 
-def validate_files(files, required_files, pattern=None):
+def validate_files(files, required_files, pattern=None, folder_name=None):
     """Validate uploaded files against required file names and an optional pattern."""
     found_files = set()
 
@@ -19,7 +20,7 @@ def validate_files(files, required_files, pattern=None):
             pattern and file_path.name.startswith(pattern)
         ):
             found_files.add(file_path.name)
-            if len(file_path.parents) != 2:
+            if not folder_name and len(file_path.parents) != 2:
                 logger.warning(
                     f"File {file.filename} is not under a single parent folder."
                 )
@@ -43,25 +44,25 @@ def extract_profiler_name(files):
     unsplit_profiler_name = str(files[0].filename)
     return unsplit_profiler_name.split("/")[0]
 
+
 def extract_npe_name(files):
     if not files:
         return None
 
-    file_path = Path(files[0].filename)
-    return file_path.stem
+    return re.sub(r"\.(json|npeviz\.zst)$", "", files[0].filename)
 
 
 def save_uploaded_files(
     files,
     target_directory,
-    profiler_name=None,
+    folder_name=None,
 ):
     """
     Save uploaded files to the target directory.
 
     :param files: List of files to be saved.
     :param target_directory: The base directory for saving the files.
-    :param profiler_name: The report name to use for the directory.
+    :param folder_name: The name to use for the directory.
     """
     for file in files:
         current_file_name = str(file.filename)
@@ -69,7 +70,10 @@ def save_uploaded_files(
 
         file_path = Path(current_file_name)
 
-        destination_file = Path(target_directory).joinpath(str(file_path))
+        if folder_name:
+            destination_file = Path(target_directory) / folder_name / str(file_path)
+        else:
+            destination_file = Path(target_directory) / str(file_path)
 
         logger.info(f"Writing file to {destination_file}")
 

ttnn_visualizer/requirements.txt

@@ -18,6 +18,7 @@ build
 PyYAML==6.0.2
 python-dotenv==1.0.1
 tt-perf-report==1.0.6
+zstd==1.5.7.0
 
 # Dev dependencies
 mypy

ttnn_visualizer/settings.py

@@ -53,7 +53,7 @@ class DefaultConfig(object):
     GUNICORN_WORKER_CLASS = os.getenv("GUNICORN_WORKER_CLASS", "gevent")
     GUNICORN_WORKERS = os.getenv("GUNICORN_WORKERS", "1")
     PORT = os.getenv("PORT", "8000")
-    HOST = "localhost"
+    HOST = os.getenv("HOST", "localhost")
     DEV_SERVER_PORT = "5173"
     DEV_SERVER_HOST = "localhost"
 

ttnn_visualizer/static/assets/{allPaths-DmImLtgS.js → allPaths-D0uOIsnD.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-Dy2mdDFG.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-CHS7htir.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-B9-8qUsI.js → allPathsLoader-B1iveKGB.js}

@@ -1,2 +1,2 @@
-const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-DmImLtgS.js","assets/index-BVMreIQm.js","assets/index-Do7YB6C4.js","assets/index-Dy2mdDFG.js","assets/index-Dhl0up3F.css"])))=>i.map(i=>d[i]);
-import{_ as o,a as n,b as i}from"./index-Dy2mdDFG.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-DmImLtgS.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-D0uOIsnD.js","assets/index-BVMreIQm.js","assets/index-Do7YB6C4.js","assets/index-CHS7htir.js","assets/index-CJJgzCye.css"])))=>i.map(i=>d[i]);
+import{_ as o,a as n,b as i}from"./index-CHS7htir.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-D0uOIsnD.js"),__vite__mapDeps([0,1,2,3,4]))];case 1:return t=r.sent().getIconPaths,[2,t(e,a)]}})})};export{_ as allPathsLoader};