Caffe2 - Python API
A deep learning, cross platform ML framework
List of all members
test_thd_distributed.TestMPI Class Reference
Inheritance diagram for test_thd_distributed.TestMPI:
common_utils.TestCase test_thd_distributed._DistTestBase expecttest.TestCase

Additional Inherited Members

- 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_thd_distributed._DistTestBase
def test_get_rank (self)
 
def test_send_recv (self)
 
def test_send_recv_any_source (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_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_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_scatter (self)
 
def test_scatter_group (self)
 
def test_gather (self)
 
def test_gather_group (self)
 
def test_all_gather (self)
 
def test_all_gather_cuda (self)
 
def test_all_gather_group (self)
 
def test_barrier (self)
 
def test_barrier_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_DistributedDataParallel (self)
 
def test_DistributedDataParallelCPU (self)
 
- 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 1165 of file test_thd_distributed.py.


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