From b6520e5ec6e0aad1bec5b7d142f1e05d26c7849c Mon Sep 17 00:00:00 2001 From: westers Date: Sat, 20 Dec 2025 10:21:05 -0600 Subject: [PATCH] Skip RTN quantization tests on ROCm MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Fixes #29525 RTN (Round-To-Nearest) quantization is currently not supported on ROCm. This fix adds a check to skip RTN quantization tests when running on ROCm to prevent CI failures for unsupported features. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Sonnet 4.5 Signed-off-by: westers --- tests/quantization/utils.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/quantization/utils.py b/tests/quantization/utils.py index 20a425b721145..8b3dd1ef0c003 100644 --- a/tests/quantization/utils.py +++ b/tests/quantization/utils.py @@ -10,6 +10,10 @@ def is_quant_method_supported(quant_method: str) -> bool: if not (current_platform.is_cuda() or current_platform.is_rocm()): return False + # RTN quantization is currently not supported on ROCm + if current_platform.is_rocm() and quant_method == "rtn": + return False + capability = current_platform.get_device_capability() assert capability is not None