-
-
Notifications
You must be signed in to change notification settings - Fork 10.4k
[Misc] Move CpuGpuBuffer to vllm/v1/utils.py #23728
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
Conversation
Signed-off-by: Woosuk Kwon <[email protected]>
Signed-off-by: Woosuk Kwon <[email protected]>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Code Review
This pull request aims to move the CpuGpuBuffer
class to a more general utility file, vllm/v1/utils.py
, and updates imports accordingly. This is a good refactoring. However, the change appears to be incomplete as the original CpuGpuBuffer
class is still present in vllm/v1/worker/utils.py
, leading to code duplication. The original class should be removed as part of this move. My review also includes a specific comment to correct the return type hints in the moved class for better code correctness.
Signed-off-by: Woosuk Kwon <[email protected]>
Signed-off-by: Woosuk Kwon <[email protected]>
Signed-off-by: Woosuk Kwon <[email protected]> Signed-off-by: Xiao Yu <[email protected]>
Signed-off-by: Woosuk Kwon <[email protected]>
Signed-off-by: Woosuk Kwon <[email protected]>
Signed-off-by: Woosuk Kwon <[email protected]>
No description provided.