update custom_all_reduce

2c560dc5 · zhuwenwen · d4df43b0 · 2c560dc5
Commit 2c560dc5 authored Jan 16, 2026 by zhuwenwen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 4 deletions

vllm/distributed/device_communicators/custom_all_reduce.py vllm/distributed/device_communicators/custom_all_reduce.py +1 -4

No files found.
--- a/vllm/distributed/device_communicators/custom_all_reduce.py
+++ b/vllm/distributed/device_communicators/custom_all_reduce.py
@@ -271,10 +271,7 @@ class CustomAllreduce:
                if envs.VLLM_CUSTOM_CACHE:
                    return self.all_reduce(input, registered=True)
                else:
-                    if not self.fully_connected:
+                    return self.all_reduce(input, registered=False)
-                        return self.all_reduce(input, registered=False)
-                    else:
-                        return self.all_reduce(input, registered=True)
            else:
                # If warm up, mimic the allocation pattern since custom
                # allreduce is out-of-place.