Update changes from #6443

microsoft · Aug 27, 2024 · 37b3dd7 · 37b3dd7
1 parent eb37cac
commit 37b3dd7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/comm/ccl.py b/deepspeed/comm/ccl.py
@@ -15,7 +15,7 @@
 
 def build_ccl_op():
     builder = get_accelerator().create_op_builder("CCLCommBuilder")
-    if builder is None or NotImplementedBuilder:
+    if builder is None or isinstance(builder, NotImplementedBuilder):
         return None
     ccl_cpp_module = builder.load()
     print(f'DeepSpeed {builder.absolute_name()} built successfully')