Skip to content

Commit f68829f

Browse files
maleksan85maleksan85
andauthored
doubled size to wa issue and preserve CAR perf (ROCm#510)
Signed-off-by: maleksan85 <[email protected]> Co-authored-by: maleksan85 <[email protected]>
1 parent 97b78bf commit f68829f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/distributed/device_communicators/custom_all_reduce.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ class CustomAllreduce:
5353
def __init__(self,
5454
group: ProcessGroup,
5555
device: Union[int, str, torch.device],
56-
max_size=8192 * 1024) -> None:
56+
max_size=2 * 8192 * 1024) -> None:
5757
"""
5858
Args:
5959
group: the process group to work on. If None, it will use the

0 commit comments

Comments
 (0)