Update gpu.js

gpu.js

@@ -1,4 +1,4 @@
-export async function initGPU() {
+export async function initGPU(canvas) {
   if (!navigator.gpu) {
     return { mode: "cpu" };
   }
@@ -7,5 +7,20 @@ export async function initGPU() {
   if (!adapter) return { mode: "cpu" };
 
   const device = await adapter.requestDevice();
+
   return { mode: "gpu", device };
+}
+
+export async function createCompute(device, wgslSource) {
+  const module = device.createShaderModule({ code: wgslSource });
+
+  const pipeline = device.createComputePipeline({
+    layout: "auto",
+    compute: {
+      module,
+      entryPoint: "main"
+    }
+  });
+
+  return pipeline;
 }