biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[CI][AMD] Skip test_permute_cols since the kernel is not used and not built for ROCm (#32444)

Browse Source 
Signed-off-by: Randall Smith <ransmith@amd.com>
This commit is contained in:
rasmith
2026-01-16 02:22:53 -06:00
committed by GitHub
parent 180e981d56
commit 6ca4f400d8
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/kernels/core/test_permute_cols.py
View File

@@ -7,6 +7,9 @@ import torch
from tests.kernels.utils import opcheck from tests.kernels.utils import opcheck
from vllm._custom_ops import permute_cols 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("shape", [(1, 512), (544, 4096), (67, 8192)])
@pytest.mark.parametrize("dtype", [torch.bfloat16]) @pytest.mark.parametrize("dtype", [torch.bfloat16])