Line data    Source code

       1             : //* This file is part of the MOOSE framework
       2             : //* https://mooseframework.inl.gov
       3             : //*
       4             : //* All rights reserved, see COPYRIGHT for full restrictions
       5             : //* https://github.com/idaholab/moose/blob/master/COPYRIGHT
       6             : //*
       7             : //* Licensed under LGPL 2.1, please see LICENSE for details
       8             : //* https://www.gnu.org/licenses/lgpl-2.1.html
       9             : 
      10             : #ifdef MOOSE_MFEM_ENABLED
      11             : 
      12             : #include "MFEMHypreGMRES.h"
      13             : #include "MFEMProblem.h"
      14             : 
      15             : registerMooseObject("MooseApp", MFEMHypreGMRES);
      16             : 
      17             : InputParameters
      18        3316 : MFEMHypreGMRES::validParams()
      19             : {
      20        3316 :   InputParameters params = Moose::MFEM::LinearSolverBase::validParams();
      21        6632 :   params.addClassDescription("Hypre solver for the iterative solution of MFEM equation systems "
      22             :                              "using the generalized minimal residual method.");
      23             : 
      24       13264 :   params.addParam<mfem::real_t>("l_tol", 1e-5, "Set the relative tolerance.");
      25       13264 :   params.addParam<mfem::real_t>("l_abs_tol", 1e-50, "Set the absolute tolerance.");
      26       13264 :   params.addParam<int>("l_max_its", 10000, "Set the maximum number of iterations.");
      27       13264 :   params.addParam<int>("kdim", 10, "Set the k-dimension.");
      28       13264 :   params.addParam<int>("print_level", 2, "Set the solver verbosity.");
      29        9948 :   params.addParam<MFEMSolverName>("preconditioner", "Optional choice of preconditioner to use.");
      30             : 
      31        3316 :   return params;
      32           0 : }
      33             : 
      34         609 : MFEMHypreGMRES::MFEMHypreGMRES(const InputParameters & parameters)
      35         609 :   : Moose::MFEM::LinearSolverBase(parameters)
      36             : {
      37         609 :   ConstructSolver();
      38         609 : }
      39             : 
      40             : void
      41         635 : MFEMHypreGMRES::ConstructSolver()
      42             : {
      43         635 :   auto solver = std::make_unique<mfem::HypreGMRES>(getMFEMProblem().getComm());
      44        1270 :   solver->iterative_mode = getParam<bool>("use_initial_guess");
      45        1270 :   solver->SetTol(getParam<mfem::real_t>("l_tol"));
      46        1270 :   solver->SetAbsTol(getParam<mfem::real_t>("l_abs_tol"));
      47        1270 :   solver->SetMaxIter(getParam<int>("l_max_its"));
      48        1270 :   solver->SetKDim(getParam<int>("kdim"));
      49        1270 :   solver->SetPrintLevel(getParam<int>("print_level"));
      50         635 :   SetPreconditioner(*solver);
      51         635 :   _solver = std::move(solver);
      52         635 : }
      53             : 
      54             : void
      55           4 : MFEMHypreGMRES::SetupLOR(mfem::ParBilinearForm & a, mfem::Array<int> & ess_bdr_markers)
      56             : {
      57           4 :   if (_lor && _preconditioner)
      58           0 :     mooseError("LOR solver cannot take a preconditioner");
      59             : 
      60           4 :   mfem::Array<int> ess_tdofs;
      61           4 :   a.ParFESpace()->GetEssentialTrueDofs(ess_bdr_markers, ess_tdofs);
      62           4 :   if (_preconditioner)
      63             :   {
      64           0 :     _preconditioner->SetupLOR(a, ess_tdofs);
      65           0 :     SetPreconditioner(static_cast<mfem::HypreGMRES &>(*_solver));
      66             :   }
      67           4 :   else if (_lor)
      68             :   {
      69           2 :     CheckSpectralEquivalence(a);
      70           2 :     mfem::ParLORDiscretization lor_disc(a, ess_tdofs);
      71           2 :     auto lor_solver = new mfem::LORSolver<mfem::HypreGMRES>(lor_disc, getMFEMProblem().getComm());
      72           4 :     lor_solver->GetSolver().SetTol(getParam<mfem::real_t>("l_tol"));
      73           4 :     lor_solver->GetSolver().SetAbsTol(getParam<mfem::real_t>("l_abs_tol"));
      74           4 :     lor_solver->GetSolver().SetMaxIter(getParam<int>("l_max_its"));
      75           4 :     lor_solver->GetSolver().SetKDim(getParam<int>("kdim"));
      76           4 :     lor_solver->GetSolver().SetPrintLevel(getParam<int>("print_level"));
      77             : 
      78           2 :     _solver.reset(lor_solver);
      79           2 :   }
      80           4 : }
      81             : 
      82             : #endif