CUDA Integration¶
CUDA Buffers¶
CudaBuffer |
An Arrow buffer with data located in a GPU device. |
new_host_buffer |
Return buffer with CUDA-accessible memory on CPU host |
HostBuffer |
Device-accessible CPU memory created using cudaHostAlloc. |
BufferReader |
File interface for zero-copy read from CUDA buffers. |
BufferWriter |
File interface for writing to CUDA buffers. |
Serialization and IPC¶
serialize_record_batch |
Write record batch message to GPU device memory |
read_record_batch |
Construct RecordBatch referencing IPC message located on CUDA device. |
read_message |
Read Arrow IPC message located on GPU device |
IpcMemHandle |
A serializable container for a CUDA IPC handle. |