ttnn-visualizer 0.51.0__py3-none-any.whl → 0.52.0__py3-none-any.whl

ttnn_visualizer/app.py

@@ -80,6 +80,7 @@ def create_app(settings_override=None):
                 "SERVER_MODE": app.config["SERVER_MODE"],
                 "BASE_PATH": app.config["BASE_PATH"],
                 "TT_METAL_HOME": app.config["TT_METAL_HOME"],
+                "REPORT_DATA_DIRECTORY": str(app.config["REPORT_DATA_DIRECTORY"]),
             }
             js = f"window.TTNN_VISUALIZER_CONFIG = {json.dumps(js_config)};"
 

ttnn_visualizer/csv_queries.py

@@ -100,7 +100,6 @@ class NPEQueries:
 
     @staticmethod
     def get_npe_manifest(instance: Instance):
-
         file_path = Path(
             instance.performance_path,
             NPEQueries.NPE_FOLDER,
@@ -417,6 +416,17 @@ class OpsPerformanceReportQueries:
         "raw_op_code",
     ]
 
+    STACKED_REPORT_COLUMNS = [
+        "percent",
+        "op_code",
+        "device_time_sum_us",
+        "ops_count",
+        "flops_min",
+        "flops_max",
+        "flops_mean",
+        "flops_std",
+    ]
+
     PASSTHROUGH_COLUMNS = {
         "pm_ideal_ns": "PM IDEAL [ns]",
     }
@@ -433,6 +443,9 @@ class OpsPerformanceReportQueries:
         raw_csv = OpsPerformanceQueries.get_raw_csv(instance)
         csv_file = StringIO(raw_csv)
         csv_output_file = tempfile.mktemp(suffix=".csv")
+        csv_stacked_output_file = tempfile.mktemp(suffix=".csv")
+        # perf_report currently generates a PNG alongside the CSV using the same temp name - we'll just delete it afterwards
+        stacked_png_file = os.path.splitext(csv_output_file)[0] + ".png"
 
         try:
             perf_report.generate_perf_report(
@@ -446,9 +459,9 @@ class OpsPerformanceReportQueries:
                 cls.DEFAULT_TRACING_MODE,
                 True,
                 True,
-                True,
-                True,
                 False,
+                True,  # no_stack_by_in0 - need to ask what this is
+                csv_stacked_output_file,
             )
         except Exception as e:
             raise DataFormatError(f"Error generating performance report: {e}") from e
@@ -489,4 +502,26 @@ class OpsPerformanceReportQueries:
         finally:
             os.unlink(csv_output_file)
 
-        return report
+        stacked_report = []
+
+        try:
+            with open(csv_stacked_output_file, newline="") as csvfile:
+                reader = csv.reader(csvfile, delimiter=",")
+                next(reader, None)
+
+                for row in reader:
+                    processed_row = {
+                        column: row[index]
+                        for index, column in enumerate(cls.STACKED_REPORT_COLUMNS)
+                        if index < len(row)
+                    }
+
+                    stacked_report.append(processed_row)
+        except csv.Error as e:
+            raise DataFormatError() from e
+        finally:
+            os.unlink(csv_stacked_output_file)
+            if os.path.exists(stacked_png_file):
+                os.unlink(stacked_png_file)
+
+        return {"report": report, "stacked_report": stacked_report}

ttnn_visualizer/decorators.py

@@ -14,7 +14,6 @@ from ttnn_visualizer.exceptions import (
     NoProjectsException,
     NoValidConnectionsError,
     RemoteConnectionException,
-    RemoteSqliteException,
     SSHException,
 )
 from ttnn_visualizer.instances import get_or_create_instance
@@ -115,14 +114,6 @@ def remote_exception_handler(func):
                 message=user_message,
             )
 
-        except RemoteSqliteException as err:
-            current_app.logger.error(f"Remote Sqlite exception: {str(err)}")
-            message = err.message
-            if "No such file" in str(err):
-                message = "Unable to open SQLite binary, check path"
-            raise RemoteConnectionException(
-                status=ConnectionTestStates.FAILED, message=message
-            )
         except IOError as err:
             message = f"Error opening remote folder: {str(err)}"
             if "Name or service not known" in str(err):

ttnn_visualizer/exceptions.py

@@ -56,13 +56,6 @@ class NoProjectsException(RemoteConnectionException):
     pass
 
 
-class RemoteSqliteException(Exception):
-    def __init__(self, message, status):
-        super().__init__(message)
-        self.message = message
-        self.status = status
-
-
 class DatabaseFileNotFoundException(Exception):
     pass
 

ttnn_visualizer/models.py

@@ -169,7 +169,6 @@ class RemoteConnection(SerializeableModel):
     port: int = Field(ge=1, le=65535)
     profilerPath: str
     performancePath: Optional[str] = None
-    sqliteBinaryPath: Optional[str] = None
 
 
 class StatusMessage(SerializeableModel):

ttnn_visualizer/static/assets/{allPaths-DucASGPg.js → allPaths-BoDgVavt.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-DFKuZosj.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-DpyzM7Uz.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-DaQoP_bJ.js

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