test_distributed.TestDistBackend Class Reference

Inheritance diagram for test_distributed.TestDistBackend:

Public Member Functions
def	setUpClass (cls)
def	setUp (self)
def	tearDown (self)
Public Member Functions inherited from common_utils.TestCase
def	__init__ (self, method_name='runTest')
def	assertLeaksNoCudaTensors (self, name=None)
def	wrap_with_cuda_memory_check (self, method)
def	setUp (self)
def	assertTensorsSlowEqual (self, x, y, prec=None, message='')
def	genSparseTensor (self, size, sparse_dim, nnz, is_uncoalesced, device='cpu')
def	safeToDense (self, t)
def	safeCoalesce (self, t)
def	assertEqual (self, x, y, prec=None, message='', allow_inf=False)
def	assertAlmostEqual (self, x, y, places=None, msg=None, delta=None, allow_inf=None)
def	assertNotEqual (self, x, y, prec=None, message='')
def	assertObjectIn (self, obj, iterable)
def	assertExpectedRaises (self, exc_type, callable, args, kwargs)
def	assertWarns (self, callable, msg='')
def	assertWarnsRegex (self, callable, regex, msg='')
def	assertExpected (self, s, subname=None)
Public Member Functions inherited from expecttest.TestCase
def	assertExpectedInline (self, actual, expect, skip=0)
Public Member Functions inherited from test_distributed._DistTestBase
def	test_get_rank (self)
def	test_get_backend (self)
def	test_Backend_enum_class (self)
def	test_destroy_group (self)
def	test_get_rank_size_group (self)
def	test_destroy_full_group (self)
def	test_get_rank_size_full_group (self)
def	test_barrier_timeout_global (self)
def	test_barrier_timeout_group (self)
def	test_barrier_timeout_full_group (self)
def	test_send_recv (self)
def	test_send_recv_any_source (self)
def	test_send_recv_with_tag (self)
def	test_isend (self)
def	test_irecv (self)
def	test_broadcast (self)
def	test_broadcast_cuda (self)
def	test_broadcast_group (self)
def	test_broadcast_full_group (self)
def	test_reduce_sum (self)
def	test_reduce_sum_cuda (self)
def	test_reduce_product (self)
def	test_reduce_min (self)
def	test_reduce_max (self)
def	test_reduce_group_sum (self)
def	test_reduce_group_product (self)
def	test_reduce_group_min (self)
def	test_reduce_group_max (self)
def	test_reduce_full_group_sum (self)
def	test_reduce_full_group_product (self)
def	test_reduce_full_group_min (self)
def	test_reduce_full_group_max (self)
def	test_all_reduce_sum (self)
def	test_all_reduce_sum_cuda (self)
def	test_all_reduce_product (self)
def	test_all_reduce_min (self)
def	test_all_reduce_max (self)
def	test_all_reduce_group_sum (self)
def	test_all_reduce_group_product (self)
def	test_all_reduce_group_min (self)
def	test_all_reduce_group_max (self)
def	test_all_reduce_full_group_sum (self)
def	test_all_reduce_full_group_product (self)
def	test_all_reduce_full_group_min (self)
def	test_all_reduce_full_group_max (self)
def	test_scatter (self)
def	test_scatter_group (self)
def	test_scatter_full_group (self)
def	test_gather (self)
def	test_gather_group (self)
def	test_gather_full_group (self)
def	test_all_gather (self)
def	test_all_gather_cuda (self)
def	test_all_gather_group (self)
def	test_all_gather_full_group (self)
def	test_barrier_cuda (self)
def	test_barrier_group_cuda (self)
def	test_barrier_full_group_cuda (self)
def	test_barrier (self)
def	test_barrier_group (self)
def	test_barrier_full_group (self)
def	test_broadcast_multigpu (self)
def	test_all_reduce_multigpu (self)
def	test_reduce_multigpu (self)
def	test_all_gather_multigpu (self)
def	test_DistributedDataParallelCPU (self)
def	test_DistributedDataParallel (self)
def	test_DistributedDataParallel_SyncBatchNorm (self)

Static Public Member Functions
def	manager_join (fn)

Public Attributes
	rank
	processes

Static Public Attributes
int	MANAGER_PROCESS_RANK = -1
Static Public Attributes inherited from common_utils.TestCase
int	precision = 1
	maxDiff = None
	assertRegex = unittest.TestCase.assertRegexpMatches
	assertRaisesRegex = unittest.TestCase.assertRaisesRegexp
Static Public Attributes inherited from expecttest.TestCase
bool	longMessage = True

Detailed Description

Definition at line 1439 of file test_distributed.py.

---

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

• test/test_distributed.py