Add TurboMalloc::thread_park() to flush and collect on thread park

crates/next-build-test/src/main.rs

@@ -92,6 +92,7 @@ fn main() {
                             *cell = Some(Instant::now());
                         }
                     });
+                    TurboMalloc::thread_park();
                 })
                 .build()
                 .unwrap()

crates/next-napi-bindings/src/lib.rs

@@ -106,6 +106,7 @@ fn init() {
                     *cell = Some(Instant::now());
                 }
             });
+            TurboMalloc::thread_park();
         })
         .worker_threads(worker_threads)
         // Avoid a limit on threads to avoid deadlocks due to usage of block_in_place

turbopack/crates/turbo-tasks-backend/fuzz/src/graph.rs

@@ -87,6 +87,9 @@ static RUNTIME: Lazy<tokio::runtime::Runtime> = Lazy::new(|| {
         .on_thread_stop(|| {
             TurboMalloc::thread_stop();
         })
+        .on_thread_park(|| {
+            TurboMalloc::thread_park();
+        })
         .build()
         .unwrap()
 });

turbopack/crates/turbo-tasks-malloc/Cargo.toml

@@ -11,6 +11,10 @@ bench = false
 
 [dependencies]
 
+[target.'cfg(not(target_family = "wasm"))'.dependencies]
+libmimalloc-sys = { version = "0.1.44", features = [
+  "extended",
+], optional = true }
 
 [target.'cfg(not(any(target_os = "linux", target_family = "wasm")))'.dependencies]
 mimalloc = { version = "0.1.48", features = [
@@ -26,5 +30,5 @@ mimalloc = { version = "0.1.48", features = [
 ], optional = true }
 
 [features]
-custom_allocator = ["dep:mimalloc"]
+custom_allocator = ["dep:mimalloc", "dep:libmimalloc-sys"]
 default = ["custom_allocator"]

turbopack/crates/turbo-tasks-malloc/src/lib.rs

@@ -93,6 +93,13 @@ impl TurboMalloc {
         flush();
     }
 
+    pub fn thread_park() {
+        #[cfg(all(feature = "custom_allocator", not(target_family = "wasm")))]
+        unsafe {
+            libmimalloc_sys::mi_collect(false);
+        }
+    }
+
     pub fn allocation_counters() -> AllocationCounters {
         self::counter::allocation_counters()
     }

turbopack/crates/turbopack-cli/benches/small_apps.rs

@@ -50,9 +50,13 @@ fn bench_small_apps(c: &mut Criterion) {
 
                 b.iter(move || {
                     let mut rt = tokio::runtime::Builder::new_multi_thread();
-                    rt.enable_all().on_thread_stop(|| {
-                        TurboMalloc::thread_stop();
-                    });
+                    rt.enable_all()
+                        .on_thread_stop(|| {
+                            TurboMalloc::thread_stop();
+                        })
+                        .on_thread_park(|| {
+                            TurboMalloc::thread_park();
+                        });
                     let rt = rt.build().unwrap();
 
                     let apps_dir = apps_dir.clone();

turbopack/crates/turbopack-cli/src/main.rs

@@ -37,6 +37,7 @@ fn main() {
                     *cell = Some(Instant::now());
                 }
             });
+            TurboMalloc::thread_park();
         });
 
     let args = Arguments::parse();