| /external/pytorch/aten/src/ATen/xpu/detail/ |
| D | XPUHooks.cpp | 70 Allocator* XPUHooks::getPinnedMemoryAllocator() const { in getPinnedMemoryAllocator() function in at::xpu::detail::XPUHooks
71 return at::xpu::getPinnedMemoryAllocator(); in getPinnedMemoryAllocator()
|
| D | XPUHooks.h | 21 Allocator* getPinnedMemoryAllocator() const override;
|
| /external/pytorch/aten/src/ATen/native/ |
| D | Memory.cpp | 62 at::globalContext().getPinnedMemoryAllocator(device.value().type()): in _pin_memory()
63 at::globalContext().getPinnedMemoryAllocator(); in _pin_memory()
|
| /external/pytorch/aten/src/ATen/xpu/ |
| D | PinnedMemoryAllocator.h | 8 inline TORCH_XPU_API at::Allocator* getPinnedMemoryAllocator() { in getPinnedMemoryAllocator() function
|
| /external/pytorch/aten/src/ATen/cuda/ |
| D | PinnedMemoryAllocator.h | 8 inline TORCH_CUDA_CPP_API at::Allocator* getPinnedMemoryAllocator() { in getPinnedMemoryAllocator() function
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | MiscUtils.h | 22 auto* allocator = cuda::getPinnedMemoryAllocator(); in pin_memory()
|
| /external/pytorch/aten/src/ATen/detail/ |
| D | AcceleratorHooksInterface.h | 49 virtual Allocator* getPinnedMemoryAllocator() const { in getPinnedMemoryAllocator() function
|
| D | PrivateUse1HooksInterface.h | 31 virtual Allocator* getPinnedMemoryAllocator() const override { in getPinnedMemoryAllocator() function
|
| D | HIPHooksInterface.h | 50 Allocator* getPinnedMemoryAllocator() const override { in getPinnedMemoryAllocator() function
|
| D | XPUHooksInterface.h | 61 Allocator* getPinnedMemoryAllocator() const override { in getPinnedMemoryAllocator() function
|
| D | MTIAHooksInterface.h | 98 Allocator* getPinnedMemoryAllocator() const override { in getPinnedMemoryAllocator() function
|
| D | MPSHooksInterface.h | 100 Allocator* getPinnedMemoryAllocator() const override { in getPinnedMemoryAllocator() function
|
| D | CUDAHooksInterface.h | 124 Allocator* getPinnedMemoryAllocator() const override { in getPinnedMemoryAllocator() function
|
| /external/pytorch/aten/src/ATen/mps/ |
| D | MPSHooks.h | 38 Allocator* getPinnedMemoryAllocator() const override;
|
| D | MPSHooks.mm | 142 Allocator* MPSHooks::getPinnedMemoryAllocator() const {
|
| /external/pytorch/aten/src/ATen/cuda/detail/ |
| D | CUDAHooks.h | 36 Allocator* getPinnedMemoryAllocator() const override;
|
| D | CUDAHooks.cpp | 252 Allocator* CUDAHooks::getPinnedMemoryAllocator() const { in getPinnedMemoryAllocator() function in at::cuda::detail::CUDAHooks
253 return at::cuda::getPinnedMemoryAllocator(); in getPinnedMemoryAllocator()
|
| /external/pytorch/aten/src/ATen/ |
| D | EmptyTensor.cpp | 20 return at::detail::getCUDAHooks().getPinnedMemoryAllocator(); in GetCPUAllocatorMaybePinned()
22 return at::detail::getXPUHooks().getPinnedMemoryAllocator(); in GetCPUAllocatorMaybePinned()
24 return at::detail::getPrivateUse1Hooks().getPinnedMemoryAllocator(); in GetCPUAllocatorMaybePinned()
|
| /external/pytorch/test/cpp_extensions/ |
| D | open_registration_extension.cpp | 599 c10::Allocator* getPinnedMemoryAllocator() const override { in getPinnedMemoryAllocator() function
|
| /external/pytorch/torch/csrc/distributed/c10d/ |
| D | ProcessGroupGloo.cpp | 314 auto* allocator = at::detail::getCUDAHooks().getPinnedMemoryAllocator(); in pinnedLike()
|