This source file includes following definitions.
- MPI_Iallgather
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_Iallgather = PMPI_Iallgather
40 #endif
41 #define MPI_Iallgather PMPI_Iallgather
42 #endif
43
44 static const char FUNC_NAME[] = "MPI_Iallgather";
45
46
47 int MPI_Iallgather(const void *sendbuf, int sendcount, MPI_Datatype sendtype,
48 void *recvbuf, int recvcount, MPI_Datatype recvtype,
49 MPI_Comm comm, MPI_Request *request)
50 {
51 int err;
52
53 SPC_RECORD(OMPI_SPC_IALLGATHER, 1);
54
55 MEMCHECKER(
56 int rank;
57 ptrdiff_t ext;
58
59 rank = ompi_comm_rank(comm);
60 ompi_datatype_type_extent(recvtype, &ext);
61
62 memchecker_datatype(recvtype);
63 memchecker_comm(comm);
64
65 if (MPI_IN_PLACE == sendbuf) {
66 memchecker_call(&opal_memchecker_base_isdefined,
67 (char *)(recvbuf)+rank*ext,
68 recvcount, recvtype);
69 } else {
70 memchecker_datatype(sendtype);
71 memchecker_call(&opal_memchecker_base_isdefined, sendbuf, sendcount, sendtype);
72 }
73
74 memchecker_call(&opal_memchecker_base_isaddressable, recvbuf, recvcount, recvtype);
75 );
76
77 if (MPI_PARAM_CHECK) {
78
79
80
81
82 err = MPI_SUCCESS;
83 OMPI_ERR_INIT_FINALIZE(FUNC_NAME);
84 if (ompi_comm_invalid(comm)) {
85 OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, MPI_ERR_COMM, FUNC_NAME);
86 } else if (MPI_DATATYPE_NULL == recvtype || NULL == recvtype) {
87 err = MPI_ERR_TYPE;
88 } else if (recvcount < 0) {
89 err = MPI_ERR_COUNT;
90 } else if ((MPI_IN_PLACE == sendbuf && OMPI_COMM_IS_INTER(comm)) ||
91 MPI_IN_PLACE == recvbuf) {
92 return OMPI_ERRHANDLER_INVOKE(comm, MPI_ERR_ARG, FUNC_NAME);
93 } else if (MPI_IN_PLACE != sendbuf) {
94 OMPI_CHECK_DATATYPE_FOR_SEND(err, sendtype, sendcount);
95 }
96 OMPI_ERRHANDLER_CHECK(err, comm, err, FUNC_NAME);
97 }
98
99 OPAL_CR_ENTER_LIBRARY();
100
101
102 err = comm->c_coll->coll_iallgather(sendbuf, sendcount, sendtype,
103 recvbuf, recvcount, recvtype, comm,
104 request, comm->c_coll->coll_iallgather_module);
105
106 OMPI_ERRHANDLER_RETURN(err, comm, err, FUNC_NAME);
107 }