diff --git a/sgl-kernel/CMakeLists.txt b/sgl-kernel/CMakeLists.txt index 72225e55b..aeefd3371 100644 --- a/sgl-kernel/CMakeLists.txt +++ b/sgl-kernel/CMakeLists.txt @@ -54,6 +54,9 @@ FetchContent_Populate(repo-cutlass) if("${CUDA_VERSION}" VERSION_EQUAL "12.8") set(DeepGEMM_REPO "https://github.com/sgl-project/DeepGEMM") set(DeepGEMM_TAG "blackwell") +elseif("${CUDA_VERSION}" VERSION_EQUAL "12.9") + set(DeepGEMM_REPO "https://github.com/sgl-project/DeepGEMM") + set(DeepGEMM_TAG "blackwell") else() set(DeepGEMM_REPO "https://github.com/deepseek-ai/DeepGEMM") set(DeepGEMM_TAG "391755ada0ffefa9a6a52b6f14dcaf22d1a463e0")