Function testMatrix2

test/features/internally-defined-matrices.js:6–26

(mode)

describe('features: internally defined matrices');

function testMatrix2(mode) {
  const gpu = new GPU({ mode });
  function getMatrix() {
    const matrix = [
      [1,2],
      [3,4]
    ];
    return matrix;
  }
  gpu.addFunction(getMatrix);
  const kernel = gpu.createKernel(function(y, x) {
    return getMatrix()[y][x];
  }, { output: [1] });

  assert.equal(kernel(0, 0)[0], 1);
  assert.equal(kernel(0, 1)[0], 2);
  assert.equal(kernel(1, 0)[0], 3);
  assert.equal(kernel(1, 1)[0], 4);

  gpu.destroy();
}

test('matrix2 auto', () => {
  testMatrix2();