Skip to content

Commit df59f71

Browse files
authored
[CMake][Bugfix] Set default value for FLASHINFER_GEN_MASK_MODES (#340)
This commit resolves a build-time error with the following message: ``` CMake Error at 3rdparty/flashinfer/CMakeLists.txt:313 (add_library): No SOURCES given to target: prefill_kernels ``` This occurred after #266, which replaces the `FLASHINFER_GEN_CASUALS` option with `FLASHINFER_GEN_MASK_MODES`. However, the definition of `flashinfer_option(FLASHINFER_GEN_CASUALS ... )` was not replaced. As a result, loop over the empty `MASK_MODES` does not produce any kernels that should be compiled. This commit updates the `flashinfer_option(FLASH_GEN_CASUALS ...)` line to instead define `FLASH_GEN_MASK_MODES`, using the same default value as `config.cmake`.
1 parent 95f507f commit df59f71

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

CMakeLists.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ flashinfer_option(FLASHINFER_GEN_KV_LAYOUTS "KV layouts to enable" 0 1)
4242
flashinfer_option(FLASHINFER_GEN_LOGITS_POST_HOOKS "Logits post hooks" 0 1)
4343
flashinfer_option(FLASHINFER_GEN_POS_ENCODING_MODES "Pos encodings to enable" 0 1 2)
4444
flashinfer_option(FLASHINFER_GEN_ALLOW_FP16_QK_REDUCTIONS "QK reductions to enable" "false" "true")
45-
flashinfer_option(FLASHINFER_GEN_CASUALS "Casual modes to enable" "false" "true")
45+
flashinfer_option(FLASHINFER_GEN_MASK_MODES "Mask modes to enable" 0 1 2)
4646

4747
if(DEFINED FLASHINFER_CUDA_ARCHITECTURES)
4848
message(STATUS "CMAKE_CUDA_ARCHITECTURES set to ${FLASHINFER_CUDA_ARCHITECTURES}.")

0 commit comments

Comments
 (0)