Ginkgo Generated from branch based on main. Ginkgo version 1.10.0
A numerical linear algebra library targeting many-core architectures
Loading...
Searching...
No Matches
batch_cg.hpp
1// SPDX-FileCopyrightText: 2017 - 2024 The Ginkgo authors
2//
3// SPDX-License-Identifier: BSD-3-Clause
4
5#ifndef GKO_PUBLIC_CORE_SOLVER_BATCH_CG_HPP_
6#define GKO_PUBLIC_CORE_SOLVER_BATCH_CG_HPP_
7
8
9#include <vector>
10
11#include <ginkgo/core/base/batch_lin_op.hpp>
12#include <ginkgo/core/base/batch_multi_vector.hpp>
13#include <ginkgo/core/base/exception_helpers.hpp>
14#include <ginkgo/core/base/lin_op.hpp>
15#include <ginkgo/core/base/types.hpp>
16#include <ginkgo/core/solver/batch_solver_base.hpp>
17#include <ginkgo/core/stop/batch_stop_enum.hpp>
18
19
20namespace gko {
21namespace batch {
22namespace solver {
23
24
49template <typename ValueType = default_precision>
50class Cg final : public EnableBatchSolver<Cg<ValueType>, ValueType> {
51 friend class EnableBatchLinOp<Cg>;
52 friend class EnablePolymorphicObject<Cg, BatchLinOp>;
53
54public:
55 using value_type = ValueType;
56 using real_type = gko::remove_complex<ValueType>;
57
58 class Factory;
59
63 GKO_ENABLE_BATCH_LIN_OP_FACTORY(Cg, parameters, Factory);
65
66private:
67 explicit Cg(std::shared_ptr<const Executor> exec);
68
69 explicit Cg(const Factory* factory,
70 std::shared_ptr<const BatchLinOp> system_matrix);
71
72 void solver_apply(
74 log::detail::log_data<real_type>* log_data) const override;
75};
76
77
78} // namespace solver
79} // namespace batch
80} // namespace gko
81
82
83#endif // GKO_PUBLIC_CORE_SOLVER_BATCH_CG_HPP_
This mixin inherits from (a subclass of) PolymorphicObject and provides a base implementation of a ne...
Definition polymorphic_object.hpp:668
Definition batch_lin_op.hpp:59
The EnableBatchLinOp mixin can be used to provide sensible default implementations of the majority of...
Definition batch_lin_op.hpp:252
MultiVector stores multiple vectors in a batched fashion and is useful for batched operations.
Definition batch_multi_vector.hpp:61
Definition batch_cg.hpp:63
#define GKO_ENABLE_BATCH_LIN_OP_FACTORY(_batch_lin_op, _parameters_name, _factory_name)
This macro will generate a default implementation of a BatchLinOpFactory for the BatchLinOp subclass ...
Definition batch_lin_op.hpp:358
#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
The Ginkgo namespace.
Definition abstract_factory.hpp:20
typename detail::remove_complex_s< T >::type remove_complex
Obtain the type which removed the complex of complex/scalar type or the template parameter of class b...
Definition math.hpp:264
Definition batch_cg.hpp:62