Home
last modified time | relevance | path

Searched refs:ipcMemHandle_to_devptr (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDACachingAllocator.cpp3571 for (auto& entry : ipcMemHandle_to_devptr) { in enablePeerAccess()
3686 ska::flat_hash_map<std::string, MemHandleCacheEntry> ipcMemHandle_to_devptr; member in c10::cuda::CUDACachingAllocator::Native::NativeCachingAllocator
3690 auto iter = ipcMemHandle_to_devptr.find(handle); in getIpcDevPtr()
3691 if (iter != ipcMemHandle_to_devptr.end()) { in getIpcDevPtr()
3701 auto inserted = ipcMemHandle_to_devptr.insert( in getIpcDevPtr()
3709 auto it = ipcMemHandle_to_devptr.find(handle); in getIpcDevPtr()
3710 TORCH_INTERNAL_ASSERT(it != ipcMemHandle_to_devptr.end()); in getIpcDevPtr()
3712 ipcMemHandle_to_devptr.erase(it); in getIpcDevPtr()