Alex Cheema 1 year ago
parent
commit
e18549e9e1
1 changed files with 0 additions and 1 deletions
  1. 0 1
      exo/viz/topology_viz.py

+ 0 - 1
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