1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
3 * (C) 2014 by Argonne National Laboratory.
4 * See COPYRIGHT in top-level directory.
11 #define GET_SIZE 100000
12 #define WIN_SIZE (PUT_SIZE+GET_SIZE)
15 int main(int argc, char **argv)
18 int i, k, rank, nproc;
19 int win_buf[WIN_SIZE], orig_get_buf[GET_SIZE], orig_put_buf[PUT_SIZE];
20 int orig_rank = 0, dest_rank = 1;
22 MPI_Group comm_group, orig_group, dest_group;
24 MPI_Init(&argc, &argv);
26 MPI_Comm_rank(MPI_COMM_WORLD, &rank);
27 MPI_Comm_size(MPI_COMM_WORLD, &nproc);
29 MPI_Comm_group(MPI_COMM_WORLD, &comm_group);
30 MPI_Group_incl(comm_group, 1, &orig_rank, &orig_group);
31 MPI_Group_incl(comm_group, 1, &dest_rank, &dest_group);
33 for (i = 0; i < PUT_SIZE; i++)
35 for (i = 0; i < GET_SIZE; i++)
37 for (i = 0; i < WIN_SIZE; i++)
40 MPI_Win_create(win_buf, sizeof(int) * WIN_SIZE, sizeof(int), MPI_INFO_NULL,
41 MPI_COMM_WORLD, &win);
43 for (k = 0; k < LOOP; k++) {
47 if (rank == orig_rank) {
48 MPI_Win_fence(MPI_MODE_NOPRECEDE, win);
49 MPI_Get(orig_get_buf, GET_SIZE, MPI_INT,
50 dest_rank, PUT_SIZE /*disp */ , GET_SIZE, MPI_INT, win);
51 MPI_Put(orig_put_buf, PUT_SIZE, MPI_INT,
52 dest_rank, 0 /*disp */ , PUT_SIZE, MPI_INT, win);
53 MPI_Win_fence(MPI_MODE_NOSUCCEED, win);
55 /* check GET result values */
56 for (i = 0; i < GET_SIZE; i++) {
57 if (orig_get_buf[i] != 1) {
58 printf("LOOP=%d, FENCE, orig_get_buf[%d] = %d, expected 1.\n",
59 k, i, orig_get_buf[i]);
64 else if (rank == dest_rank) {
65 MPI_Win_fence(MPI_MODE_NOPRECEDE, win);
66 MPI_Win_fence(MPI_MODE_NOSUCCEED, win);
68 /* modify the last element in window */
69 MPI_Win_lock(MPI_LOCK_SHARED, rank, 0, win);
70 win_buf[WIN_SIZE - 1] = 2;
71 MPI_Win_unlock(rank, win);
74 MPI_Barrier(MPI_COMM_WORLD);
76 for (i = 0; i < PUT_SIZE; i++)
78 for (i = 0; i < GET_SIZE; i++)
80 MPI_Win_lock(MPI_LOCK_SHARED, rank, 0, win);
81 for (i = 0; i < WIN_SIZE; i++)
83 MPI_Win_unlock(rank, win);
84 MPI_Barrier(MPI_COMM_WORLD);
88 if (rank == orig_rank) {
89 MPI_Win_start(dest_group, 0, win);
90 MPI_Get(orig_get_buf, GET_SIZE, MPI_INT,
91 dest_rank, PUT_SIZE /*disp */ , GET_SIZE, MPI_INT, win);
92 MPI_Put(orig_put_buf, PUT_SIZE, MPI_INT,
93 dest_rank, 0 /*disp */ , PUT_SIZE, MPI_INT, win);
94 MPI_Win_complete(win);
96 /* check GET result values */
97 for (i = 0; i < GET_SIZE; i++) {
98 if (orig_get_buf[i] != 1) {
99 printf("LOOP=%d, PSCW, orig_get_buf[%d] = %d, expected 1.\n",
100 k, i, orig_get_buf[i]);
105 else if (rank == dest_rank) {
106 MPI_Win_post(orig_group, 0, win);
109 /* modify the last element in window */
110 MPI_Win_lock(MPI_LOCK_SHARED, rank, 0, win);
111 win_buf[WIN_SIZE - 1] = 2;
112 MPI_Win_unlock(rank, win);
115 MPI_Barrier(MPI_COMM_WORLD);
117 for (i = 0; i < PUT_SIZE; i++)
119 for (i = 0; i < GET_SIZE; i++)
121 MPI_Win_lock(MPI_LOCK_SHARED, rank, 0, win);
122 for (i = 0; i < WIN_SIZE; i++)
124 MPI_Win_unlock(rank, win);
125 MPI_Barrier(MPI_COMM_WORLD);
130 MPI_Group_free(&orig_group);
131 MPI_Group_free(&dest_group);
132 MPI_Group_free(&comm_group);
134 if (rank == orig_rank && errors == 0) {
135 printf(" No Errors\n");