A high-throughput and memory-efficient inference and serving engine for LLMs
Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git
Synced Nov 5, 01:00 UTC
[CI/Build] Apply ruff formatter to pass pre-commit (#40078)
Signed-off-by: Hemmi Shinichi <shemmi@preferred.jp>
Shinichi Hemmi committed
Apr 17, 2026 at 09:54 UTC
4c47710bf707e93bdf83391948f844e461b7ba4f
1 file changed
+3
-5
tests/compile/passes/test_vllm_fusion_pattern_matcher_pass.py
+3
-5
@@ -6,14 +6,13 @@ import torch
6
7
import vllm.config
8
from tests.compile.backend import TestBackend
9
-from vllm.platforms import current_platform
9
from vllm.compilation.passes.vllm_inductor_pass import (
10
VllmFusionPatternMatcherPass,
11
VllmPatternMatcherPass,
12
VllmPatternReplacement,
13
)
14
from vllm.config import CompilationConfig, CompilationMode, VllmConfig
16
-
15
+from vllm.platforms import current_platform
16
17
18
class ReluToAbsPattern(VllmPatternReplacement):
@@ -58,7 +57,6 @@ class ExpToSqrtPattern(VllmPatternReplacement):
57
return [self.empty_fp32(4)]
58
59
61
-
60
class ReluFusionPass(VllmFusionPatternMatcherPass):
61
def __init__(self, config: VllmConfig) -> None:
62
super().__init__(config, "test_relu_fusion")
@@ -72,13 +70,13 @@ class TwoPatternFusionPass(VllmFusionPatternMatcherPass):
70
self.register(ExpToSqrtPattern())
71
72
75
-
73
@pytest.fixture
74
def vllm_config():
75
return VllmConfig(
76
compilation_config=CompilationConfig(mode=CompilationMode.VLLM_COMPILE),
77
)
78
79
+
80
@pytest.mark.skipif(not current_platform.is_cuda_alike(), reason="Requires CUDA")
81
def test_register_tracks_patterns(vllm_config):
82
"""register() appends each VllmPatternReplacement to _pattern_replacements."""
@@ -96,7 +94,7 @@ def test_uuid_stable(vllm_config):
94
with vllm.config.set_current_vllm_config(vllm_config):
95
p1 = ReluFusionPass(vllm_config)
96
p2 = ReluFusionPass(vllm_config)
99
- p3= TwoPatternFusionPass(vllm_config)
97
+ p3 = TwoPatternFusionPass(vllm_config)
98
99
assert p1.uuid() == p2.uuid()
100
assert p1.uuid() != p3.uuid()