Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/distributed/c10d/
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
DSymmetricMemory.cpp192 TORCH_API bool has_multicast_support( in has_multicast_support() function