5#ifndef GKO_PUBLIC_CORE_SOLVER_MULTIGRID_HPP_
6#define GKO_PUBLIC_CORE_SOLVER_MULTIGRID_HPP_
14#include <ginkgo/core/base/abstract_factory.hpp>
15#include <ginkgo/core/base/array.hpp>
16#include <ginkgo/core/base/exception_helpers.hpp>
17#include <ginkgo/core/base/lin_op.hpp>
18#include <ginkgo/core/base/math.hpp>
19#include <ginkgo/core/base/precision_dispatch.hpp>
20#include <ginkgo/core/base/types.hpp>
21#include <ginkgo/core/config/config.hpp>
22#include <ginkgo/core/config/registry.hpp>
23#include <ginkgo/core/config/type_descriptor.hpp>
24#include <ginkgo/core/log/logger.hpp>
25#include <ginkgo/core/matrix/dense.hpp>
26#include <ginkgo/core/matrix/identity.hpp>
27#include <ginkgo/core/multigrid/multigrid_level.hpp>
28#include <ginkgo/core/solver/solver_base.hpp>
29#include <ginkgo/core/stop/combined.hpp>
30#include <ginkgo/core/stop/criterion.hpp>
108 public EnableSolverBase<Multigrid>,
109 public EnableIterativeBase<Multigrid>,
110 public EnableApplyWithInitialGuess<Multigrid> {
113 friend class EnableApplyWithInitialGuess<Multigrid>;
124 return this->get_default_initial_guess() ==
133 std::vector<std::shared_ptr<const gko::multigrid::MultigridLevel>>
136 return mg_level_list_;
146 return pre_smoother_list_;
156 return mid_smoother_list_;
166 return post_smoother_list_;
176 return coarsest_solver_;
202 std::vector<std::shared_ptr<const LinOpFactory>>
248 std::vector<std::shared_ptr<const LinOpFactory>>
256 std::vector<std::shared_ptr<const LinOpFactory>>
266 std::vector<std::shared_ptr<const LinOpFactory>>
307 std::vector<std::shared_ptr<const LinOpFactory>>
341 multigrid::cycle::v);
401 config::make_type_descriptor<>());
404 void apply_impl(
const LinOp* b,
LinOp* x)
const override;
407 LinOp* x)
const override;
409 void apply_with_initial_guess_impl(
const LinOp* b,
LinOp* x,
412 void apply_with_initial_guess_impl(
const LinOp* alpha,
const LinOp* b,
416 template <
typename VectorType>
417 void apply_dense_impl(
const VectorType* b, VectorType* x,
426 explicit Multigrid(std::shared_ptr<const Executor> exec);
428 explicit Multigrid(
const Factory* factory,
429 std::shared_ptr<const LinOp> system_matrix);
447 void create_state()
const;
450 std::vector<std::shared_ptr<const gko::multigrid::MultigridLevel>>
452 std::vector<std::shared_ptr<const LinOp>> pre_smoother_list_{};
453 std::vector<std::shared_ptr<const LinOp>> mid_smoother_list_{};
454 std::vector<std::shared_ptr<const LinOp>> post_smoother_list_{};
455 std::shared_ptr<const LinOp> coarsest_solver_{};
469 mutable struct cache_struct {
470 cache_struct() =
default;
472 ~cache_struct() =
default;
474 cache_struct(
const cache_struct&) {}
476 cache_struct(cache_struct&&) {}
478 cache_struct& operator=(
const cache_struct&) {
return *
this; }
480 cache_struct& operator=(cache_struct&&) {
return *
this; }
484 std::shared_ptr<multigrid::detail::MultigridState> state{};
492 static int num_vectors(
const Solver&);
494 static int num_arrays(
const Solver&);
496 static std::vector<std::string> op_names(
const Solver&);
498 static std::vector<std::string> array_names(
const Solver&);
500 static std::vector<int> scalars(
const Solver&);
502 static std::vector<int> vectors(
const Solver&);
505 constexpr static int stop = 0;
The EnableLinOp mixin can be used to provide sensible default implementations of the majority of the ...
Definition lin_op.hpp:879
This mixin inherits from (a subclass of) PolymorphicObject and provides a base implementation of a ne...
Definition polymorphic_object.hpp:668
Definition lin_op.hpp:117
LinOp(const LinOp &)=default
Copy-constructs a LinOp.
pnode describes a tree of properties.
Definition property_tree.hpp:28
This class stores additional context for creating Ginkgo objects from configuration files.
Definition registry.hpp:167
This class describes the value and index types to be used when building a Ginkgo type from a configur...
Definition type_descriptor.hpp:39
Definition multigrid.hpp:381
Multigrid methods have a hierarchy of many levels, whose corase level is a subset of the fine level,...
Definition multigrid.hpp:110
std::shared_ptr< const LinOp > get_coarsest_solver() const
Gets the operator at the coarsest level.
Definition multigrid.hpp:174
std::vector< std::shared_ptr< const gko::multigrid::MultigridLevel > > get_mg_level_list() const
Gets the list of MultigridLevel operators.
Definition multigrid.hpp:134
void set_cycle(multigrid::cycle cycle)
Set the cycle of multigrid.
Definition multigrid.hpp:191
std::vector< std::shared_ptr< const LinOp > > get_mid_smoother_list() const
Gets the list of mid-smoother operators.
Definition multigrid.hpp:154
multigrid::cycle get_cycle() const
Get the cycle of multigrid.
Definition multigrid.hpp:184
bool apply_uses_initial_guess() const override
Return true as iterative solvers use the data in x as an initial guess or false if multigrid always s...
Definition multigrid.hpp:122
std::vector< std::shared_ptr< const LinOp > > get_pre_smoother_list() const
Gets the list of pre-smoother operators.
Definition multigrid.hpp:144
std::vector< std::shared_ptr< const LinOp > > get_post_smoother_list() const
Gets the list of post-smoother operators.
Definition multigrid.hpp:164
static parameters_type parse(const config::pnode &config, const config::registry &context, const config::type_descriptor &td_for_child=config::make_type_descriptor<>())
Create the parameters from the property_tree.
#define GKO_FACTORY_PARAMETER_SCALAR(_name, _default)
Creates a scalar factory parameter in the factory parameters structure.
Definition abstract_factory.hpp:445
#define GKO_ENABLE_BUILD_METHOD(_factory_name)
Defines a build method for the factory, simplifying its construction by removing the repetitive typin...
Definition abstract_factory.hpp:394
#define GKO_ENABLE_LIN_OP_FACTORY(_lin_op, _parameters_name, _factory_name)
This macro will generate a default implementation of a LinOpFactory for the LinOp subclass it is defi...
Definition lin_op.hpp:1017
The solver multigrid namespace.
Definition multigrid.hpp:42
mid_smooth_type
mid_smooth_type gives the options to handle the middle smoother behavior between the two cycles in th...
Definition multigrid.hpp:70
cycle
cycle defines which kind of multigrid cycle can be used.
Definition multigrid.hpp:54
The ginkgo Solve namespace.
Definition bicg.hpp:28
initial_guess_mode
Give a initial guess mode about the input of the apply method.
Definition solver_base.hpp:33
@ provided
the input is provided
Definition solver_base.hpp:45
@ zero
the input is zero
Definition solver_base.hpp:37
The Ginkgo namespace.
Definition abstract_factory.hpp:20
std::size_t size_type
Integral type used for allocation quantities.
Definition types.hpp:90
Definition multigrid.hpp:198
std::vector< std::shared_ptr< const LinOpFactory > > coarsest_solver
Coarsest factory list.
Definition multigrid.hpp:308
std::vector< std::shared_ptr< const LinOpFactory > > post_smoother
Post-smooth Factory list.
Definition multigrid.hpp:257
size_type max_levels
The maximum number of mg_level (without coarsest solver level) that can be used.
Definition multigrid.hpp:292
std::complex< double > smoother_relax
smoother_relax is the relaxation factor of default generated smoother.
Definition multigrid.hpp:366
size_type min_coarse_rows
The minimal number of coarse rows.
Definition multigrid.hpp:299
initial_guess_mode default_initial_guess
Default initial guess mode.
Definition multigrid.hpp:379
std::vector< std::shared_ptr< const LinOpFactory > > pre_smoother
Pre-smooth Factory list.
Definition multigrid.hpp:249
size_type kcycle_base
kcycle_base is a factor to choose how often enable FCG/GCR step.
Definition multigrid.hpp:351
size_type smoother_iters
smoother_iters is the number of iteration of default generated smoother.
Definition multigrid.hpp:372
std::function< size_type(const size_type, const LinOp *)> level_selector
Custom selector size_type (size_type level, const LinOp* fine_matrix) Selector function returns the e...
Definition multigrid.hpp:232
multigrid::cycle cycle
Multigrid cycle type.
Definition multigrid.hpp:341
multigrid::mid_smooth_type mid_case
Choose the behavior of mid smoother between two cycles close to each other in the same level.
Definition multigrid.hpp:283
bool post_uses_pre
Whether post-smoothing-related calls use corresponding pre-smoothing-related calls.
Definition multigrid.hpp:273
std::function< size_type(const size_type, const LinOp *)> solver_selector
Custom coarsest_solver selector size_type (size_type level, const LinOp *coarsest_matrix) Selector fu...
Definition multigrid.hpp:333
std::vector< std::shared_ptr< const LinOpFactory > > mid_smoother
Mid-smooth Factory list.
Definition multigrid.hpp:267
double kcycle_rel_tol
kcycle_rel_tol decides whether run the second iteration of FCG/GCR step.
Definition multigrid.hpp:360
std::vector< std::shared_ptr< const LinOpFactory > > mg_level
MultigridLevel Factory list.
Definition multigrid.hpp:203
Definition solver_base.hpp:844
Traits class providing information on the type and location of workspace vectors inside a solver.
Definition solver_base.hpp:238