[rocm-libraries] ROCm/rocm-libraries#6526 (commit 3e01710)
feat: [CK Tile] mxfp8 support for qr async pipeline MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Closes: #6526 ## Motivation Adds MXFP8 support to the QR async FMHA forward pipeline by enabling MX quant scale paths, loosening padding constraints, and updating tests/codegen to cover the new combinations. ## Technical Details Changes: - Extend BlockFmhaPipelineQRKSVSAsync to support MX quant scales (Q/K/V/P scale tiles) and adjust padding/alignment behavior accordingly. - Update FMHA fwd tests to include head-dim adjustment helpers and add a new “General” parameterized test suite. - Update codegen pipeline constraints/selection for qr_async, and adjust tile-window vectorization logic for packed types. ## Test Plan Running CI with extended set of tests ## Test Result CI passing ## Submission Checklist - [ ] Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.
Z
Zoltán Lakatos committed
cb859854a7b9f511bc043b4390f722a9457c9dfc
Parent: 4975bd0
Committed by assistant-librarian[bot] <assistant-librarian[bot]@users.noreply.github.com>
on 7/13/2026, 10:44:45 AM