src/smpi/bindings/smpi_pmpi_group.cpp

   1 /* Copyright (c) 2007-2020. 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 "private.hpp"
   7 #include "smpi_coll.hpp"
   8 #include "smpi_comm.hpp"
   9 #include "smpi_datatype_derived.hpp"
  10 #include "smpi_op.hpp"
  11 #include "src/smpi/include/smpi_actor.hpp"
  12
  13 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(smpi_pmpi);
  14
  15 /* PMPI User level calls */
  16
  17 int PMPI_Group_free(MPI_Group * group)
  18 {
  19   CHECK_NULL(1, MPI_ERR_ARG, group)
  20   if(*group != MPI_COMM_WORLD->group() && *group != MPI_GROUP_EMPTY)
  21     simgrid::smpi::Group::unref(*group);
  22   *group = MPI_GROUP_NULL;
  23   return MPI_SUCCESS;
  24 }
  25
  26 int PMPI_Group_size(MPI_Group group, int *size)
  27 {
  28   CHECK_GROUP(1, group)
  29   CHECK_NULL(2, MPI_ERR_ARG, size)
  30   *size = group->size();
  31   return MPI_SUCCESS;
  32 }
  33
  34 int PMPI_Group_rank(MPI_Group group, int *rank)
  35 {
  36   CHECK_GROUP(1, group)
  37   CHECK_NULL(2, MPI_ERR_ARG, rank)
  38   *rank = group->rank(simgrid::s4u::this_actor::get_pid());
  39   return MPI_SUCCESS;
  40 }
  41
  42 int PMPI_Group_translate_ranks(MPI_Group group1, int n, const int *ranks1, MPI_Group group2, int *ranks2)
  43 {
  44   CHECK_GROUP(1, group1)
  45   CHECK_GROUP(4, group2)
  46   for (int i = 0; i < n; i++) {
  47     if(ranks1[i]==MPI_PROC_NULL){
  48       ranks2[i]=MPI_PROC_NULL;
  49     }else{
  50       simgrid::s4u::Actor* actor = group1->actor(ranks1[i]);
  51       ranks2[i] = group2->rank(actor);
  52     }
  53   }
  54   return MPI_SUCCESS;
  55 }
  56
  57 int PMPI_Group_compare(MPI_Group group1, MPI_Group group2, int *result)
  58 {
  59   CHECK_GROUP(1, group1)
  60   CHECK_GROUP(2, group2)
  61   CHECK_NULL(3, MPI_ERR_ARG, result)
  62   *result = group1->compare(group2);
  63   return MPI_SUCCESS;
  64 }
  65
  66 int PMPI_Group_union(MPI_Group group1, MPI_Group group2, MPI_Group * newgroup)
  67 {
  68   CHECK_GROUP(1, group1)
  69   CHECK_GROUP(2, group2)
  70   CHECK_NULL(3, MPI_ERR_ARG, newgroup)
  71   return group1->group_union(group2, newgroup);
  72 }
  73
  74 int PMPI_Group_intersection(MPI_Group group1, MPI_Group group2, MPI_Group * newgroup)
  75 {
  76   CHECK_GROUP(1, group1)
  77   CHECK_GROUP(2, group2)
  78   CHECK_NULL(3, MPI_ERR_ARG, newgroup)
  79   return group1->intersection(group2,newgroup);
  80 }
  81
  82 int PMPI_Group_difference(MPI_Group group1, MPI_Group group2, MPI_Group * newgroup)
  83 {
  84   CHECK_GROUP(1, group1)
  85   CHECK_GROUP(2, group2)
  86   CHECK_NULL(3, MPI_ERR_ARG, newgroup)
  87   return group1->difference(group2,newgroup);
  88 }
  89
  90 int PMPI_Group_incl(MPI_Group group, int n, const int *ranks, MPI_Group * newgroup)
  91 {
  92   CHECK_GROUP(1, group)
  93   CHECK_NULL(4, MPI_ERR_ARG, newgroup)
  94   return group->incl(n, ranks, newgroup);
  95 }
  96
  97 int PMPI_Group_excl(MPI_Group group, int n, const int *ranks, MPI_Group * newgroup)
  98 {
  99   CHECK_GROUP(1, group)
 100   CHECK_NULL(4, MPI_ERR_ARG, newgroup)
 101   if (n == 0) {
 102     *newgroup = group;
 103     if (group != MPI_COMM_WORLD->group() && group != MPI_COMM_SELF->group() && group != MPI_GROUP_EMPTY)
 104       group->ref();
 105     return MPI_SUCCESS;
 106   } else if (n == group->size()) {
 107     *newgroup = MPI_GROUP_EMPTY;
 108     return MPI_SUCCESS;
 109   } else {
 110     return group->excl(n,ranks,newgroup);
 111   }
 112 }
 113
 114 int PMPI_Group_range_incl(MPI_Group group, int n, int ranges[][3], MPI_Group * newgroup)
 115 {
 116   CHECK_GROUP(1, group)
 117   CHECK_NULL(4, MPI_ERR_ARG, newgroup)
 118   if (n == 0) {
 119     *newgroup = MPI_GROUP_EMPTY;
 120     return MPI_SUCCESS;
 121   } else {
 122     return group->range_incl(n,ranges,newgroup);
 123   }
 124 }
 125
 126 int PMPI_Group_range_excl(MPI_Group group, int n, int ranges[][3], MPI_Group * newgroup)
 127 {
 128   CHECK_GROUP(1, group)
 129   CHECK_NULL(4, MPI_ERR_ARG, newgroup)
 130   if (n == 0) {
 131     *newgroup = group;
 132     if (group != MPI_COMM_WORLD->group() && group != MPI_COMM_SELF->group() &&
 133         group != MPI_GROUP_EMPTY)
 134       group->ref();
 135     return MPI_SUCCESS;
 136   } else {
 137     return group->range_excl(n,ranges,newgroup);
 138   }
 139 }
 140
 141 MPI_Group PMPI_Group_f2c(MPI_Fint group){
 142   if(group==-1)
 143     return MPI_GROUP_NULL;
 144   return simgrid::smpi::Group::f2c(group);
 145 }
 146
 147 MPI_Fint PMPI_Group_c2f(MPI_Group group){
 148   if(group==MPI_GROUP_NULL)
 149     return -1;
 150   return group->c2f();
 151 }