From 6ca4f400d8b6c83a40af5c338b2ac55a8effdd76 Mon Sep 17 00:00:00 2001 From: rasmith Date: Fri, 16 Jan 2026 02:22:53 -0600 Subject: [PATCH] [CI][AMD] Skip test_permute_cols since the kernel is not used and not built for ROCm (#32444) Signed-off-by: Randall Smith --- tests/kernels/core/test_permute_cols.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/kernels/core/test_permute_cols.py b/tests/kernels/core/test_permute_cols.py index 08fdd0e05..929db34ae 100644 --- a/tests/kernels/core/test_permute_cols.py +++ b/tests/kernels/core/test_permute_cols.py @@ -7,6 +7,9 @@ import torch from tests.kernels.utils import opcheck from vllm._custom_ops import permute_cols +if not hasattr(torch.ops._C, "permute_cols"): + pytest.skip(reason="permute_cols is not supported on ROCm", allow_module_level=True) + @pytest.mark.parametrize("shape", [(1, 512), (544, 4096), (67, 8192)]) @pytest.mark.parametrize("dtype", [torch.bfloat16])