[ Web Proxy ]
URL:
Viewing: https://arrow.apache.org/docs/python/generated/../../cpp/../python/integration/../api/cuda.html [Back]  [Original]

CUDA Integration — Apache Arrow v25.0.1
Back to top
Search the docs ...:

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_tsize,intdevice=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.


Web Proxy Viewer  |  New URL  |  Original Page