Inheritance diagram for AsyncBroadcastTest:

Public Member Functions
	AsyncBroadcastTest (const std::string &path, int numTensors)

std::shared_ptr< c10d::ProcessGroup::Work >	run (int rootRank, int rootTensor)

Public Member Functions inherited from AsyncInputIsOutputTest
	AsyncInputIsOutputTest (const std::string &path, int numTensors)

void	wait (std::shared_ptr< ProcessGroup::Work > &work)

std::vector< at::Tensor >	getTensors ()

Public Member Functions inherited from AsyncTest
	AsyncTest (const std::string &path)

	AsyncTest (AsyncTest &&other)

::c10d::ProcessGroupGloo &	getProcessGroup ()

void	start (int rank, int size)

Additional Inherited Members
Protected Attributes inherited from AsyncInputIsOutputTest
const int	numTensors_

const int	numDevices_

THCState *	state_

std::vector< at::Tensor >	inputs_

std::vector< CUDAStream >	streams_

Protected Attributes inherited from AsyncTest
std::string	path_

std::unique_ptr<::c10d::ProcessGroupGloo >	pg_

Detailed Description

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

Public Member Functions