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

Function put

python/pyarrow/tests/test_cuda.py:469–472  ·  view source on GitHub ↗
(*args, **kwargs)

Source from the content-addressed store, hash-verified

467 dbuf = dest_ctx.new_buffer(size)
468
469 def put(*args, **kwargs):
470 dbuf.copy_from_device(buf, *args, **kwargs)
471 rbuf = dbuf.copy_to_host()
472 return np.frombuffer(rbuf, dtype=np.uint8).tolist()
473 assert put() == lst
474 if size > 4:
475 assert put(position=size//4) == lst[:size//4]+lst[:-size//4]

Callers 2

test_copy_from_deviceFunction · 0.85
test_copy_from_hostFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected