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