[ROcm][BugFix][The Rock] Update The Rock dockerfile to most recent Triton 3.8 (#59287)
Signed-off-by: Randall Smith <Randall.Smith@amd.com>
rasmith committed
Sep 30, 2026 at 11:15 UTC
42a3dd5fec6f1ee9c7f84af7700fdf544aa4ca9c
1 file changed
+2
-2
docker/Dockerfile.rock_base
+2
-2
@@ -3,8 +3,8 @@ ARG PYTHON_VERSION=3.12
3
ARG TORCH_VERSION=2.12.0+rocm10.0.0
4
ARG TORCHVISION_VERSION=0.27.0+rocm10.0.0
5
ARG TORCHAUDIO_VERSION=2.11.0+rocm10.0.0
6
-ARG TRITON_BRANCH="c0a142ffcbb42f9a67a56bf791cad472be1d6425" # release/internal/3.8.x as of 9/21/2026
7
-ARG TRITON_VERSION=3.8.0+gitc0a142ff
6
+ARG TRITON_BRANCH="669b31ac" # release/internal/3.8.x as of 09/29/2026
7
+ARG TRITON_VERSION=3.8.0+git669b31ac
8
ARG TRITON_REPO="https://github.com/ROCm/triton.git"
9
ARG FA_BRANCH="0e60e394"
10
ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"