 Tri Dao
|
659a631f4c
Rename bwd classes to include Sm90 suffix
|
5 giorni fa |
|
Tri Dao
|
0eb8f680a0
Fix env var to disable hdims
|
2 settimane fa |
|
Tri Dao
|
29cdfedd80
Use Bulk reduce instead of TMA for dQaccum, split across WGs
|
1 mese fa |
|
Tri Dao
|
df96486c31
Decode: varlen, paged KV, leftpad
|
1 mese fa |
|
Tri Dao
|
6e8b25e426
Refactor
|
3 mesi fa |
 Ying Zhang
|
1c9717d699
address comments
|
3 mesi fa |
|
Ying Zhang
|
be6c1b98c4
small fixes
|
3 mesi fa |
|
Ying Zhang
|
dff976a84a
fixes
|
4 mesi fa |
|
Ying Zhang
|
7b4e68e04f
hopper local attention
|
4 mesi fa |
|
Tri Dao
|
3f6ff1c1c5
Remove struct : cute::aligned_struct to avoid error with gcc 12
|
5 mesi fa |
|
Tri Dao
|
bafe253042
[FA3] Bwd
|
5 mesi fa |
|
Tri Dao
|
7f67966cc7
FA3 initial code release
|
5 mesi fa |
