src/smpi/bindings/smpi_pmpi_topo.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_comm.hpp"
   8
   9 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(smpi_pmpi);
  10
  11 /* PMPI User level calls */
  12
  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)*/
  16
  17 int PMPI_Cart_create(MPI_Comm comm_old, int ndims, const int* dims, const int* periodic, int reorder, MPI_Comm* comm_cart) {
  18   CHECK_COMM2(1, comm_old)
  19   if (ndims > 0){
  20     CHECK_NULL(3, MPI_ERR_ARG, dims)
  21     CHECK_NULL(4, MPI_ERR_ARG, periodic)
  22     CHECK_NULL(6, MPI_ERR_ARG, comm_cart)
  23   }
  24   CHECK_NEGATIVE(2, MPI_ERR_ARG, ndims)
  25   for (int i = 0; i < ndims; i++)
  26     CHECK_NEGATIVE(2, MPI_ERR_ARG, dims[i])
  27   const simgrid::smpi::Topo_Cart* topo =
  28       new simgrid::smpi::Topo_Cart(comm_old, ndims, dims, periodic, reorder, comm_cart);
  29   if (*comm_cart == MPI_COMM_NULL) {
  30     delete topo;
  31   } else {
  32     xbt_assert((*comm_cart)->topo() == topo);
  33   }
  34   return MPI_SUCCESS;
  35 }
  36
  37 int PMPI_Cart_rank(MPI_Comm comm, const int* coords, int* rank) {
  38   CHECK_COMM(1)
  39   CHECK_NULL(1, MPI_ERR_TOPOLOGY, comm->topo())
  40   CHECK_NULL(2, MPI_SUCCESS, coords)
  41   MPIR_Cart_Topology topo = static_cast<MPIR_Cart_Topology>(comm->topo());
  42   if (topo==nullptr) {
  43     return MPI_ERR_ARG;
  44   }
  45   return topo->rank(coords, rank);
  46 }
  47
  48 int PMPI_Cart_shift(MPI_Comm comm, int direction, int displ, int* source, int* dest) {
  49   CHECK_COMM(1)
  50   CHECK_NULL(1, MPI_ERR_TOPOLOGY, comm->topo())
  51   CHECK_NEGATIVE(3, MPI_ERR_ARG, direction)
  52   CHECK_NULL(4, MPI_ERR_ARG, source)
  53   CHECK_NULL(5, MPI_ERR_ARG, dest)
  54   MPIR_Cart_Topology topo = static_cast<MPIR_Cart_Topology>(comm->topo());
  55   if (topo==nullptr) {
  56     return MPI_ERR_ARG;
  57   }
  58   return topo->shift(direction, displ, source, dest);
  59 }
  60
  61 int PMPI_Cart_coords(MPI_Comm comm, int rank, int maxdims, int* coords) {
  62   CHECK_COMM(1)
  63   CHECK_NULL(1, MPI_ERR_TOPOLOGY, comm->topo())
  64   CHECK_NEGATIVE(3, MPI_ERR_ARG, maxdims)
  65   CHECK_RANK(2, rank, comm)
  66   if(maxdims==0 || coords == nullptr) {
  67     return MPI_SUCCESS;
  68   }
  69   MPIR_Cart_Topology topo = static_cast<MPIR_Cart_Topology>(comm->topo());
  70   if (topo==nullptr) {
  71     return MPI_ERR_ARG;
  72   }
  73   return topo->coords(rank, maxdims, coords);
  74 }
  75
  76 int PMPI_Cart_get(MPI_Comm comm, int maxdims, int* dims, int* periods, int* coords) {
  77   if(dims == nullptr || periods == nullptr || coords == nullptr){
  78     return MPI_SUCCESS;
  79   }
  80   CHECK_COMM(1)
  81   CHECK_NULL(1, MPI_ERR_TOPOLOGY, comm->topo())
  82   CHECK_NEGATIVE(3, MPI_ERR_ARG, maxdims)
  83   MPIR_Cart_Topology topo = static_cast<MPIR_Cart_Topology>(comm->topo());
  84   if (topo==nullptr) {
  85     return MPI_ERR_ARG;
  86   }
  87   return topo->get(maxdims, dims, periods, coords);
  88 }
  89
  90 int PMPI_Cartdim_get(MPI_Comm comm, int* ndims) {
  91   CHECK_COMM(1)
  92   CHECK_NULL(1, MPI_ERR_TOPOLOGY, comm->topo())
  93   CHECK_NULL(2, MPI_ERR_ARG, ndims)
  94   MPIR_Cart_Topology topo = static_cast<MPIR_Cart_Topology>(comm->topo());
  95   if (topo==nullptr) {
  96     return MPI_ERR_ARG;
  97   }
  98   return topo->dim_get(ndims);
  99 }
 100
 101 int PMPI_Dims_create(int nnodes, int ndims, int* dims) {
 102   CHECK_NULL(3, MPI_SUCCESS, dims)
 103   if (ndims < 1 || nnodes < 1) {
 104     return MPI_ERR_DIMS;
 105   }
 106   return simgrid::smpi::Topo_Cart::Dims_create(nnodes, ndims, dims);
 107 }
 108
 109 int PMPI_Cart_sub(MPI_Comm comm, const int* remain_dims, MPI_Comm* comm_new) {
 110   CHECK_COMM(1)
 111   CHECK_NULL(1, MPI_ERR_TOPOLOGY, comm->topo())
 112   CHECK_NULL(3, MPI_ERR_ARG, comm_new)
 113   MPIR_Cart_Topology topo = static_cast<MPIR_Cart_Topology>(comm->topo());
 114   if (topo==nullptr) {
 115     return MPI_ERR_ARG;
 116   }
 117   const simgrid::smpi::Topo_Cart* cart = topo->sub(remain_dims, comm_new);
 118   if(*comm_new==MPI_COMM_NULL)
 119       delete cart;
 120   if(cart==nullptr)
 121     return  MPI_ERR_ARG;
 122   return MPI_SUCCESS;
 123 }