batch_cg.hpp

// SPDX-FileCopyrightText: 2017 - 2024 The Ginkgo authors
//
// SPDX-License-Identifier: BSD-3-Clause
 
#ifndef GKO_PUBLIC_CORE_SOLVER_BATCH_CG_HPP_
#define GKO_PUBLIC_CORE_SOLVER_BATCH_CG_HPP_
 
 
#include <vector>
 
 
#include <ginkgo/core/base/batch_lin_op.hpp>
#include <ginkgo/core/base/batch_multi_vector.hpp>
#include <ginkgo/core/base/exception_helpers.hpp>
#include <ginkgo/core/base/lin_op.hpp>
#include <ginkgo/core/base/types.hpp>
#include <ginkgo/core/solver/batch_solver_base.hpp>
#include <ginkgo/core/stop/batch_stop_enum.hpp>
 
 
namespace gko {
namespace batch {
namespace solver {
 
 
template <typename ValueType = default_precision>
class Cg final : public EnableBatchSolver<Cg<ValueType>, ValueType> {
    friend class EnableBatchLinOp<Cg>;
    friend class EnablePolymorphicObject<Cg, BatchLinOp>;
 
public:
    using value_type = ValueType;
    using real_type = gko::remove_complex<ValueType>;
 
    class Factory;
 
    struct parameters_type
        : enable_preconditioned_iterative_solver_factory_parameters<
              parameters_type, Factory> {};
    GKO_ENABLE_BATCH_LIN_OP_FACTORY(Cg, parameters, Factory);
    GKO_ENABLE_BUILD_METHOD(Factory);
 
private:
    explicit Cg(std::shared_ptr<const Executor> exec);
 
    explicit Cg(const Factory* factory,
                std::shared_ptr<const BatchLinOp> system_matrix);
 
    void solver_apply(
        const MultiVector<ValueType>* b, MultiVector<ValueType>* x,
        log::detail::log_data<real_type>* log_data) const override;
};
 
 
}  // namespace solver
}  // namespace batch
}  // namespace gko
 
 
#endif  // GKO_PUBLIC_CORE_SOLVER_BATCH_CG_HPP_