Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DCopy.cu175 bool p2p_enabled) { in copy_device_to_device()
257 static bool copy_requires_temporaries(TensorIterator& iter, bool p2p_enabled) { in copy_requires_temporaries()
295 bool p2p_enabled = maybe_enable_p2p_access(dst_device, src_device); in copy_kernel_cuda() local
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DCUDAPluggableAllocator.cpp351 bool p2p_enabled) { in memcpyAsync()
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDACachingAllocator.h440 bool p2p_enabled) { in memcpyAsync()
H A DCUDACachingAllocator.cpp3586 bool p2p_enabled) override { in memcpyAsync()