From e6cd8913ddfe63b4620e45ff8c2da1d37318dbe5 Mon Sep 17 00:00:00 2001 From: Charlie Fu Date: Fri, 19 Jun 2026 17:20:10 -0500 Subject: [PATCH] [ROCm][CI] Skip Qwen3.5-35B-A3B-MXFP4-AITER-TP2 for non gfx950 (#46109) Signed-off-by: charlifu --- .buildkite/test_areas/lm_eval.yaml | 9 +++++++++ tests/evals/gsm8k/test_gsm8k_correctness.py | 9 +++++++++ 2 files changed, 18 insertions(+) diff --git a/.buildkite/test_areas/lm_eval.yaml b/.buildkite/test_areas/lm_eval.yaml index fc8e72699e..217fc5665c 100644 --- a/.buildkite/test_areas/lm_eval.yaml +++ b/.buildkite/test_areas/lm_eval.yaml @@ -101,6 +101,15 @@ steps: num_devices: 8 commands: - pytest -s -v evals/gsm8k/test_gsm8k_correctness.py --config-list-file=configs/models-h200.txt + mirror: + amd: + device: mi300_8 + timeout_in_minutes: 180 + depends_on: + - image-build-amd + commands: + - export VLLM_WORKER_MULTIPROC_METHOD=spawn + - pytest -s -v evals/gsm8k/test_gsm8k_correctness.py --config-list-file=configs/models-mi3xx.txt - label: MoE Refactor Integration Test (H100 - TEMPORARY) key: moe-refactor-integration-test-h100-temporary diff --git a/tests/evals/gsm8k/test_gsm8k_correctness.py b/tests/evals/gsm8k/test_gsm8k_correctness.py index e7a254e760..cd90d71669 100644 --- a/tests/evals/gsm8k/test_gsm8k_correctness.py +++ b/tests/evals/gsm8k/test_gsm8k_correctness.py @@ -78,7 +78,16 @@ def test_gsm8k_correctness(config_filename): "Skipping DeepSeek-V3.2 and DeepSeek-R1 on ROCm platforms " "due to agent pool disk space issues and pod evictions." ) + if current_platform.is_rocm() and ( + "Qwen3.5-35B-A3B-MXFP4-AITER-TP2" in config_filename.name + ): + from vllm.platforms.rocm import on_gfx950 + if not on_gfx950(): + pytest.skip( + "Skipping Qwen3.5-35B-A3B-MXFP4-AITER-TP2 on non-GFX950 platforms. " + "The quantization scheme is not supported on non-GFX950 platforms." + ) # Parse server arguments from config (use shlex to handle quoted strings) server_args_str = eval_config.get("server_args", "") server_args = shlex.split(server_args_str) if server_args_str else []