pyarrow.cuda.read_record_batch#
- pyarrow.cuda.read_record_batch(buffer, schema, *, DictionaryMemo dictionary_memo=None, pool=None)#
构建引用位于 CUDA 设备上的 IPC 消息的 RecordBatch。
虽然元数据会被复制到主机内存以进行反序列化，但记录批处理（record batch）数据仍保留在设备上。
- 参数:
- buffer
包含完整 IPC 消息的设备缓冲区
- schema
Schema 记录批处理的 schema（模式）
- dictionary_memo
DictionaryMemo，可选 如果消息包含字典，则必须传入一个已填充的 DictionaryMemo
- pool
MemoryPool（可选） 用于分配元数据的内存池
- 返回：
- batch
RecordBatch 重建的记录批处理，包含设备指针
- batch