From 573a1d1119af85613ff0cb90ac063ab669cbbd7f Mon Sep 17 00:00:00 2001 From: Zhiwei <532707544@qq.com> Date: Thu, 8 Jan 2026 15:47:44 +0800 Subject: [PATCH] [ROCm]Skip test_torchao.py::test_pre_quantized_model on CDNA3 arch (#31905) Signed-off-by: ZhiweiYan-96 --- tests/quantization/test_torchao.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/quantization/test_torchao.py b/tests/quantization/test_torchao.py index f35c3973ab6..da4f6a02870 100644 --- a/tests/quantization/test_torchao.py +++ b/tests/quantization/test_torchao.py @@ -6,11 +6,17 @@ import importlib.util import pytest import torch +from vllm.platforms import current_platform + DTYPE = ["bfloat16"] TORCHAO_AVAILABLE = importlib.util.find_spec("torchao") is not None +@pytest.mark.skipif( + current_platform.is_rocm() and current_platform.is_fp8_fnuz(), + reason="Only fp8_fnuz supported on CDNA3 architecture", +) @pytest.mark.skipif(not TORCHAO_AVAILABLE, reason="torchao is not available") def test_pre_quantized_model(vllm_runner): with vllm_runner(