Skip to content

Commit dfe3216

Browse files
authored
Update the base dockerfile to match the one actually built (#623)
1 parent b6ddf62 commit dfe3216

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
@@ -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="916bf3c"
13+
ARG AITER_BRANCH="6b92d30d"
1414
ARG AITER_REPO="https://github.com/ROCm/aiter.git"
1515

1616
FROM ${BASE_IMAGE} AS base

0 commit comments

Comments
 (0)