"src/targets/vscode:/vscode.git/clone" did not exist on "a8965e517e22690cfcf070ff1f060bbb1d1bb685"
  • 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