Test cuDF spilling with enable_cudf_spill

rapidsai · Jul 26, 2024 · 86ede13 · 86ede13
1 parent 8113d72
commit 86ede13
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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: