Skip to content

Commit 6ca4f40

Browse files
authored
[CI][AMD] Skip test_permute_cols since the kernel is not used and not built for ROCm (vllm-project#32444)
Signed-off-by: Randall Smith <ransmith@amd.com>
1 parent 180e981 commit 6ca4f40

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

‎tests/kernels/core/test_permute_cols.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,9 @@
77
from tests.kernels.utils import opcheck
88
from vllm._custom_ops import permute_cols
99

10+
if not hasattr(torch.ops._C, "permute_cols"):
11+
pytest.skip(reason="permute_cols is not supported on ROCm", allow_module_level=True)
12+
1013

1114
@pytest.mark.parametrize("shape", [(1, 512), (544, 4096), (67, 8192)])
1215
@pytest.mark.parametrize("dtype", [torch.bfloat16])

0 commit comments

Comments
 (0)