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 MFEM_ENABLED
      11             : 
      12             : #include "MFEMHypreGMRES.h"
      13             : #include "MFEMProblem.h"
      14             : 
      15             : registerMooseObject("MooseApp", MFEMHypreGMRES);
      16             : 
      17             : InputParameters
      18        8766 : MFEMHypreGMRES::validParams()
      19             : {
      20        8766 :   InputParameters params = MFEMSolverBase::validParams();
      21        8766 :   params.addClassDescription("Hypre solver for the iterative solution of MFEM equation systems "
      22             :                              "using the generalized minimal residual method.");
      23             : 
      24        8766 :   params.addParam<mfem::real_t>("l_tol", 1e-5, "Set the relative tolerance.");
      25        8766 :   params.addParam<mfem::real_t>("l_abs_tol", 1e-50, "Set the absolute tolerance.");
      26        8766 :   params.addParam<int>("l_max_its", 10000, "Set the maximum number of iterations.");
      27        8766 :   params.addParam<int>("kdim", 10, "Set the k-dimension.");
      28        8766 :   params.addParam<int>("print_level", 2, "Set the solver verbosity.");
      29        8766 :   params.addParam<UserObjectName>("preconditioner", "Optional choice of preconditioner to use.");
      30             : 
      31        8766 :   return params;
      32           0 : }
      33             : 
      34          68 : MFEMHypreGMRES::MFEMHypreGMRES(const InputParameters & parameters) : MFEMSolverBase(parameters)
      35             : {
      36          68 :   constructSolver(parameters);
      37          68 : }
      38             : 
      39             : void
      40          68 : MFEMHypreGMRES::constructSolver(const InputParameters &)
      41             : {
      42             :   auto solver =
      43          68 :       std::make_unique<mfem::HypreGMRES>(getMFEMProblem().mesh().getMFEMParMesh().GetComm());
      44          68 :   solver->SetTol(getParam<mfem::real_t>("l_tol"));
      45          68 :   solver->SetAbsTol(getParam<mfem::real_t>("l_abs_tol"));
      46          68 :   solver->SetMaxIter(getParam<int>("l_max_its"));
      47          68 :   solver->SetKDim(getParam<int>("kdim"));
      48          68 :   solver->SetPrintLevel(getParam<int>("print_level"));
      49          68 :   setPreconditioner(*solver);
      50          68 :   _solver = std::move(solver);
      51          68 : }
      52             : 
      53             : void
      54          88 : MFEMHypreGMRES::updateSolver(mfem::ParBilinearForm & a, mfem::Array<int> & tdofs)
      55             : {
      56          88 :   if (_lor && _preconditioner)
      57           0 :     mooseError("LOR solver cannot take a preconditioner");
      58             : 
      59          88 :   if (_preconditioner)
      60             :   {
      61          86 :     _preconditioner->updateSolver(a, tdofs);
      62          86 :     setPreconditioner(static_cast<mfem::HypreGMRES &>(*_solver));
      63             :   }
      64           2 :   else if (_lor)
      65             :   {
      66           1 :     if (!checkSpectralEquivalence(a))
      67           0 :       mooseError("Low-Order-Refined solver requires the FESpace closed_basis to be GaussLobatto "
      68             :                  "and the open-basis to be IntegratedGLL for ND and RT elements.");
      69             : 
      70           1 :     mfem::ParLORDiscretization lor_disc(a, tdofs);
      71             :     auto lor_solver = new mfem::LORSolver<mfem::HypreGMRES>(
      72           1 :         lor_disc, getMFEMProblem().mesh().getMFEMParMesh().GetComm());
      73           1 :     lor_solver->GetSolver().SetTol(getParam<mfem::real_t>("l_tol"));
      74           1 :     lor_solver->GetSolver().SetAbsTol(getParam<mfem::real_t>("l_abs_tol"));
      75           1 :     lor_solver->GetSolver().SetMaxIter(getParam<int>("l_max_its"));
      76           1 :     lor_solver->GetSolver().SetKDim(getParam<int>("kdim"));
      77           1 :     lor_solver->GetSolver().SetPrintLevel(getParam<int>("print_level"));
      78             : 
      79           1 :     _solver.reset(lor_solver);
      80           1 :   }
      81          88 : }
      82             : 
      83             : #endif