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" 82eac72dbSBarry Smith 928ce4d24SBarry Smith /*S 1028ce4d24SBarry Smith KSP - Abstract PETSc object that manages all Krylov methods 1128ce4d24SBarry Smith 1228ce4d24SBarry Smith Level: beginner 1328ce4d24SBarry Smith 1428ce4d24SBarry Smith Concepts: Krylov methods 1528ce4d24SBarry Smith 16435da068SBarry Smith .seealso: KSPCreate(), KSPSetType(), KSPType, SNES, TS, PC, SLES 1728ce4d24SBarry Smith S*/ 18e2a1c21fSSatish Balay typedef struct _p_KSP* KSP; 192eac72dbSBarry Smith 2028ce4d24SBarry Smith /*E 2128ce4d24SBarry Smith KSPType - String with the name of a PETSc Krylov method or the creation function 2228ce4d24SBarry Smith with an optional dynamic library name, for example 2328ce4d24SBarry Smith http://www.mcs.anl.gov/petsc/lib.a:mykspcreate() 2428ce4d24SBarry Smith 2528ce4d24SBarry Smith Level: beginner 2628ce4d24SBarry Smith 2728ce4d24SBarry Smith .seealso: KSPSetType(), KSP 2828ce4d24SBarry Smith E*/ 2982bf6240SBarry Smith #define KSPRICHARDSON "richardson" 3082bf6240SBarry Smith #define KSPCHEBYCHEV "chebychev" 3182bf6240SBarry Smith #define KSPCG "cg" 3282bf6240SBarry Smith #define KSPGMRES "gmres" 3382bf6240SBarry Smith #define KSPTCQMR "tcqmr" 3482bf6240SBarry Smith #define KSPBCGS "bcgs" 3582bf6240SBarry Smith #define KSPCGS "cgs" 3682bf6240SBarry Smith #define KSPTFQMR "tfqmr" 3782bf6240SBarry Smith #define KSPCR "cr" 3882bf6240SBarry Smith #define KSPLSQR "lsqr" 3982bf6240SBarry Smith #define KSPPREONLY "preonly" 4082bf6240SBarry Smith #define KSPQCG "qcg" 41c9cf9b11SBarry Smith #define KSPBICG "bicg" 42c38d4ed2SBarry Smith #define KSPFGMRES "fgmres" 43b4ac9ba4SBarry Smith #define KSPMINRES "minres" 4401c5daebSSatish Balay #define KSPSYMMLQ "symmlq" 45bae98175SBarry Smith #define KSPLGMRES "lgmres" 4682bf6240SBarry Smith typedef char * KSPType; 472eac72dbSBarry Smith 488ba1e511SMatthew Knepley /* Logging support */ 498ba1e511SMatthew Knepley extern int KSP_COOKIE; 50d5ba7fb7SMatthew Knepley extern int KSP_GMRESOrthogonalization; 518ba1e511SMatthew Knepley 52ca44d042SBarry Smith EXTERN int KSPCreate(MPI_Comm,KSP *); 53ca44d042SBarry Smith EXTERN int KSPSetType(KSP,KSPType); 54ca44d042SBarry Smith EXTERN int KSPSetUp(KSP); 55ca44d042SBarry Smith EXTERN int KSPSolve(KSP,int *); 56ca44d042SBarry Smith EXTERN int KSPSolveTranspose(KSP,int *); 57ca44d042SBarry Smith EXTERN int KSPDestroy(KSP); 582eac72dbSBarry Smith 59b0a32e0cSBarry Smith extern PetscFList KSPList; 60ca44d042SBarry Smith EXTERN int KSPRegisterAll(char *); 61ca44d042SBarry Smith EXTERN int KSPRegisterDestroy(void); 622eac72dbSBarry Smith 63ca44d042SBarry Smith EXTERN int KSPRegister(char*,char*,char*,int(*)(KSP)); 64*30de9b25SBarry Smith 65*30de9b25SBarry Smith /*MC 66*30de9b25SBarry Smith KSPRegisterDynamic - Adds a method to the Krylov subspace solver package. 67*30de9b25SBarry Smith 68*30de9b25SBarry Smith Synopsis: 69*30de9b25SBarry Smith int KSPRegisterDynamic(char *name_solver,char *path,char *name_create,int (*routine_create)(KSP)) 70*30de9b25SBarry Smith 71*30de9b25SBarry Smith Not Collective 72*30de9b25SBarry Smith 73*30de9b25SBarry Smith Input Parameters: 74*30de9b25SBarry Smith + name_solver - name of a new user-defined solver 75*30de9b25SBarry Smith . path - path (either absolute or relative) the library containing this solver 76*30de9b25SBarry Smith . name_create - name of routine to create method context 77*30de9b25SBarry Smith - routine_create - routine to create method context 78*30de9b25SBarry Smith 79*30de9b25SBarry Smith Notes: 80*30de9b25SBarry Smith KSPRegisterDynamic() may be called multiple times to add several user-defined solvers. 81*30de9b25SBarry Smith 82*30de9b25SBarry Smith If dynamic libraries are used, then the fourth input argument (routine_create) 83*30de9b25SBarry Smith is ignored. 84*30de9b25SBarry Smith 85*30de9b25SBarry Smith Sample usage: 86*30de9b25SBarry Smith .vb 87*30de9b25SBarry Smith KSPRegisterDynamic("my_solver",/home/username/my_lib/lib/libO/solaris/mylib.a, 88*30de9b25SBarry Smith "MySolverCreate",MySolverCreate); 89*30de9b25SBarry Smith .ve 90*30de9b25SBarry Smith 91*30de9b25SBarry Smith Then, your solver can be chosen with the procedural interface via 92*30de9b25SBarry Smith $ KSPSetType(ksp,"my_solver") 93*30de9b25SBarry Smith or at runtime via the option 94*30de9b25SBarry Smith $ -ksp_type my_solver 95*30de9b25SBarry Smith 96*30de9b25SBarry Smith Level: advanced 97*30de9b25SBarry Smith 98*30de9b25SBarry Smith Notes: Environmental variables such as ${PETSC_ARCH}, ${PETSC_DIR}, ${PETSC_LIB_DIR}, ${BOPT}, 99*30de9b25SBarry Smith and others of the form ${any_environmental_variable} occuring in pathname will be 100*30de9b25SBarry Smith replaced with appropriate values. 101*30de9b25SBarry Smith If your function is not being put into a shared library then use KSPRegister() instead 102*30de9b25SBarry Smith 103*30de9b25SBarry Smith .keywords: KSP, register 104*30de9b25SBarry Smith 105*30de9b25SBarry Smith .seealso: KSPRegisterAll(), KSPRegisterDestroy() 106*30de9b25SBarry Smith 107*30de9b25SBarry Smith M*/ 108aa482453SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES) 109f1af5d2fSBarry Smith #define KSPRegisterDynamic(a,b,c,d) KSPRegister(a,b,c,0) 1106df38c32SLois Curfman McInnes #else 111f1af5d2fSBarry Smith #define KSPRegisterDynamic(a,b,c,d) KSPRegister(a,b,c,d) 1126df38c32SLois Curfman McInnes #endif 11382bf6240SBarry Smith 114ca44d042SBarry Smith EXTERN int KSPGetType(KSP,KSPType *); 115ca44d042SBarry Smith EXTERN int KSPSetPreconditionerSide(KSP,PCSide); 116ca44d042SBarry Smith EXTERN int KSPGetPreconditionerSide(KSP,PCSide*); 11787828ca2SBarry Smith EXTERN int KSPGetTolerances(KSP,PetscReal*,PetscReal*,PetscReal*,int*); 11887828ca2SBarry Smith EXTERN int KSPSetTolerances(KSP,PetscReal,PetscReal,PetscReal,int); 1193a7fca6bSBarry Smith EXTERN int KSPSetInitialGuessNonzero(KSP,PetscTruth); 120ca44d042SBarry Smith EXTERN int KSPGetInitialGuessNonzero(KSP,PetscTruth *); 1216ab422ffSBarry Smith EXTERN int KSPSetInitialGuessKnoll(KSP,PetscTruth); 1226ab422ffSBarry Smith EXTERN int KSPGetInitialGuessKnoll(KSP,PetscTruth*); 1233a7fca6bSBarry Smith EXTERN int KSPSetComputeEigenvalues(KSP,PetscTruth); 1243a7fca6bSBarry Smith EXTERN int KSPSetComputeSingularValues(KSP,PetscTruth); 125ca44d042SBarry Smith EXTERN int KSPSetRhs(KSP,Vec); 126ca44d042SBarry Smith EXTERN int KSPGetRhs(KSP,Vec *); 127ca44d042SBarry Smith EXTERN int KSPSetSolution(KSP,Vec); 128ca44d042SBarry Smith EXTERN int KSPGetSolution(KSP,Vec *); 12987828ca2SBarry Smith EXTERN int KSPGetResidualNorm(KSP,PetscReal*); 130ca44d042SBarry Smith EXTERN int KSPGetIterationNumber(KSP,int*); 1312eac72dbSBarry Smith 132ca44d042SBarry Smith EXTERN int KSPSetPC(KSP,PC); 133ca44d042SBarry Smith EXTERN int KSPGetPC(KSP,PC*); 134aabeff55SBarry Smith 13587828ca2SBarry Smith EXTERN int KSPSetMonitor(KSP,int (*)(KSP,int,PetscReal,void*),void *,int (*)(void*)); 136ca44d042SBarry Smith EXTERN int KSPClearMonitor(KSP); 137ca44d042SBarry Smith EXTERN int KSPGetMonitorContext(KSP,void **); 13887828ca2SBarry Smith EXTERN int KSPGetResidualHistory(KSP,PetscReal **,int *); 13987828ca2SBarry Smith EXTERN int KSPSetResidualHistory(KSP,PetscReal *,int,PetscTruth); 1404b0e389bSBarry Smith 1412eac72dbSBarry Smith 142ca44d042SBarry Smith EXTERN int KSPBuildSolution(KSP,Vec,Vec *); 143ca44d042SBarry Smith EXTERN int KSPBuildResidual(KSP,Vec,Vec,Vec *); 1442eac72dbSBarry Smith 14587828ca2SBarry Smith EXTERN int KSPRichardsonSetScale(KSP,PetscReal); 14687828ca2SBarry Smith EXTERN int KSPChebychevSetEigenvalues(KSP,PetscReal,PetscReal); 14787828ca2SBarry Smith EXTERN int KSPComputeExtremeSingularValues(KSP,PetscReal*,PetscReal*); 14887828ca2SBarry Smith EXTERN int KSPComputeEigenvalues(KSP,int,PetscReal*,PetscReal*,int *); 14987828ca2SBarry Smith EXTERN int KSPComputeEigenvaluesExplicitly(KSP,int,PetscReal*,PetscReal*); 1504b0e389bSBarry Smith 1519f236934SBarry Smith #define KSPGMRESSetRestart(ksp,r) PetscTryMethod((ksp),KSPGMRESSetRestart_C,(KSP,int),((ksp),(r))) 1529f236934SBarry Smith #define KSPGMRESSetHapTol(ksp,tol) PetscTryMethod((ksp),KSPGMRESSetHapTol_C,(KSP,PetscReal),((ksp),(tol))) 1539f236934SBarry Smith 154ca44d042SBarry Smith EXTERN int KSPGMRESSetPreAllocateVectors(KSP); 155ca44d042SBarry Smith EXTERN int KSPGMRESSetOrthogonalization(KSP,int (*)(KSP,int)); 156ca44d042SBarry Smith EXTERN int KSPGMRESUnmodifiedGramSchmidtOrthogonalization(KSP,int); 157ca44d042SBarry Smith EXTERN int KSPGMRESModifiedGramSchmidtOrthogonalization(KSP,int); 158ca44d042SBarry Smith EXTERN int KSPGMRESIROrthogonalization(KSP,int); 15908480c60SBarry Smith 16087828ca2SBarry Smith EXTERN int KSPFGMRESModifyPCNoChange(KSP,int,int,PetscReal,void*); 16187828ca2SBarry Smith EXTERN int KSPFGMRESModifyPCSLES(KSP,int,int,PetscReal,void*); 16287828ca2SBarry Smith EXTERN int KSPFGMRESSetModifyPC(KSP,int (*)(KSP,int,int,PetscReal,void*),void*,int(*)(void*)); 163c38d4ed2SBarry Smith 16487828ca2SBarry Smith EXTERN int KSPQCGSetTrustRegionRadius(KSP,PetscReal); 16587828ca2SBarry Smith EXTERN int KSPQCGGetQuadratic(KSP,PetscReal*); 16687828ca2SBarry Smith EXTERN int KSPQCGGetTrialStepNorm(KSP,PetscReal*); 167121fd945SKris Buschelman 168ca44d042SBarry Smith EXTERN int KSPSetFromOptions(KSP); 169ca44d042SBarry Smith EXTERN int KSPAddOptionsChecker(int (*)(KSP)); 1702eac72dbSBarry Smith 17187828ca2SBarry Smith EXTERN int KSPSingularValueMonitor(KSP,int,PetscReal,void *); 17287828ca2SBarry Smith EXTERN int KSPDefaultMonitor(KSP,int,PetscReal,void *); 17387828ca2SBarry Smith EXTERN int KSPTrueMonitor(KSP,int,PetscReal,void *); 17487828ca2SBarry Smith EXTERN int KSPDefaultSMonitor(KSP,int,PetscReal,void *); 17587828ca2SBarry Smith EXTERN int KSPVecViewMonitor(KSP,int,PetscReal,void *); 17687828ca2SBarry Smith EXTERN int KSPGMRESKrylovMonitor(KSP,int,PetscReal,void *); 17784cb2905SBarry Smith 178ca44d042SBarry Smith EXTERN int KSPUnwindPreconditioner(KSP,Vec,Vec); 179ca44d042SBarry Smith EXTERN int KSPDefaultBuildSolution(KSP,Vec,Vec*); 180ca44d042SBarry Smith EXTERN int KSPDefaultBuildResidual(KSP,Vec,Vec,Vec *); 181c01c455dSBarry Smith 182ca44d042SBarry Smith EXTERN int KSPSetOptionsPrefix(KSP,char*); 183ca44d042SBarry Smith EXTERN int KSPAppendOptionsPrefix(KSP,char*); 184ca44d042SBarry Smith EXTERN int KSPGetOptionsPrefix(KSP,char**); 1851eb62cbbSBarry Smith 186b0a32e0cSBarry Smith EXTERN int KSPView(KSP,PetscViewer); 1871eb62cbbSBarry Smith 18828ce4d24SBarry Smith /*E 1898a4b9c5cSBarry Smith KSPNormType - Norm that is passed in the Krylov convergence 1908a4b9c5cSBarry Smith test routines. 1918a4b9c5cSBarry Smith 1928a4b9c5cSBarry Smith Level: advanced 1938a4b9c5cSBarry Smith 1948a4b9c5cSBarry Smith Notes: this must match finclude/petscksp.h 1958a4b9c5cSBarry Smith 1968a4b9c5cSBarry Smith .seealso: SLESSolve(), KSPSolve(), KSPGetConvergedReason(), KSPSetNormType(), 1978a4b9c5cSBarry Smith KSPSetConvergenceTest() 1988a4b9c5cSBarry Smith E*/ 1998a4b9c5cSBarry Smith typedef enum {KSP_NO_NORM = 0, 2008a4b9c5cSBarry Smith KSP_PRECONDITIONED_NORM = 1, 2018a4b9c5cSBarry Smith KSP_UNPRECONDITIONED_NORM = 2, 2028a4b9c5cSBarry Smith KSP_NATURAL_NORM = 3} KSPNormType; 2038a4b9c5cSBarry Smith EXTERN int KSPSetNormType(KSP,KSPNormType); 2048a4b9c5cSBarry Smith 2058a4b9c5cSBarry Smith /*E 20628ce4d24SBarry Smith KSPConvergedReason - reason a Krylov method was said to 20728ce4d24SBarry Smith have converged or diverged 20828ce4d24SBarry Smith 20928ce4d24SBarry Smith Level: beginner 21028ce4d24SBarry Smith 21128ce4d24SBarry Smith Notes: this must match finclude/petscksp.h 21228ce4d24SBarry Smith 21328ce4d24SBarry Smith .seealso: SLESSolve(), KSPSolve(), KSPGetConvergedReason() 21428ce4d24SBarry Smith E*/ 215d15094e1SBarry Smith typedef enum {/* converged */ 216d15094e1SBarry Smith KSP_CONVERGED_RTOL = 2, 217d15094e1SBarry Smith KSP_CONVERGED_ATOL = 3, 218b335793eSSatish Balay KSP_CONVERGED_ITS = 4, 2198d9717b1SSatish Balay KSP_CONVERGED_QCG_NEG_CURVE = 5, 220329f5518SBarry Smith KSP_CONVERGED_QCG_CONSTRAINED = 6, 221329f5518SBarry Smith KSP_CONVERGED_STEP_LENGTH = 7, 222d15094e1SBarry Smith /* diverged */ 223d15094e1SBarry Smith KSP_DIVERGED_ITS = -3, 224d15094e1SBarry Smith KSP_DIVERGED_DTOL = -4, 225d15094e1SBarry Smith KSP_DIVERGED_BREAKDOWN = -5, 226b4ac9ba4SBarry Smith KSP_DIVERGED_BREAKDOWN_BICG = -6, 227b4ac9ba4SBarry Smith KSP_DIVERGED_NONSYMMETRIC = -7, 228b4ac9ba4SBarry Smith KSP_DIVERGED_INDEFINITE_PC = -8, 229d15094e1SBarry Smith 230d15094e1SBarry Smith KSP_CONVERGED_ITERATING = 0} KSPConvergedReason; 231d15094e1SBarry Smith 23287828ca2SBarry Smith EXTERN int KSPSetConvergenceTest(KSP,int (*)(KSP,int,PetscReal,KSPConvergedReason*,void*),void *); 233ca44d042SBarry Smith EXTERN int KSPGetConvergenceContext(KSP,void **); 23487828ca2SBarry Smith EXTERN int KSPDefaultConverged(KSP,int,PetscReal,KSPConvergedReason*,void *); 23587828ca2SBarry Smith EXTERN int KSPSkipConverged(KSP,int,PetscReal,KSPConvergedReason*,void *); 236ca44d042SBarry Smith EXTERN int KSPGetConvergedReason(KSP,KSPConvergedReason *); 237abef13c0SSatish Balay 238ca44d042SBarry Smith EXTERN int KSPComputeExplicitOperator(KSP,Mat *); 239d4fbbf0eSBarry Smith 24028ce4d24SBarry Smith /*E 24128ce4d24SBarry Smith KSPCGType - Determines what type of CG to use 24228ce4d24SBarry Smith 24328ce4d24SBarry Smith Level: beginner 24428ce4d24SBarry Smith 24528ce4d24SBarry Smith .seealso: KSPCGSetType() 24628ce4d24SBarry Smith E*/ 2476d4a8577SBarry Smith typedef enum {KSP_CG_SYMMETRIC=1,KSP_CG_HERMITIAN=2} KSPCGType; 24828ce4d24SBarry Smith 249ca44d042SBarry Smith EXTERN int KSPCGSetType(KSP,KSPCGType); 250e559a7feSLois Curfman McInnes 251ca44d042SBarry Smith EXTERN int PCPreSolve(PC,KSP); 252ca44d042SBarry Smith EXTERN int PCPostSolve(PC,KSP); 2533369ce9aSBarry Smith 254b0a32e0cSBarry Smith EXTERN int KSPLGMonitorCreate(char*,char*,int,int,int,int,PetscDrawLG*); 25587828ca2SBarry Smith EXTERN int KSPLGMonitor(KSP,int,PetscReal,void*); 256b0a32e0cSBarry Smith EXTERN int KSPLGMonitorDestroy(PetscDrawLG); 257b0a32e0cSBarry Smith EXTERN int KSPLGTrueMonitorCreate(MPI_Comm,char*,char*,int,int,int,int,PetscDrawLG*); 25887828ca2SBarry Smith EXTERN int KSPLGTrueMonitor(KSP,int,PetscReal,void*); 259b0a32e0cSBarry Smith EXTERN int KSPLGTrueMonitorDestroy(PetscDrawLG); 2601eb62cbbSBarry Smith 2612eac72dbSBarry Smith #endif 2622eac72dbSBarry Smith 2632eac72dbSBarry Smith 264