MueLu  Version of the Day
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
MueLu_TentativePFactory_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 // Andrey Prokopenko (aprokop@sandia.gov)
41 // Ray Tuminaro (rstumin@sandia.gov)
42 //
43 // ***********************************************************************
44 //
45 // @HEADER
46 #ifndef MUELU_TENTATIVEPFACTORY_KOKKOS_DECL_HPP
47 #define MUELU_TENTATIVEPFACTORY_KOKKOS_DECL_HPP
48 
49 #include "MueLu_ConfigDefs.hpp"
50 
52 
53 #include <Tpetra_KokkosCompat_ClassicNodeAPI_Wrapper.hpp>
54 
55 #include "Teuchos_ScalarTraits.hpp"
56 
58 
59 #include "MueLu_Aggregates_fwd.hpp"
61 #include "MueLu_Level_fwd.hpp"
62 #include "MueLu_PerfUtils_fwd.hpp"
63 #include "MueLu_PFactory.hpp"
64 
65 namespace MueLu {
66 
105 template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
107  public:
110  typedef typename Node::execution_space execution_space;
111  typedef Kokkos::RangePolicy<local_ordinal_type, execution_space> range_type;
112  typedef typename Node::device_type DeviceType;
113  typedef Node node_type;
116 
117  private:
118 #undef MUELU_TENTATIVEPFACTORY_KOKKOS_SHORT
119 #include "MueLu_UseShortNames.hpp"
120 
121  public:
123 
124 
127 
131 
133 
135 
136 
137  void DeclareInput(Level& fineLevel, Level& coarseLevel) const;
138 
140 
142 
143 
144  void Build(Level& fineLevel, Level& coarseLevel) const;
145  void BuildP(Level& fineLevel, Level& coarseLevel) const;
146 
148 
149  // NOTE: All of thess should really be private, but CUDA doesn't like that
150 
151  void BuildPuncoupledBlockCrs(Level& coarseLevel, RCP<Matrix> A, RCP<Aggregates> aggregates, RCP<AmalgamationInfo> amalgInfo,
152  RCP<MultiVector> fineNullspace, RCP<const Map> coarseMap, RCP<Matrix>& Ptentative, RCP<MultiVector>& coarseNullspace, const int levelID) const;
153 
154  bool isGoodMap(const Map& rowMap, const Map& colMap) const;
155 
156  void BuildPcoupled(RCP<Matrix> A, RCP<Aggregates> aggregates,
157  RCP<AmalgamationInfo> amalgInfo, RCP<MultiVector> fineNullspace,
158  RCP<const Map> coarseMap, RCP<Matrix>& Ptentative,
159  RCP<MultiVector>& coarseNullspace) const;
160 
161  void BuildPuncoupled(Level& coarseLevel, RCP<Matrix> A, RCP<Aggregates> aggregates,
162  RCP<AmalgamationInfo> amalgInfo, RCP<MultiVector> fineNullspace,
163  RCP<const Map> coarseMap, RCP<Matrix>& Ptentative,
164  RCP<MultiVector>& coarseNullspace, const int levelID) const;
165 
166  mutable bool bTransferCoordinates_ = false;
167 };
168 
169 } // namespace MueLu
170 
171 #define MUELU_TENTATIVEPFACTORY_KOKKOS_SHORT
172 #endif // MUELU_TENTATIVEPFACTORY_KOKKOS_DECL_HPP
Teuchos::ScalarTraits< Scalar >::coordinateType real_type
void BuildPcoupled(RCP< Matrix > A, RCP< Aggregates > aggregates, RCP< AmalgamationInfo > amalgInfo, RCP< MultiVector > fineNullspace, RCP< const Map > coarseMap, RCP< Matrix > &Ptentative, RCP< MultiVector > &coarseNullspace) const
MueLu::DefaultLocalOrdinal LocalOrdinal
void BuildPuncoupledBlockCrs(Level &coarseLevel, RCP< Matrix > A, RCP< Aggregates > aggregates, RCP< AmalgamationInfo > amalgInfo, RCP< MultiVector > fineNullspace, RCP< const Map > coarseMap, RCP< Matrix > &Ptentative, RCP< MultiVector > &coarseNullspace, const int levelID) const
void BuildP(Level &fineLevel, Level &coarseLevel) const
Abstract Build method.
Xpetra::MultiVector< real_type, LocalOrdinal, GlobalOrdinal, node_type > RealValuedMultiVector
MueLu::DefaultNode Node
void BuildPuncoupled(Level &coarseLevel, RCP< Matrix > A, RCP< Aggregates > aggregates, RCP< AmalgamationInfo > amalgInfo, RCP< MultiVector > fineNullspace, RCP< const Map > coarseMap, RCP< Matrix > &Ptentative, RCP< MultiVector > &coarseNullspace, const int levelID) const
MueLu::DefaultGlobalOrdinal GlobalOrdinal
Kokkos::RangePolicy< local_ordinal_type, execution_space > range_type
Class that holds all level-specific information.
Definition: MueLu_Level.hpp:99
void Build(Level &fineLevel, Level &coarseLevel) const
Build an object with this factory.
void DeclareInput(Level &fineLevel, Level &coarseLevel) const
Input.
RCP< const ParameterList > GetValidParameterList() const
Return a const parameter list of valid parameters that setParameterList() will accept.
Factory that provides an interface for a concrete implementation of a prolongation operator...
bool isGoodMap(const Map &rowMap, const Map &colMap) const