Function inputYX

test/features/input.js:84–97

(mode)

});

function inputYX(mode) {
  const gpu = new GPU({ mode });
  const kernel = gpu.createKernel(function(a) {
    return a[this.thread.y][this.thread.x];
  })
    .setOutput([3, 3]);

  const a = new Float32Array(9);
  a.set([1,2,3,4,5,6,7,8,9]);

  const result = kernel(input(a, [3, 3]));
  assert.deepEqual(result.map(function(v) { return Array.from(v); }), [[1,2,3],[4,5,6],[7,8,9]]);
  gpu.destroy();
}

test("inputYX auto", () => {
  inputYX();