Function compileProgram

tfjs-backend-webgpu/src/webgpu_program.ts:62–96

(device: GPUDevice, program: WebGPUProgram, inputsData: InputInfo[],
     output: TensorInfo, parallelCompilation: boolean)

export const compileProgram =
    (device: GPUDevice, program: WebGPUProgram, inputsData: InputInfo[],
     output: TensorInfo, parallelCompilation: boolean): GPUComputePipeline|
    Promise<GPUComputePipeline> => {
      const outputData = {dtype: output.dtype, shape: output.shape};
      const source = makeShader(inputsData, outputData, program);
      const module = device.createShaderModule(
          {code: source, label: program.constructor.name});

      let printShaderString = env().get('WEBGPU_PRINT_SHADER') as string;
      if (printShaderString !== '') {
        printShaderString = printShaderString.toLowerCase();
        const printShaderArray = printShaderString.split(',');
        if (printShaderString === 'all' ||
            printShaderArray.some(
                item => program.shaderKey.toLowerCase().includes(item))) {
          console.group(program.shaderKey);
          console.debug(source);
          console.groupEnd();
        }
      }

      if (parallelCompilation) {
        return device.createComputePipelineAsync({
          compute: {module, entryPoint: '_start'},
          label: program.constructor.name,
          layout: 'auto'
        });
      } else {
        return device.createComputePipeline({
          compute: {module, entryPoint: '_start'},
          label: program.constructor.name,
          layout: 'auto'
        });
      }
    };

export const typeSnippet = (component: number, type = 'f32') => {
  switch (component) {