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

Disable chunked prefill and/or prefix caching when MLA is enabled #12642

Merged
merged 4 commits into from
Feb 1, 2025

Conversation

simon-mo
Copy link
Collaborator

@simon-mo simon-mo commented Feb 1, 2025

From @mgoin in #12638

I cannot push to that branch, therefore a new PR to unblock release.

Copy link

github-actions bot commented Feb 1, 2025

👋 Hi! Thank you for contributing to the vLLM project.
Just a reminder: PRs would not trigger full CI run by default. Instead, it would only run fastcheck CI which starts running only a small and essential subset of CI tests to quickly catch errors. You can run other CI tests on top of those by going to your fastcheck build on Buildkite UI (linked in the PR checks section) and unblock them. If you do not have permission to unblock, ping simon-mo or khluu to add you in our Buildkite org.

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 do one of these:

  • Add ready label to the PR
  • Enable auto-merge.

🚀

Signed-off-by: simon-mo <[email protected]>
@mgoin mgoin added the ready ONLY add when PR is ready to merge/full CI is needed label Feb 1, 2025
@mgoin mgoin enabled auto-merge (squash) February 1, 2025 06:02
Comment on lines +3232 to +3233
self.scheduler_config.enable_chunked_prefill = False
self.scheduler_config.chunked_prefill_enabled = False
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I never knew we had these two different flags, what's the difference?

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

idk

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

😅

@simon-mo simon-mo disabled auto-merge February 1, 2025 07:46
@simon-mo simon-mo merged commit 4f4d427 into vllm-project:main Feb 1, 2025
34 of 46 checks passed
Isotr0py pushed a commit to Isotr0py/vllm that referenced this pull request Feb 2, 2025
…lm-project#12642)

From @mgoin in vllm-project#12638

I cannot push to that branch, therefore a new PR to unblock release.

---------

Signed-off-by: mgoin <[email protected]>
Signed-off-by: simon-mo <[email protected]>
Co-authored-by: mgoin <[email protected]>
Signed-off-by: Isotr0py <[email protected]>
youngkent pushed a commit to youngkent/vllm that referenced this pull request Feb 3, 2025
…lm-project#12642)

From @mgoin in vllm-project#12638

I cannot push to that branch, therefore a new PR to unblock release.

---------

Signed-off-by: mgoin <[email protected]>
Signed-off-by: simon-mo <[email protected]>
Co-authored-by: mgoin <[email protected]>
srikanthsrnvs pushed a commit to srikanthsrnvs/vllm that referenced this pull request Feb 3, 2025
…lm-project#12642)

From @mgoin in vllm-project#12638

I cannot push to that branch, therefore a new PR to unblock release.

---------

Signed-off-by: mgoin <[email protected]>
Signed-off-by: simon-mo <[email protected]>
Co-authored-by: mgoin <[email protected]>
Signed-off-by: Srikanth Srinivas <[email protected]>
sahelib25 pushed a commit to krai/vllm that referenced this pull request Feb 3, 2025
…lm-project#12642)

From @mgoin in vllm-project#12638

I cannot push to that branch, therefore a new PR to unblock release.

---------

Signed-off-by: mgoin <[email protected]>
Signed-off-by: simon-mo <[email protected]>
Co-authored-by: mgoin <[email protected]>
fxmarty-amd pushed a commit to fxmarty-amd/vllm that referenced this pull request Feb 7, 2025
…lm-project#12642)

From @mgoin in vllm-project#12638

I cannot push to that branch, therefore a new PR to unblock release.

---------

Signed-off-by: mgoin <[email protected]>
Signed-off-by: simon-mo <[email protected]>
Co-authored-by: mgoin <[email protected]>
Signed-off-by: Felix Marty <[email protected]>
NickLucche pushed a commit to NickLucche/vllm that referenced this pull request Feb 7, 2025
…lm-project#12642)

From @mgoin in vllm-project#12638

I cannot push to that branch, therefore a new PR to unblock release.

---------

Signed-off-by: mgoin <[email protected]>
Signed-off-by: simon-mo <[email protected]>
Co-authored-by: mgoin <[email protected]>
@siddartha-RE
Copy link

Why was this change required. We are concerned we are seeing instability in the VLLM server with chunked prefill disabled. We are observing the server hangs when it receives a large prompt while other requests are in flight. Thank you.

@mgoin
Copy link
Member

mgoin commented Feb 9, 2025

@siddartha-RE MLA does not supported chunked prefill yet, so it will error if you try to run it with that enabled. We are working on enabling this ASAP

@siddartha-RE
Copy link

Thank you for the explanation. We are testing on 16x GH200 with 96GB per GPU. Let me know if you would like me to verify / benchmark anything on this configuration.

ShangmingCai pushed a commit to ShangmingCai/vllm that referenced this pull request Feb 10, 2025
…lm-project#12642)

From @mgoin in vllm-project#12638

I cannot push to that branch, therefore a new PR to unblock release.

---------

Signed-off-by: mgoin <[email protected]>
Signed-off-by: simon-mo <[email protected]>
Co-authored-by: mgoin <[email protected]>
GWS0428 pushed a commit to GWS0428/VARserve that referenced this pull request Feb 12, 2025
…lm-project#12642)

From @mgoin in vllm-project#12638

I cannot push to that branch, therefore a new PR to unblock release.

---------

Signed-off-by: mgoin <[email protected]>
Signed-off-by: simon-mo <[email protected]>
Co-authored-by: mgoin <[email protected]>
panf2333 pushed a commit to yottalabsai/vllm that referenced this pull request Feb 18, 2025
…lm-project#12642)

From @mgoin in vllm-project#12638

I cannot push to that branch, therefore a new PR to unblock release.

---------

Signed-off-by: mgoin <[email protected]>
Signed-off-by: simon-mo <[email protected]>
Co-authored-by: mgoin <[email protected]>
kerthcet pushed a commit to kerthcet/vllm that referenced this pull request Feb 21, 2025
…lm-project#12642)

From @mgoin in vllm-project#12638

I cannot push to that branch, therefore a new PR to unblock release.

---------

Signed-off-by: mgoin <[email protected]>
Signed-off-by: simon-mo <[email protected]>
Co-authored-by: mgoin <[email protected]>
lk-chen pushed a commit to lk-chen/vllm that referenced this pull request Mar 5, 2025
…lm-project#12642)

From @mgoin in vllm-project#12638

I cannot push to that branch, therefore a new PR to unblock release.

---------

Signed-off-by: mgoin <[email protected]>
Signed-off-by: simon-mo <[email protected]>
Co-authored-by: mgoin <[email protected]>
Signed-off-by: Linkun Chen <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ready ONLY add when PR is ready to merge/full CI is needed
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants