ttnn-visualizer 0.27.3__py3-none-any.whl → 0.28.0__py3-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 (17) hide show
  1. ttnn_visualizer/serializers.py +13 -5
  2. ttnn_visualizer/sessions.py +8 -8
  3. ttnn_visualizer/settings.py +2 -2
  4. ttnn_visualizer/static/assets/{allPaths-DhoJy5jM.js → allPaths-DGdUWICT.js} +1 -1
  5. ttnn_visualizer/static/assets/{allPathsLoader-B4SALZ6P.js → allPathsLoader-BSYo-ZTH.js} +2 -2
  6. ttnn_visualizer/static/assets/{index-BME2kYRR.js → index-CYZZ70nJ.js} +207 -207
  7. ttnn_visualizer/static/assets/{index-BTfoVg9a.css → index-DWVfq2z5.css} +2 -2
  8. ttnn_visualizer/static/assets/{splitPathsBySizeLoader-B2PjTVfS.js → splitPathsBySizeLoader-hHE0y-FQ.js} +1 -1
  9. ttnn_visualizer/static/index.html +2 -2
  10. ttnn_visualizer/views.py +26 -4
  11. {ttnn_visualizer-0.27.3.dist-info → ttnn_visualizer-0.28.0.dist-info}/METADATA +11 -10
  12. {ttnn_visualizer-0.27.3.dist-info → ttnn_visualizer-0.28.0.dist-info}/RECORD +17 -17
  13. {ttnn_visualizer-0.27.3.dist-info → ttnn_visualizer-0.28.0.dist-info}/LICENSE +0 -0
  14. {ttnn_visualizer-0.27.3.dist-info → ttnn_visualizer-0.28.0.dist-info}/LICENSE_understanding.txt +0 -0
  15. {ttnn_visualizer-0.27.3.dist-info → ttnn_visualizer-0.28.0.dist-info}/WHEEL +0 -0
  16. {ttnn_visualizer-0.27.3.dist-info → ttnn_visualizer-0.28.0.dist-info}/entry_points.txt +0 -0
  17. {ttnn_visualizer-0.27.3.dist-info → ttnn_visualizer-0.28.0.dist-info}/top_level.txt +0 -0
ttnn_visualizer/static/assets/{splitPathsBySizeLoader-B2PjTVfS.js → splitPathsBySizeLoader-hHE0y-FQ.js} RENAMED
@@ -1 +1 @@
1
- import{_ as o,a as _,b as i,p as c,I as u}from"./index-BME2kYRR.js";var p=function(n,s){return o(void 0,void 0,void 0,function(){var a,r;return _(this,function(e){switch(e.label){case 0:return a=c(n),s!==u.STANDARD?[3,2]:[4,i(()=>import("./index-BVMreIQm.js").then(t=>t.I),[])];case 1:return r=e.sent(),[3,4];case 2:return[4,i(()=>import("./index-Do7YB6C4.js").then(t=>t.I),[])];case 3:r=e.sent(),e.label=4;case 4:return[2,r[a]]}})})};export{p as splitPathsBySizeLoader};
1
+ import{_ as o,a as _,b as i,p as c,I as u}from"./index-CYZZ70nJ.js";var p=function(n,s){return o(void 0,void 0,void 0,function(){var a,r;return _(this,function(e){switch(e.label){case 0:return a=c(n),s!==u.STANDARD?[3,2]:[4,i(()=>import("./index-BVMreIQm.js").then(t=>t.I),[])];case 1:return r=e.sent(),[3,4];case 2:return[4,i(()=>import("./index-Do7YB6C4.js").then(t=>t.I),[])];case 3:r=e.sent(),e.label=4;case 4:return[2,r[a]]}})})};export{p as splitPathsBySizeLoader};
ttnn_visualizer/static/index.html CHANGED
@@ -27,8 +27,8 @@
27
27
  name="theme-color"
28
28
  content="#33333d"
29
29
  />
