Function immutableSubKernelsWithoutFloats

test/issues/267-immutable-sub-kernels.js:102–145

(mode)

describe('issue #267 sub kernel');

function immutableSubKernelsWithoutFloats(mode) {
  function value1(value) {
    return value + 1;
  }

  function value2(value) {
    return value + 1;
  }

  const gpu = new GPU({ mode });
  const kernel = gpu.createKernelMap(
    {
      valueOutput1: value1,
      valueOutput2: value2
    },
    function (a, b) {
      value1(a[this.thread.x]);
      return value2(b[this.thread.x]);
    },
    {
      output: [1],
      immutable: true,
      pipeline: true,
      precision: 'unsigned',
    }
  );

  // start with a value on CPU
  const output1 = kernel([1], [2]);
  const result1 = output1.valueOutput1.toArray()[0];

  // reuse that output, simulating that this value will be monitored, and updated via the same kernel
  // this is often used in neural networks
  const output2 = kernel(output1.valueOutput1, output1.valueOutput2);
  const result2 = output2.valueOutput1.toArray()[0];

  const output3 = kernel(output2.valueOutput1, output2.valueOutput2);
  const result3 = output3.valueOutput1.toArray()[0];

  assert.equal(result1, 2);
  assert.equal(result2, 3);
  assert.equal(result3, 4);
  gpu.destroy();
}
(GPU.isKernelMapSupported ? test : skip)('Issue #267 immutable sub-kernel output - auto', () => {
  immutableSubKernelsWithoutFloats();
});