From 38538fcd8006c58371878b3c7c42b3224e400440 Mon Sep 17 00:00:00 2001
From: adrien <adrien@bouvais.lu>
Date: Sun, 17 May 2026 23:18:49 +0200
Subject: [PATCH] Removed waiting for each run and moved it in bench directly

---
 src/Vec.zig   |  1 -
 src/bench.zig | 12 ++++++++----
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/Vec.zig b/src/Vec.zig
index 345cb94..8c4370b 100644
--- a/src/Vec.zig
+++ b/src/Vec.zig
@@ -185,7 +185,6 @@ fn submitPass(
     defer c.wgpuCommandEncoderRelease(enc);
     defer c.wgpuCommandBufferRelease(cmd);
     c.wgpuQueueSubmit(gloc.device.queue, 1, &cmd);
-    _ = c.wgpuDevicePoll(gloc.device.device, 1, null); // Wait for it to be done
 }
 
 fn ceilDiv(n: usize, d: usize) usize {
diff --git a/src/bench.zig b/src/bench.zig
index c1092ee..a163889 100644
--- a/src/bench.zig
+++ b/src/bench.zig
@@ -4,6 +4,8 @@ const GpuAllocator = @import("GpuAllocator.zig");
 const GpuPipeline = @import("GpuPipeline.zig");
 const Vec = @import("Vec.zig");
 
+const c = @import("c.zig").c;
+
 pub fn main(init: std.process.Init) !void {
     const device = try GpuDevice.init();
     defer device.deinit();
@@ -33,10 +35,10 @@ pub fn main(init: std.process.Init) !void {
     const sizes = [_]usize{
         1,
         1024,
-        4096,
-        16384,
-        65536,
-        262144,
+        4 * 1024,
+        4 * 4 * 1024,
+        4 * 4 * 4 * 1024,
+        4 * 4 * 4 * 4 * 1024,
         1024 * 1024,
         4 * 1024 * 1024,
         4 * 4 * 1024 * 1024,
@@ -89,6 +91,8 @@ pub fn main(init: std.process.Init) !void {
             const sum = try a.run(&gloc, b, add_pip);
             defer sum.deinit();
 
+            _ = c.wgpuDevicePoll(device.device, 1, null);
+
             const compute_duration = compute_start.durationTo(std.Io.Clock.awake.now(init.io));
             const compute_ns = @as(u64, @intCast(compute_duration.toNanoseconds()));
             if (compute_ns < min_compute_ns) min_compute_ns = compute_ns;