v.0.6.1 (#165)

Buffers view: zoom
Buffer view: linkable operations
Buffers view: buffer table - initial rendering
Buffers pages query api for Tensor sharding visualization
Support for multiple per-tab parallel downloads
Multithreading to support background jobs
Real-time monitoring of sync/download/upload jobs
Improved session management
UX improvements to display/monitor background jobs

backend/ttnn_visualizer/decorators.py

@@ -18,6 +18,12 @@ def wrapper(*args, **kwargs):
             abort(404)
 
         session = get_or_create_tab_session(tab_id=tab_id)
+
+        remote_connection = None
+
+        if hasattr(session, "remote_connection"):
+            remote_connection = session.remote_connection
+
         active_report = session.active_report
 
         if not active_report:
@@ -27,7 +33,8 @@ def wrapper(*args, **kwargs):
             # Raise 404 if report_path is missing or does not exist
             abort(404)
 
-        report_path = get_report_path(active_report, current_app)
+        report_path = get_report_path(active_report, current_app, remote_connection)
+
         if not Path(report_path).exists():
             current_app.logger.error(
                 f"Specified report path {report_path} does not exist, returning 404"

backend/ttnn_visualizer/sftp_operations.py

@@ -232,7 +232,7 @@ def sync_test_folders(
 
     check_permissions(client, remote_folder.remotePath)
 
-    if not check_gzip_exists(client):
+    if check_gzip_exists(client):
         try:
             remote_tar_path = f"{remote_folder.remotePath}.tar.gz"
             folder_size = calculate_folder_size(client, remote_folder.remotePath)

backend/ttnn_visualizer/utils.py

@@ -24,7 +24,7 @@ def wrapper(*args, **kwargs):
     return wrapper
 
 
-def get_report_path(active_report, current_app):
+def get_report_path(active_report, current_app, remote_connection=None):
     """
     Gets the report path for the given active_report object.
     :param active_report: Dictionary representing the active report.
@@ -36,7 +36,6 @@ def get_report_path(active_report, current_app):
 
     if active_report:
         # Check if there's an associated RemoteConnection
-        remote_connection = active_report.get("remote_connection")
         if remote_connection:
             # Use the remote directory if a remote connection exists
             base_dir = Path(remote_dir).joinpath(remote_connection.get("host"))

package.json

@@ -1,7 +1,7 @@
 {
   "name": "ttnn-visualzer",
   "private": true,
-  "version": "0.6.0",
+  "version": "0.6.1",
   "type": "module",
   "scripts": {
     "dev": "vite",

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "ttnn_visualizer"
 authors = []
-version = "0.6.0"
+version = "0.6.1"
 description = "TT Visualizer"
 readme = "README.md"
 requires-python = ">=3.12"