[rocm-libraries] ROCm/rocm-libraries#8799 (commit f36ce12)
fix(ck-tile): add missing fp8/bf8 warp_gemm dispatcher entries for gfx950 (#8799) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit JIRA ID : AICK-1534 ## Motivation Enable asymmetric A/B access width (e.g. `EDouble` for A, `ESingle` for B) for fp8/bf8/f16/bf16/i8 warp gemms on gfx950, and fix related dispatcher bugs. ## Technical Details - Split `AttrNumAccess` into `AttrNumAccessA`/`AttrNumAccessB` for fp8/bf8/f16/bf16 warp gemm aliases (default `B = A`, backward compatible). - Added `__gfx950__`-guarded asymmetric dispatcher entries for fp8/bf8 at 16×16×64 and 32×32×32. - Fixed missing `bf8×bf8` 16×16×64 `EDouble` dispatcher on non-gfx950/non-gfx125 platforms. - Fixed `WarpGemmMfmaBf16Bf16F32M16N16K32` silently dropping `AttrNumAccessB` on non-gfx950. - Fixed duplicate definition of `WarpGemmMfma_i32_16x16x64_i8_i8_CTransposed`. - Fixed `IsScale16=true` dispatcher incorrectly routing to non-scale16 variant. ## Test Plan Build and run existing CK Tile GEMM unit tests for fp8/bf8/f16/bf16/i8 types.
T
Thrupti Raj Lakshmana Gowda committed
6c02d45356e76a1eddd2910e97089f84c5ac279e
Parent: f4e6fad
Committed by assistant-librarian[bot] <assistant-librarian[bot]@users.noreply.github.com>
on 7/1/2026, 4:15:36 PM