Unverified Commit 573a1d11 authored by Zhiwei's avatar Zhiwei Committed by GitHub
Browse files

[ROCm]Skip test_torchao.py::test_pre_quantized_model on CDNA3 arch (#31905)


Signed-off-by: default avatarZhiweiYan-96 <zhiwei.yan@amd.com>
parent 33156f56
......@@ -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(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment