This source file includes following definitions.
- MPI_Scatter
   1 
   2 
   3 
   4 
   5 
   6 
   7 
   8 
   9 
  10 
  11 
  12 
  13 
  14 
  15 
  16 
  17 
  18 
  19 
  20 
  21 
  22 
  23 
  24 
  25 
  26 #include "ompi_config.h"
  27 #include <stdio.h>
  28 
  29 #include "ompi/mpi/c/bindings.h"
  30 #include "ompi/runtime/params.h"
  31 #include "ompi/communicator/communicator.h"
  32 #include "ompi/errhandler/errhandler.h"
  33 #include "ompi/datatype/ompi_datatype.h"
  34 #include "ompi/memchecker.h"
  35 #include "ompi/runtime/ompi_spc.h"
  36 
  37 #if OMPI_BUILD_MPI_PROFILING
  38 #if OPAL_HAVE_WEAK_SYMBOLS
  39 #pragma weak MPI_Scatter = PMPI_Scatter
  40 #endif
  41 #define MPI_Scatter PMPI_Scatter
  42 #endif
  43 
  44 static const char FUNC_NAME[] = "MPI_Scatter";
  45 
  46 
  47 int MPI_Scatter(const void *sendbuf, int sendcount, MPI_Datatype sendtype,
  48                 void *recvbuf, int recvcount, MPI_Datatype recvtype,
  49                 int root, MPI_Comm comm)
  50 {
  51     int err;
  52 
  53     SPC_RECORD(OMPI_SPC_SCATTER, 1);
  54 
  55     MEMCHECKER(
  56         memchecker_comm(comm);
  57         if(OMPI_COMM_IS_INTRA(comm)) {
  58             if(ompi_comm_rank(comm) == root) {
  59                 memchecker_datatype(sendtype);
  60                 
  61                 memchecker_call(&opal_memchecker_base_isdefined, sendbuf, sendcount, sendtype);
  62                 if(MPI_IN_PLACE != recvbuf) {
  63                     memchecker_datatype(recvtype);
  64                     
  65                     memchecker_call(&opal_memchecker_base_isaddressable, recvbuf, recvcount, recvtype);
  66                 }
  67             } else {
  68                 memchecker_datatype(recvtype);
  69                 
  70                 memchecker_call(&opal_memchecker_base_isaddressable, recvbuf, recvcount, recvtype);
  71             }
  72         } else {
  73             if(MPI_ROOT == root) {
  74                 memchecker_datatype(sendtype);
  75                 
  76                 memchecker_call(&opal_memchecker_base_isdefined, sendbuf, sendcount, sendtype);
  77             } else if (MPI_PROC_NULL != root) {
  78                 memchecker_datatype(recvtype);
  79                 
  80                 memchecker_call(&opal_memchecker_base_isaddressable, recvbuf, recvcount, recvtype);
  81             }
  82         }
  83     );
  84 
  85     if (MPI_PARAM_CHECK) {
  86         err = MPI_SUCCESS;
  87         OMPI_ERR_INIT_FINALIZE(FUNC_NAME);
  88         if (ompi_comm_invalid(comm)) {
  89             return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, MPI_ERR_COMM,
  90                                           FUNC_NAME);
  91         } else if ((ompi_comm_rank(comm) != root && MPI_IN_PLACE == recvbuf) ||
  92                    (ompi_comm_rank(comm) == root && MPI_IN_PLACE == sendbuf)) {
  93             return OMPI_ERRHANDLER_INVOKE(comm, MPI_ERR_ARG, FUNC_NAME);
  94         }
  95 
  96         
  97 
  98         if (OMPI_COMM_IS_INTRA(comm)) {
  99 
 100           
 101 
 102           if ((root >= ompi_comm_size(comm)) || (root < 0)) {
 103               err = MPI_ERR_ROOT;
 104           } else if (MPI_IN_PLACE != recvbuf) {
 105               if (recvcount < 0) {
 106                   err = MPI_ERR_COUNT;
 107               } else if (MPI_DATATYPE_NULL == recvtype || NULL == recvtype) {
 108                   err = MPI_ERR_TYPE;
 109               }
 110           }
 111 
 112           
 113 
 114 
 115 
 116 
 117 
 118           else if (ompi_comm_rank(comm) == root) {
 119               OMPI_CHECK_DATATYPE_FOR_SEND(err, sendtype, sendcount);
 120           }
 121           OMPI_ERRHANDLER_CHECK(err, comm, err, FUNC_NAME);
 122         }
 123 
 124         
 125 
 126         else {
 127           if (! ((root >= 0 && root < ompi_comm_remote_size(comm)) ||
 128                  MPI_ROOT == root || MPI_PROC_NULL == root)) {
 129               err = MPI_ERR_ROOT;
 130           }
 131 
 132           
 133 
 134           else if (MPI_ROOT != root && MPI_PROC_NULL != root) {
 135             if (recvcount < 0) {
 136                 err = MPI_ERR_COUNT;
 137             } else if (MPI_DATATYPE_NULL == recvtype) {
 138                 err = MPI_ERR_TYPE;
 139             }
 140           }
 141 
 142           
 143 
 144 
 145 
 146           else if (MPI_ROOT == root) {
 147               OMPI_CHECK_DATATYPE_FOR_SEND(err, sendtype, sendcount);
 148           }
 149           OMPI_ERRHANDLER_CHECK(err, comm, err, FUNC_NAME);
 150         }
 151     }
 152 
 153     
 154 
 155     if ((0 == recvcount && MPI_ROOT != root &&
 156          (ompi_comm_rank(comm) != root ||
 157           (ompi_comm_rank(comm) == root && MPI_IN_PLACE != recvbuf))) ||
 158         (ompi_comm_rank(comm) == root && MPI_IN_PLACE == recvbuf &&
 159          0 == sendcount) ||
 160         (0 == sendcount && (MPI_ROOT == root || MPI_PROC_NULL == root))) {
 161         return MPI_SUCCESS;
 162     }
 163 
 164     OPAL_CR_ENTER_LIBRARY();
 165 
 166     
 167     err = comm->c_coll->coll_scatter(sendbuf, sendcount, sendtype, recvbuf,
 168                                     recvcount, recvtype, root, comm,
 169                                     comm->c_coll->coll_scatter_module);
 170     OMPI_ERRHANDLER_RETURN(err, comm, err, FUNC_NAME);
 171 }