From 921e78f4bbe56d6c9a0e9e609b9b3cfe97a4a51b Mon Sep 17 00:00:00 2001 From: Micah Williamson Date: Mon, 27 Oct 2025 12:22:33 -0500 Subject: [PATCH] [ROCm] Update AITER branch for ROCm base docker (#27586) Signed-off-by: Micah Williamson --- docker/Dockerfile.rocm_base | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile.rocm_base b/docker/Dockerfile.rocm_base index 5479eebaf7..19f7fa7e14 100644 --- a/docker/Dockerfile.rocm_base +++ b/docker/Dockerfile.rocm_base @@ -7,7 +7,7 @@ ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git" ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git" ARG FA_BRANCH="0e60e394" ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git" -ARG AITER_BRANCH="eef23c7f" +ARG AITER_BRANCH="9716b1b8" ARG AITER_REPO="https://github.com/ROCm/aiter.git" FROM ${BASE_IMAGE} AS base