Merge pull request #8089 from jemiryguo/fix

Fix#7981, Update _nccl_comm.py
cupy · Jan 16, 2024 · 02b69f0 · 02b69f0
2 parents 1737214 + 49030c8
commit 02b69f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cupyx/distributed/_nccl_comm.py b/cupyx/distributed/_nccl_comm.py
@@ -124,7 +124,7 @@ def _get_stream(self, stream):
     def _get_op(self, op, dtype):
         if op not in _nccl_ops:
             raise RuntimeError(f'Unknown op {op} for NCCL')
-        if dtype in 'FD' and op != nccl.NCCL_SUM:
+        if dtype in 'FD' and op != 'sum':
             raise ValueError(
                 'Only nccl.SUM is supported for complex arrays')
         return _nccl_ops[op]