zig

commit 9ae4e38ca27e1b14e84f58594b8be4513f1fcfbe (tree)
parent 2046e0f4be23028d18009a61a517d0712e2ca164
Author: Matthew Lugg <mlugg@mlugg.co.uk>
Date:   Sun, 21 Dec 2025 14:50:16 +0000

tools: replace thread pool with `std.Io`

Diffstat:
M
tools/update_cpu_features.zig
 | 
51
++++++++++++++++++++++++---------------------------

1 file changed, 24 insertions(+), 27 deletions(-)
diff --git a/tools/update_cpu_features.zig b/tools/update_cpu_features.zig
@@ -1882,10 +1882,18 @@ const targets = [_]ArchTarget{
 };
 
 pub fn main() anyerror!void {
-    var arena_state = std.heap.ArenaAllocator.init(std.heap.page_allocator);
+    var debug_allocator: std.heap.DebugAllocator(.{}) = .init;
+    defer _ = debug_allocator.deinit();
+    const gpa = debug_allocator.allocator();
+
+    var arena_state: std.heap.ArenaAllocator = .init(gpa);
     defer arena_state.deinit();
     const arena = arena_state.allocator();
 
+    var threaded: std.Io.Threaded = .init(gpa);
+    defer threaded.deinit();
+    const io = threaded.io();
+
     var args = try std.process.argsWithAllocator(arena);
     const args0 = args.next().?;
 
@@ -1925,34 +1933,23 @@ pub fn main() anyerror!void {
     const root_progress = std.Progress.start(.{ .estimated_total_items = targets.len });
     defer root_progress.end();
 
-    if (builtin.single_threaded) {
-        for (targets) |target| {
-            if (filter) |zig_name| if (!std.mem.eql(u8, target.zig_name, zig_name)) continue;
-            try processOneTarget(.{
-                .llvm_tblgen_exe = llvm_tblgen_exe,
-                .llvm_src_root = llvm_src_root,
-                .zig_src_dir = zig_src_dir,
-                .root_progress = root_progress,
-                .target = target,
-            });
-        }
-    } else {
-        var pool: std.Thread.Pool = undefined;
-        try pool.init(.{ .allocator = arena, .n_jobs = targets.len });
-        defer pool.deinit();
-
-        for (targets) |target| {
-            if (filter) |zig_name| if (!std.mem.eql(u8, target.zig_name, zig_name)) continue;
-            const job = Job{
-                .llvm_tblgen_exe = llvm_tblgen_exe,
-                .llvm_src_root = llvm_src_root,
-                .zig_src_dir = zig_src_dir,
-                .root_progress = root_progress,
-                .target = target,
-            };
-            try pool.spawn(processOneTarget, .{job});
+    var group: std.Io.Group = .init;
+    defer group.cancel(io);
+
+    for (targets) |target| {
+        if (filter) |zig_name| {
+            if (!std.mem.eql(u8, target.zig_name, zig_name)) continue;
         }
+        group.async(io, processOneTarget, .{.{
+            .llvm_tblgen_exe = llvm_tblgen_exe,
+            .llvm_src_root = llvm_src_root,
+            .zig_src_dir = zig_src_dir,
+            .root_progress = root_progress,
+            .target = target,
+        }});
     }
+
+    group.wait(io);
 }
 
 const Job = struct {