ROCm docker: triton update (#3584)

d973c78e · HAI · GitHub · 6ce6eabb · d973c78e
Unverified Commit d973c78e authored Feb 14, 2025 by HAI Committed by GitHub Feb 14, 2025
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

docker/Dockerfile.rocm docker/Dockerfile.rocm +2 -2

No files found.
--- 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"