Skip to content

[BugFix] Patch inductor memory plan logic#26878

Merged
DarkLight1337 merged 5 commits intovllm-project:mainfrom
BoyuanFeng:bf/partition-multi-graph-test
Oct 15, 2025
Merged

[BugFix] Patch inductor memory plan logic#26878
DarkLight1337 merged 5 commits intovllm-project:mainfrom
BoyuanFeng:bf/partition-multi-graph-test

Conversation

@BoyuanFeng
Copy link
Copy Markdown
Collaborator

This PR fixes a test failure for test_multi_graph_piecewise_compile. This is a pytorch issue and is fixed in pytorch/pytorch#165514. This PR monkey patches vLLM for PyTorch-2.9.0 only to unblock.

Signed-off-by: Boyuan Feng <boyuan@meta.com>
Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist bot left a 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 introduces a monkey patch for a PyTorch 2.9.0 issue related to inductor's memory planning, which was causing test_multi_graph_piecewise_compile to fail. The patch is applied conditionally for PyTorch 2.9.0. The changes also include enabling the failing test for PyTorch versions 2.9.0.dev and newer, assuming the fix is present in those versions. While the overall approach is sound for a temporary workaround, I've found a critical issue in the version comparison logic that could lead to the patch being applied to incorrect PyTorch versions.

Signed-off-by: Boyuan Feng <boyuan@meta.com>
Signed-off-by: Boyuan Feng <boyuan@meta.com>
ProExpertProg added a commit to neuralmagic/vllm that referenced this pull request Oct 15, 2025
commit 6b0c3c3
Author: Boyuan Feng <boyuan@meta.com>
Date:   Tue Oct 14 21:30:29 2025 -0700

    nit

    Signed-off-by: Boyuan Feng <boyuan@meta.com>

commit 1016467
Author: Boyuan Feng <boyuan@meta.com>
Date:   Tue Oct 14 21:21:47 2025 -0700

    fix multi-graph test

    Signed-off-by: Boyuan Feng <boyuan@meta.com>

Signed-off-by: ProExpertProg <lgovedic@redhat.com>
@ProExpertProg ProExpertProg added the ready ONLY add when PR is ready to merge/full CI is needed label Oct 15, 2025
@mergify
Copy link
Copy Markdown

mergify bot commented Oct 15, 2025

This pull request has merge conflicts that must be resolved before it can be
merged. Please rebase the PR, @BoyuanFeng.

https://docs.github.com/en/pull-requests/collaborating-with-pull-requests/working-with-forks/syncing-a-fork

@mergify mergify bot added the needs-rebase label Oct 15, 2025
Signed-off-by: Boyuan Feng <boyuan@meta.com>
@mergify mergify bot removed the needs-rebase label Oct 15, 2025
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
@hmellor hmellor self-requested a review as a code owner October 15, 2025 09:18
@mergify
Copy link
Copy Markdown

mergify bot commented Oct 15, 2025

Documentation preview: https://vllm--26878.org.readthedocs.build/en/26878/

@mergify mergify bot added the documentation Improvements or additions to documentation label Oct 15, 2025
@DarkLight1337 DarkLight1337 enabled auto-merge (squash) October 15, 2025 10:04
@DarkLight1337 DarkLight1337 merged commit f574383 into vllm-project:main Oct 15, 2025
46 of 47 checks passed
bbartels pushed a commit to bbartels/vllm that referenced this pull request Oct 16, 2025
Signed-off-by: Boyuan Feng <boyuan@meta.com>
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Signed-off-by: bbartels <benjamin@bartels.dev>
albertoperdomo2 pushed a commit to albertoperdomo2/vllm that referenced this pull request Oct 16, 2025
Signed-off-by: Boyuan Feng <boyuan@meta.com>
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Signed-off-by: Alberto Perdomo <aperdomo@redhat.com>
lywa1998 pushed a commit to lywa1998/vllm that referenced this pull request Oct 20, 2025
Signed-off-by: Boyuan Feng <boyuan@meta.com>
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
alhridoy pushed a commit to alhridoy/vllm that referenced this pull request Oct 24, 2025
Signed-off-by: Boyuan Feng <boyuan@meta.com>
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
0xrushi pushed a commit to 0xrushi/vllm that referenced this pull request Oct 26, 2025
Signed-off-by: Boyuan Feng <boyuan@meta.com>
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Signed-off-by: 0xrushi <6279035+0xrushi@users.noreply.github.com>
0xrushi pushed a commit to 0xrushi/vllm that referenced this pull request Oct 26, 2025
Signed-off-by: Boyuan Feng <boyuan@meta.com>
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Signed-off-by: 0xrushi <6279035+0xrushi@users.noreply.github.com>
rtourgeman pushed a commit to rtourgeman/vllm that referenced this pull request Nov 10, 2025
Signed-off-by: Boyuan Feng <boyuan@meta.com>
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
devpatelio pushed a commit to SumanthRH/vllm that referenced this pull request Nov 29, 2025
Signed-off-by: Boyuan Feng <boyuan@meta.com>
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

documentation Improvements or additions to documentation 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