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
|
/* StarPU --- Runtime system for heterogeneous multicore architectures.
*
* Copyright (C) 2009, 2010 Université de Bordeaux 1
* Copyright (C) 2010, 2011 Centre National de la Recherche Scientifique
*
* StarPU is free software; you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation; either version 2.1 of the License, or (at
* your option) any later version.
*
* StarPU is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
*
* See the GNU Lesser General Public License in COPYING.LGPL for more details.
*/
#include <starpu_mpi.h>
#include "helper.h"
#define NITER 2048
#define SIZE 16
float *tab;
starpu_data_handle_t tab_handle;
int main(int argc, char **argv)
{
int ret, rank, size;
MPI_Init(NULL, NULL);
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
MPI_Comm_size(MPI_COMM_WORLD, &size);
if (size != 2)
{
if (rank == 0)
FPRINTF(stderr, "We need exactly 2 processes.\n");
MPI_Finalize();
return STARPU_TEST_SKIPPED;
}
ret = starpu_init(NULL);
STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
ret = starpu_mpi_initialize();
STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_initialize");
tab = malloc(SIZE*sizeof(float));
starpu_vector_data_register(&tab_handle, 0, (uintptr_t)tab, SIZE, sizeof(float));
unsigned nloops = NITER;
unsigned loop;
int other_rank = (rank + 1)%2;
for (loop = 0; loop < nloops; loop++)
{
if ((loop % 2) == rank)
{
starpu_mpi_send(tab_handle, other_rank, loop, MPI_COMM_WORLD);
}
else
{
MPI_Status status;
starpu_mpi_req req;
starpu_mpi_irecv(tab_handle, &req, other_rank, loop, MPI_COMM_WORLD);
starpu_mpi_wait(&req, &status);
}
}
starpu_mpi_shutdown();
starpu_shutdown();
MPI_Finalize();
return 0;
}
|
