Skip to content

[Test] Add gfx1250 WMMA lowering tests for additional dtypes#891

Open
AiyyappanMR wants to merge 1 commit into
ROCm:mainfrom
AiyyappanMR:add-wmma-tests-gfx1250
Open

[Test] Add gfx1250 WMMA lowering tests for additional dtypes#891
AiyyappanMR wants to merge 1 commit into
ROCm:mainfrom
AiyyappanMR:add-wmma-tests-gfx1250

Conversation

@AiyyappanMR

Copy link
Copy Markdown

Motivation

The test file only covered the int4 (iu4) WMMA config. The other supported MMA shapes/dtypes for gfx1250 (f16, bf16, f32, i8, fp8) had no MLIR test coverage for their fly.mma_atom_call -> rocdl.wmma.* lowering.

Changes

  • Added CHECK-LABEL test cases for:
    • rocdl.wmma.f32.16x16x32.f16
    • rocdl.wmma.f32.16x16x32.bf16
    • rocdl.wmma.f32.16x16x4.f32
    • rocdl.wmma.i32.16x16x64.iu8
    • rocdl.wmma.f32.16x16x64.fp8_fp8

Testing

  • Lit tests added (FileCheck-based)
  • Performance benchmarks run (N/A, test-only change)

Dependencies

  • No new third-party dependencies added

Breaking Changes

None — test-only addition.

Signed-off-by: AiyyappanMR <aiyyappan2004@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant