chore: Allow for choice in memory allocator and use mimalloc for pola…

…rs-lts-cpu (#5414) Co-authored-by: Louis Groux <louis.groux@paige.ai>
pola-rs · Nov 4, 2022 · 3e34e69 · 3e34e69
1 parent ffa17c0
commit 3e34e69
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/create-py-release-manylinux-lts-cpu.yaml b/.github/workflows/create-py-release-manylinux-lts-cpu.yaml
@@ -30,7 +30,7 @@ jobs:
         uses: messense/maturin-action@v1
         env:
           MATURIN_PASSWORD: ${{ secrets.PYPI_PASS }}
-          RUSTFLAGS: -C target-feature=+fxsr,+sse,+sse2,+sse3,+ssse3,+sse4.1,+sse4.2,+popcnt
+          RUSTFLAGS: -C target-feature=+fxsr,+sse,+sse2,+sse3,+ssse3,+sse4.1,+sse4.2,+popcnt --cfg use_mimalloc
         with:
           rust-toolchain: nightly-2022-10-24
           maturin-version: '0.13.5'

diff --git a/py-polars/src/lib.rs b/py-polars/src/lib.rs
@@ -25,10 +25,10 @@ mod set;
 mod sql;
 pub mod utils;
 
-#[cfg(target_os = "linux")]
+#[cfg(all(target_os = "linux", not(use_mimalloc)))]
 use jemallocator::Jemalloc;
 use lazy::ToExprs;
-#[cfg(not(target_os = "linux"))]
+#[cfg(any(not(target_os = "linux"), use_mimalloc))]
 use mimalloc::MiMalloc;
 use polars::functions::{diag_concat_df, hor_concat_df};
 use polars::prelude::Null;
@@ -57,11 +57,11 @@ use crate::prelude::{
 use crate::series::PySeries;
 
 #[global_allocator]
-#[cfg(target_os = "linux")]
+#[cfg(all(target_os = "linux", not(use_mimalloc)))]
 static ALLOC: Jemalloc = Jemalloc;
 
 #[global_allocator]
-#[cfg(not(target_os = "linux"))]
+#[cfg(any(not(target_os = "linux"), use_mimalloc))]
 static ALLOC: MiMalloc = MiMalloc;
 
 #[pyfunction]