Module cupy
Efficient serialization GPU arrays.
source linkFunctions
@cuda_deserialize.register(cupy.ndarray)
def cuda_deserialize_cupy_ndarray(
header,
frames,
)
@cuda_serialize.register(cupy.ndarray)
def cuda_serialize_cupy_ndarray(x)
@dask_deserialize.register(cupy.ndarray)
def dask_deserialize_cupy_ndarray(
header,
frames,
)
@dask_serialize.register(cupy.ndarray)
def dask_serialize_cupy_ndarray(x)