rm print

e18549e9e1daa7b0dac951cde60d19350c3180ee · 2024-07-19 16:52:13 -0700 · Alex Cheema

Files touched

M exo/viz/topology_viz.py

Diff

commit e18549e9e1daa7b0dac951cde60d19350c3180ee
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Fri Jul 19 16:52:13 2024 -0700

    rm print
---
 exo/viz/topology_viz.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/exo/viz/topology_viz.py b/exo/viz/topology_viz.py
index a97b8298..b2016052 100644
--- a/exo/viz/topology_viz.py
+++ b/exo/viz/topology_viz.py
@@ -74,7 +74,6 @@ class TopologyViz:
         # Calculate total FLOPS and position on the bar
         total_flops = sum(self.topology.nodes.get(partition.node_id, UNKNOWN_DEVICE_CAPABILITIES).flops.fp16 for partition in self.partitions)
         bar_pos = (math.tanh(total_flops / 20 - 2) + 1) / 2
-        print(f"{bar_pos=}")
 
         # Add GPU poor/rich bar
         bar_width = 30  # Increased bar width

← 0c5a927f spacing in viz · back to Exo · fix inference_state serialization. related: #40 #44 #45 1475c735 →