[ROCm] Skip tests for quantizations incompatible with ROCm (#17905)

Signed-off-by: Hissu Hyvarinen <hissu.hyvarinen@amd.com>
This commit is contained in:
hissu-hyvarinen
2025-05-13 03:39:28 +03:00
committed by GitHub
parent d67085c2c8
commit f6518b2b48
4 changed files with 20 additions and 3 deletions

View File

@@ -2,6 +2,7 @@
import pytest
from tests.quantization.utils import is_quant_method_supported
from vllm.platforms import current_platform
# These ground truth generations were generated using `transformers==4.38.1
# aqlm==1.1.0 torch==2.2.0`
@@ -34,7 +35,9 @@ ground_truth_generations = [
]
@pytest.mark.skipif(not is_quant_method_supported("aqlm"),
@pytest.mark.skipif(not is_quant_method_supported("aqlm")
or current_platform.is_rocm()
or not current_platform.is_cuda(),
reason="AQLM is not supported on this GPU type.")
@pytest.mark.parametrize("model", ["ISTA-DASLab/Llama-2-7b-AQLM-2Bit-1x16-hf"])
@pytest.mark.parametrize("dtype", ["half"])