Tri Dao
|
2c996ca25f
Use SeqlenInfo for bwd and epilogue
|
před 1 týdnem |
Tri Dao
|
82dc825759
Don't use the unsafe convert_type function
|
před 2 týdny |
Tri Dao
|
e7b93e3902
Clean up mha_combine kernel
|
před 3 týdny |
Tri Dao
|
3d0d147940
Early stop on actual num_splits in mha_combine kernel
|
před 3 týdny |
Tri Dao
|
9fd6b977bb
Precompute the pointers in mha_combine kernel
|
před 3 týdny |
Tri Dao
|
01de45b8f8
Add combine kernel
|
před 1 měsícem |