From 45c1ca1ca1ee8fa06df263c8715e8a412ff408d4 Mon Sep 17 00:00:00 2001 From: Andreas Karatzas Date: Mon, 29 Dec 2025 01:31:10 -0600 Subject: [PATCH] [ROCm][CI] Skip DeepGemm-dependent test on ROCm platform (#31462) Signed-off-by: Andreas Karatzas --- .../moe/test_silu_mul_per_token_group_quant_fp8_colmajor.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/kernels/moe/test_silu_mul_per_token_group_quant_fp8_colmajor.py b/tests/kernels/moe/test_silu_mul_per_token_group_quant_fp8_colmajor.py index e4617072c..ace0794fe 100644 --- a/tests/kernels/moe/test_silu_mul_per_token_group_quant_fp8_colmajor.py +++ b/tests/kernels/moe/test_silu_mul_per_token_group_quant_fp8_colmajor.py @@ -67,6 +67,10 @@ def reference(x: torch.Tensor, use_ue8m0: bool) -> tuple[torch.Tensor, torch.Ten @pytest.mark.parametrize("T", [128, 256, 512]) @pytest.mark.parametrize("N", [128 * 2, 256 * 2, 768 * 2, 2048 * 2, 7168 * 2]) +@pytest.mark.skipif( + current_platform.is_rocm(), + reason="ROCm does not support DeepGemm.", +) def test_silu_mul_fp8_quant_deep_gemm(T: int, N: int): current_platform.seed_everything(42)