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