Home Esplora Aiuto
Registrati Accedi
david
/
flash-attention
mirror da https://github.com/Dao-AILab/flash-attention
1
0
Forka 0
File Problemi 0 Wiki
Ramo (Branch): varlen
Rami (Branch) Tag
flash-attention
/
hopper
Tri Dao 4ead9bd7cc [FA3] Varlen forward 4 mesi fa
..
__init__.py 7f67966cc7 FA3 initial code release 5 mesi fa
block_info.h 7f67966cc7 FA3 initial code release 5 mesi fa
epilogue_fwd_sm90_tma.hpp 4ead9bd7cc [FA3] Varlen forward 4 mesi fa
flash.h 4ead9bd7cc [FA3] Varlen forward 4 mesi fa
flash_api.cpp 4ead9bd7cc [FA3] Varlen forward 4 mesi fa
flash_attn_interface.py 4ead9bd7cc [FA3] Varlen forward 4 mesi fa
flash_bwd_hdim128_fp16_sm90.cu 7f67966cc7 FA3 initial code release 5 mesi fa
flash_bwd_hdim256_fp16_sm90.cu 7f67966cc7 FA3 initial code release 5 mesi fa
flash_bwd_hdim64_fp16_sm90.cu 7f67966cc7 FA3 initial code release 5 mesi fa
flash_bwd_kernel.h 7f67966cc7 FA3 initial code release 5 mesi fa
flash_bwd_launch_template.h 7f67966cc7 FA3 initial code release 5 mesi fa
flash_bwd_preprocess_kernel.h 7f67966cc7 FA3 initial code release 5 mesi fa
flash_fwd_hdim128_bf16_sm90.cu 74b0761ff7 [FA3] BF16 forward 5 mesi fa
flash_fwd_hdim128_fp16_sm90.cu 7f67966cc7 FA3 initial code release 5 mesi fa
flash_fwd_hdim256_bf16_sm90.cu 74b0761ff7 [FA3] BF16 forward 5 mesi fa
flash_fwd_hdim256_fp16_sm90.cu 7f67966cc7 FA3 initial code release 5 mesi fa
flash_fwd_hdim64_bf16_sm90.cu 74b0761ff7 [FA3] BF16 forward 5 mesi fa
flash_fwd_hdim64_fp16_sm90.cu 7f67966cc7 FA3 initial code release 5 mesi fa
flash_fwd_kernel.h 4ead9bd7cc [FA3] Varlen forward 4 mesi fa
flash_fwd_launch_template.h 4ead9bd7cc [FA3] Varlen forward 4 mesi fa
kernel_traits.h 7f67966cc7 FA3 initial code release 5 mesi fa
mainloop_fwd_sm90_tma_gmma_ws.hpp 4ead9bd7cc [FA3] Varlen forward 4 mesi fa
named_barrier.hpp 74b0761ff7 [FA3] BF16 forward 5 mesi fa
setup.py 74b0761ff7 [FA3] BF16 forward 5 mesi fa
softmax.h 7f67966cc7 FA3 initial code release 5 mesi fa
static_switch.h 7f67966cc7 FA3 initial code release 5 mesi fa
test_flash_attn.py 4ead9bd7cc [FA3] Varlen forward 4 mesi fa
tile_scheduler.hpp 4ead9bd7cc [FA3] Varlen forward 4 mesi fa
utils.h 7f67966cc7 FA3 initial code release 5 mesi fa