[CI][IR] Speed up vLLM IR test group (#55965)
Signed-off-by: Djordje Ramic <djoramic@amd.com>
djramic committed
Sep 9, 2026 at 04:28 UTC
21a117ad87ef2792964ef71c0603e220ce97f346
4 files changed
+11
tests/ir/test_inplace_op.py
+3
@@ -1,5 +1,6 @@
1
# SPDX-License-Identifier: Apache-2.0
2
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
+import pytest
4
import torch
5
from torch import Tensor
6
from torch.fx.experimental.proxy_tensor import make_fx
@@ -7,6 +8,8 @@ from torch.fx.experimental.proxy_tensor import make_fx
8
import vllm.ir.op
9
from vllm.ir.op import IrOp, IrOpInplaceOverload
10
11
+pytestmark = pytest.mark.skip_global_cleanup
12
+
13
14
@vllm.ir.register_op(allow_inplace=True)
15
def _custom_mm2(x: Tensor, w: Tensor) -> Tensor:
tests/ir/test_op.py
+2
@@ -13,6 +13,8 @@ from torch.fx.experimental.proxy_tensor import make_fx
13
import vllm.ir.op
14
from vllm.ir.op import RESERVED_PROVIDERS, IrOp, IrOpImpl
15
16
+pytestmark = pytest.mark.skip_global_cleanup
17
+
18
19
class CustomError(Exception):
20
pass
tests/kernels/ir/test_ir_ops.py
+4
@@ -8,9 +8,13 @@ Per-op correctness tests live alongside their op definitions
8
(e.g. tests/kernels/ir/test_layernorm.py).
9
"""
10
11
+import pytest
12
+
13
import vllm.kernels # noqa: F401 — registers provider implementations
14
from vllm.ir.op import IrOp
15
16
+pytestmark = pytest.mark.skip_global_cleanup
17
+
18
19
def test_all_ops_have_input_generator():
20
missing = [name for name, op in IrOp.registry.items() if not op.has_input_generator]
tests/kernels/ir/test_layernorm.py
+2
@@ -16,6 +16,8 @@ from tests.utils import set_random_seed
16
from vllm import ir
17
from vllm.platforms import current_platform
18
19
+pytestmark = pytest.mark.skip_global_cleanup
20
+
21
rms_norm_native = ir.ops.rms_norm.impls["native"].impl_fn
22
23
IS_GPGPU_DEVICE = current_platform.is_cuda_alike() or current_platform.is_xpu()