gpu.js/test/issues/91-create-kernel-map-array.js

(function () {
  const GPU = require('../../src/index');
  function getResult(mode) {
    const A = [
      [1, 2],
      [3, 4],
      [5, 6]
    ];

    const B = [
      [6, 5, 4],
      [3, 2, 1]
    ];

    const gpu = new GPU({ mode: mode });

    function multiply(b, a, y, x) {
      let sum = 0;
      for (let i = 0; i < 2; i++) {
        sum += b[y][i] * a[i][x];
      }
      return sum;
    }

    const kernels = gpu.createKernelMap({
      multiplyResult: multiply
    }, function (a, b) {
      return multiply(b, a, this.thread.y, this.thread.x);
    })
      .setOutput([2, 2]);
    const result = kernels(A, B).result;
    QUnit.assert.deepEqual(QUnit.extend([], result[0]), [21,32]);
    QUnit.assert.deepEqual(QUnit.extend([], result[1]), [9,14]);
    gpu.destroy();
    return kernels;
  }
  (GPU.isWebGL2Supported || (GPU.isHeadlessGLSupported && GPU.HeadlessGLKernel.features.kernelMap) ? QUnit.test : QUnit.skip)("Issue #91 - type detection (auto)", () => {
    getResult();
  });
  (GPU.isWebGL2Supported || (GPU.isHeadlessGLSupported && GPU.HeadlessGLKernel.features.kernelMap) ? QUnit.test : QUnit.skip)("Issue #91 - type detection (gpu)", () => {
    getResult('gpu');
  });
  (GPU.isWebGLSupported ? QUnit.test : QUnit.skip)("Issue #91 - type detection (webgl)", () => {
    const kernel = getResult('webgl');
    QUnit.assert.equal(kernel.kernel.constructor, GPU.WebGLKernel, 'kernel type is wrong');
  });
  (GPU.isWebGL2Supported ? QUnit.test : QUnit.skip)("Issue #91 - type detection (webgl2)", () => {
    const kernel = getResult('webgl2');
    QUnit.assert.equal(kernel.kernel.constructor, GPU.WebGL2Kernel, 'kernel type is wrong');
  });
  (GPU.isHeadlessGLSupported ? QUnit.test : QUnit.skip)("Issue #91 - type detection (headlessgl)", () => {
    QUnit.assert.throws(() => {
      getResult('headlessgl');
    });
  });
  QUnit.test("Issue #91 - type detection (cpu)", () => {
    const kernel = getResult('cpu');
    QUnit.assert.equal(kernel.kernel.constructor, GPU.CPUKernel, 'kernel type is wrong');
  });
})();