diff --git a/vllm/distributed/device_communicators/cpu_communicator.py b/vllm/distributed/device_communicators/cpu_communicator.py index 4e86396e7135..b920cd7e1acf 100644 --- a/vllm/distributed/device_communicators/cpu_communicator.py +++ b/vllm/distributed/device_communicators/cpu_communicator.py @@ -30,4 +30,5 @@ def __init__(self, pass def all_reduce(self, input_): - return self.dist_module.all_reduce(input_, group=self.device_group) + self.dist_module.all_reduce(input_, group=self.device_group) + return input_