Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Yu Cheng's avatar
    [Dev] Add mha backward example (#77) · a6fe61e2
    Yu Cheng authored 
    * [CI][Test] Add test cases for tilelang transform MultiVersionBuffer and WarpSpecialized

* Relax the mismatch ratio restrictions in the flash_linear_attention and mha tests

* [Dev] Add mha backward example
    a6fe61e2
example_mha_bwd.py 14.5 KB