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) {
18 int i, j, rank, nproc;
19 int errors = 0, all_errors = 0;
23 MPI_Init(&argc, &argv);
25 MPI_Comm_rank(MPI_COMM_WORLD, &rank);
26 MPI_Comm_size(MPI_COMM_WORLD, &nproc);
28 val_ptr = malloc(sizeof(int));
32 MPI_Win_create(val_ptr, sizeof(int), sizeof(int), MPI_INFO_NULL, MPI_COMM_WORLD, &win);
34 /* Test self communication */
36 for (i = 0; i < ITER; i++) {
37 int next = i + 1, result = -1;
38 MPI_Win_lock(MPI_LOCK_EXCLUSIVE, rank, 0, win);
39 MPI_Compare_and_swap(&next, &i, &result, MPI_INT, rank, 0, win);
40 MPI_Win_unlock(rank, win);
42 SQUELCH( printf("%d->%d -- Error: next=%d compare=%d result=%d val=%d\n", rank,
43 rank, next, i, result, *val_ptr); );
48 MPI_Win_lock(MPI_LOCK_EXCLUSIVE, rank, 0, win);
50 MPI_Win_unlock(rank, win);
52 MPI_Barrier(MPI_COMM_WORLD);
54 /* Test neighbor communication */
56 for (i = 0; i < ITER; i++) {
57 int next = i + 1, result = -1;
58 MPI_Win_lock(MPI_LOCK_EXCLUSIVE, (rank+1)%nproc, 0, win);
59 MPI_Compare_and_swap(&next, &i, &result, MPI_INT, (rank+1)%nproc, 0, win);
60 MPI_Win_unlock((rank+1)%nproc, win);
62 SQUELCH( printf("%d->%d -- Error: next=%d compare=%d result=%d val=%d\n", rank,
63 (rank+1)%nproc, next, i, result, *val_ptr); );
70 MPI_Barrier(MPI_COMM_WORLD);
71 MPI_Win_lock(MPI_LOCK_EXCLUSIVE, rank, 0, win);
73 MPI_Win_unlock(rank, win);
74 MPI_Barrier(MPI_COMM_WORLD);
80 for (i = 0; i < ITER; i++) {
81 int next = i + 1, result = -1;
82 MPI_Win_lock(MPI_LOCK_EXCLUSIVE, 0, 0, win);
83 MPI_Compare_and_swap(&next, &i, &result, MPI_INT, 0, 0, win);
84 MPI_Win_unlock(0, win);
88 MPI_Barrier(MPI_COMM_WORLD);
90 if (rank == 0 && nproc > 1) {
91 if (*val_ptr != ITER) {
92 SQUELCH( printf("%d - Error: expected=%d val=%d\n", rank, ITER, *val_ptr); );
99 MPI_Reduce(&errors, &all_errors, 1, MPI_INT, MPI_SUM, 0, MPI_COMM_WORLD);
101 if (rank == 0 && all_errors == 0)
102 printf(" No Errors\n");