xref: /petsc/include/petsctao.h (revision 6524c165f7ddaf30fd7457737f668f984c8ababf)
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