We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 839b27c commit b2c3fc5Copy full SHA for b2c3fc5
vllm/distributed/device_communicators/cpu_communicator.py
@@ -30,4 +30,5 @@ def __init__(self,
30
pass
31
32
def all_reduce(self, input_):
33
- return self.dist_module.all_reduce(input_, group=self.device_group)
+ self.dist_module.all_reduce(input_, group=self.device_group)
34
+ return input_
0 commit comments