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
4c47710b
Unverified
Commit
4c47710b
authored
Apr 17, 2026
by
Shinichi Hemmi
Committed by
GitHub
Apr 17, 2026
Browse files
[CI/Build] Apply ruff formatter to pass pre-commit (#40078)
Signed-off-by:
Hemmi Shinichi
<
shemmi@preferred.jp
>
parent
bf9a5ddb
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
5 deletions
+3
-5
tests/compile/passes/test_vllm_fusion_pattern_matcher_pass.py
...s/compile/passes/test_vllm_fusion_pattern_matcher_pass.py
+3
-5
No files found.
tests/compile/passes/test_vllm_fusion_pattern_matcher_pass.py
View file @
4c47710b
...
@@ -6,14 +6,13 @@ import torch
...
@@ -6,14 +6,13 @@ import torch
import
vllm.config
import
vllm.config
from
tests.compile.backend
import
TestBackend
from
tests.compile.backend
import
TestBackend
from
vllm.platforms
import
current_platform
from
vllm.compilation.passes.vllm_inductor_pass
import
(
from
vllm.compilation.passes.vllm_inductor_pass
import
(
VllmFusionPatternMatcherPass
,
VllmFusionPatternMatcherPass
,
VllmPatternMatcherPass
,
VllmPatternMatcherPass
,
VllmPatternReplacement
,
VllmPatternReplacement
,
)
)
from
vllm.config
import
CompilationConfig
,
CompilationMode
,
VllmConfig
from
vllm.config
import
CompilationConfig
,
CompilationMode
,
VllmConfig
from
vllm.platforms
import
current_platform
class
ReluToAbsPattern
(
VllmPatternReplacement
):
class
ReluToAbsPattern
(
VllmPatternReplacement
):
...
@@ -58,7 +57,6 @@ class ExpToSqrtPattern(VllmPatternReplacement):
...
@@ -58,7 +57,6 @@ class ExpToSqrtPattern(VllmPatternReplacement):
return
[
self
.
empty_fp32
(
4
)]
return
[
self
.
empty_fp32
(
4
)]
class
ReluFusionPass
(
VllmFusionPatternMatcherPass
):
class
ReluFusionPass
(
VllmFusionPatternMatcherPass
):
def
__init__
(
self
,
config
:
VllmConfig
)
->
None
:
def
__init__
(
self
,
config
:
VllmConfig
)
->
None
:
super
().
__init__
(
config
,
"test_relu_fusion"
)
super
().
__init__
(
config
,
"test_relu_fusion"
)
...
@@ -72,13 +70,13 @@ class TwoPatternFusionPass(VllmFusionPatternMatcherPass):
...
@@ -72,13 +70,13 @@ class TwoPatternFusionPass(VllmFusionPatternMatcherPass):
self
.
register
(
ExpToSqrtPattern
())
self
.
register
(
ExpToSqrtPattern
())
@
pytest
.
fixture
@
pytest
.
fixture
def
vllm_config
():
def
vllm_config
():
return
VllmConfig
(
return
VllmConfig
(
compilation_config
=
CompilationConfig
(
mode
=
CompilationMode
.
VLLM_COMPILE
),
compilation_config
=
CompilationConfig
(
mode
=
CompilationMode
.
VLLM_COMPILE
),
)
)
@
pytest
.
mark
.
skipif
(
not
current_platform
.
is_cuda_alike
(),
reason
=
"Requires CUDA"
)
@
pytest
.
mark
.
skipif
(
not
current_platform
.
is_cuda_alike
(),
reason
=
"Requires CUDA"
)
def
test_register_tracks_patterns
(
vllm_config
):
def
test_register_tracks_patterns
(
vllm_config
):
"""register() appends each VllmPatternReplacement to _pattern_replacements."""
"""register() appends each VllmPatternReplacement to _pattern_replacements."""
...
@@ -96,7 +94,7 @@ def test_uuid_stable(vllm_config):
...
@@ -96,7 +94,7 @@ def test_uuid_stable(vllm_config):
with
vllm
.
config
.
set_current_vllm_config
(
vllm_config
):
with
vllm
.
config
.
set_current_vllm_config
(
vllm_config
):
p1
=
ReluFusionPass
(
vllm_config
)
p1
=
ReluFusionPass
(
vllm_config
)
p2
=
ReluFusionPass
(
vllm_config
)
p2
=
ReluFusionPass
(
vllm_config
)
p3
=
TwoPatternFusionPass
(
vllm_config
)
p3
=
TwoPatternFusionPass
(
vllm_config
)
assert
p1
.
uuid
()
==
p2
.
uuid
()
assert
p1
.
uuid
()
==
p2
.
uuid
()
assert
p1
.
uuid
()
!=
p3
.
uuid
()
assert
p1
.
uuid
()
!=
p3
.
uuid
()
...
...
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