[Build] Update DeepGEMM pin to deepseek-ai nv_dev tip (#52035)
Signed-off-by: Yongye Zhu <zyy1102000@gmail.com> Co-authored-by: Kimi Code <noreply@moonshot.ai>
Yongye Zhu committed
Aug 12, 2026 at 13:49 UTC
025d56a11e8f9bdf614af26e5386cdd8d322aaef
2 files changed
+6
-6
cmake/external_projects/deepgemm.cmake
+3
-3
@@ -28,9 +28,9 @@ if(DEEPGEMM_SRC_DIR)
28
message(STATUS "DeepGEMM using local DEEPGEMM_SRC_DIR: ${deepgemm_SOURCE_DIR}")
29
else()
30
# Keep in sync with tools/install_deepgemm.sh
31
- set(_DEEPGEMM_UPSTREAM_REPO "https://github.com/vllm-project/DeepGEMM.git")
32
- # TODO: switch to nv_dev branch after it support situ
33
- set(_DEEPGEMM_UPSTREAM_TAG "e21c821f39a2056d68067a466c64ddc942200106")
31
+ set(_DEEPGEMM_UPSTREAM_REPO "https://github.com/deepseek-ai/DeepGEMM.git")
32
+ # Pinned to the tip of the nv_dev branch (SM120 support).
33
+ set(_DEEPGEMM_UPSTREAM_TAG "8b1392b978f5a03c828dd1711090d7fb50958b8a")
34
35
set(_deepgemm_fc_root "${FETCHCONTENT_BASE_DIR}")
36
if(NOT _deepgemm_fc_root)
tools/install_deepgemm.sh
+3
-3
@@ -6,9 +6,9 @@ set -e
6
7
# Default values
8
# Keep DEEPGEMM_GIT_REF in sync with cmake/external_projects/deepgemm.cmake
9
-DEEPGEMM_GIT_REPO="https://github.com/vllm-project/DeepGEMM.git"
10
-# NOTE: This is currently targeting nv-dev branch due to sm120 support
11
-DEEPGEMM_GIT_REF="e21c821f39a2056d68067a466c64ddc942200106"
9
+DEEPGEMM_GIT_REPO="https://github.com/deepseek-ai/DeepGEMM.git"
10
+# NOTE: This is currently targeting the nv_dev branch tip due to sm120 support
11
+DEEPGEMM_GIT_REF="8b1392b978f5a03c828dd1711090d7fb50958b8a"
12
WHEEL_DIR=""
13
14
# Parse command line arguments