Add pytest markers for core model and diffusion tests

Signed-off-by: 汪志鹏 <wangzhipeng628@gmail.com>
Signed-off-by: princepride <wangzhipeng628@gmail.com>
This commit is contained in:
汪志鹏 authored and princepride committed 2026-05-29 02:34:51 +00:00
1 parent f9ba582747
commit 5958b01ff7
1 file changed
+2
@@ -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
# ---------------------------------------------------------------------------