Fix async tests not waiting

gpuweb · Nov 2, 2023 · 774e392 · 774e392
1 parent b233b1e
commit 774e392
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/webgpu/shader/execution/expression/expression.ts b/src/webgpu/shader/execution/expression/expression.ts
@@ -367,6 +367,8 @@ export async function run(
     void t.queue.onSubmittedWorkDone().finally(batchFinishedCallback);
   };
 
+  const pendingBatches = [];
+
   for (let i = 0; i < cases.length; i += casesPerBatch) {
     const batchCases = cases.slice(i, Math.min(i + casesPerBatch, cases.length));
 
@@ -379,8 +381,10 @@ export async function run(
     }
     batchesInFlight += 1;
 
-    void processBatch(batchCases);
+    pendingBatches.push(processBatch(batchCases));
   }
+
+  await Promise.all(pendingBatches);
 }
 
 /**