Function withKernelAddFunctionMethod

test/features/add-custom-function.js:149–170

(mode)

});

function withKernelAddFunctionMethod(mode) {
  function customAdder(a, b) {
    return a + b;
  }
  const gpu = new GPU({ mode });
  const kernel = gpu.createKernel(function (a, b) {
    return customAdder(a[this.thread.x], b[this.thread.x]);
  }, {
    output: [6]
  })
    .addFunction(customAdder);

  const a = [1, 2, 3, 5, 6, 7];
  const b = [4, 5, 6, 1, 2, 3];

  const result = kernel(a, b);

  const expected = [5, 7, 9, 6, 8, 10];

  assert.deepEqual(Array.from(result), expected);
  gpu.destroy();
}

test('with Kernel addFunction method auto', () => {
  withKernelAddFunctionMethod(null);