pyarrow.cuda.read_record_batch

pyarrow.cuda.read_record_batch()

Construct RecordBatch referencing IPC message located on CUDA device.

While the metadata is copied to host memory for deserialization, the record batch data remains on the device.

Parameters
  • buffer – Device buffer containing the complete IPC message

  • schema (Schema) – The schema for the record batch

  • dictionary_memo (DictionaryMemo, optional) – If message contains dictionaries, must pass a populated DictionaryMemo

  • pool (MemoryPool (optional)) – Pool to allocate metadata from

Returns

batch (RecordBatch) – Reconstructed record batch, with device pointers