Function dynamicOutput3DKernelMapShrinks

test/features/dynamic-output.js:737–796

(mode)

function dynamicOutput3DKernelMapShrinks(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.z + this.thread.x + this.thread.y + this.thread.z);
  }, { dynamicOutput: true });

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