xref: /petsc/include/petsctao.h (revision d71ae5a4db6382e7f06317b8d368875286fe9008)
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 
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 *);
250*d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoSetSolution() (since version 3.17)") static inline PetscErrorCode TaoSetInitialVector(Tao t, Vec v)
251*d71ae5a4SJacob Faibussowitsch {
2529371c9d4SSatish Balay   return TaoSetSolution(t, v);
2539371c9d4SSatish Balay }
254*d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoGetSolution() (since version 3.17)") static inline PetscErrorCode TaoGetInitialVector(Tao t, Vec *v)
255*d71ae5a4SJacob Faibussowitsch {
2569371c9d4SSatish Balay   return TaoGetSolution(t, v);
2579371c9d4SSatish Balay }
258a82e8c82SStefano Zampini 
259a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetObjective(Tao, PetscErrorCode (*)(Tao, Vec, PetscReal *, void *), void *);
260a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetObjective(Tao, PetscErrorCode (**)(Tao, Vec, PetscReal *, void *), void **);
261a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetGradient(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *);
262a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetGradient(Tao, Vec *, PetscErrorCode (**)(Tao, Vec, Vec, void *), void **);
263a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetObjectiveAndGradient(Tao, Vec, PetscErrorCode (*)(Tao, Vec, PetscReal *, Vec, void *), void *);
264a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetObjectiveAndGradient(Tao, Vec *, PetscErrorCode (**)(Tao, Vec, PetscReal *, Vec, void *), void **);
265a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetHessian(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *);
266a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetHessian(Tao, Mat *, Mat *, PetscErrorCode (**)(Tao, Vec, Mat, Mat, void *), void **);
267*d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoSetObjective() (since version 3.17)") static inline PetscErrorCode TaoSetObjectiveRoutine(Tao t, PetscErrorCode (*f)(Tao, Vec, PetscReal *, void *), void *c)
268*d71ae5a4SJacob Faibussowitsch {
2699371c9d4SSatish Balay   return TaoSetObjective(t, f, c);
2709371c9d4SSatish Balay }
271*d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoGetGradient() (since version 3.17)") static inline PetscErrorCode TaoGetGradientVector(Tao t, Vec *v)
272*d71ae5a4SJacob Faibussowitsch {
2739371c9d4SSatish Balay   return TaoGetGradient(t, v, NULL, NULL);
2749371c9d4SSatish Balay }
275*d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoSetGradient() (since version 3.17)") static inline PetscErrorCode TaoSetGradientRoutine(Tao t, PetscErrorCode (*f)(Tao, Vec, Vec, void *), void *c)
276*d71ae5a4SJacob Faibussowitsch {
2779371c9d4SSatish Balay   return TaoSetGradient(t, NULL, f, c);
2789371c9d4SSatish Balay }
279*d71ae5a4SJacob 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)
280*d71ae5a4SJacob Faibussowitsch {
2819371c9d4SSatish Balay   return TaoSetObjectiveAndGradient(t, NULL, f, c);
2829371c9d4SSatish Balay }
283*d71ae5a4SJacob 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)
284*d71ae5a4SJacob Faibussowitsch {
2859371c9d4SSatish Balay   return TaoSetHessian(t, H, P, f, c);
2869371c9d4SSatish Balay }
287a82e8c82SStefano Zampini 
288a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoSetGradientNorm(Tao, Mat);
289a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoGetGradientNorm(Tao, Mat *);
290414d97d3SAlp Dener PETSC_EXTERN PetscErrorCode TaoSetLMVMMatrix(Tao, Mat);
291f5766c09SAlp Dener PETSC_EXTERN PetscErrorCode TaoGetLMVMMatrix(Tao, Mat *);
292414d97d3SAlp Dener PETSC_EXTERN PetscErrorCode TaoSetRecycleHistory(Tao, PetscBool);
293414d97d3SAlp Dener PETSC_EXTERN PetscErrorCode TaoGetRecycleHistory(Tao, PetscBool *);
294a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoLMVMSetH0(Tao, Mat);
295a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoLMVMGetH0(Tao, Mat *);
296a9603a14SPatrick Farrell PETSC_EXTERN PetscErrorCode TaoLMVMGetH0KSP(Tao, KSP *);
297b39c12a9SAlp Dener PETSC_EXTERN PetscErrorCode TaoLMVMRecycle(Tao, PetscBool);
2984a48860cSAlp Dener PETSC_EXTERN PetscErrorCode TaoSetResidualRoutine(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *);
299737f463aSAlp Dener PETSC_EXTERN PetscErrorCode TaoSetResidualWeights(Tao, Vec, PetscInt, PetscInt *, PetscInt *, PetscReal *);
300441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetConstraintsRoutine(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *);
301441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetInequalityConstraintsRoutine(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *);
302441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetEqualityConstraintsRoutine(Tao, Vec, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *);
3034ffbe8acSAlp Dener PETSC_EXTERN PetscErrorCode TaoSetJacobianResidualRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *);
304ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *);
305ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianStateRoutine(Tao, Mat, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, Mat, void *), void *);
30694ab13aaSBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianDesignRoutine(Tao, Mat, PetscErrorCode (*)(Tao, Vec, Mat, void *), void *);
307ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianInequalityRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *);
308ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetJacobianEqualityRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *);
30921ec2d5cSBarry Smith 
310a82e8c82SStefano Zampini PETSC_EXTERN PetscErrorCode TaoPythonSetType(Tao, const char[]);
311ebead697SStefano Zampini PETSC_EXTERN PetscErrorCode TaoPythonGetType(Tao, const char *[]);
312a82e8c82SStefano Zampini 
31383a0a5c3SToby Isaac PETSC_EXTERN PetscErrorCode TaoShellSetSolve(Tao, PetscErrorCode (*)(Tao));
31483a0a5c3SToby Isaac PETSC_EXTERN PetscErrorCode TaoShellSetContext(Tao, void *);
3153ec1f749SStefano Zampini PETSC_EXTERN PetscErrorCode TaoShellGetContext(Tao, void *);
31683a0a5c3SToby Isaac 
317*d71ae5a4SJacob 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)
318*d71ae5a4SJacob Faibussowitsch {
3199371c9d4SSatish Balay   return TaoSetResidualRoutine(tao, res, func, ctx);
3209371c9d4SSatish Balay }
321*d71ae5a4SJacob 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)
322*d71ae5a4SJacob Faibussowitsch {
3239371c9d4SSatish Balay   return TaoSetResidualWeights(tao, sigma_v, n, rows, cols, vals);
3249371c9d4SSatish Balay }
325737f463aSAlp Dener 
326441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetStateDesignIS(Tao, IS, IS);
32721ec2d5cSBarry Smith 
328441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeObjective(Tao, Vec, PetscReal *);
3294a48860cSAlp Dener PETSC_EXTERN PetscErrorCode TaoComputeResidual(Tao, Vec, Vec);
330412cdd55SHong Zhang PETSC_EXTERN PetscErrorCode TaoTestGradient(Tao, Vec, Vec);
331441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeGradient(Tao, Vec, Vec);
332441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeObjectiveAndGradient(Tao, Vec, PetscReal *, Vec);
333441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeConstraints(Tao, Vec, Vec);
334441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeInequalityConstraints(Tao, Vec, Vec);
335441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeEqualityConstraints(Tao, Vec, Vec);
336441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultComputeGradient(Tao, Vec, Vec, void *);
337441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoIsObjectiveDefined(Tao, PetscBool *);
338441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoIsGradientDefined(Tao, PetscBool *);
339441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoIsObjectiveAndGradientDefined(Tao, PetscBool *);
34021ec2d5cSBarry Smith 
341*d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoComputeResidual() (since version 3.11)") static inline PetscErrorCode TaoComputeSeparableObjective(Tao tao, Vec X, Vec F)
342*d71ae5a4SJacob Faibussowitsch {
3439371c9d4SSatish Balay   return TaoComputeResidual(tao, X, F);
3449371c9d4SSatish Balay }
3454a48860cSAlp Dener 
34609baa881SHong Zhang PETSC_EXTERN PetscErrorCode TaoTestHessian(Tao);
347ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeHessian(Tao, Vec, Mat, Mat);
348737f463aSAlp Dener PETSC_EXTERN PetscErrorCode TaoComputeResidualJacobian(Tao, Vec, Mat, Mat);
349ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobian(Tao, Vec, Mat, Mat);
350ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobianState(Tao, Vec, Mat, Mat, Mat);
351ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobianEquality(Tao, Vec, Mat, Mat);
352ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobianInequality(Tao, Vec, Mat, Mat);
35394ab13aaSBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeJacobianDesign(Tao, Vec, Mat);
35421ec2d5cSBarry Smith 
355ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultComputeHessian(Tao, Vec, Mat, Mat, void *);
356ffad9901SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultComputeHessianColor(Tao, Vec, Mat, Mat, void *);
357f4c1ad5cSStefano Zampini PETSC_EXTERN PetscErrorCode TaoDefaultComputeHessianMFFD(Tao, Vec, Mat, Mat, void *);
358441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeDualVariables(Tao, Vec, Vec);
359441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetVariableBounds(Tao, Vec, Vec);
360441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetVariableBounds(Tao, Vec *, Vec *);
361441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetDualVariables(Tao, Vec *, Vec *);
362441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetInequalityBounds(Tao, Vec, Vec);
363441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetInequalityBounds(Tao, Vec *, Vec *);
364441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetVariableBoundsRoutine(Tao, PetscErrorCode (*)(Tao, Vec, Vec, void *), void *);
365441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoComputeVariableBounds(Tao);
36621ec2d5cSBarry Smith 
367e52336cbSBarry Smith PETSC_EXTERN PetscErrorCode TaoGetTolerances(Tao, PetscReal *, PetscReal *, PetscReal *);
368e52336cbSBarry Smith PETSC_EXTERN PetscErrorCode TaoSetTolerances(Tao, PetscReal, PetscReal, PetscReal);
369441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetConstraintTolerances(Tao, PetscReal *, PetscReal *);
370441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetConstraintTolerances(Tao, PetscReal, PetscReal);
371441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetFunctionLowerBound(Tao, PetscReal);
372441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetInitialTrustRegionRadius(Tao, PetscReal);
373441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetMaximumIterations(Tao, PetscInt);
374441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetMaximumFunctionEvaluations(Tao, PetscInt);
375441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetFunctionLowerBound(Tao, PetscReal *);
376441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetInitialTrustRegionRadius(Tao, PetscReal *);
377441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetCurrentTrustRegionRadius(Tao, PetscReal *);
378441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetMaximumIterations(Tao, PetscInt *);
379770232b9SCe Qin PETSC_EXTERN PetscErrorCode TaoGetCurrentFunctionEvaluations(Tao, PetscInt *);
380441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetMaximumFunctionEvaluations(Tao, PetscInt *);
3818931d482SJason Sarich PETSC_EXTERN PetscErrorCode TaoGetIterationNumber(Tao, PetscInt *);
3828931d482SJason Sarich PETSC_EXTERN PetscErrorCode TaoSetIterationNumber(Tao, PetscInt);
3838931d482SJason Sarich PETSC_EXTERN PetscErrorCode TaoGetTotalIterationNumber(Tao, PetscInt *);
3848931d482SJason Sarich PETSC_EXTERN PetscErrorCode TaoSetTotalIterationNumber(Tao, PetscInt);
38579f5d8caSBarry Smith PETSC_EXTERN PetscErrorCode TaoGetResidualNorm(Tao, PetscReal *);
3868931d482SJason Sarich 
387b54963c9SStefano Zampini PETSC_EXTERN PetscErrorCode TaoAppendOptionsPrefix(Tao, const char[]);
388b54963c9SStefano Zampini PETSC_EXTERN PetscErrorCode TaoGetOptionsPrefix(Tao, const char *[]);
389441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoResetStatistics(Tao);
3908fcddce6SStefano Zampini PETSC_EXTERN PetscErrorCode TaoSetUpdate(Tao, PetscErrorCode (*)(Tao, PetscInt, void *), void *);
39121ec2d5cSBarry Smith 
392441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetKSP(Tao, KSP *);
393025e9500SJason Sarich PETSC_EXTERN PetscErrorCode TaoGetLinearSolveIterations(Tao, PetscInt *);
3940f0abf79SStefano Zampini PETSC_EXTERN PetscErrorCode TaoKSPSetUseEW(Tao, PetscBool);
395235fd6e6SBarry Smith 
396235fd6e6SBarry Smith #include <petsctaolinesearch.h>
397b54963c9SStefano Zampini 
398441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGetLineSearch(Tao, TaoLineSearch *);
39921ec2d5cSBarry Smith 
400ae93cb3cSJason Sarich PETSC_EXTERN PetscErrorCode TaoSetConvergenceHistory(Tao, PetscReal *, PetscReal *, PetscReal *, PetscInt *, PetscInt, PetscBool);
401ae93cb3cSJason Sarich PETSC_EXTERN PetscErrorCode TaoGetConvergenceHistory(Tao, PetscReal **, PetscReal **, PetscReal **, PetscInt **, PetscInt *);
402441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetMonitor(Tao, PetscErrorCode (*)(Tao, void *), void *, PetscErrorCode (*)(void **));
403441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoCancelMonitors(Tao);
40498ea980cSBarry Smith PETSC_EXTERN PetscErrorCode TaoMonitorDefault(Tao, void *);
405*d71ae5a4SJacob Faibussowitsch PETSC_DEPRECATED_FUNCTION("Use TaoMonitorDefault() (since version 3.9)") static inline PetscErrorCode TaoDefaultMonitor(Tao tao, void *ctx)
406*d71ae5a4SJacob Faibussowitsch {
4079371c9d4SSatish Balay   return TaoMonitorDefault(tao, ctx);
4089371c9d4SSatish Balay }
4098d5ead36SAlp Dener PETSC_EXTERN PetscErrorCode TaoDefaultGMonitor(Tao, void *);
410441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultSMonitor(Tao, void *);
411441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultCMonitor(Tao, void *);
412441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSolutionMonitor(Tao, void *);
413737f463aSAlp Dener PETSC_EXTERN PetscErrorCode TaoResidualMonitor(Tao, void *);
414441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoGradientMonitor(Tao, void *);
415441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoStepDirectionMonitor(Tao, void *);
416441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDrawSolutionMonitor(Tao, void *);
417441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDrawStepMonitor(Tao, void *);
418441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDrawGradientMonitor(Tao, void *);
419441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoAddLineSearchCounts(Tao);
42021ec2d5cSBarry Smith 
421441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoDefaultConvergenceTest(Tao, void *);
422441846f8SBarry Smith PETSC_EXTERN PetscErrorCode TaoSetConvergenceTest(Tao, PetscErrorCode (*)(Tao, void *), void *);
42321ec2d5cSBarry Smith 
424441846f8SBarry Smith PETSC_EXTERN PetscErrorCode          TaoLCLSetStateDesignIS(Tao, IS, IS);
4253ecd9318SAlp Dener PETSC_EXTERN PetscErrorCode          TaoMonitor(Tao, PetscInt, PetscReal, PetscReal, PetscReal, PetscReal);
426e882e171SHong Zhang typedef struct _n_TaoMonitorDrawCtx *TaoMonitorDrawCtx;
427e882e171SHong Zhang PETSC_EXTERN PetscErrorCode          TaoMonitorDrawCtxCreate(MPI_Comm, const char[], const char[], int, int, int, int, PetscInt, TaoMonitorDrawCtx *);
428e882e171SHong Zhang PETSC_EXTERN PetscErrorCode          TaoMonitorDrawCtxDestroy(TaoMonitorDrawCtx *);
429737f463aSAlp Dener 
4308e85b1b3SXiang Huang PETSC_EXTERN PetscErrorCode TaoBRGNGetSubsolver(Tao, Tao *);
431a3c390cfSAlp Dener PETSC_EXTERN PetscErrorCode TaoBRGNSetRegularizerObjectiveAndGradientRoutine(Tao, PetscErrorCode (*)(Tao, Vec, PetscReal *, Vec, void *), void *);
432a3c390cfSAlp Dener PETSC_EXTERN PetscErrorCode TaoBRGNSetRegularizerHessianRoutine(Tao, Mat, PetscErrorCode (*)(Tao, Vec, Mat, void *), void *);
433a3c390cfSAlp Dener PETSC_EXTERN PetscErrorCode TaoBRGNSetRegularizerWeight(Tao, PetscReal);
4348ac80d48SXiang Huang PETSC_EXTERN PetscErrorCode TaoBRGNSetL1SmoothEpsilon(Tao, PetscReal);
4358e85b1b3SXiang Huang PETSC_EXTERN PetscErrorCode TaoBRGNSetDictionaryMatrix(Tao, Mat);
436cd1c4666STristan Konolige PETSC_EXTERN PetscErrorCode TaoBRGNGetDampingVector(Tao, Vec *);
4376285c0a3SHansol  Suh 
4386285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMGetMisfitSubsolver(Tao, Tao *);
4396285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMGetRegularizationSubsolver(Tao, Tao *);
4406285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMGetDualVector(Tao, Vec *);
4416285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMGetSpectralPenalty(Tao, PetscReal *);
4426285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMSetSpectralPenalty(Tao, PetscReal);
4436285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoGetADMMParentTao(Tao, Tao *);
4446285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMSetConstraintVectorRHS(Tao, Vec);
4456285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerCoefficient(Tao, PetscReal);
4466285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMisfitConstraintJacobian(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *);
4476285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerConstraintJacobian(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *);
4486285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerHessianRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *);
4496285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerObjectiveAndGradientRoutine(Tao, PetscErrorCode (*)(Tao, Vec, PetscReal *, Vec, void *), void *);
4506285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMisfitHessianRoutine(Tao, Mat, Mat, PetscErrorCode (*)(Tao, Vec, Mat, Mat, void *), void *);
4516285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMisfitObjectiveAndGradientRoutine(Tao, PetscErrorCode (*)(Tao, Vec, PetscReal *, Vec, void *), void *);
4526285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMisfitHessianChangeStatus(Tao, PetscBool);
4536285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegHessianChangeStatus(Tao, PetscBool);
4546285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMSetMinimumSpectralPenalty(Tao, PetscReal);
4556285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMSetRegularizerType(Tao, TaoADMMRegularizerType);
4566285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMGetRegularizerType(Tao, TaoADMMRegularizerType *);
4576285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMSetUpdateType(Tao, TaoADMMUpdateType);
4586285c0a3SHansol  Suh PETSC_EXTERN PetscErrorCode TaoADMMGetUpdateType(Tao, TaoADMMUpdateType *);
459661095bbSAlp Dener 
460661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetType(Tao, TaoALMMType *);
461661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMSetType(Tao, TaoALMMType);
462661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetSubsolver(Tao, Tao *);
463661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMSetSubsolver(Tao, Tao);
464661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetMultipliers(Tao, Vec *);
465661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMSetMultipliers(Tao, Vec);
466661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetPrimalIS(Tao, IS *, IS *);
467661095bbSAlp Dener PETSC_EXTERN PetscErrorCode TaoALMMGetDualIS(Tao, IS *, IS *);
46821ec2d5cSBarry Smith #endif
469