Skip to content

GitLab

Unverified Commit 921e78f4 authored by Micah Williamson's avatar Micah Williamson Committed by GitHub
Browse files

[ROCm] Update AITER branch for ROCm base docker (#27586) 


Signed-off-by: default avatarMicah Williamson <micah.williamson@amd.com>
parent 6ebffafb
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docker/Dockerfile.rocm_base
View file @ 921e78f4
...@@ -7,7 +7,7 @@ ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git" ...@@ -7,7 +7,7 @@ ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git"
ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git" ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"
ARG FA_BRANCH="0e60e394" ARG FA_BRANCH="0e60e394"
ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git" ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"
ARG AITER_BRANCH="eef23c7f" ARG AITER_BRANCH="9716b1b8"
ARG AITER_REPO="https://github.com/ROCm/aiter.git" ARG AITER_REPO="https://github.com/ROCm/aiter.git"
FROM ${BASE_IMAGE} AS base FROM ${BASE_IMAGE} AS base
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment