From 58cde5c026efee42987fbc87681ecbf262f9db2b Mon Sep 17 00:00:00 2001 From: Andreas Karatzas Date: Tue, 17 Mar 2026 22:12:26 -0500 Subject: [PATCH] [ROCm][CI] Skip trtllm kvfp8 dequant tests on ROCm (#37330) Signed-off-by: Andreas Karatzas --- tests/kernels/attention/test_trtllm_kvfp8_dequant.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/kernels/attention/test_trtllm_kvfp8_dequant.py b/tests/kernels/attention/test_trtllm_kvfp8_dequant.py index a2ea372c0..c49ceb03f 100644 --- a/tests/kernels/attention/test_trtllm_kvfp8_dequant.py +++ b/tests/kernels/attention/test_trtllm_kvfp8_dequant.py @@ -12,6 +12,12 @@ import torch from vllm.platforms import current_platform +if current_platform.is_rocm(): + pytest.skip( + "trtllm kvfp8 dequant is not supported on ROCm.", + allow_module_level=True, + ) + FP8_DTYPE = current_platform.fp8_dtype() NUM_BLOCKS = 128