Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dnccl.h78 struct AutoNcclGroup { struct
82 ncclComm_t comm_;
83 bool comm_nonblocking_;
H A Dnccl.cpp417 AutoNcclGroup::AutoNcclGroup() { in AutoNcclGroup() function in torch::cuda::nccl::AutoNcclGroup
429 AutoNcclGroup::AutoNcclGroup(ncclComm_t comm, bool comm_nonblocking) { in AutoNcclGroup() function in torch::cuda::nccl::AutoNcclGroup