[webnn] Add float32 tests for WebNN API where op (#43558)

web-platform-tests · Dec 13, 2023 · ba654bf · ba654bf
1 parent b597a15
commit ba654bf
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 0 deletions.
diff --git a/webnn/gpu/where.https.any.js b/webnn/gpu/where.https.any.js
@@ -0,0 +1,10 @@
+// META: title=test WebNN API where operation
+// META: global=window,dedicatedworker
+// META: script=../resources/utils.js
+// META: timeout=long
+
+'use strict';
+
+// https://webmachinelearning.github.io/webnn/#api-mlgraphbuilder-where
+
+testWebNNOperation('where', buildWhere, 'gpu');
diff --git a/webnn/resources/utils.js b/webnn/resources/utils.js
@@ -354,6 +354,7 @@ const PrecisionMetrics = {
   squeeze: {ULP: {float32: 0, float16: 0}},
   tanh: {ATOL: {float32: 1/1024, float16: 1/512}},
   transpose: {ULP: {float32: 0, float16: 0}},
+  where: {ULP: {float32: 0, float16: 0}},
 };
 
 /**
@@ -694,6 +695,15 @@ const buildSplit = (operationName, builder, resources) => {
   return namedOutputOperand;
 };
 
+const buildWhere = (operationName, builder, resources) => {
+  // MLOperand where(MLOperand condition, MLOperand trueValues, MLOperand falseValues);
+  const namedOutputOperand = {};
+  const [conditionOperand, trueValuesOperand, falseValuesOperand] = createMultiInputOperands(builder, resources);
+  // invoke builder.where()
+  namedOutputOperand[resources.expected.name] = builder[operationName](conditionOperand, trueValuesOperand, falseValuesOperand);
+  return namedOutputOperand;
+};
+
 /**
  * Build a graph.
  * @param {String} operationName - An operation name

diff --git a/webnn/where.https.any.js b/webnn/where.https.any.js
@@ -0,0 +1,10 @@
+// META: title=test WebNN API where operation
+// META: global=window,dedicatedworker
+// META: script=./resources/utils.js
+// META: timeout=long
+
+'use strict';
+
+// https://webmachinelearning.github.io/webnn/#api-mlgraphbuilder-where
+
+testWebNNOperation('where', buildWhere);