From 770a2cf7ae2de8bbdab35f0e1aeeebac4cf2ece9 Mon Sep 17 00:00:00 2001 From: youkaichao Date: Mon, 29 Sep 2025 17:51:43 +0800 Subject: [PATCH] update to latest deepgemm for dsv3.2 (#25871) Signed-off-by: youkaichao Signed-off-by: yewentao256 --- docker/Dockerfile | 2 +- tools/install_deepgemm.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 62be0a12e988c..a6b24f69f64ed 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -456,7 +456,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \ 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}" ${DEEPGEMM_GIT_REF:+--ref "$DEEPGEMM_GIT_REF"} + VLLM_DOCKER_BUILD_CONTEXT=1 TORCH_CUDA_ARCH_LIST="9.0a 10.0a" /tmp/install_deepgemm.sh --cuda-version "${CUDA_VERSION}" ${DEEPGEMM_GIT_REF:+--ref "$DEEPGEMM_GIT_REF"} COPY tools/install_gdrcopy.sh install_gdrcopy.sh RUN set -eux; \ diff --git a/tools/install_deepgemm.sh b/tools/install_deepgemm.sh index 98427f1835ec2..4f2cd302c3eff 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="ea9c5d9270226c5dd7a577c212e9ea385f6ef048" +DEEPGEMM_GIT_REF="594953acce41793ae00a1233eb516044d604bcb6" # Parse command line arguments while [[ $# -gt 0 ]]; do