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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
/*
* Copyright (C) by Argonne National Laboratory
* See COPYRIGHT in top-level directory
*/
/* Thanks to Jim Hoekstra of Iowa State University for several important
bug fixes */
#include "mpi.h"
#include "mpitestconf.h"
#include "mpitestcxx.h"
#ifdef HAVE_IOSTREAM
// Not all C++ compilers have iostream instead of iostream.h
#include <iostream>
#ifdef HAVE_NAMESPACE_STD
// Those that do often need the std namespace; otherwise, a bare "cout"
// is likely to fail to compile
using namespace std;
#endif
#else
#include <iostream.h>
#endif
/* Needed for strcmp */
#include <string.h>
/* Rather than deal with the need to declare strncpy correctly
(avoiding possible conflicts with other header files), we
simply define a copy routine that is used for this example */
void CopyChars(char *, int, const char *);
void CopyChars(char *dest, int len, const char *src)
{
int i;
for (i = 0; i < len && *src; i++) {
dest[i] = src[i];
}
if (i < len)
dest[i] = 0;
}
/*
* This is a simple program that tests bsend. It may be run as a single
* process to simplify debugging; in addition, bsend allows send-to-self
* programs.
*/
int main(int argc, char *argv[])
{
MPI::Intracomm comm;
int dest = 0, src = 0, tag = 1;
char *buf;
void *bbuf;
char msg1[7], msg3[17];
double msg2[2];
char rmsg1[64], rmsg3[64];
double rmsg2[64];
int errs = 0, rank;
int bufsize, bsize;
MPI::Status status;
MTest_Init();
comm = MPI::COMM_WORLD;
rank = comm.Get_rank();
bufsize = 3 * MPI_BSEND_OVERHEAD + 7 + 17 + 2 * sizeof(double);
buf = new char[bufsize];
MPI::Attach_buffer(buf, bufsize);
CopyChars(msg1, sizeof(msg1), "012345");
CopyChars(msg3, sizeof(msg3), "0123401234012341");
msg2[0] = 1.23;
msg2[1] = 3.21;
if (rank == src) {
/* These message sizes are chosen to expose any alignment problems */
comm.Bsend(msg1, 7, MPI::CHAR, dest, tag);
comm.Bsend(msg2, 2, MPI::DOUBLE, dest, tag);
comm.Bsend(msg3, 17, MPI::CHAR, dest, tag);
}
if (rank == dest) {
comm.Recv(rmsg1, 7, MPI::CHAR, src, tag, status);
comm.Recv(rmsg2, 10, MPI::DOUBLE, src, tag, status);
comm.Recv(rmsg3, 17, MPI::CHAR, src, tag, status);
if (strcmp(rmsg1, msg1) != 0) {
errs++;
cerr << "message 1 (" << rmsg1 << ") should be " << msg1 << "\n";
}
if (rmsg2[0] != msg2[0] || rmsg2[1] != msg2[1]) {
errs++;
cerr << "message 2 incorrect, values are (" << rmsg2[0] << ","
<< rmsg2[1] << ") but should be (" << msg2[0] << "," << msg2[1] << "\n";
}
if (strcmp(rmsg3, msg3) != 0) {
errs++;
cerr << "message 3 (" << rmsg3 << ") should be " << msg3 << "\n";
}
}
/* We can't guarantee that messages arrive until the detach */
bsize = MPI::Detach_buffer(bbuf);
if (bbuf != (void *) buf) {
errs++;
cerr << "Returned buffer from detach is not the same as the initial buffer\n";
}
delete[]buf;
MTest_Finalize(errs);
return 0;
}
|
