XGCa
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
domain_decomposition.hpp
Go to the documentation of this file.
1 #ifndef DOMAIN_DECOMPOSITION_HPP
2 #define DOMAIN_DECOMPOSITION_HPP
3 #include "space_settings.hpp"
4 #include "NamelistReader.hpp"
5 #ifdef USE_MPI
6 # include "my_mpi.hpp"
7 #endif
9 #include "grid_weights.hpp"
10 
11 // Decomposition class
12 template<class Device>
14  public:
15 
16  // Constructor
17  DomainDecomposition(NLReader::NamelistReader& nlr, int nplanes, int nnodes);
18 
19  DomainDecomposition(int nranks_in);
20 
21 #ifdef USE_MPI
22  // MPI communicators etc
23  const MyMPI mpi;
24 #endif
25 
26  // Updater
27  void update(int new_f0_inode1);
28 
29  void update_pol_decomp();
30 
31  KOKKOS_INLINE_FUNCTION int node_to_plane_rank(int node) const;
32 
33  KOKKOS_INLINE_FUNCTION void node_to_rank(const SimdGridWeights<Order::Zero, PIT_GLOBAL>& grid_wts0, const Simd<int>& iphi, Simd<int>& pid) const;
34 
35  // Constants
36  bool pol_decomp;
39  int n_pids;
40  int my_pid;
44 
47 
48  // Variables
50  int nnodes;
51 
52  // Views
53  Kokkos::View<int*,Kokkos::LayoutRight,HostType> gvid0_pid_h;
54  Kokkos::View<int*,Kokkos::LayoutRight,Device> gvid0_pid;
55 };
56 
57 #include "domain_decomposition.tpp"
58 #endif
bool decompose_fields
Whether to decompose fields.
Definition: domain_decomposition.hpp:45
int plane_totalpe
Processors per plane?
Definition: domain_decomposition.hpp:38
int plane_index
Offset of local plane.
Definition: domain_decomposition.hpp:42
Definition: grid_weights.hpp:51
Definition: NamelistReader.hpp:193
int node_offset
Offset of first mesh node belonging to this MPI rank.
Definition: domain_decomposition.hpp:49
void update_pol_decomp()
Definition: domain_decomposition.tpp:109
DomainDecomposition(NLReader::NamelistReader &nlr, int nplanes, int nnodes)
Definition: domain_decomposition.tpp:13
KOKKOS_INLINE_FUNCTION void node_to_rank(const SimdGridWeights< Order::Zero, PIT_GLOBAL > &grid_wts0, const Simd< int > &iphi, Simd< int > &pid) const
Definition: domain_decomposition.tpp:143
int nnodes
Number of nodes belonging to this MPI rank.
Definition: domain_decomposition.hpp:50
int my_pid
ID of this processor.
Definition: domain_decomposition.hpp:40
int pe_per_plane
Processors per plane.
Definition: domain_decomposition.hpp:37
Definition: my_mpi.hpp:19
int n_pids
Processors per plane?
Definition: domain_decomposition.hpp:39
int nnodes_on_plane
Number of nodes on local plane.
Definition: domain_decomposition.hpp:43
void update(int new_f0_inode1)
Definition: domain_decomposition.tpp:89
Definition: field_decomposition.hpp:12
Kokkos::View< int *, Kokkos::LayoutRight, Device > gvid0_pid
Which processors get which vertices.
Definition: domain_decomposition.hpp:54
int my_plane_pid
Poloidal ID of this processor.
Definition: domain_decomposition.hpp:41
Definition: domain_decomposition.hpp:13
FieldDecomposition< Device > field_decomp
Definition: domain_decomposition.hpp:46
bool pol_decomp
Use poloidal decomposition.
Definition: domain_decomposition.hpp:36
KOKKOS_INLINE_FUNCTION int node_to_plane_rank(int node) const
Definition: domain_decomposition.tpp:125
Kokkos::View< int *, Kokkos::LayoutRight, HostType > gvid0_pid_h
Which processors get which vertices (host)
Definition: domain_decomposition.hpp:53