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