Function inputZYX

test/features/input.js:200–213

(mode)

});

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

  const a = new Float32Array(32);
  a.set([1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32]);

  const result = kernel(input(a, [2, 4, 4]));
  assert.deepEqual(result.map(function(v) { return v.map(function(v) { return Array.from(v); }); }), [[[1,2],[3,4],[5,6],[7,8]],[[9,10],[11,12],[13,14],[15,16]],[[17,18],[19,20],[21,22],[23,24]],[[25,26],[27,28],[29,30],[31,32]]]);
  gpu.destroy();
}

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