File size: 442 Bytes
da2b37f
1ac49d6
da2b37f
 
1ac49d6
 
 
 
16d758a
 
2cd3c22
da2b37f
16d758a
 
1ac49d6
16d758a
 
 
d21242b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
export async function initGPU() {
  if (!navigator.gpu) {
    return { mode: "cpu" };
  }

  const adapter = await navigator.gpu.requestAdapter();
  const device = await adapter.requestDevice();
  return { mode: "gpu", device };
}

export function createCompute(device, code) {
  return device.createComputePipeline({
    layout: "auto",
    compute: {
      module: device.createShaderModule({ code }),
      entryPoint: "main"
    }
  });
}