cvc/mpi_init.h at master · marcuspetschlies/cvc · GitHub

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
84
85
#ifndef _MPI_INIT_H
#define _MPI_INIT_H

namespace cvc {

#  ifdef HAVE_MPI
#    include <mpi.h>
#  endif

#  ifdef HAVE_MPI
extern MPI_Datatype gauge_point;
extern MPI_Datatype spinor_point;
extern MPI_Datatype contraction_point;


extern MPI_Datatype gauge_time_slice_cont;
extern MPI_Datatype spinor_time_slice_cont;
extern MPI_Datatype contraction_time_slice_cont;

#  if defined PARALLELTX || defined PARALLELTXY || defined PARALLELTXYZ

/* gauge slices */
extern MPI_Datatype gauge_x_slice_vector;
extern MPI_Datatype gauge_x_slice_cont;
extern MPI_Datatype gauge_x_subslice_cont;

extern MPI_Datatype gauge_y_slice_vector;
extern MPI_Datatype gauge_y_subslice_cont;
extern MPI_Datatype gauge_y_slice_cont;

extern MPI_Datatype gauge_z_slice_vector;
extern MPI_Datatype gauge_z_subslice_cont;
extern MPI_Datatype gauge_z_slice_cont;
/* spinor slices */

extern MPI_Datatype spinor_x_slice_vector;
extern MPI_Datatype spinor_x_slice_cont;
extern MPI_Datatype spinor_x_subslice_cont;

extern MPI_Datatype spinor_y_slice_vector;
extern MPI_Datatype spinor_y_subslice_cont;
extern MPI_Datatype spinor_y_slice_cont;

extern MPI_Datatype spinor_z_slice_vector;
extern MPI_Datatype spinor_z_subslice_cont;
extern MPI_Datatype spinor_z_slice_cont;

/* edges */

extern MPI_Datatype gauge_xt_edge_vector;
extern MPI_Datatype gauge_xt_edge_cont;
extern MPI_Datatype gauge_yt_edge_vector;
extern MPI_Datatype gauge_yt_edge_cont;
extern MPI_Datatype gauge_yx_edge_vector;
extern MPI_Datatype gauge_yx_edge_cont;

extern MPI_Datatype gauge_zt_edge_vector;
extern MPI_Datatype gauge_zt_edge_cont;
extern MPI_Datatype gauge_zx_edge_vector;
extern MPI_Datatype gauge_zx_edge_cont;
extern MPI_Datatype gauge_zy_edge_vector;
extern MPI_Datatype gauge_zy_edge_cont;

extern MPI_Datatype contraction_x_slice_vector;
extern MPI_Datatype contraction_x_subslice_cont;
extern MPI_Datatype contraction_x_slice_cont;

extern MPI_Datatype contraction_y_slice_vector;
extern MPI_Datatype contraction_y_subslice_cont;
extern MPI_Datatype contraction_y_slice_cont;

extern MPI_Datatype contraction_z_slice_vector;
extern MPI_Datatype contraction_z_subslice_cont;
extern MPI_Datatype contraction_z_slice_cont;

#  endif
#  endif

void mpi_init(int argc, char *argv[]);

void mpi_init_xchange_contraction(int N);
void mpi_fini_xchange_contraction(void);
}
#endif