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