pyarrow.cuda.serialize_record_batch

pyarrow.cuda.serialize_record_batch()

Write record batch message to GPU device memory

Parameters:

• batch (RecordBatch) – Record batch to write
• ctx (Context) – CUDA Context to allocate device memory from

Returns:

dbuf (CudaBuffer) – device buffer which contains the record batch message