pyarrow.cuda.read_record_batch

pyarrow.cuda.read_record_batch(buffer, schema, *, DictionaryMemo dictionary_memo=None, pool=None)

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

schemaSchema

The schema for the record batch

dictionary_memoDictionaryMemo, optional

If message contains dictionaries, must pass a populated DictionaryMemo

poolMemoryPool (optional)

Pool to allocate metadata from

Returns

batchRecordBatch

Reconstructed record batch, with device pointers

previous

pyarrow.cuda.serialize_record_batch

next

pyarrow.cuda.read_message