| 417 | } |
| 418 | |
| 419 | void DeviceManager::setMemoryManagerPinned( |
| 420 | std::unique_ptr<MemoryManagerBase> newMgr) { |
| 421 | std::lock_guard<std::mutex> l(mutex); |
| 422 | // It's possible we're setting a pinned memory manager and the default |
| 423 | // memory manager still hasn't been initialized, so initialize it anyways so |
| 424 | // we don't inadvertently reset to it when we first call |
| 425 | // pinnedMemoryManager() |
| 426 | pinnedMemoryManager(); |
| 427 | // Calls shutdown() on the existing memory manager. |
| 428 | if (pinnedMemManager) { pinnedMemManager->shutdownAllocator(); } |
| 429 | // Set the backend memory manager for this new manager to register native |
| 430 | // functions correctly. |
| 431 | pinnedMemManager = std::move(newMgr); |
| 432 | std::unique_ptr<cuda::AllocatorPinned> deviceMemoryManager( |
| 433 | new AllocatorPinned()); |
| 434 | pinnedMemManager->setAllocator(std::move(deviceMemoryManager)); |
| 435 | pinnedMemManager->initialize(); |
| 436 | } |
| 437 | |
| 438 | void DeviceManager::resetMemoryManagerPinned() { |
| 439 | // Replace with default memory manager |
no test coverage detected