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; 293ba16761SJacob Faibussowitsch if (!snes_ewdummy) PetscFunctionReturn(PETSC_SUCCESS); 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; 373ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 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; 453ba16761SJacob Faibussowitsch if (!snes_ewdummy) PetscFunctionReturn(PETSC_SUCCESS); 460f0abf79SStefano Zampini PetscCall(KSPPostSolve_SNESEW(ksp, b, x, snes_ewdummy)); 473ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 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; 563ba16761SJacob Faibussowitsch if (!tao->ksp) PetscFunctionReturn(PETSC_SUCCESS); 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; 65a4598233SStefano Zampini PetscCall(SNESEWSetFromOptions_Private(kctx, PETSC_FALSE, PetscObjectComm((PetscObject)tao), ewprefix)); 660f0abf79SStefano Zampini } else PetscCall(SNESDestroy(&tao->snes_ewdummy)); 673ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 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: 7947450a7bSBarry Smith . newtao - the new `Tao` context 80a7e14dcfSSatish Balay 8147450a7bSBarry Smith Options Database Key: 8265ba42b6SBarry Smith . -tao_type - select which method Tao should use 83a7e14dcfSSatish Balay 84a7e14dcfSSatish Balay Level: beginner 85a7e14dcfSSatish Balay 861cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoDestroy()`, `TAOSetFromOptions()`, `TAOSetType()` 87a7e14dcfSSatish Balay @*/ 88d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoCreate(MPI_Comm comm, Tao *newtao) 89d71ae5a4SJacob Faibussowitsch { 90441846f8SBarry Smith Tao tao; 91a7e14dcfSSatish Balay 92a7e14dcfSSatish Balay PetscFunctionBegin; 934f572ea9SToby Isaac PetscAssertPointer(newtao, 2); 949566063dSJacob Faibussowitsch PetscCall(TaoInitializePackage()); 959566063dSJacob Faibussowitsch PetscCall(TaoLineSearchInitializePackage()); 969566063dSJacob Faibussowitsch PetscCall(PetscHeaderCreate(tao, TAO_CLASSID, "Tao", "Optimization solver", "Tao", comm, TaoDestroy, TaoView)); 97a7e14dcfSSatish Balay 9894511df7SStefano Zampini /* Set non-NULL defaults */ 9994511df7SStefano Zampini tao->ops->convergencetest = TaoDefaultConvergenceTest; 100a7e14dcfSSatish Balay 101a7e14dcfSSatish Balay tao->max_it = 10000; 10294511df7SStefano Zampini tao->max_funcs = -1; 1036f4723b1SBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 1046f4723b1SBarry Smith tao->gatol = 1e-5; 1056f4723b1SBarry Smith tao->grtol = 1e-5; 1066f1f5f59SAlp Dener tao->crtol = 1e-5; 1076f1f5f59SAlp Dener tao->catol = 1e-5; 1086f4723b1SBarry Smith #else 109a7e14dcfSSatish Balay tao->gatol = 1e-8; 110a7e14dcfSSatish Balay tao->grtol = 1e-8; 1116f1f5f59SAlp Dener tao->crtol = 1e-8; 1126f1f5f59SAlp Dener tao->catol = 1e-8; 1136f4723b1SBarry Smith #endif 1146552cf8aSJason Sarich tao->gttol = 0.0; 1154f1535bcSTodd Munson tao->steptol = 0.0; 116e270355aSBarry Smith tao->trust0 = PETSC_INFINITY; 1176f4723b1SBarry Smith tao->fmin = PETSC_NINFINITY; 1189fa2b5dcSStefano Zampini 119a7e14dcfSSatish Balay tao->hist_reset = PETSC_TRUE; 120a7e14dcfSSatish Balay 1219566063dSJacob Faibussowitsch PetscCall(TaoResetStatistics(tao)); 122a7e14dcfSSatish Balay *newtao = tao; 1233ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 124a7e14dcfSSatish Balay } 125a7e14dcfSSatish Balay 126a7e14dcfSSatish Balay /*@ 127a7e14dcfSSatish Balay TaoSolve - Solves an optimization problem min F(x) s.t. l <= x <= u 128a7e14dcfSSatish Balay 129c3339decSBarry Smith Collective 130a7e14dcfSSatish Balay 13147450a7bSBarry Smith Input Parameter: 13247450a7bSBarry Smith . tao - the `Tao` context 133a7e14dcfSSatish Balay 13467be906fSBarry Smith Level: beginner 13567be906fSBarry Smith 136a7e14dcfSSatish Balay Notes: 13747450a7bSBarry Smith The user must set up the `Tao` object with calls to `TaoSetSolution()`, `TaoSetObjective()`, `TaoSetGradient()`, and (if using 2nd order method) `TaoSetHessian()`. 138a7e14dcfSSatish Balay 13967be906fSBarry Smith You should call `TaoGetConvergedReason()` or run with `-tao_converged_reason` to determine if the optimization algorithm actually succeeded or 140a35d58b8SBarry Smith why it failed. 141a35d58b8SBarry Smith 1421cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSetObjective()`, `TaoSetGradient()`, `TaoSetHessian()`, `TaoGetConvergedReason()`, `TaoSetUp()` 143a7e14dcfSSatish Balay @*/ 144d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSolve(Tao tao) 145d71ae5a4SJacob Faibussowitsch { 146e2379f4fSBarry Smith static PetscBool set = PETSC_FALSE; 14747a47007SBarry Smith 148a7e14dcfSSatish Balay PetscFunctionBegin; 149441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 150d0609cedSBarry Smith PetscCall(PetscCitationsRegister("@TechReport{tao-user-ref,\n" 151e2379f4fSBarry Smith "title = {Toolkit for Advanced Optimization (TAO) Users Manual},\n" 152e2379f4fSBarry Smith "author = {Todd Munson and Jason Sarich and Stefan Wild and Steve Benson and Lois Curfman McInnes},\n" 153e2379f4fSBarry Smith "Institution = {Argonne National Laboratory},\n" 154e2379f4fSBarry Smith "Year = 2014,\n" 155e2379f4fSBarry Smith "Number = {ANL/MCS-TM-322 - Revision 3.5},\n" 1569371c9d4SSatish Balay "url = {https://www.mcs.anl.gov/research/projects/tao/}\n}\n", 1579371c9d4SSatish Balay &set)); 158494bef23SAlp Dener tao->header_printed = PETSC_FALSE; 1599566063dSJacob Faibussowitsch PetscCall(TaoSetUp(tao)); 1609566063dSJacob Faibussowitsch PetscCall(TaoResetStatistics(tao)); 1611baa6e33SBarry Smith if (tao->linesearch) PetscCall(TaoLineSearchReset(tao->linesearch)); 162a7e14dcfSSatish Balay 1639566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(TAO_Solve, tao, 0, 0, 0)); 164dbbe0bcdSBarry Smith PetscTryTypeMethod(tao, solve); 1659566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(TAO_Solve, tao, 0, 0, 0)); 166a7e14dcfSSatish Balay 1679566063dSJacob Faibussowitsch PetscCall(VecViewFromOptions(tao->solution, (PetscObject)tao, "-tao_view_solution")); 1687cf37e64SBarry Smith 1698931d482SJason Sarich tao->ntotalits += tao->niter; 170a7e14dcfSSatish Balay 171a7e14dcfSSatish Balay if (tao->printreason) { 172f642d338SBarry Smith PetscViewer viewer = PETSC_VIEWER_STDOUT_(((PetscObject)tao)->comm); 173f642d338SBarry Smith PetscCall(PetscViewerASCIIAddTab(viewer, ((PetscObject)tao)->tablevel)); 174a7e14dcfSSatish Balay if (tao->reason > 0) { 175f642d338SBarry 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)); 176a7e14dcfSSatish Balay } else { 177f642d338SBarry 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)); 178a7e14dcfSSatish Balay } 179f642d338SBarry Smith PetscCall(PetscViewerASCIISubtractTab(viewer, ((PetscObject)tao)->tablevel)); 180a7e14dcfSSatish Balay } 181f642d338SBarry Smith PetscCall(TaoViewFromOptions(tao, NULL, "-tao_view")); 1823ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 183a7e14dcfSSatish Balay } 184a7e14dcfSSatish Balay 185a7e14dcfSSatish Balay /*@ 186a7e14dcfSSatish Balay TaoSetUp - Sets up the internal data structures for the later use 187a7e14dcfSSatish Balay of a Tao solver 188a7e14dcfSSatish Balay 189c3339decSBarry Smith Collective 190a7e14dcfSSatish Balay 19147450a7bSBarry Smith Input Parameter: 19247450a7bSBarry Smith . tao - the `Tao` context 193a7e14dcfSSatish Balay 19467be906fSBarry Smith Level: advanced 19567be906fSBarry Smith 19647450a7bSBarry Smith Note: 19765ba42b6SBarry Smith The user will not need to explicitly call `TaoSetUp()`, as it will 19865ba42b6SBarry Smith automatically be called in `TaoSolve()`. However, if the user 19965ba42b6SBarry Smith desires to call it explicitly, it should come after `TaoCreate()` 20065ba42b6SBarry Smith and any TaoSetSomething() routines, but before `TaoSolve()`. 201a7e14dcfSSatish Balay 2021cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSolve()` 203a7e14dcfSSatish Balay @*/ 204d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetUp(Tao tao) 205d71ae5a4SJacob Faibussowitsch { 206a7e14dcfSSatish Balay PetscFunctionBegin; 207441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2083ba16761SJacob Faibussowitsch if (tao->setupcalled) PetscFunctionReturn(PETSC_SUCCESS); 2090f0abf79SStefano Zampini PetscCall(TaoSetUpEW_Private(tao)); 2103c859ba3SBarry Smith PetscCheck(tao->solution, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_WRONGSTATE, "Must call TaoSetSolution"); 211dbbe0bcdSBarry Smith PetscTryTypeMethod(tao, setup); 212a7e14dcfSSatish Balay tao->setupcalled = PETSC_TRUE; 2133ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 214a7e14dcfSSatish Balay } 215a7e14dcfSSatish Balay 2161fb7b255SJunchao Zhang /*@C 21747450a7bSBarry Smith TaoDestroy - Destroys the `Tao` context that was created with `TaoCreate()` 218a7e14dcfSSatish Balay 219c3339decSBarry Smith Collective 220a7e14dcfSSatish Balay 221a7e14dcfSSatish Balay Input Parameter: 22247450a7bSBarry Smith . tao - the `Tao` context 223a7e14dcfSSatish Balay 224a7e14dcfSSatish Balay Level: beginner 225a7e14dcfSSatish Balay 2261cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSolve()` 227a7e14dcfSSatish Balay @*/ 228d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDestroy(Tao *tao) 229d71ae5a4SJacob Faibussowitsch { 230a7e14dcfSSatish Balay PetscFunctionBegin; 2313ba16761SJacob Faibussowitsch if (!*tao) PetscFunctionReturn(PETSC_SUCCESS); 232441846f8SBarry Smith PetscValidHeaderSpecific(*tao, TAO_CLASSID, 1); 2339371c9d4SSatish Balay if (--((PetscObject)*tao)->refct > 0) { 2349371c9d4SSatish Balay *tao = NULL; 2353ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 236a7e14dcfSSatish Balay } 2379371c9d4SSatish Balay 23848a46eb9SPierre Jolivet if ((*tao)->ops->destroy) PetscCall((*((*tao))->ops->destroy)(*tao)); 2399566063dSJacob Faibussowitsch PetscCall(KSPDestroy(&(*tao)->ksp)); 2400f0abf79SStefano Zampini PetscCall(SNESDestroy(&(*tao)->snes_ewdummy)); 2419566063dSJacob Faibussowitsch PetscCall(TaoLineSearchDestroy(&(*tao)->linesearch)); 242a7e14dcfSSatish Balay 243a7e14dcfSSatish Balay if ((*tao)->ops->convergencedestroy) { 2449566063dSJacob Faibussowitsch PetscCall((*(*tao)->ops->convergencedestroy)((*tao)->cnvP)); 24548a46eb9SPierre Jolivet if ((*tao)->jacobian_state_inv) PetscCall(MatDestroy(&(*tao)->jacobian_state_inv)); 246a7e14dcfSSatish Balay } 2479566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->solution)); 2489566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->gradient)); 2499566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->ls_res)); 250a7e14dcfSSatish Balay 251a9603a14SPatrick Farrell if ((*tao)->gradient_norm) { 2529566063dSJacob Faibussowitsch PetscCall(PetscObjectDereference((PetscObject)(*tao)->gradient_norm)); 2539566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->gradient_norm_tmp)); 254a9603a14SPatrick Farrell } 255a9603a14SPatrick Farrell 2569566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->XL)); 2579566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->XU)); 2589566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->IL)); 2599566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->IU)); 2609566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->DE)); 2619566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->DI)); 2629566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->constraints)); 2639566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->constraints_equality)); 2649566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->constraints_inequality)); 2659566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->stepdirection)); 2669566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->hessian_pre)); 2679566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->hessian)); 2689566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->ls_jac)); 2699566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->ls_jac_pre)); 2709566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_pre)); 2719566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian)); 2729566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_state_pre)); 2739566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_state)); 2749566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_state_inv)); 2759566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_design)); 2769566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_equality)); 2779566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_equality_pre)); 2789566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_inequality)); 2799566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_inequality_pre)); 2809566063dSJacob Faibussowitsch PetscCall(ISDestroy(&(*tao)->state_is)); 2819566063dSJacob Faibussowitsch PetscCall(ISDestroy(&(*tao)->design_is)); 2829566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->res_weights_v)); 283*b2dc45ddSPierre Jolivet PetscCall(TaoMonitorCancel(*tao)); 28448a46eb9SPierre Jolivet if ((*tao)->hist_malloc) PetscCall(PetscFree4((*tao)->hist_obj, (*tao)->hist_resid, (*tao)->hist_cnorm, (*tao)->hist_lits)); 285737f463aSAlp Dener if ((*tao)->res_weights_n) { 2869566063dSJacob Faibussowitsch PetscCall(PetscFree((*tao)->res_weights_rows)); 2879566063dSJacob Faibussowitsch PetscCall(PetscFree((*tao)->res_weights_cols)); 2889566063dSJacob Faibussowitsch PetscCall(PetscFree((*tao)->res_weights_w)); 289125ddc8aSJason Sarich } 2909566063dSJacob Faibussowitsch PetscCall(PetscHeaderDestroy(tao)); 2913ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 292a7e14dcfSSatish Balay } 293a7e14dcfSSatish Balay 294a7e14dcfSSatish Balay /*@ 2951d27aa22SBarry Smith TaoKSPSetUseEW - Sets `SNES` to use Eisenstat-Walker method {cite}`ew96`for computing relative tolerance for linear solvers. 2960f0abf79SStefano Zampini 297c3339decSBarry Smith Logically Collective 2980f0abf79SStefano Zampini 2990f0abf79SStefano Zampini Input Parameters: 3000f0abf79SStefano Zampini + tao - Tao context 30165ba42b6SBarry Smith - flag - `PETSC_TRUE` or `PETSC_FALSE` 3020f0abf79SStefano Zampini 30367be906fSBarry Smith Level: advanced 30467be906fSBarry Smith 30547450a7bSBarry Smith Note: 30665ba42b6SBarry Smith See `SNESKSPSetUseEW()` for customization details. 3070f0abf79SStefano Zampini 3081cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `SNESKSPSetUseEW()` 3090f0abf79SStefano Zampini @*/ 310d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoKSPSetUseEW(Tao tao, PetscBool flag) 311d71ae5a4SJacob Faibussowitsch { 3120f0abf79SStefano Zampini PetscFunctionBegin; 3130f0abf79SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 3140f0abf79SStefano Zampini PetscValidLogicalCollectiveBool(tao, flag, 2); 3150f0abf79SStefano Zampini tao->ksp_ewconv = flag; 3163ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3170f0abf79SStefano Zampini } 3180f0abf79SStefano Zampini 3190f0abf79SStefano Zampini /*@ 32047450a7bSBarry Smith TaoSetFromOptions - Sets various Tao parameters from the options database 321a7e14dcfSSatish Balay 322c3339decSBarry Smith Collective 323a7e14dcfSSatish Balay 32401d2d390SJose E. Roman Input Parameter: 32547450a7bSBarry Smith . tao - the `Tao` solver context 326a7e14dcfSSatish Balay 32747450a7bSBarry Smith Options Database Keys: 32865ba42b6SBarry Smith + -tao_type <type> - The algorithm that Tao uses (lmvm, nls, etc.) 329a7e14dcfSSatish Balay . -tao_gatol <gatol> - absolute error tolerance for ||gradient|| 330a7e14dcfSSatish Balay . -tao_grtol <grtol> - relative error tolerance for ||gradient|| 331a7e14dcfSSatish Balay . -tao_gttol <gttol> - reduction of ||gradient|| relative to initial gradient 332a7e14dcfSSatish Balay . -tao_max_it <max> - sets maximum number of iterations 333a7e14dcfSSatish Balay . -tao_max_funcs <max> - sets maximum number of function evaluations 334a7e14dcfSSatish Balay . -tao_fmin <fmin> - stop if function value reaches fmin 335a7e14dcfSSatish Balay . -tao_steptol <tol> - stop if trust region radius less than <tol> 336a7e14dcfSSatish Balay . -tao_trust0 <t> - initial trust region radius 33747450a7bSBarry Smith . -tao_monitor - prints function value and residual norm at each iteration 338a7e14dcfSSatish Balay . -tao_smonitor - same as tao_monitor, but truncates very small values 339a7e14dcfSSatish Balay . -tao_cmonitor - prints function value, residual, and constraint norm at each iteration 340a7e14dcfSSatish Balay . -tao_view_solution - prints solution vector at each iteration 3414a48860cSAlp Dener . -tao_view_ls_residual - prints least-squares residual vector at each iteration 342147403d9SBarry Smith . -tao_view_stepdirection - prints step direction vector at each iteration 343a7e14dcfSSatish Balay . -tao_view_gradient - prints gradient vector at each iteration 344a7e14dcfSSatish Balay . -tao_draw_solution - graphically view solution vector at each iteration 345a7e14dcfSSatish Balay . -tao_draw_step - graphically view step vector at each iteration 346a7e14dcfSSatish Balay . -tao_draw_gradient - graphically view gradient at each iteration 347a7e14dcfSSatish Balay . -tao_fd_gradient - use gradient computed with finite differences 348f4c1ad5cSStefano Zampini . -tao_fd_hessian - use hessian computed with finite differences 349f4c1ad5cSStefano Zampini . -tao_mf_hessian - use matrix-free hessian computed with finite differences 350*b2dc45ddSPierre Jolivet . -tao_monitor_cancel - cancels all monitors (except those set with command line) 351441846f8SBarry Smith . -tao_view - prints information about the Tao after solving 35265ba42b6SBarry Smith - -tao_converged_reason - prints the reason Tao stopped iterating 353a7e14dcfSSatish Balay 35467be906fSBarry Smith Level: beginner 35567be906fSBarry Smith 35667be906fSBarry Smith Note: 35747450a7bSBarry Smith To see all options, run your program with the `-help` option or consult the 35865ba42b6SBarry Smith user's manual. Should be called after `TaoCreate()` but before `TaoSolve()` 359a7e14dcfSSatish Balay 3601cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSolve()` 361a7e14dcfSSatish Balay @*/ 362d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetFromOptions(Tao tao) 363d71ae5a4SJacob Faibussowitsch { 364b625d6c7SJed Brown TaoType default_type = TAOLMVM; 365a7e14dcfSSatish Balay char type[256], monfilename[PETSC_MAX_PATH_LEN]; 366a7e14dcfSSatish Balay PetscViewer monviewer; 367a7e14dcfSSatish Balay PetscBool flg; 368a7e14dcfSSatish Balay MPI_Comm comm; 369a7e14dcfSSatish Balay 370a7e14dcfSSatish Balay PetscFunctionBegin; 371441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 3729566063dSJacob Faibussowitsch PetscCall(PetscObjectGetComm((PetscObject)tao, &comm)); 373125ddc8aSJason Sarich 37460b70c5cSStefano Zampini if (((PetscObject)tao)->type_name) default_type = ((PetscObject)tao)->type_name; 37560b70c5cSStefano Zampini 376d0609cedSBarry Smith PetscObjectOptionsBegin((PetscObject)tao); 377a7e14dcfSSatish Balay /* Check for type from options */ 3789566063dSJacob Faibussowitsch PetscCall(PetscOptionsFList("-tao_type", "Tao Solver type", "TaoSetType", TaoList, default_type, type, 256, &flg)); 379a7e14dcfSSatish Balay if (flg) { 3809566063dSJacob Faibussowitsch PetscCall(TaoSetType(tao, type)); 381a7e14dcfSSatish Balay } else if (!((PetscObject)tao)->type_name) { 3829566063dSJacob Faibussowitsch PetscCall(TaoSetType(tao, default_type)); 383a7e14dcfSSatish Balay } 384a7e14dcfSSatish Balay 38560b70c5cSStefano Zampini /* Tao solvers do not set the prefix, set it here if not yet done 38660b70c5cSStefano Zampini We do it after SetType since solver may have been changed */ 38760b70c5cSStefano Zampini if (tao->linesearch) { 38860b70c5cSStefano Zampini const char *prefix; 38960b70c5cSStefano Zampini PetscCall(TaoLineSearchGetOptionsPrefix(tao->linesearch, &prefix)); 39060b70c5cSStefano Zampini if (!prefix) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch, ((PetscObject)(tao))->prefix)); 39160b70c5cSStefano Zampini } 39260b70c5cSStefano Zampini 3939566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_catol", "Stop if constraints violations within", "TaoSetConstraintTolerances", tao->catol, &tao->catol, &flg)); 3946552cf8aSJason Sarich if (flg) tao->catol_changed = PETSC_TRUE; 3956aad120cSJose E. Roman PetscCall(PetscOptionsReal("-tao_crtol", "Stop if relative constraint violations within", "TaoSetConstraintTolerances", tao->crtol, &tao->crtol, &flg)); 3966552cf8aSJason Sarich if (flg) tao->crtol_changed = PETSC_TRUE; 3979566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_gatol", "Stop if norm of gradient less than", "TaoSetTolerances", tao->gatol, &tao->gatol, &flg)); 3986552cf8aSJason Sarich if (flg) tao->gatol_changed = PETSC_TRUE; 3999566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_grtol", "Stop if norm of gradient divided by the function value is less than", "TaoSetTolerances", tao->grtol, &tao->grtol, &flg)); 4006552cf8aSJason Sarich if (flg) tao->grtol_changed = PETSC_TRUE; 4019566063dSJacob 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)); 4026552cf8aSJason Sarich if (flg) tao->gttol_changed = PETSC_TRUE; 4039566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-tao_max_it", "Stop if iteration number exceeds", "TaoSetMaximumIterations", tao->max_it, &tao->max_it, &flg)); 4046552cf8aSJason Sarich if (flg) tao->max_it_changed = PETSC_TRUE; 4059566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-tao_max_funcs", "Stop if number of function evaluations exceeds", "TaoSetMaximumFunctionEvaluations", tao->max_funcs, &tao->max_funcs, &flg)); 4066552cf8aSJason Sarich if (flg) tao->max_funcs_changed = PETSC_TRUE; 4079566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_fmin", "Stop if function less than", "TaoSetFunctionLowerBound", tao->fmin, &tao->fmin, &flg)); 4086552cf8aSJason Sarich if (flg) tao->fmin_changed = PETSC_TRUE; 4099566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_steptol", "Stop if step size or trust region radius less than", "", tao->steptol, &tao->steptol, &flg)); 4106552cf8aSJason Sarich if (flg) tao->steptol_changed = PETSC_TRUE; 4119566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_trust0", "Initial trust region radius", "TaoSetTrustRegionRadius", tao->trust0, &tao->trust0, &flg)); 4126552cf8aSJason Sarich if (flg) tao->trust0_changed = PETSC_TRUE; 4139566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_view_solution", "view solution vector after each evaluation", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 414a7e14dcfSSatish Balay if (flg) { 4159566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4169566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoSolutionMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 417a7e14dcfSSatish Balay } 418a7e14dcfSSatish Balay 41965ba42b6SBarry Smith PetscCall(PetscOptionsBool("-tao_converged_reason", "Print reason for Tao converged", "TaoSolve", tao->printreason, &tao->printreason, NULL)); 4209566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_view_gradient", "view gradient vector after each evaluation", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 421a7e14dcfSSatish Balay if (flg) { 4229566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4239566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoGradientMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 424a7e14dcfSSatish Balay } 425a7e14dcfSSatish Balay 4269566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_view_stepdirection", "view step direction vector after each iteration", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 427a7e14dcfSSatish Balay if (flg) { 4289566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4299566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoStepDirectionMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 430a7e14dcfSSatish Balay } 431a7e14dcfSSatish Balay 4329566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_view_residual", "view least-squares residual 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, TaoResidualMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 436a7e14dcfSSatish Balay } 437a7e14dcfSSatish Balay 4389566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_monitor", "Use the default convergence monitor", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 439a7e14dcfSSatish Balay if (flg) { 4409566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4419566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoMonitorDefault, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 442a7e14dcfSSatish Balay } 443a7e14dcfSSatish Balay 4449566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_gmonitor", "Use the convergence monitor with extra globalization info", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 4458d5ead36SAlp Dener if (flg) { 4469566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4479566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoDefaultGMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 4488d5ead36SAlp Dener } 4498d5ead36SAlp Dener 4509566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_smonitor", "Use the short convergence monitor", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 451a7e14dcfSSatish Balay if (flg) { 4529566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4539566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoDefaultSMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 454a7e14dcfSSatish Balay } 455a7e14dcfSSatish Balay 4569566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_cmonitor", "Use the default convergence monitor with constraint norm", "TaoSetMonitor", "stdout", monfilename, sizeof(monfilename), &flg)); 457a7e14dcfSSatish Balay if (flg) { 4589566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 4599566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoDefaultCMonitor, monviewer, (PetscErrorCode(*)(void **))PetscViewerDestroy)); 460a7e14dcfSSatish Balay } 461a7e14dcfSSatish Balay 4628afaa268SBarry Smith flg = PETSC_FALSE; 463*b2dc45ddSPierre Jolivet PetscCall(PetscOptionsDeprecated("-tao_cancelmonitors", "-tao_monitor_cancel", "3.21", NULL)); 464*b2dc45ddSPierre Jolivet PetscCall(PetscOptionsBool("-tao_monitor_cancel", "cancel all monitors and call any registered destroy routines", "TaoMonitorCancel", flg, &flg, NULL)); 465*b2dc45ddSPierre Jolivet if (flg) PetscCall(TaoMonitorCancel(tao)); 466a7e14dcfSSatish Balay 4678afaa268SBarry Smith flg = PETSC_FALSE; 4689566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_draw_solution", "Plot solution vector at each iteration", "TaoSetMonitor", flg, &flg, NULL)); 469a7e14dcfSSatish Balay if (flg) { 470e882e171SHong Zhang TaoMonitorDrawCtx drawctx; 471e882e171SHong Zhang PetscInt howoften = 1; 4729566063dSJacob Faibussowitsch PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao), NULL, NULL, PETSC_DECIDE, PETSC_DECIDE, 300, 300, howoften, &drawctx)); 4739566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoDrawSolutionMonitor, drawctx, (PetscErrorCode(*)(void **))TaoMonitorDrawCtxDestroy)); 474a7e14dcfSSatish Balay } 475a7e14dcfSSatish Balay 4768afaa268SBarry Smith flg = PETSC_FALSE; 4779566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_draw_step", "plots step direction at each iteration", "TaoSetMonitor", flg, &flg, NULL)); 4781baa6e33SBarry Smith if (flg) PetscCall(TaoSetMonitor(tao, TaoDrawStepMonitor, NULL, NULL)); 479a7e14dcfSSatish Balay 4808afaa268SBarry Smith flg = PETSC_FALSE; 4819566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_draw_gradient", "plots gradient at each iteration", "TaoSetMonitor", flg, &flg, NULL)); 482a7e14dcfSSatish Balay if (flg) { 483e882e171SHong Zhang TaoMonitorDrawCtx drawctx; 484e882e171SHong Zhang PetscInt howoften = 1; 4859566063dSJacob Faibussowitsch PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao), NULL, NULL, PETSC_DECIDE, PETSC_DECIDE, 300, 300, howoften, &drawctx)); 4869566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao, TaoDrawGradientMonitor, drawctx, (PetscErrorCode(*)(void **))TaoMonitorDrawCtxDestroy)); 487a7e14dcfSSatish Balay } 4888afaa268SBarry Smith flg = PETSC_FALSE; 4899566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_fd_gradient", "compute gradient using finite differences", "TaoDefaultComputeGradient", flg, &flg, NULL)); 4901baa6e33SBarry Smith if (flg) PetscCall(TaoSetGradient(tao, NULL, TaoDefaultComputeGradient, NULL)); 491f4c1ad5cSStefano Zampini flg = PETSC_FALSE; 4929566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_fd_hessian", "compute hessian using finite differences", "TaoDefaultComputeHessian", flg, &flg, NULL)); 493f4c1ad5cSStefano Zampini if (flg) { 494f4c1ad5cSStefano Zampini Mat H; 495f4c1ad5cSStefano Zampini 4969566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)tao), &H)); 4979566063dSJacob Faibussowitsch PetscCall(MatSetType(H, MATAIJ)); 4989566063dSJacob Faibussowitsch PetscCall(TaoSetHessian(tao, H, H, TaoDefaultComputeHessian, NULL)); 4999566063dSJacob Faibussowitsch PetscCall(MatDestroy(&H)); 500f4c1ad5cSStefano Zampini } 501f4c1ad5cSStefano Zampini flg = PETSC_FALSE; 5029566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_mf_hessian", "compute matrix-free hessian using finite differences", "TaoDefaultComputeHessianMFFD", flg, &flg, NULL)); 503f4c1ad5cSStefano Zampini if (flg) { 504f4c1ad5cSStefano Zampini Mat H; 505f4c1ad5cSStefano Zampini 5069566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)tao), &H)); 5079566063dSJacob Faibussowitsch PetscCall(TaoSetHessian(tao, H, H, TaoDefaultComputeHessianMFFD, NULL)); 5089566063dSJacob Faibussowitsch PetscCall(MatDestroy(&H)); 509f4c1ad5cSStefano Zampini } 510414d97d3SAlp Dener flg = PETSC_FALSE; 5119566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_recycle_history", "enable recycling/re-using information from the previous TaoSolve() call for some algorithms", "TaoSetRecycleHistory", flg, &flg, NULL)); 5121baa6e33SBarry Smith if (flg) PetscCall(TaoSetRecycleHistory(tao, PETSC_TRUE)); 5139566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnum("-tao_subset_type", "subset type", "", TaoSubSetTypes, (PetscEnum)tao->subset_type, (PetscEnum *)&tao->subset_type, NULL)); 514a7e14dcfSSatish Balay 5150f0abf79SStefano Zampini if (tao->ksp) { 5160f0abf79SStefano Zampini PetscCall(PetscOptionsBool("-tao_ksp_ew", "Use Eisentat-Walker linear system convergence test", "TaoKSPSetUseEW", tao->ksp_ewconv, &tao->ksp_ewconv, NULL)); 5170f0abf79SStefano Zampini PetscCall(TaoKSPSetUseEW(tao, tao->ksp_ewconv)); 5180f0abf79SStefano Zampini } 5190f0abf79SStefano Zampini 520dbbe0bcdSBarry Smith PetscTryTypeMethod(tao, setfromoptions, PetscOptionsObject); 52160b70c5cSStefano Zampini 52260b70c5cSStefano Zampini /* process any options handlers added with PetscObjectAddOptionsHandler() */ 52360b70c5cSStefano Zampini PetscCall(PetscObjectProcessOptionsHandlers((PetscObject)tao, PetscOptionsObject)); 524d0609cedSBarry Smith PetscOptionsEnd(); 52560b70c5cSStefano Zampini 52660b70c5cSStefano Zampini if (tao->linesearch) PetscCall(TaoLineSearchSetFromOptions(tao->linesearch)); 5273ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 528a7e14dcfSSatish Balay } 529a7e14dcfSSatish Balay 530a7e14dcfSSatish Balay /*@C 53147450a7bSBarry Smith TaoViewFromOptions - View a `Tao` object based on values in the options database 532fe2efc57SMark 533c3339decSBarry Smith Collective 534fe2efc57SMark 535fe2efc57SMark Input Parameters: 53647450a7bSBarry Smith + A - the `Tao` context 53747450a7bSBarry Smith . obj - Optional object that provides the prefix for the options database 538736c3998SJose E. Roman - name - command line option 539fe2efc57SMark 540fe2efc57SMark Level: intermediate 54167be906fSBarry Smith 5421cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoView`, `PetscObjectViewFromOptions()`, `TaoCreate()` 543fe2efc57SMark @*/ 544d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoViewFromOptions(Tao A, PetscObject obj, const char name[]) 545d71ae5a4SJacob Faibussowitsch { 546fe2efc57SMark PetscFunctionBegin; 547fe2efc57SMark PetscValidHeaderSpecific(A, TAO_CLASSID, 1); 5489566063dSJacob Faibussowitsch PetscCall(PetscObjectViewFromOptions((PetscObject)A, obj, name)); 5493ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 550fe2efc57SMark } 551fe2efc57SMark 552fe2efc57SMark /*@C 55347450a7bSBarry Smith TaoView - Prints information about the `Tao` object 554a7e14dcfSSatish Balay 555c3339decSBarry Smith Collective 556a7e14dcfSSatish Balay 557a7e14dcfSSatish Balay Input Parameters: 55847450a7bSBarry Smith + tao - the `Tao` context 559a7e14dcfSSatish Balay - viewer - visualization context 560a7e14dcfSSatish Balay 561a7e14dcfSSatish Balay Options Database Key: 56265ba42b6SBarry Smith . -tao_view - Calls `TaoView()` at the end of `TaoSolve()` 563a7e14dcfSSatish Balay 56467be906fSBarry Smith Level: beginner 56567be906fSBarry Smith 566a7e14dcfSSatish Balay Notes: 567a7e14dcfSSatish Balay The available visualization contexts include 56865ba42b6SBarry Smith + `PETSC_VIEWER_STDOUT_SELF` - standard output (default) 56965ba42b6SBarry Smith - `PETSC_VIEWER_STDOUT_WORLD` - synchronized standard 570a7e14dcfSSatish Balay output where only the first processor opens 571a7e14dcfSSatish Balay the file. All other processors send their 572a7e14dcfSSatish Balay data to the first processor to print. 573a7e14dcfSSatish Balay 5741cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `PetscViewerASCIIOpen()` 575a7e14dcfSSatish Balay @*/ 576d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoView(Tao tao, PetscViewer viewer) 577d71ae5a4SJacob Faibussowitsch { 578a7e14dcfSSatish Balay PetscBool isascii, isstring; 579b625d6c7SJed Brown TaoType type; 58047a47007SBarry Smith 581a7e14dcfSSatish Balay PetscFunctionBegin; 582441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 58348a46eb9SPierre Jolivet if (!viewer) PetscCall(PetscViewerASCIIGetStdout(((PetscObject)tao)->comm, &viewer)); 584a7e14dcfSSatish Balay PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 585a7e14dcfSSatish Balay PetscCheckSameComm(tao, 1, viewer, 2); 586a7e14dcfSSatish Balay 5879566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii)); 5889566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERSTRING, &isstring)); 589a7e14dcfSSatish Balay if (isascii) { 5909566063dSJacob Faibussowitsch PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)tao, viewer)); 591a7e14dcfSSatish Balay 592a7e14dcfSSatish Balay if (tao->ops->view) { 5939566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 594dbbe0bcdSBarry Smith PetscUseTypeMethod(tao, view, viewer); 5959566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 596a7e14dcfSSatish Balay } 5978d3f3ddaSAlp Dener if (tao->linesearch) { 5989566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 5999566063dSJacob Faibussowitsch PetscCall(TaoLineSearchView(tao->linesearch, viewer)); 6009566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 6018d3f3ddaSAlp Dener } 602a7e14dcfSSatish Balay if (tao->ksp) { 6039566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 6049566063dSJacob Faibussowitsch PetscCall(KSPView(tao->ksp, viewer)); 60563a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total KSP iterations: %" PetscInt_FMT "\n", tao->ksp_tot_its)); 6069566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 607a7e14dcfSSatish Balay } 60819b5c101SAlp Dener 6099566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 61019b5c101SAlp Dener 61148a46eb9SPierre Jolivet if (tao->XL || tao->XU) PetscCall(PetscViewerASCIIPrintf(viewer, "Active Set subset type: %s\n", TaoSubSetTypes[tao->subset_type])); 612a7e14dcfSSatish Balay 6139566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: gatol=%g,", (double)tao->gatol)); 6149566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " steptol=%g,", (double)tao->steptol)); 6159566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " gttol=%g\n", (double)tao->gttol)); 6169566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Residual in Function/Gradient:=%g\n", (double)tao->residual)); 617a7e14dcfSSatish Balay 6186246e8cdSAlp Dener if (tao->constrained) { 6199566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances:")); 6209566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " catol=%g,", (double)tao->catol)); 6219566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " crtol=%g\n", (double)tao->crtol)); 6229566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Residual in Constraints:=%g\n", (double)tao->cnorm)); 623a7e14dcfSSatish Balay } 624a7e14dcfSSatish Balay 625a7e14dcfSSatish Balay if (tao->trust < tao->steptol) { 6269566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: steptol=%g\n", (double)tao->steptol)); 6279566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Final trust region radius:=%g\n", (double)tao->trust)); 628a7e14dcfSSatish Balay } 629a7e14dcfSSatish Balay 63048a46eb9SPierre Jolivet if (tao->fmin > -1.e25) PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: function minimum=%g\n", (double)tao->fmin)); 6319566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Objective value=%g\n", (double)tao->fc)); 632a7e14dcfSSatish Balay 63363a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total number of iterations=%" PetscInt_FMT ", ", tao->niter)); 63463a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " (max: %" PetscInt_FMT ")\n", tao->max_it)); 635a7e14dcfSSatish Balay 636a7e14dcfSSatish Balay if (tao->nfuncs > 0) { 63763a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total number of function evaluations=%" PetscInt_FMT ",", tao->nfuncs)); 63863a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " max: %" PetscInt_FMT "\n", tao->max_funcs)); 639a7e14dcfSSatish Balay } 640a7e14dcfSSatish Balay if (tao->ngrads > 0) { 64163a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total number of gradient evaluations=%" PetscInt_FMT ",", tao->ngrads)); 64263a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " max: %" PetscInt_FMT "\n", tao->max_funcs)); 643a7e14dcfSSatish Balay } 644a7e14dcfSSatish Balay if (tao->nfuncgrads > 0) { 64563a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total number of function/gradient evaluations=%" PetscInt_FMT ",", tao->nfuncgrads)); 64663a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " (max: %" PetscInt_FMT ")\n", tao->max_funcs)); 647a7e14dcfSSatish Balay } 64848a46eb9SPierre Jolivet if (tao->nhess > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of Hessian evaluations=%" PetscInt_FMT "\n", tao->nhess)); 64948a46eb9SPierre Jolivet if (tao->nconstraints > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of constraint function evaluations=%" PetscInt_FMT "\n", tao->nconstraints)); 65048a46eb9SPierre Jolivet if (tao->njac > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of Jacobian evaluations=%" PetscInt_FMT "\n", tao->njac)); 651a7e14dcfSSatish Balay 652a7e14dcfSSatish Balay if (tao->reason > 0) { 6539566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Solution converged: ")); 654a7e14dcfSSatish Balay switch (tao->reason) { 655d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_GATOL: 656d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)|| <= gatol\n")); 657d71ae5a4SJacob Faibussowitsch break; 658d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_GRTOL: 659d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)||/|f(X)| <= grtol\n")); 660d71ae5a4SJacob Faibussowitsch break; 661d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_GTTOL: 662d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)||/||g(X0)|| <= gttol\n")); 663d71ae5a4SJacob Faibussowitsch break; 664d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_STEPTOL: 665d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Steptol -- step size small\n")); 666d71ae5a4SJacob Faibussowitsch break; 667d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_MINF: 668d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Minf -- f < fmin\n")); 669d71ae5a4SJacob Faibussowitsch break; 670d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_USER: 671d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " User Terminated\n")); 672d71ae5a4SJacob Faibussowitsch break; 673d71ae5a4SJacob Faibussowitsch default: 674d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "\n")); 675d71ae5a4SJacob Faibussowitsch break; 676a7e14dcfSSatish Balay } 677a7e14dcfSSatish Balay } else { 6789566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Solver terminated: %d", tao->reason)); 679a7e14dcfSSatish Balay switch (tao->reason) { 680d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_MAXITS: 681d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum Iterations\n")); 682d71ae5a4SJacob Faibussowitsch break; 683d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_NAN: 684d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " NAN or Inf encountered\n")); 685d71ae5a4SJacob Faibussowitsch break; 686d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_MAXFCN: 687d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum Function Evaluations\n")); 688d71ae5a4SJacob Faibussowitsch break; 689d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_LS_FAILURE: 690d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Line Search Failure\n")); 691d71ae5a4SJacob Faibussowitsch break; 692d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_TR_REDUCTION: 693d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Trust Region too small\n")); 694d71ae5a4SJacob Faibussowitsch break; 695d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_USER: 696d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " User Terminated\n")); 697d71ae5a4SJacob Faibussowitsch break; 698d71ae5a4SJacob Faibussowitsch default: 699d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "\n")); 700d71ae5a4SJacob Faibussowitsch break; 701a7e14dcfSSatish Balay } 702a7e14dcfSSatish Balay } 7039566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 704a7e14dcfSSatish Balay } else if (isstring) { 7059566063dSJacob Faibussowitsch PetscCall(TaoGetType(tao, &type)); 7069566063dSJacob Faibussowitsch PetscCall(PetscViewerStringSPrintf(viewer, " %-3.3s", type)); 707a7e14dcfSSatish Balay } 7083ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 709a7e14dcfSSatish Balay } 710a7e14dcfSSatish Balay 711a7e14dcfSSatish Balay /*@ 712414d97d3SAlp Dener TaoSetRecycleHistory - Sets the boolean flag to enable/disable re-using 71365ba42b6SBarry Smith iterate information from the previous `TaoSolve()`. This feature is disabled by 714414d97d3SAlp Dener default. 715414d97d3SAlp Dener 71667be906fSBarry Smith Logically Collective 71767be906fSBarry Smith 71867be906fSBarry Smith Input Parameters: 71947450a7bSBarry Smith + tao - the `Tao` context 72067be906fSBarry Smith - recycle - boolean flag 72167be906fSBarry Smith 72247450a7bSBarry Smith Options Database Key: 72367be906fSBarry Smith . -tao_recycle_history <true,false> - reuse the history 72467be906fSBarry Smith 72567be906fSBarry Smith Level: intermediate 72667be906fSBarry Smith 72767be906fSBarry Smith Notes: 72865ba42b6SBarry Smith For conjugate gradient methods (`TAOBNCG`), this re-uses the latest search direction 72965ba42b6SBarry Smith from the previous `TaoSolve()` call when computing the first search direction in a 730414d97d3SAlp Dener new solution. By default, CG methods set the first search direction to the 731414d97d3SAlp Dener negative gradient. 732414d97d3SAlp Dener 73365ba42b6SBarry Smith For quasi-Newton family of methods (`TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`), this re-uses 73465ba42b6SBarry Smith the accumulated quasi-Newton Hessian approximation from the previous `TaoSolve()` 735414d97d3SAlp Dener call. By default, QN family of methods reset the initial Hessian approximation to 736414d97d3SAlp Dener the identity matrix. 737414d97d3SAlp Dener 738414d97d3SAlp Dener For any other algorithm, this setting has no effect. 739414d97d3SAlp Dener 7401cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL` 741414d97d3SAlp Dener @*/ 742d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetRecycleHistory(Tao tao, PetscBool recycle) 743d71ae5a4SJacob Faibussowitsch { 744414d97d3SAlp Dener PetscFunctionBegin; 745414d97d3SAlp Dener PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 74694511df7SStefano Zampini PetscValidLogicalCollectiveBool(tao, recycle, 2); 747414d97d3SAlp Dener tao->recycle = recycle; 7483ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 749414d97d3SAlp Dener } 750414d97d3SAlp Dener 751414d97d3SAlp Dener /*@ 752414d97d3SAlp Dener TaoGetRecycleHistory - Retrieve the boolean flag for re-using iterate information 75365ba42b6SBarry Smith from the previous `TaoSolve()`. This feature is disabled by default. 754414d97d3SAlp Dener 75567be906fSBarry Smith Logically Collective 756414d97d3SAlp Dener 75747450a7bSBarry Smith Input Parameter: 75847450a7bSBarry Smith . tao - the `Tao` context 759414d97d3SAlp Dener 76047450a7bSBarry Smith Output Parameter: 761147403d9SBarry Smith . recycle - boolean flag 762414d97d3SAlp Dener 763414d97d3SAlp Dener Level: intermediate 764414d97d3SAlp Dener 7651cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL` 766414d97d3SAlp Dener @*/ 767d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetRecycleHistory(Tao tao, PetscBool *recycle) 768d71ae5a4SJacob Faibussowitsch { 769414d97d3SAlp Dener PetscFunctionBegin; 770414d97d3SAlp Dener PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 7714f572ea9SToby Isaac PetscAssertPointer(recycle, 2); 772414d97d3SAlp Dener *recycle = tao->recycle; 7733ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 774414d97d3SAlp Dener } 775414d97d3SAlp Dener 776414d97d3SAlp Dener /*@ 77747450a7bSBarry Smith TaoSetTolerances - Sets parameters used in `TaoSolve()` convergence tests 778a7e14dcfSSatish Balay 77967be906fSBarry Smith Logically Collective 780a7e14dcfSSatish Balay 781a7e14dcfSSatish Balay Input Parameters: 78247450a7bSBarry Smith + tao - the `Tao` context 783a7e14dcfSSatish Balay . gatol - stop if norm of gradient is less than this 784a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this 785a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by this factor 786a7e14dcfSSatish Balay 787a7e14dcfSSatish Balay Options Database Keys: 788e52336cbSBarry Smith + -tao_gatol <gatol> - Sets gatol 789a7e14dcfSSatish Balay . -tao_grtol <grtol> - Sets grtol 790a7e14dcfSSatish Balay - -tao_gttol <gttol> - Sets gttol 791a7e14dcfSSatish Balay 7923b242c63SJacob Faibussowitsch Stopping Criteria\: 79367be906fSBarry Smith .vb 79467be906fSBarry Smith ||g(X)|| <= gatol 79567be906fSBarry Smith ||g(X)|| / |f(X)| <= grtol 79667be906fSBarry Smith ||g(X)|| / ||g(X0)|| <= gttol 79767be906fSBarry Smith .ve 798a7e14dcfSSatish Balay 799a7e14dcfSSatish Balay Level: beginner 800a7e14dcfSSatish Balay 80167be906fSBarry Smith Note: 80267be906fSBarry Smith Use `PETSC_DEFAULT` to leave one or more tolerances unchanged. 803a7e14dcfSSatish Balay 8041cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoGetTolerances()` 805a7e14dcfSSatish Balay @*/ 806d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetTolerances(Tao tao, PetscReal gatol, PetscReal grtol, PetscReal gttol) 807d71ae5a4SJacob Faibussowitsch { 808a7e14dcfSSatish Balay PetscFunctionBegin; 809441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 81094511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, gatol, 2); 81194511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, grtol, 3); 81294511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, gttol, 4); 813a7e14dcfSSatish Balay 81413bcc0bdSJacob Faibussowitsch if (gatol != (PetscReal)PETSC_DEFAULT) { 815a7e14dcfSSatish Balay if (gatol < 0) { 8169566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Tried to set negative gatol -- ignored.\n")); 817a7e14dcfSSatish Balay } else { 818a7e14dcfSSatish Balay tao->gatol = PetscMax(0, gatol); 8196552cf8aSJason Sarich tao->gatol_changed = PETSC_TRUE; 820a7e14dcfSSatish Balay } 821a7e14dcfSSatish Balay } 822a7e14dcfSSatish Balay 82313bcc0bdSJacob Faibussowitsch if (grtol != (PetscReal)PETSC_DEFAULT) { 824a7e14dcfSSatish Balay if (grtol < 0) { 8259566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Tried to set negative grtol -- ignored.\n")); 826a7e14dcfSSatish Balay } else { 827a7e14dcfSSatish Balay tao->grtol = PetscMax(0, grtol); 8286552cf8aSJason Sarich tao->grtol_changed = PETSC_TRUE; 829a7e14dcfSSatish Balay } 830a7e14dcfSSatish Balay } 831a7e14dcfSSatish Balay 83213bcc0bdSJacob Faibussowitsch if (gttol != (PetscReal)PETSC_DEFAULT) { 833a7e14dcfSSatish Balay if (gttol < 0) { 8349566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Tried to set negative gttol -- ignored.\n")); 835a7e14dcfSSatish Balay } else { 836a7e14dcfSSatish Balay tao->gttol = PetscMax(0, gttol); 8376552cf8aSJason Sarich tao->gttol_changed = PETSC_TRUE; 838a7e14dcfSSatish Balay } 839a7e14dcfSSatish Balay } 8403ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 841a7e14dcfSSatish Balay } 842a7e14dcfSSatish Balay 843a7e14dcfSSatish Balay /*@ 84447450a7bSBarry Smith TaoSetConstraintTolerances - Sets constraint tolerance parameters used in `TaoSolve()` convergence tests 845a7e14dcfSSatish Balay 84667be906fSBarry Smith Logically Collective 847a7e14dcfSSatish Balay 848a7e14dcfSSatish Balay Input Parameters: 84947450a7bSBarry Smith + tao - the `Tao` context 85067be906fSBarry Smith . catol - absolute constraint tolerance, constraint norm must be less than `catol` for used for gatol convergence criteria 85167be906fSBarry Smith - crtol - relative constraint tolerance, constraint norm must be less than `crtol` for used for gatol, gttol convergence criteria 852a7e14dcfSSatish Balay 853a7e14dcfSSatish Balay Options Database Keys: 854a7e14dcfSSatish Balay + -tao_catol <catol> - Sets catol 855a7e14dcfSSatish Balay - -tao_crtol <crtol> - Sets crtol 856a7e14dcfSSatish Balay 857a7e14dcfSSatish Balay Level: intermediate 858a7e14dcfSSatish Balay 85967be906fSBarry Smith Notes: 86067be906fSBarry Smith Use `PETSC_DEFAULT` to leave any tolerance unchanged. 861a7e14dcfSSatish Balay 8621cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoGetTolerances()`, `TaoGetConstraintTolerances()`, `TaoSetTolerances()` 863a7e14dcfSSatish Balay @*/ 864d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConstraintTolerances(Tao tao, PetscReal catol, PetscReal crtol) 865d71ae5a4SJacob Faibussowitsch { 866a7e14dcfSSatish Balay PetscFunctionBegin; 867441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 86894511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, catol, 2); 86994511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, crtol, 3); 870a7e14dcfSSatish Balay 87113bcc0bdSJacob Faibussowitsch if (catol != (PetscReal)PETSC_DEFAULT) { 872a7e14dcfSSatish Balay if (catol < 0) { 8739566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Tried to set negative catol -- ignored.\n")); 874a7e14dcfSSatish Balay } else { 875a7e14dcfSSatish Balay tao->catol = PetscMax(0, catol); 8766552cf8aSJason Sarich tao->catol_changed = PETSC_TRUE; 877a7e14dcfSSatish Balay } 878a7e14dcfSSatish Balay } 879a7e14dcfSSatish Balay 88013bcc0bdSJacob Faibussowitsch if (crtol != (PetscReal)PETSC_DEFAULT) { 881a7e14dcfSSatish Balay if (crtol < 0) { 8829566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Tried to set negative crtol -- ignored.\n")); 883a7e14dcfSSatish Balay } else { 884a7e14dcfSSatish Balay tao->crtol = PetscMax(0, crtol); 8856552cf8aSJason Sarich tao->crtol_changed = PETSC_TRUE; 886a7e14dcfSSatish Balay } 887a7e14dcfSSatish Balay } 8883ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 889a7e14dcfSSatish Balay } 890a7e14dcfSSatish Balay 89158417fe7SBarry Smith /*@ 89247450a7bSBarry Smith TaoGetConstraintTolerances - Gets constraint tolerance parameters used in `TaoSolve()` convergence tests 89358417fe7SBarry Smith 89467be906fSBarry Smith Not Collective 89558417fe7SBarry Smith 89658417fe7SBarry Smith Input Parameter: 89747450a7bSBarry Smith . tao - the `Tao` context 89858417fe7SBarry Smith 899d8d19677SJose E. Roman Output Parameters: 90067be906fSBarry Smith + catol - absolute constraint tolerance, constraint norm must be less than `catol` for used for gatol convergence criteria 90167be906fSBarry Smith - crtol - relative constraint tolerance, constraint norm must be less than `crtol` for used for gatol, gttol convergence criteria 90258417fe7SBarry Smith 90358417fe7SBarry Smith Level: intermediate 90458417fe7SBarry Smith 9051cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReasons`,`TaoGetTolerances()`, `TaoSetTolerances()`, `TaoSetConstraintTolerances()` 90658417fe7SBarry Smith @*/ 907d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConstraintTolerances(Tao tao, PetscReal *catol, PetscReal *crtol) 908d71ae5a4SJacob Faibussowitsch { 90958417fe7SBarry Smith PetscFunctionBegin; 91058417fe7SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 91158417fe7SBarry Smith if (catol) *catol = tao->catol; 91258417fe7SBarry Smith if (crtol) *crtol = tao->crtol; 9133ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 91458417fe7SBarry Smith } 91558417fe7SBarry Smith 916a7e14dcfSSatish Balay /*@ 917a7e14dcfSSatish Balay TaoSetFunctionLowerBound - Sets a bound on the solution objective value. 918a7e14dcfSSatish Balay When an approximate solution with an objective value below this number 919a7e14dcfSSatish Balay has been found, the solver will terminate. 920a7e14dcfSSatish Balay 921c3339decSBarry Smith Logically Collective 922a7e14dcfSSatish Balay 923a7e14dcfSSatish Balay Input Parameters: 924441846f8SBarry Smith + tao - the Tao solver context 925a7e14dcfSSatish Balay - fmin - the tolerance 926a7e14dcfSSatish Balay 92747450a7bSBarry Smith Options Database Key: 928a7e14dcfSSatish Balay . -tao_fmin <fmin> - sets the minimum function value 929a7e14dcfSSatish Balay 930a7e14dcfSSatish Balay Level: intermediate 931a7e14dcfSSatish Balay 9321cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetTolerances()` 933a7e14dcfSSatish Balay @*/ 934d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetFunctionLowerBound(Tao tao, PetscReal fmin) 935d71ae5a4SJacob Faibussowitsch { 936a7e14dcfSSatish Balay PetscFunctionBegin; 937441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 93894511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, fmin, 2); 939a7e14dcfSSatish Balay tao->fmin = fmin; 9406552cf8aSJason Sarich tao->fmin_changed = PETSC_TRUE; 9413ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 942a7e14dcfSSatish Balay } 943a7e14dcfSSatish Balay 944a7e14dcfSSatish Balay /*@ 9458e96d397SJason Sarich TaoGetFunctionLowerBound - Gets the bound on the solution objective value. 946a7e14dcfSSatish Balay When an approximate solution with an objective value below this number 947a7e14dcfSSatish Balay has been found, the solver will terminate. 948a7e14dcfSSatish Balay 94967be906fSBarry Smith Not Collective 950a7e14dcfSSatish Balay 95147450a7bSBarry Smith Input Parameter: 95247450a7bSBarry Smith . tao - the `Tao` solver context 953a7e14dcfSSatish Balay 95447450a7bSBarry Smith Output Parameter: 955a7e14dcfSSatish Balay . fmin - the minimum function value 956a7e14dcfSSatish Balay 957a7e14dcfSSatish Balay Level: intermediate 958a7e14dcfSSatish Balay 9591cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetFunctionLowerBound()` 960a7e14dcfSSatish Balay @*/ 961d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetFunctionLowerBound(Tao tao, PetscReal *fmin) 962d71ae5a4SJacob Faibussowitsch { 963a7e14dcfSSatish Balay PetscFunctionBegin; 964441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 9654f572ea9SToby Isaac PetscAssertPointer(fmin, 2); 966a7e14dcfSSatish Balay *fmin = tao->fmin; 9673ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 968a7e14dcfSSatish Balay } 969a7e14dcfSSatish Balay 970a7e14dcfSSatish Balay /*@ 97147450a7bSBarry Smith TaoSetMaximumFunctionEvaluations - Sets a maximum number of function evaluations allowed for a `TaoSolve()`. 972a7e14dcfSSatish Balay 973c3339decSBarry Smith Logically Collective 974a7e14dcfSSatish Balay 975a7e14dcfSSatish Balay Input Parameters: 97647450a7bSBarry Smith + tao - the `Tao` solver context 977a7e14dcfSSatish Balay - nfcn - the maximum number of function evaluations (>=0) 978a7e14dcfSSatish Balay 97947450a7bSBarry Smith Options Database Key: 980a7e14dcfSSatish Balay . -tao_max_funcs <nfcn> - sets the maximum number of function evaluations 981a7e14dcfSSatish Balay 982a7e14dcfSSatish Balay Level: intermediate 983a7e14dcfSSatish Balay 9841cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoSetMaximumIterations()` 985a7e14dcfSSatish Balay @*/ 986d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao tao, PetscInt nfcn) 987d71ae5a4SJacob Faibussowitsch { 988a7e14dcfSSatish Balay PetscFunctionBegin; 989441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 99094511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao, nfcn, 2); 9919371c9d4SSatish Balay if (nfcn >= 0) { 9929371c9d4SSatish Balay tao->max_funcs = PetscMax(0, nfcn); 9939371c9d4SSatish Balay } else { 9949371c9d4SSatish Balay tao->max_funcs = -1; 9959371c9d4SSatish Balay } 9966552cf8aSJason Sarich tao->max_funcs_changed = PETSC_TRUE; 9973ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 998a7e14dcfSSatish Balay } 999a7e14dcfSSatish Balay 1000a7e14dcfSSatish Balay /*@ 100147450a7bSBarry Smith TaoGetMaximumFunctionEvaluations - Gets a maximum number of function evaluations allowed for a `TaoSolve()` 1002a7e14dcfSSatish Balay 1003c3339decSBarry Smith Logically Collective 1004a7e14dcfSSatish Balay 100547450a7bSBarry Smith Input Parameter: 100647450a7bSBarry Smith . tao - the `Tao` solver context 1007a7e14dcfSSatish Balay 100847450a7bSBarry Smith Output Parameter: 1009a7e14dcfSSatish Balay . nfcn - the maximum number of function evaluations 1010a7e14dcfSSatish Balay 1011a7e14dcfSSatish Balay Level: intermediate 1012a7e14dcfSSatish Balay 10131cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()` 1014a7e14dcfSSatish Balay @*/ 1015d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao tao, PetscInt *nfcn) 1016d71ae5a4SJacob Faibussowitsch { 1017a7e14dcfSSatish Balay PetscFunctionBegin; 1018441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 10194f572ea9SToby Isaac PetscAssertPointer(nfcn, 2); 1020a7e14dcfSSatish Balay *nfcn = tao->max_funcs; 10213ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1022a7e14dcfSSatish Balay } 1023a7e14dcfSSatish Balay 1024770232b9SCe Qin /*@ 102547450a7bSBarry Smith TaoGetCurrentFunctionEvaluations - Get current number of function evaluations used by a `Tao` object 1026770232b9SCe Qin 1027770232b9SCe Qin Not Collective 1028770232b9SCe Qin 102947450a7bSBarry Smith Input Parameter: 103047450a7bSBarry Smith . tao - the `Tao` solver context 1031770232b9SCe Qin 103247450a7bSBarry Smith Output Parameter: 103394511df7SStefano Zampini . nfuncs - the current number of function evaluations (maximum between gradient and function evaluations) 1034770232b9SCe Qin 1035770232b9SCe Qin Level: intermediate 1036770232b9SCe Qin 10371cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()` 1038770232b9SCe Qin @*/ 1039d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao tao, PetscInt *nfuncs) 1040d71ae5a4SJacob Faibussowitsch { 1041770232b9SCe Qin PetscFunctionBegin; 1042770232b9SCe Qin PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 10434f572ea9SToby Isaac PetscAssertPointer(nfuncs, 2); 1044770232b9SCe Qin *nfuncs = PetscMax(tao->nfuncs, tao->nfuncgrads); 10453ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1046770232b9SCe Qin } 1047770232b9SCe Qin 1048a7e14dcfSSatish Balay /*@ 104947450a7bSBarry Smith TaoSetMaximumIterations - Sets a maximum number of iterates to be used in `TaoSolve()` 1050a7e14dcfSSatish Balay 1051c3339decSBarry Smith Logically Collective 1052a7e14dcfSSatish Balay 1053a7e14dcfSSatish Balay Input Parameters: 105447450a7bSBarry Smith + tao - the `Tao` solver context 1055a7e14dcfSSatish Balay - maxits - the maximum number of iterates (>=0) 1056a7e14dcfSSatish Balay 105747450a7bSBarry Smith Options Database Key: 1058a7e14dcfSSatish Balay . -tao_max_it <its> - sets the maximum number of iterations 1059a7e14dcfSSatish Balay 1060a7e14dcfSSatish Balay Level: intermediate 1061a7e14dcfSSatish Balay 10621cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoSetMaximumFunctionEvaluations()` 1063a7e14dcfSSatish Balay @*/ 1064d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMaximumIterations(Tao tao, PetscInt maxits) 1065d71ae5a4SJacob Faibussowitsch { 1066a7e14dcfSSatish Balay PetscFunctionBegin; 1067441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 106894511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao, maxits, 2); 106947a47007SBarry Smith tao->max_it = PetscMax(0, maxits); 10706552cf8aSJason Sarich tao->max_it_changed = PETSC_TRUE; 10713ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1072a7e14dcfSSatish Balay } 1073a7e14dcfSSatish Balay 1074a7e14dcfSSatish Balay /*@ 107565ba42b6SBarry Smith TaoGetMaximumIterations - Gets a maximum number of iterates that will be used 1076a7e14dcfSSatish Balay 1077a7e14dcfSSatish Balay Not Collective 1078a7e14dcfSSatish Balay 107947450a7bSBarry Smith Input Parameter: 108047450a7bSBarry Smith . tao - the `Tao` solver context 1081a7e14dcfSSatish Balay 108247450a7bSBarry Smith Output Parameter: 1083a7e14dcfSSatish Balay . maxits - the maximum number of iterates 1084a7e14dcfSSatish Balay 1085a7e14dcfSSatish Balay Level: intermediate 1086a7e14dcfSSatish Balay 10871cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumIterations()`, `TaoGetMaximumFunctionEvaluations()` 1088a7e14dcfSSatish Balay @*/ 1089d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetMaximumIterations(Tao tao, PetscInt *maxits) 1090d71ae5a4SJacob Faibussowitsch { 1091a7e14dcfSSatish Balay PetscFunctionBegin; 1092441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 10934f572ea9SToby Isaac PetscAssertPointer(maxits, 2); 1094a7e14dcfSSatish Balay *maxits = tao->max_it; 10953ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1096a7e14dcfSSatish Balay } 1097a7e14dcfSSatish Balay 1098a7e14dcfSSatish Balay /*@ 1099a7e14dcfSSatish Balay TaoSetInitialTrustRegionRadius - Sets the initial trust region radius. 1100a7e14dcfSSatish Balay 110167be906fSBarry Smith Logically Collective 1102a7e14dcfSSatish Balay 1103d8d19677SJose E. Roman Input Parameters: 110447450a7bSBarry Smith + tao - a `Tao` optimization solver 1105a7e14dcfSSatish Balay - radius - the trust region radius 1106a7e14dcfSSatish Balay 1107a7e14dcfSSatish Balay Options Database Key: 1108a7e14dcfSSatish Balay . -tao_trust0 <t0> - sets initial trust region radius 1109a7e14dcfSSatish Balay 111047450a7bSBarry Smith Level: intermediate 111147450a7bSBarry Smith 11121cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetTrustRegionRadius()`, `TaoSetTrustRegionTolerance()`, `TAONTR` 1113a7e14dcfSSatish Balay @*/ 1114d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetInitialTrustRegionRadius(Tao tao, PetscReal radius) 1115d71ae5a4SJacob Faibussowitsch { 1116a7e14dcfSSatish Balay PetscFunctionBegin; 1117441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 111894511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, radius, 2); 1119a7e14dcfSSatish Balay tao->trust0 = PetscMax(0.0, radius); 11206552cf8aSJason Sarich tao->trust0_changed = PETSC_TRUE; 11213ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1122a7e14dcfSSatish Balay } 1123a7e14dcfSSatish Balay 1124a7e14dcfSSatish Balay /*@ 112565ba42b6SBarry Smith TaoGetInitialTrustRegionRadius - Gets the initial trust region radius. 1126a7e14dcfSSatish Balay 1127a7e14dcfSSatish Balay Not Collective 1128a7e14dcfSSatish Balay 1129a7e14dcfSSatish Balay Input Parameter: 113047450a7bSBarry Smith . tao - a `Tao` optimization solver 1131a7e14dcfSSatish Balay 1132a7e14dcfSSatish Balay Output Parameter: 1133a7e14dcfSSatish Balay . radius - the trust region radius 1134a7e14dcfSSatish Balay 1135a7e14dcfSSatish Balay Level: intermediate 1136a7e14dcfSSatish Balay 11371cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetInitialTrustRegionRadius()`, `TaoGetCurrentTrustRegionRadius()`, `TAONTR` 1138a7e14dcfSSatish Balay @*/ 1139d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetInitialTrustRegionRadius(Tao tao, PetscReal *radius) 1140d71ae5a4SJacob Faibussowitsch { 1141a7e14dcfSSatish Balay PetscFunctionBegin; 1142441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 11434f572ea9SToby Isaac PetscAssertPointer(radius, 2); 1144a7e14dcfSSatish Balay *radius = tao->trust0; 11453ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1146a7e14dcfSSatish Balay } 1147a7e14dcfSSatish Balay 1148a7e14dcfSSatish Balay /*@ 1149a7e14dcfSSatish Balay TaoGetCurrentTrustRegionRadius - Gets the current trust region radius. 1150a7e14dcfSSatish Balay 1151a7e14dcfSSatish Balay Not Collective 1152a7e14dcfSSatish Balay 1153a7e14dcfSSatish Balay Input Parameter: 115447450a7bSBarry Smith . tao - a `Tao` optimization solver 1155a7e14dcfSSatish Balay 1156a7e14dcfSSatish Balay Output Parameter: 1157a7e14dcfSSatish Balay . radius - the trust region radius 1158a7e14dcfSSatish Balay 1159a7e14dcfSSatish Balay Level: intermediate 1160a7e14dcfSSatish Balay 11611cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetInitialTrustRegionRadius()`, `TaoGetInitialTrustRegionRadius()`, `TAONTR` 1162a7e14dcfSSatish Balay @*/ 1163d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao tao, PetscReal *radius) 1164d71ae5a4SJacob Faibussowitsch { 1165a7e14dcfSSatish Balay PetscFunctionBegin; 1166441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 11674f572ea9SToby Isaac PetscAssertPointer(radius, 2); 1168a7e14dcfSSatish Balay *radius = tao->trust; 11693ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1170a7e14dcfSSatish Balay } 1171a7e14dcfSSatish Balay 1172a7e14dcfSSatish Balay /*@ 117347450a7bSBarry Smith TaoGetTolerances - gets the current values of some tolerances used for the convergence testing of `TaoSolve()` 1174a7e14dcfSSatish Balay 1175a7e14dcfSSatish Balay Not Collective 1176a7e14dcfSSatish Balay 1177f899ff85SJose E. Roman Input Parameter: 117847450a7bSBarry Smith . tao - the `Tao` context 1179a7e14dcfSSatish Balay 1180a7e14dcfSSatish Balay Output Parameters: 1181e52336cbSBarry Smith + gatol - stop if norm of gradient is less than this 1182a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this 1183a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by a this factor 1184a7e14dcfSSatish Balay 1185a7e14dcfSSatish Balay Level: intermediate 1186a1cb98faSBarry Smith 1187a1cb98faSBarry Smith Note: 118847450a7bSBarry Smith `NULL` can be used as an argument if not all tolerances values are needed 1189a1cb98faSBarry Smith 11901cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()` 1191a7e14dcfSSatish Balay @*/ 1192d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetTolerances(Tao tao, PetscReal *gatol, PetscReal *grtol, PetscReal *gttol) 1193d71ae5a4SJacob Faibussowitsch { 1194a7e14dcfSSatish Balay PetscFunctionBegin; 1195441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1196a7e14dcfSSatish Balay if (gatol) *gatol = tao->gatol; 1197a7e14dcfSSatish Balay if (grtol) *grtol = tao->grtol; 1198a7e14dcfSSatish Balay if (gttol) *gttol = tao->gttol; 11993ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1200a7e14dcfSSatish Balay } 1201a7e14dcfSSatish Balay 1202a7e14dcfSSatish Balay /*@ 1203a7e14dcfSSatish Balay TaoGetKSP - Gets the linear solver used by the optimization solver. 1204a7e14dcfSSatish Balay 1205a7e14dcfSSatish Balay Not Collective 1206a7e14dcfSSatish Balay 120747450a7bSBarry Smith Input Parameter: 120847450a7bSBarry Smith . tao - the `Tao` solver 1209a7e14dcfSSatish Balay 121047450a7bSBarry Smith Output Parameter: 121147450a7bSBarry Smith . ksp - the `KSP` linear solver used in the optimization solver 1212a7e14dcfSSatish Balay 1213a7e14dcfSSatish Balay Level: intermediate 1214a7e14dcfSSatish Balay 12151cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `KSP` 1216a7e14dcfSSatish Balay @*/ 1217d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetKSP(Tao tao, KSP *ksp) 1218d71ae5a4SJacob Faibussowitsch { 1219a7e14dcfSSatish Balay PetscFunctionBegin; 122094511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 12214f572ea9SToby Isaac PetscAssertPointer(ksp, 2); 1222a7e14dcfSSatish Balay *ksp = tao->ksp; 12233ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1224a7e14dcfSSatish Balay } 1225a7e14dcfSSatish Balay 1226025e9500SJason Sarich /*@ 1227025e9500SJason Sarich TaoGetLinearSolveIterations - Gets the total number of linear iterations 122847450a7bSBarry Smith used by the `Tao` solver 1229025e9500SJason Sarich 1230025e9500SJason Sarich Not Collective 1231025e9500SJason Sarich 1232025e9500SJason Sarich Input Parameter: 123347450a7bSBarry Smith . tao - the `Tao` context 1234025e9500SJason Sarich 1235025e9500SJason Sarich Output Parameter: 1236025e9500SJason Sarich . lits - number of linear iterations 1237025e9500SJason Sarich 1238025e9500SJason Sarich Level: intermediate 1239025e9500SJason Sarich 124047450a7bSBarry Smith Note: 124147450a7bSBarry Smith This counter is reset to zero for each successive call to `TaoSolve()` 124247450a7bSBarry Smith 12431cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetKSP()` 1244025e9500SJason Sarich @*/ 1245d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetLinearSolveIterations(Tao tao, PetscInt *lits) 1246d71ae5a4SJacob Faibussowitsch { 1247025e9500SJason Sarich PetscFunctionBegin; 1248025e9500SJason Sarich PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 12494f572ea9SToby Isaac PetscAssertPointer(lits, 2); 12502d9aa51bSJason Sarich *lits = tao->ksp_tot_its; 12513ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1252025e9500SJason Sarich } 1253025e9500SJason Sarich 1254a7e14dcfSSatish Balay /*@ 1255a7e14dcfSSatish Balay TaoGetLineSearch - Gets the line search used by the optimization solver. 1256a7e14dcfSSatish Balay 1257a7e14dcfSSatish Balay Not Collective 1258a7e14dcfSSatish Balay 125947450a7bSBarry Smith Input Parameter: 126047450a7bSBarry Smith . tao - the `Tao` solver 1261a7e14dcfSSatish Balay 126247450a7bSBarry Smith Output Parameter: 1263a7e14dcfSSatish Balay . ls - the line search used in the optimization solver 1264a7e14dcfSSatish Balay 1265a7e14dcfSSatish Balay Level: intermediate 1266a7e14dcfSSatish Balay 12671cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoLineSearch`, `TaoLineSearchType` 1268a7e14dcfSSatish Balay @*/ 1269d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetLineSearch(Tao tao, TaoLineSearch *ls) 1270d71ae5a4SJacob Faibussowitsch { 1271a7e14dcfSSatish Balay PetscFunctionBegin; 127294511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 12734f572ea9SToby Isaac PetscAssertPointer(ls, 2); 1274a7e14dcfSSatish Balay *ls = tao->linesearch; 12753ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1276a7e14dcfSSatish Balay } 1277a7e14dcfSSatish Balay 1278a7e14dcfSSatish Balay /*@ 1279a7e14dcfSSatish Balay TaoAddLineSearchCounts - Adds the number of function evaluations spent 1280a7e14dcfSSatish Balay in the line search to the running total. 1281a7e14dcfSSatish Balay 1282a7e14dcfSSatish Balay Input Parameters: 12832fe279fdSBarry Smith . tao - the `Tao` solver 1284a7e14dcfSSatish Balay 1285a7e14dcfSSatish Balay Level: developer 1286a7e14dcfSSatish Balay 12871cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLineSearch()`, `TaoLineSearchApply()` 1288a7e14dcfSSatish Balay @*/ 1289d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoAddLineSearchCounts(Tao tao) 1290d71ae5a4SJacob Faibussowitsch { 1291a7e14dcfSSatish Balay PetscBool flg; 1292a7e14dcfSSatish Balay PetscInt nfeval, ngeval, nfgeval; 129347a47007SBarry Smith 1294a7e14dcfSSatish Balay PetscFunctionBegin; 1295441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1296a7e14dcfSSatish Balay if (tao->linesearch) { 12979566063dSJacob Faibussowitsch PetscCall(TaoLineSearchIsUsingTaoRoutines(tao->linesearch, &flg)); 129894ae4db5SBarry Smith if (!flg) { 12999566063dSJacob Faibussowitsch PetscCall(TaoLineSearchGetNumberFunctionEvaluations(tao->linesearch, &nfeval, &ngeval, &nfgeval)); 1300a7e14dcfSSatish Balay tao->nfuncs += nfeval; 1301a7e14dcfSSatish Balay tao->ngrads += ngeval; 1302a7e14dcfSSatish Balay tao->nfuncgrads += nfgeval; 1303a7e14dcfSSatish Balay } 1304a7e14dcfSSatish Balay } 13053ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1306a7e14dcfSSatish Balay } 1307a7e14dcfSSatish Balay 1308a7e14dcfSSatish Balay /*@ 130947450a7bSBarry Smith TaoGetSolution - Returns the vector with the current solution from the `Tao` object 1310a7e14dcfSSatish Balay 1311a7e14dcfSSatish Balay Not Collective 1312a7e14dcfSSatish Balay 1313a7e14dcfSSatish Balay Input Parameter: 131447450a7bSBarry Smith . tao - the `Tao` context 1315a7e14dcfSSatish Balay 1316a7e14dcfSSatish Balay Output Parameter: 1317a7e14dcfSSatish Balay . X - the current solution 1318a7e14dcfSSatish Balay 1319a7e14dcfSSatish Balay Level: intermediate 1320a7e14dcfSSatish Balay 132165ba42b6SBarry Smith Note: 132265ba42b6SBarry Smith The returned vector will be the same object that was passed into `TaoSetSolution()` 132365ba42b6SBarry Smith 13241cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetSolution()`, `TaoSolve()` 1325a7e14dcfSSatish Balay @*/ 1326d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetSolution(Tao tao, Vec *X) 1327d71ae5a4SJacob Faibussowitsch { 1328a7e14dcfSSatish Balay PetscFunctionBegin; 1329441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 13304f572ea9SToby Isaac PetscAssertPointer(X, 2); 1331a7e14dcfSSatish Balay *X = tao->solution; 13323ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1333a7e14dcfSSatish Balay } 1334a7e14dcfSSatish Balay 1335a7e14dcfSSatish Balay /*@ 133647450a7bSBarry Smith TaoResetStatistics - Initialize the statistics collected by the `Tao` object. 1337a7e14dcfSSatish Balay These statistics include the iteration number, residual norms, and convergence status. 1338a7e14dcfSSatish Balay This routine gets called before solving each optimization problem. 1339a7e14dcfSSatish Balay 1340c3339decSBarry Smith Collective 1341a7e14dcfSSatish Balay 134247450a7bSBarry Smith Input Parameter: 1343e056e8ceSJacob Faibussowitsch . tao - the `Tao` context 1344a7e14dcfSSatish Balay 1345a7e14dcfSSatish Balay Level: developer 1346a7e14dcfSSatish Balay 13471cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSolve()` 1348a7e14dcfSSatish Balay @*/ 1349d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoResetStatistics(Tao tao) 1350d71ae5a4SJacob Faibussowitsch { 1351a7e14dcfSSatish Balay PetscFunctionBegin; 1352441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1353a7e14dcfSSatish Balay tao->niter = 0; 1354a7e14dcfSSatish Balay tao->nfuncs = 0; 1355a7e14dcfSSatish Balay tao->nfuncgrads = 0; 1356a7e14dcfSSatish Balay tao->ngrads = 0; 1357a7e14dcfSSatish Balay tao->nhess = 0; 1358a7e14dcfSSatish Balay tao->njac = 0; 1359a7e14dcfSSatish Balay tao->nconstraints = 0; 1360a7e14dcfSSatish Balay tao->ksp_its = 0; 1361ae93cb3cSJason Sarich tao->ksp_tot_its = 0; 1362a7e14dcfSSatish Balay tao->reason = TAO_CONTINUE_ITERATING; 1363a7e14dcfSSatish Balay tao->residual = 0.0; 1364a7e14dcfSSatish Balay tao->cnorm = 0.0; 1365a7e14dcfSSatish Balay tao->step = 0.0; 1366a7e14dcfSSatish Balay tao->lsflag = PETSC_FALSE; 1367a7e14dcfSSatish Balay if (tao->hist_reset) tao->hist_len = 0; 13683ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1369a7e14dcfSSatish Balay } 1370a7e14dcfSSatish Balay 1371a7e14dcfSSatish Balay /*@C 1372e1e80dc8SAlp Dener TaoSetUpdate - Sets the general-purpose update function called 13732fe279fdSBarry Smith at the beginning of every iteration of the optimization algorithm. Called after the new solution and the gradient 1374e1e80dc8SAlp Dener is determined, but before the Hessian is computed (if applicable). 1375e1e80dc8SAlp Dener 1376c3339decSBarry Smith Logically Collective 1377e1e80dc8SAlp Dener 1378e1e80dc8SAlp Dener Input Parameters: 13792fe279fdSBarry Smith + tao - The `Tao` solver context 1380a2b725a8SWilliam Gropp - func - The function 1381e1e80dc8SAlp Dener 138220f4b53cSBarry Smith Calling sequence of `func`: 138320f4b53cSBarry Smith + tao - the optimizer context 1384e056e8ceSJacob Faibussowitsch - ctx - The current step of the iteration 1385e1e80dc8SAlp Dener 1386e1e80dc8SAlp Dener Level: advanced 1387e1e80dc8SAlp Dener 13881cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()` 1389e1e80dc8SAlp Dener @*/ 1390d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetUpdate(Tao tao, PetscErrorCode (*func)(Tao, PetscInt, void *), void *ctx) 1391d71ae5a4SJacob Faibussowitsch { 1392e1e80dc8SAlp Dener PetscFunctionBegin; 1393e1e80dc8SAlp Dener PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1394e1e80dc8SAlp Dener tao->ops->update = func; 1395e1e80dc8SAlp Dener tao->user_update = ctx; 13963ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1397e1e80dc8SAlp Dener } 1398e1e80dc8SAlp Dener 1399e1e80dc8SAlp Dener /*@C 1400a7e14dcfSSatish Balay TaoSetConvergenceTest - Sets the function that is to be used to test 1401a7e14dcfSSatish Balay for convergence o fthe iterative minimization solution. The new convergence 140265ba42b6SBarry Smith testing routine will replace Tao's default convergence test. 1403a7e14dcfSSatish Balay 1404c3339decSBarry Smith Logically Collective 1405a7e14dcfSSatish Balay 1406a7e14dcfSSatish Balay Input Parameters: 140747450a7bSBarry Smith + tao - the `Tao` object 1408a7e14dcfSSatish Balay . conv - the routine to test for convergence 1409a7e14dcfSSatish Balay - ctx - [optional] context for private data for the convergence routine 141047450a7bSBarry Smith (may be `NULL`) 1411a7e14dcfSSatish Balay 141220f4b53cSBarry Smith Calling sequence of `conv`: 141347450a7bSBarry Smith + tao - the `Tao` object 1414a7e14dcfSSatish Balay - ctx - [optional] convergence context 1415a7e14dcfSSatish Balay 141647450a7bSBarry Smith Level: advanced 141747450a7bSBarry Smith 141865ba42b6SBarry Smith Note: 141965ba42b6SBarry Smith The new convergence testing routine should call `TaoSetConvergedReason()`. 1420a7e14dcfSSatish Balay 14211cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoSetConvergedReason()`, `TaoGetSolutionStatus()`, `TaoGetTolerances()`, `TaoSetMonitor` 1422a7e14dcfSSatish Balay @*/ 1423d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergenceTest(Tao tao, PetscErrorCode (*conv)(Tao, void *), void *ctx) 1424d71ae5a4SJacob Faibussowitsch { 1425a7e14dcfSSatish Balay PetscFunctionBegin; 1426441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 142794511df7SStefano Zampini tao->ops->convergencetest = conv; 142894511df7SStefano Zampini tao->cnvP = ctx; 14293ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1430a7e14dcfSSatish Balay } 1431a7e14dcfSSatish Balay 1432a7e14dcfSSatish Balay /*@C 143365ba42b6SBarry Smith TaoSetMonitor - Sets an additional function that is to be used at every 1434a7e14dcfSSatish Balay iteration of the solver to display the iteration's 1435a7e14dcfSSatish Balay progress. 1436a7e14dcfSSatish Balay 1437c3339decSBarry Smith Logically Collective 1438a7e14dcfSSatish Balay 1439a7e14dcfSSatish Balay Input Parameters: 144047450a7bSBarry Smith + tao - the `Tao` solver context 14412fe279fdSBarry Smith . func - monitoring routine 14422fe279fdSBarry Smith . ctx - [optional] user-defined context for private data for the monitor routine (may be `NULL`) 14432fe279fdSBarry Smith - dest - [optional] function to destroy the context when the `Tao` is destroyed 1444a7e14dcfSSatish Balay 14452fe279fdSBarry Smith Calling sequence of `func`: 144647450a7bSBarry Smith + tao - the `Tao` solver context 14472fe279fdSBarry Smith - ctx - [optional] monitoring context 14482fe279fdSBarry Smith 14492fe279fdSBarry Smith Calling sequence of `dest`: 14502fe279fdSBarry Smith . ctx - monitoring context 1451a7e14dcfSSatish Balay 1452a7e14dcfSSatish Balay Options Database Keys: 145365ba42b6SBarry Smith + -tao_monitor - sets the default monitor `TaoMonitorDefault()` 1454a7e14dcfSSatish Balay . -tao_smonitor - sets short monitor 1455a7e14dcfSSatish Balay . -tao_cmonitor - same as smonitor plus constraint norm 1456a7e14dcfSSatish Balay . -tao_view_solution - view solution at each iteration 1457a7e14dcfSSatish Balay . -tao_view_gradient - view gradient at each iteration 14584a48860cSAlp Dener . -tao_view_ls_residual - view least-squares residual vector at each iteration 1459*b2dc45ddSPierre Jolivet - -tao_monitor_cancel - cancels all monitors that have been hardwired into a code by calls to TaoSetMonitor(), but does not cancel those set via the options database. 1460a7e14dcfSSatish Balay 146147450a7bSBarry Smith Level: intermediate 146247450a7bSBarry Smith 146387497f52SBarry Smith Notes: 1464a7e14dcfSSatish Balay Several different monitoring routines may be set by calling 146565ba42b6SBarry Smith `TaoSetMonitor()` multiple times; all will be called in the 1466a7e14dcfSSatish Balay order in which they were set. 1467a7e14dcfSSatish Balay 1468e056e8ceSJacob Faibussowitsch Fortran Notes: 146995452b02SPatrick Sanan Only one monitor function may be set 1470a7e14dcfSSatish Balay 1471*b2dc45ddSPierre Jolivet .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoMonitorDefault()`, `TaoMonitorCancel()`, `TaoSetDestroyRoutine()`, `TaoView()` 1472a7e14dcfSSatish Balay @*/ 1473d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMonitor(Tao tao, PetscErrorCode (*func)(Tao, void *), void *ctx, PetscErrorCode (*dest)(void **)) 1474d71ae5a4SJacob Faibussowitsch { 147508d19d1fSJason Sarich PetscInt i; 14768732526dSAlp Dener PetscBool identical; 147708d19d1fSJason Sarich 1478a7e14dcfSSatish Balay PetscFunctionBegin; 1479441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 14803c859ba3SBarry Smith PetscCheck(tao->numbermonitors < MAXTAOMONITORS, PetscObjectComm((PetscObject)tao), PETSC_ERR_SUP, "Cannot attach another monitor -- max=%d", MAXTAOMONITORS); 148108d19d1fSJason Sarich 148208d19d1fSJason Sarich for (i = 0; i < tao->numbermonitors; i++) { 14839566063dSJacob Faibussowitsch PetscCall(PetscMonitorCompare((PetscErrorCode(*)(void))func, ctx, dest, (PetscErrorCode(*)(void))tao->monitor[i], tao->monitorcontext[i], tao->monitordestroy[i], &identical)); 14843ba16761SJacob Faibussowitsch if (identical) PetscFunctionReturn(PETSC_SUCCESS); 148508d19d1fSJason Sarich } 1486a7e14dcfSSatish Balay tao->monitor[tao->numbermonitors] = func; 14878732526dSAlp Dener tao->monitorcontext[tao->numbermonitors] = (void *)ctx; 1488a7e14dcfSSatish Balay tao->monitordestroy[tao->numbermonitors] = dest; 1489a7e14dcfSSatish Balay ++tao->numbermonitors; 14903ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1491a7e14dcfSSatish Balay } 1492a7e14dcfSSatish Balay 1493a7e14dcfSSatish Balay /*@ 1494*b2dc45ddSPierre Jolivet TaoMonitorCancel - Clears all the monitor functions for a `Tao` object. 1495a7e14dcfSSatish Balay 1496c3339decSBarry Smith Logically Collective 1497a7e14dcfSSatish Balay 149847450a7bSBarry Smith Input Parameter: 149947450a7bSBarry Smith . tao - the `Tao` solver context 1500a7e14dcfSSatish Balay 15013c7db156SBarry Smith Options Database Key: 1502*b2dc45ddSPierre Jolivet . -tao_monitor_cancel - cancels all monitors that have been hardwired 150365ba42b6SBarry Smith into a code by calls to `TaoSetMonitor()`, but does not cancel those 1504a7e14dcfSSatish Balay set via the options database 1505a7e14dcfSSatish Balay 1506a7e14dcfSSatish Balay Level: advanced 1507a7e14dcfSSatish Balay 150847450a7bSBarry Smith Note: 150947450a7bSBarry Smith There is no way to clear one specific monitor from a `Tao` object. 151047450a7bSBarry Smith 15111cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoSetMonitor()` 1512a7e14dcfSSatish Balay @*/ 1513*b2dc45ddSPierre Jolivet PetscErrorCode TaoMonitorCancel(Tao tao) 1514d71ae5a4SJacob Faibussowitsch { 1515a7e14dcfSSatish Balay PetscInt i; 151647a47007SBarry Smith 1517a7e14dcfSSatish Balay PetscFunctionBegin; 1518441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1519a7e14dcfSSatish Balay for (i = 0; i < tao->numbermonitors; i++) { 152048a46eb9SPierre Jolivet if (tao->monitordestroy[i]) PetscCall((*tao->monitordestroy[i])(&tao->monitorcontext[i])); 1521a7e14dcfSSatish Balay } 1522a7e14dcfSSatish Balay tao->numbermonitors = 0; 15233ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1524a7e14dcfSSatish Balay } 1525a7e14dcfSSatish Balay 15267fab98ebSJason Sarich /*@ 152747450a7bSBarry Smith TaoMonitorDefault - Default routine for monitoring progress of `TaoSolve()` 1528a7e14dcfSSatish Balay 1529c3339decSBarry Smith Collective 1530a7e14dcfSSatish Balay 1531a7e14dcfSSatish Balay Input Parameters: 153247450a7bSBarry Smith + tao - the `Tao` context 153347450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL` 1534a7e14dcfSSatish Balay 153547450a7bSBarry Smith Options Database Key: 1536147403d9SBarry Smith . -tao_monitor - turn on default monitoring 1537a7e14dcfSSatish Balay 1538a7e14dcfSSatish Balay Level: advanced 1539a7e14dcfSSatish Balay 154047450a7bSBarry Smith Note: 154147450a7bSBarry Smith This monitor prints the function value and gradient 154247450a7bSBarry Smith norm at each iteration. 154347450a7bSBarry Smith 15441cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1545a7e14dcfSSatish Balay @*/ 1546d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDefault(Tao tao, void *ctx) 1547d71ae5a4SJacob Faibussowitsch { 154863b15415SAlp Dener PetscInt its, tabs; 1549a7e14dcfSSatish Balay PetscReal fct, gnorm; 15508163d661SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 155147a47007SBarry Smith 1552a7e14dcfSSatish Balay PetscFunctionBegin; 155394511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 15548163d661SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 1555a7e14dcfSSatish Balay its = tao->niter; 1556a7e14dcfSSatish Balay fct = tao->fc; 1557a7e14dcfSSatish Balay gnorm = tao->residual; 15589566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 15599566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 1560494bef23SAlp Dener if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) { 15619566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Iteration information for %s solve.\n", ((PetscObject)tao)->prefix)); 1562494bef23SAlp Dener tao->header_printed = PETSC_TRUE; 156363b15415SAlp Dener } 156463a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " TAO,", its)); 15659566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Function value: %g,", (double)fct)); 156608d19d1fSJason Sarich if (gnorm >= PETSC_INFINITY) { 15679566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: Inf \n")); 156808d19d1fSJason Sarich } else { 15699566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g \n", (double)gnorm)); 157008d19d1fSJason Sarich } 15719566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 15723ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1573a7e14dcfSSatish Balay } 1574a7e14dcfSSatish Balay 15757fab98ebSJason Sarich /*@ 157647450a7bSBarry Smith TaoDefaultGMonitor - Default routine for monitoring progress of `TaoSolve()` with extra detail on the globalization method. 15778d5ead36SAlp Dener 1578c3339decSBarry Smith Collective 15798d5ead36SAlp Dener 15808d5ead36SAlp Dener Input Parameters: 158147450a7bSBarry Smith + tao - the `Tao` context 158247450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL` 15838d5ead36SAlp Dener 158447450a7bSBarry Smith Options Database Key: 1585147403d9SBarry Smith . -tao_gmonitor - turn on monitoring with globalization information 15868d5ead36SAlp Dener 15878d5ead36SAlp Dener Level: advanced 15888d5ead36SAlp Dener 158947450a7bSBarry Smith Note: 159047450a7bSBarry Smith This monitor prints the function value and gradient norm at each 159147450a7bSBarry Smith iteration, as well as the step size and trust radius. Note that the 159247450a7bSBarry Smith step size and trust radius may be the same for some algorithms. 159347450a7bSBarry Smith 15941cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()` 15958d5ead36SAlp Dener @*/ 1596d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultGMonitor(Tao tao, void *ctx) 1597d71ae5a4SJacob Faibussowitsch { 15988d5ead36SAlp Dener PetscInt its, tabs; 15998d5ead36SAlp Dener PetscReal fct, gnorm, stp, tr; 16008d5ead36SAlp Dener PetscViewer viewer = (PetscViewer)ctx; 16018d5ead36SAlp Dener 16028d5ead36SAlp Dener PetscFunctionBegin; 160394511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 16048d5ead36SAlp Dener PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 16058d5ead36SAlp Dener its = tao->niter; 16068d5ead36SAlp Dener fct = tao->fc; 16078d5ead36SAlp Dener gnorm = tao->residual; 16088d5ead36SAlp Dener stp = tao->step; 16098d5ead36SAlp Dener tr = tao->trust; 16109566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 16119566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 1612494bef23SAlp Dener if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) { 16139566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Iteration information for %s solve.\n", ((PetscObject)tao)->prefix)); 1614494bef23SAlp Dener tao->header_printed = PETSC_TRUE; 16158d5ead36SAlp Dener } 161663a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " TAO,", its)); 16179566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Function value: %g,", (double)fct)); 16188d5ead36SAlp Dener if (gnorm >= PETSC_INFINITY) { 16199566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: Inf,")); 16208d5ead36SAlp Dener } else { 16219566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g,", (double)gnorm)); 16228d5ead36SAlp Dener } 16239566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Step: %g, Trust: %g\n", (double)stp, (double)tr)); 16249566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 16253ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 16268d5ead36SAlp Dener } 16278d5ead36SAlp Dener 16288d5ead36SAlp Dener /*@ 162947450a7bSBarry Smith TaoDefaultSMonitor - Default routine for monitoring progress of `TaoSolve()` 1630a7e14dcfSSatish Balay 1631c3339decSBarry Smith Collective 1632a7e14dcfSSatish Balay 1633a7e14dcfSSatish Balay Input Parameters: 163447450a7bSBarry Smith + tao - the `Tao` context 163547450a7bSBarry Smith - ctx - `PetscViewer` context of type `PETSCVIEWERASCII` 1636a7e14dcfSSatish Balay 163747450a7bSBarry Smith Options Database Key: 1638147403d9SBarry Smith . -tao_smonitor - turn on default short monitoring 1639a7e14dcfSSatish Balay 1640a7e14dcfSSatish Balay Level: advanced 1641a7e14dcfSSatish Balay 164247450a7bSBarry Smith Note: 164347450a7bSBarry Smith Same as `TaoMonitorDefault()` except 164447450a7bSBarry Smith it prints fewer digits of the residual as the residual gets smaller. 164547450a7bSBarry Smith This is because the later digits are meaningless and are often 164647450a7bSBarry Smith different on different machines; by using this routine different 164747450a7bSBarry Smith machines will usually generate the same output. 164847450a7bSBarry Smith 16491cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoSetMonitor()` 1650a7e14dcfSSatish Balay @*/ 1651d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultSMonitor(Tao tao, void *ctx) 1652d71ae5a4SJacob Faibussowitsch { 16532c9b8b83SAlp Dener PetscInt its, tabs; 1654a7e14dcfSSatish Balay PetscReal fct, gnorm; 16554d4332d5SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 165647a47007SBarry Smith 1657a7e14dcfSSatish Balay PetscFunctionBegin; 165894511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 16594d4332d5SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 1660a7e14dcfSSatish Balay its = tao->niter; 1661a7e14dcfSSatish Balay fct = tao->fc; 1662a7e14dcfSSatish Balay gnorm = tao->residual; 16639566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 16649566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 166563a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "iter = %3" PetscInt_FMT ",", its)); 16669566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Function value %g,", (double)fct)); 1667d393f493SBarry Smith if (gnorm >= PETSC_INFINITY) { 16689566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: Inf \n")); 166908d19d1fSJason Sarich } else if (gnorm > 1.e-6) { 16709566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g \n", (double)gnorm)); 1671a7e14dcfSSatish Balay } else if (gnorm > 1.e-11) { 16729566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: < 1.0e-6 \n")); 1673a7e14dcfSSatish Balay } else { 16749566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: < 1.0e-11 \n")); 1675a7e14dcfSSatish Balay } 16769566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 16773ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1678a7e14dcfSSatish Balay } 1679a7e14dcfSSatish Balay 16807fab98ebSJason Sarich /*@ 168165ba42b6SBarry Smith TaoDefaultCMonitor - same as `TaoMonitorDefault()` except 168247450a7bSBarry Smith it prints the norm of the constraint function. 1683a7e14dcfSSatish Balay 1684c3339decSBarry Smith Collective 1685a7e14dcfSSatish Balay 1686a7e14dcfSSatish Balay Input Parameters: 168747450a7bSBarry Smith + tao - the `Tao` context 168847450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL` 1689a7e14dcfSSatish Balay 169047450a7bSBarry Smith Options Database Key: 1691147403d9SBarry Smith . -tao_cmonitor - monitor the constraints 1692a7e14dcfSSatish Balay 1693a7e14dcfSSatish Balay Level: advanced 1694a7e14dcfSSatish Balay 16951cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoSetMonitor()` 1696a7e14dcfSSatish Balay @*/ 1697d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultCMonitor(Tao tao, void *ctx) 1698d71ae5a4SJacob Faibussowitsch { 16992c9b8b83SAlp Dener PetscInt its, tabs; 1700a7e14dcfSSatish Balay PetscReal fct, gnorm; 1701d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 1702a7e14dcfSSatish Balay 1703a7e14dcfSSatish Balay PetscFunctionBegin; 170494511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1705d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 1706a7e14dcfSSatish Balay its = tao->niter; 1707a7e14dcfSSatish Balay fct = tao->fc; 1708a7e14dcfSSatish Balay gnorm = tao->residual; 17099566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 17109566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 171163a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "iter = %" PetscInt_FMT ",", its)); 17129566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Function value: %g,", (double)fct)); 17139566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g ", (double)gnorm)); 17149566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Constraint: %g \n", (double)tao->cnorm)); 17159566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 17163ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1717a7e14dcfSSatish Balay } 1718a7e14dcfSSatish Balay 1719a7e14dcfSSatish Balay /*@C 172047450a7bSBarry Smith TaoSolutionMonitor - Views the solution at each iteration of `TaoSolve()` 1721a7e14dcfSSatish Balay 1722c3339decSBarry Smith Collective 1723a7e14dcfSSatish Balay 1724a7e14dcfSSatish Balay Input Parameters: 172547450a7bSBarry Smith + tao - the `Tao` context 172647450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL` 1727a7e14dcfSSatish Balay 172847450a7bSBarry Smith Options Database Key: 1729147403d9SBarry Smith . -tao_view_solution - view the solution 1730a7e14dcfSSatish Balay 1731a7e14dcfSSatish Balay Level: advanced 1732a7e14dcfSSatish Balay 17331cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1734a7e14dcfSSatish Balay @*/ 1735d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSolutionMonitor(Tao tao, void *ctx) 1736d71ae5a4SJacob Faibussowitsch { 1737feb237baSPierre Jolivet PetscViewer viewer = (PetscViewer)ctx; 173847a47007SBarry Smith 1739a7e14dcfSSatish Balay PetscFunctionBegin; 174094511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1741d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 17429566063dSJacob Faibussowitsch PetscCall(VecView(tao->solution, viewer)); 17433ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1744a7e14dcfSSatish Balay } 1745a7e14dcfSSatish Balay 1746a7e14dcfSSatish Balay /*@C 174747450a7bSBarry Smith TaoGradientMonitor - Views the gradient at each iteration of `TaoSolve()` 1748a7e14dcfSSatish Balay 1749c3339decSBarry Smith Collective 1750a7e14dcfSSatish Balay 1751a7e14dcfSSatish Balay Input Parameters: 175247450a7bSBarry Smith + tao - the `Tao` context 175347450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL` 1754a7e14dcfSSatish Balay 175547450a7bSBarry Smith Options Database Key: 1756147403d9SBarry Smith . -tao_view_gradient - view the gradient at each iteration 1757a7e14dcfSSatish Balay 1758a7e14dcfSSatish Balay Level: advanced 1759a7e14dcfSSatish Balay 17601cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1761a7e14dcfSSatish Balay @*/ 1762d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGradientMonitor(Tao tao, void *ctx) 1763d71ae5a4SJacob Faibussowitsch { 1764d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 176547a47007SBarry Smith 1766a7e14dcfSSatish Balay PetscFunctionBegin; 176794511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1768d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 17699566063dSJacob Faibussowitsch PetscCall(VecView(tao->gradient, viewer)); 17703ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1771a7e14dcfSSatish Balay } 1772a7e14dcfSSatish Balay 1773a7e14dcfSSatish Balay /*@C 177447450a7bSBarry Smith TaoStepDirectionMonitor - Views the step-direction at each iteration of `TaoSolve()` 1775a7e14dcfSSatish Balay 1776c3339decSBarry Smith Collective 1777a7e14dcfSSatish Balay 1778a7e14dcfSSatish Balay Input Parameters: 177947450a7bSBarry Smith + tao - the `Tao` context 178047450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL` 1781a7e14dcfSSatish Balay 178247450a7bSBarry Smith Options Database Key: 178347450a7bSBarry Smith . -tao_view_stepdirection - view the step direction vector at each iteration 1784a7e14dcfSSatish Balay 1785a7e14dcfSSatish Balay Level: advanced 1786a7e14dcfSSatish Balay 17871cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1788a7e14dcfSSatish Balay @*/ 1789d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoStepDirectionMonitor(Tao tao, void *ctx) 1790d71ae5a4SJacob Faibussowitsch { 1791d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 1792d393f493SBarry Smith 1793a7e14dcfSSatish Balay PetscFunctionBegin; 179494511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1795d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 17969566063dSJacob Faibussowitsch PetscCall(VecView(tao->stepdirection, viewer)); 17973ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1798a7e14dcfSSatish Balay } 1799a7e14dcfSSatish Balay 1800a7e14dcfSSatish Balay /*@C 180147450a7bSBarry Smith TaoDrawSolutionMonitor - Plots the solution at each iteration of `TaoSolve()` 1802a7e14dcfSSatish Balay 1803c3339decSBarry Smith Collective 1804a7e14dcfSSatish Balay 1805a7e14dcfSSatish Balay Input Parameters: 180647450a7bSBarry Smith + tao - the `Tao` context 180765ba42b6SBarry Smith - ctx - `TaoMonitorDraw` context 1808a7e14dcfSSatish Balay 180947450a7bSBarry Smith Options Database Key: 1810147403d9SBarry Smith . -tao_draw_solution - draw the solution at each iteration 1811a7e14dcfSSatish Balay 1812a7e14dcfSSatish Balay Level: advanced 1813a7e14dcfSSatish Balay 18141cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolutionMonitor()`, `TaoSetMonitor()`, `TaoDrawGradientMonitor`, `TaoMonitorDraw` 1815a7e14dcfSSatish Balay @*/ 1816d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawSolutionMonitor(Tao tao, void *ctx) 1817d71ae5a4SJacob Faibussowitsch { 1818e882e171SHong Zhang TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx; 181947a47007SBarry Smith 1820a7e14dcfSSatish Balay PetscFunctionBegin; 182194511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 18223ba16761SJacob Faibussowitsch if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(PETSC_SUCCESS); 18239566063dSJacob Faibussowitsch PetscCall(VecView(tao->solution, ictx->viewer)); 18243ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1825a7e14dcfSSatish Balay } 1826a7e14dcfSSatish Balay 1827a7e14dcfSSatish Balay /*@C 182847450a7bSBarry Smith TaoDrawGradientMonitor - Plots the gradient at each iteration of `TaoSolve()` 1829a7e14dcfSSatish Balay 1830c3339decSBarry Smith Collective 1831a7e14dcfSSatish Balay 1832a7e14dcfSSatish Balay Input Parameters: 183347450a7bSBarry Smith + tao - the `Tao` context 183465ba42b6SBarry Smith - ctx - `PetscViewer` context 1835a7e14dcfSSatish Balay 183647450a7bSBarry Smith Options Database Key: 1837147403d9SBarry Smith . -tao_draw_gradient - draw the gradient at each iteration 1838a7e14dcfSSatish Balay 1839a7e14dcfSSatish Balay Level: advanced 1840a7e14dcfSSatish Balay 18411cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGradientMonitor()`, `TaoSetMonitor()`, `TaoDrawSolutionMonitor` 1842a7e14dcfSSatish Balay @*/ 1843d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawGradientMonitor(Tao tao, void *ctx) 1844d71ae5a4SJacob Faibussowitsch { 1845e882e171SHong Zhang TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx; 184647a47007SBarry Smith 1847a7e14dcfSSatish Balay PetscFunctionBegin; 184894511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 18493ba16761SJacob Faibussowitsch if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(PETSC_SUCCESS); 18509566063dSJacob Faibussowitsch PetscCall(VecView(tao->gradient, ictx->viewer)); 18513ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1852a7e14dcfSSatish Balay } 1853a7e14dcfSSatish Balay 1854a7e14dcfSSatish Balay /*@C 185547450a7bSBarry Smith TaoDrawStepMonitor - Plots the step direction at each iteration of `TaoSolve()` 1856a7e14dcfSSatish Balay 1857c3339decSBarry Smith Collective 1858a7e14dcfSSatish Balay 1859a7e14dcfSSatish Balay Input Parameters: 186047450a7bSBarry Smith + tao - the `Tao` context 186147450a7bSBarry Smith - ctx - the `PetscViewer` context 1862a7e14dcfSSatish Balay 186347450a7bSBarry Smith Options Database Key: 1864147403d9SBarry Smith . -tao_draw_step - draw the step direction at each iteration 1865a7e14dcfSSatish Balay 1866a7e14dcfSSatish Balay Level: advanced 1867a7e14dcfSSatish Balay 18681cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMonitor()`, `TaoDrawSolutionMonitor` 1869a7e14dcfSSatish Balay @*/ 1870d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDrawStepMonitor(Tao tao, void *ctx) 1871d71ae5a4SJacob Faibussowitsch { 187294511df7SStefano Zampini PetscViewer viewer = (PetscViewer)ctx; 187347a47007SBarry Smith 1874a7e14dcfSSatish Balay PetscFunctionBegin; 187594511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 187694511df7SStefano Zampini PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 18779566063dSJacob Faibussowitsch PetscCall(VecView(tao->stepdirection, viewer)); 18783ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1879a7e14dcfSSatish Balay } 1880a7e14dcfSSatish Balay 1881a7e14dcfSSatish Balay /*@C 188247450a7bSBarry Smith TaoResidualMonitor - Views the least-squares residual at each iteration of `TaoSolve()` 1883a7e14dcfSSatish Balay 1884c3339decSBarry Smith Collective 1885a7e14dcfSSatish Balay 1886a7e14dcfSSatish Balay Input Parameters: 188747450a7bSBarry Smith + tao - the `Tao` context 188847450a7bSBarry Smith - ctx - the `PetscViewer` context or `NULL` 1889a7e14dcfSSatish Balay 189047450a7bSBarry Smith Options Database Key: 189147450a7bSBarry Smith . -tao_view_ls_residual - view the residual at each iteration 1892a7e14dcfSSatish Balay 1893a7e14dcfSSatish Balay Level: advanced 1894a7e14dcfSSatish Balay 18951cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1896a7e14dcfSSatish Balay @*/ 1897d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoResidualMonitor(Tao tao, void *ctx) 1898d71ae5a4SJacob Faibussowitsch { 1899d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 190047a47007SBarry Smith 1901a7e14dcfSSatish Balay PetscFunctionBegin; 190294511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1903d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 19049566063dSJacob Faibussowitsch PetscCall(VecView(tao->ls_res, viewer)); 19053ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1906a7e14dcfSSatish Balay } 1907a7e14dcfSSatish Balay 19087fab98ebSJason Sarich /*@ 1909a7e14dcfSSatish Balay TaoDefaultConvergenceTest - Determines whether the solver should continue iterating 1910a7e14dcfSSatish Balay or terminate. 1911a7e14dcfSSatish Balay 1912c3339decSBarry Smith Collective 1913a7e14dcfSSatish Balay 1914a7e14dcfSSatish Balay Input Parameters: 191547450a7bSBarry Smith + tao - the `Tao` context 1916a7e14dcfSSatish Balay - dummy - unused dummy context 1917a7e14dcfSSatish Balay 191847450a7bSBarry Smith Level: developer 191947450a7bSBarry Smith 1920a7e14dcfSSatish Balay Notes: 1921a7e14dcfSSatish Balay This routine checks the residual in the optimality conditions, the 1922a7e14dcfSSatish Balay relative residual in the optimity conditions, the number of function 1923a7e14dcfSSatish Balay evaluations, and the function value to test convergence. Some 1924a7e14dcfSSatish Balay solvers may use different convergence routines. 1925a7e14dcfSSatish Balay 19261cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoGetConvergedReason()`, `TaoSetConvergedReason()` 1927a7e14dcfSSatish Balay @*/ 1928d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultConvergenceTest(Tao tao, void *dummy) 1929d71ae5a4SJacob Faibussowitsch { 1930a7e14dcfSSatish Balay PetscInt niter = tao->niter, nfuncs = PetscMax(tao->nfuncs, tao->nfuncgrads); 1931a7e14dcfSSatish Balay PetscInt max_funcs = tao->max_funcs; 1932a7e14dcfSSatish Balay PetscReal gnorm = tao->residual, gnorm0 = tao->gnorm0; 1933a7e14dcfSSatish Balay PetscReal f = tao->fc, steptol = tao->steptol, trradius = tao->step; 1934a7e14dcfSSatish Balay PetscReal gatol = tao->gatol, grtol = tao->grtol, gttol = tao->gttol; 1935e52336cbSBarry Smith PetscReal catol = tao->catol, crtol = tao->crtol; 1936e52336cbSBarry Smith PetscReal fmin = tao->fmin, cnorm = tao->cnorm; 1937e4cb33bbSBarry Smith TaoConvergedReason reason = tao->reason; 1938a7e14dcfSSatish Balay 1939a7e14dcfSSatish Balay PetscFunctionBegin; 1940441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 19413ba16761SJacob Faibussowitsch if (reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(PETSC_SUCCESS); 1942a7e14dcfSSatish Balay 1943a7e14dcfSSatish Balay if (PetscIsInfOrNanReal(f)) { 19449566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Failed to converged, function value is Inf or NaN\n")); 1945a7e14dcfSSatish Balay reason = TAO_DIVERGED_NAN; 1946a7e14dcfSSatish Balay } else if (f <= fmin && cnorm <= catol) { 19479566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Converged due to function value %g < minimum function value %g\n", (double)f, (double)fmin)); 1948a7e14dcfSSatish Balay reason = TAO_CONVERGED_MINF; 1949a7e14dcfSSatish Balay } else if (gnorm <= gatol && cnorm <= catol) { 19509566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Converged due to residual norm ||g(X)||=%g < %g\n", (double)gnorm, (double)gatol)); 1951a7e14dcfSSatish Balay reason = TAO_CONVERGED_GATOL; 1952a7e14dcfSSatish Balay } else if (f != 0 && PetscAbsReal(gnorm / f) <= grtol && cnorm <= crtol) { 19539566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Converged due to residual ||g(X)||/|f(X)| =%g < %g\n", (double)(gnorm / f), (double)grtol)); 1954a7e14dcfSSatish Balay reason = TAO_CONVERGED_GRTOL; 1955e1d16bb9SBarry Smith } else if (gnorm0 != 0 && ((gttol == 0 && gnorm == 0) || gnorm / gnorm0 < gttol) && cnorm <= crtol) { 19569566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Converged due to relative residual norm ||g(X)||/||g(X0)|| = %g < %g\n", (double)(gnorm / gnorm0), (double)gttol)); 1957a7e14dcfSSatish Balay reason = TAO_CONVERGED_GTTOL; 195894511df7SStefano Zampini } else if (max_funcs >= 0 && nfuncs > max_funcs) { 19599566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Exceeded maximum number of function evaluations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", nfuncs, max_funcs)); 1960a7e14dcfSSatish Balay reason = TAO_DIVERGED_MAXFCN; 1961a7e14dcfSSatish Balay } else if (tao->lsflag != 0) { 19629566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Tao Line Search failure.\n")); 1963a7e14dcfSSatish Balay reason = TAO_DIVERGED_LS_FAILURE; 1964a7e14dcfSSatish Balay } else if (trradius < steptol && niter > 0) { 19659566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Trust region/step size too small: %g < %g\n", (double)trradius, (double)steptol)); 1966a7e14dcfSSatish Balay reason = TAO_CONVERGED_STEPTOL; 1967e031d6f5SAlp Dener } else if (niter >= tao->max_it) { 196863a3b9bcSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Exceeded maximum number of iterations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", niter, tao->max_it)); 1969a7e14dcfSSatish Balay reason = TAO_DIVERGED_MAXITS; 1970a7e14dcfSSatish Balay } else { 1971a7e14dcfSSatish Balay reason = TAO_CONTINUE_ITERATING; 1972a7e14dcfSSatish Balay } 1973a7e14dcfSSatish Balay tao->reason = reason; 19743ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1975a7e14dcfSSatish Balay } 1976a7e14dcfSSatish Balay 1977a7e14dcfSSatish Balay /*@C 1978a7e14dcfSSatish Balay TaoSetOptionsPrefix - Sets the prefix used for searching for all 197965ba42b6SBarry Smith Tao options in the database. 1980a7e14dcfSSatish Balay 1981c3339decSBarry Smith Logically Collective 1982a7e14dcfSSatish Balay 1983a7e14dcfSSatish Balay Input Parameters: 198447450a7bSBarry Smith + tao - the `Tao` context 1985e056e8ceSJacob Faibussowitsch - p - the prefix string to prepend to all Tao option requests 1986a7e14dcfSSatish Balay 19872fe279fdSBarry Smith Level: advanced 19882fe279fdSBarry Smith 1989a7e14dcfSSatish Balay Notes: 1990a7e14dcfSSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 1991a7e14dcfSSatish Balay The first character of all runtime options is AUTOMATICALLY the hyphen. 1992a7e14dcfSSatish Balay 1993a7e14dcfSSatish Balay For example, to distinguish between the runtime options for two 199465ba42b6SBarry Smith different Tao solvers, one could call 1995a7e14dcfSSatish Balay .vb 1996a7e14dcfSSatish Balay TaoSetOptionsPrefix(tao1,"sys1_") 1997a7e14dcfSSatish Balay TaoSetOptionsPrefix(tao2,"sys2_") 1998a7e14dcfSSatish Balay .ve 1999a7e14dcfSSatish Balay 2000a7e14dcfSSatish Balay This would enable use of different options for each system, such as 2001a7e14dcfSSatish Balay .vb 20029fa2b5dcSStefano Zampini -sys1_tao_method blmvm -sys1_tao_grtol 1.e-3 20039fa2b5dcSStefano Zampini -sys2_tao_method lmvm -sys2_tao_grtol 1.e-4 2004a7e14dcfSSatish Balay .ve 2005a7e14dcfSSatish Balay 20061cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoAppendOptionsPrefix()`, `TaoGetOptionsPrefix()` 2007a7e14dcfSSatish Balay @*/ 2008d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetOptionsPrefix(Tao tao, const char p[]) 2009d71ae5a4SJacob Faibussowitsch { 2010a7e14dcfSSatish Balay PetscFunctionBegin; 201194511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 20129566063dSJacob Faibussowitsch PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tao, p)); 20131baa6e33SBarry Smith if (tao->linesearch) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch, p)); 20141baa6e33SBarry Smith if (tao->ksp) PetscCall(KSPSetOptionsPrefix(tao->ksp, p)); 20153ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2016a7e14dcfSSatish Balay } 2017a7e14dcfSSatish Balay 2018a7e14dcfSSatish Balay /*@C 201947450a7bSBarry Smith TaoAppendOptionsPrefix - Appends to the prefix used for searching for all Tao options in the database. 2020a7e14dcfSSatish Balay 2021c3339decSBarry Smith Logically Collective 2022a7e14dcfSSatish Balay 2023a7e14dcfSSatish Balay Input Parameters: 202447450a7bSBarry Smith + tao - the `Tao` solver context 2025e056e8ceSJacob Faibussowitsch - p - the prefix string to prepend to all `Tao` option requests 2026a7e14dcfSSatish Balay 20272fe279fdSBarry Smith Level: advanced 20282fe279fdSBarry Smith 202965ba42b6SBarry Smith Note: 2030a7e14dcfSSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 203165ba42b6SBarry Smith The first character of all runtime options is automatically the hyphen. 2032a7e14dcfSSatish Balay 20331cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoSetOptionsPrefix()`, `TaoGetOptionsPrefix()` 2034a7e14dcfSSatish Balay @*/ 2035d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoAppendOptionsPrefix(Tao tao, const char p[]) 2036d71ae5a4SJacob Faibussowitsch { 2037a7e14dcfSSatish Balay PetscFunctionBegin; 203894511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 20399566063dSJacob Faibussowitsch PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao, p)); 20401baa6e33SBarry Smith if (tao->linesearch) PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao->linesearch, p)); 20411baa6e33SBarry Smith if (tao->ksp) PetscCall(KSPAppendOptionsPrefix(tao->ksp, p)); 20423ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2043a7e14dcfSSatish Balay } 2044a7e14dcfSSatish Balay 2045a7e14dcfSSatish Balay /*@C 2046a7e14dcfSSatish Balay TaoGetOptionsPrefix - Gets the prefix used for searching for all 204765ba42b6SBarry Smith Tao options in the database 2048a7e14dcfSSatish Balay 2049a7e14dcfSSatish Balay Not Collective 2050a7e14dcfSSatish Balay 205147450a7bSBarry Smith Input Parameter: 205247450a7bSBarry Smith . tao - the `Tao` context 2053a7e14dcfSSatish Balay 205447450a7bSBarry Smith Output Parameter: 2055e056e8ceSJacob Faibussowitsch . p - pointer to the prefix string used is returned 2056a7e14dcfSSatish Balay 2057e056e8ceSJacob Faibussowitsch Fortran Notes: 205847450a7bSBarry Smith Pass in a string 'prefix' of sufficient length to hold the prefix. 2059a7e14dcfSSatish Balay 2060a7e14dcfSSatish Balay Level: advanced 2061a7e14dcfSSatish Balay 20621cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoSetOptionsPrefix()`, `TaoAppendOptionsPrefix()` 2063a7e14dcfSSatish Balay @*/ 2064d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetOptionsPrefix(Tao tao, const char *p[]) 2065d71ae5a4SJacob Faibussowitsch { 206694511df7SStefano Zampini PetscFunctionBegin; 206794511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 20689566063dSJacob Faibussowitsch PetscCall(PetscObjectGetOptionsPrefix((PetscObject)tao, p)); 20693ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2070a7e14dcfSSatish Balay } 2071a7e14dcfSSatish Balay 2072a7e14dcfSSatish Balay /*@C 207347450a7bSBarry Smith TaoSetType - Sets the `TaoType` for the minimization solver. 2074a7e14dcfSSatish Balay 2075c3339decSBarry Smith Collective 2076a7e14dcfSSatish Balay 2077a7e14dcfSSatish Balay Input Parameters: 2078e056e8ceSJacob Faibussowitsch + tao - the `Tao` solver context 2079a7e14dcfSSatish Balay - type - a known method 2080a7e14dcfSSatish Balay 2081a7e14dcfSSatish Balay Options Database Key: 208258417fe7SBarry Smith . -tao_type <type> - Sets the method; use -help for a list 208358417fe7SBarry Smith of available methods (for instance, "-tao_type lmvm" or "-tao_type tron") 2084a7e14dcfSSatish Balay 2085a7e14dcfSSatish Balay Level: intermediate 2086a7e14dcfSSatish Balay 20871cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoGetType()`, `TaoType` 2088a7e14dcfSSatish Balay @*/ 2089d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetType(Tao tao, TaoType type) 2090d71ae5a4SJacob Faibussowitsch { 2091441846f8SBarry Smith PetscErrorCode (*create_xxx)(Tao); 2092a7e14dcfSSatish Balay PetscBool issame; 2093a7e14dcfSSatish Balay 2094a7e14dcfSSatish Balay PetscFunctionBegin; 2095441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2096a7e14dcfSSatish Balay 20979566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)tao, type, &issame)); 20983ba16761SJacob Faibussowitsch if (issame) PetscFunctionReturn(PETSC_SUCCESS); 2099a7e14dcfSSatish Balay 21009566063dSJacob Faibussowitsch PetscCall(PetscFunctionListFind(TaoList, type, (void (**)(void)) & create_xxx)); 21013c859ba3SBarry Smith PetscCheck(create_xxx, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unable to find requested Tao type %s", type); 2102a7e14dcfSSatish Balay 2103a7e14dcfSSatish Balay /* Destroy the existing solver information */ 2104dbbe0bcdSBarry Smith PetscTryTypeMethod(tao, destroy); 21051baa6e33SBarry Smith PetscCall(KSPDestroy(&tao->ksp)); 21069566063dSJacob Faibussowitsch PetscCall(TaoLineSearchDestroy(&tao->linesearch)); 210783c8fe1dSLisandro Dalcin tao->ops->setup = NULL; 210883c8fe1dSLisandro Dalcin tao->ops->solve = NULL; 210983c8fe1dSLisandro Dalcin tao->ops->view = NULL; 211083c8fe1dSLisandro Dalcin tao->ops->setfromoptions = NULL; 211183c8fe1dSLisandro Dalcin tao->ops->destroy = NULL; 2112a7e14dcfSSatish Balay 2113a7e14dcfSSatish Balay tao->setupcalled = PETSC_FALSE; 2114a7e14dcfSSatish Balay 21159566063dSJacob Faibussowitsch PetscCall((*create_xxx)(tao)); 21169566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject)tao, type)); 21173ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2118a7e14dcfSSatish Balay } 211947a47007SBarry Smith 212067be906fSBarry Smith /*@C 212147450a7bSBarry Smith TaoRegister - Adds a method to the Tao package for minimization. 2122a7e14dcfSSatish Balay 212320f4b53cSBarry Smith Not Collective 2124a7e14dcfSSatish Balay 2125a7e14dcfSSatish Balay Input Parameters: 2126a7e14dcfSSatish Balay + sname - name of a new user-defined solver 2127a7e14dcfSSatish Balay - func - routine to Create method context 2128a7e14dcfSSatish Balay 2129e056e8ceSJacob Faibussowitsch Example Usage: 2130a7e14dcfSSatish Balay .vb 2131441846f8SBarry Smith TaoRegister("my_solver", MySolverCreate); 2132a7e14dcfSSatish Balay .ve 2133a7e14dcfSSatish Balay 2134a7e14dcfSSatish Balay Then, your solver can be chosen with the procedural interface via 2135a7e14dcfSSatish Balay $ TaoSetType(tao, "my_solver") 2136a7e14dcfSSatish Balay or at runtime via the option 213758417fe7SBarry Smith $ -tao_type my_solver 2138a7e14dcfSSatish Balay 2139a7e14dcfSSatish Balay Level: advanced 2140a7e14dcfSSatish Balay 214167be906fSBarry Smith Note: 214267be906fSBarry Smith `TaoRegister()` may be called multiple times to add several user-defined solvers. 214367be906fSBarry Smith 21441cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetType()`, `TaoRegisterAll()`, `TaoRegisterDestroy()` 214567be906fSBarry Smith @*/ 2146d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoRegister(const char sname[], PetscErrorCode (*func)(Tao)) 2147d71ae5a4SJacob Faibussowitsch { 2148a7e14dcfSSatish Balay PetscFunctionBegin; 21499566063dSJacob Faibussowitsch PetscCall(TaoInitializePackage()); 21509566063dSJacob Faibussowitsch PetscCall(PetscFunctionListAdd(&TaoList, sname, (void (*)(void))func)); 21513ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2152a7e14dcfSSatish Balay } 2153a7e14dcfSSatish Balay 2154a7e14dcfSSatish Balay /*@C 2155441846f8SBarry Smith TaoRegisterDestroy - Frees the list of minimization solvers that were 215647450a7bSBarry Smith registered by `TaoRegister()`. 2157a7e14dcfSSatish Balay 2158a7e14dcfSSatish Balay Not Collective 2159a7e14dcfSSatish Balay 2160a7e14dcfSSatish Balay Level: advanced 2161a7e14dcfSSatish Balay 21621cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoRegisterAll()`, `TaoRegister()` 2163a7e14dcfSSatish Balay @*/ 2164d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoRegisterDestroy(void) 2165d71ae5a4SJacob Faibussowitsch { 2166a7e14dcfSSatish Balay PetscFunctionBegin; 21679566063dSJacob Faibussowitsch PetscCall(PetscFunctionListDestroy(&TaoList)); 2168441846f8SBarry Smith TaoRegisterAllCalled = PETSC_FALSE; 21693ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2170a7e14dcfSSatish Balay } 217147a47007SBarry Smith 21728931d482SJason Sarich /*@ 217347450a7bSBarry Smith TaoGetIterationNumber - Gets the number of `TaoSolve()` iterations completed 21748931d482SJason Sarich at this time. 21758931d482SJason Sarich 21768931d482SJason Sarich Not Collective 21778931d482SJason Sarich 21788931d482SJason Sarich Input Parameter: 217947450a7bSBarry Smith . tao - the `Tao` context 21808931d482SJason Sarich 21818931d482SJason Sarich Output Parameter: 21828931d482SJason Sarich . iter - iteration number 21838931d482SJason Sarich 21848931d482SJason Sarich Notes: 21858931d482SJason Sarich For example, during the computation of iteration 2 this would return 1. 21868931d482SJason Sarich 21878931d482SJason Sarich Level: intermediate 21888931d482SJason Sarich 21891cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`, `TaoGetResidualNorm()`, `TaoGetObjective()` 21908931d482SJason Sarich @*/ 2191d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetIterationNumber(Tao tao, PetscInt *iter) 2192d71ae5a4SJacob Faibussowitsch { 21938931d482SJason Sarich PetscFunctionBegin; 21948931d482SJason Sarich PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 21954f572ea9SToby Isaac PetscAssertPointer(iter, 2); 21968931d482SJason Sarich *iter = tao->niter; 21973ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 21988931d482SJason Sarich } 21998931d482SJason Sarich 22008931d482SJason Sarich /*@ 220147450a7bSBarry Smith TaoGetResidualNorm - Gets the current value of the norm of the residual (gradient) 220279f5d8caSBarry Smith at this time. 220379f5d8caSBarry Smith 220479f5d8caSBarry Smith Not Collective 220579f5d8caSBarry Smith 220679f5d8caSBarry Smith Input Parameter: 220747450a7bSBarry Smith . tao - the `Tao` context 220879f5d8caSBarry Smith 220979f5d8caSBarry Smith Output Parameter: 221079f5d8caSBarry Smith . value - the current value 221179f5d8caSBarry Smith 221279f5d8caSBarry Smith Level: intermediate 221379f5d8caSBarry Smith 2214e056e8ceSJacob Faibussowitsch Developer Notes: 221567be906fSBarry Smith This is the 2-norm of the residual, we cannot use `TaoGetGradientNorm()` because that has 221647450a7bSBarry Smith a different meaning. For some reason `Tao` sometimes calls the gradient the residual. 221779f5d8caSBarry Smith 22181cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`, `TaoGetIterationNumber()`, `TaoGetObjective()` 221979f5d8caSBarry Smith @*/ 2220d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetResidualNorm(Tao tao, PetscReal *value) 2221d71ae5a4SJacob Faibussowitsch { 222279f5d8caSBarry Smith PetscFunctionBegin; 222379f5d8caSBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 22244f572ea9SToby Isaac PetscAssertPointer(value, 2); 222579f5d8caSBarry Smith *value = tao->residual; 22263ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 222779f5d8caSBarry Smith } 222879f5d8caSBarry Smith 222979f5d8caSBarry Smith /*@ 22308931d482SJason Sarich TaoSetIterationNumber - Sets the current iteration number. 22318931d482SJason Sarich 2232c3339decSBarry Smith Logically Collective 22338931d482SJason Sarich 2234d8d19677SJose E. Roman Input Parameters: 223547450a7bSBarry Smith + tao - the `Tao` context 2236a2b725a8SWilliam Gropp - iter - iteration number 22378931d482SJason Sarich 22388931d482SJason Sarich Level: developer 22398931d482SJason Sarich 22401cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()` 22418931d482SJason Sarich @*/ 2242d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetIterationNumber(Tao tao, PetscInt iter) 2243d71ae5a4SJacob Faibussowitsch { 22448931d482SJason Sarich PetscFunctionBegin; 22458931d482SJason Sarich PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 224694511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao, iter, 2); 22479566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao)); 22488931d482SJason Sarich tao->niter = iter; 22499566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao)); 22503ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 22518931d482SJason Sarich } 22528931d482SJason Sarich 22538931d482SJason Sarich /*@ 225447450a7bSBarry Smith TaoGetTotalIterationNumber - Gets the total number of `TaoSolve()` iterations 22558931d482SJason Sarich completed. This number keeps accumulating if multiple solves 225647450a7bSBarry Smith are called with the `Tao` object. 22578931d482SJason Sarich 22588931d482SJason Sarich Not Collective 22598931d482SJason Sarich 22608931d482SJason Sarich Input Parameter: 226147450a7bSBarry Smith . tao - the `Tao` context 22628931d482SJason Sarich 22638931d482SJason Sarich Output Parameter: 226447450a7bSBarry Smith . iter - number of iterations 22658931d482SJason Sarich 22668931d482SJason Sarich Level: intermediate 22678931d482SJason Sarich 226820f4b53cSBarry Smith Note: 226967be906fSBarry Smith The total iteration count is updated after each solve, if there is a current 227047450a7bSBarry Smith `TaoSolve()` in progress then those iterations are not included in the count 227167be906fSBarry Smith 22721cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()` 22738931d482SJason Sarich @*/ 2274d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetTotalIterationNumber(Tao tao, PetscInt *iter) 2275d71ae5a4SJacob Faibussowitsch { 22768931d482SJason Sarich PetscFunctionBegin; 22778931d482SJason Sarich PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 22784f572ea9SToby Isaac PetscAssertPointer(iter, 2); 22798931d482SJason Sarich *iter = tao->ntotalits; 22803ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 22818931d482SJason Sarich } 22828931d482SJason Sarich 22838931d482SJason Sarich /*@ 22848931d482SJason Sarich TaoSetTotalIterationNumber - Sets the current total iteration number. 22858931d482SJason Sarich 2286c3339decSBarry Smith Logically Collective 22878931d482SJason Sarich 2288d8d19677SJose E. Roman Input Parameters: 228947450a7bSBarry Smith + tao - the `Tao` context 229047450a7bSBarry Smith - iter - the iteration number 22918931d482SJason Sarich 22928931d482SJason Sarich Level: developer 22938931d482SJason Sarich 22941cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()` 22958931d482SJason Sarich @*/ 2296d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetTotalIterationNumber(Tao tao, PetscInt iter) 2297d71ae5a4SJacob Faibussowitsch { 22988931d482SJason Sarich PetscFunctionBegin; 22998931d482SJason Sarich PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 230094511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao, iter, 2); 23019566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao)); 23028931d482SJason Sarich tao->ntotalits = iter; 23039566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao)); 23043ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 23058931d482SJason Sarich } 23068931d482SJason Sarich 2307a7e14dcfSSatish Balay /*@ 230847450a7bSBarry Smith TaoSetConvergedReason - Sets the termination flag on a `Tao` object 2309a7e14dcfSSatish Balay 2310c3339decSBarry Smith Logically Collective 2311a7e14dcfSSatish Balay 2312a7e14dcfSSatish Balay Input Parameters: 231347450a7bSBarry Smith + tao - the `Tao` context 231447450a7bSBarry Smith - reason - the `TaoConvergedReason` 2315a7e14dcfSSatish Balay 2316a7e14dcfSSatish Balay Level: intermediate 2317a7e14dcfSSatish Balay 23181cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason` 2319a7e14dcfSSatish Balay @*/ 2320d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergedReason(Tao tao, TaoConvergedReason reason) 2321d71ae5a4SJacob Faibussowitsch { 2322a7e14dcfSSatish Balay PetscFunctionBegin; 232394511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 232494511df7SStefano Zampini PetscValidLogicalCollectiveEnum(tao, reason, 2); 2325a7e14dcfSSatish Balay tao->reason = reason; 23263ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2327a7e14dcfSSatish Balay } 2328a7e14dcfSSatish Balay 2329a7e14dcfSSatish Balay /*@ 233047450a7bSBarry Smith TaoGetConvergedReason - Gets the reason the `TaoSolve()` was stopped. 2331a7e14dcfSSatish Balay 2332a7e14dcfSSatish Balay Not Collective 2333a7e14dcfSSatish Balay 2334a7e14dcfSSatish Balay Input Parameter: 233547450a7bSBarry Smith . tao - the `Tao` solver context 2336a7e14dcfSSatish Balay 2337a7e14dcfSSatish Balay Output Parameter: 233847450a7bSBarry Smith . reason - value of `TaoConvergedReason` 2339a7e14dcfSSatish Balay 2340a7e14dcfSSatish Balay Level: intermediate 2341a7e14dcfSSatish Balay 23421cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetConvergenceTest()`, `TaoSetTolerances()` 2343a7e14dcfSSatish Balay @*/ 2344d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConvergedReason(Tao tao, TaoConvergedReason *reason) 2345d71ae5a4SJacob Faibussowitsch { 2346a7e14dcfSSatish Balay PetscFunctionBegin; 2347441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 23484f572ea9SToby Isaac PetscAssertPointer(reason, 2); 2349a7e14dcfSSatish Balay *reason = tao->reason; 23503ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2351a7e14dcfSSatish Balay } 2352a7e14dcfSSatish Balay 2353a7e14dcfSSatish Balay /*@ 2354a7e14dcfSSatish Balay TaoGetSolutionStatus - Get the current iterate, objective value, 235547450a7bSBarry Smith residual, infeasibility, and termination from a `Tao` object 2356a7e14dcfSSatish Balay 2357a7e14dcfSSatish Balay Not Collective 2358a7e14dcfSSatish Balay 2359f899ff85SJose E. Roman Input Parameter: 236047450a7bSBarry Smith . tao - the `Tao` context 2361a7e14dcfSSatish Balay 2362a7e14dcfSSatish Balay Output Parameters: 2363e056e8ceSJacob Faibussowitsch + its - the current iterate number (>=0) 2364a7e14dcfSSatish Balay . f - the current function value 236558417fe7SBarry Smith . gnorm - the square of the gradient norm, duality gap, or other measure indicating distance from optimality. 2366a7e14dcfSSatish Balay . cnorm - the infeasibility of the current solution with regard to the constraints. 2367a7e14dcfSSatish Balay . xdiff - the step length or trust region radius of the most recent iterate. 236865ba42b6SBarry Smith - reason - The termination reason, which can equal `TAO_CONTINUE_ITERATING` 2369a7e14dcfSSatish Balay 2370a7e14dcfSSatish Balay Level: intermediate 2371a7e14dcfSSatish Balay 237265ba42b6SBarry Smith Notes: 237365ba42b6SBarry Smith Tao returns the values set by the solvers in the routine `TaoMonitor()`. 2374a7e14dcfSSatish Balay 237567be906fSBarry Smith If any of the output arguments are set to `NULL`, no corresponding value will be returned. 2376a7e14dcfSSatish Balay 23771cc06b55SBarry Smith .seealso: [](ch_tao), `TaoMonitor()`, `TaoGetConvergedReason()` 2378a7e14dcfSSatish Balay @*/ 2379d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetSolutionStatus(Tao tao, PetscInt *its, PetscReal *f, PetscReal *gnorm, PetscReal *cnorm, PetscReal *xdiff, TaoConvergedReason *reason) 2380d71ae5a4SJacob Faibussowitsch { 2381a7e14dcfSSatish Balay PetscFunctionBegin; 238294511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2383a7e14dcfSSatish Balay if (its) *its = tao->niter; 2384a7e14dcfSSatish Balay if (f) *f = tao->fc; 2385a7e14dcfSSatish Balay if (gnorm) *gnorm = tao->residual; 2386a7e14dcfSSatish Balay if (cnorm) *cnorm = tao->cnorm; 2387a7e14dcfSSatish Balay if (reason) *reason = tao->reason; 2388a7e14dcfSSatish Balay if (xdiff) *xdiff = tao->step; 23893ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2390a7e14dcfSSatish Balay } 2391a7e14dcfSSatish Balay 2392a7e14dcfSSatish Balay /*@C 239347450a7bSBarry Smith TaoGetType - Gets the current `TaoType` being used in the `Tao` object 2394a7e14dcfSSatish Balay 2395a7e14dcfSSatish Balay Not Collective 2396a7e14dcfSSatish Balay 2397a7e14dcfSSatish Balay Input Parameter: 239847450a7bSBarry Smith . tao - the `Tao` solver context 2399a7e14dcfSSatish Balay 2400a7e14dcfSSatish Balay Output Parameter: 240147450a7bSBarry Smith . type - the `TaoType` 2402a7e14dcfSSatish Balay 2403a7e14dcfSSatish Balay Level: intermediate 2404a7e14dcfSSatish Balay 24051cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoType`, `TaoSetType()` 2406a7e14dcfSSatish Balay @*/ 2407d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetType(Tao tao, TaoType *type) 2408d71ae5a4SJacob Faibussowitsch { 2409a7e14dcfSSatish Balay PetscFunctionBegin; 2410441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 24114f572ea9SToby Isaac PetscAssertPointer(type, 2); 2412a7e14dcfSSatish Balay *type = ((PetscObject)tao)->type_name; 24133ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2414a7e14dcfSSatish Balay } 2415a7e14dcfSSatish Balay 2416a7e14dcfSSatish Balay /*@C 2417a7e14dcfSSatish Balay TaoMonitor - Monitor the solver and the current solution. This 2418a7e14dcfSSatish Balay routine will record the iteration number and residual statistics, 2419670c031eSStefano Zampini and call any monitors specified by the user. 2420a7e14dcfSSatish Balay 2421a7e14dcfSSatish Balay Input Parameters: 242247450a7bSBarry Smith + tao - the `Tao` context 2423a7e14dcfSSatish Balay . its - the current iterate number (>=0) 2424a7e14dcfSSatish Balay . f - the current objective function value 24251cb3f120SPierre Jolivet . res - the gradient norm, square root of the duality gap, or other measure indicating distance from optimality. This measure will be recorded and 2426a7e14dcfSSatish Balay used for some termination tests. 2427a7e14dcfSSatish Balay . cnorm - the infeasibility of the current solution with regard to the constraints. 2428a7e14dcfSSatish Balay - steplength - multiple of the step direction added to the previous iterate. 2429a7e14dcfSSatish Balay 2430a7e14dcfSSatish Balay Options Database Key: 2431a7e14dcfSSatish Balay . -tao_monitor - Use the default monitor, which prints statistics to standard output 2432a7e14dcfSSatish Balay 2433a7e14dcfSSatish Balay Level: developer 2434a7e14dcfSSatish Balay 24351cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetConvergedReason()`, `TaoMonitorDefault()`, `TaoSetMonitor()` 2436a7e14dcfSSatish Balay @*/ 2437d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitor(Tao tao, PetscInt its, PetscReal f, PetscReal res, PetscReal cnorm, PetscReal steplength) 2438d71ae5a4SJacob Faibussowitsch { 243947a47007SBarry Smith PetscInt i; 244047a47007SBarry Smith 2441a7e14dcfSSatish Balay PetscFunctionBegin; 2442441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2443a7e14dcfSSatish Balay tao->fc = f; 2444a7e14dcfSSatish Balay tao->residual = res; 2445a7e14dcfSSatish Balay tao->cnorm = cnorm; 2446a7e14dcfSSatish Balay tao->step = steplength; 2447e52336cbSBarry Smith if (!its) { 244894511df7SStefano Zampini tao->cnorm0 = cnorm; 244994511df7SStefano Zampini tao->gnorm0 = res; 2450a7e14dcfSSatish Balay } 2451bfdcf5a2SStefano Zampini PetscCall(VecLockReadPush(tao->solution)); 245248a46eb9SPierre Jolivet for (i = 0; i < tao->numbermonitors; i++) PetscCall((*tao->monitor[i])(tao, tao->monitorcontext[i])); 2453bfdcf5a2SStefano Zampini PetscCall(VecLockReadPop(tao->solution)); 24543ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2455a7e14dcfSSatish Balay } 2456a7e14dcfSSatish Balay 2457a7e14dcfSSatish Balay /*@ 2458ae93cb3cSJason Sarich TaoSetConvergenceHistory - Sets the array used to hold the convergence history. 2459a7e14dcfSSatish Balay 2460c3339decSBarry Smith Logically Collective 2461a7e14dcfSSatish Balay 2462a7e14dcfSSatish Balay Input Parameters: 246347450a7bSBarry Smith + tao - the `Tao` solver context 2464a7e14dcfSSatish Balay . obj - array to hold objective value history 2465a7e14dcfSSatish Balay . resid - array to hold residual history 2466a7e14dcfSSatish Balay . cnorm - array to hold constraint violation history 2467be1558d8SJason Sarich . lits - integer array holds the number of linear iterations for each Tao iteration 246867be906fSBarry Smith . na - size of `obj`, `resid`, and `cnorm` 246965ba42b6SBarry Smith - reset - `PETSC_TRUE` indicates each new minimization resets the history counter to zero, 2470a7e14dcfSSatish Balay else it continues storing new values for new minimizations after the old ones 2471a7e14dcfSSatish Balay 247267be906fSBarry Smith Level: intermediate 247367be906fSBarry Smith 2474a7e14dcfSSatish Balay Notes: 247547450a7bSBarry Smith If set, `Tao` will fill the given arrays with the indicated 2476ae93cb3cSJason Sarich information at each iteration. If 'obj','resid','cnorm','lits' are 247767be906fSBarry Smith *all* `NULL` then space (using size `na`, or 1000 if na is `PETSC_DECIDE` or 247865ba42b6SBarry Smith `PETSC_DEFAULT`) is allocated for the history. 247967be906fSBarry Smith If not all are `NULL`, then only the non-`NULL` information categories 2480ae93cb3cSJason Sarich will be stored, the others will be ignored. 2481ae93cb3cSJason Sarich 2482ae93cb3cSJason Sarich Any convergence information after iteration number 'na' will not be stored. 2483a7e14dcfSSatish Balay 2484a7e14dcfSSatish Balay This routine is useful, e.g., when running a code for purposes 2485a7e14dcfSSatish Balay of accurate performance monitoring, when no I/O should be done 2486a7e14dcfSSatish Balay during the section of code that is being timed. 2487a7e14dcfSSatish Balay 24881cc06b55SBarry Smith .seealso: [](ch_tao), `TaoGetConvergenceHistory()` 2489a7e14dcfSSatish Balay @*/ 2490d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergenceHistory(Tao tao, PetscReal obj[], PetscReal resid[], PetscReal cnorm[], PetscInt lits[], PetscInt na, PetscBool reset) 2491d71ae5a4SJacob Faibussowitsch { 2492a7e14dcfSSatish Balay PetscFunctionBegin; 2493441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 24944f572ea9SToby Isaac if (obj) PetscAssertPointer(obj, 2); 24954f572ea9SToby Isaac if (resid) PetscAssertPointer(resid, 3); 24964f572ea9SToby Isaac if (cnorm) PetscAssertPointer(cnorm, 4); 24974f572ea9SToby Isaac if (lits) PetscAssertPointer(lits, 5); 2498ae93cb3cSJason Sarich 2499ae93cb3cSJason Sarich if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000; 2500ae93cb3cSJason Sarich if (!obj && !resid && !cnorm && !lits) { 25019566063dSJacob Faibussowitsch PetscCall(PetscCalloc4(na, &obj, na, &resid, na, &cnorm, na, &lits)); 2502ae93cb3cSJason Sarich tao->hist_malloc = PETSC_TRUE; 2503ae93cb3cSJason Sarich } 2504ae93cb3cSJason Sarich 2505a7e14dcfSSatish Balay tao->hist_obj = obj; 2506a7e14dcfSSatish Balay tao->hist_resid = resid; 2507a7e14dcfSSatish Balay tao->hist_cnorm = cnorm; 2508ae93cb3cSJason Sarich tao->hist_lits = lits; 2509a7e14dcfSSatish Balay tao->hist_max = na; 2510a7e14dcfSSatish Balay tao->hist_reset = reset; 2511ae93cb3cSJason Sarich tao->hist_len = 0; 25123ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2513a7e14dcfSSatish Balay } 2514a7e14dcfSSatish Balay 2515a7e14dcfSSatish Balay /*@C 251665ba42b6SBarry Smith TaoGetConvergenceHistory - Gets the arrays used that hold the convergence history. 2517a7e14dcfSSatish Balay 2518c3339decSBarry Smith Collective 2519a7e14dcfSSatish Balay 2520a7e14dcfSSatish Balay Input Parameter: 252147450a7bSBarry Smith . tao - the `Tao` context 2522a7e14dcfSSatish Balay 2523a7e14dcfSSatish Balay Output Parameters: 2524a7e14dcfSSatish Balay + obj - array used to hold objective value history 2525a7e14dcfSSatish Balay . resid - array used to hold residual history 2526a7e14dcfSSatish Balay . cnorm - array used to hold constraint violation history 2527ae93cb3cSJason Sarich . lits - integer array used to hold linear solver iteration count 252867be906fSBarry Smith - nhist - size of `obj`, `resid`, `cnorm`, and `lits` 252967be906fSBarry Smith 253067be906fSBarry Smith Level: advanced 2531a7e14dcfSSatish Balay 2532a7e14dcfSSatish Balay Notes: 253365ba42b6SBarry Smith This routine must be preceded by calls to `TaoSetConvergenceHistory()` 253465ba42b6SBarry Smith and `TaoSolve()`, otherwise it returns useless information. 2535ae93cb3cSJason Sarich 2536a7e14dcfSSatish Balay This routine is useful, e.g., when running a code for purposes 2537a7e14dcfSSatish Balay of accurate performance monitoring, when no I/O should be done 2538a7e14dcfSSatish Balay during the section of code that is being timed. 2539a7e14dcfSSatish Balay 2540e056e8ceSJacob Faibussowitsch Fortran Notes: 254167be906fSBarry Smith The calling sequence is 254267be906fSBarry Smith .vb 254367be906fSBarry Smith call TaoGetConvergenceHistory(Tao tao, PetscInt nhist, PetscErrorCode ierr) 254467be906fSBarry Smith .ve 2545a7e14dcfSSatish Balay 25461cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoSetConvergenceHistory()` 2547a7e14dcfSSatish Balay @*/ 2548d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConvergenceHistory(Tao tao, PetscReal **obj, PetscReal **resid, PetscReal **cnorm, PetscInt **lits, PetscInt *nhist) 2549d71ae5a4SJacob Faibussowitsch { 2550a7e14dcfSSatish Balay PetscFunctionBegin; 2551441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2552a7e14dcfSSatish Balay if (obj) *obj = tao->hist_obj; 2553a7e14dcfSSatish Balay if (cnorm) *cnorm = tao->hist_cnorm; 2554a7e14dcfSSatish Balay if (resid) *resid = tao->hist_resid; 25555b494b05SBarry Smith if (lits) *lits = tao->hist_lits; 2556a7e14dcfSSatish Balay if (nhist) *nhist = tao->hist_len; 25573ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2558a7e14dcfSSatish Balay } 2559a7e14dcfSSatish Balay 2560a7e14dcfSSatish Balay /*@ 256147450a7bSBarry Smith TaoSetApplicationContext - Sets the optional user-defined context for a `Tao` solver. 2562a7e14dcfSSatish Balay 2563c3339decSBarry Smith Logically Collective 2564a7e14dcfSSatish Balay 2565a7e14dcfSSatish Balay Input Parameters: 256647450a7bSBarry Smith + tao - the `Tao` context 2567a7e14dcfSSatish Balay - usrP - optional user context 2568a7e14dcfSSatish Balay 2569a7e14dcfSSatish Balay Level: intermediate 2570a7e14dcfSSatish Balay 257142747ad1SJacob Faibussowitsch .seealso: [](ch_tao), `Tao`, `TaoGetApplicationContext()` 2572a7e14dcfSSatish Balay @*/ 2573d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetApplicationContext(Tao tao, void *usrP) 2574d71ae5a4SJacob Faibussowitsch { 2575a7e14dcfSSatish Balay PetscFunctionBegin; 2576441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2577a7e14dcfSSatish Balay tao->user = usrP; 25783ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2579a7e14dcfSSatish Balay } 2580a7e14dcfSSatish Balay 2581a7e14dcfSSatish Balay /*@ 258247450a7bSBarry Smith TaoGetApplicationContext - Gets the user-defined context for a `Tao` solver 2583a7e14dcfSSatish Balay 2584a7e14dcfSSatish Balay Not Collective 2585a7e14dcfSSatish Balay 2586a7e14dcfSSatish Balay Input Parameter: 258747450a7bSBarry Smith . tao - the `Tao` context 2588a7e14dcfSSatish Balay 2589a7e14dcfSSatish Balay Output Parameter: 2590a7e14dcfSSatish Balay . usrP - user context 2591a7e14dcfSSatish Balay 2592a7e14dcfSSatish Balay Level: intermediate 2593a7e14dcfSSatish Balay 25941cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetApplicationContext()` 2595a7e14dcfSSatish Balay @*/ 2596d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetApplicationContext(Tao tao, void *usrP) 2597d71ae5a4SJacob Faibussowitsch { 2598a7e14dcfSSatish Balay PetscFunctionBegin; 2599441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 26004f572ea9SToby Isaac PetscAssertPointer(usrP, 2); 2601a7e14dcfSSatish Balay *(void **)usrP = tao->user; 26023ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2603a7e14dcfSSatish Balay } 2604a9603a14SPatrick Farrell 2605a9603a14SPatrick Farrell /*@ 260665ba42b6SBarry Smith TaoSetGradientNorm - Sets the matrix used to define the norm that measures the size of the gradient. 2607a9603a14SPatrick Farrell 2608c3339decSBarry Smith Collective 2609a9603a14SPatrick Farrell 2610a9603a14SPatrick Farrell Input Parameters: 261147450a7bSBarry Smith + tao - the `Tao` context 261265ba42b6SBarry Smith - M - matrix that defines the norm 2613a9603a14SPatrick Farrell 2614a9603a14SPatrick Farrell Level: beginner 2615a9603a14SPatrick Farrell 26161cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetGradientNorm()`, `TaoGradientNorm()` 2617a9603a14SPatrick Farrell @*/ 2618d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetGradientNorm(Tao tao, Mat M) 2619d71ae5a4SJacob Faibussowitsch { 2620a9603a14SPatrick Farrell PetscFunctionBegin; 2621a9603a14SPatrick Farrell PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 262294511df7SStefano Zampini PetscValidHeaderSpecific(M, MAT_CLASSID, 2); 26239566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)M)); 26249566063dSJacob Faibussowitsch PetscCall(MatDestroy(&tao->gradient_norm)); 26259566063dSJacob Faibussowitsch PetscCall(VecDestroy(&tao->gradient_norm_tmp)); 2626a9603a14SPatrick Farrell tao->gradient_norm = M; 26279566063dSJacob Faibussowitsch PetscCall(MatCreateVecs(M, NULL, &tao->gradient_norm_tmp)); 26283ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2629a9603a14SPatrick Farrell } 2630a9603a14SPatrick Farrell 2631a9603a14SPatrick Farrell /*@ 263265ba42b6SBarry Smith TaoGetGradientNorm - Returns the matrix used to define the norm used for measuring the size of the gradient. 2633a9603a14SPatrick Farrell 2634a9603a14SPatrick Farrell Not Collective 2635a9603a14SPatrick Farrell 2636a9603a14SPatrick Farrell Input Parameter: 263747450a7bSBarry Smith . tao - the `Tao` context 2638a9603a14SPatrick Farrell 2639a9603a14SPatrick Farrell Output Parameter: 2640a9603a14SPatrick Farrell . M - gradient norm 2641a9603a14SPatrick Farrell 2642a9603a14SPatrick Farrell Level: beginner 2643a9603a14SPatrick Farrell 26441cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetGradientNorm()`, `TaoGradientNorm()` 2645a9603a14SPatrick Farrell @*/ 2646d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetGradientNorm(Tao tao, Mat *M) 2647d71ae5a4SJacob Faibussowitsch { 2648a9603a14SPatrick Farrell PetscFunctionBegin; 2649a9603a14SPatrick Farrell PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 26504f572ea9SToby Isaac PetscAssertPointer(M, 2); 2651a9603a14SPatrick Farrell *M = tao->gradient_norm; 26523ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2653a9603a14SPatrick Farrell } 2654a9603a14SPatrick Farrell 265505aca2cdSSatish Balay /*@C 265647450a7bSBarry Smith TaoGradientNorm - Compute the norm using the `NormType`, the user has selected 2657a9603a14SPatrick Farrell 2658c3339decSBarry Smith Collective 2659a9603a14SPatrick Farrell 2660d8d19677SJose E. Roman Input Parameters: 266147450a7bSBarry Smith + tao - the `Tao` context 2662a9603a14SPatrick Farrell . gradient - the gradient to be computed 26633b242c63SJacob Faibussowitsch - type - the norm type 2664a9603a14SPatrick Farrell 2665a9603a14SPatrick Farrell Output Parameter: 2666a9603a14SPatrick Farrell . gnorm - the gradient norm 2667a9603a14SPatrick Farrell 266847450a7bSBarry Smith Level: advanced 2669a9603a14SPatrick Farrell 26701cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetGradientNorm()`, `TaoGetGradientNorm()` 2671a9603a14SPatrick Farrell @*/ 2672d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGradientNorm(Tao tao, Vec gradient, NormType type, PetscReal *gnorm) 2673d71ae5a4SJacob Faibussowitsch { 2674a9603a14SPatrick Farrell PetscFunctionBegin; 26758854b543SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 26768854b543SStefano Zampini PetscValidHeaderSpecific(gradient, VEC_CLASSID, 2); 26778854b543SStefano Zampini PetscValidLogicalCollectiveEnum(tao, type, 3); 26784f572ea9SToby Isaac PetscAssertPointer(gnorm, 4); 2679a9603a14SPatrick Farrell if (tao->gradient_norm) { 2680680e2bc7SPatrick Farrell PetscScalar gnorms; 2681680e2bc7SPatrick Farrell 26823c859ba3SBarry 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."); 26839566063dSJacob Faibussowitsch PetscCall(MatMult(tao->gradient_norm, gradient, tao->gradient_norm_tmp)); 26849566063dSJacob Faibussowitsch PetscCall(VecDot(gradient, tao->gradient_norm_tmp, &gnorms)); 26857bb79937SPatrick Farrell *gnorm = PetscRealPart(PetscSqrtScalar(gnorms)); 2686a9603a14SPatrick Farrell } else { 26879566063dSJacob Faibussowitsch PetscCall(VecNorm(gradient, type, gnorm)); 2688a9603a14SPatrick Farrell } 26893ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2690a9603a14SPatrick Farrell } 2691a9603a14SPatrick Farrell 2692e882e171SHong Zhang /*@C 269347450a7bSBarry Smith TaoMonitorDrawCtxCreate - Creates the monitor context for `TaoMonitorDrawSolution()` 2694a9603a14SPatrick Farrell 2695c3339decSBarry Smith Collective 2696e882e171SHong Zhang 26972fe279fdSBarry Smith Input Parameters: 26982fe279fdSBarry Smith + comm - the communicator to share the context 26992fe279fdSBarry Smith . host - the name of the X Windows host that will display the monitor 27002fe279fdSBarry Smith . label - the label to put at the top of the display window 27012fe279fdSBarry Smith . x - the horizontal coordinate of the lower left corner of the window to open 27022fe279fdSBarry Smith . y - the vertical coordinate of the lower left corner of the window to open 27032fe279fdSBarry Smith . m - the width of the window 27042fe279fdSBarry Smith . n - the height of the window 27052fe279fdSBarry Smith - howoften - how many `Tao` iterations between displaying the monitor information 27062fe279fdSBarry Smith 2707d5b43468SJose E. Roman Output Parameter: 2708e882e171SHong Zhang . ctx - the monitor context 2709e882e171SHong Zhang 27103c7db156SBarry Smith Options Database Key: 2711e882e171SHong Zhang . -tao_draw_solution_initial - show initial guess as well as current solution 2712e882e171SHong Zhang 2713e882e171SHong Zhang Level: intermediate 2714e882e171SHong Zhang 27151cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawCtx()` 2716e882e171SHong Zhang @*/ 2717d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDrawCtxCreate(MPI_Comm comm, const char host[], const char label[], int x, int y, int m, int n, PetscInt howoften, TaoMonitorDrawCtx *ctx) 2718d71ae5a4SJacob Faibussowitsch { 2719e882e171SHong Zhang PetscFunctionBegin; 27209566063dSJacob Faibussowitsch PetscCall(PetscNew(ctx)); 27219566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawOpen(comm, host, label, x, y, m, n, &(*ctx)->viewer)); 27229566063dSJacob Faibussowitsch PetscCall(PetscViewerSetFromOptions((*ctx)->viewer)); 2723e882e171SHong Zhang (*ctx)->howoften = howoften; 27243ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2725e882e171SHong Zhang } 2726e882e171SHong Zhang 2727e882e171SHong Zhang /*@C 272865ba42b6SBarry Smith TaoMonitorDrawCtxDestroy - Destroys the monitor context for `TaoMonitorDrawSolution()` 2729e882e171SHong Zhang 2730c3339decSBarry Smith Collective 2731e882e171SHong Zhang 273247450a7bSBarry Smith Input Parameter: 2733e056e8ceSJacob Faibussowitsch . ictx - the monitor context 2734e882e171SHong Zhang 2735e882e171SHong Zhang Level: intermediate 2736e882e171SHong Zhang 27371cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawSolution()` 2738e882e171SHong Zhang @*/ 2739d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *ictx) 2740d71ae5a4SJacob Faibussowitsch { 2741e882e171SHong Zhang PetscFunctionBegin; 27429566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&(*ictx)->viewer)); 27439566063dSJacob Faibussowitsch PetscCall(PetscFree(*ictx)); 27443ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2745e882e171SHong Zhang } 2746