Searched refs:dev_to_access (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | PeerToPeerAccess.cpp | 36 bool get_p2p_access(int dev, int dev_to_access) { in get_p2p_access() argument
41 TORCH_CHECK(dev_to_access >= 0 || dev_to_access < num_devices_, in get_p2p_access()
42 dev_to_access, " is not a device"); in get_p2p_access()
45 auto &cache = p2pAccessEnabled_[dev * num_devices_ + dev_to_access]; in get_p2p_access()
52 C10_CUDA_CHECK(cudaDeviceCanAccessPeer(&result, dev, dev_to_access)); in get_p2p_access()
55 CUDACachingAllocator::enablePeerAccess(dev, dev_to_access); in get_p2p_access()
|
| /aosp_15_r20/external/pytorch/c10/cuda/ |
| H A D | CUDACachingAllocator.h | 264 c10::DeviceIndex dev_to_access) = 0;
447 c10::DeviceIndex dev_to_access) { in enablePeerAccess() argument
448 return get()->enablePeerAccess(dev, dev_to_access); in enablePeerAccess()
|
| H A D | CUDACachingAllocator.cpp | 2086 void addPeerAccess(c10::DeviceIndex dev_to_access) { in addPeerAccess() argument
2091 dev_to_access) != devices_with_peer_access_.end()) { in addPeerAccess()
2094 devices_with_peer_access_.push_back(dev_to_access); in addPeerAccess()
2096 es->addPeer(dev_to_access); in addPeerAccess()
3559 void enablePeerAccess(c10::DeviceIndex dev, c10::DeviceIndex dev_to_access) in enablePeerAccess() argument
3562 cudaError_t err = cudaDeviceEnablePeerAccess(dev_to_access, 0); in enablePeerAccess()
3569 device_allocator[dev_to_access]->addPeerAccess(dev); in enablePeerAccess()
3572 if (entry.second.device_ == dev_to_access && in enablePeerAccess()
|
| H A D | CUDAMallocAsyncAllocator.cpp | 856 void enablePeerAccess(c10::DeviceIndex dev, c10::DeviceIndex dev_to_access) in enablePeerAccess()
864 C10_CUDA_CHECK(cudaDeviceGetDefaultMemPool(&mempool, dev_to_access)); in enablePeerAccess()
|
| /aosp_15_r20/external/pytorch/torch/csrc/cuda/ |
| H A D | CUDAPluggableAllocator.cpp | 333 c10::DeviceIndex dev_to_access) { in enablePeerAccess() argument
335 cudaError_t err = cudaDeviceEnablePeerAccess(dev_to_access, 0); in enablePeerAccess()
|
| H A D | CUDAPluggableAllocator.h | 151 void enablePeerAccess(c10::DeviceIndex dev, c10::DeviceIndex dev_to_access)
|