perf(rome_cli): replace global allocator with jemalloc (#3237)

rome · Sep 21, 2022 · 806de56 · 806de56
1 parent c7aebe3
commit 806de56
Show file tree

Hide file tree

Showing 5 changed files with 43 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/rome_cli/Cargo.toml b/crates/rome_cli/Cargo.toml
@@ -42,6 +42,9 @@ tokio = { version = "1.15.0", features = ["process"] }
 [target.'cfg(windows)'.dependencies]
 mimalloc = "0.1.29"
 
+[target.'cfg(not(windows))'.dependencies]
+tikv-jemallocator = "0.5.0"
+
 [dev-dependencies]
 insta = "1.18.2"
 tokio = { version = "1.15.0", features = ["io-util"] }
diff --git a/crates/rome_cli/src/main.rs b/crates/rome_cli/src/main.rs
@@ -12,6 +12,10 @@ use tokio::runtime::Runtime;
 #[global_allocator]
 static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
 
+#[cfg(not(target_os = "windows"))]
+#[global_allocator]
+static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
+
 fn main() -> Result<(), Termination> {
     setup_panic_handler();
 

diff --git a/xtask/bench/Cargo.toml b/xtask/bench/Cargo.toml
@@ -30,5 +30,8 @@ humansize = {version = "1.1.1", optional = true }
 [target.'cfg(target_os = "windows")'.dependencies]
 mimalloc = "0.1.29"
 
+[target.'cfg(not(windows))'.dependencies]
+tikv-jemallocator = "0.5.0"
+
 [features]
 dhat-heap = ["dhat", "humansize"]
diff --git a/xtask/bench/src/main.rs b/xtask/bench/src/main.rs
@@ -10,6 +10,10 @@ static ALLOCATOR: dhat::Alloc = dhat::Alloc;
 #[global_allocator]
 static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
 
+#[cfg(all(not(target_os = "windows"), not(feature = "dhat-heap")))]
+#[global_allocator]
+static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
+
 fn main() -> Result<(), pico_args::Error> {
     #[cfg(feature = "dhat-heap")]
     let _profiler = dhat::Profiler::new_heap();