pub fn set_current_pool(device: &Device, pool: &MemoryPool) -> Result<()>Expand description
Replace the pool used by cudaMallocAsync on device.
pub fn set_current_pool(device: &Device, pool: &MemoryPool) -> Result<()>Replace the pool used by cudaMallocAsync on device.