 Antoni Viros
|
83e41b3ca4
Add custom ops for compatibility with PT Compile (#1139)
|
il y a 3 mois |
 Tri Dao
|
65f723bb9a
Split bwd into more .cu files to speed up compilation
|
il y a 5 mois |
|
Tri Dao
|
751c762c9c
Don't specialize for hdim 224 to speed up compilation
|
il y a 5 mois |
 Driss Guessous
|
1c275eb070
Fix ima for split-kv kernel (#1085)
|
il y a 5 mois |
 Jorge António
|
4df62e1440
catch typo (#1058)
|
il y a 5 mois |
|
Tri Dao
|
40e534a7f6
Implement cache_leftpad
|
il y a 6 mois |
|
Tri Dao
|
dca6d89da4
Don't support softcap and dropout at the same time
|
il y a 6 mois |
|
Tri Dao
|
908511b2b6
Split into more .cu files to speed up compilation
|
il y a 6 mois |
|
Tri Dao
|
1d536d7de5
Minor cleanup of softcapping
|
il y a 6 mois |
 Nicolas Patry
|
8f873cc6ac
Implement softcapping. (#1025)
|
il y a 6 mois |
|
Nicolas Patry
|
5bf201966a
Fixing argument checking when using `seqlenq_ngroups_swapped`. (#976)
|
il y a 6 mois |
 Grigory Sizov
|
f816dee63c
Support unpadded LSE layout (#970)
|
il y a 6 mois |
|
Tri Dao
|
9eb3d099c1
Transpose out when swapping seqlen_q and num_groups
|
il y a 9 mois |
|
Driss Guessous
|
4a73e903da
Add in, macrosf for defining __grid_constant__ (#852)
|
il y a 10 mois |
|
Grigory Sizov
|
2a15840f09
Enable paged attention in varlen forward (#831)
|
il y a 10 mois |
|
Tri Dao
|
2406f28805
Enable headdim 256 backward on consumer GPUs (Ampere, Ada)
|
il y a 10 mois |
|
Tri Dao
|
d9a5cb291c
Fix dv = torch::empty_like(k) for mha_bwd_varlen as well
|
il y a 11 mois |
 Brian Hirsh
|
2423cca3ad
fix backward for when query and key have different contiguity (#818)
|
il y a 11 mois |
|
Grigory Sizov
|
4687936413
Fix Windows build (#816)
|
il y a 11 mois |
 Jeremy Reizenstein
|
0658e320f6
Preprocessor switches to control functionality (#788)
|
il y a 11 mois |
|
Tri Dao
|
54e80a3829
Implement page KV cache
|
il y a 11 mois |
|
Tri Dao
|
ea8a25ca38
Remove configure in bwd kernel launch
|
il y a 11 mois |
 Grigory Sizov
|
af01244ddd
Add split-kv and M<->H swap to varlen forward decoding attention (#754)
|
il y a 11 mois |
|
Tri Dao
|
0842ec0da4
Don't dispatch to local if window size >= seqlen_k
|
il y a 1 an |
|
Tri Dao
|
732654583c
Implement deterministic backward (thanks to Meituan)
|
il y a 1 an |
|
Tri Dao
|
5ab9b3667b
Clean up alibi, implement non-causal alibi
|
il y a 1 an |
 Sanghun Cho
|
e4f726fc44
Support alibi, by Sanghun Cho from Kakao Brain
|
il y a 1 an |
|
Jeremy Reizenstein
|
ce3e7280f8
Allow varlen_fwd to take optional seqused_k (#647)
|
il y a 1 an |
|
Tri Dao
|
db2f80692c
Write zero to out / grad if seqlen_q or seqlen_k is zero
|
il y a 1 an |
|
Tri Dao
|
e279bf8ed9
[Gen] Accept cache_batch_idx to index into the KV cache
|
il y a 1 an |
