Update

Deep-Learning-Profiling-Tools · Nov 8, 2024 · 191ba18 · 191ba18
1 parent 434fa20
commit 191ba18
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/triton_viz/draw.py b/triton_viz/draw.py
@@ -139,7 +139,7 @@ def draw_record(x):
     dr = dr.center_xy()
     env = dr.get_envelope()
     dr = rectangle(env.width + 1, env.height + 1).fill_color(BG).center_xy() + dr
-    dr.render_svg(base, 2500)
+    dr.render(base, 2500)
     return env.width, env.height
 
 

diff --git a/triton_viz/interface.py b/triton_viz/interface.py
@@ -16,7 +16,7 @@ def launch(share=True):
         m[0] = max(k[0] + 1, m[0])
         m[1] = max(k[1] + 1, m[1])
         m[2] = max(k[2] + 1, m[2])
-    w, h = triton_viz.draw_record(program_records[(0, 0, 0)], tt, "tmp.svg")
+    w, h = triton_viz.draw_record(program_records[(0, 0, 0)], tt, "tmp.png")
     size[0] = w
     size[1] = h
     height = 600 * size[1] / size[0]
@@ -52,7 +52,7 @@ def launch(share=True):
                     )
 
         def cache_block(idx):
-            name = tempfile.NamedTemporaryFile(suffix=".svg")
+            name = tempfile.NamedTemporaryFile(suffix=".png")
             w, h = triton_viz.draw_record(program_records[idx], tt, name.name)
             size[0] = w
             size[1] = h