Function dynamicOutput1DKernelMapShrinks

test/features/dynamic-output.js:151–178

(mode)

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

  kernel.setOutput([10]);
  let result = kernel();
  assert.equal(result.result.length, 10);
  assert.equal(result.result2.length, 10);
  assert.deepEqual(Array.from(result.result), [10,11,12,13,14,15,16,17,18,19]);
  assert.deepEqual(Array.from(result.result2), [10,11,12,13,14,15,16,17,18,19]);
  assert.deepEqual(Array.from(kernel.output), [10]);

  kernel.setOutput([5]);
  result = kernel();
  assert.equal(result.result.length, 5);
  assert.equal(result.result2.length, 5);
  assert.deepEqual(Array.from(result.result), [5,6,7,8,9]);
  assert.deepEqual(Array.from(result.result2), [5,6,7,8,9]);
  assert.deepEqual(Array.from(kernel.output), [5]);

  gpu.destroy();
}

test('dynamic output 1d kernel map shrinks auto', () => {
  dynamicOutput1DKernelMapShrinks();