1dba47a55SKris Buschelman #define PETSCKSP_DLL 2dba47a55SKris Buschelman 34b9ad928SBarry Smith /* 44b9ad928SBarry Smith Additive Multigrid V Cycle routine 54b9ad928SBarry Smith */ 6*7c4f633dSBarry Smith #include "../src/ksp/pc/impls/mg/mgimpl.h" 74b9ad928SBarry Smith 84b9ad928SBarry Smith #undef __FUNCT__ 99dcbbd2bSBarry Smith #define __FUNCT__ "PCMGACycle_Private" 109dcbbd2bSBarry Smith PetscErrorCode PCMGACycle_Private(PC_MG **mg) 114b9ad928SBarry Smith { 126849ba73SBarry Smith PetscErrorCode ierr; 1379416396SBarry Smith PetscInt i,l = mg[0]->levels; 144b9ad928SBarry Smith 154b9ad928SBarry Smith PetscFunctionBegin; 164b9ad928SBarry Smith /* compute RHS on each level */ 174b9ad928SBarry Smith for (i=l-1; i>0; i--) { 1832cf1786SBarry Smith if (mg[i]->eventinterprestrict) {ierr = PetscLogEventBegin(mg[i]->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);} 194b9ad928SBarry Smith ierr = MatRestrict(mg[i]->restrct,mg[i]->b,mg[i-1]->b);CHKERRQ(ierr); 2032cf1786SBarry Smith if (mg[i]->eventinterprestrict) {ierr = PetscLogEventEnd(mg[i]->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);} 214b9ad928SBarry Smith } 22a8c7a070SBarry Smith /* solve separately on each level */ 234b9ad928SBarry Smith for (i=0; i<l; i++) { 24efb30889SBarry Smith ierr = VecSet(mg[i]->x,0.0);CHKERRQ(ierr); 2532cf1786SBarry Smith if (mg[i]->eventsmoothsolve) {ierr = PetscLogEventBegin(mg[i]->eventsmoothsolve,0,0,0,0);CHKERRQ(ierr);} 2623ce1328SBarry Smith ierr = KSPSolve(mg[i]->smoothd,mg[i]->b,mg[i]->x);CHKERRQ(ierr); 2732cf1786SBarry Smith if (mg[i]->eventsmoothsolve) {ierr = PetscLogEventEnd(mg[i]->eventsmoothsolve,0,0,0,0);CHKERRQ(ierr);} 284b9ad928SBarry Smith } 294b9ad928SBarry Smith for (i=1; i<l; i++) { 3032cf1786SBarry Smith if (mg[i]->eventinterprestrict) {ierr = PetscLogEventBegin(mg[i]->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);} 314b9ad928SBarry Smith ierr = MatInterpolateAdd(mg[i]->interpolate,mg[i-1]->x,mg[i]->x,mg[i]->x);CHKERRQ(ierr); 3232cf1786SBarry Smith if (mg[i]->eventinterprestrict) {ierr = PetscLogEventEnd(mg[i]->eventinterprestrict,0,0,0,0);CHKERRQ(ierr);} 334b9ad928SBarry Smith } 344b9ad928SBarry Smith PetscFunctionReturn(0); 354b9ad928SBarry Smith } 36