Function testSinglePrecisionMatrix

test/features/dynamic-arguments.js:1096–1128

(mode)

});

function testSinglePrecisionMatrix(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 = [
    [1,2,3,4],
    [5,6,7,8],
    [9,10,11,12],
    [13,14,15,16]
  ];
  assert.deepEqual(kernel(matrix), matrix.map(row => new Float32Array(row)));

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

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

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