build(sgl-kernel): support configurable mirrors for restricted networks (#27387)

This commit is contained in:
Yingchun Lai
2026-06-08 08:52:26 -07:00
committed by GitHub
parent 3607cbd65a
commit b047bb3e92
5 changed files with 82 additions and 27 deletions
+11 -3
View File
@@ -1,12 +1,20 @@
# flash_mla
FetchContent_Declare(
repo-flashmla
GIT_REPOSITORY https://github.com/sgl-project/FlashMLA
GIT_TAG df022ebafb88578eab9f0300606ee765608d8b5c
GIT_SHALLOW OFF
URL https://${GITHUB_ARTIFACTORY}/sgl-project/FlashMLA/archive/df022ebafb88578eab9f0300606ee765608d8b5c.tar.gz
URL_HASH SHA256=45992d7de7d051dc897aff33156a2b7515d745fc489008adc7924ab884578d52
)
FetchContent_Populate(repo-flashmla)
# flashmla submodule pin: NVIDIA/cutlass @ 147f5673d0c1c3dcf66f78d677fd647e4a020219
FetchContent_Declare(
repo-flashmla-cutlass
URL https://${GITHUB_ARTIFACTORY}/NVIDIA/cutlass/archive/147f5673d0c1c3dcf66f78d677fd647e4a020219.tar.gz
URL_HASH SHA256=9f6c53320a85b4a570975e557918cde65168cd311f081920446c238437347dc6
SOURCE_DIR ${repo-flashmla_SOURCE_DIR}/csrc/cutlass
)
FetchContent_Populate(repo-flashmla-cutlass)
set(FLASHMLA_CUDA_FLAGS
"--expt-relaxed-constexpr"
"--expt-extended-lambda"