From 025d56a11e8f9bdf614af26e5386cdd8d322aaef Mon Sep 17 00:00:00 2001 From: Yongye Zhu Date: Wed, 12 Aug 2026 13:49:04 -0700 Subject: [PATCH] [Build] Update DeepGEMM pin to deepseek-ai nv_dev tip (#52035) Signed-off-by: Yongye Zhu Co-authored-by: Kimi Code --- cmake/external_projects/deepgemm.cmake | 6 +++--- tools/install_deepgemm.sh | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/cmake/external_projects/deepgemm.cmake b/cmake/external_projects/deepgemm.cmake index be0eccd45a..72f7f5eb66 100644 --- a/cmake/external_projects/deepgemm.cmake +++ b/cmake/external_projects/deepgemm.cmake @@ -28,9 +28,9 @@ if(DEEPGEMM_SRC_DIR) message(STATUS "DeepGEMM using local DEEPGEMM_SRC_DIR: ${deepgemm_SOURCE_DIR}") else() # Keep in sync with tools/install_deepgemm.sh - set(_DEEPGEMM_UPSTREAM_REPO "https://github.com/vllm-project/DeepGEMM.git") - # TODO: switch to nv_dev branch after it support situ - set(_DEEPGEMM_UPSTREAM_TAG "e21c821f39a2056d68067a466c64ddc942200106") + set(_DEEPGEMM_UPSTREAM_REPO "https://github.com/deepseek-ai/DeepGEMM.git") + # Pinned to the tip of the nv_dev branch (SM120 support). + set(_DEEPGEMM_UPSTREAM_TAG "8b1392b978f5a03c828dd1711090d7fb50958b8a") set(_deepgemm_fc_root "${FETCHCONTENT_BASE_DIR}") if(NOT _deepgemm_fc_root) diff --git a/tools/install_deepgemm.sh b/tools/install_deepgemm.sh index 3f80ab5673..f6a64d8796 100755 --- a/tools/install_deepgemm.sh +++ b/tools/install_deepgemm.sh @@ -6,9 +6,9 @@ set -e # Default values # Keep DEEPGEMM_GIT_REF in sync with cmake/external_projects/deepgemm.cmake -DEEPGEMM_GIT_REPO="https://github.com/vllm-project/DeepGEMM.git" -# NOTE: This is currently targeting nv-dev branch due to sm120 support -DEEPGEMM_GIT_REF="e21c821f39a2056d68067a466c64ddc942200106" +DEEPGEMM_GIT_REPO="https://github.com/deepseek-ai/DeepGEMM.git" +# NOTE: This is currently targeting the nv_dev branch tip due to sm120 support +DEEPGEMM_GIT_REF="8b1392b978f5a03c828dd1711090d7fb50958b8a" WHEEL_DIR="" # Parse command line arguments