We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6ebffaf commit 921e78fCopy full SHA for 921e78f
docker/Dockerfile.rocm_base
@@ -7,7 +7,7 @@ ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git"
7
ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"
8
ARG FA_BRANCH="0e60e394"
9
ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"
10
-ARG AITER_BRANCH="eef23c7f"
+ARG AITER_BRANCH="9716b1b8"
11
ARG AITER_REPO="https://github.com/ROCm/aiter.git"
12
13
FROM ${BASE_IMAGE} AS base
0 commit comments