Skip to content

Commit

Permalink
Remove innermost tqdm
Browse files Browse the repository at this point in the history
It flickers past too fast anyhow.
  • Loading branch information
Waino committed Dec 18, 2023
1 parent 3f1f8b9 commit 6bb5434
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tools/gpu_assignment.py
Original file line number Diff line number Diff line change
Expand Up @@ -346,7 +346,7 @@ def _split_lps_cost(self, assignment: Assignment) -> float:
def best_swap_for(self, slot_a: GpuSlot, assignment, current_cost, slot_subset=None):
slot_subset = self.gpu_slots if slot_subset is None else slot_subset
costs = [(current_cost, slot_a)]
for i, slot_b in enumerate(tqdm(slot_subset, desc='best_swap_for', leave=False)):
for i, slot_b in enumerate(slot_subset):
if slot_a.node == slot_b.node and slot_a.gpu == slot_b.gpu:
# No point swapping pairs already on the same device
continue
Expand Down

0 comments on commit 6bb5434

Please sign in to comment.