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: