Skip to content

Commit a9c0ffb

Browse files
Isotr0pyAkshat-Tripathi
authored andcommitted
[Bugfix][CPU] Fix cpu all-reduce using native pytorch implementation (vllm-project#13586)
1 parent 0161781 commit a9c0ffb

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vllm/distributed/device_communicators/cpu_communicator.py

Lines changed: 2 additions & 1 deletion
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)