Function getStartHeaderString

tfjs-backend-webgpu/src/webgpu_program.ts:170–189

(
    useGlobalIndex: boolean, program: WebGPUProgram)

}

export function getStartHeaderString(
    useGlobalIndex: boolean, program: WebGPUProgram): string {
  let snippet: string;
  snippet = `
     ${getWorkgroupSizeString(program)}
      fn _start(@builtin(local_invocation_id) LocalId : vec3<u32>,
                @builtin(global_invocation_id) GlobalId : vec3<u32>,
                @builtin(local_invocation_index) LocalIndex: u32,
                @builtin(workgroup_id) WorkgroupId : vec3<u32>,
                @builtin(num_workgroups) NumWorkgroups : vec3<u32>) {
        localId = LocalId;
        localIndex = LocalIndex;
        globalId = GlobalId;
        numWorkgroups = NumWorkgroups;
        workgroupId = WorkgroupId;
        ${useGlobalIndex ? `main(getGlobalIndex());` : `main();`};
      }
    `;
  return snippet;
}

export function getWorkgroupSizeString(program: WebGPUProgram): string {
  return `