1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
4 * (C) 2003 by Argonne National Laboratory.
5 * See COPYRIGHT in top-level directory.
13 #define MAX_INT 0x7fffffff
17 static char MTEST_Descrip[] = "Test MPI_Accumulate with fence";
20 int main( int argc, char *argv[] )
23 int rank, size, source, dest;
24 int minsize = 2, count, i;
27 MPI_Datatype datatype;
30 MTest_Init( &argc, &argv );
32 /* The following illustrates the use of the routines to
33 run through a selection of communicators and datatypes.
34 Use subsets of these for tests that do not involve combinations
35 of communicators, datatypes, and counts of datatypes */
36 while (MTestGetIntracommGeneral( &comm, minsize, 1 )) {
37 if (comm == MPI_COMM_NULL) continue;
38 /* Determine the sender and receiver */
39 MPI_Comm_rank( comm, &rank );
40 MPI_Comm_size( comm, &size );
44 for (count = 1; count < 65000; count = count * 2) {
46 /* We compare with an integer value that can be as large as
47 size * (count * count + (1/2)*(size-1))
48 For large machines (size large), this can exceed the
49 maximum integer for some large values of count. We check
50 that in advance and break this loop if the above value
51 would exceed MAX_INT. Specifically,
53 size*count*count + (1/2)*size*(size-1) > MAX_INT
54 count*count > (MAX_INT/size - (1/2)*(size-1))
56 if (count * count > (MAX_INT/size - (size-1)/2)) break;
57 winbuf = (int *)malloc( count * sizeof(int) );
58 sbuf = (int *)malloc( count * sizeof(int) );
60 for (i=0; i<count; i++) winbuf[i] = 0;
61 for (i=0; i<count; i++) sbuf[i] = rank + i * count;
62 MPI_Win_create( winbuf, count * sizeof(int), sizeof(int),
63 MPI_INFO_NULL, comm, &win );
64 MPI_Win_fence( 0, win );
65 MPI_Accumulate( sbuf, count, MPI_INT, source, 0, count, MPI_INT,
67 MPI_Win_fence( 0, win );
69 /* Check the results */
70 for (i=0; i<count; i++) {
71 int result = i * count * size + (size*(size-1))/2;
72 if (winbuf[i] != result) {
74 fprintf( stderr, "Winbuf[%d] = %d, expected %d (count = %d, size = %d)\n",
75 i, winbuf[i], result, count, size );
88 MTest_Finalize( errs );