ROCm docker: triton update (#3584)

2025-02-14 10:26:32 -08:00
parent 6ce6eabbcc
commit d973c78e79
1 changed files with 2 additions and 2 deletions
--- 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"