Function rcudnn_sys::cudaDeviceSetMemPool [−][src]
pub unsafe extern "C" fn cudaDeviceSetMemPool(
device: c_int,
memPool: cudaMemPool_t
) -> cudaError_t
Expand description
\brief Sets the current memory pool of a device
The memory pool must be local to the specified device. Unless a mempool is specified in the ::cudaMallocAsync call, ::cudaMallocAsync allocates from the current mempool of the provided stream’s device. By default, a device’s current memory pool is its default memory pool.
\note Use ::cudaMallocFromPoolAsync to specify asynchronous allocations from a device different than the one the stream runs on.
\returns ::cudaSuccess, ::cudaErrorInvalidValue ::cudaErrorInvalidDevice ::cudaErrorNotSupported \notefnerr \note_callback
\sa ::cuDeviceSetDefaultMemPool, ::cudaDeviceGetMemPool, ::cudaDeviceGetDefaultMemPool, ::cudaMemPoolCreate, ::cudaMemPoolDestroy, ::cudaMallocFromPoolAsync