1 files changed, 6 insertions, 23 deletions
diff --git a/cli/tests/unit/webgpu_test.ts b/cli/tests/unit/webgpu_test.ts
index 7b761ee3c..4fbbd5cde 100644
--- a/cli/tests/unit/webgpu_test.ts
+++ b/cli/tests/unit/webgpu_test.ts
@@ -49,18 +49,13 @@ unitTest({
 
   storageBuffer.unmap();
 
-  const bindGroupLayout = device.createBindGroupLayout({
-    entries: [
-      {
-        binding: 0,
-        visibility: 4,
-        buffer: {
-          type: "storage",
-          minBindingSize: 4,
-        },
-      },
-    ],
+  const computePipeline = device.createComputePipeline({
+    compute: {
+      module: shaderModule,
+      entryPoint: "main",
+    },
   });
+  const bindGroupLayout = computePipeline.getBindGroupLayout(0);
 
   const bindGroup = device.createBindGroup({
     layout: bindGroupLayout,
@@ -74,18 +69,6 @@ unitTest({
     ],
   });
 
-  const pipelineLayout = device.createPipelineLayout({
-    bindGroupLayouts: [bindGroupLayout],
-  });
-
-  const computePipeline = device.createComputePipeline({
-    layout: pipelineLayout,
-    compute: {
-      module: shaderModule,
-      entryPoint: "main",
-    },
-  });
-
   const encoder = device.createCommandEncoder();
 
   const computePass = encoder.beginComputePass();