diff --git a/deepspeed/comm/ccl.py b/deepspeed/comm/ccl.py index cdf4c030f5d6..f0296821d235 100644 --- 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')