Tri Dao
|
2c996ca25f
Use SeqlenInfo for bwd and epilogue
|
hace 1 semana |
Tri Dao
|
82dc825759
Don't use the unsafe convert_type function
|
hace 2 semanas |
Tri Dao
|
e7b93e3902
Clean up mha_combine kernel
|
hace 3 semanas |
Tri Dao
|
3d0d147940
Early stop on actual num_splits in mha_combine kernel
|
hace 3 semanas |
Tri Dao
|
9fd6b977bb
Precompute the pointers in mha_combine kernel
|
hace 3 semanas |
Tri Dao
|
01de45b8f8
Add combine kernel
|
hace 1 mes |