ttnn-visualizer 0.40.3__py3-none-any.whl → 0.41.0__py3-none-any.whl

ttnn_visualizer/app.py

@@ -16,7 +16,7 @@ from typing import cast
 
 import flask
 from dotenv import load_dotenv
-from flask import Flask, jsonify
+from flask import Flask, abort, jsonify
 from flask_cors import CORS
 from werkzeug.debug import DebuggedApplication
 from werkzeug.middleware.proxy_fix import ProxyFix
@@ -46,7 +46,11 @@ def create_app(settings_override=None):
 
     config = cast(DefaultConfig, Config())
 
-    app = Flask(__name__, static_folder=config.STATIC_ASSETS_DIR, static_url_path="/")
+    app = Flask(
+        __name__,
+        static_folder=config.STATIC_ASSETS_DIR,
+        static_url_path=f"{config.BASE_PATH}static",
+    )
     logging.basicConfig(level=app.config.get("LOG_LEVEL", "INFO"))
 
     app.config.from_object(config)
@@ -56,14 +60,17 @@ def create_app(settings_override=None):
 
     middleware(app)
 
-    app.register_blueprint(api, url_prefix=f"{app.config['BASE_PATH']}/api")
+    app.register_blueprint(api, url_prefix=f"{app.config['BASE_PATH']}api")
 
     extensions(app)
 
     if flask_env == "production":
-        @app.route(f"{app.config['BASE_PATH']}/", defaults={"path": ""})
-        @app.route("/<path:path>")
+        @app.route(f"{app.config['BASE_PATH']}", defaults={"path": ""})
+        @app.route(f"{app.config['BASE_PATH']}<path:path>")
         def catch_all(path):
+            if path.startswith("static/"):
+                abort(404)  # Pass control to Flask's static view
+
             js_config = {
                 "SERVER_MODE": app.config["SERVER_MODE"],
                 "BASE_PATH": app.config["BASE_PATH"],
@@ -78,7 +85,15 @@ def create_app(settings_override=None):
                     js,
                 )
 
-            return flask.Response(html_with_config, mimetype="text/html")
+            return flask.Response(
+                html_with_config,
+                mimetype="text/html",
+                headers={
+                    "Cache-Control": "no-store, no-cache, must-revalidate, max-age=0",
+                    "Pragma": "no-cache",
+                    "Expires": "0",
+                },
+            )
 
     return app
 

ttnn_visualizer/csv_queries.py

@@ -2,6 +2,7 @@
 #
 # SPDX-FileCopyrightText: © 2025 Tenstorrent Inc.
 import csv
+import json
 import os
 import tempfile
 from io import StringIO
@@ -14,7 +15,7 @@ from tt_perf_report import perf_report
 from ttnn_visualizer.exceptions import DataFormatError
 from ttnn_visualizer.models import Instance
 from ttnn_visualizer.ssh_client import get_client
-
+from ttnn_visualizer.sftp_operations import read_remote_file
 
 class LocalCSVQueryRunner:
     def __init__(self, file_path: str, offset: int = 0):
@@ -211,7 +212,7 @@ class RemoteCSVQueryRunner:
         if error:
             raise RuntimeError(f"Error fetching raw rows: {error}")
 
-        return output.splitlines()[self.offset :]
+        return output.splitlines()[self.offset:]
 
     def get_csv_header(self) -> Dict[str, int]:
         """
@@ -259,6 +260,32 @@ class RemoteCSVQueryRunner:
             self.ssh_client.close()
 
 
+class NPEQueries:
+    NPE_FOLDER = "npe_viz"
+    MANIFEST_FILE = "manifest.json"
+
+    @staticmethod
+    def get_npe_manifest(instance: Instance):
+
+
+        if (
+            not instance.remote_connection
+            or instance.remote_connection
+            and not instance.remote_connection.useRemoteQuerying
+        ):
+            file_path = Path(
+                instance.performance_path, NPEQueries.NPE_FOLDER, NPEQueries.MANIFEST_FILE
+            )
+            with open(file_path, "r") as f:
+                return json.load(f)
+        else:
+            profiler_folder = instance.remote_profile_folder
+            return read_remote_file(
+                instance.remote_connection,
+                f"{profiler_folder.remotePath}/{NPEQueries.NPE_FOLDER}/{NPEQueries.MANIFEST_FILE}",
+            )
+
+
 class DeviceLogProfilerQueries:
     DEVICE_LOG_FILE = "profile_log_device.csv"
     DEVICE_LOG_COLUMNS = [
@@ -640,7 +667,7 @@ class OpsPerformanceReportQueries:
 
                     for key, value in cls.PASSTHROUGH_COLUMNS.items():
                         op_id = int(row[0])
-                        idx = op_id - 2 # IDs in result column one correspond to row numbers in ops perf results csv
+                        idx = op_id - 2  # IDs in result column one correspond to row numbers in ops perf results csv
                         processed_row[key] = ops_perf_results[idx][value]
 
                     report.append(processed_row)

ttnn_visualizer/static/assets/{allPaths-DSFl5HNA.js → allPaths-4_pFqSAW.js}

@@ -1 +1 @@
-import{I as n}from"./index-BKzgFDAn.js";import{I as e}from"./index-BvSuWPlB.js";import{p as r,I as s}from"./index-CWerbNbe.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-BKzgFDAn.js";import{I as e}from"./index-BvSuWPlB.js";import{p as r,I as s}from"./index-DFVwehlj.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-BwkPDbOI.js → allPathsLoader-CpLPTLlt.js}

@@ -1,2 +1,2 @@
-const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f||(m.f=["assets/allPaths-DSFl5HNA.js","assets/index-BKzgFDAn.js","assets/index-BvSuWPlB.js","assets/index-CWerbNbe.js","assets/index-C1rJBrMl.css"])))=>i.map(i=>d[i]);
-import{_ as o,a as n,b as i}from"./index-CWerbNbe.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-DSFl5HNA.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-4_pFqSAW.js","assets/index-BKzgFDAn.js","assets/index-BvSuWPlB.js","assets/index-DFVwehlj.js","assets/index-C1rJBrMl.css"])))=>i.map(i=>d[i]);
+import{_ as o,a as n,b as i}from"./index-DFVwehlj.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-4_pFqSAW.js"),__vite__mapDeps([0,1,2,3,4]))];case 1:return t=r.sent().getIconPaths,[2,t(e,a)]}})})};export{_ as allPathsLoader};