Module test_cupy

Functions

▷ def test_serialize_cupy(shape, dtype, order, serializers) 4 decorators
@pytest.mark.parametrize("shape", [(0,), (5,), (4, 6), (10, 11), (2, 3, 5)])
@pytest.mark.parametrize("dtype", ["u1", "u4", "u8", "f4"])
@pytest.mark.parametrize("order", ["C", "F"])
@pytest.mark.parametrize("serializers", [("cuda",), ("dask",), ("pickle",)])
def test_serialize_cupy(
shape,
dtype,
order,
serializers,
)
source link
▷ def test_serialize_cupy_from_numba(dtype) @pytest.mark.parametrize("dtype", ["u1", "u4", "u8", "f4"])
@pytest.mark.parametrize("dtype", ["u1", "u4", "u8", "f4"])
def test_serialize_cupy_from_numba(dtype)
source link
▷ def test_serialize_cupy_from_rmm(size) @pytest.mark.parametrize("size", [0, 3, 10])
@pytest.mark.parametrize("size", [0, 3, 10])
def test_serialize_cupy_from_rmm(size)
source link
▷ def test_serialize_cupy_sparse(sparse_name, dtype, serializer) 3 decorators
@pytest.mark.parametrize( "sparse_name", ["coo_matrix", "csc_matrix", "csr_matrix", "dia_matrix"] )
@pytest.mark.parametrize( "dtype", [numpy.dtype("<f4"), numpy.dtype(">f4"), numpy.dtype("<f8"), numpy.dtype(">f8")], )
@pytest.mark.parametrize("serializer", ["cuda", "dask", "pickle"])
def test_serialize_cupy_sparse(
sparse_name,
dtype,
serializer,
)
source link