9408ebcb3f9ef1420640febf67b48c48fdcd2dd8,distributed/protocol/cupy.py,,serialize_cupy_ndarray,#Any#,40

Before Change

        x = cupy.array(x, copy=True)

    header = x.__cuda_array_interface__.copy()
    return header, [x]


@cuda_deserialize.register(cupy.ndarray)

After Change

    header = x.__cuda_array_interface__.copy()
    header["strides"] = tuple(x.strides)
    frames = [
        cupy.ndarray(
            shape=(x.nbytes,), dtype=cupy.dtype("u1"), memptr=x.data, strides=(1,)
        )
    ]

    return header, frames

Instances