 Tri Dao
|
7f67966cc7
FA3 initial code release
|
5 hónapja |
|
Tri Dao
|
da11d1b853
Bump v2.6.0
|
5 hónapja |
|
Tri Dao
|
320fb59487
Update citation
|
6 hónapja |
|
Tri Dao
|
2406f28805
Enable headdim 256 backward on consumer GPUs (Ampere, Ada)
|
9 hónapja |
 Tao He
|
204c3c6d1b
Fixes an error in comment (#785)
|
10 hónapja |
|
Tri Dao
|
54e80a3829
Implement page KV cache
|
10 hónapja |
 Erich Schubert
|
99ea4baa1d
Typo in README (#760)
|
11 hónapja |
|
Tri Dao
|
732654583c
Implement deterministic backward (thanks to Meituan)
|
11 hónapja |
|
Tri Dao
|
50d144c906
Mention Alibi in README
|
11 hónapja |
|
Tri Dao
|
7f31e7c16a
Bump to v2.3.2
|
1 éve |
|
Tri Dao
|
5a83425442
Change constexpr int to constexpr static int
|
1 éve |
|
Tri Dao
|
3a9fe7b0fa
Add change log
|
1 éve |
|
Tri Dao
|
aa4fd2d166
Clarify that Windows is not supported right now
|
1 éve |
|
Tri Dao
|
0c04943fa2
Require CUDA 11.6+, clean up setup.py
|
1 éve |
 Jeffrey Quesnelle
|
1d817a8ffc
fix citation in README (#501)
|
1 éve |
|
Tri Dao
|
45ba93cd96
Add newlines to README
|
1 éve |
|
Tri Dao
|
9e5e8bc91e
Change causal mask to be aligned to bottom-right instead of top-left
|
1 éve |
|
Tri Dao
|
d30f2e1cd5
Bump to v2.0.4
|
1 éve |
 Ian Timmis
|
cbf982afa5
README syntax highlighting (#365)
|
1 éve |
|
Tri Dao
|
d1a3b52f17
Add instruction about limiting number of ninja jobs
|
1 éve |
|
Tri Dao
|
b4cc152e97
Make sure dout is contiguous
|
1 éve |
|
Tri Dao
|
4f285b3547
FlashAttention-2 release
|
1 éve |
|
Tri Dao
|
ce68305c84
Update installation instruction
|
1 éve |
|
Tri Dao
|
f0c40b7ddb
Recommend Nvidia's Pytorch container
|
1 éve |
|
Tri Dao
|
40a25c8ee7
Update roadmap
|
1 éve |
 Anthony Hu
|
d63cfc3551
Use pyproject.toml to specify build dependencies
|
1 éve |
|
Tri Dao
|
74af023316
Bump version to 1.0.0
|
1 éve |
|
Tri Dao
|
1b18f1b7a1
Support H100
|
1 éve |
|
Tri Dao
|
f28d61cb2a
Update README on requirements (nvcc and Pytorch)
|
1 éve |
 Tri Dao
|
57ee618170
Merge pull request #94 from calebthomas259/main
|
1 éve |
