Skip to content

Commit 2fa62a9

Browse files
missing symbol
Signed-off-by: Lucas Wilkinson <lwilkinson@neuralmagic.com>
1 parent 642456f commit 2fa62a9

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
@@ -38,7 +38,8 @@ 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
4040
${flashmla_SOURCE_DIR}/csrc/flash_fwd_mla_bf16_sm90.cu
41-
${flashmla_SOURCE_DIR}/csrc/flash_fwd_mla_fp16_sm90.cu)
41+
${flashmla_SOURCE_DIR}/csrc/flash_fwd_mla_fp16_sm90.cu
42+
${flashmla_SOURCE_DIR}/csrc/flash_fwd_mla_metadata.cu)
4243

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

0 commit comments

Comments
 (0)