FlashMLA/csrc/flash_api
zhan3916 30f7561775 Performance: Add new tile for small bs
Signed-off-by: zhan3916 <zhan3916@gmail.com>
2025-03-25 11:54:33 +08:00
..
flash_api.cpp Performance: Add new tile for small bs 2025-03-25 11:54:33 +08:00
flash_fwd_splitkv_hdimqk576_hdimv512_m16n16_bf16_True_True_sm80.cu Performance: Add new tile for small bs 2025-03-25 11:54:33 +08:00
flash_fwd_splitkv_hdimqk576_hdimv512_m16n16_bf16_True_True_split_sm80.cu Performance: Add new tile for small bs 2025-03-25 11:54:33 +08:00
flash_fwd_splitkv_hdimqk576_hdimv512_m16n16_fp16_True_True_sm80.cu Performance: Add new tile for small bs 2025-03-25 11:54:33 +08:00
flash_fwd_splitkv_hdimqk576_hdimv512_m16n16_fp16_True_True_split_sm80.cu Performance: Add new tile for small bs 2025-03-25 11:54:33 +08:00
flash_fwd_splitkv_hdimqk576_hdimv512_m32n16_bf16_True_True_sm80.cu Performance: improve perf by new tile 2025-03-14 19:00:47 +08:00
flash_fwd_splitkv_hdimqk576_hdimv512_m32n16_bf16_True_True_split_sm80.cu Performance: improve perf by new tile 2025-03-14 19:00:47 +08:00
flash_fwd_splitkv_hdimqk576_hdimv512_m32n16_fp16_True_True_sm80.cu Performance: improve perf by new tile 2025-03-14 19:00:47 +08:00
flash_fwd_splitkv_hdimqk576_hdimv512_m32n16_fp16_True_True_split_sm80.cu Performance: improve perf by new tile 2025-03-14 19:00:47 +08:00
flash_fwd_splitkv_hdimqk576_hdimv512_m64n16_bf16_True_True_sm80.cu Performance: improve perf by new tile 2025-03-14 19:00:47 +08:00
flash_fwd_splitkv_hdimqk576_hdimv512_m64n16_bf16_True_True_split_sm80.cu Performance: improve perf by new tile 2025-03-14 19:00:47 +08:00
flash_fwd_splitkv_hdimqk576_hdimv512_m64n16_fp16_True_True_sm80.cu Performance: improve perf by new tile 2025-03-14 19:00:47 +08:00
flash_fwd_splitkv_hdimqk576_hdimv512_m64n16_fp16_True_True_split_sm80.cu Performance: improve perf by new tile 2025-03-14 19:00:47 +08:00
flash_mla.h Support splitKV>1 2025-03-06 13:57:41 +08:00