def test_compare_bcm_cm_local_index(self): """Test Block-Cyclic against Cyclic map.""" start = 1 size = 16 grid = 4 block = 1 dimdict = dict(start=start, size=size, proc_grid_size=grid, block_size=block, proc_grid_rank=start) bcm = maps.map_from_dim_dict(dict(list(dimdict.items()) + [('dist_type', 'c')])) cm = maps.map_from_dim_dict(dict(list(dimdict.items()) + [('dist_type', 'c')])) bcm_lis = [bcm.local_from_global_index(e) for e in range(1, 16, 4)] cm_lis = [cm.local_from_global_index(e) for e in range(1, 16, 4)] self.assertSequenceEqual(bcm_lis, cm_lis)
def setUp(self): dimdict = dict(dist_type='c', start=2, size=16, proc_grid_size=4, proc_grid_rank=2) self.m = maps.map_from_dim_dict(dimdict)
def test_compare_bcm_bm_local_index(self): """Test Block-Cyclic against Block map.""" start = 4 size = 16 grid = 4 block = size // grid dimdict = dict(start=start, size=size, proc_grid_size=grid) bcm = maps.map_from_dim_dict(dict(list(dimdict.items()) + [('dist_type', 'c'), ('block_size', block)])) bm = maps.map_from_dim_dict(dict(list(dimdict.items()) + [('dist_type', 'b'), ('stop', size // grid + start)])) bcm_lis = [bcm.local_from_global_index(e) for e in range(4, 8)] bm_lis = [bm.local_from_global_index(e) for e in range(4, 8)] self.assertSequenceEqual(bcm_lis, bm_lis)
def test_compare_bcm_cm_local_index(self): """Test Block-Cyclic against Cyclic map.""" start = 1 size = 16 grid = 4 block = 1 dimdict = dict(start=start, size=size, proc_grid_size=grid, block_size=block, proc_grid_rank=start) bcm = maps.map_from_dim_dict( dict(list(dimdict.items()) + [('dist_type', 'c')])) cm = maps.map_from_dim_dict( dict(list(dimdict.items()) + [('dist_type', 'c')])) bcm_lis = [bcm.local_from_global_index(e) for e in range(1, 16, 4)] cm_lis = [cm.local_from_global_index(e) for e in range(1, 16, 4)] self.assertSequenceEqual(bcm_lis, cm_lis)
def test_compare_bcm_bm_local_index(self): """Test Block-Cyclic against Block map.""" start = 4 size = 16 grid = 4 block = size // grid dimdict = dict(start=start, size=size, proc_grid_size=grid) bcm = maps.map_from_dim_dict( dict( list(dimdict.items()) + [('dist_type', 'c'), ('block_size', block)])) bm = maps.map_from_dim_dict( dict( list(dimdict.items()) + [('dist_type', 'b'), ('stop', size // grid + start)])) bcm_lis = [bcm.local_from_global_index(e) for e in range(4, 8)] bm_lis = [bm.local_from_global_index(e) for e in range(4, 8)] self.assertSequenceEqual(bcm_lis, bm_lis)
def unstructured_index(dd): return list(maps.map_from_dim_dict(dd).global_iter)
def cyclic_index(dd): if ('block_size' not in dd) or (dd['block_size'] == 1): return slice(dd['start'], None, dd['proc_grid_size']) else: return list(maps.map_from_dim_dict(dd).global_iter)
def setUp(self): dimdict = dict(dist_type='b', size=(39-16), start=16, stop=39) self.m = maps.map_from_dim_dict(dimdict)
def setUp(self): size = 20 dimdict = dict(dist_type='n', size=size) self.m = maps.map_from_dim_dict(dimdict)
def setUp(self): dimdict = dict(dist_type='b', size=(39 - 16), start=16, stop=39) self.m = maps.map_from_dim_dict(dimdict)