Skip to content

Commit 3458abc

Browse files
committed
rebase
Signed-off-by: Haoyang Li <Haoyang.Li@amd.com>
1 parent bb8f513 commit 3458abc

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vllm/distributed/device_communicators/quick_all_reduce.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,8 @@ def __init__(self,
139139
"specify disable_quick_all_reduce=0 explicitly.")
140140
return
141141

142-
self.max_size = max_size if envs.VLLM_QUICK_ALLREDUCE > 0 else max_size / self.world_size * 2
142+
self.max_size = (max_size if envs.VLLM_QUICK_ALLREDUCE > 0 else
143+
max_size / self.world_size * 2)
143144
self.min_size = min_size
144145
self._ptr = ops.init_quick_ar(world_size, rank)
145146
my_handle = ops.qr_get_comm_handle(self._ptr)

0 commit comments

Comments
 (0)