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 }