From 00dd34e37cee5b79fe776098d1bc9e826a6fb789 Mon Sep 17 00:00:00 2001 From: Yizhou Wang Date: Tue, 14 May 2024 07:06:21 +0000 Subject: [PATCH] fix comm/ccl.py return issue --- deepspeed/comm/ccl.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/deepspeed/comm/ccl.py b/deepspeed/comm/ccl.py index f2a6cb6a36dc..cdf4c030f5d6 100644 --- 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')