MCPcopy Create free account
hub / github.com/apache/arrow / test_IPC

Function test_IPC

python/pyarrow/tests/test_cuda.py:822–832  ·  view source on GitHub ↗
(size)

Source from the content-addressed store, hash-verified

820@cuda_ipc
821@pytest.mark.parametrize("size", [0, 1, 1000])
822def test_IPC(size):
823 import multiprocessing
824 ctx = multiprocessing.get_context('spawn')
825 arr, cbuf = make_random_buffer(size=size, target='device')
826 ipc_handle = cbuf.export_for_ipc()
827 handle_buffer = ipc_handle.serialize()
828 p = ctx.Process(target=other_process_for_test_IPC,
829 args=(handle_buffer, arr))
830 p.start()
831 p.join()
832 assert p.exitcode == 0
833
834
835def test_copy_to():

Callers

nothing calls this directly

Calls 4

make_random_bufferFunction · 0.70
ProcessMethod · 0.45
startMethod · 0.45
joinMethod · 0.45

Tested by

no test coverage detected