gpu.js/test/features/add-custom-native-function.js

(function() {
  var glslDivide = `float divide(float a, float b) {
    return a / b;
  }`;
  var jsDivide = `function divide(a, b) {
    return a / b;
  }`;

  function addCustomNativeFunctionDivide(mode, fn) {
    var gpu = new GPU({ mode: mode });

    gpu.addNativeFunction('divide', fn);

    var f = gpu.createKernel(function(a, b) {
      return divide(a[this.thread.x], b[this.thread.x]);
    }, {
      output : [6]
    });

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

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

    var res = f(a,b);
    var exp = [0.25, 0.4, 0.5, 5, 3, 2.33];

    for(var i = 0; i < exp.length; ++i) {
      QUnit.assert.close(res[i], exp[i], 0.1, 'Result arr idx: '+i);
    }
  }

  QUnit.test( 'addCustomNativeFunctionDivide (auto)', function() {
    addCustomNativeFunctionDivide(null, glslDivide);
  });

  QUnit.test( 'addCustomNativeFunctionDivide (gpu)', function() {
    addCustomNativeFunctionDivide('gpu', glslDivide);
  });

  QUnit.test( 'addCustomNativeFunctionDivide (webgl)', function() {
    addCustomNativeFunctionDivide('webgl', glslDivide);
  });

  QUnit.test( 'addCustomNativeFunctionDivide (webgl2)', function() {
    addCustomNativeFunctionDivide('webgl2', glslDivide);
  });

  QUnit.test( 'addCustomNativeFunctionDivide (cpu)', function() {
    addCustomNativeFunctionDivide('cpu', jsDivide);
  });

  function addCustomNativeFunctionDivideFallback(mode) {
    var gpu = new GPU({ mode: mode });

    gpu.addNativeFunction('divide', `float divide(float a, float b) {
    return a / b;
  }`);

    function divide(a,b) {
      return a / b;
    }

    var f = gpu.createKernel(function(a, b) {
      return divide(a[this.thread.x], b[this.thread.x]);
    }, {
      functions: [divide],
      output : [6]
    });

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

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

    var res = f(a,b);
    var exp = [0.25, 0.4, 0.5, 5, 3, 2.33];

    for(var i = 0; i < exp.length; ++i) {
      QUnit.assert.close(res[i], exp[i], 0.1, 'Result arr idx: '+i);
    }
  }

  QUnit.test( 'addCustomNativeFunctionDivideFallback (GPU only) (auto)', function() {
    addCustomNativeFunctionDivideFallback(null);
  });

  QUnit.test( 'addCustomNativeFunctionDivideFallback (GPU only) (gpu)', function() {
    addCustomNativeFunctionDivideFallback('gpu');
  });

  QUnit.test( 'addCustomNativeFunctionDivideFallback (GPU only) (webgl)', function() {
    addCustomNativeFunctionDivideFallback('webgl');
  });

  QUnit.test( 'addCustomNativeFunctionDivideFallback (GPU only) (webgl2)', function() {
    addCustomNativeFunctionDivideFallback('webgl2');
  });
})();