From d973c78e7986941ce7f96871cdf30711baecc5ce Mon Sep 17 00:00:00 2001 From: HAI Date: Fri, 14 Feb 2025 10:26:32 -0800 Subject: [PATCH] ROCm docker: triton update (#3584) --- docker/Dockerfile.rocm | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm index 44217d227..1c55bd31d 100644 --- a/docker/Dockerfile.rocm +++ b/docker/Dockerfile.rocm @@ -13,8 +13,8 @@ ARG SGL_REPO="https://github.com/sgl-project/sglang" ENV SGL_DEFAULT="main" ARG SGL_BRANCH=${SGL_DEFAULT} -ARG TRITON_REPO="https://github.com/triton-lang/triton.git" -ARG TRITON_COMMIT="845d75a" +ARG TRITON_REPO="https://github.com/ROCm/triton.git" +ARG TRITON_COMMIT="improve_fa_decode_3.0.0" ARG ATER_REPO="https://github.com/HaiShaw/ater"