[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

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

1

third_party/aiter vendored Submodule

Submodule third_party/aiter added at 01aae101b9

2

third_party/composable_kernel vendored

Submodule third_party/composable_kernel updated: 8086bbe3a7...434d19f696