Update DeepGEMM tag to point to latest nv-dev branch for sm120 support (#47304)
Signed-off-by: mgoin <mgoin64@gmail.com>
Michael Goin committed
Jul 2, 2026 at 06:33 UTC
84b9c2762f5fdfa6ec1b133397e9ea460bd19152
2 files changed
+4
-2
cmake/external_projects/deepgemm.cmake
+2
-1
@@ -29,7 +29,8 @@ if(DEEPGEMM_SRC_DIR)
29
else()
30
# Keep in sync with tools/install_deepgemm.sh
31
set(_DEEPGEMM_UPSTREAM_REPO "https://github.com/deepseek-ai/DeepGEMM.git")
32
- set(_DEEPGEMM_UPSTREAM_TAG "891d57b4db1071624b5c8fa0d1e51cb317fa709f")
32
+ # NOTE: This is currently targeting nv-dev branch due to sm120 support
33
+ set(_DEEPGEMM_UPSTREAM_TAG "a6b593d2826719dcf4892609af7b84ee23aaf32a")
34
35
set(_deepgemm_fc_root "${FETCHCONTENT_BASE_DIR}")
36
if(NOT _deepgemm_fc_root)
tools/install_deepgemm.sh
+2
-1
@@ -7,7 +7,8 @@ set -e
7
# Default values
8
# Keep DEEPGEMM_GIT_REF in sync with cmake/external_projects/deepgemm.cmake
9
DEEPGEMM_GIT_REPO="https://github.com/deepseek-ai/DeepGEMM.git"
10
-DEEPGEMM_GIT_REF="891d57b4db1071624b5c8fa0d1e51cb317fa709f"
10
+# NOTE: This is currently targeting nv-dev branch due to sm120 support
11
+DEEPGEMM_GIT_REF="a6b593d2826719dcf4892609af7b84ee23aaf32a"
12
WHEEL_DIR=""
13
14
# Parse command line arguments