Example #1
0
 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)
Example #2
0
 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)
Example #3
0
    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)
Example #4
0
 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)
Example #5
0
    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)
Example #6
0
 def unstructured_index(dd):
     return list(maps.map_from_dim_dict(dd).global_iter)
Example #7
0
 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)
Example #8
0
 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)
Example #9
0
 def setUp(self):
     dimdict = dict(dist_type='b', size=(39-16), start=16, stop=39)
     self.m = maps.map_from_dim_dict(dimdict)
Example #10
0
 def setUp(self):
     size = 20
     dimdict = dict(dist_type='n', size=size)
     self.m = maps.map_from_dim_dict(dimdict)
Example #11
0
 def unstructured_index(dd):
     return list(maps.map_from_dim_dict(dd).global_iter)
Example #12
0
 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)
Example #13
0
 def setUp(self):
     dimdict = dict(dist_type='b', size=(39 - 16), start=16, stop=39)
     self.m = maps.map_from_dim_dict(dimdict)
Example #14
0
 def setUp(self):
     size = 20
     dimdict = dict(dist_type='n', size=size)
     self.m = maps.map_from_dim_dict(dimdict)