16524c165SJacob Faibussowitsch #ifndef PETSCTAO_H 23028902dSLisandro Dalcin #define PETSCTAO_H 321ec2d5cSBarry Smith 4aad13602SShrirang Abhyankar #include <petscsnes.h> 521ec2d5cSBarry Smith 6ac09b921SBarry Smith /* SUBMANSEC = Tao */ 7ac09b921SBarry Smith 8b54963c9SStefano Zampini PETSC_EXTERN PetscErrorCode VecFischer(Vec, Vec, Vec, Vec, Vec); 9b54963c9SStefano Zampini PETSC_EXTERN PetscErrorCode VecSFischer(Vec, Vec, Vec, Vec, PetscReal, Vec); 10b54963c9SStefano Zampini PETSC_EXTERN PetscErrorCode MatDFischer(Mat, Vec, Vec, Vec, Vec, Vec, Vec, Vec, Vec); 11b54963c9SStefano Zampini PETSC_EXTERN PetscErrorCode MatDSFischer(Mat, Vec, Vec, Vec, Vec, PetscReal, Vec, Vec, Vec, Vec, Vec); 128370d7cdSHansol Suh PETSC_EXTERN PetscErrorCode TaoSoftThreshold(Vec, PetscReal, PetscReal, Vec); 1321ec2d5cSBarry Smith 1421ec2d5cSBarry Smith /*E 1565ba42b6SBarry Smith TaoSubsetType - Type representing the way Tao handles active sets 1621ec2d5cSBarry Smith 1787497f52SBarry Smith + `TAO_SUBSET_SUBVEC` - Tao uses `MatCreateSubMatrix()` and `VecGetSubVector()` 1865ba42b6SBarry Smith . `TAO_SUBSET_MASK` - Matrices are zeroed out corresponding to active set entries 1965ba42b6SBarry Smith - `TAO_SUBSET_MATRIXFREE` - Same as `TAO_SUBSET_MASK` but it can be applied to matrix-free operators 2021ec2d5cSBarry Smith 2121ec2d5cSBarry Smith Options database keys: 2265ba42b6SBarry Smith . -different_hessian - Tao will use a copy of the hessian operator for masking. By default 2365ba42b6SBarry Smith Tao will directly alter the hessian operator. 241eb8069cSJason Sarich Level: intermediate 2521ec2d5cSBarry Smith 2621ec2d5cSBarry Smith E*/ 279371c9d4SSatish Balay typedef enum { 289371c9d4SSatish Balay TAO_SUBSET_SUBVEC, 299371c9d4SSatish Balay TAO_SUBSET_MASK, 309371c9d4SSatish Balay TAO_SUBSET_MATRIXFREE 319371c9d4SSatish Balay } TaoSubsetType; 3221ec2d5cSBarry Smith PETSC_EXTERN const char *const TaoSubsetTypes[]; 3365ba42b6SBarry Smith 341eb8069cSJason Sarich /*S 351eb8069cSJason Sarich Tao - Abstract PETSc object that manages nonlinear optimization solves 361eb8069cSJason Sarich 371eb8069cSJason Sarich Level: advanced 381eb8069cSJason Sarich 39*a1cb98faSBarry Smith .seealso: `TaoCreate()`, `TaoDestroy()`, `TaoSetType()`, `TaoType` 401eb8069cSJason Sarich S*/ 4165ba42b6SBarry Smith typedef struct _p_Tao *Tao; 4221ec2d5cSBarry Smith 436285c0a3SHansol Suh /*E 4487497f52SBarry Smith TaoADMMUpdateType - Determine spectral penalty update routine for Lagrange augmented term for `TAOADMM`. 456285c0a3SHansol Suh 466285c0a3SHansol Suh Level: advanced 476285c0a3SHansol Suh 48*a1cb98faSBarry Smith .seealso: `Tao`, `TAOADMM`, `TaoADMMSetUpdateType()` 496285c0a3SHansol Suh E*/ 509371c9d4SSatish Balay typedef enum { 519371c9d4SSatish Balay TAO_ADMM_UPDATE_BASIC, 529371c9d4SSatish Balay TAO_ADMM_UPDATE_ADAPTIVE, 539371c9d4SSatish Balay TAO_ADMM_UPDATE_ADAPTIVE_RELAXED 549371c9d4SSatish Balay } TaoADMMUpdateType; 556285c0a3SHansol Suh PETSC_EXTERN const char *const TaoADMMUpdateTypes[]; 5665ba42b6SBarry Smith 576285c0a3SHansol Suh /*MC 586285c0a3SHansol Suh TAO_ADMM_UPDATE_BASIC - Use same spectral penalty set at the beginning. No update 596285c0a3SHansol Suh 606285c0a3SHansol Suh Level: advanced 616285c0a3SHansol Suh 6287497f52SBarry Smith Note: 6387497f52SBarry Smith Most basic implementation. Generally slower than adaptive or adaptive relaxed version. 646285c0a3SHansol Suh 65db781477SPatrick Sanan .seealso: `TaoADMMSetUpdateType()`, `TAO_ADMM_UPDATE_ADAPTIVE`, `TAO_ADMM_UPDATE_ADAPTIVE_RELAXED` 666285c0a3SHansol Suh M*/ 676285c0a3SHansol Suh 686285c0a3SHansol Suh /*MC 696285c0a3SHansol Suh TAO_ADMM_UPDATE_ADAPTIVE - Adaptively update spectral penalty 706285c0a3SHansol Suh 716285c0a3SHansol Suh Level: advanced 726285c0a3SHansol Suh 7387497f52SBarry Smith Note: 7487497f52SBarry Smith Adaptively updates spectral penalty by using both steepest descent and minimum gradient. 756285c0a3SHansol Suh 76db781477SPatrick Sanan .seealso: `TaoADMMSetUpdateType()`, `TAO_ADMM_UPDATE_BASIC`, `TAO_ADMM_UPDATE_ADAPTIVE_RELAXED` 776285c0a3SHansol Suh M*/ 786285c0a3SHansol Suh 796285c0a3SHansol Suh /*MC 806285c0a3SHansol Suh ADMM_UPDATE_ADAPTIVE_RELAXED - Adaptively update spectral penalty, and relaxes parameter update 816285c0a3SHansol Suh 826285c0a3SHansol Suh Level: advanced 836285c0a3SHansol Suh 8487497f52SBarry Smith Note: 8587497f52SBarry Smith With adaptive spectral penalty update, it also relaxes x vector update by a factor. 866285c0a3SHansol Suh 87db781477SPatrick Sanan .seealso: `TaoADMMSetUpdateType()`, `TAO_ADMM_UPDATE_BASIC`, `TAO_ADMM_UPDATE_ADAPTIVE` 886285c0a3SHansol Suh M*/ 896285c0a3SHansol Suh 906285c0a3SHansol Suh /*E 916285c0a3SHansol Suh TaoADMMRegularizerType - Determine regularizer routine - either user provided or soft threshold 926285c0a3SHansol Suh 936285c0a3SHansol Suh Level: advanced 946285c0a3SHansol Suh 95*a1cb98faSBarry Smith .seealso: `Tao`, `TAOADMM`, `TaoADMMSetRegularizerType()` 966285c0a3SHansol Suh E*/ 979371c9d4SSatish Balay typedef enum { 989371c9d4SSatish Balay TAO_ADMM_REGULARIZER_USER, 999371c9d4SSatish Balay TAO_ADMM_REGULARIZER_SOFT_THRESH 1009371c9d4SSatish Balay } TaoADMMRegularizerType; 1016285c0a3SHansol Suh PETSC_EXTERN const char *const TaoADMMRegularizerTypes[]; 10265ba42b6SBarry Smith 1036285c0a3SHansol Suh /*MC 10487497f52SBarry Smith TAO_ADMM_REGULARIZER_USER - User provided routines for regularizer part of `TAOADMM` 1056285c0a3SHansol Suh 1066285c0a3SHansol Suh Level: advanced 1076285c0a3SHansol Suh 10887497f52SBarry Smith Note: 10987497f52SBarry Smith User needs to provided appropriate routines and type for regularizer solver 1106285c0a3SHansol Suh 111db781477SPatrick Sanan .seealso: `TaoADMMSetRegularizerType()`, `TAO_ADMM_REGULARIZER_SOFT_THRESH` 1126285c0a3SHansol Suh M*/ 1136285c0a3SHansol Suh 1146285c0a3SHansol Suh /*MC 11587497f52SBarry Smith TAO_ADMM_REGULARIZER_SOFT_THRESH - Soft threshold to solve regularizer part of `TAOADMM` 1166285c0a3SHansol Suh 1176285c0a3SHansol Suh Level: advanced 1186285c0a3SHansol Suh 11987497f52SBarry Smith Note: 12087497f52SBarry Smith Utilizes built-in SoftThreshold routines 1216285c0a3SHansol Suh 122db781477SPatrick Sanan .seealso: `TaoSoftThreshold()`, `TaoADMMSetRegularizerObjectiveAndGradientRoutine()`, 123db781477SPatrick Sanan `TaoADMMSetRegularizerHessianRoutine()`, `TaoADMMSetRegularizerType()`, `TAO_ADMM_REGULARIZER_USER` 1246285c0a3SHansol Suh M*/ 1256285c0a3SHansol Suh 126661095bbSAlp Dener /*E 12787497f52SBarry Smith TaoALMMType - Determine the augmented Lagrangian formulation used in the `TAOALMM` subproblem. 128661095bbSAlp Dener 129*a1cb98faSBarry Smith Values: 130*a1cb98faSBarry Smith + `TAO_ALMM_CLASSIC` - classic augmented Lagrangian definition including slack variables for inequality constraints 131*a1cb98faSBarry Smith - `TAO_ALMM_PHR` - Powell-Hestenes-Rockafellar formulation without slack variables, uses pointwise min() for inequalities 132661095bbSAlp Dener 133661095bbSAlp Dener Level: advanced 134661095bbSAlp Dener 135*a1cb98faSBarry Smith .seealso: `Tao`, `TAOALMM`, `TaoALMMSetType()`, `TaoALMMGetType()` 136661095bbSAlp Dener E*/ 1379371c9d4SSatish Balay typedef enum { 1389371c9d4SSatish Balay TAO_ALMM_CLASSIC, 1399371c9d4SSatish Balay TAO_ALMM_PHR 1409371c9d4SSatish Balay } TaoALMMType; 141661095bbSAlp Dener PETSC_EXTERN const char *const TaoALMMTypes[]; 142661095bbSAlp Dener 1431eb8069cSJason Sarich /*J 14487497f52SBarry Smith TaoType - String with the name of a `Tao` method 1451eb8069cSJason Sarich 1461eb8069cSJason Sarich Level: beginner 1471eb8069cSJason Sarich 148*a1cb98faSBarry Smith .seealso: `Tao`, `TaoCreate()`, `TaoSetType()` 1491eb8069cSJason Sarich J*/ 150b625d6c7SJed Brown typedef const char *TaoType; 15158417fe7SBarry Smith #define TAOLMVM "lmvm" 15258417fe7SBarry Smith #define TAONLS "nls" 15358417fe7SBarry Smith #define TAONTR "ntr" 15458417fe7SBarry Smith #define TAONTL "ntl" 15558417fe7SBarry Smith #define TAOCG "cg" 15658417fe7SBarry Smith #define TAOTRON "tron" 15758417fe7SBarry Smith #define TAOOWLQN "owlqn" 15858417fe7SBarry Smith #define TAOBMRM "bmrm" 15958417fe7SBarry Smith #define TAOBLMVM "blmvm" 1606b591159SAlp Dener #define TAOBQNLS "bqnls" 161ac9112b8SAlp Dener #define TAOBNCG "bncg" 162eb910715SAlp Dener #define TAOBNLS "bnls" 163fed79b8eSAlp Dener #define TAOBNTR "bntr" 164c14b763aSAlp Dener #define TAOBNTL "bntl" 165e0ed867bSAlp Dener #define TAOBQNKLS "bqnkls" 166e0ed867bSAlp Dener #define TAOBQNKTR "bqnktr" 167e0ed867bSAlp Dener #define TAOBQNKTL "bqnktl" 16858417fe7SBarry Smith #define TAOBQPIP "bqpip" 16958417fe7SBarry Smith #define TAOGPCG "gpcg" 17058417fe7SBarry Smith #define TAONM "nm" 17158417fe7SBarry Smith #define TAOPOUNDERS "pounders" 172737f463aSAlp Dener #define TAOBRGN "brgn" 17358417fe7SBarry Smith #define TAOLCL "lcl" 17458417fe7SBarry Smith #define TAOSSILS "ssils" 17558417fe7SBarry Smith #define TAOSSFLS "ssfls" 17658417fe7SBarry Smith #define TAOASILS "asils" 17758417fe7SBarry Smith #define TAOASFLS "asfls" 17858417fe7SBarry Smith #define TAOIPM "ipm" 179aad13602SShrirang Abhyankar #define TAOPDIPM "pdipm" 18083a0a5c3SToby Isaac #define TAOSHELL "shell" 1816285c0a3SHansol Suh #define TAOADMM "admm" 182661095bbSAlp Dener #define TAOALMM "almm" 183a82e8c82SStefano Zampini #define TAOPYTHON "python" 18458417fe7SBarry Smith 185441846f8SBarry Smith PETSC_EXTERN PetscClassId TAO_CLASSID; 186441846f8SBarry Smith PETSC_EXTERN PetscFunctionList TaoList; 18721ec2d5cSBarry Smith 188a35d58b8SBarry Smith /*E 18965ba42b6SBarry Smith TaoConvergedReason - reason a Tao method was said to have converged or diverged 190a35d58b8SBarry Smith 191a35d58b8SBarry Smith Level: beginner 192a35d58b8SBarry Smith 193a35d58b8SBarry Smith The two most common reasons for divergence are 194a35d58b8SBarry Smith $ 1) an incorrectly coded or computed gradient or Hessian 195a35d58b8SBarry Smith $ 2) failure or lack of convergence in the linear system (in this case we recommend 196a35d58b8SBarry Smith $ testing with -pc_type lu to eliminate the linear solver as the cause of the problem). 197a35d58b8SBarry Smith 19895452b02SPatrick Sanan Developer Notes: 19987497f52SBarry Smith This must match petsc/finclude/petsctao.h 200a35d58b8SBarry Smith 20187497f52SBarry Smith The string versions of these are in `TAOConvergedReasons`, if you change any value here you must 202a35d58b8SBarry Smith also adjust that array. 203a35d58b8SBarry Smith 20465ba42b6SBarry Smith .seealso: `Tao`, `TaoSolve()`, `TaoGetConvergedReason()`, `KSPConvergedReason`, `SNESConvergedReason`, `TSConvergedReason` 205a35d58b8SBarry Smith E*/ 20621ec2d5cSBarry Smith typedef enum { /* converged */ 20721ec2d5cSBarry Smith TAO_CONVERGED_GATOL = 3, /* ||g(X)|| < gatol */ 20821ec2d5cSBarry Smith TAO_CONVERGED_GRTOL = 4, /* ||g(X)|| / f(X) < grtol */ 20921ec2d5cSBarry Smith TAO_CONVERGED_GTTOL = 5, /* ||g(X)|| / ||g(X0)|| < gttol */ 21021ec2d5cSBarry Smith TAO_CONVERGED_STEPTOL = 6, /* step size small */ 21121ec2d5cSBarry Smith TAO_CONVERGED_MINF = 7, /* F < F_min */ 21221ec2d5cSBarry Smith TAO_CONVERGED_USER = 8, /* User defined */ 21321ec2d5cSBarry Smith /* diverged */ 21421ec2d5cSBarry Smith TAO_DIVERGED_MAXITS = -2, 21521ec2d5cSBarry Smith TAO_DIVERGED_NAN = -4, 21621ec2d5cSBarry Smith TAO_DIVERGED_MAXFCN = -5, 21721ec2d5cSBarry Smith TAO_DIVERGED_LS_FAILURE = -6, 21821ec2d5cSBarry Smith TAO_DIVERGED_TR_REDUCTION = -7, 21921ec2d5cSBarry Smith TAO_DIVERGED_USER = -8, /* User defined */ 22021ec2d5cSBarry Smith /* keep going */ 2219371c9d4SSatish Balay TAO_CONTINUE_ITERATING = 0 2229371c9d4SSatish Balay } TaoConvergedReason; 22321ec2d5cSBarry Smith 224e4cb33bbSBarry Smith PETSC_EXTERN const char **TaoConvergedReasons; 22521ec2d5cSBarry Smith 22621ec2d5cSBarry Smith PETSC_EXTERN PetscErrorCode TaoInitializePackage(void); 22721ec2d5cSBarry Smith PETSC_EXTERN PetscErrorCode TaoFinalizePackage(void); 228441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoCreate(MPI_Comm, Tao *); 229441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetFromOptions(Tao); 230441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetUp(Tao); 231b625d6c7SJed Brown PETSC_EXTERN PetscErrorCode TaoSetType(Tao, TaoType); 232b625d6c7SJed Brown PETSC_EXTERN PetscErrorCode TaoGetType(Tao, TaoType *); 233441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetApplicationContext(Tao, void *); 234441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetApplicationContext(Tao, void *); 235441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDestroy(Tao *); 23621ec2d5cSBarry Smith 237441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetOptionsPrefix(Tao, const char[]); 238441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoView(Tao, PetscViewer); 239fe2efc57SMark PETSC_EXTERN PetscErrorCode TaoViewFromOptions(Tao, PetscObject, const char[]); 24021ec2d5cSBarry Smith 241441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSolve(Tao); 24221ec2d5cSBarry Smith 243441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoRegister(const char[], PetscErrorCode (*)(Tao)); 244441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoRegisterDestroy(void); 24521ec2d5cSBarry Smith 246e4cb33bbSBarry Smith PETSC_EXTERN PetscErrorCode TaoGetConvergedReason(Tao, TaoConvergedReason *); 247e4cb33bbSBarry Smith PETSC_EXTERN PetscErrorCode TaoGetSolutionStatus(Tao, PetscInt *, PetscReal *, PetscReal *, PetscReal *, PetscReal *, TaoConvergedReason *); 248e4cb33bbSBarry Smith PETSC_EXTERN PetscErrorCode TaoSetConvergedReason(Tao, TaoConvergedReason); 249a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetSolution(Tao, Vec); 250a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetSolution(Tao, Vec *); 251d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoSetSolution() (since version 3.17)") static inline PetscErrorCode TaoSetInitialVector(Tao t, Vec v) 252d71ae5a4SJacob Faibussowitsch { 2539371c9d4SSatish Balay return TaoSetSolution(t, v); 2549371c9d4SSatish Balay } 255d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoGetSolution() (since version 3.17)") static inline PetscErrorCode TaoGetInitialVector(Tao t, Vec *v) 256d71ae5a4SJacob Faibussowitsch { 2579371c9d4SSatish Balay return TaoGetSolution(t, v); 2589371c9d4SSatish Balay } 259a82e8c82SStefano Zampini 260a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetObjective(Tao, PetscErrorCode (*)(Tao, Vec, PetscReal *, void *), void *); 261a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetObjective(Tao, PetscErrorCode (**)(Tao, Vec, PetscReal *, void *), void **); 262a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetGradient(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *); 263a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetGradient(Tao, Vec *, PetscErrorCode (**)(Tao, Vec, Vec, void *), void **); 264a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetObjectiveAndGradient(Tao, Vec, PetscErrorCode (*)(Tao, Vec, PetscReal *, Vec, void *), void *); 265a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetObjectiveAndGradient(Tao, Vec *, PetscErrorCode (**)(Tao, Vec, PetscReal *, Vec, void *), void **); 266a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetHessian(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 267a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetHessian(Tao, Mat *, Mat *, PetscErrorCode (**)(Tao, Vec, Mat, Mat, void *), void **); 268d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoSetObjective() (since version 3.17)") static inline PetscErrorCode TaoSetObjectiveRoutine(Tao t, PetscErrorCode (*f)(Tao, Vec, PetscReal *, void *), void *c) 269d71ae5a4SJacob Faibussowitsch { 2709371c9d4SSatish Balay return TaoSetObjective(t, f, c); 2719371c9d4SSatish Balay } 272d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoGetGradient() (since version 3.17)") static inline PetscErrorCode TaoGetGradientVector(Tao t, Vec *v) 273d71ae5a4SJacob Faibussowitsch { 2749371c9d4SSatish Balay return TaoGetGradient(t, v, NULL, NULL); 2759371c9d4SSatish Balay } 276d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoSetGradient() (since version 3.17)") static inline PetscErrorCode TaoSetGradientRoutine(Tao t, PetscErrorCode (*f)(Tao, Vec, Vec, void *), void *c) 277d71ae5a4SJacob Faibussowitsch { 2789371c9d4SSatish Balay return TaoSetGradient(t, NULL, f, c); 2799371c9d4SSatish Balay } 280d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoSetObjectiveAndGradient() (since version 3.17)") static inline PetscErrorCode TaoSetObjectiveAndGradientRoutine(Tao t, PetscErrorCode (*f)(Tao, Vec, PetscReal *, Vec, void *), void *c) 281d71ae5a4SJacob Faibussowitsch { 2829371c9d4SSatish Balay return TaoSetObjectiveAndGradient(t, NULL, f, c); 2839371c9d4SSatish Balay } 284d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoSetHessian() (since version 3.17)") static inline PetscErrorCode TaoSetHessianRoutine(Tao t, Mat H, Mat P, PetscErrorCode (*f)(Tao, Vec, Mat, Mat, void *), void *c) 285d71ae5a4SJacob Faibussowitsch { 2869371c9d4SSatish Balay return TaoSetHessian(t, H, P, f, c); 2879371c9d4SSatish Balay } 288a82e8c82SStefano Zampini 289a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoSetGradientNorm(Tao, Mat); 290a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoGetGradientNorm(Tao, Mat *); 291414d97d3SAlp Dener PETSC_EXTERN PetscErrorCode TaoSetLMVMMatrix(Tao, Mat); 292f5766c09SAlp Dener PETSC_EXTERN PetscErrorCode TaoGetLMVMMatrix(Tao, Mat *); 293414d97d3SAlp Dener PETSC_EXTERN PetscErrorCode TaoSetRecycleHistory(Tao, PetscBool); 294414d97d3SAlp Dener PETSC_EXTERN PetscErrorCode TaoGetRecycleHistory(Tao, PetscBool *); 295a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoLMVMSetH0(Tao, Mat); 296a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoLMVMGetH0(Tao, Mat *); 297a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoLMVMGetH0KSP(Tao, KSP *); 298b39c12a9SAlp Dener PETSC_EXTERN PetscErrorCode TaoLMVMRecycle(Tao, PetscBool); 2994a48860cSAlp Dener PETSC_EXTERN PetscErrorCode TaoSetResidualRoutine(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *); 300737f463aSAlp Dener PETSC_EXTERN PetscErrorCode TaoSetResidualWeights(Tao, Vec, PetscInt, PetscInt *, PetscInt *, PetscReal *); 301441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetConstraintsRoutine(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *); 302441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetInequalityConstraintsRoutine(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *); 303441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetEqualityConstraintsRoutine(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *); 3044ffbe8acSAlp Dener PETSC_EXTERN PetscErrorCode TaoSetJacobianResidualRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 305ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 306ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianStateRoutine(Tao, Mat, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, Mat, void *), void *); 30794ab13aaSBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianDesignRoutine(Tao, Mat, PetscErrorCode (*)(Tao, Vec, Mat, void *), void *); 308ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianInequalityRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 309ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianEqualityRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 31021ec2d5cSBarry Smith 311a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoPythonSetType(Tao, const char[]); 312ebead697SStefano Zampini PETSC_EXTERN PetscErrorCode TaoPythonGetType(Tao, const char *[]); 313a82e8c82SStefano Zampini 31483a0a5c3SToby Isaac PETSC_EXTERN PetscErrorCode TaoShellSetSolve(Tao, PetscErrorCode (*)(Tao)); 31583a0a5c3SToby Isaac PETSC_EXTERN PetscErrorCode TaoShellSetContext(Tao, void *); 3163ec1f749SStefano Zampini PETSC_EXTERN PetscErrorCode TaoShellGetContext(Tao, void *); 31783a0a5c3SToby Isaac 318d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoSetResidualRoutine() (since version 3.11)") static inline PetscErrorCode TaoSetSeparableObjectiveRoutine(Tao tao, Vec res, PetscErrorCode (*func)(Tao, Vec, Vec, void *), void *ctx) 319d71ae5a4SJacob Faibussowitsch { 3209371c9d4SSatish Balay return TaoSetResidualRoutine(tao, res, func, ctx); 3219371c9d4SSatish Balay } 322d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoSetResidualWeights() (since version 3.11)") static inline PetscErrorCode TaoSetSeparableObjectiveWeights(Tao tao, Vec sigma_v, PetscInt n, PetscInt *rows, PetscInt *cols, PetscReal *vals) 323d71ae5a4SJacob Faibussowitsch { 3249371c9d4SSatish Balay return TaoSetResidualWeights(tao, sigma_v, n, rows, cols, vals); 3259371c9d4SSatish Balay } 326737f463aSAlp Dener 327441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetStateDesignIS(Tao, IS, IS); 32821ec2d5cSBarry Smith 329441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeObjective(Tao, Vec, PetscReal *); 3304a48860cSAlp Dener PETSC_EXTERN PetscErrorCode TaoComputeResidual(Tao, Vec, Vec); 331412cdd55SHong Zhang PETSC_EXTERN PetscErrorCode TaoTestGradient(Tao, Vec, Vec); 332441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeGradient(Tao, Vec, Vec); 333441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeObjectiveAndGradient(Tao, Vec, PetscReal *, Vec); 334441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeConstraints(Tao, Vec, Vec); 335441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeInequalityConstraints(Tao, Vec, Vec); 336441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeEqualityConstraints(Tao, Vec, Vec); 337441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultComputeGradient(Tao, Vec, Vec, void *); 338441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoIsObjectiveDefined(Tao, PetscBool *); 339441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoIsGradientDefined(Tao, PetscBool *); 340441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoIsObjectiveAndGradientDefined(Tao, PetscBool *); 34121ec2d5cSBarry Smith 342d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoComputeResidual() (since version 3.11)") static inline PetscErrorCode TaoComputeSeparableObjective(Tao tao, Vec X, Vec F) 343d71ae5a4SJacob Faibussowitsch { 3449371c9d4SSatish Balay return TaoComputeResidual(tao, X, F); 3459371c9d4SSatish Balay } 3464a48860cSAlp Dener 34709baa881SHong Zhang PETSC_EXTERN PetscErrorCode TaoTestHessian(Tao); 348ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeHessian(Tao, Vec, Mat, Mat); 349737f463aSAlp Dener PETSC_EXTERN PetscErrorCode TaoComputeResidualJacobian(Tao, Vec, Mat, Mat); 350ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobian(Tao, Vec, Mat, Mat); 351ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobianState(Tao, Vec, Mat, Mat, Mat); 352ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobianEquality(Tao, Vec, Mat, Mat); 353ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobianInequality(Tao, Vec, Mat, Mat); 35494ab13aaSBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobianDesign(Tao, Vec, Mat); 35521ec2d5cSBarry Smith 356ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultComputeHessian(Tao, Vec, Mat, Mat, void *); 357ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultComputeHessianColor(Tao, Vec, Mat, Mat, void *); 358f4c1ad5cSStefano Zampini PETSC_EXTERN PetscErrorCode TaoDefaultComputeHessianMFFD(Tao, Vec, Mat, Mat, void *); 359441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeDualVariables(Tao, Vec, Vec); 360441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetVariableBounds(Tao, Vec, Vec); 361441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetVariableBounds(Tao, Vec *, Vec *); 362441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetDualVariables(Tao, Vec *, Vec *); 363441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetInequalityBounds(Tao, Vec, Vec); 364441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetInequalityBounds(Tao, Vec *, Vec *); 365441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetVariableBoundsRoutine(Tao, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *); 366441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeVariableBounds(Tao); 36721ec2d5cSBarry Smith 368e52336cbSBarry Smith PETSC_EXTERN PetscErrorCode TaoGetTolerances(Tao, PetscReal *, PetscReal *, PetscReal *); 369e52336cbSBarry Smith PETSC_EXTERN PetscErrorCode TaoSetTolerances(Tao, PetscReal, PetscReal, PetscReal); 370441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetConstraintTolerances(Tao, PetscReal *, PetscReal *); 371441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetConstraintTolerances(Tao, PetscReal, PetscReal); 372441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetFunctionLowerBound(Tao, PetscReal); 373441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetInitialTrustRegionRadius(Tao, PetscReal); 374441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetMaximumIterations(Tao, PetscInt); 375441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao, PetscInt); 376441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetFunctionLowerBound(Tao, PetscReal *); 377441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetInitialTrustRegionRadius(Tao, PetscReal *); 378441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao, PetscReal *); 379441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetMaximumIterations(Tao, PetscInt *); 380770232b9SCe Qin PETSC_EXTERN PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao, PetscInt *); 381441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao, PetscInt *); 3828931d482SJason Sarich PETSC_EXTERN PetscErrorCode TaoGetIterationNumber(Tao, PetscInt *); 3838931d482SJason Sarich PETSC_EXTERN PetscErrorCode TaoSetIterationNumber(Tao, PetscInt); 3848931d482SJason Sarich PETSC_EXTERN PetscErrorCode TaoGetTotalIterationNumber(Tao, PetscInt *); 3858931d482SJason Sarich PETSC_EXTERN PetscErrorCode TaoSetTotalIterationNumber(Tao, PetscInt); 38679f5d8caSBarry Smith PETSC_EXTERN PetscErrorCode TaoGetResidualNorm(Tao, PetscReal *); 3878931d482SJason Sarich 388b54963c9SStefano Zampini PETSC_EXTERN PetscErrorCode TaoAppendOptionsPrefix(Tao, const char[]); 389b54963c9SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetOptionsPrefix(Tao, const char *[]); 390441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoResetStatistics(Tao); 3918fcddce6SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetUpdate(Tao, PetscErrorCode (*)(Tao, PetscInt, void *), void *); 39221ec2d5cSBarry Smith 393441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetKSP(Tao, KSP *); 394025e9500SJason Sarich PETSC_EXTERN PetscErrorCode TaoGetLinearSolveIterations(Tao, PetscInt *); 3950f0abf79SStefano Zampini PETSC_EXTERN PetscErrorCode TaoKSPSetUseEW(Tao, PetscBool); 396235fd6e6SBarry Smith 397235fd6e6SBarry Smith #include <petsctaolinesearch.h> 398b54963c9SStefano Zampini 399441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetLineSearch(Tao, TaoLineSearch *); 40021ec2d5cSBarry Smith 401ae93cb3cSJason Sarich PETSC_EXTERN PetscErrorCode TaoSetConvergenceHistory(Tao, PetscReal *, PetscReal *, PetscReal *, PetscInt *, PetscInt, PetscBool); 402ae93cb3cSJason Sarich PETSC_EXTERN PetscErrorCode TaoGetConvergenceHistory(Tao, PetscReal **, PetscReal **, PetscReal **, PetscInt **, PetscInt *); 403441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetMonitor(Tao, PetscErrorCode (*)(Tao, void *), void *, PetscErrorCode (*)(void **)); 404441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoCancelMonitors(Tao); 40598ea980cSBarry Smith PETSC_EXTERN PetscErrorCode TaoMonitorDefault(Tao, void *); 406d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoMonitorDefault() (since version 3.9)") static inline PetscErrorCode TaoDefaultMonitor(Tao tao, void *ctx) 407d71ae5a4SJacob Faibussowitsch { 4089371c9d4SSatish Balay return TaoMonitorDefault(tao, ctx); 4099371c9d4SSatish Balay } 4108d5ead36SAlp Dener PETSC_EXTERN PetscErrorCode TaoDefaultGMonitor(Tao, void *); 411441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultSMonitor(Tao, void *); 412441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultCMonitor(Tao, void *); 413441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSolutionMonitor(Tao, void *); 414737f463aSAlp Dener PETSC_EXTERN PetscErrorCode TaoResidualMonitor(Tao, void *); 415441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGradientMonitor(Tao, void *); 416441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoStepDirectionMonitor(Tao, void *); 417441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDrawSolutionMonitor(Tao, void *); 418441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDrawStepMonitor(Tao, void *); 419441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDrawGradientMonitor(Tao, void *); 420441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoAddLineSearchCounts(Tao); 42121ec2d5cSBarry Smith 422441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultConvergenceTest(Tao, void *); 423441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetConvergenceTest(Tao, PetscErrorCode (*)(Tao, void *), void *); 42421ec2d5cSBarry Smith 425441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoLCLSetStateDesignIS(Tao, IS, IS); 4263ecd9318SAlp Dener PETSC_EXTERN PetscErrorCode TaoMonitor(Tao, PetscInt, PetscReal, PetscReal, PetscReal, PetscReal); 427e882e171SHong Zhang typedef struct _n_TaoMonitorDrawCtx *TaoMonitorDrawCtx; 428e882e171SHong Zhang PETSC_EXTERN PetscErrorCode TaoMonitorDrawCtxCreate(MPI_Comm, const char[], const char[], int, int, int, int, PetscInt, TaoMonitorDrawCtx *); 429e882e171SHong Zhang PETSC_EXTERN PetscErrorCode TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *); 430737f463aSAlp Dener 4318e85b1b3SXiang Huang PETSC_EXTERN PetscErrorCode TaoBRGNGetSubsolver(Tao, Tao *); 432a3c390cfSAlp Dener PETSC_EXTERN PetscErrorCode TaoBRGNSetRegularizerObjectiveAndGradientRoutine(Tao, PetscErrorCode (*)(Tao, Vec, PetscReal *, Vec, void *), void *); 433a3c390cfSAlp Dener PETSC_EXTERN PetscErrorCode TaoBRGNSetRegularizerHessianRoutine(Tao, Mat, PetscErrorCode (*)(Tao, Vec, Mat, void *), void *); 434a3c390cfSAlp Dener PETSC_EXTERN PetscErrorCode TaoBRGNSetRegularizerWeight(Tao, PetscReal); 4358ac80d48SXiang Huang PETSC_EXTERN PetscErrorCode TaoBRGNSetL1SmoothEpsilon(Tao, PetscReal); 4368e85b1b3SXiang Huang PETSC_EXTERN PetscErrorCode TaoBRGNSetDictionaryMatrix(Tao, Mat); 437cd1c4666STristan Konolige PETSC_EXTERN PetscErrorCode TaoBRGNGetDampingVector(Tao, Vec *); 4386285c0a3SHansol Suh 4396285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMGetMisfitSubsolver(Tao, Tao *); 4406285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMGetRegularizationSubsolver(Tao, Tao *); 4416285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMGetDualVector(Tao, Vec *); 4426285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMGetSpectralPenalty(Tao, PetscReal *); 4436285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetSpectralPenalty(Tao, PetscReal); 4446285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoGetADMMParentTao(Tao, Tao *); 4456285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetConstraintVectorRHS(Tao, Vec); 4466285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerCoefficient(Tao, PetscReal); 4476285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMisfitConstraintJacobian(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 4486285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerConstraintJacobian(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 4496285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerHessianRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 4506285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerObjectiveAndGradientRoutine(Tao, PetscErrorCode (*)(Tao, Vec, PetscReal *, Vec, void *), void *); 4516285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMisfitHessianRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 4526285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMisfitObjectiveAndGradientRoutine(Tao, PetscErrorCode (*)(Tao, Vec, PetscReal *, Vec, void *), void *); 4536285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMisfitHessianChangeStatus(Tao, PetscBool); 4546285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegHessianChangeStatus(Tao, PetscBool); 4556285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMinimumSpectralPenalty(Tao, PetscReal); 4566285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerType(Tao, TaoADMMRegularizerType); 4576285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMGetRegularizerType(Tao, TaoADMMRegularizerType *); 4586285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetUpdateType(Tao, TaoADMMUpdateType); 4596285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMGetUpdateType(Tao, TaoADMMUpdateType *); 460661095bbSAlp Dener 461661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetType(Tao, TaoALMMType *); 462661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMSetType(Tao, TaoALMMType); 463661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetSubsolver(Tao, Tao *); 464661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMSetSubsolver(Tao, Tao); 465661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetMultipliers(Tao, Vec *); 466661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMSetMultipliers(Tao, Vec); 467661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetPrimalIS(Tao, IS *, IS *); 468661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetDualIS(Tao, IS *, IS *); 46921ec2d5cSBarry Smith #endif 470