Merge pull request #260 from madsbk/spill_dicts

Support spilling of device objects in dictionaries
rapidsai · Mar 24, 2020 · 259f36a · 259f36a
2 parents bf6f868 + 0a94220
commit 259f36a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/dask_cuda/is_device_object.py b/dask_cuda/is_device_object.py
@@ -18,6 +18,11 @@ def is_device_object_python_collection(seq):
     return any([is_device_object(s) for s in seq])
 
 
+@is_device_object.register(dict)
+def is_device_object_python_dict(seq):
+    return any([is_device_object(s) for s in seq.items()])
+
+
 @is_device_object.register_lazy("cudf")
 def register_cudf():
     import cudf