Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Core][Distributed] use cpu group to broadcast metadata in cpu #4444

Merged
merged 11 commits into from
Apr 29, 2024

Conversation

youkaichao
Copy link
Member

As discussed in #4440 , broadcasting metadata through cpu is better, because it avoids moving data back and forth between cpu and gpu.

vllm/distributed/communication_op.py Outdated Show resolved Hide resolved
tests/worker/test_model_runner.py Outdated Show resolved Hide resolved
@youkaichao youkaichao merged commit f4f921b into vllm-project:main Apr 29, 2024
48 checks passed
@youkaichao youkaichao deleted the broadcast_cpu branch April 29, 2024 20:52
z103cb pushed a commit to z103cb/opendatahub_vllm that referenced this pull request May 7, 2024
mawong-amd added a commit to ROCm/vllm that referenced this pull request Jun 4, 2024
Partially reverts [Core][Distributed] use cpu group to broadcast
metadata in cpu (vllm-project#4444)
mawong-amd added a commit to ROCm/vllm that referenced this pull request Jun 4, 2024
Partially reverts [Core][Distributed] use cpu group to broadcast
metadata in cpu (vllm-project#4444)
shaojiewang pushed a commit to shaojiewang/vllm-rocm that referenced this pull request Jul 3, 2024
Partially reverts [Core][Distributed] use cpu group to broadcast
metadata in cpu (vllm-project/vllm#4444)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants