CUDA Integration#
CUDA Contexts#
Context(*args, **kwargs)
CUDA driver context.
CUDA Buffers#
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
Device-accessible CPU memory created using cudaHostAlloc.
File interface for zero-copy read from CUDA buffers.
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
A serializable container for a CUDA IPC handle.