src/smpi/mpi/smpi_group.cpp

   1 /* Copyright (c) 2010-2017. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #include "smpi_group.hpp"
   7 #include "smpi_comm.hpp"
   8 #include <string>
   9 #include <xbt/log.h>
  10
  11 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_group, smpi, "Logging specific to SMPI (group)");
  12
  13 simgrid::smpi::Group mpi_MPI_GROUP_EMPTY;
  14 MPI_Group MPI_GROUP_EMPTY=&mpi_MPI_GROUP_EMPTY;
  15
  16 namespace simgrid{
  17 namespace smpi{
  18
  19 Group::Group()
  20 {
  21   size_              = 0;       /* size */
  22   refcount_          = 1;       /* refcount_: start > 0 so that this group never gets freed */
  23 }
  24
  25 Group::Group(int n) : size_(n), rank_to_index_map_(size_, MPI_UNDEFINED)
  26 {
  27   refcount_ = 1;
  28 }
  29
  30 Group::Group(MPI_Group origin)
  31 {
  32   if (origin != MPI_GROUP_NULL && origin != MPI_GROUP_EMPTY) {
  33     size_              = origin->size();
  34     refcount_          = 1;
  35     rank_to_index_map_ = origin->rank_to_index_map_;
  36     index_to_rank_map_ = origin->index_to_rank_map_;
  37   }
  38 }
  39
  40 void Group::set_mapping(int index, int rank)
  41 {
  42   if (0 <= rank && rank < size_) {
  43     rank_to_index_map_[rank] = index;
  44     if (index != MPI_UNDEFINED) {
  45       if ((unsigned)index >= index_to_rank_map_.size())
  46         index_to_rank_map_.resize(index + 1, MPI_UNDEFINED);
  47       index_to_rank_map_[index] = rank;
  48     }
  49   }
  50 }
  51
  52 int Group::index(int rank)
  53 {
  54   int index;
  55   if (0 <= rank && rank < size_)
  56     index = rank_to_index_map_[rank];
  57   else
  58     index = MPI_UNDEFINED;
  59   return index;
  60 }
  61
  62 int Group::rank(int index)
  63 {
  64   int rank;
  65   if (0 <= index && (unsigned)index < index_to_rank_map_.size())
  66     rank = index_to_rank_map_[index];
  67   else
  68     rank = MPI_UNDEFINED;
  69   return rank;
  70 }
  71
  72 void Group::ref()
  73 {
  74   refcount_++;
  75 }
  76
  77 void Group::unref(Group* group)
  78 {
  79   group->refcount_--;
  80   if (group->refcount_ <= 0) {
  81     delete group;
  82   }
  83 }
  84
  85 int Group::size()
  86 {
  87   return size_;
  88 }
  89
  90 int Group::compare(MPI_Group group2)
  91 {
  92   int result;
  93
  94   result = MPI_IDENT;
  95   if (size_ != group2->size()) {
  96     result = MPI_UNEQUAL;
  97   } else {
  98     int sz = group2->size();
  99     for (int i = 0; i < sz; i++) {
 100       int index = this->index(i);
 101       int rank = group2->rank(index);
 102       if (rank == MPI_UNDEFINED) {
 103         result = MPI_UNEQUAL;
 104         break;
 105       }
 106       if (rank != i) {
 107         result = MPI_SIMILAR;
 108       }
 109     }
 110   }
 111   return result;
 112 }
 113
 114 int Group::incl(int n, int* ranks, MPI_Group* newgroup)
 115 {
 116   int i=0;
 117   int index=0;
 118   if (n == 0) {
 119     *newgroup = MPI_GROUP_EMPTY;
 120   } else if (n == size_) {
 121     *newgroup = this;
 122     if (this != MPI_COMM_WORLD->group() && this != MPI_COMM_SELF->group() && this != MPI_GROUP_EMPTY)
 123       this->ref();
 124   } else {
 125     *newgroup = new Group(n);
 126     for (i = 0; i < n; i++) {
 127       index = this->index(ranks[i]);
 128       (*newgroup)->set_mapping(index, i);
 129     }
 130   }
 131   return MPI_SUCCESS;
 132 }
 133
 134 int Group::group_union(MPI_Group group2, MPI_Group* newgroup)
 135 {
 136   int size1 = size_;
 137   int size2 = group2->size();
 138   for (int i = 0; i < size2; i++) {
 139     int proc2 = group2->index(i);
 140     int proc1 = this->rank(proc2);
 141     if (proc1 == MPI_UNDEFINED) {
 142       size1++;
 143     }
 144   }
 145   if (size1 == 0) {
 146     *newgroup = MPI_GROUP_EMPTY;
 147   } else {
 148     *newgroup = new  Group(size1);
 149     size2 = this->size();
 150     for (int i = 0; i < size2; i++) {
 151       int proc1 = this->index(i);
 152       (*newgroup)->set_mapping(proc1, i);
 153     }
 154     for (int i = size2; i < size1; i++) {
 155       int proc2 = group2->index(i - size2);
 156       (*newgroup)->set_mapping(proc2, i);
 157     }
 158   }
 159   return MPI_SUCCESS;
 160 }
 161
 162 int Group::intersection(MPI_Group group2, MPI_Group* newgroup)
 163 {
 164   int size2 = group2->size();
 165   for (int i = 0; i < size2; i++) {
 166     int proc2 = group2->index(i);
 167     int proc1 = this->rank(proc2);
 168     if (proc1 == MPI_UNDEFINED) {
 169       size2--;
 170     }
 171   }
 172   if (size2 == 0) {
 173     *newgroup = MPI_GROUP_EMPTY;
 174   } else {
 175     *newgroup = new  Group(size2);
 176     int j=0;
 177     for (int i = 0; i < group2->size(); i++) {
 178       int proc2 = group2->index(i);
 179       int proc1 = this->rank(proc2);
 180       if (proc1 != MPI_UNDEFINED) {
 181         (*newgroup)->set_mapping(proc2, j);
 182         j++;
 183       }
 184     }
 185   }
 186   return MPI_SUCCESS;
 187 }
 188
 189 int Group::difference(MPI_Group group2, MPI_Group* newgroup)
 190 {
 191   int newsize = size_;
 192   int size2 = size_;
 193   for (int i = 0; i < size2; i++) {
 194     int proc1 = this->index(i);
 195     int proc2 = group2->rank(proc1);
 196     if (proc2 != MPI_UNDEFINED) {
 197       newsize--;
 198     }
 199   }
 200   if (newsize == 0) {
 201     *newgroup = MPI_GROUP_EMPTY;
 202   } else {
 203     *newgroup = new  Group(newsize);
 204     for (int i = 0; i < size2; i++) {
 205       int proc1 = this->index(i);
 206       int proc2 = group2->rank(proc1);
 207       if (proc2 == MPI_UNDEFINED) {
 208         (*newgroup)->set_mapping(proc1, i);
 209       }
 210     }
 211   }
 212   return MPI_SUCCESS;
 213 }
 214
 215 int Group::excl(int n, int *ranks, MPI_Group * newgroup){
 216   int oldsize = size_;
 217   int newsize = oldsize - n;
 218   *newgroup = new  Group(newsize);
 219   int* to_exclude = new int[size_];
 220   for (int i     = 0; i < oldsize; i++)
 221     to_exclude[i]=0;
 222   for (int i            = 0; i < n; i++)
 223     to_exclude[ranks[i]]=1;
 224   int j = 0;
 225   for (int i = 0; i < oldsize; i++) {
 226     if(to_exclude[i]==0){
 227       int index = this->index(i);
 228       (*newgroup)->set_mapping(index, j);
 229       j++;
 230     }
 231   }
 232   delete[] to_exclude;
 233   return MPI_SUCCESS;
 234
 235 }
 236
 237 int Group::range_incl(int n, int ranges[][3], MPI_Group * newgroup){
 238   int newsize = 0;
 239   for (int i = 0; i < n; i++) {
 240     for (int rank = ranges[i][0];                    /* First */
 241          rank >= 0 && rank < size_; /* Last */
 242          ) {
 243       newsize++;
 244       if(rank == ranges[i][1]){/*already last ?*/
 245         break;
 246       }
 247       rank += ranges[i][2]; /* Stride */
 248       if (ranges[i][0] < ranges[i][1]) {
 249         if (rank > ranges[i][1])
 250           break;
 251       } else {
 252         if (rank < ranges[i][1])
 253           break;
 254       }
 255     }
 256   }
 257   *newgroup = new  Group(newsize);
 258   int j     = 0;
 259   for (int i = 0; i < n; i++) {
 260     for (int rank = ranges[i][0];                    /* First */
 261          rank >= 0 && rank < size_; /* Last */
 262          ) {
 263       int index = this->index(rank);
 264       (*newgroup)->set_mapping(index, j);
 265       j++;
 266       if(rank == ranges[i][1]){/*already last ?*/
 267         break;
 268       }
 269       rank += ranges[i][2]; /* Stride */
 270       if (ranges[i][0] < ranges[i][1]) {
 271         if (rank > ranges[i][1])
 272           break;
 273       } else {
 274         if (rank < ranges[i][1])
 275           break;
 276       }
 277     }
 278   }
 279   return MPI_SUCCESS;
 280 }
 281
 282 int Group::range_excl(int n, int ranges[][3], MPI_Group * newgroup){
 283   int newsize = size_;
 284   for (int i = 0; i < n; i++) {
 285     for (int rank = ranges[i][0];                    /* First */
 286          rank >= 0 && rank < size_; /* Last */
 287          ) {
 288       newsize--;
 289       if(rank == ranges[i][1]){/*already last ?*/
 290         break;
 291       }
 292       rank += ranges[i][2]; /* Stride */
 293       if (ranges[i][0] < ranges[i][1]) {
 294         if (rank > ranges[i][1])
 295           break;
 296       } else {
 297         if (rank < ranges[i][1])
 298           break;
 299       }
 300     }
 301   }
 302   if (newsize == 0) {
 303     *newgroup = MPI_GROUP_EMPTY;
 304   } else {
 305     *newgroup = new  Group(newsize);
 306     int newrank = 0;
 307     int oldrank = 0;
 308     while (newrank < newsize) {
 309       int add = 1;
 310       for (int i = 0; i < n; i++) {
 311         for (int rank = ranges[i][0]; rank >= 0 && rank < size_;) {
 312           if(rank==oldrank){
 313             add = 0;
 314             break;
 315           }
 316           if(rank == ranges[i][1]){/*already last ?*/
 317             break;
 318           }
 319           rank += ranges[i][2]; /* Stride */
 320           if (ranges[i][0]<ranges[i][1]){
 321             if (rank > ranges[i][1])
 322               break;
 323           }else{
 324             if (rank < ranges[i][1])
 325               break;
 326           }
 327         }
 328       }
 329       if(add==1){
 330         int index = this->index(oldrank);
 331         (*newgroup)->set_mapping(index, newrank);
 332         newrank++;
 333       }
 334       oldrank++;
 335     }
 336   }
 337   return MPI_SUCCESS;
 338 }
 339
 340 MPI_Group Group::f2c(int id) {
 341   if(id == -2) {
 342     return MPI_GROUP_EMPTY;
 343   } else if(F2C::f2c_lookup() != nullptr && id >= 0) {
 344     char key[KEY_SIZE];
 345     return static_cast<MPI_Group>(F2C::f2c_lookup()->at(get_key(key, id)));
 346   } else {
 347     return static_cast<MPI_Group>(MPI_GROUP_NULL);
 348   }
 349 }
 350
 351 }
 352 }