1 files changed, 61 insertions, 0 deletions
diff --git a/testing/web-platform/tests/webnn/conformance_tests/compute-arraybufferview-with-bigger-arraybuffer.https.any.js b/testing/web-platform/tests/webnn/conformance_tests/compute-arraybufferview-with-bigger-arraybuffer.https.any.js
new file mode 100644
index 0000000000..62ce16c93e
--- /dev/null
+++ b/testing/web-platform/tests/webnn/conformance_tests/compute-arraybufferview-with-bigger-arraybuffer.https.any.js
@@ -0,0 +1,61 @@
+// META: title=test WebNN MLContext.compute() for ArrayBufferView created from bigger ArrayBuffer
+// META: global=window,dedicatedworker
+// META: variant=?gpu
+// META: script=../resources/utils.js
+
+'use strict';
+
+// These tests are used to reproduce the Chromium issue:
+// https://issues.chromium.org/issues/332151809
+
+if (navigator.ml) {
+  const variant = location.search.substring(1);
+  const contextOptions = kContextOptionsForVariant[variant];
+
+  let context;
+  let builder;
+
+  promise_setup(async () => {
+    let supported = false;
+    try {
+      context = await navigator.ml.createContext(contextOptions);
+      supported = true;
+    } catch (e) {
+    }
+    assert_implements(
+        supported, `Unable to create context for ${variant} variant`);
+    builder = new MLGraphBuilder(context);
+  });
+
+  promise_test(async t => {
+    const a = builder.input('a', {dataType: 'float32', dimensions: [2]});
+    const b = builder.relu(a);
+    const graph = await builder.build({b});
+    const arraybuffer = new ArrayBuffer(100);
+    const aBuffer =
+        new Float32Array(arraybuffer, /*byteOffset*/ 4, /*length*/ 2)
+    aBuffer.set([1, -1]);
+    const bBuffer = new Float32Array(2);
+    const results =
+        await context.compute(graph, {'a': aBuffer}, {'b': bBuffer});
+    assert_array_approx_equals_ulp(
+        results.outputs.b, [1, 0], /*nulp*/ 0, 'float32');
+  }, 'Test compute() working for input ArrayBufferView created from bigger ArrayBuffer');
+
+  promise_test(async t => {
+    const a = builder.input('a', {dataType: 'float32', dimensions: [2]});
+    const b = builder.relu(a);
+    const graph = await builder.build({b});
+    const aBuffer = new Float32Array(2);
+    aBuffer.set([1, -1]);
+    const arraybuffer = new ArrayBuffer(100);
+    const bBuffer =
+        new Float32Array(arraybuffer, /*byteOffset*/ 8, /*length*/ 2);
+    const results =
+        await context.compute(graph, {'a': aBuffer}, {'b': bBuffer});
+    assert_array_approx_equals_ulp(
+        results.outputs.b, [1, 0], /*nulp*/ 0, 'float32');
+  }, 'Test compute() working for output ArrayBufferView created from bigger ArrayBuffer');
+} else {
+  test(() => assert_implements(navigator.ml, 'missing navigator.ml'));
+}