Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
AllreduceNCCLTest Class Reference
Inheritance diagram for AllreduceNCCLTest:
NCCLTest NCCLTestBase

Public Member Functions

 AllreduceNCCLTest (const std::string &path, int worldSize)
 
std::shared_ptr< c10d::ProcessGroup::Work > run ()
 
- Public Member Functions inherited from NCCLTest
 NCCLTest (const std::string &path, int worldSize)
 
void wait (std::shared_ptr< ProcessGroup::Work > &work)
 
std::vector< at::TensorgetTensors ()
 
std::vector< std::vector< at::Tensor > > getOutputTensors ()
 
int numDevices () const
 
- Public Member Functions inherited from NCCLTestBase
 NCCLTestBase (const std::string &path)
 
 NCCLTestBase (NCCLTestBase &&other)
 
::c10d::ProcessGroupNCCL & getProcessGroup ()
 
void initialize (int rank, int size)
 

Additional Inherited Members

- Protected Attributes inherited from NCCLTest
const int numDevices_
 
THCState * state_
 
int worldSize_
 
std::vector< at::Tensorinputs_
 
std::vector< std::vector< at::Tensor > > outputs_
 
std::vector< CUDAStream > streams_
 
- Protected Attributes inherited from NCCLTestBase
std::string path_
 
std::unique_ptr<::c10d::ProcessGroupNCCL > pg_
 

Detailed Description

Definition at line 128 of file ProcessGroupNCCLTest.cpp.


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