From 5958b01ff714c13252ff774b7c5c24b940ad0686 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B1=AA=E5=BF=97=E9=B9=8F?= Date: Tue, 10 Mar 2026 23:11:30 +0800 Subject: [PATCH] Add pytest markers for core model and diffusion tests MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 汪志鹏 Signed-off-by: princepride --- tests/diffusion/kernels/mot/test_mot_linear.py | 2 ++ 1 file changed, 2 insertions(+) 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 # ---------------------------------------------------------------------------