index
:
forks/candle.git
main
summary
refs
log
tree
commit
diff
log msg
author
committer
range
path:
root
/
candle-flash-attn
/
kernels
Mode
Name
Size
-rw-r--r--
alibi.h
2499
log
plain
-rw-r--r--
block_info.h
2242
log
plain
-rw-r--r--
flash.h
5302
log
plain
-rw-r--r--
flash_api.cu
3631
log
plain
-rw-r--r--
flash_fwd_hdim128_bf16_sm80.cu
386
log
plain
-rw-r--r--
flash_fwd_hdim128_fp16_sm80.cu
378
log
plain
-rw-r--r--
flash_fwd_hdim160_bf16_sm80.cu
386
log
plain
-rw-r--r--
flash_fwd_hdim160_fp16_sm80.cu
378
log
plain
-rw-r--r--
flash_fwd_hdim192_bf16_sm80.cu
386
log
plain
-rw-r--r--
flash_fwd_hdim192_fp16_sm80.cu
378
log
plain
-rw-r--r--
flash_fwd_hdim224_bf16_sm80.cu
386
log
plain
-rw-r--r--
flash_fwd_hdim224_fp16_sm80.cu
378
log
plain
-rw-r--r--
flash_fwd_hdim256_bf16_sm80.cu
386
log
plain
-rw-r--r--
flash_fwd_hdim256_fp16_sm80.cu
378
log
plain
-rw-r--r--
flash_fwd_hdim32_bf16_sm80.cu
384
log
plain
-rw-r--r--
flash_fwd_hdim32_fp16_sm80.cu
376
log
plain
-rw-r--r--
flash_fwd_hdim64_bf16_sm80.cu
384
log
plain
-rw-r--r--
flash_fwd_hdim64_fp16_sm80.cu
376
log
plain
-rw-r--r--
flash_fwd_hdim96_bf16_sm80.cu
384
log
plain
-rw-r--r--
flash_fwd_hdim96_fp16_sm80.cu
376
log
plain
-rw-r--r--
flash_fwd_kernel.h
33338
log
plain
-rw-r--r--
flash_fwd_launch_template.h
16266
log
plain
-rw-r--r--
kernel_traits.h
20443
log
plain
-rw-r--r--
kernel_traits_sm90.h
7555
log
plain
-rw-r--r--
philox.cuh
5372
log
plain
-rw-r--r--
softmax.h
15114
log
plain
-rw-r--r--
static_switch.h
2598
log
plain
-rw-r--r--
utils.h
15787
log
plain
