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
  • size (int) – Specify the number of bytes to be allocated.

  • device (int) – Specify GPU device number.

Returns

dbuf (HostBuffer) – Allocated host buffer