Skip to content

Commit 7ecc5da

Browse files
authored
Update aiter version to match 0909_rc2 (#655)
1 parent f8c9a38 commit 7ecc5da

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docker/Dockerfile.rocm_base

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git"
1010
ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"
1111
ARG FA_BRANCH="1a7f4dfa"
1212
ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"
13-
ARG AITER_BRANCH="6b92d30d"
13+
ARG AITER_BRANCH="4822e675"
1414
ARG AITER_REPO="https://github.com/ROCm/aiter.git"
1515

1616
FROM ${BASE_IMAGE} AS base
@@ -178,4 +178,4 @@ RUN echo "BASE_IMAGE: ${BASE_IMAGE}" > /app/versions.txt \
178178
&& echo "PYTORCH_VISION_REPO: ${PYTORCH_VISION_REPO}" >> /app/versions.txt \
179179
&& echo "FA_BRANCH: ${FA_BRANCH}" >> /app/versions.txt \
180180
&& echo "AITER_BRANCH: ${AITER_BRANCH}" >> /app/versions.txt \
181-
&& echo "AITER_REPO: ${AITER_REPO}" >> /app/versions.txt
181+
&& echo "AITER_REPO: ${AITER_REPO}" >> /app/versions.txt

0 commit comments

Comments
 (0)