Revert unrelated ROCm CI test and timeout changes

Signed-off-by: Sy03 <1370724210@qq.com>
This commit is contained in:
Sy03 committed 2026-09-16 21:02:33 +08:00
1 parent ec0feea2bd
commit d8458b8417
4 files changed
+3 -78

No files matched your search

@@ -14,17 +14,12 @@ from tests.diffusion.models.magi2.test_native_packing import (
)
from tests.helpers.mark import hardware_marks
from vllm_omni.diffusion.models.magi2.sampler_magi2 import CFGConfig
from vllm_omni.platforms import current_omni_platform
pytestmark = [
pytest.mark.diffusion,
pytest.mark.core_model,
*hardware_marks(res={"cuda": "L4"}, num_cards=1),
# ROCm also exposes torch.cuda, but this path uses vLLM CUDA FlashAttention.
pytest.mark.skipif(
not current_omni_platform.is_cuda() or not torch.cuda.is_available(),
reason="requires vLLM CUDA FlashAttention",
),
pytest.mark.skipif(not torch.cuda.is_available(), reason="requires CUDA"),
]