From b52da5455f3ed00b43641066781ec57a5560e3ea 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:59 +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_norm.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/diffusion/kernels/mot/test_mot_norm.py b/tests/diffusion/kernels/mot/test_mot_norm.py index c3b8bf53d..af6724d1e 100644 --- a/tests/diffusion/kernels/mot/test_mot_norm.py +++ b/tests/diffusion/kernels/mot/test_mot_norm.py @@ -22,6 +22,8 @@ from vllm.model_executor.layers.layernorm import rms_norm from vllm_omni.diffusion.layers.mot.mot_layernorm import MoTRMSNorm +pytestmark = [pytest.mark.core_model, pytest.mark.diffusion, pytest.mark.gpu] + # --------------------------------------------------------------------------- # Constants # ---------------------------------------------------------------------------