Mark CUDA frames as neither readonly no writeable

This special case of `None` will bypass both cases.

Author: jakirkham

distributed/protocol/cuda.py

@@ -18,8 +18,6 @@ def cuda_dumps(x):
     header, frames = dumps(x)
     header["type-serialized"] = pickle.dumps(type(x))
     header["serializer"] = "cuda"
-    header["compression"] = (False,) * len(frames)  # no compression for gpu data
-    header["writeable"] = (False,) * len(frames)  # no need to enforce writeable frames
     return header, frames
 
 

distributed/protocol/cupy.py

@@ -23,6 +23,7 @@ def cuda_serialize_cupy_ndarray(x):
     header = x.__cuda_array_interface__.copy()
     header["strides"] = tuple(x.strides)
     header["lengths"] = [x.nbytes]
+    header["compression"] = (False,) * len(frames)
     frames = [
         cupy.ndarray(
             shape=(x.nbytes,), dtype=cupy.dtype("u1"), memptr=x.data, strides=(1,)
@@ -47,6 +48,7 @@ def cuda_deserialize_cupy_ndarray(header, frames):
 @dask_serialize.register(cupy.ndarray)
 def dask_serialize_cupy_ndarray(x):
     header, frames = cuda_serialize_cupy_ndarray(x)
+    header["writeable"] = (None,) * len(frames)
     frames = [memoryview(cupy.asnumpy(f)) for f in frames]
     return header, frames
 

distributed/protocol/numba.py

@@ -51,6 +51,7 @@ def cuda_deserialize_numba_ndarray(header, frames):
 @dask_serialize.register(numba.cuda.devicearray.DeviceNDArray)
 def dask_serialize_numba_ndarray(x):
     header, frames = cuda_serialize_numba_ndarray(x)
+    header["writeable"] = (None,) * len(frames)
     frames = [memoryview(f.copy_to_host()) for f in frames]
     return header, frames
 

distributed/protocol/rmm.py

@@ -30,6 +30,7 @@ def cuda_deserialize_rmm_device_buffer(header, frames):
     @dask_serialize.register(rmm.DeviceBuffer)
     def dask_serialize_rmm_device_buffer(x):
         header, frames = cuda_serialize_rmm_device_buffer(x)
+        header["writeable"] = (None,) * len(frames)
         frames = [numba.cuda.as_cuda_array(f).copy_to_host().data for f in frames]
         return header, frames