Skip to content

Commit c215c6a

Browse files
add fp16 source
Signed-off-by: Lucas Wilkinson <lwilkins@redhat.com>
1 parent 48207c9 commit c215c6a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

cmake/external_projects/flashmla.cmake

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,8 @@ cuda_archs_loose_intersection(FLASH_MLA_ARCHS "9.0a" "${CUDA_ARCHS}")
3737
if(${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/flash_fwd_mla_bf16_sm90.cu)
40+
${flashmla_SOURCE_DIR}/csrc/flash_fwd_mla_bf16_sm90.cu
41+
${flashmla_SOURCE_DIR}/csrc/flash_fwd_mla_fp16_sm90.cu)
4142

4243
set(FlashMLA_INCLUDES
4344
${flashmla_SOURCE_DIR}/csrc/cutlass/include

0 commit comments

Comments
 (0)