vllm-flash-attn3 / flash-attn

1.01 MB

Ctrl+K

6 contributors

History: 3 commits

danieldk HF Staff

Sync with upstream

a64d4b2 9 months ago

instantiations
Convert FA3 to Kernel Hub format 10 months ago
block.h

5.58 kB
Convert FA3 to Kernel Hub format 10 months ago
copy_sm90_bulk_reduce.hpp

1.78 kB
Convert FA3 to Kernel Hub format 10 months ago
cuda_check.h

1.03 kB
Convert FA3 to Kernel Hub format 10 months ago
epilogue_bwd.hpp

30.2 kB
Convert FA3 to Kernel Hub format 10 months ago
epilogue_fwd.hpp

29.2 kB
Convert FA3 to Kernel Hub format 10 months ago
flash.h

6.76 kB
Convert FA3 to Kernel Hub format 10 months ago
flash_api.cpp

83.8 kB
Sync with upstream 9 months ago
flash_bwd_kernel_sm80.h

6.84 kB
Convert FA3 to Kernel Hub format 10 months ago
flash_bwd_kernel_sm90.h

13.7 kB
Convert FA3 to Kernel Hub format 10 months ago
flash_bwd_launch_template.h

24 kB
Convert FA3 to Kernel Hub format 10 months ago
flash_bwd_postprocess_kernel.h

13.6 kB
Convert FA3 to Kernel Hub format 10 months ago
flash_bwd_preprocess_kernel.h

12.4 kB
Convert FA3 to Kernel Hub format 10 months ago
flash_fwd_combine.cu

929 Bytes
Convert FA3 to Kernel Hub format 10 months ago
flash_fwd_combine_kernel.h

35.5 kB
Sync with upstream 9 months ago
flash_fwd_combine_launch_template.h

5.15 kB
Sync with upstream 9 months ago
flash_fwd_kernel_sm80.h

10 kB
Convert FA3 to Kernel Hub format 10 months ago
flash_fwd_kernel_sm90.h

26.5 kB
Convert FA3 to Kernel Hub format 10 months ago
flash_fwd_launch_template.h

15.7 kB
Convert FA3 to Kernel Hub format 10 months ago
flash_prepare_scheduler.cu

5.97 kB
Convert FA3 to Kernel Hub format 10 months ago
heuristics.h

3.36 kB
Convert FA3 to Kernel Hub format 10 months ago
mainloop_bwd_sm80.hpp

55.1 kB
Convert FA3 to Kernel Hub format 10 months ago
mainloop_bwd_sm90_tma_gmma_ws.hpp

65.1 kB
Convert FA3 to Kernel Hub format 10 months ago
mainloop_fwd_sm80.hpp

51.1 kB
Convert FA3 to Kernel Hub format 10 months ago
mainloop_fwd_sm90_tma_gmma_ws.hpp

109 kB
Convert FA3 to Kernel Hub format 10 months ago
mask.h

8.85 kB
Convert FA3 to Kernel Hub format 10 months ago
named_barrier.hpp

2.87 kB
Convert FA3 to Kernel Hub format 10 months ago
pack_gqa.h

15.4 kB
Convert FA3 to Kernel Hub format 10 months ago
paged_kv.h

20.9 kB
Convert FA3 to Kernel Hub format 10 months ago
rotary.h

28.9 kB
Convert FA3 to Kernel Hub format 10 months ago
seqlen.h

4.55 kB
Convert FA3 to Kernel Hub format 10 months ago
sm90_pipeline_no_cluster.hpp

4.24 kB
Convert FA3 to Kernel Hub format 10 months ago
softmax.h

9.43 kB
Convert FA3 to Kernel Hub format 10 months ago
static_switch.h

12 kB
Convert FA3 to Kernel Hub format 10 months ago
tile_scheduler.hpp

28.3 kB
Convert FA3 to Kernel Hub format 10 months ago
tile_size.h

4.51 kB
Convert FA3 to Kernel Hub format 10 months ago
utils.h

32.3 kB
Sync with upstream 9 months ago