1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
3 * (C) 2015 by Argonne National Laboratory.
4 * See COPYRIGHT in top-level directory.
7 /* This code tests the case when one process issues large number
8 * of MPI_Accumulate operations (with large derived datatype) and
9 * issues a MPI_Win_flush_local at end. */
11 /* FIXME: we should merge this into a comprehensive test for RMA
12 * operations + MPI_Win_flush_local. */
19 #define DATA_SIZE 1000000
21 #define BLOCKLENGTH (DATA_SIZE/COUNT)
22 #define STRIDE BLOCKLENGTH
25 int main(int argc, char *argv[])
32 MPI_Datatype derived_dtp;
35 MPI_Init(&argc, &argv);
37 MPI_Comm_size(MPI_COMM_WORLD, &nproc);
38 MPI_Comm_rank(MPI_COMM_WORLD, &rank);
41 fprintf(stderr, "Run this program with at least 3 processes\n");
42 MPI_Abort(MPI_COMM_WORLD, 1);
45 MPI_Alloc_mem(sizeof(int) * DATA_SIZE, MPI_INFO_NULL, &orig_buf);
46 MPI_Alloc_mem(sizeof(int) * DATA_SIZE, MPI_INFO_NULL, &tar_buf);
48 for (i = 0; i < DATA_SIZE; i++) {
53 MPI_Type_vector(COUNT, BLOCKLENGTH - 1, STRIDE, MPI_INT, &derived_dtp);
54 MPI_Type_commit(&derived_dtp);
56 MPI_Win_create(tar_buf, sizeof(int) * DATA_SIZE, sizeof(int),
57 MPI_INFO_NULL, MPI_COMM_WORLD, &win);
59 /***** test between rank 0 and rank 1 *****/
62 MPI_Win_lock(MPI_LOCK_SHARED, 0, 0, win);
64 for (i = 0; i < OPS_NUM; i++) {
65 MPI_Accumulate(orig_buf, 1, derived_dtp,
66 0, 0, DATA_SIZE - COUNT, MPI_INT, MPI_SUM, win);
67 MPI_Win_flush_local(0, win);
70 MPI_Win_unlock(0, win);
73 MPI_Barrier(MPI_COMM_WORLD);
77 for (i = 0; i < DATA_SIZE - COUNT; i++) {
78 if (tar_buf[i] != OPS_NUM) {
79 printf("tar_buf[%d] = %d, expected %d\n", i, tar_buf[i], OPS_NUM);
85 for (i = 0; i < DATA_SIZE; i++) {
89 MPI_Barrier(MPI_COMM_WORLD);
91 /***** test between rank 0 and rank 2 *****/
94 MPI_Win_lock(MPI_LOCK_SHARED, 0, 0, win);
96 for (i = 0; i < OPS_NUM; i++) {
97 MPI_Accumulate(orig_buf, 1, derived_dtp,
98 0, 0, DATA_SIZE - COUNT, MPI_INT, MPI_SUM, win);
99 MPI_Win_flush_local(0, win);
102 MPI_Win_unlock(0, win);
105 MPI_Barrier(MPI_COMM_WORLD);
109 for (i = 0; i < DATA_SIZE - COUNT; i++) {
110 if (tar_buf[i] != OPS_NUM) {
111 printf("tar_buf[%d] = %d, expected %d\n", i, tar_buf[i], OPS_NUM);
117 printf(" No Errors\n");
122 MPI_Type_free(&derived_dtp);
124 MPI_Free_mem(orig_buf);
125 MPI_Free_mem(tar_buf);