SIGN IN SIGN UP

[rocm-libraries] ROCm/rocm-libraries#7016 (commit 2b73c00)

[CK] Fix RDNA3 FMHA tile-load paths (#7016)

## Summary

Fix CK tile FMHA paths needed for RDNA3/RDNA4 targets.

## Details

This PR addresses RDNA-specific issues hit while enabling xFormers CK
FMHA on gfx11/gfx12:

- On RDNA3, update FMHA P tile handling so the layout consumed by the
second GEMM matches the WMMA path.

## Testing

Validated downstream with xFormers CK/FMHA on gfx1201/gfx1151.

```text
pytest --import-mode=importlib -q \
  tests/test_mem_eff_attention.py::test_forward \
  tests/test_mem_eff_attention.py::test_backward \
  tests/test_mem_eff_attention.py::test_dropout_ck

3844 passed, 5244 skipped, 26 warnings

---------

Co-authored-by: Po Yen Chen <PoYen.Chen@amd.com>
A
Aaryaman Vasishta committed
fad83d9c907bdfe966d5f6de8dae1756b24439fe
Parent: 5169cd1
Committed by GitHub <noreply@github.com> on 5/19/2026, 1:41:36 PM