Function rcudnn::cudaDeviceSetMemPool [−][src]
pub unsafe extern "C" fn cudaDeviceSetMemPool(
device: i32,
memPool: *mut CUmemPoolHandle_st
) -> cudaError
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