Home
last modified time | relevance | path

Searched defs:getIpcDevPtr (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DCUDAPluggableAllocator.cpp251 std::shared_ptr<void> CUDAPluggableAllocator::getIpcDevPtr(std::string handle) { in getIpcDevPtr() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAMallocAsyncAllocator.cpp617 std::shared_ptr<void> getIpcDevPtr(std::string handle) override { in getIpcDevPtr() function
H A DCUDACachingAllocator.h421 inline std::shared_ptr<void> getIpcDevPtr(std::string handle) { in getIpcDevPtr() function
H A DCUDACachingAllocator.cpp3687 std::shared_ptr<void> getIpcDevPtr(std::string handle) override { in getIpcDevPtr() function in c10::cuda::CUDACachingAllocator::Native::NativeCachingAllocator