Skip to content

Commit 809abaa

Browse files
authored
cmake: fix DECODE_F8_DTYPES and DECODE_FP8_DTYPES discrepancy (#287)
As a result of typo, it appeared two variables: `DECODE_F8_DTYPES` and `DECODE_FP8_DTYPES` in CMakeLists.txt. But it should be the same. This commit fixes this discrepancy and rename `DECODE_F8_DTYPES` -> `DECODE_FP8_DTYPES`.
1 parent 5a38066 commit 809abaa

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

CMakeLists.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -84,12 +84,12 @@ set (ALLOW_FP16_QK_REDUCTIONS ${FLASHINFER_GEN_ALLOW_FP16_QK_REDUCTIONS})
8484
set (MASK_MODES ${FLASHINFER_GEN_MASK_MODES})
8585
set (DECODE_DTYPES "f16")
8686
set (PREFILL_DTYPES "f16")
87-
set (DECODE_F8_DTYPES)
87+
set (DECODE_FP8_DTYPES)
8888
set (IDTYPES "i32")
8989

9090
if(FLASHINFER_ENABLE_FP8)
9191
list(APPEND DECODE_DTYPES "e4m3" "e5m2")
92-
list(APPEND DECODE_F8_DTYPES "e4m3" "e5m2")
92+
list(APPEND DECODE_FP8_DTYPES "e4m3" "e5m2")
9393
endif(FLASHINFER_ENABLE_FP8)
9494

9595
if(FLASHINFER_ENABLE_BF16)
@@ -136,7 +136,7 @@ foreach(group_size IN LISTS GROUP_SIZES)
136136
endforeach(dtype)
137137

138138
# fp8 in, fp16 out
139-
foreach(dtype IN LISTS DECODE_F8_DTYPES)
139+
foreach(dtype IN LISTS DECODE_FP8_DTYPES)
140140
set(generated_kernel_src ${PROJECT_SOURCE_DIR}/src/generated/single_decode_group_${group_size}_head_${head_dim}_layout_${kv_layout}_posenc_${pos_encoding_mode}_dtypein_${dtype}_dtypeout_f16.cu)
141141
add_custom_command(
142142
OUTPUT ${generated_kernel_src}

0 commit comments

Comments
 (0)