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