Function init

gpu-pipeline/webgpu/index.js:18–154

* @license * Copyright 2022 Google LLC. All Rights Reserved. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unles

()

 */

async function init() {
  const customBackendName = 'custom-webgpu';

  const kernels = tf.getKernelsForBackend('webgpu');
  kernels.forEach(kernelConfig => {
    const newKernelConfig = { ...kernelConfig, backendName: customBackendName };
    tf.registerKernel(newKernelConfig);
  });

  adapter = await navigator.gpu.requestAdapter();
  device = await adapter.requestDevice();

  tf.registerBackend(customBackendName, async () => {
    return new tf.WebGPUBackend(device);
  });
  await tf.setBackend(customBackendName);

  const context = canvasEl.getContext('webgpu');
  const presentationFormat = navigator.gpu.getPreferredCanvasFormat();
  const presentationSize = [
    canvasEl.width,
    canvasEl.height,
  ];

  context.configure({
    device,
    size: presentationSize,
    format: presentationFormat,
    alphaMode: 'opaque',
  });

  const pipeline = device.createRenderPipeline({
    layout: 'auto',
    vertex: {
      module: device.createShaderModule({
        code: VERTEX_SHADER,
      }),
      entryPoint: 'main',
    },
    fragment: {
      module: device.createShaderModule({
        code: PIXEL_SHADER,
      }),
      entryPoint: 'main',
      targets: [
        {
          format: presentationFormat,
        },
      ],
    },
    primitive: {
      topology: 'triangle-list',
    },
  });

  const sampler = device.createSampler({
    magFilter: 'linear',
    minFilter: 'linear',