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)
16 int my_rank, shared_rank;
22 int shm_win_size = 1024 * 1024 * 1024 * sizeof(char); /* 1GB */
24 MPI_Init(&argc, &argv);
26 MPI_Comm_rank(MPI_COMM_WORLD, &my_rank);
28 for (i = 0; i < 2; i++) {
30 MPI_Info_create(&win_info);
31 MPI_Info_set(win_info, (char *) "alloc_shm", (char *) "true");
34 win_info = MPI_INFO_NULL;
37 MPI_Comm_split_type(MPI_COMM_WORLD, MPI_COMM_TYPE_SHARED, my_rank, MPI_INFO_NULL,
40 MPI_Comm_rank(shared_comm, &shared_rank);
42 /* every processes allocate 1GB window memory */
43 MPI_Win_allocate(shm_win_size, sizeof(char), win_info, MPI_COMM_WORLD, &mybase, &win);
47 MPI_Win_allocate_shared(shm_win_size, sizeof(char), win_info, shared_comm, &mybase, &win);
51 /* some processes allocate 1GB and some processes allocate zero bytes */
53 MPI_Win_allocate(shm_win_size, sizeof(char), win_info, MPI_COMM_WORLD, &mybase, &win);
55 MPI_Win_allocate(0, sizeof(char), win_info, MPI_COMM_WORLD, &mybase, &win);
59 if (shared_rank % 2 == 0)
60 MPI_Win_allocate_shared(shm_win_size, sizeof(char), win_info, shared_comm, &mybase,
63 MPI_Win_allocate_shared(0, sizeof(char), win_info, shared_comm, &mybase, &win);
67 /* some processes allocate 1GB and some processes allocate smaller bytes */
69 MPI_Win_allocate(shm_win_size, sizeof(char), win_info, MPI_COMM_WORLD, &mybase, &win);
71 MPI_Win_allocate(shm_win_size / 2, sizeof(char), win_info, MPI_COMM_WORLD, &mybase,
76 /* some processes allocate 1GB and some processes allocate smaller bytes */
77 if (shared_rank % 2 == 0)
78 MPI_Win_allocate_shared(shm_win_size, sizeof(char), win_info, shared_comm, &mybase,
81 MPI_Win_allocate_shared(shm_win_size / 2, sizeof(char), win_info, shared_comm, &mybase,
86 MPI_Comm_free(&shared_comm);
89 MPI_Info_free(&win_info);
93 printf(" No Errors\n");