Custom thread pool

.gitignore

@@ -1,4 +1,6 @@
 /target
+/target-*
+/.lm-*
 .vscode
 /docs/benchmark_graphs/.venv
 minimal_zkVM.synctex.gz

Cargo.toml

@@ -19,7 +19,7 @@ members = [
     "crates/backend/fiat-shamir",
     "crates/backend/sumcheck",
     "crates/backend/system-info",
-    "crates/backend/zk-alloc",
+    "crates/backend/parallel",
 ]
 
 [workspace.lints]
@@ -61,14 +61,13 @@ lean_compiler = { path = "crates/lean_compiler" }
 lean_prover = { path = "crates/lean_prover" }
 rec_aggregation = { path = "crates/rec_aggregation" }
 backend = { path = "crates/backend" }
-zk-alloc = { path = "crates/backend/zk-alloc" }
 system-info = { path = "crates/backend/system-info" }
+parallel = { path = "crates/backend/parallel" }
 
 # External
 sha3 = "0.11.0"
 clap = { version = "4.5.59", features = ["derive"] }
 rand = "0.10.0"
-rayon = "1.11.0"
 pest = "2.7"
 pest_derive = "2.7"
 itertools = "0.14.0"
@@ -83,12 +82,15 @@ include_dir = "0.7"
 
 [features]
 prox-gaps-conjecture = ["rec_aggregation/prox-gaps-conjecture"]
-standard-alloc = ["rec_aggregation/standard-alloc"]
+# Build with the plain system allocator instead of mimalloc (for comparison/debugging).
+standard-alloc = []
 
 [dependencies]
 clap.workspace = true
 rec_aggregation.workspace = true
-zk-alloc.workspace = true
+mimalloc = "0.1"
+libmimalloc-sys = { version = "0.1", features = ["extended"] }
+libc = "0.2"
 rand.workspace = true
 sub_protocols.workspace = true
 utils.workspace = true

crates/backend/Cargo.toml

@@ -9,7 +9,7 @@ poly = { path = "poly", package = "mt-poly" }
 sumcheck = { path = "sumcheck", package = "mt-sumcheck" }
 field = { path = "field", package = "mt-field" }
 air = { path = "air", package = "mt-air" }
-rayon.workspace = true
+parallel.workspace = true
 whir = { path = "../whir", package = "mt-whir" }
 tracing.workspace = true
 fiat-shamir = { path = "fiat-shamir", package = "mt-fiat-shamir" }

crates/backend/fiat-shamir/Cargo.toml

@@ -10,4 +10,4 @@ symetric = { path = "../symetric", package = "mt-symetric" }
 utils = { path = "../utils", package = "mt-utils" }
 tracing.workspace = true
 serde.workspace = true
-rayon.workspace = true
+parallel.workspace = true