Function inputZYXVariables

test/features/input.js:240–256

(mode)

function inputZYXVariables(mode) {
  const gpu = new GPU({ mode });
  const kernel = gpu.createKernel(function(a, x, y, z) {
    return a[z][y][x];
  })
    .setOutput([1]);

  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 aInput = input(a, [2, 4, 4]);
  assert.deepEqual(Array.from(kernel(aInput, 1, 2, 3)), [30]);
  assert.deepEqual(Array.from(kernel(aInput, 0, 2, 3)), [29]);
  assert.deepEqual(Array.from(kernel(aInput, 0, 2, 1)), [13]);
  assert.deepEqual(Array.from(kernel(aInput, 1, 2, 2)), [22]);
  assert.deepEqual(Array.from(kernel(aInput, 0, 2, 2)), [21]);
  gpu.destroy();
}

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