fix condition issue in ccl.py

microsoft · Aug 26, 2024 · ec32a0a · ec32a0a
1 parent e2654bf
commit ec32a0a
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 builder is NotImplementedBuilder:
         return None
     ccl_cpp_module = builder.load()
     print(f'DeepSpeed {builder.absolute_name()} built successfully')