add debug cruft

Signed-off-by: Tyler Michael Smith <tysmith@redhat.com>
This commit is contained in:
Tyler Michael Smith
2025-06-20 20:37:37 +00:00
parent 850dafea92
commit fcec8c8827

View File

@ -143,6 +143,24 @@ class DeepEPAll2AllManagerBase(All2AllManagerBase):
device = self.dp_group.device
props = torch.cuda.get_device_properties(device)
self.num_sms = props.multi_processor_count
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
print(f"Setting num sms to {self.num_sms}")
def get_handle(self, kwargs):
raise NotImplementedError