Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
17c540a9
Unverified
Commit
17c540a9
authored
Oct 16, 2025
by
Boyuan Feng
Committed by
GitHub
Oct 16, 2025
Browse files
[torch.compile] fix simple inductor graph partition test (#27050)
Signed-off-by:
Boyuan Feng
<
boyuan@meta.com
>
parent
4d4d6bad
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
4 deletions
+2
-4
tests/compile/piecewise/test_simple.py
tests/compile/piecewise/test_simple.py
+2
-4
No files found.
tests/compile/piecewise/test_simple.py
View file @
17c540a9
...
...
@@ -142,8 +142,7 @@ def test_simple_piecewise_compile(use_inductor):
@
torch
.
inference_mode
()
@
pytest
.
mark
.
parametrize
(
"splitting_ops"
,
[[
"silly::attention"
],
[]])
def
test_simple_inductor_graph_partition
(
splitting_ops
,
monkeypatch
):
def
test_simple_inductor_graph_partition
(
monkeypatch
):
if
not
is_torch_equal_or_newer
(
"2.9.0.dev"
):
pytest
.
skip
(
"inductor graph partition is only available in PyTorch 2.9+"
)
...
...
@@ -152,8 +151,7 @@ def test_simple_inductor_graph_partition(splitting_ops, monkeypatch):
monkeypatch
.
setenv
(
"VLLM_DISABLE_COMPILE_CACHE"
,
"1"
)
_run_simple_model
(
# Inductor graph partition automatically resets splitting_ops to an empty list
splitting_ops
=
splitting_ops
,
splitting_ops
=
[
"silly::attention"
],
use_inductor_graph_partition
=
True
,
use_inductor
=
True
,
# Since not splitting at fx graph level
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment