 Tri Dao
|
29cdfedd80
Use Bulk reduce instead of TMA for dQaccum, split across WGs
|
il y a 1 mois |
|
Tri Dao
|
c5ba47b3d5
Add fence.async to epilogue
|
il y a 1 mois |
|
Tri Dao
|
9f82a326ad
Implement rotary for attn decode
|
il y a 1 mois |
|
Tri Dao
|
4d00645c76
Implement appending new KV to KV cache
|
il y a 1 mois |
|
Tri Dao
|
df96486c31
Decode: varlen, paged KV, leftpad
|
il y a 1 mois |
|
Tri Dao
|
6e8b25e426
Refactor
|
il y a 3 mois |
|
Tri Dao
|
bafe253042
[FA3] Bwd
|
il y a 5 mois |
 jayhshah
|
5018ac6ac5
Fp8 kernel with "in-kernel" transpose of V in producer (#1100)
|
il y a 5 mois |
|
Tri Dao
|
74b0761ff7
[FA3] BF16 forward
|
il y a 5 mois |
