pyarrow.cuda.serialize_record_batch

pyarrow.cuda.serialize_record_batch(batch, ctx)

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