diff --git a/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_32x64_pf3.co b/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_32x64_pf3.co index d0d0d7c53ac..273e574e417 100755 Binary files a/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_32x64_pf3.co and b/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_32x64_pf3.co differ diff --git a/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_48x64_pf3.co b/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_48x64_pf3.co index 4d707f91966..cedc1e2455a 100755 Binary files a/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_48x64_pf3.co and b/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_48x64_pf3.co differ diff --git a/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_64x64_pf3.co b/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_64x64_pf3.co index f8a32f8131a..ee41f2e493e 100755 Binary files a/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_64x64_pf3.co and b/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_64x64_pf3.co differ diff --git a/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_96x64_pf3.co b/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_96x64_pf3.co index fa04213ceb7..539490ca4d7 100755 Binary files a/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_96x64_pf3.co and b/hsa/gfx942/bf16gemm/bf16gemm_outf32_tn_96x64_pf3.co differ diff --git a/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_32x64_pf3.co b/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_32x64_pf3.co index 03a90dc538e..e8b66a682a0 100755 Binary files a/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_32x64_pf3.co and b/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_32x64_pf3.co differ diff --git a/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_48x64_pf3.co b/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_48x64_pf3.co index ed67334c7bf..e80adfdd5f0 100755 Binary files a/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_48x64_pf3.co and b/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_48x64_pf3.co differ diff --git a/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_64x64_pf3.co b/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_64x64_pf3.co index fac6c76fcfc..8920454b677 100755 Binary files a/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_64x64_pf3.co and b/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_64x64_pf3.co differ diff --git a/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_96x64_pf3.co b/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_96x64_pf3.co index 9da1a628135..2d1be397f6e 100755 Binary files a/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_96x64_pf3.co and b/hsa/gfx950/bf16gemm/bf16gemm_outf32_tn_96x64_pf3.co differ diff --git a/op_tests/test_gemm_a16w16.py b/op_tests/test_gemm_a16w16.py index 6f9dde21ab7..ed0d44a341d 100755 --- a/op_tests/test_gemm_a16w16.py +++ b/op_tests/test_gemm_a16w16.py @@ -143,11 +143,11 @@ def test_gemm(dtype, m, n, k, bias=False, otype=None, scaleA=None, scaleB=None): dtype == dtypes.bf16 and otype == dtypes.fp32 and (k % 64 == 0) - # and (n % 64 == 0) - and (m in [64, 80, 128, 150, 192, 220, 256, 384, 448, 512]) - and (n == 256) - and (k == 5120 or k == 7168) - and bias == None + and (n % 64 == 0) # N % tileN == 0 + # and (m in [64, 80, 128, 150, 192, 220, 256, 384, 448, 512]) + # and (n == 256) + # and (k == 5120 or k == 7168) + and bias is None ): # wshuffle = shuffle_weight(weight, layout=(16, 16)) # out_asm = torch.empty((m + 191) // 192 * 192, n, dtype=otype)