Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
torch::cuda::nccl::detail::NcclCommList Struct Reference

Public Member Functions

 NcclCommList (const std::vector< int > &devices)
 
 NcclCommList (NcclCommList &&foo)=default
 
ArrayRef< ncclComm_t > ref () const
 

Data Fields

std::unique_ptr< ncclComm_t[]> comms
 
int ndevices
 

Detailed Description

Definition at line 31 of file nccl.cpp.


The documentation for this struct was generated from the following file: