File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed
Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -37,13 +37,14 @@ cuda_archs_loose_intersection(FLASH_MLA_ARCHS "9.0a" "${CUDA_ARCHS}")
3737if (${CMAKE_CUDA_COMPILER_VERSION} VERSION_GREATER 12.3 AND FLASH_MLA_ARCHS)
3838 set (FlashMLA_SOURCES
3939 ${flashmla_SOURCE_DIR} /csrc/flash_api.cpp
40- ${flashmla_SOURCE_DIR} /csrc/kernels/splitkv_mla .cu
40+ ${flashmla_SOURCE_DIR} /csrc/kernels/get_mla_metadata .cu
4141 ${flashmla_SOURCE_DIR} /csrc/kernels/mla_combine.cu
42- ${flashmla_SOURCE_DIR} /csrc/kernels/get_mla_metadata.cu)
42+ ${flashmla_SOURCE_DIR} /csrc/kernels/splitkv_mla.cu
43+ ${flashmla_SOURCE_DIR} /csrc/kernels_fp8/flash_fwd_mla_fp8_sm90.cu)
4344
4445 set (FlashMLA_INCLUDES
4546 ${flashmla_SOURCE_DIR} /csrc/cutlass/include
46- ${flashmla_SOURCE_DIR} /csrc/include )
47+ ${flashmla_SOURCE_DIR} /csrc)
4748
4849 set_gencode_flags_for_srcs (
4950 SRCS "${FlashMLA_SOURCES} "
You can’t perform that action at this time.
0 commit comments