30
- <script type="module" crossorigin src="/assets/index-BME2kYRR.js"></script>
31
- <link rel="stylesheet" crossorigin href="/assets/index-BTfoVg9a.css">
30
+ <script type="module" crossorigin src="/assets/index-CYZZ70nJ.js"></script>
31
+ <link rel="stylesheet" crossorigin href="/assets/index-DWVfq2z5.css">
32
32
  </head>
33
33
  <body>
34
34
  <div id="root"></div>
ttnn_visualizer/views.py CHANGED
@@ -10,14 +10,13 @@ from http import HTTPStatus
10
10
  from pathlib import Path
11
11
  from typing import List
12
12
 
13
- import yaml
14
- from flask import Blueprint, Response, jsonify
13
+ from flask import Blueprint
15
14
  from flask import request, current_app
16
15
 
17
16
  from ttnn_visualizer.csv_queries import DeviceLogProfilerQueries, OpsPerformanceQueries, OpsPerformanceReportQueries
18
17
  from ttnn_visualizer.decorators import with_session
19
- from ttnn_visualizer.exceptions import DataFormatError
20
18
  from ttnn_visualizer.enums import ConnectionTestStates
19
+ from ttnn_visualizer.exceptions import DataFormatError
21
20
  from ttnn_visualizer.exceptions import RemoteConnectionException
22
21
  from ttnn_visualizer.file_uploads import (
23
22
  extract_report_name,
@@ -40,7 +39,7 @@ from ttnn_visualizer.serializers import (
40
39
  serialize_buffer_pages,
41
40
  serialize_operation_buffers,
42
41
  serialize_operations_buffers,
43
- serialize_devices,
42
+ serialize_devices, serialize_buffer,
44
43
  )
45
44
  from ttnn_visualizer.sessions import (
46
45
  update_instance,
@@ -306,6 +305,26 @@ def tensor_detail(tensor_id, session: Instance):
306
305
  return dataclasses.asdict(tensors[0])
307
306
 
308
307
 
308
+ @api.route("/buffers", methods=["GET"])
309
+ @with_session
310
+ def get_all_buffers(session: Instance):
311
+ buffer_type = request.args.get("buffer_type", "")
312
+ device_id = request.args.get("device_id", None)
313
+ if buffer_type and str.isdigit(buffer_type):
314
+ buffer_type = int(buffer_type)
315
+ else:
316
+ buffer_type = None
317
+
318
+ with DatabaseQueries(session) as db:
319
+ buffers = list(
320
+ db.query_buffers(
321
+ filters={"buffer_type": buffer_type, "device_id": device_id}
322
+ )
323
+ )
324
+ serialized = [serialize_buffer(b) for b in buffers]
325
+ return jsonify(serialized)
326
+
327
+
309
328
  @api.route("/operation-buffers", methods=["GET"])
310
329
  @with_session
311
330
  def get_operations_buffers(session: Instance):
@@ -457,6 +476,7 @@ def create_report_files():
457
476
  status=ConnectionTestStates.OK, message="Success."
458
477
  ).model_dump()
459
478
 
479
+
460
480
  @api.route("/local/upload/profile", methods=["POST"])
461
481
  def create_profile_files():
462
482
  files = request.files.getlist("files")
@@ -532,6 +552,7 @@ def create_npe_files():
532
552
  status=ConnectionTestStates.OK, message="Success"
533
553
  ).model_dump()
534
554
 
555
+
535
556
  @api.route("/remote/folder", methods=["POST"])
536
557
  def get_remote_folders():
537
558
  connection = RemoteConnection.model_validate(request.json, strict=False)
@@ -818,6 +839,7 @@ def get_instance(session: Instance):
818
839
  # Used to gate UI functions if no report is active
819
840
  return session.model_dump()
820
841
 
842
+
821
843
  @api.route("/npe", methods=["GET"])
822
844
  @with_session
823
845
  @timer
{ttnn_visualizer-0.27.3.dist-info → ttnn_visualizer-0.28.0.dist-info}/METADATA RENAMED
@@ -1,7 +1,7 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ttnn_visualizer
3
- Version: 0.27.3
4
- Summary: TT Visualizer
3
+ Version: 0.28.0
4
+ Summary: TT-NN Visualizer
5
5
  Classifier: Programming Language :: Python :: 3
