PyPI - ttnn-visualizer - Versions diffs - 0.47.0__py3-none-any.whl → 0.49.0__py3-none-any.whl - Mend

ttnn-visualizer 0.47.0py3-none-any.whl → 0.49.0py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (27) hide show

ttnn_visualizer/static/assets/splitPathsBySizeLoader-ivxxaHxa.js ADDED Viewed

	@@ -0,0 +1 @@
1	+ import{p as r,I as s,_ as a}from"./index-DVrPLQJ7.js";const n=async(o,_)=>{const i=r(o);let t;return _===s.STANDARD?t=await a(()=>import("./index-CnPrfHYh.js").then(e=>e.I),[]):t=await a(()=>import("./index-Cnc1EkDo.js").then(e=>e.I),[]),t[i]};export{n as splitPathsBySizeLoader};

ttnn_visualizer/static/index.html CHANGED Viewed

@@ -34,8 +34,8 @@
             /* SERVER_CONFIG */
         </script>
-      <script type="module" crossorigin src="/static/assets/index-CmA3KkTi.js"></script>
-      <link rel="stylesheet" crossorigin href="/static/assets/index-DJA68-a6.css">
+      <script type="module" crossorigin src="/static/assets/index-DVrPLQJ7.js"></script>
+      <link rel="stylesheet" crossorigin href="/static/assets/index-UuXdrHif.css">
     </head>
     <body>

ttnn_visualizer/tests/test_serializers.py CHANGED Viewed

@@ -161,12 +161,14 @@ class TestSerializers(unittest.TestCase):
                         "device_id": 1,
                         "address": 1000,
                         "buffer_type": 0,
+                        "buffer_layout": None,
                         "size": 256,
                     },
                     {
                         "device_id": 2,
                         "address": 2000,
                         "buffer_type": 1,
+                        "buffer_layout": None,
                         "size": 512,
                     },
                 ],
@@ -179,6 +181,7 @@ class TestSerializers(unittest.TestCase):
                         "device_id": 3,
                         "address": 3000,
                         "buffer_type": 1,
+                        "buffer_layout": None,
                         "size": 1024,
                     },
                 ],
@@ -260,12 +263,14 @@ class TestSerializers(unittest.TestCase):
                     "device_id": 1,
                     "address": 1000,
                     "buffer_type": 0,
+                    "buffer_layout": None,
                     "size": 256,
                 },
                 {
                     "device_id": 2,
                     "address": 2000,
                     "buffer_type": 1,
+                    "buffer_layout": None,
                     "size": 512,
                 },
             ],
