46 #ifndef MUELU_GEOMETRICINTERPOLATIONPFACTORY_KOKKOS_DECL_HPP 47 #define MUELU_GEOMETRICINTERPOLATIONPFACTORY_KOKKOS_DECL_HPP 50 #include <Teuchos_SerialDenseMatrix.hpp> 51 #include <Teuchos_SerialDenseVector.hpp> 52 #include <Teuchos_SerialDenseSolver.hpp> 54 #include "Xpetra_CrsGraph.hpp" 56 #include "MueLu_PFactory.hpp" 62 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
64 #undef MUELU_GEOMETRICINTERPOLATIONPFACTORY_KOKKOS_SHORT 70 using real_type =
typename Teuchos::ScalarTraits<SC>::coordinateType;
119 KOKKOS_INLINE_FUNCTION
127 void BuildConstantP(RCP<Matrix>& P, RCP<const CrsGraph>& prolongatorGraph, RCP<Matrix>& A)
const;
128 void BuildLinearP(RCP<Matrix>& A, RCP<const CrsGraph>& prolongatorGraph,
129 RCP<realvaluedmultivector_type>& fineCoordinates,
130 RCP<realvaluedmultivector_type>& ghostCoordinates,
131 const int numDimensions, RCP<Matrix>& P)
const;
133 const Array<Array<real_type> > coord,
134 Array<real_type>& stencil)
const;
136 const Teuchos::SerialDenseVector<LO,real_type> parametricCoordinates,
143 #define MUELU_GEOMETRICINTERPOLATIONPFACTORY_KOKKOS_SHORT 144 #endif // MUELU_GEOMETRICINTERPOLATIONPFACTORY_KOKKOS_DECL_HPP IndexManager_kokkos geoData_
void GetInterpolationFunctions(const LO numDimensions, const Teuchos::SerialDenseVector< LO, real_type > parametricCoordinates, real_type functions[4][8]) const
typename Kokkos::Details::ArithTraits< real_type >::val_type impl_scalar_type
coarseCoordinatesBuilderFunctor(RCP< IndexManager_kokkos > geoData, coord_view_type fineCoordView, coord_view_type coarseCoordView)
void BuildP(Level &fineLevel, Level &coarseLevel) const
Abstract Build method.
coord_view_type coarseCoordView_
virtual ~GeometricInterpolationPFactory_kokkos()
Destructor.
void Build(Level &fineLevel, Level &coarseLevel) const
Build an object with this factory.
Namespace for MueLu classes and methods.
typename Teuchos::ScalarTraits< SC >::coordinateType real_type
Xpetra::MultiVector< real_type, LO, GO, Node > realvaluedmultivector_type
void BuildLinearP(RCP< Matrix > &A, RCP< const CrsGraph > &prolongatorGraph, RCP< realvaluedmultivector_type > &fineCoordinates, RCP< realvaluedmultivector_type > &ghostCoordinates, const int numDimensions, RCP< Matrix > &P) const
typename Node::device_type device_type
GeometricInterpolationPFactory_kokkos()
Constructor.
coord_view_type fineCoordView_
RCP< const ParameterList > GetValidParameterList() const
Return a const parameter list of valid parameters that setParameterList() will accept.
Class that holds all level-specific information.
typename Kokkos::View< impl_scalar_type **, Kokkos::LayoutLeft, device_type > coord_view_type
void DeclareInput(Level &fineLevel, Level &coarseLevel) const
Input.
void BuildConstantP(RCP< Matrix > &P, RCP< const CrsGraph > &prolongatorGraph, RCP< Matrix > &A) const
void ComputeLinearInterpolationStencil(const int numDimensions, const int numInterpolationPoints, const Array< Array< real_type > > coord, Array< real_type > &stencil) const
Factory that provides an interface for a concrete implementation of a prolongation operator...
KOKKOS_INLINE_FUNCTION void operator()(const LO nodeIdx) const
typename Node::execution_space execution_space