 juejuezi
|
e371bea04f
feat: change minimal supported CUDA version to 11.7 (#1206)
|
hace 3 meses |
 Garrett Byrd
|
16025d8cc9
Clearer install instructions for CUDA and ROCm backends (#1147)
|
hace 4 meses |
 Tri Dao
|
5ca83a9c71
Clean up softcapping bwd a bit
|
hace 4 meses |
 rocking
|
d8f104e97a
Support AMD ROCm on FlashAttention 2 (#1010)
|
hace 4 meses |
|
Tri Dao
|
7f67966cc7
FA3 initial code release
|
hace 5 meses |
|
Tri Dao
|
da11d1b853
Bump v2.6.0
|
hace 5 meses |
|
Tri Dao
|
320fb59487
Update citation
|
hace 6 meses |
|
Tri Dao
|
2406f28805
Enable headdim 256 backward on consumer GPUs (Ampere, Ada)
|
hace 9 meses |
 Tao He
|
204c3c6d1b
Fixes an error in comment (#785)
|
hace 10 meses |
|
Tri Dao
|
54e80a3829
Implement page KV cache
|
hace 10 meses |
 Erich Schubert
|
99ea4baa1d
Typo in README (#760)
|
hace 11 meses |
|
Tri Dao
|
732654583c
Implement deterministic backward (thanks to Meituan)
|
hace 11 meses |
|
Tri Dao
|
50d144c906
Mention Alibi in README
|
hace 11 meses |
|
Tri Dao
|
7f31e7c16a
Bump to v2.3.2
|
hace 1 año |
|
Tri Dao
|
5a83425442
Change constexpr int to constexpr static int
|
hace 1 año |
|
Tri Dao
|
3a9fe7b0fa
Add change log
|
hace 1 año |
|
Tri Dao
|
aa4fd2d166
Clarify that Windows is not supported right now
|
hace 1 año |
|
Tri Dao
|
0c04943fa2
Require CUDA 11.6+, clean up setup.py
|
hace 1 año |
 Jeffrey Quesnelle
|
1d817a8ffc
fix citation in README (#501)
|
hace 1 año |
|
Tri Dao
|
45ba93cd96
Add newlines to README
|
hace 1 año |
|
Tri Dao
|
9e5e8bc91e
Change causal mask to be aligned to bottom-right instead of top-left
|
hace 1 año |
|
Tri Dao
|
d30f2e1cd5
Bump to v2.0.4
|
hace 1 año |
 Ian Timmis
|
cbf982afa5
README syntax highlighting (#365)
|
hace 1 año |
|
Tri Dao
|
d1a3b52f17
Add instruction about limiting number of ninja jobs
|
hace 1 año |
|
Tri Dao
|
b4cc152e97
Make sure dout is contiguous
|
hace 1 año |
|
Tri Dao
|
4f285b3547
FlashAttention-2 release
|
hace 1 año |
|
Tri Dao
|
ce68305c84
Update installation instruction
|
hace 1 año |
|
Tri Dao
|
f0c40b7ddb
Recommend Nvidia's Pytorch container
|
hace 1 año |
|
Tri Dao
|
40a25c8ee7
Update roadmap
|
hace 1 año |
 Anthony Hu
|
d63cfc3551
Use pyproject.toml to specify build dependencies
|
hace 1 año |
