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 24f1e99121SPierre Jolivet static PetscErrorCode KSPPreSolve_TAOEW_Private(KSP ksp, Vec b, Vec x, void *ctx) 25d71ae5a4SJacob Faibussowitsch { 26f1e99121SPierre Jolivet Tao tao = (Tao)ctx; 270f0abf79SStefano Zampini SNES snes_ewdummy = tao->snes_ewdummy; 280f0abf79SStefano Zampini 290f0abf79SStefano Zampini PetscFunctionBegin; 303ba16761SJacob Faibussowitsch if (!snes_ewdummy) PetscFunctionReturn(PETSC_SUCCESS); 310f0abf79SStefano Zampini /* populate snes_ewdummy struct values used in KSPPreSolve_SNESEW */ 320f0abf79SStefano Zampini snes_ewdummy->vec_func = b; 330f0abf79SStefano Zampini snes_ewdummy->rtol = tao->gttol; 340f0abf79SStefano Zampini snes_ewdummy->iter = tao->niter; 350f0abf79SStefano Zampini PetscCall(VecNorm(b, NORM_2, &snes_ewdummy->norm)); 360f0abf79SStefano Zampini PetscCall(KSPPreSolve_SNESEW(ksp, b, x, snes_ewdummy)); 370f0abf79SStefano Zampini snes_ewdummy->vec_func = NULL; 383ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 390f0abf79SStefano Zampini } 400f0abf79SStefano Zampini 41f1e99121SPierre Jolivet static PetscErrorCode KSPPostSolve_TAOEW_Private(KSP ksp, Vec b, Vec x, void *ctx) 42d71ae5a4SJacob Faibussowitsch { 43f1e99121SPierre Jolivet Tao tao = (Tao)ctx; 440f0abf79SStefano Zampini SNES snes_ewdummy = tao->snes_ewdummy; 450f0abf79SStefano Zampini 460f0abf79SStefano Zampini PetscFunctionBegin; 473ba16761SJacob Faibussowitsch if (!snes_ewdummy) PetscFunctionReturn(PETSC_SUCCESS); 480f0abf79SStefano Zampini PetscCall(KSPPostSolve_SNESEW(ksp, b, x, snes_ewdummy)); 493ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 500f0abf79SStefano Zampini } 510f0abf79SStefano Zampini 52d71ae5a4SJacob Faibussowitsch static PetscErrorCode TaoSetUpEW_Private(Tao tao) 53d71ae5a4SJacob Faibussowitsch { 540f0abf79SStefano Zampini SNESKSPEW *kctx; 550f0abf79SStefano Zampini const char *ewprefix; 560f0abf79SStefano Zampini 570f0abf79SStefano Zampini PetscFunctionBegin; 583ba16761SJacob Faibussowitsch if (!tao->ksp) PetscFunctionReturn(PETSC_SUCCESS); 590f0abf79SStefano Zampini if (tao->ksp_ewconv) { 600f0abf79SStefano Zampini if (!tao->snes_ewdummy) PetscCall(SNESCreate(PetscObjectComm((PetscObject)tao), &tao->snes_ewdummy)); 610f0abf79SStefano Zampini tao->snes_ewdummy->ksp_ewconv = PETSC_TRUE; 62f1e99121SPierre Jolivet PetscCall(KSPSetPreSolve(tao->ksp, KSPPreSolve_TAOEW_Private, tao)); 63f1e99121SPierre Jolivet PetscCall(KSPSetPostSolve(tao->ksp, KSPPostSolve_TAOEW_Private, tao)); 640f0abf79SStefano Zampini 650f0abf79SStefano Zampini PetscCall(KSPGetOptionsPrefix(tao->ksp, &ewprefix)); 660f0abf79SStefano Zampini kctx = (SNESKSPEW *)tao->snes_ewdummy->kspconvctx; 67a4598233SStefano Zampini PetscCall(SNESEWSetFromOptions_Private(kctx, PETSC_FALSE, PetscObjectComm((PetscObject)tao), ewprefix)); 680f0abf79SStefano Zampini } else PetscCall(SNESDestroy(&tao->snes_ewdummy)); 693ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 700f0abf79SStefano Zampini } 710f0abf79SStefano Zampini 72a7e14dcfSSatish Balay /*@ 73606f75f6SBarry Smith TaoParametersInitialize - Sets all the parameters in `tao` to their default value (when `TaoCreate()` was called) if they 74606f75f6SBarry Smith currently contain default values. Default values are the parameter values when the object's type is set. 75606f75f6SBarry Smith 76606f75f6SBarry Smith Collective 77606f75f6SBarry Smith 78606f75f6SBarry Smith Input Parameter: 79606f75f6SBarry Smith . tao - the `Tao` object 80606f75f6SBarry Smith 81606f75f6SBarry Smith Level: developer 82606f75f6SBarry Smith 83606f75f6SBarry Smith Developer Note: 84606f75f6SBarry Smith This is called by all the `TaoCreate_XXX()` routines. 85606f75f6SBarry Smith 86606f75f6SBarry Smith .seealso: [](ch_snes), `Tao`, `TaoSolve()`, `TaoDestroy()`, 87606f75f6SBarry Smith `PetscObjectParameterSetDefault()` 88606f75f6SBarry Smith @*/ 89606f75f6SBarry Smith PetscErrorCode TaoParametersInitialize(Tao tao) 90606f75f6SBarry Smith { 91606f75f6SBarry Smith PetscObjectParameterSetDefault(tao, max_it, 10000); 92606f75f6SBarry Smith PetscObjectParameterSetDefault(tao, max_funcs, PETSC_UNLIMITED); 93606f75f6SBarry Smith PetscObjectParameterSetDefault(tao, gatol, PetscDefined(USE_REAL_SINGLE) ? 1e-5 : 1e-8); 94606f75f6SBarry Smith PetscObjectParameterSetDefault(tao, grtol, PetscDefined(USE_REAL_SINGLE) ? 1e-5 : 1e-8); 95606f75f6SBarry Smith PetscObjectParameterSetDefault(tao, crtol, PetscDefined(USE_REAL_SINGLE) ? 1e-5 : 1e-8); 96606f75f6SBarry Smith PetscObjectParameterSetDefault(tao, catol, PetscDefined(USE_REAL_SINGLE) ? 1e-5 : 1e-8); 97606f75f6SBarry Smith PetscObjectParameterSetDefault(tao, gttol, 0.0); 98606f75f6SBarry Smith PetscObjectParameterSetDefault(tao, steptol, 0.0); 99606f75f6SBarry Smith PetscObjectParameterSetDefault(tao, fmin, PETSC_NINFINITY); 100606f75f6SBarry Smith PetscObjectParameterSetDefault(tao, trust0, PETSC_INFINITY); 101606f75f6SBarry Smith return PETSC_SUCCESS; 102606f75f6SBarry Smith } 103606f75f6SBarry Smith 104606f75f6SBarry Smith /*@ 10565ba42b6SBarry Smith TaoCreate - Creates a Tao solver 106a7e14dcfSSatish Balay 107d083f849SBarry Smith Collective 108a7e14dcfSSatish Balay 109a7e14dcfSSatish Balay Input Parameter: 110a7e14dcfSSatish Balay . comm - MPI communicator 111a7e14dcfSSatish Balay 112a7e14dcfSSatish Balay Output Parameter: 11347450a7bSBarry Smith . newtao - the new `Tao` context 114a7e14dcfSSatish Balay 11547450a7bSBarry Smith Options Database Key: 11665ba42b6SBarry Smith . -tao_type - select which method Tao should use 117a7e14dcfSSatish Balay 118a7e14dcfSSatish Balay Level: beginner 119a7e14dcfSSatish Balay 1200241b274SPierre Jolivet .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoDestroy()`, `TaoSetFromOptions()`, `TaoSetType()` 121a7e14dcfSSatish Balay @*/ 122d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoCreate(MPI_Comm comm, Tao *newtao) 123d71ae5a4SJacob Faibussowitsch { 124441846f8SBarry Smith Tao tao; 125a7e14dcfSSatish Balay 126a7e14dcfSSatish Balay PetscFunctionBegin; 1274f572ea9SToby Isaac PetscAssertPointer(newtao, 2); 1289566063dSJacob Faibussowitsch PetscCall(TaoInitializePackage()); 1299566063dSJacob Faibussowitsch PetscCall(TaoLineSearchInitializePackage()); 130a7e14dcfSSatish Balay 131606f75f6SBarry Smith PetscCall(PetscHeaderCreate(tao, TAO_CLASSID, "Tao", "Optimization solver", "Tao", comm, TaoDestroy, TaoView)); 13294511df7SStefano Zampini tao->ops->convergencetest = TaoDefaultConvergenceTest; 133606f75f6SBarry Smith 134a7e14dcfSSatish Balay tao->hist_reset = PETSC_TRUE; 1359566063dSJacob Faibussowitsch PetscCall(TaoResetStatistics(tao)); 136a7e14dcfSSatish Balay *newtao = tao; 1373ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 138a7e14dcfSSatish Balay } 139a7e14dcfSSatish Balay 140a7e14dcfSSatish Balay /*@ 141a7e14dcfSSatish Balay TaoSolve - Solves an optimization problem min F(x) s.t. l <= x <= u 142a7e14dcfSSatish Balay 143c3339decSBarry Smith Collective 144a7e14dcfSSatish Balay 14547450a7bSBarry Smith Input Parameter: 14647450a7bSBarry Smith . tao - the `Tao` context 147a7e14dcfSSatish Balay 14867be906fSBarry Smith Level: beginner 14967be906fSBarry Smith 150a7e14dcfSSatish Balay Notes: 15147450a7bSBarry Smith The user must set up the `Tao` object with calls to `TaoSetSolution()`, `TaoSetObjective()`, `TaoSetGradient()`, and (if using 2nd order method) `TaoSetHessian()`. 152a7e14dcfSSatish Balay 15367be906fSBarry Smith You should call `TaoGetConvergedReason()` or run with `-tao_converged_reason` to determine if the optimization algorithm actually succeeded or 154a35d58b8SBarry Smith why it failed. 155a35d58b8SBarry Smith 1561cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSetObjective()`, `TaoSetGradient()`, `TaoSetHessian()`, `TaoGetConvergedReason()`, `TaoSetUp()` 157a7e14dcfSSatish Balay @*/ 158d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSolve(Tao tao) 159d71ae5a4SJacob Faibussowitsch { 160e2379f4fSBarry Smith static PetscBool set = PETSC_FALSE; 16147a47007SBarry Smith 162a7e14dcfSSatish Balay PetscFunctionBegin; 163441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 164d0609cedSBarry Smith PetscCall(PetscCitationsRegister("@TechReport{tao-user-ref,\n" 165e2379f4fSBarry Smith "title = {Toolkit for Advanced Optimization (TAO) Users Manual},\n" 166e2379f4fSBarry Smith "author = {Todd Munson and Jason Sarich and Stefan Wild and Steve Benson and Lois Curfman McInnes},\n" 167e2379f4fSBarry Smith "Institution = {Argonne National Laboratory},\n" 168e2379f4fSBarry Smith "Year = 2014,\n" 169e2379f4fSBarry Smith "Number = {ANL/MCS-TM-322 - Revision 3.5},\n" 1709371c9d4SSatish Balay "url = {https://www.mcs.anl.gov/research/projects/tao/}\n}\n", 1719371c9d4SSatish Balay &set)); 172494bef23SAlp Dener tao->header_printed = PETSC_FALSE; 1739566063dSJacob Faibussowitsch PetscCall(TaoSetUp(tao)); 1749566063dSJacob Faibussowitsch PetscCall(TaoResetStatistics(tao)); 1751baa6e33SBarry Smith if (tao->linesearch) PetscCall(TaoLineSearchReset(tao->linesearch)); 176a7e14dcfSSatish Balay 1779566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(TAO_Solve, tao, 0, 0, 0)); 178dbbe0bcdSBarry Smith PetscTryTypeMethod(tao, solve); 1799566063dSJacob Faibussowitsch PetscCall(PetscLogEventEnd(TAO_Solve, tao, 0, 0, 0)); 180a7e14dcfSSatish Balay 1819566063dSJacob Faibussowitsch PetscCall(VecViewFromOptions(tao->solution, (PetscObject)tao, "-tao_view_solution")); 1827cf37e64SBarry Smith 1838931d482SJason Sarich tao->ntotalits += tao->niter; 184a7e14dcfSSatish Balay 185a7e14dcfSSatish Balay if (tao->printreason) { 186f642d338SBarry Smith PetscViewer viewer = PETSC_VIEWER_STDOUT_(((PetscObject)tao)->comm); 187f642d338SBarry Smith PetscCall(PetscViewerASCIIAddTab(viewer, ((PetscObject)tao)->tablevel)); 188a7e14dcfSSatish Balay if (tao->reason > 0) { 189f642d338SBarry 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)); 190a7e14dcfSSatish Balay } else { 191f642d338SBarry 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)); 192a7e14dcfSSatish Balay } 193f642d338SBarry Smith PetscCall(PetscViewerASCIISubtractTab(viewer, ((PetscObject)tao)->tablevel)); 194a7e14dcfSSatish Balay } 195f642d338SBarry Smith PetscCall(TaoViewFromOptions(tao, NULL, "-tao_view")); 1963ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 197a7e14dcfSSatish Balay } 198a7e14dcfSSatish Balay 199a7e14dcfSSatish Balay /*@ 200a7e14dcfSSatish Balay TaoSetUp - Sets up the internal data structures for the later use 201a7e14dcfSSatish Balay of a Tao solver 202a7e14dcfSSatish Balay 203c3339decSBarry Smith Collective 204a7e14dcfSSatish Balay 20547450a7bSBarry Smith Input Parameter: 20647450a7bSBarry Smith . tao - the `Tao` context 207a7e14dcfSSatish Balay 20867be906fSBarry Smith Level: advanced 20967be906fSBarry Smith 21047450a7bSBarry Smith Note: 21165ba42b6SBarry Smith The user will not need to explicitly call `TaoSetUp()`, as it will 21265ba42b6SBarry Smith automatically be called in `TaoSolve()`. However, if the user 21365ba42b6SBarry Smith desires to call it explicitly, it should come after `TaoCreate()` 21465ba42b6SBarry Smith and any TaoSetSomething() routines, but before `TaoSolve()`. 215a7e14dcfSSatish Balay 2161cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSolve()` 217a7e14dcfSSatish Balay @*/ 218d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetUp(Tao tao) 219d71ae5a4SJacob Faibussowitsch { 220a7e14dcfSSatish Balay PetscFunctionBegin; 221441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2223ba16761SJacob Faibussowitsch if (tao->setupcalled) PetscFunctionReturn(PETSC_SUCCESS); 2230f0abf79SStefano Zampini PetscCall(TaoSetUpEW_Private(tao)); 2243c859ba3SBarry Smith PetscCheck(tao->solution, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_WRONGSTATE, "Must call TaoSetSolution"); 225dbbe0bcdSBarry Smith PetscTryTypeMethod(tao, setup); 226a7e14dcfSSatish Balay tao->setupcalled = PETSC_TRUE; 2273ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 228a7e14dcfSSatish Balay } 229a7e14dcfSSatish Balay 2300764c050SBarry Smith /*@ 23147450a7bSBarry Smith TaoDestroy - Destroys the `Tao` context that was created with `TaoCreate()` 232a7e14dcfSSatish Balay 233c3339decSBarry Smith Collective 234a7e14dcfSSatish Balay 235a7e14dcfSSatish Balay Input Parameter: 23647450a7bSBarry Smith . tao - the `Tao` context 237a7e14dcfSSatish Balay 238a7e14dcfSSatish Balay Level: beginner 239a7e14dcfSSatish Balay 2401cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSolve()` 241a7e14dcfSSatish Balay @*/ 242d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDestroy(Tao *tao) 243d71ae5a4SJacob Faibussowitsch { 244a7e14dcfSSatish Balay PetscFunctionBegin; 2453ba16761SJacob Faibussowitsch if (!*tao) PetscFunctionReturn(PETSC_SUCCESS); 246441846f8SBarry Smith PetscValidHeaderSpecific(*tao, TAO_CLASSID, 1); 2479371c9d4SSatish Balay if (--((PetscObject)*tao)->refct > 0) { 2489371c9d4SSatish Balay *tao = NULL; 2493ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 250a7e14dcfSSatish Balay } 2519371c9d4SSatish Balay 252213acdd3SPierre Jolivet PetscTryTypeMethod(*tao, destroy); 2539566063dSJacob Faibussowitsch PetscCall(KSPDestroy(&(*tao)->ksp)); 2540f0abf79SStefano Zampini PetscCall(SNESDestroy(&(*tao)->snes_ewdummy)); 2559566063dSJacob Faibussowitsch PetscCall(TaoLineSearchDestroy(&(*tao)->linesearch)); 256a7e14dcfSSatish Balay 257a7e14dcfSSatish Balay if ((*tao)->ops->convergencedestroy) { 2589566063dSJacob Faibussowitsch PetscCall((*(*tao)->ops->convergencedestroy)((*tao)->cnvP)); 25948a46eb9SPierre Jolivet if ((*tao)->jacobian_state_inv) PetscCall(MatDestroy(&(*tao)->jacobian_state_inv)); 260a7e14dcfSSatish Balay } 2619566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->solution)); 2629566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->gradient)); 2639566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->ls_res)); 264a7e14dcfSSatish Balay 265a9603a14SPatrick Farrell if ((*tao)->gradient_norm) { 2669566063dSJacob Faibussowitsch PetscCall(PetscObjectDereference((PetscObject)(*tao)->gradient_norm)); 2679566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->gradient_norm_tmp)); 268a9603a14SPatrick Farrell } 269a9603a14SPatrick Farrell 2709566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->XL)); 2719566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->XU)); 2729566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->IL)); 2739566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->IU)); 2749566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->DE)); 2759566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->DI)); 2769566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->constraints)); 2779566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->constraints_equality)); 2789566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->constraints_inequality)); 2799566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->stepdirection)); 2809566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->hessian_pre)); 2819566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->hessian)); 2829566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->ls_jac)); 2839566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->ls_jac_pre)); 2849566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_pre)); 2859566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian)); 2869566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_state_pre)); 2879566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_state)); 2889566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_state_inv)); 2899566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_design)); 2909566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_equality)); 2919566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_equality_pre)); 2929566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_inequality)); 2939566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_inequality_pre)); 2949566063dSJacob Faibussowitsch PetscCall(ISDestroy(&(*tao)->state_is)); 2959566063dSJacob Faibussowitsch PetscCall(ISDestroy(&(*tao)->design_is)); 2969566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->res_weights_v)); 297b2dc45ddSPierre Jolivet PetscCall(TaoMonitorCancel(*tao)); 29848a46eb9SPierre Jolivet if ((*tao)->hist_malloc) PetscCall(PetscFree4((*tao)->hist_obj, (*tao)->hist_resid, (*tao)->hist_cnorm, (*tao)->hist_lits)); 299737f463aSAlp Dener if ((*tao)->res_weights_n) { 3009566063dSJacob Faibussowitsch PetscCall(PetscFree((*tao)->res_weights_rows)); 3019566063dSJacob Faibussowitsch PetscCall(PetscFree((*tao)->res_weights_cols)); 3029566063dSJacob Faibussowitsch PetscCall(PetscFree((*tao)->res_weights_w)); 303125ddc8aSJason Sarich } 3049566063dSJacob Faibussowitsch PetscCall(PetscHeaderDestroy(tao)); 3053ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 306a7e14dcfSSatish Balay } 307a7e14dcfSSatish Balay 308a7e14dcfSSatish Balay /*@ 3091d27aa22SBarry Smith TaoKSPSetUseEW - Sets `SNES` to use Eisenstat-Walker method {cite}`ew96`for computing relative tolerance for linear solvers. 3100f0abf79SStefano Zampini 311c3339decSBarry Smith Logically Collective 3120f0abf79SStefano Zampini 3130f0abf79SStefano Zampini Input Parameters: 3140f0abf79SStefano Zampini + tao - Tao context 31565ba42b6SBarry Smith - flag - `PETSC_TRUE` or `PETSC_FALSE` 3160f0abf79SStefano Zampini 31767be906fSBarry Smith Level: advanced 31867be906fSBarry Smith 31947450a7bSBarry Smith Note: 32065ba42b6SBarry Smith See `SNESKSPSetUseEW()` for customization details. 3210f0abf79SStefano Zampini 3221cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `SNESKSPSetUseEW()` 3230f0abf79SStefano Zampini @*/ 324d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoKSPSetUseEW(Tao tao, PetscBool flag) 325d71ae5a4SJacob Faibussowitsch { 3260f0abf79SStefano Zampini PetscFunctionBegin; 3270f0abf79SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 3280f0abf79SStefano Zampini PetscValidLogicalCollectiveBool(tao, flag, 2); 3290f0abf79SStefano Zampini tao->ksp_ewconv = flag; 3303ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 3310f0abf79SStefano Zampini } 3320f0abf79SStefano Zampini 3330f0abf79SStefano Zampini /*@ 33447450a7bSBarry Smith TaoSetFromOptions - Sets various Tao parameters from the options database 335a7e14dcfSSatish Balay 336c3339decSBarry Smith Collective 337a7e14dcfSSatish Balay 33801d2d390SJose E. Roman Input Parameter: 33947450a7bSBarry Smith . tao - the `Tao` solver context 340a7e14dcfSSatish Balay 34147450a7bSBarry Smith Options Database Keys: 34265ba42b6SBarry Smith + -tao_type <type> - The algorithm that Tao uses (lmvm, nls, etc.) 343a7e14dcfSSatish Balay . -tao_gatol <gatol> - absolute error tolerance for ||gradient|| 344a7e14dcfSSatish Balay . -tao_grtol <grtol> - relative error tolerance for ||gradient|| 345a7e14dcfSSatish Balay . -tao_gttol <gttol> - reduction of ||gradient|| relative to initial gradient 346a7e14dcfSSatish Balay . -tao_max_it <max> - sets maximum number of iterations 347a7e14dcfSSatish Balay . -tao_max_funcs <max> - sets maximum number of function evaluations 348a7e14dcfSSatish Balay . -tao_fmin <fmin> - stop if function value reaches fmin 349a7e14dcfSSatish Balay . -tao_steptol <tol> - stop if trust region radius less than <tol> 350a7e14dcfSSatish Balay . -tao_trust0 <t> - initial trust region radius 35110978b7dSBarry Smith . -tao_view_solution - view the solution at the end of the optimization process 35247450a7bSBarry Smith . -tao_monitor - prints function value and residual norm at each iteration 35310978b7dSBarry Smith . -tao_monitor_short - same as `-tao_monitor`, but truncates very small values 35410978b7dSBarry Smith . -tao_monitor_constraint_norm - prints objective value, gradient, and constraint norm at each iteration 35510978b7dSBarry Smith . -tao_monitor_globalization - prints information about the globalization at each iteration 35610978b7dSBarry Smith . -tao_monitor_solution - prints solution vector at each iteration 35710978b7dSBarry Smith . -tao_monitor_ls_residual - prints least-squares residual vector at each iteration 35810978b7dSBarry Smith . -tao_monitor_step - prints step vector at each iteration 35910978b7dSBarry Smith . -tao_monitor_gradient - prints gradient vector at each iteration 36010978b7dSBarry Smith . -tao_monitor_solution_draw - graphically view solution vector at each iteration 36110978b7dSBarry Smith . -tao_monitor_step_draw - graphically view step vector at each iteration 36210978b7dSBarry Smith . -tao_monitor_gradient_draw - graphically view gradient at each iteration 36310978b7dSBarry Smith . -tao_monitor_cancel - cancels all monitors (except those set with command line) 364a7e14dcfSSatish Balay . -tao_fd_gradient - use gradient computed with finite differences 365f4c1ad5cSStefano Zampini . -tao_fd_hessian - use hessian computed with finite differences 36610978b7dSBarry Smith . -tao_mf_hessian - use matrix-free Hessian computed with finite differences 367441846f8SBarry Smith . -tao_view - prints information about the Tao after solving 36865ba42b6SBarry Smith - -tao_converged_reason - prints the reason Tao stopped iterating 369a7e14dcfSSatish Balay 37067be906fSBarry Smith Level: beginner 37167be906fSBarry Smith 37267be906fSBarry Smith Note: 37347450a7bSBarry Smith To see all options, run your program with the `-help` option or consult the 37465ba42b6SBarry Smith user's manual. Should be called after `TaoCreate()` but before `TaoSolve()` 375a7e14dcfSSatish Balay 3761cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSolve()` 377a7e14dcfSSatish Balay @*/ 378d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetFromOptions(Tao tao) 379d71ae5a4SJacob Faibussowitsch { 380b625d6c7SJed Brown TaoType default_type = TAOLMVM; 381a7e14dcfSSatish Balay char type[256], monfilename[PETSC_MAX_PATH_LEN]; 382a7e14dcfSSatish Balay PetscViewer monviewer; 383e876fe75SStephan Köhler PetscBool flg, found; 384a7e14dcfSSatish Balay MPI_Comm comm; 385606f75f6SBarry Smith PetscReal catol, crtol, gatol, grtol, gttol; 386a7e14dcfSSatish Balay 387a7e14dcfSSatish Balay PetscFunctionBegin; 388441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 3899566063dSJacob Faibussowitsch PetscCall(PetscObjectGetComm((PetscObject)tao, &comm)); 390125ddc8aSJason Sarich 39160b70c5cSStefano Zampini if (((PetscObject)tao)->type_name) default_type = ((PetscObject)tao)->type_name; 39260b70c5cSStefano Zampini 393d0609cedSBarry Smith PetscObjectOptionsBegin((PetscObject)tao); 394a7e14dcfSSatish Balay /* Check for type from options */ 3959566063dSJacob Faibussowitsch PetscCall(PetscOptionsFList("-tao_type", "Tao Solver type", "TaoSetType", TaoList, default_type, type, 256, &flg)); 396a7e14dcfSSatish Balay if (flg) { 3979566063dSJacob Faibussowitsch PetscCall(TaoSetType(tao, type)); 398a7e14dcfSSatish Balay } else if (!((PetscObject)tao)->type_name) { 3999566063dSJacob Faibussowitsch PetscCall(TaoSetType(tao, default_type)); 400a7e14dcfSSatish Balay } 401a7e14dcfSSatish Balay 40260b70c5cSStefano Zampini /* Tao solvers do not set the prefix, set it here if not yet done 40360b70c5cSStefano Zampini We do it after SetType since solver may have been changed */ 40460b70c5cSStefano Zampini if (tao->linesearch) { 40560b70c5cSStefano Zampini const char *prefix; 40660b70c5cSStefano Zampini PetscCall(TaoLineSearchGetOptionsPrefix(tao->linesearch, &prefix)); 407f4f49eeaSPierre Jolivet if (!prefix) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch, ((PetscObject)tao)->prefix)); 40860b70c5cSStefano Zampini } 40960b70c5cSStefano Zampini 410606f75f6SBarry Smith catol = tao->catol; 411606f75f6SBarry Smith crtol = tao->crtol; 412606f75f6SBarry Smith PetscCall(PetscOptionsReal("-tao_catol", "Stop if constraints violations within", "TaoSetConstraintTolerances", tao->catol, &catol, NULL)); 413606f75f6SBarry Smith PetscCall(PetscOptionsReal("-tao_crtol", "Stop if relative constraint violations within", "TaoSetConstraintTolerances", tao->crtol, &crtol, NULL)); 414606f75f6SBarry Smith PetscCall(TaoSetConstraintTolerances(tao, catol, crtol)); 415606f75f6SBarry Smith 416606f75f6SBarry Smith gatol = tao->gatol; 417606f75f6SBarry Smith grtol = tao->grtol; 418606f75f6SBarry Smith gttol = tao->gttol; 419606f75f6SBarry Smith PetscCall(PetscOptionsReal("-tao_gatol", "Stop if norm of gradient less than", "TaoSetTolerances", tao->gatol, &gatol, NULL)); 420606f75f6SBarry Smith PetscCall(PetscOptionsReal("-tao_grtol", "Stop if norm of gradient divided by the function value is less than", "TaoSetTolerances", tao->grtol, &grtol, NULL)); 421606f75f6SBarry Smith 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, >tol, NULL)); 422606f75f6SBarry Smith PetscCall(TaoSetTolerances(tao, gatol, grtol, gttol)); 423606f75f6SBarry Smith 4249566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-tao_max_it", "Stop if iteration number exceeds", "TaoSetMaximumIterations", tao->max_it, &tao->max_it, &flg)); 425606f75f6SBarry Smith if (flg) PetscCall(TaoSetMaximumIterations(tao, tao->max_it)); 426606f75f6SBarry Smith 4279566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-tao_max_funcs", "Stop if number of function evaluations exceeds", "TaoSetMaximumFunctionEvaluations", tao->max_funcs, &tao->max_funcs, &flg)); 428606f75f6SBarry Smith if (flg) PetscCall(TaoSetMaximumFunctionEvaluations(tao, tao->max_funcs)); 429606f75f6SBarry Smith 430606f75f6SBarry Smith PetscCall(PetscOptionsReal("-tao_fmin", "Stop if function less than", "TaoSetFunctionLowerBound", tao->fmin, &tao->fmin, NULL)); 431606f75f6SBarry Smith PetscCall(PetscOptionsBoundedReal("-tao_steptol", "Stop if step size or trust region radius less than", "", tao->steptol, &tao->steptol, NULL, 0)); 432606f75f6SBarry Smith PetscCall(PetscOptionsReal("-tao_trust0", "Initial trust region radius", "TaoSetInitialTrustRegionRadius", tao->trust0, &tao->trust0, &flg)); 433606f75f6SBarry Smith if (flg) PetscCall(TaoSetInitialTrustRegionRadius(tao, tao->trust0)); 43410978b7dSBarry Smith 43510978b7dSBarry Smith PetscCall(PetscOptionsDeprecated("-tao_solution_monitor", "-tao_monitor_solution", "3.21", NULL)); 43610978b7dSBarry Smith PetscCall(PetscOptionsDeprecated("-tao_gradient_monitor", "-tao_monitor_gradient", "3.21", NULL)); 43710978b7dSBarry Smith PetscCall(PetscOptionsDeprecated("-tao_stepdirection_monitor", "-tao_monitor_step", "3.21", NULL)); 43810978b7dSBarry Smith PetscCall(PetscOptionsDeprecated("-tao_residual_monitor", "-tao_monitor_residual", "3.21", NULL)); 43910978b7dSBarry Smith PetscCall(PetscOptionsDeprecated("-tao_smonitor", "-tao_monitor_short", "3.21", NULL)); 44010978b7dSBarry Smith PetscCall(PetscOptionsDeprecated("-tao_cmonitor", "-tao_monitor_constraint_norm", "3.21", NULL)); 44110978b7dSBarry Smith PetscCall(PetscOptionsDeprecated("-tao_gmonitor", "-tao_monitor_globalization", "3.21", NULL)); 44210978b7dSBarry Smith PetscCall(PetscOptionsDeprecated("-tao_draw_solution", "-tao_monitor_solution_draw", "3.21", NULL)); 44310978b7dSBarry Smith PetscCall(PetscOptionsDeprecated("-tao_draw_gradient", "-tao_monitor_gradient_draw", "3.21", NULL)); 44410978b7dSBarry Smith PetscCall(PetscOptionsDeprecated("-tao_draw_step", "-tao_monitor_step_draw", "3.21", NULL)); 44510978b7dSBarry Smith 44610978b7dSBarry Smith PetscCall(PetscOptionsString("-tao_monitor_solution", "View solution vector after each iteration", "TaoMonitorSet", "stdout", monfilename, sizeof(monfilename), &flg)); 447a7e14dcfSSatish Balay if (flg) { 4489566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 449*49abdd8aSBarry Smith PetscCall(TaoMonitorSet(tao, TaoMonitorSolution, monviewer, (PetscCtxDestroyFn *)PetscViewerDestroy)); 450a7e14dcfSSatish Balay } 451a7e14dcfSSatish Balay 45265ba42b6SBarry Smith PetscCall(PetscOptionsBool("-tao_converged_reason", "Print reason for Tao converged", "TaoSolve", tao->printreason, &tao->printreason, NULL)); 45310978b7dSBarry Smith PetscCall(PetscOptionsString("-tao_monitor_gradient", "View gradient vector for each iteration", "TaoMonitorSet", "stdout", monfilename, sizeof(monfilename), &flg)); 454a7e14dcfSSatish Balay if (flg) { 4559566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 456*49abdd8aSBarry Smith PetscCall(TaoMonitorSet(tao, TaoMonitorGradient, monviewer, (PetscCtxDestroyFn *)PetscViewerDestroy)); 457a7e14dcfSSatish Balay } 458a7e14dcfSSatish Balay 45910978b7dSBarry Smith PetscCall(PetscOptionsString("-tao_monitor_step", "View step vector after each iteration", "TaoMonitorSet", "stdout", monfilename, sizeof(monfilename), &flg)); 460a7e14dcfSSatish Balay if (flg) { 4619566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 462*49abdd8aSBarry Smith PetscCall(TaoMonitorSet(tao, TaoMonitorStep, monviewer, (PetscCtxDestroyFn *)PetscViewerDestroy)); 463a7e14dcfSSatish Balay } 464a7e14dcfSSatish Balay 46510978b7dSBarry Smith PetscCall(PetscOptionsString("-tao_monitor_residual", "View least-squares residual vector after each iteration", "TaoMonitorSet", "stdout", monfilename, sizeof(monfilename), &flg)); 466a7e14dcfSSatish Balay if (flg) { 4679566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 468*49abdd8aSBarry Smith PetscCall(TaoMonitorSet(tao, TaoMonitorResidual, monviewer, (PetscCtxDestroyFn *)PetscViewerDestroy)); 469a7e14dcfSSatish Balay } 470a7e14dcfSSatish Balay 47110978b7dSBarry Smith PetscCall(PetscOptionsString("-tao_monitor", "Use the default convergence monitor", "TaoMonitorSet", "stdout", monfilename, sizeof(monfilename), &flg)); 472a7e14dcfSSatish Balay if (flg) { 4739566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 474*49abdd8aSBarry Smith PetscCall(TaoMonitorSet(tao, TaoMonitorDefault, monviewer, (PetscCtxDestroyFn *)PetscViewerDestroy)); 475a7e14dcfSSatish Balay } 476a7e14dcfSSatish Balay 47710978b7dSBarry Smith PetscCall(PetscOptionsString("-tao_monitor_globalization", "Use the convergence monitor with extra globalization info", "TaoMonitorSet", "stdout", monfilename, sizeof(monfilename), &flg)); 4788d5ead36SAlp Dener if (flg) { 4799566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 480*49abdd8aSBarry Smith PetscCall(TaoMonitorSet(tao, TaoMonitorGlobalization, monviewer, (PetscCtxDestroyFn *)PetscViewerDestroy)); 4818d5ead36SAlp Dener } 4828d5ead36SAlp Dener 48310978b7dSBarry Smith PetscCall(PetscOptionsString("-tao_monitor_short", "Use the short convergence monitor", "TaoMonitorSet", "stdout", monfilename, sizeof(monfilename), &flg)); 484a7e14dcfSSatish Balay if (flg) { 4859566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 486*49abdd8aSBarry Smith PetscCall(TaoMonitorSet(tao, TaoMonitorDefaultShort, monviewer, (PetscCtxDestroyFn *)PetscViewerDestroy)); 487a7e14dcfSSatish Balay } 488a7e14dcfSSatish Balay 48910978b7dSBarry Smith PetscCall(PetscOptionsString("-tao_monitor_constraint_norm", "Use the default convergence monitor with constraint norm", "TaoMonitorSet", "stdout", monfilename, sizeof(monfilename), &flg)); 490a7e14dcfSSatish Balay if (flg) { 4919566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm, monfilename, &monviewer)); 492*49abdd8aSBarry Smith PetscCall(TaoMonitorSet(tao, TaoMonitorConstraintNorm, monviewer, (PetscCtxDestroyFn *)PetscViewerDestroy)); 493a7e14dcfSSatish Balay } 494a7e14dcfSSatish Balay 4958afaa268SBarry Smith flg = PETSC_FALSE; 496b2dc45ddSPierre Jolivet PetscCall(PetscOptionsDeprecated("-tao_cancelmonitors", "-tao_monitor_cancel", "3.21", NULL)); 497b2dc45ddSPierre Jolivet PetscCall(PetscOptionsBool("-tao_monitor_cancel", "cancel all monitors and call any registered destroy routines", "TaoMonitorCancel", flg, &flg, NULL)); 498b2dc45ddSPierre Jolivet if (flg) PetscCall(TaoMonitorCancel(tao)); 499a7e14dcfSSatish Balay 5008afaa268SBarry Smith flg = PETSC_FALSE; 50110978b7dSBarry Smith PetscCall(PetscOptionsBool("-tao_monitor_solution_draw", "Plot solution vector at each iteration", "TaoMonitorSet", flg, &flg, NULL)); 502a7e14dcfSSatish Balay if (flg) { 503e882e171SHong Zhang TaoMonitorDrawCtx drawctx; 504e882e171SHong Zhang PetscInt howoften = 1; 5059566063dSJacob Faibussowitsch PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao), NULL, NULL, PETSC_DECIDE, PETSC_DECIDE, 300, 300, howoften, &drawctx)); 506*49abdd8aSBarry Smith PetscCall(TaoMonitorSet(tao, TaoMonitorSolutionDraw, drawctx, (PetscCtxDestroyFn *)TaoMonitorDrawCtxDestroy)); 507a7e14dcfSSatish Balay } 508a7e14dcfSSatish Balay 5098afaa268SBarry Smith flg = PETSC_FALSE; 51010978b7dSBarry Smith PetscCall(PetscOptionsBool("-tao_monitor_step_draw", "Plots step at each iteration", "TaoMonitorSet", flg, &flg, NULL)); 51110978b7dSBarry Smith if (flg) PetscCall(TaoMonitorSet(tao, TaoMonitorStepDraw, NULL, NULL)); 512a7e14dcfSSatish Balay 5138afaa268SBarry Smith flg = PETSC_FALSE; 51410978b7dSBarry Smith PetscCall(PetscOptionsBool("-tao_monitor_gradient_draw", "plots gradient at each iteration", "TaoMonitorSet", flg, &flg, NULL)); 515a7e14dcfSSatish Balay if (flg) { 516e882e171SHong Zhang TaoMonitorDrawCtx drawctx; 517e882e171SHong Zhang PetscInt howoften = 1; 5189566063dSJacob Faibussowitsch PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao), NULL, NULL, PETSC_DECIDE, PETSC_DECIDE, 300, 300, howoften, &drawctx)); 519*49abdd8aSBarry Smith PetscCall(TaoMonitorSet(tao, TaoMonitorGradientDraw, drawctx, (PetscCtxDestroyFn *)TaoMonitorDrawCtxDestroy)); 520a7e14dcfSSatish Balay } 5218afaa268SBarry Smith flg = PETSC_FALSE; 5229566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_fd_gradient", "compute gradient using finite differences", "TaoDefaultComputeGradient", flg, &flg, NULL)); 5231baa6e33SBarry Smith if (flg) PetscCall(TaoSetGradient(tao, NULL, TaoDefaultComputeGradient, NULL)); 524f4c1ad5cSStefano Zampini flg = PETSC_FALSE; 52510978b7dSBarry Smith PetscCall(PetscOptionsBool("-tao_fd_hessian", "compute Hessian using finite differences", "TaoDefaultComputeHessian", flg, &flg, NULL)); 526f4c1ad5cSStefano Zampini if (flg) { 527f4c1ad5cSStefano Zampini Mat H; 528f4c1ad5cSStefano Zampini 5299566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)tao), &H)); 5309566063dSJacob Faibussowitsch PetscCall(MatSetType(H, MATAIJ)); 5319566063dSJacob Faibussowitsch PetscCall(TaoSetHessian(tao, H, H, TaoDefaultComputeHessian, NULL)); 5329566063dSJacob Faibussowitsch PetscCall(MatDestroy(&H)); 533f4c1ad5cSStefano Zampini } 534f4c1ad5cSStefano Zampini flg = PETSC_FALSE; 53510978b7dSBarry Smith PetscCall(PetscOptionsBool("-tao_mf_hessian", "compute matrix-free Hessian using finite differences", "TaoDefaultComputeHessianMFFD", flg, &flg, NULL)); 536f4c1ad5cSStefano Zampini if (flg) { 537f4c1ad5cSStefano Zampini Mat H; 538f4c1ad5cSStefano Zampini 5399566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)tao), &H)); 5409566063dSJacob Faibussowitsch PetscCall(TaoSetHessian(tao, H, H, TaoDefaultComputeHessianMFFD, NULL)); 5419566063dSJacob Faibussowitsch PetscCall(MatDestroy(&H)); 542f4c1ad5cSStefano Zampini } 543e876fe75SStephan Köhler PetscCall(PetscOptionsBool("-tao_recycle_history", "enable recycling/re-using information from the previous TaoSolve() call for some algorithms", "TaoSetRecycleHistory", flg, &flg, &found)); 544e876fe75SStephan Köhler if (found) PetscCall(TaoSetRecycleHistory(tao, flg)); 5459566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnum("-tao_subset_type", "subset type", "", TaoSubSetTypes, (PetscEnum)tao->subset_type, (PetscEnum *)&tao->subset_type, NULL)); 546a7e14dcfSSatish Balay 5470f0abf79SStefano Zampini if (tao->ksp) { 5480f0abf79SStefano Zampini PetscCall(PetscOptionsBool("-tao_ksp_ew", "Use Eisentat-Walker linear system convergence test", "TaoKSPSetUseEW", tao->ksp_ewconv, &tao->ksp_ewconv, NULL)); 5490f0abf79SStefano Zampini PetscCall(TaoKSPSetUseEW(tao, tao->ksp_ewconv)); 5500f0abf79SStefano Zampini } 5510f0abf79SStefano Zampini 552dbbe0bcdSBarry Smith PetscTryTypeMethod(tao, setfromoptions, PetscOptionsObject); 55360b70c5cSStefano Zampini 55460b70c5cSStefano Zampini /* process any options handlers added with PetscObjectAddOptionsHandler() */ 55560b70c5cSStefano Zampini PetscCall(PetscObjectProcessOptionsHandlers((PetscObject)tao, PetscOptionsObject)); 556d0609cedSBarry Smith PetscOptionsEnd(); 55760b70c5cSStefano Zampini 55860b70c5cSStefano Zampini if (tao->linesearch) PetscCall(TaoLineSearchSetFromOptions(tao->linesearch)); 5593ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 560a7e14dcfSSatish Balay } 561a7e14dcfSSatish Balay 562ffeef943SBarry Smith /*@ 56347450a7bSBarry Smith TaoViewFromOptions - View a `Tao` object based on values in the options database 564fe2efc57SMark 565c3339decSBarry Smith Collective 566fe2efc57SMark 567fe2efc57SMark Input Parameters: 56847450a7bSBarry Smith + A - the `Tao` context 56947450a7bSBarry Smith . obj - Optional object that provides the prefix for the options database 570736c3998SJose E. Roman - name - command line option 571fe2efc57SMark 572fe2efc57SMark Level: intermediate 57367be906fSBarry Smith 5741cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoView`, `PetscObjectViewFromOptions()`, `TaoCreate()` 575fe2efc57SMark @*/ 576d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoViewFromOptions(Tao A, PetscObject obj, const char name[]) 577d71ae5a4SJacob Faibussowitsch { 578fe2efc57SMark PetscFunctionBegin; 579fe2efc57SMark PetscValidHeaderSpecific(A, TAO_CLASSID, 1); 5809566063dSJacob Faibussowitsch PetscCall(PetscObjectViewFromOptions((PetscObject)A, obj, name)); 5813ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 582fe2efc57SMark } 583fe2efc57SMark 584ffeef943SBarry Smith /*@ 58547450a7bSBarry Smith TaoView - Prints information about the `Tao` object 586a7e14dcfSSatish Balay 587c3339decSBarry Smith Collective 588a7e14dcfSSatish Balay 589a7e14dcfSSatish Balay Input Parameters: 59047450a7bSBarry Smith + tao - the `Tao` context 591a7e14dcfSSatish Balay - viewer - visualization context 592a7e14dcfSSatish Balay 593a7e14dcfSSatish Balay Options Database Key: 59465ba42b6SBarry Smith . -tao_view - Calls `TaoView()` at the end of `TaoSolve()` 595a7e14dcfSSatish Balay 59667be906fSBarry Smith Level: beginner 59767be906fSBarry Smith 598a7e14dcfSSatish Balay Notes: 599a7e14dcfSSatish Balay The available visualization contexts include 60065ba42b6SBarry Smith + `PETSC_VIEWER_STDOUT_SELF` - standard output (default) 60165ba42b6SBarry Smith - `PETSC_VIEWER_STDOUT_WORLD` - synchronized standard 602a7e14dcfSSatish Balay output where only the first processor opens 603a7e14dcfSSatish Balay the file. All other processors send their 604a7e14dcfSSatish Balay data to the first processor to print. 605a7e14dcfSSatish Balay 6061cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `PetscViewerASCIIOpen()` 607a7e14dcfSSatish Balay @*/ 608d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoView(Tao tao, PetscViewer viewer) 609d71ae5a4SJacob Faibussowitsch { 610a7e14dcfSSatish Balay PetscBool isascii, isstring; 611b625d6c7SJed Brown TaoType type; 61247a47007SBarry Smith 613a7e14dcfSSatish Balay PetscFunctionBegin; 614441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 61548a46eb9SPierre Jolivet if (!viewer) PetscCall(PetscViewerASCIIGetStdout(((PetscObject)tao)->comm, &viewer)); 616a7e14dcfSSatish Balay PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 617a7e14dcfSSatish Balay PetscCheckSameComm(tao, 1, viewer, 2); 618a7e14dcfSSatish Balay 6199566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERASCII, &isascii)); 6209566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer, PETSCVIEWERSTRING, &isstring)); 621a7e14dcfSSatish Balay if (isascii) { 6229566063dSJacob Faibussowitsch PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)tao, viewer)); 623a7e14dcfSSatish Balay 6249566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 625606f75f6SBarry Smith PetscTryTypeMethod(tao, view, viewer); 626606f75f6SBarry Smith if (tao->linesearch) PetscCall(TaoLineSearchView(tao->linesearch, viewer)); 627a7e14dcfSSatish Balay if (tao->ksp) { 6289566063dSJacob Faibussowitsch PetscCall(KSPView(tao->ksp, viewer)); 62963a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total KSP iterations: %" PetscInt_FMT "\n", tao->ksp_tot_its)); 630a7e14dcfSSatish Balay } 63119b5c101SAlp Dener 63248a46eb9SPierre Jolivet if (tao->XL || tao->XU) PetscCall(PetscViewerASCIIPrintf(viewer, "Active Set subset type: %s\n", TaoSubSetTypes[tao->subset_type])); 633a7e14dcfSSatish Balay 6349566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: gatol=%g,", (double)tao->gatol)); 635606f75f6SBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, " grtol=%g,", (double)tao->grtol)); 6369566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " steptol=%g,", (double)tao->steptol)); 6379566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " gttol=%g\n", (double)tao->gttol)); 6389566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Residual in Function/Gradient:=%g\n", (double)tao->residual)); 639a7e14dcfSSatish Balay 6406246e8cdSAlp Dener if (tao->constrained) { 6419566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances:")); 6429566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " catol=%g,", (double)tao->catol)); 6439566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " crtol=%g\n", (double)tao->crtol)); 6449566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Residual in Constraints:=%g\n", (double)tao->cnorm)); 645a7e14dcfSSatish Balay } 646a7e14dcfSSatish Balay 647a7e14dcfSSatish Balay if (tao->trust < tao->steptol) { 6489566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: steptol=%g\n", (double)tao->steptol)); 6499566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Final trust region radius:=%g\n", (double)tao->trust)); 650a7e14dcfSSatish Balay } 651a7e14dcfSSatish Balay 65248a46eb9SPierre Jolivet if (tao->fmin > -1.e25) PetscCall(PetscViewerASCIIPrintf(viewer, "convergence tolerances: function minimum=%g\n", (double)tao->fmin)); 6539566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Objective value=%g\n", (double)tao->fc)); 654a7e14dcfSSatish Balay 65563a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total number of iterations=%" PetscInt_FMT ", ", tao->niter)); 65663a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " (max: %" PetscInt_FMT ")\n", tao->max_it)); 657a7e14dcfSSatish Balay 658a7e14dcfSSatish Balay if (tao->nfuncs > 0) { 65963a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total number of function evaluations=%" PetscInt_FMT ",", tao->nfuncs)); 660606f75f6SBarry Smith if (tao->max_funcs == PETSC_UNLIMITED) PetscCall(PetscViewerASCIIPrintf(viewer, " (max: unlimited)\n")); 661606f75f6SBarry Smith else PetscCall(PetscViewerASCIIPrintf(viewer, " (max: %" PetscInt_FMT ")\n", tao->max_funcs)); 662a7e14dcfSSatish Balay } 663a7e14dcfSSatish Balay if (tao->ngrads > 0) { 66463a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total number of gradient evaluations=%" PetscInt_FMT ",", tao->ngrads)); 665606f75f6SBarry Smith if (tao->max_funcs == PETSC_UNLIMITED) PetscCall(PetscViewerASCIIPrintf(viewer, " (max: unlimited)\n")); 666606f75f6SBarry Smith else PetscCall(PetscViewerASCIIPrintf(viewer, " (max: %" PetscInt_FMT ")\n", tao->max_funcs)); 667a7e14dcfSSatish Balay } 668a7e14dcfSSatish Balay if (tao->nfuncgrads > 0) { 66963a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "total number of function/gradient evaluations=%" PetscInt_FMT ",", tao->nfuncgrads)); 670606f75f6SBarry Smith if (tao->max_funcs == PETSC_UNLIMITED) PetscCall(PetscViewerASCIIPrintf(viewer, " (max: unlimited)\n")); 671606f75f6SBarry Smith else PetscCall(PetscViewerASCIIPrintf(viewer, " (max: %" PetscInt_FMT ")\n", tao->max_funcs)); 672a7e14dcfSSatish Balay } 67348a46eb9SPierre Jolivet if (tao->nhess > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of Hessian evaluations=%" PetscInt_FMT "\n", tao->nhess)); 67448a46eb9SPierre Jolivet if (tao->nconstraints > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of constraint function evaluations=%" PetscInt_FMT "\n", tao->nconstraints)); 67548a46eb9SPierre Jolivet if (tao->njac > 0) PetscCall(PetscViewerASCIIPrintf(viewer, "total number of Jacobian evaluations=%" PetscInt_FMT "\n", tao->njac)); 676a7e14dcfSSatish Balay 677a7e14dcfSSatish Balay if (tao->reason > 0) { 6789566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Solution converged: ")); 679a7e14dcfSSatish Balay switch (tao->reason) { 680d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_GATOL: 681d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)|| <= gatol\n")); 682d71ae5a4SJacob Faibussowitsch break; 683d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_GRTOL: 684d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)||/|f(X)| <= grtol\n")); 685d71ae5a4SJacob Faibussowitsch break; 686d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_GTTOL: 687d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " ||g(X)||/||g(X0)|| <= gttol\n")); 688d71ae5a4SJacob Faibussowitsch break; 689d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_STEPTOL: 690d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Steptol -- step size small\n")); 691d71ae5a4SJacob Faibussowitsch break; 692d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_MINF: 693d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Minf -- f < fmin\n")); 694d71ae5a4SJacob Faibussowitsch break; 695d71ae5a4SJacob Faibussowitsch case TAO_CONVERGED_USER: 696d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " User Terminated\n")); 697d71ae5a4SJacob Faibussowitsch break; 698d71ae5a4SJacob Faibussowitsch default: 699606f75f6SBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, " %d\n", tao->reason)); 700d71ae5a4SJacob Faibussowitsch break; 701a7e14dcfSSatish Balay } 702606f75f6SBarry Smith } else if (tao->reason == TAO_CONTINUE_ITERATING) { 703606f75f6SBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, "Solver never run\n")); 704a7e14dcfSSatish Balay } else { 705606f75f6SBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, "Solver failed: ")); 706a7e14dcfSSatish Balay switch (tao->reason) { 707d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_MAXITS: 708d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum Iterations\n")); 709d71ae5a4SJacob Faibussowitsch break; 710d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_NAN: 711d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " NAN or Inf encountered\n")); 712d71ae5a4SJacob Faibussowitsch break; 713d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_MAXFCN: 714d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Maximum Function Evaluations\n")); 715d71ae5a4SJacob Faibussowitsch break; 716d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_LS_FAILURE: 717d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Line Search Failure\n")); 718d71ae5a4SJacob Faibussowitsch break; 719d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_TR_REDUCTION: 720d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Trust Region too small\n")); 721d71ae5a4SJacob Faibussowitsch break; 722d71ae5a4SJacob Faibussowitsch case TAO_DIVERGED_USER: 723d71ae5a4SJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " User Terminated\n")); 724d71ae5a4SJacob Faibussowitsch break; 725d71ae5a4SJacob Faibussowitsch default: 726606f75f6SBarry Smith PetscCall(PetscViewerASCIIPrintf(viewer, " %d\n", tao->reason)); 727d71ae5a4SJacob Faibussowitsch break; 728a7e14dcfSSatish Balay } 729a7e14dcfSSatish Balay } 7309566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 731a7e14dcfSSatish Balay } else if (isstring) { 7329566063dSJacob Faibussowitsch PetscCall(TaoGetType(tao, &type)); 7339566063dSJacob Faibussowitsch PetscCall(PetscViewerStringSPrintf(viewer, " %-3.3s", type)); 734a7e14dcfSSatish Balay } 7353ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 736a7e14dcfSSatish Balay } 737a7e14dcfSSatish Balay 738a7e14dcfSSatish Balay /*@ 739414d97d3SAlp Dener TaoSetRecycleHistory - Sets the boolean flag to enable/disable re-using 74065ba42b6SBarry Smith iterate information from the previous `TaoSolve()`. This feature is disabled by 741414d97d3SAlp Dener default. 742414d97d3SAlp Dener 74367be906fSBarry Smith Logically Collective 74467be906fSBarry Smith 74567be906fSBarry Smith Input Parameters: 74647450a7bSBarry Smith + tao - the `Tao` context 74767be906fSBarry Smith - recycle - boolean flag 74867be906fSBarry Smith 74947450a7bSBarry Smith Options Database Key: 75067be906fSBarry Smith . -tao_recycle_history <true,false> - reuse the history 75167be906fSBarry Smith 75267be906fSBarry Smith Level: intermediate 75367be906fSBarry Smith 75467be906fSBarry Smith Notes: 75565ba42b6SBarry Smith For conjugate gradient methods (`TAOBNCG`), this re-uses the latest search direction 75665ba42b6SBarry Smith from the previous `TaoSolve()` call when computing the first search direction in a 757414d97d3SAlp Dener new solution. By default, CG methods set the first search direction to the 758414d97d3SAlp Dener negative gradient. 759414d97d3SAlp Dener 76065ba42b6SBarry Smith For quasi-Newton family of methods (`TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL`), this re-uses 76165ba42b6SBarry Smith the accumulated quasi-Newton Hessian approximation from the previous `TaoSolve()` 762414d97d3SAlp Dener call. By default, QN family of methods reset the initial Hessian approximation to 763414d97d3SAlp Dener the identity matrix. 764414d97d3SAlp Dener 765414d97d3SAlp Dener For any other algorithm, this setting has no effect. 766414d97d3SAlp Dener 7671cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL` 768414d97d3SAlp Dener @*/ 769d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetRecycleHistory(Tao tao, PetscBool recycle) 770d71ae5a4SJacob Faibussowitsch { 771414d97d3SAlp Dener PetscFunctionBegin; 772414d97d3SAlp Dener PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 77394511df7SStefano Zampini PetscValidLogicalCollectiveBool(tao, recycle, 2); 774414d97d3SAlp Dener tao->recycle = recycle; 7753ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 776414d97d3SAlp Dener } 777414d97d3SAlp Dener 778414d97d3SAlp Dener /*@ 779414d97d3SAlp Dener TaoGetRecycleHistory - Retrieve the boolean flag for re-using iterate information 78065ba42b6SBarry Smith from the previous `TaoSolve()`. This feature is disabled by default. 781414d97d3SAlp Dener 78267be906fSBarry Smith Logically Collective 783414d97d3SAlp Dener 78447450a7bSBarry Smith Input Parameter: 78547450a7bSBarry Smith . tao - the `Tao` context 786414d97d3SAlp Dener 78747450a7bSBarry Smith Output Parameter: 788147403d9SBarry Smith . recycle - boolean flag 789414d97d3SAlp Dener 790414d97d3SAlp Dener Level: intermediate 791414d97d3SAlp Dener 7921cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL` 793414d97d3SAlp Dener @*/ 794d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetRecycleHistory(Tao tao, PetscBool *recycle) 795d71ae5a4SJacob Faibussowitsch { 796414d97d3SAlp Dener PetscFunctionBegin; 797414d97d3SAlp Dener PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 7984f572ea9SToby Isaac PetscAssertPointer(recycle, 2); 799414d97d3SAlp Dener *recycle = tao->recycle; 8003ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 801414d97d3SAlp Dener } 802414d97d3SAlp Dener 803414d97d3SAlp Dener /*@ 80447450a7bSBarry Smith TaoSetTolerances - Sets parameters used in `TaoSolve()` convergence tests 805a7e14dcfSSatish Balay 80667be906fSBarry Smith Logically Collective 807a7e14dcfSSatish Balay 808a7e14dcfSSatish Balay Input Parameters: 80947450a7bSBarry Smith + tao - the `Tao` context 810a7e14dcfSSatish Balay . gatol - stop if norm of gradient is less than this 811a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this 812a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by this factor 813a7e14dcfSSatish Balay 814a7e14dcfSSatish Balay Options Database Keys: 815e52336cbSBarry Smith + -tao_gatol <gatol> - Sets gatol 816a7e14dcfSSatish Balay . -tao_grtol <grtol> - Sets grtol 817a7e14dcfSSatish Balay - -tao_gttol <gttol> - Sets gttol 818a7e14dcfSSatish Balay 8193b242c63SJacob Faibussowitsch Stopping Criteria\: 82067be906fSBarry Smith .vb 82167be906fSBarry Smith ||g(X)|| <= gatol 82267be906fSBarry Smith ||g(X)|| / |f(X)| <= grtol 82367be906fSBarry Smith ||g(X)|| / ||g(X0)|| <= gttol 82467be906fSBarry Smith .ve 825a7e14dcfSSatish Balay 826a7e14dcfSSatish Balay Level: beginner 827a7e14dcfSSatish Balay 828606f75f6SBarry Smith Notes: 829606f75f6SBarry Smith Use `PETSC_CURRENT` to leave one or more tolerances unchanged. 830606f75f6SBarry Smith 831606f75f6SBarry Smith Use `PETSC_DETERMINE` to set one or more tolerances to their values when the `tao`object's type was set 832606f75f6SBarry Smith 833606f75f6SBarry Smith Fortran Note: 834606f75f6SBarry Smith Use `PETSC_CURRENT_REAL` or `PETSC_DETERMINE_REAL` 835a7e14dcfSSatish Balay 8361cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoGetTolerances()` 837a7e14dcfSSatish Balay @*/ 838d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetTolerances(Tao tao, PetscReal gatol, PetscReal grtol, PetscReal gttol) 839d71ae5a4SJacob Faibussowitsch { 840a7e14dcfSSatish Balay PetscFunctionBegin; 841441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 84294511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, gatol, 2); 84394511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, grtol, 3); 84494511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, gttol, 4); 845a7e14dcfSSatish Balay 846606f75f6SBarry Smith if (gatol == (PetscReal)PETSC_DETERMINE) { 847606f75f6SBarry Smith tao->gatol = tao->default_gatol; 848606f75f6SBarry Smith } else if (gatol != (PetscReal)PETSC_CURRENT) { 849606f75f6SBarry Smith PetscCheck(gatol >= 0, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_OUTOFRANGE, "Negative gatol not allowed"); 850606f75f6SBarry Smith tao->gatol = gatol; 851a7e14dcfSSatish Balay } 852a7e14dcfSSatish Balay 853606f75f6SBarry Smith if (grtol == (PetscReal)PETSC_DETERMINE) { 854606f75f6SBarry Smith tao->grtol = tao->default_grtol; 855606f75f6SBarry Smith } else if (grtol != (PetscReal)PETSC_CURRENT) { 856606f75f6SBarry Smith PetscCheck(grtol >= 0, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_OUTOFRANGE, "Negative grtol not allowed"); 857606f75f6SBarry Smith tao->grtol = grtol; 858a7e14dcfSSatish Balay } 859a7e14dcfSSatish Balay 860606f75f6SBarry Smith if (gttol == (PetscReal)PETSC_DETERMINE) { 861606f75f6SBarry Smith tao->gttol = tao->default_gttol; 862606f75f6SBarry Smith } else if (gttol != (PetscReal)PETSC_CURRENT) { 863606f75f6SBarry Smith PetscCheck(gttol >= 0, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_OUTOFRANGE, "Negative gttol not allowed"); 864606f75f6SBarry Smith tao->gttol = gttol; 865a7e14dcfSSatish Balay } 8663ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 867a7e14dcfSSatish Balay } 868a7e14dcfSSatish Balay 869a7e14dcfSSatish Balay /*@ 87047450a7bSBarry Smith TaoSetConstraintTolerances - Sets constraint tolerance parameters used in `TaoSolve()` convergence tests 871a7e14dcfSSatish Balay 87267be906fSBarry Smith Logically Collective 873a7e14dcfSSatish Balay 874a7e14dcfSSatish Balay Input Parameters: 87547450a7bSBarry Smith + tao - the `Tao` context 876606f75f6SBarry Smith . catol - absolute constraint tolerance, constraint norm must be less than `catol` for used for `gatol` convergence criteria 877606f75f6SBarry Smith - crtol - relative constraint tolerance, constraint norm must be less than `crtol` for used for `gatol`, `gttol` convergence criteria 878a7e14dcfSSatish Balay 879a7e14dcfSSatish Balay Options Database Keys: 880a7e14dcfSSatish Balay + -tao_catol <catol> - Sets catol 881a7e14dcfSSatish Balay - -tao_crtol <crtol> - Sets crtol 882a7e14dcfSSatish Balay 883a7e14dcfSSatish Balay Level: intermediate 884a7e14dcfSSatish Balay 88567be906fSBarry Smith Notes: 886606f75f6SBarry Smith Use `PETSC_CURRENT` to leave one or tolerance unchanged. 887606f75f6SBarry Smith 888606f75f6SBarry Smith Use `PETSC_DETERMINE` to set one or more tolerances to their values when the `tao` object's type was set 889606f75f6SBarry Smith 890606f75f6SBarry Smith Fortran Note: 891606f75f6SBarry Smith Use `PETSC_CURRENT_REAL` or `PETSC_DETERMINE_REAL` 892a7e14dcfSSatish Balay 8931cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoGetTolerances()`, `TaoGetConstraintTolerances()`, `TaoSetTolerances()` 894a7e14dcfSSatish Balay @*/ 895d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConstraintTolerances(Tao tao, PetscReal catol, PetscReal crtol) 896d71ae5a4SJacob Faibussowitsch { 897a7e14dcfSSatish Balay PetscFunctionBegin; 898441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 89994511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, catol, 2); 90094511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, crtol, 3); 901a7e14dcfSSatish Balay 902606f75f6SBarry Smith if (catol == (PetscReal)PETSC_DETERMINE) { 903606f75f6SBarry Smith tao->catol = tao->default_catol; 904606f75f6SBarry Smith } else if (catol != (PetscReal)PETSC_CURRENT) { 905606f75f6SBarry Smith PetscCheck(catol >= 0, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_OUTOFRANGE, "Negative catol not allowed"); 906606f75f6SBarry Smith tao->catol = catol; 907a7e14dcfSSatish Balay } 908a7e14dcfSSatish Balay 909606f75f6SBarry Smith if (crtol == (PetscReal)PETSC_DETERMINE) { 910606f75f6SBarry Smith tao->crtol = tao->default_crtol; 911606f75f6SBarry Smith } else if (crtol != (PetscReal)PETSC_CURRENT) { 912606f75f6SBarry Smith PetscCheck(crtol >= 0, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_OUTOFRANGE, "Negative crtol not allowed"); 913606f75f6SBarry Smith tao->crtol = crtol; 914a7e14dcfSSatish Balay } 9153ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 916a7e14dcfSSatish Balay } 917a7e14dcfSSatish Balay 91858417fe7SBarry Smith /*@ 91947450a7bSBarry Smith TaoGetConstraintTolerances - Gets constraint tolerance parameters used in `TaoSolve()` convergence tests 92058417fe7SBarry Smith 92167be906fSBarry Smith Not Collective 92258417fe7SBarry Smith 92358417fe7SBarry Smith Input Parameter: 92447450a7bSBarry Smith . tao - the `Tao` context 92558417fe7SBarry Smith 926d8d19677SJose E. Roman Output Parameters: 927606f75f6SBarry Smith + catol - absolute constraint tolerance, constraint norm must be less than `catol` for used for `gatol` convergence criteria 928606f75f6SBarry Smith - crtol - relative constraint tolerance, constraint norm must be less than `crtol` for used for `gatol`, `gttol` convergence criteria 92958417fe7SBarry Smith 93058417fe7SBarry Smith Level: intermediate 93158417fe7SBarry Smith 9321cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReasons`,`TaoGetTolerances()`, `TaoSetTolerances()`, `TaoSetConstraintTolerances()` 93358417fe7SBarry Smith @*/ 934d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConstraintTolerances(Tao tao, PetscReal *catol, PetscReal *crtol) 935d71ae5a4SJacob Faibussowitsch { 93658417fe7SBarry Smith PetscFunctionBegin; 93758417fe7SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 93858417fe7SBarry Smith if (catol) *catol = tao->catol; 93958417fe7SBarry Smith if (crtol) *crtol = tao->crtol; 9403ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 94158417fe7SBarry Smith } 94258417fe7SBarry Smith 943a7e14dcfSSatish Balay /*@ 944a7e14dcfSSatish Balay TaoSetFunctionLowerBound - Sets a bound on the solution objective value. 945a7e14dcfSSatish Balay When an approximate solution with an objective value below this number 946a7e14dcfSSatish Balay has been found, the solver will terminate. 947a7e14dcfSSatish Balay 948c3339decSBarry Smith Logically Collective 949a7e14dcfSSatish Balay 950a7e14dcfSSatish Balay Input Parameters: 951441846f8SBarry Smith + tao - the Tao solver context 952a7e14dcfSSatish Balay - fmin - the tolerance 953a7e14dcfSSatish Balay 95447450a7bSBarry Smith Options Database Key: 955a7e14dcfSSatish Balay . -tao_fmin <fmin> - sets the minimum function value 956a7e14dcfSSatish Balay 957a7e14dcfSSatish Balay Level: intermediate 958a7e14dcfSSatish Balay 9591cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetTolerances()` 960a7e14dcfSSatish Balay @*/ 961d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetFunctionLowerBound(Tao tao, PetscReal fmin) 962d71ae5a4SJacob Faibussowitsch { 963a7e14dcfSSatish Balay PetscFunctionBegin; 964441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 96594511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, fmin, 2); 966a7e14dcfSSatish Balay tao->fmin = fmin; 9673ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 968a7e14dcfSSatish Balay } 969a7e14dcfSSatish Balay 970a7e14dcfSSatish Balay /*@ 9718e96d397SJason Sarich TaoGetFunctionLowerBound - Gets the bound on the solution objective value. 972a7e14dcfSSatish Balay When an approximate solution with an objective value below this number 973a7e14dcfSSatish Balay has been found, the solver will terminate. 974a7e14dcfSSatish Balay 97567be906fSBarry Smith Not Collective 976a7e14dcfSSatish Balay 97747450a7bSBarry Smith Input Parameter: 97847450a7bSBarry Smith . tao - the `Tao` solver context 979a7e14dcfSSatish Balay 98047450a7bSBarry Smith Output Parameter: 981a7e14dcfSSatish Balay . fmin - the minimum function value 982a7e14dcfSSatish Balay 983a7e14dcfSSatish Balay Level: intermediate 984a7e14dcfSSatish Balay 9851cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetFunctionLowerBound()` 986a7e14dcfSSatish Balay @*/ 987d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetFunctionLowerBound(Tao tao, PetscReal *fmin) 988d71ae5a4SJacob Faibussowitsch { 989a7e14dcfSSatish Balay PetscFunctionBegin; 990441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 9914f572ea9SToby Isaac PetscAssertPointer(fmin, 2); 992a7e14dcfSSatish Balay *fmin = tao->fmin; 9933ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 994a7e14dcfSSatish Balay } 995a7e14dcfSSatish Balay 996a7e14dcfSSatish Balay /*@ 99747450a7bSBarry Smith TaoSetMaximumFunctionEvaluations - Sets a maximum number of function evaluations allowed for a `TaoSolve()`. 998a7e14dcfSSatish Balay 999c3339decSBarry Smith Logically Collective 1000a7e14dcfSSatish Balay 1001a7e14dcfSSatish Balay Input Parameters: 100247450a7bSBarry Smith + tao - the `Tao` solver context 1003606f75f6SBarry Smith - nfcn - the maximum number of function evaluations (>=0), use `PETSC_UNLIMITED` to have no bound 1004a7e14dcfSSatish Balay 100547450a7bSBarry Smith Options Database Key: 1006a7e14dcfSSatish Balay . -tao_max_funcs <nfcn> - sets the maximum number of function evaluations 1007a7e14dcfSSatish Balay 1008a7e14dcfSSatish Balay Level: intermediate 1009a7e14dcfSSatish Balay 1010606f75f6SBarry Smith Note: 1011606f75f6SBarry Smith Use `PETSC_DETERMINE` to use the default maximum number of function evaluations that was set when the object type was set. 1012606f75f6SBarry Smith 1013606f75f6SBarry Smith Developer Note: 1014606f75f6SBarry Smith Deprecated support for an unlimited number of function evaluations by passing a negative value. 1015606f75f6SBarry Smith 10161cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoSetMaximumIterations()` 1017a7e14dcfSSatish Balay @*/ 1018d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao tao, PetscInt nfcn) 1019d71ae5a4SJacob Faibussowitsch { 1020a7e14dcfSSatish Balay PetscFunctionBegin; 1021441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 102294511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao, nfcn, 2); 1023606f75f6SBarry Smith if (nfcn == PETSC_DETERMINE) { 1024606f75f6SBarry Smith tao->max_funcs = tao->default_max_funcs; 1025606f75f6SBarry Smith } else if (nfcn == PETSC_UNLIMITED || nfcn < 0) { 1026606f75f6SBarry Smith tao->max_funcs = PETSC_UNLIMITED; 10279371c9d4SSatish Balay } else { 1028d7c1f440SPierre Jolivet PetscCheck(nfcn >= 0, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_OUTOFRANGE, "Maximum number of function evaluations must be positive"); 1029606f75f6SBarry Smith tao->max_funcs = nfcn; 10309371c9d4SSatish Balay } 10313ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1032a7e14dcfSSatish Balay } 1033a7e14dcfSSatish Balay 1034a7e14dcfSSatish Balay /*@ 103547450a7bSBarry Smith TaoGetMaximumFunctionEvaluations - Gets a maximum number of function evaluations allowed for a `TaoSolve()` 1036a7e14dcfSSatish Balay 1037c3339decSBarry Smith Logically Collective 1038a7e14dcfSSatish Balay 103947450a7bSBarry Smith Input Parameter: 104047450a7bSBarry Smith . tao - the `Tao` solver context 1041a7e14dcfSSatish Balay 104247450a7bSBarry Smith Output Parameter: 1043a7e14dcfSSatish Balay . nfcn - the maximum number of function evaluations 1044a7e14dcfSSatish Balay 1045a7e14dcfSSatish Balay Level: intermediate 1046a7e14dcfSSatish Balay 10471cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()` 1048a7e14dcfSSatish Balay @*/ 1049d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao tao, PetscInt *nfcn) 1050d71ae5a4SJacob Faibussowitsch { 1051a7e14dcfSSatish Balay PetscFunctionBegin; 1052441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 10534f572ea9SToby Isaac PetscAssertPointer(nfcn, 2); 1054a7e14dcfSSatish Balay *nfcn = tao->max_funcs; 10553ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1056a7e14dcfSSatish Balay } 1057a7e14dcfSSatish Balay 1058770232b9SCe Qin /*@ 105947450a7bSBarry Smith TaoGetCurrentFunctionEvaluations - Get current number of function evaluations used by a `Tao` object 1060770232b9SCe Qin 1061770232b9SCe Qin Not Collective 1062770232b9SCe Qin 106347450a7bSBarry Smith Input Parameter: 106447450a7bSBarry Smith . tao - the `Tao` solver context 1065770232b9SCe Qin 106647450a7bSBarry Smith Output Parameter: 106794511df7SStefano Zampini . nfuncs - the current number of function evaluations (maximum between gradient and function evaluations) 1068770232b9SCe Qin 1069770232b9SCe Qin Level: intermediate 1070770232b9SCe Qin 10711cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()` 1072770232b9SCe Qin @*/ 1073d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao tao, PetscInt *nfuncs) 1074d71ae5a4SJacob Faibussowitsch { 1075770232b9SCe Qin PetscFunctionBegin; 1076770232b9SCe Qin PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 10774f572ea9SToby Isaac PetscAssertPointer(nfuncs, 2); 1078770232b9SCe Qin *nfuncs = PetscMax(tao->nfuncs, tao->nfuncgrads); 10793ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1080770232b9SCe Qin } 1081770232b9SCe Qin 1082a7e14dcfSSatish Balay /*@ 108347450a7bSBarry Smith TaoSetMaximumIterations - Sets a maximum number of iterates to be used in `TaoSolve()` 1084a7e14dcfSSatish Balay 1085c3339decSBarry Smith Logically Collective 1086a7e14dcfSSatish Balay 1087a7e14dcfSSatish Balay Input Parameters: 108847450a7bSBarry Smith + tao - the `Tao` solver context 1089606f75f6SBarry Smith - maxits - the maximum number of iterates (>=0), use `PETSC_UNLIMITED` to have no bound 1090a7e14dcfSSatish Balay 109147450a7bSBarry Smith Options Database Key: 1092a7e14dcfSSatish Balay . -tao_max_it <its> - sets the maximum number of iterations 1093a7e14dcfSSatish Balay 1094a7e14dcfSSatish Balay Level: intermediate 1095a7e14dcfSSatish Balay 1096606f75f6SBarry Smith Note: 1097606f75f6SBarry Smith Use `PETSC_DETERMINE` to use the default maximum number of iterations that was set when the object's type was set. 1098606f75f6SBarry Smith 1099606f75f6SBarry Smith Developer Note: 1100606f75f6SBarry Smith DeprAlso accepts the deprecated negative values to indicate no limit 1101606f75f6SBarry Smith 11021cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoSetMaximumFunctionEvaluations()` 1103a7e14dcfSSatish Balay @*/ 1104d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetMaximumIterations(Tao tao, PetscInt maxits) 1105d71ae5a4SJacob Faibussowitsch { 1106a7e14dcfSSatish Balay PetscFunctionBegin; 1107441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 110894511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao, maxits, 2); 1109606f75f6SBarry Smith if (maxits == PETSC_DETERMINE) { 1110606f75f6SBarry Smith tao->max_it = tao->default_max_it; 1111606f75f6SBarry Smith } else if (maxits == PETSC_UNLIMITED) { 11121690c2aeSBarry Smith tao->max_it = PETSC_INT_MAX; 1113606f75f6SBarry Smith } else { 1114d7c1f440SPierre Jolivet PetscCheck(maxits > 0, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_OUTOFRANGE, "Maximum number of iterations must be positive"); 1115606f75f6SBarry Smith tao->max_it = maxits; 1116606f75f6SBarry Smith } 11173ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1118a7e14dcfSSatish Balay } 1119a7e14dcfSSatish Balay 1120a7e14dcfSSatish Balay /*@ 112165ba42b6SBarry Smith TaoGetMaximumIterations - Gets a maximum number of iterates that will be used 1122a7e14dcfSSatish Balay 1123a7e14dcfSSatish Balay Not Collective 1124a7e14dcfSSatish Balay 112547450a7bSBarry Smith Input Parameter: 112647450a7bSBarry Smith . tao - the `Tao` solver context 1127a7e14dcfSSatish Balay 112847450a7bSBarry Smith Output Parameter: 1129a7e14dcfSSatish Balay . maxits - the maximum number of iterates 1130a7e14dcfSSatish Balay 1131a7e14dcfSSatish Balay Level: intermediate 1132a7e14dcfSSatish Balay 11331cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetMaximumIterations()`, `TaoGetMaximumFunctionEvaluations()` 1134a7e14dcfSSatish Balay @*/ 1135d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetMaximumIterations(Tao tao, PetscInt *maxits) 1136d71ae5a4SJacob Faibussowitsch { 1137a7e14dcfSSatish Balay PetscFunctionBegin; 1138441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 11394f572ea9SToby Isaac PetscAssertPointer(maxits, 2); 1140a7e14dcfSSatish Balay *maxits = tao->max_it; 11413ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1142a7e14dcfSSatish Balay } 1143a7e14dcfSSatish Balay 1144a7e14dcfSSatish Balay /*@ 1145a7e14dcfSSatish Balay TaoSetInitialTrustRegionRadius - Sets the initial trust region radius. 1146a7e14dcfSSatish Balay 114767be906fSBarry Smith Logically Collective 1148a7e14dcfSSatish Balay 1149d8d19677SJose E. Roman Input Parameters: 115047450a7bSBarry Smith + tao - a `Tao` optimization solver 1151a7e14dcfSSatish Balay - radius - the trust region radius 1152a7e14dcfSSatish Balay 1153a7e14dcfSSatish Balay Options Database Key: 1154a7e14dcfSSatish Balay . -tao_trust0 <t0> - sets initial trust region radius 1155a7e14dcfSSatish Balay 115647450a7bSBarry Smith Level: intermediate 115747450a7bSBarry Smith 1158606f75f6SBarry Smith Note: 1159606f75f6SBarry Smith Use `PETSC_DETERMINE` to use the default radius that was set when the object's type was set. 1160606f75f6SBarry Smith 11611cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetTrustRegionRadius()`, `TaoSetTrustRegionTolerance()`, `TAONTR` 1162a7e14dcfSSatish Balay @*/ 1163d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetInitialTrustRegionRadius(Tao tao, PetscReal radius) 1164d71ae5a4SJacob Faibussowitsch { 1165a7e14dcfSSatish Balay PetscFunctionBegin; 1166441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 116794511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao, radius, 2); 1168606f75f6SBarry Smith if (radius == PETSC_DETERMINE) { 1169606f75f6SBarry Smith tao->trust0 = tao->default_trust0; 1170606f75f6SBarry Smith } else { 1171606f75f6SBarry Smith PetscCheck(radius > 0, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_OUTOFRANGE, "Radius must be positive"); 1172606f75f6SBarry Smith tao->trust0 = radius; 1173606f75f6SBarry Smith } 11743ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1175a7e14dcfSSatish Balay } 1176a7e14dcfSSatish Balay 1177a7e14dcfSSatish Balay /*@ 117865ba42b6SBarry Smith TaoGetInitialTrustRegionRadius - Gets the initial trust region radius. 1179a7e14dcfSSatish Balay 1180a7e14dcfSSatish Balay Not Collective 1181a7e14dcfSSatish Balay 1182a7e14dcfSSatish Balay Input Parameter: 118347450a7bSBarry Smith . tao - a `Tao` optimization solver 1184a7e14dcfSSatish Balay 1185a7e14dcfSSatish Balay Output Parameter: 1186a7e14dcfSSatish Balay . radius - the trust region radius 1187a7e14dcfSSatish Balay 1188a7e14dcfSSatish Balay Level: intermediate 1189a7e14dcfSSatish Balay 11901cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetInitialTrustRegionRadius()`, `TaoGetCurrentTrustRegionRadius()`, `TAONTR` 1191a7e14dcfSSatish Balay @*/ 1192d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetInitialTrustRegionRadius(Tao tao, PetscReal *radius) 1193d71ae5a4SJacob Faibussowitsch { 1194a7e14dcfSSatish Balay PetscFunctionBegin; 1195441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 11964f572ea9SToby Isaac PetscAssertPointer(radius, 2); 1197a7e14dcfSSatish Balay *radius = tao->trust0; 11983ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1199a7e14dcfSSatish Balay } 1200a7e14dcfSSatish Balay 1201a7e14dcfSSatish Balay /*@ 1202a7e14dcfSSatish Balay TaoGetCurrentTrustRegionRadius - Gets the current trust region radius. 1203a7e14dcfSSatish Balay 1204a7e14dcfSSatish Balay Not Collective 1205a7e14dcfSSatish Balay 1206a7e14dcfSSatish Balay Input Parameter: 120747450a7bSBarry Smith . tao - a `Tao` optimization solver 1208a7e14dcfSSatish Balay 1209a7e14dcfSSatish Balay Output Parameter: 1210a7e14dcfSSatish Balay . radius - the trust region radius 1211a7e14dcfSSatish Balay 1212a7e14dcfSSatish Balay Level: intermediate 1213a7e14dcfSSatish Balay 12141cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetInitialTrustRegionRadius()`, `TaoGetInitialTrustRegionRadius()`, `TAONTR` 1215a7e14dcfSSatish Balay @*/ 1216d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao tao, PetscReal *radius) 1217d71ae5a4SJacob Faibussowitsch { 1218a7e14dcfSSatish Balay PetscFunctionBegin; 1219441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 12204f572ea9SToby Isaac PetscAssertPointer(radius, 2); 1221a7e14dcfSSatish Balay *radius = tao->trust; 12223ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1223a7e14dcfSSatish Balay } 1224a7e14dcfSSatish Balay 1225a7e14dcfSSatish Balay /*@ 122647450a7bSBarry Smith TaoGetTolerances - gets the current values of some tolerances used for the convergence testing of `TaoSolve()` 1227a7e14dcfSSatish Balay 1228a7e14dcfSSatish Balay Not Collective 1229a7e14dcfSSatish Balay 1230f899ff85SJose E. Roman Input Parameter: 123147450a7bSBarry Smith . tao - the `Tao` context 1232a7e14dcfSSatish Balay 1233a7e14dcfSSatish Balay Output Parameters: 1234e52336cbSBarry Smith + gatol - stop if norm of gradient is less than this 1235a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this 1236a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by a this factor 1237a7e14dcfSSatish Balay 1238a7e14dcfSSatish Balay Level: intermediate 1239a1cb98faSBarry Smith 1240a1cb98faSBarry Smith Note: 124147450a7bSBarry Smith `NULL` can be used as an argument if not all tolerances values are needed 1242a1cb98faSBarry Smith 12431cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()` 1244a7e14dcfSSatish Balay @*/ 1245d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetTolerances(Tao tao, PetscReal *gatol, PetscReal *grtol, PetscReal *gttol) 1246d71ae5a4SJacob Faibussowitsch { 1247a7e14dcfSSatish Balay PetscFunctionBegin; 1248441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1249a7e14dcfSSatish Balay if (gatol) *gatol = tao->gatol; 1250a7e14dcfSSatish Balay if (grtol) *grtol = tao->grtol; 1251a7e14dcfSSatish Balay if (gttol) *gttol = tao->gttol; 12523ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1253a7e14dcfSSatish Balay } 1254a7e14dcfSSatish Balay 1255a7e14dcfSSatish Balay /*@ 1256a7e14dcfSSatish Balay TaoGetKSP - Gets the linear solver used by the optimization solver. 1257a7e14dcfSSatish Balay 1258a7e14dcfSSatish Balay Not Collective 1259a7e14dcfSSatish Balay 126047450a7bSBarry Smith Input Parameter: 126147450a7bSBarry Smith . tao - the `Tao` solver 1262a7e14dcfSSatish Balay 126347450a7bSBarry Smith Output Parameter: 126447450a7bSBarry Smith . ksp - the `KSP` linear solver used in the optimization solver 1265a7e14dcfSSatish Balay 1266a7e14dcfSSatish Balay Level: intermediate 1267a7e14dcfSSatish Balay 12681cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `KSP` 1269a7e14dcfSSatish Balay @*/ 1270d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetKSP(Tao tao, KSP *ksp) 1271d71ae5a4SJacob Faibussowitsch { 1272a7e14dcfSSatish Balay PetscFunctionBegin; 127394511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 12744f572ea9SToby Isaac PetscAssertPointer(ksp, 2); 1275a7e14dcfSSatish Balay *ksp = tao->ksp; 12763ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1277a7e14dcfSSatish Balay } 1278a7e14dcfSSatish Balay 1279025e9500SJason Sarich /*@ 1280025e9500SJason Sarich TaoGetLinearSolveIterations - Gets the total number of linear iterations 128147450a7bSBarry Smith used by the `Tao` solver 1282025e9500SJason Sarich 1283025e9500SJason Sarich Not Collective 1284025e9500SJason Sarich 1285025e9500SJason Sarich Input Parameter: 128647450a7bSBarry Smith . tao - the `Tao` context 1287025e9500SJason Sarich 1288025e9500SJason Sarich Output Parameter: 1289025e9500SJason Sarich . lits - number of linear iterations 1290025e9500SJason Sarich 1291025e9500SJason Sarich Level: intermediate 1292025e9500SJason Sarich 129347450a7bSBarry Smith Note: 129447450a7bSBarry Smith This counter is reset to zero for each successive call to `TaoSolve()` 129547450a7bSBarry Smith 12961cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetKSP()` 1297025e9500SJason Sarich @*/ 1298d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetLinearSolveIterations(Tao tao, PetscInt *lits) 1299d71ae5a4SJacob Faibussowitsch { 1300025e9500SJason Sarich PetscFunctionBegin; 1301025e9500SJason Sarich PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 13024f572ea9SToby Isaac PetscAssertPointer(lits, 2); 13032d9aa51bSJason Sarich *lits = tao->ksp_tot_its; 13043ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1305025e9500SJason Sarich } 1306025e9500SJason Sarich 1307a7e14dcfSSatish Balay /*@ 1308a7e14dcfSSatish Balay TaoGetLineSearch - Gets the line search used by the optimization solver. 1309a7e14dcfSSatish Balay 1310a7e14dcfSSatish Balay Not Collective 1311a7e14dcfSSatish Balay 131247450a7bSBarry Smith Input Parameter: 131347450a7bSBarry Smith . tao - the `Tao` solver 1314a7e14dcfSSatish Balay 131547450a7bSBarry Smith Output Parameter: 1316a7e14dcfSSatish Balay . ls - the line search used in the optimization solver 1317a7e14dcfSSatish Balay 1318a7e14dcfSSatish Balay Level: intermediate 1319a7e14dcfSSatish Balay 13201cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoLineSearch`, `TaoLineSearchType` 1321a7e14dcfSSatish Balay @*/ 1322d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetLineSearch(Tao tao, TaoLineSearch *ls) 1323d71ae5a4SJacob Faibussowitsch { 1324a7e14dcfSSatish Balay PetscFunctionBegin; 132594511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 13264f572ea9SToby Isaac PetscAssertPointer(ls, 2); 1327a7e14dcfSSatish Balay *ls = tao->linesearch; 13283ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1329a7e14dcfSSatish Balay } 1330a7e14dcfSSatish Balay 1331a7e14dcfSSatish Balay /*@ 1332a7e14dcfSSatish Balay TaoAddLineSearchCounts - Adds the number of function evaluations spent 1333a7e14dcfSSatish Balay in the line search to the running total. 1334a7e14dcfSSatish Balay 1335a7e14dcfSSatish Balay Input Parameters: 13362fe279fdSBarry Smith . tao - the `Tao` solver 1337a7e14dcfSSatish Balay 1338a7e14dcfSSatish Balay Level: developer 1339a7e14dcfSSatish Balay 13401cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLineSearch()`, `TaoLineSearchApply()` 1341a7e14dcfSSatish Balay @*/ 1342d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoAddLineSearchCounts(Tao tao) 1343d71ae5a4SJacob Faibussowitsch { 1344a7e14dcfSSatish Balay PetscBool flg; 1345a7e14dcfSSatish Balay PetscInt nfeval, ngeval, nfgeval; 134647a47007SBarry Smith 1347a7e14dcfSSatish Balay PetscFunctionBegin; 1348441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1349a7e14dcfSSatish Balay if (tao->linesearch) { 13509566063dSJacob Faibussowitsch PetscCall(TaoLineSearchIsUsingTaoRoutines(tao->linesearch, &flg)); 135194ae4db5SBarry Smith if (!flg) { 13529566063dSJacob Faibussowitsch PetscCall(TaoLineSearchGetNumberFunctionEvaluations(tao->linesearch, &nfeval, &ngeval, &nfgeval)); 1353a7e14dcfSSatish Balay tao->nfuncs += nfeval; 1354a7e14dcfSSatish Balay tao->ngrads += ngeval; 1355a7e14dcfSSatish Balay tao->nfuncgrads += nfgeval; 1356a7e14dcfSSatish Balay } 1357a7e14dcfSSatish Balay } 13583ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1359a7e14dcfSSatish Balay } 1360a7e14dcfSSatish Balay 1361a7e14dcfSSatish Balay /*@ 136247450a7bSBarry Smith TaoGetSolution - Returns the vector with the current solution from the `Tao` object 1363a7e14dcfSSatish Balay 1364a7e14dcfSSatish Balay Not Collective 1365a7e14dcfSSatish Balay 1366a7e14dcfSSatish Balay Input Parameter: 136747450a7bSBarry Smith . tao - the `Tao` context 1368a7e14dcfSSatish Balay 1369a7e14dcfSSatish Balay Output Parameter: 1370a7e14dcfSSatish Balay . X - the current solution 1371a7e14dcfSSatish Balay 1372a7e14dcfSSatish Balay Level: intermediate 1373a7e14dcfSSatish Balay 137465ba42b6SBarry Smith Note: 137565ba42b6SBarry Smith The returned vector will be the same object that was passed into `TaoSetSolution()` 137665ba42b6SBarry Smith 13771cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetSolution()`, `TaoSolve()` 1378a7e14dcfSSatish Balay @*/ 1379d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetSolution(Tao tao, Vec *X) 1380d71ae5a4SJacob Faibussowitsch { 1381a7e14dcfSSatish Balay PetscFunctionBegin; 1382441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 13834f572ea9SToby Isaac PetscAssertPointer(X, 2); 1384a7e14dcfSSatish Balay *X = tao->solution; 13853ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1386a7e14dcfSSatish Balay } 1387a7e14dcfSSatish Balay 1388a7e14dcfSSatish Balay /*@ 138947450a7bSBarry Smith TaoResetStatistics - Initialize the statistics collected by the `Tao` object. 1390a7e14dcfSSatish Balay These statistics include the iteration number, residual norms, and convergence status. 1391a7e14dcfSSatish Balay This routine gets called before solving each optimization problem. 1392a7e14dcfSSatish Balay 1393c3339decSBarry Smith Collective 1394a7e14dcfSSatish Balay 139547450a7bSBarry Smith Input Parameter: 1396e056e8ceSJacob Faibussowitsch . tao - the `Tao` context 1397a7e14dcfSSatish Balay 1398a7e14dcfSSatish Balay Level: developer 1399a7e14dcfSSatish Balay 14001cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoSolve()` 1401a7e14dcfSSatish Balay @*/ 1402d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoResetStatistics(Tao tao) 1403d71ae5a4SJacob Faibussowitsch { 1404a7e14dcfSSatish Balay PetscFunctionBegin; 1405441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1406a7e14dcfSSatish Balay tao->niter = 0; 1407a7e14dcfSSatish Balay tao->nfuncs = 0; 1408a7e14dcfSSatish Balay tao->nfuncgrads = 0; 1409a7e14dcfSSatish Balay tao->ngrads = 0; 1410a7e14dcfSSatish Balay tao->nhess = 0; 1411a7e14dcfSSatish Balay tao->njac = 0; 1412a7e14dcfSSatish Balay tao->nconstraints = 0; 1413a7e14dcfSSatish Balay tao->ksp_its = 0; 1414ae93cb3cSJason Sarich tao->ksp_tot_its = 0; 1415a7e14dcfSSatish Balay tao->reason = TAO_CONTINUE_ITERATING; 1416a7e14dcfSSatish Balay tao->residual = 0.0; 1417a7e14dcfSSatish Balay tao->cnorm = 0.0; 1418a7e14dcfSSatish Balay tao->step = 0.0; 1419a7e14dcfSSatish Balay tao->lsflag = PETSC_FALSE; 1420a7e14dcfSSatish Balay if (tao->hist_reset) tao->hist_len = 0; 14213ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1422a7e14dcfSSatish Balay } 1423a7e14dcfSSatish Balay 1424a7e14dcfSSatish Balay /*@C 1425e1e80dc8SAlp Dener TaoSetUpdate - Sets the general-purpose update function called 14262fe279fdSBarry Smith at the beginning of every iteration of the optimization algorithm. Called after the new solution and the gradient 1427e1e80dc8SAlp Dener is determined, but before the Hessian is computed (if applicable). 1428e1e80dc8SAlp Dener 1429c3339decSBarry Smith Logically Collective 1430e1e80dc8SAlp Dener 1431e1e80dc8SAlp Dener Input Parameters: 1432270bebe6SStefano Zampini + tao - The `Tao` solver 1433270bebe6SStefano Zampini . func - The function 1434270bebe6SStefano Zampini - ctx - The update function context 1435e1e80dc8SAlp Dener 143620f4b53cSBarry Smith Calling sequence of `func`: 1437270bebe6SStefano Zampini + tao - The optimizer context 1438270bebe6SStefano Zampini . it - The current iteration index 1439270bebe6SStefano Zampini - ctx - The update context 1440e1e80dc8SAlp Dener 1441e1e80dc8SAlp Dener Level: advanced 1442e1e80dc8SAlp Dener 1443270bebe6SStefano Zampini Notes: 1444270bebe6SStefano Zampini Users can modify the gradient direction or any other vector associated to the specific solver used. 1445270bebe6SStefano Zampini The objective function value is always recomputed after a call to the update hook. 1446270bebe6SStefano Zampini 14471cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()` 1448e1e80dc8SAlp Dener @*/ 1449270bebe6SStefano Zampini PetscErrorCode TaoSetUpdate(Tao tao, PetscErrorCode (*func)(Tao tao, PetscInt it, void *ctx), void *ctx) 1450d71ae5a4SJacob Faibussowitsch { 1451e1e80dc8SAlp Dener PetscFunctionBegin; 1452e1e80dc8SAlp Dener PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1453e1e80dc8SAlp Dener tao->ops->update = func; 1454e1e80dc8SAlp Dener tao->user_update = ctx; 14553ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1456e1e80dc8SAlp Dener } 1457e1e80dc8SAlp Dener 1458e1e80dc8SAlp Dener /*@C 1459a7e14dcfSSatish Balay TaoSetConvergenceTest - Sets the function that is to be used to test 1460a7e14dcfSSatish Balay for convergence of the iterative minimization solution. The new convergence 146165ba42b6SBarry Smith testing routine will replace Tao's default convergence test. 1462a7e14dcfSSatish Balay 1463c3339decSBarry Smith Logically Collective 1464a7e14dcfSSatish Balay 1465a7e14dcfSSatish Balay Input Parameters: 146647450a7bSBarry Smith + tao - the `Tao` object 1467a7e14dcfSSatish Balay . conv - the routine to test for convergence 1468a7e14dcfSSatish Balay - ctx - [optional] context for private data for the convergence routine 146947450a7bSBarry Smith (may be `NULL`) 1470a7e14dcfSSatish Balay 147120f4b53cSBarry Smith Calling sequence of `conv`: 147247450a7bSBarry Smith + tao - the `Tao` object 1473a7e14dcfSSatish Balay - ctx - [optional] convergence context 1474a7e14dcfSSatish Balay 147547450a7bSBarry Smith Level: advanced 147647450a7bSBarry Smith 147765ba42b6SBarry Smith Note: 147865ba42b6SBarry Smith The new convergence testing routine should call `TaoSetConvergedReason()`. 1479a7e14dcfSSatish Balay 148010978b7dSBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoSetConvergedReason()`, `TaoGetSolutionStatus()`, `TaoGetTolerances()`, `TaoMonitorSet()` 1481a7e14dcfSSatish Balay @*/ 1482d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergenceTest(Tao tao, PetscErrorCode (*conv)(Tao, void *), void *ctx) 1483d71ae5a4SJacob Faibussowitsch { 1484a7e14dcfSSatish Balay PetscFunctionBegin; 1485441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 148694511df7SStefano Zampini tao->ops->convergencetest = conv; 148794511df7SStefano Zampini tao->cnvP = ctx; 14883ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1489a7e14dcfSSatish Balay } 1490a7e14dcfSSatish Balay 1491a7e14dcfSSatish Balay /*@C 149210978b7dSBarry Smith TaoMonitorSet - Sets an additional function that is to be used at every 1493a7e14dcfSSatish Balay iteration of the solver to display the iteration's 1494a7e14dcfSSatish Balay progress. 1495a7e14dcfSSatish Balay 1496c3339decSBarry Smith Logically Collective 1497a7e14dcfSSatish Balay 1498a7e14dcfSSatish Balay Input Parameters: 149947450a7bSBarry Smith + tao - the `Tao` solver context 15002fe279fdSBarry Smith . func - monitoring routine 15012fe279fdSBarry Smith . ctx - [optional] user-defined context for private data for the monitor routine (may be `NULL`) 1502*49abdd8aSBarry Smith - dest - [optional] function to destroy the context when the `Tao` is destroyed, see `PetscCtxDestroyFn` for the calling sequence 1503a7e14dcfSSatish Balay 15042fe279fdSBarry Smith Calling sequence of `func`: 150547450a7bSBarry Smith + tao - the `Tao` solver context 15062fe279fdSBarry Smith - ctx - [optional] monitoring context 15072fe279fdSBarry Smith 150847450a7bSBarry Smith Level: intermediate 150947450a7bSBarry Smith 151087497f52SBarry Smith Notes: 151110978b7dSBarry Smith See `TaoSetFromOptions()` for a monitoring options. 151210978b7dSBarry Smith 1513a7e14dcfSSatish Balay Several different monitoring routines may be set by calling 151410978b7dSBarry Smith `TaoMonitorSet()` multiple times; all will be called in the 1515a7e14dcfSSatish Balay order in which they were set. 1516a7e14dcfSSatish Balay 1517e056e8ceSJacob Faibussowitsch Fortran Notes: 151895452b02SPatrick Sanan Only one monitor function may be set 1519a7e14dcfSSatish Balay 1520*49abdd8aSBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoMonitorDefault()`, `TaoMonitorCancel()`, `TaoSetDestroyRoutine()`, `TaoView()`, `PetscCtxDestroyFn` 1521a7e14dcfSSatish Balay @*/ 1522*49abdd8aSBarry Smith PetscErrorCode TaoMonitorSet(Tao tao, PetscErrorCode (*func)(Tao, void *), void *ctx, PetscCtxDestroyFn *dest) 1523d71ae5a4SJacob Faibussowitsch { 152408d19d1fSJason Sarich PetscInt i; 15258732526dSAlp Dener PetscBool identical; 152608d19d1fSJason Sarich 1527a7e14dcfSSatish Balay PetscFunctionBegin; 1528441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 15293c859ba3SBarry Smith PetscCheck(tao->numbermonitors < MAXTAOMONITORS, PetscObjectComm((PetscObject)tao), PETSC_ERR_SUP, "Cannot attach another monitor -- max=%d", MAXTAOMONITORS); 153008d19d1fSJason Sarich 153108d19d1fSJason Sarich for (i = 0; i < tao->numbermonitors; i++) { 15329566063dSJacob Faibussowitsch PetscCall(PetscMonitorCompare((PetscErrorCode (*)(void))func, ctx, dest, (PetscErrorCode (*)(void))tao->monitor[i], tao->monitorcontext[i], tao->monitordestroy[i], &identical)); 15333ba16761SJacob Faibussowitsch if (identical) PetscFunctionReturn(PETSC_SUCCESS); 153408d19d1fSJason Sarich } 1535a7e14dcfSSatish Balay tao->monitor[tao->numbermonitors] = func; 15368732526dSAlp Dener tao->monitorcontext[tao->numbermonitors] = (void *)ctx; 1537a7e14dcfSSatish Balay tao->monitordestroy[tao->numbermonitors] = dest; 1538a7e14dcfSSatish Balay ++tao->numbermonitors; 15393ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1540a7e14dcfSSatish Balay } 1541a7e14dcfSSatish Balay 1542a7e14dcfSSatish Balay /*@ 1543b2dc45ddSPierre Jolivet TaoMonitorCancel - Clears all the monitor functions for a `Tao` object. 1544a7e14dcfSSatish Balay 1545c3339decSBarry Smith Logically Collective 1546a7e14dcfSSatish Balay 154747450a7bSBarry Smith Input Parameter: 154847450a7bSBarry Smith . tao - the `Tao` solver context 1549a7e14dcfSSatish Balay 15503c7db156SBarry Smith Options Database Key: 1551b2dc45ddSPierre Jolivet . -tao_monitor_cancel - cancels all monitors that have been hardwired 155210978b7dSBarry Smith into a code by calls to `TaoMonitorSet()`, but does not cancel those 1553a7e14dcfSSatish Balay set via the options database 1554a7e14dcfSSatish Balay 1555a7e14dcfSSatish Balay Level: advanced 1556a7e14dcfSSatish Balay 155747450a7bSBarry Smith Note: 155847450a7bSBarry Smith There is no way to clear one specific monitor from a `Tao` object. 155947450a7bSBarry Smith 156010978b7dSBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoMonitorSet()` 1561a7e14dcfSSatish Balay @*/ 1562b2dc45ddSPierre Jolivet PetscErrorCode TaoMonitorCancel(Tao tao) 1563d71ae5a4SJacob Faibussowitsch { 1564a7e14dcfSSatish Balay PetscInt i; 156547a47007SBarry Smith 1566a7e14dcfSSatish Balay PetscFunctionBegin; 1567441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1568a7e14dcfSSatish Balay for (i = 0; i < tao->numbermonitors; i++) { 156948a46eb9SPierre Jolivet if (tao->monitordestroy[i]) PetscCall((*tao->monitordestroy[i])(&tao->monitorcontext[i])); 1570a7e14dcfSSatish Balay } 1571a7e14dcfSSatish Balay tao->numbermonitors = 0; 15723ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1573a7e14dcfSSatish Balay } 1574a7e14dcfSSatish Balay 15757fab98ebSJason Sarich /*@ 157647450a7bSBarry Smith TaoMonitorDefault - Default routine for monitoring progress of `TaoSolve()` 1577a7e14dcfSSatish Balay 1578c3339decSBarry Smith Collective 1579a7e14dcfSSatish Balay 1580a7e14dcfSSatish Balay Input Parameters: 158147450a7bSBarry Smith + tao - the `Tao` context 158247450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL` 1583a7e14dcfSSatish Balay 158447450a7bSBarry Smith Options Database Key: 1585147403d9SBarry Smith . -tao_monitor - turn on default monitoring 1586a7e14dcfSSatish Balay 1587a7e14dcfSSatish Balay Level: advanced 1588a7e14dcfSSatish Balay 158947450a7bSBarry Smith Note: 159047450a7bSBarry Smith This monitor prints the function value and gradient 159147450a7bSBarry Smith norm at each iteration. 159247450a7bSBarry Smith 159310978b7dSBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefaultShort()`, `TaoMonitorSet()` 1594a7e14dcfSSatish Balay @*/ 1595d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDefault(Tao tao, void *ctx) 1596d71ae5a4SJacob Faibussowitsch { 159763b15415SAlp Dener PetscInt its, tabs; 1598a7e14dcfSSatish Balay PetscReal fct, gnorm; 15998163d661SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 160047a47007SBarry Smith 1601a7e14dcfSSatish Balay PetscFunctionBegin; 160294511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 16038163d661SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 1604a7e14dcfSSatish Balay its = tao->niter; 1605a7e14dcfSSatish Balay fct = tao->fc; 1606a7e14dcfSSatish Balay gnorm = tao->residual; 16079566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 16089566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 1609494bef23SAlp Dener if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) { 16109566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Iteration information for %s solve.\n", ((PetscObject)tao)->prefix)); 1611494bef23SAlp Dener tao->header_printed = PETSC_TRUE; 161263b15415SAlp Dener } 161363a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " TAO,", its)); 16149566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Function value: %g,", (double)fct)); 161508d19d1fSJason Sarich if (gnorm >= PETSC_INFINITY) { 16169566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: Inf \n")); 161708d19d1fSJason Sarich } else { 16189566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g \n", (double)gnorm)); 161908d19d1fSJason Sarich } 16209566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 16213ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1622a7e14dcfSSatish Balay } 1623a7e14dcfSSatish Balay 16247fab98ebSJason Sarich /*@ 162510978b7dSBarry Smith TaoMonitorGlobalization - Default routine for monitoring progress of `TaoSolve()` with extra detail on the globalization method. 16268d5ead36SAlp Dener 1627c3339decSBarry Smith Collective 16288d5ead36SAlp Dener 16298d5ead36SAlp Dener Input Parameters: 163047450a7bSBarry Smith + tao - the `Tao` context 163147450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL` 16328d5ead36SAlp Dener 163347450a7bSBarry Smith Options Database Key: 163410978b7dSBarry Smith . -tao_monitor_globalization - turn on monitoring with globalization information 16358d5ead36SAlp Dener 16368d5ead36SAlp Dener Level: advanced 16378d5ead36SAlp Dener 163847450a7bSBarry Smith Note: 163947450a7bSBarry Smith This monitor prints the function value and gradient norm at each 164047450a7bSBarry Smith iteration, as well as the step size and trust radius. Note that the 164147450a7bSBarry Smith step size and trust radius may be the same for some algorithms. 164247450a7bSBarry Smith 164310978b7dSBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefaultShort()`, `TaoMonitorSet()` 16448d5ead36SAlp Dener @*/ 164510978b7dSBarry Smith PetscErrorCode TaoMonitorGlobalization(Tao tao, void *ctx) 1646d71ae5a4SJacob Faibussowitsch { 16478d5ead36SAlp Dener PetscInt its, tabs; 16488d5ead36SAlp Dener PetscReal fct, gnorm, stp, tr; 16498d5ead36SAlp Dener PetscViewer viewer = (PetscViewer)ctx; 16508d5ead36SAlp Dener 16518d5ead36SAlp Dener PetscFunctionBegin; 165294511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 16538d5ead36SAlp Dener PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 16548d5ead36SAlp Dener its = tao->niter; 16558d5ead36SAlp Dener fct = tao->fc; 16568d5ead36SAlp Dener gnorm = tao->residual; 16578d5ead36SAlp Dener stp = tao->step; 16588d5ead36SAlp Dener tr = tao->trust; 16599566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 16609566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 1661494bef23SAlp Dener if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) { 16629566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Iteration information for %s solve.\n", ((PetscObject)tao)->prefix)); 1663494bef23SAlp Dener tao->header_printed = PETSC_TRUE; 16648d5ead36SAlp Dener } 166563a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "%3" PetscInt_FMT " TAO,", its)); 16669566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Function value: %g,", (double)fct)); 16678d5ead36SAlp Dener if (gnorm >= PETSC_INFINITY) { 16689566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: Inf,")); 16698d5ead36SAlp Dener } else { 16709566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g,", (double)gnorm)); 16718d5ead36SAlp Dener } 16729566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Step: %g, Trust: %g\n", (double)stp, (double)tr)); 16739566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 16743ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 16758d5ead36SAlp Dener } 16768d5ead36SAlp Dener 16778d5ead36SAlp Dener /*@ 167810978b7dSBarry Smith TaoMonitorDefaultShort - Routine for monitoring progress of `TaoSolve()` that displays fewer digits than `TaoMonitorDefault()` 1679a7e14dcfSSatish Balay 1680c3339decSBarry Smith Collective 1681a7e14dcfSSatish Balay 1682a7e14dcfSSatish Balay Input Parameters: 168347450a7bSBarry Smith + tao - the `Tao` context 168447450a7bSBarry Smith - ctx - `PetscViewer` context of type `PETSCVIEWERASCII` 1685a7e14dcfSSatish Balay 168647450a7bSBarry Smith Options Database Key: 168710978b7dSBarry Smith . -tao_monitor_short - turn on default short monitoring 1688a7e14dcfSSatish Balay 1689a7e14dcfSSatish Balay Level: advanced 1690a7e14dcfSSatish Balay 169147450a7bSBarry Smith Note: 169247450a7bSBarry Smith Same as `TaoMonitorDefault()` except 169347450a7bSBarry Smith it prints fewer digits of the residual as the residual gets smaller. 169447450a7bSBarry Smith This is because the later digits are meaningless and are often 169547450a7bSBarry Smith different on different machines; by using this routine different 169647450a7bSBarry Smith machines will usually generate the same output. 169747450a7bSBarry Smith 169810978b7dSBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoMonitorSet()` 1699a7e14dcfSSatish Balay @*/ 170010978b7dSBarry Smith PetscErrorCode TaoMonitorDefaultShort(Tao tao, void *ctx) 1701d71ae5a4SJacob Faibussowitsch { 17022c9b8b83SAlp Dener PetscInt its, tabs; 1703a7e14dcfSSatish Balay PetscReal fct, gnorm; 17044d4332d5SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 170547a47007SBarry Smith 1706a7e14dcfSSatish Balay PetscFunctionBegin; 170794511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 17084d4332d5SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 1709a7e14dcfSSatish Balay its = tao->niter; 1710a7e14dcfSSatish Balay fct = tao->fc; 1711a7e14dcfSSatish Balay gnorm = tao->residual; 17129566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 17139566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 171463a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "iter = %3" PetscInt_FMT ",", its)); 17159566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Function value %g,", (double)fct)); 1716d393f493SBarry Smith if (gnorm >= PETSC_INFINITY) { 17179566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: Inf \n")); 171808d19d1fSJason Sarich } else if (gnorm > 1.e-6) { 17199566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g \n", (double)gnorm)); 1720a7e14dcfSSatish Balay } else if (gnorm > 1.e-11) { 17219566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: < 1.0e-6 \n")); 1722a7e14dcfSSatish Balay } else { 17239566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: < 1.0e-11 \n")); 1724a7e14dcfSSatish Balay } 17259566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 17263ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1727a7e14dcfSSatish Balay } 1728a7e14dcfSSatish Balay 17297fab98ebSJason Sarich /*@ 173010978b7dSBarry Smith TaoMonitorConstraintNorm - same as `TaoMonitorDefault()` except 173147450a7bSBarry Smith it prints the norm of the constraint function. 1732a7e14dcfSSatish Balay 1733c3339decSBarry Smith Collective 1734a7e14dcfSSatish Balay 1735a7e14dcfSSatish Balay Input Parameters: 173647450a7bSBarry Smith + tao - the `Tao` context 173747450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL` 1738a7e14dcfSSatish Balay 173947450a7bSBarry Smith Options Database Key: 174010978b7dSBarry Smith . -tao_monitor_constraint_norm - monitor the constraints 1741a7e14dcfSSatish Balay 1742a7e14dcfSSatish Balay Level: advanced 1743a7e14dcfSSatish Balay 174410978b7dSBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefault()`, `TaoMonitorSet()` 1745a7e14dcfSSatish Balay @*/ 174610978b7dSBarry Smith PetscErrorCode TaoMonitorConstraintNorm(Tao tao, void *ctx) 1747d71ae5a4SJacob Faibussowitsch { 17482c9b8b83SAlp Dener PetscInt its, tabs; 1749a7e14dcfSSatish Balay PetscReal fct, gnorm; 1750d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 1751a7e14dcfSSatish Balay 1752a7e14dcfSSatish Balay PetscFunctionBegin; 175394511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1754d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 1755a7e14dcfSSatish Balay its = tao->niter; 1756a7e14dcfSSatish Balay fct = tao->fc; 1757a7e14dcfSSatish Balay gnorm = tao->residual; 17589566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 17599566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 176063a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "iter = %" PetscInt_FMT ",", its)); 17619566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Function value: %g,", (double)fct)); 17629566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Residual: %g ", (double)gnorm)); 17639566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, " Constraint: %g \n", (double)tao->cnorm)); 17649566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 17653ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1766a7e14dcfSSatish Balay } 1767a7e14dcfSSatish Balay 1768a7e14dcfSSatish Balay /*@C 176910978b7dSBarry Smith TaoMonitorSolution - Views the solution at each iteration of `TaoSolve()` 1770a7e14dcfSSatish Balay 1771c3339decSBarry Smith Collective 1772a7e14dcfSSatish Balay 1773a7e14dcfSSatish Balay Input Parameters: 177447450a7bSBarry Smith + tao - the `Tao` context 177547450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL` 1776a7e14dcfSSatish Balay 177747450a7bSBarry Smith Options Database Key: 177810978b7dSBarry Smith . -tao_monitor_solution - view the solution 1779a7e14dcfSSatish Balay 1780a7e14dcfSSatish Balay Level: advanced 1781a7e14dcfSSatish Balay 178210978b7dSBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefaultShort()`, `TaoMonitorSet()` 1783a7e14dcfSSatish Balay @*/ 178410978b7dSBarry Smith PetscErrorCode TaoMonitorSolution(Tao tao, void *ctx) 1785d71ae5a4SJacob Faibussowitsch { 1786feb237baSPierre Jolivet PetscViewer viewer = (PetscViewer)ctx; 178747a47007SBarry Smith 1788a7e14dcfSSatish Balay PetscFunctionBegin; 178994511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1790d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 17919566063dSJacob Faibussowitsch PetscCall(VecView(tao->solution, viewer)); 17923ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1793a7e14dcfSSatish Balay } 1794a7e14dcfSSatish Balay 1795a7e14dcfSSatish Balay /*@C 179610978b7dSBarry Smith TaoMonitorGradient - Views the gradient at each iteration of `TaoSolve()` 1797a7e14dcfSSatish Balay 1798c3339decSBarry Smith Collective 1799a7e14dcfSSatish Balay 1800a7e14dcfSSatish Balay Input Parameters: 180147450a7bSBarry Smith + tao - the `Tao` context 180247450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL` 1803a7e14dcfSSatish Balay 180447450a7bSBarry Smith Options Database Key: 180510978b7dSBarry Smith . -tao_monitor_gradient - view the gradient at each iteration 1806a7e14dcfSSatish Balay 1807a7e14dcfSSatish Balay Level: advanced 1808a7e14dcfSSatish Balay 180910978b7dSBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefaultShort()`, `TaoMonitorSet()` 1810a7e14dcfSSatish Balay @*/ 181110978b7dSBarry Smith PetscErrorCode TaoMonitorGradient(Tao tao, void *ctx) 1812d71ae5a4SJacob Faibussowitsch { 1813d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 181447a47007SBarry Smith 1815a7e14dcfSSatish Balay PetscFunctionBegin; 181694511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1817d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 18189566063dSJacob Faibussowitsch PetscCall(VecView(tao->gradient, viewer)); 18193ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1820a7e14dcfSSatish Balay } 1821a7e14dcfSSatish Balay 1822a7e14dcfSSatish Balay /*@C 182310978b7dSBarry Smith TaoMonitorStep - Views the step-direction at each iteration of `TaoSolve()` 1824a7e14dcfSSatish Balay 1825c3339decSBarry Smith Collective 1826a7e14dcfSSatish Balay 1827a7e14dcfSSatish Balay Input Parameters: 182847450a7bSBarry Smith + tao - the `Tao` context 182947450a7bSBarry Smith - ctx - `PetscViewer` context or `NULL` 1830a7e14dcfSSatish Balay 183147450a7bSBarry Smith Options Database Key: 183210978b7dSBarry Smith . -tao_monitor_step - view the step vector at each iteration 1833a7e14dcfSSatish Balay 1834a7e14dcfSSatish Balay Level: advanced 1835a7e14dcfSSatish Balay 183610978b7dSBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefaultShort()`, `TaoMonitorSet()` 1837a7e14dcfSSatish Balay @*/ 183810978b7dSBarry Smith PetscErrorCode TaoMonitorStep(Tao tao, void *ctx) 1839d71ae5a4SJacob Faibussowitsch { 1840d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 1841d393f493SBarry Smith 1842a7e14dcfSSatish Balay PetscFunctionBegin; 184394511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1844d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 18459566063dSJacob Faibussowitsch PetscCall(VecView(tao->stepdirection, viewer)); 18463ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1847a7e14dcfSSatish Balay } 1848a7e14dcfSSatish Balay 1849a7e14dcfSSatish Balay /*@C 185010978b7dSBarry Smith TaoMonitorSolutionDraw - Plots the solution at each iteration of `TaoSolve()` 1851a7e14dcfSSatish Balay 1852c3339decSBarry Smith Collective 1853a7e14dcfSSatish Balay 1854a7e14dcfSSatish Balay Input Parameters: 185547450a7bSBarry Smith + tao - the `Tao` context 185665ba42b6SBarry Smith - ctx - `TaoMonitorDraw` context 1857a7e14dcfSSatish Balay 185847450a7bSBarry Smith Options Database Key: 185910978b7dSBarry Smith . -tao_monitor_solution_draw - draw the solution at each iteration 1860a7e14dcfSSatish Balay 1861a7e14dcfSSatish Balay Level: advanced 1862a7e14dcfSSatish Balay 1863a81c3919SBarry Smith Note: 1864a81c3919SBarry Smith The context created by `TaoMonitorDrawCtxCreate()`, along with `TaoMonitorSolutionDraw()`, and `TaoMonitorDrawCtxDestroy()` 1865a81c3919SBarry Smith are passed to `TaoMonitorSet()` to monitor the solution graphically. 1866a81c3919SBarry Smith 1867a81c3919SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorSolution()`, `TaoMonitorSet()`, `TaoMonitorGradientDraw()`, `TaoMonitorDrawCtxCreate()`, 1868a81c3919SBarry Smith `TaoMonitorDrawCtxDestroy()` 1869a7e14dcfSSatish Balay @*/ 187010978b7dSBarry Smith PetscErrorCode TaoMonitorSolutionDraw(Tao tao, void *ctx) 1871d71ae5a4SJacob Faibussowitsch { 1872e882e171SHong Zhang TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx; 187347a47007SBarry Smith 1874a7e14dcfSSatish Balay PetscFunctionBegin; 187594511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 18763ba16761SJacob Faibussowitsch if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(PETSC_SUCCESS); 18779566063dSJacob Faibussowitsch PetscCall(VecView(tao->solution, ictx->viewer)); 18783ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1879a7e14dcfSSatish Balay } 1880a7e14dcfSSatish Balay 1881a7e14dcfSSatish Balay /*@C 188210978b7dSBarry Smith TaoMonitorGradientDraw - Plots the gradient at each iteration of `TaoSolve()` 1883a7e14dcfSSatish Balay 1884c3339decSBarry Smith Collective 1885a7e14dcfSSatish Balay 1886a7e14dcfSSatish Balay Input Parameters: 188747450a7bSBarry Smith + tao - the `Tao` context 188865ba42b6SBarry Smith - ctx - `PetscViewer` context 1889a7e14dcfSSatish Balay 189047450a7bSBarry Smith Options Database Key: 189110978b7dSBarry Smith . -tao_monitor_gradient_draw - draw the gradient at each iteration 1892a7e14dcfSSatish Balay 1893a7e14dcfSSatish Balay Level: advanced 1894a7e14dcfSSatish Balay 189510978b7dSBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorGradient()`, `TaoMonitorSet()`, `TaoMonitorSolutionDraw()` 1896a7e14dcfSSatish Balay @*/ 189710978b7dSBarry Smith PetscErrorCode TaoMonitorGradientDraw(Tao tao, void *ctx) 1898d71ae5a4SJacob Faibussowitsch { 1899e882e171SHong Zhang TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx; 190047a47007SBarry Smith 1901a7e14dcfSSatish Balay PetscFunctionBegin; 190294511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 19033ba16761SJacob Faibussowitsch if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(PETSC_SUCCESS); 19049566063dSJacob Faibussowitsch PetscCall(VecView(tao->gradient, ictx->viewer)); 19053ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1906a7e14dcfSSatish Balay } 1907a7e14dcfSSatish Balay 1908a7e14dcfSSatish Balay /*@C 190910978b7dSBarry Smith TaoMonitorStepDraw - Plots the step direction at each iteration of `TaoSolve()` 1910a7e14dcfSSatish Balay 1911c3339decSBarry Smith Collective 1912a7e14dcfSSatish Balay 1913a7e14dcfSSatish Balay Input Parameters: 191447450a7bSBarry Smith + tao - the `Tao` context 191547450a7bSBarry Smith - ctx - the `PetscViewer` context 1916a7e14dcfSSatish Balay 191747450a7bSBarry Smith Options Database Key: 191810978b7dSBarry Smith . -tao_monitor_step_draw - draw the step direction at each iteration 1919a7e14dcfSSatish Balay 1920a7e14dcfSSatish Balay Level: advanced 1921a7e14dcfSSatish Balay 192210978b7dSBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorSet()`, `TaoMonitorSolutionDraw` 1923a7e14dcfSSatish Balay @*/ 192410978b7dSBarry Smith PetscErrorCode TaoMonitorStepDraw(Tao tao, void *ctx) 1925d71ae5a4SJacob Faibussowitsch { 192694511df7SStefano Zampini PetscViewer viewer = (PetscViewer)ctx; 192747a47007SBarry Smith 1928a7e14dcfSSatish Balay PetscFunctionBegin; 192994511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 193094511df7SStefano Zampini PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 19319566063dSJacob Faibussowitsch PetscCall(VecView(tao->stepdirection, viewer)); 19323ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1933a7e14dcfSSatish Balay } 1934a7e14dcfSSatish Balay 1935a7e14dcfSSatish Balay /*@C 193610978b7dSBarry Smith TaoMonitorResidual - Views the least-squares residual at each iteration of `TaoSolve()` 1937a7e14dcfSSatish Balay 1938c3339decSBarry Smith Collective 1939a7e14dcfSSatish Balay 1940a7e14dcfSSatish Balay Input Parameters: 194147450a7bSBarry Smith + tao - the `Tao` context 194247450a7bSBarry Smith - ctx - the `PetscViewer` context or `NULL` 1943a7e14dcfSSatish Balay 194447450a7bSBarry Smith Options Database Key: 194510978b7dSBarry Smith . -tao_monitor_ls_residual - view the residual at each iteration 1946a7e14dcfSSatish Balay 1947a7e14dcfSSatish Balay Level: advanced 1948a7e14dcfSSatish Balay 194910978b7dSBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorDefaultShort()`, `TaoMonitorSet()` 1950a7e14dcfSSatish Balay @*/ 195110978b7dSBarry Smith PetscErrorCode TaoMonitorResidual(Tao tao, void *ctx) 1952d71ae5a4SJacob Faibussowitsch { 1953d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 195447a47007SBarry Smith 1955a7e14dcfSSatish Balay PetscFunctionBegin; 195694511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 1957d393f493SBarry Smith PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2); 19589566063dSJacob Faibussowitsch PetscCall(VecView(tao->ls_res, viewer)); 19593ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 1960a7e14dcfSSatish Balay } 1961a7e14dcfSSatish Balay 19627fab98ebSJason Sarich /*@ 1963a7e14dcfSSatish Balay TaoDefaultConvergenceTest - Determines whether the solver should continue iterating 1964a7e14dcfSSatish Balay or terminate. 1965a7e14dcfSSatish Balay 1966c3339decSBarry Smith Collective 1967a7e14dcfSSatish Balay 1968a7e14dcfSSatish Balay Input Parameters: 196947450a7bSBarry Smith + tao - the `Tao` context 1970a7e14dcfSSatish Balay - dummy - unused dummy context 1971a7e14dcfSSatish Balay 197247450a7bSBarry Smith Level: developer 197347450a7bSBarry Smith 1974a7e14dcfSSatish Balay Notes: 1975a7e14dcfSSatish Balay This routine checks the residual in the optimality conditions, the 1976a7e14dcfSSatish Balay relative residual in the optimity conditions, the number of function 1977a7e14dcfSSatish Balay evaluations, and the function value to test convergence. Some 1978a7e14dcfSSatish Balay solvers may use different convergence routines. 1979a7e14dcfSSatish Balay 19801cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetTolerances()`, `TaoGetConvergedReason()`, `TaoSetConvergedReason()` 1981a7e14dcfSSatish Balay @*/ 1982d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoDefaultConvergenceTest(Tao tao, void *dummy) 1983d71ae5a4SJacob Faibussowitsch { 1984a7e14dcfSSatish Balay PetscInt niter = tao->niter, nfuncs = PetscMax(tao->nfuncs, tao->nfuncgrads); 1985a7e14dcfSSatish Balay PetscInt max_funcs = tao->max_funcs; 1986a7e14dcfSSatish Balay PetscReal gnorm = tao->residual, gnorm0 = tao->gnorm0; 1987a7e14dcfSSatish Balay PetscReal f = tao->fc, steptol = tao->steptol, trradius = tao->step; 1988a7e14dcfSSatish Balay PetscReal gatol = tao->gatol, grtol = tao->grtol, gttol = tao->gttol; 1989e52336cbSBarry Smith PetscReal catol = tao->catol, crtol = tao->crtol; 1990e52336cbSBarry Smith PetscReal fmin = tao->fmin, cnorm = tao->cnorm; 1991e4cb33bbSBarry Smith TaoConvergedReason reason = tao->reason; 1992a7e14dcfSSatish Balay 1993a7e14dcfSSatish Balay PetscFunctionBegin; 1994441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 19953ba16761SJacob Faibussowitsch if (reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(PETSC_SUCCESS); 1996a7e14dcfSSatish Balay 1997a7e14dcfSSatish Balay if (PetscIsInfOrNanReal(f)) { 19989566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Failed to converged, function value is Inf or NaN\n")); 1999a7e14dcfSSatish Balay reason = TAO_DIVERGED_NAN; 2000a7e14dcfSSatish Balay } else if (f <= fmin && cnorm <= catol) { 20019566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Converged due to function value %g < minimum function value %g\n", (double)f, (double)fmin)); 2002a7e14dcfSSatish Balay reason = TAO_CONVERGED_MINF; 2003a7e14dcfSSatish Balay } else if (gnorm <= gatol && cnorm <= catol) { 20049566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Converged due to residual norm ||g(X)||=%g < %g\n", (double)gnorm, (double)gatol)); 2005a7e14dcfSSatish Balay reason = TAO_CONVERGED_GATOL; 2006a7e14dcfSSatish Balay } else if (f != 0 && PetscAbsReal(gnorm / f) <= grtol && cnorm <= crtol) { 20079566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Converged due to residual ||g(X)||/|f(X)| =%g < %g\n", (double)(gnorm / f), (double)grtol)); 2008a7e14dcfSSatish Balay reason = TAO_CONVERGED_GRTOL; 2009e1d16bb9SBarry Smith } else if (gnorm0 != 0 && ((gttol == 0 && gnorm == 0) || gnorm / gnorm0 < gttol) && cnorm <= crtol) { 20109566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Converged due to relative residual norm ||g(X)||/||g(X0)|| = %g < %g\n", (double)(gnorm / gnorm0), (double)gttol)); 2011a7e14dcfSSatish Balay reason = TAO_CONVERGED_GTTOL; 2012606f75f6SBarry Smith } else if (max_funcs != PETSC_UNLIMITED && nfuncs > max_funcs) { 20139566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Exceeded maximum number of function evaluations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", nfuncs, max_funcs)); 2014a7e14dcfSSatish Balay reason = TAO_DIVERGED_MAXFCN; 2015a7e14dcfSSatish Balay } else if (tao->lsflag != 0) { 20169566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Tao Line Search failure.\n")); 2017a7e14dcfSSatish Balay reason = TAO_DIVERGED_LS_FAILURE; 2018a7e14dcfSSatish Balay } else if (trradius < steptol && niter > 0) { 20199566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Trust region/step size too small: %g < %g\n", (double)trradius, (double)steptol)); 2020a7e14dcfSSatish Balay reason = TAO_CONVERGED_STEPTOL; 2021e031d6f5SAlp Dener } else if (niter >= tao->max_it) { 202263a3b9bcSJacob Faibussowitsch PetscCall(PetscInfo(tao, "Exceeded maximum number of iterations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", niter, tao->max_it)); 2023a7e14dcfSSatish Balay reason = TAO_DIVERGED_MAXITS; 2024a7e14dcfSSatish Balay } else { 2025a7e14dcfSSatish Balay reason = TAO_CONTINUE_ITERATING; 2026a7e14dcfSSatish Balay } 2027a7e14dcfSSatish Balay tao->reason = reason; 20283ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2029a7e14dcfSSatish Balay } 2030a7e14dcfSSatish Balay 2031cc4c1da9SBarry Smith /*@ 2032a7e14dcfSSatish Balay TaoSetOptionsPrefix - Sets the prefix used for searching for all 203365ba42b6SBarry Smith Tao options in the database. 2034a7e14dcfSSatish Balay 2035c3339decSBarry Smith Logically Collective 2036a7e14dcfSSatish Balay 2037a7e14dcfSSatish Balay Input Parameters: 203847450a7bSBarry Smith + tao - the `Tao` context 2039e056e8ceSJacob Faibussowitsch - p - the prefix string to prepend to all Tao option requests 2040a7e14dcfSSatish Balay 20412fe279fdSBarry Smith Level: advanced 20422fe279fdSBarry Smith 2043a7e14dcfSSatish Balay Notes: 2044a7e14dcfSSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 2045a7e14dcfSSatish Balay The first character of all runtime options is AUTOMATICALLY the hyphen. 2046a7e14dcfSSatish Balay 2047a7e14dcfSSatish Balay For example, to distinguish between the runtime options for two 204865ba42b6SBarry Smith different Tao solvers, one could call 2049a7e14dcfSSatish Balay .vb 2050a7e14dcfSSatish Balay TaoSetOptionsPrefix(tao1,"sys1_") 2051a7e14dcfSSatish Balay TaoSetOptionsPrefix(tao2,"sys2_") 2052a7e14dcfSSatish Balay .ve 2053a7e14dcfSSatish Balay 2054a7e14dcfSSatish Balay This would enable use of different options for each system, such as 2055a7e14dcfSSatish Balay .vb 20569fa2b5dcSStefano Zampini -sys1_tao_method blmvm -sys1_tao_grtol 1.e-3 20579fa2b5dcSStefano Zampini -sys2_tao_method lmvm -sys2_tao_grtol 1.e-4 2058a7e14dcfSSatish Balay .ve 2059a7e14dcfSSatish Balay 20601cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoAppendOptionsPrefix()`, `TaoGetOptionsPrefix()` 2061a7e14dcfSSatish Balay @*/ 2062d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetOptionsPrefix(Tao tao, const char p[]) 2063d71ae5a4SJacob Faibussowitsch { 2064a7e14dcfSSatish Balay PetscFunctionBegin; 206594511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 20669566063dSJacob Faibussowitsch PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tao, p)); 20671baa6e33SBarry Smith if (tao->linesearch) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch, p)); 20681baa6e33SBarry Smith if (tao->ksp) PetscCall(KSPSetOptionsPrefix(tao->ksp, p)); 20693ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2070a7e14dcfSSatish Balay } 2071a7e14dcfSSatish Balay 2072cc4c1da9SBarry Smith /*@ 207347450a7bSBarry Smith TaoAppendOptionsPrefix - Appends to the prefix used for searching for all Tao options in the database. 2074a7e14dcfSSatish Balay 2075c3339decSBarry Smith Logically Collective 2076a7e14dcfSSatish Balay 2077a7e14dcfSSatish Balay Input Parameters: 207847450a7bSBarry Smith + tao - the `Tao` solver context 2079e056e8ceSJacob Faibussowitsch - p - the prefix string to prepend to all `Tao` option requests 2080a7e14dcfSSatish Balay 20812fe279fdSBarry Smith Level: advanced 20822fe279fdSBarry Smith 208365ba42b6SBarry Smith Note: 2084a7e14dcfSSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 208565ba42b6SBarry Smith The first character of all runtime options is automatically the hyphen. 2086a7e14dcfSSatish Balay 20871cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoSetOptionsPrefix()`, `TaoGetOptionsPrefix()` 2088a7e14dcfSSatish Balay @*/ 2089d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoAppendOptionsPrefix(Tao tao, const char p[]) 2090d71ae5a4SJacob Faibussowitsch { 2091a7e14dcfSSatish Balay PetscFunctionBegin; 209294511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 20939566063dSJacob Faibussowitsch PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao, p)); 20941baa6e33SBarry Smith if (tao->linesearch) PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao->linesearch, p)); 20951baa6e33SBarry Smith if (tao->ksp) PetscCall(KSPAppendOptionsPrefix(tao->ksp, p)); 20963ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2097a7e14dcfSSatish Balay } 2098a7e14dcfSSatish Balay 2099cc4c1da9SBarry Smith /*@ 2100a7e14dcfSSatish Balay TaoGetOptionsPrefix - Gets the prefix used for searching for all 210165ba42b6SBarry Smith Tao options in the database 2102a7e14dcfSSatish Balay 2103a7e14dcfSSatish Balay Not Collective 2104a7e14dcfSSatish Balay 210547450a7bSBarry Smith Input Parameter: 210647450a7bSBarry Smith . tao - the `Tao` context 2107a7e14dcfSSatish Balay 210847450a7bSBarry Smith Output Parameter: 2109e056e8ceSJacob Faibussowitsch . p - pointer to the prefix string used is returned 2110a7e14dcfSSatish Balay 2111e056e8ceSJacob Faibussowitsch Fortran Notes: 211247450a7bSBarry Smith Pass in a string 'prefix' of sufficient length to hold the prefix. 2113a7e14dcfSSatish Balay 2114a7e14dcfSSatish Balay Level: advanced 2115a7e14dcfSSatish Balay 21161cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetFromOptions()`, `TaoSetOptionsPrefix()`, `TaoAppendOptionsPrefix()` 2117a7e14dcfSSatish Balay @*/ 2118d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetOptionsPrefix(Tao tao, const char *p[]) 2119d71ae5a4SJacob Faibussowitsch { 212094511df7SStefano Zampini PetscFunctionBegin; 212194511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 21229566063dSJacob Faibussowitsch PetscCall(PetscObjectGetOptionsPrefix((PetscObject)tao, p)); 21233ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2124a7e14dcfSSatish Balay } 2125a7e14dcfSSatish Balay 2126cc4c1da9SBarry Smith /*@ 212747450a7bSBarry Smith TaoSetType - Sets the `TaoType` for the minimization solver. 2128a7e14dcfSSatish Balay 2129c3339decSBarry Smith Collective 2130a7e14dcfSSatish Balay 2131a7e14dcfSSatish Balay Input Parameters: 2132e056e8ceSJacob Faibussowitsch + tao - the `Tao` solver context 2133a7e14dcfSSatish Balay - type - a known method 2134a7e14dcfSSatish Balay 2135a7e14dcfSSatish Balay Options Database Key: 213658417fe7SBarry Smith . -tao_type <type> - Sets the method; use -help for a list 213758417fe7SBarry Smith of available methods (for instance, "-tao_type lmvm" or "-tao_type tron") 2138a7e14dcfSSatish Balay 2139a7e14dcfSSatish Balay Level: intermediate 2140a7e14dcfSSatish Balay 21411cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoCreate()`, `TaoGetType()`, `TaoType` 2142a7e14dcfSSatish Balay @*/ 2143d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetType(Tao tao, TaoType type) 2144d71ae5a4SJacob Faibussowitsch { 2145441846f8SBarry Smith PetscErrorCode (*create_xxx)(Tao); 2146a7e14dcfSSatish Balay PetscBool issame; 2147a7e14dcfSSatish Balay 2148a7e14dcfSSatish Balay PetscFunctionBegin; 2149441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2150a7e14dcfSSatish Balay 21519566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)tao, type, &issame)); 21523ba16761SJacob Faibussowitsch if (issame) PetscFunctionReturn(PETSC_SUCCESS); 2153a7e14dcfSSatish Balay 21549566063dSJacob Faibussowitsch PetscCall(PetscFunctionListFind(TaoList, type, (void (**)(void))&create_xxx)); 21553c859ba3SBarry Smith PetscCheck(create_xxx, PetscObjectComm((PetscObject)tao), PETSC_ERR_ARG_UNKNOWN_TYPE, "Unable to find requested Tao type %s", type); 2156a7e14dcfSSatish Balay 2157a7e14dcfSSatish Balay /* Destroy the existing solver information */ 2158dbbe0bcdSBarry Smith PetscTryTypeMethod(tao, destroy); 21591baa6e33SBarry Smith PetscCall(KSPDestroy(&tao->ksp)); 21609566063dSJacob Faibussowitsch PetscCall(TaoLineSearchDestroy(&tao->linesearch)); 216183c8fe1dSLisandro Dalcin tao->ops->setup = NULL; 216283c8fe1dSLisandro Dalcin tao->ops->solve = NULL; 216383c8fe1dSLisandro Dalcin tao->ops->view = NULL; 216483c8fe1dSLisandro Dalcin tao->ops->setfromoptions = NULL; 216583c8fe1dSLisandro Dalcin tao->ops->destroy = NULL; 2166a7e14dcfSSatish Balay 2167a7e14dcfSSatish Balay tao->setupcalled = PETSC_FALSE; 2168a7e14dcfSSatish Balay 21699566063dSJacob Faibussowitsch PetscCall((*create_xxx)(tao)); 21709566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject)tao, type)); 21713ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2172a7e14dcfSSatish Balay } 217347a47007SBarry Smith 217467be906fSBarry Smith /*@C 217547450a7bSBarry Smith TaoRegister - Adds a method to the Tao package for minimization. 2176a7e14dcfSSatish Balay 2177cc4c1da9SBarry Smith Not Collective, No Fortran Support 2178a7e14dcfSSatish Balay 2179a7e14dcfSSatish Balay Input Parameters: 2180a7e14dcfSSatish Balay + sname - name of a new user-defined solver 2181a7e14dcfSSatish Balay - func - routine to Create method context 2182a7e14dcfSSatish Balay 2183e056e8ceSJacob Faibussowitsch Example Usage: 2184a7e14dcfSSatish Balay .vb 2185441846f8SBarry Smith TaoRegister("my_solver", MySolverCreate); 2186a7e14dcfSSatish Balay .ve 2187a7e14dcfSSatish Balay 2188a7e14dcfSSatish Balay Then, your solver can be chosen with the procedural interface via 2189a7e14dcfSSatish Balay $ TaoSetType(tao, "my_solver") 2190a7e14dcfSSatish Balay or at runtime via the option 219158417fe7SBarry Smith $ -tao_type my_solver 2192a7e14dcfSSatish Balay 2193a7e14dcfSSatish Balay Level: advanced 2194a7e14dcfSSatish Balay 219567be906fSBarry Smith Note: 219667be906fSBarry Smith `TaoRegister()` may be called multiple times to add several user-defined solvers. 219767be906fSBarry Smith 21981cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetType()`, `TaoRegisterAll()`, `TaoRegisterDestroy()` 219967be906fSBarry Smith @*/ 2200d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoRegister(const char sname[], PetscErrorCode (*func)(Tao)) 2201d71ae5a4SJacob Faibussowitsch { 2202a7e14dcfSSatish Balay PetscFunctionBegin; 22039566063dSJacob Faibussowitsch PetscCall(TaoInitializePackage()); 22049566063dSJacob Faibussowitsch PetscCall(PetscFunctionListAdd(&TaoList, sname, (void (*)(void))func)); 22053ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2206a7e14dcfSSatish Balay } 2207a7e14dcfSSatish Balay 2208a7e14dcfSSatish Balay /*@C 2209441846f8SBarry Smith TaoRegisterDestroy - Frees the list of minimization solvers that were 221047450a7bSBarry Smith registered by `TaoRegister()`. 2211a7e14dcfSSatish Balay 2212a7e14dcfSSatish Balay Not Collective 2213a7e14dcfSSatish Balay 2214a7e14dcfSSatish Balay Level: advanced 2215a7e14dcfSSatish Balay 22161cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoRegisterAll()`, `TaoRegister()` 2217a7e14dcfSSatish Balay @*/ 2218d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoRegisterDestroy(void) 2219d71ae5a4SJacob Faibussowitsch { 2220a7e14dcfSSatish Balay PetscFunctionBegin; 22219566063dSJacob Faibussowitsch PetscCall(PetscFunctionListDestroy(&TaoList)); 2222441846f8SBarry Smith TaoRegisterAllCalled = PETSC_FALSE; 22233ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2224a7e14dcfSSatish Balay } 222547a47007SBarry Smith 22268931d482SJason Sarich /*@ 222747450a7bSBarry Smith TaoGetIterationNumber - Gets the number of `TaoSolve()` iterations completed 22288931d482SJason Sarich at this time. 22298931d482SJason Sarich 22308931d482SJason Sarich Not Collective 22318931d482SJason Sarich 22328931d482SJason Sarich Input Parameter: 223347450a7bSBarry Smith . tao - the `Tao` context 22348931d482SJason Sarich 22358931d482SJason Sarich Output Parameter: 22368931d482SJason Sarich . iter - iteration number 22378931d482SJason Sarich 22388931d482SJason Sarich Notes: 22398931d482SJason Sarich For example, during the computation of iteration 2 this would return 1. 22408931d482SJason Sarich 22418931d482SJason Sarich Level: intermediate 22428931d482SJason Sarich 22431cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`, `TaoGetResidualNorm()`, `TaoGetObjective()` 22448931d482SJason Sarich @*/ 2245d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetIterationNumber(Tao tao, PetscInt *iter) 2246d71ae5a4SJacob Faibussowitsch { 22478931d482SJason Sarich PetscFunctionBegin; 22488931d482SJason Sarich PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 22494f572ea9SToby Isaac PetscAssertPointer(iter, 2); 22508931d482SJason Sarich *iter = tao->niter; 22513ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 22528931d482SJason Sarich } 22538931d482SJason Sarich 22548931d482SJason Sarich /*@ 225547450a7bSBarry Smith TaoGetResidualNorm - Gets the current value of the norm of the residual (gradient) 225679f5d8caSBarry Smith at this time. 225779f5d8caSBarry Smith 225879f5d8caSBarry Smith Not Collective 225979f5d8caSBarry Smith 226079f5d8caSBarry Smith Input Parameter: 226147450a7bSBarry Smith . tao - the `Tao` context 226279f5d8caSBarry Smith 226379f5d8caSBarry Smith Output Parameter: 226479f5d8caSBarry Smith . value - the current value 226579f5d8caSBarry Smith 226679f5d8caSBarry Smith Level: intermediate 226779f5d8caSBarry Smith 2268e056e8ceSJacob Faibussowitsch Developer Notes: 226967be906fSBarry Smith This is the 2-norm of the residual, we cannot use `TaoGetGradientNorm()` because that has 227047450a7bSBarry Smith a different meaning. For some reason `Tao` sometimes calls the gradient the residual. 227179f5d8caSBarry Smith 22721cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()`, `TaoGetIterationNumber()`, `TaoGetObjective()` 227379f5d8caSBarry Smith @*/ 2274d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetResidualNorm(Tao tao, PetscReal *value) 2275d71ae5a4SJacob Faibussowitsch { 227679f5d8caSBarry Smith PetscFunctionBegin; 227779f5d8caSBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 22784f572ea9SToby Isaac PetscAssertPointer(value, 2); 227979f5d8caSBarry Smith *value = tao->residual; 22803ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 228179f5d8caSBarry Smith } 228279f5d8caSBarry Smith 228379f5d8caSBarry Smith /*@ 22848931d482SJason Sarich TaoSetIterationNumber - Sets the current iteration number. 22858931d482SJason Sarich 2286c3339decSBarry Smith Logically Collective 22878931d482SJason Sarich 2288d8d19677SJose E. Roman Input Parameters: 228947450a7bSBarry Smith + tao - the `Tao` context 2290a2b725a8SWilliam Gropp - iter - iteration number 22918931d482SJason Sarich 22928931d482SJason Sarich Level: developer 22938931d482SJason Sarich 22941cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()` 22958931d482SJason Sarich @*/ 2296d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetIterationNumber(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->niter = iter; 23039566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao)); 23043ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 23058931d482SJason Sarich } 23068931d482SJason Sarich 23078931d482SJason Sarich /*@ 230847450a7bSBarry Smith TaoGetTotalIterationNumber - Gets the total number of `TaoSolve()` iterations 23098931d482SJason Sarich completed. This number keeps accumulating if multiple solves 231047450a7bSBarry Smith are called with the `Tao` object. 23118931d482SJason Sarich 23128931d482SJason Sarich Not Collective 23138931d482SJason Sarich 23148931d482SJason Sarich Input Parameter: 231547450a7bSBarry Smith . tao - the `Tao` context 23168931d482SJason Sarich 23178931d482SJason Sarich Output Parameter: 231847450a7bSBarry Smith . iter - number of iterations 23198931d482SJason Sarich 23208931d482SJason Sarich Level: intermediate 23218931d482SJason Sarich 232220f4b53cSBarry Smith Note: 232367be906fSBarry Smith The total iteration count is updated after each solve, if there is a current 232447450a7bSBarry Smith `TaoSolve()` in progress then those iterations are not included in the count 232567be906fSBarry Smith 23261cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()` 23278931d482SJason Sarich @*/ 2328d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetTotalIterationNumber(Tao tao, PetscInt *iter) 2329d71ae5a4SJacob Faibussowitsch { 23308931d482SJason Sarich PetscFunctionBegin; 23318931d482SJason Sarich PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 23324f572ea9SToby Isaac PetscAssertPointer(iter, 2); 23338931d482SJason Sarich *iter = tao->ntotalits; 23343ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 23358931d482SJason Sarich } 23368931d482SJason Sarich 23378931d482SJason Sarich /*@ 23388931d482SJason Sarich TaoSetTotalIterationNumber - Sets the current total iteration number. 23398931d482SJason Sarich 2340c3339decSBarry Smith Logically Collective 23418931d482SJason Sarich 2342d8d19677SJose E. Roman Input Parameters: 234347450a7bSBarry Smith + tao - the `Tao` context 234447450a7bSBarry Smith - iter - the iteration number 23458931d482SJason Sarich 23468931d482SJason Sarich Level: developer 23478931d482SJason Sarich 23481cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetLinearSolveIterations()` 23498931d482SJason Sarich @*/ 2350d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetTotalIterationNumber(Tao tao, PetscInt iter) 2351d71ae5a4SJacob Faibussowitsch { 23528931d482SJason Sarich PetscFunctionBegin; 23538931d482SJason Sarich PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 235494511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao, iter, 2); 23559566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao)); 23568931d482SJason Sarich tao->ntotalits = iter; 23579566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao)); 23583ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 23598931d482SJason Sarich } 23608931d482SJason Sarich 2361a7e14dcfSSatish Balay /*@ 236247450a7bSBarry Smith TaoSetConvergedReason - Sets the termination flag on a `Tao` object 2363a7e14dcfSSatish Balay 2364c3339decSBarry Smith Logically Collective 2365a7e14dcfSSatish Balay 2366a7e14dcfSSatish Balay Input Parameters: 236747450a7bSBarry Smith + tao - the `Tao` context 236847450a7bSBarry Smith - reason - the `TaoConvergedReason` 2369a7e14dcfSSatish Balay 2370a7e14dcfSSatish Balay Level: intermediate 2371a7e14dcfSSatish Balay 23721cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason` 2373a7e14dcfSSatish Balay @*/ 2374d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergedReason(Tao tao, TaoConvergedReason reason) 2375d71ae5a4SJacob Faibussowitsch { 2376a7e14dcfSSatish Balay PetscFunctionBegin; 237794511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 237894511df7SStefano Zampini PetscValidLogicalCollectiveEnum(tao, reason, 2); 2379a7e14dcfSSatish Balay tao->reason = reason; 23803ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2381a7e14dcfSSatish Balay } 2382a7e14dcfSSatish Balay 2383a7e14dcfSSatish Balay /*@ 238447450a7bSBarry Smith TaoGetConvergedReason - Gets the reason the `TaoSolve()` was stopped. 2385a7e14dcfSSatish Balay 2386a7e14dcfSSatish Balay Not Collective 2387a7e14dcfSSatish Balay 2388a7e14dcfSSatish Balay Input Parameter: 238947450a7bSBarry Smith . tao - the `Tao` solver context 2390a7e14dcfSSatish Balay 2391a7e14dcfSSatish Balay Output Parameter: 239247450a7bSBarry Smith . reason - value of `TaoConvergedReason` 2393a7e14dcfSSatish Balay 2394a7e14dcfSSatish Balay Level: intermediate 2395a7e14dcfSSatish Balay 23961cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoConvergedReason`, `TaoSetConvergenceTest()`, `TaoSetTolerances()` 2397a7e14dcfSSatish Balay @*/ 2398d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConvergedReason(Tao tao, TaoConvergedReason *reason) 2399d71ae5a4SJacob Faibussowitsch { 2400a7e14dcfSSatish Balay PetscFunctionBegin; 2401441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 24024f572ea9SToby Isaac PetscAssertPointer(reason, 2); 2403a7e14dcfSSatish Balay *reason = tao->reason; 24043ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2405a7e14dcfSSatish Balay } 2406a7e14dcfSSatish Balay 2407a7e14dcfSSatish Balay /*@ 2408a7e14dcfSSatish Balay TaoGetSolutionStatus - Get the current iterate, objective value, 240947450a7bSBarry Smith residual, infeasibility, and termination from a `Tao` object 2410a7e14dcfSSatish Balay 2411a7e14dcfSSatish Balay Not Collective 2412a7e14dcfSSatish Balay 2413f899ff85SJose E. Roman Input Parameter: 241447450a7bSBarry Smith . tao - the `Tao` context 2415a7e14dcfSSatish Balay 2416a7e14dcfSSatish Balay Output Parameters: 2417e056e8ceSJacob Faibussowitsch + its - the current iterate number (>=0) 2418a7e14dcfSSatish Balay . f - the current function value 241958417fe7SBarry Smith . gnorm - the square of the gradient norm, duality gap, or other measure indicating distance from optimality. 2420a7e14dcfSSatish Balay . cnorm - the infeasibility of the current solution with regard to the constraints. 2421a7e14dcfSSatish Balay . xdiff - the step length or trust region radius of the most recent iterate. 242265ba42b6SBarry Smith - reason - The termination reason, which can equal `TAO_CONTINUE_ITERATING` 2423a7e14dcfSSatish Balay 2424a7e14dcfSSatish Balay Level: intermediate 2425a7e14dcfSSatish Balay 242665ba42b6SBarry Smith Notes: 242765ba42b6SBarry Smith Tao returns the values set by the solvers in the routine `TaoMonitor()`. 2428a7e14dcfSSatish Balay 242967be906fSBarry Smith If any of the output arguments are set to `NULL`, no corresponding value will be returned. 2430a7e14dcfSSatish Balay 24311cc06b55SBarry Smith .seealso: [](ch_tao), `TaoMonitor()`, `TaoGetConvergedReason()` 2432a7e14dcfSSatish Balay @*/ 2433d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetSolutionStatus(Tao tao, PetscInt *its, PetscReal *f, PetscReal *gnorm, PetscReal *cnorm, PetscReal *xdiff, TaoConvergedReason *reason) 2434d71ae5a4SJacob Faibussowitsch { 2435a7e14dcfSSatish Balay PetscFunctionBegin; 243694511df7SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2437a7e14dcfSSatish Balay if (its) *its = tao->niter; 2438a7e14dcfSSatish Balay if (f) *f = tao->fc; 2439a7e14dcfSSatish Balay if (gnorm) *gnorm = tao->residual; 2440a7e14dcfSSatish Balay if (cnorm) *cnorm = tao->cnorm; 2441a7e14dcfSSatish Balay if (reason) *reason = tao->reason; 2442a7e14dcfSSatish Balay if (xdiff) *xdiff = tao->step; 24433ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2444a7e14dcfSSatish Balay } 2445a7e14dcfSSatish Balay 2446cc4c1da9SBarry Smith /*@ 244747450a7bSBarry Smith TaoGetType - Gets the current `TaoType` being used in the `Tao` object 2448a7e14dcfSSatish Balay 2449a7e14dcfSSatish Balay Not Collective 2450a7e14dcfSSatish Balay 2451a7e14dcfSSatish Balay Input Parameter: 245247450a7bSBarry Smith . tao - the `Tao` solver context 2453a7e14dcfSSatish Balay 2454a7e14dcfSSatish Balay Output Parameter: 245547450a7bSBarry Smith . type - the `TaoType` 2456a7e14dcfSSatish Balay 2457a7e14dcfSSatish Balay Level: intermediate 2458a7e14dcfSSatish Balay 24591cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoType`, `TaoSetType()` 2460a7e14dcfSSatish Balay @*/ 2461d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetType(Tao tao, TaoType *type) 2462d71ae5a4SJacob Faibussowitsch { 2463a7e14dcfSSatish Balay PetscFunctionBegin; 2464441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 24654f572ea9SToby Isaac PetscAssertPointer(type, 2); 2466a7e14dcfSSatish Balay *type = ((PetscObject)tao)->type_name; 24673ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2468a7e14dcfSSatish Balay } 2469a7e14dcfSSatish Balay 2470a7e14dcfSSatish Balay /*@C 2471a7e14dcfSSatish Balay TaoMonitor - Monitor the solver and the current solution. This 2472a7e14dcfSSatish Balay routine will record the iteration number and residual statistics, 2473670c031eSStefano Zampini and call any monitors specified by the user. 2474a7e14dcfSSatish Balay 2475a7e14dcfSSatish Balay Input Parameters: 247647450a7bSBarry Smith + tao - the `Tao` context 2477a7e14dcfSSatish Balay . its - the current iterate number (>=0) 2478a7e14dcfSSatish Balay . f - the current objective function value 24791cb3f120SPierre Jolivet . res - the gradient norm, square root of the duality gap, or other measure indicating distance from optimality. This measure will be recorded and 2480a7e14dcfSSatish Balay used for some termination tests. 2481a7e14dcfSSatish Balay . cnorm - the infeasibility of the current solution with regard to the constraints. 2482a7e14dcfSSatish Balay - steplength - multiple of the step direction added to the previous iterate. 2483a7e14dcfSSatish Balay 2484a7e14dcfSSatish Balay Options Database Key: 2485a7e14dcfSSatish Balay . -tao_monitor - Use the default monitor, which prints statistics to standard output 2486a7e14dcfSSatish Balay 2487a7e14dcfSSatish Balay Level: developer 2488a7e14dcfSSatish Balay 248910978b7dSBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetConvergedReason()`, `TaoMonitorDefault()`, `TaoMonitorSet()` 2490a7e14dcfSSatish Balay @*/ 2491d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitor(Tao tao, PetscInt its, PetscReal f, PetscReal res, PetscReal cnorm, PetscReal steplength) 2492d71ae5a4SJacob Faibussowitsch { 249347a47007SBarry Smith PetscInt i; 249447a47007SBarry Smith 2495a7e14dcfSSatish Balay PetscFunctionBegin; 2496441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2497a7e14dcfSSatish Balay tao->fc = f; 2498a7e14dcfSSatish Balay tao->residual = res; 2499a7e14dcfSSatish Balay tao->cnorm = cnorm; 2500a7e14dcfSSatish Balay tao->step = steplength; 2501e52336cbSBarry Smith if (!its) { 250294511df7SStefano Zampini tao->cnorm0 = cnorm; 250394511df7SStefano Zampini tao->gnorm0 = res; 2504a7e14dcfSSatish Balay } 2505bfdcf5a2SStefano Zampini PetscCall(VecLockReadPush(tao->solution)); 250648a46eb9SPierre Jolivet for (i = 0; i < tao->numbermonitors; i++) PetscCall((*tao->monitor[i])(tao, tao->monitorcontext[i])); 2507bfdcf5a2SStefano Zampini PetscCall(VecLockReadPop(tao->solution)); 25083ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2509a7e14dcfSSatish Balay } 2510a7e14dcfSSatish Balay 2511a7e14dcfSSatish Balay /*@ 2512ae93cb3cSJason Sarich TaoSetConvergenceHistory - Sets the array used to hold the convergence history. 2513a7e14dcfSSatish Balay 2514c3339decSBarry Smith Logically Collective 2515a7e14dcfSSatish Balay 2516a7e14dcfSSatish Balay Input Parameters: 251747450a7bSBarry Smith + tao - the `Tao` solver context 2518a7e14dcfSSatish Balay . obj - array to hold objective value history 2519a7e14dcfSSatish Balay . resid - array to hold residual history 2520a7e14dcfSSatish Balay . cnorm - array to hold constraint violation history 2521be1558d8SJason Sarich . lits - integer array holds the number of linear iterations for each Tao iteration 252267be906fSBarry Smith . na - size of `obj`, `resid`, and `cnorm` 252365ba42b6SBarry Smith - reset - `PETSC_TRUE` indicates each new minimization resets the history counter to zero, 2524a7e14dcfSSatish Balay else it continues storing new values for new minimizations after the old ones 2525a7e14dcfSSatish Balay 252667be906fSBarry Smith Level: intermediate 252767be906fSBarry Smith 2528a7e14dcfSSatish Balay Notes: 252947450a7bSBarry Smith If set, `Tao` will fill the given arrays with the indicated 2530ae93cb3cSJason Sarich information at each iteration. If 'obj','resid','cnorm','lits' are 2531606f75f6SBarry Smith *all* `NULL` then space (using size `na`, or 1000 if `na` is `PETSC_DECIDE`) is allocated for the history. 253267be906fSBarry Smith If not all are `NULL`, then only the non-`NULL` information categories 2533ae93cb3cSJason Sarich will be stored, the others will be ignored. 2534ae93cb3cSJason Sarich 2535ae93cb3cSJason Sarich Any convergence information after iteration number 'na' will not be stored. 2536a7e14dcfSSatish Balay 2537a7e14dcfSSatish Balay This routine is useful, e.g., when running a code for purposes 2538a7e14dcfSSatish Balay of accurate performance monitoring, when no I/O should be done 2539a7e14dcfSSatish Balay during the section of code that is being timed. 2540a7e14dcfSSatish Balay 25411cc06b55SBarry Smith .seealso: [](ch_tao), `TaoGetConvergenceHistory()` 2542a7e14dcfSSatish Balay @*/ 2543d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetConvergenceHistory(Tao tao, PetscReal obj[], PetscReal resid[], PetscReal cnorm[], PetscInt lits[], PetscInt na, PetscBool reset) 2544d71ae5a4SJacob Faibussowitsch { 2545a7e14dcfSSatish Balay PetscFunctionBegin; 2546441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 25474f572ea9SToby Isaac if (obj) PetscAssertPointer(obj, 2); 25484f572ea9SToby Isaac if (resid) PetscAssertPointer(resid, 3); 25494f572ea9SToby Isaac if (cnorm) PetscAssertPointer(cnorm, 4); 25504f572ea9SToby Isaac if (lits) PetscAssertPointer(lits, 5); 2551ae93cb3cSJason Sarich 2552606f75f6SBarry Smith if (na == PETSC_DECIDE || na == PETSC_CURRENT) na = 1000; 2553ae93cb3cSJason Sarich if (!obj && !resid && !cnorm && !lits) { 25549566063dSJacob Faibussowitsch PetscCall(PetscCalloc4(na, &obj, na, &resid, na, &cnorm, na, &lits)); 2555ae93cb3cSJason Sarich tao->hist_malloc = PETSC_TRUE; 2556ae93cb3cSJason Sarich } 2557ae93cb3cSJason Sarich 2558a7e14dcfSSatish Balay tao->hist_obj = obj; 2559a7e14dcfSSatish Balay tao->hist_resid = resid; 2560a7e14dcfSSatish Balay tao->hist_cnorm = cnorm; 2561ae93cb3cSJason Sarich tao->hist_lits = lits; 2562a7e14dcfSSatish Balay tao->hist_max = na; 2563a7e14dcfSSatish Balay tao->hist_reset = reset; 2564ae93cb3cSJason Sarich tao->hist_len = 0; 25653ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2566a7e14dcfSSatish Balay } 2567a7e14dcfSSatish Balay 2568a7e14dcfSSatish Balay /*@C 256965ba42b6SBarry Smith TaoGetConvergenceHistory - Gets the arrays used that hold the convergence history. 2570a7e14dcfSSatish Balay 2571c3339decSBarry Smith Collective 2572a7e14dcfSSatish Balay 2573a7e14dcfSSatish Balay Input Parameter: 257447450a7bSBarry Smith . tao - the `Tao` context 2575a7e14dcfSSatish Balay 2576a7e14dcfSSatish Balay Output Parameters: 2577a7e14dcfSSatish Balay + obj - array used to hold objective value history 2578a7e14dcfSSatish Balay . resid - array used to hold residual history 2579a7e14dcfSSatish Balay . cnorm - array used to hold constraint violation history 2580ae93cb3cSJason Sarich . lits - integer array used to hold linear solver iteration count 258167be906fSBarry Smith - nhist - size of `obj`, `resid`, `cnorm`, and `lits` 258267be906fSBarry Smith 258367be906fSBarry Smith Level: advanced 2584a7e14dcfSSatish Balay 2585a7e14dcfSSatish Balay Notes: 258665ba42b6SBarry Smith This routine must be preceded by calls to `TaoSetConvergenceHistory()` 258765ba42b6SBarry Smith and `TaoSolve()`, otherwise it returns useless information. 2588ae93cb3cSJason Sarich 2589a7e14dcfSSatish Balay This routine is useful, e.g., when running a code for purposes 2590a7e14dcfSSatish Balay of accurate performance monitoring, when no I/O should be done 2591a7e14dcfSSatish Balay during the section of code that is being timed. 2592a7e14dcfSSatish Balay 2593e056e8ceSJacob Faibussowitsch Fortran Notes: 259467be906fSBarry Smith The calling sequence is 259567be906fSBarry Smith .vb 259667be906fSBarry Smith call TaoGetConvergenceHistory(Tao tao, PetscInt nhist, PetscErrorCode ierr) 259767be906fSBarry Smith .ve 2598a3b724e8SBarry Smith In other words this gets the current number of entries in the history. Access the history through the array you passed to `TaoSetConvergenceHistory()` 2599a7e14dcfSSatish Balay 26001cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSolve()`, `TaoSetConvergenceHistory()` 2601a7e14dcfSSatish Balay @*/ 2602d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetConvergenceHistory(Tao tao, PetscReal **obj, PetscReal **resid, PetscReal **cnorm, PetscInt **lits, PetscInt *nhist) 2603d71ae5a4SJacob Faibussowitsch { 2604a7e14dcfSSatish Balay PetscFunctionBegin; 2605441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2606a7e14dcfSSatish Balay if (obj) *obj = tao->hist_obj; 2607a7e14dcfSSatish Balay if (cnorm) *cnorm = tao->hist_cnorm; 2608a7e14dcfSSatish Balay if (resid) *resid = tao->hist_resid; 26095b494b05SBarry Smith if (lits) *lits = tao->hist_lits; 2610a7e14dcfSSatish Balay if (nhist) *nhist = tao->hist_len; 26113ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2612a7e14dcfSSatish Balay } 2613a7e14dcfSSatish Balay 2614a7e14dcfSSatish Balay /*@ 261547450a7bSBarry Smith TaoSetApplicationContext - Sets the optional user-defined context for a `Tao` solver. 2616a7e14dcfSSatish Balay 2617c3339decSBarry Smith Logically Collective 2618a7e14dcfSSatish Balay 2619a7e14dcfSSatish Balay Input Parameters: 262047450a7bSBarry Smith + tao - the `Tao` context 2621*49abdd8aSBarry Smith - ctx - optional user context 2622a7e14dcfSSatish Balay 2623a7e14dcfSSatish Balay Level: intermediate 2624a7e14dcfSSatish Balay 262542747ad1SJacob Faibussowitsch .seealso: [](ch_tao), `Tao`, `TaoGetApplicationContext()` 2626a7e14dcfSSatish Balay @*/ 2627*49abdd8aSBarry Smith PetscErrorCode TaoSetApplicationContext(Tao tao, void *ctx) 2628d71ae5a4SJacob Faibussowitsch { 2629a7e14dcfSSatish Balay PetscFunctionBegin; 2630441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2631*49abdd8aSBarry Smith tao->ctx = ctx; 26323ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2633a7e14dcfSSatish Balay } 2634a7e14dcfSSatish Balay 2635a7e14dcfSSatish Balay /*@ 263647450a7bSBarry Smith TaoGetApplicationContext - Gets the user-defined context for a `Tao` solver 2637a7e14dcfSSatish Balay 2638a7e14dcfSSatish Balay Not Collective 2639a7e14dcfSSatish Balay 2640a7e14dcfSSatish Balay Input Parameter: 264147450a7bSBarry Smith . tao - the `Tao` context 2642a7e14dcfSSatish Balay 2643a7e14dcfSSatish Balay Output Parameter: 2644*49abdd8aSBarry Smith . ctx - user context 2645a7e14dcfSSatish Balay 2646a7e14dcfSSatish Balay Level: intermediate 2647a7e14dcfSSatish Balay 26481cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetApplicationContext()` 2649a7e14dcfSSatish Balay @*/ 2650*49abdd8aSBarry Smith PetscErrorCode TaoGetApplicationContext(Tao tao, void *ctx) 2651d71ae5a4SJacob Faibussowitsch { 2652a7e14dcfSSatish Balay PetscFunctionBegin; 2653441846f8SBarry Smith PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 2654*49abdd8aSBarry Smith PetscAssertPointer(ctx, 2); 2655*49abdd8aSBarry Smith *(void **)ctx = tao->ctx; 26563ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2657a7e14dcfSSatish Balay } 2658a9603a14SPatrick Farrell 2659a9603a14SPatrick Farrell /*@ 2660a81c3919SBarry Smith TaoSetGradientNorm - Sets the matrix used to define the norm that measures the size of the gradient in some of the `Tao` algorithms 2661a9603a14SPatrick Farrell 2662c3339decSBarry Smith Collective 2663a9603a14SPatrick Farrell 2664a9603a14SPatrick Farrell Input Parameters: 266547450a7bSBarry Smith + tao - the `Tao` context 266665ba42b6SBarry Smith - M - matrix that defines the norm 2667a9603a14SPatrick Farrell 2668a9603a14SPatrick Farrell Level: beginner 2669a9603a14SPatrick Farrell 26701cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoGetGradientNorm()`, `TaoGradientNorm()` 2671a9603a14SPatrick Farrell @*/ 2672d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoSetGradientNorm(Tao tao, Mat M) 2673d71ae5a4SJacob Faibussowitsch { 2674a9603a14SPatrick Farrell PetscFunctionBegin; 2675a9603a14SPatrick Farrell PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 267694511df7SStefano Zampini PetscValidHeaderSpecific(M, MAT_CLASSID, 2); 26779566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)M)); 26789566063dSJacob Faibussowitsch PetscCall(MatDestroy(&tao->gradient_norm)); 26799566063dSJacob Faibussowitsch PetscCall(VecDestroy(&tao->gradient_norm_tmp)); 2680a9603a14SPatrick Farrell tao->gradient_norm = M; 26819566063dSJacob Faibussowitsch PetscCall(MatCreateVecs(M, NULL, &tao->gradient_norm_tmp)); 26823ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2683a9603a14SPatrick Farrell } 2684a9603a14SPatrick Farrell 2685a9603a14SPatrick Farrell /*@ 2686a81c3919SBarry Smith TaoGetGradientNorm - Returns the matrix used to define the norm used for measuring the size of the gradient in some of the `Tao` algorithms 2687a9603a14SPatrick Farrell 2688a9603a14SPatrick Farrell Not Collective 2689a9603a14SPatrick Farrell 2690a9603a14SPatrick Farrell Input Parameter: 269147450a7bSBarry Smith . tao - the `Tao` context 2692a9603a14SPatrick Farrell 2693a9603a14SPatrick Farrell Output Parameter: 2694a9603a14SPatrick Farrell . M - gradient norm 2695a9603a14SPatrick Farrell 2696a9603a14SPatrick Farrell Level: beginner 2697a9603a14SPatrick Farrell 26981cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetGradientNorm()`, `TaoGradientNorm()` 2699a9603a14SPatrick Farrell @*/ 2700d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGetGradientNorm(Tao tao, Mat *M) 2701d71ae5a4SJacob Faibussowitsch { 2702a9603a14SPatrick Farrell PetscFunctionBegin; 2703a9603a14SPatrick Farrell PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 27044f572ea9SToby Isaac PetscAssertPointer(M, 2); 2705a9603a14SPatrick Farrell *M = tao->gradient_norm; 27063ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2707a9603a14SPatrick Farrell } 2708a9603a14SPatrick Farrell 2709cc4c1da9SBarry Smith /*@ 271047450a7bSBarry Smith TaoGradientNorm - Compute the norm using the `NormType`, the user has selected 2711a9603a14SPatrick Farrell 2712c3339decSBarry Smith Collective 2713a9603a14SPatrick Farrell 2714d8d19677SJose E. Roman Input Parameters: 271547450a7bSBarry Smith + tao - the `Tao` context 2716a81c3919SBarry Smith . gradient - the gradient 27173b242c63SJacob Faibussowitsch - type - the norm type 2718a9603a14SPatrick Farrell 2719a9603a14SPatrick Farrell Output Parameter: 2720a9603a14SPatrick Farrell . gnorm - the gradient norm 2721a9603a14SPatrick Farrell 272247450a7bSBarry Smith Level: advanced 2723a9603a14SPatrick Farrell 2724a81c3919SBarry Smith Note: 2725a81c3919SBarry Smith If `TaoSetGradientNorm()` has been set and `type` is `NORM_2` then the norm provided with `TaoSetGradientNorm()` is used. 2726a81c3919SBarry Smith 2727a81c3919SBarry Smith Developer Notes: 2728a81c3919SBarry Smith Should be named `TaoComputeGradientNorm()`. 2729a81c3919SBarry Smith 2730a81c3919SBarry Smith The usage is a bit confusing, with `TaoSetGradientNorm()` plus `NORM_2` resulting in the computation of the user provided 2731a81c3919SBarry Smith norm, perhaps a refactorization is in order. 2732a81c3919SBarry Smith 27331cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoSetGradientNorm()`, `TaoGetGradientNorm()` 2734a9603a14SPatrick Farrell @*/ 2735d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoGradientNorm(Tao tao, Vec gradient, NormType type, PetscReal *gnorm) 2736d71ae5a4SJacob Faibussowitsch { 2737a9603a14SPatrick Farrell PetscFunctionBegin; 27388854b543SStefano Zampini PetscValidHeaderSpecific(tao, TAO_CLASSID, 1); 27398854b543SStefano Zampini PetscValidHeaderSpecific(gradient, VEC_CLASSID, 2); 27408854b543SStefano Zampini PetscValidLogicalCollectiveEnum(tao, type, 3); 27414f572ea9SToby Isaac PetscAssertPointer(gnorm, 4); 2742a9603a14SPatrick Farrell if (tao->gradient_norm) { 2743680e2bc7SPatrick Farrell PetscScalar gnorms; 2744680e2bc7SPatrick Farrell 27453c859ba3SBarry 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."); 27469566063dSJacob Faibussowitsch PetscCall(MatMult(tao->gradient_norm, gradient, tao->gradient_norm_tmp)); 27479566063dSJacob Faibussowitsch PetscCall(VecDot(gradient, tao->gradient_norm_tmp, &gnorms)); 27487bb79937SPatrick Farrell *gnorm = PetscRealPart(PetscSqrtScalar(gnorms)); 2749a9603a14SPatrick Farrell } else { 27509566063dSJacob Faibussowitsch PetscCall(VecNorm(gradient, type, gnorm)); 2751a9603a14SPatrick Farrell } 27523ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2753a9603a14SPatrick Farrell } 2754a9603a14SPatrick Farrell 2755e882e171SHong Zhang /*@C 2756a81c3919SBarry Smith TaoMonitorDrawCtxCreate - Creates the monitor context for `TaoMonitorSolutionDraw()` 2757a9603a14SPatrick Farrell 2758c3339decSBarry Smith Collective 2759e882e171SHong Zhang 27602fe279fdSBarry Smith Input Parameters: 27612fe279fdSBarry Smith + comm - the communicator to share the context 27622fe279fdSBarry Smith . host - the name of the X Windows host that will display the monitor 27632fe279fdSBarry Smith . label - the label to put at the top of the display window 27642fe279fdSBarry Smith . x - the horizontal coordinate of the lower left corner of the window to open 27652fe279fdSBarry Smith . y - the vertical coordinate of the lower left corner of the window to open 27662fe279fdSBarry Smith . m - the width of the window 27672fe279fdSBarry Smith . n - the height of the window 27682fe279fdSBarry Smith - howoften - how many `Tao` iterations between displaying the monitor information 27692fe279fdSBarry Smith 2770d5b43468SJose E. Roman Output Parameter: 2771e882e171SHong Zhang . ctx - the monitor context 2772e882e171SHong Zhang 2773a81c3919SBarry Smith Options Database Keys: 2774a81c3919SBarry Smith + -tao_monitor_solution_draw - use `TaoMonitorSolutionDraw()` to monitor the solution 2775a81c3919SBarry Smith - -tao_draw_solution_initial - show initial guess as well as current solution 2776e882e171SHong Zhang 2777e882e171SHong Zhang Level: intermediate 2778e882e171SHong Zhang 2779a81c3919SBarry Smith Note: 2780a81c3919SBarry Smith The context this creates, along with `TaoMonitorSolutionDraw()`, and `TaoMonitorDrawCtxDestroy()` 2781a81c3919SBarry Smith are passed to `TaoMonitorSet()`. 2782a81c3919SBarry Smith 27831cc06b55SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawCtx()` 2784e882e171SHong Zhang @*/ 2785d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDrawCtxCreate(MPI_Comm comm, const char host[], const char label[], int x, int y, int m, int n, PetscInt howoften, TaoMonitorDrawCtx *ctx) 2786d71ae5a4SJacob Faibussowitsch { 2787e882e171SHong Zhang PetscFunctionBegin; 27889566063dSJacob Faibussowitsch PetscCall(PetscNew(ctx)); 27899566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawOpen(comm, host, label, x, y, m, n, &(*ctx)->viewer)); 27909566063dSJacob Faibussowitsch PetscCall(PetscViewerSetFromOptions((*ctx)->viewer)); 2791e882e171SHong Zhang (*ctx)->howoften = howoften; 27923ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2793e882e171SHong Zhang } 2794e882e171SHong Zhang 2795e882e171SHong Zhang /*@C 2796a81c3919SBarry Smith TaoMonitorDrawCtxDestroy - Destroys the monitor context for `TaoMonitorSolutionDraw()` 2797e882e171SHong Zhang 2798c3339decSBarry Smith Collective 2799e882e171SHong Zhang 280047450a7bSBarry Smith Input Parameter: 2801e056e8ceSJacob Faibussowitsch . ictx - the monitor context 2802e882e171SHong Zhang 2803e882e171SHong Zhang Level: intermediate 2804e882e171SHong Zhang 2805a81c3919SBarry Smith Note: 2806a81c3919SBarry Smith This is passed to `TaoMonitorSet()` as the final argument, along with `TaoMonitorSolutionDraw()`, and the context 2807a81c3919SBarry Smith obtained with `TaoMonitorDrawCtxCreate()`. 2808a81c3919SBarry Smith 2809a81c3919SBarry Smith .seealso: [](ch_tao), `Tao`, `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorSolutionDraw()` 2810e882e171SHong Zhang @*/ 2811d71ae5a4SJacob Faibussowitsch PetscErrorCode TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *ictx) 2812d71ae5a4SJacob Faibussowitsch { 2813e882e171SHong Zhang PetscFunctionBegin; 28149566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&(*ictx)->viewer)); 28159566063dSJacob Faibussowitsch PetscCall(PetscFree(*ictx)); 28163ba16761SJacob Faibussowitsch PetscFunctionReturn(PETSC_SUCCESS); 2817e882e171SHong Zhang } 2818