Aaryaman Vasishta 457f153b69 [rocm-libraries] ROCm/rocm-libraries#7016 (commit 2b73c00) ...

[CK] Fix RDNA3 FMHA tile-load paths

## 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

2026-05-19 13:42:43 +00:00

..

block

[rocm-libraries] ROCm/rocm-libraries#7256 (commit 1fc20eb)

2026-05-13 09:42:28 +00:00

kernel

[rocm-libraries] ROCm/rocm-libraries#7331 (commit 5692db0)

2026-05-14 13:34:32 +00:00

pipeline

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

2026-05-19 13:42:43 +00:00