Function dynamicOutput3DShrinks

test/features/dynamic-output.js:590–625

(mode)

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

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

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

  gpu.destroy();
}

test('dynamic output 3d shrinks auto', () => {
  dynamicOutput3DShrinks();