Skip to content

Commit 921e78f

Browse files
authored
[ROCm] Update AITER branch for ROCm base docker (#27586)
Signed-off-by: Micah Williamson <[email protected]>
1 parent 6ebffaf commit 921e78f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docker/Dockerfile.rocm_base

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git"
77
ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"
88
ARG FA_BRANCH="0e60e394"
99
ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"
10-
ARG AITER_BRANCH="eef23c7f"
10+
ARG AITER_BRANCH="9716b1b8"
1111
ARG AITER_REPO="https://github.com/ROCm/aiter.git"
1212

1313
FROM ${BASE_IMAGE} AS base

0 commit comments

Comments
 (0)