 Tri Dao
|
ae3c1fb3e0
Simplify bwd by setting NumdQWarpGroups = NumMmaWarpGroups
|
il y a 4 semaines |
|
Tri Dao
|
2c996ca25f
Use SeqlenInfo for bwd and epilogue
|
il y a 1 mois |
|
Tri Dao
|
3b6ac2b954
Use compile time constants in local mask
|
il y a 1 mois |
|
Tri Dao
|
bfbaafd043
Fix bwd reading out of out LSE
|
il y a 1 mois |
|
Tri Dao
|
29cdfedd80
Use Bulk reduce instead of TMA for dQaccum, split across WGs
|
il y a 1 mois |
|
Tri Dao
|
e8a1edbeb2
Clean up some #include
|
il y a 1 mois |
|
Tri Dao
|
3ed79742fb
Add option to shuffle LSE and dPsum in the bwd
|
il y a 1 mois |
|
Tri Dao
|
82dc825759
Don't use the unsafe convert_type function
|
il y a 1 mois |
|
Tri Dao
|
df96486c31
Decode: varlen, paged KV, leftpad
|
il y a 1 mois |
|
Tri Dao
|
ea7a98f15d
Fix backward with softcap
|
il y a 2 mois |
|
Tri Dao
|
6e8b25e426
Refactor
|
il y a 3 mois |
 Ying Zhang
|
1c9717d699
address comments
|
il y a 3 mois |
|
Ying Zhang
|
dff976a84a
fixes
|
il y a 4 mois |
|
Ying Zhang
|
7b4e68e04f
hopper local attention
|
il y a 4 mois |
|
Ying Zhang
|
db80387343
Add seqused_q in fwd / bwd and seqused_k in bwd.
|
il y a 4 mois |
|
Tri Dao
|
bafe253042
[FA3] Bwd
|
il y a 5 mois |
