1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
3 * (C) 2001 by Argonne National Laboratory.
4 * See COPYRIGHT in top-level directory.
7 /* test MPI_WIN_ALLOCATE and MPI_WIN_ALLOCATE_SHARED when allocating
8 SHM memory with size of 1GB per process */
14 int main(int argc, char **argv) {
15 int my_rank, shared_rank;
20 int shm_win_size = 1024 * 1024 * 1024 * sizeof(char); /* 1GB */
22 MPI_Init(&argc, &argv);
24 MPI_Comm_rank(MPI_COMM_WORLD, &my_rank);
26 MPI_Info_create(&win_info);
27 MPI_Info_set(win_info, (char*)"alloc_shm", (char*)"true");
29 MPI_Comm_split_type(MPI_COMM_WORLD, MPI_COMM_TYPE_SHARED, my_rank, MPI_INFO_NULL, &shared_comm);
31 MPI_Comm_rank(shared_comm, &shared_rank);
33 /* every processes allocate 1GB window memory */
34 MPI_Win_allocate(shm_win_size, sizeof(char), win_info, MPI_COMM_WORLD, &mybase, &win);
38 MPI_Win_allocate_shared(shm_win_size, sizeof(char), win_info, shared_comm, &mybase, &win);
42 /* some processes allocate 1GB and some processes allocate zero bytes */
44 MPI_Win_allocate(shm_win_size, sizeof(char), win_info, MPI_COMM_WORLD, &mybase, &win);
46 MPI_Win_allocate(0, sizeof(char), win_info, MPI_COMM_WORLD, &mybase, &win);
50 if (shared_rank % 2 == 0)
51 MPI_Win_allocate_shared(shm_win_size, sizeof(char), win_info, shared_comm, &mybase, &win);
53 MPI_Win_allocate_shared(0, sizeof(char), win_info, shared_comm, &mybase, &win);
57 /* some processes allocate 1GB and some processes allocate smaller bytes */
59 MPI_Win_allocate(shm_win_size, sizeof(char), win_info, MPI_COMM_WORLD, &mybase, &win);
61 MPI_Win_allocate(shm_win_size/2, sizeof(char), win_info, MPI_COMM_WORLD, &mybase, &win);
65 /* some processes allocate 1GB and some processes allocate smaller bytes */
66 if (shared_rank % 2 == 0)
67 MPI_Win_allocate_shared(shm_win_size, sizeof(char), win_info, shared_comm, &mybase, &win);
69 MPI_Win_allocate_shared(shm_win_size/2, sizeof(char), win_info, shared_comm, &mybase, &win);
73 MPI_Comm_free(&shared_comm);
75 MPI_Info_free(&win_info);
78 printf(" No Errors\n");