Skip to content

Commit b2c3fc5

Browse files
authored
[Bugfix][CPU] Fix cpu all-reduce using native pytorch implementation (vllm-project#13586)
1 parent 839b27c commit b2c3fc5

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vllm/distributed/device_communicators/cpu_communicator.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -30,4 +30,5 @@ def __init__(self,
3030
pass
3131

3232
def all_reduce(self, input_):
33-
return self.dist_module.all_reduce(input_, group=self.device_group)
33+
self.dist_module.all_reduce(input_, group=self.device_group)
34+
return input_

0 commit comments

Comments
 (0)