build(sgl-kernel): bump FlashMLA pin + fix cccl include for CUDA 13 (#29067)
This commit is contained in:
@@ -1,8 +1,9 @@
|
|||||||
# flash_mla
|
# flash_mla
|
||||||
|
# sm90 dense decode HEAD_DIM_K=512 support (sgl-project/FlashMLA#9, merged).
|
||||||
FetchContent_Declare(
|
FetchContent_Declare(
|
||||||
repo-flashmla
|
repo-flashmla
|
||||||
URL https://${GITHUB_ARTIFACTORY}/sgl-project/FlashMLA/archive/df022ebafb88578eab9f0300606ee765608d8b5c.tar.gz
|
URL https://${GITHUB_ARTIFACTORY}/sgl-project/FlashMLA/archive/05e26647fe840b8baedae486c2d86d5ce4efeb7c.tar.gz
|
||||||
URL_HASH SHA256=45992d7de7d051dc897aff33156a2b7515d745fc489008adc7924ab884578d52
|
URL_HASH SHA256=ce369489bbfc42cdfbba9aa949de0270e64469d530748dea9f4f60b3c69dea9b
|
||||||
)
|
)
|
||||||
FetchContent_Populate(repo-flashmla)
|
FetchContent_Populate(repo-flashmla)
|
||||||
|
|
||||||
@@ -156,6 +157,16 @@ target_compile_options(flashmla_ops PRIVATE
|
|||||||
if(FLASHMLA_ENABLE_SM100)
|
if(FLASHMLA_ENABLE_SM100)
|
||||||
target_compile_definitions(flashmla_ops PRIVATE FLASHMLA_ENABLE_SM100)
|
target_compile_definitions(flashmla_ops PRIVATE FLASHMLA_ENABLE_SM100)
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
|
# CUDA 13 moved cuda/std/* under cccl/cuda/std/*. The vendored cutlass routes
|
||||||
|
# <cuda/std/...> to <cccl/cuda/std/...> when __CUDACC_VER_MAJOR__ >= 13, so the
|
||||||
|
# host C++ TU (compiled by g++, where that macro is unset for the legacy path)
|
||||||
|
# needs the cccl include root on the search path.
|
||||||
|
if(CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL "13.0")
|
||||||
|
find_path(FLASHMLA_CCCL_INCLUDE NAMES cuda/std/utility
|
||||||
|
HINTS ${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES}
|
||||||
|
${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES}/cccl)
|
||||||
|
endif()
|
||||||
target_include_directories(flashmla_ops PRIVATE
|
target_include_directories(flashmla_ops PRIVATE
|
||||||
${repo-flashmla_SOURCE_DIR}/csrc
|
${repo-flashmla_SOURCE_DIR}/csrc
|
||||||
${repo-flashmla_SOURCE_DIR}/csrc/kerutils/include
|
${repo-flashmla_SOURCE_DIR}/csrc/kerutils/include
|
||||||
@@ -163,6 +174,7 @@ target_include_directories(flashmla_ops PRIVATE
|
|||||||
${repo-flashmla_SOURCE_DIR}/csrc/extension/sm90/dense_fp8/
|
${repo-flashmla_SOURCE_DIR}/csrc/extension/sm90/dense_fp8/
|
||||||
${repo-flashmla_SOURCE_DIR}/csrc/cutlass/include
|
${repo-flashmla_SOURCE_DIR}/csrc/cutlass/include
|
||||||
${repo-flashmla_SOURCE_DIR}/csrc/cutlass/tools/util/include
|
${repo-flashmla_SOURCE_DIR}/csrc/cutlass/tools/util/include
|
||||||
|
${FLASHMLA_CCCL_INCLUDE}
|
||||||
)
|
)
|
||||||
|
|
||||||
target_link_libraries(flashmla_ops PRIVATE ${TORCH_LIBRARIES} c10 cuda)
|
target_link_libraries(flashmla_ops PRIVATE ${TORCH_LIBRARIES} c10 cuda)
|
||||||
|
|||||||
Reference in New Issue
Block a user