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 240f0abf79SStefano Zampini static PetscErrorCode KSPPreSolve_TAOEW_Private(KSP ksp, Vec b, Vec x, Tao tao) 250f0abf79SStefano Zampini { 260f0abf79SStefano Zampini SNES snes_ewdummy = tao->snes_ewdummy; 270f0abf79SStefano Zampini 280f0abf79SStefano Zampini PetscFunctionBegin; 290f0abf79SStefano Zampini if (!snes_ewdummy) PetscFunctionReturn(0); 300f0abf79SStefano Zampini /* populate snes_ewdummy struct values used in KSPPreSolve_SNESEW */ 310f0abf79SStefano Zampini snes_ewdummy->vec_func = b; 320f0abf79SStefano Zampini snes_ewdummy->rtol = tao->gttol; 330f0abf79SStefano Zampini snes_ewdummy->iter = tao->niter; 340f0abf79SStefano Zampini PetscCall(VecNorm(b,NORM_2,&snes_ewdummy->norm)); 350f0abf79SStefano Zampini PetscCall(KSPPreSolve_SNESEW(ksp,b,x,snes_ewdummy)); 360f0abf79SStefano Zampini snes_ewdummy->vec_func = NULL; 370f0abf79SStefano Zampini PetscFunctionReturn(0); 380f0abf79SStefano Zampini } 390f0abf79SStefano Zampini 400f0abf79SStefano Zampini static PetscErrorCode KSPPostSolve_TAOEW_Private(KSP ksp, Vec b, Vec x, Tao tao) 410f0abf79SStefano Zampini { 420f0abf79SStefano Zampini SNES snes_ewdummy = tao->snes_ewdummy; 430f0abf79SStefano Zampini 440f0abf79SStefano Zampini PetscFunctionBegin; 450f0abf79SStefano Zampini if (!snes_ewdummy) PetscFunctionReturn(0); 460f0abf79SStefano Zampini PetscCall(KSPPostSolve_SNESEW(ksp,b,x,snes_ewdummy)); 470f0abf79SStefano Zampini PetscFunctionReturn(0); 480f0abf79SStefano Zampini } 490f0abf79SStefano Zampini 500f0abf79SStefano Zampini static PetscErrorCode TaoSetUpEW_Private(Tao tao) 510f0abf79SStefano Zampini { 520f0abf79SStefano Zampini SNESKSPEW *kctx; 530f0abf79SStefano Zampini const char *ewprefix; 540f0abf79SStefano Zampini 550f0abf79SStefano Zampini PetscFunctionBegin; 560f0abf79SStefano Zampini if (!tao->ksp) PetscFunctionReturn(0); 570f0abf79SStefano Zampini if (tao->ksp_ewconv) { 580f0abf79SStefano Zampini if (!tao->snes_ewdummy) PetscCall(SNESCreate(PetscObjectComm((PetscObject)tao),&tao->snes_ewdummy)); 590f0abf79SStefano Zampini tao->snes_ewdummy->ksp_ewconv = PETSC_TRUE; 600f0abf79SStefano Zampini PetscCall(KSPSetPreSolve(tao->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPreSolve_TAOEW_Private,tao)); 610f0abf79SStefano Zampini PetscCall(KSPSetPostSolve(tao->ksp,(PetscErrorCode (*)(KSP,Vec,Vec,void*))KSPPostSolve_TAOEW_Private,tao)); 620f0abf79SStefano Zampini 630f0abf79SStefano Zampini PetscCall(KSPGetOptionsPrefix(tao->ksp,&ewprefix)); 640f0abf79SStefano Zampini kctx = (SNESKSPEW*)tao->snes_ewdummy->kspconvctx; 650f0abf79SStefano Zampini PetscCall(SNESEWSetFromOptions_Private(kctx,PetscObjectComm((PetscObject)tao),ewprefix)); 660f0abf79SStefano Zampini } else PetscCall(SNESDestroy(&tao->snes_ewdummy)); 670f0abf79SStefano Zampini PetscFunctionReturn(0); 680f0abf79SStefano Zampini } 690f0abf79SStefano Zampini 70a7e14dcfSSatish Balay /*@ 71a7e14dcfSSatish Balay TaoCreate - Creates a TAO solver 72a7e14dcfSSatish Balay 73d083f849SBarry Smith Collective 74a7e14dcfSSatish Balay 75a7e14dcfSSatish Balay Input Parameter: 76a7e14dcfSSatish Balay . comm - MPI communicator 77a7e14dcfSSatish Balay 78a7e14dcfSSatish Balay Output Parameter: 79441846f8SBarry Smith . newtao - the new Tao context 80a7e14dcfSSatish Balay 81a7e14dcfSSatish Balay Available methods include: 8258417fe7SBarry Smith + nls - Newton's method with line search for unconstrained minimization 8358417fe7SBarry Smith . ntr - Newton's method with trust region for unconstrained minimization 8458417fe7SBarry Smith . ntl - Newton's method with trust region, line search for unconstrained minimization 8558417fe7SBarry Smith . lmvm - Limited memory variable metric method for unconstrained minimization 8658417fe7SBarry Smith . cg - Nonlinear conjugate gradient method for unconstrained minimization 8758417fe7SBarry Smith . nm - Nelder-Mead algorithm for derivate-free unconstrained minimization 8858417fe7SBarry Smith . tron - Newton Trust Region method for bound constrained minimization 8958417fe7SBarry Smith . gpcg - Newton Trust Region method for quadratic bound constrained minimization 9058417fe7SBarry Smith . blmvm - Limited memory variable metric method for bound constrained minimization 9158417fe7SBarry Smith . lcl - Linearly constrained Lagrangian method for pde-constrained minimization 9258417fe7SBarry Smith - pounders - Model-based algorithm for nonlinear least squares 93a7e14dcfSSatish Balay 94a7e14dcfSSatish Balay Options Database Keys: 9558417fe7SBarry Smith . -tao_type - select which method TAO should use 96a7e14dcfSSatish Balay 97a7e14dcfSSatish Balay Level: beginner 98a7e14dcfSSatish Balay 99db781477SPatrick Sanan .seealso: `TaoSolve()`, `TaoDestroy()` 100a7e14dcfSSatish Balay @*/ 101441846f8SBarry Smith PetscErrorCode TaoCreate(MPI_Comm comm, Tao *newtao) 102a7e14dcfSSatish Balay { 103441846f8SBarry Smith Tao tao; 104a7e14dcfSSatish Balay 105a7e14dcfSSatish Balay PetscFunctionBegin; 106a7e14dcfSSatish Balay PetscValidPointer(newtao,2); 1079566063dSJacob Faibussowitsch PetscCall(TaoInitializePackage()); 1089566063dSJacob Faibussowitsch PetscCall(TaoLineSearchInitializePackage()); 1099566063dSJacob Faibussowitsch PetscCall(PetscHeaderCreate(tao,TAO_CLASSID,"Tao","Optimization solver","Tao",comm,TaoDestroy,TaoView)); 110a7e14dcfSSatish Balay 11194511df7SStefano Zampini /* Set non-NULL defaults */ 11294511df7SStefano Zampini tao->ops->convergencetest = TaoDefaultConvergenceTest; 113a7e14dcfSSatish Balay 114a7e14dcfSSatish Balay tao->max_it = 10000; 11594511df7SStefano Zampini tao->max_funcs = -1; 1166f4723b1SBarry Smith #if defined(PETSC_USE_REAL_SINGLE) 1176f4723b1SBarry Smith tao->gatol = 1e-5; 1186f4723b1SBarry Smith tao->grtol = 1e-5; 1196f1f5f59SAlp Dener tao->crtol = 1e-5; 1206f1f5f59SAlp Dener tao->catol = 1e-5; 1216f4723b1SBarry Smith #else 122a7e14dcfSSatish Balay tao->gatol = 1e-8; 123a7e14dcfSSatish Balay tao->grtol = 1e-8; 1246f1f5f59SAlp Dener tao->crtol = 1e-8; 1256f1f5f59SAlp Dener tao->catol = 1e-8; 1266f4723b1SBarry Smith #endif 1276552cf8aSJason Sarich tao->gttol = 0.0; 1284f1535bcSTodd Munson tao->steptol = 0.0; 129e270355aSBarry Smith tao->trust0 = PETSC_INFINITY; 1306f4723b1SBarry Smith tao->fmin = PETSC_NINFINITY; 1319fa2b5dcSStefano Zampini 132a7e14dcfSSatish Balay tao->hist_reset = PETSC_TRUE; 133a7e14dcfSSatish Balay 1349566063dSJacob Faibussowitsch PetscCall(TaoResetStatistics(tao)); 135a7e14dcfSSatish Balay *newtao = tao; 136a7e14dcfSSatish Balay PetscFunctionReturn(0); 137a7e14dcfSSatish Balay } 138a7e14dcfSSatish Balay 139a7e14dcfSSatish Balay /*@ 140a7e14dcfSSatish Balay TaoSolve - Solves an optimization problem min F(x) s.t. l <= x <= u 141a7e14dcfSSatish Balay 142441846f8SBarry Smith Collective on Tao 143a7e14dcfSSatish Balay 144a7e14dcfSSatish Balay Input Parameters: 145441846f8SBarry Smith . tao - the Tao context 146a7e14dcfSSatish Balay 147a7e14dcfSSatish Balay Notes: 148a82e8c82SStefano Zampini The user must set up the Tao with calls to TaoSetSolution(), 149a82e8c82SStefano Zampini TaoSetObjective(), 150a82e8c82SStefano Zampini TaoSetGradient(), and (if using 2nd order method) TaoSetHessian(). 151a7e14dcfSSatish Balay 152a35d58b8SBarry Smith You should call TaoGetConvergedReason() or run with -tao_converged_reason to determine if the optimization algorithm actually succeeded or 153a35d58b8SBarry Smith why it failed. 154a35d58b8SBarry Smith 155a7e14dcfSSatish Balay Level: beginner 156a7e14dcfSSatish Balay 157db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSetObjective()`, `TaoSetGradient()`, `TaoSetHessian()`, `TaoGetConvergedReason()` 158a7e14dcfSSatish Balay @*/ 159441846f8SBarry Smith PetscErrorCode TaoSolve(Tao tao) 160a7e14dcfSSatish Balay { 161e2379f4fSBarry Smith static PetscBool set = PETSC_FALSE; 16247a47007SBarry Smith 163a7e14dcfSSatish Balay PetscFunctionBegin; 164441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 165d0609cedSBarry Smith PetscCall(PetscCitationsRegister("@TechReport{tao-user-ref,\n" 166e2379f4fSBarry Smith "title = {Toolkit for Advanced Optimization (TAO) Users Manual},\n" 167e2379f4fSBarry Smith "author = {Todd Munson and Jason Sarich and Stefan Wild and Steve Benson and Lois Curfman McInnes},\n" 168e2379f4fSBarry Smith "Institution = {Argonne National Laboratory},\n" 169e2379f4fSBarry Smith "Year = 2014,\n" 170e2379f4fSBarry Smith "Number = {ANL/MCS-TM-322 - Revision 3.5},\n" 171d0609cedSBarry Smith "url = {https://www.mcs.anl.gov/research/projects/tao/}\n}\n",&set)); 172494bef23SAlp Dener tao->header_printed = PETSC_FALSE; 1739566063dSJacob Faibussowitsch PetscCall(TaoSetUp(tao)); 1749566063dSJacob Faibussowitsch PetscCall(TaoResetStatistics(tao)); 175*1baa6e33SBarry Smith if (tao->linesearch) PetscCall(TaoLineSearchReset(tao->linesearch)); 176a7e14dcfSSatish Balay 1779566063dSJacob Faibussowitsch PetscCall(PetscLogEventBegin(TAO_Solve,tao,0,0,0)); 1789566063dSJacob Faibussowitsch if (tao->ops->solve) PetscCall((*tao->ops->solve)(tao)); 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; 1849566063dSJacob Faibussowitsch PetscCall(TaoViewFromOptions(tao,NULL,"-tao_view")); 185a7e14dcfSSatish Balay 186a7e14dcfSSatish Balay if (tao->printreason) { 187a7e14dcfSSatish Balay if (tao->reason > 0) { 18863a3b9bcSJacob Faibussowitsch PetscCall(PetscPrintf(((PetscObject)tao)->comm,"TAO solve converged due to %s iterations %" PetscInt_FMT "\n",TaoConvergedReasons[tao->reason],tao->niter)); 189a7e14dcfSSatish Balay } else { 19063a3b9bcSJacob Faibussowitsch PetscCall(PetscPrintf(((PetscObject)tao)->comm,"TAO solve did not converge due to %s iteration %" PetscInt_FMT "\n",TaoConvergedReasons[tao->reason],tao->niter)); 191a7e14dcfSSatish Balay } 192a7e14dcfSSatish Balay } 193a7e14dcfSSatish Balay PetscFunctionReturn(0); 194a7e14dcfSSatish Balay } 195a7e14dcfSSatish Balay 196a7e14dcfSSatish Balay /*@ 197a7e14dcfSSatish Balay TaoSetUp - Sets up the internal data structures for the later use 198a7e14dcfSSatish Balay of a Tao solver 199a7e14dcfSSatish Balay 200a7e14dcfSSatish Balay Collective on tao 201a7e14dcfSSatish Balay 202a7e14dcfSSatish Balay Input Parameters: 203a7e14dcfSSatish Balay . tao - the TAO context 204a7e14dcfSSatish Balay 205a7e14dcfSSatish Balay Notes: 206a7e14dcfSSatish Balay The user will not need to explicitly call TaoSetUp(), as it will 207a7e14dcfSSatish Balay automatically be called in TaoSolve(). However, if the user 208a7e14dcfSSatish Balay desires to call it explicitly, it should come after TaoCreate() 209a7e14dcfSSatish Balay and any TaoSetSomething() routines, but before TaoSolve(). 210a7e14dcfSSatish Balay 211a7e14dcfSSatish Balay Level: advanced 212a7e14dcfSSatish Balay 213db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSolve()` 214a7e14dcfSSatish Balay @*/ 215441846f8SBarry Smith PetscErrorCode TaoSetUp(Tao tao) 216a7e14dcfSSatish Balay { 217a7e14dcfSSatish Balay PetscFunctionBegin; 218441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 219a7e14dcfSSatish Balay if (tao->setupcalled) PetscFunctionReturn(0); 2200f0abf79SStefano Zampini PetscCall(TaoSetUpEW_Private(tao)); 2213c859ba3SBarry Smith PetscCheck(tao->solution,PetscObjectComm((PetscObject)tao),PETSC_ERR_ARG_WRONGSTATE,"Must call TaoSetSolution"); 2220f0abf79SStefano Zampini if (tao->ops->setup) PetscCall((*tao->ops->setup)(tao)); 223a7e14dcfSSatish Balay tao->setupcalled = PETSC_TRUE; 224a7e14dcfSSatish Balay PetscFunctionReturn(0); 225a7e14dcfSSatish Balay } 226a7e14dcfSSatish Balay 2271fb7b255SJunchao Zhang /*@C 228a7e14dcfSSatish Balay TaoDestroy - Destroys the TAO context that was created with 229a7e14dcfSSatish Balay TaoCreate() 230a7e14dcfSSatish Balay 231441846f8SBarry Smith Collective on Tao 232a7e14dcfSSatish Balay 233a7e14dcfSSatish Balay Input Parameter: 234441846f8SBarry Smith . tao - the Tao context 235a7e14dcfSSatish Balay 236a7e14dcfSSatish Balay Level: beginner 237a7e14dcfSSatish Balay 238db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSolve()` 239a7e14dcfSSatish Balay @*/ 240441846f8SBarry Smith PetscErrorCode TaoDestroy(Tao *tao) 241a7e14dcfSSatish Balay { 242a7e14dcfSSatish Balay PetscFunctionBegin; 243a7e14dcfSSatish Balay if (!*tao) PetscFunctionReturn(0); 244441846f8SBarry Smith PetscValidHeaderSpecific(*tao,TAO_CLASSID,1); 24583c8fe1dSLisandro Dalcin if (--((PetscObject)*tao)->refct > 0) {*tao = NULL;PetscFunctionReturn(0);} 246a7e14dcfSSatish Balay 247a7e14dcfSSatish Balay if ((*tao)->ops->destroy) { 2489566063dSJacob Faibussowitsch PetscCall((*((*tao))->ops->destroy)(*tao)); 249a7e14dcfSSatish Balay } 2509566063dSJacob Faibussowitsch PetscCall(KSPDestroy(&(*tao)->ksp)); 2510f0abf79SStefano Zampini PetscCall(SNESDestroy(&(*tao)->snes_ewdummy)); 2529566063dSJacob Faibussowitsch PetscCall(TaoLineSearchDestroy(&(*tao)->linesearch)); 253a7e14dcfSSatish Balay 254a7e14dcfSSatish Balay if ((*tao)->ops->convergencedestroy) { 2559566063dSJacob Faibussowitsch PetscCall((*(*tao)->ops->convergencedestroy)((*tao)->cnvP)); 256a7e14dcfSSatish Balay if ((*tao)->jacobian_state_inv) { 2579566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_state_inv)); 258a7e14dcfSSatish Balay } 259a7e14dcfSSatish Balay } 2609566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->solution)); 2619566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->gradient)); 2629566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->ls_res)); 263a7e14dcfSSatish Balay 264a9603a14SPatrick Farrell if ((*tao)->gradient_norm) { 2659566063dSJacob Faibussowitsch PetscCall(PetscObjectDereference((PetscObject)(*tao)->gradient_norm)); 2669566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->gradient_norm_tmp)); 267a9603a14SPatrick Farrell } 268a9603a14SPatrick Farrell 2699566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->XL)); 2709566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->XU)); 2719566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->IL)); 2729566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->IU)); 2739566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->DE)); 2749566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->DI)); 2759566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->constraints)); 2769566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->constraints_equality)); 2779566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->constraints_inequality)); 2789566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->stepdirection)); 2799566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->hessian_pre)); 2809566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->hessian)); 2819566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->ls_jac)); 2829566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->ls_jac_pre)); 2839566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_pre)); 2849566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian)); 2859566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_state_pre)); 2869566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_state)); 2879566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_state_inv)); 2889566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_design)); 2899566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_equality)); 2909566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_equality_pre)); 2919566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_inequality)); 2929566063dSJacob Faibussowitsch PetscCall(MatDestroy(&(*tao)->jacobian_inequality_pre)); 2939566063dSJacob Faibussowitsch PetscCall(ISDestroy(&(*tao)->state_is)); 2949566063dSJacob Faibussowitsch PetscCall(ISDestroy(&(*tao)->design_is)); 2959566063dSJacob Faibussowitsch PetscCall(VecDestroy(&(*tao)->res_weights_v)); 2969566063dSJacob Faibussowitsch PetscCall(TaoCancelMonitors(*tao)); 297ae93cb3cSJason Sarich if ((*tao)->hist_malloc) { 2989566063dSJacob Faibussowitsch PetscCall(PetscFree4((*tao)->hist_obj,(*tao)->hist_resid,(*tao)->hist_cnorm,(*tao)->hist_lits)); 299ae93cb3cSJason Sarich } 300737f463aSAlp Dener if ((*tao)->res_weights_n) { 3019566063dSJacob Faibussowitsch PetscCall(PetscFree((*tao)->res_weights_rows)); 3029566063dSJacob Faibussowitsch PetscCall(PetscFree((*tao)->res_weights_cols)); 3039566063dSJacob Faibussowitsch PetscCall(PetscFree((*tao)->res_weights_w)); 304125ddc8aSJason Sarich } 3059566063dSJacob Faibussowitsch PetscCall(PetscHeaderDestroy(tao)); 306a7e14dcfSSatish Balay PetscFunctionReturn(0); 307a7e14dcfSSatish Balay } 308a7e14dcfSSatish Balay 309a7e14dcfSSatish Balay /*@ 3100f0abf79SStefano Zampini TaoKSPSetUseEW - Sets SNES use Eisenstat-Walker method for 3110f0abf79SStefano Zampini computing relative tolerance for linear solvers. 3120f0abf79SStefano Zampini 3130f0abf79SStefano Zampini Logically Collective on TAO 3140f0abf79SStefano Zampini 3150f0abf79SStefano Zampini Input Parameters: 3160f0abf79SStefano Zampini + tao - Tao context 3170f0abf79SStefano Zampini - flag - PETSC_TRUE or PETSC_FALSE 3180f0abf79SStefano Zampini 3190f0abf79SStefano Zampini Notes: 3200f0abf79SStefano Zampini See SNESKSPSetUseEW() for customization details. 3210f0abf79SStefano Zampini 3220f0abf79SStefano Zampini Level: advanced 3230f0abf79SStefano Zampini 3240f0abf79SStefano Zampini Reference: 3250f0abf79SStefano Zampini S. C. Eisenstat and H. F. Walker, "Choosing the forcing terms in an 3260f0abf79SStefano Zampini inexact Newton method", SISC 17 (1), pp.16-32, 1996. 3270f0abf79SStefano Zampini 3280f0abf79SStefano Zampini .seealso: `SNESKSPSetUseEW()` 3290f0abf79SStefano Zampini @*/ 3300f0abf79SStefano Zampini PetscErrorCode TaoKSPSetUseEW(Tao tao,PetscBool flag) 3310f0abf79SStefano Zampini { 3320f0abf79SStefano Zampini PetscFunctionBegin; 3330f0abf79SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 3340f0abf79SStefano Zampini PetscValidLogicalCollectiveBool(tao,flag,2); 3350f0abf79SStefano Zampini tao->ksp_ewconv = flag; 3360f0abf79SStefano Zampini PetscFunctionReturn(0); 3370f0abf79SStefano Zampini } 3380f0abf79SStefano Zampini 3390f0abf79SStefano Zampini /*@ 340441846f8SBarry Smith TaoSetFromOptions - Sets various Tao parameters from user 341a7e14dcfSSatish Balay options. 342a7e14dcfSSatish Balay 343441846f8SBarry Smith Collective on Tao 344a7e14dcfSSatish Balay 34501d2d390SJose E. Roman Input Parameter: 346441846f8SBarry Smith . tao - the Tao solver context 347a7e14dcfSSatish Balay 348a7e14dcfSSatish Balay options Database Keys: 34958417fe7SBarry Smith + -tao_type <type> - The algorithm that TAO uses (lmvm, nls, etc.) 350a7e14dcfSSatish Balay . -tao_gatol <gatol> - absolute error tolerance for ||gradient|| 351a7e14dcfSSatish Balay . -tao_grtol <grtol> - relative error tolerance for ||gradient|| 352a7e14dcfSSatish Balay . -tao_gttol <gttol> - reduction of ||gradient|| relative to initial gradient 353a7e14dcfSSatish Balay . -tao_max_it <max> - sets maximum number of iterations 354a7e14dcfSSatish Balay . -tao_max_funcs <max> - sets maximum number of function evaluations 355a7e14dcfSSatish Balay . -tao_fmin <fmin> - stop if function value reaches fmin 356a7e14dcfSSatish Balay . -tao_steptol <tol> - stop if trust region radius less than <tol> 357a7e14dcfSSatish Balay . -tao_trust0 <t> - initial trust region radius 358a7e14dcfSSatish Balay . -tao_monitor - prints function value and residual at each iteration 359a7e14dcfSSatish Balay . -tao_smonitor - same as tao_monitor, but truncates very small values 360a7e14dcfSSatish Balay . -tao_cmonitor - prints function value, residual, and constraint norm at each iteration 361a7e14dcfSSatish Balay . -tao_view_solution - prints solution vector at each iteration 3624a48860cSAlp Dener . -tao_view_ls_residual - prints least-squares residual vector at each iteration 363147403d9SBarry Smith . -tao_view_stepdirection - prints step direction vector at each iteration 364a7e14dcfSSatish Balay . -tao_view_gradient - prints gradient vector at each iteration 365a7e14dcfSSatish Balay . -tao_draw_solution - graphically view solution vector at each iteration 366a7e14dcfSSatish Balay . -tao_draw_step - graphically view step vector at each iteration 367a7e14dcfSSatish Balay . -tao_draw_gradient - graphically view gradient at each iteration 368a7e14dcfSSatish Balay . -tao_fd_gradient - use gradient computed with finite differences 369f4c1ad5cSStefano Zampini . -tao_fd_hessian - use hessian computed with finite differences 370f4c1ad5cSStefano Zampini . -tao_mf_hessian - use matrix-free hessian computed with finite differences 371a7e14dcfSSatish Balay . -tao_cancelmonitors - cancels all monitors (except those set with command line) 372441846f8SBarry Smith . -tao_view - prints information about the Tao after solving 373a7e14dcfSSatish Balay - -tao_converged_reason - prints the reason TAO stopped iterating 374a7e14dcfSSatish Balay 375a7e14dcfSSatish Balay Notes: 376a7e14dcfSSatish Balay To see all options, run your program with the -help option or consult the 377a7e14dcfSSatish Balay user's manual. Should be called after TaoCreate() but before TaoSolve() 378a7e14dcfSSatish Balay 379a7e14dcfSSatish Balay Level: beginner 380a7e14dcfSSatish Balay @*/ 381441846f8SBarry Smith PetscErrorCode TaoSetFromOptions(Tao tao) 382a7e14dcfSSatish Balay { 383b625d6c7SJed Brown TaoType default_type = TAOLMVM; 384a7e14dcfSSatish Balay char type[256], monfilename[PETSC_MAX_PATH_LEN]; 385a7e14dcfSSatish Balay PetscViewer monviewer; 386a7e14dcfSSatish Balay PetscBool flg; 387a7e14dcfSSatish Balay MPI_Comm comm; 388a7e14dcfSSatish Balay 389a7e14dcfSSatish Balay PetscFunctionBegin; 390441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 3919566063dSJacob Faibussowitsch PetscCall(PetscObjectGetComm((PetscObject)tao,&comm)); 392125ddc8aSJason Sarich 393a7e14dcfSSatish Balay /* So no warnings are given about unused options */ 3949566063dSJacob Faibussowitsch PetscCall(PetscOptionsHasName(((PetscObject)tao)->options,((PetscObject)tao)->prefix,"-tao_ls_type",&flg)); 395a7e14dcfSSatish Balay 396d0609cedSBarry Smith PetscObjectOptionsBegin((PetscObject)tao); 397a7e14dcfSSatish Balay { 398a82e8c82SStefano Zampini if (((PetscObject)tao)->type_name) default_type = ((PetscObject)tao)->type_name; 399a7e14dcfSSatish Balay /* Check for type from options */ 4009566063dSJacob Faibussowitsch PetscCall(PetscOptionsFList("-tao_type","Tao Solver type","TaoSetType",TaoList,default_type,type,256,&flg)); 401a7e14dcfSSatish Balay if (flg) { 4029566063dSJacob Faibussowitsch PetscCall(TaoSetType(tao,type)); 403a7e14dcfSSatish Balay } else if (!((PetscObject)tao)->type_name) { 4049566063dSJacob Faibussowitsch PetscCall(TaoSetType(tao,default_type)); 405a7e14dcfSSatish Balay } 406a7e14dcfSSatish Balay 4079566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_catol","Stop if constraints violations within","TaoSetConstraintTolerances",tao->catol,&tao->catol,&flg)); 4086552cf8aSJason Sarich if (flg) tao->catol_changed = PETSC_TRUE; 4096aad120cSJose E. Roman PetscCall(PetscOptionsReal("-tao_crtol","Stop if relative constraint violations within","TaoSetConstraintTolerances",tao->crtol,&tao->crtol,&flg)); 4106552cf8aSJason Sarich if (flg) tao->crtol_changed = PETSC_TRUE; 4119566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_gatol","Stop if norm of gradient less than","TaoSetTolerances",tao->gatol,&tao->gatol,&flg)); 4126552cf8aSJason Sarich if (flg) tao->gatol_changed = PETSC_TRUE; 4139566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_grtol","Stop if norm of gradient divided by the function value is less than","TaoSetTolerances",tao->grtol,&tao->grtol,&flg)); 4146552cf8aSJason Sarich if (flg) tao->grtol_changed = PETSC_TRUE; 4159566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_gttol","Stop if the norm of the gradient is less than the norm of the initial gradient times tol","TaoSetTolerances",tao->gttol,&tao->gttol,&flg)); 4166552cf8aSJason Sarich if (flg) tao->gttol_changed = PETSC_TRUE; 4179566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-tao_max_it","Stop if iteration number exceeds","TaoSetMaximumIterations",tao->max_it,&tao->max_it,&flg)); 4186552cf8aSJason Sarich if (flg) tao->max_it_changed = PETSC_TRUE; 4199566063dSJacob Faibussowitsch PetscCall(PetscOptionsInt("-tao_max_funcs","Stop if number of function evaluations exceeds","TaoSetMaximumFunctionEvaluations",tao->max_funcs,&tao->max_funcs,&flg)); 4206552cf8aSJason Sarich if (flg) tao->max_funcs_changed = PETSC_TRUE; 4219566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_fmin","Stop if function less than","TaoSetFunctionLowerBound",tao->fmin,&tao->fmin,&flg)); 4226552cf8aSJason Sarich if (flg) tao->fmin_changed = PETSC_TRUE; 4239566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_steptol","Stop if step size or trust region radius less than","",tao->steptol,&tao->steptol,&flg)); 4246552cf8aSJason Sarich if (flg) tao->steptol_changed = PETSC_TRUE; 4259566063dSJacob Faibussowitsch PetscCall(PetscOptionsReal("-tao_trust0","Initial trust region radius","TaoSetTrustRegionRadius",tao->trust0,&tao->trust0,&flg)); 4266552cf8aSJason Sarich if (flg) tao->trust0_changed = PETSC_TRUE; 4279566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_view_solution","view solution vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg)); 428a7e14dcfSSatish Balay if (flg) { 4299566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer)); 4309566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao,TaoSolutionMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy)); 431a7e14dcfSSatish Balay } 432a7e14dcfSSatish Balay 4339566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_converged_reason","Print reason for TAO converged","TaoSolve",tao->printreason,&tao->printreason,NULL)); 4349566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_view_gradient","view gradient vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg)); 435a7e14dcfSSatish Balay if (flg) { 4369566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer)); 4379566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao,TaoGradientMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy)); 438a7e14dcfSSatish Balay } 439a7e14dcfSSatish Balay 4409566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_view_stepdirection","view step direction vector after each iteration","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg)); 441a7e14dcfSSatish Balay if (flg) { 4429566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer)); 4439566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao,TaoStepDirectionMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy)); 444a7e14dcfSSatish Balay } 445a7e14dcfSSatish Balay 4469566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_view_residual","view least-squares residual vector after each evaluation","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg)); 447a7e14dcfSSatish Balay if (flg) { 4489566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer)); 4499566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao,TaoResidualMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy)); 450a7e14dcfSSatish Balay } 451a7e14dcfSSatish Balay 4529566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_monitor","Use the default convergence monitor","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg)); 453a7e14dcfSSatish Balay if (flg) { 4549566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer)); 4559566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao,TaoMonitorDefault,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy)); 456a7e14dcfSSatish Balay } 457a7e14dcfSSatish Balay 4589566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_gmonitor","Use the convergence monitor with extra globalization info","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg)); 4598d5ead36SAlp Dener if (flg) { 4609566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer)); 4619566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao,TaoDefaultGMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy)); 4628d5ead36SAlp Dener } 4638d5ead36SAlp Dener 4649566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_smonitor","Use the short convergence monitor","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg)); 465a7e14dcfSSatish Balay if (flg) { 4669566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer)); 4679566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao,TaoDefaultSMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy)); 468a7e14dcfSSatish Balay } 469a7e14dcfSSatish Balay 4709566063dSJacob Faibussowitsch PetscCall(PetscOptionsString("-tao_cmonitor","Use the default convergence monitor with constraint norm","TaoSetMonitor","stdout",monfilename,sizeof(monfilename),&flg)); 471a7e14dcfSSatish Balay if (flg) { 4729566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIOpen(comm,monfilename,&monviewer)); 4739566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao,TaoDefaultCMonitor,monviewer,(PetscErrorCode (*)(void**))PetscViewerDestroy)); 474a7e14dcfSSatish Balay } 475a7e14dcfSSatish Balay 4768afaa268SBarry Smith flg = PETSC_FALSE; 4779566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_cancelmonitors","cancel all monitors and call any registered destroy routines","TaoCancelMonitors",flg,&flg,NULL)); 4789566063dSJacob Faibussowitsch if (flg) PetscCall(TaoCancelMonitors(tao)); 479a7e14dcfSSatish Balay 4808afaa268SBarry Smith flg = PETSC_FALSE; 4819566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_draw_solution","Plot solution vector at each iteration","TaoSetMonitor",flg,&flg,NULL)); 482a7e14dcfSSatish Balay if (flg) { 483e882e171SHong Zhang TaoMonitorDrawCtx drawctx; 484e882e171SHong Zhang PetscInt howoften = 1; 4859566063dSJacob Faibussowitsch PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,300,300,howoften,&drawctx)); 4869566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao,TaoDrawSolutionMonitor,drawctx,(PetscErrorCode (*)(void**))TaoMonitorDrawCtxDestroy)); 487a7e14dcfSSatish Balay } 488a7e14dcfSSatish Balay 4898afaa268SBarry Smith flg = PETSC_FALSE; 4909566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_draw_step","plots step direction at each iteration","TaoSetMonitor",flg,&flg,NULL)); 491*1baa6e33SBarry Smith if (flg) PetscCall(TaoSetMonitor(tao,TaoDrawStepMonitor,NULL,NULL)); 492a7e14dcfSSatish Balay 4938afaa268SBarry Smith flg = PETSC_FALSE; 4949566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_draw_gradient","plots gradient at each iteration","TaoSetMonitor",flg,&flg,NULL)); 495a7e14dcfSSatish Balay if (flg) { 496e882e171SHong Zhang TaoMonitorDrawCtx drawctx; 497e882e171SHong Zhang PetscInt howoften = 1; 4989566063dSJacob Faibussowitsch PetscCall(TaoMonitorDrawCtxCreate(PetscObjectComm((PetscObject)tao),NULL,NULL,PETSC_DECIDE,PETSC_DECIDE,300,300,howoften,&drawctx)); 4999566063dSJacob Faibussowitsch PetscCall(TaoSetMonitor(tao,TaoDrawGradientMonitor,drawctx,(PetscErrorCode (*)(void**))TaoMonitorDrawCtxDestroy)); 500a7e14dcfSSatish Balay } 5018afaa268SBarry Smith flg = PETSC_FALSE; 5029566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_fd_gradient","compute gradient using finite differences","TaoDefaultComputeGradient",flg,&flg,NULL)); 503*1baa6e33SBarry Smith if (flg) PetscCall(TaoSetGradient(tao,NULL,TaoDefaultComputeGradient,NULL)); 504f4c1ad5cSStefano Zampini flg = PETSC_FALSE; 5059566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_fd_hessian","compute hessian using finite differences","TaoDefaultComputeHessian",flg,&flg,NULL)); 506f4c1ad5cSStefano Zampini if (flg) { 507f4c1ad5cSStefano Zampini Mat H; 508f4c1ad5cSStefano Zampini 5099566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)tao),&H)); 5109566063dSJacob Faibussowitsch PetscCall(MatSetType(H,MATAIJ)); 5119566063dSJacob Faibussowitsch PetscCall(TaoSetHessian(tao,H,H,TaoDefaultComputeHessian,NULL)); 5129566063dSJacob Faibussowitsch PetscCall(MatDestroy(&H)); 513f4c1ad5cSStefano Zampini } 514f4c1ad5cSStefano Zampini flg = PETSC_FALSE; 5159566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_mf_hessian","compute matrix-free hessian using finite differences","TaoDefaultComputeHessianMFFD",flg,&flg,NULL)); 516f4c1ad5cSStefano Zampini if (flg) { 517f4c1ad5cSStefano Zampini Mat H; 518f4c1ad5cSStefano Zampini 5199566063dSJacob Faibussowitsch PetscCall(MatCreate(PetscObjectComm((PetscObject)tao),&H)); 5209566063dSJacob Faibussowitsch PetscCall(TaoSetHessian(tao,H,H,TaoDefaultComputeHessianMFFD,NULL)); 5219566063dSJacob Faibussowitsch PetscCall(MatDestroy(&H)); 522f4c1ad5cSStefano Zampini } 523414d97d3SAlp Dener flg = PETSC_FALSE; 5249566063dSJacob Faibussowitsch PetscCall(PetscOptionsBool("-tao_recycle_history","enable recycling/re-using information from the previous TaoSolve() call for some algorithms","TaoSetRecycleHistory",flg,&flg,NULL)); 525*1baa6e33SBarry Smith if (flg) PetscCall(TaoSetRecycleHistory(tao,PETSC_TRUE)); 5269566063dSJacob Faibussowitsch PetscCall(PetscOptionsEnum("-tao_subset_type","subset type","",TaoSubSetTypes,(PetscEnum)tao->subset_type,(PetscEnum*)&tao->subset_type,NULL)); 527a7e14dcfSSatish Balay 5280f0abf79SStefano Zampini if (tao->ksp) { 5290f0abf79SStefano Zampini PetscCall(PetscOptionsBool("-tao_ksp_ew","Use Eisentat-Walker linear system convergence test","TaoKSPSetUseEW",tao->ksp_ewconv,&tao->ksp_ewconv,NULL)); 5300f0abf79SStefano Zampini PetscCall(TaoKSPSetUseEW(tao,tao->ksp_ewconv)); 5310f0abf79SStefano Zampini } 5320f0abf79SStefano Zampini 533*1baa6e33SBarry Smith if (tao->linesearch) PetscCall(TaoLineSearchSetFromOptions(tao->linesearch)); 534a82e8c82SStefano Zampini 535*1baa6e33SBarry Smith if (tao->ops->setfromoptions) PetscCall((*tao->ops->setfromoptions)(PetscOptionsObject,tao)); 536a7e14dcfSSatish Balay } 537d0609cedSBarry Smith PetscOptionsEnd(); 538a7e14dcfSSatish Balay PetscFunctionReturn(0); 539a7e14dcfSSatish Balay } 540a7e14dcfSSatish Balay 541a7e14dcfSSatish Balay /*@C 542fe2efc57SMark TaoViewFromOptions - View from Options 543fe2efc57SMark 544fe2efc57SMark Collective on Tao 545fe2efc57SMark 546fe2efc57SMark Input Parameters: 547fe2efc57SMark + A - the Tao context 548736c3998SJose E. Roman . obj - Optional object 549736c3998SJose E. Roman - name - command line option 550fe2efc57SMark 551fe2efc57SMark Level: intermediate 552db781477SPatrick Sanan .seealso: `Tao`, `TaoView`, `PetscObjectViewFromOptions()`, `TaoCreate()` 553fe2efc57SMark @*/ 554fe2efc57SMark PetscErrorCode TaoViewFromOptions(Tao A,PetscObject obj,const char name[]) 555fe2efc57SMark { 556fe2efc57SMark PetscFunctionBegin; 557fe2efc57SMark PetscValidHeaderSpecific(A,TAO_CLASSID,1); 5589566063dSJacob Faibussowitsch PetscCall(PetscObjectViewFromOptions((PetscObject)A,obj,name)); 559fe2efc57SMark PetscFunctionReturn(0); 560fe2efc57SMark } 561fe2efc57SMark 562fe2efc57SMark /*@C 563441846f8SBarry Smith TaoView - Prints information about the Tao 564a7e14dcfSSatish Balay 565441846f8SBarry Smith Collective on Tao 566a7e14dcfSSatish Balay 567a7e14dcfSSatish Balay InputParameters: 568441846f8SBarry Smith + tao - the Tao context 569a7e14dcfSSatish Balay - viewer - visualization context 570a7e14dcfSSatish Balay 571a7e14dcfSSatish Balay Options Database Key: 572a7e14dcfSSatish Balay . -tao_view - Calls TaoView() at the end of TaoSolve() 573a7e14dcfSSatish Balay 574a7e14dcfSSatish Balay Notes: 575a7e14dcfSSatish Balay The available visualization contexts include 576a7e14dcfSSatish Balay + PETSC_VIEWER_STDOUT_SELF - standard output (default) 577a7e14dcfSSatish Balay - PETSC_VIEWER_STDOUT_WORLD - synchronized standard 578a7e14dcfSSatish Balay output where only the first processor opens 579a7e14dcfSSatish Balay the file. All other processors send their 580a7e14dcfSSatish Balay data to the first processor to print. 581a7e14dcfSSatish Balay 582a7e14dcfSSatish Balay Level: beginner 583a7e14dcfSSatish Balay 584db781477SPatrick Sanan .seealso: `PetscViewerASCIIOpen()` 585a7e14dcfSSatish Balay @*/ 586441846f8SBarry Smith PetscErrorCode TaoView(Tao tao, PetscViewer viewer) 587a7e14dcfSSatish Balay { 588a7e14dcfSSatish Balay PetscBool isascii,isstring; 589b625d6c7SJed Brown TaoType type; 59047a47007SBarry Smith 591a7e14dcfSSatish Balay PetscFunctionBegin; 592441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 593a7e14dcfSSatish Balay if (!viewer) { 5949566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetStdout(((PetscObject)tao)->comm,&viewer)); 595a7e14dcfSSatish Balay } 596a7e14dcfSSatish Balay PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 597a7e14dcfSSatish Balay PetscCheckSameComm(tao,1,viewer,2); 598a7e14dcfSSatish Balay 5999566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii)); 6009566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring)); 601a7e14dcfSSatish Balay if (isascii) { 6029566063dSJacob Faibussowitsch PetscCall(PetscObjectPrintClassNamePrefixType((PetscObject)tao,viewer)); 603a7e14dcfSSatish Balay 604a7e14dcfSSatish Balay if (tao->ops->view) { 6059566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 6069566063dSJacob Faibussowitsch PetscCall((*tao->ops->view)(tao,viewer)); 6079566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 608a7e14dcfSSatish Balay } 6098d3f3ddaSAlp Dener if (tao->linesearch) { 6109566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 6119566063dSJacob Faibussowitsch PetscCall(TaoLineSearchView(tao->linesearch,viewer)); 6129566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 6138d3f3ddaSAlp Dener } 614a7e14dcfSSatish Balay if (tao->ksp) { 6159566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 6169566063dSJacob Faibussowitsch PetscCall(KSPView(tao->ksp,viewer)); 61763a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"total KSP iterations: %" PetscInt_FMT "\n",tao->ksp_tot_its)); 6189566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 619a7e14dcfSSatish Balay } 62019b5c101SAlp Dener 6219566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPushTab(viewer)); 62219b5c101SAlp Dener 623a7e14dcfSSatish Balay if (tao->XL || tao->XU) { 6249566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"Active Set subset type: %s\n",TaoSubSetTypes[tao->subset_type])); 625a7e14dcfSSatish Balay } 626a7e14dcfSSatish Balay 6279566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: gatol=%g,",(double)tao->gatol)); 6289566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," steptol=%g,",(double)tao->steptol)); 6299566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," gttol=%g\n",(double)tao->gttol)); 6309566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"Residual in Function/Gradient:=%g\n",(double)tao->residual)); 631a7e14dcfSSatish Balay 6326246e8cdSAlp Dener if (tao->constrained) { 6339566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances:")); 6349566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," catol=%g,",(double)tao->catol)); 6359566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," crtol=%g\n",(double)tao->crtol)); 6369566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"Residual in Constraints:=%g\n",(double)tao->cnorm)); 637a7e14dcfSSatish Balay } 638a7e14dcfSSatish Balay 639a7e14dcfSSatish Balay if (tao->trust < tao->steptol) { 6409566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: steptol=%g\n",(double)tao->steptol)); 6419566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"Final trust region radius:=%g\n",(double)tao->trust)); 642a7e14dcfSSatish Balay } 643a7e14dcfSSatish Balay 644a7e14dcfSSatish Balay if (tao->fmin>-1.e25) { 6459566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"convergence tolerances: function minimum=%g\n",(double)tao->fmin)); 646a7e14dcfSSatish Balay } 6479566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"Objective value=%g\n",(double)tao->fc)); 648a7e14dcfSSatish Balay 64963a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"total number of iterations=%" PetscInt_FMT ", ",tao->niter)); 65063a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (max: %" PetscInt_FMT ")\n",tao->max_it)); 651a7e14dcfSSatish Balay 652a7e14dcfSSatish Balay if (tao->nfuncs>0) { 65363a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"total number of function evaluations=%" PetscInt_FMT ",",tao->nfuncs)); 65463a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," max: %" PetscInt_FMT "\n",tao->max_funcs)); 655a7e14dcfSSatish Balay } 656a7e14dcfSSatish Balay if (tao->ngrads>0) { 65763a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"total number of gradient evaluations=%" PetscInt_FMT ",",tao->ngrads)); 65863a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," max: %" PetscInt_FMT "\n",tao->max_funcs)); 659a7e14dcfSSatish Balay } 660a7e14dcfSSatish Balay if (tao->nfuncgrads>0) { 66163a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"total number of function/gradient evaluations=%" PetscInt_FMT ",",tao->nfuncgrads)); 66263a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," (max: %" PetscInt_FMT ")\n",tao->max_funcs)); 663a7e14dcfSSatish Balay } 664a7e14dcfSSatish Balay if (tao->nhess>0) { 66563a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"total number of Hessian evaluations=%" PetscInt_FMT "\n",tao->nhess)); 666a7e14dcfSSatish Balay } 667a7e14dcfSSatish Balay if (tao->nconstraints>0) { 66863a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"total number of constraint function evaluations=%" PetscInt_FMT "\n",tao->nconstraints)); 669a7e14dcfSSatish Balay } 670a7e14dcfSSatish Balay if (tao->njac>0) { 67163a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"total number of Jacobian evaluations=%" PetscInt_FMT "\n",tao->njac)); 672a7e14dcfSSatish Balay } 673a7e14dcfSSatish Balay 674a7e14dcfSSatish Balay if (tao->reason>0) { 6759566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer, "Solution converged: ")); 676a7e14dcfSSatish Balay switch (tao->reason) { 677a7e14dcfSSatish Balay case TAO_CONVERGED_GATOL: 6789566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)|| <= gatol\n")); 679a7e14dcfSSatish Balay break; 680a7e14dcfSSatish Balay case TAO_CONVERGED_GRTOL: 6819566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)||/|f(X)| <= grtol\n")); 682a7e14dcfSSatish Balay break; 683a7e14dcfSSatish Balay case TAO_CONVERGED_GTTOL: 6849566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," ||g(X)||/||g(X0)|| <= gttol\n")); 685a7e14dcfSSatish Balay break; 686a7e14dcfSSatish Balay case TAO_CONVERGED_STEPTOL: 6879566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Steptol -- step size small\n")); 688a7e14dcfSSatish Balay break; 689a7e14dcfSSatish Balay case TAO_CONVERGED_MINF: 6909566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Minf -- f < fmin\n")); 691a7e14dcfSSatish Balay break; 692a7e14dcfSSatish Balay case TAO_CONVERGED_USER: 6939566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," User Terminated\n")); 694a7e14dcfSSatish Balay break; 695a7e14dcfSSatish Balay default: 6969566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"\n")); 697a7e14dcfSSatish Balay break; 698a7e14dcfSSatish Balay } 699a7e14dcfSSatish Balay } else { 7009566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"Solver terminated: %d",tao->reason)); 701a7e14dcfSSatish Balay switch (tao->reason) { 702a7e14dcfSSatish Balay case TAO_DIVERGED_MAXITS: 7039566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Maximum Iterations\n")); 704a7e14dcfSSatish Balay break; 705a7e14dcfSSatish Balay case TAO_DIVERGED_NAN: 7069566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," NAN or Inf encountered\n")); 707a7e14dcfSSatish Balay break; 708a7e14dcfSSatish Balay case TAO_DIVERGED_MAXFCN: 7099566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Maximum Function Evaluations\n")); 710a7e14dcfSSatish Balay break; 711a7e14dcfSSatish Balay case TAO_DIVERGED_LS_FAILURE: 7129566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Line Search Failure\n")); 713a7e14dcfSSatish Balay break; 714a7e14dcfSSatish Balay case TAO_DIVERGED_TR_REDUCTION: 7159566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Trust Region too small\n")); 716a7e14dcfSSatish Balay break; 717a7e14dcfSSatish Balay case TAO_DIVERGED_USER: 7189566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," User Terminated\n")); 719a7e14dcfSSatish Balay break; 720a7e14dcfSSatish Balay default: 7219566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"\n")); 722a7e14dcfSSatish Balay break; 723a7e14dcfSSatish Balay } 724a7e14dcfSSatish Balay } 7259566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPopTab(viewer)); 726a7e14dcfSSatish Balay } else if (isstring) { 7279566063dSJacob Faibussowitsch PetscCall(TaoGetType(tao,&type)); 7289566063dSJacob Faibussowitsch PetscCall(PetscViewerStringSPrintf(viewer," %-3.3s",type)); 729a7e14dcfSSatish Balay } 730a7e14dcfSSatish Balay PetscFunctionReturn(0); 731a7e14dcfSSatish Balay } 732a7e14dcfSSatish Balay 733a7e14dcfSSatish Balay /*@ 734414d97d3SAlp Dener TaoSetRecycleHistory - Sets the boolean flag to enable/disable re-using 735414d97d3SAlp Dener iterate information from the previous TaoSolve(). This feature is disabled by 736414d97d3SAlp Dener default. 737414d97d3SAlp Dener 738414d97d3SAlp Dener For conjugate gradient methods (BNCG), this re-uses the latest search direction 739414d97d3SAlp Dener from the previous TaoSolve() call when computing the first search direction in a 740414d97d3SAlp Dener new solution. By default, CG methods set the first search direction to the 741414d97d3SAlp Dener negative gradient. 742414d97d3SAlp Dener 743414d97d3SAlp Dener For quasi-Newton family of methods (BQNLS, BQNKLS, BQNKTR, BQNKTL), this re-uses 744414d97d3SAlp Dener the accumulated quasi-Newton Hessian approximation from the previous TaoSolve() 745414d97d3SAlp Dener call. By default, QN family of methods reset the initial Hessian approximation to 746414d97d3SAlp Dener the identity matrix. 747414d97d3SAlp Dener 748414d97d3SAlp Dener For any other algorithm, this setting has no effect. 749414d97d3SAlp Dener 750414d97d3SAlp Dener Logically collective on Tao 751414d97d3SAlp Dener 752414d97d3SAlp Dener Input Parameters: 753414d97d3SAlp Dener + tao - the Tao context 754414d97d3SAlp Dener - recycle - boolean flag 755414d97d3SAlp Dener 756414d97d3SAlp Dener Options Database Keys: 757147403d9SBarry Smith . -tao_recycle_history <true,false> - reuse the history 758414d97d3SAlp Dener 759414d97d3SAlp Dener Level: intermediate 760414d97d3SAlp Dener 761db781477SPatrick Sanan .seealso: `TaoSetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL` 762414d97d3SAlp Dener 763414d97d3SAlp Dener @*/ 764414d97d3SAlp Dener PetscErrorCode TaoSetRecycleHistory(Tao tao, PetscBool recycle) 765414d97d3SAlp Dener { 766414d97d3SAlp Dener PetscFunctionBegin; 767414d97d3SAlp Dener PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 76894511df7SStefano Zampini PetscValidLogicalCollectiveBool(tao,recycle,2); 769414d97d3SAlp Dener tao->recycle = recycle; 770414d97d3SAlp Dener PetscFunctionReturn(0); 771414d97d3SAlp Dener } 772414d97d3SAlp Dener 773414d97d3SAlp Dener /*@ 774414d97d3SAlp Dener TaoGetRecycleHistory - Retrieve the boolean flag for re-using iterate information 775414d97d3SAlp Dener from the previous TaoSolve(). This feature is disabled by default. 776414d97d3SAlp Dener 777414d97d3SAlp Dener Logically collective on Tao 778414d97d3SAlp Dener 779414d97d3SAlp Dener Input Parameters: 780147403d9SBarry Smith . tao - the Tao context 781414d97d3SAlp Dener 782414d97d3SAlp Dener Output Parameters: 783147403d9SBarry Smith . recycle - boolean flag 784414d97d3SAlp Dener 785414d97d3SAlp Dener Level: intermediate 786414d97d3SAlp Dener 787db781477SPatrick Sanan .seealso: `TaoGetRecycleHistory()`, `TAOBNCG`, `TAOBQNLS`, `TAOBQNKLS`, `TAOBQNKTR`, `TAOBQNKTL` 788414d97d3SAlp Dener 789414d97d3SAlp Dener @*/ 790414d97d3SAlp Dener PetscErrorCode TaoGetRecycleHistory(Tao tao, PetscBool *recycle) 791414d97d3SAlp Dener { 792414d97d3SAlp Dener PetscFunctionBegin; 793414d97d3SAlp Dener PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 79494511df7SStefano Zampini PetscValidBoolPointer(recycle,2); 795414d97d3SAlp Dener *recycle = tao->recycle; 796414d97d3SAlp Dener PetscFunctionReturn(0); 797414d97d3SAlp Dener } 798414d97d3SAlp Dener 799414d97d3SAlp Dener /*@ 800a7e14dcfSSatish Balay TaoSetTolerances - Sets parameters used in TAO convergence tests 801a7e14dcfSSatish Balay 802441846f8SBarry Smith Logically collective on Tao 803a7e14dcfSSatish Balay 804a7e14dcfSSatish Balay Input Parameters: 805441846f8SBarry Smith + tao - the Tao context 806a7e14dcfSSatish Balay . gatol - stop if norm of gradient is less than this 807a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this 808a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by this factor 809a7e14dcfSSatish Balay 810a7e14dcfSSatish Balay Options Database Keys: 811e52336cbSBarry Smith + -tao_gatol <gatol> - Sets gatol 812a7e14dcfSSatish Balay . -tao_grtol <grtol> - Sets grtol 813a7e14dcfSSatish Balay - -tao_gttol <gttol> - Sets gttol 814a7e14dcfSSatish Balay 815a7e14dcfSSatish Balay Stopping Criteria: 816a7e14dcfSSatish Balay $ ||g(X)|| <= gatol 817a7e14dcfSSatish Balay $ ||g(X)|| / |f(X)| <= grtol 818a7e14dcfSSatish Balay $ ||g(X)|| / ||g(X0)|| <= gttol 819a7e14dcfSSatish Balay 820a7e14dcfSSatish Balay Notes: 821a7e14dcfSSatish Balay Use PETSC_DEFAULT to leave one or more tolerances unchanged. 822a7e14dcfSSatish Balay 823a7e14dcfSSatish Balay Level: beginner 824a7e14dcfSSatish Balay 825db781477SPatrick Sanan .seealso: `TaoGetTolerances()` 826a7e14dcfSSatish Balay 827a7e14dcfSSatish Balay @*/ 828e52336cbSBarry Smith PetscErrorCode TaoSetTolerances(Tao tao, PetscReal gatol, PetscReal grtol, PetscReal gttol) 829a7e14dcfSSatish Balay { 830a7e14dcfSSatish Balay PetscFunctionBegin; 831441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 83294511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao,gatol,2); 83394511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao,grtol,3); 83494511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao,gttol,4); 835a7e14dcfSSatish Balay 836a7e14dcfSSatish Balay if (gatol != PETSC_DEFAULT) { 837a7e14dcfSSatish Balay if (gatol<0) { 8389566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao,"Tried to set negative gatol -- ignored.\n")); 839a7e14dcfSSatish Balay } else { 840a7e14dcfSSatish Balay tao->gatol = PetscMax(0,gatol); 8416552cf8aSJason Sarich tao->gatol_changed = PETSC_TRUE; 842a7e14dcfSSatish Balay } 843a7e14dcfSSatish Balay } 844a7e14dcfSSatish Balay 845a7e14dcfSSatish Balay if (grtol != PETSC_DEFAULT) { 846a7e14dcfSSatish Balay if (grtol<0) { 8479566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao,"Tried to set negative grtol -- ignored.\n")); 848a7e14dcfSSatish Balay } else { 849a7e14dcfSSatish Balay tao->grtol = PetscMax(0,grtol); 8506552cf8aSJason Sarich tao->grtol_changed = PETSC_TRUE; 851a7e14dcfSSatish Balay } 852a7e14dcfSSatish Balay } 853a7e14dcfSSatish Balay 854a7e14dcfSSatish Balay if (gttol != PETSC_DEFAULT) { 855a7e14dcfSSatish Balay if (gttol<0) { 8569566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao,"Tried to set negative gttol -- ignored.\n")); 857a7e14dcfSSatish Balay } else { 858a7e14dcfSSatish Balay tao->gttol = PetscMax(0,gttol); 8596552cf8aSJason Sarich tao->gttol_changed = PETSC_TRUE; 860a7e14dcfSSatish Balay } 861a7e14dcfSSatish Balay } 862a7e14dcfSSatish Balay PetscFunctionReturn(0); 863a7e14dcfSSatish Balay } 864a7e14dcfSSatish Balay 865a7e14dcfSSatish Balay /*@ 86658417fe7SBarry Smith TaoSetConstraintTolerances - Sets constraint tolerance parameters used in TAO convergence tests 867a7e14dcfSSatish Balay 868441846f8SBarry Smith Logically collective on Tao 869a7e14dcfSSatish Balay 870a7e14dcfSSatish Balay Input Parameters: 871441846f8SBarry Smith + tao - the Tao context 872e52336cbSBarry Smith . catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria 8736aad120cSJose E. Roman - crtol - relative constraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria 874a7e14dcfSSatish Balay 875a7e14dcfSSatish Balay Options Database Keys: 876a7e14dcfSSatish Balay + -tao_catol <catol> - Sets catol 877a7e14dcfSSatish Balay - -tao_crtol <crtol> - Sets crtol 878a7e14dcfSSatish Balay 8796552cf8aSJason Sarich Notes: 8806552cf8aSJason Sarich Use PETSC_DEFAULT to leave any tolerance unchanged. 8816552cf8aSJason Sarich 882a7e14dcfSSatish Balay Level: intermediate 883a7e14dcfSSatish Balay 884db781477SPatrick Sanan .seealso: `TaoGetTolerances()`, `TaoGetConstraintTolerances()`, `TaoSetTolerances()` 885a7e14dcfSSatish Balay 886a7e14dcfSSatish Balay @*/ 887441846f8SBarry Smith PetscErrorCode TaoSetConstraintTolerances(Tao tao, PetscReal catol, PetscReal crtol) 888a7e14dcfSSatish Balay { 889a7e14dcfSSatish Balay PetscFunctionBegin; 890441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 89194511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao,catol,2); 89294511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao,crtol,3); 893a7e14dcfSSatish Balay 894a7e14dcfSSatish Balay if (catol != PETSC_DEFAULT) { 895a7e14dcfSSatish Balay if (catol<0) { 8969566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao,"Tried to set negative catol -- ignored.\n")); 897a7e14dcfSSatish Balay } else { 898a7e14dcfSSatish Balay tao->catol = PetscMax(0,catol); 8996552cf8aSJason Sarich tao->catol_changed = PETSC_TRUE; 900a7e14dcfSSatish Balay } 901a7e14dcfSSatish Balay } 902a7e14dcfSSatish Balay 903a7e14dcfSSatish Balay if (crtol != PETSC_DEFAULT) { 904a7e14dcfSSatish Balay if (crtol<0) { 9059566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao,"Tried to set negative crtol -- ignored.\n")); 906a7e14dcfSSatish Balay } else { 907a7e14dcfSSatish Balay tao->crtol = PetscMax(0,crtol); 9086552cf8aSJason Sarich tao->crtol_changed = PETSC_TRUE; 909a7e14dcfSSatish Balay } 910a7e14dcfSSatish Balay } 911a7e14dcfSSatish Balay PetscFunctionReturn(0); 912a7e14dcfSSatish Balay } 913a7e14dcfSSatish Balay 91458417fe7SBarry Smith /*@ 91558417fe7SBarry Smith TaoGetConstraintTolerances - Gets constraint tolerance parameters used in TAO convergence tests 91658417fe7SBarry Smith 91758417fe7SBarry Smith Not ollective 91858417fe7SBarry Smith 91958417fe7SBarry Smith Input Parameter: 92058417fe7SBarry Smith . tao - the Tao context 92158417fe7SBarry Smith 922d8d19677SJose E. Roman Output Parameters: 923e52336cbSBarry Smith + catol - absolute constraint tolerance, constraint norm must be less than catol for used for gatol convergence criteria 9246aad120cSJose E. Roman - crtol - relative constraint tolerance, constraint norm must be less than crtol for used for gatol, gttol convergence criteria 92558417fe7SBarry Smith 92658417fe7SBarry Smith Level: intermediate 92758417fe7SBarry Smith 928db781477SPatrick Sanan .seealso: `TaoGetTolerances()`, `TaoSetTolerances()`, `TaoSetConstraintTolerances()` 92958417fe7SBarry Smith 93058417fe7SBarry Smith @*/ 93158417fe7SBarry Smith PetscErrorCode TaoGetConstraintTolerances(Tao tao, PetscReal *catol, PetscReal *crtol) 93258417fe7SBarry Smith { 93358417fe7SBarry Smith PetscFunctionBegin; 93458417fe7SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 93558417fe7SBarry Smith if (catol) *catol = tao->catol; 93658417fe7SBarry Smith if (crtol) *crtol = tao->crtol; 93758417fe7SBarry Smith PetscFunctionReturn(0); 93858417fe7SBarry Smith } 93958417fe7SBarry Smith 940a7e14dcfSSatish Balay /*@ 941a7e14dcfSSatish Balay TaoSetFunctionLowerBound - Sets a bound on the solution objective value. 942a7e14dcfSSatish Balay When an approximate solution with an objective value below this number 943a7e14dcfSSatish Balay has been found, the solver will terminate. 944a7e14dcfSSatish Balay 945441846f8SBarry Smith Logically Collective on Tao 946a7e14dcfSSatish Balay 947a7e14dcfSSatish Balay Input Parameters: 948441846f8SBarry Smith + tao - the Tao solver context 949a7e14dcfSSatish Balay - fmin - the tolerance 950a7e14dcfSSatish Balay 951a7e14dcfSSatish Balay Options Database Keys: 952a7e14dcfSSatish Balay . -tao_fmin <fmin> - sets the minimum function value 953a7e14dcfSSatish Balay 954a7e14dcfSSatish Balay Level: intermediate 955a7e14dcfSSatish Balay 956db781477SPatrick Sanan .seealso: `TaoSetTolerances()` 957a7e14dcfSSatish Balay @*/ 958441846f8SBarry Smith PetscErrorCode TaoSetFunctionLowerBound(Tao tao,PetscReal fmin) 959a7e14dcfSSatish Balay { 960a7e14dcfSSatish Balay PetscFunctionBegin; 961441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 96294511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao,fmin,2); 963a7e14dcfSSatish Balay tao->fmin = fmin; 9646552cf8aSJason Sarich tao->fmin_changed = PETSC_TRUE; 965a7e14dcfSSatish Balay PetscFunctionReturn(0); 966a7e14dcfSSatish Balay } 967a7e14dcfSSatish Balay 968a7e14dcfSSatish Balay /*@ 9698e96d397SJason Sarich TaoGetFunctionLowerBound - Gets the bound on the solution objective value. 970a7e14dcfSSatish Balay When an approximate solution with an objective value below this number 971a7e14dcfSSatish Balay has been found, the solver will terminate. 972a7e14dcfSSatish Balay 973441846f8SBarry Smith Not collective on Tao 974a7e14dcfSSatish Balay 975a7e14dcfSSatish Balay Input Parameters: 976441846f8SBarry Smith . tao - the Tao solver context 977a7e14dcfSSatish Balay 978a7e14dcfSSatish Balay OutputParameters: 979a7e14dcfSSatish Balay . fmin - the minimum function value 980a7e14dcfSSatish Balay 981a7e14dcfSSatish Balay Level: intermediate 982a7e14dcfSSatish Balay 983db781477SPatrick Sanan .seealso: `TaoSetFunctionLowerBound()` 984a7e14dcfSSatish Balay @*/ 985441846f8SBarry Smith PetscErrorCode TaoGetFunctionLowerBound(Tao tao,PetscReal *fmin) 986a7e14dcfSSatish Balay { 987a7e14dcfSSatish Balay PetscFunctionBegin; 988441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 98994511df7SStefano Zampini PetscValidRealPointer(fmin,2); 990a7e14dcfSSatish Balay *fmin = tao->fmin; 991a7e14dcfSSatish Balay PetscFunctionReturn(0); 992a7e14dcfSSatish Balay } 993a7e14dcfSSatish Balay 994a7e14dcfSSatish Balay /*@ 995a7e14dcfSSatish Balay TaoSetMaximumFunctionEvaluations - Sets a maximum number of 996a7e14dcfSSatish Balay function evaluations. 997a7e14dcfSSatish Balay 998441846f8SBarry Smith Logically Collective on Tao 999a7e14dcfSSatish Balay 1000a7e14dcfSSatish Balay Input Parameters: 1001441846f8SBarry Smith + tao - the Tao solver context 1002a7e14dcfSSatish Balay - nfcn - the maximum number of function evaluations (>=0) 1003a7e14dcfSSatish Balay 1004a7e14dcfSSatish Balay Options Database Keys: 1005a7e14dcfSSatish Balay . -tao_max_funcs <nfcn> - sets the maximum number of function evaluations 1006a7e14dcfSSatish Balay 1007a7e14dcfSSatish Balay Level: intermediate 1008a7e14dcfSSatish Balay 1009db781477SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoSetMaximumIterations()` 1010a7e14dcfSSatish Balay @*/ 1011a7e14dcfSSatish Balay 1012441846f8SBarry Smith PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao tao,PetscInt nfcn) 1013a7e14dcfSSatish Balay { 1014a7e14dcfSSatish Balay PetscFunctionBegin; 1015441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 101694511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao,nfcn,2); 101794511df7SStefano Zampini if (nfcn >= 0) { tao->max_funcs = PetscMax(0,nfcn); } 101894511df7SStefano Zampini else { tao->max_funcs = -1; } 10196552cf8aSJason Sarich tao->max_funcs_changed = PETSC_TRUE; 1020a7e14dcfSSatish Balay PetscFunctionReturn(0); 1021a7e14dcfSSatish Balay } 1022a7e14dcfSSatish Balay 1023a7e14dcfSSatish Balay /*@ 1024a7e14dcfSSatish Balay TaoGetMaximumFunctionEvaluations - Sets a maximum number of 1025a7e14dcfSSatish Balay function evaluations. 1026a7e14dcfSSatish Balay 1027a7e14dcfSSatish Balay Not Collective 1028a7e14dcfSSatish Balay 1029a7e14dcfSSatish Balay Input Parameters: 1030441846f8SBarry Smith . tao - the Tao solver context 1031a7e14dcfSSatish Balay 1032a7e14dcfSSatish Balay Output Parameters: 1033a7e14dcfSSatish Balay . nfcn - the maximum number of function evaluations 1034a7e14dcfSSatish Balay 1035a7e14dcfSSatish Balay Level: intermediate 1036a7e14dcfSSatish Balay 1037db781477SPatrick Sanan .seealso: `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()` 1038a7e14dcfSSatish Balay @*/ 1039a7e14dcfSSatish Balay 1040441846f8SBarry Smith PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao tao,PetscInt *nfcn) 1041a7e14dcfSSatish Balay { 1042a7e14dcfSSatish Balay PetscFunctionBegin; 1043441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 104494511df7SStefano Zampini PetscValidIntPointer(nfcn,2); 1045a7e14dcfSSatish Balay *nfcn = tao->max_funcs; 1046a7e14dcfSSatish Balay PetscFunctionReturn(0); 1047a7e14dcfSSatish Balay } 1048a7e14dcfSSatish Balay 1049770232b9SCe Qin /*@ 1050770232b9SCe Qin TaoGetCurrentFunctionEvaluations - Get current number of 1051770232b9SCe Qin function evaluations. 1052770232b9SCe Qin 1053770232b9SCe Qin Not Collective 1054770232b9SCe Qin 1055770232b9SCe Qin Input Parameters: 1056770232b9SCe Qin . tao - the Tao solver context 1057770232b9SCe Qin 1058770232b9SCe Qin Output Parameters: 105994511df7SStefano Zampini . nfuncs - the current number of function evaluations (maximum between gradient and function evaluations) 1060770232b9SCe Qin 1061770232b9SCe Qin Level: intermediate 1062770232b9SCe Qin 1063db781477SPatrick Sanan .seealso: `TaoSetMaximumFunctionEvaluations()`, `TaoGetMaximumFunctionEvaluations()`, `TaoGetMaximumIterations()` 1064770232b9SCe Qin @*/ 1065770232b9SCe Qin 1066770232b9SCe Qin PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao tao,PetscInt *nfuncs) 1067770232b9SCe Qin { 1068770232b9SCe Qin PetscFunctionBegin; 1069770232b9SCe Qin PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 107094511df7SStefano Zampini PetscValidIntPointer(nfuncs,2); 1071770232b9SCe Qin *nfuncs = PetscMax(tao->nfuncs,tao->nfuncgrads); 1072770232b9SCe Qin PetscFunctionReturn(0); 1073770232b9SCe Qin } 1074770232b9SCe Qin 1075a7e14dcfSSatish Balay /*@ 1076a7e14dcfSSatish Balay TaoSetMaximumIterations - Sets a maximum number of iterates. 1077a7e14dcfSSatish Balay 1078441846f8SBarry Smith Logically Collective on Tao 1079a7e14dcfSSatish Balay 1080a7e14dcfSSatish Balay Input Parameters: 1081441846f8SBarry Smith + tao - the Tao solver context 1082a7e14dcfSSatish Balay - maxits - the maximum number of iterates (>=0) 1083a7e14dcfSSatish Balay 1084a7e14dcfSSatish Balay Options Database Keys: 1085a7e14dcfSSatish Balay . -tao_max_it <its> - sets the maximum number of iterations 1086a7e14dcfSSatish Balay 1087a7e14dcfSSatish Balay Level: intermediate 1088a7e14dcfSSatish Balay 1089db781477SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoSetMaximumFunctionEvaluations()` 1090a7e14dcfSSatish Balay @*/ 1091441846f8SBarry Smith PetscErrorCode TaoSetMaximumIterations(Tao tao,PetscInt maxits) 1092a7e14dcfSSatish Balay { 1093a7e14dcfSSatish Balay PetscFunctionBegin; 1094441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 109594511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao,maxits,2); 109647a47007SBarry Smith tao->max_it = PetscMax(0,maxits); 10976552cf8aSJason Sarich tao->max_it_changed = PETSC_TRUE; 1098a7e14dcfSSatish Balay PetscFunctionReturn(0); 1099a7e14dcfSSatish Balay } 1100a7e14dcfSSatish Balay 1101a7e14dcfSSatish Balay /*@ 1102a7e14dcfSSatish Balay TaoGetMaximumIterations - Sets a maximum number of iterates. 1103a7e14dcfSSatish Balay 1104a7e14dcfSSatish Balay Not Collective 1105a7e14dcfSSatish Balay 1106a7e14dcfSSatish Balay Input Parameters: 1107441846f8SBarry Smith . tao - the Tao solver context 1108a7e14dcfSSatish Balay 1109a7e14dcfSSatish Balay Output Parameters: 1110a7e14dcfSSatish Balay . maxits - the maximum number of iterates 1111a7e14dcfSSatish Balay 1112a7e14dcfSSatish Balay Level: intermediate 1113a7e14dcfSSatish Balay 1114db781477SPatrick Sanan .seealso: `TaoSetMaximumIterations()`, `TaoGetMaximumFunctionEvaluations()` 1115a7e14dcfSSatish Balay @*/ 1116441846f8SBarry Smith PetscErrorCode TaoGetMaximumIterations(Tao tao,PetscInt *maxits) 1117a7e14dcfSSatish Balay { 1118a7e14dcfSSatish Balay PetscFunctionBegin; 1119441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 112094511df7SStefano Zampini PetscValidIntPointer(maxits,2); 1121a7e14dcfSSatish Balay *maxits = tao->max_it; 1122a7e14dcfSSatish Balay PetscFunctionReturn(0); 1123a7e14dcfSSatish Balay } 1124a7e14dcfSSatish Balay 1125a7e14dcfSSatish Balay /*@ 1126a7e14dcfSSatish Balay TaoSetInitialTrustRegionRadius - Sets the initial trust region radius. 1127a7e14dcfSSatish Balay 1128441846f8SBarry Smith Logically collective on Tao 1129a7e14dcfSSatish Balay 1130d8d19677SJose E. Roman Input Parameters: 1131a7e14dcfSSatish Balay + tao - a TAO optimization solver 1132a7e14dcfSSatish Balay - radius - the trust region radius 1133a7e14dcfSSatish Balay 1134a7e14dcfSSatish Balay Level: intermediate 1135a7e14dcfSSatish Balay 1136a7e14dcfSSatish Balay Options Database Key: 1137a7e14dcfSSatish Balay . -tao_trust0 <t0> - sets initial trust region radius 1138a7e14dcfSSatish Balay 1139db781477SPatrick Sanan .seealso: `TaoGetTrustRegionRadius()`, `TaoSetTrustRegionTolerance()` 1140a7e14dcfSSatish Balay @*/ 1141441846f8SBarry Smith PetscErrorCode TaoSetInitialTrustRegionRadius(Tao tao, PetscReal radius) 1142a7e14dcfSSatish Balay { 1143a7e14dcfSSatish Balay PetscFunctionBegin; 1144441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 114594511df7SStefano Zampini PetscValidLogicalCollectiveReal(tao,radius,2); 1146a7e14dcfSSatish Balay tao->trust0 = PetscMax(0.0,radius); 11476552cf8aSJason Sarich tao->trust0_changed = PETSC_TRUE; 1148a7e14dcfSSatish Balay PetscFunctionReturn(0); 1149a7e14dcfSSatish Balay } 1150a7e14dcfSSatish Balay 1151a7e14dcfSSatish Balay /*@ 1152a7e14dcfSSatish Balay TaoGetInitialTrustRegionRadius - Sets the initial trust region radius. 1153a7e14dcfSSatish Balay 1154a7e14dcfSSatish Balay Not Collective 1155a7e14dcfSSatish Balay 1156a7e14dcfSSatish Balay Input Parameter: 1157a7e14dcfSSatish Balay . tao - a TAO optimization solver 1158a7e14dcfSSatish Balay 1159a7e14dcfSSatish Balay Output Parameter: 1160a7e14dcfSSatish Balay . radius - the trust region radius 1161a7e14dcfSSatish Balay 1162a7e14dcfSSatish Balay Level: intermediate 1163a7e14dcfSSatish Balay 1164db781477SPatrick Sanan .seealso: `TaoSetInitialTrustRegionRadius()`, `TaoGetCurrentTrustRegionRadius()` 1165a7e14dcfSSatish Balay @*/ 1166441846f8SBarry Smith PetscErrorCode TaoGetInitialTrustRegionRadius(Tao tao, PetscReal *radius) 1167a7e14dcfSSatish Balay { 1168a7e14dcfSSatish Balay PetscFunctionBegin; 1169441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 117094511df7SStefano Zampini PetscValidRealPointer(radius,2); 1171a7e14dcfSSatish Balay *radius = tao->trust0; 1172a7e14dcfSSatish Balay PetscFunctionReturn(0); 1173a7e14dcfSSatish Balay } 1174a7e14dcfSSatish Balay 1175a7e14dcfSSatish Balay /*@ 1176a7e14dcfSSatish Balay TaoGetCurrentTrustRegionRadius - Gets the current trust region radius. 1177a7e14dcfSSatish Balay 1178a7e14dcfSSatish Balay Not Collective 1179a7e14dcfSSatish Balay 1180a7e14dcfSSatish Balay Input Parameter: 1181a7e14dcfSSatish Balay . tao - a TAO optimization solver 1182a7e14dcfSSatish Balay 1183a7e14dcfSSatish Balay Output Parameter: 1184a7e14dcfSSatish Balay . radius - the trust region radius 1185a7e14dcfSSatish Balay 1186a7e14dcfSSatish Balay Level: intermediate 1187a7e14dcfSSatish Balay 1188db781477SPatrick Sanan .seealso: `TaoSetInitialTrustRegionRadius()`, `TaoGetInitialTrustRegionRadius()` 1189a7e14dcfSSatish Balay @*/ 1190441846f8SBarry Smith PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao tao, PetscReal *radius) 1191a7e14dcfSSatish Balay { 1192a7e14dcfSSatish Balay PetscFunctionBegin; 1193441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 119494511df7SStefano Zampini PetscValidRealPointer(radius,2); 1195a7e14dcfSSatish Balay *radius = tao->trust; 1196a7e14dcfSSatish Balay PetscFunctionReturn(0); 1197a7e14dcfSSatish Balay } 1198a7e14dcfSSatish Balay 1199a7e14dcfSSatish Balay /*@ 1200a7e14dcfSSatish Balay TaoGetTolerances - gets the current values of tolerances 1201a7e14dcfSSatish Balay 1202a7e14dcfSSatish Balay Not Collective 1203a7e14dcfSSatish Balay 1204f899ff85SJose E. Roman Input Parameter: 1205441846f8SBarry Smith . tao - the Tao context 1206a7e14dcfSSatish Balay 1207a7e14dcfSSatish Balay Output Parameters: 1208e52336cbSBarry Smith + gatol - stop if norm of gradient is less than this 1209a7e14dcfSSatish Balay . grtol - stop if relative norm of gradient is less than this 1210a7e14dcfSSatish Balay - gttol - stop if norm of gradient is reduced by a this factor 1211a7e14dcfSSatish Balay 121247a47007SBarry Smith Note: NULL can be used as an argument if not all tolerances values are needed 1213a7e14dcfSSatish Balay 1214db781477SPatrick Sanan .seealso `TaoSetTolerances()` 1215a7e14dcfSSatish Balay 1216a7e14dcfSSatish Balay Level: intermediate 1217a7e14dcfSSatish Balay @*/ 1218e52336cbSBarry Smith PetscErrorCode TaoGetTolerances(Tao tao, PetscReal *gatol, PetscReal *grtol, PetscReal *gttol) 1219a7e14dcfSSatish Balay { 1220a7e14dcfSSatish Balay PetscFunctionBegin; 1221441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 1222a7e14dcfSSatish Balay if (gatol) *gatol = tao->gatol; 1223a7e14dcfSSatish Balay if (grtol) *grtol = tao->grtol; 1224a7e14dcfSSatish Balay if (gttol) *gttol = tao->gttol; 1225a7e14dcfSSatish Balay PetscFunctionReturn(0); 1226a7e14dcfSSatish Balay } 1227a7e14dcfSSatish Balay 1228a7e14dcfSSatish Balay /*@ 1229a7e14dcfSSatish Balay TaoGetKSP - Gets the linear solver used by the optimization solver. 1230a7e14dcfSSatish Balay Application writers should use TaoGetKSP if they need direct access 1231a7e14dcfSSatish Balay to the PETSc KSP object. 1232a7e14dcfSSatish Balay 1233a7e14dcfSSatish Balay Not Collective 1234a7e14dcfSSatish Balay 1235a7e14dcfSSatish Balay Input Parameters: 1236a7e14dcfSSatish Balay . tao - the TAO solver 1237a7e14dcfSSatish Balay 1238a7e14dcfSSatish Balay Output Parameters: 1239a7e14dcfSSatish Balay . ksp - the KSP linear solver used in the optimization solver 1240a7e14dcfSSatish Balay 1241a7e14dcfSSatish Balay Level: intermediate 1242a7e14dcfSSatish Balay 1243a7e14dcfSSatish Balay @*/ 1244441846f8SBarry Smith PetscErrorCode TaoGetKSP(Tao tao, KSP *ksp) 124547a47007SBarry Smith { 1246a7e14dcfSSatish Balay PetscFunctionBegin; 124794511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 124894511df7SStefano Zampini PetscValidPointer(ksp,2); 1249a7e14dcfSSatish Balay *ksp = tao->ksp; 1250a7e14dcfSSatish Balay PetscFunctionReturn(0); 1251a7e14dcfSSatish Balay } 1252a7e14dcfSSatish Balay 1253025e9500SJason Sarich /*@ 1254025e9500SJason Sarich TaoGetLinearSolveIterations - Gets the total number of linear iterations 1255025e9500SJason Sarich used by the TAO solver 1256025e9500SJason Sarich 1257025e9500SJason Sarich Not Collective 1258025e9500SJason Sarich 1259025e9500SJason Sarich Input Parameter: 1260025e9500SJason Sarich . tao - TAO context 1261025e9500SJason Sarich 1262025e9500SJason Sarich Output Parameter: 1263025e9500SJason Sarich . lits - number of linear iterations 1264025e9500SJason Sarich 1265025e9500SJason Sarich Notes: 1266025e9500SJason Sarich This counter is reset to zero for each successive call to TaoSolve() 1267025e9500SJason Sarich 1268025e9500SJason Sarich Level: intermediate 1269025e9500SJason Sarich 1270db781477SPatrick Sanan .seealso: `TaoGetKSP()` 1271025e9500SJason Sarich @*/ 1272025e9500SJason Sarich PetscErrorCode TaoGetLinearSolveIterations(Tao tao, PetscInt *lits) 1273025e9500SJason Sarich { 1274025e9500SJason Sarich PetscFunctionBegin; 1275025e9500SJason Sarich PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 1276025e9500SJason Sarich PetscValidIntPointer(lits,2); 12772d9aa51bSJason Sarich *lits = tao->ksp_tot_its; 1278025e9500SJason Sarich PetscFunctionReturn(0); 1279025e9500SJason Sarich } 1280025e9500SJason Sarich 1281a7e14dcfSSatish Balay /*@ 1282a7e14dcfSSatish Balay TaoGetLineSearch - Gets the line search used by the optimization solver. 1283a7e14dcfSSatish Balay Application writers should use TaoGetLineSearch if they need direct access 1284a7e14dcfSSatish Balay to the TaoLineSearch object. 1285a7e14dcfSSatish Balay 1286a7e14dcfSSatish Balay Not Collective 1287a7e14dcfSSatish Balay 1288a7e14dcfSSatish Balay Input Parameters: 1289a7e14dcfSSatish Balay . tao - the TAO solver 1290a7e14dcfSSatish Balay 1291a7e14dcfSSatish Balay Output Parameters: 1292a7e14dcfSSatish Balay . ls - the line search used in the optimization solver 1293a7e14dcfSSatish Balay 1294a7e14dcfSSatish Balay Level: intermediate 1295a7e14dcfSSatish Balay 1296a7e14dcfSSatish Balay @*/ 1297441846f8SBarry Smith PetscErrorCode TaoGetLineSearch(Tao tao, TaoLineSearch *ls) 129847a47007SBarry Smith { 1299a7e14dcfSSatish Balay PetscFunctionBegin; 130094511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 130194511df7SStefano Zampini PetscValidPointer(ls,2); 1302a7e14dcfSSatish Balay *ls = tao->linesearch; 1303a7e14dcfSSatish Balay PetscFunctionReturn(0); 1304a7e14dcfSSatish Balay } 1305a7e14dcfSSatish Balay 1306a7e14dcfSSatish Balay /*@ 1307a7e14dcfSSatish Balay TaoAddLineSearchCounts - Adds the number of function evaluations spent 1308a7e14dcfSSatish Balay in the line search to the running total. 1309a7e14dcfSSatish Balay 1310a7e14dcfSSatish Balay Input Parameters: 1311a7e14dcfSSatish Balay + tao - the TAO solver 1312a7e14dcfSSatish Balay - ls - the line search used in the optimization solver 1313a7e14dcfSSatish Balay 1314a7e14dcfSSatish Balay Level: developer 1315a7e14dcfSSatish Balay 1316db781477SPatrick Sanan .seealso: `TaoLineSearchApply()` 1317a7e14dcfSSatish Balay @*/ 1318441846f8SBarry Smith PetscErrorCode TaoAddLineSearchCounts(Tao tao) 131947a47007SBarry Smith { 1320a7e14dcfSSatish Balay PetscBool flg; 1321a7e14dcfSSatish Balay PetscInt nfeval,ngeval,nfgeval; 132247a47007SBarry Smith 1323a7e14dcfSSatish Balay PetscFunctionBegin; 1324441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 1325a7e14dcfSSatish Balay if (tao->linesearch) { 13269566063dSJacob Faibussowitsch PetscCall(TaoLineSearchIsUsingTaoRoutines(tao->linesearch,&flg)); 132794ae4db5SBarry Smith if (!flg) { 13289566063dSJacob Faibussowitsch PetscCall(TaoLineSearchGetNumberFunctionEvaluations(tao->linesearch,&nfeval,&ngeval,&nfgeval)); 1329a7e14dcfSSatish Balay tao->nfuncs += nfeval; 1330a7e14dcfSSatish Balay tao->ngrads += ngeval; 1331a7e14dcfSSatish Balay tao->nfuncgrads += nfgeval; 1332a7e14dcfSSatish Balay } 1333a7e14dcfSSatish Balay } 1334a7e14dcfSSatish Balay PetscFunctionReturn(0); 1335a7e14dcfSSatish Balay } 1336a7e14dcfSSatish Balay 1337a7e14dcfSSatish Balay /*@ 1338a82e8c82SStefano Zampini TaoGetSolution - Returns the vector with the current TAO solution 1339a7e14dcfSSatish Balay 1340a7e14dcfSSatish Balay Not Collective 1341a7e14dcfSSatish Balay 1342a7e14dcfSSatish Balay Input Parameter: 1343441846f8SBarry Smith . tao - the Tao context 1344a7e14dcfSSatish Balay 1345a7e14dcfSSatish Balay Output Parameter: 1346a7e14dcfSSatish Balay . X - the current solution 1347a7e14dcfSSatish Balay 1348a7e14dcfSSatish Balay Level: intermediate 1349a7e14dcfSSatish Balay 1350a82e8c82SStefano Zampini Note: The returned vector will be the same object that was passed into TaoSetSolution() 1351a7e14dcfSSatish Balay @*/ 1352a82e8c82SStefano Zampini PetscErrorCode TaoGetSolution(Tao tao, Vec *X) 1353a7e14dcfSSatish Balay { 1354a7e14dcfSSatish Balay PetscFunctionBegin; 1355441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 135694511df7SStefano Zampini PetscValidPointer(X,2); 1357a7e14dcfSSatish Balay *X = tao->solution; 1358a7e14dcfSSatish Balay PetscFunctionReturn(0); 1359a7e14dcfSSatish Balay } 1360a7e14dcfSSatish Balay 1361a7e14dcfSSatish Balay /*@ 1362a7e14dcfSSatish Balay TaoResetStatistics - Initialize the statistics used by TAO for all of the solvers. 1363a7e14dcfSSatish Balay These statistics include the iteration number, residual norms, and convergence status. 1364a7e14dcfSSatish Balay This routine gets called before solving each optimization problem. 1365a7e14dcfSSatish Balay 1366441846f8SBarry Smith Collective on Tao 1367a7e14dcfSSatish Balay 1368a7e14dcfSSatish Balay Input Parameters: 1369441846f8SBarry Smith . solver - the Tao context 1370a7e14dcfSSatish Balay 1371a7e14dcfSSatish Balay Level: developer 1372a7e14dcfSSatish Balay 1373db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoSolve()` 1374a7e14dcfSSatish Balay @*/ 1375441846f8SBarry Smith PetscErrorCode TaoResetStatistics(Tao tao) 1376a7e14dcfSSatish Balay { 1377a7e14dcfSSatish Balay PetscFunctionBegin; 1378441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 1379a7e14dcfSSatish Balay tao->niter = 0; 1380a7e14dcfSSatish Balay tao->nfuncs = 0; 1381a7e14dcfSSatish Balay tao->nfuncgrads = 0; 1382a7e14dcfSSatish Balay tao->ngrads = 0; 1383a7e14dcfSSatish Balay tao->nhess = 0; 1384a7e14dcfSSatish Balay tao->njac = 0; 1385a7e14dcfSSatish Balay tao->nconstraints = 0; 1386a7e14dcfSSatish Balay tao->ksp_its = 0; 1387ae93cb3cSJason Sarich tao->ksp_tot_its = 0; 1388a7e14dcfSSatish Balay tao->reason = TAO_CONTINUE_ITERATING; 1389a7e14dcfSSatish Balay tao->residual = 0.0; 1390a7e14dcfSSatish Balay tao->cnorm = 0.0; 1391a7e14dcfSSatish Balay tao->step = 0.0; 1392a7e14dcfSSatish Balay tao->lsflag = PETSC_FALSE; 1393a7e14dcfSSatish Balay if (tao->hist_reset) tao->hist_len = 0; 1394a7e14dcfSSatish Balay PetscFunctionReturn(0); 1395a7e14dcfSSatish Balay } 1396a7e14dcfSSatish Balay 1397a7e14dcfSSatish Balay /*@C 1398e1e80dc8SAlp Dener TaoSetUpdate - Sets the general-purpose update function called 1399e1e80dc8SAlp Dener at the beginning of every iteration of the nonlinear solve. Specifically 1400e1e80dc8SAlp Dener it is called at the top of every iteration, after the new solution and the gradient 1401e1e80dc8SAlp Dener is determined, but before the Hessian is computed (if applicable). 1402e1e80dc8SAlp Dener 1403e1e80dc8SAlp Dener Logically Collective on Tao 1404e1e80dc8SAlp Dener 1405e1e80dc8SAlp Dener Input Parameters: 1406a2b725a8SWilliam Gropp + tao - The tao solver context 1407a2b725a8SWilliam Gropp - func - The function 1408e1e80dc8SAlp Dener 1409e1e80dc8SAlp Dener Calling sequence of func: 1410a2b725a8SWilliam Gropp $ func (Tao tao, PetscInt step); 1411e1e80dc8SAlp Dener 1412e1e80dc8SAlp Dener . step - The current step of the iteration 1413e1e80dc8SAlp Dener 1414e1e80dc8SAlp Dener Level: advanced 1415e1e80dc8SAlp Dener 1416db781477SPatrick Sanan .seealso `TaoSolve()` 1417e1e80dc8SAlp Dener @*/ 14188fcddce6SStefano Zampini PetscErrorCode TaoSetUpdate(Tao tao, PetscErrorCode (*func)(Tao, PetscInt, void*), void *ctx) 1419e1e80dc8SAlp Dener { 1420e1e80dc8SAlp Dener PetscFunctionBegin; 1421e1e80dc8SAlp Dener PetscValidHeaderSpecific(tao, TAO_CLASSID,1); 1422e1e80dc8SAlp Dener tao->ops->update = func; 1423e1e80dc8SAlp Dener tao->user_update = ctx; 1424e1e80dc8SAlp Dener PetscFunctionReturn(0); 1425e1e80dc8SAlp Dener } 1426e1e80dc8SAlp Dener 1427e1e80dc8SAlp Dener /*@C 1428a7e14dcfSSatish Balay TaoSetConvergenceTest - Sets the function that is to be used to test 1429a7e14dcfSSatish Balay for convergence o fthe iterative minimization solution. The new convergence 1430a7e14dcfSSatish Balay testing routine will replace TAO's default convergence test. 1431a7e14dcfSSatish Balay 1432441846f8SBarry Smith Logically Collective on Tao 1433a7e14dcfSSatish Balay 1434a7e14dcfSSatish Balay Input Parameters: 1435441846f8SBarry Smith + tao - the Tao object 1436a7e14dcfSSatish Balay . conv - the routine to test for convergence 1437a7e14dcfSSatish Balay - ctx - [optional] context for private data for the convergence routine 143847a47007SBarry Smith (may be NULL) 1439a7e14dcfSSatish Balay 1440a7e14dcfSSatish Balay Calling sequence of conv: 1441441846f8SBarry Smith $ PetscErrorCode conv(Tao tao, void *ctx) 1442a7e14dcfSSatish Balay 1443441846f8SBarry Smith + tao - the Tao object 1444a7e14dcfSSatish Balay - ctx - [optional] convergence context 1445a7e14dcfSSatish Balay 1446e4cb33bbSBarry Smith Note: The new convergence testing routine should call TaoSetConvergedReason(). 1447a7e14dcfSSatish Balay 1448a7e14dcfSSatish Balay Level: advanced 1449a7e14dcfSSatish Balay 1450db781477SPatrick Sanan .seealso: `TaoSetConvergedReason()`, `TaoGetSolutionStatus()`, `TaoGetTolerances()`, `TaoSetMonitor` 1451a7e14dcfSSatish Balay 1452a7e14dcfSSatish Balay @*/ 1453441846f8SBarry Smith PetscErrorCode TaoSetConvergenceTest(Tao tao, PetscErrorCode (*conv)(Tao, void*), void *ctx) 1454a7e14dcfSSatish Balay { 1455a7e14dcfSSatish Balay PetscFunctionBegin; 1456441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 145794511df7SStefano Zampini tao->ops->convergencetest = conv; 145894511df7SStefano Zampini tao->cnvP = ctx; 1459a7e14dcfSSatish Balay PetscFunctionReturn(0); 1460a7e14dcfSSatish Balay } 1461a7e14dcfSSatish Balay 1462a7e14dcfSSatish Balay /*@C 1463a7e14dcfSSatish Balay TaoSetMonitor - Sets an ADDITIONAL function that is to be used at every 1464a7e14dcfSSatish Balay iteration of the solver to display the iteration's 1465a7e14dcfSSatish Balay progress. 1466a7e14dcfSSatish Balay 1467441846f8SBarry Smith Logically Collective on Tao 1468a7e14dcfSSatish Balay 1469a7e14dcfSSatish Balay Input Parameters: 1470441846f8SBarry Smith + tao - the Tao solver context 1471a7e14dcfSSatish Balay . mymonitor - monitoring routine 1472a7e14dcfSSatish Balay - mctx - [optional] user-defined context for private data for the 147347a47007SBarry Smith monitor routine (may be NULL) 1474a7e14dcfSSatish Balay 1475a7e14dcfSSatish Balay Calling sequence of mymonitor: 1476147403d9SBarry Smith .vb 1477147403d9SBarry Smith PetscErrorCode mymonitor(Tao tao,void *mctx) 1478147403d9SBarry Smith .ve 1479a7e14dcfSSatish Balay 1480441846f8SBarry Smith + tao - the Tao solver context 1481a7e14dcfSSatish Balay - mctx - [optional] monitoring context 1482a7e14dcfSSatish Balay 1483a7e14dcfSSatish Balay Options Database Keys: 148498ea980cSBarry Smith + -tao_monitor - sets TaoMonitorDefault() 1485a7e14dcfSSatish Balay . -tao_smonitor - sets short monitor 1486a7e14dcfSSatish Balay . -tao_cmonitor - same as smonitor plus constraint norm 1487a7e14dcfSSatish Balay . -tao_view_solution - view solution at each iteration 1488a7e14dcfSSatish Balay . -tao_view_gradient - view gradient at each iteration 14894a48860cSAlp Dener . -tao_view_ls_residual - view least-squares residual vector at each iteration 1490a7e14dcfSSatish Balay - -tao_cancelmonitors - cancels all monitors that have been hardwired into a code by calls to TaoSetMonitor(), but does not cancel those set via the options database. 1491a7e14dcfSSatish Balay 1492a7e14dcfSSatish Balay Notes: 1493a7e14dcfSSatish Balay Several different monitoring routines may be set by calling 1494a7e14dcfSSatish Balay TaoSetMonitor() multiple times; all will be called in the 1495a7e14dcfSSatish Balay order in which they were set. 1496a7e14dcfSSatish Balay 149795452b02SPatrick Sanan Fortran Notes: 149895452b02SPatrick Sanan Only one monitor function may be set 1499a7e14dcfSSatish Balay 1500a7e14dcfSSatish Balay Level: intermediate 1501a7e14dcfSSatish Balay 1502db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoCancelMonitors()`, `TaoSetDestroyRoutine()` 1503a7e14dcfSSatish Balay @*/ 1504441846f8SBarry Smith PetscErrorCode TaoSetMonitor(Tao tao, PetscErrorCode (*func)(Tao, void*), void *ctx,PetscErrorCode (*dest)(void**)) 1505a7e14dcfSSatish Balay { 150608d19d1fSJason Sarich PetscInt i; 15078732526dSAlp Dener PetscBool identical; 150808d19d1fSJason Sarich 1509a7e14dcfSSatish Balay PetscFunctionBegin; 1510441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 15113c859ba3SBarry Smith PetscCheck(tao->numbermonitors < MAXTAOMONITORS,PetscObjectComm((PetscObject)tao),PETSC_ERR_SUP,"Cannot attach another monitor -- max=%d",MAXTAOMONITORS); 151208d19d1fSJason Sarich 151308d19d1fSJason Sarich for (i=0; i<tao->numbermonitors;i++) { 15149566063dSJacob Faibussowitsch PetscCall(PetscMonitorCompare((PetscErrorCode (*)(void))func,ctx,dest,(PetscErrorCode (*)(void))tao->monitor[i],tao->monitorcontext[i],tao->monitordestroy[i],&identical)); 15158732526dSAlp Dener if (identical) PetscFunctionReturn(0); 151608d19d1fSJason Sarich } 1517a7e14dcfSSatish Balay tao->monitor[tao->numbermonitors] = func; 15188732526dSAlp Dener tao->monitorcontext[tao->numbermonitors] = (void*)ctx; 1519a7e14dcfSSatish Balay tao->monitordestroy[tao->numbermonitors] = dest; 1520a7e14dcfSSatish Balay ++tao->numbermonitors; 1521a7e14dcfSSatish Balay PetscFunctionReturn(0); 1522a7e14dcfSSatish Balay } 1523a7e14dcfSSatish Balay 1524a7e14dcfSSatish Balay /*@ 1525441846f8SBarry Smith TaoCancelMonitors - Clears all the monitor functions for a Tao object. 1526a7e14dcfSSatish Balay 1527441846f8SBarry Smith Logically Collective on Tao 1528a7e14dcfSSatish Balay 1529a7e14dcfSSatish Balay Input Parameters: 1530441846f8SBarry Smith . tao - the Tao solver context 1531a7e14dcfSSatish Balay 1532a7e14dcfSSatish Balay Options Database: 1533a7e14dcfSSatish Balay . -tao_cancelmonitors - cancels all monitors that have been hardwired 1534a7e14dcfSSatish Balay into a code by calls to TaoSetMonitor(), but does not cancel those 1535a7e14dcfSSatish Balay set via the options database 1536a7e14dcfSSatish Balay 1537a7e14dcfSSatish Balay Notes: 1538441846f8SBarry Smith There is no way to clear one specific monitor from a Tao object. 1539a7e14dcfSSatish Balay 1540a7e14dcfSSatish Balay Level: advanced 1541a7e14dcfSSatish Balay 1542db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()` 1543a7e14dcfSSatish Balay @*/ 1544441846f8SBarry Smith PetscErrorCode TaoCancelMonitors(Tao tao) 1545a7e14dcfSSatish Balay { 1546a7e14dcfSSatish Balay PetscInt i; 154747a47007SBarry Smith 1548a7e14dcfSSatish Balay PetscFunctionBegin; 1549441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 1550a7e14dcfSSatish Balay for (i=0;i<tao->numbermonitors;i++) { 1551a7e14dcfSSatish Balay if (tao->monitordestroy[i]) { 15529566063dSJacob Faibussowitsch PetscCall((*tao->monitordestroy[i])(&tao->monitorcontext[i])); 1553a7e14dcfSSatish Balay } 1554a7e14dcfSSatish Balay } 1555a7e14dcfSSatish Balay tao->numbermonitors = 0; 1556a7e14dcfSSatish Balay PetscFunctionReturn(0); 1557a7e14dcfSSatish Balay } 1558a7e14dcfSSatish Balay 15597fab98ebSJason Sarich /*@ 156098ea980cSBarry Smith TaoMonitorDefault - Default routine for monitoring progress of the 1561441846f8SBarry Smith Tao solvers (default). This monitor prints the function value and gradient 1562a7e14dcfSSatish Balay norm at each iteration. It can be turned on from the command line using the 1563a7e14dcfSSatish Balay -tao_monitor option 1564a7e14dcfSSatish Balay 1565441846f8SBarry Smith Collective on Tao 1566a7e14dcfSSatish Balay 1567a7e14dcfSSatish Balay Input Parameters: 1568441846f8SBarry Smith + tao - the Tao context 156947a47007SBarry Smith - ctx - PetscViewer context or NULL 1570a7e14dcfSSatish Balay 1571a7e14dcfSSatish Balay Options Database Keys: 1572147403d9SBarry Smith . -tao_monitor - turn on default monitoring 1573a7e14dcfSSatish Balay 1574a7e14dcfSSatish Balay Level: advanced 1575a7e14dcfSSatish Balay 1576db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1577a7e14dcfSSatish Balay @*/ 157898ea980cSBarry Smith PetscErrorCode TaoMonitorDefault(Tao tao, void *ctx) 1579a7e14dcfSSatish Balay { 158063b15415SAlp Dener PetscInt its, tabs; 1581a7e14dcfSSatish Balay PetscReal fct,gnorm; 15828163d661SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 158347a47007SBarry Smith 1584a7e14dcfSSatish Balay PetscFunctionBegin; 158594511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 15868163d661SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 1587a7e14dcfSSatish Balay its = tao->niter; 1588a7e14dcfSSatish Balay fct = tao->fc; 1589a7e14dcfSSatish Balay gnorm = tao->residual; 15909566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 15919566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 1592494bef23SAlp Dener if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) { 15939566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Iteration information for %s solve.\n",((PetscObject)tao)->prefix)); 1594494bef23SAlp Dener tao->header_printed = PETSC_TRUE; 159563b15415SAlp Dener } 159663a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"%3" PetscInt_FMT " TAO,",its)); 15979566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Function value: %g,",(double)fct)); 159808d19d1fSJason Sarich if (gnorm >= PETSC_INFINITY) { 15999566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Residual: Inf \n")); 160008d19d1fSJason Sarich } else { 16019566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Residual: %g \n",(double)gnorm)); 160208d19d1fSJason Sarich } 16039566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 1604a7e14dcfSSatish Balay PetscFunctionReturn(0); 1605a7e14dcfSSatish Balay } 1606a7e14dcfSSatish Balay 16077fab98ebSJason Sarich /*@ 16088d5ead36SAlp Dener TaoDefaultGMonitor - Default routine for monitoring progress of the 16098d5ead36SAlp Dener Tao solvers (default) with extra detail on the globalization method. 16108d5ead36SAlp Dener This monitor prints the function value and gradient norm at each 16118d5ead36SAlp Dener iteration, as well as the step size and trust radius. Note that the 16128d5ead36SAlp Dener step size and trust radius may be the same for some algorithms. 16138d5ead36SAlp Dener It can be turned on from the command line using the 16148d5ead36SAlp Dener -tao_gmonitor option 16158d5ead36SAlp Dener 16168d5ead36SAlp Dener Collective on Tao 16178d5ead36SAlp Dener 16188d5ead36SAlp Dener Input Parameters: 16198d5ead36SAlp Dener + tao - the Tao context 16208d5ead36SAlp Dener - ctx - PetscViewer context or NULL 16218d5ead36SAlp Dener 16228d5ead36SAlp Dener Options Database Keys: 1623147403d9SBarry Smith . -tao_gmonitor - turn on monitoring with globalization information 16248d5ead36SAlp Dener 16258d5ead36SAlp Dener Level: advanced 16268d5ead36SAlp Dener 1627db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()` 16288d5ead36SAlp Dener @*/ 16298d5ead36SAlp Dener PetscErrorCode TaoDefaultGMonitor(Tao tao, void *ctx) 16308d5ead36SAlp Dener { 16318d5ead36SAlp Dener PetscInt its, tabs; 16328d5ead36SAlp Dener PetscReal fct,gnorm,stp,tr; 16338d5ead36SAlp Dener PetscViewer viewer = (PetscViewer)ctx; 16348d5ead36SAlp Dener 16358d5ead36SAlp Dener PetscFunctionBegin; 163694511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 16378d5ead36SAlp Dener PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 16388d5ead36SAlp Dener its = tao->niter; 16398d5ead36SAlp Dener fct = tao->fc; 16408d5ead36SAlp Dener gnorm = tao->residual; 16418d5ead36SAlp Dener stp = tao->step; 16428d5ead36SAlp Dener tr = tao->trust; 16439566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 16449566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 1645494bef23SAlp Dener if (its == 0 && ((PetscObject)tao)->prefix && !tao->header_printed) { 16469566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Iteration information for %s solve.\n",((PetscObject)tao)->prefix)); 1647494bef23SAlp Dener tao->header_printed = PETSC_TRUE; 16488d5ead36SAlp Dener } 164963a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"%3" PetscInt_FMT " TAO,",its)); 16509566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Function value: %g,",(double)fct)); 16518d5ead36SAlp Dener if (gnorm >= PETSC_INFINITY) { 16529566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Residual: Inf,")); 16538d5ead36SAlp Dener } else { 16549566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Residual: %g,",(double)gnorm)); 16558d5ead36SAlp Dener } 16569566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Step: %g, Trust: %g\n",(double)stp,(double)tr)); 16579566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 16588d5ead36SAlp Dener PetscFunctionReturn(0); 16598d5ead36SAlp Dener } 16608d5ead36SAlp Dener 16618d5ead36SAlp Dener /*@ 1662a7e14dcfSSatish Balay TaoDefaultSMonitor - Default routine for monitoring progress of the 166398ea980cSBarry Smith solver. Same as TaoMonitorDefault() except 1664a7e14dcfSSatish Balay it prints fewer digits of the residual as the residual gets smaller. 1665a7e14dcfSSatish Balay This is because the later digits are meaningless and are often 1666a7e14dcfSSatish Balay different on different machines; by using this routine different 1667a7e14dcfSSatish Balay machines will usually generate the same output. It can be turned on 1668a7e14dcfSSatish Balay by using the -tao_smonitor option 1669a7e14dcfSSatish Balay 1670441846f8SBarry Smith Collective on Tao 1671a7e14dcfSSatish Balay 1672a7e14dcfSSatish Balay Input Parameters: 1673441846f8SBarry Smith + tao - the Tao context 16744d4332d5SBarry Smith - ctx - PetscViewer context of type ASCII 1675a7e14dcfSSatish Balay 1676a7e14dcfSSatish Balay Options Database Keys: 1677147403d9SBarry Smith . -tao_smonitor - turn on default short monitoring 1678a7e14dcfSSatish Balay 1679a7e14dcfSSatish Balay Level: advanced 1680a7e14dcfSSatish Balay 1681db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()` 1682a7e14dcfSSatish Balay @*/ 1683441846f8SBarry Smith PetscErrorCode TaoDefaultSMonitor(Tao tao, void *ctx) 1684a7e14dcfSSatish Balay { 16852c9b8b83SAlp Dener PetscInt its, tabs; 1686a7e14dcfSSatish Balay PetscReal fct,gnorm; 16874d4332d5SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 168847a47007SBarry Smith 1689a7e14dcfSSatish Balay PetscFunctionBegin; 169094511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 16914d4332d5SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 1692a7e14dcfSSatish Balay its = tao->niter; 1693a7e14dcfSSatish Balay fct = tao->fc; 1694a7e14dcfSSatish Balay gnorm = tao->residual; 16959566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 16969566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 169763a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"iter = %3" PetscInt_FMT ",",its)); 16989566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Function value %g,",(double)fct)); 1699d393f493SBarry Smith if (gnorm >= PETSC_INFINITY) { 17009566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Residual: Inf \n")); 170108d19d1fSJason Sarich } else if (gnorm > 1.e-6) { 17029566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Residual: %g \n",(double)gnorm)); 1703a7e14dcfSSatish Balay } else if (gnorm > 1.e-11) { 17049566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Residual: < 1.0e-6 \n")); 1705a7e14dcfSSatish Balay } else { 17069566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Residual: < 1.0e-11 \n")); 1707a7e14dcfSSatish Balay } 17089566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 1709a7e14dcfSSatish Balay PetscFunctionReturn(0); 1710a7e14dcfSSatish Balay } 1711a7e14dcfSSatish Balay 17127fab98ebSJason Sarich /*@ 171398ea980cSBarry Smith TaoDefaultCMonitor - same as TaoMonitorDefault() except 1714a7e14dcfSSatish Balay it prints the norm of the constraints function. It can be turned on 1715a7e14dcfSSatish Balay from the command line using the -tao_cmonitor option 1716a7e14dcfSSatish Balay 1717441846f8SBarry Smith Collective on Tao 1718a7e14dcfSSatish Balay 1719a7e14dcfSSatish Balay Input Parameters: 1720441846f8SBarry Smith + tao - the Tao context 172147a47007SBarry Smith - ctx - PetscViewer context or NULL 1722a7e14dcfSSatish Balay 1723a7e14dcfSSatish Balay Options Database Keys: 1724147403d9SBarry Smith . -tao_cmonitor - monitor the constraints 1725a7e14dcfSSatish Balay 1726a7e14dcfSSatish Balay Level: advanced 1727a7e14dcfSSatish Balay 1728db781477SPatrick Sanan .seealso: `TaoMonitorDefault()`, `TaoSetMonitor()` 1729a7e14dcfSSatish Balay @*/ 1730441846f8SBarry Smith PetscErrorCode TaoDefaultCMonitor(Tao tao, void *ctx) 1731a7e14dcfSSatish Balay { 17322c9b8b83SAlp Dener PetscInt its, tabs; 1733a7e14dcfSSatish Balay PetscReal fct,gnorm; 1734d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 1735a7e14dcfSSatish Balay 1736a7e14dcfSSatish Balay PetscFunctionBegin; 173794511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 1738d393f493SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 1739a7e14dcfSSatish Balay its = tao->niter; 1740a7e14dcfSSatish Balay fct = tao->fc; 1741a7e14dcfSSatish Balay gnorm = tao->residual; 17429566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIGetTab(viewer, &tabs)); 17439566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, ((PetscObject)tao)->tablevel)); 174463a3b9bcSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer,"iter = %" PetscInt_FMT ",",its)); 17459566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Function value: %g,",(double)fct)); 17469566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Residual: %g ",(double)gnorm)); 17479566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIIPrintf(viewer," Constraint: %g \n",(double)tao->cnorm)); 17489566063dSJacob Faibussowitsch PetscCall(PetscViewerASCIISetTab(viewer, tabs)); 1749a7e14dcfSSatish Balay PetscFunctionReturn(0); 1750a7e14dcfSSatish Balay } 1751a7e14dcfSSatish Balay 1752a7e14dcfSSatish Balay /*@C 1753a7e14dcfSSatish Balay TaoSolutionMonitor - Views the solution at each iteration 1754a7e14dcfSSatish Balay It can be turned on from the command line using the 1755a7e14dcfSSatish Balay -tao_view_solution option 1756a7e14dcfSSatish Balay 1757441846f8SBarry Smith Collective on Tao 1758a7e14dcfSSatish Balay 1759a7e14dcfSSatish Balay Input Parameters: 1760441846f8SBarry Smith + tao - the Tao context 176147a47007SBarry Smith - ctx - PetscViewer context or NULL 1762a7e14dcfSSatish Balay 1763a7e14dcfSSatish Balay Options Database Keys: 1764147403d9SBarry Smith . -tao_view_solution - view the solution 1765a7e14dcfSSatish Balay 1766a7e14dcfSSatish Balay Level: advanced 1767a7e14dcfSSatish Balay 1768db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1769a7e14dcfSSatish Balay @*/ 1770441846f8SBarry Smith PetscErrorCode TaoSolutionMonitor(Tao tao, void *ctx) 1771a7e14dcfSSatish Balay { 1772feb237baSPierre Jolivet PetscViewer viewer = (PetscViewer)ctx; 177347a47007SBarry Smith 1774a7e14dcfSSatish Balay PetscFunctionBegin; 177594511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 1776d393f493SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 17779566063dSJacob Faibussowitsch PetscCall(VecView(tao->solution,viewer)); 1778a7e14dcfSSatish Balay PetscFunctionReturn(0); 1779a7e14dcfSSatish Balay } 1780a7e14dcfSSatish Balay 1781a7e14dcfSSatish Balay /*@C 1782a7e14dcfSSatish Balay TaoGradientMonitor - Views the gradient at each iteration 1783a7e14dcfSSatish Balay It can be turned on from the command line using the 1784a7e14dcfSSatish Balay -tao_view_gradient option 1785a7e14dcfSSatish Balay 1786441846f8SBarry Smith Collective on Tao 1787a7e14dcfSSatish Balay 1788a7e14dcfSSatish Balay Input Parameters: 1789441846f8SBarry Smith + tao - the Tao context 179047a47007SBarry Smith - ctx - PetscViewer context or NULL 1791a7e14dcfSSatish Balay 1792a7e14dcfSSatish Balay Options Database Keys: 1793147403d9SBarry Smith . -tao_view_gradient - view the gradient at each iteration 1794a7e14dcfSSatish Balay 1795a7e14dcfSSatish Balay Level: advanced 1796a7e14dcfSSatish Balay 1797db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1798a7e14dcfSSatish Balay @*/ 1799441846f8SBarry Smith PetscErrorCode TaoGradientMonitor(Tao tao, void *ctx) 1800a7e14dcfSSatish Balay { 1801d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 180247a47007SBarry Smith 1803a7e14dcfSSatish Balay PetscFunctionBegin; 180494511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 1805d393f493SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 18069566063dSJacob Faibussowitsch PetscCall(VecView(tao->gradient,viewer)); 1807a7e14dcfSSatish Balay PetscFunctionReturn(0); 1808a7e14dcfSSatish Balay } 1809a7e14dcfSSatish Balay 1810a7e14dcfSSatish Balay /*@C 1811147403d9SBarry Smith TaoStepDirectionMonitor - Views the step-direction at each iteration 1812a7e14dcfSSatish Balay 1813441846f8SBarry Smith Collective on Tao 1814a7e14dcfSSatish Balay 1815a7e14dcfSSatish Balay Input Parameters: 1816441846f8SBarry Smith + tao - the Tao context 181747a47007SBarry Smith - ctx - PetscViewer context or NULL 1818a7e14dcfSSatish Balay 1819a7e14dcfSSatish Balay Options Database Keys: 1820147403d9SBarry Smith . -tao_view_gradient - view the gradient at each iteration 1821a7e14dcfSSatish Balay 1822a7e14dcfSSatish Balay Level: advanced 1823a7e14dcfSSatish Balay 1824db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1825a7e14dcfSSatish Balay @*/ 1826441846f8SBarry Smith PetscErrorCode TaoStepDirectionMonitor(Tao tao, void *ctx) 1827a7e14dcfSSatish Balay { 1828d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 1829d393f493SBarry Smith 1830a7e14dcfSSatish Balay PetscFunctionBegin; 183194511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 1832d393f493SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 18339566063dSJacob Faibussowitsch PetscCall(VecView(tao->stepdirection,viewer)); 1834a7e14dcfSSatish Balay PetscFunctionReturn(0); 1835a7e14dcfSSatish Balay } 1836a7e14dcfSSatish Balay 1837a7e14dcfSSatish Balay /*@C 1838a7e14dcfSSatish Balay TaoDrawSolutionMonitor - Plots the solution at each iteration 1839a7e14dcfSSatish Balay It can be turned on from the command line using the 1840a7e14dcfSSatish Balay -tao_draw_solution option 1841a7e14dcfSSatish Balay 1842441846f8SBarry Smith Collective on Tao 1843a7e14dcfSSatish Balay 1844a7e14dcfSSatish Balay Input Parameters: 1845441846f8SBarry Smith + tao - the Tao context 1846e882e171SHong Zhang - ctx - TaoMonitorDraw context 1847a7e14dcfSSatish Balay 1848a7e14dcfSSatish Balay Options Database Keys: 1849147403d9SBarry Smith . -tao_draw_solution - draw the solution at each iteration 1850a7e14dcfSSatish Balay 1851a7e14dcfSSatish Balay Level: advanced 1852a7e14dcfSSatish Balay 1853db781477SPatrick Sanan .seealso: `TaoSolutionMonitor()`, `TaoSetMonitor()`, `TaoDrawGradientMonitor` 1854a7e14dcfSSatish Balay @*/ 1855441846f8SBarry Smith PetscErrorCode TaoDrawSolutionMonitor(Tao tao, void *ctx) 1856a7e14dcfSSatish Balay { 1857e882e171SHong Zhang TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx; 185847a47007SBarry Smith 1859a7e14dcfSSatish Balay PetscFunctionBegin; 186094511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 1861e882e171SHong Zhang if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0); 18629566063dSJacob Faibussowitsch PetscCall(VecView(tao->solution,ictx->viewer)); 1863a7e14dcfSSatish Balay PetscFunctionReturn(0); 1864a7e14dcfSSatish Balay } 1865a7e14dcfSSatish Balay 1866a7e14dcfSSatish Balay /*@C 1867a7e14dcfSSatish Balay TaoDrawGradientMonitor - Plots the gradient at each iteration 1868a7e14dcfSSatish Balay It can be turned on from the command line using the 1869a7e14dcfSSatish Balay -tao_draw_gradient option 1870a7e14dcfSSatish Balay 1871441846f8SBarry Smith Collective on Tao 1872a7e14dcfSSatish Balay 1873a7e14dcfSSatish Balay Input Parameters: 1874441846f8SBarry Smith + tao - the Tao context 1875f55353a2SBarry Smith - ctx - PetscViewer context 1876a7e14dcfSSatish Balay 1877a7e14dcfSSatish Balay Options Database Keys: 1878147403d9SBarry Smith . -tao_draw_gradient - draw the gradient at each iteration 1879a7e14dcfSSatish Balay 1880a7e14dcfSSatish Balay Level: advanced 1881a7e14dcfSSatish Balay 1882db781477SPatrick Sanan .seealso: `TaoGradientMonitor()`, `TaoSetMonitor()`, `TaoDrawSolutionMonitor` 1883a7e14dcfSSatish Balay @*/ 1884441846f8SBarry Smith PetscErrorCode TaoDrawGradientMonitor(Tao tao, void *ctx) 1885a7e14dcfSSatish Balay { 1886e882e171SHong Zhang TaoMonitorDrawCtx ictx = (TaoMonitorDrawCtx)ctx; 188747a47007SBarry Smith 1888a7e14dcfSSatish Balay PetscFunctionBegin; 188994511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 1890e882e171SHong Zhang if (!(((ictx->howoften > 0) && (!(tao->niter % ictx->howoften))) || ((ictx->howoften == -1) && tao->reason))) PetscFunctionReturn(0); 18919566063dSJacob Faibussowitsch PetscCall(VecView(tao->gradient,ictx->viewer)); 1892a7e14dcfSSatish Balay PetscFunctionReturn(0); 1893a7e14dcfSSatish Balay } 1894a7e14dcfSSatish Balay 1895a7e14dcfSSatish Balay /*@C 1896a7e14dcfSSatish Balay TaoDrawStepMonitor - Plots the step direction at each iteration 1897a7e14dcfSSatish Balay 1898441846f8SBarry Smith Collective on Tao 1899a7e14dcfSSatish Balay 1900a7e14dcfSSatish Balay Input Parameters: 1901441846f8SBarry Smith + tao - the Tao context 1902f55353a2SBarry Smith - ctx - PetscViewer context 1903a7e14dcfSSatish Balay 1904a7e14dcfSSatish Balay Options Database Keys: 1905147403d9SBarry Smith . -tao_draw_step - draw the step direction at each iteration 1906a7e14dcfSSatish Balay 1907a7e14dcfSSatish Balay Level: advanced 1908a7e14dcfSSatish Balay 1909db781477SPatrick Sanan .seealso: `TaoSetMonitor()`, `TaoDrawSolutionMonitor` 1910a7e14dcfSSatish Balay @*/ 1911441846f8SBarry Smith PetscErrorCode TaoDrawStepMonitor(Tao tao, void *ctx) 1912a7e14dcfSSatish Balay { 191394511df7SStefano Zampini PetscViewer viewer = (PetscViewer)ctx; 191447a47007SBarry Smith 1915a7e14dcfSSatish Balay PetscFunctionBegin; 191694511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 191794511df7SStefano Zampini PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 19189566063dSJacob Faibussowitsch PetscCall(VecView(tao->stepdirection,viewer)); 1919a7e14dcfSSatish Balay PetscFunctionReturn(0); 1920a7e14dcfSSatish Balay } 1921a7e14dcfSSatish Balay 1922a7e14dcfSSatish Balay /*@C 1923737f463aSAlp Dener TaoResidualMonitor - Views the least-squares residual at each iteration 1924a7e14dcfSSatish Balay 1925441846f8SBarry Smith Collective on Tao 1926a7e14dcfSSatish Balay 1927a7e14dcfSSatish Balay Input Parameters: 1928441846f8SBarry Smith + tao - the Tao context 192947a47007SBarry Smith - ctx - PetscViewer context or NULL 1930a7e14dcfSSatish Balay 1931a7e14dcfSSatish Balay Options Database Keys: 1932147403d9SBarry Smith . -tao_view_ls_residual - view the least-squares residual at each iteration 1933a7e14dcfSSatish Balay 1934a7e14dcfSSatish Balay Level: advanced 1935a7e14dcfSSatish Balay 1936db781477SPatrick Sanan .seealso: `TaoDefaultSMonitor()`, `TaoSetMonitor()` 1937a7e14dcfSSatish Balay @*/ 1938737f463aSAlp Dener PetscErrorCode TaoResidualMonitor(Tao tao, void *ctx) 1939a7e14dcfSSatish Balay { 1940d393f493SBarry Smith PetscViewer viewer = (PetscViewer)ctx; 194147a47007SBarry Smith 1942a7e14dcfSSatish Balay PetscFunctionBegin; 194394511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 1944d393f493SBarry Smith PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2); 19459566063dSJacob Faibussowitsch PetscCall(VecView(tao->ls_res,viewer)); 1946a7e14dcfSSatish Balay PetscFunctionReturn(0); 1947a7e14dcfSSatish Balay } 1948a7e14dcfSSatish Balay 19497fab98ebSJason Sarich /*@ 1950a7e14dcfSSatish Balay TaoDefaultConvergenceTest - Determines whether the solver should continue iterating 1951a7e14dcfSSatish Balay or terminate. 1952a7e14dcfSSatish Balay 1953441846f8SBarry Smith Collective on Tao 1954a7e14dcfSSatish Balay 1955a7e14dcfSSatish Balay Input Parameters: 1956441846f8SBarry Smith + tao - the Tao context 1957a7e14dcfSSatish Balay - dummy - unused dummy context 1958a7e14dcfSSatish Balay 1959a7e14dcfSSatish Balay Output Parameter: 1960a7e14dcfSSatish Balay . reason - for terminating 1961a7e14dcfSSatish Balay 1962a7e14dcfSSatish Balay Notes: 1963a7e14dcfSSatish Balay This routine checks the residual in the optimality conditions, the 1964a7e14dcfSSatish Balay relative residual in the optimity conditions, the number of function 1965a7e14dcfSSatish Balay evaluations, and the function value to test convergence. Some 1966a7e14dcfSSatish Balay solvers may use different convergence routines. 1967a7e14dcfSSatish Balay 1968a7e14dcfSSatish Balay Level: developer 1969a7e14dcfSSatish Balay 1970c2e3fba1SPatrick Sanan .seealso: `TaoSetTolerances()`, `TaoGetConvergedReason()`, `TaoSetConvergedReason()` 1971a7e14dcfSSatish Balay @*/ 1972a7e14dcfSSatish Balay 1973441846f8SBarry Smith PetscErrorCode TaoDefaultConvergenceTest(Tao tao,void *dummy) 1974a7e14dcfSSatish Balay { 1975a7e14dcfSSatish Balay PetscInt niter=tao->niter, nfuncs=PetscMax(tao->nfuncs,tao->nfuncgrads); 1976a7e14dcfSSatish Balay PetscInt max_funcs=tao->max_funcs; 1977a7e14dcfSSatish Balay PetscReal gnorm=tao->residual, gnorm0=tao->gnorm0; 1978a7e14dcfSSatish Balay PetscReal f=tao->fc, steptol=tao->steptol,trradius=tao->step; 1979a7e14dcfSSatish Balay PetscReal gatol=tao->gatol,grtol=tao->grtol,gttol=tao->gttol; 1980e52336cbSBarry Smith PetscReal catol=tao->catol,crtol=tao->crtol; 1981e52336cbSBarry Smith PetscReal fmin=tao->fmin, cnorm=tao->cnorm; 1982e4cb33bbSBarry Smith TaoConvergedReason reason=tao->reason; 1983a7e14dcfSSatish Balay 1984a7e14dcfSSatish Balay PetscFunctionBegin; 1985441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 198694511df7SStefano Zampini if (reason != TAO_CONTINUE_ITERATING) PetscFunctionReturn(0); 1987a7e14dcfSSatish Balay 1988a7e14dcfSSatish Balay if (PetscIsInfOrNanReal(f)) { 19899566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao,"Failed to converged, function value is Inf or NaN\n")); 1990a7e14dcfSSatish Balay reason = TAO_DIVERGED_NAN; 1991a7e14dcfSSatish Balay } else if (f <= fmin && cnorm <=catol) { 19929566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao,"Converged due to function value %g < minimum function value %g\n", (double)f,(double)fmin)); 1993a7e14dcfSSatish Balay reason = TAO_CONVERGED_MINF; 1994a7e14dcfSSatish Balay } else if (gnorm<= gatol && cnorm <=catol) { 19959566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao,"Converged due to residual norm ||g(X)||=%g < %g\n",(double)gnorm,(double)gatol)); 1996a7e14dcfSSatish Balay reason = TAO_CONVERGED_GATOL; 1997a7e14dcfSSatish Balay } else if (f!=0 && PetscAbsReal(gnorm/f) <= grtol && cnorm <= crtol) { 19989566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao,"Converged due to residual ||g(X)||/|f(X)| =%g < %g\n",(double)(gnorm/f),(double)grtol)); 1999a7e14dcfSSatish Balay reason = TAO_CONVERGED_GRTOL; 2000e1d16bb9SBarry Smith } else if (gnorm0 != 0 && ((gttol == 0 && gnorm == 0) || gnorm/gnorm0 < gttol) && cnorm <= crtol) { 20019566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao,"Converged due to relative residual norm ||g(X)||/||g(X0)|| = %g < %g\n",(double)(gnorm/gnorm0),(double)gttol)); 2002a7e14dcfSSatish Balay reason = TAO_CONVERGED_GTTOL; 200394511df7SStefano Zampini } else if (max_funcs >=0 && nfuncs > max_funcs) { 20049566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao,"Exceeded maximum number of function evaluations: %" PetscInt_FMT " > %" PetscInt_FMT "\n", nfuncs,max_funcs)); 2005a7e14dcfSSatish Balay reason = TAO_DIVERGED_MAXFCN; 2006a7e14dcfSSatish Balay } else if (tao->lsflag != 0) { 20079566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao,"Tao Line Search failure.\n")); 2008a7e14dcfSSatish Balay reason = TAO_DIVERGED_LS_FAILURE; 2009a7e14dcfSSatish Balay } else if (trradius < steptol && niter > 0) { 20109566063dSJacob Faibussowitsch PetscCall(PetscInfo(tao,"Trust region/step size too small: %g < %g\n", (double)trradius,(double)steptol)); 2011a7e14dcfSSatish Balay reason = TAO_CONVERGED_STEPTOL; 2012e031d6f5SAlp Dener } else if (niter >= tao->max_it) { 201363a3b9bcSJacob Faibussowitsch PetscCall(PetscInfo(tao,"Exceeded maximum number of iterations: %" PetscInt_FMT " > %" PetscInt_FMT "\n",niter,tao->max_it)); 2014a7e14dcfSSatish Balay reason = TAO_DIVERGED_MAXITS; 2015a7e14dcfSSatish Balay } else { 2016a7e14dcfSSatish Balay reason = TAO_CONTINUE_ITERATING; 2017a7e14dcfSSatish Balay } 2018a7e14dcfSSatish Balay tao->reason = reason; 2019a7e14dcfSSatish Balay PetscFunctionReturn(0); 2020a7e14dcfSSatish Balay } 2021a7e14dcfSSatish Balay 2022a7e14dcfSSatish Balay /*@C 2023a7e14dcfSSatish Balay TaoSetOptionsPrefix - Sets the prefix used for searching for all 2024a7e14dcfSSatish Balay TAO options in the database. 2025a7e14dcfSSatish Balay 2026441846f8SBarry Smith Logically Collective on Tao 2027a7e14dcfSSatish Balay 2028a7e14dcfSSatish Balay Input Parameters: 2029441846f8SBarry Smith + tao - the Tao context 2030a7e14dcfSSatish Balay - prefix - the prefix string to prepend to all TAO option requests 2031a7e14dcfSSatish Balay 2032a7e14dcfSSatish Balay Notes: 2033a7e14dcfSSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 2034a7e14dcfSSatish Balay The first character of all runtime options is AUTOMATICALLY the hyphen. 2035a7e14dcfSSatish Balay 2036a7e14dcfSSatish Balay For example, to distinguish between the runtime options for two 2037a7e14dcfSSatish Balay different TAO solvers, one could call 2038a7e14dcfSSatish Balay .vb 2039a7e14dcfSSatish Balay TaoSetOptionsPrefix(tao1,"sys1_") 2040a7e14dcfSSatish Balay TaoSetOptionsPrefix(tao2,"sys2_") 2041a7e14dcfSSatish Balay .ve 2042a7e14dcfSSatish Balay 2043a7e14dcfSSatish Balay This would enable use of different options for each system, such as 2044a7e14dcfSSatish Balay .vb 20459fa2b5dcSStefano Zampini -sys1_tao_method blmvm -sys1_tao_grtol 1.e-3 20469fa2b5dcSStefano Zampini -sys2_tao_method lmvm -sys2_tao_grtol 1.e-4 2047a7e14dcfSSatish Balay .ve 2048a7e14dcfSSatish Balay 2049a7e14dcfSSatish Balay Level: advanced 2050a7e14dcfSSatish Balay 2051db781477SPatrick Sanan .seealso: `TaoAppendOptionsPrefix()`, `TaoGetOptionsPrefix()` 2052a7e14dcfSSatish Balay @*/ 2053a7e14dcfSSatish Balay 2054441846f8SBarry Smith PetscErrorCode TaoSetOptionsPrefix(Tao tao, const char p[]) 2055a7e14dcfSSatish Balay { 2056a7e14dcfSSatish Balay PetscFunctionBegin; 205794511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 20589566063dSJacob Faibussowitsch PetscCall(PetscObjectSetOptionsPrefix((PetscObject)tao,p)); 2059*1baa6e33SBarry Smith if (tao->linesearch) PetscCall(TaoLineSearchSetOptionsPrefix(tao->linesearch,p)); 2060*1baa6e33SBarry Smith if (tao->ksp) PetscCall(KSPSetOptionsPrefix(tao->ksp,p)); 2061a7e14dcfSSatish Balay PetscFunctionReturn(0); 2062a7e14dcfSSatish Balay } 2063a7e14dcfSSatish Balay 2064a7e14dcfSSatish Balay /*@C 2065a7e14dcfSSatish Balay TaoAppendOptionsPrefix - Appends to the prefix used for searching for all 2066a7e14dcfSSatish Balay TAO options in the database. 2067a7e14dcfSSatish Balay 2068441846f8SBarry Smith Logically Collective on Tao 2069a7e14dcfSSatish Balay 2070a7e14dcfSSatish Balay Input Parameters: 2071441846f8SBarry Smith + tao - the Tao solver context 2072a7e14dcfSSatish Balay - prefix - the prefix string to prepend to all TAO option requests 2073a7e14dcfSSatish Balay 2074a7e14dcfSSatish Balay Notes: 2075a7e14dcfSSatish Balay A hyphen (-) must NOT be given at the beginning of the prefix name. 2076a7e14dcfSSatish Balay The first character of all runtime options is AUTOMATICALLY the hyphen. 2077a7e14dcfSSatish Balay 2078a7e14dcfSSatish Balay Level: advanced 2079a7e14dcfSSatish Balay 2080db781477SPatrick Sanan .seealso: `TaoSetOptionsPrefix()`, `TaoGetOptionsPrefix()` 2081a7e14dcfSSatish Balay @*/ 2082441846f8SBarry Smith PetscErrorCode TaoAppendOptionsPrefix(Tao tao, const char p[]) 2083a7e14dcfSSatish Balay { 2084a7e14dcfSSatish Balay PetscFunctionBegin; 208594511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 20869566063dSJacob Faibussowitsch PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao,p)); 2087*1baa6e33SBarry Smith if (tao->linesearch) PetscCall(PetscObjectAppendOptionsPrefix((PetscObject)tao->linesearch,p)); 2088*1baa6e33SBarry Smith if (tao->ksp) PetscCall(KSPAppendOptionsPrefix(tao->ksp,p)); 2089a7e14dcfSSatish Balay PetscFunctionReturn(0); 2090a7e14dcfSSatish Balay } 2091a7e14dcfSSatish Balay 2092a7e14dcfSSatish Balay /*@C 2093a7e14dcfSSatish Balay TaoGetOptionsPrefix - Gets the prefix used for searching for all 2094a7e14dcfSSatish Balay TAO options in the database 2095a7e14dcfSSatish Balay 2096a7e14dcfSSatish Balay Not Collective 2097a7e14dcfSSatish Balay 2098a7e14dcfSSatish Balay Input Parameters: 2099441846f8SBarry Smith . tao - the Tao context 2100a7e14dcfSSatish Balay 2101a7e14dcfSSatish Balay Output Parameters: 2102a7e14dcfSSatish Balay . prefix - pointer to the prefix string used is returned 2103a7e14dcfSSatish Balay 210495452b02SPatrick Sanan Notes: 210595452b02SPatrick Sanan On the fortran side, the user should pass in a string 'prefix' of 2106a7e14dcfSSatish Balay sufficient length to hold the prefix. 2107a7e14dcfSSatish Balay 2108a7e14dcfSSatish Balay Level: advanced 2109a7e14dcfSSatish Balay 2110db781477SPatrick Sanan .seealso: `TaoSetOptionsPrefix()`, `TaoAppendOptionsPrefix()` 2111a7e14dcfSSatish Balay @*/ 2112441846f8SBarry Smith PetscErrorCode TaoGetOptionsPrefix(Tao tao, const char *p[]) 2113a7e14dcfSSatish Balay { 211494511df7SStefano Zampini PetscFunctionBegin; 211594511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 21169566063dSJacob Faibussowitsch PetscCall(PetscObjectGetOptionsPrefix((PetscObject)tao,p)); 211794511df7SStefano Zampini PetscFunctionReturn(0); 2118a7e14dcfSSatish Balay } 2119a7e14dcfSSatish Balay 2120a7e14dcfSSatish Balay /*@C 2121a7e14dcfSSatish Balay TaoSetType - Sets the method for the unconstrained minimization solver. 2122a7e14dcfSSatish Balay 2123441846f8SBarry Smith Collective on Tao 2124a7e14dcfSSatish Balay 2125a7e14dcfSSatish Balay Input Parameters: 2126441846f8SBarry Smith + solver - the Tao solver context 2127a7e14dcfSSatish Balay - type - a known method 2128a7e14dcfSSatish Balay 2129a7e14dcfSSatish Balay Options Database Key: 213058417fe7SBarry Smith . -tao_type <type> - Sets the method; use -help for a list 213158417fe7SBarry Smith of available methods (for instance, "-tao_type lmvm" or "-tao_type tron") 2132a7e14dcfSSatish Balay 2133a7e14dcfSSatish Balay Available methods include: 213458417fe7SBarry Smith + nls - Newton's method with line search for unconstrained minimization 213558417fe7SBarry Smith . ntr - Newton's method with trust region for unconstrained minimization 213658417fe7SBarry Smith . ntl - Newton's method with trust region, line search for unconstrained minimization 213758417fe7SBarry Smith . lmvm - Limited memory variable metric method for unconstrained minimization 213858417fe7SBarry Smith . cg - Nonlinear conjugate gradient method for unconstrained minimization 213958417fe7SBarry Smith . nm - Nelder-Mead algorithm for derivate-free unconstrained minimization 214058417fe7SBarry Smith . tron - Newton Trust Region method for bound constrained minimization 214158417fe7SBarry Smith . gpcg - Newton Trust Region method for quadratic bound constrained minimization 214258417fe7SBarry Smith . blmvm - Limited memory variable metric method for bound constrained minimization 21431eb8069cSJason Sarich - pounders - Model-based algorithm pounder extended for nonlinear least squares 2144a7e14dcfSSatish Balay 2145a7e14dcfSSatish Balay Level: intermediate 2146a7e14dcfSSatish Balay 2147db781477SPatrick Sanan .seealso: `TaoCreate()`, `TaoGetType()`, `TaoType` 2148a7e14dcfSSatish Balay 2149a7e14dcfSSatish Balay @*/ 2150b625d6c7SJed Brown PetscErrorCode TaoSetType(Tao tao, TaoType type) 2151a7e14dcfSSatish Balay { 2152441846f8SBarry Smith PetscErrorCode (*create_xxx)(Tao); 2153a7e14dcfSSatish Balay PetscBool issame; 2154a7e14dcfSSatish Balay 2155a7e14dcfSSatish Balay PetscFunctionBegin; 2156441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 2157a7e14dcfSSatish Balay 21589566063dSJacob Faibussowitsch PetscCall(PetscObjectTypeCompare((PetscObject)tao,type,&issame)); 2159a7e14dcfSSatish Balay if (issame) PetscFunctionReturn(0); 2160a7e14dcfSSatish Balay 21619566063dSJacob Faibussowitsch PetscCall(PetscFunctionListFind(TaoList, type, (void(**)(void))&create_xxx)); 21623c859ba3SBarry Smith PetscCheck(create_xxx,PetscObjectComm((PetscObject)tao),PETSC_ERR_ARG_UNKNOWN_TYPE,"Unable to find requested Tao type %s",type); 2163a7e14dcfSSatish Balay 2164a7e14dcfSSatish Balay /* Destroy the existing solver information */ 2165a958fbfcSStefano Zampini if (tao->ops->destroy) PetscCall((*tao->ops->destroy)(tao)); 2166*1baa6e33SBarry Smith PetscCall(KSPDestroy(&tao->ksp)); 21679566063dSJacob Faibussowitsch PetscCall(TaoLineSearchDestroy(&tao->linesearch)); 216883c8fe1dSLisandro Dalcin tao->ops->setup = NULL; 216983c8fe1dSLisandro Dalcin tao->ops->solve = NULL; 217083c8fe1dSLisandro Dalcin tao->ops->view = NULL; 217183c8fe1dSLisandro Dalcin tao->ops->setfromoptions = NULL; 217283c8fe1dSLisandro Dalcin tao->ops->destroy = NULL; 2173a7e14dcfSSatish Balay 2174a7e14dcfSSatish Balay tao->setupcalled = PETSC_FALSE; 2175a7e14dcfSSatish Balay 21769566063dSJacob Faibussowitsch PetscCall((*create_xxx)(tao)); 21779566063dSJacob Faibussowitsch PetscCall(PetscObjectChangeTypeName((PetscObject)tao,type)); 2178a7e14dcfSSatish Balay PetscFunctionReturn(0); 2179a7e14dcfSSatish Balay } 218047a47007SBarry Smith 2181a7e14dcfSSatish Balay /*MC 2182441846f8SBarry Smith TaoRegister - Adds a method to the TAO package for unconstrained minimization. 2183a7e14dcfSSatish Balay 2184a7e14dcfSSatish Balay Synopsis: 21855bd1e576SStefano Zampini TaoRegister(char *name_solver,char *path,char *name_Create,PetscErrorCode (*routine_Create)(Tao)) 2186a7e14dcfSSatish Balay 2187a7e14dcfSSatish Balay Not collective 2188a7e14dcfSSatish Balay 2189a7e14dcfSSatish Balay Input Parameters: 2190a7e14dcfSSatish Balay + sname - name of a new user-defined solver 2191a7e14dcfSSatish Balay - func - routine to Create method context 2192a7e14dcfSSatish Balay 2193a7e14dcfSSatish Balay Notes: 2194441846f8SBarry Smith TaoRegister() may be called multiple times to add several user-defined solvers. 2195a7e14dcfSSatish Balay 2196a7e14dcfSSatish Balay Sample usage: 2197a7e14dcfSSatish Balay .vb 2198441846f8SBarry Smith TaoRegister("my_solver",MySolverCreate); 2199a7e14dcfSSatish Balay .ve 2200a7e14dcfSSatish Balay 2201a7e14dcfSSatish Balay Then, your solver can be chosen with the procedural interface via 2202a7e14dcfSSatish Balay $ TaoSetType(tao,"my_solver") 2203a7e14dcfSSatish Balay or at runtime via the option 220458417fe7SBarry Smith $ -tao_type my_solver 2205a7e14dcfSSatish Balay 2206a7e14dcfSSatish Balay Level: advanced 2207a7e14dcfSSatish Balay 2208db781477SPatrick Sanan .seealso: `TaoRegisterAll()`, `TaoRegisterDestroy()` 2209a7e14dcfSSatish Balay M*/ 2210441846f8SBarry Smith PetscErrorCode TaoRegister(const char sname[], PetscErrorCode (*func)(Tao)) 2211a7e14dcfSSatish Balay { 2212a7e14dcfSSatish Balay PetscFunctionBegin; 22139566063dSJacob Faibussowitsch PetscCall(TaoInitializePackage()); 22149566063dSJacob Faibussowitsch PetscCall(PetscFunctionListAdd(&TaoList,sname,(void (*)(void))func)); 2215a7e14dcfSSatish Balay PetscFunctionReturn(0); 2216a7e14dcfSSatish Balay } 2217a7e14dcfSSatish Balay 2218a7e14dcfSSatish Balay /*@C 2219441846f8SBarry Smith TaoRegisterDestroy - Frees the list of minimization solvers that were 2220441846f8SBarry Smith registered by TaoRegisterDynamic(). 2221a7e14dcfSSatish Balay 2222a7e14dcfSSatish Balay Not Collective 2223a7e14dcfSSatish Balay 2224a7e14dcfSSatish Balay Level: advanced 2225a7e14dcfSSatish Balay 2226db781477SPatrick Sanan .seealso: `TaoRegisterAll()`, `TaoRegister()` 2227a7e14dcfSSatish Balay @*/ 2228441846f8SBarry Smith PetscErrorCode TaoRegisterDestroy(void) 2229a7e14dcfSSatish Balay { 2230a7e14dcfSSatish Balay PetscFunctionBegin; 22319566063dSJacob Faibussowitsch PetscCall(PetscFunctionListDestroy(&TaoList)); 2232441846f8SBarry Smith TaoRegisterAllCalled = PETSC_FALSE; 2233a7e14dcfSSatish Balay PetscFunctionReturn(0); 2234a7e14dcfSSatish Balay } 223547a47007SBarry Smith 22368931d482SJason Sarich /*@ 22378931d482SJason Sarich TaoGetIterationNumber - Gets the number of Tao iterations completed 22388931d482SJason Sarich at this time. 22398931d482SJason Sarich 22408931d482SJason Sarich Not Collective 22418931d482SJason Sarich 22428931d482SJason Sarich Input Parameter: 22438931d482SJason Sarich . tao - Tao context 22448931d482SJason Sarich 22458931d482SJason Sarich Output Parameter: 22468931d482SJason Sarich . iter - iteration number 22478931d482SJason Sarich 22488931d482SJason Sarich Notes: 22498931d482SJason Sarich For example, during the computation of iteration 2 this would return 1. 22508931d482SJason Sarich 22518931d482SJason Sarich Level: intermediate 22528931d482SJason Sarich 2253db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`, `TaoGetResidualNorm()`, `TaoGetObjective()` 22548931d482SJason Sarich @*/ 22558931d482SJason Sarich PetscErrorCode TaoGetIterationNumber(Tao tao,PetscInt *iter) 22568931d482SJason Sarich { 22578931d482SJason Sarich PetscFunctionBegin; 22588931d482SJason Sarich PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 22598931d482SJason Sarich PetscValidIntPointer(iter,2); 22608931d482SJason Sarich *iter = tao->niter; 22618931d482SJason Sarich PetscFunctionReturn(0); 22628931d482SJason Sarich } 22638931d482SJason Sarich 22648931d482SJason Sarich /*@ 226579f5d8caSBarry Smith TaoGetResidualNorm - Gets the current value of the norm of the residual 226679f5d8caSBarry Smith at this time. 226779f5d8caSBarry Smith 226879f5d8caSBarry Smith Not Collective 226979f5d8caSBarry Smith 227079f5d8caSBarry Smith Input Parameter: 227179f5d8caSBarry Smith . tao - Tao context 227279f5d8caSBarry Smith 227379f5d8caSBarry Smith Output Parameter: 227479f5d8caSBarry Smith . value - the current value 227579f5d8caSBarry Smith 227679f5d8caSBarry Smith Level: intermediate 227779f5d8caSBarry Smith 227879f5d8caSBarry Smith Developer Note: This is the 2-norm of the residual, we cannot use TaoGetGradientNorm() because that has 227979f5d8caSBarry Smith a different meaning. For some reason Tao sometimes calls the gradient the residual. 228079f5d8caSBarry Smith 2281db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()`, `TaoGetIterationNumber()`, `TaoGetObjective()` 228279f5d8caSBarry Smith @*/ 228379f5d8caSBarry Smith PetscErrorCode TaoGetResidualNorm(Tao tao,PetscReal *value) 228479f5d8caSBarry Smith { 228579f5d8caSBarry Smith PetscFunctionBegin; 228679f5d8caSBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 228779f5d8caSBarry Smith PetscValidRealPointer(value,2); 228879f5d8caSBarry Smith *value = tao->residual; 228979f5d8caSBarry Smith PetscFunctionReturn(0); 229079f5d8caSBarry Smith } 229179f5d8caSBarry Smith 229279f5d8caSBarry Smith /*@ 22938931d482SJason Sarich TaoSetIterationNumber - Sets the current iteration number. 22948931d482SJason Sarich 229594511df7SStefano Zampini Logically Collective on Tao 22968931d482SJason Sarich 2297d8d19677SJose E. Roman Input Parameters: 2298a2b725a8SWilliam Gropp + tao - Tao context 2299a2b725a8SWilliam Gropp - iter - iteration number 23008931d482SJason Sarich 23018931d482SJason Sarich Level: developer 23028931d482SJason Sarich 2303db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()` 23048931d482SJason Sarich @*/ 23058931d482SJason Sarich PetscErrorCode TaoSetIterationNumber(Tao tao,PetscInt iter) 23068931d482SJason Sarich { 23078931d482SJason Sarich PetscFunctionBegin; 23088931d482SJason Sarich PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 230994511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao,iter,2); 23109566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao)); 23118931d482SJason Sarich tao->niter = iter; 23129566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao)); 23138931d482SJason Sarich PetscFunctionReturn(0); 23148931d482SJason Sarich } 23158931d482SJason Sarich 23168931d482SJason Sarich /*@ 23178931d482SJason Sarich TaoGetTotalIterationNumber - Gets the total number of Tao iterations 23188931d482SJason Sarich completed. This number keeps accumulating if multiple solves 23198931d482SJason Sarich are called with the Tao object. 23208931d482SJason Sarich 23218931d482SJason Sarich Not Collective 23228931d482SJason Sarich 23238931d482SJason Sarich Input Parameter: 23248931d482SJason Sarich . tao - Tao context 23258931d482SJason Sarich 23268931d482SJason Sarich Output Parameter: 23278931d482SJason Sarich . iter - iteration number 23288931d482SJason Sarich 23298931d482SJason Sarich Notes: 23308931d482SJason Sarich The total iteration count is updated after each solve, if there is a current 23318931d482SJason Sarich TaoSolve() in progress then those iterations are not yet counted. 23328931d482SJason Sarich 23338931d482SJason Sarich Level: intermediate 23348931d482SJason Sarich 2335db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()` 23368931d482SJason Sarich @*/ 23378931d482SJason Sarich PetscErrorCode TaoGetTotalIterationNumber(Tao tao,PetscInt *iter) 23388931d482SJason Sarich { 23398931d482SJason Sarich PetscFunctionBegin; 23408931d482SJason Sarich PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 23418931d482SJason Sarich PetscValidIntPointer(iter,2); 23428931d482SJason Sarich *iter = tao->ntotalits; 23438931d482SJason Sarich PetscFunctionReturn(0); 23448931d482SJason Sarich } 23458931d482SJason Sarich 23468931d482SJason Sarich /*@ 23478931d482SJason Sarich TaoSetTotalIterationNumber - Sets the current total iteration number. 23488931d482SJason Sarich 234994511df7SStefano Zampini Logically Collective on Tao 23508931d482SJason Sarich 2351d8d19677SJose E. Roman Input Parameters: 2352a2b725a8SWilliam Gropp + tao - Tao context 2353a2b725a8SWilliam Gropp - iter - iteration number 23548931d482SJason Sarich 23558931d482SJason Sarich Level: developer 23568931d482SJason Sarich 2357db781477SPatrick Sanan .seealso: `TaoGetLinearSolveIterations()` 23588931d482SJason Sarich @*/ 23598931d482SJason Sarich PetscErrorCode TaoSetTotalIterationNumber(Tao tao,PetscInt iter) 23608931d482SJason Sarich { 23618931d482SJason Sarich PetscFunctionBegin; 23628931d482SJason Sarich PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 236394511df7SStefano Zampini PetscValidLogicalCollectiveInt(tao,iter,2); 23649566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsTakeAccess((PetscObject)tao)); 23658931d482SJason Sarich tao->ntotalits = iter; 23669566063dSJacob Faibussowitsch PetscCall(PetscObjectSAWsGrantAccess((PetscObject)tao)); 23678931d482SJason Sarich PetscFunctionReturn(0); 23688931d482SJason Sarich } 23698931d482SJason Sarich 2370a7e14dcfSSatish Balay /*@ 2371e4cb33bbSBarry Smith TaoSetConvergedReason - Sets the termination flag on a Tao object 2372a7e14dcfSSatish Balay 2373441846f8SBarry Smith Logically Collective on Tao 2374a7e14dcfSSatish Balay 2375a7e14dcfSSatish Balay Input Parameters: 2376441846f8SBarry Smith + tao - the Tao context 2377a7e14dcfSSatish Balay - reason - one of 2378a7e14dcfSSatish Balay $ TAO_CONVERGED_ATOL (2), 2379a7e14dcfSSatish Balay $ TAO_CONVERGED_RTOL (3), 2380a7e14dcfSSatish Balay $ TAO_CONVERGED_STEPTOL (4), 2381a7e14dcfSSatish Balay $ TAO_CONVERGED_MINF (5), 2382a7e14dcfSSatish Balay $ TAO_CONVERGED_USER (6), 2383a7e14dcfSSatish Balay $ TAO_DIVERGED_MAXITS (-2), 2384a7e14dcfSSatish Balay $ TAO_DIVERGED_NAN (-4), 2385a7e14dcfSSatish Balay $ TAO_DIVERGED_MAXFCN (-5), 2386a7e14dcfSSatish Balay $ TAO_DIVERGED_LS_FAILURE (-6), 2387a7e14dcfSSatish Balay $ TAO_DIVERGED_TR_REDUCTION (-7), 2388a7e14dcfSSatish Balay $ TAO_DIVERGED_USER (-8), 2389a7e14dcfSSatish Balay $ TAO_CONTINUE_ITERATING (0) 2390a7e14dcfSSatish Balay 2391a7e14dcfSSatish Balay Level: intermediate 2392a7e14dcfSSatish Balay 2393a7e14dcfSSatish Balay @*/ 2394e4cb33bbSBarry Smith PetscErrorCode TaoSetConvergedReason(Tao tao, TaoConvergedReason reason) 2395a7e14dcfSSatish Balay { 2396a7e14dcfSSatish Balay PetscFunctionBegin; 239794511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 239894511df7SStefano Zampini PetscValidLogicalCollectiveEnum(tao,reason,2); 2399a7e14dcfSSatish Balay tao->reason = reason; 2400a7e14dcfSSatish Balay PetscFunctionReturn(0); 2401a7e14dcfSSatish Balay } 2402a7e14dcfSSatish Balay 2403a7e14dcfSSatish Balay /*@ 2404e4cb33bbSBarry Smith TaoGetConvergedReason - Gets the reason the Tao iteration was stopped. 2405a7e14dcfSSatish Balay 2406a7e14dcfSSatish Balay Not Collective 2407a7e14dcfSSatish Balay 2408a7e14dcfSSatish Balay Input Parameter: 2409441846f8SBarry Smith . tao - the Tao solver context 2410a7e14dcfSSatish Balay 2411a7e14dcfSSatish Balay Output Parameter: 2412a7e14dcfSSatish Balay . reason - one of 2413a7e14dcfSSatish Balay $ TAO_CONVERGED_GATOL (3) ||g(X)|| < gatol 2414a7e14dcfSSatish Balay $ TAO_CONVERGED_GRTOL (4) ||g(X)|| / f(X) < grtol 2415a7e14dcfSSatish Balay $ TAO_CONVERGED_GTTOL (5) ||g(X)|| / ||g(X0)|| < gttol 2416a7e14dcfSSatish Balay $ TAO_CONVERGED_STEPTOL (6) step size small 2417a7e14dcfSSatish Balay $ TAO_CONVERGED_MINF (7) F < F_min 2418a7e14dcfSSatish Balay $ TAO_CONVERGED_USER (8) User defined 2419a7e14dcfSSatish Balay $ TAO_DIVERGED_MAXITS (-2) its > maxits 2420a7e14dcfSSatish Balay $ TAO_DIVERGED_NAN (-4) Numerical problems 2421a7e14dcfSSatish Balay $ TAO_DIVERGED_MAXFCN (-5) fevals > max_funcsals 2422a7e14dcfSSatish Balay $ TAO_DIVERGED_LS_FAILURE (-6) line search failure 2423a7e14dcfSSatish Balay $ TAO_DIVERGED_TR_REDUCTION (-7) trust region failure 2424a7e14dcfSSatish Balay $ TAO_DIVERGED_USER(-8) (user defined) 2425a7e14dcfSSatish Balay $ TAO_CONTINUE_ITERATING (0) 2426a7e14dcfSSatish Balay 2427a7e14dcfSSatish Balay where 2428a7e14dcfSSatish Balay + X - current solution 2429a7e14dcfSSatish Balay . X0 - initial guess 2430a7e14dcfSSatish Balay . f(X) - current function value 2431a7e14dcfSSatish Balay . f(X*) - true solution (estimated) 2432a7e14dcfSSatish Balay . g(X) - current gradient 2433a7e14dcfSSatish Balay . its - current iterate number 2434a7e14dcfSSatish Balay . maxits - maximum number of iterates 2435a7e14dcfSSatish Balay . fevals - number of function evaluations 2436a7e14dcfSSatish Balay - max_funcsals - maximum number of function evaluations 2437a7e14dcfSSatish Balay 2438a7e14dcfSSatish Balay Level: intermediate 2439a7e14dcfSSatish Balay 2440db781477SPatrick Sanan .seealso: `TaoSetConvergenceTest()`, `TaoSetTolerances()` 2441a7e14dcfSSatish Balay 2442a7e14dcfSSatish Balay @*/ 2443e4cb33bbSBarry Smith PetscErrorCode TaoGetConvergedReason(Tao tao, TaoConvergedReason *reason) 2444a7e14dcfSSatish Balay { 2445a7e14dcfSSatish Balay PetscFunctionBegin; 2446441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 2447a7e14dcfSSatish Balay PetscValidPointer(reason,2); 2448a7e14dcfSSatish Balay *reason = tao->reason; 2449a7e14dcfSSatish Balay PetscFunctionReturn(0); 2450a7e14dcfSSatish Balay } 2451a7e14dcfSSatish Balay 2452a7e14dcfSSatish Balay /*@ 2453a7e14dcfSSatish Balay TaoGetSolutionStatus - Get the current iterate, objective value, 2454a7e14dcfSSatish Balay residual, infeasibility, and termination 2455a7e14dcfSSatish Balay 2456a7e14dcfSSatish Balay Not Collective 2457a7e14dcfSSatish Balay 2458f899ff85SJose E. Roman Input Parameter: 2459441846f8SBarry Smith . tao - the Tao context 2460a7e14dcfSSatish Balay 2461a7e14dcfSSatish Balay Output Parameters: 2462a7e14dcfSSatish Balay + iterate - the current iterate number (>=0) 2463a7e14dcfSSatish Balay . f - the current function value 246458417fe7SBarry Smith . gnorm - the square of the gradient norm, duality gap, or other measure indicating distance from optimality. 2465a7e14dcfSSatish Balay . cnorm - the infeasibility of the current solution with regard to the constraints. 2466a7e14dcfSSatish Balay . xdiff - the step length or trust region radius of the most recent iterate. 2467a7e14dcfSSatish Balay - reason - The termination reason, which can equal TAO_CONTINUE_ITERATING 2468a7e14dcfSSatish Balay 2469a7e14dcfSSatish Balay Level: intermediate 2470a7e14dcfSSatish Balay 2471a7e14dcfSSatish Balay Note: 2472a7e14dcfSSatish Balay TAO returns the values set by the solvers in the routine TaoMonitor(). 2473a7e14dcfSSatish Balay 2474a7e14dcfSSatish Balay Note: 247547a47007SBarry Smith If any of the output arguments are set to NULL, no corresponding value will be returned. 2476a7e14dcfSSatish Balay 2477db781477SPatrick Sanan .seealso: `TaoMonitor()`, `TaoGetConvergedReason()` 2478a7e14dcfSSatish Balay @*/ 2479e4cb33bbSBarry Smith PetscErrorCode TaoGetSolutionStatus(Tao tao, PetscInt *its, PetscReal *f, PetscReal *gnorm, PetscReal *cnorm, PetscReal *xdiff, TaoConvergedReason *reason) 2480a7e14dcfSSatish Balay { 2481a7e14dcfSSatish Balay PetscFunctionBegin; 248294511df7SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 2483a7e14dcfSSatish Balay if (its) *its = tao->niter; 2484a7e14dcfSSatish Balay if (f) *f = tao->fc; 2485a7e14dcfSSatish Balay if (gnorm) *gnorm = tao->residual; 2486a7e14dcfSSatish Balay if (cnorm) *cnorm = tao->cnorm; 2487a7e14dcfSSatish Balay if (reason) *reason = tao->reason; 2488a7e14dcfSSatish Balay if (xdiff) *xdiff = tao->step; 2489a7e14dcfSSatish Balay PetscFunctionReturn(0); 2490a7e14dcfSSatish Balay } 2491a7e14dcfSSatish Balay 2492a7e14dcfSSatish Balay /*@C 2493441846f8SBarry Smith TaoGetType - Gets the current Tao algorithm. 2494a7e14dcfSSatish Balay 2495a7e14dcfSSatish Balay Not Collective 2496a7e14dcfSSatish Balay 2497a7e14dcfSSatish Balay Input Parameter: 2498441846f8SBarry Smith . tao - the Tao solver context 2499a7e14dcfSSatish Balay 2500a7e14dcfSSatish Balay Output Parameter: 2501441846f8SBarry Smith . type - Tao method 2502a7e14dcfSSatish Balay 2503a7e14dcfSSatish Balay Level: intermediate 2504a7e14dcfSSatish Balay 2505a7e14dcfSSatish Balay @*/ 2506b625d6c7SJed Brown PetscErrorCode TaoGetType(Tao tao,TaoType *type) 2507a7e14dcfSSatish Balay { 2508a7e14dcfSSatish Balay PetscFunctionBegin; 2509441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 2510a7e14dcfSSatish Balay PetscValidPointer(type,2); 2511a7e14dcfSSatish Balay *type = ((PetscObject)tao)->type_name; 2512a7e14dcfSSatish Balay PetscFunctionReturn(0); 2513a7e14dcfSSatish Balay } 2514a7e14dcfSSatish Balay 2515a7e14dcfSSatish Balay /*@C 2516a7e14dcfSSatish Balay TaoMonitor - Monitor the solver and the current solution. This 2517a7e14dcfSSatish Balay routine will record the iteration number and residual statistics, 2518a7e14dcfSSatish Balay call any monitors specified by the user, and calls the convergence-check routine. 2519a7e14dcfSSatish Balay 2520a7e14dcfSSatish Balay Input Parameters: 2521441846f8SBarry Smith + tao - the Tao context 2522a7e14dcfSSatish Balay . its - the current iterate number (>=0) 2523a7e14dcfSSatish Balay . f - the current objective function value 252458417fe7SBarry Smith . res - the gradient norm, square root of the duality gap, or other measure indicating distince from optimality. This measure will be recorded and 2525a7e14dcfSSatish Balay used for some termination tests. 2526a7e14dcfSSatish Balay . cnorm - the infeasibility of the current solution with regard to the constraints. 2527a7e14dcfSSatish Balay - steplength - multiple of the step direction added to the previous iterate. 2528a7e14dcfSSatish Balay 2529a7e14dcfSSatish Balay Output Parameters: 2530a7e14dcfSSatish Balay . reason - The termination reason, which can equal TAO_CONTINUE_ITERATING 2531a7e14dcfSSatish Balay 2532a7e14dcfSSatish Balay Options Database Key: 2533a7e14dcfSSatish Balay . -tao_monitor - Use the default monitor, which prints statistics to standard output 2534a7e14dcfSSatish Balay 2535db781477SPatrick Sanan .seealso `TaoGetConvergedReason()`, `TaoMonitorDefault()`, `TaoSetMonitor()` 2536a7e14dcfSSatish Balay 2537a7e14dcfSSatish Balay Level: developer 2538a7e14dcfSSatish Balay 2539a7e14dcfSSatish Balay @*/ 25403ecd9318SAlp Dener PetscErrorCode TaoMonitor(Tao tao, PetscInt its, PetscReal f, PetscReal res, PetscReal cnorm, PetscReal steplength) 2541a7e14dcfSSatish Balay { 254247a47007SBarry Smith PetscInt i; 254347a47007SBarry Smith 2544a7e14dcfSSatish Balay PetscFunctionBegin; 2545441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 2546a7e14dcfSSatish Balay tao->fc = f; 2547a7e14dcfSSatish Balay tao->residual = res; 2548a7e14dcfSSatish Balay tao->cnorm = cnorm; 2549a7e14dcfSSatish Balay tao->step = steplength; 2550e52336cbSBarry Smith if (!its) { 255194511df7SStefano Zampini tao->cnorm0 = cnorm; 255294511df7SStefano Zampini tao->gnorm0 = res; 2553a7e14dcfSSatish Balay } 25543c859ba3SBarry Smith PetscCheck(!PetscIsInfOrNanReal(f) && !PetscIsInfOrNanReal(res),PetscObjectComm((PetscObject)tao),PETSC_ERR_USER, "User provided compute function generated Inf or NaN"); 2555a7e14dcfSSatish Balay for (i=0;i<tao->numbermonitors;i++) { 25569566063dSJacob Faibussowitsch PetscCall((*tao->monitor[i])(tao,tao->monitorcontext[i])); 2557a7e14dcfSSatish Balay } 2558a7e14dcfSSatish Balay PetscFunctionReturn(0); 2559a7e14dcfSSatish Balay } 2560a7e14dcfSSatish Balay 2561a7e14dcfSSatish Balay /*@ 2562ae93cb3cSJason Sarich TaoSetConvergenceHistory - Sets the array used to hold the convergence history. 2563a7e14dcfSSatish Balay 2564441846f8SBarry Smith Logically Collective on Tao 2565a7e14dcfSSatish Balay 2566a7e14dcfSSatish Balay Input Parameters: 2567441846f8SBarry Smith + tao - the Tao solver context 2568a7e14dcfSSatish Balay . obj - array to hold objective value history 2569a7e14dcfSSatish Balay . resid - array to hold residual history 2570a7e14dcfSSatish Balay . cnorm - array to hold constraint violation history 2571be1558d8SJason Sarich . lits - integer array holds the number of linear iterations for each Tao iteration 2572a7e14dcfSSatish Balay . na - size of obj, resid, and cnorm 2573a7e14dcfSSatish Balay - reset - PetscTrue indicates each new minimization resets the history counter to zero, 2574a7e14dcfSSatish Balay else it continues storing new values for new minimizations after the old ones 2575a7e14dcfSSatish Balay 2576a7e14dcfSSatish Balay Notes: 2577a7e14dcfSSatish Balay If set, TAO will fill the given arrays with the indicated 2578ae93cb3cSJason Sarich information at each iteration. If 'obj','resid','cnorm','lits' are 2579ae93cb3cSJason Sarich *all* NULL then space (using size na, or 1000 if na is PETSC_DECIDE or 2580ae93cb3cSJason Sarich PETSC_DEFAULT) is allocated for the history. 2581be1558d8SJason Sarich If not all are NULL, then only the non-NULL information categories 2582ae93cb3cSJason Sarich will be stored, the others will be ignored. 2583ae93cb3cSJason Sarich 2584ae93cb3cSJason Sarich Any convergence information after iteration number 'na' will not be stored. 2585a7e14dcfSSatish Balay 2586a7e14dcfSSatish Balay This routine is useful, e.g., when running a code for purposes 2587a7e14dcfSSatish Balay of accurate performance monitoring, when no I/O should be done 2588a7e14dcfSSatish Balay during the section of code that is being timed. 2589a7e14dcfSSatish Balay 2590a7e14dcfSSatish Balay Level: intermediate 2591a7e14dcfSSatish Balay 2592db781477SPatrick Sanan .seealso: `TaoGetConvergenceHistory()` 2593a7e14dcfSSatish Balay 2594a7e14dcfSSatish Balay @*/ 25951b266c99SBarry Smith PetscErrorCode TaoSetConvergenceHistory(Tao tao, PetscReal obj[], PetscReal resid[], PetscReal cnorm[], PetscInt lits[], PetscInt na,PetscBool reset) 2596a7e14dcfSSatish Balay { 2597a7e14dcfSSatish Balay PetscFunctionBegin; 2598441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 2599064a246eSJacob Faibussowitsch if (obj) PetscValidRealPointer(obj,2); 2600064a246eSJacob Faibussowitsch if (resid) PetscValidRealPointer(resid,3); 2601064a246eSJacob Faibussowitsch if (cnorm) PetscValidRealPointer(cnorm,4); 2602ae93cb3cSJason Sarich if (lits) PetscValidIntPointer(lits,5); 2603ae93cb3cSJason Sarich 2604ae93cb3cSJason Sarich if (na == PETSC_DECIDE || na == PETSC_DEFAULT) na = 1000; 2605ae93cb3cSJason Sarich if (!obj && !resid && !cnorm && !lits) { 26069566063dSJacob Faibussowitsch PetscCall(PetscCalloc4(na,&obj,na,&resid,na,&cnorm,na,&lits)); 2607ae93cb3cSJason Sarich tao->hist_malloc = PETSC_TRUE; 2608ae93cb3cSJason Sarich } 2609ae93cb3cSJason Sarich 2610a7e14dcfSSatish Balay tao->hist_obj = obj; 2611a7e14dcfSSatish Balay tao->hist_resid = resid; 2612a7e14dcfSSatish Balay tao->hist_cnorm = cnorm; 2613ae93cb3cSJason Sarich tao->hist_lits = lits; 2614a7e14dcfSSatish Balay tao->hist_max = na; 2615a7e14dcfSSatish Balay tao->hist_reset = reset; 2616ae93cb3cSJason Sarich tao->hist_len = 0; 2617a7e14dcfSSatish Balay PetscFunctionReturn(0); 2618a7e14dcfSSatish Balay } 2619a7e14dcfSSatish Balay 2620a7e14dcfSSatish Balay /*@C 2621ae93cb3cSJason Sarich TaoGetConvergenceHistory - Gets the arrays used to hold the convergence history. 2622a7e14dcfSSatish Balay 2623441846f8SBarry Smith Collective on Tao 2624a7e14dcfSSatish Balay 2625a7e14dcfSSatish Balay Input Parameter: 2626441846f8SBarry Smith . tao - the Tao context 2627a7e14dcfSSatish Balay 2628a7e14dcfSSatish Balay Output Parameters: 2629a7e14dcfSSatish Balay + obj - array used to hold objective value history 2630a7e14dcfSSatish Balay . resid - array used to hold residual history 2631a7e14dcfSSatish Balay . cnorm - array used to hold constraint violation history 2632ae93cb3cSJason Sarich . lits - integer array used to hold linear solver iteration count 2633dd25e5eeSStefano Zampini - nhist - size of obj, resid, cnorm, and lits 2634a7e14dcfSSatish Balay 2635a7e14dcfSSatish Balay Notes: 2636ae93cb3cSJason Sarich This routine must be preceded by calls to TaoSetConvergenceHistory() 2637ae93cb3cSJason Sarich and TaoSolve(), otherwise it returns useless information. 2638ae93cb3cSJason Sarich 2639a7e14dcfSSatish Balay The calling sequence for this routine in Fortran is 2640ae93cb3cSJason Sarich $ call TaoGetConvergenceHistory(Tao tao, PetscInt nhist, PetscErrorCode ierr) 2641a7e14dcfSSatish Balay 2642a7e14dcfSSatish Balay This routine is useful, e.g., when running a code for purposes 2643a7e14dcfSSatish Balay of accurate performance monitoring, when no I/O should be done 2644a7e14dcfSSatish Balay during the section of code that is being timed. 2645a7e14dcfSSatish Balay 2646a7e14dcfSSatish Balay Level: advanced 2647a7e14dcfSSatish Balay 2648db781477SPatrick Sanan .seealso: `TaoSetConvergenceHistory()` 2649a7e14dcfSSatish Balay 2650a7e14dcfSSatish Balay @*/ 2651ae93cb3cSJason Sarich PetscErrorCode TaoGetConvergenceHistory(Tao tao, PetscReal **obj, PetscReal **resid, PetscReal **cnorm, PetscInt **lits, PetscInt *nhist) 2652a7e14dcfSSatish Balay { 2653a7e14dcfSSatish Balay PetscFunctionBegin; 2654441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 2655a7e14dcfSSatish Balay if (obj) *obj = tao->hist_obj; 2656a7e14dcfSSatish Balay if (cnorm) *cnorm = tao->hist_cnorm; 2657a7e14dcfSSatish Balay if (resid) *resid = tao->hist_resid; 2658a7e14dcfSSatish Balay if (nhist) *nhist = tao->hist_len; 2659a7e14dcfSSatish Balay PetscFunctionReturn(0); 2660a7e14dcfSSatish Balay } 2661a7e14dcfSSatish Balay 2662a7e14dcfSSatish Balay /*@ 2663a7e14dcfSSatish Balay TaoSetApplicationContext - Sets the optional user-defined context for 2664a7e14dcfSSatish Balay a solver. 2665a7e14dcfSSatish Balay 2666441846f8SBarry Smith Logically Collective on Tao 2667a7e14dcfSSatish Balay 2668a7e14dcfSSatish Balay Input Parameters: 2669441846f8SBarry Smith + tao - the Tao context 2670a7e14dcfSSatish Balay - usrP - optional user context 2671a7e14dcfSSatish Balay 2672a7e14dcfSSatish Balay Level: intermediate 2673a7e14dcfSSatish Balay 2674db781477SPatrick Sanan .seealso: `TaoGetApplicationContext()`, `TaoSetApplicationContext()` 2675a7e14dcfSSatish Balay @*/ 2676441846f8SBarry Smith PetscErrorCode TaoSetApplicationContext(Tao tao,void *usrP) 2677a7e14dcfSSatish Balay { 2678a7e14dcfSSatish Balay PetscFunctionBegin; 2679441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 2680a7e14dcfSSatish Balay tao->user = usrP; 2681a7e14dcfSSatish Balay PetscFunctionReturn(0); 2682a7e14dcfSSatish Balay } 2683a7e14dcfSSatish Balay 2684a7e14dcfSSatish Balay /*@ 2685a7e14dcfSSatish Balay TaoGetApplicationContext - Gets the user-defined context for a 2686a7e14dcfSSatish Balay TAO solvers. 2687a7e14dcfSSatish Balay 2688a7e14dcfSSatish Balay Not Collective 2689a7e14dcfSSatish Balay 2690a7e14dcfSSatish Balay Input Parameter: 2691441846f8SBarry Smith . tao - Tao context 2692a7e14dcfSSatish Balay 2693a7e14dcfSSatish Balay Output Parameter: 2694a7e14dcfSSatish Balay . usrP - user context 2695a7e14dcfSSatish Balay 2696a7e14dcfSSatish Balay Level: intermediate 2697a7e14dcfSSatish Balay 2698db781477SPatrick Sanan .seealso: `TaoSetApplicationContext()` 2699a7e14dcfSSatish Balay @*/ 2700441846f8SBarry Smith PetscErrorCode TaoGetApplicationContext(Tao tao,void *usrP) 2701a7e14dcfSSatish Balay { 2702a7e14dcfSSatish Balay PetscFunctionBegin; 2703441846f8SBarry Smith PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 270494511df7SStefano Zampini PetscValidPointer(usrP,2); 2705a7e14dcfSSatish Balay *(void**)usrP = tao->user; 2706a7e14dcfSSatish Balay PetscFunctionReturn(0); 2707a7e14dcfSSatish Balay } 2708a9603a14SPatrick Farrell 2709a9603a14SPatrick Farrell /*@ 2710a9603a14SPatrick Farrell TaoSetGradientNorm - Sets the matrix used to define the inner product that measures the size of the gradient. 2711a9603a14SPatrick Farrell 2712a9603a14SPatrick Farrell Collective on tao 2713a9603a14SPatrick Farrell 2714a9603a14SPatrick Farrell Input Parameters: 2715a9603a14SPatrick Farrell + tao - the Tao context 2716a9603a14SPatrick Farrell - M - gradient norm 2717a9603a14SPatrick Farrell 2718a9603a14SPatrick Farrell Level: beginner 2719a9603a14SPatrick Farrell 2720db781477SPatrick Sanan .seealso: `TaoGetGradientNorm()`, `TaoGradientNorm()` 2721a9603a14SPatrick Farrell @*/ 2722a9603a14SPatrick Farrell PetscErrorCode TaoSetGradientNorm(Tao tao, Mat M) 2723a9603a14SPatrick Farrell { 2724a9603a14SPatrick Farrell PetscFunctionBegin; 2725a9603a14SPatrick Farrell PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 272694511df7SStefano Zampini PetscValidHeaderSpecific(M,MAT_CLASSID,2); 27279566063dSJacob Faibussowitsch PetscCall(PetscObjectReference((PetscObject)M)); 27289566063dSJacob Faibussowitsch PetscCall(MatDestroy(&tao->gradient_norm)); 27299566063dSJacob Faibussowitsch PetscCall(VecDestroy(&tao->gradient_norm_tmp)); 2730a9603a14SPatrick Farrell tao->gradient_norm = M; 27319566063dSJacob Faibussowitsch PetscCall(MatCreateVecs(M, NULL, &tao->gradient_norm_tmp)); 2732a9603a14SPatrick Farrell PetscFunctionReturn(0); 2733a9603a14SPatrick Farrell } 2734a9603a14SPatrick Farrell 2735a9603a14SPatrick Farrell /*@ 2736a9603a14SPatrick Farrell TaoGetGradientNorm - Returns the matrix used to define the inner product for measuring the size of the gradient. 2737a9603a14SPatrick Farrell 2738a9603a14SPatrick Farrell Not Collective 2739a9603a14SPatrick Farrell 2740a9603a14SPatrick Farrell Input Parameter: 2741a9603a14SPatrick Farrell . tao - Tao context 2742a9603a14SPatrick Farrell 2743a9603a14SPatrick Farrell Output Parameter: 2744a9603a14SPatrick Farrell . M - gradient norm 2745a9603a14SPatrick Farrell 2746a9603a14SPatrick Farrell Level: beginner 2747a9603a14SPatrick Farrell 2748db781477SPatrick Sanan .seealso: `TaoSetGradientNorm()`, `TaoGradientNorm()` 2749a9603a14SPatrick Farrell @*/ 2750a9603a14SPatrick Farrell PetscErrorCode TaoGetGradientNorm(Tao tao, Mat *M) 2751a9603a14SPatrick Farrell { 2752a9603a14SPatrick Farrell PetscFunctionBegin; 2753a9603a14SPatrick Farrell PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 275494511df7SStefano Zampini PetscValidPointer(M,2); 2755a9603a14SPatrick Farrell *M = tao->gradient_norm; 2756a9603a14SPatrick Farrell PetscFunctionReturn(0); 2757a9603a14SPatrick Farrell } 2758a9603a14SPatrick Farrell 275905aca2cdSSatish Balay /*@C 2760a9603a14SPatrick Farrell TaoGradientNorm - Compute the norm with respect to the inner product the user has set. 2761a9603a14SPatrick Farrell 2762a9603a14SPatrick Farrell Collective on tao 2763a9603a14SPatrick Farrell 2764d8d19677SJose E. Roman Input Parameters: 27652653c764SPierre Jolivet + tao - the Tao context 2766a9603a14SPatrick Farrell . gradient - the gradient to be computed 27672653c764SPierre Jolivet - norm - the norm type 2768a9603a14SPatrick Farrell 2769a9603a14SPatrick Farrell Output Parameter: 2770a9603a14SPatrick Farrell . gnorm - the gradient norm 2771a9603a14SPatrick Farrell 2772a9603a14SPatrick Farrell Level: developer 2773a9603a14SPatrick Farrell 2774db781477SPatrick Sanan .seealso: `TaoSetGradientNorm()`, `TaoGetGradientNorm()` 2775a9603a14SPatrick Farrell @*/ 2776a9603a14SPatrick Farrell PetscErrorCode TaoGradientNorm(Tao tao, Vec gradient, NormType type, PetscReal *gnorm) 2777a9603a14SPatrick Farrell { 2778a9603a14SPatrick Farrell PetscFunctionBegin; 27798854b543SStefano Zampini PetscValidHeaderSpecific(tao,TAO_CLASSID,1); 27808854b543SStefano Zampini PetscValidHeaderSpecific(gradient,VEC_CLASSID,2); 27818854b543SStefano Zampini PetscValidLogicalCollectiveEnum(tao,type,3); 278294511df7SStefano Zampini PetscValidRealPointer(gnorm,4); 2783a9603a14SPatrick Farrell if (tao->gradient_norm) { 2784680e2bc7SPatrick Farrell PetscScalar gnorms; 2785680e2bc7SPatrick Farrell 27863c859ba3SBarry 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."); 27879566063dSJacob Faibussowitsch PetscCall(MatMult(tao->gradient_norm, gradient, tao->gradient_norm_tmp)); 27889566063dSJacob Faibussowitsch PetscCall(VecDot(gradient, tao->gradient_norm_tmp, &gnorms)); 27897bb79937SPatrick Farrell *gnorm = PetscRealPart(PetscSqrtScalar(gnorms)); 2790a9603a14SPatrick Farrell } else { 27919566063dSJacob Faibussowitsch PetscCall(VecNorm(gradient, type, gnorm)); 2792a9603a14SPatrick Farrell } 2793a9603a14SPatrick Farrell PetscFunctionReturn(0); 2794a9603a14SPatrick Farrell } 2795a9603a14SPatrick Farrell 2796e882e171SHong Zhang /*@C 2797e882e171SHong Zhang TaoMonitorDrawCtxCreate - Creates the monitor context for TaoMonitorDrawCtx 2798a9603a14SPatrick Farrell 2799e882e171SHong Zhang Collective on Tao 2800e882e171SHong Zhang 2801e882e171SHong Zhang Output Patameter: 2802e882e171SHong Zhang . ctx - the monitor context 2803e882e171SHong Zhang 2804e882e171SHong Zhang Options Database: 2805e882e171SHong Zhang . -tao_draw_solution_initial - show initial guess as well as current solution 2806e882e171SHong Zhang 2807e882e171SHong Zhang Level: intermediate 2808e882e171SHong Zhang 2809db781477SPatrick Sanan .seealso: `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawCtx()` 2810e882e171SHong Zhang @*/ 2811e882e171SHong Zhang PetscErrorCode TaoMonitorDrawCtxCreate(MPI_Comm comm,const char host[],const char label[],int x,int y,int m,int n,PetscInt howoften,TaoMonitorDrawCtx *ctx) 2812e882e171SHong Zhang { 2813e882e171SHong Zhang PetscFunctionBegin; 28149566063dSJacob Faibussowitsch PetscCall(PetscNew(ctx)); 28159566063dSJacob Faibussowitsch PetscCall(PetscViewerDrawOpen(comm,host,label,x,y,m,n,&(*ctx)->viewer)); 28169566063dSJacob Faibussowitsch PetscCall(PetscViewerSetFromOptions((*ctx)->viewer)); 2817e882e171SHong Zhang (*ctx)->howoften = howoften; 2818e882e171SHong Zhang PetscFunctionReturn(0); 2819e882e171SHong Zhang } 2820e882e171SHong Zhang 2821e882e171SHong Zhang /*@C 2822e882e171SHong Zhang TaoMonitorDrawCtxDestroy - Destroys the monitor context for TaoMonitorDrawSolution() 2823e882e171SHong Zhang 2824e882e171SHong Zhang Collective on Tao 2825e882e171SHong Zhang 2826e882e171SHong Zhang Input Parameters: 2827e882e171SHong Zhang . ctx - the monitor context 2828e882e171SHong Zhang 2829e882e171SHong Zhang Level: intermediate 2830e882e171SHong Zhang 2831db781477SPatrick Sanan .seealso: `TaoMonitorSet()`, `TaoMonitorDefault()`, `VecView()`, `TaoMonitorDrawSolution()` 2832e882e171SHong Zhang @*/ 2833e882e171SHong Zhang PetscErrorCode TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *ictx) 2834e882e171SHong Zhang { 2835e882e171SHong Zhang PetscFunctionBegin; 28369566063dSJacob Faibussowitsch PetscCall(PetscViewerDestroy(&(*ictx)->viewer)); 28379566063dSJacob Faibussowitsch PetscCall(PetscFree(*ictx)); 2838e882e171SHong Zhang PetscFunctionReturn(0); 2839e882e171SHong Zhang } 2840