diff --git a/Dockerfile.base_navi b/Dockerfile.base_navi index 4041024189f5d..8c571ae219794 100644 --- a/Dockerfile.base_navi +++ b/Dockerfile.base_navi @@ -1,11 +1,11 @@ ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:6.3-complete -ARG HIPBLASLT_BRANCH="d43d84a" +ARG HIPBLASLT_BRANCH="4d40e36" ARG LEGACY_HIPBLASLT_OPTION= -ARG RCCL_BRANCH="dfe4a3e" +ARG RCCL_BRANCH="648a58d" ARG RCCL_REPO="https://github.com/ROCm/rccl" -ARG TRITON_BRANCH="e192dba" +ARG TRITON_BRANCH="e5be006" ARG TRITON_REPO="https://github.com/triton-lang/triton.git" -ARG PYTORCH_BRANCH="8bc4033" +ARG PYTORCH_BRANCH="8d4926e" ARG PYTORCH_VISION_BRANCH="v0.19.1" ARG PYTORCH_REPO="https://github.com/pytorch/pytorch.git" ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"