Switch from unmaintained jemalloctor to maintained tikv-jemallocator. (…

…#3141)
pola-rs · Apr 14, 2022 · 00e5c00 · 00e5c00
1 parent f164d7b
commit 00e5c00
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 26 deletions.
diff --git a/polars/src/lib.rs b/polars/src/lib.rs
@@ -261,7 +261,9 @@
 //!
 //! ### Custom allocator
 //! A DataFrame library naturally does a lot of heap allocations. It is recommended to use a custom
-//! allocator. [Mimalloc](https://docs.rs/mimalloc/0.1.25/mimalloc/) and [JeMalloc](https://crates.io/crates/jemallocator) for instance, show a significant
+//! allocator.
+//! [Mimalloc](https://crates.io/crates/mimalloc) and
+//! [JeMalloc](https://crates.io/crates/tikv-jemallocator) for instance, show a significant
 //! performance gain in runtime as well as memory usage.
 //!
 //! #### Usage

diff --git a/py-polars/Cargo.lock b/py-polars/Cargo.lock
diff --git a/py-polars/Cargo.toml b/py-polars/Cargo.toml
@@ -19,7 +19,7 @@ description = "Blazingly fast DataFrame library"
 mimalloc = { version = "*", default-features = false }
 
 [target.'cfg(target_os="linux")'.dependencies]
-jemallocator = { version = "0.3", features = ["disable_initial_exec_tls"] }
+tikv-jemallocator = { version = "0.4", features = ["disable_initial_exec_tls"] }
 
 [dependencies]
 ahash = "0.7"

diff --git a/py-polars/src/lib.rs b/py-polars/src/lib.rs
@@ -39,8 +39,6 @@ use crate::error::{
 use crate::file::get_either_file;
 use crate::prelude::{ClosedWindow, DataType, DatetimeArgs, Duration, DurationArgs, PyDataType};
 use dsl::ToExprs;
-#[cfg(target_os = "linux")]
-use jemallocator::Jemalloc;
 #[cfg(not(target_os = "linux"))]
 use mimalloc::MiMalloc;
 use polars::functions::{diag_concat_df, hor_concat_df};
@@ -49,6 +47,8 @@ use polars_core::datatypes::TimeUnit;
 use polars_core::export::arrow::io::ipc::read::read_file_metadata;
 use polars_core::prelude::IntoSeries;
 use pyo3::types::{PyBool, PyDict, PyFloat, PyInt, PyString};
+#[cfg(target_os = "linux")]
+use tikv_jemallocator::Jemalloc;
 
 #[global_allocator]
 #[cfg(target_os = "linux")]