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