CodexCapsule / gpu.js
LordXido's picture
Update gpu.js
1ac49d6 verified
raw
history blame contribute delete
442 Bytes
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"
}
});
}