@@ -389,6 +394,7 @@ class TestSerializers(unittest.TestCase):
             "buffers": [
                 {
                     "address": 1000,
+                    "buffer_layout": None,
                     "buffer_type": 0,
                     "device_id": 1,
                     "max_size_per_bank": 256,

ttnn_visualizer/views.py CHANGED Viewed

@@ -12,6 +12,7 @@ from http import HTTPStatus
 from pathlib import Path
 from typing import List
+import orjson
 import yaml
 import zstd
 from flask import Blueprint, Response, current_app, jsonify, request, session
@@ -865,12 +866,10 @@ def create_npe_files():
     data_directory = current_app.config["LOCAL_DATA_DIRECTORY"]
     for file in files:
-        if not file.filename.endswith(".json") and not file.filename.endswith(
-            ".npeviz.zst"
-        ):
+        if not file.filename.endswith(".json") and not file.filename.endswith(".zst"):
             return StatusMessage(
                 status=ConnectionTestStates.FAILED,
-                message="NPE requires a valid .json or .npeviz.zst file",
+                message="NPE requires a valid .json or .zst file",
             ).model_dump()
     npe_name = extract_npe_name(files)
@@ -1263,13 +1262,80 @@ def get_npe_data(instance: Instance):
         )
         return Response(status=HTTPStatus.NOT_FOUND)
-    if compressed_path and compressed_path.exists():
-        with open(compressed_path, "rb") as file:
-            compressed_data = file.read()
-            uncompressed_data = zstd.uncompress(compressed_data)
-            npe_data = json.loads(uncompressed_data)
-    else:
-        with open(uncompressed_path, "r") as file:
-            npe_data = json.load(file)
+    try:
+        if compressed_path and compressed_path.exists():
+            with open(compressed_path, "rb") as file:
+                compressed_data = file.read()
+                uncompressed_data = zstd.uncompress(compressed_data)
+                npe_data = json.loads(uncompressed_data)
+        else:
+            with open(uncompressed_path, "r") as file:
+                npe_data = json.load(file)
+    except json.JSONDecodeError as e:
+        logger.error(f"Invalid JSON in NPE file: {e}")
+        return Response(status=HTTPStatus.UNPROCESSABLE_ENTITY)
+    # Use orjson for much faster JSON serialization of large files
+    return Response(orjson.dumps(npe_data), mimetype="application/json")
+@api.route("/notify", methods=["POST"])
+def notify_report_update():
+    """
+    Endpoint to receive notifications about report updates and broadcast them via websockets.
+    """
+    from ttnn_visualizer.sockets import (
+        ExitStatus,
+        ReportGenerated,
+        emit_report_generated,
+    )
+    try:
+        data = request.get_json()
+        if not data:
+            return jsonify({"error": "No JSON data provided"}), 400
+        report_name = data.get("report_name")
+        exit_status_str = data.get("exit_status")
-    return jsonify(npe_data)
+        if not report_name:
+            return jsonify({"error": "report_name is required"}), 400
+        # Validate status
+        try:
+            exit_status = (
+                ExitStatus(exit_status_str.upper()) if exit_status_str else None
+            )
+        except ValueError:
+            return (
+                jsonify({"error": "Invalid exit_status."}),
+                400,
+            )
+        # Create and emit the report update
+        report_generated = ReportGenerated(
+            report_name=report_name,
+            exit_status=exit_status,
+            profiler_path=data.get("profiler_path"),
+            performance_path=data.get("performance_path"),
+        )
+        emit_report_generated(report_generated)
+        logger.info(f"Report generated notification processed: {report_name}")
+        return (
+            jsonify(
+                {
+                    "report_name": report_name,
+                    "profiler_path": report_generated.profiler_path,
+                    "performance_path": report_generated.performance_path,
+                    "exit_status": exit_status.value if exit_status else None,
+                    "timestamp": report_generated.timestamp,
+                }
+            ),
+            200,
+        )
+    except Exception as e:
+        logger.error(f"Error processing report update notification: {str(e)}")
+        return jsonify({"error": "Internal server error"}), 500

{ttnn_visualizer-0.47.0.dist-info → ttnn_visualizer-0.49.0.dist-info}/METADATA RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ttnn_visualizer
-Version: 0.47.0
+Version: 0.49.0
 Summary: TT-NN Visualizer
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
@@ -17,6 +17,7 @@ Requires-Dist: Flask-Static-Digest==0.4.1
 Requires-Dist: Flask==3.1.1
 Requires-Dist: gevent==24.10.2
 Requires-Dist: gunicorn~=23.0.0
+Requires-Dist: orjson>=3.9.0
 Requires-Dist: pandas==2.2.3
 Requires-Dist: pydantic_core==2.27.1
 Requires-Dist: pydantic==2.10.3

{ttnn_visualizer-0.47.0.dist-info → ttnn_visualizer-0.49.0.dist-info}/RECORD RENAMED Viewed

@@ -7,25 +7,26 @@ ttnn_visualizer/exceptions.py,sha256=XwTIykJpdvZV8nqrd9JZdHIYL0EBFBhTbE9H09VZluA
 ttnn_visualizer/extensions.py,sha256=6OIRJ8-_ccfjOaXSruRXiS29jEbxp4Pyk-0JlD8IHBQ,379
 ttnn_visualizer/file_uploads.py,sha256=HFcC6TBt5I0oBkiKgM2Qw1W7hpixE8TOTACS5N-rmGE,5013
 ttnn_visualizer/instances.py,sha256=XctQgQXdlwtuXWFXFletRoX1m1lGUZdiW3TwIIjY0uw,11564
-ttnn_visualizer/models.py,sha256=oxZVvWjtBDP0X6GqoYaH4wDa5PqJF_vG_I3Q_YWHMmo,7814
-ttnn_visualizer/queries.py,sha256=3Nv0jXIDNVH-qKx9xc9zSINy4FdmDidu-1I6f-VYV48,9603
+ttnn_visualizer/models.py,sha256=QCL64uynSS3WbUI05UoHIgex6eWgMGqf0-YTm1XwCwI,7854
+ttnn_visualizer/pytest_plugin.py,sha256=bEG0cbqH0HUuZT5Ox9tFoexFNTyimBBPpI_jp75b54c,2629
+ttnn_visualizer/queries.py,sha256=0s2zGA7CH1fbV6BWB-MhdHfWW7GhkDti4gVJyDmeCE8,9607
 ttnn_visualizer/remote_sqlite_setup.py,sha256=VdJk5LfkaJo1XPC-yxy909I0AOgJF1GUjryj0Oe0O14,3498
 ttnn_visualizer/serializers.py,sha256=LmjHIrFg8BLx1JKVFh9Nd_TcA7nyy1MwY2BOGnX1MKw,8029
 ttnn_visualizer/settings.py,sha256=W0PCeBCGXONNfFHqI7ATCVGhKv0a02-BhwBwrB3Clag,4660
 ttnn_visualizer/sftp_operations.py,sha256=9HwbPJPSO1UUQ98d5zeWAkEwR0zFPryUakcI68GqkVw,30181
-ttnn_visualizer/sockets.py,sha256=W5pK0QmlsU58EH_Qnra6HlcswLS7vsw5C_pk4avVYvs,3706
+ttnn_visualizer/sockets.py,sha256=_Hdne33r4FrB2tg58Vw87FWLbgQ_ikICVp4o1Mkv2mo,4789
 ttnn_visualizer/ssh_client.py,sha256=-GS2_1tdlUqVoLfRS02i3_o1fQaM39UQN-jtAnPBmzQ,13511
 ttnn_visualizer/utils.py,sha256=_W990vRnup9zlWJ-g_Bggyo-wDjCYF49p0nPxWNgxrE,12934
-ttnn_visualizer/views.py,sha256=4l9bYIdDMyui1WjBDOn7rJ9V4egc439jeN1uG47Yk7Y,43072
-ttnn_visualizer/static/index.html,sha256=5kiMyUyhbOhgwPgREMpeH3v-AGdP5G4Qtb10TcV7A-w,1135
-ttnn_visualizer/static/assets/allPaths-OR2-IW-_.js,sha256=Oh2drbxvlmAIFWE4Srb9Pi8hdb__NF3_mVn2Vm60UI4,255
-ttnn_visualizer/static/assets/allPathsLoader-xRXweacG.js,sha256=co2dBtSW_xZQ_XduamaE0h5AVsH-Ldswk44uwD7QvNc,477
-ttnn_visualizer/static/assets/index-03c8d4Gh.js,sha256=k0jIi5q-lzXcPCqAD7K091vBMOkJHmx_fYcYKh-fOnM,285602
-ttnn_visualizer/static/assets/index-CmA3KkTi.js,sha256=rp_h66A8t5Jl7OwbDKnM6bj03uE_H2bwqBAc6aIuIKY,7882665
-ttnn_visualizer/static/assets/index-DJA68-a6.css,sha256=sPrF4HPkGtkuy0Ae8wkQXcUUzGGRGeKNeymUY6YPtwo,622830
-ttnn_visualizer/static/assets/index-PKNBViIU.js,sha256=dUuCyAPr_QvvTY0Xula4q4rgOm1J7xgdbppL67off4k,294225
+ttnn_visualizer/views.py,sha256=R9JD6ajxd4Z0zgbjqYGeX4L8fOUSVqDsiuQ7FSDLDvA,45331
+ttnn_visualizer/static/index.html,sha256=9Gct5mpFLirIshWl30b7_TgT4fcYWq_OeYRsNPky6Iw,1135
+ttnn_visualizer/static/assets/allPaths-G_CNx_x1.js,sha256=l6Ec9RqUrxT3o1AFEAPfG40uv2iZgLBV5UsA7KhIsms,255
+ttnn_visualizer/static/assets/allPathsLoader-s_Yfmxfp.js,sha256=bxXd5uj9p2PVTrwhCQrFItzS3VXDwbhssIwBW8wqFhM,477
+ttnn_visualizer/static/assets/index-CnPrfHYh.js,sha256=xtrmUrKcMSwTBA7zqXJ6I_L70LzG9DkO-zbC3aWJS8g,289282
+ttnn_visualizer/static/assets/index-Cnc1EkDo.js,sha256=m-gmIOT0AsJcaW0TVjRpOwSm52Dfp4ZDcrdnIS4qBHA,298140
+ttnn_visualizer/static/assets/index-DVrPLQJ7.js,sha256=YL1kan4zv0M5x5oZeSkl-q7PlCHnCgQb7w8rj1GNcGU,7890913
+ttnn_visualizer/static/assets/index-UuXdrHif.css,sha256=gdueUx5rG_VioiyHhyblHRYqo8AkOtROM3xvEqIovGk,623039
 ttnn_visualizer/static/assets/site-BTBrvHC5.webmanifest,sha256=Uy_XmnGuYFVf-OZuma2NvgEPdrCrevb3HZvaxSIHoA0,456
-ttnn_visualizer/static/assets/splitPathsBySizeLoader-P9sdNg6R.js,sha256=hrgrHehC3Bwy8EF3pWi11eVXlDm4JZ9Y2-j7CSX0WCY,281
+ttnn_visualizer/static/assets/splitPathsBySizeLoader-ivxxaHxa.js,sha256=wl69al5zPgTKgIUKXz6yBRpjaB9rTPHVD-hqrz09ryk,281
 ttnn_visualizer/static/favicon/android-chrome-192x192.png,sha256=BZWA09Zxaa3fXbaeS6nhWo2e-DUSjm9ElzNQ_xTB5XU,6220
 ttnn_visualizer/static/favicon/android-chrome-512x512.png,sha256=HBiJSZyguB3o8fMJuqIGcpeBy_9JOdImme3wD02UYCw,62626
 ttnn_visualizer/static/favicon/favicon-32x32.png,sha256=Zw201qUsczQv1UvoQvJf5smQ2ss10xaTeWxmQNYCGtY,480
@@ -33,11 +34,11 @@ ttnn_visualizer/static/favicon/favicon.svg,sha256=wDPY3VrekJ_DE1TnJ2vUy602K3S4Xe
 ttnn_visualizer/static/sample-data/cluster-desc.yaml,sha256=LMxOmsRUXtVVU5ogzYkXUozB3dg2IzqIRJQpV_O5qMU,29618
 ttnn_visualizer/tests/__init__.py,sha256=FCQeTWnXsf-Wx-fay53-lQsm0y5-GcPMUmzhE5upDx0,93
 ttnn_visualizer/tests/test_queries.py,sha256=HqaDXwudZpXiigJdHkdJP8oiUc-PtHASbpLnQQpbD7A,13792
-ttnn_visualizer/tests/test_serializers.py,sha256=AF6m6tvewnZ_OSQMgMTUhsOI26GdJHPajvnRGIa9P4U,18492
-ttnn_visualizer-0.47.0.dist-info/licenses/LICENSE,sha256=gcO1GgS_1J-gOVcHALMiIPsyPv2Pu7hSSHhsgkp6F_Y,19345
-ttnn_visualizer-0.47.0.dist-info/licenses/LICENSE_understanding.txt,sha256=pymi-yb_RvYM9p2ZA4iSNsImcvhDBBxlGuJCY9dTq7M,233
-ttnn_visualizer-0.47.0.dist-info/METADATA,sha256=ugKvf2bHr5oB9S0nXFfYV8tRTBr652XPa3t43IrnFRI,7751
-ttnn_visualizer-0.47.0.dist-info/WHEEL,sha256=lTU6B6eIfYoiQJTZNc-fyaR6BpL6ehTzU3xGYxn2n8k,91
-ttnn_visualizer-0.47.0.dist-info/entry_points.txt,sha256=QpuUpkmQ_mEHJTMqOBdU0MH2Z4WF_9iFsGACeyyAO1E,61
-ttnn_visualizer-0.47.0.dist-info/top_level.txt,sha256=M1EGkvDOuIfbhDbcUdz2-TSdmCtDoQ2Uyag9k5JLDSY,16
-ttnn_visualizer-0.47.0.dist-info/RECORD,,
+ttnn_visualizer/tests/test_serializers.py,sha256=IJekAZRBpyOr_Ffp0dqSrnhFOU_ZZ8pHma_JO0j23TQ,18762
+ttnn_visualizer-0.49.0.dist-info/licenses/LICENSE,sha256=bapl7NysYmv8aYSVxr5qFDya0vCqppyjkyyNJYosBdc,20044
+ttnn_visualizer-0.49.0.dist-info/licenses/LICENSE_understanding.txt,sha256=pymi-yb_RvYM9p2ZA4iSNsImcvhDBBxlGuJCY9dTq7M,233
+ttnn_visualizer-0.49.0.dist-info/METADATA,sha256=f6Pxll0vaRJggbBD6qEqYuVGGIKw0dt4sMX-AaWVy_k,7780
+ttnn_visualizer-0.49.0.dist-info/WHEEL,sha256=lTU6B6eIfYoiQJTZNc-fyaR6BpL6ehTzU3xGYxn2n8k,91
+ttnn_visualizer-0.49.0.dist-info/entry_points.txt,sha256=QpuUpkmQ_mEHJTMqOBdU0MH2Z4WF_9iFsGACeyyAO1E,61
+ttnn_visualizer-0.49.0.dist-info/top_level.txt,sha256=M1EGkvDOuIfbhDbcUdz2-TSdmCtDoQ2Uyag9k5JLDSY,16
+ttnn_visualizer-0.49.0.dist-info/RECORD,,

{ttnn_visualizer-0.47.0.dist-info → ttnn_visualizer-0.49.0.dist-info}/licenses/LICENSE RENAMED Viewed

@@ -155,22 +155,30 @@ The following separate and independent dependencies are utilized by this project
 - vitest - MIT - https://github.com/vitest-dev/vitest/blob/main/LICENSE
 - vitest-canvas-mock - MIT - https://github.com/wobsoriano/vitest-canvas-mock/blob/main/LICENSE
-- build - MIT https://opensource.org/license/mit
+- black - MIT - https://github.com/psf/black/blob/main/LICENSE
+- build - MIT - https://opensource.org/license/mit
 - Flask - BSD‑3‑Clause - https://github.com/pallets/flask/blob/main/LICENSE.txt
-- flask_cors - MIT - https://github.com/corydolphin/flask-cors/blob/main/LICENSE
+- Flask-Cors - MIT - https://github.com/corydolphin/flask-cors/blob/main/LICENSE
 - flask-socketio - MIT - https://github.com/miguelgrinberg/Flask-SocketIO/blob/main/LICENSE
 - flask-sqlalchemy - BSD‑3‑Clause - https://github.com/pallets-eco/flask-sqlalchemy/blob/main/LICENSE
+- Flask-Static-Digest - MIT - https://github.com/nickjj/flask-static-digest/blob/master/LICENSE
 - gevent - MIT - https://github.com/gevent/gevent/blob/master/LICENSE
 - gunicorn - MIT - https://github.com/benoitc/gunicorn/blob/master/LICENSE
+- isort - MIT - https://github.com/PyCQA/isort/blob/main/LICENSE
+- mypy - MIT - https://github.com/python/mypy/blob/master/LICENSE
+- orjson - Dual-license Apache-2.0 or MIT - https://github.com/ijl/orjson/blob/master/LICENSE-APACHE and https://github.com/ijl/orjson/blob/master/LICENSE-MIT
 - pandas - BSD‑3‑Clause - https://github.com/pandas-dev/pandas/blob/main/LICENSE
 - playwright - Apache-2.0 - https://github.com/microsoft/playwright-python/blob/main/LICENSE
+- playwright - Apache-2.0 - https://github.com/microsoft/playwright/blob/main/LICENSE
 - pydantic - MIT - https://github.com/pydantic/pydantic-settings/blob/main/LICENSE
+- pydantic_core - MIT - https://github.com/pydantic/pydantic-core/blob/main/LICENSE
+- pytest - MIT - https://github.com/pytest-dev/pytest/blob/main/LICENSE
 - python-dotenv - BSD‑3‑Clause - https://github.com/theskumar/python-dotenv/blob/main/LICENSE
 - PyYAML - MIT - https://github.com/yaml/pyyaml/blob/main/LICENSE
 - setuptools - MIT - https://github.com/pypa/setuptools/blob/main/LICENSE
-- tt-perf-report - MIT https://opensource.org/license/mit
+- tt-perf-report - MIT - https://opensource.org/license/mit
 - uvicorn - BSD‑3‑Clause - https://github.com/encode/uvicorn/blob/master/LICENSE.md
-- wheel - MIT https://opensource.org/license/mit
-- zstd (python-zstandard) - BSD‑3‑Clause - https://github.com/indygreg/python-zstandard/blob/main/LICENSE
+- wheel - MIT - https://opensource.org/license/mit
+- zstd - (python-zstandard) - BSD‑3‑Clause - https://github.com/indygreg/python-zstandard/blob/main/LICENSE
 - Upload Release Assets - GitHub Action - MIT - https://github.com/AButler/upload-release-assets/blob/master/LICENSE

ttnn_visualizer/static/assets/allPaths-OR2-IW-_.js DELETED Viewed

	@@ -1 +0,0 @@
1	- import{I as s}from"./index-03c8d4Gh.js";import{I as r}from"./index-PKNBViIU.js";import{p as n,I as c}from"./index-CmA3KkTi.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-xRXweacG.js DELETED Viewed

	@@ -1,2 +0,0 @@
1	- const __vite__mapDeps=(i,m=__vite__mapDeps,d=(m.f\|\|(m.f=["assets/allPaths-OR2-IW-_.js","assets/index-03c8d4Gh.js","assets/index-PKNBViIU.js","assets/index-CmA3KkTi.js","assets/index-DJA68-a6.css"])))=>i.map(i=>d[i]);
2	- import{_ as e}from"./index-CmA3KkTi.js";const s=async(t,a)=>{const{getIconPaths:o}=await e(async()=>{const{getIconPaths:r}=await import("./allPaths-OR2-IW-_.js");return{getIconPaths:r}},__vite__mapDeps([0,1,2,3,4]));return o(t,a)};export{s as allPathsLoader};

ttnn-visualizer 0.47.0__py3-none-any.whl → 0.49.0__py3-none-any.whl

ttnn-visualizer 0.47.0py3-none-any.whl → 0.49.0py3-none-any.whl