MueLu_StructuredAggregationFactory_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_STRUCTUREDAGGREGATIONFACTORY_KOKKOS_DECL_HPP
#define MUELU_STRUCTUREDAGGREGATIONFACTORY_KOKKOS_DECL_HPP
 
#include "MueLu_ConfigDefs.hpp"
#include "MueLu_SingleLevelFactoryBase.hpp"
#include "MueLu_Level_fwd.hpp"
#include "MueLu_Exceptions.hpp"
 
#include "MueLu_AggregationStructuredAlgorithm_fwd.hpp"
 
#include <Tpetra_KokkosCompat_ClassicNodeAPI_Wrapper.hpp>
 
namespace MueLu {

 
template <class LocalOrdinal  = DefaultLocalOrdinal,
          class GlobalOrdinal = DefaultGlobalOrdinal,
          class Node          = DefaultNode>
class StructuredAggregationFactory_kokkos : public SingleLevelFactoryBase {
#undef MUELU_STRUCTUREDAGGREGATIONFACTORY_KOKKOS_SHORT
#include "MueLu_UseShortNamesOrdinal.hpp"
 
 public:


  StructuredAggregationFactory_kokkos();

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



  // set information about 1-node aggregates (map name and generating factory)
  void SetOnePtMapName(const std::string name, Teuchos::RCP<const FactoryBase> mapFact) {
    SetParameter("OnePt aggregate map name", ParameterEntry(std::string(name)));  // revalidate
    SetFactory("OnePt aggregate map factory", mapFact);
  }



 
  void DeclareInput(Level& currentLevel) const;




  void Build(Level& currentLevel) const;

 
 private:
  mutable bool bDefinitionPhase_;
 
};  // class StructuredAggregationFactory
 
}  // namespace MueLu
 
#define MUELU_STRUCTUREDAGGREGATIONFACTORY_KOKKOS_SHORT
#endif  // MUELU_UNCOUPLEDAGGREGATIONFACTORY_KOKKOS_DECL_HPP