diff --git a/Dockerfile b/Dockerfile index 00e388d..cda048c 100644 --- a/Dockerfile +++ b/Dockerfile @@ -102,7 +102,7 @@ RUN git clone https://github.com/triton-lang/triton.git # We expect TRITON_REF to be passed from the command line to break the cache # Set to v3.5.1 tag by default -ARG TRITON_REF=v3.5.1 +ARG TRITON_REF=v3.6.0 WORKDIR $VLLM_BASE_DIR/triton diff --git a/build-and-copy.sh b/build-and-copy.sh index d707d72..ad01c78 100755 --- a/build-and-copy.sh +++ b/build-and-copy.sh @@ -11,7 +11,7 @@ REBUILD_VLLM=false COPY_HOSTS=() SSH_USER="$USER" NO_BUILD=false -TRITON_REF="v3.5.1" +TRITON_REF="v3.6.0" VLLM_REF="main" TMP_IMAGE="" PARALLEL_COPY=false