From dc1a53186d8dbb3b450174a38a10c474b55f22e9 Mon Sep 17 00:00:00 2001 From: Jee Jee Li Date: Mon, 1 Sep 2025 17:38:04 +0800 Subject: [PATCH] [Kernel] Update DeepGEMM to latest commit (#23915) Signed-off-by: Jee Jee Li Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com> --- docker/Dockerfile | 5 ++--- tools/install_deepgemm.sh | 2 +- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 2e272cbca8417..75e8fa49f86c9 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -432,11 +432,10 @@ RUN --mount=type=cache,target=/root/.cache/uv \ --extra-index-url ${PYTORCH_CUDA_INDEX_BASE_URL}/cu$(echo $CUDA_VERSION | cut -d. -f1,2 | tr -d '.') # Install DeepGEMM from source -ARG DEEPGEMM_GIT_REF="7b6b5563b9d4c1ae07ffbce7f78ad3ac9204827c" +ARG DEEPGEMM_GIT_REF COPY tools/install_deepgemm.sh /tmp/install_deepgemm.sh RUN --mount=type=cache,target=/root/.cache/uv \ - VLLM_DOCKER_BUILD_CONTEXT=1 /tmp/install_deepgemm.sh --cuda-version "${CUDA_VERSION}" --ref "${DEEPGEMM_GIT_REF}" \ - && rm /tmp/install_deepgemm.sh + VLLM_DOCKER_BUILD_CONTEXT=1 /tmp/install_deepgemm.sh --cuda-version "${CUDA_VERSION}" ${DEEPGEMM_GIT_REF:+--ref "$DEEPGEMM_GIT_REF"} # Install EP kernels(pplx-kernels and DeepEP), NixL COPY tools/ep_kernels/install_python_libraries.sh install_python_libraries.sh diff --git a/tools/install_deepgemm.sh b/tools/install_deepgemm.sh index 33849581d2c0e..b125cda96f179 100755 --- a/tools/install_deepgemm.sh +++ b/tools/install_deepgemm.sh @@ -6,7 +6,7 @@ set -e # Default values DEEPGEMM_GIT_REPO="https://github.com/deepseek-ai/DeepGEMM.git" -DEEPGEMM_GIT_REF="7b6b5563b9d4c1ae07ffbce7f78ad3ac9204827c" +DEEPGEMM_GIT_REF="ea9c5d9270226c5dd7a577c212e9ea385f6ef048" # Parse command line arguments while [[ $# -gt 0 ]]; do