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.

previous

pyarrow.dataset.get_partition_keys

next

pyarrow.cuda.Context