1*6524c165SJacob 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 39db781477SPatrick Sanan .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 48db781477SPatrick Sanan .seealso `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 95db781477SPatrick Sanan .seealso `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 12987497f52SBarry Smith $ `TAO_ALMM_CLASSIC` - classic augmented Lagrangian definition including slack variables for inequality constraints 13087497f52SBarry Smith $ `TAO_ALMM_PHR` - Powell-Hestenes-Rockafellar formulation without slack variables, uses pointwise min() for inequalities 131661095bbSAlp Dener 132661095bbSAlp Dener Level: advanced 133661095bbSAlp Dener 134db781477SPatrick Sanan .seealso `TAOALMM`, `TaoALMMSetType()`, `TaoALMMGetType()` 135661095bbSAlp Dener E*/ 1369371c9d4SSatish Balay typedef enum { 1379371c9d4SSatish Balay TAO_ALMM_CLASSIC, 1389371c9d4SSatish Balay TAO_ALMM_PHR 1399371c9d4SSatish Balay } TaoALMMType; 140661095bbSAlp Dener PETSC_EXTERN const char *const TaoALMMTypes[]; 141661095bbSAlp Dener 1421eb8069cSJason Sarich /*J 14387497f52SBarry Smith TaoType - String with the name of a `Tao` method 1441eb8069cSJason Sarich 1451eb8069cSJason Sarich Level: beginner 1461eb8069cSJason Sarich 14765ba42b6SBarry Smith .seealso `Tao`, `TaoCreate()`, `TaoSetType()` 1481eb8069cSJason Sarich J*/ 149b625d6c7SJed Brown typedef const char *TaoType; 15058417fe7SBarry Smith #define TAOLMVM "lmvm" 15158417fe7SBarry Smith #define TAONLS "nls" 15258417fe7SBarry Smith #define TAONTR "ntr" 15358417fe7SBarry Smith #define TAONTL "ntl" 15458417fe7SBarry Smith #define TAOCG "cg" 15558417fe7SBarry Smith #define TAOTRON "tron" 15658417fe7SBarry Smith #define TAOOWLQN "owlqn" 15758417fe7SBarry Smith #define TAOBMRM "bmrm" 15858417fe7SBarry Smith #define TAOBLMVM "blmvm" 1596b591159SAlp Dener #define TAOBQNLS "bqnls" 160ac9112b8SAlp Dener #define TAOBNCG "bncg" 161eb910715SAlp Dener #define TAOBNLS "bnls" 162fed79b8eSAlp Dener #define TAOBNTR "bntr" 163c14b763aSAlp Dener #define TAOBNTL "bntl" 164e0ed867bSAlp Dener #define TAOBQNKLS "bqnkls" 165e0ed867bSAlp Dener #define TAOBQNKTR "bqnktr" 166e0ed867bSAlp Dener #define TAOBQNKTL "bqnktl" 16758417fe7SBarry Smith #define TAOBQPIP "bqpip" 16858417fe7SBarry Smith #define TAOGPCG "gpcg" 16958417fe7SBarry Smith #define TAONM "nm" 17058417fe7SBarry Smith #define TAOPOUNDERS "pounders" 171737f463aSAlp Dener #define TAOBRGN "brgn" 17258417fe7SBarry Smith #define TAOLCL "lcl" 17358417fe7SBarry Smith #define TAOSSILS "ssils" 17458417fe7SBarry Smith #define TAOSSFLS "ssfls" 17558417fe7SBarry Smith #define TAOASILS "asils" 17658417fe7SBarry Smith #define TAOASFLS "asfls" 17758417fe7SBarry Smith #define TAOIPM "ipm" 178aad13602SShrirang Abhyankar #define TAOPDIPM "pdipm" 17983a0a5c3SToby Isaac #define TAOSHELL "shell" 1806285c0a3SHansol Suh #define TAOADMM "admm" 181661095bbSAlp Dener #define TAOALMM "almm" 182a82e8c82SStefano Zampini #define TAOPYTHON "python" 18358417fe7SBarry Smith 184441846f8SBarry Smith PETSC_EXTERN PetscClassId TAO_CLASSID; 185441846f8SBarry Smith PETSC_EXTERN PetscFunctionList TaoList; 18621ec2d5cSBarry Smith 187a35d58b8SBarry Smith /*E 18865ba42b6SBarry Smith TaoConvergedReason - reason a Tao method was said to have converged or diverged 189a35d58b8SBarry Smith 190a35d58b8SBarry Smith Level: beginner 191a35d58b8SBarry Smith 192a35d58b8SBarry Smith The two most common reasons for divergence are 193a35d58b8SBarry Smith $ 1) an incorrectly coded or computed gradient or Hessian 194a35d58b8SBarry Smith $ 2) failure or lack of convergence in the linear system (in this case we recommend 195a35d58b8SBarry Smith $ testing with -pc_type lu to eliminate the linear solver as the cause of the problem). 196a35d58b8SBarry Smith 19795452b02SPatrick Sanan Developer Notes: 19887497f52SBarry Smith This must match petsc/finclude/petsctao.h 199a35d58b8SBarry Smith 20087497f52SBarry Smith The string versions of these are in `TAOConvergedReasons`, if you change any value here you must 201a35d58b8SBarry Smith also adjust that array. 202a35d58b8SBarry Smith 20365ba42b6SBarry Smith .seealso: `Tao`, `TaoSolve()`, `TaoGetConvergedReason()`, `KSPConvergedReason`, `SNESConvergedReason`, `TSConvergedReason` 204a35d58b8SBarry Smith E*/ 20521ec2d5cSBarry Smith typedef enum { /* converged */ 20621ec2d5cSBarry Smith TAO_CONVERGED_GATOL = 3, /* ||g(X)|| < gatol */ 20721ec2d5cSBarry Smith TAO_CONVERGED_GRTOL = 4, /* ||g(X)|| / f(X) < grtol */ 20821ec2d5cSBarry Smith TAO_CONVERGED_GTTOL = 5, /* ||g(X)|| / ||g(X0)|| < gttol */ 20921ec2d5cSBarry Smith TAO_CONVERGED_STEPTOL = 6, /* step size small */ 21021ec2d5cSBarry Smith TAO_CONVERGED_MINF = 7, /* F < F_min */ 21121ec2d5cSBarry Smith TAO_CONVERGED_USER = 8, /* User defined */ 21221ec2d5cSBarry Smith /* diverged */ 21321ec2d5cSBarry Smith TAO_DIVERGED_MAXITS = -2, 21421ec2d5cSBarry Smith TAO_DIVERGED_NAN = -4, 21521ec2d5cSBarry Smith TAO_DIVERGED_MAXFCN = -5, 21621ec2d5cSBarry Smith TAO_DIVERGED_LS_FAILURE = -6, 21721ec2d5cSBarry Smith TAO_DIVERGED_TR_REDUCTION = -7, 21821ec2d5cSBarry Smith TAO_DIVERGED_USER = -8, /* User defined */ 21921ec2d5cSBarry Smith /* keep going */ 2209371c9d4SSatish Balay TAO_CONTINUE_ITERATING = 0 2219371c9d4SSatish Balay } TaoConvergedReason; 22221ec2d5cSBarry Smith 223e4cb33bbSBarry Smith PETSC_EXTERN const char **TaoConvergedReasons; 22421ec2d5cSBarry Smith 22521ec2d5cSBarry Smith PETSC_EXTERN PetscErrorCode TaoInitializePackage(void); 22621ec2d5cSBarry Smith PETSC_EXTERN PetscErrorCode TaoFinalizePackage(void); 227441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoCreate(MPI_Comm, Tao *); 228441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetFromOptions(Tao); 229441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetUp(Tao); 230b625d6c7SJed Brown PETSC_EXTERN PetscErrorCode TaoSetType(Tao, TaoType); 231b625d6c7SJed Brown PETSC_EXTERN PetscErrorCode TaoGetType(Tao, TaoType *); 232441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetApplicationContext(Tao, void *); 233441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetApplicationContext(Tao, void *); 234441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDestroy(Tao *); 23521ec2d5cSBarry Smith 236441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetOptionsPrefix(Tao, const char[]); 237441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoView(Tao, PetscViewer); 238fe2efc57SMark PETSC_EXTERN PetscErrorCode TaoViewFromOptions(Tao, PetscObject, const char[]); 23921ec2d5cSBarry Smith 240441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSolve(Tao); 24121ec2d5cSBarry Smith 242441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoRegister(const char[], PetscErrorCode (*)(Tao)); 243441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoRegisterDestroy(void); 24421ec2d5cSBarry Smith 245e4cb33bbSBarry Smith PETSC_EXTERN PetscErrorCode TaoGetConvergedReason(Tao, TaoConvergedReason *); 246e4cb33bbSBarry Smith PETSC_EXTERN PetscErrorCode TaoGetSolutionStatus(Tao, PetscInt *, PetscReal *, PetscReal *, PetscReal *, PetscReal *, TaoConvergedReason *); 247e4cb33bbSBarry Smith PETSC_EXTERN PetscErrorCode TaoSetConvergedReason(Tao, TaoConvergedReason); 248a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetSolution(Tao, Vec); 249a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetSolution(Tao, Vec *); 2509371c9d4SSatish Balay PETSC_DEPRECATED_FUNCTION("Use TaoSetSolution() (since version 3.17)") static inline PetscErrorCode TaoSetInitialVector(Tao t, Vec v) { 2519371c9d4SSatish Balay return TaoSetSolution(t, v); 2529371c9d4SSatish Balay } 2539371c9d4SSatish Balay PETSC_DEPRECATED_FUNCTION("Use TaoGetSolution() (since version 3.17)") static inline PetscErrorCode TaoGetInitialVector(Tao t, Vec *v) { 2549371c9d4SSatish Balay return TaoGetSolution(t, v); 2559371c9d4SSatish Balay } 256a82e8c82SStefano Zampini 257a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetObjective(Tao, PetscErrorCode (*)(Tao, Vec, PetscReal *, void *), void *); 258a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetObjective(Tao, PetscErrorCode (**)(Tao, Vec, PetscReal *, void *), void **); 259a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetGradient(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *); 260a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetGradient(Tao, Vec *, PetscErrorCode (**)(Tao, Vec, Vec, void *), void **); 261a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetObjectiveAndGradient(Tao, Vec, PetscErrorCode (*)(Tao, Vec, PetscReal *, Vec, void *), void *); 262a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetObjectiveAndGradient(Tao, Vec *, PetscErrorCode (**)(Tao, Vec, PetscReal *, Vec, void *), void **); 263a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetHessian(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 264a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetHessian(Tao, Mat *, Mat *, PetscErrorCode (**)(Tao, Vec, Mat, Mat, void *), void **); 2659371c9d4SSatish Balay PETSC_DEPRECATED_FUNCTION("Use TaoSetObjective() (since version 3.17)") static inline PetscErrorCode TaoSetObjectiveRoutine(Tao t, PetscErrorCode (*f)(Tao, Vec, PetscReal *, void *), void *c) { 2669371c9d4SSatish Balay return TaoSetObjective(t, f, c); 2679371c9d4SSatish Balay } 2689371c9d4SSatish Balay PETSC_DEPRECATED_FUNCTION("Use TaoGetGradient() (since version 3.17)") static inline PetscErrorCode TaoGetGradientVector(Tao t, Vec *v) { 2699371c9d4SSatish Balay return TaoGetGradient(t, v, NULL, NULL); 2709371c9d4SSatish Balay } 2719371c9d4SSatish Balay PETSC_DEPRECATED_FUNCTION("Use TaoSetGradient() (since version 3.17)") static inline PetscErrorCode TaoSetGradientRoutine(Tao t, PetscErrorCode (*f)(Tao, Vec, Vec, void *), void *c) { 2729371c9d4SSatish Balay return TaoSetGradient(t, NULL, f, c); 2739371c9d4SSatish Balay } 2749371c9d4SSatish Balay PETSC_DEPRECATED_FUNCTION("Use TaoSetObjectiveAndGradient() (since version 3.17)") static inline PetscErrorCode TaoSetObjectiveAndGradientRoutine(Tao t, PetscErrorCode (*f)(Tao, Vec, PetscReal *, Vec, void *), void *c) { 2759371c9d4SSatish Balay return TaoSetObjectiveAndGradient(t, NULL, f, c); 2769371c9d4SSatish Balay } 2779371c9d4SSatish Balay 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) { 2789371c9d4SSatish Balay return TaoSetHessian(t, H, P, f, c); 2799371c9d4SSatish Balay } 280a82e8c82SStefano Zampini 281a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoSetGradientNorm(Tao, Mat); 282a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoGetGradientNorm(Tao, Mat *); 283414d97d3SAlp Dener PETSC_EXTERN PetscErrorCode TaoSetLMVMMatrix(Tao, Mat); 284f5766c09SAlp Dener PETSC_EXTERN PetscErrorCode TaoGetLMVMMatrix(Tao, Mat *); 285414d97d3SAlp Dener PETSC_EXTERN PetscErrorCode TaoSetRecycleHistory(Tao, PetscBool); 286414d97d3SAlp Dener PETSC_EXTERN PetscErrorCode TaoGetRecycleHistory(Tao, PetscBool *); 287a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoLMVMSetH0(Tao, Mat); 288a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoLMVMGetH0(Tao, Mat *); 289a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoLMVMGetH0KSP(Tao, KSP *); 290b39c12a9SAlp Dener PETSC_EXTERN PetscErrorCode TaoLMVMRecycle(Tao, PetscBool); 2914a48860cSAlp Dener PETSC_EXTERN PetscErrorCode TaoSetResidualRoutine(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *); 292737f463aSAlp Dener PETSC_EXTERN PetscErrorCode TaoSetResidualWeights(Tao, Vec, PetscInt, PetscInt *, PetscInt *, PetscReal *); 293441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetConstraintsRoutine(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *); 294441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetInequalityConstraintsRoutine(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *); 295441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetEqualityConstraintsRoutine(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *); 2964ffbe8acSAlp Dener PETSC_EXTERN PetscErrorCode TaoSetJacobianResidualRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 297ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 298ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianStateRoutine(Tao, Mat, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, Mat, void *), void *); 29994ab13aaSBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianDesignRoutine(Tao, Mat, PetscErrorCode (*)(Tao, Vec, Mat, void *), void *); 300ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianInequalityRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 301ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianEqualityRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 30221ec2d5cSBarry Smith 303a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoPythonSetType(Tao, const char[]); 304ebead697SStefano Zampini PETSC_EXTERN PetscErrorCode TaoPythonGetType(Tao, const char *[]); 305a82e8c82SStefano Zampini 30683a0a5c3SToby Isaac PETSC_EXTERN PetscErrorCode TaoShellSetSolve(Tao, PetscErrorCode (*)(Tao)); 30783a0a5c3SToby Isaac PETSC_EXTERN PetscErrorCode TaoShellSetContext(Tao, void *); 3083ec1f749SStefano Zampini PETSC_EXTERN PetscErrorCode TaoShellGetContext(Tao, void *); 30983a0a5c3SToby Isaac 3109371c9d4SSatish Balay 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) { 3119371c9d4SSatish Balay return TaoSetResidualRoutine(tao, res, func, ctx); 3129371c9d4SSatish Balay } 3139371c9d4SSatish Balay 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) { 3149371c9d4SSatish Balay return TaoSetResidualWeights(tao, sigma_v, n, rows, cols, vals); 3159371c9d4SSatish Balay } 316737f463aSAlp Dener 317441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetStateDesignIS(Tao, IS, IS); 31821ec2d5cSBarry Smith 319441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeObjective(Tao, Vec, PetscReal *); 3204a48860cSAlp Dener PETSC_EXTERN PetscErrorCode TaoComputeResidual(Tao, Vec, Vec); 321412cdd55SHong Zhang PETSC_EXTERN PetscErrorCode TaoTestGradient(Tao, Vec, Vec); 322441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeGradient(Tao, Vec, Vec); 323441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeObjectiveAndGradient(Tao, Vec, PetscReal *, Vec); 324441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeConstraints(Tao, Vec, Vec); 325441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeInequalityConstraints(Tao, Vec, Vec); 326441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeEqualityConstraints(Tao, Vec, Vec); 327441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultComputeGradient(Tao, Vec, Vec, void *); 328441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoIsObjectiveDefined(Tao, PetscBool *); 329441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoIsGradientDefined(Tao, PetscBool *); 330441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoIsObjectiveAndGradientDefined(Tao, PetscBool *); 33121ec2d5cSBarry Smith 3329371c9d4SSatish Balay PETSC_DEPRECATED_FUNCTION("Use TaoComputeResidual() (since version 3.11)") static inline PetscErrorCode TaoComputeSeparableObjective(Tao tao, Vec X, Vec F) { 3339371c9d4SSatish Balay return TaoComputeResidual(tao, X, F); 3349371c9d4SSatish Balay } 3354a48860cSAlp Dener 33609baa881SHong Zhang PETSC_EXTERN PetscErrorCode TaoTestHessian(Tao); 337ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeHessian(Tao, Vec, Mat, Mat); 338737f463aSAlp Dener PETSC_EXTERN PetscErrorCode TaoComputeResidualJacobian(Tao, Vec, Mat, Mat); 339ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobian(Tao, Vec, Mat, Mat); 340ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobianState(Tao, Vec, Mat, Mat, Mat); 341ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobianEquality(Tao, Vec, Mat, Mat); 342ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobianInequality(Tao, Vec, Mat, Mat); 34394ab13aaSBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobianDesign(Tao, Vec, Mat); 34421ec2d5cSBarry Smith 345ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultComputeHessian(Tao, Vec, Mat, Mat, void *); 346ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultComputeHessianColor(Tao, Vec, Mat, Mat, void *); 347f4c1ad5cSStefano Zampini PETSC_EXTERN PetscErrorCode TaoDefaultComputeHessianMFFD(Tao, Vec, Mat, Mat, void *); 348441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeDualVariables(Tao, Vec, Vec); 349441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetVariableBounds(Tao, Vec, Vec); 350441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetVariableBounds(Tao, Vec *, Vec *); 351441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetDualVariables(Tao, Vec *, Vec *); 352441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetInequalityBounds(Tao, Vec, Vec); 353441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetInequalityBounds(Tao, Vec *, Vec *); 354441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetVariableBoundsRoutine(Tao, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *); 355441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeVariableBounds(Tao); 35621ec2d5cSBarry Smith 357e52336cbSBarry Smith PETSC_EXTERN PetscErrorCode TaoGetTolerances(Tao, PetscReal *, PetscReal *, PetscReal *); 358e52336cbSBarry Smith PETSC_EXTERN PetscErrorCode TaoSetTolerances(Tao, PetscReal, PetscReal, PetscReal); 359441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetConstraintTolerances(Tao, PetscReal *, PetscReal *); 360441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetConstraintTolerances(Tao, PetscReal, PetscReal); 361441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetFunctionLowerBound(Tao, PetscReal); 362441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetInitialTrustRegionRadius(Tao, PetscReal); 363441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetMaximumIterations(Tao, PetscInt); 364441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao, PetscInt); 365441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetFunctionLowerBound(Tao, PetscReal *); 366441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetInitialTrustRegionRadius(Tao, PetscReal *); 367441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao, PetscReal *); 368441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetMaximumIterations(Tao, PetscInt *); 369770232b9SCe Qin PETSC_EXTERN PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao, PetscInt *); 370441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao, PetscInt *); 3718931d482SJason Sarich PETSC_EXTERN PetscErrorCode TaoGetIterationNumber(Tao, PetscInt *); 3728931d482SJason Sarich PETSC_EXTERN PetscErrorCode TaoSetIterationNumber(Tao, PetscInt); 3738931d482SJason Sarich PETSC_EXTERN PetscErrorCode TaoGetTotalIterationNumber(Tao, PetscInt *); 3748931d482SJason Sarich PETSC_EXTERN PetscErrorCode TaoSetTotalIterationNumber(Tao, PetscInt); 37579f5d8caSBarry Smith PETSC_EXTERN PetscErrorCode TaoGetResidualNorm(Tao, PetscReal *); 3768931d482SJason Sarich 377b54963c9SStefano Zampini PETSC_EXTERN PetscErrorCode TaoAppendOptionsPrefix(Tao, const char[]); 378b54963c9SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetOptionsPrefix(Tao, const char *[]); 379441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoResetStatistics(Tao); 3808fcddce6SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetUpdate(Tao, PetscErrorCode (*)(Tao, PetscInt, void *), void *); 38121ec2d5cSBarry Smith 382441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetKSP(Tao, KSP *); 383025e9500SJason Sarich PETSC_EXTERN PetscErrorCode TaoGetLinearSolveIterations(Tao, PetscInt *); 3840f0abf79SStefano Zampini PETSC_EXTERN PetscErrorCode TaoKSPSetUseEW(Tao, PetscBool); 385235fd6e6SBarry Smith 386235fd6e6SBarry Smith #include <petsctaolinesearch.h> 387b54963c9SStefano Zampini 388441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetLineSearch(Tao, TaoLineSearch *); 38921ec2d5cSBarry Smith 390ae93cb3cSJason Sarich PETSC_EXTERN PetscErrorCode TaoSetConvergenceHistory(Tao, PetscReal *, PetscReal *, PetscReal *, PetscInt *, PetscInt, PetscBool); 391ae93cb3cSJason Sarich PETSC_EXTERN PetscErrorCode TaoGetConvergenceHistory(Tao, PetscReal **, PetscReal **, PetscReal **, PetscInt **, PetscInt *); 392441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetMonitor(Tao, PetscErrorCode (*)(Tao, void *), void *, PetscErrorCode (*)(void **)); 393441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoCancelMonitors(Tao); 39498ea980cSBarry Smith PETSC_EXTERN PetscErrorCode TaoMonitorDefault(Tao, void *); 3959371c9d4SSatish Balay PETSC_DEPRECATED_FUNCTION("Use TaoMonitorDefault() (since version 3.9)") static inline PetscErrorCode TaoDefaultMonitor(Tao tao, void *ctx) { 3969371c9d4SSatish Balay return TaoMonitorDefault(tao, ctx); 3979371c9d4SSatish Balay } 3988d5ead36SAlp Dener PETSC_EXTERN PetscErrorCode TaoDefaultGMonitor(Tao, void *); 399441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultSMonitor(Tao, void *); 400441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultCMonitor(Tao, void *); 401441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSolutionMonitor(Tao, void *); 402737f463aSAlp Dener PETSC_EXTERN PetscErrorCode TaoResidualMonitor(Tao, void *); 403441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGradientMonitor(Tao, void *); 404441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoStepDirectionMonitor(Tao, void *); 405441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDrawSolutionMonitor(Tao, void *); 406441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDrawStepMonitor(Tao, void *); 407441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDrawGradientMonitor(Tao, void *); 408441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoAddLineSearchCounts(Tao); 40921ec2d5cSBarry Smith 410441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultConvergenceTest(Tao, void *); 411441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetConvergenceTest(Tao, PetscErrorCode (*)(Tao, void *), void *); 41221ec2d5cSBarry Smith 413441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoLCLSetStateDesignIS(Tao, IS, IS); 4143ecd9318SAlp Dener PETSC_EXTERN PetscErrorCode TaoMonitor(Tao, PetscInt, PetscReal, PetscReal, PetscReal, PetscReal); 415e882e171SHong Zhang typedef struct _n_TaoMonitorDrawCtx *TaoMonitorDrawCtx; 416e882e171SHong Zhang PETSC_EXTERN PetscErrorCode TaoMonitorDrawCtxCreate(MPI_Comm, const char[], const char[], int, int, int, int, PetscInt, TaoMonitorDrawCtx *); 417e882e171SHong Zhang PETSC_EXTERN PetscErrorCode TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *); 418737f463aSAlp Dener 4198e85b1b3SXiang Huang PETSC_EXTERN PetscErrorCode TaoBRGNGetSubsolver(Tao, Tao *); 420a3c390cfSAlp Dener PETSC_EXTERN PetscErrorCode TaoBRGNSetRegularizerObjectiveAndGradientRoutine(Tao, PetscErrorCode (*)(Tao, Vec, PetscReal *, Vec, void *), void *); 421a3c390cfSAlp Dener PETSC_EXTERN PetscErrorCode TaoBRGNSetRegularizerHessianRoutine(Tao, Mat, PetscErrorCode (*)(Tao, Vec, Mat, void *), void *); 422a3c390cfSAlp Dener PETSC_EXTERN PetscErrorCode TaoBRGNSetRegularizerWeight(Tao, PetscReal); 4238ac80d48SXiang Huang PETSC_EXTERN PetscErrorCode TaoBRGNSetL1SmoothEpsilon(Tao, PetscReal); 4248e85b1b3SXiang Huang PETSC_EXTERN PetscErrorCode TaoBRGNSetDictionaryMatrix(Tao, Mat); 425cd1c4666STristan Konolige PETSC_EXTERN PetscErrorCode TaoBRGNGetDampingVector(Tao, Vec *); 4266285c0a3SHansol Suh 4276285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMGetMisfitSubsolver(Tao, Tao *); 4286285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMGetRegularizationSubsolver(Tao, Tao *); 4296285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMGetDualVector(Tao, Vec *); 4306285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMGetSpectralPenalty(Tao, PetscReal *); 4316285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetSpectralPenalty(Tao, PetscReal); 4326285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoGetADMMParentTao(Tao, Tao *); 4336285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetConstraintVectorRHS(Tao, Vec); 4346285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerCoefficient(Tao, PetscReal); 4356285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMisfitConstraintJacobian(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 4366285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerConstraintJacobian(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 4376285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerHessianRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 4386285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerObjectiveAndGradientRoutine(Tao, PetscErrorCode (*)(Tao, Vec, PetscReal *, Vec, void *), void *); 4396285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMisfitHessianRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *); 4406285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMisfitObjectiveAndGradientRoutine(Tao, PetscErrorCode (*)(Tao, Vec, PetscReal *, Vec, void *), void *); 4416285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMisfitHessianChangeStatus(Tao, PetscBool); 4426285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegHessianChangeStatus(Tao, PetscBool); 4436285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMinimumSpectralPenalty(Tao, PetscReal); 4446285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerType(Tao, TaoADMMRegularizerType); 4456285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMGetRegularizerType(Tao, TaoADMMRegularizerType *); 4466285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMSetUpdateType(Tao, TaoADMMUpdateType); 4476285c0a3SHansol Suh PETSC_EXTERN PetscErrorCode TaoADMMGetUpdateType(Tao, TaoADMMUpdateType *); 448661095bbSAlp Dener 449661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetType(Tao, TaoALMMType *); 450661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMSetType(Tao, TaoALMMType); 451661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetSubsolver(Tao, Tao *); 452661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMSetSubsolver(Tao, Tao); 453661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetMultipliers(Tao, Vec *); 454661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMSetMultipliers(Tao, Vec); 455661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetPrimalIS(Tao, IS *, IS *); 456661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetDualIS(Tao, IS *, IS *); 45721ec2d5cSBarry Smith #endif 458