1 files changed, 6 insertions, 4 deletions
diff --git a/cli/tests/unit/webgpu_test.ts b/cli/tests/unit/webgpu_test.ts
index 6ecf27fd0..fa71ffcc8 100644
--- a/cli/tests/unit/webgpu_test.ts
+++ b/cli/tests/unit/webgpu_test.ts
@@ -56,6 +56,7 @@ Deno.test({
   storageBuffer.unmap();
 
   const computePipeline = device.createComputePipeline({
+    layout: "auto",
     compute: {
       module: shaderModule,
       entryPoint: "main",
@@ -81,8 +82,8 @@ Deno.test({
   computePass.setPipeline(computePipeline);
   computePass.setBindGroup(0, bindGroup);
   computePass.insertDebugMarker("compute collatz iterations");
-  computePass.dispatch(numbers.length);
-  computePass.endPass();
+  computePass.dispatchWorkgroups(numbers.length);
+  computePass.end();
 
   encoder.copyBufferToBuffer(storageBuffer, 0, stagingBuffer, 0, size);
 
@@ -172,13 +173,14 @@ Deno.test({
       {
         view,
         storeOp: "store",
-        loadValue: [0, 1, 0, 1],
+        loadOp: "clear",
+        clearValue: [0, 1, 0, 1],
       },
     ],
   });
   renderPass.setPipeline(renderPipeline);
   renderPass.draw(3, 1);
-  renderPass.endPass();
+  renderPass.end();
 
   encoder.copyTextureToBuffer(
     {