cleaner node info =

a2a7ca1f8bbf8c36e92770458748b10c3b0987a3 · 2024-07-20 12:33:38 -0700 · Alex Cheema

Files touched

M exo/viz/topology_viz.py

Diff

commit a2a7ca1f8bbf8c36e92770458748b10c3b0987a3
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Sat Jul 20 12:33:38 2024 -0700

    cleaner node info =
---
 exo/viz/topology_viz.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/exo/viz/topology_viz.py b/exo/viz/topology_viz.py
index b2016052..f3f68526 100644
--- a/exo/viz/topology_viz.py
+++ b/exo/viz/topology_viz.py
@@ -117,12 +117,11 @@ class TopologyViz:
             else:
                 visualization[y][x] = '🔵'  # Blue circle for inactive nodes
 
-            # Place node info (ID, start_layer, end_layer)
+            # Place node info (model, memory, TFLOPS, partition) on three lines
             node_info = [
-                f"Model: {device_capabilities.model}",
-                f"Mem: {device_capabilities.memory // 1024}GB",
-                f"FLOPS: {device_capabilities.flops.fp16}T",
-                f"Part: {partition.start:.2f}-{partition.end:.2f}"
+                f"{device_capabilities.model} {device_capabilities.memory // 1024}GB",
+                f"{device_capabilities.flops.fp16}TFLOPS",
+                f"[{partition.start:.2f}-{partition.end:.2f}]"
             ]
 
             # Calculate info position based on angle

← 04f2aa2a try with METAL_XCODE=1 for tinygrad metal · back to Exo · add space between outputs in github workflow integration tes 774e6209 →