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 <Teuchos_ScalarTraits.hpp>
#include <Tpetra_KokkosCompat_ClassicNodeAPI_Wrapper.hpp>
 
#include <Xpetra_CrsGraphFactory_fwd.hpp>
 
#include "MueLu_ConfigDefs.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"
#include "MueLu_Utilities_fwd.hpp"
 
namespace MueLu {
 
template <class Scalar        = DefaultScalar,
          class LocalOrdinal  = DefaultLocalOrdinal,
          class GlobalOrdinal = DefaultGlobalOrdinal,
          class Node          = DefaultNode>
class TentativePFactory_kokkos : public PFactory {
 public:
  using execution_space = typename Node::execution_space;
  using range_type      = Kokkos::RangePolicy<LocalOrdinal, execution_space>;
  using DeviceType      = typename Node::device_type;
 
 private:
#undef MUELU_TENTATIVEPFACTORY_KOKKOS_SHORT
#include "MueLu_UseShortNames.hpp"
 
 public:
 
 
  TentativePFactory_kokkos();
 
  ~TentativePFactory_kokkos() override;
 
  RCP<const ParameterList> GetValidParameterList() const override;
 
 
 
  void DeclareInput(Level& fineLevel, Level& coarseLevel) const override;
 
 
 
 
  void Build(Level& fineLevel, Level& coarseLevel) const override;
  void BuildP(Level& fineLevel, Level& coarseLevel) const override;
 
 
  // NOTE: All of these should really be private, but CUDA doesn't like that
 
  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, int levelID) const;
 
  void BuildPuncoupledBlockCrs(Level& coarseLevel, RCP<Matrix> A, RCP<Aggregates> aggregates, RCP<AmalgamationInfo> amalgInfo,
                               RCP<MultiVector> fineNullspace, RCP<const Map> coarsePointMap, RCP<Matrix>& Ptentative, RCP<MultiVector>& coarseNullspace, int levelID) 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;
 
  mutable bool bTransferCoordinates_ = false;
};
 
}  // namespace MueLu
 
#define MUELU_TENTATIVEPFACTORY_KOKKOS_SHORT
#endif  // MUELU_TENTATIVEPFACTORY_KOKKOS_DECL_HPP