Merge pull request #290 from neherlab/perf/jemalloc

perf: set jemalloc as global allocator
neherlab · Dec 6, 2024 · 4538fd6 · 4538fd6
2 parents 20c5b36 + 1acf00b
commit 4538fd6
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -123,6 +123,7 @@ statrs = "=0.17.1"
 strum = "=0.26.3"
 strum_macros = "=0.26.4"
 textplots = "=0.8.6"
+tikv-jemallocator = "=0.6.0"
 time = { version = "=0.3.36", features = ["serde", "macros", "parsing"] }
 traversal = "=0.1.2"
 typify = { version = "=0.2.0", features = ["macro"] }

diff --git a/packages/treetime/Cargo.toml b/packages/treetime/Cargo.toml
@@ -79,9 +79,23 @@ serde_yaml = { workspace = true }
 smart-default = { workspace = true }
 strum = { workspace = true }
 strum_macros = { workspace = true }
+tikv-jemallocator = { workspace = true, optional = true }
 time = { workspace = true }
 traversal = { workspace = true }
 
+[target.'cfg(all(target_arch = "x86_64", target_os = "linux", target_env = "gnu"))'.dependencies]
+tikv-jemallocator = { workspace = true }
+
+[target.'cfg(all(target_arch = "aarch64", target_os = "linux", target_env = "gnu"))'.dependencies]
+tikv-jemallocator = { workspace = true }
+
+[target.'cfg(all(target_arch = "x86_64", target_os = "linux", target_env = "musl"))'.dependencies]
+tikv-jemallocator = { workspace = true }
+
+[target.'cfg(all(target_arch = "aarch64", target_os = "linux", target_env = "musl"))'.dependencies]
+tikv-jemallocator = { workspace = true }
+
+
 [features]
 "png" = ["image", "plotters/bitmap_backend", "plotters/ttf"]
 

diff --git a/packages/treetime/src/lib.rs b/packages/treetime/src/lib.rs
@@ -10,6 +10,15 @@ pub mod representation;
 pub mod seq;
 pub mod utils;
 
+#[cfg(any(
+  all(target_arch = "aarch64", target_os = "linux", target_env = "gnu"),
+  all(target_arch = "aarch64", target_os = "linux", target_env = "musl"),
+  all(target_arch = "x86_64", target_os = "linux", target_env = "gnu"),
+  all(target_arch = "x86_64", target_os = "linux", target_env = "musl"),
+))]
+#[global_allocator]
+static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
+
 #[cfg(test)]
 mod tests {
   use crate::utils::global_init::global_init;