CUDA Integration

CUDA Contexts

Context(*args, **kwargs)

CUDA driver context.

CUDA Buffers

CudaBuffer()

An Arrow buffer with data located in a GPU device.

new_host_buffer(int64_t size, int device=0)

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(batch, ctx)

Write record batch message to GPU device memory

read_record_batch(buffer, schema, *, …[, pool])

Construct RecordBatch referencing IPC message located on CUDA device.

read_message(source[, pool])

Read Arrow IPC message located on GPU device

IpcMemHandle

A serializable container for a CUDA IPC handle.