pyarrow.cuda.new_host_buffer

pyarrow.cuda.new_host_buffer(int64_t size, int device=0)

Return buffer with CUDA-accessible memory on CPU host

Parameters
sizeint

Specify the number of bytes to be allocated.

deviceint

Specify GPU device number.

Returns
dbufHostBuffer

Allocated host buffer