From ec8a3725668418b156fe2fd5c85ae3b1b9e37ad9 Mon Sep 17 00:00:00 2001 From: Ronald Date: Wed, 14 Oct 2026 15:19:38 +0800 Subject: [PATCH] [ROCm][Bugfix] Pin AITER back to v0.1.24.post1 for DSv4.1 mono decode numerics (#61559) Signed-off-by: Ronald1995 --- .buildkite/test_areas/models_basic.yaml | 1 + docker/Dockerfile.rock_base | 2 +- docker/Dockerfile.rocm_base | 2 +- 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.buildkite/test_areas/models_basic.yaml b/.buildkite/test_areas/models_basic.yaml index a500b838c5..3e0d840c6f 100644 --- a/.buildkite/test_areas/models_basic.yaml +++ b/.buildkite/test_areas/models_basic.yaml @@ -273,6 +273,7 @@ steps: depends_on: - image-build-amd source_file_dependencies: + - docker/Dockerfile.rocm_base - vllm/models/deepseek_v41/amd/ - tests/models/test_dsv41_mono_numerics.py - tests/models/test_dsv41_mono_layers.py diff --git a/docker/Dockerfile.rock_base b/docker/Dockerfile.rock_base index 57bf6ae6fe..40100a1bbf 100644 --- a/docker/Dockerfile.rock_base +++ b/docker/Dockerfile.rock_base @@ -8,7 +8,7 @@ ARG TRITON_VERSION=3.8.0+git669b31ac ARG TRITON_REPO="https://github.com/ROCm/triton.git" ARG FA_BRANCH="0e60e394" ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git" -ARG AITER_BRANCH="v0.1.25" +ARG AITER_BRANCH="v0.1.24.post1" ARG AITER_REPO="https://github.com/ROCm/aiter.git" ARG MORI_BRANCH="v1.2.3.post1" ARG MORI_REPO="https://github.com/ROCm/mori.git" diff --git a/docker/Dockerfile.rocm_base b/docker/Dockerfile.rocm_base index 33dd0f6b10..bd8099dd15 100644 --- a/docker/Dockerfile.rocm_base +++ b/docker/Dockerfile.rocm_base @@ -9,7 +9,7 @@ ARG PYTORCH_AUDIO_BRANCH="v2.11.0" ARG PYTORCH_AUDIO_REPO="https://github.com/pytorch/audio.git" ARG FA_BRANCH="0e60e394" ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git" -ARG AITER_BRANCH="v0.1.25" +ARG AITER_BRANCH="v0.1.24.post1" ARG AITER_REPO="https://github.com/ROCm/aiter.git" ARG MORI_BRANCH="v1.2.3.post1" ARG MORI_REPO="https://github.com/ROCm/mori.git"