CUDA Integration

CUDA Contexts

Context

CUDA driver context.

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.