chore: Use tikv-jemallocator (#21486)

pola-rs · Feb 28, 2025 · e8ee335 · e8ee335
1 parent 4e4d259
commit e8ee335
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 27 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/crates/polars/src/lib.rs b/crates/polars/src/lib.rs
@@ -325,13 +325,13 @@
 //! ### Custom allocator
 //! An OLAP query engine does a lot of heap allocations. It is recommended to use a custom
 //! allocator, (we have found this to have up to ~25% runtime influence).
-//! [JeMalloc](https://crates.io/crates/jemallocator) and
+//! [JeMalloc](https://crates.io/crates/tikv-jemallocator) and
 //! [Mimalloc](https://crates.io/crates/mimalloc) for instance, show a significant
 //! performance gain in runtime as well as memory usage.
 //!
 //! #### Jemalloc Usage
 //! ```ignore
-//! use jemallocator::Jemalloc;
+//! use tikv_jemallocator::Jemalloc;
 //!
 //! #[global_allocator]
 //! static GLOBAL: Jemalloc = Jemalloc;
@@ -340,7 +340,7 @@
 //! #### Cargo.toml
 //! ```toml
 //! [dependencies]
-//! jemallocator = { version = "*" }
+//! tikv-jemallocator = { version = "*" }
 //! ```
 //!
 //! #### Mimalloc Usage

diff --git a/py-polars/Cargo.toml b/py-polars/Cargo.toml
@@ -19,10 +19,10 @@ mimalloc = { version = "0.1", default-features = false }
 
 # Feature background_threads is unsupported on MacOS (https://github.com/jemalloc/jemalloc/issues/843).
 [target.'cfg(all(target_family = "unix", not(target_os = "macos"), not(target_os = "emscripten"), not(allocator = "mimalloc"), not(allocator = "default")))'.dependencies]
-jemallocator = { version = "0.5", features = ["disable_initial_exec_tls", "background_threads"] }
+tikv-jemallocator = { version = "0.6.0", features = ["disable_initial_exec_tls", "background_threads"] }
 
 [target.'cfg(all(target_family = "unix", target_os = "macos", not(allocator = "mimalloc"), not(allocator = "default")))'.dependencies]
-jemallocator = { version = "0.5", features = ["disable_initial_exec_tls"] }
+tikv-jemallocator = { version = "0.6.0", features = ["disable_initial_exec_tls"] }
 either = { workspace = true }
 
 [features]

diff --git a/py-polars/src/allocator.rs b/py-polars/src/allocator.rs
@@ -5,7 +5,7 @@
     target_family = "unix",
     not(target_os = "emscripten"),
 ))]
-static ALLOC: jemallocator::Jemalloc = jemallocator::Jemalloc;
+static ALLOC: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
 
 #[global_allocator]
 #[cfg(all(