Module test_rmm

distributed.protocol.tests.test_rmm

Functions

▷ def test_serialize_rmm_device_buffer(size, serializers) @pytest.mark.parametrize("size", [0, 3, 10]) @pytest.mark.parametrize("serializers", [("cuda",), ("dask",), ("pickle",)])
@pytest.mark.parametrize("size", [0, 3, 10])
@pytest.mark.parametrize("serializers", [("cuda",), ("dask",), ("pickle",)])
def test_serialize_rmm_device_buffer(
size,
serializers,
)
source link