webnn: Enforce input data type constraints for gemm and matmul

shiyi9801 · chromium-wpt-export-bot · commit 03cd3c5b425d · 2024-04-27T22:56:35.000-07:00
As specified in webmachinelearning/webnn#646 Bug: 328567884 Change-Id: Ia55a214e7ad281ec3c8911e9116f388fac209d05
diff --git a/webnn/validation_tests/gemm.https.any.js b/webnn/validation_tests/gemm.https.any.js
@@ -34,7 +34,7 @@ const tests = [
     b: {dataType: 'float32', dimensions: [2, 4]},
   },
   {
-    name: 'Test building gemm with aTranspose=true.',
+    name: '[gemm] Test building gemm with aTranspose=true.',
     a: {dataType: 'float32', dimensions: [2, 3]},
     b: {dataType: 'float32', dimensions: [2, 4]},
     options: {
@@ -44,15 +44,15 @@ const tests = [
   },
   {
     name:
-        'Throw if inputShapeA[0] is not equal to inputShapeB[0] with aTranspose=true.',
+        '[gemm] Throw if inputShapeA[0] is not equal to inputShapeB[0] with aTranspose=true.',
     a: {dataType: 'float32', dimensions: [2, 3]},
     b: {dataType: 'float32', dimensions: [3, 4]},
     options: {
       aTranspose: true,
     },
   },
   {
-    name: 'Test building gemm with bTranspose=true.',
+    name: '[gemm] Test building gemm with bTranspose=true.',
     a: {dataType: 'float32', dimensions: [2, 3]},
     b: {dataType: 'float32', dimensions: [4, 3]},
     options: {
@@ -62,30 +62,30 @@ const tests = [
   },
   {
     name:
-        'Throw if inputShapeA[0] is not equal to inputShapeB[0] with bTranspose=true.',
+        '[gemm] Throw if inputShapeA[0] is not equal to inputShapeB[0] with bTranspose=true.',
     a: {dataType: 'float32', dimensions: [2, 3]},
     b: {dataType: 'float32', dimensions: [3, 4]},
     options: {
       bTranspose: true,
     },
   },
   {
-    name: 'Throw if the rank of inputA is not 2.',
+    name: '[gemm] Throw if the rank of inputA is not 2.',
     a: {dataType: 'float32', dimensions: [2, 3, 1]},
     b: {dataType: 'float32', dimensions: [2, 4]},
   },
   {
-    name: 'Throw if the rank of inputB is not 2.',
+    name: '[gemm] Throw if the rank of inputB is not 2.',
     a: {dataType: 'float32', dimensions: [2, 4]},
     b: {dataType: 'float32', dimensions: [2, 3, 1]},
   },
   {
-    name: 'Throw if data types of two inputs do not match.',
+    name: '[gemm] Throw if data types of two inputs do not match.',
     a: {dataType: 'float32', dimensions: [2, 3]},
     b: {dataType: 'int32', dimensions: [3, 4]},
   },
   {
-    name: 'Test building gemm with inputC.',
+    name: '[gemm] Test building gemm with inputC.',
     a: {dataType: 'float32', dimensions: [2, 3]},
     b: {dataType: 'float32', dimensions: [3, 4]},
     options: {
@@ -94,7 +94,7 @@ const tests = [
     output: {dataType: 'float32', dimensions: [2, 4]}
   },
   {
-    name: 'Test building gemm with scalar inputC.',
+    name: '[gemm] Test building gemm with scalar inputC.',
     a: {dataType: 'float32', dimensions: [2, 3]},
     b: {dataType: 'float32', dimensions: [3, 4]},
     options: {
@@ -104,16 +104,21 @@ const tests = [
   },
   {
     name:
-        'Throw if inputShapeC is not unidirectionally broadcastable to the output shape [inputShapeA[0], inputShapeB[1]].',
+        '[gemm] Throw if inputShapeC is not unidirectionally broadcastable to the output shape [inputShapeA[0], inputShapeB[1]].',
     a: {dataType: 'float32', dimensions: [2, 3]},
     b: {dataType: 'float32', dimensions: [3, 4]},
     options: {
       c: {dataType: 'float32', dimensions: [2, 3]},
     },
   },
+  {
+    name: '[gemm] Throw if the input data type is not floating point.',
+    a: {dataType: 'int32', dimensions: [2, 3]},
+    b: {dataType: 'int32', dimensions: [3, 4]}
+  },
   {
     name:
-        'Throw if data type of inputC does not match ones of inputA and inputB.',
+        '[gemm] Throw if data type of inputC does not match ones of inputA and inputB.',
     a: {dataType: 'float32', dimensions: [3, 2]},
     b: {dataType: 'float32', dimensions: [4, 3]},
     options: {
@@ -123,7 +128,7 @@ const tests = [
     },
   },
   {
-    name: 'Throw if the rank of inputC is 3.',
+    name: '[gemm] Throw if the rank of inputC is 3.',
     a: {dataType: 'float32', dimensions: [3, 2]},
     b: {dataType: 'float32', dimensions: [4, 3]},
     options: {
diff --git a/webnn/validation_tests/matmul.https.any.js b/webnn/validation_tests/matmul.https.any.js
@@ -67,6 +67,13 @@ const tests = [
     },
     output: {dataType: 'float32', dimensions: [2, 3, 5]}
   },
+  {
+    name: '[matmul] Throw if the input data type is not floating point',
+    inputs: {
+      a: {dataType: 'uint32', dimensions: [2, 3, 4]},
+      b: {dataType: 'uint32', dimensions: [2, 4, 5]}
+    }
+  },
   {
     name: '[matmul] Throw if data type of two inputs don\'t match',
     inputs: {
