1 /* Copyright (c) 2007-2019. The SimGrid Team. All rights reserved. */
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. */
7 #include "smpi_comm.hpp"
9 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(smpi_pmpi);
11 /* PMPI User level calls */
13 /* The topo part of MPI_COMM_WORLD should always be nullptr. When other topologies will be implemented, not only should we
14 * check if the topology is nullptr, but we should check if it is the good topology type (so we have to add a
15 * MPIR_Topo_Type field, and replace the MPI_Topology field by an union)*/
17 int PMPI_Cart_create(MPI_Comm comm_old, int ndims, const int* dims, const int* periodic, int reorder, MPI_Comm* comm_cart) {
18 if (comm_old == MPI_COMM_NULL){
20 } else if (ndims < 0 || (ndims > 0 && (dims == nullptr || periodic == nullptr)) || comm_cart == nullptr) {
23 simgrid::smpi::Topo_Cart* topo = new simgrid::smpi::Topo_Cart(comm_old, ndims, dims, periodic, reorder, comm_cart);
24 if (*comm_cart == MPI_COMM_NULL) {
27 xbt_assert((*comm_cart)->topo() == topo);
33 int PMPI_Cart_rank(MPI_Comm comm, const int* coords, int* rank) {
34 if(comm == MPI_COMM_NULL || comm->topo() == nullptr) {
35 return MPI_ERR_TOPOLOGY;
37 if (coords == nullptr) {
40 MPIR_Cart_Topology topo = static_cast<MPIR_Cart_Topology>(comm->topo());
44 return topo->rank(coords, rank);
47 int PMPI_Cart_shift(MPI_Comm comm, int direction, int displ, int* source, int* dest) {
48 if(comm == MPI_COMM_NULL || comm->topo() == nullptr) {
49 return MPI_ERR_TOPOLOGY;
51 if (source == nullptr || dest == nullptr || direction < 0 ) {
54 MPIR_Cart_Topology topo = static_cast<MPIR_Cart_Topology>(comm->topo());
58 return topo->shift(direction, displ, source, dest);
61 int PMPI_Cart_coords(MPI_Comm comm, int rank, int maxdims, int* coords) {
62 if(comm == MPI_COMM_NULL || comm->topo() == nullptr) {
63 return MPI_ERR_TOPOLOGY;
65 if (rank < 0 || rank >= comm->size()) {
71 if(coords == nullptr) {
74 MPIR_Cart_Topology topo = static_cast<MPIR_Cart_Topology>(comm->topo());
78 return topo->coords(rank, maxdims, coords);
81 int PMPI_Cart_get(MPI_Comm comm, int maxdims, int* dims, int* periods, int* coords) {
82 if(comm == nullptr || comm->topo() == nullptr) {
83 return MPI_ERR_TOPOLOGY;
85 if(maxdims <= 0 || dims == nullptr || periods == nullptr || coords == nullptr) {
88 MPIR_Cart_Topology topo = static_cast<MPIR_Cart_Topology>(comm->topo());
92 return topo->get(maxdims, dims, periods, coords);
95 int PMPI_Cartdim_get(MPI_Comm comm, int* ndims) {
96 if (comm == MPI_COMM_NULL || comm->topo() == nullptr) {
97 return MPI_ERR_TOPOLOGY;
99 if (ndims == nullptr) {
102 MPIR_Cart_Topology topo = static_cast<MPIR_Cart_Topology>(comm->topo());
106 return topo->dim_get(ndims);
109 int PMPI_Dims_create(int nnodes, int ndims, int* dims) {
110 if(dims == nullptr) {
113 if (ndims < 1 || nnodes < 1) {
116 return simgrid::smpi::Topo_Cart::Dims_create(nnodes, ndims, dims);
119 int PMPI_Cart_sub(MPI_Comm comm, const int* remain_dims, MPI_Comm* comm_new) {
120 if(comm == MPI_COMM_NULL || comm->topo() == nullptr) {
121 return MPI_ERR_TOPOLOGY;
123 if (comm_new == nullptr) {
126 MPIR_Cart_Topology topo = static_cast<MPIR_Cart_Topology>(comm->topo());
130 MPIR_Cart_Topology cart = topo->sub(remain_dims, comm_new);
131 if(*comm_new==MPI_COMM_NULL)