Skip to content

Commit 017a7c0

Browse files
xhcaojavier-intel
authored andcommitted
[webgpu] add cases to cover the subgroup matrix path (microsoft#24970)
### Description <!-- Describe your changes. --> ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. -->
1 parent f1771dc commit 017a7c0

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

onnxruntime/test/contrib_ops/matmul_4bits_test.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -446,6 +446,8 @@ TEST(MatMulNBits, Float16_Accuracy4) {
446446
TestMatMulNBitsTyped<MLFloat16, 100, 32, 16, 16, 4>();
447447
TestMatMulNBitsTyped<MLFloat16, 100, 32, 32, 16, 4>();
448448
TestMatMulNBitsTyped<MLFloat16, 100, 32, 16, 128, 4>();
449+
TestMatMulNBitsTyped<MLFloat16, 100, 64, 32, 32, 4>();
450+
TestMatMulNBitsTyped<MLFloat16, 100, 128, 128, 32, 4>();
449451
TestMatMulNBitsTyped<MLFloat16, 100, 288, 16, 16, 4>();
450452
TestMatMulNBitsTyped<MLFloat16, 100, 288, 1024, 16, 4>();
451453
TestMatMulNBitsTyped<MLFloat16, 100, 288, 1024, 128, 4>();

onnxruntime/test/contrib_ops/matmul_8bits_test.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -289,6 +289,8 @@ TEST(MatMulNBits, Float16_8b_AccuracyLevel4) {
289289
constexpr float abs_error = 0.055f;
290290
constexpr float rel_error = 0.02f;
291291
TestMatMul8BitsTyped<MLFloat16, 2, 4, 32, 16, 4>(abs_error, rel_error);
292+
TestMatMul8BitsTyped<MLFloat16, 100, 64, 32, 32, 4>(abs_error, rel_error);
293+
TestMatMul8BitsTyped<MLFloat16, 100, 128, 128, 32, 4>(abs_error, rel_error);
292294
TestMatMul8BitsTyped<MLFloat16, 199, 40, 576, 32, 4>(abs_error, rel_error);
293295
}
294296
#endif

0 commit comments

Comments
 (0)