Function dynamicOutput2DKernelMapShrinks

test/features/dynamic-output.js:349–376

(mode)

function dynamicOutput2DKernelMapShrinks(mode) {
  const gpu = new GPU({ mode });
  const kernel = gpu.createKernelMap({
    result1: function map(v) {
      return v;
    }
  }, function() {
    return map(this.output.x + this.output.y + this.thread.x + this.thread.y);
  }, { dynamicOutput: true });

  kernel.setOutput([3,3]);
  let result = kernel();
  assert.equal(result.result.length, 3);
  assert.equal(result.result1.length, 3);
  assert.deepEqual(result.result.map(row => Array.from(row)), [[6,7,8],[7,8,9],[8,9,10]]);
  assert.deepEqual(result.result1.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.result.length, 2);
  assert.equal(result.result1.length, 2);
  assert.deepEqual(result.result.map(row => Array.from(row)), [[4,5],[5,6]]);
  assert.deepEqual(result.result1.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();