root/ompi/mpi/c/cart_shift.c

DEFINITIONS

1. MPI_Cart_shift

   1 /*
   2  * Copyright (c) 2004-2007 The Trustees of Indiana University and Indiana
   3  *                         University Research and Technology
   4  *                         Corporation.  All rights reserved.
   5  * Copyright (c) 2004-2013 The University of Tennessee and The University
   6  *                         of Tennessee Research Foundation.  All rights
   7  *                         reserved.
   8  * Copyright (c) 2004-2008 High Performance Computing Center Stuttgart,
   9  *                         University of Stuttgart.  All rights reserved.
  10  * Copyright (c) 2004-2005 The Regents of the University of California.
  11  *                         All rights reserved.
  12  * Copyright (c) 2007      Cisco Systems, Inc.  All rights reserved.
  13  * Copyright (c) 2012-2013 Inria.  All rights reserved.
  14  * Copyright (c) 2015      Research Organization for Information Science
  15  *                         and Technology (RIST). All rights reserved.
  16  * $COPYRIGHT$
  17  *
  18  * Additional copyrights may follow
  19  *
  20  * $HEADER$
  21  */
  22 #include "ompi_config.h"
  23 #include <stdio.h>
  24 
  25 #include "ompi/mpi/c/bindings.h"
  26 #include "ompi/runtime/params.h"
  27 #include "ompi/communicator/communicator.h"
  28 #include "ompi/errhandler/errhandler.h"
  29 #include "ompi/mca/topo/topo.h"
  30 #include "ompi/memchecker.h"
  31 
  32 #if OMPI_BUILD_MPI_PROFILING
  33 #if OPAL_HAVE_WEAK_SYMBOLS
  34 #pragma weak MPI_Cart_shift = PMPI_Cart_shift
  35 #endif
  36 #define MPI_Cart_shift PMPI_Cart_shift
  37 #endif
  38 
  39 static const char FUNC_NAME[] = "MPI_Cart_shift";
  40 
  41 
  42 int MPI_Cart_shift(MPI_Comm comm, int direction, int disp,
  43                    int *rank_source, int *rank_dest)
  44 {
  45     int err;
  46 
  47     MEMCHECKER(
  48         memchecker_comm(comm);
  49     );
  50 
  51     /* check the arguments */
  52     if (MPI_PARAM_CHECK) {
  53         OMPI_ERR_INIT_FINALIZE(FUNC_NAME);
  54         if (ompi_comm_invalid(comm)) {
  55             return OMPI_ERRHANDLER_INVOKE (MPI_COMM_WORLD, MPI_ERR_COMM,
  56                                           FUNC_NAME);
  57         }
  58         if (OMPI_COMM_IS_INTER(comm)) {
  59             return OMPI_ERRHANDLER_INVOKE (comm, MPI_ERR_COMM,
  60                                           FUNC_NAME);
  61         }
  62         if (0 > direction) { /* yet to detect direction >= comm->c_topo_ndims */
  63             return OMPI_ERRHANDLER_INVOKE (comm, MPI_ERR_DIMS,
  64                                           FUNC_NAME);
  65         }
  66         if (NULL == rank_source || NULL == rank_dest) {
  67             return OMPI_ERRHANDLER_INVOKE (comm, MPI_ERR_ARG,
  68                                           FUNC_NAME);
  69         }
  70     }
  71 
  72     if (!OMPI_COMM_IS_CART(comm)) {
  73         return OMPI_ERRHANDLER_INVOKE (comm, MPI_ERR_TOPOLOGY,
  74                                       FUNC_NAME);
  75     }
  76     OPAL_CR_ENTER_LIBRARY();
  77 
  78     /* call the function */
  79     err = comm->c_topo->topo.cart.cart_shift(comm, direction, disp, rank_source, rank_dest);
  80     OPAL_CR_EXIT_LIBRARY();
  81 
  82     OMPI_ERRHANDLER_RETURN(err, comm, err, FUNC_NAME);
  83 }