1a5eb4965SSatish Balay #ifdef PETSC_RCS_HEADER 2*65f2ba5bSLois Curfman McInnes static char vcid[] = "$Id: snesmfj.c,v 1.81 1999/03/19 21:22:38 bsmith Exp curfman $"; 381e6777dSBarry Smith #endif 481e6777dSBarry Smith 59a6cb015SBarry Smith #include "src/snes/snesimpl.h" 69a6cb015SBarry Smith #include "src/snes/mf/snesmfj.h" /*I "snes.h" I*/ 781e6777dSBarry Smith 85a655dc6SBarry Smith FList MatSNESMFList = 0; 95a655dc6SBarry Smith int MatSNESMFRegisterAllCalled = 0; 10a4d4d686SBarry Smith 115615d1e5SSatish Balay #undef __FUNC__ 125a655dc6SBarry Smith #define __FUNC__ "MatSNESMFSetType" 139a6cb015SBarry Smith /*@ 14*65f2ba5bSLois Curfman McInnes MatSNESMFSetType - Sets the method that is used to compute the 15*65f2ba5bSLois Curfman McInnes differencing parameter for finite difference matrix-free formulations. 169a6cb015SBarry Smith 179a6cb015SBarry Smith Input Parameters: 185a655dc6SBarry Smith + mat - the matrix free matrix created via MatCreateSNESMF() 199a6cb015SBarry Smith - ftype - the type requested 209a6cb015SBarry Smith 2115091d37SBarry Smith Level: advanced 2215091d37SBarry Smith 23*65f2ba5bSLois Curfman McInnes Notes: 24*65f2ba5bSLois Curfman McInnes For example, such routines can compute h for use in 25*65f2ba5bSLois Curfman McInnes Jacobian-vector products of the form 26*65f2ba5bSLois Curfman McInnes 27*65f2ba5bSLois Curfman McInnes F(x+ha) - F(x) 28*65f2ba5bSLois Curfman McInnes F'(u)a =approx ---------------- 29*65f2ba5bSLois Curfman McInnes h 30*65f2ba5bSLois Curfman McInnes 315a655dc6SBarry Smith .seealso: MatCreateSNESMF(), MatSNESMFRegister() 329a6cb015SBarry Smith @*/ 335a655dc6SBarry Smith int MatSNESMFSetType(Mat mat,char *ftype) 34b9fa9cd0SBarry Smith { 355a655dc6SBarry Smith int ierr, (*r)(MatSNESMFCtx); 365a655dc6SBarry Smith MatSNESMFCtx ctx; 37a4d4d686SBarry Smith 38a4d4d686SBarry Smith PetscFunctionBegin; 39a4d4d686SBarry Smith ierr = MatShellGetContext(mat,(void **)&ctx); CHKERRQ(ierr); 40a4d4d686SBarry Smith 419a6cb015SBarry Smith /* already set, so just return */ 423f1db9ecSBarry Smith if (PetscTypeCompare(ctx->type_name,ftype)) PetscFunctionReturn(0); 43a4d4d686SBarry Smith 449a6cb015SBarry Smith /* destroy the old one if it exists */ 459a6cb015SBarry Smith if (ctx->ops->destroy) { 469a6cb015SBarry Smith ierr = (*ctx->ops->destroy)(ctx);CHKERRQ(ierr); 479a6cb015SBarry Smith } 489a6cb015SBarry Smith 49*65f2ba5bSLois Curfman McInnes /* Get the function pointers for the requrested method */ 505a655dc6SBarry Smith if (!MatSNESMFRegisterAllCalled) {ierr = MatSNESMFRegisterAll(PETSC_NULL); CHKERRQ(ierr);} 519a6cb015SBarry Smith 525a655dc6SBarry Smith ierr = FListFind(ctx->comm, MatSNESMFList, ftype,(int (**)(void *)) &r );CHKERRQ(ierr); 539a6cb015SBarry Smith 545a655dc6SBarry Smith if (!r) SETERRQ(1,1,"Unknown MatSNESMF type given"); 559a6cb015SBarry Smith 569a6cb015SBarry Smith ierr = (*r)(ctx); CHKERRQ(ierr); 579a6cb015SBarry Smith ierr = PetscStrncpy(ctx->type_name,ftype,256);CHKERRQ(ierr); 589a6cb015SBarry Smith 599a6cb015SBarry Smith PetscFunctionReturn(0); 609a6cb015SBarry Smith } 619a6cb015SBarry Smith 629a6cb015SBarry Smith /*MC 63*65f2ba5bSLois Curfman McInnes MatSNESMFRegister - Adds a method to the MatSNESMF registry. 649a6cb015SBarry Smith 659a6cb015SBarry Smith Synopsis: 665a655dc6SBarry Smith MatSNESMFRegister(char *name_solver,char *path,char *name_create,int (*routine_create)(MatSNESMF)) 679a6cb015SBarry Smith 689a6cb015SBarry Smith Not Collective 699a6cb015SBarry Smith 709a6cb015SBarry Smith Input Parameters: 719a6cb015SBarry Smith + name_solver - name of a new user-defined compute-h module 729a6cb015SBarry Smith . path - path (either absolute or relative) the library containing this solver 739a6cb015SBarry Smith . name_create - name of routine to create method context 749a6cb015SBarry Smith - routine_create - routine to create method context 759a6cb015SBarry Smith 7615091d37SBarry Smith Level: developer 7715091d37SBarry Smith 789a6cb015SBarry Smith Notes: 795a655dc6SBarry Smith MatSNESMFRegister() may be called multiple times to add several user-defined solvers. 809a6cb015SBarry Smith 819a6cb015SBarry Smith If dynamic libraries are used, then the fourth input argument (routine_create) 829a6cb015SBarry Smith is ignored. 839a6cb015SBarry Smith 849a6cb015SBarry Smith Sample usage: 859a6cb015SBarry Smith .vb 865a655dc6SBarry Smith MatSNESMFRegister("my_h",/home/username/my_lib/lib/libO/solaris/mylib.a, 879a6cb015SBarry Smith "MyHCreate",MyHCreate); 889a6cb015SBarry Smith .ve 899a6cb015SBarry Smith 909a6cb015SBarry Smith Then, your solver can be chosen with the procedural interface via 915a655dc6SBarry Smith $ MatSNESMFSetType(mfctx,"my_h") 929a6cb015SBarry Smith or at runtime via the option 939a6cb015SBarry Smith $ -snes_mf_type my_h 949a6cb015SBarry Smith 955a655dc6SBarry Smith .keywords: MatSNESMF, register 969a6cb015SBarry Smith 975a655dc6SBarry Smith .seealso: MatSNESMFRegisterAll(), MatSNESMFRegisterDestroy() 989a6cb015SBarry Smith M*/ 999a6cb015SBarry Smith 1009a6cb015SBarry Smith #undef __FUNC__ 1015a655dc6SBarry Smith #define __FUNC__ "MatSNESMFRegister_Private" 1025a655dc6SBarry Smith int MatSNESMFRegister_Private(char *sname,char *path,char *name,int (*function)(MatSNESMFCtx)) 1039a6cb015SBarry Smith { 1049a6cb015SBarry Smith int ierr; 1059a6cb015SBarry Smith char fullname[256]; 1069a6cb015SBarry Smith 1079a6cb015SBarry Smith PetscFunctionBegin; 1089a6cb015SBarry Smith PetscStrcpy(fullname,path); PetscStrcat(fullname,":");PetscStrcat(fullname,name); 1095a655dc6SBarry Smith ierr = FListAdd_Private(&MatSNESMFList,sname,fullname,(int (*)(void*))function);CHKERRQ(ierr); 1109a6cb015SBarry Smith PetscFunctionReturn(0); 1119a6cb015SBarry Smith } 1129a6cb015SBarry Smith 1139a6cb015SBarry Smith 1149a6cb015SBarry Smith #undef __FUNC__ 1155a655dc6SBarry Smith #define __FUNC__ "MatSNESMFRegisterDestroy" 1169a6cb015SBarry Smith /*@C 1175a655dc6SBarry Smith MatSNESMFRegisterDestroy - Frees the list of MatSNESMF methods that were 1185a655dc6SBarry Smith registered by MatSNESMFRegister(). 1199a6cb015SBarry Smith 1209a6cb015SBarry Smith Not Collective 1219a6cb015SBarry Smith 12215091d37SBarry Smith Level: developer 12315091d37SBarry Smith 1245a655dc6SBarry Smith .keywords: MatSNESMF, register, destroy 1259a6cb015SBarry Smith 1265a655dc6SBarry Smith .seealso: MatSNESMFRegister(), MatSNESMFRegisterAll() 1279a6cb015SBarry Smith @*/ 1285a655dc6SBarry Smith int MatSNESMFRegisterDestroy(void) 1299a6cb015SBarry Smith { 1309a6cb015SBarry Smith int ierr; 1319a6cb015SBarry Smith 1329a6cb015SBarry Smith PetscFunctionBegin; 1335a655dc6SBarry Smith if (MatSNESMFList) { 1345a655dc6SBarry Smith ierr = FListDestroy( MatSNESMFList );CHKERRQ(ierr); 1355a655dc6SBarry Smith MatSNESMFList = 0; 1369a6cb015SBarry Smith } 1375a655dc6SBarry Smith MatSNESMFRegisterAllCalled = 0; 1389a6cb015SBarry Smith PetscFunctionReturn(0); 1399a6cb015SBarry Smith } 1409a6cb015SBarry Smith 1419a6cb015SBarry Smith /* ----------------------------------------------------------------------------------------*/ 142a4d4d686SBarry Smith #undef __FUNC__ 1435a655dc6SBarry Smith #define __FUNC__ "MatSNESMFDestroy_Private" 1445a655dc6SBarry Smith int MatSNESMFDestroy_Private(Mat mat) 145a4d4d686SBarry Smith { 146a4d4d686SBarry Smith int ierr; 1475a655dc6SBarry Smith MatSNESMFCtx ctx; 148fae171e0SBarry Smith 1493a40ed3dSBarry Smith PetscFunctionBegin; 1500a25c783SBarry Smith ierr = MatShellGetContext(mat,(void **)&ctx);CHKERRQ(ierr); 151b9fa9cd0SBarry Smith ierr = VecDestroy(ctx->w); CHKERRQ(ierr); 1529a6cb015SBarry Smith if (ctx->ops->destroy) {ierr = (*ctx->ops->destroy)(ctx);CHKERRQ(ierr);} 153b4fd4287SBarry Smith if (ctx->sp) {ierr = PCNullSpaceDestroy(ctx->sp);CHKERRQ(ierr);} 1549a6cb015SBarry Smith PetscFree(ctx->ops); 155fae171e0SBarry Smith PetscFree(ctx); 1563a40ed3dSBarry Smith PetscFunctionReturn(0); 157b9fa9cd0SBarry Smith } 15850361f65SLois Curfman McInnes 1595615d1e5SSatish Balay #undef __FUNC__ 1605a655dc6SBarry Smith #define __FUNC__ "MatSNESMFView_Private" 16139e2f89bSBarry Smith /* 1625a655dc6SBarry Smith MatSNESMFView_Private - Views matrix-free parameters. 1638f6e3e37SBarry Smith 16439e2f89bSBarry Smith */ 1655a655dc6SBarry Smith int MatSNESMFView_Private(Mat J,Viewer viewer) 166eb9086c3SLois Curfman McInnes { 167eb9086c3SLois Curfman McInnes int ierr; 1685a655dc6SBarry Smith MatSNESMFCtx ctx; 169eb9086c3SLois Curfman McInnes MPI_Comm comm; 170eb9086c3SLois Curfman McInnes FILE *fd; 171eb9086c3SLois Curfman McInnes ViewerType vtype; 172eb9086c3SLois Curfman McInnes 1733a40ed3dSBarry Smith PetscFunctionBegin; 1742d0c0e3bSBarry Smith ierr = PetscObjectGetComm((PetscObject)J,&comm);CHKERRQ(ierr); 175eb9086c3SLois Curfman McInnes ierr = MatShellGetContext(J,(void **)&ctx); CHKERRQ(ierr); 176eb9086c3SLois Curfman McInnes ierr = ViewerGetType(viewer,&vtype); CHKERRQ(ierr); 177eb9086c3SLois Curfman McInnes ierr = ViewerASCIIGetPointer(viewer,&fd); CHKERRQ(ierr); 1783f1db9ecSBarry Smith if (PetscTypeCompare(vtype,ASCII_VIEWER)) { 179eb9086c3SLois Curfman McInnes PetscFPrintf(comm,fd," SNES matrix-free approximation:\n"); 180eb9086c3SLois Curfman McInnes PetscFPrintf(comm,fd," err=%g (relative error in function evaluation)\n",ctx->error_rel); 1819a6cb015SBarry Smith PetscFPrintf(ctx->comm,fd," Using %s compute h routine\n",ctx->type_name); 1829a6cb015SBarry Smith if (ctx->ops->view) { 1839a6cb015SBarry Smith ierr = (*ctx->ops->view)(ctx,viewer);CHKERRQ(ierr); 1849a6cb015SBarry Smith } 1855cd90555SBarry Smith } else { 1865cd90555SBarry Smith SETERRQ(1,1,"Viewer type not supported for this object"); 187eb9086c3SLois Curfman McInnes } 1883a40ed3dSBarry Smith PetscFunctionReturn(0); 189eb9086c3SLois Curfman McInnes } 190eb9086c3SLois Curfman McInnes 191be726c96SBarry Smith #undef __FUNC__ 1925a655dc6SBarry Smith #define __FUNC__ "MatSNESMFAssemblyEnd_Private" 193be726c96SBarry Smith /* 1945a655dc6SBarry Smith MatSNESMFAssemblyEnd_Private - Resets the ctx->ncurrenth to zero. This 195*65f2ba5bSLois Curfman McInnes allows the user to indicate the beginning of a new linear solve by calling 196be726c96SBarry Smith MatAssemblyXXX() on the matrix free matrix. This then allows the 197*65f2ba5bSLois Curfman McInnes MatSNESMFCreate_WP() to properly compute ||U|| only the first time 198*65f2ba5bSLois Curfman McInnes in the linear solver rather than every time. 199be726c96SBarry Smith */ 2005a655dc6SBarry Smith int MatSNESMFAssemblyEnd_Private(Mat J) 201be726c96SBarry Smith { 202be726c96SBarry Smith int ierr; 203be726c96SBarry Smith 204be726c96SBarry Smith PetscFunctionBegin; 2055a655dc6SBarry Smith ierr = MatSNESMFResetHHistory(J);CHKERRQ(ierr); 206be726c96SBarry Smith PetscFunctionReturn(0); 207be726c96SBarry Smith } 208be726c96SBarry Smith 209c481317fSBarry Smith 2105615d1e5SSatish Balay #undef __FUNC__ 2115a655dc6SBarry Smith #define __FUNC__ "MatSNESMFMult_Private" 212eb9086c3SLois Curfman McInnes /* 2135a655dc6SBarry Smith MatSNESMFMult_Private - Default matrix-free form for Jacobian-vector 214eb9086c3SLois Curfman McInnes product, y = F'(u)*a: 215a4d4d686SBarry Smith 2169a6cb015SBarry Smith y ~= ( F(u + ha) - F(u) )/h, 217eb9086c3SLois Curfman McInnes where F = nonlinear function, as set by SNESSetFunction() 218eb9086c3SLois Curfman McInnes u = current iterate 219eb9086c3SLois Curfman McInnes h = difference interval 220eb9086c3SLois Curfman McInnes */ 2215a655dc6SBarry Smith int MatSNESMFMult_Private(Mat mat,Vec a,Vec y) 22239e2f89bSBarry Smith { 2235a655dc6SBarry Smith MatSNESMFCtx ctx; 224fae171e0SBarry Smith SNES snes; 225a4d4d686SBarry Smith Scalar h, mone = -1.0; 226fae171e0SBarry Smith Vec w,U,F; 227a305c92eSSatish Balay int ierr, (*eval_fct)(SNES,Vec,Vec)=0; 22839e2f89bSBarry Smith 2293a40ed3dSBarry Smith PetscFunctionBegin; 2309a6cb015SBarry Smith /* We log matrix-free matrix-vector products separately, so that we can 2319a6cb015SBarry Smith separate the performance monitoring from the cases that use conventional 2329a6cb015SBarry Smith storage. We may eventually modify event logging to associate events 2339a6cb015SBarry Smith with particular objects, hence alleviating the more general problem. */ 23456cd22aeSBarry Smith PLogEventBegin(MAT_MatrixFreeMult,a,y,0,0); 23556cd22aeSBarry Smith 2366e38a044SBarry Smith ierr = MatShellGetContext(mat,(void **)&ctx); CHKERRQ(ierr); 237fae171e0SBarry Smith snes = ctx->snes; 238fae171e0SBarry Smith w = ctx->w; 239fae171e0SBarry Smith 24078b31e54SBarry Smith ierr = SNESGetSolution(snes,&U); CHKERRQ(ierr); 24183e56afdSLois Curfman McInnes if (snes->method_class == SNES_NONLINEAR_EQUATIONS) { 24283e56afdSLois Curfman McInnes eval_fct = SNESComputeFunction; 24378b31e54SBarry Smith ierr = SNESGetFunction(snes,&F); CHKERRQ(ierr); 244a4d4d686SBarry Smith } else if (snes->method_class == SNES_UNCONSTRAINED_MINIMIZATION) { 24583e56afdSLois Curfman McInnes eval_fct = SNESComputeGradient; 24683e56afdSLois Curfman McInnes ierr = SNESGetGradient(snes,&F); CHKERRQ(ierr); 247a4d4d686SBarry Smith } else SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Invalid method class"); 24850361f65SLois Curfman McInnes 2499a6cb015SBarry Smith if (!ctx->ops->compute) { 2505a655dc6SBarry Smith ierr = MatSNESMFSetType(mat,"default");CHKERRQ(ierr); 2515a655dc6SBarry Smith ierr = MatSNESMFSetFromOptions(mat);CHKERRQ(ierr); 2529a6cb015SBarry Smith } 2539a6cb015SBarry Smith ierr = (*ctx->ops->compute)(ctx,U,a,&h); CHKERRQ(ierr); 254a4d4d686SBarry Smith 255a4d4d686SBarry Smith /* keep a record of the current differencing parameter h */ 256a4d4d686SBarry Smith ctx->currenth = h; 257a4d4d686SBarry Smith #if defined(USE_PETSC_COMPLEX) 258a4d4d686SBarry Smith PLogInfo(mat,"Current differencing parameter: %g + %g i\n",PetscReal(h),PetscImaginary(h)); 259a4d4d686SBarry Smith #else 260a4d4d686SBarry Smith PLogInfo(mat,"Current differencing parameter: %g\n",h); 261a4d4d686SBarry Smith #endif 262a4d4d686SBarry Smith if (ctx->historyh && ctx->ncurrenth < ctx->maxcurrenth) { 263a4d4d686SBarry Smith ctx->historyh[ctx->ncurrenth++] = h; 264be726c96SBarry Smith } else { 265be726c96SBarry Smith ctx->ncurrenth++; 266a4d4d686SBarry Smith } 267a4d4d686SBarry Smith 268a4d4d686SBarry Smith /* Evaluate function at F(u + ha) */ 269a4d4d686SBarry Smith ierr = VecWAXPY(&h,a,U,w); CHKERRQ(ierr); 270a4d4d686SBarry Smith ierr = eval_fct(snes,w,y); CHKERRQ(ierr); 271a4d4d686SBarry Smith 272a4d4d686SBarry Smith ierr = VecAXPY(&mone,F,y); CHKERRQ(ierr); 273a4d4d686SBarry Smith h = 1.0/h; 274a4d4d686SBarry Smith ierr = VecScale(&h,y); CHKERRQ(ierr); 275a4d4d686SBarry Smith if (ctx->sp) {ierr = PCNullSpaceRemove(ctx->sp,y); CHKERRQ(ierr);} 276a4d4d686SBarry Smith 277a4d4d686SBarry Smith PLogEventEnd(MAT_MatrixFreeMult,a,y,0,0); 278a4d4d686SBarry Smith PetscFunctionReturn(0); 279a4d4d686SBarry Smith } 280a4d4d686SBarry Smith 281a4d4d686SBarry Smith #undef __FUNC__ 2825a655dc6SBarry Smith #define __FUNC__ "MatCreateSNESMF" 283a4d4d686SBarry Smith /*@C 284*65f2ba5bSLois Curfman McInnes MatCreateSNESMF - Creates a matrix-free matrix context for use with 285*65f2ba5bSLois Curfman McInnes a SNES solver. This matrix can be used as the Jacobian argument for 286*65f2ba5bSLois Curfman McInnes the routine SNESSetJacobian(). 287a4d4d686SBarry Smith 288a4d4d686SBarry Smith Collective on SNES and Vec 289a4d4d686SBarry Smith 290a4d4d686SBarry Smith Input Parameters: 291a4d4d686SBarry Smith + snes - the SNES context 292a4d4d686SBarry Smith - x - vector where SNES solution is to be stored. 293a4d4d686SBarry Smith 294a4d4d686SBarry Smith Output Parameter: 295a4d4d686SBarry Smith . J - the matrix-free matrix 296a4d4d686SBarry Smith 29715091d37SBarry Smith Level: advanced 29815091d37SBarry Smith 299a4d4d686SBarry Smith Notes: 300a4d4d686SBarry Smith The matrix-free matrix context merely contains the function pointers 301a4d4d686SBarry Smith and work space for performing finite difference approximations of 302*65f2ba5bSLois Curfman McInnes Jacobian-vector products, F'(u)*a, 3039a6cb015SBarry Smith 3049a6cb015SBarry Smith The default code uses the following approach to compute h 305a4d4d686SBarry Smith 306a4d4d686SBarry Smith .vb 307*65f2ba5bSLois Curfman McInnes F'(u)*a = [F(u+h*a) - F(u)]/h where 308a4d4d686SBarry Smith h = error_rel*u'a/||a||^2 if |u'a| > umin*||a||_{1} 309a4d4d686SBarry Smith = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2 otherwise 310a4d4d686SBarry Smith where 311a4d4d686SBarry Smith error_rel = square root of relative error in function evaluation 312a4d4d686SBarry Smith umin = minimum iterate parameter 313a4d4d686SBarry Smith .ve 314a4d4d686SBarry Smith 3155a655dc6SBarry Smith The user can set the error_rel via MatSNESMFSetFunctionError() and 316*65f2ba5bSLois Curfman McInnes umin via MatSNESMFDefaultSetUmin(); see the nonlinear solvers chapter 317*65f2ba5bSLois Curfman McInnes of the users manual for details. 318a4d4d686SBarry Smith 319a4d4d686SBarry Smith The user should call MatDestroy() when finished with the matrix-free 320a4d4d686SBarry Smith matrix context. 321a4d4d686SBarry Smith 322a4d4d686SBarry Smith Options Database Keys: 323a4d4d686SBarry Smith + -snes_mf_err <error_rel> - Sets error_rel 3249a6cb015SBarry Smith . -snes_mf_unim <umin> - Sets umin (for default PETSc routine that computes h only) 325a4d4d686SBarry Smith - -snes_mf_ksp_monitor - KSP monitor routine that prints differencing h 326a4d4d686SBarry Smith 327a4d4d686SBarry Smith .keywords: SNES, default, matrix-free, create, matrix 328a4d4d686SBarry Smith 3295a655dc6SBarry Smith .seealso: MatDestroy(), MatSNESMFSetFunctionError(), MatSNESMFDefaultSetUmin() 3305a655dc6SBarry Smith MatSNESMFSetHHistory(), MatSNESMFResetHHistory(), 3315a655dc6SBarry Smith MatSNESMFGetH(),MatSNESMFKSPMonitor(), MatSNESMFRegister() 332a4d4d686SBarry Smith 333a4d4d686SBarry Smith @*/ 3345a655dc6SBarry Smith int MatCreateSNESMF(SNES snes,Vec x, Mat *J) 335a4d4d686SBarry Smith { 336a4d4d686SBarry Smith MPI_Comm comm; 3375a655dc6SBarry Smith MatSNESMFCtx mfctx; 3389a6cb015SBarry Smith int n, nloc, ierr; 339a4d4d686SBarry Smith 340a4d4d686SBarry Smith PetscFunctionBegin; 3415a655dc6SBarry Smith mfctx = (MatSNESMFCtx) PetscMalloc(sizeof(struct _p_MatSNESMFCtx)); CHKPTRQ(mfctx); 3425a655dc6SBarry Smith PLogObjectMemory(snes,sizeof(MatSNESMFCtx)); 3439a6cb015SBarry Smith mfctx->comm = snes->comm; 344a4d4d686SBarry Smith mfctx->sp = 0; 345a4d4d686SBarry Smith mfctx->snes = snes; 346a4d4d686SBarry Smith mfctx->error_rel = 1.e-8; /* assumes double precision */ 347a4d4d686SBarry Smith mfctx->currenth = 0.0; 348a4d4d686SBarry Smith mfctx->historyh = PETSC_NULL; 349a4d4d686SBarry Smith mfctx->ncurrenth = 0; 350a4d4d686SBarry Smith mfctx->maxcurrenth = 0; 351a4d4d686SBarry Smith 3529a6cb015SBarry Smith /* 3539a6cb015SBarry Smith Create the empty data structure to contain compute-h routines. 3549a6cb015SBarry Smith These will be filled in below from the command line options or 3555a655dc6SBarry Smith a later call with MatSNESMFSetType() or if that is not called 3565a655dc6SBarry Smith then it will default in the first use of MatSNESMFMult_private() 3579a6cb015SBarry Smith */ 3589a6cb015SBarry Smith mfctx->ops = (MFOps *)PetscMalloc(sizeof(MFOps)); CHKPTRQ(mfctx->ops); 3599a6cb015SBarry Smith mfctx->ops->compute = 0; 3609a6cb015SBarry Smith mfctx->ops->destroy = 0; 3619a6cb015SBarry Smith mfctx->ops->view = 0; 3629a6cb015SBarry Smith mfctx->ops->printhelp = 0; 3639a6cb015SBarry Smith mfctx->ops->setfromoptions = 0; 3649a6cb015SBarry Smith mfctx->hctx = 0; 3659a6cb015SBarry Smith 366a4d4d686SBarry Smith ierr = VecDuplicate(x,&mfctx->w); CHKERRQ(ierr); 367a4d4d686SBarry Smith ierr = PetscObjectGetComm((PetscObject)x,&comm); CHKERRQ(ierr); 368a4d4d686SBarry Smith ierr = VecGetSize(x,&n); CHKERRQ(ierr); 369a4d4d686SBarry Smith ierr = VecGetLocalSize(x,&nloc); CHKERRQ(ierr); 370ffa0fd9eSBarry Smith ierr = MatCreateShell(comm,nloc,nloc,n,n,mfctx,J); CHKERRQ(ierr); 3715a655dc6SBarry Smith ierr = MatShellSetOperation(*J,MATOP_MULT,(void*)MatSNESMFMult_Private);CHKERRQ(ierr); 3725a655dc6SBarry Smith ierr = MatShellSetOperation(*J,MATOP_DESTROY,(void *)MatSNESMFDestroy_Private);CHKERRQ(ierr); 3735a655dc6SBarry Smith ierr = MatShellSetOperation(*J,MATOP_VIEW,(void *)MatSNESMFView_Private); CHKERRQ(ierr); 3745a655dc6SBarry Smith ierr = MatShellSetOperation(*J,MATOP_ASSEMBLY_END,(void *)MatSNESMFAssemblyEnd_Private);CHKERRQ(ierr); 375a4d4d686SBarry Smith PLogObjectParent(*J,mfctx->w); 376a4d4d686SBarry Smith PLogObjectParent(snes,*J); 3779a6cb015SBarry Smith 3789a6cb015SBarry Smith mfctx->mat = *J; 3799a6cb015SBarry Smith 3809a6cb015SBarry Smith 3819a6cb015SBarry Smith PetscFunctionReturn(0); 3829a6cb015SBarry Smith } 3839a6cb015SBarry Smith 3849a6cb015SBarry Smith #undef __FUNC__ 3855a655dc6SBarry Smith #define __FUNC__ "MatSNESMFSetFromOptions" 3869a6cb015SBarry Smith /*@ 3875a655dc6SBarry Smith MatSNESMFSetFromOptions - Sets the MatSNESMF options from the command line 3889a6cb015SBarry Smith parameter. 3899a6cb015SBarry Smith 3909a6cb015SBarry Smith Collective on Mat 3919a6cb015SBarry Smith 3929a6cb015SBarry Smith Input Parameters: 3935a655dc6SBarry Smith . mat - the matrix obtained with MatCreateSNESMF() 3945a655dc6SBarry Smith 3955a655dc6SBarry Smith Options Database Keys: 3965a655dc6SBarry Smith + -snes_mf_type - <default,wp> 3975a655dc6SBarry Smith - -snes_mf_err - square root of estimated relative error in function evaluation 3989a6cb015SBarry Smith 39915091d37SBarry Smith Level: advanced 40015091d37SBarry Smith 4019a6cb015SBarry Smith .keywords: SNES, matrix-free, parameters 4029a6cb015SBarry Smith 4035a655dc6SBarry Smith .seealso: MatCreateSNESMF(),MatSNESMFSetHHistory(), 4045a655dc6SBarry Smith MatSNESMFResetHHistory(), MatSNESMFKSPMonitor() 4059a6cb015SBarry Smith @*/ 4065a655dc6SBarry Smith int MatSNESMFSetFromOptions(Mat mat) 4079a6cb015SBarry Smith { 4085a655dc6SBarry Smith MatSNESMFCtx mfctx; 4099a6cb015SBarry Smith int ierr,flg; 4109a6cb015SBarry Smith char ftype[256],p[64]; 4119a6cb015SBarry Smith 4129a6cb015SBarry Smith PetscFunctionBegin; 4139a6cb015SBarry Smith ierr = MatShellGetContext(mat,(void **)&mfctx); CHKERRQ(ierr); 4149a6cb015SBarry Smith if (mfctx) { 4159a6cb015SBarry Smith /* allow user to set the type */ 4169a6cb015SBarry Smith ierr = OptionsGetString(mfctx->snes->prefix,"-snes_mf_type",ftype,256,&flg);CHKERRQ(ierr); 4179a6cb015SBarry Smith if (flg) { 4185a655dc6SBarry Smith ierr = MatSNESMFSetType(mat,ftype);CHKERRQ(ierr); 4199a6cb015SBarry Smith } 4209a6cb015SBarry Smith 4219a6cb015SBarry Smith ierr = OptionsGetDouble(mfctx->snes->prefix,"-snes_mf_err",&mfctx->error_rel,&flg);CHKERRQ(ierr); 4229a6cb015SBarry Smith if (mfctx->ops->setfromoptions) { 4239a6cb015SBarry Smith ierr = (*mfctx->ops->setfromoptions)(mfctx);CHKERRQ(ierr); 4249a6cb015SBarry Smith } 4259a6cb015SBarry Smith 4269a6cb015SBarry Smith ierr = OptionsHasName(PETSC_NULL,"-help",&flg); CHKERRQ(ierr); 4279a6cb015SBarry Smith PetscStrcpy(p,"-"); 4289a6cb015SBarry Smith if (mfctx->snes->prefix) PetscStrcat(p,mfctx->snes->prefix); 4299a6cb015SBarry Smith if (flg) { 4309a6cb015SBarry Smith (*PetscHelpPrintf)(mfctx->snes->comm," %ssnes_mf_err <err>: set sqrt rel error in function (default %g)\n",p,mfctx->error_rel); 4319a6cb015SBarry Smith if (mfctx->ops->printhelp) { 4329a6cb015SBarry Smith (*mfctx->ops->printhelp)(mfctx);CHKERRQ(ierr); 4339a6cb015SBarry Smith } 4349a6cb015SBarry Smith } 4359a6cb015SBarry Smith } 436a4d4d686SBarry Smith PetscFunctionReturn(0); 437a4d4d686SBarry Smith } 438a4d4d686SBarry Smith 439a4d4d686SBarry Smith #undef __FUNC__ 4405a655dc6SBarry Smith #define __FUNC__ "MatSNESMFGetH" 441a4d4d686SBarry Smith /*@ 442*65f2ba5bSLois Curfman McInnes MatSNESMFGetH - Gets the last value that was used as the differencing 443a4d4d686SBarry Smith parameter. 444a4d4d686SBarry Smith 445a4d4d686SBarry Smith Not Collective 446a4d4d686SBarry Smith 447a4d4d686SBarry Smith Input Parameters: 4485a655dc6SBarry Smith . mat - the matrix obtained with MatCreateSNESMF() 449a4d4d686SBarry Smith 450a4d4d686SBarry Smith Output Paramter: 451a4d4d686SBarry Smith . h - the differencing step size 452a4d4d686SBarry Smith 45315091d37SBarry Smith Level: advanced 45415091d37SBarry Smith 455a4d4d686SBarry Smith .keywords: SNES, matrix-free, parameters 456a4d4d686SBarry Smith 4575a655dc6SBarry Smith .seealso: MatCreateSNESMF(),MatSNESMFSetHHistory(), 4585a655dc6SBarry Smith MatSNESMFResetHHistory(),MatSNESMFKSPMonitor() 459a4d4d686SBarry Smith @*/ 4605a655dc6SBarry Smith int MatSNESMFGetH(Mat mat,Scalar *h) 461a4d4d686SBarry Smith { 4625a655dc6SBarry Smith MatSNESMFCtx ctx; 463a4d4d686SBarry Smith int ierr; 464a4d4d686SBarry Smith 465a4d4d686SBarry Smith PetscFunctionBegin; 466a4d4d686SBarry Smith ierr = MatShellGetContext(mat,(void **)&ctx); CHKERRQ(ierr); 467a4d4d686SBarry Smith if (ctx) { 468a4d4d686SBarry Smith *h = ctx->currenth; 469a4d4d686SBarry Smith } 470a4d4d686SBarry Smith PetscFunctionReturn(0); 471a4d4d686SBarry Smith } 472a4d4d686SBarry Smith 473a4d4d686SBarry Smith #undef __FUNC__ 4745a655dc6SBarry Smith #define __FUNC__ "MatSNESMFKSPMonitor" 475a4d4d686SBarry Smith /* 4765a655dc6SBarry Smith MatSNESMFKSPMonitor - A KSP monitor for use with the default PETSc 477*65f2ba5bSLois Curfman McInnes SNES matrix free routines. Prints the differencing parameter used at 478*65f2ba5bSLois Curfman McInnes each step. 479a4d4d686SBarry Smith */ 4805a655dc6SBarry Smith int MatSNESMFKSPMonitor(KSP ksp,int n,double rnorm,void *dummy) 481a4d4d686SBarry Smith { 482a4d4d686SBarry Smith PC pc; 4835a655dc6SBarry Smith MatSNESMFCtx ctx; 484a4d4d686SBarry Smith int ierr; 485a4d4d686SBarry Smith Mat mat; 486a4d4d686SBarry Smith MPI_Comm comm; 487a4d4d686SBarry Smith PetscTruth nonzeroinitialguess; 488a4d4d686SBarry Smith 489a4d4d686SBarry Smith PetscFunctionBegin; 490a4d4d686SBarry Smith ierr = PetscObjectGetComm((PetscObject)ksp,&comm); CHKERRQ(ierr); 491a4d4d686SBarry Smith ierr = KSPGetPC(ksp,&pc); CHKERRQ(ierr); 492a4d4d686SBarry Smith ierr = KSPGetInitialGuessNonzero(ksp,&nonzeroinitialguess); CHKERRQ(ierr); 493a4d4d686SBarry Smith ierr = PCGetOperators(pc,&mat,PETSC_NULL,PETSC_NULL); CHKERRQ(ierr); 494a4d4d686SBarry Smith ierr = MatShellGetContext(mat,(void **)&ctx); CHKERRQ(ierr); 4959a6cb015SBarry Smith if (!ctx) { 4969a6cb015SBarry Smith SETERRQ(1,1,"Matrix is not a matrix free shell matrix"); 4979a6cb015SBarry Smith } 498a4d4d686SBarry Smith if (n > 0 || nonzeroinitialguess) { 499a4d4d686SBarry Smith #if defined(USE_PETSC_COMPLEX) 500a4d4d686SBarry Smith PetscPrintf(comm,"%d KSP Residual norm %14.12e h %g + %g i\n",n,rnorm, 501a4d4d686SBarry Smith PetscReal(ctx->currenth),PetscImaginary(ctx->currenth)); 502a4d4d686SBarry Smith #else 503a4d4d686SBarry Smith PetscPrintf(comm,"%d KSP Residual norm %14.12e h %g \n",n,rnorm,ctx->currenth); 504a4d4d686SBarry Smith #endif 505a4d4d686SBarry Smith } else { 506a4d4d686SBarry Smith PetscPrintf(comm,"%d KSP Residual norm %14.12e\n",n,rnorm); 507a4d4d686SBarry Smith } 508a4d4d686SBarry Smith PetscFunctionReturn(0); 509a4d4d686SBarry Smith } 510a4d4d686SBarry Smith 511a4d4d686SBarry Smith #undef __FUNC__ 5125a655dc6SBarry Smith #define __FUNC__ "MatSNESMFSetFunctionError" 513a4d4d686SBarry Smith /*@ 5145a655dc6SBarry Smith MatSNESMFSetFunctionError - Sets the error_rel for the approximation of 515a4d4d686SBarry Smith matrix-vector products using finite differences. 516a4d4d686SBarry Smith 517a4d4d686SBarry Smith Collective on Mat 518a4d4d686SBarry Smith 519a4d4d686SBarry Smith Input Parameters: 5205a655dc6SBarry Smith + mat - the matrix free matrix created via MatCreateSNESMF() 5219a6cb015SBarry Smith - error_rel - relative error (should be set to the square root of 522a4d4d686SBarry Smith the relative error in the function evaluations) 523a4d4d686SBarry Smith 52415091d37SBarry Smith Options Database Keys: 52515091d37SBarry Smith + -snes_mf_err <error_rel> - Sets error_rel 52615091d37SBarry Smith 52715091d37SBarry Smith Level: advanced 52815091d37SBarry Smith 529a4d4d686SBarry Smith Notes: 530a4d4d686SBarry Smith The default matrix-free matrix-vector product routine computes 531a4d4d686SBarry Smith .vb 532*65f2ba5bSLois Curfman McInnes F'(u)*a = [F(u+h*a) - F(u)]/h where 533a4d4d686SBarry Smith h = error_rel*u'a/||a||^2 if |u'a| > umin*||a||_{1} 534a4d4d686SBarry Smith = error_rel*umin*sign(u'a)*||a||_{1}/||a||^2 else 535a4d4d686SBarry Smith .ve 536a4d4d686SBarry Smith 537a4d4d686SBarry Smith .keywords: SNES, matrix-free, parameters 538a4d4d686SBarry Smith 5395a655dc6SBarry Smith .seealso: MatCreateSNESMF(),MatSNESMFGetH(), 5405a655dc6SBarry Smith MatSNESMFSetHHistory(), MatSNESMFResetHHistory(), 5415a655dc6SBarry Smith MatSNESMFKSPMonitor() 542a4d4d686SBarry Smith @*/ 5435a655dc6SBarry Smith int MatSNESMFSetFunctionError(Mat mat,double error) 544a4d4d686SBarry Smith { 5455a655dc6SBarry Smith MatSNESMFCtx ctx; 546a4d4d686SBarry Smith int ierr; 547a4d4d686SBarry Smith 548a4d4d686SBarry Smith PetscFunctionBegin; 549a4d4d686SBarry Smith ierr = MatShellGetContext(mat,(void **)&ctx); CHKERRQ(ierr); 550a4d4d686SBarry Smith if (ctx) { 551a4d4d686SBarry Smith if (error != PETSC_DEFAULT) ctx->error_rel = error; 552a4d4d686SBarry Smith } 553a4d4d686SBarry Smith PetscFunctionReturn(0); 554a4d4d686SBarry Smith } 555a4d4d686SBarry Smith 556a4d4d686SBarry Smith #undef __FUNC__ 5575a655dc6SBarry Smith #define __FUNC__ "MatSNESMFAddNullSpace" 558a4d4d686SBarry Smith /*@ 559*65f2ba5bSLois Curfman McInnes MatSNESMFAddNullSpace - Provides a null space that an operator is 560*65f2ba5bSLois Curfman McInnes supposed to have. Since roundoff will create a small component in 561*65f2ba5bSLois Curfman McInnes the null space, if you know the null space you may have it 562*65f2ba5bSLois Curfman McInnes automatically removed. 563a4d4d686SBarry Smith 564a4d4d686SBarry Smith Collective on Mat 565a4d4d686SBarry Smith 566a4d4d686SBarry Smith Input Parameters: 567a4d4d686SBarry Smith + J - the matrix-free matrix context 568*65f2ba5bSLois Curfman McInnes . has_cnst - PETSC_TRUE or PETSC_FALSE, indicating whether null space has constants 569a4d4d686SBarry Smith . n - number of vectors (excluding constant vector) in null space 570a4d4d686SBarry Smith - vecs - the vectors that span the null space (excluding the constant vector); 571a4d4d686SBarry Smith these vectors must be orthonormal 572a4d4d686SBarry Smith 57315091d37SBarry Smith Level: advanced 57415091d37SBarry Smith 575a4d4d686SBarry Smith .keywords: SNES, matrix-free, null space 576a4d4d686SBarry Smith 5775a655dc6SBarry Smith .seealso: MatSNESMFGetH(), MatCreateSNESMF(), 5785a655dc6SBarry Smith MatSNESMFSetHHistory(), MatSNESMFResetHHistory(), 5795a655dc6SBarry Smith MatSNESMFKSPMonitor(), MatSNESMFErrorRel() 580a4d4d686SBarry Smith 581a4d4d686SBarry Smith @*/ 5825a655dc6SBarry Smith int MatSNESMFAddNullSpace(Mat J,int has_cnst,int n,Vec *vecs) 583a4d4d686SBarry Smith { 584a4d4d686SBarry Smith int ierr; 5855a655dc6SBarry Smith MatSNESMFCtx ctx; 586a4d4d686SBarry Smith MPI_Comm comm; 587a4d4d686SBarry Smith 588a4d4d686SBarry Smith PetscFunctionBegin; 5892d0c0e3bSBarry Smith ierr = PetscObjectGetComm((PetscObject)J,&comm);CHKERRQ(ierr); 590a4d4d686SBarry Smith 591a4d4d686SBarry Smith ierr = MatShellGetContext(J,(void **)&ctx); CHKERRQ(ierr); 592a4d4d686SBarry Smith /* no context indicates that it is not the "matrix free" matrix type */ 593a4d4d686SBarry Smith if (!ctx) PetscFunctionReturn(0); 594a4d4d686SBarry Smith ierr = PCNullSpaceCreate(comm,has_cnst,n,vecs,&ctx->sp); CHKERRQ(ierr); 595a4d4d686SBarry Smith PetscFunctionReturn(0); 596a4d4d686SBarry Smith } 597a4d4d686SBarry Smith 598a4d4d686SBarry Smith #undef __FUNC__ 5995a655dc6SBarry Smith #define __FUNC__ "MatSNESMFSetHHistory" 600a4d4d686SBarry Smith /*@ 601*65f2ba5bSLois Curfman McInnes MatSNESMFSetHHistory - Sets an array to collect a history of the 602*65f2ba5bSLois Curfman McInnes differencing values (h) computed for the matrix-free product. 603a4d4d686SBarry Smith 604a4d4d686SBarry Smith Collective on Mat 605a4d4d686SBarry Smith 606a4d4d686SBarry Smith Input Parameters: 607a4d4d686SBarry Smith + J - the matrix-free matrix context 608*65f2ba5bSLois Curfman McInnes . histroy - space to hold the history 609*65f2ba5bSLois Curfman McInnes - nhistory - number of entries in history, if more entries are generated than 610*65f2ba5bSLois Curfman McInnes nhistory, then the later ones are discarded 611a4d4d686SBarry Smith 61215091d37SBarry Smith Level: advanced 61315091d37SBarry Smith 614a4d4d686SBarry Smith Notes: 615*65f2ba5bSLois Curfman McInnes Use MatSNESMFResetHHistory() to reset the history counter and collect 616*65f2ba5bSLois Curfman McInnes a new batch of differencing parameters, h. 617a4d4d686SBarry Smith 618a4d4d686SBarry Smith .keywords: SNES, matrix-free, h history, differencing history 619a4d4d686SBarry Smith 6205a655dc6SBarry Smith .seealso: MatSNESMFGetH(), MatCreateSNESMF(), 6215a655dc6SBarry Smith MatSNESMFResetHHistory(), 6225a655dc6SBarry Smith MatSNESMFKSPMonitor(), MatSNESMFSetFunctionError() 623a4d4d686SBarry Smith 624a4d4d686SBarry Smith @*/ 6255a655dc6SBarry Smith int MatSNESMFSetHHistory(Mat J,Scalar *history,int nhistory) 626a4d4d686SBarry Smith { 627a4d4d686SBarry Smith int ierr; 6285a655dc6SBarry Smith MatSNESMFCtx ctx; 629a4d4d686SBarry Smith 630a4d4d686SBarry Smith PetscFunctionBegin; 631a4d4d686SBarry Smith 632a4d4d686SBarry Smith ierr = MatShellGetContext(J,(void **)&ctx); CHKERRQ(ierr); 633a4d4d686SBarry Smith /* no context indicates that it is not the "matrix free" matrix type */ 634a4d4d686SBarry Smith if (!ctx) PetscFunctionReturn(0); 635a4d4d686SBarry Smith ctx->historyh = history; 636a4d4d686SBarry Smith ctx->maxcurrenth = nhistory; 637a4d4d686SBarry Smith ctx->currenth = 0; 638a4d4d686SBarry Smith 639a4d4d686SBarry Smith PetscFunctionReturn(0); 640a4d4d686SBarry Smith } 641a4d4d686SBarry Smith 642a4d4d686SBarry Smith #undef __FUNC__ 6435a655dc6SBarry Smith #define __FUNC__ "MatSNESMFResetHHistory" 644a4d4d686SBarry Smith /*@ 6455a655dc6SBarry Smith MatSNESMFResetHHistory - Resets the counter to zero to begin 646a4d4d686SBarry Smith collecting a new set of differencing histories. 647a4d4d686SBarry Smith 648a4d4d686SBarry Smith Collective on Mat 649a4d4d686SBarry Smith 650a4d4d686SBarry Smith Input Parameters: 651a4d4d686SBarry Smith . J - the matrix-free matrix context 652a4d4d686SBarry Smith 65315091d37SBarry Smith Level: advanced 65415091d37SBarry Smith 655a4d4d686SBarry Smith Notes: 656*65f2ba5bSLois Curfman McInnes Use MatSNESMFSetHHistory() to create the original history counter. 657a4d4d686SBarry Smith 658a4d4d686SBarry Smith .keywords: SNES, matrix-free, h history, differencing history 659a4d4d686SBarry Smith 6605a655dc6SBarry Smith .seealso: MatSNESMFGetH(), MatCreateSNESMF(), 6615a655dc6SBarry Smith MatSNESMFSetHHistory(), 6625a655dc6SBarry Smith MatSNESMFKSPMonitor(), MatSNESMFSetFunctionError() 663a4d4d686SBarry Smith 664a4d4d686SBarry Smith @*/ 6655a655dc6SBarry Smith int MatSNESMFResetHHistory(Mat J) 666a4d4d686SBarry Smith { 667a4d4d686SBarry Smith int ierr; 6685a655dc6SBarry Smith MatSNESMFCtx ctx; 669a4d4d686SBarry Smith 670a4d4d686SBarry Smith PetscFunctionBegin; 671a4d4d686SBarry Smith 672a4d4d686SBarry Smith ierr = MatShellGetContext(J,(void **)&ctx); CHKERRQ(ierr); 673a4d4d686SBarry Smith /* no context indicates that it is not the "matrix free" matrix type */ 674a4d4d686SBarry Smith if (!ctx) PetscFunctionReturn(0); 675be726c96SBarry Smith ctx->ncurrenth = 0; 676a4d4d686SBarry Smith 677a4d4d686SBarry Smith PetscFunctionReturn(0); 678a4d4d686SBarry Smith } 679a4d4d686SBarry Smith 680