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