1af0996ceSBarry Smith #include <petsc/private/taoimpl.h> /*I "petsctao.h" I*/ 20f0abf79SStefano Zampini #include <petsc/private/snesimpl.h> 3a7e14dcfSSatish Balay 4441846f8SBarry Smith PetscBool TaoRegisterAllCalled = PETSC_FALSE; 5441846f8SBarry Smith PetscFunctionList TaoList = NULL; 6a7e14dcfSSatish Balay 7441846f8SBarry Smith PetscClassId TAO_CLASSID; 80ebee16dSLisandro Dalcin 90ebee16dSLisandro Dalcin PetscLogEvent TAO_Solve; 100ebee16dSLisandro Dalcin PetscLogEvent TAO_ObjectiveEval; 110ebee16dSLisandro Dalcin PetscLogEvent TAO_GradientEval; 120ebee16dSLisandro Dalcin PetscLogEvent TAO_ObjGradEval; 130ebee16dSLisandro Dalcin PetscLogEvent TAO_HessianEval; 140ebee16dSLisandro Dalcin PetscLogEvent TAO_JacobianEval; 150ebee16dSLisandro Dalcin PetscLogEvent TAO_ConstraintsEval; 16a7e14dcfSSatish Balay 1783c8fe1dSLisandro Dalcin const char *TaoSubSetTypes[] = {"subvec", "mask", "matrixfree", "TaoSubSetType", "TAO_SUBSET_", NULL}; 18a7e14dcfSSatish Balay 19e882e171SHong Zhang struct _n_TaoMonitorDrawCtx { 20e882e171SHong Zhang PetscViewer viewer; 21e882e171SHong Zhang PetscInt howoften; /* when > 0 uses iteration % howoften, when negative only final solution plotted */ 22e882e171SHong Zhang }; 23e882e171SHong Zhang 24d71ae5a4SJacob Faibussowitsch static PetscErrorCode KSPPreSolve_TAOEW_Private(KSP ksp, Vec b, Vec x, Tao tao) 25d71ae5a4SJacob Faibussowitsch { 260f0abf79SStefano Zampini SNES snes_ewdummy = tao->snes_ewdummy; 270f0abf79SStefano Zampini 280f0abf79SStefano Zampini PetscFunctionBegin; 290f0abf79SStefano Zampini if (!snes_ewdummy) PetscFunctionReturn(0); 300f0abf79SStefano Zampini /* populate snes_ewdummy struct values used in KSPPreSolve_SNESEW */ 310f0abf79SStefano Zampini snes_ewdummy->vec_func = b; 320f0abf79SStefano Zampini snes_ewdummy->rtol = tao->gttol; 330f0abf79SStefano Zampini snes_ewdummy->iter = tao->niter; 340f0abf79SStefano Zampini PetscCall(VecNorm(b, NORM_2, &snes_ewdummy->norm)); 350f0abf79SStefano Zampini PetscCall(KSPPreSolve_SNESEW(ksp, b, x, snes_ewdummy)); 360f0abf79SStefano Zampini snes_ewdummy->vec_func = NULL; 370f0abf79SStefano Zampini PetscFunctionReturn(0); 380f0abf79SStefano Zampini } 390f0abf79SStefano Zampini 40d71ae5a4SJacob Faibussowitsch static PetscErrorCode KSPPostSolve_TAOEW_Private(KSP ksp, Vec b, Vec x, Tao tao) 41d71ae5a4SJacob Faibussowitsch { 420f0abf79SStefano Zampini SNES snes_ewdummy = tao->snes_ewdummy; 430f0abf79SStefano Zampini 440f0abf79SStefano Zampini PetscFunctionBegin; 450f0abf79SStefano Zampini if (!snes_ewdummy) PetscFunctionReturn(0); 460f0abf79SStefano Zampini PetscCall(KSPPostSolve_SNESEW(ksp, b, x, snes_ewdummy)); 470f0abf79SStefano Zampini PetscFunctionReturn(0); 480f0abf79SStefano Zampini } 490f0abf79SStefano Zampini 50d71ae5a4SJacob Faibussowitsch static PetscErrorCode TaoSetUpEW_Private(Tao tao) 51d71ae5a4SJacob Faibussowitsch { 520f0abf79SStefano Zampini SNESKSPEW *kctx; 530f0abf79SStefano Zampini const char *ewprefix; 540f0abf79SStefano Zampini 550f0abf79SStefano Zampini PetscFunctionBegin; 560f0abf79SStefano Zampini if (!tao->ksp) PetscFunctionReturn(0); 570f0abf79SStefano Zampini if (tao->ksp_ewconv) { 580f0abf79SStefano Zampini if (!tao->snes_ewdummy) PetscCall(SNESCreate(PetscObjectComm((PetscObject)tao), &tao->snes_ewdummy)); 590f0abf79SStefano Zampini tao->snes_ewdummy->ksp_ewconv = PETSC_TRUE; 600f0abf79SStefano Zampini PetscCall(KSPSetPreSolve(tao->ksp, (PetscErrorCode(*)(KSP, Vec, Vec, void *))KSPPreSolve_TAOEW_Private, tao)); 610f0abf79SStefano Zampini PetscCall(KSPSetPostSolve(tao->ksp, (PetscErrorCode(*)(KSP, Vec, Vec, void *))KSPPostSolve_TAOEW_Private, tao)); 620f0abf79SStefano Zampini 630f0abf79SStefano Zampini PetscCall(KSPGetOptionsPrefix(tao->ksp, &ewprefix)); 640f0abf79SStefano Zampini kctx = (SNESKSPEW *)tao->snes_ewdummy->kspconvctx; 650f0abf79SStefano Zampini PetscCall(SNESEWSetFromOptions_Private(kctx, PetscObjectComm((PetscObject)tao), ewprefix)); 660f0abf79SStefano Zampini } else PetscCall(SNESDestroy(&tao->snes_ewdummy)); 670f0abf79SStefano Zampini PetscFunctionReturn(0); 680f0abf79SStefano Zampini } 690f0abf79SStefano Zampini 70a7e14dcfSSatish Balay /*@ 7165ba42b6SBarry Smith TaoCreate - Creates a Tao solver 72a7e14dcfSSatish Balay 73d083f849SBarry Smith Collective 74a7e14dcfSSatish Balay 75a7e14dcfSSatish Balay Input Parameter: 76a7e14dcfSSatish Balay . comm - MPI communicator 77a7e14dcfSSatish Balay 78a7e14dcfSSatish Balay Output Parameter: 79441846f8SBarry Smith . newtao - the new Tao context 80a7e14dcfSSatish Balay 81a7e14dcfSSatish Balay Available methods include: 8265ba42b6SBarry Smith + `TAONLS` - nls Newton's method with line search for unconstrained minimization 8365ba42b6SBarry Smith . `TAONTR` - ntr Newton's method with trust region for unconstrained minimization 8465ba42b6SBarry Smith . `TAONTL` - ntl Newton's method with trust region, line search for unconstrained minimization 8565ba42b6SBarry Smith . `TAOLMVM` - lmvm Limited memory variable metric method for unconstrained minimization 8665ba42b6SBarry Smith . `TAOCG` - cg Nonlinear conjugate gradient method for unconstrained minimization 8765ba42b6SBarry Smith . `TAONM` - nm Nelder-Mead algorithm for derivate-free unconstrained minimization 8865ba42b6SBarry Smith . `TAOTRON` - tron Newton Trust Region method for bound constrained minimization 8965ba42b6SBarry Smith . `TAOGPCG` - gpcg Newton Trust Region method for quadratic bound constrained minimization 9065ba42b6SBarry Smith . `TAOBLMVM` - blmvm Limited memory variable metric method for bound constrained minimization 9165ba42b6SBarry Smith . `TAOLCL` - lcl Linearly constrained Lagrangian method for pde-constrained minimization 9265ba42b6SBarry Smith - `TAOPOUNDERS` - pounders Model-based algorithm for nonlinear least squares 93a7e14dcfSSatish Balay 94a7e14dcfSSatish Balay Options Database Keys: 9565ba42b6SBarry Smith . -tao_type - select which method Tao should use 96a7e14dcfSSatish Balay 97a7e14dcfSSatish Balay Level: beginner 98a7e14dcfSSatish Balay 9965ba42b6SBarry Smith .seealso: `Tao`, `TaoSolve()`, `TaoDestroy()`, `TAOSetFromOptions()`, `TAOSetType()` 100a7e14dcfSSatish Balay @*/ 101d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoCreate(MPI_Comm comm, Tao *newtao) 102d71ae5a4SJacob Faibussowitsch { 103441846f8SBarry Smith Tao tao; 104a7e14dcfSSatish Balay 105a7e14dcfSSatish Balay PetscFunctionBegin; 106a7e14dcfSSatish Balay PetscValidPointer(newtao, 2); 1079566063dSJacob Faibussowitsch PetscCall(TaoInitializePackage()); 1089566063dSJacob Faibussowitsch PetscCall(TaoLineSearchInitializePackage()); 1099566063dSJacob Faibussowitsch PetscCall(PetscHeaderCreate(tao, TAO_CLASSID, "Tao", "Optimization solver", "Tao", comm, TaoDestroy, TaoView)); 110a7e14dcfSSatish Balay 11194511df7SStefano Zampini /* Set non-NULL defaults */ 11294511df7SStefano Zampini tao->ops->convergencetest = TaoDefaultConvergenceTest; 113a7e14dcfSSatish Balay 114a7e14dcfSSatish Balay tao->max_it = 10000; 11594511df7SStefano Zampini tao->max_funcs = -1; 1166f4723b1SBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 1176f4723b1SBarry Smith tao->gatol = 1e-5; 1186f4723b1SBarry Smith tao->grtol = 1e-5; 1196f1f5f59SAlp Dener tao->crtol = 1e-5; 1206f1f5f59SAlp Dener tao->catol = 1e-5; 1216f4723b1SBarry Smith #else 122a7e14dcfSSatish Balay tao->gatol = 1e-8; 123a7e14dcfSSatish Balay tao->grtol = 1e-8; 1246f1f5f59SAlp Dener tao->crtol = 1e-8; 1256f1f5f59SAlp Dener tao->catol = 1e-8; 1266f4723b1SBarry Smith #endif 1276552cf8aSJason Sarich tao->gttol = 0.0; 1284f1535bcSTodd Munson tao->steptol = 0.0; 129e270355aSBarry Smith tao->trust0 = PETSC_INFINITY; 1306f4723b1SBarry Smith tao->fmin = PETSC_NINFINITY; 1319fa2b5dcSStefano Zampini 132a7e14dcfSSatish Balay tao->hist_reset = PETSC_TRUE; 133a7e14dcfSSatish Balay 1349566063dSJacob Faibussowitsch PetscCall(TaoResetStatistics(tao)); 135a7e14dcfSSatish Balay *newtao = tao; 136a7e14dcfSSatish Balay PetscFunctionReturn(0); 137a7e14dcfSSatish Balay } 138a7e14dcfSSatish Balay 139a7e14dcfSSatish Balay /*@ 140a7e14dcfSSatish Balay TaoSolve - Solves an optimization problem min F(x) s.t. l <= x <= u 141a7e14dcfSSatish Balay 142c3339decSBarry Smith Collective 143a7e14dcfSSatish Balay 144a7e14dcfSSatish Balay Input Parameters: 145441846f8SBarry Smith . tao - the Tao context 146a7e14dcfSSatish Balay 147*67be906fSBarry Smith Level: beginner 148*67be906fSBarry Smith 149a7e14dcfSSatish Balay Notes: 15065ba42b6SBarry Smith The user must set up the Tao with calls to `TaoSetSolution()`, `TaoSetObjective()`, `TaoSetGradient()`, and (if using 2nd order method) `TaoSetHessian()`. 151a7e14dcfSSatish Balay 152*67be906fSBarry Smith You should call `TaoGetConvergedReason()` or run with `-tao_converged_reason` to determine if the optimization algorithm actually succeeded or 153a35d58b8SBarry Smith why it failed. 154a35d58b8SBarry Smith 15565ba42b6SBarry Smith .seealso: `Tao`, `TaoCreate()`, `TaoSetObjective()`, `TaoSetGradient()`, `TaoSetHessian()`, `TaoGetConvergedReason()`, `TaoSetUp()` 156a7e14dcfSSatish Balay @*/ 157d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSolve(Tao tao) 158d71ae5a4SJacob Faibussowitsch { 159e2379f4fSBarry Smith static PetscBool set = PETSC_FALSE; 16047a47007SBarry Smith 161a7e14dcfSSatish Balay PetscFunctionBegin; 162441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 163d0609cedSBarry Smith PetscCall(PetscCitationsRegister("@TechReport{tao-user-ref,\n" 164e2379f4fSBarry Smith "title = {Toolkit for Advanced Optimization (TAO) Users Manual},\n" 165e2379f4fSBarry Smith "author = {Todd Munson and Jason Sarich and Stefan Wild and Steve Benson and Lois Curfman McInnes},\n" 166e2379f4fSBarry Smith "Institution = {Argonne National Laboratory},\n" 167e2379f4fSBarry Smith "Year = 2014,\n" 168e2379f4fSBarry Smith "Number = {ANL/MCS-TM-322 - Revision 3.5},\n" 1699371c9d4SSatish Balay "url = {https://www.mcs.anl.gov/research/projects/tao/}\n}\n", 1709371c9d4SSatish Balay &set)); 171494bef23SAlp Dener tao->header_printed = PETSC_FALSE; 1729566063dSJacob Faibussowitsch PetscCall(TaoSetUp(tao)); 1739566063dSJacob Faibussowitsch PetscCall(TaoResetStatistics(tao)); 1741baa6e33SBarry Smith if (tao->linesearch) PetscCall(TaoLineSearchReset(tao->linesearch)); 175a7e14dcfSSatish Balay 1769566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(TAO_Solve, tao, 0, 0, 0)); 177dbbe0bcdSBarry Smith PetscTryTypeMethod(tao, solve); 1789566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(TAO_Solve, tao, 0, 0, 0)); 179a7e14dcfSSatish Balay 1809566063dSJacob Faibussowitsch PetscCall(VecViewFromOptions(tao->solution, (PetscObject)tao, "-tao_view_solution")); 1817cf37e64SBarry Smith 1828931d482SJason Sarich tao->ntotalits += tao->niter; 183a7e14dcfSSatish Balay 184a7e14dcfSSatish Balay if (tao->printreason) { 185f642d338SBarry Smith PetscViewer viewer = PETSC_VIEWER_STDOUT_(((PetscObject)tao)->comm); 186f642d338SBarry Smith PetscCall(PetscViewerASCIIAddTab(viewer, ((PetscObject)tao)->tablevel)); 187a7e14dcfSSatish Balay if (tao->reason > 0) { 188f642d338SBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, " TAO %s solve converged due to %s iterations %" PetscInt_FMT "\n", ((PetscObject)tao)->prefix ? ((PetscObject)tao)->prefix : "", TaoConvergedReasons[tao->reason], tao->niter)); 189a7e14dcfSSatish Balay } else { 190f642d338SBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, " TAO %s solve did not converge due to %s iteration %" PetscInt_FMT "\n", ((PetscObject)tao)->prefix ? ((PetscObject)tao)->prefix : "", TaoConvergedReasons[tao->reason], tao->niter)); 191a7e14dcfSSatish Balay } 192f642d338SBarry Smith PetscCall(PetscViewerASCIISubtractTab(viewer, ((PetscObject)tao)->tablevel)); 193a7e14dcfSSatish Balay } 194f642d338SBarry Smith PetscCall(TaoViewFromOptions(tao, NULL, "-tao_view")); 195a7e14dcfSSatish Balay PetscFunctionReturn(0); 196a7e14dcfSSatish Balay } 197a7e14dcfSSatish Balay 198a7e14dcfSSatish Balay /*@ 199a7e14dcfSSatish Balay TaoSetUp - Sets up the internal data structures for the later use 200a7e14dcfSSatish Balay of a Tao solver 201a7e14dcfSSatish Balay 202c3339decSBarry Smith Collective 203a7e14dcfSSatish Balay 204a7e14dcfSSatish Balay Input Parameters: 20565ba42b6SBarry Smith . tao - the Tao context 206a7e14dcfSSatish Balay 207*67be906fSBarry Smith Level: advanced 208*67be906fSBarry Smith 209a7e14dcfSSatish Balay Notes: 21065ba42b6SBarry Smith The user will not need to explicitly call `TaoSetUp()`, as it will 21165ba42b6SBarry Smith automatically be called in `TaoSolve()`. However, if the user 21265ba42b6SBarry Smith desires to call it explicitly, it should come after `TaoCreate()` 21365ba42b6SBarry Smith and any TaoSetSomething() routines, but before `TaoSolve()`. 214a7e14dcfSSatish Balay 21565ba42b6SBarry Smith .seealso: `Tao`, `TaoCreate()`, `TaoSolve()` 216a7e14dcfSSatish Balay @*/ 217d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetUp(Tao tao) 218d71ae5a4SJacob Faibussowitsch { 219a7e14dcfSSatish Balay PetscFunctionBegin; 220441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 221a7e14dcfSSatish Balay if (tao->setupcalled) PetscFunctionReturn(0); 2220f0abf79SStefano Zampini PetscCall(TaoSetUpEW_Private(tao)); 2233c859ba3SBarry Smith PetscCheck(tao->solution, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_WRONGSTATE, "Must call TaoSetSolution"); 224dbbe0bcdSBarry Smith PetscTryTypeMethod(tao, setup); 225a7e14dcfSSatish Balay tao->setupcalled = PETSC_TRUE; 226a7e14dcfSSatish Balay PetscFunctionReturn(0); 227a7e14dcfSSatish Balay } 228a7e14dcfSSatish Balay 2291fb7b255SJunchao Zhang /*@C 23065ba42b6SBarry Smith TaoDestroy - Destroys the Tao context that was created with `TaoCreate()` 231a7e14dcfSSatish Balay 232c3339decSBarry Smith Collective 233a7e14dcfSSatish Balay 234a7e14dcfSSatish Balay Input Parameter: 235441846f8SBarry Smith . tao - the Tao context 236a7e14dcfSSatish Balay 237a7e14dcfSSatish Balay Level: beginner 238a7e14dcfSSatish Balay 23965ba42b6SBarry Smith .seealso: `Tao`, `TaoCreate()`, `TaoSolve()` 240a7e14dcfSSatish Balay @*/ 241d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDestroy(Tao *tao) 242d71ae5a4SJacob Faibussowitsch { 243a7e14dcfSSatish Balay PetscFunctionBegin; 244a7e14dcfSSatish Balay if (!*tao) PetscFunctionReturn(0); 245441846f8SBarry Smith PetscValidHeaderSpecific(*tao, TAO_CLASSID, 1); 2469371c9d4SSatish Balay if (--((PetscObject)*tao)->refct > 0) { 2479371c9d4SSatish Balay *tao = NULL; 2489371c9d4SSatish Balay PetscFunctionReturn(0); 249a7e14dcfSSatish Balay } 2509371c9d4SSatish Balay 25148a46eb9SPierre Jolivet if ((*tao)->ops->destroy) PetscCall((*((*tao))->ops->destroy)(*tao)); 2529566063dSJacob Faibussowitsch PetscCall(KSPDestroy(&(*tao)->ksp)); 2530f0abf79SStefano Zampini PetscCall(SNESDestroy(&(*tao)->snes_ewdummy)); 2549566063dSJacob Faibussowitsch PetscCall(TaoLineSearchDestroy(&(*tao)->linesearch)); 255a7e14dcfSSatish Balay 256a7e14dcfSSatish Balay if ((*tao)->ops->convergencedestroy) { 2579566063dSJacob Faibussowitsch PetscCall((*(*tao)->ops->convergencedestroy)((*tao)->cnvP)); 25848a46eb9SPierre Jolivet if ((*tao)->jacobian_state_inv) PetscCall(MatDestroy(&(*tao)->jacobian_state_inv)); 259a7e14dcfSSatish Balay } 2609566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->solution)); 2619566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->gradient)); 2629566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->ls_res)); 263a7e14dcfSSatish Balay 264a9603a14SPatrick Farrell if ((*tao)->gradient_norm) { 2659566063dSJacob Faibussowitsch PetscCall(PetscObjectDereference((PetscObject)(*tao)->gradient_norm)); 2669566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->gradient_norm_tmp)); 267a9603a14SPatrick Farrell } 268a9603a14SPatrick Farrell 2699566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->XL)); 2709566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->XU)); 2719566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->IL)); 2729566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->IU)); 2739566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->DE)); 2749566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->DI)); 2759566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->constraints)); 2769566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->constraints_equality)); 2779566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->constraints_inequality)); 2789566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->stepdirection)); 2799566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->hessian_pre)); 2809566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->hessian)); 2819566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->ls_jac)); 2829566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->ls_jac_pre)); 2839566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_pre)); 2849566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian)); 2859566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_state_pre)); 2869566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_state)); 2879566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_state_inv)); 2889566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_design)); 2899566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_equality)); 2909566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_equality_pre)); 2919566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_inequality)); 2929566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_inequality_pre)); 2939566063dSJacob Faibussowitsch PetscCall(ISDestroy(&(*tao)->state_is)); 2949566063dSJacob Faibussowitsch PetscCall(ISDestroy(&(*tao)->design_is)); 2959566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->res_weights_v)); 2969566063dSJacob Faibussowitsch PetscCall(TaoCancelMonitors(*tao)); 29748a46eb9SPierre Jolivet if ((*tao)->hist_malloc) PetscCall(PetscFree4((*tao)->hist_obj, (*tao)->hist_resid, (*tao)->hist_cnorm, (*tao)->hist_lits)); 298737f463aSAlp Dener if ((*tao)->res_weights_n) { 2999566063dSJacob Faibussowitsch PetscCall(PetscFree((*tao)->res_weights_rows)); 3009566063dSJacob Faibussowitsch PetscCall(PetscFree((*tao)->res_weights_cols)); 3019566063dSJacob Faibussowitsch PetscCall(PetscFree((*tao)->res_weights_w)); 302125ddc8aSJason Sarich } 3039566063dSJacob Faibussowitsch PetscCall(PetscHeaderDestroy(tao)); 304a7e14dcfSSatish Balay PetscFunctionReturn(0); 305a7e14dcfSSatish Balay } 306a7e14dcfSSatish Balay 307a7e14dcfSSatish Balay /*@ 30865ba42b6SBarry Smith TaoKSPSetUseEW - Sets `SNES` use Eisenstat-Walker method for 3090f0abf79SStefano Zampini computing relative tolerance for linear solvers. 3100f0abf79SStefano Zampini 311c3339decSBarry Smith Logically Collective 3120f0abf79SStefano Zampini 3130f0abf79SStefano Zampini Input Parameters: 3140f0abf79SStefano Zampini + tao - Tao context 31565ba42b6SBarry Smith - flag - `PETSC_TRUE` or `PETSC_FALSE` 3160f0abf79SStefano Zampini 317*67be906fSBarry Smith Level: advanced 318*67be906fSBarry Smith 3190f0abf79SStefano Zampini Notes: 32065ba42b6SBarry Smith See `SNESKSPSetUseEW()` for customization details. 3210f0abf79SStefano Zampini 3220f0abf79SStefano Zampini Reference: 3230f0abf79SStefano Zampini S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 3240f0abf79SStefano Zampini inexact Newton method", SISC 17 (1), pp.16-32, 1996. 3250f0abf79SStefano Zampini 32665ba42b6SBarry Smith .seealso: `Tao`, `SNESKSPSetUseEW()` 3270f0abf79SStefano Zampini @*/ 328d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoKSPSetUseEW(Tao tao, PetscBool flag) 329d71ae5a4SJacob Faibussowitsch { 3300f0abf79SStefano Zampini PetscFunctionBegin; 3310f0abf79SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 3320f0abf79SStefano Zampini PetscValidLogicalCollectiveBool(tao, flag, 2); 3330f0abf79SStefano Zampini tao->ksp_ewconv = flag; 3340f0abf79SStefano Zampini PetscFunctionReturn(0); 3350f0abf79SStefano Zampini } 3360f0abf79SStefano Zampini 3370f0abf79SStefano Zampini /*@ 338441846f8SBarry Smith TaoSetFromOptions - Sets various Tao parameters from user 339a7e14dcfSSatish Balay options. 340a7e14dcfSSatish Balay 341c3339decSBarry Smith Collective 342a7e14dcfSSatish Balay 34301d2d390SJose E. Roman Input Parameter: 344441846f8SBarry Smith . tao - the Tao solver context 345a7e14dcfSSatish Balay 346a7e14dcfSSatish Balay options Database Keys: 34765ba42b6SBarry Smith + -tao_type <type> - The algorithm that Tao uses (lmvm, nls, etc.) 348a7e14dcfSSatish Balay . -tao_gatol <gatol> - absolute error tolerance for ||gradient|| 349a7e14dcfSSatish Balay . -tao_grtol <grtol> - relative error tolerance for ||gradient|| 350a7e14dcfSSatish Balay . -tao_gttol <gttol> - reduction of ||gradient|| relative to initial gradient 351a7e14dcfSSatish Balay . -tao_max_it <max> - sets maximum number of iterations 352a7e14dcfSSatish Balay . -tao_max_funcs <max> - sets maximum number of function evaluations 353a7e14dcfSSatish Balay . -tao_fmin <fmin> - stop if function value reaches fmin 354a7e14dcfSSatish Balay . -tao_steptol <tol> - stop if trust region radius less than <tol> 355a7e14dcfSSatish Balay . -tao_trust0 <t> - initial trust region radius 356a7e14dcfSSatish Balay . -tao_monitor - prints function value and residual at each iteration 357a7e14dcfSSatish Balay . -tao_smonitor - same as tao_monitor, but truncates very small values 358a7e14dcfSSatish Balay . -tao_cmonitor - prints function value, residual, and constraint norm at each iteration 359a7e14dcfSSatish Balay . -tao_view_solution - prints solution vector at each iteration 3604a48860cSAlp Dener . -tao_view_ls_residual - prints least-squares residual vector at each iteration 361147403d9SBarry Smith . -tao_view_stepdirection - prints step direction vector at each iteration 362a7e14dcfSSatish Balay . -tao_view_gradient - prints gradient vector at each iteration 363a7e14dcfSSatish Balay . -tao_draw_solution - graphically view solution vector at each iteration 364a7e14dcfSSatish Balay . -tao_draw_step - graphically view step vector at each iteration 365a7e14dcfSSatish Balay . -tao_draw_gradient - graphically view gradient at each iteration 366a7e14dcfSSatish Balay . -tao_fd_gradient - use gradient computed with finite differences 367f4c1ad5cSStefano Zampini . -tao_fd_hessian - use hessian computed with finite differences 368f4c1ad5cSStefano Zampini . -tao_mf_hessian - use matrix-free hessian computed with finite differences 369a7e14dcfSSatish Balay . -tao_cancelmonitors - cancels all monitors (except those set with command line) 370441846f8SBarry Smith . -tao_view - prints information about the Tao after solving 37165ba42b6SBarry Smith - -tao_converged_reason - prints the reason Tao stopped iterating 372a7e14dcfSSatish Balay 373*67be906fSBarry Smith Level: beginner 374*67be906fSBarry Smith 375*67be906fSBarry Smith Note: 376a7e14dcfSSatish Balay To see all options, run your program with the -help option or consult the 37765ba42b6SBarry Smith user's manual. Should be called after `TaoCreate()` but before `TaoSolve()` 378a7e14dcfSSatish Balay 37965ba42b6SBarry Smith .seealso: `Tao`, `TaoCreate()`, `TaoSolve()` 380a7e14dcfSSatish Balay @*/ 381d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetFromOptions(Tao tao) 382d71ae5a4SJacob Faibussowitsch { 383b625d6c7SJed Brown TaoType default_type = TAOLMVM; 384a7e14dcfSSatish Balay char type[256], monfilename[PETSC_MAX_PATH_LEN]; 385a7e14dcfSSatish Balay PetscViewer monviewer; 386a7e14dcfSSatish Balay PetscBool flg; 387a7e14dcfSSatish Balay MPI_Comm comm; 388a7e14dcfSSatish Balay 389a7e14dcfSSatish Balay PetscFunctionBegin; 390441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 3919566063dSJacob Faibussowitsch PetscCall(PetscObjectGetComm((PetscObject)tao, &comm)); 392125ddc8aSJason Sarich 39360b70c5cSStefano Zampini if (((PetscObject)tao)->type_name) default_type = ((PetscObject)tao)->type_name; 39460b70c5cSStefano Zampini 395d0609cedSBarry Smith PetscObjectOptionsBegin((PetscObject)tao); 396a7e14dcfSSatish Balay /* Check for type from options */ 3979566063dSJacob Faibussowitsch PetscCall(PetscOptionsFList("-tao_type", "Tao Solver type", "TaoSetType", TaoList, default_type, type, 256, &flg)); 398a7e14dcfSSatish Balay if (flg) { 3999566063dSJacob Faibussowitsch PetscCall(TaoSetType(tao, type)); 400a7e14dcfSSatish Balay } else if (!((PetscObject)tao)->type_name) { 4019566063dSJacob Faibussowitsch PetscCall(TaoSetType(tao, default_type)); 402a7e14dcfSSatish Balay } 403a7e14dcfSSatish Balay 40460b70c5cSStefano Zampini /* Tao solvers do not set the prefix, set it here if not yet done 40560b70c5cSStefano Zampini We do it after SetType since solver may have been changed */ 40660b70c5cSStefano Zampini if (tao->linesearch) { 40760b70c5cSStefano Zampini const char *prefix; 40860b70c5cSStefano Zampini PetscCall(TaoLineSearchGetOptionsPrefix(tao->linesearch, &prefix)); 40960b70c5cSStefano Zampini if (!prefix) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch, ((PetscObject)(tao))->prefix)); 41060b70c5cSStefano Zampini } 41160b70c5cSStefano Zampini 4129566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_catol", "Stop if constraints violations within", "TaoSetConstraintTolerances", tao->catol, &tao->catol, &flg)); 4136552cf8aSJason Sarich if (flg) tao->catol_changed = PETSC_TRUE; 4146aad120cSJose E. Roman PetscCall(PetscOptionsReal("-tao_crtol", "Stop if relative constraint violations within", "TaoSetConstraintTolerances", tao->crtol, &tao->crtol, &flg)); 4156552cf8aSJason Sarich if (flg) tao->crtol_changed = PETSC_TRUE; 4169566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_gatol", "Stop if norm of gradient less than", "TaoSetTolerances", tao->gatol, &tao->gatol, &flg)); 4176552cf8aSJason Sarich if (flg) tao->gatol_changed = PETSC_TRUE; 4189566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_grtol", "Stop if norm of gradient divided by the function value is less than", "TaoSetTolerances", tao->grtol, &tao->grtol, &flg)); 4196552cf8aSJason Sarich if (flg) tao->grtol_changed = PETSC_TRUE; 4209566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_gttol", "Stop if the norm of the gradient is less than the norm of the initial gradient times tol", "TaoSetTolerances", tao->gttol, &tao->gttol, &flg)); 4216552cf8aSJason Sarich if (flg) tao->gttol_changed = PETSC_TRUE; 4229566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-tao_max_it", "Stop if iteration number exceeds", "TaoSetMaximumIterations", tao->max_it, &tao->max_it, &flg)); 4236552cf8aSJason Sarich if (flg) tao->max_it_changed = PETSC_TRUE; 4249566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-tao_max_funcs", "Stop if number of function evaluations exceeds", "TaoSetMaximumFunctionEvaluations", tao->max_funcs, &tao->max_funcs, &flg)); 4256552cf8aSJason Sarich if (flg) tao->max_funcs_changed = PETSC_TRUE; 4269566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_fmin", "Stop if function less than", "TaoSetFunctionLowerBound", tao->fmin, &tao->fmin, &flg)); 4276552cf8aSJason Sarich if (flg) tao->fmin_changed = PETSC_TRUE; 4289566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_steptol", "Stop if step size or trust region radius less than", "", tao->steptol, &tao->steptol, &flg)); 4296552cf8aSJason Sarich if (flg) tao->steptol_changed = PETSC_TRUE; 4309566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_trust0", "Initial trust region radius", "TaoSetTrustRegionRadius", tao->trust0, &tao->trust0, &flg)); 4316552cf8aSJason Sarich if (flg) tao->trust0_changed = PETSC_TRUE; 4329566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_view_solution", "view solution vector after each evaluation", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 433a7e14dcfSSatish Balay if (flg) { 4349566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4359566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoSolutionMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 436a7e14dcfSSatish Balay } 437a7e14dcfSSatish Balay 43865ba42b6SBarry Smith PetscCall(PetscOptionsBool("-tao_converged_reason", "Print reason for Tao converged", "TaoSolve", tao->printreason, &tao->printreason, NULL)); 4399566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_view_gradient", "view gradient vector after each evaluation", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 440a7e14dcfSSatish Balay if (flg) { 4419566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4429566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoGradientMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 443a7e14dcfSSatish Balay } 444a7e14dcfSSatish Balay 4459566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_view_stepdirection", "view step direction vector after each iteration", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 446a7e14dcfSSatish Balay if (flg) { 4479566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4489566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoStepDirectionMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 449a7e14dcfSSatish Balay } 450a7e14dcfSSatish Balay 4519566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_view_residual", "view least-squares residual vector after each evaluation", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 452a7e14dcfSSatish Balay if (flg) { 4539566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4549566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoResidualMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 455a7e14dcfSSatish Balay } 456a7e14dcfSSatish Balay 4579566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_monitor", "Use the default convergence monitor", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 458a7e14dcfSSatish Balay if (flg) { 4599566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4609566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoMonitorDefault, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 461a7e14dcfSSatish Balay } 462a7e14dcfSSatish Balay 4639566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_gmonitor", "Use the convergence monitor with extra globalization info", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 4648d5ead36SAlp Dener if (flg) { 4659566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4669566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoDefaultGMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 4678d5ead36SAlp Dener } 4688d5ead36SAlp Dener 4699566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_smonitor", "Use the short convergence monitor", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 470a7e14dcfSSatish Balay if (flg) { 4719566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4729566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoDefaultSMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 473a7e14dcfSSatish Balay } 474a7e14dcfSSatish Balay 4759566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_cmonitor", "Use the default convergence monitor with constraint norm", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 476a7e14dcfSSatish Balay if (flg) { 4779566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4789566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoDefaultCMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 479a7e14dcfSSatish Balay } 480a7e14dcfSSatish Balay 4818afaa268SBarry Smith flg = PETSC_FALSE; 4829566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_cancelmonitors", "cancel all monitors and call any registered destroy routines", "TaoCancelMonitors", flg, &flg, NULL)); 4839566063dSJacob Faibussowitsch if (flg) PetscCall(TaoCancelMonitors(tao)); 484a7e14dcfSSatish Balay 4858afaa268SBarry Smith flg = PETSC_FALSE; 4869566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_draw_solution", "Plot solution vector at each iteration", "TaoSetMonitor", flg, &flg, NULL)); 487a7e14dcfSSatish Balay if (flg) { 488e882e171SHong Zhang TaoMonitorDrawCtx drawctx; 489e882e171SHong Zhang PetscInt howoften = 1; 4909566063dSJacob Faibussowitsch PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao), NULL, NULL, PETSC_DECIDE, PETSC_DECIDE, 300, 300, howoften, &drawctx)); 4919566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoDrawSolutionMonitor, drawctx, (PetscErrorCode(*)(void **))TaoMonitorDrawCtxDestroy)); 492a7e14dcfSSatish Balay } 493a7e14dcfSSatish Balay 4948afaa268SBarry Smith flg = PETSC_FALSE; 4959566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_draw_step", "plots step direction at each iteration", "TaoSetMonitor", flg, &flg, NULL)); 4961baa6e33SBarry Smith if (flg) PetscCall(TaoSetMonitor(tao, TaoDrawStepMonitor, NULL, NULL)); 497a7e14dcfSSatish Balay 4988afaa268SBarry Smith flg = PETSC_FALSE; 4999566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_draw_gradient", "plots gradient at each iteration", "TaoSetMonitor", flg, &flg, NULL)); 500a7e14dcfSSatish Balay if (flg) { 501e882e171SHong Zhang TaoMonitorDrawCtx drawctx; 502e882e171SHong Zhang PetscInt howoften = 1; 5039566063dSJacob Faibussowitsch PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao), NULL, NULL, PETSC_DECIDE, PETSC_DECIDE, 300, 300, howoften, &drawctx)); 5049566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoDrawGradientMonitor, drawctx, (PetscErrorCode(*)(void **))TaoMonitorDrawCtxDestroy)); 505a7e14dcfSSatish Balay } 5068afaa268SBarry Smith flg = PETSC_FALSE; 5079566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_fd_gradient", "compute gradient using finite differences", "TaoDefaultComputeGradient", flg, &flg, NULL)); 5081baa6e33SBarry Smith if (flg) PetscCall(TaoSetGradient(tao, NULL, TaoDefaultComputeGradient, NULL)); 509f4c1ad5cSStefano Zampini flg = PETSC_FALSE; 5109566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_fd_hessian", "compute hessian using finite differences", "TaoDefaultComputeHessian", flg, &flg, NULL)); 511f4c1ad5cSStefano Zampini if (flg) { 512f4c1ad5cSStefano Zampini Mat H; 513f4c1ad5cSStefano Zampini 5149566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)tao), &H)); 5159566063dSJacob Faibussowitsch PetscCall(MatSetType(H, MATAIJ)); 5169566063dSJacob Faibussowitsch PetscCall(TaoSetHessian(tao, H, H, TaoDefaultComputeHessian, NULL)); 5179566063dSJacob Faibussowitsch PetscCall(MatDestroy(&H)); 518f4c1ad5cSStefano Zampini } 519f4c1ad5cSStefano Zampini flg = PETSC_FALSE; 5209566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_mf_hessian", "compute matrix-free hessian using finite differences", "TaoDefaultComputeHessianMFFD", flg, &flg, NULL)); 521f4c1ad5cSStefano Zampini if (flg) { 522f4c1ad5cSStefano Zampini Mat H; 523f4c1ad5cSStefano Zampini 5249566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)tao), &H)); 5259566063dSJacob Faibussowitsch PetscCall(TaoSetHessian(tao, H, H, TaoDefaultComputeHessianMFFD, NULL)); 5269566063dSJacob Faibussowitsch PetscCall(MatDestroy(&H)); 527f4c1ad5cSStefano Zampini } 528414d97d3SAlp Dener flg = PETSC_FALSE; 5299566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_recycle_history", "enable recycling/re-using information from the previous TaoSolve() call for some algorithms", "TaoSetRecycleHistory", flg, &flg, NULL)); 5301baa6e33SBarry Smith if (flg) PetscCall(TaoSetRecycleHistory(tao, PETSC_TRUE)); 5319566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnum("-tao_subset_type", "subset type", "", TaoSubSetTypes, (PetscEnum)tao->subset_type, (PetscEnum *)&tao->subset_type, NULL)); 532a7e14dcfSSatish Balay 5330f0abf79SStefano Zampini if (tao->ksp) { 5340f0abf79SStefano Zampini PetscCall(PetscOptionsBool("-tao_ksp_ew", "Use Eisentat-Walker linear system convergence test", "TaoKSPSetUseEW", tao->ksp_ewconv, &tao->ksp_ewconv, NULL)); 5350f0abf79SStefano Zampini PetscCall(TaoKSPSetUseEW(tao, tao->ksp_ewconv)); 5360f0abf79SStefano Zampini } 5370f0abf79SStefano Zampini 538dbbe0bcdSBarry Smith PetscTryTypeMethod(tao, setfromoptions, PetscOptionsObject); 53960b70c5cSStefano Zampini 54060b70c5cSStefano Zampini /* process any options handlers added with PetscObjectAddOptionsHandler() */ 54160b70c5cSStefano Zampini PetscCall(PetscObjectProcessOptionsHandlers((PetscObject)tao, PetscOptionsObject)); 542d0609cedSBarry Smith PetscOptionsEnd(); 54360b70c5cSStefano Zampini 54460b70c5cSStefano Zampini if (tao->linesearch) PetscCall(TaoLineSearchSetFromOptions(tao->linesearch)); 545a7e14dcfSSatish Balay PetscFunctionReturn(0); 546a7e14dcfSSatish Balay } 547a7e14dcfSSatish Balay 548a7e14dcfSSatish Balay /*@C 54965ba42b6SBarry Smith TaoViewFromOptions - View a Tao options from the options database 550fe2efc57SMark 551c3339decSBarry Smith Collective 552fe2efc57SMark 553fe2efc57SMark Input Parameters: 554fe2efc57SMark + A - the Tao context 555736c3998SJose E. Roman . obj - Optional object 556736c3998SJose E. Roman - name - command line option 557fe2efc57SMark 558fe2efc57SMark Level: intermediate 559*67be906fSBarry Smith 560db781477SPatrick Sanan .seealso: `Tao`, `TaoView`, `PetscObjectViewFromOptions()`, `TaoCreate()` 561fe2efc57SMark @*/ 562d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoViewFromOptions(Tao A, PetscObject obj, const char name[]) 563d71ae5a4SJacob Faibussowitsch { 564fe2efc57SMark PetscFunctionBegin; 565fe2efc57SMark PetscValidHeaderSpecific(A, TAO_CLASSID, 1); 5669566063dSJacob Faibussowitsch PetscCall(PetscObjectViewFromOptions((PetscObject)A, obj, name)); 567fe2efc57SMark PetscFunctionReturn(0); 568fe2efc57SMark } 569fe2efc57SMark 570fe2efc57SMark /*@C 57165ba42b6SBarry Smith TaoView - Prints information about the Tao object 572a7e14dcfSSatish Balay 573c3339decSBarry Smith Collective 574a7e14dcfSSatish Balay 575a7e14dcfSSatish Balay InputParameters: 576441846f8SBarry Smith + tao - the Tao context 577a7e14dcfSSatish Balay - viewer - visualization context 578a7e14dcfSSatish Balay 579a7e14dcfSSatish Balay Options Database Key: 58065ba42b6SBarry Smith . -tao_view - Calls `TaoView()` at the end of `TaoSolve()` 581a7e14dcfSSatish Balay 582*67be906fSBarry Smith Level: beginner 583*67be906fSBarry Smith 584a7e14dcfSSatish Balay Notes: 585a7e14dcfSSatish Balay The available visualization contexts include 58665ba42b6SBarry Smith + `PETSC_VIEWER_STDOUT_SELF` - standard output (default) 58765ba42b6SBarry Smith - `PETSC_VIEWER_STDOUT_WORLD` - synchronized standard 588a7e14dcfSSatish Balay output where only the first processor opens 589a7e14dcfSSatish Balay the file. All other processors send their 590a7e14dcfSSatish Balay data to the first processor to print. 591a7e14dcfSSatish Balay 592db781477SPatrick Sanan .seealso: `PetscViewerASCIIOpen()` 593a7e14dcfSSatish Balay @*/ 594d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoView(Tao tao, PetscViewer viewer) 595d71ae5a4SJacob Faibussowitsch { 596a7e14dcfSSatish Balay PetscBool isascii, isstring; 597b625d6c7SJed Brown TaoType type; 59847a47007SBarry Smith 599a7e14dcfSSatish Balay PetscFunctionBegin; 600441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 60148a46eb9SPierre Jolivet if (!viewer) PetscCall(PetscViewerASCIIGetStdout(((PetscObject)tao)->comm, &viewer)); 602a7e14dcfSSatish Balay PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 603a7e14dcfSSatish Balay PetscCheckSameComm(tao, 1, viewer, 2); 604a7e14dcfSSatish Balay 6059566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii)); 6069566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERSTRING, &isstring)); 607a7e14dcfSSatish Balay if (isascii) { 6089566063dSJacob Faibussowitsch PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)tao, viewer)); 609a7e14dcfSSatish Balay 610a7e14dcfSSatish Balay if (tao->ops->view) { 6119566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 612dbbe0bcdSBarry Smith PetscUseTypeMethod(tao, view, viewer); 6139566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 614a7e14dcfSSatish Balay } 6158d3f3ddaSAlp Dener if (tao->linesearch) { 6169566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 6179566063dSJacob Faibussowitsch PetscCall(TaoLineSearchView(tao->linesearch, viewer)); 6189566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 6198d3f3ddaSAlp Dener } 620a7e14dcfSSatish Balay if (tao->ksp) { 6219566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 6229566063dSJacob Faibussowitsch PetscCall(KSPView(tao->ksp, viewer)); 62363a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total KSP iterations: %" PetscInt_FMT "\n", tao->ksp_tot_its)); 6249566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 625a7e14dcfSSatish Balay } 62619b5c101SAlp Dener 6279566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 62819b5c101SAlp Dener 62948a46eb9SPierre Jolivet if (tao->XL || tao->XU) PetscCall(PetscViewerASCIIPrintf(viewer, "Active Set subset type: %s\n", TaoSubSetTypes[tao->subset_type])); 630a7e14dcfSSatish Balay 6319566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: gatol=%g,", (double)tao->gatol)); 6329566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " steptol=%g,", (double)tao->steptol)); 6339566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " gttol=%g\n", (double)tao->gttol)); 6349566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Residual in Function/Gradient:=%g\n", (double)tao->residual)); 635a7e14dcfSSatish Balay 6366246e8cdSAlp Dener if (tao->constrained) { 6379566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances:")); 6389566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " catol=%g,", (double)tao->catol)); 6399566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " crtol=%g\n", (double)tao->crtol)); 6409566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Residual in Constraints:=%g\n", (double)tao->cnorm)); 641a7e14dcfSSatish Balay } 642a7e14dcfSSatish Balay 643a7e14dcfSSatish Balay if (tao->trust < tao->steptol) { 6449566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: steptol=%g\n", (double)tao->steptol)); 6459566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Final trust region radius:=%g\n", (double)tao->trust)); 646a7e14dcfSSatish Balay } 647a7e14dcfSSatish Balay 64848a46eb9SPierre Jolivet if (tao->fmin > -1.e25) PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: function minimum=%g\n", (double)tao->fmin)); 6499566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Objective value=%g\n", (double)tao->fc)); 650a7e14dcfSSatish Balay 65163a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total number of iterations=%" PetscInt_FMT ", ", tao->niter)); 65263a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " (max: %" PetscInt_FMT ")\n", tao->max_it)); 653a7e14dcfSSatish Balay 654a7e14dcfSSatish Balay if (tao->nfuncs > 0) { 65563a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total number of function evaluations=%" PetscInt_FMT ",", tao->nfuncs)); 65663a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " max: %" PetscInt_FMT "\n", tao->max_funcs)); 657a7e14dcfSSatish Balay } 658a7e14dcfSSatish Balay if (tao->ngrads > 0) { 65963a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total number of gradient evaluations=%" PetscInt_FMT ",", tao->ngrads)); 66063a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " max: %" PetscInt_FMT "\n", tao->max_funcs)); 661a7e14dcfSSatish Balay } 662a7e14dcfSSatish Balay if (tao->nfuncgrads > 0) { 66363a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total number of function/gradient evaluations=%" PetscInt_FMT ",", tao->nfuncgrads)); 66463a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " (max: %" PetscInt_FMT ")\n", tao->max_funcs)); 665a7e14dcfSSatish Balay } 66648a46eb9SPierre Jolivet if (tao->nhess > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of Hessian evaluations=%" PetscInt_FMT "\n", tao->nhess)); 66748a46eb9SPierre Jolivet if (tao->nconstraints > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of constraint function evaluations=%" PetscInt_FMT "\n", tao->nconstraints)); 66848a46eb9SPierre Jolivet if (tao->njac > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of Jacobian evaluations=%" PetscInt_FMT "\n", tao->njac)); 669a7e14dcfSSatish Balay 670a7e14dcfSSatish Balay if (tao->reason > 0) { 6719566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Solution converged: ")); 672a7e14dcfSSatish Balay switch (tao->reason) { 673d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_GATOL: 674d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)|| <= gatol\n")); 675d71ae5a4SJacob Faibussowitsch break; 676d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_GRTOL: 677d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)||/|f(X)| <= grtol\n")); 678d71ae5a4SJacob Faibussowitsch break; 679d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_GTTOL: 680d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)||/||g(X0)|| <= gttol\n")); 681d71ae5a4SJacob Faibussowitsch break; 682d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_STEPTOL: 683d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Steptol -- step size small\n")); 684d71ae5a4SJacob Faibussowitsch break; 685d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_MINF: 686d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Minf -- f < fmin\n")); 687d71ae5a4SJacob Faibussowitsch break; 688d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_USER: 689d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " User Terminated\n")); 690d71ae5a4SJacob Faibussowitsch break; 691d71ae5a4SJacob Faibussowitsch default: 692d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "\n")); 693d71ae5a4SJacob Faibussowitsch break; 694a7e14dcfSSatish Balay } 695a7e14dcfSSatish Balay } else { 6969566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Solver terminated: %d", tao->reason)); 697a7e14dcfSSatish Balay switch (tao->reason) { 698d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_MAXITS: 699d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum Iterations\n")); 700d71ae5a4SJacob Faibussowitsch break; 701d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_NAN: 702d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " NAN or Inf encountered\n")); 703d71ae5a4SJacob Faibussowitsch break; 704d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_MAXFCN: 705d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum Function Evaluations\n")); 706d71ae5a4SJacob Faibussowitsch break; 707d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_LS_FAILURE: 708d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Line Search Failure\n")); 709d71ae5a4SJacob Faibussowitsch break; 710d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_TR_REDUCTION: 711d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Trust Region too small\n")); 712d71ae5a4SJacob Faibussowitsch break; 713d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_USER: 714d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " User Terminated\n")); 715d71ae5a4SJacob Faibussowitsch break; 716d71ae5a4SJacob Faibussowitsch default: 717d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "\n")); 718d71ae5a4SJacob Faibussowitsch break; 719a7e14dcfSSatish Balay } 720a7e14dcfSSatish Balay } 7219566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 722a7e14dcfSSatish Balay } else if (isstring) { 7239566063dSJacob Faibussowitsch PetscCall(TaoGetType(tao, &type)); 7249566063dSJacob Faibussowitsch PetscCall(PetscViewerStringSPrintf(viewer, " %-3.3s", type)); 725a7e14dcfSSatish Balay } 726a7e14dcfSSatish Balay PetscFunctionReturn(0); 727a7e14dcfSSatish Balay } 728a7e14dcfSSatish Balay 729a7e14dcfSSatish Balay /*@ 730414d97d3SAlp Dener TaoSetRecycleHistory - Sets the boolean flag to enable/disable re-using 73165ba42b6SBarry Smith iterate information from the previous `TaoSolve()`. This feature is disabled by 732414d97d3SAlp Dener default. 733414d97d3SAlp Dener 734*67be906fSBarry Smith Logically Collective 735*67be906fSBarry Smith 736*67be906fSBarry Smith Input Parameters: 737*67be906fSBarry Smith + tao - the Tao context 738*67be906fSBarry Smith - recycle - boolean flag 739*67be906fSBarry Smith 740*67be906fSBarry Smith Options Database Keys: 741*67be906fSBarry Smith . -tao_recycle_history <true,false> - reuse the history 742*67be906fSBarry Smith 743*67be906fSBarry Smith Level: intermediate 744*67be906fSBarry Smith 745*67be906fSBarry Smith Notes: 74665ba42b6SBarry Smith For conjugate gradient methods (`TAOBNCG`), this re-uses the latest search direction 74765ba42b6SBarry Smith from the previous `TaoSolve()` call when computing the first search direction in a 748414d97d3SAlp Dener new solution. By default, CG methods set the first search direction to the 749414d97d3SAlp Dener negative gradient. 750414d97d3SAlp Dener 75165ba42b6SBarry Smith For quasi-Newton family of methods (`TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`), this re-uses 75265ba42b6SBarry Smith the accumulated quasi-Newton Hessian approximation from the previous `TaoSolve()` 753414d97d3SAlp Dener call. By default, QN family of methods reset the initial Hessian approximation to 754414d97d3SAlp Dener the identity matrix. 755414d97d3SAlp Dener 756414d97d3SAlp Dener For any other algorithm, this setting has no effect. 757414d97d3SAlp Dener 75865ba42b6SBarry Smith .seealso: `TaoGetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL` 759414d97d3SAlp Dener @*/ 760d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetRecycleHistory(Tao tao, PetscBool recycle) 761d71ae5a4SJacob Faibussowitsch { 762414d97d3SAlp Dener PetscFunctionBegin; 763414d97d3SAlp Dener PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 76494511df7SStefano Zampini PetscValidLogicalCollectiveBool(tao, recycle, 2); 765414d97d3SAlp Dener tao->recycle = recycle; 766414d97d3SAlp Dener PetscFunctionReturn(0); 767414d97d3SAlp Dener } 768414d97d3SAlp Dener 769414d97d3SAlp Dener /*@ 770414d97d3SAlp Dener TaoGetRecycleHistory - Retrieve the boolean flag for re-using iterate information 77165ba42b6SBarry Smith from the previous `TaoSolve()`. This feature is disabled by default. 772414d97d3SAlp Dener 773*67be906fSBarry Smith Logically Collective 774414d97d3SAlp Dener 775414d97d3SAlp Dener Input Parameters: 776147403d9SBarry Smith . tao - the Tao context 777414d97d3SAlp Dener 778414d97d3SAlp Dener Output Parameters: 779147403d9SBarry Smith . recycle - boolean flag 780414d97d3SAlp Dener 781414d97d3SAlp Dener Level: intermediate 782414d97d3SAlp Dener 78365ba42b6SBarry Smith .seealso: `TaoSetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL` 784414d97d3SAlp Dener @*/ 785d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetRecycleHistory(Tao tao, PetscBool *recycle) 786d71ae5a4SJacob Faibussowitsch { 787414d97d3SAlp Dener PetscFunctionBegin; 788414d97d3SAlp Dener PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 78994511df7SStefano Zampini PetscValidBoolPointer(recycle, 2); 790414d97d3SAlp Dener *recycle = tao->recycle; 791414d97d3SAlp Dener PetscFunctionReturn(0); 792414d97d3SAlp Dener } 793414d97d3SAlp Dener 794414d97d3SAlp Dener /*@ 79565ba42b6SBarry Smith TaoSetTolerances - Sets parameters used in Tao convergence tests 796a7e14dcfSSatish Balay 797*67be906fSBarry Smith Logically Collective 798a7e14dcfSSatish Balay 799a7e14dcfSSatish Balay Input Parameters: 800441846f8SBarry Smith + tao - the Tao context 801a7e14dcfSSatish Balay . gatol - stop if norm of gradient is less than this 802a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this 803a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by this factor 804a7e14dcfSSatish Balay 805a7e14dcfSSatish Balay Options Database Keys: 806e52336cbSBarry Smith + -tao_gatol <gatol> - Sets gatol 807a7e14dcfSSatish Balay . -tao_grtol <grtol> - Sets grtol 808a7e14dcfSSatish Balay - -tao_gttol <gttol> - Sets gttol 809a7e14dcfSSatish Balay 810a7e14dcfSSatish Balay Stopping Criteria: 811*67be906fSBarry Smith .vb 812*67be906fSBarry Smith ||g(X)|| <= gatol 813*67be906fSBarry Smith ||g(X)|| / |f(X)| <= grtol 814*67be906fSBarry Smith ||g(X)|| / ||g(X0)|| <= gttol 815*67be906fSBarry Smith .ve 816a7e14dcfSSatish Balay 817a7e14dcfSSatish Balay Level: beginner 818a7e14dcfSSatish Balay 819*67be906fSBarry Smith Note: 820*67be906fSBarry Smith Use `PETSC_DEFAULT` to leave one or more tolerances unchanged. 821a7e14dcfSSatish Balay 822*67be906fSBarry Smith .seealso: `TaoGetTolerances()` 823a7e14dcfSSatish Balay @*/ 824d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetTolerances(Tao tao, PetscReal gatol, PetscReal grtol, PetscReal gttol) 825d71ae5a4SJacob Faibussowitsch { 826a7e14dcfSSatish Balay PetscFunctionBegin; 827441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 82894511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, gatol, 2); 82994511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, grtol, 3); 83094511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, gttol, 4); 831a7e14dcfSSatish Balay 832a7e14dcfSSatish Balay if (gatol != PETSC_DEFAULT) { 833a7e14dcfSSatish Balay if (gatol < 0) { 8349566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Tried to set negative gatol -- ignored.\n")); 835a7e14dcfSSatish Balay } else { 836a7e14dcfSSatish Balay tao->gatol = PetscMax(0, gatol); 8376552cf8aSJason Sarich tao->gatol_changed = PETSC_TRUE; 838a7e14dcfSSatish Balay } 839a7e14dcfSSatish Balay } 840a7e14dcfSSatish Balay 841a7e14dcfSSatish Balay if (grtol != PETSC_DEFAULT) { 842a7e14dcfSSatish Balay if (grtol < 0) { 8439566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Tried to set negative grtol -- ignored.\n")); 844a7e14dcfSSatish Balay } else { 845a7e14dcfSSatish Balay tao->grtol = PetscMax(0, grtol); 8466552cf8aSJason Sarich tao->grtol_changed = PETSC_TRUE; 847a7e14dcfSSatish Balay } 848a7e14dcfSSatish Balay } 849a7e14dcfSSatish Balay 850a7e14dcfSSatish Balay if (gttol != PETSC_DEFAULT) { 851a7e14dcfSSatish Balay if (gttol < 0) { 8529566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Tried to set negative gttol -- ignored.\n")); 853a7e14dcfSSatish Balay } else { 854a7e14dcfSSatish Balay tao->gttol = PetscMax(0, gttol); 8556552cf8aSJason Sarich tao->gttol_changed = PETSC_TRUE; 856a7e14dcfSSatish Balay } 857a7e14dcfSSatish Balay } 858a7e14dcfSSatish Balay PetscFunctionReturn(0); 859a7e14dcfSSatish Balay } 860a7e14dcfSSatish Balay 861a7e14dcfSSatish Balay /*@ 86265ba42b6SBarry Smith TaoSetConstraintTolerances - Sets constraint tolerance parameters used in Tao convergence tests 863a7e14dcfSSatish Balay 864*67be906fSBarry Smith Logically Collective 865a7e14dcfSSatish Balay 866a7e14dcfSSatish Balay Input Parameters: 867441846f8SBarry Smith + tao - the Tao context 868*67be906fSBarry Smith . catol - absolute constraint tolerance, constraint norm must be less than `catol` for used for gatol convergence criteria 869*67be906fSBarry Smith - crtol - relative constraint tolerance, constraint norm must be less than `crtol` for used for gatol, gttol convergence criteria 870a7e14dcfSSatish Balay 871a7e14dcfSSatish Balay Options Database Keys: 872a7e14dcfSSatish Balay + -tao_catol <catol> - Sets catol 873a7e14dcfSSatish Balay - -tao_crtol <crtol> - Sets crtol 874a7e14dcfSSatish Balay 875a7e14dcfSSatish Balay Level: intermediate 876a7e14dcfSSatish Balay 877*67be906fSBarry Smith Notes: 878*67be906fSBarry Smith Use `PETSC_DEFAULT` to leave any tolerance unchanged. 879a7e14dcfSSatish Balay 880*67be906fSBarry Smith .seealso: `TaoGetTolerances()`, `TaoGetConstraintTolerances()`, `TaoSetTolerances()` 881a7e14dcfSSatish Balay @*/ 882d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConstraintTolerances(Tao tao, PetscReal catol, PetscReal crtol) 883d71ae5a4SJacob Faibussowitsch { 884a7e14dcfSSatish Balay PetscFunctionBegin; 885441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 88694511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, catol, 2); 88794511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, crtol, 3); 888a7e14dcfSSatish Balay 889a7e14dcfSSatish Balay if (catol != PETSC_DEFAULT) { 890a7e14dcfSSatish Balay if (catol < 0) { 8919566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Tried to set negative catol -- ignored.\n")); 892a7e14dcfSSatish Balay } else { 893a7e14dcfSSatish Balay tao->catol = PetscMax(0, catol); 8946552cf8aSJason Sarich tao->catol_changed = PETSC_TRUE; 895a7e14dcfSSatish Balay } 896a7e14dcfSSatish Balay } 897a7e14dcfSSatish Balay 898a7e14dcfSSatish Balay if (crtol != PETSC_DEFAULT) { 899a7e14dcfSSatish Balay if (crtol < 0) { 9009566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Tried to set negative crtol -- ignored.\n")); 901a7e14dcfSSatish Balay } else { 902a7e14dcfSSatish Balay tao->crtol = PetscMax(0, crtol); 9036552cf8aSJason Sarich tao->crtol_changed = PETSC_TRUE; 904a7e14dcfSSatish Balay } 905a7e14dcfSSatish Balay } 906a7e14dcfSSatish Balay PetscFunctionReturn(0); 907a7e14dcfSSatish Balay } 908a7e14dcfSSatish Balay 90958417fe7SBarry Smith /*@ 91065ba42b6SBarry Smith TaoGetConstraintTolerances - Gets constraint tolerance parameters used in Tao convergence tests 91158417fe7SBarry Smith 912*67be906fSBarry Smith Not Collective 91358417fe7SBarry Smith 91458417fe7SBarry Smith Input Parameter: 91558417fe7SBarry Smith . tao - the Tao context 91658417fe7SBarry Smith 917d8d19677SJose E. Roman Output Parameters: 918*67be906fSBarry Smith + catol - absolute constraint tolerance, constraint norm must be less than `catol` for used for gatol convergence criteria 919*67be906fSBarry Smith - crtol - relative constraint tolerance, constraint norm must be less than `crtol` for used for gatol, gttol convergence criteria 92058417fe7SBarry Smith 92158417fe7SBarry Smith Level: intermediate 92258417fe7SBarry Smith 923db781477SPatrick Sanan .seealso: `TaoGetTolerances()`, `TaoSetTolerances()`, `TaoSetConstraintTolerances()` 92458417fe7SBarry Smith 92558417fe7SBarry Smith @*/ 926d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConstraintTolerances(Tao tao, PetscReal *catol, PetscReal *crtol) 927d71ae5a4SJacob Faibussowitsch { 92858417fe7SBarry Smith PetscFunctionBegin; 92958417fe7SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 93058417fe7SBarry Smith if (catol) *catol = tao->catol; 93158417fe7SBarry Smith if (crtol) *crtol = tao->crtol; 93258417fe7SBarry Smith PetscFunctionReturn(0); 93358417fe7SBarry Smith } 93458417fe7SBarry Smith 935a7e14dcfSSatish Balay /*@ 936a7e14dcfSSatish Balay TaoSetFunctionLowerBound - Sets a bound on the solution objective value. 937a7e14dcfSSatish Balay When an approximate solution with an objective value below this number 938a7e14dcfSSatish Balay has been found, the solver will terminate. 939a7e14dcfSSatish Balay 940c3339decSBarry Smith Logically Collective 941a7e14dcfSSatish Balay 942a7e14dcfSSatish Balay Input Parameters: 943441846f8SBarry Smith + tao - the Tao solver context 944a7e14dcfSSatish Balay - fmin - the tolerance 945a7e14dcfSSatish Balay 946a7e14dcfSSatish Balay Options Database Keys: 947a7e14dcfSSatish Balay . -tao_fmin <fmin> - sets the minimum function value 948a7e14dcfSSatish Balay 949a7e14dcfSSatish Balay Level: intermediate 950a7e14dcfSSatish Balay 951db781477SPatrick Sanan .seealso: `TaoSetTolerances()` 952a7e14dcfSSatish Balay @*/ 953d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetFunctionLowerBound(Tao tao, PetscReal fmin) 954d71ae5a4SJacob Faibussowitsch { 955a7e14dcfSSatish Balay PetscFunctionBegin; 956441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 95794511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, fmin, 2); 958a7e14dcfSSatish Balay tao->fmin = fmin; 9596552cf8aSJason Sarich tao->fmin_changed = PETSC_TRUE; 960a7e14dcfSSatish Balay PetscFunctionReturn(0); 961a7e14dcfSSatish Balay } 962a7e14dcfSSatish Balay 963a7e14dcfSSatish Balay /*@ 9648e96d397SJason Sarich TaoGetFunctionLowerBound - Gets the bound on the solution objective value. 965a7e14dcfSSatish Balay When an approximate solution with an objective value below this number 966a7e14dcfSSatish Balay has been found, the solver will terminate. 967a7e14dcfSSatish Balay 968*67be906fSBarry Smith Not Collective 969a7e14dcfSSatish Balay 970a7e14dcfSSatish Balay Input Parameters: 971441846f8SBarry Smith . tao - the Tao solver context 972a7e14dcfSSatish Balay 973a7e14dcfSSatish Balay OutputParameters: 974a7e14dcfSSatish Balay . fmin - the minimum function value 975a7e14dcfSSatish Balay 976a7e14dcfSSatish Balay Level: intermediate 977a7e14dcfSSatish Balay 978db781477SPatrick Sanan .seealso: `TaoSetFunctionLowerBound()` 979a7e14dcfSSatish Balay @*/ 980d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetFunctionLowerBound(Tao tao, PetscReal *fmin) 981d71ae5a4SJacob Faibussowitsch { 982a7e14dcfSSatish Balay PetscFunctionBegin; 983441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 98494511df7SStefano Zampini PetscValidRealPointer(fmin, 2); 985a7e14dcfSSatish Balay *fmin = tao->fmin; 986a7e14dcfSSatish Balay PetscFunctionReturn(0); 987a7e14dcfSSatish Balay } 988a7e14dcfSSatish Balay 989a7e14dcfSSatish Balay /*@ 990a7e14dcfSSatish Balay TaoSetMaximumFunctionEvaluations - Sets a maximum number of 991a7e14dcfSSatish Balay function evaluations. 992a7e14dcfSSatish Balay 993c3339decSBarry Smith Logically Collective 994a7e14dcfSSatish Balay 995a7e14dcfSSatish Balay Input Parameters: 996441846f8SBarry Smith + tao - the Tao solver context 997a7e14dcfSSatish Balay - nfcn - the maximum number of function evaluations (>=0) 998a7e14dcfSSatish Balay 999a7e14dcfSSatish Balay Options Database Keys: 1000a7e14dcfSSatish Balay . -tao_max_funcs <nfcn> - sets the maximum number of function evaluations 1001a7e14dcfSSatish Balay 1002a7e14dcfSSatish Balay Level: intermediate 1003a7e14dcfSSatish Balay 1004db781477SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoSetMaximumIterations()` 1005a7e14dcfSSatish Balay @*/ 1006a7e14dcfSSatish Balay 1007d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao tao, PetscInt nfcn) 1008d71ae5a4SJacob Faibussowitsch { 1009a7e14dcfSSatish Balay PetscFunctionBegin; 1010441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 101194511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao, nfcn, 2); 10129371c9d4SSatish Balay if (nfcn >= 0) { 10139371c9d4SSatish Balay tao->max_funcs = PetscMax(0, nfcn); 10149371c9d4SSatish Balay } else { 10159371c9d4SSatish Balay tao->max_funcs = -1; 10169371c9d4SSatish Balay } 10176552cf8aSJason Sarich tao->max_funcs_changed = PETSC_TRUE; 1018a7e14dcfSSatish Balay PetscFunctionReturn(0); 1019a7e14dcfSSatish Balay } 1020a7e14dcfSSatish Balay 1021a7e14dcfSSatish Balay /*@ 102265ba42b6SBarry Smith TaoGetMaximumFunctionEvaluations - Gets a maximum number of 1023a7e14dcfSSatish Balay function evaluations. 1024a7e14dcfSSatish Balay 1025c3339decSBarry Smith Logically Collective 1026a7e14dcfSSatish Balay 1027a7e14dcfSSatish Balay Input Parameters: 1028441846f8SBarry Smith . tao - the Tao solver context 1029a7e14dcfSSatish Balay 1030a7e14dcfSSatish Balay Output Parameters: 1031a7e14dcfSSatish Balay . nfcn - the maximum number of function evaluations 1032a7e14dcfSSatish Balay 1033a7e14dcfSSatish Balay Level: intermediate 1034a7e14dcfSSatish Balay 1035db781477SPatrick Sanan .seealso: `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()` 1036a7e14dcfSSatish Balay @*/ 1037a7e14dcfSSatish Balay 1038d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao tao, PetscInt *nfcn) 1039d71ae5a4SJacob Faibussowitsch { 1040a7e14dcfSSatish Balay PetscFunctionBegin; 1041441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 104294511df7SStefano Zampini PetscValidIntPointer(nfcn, 2); 1043a7e14dcfSSatish Balay *nfcn = tao->max_funcs; 1044a7e14dcfSSatish Balay PetscFunctionReturn(0); 1045a7e14dcfSSatish Balay } 1046a7e14dcfSSatish Balay 1047770232b9SCe Qin /*@ 1048770232b9SCe Qin TaoGetCurrentFunctionEvaluations - Get current number of 1049770232b9SCe Qin function evaluations. 1050770232b9SCe Qin 1051770232b9SCe Qin Not Collective 1052770232b9SCe Qin 1053770232b9SCe Qin Input Parameters: 1054770232b9SCe Qin . tao - the Tao solver context 1055770232b9SCe Qin 1056770232b9SCe Qin Output Parameters: 105794511df7SStefano Zampini . nfuncs - the current number of function evaluations (maximum between gradient and function evaluations) 1058770232b9SCe Qin 1059770232b9SCe Qin Level: intermediate 1060770232b9SCe Qin 1061db781477SPatrick Sanan .seealso: `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()` 1062770232b9SCe Qin @*/ 1063770232b9SCe Qin 1064d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao tao, PetscInt *nfuncs) 1065d71ae5a4SJacob Faibussowitsch { 1066770232b9SCe Qin PetscFunctionBegin; 1067770232b9SCe Qin PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 106894511df7SStefano Zampini PetscValidIntPointer(nfuncs, 2); 1069770232b9SCe Qin *nfuncs = PetscMax(tao->nfuncs, tao->nfuncgrads); 1070770232b9SCe Qin PetscFunctionReturn(0); 1071770232b9SCe Qin } 1072770232b9SCe Qin 1073a7e14dcfSSatish Balay /*@ 1074a7e14dcfSSatish Balay TaoSetMaximumIterations - Sets a maximum number of iterates. 1075a7e14dcfSSatish Balay 1076c3339decSBarry Smith Logically Collective 1077a7e14dcfSSatish Balay 1078a7e14dcfSSatish Balay Input Parameters: 1079441846f8SBarry Smith + tao - the Tao solver context 1080a7e14dcfSSatish Balay - maxits - the maximum number of iterates (>=0) 1081a7e14dcfSSatish Balay 1082a7e14dcfSSatish Balay Options Database Keys: 1083a7e14dcfSSatish Balay . -tao_max_it <its> - sets the maximum number of iterations 1084a7e14dcfSSatish Balay 1085a7e14dcfSSatish Balay Level: intermediate 1086a7e14dcfSSatish Balay 1087db781477SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoSetMaximumFunctionEvaluations()` 1088a7e14dcfSSatish Balay @*/ 1089d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMaximumIterations(Tao tao, PetscInt maxits) 1090d71ae5a4SJacob Faibussowitsch { 1091a7e14dcfSSatish Balay PetscFunctionBegin; 1092441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 109394511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao, maxits, 2); 109447a47007SBarry Smith tao->max_it = PetscMax(0, maxits); 10956552cf8aSJason Sarich tao->max_it_changed = PETSC_TRUE; 1096a7e14dcfSSatish Balay PetscFunctionReturn(0); 1097a7e14dcfSSatish Balay } 1098a7e14dcfSSatish Balay 1099a7e14dcfSSatish Balay /*@ 110065ba42b6SBarry Smith TaoGetMaximumIterations - Gets a maximum number of iterates that will be used 1101a7e14dcfSSatish Balay 1102a7e14dcfSSatish Balay Not Collective 1103a7e14dcfSSatish Balay 1104a7e14dcfSSatish Balay Input Parameters: 1105441846f8SBarry Smith . tao - the Tao solver context 1106a7e14dcfSSatish Balay 1107a7e14dcfSSatish Balay Output Parameters: 1108a7e14dcfSSatish Balay . maxits - the maximum number of iterates 1109a7e14dcfSSatish Balay 1110a7e14dcfSSatish Balay Level: intermediate 1111a7e14dcfSSatish Balay 1112db781477SPatrick Sanan .seealso: `TaoSetMaximumIterations()`, `TaoGetMaximumFunctionEvaluations()` 1113a7e14dcfSSatish Balay @*/ 1114d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetMaximumIterations(Tao tao, PetscInt *maxits) 1115d71ae5a4SJacob Faibussowitsch { 1116a7e14dcfSSatish Balay PetscFunctionBegin; 1117441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 111894511df7SStefano Zampini PetscValidIntPointer(maxits, 2); 1119a7e14dcfSSatish Balay *maxits = tao->max_it; 1120a7e14dcfSSatish Balay PetscFunctionReturn(0); 1121a7e14dcfSSatish Balay } 1122a7e14dcfSSatish Balay 1123a7e14dcfSSatish Balay /*@ 1124a7e14dcfSSatish Balay TaoSetInitialTrustRegionRadius - Sets the initial trust region radius. 1125a7e14dcfSSatish Balay 1126*67be906fSBarry Smith Logically Collective 1127a7e14dcfSSatish Balay 1128d8d19677SJose E. Roman Input Parameters: 112965ba42b6SBarry Smith + tao - a Tao optimization solver 1130a7e14dcfSSatish Balay - radius - the trust region radius 1131a7e14dcfSSatish Balay 1132a7e14dcfSSatish Balay Level: intermediate 1133a7e14dcfSSatish Balay 1134a7e14dcfSSatish Balay Options Database Key: 1135a7e14dcfSSatish Balay . -tao_trust0 <t0> - sets initial trust region radius 1136a7e14dcfSSatish Balay 113765ba42b6SBarry Smith .seealso: `TaoGetTrustRegionRadius()`, `TaoSetTrustRegionTolerance()`, `TAONTR` 1138a7e14dcfSSatish Balay @*/ 1139d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetInitialTrustRegionRadius(Tao tao, PetscReal radius) 1140d71ae5a4SJacob Faibussowitsch { 1141a7e14dcfSSatish Balay PetscFunctionBegin; 1142441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 114394511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, radius, 2); 1144a7e14dcfSSatish Balay tao->trust0 = PetscMax(0.0, radius); 11456552cf8aSJason Sarich tao->trust0_changed = PETSC_TRUE; 1146a7e14dcfSSatish Balay PetscFunctionReturn(0); 1147a7e14dcfSSatish Balay } 1148a7e14dcfSSatish Balay 1149a7e14dcfSSatish Balay /*@ 115065ba42b6SBarry Smith TaoGetInitialTrustRegionRadius - Gets the initial trust region radius. 1151a7e14dcfSSatish Balay 1152a7e14dcfSSatish Balay Not Collective 1153a7e14dcfSSatish Balay 1154a7e14dcfSSatish Balay Input Parameter: 115565ba42b6SBarry Smith . tao - a Tao optimization solver 1156a7e14dcfSSatish Balay 1157a7e14dcfSSatish Balay Output Parameter: 1158a7e14dcfSSatish Balay . radius - the trust region radius 1159a7e14dcfSSatish Balay 1160a7e14dcfSSatish Balay Level: intermediate 1161a7e14dcfSSatish Balay 116265ba42b6SBarry Smith .seealso: `TaoSetInitialTrustRegionRadius()`, `TaoGetCurrentTrustRegionRadius()`, `TAONTR` 1163a7e14dcfSSatish Balay @*/ 1164d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetInitialTrustRegionRadius(Tao tao, PetscReal *radius) 1165d71ae5a4SJacob Faibussowitsch { 1166a7e14dcfSSatish Balay PetscFunctionBegin; 1167441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 116894511df7SStefano Zampini PetscValidRealPointer(radius, 2); 1169a7e14dcfSSatish Balay *radius = tao->trust0; 1170a7e14dcfSSatish Balay PetscFunctionReturn(0); 1171a7e14dcfSSatish Balay } 1172a7e14dcfSSatish Balay 1173a7e14dcfSSatish Balay /*@ 1174a7e14dcfSSatish Balay TaoGetCurrentTrustRegionRadius - Gets the current trust region radius. 1175a7e14dcfSSatish Balay 1176a7e14dcfSSatish Balay Not Collective 1177a7e14dcfSSatish Balay 1178a7e14dcfSSatish Balay Input Parameter: 117965ba42b6SBarry Smith . tao - a Tao optimization solver 1180a7e14dcfSSatish Balay 1181a7e14dcfSSatish Balay Output Parameter: 1182a7e14dcfSSatish Balay . radius - the trust region radius 1183a7e14dcfSSatish Balay 1184a7e14dcfSSatish Balay Level: intermediate 1185a7e14dcfSSatish Balay 118665ba42b6SBarry Smith .seealso: `TaoSetInitialTrustRegionRadius()`, `TaoGetInitialTrustRegionRadius()`, `TAONTR` 1187a7e14dcfSSatish Balay @*/ 1188d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao tao, PetscReal *radius) 1189d71ae5a4SJacob Faibussowitsch { 1190a7e14dcfSSatish Balay PetscFunctionBegin; 1191441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 119294511df7SStefano Zampini PetscValidRealPointer(radius, 2); 1193a7e14dcfSSatish Balay *radius = tao->trust; 1194a7e14dcfSSatish Balay PetscFunctionReturn(0); 1195a7e14dcfSSatish Balay } 1196a7e14dcfSSatish Balay 1197a7e14dcfSSatish Balay /*@ 1198a7e14dcfSSatish Balay TaoGetTolerances - gets the current values of tolerances 1199a7e14dcfSSatish Balay 1200a7e14dcfSSatish Balay Not Collective 1201a7e14dcfSSatish Balay 1202f899ff85SJose E. Roman Input Parameter: 1203441846f8SBarry Smith . tao - the Tao context 1204a7e14dcfSSatish Balay 1205a7e14dcfSSatish Balay Output Parameters: 1206e52336cbSBarry Smith + gatol - stop if norm of gradient is less than this 1207a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this 1208a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by a this factor 1209a7e14dcfSSatish Balay 1210a7e14dcfSSatish Balay Level: intermediate 1211a1cb98faSBarry Smith 1212a1cb98faSBarry Smith Note: 1213a1cb98faSBarry Smith NULL can be used as an argument if not all tolerances values are needed 1214a1cb98faSBarry Smith 1215a1cb98faSBarry Smith .seealso: `Tao`, `TaoSetTolerances()` 1216a7e14dcfSSatish Balay @*/ 1217d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetTolerances(Tao tao, PetscReal *gatol, PetscReal *grtol, PetscReal *gttol) 1218d71ae5a4SJacob Faibussowitsch { 1219a7e14dcfSSatish Balay PetscFunctionBegin; 1220441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1221a7e14dcfSSatish Balay if (gatol) *gatol = tao->gatol; 1222a7e14dcfSSatish Balay if (grtol) *grtol = tao->grtol; 1223a7e14dcfSSatish Balay if (gttol) *gttol = tao->gttol; 1224a7e14dcfSSatish Balay PetscFunctionReturn(0); 1225a7e14dcfSSatish Balay } 1226a7e14dcfSSatish Balay 1227a7e14dcfSSatish Balay /*@ 1228a7e14dcfSSatish Balay TaoGetKSP - Gets the linear solver used by the optimization solver. 122965ba42b6SBarry Smith Application writers should use `TaoGetKSP()` if they need direct access 123065ba42b6SBarry Smith to the PETSc `KSP` object. 1231a7e14dcfSSatish Balay 1232a7e14dcfSSatish Balay Not Collective 1233a7e14dcfSSatish Balay 1234a7e14dcfSSatish Balay Input Parameters: 123565ba42b6SBarry Smith . tao - the Tao solver 1236a7e14dcfSSatish Balay 1237a7e14dcfSSatish Balay Output Parameters: 1238a7e14dcfSSatish Balay . ksp - the KSP linear solver used in the optimization solver 1239a7e14dcfSSatish Balay 1240a7e14dcfSSatish Balay Level: intermediate 1241a7e14dcfSSatish Balay 124265ba42b6SBarry Smith .seealso: `Tao`, `KSP` 1243a7e14dcfSSatish Balay @*/ 1244d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetKSP(Tao tao, KSP *ksp) 1245d71ae5a4SJacob Faibussowitsch { 1246a7e14dcfSSatish Balay PetscFunctionBegin; 124794511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 124894511df7SStefano Zampini PetscValidPointer(ksp, 2); 1249a7e14dcfSSatish Balay *ksp = tao->ksp; 1250a7e14dcfSSatish Balay PetscFunctionReturn(0); 1251a7e14dcfSSatish Balay } 1252a7e14dcfSSatish Balay 1253025e9500SJason Sarich /*@ 1254025e9500SJason Sarich TaoGetLinearSolveIterations - Gets the total number of linear iterations 125565ba42b6SBarry Smith used by the Tao solver 1256025e9500SJason Sarich 1257025e9500SJason Sarich Not Collective 1258025e9500SJason Sarich 1259025e9500SJason Sarich Input Parameter: 126065ba42b6SBarry Smith . tao - Tao context 1261025e9500SJason Sarich 1262025e9500SJason Sarich Output Parameter: 1263025e9500SJason Sarich . lits - number of linear iterations 1264025e9500SJason Sarich 1265025e9500SJason Sarich Notes: 1266025e9500SJason Sarich This counter is reset to zero for each successive call to TaoSolve() 1267025e9500SJason Sarich 1268025e9500SJason Sarich Level: intermediate 1269025e9500SJason Sarich 127065ba42b6SBarry Smith .seealso: `Tao`, `TaoGetKSP()` 1271025e9500SJason Sarich @*/ 1272d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetLinearSolveIterations(Tao tao, PetscInt *lits) 1273d71ae5a4SJacob Faibussowitsch { 1274025e9500SJason Sarich PetscFunctionBegin; 1275025e9500SJason Sarich PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1276025e9500SJason Sarich PetscValidIntPointer(lits, 2); 12772d9aa51bSJason Sarich *lits = tao->ksp_tot_its; 1278025e9500SJason Sarich PetscFunctionReturn(0); 1279025e9500SJason Sarich } 1280025e9500SJason Sarich 1281a7e14dcfSSatish Balay /*@ 1282a7e14dcfSSatish Balay TaoGetLineSearch - Gets the line search used by the optimization solver. 128365ba42b6SBarry Smith Application writers should use `TaoGetLineSearch()` if they need direct access 1284a7e14dcfSSatish Balay to the TaoLineSearch object. 1285a7e14dcfSSatish Balay 1286a7e14dcfSSatish Balay Not Collective 1287a7e14dcfSSatish Balay 1288a7e14dcfSSatish Balay Input Parameters: 128965ba42b6SBarry Smith . tao - the Tao solver 1290a7e14dcfSSatish Balay 1291a7e14dcfSSatish Balay Output Parameters: 1292a7e14dcfSSatish Balay . ls - the line search used in the optimization solver 1293a7e14dcfSSatish Balay 1294a7e14dcfSSatish Balay Level: intermediate 1295a7e14dcfSSatish Balay 1296a7e14dcfSSatish Balay @*/ 1297d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetLineSearch(Tao tao, TaoLineSearch *ls) 1298d71ae5a4SJacob Faibussowitsch { 1299a7e14dcfSSatish Balay PetscFunctionBegin; 130094511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 130194511df7SStefano Zampini PetscValidPointer(ls, 2); 1302a7e14dcfSSatish Balay *ls = tao->linesearch; 1303a7e14dcfSSatish Balay PetscFunctionReturn(0); 1304a7e14dcfSSatish Balay } 1305a7e14dcfSSatish Balay 1306a7e14dcfSSatish Balay /*@ 1307a7e14dcfSSatish Balay TaoAddLineSearchCounts - Adds the number of function evaluations spent 1308a7e14dcfSSatish Balay in the line search to the running total. 1309a7e14dcfSSatish Balay 1310a7e14dcfSSatish Balay Input Parameters: 131165ba42b6SBarry Smith + tao - the Tao solver 1312a7e14dcfSSatish Balay - ls - the line search used in the optimization solver 1313a7e14dcfSSatish Balay 1314a7e14dcfSSatish Balay Level: developer 1315a7e14dcfSSatish Balay 131665ba42b6SBarry Smith .seealso: `TaoGetLineSearch()`, `TaoLineSearchApply()` 1317a7e14dcfSSatish Balay @*/ 1318d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoAddLineSearchCounts(Tao tao) 1319d71ae5a4SJacob Faibussowitsch { 1320a7e14dcfSSatish Balay PetscBool flg; 1321a7e14dcfSSatish Balay PetscInt nfeval, ngeval, nfgeval; 132247a47007SBarry Smith 1323a7e14dcfSSatish Balay PetscFunctionBegin; 1324441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1325a7e14dcfSSatish Balay if (tao->linesearch) { 13269566063dSJacob Faibussowitsch PetscCall(TaoLineSearchIsUsingTaoRoutines(tao->linesearch, &flg)); 132794ae4db5SBarry Smith if (!flg) { 13289566063dSJacob Faibussowitsch PetscCall(TaoLineSearchGetNumberFunctionEvaluations(tao->linesearch, &nfeval, &ngeval, &nfgeval)); 1329a7e14dcfSSatish Balay tao->nfuncs += nfeval; 1330a7e14dcfSSatish Balay tao->ngrads += ngeval; 1331a7e14dcfSSatish Balay tao->nfuncgrads += nfgeval; 1332a7e14dcfSSatish Balay } 1333a7e14dcfSSatish Balay } 1334a7e14dcfSSatish Balay PetscFunctionReturn(0); 1335a7e14dcfSSatish Balay } 1336a7e14dcfSSatish Balay 1337a7e14dcfSSatish Balay /*@ 133865ba42b6SBarry Smith TaoGetSolution - Returns the vector with the current Tao solution 1339a7e14dcfSSatish Balay 1340a7e14dcfSSatish Balay Not Collective 1341a7e14dcfSSatish Balay 1342a7e14dcfSSatish Balay Input Parameter: 1343441846f8SBarry Smith . tao - the Tao context 1344a7e14dcfSSatish Balay 1345a7e14dcfSSatish Balay Output Parameter: 1346a7e14dcfSSatish Balay . X - the current solution 1347a7e14dcfSSatish Balay 1348a7e14dcfSSatish Balay Level: intermediate 1349a7e14dcfSSatish Balay 135065ba42b6SBarry Smith Note: 135165ba42b6SBarry Smith The returned vector will be the same object that was passed into `TaoSetSolution()` 135265ba42b6SBarry Smith 135365ba42b6SBarry Smith .seealso: `Tao`, `TaoSetSolution()`, `TaoSolve()` 1354a7e14dcfSSatish Balay @*/ 1355d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetSolution(Tao tao, Vec *X) 1356d71ae5a4SJacob Faibussowitsch { 1357a7e14dcfSSatish Balay PetscFunctionBegin; 1358441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 135994511df7SStefano Zampini PetscValidPointer(X, 2); 1360a7e14dcfSSatish Balay *X = tao->solution; 1361a7e14dcfSSatish Balay PetscFunctionReturn(0); 1362a7e14dcfSSatish Balay } 1363a7e14dcfSSatish Balay 1364a7e14dcfSSatish Balay /*@ 136565ba42b6SBarry Smith TaoResetStatistics - Initialize the statistics used by Tao for all of the solvers. 1366a7e14dcfSSatish Balay These statistics include the iteration number, residual norms, and convergence status. 1367a7e14dcfSSatish Balay This routine gets called before solving each optimization problem. 1368a7e14dcfSSatish Balay 1369c3339decSBarry Smith Collective 1370a7e14dcfSSatish Balay 1371a7e14dcfSSatish Balay Input Parameters: 1372441846f8SBarry Smith . solver - the Tao context 1373a7e14dcfSSatish Balay 1374a7e14dcfSSatish Balay Level: developer 1375a7e14dcfSSatish Balay 137665ba42b6SBarry Smith .seealso: `Tao`, `TaoCreate()`, `TaoSolve()` 1377a7e14dcfSSatish Balay @*/ 1378d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoResetStatistics(Tao tao) 1379d71ae5a4SJacob Faibussowitsch { 1380a7e14dcfSSatish Balay PetscFunctionBegin; 1381441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1382a7e14dcfSSatish Balay tao->niter = 0; 1383a7e14dcfSSatish Balay tao->nfuncs = 0; 1384a7e14dcfSSatish Balay tao->nfuncgrads = 0; 1385a7e14dcfSSatish Balay tao->ngrads = 0; 1386a7e14dcfSSatish Balay tao->nhess = 0; 1387a7e14dcfSSatish Balay tao->njac = 0; 1388a7e14dcfSSatish Balay tao->nconstraints = 0; 1389a7e14dcfSSatish Balay tao->ksp_its = 0; 1390ae93cb3cSJason Sarich tao->ksp_tot_its = 0; 1391a7e14dcfSSatish Balay tao->reason = TAO_CONTINUE_ITERATING; 1392a7e14dcfSSatish Balay tao->residual = 0.0; 1393a7e14dcfSSatish Balay tao->cnorm = 0.0; 1394a7e14dcfSSatish Balay tao->step = 0.0; 1395a7e14dcfSSatish Balay tao->lsflag = PETSC_FALSE; 1396a7e14dcfSSatish Balay if (tao->hist_reset) tao->hist_len = 0; 1397a7e14dcfSSatish Balay PetscFunctionReturn(0); 1398a7e14dcfSSatish Balay } 1399a7e14dcfSSatish Balay 1400a7e14dcfSSatish Balay /*@C 1401e1e80dc8SAlp Dener TaoSetUpdate - Sets the general-purpose update function called 140265ba42b6SBarry Smith at the beginning of every iteration of the optimization algorithm. Specifically 1403e1e80dc8SAlp Dener it is called at the top of every iteration, after the new solution and the gradient 1404e1e80dc8SAlp Dener is determined, but before the Hessian is computed (if applicable). 1405e1e80dc8SAlp Dener 1406c3339decSBarry Smith Logically Collective 1407e1e80dc8SAlp Dener 1408e1e80dc8SAlp Dener Input Parameters: 1409a2b725a8SWilliam Gropp + tao - The tao solver context 1410a2b725a8SWilliam Gropp - func - The function 1411e1e80dc8SAlp Dener 1412e1e80dc8SAlp Dener Calling sequence of func: 1413a2b725a8SWilliam Gropp $ func (Tao tao, PetscInt step); 1414e1e80dc8SAlp Dener 1415e1e80dc8SAlp Dener . step - The current step of the iteration 1416e1e80dc8SAlp Dener 1417e1e80dc8SAlp Dener Level: advanced 1418e1e80dc8SAlp Dener 1419a1cb98faSBarry Smith .seealso: `Tao`, `TaoSolve()` 1420e1e80dc8SAlp Dener @*/ 1421d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetUpdate(Tao tao, PetscErrorCode (*func)(Tao, PetscInt, void *), void *ctx) 1422d71ae5a4SJacob Faibussowitsch { 1423e1e80dc8SAlp Dener PetscFunctionBegin; 1424e1e80dc8SAlp Dener PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1425e1e80dc8SAlp Dener tao->ops->update = func; 1426e1e80dc8SAlp Dener tao->user_update = ctx; 1427e1e80dc8SAlp Dener PetscFunctionReturn(0); 1428e1e80dc8SAlp Dener } 1429e1e80dc8SAlp Dener 1430e1e80dc8SAlp Dener /*@C 1431a7e14dcfSSatish Balay TaoSetConvergenceTest - Sets the function that is to be used to test 1432a7e14dcfSSatish Balay for convergence o fthe iterative minimization solution. The new convergence 143365ba42b6SBarry Smith testing routine will replace Tao's default convergence test. 1434a7e14dcfSSatish Balay 1435c3339decSBarry Smith Logically Collective 1436a7e14dcfSSatish Balay 1437a7e14dcfSSatish Balay Input Parameters: 1438441846f8SBarry Smith + tao - the Tao object 1439a7e14dcfSSatish Balay . conv - the routine to test for convergence 1440a7e14dcfSSatish Balay - ctx - [optional] context for private data for the convergence routine 144147a47007SBarry Smith (may be NULL) 1442a7e14dcfSSatish Balay 1443a7e14dcfSSatish Balay Calling sequence of conv: 1444441846f8SBarry Smith $ PetscErrorCode conv(Tao tao, void *ctx) 1445a7e14dcfSSatish Balay 1446441846f8SBarry Smith + tao - the Tao object 1447a7e14dcfSSatish Balay - ctx - [optional] convergence context 1448a7e14dcfSSatish Balay 144965ba42b6SBarry Smith Note: 145065ba42b6SBarry Smith The new convergence testing routine should call `TaoSetConvergedReason()`. 1451a7e14dcfSSatish Balay 1452a7e14dcfSSatish Balay Level: advanced 1453a7e14dcfSSatish Balay 145465ba42b6SBarry Smith .seealso: `Tao`, `TaoSolve()`, `TaoSetConvergedReason()`, `TaoGetSolutionStatus()`, `TaoGetTolerances()`, `TaoSetMonitor` 1455a7e14dcfSSatish Balay 1456a7e14dcfSSatish Balay @*/ 1457d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergenceTest(Tao tao, PetscErrorCode (*conv)(Tao, void *), void *ctx) 1458d71ae5a4SJacob Faibussowitsch { 1459a7e14dcfSSatish Balay PetscFunctionBegin; 1460441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 146194511df7SStefano Zampini tao->ops->convergencetest = conv; 146294511df7SStefano Zampini tao->cnvP = ctx; 1463a7e14dcfSSatish Balay PetscFunctionReturn(0); 1464a7e14dcfSSatish Balay } 1465a7e14dcfSSatish Balay 1466a7e14dcfSSatish Balay /*@C 146765ba42b6SBarry Smith TaoSetMonitor - Sets an additional function that is to be used at every 1468a7e14dcfSSatish Balay iteration of the solver to display the iteration's 1469a7e14dcfSSatish Balay progress. 1470a7e14dcfSSatish Balay 1471c3339decSBarry Smith Logically Collective 1472a7e14dcfSSatish Balay 1473a7e14dcfSSatish Balay Input Parameters: 1474441846f8SBarry Smith + tao - the Tao solver context 1475a7e14dcfSSatish Balay . mymonitor - monitoring routine 1476a7e14dcfSSatish Balay - mctx - [optional] user-defined context for private data for the 147747a47007SBarry Smith monitor routine (may be NULL) 1478a7e14dcfSSatish Balay 1479a7e14dcfSSatish Balay Calling sequence of mymonitor: 1480147403d9SBarry Smith .vb 1481147403d9SBarry Smith PetscErrorCode mymonitor(Tao tao,void *mctx) 1482147403d9SBarry Smith .ve 1483a7e14dcfSSatish Balay 1484441846f8SBarry Smith + tao - the Tao solver context 1485a7e14dcfSSatish Balay - mctx - [optional] monitoring context 1486a7e14dcfSSatish Balay 1487a7e14dcfSSatish Balay Options Database Keys: 148865ba42b6SBarry Smith + -tao_monitor - sets the default monitor `TaoMonitorDefault()` 1489a7e14dcfSSatish Balay . -tao_smonitor - sets short monitor 1490a7e14dcfSSatish Balay . -tao_cmonitor - same as smonitor plus constraint norm 1491a7e14dcfSSatish Balay . -tao_view_solution - view solution at each iteration 1492a7e14dcfSSatish Balay . -tao_view_gradient - view gradient at each iteration 14934a48860cSAlp Dener . -tao_view_ls_residual - view least-squares residual vector at each iteration 1494a7e14dcfSSatish Balay - -tao_cancelmonitors - cancels all monitors that have been hardwired into a code by calls to TaoSetMonitor(), but does not cancel those set via the options database. 1495a7e14dcfSSatish Balay 149687497f52SBarry Smith Notes: 1497a7e14dcfSSatish Balay Several different monitoring routines may be set by calling 149865ba42b6SBarry Smith `TaoSetMonitor()` multiple times; all will be called in the 1499a7e14dcfSSatish Balay order in which they were set. 1500a7e14dcfSSatish Balay 150165ba42b6SBarry Smith Fortran Note: 150295452b02SPatrick Sanan Only one monitor function may be set 1503a7e14dcfSSatish Balay 1504a7e14dcfSSatish Balay Level: intermediate 1505a7e14dcfSSatish Balay 150665ba42b6SBarry Smith .seealso: `Tao`, `TaoSolve()`, `TaoMonitorDefault()`, `TaoCancelMonitors()`, `TaoSetDestroyRoutine()`, `TaoView()` 1507a7e14dcfSSatish Balay @*/ 1508d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMonitor(Tao tao, PetscErrorCode (*func)(Tao, void *), void *ctx, PetscErrorCode (*dest)(void **)) 1509d71ae5a4SJacob Faibussowitsch { 151008d19d1fSJason Sarich PetscInt i; 15118732526dSAlp Dener PetscBool identical; 151208d19d1fSJason Sarich 1513a7e14dcfSSatish Balay PetscFunctionBegin; 1514441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 15153c859ba3SBarry Smith PetscCheck(tao->numbermonitors < MAXTAOMONITORS, PetscObjectComm((PetscObject)tao), PETSC_ERR_SUP, "Cannot attach another monitor -- max=%d", MAXTAOMONITORS); 151608d19d1fSJason Sarich 151708d19d1fSJason Sarich for (i = 0; i < tao->numbermonitors; i++) { 15189566063dSJacob Faibussowitsch PetscCall(PetscMonitorCompare((PetscErrorCode(*)(void))func, ctx, dest, (PetscErrorCode(*)(void))tao->monitor[i], tao->monitorcontext[i], tao->monitordestroy[i], &identical)); 15198732526dSAlp Dener if (identical) PetscFunctionReturn(0); 152008d19d1fSJason Sarich } 1521a7e14dcfSSatish Balay tao->monitor[tao->numbermonitors] = func; 15228732526dSAlp Dener tao->monitorcontext[tao->numbermonitors] = (void *)ctx; 1523a7e14dcfSSatish Balay tao->monitordestroy[tao->numbermonitors] = dest; 1524a7e14dcfSSatish Balay ++tao->numbermonitors; 1525a7e14dcfSSatish Balay PetscFunctionReturn(0); 1526a7e14dcfSSatish Balay } 1527a7e14dcfSSatish Balay 1528a7e14dcfSSatish Balay /*@ 1529441846f8SBarry Smith TaoCancelMonitors - Clears all the monitor functions for a Tao object. 1530a7e14dcfSSatish Balay 1531c3339decSBarry Smith Logically Collective 1532a7e14dcfSSatish Balay 1533a7e14dcfSSatish Balay Input Parameters: 1534441846f8SBarry Smith . tao - the Tao solver context 1535a7e14dcfSSatish Balay 15363c7db156SBarry Smith Options Database Key: 1537a7e14dcfSSatish Balay . -tao_cancelmonitors - cancels all monitors that have been hardwired 153865ba42b6SBarry Smith into a code by calls to `TaoSetMonitor()`, but does not cancel those 1539a7e14dcfSSatish Balay set via the options database 1540a7e14dcfSSatish Balay 1541a7e14dcfSSatish Balay Notes: 1542441846f8SBarry Smith There is no way to clear one specific monitor from a Tao object. 1543a7e14dcfSSatish Balay 1544a7e14dcfSSatish Balay Level: advanced 1545a7e14dcfSSatish Balay 154665ba42b6SBarry Smith .seealso: `Tao`, `TaoMonitorDefault()`, `TaoSetMonitor()` 1547a7e14dcfSSatish Balay @*/ 1548d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoCancelMonitors(Tao tao) 1549d71ae5a4SJacob Faibussowitsch { 1550a7e14dcfSSatish Balay PetscInt i; 155147a47007SBarry Smith 1552a7e14dcfSSatish Balay PetscFunctionBegin; 1553441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1554a7e14dcfSSatish Balay for (i = 0; i < tao->numbermonitors; i++) { 155548a46eb9SPierre Jolivet if (tao->monitordestroy[i]) PetscCall((*tao->monitordestroy[i])(&tao->monitorcontext[i])); 1556a7e14dcfSSatish Balay } 1557a7e14dcfSSatish Balay tao->numbermonitors = 0; 1558a7e14dcfSSatish Balay PetscFunctionReturn(0); 1559a7e14dcfSSatish Balay } 1560a7e14dcfSSatish Balay 15617fab98ebSJason Sarich /*@ 156298ea980cSBarry Smith TaoMonitorDefault - Default routine for monitoring progress of the 1563441846f8SBarry Smith Tao solvers (default). This monitor prints the function value and gradient 1564a7e14dcfSSatish Balay norm at each iteration. It can be turned on from the command line using the 1565a7e14dcfSSatish Balay -tao_monitor option 1566a7e14dcfSSatish Balay 1567c3339decSBarry Smith Collective 1568a7e14dcfSSatish Balay 1569a7e14dcfSSatish Balay Input Parameters: 1570441846f8SBarry Smith + tao - the Tao context 157165ba42b6SBarry Smith - ctx - `PetscViewer` context or NULL 1572a7e14dcfSSatish Balay 1573a7e14dcfSSatish Balay Options Database Keys: 1574147403d9SBarry Smith . -tao_monitor - turn on default monitoring 1575a7e14dcfSSatish Balay 1576a7e14dcfSSatish Balay Level: advanced 1577a7e14dcfSSatish Balay 1578db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1579a7e14dcfSSatish Balay @*/ 1580d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDefault(Tao tao, void *ctx) 1581d71ae5a4SJacob Faibussowitsch { 158263b15415SAlp Dener PetscInt its, tabs; 1583a7e14dcfSSatish Balay PetscReal fct, gnorm; 15848163d661SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 158547a47007SBarry Smith 1586a7e14dcfSSatish Balay PetscFunctionBegin; 158794511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 15888163d661SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 1589a7e14dcfSSatish Balay its = tao->niter; 1590a7e14dcfSSatish Balay fct = tao->fc; 1591a7e14dcfSSatish Balay gnorm = tao->residual; 15929566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 15939566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 1594494bef23SAlp Dener if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) { 15959566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Iteration information for %s solve.\n", ((PetscObject)tao)->prefix)); 1596494bef23SAlp Dener tao->header_printed = PETSC_TRUE; 159763b15415SAlp Dener } 159863a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " TAO,", its)); 15999566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Function value: %g,", (double)fct)); 160008d19d1fSJason Sarich if (gnorm >= PETSC_INFINITY) { 16019566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: Inf \n")); 160208d19d1fSJason Sarich } else { 16039566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g \n", (double)gnorm)); 160408d19d1fSJason Sarich } 16059566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 1606a7e14dcfSSatish Balay PetscFunctionReturn(0); 1607a7e14dcfSSatish Balay } 1608a7e14dcfSSatish Balay 16097fab98ebSJason Sarich /*@ 16108d5ead36SAlp Dener TaoDefaultGMonitor - Default routine for monitoring progress of the 16118d5ead36SAlp Dener Tao solvers (default) with extra detail on the globalization method. 16128d5ead36SAlp Dener This monitor prints the function value and gradient norm at each 16138d5ead36SAlp Dener iteration, as well as the step size and trust radius. Note that the 16148d5ead36SAlp Dener step size and trust radius may be the same for some algorithms. 16158d5ead36SAlp Dener It can be turned on from the command line using the 16168d5ead36SAlp Dener -tao_gmonitor option 16178d5ead36SAlp Dener 1618c3339decSBarry Smith Collective 16198d5ead36SAlp Dener 16208d5ead36SAlp Dener Input Parameters: 16218d5ead36SAlp Dener + tao - the Tao context 162265ba42b6SBarry Smith - ctx - `PetscViewer` context or NULL 16238d5ead36SAlp Dener 16248d5ead36SAlp Dener Options Database Keys: 1625147403d9SBarry Smith . -tao_gmonitor - turn on monitoring with globalization information 16268d5ead36SAlp Dener 16278d5ead36SAlp Dener Level: advanced 16288d5ead36SAlp Dener 1629db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()` 16308d5ead36SAlp Dener @*/ 1631d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultGMonitor(Tao tao, void *ctx) 1632d71ae5a4SJacob Faibussowitsch { 16338d5ead36SAlp Dener PetscInt its, tabs; 16348d5ead36SAlp Dener PetscReal fct, gnorm, stp, tr; 16358d5ead36SAlp Dener PetscViewer viewer = (PetscViewer)ctx; 16368d5ead36SAlp Dener 16378d5ead36SAlp Dener PetscFunctionBegin; 163894511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 16398d5ead36SAlp Dener PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 16408d5ead36SAlp Dener its = tao->niter; 16418d5ead36SAlp Dener fct = tao->fc; 16428d5ead36SAlp Dener gnorm = tao->residual; 16438d5ead36SAlp Dener stp = tao->step; 16448d5ead36SAlp Dener tr = tao->trust; 16459566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 16469566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 1647494bef23SAlp Dener if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) { 16489566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Iteration information for %s solve.\n", ((PetscObject)tao)->prefix)); 1649494bef23SAlp Dener tao->header_printed = PETSC_TRUE; 16508d5ead36SAlp Dener } 165163a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " TAO,", its)); 16529566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Function value: %g,", (double)fct)); 16538d5ead36SAlp Dener if (gnorm >= PETSC_INFINITY) { 16549566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: Inf,")); 16558d5ead36SAlp Dener } else { 16569566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g,", (double)gnorm)); 16578d5ead36SAlp Dener } 16589566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Step: %g, Trust: %g\n", (double)stp, (double)tr)); 16599566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 16608d5ead36SAlp Dener PetscFunctionReturn(0); 16618d5ead36SAlp Dener } 16628d5ead36SAlp Dener 16638d5ead36SAlp Dener /*@ 1664a7e14dcfSSatish Balay TaoDefaultSMonitor - Default routine for monitoring progress of the 166565ba42b6SBarry Smith solver. Same as `TaoMonitorDefault()` except 1666a7e14dcfSSatish Balay it prints fewer digits of the residual as the residual gets smaller. 1667a7e14dcfSSatish Balay This is because the later digits are meaningless and are often 1668a7e14dcfSSatish Balay different on different machines; by using this routine different 1669a7e14dcfSSatish Balay machines will usually generate the same output. It can be turned on 1670a7e14dcfSSatish Balay by using the -tao_smonitor option 1671a7e14dcfSSatish Balay 1672c3339decSBarry Smith Collective 1673a7e14dcfSSatish Balay 1674a7e14dcfSSatish Balay Input Parameters: 1675441846f8SBarry Smith + tao - the Tao context 16764d4332d5SBarry Smith - ctx - PetscViewer context of type ASCII 1677a7e14dcfSSatish Balay 1678a7e14dcfSSatish Balay Options Database Keys: 1679147403d9SBarry Smith . -tao_smonitor - turn on default short monitoring 1680a7e14dcfSSatish Balay 1681a7e14dcfSSatish Balay Level: advanced 1682a7e14dcfSSatish Balay 1683db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()` 1684a7e14dcfSSatish Balay @*/ 1685d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultSMonitor(Tao tao, void *ctx) 1686d71ae5a4SJacob Faibussowitsch { 16872c9b8b83SAlp Dener PetscInt its, tabs; 1688a7e14dcfSSatish Balay PetscReal fct, gnorm; 16894d4332d5SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 169047a47007SBarry Smith 1691a7e14dcfSSatish Balay PetscFunctionBegin; 169294511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 16934d4332d5SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 1694a7e14dcfSSatish Balay its = tao->niter; 1695a7e14dcfSSatish Balay fct = tao->fc; 1696a7e14dcfSSatish Balay gnorm = tao->residual; 16979566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 16989566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 169963a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "iter = %3" PetscInt_FMT ",", its)); 17009566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Function value %g,", (double)fct)); 1701d393f493SBarry Smith if (gnorm >= PETSC_INFINITY) { 17029566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: Inf \n")); 170308d19d1fSJason Sarich } else if (gnorm > 1.e-6) { 17049566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g \n", (double)gnorm)); 1705a7e14dcfSSatish Balay } else if (gnorm > 1.e-11) { 17069566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: < 1.0e-6 \n")); 1707a7e14dcfSSatish Balay } else { 17089566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: < 1.0e-11 \n")); 1709a7e14dcfSSatish Balay } 17109566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 1711a7e14dcfSSatish Balay PetscFunctionReturn(0); 1712a7e14dcfSSatish Balay } 1713a7e14dcfSSatish Balay 17147fab98ebSJason Sarich /*@ 171565ba42b6SBarry Smith TaoDefaultCMonitor - same as `TaoMonitorDefault()` except 1716a7e14dcfSSatish Balay it prints the norm of the constraints function. It can be turned on 1717a7e14dcfSSatish Balay from the command line using the -tao_cmonitor option 1718a7e14dcfSSatish Balay 1719c3339decSBarry Smith Collective 1720a7e14dcfSSatish Balay 1721a7e14dcfSSatish Balay Input Parameters: 1722441846f8SBarry Smith + tao - the Tao context 172365ba42b6SBarry Smith - ctx - `PetscViewer` context or NULL 1724a7e14dcfSSatish Balay 1725a7e14dcfSSatish Balay Options Database Keys: 1726147403d9SBarry Smith . -tao_cmonitor - monitor the constraints 1727a7e14dcfSSatish Balay 1728a7e14dcfSSatish Balay Level: advanced 1729a7e14dcfSSatish Balay 1730db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()` 1731a7e14dcfSSatish Balay @*/ 1732d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultCMonitor(Tao tao, void *ctx) 1733d71ae5a4SJacob Faibussowitsch { 17342c9b8b83SAlp Dener PetscInt its, tabs; 1735a7e14dcfSSatish Balay PetscReal fct, gnorm; 1736d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 1737a7e14dcfSSatish Balay 1738a7e14dcfSSatish Balay PetscFunctionBegin; 173994511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1740d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 1741a7e14dcfSSatish Balay its = tao->niter; 1742a7e14dcfSSatish Balay fct = tao->fc; 1743a7e14dcfSSatish Balay gnorm = tao->residual; 17449566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 17459566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 174663a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "iter = %" PetscInt_FMT ",", its)); 17479566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Function value: %g,", (double)fct)); 17489566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g ", (double)gnorm)); 17499566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Constraint: %g \n", (double)tao->cnorm)); 17509566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 1751a7e14dcfSSatish Balay PetscFunctionReturn(0); 1752a7e14dcfSSatish Balay } 1753a7e14dcfSSatish Balay 1754a7e14dcfSSatish Balay /*@C 1755a7e14dcfSSatish Balay TaoSolutionMonitor - Views the solution at each iteration 1756a7e14dcfSSatish Balay It can be turned on from the command line using the 1757a7e14dcfSSatish Balay -tao_view_solution option 1758a7e14dcfSSatish Balay 1759c3339decSBarry Smith Collective 1760a7e14dcfSSatish Balay 1761a7e14dcfSSatish Balay Input Parameters: 1762441846f8SBarry Smith + tao - the Tao context 176365ba42b6SBarry Smith - ctx - `PetscViewer` context or NULL 1764a7e14dcfSSatish Balay 1765a7e14dcfSSatish Balay Options Database Keys: 1766147403d9SBarry Smith . -tao_view_solution - view the solution 1767a7e14dcfSSatish Balay 1768a7e14dcfSSatish Balay Level: advanced 1769a7e14dcfSSatish Balay 1770db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1771a7e14dcfSSatish Balay @*/ 1772d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSolutionMonitor(Tao tao, void *ctx) 1773d71ae5a4SJacob Faibussowitsch { 1774feb237baSPierre Jolivet PetscViewer viewer = (PetscViewer)ctx; 177547a47007SBarry Smith 1776a7e14dcfSSatish Balay PetscFunctionBegin; 177794511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1778d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 17799566063dSJacob Faibussowitsch PetscCall(VecView(tao->solution, viewer)); 1780a7e14dcfSSatish Balay PetscFunctionReturn(0); 1781a7e14dcfSSatish Balay } 1782a7e14dcfSSatish Balay 1783a7e14dcfSSatish Balay /*@C 1784a7e14dcfSSatish Balay TaoGradientMonitor - Views the gradient at each iteration 1785a7e14dcfSSatish Balay It can be turned on from the command line using the 1786a7e14dcfSSatish Balay -tao_view_gradient option 1787a7e14dcfSSatish Balay 1788c3339decSBarry Smith Collective 1789a7e14dcfSSatish Balay 1790a7e14dcfSSatish Balay Input Parameters: 1791441846f8SBarry Smith + tao - the Tao context 179265ba42b6SBarry Smith - ctx - `PetscViewer` context or NULL 1793a7e14dcfSSatish Balay 1794a7e14dcfSSatish Balay Options Database Keys: 1795147403d9SBarry Smith . -tao_view_gradient - view the gradient at each iteration 1796a7e14dcfSSatish Balay 1797a7e14dcfSSatish Balay Level: advanced 1798a7e14dcfSSatish Balay 1799db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1800a7e14dcfSSatish Balay @*/ 1801d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGradientMonitor(Tao tao, void *ctx) 1802d71ae5a4SJacob Faibussowitsch { 1803d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 180447a47007SBarry Smith 1805a7e14dcfSSatish Balay PetscFunctionBegin; 180694511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1807d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 18089566063dSJacob Faibussowitsch PetscCall(VecView(tao->gradient, viewer)); 1809a7e14dcfSSatish Balay PetscFunctionReturn(0); 1810a7e14dcfSSatish Balay } 1811a7e14dcfSSatish Balay 1812a7e14dcfSSatish Balay /*@C 1813147403d9SBarry Smith TaoStepDirectionMonitor - Views the step-direction at each iteration 1814a7e14dcfSSatish Balay 1815c3339decSBarry Smith Collective 1816a7e14dcfSSatish Balay 1817a7e14dcfSSatish Balay Input Parameters: 1818441846f8SBarry Smith + tao - the Tao context 181965ba42b6SBarry Smith - ctx - `PetscViewer` context or NULL 1820a7e14dcfSSatish Balay 1821a7e14dcfSSatish Balay Options Database Keys: 1822147403d9SBarry Smith . -tao_view_gradient - view the gradient at each iteration 1823a7e14dcfSSatish Balay 1824a7e14dcfSSatish Balay Level: advanced 1825a7e14dcfSSatish Balay 1826db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1827a7e14dcfSSatish Balay @*/ 1828d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoStepDirectionMonitor(Tao tao, void *ctx) 1829d71ae5a4SJacob Faibussowitsch { 1830d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 1831d393f493SBarry Smith 1832a7e14dcfSSatish Balay PetscFunctionBegin; 183394511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1834d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 18359566063dSJacob Faibussowitsch PetscCall(VecView(tao->stepdirection, viewer)); 1836a7e14dcfSSatish Balay PetscFunctionReturn(0); 1837a7e14dcfSSatish Balay } 1838a7e14dcfSSatish Balay 1839a7e14dcfSSatish Balay /*@C 1840a7e14dcfSSatish Balay TaoDrawSolutionMonitor - Plots the solution at each iteration 1841a7e14dcfSSatish Balay It can be turned on from the command line using the 1842a7e14dcfSSatish Balay -tao_draw_solution option 1843a7e14dcfSSatish Balay 1844c3339decSBarry Smith Collective 1845a7e14dcfSSatish Balay 1846a7e14dcfSSatish Balay Input Parameters: 1847441846f8SBarry Smith + tao - the Tao context 184865ba42b6SBarry Smith - ctx - `TaoMonitorDraw` context 1849a7e14dcfSSatish Balay 1850a7e14dcfSSatish Balay Options Database Keys: 1851147403d9SBarry Smith . -tao_draw_solution - draw the solution at each iteration 1852a7e14dcfSSatish Balay 1853a7e14dcfSSatish Balay Level: advanced 1854a7e14dcfSSatish Balay 185565ba42b6SBarry Smith .seealso: `TaoSolutionMonitor()`, `TaoSetMonitor()`, `TaoDrawGradientMonitor`, `TaoMonitorDraw` 1856a7e14dcfSSatish Balay @*/ 1857d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawSolutionMonitor(Tao tao, void *ctx) 1858d71ae5a4SJacob Faibussowitsch { 1859e882e171SHong Zhang TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx; 186047a47007SBarry Smith 1861a7e14dcfSSatish Balay PetscFunctionBegin; 186294511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1863e882e171SHong Zhang if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0); 18649566063dSJacob Faibussowitsch PetscCall(VecView(tao->solution, ictx->viewer)); 1865a7e14dcfSSatish Balay PetscFunctionReturn(0); 1866a7e14dcfSSatish Balay } 1867a7e14dcfSSatish Balay 1868a7e14dcfSSatish Balay /*@C 1869a7e14dcfSSatish Balay TaoDrawGradientMonitor - Plots the gradient at each iteration 1870a7e14dcfSSatish Balay It can be turned on from the command line using the 1871a7e14dcfSSatish Balay -tao_draw_gradient option 1872a7e14dcfSSatish Balay 1873c3339decSBarry Smith Collective 1874a7e14dcfSSatish Balay 1875a7e14dcfSSatish Balay Input Parameters: 1876441846f8SBarry Smith + tao - the Tao context 187765ba42b6SBarry Smith - ctx - `PetscViewer` context 1878a7e14dcfSSatish Balay 1879a7e14dcfSSatish Balay Options Database Keys: 1880147403d9SBarry Smith . -tao_draw_gradient - draw the gradient at each iteration 1881a7e14dcfSSatish Balay 1882a7e14dcfSSatish Balay Level: advanced 1883a7e14dcfSSatish Balay 1884db781477SPatrick Sanan .seealso: `TaoGradientMonitor()`, `TaoSetMonitor()`, `TaoDrawSolutionMonitor` 1885a7e14dcfSSatish Balay @*/ 1886d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawGradientMonitor(Tao tao, void *ctx) 1887d71ae5a4SJacob Faibussowitsch { 1888e882e171SHong Zhang TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx; 188947a47007SBarry Smith 1890a7e14dcfSSatish Balay PetscFunctionBegin; 189194511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1892e882e171SHong Zhang if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0); 18939566063dSJacob Faibussowitsch PetscCall(VecView(tao->gradient, ictx->viewer)); 1894a7e14dcfSSatish Balay PetscFunctionReturn(0); 1895a7e14dcfSSatish Balay } 1896a7e14dcfSSatish Balay 1897a7e14dcfSSatish Balay /*@C 1898a7e14dcfSSatish Balay TaoDrawStepMonitor - Plots the step direction at each iteration 1899a7e14dcfSSatish Balay 1900c3339decSBarry Smith Collective 1901a7e14dcfSSatish Balay 1902a7e14dcfSSatish Balay Input Parameters: 1903441846f8SBarry Smith + tao - the Tao context 1904f55353a2SBarry Smith - ctx - PetscViewer context 1905a7e14dcfSSatish Balay 1906a7e14dcfSSatish Balay Options Database Keys: 1907147403d9SBarry Smith . -tao_draw_step - draw the step direction at each iteration 1908a7e14dcfSSatish Balay 1909a7e14dcfSSatish Balay Level: advanced 1910a7e14dcfSSatish Balay 1911db781477SPatrick Sanan .seealso: `TaoSetMonitor()`, `TaoDrawSolutionMonitor` 1912a7e14dcfSSatish Balay @*/ 1913d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawStepMonitor(Tao tao, void *ctx) 1914d71ae5a4SJacob Faibussowitsch { 191594511df7SStefano Zampini PetscViewer viewer = (PetscViewer)ctx; 191647a47007SBarry Smith 1917a7e14dcfSSatish Balay PetscFunctionBegin; 191894511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 191994511df7SStefano Zampini PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 19209566063dSJacob Faibussowitsch PetscCall(VecView(tao->stepdirection, viewer)); 1921a7e14dcfSSatish Balay PetscFunctionReturn(0); 1922a7e14dcfSSatish Balay } 1923a7e14dcfSSatish Balay 1924a7e14dcfSSatish Balay /*@C 1925737f463aSAlp Dener TaoResidualMonitor - Views the least-squares residual at each iteration 1926a7e14dcfSSatish Balay 1927c3339decSBarry Smith Collective 1928a7e14dcfSSatish Balay 1929a7e14dcfSSatish Balay Input Parameters: 1930441846f8SBarry Smith + tao - the Tao context 193165ba42b6SBarry Smith - ctx - `PetscViewer` context or NULL 1932a7e14dcfSSatish Balay 1933a7e14dcfSSatish Balay Options Database Keys: 1934147403d9SBarry Smith . -tao_view_ls_residual - view the least-squares residual at each iteration 1935a7e14dcfSSatish Balay 1936a7e14dcfSSatish Balay Level: advanced 1937a7e14dcfSSatish Balay 1938db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1939a7e14dcfSSatish Balay @*/ 1940d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoResidualMonitor(Tao tao, void *ctx) 1941d71ae5a4SJacob Faibussowitsch { 1942d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 194347a47007SBarry Smith 1944a7e14dcfSSatish Balay PetscFunctionBegin; 194594511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1946d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 19479566063dSJacob Faibussowitsch PetscCall(VecView(tao->ls_res, viewer)); 1948a7e14dcfSSatish Balay PetscFunctionReturn(0); 1949a7e14dcfSSatish Balay } 1950a7e14dcfSSatish Balay 19517fab98ebSJason Sarich /*@ 1952a7e14dcfSSatish Balay TaoDefaultConvergenceTest - Determines whether the solver should continue iterating 1953a7e14dcfSSatish Balay or terminate. 1954a7e14dcfSSatish Balay 1955c3339decSBarry Smith Collective 1956a7e14dcfSSatish Balay 1957a7e14dcfSSatish Balay Input Parameters: 1958441846f8SBarry Smith + tao - the Tao context 1959a7e14dcfSSatish Balay - dummy - unused dummy context 1960a7e14dcfSSatish Balay 1961a7e14dcfSSatish Balay Output Parameter: 1962a7e14dcfSSatish Balay . reason - for terminating 1963a7e14dcfSSatish Balay 1964a7e14dcfSSatish Balay Notes: 1965a7e14dcfSSatish Balay This routine checks the residual in the optimality conditions, the 1966a7e14dcfSSatish Balay relative residual in the optimity conditions, the number of function 1967a7e14dcfSSatish Balay evaluations, and the function value to test convergence. Some 1968a7e14dcfSSatish Balay solvers may use different convergence routines. 1969a7e14dcfSSatish Balay 1970a7e14dcfSSatish Balay Level: developer 1971a7e14dcfSSatish Balay 1972c2e3fba1SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoGetConvergedReason()`, `TaoSetConvergedReason()` 1973a7e14dcfSSatish Balay @*/ 1974a7e14dcfSSatish Balay 1975d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultConvergenceTest(Tao tao, void *dummy) 1976d71ae5a4SJacob Faibussowitsch { 1977a7e14dcfSSatish Balay PetscInt niter = tao->niter, nfuncs = PetscMax(tao->nfuncs, tao->nfuncgrads); 1978a7e14dcfSSatish Balay PetscInt max_funcs = tao->max_funcs; 1979a7e14dcfSSatish Balay PetscReal gnorm = tao->residual, gnorm0 = tao->gnorm0; 1980a7e14dcfSSatish Balay PetscReal f = tao->fc, steptol = tao->steptol, trradius = tao->step; 1981a7e14dcfSSatish Balay PetscReal gatol = tao->gatol, grtol = tao->grtol, gttol = tao->gttol; 1982e52336cbSBarry Smith PetscReal catol = tao->catol, crtol = tao->crtol; 1983e52336cbSBarry Smith PetscReal fmin = tao->fmin, cnorm = tao->cnorm; 1984e4cb33bbSBarry Smith TaoConvergedReason reason = tao->reason; 1985a7e14dcfSSatish Balay 1986a7e14dcfSSatish Balay PetscFunctionBegin; 1987441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 198894511df7SStefano Zampini if (reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(0); 1989a7e14dcfSSatish Balay 1990a7e14dcfSSatish Balay if (PetscIsInfOrNanReal(f)) { 19919566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Failed to converged, function value is Inf or NaN\n")); 1992a7e14dcfSSatish Balay reason = TAO_DIVERGED_NAN; 1993a7e14dcfSSatish Balay } else if (f <= fmin && cnorm <= catol) { 19949566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Converged due to function value %g < minimum function value %g\n", (double)f, (double)fmin)); 1995a7e14dcfSSatish Balay reason = TAO_CONVERGED_MINF; 1996a7e14dcfSSatish Balay } else if (gnorm <= gatol && cnorm <= catol) { 19979566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Converged due to residual norm ||g(X)||=%g < %g\n", (double)gnorm, (double)gatol)); 1998a7e14dcfSSatish Balay reason = TAO_CONVERGED_GATOL; 1999a7e14dcfSSatish Balay } else if (f != 0 && PetscAbsReal(gnorm / f) <= grtol && cnorm <= crtol) { 20009566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Converged due to residual ||g(X)||/|f(X)| =%g < %g\n", (double)(gnorm / f), (double)grtol)); 2001a7e14dcfSSatish Balay reason = TAO_CONVERGED_GRTOL; 2002e1d16bb9SBarry Smith } else if (gnorm0 != 0 && ((gttol == 0 && gnorm == 0) || gnorm / gnorm0 < gttol) && cnorm <= crtol) { 20039566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Converged due to relative residual norm ||g(X)||/||g(X0)|| = %g < %g\n", (double)(gnorm / gnorm0), (double)gttol)); 2004a7e14dcfSSatish Balay reason = TAO_CONVERGED_GTTOL; 200594511df7SStefano Zampini } else if (max_funcs >= 0 && nfuncs > max_funcs) { 20069566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Exceeded maximum number of function evaluations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", nfuncs, max_funcs)); 2007a7e14dcfSSatish Balay reason = TAO_DIVERGED_MAXFCN; 2008a7e14dcfSSatish Balay } else if (tao->lsflag != 0) { 20099566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Tao Line Search failure.\n")); 2010a7e14dcfSSatish Balay reason = TAO_DIVERGED_LS_FAILURE; 2011a7e14dcfSSatish Balay } else if (trradius < steptol && niter > 0) { 20129566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Trust region/step size too small: %g < %g\n", (double)trradius, (double)steptol)); 2013a7e14dcfSSatish Balay reason = TAO_CONVERGED_STEPTOL; 2014e031d6f5SAlp Dener } else if (niter >= tao->max_it) { 201563a3b9bcSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Exceeded maximum number of iterations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", niter, tao->max_it)); 2016a7e14dcfSSatish Balay reason = TAO_DIVERGED_MAXITS; 2017a7e14dcfSSatish Balay } else { 2018a7e14dcfSSatish Balay reason = TAO_CONTINUE_ITERATING; 2019a7e14dcfSSatish Balay } 2020a7e14dcfSSatish Balay tao->reason = reason; 2021a7e14dcfSSatish Balay PetscFunctionReturn(0); 2022a7e14dcfSSatish Balay } 2023a7e14dcfSSatish Balay 2024a7e14dcfSSatish Balay /*@C 2025a7e14dcfSSatish Balay TaoSetOptionsPrefix - Sets the prefix used for searching for all 202665ba42b6SBarry Smith Tao options in the database. 2027a7e14dcfSSatish Balay 2028c3339decSBarry Smith Logically Collective 2029a7e14dcfSSatish Balay 2030a7e14dcfSSatish Balay Input Parameters: 2031441846f8SBarry Smith + tao - the Tao context 203265ba42b6SBarry Smith - prefix - the prefix string to prepend to all Tao option requests 2033a7e14dcfSSatish Balay 2034a7e14dcfSSatish Balay Notes: 2035a7e14dcfSSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 2036a7e14dcfSSatish Balay The first character of all runtime options is AUTOMATICALLY the hyphen. 2037a7e14dcfSSatish Balay 2038a7e14dcfSSatish Balay For example, to distinguish between the runtime options for two 203965ba42b6SBarry Smith different Tao solvers, one could call 2040a7e14dcfSSatish Balay .vb 2041a7e14dcfSSatish Balay TaoSetOptionsPrefix(tao1,"sys1_") 2042a7e14dcfSSatish Balay TaoSetOptionsPrefix(tao2,"sys2_") 2043a7e14dcfSSatish Balay .ve 2044a7e14dcfSSatish Balay 2045a7e14dcfSSatish Balay This would enable use of different options for each system, such as 2046a7e14dcfSSatish Balay .vb 20479fa2b5dcSStefano Zampini -sys1_tao_method blmvm -sys1_tao_grtol 1.e-3 20489fa2b5dcSStefano Zampini -sys2_tao_method lmvm -sys2_tao_grtol 1.e-4 2049a7e14dcfSSatish Balay .ve 2050a7e14dcfSSatish Balay 2051a7e14dcfSSatish Balay Level: advanced 2052a7e14dcfSSatish Balay 205365ba42b6SBarry Smith .seealso: `TaoSetFromOptions()`, `TaoAppendOptionsPrefix()`, `TaoGetOptionsPrefix()` 2054a7e14dcfSSatish Balay @*/ 2055a7e14dcfSSatish Balay 2056d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetOptionsPrefix(Tao tao, const char p[]) 2057d71ae5a4SJacob Faibussowitsch { 2058a7e14dcfSSatish Balay PetscFunctionBegin; 205994511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 20609566063dSJacob Faibussowitsch PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tao, p)); 20611baa6e33SBarry Smith if (tao->linesearch) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch, p)); 20621baa6e33SBarry Smith if (tao->ksp) PetscCall(KSPSetOptionsPrefix(tao->ksp, p)); 2063a7e14dcfSSatish Balay PetscFunctionReturn(0); 2064a7e14dcfSSatish Balay } 2065a7e14dcfSSatish Balay 2066a7e14dcfSSatish Balay /*@C 2067a7e14dcfSSatish Balay TaoAppendOptionsPrefix - Appends to the prefix used for searching for all 206865ba42b6SBarry Smith Tao options in the database. 2069a7e14dcfSSatish Balay 2070c3339decSBarry Smith Logically Collective 2071a7e14dcfSSatish Balay 2072a7e14dcfSSatish Balay Input Parameters: 2073441846f8SBarry Smith + tao - the Tao solver context 207465ba42b6SBarry Smith - prefix - the prefix string to prepend to all Tao option requests 2075a7e14dcfSSatish Balay 207665ba42b6SBarry Smith Note: 2077a7e14dcfSSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 207865ba42b6SBarry Smith The first character of all runtime options is automatically the hyphen. 2079a7e14dcfSSatish Balay 2080a7e14dcfSSatish Balay Level: advanced 2081a7e14dcfSSatish Balay 208265ba42b6SBarry Smith .seealso: `TaoSetFromOptions()`, `TaoSetOptionsPrefix()`, `TaoGetOptionsPrefix()` 2083a7e14dcfSSatish Balay @*/ 2084d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoAppendOptionsPrefix(Tao tao, const char p[]) 2085d71ae5a4SJacob Faibussowitsch { 2086a7e14dcfSSatish Balay PetscFunctionBegin; 208794511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 20889566063dSJacob Faibussowitsch PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao, p)); 20891baa6e33SBarry Smith if (tao->linesearch) PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao->linesearch, p)); 20901baa6e33SBarry Smith if (tao->ksp) PetscCall(KSPAppendOptionsPrefix(tao->ksp, p)); 2091a7e14dcfSSatish Balay PetscFunctionReturn(0); 2092a7e14dcfSSatish Balay } 2093a7e14dcfSSatish Balay 2094a7e14dcfSSatish Balay /*@C 2095a7e14dcfSSatish Balay TaoGetOptionsPrefix - Gets the prefix used for searching for all 209665ba42b6SBarry Smith Tao options in the database 2097a7e14dcfSSatish Balay 2098a7e14dcfSSatish Balay Not Collective 2099a7e14dcfSSatish Balay 2100a7e14dcfSSatish Balay Input Parameters: 2101441846f8SBarry Smith . tao - the Tao context 2102a7e14dcfSSatish Balay 2103a7e14dcfSSatish Balay Output Parameters: 2104a7e14dcfSSatish Balay . prefix - pointer to the prefix string used is returned 2105a7e14dcfSSatish Balay 210665ba42b6SBarry Smith Fortran Note: 210795452b02SPatrick Sanan On the fortran side, the user should pass in a string 'prefix' of 2108a7e14dcfSSatish Balay sufficient length to hold the prefix. 2109a7e14dcfSSatish Balay 2110a7e14dcfSSatish Balay Level: advanced 2111a7e14dcfSSatish Balay 211265ba42b6SBarry Smith .seealso: `TaoSetFromOptions()`, `TaoSetOptionsPrefix()`, `TaoAppendOptionsPrefix()` 2113a7e14dcfSSatish Balay @*/ 2114d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetOptionsPrefix(Tao tao, const char *p[]) 2115d71ae5a4SJacob Faibussowitsch { 211694511df7SStefano Zampini PetscFunctionBegin; 211794511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 21189566063dSJacob Faibussowitsch PetscCall(PetscObjectGetOptionsPrefix((PetscObject)tao, p)); 211994511df7SStefano Zampini PetscFunctionReturn(0); 2120a7e14dcfSSatish Balay } 2121a7e14dcfSSatish Balay 2122a7e14dcfSSatish Balay /*@C 2123a7e14dcfSSatish Balay TaoSetType - Sets the method for the unconstrained minimization solver. 2124a7e14dcfSSatish Balay 2125c3339decSBarry Smith Collective 2126a7e14dcfSSatish Balay 2127a7e14dcfSSatish Balay Input Parameters: 2128441846f8SBarry Smith + solver - the Tao solver context 2129a7e14dcfSSatish Balay - type - a known method 2130a7e14dcfSSatish Balay 2131a7e14dcfSSatish Balay Options Database Key: 213258417fe7SBarry Smith . -tao_type <type> - Sets the method; use -help for a list 213358417fe7SBarry Smith of available methods (for instance, "-tao_type lmvm" or "-tao_type tron") 2134a7e14dcfSSatish Balay 2135a7e14dcfSSatish Balay Available methods include: 213665ba42b6SBarry Smith + `TAONLS` - nls Newton's method with line search for unconstrained minimization 213765ba42b6SBarry Smith . `TAONTR` - ntr Newton's method with trust region for unconstrained minimization 213865ba42b6SBarry Smith . `TAONTL` - ntl Newton's method with trust region, line search for unconstrained minimization 213965ba42b6SBarry Smith . `TAOLMVM` - lmvm Limited memory variable metric method for unconstrained minimization 214065ba42b6SBarry Smith . `TAOCG` - cg Nonlinear conjugate gradient method for unconstrained minimization 214165ba42b6SBarry Smith . `TAONM` - nm Nelder-Mead algorithm for derivate-free unconstrained minimization 214265ba42b6SBarry Smith . `TAOTRON` - tron Newton Trust Region method for bound constrained minimization 214365ba42b6SBarry Smith . `TAOGPCG` - gpcg Newton Trust Region method for quadratic bound constrained minimization 214465ba42b6SBarry Smith . `TAOBLMVM` - blmvm Limited memory variable metric method for bound constrained minimization 214565ba42b6SBarry Smith . `TAOLCL` - lcl Linearly constrained Lagrangian method for pde-constrained minimization 214665ba42b6SBarry Smith - `TAOPOUNDERS` - pounders Model-based algorithm for nonlinear least squares 2147a7e14dcfSSatish Balay 2148a7e14dcfSSatish Balay Level: intermediate 2149a7e14dcfSSatish Balay 215065ba42b6SBarry Smith .seealso: `Tao`, `TaoCreate()`, `TaoGetType()`, `TaoType` 2151a7e14dcfSSatish Balay 2152a7e14dcfSSatish Balay @*/ 2153d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetType(Tao tao, TaoType type) 2154d71ae5a4SJacob Faibussowitsch { 2155441846f8SBarry Smith PetscErrorCode (*create_xxx)(Tao); 2156a7e14dcfSSatish Balay PetscBool issame; 2157a7e14dcfSSatish Balay 2158a7e14dcfSSatish Balay PetscFunctionBegin; 2159441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2160a7e14dcfSSatish Balay 21619566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)tao, type, &issame)); 2162a7e14dcfSSatish Balay if (issame) PetscFunctionReturn(0); 2163a7e14dcfSSatish Balay 21649566063dSJacob Faibussowitsch PetscCall(PetscFunctionListFind(TaoList, type, (void (**)(void)) & create_xxx)); 21653c859ba3SBarry Smith PetscCheck(create_xxx, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unable to find requested Tao type %s", type); 2166a7e14dcfSSatish Balay 2167a7e14dcfSSatish Balay /* Destroy the existing solver information */ 2168dbbe0bcdSBarry Smith PetscTryTypeMethod(tao, destroy); 21691baa6e33SBarry Smith PetscCall(KSPDestroy(&tao->ksp)); 21709566063dSJacob Faibussowitsch PetscCall(TaoLineSearchDestroy(&tao->linesearch)); 217183c8fe1dSLisandro Dalcin tao->ops->setup = NULL; 217283c8fe1dSLisandro Dalcin tao->ops->solve = NULL; 217383c8fe1dSLisandro Dalcin tao->ops->view = NULL; 217483c8fe1dSLisandro Dalcin tao->ops->setfromoptions = NULL; 217583c8fe1dSLisandro Dalcin tao->ops->destroy = NULL; 2176a7e14dcfSSatish Balay 2177a7e14dcfSSatish Balay tao->setupcalled = PETSC_FALSE; 2178a7e14dcfSSatish Balay 21799566063dSJacob Faibussowitsch PetscCall((*create_xxx)(tao)); 21809566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject)tao, type)); 2181a7e14dcfSSatish Balay PetscFunctionReturn(0); 2182a7e14dcfSSatish Balay } 218347a47007SBarry Smith 2184*67be906fSBarry Smith /*@C 218565ba42b6SBarry Smith TaoRegister - Adds a method to the Tao package for unconstrained minimization. 2186a7e14dcfSSatish Balay 2187a7e14dcfSSatish Balay Synopsis: 21885bd1e576SStefano Zampini TaoRegister(char *name_solver,char *path,char *name_Create,PetscErrorCode (*routine_Create)(Tao)) 2189a7e14dcfSSatish Balay 2190a7e14dcfSSatish Balay Not collective 2191a7e14dcfSSatish Balay 2192a7e14dcfSSatish Balay Input Parameters: 2193a7e14dcfSSatish Balay + sname - name of a new user-defined solver 2194a7e14dcfSSatish Balay - func - routine to Create method context 2195a7e14dcfSSatish Balay 2196a7e14dcfSSatish Balay Sample usage: 2197a7e14dcfSSatish Balay .vb 2198441846f8SBarry Smith TaoRegister("my_solver",MySolverCreate); 2199a7e14dcfSSatish Balay .ve 2200a7e14dcfSSatish Balay 2201a7e14dcfSSatish Balay Then, your solver can be chosen with the procedural interface via 2202a7e14dcfSSatish Balay $ TaoSetType(tao,"my_solver") 2203a7e14dcfSSatish Balay or at runtime via the option 220458417fe7SBarry Smith $ -tao_type my_solver 2205a7e14dcfSSatish Balay 2206a7e14dcfSSatish Balay Level: advanced 2207a7e14dcfSSatish Balay 2208*67be906fSBarry Smith Note: 2209*67be906fSBarry Smith `TaoRegister()` may be called multiple times to add several user-defined solvers. 2210*67be906fSBarry Smith 221165ba42b6SBarry Smith .seealso: `Tao`, `TaoSetType()`, `TaoRegisterAll()`, `TaoRegisterDestroy()` 2212*67be906fSBarry Smith @*/ 2213d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoRegister(const char sname[], PetscErrorCode (*func)(Tao)) 2214d71ae5a4SJacob Faibussowitsch { 2215a7e14dcfSSatish Balay PetscFunctionBegin; 22169566063dSJacob Faibussowitsch PetscCall(TaoInitializePackage()); 22179566063dSJacob Faibussowitsch PetscCall(PetscFunctionListAdd(&TaoList, sname, (void (*)(void))func)); 2218a7e14dcfSSatish Balay PetscFunctionReturn(0); 2219a7e14dcfSSatish Balay } 2220a7e14dcfSSatish Balay 2221a7e14dcfSSatish Balay /*@C 2222441846f8SBarry Smith TaoRegisterDestroy - Frees the list of minimization solvers that were 222365ba42b6SBarry Smith registered by `TaoRegisterDynamic()`. 2224a7e14dcfSSatish Balay 2225a7e14dcfSSatish Balay Not Collective 2226a7e14dcfSSatish Balay 2227a7e14dcfSSatish Balay Level: advanced 2228a7e14dcfSSatish Balay 2229*67be906fSBarry Smith .seealso: `Tao`, `TaoRegisterAll()`, `TaoRegister()` 2230a7e14dcfSSatish Balay @*/ 2231d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoRegisterDestroy(void) 2232d71ae5a4SJacob Faibussowitsch { 2233a7e14dcfSSatish Balay PetscFunctionBegin; 22349566063dSJacob Faibussowitsch PetscCall(PetscFunctionListDestroy(&TaoList)); 2235441846f8SBarry Smith TaoRegisterAllCalled = PETSC_FALSE; 2236a7e14dcfSSatish Balay PetscFunctionReturn(0); 2237a7e14dcfSSatish Balay } 223847a47007SBarry Smith 22398931d482SJason Sarich /*@ 22408931d482SJason Sarich TaoGetIterationNumber - Gets the number of Tao iterations completed 22418931d482SJason Sarich at this time. 22428931d482SJason Sarich 22438931d482SJason Sarich Not Collective 22448931d482SJason Sarich 22458931d482SJason Sarich Input Parameter: 22468931d482SJason Sarich . tao - Tao context 22478931d482SJason Sarich 22488931d482SJason Sarich Output Parameter: 22498931d482SJason Sarich . iter - iteration number 22508931d482SJason Sarich 22518931d482SJason Sarich Notes: 22528931d482SJason Sarich For example, during the computation of iteration 2 this would return 1. 22538931d482SJason Sarich 22548931d482SJason Sarich Level: intermediate 22558931d482SJason Sarich 2256*67be906fSBarry Smith .seealso: `Tao`, `TaoGetLinearSolveIterations()`, `TaoGetResidualNorm()`, `TaoGetObjective()` 22578931d482SJason Sarich @*/ 2258d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetIterationNumber(Tao tao, PetscInt *iter) 2259d71ae5a4SJacob Faibussowitsch { 22608931d482SJason Sarich PetscFunctionBegin; 22618931d482SJason Sarich PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 22628931d482SJason Sarich PetscValidIntPointer(iter, 2); 22638931d482SJason Sarich *iter = tao->niter; 22648931d482SJason Sarich PetscFunctionReturn(0); 22658931d482SJason Sarich } 22668931d482SJason Sarich 22678931d482SJason Sarich /*@ 226879f5d8caSBarry Smith TaoGetResidualNorm - Gets the current value of the norm of the residual 226979f5d8caSBarry Smith at this time. 227079f5d8caSBarry Smith 227179f5d8caSBarry Smith Not Collective 227279f5d8caSBarry Smith 227379f5d8caSBarry Smith Input Parameter: 227479f5d8caSBarry Smith . tao - Tao context 227579f5d8caSBarry Smith 227679f5d8caSBarry Smith Output Parameter: 227779f5d8caSBarry Smith . value - the current value 227879f5d8caSBarry Smith 227979f5d8caSBarry Smith Level: intermediate 228079f5d8caSBarry Smith 2281*67be906fSBarry Smith Developer Note: 2282*67be906fSBarry Smith This is the 2-norm of the residual, we cannot use `TaoGetGradientNorm()` because that has 228379f5d8caSBarry Smith a different meaning. For some reason Tao sometimes calls the gradient the residual. 228479f5d8caSBarry Smith 2285*67be906fSBarry Smith .seealso: `Tao`, `TaoGetLinearSolveIterations()`, `TaoGetIterationNumber()`, `TaoGetObjective()` 228679f5d8caSBarry Smith @*/ 2287d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetResidualNorm(Tao tao, PetscReal *value) 2288d71ae5a4SJacob Faibussowitsch { 228979f5d8caSBarry Smith PetscFunctionBegin; 229079f5d8caSBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 229179f5d8caSBarry Smith PetscValidRealPointer(value, 2); 229279f5d8caSBarry Smith *value = tao->residual; 229379f5d8caSBarry Smith PetscFunctionReturn(0); 229479f5d8caSBarry Smith } 229579f5d8caSBarry Smith 229679f5d8caSBarry Smith /*@ 22978931d482SJason Sarich TaoSetIterationNumber - Sets the current iteration number. 22988931d482SJason Sarich 2299c3339decSBarry Smith Logically Collective 23008931d482SJason Sarich 2301d8d19677SJose E. Roman Input Parameters: 2302a2b725a8SWilliam Gropp + tao - Tao context 2303a2b725a8SWilliam Gropp - iter - iteration number 23048931d482SJason Sarich 23058931d482SJason Sarich Level: developer 23068931d482SJason Sarich 2307*67be906fSBarry Smith .seealso: `Tao`, `TaoGetLinearSolveIterations()` 23088931d482SJason Sarich @*/ 2309d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetIterationNumber(Tao tao, PetscInt iter) 2310d71ae5a4SJacob Faibussowitsch { 23118931d482SJason Sarich PetscFunctionBegin; 23128931d482SJason Sarich PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 231394511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao, iter, 2); 23149566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao)); 23158931d482SJason Sarich tao->niter = iter; 23169566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao)); 23178931d482SJason Sarich PetscFunctionReturn(0); 23188931d482SJason Sarich } 23198931d482SJason Sarich 23208931d482SJason Sarich /*@ 23218931d482SJason Sarich TaoGetTotalIterationNumber - Gets the total number of Tao iterations 23228931d482SJason Sarich completed. This number keeps accumulating if multiple solves 23238931d482SJason Sarich are called with the Tao object. 23248931d482SJason Sarich 23258931d482SJason Sarich Not Collective 23268931d482SJason Sarich 23278931d482SJason Sarich Input Parameter: 23288931d482SJason Sarich . tao - Tao context 23298931d482SJason Sarich 23308931d482SJason Sarich Output Parameter: 23318931d482SJason Sarich . iter - iteration number 23328931d482SJason Sarich 23338931d482SJason Sarich Level: intermediate 23348931d482SJason Sarich 2335*67be906fSBarry Smith Notes: 2336*67be906fSBarry Smith The total iteration count is updated after each solve, if there is a current 2337*67be906fSBarry Smith `TaoSolve()` in progress then those iterations are not yet counted. 2338*67be906fSBarry Smith 2339*67be906fSBarry Smith .seealso: `Tao`, `TaoGetLinearSolveIterations()` 23408931d482SJason Sarich @*/ 2341d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetTotalIterationNumber(Tao tao, PetscInt *iter) 2342d71ae5a4SJacob Faibussowitsch { 23438931d482SJason Sarich PetscFunctionBegin; 23448931d482SJason Sarich PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 23458931d482SJason Sarich PetscValidIntPointer(iter, 2); 23468931d482SJason Sarich *iter = tao->ntotalits; 23478931d482SJason Sarich PetscFunctionReturn(0); 23488931d482SJason Sarich } 23498931d482SJason Sarich 23508931d482SJason Sarich /*@ 23518931d482SJason Sarich TaoSetTotalIterationNumber - Sets the current total iteration number. 23528931d482SJason Sarich 2353c3339decSBarry Smith Logically Collective 23548931d482SJason Sarich 2355d8d19677SJose E. Roman Input Parameters: 2356a2b725a8SWilliam Gropp + tao - Tao context 2357a2b725a8SWilliam Gropp - iter - iteration number 23588931d482SJason Sarich 23598931d482SJason Sarich Level: developer 23608931d482SJason Sarich 2361*67be906fSBarry Smith .seealso: `Tao`, `TaoGetLinearSolveIterations()` 23628931d482SJason Sarich @*/ 2363d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetTotalIterationNumber(Tao tao, PetscInt iter) 2364d71ae5a4SJacob Faibussowitsch { 23658931d482SJason Sarich PetscFunctionBegin; 23668931d482SJason Sarich PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 236794511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao, iter, 2); 23689566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao)); 23698931d482SJason Sarich tao->ntotalits = iter; 23709566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao)); 23718931d482SJason Sarich PetscFunctionReturn(0); 23728931d482SJason Sarich } 23738931d482SJason Sarich 2374a7e14dcfSSatish Balay /*@ 2375e4cb33bbSBarry Smith TaoSetConvergedReason - Sets the termination flag on a Tao object 2376a7e14dcfSSatish Balay 2377c3339decSBarry Smith Logically Collective 2378a7e14dcfSSatish Balay 2379a7e14dcfSSatish Balay Input Parameters: 2380441846f8SBarry Smith + tao - the Tao context 2381a7e14dcfSSatish Balay - reason - one of 2382*67be906fSBarry Smith .vb 2383*67be906fSBarry Smith TAO_CONVERGED_ATOL (2), 2384*67be906fSBarry Smith TAO_CONVERGED_RTOL (3), 2385*67be906fSBarry Smith TAO_CONVERGED_STEPTOL (4), 2386*67be906fSBarry Smith TAO_CONVERGED_MINF (5), 2387*67be906fSBarry Smith TAO_CONVERGED_USER (6), 2388*67be906fSBarry Smith TAO_DIVERGED_MAXITS (-2), 2389*67be906fSBarry Smith TAO_DIVERGED_NAN (-4), 2390*67be906fSBarry Smith TAO_DIVERGED_MAXFCN (-5), 2391*67be906fSBarry Smith TAO_DIVERGED_LS_FAILURE (-6), 2392*67be906fSBarry Smith TAO_DIVERGED_TR_REDUCTION (-7), 2393*67be906fSBarry Smith TAO_DIVERGED_USER (-8), 2394*67be906fSBarry Smith TAO_CONTINUE_ITERATING (0) 2395*67be906fSBarry Smith .ve 2396a7e14dcfSSatish Balay 2397a7e14dcfSSatish Balay Level: intermediate 2398a7e14dcfSSatish Balay 2399*67be906fSBarry Smith .seealso: `Tao` 2400a7e14dcfSSatish Balay @*/ 2401d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergedReason(Tao tao, TaoConvergedReason reason) 2402d71ae5a4SJacob Faibussowitsch { 2403a7e14dcfSSatish Balay PetscFunctionBegin; 240494511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 240594511df7SStefano Zampini PetscValidLogicalCollectiveEnum(tao, reason, 2); 2406a7e14dcfSSatish Balay tao->reason = reason; 2407a7e14dcfSSatish Balay PetscFunctionReturn(0); 2408a7e14dcfSSatish Balay } 2409a7e14dcfSSatish Balay 2410a7e14dcfSSatish Balay /*@ 2411e4cb33bbSBarry Smith TaoGetConvergedReason - Gets the reason the Tao iteration was stopped. 2412a7e14dcfSSatish Balay 2413a7e14dcfSSatish Balay Not Collective 2414a7e14dcfSSatish Balay 2415a7e14dcfSSatish Balay Input Parameter: 2416441846f8SBarry Smith . tao - the Tao solver context 2417a7e14dcfSSatish Balay 2418a7e14dcfSSatish Balay Output Parameter: 2419a7e14dcfSSatish Balay . reason - one of 2420*67be906fSBarry Smith .vb 2421*67be906fSBarry Smith TAO_CONVERGED_GATOL (3) ||g(X)|| < gatol 2422*67be906fSBarry Smith TAO_CONVERGED_GRTOL (4) ||g(X)|| / f(X) < grtol 2423*67be906fSBarry Smith TAO_CONVERGED_GTTOL (5) ||g(X)|| / ||g(X0)|| < gttol 2424*67be906fSBarry Smith TAO_CONVERGED_STEPTOL (6) step size small 2425*67be906fSBarry Smith TAO_CONVERGED_MINF (7) F < F_min 2426*67be906fSBarry Smith TAO_CONVERGED_USER (8) User defined 2427*67be906fSBarry Smith TAO_DIVERGED_MAXITS (-2) its > maxits 2428*67be906fSBarry Smith TAO_DIVERGED_NAN (-4) Numerical problems 2429*67be906fSBarry Smith TAO_DIVERGED_MAXFCN (-5) fevals > max_funcsals 2430*67be906fSBarry Smith TAO_DIVERGED_LS_FAILURE (-6) line search failure 2431*67be906fSBarry Smith TAO_DIVERGED_TR_REDUCTION (-7) trust region failure 2432*67be906fSBarry Smith TAO_DIVERGED_USER (-8) (user defined) 2433*67be906fSBarry Smith TAO_CONTINUE_ITERATING (0) 2434*67be906fSBarry Smith .ve 2435a7e14dcfSSatish Balay 2436a7e14dcfSSatish Balay where 2437a7e14dcfSSatish Balay + X - current solution 2438a7e14dcfSSatish Balay . X0 - initial guess 2439a7e14dcfSSatish Balay . f(X) - current function value 2440a7e14dcfSSatish Balay . f(X*) - true solution (estimated) 2441a7e14dcfSSatish Balay . g(X) - current gradient 2442a7e14dcfSSatish Balay . its - current iterate number 2443a7e14dcfSSatish Balay . maxits - maximum number of iterates 2444a7e14dcfSSatish Balay . fevals - number of function evaluations 2445a7e14dcfSSatish Balay - max_funcsals - maximum number of function evaluations 2446a7e14dcfSSatish Balay 2447a7e14dcfSSatish Balay Level: intermediate 2448a7e14dcfSSatish Balay 2449db781477SPatrick Sanan .seealso: `TaoSetConvergenceTest()`, `TaoSetTolerances()` 2450a7e14dcfSSatish Balay @*/ 2451d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConvergedReason(Tao tao, TaoConvergedReason *reason) 2452d71ae5a4SJacob Faibussowitsch { 2453a7e14dcfSSatish Balay PetscFunctionBegin; 2454441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2455a7e14dcfSSatish Balay PetscValidPointer(reason, 2); 2456a7e14dcfSSatish Balay *reason = tao->reason; 2457a7e14dcfSSatish Balay PetscFunctionReturn(0); 2458a7e14dcfSSatish Balay } 2459a7e14dcfSSatish Balay 2460a7e14dcfSSatish Balay /*@ 2461a7e14dcfSSatish Balay TaoGetSolutionStatus - Get the current iterate, objective value, 2462a7e14dcfSSatish Balay residual, infeasibility, and termination 2463a7e14dcfSSatish Balay 2464a7e14dcfSSatish Balay Not Collective 2465a7e14dcfSSatish Balay 2466f899ff85SJose E. Roman Input Parameter: 2467441846f8SBarry Smith . tao - the Tao context 2468a7e14dcfSSatish Balay 2469a7e14dcfSSatish Balay Output Parameters: 2470a7e14dcfSSatish Balay + iterate - the current iterate number (>=0) 2471a7e14dcfSSatish Balay . f - the current function value 247258417fe7SBarry Smith . gnorm - the square of the gradient norm, duality gap, or other measure indicating distance from optimality. 2473a7e14dcfSSatish Balay . cnorm - the infeasibility of the current solution with regard to the constraints. 2474a7e14dcfSSatish Balay . xdiff - the step length or trust region radius of the most recent iterate. 247565ba42b6SBarry Smith - reason - The termination reason, which can equal `TAO_CONTINUE_ITERATING` 2476a7e14dcfSSatish Balay 2477a7e14dcfSSatish Balay Level: intermediate 2478a7e14dcfSSatish Balay 247965ba42b6SBarry Smith Notes: 248065ba42b6SBarry Smith Tao returns the values set by the solvers in the routine `TaoMonitor()`. 2481a7e14dcfSSatish Balay 2482*67be906fSBarry Smith If any of the output arguments are set to `NULL`, no corresponding value will be returned. 2483a7e14dcfSSatish Balay 2484db781477SPatrick Sanan .seealso: `TaoMonitor()`, `TaoGetConvergedReason()` 2485a7e14dcfSSatish Balay @*/ 2486d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetSolutionStatus(Tao tao, PetscInt *its, PetscReal *f, PetscReal *gnorm, PetscReal *cnorm, PetscReal *xdiff, TaoConvergedReason *reason) 2487d71ae5a4SJacob Faibussowitsch { 2488a7e14dcfSSatish Balay PetscFunctionBegin; 248994511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2490a7e14dcfSSatish Balay if (its) *its = tao->niter; 2491a7e14dcfSSatish Balay if (f) *f = tao->fc; 2492a7e14dcfSSatish Balay if (gnorm) *gnorm = tao->residual; 2493a7e14dcfSSatish Balay if (cnorm) *cnorm = tao->cnorm; 2494a7e14dcfSSatish Balay if (reason) *reason = tao->reason; 2495a7e14dcfSSatish Balay if (xdiff) *xdiff = tao->step; 2496a7e14dcfSSatish Balay PetscFunctionReturn(0); 2497a7e14dcfSSatish Balay } 2498a7e14dcfSSatish Balay 2499a7e14dcfSSatish Balay /*@C 2500441846f8SBarry Smith TaoGetType - Gets the current Tao algorithm. 2501a7e14dcfSSatish Balay 2502a7e14dcfSSatish Balay Not Collective 2503a7e14dcfSSatish Balay 2504a7e14dcfSSatish Balay Input Parameter: 2505441846f8SBarry Smith . tao - the Tao solver context 2506a7e14dcfSSatish Balay 2507a7e14dcfSSatish Balay Output Parameter: 2508441846f8SBarry Smith . type - Tao method 2509a7e14dcfSSatish Balay 2510a7e14dcfSSatish Balay Level: intermediate 2511a7e14dcfSSatish Balay 251265ba42b6SBarry Smith .seealso: `Tao`, `TaoType`, `TaoSetType()` 2513a7e14dcfSSatish Balay @*/ 2514d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetType(Tao tao, TaoType *type) 2515d71ae5a4SJacob Faibussowitsch { 2516a7e14dcfSSatish Balay PetscFunctionBegin; 2517441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2518a7e14dcfSSatish Balay PetscValidPointer(type, 2); 2519a7e14dcfSSatish Balay *type = ((PetscObject)tao)->type_name; 2520a7e14dcfSSatish Balay PetscFunctionReturn(0); 2521a7e14dcfSSatish Balay } 2522a7e14dcfSSatish Balay 2523a7e14dcfSSatish Balay /*@C 2524a7e14dcfSSatish Balay TaoMonitor - Monitor the solver and the current solution. This 2525a7e14dcfSSatish Balay routine will record the iteration number and residual statistics, 2526670c031eSStefano Zampini and call any monitors specified by the user. 2527a7e14dcfSSatish Balay 2528a7e14dcfSSatish Balay Input Parameters: 2529441846f8SBarry Smith + tao - the Tao context 2530a7e14dcfSSatish Balay . its - the current iterate number (>=0) 2531a7e14dcfSSatish Balay . f - the current objective function value 253258417fe7SBarry Smith . res - the gradient norm, square root of the duality gap, or other measure indicating distince from optimality. This measure will be recorded and 2533a7e14dcfSSatish Balay used for some termination tests. 2534a7e14dcfSSatish Balay . cnorm - the infeasibility of the current solution with regard to the constraints. 2535a7e14dcfSSatish Balay - steplength - multiple of the step direction added to the previous iterate. 2536a7e14dcfSSatish Balay 2537a7e14dcfSSatish Balay Output Parameters: 253865ba42b6SBarry Smith . reason - The termination reason, which can equal `TAO_CONTINUE_ITERATING` 2539a7e14dcfSSatish Balay 2540a7e14dcfSSatish Balay Options Database Key: 2541a7e14dcfSSatish Balay . -tao_monitor - Use the default monitor, which prints statistics to standard output 2542a7e14dcfSSatish Balay 2543a7e14dcfSSatish Balay Level: developer 2544a7e14dcfSSatish Balay 2545a1cb98faSBarry Smith .seealso: `Tao`, `TaoGetConvergedReason()`, `TaoMonitorDefault()`, `TaoSetMonitor()` 2546a7e14dcfSSatish Balay @*/ 2547d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitor(Tao tao, PetscInt its, PetscReal f, PetscReal res, PetscReal cnorm, PetscReal steplength) 2548d71ae5a4SJacob Faibussowitsch { 254947a47007SBarry Smith PetscInt i; 255047a47007SBarry Smith 2551a7e14dcfSSatish Balay PetscFunctionBegin; 2552441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2553a7e14dcfSSatish Balay tao->fc = f; 2554a7e14dcfSSatish Balay tao->residual = res; 2555a7e14dcfSSatish Balay tao->cnorm = cnorm; 2556a7e14dcfSSatish Balay tao->step = steplength; 2557e52336cbSBarry Smith if (!its) { 255894511df7SStefano Zampini tao->cnorm0 = cnorm; 255994511df7SStefano Zampini tao->gnorm0 = res; 2560a7e14dcfSSatish Balay } 25613c859ba3SBarry Smith PetscCheck(!PetscIsInfOrNanReal(f) && !PetscIsInfOrNanReal(res), PetscObjectComm((PetscObject)tao), PETSC_ERR_USER, "User provided compute function generated Inf or NaN"); 256248a46eb9SPierre Jolivet for (i = 0; i < tao->numbermonitors; i++) PetscCall((*tao->monitor[i])(tao, tao->monitorcontext[i])); 2563a7e14dcfSSatish Balay PetscFunctionReturn(0); 2564a7e14dcfSSatish Balay } 2565a7e14dcfSSatish Balay 2566a7e14dcfSSatish Balay /*@ 2567ae93cb3cSJason Sarich TaoSetConvergenceHistory - Sets the array used to hold the convergence history. 2568a7e14dcfSSatish Balay 2569c3339decSBarry Smith Logically Collective 2570a7e14dcfSSatish Balay 2571a7e14dcfSSatish Balay Input Parameters: 2572441846f8SBarry Smith + tao - the Tao solver context 2573a7e14dcfSSatish Balay . obj - array to hold objective value history 2574a7e14dcfSSatish Balay . resid - array to hold residual history 2575a7e14dcfSSatish Balay . cnorm - array to hold constraint violation history 2576be1558d8SJason Sarich . lits - integer array holds the number of linear iterations for each Tao iteration 2577*67be906fSBarry Smith . na - size of `obj`, `resid`, and `cnorm` 257865ba42b6SBarry Smith - reset - `PETSC_TRUE` indicates each new minimization resets the history counter to zero, 2579a7e14dcfSSatish Balay else it continues storing new values for new minimizations after the old ones 2580a7e14dcfSSatish Balay 2581*67be906fSBarry Smith Level: intermediate 2582*67be906fSBarry Smith 2583a7e14dcfSSatish Balay Notes: 258465ba42b6SBarry Smith If set, Tao will fill the given arrays with the indicated 2585ae93cb3cSJason Sarich information at each iteration. If 'obj','resid','cnorm','lits' are 2586*67be906fSBarry Smith *all* `NULL` then space (using size `na`, or 1000 if na is `PETSC_DECIDE` or 258765ba42b6SBarry Smith `PETSC_DEFAULT`) is allocated for the history. 2588*67be906fSBarry Smith If not all are `NULL`, then only the non-`NULL` information categories 2589ae93cb3cSJason Sarich will be stored, the others will be ignored. 2590ae93cb3cSJason Sarich 2591ae93cb3cSJason Sarich Any convergence information after iteration number 'na' will not be stored. 2592a7e14dcfSSatish Balay 2593a7e14dcfSSatish Balay This routine is useful, e.g., when running a code for purposes 2594a7e14dcfSSatish Balay of accurate performance monitoring, when no I/O should be done 2595a7e14dcfSSatish Balay during the section of code that is being timed. 2596a7e14dcfSSatish Balay 2597db781477SPatrick Sanan .seealso: `TaoGetConvergenceHistory()` 2598a7e14dcfSSatish Balay @*/ 2599d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergenceHistory(Tao tao, PetscReal obj[], PetscReal resid[], PetscReal cnorm[], PetscInt lits[], PetscInt na, PetscBool reset) 2600d71ae5a4SJacob Faibussowitsch { 2601a7e14dcfSSatish Balay PetscFunctionBegin; 2602441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2603064a246eSJacob Faibussowitsch if (obj) PetscValidRealPointer(obj, 2); 2604064a246eSJacob Faibussowitsch if (resid) PetscValidRealPointer(resid, 3); 2605064a246eSJacob Faibussowitsch if (cnorm) PetscValidRealPointer(cnorm, 4); 2606ae93cb3cSJason Sarich if (lits) PetscValidIntPointer(lits, 5); 2607ae93cb3cSJason Sarich 2608ae93cb3cSJason Sarich if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000; 2609ae93cb3cSJason Sarich if (!obj && !resid && !cnorm && !lits) { 26109566063dSJacob Faibussowitsch PetscCall(PetscCalloc4(na, &obj, na, &resid, na, &cnorm, na, &lits)); 2611ae93cb3cSJason Sarich tao->hist_malloc = PETSC_TRUE; 2612ae93cb3cSJason Sarich } 2613ae93cb3cSJason Sarich 2614a7e14dcfSSatish Balay tao->hist_obj = obj; 2615a7e14dcfSSatish Balay tao->hist_resid = resid; 2616a7e14dcfSSatish Balay tao->hist_cnorm = cnorm; 2617ae93cb3cSJason Sarich tao->hist_lits = lits; 2618a7e14dcfSSatish Balay tao->hist_max = na; 2619a7e14dcfSSatish Balay tao->hist_reset = reset; 2620ae93cb3cSJason Sarich tao->hist_len = 0; 2621a7e14dcfSSatish Balay PetscFunctionReturn(0); 2622a7e14dcfSSatish Balay } 2623a7e14dcfSSatish Balay 2624a7e14dcfSSatish Balay /*@C 262565ba42b6SBarry Smith TaoGetConvergenceHistory - Gets the arrays used that hold the convergence history. 2626a7e14dcfSSatish Balay 2627c3339decSBarry Smith Collective 2628a7e14dcfSSatish Balay 2629a7e14dcfSSatish Balay Input Parameter: 2630441846f8SBarry Smith . tao - the Tao context 2631a7e14dcfSSatish Balay 2632a7e14dcfSSatish Balay Output Parameters: 2633a7e14dcfSSatish Balay + obj - array used to hold objective value history 2634a7e14dcfSSatish Balay . resid - array used to hold residual history 2635a7e14dcfSSatish Balay . cnorm - array used to hold constraint violation history 2636ae93cb3cSJason Sarich . lits - integer array used to hold linear solver iteration count 2637*67be906fSBarry Smith - nhist - size of `obj`, `resid`, `cnorm`, and `lits` 2638*67be906fSBarry Smith 2639*67be906fSBarry Smith Level: advanced 2640a7e14dcfSSatish Balay 2641a7e14dcfSSatish Balay Notes: 264265ba42b6SBarry Smith This routine must be preceded by calls to `TaoSetConvergenceHistory()` 264365ba42b6SBarry Smith and `TaoSolve()`, otherwise it returns useless information. 2644ae93cb3cSJason Sarich 2645a7e14dcfSSatish Balay This routine is useful, e.g., when running a code for purposes 2646a7e14dcfSSatish Balay of accurate performance monitoring, when no I/O should be done 2647a7e14dcfSSatish Balay during the section of code that is being timed. 2648a7e14dcfSSatish Balay 2649*67be906fSBarry Smith Fortran Note: 2650*67be906fSBarry Smith The calling sequence is 2651*67be906fSBarry Smith .vb 2652*67be906fSBarry Smith call TaoGetConvergenceHistory(Tao tao, PetscInt nhist, PetscErrorCode ierr) 2653*67be906fSBarry Smith .ve 2654a7e14dcfSSatish Balay 265565ba42b6SBarry Smith .seealso: `Tao`, `TaoSolve()`, `TaoSetConvergenceHistory()` 2656a7e14dcfSSatish Balay @*/ 2657d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConvergenceHistory(Tao tao, PetscReal **obj, PetscReal **resid, PetscReal **cnorm, PetscInt **lits, PetscInt *nhist) 2658d71ae5a4SJacob Faibussowitsch { 2659a7e14dcfSSatish Balay PetscFunctionBegin; 2660441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2661a7e14dcfSSatish Balay if (obj) *obj = tao->hist_obj; 2662a7e14dcfSSatish Balay if (cnorm) *cnorm = tao->hist_cnorm; 2663a7e14dcfSSatish Balay if (resid) *resid = tao->hist_resid; 26645b494b05SBarry Smith if (lits) *lits = tao->hist_lits; 2665a7e14dcfSSatish Balay if (nhist) *nhist = tao->hist_len; 2666a7e14dcfSSatish Balay PetscFunctionReturn(0); 2667a7e14dcfSSatish Balay } 2668a7e14dcfSSatish Balay 2669a7e14dcfSSatish Balay /*@ 2670a7e14dcfSSatish Balay TaoSetApplicationContext - Sets the optional user-defined context for 2671a7e14dcfSSatish Balay a solver. 2672a7e14dcfSSatish Balay 2673c3339decSBarry Smith Logically Collective 2674a7e14dcfSSatish Balay 2675a7e14dcfSSatish Balay Input Parameters: 2676441846f8SBarry Smith + tao - the Tao context 2677a7e14dcfSSatish Balay - usrP - optional user context 2678a7e14dcfSSatish Balay 2679a7e14dcfSSatish Balay Level: intermediate 2680a7e14dcfSSatish Balay 268165ba42b6SBarry Smith .seealso: `Tao`, `TaoGetApplicationContext()`, `TaoSetApplicationContext()` 2682a7e14dcfSSatish Balay @*/ 2683d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetApplicationContext(Tao tao, void *usrP) 2684d71ae5a4SJacob Faibussowitsch { 2685a7e14dcfSSatish Balay PetscFunctionBegin; 2686441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2687a7e14dcfSSatish Balay tao->user = usrP; 2688a7e14dcfSSatish Balay PetscFunctionReturn(0); 2689a7e14dcfSSatish Balay } 2690a7e14dcfSSatish Balay 2691a7e14dcfSSatish Balay /*@ 2692a7e14dcfSSatish Balay TaoGetApplicationContext - Gets the user-defined context for a 269365ba42b6SBarry Smith Tao solvers. 2694a7e14dcfSSatish Balay 2695a7e14dcfSSatish Balay Not Collective 2696a7e14dcfSSatish Balay 2697a7e14dcfSSatish Balay Input Parameter: 2698441846f8SBarry Smith . tao - Tao context 2699a7e14dcfSSatish Balay 2700a7e14dcfSSatish Balay Output Parameter: 2701a7e14dcfSSatish Balay . usrP - user context 2702a7e14dcfSSatish Balay 2703a7e14dcfSSatish Balay Level: intermediate 2704a7e14dcfSSatish Balay 2705db781477SPatrick Sanan .seealso: `TaoSetApplicationContext()` 2706a7e14dcfSSatish Balay @*/ 2707d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetApplicationContext(Tao tao, void *usrP) 2708d71ae5a4SJacob Faibussowitsch { 2709a7e14dcfSSatish Balay PetscFunctionBegin; 2710441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 271194511df7SStefano Zampini PetscValidPointer(usrP, 2); 2712a7e14dcfSSatish Balay *(void **)usrP = tao->user; 2713a7e14dcfSSatish Balay PetscFunctionReturn(0); 2714a7e14dcfSSatish Balay } 2715a9603a14SPatrick Farrell 2716a9603a14SPatrick Farrell /*@ 271765ba42b6SBarry Smith TaoSetGradientNorm - Sets the matrix used to define the norm that measures the size of the gradient. 2718a9603a14SPatrick Farrell 2719c3339decSBarry Smith Collective 2720a9603a14SPatrick Farrell 2721a9603a14SPatrick Farrell Input Parameters: 2722a9603a14SPatrick Farrell + tao - the Tao context 272365ba42b6SBarry Smith - M - matrix that defines the norm 2724a9603a14SPatrick Farrell 2725a9603a14SPatrick Farrell Level: beginner 2726a9603a14SPatrick Farrell 272765ba42b6SBarry Smith .seealso: `Tao`, `TaoGetGradientNorm()`, `TaoGradientNorm()` 2728a9603a14SPatrick Farrell @*/ 2729d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetGradientNorm(Tao tao, Mat M) 2730d71ae5a4SJacob Faibussowitsch { 2731a9603a14SPatrick Farrell PetscFunctionBegin; 2732a9603a14SPatrick Farrell PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 273394511df7SStefano Zampini PetscValidHeaderSpecific(M, MAT_CLASSID, 2); 27349566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)M)); 27359566063dSJacob Faibussowitsch PetscCall(MatDestroy(&tao->gradient_norm)); 27369566063dSJacob Faibussowitsch PetscCall(VecDestroy(&tao->gradient_norm_tmp)); 2737a9603a14SPatrick Farrell tao->gradient_norm = M; 27389566063dSJacob Faibussowitsch PetscCall(MatCreateVecs(M, NULL, &tao->gradient_norm_tmp)); 2739a9603a14SPatrick Farrell PetscFunctionReturn(0); 2740a9603a14SPatrick Farrell } 2741a9603a14SPatrick Farrell 2742a9603a14SPatrick Farrell /*@ 274365ba42b6SBarry Smith TaoGetGradientNorm - Returns the matrix used to define the norm used for measuring the size of the gradient. 2744a9603a14SPatrick Farrell 2745a9603a14SPatrick Farrell Not Collective 2746a9603a14SPatrick Farrell 2747a9603a14SPatrick Farrell Input Parameter: 2748a9603a14SPatrick Farrell . tao - Tao context 2749a9603a14SPatrick Farrell 2750a9603a14SPatrick Farrell Output Parameter: 2751a9603a14SPatrick Farrell . M - gradient norm 2752a9603a14SPatrick Farrell 2753a9603a14SPatrick Farrell Level: beginner 2754a9603a14SPatrick Farrell 275565ba42b6SBarry Smith .seealso: `Tao`, `TaoSetGradientNorm()`, `TaoGradientNorm()` 2756a9603a14SPatrick Farrell @*/ 2757d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetGradientNorm(Tao tao, Mat *M) 2758d71ae5a4SJacob Faibussowitsch { 2759a9603a14SPatrick Farrell PetscFunctionBegin; 2760a9603a14SPatrick Farrell PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 276194511df7SStefano Zampini PetscValidPointer(M, 2); 2762a9603a14SPatrick Farrell *M = tao->gradient_norm; 2763a9603a14SPatrick Farrell PetscFunctionReturn(0); 2764a9603a14SPatrick Farrell } 2765a9603a14SPatrick Farrell 276605aca2cdSSatish Balay /*@C 276765ba42b6SBarry Smith TaoGradientNorm - Compute the norm with respect to the norm the user has set. 2768a9603a14SPatrick Farrell 2769c3339decSBarry Smith Collective 2770a9603a14SPatrick Farrell 2771d8d19677SJose E. Roman Input Parameters: 27722653c764SPierre Jolivet + tao - the Tao context 2773a9603a14SPatrick Farrell . gradient - the gradient to be computed 27742653c764SPierre Jolivet - norm - the norm type 2775a9603a14SPatrick Farrell 2776a9603a14SPatrick Farrell Output Parameter: 2777a9603a14SPatrick Farrell . gnorm - the gradient norm 2778a9603a14SPatrick Farrell 2779a9603a14SPatrick Farrell Level: developer 2780a9603a14SPatrick Farrell 278165ba42b6SBarry Smith .seealso: `Tao`, `TaoSetGradientNorm()`, `TaoGetGradientNorm()` 2782a9603a14SPatrick Farrell @*/ 2783d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGradientNorm(Tao tao, Vec gradient, NormType type, PetscReal *gnorm) 2784d71ae5a4SJacob Faibussowitsch { 2785a9603a14SPatrick Farrell PetscFunctionBegin; 27868854b543SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 27878854b543SStefano Zampini PetscValidHeaderSpecific(gradient, VEC_CLASSID, 2); 27888854b543SStefano Zampini PetscValidLogicalCollectiveEnum(tao, type, 3); 278994511df7SStefano Zampini PetscValidRealPointer(gnorm, 4); 2790a9603a14SPatrick Farrell if (tao->gradient_norm) { 2791680e2bc7SPatrick Farrell PetscScalar gnorms; 2792680e2bc7SPatrick Farrell 27933c859ba3SBarry Smith PetscCheck(type == NORM_2, PetscObjectComm((PetscObject)gradient), PETSC_ERR_ARG_WRONG, "Norm type must be NORM_2 if an inner product for the gradient norm is set."); 27949566063dSJacob Faibussowitsch PetscCall(MatMult(tao->gradient_norm, gradient, tao->gradient_norm_tmp)); 27959566063dSJacob Faibussowitsch PetscCall(VecDot(gradient, tao->gradient_norm_tmp, &gnorms)); 27967bb79937SPatrick Farrell *gnorm = PetscRealPart(PetscSqrtScalar(gnorms)); 2797a9603a14SPatrick Farrell } else { 27989566063dSJacob Faibussowitsch PetscCall(VecNorm(gradient, type, gnorm)); 2799a9603a14SPatrick Farrell } 2800a9603a14SPatrick Farrell PetscFunctionReturn(0); 2801a9603a14SPatrick Farrell } 2802a9603a14SPatrick Farrell 2803e882e171SHong Zhang /*@C 280465ba42b6SBarry Smith TaoMonitorDrawCtxCreate - Creates the monitor context `TaoMonitorDrawSolution()` 2805a9603a14SPatrick Farrell 2806c3339decSBarry Smith Collective 2807e882e171SHong Zhang 2808d5b43468SJose E. Roman Output Parameter: 2809e882e171SHong Zhang . ctx - the monitor context 2810e882e171SHong Zhang 28113c7db156SBarry Smith Options Database Key: 2812e882e171SHong Zhang . -tao_draw_solution_initial - show initial guess as well as current solution 2813e882e171SHong Zhang 2814e882e171SHong Zhang Level: intermediate 2815e882e171SHong Zhang 281665ba42b6SBarry Smith .seealso: `Tao`, `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawCtx()` 2817e882e171SHong Zhang @*/ 2818d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDrawCtxCreate(MPI_Comm comm, const char host[], const char label[], int x, int y, int m, int n, PetscInt howoften, TaoMonitorDrawCtx *ctx) 2819d71ae5a4SJacob Faibussowitsch { 2820e882e171SHong Zhang PetscFunctionBegin; 28219566063dSJacob Faibussowitsch PetscCall(PetscNew(ctx)); 28229566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawOpen(comm, host, label, x, y, m, n, &(*ctx)->viewer)); 28239566063dSJacob Faibussowitsch PetscCall(PetscViewerSetFromOptions((*ctx)->viewer)); 2824e882e171SHong Zhang (*ctx)->howoften = howoften; 2825e882e171SHong Zhang PetscFunctionReturn(0); 2826e882e171SHong Zhang } 2827e882e171SHong Zhang 2828e882e171SHong Zhang /*@C 282965ba42b6SBarry Smith TaoMonitorDrawCtxDestroy - Destroys the monitor context for `TaoMonitorDrawSolution()` 2830e882e171SHong Zhang 2831c3339decSBarry Smith Collective 2832e882e171SHong Zhang 2833e882e171SHong Zhang Input Parameters: 2834e882e171SHong Zhang . ctx - the monitor context 2835e882e171SHong Zhang 2836e882e171SHong Zhang Level: intermediate 2837e882e171SHong Zhang 2838db781477SPatrick Sanan .seealso: `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawSolution()` 2839e882e171SHong Zhang @*/ 2840d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *ictx) 2841d71ae5a4SJacob Faibussowitsch { 2842e882e171SHong Zhang PetscFunctionBegin; 28439566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&(*ictx)->viewer)); 28449566063dSJacob Faibussowitsch PetscCall(PetscFree(*ictx)); 2845e882e171SHong Zhang PetscFunctionReturn(0); 2846e882e171SHong Zhang } 2847