Function addCustomPrivate

test/features/add-custom-function.js:311–337

(mode)

describe('features: add custom private');
function addCustomPrivate(mode) {
  const gpu = new GPU({ mode });

  const kernel = gpu.createKernel(function(a, b) {
    function customAdder(a, b) {
      let sum = 0;
      for (let i = 0; i < this.output.x; i++) {
        sum += a[this.thread.x] + b[this.thread.x];
      }
      return sum;
    }
    return customAdder(a, b);
  }, {
    output : [6],
  });

  assert.ok(kernel !== null, 'function generated test');

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

  const result = kernel(a,b);
  const expected = [12, 18, 24, 36, 42, 48];

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

test('auto', () => {
  addCustomPrivate(null);