-
-
Notifications
You must be signed in to change notification settings - Fork 11.5k
[Bugfix][CUDA] fixes CUDA FP8 kv cache dtype supported #21420
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
[Bugfix][CUDA] fixes CUDA FP8 kv cache dtype supported #21420
Conversation
Signed-off-by: elvischenv <[email protected]>
|
👋 Hi! Thank you for contributing to the vLLM project. 💬 Join our developer Slack at https://slack.vllm.ai to discuss your PR in #pr-reviews, coordinate on features in #feat- channels, or join special interest groups in #sig- channels. Just a reminder: PRs would not trigger full CI run by default. Instead, it would only run Once the PR is approved and ready to go, your PR reviewer(s) can run CI to test the changes comprehensively before merging. To run CI, PR reviewers can either: Add 🚀 |
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
The pull request correctly moves the is_kv_cache_dtype_supported method to the CudaPlatformBase class to be shared by subclasses. However, I found a logical issue in the method itself that could lead to incorrect behavior. I've provided a suggestion to fix this and improve the clarity of the code.
|
@mgoin @DarkLight1337 @pavanimajety for viz. Thanks. |
mgoin
left a comment
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.
Thanks
…21420) Signed-off-by: elvischenv <[email protected]> Signed-off-by: qizixi <[email protected]>
…21420) Signed-off-by: elvischenv <[email protected]> Signed-off-by: x22x22 <[email protected]>
…21420) Signed-off-by: elvischenv <[email protected]>
…21420) Signed-off-by: elvischenv <[email protected]>
…21420) Signed-off-by: elvischenv <[email protected]> Signed-off-by: Jinzhen Lin <[email protected]>
…21420) Signed-off-by: elvischenv <[email protected]> Signed-off-by: Paul Pak <[email protected]>
…21420) Signed-off-by: elvischenv <[email protected]> Signed-off-by: Diego-Castan <[email protected]>
…21420) Signed-off-by: elvischenv <[email protected]>
Essential Elements of an Effective PR Description Checklist
supported_models.mdandexamplesfor a new model.Purpose
Previous PR(#21302) defined a class method
is_kv_cache_dtype_supported()underNonNvmlCudaPlatformclass, butNvmlCudaPlatformalso needs this method.So this PR moved the method to the parent class
CudaPlatformBase.Test Plan
Test Result
(Optional) Documentation Update