Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
[travis] detect linux as we should
[simgrid.git] / src / smpi / smpi_group.c
1 /* Copyright (c) 2010, 2013-2015. The SimGrid Team.
2  * All rights reserved.                                                     */
3
4 /* This program is free software; you can redistribute it and/or modify it
5  * under the terms of the license (GNU LGPL) which comes with this package. */
6
7 #include "private.h"
8
9 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_group, smpi,
10                                 "Logging specific to SMPI (group)");
11
12 typedef struct s_smpi_mpi_group {
13   int size;
14   int *rank_to_index_map;
15   xbt_dict_t index_to_rank_map;
16   int refcount;
17 } s_smpi_mpi_group_t;
18
19 static s_smpi_mpi_group_t mpi_MPI_GROUP_EMPTY = {
20   0,                            /* size */
21   NULL,                         /* rank_to_index_map */
22   NULL,                         /* index_to_rank_map */
23   1,                            /* refcount: start > 0 so that this group never gets freed */
24 };
25
26 MPI_Group MPI_GROUP_EMPTY = &mpi_MPI_GROUP_EMPTY;
27
28 MPI_Group smpi_group_new(int size)
29 {
30   MPI_Group group;
31   int i;
32
33   group = xbt_new(s_smpi_mpi_group_t, 1);
34   group->size = size;
35   group->rank_to_index_map = xbt_new(int, size);
36   group->index_to_rank_map = xbt_dict_new_homogeneous(xbt_free_f);
37   group->refcount = 1;
38   for (i = 0; i < size; i++) {
39     group->rank_to_index_map[i] = MPI_UNDEFINED;
40   }
41
42   return group;
43 }
44
45 MPI_Group smpi_group_copy(MPI_Group origin)
46 {
47   MPI_Group group=origin;
48   char *key;
49   char *ptr_rank;
50   xbt_dict_cursor_t cursor = NULL;
51   
52   int i;
53   if(origin!= smpi_comm_group(MPI_COMM_WORLD)
54             && origin != MPI_GROUP_NULL
55             && origin != smpi_comm_group(MPI_COMM_SELF)
56             && origin != MPI_GROUP_EMPTY)
57     {
58       group = xbt_new(s_smpi_mpi_group_t, 1);
59       group->size = origin->size;
60       group->rank_to_index_map = xbt_new(int, group->size);
61       group->index_to_rank_map = xbt_dict_new_homogeneous(xbt_free_f);
62       group->refcount = 1;
63       for (i = 0; i < group->size; i++) {
64         group->rank_to_index_map[i] = origin->rank_to_index_map[i];
65       }
66
67       xbt_dict_foreach(origin->index_to_rank_map, cursor, key, ptr_rank) {
68         xbt_dict_set(group->index_to_rank_map, key, ptr_rank, NULL);
69       }
70     }
71
72   return group;
73 }
74
75
76 void smpi_group_destroy(MPI_Group group)
77 {
78   if(group!= smpi_comm_group(MPI_COMM_WORLD)
79           && group != MPI_GROUP_NULL
80           && group != smpi_comm_group(MPI_COMM_SELF)
81           && group != MPI_GROUP_EMPTY)
82   smpi_group_unuse(group);
83 }
84
85 void smpi_group_set_mapping(MPI_Group group, int index, int rank)
86 {
87   char * key;
88   int * val_rank;
89
90   if (rank < group->size) {
91     group->rank_to_index_map[rank] = index;
92     if (index!=MPI_UNDEFINED ) {
93       val_rank = (int *) malloc(sizeof(int));
94       *val_rank = rank; 
95       int size = asprintf(&key, "%d", index);
96       if (size!=-1){
97         xbt_dict_set(group->index_to_rank_map, key, val_rank, NULL);
98         free(key);
99       } else {
100         xbt_die("could not allocate memory for asprintf");
101       }
102     }
103   }
104 }
105
106 int smpi_group_index(MPI_Group group, int rank)
107 {
108   int index = MPI_UNDEFINED;
109
110   if (0 <= rank && rank < group->size) {
111     index = group->rank_to_index_map[rank];
112   }
113   return index;
114 }
115
116 int smpi_group_rank(MPI_Group group, int index)
117 {
118   int * ptr_rank = NULL;
119   char * key;
120   int size = asprintf(&key, "%d", index);
121   if (size!=-1){
122     ptr_rank = xbt_dict_get_or_null(group->index_to_rank_map, key);
123     xbt_free(key);
124   }else
125     xbt_die("could not allocate memory for asprintf");
126   if (!ptr_rank)
127     return MPI_UNDEFINED;
128   return *ptr_rank;
129 }
130
131 int smpi_group_use(MPI_Group group)
132 {
133   group->refcount++;
134   return group->refcount;
135 }
136
137 int smpi_group_unuse(MPI_Group group)
138 {
139   group->refcount--;
140   if (group->refcount <= 0) {
141     xbt_free(group->rank_to_index_map);
142     xbt_dict_free(&group->index_to_rank_map);
143     xbt_free(group);
144     return 0;
145   }
146   return group->refcount;
147
148 }
149
150 int smpi_group_size(MPI_Group group)
151 {
152   return group->size;
153 }
154
155 int smpi_group_compare(MPI_Group group1, MPI_Group group2)
156 {
157   int result;
158   int i, index, rank, size;
159
160   result = MPI_IDENT;
161   if (smpi_group_size(group1) != smpi_group_size(group2)) {
162     result = MPI_UNEQUAL;
163   } else {
164     size = smpi_group_size(group2);
165     for (i = 0; i < size; i++) {
166       index = smpi_group_index(group1, i);
167       rank = smpi_group_rank(group2, index);
168       if (rank == MPI_UNDEFINED) {
169         result = MPI_UNEQUAL;
170         break;
171       }
172       if (rank != i) {
173         result = MPI_SIMILAR;
174       }
175     }
176   }
177   return result;
178 }
179
180 int smpi_group_incl(MPI_Group group, int n, int* ranks, MPI_Group* newgroup)
181 {
182   int i=0, index=0;
183   if (n == 0) {
184     *newgroup = MPI_GROUP_EMPTY;
185   } else if (n == smpi_group_size(group)) {
186     *newgroup = group;
187     if(group!= smpi_comm_group(MPI_COMM_WORLD)
188               && group != MPI_GROUP_NULL
189               && group != smpi_comm_group(MPI_COMM_SELF)
190               && group != MPI_GROUP_EMPTY)
191     smpi_group_use(group);
192   } else {
193     *newgroup = smpi_group_new(n);
194     for (i = 0; i < n; i++) {
195       index = smpi_group_index(group, ranks[i]);
196       smpi_group_set_mapping(*newgroup, index, i);
197     }
198   }
199   return MPI_SUCCESS;
200 }