kernels « candle-flash-attn - forks/candle.git -

Mode	Name	Size
-rw-r--r--	alibi.h	2824	log plain
-rw-r--r--	block_info.h	2426	log plain
-rw-r--r--	dropout.h	5773	log plain
-rw-r--r--	error.h	313	log plain
-rw-r--r--	flash.h	6005	log plain
-rw-r--r--	flash_api.cu	3831	log plain
-rw-r--r--	flash_fwd_hdim128_bf16_causal_sm80.cu	398	log plain
-rw-r--r--	flash_fwd_hdim128_bf16_sm80.cu	400	log plain
-rw-r--r--	flash_fwd_hdim128_fp16_causal_sm80.cu	390	log plain
-rw-r--r--	flash_fwd_hdim128_fp16_sm80.cu	392	log plain
-rw-r--r--	flash_fwd_hdim160_bf16_causal_sm80.cu	398	log plain
-rw-r--r--	flash_fwd_hdim160_bf16_sm80.cu	400	log plain
-rw-r--r--	flash_fwd_hdim160_fp16_causal_sm80.cu	390	log plain
-rw-r--r--	flash_fwd_hdim160_fp16_sm80.cu	392	log plain
-rw-r--r--	flash_fwd_hdim192_bf16_causal_sm80.cu	398	log plain
-rw-r--r--	flash_fwd_hdim192_bf16_sm80.cu	400	log plain
-rw-r--r--	flash_fwd_hdim192_fp16_causal_sm80.cu	390	log plain
-rw-r--r--	flash_fwd_hdim192_fp16_sm80.cu	392	log plain
-rw-r--r--	flash_fwd_hdim224_bf16_causal_sm80.cu	398	log plain
-rw-r--r--	flash_fwd_hdim224_bf16_sm80.cu	400	log plain
-rw-r--r--	flash_fwd_hdim224_fp16_causal_sm80.cu	390	log plain
-rw-r--r--	flash_fwd_hdim224_fp16_sm80.cu	392	log plain
-rw-r--r--	flash_fwd_hdim256_bf16_causal_sm80.cu	398	log plain
-rw-r--r--	flash_fwd_hdim256_bf16_sm80.cu	400	log plain
-rw-r--r--	flash_fwd_hdim256_fp16_causal_sm80.cu	390	log plain
-rw-r--r--	flash_fwd_hdim256_fp16_sm80.cu	392	log plain
-rw-r--r--	flash_fwd_hdim32_bf16_causal_sm80.cu	396	log plain
-rw-r--r--	flash_fwd_hdim32_bf16_sm80.cu	398	log plain
-rw-r--r--	flash_fwd_hdim32_fp16_causal_sm80.cu	388	log plain
-rw-r--r--	flash_fwd_hdim32_fp16_sm80.cu	390	log plain
-rw-r--r--	flash_fwd_hdim64_bf16_causal_sm80.cu	396	log plain
-rw-r--r--	flash_fwd_hdim64_bf16_sm80.cu	398	log plain
-rw-r--r--	flash_fwd_hdim64_fp16_causal_sm80.cu	388	log plain
-rw-r--r--	flash_fwd_hdim64_fp16_sm80.cu	390	log plain
-rw-r--r--	flash_fwd_hdim96_bf16_causal_sm80.cu	396	log plain
-rw-r--r--	flash_fwd_hdim96_bf16_sm80.cu	398	log plain
-rw-r--r--	flash_fwd_hdim96_fp16_causal_sm80.cu	388	log plain
-rw-r--r--	flash_fwd_hdim96_fp16_sm80.cu	390	log plain
-rw-r--r--	flash_fwd_kernel.h	76037	log plain
-rw-r--r--	flash_fwd_launch_template.h	23408	log plain
-rw-r--r--	hardware_info.h	1652	log plain
-rw-r--r--	kernel_helpers.h	1568	log plain
-rw-r--r--	kernel_traits.h	17697	log plain
-rw-r--r--	kernel_traits_sm90.h	7555	log plain
-rw-r--r--	kernels.h	1117	log plain
-rw-r--r--	mask.h	11224	log plain
-rw-r--r--	philox.cuh	1679	log plain
-rw-r--r--	rotary.h	8952	log plain
-rw-r--r--	softmax.h	9324	log plain
-rw-r--r--	static_switch.h	4065	log plain
-rw-r--r--	utils.h	18415	log plain