1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
4 * (C) 2012 by Argonne National Laboratory.
5 * See COPYRIGHT in top-level directory.
17 int main(int argc, char **argv)
20 int errors = 0, all_errors = 0;
24 MPI_Init(&argc, &argv);
26 MPI_Comm_rank(MPI_COMM_WORLD, &rank);
27 MPI_Comm_size(MPI_COMM_WORLD, &nproc);
29 val_ptr = malloc(sizeof(int));
33 MPI_Win_create(val_ptr, sizeof(int), sizeof(int), MPI_INFO_NULL, MPI_COMM_WORLD, &win);
35 /* Test self communication */
37 for (i = 0; i < ITER; i++) {
38 int next = i + 1, result = -1;
39 MPI_Win_lock(MPI_LOCK_EXCLUSIVE, rank, 0, win);
40 MPI_Compare_and_swap(&next, &i, &result, MPI_INT, rank, 0, win);
41 MPI_Win_unlock(rank, win);
43 SQUELCH(printf("%d->%d -- Error: next=%d compare=%d result=%d val=%d\n", rank,
44 rank, next, i, result, *val_ptr););
49 MPI_Win_lock(MPI_LOCK_EXCLUSIVE, rank, 0, win);
51 MPI_Win_unlock(rank, win);
53 MPI_Barrier(MPI_COMM_WORLD);
55 /* Test neighbor communication */
57 for (i = 0; i < ITER; i++) {
58 int next = i + 1, result = -1;
59 MPI_Win_lock(MPI_LOCK_EXCLUSIVE, (rank + 1) % nproc, 0, win);
60 MPI_Compare_and_swap(&next, &i, &result, MPI_INT, (rank + 1) % nproc, 0, win);
61 MPI_Win_unlock((rank + 1) % nproc, win);
63 SQUELCH(printf("%d->%d -- Error: next=%d compare=%d result=%d val=%d\n", rank,
64 (rank + 1) % nproc, next, i, result, *val_ptr););
71 MPI_Barrier(MPI_COMM_WORLD);
72 MPI_Win_lock(MPI_LOCK_EXCLUSIVE, rank, 0, win);
74 MPI_Win_unlock(rank, win);
75 MPI_Barrier(MPI_COMM_WORLD);
81 for (i = 0; i < ITER; i++) {
82 int next = i + 1, result = -1;
83 MPI_Win_lock(MPI_LOCK_EXCLUSIVE, 0, 0, win);
84 MPI_Compare_and_swap(&next, &i, &result, MPI_INT, 0, 0, win);
85 MPI_Win_unlock(0, win);
89 MPI_Barrier(MPI_COMM_WORLD);
91 if (rank == 0 && nproc > 1) {
92 if (*val_ptr != ITER) {
93 SQUELCH(printf("%d - Error: expected=%d val=%d\n", rank, ITER, *val_ptr););
100 MPI_Reduce(&errors, &all_errors, 1, MPI_INT, MPI_SUM, 0, MPI_COMM_WORLD);
102 if (rank == 0 && all_errors == 0)
103 printf(" No Errors\n");