1dba47a55SKris Buschelman 24b9ad928SBarry Smith /* 34b9ad928SBarry Smith Additive Multigrid V Cycle routine 44b9ad928SBarry Smith */ 5af0996ceSBarry Smith #include <petsc/private/pcmgimpl.h> 64b9ad928SBarry Smith 731567311SBarry Smith PetscErrorCode PCMGACycle_Private(PC pc,PC_MG_Levels **mglevels) 84b9ad928SBarry Smith { 96849ba73SBarry Smith PetscErrorCode ierr; 10f3fbd535SBarry Smith PetscInt i,l = mglevels[0]->levels; 114b9ad928SBarry Smith 124b9ad928SBarry Smith PetscFunctionBegin; 134b9ad928SBarry Smith /* compute RHS on each level */ 144b9ad928SBarry Smith for (i=l-1; i>0; i--) { 1563e6d426SJed Brown if (mglevels[i]->eventinterprestrict) {ierr = PetscLogEventBegin(mglevels[i]->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);} 16f3fbd535SBarry Smith ierr = MatRestrict(mglevels[i]->restrct,mglevels[i]->b,mglevels[i-1]->b);CHKERRQ(ierr); 1763e6d426SJed Brown if (mglevels[i]->eventinterprestrict) {ierr = PetscLogEventEnd(mglevels[i]->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);} 184b9ad928SBarry Smith } 19a8c7a070SBarry Smith /* solve separately on each level */ 204b9ad928SBarry Smith for (i=0; i<l; i++) { 21f3fbd535SBarry Smith ierr = VecSet(mglevels[i]->x,0.0);CHKERRQ(ierr); 2263e6d426SJed Brown if (mglevels[i]->eventsmoothsolve) {ierr = PetscLogEventBegin(mglevels[i]->eventsmoothsolve,0,0,0,0);CHKERRQ(ierr);} 23f3fbd535SBarry Smith ierr = KSPSolve(mglevels[i]->smoothd,mglevels[i]->b,mglevels[i]->x);CHKERRQ(ierr); 24*c0decd05SBarry Smith ierr = KSPCheckSolve(mglevels[i]->smoothd,pc,mglevels[i]->x);CHKERRQ(ierr); 2563e6d426SJed Brown if (mglevels[i]->eventsmoothsolve) {ierr = PetscLogEventEnd(mglevels[i]->eventsmoothsolve,0,0,0,0);CHKERRQ(ierr);} 264b9ad928SBarry Smith } 274b9ad928SBarry Smith for (i=1; i<l; i++) { 2863e6d426SJed Brown if (mglevels[i]->eventinterprestrict) {ierr = PetscLogEventBegin(mglevels[i]->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);} 29f3fbd535SBarry Smith ierr = MatInterpolateAdd(mglevels[i]->interpolate,mglevels[i-1]->x,mglevels[i]->x,mglevels[i]->x);CHKERRQ(ierr); 3063e6d426SJed Brown if (mglevels[i]->eventinterprestrict) {ierr = PetscLogEventEnd(mglevels[i]->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);} 314b9ad928SBarry Smith } 324b9ad928SBarry Smith PetscFunctionReturn(0); 334b9ad928SBarry Smith } 34