Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Add jarfile generation script for github, using actions.
[simgrid.git] / teshsuite / smpi / macro-partial-shared-communication / macro-partial-shared-communication.c
1 /* Copyright (c) 2017-2020. The SimGrid Team. All rights reserved.          */
2
3 /* This program is free software; you can redistribute it and/or modify it
4  * under the terms of the license (GNU LGPL) which comes with this package. */
5
6 #include <stdio.h>
7 #include <mpi.h>
8 #include <stdint.h>
9 #include <inttypes.h>
10 #include <assert.h>
11
12 // Set the elements between buf[start] and buf[stop-1] to (i+value)%256
13 static void set(uint8_t *buf, size_t start, size_t stop, uint8_t value) {
14   for(size_t i = start; i < stop; i++) {
15     buf[i] = (uint8_t)((i + value) % 256);
16   }
17 }
18
19 // Return the number of times that an element is equal to (i+value)%256 between buf[start] and buf[stop-1].
20 static size_t count_all(const uint8_t* buf, size_t start, size_t stop, uint8_t value)
21 {
22   size_t occ = 0;
23   for(size_t i = start ; i < stop ; i++) {
24     if(buf[i] == (i+value)%256) {
25       occ ++;
26     }
27   }
28   return occ;
29 }
30
31 // Return true iff the values from buf[start] to buf[stop-1] are all equal to (i+value)%256.
32 static int check_all(const uint8_t* buf, size_t start, size_t stop, uint8_t value)
33 {
34   size_t occ = count_all(buf, start, stop, value);
35   return occ == stop-start;
36 }
37
38 int main(int argc, char *argv[])
39 {
40   MPI_Init(&argc, &argv);
41   int rank;
42   int size;
43   size_t mem_size = 0x1000000;
44   size_t shared_blocks[] = {
45     0,         0x123456,
46     0x130000, 0x130001,
47     0x345678, 0x345789,
48     0x444444, 0x555555,
49     0x555556, 0x560000,
50     0x800000, 0x1000000
51   };
52   int nb_blocks = (sizeof(shared_blocks)/sizeof(size_t))/2;
53   MPI_Comm_rank(MPI_COMM_WORLD, &rank);
54   MPI_Comm_size(MPI_COMM_WORLD, &size);
55   //Let's Allocate a shared memory buffer
56   assert(size%2 == 0);
57   uint8_t *buf;
58   buf = SMPI_PARTIAL_SHARED_MALLOC(mem_size, shared_blocks, nb_blocks);
59   memset(buf, rank, mem_size);
60   MPI_Barrier(MPI_COMM_WORLD);
61
62   // Even processes write their rank in private blocks
63   if(rank%2 == 0) {
64     for(int i = 0; i < nb_blocks-1; i++) {
65       size_t start = shared_blocks[2*i+1];
66       size_t stop = shared_blocks[2*i+2];
67       set(buf, start, stop, (uint8_t)rank);
68     }
69   }
70   // Then, even processes send their buffer to their successor
71   if(rank%2 == 0) {
72     MPI_Send(buf, (int)mem_size, MPI_UINT8_T, rank + 1, 0, MPI_COMM_WORLD);
73   }
74   else {
75     MPI_Recv(buf, (int)mem_size, MPI_UINT8_T, rank - 1, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
76   }
77
78
79   // Odd processes verify that they successfully received the message
80   if(rank%2 == 1) {
81     for(int i = 0; i < nb_blocks-1; i++) {
82       size_t start = shared_blocks[2*i+1];
83       size_t stop = shared_blocks[2*i+2];
84       int comm     = check_all(buf, start, stop, (uint8_t)(rank - 1));
85       printf("[%d] The result of the (normal) communication check for block (0x%zx, 0x%zx) is: %d\n", rank, start, stop, comm);
86     }
87     memset(buf, rank, mem_size);
88   }
89
90   MPI_Barrier(MPI_COMM_WORLD);
91
92   // Then, even processes send a sub-part of their buffer their successor
93   // Note that the last block should not be copied entirely
94   if(rank%2 == 0) {
95     MPI_Send(buf + 0x10000, (int)(mem_size - 0xa00000), MPI_UINT8_T, rank + 1, 0, MPI_COMM_WORLD);
96   }
97   else {
98     MPI_Recv(buf + 0x10000, (int)(mem_size - 0xa00000), MPI_UINT8_T, rank - 1, 0, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
99   }
100
101
102   // Odd processes verify that they successfully received the message
103   if(rank%2 == 1) {
104     for(int i = 0; i < nb_blocks-1; i++) {
105       size_t start = shared_blocks[2*i+1];
106       size_t stop = shared_blocks[2*i+2];
107       int comm     = check_all(buf, start, stop, (uint8_t)(rank - 1));
108       printf("[%d] The result of the (shifted) communication check for block (0x%zx, 0x%zx) is: %d\n", rank, start, stop, comm);
109     }
110   }
111
112   SMPI_SHARED_FREE(buf);
113
114   MPI_Finalize();
115   return 0;
116 }