diff --git a/cmake/external_projects/deepgemm.cmake b/cmake/external_projects/deepgemm.cmake index 15a7d0cde6..be0eccd45a 100644 --- a/cmake/external_projects/deepgemm.cmake +++ b/cmake/external_projects/deepgemm.cmake @@ -30,7 +30,7 @@ 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 "f5a76426fa084087169693fd0cd815223576d6e9") + set(_DEEPGEMM_UPSTREAM_TAG "e21c821f39a2056d68067a466c64ddc942200106") 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 1d6ff5a80b..3f80ab5673 100755 --- a/tools/install_deepgemm.sh +++ b/tools/install_deepgemm.sh @@ -8,7 +8,7 @@ set -e # 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="f5a76426fa084087169693fd0cd815223576d6e9" +DEEPGEMM_GIT_REF="e21c821f39a2056d68067a466c64ddc942200106" WHEEL_DIR="" # Parse command line arguments