MueLu_GeometricInterpolationPFactory_kokkos_decl.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//        MueLu: A package for multigrid based preconditioning
//
// Copyright 2012 NTESS and the MueLu contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef MUELU_GEOMETRICINTERPOLATIONPFACTORY_KOKKOS_DECL_HPP
#define MUELU_GEOMETRICINTERPOLATIONPFACTORY_KOKKOS_DECL_HPP

// Teuchos includes for dense linear algebra
#include <Teuchos_SerialDenseMatrix.hpp>
#include <Teuchos_SerialDenseVector.hpp>
#include <Teuchos_SerialDenseSolver.hpp>

#include "Xpetra_CrsGraph_fwd.hpp"

#include "MueLu_PFactory.hpp"
#include "MueLu_Level_fwd.hpp"
#include "MueLu_IndexManager_kokkos_fwd.hpp"

namespace MueLu {

template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
class GeometricInterpolationPFactory_kokkos : public PFactory {
#undef MUELU_GEOMETRICINTERPOLATIONPFACTORY_KOKKOS_SHORT
#include "MueLu_UseShortNames.hpp"

 public:
  // Declare useful types
  using real_type                  = typename Teuchos::ScalarTraits<SC>::coordinateType;
  using realvaluedmultivector_type = Xpetra::MultiVector<real_type, LO, GO, Node>;
  using device_type                = typename Node::device_type;
  using execution_space            = typename Node::execution_space;
#if KOKKOS_VERSION >= 40799
  using impl_scalar_type = typename KokkosKernels::ArithTraits<real_type>::val_type;
#else
  using impl_scalar_type = typename Kokkos::ArithTraits<real_type>::val_type;
#endif
  using coord_view_type = typename Kokkos::View<impl_scalar_type**,
                                                Kokkos::LayoutLeft,
                                                device_type>;



  GeometricInterpolationPFactory_kokkos() {}

  virtual ~GeometricInterpolationPFactory_kokkos() {}

  RCP<const ParameterList> GetValidParameterList() const;



  void DeclareInput(Level& fineLevel, Level& coarseLevel) const;




  void Build(Level& fineLevel, Level& coarseLevel) const;
  void BuildP(Level& fineLevel, Level& coarseLevel) const;




  struct coarseCoordinatesBuilderFunctor {
    IndexManager_kokkos geoData_;
    coord_view_type fineCoordView_;
    coord_view_type coarseCoordView_;

    coarseCoordinatesBuilderFunctor(RCP<IndexManager_kokkos> geoData,
                                    coord_view_type fineCoordView,
                                    coord_view_type coarseCoordView);

    KOKKOS_INLINE_FUNCTION
    void operator()(const LO nodeIdx) const;

  };  // struct coarseCoordinatesBuilderFunctor


  void BuildConstantP(RCP<Matrix>& P, RCP<const CrsGraph>& prolongatorGraph, RCP<Matrix>& A) const;

 private:
  void BuildLinearP(RCP<Matrix>& A, RCP<const CrsGraph>& prolongatorGraph,
                    RCP<realvaluedmultivector_type>& fineCoordinates,
                    RCP<realvaluedmultivector_type>& ghostCoordinates,
                    const int numDimensions, RCP<Matrix>& P) const;
  void ComputeLinearInterpolationStencil(const int numDimensions, const int numInterpolationPoints,
                                         const Array<Array<real_type> > coord,
                                         Array<real_type>& stencil) const;
  void GetInterpolationFunctions(const LO numDimensions,
                                 const Teuchos::SerialDenseVector<LO, real_type> parametricCoordinates,
                                 real_type functions[4][8]) const;

};  // class GeometricInterpolationPFactory_kokkos

}  // namespace MueLu

#define MUELU_GEOMETRICINTERPOLATIONPFACTORY_KOKKOS_SHORT
#endif  // MUELU_GEOMETRICINTERPOLATIONPFACTORY_KOKKOS_DECL_HPP