From 635b5e0eceeb63460bd28c1d3655b6bd83a49cc1 Mon Sep 17 00:00:00 2001 From: Vyas Ramasubramani Date: Wed, 9 Oct 2024 20:43:40 +0000 Subject: [PATCH] Try specifying a different allocator --- ci/test_wheel_cudf.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/ci/test_wheel_cudf.sh b/ci/test_wheel_cudf.sh index b03e6b1a32c..19d55292ff5 100755 --- 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"