pyarrow.cuda.new_host_buffer

pyarrow.cuda.new_host_buffer()

Return buffer with CUDA-accessible memory on CPU host

Parameters:

• size (int) – Specify the number of bytes to be allocated.
• device (int) – Specify GPU device number.

Returns:

dbuf (HostBuffer) – Allocated host buffer