Function floatPipeline3DOutput

test/features/optimize-float-memory.js:291–316

(mode)

function floatPipeline3DOutput(mode) {
  const gpu = new GPU({ mode });
  const cube = [
    [
      [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],
    ]
  ];
  const kernel = gpu.createKernel(function(cube) {
    return cube[this.thread.z][this.thread.y][this.thread.x];
  }, {
    output: [5, 3, 2],
    optimizeFloatMemory: true,
    precision: 'single',
    pipeline: true,
  });
  const result = kernel(cube).toArray();
  assert.deepEqual(result.map(matrix => matrix.map(row => Array.from(row))), cube);
  gpu.destroy();
}


(GPU.isSinglePrecisionSupported && GPU.isGPUSupported ? test : skip)('float pipeline 3d output gpu (GPU only)', () => {