From 86ede13f88556f30ba581b07a28bba259d87e2cf Mon Sep 17 00:00:00 2001 From: Peter Andreas Entschev Date: Fri, 26 Jul 2024 04:40:49 -0700 Subject: [PATCH] Test cuDF spilling with `enable_cudf_spill` --- dask_cuda/tests/test_spill.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/dask_cuda/tests/test_spill.py b/dask_cuda/tests/test_spill.py index f8df7e04f..31220fb43 100644 --- a/dask_cuda/tests/test_spill.py +++ b/dask_cuda/tests/test_spill.py @@ -243,8 +243,9 @@ async def test_cupy_cluster_device_spill(params): }, ], ) +@pytest.mark.parametrize("enable_cudf_spill", [True, False]) @gen_test(timeout=30) -async def test_cudf_cluster_device_spill(params): +async def test_cudf_cluster_device_spill(params, enable_cudf_spill): cudf = pytest.importorskip("cudf") with dask.config.set( @@ -266,6 +267,7 @@ async def test_cudf_cluster_device_spill(params): device_memory_limit=params["device_memory_limit"], memory_limit=params["memory_limit"], worker_class=IncreasedCloseTimeoutNanny, + enable_cudf_spill=enable_cudf_spill, ) as cluster: async with Client(cluster, asynchronous=True) as client: