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

Mode	Name	Size
-rw-r--r--	block_info.h	1664	log plain
-rw-r--r--	flash.h	4132	log plain
-rw-r--r--	flash_api.cu	3126	log plain
-rw-r--r--	flash_fwd_hdim128_bf16_sm80.cu	783	log plain
-rw-r--r--	flash_fwd_hdim128_fp16_sm80.cu	1951	log plain
-rw-r--r--	flash_fwd_hdim160_bf16_sm80.cu	682	log plain
-rw-r--r--	flash_fwd_hdim160_fp16_sm80.cu	1674	log plain
-rw-r--r--	flash_fwd_hdim192_bf16_sm80.cu	682	log plain
-rw-r--r--	flash_fwd_hdim192_fp16_sm80.cu	1647	log plain
-rw-r--r--	flash_fwd_hdim224_bf16_sm80.cu	329	log plain
-rw-r--r--	flash_fwd_hdim224_fp16_sm80.cu	321	log plain
-rw-r--r--	flash_fwd_hdim256_bf16_sm80.cu	329	log plain
-rw-r--r--	flash_fwd_hdim256_fp16_sm80.cu	321	log plain
-rw-r--r--	flash_fwd_hdim32_bf16_sm80.cu	326	log plain
-rw-r--r--	flash_fwd_hdim32_fp16_sm80.cu	1151	log plain
-rw-r--r--	flash_fwd_hdim64_bf16_sm80.cu	777	log plain
-rw-r--r--	flash_fwd_hdim64_fp16_sm80.cu	1463	log plain
-rw-r--r--	flash_fwd_hdim96_bf16_sm80.cu	677	log plain
-rw-r--r--	flash_fwd_hdim96_fp16_sm80.cu	1328	log plain
-rw-r--r--	flash_fwd_kernel.h	30304	log plain
-rw-r--r--	flash_fwd_launch_template.h	15262	log plain
-rw-r--r--	kernel_traits.h	18396	log plain
-rw-r--r--	philox.cuh	5372	log plain
-rw-r--r--	softmax.h	14205	log plain
-rw-r--r--	static_switch.h	2598	log plain
-rw-r--r--	utils.h	16378	log plain