From 1f42c7d9d64a87e07fd859217555653c7eece0d6 Mon Sep 17 00:00:00 2001 From: John Kirkham Date: Tue, 26 May 2020 20:54:18 -0700 Subject: [PATCH] Register `dask_cudf` serializers As `dask_cudf` simply relies on the same serialization machinery that is in `cudf` (plus whatever is implemented by `dask_cudf` objects themselves), just reuse the `cudf` serialization import to handle `dask_cudf` object serialization for simplicity. --- distributed/protocol/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/distributed/protocol/__init__.py b/distributed/protocol/__init__.py index 79e77e9703b..be1c498c35c 100644 --- a/distributed/protocol/__init__.py +++ b/distributed/protocol/__init__.py @@ -105,6 +105,10 @@ def _register_rmm(): @cuda_deserialize.register_lazy("cudf") @dask_serialize.register_lazy("cudf") @dask_deserialize.register_lazy("cudf") +@cuda_serialize.register_lazy("dask_cudf") +@cuda_deserialize.register_lazy("dask_cudf") +@dask_serialize.register_lazy("dask_cudf") +@dask_deserialize.register_lazy("dask_cudf") def _register_cudf(): from cudf.comm import serialize