Commit History

作者 SHA1 備註 提交日期
  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 月之前