composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 03:07:02 +00:00

Files

Aviral Goel 93e0d79103 [rocm-libraries] ROCm/rocm-libraries#8035 (commit 45186b8)

[CK_Tile] Add wmma_bf16f32_16x16x32_bf16 warp-gemm test
 (#8035)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

## Summary

Adds the warp-gemm unit test for `wmma_bf16f32_16x16x32_bf16`. **Stacked
on #8028** (the API change) and based on its branch, so #8028 shows the
isolated API diff and this PR shows just the test.

## Test

gfx125-guarded `WmmaBf16f32.ResidualPrecisionContrast`: computes `Y_bf16
= X_bf16·W_bf16 + R_fp32` via `WarpGemm::mac_downconvert`, compares
against an fp32 reference (within bf16 tolerance), and asserts it is at
least as accurate as the bf16-accumulate path — i.e. it demonstrates the
precision benefit of the fp32 accumulator (`C`) carried into the fused
bf16 down-convert.

Passes on gfx1250.

2026-06-09 15:52:45 +00:00

CMakeLists.txt

[rocm-libraries] ROCm/rocm-libraries#8035 (commit 45186b8)

2026-06-09 15:52:45 +00:00

test_f32_16x16x128_fp4.cpp

[rocm-libraries] ROCm/rocm-libraries#7677 (commit 308af93)

2026-05-30 01:28:48 +00:00

test_f32_16x16x128_fp8_scale16.cpp

[rocm-libraries] ROCm/rocm-libraries#7725 (commit eef7e12)

2026-06-03 22:05:29 +00:00

test_f32_32x16x128_fp4.cpp

[rocm-libraries] ROCm/rocm-libraries#7677 (commit 308af93)