disable lifo slot to make all tasks stealable

vercel · Jan 29, 2025 · b356147 · b356147
1 parent e1e9b26
commit b356147
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/crates/napi/src/lib.rs b/crates/napi/src/lib.rs
@@ -76,13 +76,25 @@ static ALLOC: dhat::Alloc = dhat::Alloc;
 #[napi::module_init]
 
 fn init() {
+    use tokio::runtime::Builder;
+    use turbo_tasks_malloc::TurboMalloc;
+
     set_hook(Box::new(|panic_info| {
         util::log_internal_error_and_inform(&format!(
             "Panic: {}\nBacktrace: {:?}",
             panic_info,
             Backtrace::new()
         ));
     }));
+    let rt = Builder::new_multi_thread()
+        .enable_all()
+        .on_thread_stop(|| {
+            TurboMalloc::thread_stop();
+        })
+        .disable_lifo_slot()
+        .build()
+        .unwrap();
+    create_custom_tokio_runtime(rt);
 }
 
 #[inline]

diff --git a/turbopack/crates/turbopack-cli/src/main.rs b/turbopack/crates/turbopack-cli/src/main.rs
@@ -29,6 +29,7 @@ fn main() {
         .on_thread_stop(|| {
             TurboMalloc::thread_stop();
         })
+        .disable_lifo_slot()
         .build()
         .unwrap()
         .block_on(main_inner(args))