fix: use larger recursive stack in debug mode

pola-rs · Apr 11, 2024 · 4aa91cb · 4aa91cb
1 parent 05d980f
commit 4aa91cb
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/py-polars/Cargo.toml b/py-polars/Cargo.toml
@@ -28,6 +28,7 @@ numpy = { version = "0.20", default-features = false }
 once_cell = { workspace = true }
 pyo3 = { workspace = true, features = ["abi3-py38", "extension-module", "multiple-pymethods"] }
 pyo3-built = { version = "0.4", optional = true }
+recursive = { workspace = true }
 serde_json = { workspace = true, optional = true }
 smartstring = { workspace = true }
 thiserror = { workspace = true }

diff --git a/py-polars/src/on_startup.rs b/py-polars/src/on_startup.rs
@@ -71,6 +71,13 @@ fn warning_function(msg: &str, warning: PolarsWarning) {
 #[pyfunction]
 pub fn __register_startup_deps() {
     if !registry::is_object_builder_registered() {
+        // Stack frames can get really large in debug mode.
+        #[cfg(debug_assertions)]
+        {
+            recursive::set_minimum_stack_size(1024 * 1024);
+            recursive::set_stack_allocation_size(1024 * 1024 * 16);
+        }
+
         // register object type builder
         let object_builder = Box::new(|name: &str, capacity: usize| {
             Box::new(ObjectChunkedBuilder::<ObjectValue>::new(name, capacity))