MueLu  Version of the Day
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
MueLu_IndexManager_kokkos_decl.hpp
Go to the documentation of this file.
1 // @HEADER
2 //
3 // ***********************************************************************
4 //
5 // MueLu: A package for multigrid based preconditioning
6 // Copyright 2012 Sandia Corporation
7 //
8 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
9 // the U.S. Government retains certain rights in this software.
10 //
11 // Redistribution and use in source and binary forms, with or without
12 // modification, are permitted provided that the following conditions are
13 // met:
14 //
15 // 1. Redistributions of source code must retain the above copyright
16 // notice, this list of conditions and the following disclaimer.
17 //
18 // 2. Redistributions in binary form must reproduce the above copyright
19 // notice, this list of conditions and the following disclaimer in the
20 // documentation and/or other materials provided with the distribution.
21 //
22 // 3. Neither the name of the Corporation nor the names of the
23 // contributors may be used to endorse or promote products derived from
24 // this software without specific prior written permission.
25 //
26 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
27 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
28 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
30 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
31 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
32 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
33 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
34 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
35 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
36 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
37 //
38 // Questions? Contact
39 // Jonathan Hu (jhu@sandia.gov)
40 // Ray Tuminaro (rstumin@sandia.gov)
41 // Luc Berger-Vergiat (lberge@sandia.gov)
42 //
43 // ***********************************************************************
44 //
45 // @HEADER
46 #ifndef MUELU_INDEXMANAGER_KOKKOS_DECL_HPP
47 #define MUELU_INDEXMANAGER_KOKKOS_DECL_HPP
48 
49 #include "MueLu_ConfigDefs.hpp"
50 #include "MueLu_Types.hpp"
51 
52 #ifdef HAVE_MUELU_KOKKOS_REFACTOR
53 #include <KokkosCompat_ClassicNodeAPI_Wrapper.hpp>
54 #include <Kokkos_View.hpp>
55 
57 
58 #include <Xpetra_Map_fwd.hpp>
59 #include <Xpetra_Vector_fwd.hpp>
60 #include <Xpetra_VectorFactory_fwd.hpp>
61 
62 #include "MueLu_BaseClass.hpp"
64 
65 /*****************************************************************************
66 
67 ****************************************************************************/
68 
69 namespace MueLu {
70 
82  template <class LocalOrdinal, class GlobalOrdinal, class Node>
83  class IndexManager_kokkos : public BaseClass {
84 #undef MUELU_INDEXMANAGER_KOKKOS_SHORT
86 
87  public:
88  typedef typename Node::execution_space execution_space;
89  typedef typename execution_space::memory_space memory_space;
90  typedef typename Kokkos::View<int[3], memory_space> intTupleView;
91  typedef typename Kokkos::View<LO[3], memory_space> LOTupleView;
92 
93  private:
94 
95  const int meshLayout = UNCOUPLED;
96  int myRank = -1;
97  int numDimensions;
98  int interpolationOrder_;
99  intTupleView coarseRate;
100  intTupleView endRate;
101 
102  LO lNumFineNodes;
103  LO lNumFineNodes10;
104  LOTupleView lFineNodesPerDir;
105 
106  LO numCoarseNodes;
107  LO numCoarseNodes10;
108  LOTupleView coarseNodesPerDir;
109 
110  public:
111 
113  IndexManager_kokkos() = default;
114 
116  IndexManager_kokkos(const int NumDimensions,
117  const int interpolationOrder,
118  const int MyRank,
119  const ArrayView<const LO> LFineNodesPerDir,
120  const ArrayView<const int> CoarseRate);
121 
122  virtual ~IndexManager_kokkos() {}
123 
125  void setupIM(const int NumDimensions,
126  const int interpolationOrder,
127  const ArrayView<const int> coarseRate,
128  const ArrayView<const LO> LFineNodesPerDir);
129 
132  void computeMeshParameters();
133 
134  int getNumDimensions() const {return numDimensions;}
135 
136  int getInterpolationOrder() const {return interpolationOrder_;}
137 
138  LO getNumLocalFineNodes() const {return lNumFineNodes;}
139 
140  LO getNumCoarseNodes() const {return numCoarseNodes;}
141 
142  KOKKOS_INLINE_FUNCTION
143  intTupleView getCoarseningRates() const {return coarseRate;}
144 
145  KOKKOS_INLINE_FUNCTION
146  intTupleView getCoarseningEndRates() const {return endRate;}
147 
148  KOKKOS_INLINE_FUNCTION
149  LOTupleView getLocalFineNodesPerDir() const {return lFineNodesPerDir;}
150 
151  KOKKOS_INLINE_FUNCTION
152  LOTupleView getCoarseNodesPerDir() const {return coarseNodesPerDir;}
153 
154  Array<LO> getCoarseNodesPerDirArray() const;
155 
156  KOKKOS_INLINE_FUNCTION
157  void getFineLID2FineTuple(const LO myLID, LO (&tuple)[3]) const {
158  LO tmp;
159  tuple[2] = myLID / (lFineNodesPerDir(1)*lFineNodesPerDir(0));
160  tmp = myLID % (lFineNodesPerDir(1)*lFineNodesPerDir(0));
161  tuple[1] = tmp / lFineNodesPerDir(0);
162  tuple[0] = tmp % lFineNodesPerDir(0);
163  } // getFineNodeLocalTuple
164 
165  KOKKOS_INLINE_FUNCTION
166  void getFineTuple2FineLID(const LO tuple[3], LO& myLID) const {
167  myLID = tuple[2]*lNumFineNodes10 + tuple[1]*lFineNodesPerDir[0] + tuple[0];
168  } // getFineNodeLID
169 
170  KOKKOS_INLINE_FUNCTION
171  void getCoarseLID2CoarseTuple(const LO myLID, LO (&tuple)[3]) const {
172  LO tmp;
173  tuple[2] = myLID / numCoarseNodes10;
174  tmp = myLID % numCoarseNodes10;
175  tuple[1] = tmp / coarseNodesPerDir[0];
176  tuple[0] = tmp % coarseNodesPerDir[0];
177  } // getCoarseNodeLocalTuple
178 
179  KOKKOS_INLINE_FUNCTION
180  void getCoarseTuple2CoarseLID(const LO i, const LO j, const LO k, LO& myLID) const {
181  myLID = k*numCoarseNodes10 + j*coarseNodesPerDir[0] + i;
182  } // getCoarseNodeLID
183 
184  };
185 
186 } //namespace MueLu
187 
188 #define MUELU_INDEXMANAGER_KOKKOS_SHORT
189 #endif // HAVE_MUELU_KOKKOS_REFACTOR
190 #endif // MUELU_INDEXMANAGER_KOKKOS_DECL_HPP