fix a typo

OpenBMB · Apr 22, 2024 · 151f679 · 151f679
1 parent 5dad728
commit 151f679
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bmtrain/synchronize.py b/bmtrain/synchronize.py
@@ -25,7 +25,7 @@ def wait_loader():
     config['calc_stream'].record_event(config['load_event'])
 
 
-def sum_loss(loss : torch.Tensor, comm: Optional[nccl.Communicator] = None):
+def sum_loss(loss : torch.Tensor, comm: Optional[nccl.NCCLCommunicator] = None):
     """
     Sum the loss across all workers.