1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
|
// META: title=ensure MLGraphBuilder.constant() handles buffers which change
// META: global=window,worker
// META: variant=?cpu
// META: variant=?gpu
// META: variant=?npu
// META: script=../resources/utils_validation.js
promise_test(async t => {
const builder = new MLGraphBuilder(context);
let backingBuffer = new ArrayBuffer(8);
let aBuffer = new Float32Array(backingBuffer, 0, 2);
aBuffer[0] = 2;
aBuffer[1] = 3;
const a = builder.constant({dataType: 'float32', shape: [2]}, aBuffer);
// Detach `aBuffer`. Constant data should already be copied, so changes to
// the buffer afterwards should not be reflected in the graph.
const unusedBuffer = backingBuffer.transfer();
const b = builder.input('b', {dataType: 'float32', shape: [2]});
const c = builder.add(a, b);
const [graph, bTensor, cTensor] = await Promise.all([
builder.build({c}),
context.createTensor({dataType: 'float32', shape: [2], writable: true}),
context.createTensor({dataType: 'float32', shape: [2], readable: true})
]);
context.writeTensor(bTensor, new Float32Array([5, 7]));
context.dispatch(graph, {'b': bTensor}, {'c': cTensor});
const result = new Float32Array(await context.readTensor(cTensor));
assert_array_equals(result, new Float32Array([7, 10]));
}, 'Constant data is unaffected by detaching the buffer');
promise_test(async t => {
const builder = new MLGraphBuilder(context);
let aBuffer = new Float32Array([2, 3]);
const a = builder.constant({dataType: 'float32', shape: [2]}, aBuffer);
// Rewrite `aBuffer` contents. Constant data should already be copied, so
// changes to the buffer afterwards should not be reflected in the graph.
aBuffer[0] = 10;
aBuffer[1] = 20;
const b = builder.input('b', {dataType: 'float32', shape: [2]});
const c = builder.add(a, b);
const [graph, bTensor, cTensor] = await Promise.all([
builder.build({c}),
context.createTensor({dataType: 'float32', shape: [2], writable: true}),
context.createTensor({dataType: 'float32', shape: [2], readable: true})
]);
context.writeTensor(bTensor, new Float32Array([5, 7]));
context.dispatch(graph, {'b': bTensor}, {'c': cTensor});
const result = new Float32Array(await context.readTensor(cTensor));
assert_array_equals(result, new Float32Array([7, 10]));
}, 'Constant data is unaffected by changes to the buffer contents');
promise_test(async t => {
const builder = new MLGraphBuilder(context);
let backingBuffer = new ArrayBuffer(8);
const aBuffer = new Float32Array(backingBuffer, 0, 2);
// Detach `aBuffer` _before_ calling `constant()`. This should throw, since
// detached buffers have a length of zero, which does not match the length of
// the descriptor. See
// https://webidl.spec.whatwg.org/#dfn-get-buffer-source-copy
const unusedBuffer = backingBuffer.transfer();
assert_throws_js(
TypeError,
() => builder.constant({dataType: 'float32', shape: [2]}, aBuffer));
}, 'Constant data cannot use a detached buffer, which is empty');
|