MueLu Version of the Day
Loading...
Searching...
No Matches
MueLu_RegionRFactory_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
47#ifndef MUELU_REGIONRFACTORY_KOKKOS_DECL_HPP
48#define MUELU_REGIONRFACTORY_KOKKOS_DECL_HPP
49
50#include "MueLu_ConfigDefs.hpp"
51#include "MueLu_Types.hpp"
52
53#include <KokkosCompat_ClassicNodeAPI_Wrapper.hpp>
54#include <Kokkos_Core.hpp>
55
58
59namespace MueLu {
60
66 template <class Scalar = DefaultScalar,
69 class Node = DefaultNode>
71#undef MUELU_REGIONRFACTORY_KOKKOS_SHORT
73
74 public:
75 using real_type = typename Teuchos::ScalarTraits<SC>::coordinateType;
76 using realvaluedmultivector_type = typename Xpetra::MultiVector<real_type, LO, GO, NO>;
77 using execution_space = typename Node::execution_space;
78 using memory_space = typename Node::memory_space;
79 using device_type = Kokkos::Device<execution_space, memory_space>;
80 using intTupleView = typename Kokkos::View<int[3], device_type>;
81 using LOTupleView = typename Kokkos::View<LO[3], device_type>;
82
84
85
88
90 virtual ~RegionRFactory_kokkos() = default;
92
94
95 RCP<const ParameterList> GetValidParameterList() const;
96
97 void DeclareInput(Level& fineLevel, Level& coarseLevel) const;
98
100
102
103
104 void Build(Level& fineLevel, Level& coarseLevel) const;
105
106 void Build3D(const int numDimensions,
107 Array<LO>& lFineNodesPerDim,
108 const RCP<Matrix>& A,
109 const RCP<realvaluedmultivector_type>& fineCoordinates,
110 RCP<Matrix>& R,
111 RCP<realvaluedmultivector_type>& coarseCoordinates,
112 Array<LO>& lCoarseNodesPerDim) const;
113
115
116 }; // class RegionRFactory_kokkos
117
118} // namespace MueLu
119
120#define MUELU_REGIONRFACTORY_KOKKOS_SHORT
121#endif // MUELU_REGIONRFACTORY_KOKKOS_DECL_HPP
MueLu::DefaultLocalOrdinal LocalOrdinal
MueLu::DefaultScalar Scalar
MueLu::DefaultGlobalOrdinal GlobalOrdinal
MueLu::DefaultNode Node
Class that holds all level-specific information.
Definition: MueLu_Level.hpp:99
Factory that builds a restriction operator for region multigrid.
typename Kokkos::View< LO[3], device_type > LOTupleView
RegionRFactory_kokkos()=default
Default Constructor.
typename Teuchos::ScalarTraits< SC >::coordinateType real_type
typename Xpetra::MultiVector< real_type, LO, GO, NO > realvaluedmultivector_type
typename Kokkos::View< int[3], device_type > intTupleView
Kokkos::Device< execution_space, memory_space > device_type
void Build3D(const int numDimensions, Array< LO > &lFineNodesPerDim, const RCP< Matrix > &A, const RCP< realvaluedmultivector_type > &fineCoordinates, RCP< Matrix > &R, RCP< realvaluedmultivector_type > &coarseCoordinates, Array< LO > &lCoarseNodesPerDim) const
typename Node::execution_space execution_space
void DeclareInput(Level &fineLevel, Level &coarseLevel) const
Input.
RCP< const ParameterList > GetValidParameterList() const
Input.
void Build(Level &fineLevel, Level &coarseLevel) const
Build an object with this factory.
virtual ~RegionRFactory_kokkos()=default
Destructor.
Base class for factories that use two levels (fineLevel and coarseLevel).
Namespace for MueLu classes and methods.
KokkosClassic::DefaultNode::DefaultNodeType DefaultNode
Tpetra::Details::DefaultTypes::scalar_type DefaultScalar