Try specifying a different allocator

rapidsai · Oct 9, 2024 · 635b5e0 · 635b5e0
1 parent aa37dcf
commit 635b5e0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ci/test_wheel_cudf.sh b/ci/test_wheel_cudf.sh
@@ -35,4 +35,5 @@ RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${RESULTS_DIR}/test-results"}/
 mkdir -p "${RAPIDS_TESTS_DIR}"
 
 rapids-logger "pytest cudf"
+export ARROW_DEFAULT_MEMORY_POOL="mimalloc"
 python -c "import cupy; import cudf"