SIGN IN SIGN UP

[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