pyarrow.cuda.serialize_record_batch

pyarrow.cuda.serialize_record_batch(batch, ctx)

Write record batch message to GPU device memory

Parameters
batchRecordBatch

Record batch to write

ctxContext

CUDA Context to allocate device memory from

Returns
dbufCudaBuffer

device buffer which contains the record batch message