static PetscErrorCode TSRKTableauReset(TS ts)
{
TS_RK *rk = (TS_RK*)ts->data;
RKTableau tab = rk->tableau;
PetscErrorCode ierr;
PetscFunctionBegin;
if (!tab) PetscFunctionReturn(0);
ierr = PetscFree(rk->work);CHKERRQ(ierr);
ierr = VecDestroyVecs(tab->s,&rk->Y);CHKERRQ(ierr);
ierr = VecDestroyVecs(tab->s,&rk->YdotRHS);CHKERRQ(ierr);
ierr = VecDestroyVecs(tab->s*ts->numcost,&rk->VecDeltaLam);CHKERRQ(ierr);
ierr = VecDestroyVecs(tab->s*ts->numcost,&rk->VecDeltaMu);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode KSPReset_GCR( KSP ksp )
{
PetscErrorCode ierr;
KSP_GCR *ctx = (KSP_GCR*)ksp->data;
PetscFunctionBegin;
ierr = VecDestroy(&ctx->R);CHKERRQ(ierr);
ierr = VecDestroyVecs(ctx->restart,&ctx->VV);CHKERRQ(ierr);
ierr = VecDestroyVecs(ctx->restart,&ctx->SS);CHKERRQ(ierr);
if (ctx->modifypc_destroy) {
ierr = (*ctx->modifypc_destroy)(ctx->modifypc_ctx);CHKERRQ(ierr);
}
ierr = PetscFree(ctx->val);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode KSPReset_LCD(KSP ksp)
{
KSP_LCD *lcd = (KSP_LCD*)ksp->data;
PetscErrorCode ierr;
PetscFunctionBegin;
if (lcd->P) {
ierr = VecDestroyVecs(lcd->restart+1,&lcd->P);
CHKERRQ(ierr);
}
if (lcd->Q) {
ierr = VecDestroyVecs(lcd->restart+1,&lcd->Q);
CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
PetscErrorCode KSPDestroy_AGMRES(KSP ksp)
{
PetscErrorCode ierr;
KSP_AGMRES *agmres = (KSP_AGMRES*)ksp->data;
PetscFunctionBegin;
ierr = PetscFree(agmres->hh_origin);CHKERRQ(ierr);
ierr = PetscFree(agmres->nrs);CHKERRQ(ierr);
ierr = PetscFree(agmres->Qloc);CHKERRQ(ierr);
ierr = PetscFree(agmres->Rloc);CHKERRQ(ierr);
ierr = PetscFree(agmres->sgn);CHKERRQ(ierr);
ierr = PetscFree(agmres->tloc);CHKERRQ(ierr);
ierr = PetscFree(agmres->Rshift);CHKERRQ(ierr);
ierr = PetscFree(agmres->Ishift);CHKERRQ(ierr);
ierr = PetscFree(agmres->Scale);CHKERRQ(ierr);
ierr = PetscFree(agmres->wbufptr);CHKERRQ(ierr);
ierr = PetscFree(agmres->tau);CHKERRQ(ierr);
ierr = PetscFree(agmres->work);CHKERRQ(ierr);
ierr = PetscFree(agmres->temp);CHKERRQ(ierr);
ierr = PetscFree(agmres->select);CHKERRQ(ierr);
ierr = PetscFree(agmres->wr);CHKERRQ(ierr);
ierr = PetscFree(agmres->wi);CHKERRQ(ierr);
if (agmres->neig) {
ierr = VecDestroyVecs(MAXKSPSIZE,&agmres->TmpU);CHKERRQ(ierr);
ierr = PetscFree(agmres->perm);CHKERRQ(ierr);
ierr = PetscFree(agmres->MatEigL);CHKERRQ(ierr);
ierr = PetscFree(agmres->MatEigR);CHKERRQ(ierr);
ierr = PetscFree(agmres->Q);CHKERRQ(ierr);
ierr = PetscFree(agmres->Z);CHKERRQ(ierr);
ierr = PetscFree(agmres->beta);CHKERRQ(ierr);
}
ierr = KSPDestroy_DGMRES(ksp);
PetscFunctionReturn(0);
}
PetscErrorCode KSPReset_GMRES(KSP ksp)
{
KSP_GMRES *gmres = (KSP_GMRES*)ksp->data;
PetscErrorCode ierr;
PetscInt i;
PetscFunctionBegin;
/* Free the Hessenberg matrices */
ierr = PetscFree5(gmres->hh_origin,gmres->hes_origin,gmres->rs_origin,gmres->cc_origin,gmres->ss_origin);CHKERRQ(ierr);
/* free work vectors */
ierr = PetscFree(gmres->vecs);CHKERRQ(ierr);
for (i=0; i<gmres->nwork_alloc; i++) {
ierr = VecDestroyVecs(gmres->mwork_alloc[i],&gmres->user_work[i]);CHKERRQ(ierr);
}
gmres->nwork_alloc = 0;
ierr = PetscFree(gmres->user_work);CHKERRQ(ierr);
ierr = PetscFree(gmres->mwork_alloc);CHKERRQ(ierr);
ierr = PetscFree(gmres->nrs);CHKERRQ(ierr);
ierr = VecDestroy(&gmres->sol_temp);CHKERRQ(ierr);
ierr = PetscFree(gmres->Rsvd);CHKERRQ(ierr);
ierr = PetscFree(gmres->Dsvd);CHKERRQ(ierr);
ierr = PetscFree(gmres->orthogwork);CHKERRQ(ierr);
gmres->sol_temp = 0;
gmres->vv_allocated = 0;
gmres->vecs_allocated = 0;
gmres->sol_temp = 0;
PetscFunctionReturn(0);
}
PetscErrorCode MatPoissonDestroy (Mat A) {
PetscErrorCode ierr;
MatPoisson *c;
PetscFunctionBegin;
ierr = MatShellGetContext(A, (void *)&c); CHKERRQ(ierr);
for (int i = 0; i < c->rank; i++) {
ierr = MatDestroy(c->A[i]); CHKERRQ(ierr);
}
ierr = PetscFree(c->A); CHKERRQ(ierr);
ierr = PetscFree(c->dim); CHKERRQ(ierr);
ierr = VecDestroyVecs(c->u, c->rank); CHKERRQ(ierr);
ierr = VecDestroyVecs(c->v, c->rank); CHKERRQ(ierr);
ierr = PetscFree(c);
PetscFunctionReturn(0);
}
static PetscErrorCode KSPDestroy_CGLS(KSP ksp)
{
KSP_CGLS *cgls = (KSP_CGLS*)ksp->data;
PetscErrorCode ierr;
PetscFunctionBegin;
/* Free work vectors */
if (cgls->vwork_n) {
ierr = VecDestroyVecs(cgls->nwork_n,&cgls->vwork_n);CHKERRQ(ierr);
}
if (cgls->vwork_m) {
ierr = VecDestroyVecs(cgls->nwork_m,&cgls->vwork_m);CHKERRQ(ierr);
}
ierr = PetscFree(ksp->data);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode MatDestroy_LMVM(Mat B)
{
Mat_LMVM *lmvm = (Mat_LMVM*)B->data;
PetscErrorCode ierr;
PetscFunctionBegin;
if (lmvm->allocated) {
ierr = VecDestroyVecs(lmvm->m, &lmvm->S);CHKERRQ(ierr);
ierr = VecDestroyVecs(lmvm->m, &lmvm->Y);CHKERRQ(ierr);
ierr = VecDestroy(&lmvm->Xprev);CHKERRQ(ierr);
ierr = VecDestroy(&lmvm->Fprev);CHKERRQ(ierr);
}
ierr = KSPDestroy(&lmvm->J0ksp);CHKERRQ(ierr);
ierr = MatLMVMClearJ0(B);CHKERRQ(ierr);
ierr = PetscFree(B->data);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
static PetscErrorCode TSReset_RK(TS ts)
{
TS_RK *rk = (TS_RK*)ts->data;
PetscInt s;
PetscErrorCode ierr;
PetscFunctionBegin;
if (!rk->tableau) PetscFunctionReturn(0);
s = rk->tableau->s;
ierr = VecDestroyVecs(s,&rk->Y);CHKERRQ(ierr);
ierr = VecDestroyVecs(s,&rk->YdotRHS);CHKERRQ(ierr);
ierr = VecDestroyVecs(s*ts->numcost,&rk->VecDeltaLam);CHKERRQ(ierr);
ierr = VecDestroyVecs(s*ts->numcost,&rk->VecDeltaMu);CHKERRQ(ierr);
ierr = VecDestroyVecs(ts->numcost,&rk->VecSensiTemp);CHKERRQ(ierr);
ierr = PetscFree(rk->work);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
static PetscErrorCode CheckMatrices(Mat A,Mat B,Vec left,Vec right,Vec X,Vec Y,Vec X1,Vec Y1)
{
PetscErrorCode ierr;
Vec *ltmp,*rtmp;
PetscFunctionBegin;
ierr = VecDuplicateVecs(right,2,&rtmp);CHKERRQ(ierr);
ierr = VecDuplicateVecs(left,2,<mp);CHKERRQ(ierr);
ierr = MatScale(A,PETSC_PI);CHKERRQ(ierr);
ierr = MatScale(B,PETSC_PI);CHKERRQ(ierr);
ierr = MatDiagonalScale(A,left,right);CHKERRQ(ierr);
ierr = MatDiagonalScale(B,left,right);CHKERRQ(ierr);
ierr = MatMult(A,X,ltmp[0]);CHKERRQ(ierr);
ierr = MatMult(B,X,ltmp[1]);CHKERRQ(ierr);
ierr = Compare2(ltmp,"MatMult");CHKERRQ(ierr);
ierr = MatMultTranspose(A,Y,rtmp[0]);CHKERRQ(ierr);
ierr = MatMultTranspose(B,Y,rtmp[1]);CHKERRQ(ierr);
ierr = Compare2(rtmp,"MatMultTranspose");CHKERRQ(ierr);
ierr = VecCopy(Y1,ltmp[0]);CHKERRQ(ierr);
ierr = VecCopy(Y1,ltmp[1]);CHKERRQ(ierr);
ierr = MatMultAdd(A,X,ltmp[0],ltmp[0]);CHKERRQ(ierr);
ierr = MatMultAdd(B,X,ltmp[1],ltmp[1]);CHKERRQ(ierr);
ierr = Compare2(ltmp,"MatMultAdd v2==v3");CHKERRQ(ierr);
ierr = MatMultAdd(A,X,Y1,ltmp[0]);CHKERRQ(ierr);
ierr = MatMultAdd(B,X,Y1,ltmp[1]);CHKERRQ(ierr);
ierr = Compare2(ltmp,"MatMultAdd v2!=v3");CHKERRQ(ierr);
ierr = VecCopy(X1,rtmp[0]);CHKERRQ(ierr);
ierr = VecCopy(X1,rtmp[1]);CHKERRQ(ierr);
ierr = MatMultTransposeAdd(A,Y,rtmp[0],rtmp[0]);CHKERRQ(ierr);
ierr = MatMultTransposeAdd(B,Y,rtmp[1],rtmp[1]);CHKERRQ(ierr);
ierr = Compare2(rtmp,"MatMultTransposeAdd v2==v3");CHKERRQ(ierr);
ierr = MatMultTransposeAdd(A,Y,X1,rtmp[0]);CHKERRQ(ierr);
ierr = MatMultTransposeAdd(B,Y,X1,rtmp[1]);CHKERRQ(ierr);
ierr = Compare2(rtmp,"MatMultTransposeAdd v2!=v3");CHKERRQ(ierr);
ierr = VecDestroyVecs(2,<mp);CHKERRQ(ierr);
ierr = VecDestroyVecs(2,&rtmp);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
int main(int argc, char **argv)
{
PetscErrorCode ierr;
Vec *V,t;
PetscInt i,j,reps,n=15,k=6;
PetscRandom rctx;
PetscScalar *val_dot,*val_mdot,*tval_dot,*tval_mdot;
ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr;
ierr = PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL);CHKERRQ(ierr);
ierr = PetscOptionsGetInt(NULL,NULL,"-k",&k,NULL);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_WORLD,"Test with %D random vectors of length %D",k,n);CHKERRQ(ierr);
ierr = PetscPrintf(PETSC_COMM_WORLD,"\n",k,n);CHKERRQ(ierr);
ierr = PetscRandomCreate(PETSC_COMM_WORLD,&rctx);CHKERRQ(ierr);
ierr = PetscRandomSetFromOptions(rctx);CHKERRQ(ierr);
ierr = VecCreate(PETSC_COMM_WORLD,&t);CHKERRQ(ierr);
ierr = VecSetSizes(t,n,PETSC_DECIDE);CHKERRQ(ierr);
ierr = VecSetFromOptions(t);CHKERRQ(ierr);
ierr = VecDuplicateVecs(t,k,&V);CHKERRQ(ierr);
ierr = VecSetRandom(t,rctx);CHKERRQ(ierr);
ierr = PetscMalloc1(k,&val_dot);CHKERRQ(ierr);
ierr = PetscMalloc1(k,&val_mdot);CHKERRQ(ierr);
ierr = PetscMalloc1(k,&tval_dot);CHKERRQ(ierr);
ierr = PetscMalloc1(k,&tval_mdot);CHKERRQ(ierr);
for (i=0; i<k; i++) { ierr = VecSetRandom(V[i],rctx);CHKERRQ(ierr); }
for (reps=0; reps<20; reps++) {
for (i=1; i<k; i++) {
ierr = VecMDot(t,i,V,val_mdot);CHKERRQ(ierr);
ierr = VecMTDot(t,i,V,tval_mdot);CHKERRQ(ierr);
for (j=0;j<i;j++) {
ierr = VecDot(t,V[j],&val_dot[j]);CHKERRQ(ierr);
ierr = VecTDot(t,V[j],&tval_dot[j]);CHKERRQ(ierr);
}
/* Check result */
for (j=0;j<i;j++) {
if (PetscAbsScalar(val_mdot[j] - val_dot[j])/PetscAbsScalar(val_dot[j]) > 1e-5) {
ierr = PetscPrintf(PETSC_COMM_WORLD, "[TEST FAILED] i=%D, j=%D, val_mdot[j]=%g, val_dot[j]=%g\n",i,j,(double)PetscAbsScalar(val_mdot[j]), (double)PetscAbsScalar(val_dot[j]));CHKERRQ(ierr);
break;
}
if (PetscAbsScalar(tval_mdot[j] - tval_dot[j])/PetscAbsScalar(tval_dot[j]) > 1e-5) {
ierr = PetscPrintf(PETSC_COMM_WORLD, "[TEST FAILED] i=%D, j=%D, tval_mdot[j]=%g, tval_dot[j]=%g\n",i,j,(double)PetscAbsScalar(tval_mdot[j]), (double)PetscAbsScalar(tval_dot[j]));CHKERRQ(ierr);
break;
}
}
}
}
ierr = PetscPrintf(PETSC_COMM_WORLD,"Test completed successfully!\n",k,n);CHKERRQ(ierr);
ierr = PetscFree(val_dot);CHKERRQ(ierr);
ierr = PetscFree(val_mdot);CHKERRQ(ierr);
ierr = PetscFree(tval_dot);CHKERRQ(ierr);
ierr = PetscFree(tval_mdot);CHKERRQ(ierr);
ierr = VecDestroyVecs(k,&V);CHKERRQ(ierr);
ierr = VecDestroy(&t);CHKERRQ(ierr);
ierr = PetscRandomDestroy(&rctx);CHKERRQ(ierr);
ierr = PetscFinalize();
return ierr;
}
static PetscErrorCode KSPSetUp_CGLS(KSP ksp)
{
PetscErrorCode ierr;
KSP_CGLS *cgls = (KSP_CGLS*)ksp->data;
PetscFunctionBegin;
cgls->nwork_m = 2;
if (cgls->vwork_m) {
ierr = VecDestroyVecs(cgls->nwork_m,&cgls->vwork_m);CHKERRQ(ierr);
}
cgls->nwork_n = 2;
if (cgls->vwork_n) {
ierr = VecDestroyVecs(cgls->nwork_n,&cgls->vwork_n);CHKERRQ(ierr);
}
ierr = KSPCreateVecs(ksp,cgls->nwork_n,&cgls->vwork_n,cgls->nwork_m,&cgls->vwork_m);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode KSPDestroy_LSQR(KSP ksp)
{
KSP_LSQR *lsqr = (KSP_LSQR*)ksp->data;
PetscErrorCode ierr;
PetscFunctionBegin;
/* Free work vectors */
if (lsqr->vwork_n) {
ierr = VecDestroyVecs(lsqr->nwork_n,&lsqr->vwork_n);CHKERRQ(ierr);
}
if (lsqr->vwork_m) {
ierr = VecDestroyVecs(lsqr->nwork_m,&lsqr->vwork_m);CHKERRQ(ierr);
}
if (lsqr->se_flg) {
ierr = VecDestroy(&lsqr->se);CHKERRQ(ierr);
}
ierr = PetscFree(ksp->data);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode destroyPeriodicVec(PeriodicVec *user)
{
/* Function to destroy Vec's in a PeriodicVec struct */
PetscErrorCode ierr;
ierr = VecDestroyVecs(user->up,user->numPeriods);CHKERRQ(ierr);
return 0;
}
PetscErrorCode KSPReset_PIPEFGMRES(KSP ksp)
{
KSP_PIPEFGMRES *pipefgmres = (KSP_PIPEFGMRES*)ksp->data;
PetscErrorCode ierr;
PetscInt i;
PetscFunctionBegin;
ierr = PetscFree(pipefgmres->prevecs);CHKERRQ(ierr);
ierr = PetscFree(pipefgmres->zvecs);CHKERRQ(ierr);
for (i=0; i<pipefgmres->nwork_alloc; i++) {
ierr = VecDestroyVecs(pipefgmres->mwork_alloc[i],&pipefgmres->prevecs_user_work[i]);CHKERRQ(ierr);
ierr = VecDestroyVecs(pipefgmres->mwork_alloc[i],&pipefgmres->zvecs_user_work[i]);CHKERRQ(ierr);
}
ierr = PetscFree(pipefgmres->prevecs_user_work);CHKERRQ(ierr);
ierr = PetscFree(pipefgmres->zvecs_user_work);CHKERRQ(ierr);
ierr = PetscFree(pipefgmres->redux);CHKERRQ(ierr);
ierr = KSPReset_GMRES(ksp);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
static PetscErrorCode TSReset_RK(TS ts)
{
TS_RK *rk = (TS_RK*)ts->data;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = TSRKTableauReset(ts);CHKERRQ(ierr);
ierr = VecDestroy(&rk->VecCostIntegral0);CHKERRQ(ierr);
ierr = VecDestroyVecs(ts->numcost,&rk->VecSensiTemp);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
static PetscErrorCode TSReset_SSP(TS ts)
{
TS_SSP *ssp = (TS_SSP*)ts->data;
PetscErrorCode ierr;
PetscFunctionBegin;
if (ssp->work) {ierr = VecDestroyVecs(ssp->nwork,&ssp->work);CHKERRQ(ierr);}
ssp->nwork = 0;
ssp->workout = PETSC_FALSE;
PetscFunctionReturn(0);
}
/*@
SNESDefaultGetWork - Gets a number of work vectors.
Input Parameters:
. snes - the SNES context
. nw - number of work vectors to allocate
Level: developer
Notes:
Call this only if no work vectors have been allocated
@*/
PetscErrorCode SNESDefaultGetWork(SNES snes,PetscInt nw)
{
PetscErrorCode ierr;
PetscFunctionBegin;
if (snes->work) {ierr = VecDestroyVecs(snes->nwork,&snes->work);CHKERRQ(ierr);}
snes->nwork = nw;
ierr = VecDuplicateVecs(snes->vec_sol,snes->nwork,&snes->work);CHKERRQ(ierr);
ierr = PetscLogObjectParents(snes,nw,snes->work);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*------------------------------------------------------------*/
static PetscErrorCode TSReset_Theta(TS ts)
{
TS_Theta *th = (TS_Theta*)ts->data;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = VecDestroy(&th->X);CHKERRQ(ierr);
ierr = VecDestroy(&th->Xdot);CHKERRQ(ierr);
ierr = VecDestroy(&th->X0);CHKERRQ(ierr);
ierr = VecDestroy(&th->affine);CHKERRQ(ierr);
ierr = VecDestroy(&th->vec_sol_prev);CHKERRQ(ierr);
ierr = VecDestroy(&th->vec_lte_work);CHKERRQ(ierr);
ierr = VecDestroy(&th->VecCostIntegral0);CHKERRQ(ierr);
ierr = VecDestroyVecs(ts->numcost,&th->VecsDeltaLam);CHKERRQ(ierr);
ierr = VecDestroyVecs(ts->numcost,&th->VecsDeltaMu);CHKERRQ(ierr);
ierr = VecDestroyVecs(ts->numcost,&th->VecsSensiTemp);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode KSPReset_BCGSL(KSP ksp)
{
KSP_BCGSL *bcgsl = (KSP_BCGSL*)ksp->data;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = VecDestroyVecs(ksp->nwork,&ksp->work);CHKERRQ(ierr);
ierr = PetscFree5(AY0c,AYlc,AYtc,MZa,MZb);CHKERRQ(ierr);
ierr = PetscFree5(bcgsl->work,bcgsl->s,bcgsl->u,bcgsl->v,bcgsl->realwork);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode BVDestroy_Contiguous(BV bv)
{
PetscErrorCode ierr;
BV_CONTIGUOUS *ctx = (BV_CONTIGUOUS*)bv->data;
PetscFunctionBegin;
ierr = VecDestroyVecs(bv->nc+bv->m,&ctx->V);CHKERRQ(ierr);
ierr = PetscFree(ctx->array);CHKERRQ(ierr);
ierr = PetscFree(bv->data);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*
KSPSetWorkVecs - Sets a number of work vectors into a KSP object
Input Parameters:
. ksp - iterative context
. nw - number of work vectors to allocate
Developers Note: This is PETSC_EXTERN because it may be used by user written plugin KSP implementations
*/
PetscErrorCode KSPSetWorkVecs(KSP ksp,PetscInt nw)
{
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = VecDestroyVecs(ksp->nwork,&ksp->work);CHKERRQ(ierr);
ksp->nwork = nw;
ierr = KSPCreateVecs(ksp,nw,&ksp->work,0,NULL);CHKERRQ(ierr);
ierr = PetscLogObjectParents(ksp,nw,ksp->work);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
static PetscErrorCode SNESReset_QN(SNES snes)
{
PetscErrorCode ierr;
SNES_QN *qn;
PetscFunctionBegin;
if (snes->data) {
qn = (SNES_QN*)snes->data;
if (qn->U) {
ierr = VecDestroyVecs(qn->m, &qn->U);CHKERRQ(ierr);
}
if (qn->V) {
ierr = VecDestroyVecs(qn->m, &qn->V);CHKERRQ(ierr);
}
if (qn->singlereduction) {
ierr = PetscFree3(qn->dXdFmat, qn->dFtdX, qn->YtdX);CHKERRQ(ierr);
}
ierr = PetscFree4(qn->alpha,qn->beta,qn->dXtdF,qn->lambda);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
PetscErrorCode KSPFischerGuessDestroy_Method2(KSPFischerGuess_Method2 *itg)
{
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = PetscFree(itg->alpha);CHKERRQ(ierr);
ierr = VecDestroyVecs(itg->maxl,&itg->xtilde);CHKERRQ(ierr);
ierr = VecDestroy(&itg->Ax);CHKERRQ(ierr);
ierr = VecDestroy(&itg->guess);CHKERRQ(ierr);
ierr = PetscFree(itg);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*@C
DMMGSetNullSpace - Indicates the null space in the linear operator (this is needed by the linear solver)
Collective on DMMG
Input Parameter:
+ dmmg - the context
. has_cnst - is the constant vector in the null space
. n - number of null vectors (excluding the possible constant vector)
- func - a function that fills an array of vectors with the null vectors (must be orthonormal), may be PETSC_NULL
Level: advanced
.seealso DMMGCreate(), DMMGDestroy, DMMGSetDM(), DMMGSolve(), MatNullSpaceCreate(), KSPSetNullSpace(), DMMGSetMatType()
@*/
PetscErrorCode PETSCSNES_DLLEXPORT DMMGSetNullSpace(DMMG *dmmg,PetscTruth has_cnst,PetscInt n,PetscErrorCode (*func)(DMMG,Vec[]))
{
PetscErrorCode ierr;
PetscInt i,j,nlevels = dmmg[0]->nlevels;
Vec *nulls = 0;
MatNullSpace nullsp;
KSP iksp;
PC pc,ipc;
PetscTruth ismg,isred;
PetscFunctionBegin;
if (!dmmg) SETERRQ(PETSC_ERR_ARG_NULL,"Passing null as DMMG");
if (!dmmg[0]->ksp) SETERRQ(PETSC_ERR_ORDER,"Must call AFTER DMMGSetKSP() or DMMGSetSNES()");
if ((n && !func) || (!n && func)) SETERRQ(PETSC_ERR_ARG_INCOMP,"Both n and func() must be set together");
if (n < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Cannot have negative number of vectors in null space n = %D",n)
for (i=0; i<nlevels; i++) {
if (n) {
ierr = VecDuplicateVecs(dmmg[i]->b,n,&nulls);CHKERRQ(ierr);
ierr = (*func)(dmmg[i],nulls);CHKERRQ(ierr);
}
ierr = MatNullSpaceCreate(dmmg[i]->comm,has_cnst,n,nulls,&nullsp);CHKERRQ(ierr);
ierr = KSPSetNullSpace(dmmg[i]->ksp,nullsp);CHKERRQ(ierr);
for (j=i; j<nlevels; j++) {
ierr = KSPGetPC(dmmg[j]->ksp,&pc);CHKERRQ(ierr);
ierr = PetscTypeCompare((PetscObject)pc,PCMG,&ismg);CHKERRQ(ierr);
if (ismg) {
ierr = PCMGGetSmoother(pc,i,&iksp);CHKERRQ(ierr);
ierr = KSPSetNullSpace(iksp, nullsp);CHKERRQ(ierr);
}
}
ierr = MatNullSpaceDestroy(nullsp);CHKERRQ(ierr);
if (n) {
ierr = VecDestroyVecs(nulls,n);CHKERRQ(ierr);
}
}
/* make all the coarse grid solvers have LU shift since they are singular */
for (i=0; i<nlevels; i++) {
ierr = KSPGetPC(dmmg[i]->ksp,&pc);CHKERRQ(ierr);
ierr = PetscTypeCompare((PetscObject)pc,PCMG,&ismg);CHKERRQ(ierr);
if (ismg) {
ierr = PCMGGetSmoother(pc,0,&iksp);CHKERRQ(ierr);
ierr = KSPGetPC(iksp,&ipc);CHKERRQ(ierr);
ierr = PetscTypeCompare((PetscObject)ipc,PCREDUNDANT,&isred);CHKERRQ(ierr);
if (isred) {
ierr = PCRedundantGetPC(ipc,&ipc);CHKERRQ(ierr);
}
ierr = PCFactorSetShiftType(ipc,MAT_SHIFT_POSITIVE_DEFINITE);CHKERRQ(ierr);
}
}
PetscFunctionReturn(0);
}
PetscErrorCode
Metos3DTimeStepFunction(SNES snes, Vec ynBD, Vec fnBD, void *ctx)
{
Metos3D *metos3d = (Metos3D*)ctx;
// geometry
PetscInt nvec = metos3d->vectorLength;
// bgc
PetscInt ntracer = metos3d->tracerCount;
// load
PetscInt nvecloc = metos3d->vectorLengthLocal;
// parameter
PetscInt nparam = metos3d->parameterCount;
PetscReal *u0 = metos3d->u0;
// work vars
PetscInt itracer;
Vec *yin, *yinold, *yout;
PetscFunctionBegin;
// create work vectors
Metos3DUtilVecCreateAndSetValue(metos3d, ntracer, nvec, nvecloc, &yin, 0.0);
Metos3DUtilVecCreateAndSetValue(metos3d, ntracer, nvec, nvecloc, &yinold, 0.0);
Metos3DUtilVecCreateAndSetValue(metos3d, ntracer, nvec, nvecloc, &yout, 0.0);
// yin = ynBD
// yinold = ynBD
// yout = Phi(yin)
// yout = - yout + yinold;
// fnBD = yout
Metos3DUtilVecCopyDiagonalToSeparate(metos3d, ntracer, nvecloc, &ynBD, yin);
Metos3DUtilVecCopyDiagonalToSeparate(metos3d, ntracer, nvecloc, &ynBD, yinold);
Metos3DTimeStepPhi(metos3d, yin, yout, nparam, u0);
for (itracer = 0; itracer < ntracer; itracer++) VecAYPX(yout[itracer], -1.0, yinold[itracer]);
Metos3DUtilVecCopySeparateToDiagonal(metos3d, ntracer, nvecloc, yout, &fnBD);
// free work vectors
VecDestroyVecs(ntracer, &yin);
VecDestroyVecs(ntracer, &yinold);
VecDestroyVecs(ntracer, &yout);
// debug
Metos3DDebug(metos3d, kDebugLevel, "Metos3DTimeStepFunction\n");
PetscFunctionReturn(0);
}
static PetscErrorCode TSReset_RK(TS ts)
{
TS_RK *rk = (TS_RK*)ts->data;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = VecDestroy(&rk->y1);CHKERRQ(ierr);
ierr = VecDestroy(&rk->y2);CHKERRQ(ierr);
ierr = VecDestroy(&rk->tmp);CHKERRQ(ierr);
ierr = VecDestroy(&rk->tmp_y);CHKERRQ(ierr);
if (rk->k) {ierr = VecDestroyVecs(rk->s,&rk->k);CHKERRQ(ierr);}
PetscFunctionReturn(0);
}
extern PetscErrorCode MatDestroy_LMVM(Mat M)
{
MatLMVMCtx *ctx;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = MatShellGetContext(M,(void**)&ctx);CHKERRQ(ierr);
if (ctx->allocated) {
if (ctx->Xprev) {
ierr = PetscObjectDereference((PetscObject)ctx->Xprev);CHKERRQ(ierr);
}
if (ctx->Gprev) {
ierr = PetscObjectDereference((PetscObject)ctx->Gprev);CHKERRQ(ierr);
}
ierr = VecDestroyVecs(ctx->lm+1,&ctx->S);CHKERRQ(ierr);
ierr = VecDestroyVecs(ctx->lm+1,&ctx->Y);CHKERRQ(ierr);
ierr = VecDestroy(&ctx->D);CHKERRQ(ierr);
ierr = VecDestroy(&ctx->U);CHKERRQ(ierr);
ierr = VecDestroy(&ctx->V);CHKERRQ(ierr);
ierr = VecDestroy(&ctx->W);CHKERRQ(ierr);
ierr = VecDestroy(&ctx->P);CHKERRQ(ierr);
ierr = VecDestroy(&ctx->Q);CHKERRQ(ierr);
if (ctx->scale) {
ierr = VecDestroy(&ctx->scale);CHKERRQ(ierr);
}
}
ierr = PetscFree(ctx->rho);CHKERRQ(ierr);
ierr = PetscFree(ctx->beta);CHKERRQ(ierr);
ierr = PetscFree(ctx->yy_history);CHKERRQ(ierr);
ierr = PetscFree(ctx->ys_history);CHKERRQ(ierr);
ierr = PetscFree(ctx->ss_history);CHKERRQ(ierr);
ierr = PetscFree(ctx->yy_rhistory);CHKERRQ(ierr);
ierr = PetscFree(ctx->ys_rhistory);CHKERRQ(ierr);
ierr = PetscFree(ctx->ss_rhistory);CHKERRQ(ierr);
ierr = PetscFree(ctx);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
PetscErrorCode SNESDiffParameterDestroy_More(void *nePv)
{
DIFFPAR_MORE *neP = (DIFFPAR_MORE*)nePv;
PetscErrorCode ierr;
int err;
PetscFunctionBegin;
/* Destroy work vectors and close output file */
ierr = VecDestroyVecs(3,&neP->workv);CHKERRQ(ierr);
err = fclose(neP->fp);
if (err) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SYS,"fclose() failed on file");
ierr = PetscFree(neP);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
static PetscErrorCode SNESReset_Composite(SNES snes)
{
SNES_Composite *jac = (SNES_Composite*)snes->data;
PetscErrorCode ierr;
SNES_CompositeLink next = jac->head;
PetscFunctionBegin;
while (next) {
ierr = SNESReset(next->snes);CHKERRQ(ierr);
next = next->next;
}
ierr = VecDestroy(&jac->Xorig);CHKERRQ(ierr);
if (jac->Xes) {ierr = VecDestroyVecs(jac->nsnes,&jac->Xes);CHKERRQ(ierr);}
if (jac->Fes) {ierr = VecDestroyVecs(jac->nsnes,&jac->Fes);CHKERRQ(ierr);}
ierr = PetscFree(jac->fnorms);CHKERRQ(ierr);
ierr = PetscFree(jac->h);CHKERRQ(ierr);
ierr = PetscFree(jac->s);CHKERRQ(ierr);
ierr = PetscFree(jac->g);CHKERRQ(ierr);
ierr = PetscFree(jac->beta);CHKERRQ(ierr);
ierr = PetscFree(jac->work);CHKERRQ(ierr);
ierr = PetscFree(jac->rwork);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
