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