Spaces:
Running
Running
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"
}
});
} |