Skip to main content

Module mempool

Module mempool 

Source
Expand description

Stream-ordered memory pools (Runtime API, CUDA 11.2+).

Mirrors [baracuda_driver::mempool] — a pool is a device-backed allocator with a configurable release threshold, accessed via cudaMallocFromPoolAsync and returned with cudaFreeAsync. Each device exposes a default pool via default_pool.

Structs§

MemoryPool
A memory pool. Owned pools are destroyed on last-clone drop; borrowed pools (returned by default_pool / current_pool) are not.

Enums§

AccessFlags
Access rights granted to a device for a pool’s allocations.

Functions§

current_pool
Return the pool currently used by cudaMallocAsync on device.
default_pool
Return the device’s default memory pool (borrowed — not destroyed on drop).
set_current_pool
Replace the pool used by cudaMallocAsync on device.