We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 16c8185 commit 1aec156Copy full SHA for 1aec156
Dockerfile.rocm_base
@@ -12,7 +12,7 @@ ARG PYTORCH_REPO="https://github.com/pytorch/pytorch.git"
12
ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"
13
ARG FA_BRANCH="1a7f4dfa"
14
ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"
15
-ARG AITER_BRANCH="c0da3d23"
+ARG AITER_BRANCH="8970b25b"
16
ARG AITER_REPO="https://github.com/ROCm/aiter.git"
17
18
FROM ${BASE_IMAGE} AS base
0 commit comments