xref: /petsc/include/petscksp.h (revision 94b7f48cc472a54ea2ce57edf1fe19e8a254237c)
173f4d377SMatthew Knepley /* $Id: petscksp.h,v 1.107 2001/08/06 21:16:38 bsmith Exp $ */
2f26ada1bSBarry Smith /*
3f26ada1bSBarry Smith    Defines the interface functions for the Krylov subspace accelerators.
4f26ada1bSBarry Smith */
50a835dfdSSatish Balay #ifndef __PETSCKSP_H
60a835dfdSSatish Balay #define __PETSCKSP_H
70a835dfdSSatish Balay #include "petscpc.h"
8e9fa29b7SSatish Balay PETSC_EXTERN_CXX_BEGIN
92eac72dbSBarry Smith 
101dbb0a54SBarry Smith EXTERN int KSPInitializePackage(const char[]);
111dbb0a54SBarry Smith 
1228ce4d24SBarry Smith /*S
1328ce4d24SBarry Smith      KSP - Abstract PETSc object that manages all Krylov methods
1428ce4d24SBarry Smith 
1528ce4d24SBarry Smith    Level: beginner
1628ce4d24SBarry Smith 
1728ce4d24SBarry Smith   Concepts: Krylov methods
1828ce4d24SBarry Smith 
19*94b7f48cSBarry Smith .seealso:  KSPCreate(), KSPSetType(), KSPType, SNES, TS, PC, KSP
2028ce4d24SBarry Smith S*/
21e2a1c21fSSatish Balay typedef struct _p_KSP*     KSP;
222eac72dbSBarry Smith 
2328ce4d24SBarry Smith /*E
2428ce4d24SBarry Smith     KSPType - String with the name of a PETSc Krylov method or the creation function
2528ce4d24SBarry Smith        with an optional dynamic library name, for example
2628ce4d24SBarry Smith        http://www.mcs.anl.gov/petsc/lib.a:mykspcreate()
2728ce4d24SBarry Smith 
2828ce4d24SBarry Smith    Level: beginner
2928ce4d24SBarry Smith 
3028ce4d24SBarry Smith .seealso: KSPSetType(), KSP
3128ce4d24SBarry Smith E*/
3282bf6240SBarry Smith #define KSPRICHARDSON "richardson"
3382bf6240SBarry Smith #define KSPCHEBYCHEV  "chebychev"
3482bf6240SBarry Smith #define KSPCG         "cg"
35df2a969aSvictorle #define KSPCGNE       "cgne"
3682bf6240SBarry Smith #define KSPGMRES      "gmres"
3782bf6240SBarry Smith #define KSPTCQMR      "tcqmr"
3882bf6240SBarry Smith #define KSPBCGS       "bcgs"
3982bf6240SBarry Smith #define KSPCGS        "cgs"
4082bf6240SBarry Smith #define KSPTFQMR      "tfqmr"
4182bf6240SBarry Smith #define KSPCR         "cr"
4282bf6240SBarry Smith #define KSPLSQR       "lsqr"
4382bf6240SBarry Smith #define KSPPREONLY    "preonly"
4482bf6240SBarry Smith #define KSPQCG        "qcg"
45c9cf9b11SBarry Smith #define KSPBICG       "bicg"
46c38d4ed2SBarry Smith #define KSPFGMRES     "fgmres"
47b4ac9ba4SBarry Smith #define KSPMINRES     "minres"
4801c5daebSSatish Balay #define KSPSYMMLQ     "symmlq"
49bae98175SBarry Smith #define KSPLGMRES     "lgmres"
5082bf6240SBarry Smith typedef char * KSPType;
512eac72dbSBarry Smith 
528ba1e511SMatthew Knepley /* Logging support */
538ba1e511SMatthew Knepley extern int KSP_COOKIE;
54d5ba7fb7SMatthew Knepley extern int KSP_GMRESOrthogonalization;
551dbb0a54SBarry Smith extern int KSP_SetUp, KSP_Solve;
568ba1e511SMatthew Knepley 
57ca44d042SBarry Smith EXTERN int KSPCreate(MPI_Comm,KSP *);
58ca44d042SBarry Smith EXTERN int KSPSetType(KSP,KSPType);
59ca44d042SBarry Smith EXTERN int KSPSetUp(KSP);
60b5d62d44SBarry Smith EXTERN int KSPSetUpOnBlocks(KSP);
61c293cc10SBarry Smith EXTERN int KSPSolve(KSP);
62c293cc10SBarry Smith EXTERN int KSPSolveTranspose(KSP);
63ca44d042SBarry Smith EXTERN int KSPDestroy(KSP);
642eac72dbSBarry Smith 
65b0a32e0cSBarry Smith extern PetscFList KSPList;
661836bdbcSSatish Balay EXTERN int KSPRegisterAll(const char[]);
67ca44d042SBarry Smith EXTERN int KSPRegisterDestroy(void);
682eac72dbSBarry Smith 
6979cabe49SKris Buschelman EXTERN int KSPRegister(const char[],const char[],const char[],int(*)(KSP));
7030de9b25SBarry Smith 
7130de9b25SBarry Smith /*MC
7230de9b25SBarry Smith    KSPRegisterDynamic - Adds a method to the Krylov subspace solver package.
7330de9b25SBarry Smith 
7430de9b25SBarry Smith    Synopsis:
7530de9b25SBarry Smith    int KSPRegisterDynamic(char *name_solver,char *path,char *name_create,int (*routine_create)(KSP))
7630de9b25SBarry Smith 
7730de9b25SBarry Smith    Not Collective
7830de9b25SBarry Smith 
7930de9b25SBarry Smith    Input Parameters:
8030de9b25SBarry Smith +  name_solver - name of a new user-defined solver
8130de9b25SBarry Smith .  path - path (either absolute or relative) the library containing this solver
8230de9b25SBarry Smith .  name_create - name of routine to create method context
8330de9b25SBarry Smith -  routine_create - routine to create method context
8430de9b25SBarry Smith 
8530de9b25SBarry Smith    Notes:
8630de9b25SBarry Smith    KSPRegisterDynamic() may be called multiple times to add several user-defined solvers.
8730de9b25SBarry Smith 
8830de9b25SBarry Smith    If dynamic libraries are used, then the fourth input argument (routine_create)
8930de9b25SBarry Smith    is ignored.
9030de9b25SBarry Smith 
9130de9b25SBarry Smith    Sample usage:
9230de9b25SBarry Smith .vb
9330de9b25SBarry Smith    KSPRegisterDynamic("my_solver",/home/username/my_lib/lib/libO/solaris/mylib.a,
9430de9b25SBarry Smith                "MySolverCreate",MySolverCreate);
9530de9b25SBarry Smith .ve
9630de9b25SBarry Smith 
9730de9b25SBarry Smith    Then, your solver can be chosen with the procedural interface via
9830de9b25SBarry Smith $     KSPSetType(ksp,"my_solver")
9930de9b25SBarry Smith    or at runtime via the option
10030de9b25SBarry Smith $     -ksp_type my_solver
10130de9b25SBarry Smith 
10230de9b25SBarry Smith    Level: advanced
10330de9b25SBarry Smith 
10430de9b25SBarry Smith    Notes: Environmental variables such as ${PETSC_ARCH}, ${PETSC_DIR}, ${PETSC_LIB_DIR}, ${BOPT},
10530de9b25SBarry Smith           and others of the form ${any_environmental_variable} occuring in pathname will be
10630de9b25SBarry Smith           replaced with appropriate values.
10730de9b25SBarry Smith          If your function is not being put into a shared library then use KSPRegister() instead
10830de9b25SBarry Smith 
10930de9b25SBarry Smith .keywords: KSP, register
11030de9b25SBarry Smith 
11130de9b25SBarry Smith .seealso: KSPRegisterAll(), KSPRegisterDestroy()
11230de9b25SBarry Smith 
11330de9b25SBarry Smith M*/
114aa482453SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES)
115f1af5d2fSBarry Smith #define KSPRegisterDynamic(a,b,c,d) KSPRegister(a,b,c,0)
1166df38c32SLois Curfman McInnes #else
117f1af5d2fSBarry Smith #define KSPRegisterDynamic(a,b,c,d) KSPRegister(a,b,c,d)
1186df38c32SLois Curfman McInnes #endif
11982bf6240SBarry Smith 
120ca44d042SBarry Smith EXTERN int KSPGetType(KSP,KSPType *);
121ca44d042SBarry Smith EXTERN int KSPSetPreconditionerSide(KSP,PCSide);
122ca44d042SBarry Smith EXTERN int KSPGetPreconditionerSide(KSP,PCSide*);
12387828ca2SBarry Smith EXTERN int KSPGetTolerances(KSP,PetscReal*,PetscReal*,PetscReal*,int*);
12487828ca2SBarry Smith EXTERN int KSPSetTolerances(KSP,PetscReal,PetscReal,PetscReal,int);
1253a7fca6bSBarry Smith EXTERN int KSPSetInitialGuessNonzero(KSP,PetscTruth);
126ca44d042SBarry Smith EXTERN int KSPGetInitialGuessNonzero(KSP,PetscTruth *);
1276ab422ffSBarry Smith EXTERN int KSPSetInitialGuessKnoll(KSP,PetscTruth);
1286ab422ffSBarry Smith EXTERN int KSPGetInitialGuessKnoll(KSP,PetscTruth*);
1293a7fca6bSBarry Smith EXTERN int KSPSetComputeEigenvalues(KSP,PetscTruth);
1303a7fca6bSBarry Smith EXTERN int KSPSetComputeSingularValues(KSP,PetscTruth);
131ca44d042SBarry Smith EXTERN int KSPSetRhs(KSP,Vec);
132ca44d042SBarry Smith EXTERN int KSPGetRhs(KSP,Vec *);
133ca44d042SBarry Smith EXTERN int KSPSetSolution(KSP,Vec);
134ca44d042SBarry Smith EXTERN int KSPGetSolution(KSP,Vec *);
13587828ca2SBarry Smith EXTERN int KSPGetResidualNorm(KSP,PetscReal*);
136ca44d042SBarry Smith EXTERN int KSPGetIterationNumber(KSP,int*);
1372eac72dbSBarry Smith 
138ca44d042SBarry Smith EXTERN int KSPSetPC(KSP,PC);
139ca44d042SBarry Smith EXTERN int KSPGetPC(KSP,PC*);
140aabeff55SBarry Smith 
14187828ca2SBarry Smith EXTERN int KSPSetMonitor(KSP,int (*)(KSP,int,PetscReal,void*),void *,int (*)(void*));
142ca44d042SBarry Smith EXTERN int KSPClearMonitor(KSP);
143ca44d042SBarry Smith EXTERN int KSPGetMonitorContext(KSP,void **);
1441836bdbcSSatish Balay EXTERN int KSPGetResidualHistory(KSP,PetscReal*[],int *);
1451836bdbcSSatish Balay EXTERN int KSPSetResidualHistory(KSP,PetscReal[],int,PetscTruth);
1464b0e389bSBarry Smith 
1472eac72dbSBarry Smith 
148ca44d042SBarry Smith EXTERN int KSPBuildSolution(KSP,Vec,Vec *);
149ca44d042SBarry Smith EXTERN int KSPBuildResidual(KSP,Vec,Vec,Vec *);
1502eac72dbSBarry Smith 
15187828ca2SBarry Smith EXTERN int KSPRichardsonSetScale(KSP,PetscReal);
15287828ca2SBarry Smith EXTERN int KSPChebychevSetEigenvalues(KSP,PetscReal,PetscReal);
15387828ca2SBarry Smith EXTERN int KSPComputeExtremeSingularValues(KSP,PetscReal*,PetscReal*);
15487828ca2SBarry Smith EXTERN int KSPComputeEigenvalues(KSP,int,PetscReal*,PetscReal*,int *);
15587828ca2SBarry Smith EXTERN int KSPComputeEigenvaluesExplicitly(KSP,int,PetscReal*,PetscReal*);
1564b0e389bSBarry Smith 
1579f236934SBarry Smith #define KSPGMRESSetRestart(ksp,r) PetscTryMethod((ksp),KSPGMRESSetRestart_C,(KSP,int),((ksp),(r)))
1589f236934SBarry Smith #define KSPGMRESSetHapTol(ksp,tol) PetscTryMethod((ksp),KSPGMRESSetHapTol_C,(KSP,PetscReal),((ksp),(tol)))
1599f236934SBarry Smith 
160ca44d042SBarry Smith EXTERN int KSPGMRESSetPreAllocateVectors(KSP);
161ca44d042SBarry Smith EXTERN int KSPGMRESSetOrthogonalization(KSP,int (*)(KSP,int));
162ca44d042SBarry Smith EXTERN int KSPGMRESModifiedGramSchmidtOrthogonalization(KSP,int);
163b49fd9e1SBarry Smith EXTERN int KSPGMRESClassicalGramSchmidtOrthogonalization(KSP,int);
164b49fd9e1SBarry Smith /*E
165b49fd9e1SBarry Smith     KSPGMRESCGSRefinementType - How the classical (unmodified) Gram-Schmidt is performed.
166b49fd9e1SBarry Smith 
167b49fd9e1SBarry Smith    Level: advanced
168b49fd9e1SBarry Smith 
169b49fd9e1SBarry Smith .seealso: KSPGMRESClassicalGramSchmidtOrthogonalization(), KSPGMRESSetOrthogonalization(),
170b49fd9e1SBarry Smith           KSPGMRESSetCGSRefinementType()
171b49fd9e1SBarry Smith 
172b49fd9e1SBarry Smith E*/
173b49fd9e1SBarry Smith typedef enum {KSP_GMRES_CGS_REFINEMENT_NONE, KSP_GMRES_CGS_REFINEMENT_IFNEEDED, KSP_GMRES_CGS_REFINEMENT_ALWAYS} KSPGMRESCGSRefinementType;
174b49fd9e1SBarry Smith EXTERN int KSPGMRESSetCGSRefinementType(KSP,KSPGMRESCGSRefinementType);
17508480c60SBarry Smith 
17687828ca2SBarry Smith EXTERN int KSPFGMRESModifyPCNoChange(KSP,int,int,PetscReal,void*);
177*94b7f48cSBarry Smith EXTERN int KSPFGMRESModifyPCKSP(KSP,int,int,PetscReal,void*);
17887828ca2SBarry Smith EXTERN int KSPFGMRESSetModifyPC(KSP,int (*)(KSP,int,int,PetscReal,void*),void*,int(*)(void*));
179c38d4ed2SBarry Smith 
18087828ca2SBarry Smith EXTERN int KSPQCGSetTrustRegionRadius(KSP,PetscReal);
18187828ca2SBarry Smith EXTERN int KSPQCGGetQuadratic(KSP,PetscReal*);
18287828ca2SBarry Smith EXTERN int KSPQCGGetTrialStepNorm(KSP,PetscReal*);
183121fd945SKris Buschelman 
184ca44d042SBarry Smith EXTERN int KSPSetFromOptions(KSP);
185ca44d042SBarry Smith EXTERN int KSPAddOptionsChecker(int (*)(KSP));
1862eac72dbSBarry Smith 
18787828ca2SBarry Smith EXTERN int KSPSingularValueMonitor(KSP,int,PetscReal,void *);
18887828ca2SBarry Smith EXTERN int KSPDefaultMonitor(KSP,int,PetscReal,void *);
18987828ca2SBarry Smith EXTERN int KSPTrueMonitor(KSP,int,PetscReal,void *);
19087828ca2SBarry Smith EXTERN int KSPDefaultSMonitor(KSP,int,PetscReal,void *);
19187828ca2SBarry Smith EXTERN int KSPVecViewMonitor(KSP,int,PetscReal,void *);
19287828ca2SBarry Smith EXTERN int KSPGMRESKrylovMonitor(KSP,int,PetscReal,void *);
19384cb2905SBarry Smith 
194ca44d042SBarry Smith EXTERN int KSPUnwindPreconditioner(KSP,Vec,Vec);
195ca44d042SBarry Smith EXTERN int KSPDefaultBuildSolution(KSP,Vec,Vec*);
196ca44d042SBarry Smith EXTERN int KSPDefaultBuildResidual(KSP,Vec,Vec,Vec *);
197c01c455dSBarry Smith 
198f0821613SBarry Smith EXTERN int KSPSetOperators(KSP,Mat,Mat,MatStructure);
1991836bdbcSSatish Balay EXTERN int KSPSetOptionsPrefix(KSP,const char[]);
2001836bdbcSSatish Balay EXTERN int KSPAppendOptionsPrefix(KSP,const char[]);
2011836bdbcSSatish Balay EXTERN int KSPGetOptionsPrefix(KSP,char*[]);
2021eb62cbbSBarry Smith 
2031f7f0c4fSBarry Smith EXTERN int KSPSetDiagonalScale(KSP,PetscTruth);
2041f7f0c4fSBarry Smith EXTERN int KSPGetDiagonalScale(KSP,PetscTruth*);
2051f7f0c4fSBarry Smith EXTERN int KSPSetDiagonalScaleFix(KSP,PetscTruth);
2061f7f0c4fSBarry Smith EXTERN int KSPGetDiagonalScaleFix(KSP,PetscTruth*);
2071f7f0c4fSBarry Smith 
208b0a32e0cSBarry Smith EXTERN int KSPView(KSP,PetscViewer);
2091eb62cbbSBarry Smith 
21028ce4d24SBarry Smith /*E
2118a4b9c5cSBarry Smith     KSPNormType - Norm that is passed in the Krylov convergence
2128a4b9c5cSBarry Smith        test routines.
2138a4b9c5cSBarry Smith 
2148a4b9c5cSBarry Smith    Level: advanced
2158a4b9c5cSBarry Smith 
2168a4b9c5cSBarry Smith    Notes: this must match finclude/petscksp.h
2178a4b9c5cSBarry Smith 
218*94b7f48cSBarry Smith .seealso: KSPSolve(), KSPGetConvergedReason(), KSPSetNormType(),
2198a4b9c5cSBarry Smith           KSPSetConvergenceTest()
2208a4b9c5cSBarry Smith E*/
2218a4b9c5cSBarry Smith typedef enum {KSP_NO_NORM               = 0,
2228a4b9c5cSBarry Smith               KSP_PRECONDITIONED_NORM   = 1,
2238a4b9c5cSBarry Smith               KSP_UNPRECONDITIONED_NORM = 2,
2248a4b9c5cSBarry Smith               KSP_NATURAL_NORM          = 3} KSPNormType;
2258a4b9c5cSBarry Smith EXTERN int KSPSetNormType(KSP,KSPNormType);
2268a4b9c5cSBarry Smith 
2278a4b9c5cSBarry Smith /*E
22828ce4d24SBarry Smith     KSPConvergedReason - reason a Krylov method was said to
22928ce4d24SBarry Smith          have converged or diverged
23028ce4d24SBarry Smith 
23128ce4d24SBarry Smith    Level: beginner
23228ce4d24SBarry Smith 
23328ce4d24SBarry Smith    Notes: this must match finclude/petscksp.h
23428ce4d24SBarry Smith 
23586c02ca4SBarry Smith    Developer note: The string versions of these are in
23686c02ca4SBarry Smith      src/sles/ksp/interface/itfunc.c called convergedreasons.
23786c02ca4SBarry Smith      If these enums are changed you much change those.
23886c02ca4SBarry Smith 
23928ce4d24SBarry Smith .seealso: SLESSolve(), KSPSolve(), KSPGetConvergedReason()
24028ce4d24SBarry Smith E*/
241d15094e1SBarry Smith typedef enum {/* converged */
242d15094e1SBarry Smith               KSP_CONVERGED_RTOL               =  2,
243d15094e1SBarry Smith               KSP_CONVERGED_ATOL               =  3,
244b335793eSSatish Balay               KSP_CONVERGED_ITS                =  4,
2458d9717b1SSatish Balay               KSP_CONVERGED_QCG_NEG_CURVE      =  5,
246329f5518SBarry Smith               KSP_CONVERGED_QCG_CONSTRAINED    =  6,
247329f5518SBarry Smith               KSP_CONVERGED_STEP_LENGTH        =  7,
248d15094e1SBarry Smith               /* diverged */
249d15094e1SBarry Smith               KSP_DIVERGED_ITS                 = -3,
250d15094e1SBarry Smith               KSP_DIVERGED_DTOL                = -4,
251d15094e1SBarry Smith               KSP_DIVERGED_BREAKDOWN           = -5,
252b4ac9ba4SBarry Smith               KSP_DIVERGED_BREAKDOWN_BICG      = -6,
253b4ac9ba4SBarry Smith               KSP_DIVERGED_NONSYMMETRIC        = -7,
254b4ac9ba4SBarry Smith               KSP_DIVERGED_INDEFINITE_PC       = -8,
255d15094e1SBarry Smith 
256d15094e1SBarry Smith               KSP_CONVERGED_ITERATING          =  0} KSPConvergedReason;
257d15094e1SBarry Smith 
25887828ca2SBarry Smith EXTERN int KSPSetConvergenceTest(KSP,int (*)(KSP,int,PetscReal,KSPConvergedReason*,void*),void *);
259ca44d042SBarry Smith EXTERN int KSPGetConvergenceContext(KSP,void **);
26087828ca2SBarry Smith EXTERN int KSPDefaultConverged(KSP,int,PetscReal,KSPConvergedReason*,void *);
26187828ca2SBarry Smith EXTERN int KSPSkipConverged(KSP,int,PetscReal,KSPConvergedReason*,void *);
262ca44d042SBarry Smith EXTERN int KSPGetConvergedReason(KSP,KSPConvergedReason *);
263abef13c0SSatish Balay 
264ca44d042SBarry Smith EXTERN int KSPComputeExplicitOperator(KSP,Mat *);
265d4fbbf0eSBarry Smith 
26628ce4d24SBarry Smith /*E
26728ce4d24SBarry Smith     KSPCGType - Determines what type of CG to use
26828ce4d24SBarry Smith 
26928ce4d24SBarry Smith    Level: beginner
27028ce4d24SBarry Smith 
27128ce4d24SBarry Smith .seealso: KSPCGSetType()
27228ce4d24SBarry Smith E*/
2736d4a8577SBarry Smith typedef enum {KSP_CG_SYMMETRIC=1,KSP_CG_HERMITIAN=2} KSPCGType;
27428ce4d24SBarry Smith 
275ca44d042SBarry Smith EXTERN int KSPCGSetType(KSP,KSPCGType);
276e559a7feSLois Curfman McInnes 
277ca44d042SBarry Smith EXTERN int PCPreSolve(PC,KSP);
278ca44d042SBarry Smith EXTERN int PCPostSolve(PC,KSP);
2793369ce9aSBarry Smith 
2801836bdbcSSatish Balay EXTERN int KSPLGMonitorCreate(const char[],const char[],int,int,int,int,PetscDrawLG*);
28187828ca2SBarry Smith EXTERN int KSPLGMonitor(KSP,int,PetscReal,void*);
282b0a32e0cSBarry Smith EXTERN int KSPLGMonitorDestroy(PetscDrawLG);
2831836bdbcSSatish Balay EXTERN int KSPLGTrueMonitorCreate(MPI_Comm,const char[],const char[],int,int,int,int,PetscDrawLG*);
28487828ca2SBarry Smith EXTERN int KSPLGTrueMonitor(KSP,int,PetscReal,void*);
285b0a32e0cSBarry Smith EXTERN int KSPLGTrueMonitorDestroy(PetscDrawLG);
2861eb62cbbSBarry Smith 
287e9fa29b7SSatish Balay PETSC_EXTERN_CXX_END
2882eac72dbSBarry Smith #endif
289