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