From 0a618e42d0b304c650e2d757d18e21242066d4bf Mon Sep 17 00:00:00 2001 From: Chris Leonard Date: Tue, 13 Oct 2026 19:55:15 +0000 Subject: [PATCH 1/4] update deep_gemm git tag for locality_domain update Signed-off-by: Chris Leonard --- cmake/external_projects/deepgemm.cmake | 6 +++--- tools/install_deepgemm.sh | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/cmake/external_projects/deepgemm.cmake b/cmake/external_projects/deepgemm.cmake index b796623b7ac1..39a3c01db10e 100644 --- a/cmake/external_projects/deepgemm.cmake +++ b/cmake/external_projects/deepgemm.cmake @@ -28,12 +28,12 @@ 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") + set(_DEEPGEMM_UPSTREAM_REPO "https://github.com/cleonard530/DeepGEMM.git") # Pinned to the fork's dev branch: upstream's 26/09/30 release (2.8.1) # plus the SM120 port, the SM90 paged-MQA kv_block=32/next_n=4 port, # configurable SwiGLU alpha/beta, SiTU for FP8/FP4 Mega MoE, and NVFP4 - # Mega MoE. TORCH_LIBRARY build with a CPython 3.10 stable-ABI floor. - set(_DEEPGEMM_UPSTREAM_TAG "6ce1c5ec6627a6bf11dbcc698b0b1e19c9248432") + # Mega MoE. TORCH_STABLE_LIBRARY build with a CPython 3.10 stable-ABI floor. + set(_DEEPGEMM_UPSTREAM_TAG "6af2d3a8a74970dc8994a07dac4b2e37ea197a80") 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 b617e6d5679f..35650c48a04c 100755 --- a/tools/install_deepgemm.sh +++ b/tools/install_deepgemm.sh @@ -6,11 +6,11 @@ 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" +DEEPGEMM_GIT_REPO="https://github.com/cleonard530/DeepGEMM.git" # NOTE: This targets the vLLM fork's dev branch: upstream's 26/09/30 release # plus the sm120 and sm90 paged-MQA ports, the SwiGLU alpha/beta and SiTU # Mega MoE activations, and NVFP4 Mega MoE. -DEEPGEMM_GIT_REF="6ce1c5ec6627a6bf11dbcc698b0b1e19c9248432" +DEEPGEMM_GIT_REF="6af2d3a8a74970dc8994a07dac4b2e37ea197a80" WHEEL_DIR="" # Parse command line arguments From 2ea176d911e9a02556e412d4aa483578298399ce Mon Sep 17 00:00:00 2001 From: Chris Leonard Date: Wed, 14 Oct 2026 02:22:58 +0000 Subject: [PATCH 2/4] update deep_gemm git tag Signed-off-by: Chris Leonard --- cmake/external_projects/deepgemm.cmake | 2 +- tools/install_deepgemm.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cmake/external_projects/deepgemm.cmake b/cmake/external_projects/deepgemm.cmake index 39a3c01db10e..1bbae7e40e20 100644 --- a/cmake/external_projects/deepgemm.cmake +++ b/cmake/external_projects/deepgemm.cmake @@ -33,7 +33,7 @@ else() # plus the SM120 port, the SM90 paged-MQA kv_block=32/next_n=4 port, # configurable SwiGLU alpha/beta, SiTU for FP8/FP4 Mega MoE, and NVFP4 # Mega MoE. TORCH_STABLE_LIBRARY build with a CPython 3.10 stable-ABI floor. - set(_DEEPGEMM_UPSTREAM_TAG "6af2d3a8a74970dc8994a07dac4b2e37ea197a80") + set(_DEEPGEMM_UPSTREAM_TAG "695dbadc176ca50340483220bce84c521ef2420c") 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 35650c48a04c..49c40e18b9ce 100755 --- a/tools/install_deepgemm.sh +++ b/tools/install_deepgemm.sh @@ -10,7 +10,7 @@ DEEPGEMM_GIT_REPO="https://github.com/cleonard530/DeepGEMM.git" # NOTE: This targets the vLLM fork's dev branch: upstream's 26/09/30 release # plus the sm120 and sm90 paged-MQA ports, the SwiGLU alpha/beta and SiTU # Mega MoE activations, and NVFP4 Mega MoE. -DEEPGEMM_GIT_REF="6af2d3a8a74970dc8994a07dac4b2e37ea197a80" +DEEPGEMM_GIT_REF="695dbadc176ca50340483220bce84c521ef2420c" WHEEL_DIR="" # Parse command line arguments From 2a29cc5ba4660f586a0b7319ede8fa330e673026 Mon Sep 17 00:00:00 2001 From: Chris Leonard Date: Wed, 14 Oct 2026 03:10:46 +0000 Subject: [PATCH 3/4] update deep_gemm git tag again Signed-off-by: Chris Leonard --- cmake/external_projects/deepgemm.cmake | 4 ++-- tools/install_deepgemm.sh | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/cmake/external_projects/deepgemm.cmake b/cmake/external_projects/deepgemm.cmake index 1bbae7e40e20..59b453b6e6e7 100644 --- a/cmake/external_projects/deepgemm.cmake +++ b/cmake/external_projects/deepgemm.cmake @@ -28,12 +28,12 @@ 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/cleonard530/DeepGEMM.git") + set(_DEEPGEMM_UPSTREAM_REPO "https://github.com/vllm-project/DeepGEMM.git") # Pinned to the fork's dev branch: upstream's 26/09/30 release (2.8.1) # plus the SM120 port, the SM90 paged-MQA kv_block=32/next_n=4 port, # configurable SwiGLU alpha/beta, SiTU for FP8/FP4 Mega MoE, and NVFP4 # Mega MoE. TORCH_STABLE_LIBRARY build with a CPython 3.10 stable-ABI floor. - set(_DEEPGEMM_UPSTREAM_TAG "695dbadc176ca50340483220bce84c521ef2420c") + set(_DEEPGEMM_UPSTREAM_TAG "90a20e2df1d6c79a0365ec8ef964f81a51a2cbb7") 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 49c40e18b9ce..c22d331dbae4 100755 --- a/tools/install_deepgemm.sh +++ b/tools/install_deepgemm.sh @@ -6,11 +6,11 @@ set -e # Default values # Keep DEEPGEMM_GIT_REF in sync with cmake/external_projects/deepgemm.cmake -DEEPGEMM_GIT_REPO="https://github.com/cleonard530/DeepGEMM.git" +DEEPGEMM_GIT_REPO="https://github.com/vllm-project/DeepGEMM.git" # NOTE: This targets the vLLM fork's dev branch: upstream's 26/09/30 release # plus the sm120 and sm90 paged-MQA ports, the SwiGLU alpha/beta and SiTU # Mega MoE activations, and NVFP4 Mega MoE. -DEEPGEMM_GIT_REF="695dbadc176ca50340483220bce84c521ef2420c" +DEEPGEMM_GIT_REF="90a20e2df1d6c79a0365ec8ef964f81a51a2cbb7" WHEEL_DIR="" # Parse command line arguments From e72445e82f28a5aec276b0b9d88bd8b0f425f82a Mon Sep 17 00:00:00 2001 From: Shengqi Chen Date: Wed, 14 Oct 2026 11:59:59 +0800 Subject: [PATCH 4/4] Remove deep_gemm shared library from ABI check Signed-off-by: Shengqi Chen --- .buildkite/check-torch-abi.py | 1 - 1 file changed, 1 deletion(-) diff --git a/.buildkite/check-torch-abi.py b/.buildkite/check-torch-abi.py index 37e2b8dadbf8..5bfec8abd077 100644 --- a/.buildkite/check-torch-abi.py +++ b/.buildkite/check-torch-abi.py @@ -17,7 +17,6 @@ # _C_stable_libtorch, but these two modules still bind libtorch directly. "_C.abi3.so", "_rocm_C.abi3.so", - "third_party/deep_gemm/_C*.so", )