diff --git a/tests/diffusion/kernels/mot/test_mot_linear.py b/tests/diffusion/kernels/mot/test_mot_linear.py index 9fd0dab58..0ae55dfa7 100644 --- a/tests/diffusion/kernels/mot/test_mot_linear.py +++ b/tests/diffusion/kernels/mot/test_mot_linear.py @@ -48,6 +48,8 @@ from vllm_omni.diffusion.layers.mot.ops.mot_gemm import ( get_best_mot_config, ) +pytestmark = [pytest.mark.core_model, pytest.mark.diffusion, pytest.mark.gpu] + # --------------------------------------------------------------------------- # Constants # ---------------------------------------------------------------------------