From aa1356ec53a65a79a0027e8c265c76d84de8c046 Mon Sep 17 00:00:00 2001 From: Micah Williamson Date: Tue, 21 Oct 2025 11:01:23 -0500 Subject: [PATCH] [ROCm] Update Triton, Torch, and AITER branches for ROCm base Dockerfile (#27206) Signed-off-by: Micah Williamson --- docker/Dockerfile.rocm_base | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/Dockerfile.rocm_base b/docker/Dockerfile.rocm_base index 873c2fbcd4d3..5479eebaf795 100644 --- a/docker/Dockerfile.rocm_base +++ b/docker/Dockerfile.rocm_base @@ -1,13 +1,13 @@ ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.0-complete -ARG TRITON_BRANCH="f9e5bf54" +ARG TRITON_BRANCH="57c693b6" ARG TRITON_REPO="https://github.com/ROCm/triton.git" -ARG PYTORCH_BRANCH="b2fb6885" +ARG PYTORCH_BRANCH="1c57644d" ARG PYTORCH_VISION_BRANCH="v0.23.0" ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git" ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git" ARG FA_BRANCH="0e60e394" ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git" -ARG AITER_BRANCH="2ab9f4cd" +ARG AITER_BRANCH="eef23c7f" ARG AITER_REPO="https://github.com/ROCm/aiter.git" FROM ${BASE_IMAGE} AS base