6
6
  Classifier: License :: OSI Approved :: MIT License
7
7
  Classifier: Operating System :: OS Independent
@@ -85,35 +85,36 @@ For the latest updates and features, please see [releases](https://github.com/te
85
85
 
86
86
  #### Application demo
87
87
 
88
- https://github.com/user-attachments/assets/d00a2629-0bd1-4ee1-bb12-bd796d85221d
88
+ https://github.com/user-attachments/assets/4e51a636-c6d6-46df-bf34-a06bca13c0b3
89
89
 
90
90
  | L1 Summary with Tensor highlight | Operation inputs and outputs |
91
91
  |-----------------------------------------------|------------------------------------------|
92
- | <img width="400" alt="L1 Summary with Tensor highlight" src="https://github.com/user-attachments/assets/73c0aff8-16b2-4d1e-85a8-81c434012d36" /> | <img width="400" alt="Operation inputs and outputs" src="https://github.com/user-attachments/assets/e9480ad2-7e24-436b-932b-050c6b489516" /> |
92
+ | <img width="400" alt="L1 Summary with Tensor highlight" src="https://github.com/user-attachments/assets/6679e3ab-3819-4037-bf63-15aac0f8b625" /> | <img width="400" alt="Operation inputs and outputs" src="https://github.com/user-attachments/assets/7b0bf6da-d6e5-4b8d-b2cd-457181ac6b99" /> |
93
93
 
94
94
  | Device operations with memory consumption | DRAM memory allocation |
95
95
  |-----------------------------------------------|------------------------------------------|
96
- | <img width="400" alt="Device operations with memory consumption" src="https://github.com/user-attachments/assets/33f83d13-4597-4ad6-bf93-3e81d888a16c" /> | <img width="400" alt="DRAM memory allocations" src="https://github.com/user-attachments/assets/39e73896-2a6a-4404-8a82-c76c10662ca3" /> |
96
+ | <img width="400" alt="Device operations with memory consumption" src="https://github.com/user-attachments/assets/46f51cde-86ec-4251-8f41-261b5e14e1b7" />| <img width="400" alt="DRAM memory allocations" src="https://github.com/user-attachments/assets/588dbb57-3964-48f8-aa0d-bafc614706d3" /> |
97
97
 
98
98
  | Operation graph view | Model buffer summary |
99
99
  |-----------------------------------------------|------------------------------------------|
100
- | <img width="400" alt="Operation graph view" src="https://github.com/user-attachments/assets/04813765-8dd1-48af-b054-4de3fe1afba3" /> | <img width="400" alt="Model buffer summary" src="https://github.com/user-attachments/assets/0371e8bc-1f08-43db-a5a1-6191a611d411" /> |
100
+ | <img width="400" alt="Operation graph view" src="https://github.com/user-attachments/assets/a77b099e-9caa-4758-8dee-8b30cad4dad8" /> | <img width="400" alt="Model buffer summary" src="https://github.com/user-attachments/assets/90a4c0e6-58cb-4031-b224-a91f3beadc51" /> |
101
101
 
102
102
  | Per core allocation details | Per core allocation details for individual tensors |
103
103
  |-----------------------------------------------|------------------------------------------|
104
- | <img width="400" alt="Per core allocation details" src="https://github.com/user-attachments/assets/e1647b4d-9bf9-4e30-98a5-8cfd48ef495f" /> | <img width="400" alt="Per core allocation details for individual tensor" src="https://github.com/user-attachments/assets/c160744e-0c23-42da-a3a4-ad4bbe513061" /> |
104
+ | <img width="400" alt="Per core allocation details" src="https://github.com/user-attachments/assets/0d4a833a-d9b6-4239-a953-42c0b84d4f80" /> | <img width="400" alt="Per core allocation details for individual tensor" src="https://github.com/user-attachments/assets/06b69fbb-c6fc-45fe-9360-36722e68bee5" /> |
105
105
 
106
106
  | Tensor details list | Performance report |
107
107
  |-----------------------------------------------|------------------------------------------|
108
- | <img width="400" alt="Tensor details list" src="https://github.com/user-attachments/assets/d19d770b-a6cd-4c5d-96d5-181c2a5e9692" /> | <img width="400" alt="Performance report" src="https://github.com/user-attachments/assets/882ca7e3-3ef8-47cd-9e78-f69474f13e86" /> |
108
+ | <img width="400" alt="Tensor details list" src="https://github.com/user-attachments/assets/919da94e-45f9-432e-9eb4-c584b4140663" /> | <img width="400" alt="Performnance analysis" src="https://github.com/user-attachments/assets/468b0acb-733e-4891-8e16-781c47889017" /> |
109
109
 
110
110
  | Performance charts | |
111
111
  |-----------------------------------------------|------------------------------------------|
112
- | <img width="400" alt="Performance analysis" src="https://github.com/user-attachments/assets/03f64a7a-262a-4b2a-a0b6-c70f3f14705c" /> | <img width="400" alt="" src="https://github.com/user-attachments/assets/224d14f7-c7b3-4f4c-99c6-528c800b1a84" /> |
112
+ | <img width="400" alt="Performance charts" src="https://github.com/user-attachments/assets/23e6b1d0-2def-490a-8921-3477ded2c8ce" /> | <img width="400" alt="Performance charts" src="https://github.com/user-attachments/assets/ee620711-10be-4582-ab6f-635c896a1101" /> |
113
113
 
114
114
  | NPE | |
115
115
  |-----------------------------------------------|------------------------------------------|
116
- | <img width="400" alt="NPE" src="https://github.com/user-attachments/assets/1f4441c6-9d9e-4834-9e71-edec1955243c" /> | <img width="400" alt="NPE" src="https://github.com/user-attachments/assets/7159992e-7691-41cf-a152-8bc6a3606ade" /> |
116
+ | <img width="400" alt="NPE" src="https://github.com/user-attachments/assets/5a25a79c-0fcc-4298-aaae-8e8656b99ab5" /> | <img width="400" alt="NPE" src="https://github.com/user-attachments/assets/b2845948-02fa-4255-81f4-fc3257a282da" />
117
+
117
118
 
118
119
  ## Sample reports
119
120
 
{ttnn_visualizer-0.27.3.dist-info → ttnn_visualizer-0.28.0.dist-info}/RECORD RENAMED
@@ -10,25 +10,25 @@ ttnn_visualizer/models.py,sha256=fsokoPLzjTLlaaG0gi5xzRWBhuCmvUOgeQFMpc1ocPc,758
10
10
  ttnn_visualizer/queries.py,sha256=oifyj_ROmQwUkQCFO1SteuphaoZbROcnizG1ix-1r3U,13616
11
11
  ttnn_visualizer/remote_sqlite_setup.py,sha256=Zd7U7q_N92rD0P2qb1GIXuZjhV4LXqfq7Bhg0MTLS5k,3269
12
12
  ttnn_visualizer/requirements.txt,sha256=LiycuPs-Oh8PZ1qM6v1KlUPQG86nQyQFz_x7dU8DSpA,370
13
- ttnn_visualizer/serializers.py,sha256=smY3CgbqaAXsovoYZ-A9fgPXih5lUS2HD_3aIZMcFS4,7821
14
- ttnn_visualizer/sessions.py,sha256=MSVsc0lX2bZGmoBG92YDjRfEjApBtGi5pLrpoaCzvCs,9044
15
- ttnn_visualizer/settings.py,sha256=kC0mBNrZDq1S7RPeRGnPcGZxGuM515A1upsWRrKl5k8,3841
13
+ ttnn_visualizer/serializers.py,sha256=w7IKTetpZHmTPqurxhVSQk9VXakqaL4O1XQMezUgt3g,8030
14
+ ttnn_visualizer/sessions.py,sha256=PI_FXY-b0jRWutQVv4DOonPa30MemXU-hhRHfcIUe5E,8863
15
+ ttnn_visualizer/settings.py,sha256=oKjsPgk9PcPM5sBGgtPW30nOcW72Tti-JVS05rHS3ck,3890
16
16
  ttnn_visualizer/sftp_operations.py,sha256=RrzCGo__2sQ2g4r90qlyFBEq4v6ooX2ntJ7VDKuynC0,17659
17
17
  ttnn_visualizer/sockets.py,sha256=o0oTG26LCjTJL8ajHR28ZBp9Z8RfWDehaS8Orns1HoQ,3624
18
18
  ttnn_visualizer/ssh_client.py,sha256=KRLuIk6wxrZZQUQKfC8QWMhXGJQvfKKeyxLThgRX6ak,2679
19
19
  ttnn_visualizer/utils.py,sha256=zD8_B7Dd8sgOQF52LNQ5HrLxuvfZg-YnxLrZOpuA5Lw,6372
20
- ttnn_visualizer/views.py,sha256=nYpWsXnBG6yQEoBYcQk-nrdDFUIsYiYoV8ibSw1h960,27986
20
+ ttnn_visualizer/views.py,sha256=z2EbnX1cIkwMHDHXuacklQvdkjOJUbjJuYh_W2Lzvhg,28595
21
21
  ttnn_visualizer/bin/docker-entrypoint-web,sha256=uuv6aubpMCfOcuvDBxwBDITE8PN39teuwyJ2zA5KWuw,413
22
22
  ttnn_visualizer/bin/pip3-install,sha256=nbSRT4GfJQIQ9KTNO3j-6b5WM4lrx9XA4GBlAURRMws,502
23
- ttnn_visualizer/static/index.html,sha256=wRBO5UpcdbjedKmNHdb594L3TWscLv5QX-vT7xEcvts,923
24
- ttnn_visualizer/static/assets/allPaths-DhoJy5jM.js,sha256=NSfcM6pI0RNkGesTm5sD56t0d3WlbXQ6DQv-VCdBJ6g,309
25
- ttnn_visualizer/static/assets/allPathsLoader-B4SALZ6P.js,sha256=OQxQ4ba4PjWQSlMtLD-BA9suiK1bLDDLY7z9KAX4wmo,550
26
- ttnn_visualizer/static/assets/index-BME2kYRR.js,sha256=6edFFfpx53bMtYogaCWjK8lTJIpcnEI61d3pe4xrFAg,6985407
27
- ttnn_visualizer/static/assets/index-BTfoVg9a.css,sha256=3Ft7KsW9G6k2DYkHJvRwadGpYzSjTPVBAetC0r5xnVQ,615889
23
+ ttnn_visualizer/static/index.html,sha256=AjLxPLvSwTWXP3Mv4Nav3vjd0RgeYDQ9T3Hn7OZ7wuc,923
24
+ ttnn_visualizer/static/assets/allPaths-DGdUWICT.js,sha256=9nF2h2lGmWkA605ZBltxNKpYV6xTGMzuSVPtoBEfDs8,309
25
+ ttnn_visualizer/static/assets/allPathsLoader-BSYo-ZTH.js,sha256=9cSsk2yZOpbRyX3T2D8Ba6AnrIIGTbwmN2foK0Pp2mw,550
28
26
  ttnn_visualizer/static/assets/index-BVMreIQm.js,sha256=QJTBb4VVCMoLPYsWdru3heJX1VtMJQYJGx-tq8gZNTw,280965
27
+ ttnn_visualizer/static/assets/index-CYZZ70nJ.js,sha256=uQKNNCRxErW1Rp4LCNFEn_AM78JijCw7o-0j-e8VOKg,6988295
28
+ ttnn_visualizer/static/assets/index-DWVfq2z5.css,sha256=_kltqYll0FISXlP53fbRNWCPQ_e3ThO4ipLIWKK-6XE,616005
29
29
  ttnn_visualizer/static/assets/index-Do7YB6C4.js,sha256=10jCIy7zph8mPB2htGfhXJBV7LO2FFrGhfz7xoQyh00,289378
30
30
  ttnn_visualizer/static/assets/site-BTBrvHC5.webmanifest,sha256=Uy_XmnGuYFVf-OZuma2NvgEPdrCrevb3HZvaxSIHoA0,456
31
- ttnn_visualizer/static/assets/splitPathsBySizeLoader-B2PjTVfS.js,sha256=YhpJ9W2wkV2Y7cvTOCZMg-EbEufzf_5Iiu8LFexXvrc,472
31
+ ttnn_visualizer/static/assets/splitPathsBySizeLoader-hHE0y-FQ.js,sha256=8mIzns66ubh0WQYdEESOa7umZY9CXorcd6DkIhXydKo,472
32
32
  ttnn_visualizer/static/favicon/android-chrome-192x192.png,sha256=BZWA09Zxaa3fXbaeS6nhWo2e-DUSjm9ElzNQ_xTB5XU,6220
33
33
  ttnn_visualizer/static/favicon/android-chrome-512x512.png,sha256=HBiJSZyguB3o8fMJuqIGcpeBy_9JOdImme3wD02UYCw,62626
34
34
  ttnn_visualizer/static/favicon/favicon-32x32.png,sha256=Zw201qUsczQv1UvoQvJf5smQ2ss10xaTeWxmQNYCGtY,480
@@ -37,10 +37,10 @@ ttnn_visualizer/static/sample-data/cluster-desc.yaml,sha256=LMxOmsRUXtVVU5ogzYkX
37
37
  ttnn_visualizer/tests/__init__.py,sha256=qn9AwfHTxYYK0Icz-q2Q41Xo9ET6oc6anZzgh02PhK4,94
38
38
  ttnn_visualizer/tests/test_queries.py,sha256=OoINMZezyJkpiQuM1Owp4DlMoJGTl6OgsUbPa2Rxqc0,16546
39
39
  ttnn_visualizer/tests/test_serializers.py,sha256=966AJkXLAwzsceSQ9QR-Sy7VrEbE71AtfMF3y9ZIIWc,18490
40
- ttnn_visualizer-0.27.3.dist-info/LICENSE,sha256=7_uV4foXIbLyroI1M6NdIySajtLuPwayZr1UN2ItErI,11353
41
- ttnn_visualizer-0.27.3.dist-info/LICENSE_understanding.txt,sha256=pymi-yb_RvYM9p2ZA4iSNsImcvhDBBxlGuJCY9dTq7M,233
42
- ttnn_visualizer-0.27.3.dist-info/METADATA,sha256=9jgDIEIiWPJ-BK6m22joce4ivTJC9ONB8ldcbegN8KM,7297
43
- ttnn_visualizer-0.27.3.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
44
- ttnn_visualizer-0.27.3.dist-info/entry_points.txt,sha256=QpuUpkmQ_mEHJTMqOBdU0MH2Z4WF_9iFsGACeyyAO1E,61
45
- ttnn_visualizer-0.27.3.dist-info/top_level.txt,sha256=M1EGkvDOuIfbhDbcUdz2-TSdmCtDoQ2Uyag9k5JLDSY,16
46
- ttnn_visualizer-0.27.3.dist-info/RECORD,,
40
+ ttnn_visualizer-0.28.0.dist-info/LICENSE,sha256=7_uV4foXIbLyroI1M6NdIySajtLuPwayZr1UN2ItErI,11353
41
+ ttnn_visualizer-0.28.0.dist-info/LICENSE_understanding.txt,sha256=pymi-yb_RvYM9p2ZA4iSNsImcvhDBBxlGuJCY9dTq7M,233
42
+ ttnn_visualizer-0.28.0.dist-info/METADATA,sha256=lrE0I1wjCNDmkG6y1kir6JNkfkMp760zwPCuJ9KRqkE,7317
43
+ ttnn_visualizer-0.28.0.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
44
+ ttnn_visualizer-0.28.0.dist-info/entry_points.txt,sha256=QpuUpkmQ_mEHJTMqOBdU0MH2Z4WF_9iFsGACeyyAO1E,61
45
+ ttnn_visualizer-0.28.0.dist-info/top_level.txt,sha256=M1EGkvDOuIfbhDbcUdz2-TSdmCtDoQ2Uyag9k5JLDSY,16
46
+ ttnn_visualizer-0.28.0.dist-info/RECORD,,