Python: only use jemalloc on linux (#3126)

pola-rs · Apr 12, 2022 · 95096a7 · 95096a7
1 parent d0656a4
commit 95096a7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/py-polars/Cargo.toml b/py-polars/Cargo.toml
@@ -15,10 +15,10 @@ description = "Blazingly fast DataFrame library"
 [workspace]
 # prevents package from thinking it's in the workspace
 
-[target.'cfg(windows)'.dependencies]
+[target.'cfg(not(target_os="linux"))'.dependencies]
 mimalloc = { version = "*", default-features = false }
 
-[target.'cfg(not(windows))'.dependencies]
+[target.'cfg(target_os="linux")'.dependencies]
 jemallocator = { version = "0.3", features = ["disable_initial_exec_tls"] }
 
 [dependencies]

diff --git a/py-polars/src/lib.rs b/py-polars/src/lib.rs
@@ -39,9 +39,9 @@ use crate::error::{
 use crate::file::get_either_file;
 use crate::prelude::{ClosedWindow, DataType, DatetimeArgs, Duration, DurationArgs, PyDataType};
 use dsl::ToExprs;
-#[cfg(not(target_os = "windows"))]
+#[cfg(target_os = "linux")]
 use jemallocator::Jemalloc;
-#[cfg(target_os = "windows")]
+#[cfg(not(target_os = "linux"))]
 use mimalloc::MiMalloc;
 use polars::functions::{diag_concat_df, hor_concat_df};
 use polars::prelude::Null;
@@ -51,11 +51,11 @@ use polars_core::prelude::IntoSeries;
 use pyo3::types::{PyBool, PyDict, PyFloat, PyInt, PyString};
 
 #[global_allocator]
-#[cfg(not(target_os = "windows"))]
+#[cfg(target_os = "linux")]
 static ALLOC: Jemalloc = Jemalloc;
 
 #[global_allocator]
-#[cfg(target_os = "windows")]
+#[cfg(not(target_os = "linux"))]
 static ALLOC: MiMalloc = MiMalloc;
 
 #[pyfunction]