frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ROCm] Initial AITER Integration for mha_bwd asm kernels (#152630)

Browse Source 
Generates AITER plumbing via cmake. Calls into fav3 asm bwd CK kernels.

Update submodule composable kernel for this change

Pull Request resolved: https://github.com/pytorch/pytorch/pull/152630
Approved by: https://github.com/xw285cornell, https://github.com/yoyoyocmu
This commit is contained in:
Andy Lugo
2025-07-01 02:53:22 +00:00
committed by PyTorch MergeBot
parent f40efde2a4
commit b5ce77c1f5
8 changed files with 80 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.gitmodules vendored
View File

@ -129,3 +129,6 @@
[submodule "third_party/flash-attention"]
path = third_party/flash-attention
url = https://github.com/Dao-AILab/flash-attention.git
[submodule "third_party/aiter"]
path = third_party/aiter
url = https://github.com/ROCm/aiter.git