[CI] Fix `test_auto_gptq` on ROCm CI (#46164)
Signed-off-by: Felix Marty <Felix.Marty@amd.com> Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
fxmarty-amd committed
Jun 23, 2026 at 23:38 UTC
84f13374b350fba78872f8612c3a839256815a4a
1 file changed
+2
-2
tests/quantization/test_configs.py
+2
-2
@@ -28,7 +28,7 @@ MODEL_ARG_EXPTYPES = [
28
(
29
"TheBloke/Llama-2-7B-Chat-GPTQ",
30
"marlin",
31
- "auto_gptq" if current_platform.is_cuda() else "ERROR",
31
+ "auto_gptq" if current_platform.is_cuda_alike() else "ERROR",
32
),
33
("TheBloke/Llama-2-7B-Chat-GPTQ", "gptq", "auto_gptq"),
34
("TheBloke/Llama-2-7B-Chat-GPTQ", "awq", "ERROR"),
@@ -38,7 +38,7 @@ MODEL_ARG_EXPTYPES = [
38
(
39
"LnL-AI/TinyLlama-1.1B-Chat-v1.0-GPTQ-4bit",
40
"marlin",
41
- "auto_gptq" if current_platform.is_cuda() else "ERROR",
41
+ "auto_gptq" if current_platform.is_cuda_alike() else "ERROR",
42
),
43
("LnL-AI/TinyLlama-1.1B-Chat-v1.0-GPTQ-4bit", "gptq", "auto_gptq"),
44
("LnL-AI/TinyLlama-1.1B-Chat-v1.0-GPTQ-4bit", "awq", "ERROR"),