MueLu_CoalesceDropFactory_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_COALESCEDROPFACTORY_KOKKOS_DECL_HPP
#define MUELU_COALESCEDROPFACTORY_KOKKOS_DECL_HPP
 
#include "MueLu_ConfigDefs.hpp"
 
#include <Tpetra_KokkosCompat_ClassicNodeAPI_Wrapper.hpp>
 
#include "Xpetra_Matrix_fwd.hpp"
 
#include "MueLu_CoalesceDropFactory_kokkos_fwd.hpp"
 
#include "MueLu_AmalgamationInfo_fwd.hpp"
#include "MueLu_Level_fwd.hpp"
#include "MueLu_LWGraph_kokkos_decl.hpp"
#include "MueLu_SingleLevelFactoryBase.hpp"
#include "MueLu_Utilities_fwd.hpp"
 
namespace MueLu {

template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
class CoalesceDropFactory_kokkos
  : public SingleLevelFactoryBase {
 public:
  using local_ordinal_type  = LocalOrdinal;
  using global_ordinal_type = GlobalOrdinal;
  using execution_space     = typename Node::execution_space;
  using range_type          = Kokkos::RangePolicy<local_ordinal_type, execution_space>;
  using node_type           = Node;
 
 private:
  using boundary_nodes_type = typename MueLu::LWGraph_kokkos<LocalOrdinal, GlobalOrdinal, Node>::boundary_nodes_type;
 
  // For compatibility
#undef MUELU_COALESCEDROPFACTORY_KOKKOS_SHORT
#include "MueLu_UseShortNames.hpp"
 
 public:


  CoalesceDropFactory_kokkos() {}

  virtual ~CoalesceDropFactory_kokkos() {}
 
  RCP<const ParameterList> GetValidParameterList() const;



 
  void DeclareInput(Level& currentLevel) const;

 
  void Build(Level& currentLevel) const;
 
  std::tuple<RCP<LocalOrdinalVector>, RCP<LocalOrdinalVector> > GetBlockNumberMVs(Level& currentLevel) const;
 
  std::tuple<GlobalOrdinal, boundary_nodes_type> BuildScalar(Level& currentLevel) const;
 
  std::tuple<GlobalOrdinal, boundary_nodes_type> BuildVector(Level& currentLevel) const;
};
 
}  // namespace MueLu
 
#define MUELU_COALESCEDROPFACTORY_KOKKOS_SHORT
#endif  // MUELU_COALESCEDROPFACTORY_KOKKOS_DECL_HPP