Function immutableKernelWithoutFloats

test/issues/267-immutable-sub-kernels.js:6–33

(mode)

describe('issue #267 kernel');

function immutableKernelWithoutFloats(mode) {
  const gpu = new GPU({ mode });
  const kernel = gpu.createKernel(function (v) {
    return v[this.thread.x] + 1;
  }, {
    output: [1],
    immutable: true,
    pipeline: true,
    precision: 'unsigned',
  });

  // start with a value on CPU
  const output1 = kernel([1]);
  const result1 = output1.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);
  const result2 = output2.toArray()[0];

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

  assert.equal(result1, 2);
  assert.equal(result2, 3);
  assert.equal(result3, 4);
  gpu.destroy();
}

test('Issue #267 immutable kernel output without floats - auto', () => {
  immutableKernelWithoutFloats();