diff --git a/docker/Dockerfile.rock_base b/docker/Dockerfile.rock_base
index 80b6aaca29..adde16af3c 100644
--- a/docker/Dockerfile.rock_base
+++ b/docker/Dockerfile.rock_base
@@ -6,7 +6,7 @@ ARG TORCHAUDIO_VERSION=2.11.0+rocm10.0.0
 ARG TRITON_VERSION=3.8.0+git4cff872c.rocm10.0.0
 ARG FA_BRANCH="0e60e394"
 ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"
-ARG AITER_BRANCH="v0.1.21.post1"
+ARG AITER_BRANCH="v0.1.22.post1"
 ARG AITER_REPO="https://github.com/ROCm/aiter.git"
 ARG MORI_BRANCH="f95cd5e0ca7b730f92f9ffdfb358484d6965d9f7" # TheRock compatible v1.1.0
 ARG MORI_REPO="https://github.com/ROCm/mori.git"
diff --git a/docker/Dockerfile.rocm_base b/docker/Dockerfile.rocm_base
index fa0a88fd20..eb4bb5cc97 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.21.post2"
+ARG AITER_BRANCH="v0.1.22.post1"
 ARG AITER_REPO="https://github.com/ROCm/aiter.git"
 ARG MORI_BRANCH="v1.1.0"
 ARG MORI_REPO="https://github.com/ROCm/mori.git"