cuda.bindings._v2.driver.device_set_mem_pool#

cuda.bindings._v2.driver.device_set_mem_pool(int dev, intptr_t pool)#

Sets the current memory pool of a device.

The memory pool must be local to the specified device. cuMemAllocAsync allocates from the current mempool of the provided stream’s device. By default, a device’s current memory pool is its default memory pool.

Parameters:

dev (int) – Device to set the current memory pool for.
pool (intptr_t) – Memory pool to use as the device’s current memory pool.

Note

Use cuMemAllocFromPoolAsync to specify asynchronous allocations from a device different than the one the stream runs on.