45 #ifndef THYRA_BELOS_LINEAR_OP_WITH_SOLVE_FACTORY_HPP 46 #define THYRA_BELOS_LINEAR_OP_WITH_SOLVE_FACTORY_HPP 49 #include "Thyra_BelosLinearOpWithSolveFactory_decl.hpp" 50 #include "Thyra_BelosLinearOpWithSolve.hpp" 51 #include "Thyra_ScaledAdjointLinearOpBase.hpp" 64 #include "Teuchos_VerboseObjectParameterListHelpers.hpp" 65 #include "Teuchos_StandardParameterEntryValidators.hpp" 66 #include "Teuchos_ParameterList.hpp" 67 #include "Teuchos_dyn_cast.hpp" 68 #include "Teuchos_ValidatorXMLConverterDB.hpp" 69 #include "Teuchos_StandardValidatorXMLConverters.hpp" 77 template<
class Scalar>
79 template<
class Scalar>
81 template<
class Scalar>
83 template<
class Scalar>
85 template<
class Scalar>
87 template<
class Scalar>
89 template<
class Scalar>
91 template<
class Scalar>
93 template<
class Scalar>
95 template<
class Scalar>
97 template<
class Scalar>
99 template<
class Scalar>
101 template<
class Scalar>
105 const std::string LeftPreconditionerIfUnspecified_name =
"Left Preconditioner If Unspecified";
111 template<
class Scalar>
114 convergenceTestFrequency_(1)
116 updateThisValidParamList();
120 template<
class Scalar>
133 template<
class Scalar>
140 template<
class Scalar>
143 const std::string &precFactoryName
148 precFactoryValidPL = precFactory->getValidParameters();
149 const std::string _precFactoryName =
150 ( precFactoryName !=
"" 152 : ( precFactoryValidPL.get() ? precFactoryValidPL->name() :
"GENERIC PRECONDITIONER FACTORY" )
154 precFactory_ = precFactory;
155 precFactoryName_ = _precFactoryName;
156 updateThisValidParamList();
160 template<
class Scalar>
168 template<
class Scalar>
171 std::string *precFactoryName
174 if(precFactory) *precFactory = precFactory_;
175 if(precFactoryName) *precFactoryName = precFactoryName_;
176 precFactory_ = Teuchos::null;
177 precFactoryName_ =
"";
178 updateThisValidParamList();
182 template<
class Scalar>
187 if(precFactory_.get())
188 return precFactory_->isCompatible(fwdOpSrc);
193 template<
class Scalar>
201 template<
class Scalar>
209 initializeOpImpl(fwdOpSrc,null,null,
false,Op,supportSolveUse);
213 template<
class Scalar>
224 template<
class Scalar>
229 if(precFactory_.get())
235 template<
class Scalar>
244 initializeOpImpl(fwdOpSrc,null,prec,
false,Op,supportSolveUse);
248 template<
class Scalar>
257 initializeOpImpl(fwdOpSrc,approxFwdOpSrc,null,
false,Op,supportSolveUse);
261 template<
class Scalar>
286 if(fwdOpSrc) *fwdOpSrc = _fwdOpSrc;
287 if(prec) *prec = _prec;
288 if(approxFwdOpSrc) *approxFwdOpSrc = _approxFwdOpSrc;
289 if(supportSolveUse) *supportSolveUse = _supportSolveUse;
296 template<
class Scalar>
303 paramList_ = paramList;
305 Teuchos::getIntegralValue<EBelosSolverType>(*paramList_, SolverType_name);
306 convergenceTestFrequency_ =
307 Teuchos::getParameter<int>(*paramList_, ConvergenceTestFrequency_name);
308 Teuchos::readVerboseObjectSublist(&*paramList_,
this);
312 template<
class Scalar>
320 template<
class Scalar>
325 paramList_ = Teuchos::null;
330 template<
class Scalar>
338 template<
class Scalar>
342 return thisValidParamList_;
349 template<
class Scalar>
352 std::ostringstream oss;
353 oss <<
"Thyra::BelosLinearOpWithSolveFactory";
364 template<
class Scalar>
369 using Teuchos::tuple;
370 using Teuchos::setStringToIntegralParameter;
376 EBelosSolverType> >::getDummyObject());
381 if(validParamList.
get()==NULL) {
383 setStringToIntegralParameter<EBelosSolverType>(
384 SolverType_name, SolverType_default,
385 "Type of linear solver algorithm to use.",
388 "Pseudo Block GMRES",
391 "Pseudo Block Stochastic CG",
398 "Block GMRES solver for nonsymmetric linear systems. It can also solve " 399 "single right-hand side systems, and can also perform Flexible GMRES " 400 "(where the preconditioner may change at every iteration, for example " 401 "for inner-outer iterations), by setting options in the \"Block GMRES\" " 404 "GMRES solver for nonsymmetric linear systems, that performs single " 405 "right-hand side solves on multiple right-hand sides at once. It " 406 "exploits operator multivector multiplication in order to amortize " 407 "global communication costs. Individual linear systems are deflated " 408 "out as they are solved.",
410 "Block CG solver for symmetric (Hermitian in complex arithmetic) " 411 "positive definite linear systems. It can also solve single " 412 "right-hand-side systems.",
414 "CG solver that performs single right-hand side CG on multiple right-hand " 415 "sides at once. It exploits operator multivector multiplication in order " 416 "to amortize global communication costs. Individual linear systems are " 417 "deflated out as they are solved.",
419 "stochastic CG solver that performs single right-hand side CG on multiple right-hand " 420 "sides at once. It exploits operator multivector multiplication in order " 421 "to amortize global communication costs. Individual linear systems are " 422 "deflated out as they are solved. [EXPERIMENTAL]",
424 "Variant of GMRES that performs subspace recycling to accelerate " 425 "convergence for sequences of solves with related linear systems. " 426 "Individual linear systems are deflated out as they are solved. " 427 "The current implementation only supports real-valued Scalar types.",
429 "CG solver for symmetric (Hermitian in complex arithmetic) positive " 430 "definite linear systems, that performs subspace recycling to " 431 "accelerate convergence for sequences of related linear systems.",
433 "MINRES solver for symmetric indefinite linear systems. It performs " 434 "single-right-hand-side solves on multiple right-hand sides sequentially.",
436 "TFQMR (Transpose-Free QMR) solver for nonsymmetric linear systems." 438 tuple<EBelosSolverType>(
443 SOLVER_TYPE_PSEUDO_BLOCK_STOCHASTIC_CG,
451 validParamList->
set(ConvergenceTestFrequency_name, as<int>(1),
452 "Number of linear solver iterations to skip between applying" 453 " user-defined convergence test.");
455 LeftPreconditionerIfUnspecified_name,
false,
456 "If the preconditioner does not specify if it is left or right, and this\n" 457 "option is set to true, put the preconditioner on the left side.\n" 458 "Historically, preconditioning is on the right. Some solvers may not\n" 459 "support left preconditioning.");
461 &solverTypesSL = validParamList->
sublist(SolverTypes_name);
477 *mgr.getValidParameters()
483 *mgr.getValidParameters()
495 *mgr.getValidParameters()
501 *mgr.getValidParameters()
517 return validParamList;
521 template<
class Scalar>
522 void BelosLinearOpWithSolveFactory<Scalar>::updateThisValidParamList()
527 Teuchos::setupVerboseObjectSublist(&*thisValidParamList_);
531 template<
class Scalar>
532 void BelosLinearOpWithSolveFactory<Scalar>::initializeOpImpl(
533 const RCP<
const LinearOpSourceBase<Scalar> > &fwdOpSrc,
534 const RCP<
const LinearOpSourceBase<Scalar> > &approxFwdOpSrc,
535 const RCP<
const PreconditionerBase<Scalar> > &prec_in,
536 const bool reusePrec,
537 LinearOpWithSolveBase<Scalar> *Op,
543 using Teuchos::set_extra_data;
545 typedef MultiVectorBase<Scalar> MV_t;
546 typedef LinearOpBase<Scalar> LO_t;
548 const RCP<Teuchos::FancyOStream> out = this->getOStream();
551 if(out.get() &&
static_cast<int>(verbLevel) > static_cast<int>(
Teuchos::VERB_LOW))
552 *out <<
"\nEntering Thyra::BelosLinearOpWithSolveFactory<"<<ST::name()<<
">::initializeOpImpl(...) ...\n";
562 RCP<const LinearOpBase<Scalar> >
563 fwdOp = fwdOpSrc->getOp(),
564 approxFwdOp = ( approxFwdOpSrc.get() ? approxFwdOpSrc->getOp() : Teuchos::null );
570 BelosLinearOpWithSolve<Scalar>
577 RCP<PreconditionerBase<Scalar> > myPrec = Teuchos::null;
578 RCP<const PreconditionerBase<Scalar> > prec = Teuchos::null;
585 if(precFactory_.get()) {
587 ( !belosOp->isExternalPrec()
588 ? Teuchos::rcp_const_cast<PreconditionerBase<Scalar> >(belosOp->extract_prec())
591 bool hasExistingPrec =
false;
593 hasExistingPrec =
true;
598 hasExistingPrec =
false;
599 myPrec = precFactory_->createPrec();
601 if( hasExistingPrec && reusePrec ) {
606 if(approxFwdOp.get())
607 precFactory_->initializePrec(approxFwdOpSrc,&*myPrec);
609 precFactory_->initializePrec(fwdOpSrc,&*myPrec);
619 bool oldIsExternalPrec =
false;
620 RCP<Belos::LinearProblem<Scalar,MV_t,LO_t> > oldLP = Teuchos::null;
621 RCP<Belos::SolverManager<Scalar,MV_t,LO_t> > oldIterSolver = Teuchos::null;
622 RCP<const LinearOpSourceBase<Scalar> > oldFwdOpSrc = Teuchos::null;
623 RCP<const LinearOpSourceBase<Scalar> > oldApproxFwdOpSrc = Teuchos::null;
626 belosOp->uninitialize( &oldLP, NULL, &oldIterSolver, &oldFwdOpSrc,
627 NULL, &oldIsExternalPrec, &oldApproxFwdOpSrc, &oldSupportSolveUse );
636 if (oldLP != Teuchos::null) {
640 lp =
rcp(
new LP_t());
647 lp->setOperator(fwdOp);
654 RCP<const LinearOpBase<Scalar> > unspecified = prec->getUnspecifiedPrecOp();
655 RCP<const LinearOpBase<Scalar> > left = prec->getLeftPrecOp();
656 RCP<const LinearOpBase<Scalar> > right = prec->getRightPrecOp();
658 !( left.get() || right.get() || unspecified.get() ), std::logic_error
659 ,
"Error, at least one preconditoner linear operator objects must be set!" 661 if(unspecified.get()) {
662 if (paramList_->get<
bool>(LeftPreconditionerIfUnspecified_name,
false))
663 lp->setLeftPrec(unspecified);
665 lp->setRightPrec(unspecified);
670 left.get(),std::logic_error
671 ,
"Error, we can not currently handle a left preconditioner!" 673 lp->setRightPrec(right);
677 set_extra_data<RCP<PreconditionerBase<Scalar> > >(myPrec,
"Belos::InternalPrec",
678 Teuchos::inOutArg(lp), Teuchos::POST_DESTROY,
false);
680 else if(prec.get()) {
681 set_extra_data<RCP<const PreconditionerBase<Scalar> > >(prec,
"Belos::ExternalPrec",
682 Teuchos::inOutArg(lp), Teuchos::POST_DESTROY,
false);
690 RCP<IterativeSolver_t> iterativeSolver = Teuchos::null;
693 switch(solverType_) {
697 if(paramList_.get()) {
703 if (oldIterSolver != Teuchos::null) {
704 iterativeSolver = oldIterSolver;
705 iterativeSolver->setProblem( lp );
706 iterativeSolver->setParameters( solverPL );
716 if(paramList_.get()) {
724 if (oldIterSolver != Teuchos::null) {
725 iterativeSolver = oldIterSolver;
726 iterativeSolver->setProblem( lp );
727 iterativeSolver->setParameters( solverPL );
737 if(paramList_.get()) {
743 if (oldIterSolver != Teuchos::null) {
744 iterativeSolver = oldIterSolver;
745 iterativeSolver->setProblem( lp );
746 iterativeSolver->setParameters( solverPL );
756 if(paramList_.get()) {
764 if (oldIterSolver != Teuchos::null) {
765 iterativeSolver = oldIterSolver;
766 iterativeSolver->setProblem( lp );
767 iterativeSolver->setParameters( solverPL );
774 case SOLVER_TYPE_PSEUDO_BLOCK_STOCHASTIC_CG:
777 if(paramList_.get()) {
785 if (oldIterSolver != Teuchos::null) {
786 iterativeSolver = oldIterSolver;
787 iterativeSolver->setProblem( lp );
788 iterativeSolver->setParameters( solverPL );
798 if(paramList_.get()) {
804 if (oldIterSolver != Teuchos::null) {
805 iterativeSolver = oldIterSolver;
806 iterativeSolver->setProblem( lp );
807 iterativeSolver->setParameters( solverPL );
817 if(paramList_.get()) {
823 if (oldIterSolver != Teuchos::null) {
824 iterativeSolver = oldIterSolver;
825 iterativeSolver->setProblem( lp );
826 iterativeSolver->setParameters( solverPL );
836 if(paramList_.get()) {
842 if (oldIterSolver != Teuchos::null) {
843 iterativeSolver = oldIterSolver;
844 iterativeSolver->setProblem( lp );
845 iterativeSolver->setParameters( solverPL );
855 if(paramList_.get()) {
861 if (oldIterSolver != Teuchos::null) {
862 iterativeSolver = oldIterSolver;
863 iterativeSolver->setProblem( lp );
864 iterativeSolver->setParameters( solverPL );
883 lp, solverPL, iterativeSolver,
884 fwdOpSrc, prec, myPrec.get()==NULL, approxFwdOpSrc,
885 supportSolveUse, convergenceTestFrequency_
887 belosOp->setOStream(out);
888 belosOp->setVerbLevel(verbLevel);
890 if(paramList_.get()) {
892 paramList_->validateParameters(*this->getValidParameters(),1);
895 if(out.get() &&
static_cast<int>(verbLevel) > static_cast<int>(
Teuchos::VERB_LOW))
896 *out <<
"\nLeaving Thyra::BelosLinearOpWithSolveFactory<"<<ST::name()<<
">::initializeOpImpl(...) ...\n";
904 #endif // THYRA_BELOS_LINEAR_OP_WITH_SOLVE_FACTORY_HPP SUPPORT_SOLVE_UNSPECIFIED
static const std::string BlockGMRES_name
bool isCompatible(const LinearOpSourceBase< Scalar > &fwdOpSrc) const
static const std::string RCG_name
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters() const
RCP< const LinearOpSourceBase< Scalar > > extract_approxFwdOpSrc()
static void addConverter(RCP< const ParameterEntryValidator > validator, RCP< ValidatorXMLConverter > converterToAdd)
static const std::string MINRES_name
Concrete LinearOpWithSolveBase subclass in terms of Belos.
ParameterList & set(std::string const &name, T const &value, std::string const &docString="", RCP< const ParameterEntryValidator > const &validator=null)
void initializeAndReuseOp(const Teuchos::RCP< const LinearOpSourceBase< Scalar > > &fwdOpSrc, LinearOpWithSolveBase< Scalar > *Op) const
#define TEUCHOS_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
RCP< const PreconditionerBase< Scalar > > extract_prec()
T_To & dyn_cast(T_From &from)
bool supportsPreconditionerInputType(const EPreconditionerInputType precOpType) const
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters() const
static const std::string SolverTypes_name
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters() const
void unsetPreconditionerFactory(Teuchos::RCP< PreconditionerFactoryBase< Scalar > > *precFactory, std::string *precFactoryName)
Thyra specializations of MultiVecTraits and OperatorTraits.
void setParameterList(Teuchos::RCP< Teuchos::ParameterList > const ¶mList)
Teuchos::ParameterList * get() const
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters() const
static const std::string ConvergenceTestFrequency_name
void initializePreconditionedOp(const Teuchos::RCP< const LinearOpSourceBase< Scalar > > &fwdOpSrc, const Teuchos::RCP< const PreconditionerBase< Scalar > > &prec, LinearOpWithSolveBase< Scalar > *Op, const ESupportSolveUse supportSolveUse) const
static const std::string BlockCG_name
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
static const std::string PseudoBlockGMRES_name
SOLVER_TYPE_PSEUDO_BLOCK_CG
bool isExternalPrec() const
static const std::string PseudoBlockCG_name
Teuchos::RCP< Teuchos::ParameterList > getNonconstParameterList()
void validateParametersAndSetDefaults(ParameterList const &validParamList, int const depth=1000)
void setPreconditionerFactory(const Teuchos::RCP< PreconditionerFactoryBase< Scalar > > &precFactory, const std::string &precFactoryName)
Teuchos::RCP< const Teuchos::ParameterList > getParameterList() const
bool acceptsPreconditionerFactory() const
Returns true .
Teuchos::RCP< PreconditionerFactoryBase< Scalar > > getPreconditionerFactory() const
LinearOpWithSolveFactoryBase subclass implemented in terms of Belos.
ParameterList & setParameters(const ParameterList &source)
RCP< const LinearOpSourceBase< Scalar > > extract_fwdOpSrc()
static const std::string PseudoBlockStochasticCG_name
PRECONDITIONER_INPUT_TYPE_AS_OPERATOR
TypeTo as(const TypeFrom &t)
static const std::string SolverType_default
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters() const
std::string description() const
ParameterList & sublist(const std::string &name, bool mustAlreadyExist=false, const std::string &docString="")
BelosLinearOpWithSolveFactory()
Construct without preconditioner factory.
ESupportSolveUse supportSolveUse() const
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters() const
Teuchos::RCP< LinearOpWithSolveBase< Scalar > > createOp() const
void uninitializeOp(LinearOpWithSolveBase< Scalar > *Op, Teuchos::RCP< const LinearOpSourceBase< Scalar > > *fwdOpSrc, Teuchos::RCP< const PreconditionerBase< Scalar > > *prec, Teuchos::RCP< const LinearOpSourceBase< Scalar > > *approxFwdOpSrc, ESupportSolveUse *supportSolveUse) const
SOLVER_TYPE_PSEUDO_BLOCK_GMRES
Teuchos::RCP< Teuchos::ParameterList > unsetParameterList()
static const std::string SolverType_name
static const std::string TFQMR_name
void initializeApproxPreconditionedOp(const Teuchos::RCP< const LinearOpSourceBase< Scalar > > &fwdOpSrc, const Teuchos::RCP< const LinearOpSourceBase< Scalar > > &approxFwdOpSrc, LinearOpWithSolveBase< Scalar > *Op, const ESupportSolveUse supportSolveUse) const
#define TEUCHOS_TEST_FOR_EXCEPT(throw_exception_test)
void initializeOp(const Teuchos::RCP< const LinearOpSourceBase< Scalar > > &fwdOpSrc, LinearOpWithSolveBase< Scalar > *Op, const ESupportSolveUse supportSolveUse) const
static const std::string GCRODR_name