ae74b5ea18fac7e272c9f25b5d9f2775956aa943,distributed/protocol/tests/test_cupy.py,,test_serialize_cupy_sparse,#Any#Any#Any#,82

Before Change

    a_host = numpy.array([[0, 1, 0], [2, 0, 3], [0, 4, 0]], dtype=dtype)
    a = cupy.asarray(a_host)

    anz = a.nonzero()
    acoo = cupy_sparse.coo_matrix((a[anz], anz))
    asp = sparse_type(acoo)

    header, frames = serialize(asp, serializers=[serializer])
    asp2 = deserialize(header, frames)

After Change

    a_host = numpy.array([[0, 1, 0], [2, 0, 3], [0, 4, 0]], dtype=dtype)
    asp_host = scipy_sparse_type(a_host)
    if sparse_name == "dia_matrix":
        // CuPy `dia_matrix` cannot be created from SciPy one
        // xref: https://github.com/cupy/cupy/issues/3158
        asp_dev = cupy_sparse_type(
            (asp_host.data, asp_host.offsets),
            shape=asp_host.shape,
            dtype=asp_host.dtype,
        )
    else:
        asp_dev = cupy_sparse_type(asp_host)

    header, frames = serialize(asp_dev, serializers=[serializer])
    a2sp_dev = deserialize(header, frames)

    a2sp_host = a2sp_dev.get()

Instances