fix comm/ccl.py return issue

microsoft · May 14, 2024 · 00dd34e · 00dd34e
1 parent 86abd39
commit 00dd34e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deepspeed/comm/ccl.py b/deepspeed/comm/ccl.py
@@ -8,13 +8,14 @@
 
 import torch
 from deepspeed.accelerator import get_accelerator
+from deepspeed.ops.op_builder import NotImplementedBuilder
 from .reduce_op import ReduceOp
 from .torch import TorchBackend
 
 
 def build_ccl_op():
     builder = get_accelerator().create_op_builder("CCLCommBuilder")
-    if builder is None:
+    if builder is None or NotImplementedBuilder:
         return None
     ccl_cpp_module = builder.load()
     print(f'DeepSpeed {builder.absolute_name()} built successfully')