large-small-acc.c 2.3 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
/*
 *  (C) 2015 by Argonne National Laboratory.
 *      See COPYRIGHT in top-level directory.
 */

/* This code tests the case when a large ACC is issued, and then
 * several small ACCs is issued between the same origin and target.
 * The purpose of this test is to check if the ordering of ACCs
 * is guaranteed. */

#include "mpi.h"
#include <stdio.h>
#include <stdint.h>

#define LOOP 5
#define DATA_COUNT 8192

int main(int argc, char *argv[]){
    int rank, nprocs;
    MPI_Win win;
    uint64_t buf[DATA_COUNT], orig_buf[DATA_COUNT];
    uint64_t small_orig_buf_1 = 2, small_orig_buf_2[2] = {3, 3};
    int i, j, error = 0;

    MPI_Init(&argc, &argv);
    MPI_Comm_size(MPI_COMM_WORLD, &nprocs);
    MPI_Comm_rank(MPI_COMM_WORLD, &rank);

    for (j = 0; j < LOOP; j++) {

        error = 0;

        for (i = 0; i < DATA_COUNT; i++) {
            buf[i] = 0;
            orig_buf[i] = 1;
        }

        MPI_Win_create(buf, sizeof(uint64_t)*DATA_COUNT, sizeof(uint64_t),
                       MPI_INFO_NULL, MPI_COMM_WORLD, &win);

        MPI_Win_fence(0, win);

        if (rank == 0) {
            /* ACC (atomic PUT) to win_buf[0...DATA_COUNT-1] */
            MPI_Accumulate(orig_buf, DATA_COUNT, MPI_UINT64_T, 1, 0, DATA_COUNT, MPI_UINT64_T, MPI_REPLACE, win);
            /* ACC (atomic PUT) to win_buf[0] */
            MPI_Accumulate(&small_orig_buf_1, 1, MPI_UINT64_T, 1, 0, 1, MPI_UINT64_T, MPI_REPLACE, win);
            /* ACC (atomic PUT) to win_buf[1,2] */
            MPI_Accumulate(&small_orig_buf_2, 2, MPI_UINT64_T, 1, 1, 2, MPI_UINT64_T, MPI_REPLACE, win);
        }

        MPI_Win_fence(0, win);

        if (rank == 1) {
            for (i = 0; i < DATA_COUNT; i++) {
                if (i == 0) {
                    if (buf[i] != 2) {
                        error++;
                    }
                }
                else if (i == 1 || i == 2) {
                    if (buf[i] != 3) {
                        error++;
                    }
                }
                else {
                    if (buf[i] != 1) {
                        error++;
                    }
                }
            }
        }

        MPI_Win_free(&win);
    }

    if (rank == 1 && error == 0) {
        printf(" No Errors\n");
    }

    MPI_Finalize();
}