mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-07-07 15:47:05 +00:00
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.