Home
last modified time | relevance | path

Searched defs:has_multicast_support (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DCUDASymmetricMemory.cu350 bool CUDASymmetricMemory::has_multicast_support() { in has_multicast_support() function in c10d::symmetric_memory::CUDASymmetricMemory
587 bool has_multicast_support; member
801 bool CUDASymmetricMemoryAllocator::has_multicast_support(int device_idx) { in has_multicast_support() function in c10d::symmetric_memory::CUDASymmetricMemoryAllocator
H A DSymmetricMemory.cpp192 TORCH_API bool has_multicast_support( in has_multicast_support() function