1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
4 * (C) 2014 by Argonne National Laboratory.
5 * See COPYRIGHT in top-level directory.
13 #define MAX_SIZE 65536
15 int main(int argc, char *argv[])
18 int errors = 0, all_errors = 0;
19 int *buf = NULL, *winbuf = NULL;
22 MPI_Init(&argc, &argv);
23 MPI_Comm_rank(MPI_COMM_WORLD, &rank);
24 MPI_Comm_size(MPI_COMM_WORLD, &nproc);
28 printf("Error: must be run with two or more processes\n");
29 MPI_Abort(MPI_COMM_WORLD, 1);
32 MPI_Alloc_mem(MAX_SIZE * sizeof(int), MPI_INFO_NULL, &buf);
33 MPI_Alloc_mem(MAX_SIZE * sizeof(int), MPI_INFO_NULL, &winbuf);
34 MPI_Win_create(winbuf, MAX_SIZE * sizeof(int), sizeof(int), MPI_INFO_NULL,
35 MPI_COMM_WORLD, &window);
37 MPI_Win_lock_all(0, window);
39 /* Test Raccumulate local completion with small data.
40 * Small data is always copied to header packet as immediate data. */
42 for (i = 0; i < ITER; i++) {
47 MPI_Raccumulate(&buf[0], 1, MPI_INT, 0, 0, 1, MPI_INT, MPI_MAX, window, &acc_req);
48 MPI_Wait(&acc_req, MPI_STATUS_IGNORE);
50 /* reset local buffer to check local completion */
52 MPI_Win_flush(0, window);
54 MPI_Get(&val, 1, MPI_INT, 0, 0, 1, MPI_INT, window);
55 MPI_Win_flush(0, window);
57 if (val != rank * i) {
58 printf("%d - Got %d in small Raccumulate test, expected %d (%d * %d)\n", rank, val,
65 MPI_Barrier(MPI_COMM_WORLD);
67 /* Test Raccumulate local completion with large data .
68 * Large data is not suitable for 1-copy optimization, and always sent out
69 * from user buffer. */
71 for (i = 0; i < ITER; i++) {
73 int val0 = -1, val1 = -1, val2 = -1;
77 for (j = 0; j < MAX_SIZE; j++) {
78 buf[j] = rank + j + i;
81 MPI_Raccumulate(buf, MAX_SIZE, MPI_INT, 0, 0, MAX_SIZE, MPI_INT, MPI_REPLACE, window,
83 MPI_Wait(&acc_req, MPI_STATUS_IGNORE);
85 /* reset local buffer to check local completion */
87 buf[MAX_SIZE - 1] = 0;
88 buf[MAX_SIZE / 2] = 0;
89 MPI_Win_flush(0, window);
91 /* get remote values which are modified in local buffer after wait */
92 MPI_Get(&val0, 1, MPI_INT, 0, 0, 1, MPI_INT, window);
93 MPI_Get(&val1, 1, MPI_INT, 0, MAX_SIZE - 1, 1, MPI_INT, window);
94 MPI_Get(&val2, 1, MPI_INT, 0, MAX_SIZE / 2, 1, MPI_INT, window);
95 MPI_Win_flush(0, window);
97 if (val0 != rank + i) {
98 printf("%d - Got %d in large Raccumulate test, expected %d\n", rank,
102 if (val1 != rank + MAX_SIZE - 1 + i) {
103 printf("%d - Got %d in large Raccumulate test, expected %d\n", rank,
104 val1, rank + MAX_SIZE - 1 + i);
107 if (val2 != rank + MAX_SIZE / 2 + i) {
108 printf("%d - Got %d in large Raccumulate test, expected %d\n", rank,
109 val2, rank + MAX_SIZE / 2 + i);
115 MPI_Win_unlock_all(window);
116 MPI_Barrier(MPI_COMM_WORLD);
118 MPI_Win_free(&window);
122 MPI_Free_mem(winbuf);
124 MPI_Reduce(&errors, &all_errors, 1, MPI_INT, MPI_SUM, 0, MPI_COMM_WORLD);
126 if (rank == 0 && all_errors == 0)
127 printf(" No Errors\n");