MueLu_TentativePFactory_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_TENTATIVEPFACTORY_KOKKOS_DECL_HPP
#define MUELU_TENTATIVEPFACTORY_KOKKOS_DECL_HPP
 
#include "MueLu_ConfigDefs.hpp"
 
#include "MueLu_TentativePFactory_kokkos_fwd.hpp"
 
#include <Tpetra_KokkosCompat_ClassicNodeAPI_Wrapper.hpp>
 
#include "Teuchos_ScalarTraits.hpp"
 
#include "Xpetra_CrsGraphFactory_fwd.hpp"
 
#include "MueLu_Aggregates_fwd.hpp"
#include "MueLu_AmalgamationInfo_fwd.hpp"
#include "MueLu_Level_fwd.hpp"
#include "MueLu_PerfUtils_fwd.hpp"
#include "MueLu_PFactory.hpp"
 
namespace MueLu {
 
template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
class TentativePFactory_kokkos : public PFactory {
 public:
  typedef LocalOrdinal local_ordinal_type;
  typedef GlobalOrdinal global_ordinal_type;
  typedef typename Node::execution_space execution_space;
  typedef Kokkos::RangePolicy<local_ordinal_type, execution_space> range_type;
  typedef typename Node::device_type DeviceType;
  typedef Node node_type;
  typedef typename Teuchos::ScalarTraits<Scalar>::coordinateType real_type;
  typedef Xpetra::MultiVector<real_type, LocalOrdinal, GlobalOrdinal, node_type> RealValuedMultiVector;
 
 private:
#undef MUELU_TENTATIVEPFACTORY_KOKKOS_SHORT
#include "MueLu_UseShortNames.hpp"
 
 public:
 
 
  TentativePFactory_kokkos() {}
 
  virtual ~TentativePFactory_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;
 
 
  // NOTE: All of thess should really be private, but CUDA doesn't like that
 
  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;
 
  bool isGoodMap(const Map& rowMap, const Map& colMap) const;
 
  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;
 
  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;
 
  mutable bool bTransferCoordinates_ = false;
};
 
}  // namespace MueLu
 
#define MUELU_TENTATIVEPFACTORY_KOKKOS_SHORT
#endif  // MUELU_TENTATIVEPFACTORY_KOKKOS_DECL_HPP