Function dynamicOutput2DShrinks

test/features/dynamic-output.js:250–269

(mode)

function dynamicOutput2DShrinks(mode) {
  const gpu = new GPU({ mode });
  const kernel = gpu.createKernel(function() {
    return this.output.x + this.output.y + this.thread.x + this.thread.y;
  }, { dynamicOutput: true });

  kernel.setOutput([3,3]);
  let result = kernel();
  assert.equal(result.length, 3);
  assert.deepEqual(result.map(row => Array.from(row)), [[6,7,8],[7,8,9],[8,9,10]]);
  assert.deepEqual(Array.from(kernel.output), [3,3]);

  kernel.setOutput([2,2]);
  result = kernel();
  assert.equal(result.length, 2);
  assert.deepEqual(result.map(row => Array.from(row)), [[4,5],[5,6]]);
  assert.deepEqual(Array.from(kernel.output), [2,2]);

  gpu.destroy();
}

test('dynamic output 2d shrinks auto', () => {
  dynamicOutput2DShrinks();