Wrap nccl.get_unique_id in try/except (#549)

ptycho · Mar 21, 2024 · 36b9370 · 36b9370
1 parent c601059
commit 36b9370
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ptypy/accelerate/cuda_cupy/multi_gpu.py b/ptypy/accelerate/cuda_cupy/multi_gpu.py
@@ -98,7 +98,10 @@ def __init__(self):
         # get a unique identifier for the NCCL communicator and
         # broadcast it to all MPI processes (assuming one device per process)
         if self.rank == 0:
-            self.id = nccl.get_unique_id()
+            try:
+                self.id = nccl.get_unique_id()
+            except:
+                self.id = None
         else:
             self.id = None