Skip to content

Commit 9f1619d

Browse files
committed
add quickreduce
Signed-off-by: Haoyang Li <[email protected]>
1 parent f4a2562 commit 9f1619d

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

vllm/distributed/device_communicators/quick_all_reduce.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -65,11 +65,9 @@ def __init__(self,
6565
"""
6666
self._IS_CAPTURING = False
6767
self.disabled = True
68-
assert \
69-
envs.VLLM_QUICK_ALLREDUCE in QuickAllreduce._SUPPORTED_LEVEL, (
68+
assert envs.VLLM_QUICK_ALLREDUCE in QuickAllreduce._SUPPORTED_LEVEL, (
7069
"quick allreduce level must be in [0, 1, 2, 3, 4, 5], "
71-
f"but got {envs.VLLM_QUICK_ALLREDUCE}"
72-
)
70+
f"but got {envs.VLLM_QUICK_ALLREDUCE}")
7371

7472
if not quick_ar:
7573
# disable because of missing quick allreduce library

0 commit comments

Comments
 (0)