Trang chủ Khám phá Trợ giúp
Đăng ký Đăng nhập
david
/
flash-attention
mirror of https://github.com/Dao-AILab/flash-attention
1
0
Fork 0
Các tập tin Các vấn đề 0 Wiki
Branch: varlen
Branches Tags
flash-attention
/
csrc
Tri Dao 74b0761ff7 [FA3] BF16 forward 5 tháng trước cách đây
..
cutlass @ 756c351b49 74b0761ff7 [FA3] BF16 forward 5 tháng trước cách đây
flash_attn 40e534a7f6 Implement cache_leftpad 5 tháng trước cách đây
ft_attention 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 tháng trước cách đây
fused_dense_lib 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 tháng trước cách đây
fused_softmax 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 tháng trước cách đây
layer_norm 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 tháng trước cách đây
rotary 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 tháng trước cách đây
xentropy 50896ec574 Make nvcc threads configurable via environment variable (#885) 9 tháng trước cách đây