Function testSinglePrecisionInputMatrix

test/features/dynamic-arguments.js:1225–1269

(mode)

});

function testSinglePrecisionInputMatrix(mode) {
  const gpu = new GPU({ mode });
  const kernel = gpu.createKernel(function(input) {
    return input[this.thread.y][this.thread.x];
  })
    .setDynamicArguments(true)
    .setDynamicOutput(true)
    .setOutput([4,4]);

  let matrix = input([
    1,2,3,4,
    5,6,7,8,
    9,10,11,12,
    13,14,15,16
  ], [4, 4]);
  assert.deepEqual(kernel(matrix), [
    new Float32Array([1,2,3,4]),
    new Float32Array([5,6,7,8]),
    new Float32Array([9,10,11,12]),
    new Float32Array([13,14,15,16])
  ]);

  kernel.setOutput([3,3]);
  matrix = input([
    1,2,3,
    4,5,6,
    7,8,9
  ], [3, 3]);
  assert.deepEqual(kernel(matrix), [
    new Float32Array([1,2,3]),
    new Float32Array([4,5,6]),
    new Float32Array([7,8,9])
  ]);

  kernel.setOutput([2,2]);
  matrix = input([
    1,2,
    3,4
  ], [2,2]);
  assert.deepEqual(kernel(matrix), [
    new Float32Array([1,2]),
    new Float32Array([3,4])
  ]);
  gpu.destroy();
}

(GPU.isSinglePrecisionSupported ? test : skip)('single precision Input Matrix auto', () => {
  testSinglePrecisionInputMatrix();