[ROCm][CI] Language Models tests tiny-mixtral with aiter fix (#49551)
Signed-off-by: Dino Music <Dino.Music@amd.com>
music-dino committed
Jul 23, 2026 at 21:01 UTC
4501a6d56b2c45dd9d2af816bef631da50374828
1 file changed
-8
tests/models/language/generation/test_common.py
-8
@@ -128,14 +128,6 @@ def test_models(
128
129
if use_rocm_aiter and (model in AITER_MODEL_LIST):
130
monkeypatch.setenv("VLLM_ROCM_USE_AITER", "1")
131
- if model == "TitanML/tiny-mixtral":
132
- # Untrained model: near-uniform logits make argmax sensitive to
133
- # AITER's bfloat16 rounding error. Route the plain rms_norm and the
134
- # fused MoE (whose near-uniform router logits flip expert selection
135
- # under ~1 ULP drift) through the native kernels for this model.
136
- # See ROCm/aiter#3806 for the tracking issue and minimal repro.
137
- monkeypatch.setenv("VLLM_ROCM_USE_AITER_RMSNORM", "0")
138
- monkeypatch.setenv("VLLM_ROCM_USE_AITER_MOE", "0")
131
elif use_rocm_aiter and model not in AITER_MODEL_LIST:
132
# Skip model that are not using AITER tests.
133
# When more AITER kernels are added, this list will not be