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
113
114
115
116
|
#if HAVE_CONFIG_H
# include "config.h"
#endif
/* $Id: ipctest.c,v 1.2 1999-07-28 00:47:55 d3h325 Exp $ */
#if HAVE_STDIO_H
# include <stdio.h>
#endif
#include <mpi.h>
#if HAVE_ASSERT_H
# include <assert.h>
#endif
#include "armcip.h"
#include "armci_shmem.h"
#include "locks.h"
#include "message.h"
int me, nproc;
void test()
{
double *a = NULL, start = 1., end = -1.;
int len = 100;
long size = len * sizeof(double);
long idlist[SHMIDLEN];
int numlock = 10, i;
lockset_t lockid;
/* shared memory test */
if (me == 0) {
printf("Test shared memory\n");
a = (double *)Create_Shared_Region(idlist + 1, size, idlist);
assert(a);
a[0] = start;
a[len-1] = end;
}
MPI_Bcast(idlist, SHMIDLEN, MPI_LONG, 0, MPI_COMM_WORLD);
if (me) {
a = (double *)Attach_Shared_Region(idlist + 1, size, idlist[0]);
assert(a);
}
if (me == nproc - 1) {
printf("%d: start=%f end=%f\n", me, a[0], a[len-1]);
if (a[0] == start && a[len-1] == end) {
printf("Works!\n");
}
}
/*printf("%d: a=%x\n",me,a); */
MPI_Barrier(MPI_COMM_WORLD);
/* allocate locks */
if (me == 0) {
a[0] = 0.;
CreateInitLocks(numlock, &lockid);
printf("\nMutual exclusion test\n");
}
MPI_Bcast(&lockid, sizeof(lockid), MPI_BYTE, 0, MPI_COMM_WORLD);
if (me) {
InitLocks(numlock, lockid);
}
/* mutual exclusion test:
* everybody increments shared variable 1000 times
*/
# define TIMES 1000
MPI_Barrier(MPI_COMM_WORLD);
for (i = 0; i < TIMES; i++) {
NATIVE_LOCK(0, me);
a[0]++;
NATIVE_UNLOCK(0, me);
}
MPI_Barrier(MPI_COMM_WORLD);
if (me == nproc - 1) {
printf("value of shared variable =%f should be %f\n",
a[0], 1.0 * nproc * TIMES);
if (a[0] == 1.0 * nproc * TIMES) {
printf("Works!\n\n");
}
}
/* cleanup of IPC resources */
if (me == 0) {
DeleteLocks(lockid);
Delete_All_Regions();
}
MPI_Barrier(MPI_COMM_WORLD);
}
int main(int argc, char **argv)
{
armci_msg_init(&argc, &argv);
ARMCI_Init_args(&argc, &argv);
MPI_Comm_size(MPI_COMM_WORLD, &nproc);
MPI_Comm_rank(MPI_COMM_WORLD, &me);
if (me == 0) {
printf("Testing IPCs (%d MPI processes)\n\n", nproc);
}
test();
ARMCI_Finalize();
armci_msg_finalize();
return 0;
}
|
