+++ /dev/null
-/* Copyright (c) 2010, 2013-2015. The SimGrid Team.
- * All rights reserved. */
-
-/* This program is free software; you can redistribute it and/or modify it
- * under the terms of the license (GNU LGPL) which comes with this package. */
-
-#include "private.h"
-
-XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_group, smpi,
- "Logging specific to SMPI (group)");
-
-typedef struct s_smpi_mpi_group {
- int size;
- int *rank_to_index_map;
- xbt_dict_t index_to_rank_map;
- int refcount;
-} s_smpi_mpi_group_t;
-
-static s_smpi_mpi_group_t mpi_MPI_GROUP_EMPTY = {
- 0, /* size */
- NULL, /* rank_to_index_map */
- NULL, /* index_to_rank_map */
- 1, /* refcount: start > 0 so that this group never gets freed */
-};
-
-MPI_Group MPI_GROUP_EMPTY = &mpi_MPI_GROUP_EMPTY;
-
-MPI_Group smpi_group_new(int size)
-{
- MPI_Group group;
- int i;
-
- group = xbt_new(s_smpi_mpi_group_t, 1);
- group->size = size;
- group->rank_to_index_map = xbt_new(int, size);
- group->index_to_rank_map = xbt_dict_new_homogeneous(xbt_free_f);
- group->refcount = 1;
- for (i = 0; i < size; i++) {
- group->rank_to_index_map[i] = MPI_UNDEFINED;
- }
-
- return group;
-}
-
-MPI_Group smpi_group_copy(MPI_Group origin)
-{
- MPI_Group group=origin;
- char *key;
- char *ptr_rank;
- xbt_dict_cursor_t cursor = NULL;
-
- int i;
- if(origin!= smpi_comm_group(MPI_COMM_WORLD)
- && origin != MPI_GROUP_NULL
- && origin != smpi_comm_group(MPI_COMM_SELF)
- && origin != MPI_GROUP_EMPTY)
- {
- group = xbt_new(s_smpi_mpi_group_t, 1);
- group->size = origin->size;
- group->rank_to_index_map = xbt_new(int, group->size);
- group->index_to_rank_map = xbt_dict_new_homogeneous(xbt_free_f);
- group->refcount = 1;
- for (i = 0; i < group->size; i++) {
- group->rank_to_index_map[i] = origin->rank_to_index_map[i];
- }
-
- xbt_dict_foreach(origin->index_to_rank_map, cursor, key, ptr_rank) {
- xbt_dict_set(group->index_to_rank_map, key, ptr_rank, NULL);
- }
- }
-
- return group;
-}
-
-
-void smpi_group_destroy(MPI_Group group)
-{
- if(group!= smpi_comm_group(MPI_COMM_WORLD)
- && group != MPI_GROUP_NULL
- && group != smpi_comm_group(MPI_COMM_SELF)
- && group != MPI_GROUP_EMPTY)
- smpi_group_unuse(group);
-}
-
-void smpi_group_set_mapping(MPI_Group group, int index, int rank)
-{
- char * key;
- int * val_rank;
-
- if (rank < group->size) {
- group->rank_to_index_map[rank] = index;
- if (index!=MPI_UNDEFINED ) {
- val_rank = (int *) malloc(sizeof(int));
- *val_rank = rank;
- int size = asprintf(&key, "%d", index);
- if (size!=-1){
- xbt_dict_set(group->index_to_rank_map, key, val_rank, NULL);
- free(key);
- } else {
- xbt_die("could not allocate memory for asprintf");
- }
- }
- }
-}
-
-int smpi_group_index(MPI_Group group, int rank)
-{
- int index = MPI_UNDEFINED;
-
- if (0 <= rank && rank < group->size) {
- index = group->rank_to_index_map[rank];
- }
- return index;
-}
-
-int smpi_group_rank(MPI_Group group, int index)
-{
- int * ptr_rank = NULL;
- char * key;
- int size = asprintf(&key, "%d", index);
- if (size!=-1){
- ptr_rank = xbt_dict_get_or_null(group->index_to_rank_map, key);
- xbt_free(key);
- }else
- xbt_die("could not allocate memory for asprintf");
- if (!ptr_rank)
- return MPI_UNDEFINED;
- return *ptr_rank;
-}
-
-int smpi_group_use(MPI_Group group)
-{
- group->refcount++;
- return group->refcount;
-}
-
-int smpi_group_unuse(MPI_Group group)
-{
- group->refcount--;
- if (group->refcount <= 0) {
- xbt_free(group->rank_to_index_map);
- xbt_dict_free(&group->index_to_rank_map);
- xbt_free(group);
- return 0;
- }
- return group->refcount;
-
-}
-
-int smpi_group_size(MPI_Group group)
-{
- return group->size;
-}
-
-int smpi_group_compare(MPI_Group group1, MPI_Group group2)
-{
- int result;
- int i, index, rank, size;
-
- result = MPI_IDENT;
- if (smpi_group_size(group1) != smpi_group_size(group2)) {
- result = MPI_UNEQUAL;
- } else {
- size = smpi_group_size(group2);
- for (i = 0; i < size; i++) {
- index = smpi_group_index(group1, i);
- rank = smpi_group_rank(group2, index);
- if (rank == MPI_UNDEFINED) {
- result = MPI_UNEQUAL;
- break;
- }
- if (rank != i) {
- result = MPI_SIMILAR;
- }
- }
- }
- return result;
-}
-
-int smpi_group_incl(MPI_Group group, int n, int* ranks, MPI_Group* newgroup)
-{
- int i=0, index=0;
- if (n == 0) {
- *newgroup = MPI_GROUP_EMPTY;
- } else if (n == smpi_group_size(group)) {
- *newgroup = group;
- if(group!= smpi_comm_group(MPI_COMM_WORLD)
- && group != MPI_GROUP_NULL
- && group != smpi_comm_group(MPI_COMM_SELF)
- && group != MPI_GROUP_EMPTY)
- smpi_group_use(group);
- } else {
- *newgroup = smpi_group_new(n);
- for (i = 0; i < n; i++) {
- index = smpi_group_index(group, ranks[i]);
- smpi_group_set_mapping(*newgroup, index, i);
- }
- }
- return MPI_SUCCESS;
-}
