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

Public Member Functions

 NCCLTestBase (const std::string &path)
 
 NCCLTestBase (NCCLTestBase &&other)
 
::c10d::ProcessGroupNCCL & getProcessGroup ()
 
void initialize (int rank, int size)
 

Protected Attributes

std::string path_
 
std::unique_ptr<::c10d::ProcessGroupNCCL > pg_
 

Detailed Description

Definition at line 17 of file ProcessGroupNCCLTest.cpp.


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