Add custom global allocator (#30)

* Add sn-malloc * Mimalloc
datafusion-contrib · Feb 25, 2022 · 368b50e · 368b50e
1 parent 6c1ebe9
commit 368b50e
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -35,6 +35,7 @@ datafusion = { version = "^7.0.0", features = ["pyarrow"] }
 datafusion-expr = { version = "^7.0.0" }
 datafusion-common = { version = "^7.0.0", features = ["pyarrow"] }
 uuid = { version = "0.8", features = ["v4"] }
+mimalloc = { version = "*", default-features = false }
 
 [lib]
 name = "_internal"

diff --git a/src/lib.rs b/src/lib.rs
@@ -15,6 +15,7 @@
 // specific language governing permissions and limitations
 // under the License.
 
+use mimalloc::MiMalloc;
 use pyo3::prelude::*;
 
 mod catalog;
@@ -27,6 +28,9 @@ mod udaf;
 mod udf;
 mod utils;
 
+#[global_allocator]
+static GLOBAL: MiMalloc = MiMalloc;
+
 /// Low-level DataFusion internal package.
 ///
 /// The higher-level public API is defined in pure python files under the