FUnTiDES/src/discretization/fd/kernels/include/fd_macros.h at fdae1ab71fca18390fc2f636292f4123f3e62af9 · FUnTiDES-sim/FUnTiDES · 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
#ifndef FUNTIDES_DISCRETIZATION_FD_KERNELS_INCLUDE_FD_MACROS_H_
#define FUNTIDES_DISCRETIZATION_FD_KERNELS_INCLUDE_FD_MACROS_H_
#define POW2(x) ((x) * (x))
#define IDX3(i, j, k) (nz * ny * (i) + nz * (j) + (k))
#define IDX3_l(i, j, k)                                                      \
  ((nz + 2 * lz) * (ny + 2 * ly) * ((i) + lx) + (nz + 2 * lz) * ((j) + ly) + \
   ((k) + lz))
#define IDX3_eta1(i, j, k) \
  ((nz + 2) * (ny + 2) * ((i) + 1) + (nz + 2) * ((j) + 1) + ((k) + 1))

#define LOOP3DHEAD(x3, y3, z3, x4, y4, z4) Kokkos::parallel_for(Kokkos::MDRangePolicy<Kokkos::Rank<3>>({z3,x3,y3},{z4,x4,y4}),KOKKOS_LAMBDA(int k,int i,int j) {

#define LOOP3DEND \
  });

#define CREATEVIEWINNER              \
  vectorReal coefx = myModels.coefx; \
  vectorReal coefy = myModels.coefy; \
  vectorReal coefz = myModels.coefz; \
  vectorReal vp = myModels.vp;       \
  double coef0 = myModels.coef0;
#define CREATEVIEWPML            \
  CREATEVIEWINNER                \
  vectorReal phi = myModels.phi; \
  vectorReal eta = myModels.eta;
#define CREATEVIEWRHS                    \
  vectorReal RHSTerm = myModels.RHSTerm; \
  vectorReal pn = myModels.pn;           \
  vectorReal vp = myModels.vp;
#define CREATEVIEWSPONGE vectorReal spongeArray = myModels.spongeArray;
#define PN_Global pnGlobal

#define FDFENCE Kokkos::fence();
#endif  // FUNTIDES_DISCRETIZATION_FD_KERNELS_INCLUDE_FD_MACROS_H_