Function insertAlignment

tfjs-backend-webgpu/src/webgpu_program.ts:829–842

(uniformShader: string)

}

function insertAlignment(uniformShader: string) {
  // insert alignment when current pattern is vec5 or vec6
  const curInsertRe = /(\w+)\s*:\s*vec(5|6)/g;
  uniformShader = uniformShader.replace(curInsertRe, (match) => {
    return '@align(16) ' + match;
  });

  // insert alignment when previous pattern is vec5 or vec6
  const preInsertRe = /vec(5|6)\s*,\s*(\w+)/g;
  uniformShader = uniformShader.replace(preInsertRe, (_, p1, p2) => {
    return `vec${p1}, @align(16) ${p2}`;
  });
  return uniformShader;
}
function isFlatDispatchLayout(program: WebGPUProgram): boolean {
  if (program.dispatchLayout.hasOwnProperty('y') &&
      program.dispatchLayout.y.length !== 0) {