Searched defs:enablePeerAccess (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/csrc/cuda/ | ||
| D | CUDAPluggableAllocator.cpp | 331 void CUDAPluggableAllocator::enablePeerAccess( in enablePeerAccess() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator |
| /external/pytorch/c10/cuda/ | ||
| D | CUDAMallocAsyncAllocator.cpp | 856 void enablePeerAccess(c10::DeviceIndex dev, c10::DeviceIndex dev_to_access) in enablePeerAccess() function |
| D | CUDACachingAllocator.h | 445 inline void enablePeerAccess( in enablePeerAccess() function |
| D | CUDACachingAllocator.cpp | 3559 void enablePeerAccess(c10::DeviceIndex dev, c10::DeviceIndex dev_to_access) in enablePeerAccess() function in c10::cuda::CUDACachingAllocator::Native::NativeCachingAllocator |