#include <../src/mat/impls/aij/seq/aij.h> #include <../src/ksp/pc/impls/bddc/bddc.h> #include <../src/ksp/pc/impls/bddc/bddcprivate.h> #include #include static PetscErrorCode MatMPIAIJRestrict(Mat,MPI_Comm,Mat*); /* if range is true, it returns B s.t. span{B} = range(A) if range is false, it returns B s.t. range(B) _|_ range(A) */ PetscErrorCode MatDenseOrthogonalRangeOrComplement(Mat A, PetscBool range, PetscInt lw, PetscScalar *work, PetscReal *rwork, Mat *B) { #if !defined(PETSC_USE_COMPLEX) PetscScalar *uwork,*data,*U, ds = 0.; PetscReal *sing; PetscBLASInt bM,bN,lwork,lierr,di = 1; PetscInt ulw,i,nr,nc,n; PetscErrorCode ierr; PetscFunctionBegin; #if defined(PETSC_MISSING_LAPACK_GESVD) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"LAPACK _GESVD not available"); #else ierr = MatGetSize(A,&nr,&nc);CHKERRQ(ierr); if (!nr || !nc) PetscFunctionReturn(0); /* workspace */ if (!work) { ulw = PetscMax(PetscMax(1,5*PetscMin(nr,nc)),3*PetscMin(nr,nc)+PetscMax(nr,nc)); ierr = PetscMalloc1(ulw,&uwork);CHKERRQ(ierr); } else { ulw = lw; uwork = work; } n = PetscMin(nr,nc); if (!rwork) { ierr = PetscMalloc1(n,&sing);CHKERRQ(ierr); } else { sing = rwork; } /* SVD */ ierr = PetscMalloc1(nr*nr,&U);CHKERRQ(ierr); ierr = PetscBLASIntCast(nr,&bM);CHKERRQ(ierr); ierr = PetscBLASIntCast(nc,&bN);CHKERRQ(ierr); ierr = PetscBLASIntCast(ulw,&lwork);CHKERRQ(ierr); ierr = MatDenseGetArray(A,&data);CHKERRQ(ierr); ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); PetscStackCallBLAS("LAPACKgesvd",LAPACKgesvd_("A","N",&bM,&bN,data,&bM,sing,U,&bM,&ds,&di,uwork,&lwork,&lierr)); ierr = PetscFPTrapPop();CHKERRQ(ierr); if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in GESVD Lapack routine %d",(int)lierr); ierr = MatDenseRestoreArray(A,&data);CHKERRQ(ierr); for (i=0;idata; Mat_IS *matis = (Mat_IS*)pc->pmat->data; Mat G,T,conn,lG,lGt,lGis,lGall,lGe,lGinit; Vec tvec; PetscSF sfv; ISLocalToGlobalMapping el2g,vl2g,fl2g,al2g; MPI_Comm comm; IS lned,primals,allprimals,nedfieldlocal; IS *eedges,*extrows,*extcols,*alleedges; PetscBT btv,bte,btvc,btb,btbd,btvcand,btvi,btee,bter; PetscScalar *vals,*work; PetscReal *rwork; const PetscInt *idxs,*ii,*jj,*iit,*jjt; PetscInt ne,nv,Lv,order,n,field; PetscInt n_neigh,*neigh,*n_shared,**shared; PetscInt i,j,extmem,cum,maxsize,nee; PetscInt *extrow,*extrowcum,*marks,*vmarks,*gidxs; PetscInt *sfvleaves,*sfvroots; PetscInt *corners,*cedges; PetscInt *ecount,**eneighs,*vcount,**vneighs; #if defined(PETSC_USE_DEBUG) PetscInt *emarks; #endif PetscBool print,eerr,done,lrc[2],conforming,global,singular,setprimal; PetscErrorCode ierr; PetscFunctionBegin; /* If the discrete gradient is defined for a subset of dofs and global is true, it assumes G is given in global ordering for all the dofs. Otherwise, the ordering is global for the Nedelec field */ order = pcbddc->nedorder; conforming = pcbddc->conforming; field = pcbddc->nedfield; global = pcbddc->nedglobal; setprimal = PETSC_FALSE; print = PETSC_FALSE; singular = PETSC_FALSE; /* Command line customization */ ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)pc),((PetscObject)pc)->prefix,"BDDC Nedelec options","PC");CHKERRQ(ierr); ierr = PetscOptionsBool("-pc_bddc_nedelec_field_primal","All edge dofs set as primals: Toselli's algorithm C",NULL,setprimal,&setprimal,NULL);CHKERRQ(ierr); ierr = PetscOptionsBool("-pc_bddc_nedelec_singular","Infer nullspace from discrete gradient",NULL,singular,&singular,NULL);CHKERRQ(ierr); ierr = PetscOptionsInt ("-pc_bddc_nedelec_order","Test variable order code (to be removed)",NULL,order,&order,NULL);CHKERRQ(ierr); /* print debug info TODO: to be removed */ ierr = PetscOptionsBool("-pc_bddc_nedelec_print","Print debug info",NULL,print,&print,NULL);CHKERRQ(ierr); ierr = PetscOptionsEnd();CHKERRQ(ierr); /* Return if there are no edges in the decomposition and the problem is not singular */ ierr = MatGetLocalToGlobalMapping(pc->pmat,&al2g,NULL);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingGetSize(al2g,&n);CHKERRQ(ierr); ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); if (!singular) { ierr = VecGetArrayRead(matis->counter,(const PetscScalar**)&vals);CHKERRQ(ierr); lrc[0] = PETSC_FALSE; for (i=0;i 2.) { lrc[0] = PETSC_TRUE; break; } } ierr = VecRestoreArrayRead(matis->counter,(const PetscScalar**)&vals);CHKERRQ(ierr); ierr = MPIU_Allreduce(&lrc[0],&lrc[1],1,MPIU_BOOL,MPI_LOR,comm);CHKERRQ(ierr); if (!lrc[1]) PetscFunctionReturn(0); } /* Get Nedelec field */ ierr = MatISSetUpSF(pc->pmat);CHKERRQ(ierr); if (pcbddc->n_ISForDofsLocal && field >= pcbddc->n_ISForDofsLocal) SETERRQ2(comm,PETSC_ERR_USER,"Invalid field for Nedelec %d: number of fields is %d",field,pcbddc->n_ISForDofsLocal); if (pcbddc->n_ISForDofsLocal && field >= 0) { ierr = PetscObjectReference((PetscObject)pcbddc->ISForDofsLocal[field]);CHKERRQ(ierr); nedfieldlocal = pcbddc->ISForDofsLocal[field]; ierr = ISGetLocalSize(nedfieldlocal,&ne);CHKERRQ(ierr); } else if (!pcbddc->n_ISForDofsLocal && field != PETSC_DECIDE) { ne = n; nedfieldlocal = NULL; global = PETSC_TRUE; } else if (field == PETSC_DECIDE) { PetscInt rst,ren,*idx; ierr = PetscMemzero(matis->sf_leafdata,n*sizeof(PetscInt));CHKERRQ(ierr); ierr = PetscMemzero(matis->sf_rootdata,pc->pmat->rmap->n*sizeof(PetscInt));CHKERRQ(ierr); ierr = MatGetOwnershipRange(pcbddc->discretegradient,&rst,&ren);CHKERRQ(ierr); for (i=rst;idiscretegradient,i,&nc,NULL,NULL);CHKERRQ(ierr); if (nc > 1) matis->sf_rootdata[i-rst] = 1; ierr = MatRestoreRow(pcbddc->discretegradient,i,&nc,NULL,NULL);CHKERRQ(ierr); } ierr = PetscSFBcastBegin(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata);CHKERRQ(ierr); ierr = PetscSFBcastEnd(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata);CHKERRQ(ierr); ierr = PetscMalloc1(n,&idx);CHKERRQ(ierr); for (i=0,ne=0;isf_leafdata[i]) idx[ne++] = i; ierr = ISCreateGeneral(comm,ne,idx,PETSC_OWN_POINTER,&nedfieldlocal);CHKERRQ(ierr); } else { SETERRQ(comm,PETSC_ERR_USER,"When multiple fields are present, the Nedelec field has to be specified"); } /* Sanity checks */ if (!order && !conforming) SETERRQ(comm,PETSC_ERR_SUP,"Variable order and non-conforming spaces are not supported at the same time"); if (pcbddc->user_ChangeOfBasisMatrix) SETERRQ(comm,PETSC_ERR_SUP,"Cannot generate Nedelec support with user defined change of basis"); if (order && ne%order) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"The number of local edge dofs %d it's not a multiple of the order %d",ne,order); /* Just set primal dofs and return */ if (setprimal) { IS enedfieldlocal; PetscInt *eidxs; ierr = PetscMalloc1(ne,&eidxs);CHKERRQ(ierr); ierr = VecGetArrayRead(matis->counter,(const PetscScalar**)&vals);CHKERRQ(ierr); if (nedfieldlocal) { ierr = ISGetIndices(nedfieldlocal,&idxs);CHKERRQ(ierr); for (i=0,cum=0;i 2.) { eidxs[cum++] = idxs[i]; } } ierr = ISRestoreIndices(nedfieldlocal,&idxs);CHKERRQ(ierr); } else { for (i=0,cum=0;i 2.) { eidxs[cum++] = i; } } } ierr = VecRestoreArrayRead(matis->counter,(const PetscScalar**)&vals);CHKERRQ(ierr); ierr = ISCreateGeneral(comm,cum,eidxs,PETSC_COPY_VALUES,&enedfieldlocal);CHKERRQ(ierr); ierr = PCBDDCSetPrimalVerticesLocalIS(pc,enedfieldlocal);CHKERRQ(ierr); ierr = PetscFree(eidxs);CHKERRQ(ierr); ierr = ISDestroy(&nedfieldlocal);CHKERRQ(ierr); ierr = ISDestroy(&enedfieldlocal);CHKERRQ(ierr); PetscFunctionReturn(0); } /* Compute some l2g maps */ if (nedfieldlocal) { IS is; /* need to map from the local Nedelec field to local numbering */ ierr = ISLocalToGlobalMappingCreateIS(nedfieldlocal,&fl2g);CHKERRQ(ierr); /* need to map from the local Nedelec field to global numbering for the whole dofs*/ ierr = ISLocalToGlobalMappingApplyIS(al2g,nedfieldlocal,&is);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingCreateIS(is,&al2g);CHKERRQ(ierr); /* need to map from the local Nedelec field to global numbering (for Nedelec only) */ if (global) { ierr = PetscObjectReference((PetscObject)al2g);CHKERRQ(ierr); el2g = al2g; } else { IS gis; ierr = ISRenumber(is,NULL,NULL,&gis);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingCreateIS(gis,&el2g);CHKERRQ(ierr); ierr = ISDestroy(&gis);CHKERRQ(ierr); } ierr = ISDestroy(&is);CHKERRQ(ierr); } else { /* restore default */ pcbddc->nedfield = -1; /* one ref for the destruction of al2g, one for el2g */ ierr = PetscObjectReference((PetscObject)al2g);CHKERRQ(ierr); ierr = PetscObjectReference((PetscObject)al2g);CHKERRQ(ierr); el2g = al2g; fl2g = NULL; } /* Start communication to drop connections for interior edges (for cc analysis only) */ ierr = PetscMemzero(matis->sf_leafdata,n*sizeof(PetscInt));CHKERRQ(ierr); ierr = PetscMemzero(matis->sf_rootdata,pc->pmat->rmap->n*sizeof(PetscInt));CHKERRQ(ierr); if (nedfieldlocal) { ierr = ISGetIndices(nedfieldlocal,&idxs);CHKERRQ(ierr); for (i=0;isf_leafdata[idxs[i]] = 1; ierr = ISRestoreIndices(nedfieldlocal,&idxs);CHKERRQ(ierr); } else { for (i=0;isf_leafdata[i] = 1; } ierr = PetscSFReduceBegin(matis->sf,MPIU_INT,matis->sf_leafdata,matis->sf_rootdata,MPI_SUM);CHKERRQ(ierr); ierr = PetscSFReduceEnd(matis->sf,MPIU_INT,matis->sf_leafdata,matis->sf_rootdata,MPI_SUM);CHKERRQ(ierr); if (!singular) { /* drop connections with interior edges to avoid unneeded communications and memory movements */ ierr = MatDuplicate(pcbddc->discretegradient,MAT_COPY_VALUES,&G);CHKERRQ(ierr); ierr = MatSetOption(G,MAT_KEEP_NONZERO_PATTERN,PETSC_FALSE);CHKERRQ(ierr); if (global) { PetscInt rst; ierr = MatGetOwnershipRange(G,&rst,NULL);CHKERRQ(ierr); for (i=0,cum=0;ipmat->rmap->n;i++) { if (matis->sf_rootdata[i] < 2) { matis->sf_rootdata[cum++] = i + rst; } } ierr = MatSetOption(G,MAT_NO_OFF_PROC_ZERO_ROWS,PETSC_TRUE);CHKERRQ(ierr); ierr = MatZeroRows(G,cum,matis->sf_rootdata,0.,NULL,NULL);CHKERRQ(ierr); } else { PetscInt *tbz; ierr = PetscMalloc1(ne,&tbz);CHKERRQ(ierr); ierr = PetscSFBcastBegin(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata);CHKERRQ(ierr); ierr = PetscSFBcastEnd(matis->sf,MPIU_INT,matis->sf_rootdata,matis->sf_leafdata);CHKERRQ(ierr); ierr = ISGetIndices(nedfieldlocal,&idxs);CHKERRQ(ierr); for (i=0,cum=0;isf_leafdata[idxs[i]] == 1) tbz[cum++] = i; ierr = ISRestoreIndices(nedfieldlocal,&idxs);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingApply(el2g,cum,tbz,tbz);CHKERRQ(ierr); ierr = MatZeroRows(G,cum,tbz,0.,NULL,NULL);CHKERRQ(ierr); ierr = PetscFree(tbz);CHKERRQ(ierr); } } else { /* we need the entire G to infer the nullspace */ ierr = PetscObjectReference((PetscObject)pcbddc->discretegradient);CHKERRQ(ierr); G = pcbddc->discretegradient; } /* Extract subdomain relevant rows of G */ ierr = ISLocalToGlobalMappingGetIndices(el2g,&idxs);CHKERRQ(ierr); ierr = ISCreateGeneral(comm,ne,idxs,PETSC_USE_POINTER,&lned);CHKERRQ(ierr); ierr = MatCreateSubMatrix(G,lned,NULL,MAT_INITIAL_MATRIX,&lGall);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingRestoreIndices(el2g,&idxs);CHKERRQ(ierr); ierr = ISDestroy(&lned);CHKERRQ(ierr); ierr = MatConvert(lGall,MATIS,MAT_INITIAL_MATRIX,&lGis);CHKERRQ(ierr); ierr = MatDestroy(&lGall);CHKERRQ(ierr); ierr = MatISGetLocalMat(lGis,&lG);CHKERRQ(ierr); /* SF for nodal dofs communications */ ierr = MatGetLocalSize(G,NULL,&Lv);CHKERRQ(ierr); ierr = MatGetLocalToGlobalMapping(lGis,NULL,&vl2g);CHKERRQ(ierr); ierr = PetscObjectReference((PetscObject)vl2g);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingGetSize(vl2g,&nv);CHKERRQ(ierr); ierr = PetscSFCreate(comm,&sfv);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingGetIndices(vl2g,&idxs);CHKERRQ(ierr); ierr = PetscSFSetGraphLayout(sfv,lGis->cmap,nv,NULL,PETSC_OWN_POINTER,idxs);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingRestoreIndices(vl2g,&idxs);CHKERRQ(ierr); i = singular ? 2 : 1; ierr = PetscMalloc2(i*nv,&sfvleaves,i*Lv,&sfvroots);CHKERRQ(ierr); /* Destroy temporary G created in MATIS format and modified G */ ierr = PetscObjectReference((PetscObject)lG);CHKERRQ(ierr); ierr = MatDestroy(&lGis);CHKERRQ(ierr); ierr = MatDestroy(&G);CHKERRQ(ierr); if (print) { ierr = PetscObjectSetName((PetscObject)lG,"initial_lG");CHKERRQ(ierr); ierr = MatView(lG,NULL);CHKERRQ(ierr); } /* Save lG for values insertion in change of basis */ ierr = MatDuplicate(lG,MAT_COPY_VALUES,&lGinit);CHKERRQ(ierr); /* Analyze the edge-nodes connections (duplicate lG) */ ierr = MatDuplicate(lG,MAT_COPY_VALUES,&lGe);CHKERRQ(ierr); ierr = MatSetOption(lGe,MAT_KEEP_NONZERO_PATTERN,PETSC_FALSE);CHKERRQ(ierr); ierr = PetscBTCreate(nv,&btv);CHKERRQ(ierr); ierr = PetscBTCreate(ne,&bte);CHKERRQ(ierr); ierr = PetscBTCreate(ne,&btb);CHKERRQ(ierr); ierr = PetscBTCreate(ne,&btbd);CHKERRQ(ierr); ierr = PetscBTCreate(nv,&btvcand);CHKERRQ(ierr); /* need to import the boundary specification to ensure the proper detection of coarse edges' endpoints */ if (pcbddc->DirichletBoundariesLocal) { IS is; if (fl2g) { ierr = ISGlobalToLocalMappingApplyIS(fl2g,IS_GTOLM_MASK,pcbddc->DirichletBoundariesLocal,&is);CHKERRQ(ierr); } else { is = pcbddc->DirichletBoundariesLocal; } ierr = ISGetLocalSize(is,&cum);CHKERRQ(ierr); ierr = ISGetIndices(is,&idxs);CHKERRQ(ierr); for (i=0;i= 0) { ierr = PetscBTSet(btb,idxs[i]);CHKERRQ(ierr); ierr = PetscBTSet(btbd,idxs[i]);CHKERRQ(ierr); } } ierr = ISRestoreIndices(is,&idxs);CHKERRQ(ierr); if (fl2g) { ierr = ISDestroy(&is);CHKERRQ(ierr); } } if (pcbddc->NeumannBoundariesLocal) { IS is; if (fl2g) { ierr = ISGlobalToLocalMappingApplyIS(fl2g,IS_GTOLM_MASK,pcbddc->NeumannBoundariesLocal,&is);CHKERRQ(ierr); } else { is = pcbddc->NeumannBoundariesLocal; } ierr = ISGetLocalSize(is,&cum);CHKERRQ(ierr); ierr = ISGetIndices(is,&idxs);CHKERRQ(ierr); for (i=0;i= 0) { ierr = PetscBTSet(btb,idxs[i]);CHKERRQ(ierr); } } ierr = ISRestoreIndices(is,&idxs);CHKERRQ(ierr); if (fl2g) { ierr = ISDestroy(&is);CHKERRQ(ierr); } } /* Count neighs per dof */ ierr = PetscCalloc1(ne,&ecount);CHKERRQ(ierr); ierr = PetscMalloc1(ne,&eneighs);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingGetInfo(el2g,&n_neigh,&neigh,&n_shared,&shared);CHKERRQ(ierr); for (i=1,cum=0;i 1 && !PetscBTLookup(btbd,i)) || (ecount[i] == 1 && PetscBTLookup(btb,i))) { ierr = PetscBTSet(btee,i);CHKERRQ(ierr); } } ierr = PetscMalloc1(ne,&marks);CHKERRQ(ierr); if (!conforming) { ierr = MatTranspose(lGe,MAT_INITIAL_MATRIX,&lGt);CHKERRQ(ierr); ierr = MatGetRowIJ(lGt,0,PETSC_FALSE,PETSC_FALSE,&i,&iit,&jjt,&done);CHKERRQ(ierr); } ierr = MatGetRowIJ(lGe,0,PETSC_FALSE,PETSC_FALSE,&i,&ii,&jj,&done);CHKERRQ(ierr); ierr = MatSeqAIJGetArray(lGe,&vals);CHKERRQ(ierr); cum = 0; for (i=0;i order) ends++; else if (nconn == order) ints++; else undef++; } if (undef || ends > 2 || ints != order -1) { marks[cum++] = i; ierr = PetscBTSet(bte,i);CHKERRQ(ierr); for (j=ii[i];j PETSC_SQRT_MACHINE_EPSILON) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected value for vorder: %g (%d)",vorder,test); ord = 1; } #if defined(PETSC_USE_DEBUG) if (test%ord) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected number of edge dofs %d connected with nodal dof %d with order %d",test,i,ord); #endif for (j=ii[i];j= 3*ord || bdir) { /* splitpoints */ if (print) PetscPrintf(PETSC_COMM_SELF,"SPLITPOINT %d (%d %d %d)\n",i,!sneighs,test >= 3*ord,bdir); ierr = PetscBTSet(btv,i);CHKERRQ(ierr); } else if (test == ord) { if (order == 1 || (!order && ii[i+1]-ii[i] == 1)) { if (print) PetscPrintf(PETSC_COMM_SELF,"ENDPOINT %d\n",i); ierr = PetscBTSet(btv,i);CHKERRQ(ierr); } else { if (print) PetscPrintf(PETSC_COMM_SELF,"CORNER CANDIDATE %d\n",i); ierr = PetscBTSet(btvcand,i);CHKERRQ(ierr); } } } ierr = PetscFree(ecount);CHKERRQ(ierr); ierr = PetscFree(vcount);CHKERRQ(ierr); if (ne) { ierr = PetscFree(eneighs[0]);CHKERRQ(ierr); } if (nv) { ierr = PetscFree(vneighs[0]);CHKERRQ(ierr); } ierr = PetscFree(eneighs);CHKERRQ(ierr); ierr = PetscFree(vneighs);CHKERRQ(ierr); ierr = PetscBTDestroy(&btbd);CHKERRQ(ierr); /* a candidate is valid if it is connected to another candidate via a non-primal edge dof */ if (order != 1) { if (print) PetscPrintf(PETSC_COMM_SELF,"INSPECTING CANDIDATES\n"); ierr = MatGetRowIJ(lGe,0,PETSC_FALSE,PETSC_FALSE,&i,&iit,&jjt,&done);CHKERRQ(ierr); for (i=0;imat_graph->nvtxs_csr) { /* the user has passed in a CSR graph */ if (pcbddc->mat_graph->nvtxs_csr != n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"Invalid size of CSR graph %d. Should be %d\n",pcbddc->mat_graph->nvtxs_csr,n); iiu = pcbddc->mat_graph->xadj; jju = pcbddc->mat_graph->adjncy; } else if (pcbddc->use_local_adj) { rest = PETSC_TRUE; ierr = MatGetRowIJ(matis->A,0,PETSC_TRUE,PETSC_FALSE,&i,(const PetscInt**)&iiu,(const PetscInt**)&jju,&done);CHKERRQ(ierr); } else { free = PETSC_TRUE; ierr = PetscMalloc2(n+1,&iiu,n,&jju);CHKERRQ(ierr); iiu[0] = 0; for (i=0;iA,0,PETSC_TRUE,PETSC_FALSE,&i,(const PetscInt**)&iiu,(const PetscInt**)&jju,&done);CHKERRQ(ierr); } if (free) { ierr = PetscFree2(iiu,jju);CHKERRQ(ierr); } ierr = PetscBTDestroy(&btf);CHKERRQ(ierr); } else { ierr = PCBDDCSetLocalAdjacencyGraph(pc,n,ii,jj,PETSC_USE_POINTER);CHKERRQ(ierr); } /* Analyze interface for edge dofs */ ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); pcbddc->mat_graph->twodim = PETSC_FALSE; /* Get coarse edges in the edge space */ ierr = PCBDDCGraphGetCandidatesIS(pcbddc->mat_graph,NULL,NULL,&nee,&alleedges,&allprimals);CHKERRQ(ierr); ierr = MatRestoreRowIJ(conn,0,PETSC_FALSE,PETSC_FALSE,&i,&ii,&jj,&done);CHKERRQ(ierr); if (fl2g) { ierr = ISGlobalToLocalMappingApplyIS(fl2g,IS_GTOLM_DROP,allprimals,&primals);CHKERRQ(ierr); ierr = PetscMalloc1(nee,&eedges);CHKERRQ(ierr); for (i=0;i EXTCOL FIRST PASS"); */ ierr = MPIU_Allreduce(&eerr,&done,1,MPIU_BOOL,MPI_LOR,comm);CHKERRQ(ierr); if (done) { PetscInt *newprimals; ierr = PetscMalloc1(ne,&newprimals);CHKERRQ(ierr); ierr = ISGetLocalSize(primals,&cum);CHKERRQ(ierr); ierr = ISGetIndices(primals,&idxs);CHKERRQ(ierr); ierr = PetscMemcpy(newprimals,idxs,cum*sizeof(PetscInt));CHKERRQ(ierr); ierr = ISRestoreIndices(primals,&idxs);CHKERRQ(ierr); ierr = MatGetRowIJ(lGt,0,PETSC_FALSE,PETSC_FALSE,&i,&iit,&jjt,&done);CHKERRQ(ierr); if (print) PetscPrintf(PETSC_COMM_SELF,"DOING SECOND PASS (eerr %d)\n",eerr); for (i=0;i 1) { if (print) PetscPrintf(PETSC_COMM_SELF," Edge dof set to primal %d\n",idxs[j]); newprimals[cum++] = idxs[j]; } } ierr = PetscFree(tmarks);CHKERRQ(ierr); } ierr = ISRestoreIndices(eedges[i],&idxs);CHKERRQ(ierr); } ierr = ISDestroy(&extcols[i]);CHKERRQ(ierr); } ierr = PetscFree(extcols);CHKERRQ(ierr); ierr = MatRestoreRowIJ(lGt,0,PETSC_FALSE,PETSC_FALSE,&i,&iit,&jjt,&done);CHKERRQ(ierr); ierr = PetscSortRemoveDupsInt(&cum,newprimals);CHKERRQ(ierr); if (fl2g) { ierr = ISLocalToGlobalMappingApply(fl2g,cum,newprimals,newprimals);CHKERRQ(ierr); ierr = ISDestroy(&primals);CHKERRQ(ierr); for (i=0;imat_graph,NULL,NULL,&nee,&alleedges,&allprimals);CHKERRQ(ierr); ierr = ISCreateGeneral(comm,cum,newprimals,PETSC_COPY_VALUES,&primals);CHKERRQ(ierr); ierr = PetscFree(newprimals);CHKERRQ(ierr); ierr = PCBDDCSetPrimalVerticesLocalIS(pc,primals);CHKERRQ(ierr); ierr = ISDestroy(&primals);CHKERRQ(ierr); ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); pcbddc->mat_graph->twodim = PETSC_FALSE; ierr = PCBDDCGraphGetCandidatesIS(pcbddc->mat_graph,NULL,NULL,&nee,&alleedges,&allprimals);CHKERRQ(ierr); if (fl2g) { ierr = ISGlobalToLocalMappingApplyIS(fl2g,IS_GTOLM_DROP,allprimals,&primals);CHKERRQ(ierr); ierr = PetscMalloc1(nee,&eedges);CHKERRQ(ierr); for (i=0;imat_graph,5,PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr); } /* an error should not occur at this point */ if (eerr) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected SIZE OF EDGE > EXTCOL SECOND PASS"); /* Check the number of endpoints */ ierr = MatGetRowIJ(lG,0,PETSC_FALSE,PETSC_FALSE,&i,&ii,&jj,&done);CHKERRQ(ierr); ierr = PetscMalloc1(2*nee,&corners);CHKERRQ(ierr); ierr = PetscMalloc1(nee,&cedges);CHKERRQ(ierr); for (i=0;i gc[1]) { PetscInt swap = corners[2*i]; corners[2*i] = corners[2*i+1]; corners[2*i+1] = swap; } cedges[i] = idxs[size-1]; ierr = ISRestoreIndices(eedges[i],&idxs);CHKERRQ(ierr); if (print) PetscPrintf(PETSC_COMM_SELF,"EDGE %d: ce %d, corners (%d,%d)\n",i,cedges[i],corners[2*i],corners[2*i+1]); } ierr = MatRestoreRowIJ(lG,0,PETSC_FALSE,PETSC_FALSE,&i,&ii,&jj,&done);CHKERRQ(ierr); ierr = PetscBTDestroy(&btvc);CHKERRQ(ierr); #if defined(PETSC_USE_DEBUG) /* Inspects columns of lG (rows of lGt) and make sure the change of basis will not interfere with neighbouring coarse edges */ ierr = PetscMalloc1(nee+1,&emarks);CHKERRQ(ierr); ierr = MatGetRowIJ(lGt,0,PETSC_FALSE,PETSC_FALSE,&i,&ii,&jj,&done);CHKERRQ(ierr); for (i=0;i extmem) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Not enough memory allocated %d > %d",extrowcum[mark] + size,extmem); ierr = PetscMemcpy(extrow+start,jj+ii[i],size*sizeof(PetscInt));CHKERRQ(ierr); extrowcum[mark] += size; } ierr = MatRestoreRowIJ(lGt,0,PETSC_FALSE,PETSC_FALSE,&i,&ii,&jj,&done);CHKERRQ(ierr); ierr = MatDestroy(&lGt);CHKERRQ(ierr); ierr = PetscFree(marks);CHKERRQ(ierr); /* Compress extrows */ cum = 0; for (i=0;ipmat->rmap->n,pc->pmat->rmap->n, pc->pmat->rmap->N,pc->pmat->rmap->N);CHKERRQ(ierr); ierr = MatSetType(T,MATAIJ);CHKERRQ(ierr); ierr = MatSeqAIJSetPreallocation(T,10,NULL);CHKERRQ(ierr); ierr = MatMPIAIJSetPreallocation(T,10,NULL,10,NULL);CHKERRQ(ierr); ierr = MatSetLocalToGlobalMapping(T,al2g,al2g);CHKERRQ(ierr); ierr = MatSetOption(T,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_FALSE);CHKERRQ(ierr); ierr = MatSetOption(T,MAT_ROW_ORIENTED,PETSC_FALSE);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingDestroy(&al2g);CHKERRQ(ierr); /* Defaults to identity */ ierr = MatCreateVecs(pc->pmat,&tvec,NULL);CHKERRQ(ierr); ierr = VecSet(tvec,1.0);CHKERRQ(ierr); ierr = MatDiagonalSet(T,tvec,INSERT_VALUES);CHKERRQ(ierr); ierr = VecDestroy(&tvec);CHKERRQ(ierr); /* Create discrete gradient for the coarser level if needed */ ierr = MatDestroy(&pcbddc->nedcG);CHKERRQ(ierr); ierr = ISDestroy(&pcbddc->nedclocal);CHKERRQ(ierr); if (pcbddc->current_level < pcbddc->max_levels) { ISLocalToGlobalMapping cel2g,cvl2g; IS wis,gwis; PetscInt cnv,cne; ierr = ISCreateGeneral(comm,nee,cedges,PETSC_COPY_VALUES,&wis);CHKERRQ(ierr); if (fl2g) { ierr = ISLocalToGlobalMappingApplyIS(fl2g,wis,&pcbddc->nedclocal);CHKERRQ(ierr); } else { ierr = PetscObjectReference((PetscObject)wis);CHKERRQ(ierr); pcbddc->nedclocal = wis; } ierr = ISLocalToGlobalMappingApplyIS(el2g,wis,&gwis);CHKERRQ(ierr); ierr = ISDestroy(&wis);CHKERRQ(ierr); ierr = ISRenumber(gwis,NULL,&cne,&wis);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingCreateIS(wis,&cel2g);CHKERRQ(ierr); ierr = ISDestroy(&wis);CHKERRQ(ierr); ierr = ISDestroy(&gwis);CHKERRQ(ierr); ierr = ISCreateGeneral(comm,2*nee,corners,PETSC_USE_POINTER,&wis);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingApplyIS(vl2g,wis,&gwis);CHKERRQ(ierr); ierr = ISDestroy(&wis);CHKERRQ(ierr); ierr = ISRenumber(gwis,NULL,&cnv,&wis);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingCreateIS(wis,&cvl2g);CHKERRQ(ierr); ierr = ISDestroy(&wis);CHKERRQ(ierr); ierr = ISDestroy(&gwis);CHKERRQ(ierr); ierr = MatCreate(comm,&pcbddc->nedcG);CHKERRQ(ierr); ierr = MatSetSizes(pcbddc->nedcG,PETSC_DECIDE,PETSC_DECIDE,cne,cnv);CHKERRQ(ierr); ierr = MatSetType(pcbddc->nedcG,MATAIJ);CHKERRQ(ierr); ierr = MatSeqAIJSetPreallocation(pcbddc->nedcG,2,NULL);CHKERRQ(ierr); ierr = MatMPIAIJSetPreallocation(pcbddc->nedcG,2,NULL,2,NULL);CHKERRQ(ierr); ierr = MatSetLocalToGlobalMapping(pcbddc->nedcG,cel2g,cvl2g);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingDestroy(&cel2g);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingDestroy(&cvl2g);CHKERRQ(ierr); } ierr = ISLocalToGlobalMappingDestroy(&vl2g);CHKERRQ(ierr); #if defined(PRINT_GDET) inc = 0; lev = pcbddc->current_level; #endif /* Insert values in the change of basis matrix */ for (i=0;inedcG) { ierr = ISCreateGeneral(PETSC_COMM_SELF,2,corners+2*i,PETSC_USE_POINTER,&cornersis);CHKERRQ(ierr); } ierr = PCBDDCComputeNedelecChangeEdge(lG,eedges[i],extrows[i],extcols[i],cornersis,&Gins,&GKins,cvals,work,rwork);CHKERRQ(ierr); if (Gins && GKins) { PetscScalar *data; const PetscInt *rows,*cols; PetscInt nrh,nch,nrc,ncc; ierr = ISGetIndices(eedges[i],&cols);CHKERRQ(ierr); /* H1 */ ierr = ISGetIndices(extrows[i],&rows);CHKERRQ(ierr); ierr = MatGetSize(Gins,&nrh,&nch);CHKERRQ(ierr); ierr = MatDenseGetArray(Gins,&data);CHKERRQ(ierr); ierr = MatSetValuesLocal(T,nrh,rows,nch,cols,data,INSERT_VALUES);CHKERRQ(ierr); ierr = MatDenseRestoreArray(Gins,&data);CHKERRQ(ierr); ierr = ISRestoreIndices(extrows[i],&rows);CHKERRQ(ierr); /* complement */ ierr = MatGetSize(GKins,&nrc,&ncc);CHKERRQ(ierr); if (!ncc) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Constant function has not been generated for coarse edge %d",i); if (ncc + nch != nrc) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_PLIB,"The sum of the number of columns of GKins %d and Gins %d does not match %d for coarse edge %d",ncc,nch,nrc,i); if (ncc != 1 && pcbddc->nedcG) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot generate the coarse discrete gradient for coarse edge %d with ncc %d",i,ncc); ierr = MatDenseGetArray(GKins,&data);CHKERRQ(ierr); ierr = MatSetValuesLocal(T,nrc,cols,ncc,cols+nch,data,INSERT_VALUES);CHKERRQ(ierr); ierr = MatDenseRestoreArray(GKins,&data);CHKERRQ(ierr); /* coarse discrete gradient */ if (pcbddc->nedcG) { PetscInt cols[2]; cols[0] = 2*i; cols[1] = 2*i+1; ierr = MatSetValuesLocal(pcbddc->nedcG,1,&i,2,cols,cvals,INSERT_VALUES);CHKERRQ(ierr); } ierr = ISRestoreIndices(eedges[i],&cols);CHKERRQ(ierr); } ierr = ISDestroy(&extrows[i]);CHKERRQ(ierr); ierr = ISDestroy(&extcols[i]);CHKERRQ(ierr); ierr = ISDestroy(&cornersis);CHKERRQ(ierr); ierr = MatDestroy(&Gins);CHKERRQ(ierr); ierr = MatDestroy(&GKins);CHKERRQ(ierr); } ierr = ISLocalToGlobalMappingDestroy(&el2g);CHKERRQ(ierr); /* Start assembling */ ierr = MatAssemblyBegin(T,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); if (pcbddc->nedcG) { ierr = MatAssemblyBegin(pcbddc->nedcG,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); } /* Free */ if (fl2g) { ierr = ISDestroy(&primals);CHKERRQ(ierr); for (i=0;imat_graph; PetscInt *oqueue = graph->queue; PetscInt *ocptr = graph->cptr; PetscInt ncc,*idxs; /* find first primal edge */ if (pcbddc->nedclocal) { ierr = ISGetIndices(pcbddc->nedclocal,(const PetscInt**)&idxs);CHKERRQ(ierr); } else { if (fl2g) { ierr = ISLocalToGlobalMappingApply(fl2g,nee,cedges,cedges);CHKERRQ(ierr); } idxs = cedges; } cum = 0; while (cum < nee && cedges[cum] < 0) cum++; /* adapt connected components */ ierr = PetscMalloc2(graph->nvtxs+1,&graph->cptr,ocptr[graph->ncc],&graph->queue);CHKERRQ(ierr); graph->cptr[0] = 0; for (i=0,ncc=0;incc;i++) { PetscInt lc = ocptr[i+1]-ocptr[i]; if (cum != nee && oqueue[ocptr[i+1]-1] == cedges[cum]) { /* this cc has a primal dof */ graph->cptr[ncc+1] = graph->cptr[ncc]+1; graph->queue[graph->cptr[ncc]] = cedges[cum]; ncc++; lc--; cum++; while (cum < nee && cedges[cum] < 0) cum++; } graph->cptr[ncc+1] = graph->cptr[ncc] + lc; for (j=0;jqueue[graph->cptr[ncc]+j] = oqueue[ocptr[i]+j]; ncc++; } graph->ncc = ncc; if (pcbddc->nedclocal) { ierr = ISRestoreIndices(pcbddc->nedclocal,(const PetscInt**)&idxs);CHKERRQ(ierr); } ierr = PetscFree2(ocptr,oqueue);CHKERRQ(ierr); } ierr = ISLocalToGlobalMappingDestroy(&fl2g);CHKERRQ(ierr); ierr = PCBDDCGraphRestoreCandidatesIS(pcbddc->mat_graph,NULL,NULL,&nee,&alleedges,&allprimals);CHKERRQ(ierr); ierr = PCBDDCGraphResetCSR(pcbddc->mat_graph);CHKERRQ(ierr); ierr = MatDestroy(&conn);CHKERRQ(ierr); ierr = ISDestroy(&nedfieldlocal);CHKERRQ(ierr); ierr = PetscFree(extrow);CHKERRQ(ierr); ierr = PetscFree2(work,rwork);CHKERRQ(ierr); ierr = PetscFree(corners);CHKERRQ(ierr); ierr = PetscFree(cedges);CHKERRQ(ierr); ierr = PetscFree(extrows);CHKERRQ(ierr); ierr = PetscFree(extcols);CHKERRQ(ierr); ierr = MatDestroy(&lG);CHKERRQ(ierr); /* Complete assembling */ ierr = MatAssemblyEnd(T,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); if (pcbddc->nedcG) { ierr = MatAssemblyEnd(pcbddc->nedcG,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); #if 0 ierr = PetscObjectSetName((PetscObject)pcbddc->nedcG,"coarse_G");CHKERRQ(ierr); ierr = MatView(pcbddc->nedcG,NULL);CHKERRQ(ierr); #endif } /* set change of basis */ ierr = PCBDDCSetChangeOfBasisMat(pc,T,singular);CHKERRQ(ierr); ierr = MatDestroy(&T);CHKERRQ(ierr); PetscFunctionReturn(0); } /* the near-null space of BDDC carries information on quadrature weights, and these can be collinear -> so cheat with MatNullSpaceCreate and create a suitable set of basis vectors first */ PetscErrorCode PCBDDCNullSpaceCreate(MPI_Comm comm, PetscBool has_const, PetscInt nvecs, Vec quad_vecs[], MatNullSpace *nnsp) { PetscErrorCode ierr; PetscInt i; PetscFunctionBegin; for (i=0;i=first && i < last) { PetscScalar *data; ierr = VecGetArray(quad_vecs[i],&data);CHKERRQ(ierr); if (!has_const) { data[i-first] = 1.; } else { data[2*i-first] = 1./PetscSqrtReal(2.); data[2*i-first+1] = -1./PetscSqrtReal(2.); } ierr = VecRestoreArray(quad_vecs[i],&data);CHKERRQ(ierr); } ierr = PetscObjectStateIncrease((PetscObject)quad_vecs[i]);CHKERRQ(ierr); } ierr = MatNullSpaceCreate(comm,has_const,nvecs,quad_vecs,nnsp);CHKERRQ(ierr); for (i=0;i=first && i < last) { PetscScalar *data; ierr = VecGetArray(quad_vecs[i],&data);CHKERRQ(ierr); if (!has_const) { data[i-first] = 0.; } else { data[2*i-first] = 0.; data[2*i-first+1] = 0.; } ierr = VecRestoreArray(quad_vecs[i],&data);CHKERRQ(ierr); } ierr = PetscObjectStateIncrease((PetscObject)quad_vecs[i]);CHKERRQ(ierr); ierr = VecLockPush(quad_vecs[i]);CHKERRQ(ierr); } PetscFunctionReturn(0); } PetscErrorCode PCBDDCComputeNoNetFlux(Mat A, Mat divudotp, PetscBool transpose, IS vl2l, PCBDDCGraph graph, MatNullSpace *nnsp) { Mat loc_divudotp; Vec p,v,vins,quad_vec,*quad_vecs; ISLocalToGlobalMapping map; IS *faces,*edges; PetscScalar *vals; const PetscScalar *array; PetscInt i,maxneighs,lmaxneighs,maxsize,nf,ne; PetscMPIInt rank; PetscErrorCode ierr; PetscFunctionBegin; ierr = PCBDDCGraphGetCandidatesIS(graph,&nf,&faces,&ne,&edges,NULL);CHKERRQ(ierr); if (graph->twodim) { lmaxneighs = 2; } else { lmaxneighs = 1; for (i=0;icount[idxs[0]]); ierr = ISRestoreIndices(edges[i],&idxs);CHKERRQ(ierr); } lmaxneighs++; /* graph count does not include self */ } ierr = MPIU_Allreduce(&lmaxneighs,&maxneighs,1,MPIU_INT,MPI_MAX,PetscObjectComm((PetscObject)A));CHKERRQ(ierr); maxsize = 0; for (i=0;icount[idxs[0]],graph->neighbours_set[idxs[0]],&idx);CHKERRQ(ierr); idx = -(idx+1); ierr = VecSetValuesLocal(quad_vecs[idx],nn,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); ierr = ISRestoreIndices(faces[i],&idxs);CHKERRQ(ierr); } for (i=0;icount[idxs[0]],graph->neighbours_set[idxs[0]],&idx);CHKERRQ(ierr); idx = -(idx+1); ierr = VecSetValuesLocal(quad_vecs[idx],nn,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); ierr = ISRestoreIndices(edges[i],&idxs);CHKERRQ(ierr); } ierr = PCBDDCGraphRestoreCandidatesIS(graph,&nf,&faces,&ne,&edges,NULL);CHKERRQ(ierr); ierr = VecRestoreArrayRead(vins,&array);CHKERRQ(ierr); if (vl2l) { ierr = VecRestoreSubVector(v,vl2l,&vins);CHKERRQ(ierr); } ierr = VecDestroy(&v);CHKERRQ(ierr); ierr = PetscFree(vals);CHKERRQ(ierr); /* assemble near null space */ for (i=0;idata; Mat_IS *matis = (Mat_IS*)pc->pmat->data; PetscFunctionBegin; ierr = MatCreateVecs(pc->pmat,&global,NULL);CHKERRQ(ierr); /* need to convert from global to local topology information and remove references to information in global ordering */ ierr = MatCreateVecs(matis->A,&local,NULL);CHKERRQ(ierr); if (pcbddc->user_provided_isfordofs) { if (pcbddc->n_ISForDofs) { PetscInt i; ierr = PetscMalloc1(pcbddc->n_ISForDofs,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); for (i=0;in_ISForDofs;i++) { ierr = PCBDDCGlobalToLocal(matis->rctx,global,local,pcbddc->ISForDofs[i],&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); } pcbddc->n_ISForDofsLocal = pcbddc->n_ISForDofs; pcbddc->n_ISForDofs = 0; ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); } } else { if (!pcbddc->n_ISForDofsLocal) { /* field split not present, create it in local ordering if bs > 1 */ PetscInt i, n = matis->A->rmap->n; ierr = MatGetBlockSize(pc->pmat,&i);CHKERRQ(ierr); if (i > 1) { pcbddc->n_ISForDofsLocal = i; ierr = PetscMalloc1(pcbddc->n_ISForDofsLocal,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); for (i=0;in_ISForDofsLocal;i++) { ierr = ISCreateStride(PetscObjectComm((PetscObject)pc),n/pcbddc->n_ISForDofsLocal,i,pcbddc->n_ISForDofsLocal,&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); } } } else { PetscInt i; for (i=0;in_ISForDofsLocal;i++) { ierr = PCBDDCConsistencyCheckIS(pc,MPI_LAND,&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); } } } if (!pcbddc->DirichletBoundariesLocal && pcbddc->DirichletBoundaries) { ierr = PCBDDCGlobalToLocal(matis->rctx,global,local,pcbddc->DirichletBoundaries,&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); } else if (pcbddc->DirichletBoundariesLocal) { ierr = PCBDDCConsistencyCheckIS(pc,MPI_LAND,&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); } if (!pcbddc->NeumannBoundariesLocal && pcbddc->NeumannBoundaries) { ierr = PCBDDCGlobalToLocal(matis->rctx,global,local,pcbddc->NeumannBoundaries,&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); } else if (pcbddc->NeumannBoundariesLocal) { ierr = PCBDDCConsistencyCheckIS(pc,MPI_LOR,&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); } if (!pcbddc->user_primal_vertices_local && pcbddc->user_primal_vertices) { ierr = PCBDDCGlobalToLocal(matis->rctx,global,local,pcbddc->user_primal_vertices,&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); } ierr = VecDestroy(&global);CHKERRQ(ierr); ierr = VecDestroy(&local);CHKERRQ(ierr); PetscFunctionReturn(0); } PetscErrorCode PCBDDCConsistencyCheckIS(PC pc, MPI_Op mop, IS *is) { Mat_IS *matis = (Mat_IS*)(pc->pmat->data); PetscErrorCode ierr; IS nis; const PetscInt *idxs; PetscInt i,nd,n = matis->A->rmap->n,*nidxs,nnd; PetscBool *ld; PetscFunctionBegin; if (mop != MPI_LAND && mop != MPI_LOR) SETERRQ(PetscObjectComm((PetscObject)(pc)),PETSC_ERR_SUP,"Supported are MPI_LAND and MPI_LOR"); ierr = MatISSetUpSF(pc->pmat);CHKERRQ(ierr); if (mop == MPI_LAND) { /* init rootdata with true */ ld = (PetscBool*) matis->sf_rootdata; for (i=0;ipmat->rmap->n;i++) ld[i] = PETSC_TRUE; } else { ierr = PetscMemzero(matis->sf_rootdata,pc->pmat->rmap->n*sizeof(PetscBool));CHKERRQ(ierr); } ierr = PetscMemzero(matis->sf_leafdata,n*sizeof(PetscBool));CHKERRQ(ierr); ierr = ISGetLocalSize(*is,&nd);CHKERRQ(ierr); ierr = ISGetIndices(*is,&idxs);CHKERRQ(ierr); ld = (PetscBool*) matis->sf_leafdata; for (i=0;isf,MPIU_BOOL,matis->sf_leafdata,matis->sf_rootdata,mop);CHKERRQ(ierr); ierr = PetscSFReduceEnd(matis->sf,MPIU_BOOL,matis->sf_leafdata,matis->sf_rootdata,mop);CHKERRQ(ierr); ierr = PetscSFBcastBegin(matis->sf,MPIU_BOOL,matis->sf_rootdata,matis->sf_leafdata);CHKERRQ(ierr); ierr = PetscSFBcastEnd(matis->sf,MPIU_BOOL,matis->sf_rootdata,matis->sf_leafdata);CHKERRQ(ierr); if (mop == MPI_LAND) { ierr = PetscMalloc1(nd,&nidxs);CHKERRQ(ierr); } else { ierr = PetscMalloc1(n,&nidxs);CHKERRQ(ierr); } for (i=0,nnd=0;idata); PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); PetscErrorCode ierr; PetscFunctionBegin; if (!pcbddc->benign_have_null) { PetscFunctionReturn(0); } if (pcbddc->ChangeOfBasisMatrix) { Vec swap; ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); swap = pcbddc->work_change; pcbddc->work_change = r; r = swap; } ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); ierr = VecSet(z,0.);CHKERRQ(ierr); ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); if (pcbddc->ChangeOfBasisMatrix) { pcbddc->work_change = r; ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); } PetscFunctionReturn(0); } PetscErrorCode PCBDDCBenignMatMult_Private_Private(Mat A, Vec x, Vec y, PetscBool transpose) { PCBDDCBenignMatMult_ctx ctx; PetscErrorCode ierr; PetscBool apply_right,apply_left,reset_x; PetscFunctionBegin; ierr = MatShellGetContext(A,&ctx);CHKERRQ(ierr); if (transpose) { apply_right = ctx->apply_left; apply_left = ctx->apply_right; } else { apply_right = ctx->apply_right; apply_left = ctx->apply_left; } reset_x = PETSC_FALSE; if (apply_right) { const PetscScalar *ax; PetscInt nl,i; ierr = VecGetLocalSize(x,&nl);CHKERRQ(ierr); ierr = VecGetArrayRead(x,&ax);CHKERRQ(ierr); ierr = PetscMemcpy(ctx->work,ax,nl*sizeof(PetscScalar));CHKERRQ(ierr); ierr = VecRestoreArrayRead(x,&ax);CHKERRQ(ierr); for (i=0;ibenign_n;i++) { PetscScalar sum,val; const PetscInt *idxs; PetscInt nz,j; ierr = ISGetLocalSize(ctx->benign_zerodiag_subs[i],&nz);CHKERRQ(ierr); ierr = ISGetIndices(ctx->benign_zerodiag_subs[i],&idxs);CHKERRQ(ierr); sum = 0.; if (ctx->apply_p0) { val = ctx->work[idxs[nz-1]]; for (j=0;jwork[idxs[j]]; ctx->work[idxs[j]] += val; } } else { for (j=0;jwork[idxs[j]]; } } ctx->work[idxs[nz-1]] -= sum; ierr = ISRestoreIndices(ctx->benign_zerodiag_subs[i],&idxs);CHKERRQ(ierr); } ierr = VecPlaceArray(x,ctx->work);CHKERRQ(ierr); reset_x = PETSC_TRUE; } if (transpose) { ierr = MatMultTranspose(ctx->A,x,y);CHKERRQ(ierr); } else { ierr = MatMult(ctx->A,x,y);CHKERRQ(ierr); } if (reset_x) { ierr = VecResetArray(x);CHKERRQ(ierr); } if (apply_left) { PetscScalar *ay; PetscInt i; ierr = VecGetArray(y,&ay);CHKERRQ(ierr); for (i=0;ibenign_n;i++) { PetscScalar sum,val; const PetscInt *idxs; PetscInt nz,j; ierr = ISGetLocalSize(ctx->benign_zerodiag_subs[i],&nz);CHKERRQ(ierr); ierr = ISGetIndices(ctx->benign_zerodiag_subs[i],&idxs);CHKERRQ(ierr); val = -ay[idxs[nz-1]]; if (ctx->apply_p0) { sum = 0.; for (j=0;jbenign_zerodiag_subs[i],&idxs);CHKERRQ(ierr); } ierr = VecRestoreArray(y,&ay);CHKERRQ(ierr); } PetscFunctionReturn(0); } PetscErrorCode PCBDDCBenignMatMultTranspose_Private(Mat A, Vec x, Vec y) { PetscErrorCode ierr; PetscFunctionBegin; ierr = PCBDDCBenignMatMult_Private_Private(A,x,y,PETSC_TRUE);CHKERRQ(ierr); PetscFunctionReturn(0); } PetscErrorCode PCBDDCBenignMatMult_Private(Mat A, Vec x, Vec y) { PetscErrorCode ierr; PetscFunctionBegin; ierr = PCBDDCBenignMatMult_Private_Private(A,x,y,PETSC_FALSE);CHKERRQ(ierr); PetscFunctionReturn(0); } PetscErrorCode PCBDDCBenignShellMat(PC pc, PetscBool restore) { PC_IS *pcis = (PC_IS*)pc->data; PC_BDDC *pcbddc = (PC_BDDC*)pc->data; PCBDDCBenignMatMult_ctx ctx; PetscErrorCode ierr; PetscFunctionBegin; if (!restore) { Mat A_IB,A_BI; PetscScalar *work; PCBDDCReuseSolvers reuse = pcbddc->sub_schurs ? pcbddc->sub_schurs->reuse_solver : NULL; if (pcbddc->benign_original_mat) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Benign original mat has not been restored"); if (!pcbddc->benign_change || !pcbddc->benign_n || pcbddc->benign_change_explicit) PetscFunctionReturn(0); ierr = PetscMalloc1(pcis->n,&work);CHKERRQ(ierr); ierr = MatCreate(PETSC_COMM_SELF,&A_IB);CHKERRQ(ierr); ierr = MatSetSizes(A_IB,pcis->n-pcis->n_B,pcis->n_B,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); ierr = MatSetType(A_IB,MATSHELL);CHKERRQ(ierr); ierr = MatShellSetOperation(A_IB,MATOP_MULT,(void (*)(void))PCBDDCBenignMatMult_Private);CHKERRQ(ierr); ierr = MatShellSetOperation(A_IB,MATOP_MULT_TRANSPOSE,(void (*)(void))PCBDDCBenignMatMultTranspose_Private);CHKERRQ(ierr); ierr = PetscNew(&ctx);CHKERRQ(ierr); ierr = MatShellSetContext(A_IB,ctx);CHKERRQ(ierr); ctx->apply_left = PETSC_TRUE; ctx->apply_right = PETSC_FALSE; ctx->apply_p0 = PETSC_FALSE; ctx->benign_n = pcbddc->benign_n; if (reuse) { ctx->benign_zerodiag_subs = reuse->benign_zerodiag_subs; ctx->free = PETSC_FALSE; } else { /* TODO: could be optimized for successive solves */ ISLocalToGlobalMapping N_to_D; PetscInt i; ierr = ISLocalToGlobalMappingCreateIS(pcis->is_I_local,&N_to_D);CHKERRQ(ierr); ierr = PetscMalloc1(pcbddc->benign_n,&ctx->benign_zerodiag_subs);CHKERRQ(ierr); for (i=0;ibenign_n;i++) { ierr = ISGlobalToLocalMappingApplyIS(N_to_D,IS_GTOLM_DROP,pcbddc->benign_zerodiag_subs[i],&ctx->benign_zerodiag_subs[i]);CHKERRQ(ierr); } ierr = ISLocalToGlobalMappingDestroy(&N_to_D);CHKERRQ(ierr); ctx->free = PETSC_TRUE; } ctx->A = pcis->A_IB; ctx->work = work; ierr = MatSetUp(A_IB);CHKERRQ(ierr); ierr = MatAssemblyBegin(A_IB,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); ierr = MatAssemblyEnd(A_IB,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); pcis->A_IB = A_IB; /* A_BI as A_IB^T */ ierr = MatCreateTranspose(A_IB,&A_BI);CHKERRQ(ierr); pcbddc->benign_original_mat = pcis->A_BI; pcis->A_BI = A_BI; } else { if (!pcbddc->benign_original_mat) { PetscFunctionReturn(0); } ierr = MatShellGetContext(pcis->A_IB,&ctx);CHKERRQ(ierr); ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); pcis->A_IB = ctx->A; ctx->A = NULL; ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); pcis->A_BI = pcbddc->benign_original_mat; pcbddc->benign_original_mat = NULL; if (ctx->free) { PetscInt i; for (i=0;ibenign_n;i++) { ierr = ISDestroy(&ctx->benign_zerodiag_subs[i]);CHKERRQ(ierr); } ierr = PetscFree(ctx->benign_zerodiag_subs);CHKERRQ(ierr); } ierr = PetscFree(ctx->work);CHKERRQ(ierr); ierr = PetscFree(ctx);CHKERRQ(ierr); } PetscFunctionReturn(0); } /* used just in bddc debug mode */ PetscErrorCode PCBDDCBenignProject(PC pc, IS is1, IS is2, Mat *B) { PC_BDDC *pcbddc = (PC_BDDC*)pc->data; Mat_IS *matis = (Mat_IS*)pc->pmat->data; Mat An; PetscErrorCode ierr; PetscFunctionBegin; ierr = MatPtAP(matis->A,pcbddc->benign_change,MAT_INITIAL_MATRIX,2.0,&An);CHKERRQ(ierr); ierr = MatZeroRowsColumns(An,pcbddc->benign_n,pcbddc->benign_p0_lidx,1.0,NULL,NULL);CHKERRQ(ierr); if (is1) { ierr = MatCreateSubMatrix(An,is1,is2,MAT_INITIAL_MATRIX,B);CHKERRQ(ierr); ierr = MatDestroy(&An);CHKERRQ(ierr); } else { *B = An; } PetscFunctionReturn(0); } /* TODO: add reuse flag */ PetscErrorCode MatSeqAIJCompress(Mat A, Mat *B) { Mat Bt; PetscScalar *a,*bdata; const PetscInt *ii,*ij; PetscInt m,n,i,nnz,*bii,*bij; PetscBool flg_row; PetscErrorCode ierr; PetscFunctionBegin; ierr = MatGetSize(A,&n,&m);CHKERRQ(ierr); ierr = MatGetRowIJ(A,0,PETSC_FALSE,PETSC_FALSE,&n,&ii,&ij,&flg_row);CHKERRQ(ierr); ierr = MatSeqAIJGetArray(A,&a);CHKERRQ(ierr); nnz = n; for (i=0;i PETSC_SMALL)) nnz++; } ierr = PetscMalloc1(n+1,&bii);CHKERRQ(ierr); ierr = PetscMalloc1(nnz,&bij);CHKERRQ(ierr); ierr = PetscMalloc1(nnz,&bdata);CHKERRQ(ierr); nnz = 0; bii[0] = 0; for (i=0;i PETSC_SMALL) || ij[j] == i) { bij[nnz] = ij[j]; bdata[nnz] = entry; nnz++; } } bii[i+1] = nnz; } ierr = MatSeqAIJRestoreArray(A,&a);CHKERRQ(ierr); ierr = MatCreateSeqAIJWithArrays(PetscObjectComm((PetscObject)A),n,m,bii,bij,bdata,&Bt);CHKERRQ(ierr); ierr = MatRestoreRowIJ(A,0,PETSC_FALSE,PETSC_FALSE,&n,&ii,&ij,&flg_row);CHKERRQ(ierr); { Mat_SeqAIJ *b = (Mat_SeqAIJ*)(Bt->data); b->free_a = PETSC_TRUE; b->free_ij = PETSC_TRUE; } *B = Bt; PetscFunctionReturn(0); } PetscErrorCode MatDetectDisconnectedComponents(Mat A, PetscBool filter, PetscInt *ncc, IS* cc[]) { Mat B; IS is_dummy,*cc_n; ISLocalToGlobalMapping l2gmap_dummy; PCBDDCGraph graph; PetscInt i,n; PetscInt *xadj,*adjncy; PetscInt *xadj_filtered,*adjncy_filtered; PetscBool flg_row,isseqaij; PetscErrorCode ierr; PetscFunctionBegin; if (!A->rmap->N || !A->cmap->N) { *ncc = 0; *cc = NULL; PetscFunctionReturn(0); } ierr = PetscObjectTypeCompare((PetscObject)A,MATSEQAIJ,&isseqaij);CHKERRQ(ierr); if (!isseqaij && filter) { PetscBool isseqdense; ierr = PetscObjectTypeCompare((PetscObject)A,MATSEQDENSE,&isseqdense);CHKERRQ(ierr); if (!isseqdense) { ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr); } else { /* TODO: rectangular case and LDA */ PetscScalar *array; PetscReal chop=1.e-6; ierr = MatDuplicate(A,MAT_COPY_VALUES,&B);CHKERRQ(ierr); ierr = MatDenseGetArray(B,&array);CHKERRQ(ierr); ierr = MatGetSize(B,&n,NULL);CHKERRQ(ierr); for (i=0;ixadj = xadj_filtered; graph->adjncy = adjncy_filtered; } else { graph->xadj = xadj; graph->adjncy = adjncy; } ierr = PCBDDCGraphSetUp(graph,1,NULL,NULL,0,NULL,NULL);CHKERRQ(ierr); ierr = PCBDDCGraphComputeConnectedComponents(graph);CHKERRQ(ierr); /* partial clean up */ ierr = PetscFree2(xadj_filtered,adjncy_filtered);CHKERRQ(ierr); ierr = MatRestoreRowIJ(B,0,PETSC_TRUE,PETSC_FALSE,&n,(const PetscInt**)&xadj,(const PetscInt**)&adjncy,&flg_row);CHKERRQ(ierr); if (A != B) { ierr = MatDestroy(&B);CHKERRQ(ierr); } /* get back data */ if (ncc) *ncc = graph->ncc; if (cc) { ierr = PetscMalloc1(graph->ncc,&cc_n);CHKERRQ(ierr); for (i=0;incc;i++) { ierr = ISCreateGeneral(PETSC_COMM_SELF,graph->cptr[i+1]-graph->cptr[i],graph->queue+graph->cptr[i],PETSC_COPY_VALUES,&cc_n[i]);CHKERRQ(ierr); } *cc = cc_n; } /* clean up graph */ graph->xadj = 0; graph->adjncy = 0; ierr = PCBDDCGraphDestroy(&graph);CHKERRQ(ierr); PetscFunctionReturn(0); } PetscErrorCode PCBDDCBenignCheck(PC pc, IS zerodiag) { PC_BDDC* pcbddc = (PC_BDDC*)pc->data; PC_IS* pcis = (PC_IS*)(pc->data); IS dirIS = NULL; PetscInt i; PetscErrorCode ierr; PetscFunctionBegin; ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); if (zerodiag) { Mat A; Vec vec3_N; PetscScalar *vals; const PetscInt *idxs; PetscInt nz,*count; /* p0 */ ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); ierr = PetscMalloc1(pcis->n,&vals);CHKERRQ(ierr); ierr = ISGetLocalSize(zerodiag,&nz);CHKERRQ(ierr); ierr = ISGetIndices(zerodiag,&idxs);CHKERRQ(ierr); for (i=0;ivec1_N,nz,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); ierr = VecAssemblyBegin(pcis->vec1_N);CHKERRQ(ierr); ierr = VecAssemblyEnd(pcis->vec1_N);CHKERRQ(ierr); /* v_I */ ierr = VecSetRandom(pcis->vec2_N,NULL);CHKERRQ(ierr); for (i=0;ivec2_N,nz,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); ierr = ISRestoreIndices(zerodiag,&idxs);CHKERRQ(ierr); ierr = ISGetIndices(pcis->is_B_local,&idxs);CHKERRQ(ierr); for (i=0;in_B;i++) vals[i] = 0.; ierr = VecSetValues(pcis->vec2_N,pcis->n_B,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); ierr = ISRestoreIndices(pcis->is_B_local,&idxs);CHKERRQ(ierr); if (dirIS) { PetscInt n; ierr = ISGetLocalSize(dirIS,&n);CHKERRQ(ierr); ierr = ISGetIndices(dirIS,&idxs);CHKERRQ(ierr); for (i=0;ivec2_N,n,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); ierr = ISRestoreIndices(dirIS,&idxs);CHKERRQ(ierr); } ierr = VecAssemblyBegin(pcis->vec2_N);CHKERRQ(ierr); ierr = VecAssemblyEnd(pcis->vec2_N);CHKERRQ(ierr); ierr = VecDuplicate(pcis->vec1_N,&vec3_N);CHKERRQ(ierr); ierr = VecSet(vec3_N,0.);CHKERRQ(ierr); ierr = MatISGetLocalMat(pc->pmat,&A);CHKERRQ(ierr); ierr = MatMult(A,pcis->vec1_N,vec3_N);CHKERRQ(ierr); ierr = VecDot(vec3_N,pcis->vec2_N,&vals[0]);CHKERRQ(ierr); if (PetscAbsScalar(vals[0]) > 1.e-1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Benign trick can not be applied! b(v_I,p_0) = %1.6e (should be numerically 0.)",PetscAbsScalar(vals[0])); ierr = PetscFree(vals);CHKERRQ(ierr); ierr = VecDestroy(&vec3_N);CHKERRQ(ierr); /* there should not be any pressure dofs lying on the interface */ ierr = PetscCalloc1(pcis->n,&count);CHKERRQ(ierr); ierr = ISGetIndices(pcis->is_B_local,&idxs);CHKERRQ(ierr); for (i=0;in_B;i++) count[idxs[i]]++; ierr = ISRestoreIndices(pcis->is_B_local,&idxs);CHKERRQ(ierr); ierr = ISGetIndices(zerodiag,&idxs);CHKERRQ(ierr); for (i=0;ivec1_global,NULL);CHKERRQ(ierr); for (i=0;ibenign_n;i++) pcbddc->benign_p0[i] = -PetscGlobalRank-i; ierr = PCBDDCBenignGetOrSetP0(pc,pcis->vec1_global,PETSC_FALSE);CHKERRQ(ierr); for (i=0;ibenign_n;i++) pcbddc->benign_p0[i] = 1; ierr = PCBDDCBenignGetOrSetP0(pc,pcis->vec1_global,PETSC_TRUE);CHKERRQ(ierr); for (i=0;ibenign_n;i++) { PetscInt val = PetscRealPart(pcbddc->benign_p0[i]); if (val != -PetscGlobalRank-i) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Error testing PCBDDCBenignGetOrSetP0! Found %g at %d instead of %g\n",PetscRealPart(pcbddc->benign_p0[i]),i,-PetscGlobalRank-i);CHKERRQ(ierr); } PetscFunctionReturn(0); } PetscErrorCode PCBDDCBenignDetectSaddlePoint(PC pc, IS *zerodiaglocal) { PC_BDDC* pcbddc = (PC_BDDC*)pc->data; IS pressures,zerodiag,zerodiag_save,*zerodiag_subs; PetscInt nz,n; PetscInt *interior_dofs,n_interior_dofs,nneu; PetscBool sorted,have_null,has_null_pressures,recompute_zerodiag,checkb; PetscErrorCode ierr; PetscFunctionBegin; ierr = PetscSFDestroy(&pcbddc->benign_sf);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->benign_B0);CHKERRQ(ierr); for (n=0;nbenign_n;n++) { ierr = ISDestroy(&pcbddc->benign_zerodiag_subs[n]);CHKERRQ(ierr); } ierr = PetscFree(pcbddc->benign_zerodiag_subs);CHKERRQ(ierr); pcbddc->benign_n = 0; /* if a local info on dofs is present, uses the last field for "pressures" (or fid by command line) otherwise, it uses only zerodiagonal dofs (ok if the pressure block is all zero; it could fail if it is not) Checks if all the pressure dofs in each subdomain have a zero diagonal If not, a change of basis on pressures is not needed since the local Schur complements are already SPD */ has_null_pressures = PETSC_TRUE; have_null = PETSC_TRUE; if (pcbddc->n_ISForDofsLocal) { PetscInt npl,*idxs,p = pcbddc->n_ISForDofsLocal-1; ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)pc),((PetscObject)pc)->prefix,"BDDC benign options","PC");CHKERRQ(ierr); ierr = PetscOptionsInt ("-pc_bddc_pressure_field","Field id for pressures",NULL,p,&p,NULL);CHKERRQ(ierr); ierr = PetscOptionsEnd();CHKERRQ(ierr); if (p < 0 || p > pcbddc->n_ISForDofsLocal-1) SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_USER,"Invalid field id for pressures %D",p); /* Dofs splitting for BDDC cannot have PETSC_COMM_SELF, so create a sequential IS */ ierr = ISGetLocalSize(pcbddc->ISForDofsLocal[p],&npl);CHKERRQ(ierr); ierr = ISGetIndices(pcbddc->ISForDofsLocal[p],(const PetscInt**)&idxs);CHKERRQ(ierr); ierr = ISCreateGeneral(PETSC_COMM_SELF,npl,idxs,PETSC_COPY_VALUES,&pressures);CHKERRQ(ierr); ierr = ISRestoreIndices(pcbddc->ISForDofsLocal[p],(const PetscInt**)&idxs);CHKERRQ(ierr); ierr = ISSorted(pressures,&sorted);CHKERRQ(ierr); if (!sorted) { ierr = ISSort(pressures);CHKERRQ(ierr); } } else { pressures = NULL; } /* pcis has not been setup yet, so get the local size from the subdomain matrix */ ierr = MatGetLocalSize(pcbddc->local_mat,&n,NULL);CHKERRQ(ierr); if (!n) pcbddc->benign_change_explicit = PETSC_TRUE; ierr = MatFindZeroDiagonals(pcbddc->local_mat,&zerodiag);CHKERRQ(ierr); ierr = ISSorted(zerodiag,&sorted);CHKERRQ(ierr); if (!sorted) { ierr = ISSort(zerodiag);CHKERRQ(ierr); } ierr = PetscObjectReference((PetscObject)zerodiag);CHKERRQ(ierr); zerodiag_save = zerodiag; ierr = ISGetLocalSize(zerodiag,&nz);CHKERRQ(ierr); if (!nz) { if (n) have_null = PETSC_FALSE; has_null_pressures = PETSC_FALSE; ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); } recompute_zerodiag = PETSC_FALSE; /* in case disconnected subdomains info is present, split the pressures accordingly (otherwise the benign trick could fail) */ zerodiag_subs = NULL; pcbddc->benign_n = 0; n_interior_dofs = 0; interior_dofs = NULL; nneu = 0; if (pcbddc->NeumannBoundariesLocal) { ierr = ISGetLocalSize(pcbddc->NeumannBoundariesLocal,&nneu);CHKERRQ(ierr); } checkb = (PetscBool)(!pcbddc->NeumannBoundariesLocal || pcbddc->current_level); if (checkb) { /* need to compute interior nodes */ PetscInt n,i,j; PetscInt n_neigh,*neigh,*n_shared,**shared; PetscInt *iwork; ierr = ISLocalToGlobalMappingGetSize(pc->pmat->rmap->mapping,&n);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingGetInfo(pc->pmat->rmap->mapping,&n_neigh,&neigh,&n_shared,&shared);CHKERRQ(ierr); ierr = PetscCalloc1(n,&iwork);CHKERRQ(ierr); ierr = PetscMalloc1(n,&interior_dofs);CHKERRQ(ierr); for (i=1;ipmat->rmap->mapping,&n_neigh,&neigh,&n_shared,&shared);CHKERRQ(ierr); } if (has_null_pressures) { IS *subs; PetscInt nsubs,i,j,nl; const PetscInt *idxs; PetscScalar *array; Vec *work; Mat_IS* matis = (Mat_IS*)(pc->pmat->data); subs = pcbddc->local_subs; nsubs = pcbddc->n_local_subs; /* these vectors are needed to check if the constant on pressures is in the kernel of the local operator B (i.e. B(v_I,p0) should be zero) */ if (checkb) { ierr = VecDuplicateVecs(matis->y,2,&work);CHKERRQ(ierr); ierr = ISGetLocalSize(zerodiag,&nl);CHKERRQ(ierr); ierr = ISGetIndices(zerodiag,&idxs);CHKERRQ(ierr); /* work[0] = 1_p */ ierr = VecSet(work[0],0.);CHKERRQ(ierr); ierr = VecGetArray(work[0],&array);CHKERRQ(ierr); for (j=0;j 1) { ierr = PetscCalloc1(nsubs,&zerodiag_subs);CHKERRQ(ierr); for (i=0;ix,0);CHKERRQ(ierr); ierr = ISGetLocalSize(subs[i],&nl);CHKERRQ(ierr); ierr = ISGetIndices(subs[i],&idxs);CHKERRQ(ierr); ierr = VecGetArray(matis->x,&array);CHKERRQ(ierr); for (j=0;jx,&array);CHKERRQ(ierr); ierr = ISRestoreIndices(subs[i],&idxs);CHKERRQ(ierr); ierr = VecPointwiseMult(matis->x,work[0],matis->x);CHKERRQ(ierr); ierr = MatMult(matis->A,matis->x,matis->y);CHKERRQ(ierr); ierr = VecPointwiseMult(matis->y,work[1],matis->y);CHKERRQ(ierr); ierr = VecGetArray(matis->y,&array);CHKERRQ(ierr); for (j=0;j PETSC_SMALL) { valid = PETSC_FALSE; break; } } ierr = VecRestoreArray(matis->y,&array);CHKERRQ(ierr); } if (valid && nneu) { const PetscInt *idxs; PetscInt nzb; ierr = ISGetIndices(pcbddc->NeumannBoundariesLocal,&idxs);CHKERRQ(ierr); ierr = ISGlobalToLocalMappingApply(l2g,IS_GTOLM_DROP,nneu,idxs,&nzb,NULL);CHKERRQ(ierr); ierr = ISRestoreIndices(pcbddc->NeumannBoundariesLocal,&idxs);CHKERRQ(ierr); if (nzb) valid = PETSC_FALSE; } if (valid && pressures) { IS t_pressure_subs; ierr = ISGlobalToLocalMappingApplyIS(l2g,IS_GTOLM_DROP,pressures,&t_pressure_subs);CHKERRQ(ierr); ierr = ISEqual(t_pressure_subs,t_zerodiag_subs,&valid);CHKERRQ(ierr); ierr = ISDestroy(&t_pressure_subs);CHKERRQ(ierr); } if (valid) { ierr = ISLocalToGlobalMappingApplyIS(l2g,t_zerodiag_subs,&zerodiag_subs[pcbddc->benign_n]);CHKERRQ(ierr); pcbddc->benign_n++; } else { recompute_zerodiag = PETSC_TRUE; } } ierr = ISDestroy(&t_zerodiag_subs);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingDestroy(&l2g);CHKERRQ(ierr); } } else { /* there's just one subdomain (or zero if they have not been detected */ PetscBool valid = PETSC_TRUE; if (nneu) valid = PETSC_FALSE; if (valid && pressures) { ierr = ISEqual(pressures,zerodiag,&valid);CHKERRQ(ierr); } if (valid && checkb) { ierr = MatMult(matis->A,work[0],matis->x);CHKERRQ(ierr); ierr = VecPointwiseMult(matis->x,work[1],matis->x);CHKERRQ(ierr); ierr = VecGetArray(matis->x,&array);CHKERRQ(ierr); for (j=0;j PETSC_SMALL) { valid = PETSC_FALSE; break; } } ierr = VecRestoreArray(matis->x,&array);CHKERRQ(ierr); } if (valid) { pcbddc->benign_n = 1; ierr = PetscMalloc1(pcbddc->benign_n,&zerodiag_subs);CHKERRQ(ierr); ierr = PetscObjectReference((PetscObject)zerodiag);CHKERRQ(ierr); zerodiag_subs[0] = zerodiag; } } if (checkb) { ierr = VecDestroyVecs(2,&work);CHKERRQ(ierr); } } ierr = PetscFree(interior_dofs);CHKERRQ(ierr); if (!pcbddc->benign_n) { PetscInt n; ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); recompute_zerodiag = PETSC_FALSE; ierr = MatGetLocalSize(pcbddc->local_mat,&n,NULL);CHKERRQ(ierr); if (n) { has_null_pressures = PETSC_FALSE; have_null = PETSC_FALSE; } } /* final check for null pressures */ if (zerodiag && pressures) { PetscInt nz,np; ierr = ISGetLocalSize(zerodiag,&nz);CHKERRQ(ierr); ierr = ISGetLocalSize(pressures,&np);CHKERRQ(ierr); if (nz != np) have_null = PETSC_FALSE; } if (recompute_zerodiag) { ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); if (pcbddc->benign_n == 1) { ierr = PetscObjectReference((PetscObject)zerodiag_subs[0]);CHKERRQ(ierr); zerodiag = zerodiag_subs[0]; } else { PetscInt i,nzn,*new_idxs; nzn = 0; for (i=0;ibenign_n;i++) { PetscInt ns; ierr = ISGetLocalSize(zerodiag_subs[i],&ns);CHKERRQ(ierr); nzn += ns; } ierr = PetscMalloc1(nzn,&new_idxs);CHKERRQ(ierr); nzn = 0; for (i=0;ibenign_n;i++) { PetscInt ns,*idxs; ierr = ISGetLocalSize(zerodiag_subs[i],&ns);CHKERRQ(ierr); ierr = ISGetIndices(zerodiag_subs[i],(const PetscInt**)&idxs);CHKERRQ(ierr); ierr = PetscMemcpy(new_idxs+nzn,idxs,ns*sizeof(PetscInt));CHKERRQ(ierr); ierr = ISRestoreIndices(zerodiag_subs[i],(const PetscInt**)&idxs);CHKERRQ(ierr); nzn += ns; } ierr = PetscSortInt(nzn,new_idxs);CHKERRQ(ierr); ierr = ISCreateGeneral(PETSC_COMM_SELF,nzn,new_idxs,PETSC_OWN_POINTER,&zerodiag);CHKERRQ(ierr); } have_null = PETSC_FALSE; } /* Prepare matrix to compute no-net-flux */ if (pcbddc->compute_nonetflux && !pcbddc->divudotp) { Mat A,loc_divudotp; ISLocalToGlobalMapping rl2g,cl2g,l2gmap; IS row,col,isused = NULL; PetscInt M,N,n,st,n_isused; if (pressures) { isused = pressures; } else { isused = zerodiag_save; } ierr = MatGetLocalToGlobalMapping(pc->pmat,&l2gmap,NULL);CHKERRQ(ierr); ierr = MatISGetLocalMat(pc->pmat,&A);CHKERRQ(ierr); ierr = MatGetLocalSize(A,&n,NULL);CHKERRQ(ierr); if (!isused && n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_USER,"Don't know how to extract div u dot p! Please provide the pressure field"); n_isused = 0; if (isused) { ierr = ISGetLocalSize(isused,&n_isused);CHKERRQ(ierr); } ierr = MPI_Scan(&n_isused,&st,1,MPIU_INT,MPI_SUM,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); st = st-n_isused; if (n) { const PetscInt *gidxs; ierr = MatCreateSubMatrix(A,isused,NULL,MAT_INITIAL_MATRIX,&loc_divudotp);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingGetIndices(l2gmap,&gidxs);CHKERRQ(ierr); /* TODO: extend ISCreateStride with st = PETSC_DECIDE */ ierr = ISCreateStride(PetscObjectComm((PetscObject)pc),n_isused,st,1,&row);CHKERRQ(ierr); ierr = ISCreateGeneral(PetscObjectComm((PetscObject)pc),n,gidxs,PETSC_COPY_VALUES,&col);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingRestoreIndices(l2gmap,&gidxs);CHKERRQ(ierr); } else { ierr = MatCreateSeqAIJ(PETSC_COMM_SELF,0,0,1,NULL,&loc_divudotp);CHKERRQ(ierr); ierr = ISCreateStride(PetscObjectComm((PetscObject)pc),n_isused,st,1,&row);CHKERRQ(ierr); ierr = ISCreateGeneral(PetscObjectComm((PetscObject)pc),0,NULL,PETSC_COPY_VALUES,&col);CHKERRQ(ierr); } ierr = MatGetSize(pc->pmat,NULL,&N);CHKERRQ(ierr); ierr = ISGetSize(row,&M);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingCreateIS(row,&rl2g);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingCreateIS(col,&cl2g);CHKERRQ(ierr); ierr = ISDestroy(&row);CHKERRQ(ierr); ierr = ISDestroy(&col);CHKERRQ(ierr); ierr = MatCreate(PetscObjectComm((PetscObject)pc),&pcbddc->divudotp);CHKERRQ(ierr); ierr = MatSetType(pcbddc->divudotp,MATIS);CHKERRQ(ierr); ierr = MatSetSizes(pcbddc->divudotp,PETSC_DECIDE,PETSC_DECIDE,M,N);CHKERRQ(ierr); ierr = MatSetLocalToGlobalMapping(pcbddc->divudotp,rl2g,cl2g);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr); ierr = MatISSetLocalMat(pcbddc->divudotp,loc_divudotp);CHKERRQ(ierr); ierr = MatDestroy(&loc_divudotp);CHKERRQ(ierr); ierr = MatAssemblyBegin(pcbddc->divudotp,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); ierr = MatAssemblyEnd(pcbddc->divudotp,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); } ierr = ISDestroy(&zerodiag_save);CHKERRQ(ierr); /* change of basis and p0 dofs */ if (has_null_pressures) { IS zerodiagc; const PetscInt *idxs,*idxsc; PetscInt i,s,*nnz; ierr = ISGetLocalSize(zerodiag,&nz);CHKERRQ(ierr); ierr = ISComplement(zerodiag,0,n,&zerodiagc);CHKERRQ(ierr); ierr = ISGetIndices(zerodiagc,&idxsc);CHKERRQ(ierr); /* local change of basis for pressures */ ierr = MatDestroy(&pcbddc->benign_change);CHKERRQ(ierr); ierr = MatCreate(PetscObjectComm((PetscObject)pcbddc->local_mat),&pcbddc->benign_change);CHKERRQ(ierr); ierr = MatSetType(pcbddc->benign_change,MATAIJ);CHKERRQ(ierr); ierr = MatSetSizes(pcbddc->benign_change,n,n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); ierr = PetscMalloc1(n,&nnz);CHKERRQ(ierr); for (i=0;ibenign_n;i++) { PetscInt nzs,j; ierr = ISGetLocalSize(zerodiag_subs[i],&nzs);CHKERRQ(ierr); ierr = ISGetIndices(zerodiag_subs[i],&idxs);CHKERRQ(ierr); for (j=0;jbenign_change,0,nnz);CHKERRQ(ierr); ierr = PetscFree(nnz);CHKERRQ(ierr); /* set identity on velocities */ for (i=0;ibenign_change,idxsc[i],idxsc[i],1.,INSERT_VALUES);CHKERRQ(ierr); } ierr = ISRestoreIndices(zerodiagc,&idxsc);CHKERRQ(ierr); ierr = ISDestroy(&zerodiagc);CHKERRQ(ierr); ierr = PetscFree3(pcbddc->benign_p0_lidx,pcbddc->benign_p0_gidx,pcbddc->benign_p0);CHKERRQ(ierr); ierr = PetscMalloc3(pcbddc->benign_n,&pcbddc->benign_p0_lidx,pcbddc->benign_n,&pcbddc->benign_p0_gidx,pcbddc->benign_n,&pcbddc->benign_p0);CHKERRQ(ierr); /* set change on pressures */ for (s=0;sbenign_n;s++) { PetscScalar *array; PetscInt nzs; ierr = ISGetLocalSize(zerodiag_subs[s],&nzs);CHKERRQ(ierr); ierr = ISGetIndices(zerodiag_subs[s],&idxs);CHKERRQ(ierr); for (i=0;ibenign_change,1,cols,2,cols,vals,INSERT_VALUES);CHKERRQ(ierr); } ierr = PetscMalloc1(nzs,&array);CHKERRQ(ierr); for (i=0;ibenign_change,1,idxs+nzs-1,nzs,idxs,array,INSERT_VALUES);CHKERRQ(ierr); /* store local idxs for p0 */ pcbddc->benign_p0_lidx[s] = idxs[nzs-1]; ierr = ISRestoreIndices(zerodiag_subs[s],&idxs);CHKERRQ(ierr); ierr = PetscFree(array);CHKERRQ(ierr); } ierr = MatAssemblyBegin(pcbddc->benign_change,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); ierr = MatAssemblyEnd(pcbddc->benign_change,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); /* project if needed */ if (pcbddc->benign_change_explicit) { Mat M; ierr = MatPtAP(pcbddc->local_mat,pcbddc->benign_change,MAT_INITIAL_MATRIX,2.0,&M);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); ierr = MatSeqAIJCompress(M,&pcbddc->local_mat);CHKERRQ(ierr); ierr = MatDestroy(&M);CHKERRQ(ierr); } /* store global idxs for p0 */ ierr = ISLocalToGlobalMappingApply(pc->pmat->rmap->mapping,pcbddc->benign_n,pcbddc->benign_p0_lidx,pcbddc->benign_p0_gidx);CHKERRQ(ierr); } pcbddc->benign_zerodiag_subs = zerodiag_subs; ierr = ISDestroy(&pressures);CHKERRQ(ierr); /* determines if the coarse solver will be singular or not */ ierr = MPI_Allreduce(&have_null,&pcbddc->benign_null,1,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); /* determines if the problem has subdomains with 0 pressure block */ ierr = MPI_Allreduce(&have_null,&pcbddc->benign_have_null,1,MPIU_BOOL,MPI_LOR,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); *zerodiaglocal = zerodiag; PetscFunctionReturn(0); } PetscErrorCode PCBDDCBenignGetOrSetP0(PC pc, Vec v, PetscBool get) { PC_BDDC* pcbddc = (PC_BDDC*)pc->data; PetscScalar *array; PetscErrorCode ierr; PetscFunctionBegin; if (!pcbddc->benign_sf) { ierr = PetscSFCreate(PetscObjectComm((PetscObject)pc),&pcbddc->benign_sf);CHKERRQ(ierr); ierr = PetscSFSetGraphLayout(pcbddc->benign_sf,pc->pmat->rmap,pcbddc->benign_n,NULL,PETSC_OWN_POINTER,pcbddc->benign_p0_gidx);CHKERRQ(ierr); } if (get) { ierr = VecGetArrayRead(v,(const PetscScalar**)&array);CHKERRQ(ierr); ierr = PetscSFBcastBegin(pcbddc->benign_sf,MPIU_SCALAR,array,pcbddc->benign_p0);CHKERRQ(ierr); ierr = PetscSFBcastEnd(pcbddc->benign_sf,MPIU_SCALAR,array,pcbddc->benign_p0);CHKERRQ(ierr); ierr = VecRestoreArrayRead(v,(const PetscScalar**)&array);CHKERRQ(ierr); } else { ierr = VecGetArray(v,&array);CHKERRQ(ierr); ierr = PetscSFReduceBegin(pcbddc->benign_sf,MPIU_SCALAR,pcbddc->benign_p0,array,MPIU_REPLACE);CHKERRQ(ierr); ierr = PetscSFReduceEnd(pcbddc->benign_sf,MPIU_SCALAR,pcbddc->benign_p0,array,MPIU_REPLACE);CHKERRQ(ierr); ierr = VecRestoreArray(v,&array);CHKERRQ(ierr); } PetscFunctionReturn(0); } PetscErrorCode PCBDDCBenignPopOrPushB0(PC pc, PetscBool pop) { PC_BDDC* pcbddc = (PC_BDDC*)pc->data; PetscErrorCode ierr; PetscFunctionBegin; /* TODO: add error checking - avoid nested pop (or push) calls. - cannot push before pop. - cannot call this if pcbddc->local_mat is NULL */ if (!pcbddc->benign_n) { PetscFunctionReturn(0); } if (pop) { if (pcbddc->benign_change_explicit) { IS is_p0; MatReuse reuse; /* extract B_0 */ reuse = MAT_INITIAL_MATRIX; if (pcbddc->benign_B0) { reuse = MAT_REUSE_MATRIX; } ierr = ISCreateGeneral(PETSC_COMM_SELF,pcbddc->benign_n,pcbddc->benign_p0_lidx,PETSC_COPY_VALUES,&is_p0);CHKERRQ(ierr); ierr = MatCreateSubMatrix(pcbddc->local_mat,is_p0,NULL,reuse,&pcbddc->benign_B0);CHKERRQ(ierr); /* remove rows and cols from local problem */ ierr = MatSetOption(pcbddc->local_mat,MAT_KEEP_NONZERO_PATTERN,PETSC_TRUE);CHKERRQ(ierr); ierr = MatSetOption(pcbddc->local_mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_FALSE);CHKERRQ(ierr); ierr = MatZeroRowsColumnsIS(pcbddc->local_mat,is_p0,1.0,NULL,NULL);CHKERRQ(ierr); ierr = ISDestroy(&is_p0);CHKERRQ(ierr); } else { Mat_IS *matis = (Mat_IS*)pc->pmat->data; PetscScalar *vals; PetscInt i,n,*idxs_ins; ierr = VecGetLocalSize(matis->y,&n);CHKERRQ(ierr); ierr = PetscMalloc2(n,&idxs_ins,n,&vals);CHKERRQ(ierr); if (!pcbddc->benign_B0) { PetscInt *nnz; ierr = MatCreate(PetscObjectComm((PetscObject)pcbddc->local_mat),&pcbddc->benign_B0);CHKERRQ(ierr); ierr = MatSetType(pcbddc->benign_B0,MATAIJ);CHKERRQ(ierr); ierr = MatSetSizes(pcbddc->benign_B0,pcbddc->benign_n,n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); ierr = PetscMalloc1(pcbddc->benign_n,&nnz);CHKERRQ(ierr); for (i=0;ibenign_n;i++) { ierr = ISGetLocalSize(pcbddc->benign_zerodiag_subs[i],&nnz[i]);CHKERRQ(ierr); nnz[i] = n - nnz[i]; } ierr = MatSeqAIJSetPreallocation(pcbddc->benign_B0,0,nnz);CHKERRQ(ierr); ierr = PetscFree(nnz);CHKERRQ(ierr); } for (i=0;ibenign_n;i++) { PetscScalar *array; PetscInt *idxs,j,nz,cum; ierr = VecSet(matis->x,0.);CHKERRQ(ierr); ierr = ISGetLocalSize(pcbddc->benign_zerodiag_subs[i],&nz);CHKERRQ(ierr); ierr = ISGetIndices(pcbddc->benign_zerodiag_subs[i],(const PetscInt**)&idxs);CHKERRQ(ierr); for (j=0;jx,nz,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); ierr = VecAssemblyBegin(matis->x);CHKERRQ(ierr); ierr = VecAssemblyEnd(matis->x);CHKERRQ(ierr); ierr = VecSet(matis->y,0.);CHKERRQ(ierr); ierr = MatMult(matis->A,matis->x,matis->y);CHKERRQ(ierr); ierr = VecGetArray(matis->y,&array);CHKERRQ(ierr); cum = 0; for (j=0;j PETSC_SMALL)) { vals[cum] = array[j]; idxs_ins[cum] = j; cum++; } } ierr = MatSetValues(pcbddc->benign_B0,1,&i,cum,idxs_ins,vals,INSERT_VALUES);CHKERRQ(ierr); ierr = VecRestoreArray(matis->y,&array);CHKERRQ(ierr); ierr = ISRestoreIndices(pcbddc->benign_zerodiag_subs[i],(const PetscInt**)&idxs);CHKERRQ(ierr); } ierr = MatAssemblyBegin(pcbddc->benign_B0,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); ierr = MatAssemblyEnd(pcbddc->benign_B0,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); ierr = PetscFree2(idxs_ins,vals);CHKERRQ(ierr); } } else { /* push */ if (pcbddc->benign_change_explicit) { PetscInt i; for (i=0;ibenign_n;i++) { PetscScalar *B0_vals; PetscInt *B0_cols,B0_ncol; ierr = MatGetRow(pcbddc->benign_B0,i,&B0_ncol,(const PetscInt**)&B0_cols,(const PetscScalar**)&B0_vals);CHKERRQ(ierr); ierr = MatSetValues(pcbddc->local_mat,1,pcbddc->benign_p0_lidx+i,B0_ncol,B0_cols,B0_vals,INSERT_VALUES);CHKERRQ(ierr); ierr = MatSetValues(pcbddc->local_mat,B0_ncol,B0_cols,1,pcbddc->benign_p0_lidx+i,B0_vals,INSERT_VALUES);CHKERRQ(ierr); ierr = MatSetValue(pcbddc->local_mat,pcbddc->benign_p0_lidx[i],pcbddc->benign_p0_lidx[i],0.0,INSERT_VALUES);CHKERRQ(ierr); ierr = MatRestoreRow(pcbddc->benign_B0,i,&B0_ncol,(const PetscInt**)&B0_cols,(const PetscScalar**)&B0_vals);CHKERRQ(ierr); } ierr = MatAssemblyBegin(pcbddc->local_mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); ierr = MatAssemblyEnd(pcbddc->local_mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); } else { SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Cannot push B0!\n"); } } PetscFunctionReturn(0); } PetscErrorCode PCBDDCAdaptiveSelection(PC pc) { PC_BDDC* pcbddc = (PC_BDDC*)pc->data; PCBDDCSubSchurs sub_schurs = pcbddc->sub_schurs; PetscBLASInt B_dummyint,B_neigs,B_ierr,B_lwork; PetscBLASInt *B_iwork,*B_ifail; PetscScalar *work,lwork; PetscScalar *St,*S,*eigv; PetscScalar *Sarray,*Starray; PetscReal *eigs,thresh; PetscInt i,nmax,nmin,nv,cum,mss,cum2,cumarray,maxneigs; PetscBool allocated_S_St; #if defined(PETSC_USE_COMPLEX) PetscReal *rwork; #endif PetscErrorCode ierr; PetscFunctionBegin; if (!sub_schurs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Adaptive selection of constraints requires SubSchurs data"); if (!sub_schurs->schur_explicit) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Adaptive selection of constraints requires MUMPS and/or MKL_CPARDISO"); if (sub_schurs->n_subs && (!sub_schurs->is_hermitian || !sub_schurs->is_posdef)) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Adaptive selection not yet implemented for general matrix pencils (herm %d, posdef %d)\nRerun with -sub_schurs_hermitian 1 -sub_schurs_posdef 1 if the problem is SPD",sub_schurs->is_hermitian,sub_schurs->is_posdef); if (pcbddc->dbg_flag) { ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"--------------------------------------------------\n");CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"Check adaptive selection of constraints\n");CHKERRQ(ierr); ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); } if (pcbddc->dbg_flag) { PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d cc %d (%d,%d).\n",PetscGlobalRank,sub_schurs->n_subs,sub_schurs->is_hermitian,sub_schurs->is_posdef); } /* max size of subsets */ mss = 0; for (i=0;in_subs;i++) { PetscInt subset_size; ierr = ISGetLocalSize(sub_schurs->is_subs[i],&subset_size);CHKERRQ(ierr); mss = PetscMax(mss,subset_size); } /* min/max and threshold */ nmax = pcbddc->adaptive_nmax > 0 ? pcbddc->adaptive_nmax : mss; nmin = pcbddc->adaptive_nmin > 0 ? pcbddc->adaptive_nmin : 0; nmax = PetscMax(nmin,nmax); allocated_S_St = PETSC_FALSE; if (nmin) { allocated_S_St = PETSC_TRUE; } /* allocate lapack workspace */ cum = cum2 = 0; maxneigs = 0; for (i=0;in_subs;i++) { PetscInt n,subset_size; ierr = ISGetLocalSize(sub_schurs->is_subs[i],&subset_size);CHKERRQ(ierr); n = PetscMin(subset_size,nmax); cum += subset_size; cum2 += subset_size*n; maxneigs = PetscMax(maxneigs,n); } if (mss) { if (sub_schurs->is_hermitian && sub_schurs->is_posdef) { PetscBLASInt B_itype = 1; PetscBLASInt B_N = mss; PetscReal zero = 0.0; PetscReal eps = 0.0; /* dlamch? */ B_lwork = -1; S = NULL; St = NULL; eigs = NULL; eigv = NULL; B_iwork = NULL; B_ifail = NULL; #if defined(PETSC_USE_COMPLEX) rwork = NULL; #endif thresh = 1.0; ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); #if defined(PETSC_USE_COMPLEX) PetscStackCallBLAS("LAPACKsygvx",LAPACKsygvx_(&B_itype,"V","V","L",&B_N,St,&B_N,S,&B_N,&zero,&thresh,&B_dummyint,&B_dummyint,&eps,&B_neigs,eigs,eigv,&B_N,&lwork,&B_lwork,rwork,B_iwork,B_ifail,&B_ierr)); #else PetscStackCallBLAS("LAPACKsygvx",LAPACKsygvx_(&B_itype,"V","V","L",&B_N,St,&B_N,S,&B_N,&zero,&thresh,&B_dummyint,&B_dummyint,&eps,&B_neigs,eigs,eigv,&B_N,&lwork,&B_lwork,B_iwork,B_ifail,&B_ierr)); #endif if (B_ierr != 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in query to SYGVX Lapack routine %d",(int)B_ierr); ierr = PetscFPTrapPop();CHKERRQ(ierr); } else { /* TODO */ } } else { lwork = 0; } nv = 0; if (sub_schurs->is_vertices && pcbddc->use_vertices) { /* complement set of active subsets, each entry is a vertex (boundary made by active subsets, vertices and dirichlet dofs) */ ierr = ISGetLocalSize(sub_schurs->is_vertices,&nv);CHKERRQ(ierr); } ierr = PetscBLASIntCast((PetscInt)PetscRealPart(lwork),&B_lwork);CHKERRQ(ierr); if (allocated_S_St) { ierr = PetscMalloc2(mss*mss,&S,mss*mss,&St);CHKERRQ(ierr); } ierr = PetscMalloc5(mss*mss,&eigv,mss,&eigs,B_lwork,&work,5*mss,&B_iwork,mss,&B_ifail);CHKERRQ(ierr); #if defined(PETSC_USE_COMPLEX) ierr = PetscMalloc1(7*mss,&rwork);CHKERRQ(ierr); #endif ierr = PetscMalloc5(nv+sub_schurs->n_subs,&pcbddc->adaptive_constraints_n, nv+sub_schurs->n_subs+1,&pcbddc->adaptive_constraints_idxs_ptr, nv+sub_schurs->n_subs+1,&pcbddc->adaptive_constraints_data_ptr, nv+cum,&pcbddc->adaptive_constraints_idxs, nv+cum2,&pcbddc->adaptive_constraints_data);CHKERRQ(ierr); ierr = PetscMemzero(pcbddc->adaptive_constraints_n,(nv+sub_schurs->n_subs)*sizeof(PetscInt));CHKERRQ(ierr); maxneigs = 0; cum = cumarray = 0; pcbddc->adaptive_constraints_idxs_ptr[0] = 0; pcbddc->adaptive_constraints_data_ptr[0] = 0; if (sub_schurs->is_vertices && pcbddc->use_vertices) { const PetscInt *idxs; ierr = ISGetIndices(sub_schurs->is_vertices,&idxs);CHKERRQ(ierr); for (cum=0;cumadaptive_constraints_n[cum] = 1; pcbddc->adaptive_constraints_idxs[cum] = idxs[cum]; pcbddc->adaptive_constraints_data[cum] = 1.0; pcbddc->adaptive_constraints_idxs_ptr[cum+1] = pcbddc->adaptive_constraints_idxs_ptr[cum]+1; pcbddc->adaptive_constraints_data_ptr[cum+1] = pcbddc->adaptive_constraints_data_ptr[cum]+1; } ierr = ISRestoreIndices(sub_schurs->is_vertices,&idxs);CHKERRQ(ierr); } if (mss) { /* multilevel */ ierr = MatSeqAIJGetArray(sub_schurs->sum_S_Ej_inv_all,&Sarray);CHKERRQ(ierr); ierr = MatSeqAIJGetArray(sub_schurs->sum_S_Ej_tilda_all,&Starray);CHKERRQ(ierr); } thresh = pcbddc->adaptive_threshold; for (i=0;in_subs;i++) { const PetscInt *idxs; PetscReal upper,lower; PetscInt j,subset_size,eigs_start = 0; PetscBLASInt B_N; PetscBool same_data = PETSC_FALSE; if (pcbddc->use_deluxe_scaling) { upper = PETSC_MAX_REAL; lower = thresh; } else { upper = 1./thresh; lower = 0.; } ierr = ISGetLocalSize(sub_schurs->is_subs[i],&subset_size);CHKERRQ(ierr); ierr = ISGetIndices(sub_schurs->is_subs[i],&idxs);CHKERRQ(ierr); ierr = PetscBLASIntCast(subset_size,&B_N);CHKERRQ(ierr); if (allocated_S_St) { /* S and S_t should be copied since we could need them later */ if (sub_schurs->is_hermitian) { PetscInt j,k; if (sub_schurs->n_subs == 1) { /* zeroing memory to use PetscMemcmp later */ ierr = PetscMemzero(S,subset_size*subset_size*sizeof(PetscScalar));CHKERRQ(ierr); ierr = PetscMemzero(St,subset_size*subset_size*sizeof(PetscScalar));CHKERRQ(ierr); } for (j=0;jn_subs == 1 && pcbddc->use_deluxe_scaling) { ierr = PetscMemcmp(S,St,subset_size*subset_size*sizeof(PetscScalar),&same_data);CHKERRQ(ierr); } if (same_data && !sub_schurs->change) { /* there's no need of constraints here */ B_neigs = 0; } else { if (sub_schurs->is_hermitian && sub_schurs->is_posdef) { PetscBLASInt B_itype = 1; PetscBLASInt B_IL, B_IU; PetscReal eps = -1.0; /* dlamch? */ PetscInt nmin_s; PetscBool compute_range = PETSC_FALSE; if (pcbddc->dbg_flag) { PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Computing for sub %d/%d size %d count %d fid %d.\n",i,sub_schurs->n_subs,subset_size,pcbddc->mat_graph->count[idxs[0]]+1,pcbddc->mat_graph->which_dof[idxs[0]]); } compute_range = PETSC_FALSE; if (thresh > 1.+PETSC_SMALL && !same_data) { compute_range = PETSC_TRUE; } ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); if (compute_range) { /* ask for eigenvalues larger than thresh */ #if defined(PETSC_USE_COMPLEX) PetscStackCallBLAS("LAPACKsygvx",LAPACKsygvx_(&B_itype,"V","V","L",&B_N,St,&B_N,S,&B_N,&lower,&upper,&B_IL,&B_IU,&eps,&B_neigs,eigs,eigv,&B_N,work,&B_lwork,rwork,B_iwork,B_ifail,&B_ierr)); #else PetscStackCallBLAS("LAPACKsygvx",LAPACKsygvx_(&B_itype,"V","V","L",&B_N,St,&B_N,S,&B_N,&lower,&upper,&B_IL,&B_IU,&eps,&B_neigs,eigs,eigv,&B_N,work,&B_lwork,B_iwork,B_ifail,&B_ierr)); #endif } else if (!same_data) { B_IU = PetscMax(1,PetscMin(B_N,nmax)); B_IL = 1; #if defined(PETSC_USE_COMPLEX) PetscStackCallBLAS("LAPACKsygvx",LAPACKsygvx_(&B_itype,"V","I","L",&B_N,St,&B_N,S,&B_N,&lower,&upper,&B_IL,&B_IU,&eps,&B_neigs,eigs,eigv,&B_N,work,&B_lwork,rwork,B_iwork,B_ifail,&B_ierr)); #else PetscStackCallBLAS("LAPACKsygvx",LAPACKsygvx_(&B_itype,"V","I","L",&B_N,St,&B_N,S,&B_N,&lower,&upper,&B_IL,&B_IU,&eps,&B_neigs,eigs,eigv,&B_N,work,&B_lwork,B_iwork,B_ifail,&B_ierr)); #endif } else { /* same_data is true, so just get the adaptive functional requested by the user */ PetscInt k; if (!sub_schurs->change_primal_sub) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"This should not happen"); ierr = ISGetLocalSize(sub_schurs->change_primal_sub[i],&nmax);CHKERRQ(ierr); ierr = PetscBLASIntCast(nmax,&B_neigs);CHKERRQ(ierr); nmin = nmax; ierr = PetscMemzero(eigv,subset_size*nmax*sizeof(PetscScalar));CHKERRQ(ierr); for (k=0;k nmax) { if (pcbddc->dbg_flag) { PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer," found %d eigs, more than maximum required %d.\n",B_neigs,nmax); } if (pcbddc->use_deluxe_scaling) eigs_start = B_neigs -nmax; B_neigs = nmax; } nmin_s = PetscMin(nmin,B_N); if (B_neigs < nmin_s) { PetscBLASInt B_neigs2; if (pcbddc->use_deluxe_scaling) { B_IL = B_N - nmin_s + 1; B_IU = B_N - B_neigs; } else { B_IL = B_neigs + 1; B_IU = nmin_s; } if (pcbddc->dbg_flag) { PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer," found %d eigs, less than minimum required %d. Asking for %d to %d incl (fortran like)\n",B_neigs,nmin,B_IL,B_IU); } if (sub_schurs->is_hermitian) { PetscInt j,k; for (j=0;jdbg_flag) { ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer," -> Got %d eigs\n",B_neigs);CHKERRQ(ierr); for (j=0;jdbg_viewer," Inf\n");CHKERRQ(ierr); } else { if (pcbddc->use_deluxe_scaling) { ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer," %1.6e\n",eigs[j+eigs_start]);CHKERRQ(ierr); } else { ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer," %1.6e\n",1./eigs[j+eigs_start]);CHKERRQ(ierr); } } } } } else { /* TODO */ } } /* change the basis back to the original one */ if (sub_schurs->change) { Mat change,phi,phit; if (pcbddc->dbg_flag > 1) { PetscInt ii; for (ii=0;iidbg_viewer," -> Eigenvector (old basis) %d/%d (%d)\n",ii,B_neigs,B_N);CHKERRQ(ierr); for (j=0;jdbg_viewer," %1.4e + %1.4e i\n",r,c);CHKERRQ(ierr); #else ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer," %1.4e\n",eigv[(ii+eigs_start)*subset_size+j]);CHKERRQ(ierr); #endif } } } ierr = KSPGetOperators(sub_schurs->change[i],&change,NULL);CHKERRQ(ierr); ierr = MatCreateSeqDense(PETSC_COMM_SELF,subset_size,B_neigs,eigv+eigs_start*subset_size,&phit);CHKERRQ(ierr); ierr = MatMatMult(change,phit,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&phi);CHKERRQ(ierr); ierr = MatCopy(phi,phit,SAME_NONZERO_PATTERN);CHKERRQ(ierr); ierr = MatDestroy(&phit);CHKERRQ(ierr); ierr = MatDestroy(&phi);CHKERRQ(ierr); } maxneigs = PetscMax(B_neigs,maxneigs); pcbddc->adaptive_constraints_n[i+nv] = B_neigs; if (B_neigs) { ierr = PetscMemcpy(pcbddc->adaptive_constraints_data+pcbddc->adaptive_constraints_data_ptr[cum],eigv+eigs_start*subset_size,B_neigs*subset_size*sizeof(PetscScalar));CHKERRQ(ierr); if (pcbddc->dbg_flag > 1) { PetscInt ii; for (ii=0;iidbg_viewer," -> Eigenvector %d/%d (%d)\n",ii,B_neigs,B_N);CHKERRQ(ierr); for (j=0;jadaptive_constraints_data[ii*subset_size+j+pcbddc->adaptive_constraints_data_ptr[cum]]); PetscReal c = PetscImaginaryPart(pcbddc->adaptive_constraints_data[ii*subset_size+j+pcbddc->adaptive_constraints_data_ptr[cum]]); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer," %1.4e + %1.4e i\n",r,c);CHKERRQ(ierr); #else ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer," %1.4e\n",pcbddc->adaptive_constraints_data[ii*subset_size+j+pcbddc->adaptive_constraints_data_ptr[cum]]);CHKERRQ(ierr); #endif } } } ierr = PetscMemcpy(pcbddc->adaptive_constraints_idxs+pcbddc->adaptive_constraints_idxs_ptr[cum],idxs,subset_size*sizeof(PetscInt));CHKERRQ(ierr); pcbddc->adaptive_constraints_idxs_ptr[cum+1] = pcbddc->adaptive_constraints_idxs_ptr[cum] + subset_size; pcbddc->adaptive_constraints_data_ptr[cum+1] = pcbddc->adaptive_constraints_data_ptr[cum] + subset_size*B_neigs; cum++; } ierr = ISRestoreIndices(sub_schurs->is_subs[i],&idxs);CHKERRQ(ierr); /* shift for next computation */ cumarray += subset_size*subset_size; } if (pcbddc->dbg_flag) { ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); } if (mss) { ierr = MatSeqAIJRestoreArray(sub_schurs->sum_S_Ej_inv_all,&Sarray);CHKERRQ(ierr); ierr = MatSeqAIJRestoreArray(sub_schurs->sum_S_Ej_tilda_all,&Starray);CHKERRQ(ierr); /* destroy matrices (junk) */ ierr = MatDestroy(&sub_schurs->sum_S_Ej_inv_all);CHKERRQ(ierr); ierr = MatDestroy(&sub_schurs->sum_S_Ej_tilda_all);CHKERRQ(ierr); } if (allocated_S_St) { ierr = PetscFree2(S,St);CHKERRQ(ierr); } ierr = PetscFree5(eigv,eigs,work,B_iwork,B_ifail);CHKERRQ(ierr); #if defined(PETSC_USE_COMPLEX) ierr = PetscFree(rwork);CHKERRQ(ierr); #endif if (pcbddc->dbg_flag) { PetscInt maxneigs_r; ierr = MPIU_Allreduce(&maxneigs,&maxneigs_r,1,MPIU_INT,MPI_MAX,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"Maximum number of constraints per cc %d\n",maxneigs_r);CHKERRQ(ierr); } PetscFunctionReturn(0); } PetscErrorCode PCBDDCSetUpSolvers(PC pc) { PetscScalar *coarse_submat_vals; PetscErrorCode ierr; PetscFunctionBegin; /* Setup local scatters R_to_B and (optionally) R_to_D */ /* PCBDDCSetUpLocalWorkVectors should be called first! */ ierr = PCBDDCSetUpLocalScatters(pc);CHKERRQ(ierr); /* Setup local neumann solver ksp_R */ /* PCBDDCSetUpLocalScatters should be called first! */ ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_FALSE,PETSC_TRUE);CHKERRQ(ierr); /* Setup local correction and local part of coarse basis. Gives back the dense local part of the coarse matrix in column major ordering */ ierr = PCBDDCSetUpCorrection(pc,&coarse_submat_vals);CHKERRQ(ierr); /* Compute total number of coarse nodes and setup coarse solver */ ierr = PCBDDCSetUpCoarseSolver(pc,coarse_submat_vals);CHKERRQ(ierr); /* free */ ierr = PetscFree(coarse_submat_vals);CHKERRQ(ierr); PetscFunctionReturn(0); } PetscErrorCode PCBDDCResetCustomization(PC pc) { PC_BDDC *pcbddc = (PC_BDDC*)pc->data; PetscErrorCode ierr; PetscFunctionBegin; ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); ierr = MatNullSpaceDestroy(&pcbddc->onearnullspace);CHKERRQ(ierr); ierr = PetscFree(pcbddc->onearnullvecs_state);CHKERRQ(ierr); ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); ierr = PCBDDCSetDofsSplitting(pc,0,NULL);CHKERRQ(ierr); ierr = PCBDDCSetDofsSplittingLocal(pc,0,NULL);CHKERRQ(ierr); PetscFunctionReturn(0); } PetscErrorCode PCBDDCResetTopography(PC pc) { PC_BDDC *pcbddc = (PC_BDDC*)pc->data; PetscInt i; PetscErrorCode ierr; PetscFunctionBegin; ierr = MatDestroy(&pcbddc->nedcG);CHKERRQ(ierr); ierr = ISDestroy(&pcbddc->nedclocal);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->discretegradient);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->switch_static_change);CHKERRQ(ierr); ierr = VecDestroy(&pcbddc->work_change);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->ConstraintMatrix);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->divudotp);CHKERRQ(ierr); ierr = ISDestroy(&pcbddc->divudotp_vl2l);CHKERRQ(ierr); ierr = PCBDDCGraphDestroy(&pcbddc->mat_graph);CHKERRQ(ierr); for (i=0;in_local_subs;i++) { ierr = ISDestroy(&pcbddc->local_subs[i]);CHKERRQ(ierr); } pcbddc->n_local_subs = 0; ierr = PetscFree(pcbddc->local_subs);CHKERRQ(ierr); ierr = PCBDDCSubSchursDestroy(&pcbddc->sub_schurs);CHKERRQ(ierr); pcbddc->graphanalyzed = PETSC_FALSE; pcbddc->recompute_topography = PETSC_TRUE; PetscFunctionReturn(0); } PetscErrorCode PCBDDCResetSolvers(PC pc) { PC_BDDC *pcbddc = (PC_BDDC*)pc->data; PetscErrorCode ierr; PetscFunctionBegin; ierr = VecDestroy(&pcbddc->coarse_vec);CHKERRQ(ierr); if (pcbddc->coarse_phi_B) { PetscScalar *array; ierr = MatDenseGetArray(pcbddc->coarse_phi_B,&array);CHKERRQ(ierr); ierr = PetscFree(array);CHKERRQ(ierr); } ierr = MatDestroy(&pcbddc->coarse_phi_B);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->coarse_phi_D);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->coarse_psi_B);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->coarse_psi_D);CHKERRQ(ierr); ierr = VecDestroy(&pcbddc->vec1_P);CHKERRQ(ierr); ierr = VecDestroy(&pcbddc->vec1_C);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->local_auxmat2);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->local_auxmat1);CHKERRQ(ierr); ierr = VecDestroy(&pcbddc->vec1_R);CHKERRQ(ierr); ierr = VecDestroy(&pcbddc->vec2_R);CHKERRQ(ierr); ierr = ISDestroy(&pcbddc->is_R_local);CHKERRQ(ierr); ierr = VecScatterDestroy(&pcbddc->R_to_B);CHKERRQ(ierr); ierr = VecScatterDestroy(&pcbddc->R_to_D);CHKERRQ(ierr); ierr = VecScatterDestroy(&pcbddc->coarse_loc_to_glob);CHKERRQ(ierr); ierr = KSPReset(pcbddc->ksp_D);CHKERRQ(ierr); ierr = KSPReset(pcbddc->ksp_R);CHKERRQ(ierr); ierr = KSPReset(pcbddc->coarse_ksp);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); ierr = PetscFree(pcbddc->primal_indices_local_idxs);CHKERRQ(ierr); ierr = PetscFree2(pcbddc->local_primal_ref_node,pcbddc->local_primal_ref_mult);CHKERRQ(ierr); ierr = PetscFree(pcbddc->global_primal_indices);CHKERRQ(ierr); ierr = ISDestroy(&pcbddc->coarse_subassembling);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->benign_change);CHKERRQ(ierr); ierr = VecDestroy(&pcbddc->benign_vec);CHKERRQ(ierr); ierr = PCBDDCBenignShellMat(pc,PETSC_TRUE);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->benign_B0);CHKERRQ(ierr); ierr = PetscSFDestroy(&pcbddc->benign_sf);CHKERRQ(ierr); if (pcbddc->benign_zerodiag_subs) { PetscInt i; for (i=0;ibenign_n;i++) { ierr = ISDestroy(&pcbddc->benign_zerodiag_subs[i]);CHKERRQ(ierr); } ierr = PetscFree(pcbddc->benign_zerodiag_subs);CHKERRQ(ierr); } ierr = PetscFree3(pcbddc->benign_p0_lidx,pcbddc->benign_p0_gidx,pcbddc->benign_p0);CHKERRQ(ierr); PetscFunctionReturn(0); } PetscErrorCode PCBDDCSetUpLocalWorkVectors(PC pc) { PC_BDDC *pcbddc = (PC_BDDC*)pc->data; PC_IS *pcis = (PC_IS*)pc->data; VecType impVecType; PetscInt n_constraints,n_R,old_size; PetscErrorCode ierr; PetscFunctionBegin; n_constraints = pcbddc->local_primal_size - pcbddc->benign_n - pcbddc->n_vertices; n_R = pcis->n - pcbddc->n_vertices; ierr = VecGetType(pcis->vec1_N,&impVecType);CHKERRQ(ierr); /* local work vectors (try to avoid unneeded work)*/ /* R nodes */ old_size = -1; if (pcbddc->vec1_R) { ierr = VecGetSize(pcbddc->vec1_R,&old_size);CHKERRQ(ierr); } if (n_R != old_size) { ierr = VecDestroy(&pcbddc->vec1_R);CHKERRQ(ierr); ierr = VecDestroy(&pcbddc->vec2_R);CHKERRQ(ierr); ierr = VecCreate(PetscObjectComm((PetscObject)pcis->vec1_N),&pcbddc->vec1_R);CHKERRQ(ierr); ierr = VecSetSizes(pcbddc->vec1_R,PETSC_DECIDE,n_R);CHKERRQ(ierr); ierr = VecSetType(pcbddc->vec1_R,impVecType);CHKERRQ(ierr); ierr = VecDuplicate(pcbddc->vec1_R,&pcbddc->vec2_R);CHKERRQ(ierr); } /* local primal dofs */ old_size = -1; if (pcbddc->vec1_P) { ierr = VecGetSize(pcbddc->vec1_P,&old_size);CHKERRQ(ierr); } if (pcbddc->local_primal_size != old_size) { ierr = VecDestroy(&pcbddc->vec1_P);CHKERRQ(ierr); ierr = VecCreate(PetscObjectComm((PetscObject)pcis->vec1_N),&pcbddc->vec1_P);CHKERRQ(ierr); ierr = VecSetSizes(pcbddc->vec1_P,PETSC_DECIDE,pcbddc->local_primal_size);CHKERRQ(ierr); ierr = VecSetType(pcbddc->vec1_P,impVecType);CHKERRQ(ierr); } /* local explicit constraints */ old_size = -1; if (pcbddc->vec1_C) { ierr = VecGetSize(pcbddc->vec1_C,&old_size);CHKERRQ(ierr); } if (n_constraints && n_constraints != old_size) { ierr = VecDestroy(&pcbddc->vec1_C);CHKERRQ(ierr); ierr = VecCreate(PetscObjectComm((PetscObject)pcis->vec1_N),&pcbddc->vec1_C);CHKERRQ(ierr); ierr = VecSetSizes(pcbddc->vec1_C,PETSC_DECIDE,n_constraints);CHKERRQ(ierr); ierr = VecSetType(pcbddc->vec1_C,impVecType);CHKERRQ(ierr); } PetscFunctionReturn(0); } PetscErrorCode PCBDDCSetUpCorrection(PC pc, PetscScalar **coarse_submat_vals_n) { PetscErrorCode ierr; /* pointers to pcis and pcbddc */ PC_IS* pcis = (PC_IS*)pc->data; PC_BDDC* pcbddc = (PC_BDDC*)pc->data; PCBDDCSubSchurs sub_schurs = pcbddc->sub_schurs; /* submatrices of local problem */ Mat A_RV,A_VR,A_VV,local_auxmat2_R; /* submatrices of local coarse problem */ Mat S_VV,S_CV,S_VC,S_CC; /* working matrices */ Mat C_CR; /* additional working stuff */ PC pc_R; Mat F; Vec dummy_vec; PetscBool isLU,isCHOL,isILU,need_benign_correction; PetscScalar *coarse_submat_vals; /* TODO: use a PETSc matrix */ PetscScalar *work; PetscInt *idx_V_B; PetscInt lda_rhs,n,n_vertices,n_constraints,*p0_lidx_I; PetscInt i,n_R,n_D,n_B; /* some shortcuts to scalars */ PetscScalar one=1.0,m_one=-1.0; PetscFunctionBegin; if (!pcbddc->symmetric_primal && pcbddc->benign_n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Non-symmetric primal basis computation with benign trick not yet implemented"); /* Set Non-overlapping dimensions */ n_vertices = pcbddc->n_vertices; n_constraints = pcbddc->local_primal_size - pcbddc->benign_n - n_vertices; n_B = pcis->n_B; n_D = pcis->n - n_B; n_R = pcis->n - n_vertices; /* vertices in boundary numbering */ ierr = PetscMalloc1(n_vertices,&idx_V_B);CHKERRQ(ierr); ierr = ISGlobalToLocalMappingApply(pcis->BtoNmap,IS_GTOLM_DROP,n_vertices,pcbddc->local_primal_ref_node,&i,idx_V_B);CHKERRQ(ierr); if (i != n_vertices) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Error in boundary numbering for BDDC vertices! %D != %D\n",n_vertices,i); /* Subdomain contribution (Non-overlapping) to coarse matrix */ ierr = PetscCalloc1(pcbddc->local_primal_size*pcbddc->local_primal_size,&coarse_submat_vals);CHKERRQ(ierr); ierr = MatCreateSeqDense(PETSC_COMM_SELF,n_vertices,n_vertices,coarse_submat_vals,&S_VV);CHKERRQ(ierr); ierr = MatSeqDenseSetLDA(S_VV,pcbddc->local_primal_size);CHKERRQ(ierr); ierr = MatCreateSeqDense(PETSC_COMM_SELF,n_constraints,n_vertices,coarse_submat_vals+n_vertices,&S_CV);CHKERRQ(ierr); ierr = MatSeqDenseSetLDA(S_CV,pcbddc->local_primal_size);CHKERRQ(ierr); ierr = MatCreateSeqDense(PETSC_COMM_SELF,n_vertices,n_constraints,coarse_submat_vals+pcbddc->local_primal_size*n_vertices,&S_VC);CHKERRQ(ierr); ierr = MatSeqDenseSetLDA(S_VC,pcbddc->local_primal_size);CHKERRQ(ierr); ierr = MatCreateSeqDense(PETSC_COMM_SELF,n_constraints,n_constraints,coarse_submat_vals+(pcbddc->local_primal_size+1)*n_vertices,&S_CC);CHKERRQ(ierr); ierr = MatSeqDenseSetLDA(S_CC,pcbddc->local_primal_size);CHKERRQ(ierr); /* determine if can use MatSolve routines instead of calling KSPSolve on ksp_R */ ierr = KSPGetPC(pcbddc->ksp_R,&pc_R);CHKERRQ(ierr); ierr = PetscObjectTypeCompare((PetscObject)pc_R,PCLU,&isLU);CHKERRQ(ierr); ierr = PetscObjectTypeCompare((PetscObject)pc_R,PCILU,&isILU);CHKERRQ(ierr); ierr = PetscObjectTypeCompare((PetscObject)pc_R,PCCHOLESKY,&isCHOL);CHKERRQ(ierr); lda_rhs = n_R; need_benign_correction = PETSC_FALSE; if (isLU || isILU || isCHOL) { ierr = PCFactorGetMatrix(pc_R,&F);CHKERRQ(ierr); } else if (sub_schurs && sub_schurs->reuse_solver) { PCBDDCReuseSolvers reuse_solver = sub_schurs->reuse_solver; MatFactorType type; F = reuse_solver->F; ierr = MatGetFactorType(F,&type);CHKERRQ(ierr); if (type == MAT_FACTOR_CHOLESKY) isCHOL = PETSC_TRUE; ierr = MatGetSize(F,&lda_rhs,NULL);CHKERRQ(ierr); need_benign_correction = (PetscBool)(!!reuse_solver->benign_n); } else { F = NULL; } /* allocate workspace */ n = 0; if (n_constraints) { n += lda_rhs*n_constraints; } if (n_vertices) { n = PetscMax(2*lda_rhs*n_vertices,n); n = PetscMax((lda_rhs+n_B)*n_vertices,n); } if (!pcbddc->symmetric_primal) { n = PetscMax(2*lda_rhs*pcbddc->local_primal_size,n); } ierr = PetscMalloc1(n,&work);CHKERRQ(ierr); /* create dummy vector to modify rhs and sol of MatMatSolve (work array will never be used) */ dummy_vec = NULL; if (need_benign_correction && lda_rhs != n_R && F) { ierr = VecCreateSeqWithArray(PETSC_COMM_SELF,1,lda_rhs,work,&dummy_vec);CHKERRQ(ierr); } /* Precompute stuffs needed for preprocessing and application of BDDC*/ if (n_constraints) { Mat M1,M2,M3,C_B; IS is_aux; PetscScalar *array,*array2; ierr = MatDestroy(&pcbddc->local_auxmat1);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->local_auxmat2);CHKERRQ(ierr); /* Extract constraints on R nodes: C_{CR} */ ierr = ISCreateStride(PETSC_COMM_SELF,n_constraints,n_vertices,1,&is_aux);CHKERRQ(ierr); ierr = MatCreateSubMatrix(pcbddc->ConstraintMatrix,is_aux,pcbddc->is_R_local,MAT_INITIAL_MATRIX,&C_CR);CHKERRQ(ierr); ierr = MatCreateSubMatrix(pcbddc->ConstraintMatrix,is_aux,pcis->is_B_local,MAT_INITIAL_MATRIX,&C_B);CHKERRQ(ierr); /* Assemble local_auxmat2_R = (- A_{RR}^{-1} C^T_{CR}) needed by BDDC setup */ /* Assemble pcbddc->local_auxmat2 = R_to_B (- A_{RR}^{-1} C^T_{CR}) needed by BDDC application */ ierr = PetscMemzero(work,lda_rhs*n_constraints*sizeof(PetscScalar));CHKERRQ(ierr); for (i=0;ireuse_solver; /* rhs is already zero on interior dofs, no need to change the rhs */ ierr = PetscMemzero(reuse_solver->benign_save_vals,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); } ierr = MatMatSolve(F,B,local_auxmat2_R);CHKERRQ(ierr); if (need_benign_correction) { PetscScalar *marr; PCBDDCReuseSolvers reuse_solver = sub_schurs->reuse_solver; ierr = MatDenseGetArray(local_auxmat2_R,&marr);CHKERRQ(ierr); if (lda_rhs != n_R) { for (i=0;ivec1_R,marr+i*lda_rhs);CHKERRQ(ierr); ierr = PCBDDCReuseSolversBenignAdapt(reuse_solver,pcbddc->vec1_R,NULL,PETSC_TRUE,PETSC_TRUE);CHKERRQ(ierr); ierr = VecResetArray(pcbddc->vec1_R);CHKERRQ(ierr); } } ierr = MatDenseRestoreArray(local_auxmat2_R,&marr);CHKERRQ(ierr); } ierr = MatDestroy(&B);CHKERRQ(ierr); } else { PetscScalar *marr; ierr = MatDenseGetArray(local_auxmat2_R,&marr);CHKERRQ(ierr); for (i=0;ivec1_R,work+i*lda_rhs);CHKERRQ(ierr); ierr = VecPlaceArray(pcbddc->vec2_R,marr+i*lda_rhs);CHKERRQ(ierr); ierr = KSPSolve(pcbddc->ksp_R,pcbddc->vec1_R,pcbddc->vec2_R);CHKERRQ(ierr); ierr = VecResetArray(pcbddc->vec1_R);CHKERRQ(ierr); ierr = VecResetArray(pcbddc->vec2_R);CHKERRQ(ierr); } ierr = MatDenseRestoreArray(local_auxmat2_R,&marr);CHKERRQ(ierr); } if (!pcbddc->switch_static) { ierr = MatCreateSeqDense(PETSC_COMM_SELF,n_B,n_constraints,NULL,&pcbddc->local_auxmat2);CHKERRQ(ierr); ierr = MatDenseGetArray(pcbddc->local_auxmat2,&array);CHKERRQ(ierr); ierr = MatDenseGetArray(local_auxmat2_R,&array2);CHKERRQ(ierr); for (i=0;ivec1_R,array2+i*lda_rhs);CHKERRQ(ierr); ierr = VecPlaceArray(pcis->vec1_B,array+i*n_B);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecResetArray(pcis->vec1_B);CHKERRQ(ierr); ierr = VecResetArray(pcbddc->vec1_R);CHKERRQ(ierr); } ierr = MatDenseRestoreArray(local_auxmat2_R,&array2);CHKERRQ(ierr); ierr = MatDenseRestoreArray(pcbddc->local_auxmat2,&array);CHKERRQ(ierr); ierr = MatMatMult(C_B,pcbddc->local_auxmat2,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&M3);CHKERRQ(ierr); } else { if (lda_rhs != n_R) { IS dummy; ierr = ISCreateStride(PETSC_COMM_SELF,n_R,0,1,&dummy);CHKERRQ(ierr); ierr = MatCreateSubMatrix(local_auxmat2_R,dummy,NULL,MAT_INITIAL_MATRIX,&pcbddc->local_auxmat2);CHKERRQ(ierr); ierr = ISDestroy(&dummy);CHKERRQ(ierr); } else { ierr = PetscObjectReference((PetscObject)local_auxmat2_R);CHKERRQ(ierr); pcbddc->local_auxmat2 = local_auxmat2_R; } ierr = MatMatMult(C_CR,pcbddc->local_auxmat2,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&M3);CHKERRQ(ierr); } ierr = ISDestroy(&is_aux);CHKERRQ(ierr); /* Assemble explicitly S_CC = ( C_{CR} A_{RR}^{-1} C^T_{CR} )^{-1} */ ierr = MatScale(M3,m_one);CHKERRQ(ierr); ierr = MatDuplicate(M3,MAT_DO_NOT_COPY_VALUES,&M1);CHKERRQ(ierr); ierr = MatDuplicate(M3,MAT_DO_NOT_COPY_VALUES,&M2);CHKERRQ(ierr); if (isCHOL) { ierr = MatCholeskyFactor(M3,NULL,NULL);CHKERRQ(ierr); } else { ierr = MatLUFactor(M3,NULL,NULL,NULL);CHKERRQ(ierr); } ierr = VecSet(pcbddc->vec1_C,one);CHKERRQ(ierr); ierr = MatDiagonalSet(M2,pcbddc->vec1_C,INSERT_VALUES);CHKERRQ(ierr); ierr = MatMatSolve(M3,M2,M1);CHKERRQ(ierr); ierr = MatDestroy(&M2);CHKERRQ(ierr); ierr = MatDestroy(&M3);CHKERRQ(ierr); /* Assemble local_auxmat1 = S_CC*C_{CB} needed by BDDC application in KSP and in preproc */ ierr = MatMatMult(M1,C_B,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&pcbddc->local_auxmat1);CHKERRQ(ierr); ierr = MatDestroy(&C_B);CHKERRQ(ierr); ierr = MatCopy(M1,S_CC,SAME_NONZERO_PATTERN);CHKERRQ(ierr); /* S_CC can have a different LDA, MatMatSolve doesn't support it */ ierr = MatDestroy(&M1);CHKERRQ(ierr); } /* Get submatrices from subdomain matrix */ if (n_vertices) { IS is_aux; if (sub_schurs && sub_schurs->reuse_solver) { /* is_R_local is not sorted, ISComplement doesn't like it */ IS tis; ierr = ISDuplicate(pcbddc->is_R_local,&tis);CHKERRQ(ierr); ierr = ISSort(tis);CHKERRQ(ierr); ierr = ISComplement(tis,0,pcis->n,&is_aux);CHKERRQ(ierr); ierr = ISDestroy(&tis);CHKERRQ(ierr); } else { ierr = ISComplement(pcbddc->is_R_local,0,pcis->n,&is_aux);CHKERRQ(ierr); } ierr = MatCreateSubMatrix(pcbddc->local_mat,pcbddc->is_R_local,is_aux,MAT_INITIAL_MATRIX,&A_RV);CHKERRQ(ierr); ierr = MatCreateSubMatrix(pcbddc->local_mat,is_aux,pcbddc->is_R_local,MAT_INITIAL_MATRIX,&A_VR);CHKERRQ(ierr); ierr = MatCreateSubMatrix(pcbddc->local_mat,is_aux,is_aux,MAT_INITIAL_MATRIX,&A_VV);CHKERRQ(ierr); ierr = ISDestroy(&is_aux);CHKERRQ(ierr); } /* Matrix of coarse basis functions (local) */ if (pcbddc->coarse_phi_B) { PetscInt on_B,on_primal,on_D=n_D; if (pcbddc->coarse_phi_D) { ierr = MatGetSize(pcbddc->coarse_phi_D,&on_D,NULL);CHKERRQ(ierr); } ierr = MatGetSize(pcbddc->coarse_phi_B,&on_B,&on_primal);CHKERRQ(ierr); if (on_B != n_B || on_primal != pcbddc->local_primal_size || on_D != n_D) { PetscScalar *marray; ierr = MatDenseGetArray(pcbddc->coarse_phi_B,&marray);CHKERRQ(ierr); ierr = PetscFree(marray);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->coarse_phi_B);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->coarse_psi_B);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->coarse_phi_D);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->coarse_psi_D);CHKERRQ(ierr); } } if (!pcbddc->coarse_phi_B) { PetscScalar *marr; /* memory size */ n = n_B*pcbddc->local_primal_size; if (pcbddc->switch_static || pcbddc->dbg_flag) n += n_D*pcbddc->local_primal_size; if (!pcbddc->symmetric_primal) n *= 2; ierr = PetscCalloc1(n,&marr);CHKERRQ(ierr); ierr = MatCreateSeqDense(PETSC_COMM_SELF,n_B,pcbddc->local_primal_size,marr,&pcbddc->coarse_phi_B);CHKERRQ(ierr); marr += n_B*pcbddc->local_primal_size; if (pcbddc->switch_static || pcbddc->dbg_flag) { ierr = MatCreateSeqDense(PETSC_COMM_SELF,n_D,pcbddc->local_primal_size,marr,&pcbddc->coarse_phi_D);CHKERRQ(ierr); marr += n_D*pcbddc->local_primal_size; } if (!pcbddc->symmetric_primal) { ierr = MatCreateSeqDense(PETSC_COMM_SELF,n_B,pcbddc->local_primal_size,marr,&pcbddc->coarse_psi_B);CHKERRQ(ierr); marr += n_B*pcbddc->local_primal_size; if (pcbddc->switch_static || pcbddc->dbg_flag) { ierr = MatCreateSeqDense(PETSC_COMM_SELF,n_D,pcbddc->local_primal_size,marr,&pcbddc->coarse_psi_D);CHKERRQ(ierr); } } else { ierr = PetscObjectReference((PetscObject)pcbddc->coarse_phi_B);CHKERRQ(ierr); pcbddc->coarse_psi_B = pcbddc->coarse_phi_B; if (pcbddc->switch_static || pcbddc->dbg_flag) { ierr = PetscObjectReference((PetscObject)pcbddc->coarse_phi_D);CHKERRQ(ierr); pcbddc->coarse_psi_D = pcbddc->coarse_phi_D; } } } /* We are now ready to evaluate coarse basis functions and subdomain contribution to coarse problem */ p0_lidx_I = NULL; if (pcbddc->benign_n && (pcbddc->switch_static || pcbddc->dbg_flag)) { const PetscInt *idxs; ierr = ISGetIndices(pcis->is_I_local,&idxs);CHKERRQ(ierr); ierr = PetscMalloc1(pcbddc->benign_n,&p0_lidx_I);CHKERRQ(ierr); for (i=0;ibenign_n;i++) { ierr = PetscFindInt(pcbddc->benign_p0_lidx[i],pcis->n-pcis->n_B,idxs,&p0_lidx_I[i]);CHKERRQ(ierr); } ierr = ISRestoreIndices(pcis->is_I_local,&idxs);CHKERRQ(ierr); } /* vertices */ if (n_vertices) { ierr = MatConvert(A_VV,MATDENSE,MAT_INPLACE_MATRIX,&A_VV);CHKERRQ(ierr); if (n_R) { Mat A_RRmA_RV,A_RV_bcorr=NULL,S_VVt; /* S_VVt with LDA=N */ PetscBLASInt B_N,B_one = 1; PetscScalar *x,*y; PetscBool isseqaij; ierr = MatScale(A_RV,m_one);CHKERRQ(ierr); if (need_benign_correction) { ISLocalToGlobalMapping RtoN; IS is_p0; PetscInt *idxs_p0,n; ierr = PetscMalloc1(pcbddc->benign_n,&idxs_p0);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingCreateIS(pcbddc->is_R_local,&RtoN);CHKERRQ(ierr); ierr = ISGlobalToLocalMappingApply(RtoN,IS_GTOLM_DROP,pcbddc->benign_n,pcbddc->benign_p0_lidx,&n,idxs_p0);CHKERRQ(ierr); if (n != pcbddc->benign_n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Error in R numbering for benign p0! %d != %d\n",n,pcbddc->benign_n); ierr = ISLocalToGlobalMappingDestroy(&RtoN);CHKERRQ(ierr); ierr = ISCreateGeneral(PETSC_COMM_SELF,n,idxs_p0,PETSC_OWN_POINTER,&is_p0);CHKERRQ(ierr); ierr = MatCreateSubMatrix(A_RV,is_p0,NULL,MAT_INITIAL_MATRIX,&A_RV_bcorr);CHKERRQ(ierr); ierr = ISDestroy(&is_p0);CHKERRQ(ierr); } if (lda_rhs == n_R) { ierr = MatConvert(A_RV,MATDENSE,MAT_INPLACE_MATRIX,&A_RV);CHKERRQ(ierr); } else { PetscScalar *av,*array; const PetscInt *xadj,*adjncy; PetscInt n; PetscBool flg_row; array = work+lda_rhs*n_vertices; ierr = PetscMemzero(array,lda_rhs*n_vertices*sizeof(PetscScalar));CHKERRQ(ierr); ierr = MatConvert(A_RV,MATSEQAIJ,MAT_INPLACE_MATRIX,&A_RV);CHKERRQ(ierr); ierr = MatGetRowIJ(A_RV,0,PETSC_FALSE,PETSC_FALSE,&n,&xadj,&adjncy,&flg_row);CHKERRQ(ierr); ierr = MatSeqAIJGetArray(A_RV,&av);CHKERRQ(ierr); for (i=0;ireuse_solver; PetscScalar *marr; ierr = MatDenseGetArray(A_RV,&marr);CHKERRQ(ierr); /* need \Phi^T A_RV = (I+L)A_RV, L given by | 0 0 0 | (V) L = | 0 0 -1 | (P-p0) | 0 0 -1 | (p0) */ for (i=0;ibenign_n;i++) { const PetscScalar *vals; const PetscInt *idxs,*idxs_zero; PetscInt n,j,nz; ierr = ISGetLocalSize(reuse_solver->benign_zerodiag_subs[i],&nz);CHKERRQ(ierr); ierr = ISGetIndices(reuse_solver->benign_zerodiag_subs[i],&idxs_zero);CHKERRQ(ierr); ierr = MatGetRow(A_RV_bcorr,i,&n,&idxs,&vals);CHKERRQ(ierr); for (j=0;jbenign_zerodiag_subs[i],&idxs_zero);CHKERRQ(ierr); } ierr = MatDenseRestoreArray(A_RV,&marr);CHKERRQ(ierr); } if (F) { /* need to correct the rhs */ if (need_benign_correction) { PCBDDCReuseSolvers reuse_solver = sub_schurs->reuse_solver; PetscScalar *marr; ierr = MatDenseGetArray(A_RV,&marr);CHKERRQ(ierr); if (lda_rhs != n_R) { for (i=0;ivec1_R,marr+i*lda_rhs);CHKERRQ(ierr); ierr = PCBDDCReuseSolversBenignAdapt(reuse_solver,pcbddc->vec1_R,NULL,PETSC_FALSE,PETSC_TRUE);CHKERRQ(ierr); ierr = VecResetArray(pcbddc->vec1_R);CHKERRQ(ierr); } } ierr = MatDenseRestoreArray(A_RV,&marr);CHKERRQ(ierr); } ierr = MatMatSolve(F,A_RV,A_RRmA_RV);CHKERRQ(ierr); /* need to correct the solution */ if (need_benign_correction) { PCBDDCReuseSolvers reuse_solver = sub_schurs->reuse_solver; PetscScalar *marr; ierr = MatDenseGetArray(A_RRmA_RV,&marr);CHKERRQ(ierr); if (lda_rhs != n_R) { for (i=0;ivec1_R,marr+i*lda_rhs);CHKERRQ(ierr); ierr = PCBDDCReuseSolversBenignAdapt(reuse_solver,pcbddc->vec1_R,NULL,PETSC_TRUE,PETSC_TRUE);CHKERRQ(ierr); ierr = VecResetArray(pcbddc->vec1_R);CHKERRQ(ierr); } } ierr = MatDenseRestoreArray(A_RRmA_RV,&marr);CHKERRQ(ierr); } } else { ierr = MatDenseGetArray(A_RV,&y);CHKERRQ(ierr); for (i=0;ivec1_R,y+i*lda_rhs);CHKERRQ(ierr); ierr = VecPlaceArray(pcbddc->vec2_R,work+i*lda_rhs);CHKERRQ(ierr); ierr = KSPSolve(pcbddc->ksp_R,pcbddc->vec1_R,pcbddc->vec2_R);CHKERRQ(ierr); ierr = VecResetArray(pcbddc->vec1_R);CHKERRQ(ierr); ierr = VecResetArray(pcbddc->vec2_R);CHKERRQ(ierr); } ierr = MatDenseRestoreArray(A_RV,&y);CHKERRQ(ierr); } ierr = MatDestroy(&A_RV);CHKERRQ(ierr); /* S_VV and S_CV */ if (n_constraints) { Mat B; ierr = PetscMemzero(work+lda_rhs*n_vertices,n_B*n_vertices*sizeof(PetscScalar));CHKERRQ(ierr); for (i=0;ivec1_R,work+i*lda_rhs);CHKERRQ(ierr); ierr = VecPlaceArray(pcis->vec1_B,work+lda_rhs*n_vertices+i*n_B);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecResetArray(pcis->vec1_B);CHKERRQ(ierr); ierr = VecResetArray(pcbddc->vec1_R);CHKERRQ(ierr); } ierr = MatCreateSeqDense(PETSC_COMM_SELF,n_B,n_vertices,work+lda_rhs*n_vertices,&B);CHKERRQ(ierr); ierr = MatMatMult(pcbddc->local_auxmat1,B,MAT_REUSE_MATRIX,PETSC_DEFAULT,&S_CV);CHKERRQ(ierr); ierr = MatDestroy(&B);CHKERRQ(ierr); ierr = MatCreateSeqDense(PETSC_COMM_SELF,lda_rhs,n_vertices,work+lda_rhs*n_vertices,&B);CHKERRQ(ierr); ierr = MatMatMult(local_auxmat2_R,S_CV,MAT_REUSE_MATRIX,PETSC_DEFAULT,&B);CHKERRQ(ierr); ierr = MatScale(S_CV,m_one);CHKERRQ(ierr); ierr = PetscBLASIntCast(lda_rhs*n_vertices,&B_N);CHKERRQ(ierr); PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&B_N,&one,work+lda_rhs*n_vertices,&B_one,work,&B_one)); ierr = MatDestroy(&B);CHKERRQ(ierr); } ierr = PetscObjectTypeCompare((PetscObject)A_VR,MATSEQAIJ,&isseqaij);CHKERRQ(ierr); if (!isseqaij) { /* MatMatMult with SEQ(S)BAIJ below will raise an error */ ierr = MatConvert(A_VR,MATSEQAIJ,MAT_INPLACE_MATRIX,&A_VR);CHKERRQ(ierr); } if (lda_rhs != n_R) { ierr = MatDestroy(&A_RRmA_RV);CHKERRQ(ierr); ierr = MatCreateSeqDense(PETSC_COMM_SELF,n_R,n_vertices,work,&A_RRmA_RV);CHKERRQ(ierr); ierr = MatSeqDenseSetLDA(A_RRmA_RV,lda_rhs);CHKERRQ(ierr); } ierr = MatMatMult(A_VR,A_RRmA_RV,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&S_VVt);CHKERRQ(ierr); /* need A_VR * \Phi * A_RRmA_RV = A_VR * (I+L)^T * A_RRmA_RV, L given as before */ if (need_benign_correction) { PCBDDCReuseSolvers reuse_solver = sub_schurs->reuse_solver; PetscScalar *marr,*sums; ierr = PetscMalloc1(n_vertices,&sums);CHKERRQ(ierr); ierr = MatDenseGetArray(S_VVt,&marr);CHKERRQ(ierr); for (i=0;ibenign_n;i++) { const PetscScalar *vals; const PetscInt *idxs,*idxs_zero; PetscInt n,j,nz; ierr = ISGetLocalSize(reuse_solver->benign_zerodiag_subs[i],&nz);CHKERRQ(ierr); ierr = ISGetIndices(reuse_solver->benign_zerodiag_subs[i],&idxs_zero);CHKERRQ(ierr); for (j=0;jbenign_zerodiag_subs[i],&idxs_zero);CHKERRQ(ierr); } ierr = PetscFree(sums);CHKERRQ(ierr); ierr = MatDenseRestoreArray(S_VVt,&marr);CHKERRQ(ierr); ierr = MatDestroy(&A_RV_bcorr);CHKERRQ(ierr); } ierr = MatDestroy(&A_RRmA_RV);CHKERRQ(ierr); ierr = PetscBLASIntCast(n_vertices*n_vertices,&B_N);CHKERRQ(ierr); ierr = MatDenseGetArray(A_VV,&x);CHKERRQ(ierr); ierr = MatDenseGetArray(S_VVt,&y);CHKERRQ(ierr); PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&B_N,&one,x,&B_one,y,&B_one)); ierr = MatDenseRestoreArray(A_VV,&x);CHKERRQ(ierr); ierr = MatDenseRestoreArray(S_VVt,&y);CHKERRQ(ierr); ierr = MatCopy(S_VVt,S_VV,SAME_NONZERO_PATTERN);CHKERRQ(ierr); ierr = MatDestroy(&S_VVt);CHKERRQ(ierr); } else { ierr = MatCopy(A_VV,S_VV,SAME_NONZERO_PATTERN);CHKERRQ(ierr); } ierr = MatDestroy(&A_VV);CHKERRQ(ierr); /* coarse basis functions */ for (i=0;ivec1_R,work+lda_rhs*i);CHKERRQ(ierr); ierr = MatDenseGetArray(pcbddc->coarse_phi_B,&y);CHKERRQ(ierr); ierr = VecPlaceArray(pcis->vec1_B,y+n_B*i);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); y[n_B*i+idx_V_B[i]] = 1.0; ierr = MatDenseRestoreArray(pcbddc->coarse_phi_B,&y);CHKERRQ(ierr); ierr = VecResetArray(pcis->vec1_B);CHKERRQ(ierr); if (pcbddc->switch_static || pcbddc->dbg_flag) { PetscInt j; ierr = MatDenseGetArray(pcbddc->coarse_phi_D,&y);CHKERRQ(ierr); ierr = VecPlaceArray(pcis->vec1_D,y+n_D*i);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecResetArray(pcis->vec1_D);CHKERRQ(ierr); for (j=0;jbenign_n;j++) y[n_D*i+p0_lidx_I[j]] = 0.0; ierr = MatDenseRestoreArray(pcbddc->coarse_phi_D,&y);CHKERRQ(ierr); } ierr = VecResetArray(pcbddc->vec1_R);CHKERRQ(ierr); } /* if n_R == 0 the object is not destroyed */ ierr = MatDestroy(&A_RV);CHKERRQ(ierr); } ierr = VecDestroy(&dummy_vec);CHKERRQ(ierr); if (n_constraints) { Mat B; ierr = MatCreateSeqDense(PETSC_COMM_SELF,lda_rhs,n_constraints,work,&B);CHKERRQ(ierr); ierr = MatScale(S_CC,m_one);CHKERRQ(ierr); ierr = MatMatMult(local_auxmat2_R,S_CC,MAT_REUSE_MATRIX,PETSC_DEFAULT,&B);CHKERRQ(ierr); ierr = MatScale(S_CC,m_one);CHKERRQ(ierr); if (n_vertices) { if (isCHOL) { /* if we can solve the interior problem with cholesky, we should also be fine with transposing here */ ierr = MatTranspose(S_CV,MAT_REUSE_MATRIX,&S_VC);CHKERRQ(ierr); } else { Mat S_VCt; if (lda_rhs != n_R) { ierr = MatDestroy(&B);CHKERRQ(ierr); ierr = MatCreateSeqDense(PETSC_COMM_SELF,n_R,n_constraints,work,&B);CHKERRQ(ierr); ierr = MatSeqDenseSetLDA(B,lda_rhs);CHKERRQ(ierr); } ierr = MatMatMult(A_VR,B,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&S_VCt);CHKERRQ(ierr); ierr = MatCopy(S_VCt,S_VC,SAME_NONZERO_PATTERN);CHKERRQ(ierr); ierr = MatDestroy(&S_VCt);CHKERRQ(ierr); } } ierr = MatDestroy(&B);CHKERRQ(ierr); /* coarse basis functions */ for (i=0;ivec1_R,work+lda_rhs*i);CHKERRQ(ierr); ierr = MatDenseGetArray(pcbddc->coarse_phi_B,&y);CHKERRQ(ierr); ierr = VecPlaceArray(pcis->vec1_B,y+n_B*(i+n_vertices));CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = MatDenseRestoreArray(pcbddc->coarse_phi_B,&y);CHKERRQ(ierr); ierr = VecResetArray(pcis->vec1_B);CHKERRQ(ierr); if (pcbddc->switch_static || pcbddc->dbg_flag) { PetscInt j; ierr = MatDenseGetArray(pcbddc->coarse_phi_D,&y);CHKERRQ(ierr); ierr = VecPlaceArray(pcis->vec1_D,y+n_D*(i+n_vertices));CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecResetArray(pcis->vec1_D);CHKERRQ(ierr); for (j=0;jbenign_n;j++) y[n_D*i+p0_lidx_I[j]] = 0.0; ierr = MatDenseRestoreArray(pcbddc->coarse_phi_D,&y);CHKERRQ(ierr); } ierr = VecResetArray(pcbddc->vec1_R);CHKERRQ(ierr); } } if (n_constraints) { ierr = MatDestroy(&local_auxmat2_R);CHKERRQ(ierr); } ierr = PetscFree(p0_lidx_I);CHKERRQ(ierr); /* coarse matrix entries relative to B_0 */ if (pcbddc->benign_n) { Mat B0_B,B0_BPHI; IS is_dummy; PetscScalar *data; PetscInt j; ierr = ISCreateStride(PETSC_COMM_SELF,pcbddc->benign_n,0,1,&is_dummy);CHKERRQ(ierr); ierr = MatCreateSubMatrix(pcbddc->benign_B0,is_dummy,pcis->is_B_local,MAT_INITIAL_MATRIX,&B0_B);CHKERRQ(ierr); ierr = ISDestroy(&is_dummy);CHKERRQ(ierr); ierr = MatMatMult(B0_B,pcbddc->coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&B0_BPHI);CHKERRQ(ierr); ierr = MatConvert(B0_BPHI,MATSEQDENSE,MAT_INPLACE_MATRIX,&B0_BPHI);CHKERRQ(ierr); ierr = MatDenseGetArray(B0_BPHI,&data);CHKERRQ(ierr); for (j=0;jbenign_n;j++) { PetscInt primal_idx = pcbddc->local_primal_size - pcbddc->benign_n + j; for (i=0;ilocal_primal_size;i++) { coarse_submat_vals[primal_idx*pcbddc->local_primal_size+i] = data[i*pcbddc->benign_n+j]; coarse_submat_vals[i*pcbddc->local_primal_size+primal_idx] = data[i*pcbddc->benign_n+j]; } } ierr = MatDenseRestoreArray(B0_BPHI,&data);CHKERRQ(ierr); ierr = MatDestroy(&B0_B);CHKERRQ(ierr); ierr = MatDestroy(&B0_BPHI);CHKERRQ(ierr); } /* compute other basis functions for non-symmetric problems */ if (!pcbddc->symmetric_primal) { Mat B_V=NULL,B_C=NULL; PetscScalar *marray; if (n_constraints) { Mat S_CCT,C_CRT; ierr = MatTranspose(C_CR,MAT_INITIAL_MATRIX,&C_CRT);CHKERRQ(ierr); ierr = MatTranspose(S_CC,MAT_INITIAL_MATRIX,&S_CCT);CHKERRQ(ierr); ierr = MatMatMult(C_CRT,S_CCT,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&B_C);CHKERRQ(ierr); ierr = MatDestroy(&S_CCT);CHKERRQ(ierr); if (n_vertices) { Mat S_VCT; ierr = MatTranspose(S_VC,MAT_INITIAL_MATRIX,&S_VCT);CHKERRQ(ierr); ierr = MatMatMult(C_CRT,S_VCT,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&B_V);CHKERRQ(ierr); ierr = MatDestroy(&S_VCT);CHKERRQ(ierr); } ierr = MatDestroy(&C_CRT);CHKERRQ(ierr); } else { ierr = MatCreateSeqDense(PETSC_COMM_SELF,n_R,n_vertices,NULL,&B_V);CHKERRQ(ierr); } if (n_vertices && n_R) { PetscScalar *av,*marray; const PetscInt *xadj,*adjncy; PetscInt n; PetscBool flg_row; /* B_V = B_V - A_VR^T */ ierr = MatConvert(A_VR,MATSEQAIJ,MAT_INPLACE_MATRIX,&A_VR);CHKERRQ(ierr); ierr = MatGetRowIJ(A_VR,0,PETSC_FALSE,PETSC_FALSE,&n,&xadj,&adjncy,&flg_row);CHKERRQ(ierr); ierr = MatSeqAIJGetArray(A_VR,&av);CHKERRQ(ierr); ierr = MatDenseGetArray(B_V,&marray);CHKERRQ(ierr); for (i=0;ivec1_R,marray+i*n_R);CHKERRQ(ierr); ierr = VecPlaceArray(pcbddc->vec2_R,work+i*n_R);CHKERRQ(ierr); ierr = KSPSolveTranspose(pcbddc->ksp_R,pcbddc->vec1_R,pcbddc->vec2_R);CHKERRQ(ierr); ierr = VecResetArray(pcbddc->vec1_R);CHKERRQ(ierr); ierr = VecResetArray(pcbddc->vec2_R);CHKERRQ(ierr); } ierr = MatDenseRestoreArray(B_V,&marray);CHKERRQ(ierr); } if (B_C) { ierr = MatDenseGetArray(B_C,&marray);CHKERRQ(ierr); for (i=n_vertices;ivec1_R,marray+(i-n_vertices)*n_R);CHKERRQ(ierr); ierr = VecPlaceArray(pcbddc->vec2_R,work+i*n_R);CHKERRQ(ierr); ierr = KSPSolveTranspose(pcbddc->ksp_R,pcbddc->vec1_R,pcbddc->vec2_R);CHKERRQ(ierr); ierr = VecResetArray(pcbddc->vec1_R);CHKERRQ(ierr); ierr = VecResetArray(pcbddc->vec2_R);CHKERRQ(ierr); } ierr = MatDenseRestoreArray(B_C,&marray);CHKERRQ(ierr); } /* coarse basis functions */ for (i=0;ilocal_primal_size;i++) { PetscScalar *y; ierr = VecPlaceArray(pcbddc->vec1_R,work+i*n_R);CHKERRQ(ierr); ierr = MatDenseGetArray(pcbddc->coarse_psi_B,&y);CHKERRQ(ierr); ierr = VecPlaceArray(pcis->vec1_B,y+n_B*i);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); if (icoarse_psi_B,&y);CHKERRQ(ierr); ierr = VecResetArray(pcis->vec1_B);CHKERRQ(ierr); if (pcbddc->switch_static || pcbddc->dbg_flag) { ierr = MatDenseGetArray(pcbddc->coarse_psi_D,&y);CHKERRQ(ierr); ierr = VecPlaceArray(pcis->vec1_D,y+n_D*i);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecResetArray(pcis->vec1_D);CHKERRQ(ierr); ierr = MatDenseRestoreArray(pcbddc->coarse_psi_D,&y);CHKERRQ(ierr); } ierr = VecResetArray(pcbddc->vec1_R);CHKERRQ(ierr); } ierr = MatDestroy(&B_V);CHKERRQ(ierr); ierr = MatDestroy(&B_C);CHKERRQ(ierr); } /* free memory */ ierr = PetscFree(idx_V_B);CHKERRQ(ierr); ierr = MatDestroy(&S_VV);CHKERRQ(ierr); ierr = MatDestroy(&S_CV);CHKERRQ(ierr); ierr = MatDestroy(&S_VC);CHKERRQ(ierr); ierr = MatDestroy(&S_CC);CHKERRQ(ierr); ierr = PetscFree(work);CHKERRQ(ierr); if (n_vertices) { ierr = MatDestroy(&A_VR);CHKERRQ(ierr); } if (n_constraints) { ierr = MatDestroy(&C_CR);CHKERRQ(ierr); } /* Checking coarse_sub_mat and coarse basis functios */ /* Symmetric case : It should be \Phi^{(j)^T} A^{(j)} \Phi^{(j)}=coarse_sub_mat */ /* Non-symmetric case : It should be \Psi^{(j)^T} A^{(j)} \Phi^{(j)}=coarse_sub_mat */ if (pcbddc->dbg_flag) { Mat coarse_sub_mat; Mat AUXMAT,TM1,TM2,TM3,TM4; Mat coarse_phi_D,coarse_phi_B; Mat coarse_psi_D,coarse_psi_B; Mat A_II,A_BB,A_IB,A_BI; Mat C_B,CPHI; IS is_dummy; Vec mones; MatType checkmattype=MATSEQAIJ; PetscReal real_value; if (pcbddc->benign_n && !pcbddc->benign_change_explicit) { Mat A; ierr = PCBDDCBenignProject(pc,NULL,NULL,&A);CHKERRQ(ierr); ierr = MatCreateSubMatrix(A,pcis->is_I_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&A_II);CHKERRQ(ierr); ierr = MatCreateSubMatrix(A,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&A_IB);CHKERRQ(ierr); ierr = MatCreateSubMatrix(A,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&A_BI);CHKERRQ(ierr); ierr = MatCreateSubMatrix(A,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&A_BB);CHKERRQ(ierr); ierr = MatDestroy(&A);CHKERRQ(ierr); } else { ierr = MatConvert(pcis->A_II,checkmattype,MAT_INITIAL_MATRIX,&A_II);CHKERRQ(ierr); ierr = MatConvert(pcis->A_IB,checkmattype,MAT_INITIAL_MATRIX,&A_IB);CHKERRQ(ierr); ierr = MatConvert(pcis->A_BI,checkmattype,MAT_INITIAL_MATRIX,&A_BI);CHKERRQ(ierr); ierr = MatConvert(pcis->A_BB,checkmattype,MAT_INITIAL_MATRIX,&A_BB);CHKERRQ(ierr); } ierr = MatConvert(pcbddc->coarse_phi_D,checkmattype,MAT_INITIAL_MATRIX,&coarse_phi_D);CHKERRQ(ierr); ierr = MatConvert(pcbddc->coarse_phi_B,checkmattype,MAT_INITIAL_MATRIX,&coarse_phi_B);CHKERRQ(ierr); if (!pcbddc->symmetric_primal) { ierr = MatConvert(pcbddc->coarse_psi_D,checkmattype,MAT_INITIAL_MATRIX,&coarse_psi_D);CHKERRQ(ierr); ierr = MatConvert(pcbddc->coarse_psi_B,checkmattype,MAT_INITIAL_MATRIX,&coarse_psi_B);CHKERRQ(ierr); } ierr = MatCreateSeqDense(PETSC_COMM_SELF,pcbddc->local_primal_size,pcbddc->local_primal_size,coarse_submat_vals,&coarse_sub_mat);CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"--------------------------------------------------\n");CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"Check coarse sub mat computation (symmetric %d)\n",pcbddc->symmetric_primal);CHKERRQ(ierr); ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); if (!pcbddc->symmetric_primal) { ierr = MatMatMult(A_II,coarse_phi_D,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr); ierr = MatTransposeMatMult(coarse_psi_D,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM1);CHKERRQ(ierr); ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr); ierr = MatMatMult(A_BB,coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr); ierr = MatTransposeMatMult(coarse_psi_B,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM2);CHKERRQ(ierr); ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr); ierr = MatMatMult(A_IB,coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr); ierr = MatTransposeMatMult(coarse_psi_D,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM3);CHKERRQ(ierr); ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr); ierr = MatMatMult(A_BI,coarse_phi_D,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr); ierr = MatTransposeMatMult(coarse_psi_B,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM4);CHKERRQ(ierr); ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr); } else { ierr = MatPtAP(A_II,coarse_phi_D,MAT_INITIAL_MATRIX,1.0,&TM1);CHKERRQ(ierr); ierr = MatPtAP(A_BB,coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&TM2);CHKERRQ(ierr); ierr = MatMatMult(A_IB,coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr); ierr = MatTransposeMatMult(coarse_phi_D,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM3);CHKERRQ(ierr); ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr); ierr = MatMatMult(A_BI,coarse_phi_D,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr); ierr = MatTransposeMatMult(coarse_phi_B,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM4);CHKERRQ(ierr); ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr); } ierr = MatAXPY(TM1,one,TM2,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr); ierr = MatAXPY(TM1,one,TM3,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr); ierr = MatAXPY(TM1,one,TM4,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr); ierr = MatConvert(TM1,MATSEQDENSE,MAT_INPLACE_MATRIX,&TM1);CHKERRQ(ierr); if (pcbddc->benign_n) { Mat B0_B,B0_BPHI; PetscScalar *data,*data2; PetscInt j; ierr = ISCreateStride(PETSC_COMM_SELF,pcbddc->benign_n,0,1,&is_dummy);CHKERRQ(ierr); ierr = MatCreateSubMatrix(pcbddc->benign_B0,is_dummy,pcis->is_B_local,MAT_INITIAL_MATRIX,&B0_B);CHKERRQ(ierr); ierr = MatMatMult(B0_B,coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&B0_BPHI);CHKERRQ(ierr); ierr = MatConvert(B0_BPHI,MATSEQDENSE,MAT_INPLACE_MATRIX,&B0_BPHI);CHKERRQ(ierr); ierr = MatDenseGetArray(TM1,&data);CHKERRQ(ierr); ierr = MatDenseGetArray(B0_BPHI,&data2);CHKERRQ(ierr); for (j=0;jbenign_n;j++) { PetscInt primal_idx = pcbddc->local_primal_size - pcbddc->benign_n + j; for (i=0;ilocal_primal_size;i++) { data[primal_idx*pcbddc->local_primal_size+i] += data2[i*pcbddc->benign_n+j]; data[i*pcbddc->local_primal_size+primal_idx] += data2[i*pcbddc->benign_n+j]; } } ierr = MatDenseRestoreArray(TM1,&data);CHKERRQ(ierr); ierr = MatDenseRestoreArray(B0_BPHI,&data2);CHKERRQ(ierr); ierr = MatDestroy(&B0_B);CHKERRQ(ierr); ierr = ISDestroy(&is_dummy);CHKERRQ(ierr); ierr = MatDestroy(&B0_BPHI);CHKERRQ(ierr); } #if 0 { PetscViewer viewer; char filename[256]; sprintf(filename,"details_local_coarse_mat%d_level%d.m",PetscGlobalRank,pcbddc->current_level); ierr = PetscViewerASCIIOpen(PETSC_COMM_SELF,filename,&viewer);CHKERRQ(ierr); ierr = PetscViewerSetFormat(viewer,PETSC_VIEWER_ASCII_MATLAB);CHKERRQ(ierr); ierr = PetscObjectSetName((PetscObject)coarse_sub_mat,"computed");CHKERRQ(ierr); ierr = MatView(coarse_sub_mat,viewer);CHKERRQ(ierr); ierr = PetscObjectSetName((PetscObject)TM1,"projected");CHKERRQ(ierr); ierr = MatView(TM1,viewer);CHKERRQ(ierr); if (save_change) { Mat phi_B; ierr = MatMatMult(save_change,pcbddc->coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&phi_B);CHKERRQ(ierr); ierr = PetscObjectSetName((PetscObject)phi_B,"phi_B");CHKERRQ(ierr); ierr = MatView(phi_B,viewer);CHKERRQ(ierr); ierr = MatDestroy(&phi_B);CHKERRQ(ierr); } else { ierr = PetscObjectSetName((PetscObject)pcbddc->coarse_phi_B,"phi_B");CHKERRQ(ierr); ierr = MatView(pcbddc->coarse_phi_B,viewer);CHKERRQ(ierr); } if (pcbddc->coarse_phi_D) { ierr = PetscObjectSetName((PetscObject)pcbddc->coarse_phi_D,"phi_D");CHKERRQ(ierr); ierr = MatView(pcbddc->coarse_phi_D,viewer);CHKERRQ(ierr); } if (pcbddc->coarse_psi_B) { ierr = PetscObjectSetName((PetscObject)pcbddc->coarse_psi_B,"psi_B");CHKERRQ(ierr); ierr = MatView(pcbddc->coarse_psi_B,viewer);CHKERRQ(ierr); } if (pcbddc->coarse_psi_D) { ierr = PetscObjectSetName((PetscObject)pcbddc->coarse_psi_D,"psi_D");CHKERRQ(ierr); ierr = MatView(pcbddc->coarse_psi_D,viewer);CHKERRQ(ierr); } ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); } #endif ierr = MatAXPY(TM1,m_one,coarse_sub_mat,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr); ierr = MatNorm(TM1,NORM_FROBENIUS,&real_value);CHKERRQ(ierr); ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d matrix error % 1.14e\n",PetscGlobalRank,real_value);CHKERRQ(ierr); /* check constraints */ ierr = ISCreateStride(PETSC_COMM_SELF,pcbddc->local_primal_size-pcbddc->benign_n,0,1,&is_dummy);CHKERRQ(ierr); ierr = MatCreateSubMatrix(pcbddc->ConstraintMatrix,is_dummy,pcis->is_B_local,MAT_INITIAL_MATRIX,&C_B);CHKERRQ(ierr); if (!pcbddc->benign_n) { /* TODO: add benign case */ ierr = MatMatMult(C_B,coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&CPHI);CHKERRQ(ierr); } else { PetscScalar *data; Mat tmat; ierr = MatDenseGetArray(pcbddc->coarse_phi_B,&data);CHKERRQ(ierr); ierr = MatCreateSeqDense(PETSC_COMM_SELF,pcis->n_B,pcbddc->local_primal_size-pcbddc->benign_n,data,&tmat);CHKERRQ(ierr); ierr = MatDenseRestoreArray(pcbddc->coarse_phi_B,&data);CHKERRQ(ierr); ierr = MatMatMult(C_B,tmat,MAT_INITIAL_MATRIX,1.0,&CPHI);CHKERRQ(ierr); ierr = MatDestroy(&tmat);CHKERRQ(ierr); } ierr = MatCreateVecs(CPHI,&mones,NULL);CHKERRQ(ierr); ierr = VecSet(mones,-1.0);CHKERRQ(ierr); ierr = MatDiagonalSet(CPHI,mones,ADD_VALUES);CHKERRQ(ierr); ierr = MatNorm(CPHI,NORM_FROBENIUS,&real_value);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d phi constraints error % 1.14e\n",PetscGlobalRank,real_value);CHKERRQ(ierr); if (!pcbddc->symmetric_primal) { ierr = MatMatMult(C_B,coarse_psi_B,MAT_REUSE_MATRIX,1.0,&CPHI);CHKERRQ(ierr); ierr = VecSet(mones,-1.0);CHKERRQ(ierr); ierr = MatDiagonalSet(CPHI,mones,ADD_VALUES);CHKERRQ(ierr); ierr = MatNorm(CPHI,NORM_FROBENIUS,&real_value);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d psi constraints error % 1.14e\n",PetscGlobalRank,real_value);CHKERRQ(ierr); } ierr = MatDestroy(&C_B);CHKERRQ(ierr); ierr = MatDestroy(&CPHI);CHKERRQ(ierr); ierr = ISDestroy(&is_dummy);CHKERRQ(ierr); ierr = VecDestroy(&mones);CHKERRQ(ierr); ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = MatDestroy(&A_II);CHKERRQ(ierr); ierr = MatDestroy(&A_BB);CHKERRQ(ierr); ierr = MatDestroy(&A_IB);CHKERRQ(ierr); ierr = MatDestroy(&A_BI);CHKERRQ(ierr); ierr = MatDestroy(&TM1);CHKERRQ(ierr); ierr = MatDestroy(&TM2);CHKERRQ(ierr); ierr = MatDestroy(&TM3);CHKERRQ(ierr); ierr = MatDestroy(&TM4);CHKERRQ(ierr); ierr = MatDestroy(&coarse_phi_D);CHKERRQ(ierr); ierr = MatDestroy(&coarse_phi_B);CHKERRQ(ierr); if (!pcbddc->symmetric_primal) { ierr = MatDestroy(&coarse_psi_D);CHKERRQ(ierr); ierr = MatDestroy(&coarse_psi_B);CHKERRQ(ierr); } ierr = MatDestroy(&coarse_sub_mat);CHKERRQ(ierr); } /* get back data */ *coarse_submat_vals_n = coarse_submat_vals; PetscFunctionReturn(0); } PetscErrorCode MatCreateSubMatrixUnsorted(Mat A, IS isrow, IS iscol, Mat* B) { Mat *work_mat; IS isrow_s,iscol_s; PetscBool rsorted,csorted; PetscInt rsize,*idxs_perm_r=NULL,csize,*idxs_perm_c=NULL; PetscErrorCode ierr; PetscFunctionBegin; ierr = ISSorted(isrow,&rsorted);CHKERRQ(ierr); ierr = ISSorted(iscol,&csorted);CHKERRQ(ierr); ierr = ISGetLocalSize(isrow,&rsize);CHKERRQ(ierr); ierr = ISGetLocalSize(iscol,&csize);CHKERRQ(ierr); if (!rsorted) { const PetscInt *idxs; PetscInt *idxs_sorted,i; ierr = PetscMalloc1(rsize,&idxs_perm_r);CHKERRQ(ierr); ierr = PetscMalloc1(rsize,&idxs_sorted);CHKERRQ(ierr); for (i=0;ipmat->data; PC_BDDC* pcbddc = (PC_BDDC*)pc->data; Mat new_mat,lA; IS is_local,is_global; PetscInt local_size; PetscBool isseqaij; PetscErrorCode ierr; PetscFunctionBegin; ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); ierr = MatGetSize(matis->A,&local_size,NULL);CHKERRQ(ierr); ierr = ISCreateStride(PetscObjectComm((PetscObject)matis->A),local_size,0,1,&is_local);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingApplyIS(pc->pmat->rmap->mapping,is_local,&is_global);CHKERRQ(ierr); ierr = ISDestroy(&is_local);CHKERRQ(ierr); ierr = MatCreateSubMatrixUnsorted(ChangeOfBasisMatrix,is_global,is_global,&new_mat);CHKERRQ(ierr); ierr = ISDestroy(&is_global);CHKERRQ(ierr); /* check */ if (pcbddc->dbg_flag) { Vec x,x_change; PetscReal error; ierr = MatCreateVecs(ChangeOfBasisMatrix,&x,&x_change);CHKERRQ(ierr); ierr = VecSetRandom(x,NULL);CHKERRQ(ierr); ierr = MatMult(ChangeOfBasisMatrix,x,x_change);CHKERRQ(ierr); ierr = VecScatterBegin(matis->cctx,x,matis->x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(matis->cctx,x,matis->x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = MatMult(new_mat,matis->x,matis->y);CHKERRQ(ierr); if (!pcbddc->change_interior) { const PetscScalar *x,*y,*v; PetscReal lerror = 0.; PetscInt i; ierr = VecGetArrayRead(matis->x,&x);CHKERRQ(ierr); ierr = VecGetArrayRead(matis->y,&y);CHKERRQ(ierr); ierr = VecGetArrayRead(matis->counter,&v);CHKERRQ(ierr); for (i=0;i lerror) lerror = PetscAbsScalar(x[i]-y[i]); ierr = VecRestoreArrayRead(matis->x,&x);CHKERRQ(ierr); ierr = VecRestoreArrayRead(matis->y,&y);CHKERRQ(ierr); ierr = VecRestoreArrayRead(matis->counter,&v);CHKERRQ(ierr); ierr = MPIU_Allreduce(&lerror,&error,1,MPIU_REAL,MPI_MAX,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); if (error > PETSC_SMALL) { if (!pcbddc->user_ChangeOfBasisMatrix || pcbddc->current_level) { SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_PLIB,"Error global vs local change on I: %1.6e\n",error); } else { SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_USER,"Error global vs local change on I: %1.6e\n",error); } } } ierr = VecScatterBegin(matis->rctx,matis->y,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(matis->rctx,matis->y,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecAXPY(x,-1.0,x_change);CHKERRQ(ierr); ierr = VecNorm(x,NORM_INFINITY,&error);CHKERRQ(ierr); if (error > PETSC_SMALL) { if (!pcbddc->user_ChangeOfBasisMatrix || pcbddc->current_level) { SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_PLIB,"Error global vs local change on N: %1.6e\n",error); } else { SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_USER,"Error global vs local change on N: %1.6e\n",error); } } ierr = VecDestroy(&x);CHKERRQ(ierr); ierr = VecDestroy(&x_change);CHKERRQ(ierr); } /* lA is present if we are setting up an inner BDDC for a saddle point FETI-DP */ ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lA" ,(PetscObject*)&lA);CHKERRQ(ierr); /* TODO: HOW TO WORK WITH BAIJ and SBAIJ and SEQDENSE? */ ierr = PetscObjectTypeCompare((PetscObject)matis->A,MATSEQAIJ,&isseqaij);CHKERRQ(ierr); if (isseqaij) { ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); ierr = MatPtAP(matis->A,new_mat,MAT_INITIAL_MATRIX,2.0,&pcbddc->local_mat);CHKERRQ(ierr); if (lA) { Mat work; ierr = MatPtAP(lA,new_mat,MAT_INITIAL_MATRIX,2.0,&work);CHKERRQ(ierr); ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_lA" ,(PetscObject)work);CHKERRQ(ierr); ierr = MatDestroy(&work);CHKERRQ(ierr); } } else { Mat work_mat; ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); ierr = MatConvert(matis->A,MATSEQAIJ,MAT_INITIAL_MATRIX,&work_mat);CHKERRQ(ierr); ierr = MatPtAP(work_mat,new_mat,MAT_INITIAL_MATRIX,2.0,&pcbddc->local_mat);CHKERRQ(ierr); ierr = MatDestroy(&work_mat);CHKERRQ(ierr); if (lA) { Mat work; ierr = MatConvert(lA,MATSEQAIJ,MAT_INITIAL_MATRIX,&work_mat);CHKERRQ(ierr); ierr = MatPtAP(work_mat,new_mat,MAT_INITIAL_MATRIX,2.0,&work);CHKERRQ(ierr); ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_lA" ,(PetscObject)work);CHKERRQ(ierr); ierr = MatDestroy(&work);CHKERRQ(ierr); } } if (matis->A->symmetric_set) { ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr); #if !defined(PETSC_USE_COMPLEX) ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); #endif } ierr = MatDestroy(&new_mat);CHKERRQ(ierr); PetscFunctionReturn(0); } PetscErrorCode PCBDDCSetUpLocalScatters(PC pc) { PC_IS* pcis = (PC_IS*)(pc->data); PC_BDDC* pcbddc = (PC_BDDC*)pc->data; PCBDDCSubSchurs sub_schurs = pcbddc->sub_schurs; PetscInt *idx_R_local=NULL; PetscInt n_vertices,i,j,n_R,n_D,n_B; PetscInt vbs,bs; PetscBT bitmask=NULL; PetscErrorCode ierr; PetscFunctionBegin; /* No need to setup local scatters if - primal space is unchanged AND - we actually have locally some primal dofs (could not be true in multilevel or for isolated subdomains) AND - we are not in debugging mode (this is needed since there are Synchronized prints at the end of the subroutine */ if (!pcbddc->new_primal_space_local && pcbddc->local_primal_size && !pcbddc->dbg_flag) { PetscFunctionReturn(0); } /* destroy old objects */ ierr = ISDestroy(&pcbddc->is_R_local);CHKERRQ(ierr); ierr = VecScatterDestroy(&pcbddc->R_to_B);CHKERRQ(ierr); ierr = VecScatterDestroy(&pcbddc->R_to_D);CHKERRQ(ierr); /* Set Non-overlapping dimensions */ n_B = pcis->n_B; n_D = pcis->n - n_B; n_vertices = pcbddc->n_vertices; /* Dohrmann's notation: dofs splitted in R (Remaining: all dofs but the vertices) and V (Vertices) */ /* create auxiliary bitmask and allocate workspace */ if (!sub_schurs || !sub_schurs->reuse_solver) { ierr = PetscMalloc1(pcis->n-n_vertices,&idx_R_local);CHKERRQ(ierr); ierr = PetscBTCreate(pcis->n,&bitmask);CHKERRQ(ierr); for (i=0;ilocal_primal_ref_node[i]);CHKERRQ(ierr); } for (i=0, n_R=0; in; i++) { if (!PetscBTLookup(bitmask,i)) { idx_R_local[n_R++] = i; } } } else { /* A different ordering (already computed) is present if we are reusing the Schur solver */ PCBDDCReuseSolvers reuse_solver = sub_schurs->reuse_solver; ierr = ISGetIndices(reuse_solver->is_R,(const PetscInt**)&idx_R_local);CHKERRQ(ierr); ierr = ISGetLocalSize(reuse_solver->is_R,&n_R);CHKERRQ(ierr); } /* Block code */ vbs = 1; ierr = MatGetBlockSize(pcbddc->local_mat,&bs);CHKERRQ(ierr); if (bs>1 && !(n_vertices%bs)) { PetscBool is_blocked = PETSC_TRUE; PetscInt *vary; if (!sub_schurs || !sub_schurs->reuse_solver) { ierr = PetscMalloc1(pcis->n/bs,&vary);CHKERRQ(ierr); ierr = PetscMemzero(vary,pcis->n/bs*sizeof(PetscInt));CHKERRQ(ierr); /* Verify that the vertex indices correspond to each element in a block (code taken from sbaij2.c) */ /* it is ok to check this way since local_primal_ref_node are always sorted by local numbering and idx_R_local is obtained as a complement */ for (i=0; ilocal_primal_ref_node[i]/bs]++; for (i=0; in/bs; i++) { if (vary[i]!=0 && vary[i]!=bs) { is_blocked = PETSC_FALSE; break; } } ierr = PetscFree(vary);CHKERRQ(ierr); } else { /* Verify directly the R set */ for (i=0; iis_R_local);CHKERRQ(ierr); if (sub_schurs && sub_schurs->reuse_solver) { PCBDDCReuseSolvers reuse_solver = sub_schurs->reuse_solver; ierr = ISRestoreIndices(reuse_solver->is_R,(const PetscInt**)&idx_R_local);CHKERRQ(ierr); ierr = ISDestroy(&reuse_solver->is_R);CHKERRQ(ierr); ierr = PetscObjectReference((PetscObject)pcbddc->is_R_local);CHKERRQ(ierr); reuse_solver->is_R = pcbddc->is_R_local; } else { ierr = PetscFree(idx_R_local);CHKERRQ(ierr); } /* print some info if requested */ if (pcbddc->dbg_flag) { ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"--------------------------------------------------\n");CHKERRQ(ierr); ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d local dimensions\n",PetscGlobalRank);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"local_size = %d, dirichlet_size = %d, boundary_size = %d\n",pcis->n,n_D,n_B);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"r_size = %d, v_size = %d, constraints = %d, local_primal_size = %d\n",n_R,n_vertices,pcbddc->local_primal_size-n_vertices-pcbddc->benign_n,pcbddc->local_primal_size);CHKERRQ(ierr); ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); } /* VecScatters pcbddc->R_to_B and (optionally) pcbddc->R_to_D */ if (!sub_schurs || !sub_schurs->reuse_solver) { IS is_aux1,is_aux2; PetscInt *aux_array1,*aux_array2,*is_indices,*idx_R_local; ierr = ISGetIndices(pcbddc->is_R_local,(const PetscInt**)&idx_R_local);CHKERRQ(ierr); ierr = PetscMalloc1(pcis->n_B-n_vertices,&aux_array1);CHKERRQ(ierr); ierr = PetscMalloc1(pcis->n_B-n_vertices,&aux_array2);CHKERRQ(ierr); ierr = ISGetIndices(pcis->is_I_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); for (i=0; iis_I_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); for (i=0, j=0; iis_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); for (i=0, j=0; iis_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); ierr = ISCreateGeneral(PETSC_COMM_SELF,j,aux_array2,PETSC_OWN_POINTER,&is_aux2);CHKERRQ(ierr); ierr = VecScatterCreate(pcbddc->vec1_R,is_aux1,pcis->vec1_B,is_aux2,&pcbddc->R_to_B);CHKERRQ(ierr); ierr = ISDestroy(&is_aux1);CHKERRQ(ierr); ierr = ISDestroy(&is_aux2);CHKERRQ(ierr); if (pcbddc->switch_static || pcbddc->dbg_flag) { ierr = PetscMalloc1(n_D,&aux_array1);CHKERRQ(ierr); for (i=0, j=0; ivec1_R,is_aux1,pcis->vec1_D,(IS)0,&pcbddc->R_to_D);CHKERRQ(ierr); ierr = ISDestroy(&is_aux1);CHKERRQ(ierr); } ierr = PetscBTDestroy(&bitmask);CHKERRQ(ierr); ierr = ISRestoreIndices(pcbddc->is_R_local,(const PetscInt**)&idx_R_local);CHKERRQ(ierr); } else { PCBDDCReuseSolvers reuse_solver = sub_schurs->reuse_solver; IS tis; PetscInt schur_size; ierr = ISGetLocalSize(reuse_solver->is_B,&schur_size);CHKERRQ(ierr); ierr = ISCreateStride(PETSC_COMM_SELF,schur_size,n_D,1,&tis);CHKERRQ(ierr); ierr = VecScatterCreate(pcbddc->vec1_R,tis,pcis->vec1_B,reuse_solver->is_B,&pcbddc->R_to_B);CHKERRQ(ierr); ierr = ISDestroy(&tis);CHKERRQ(ierr); if (pcbddc->switch_static || pcbddc->dbg_flag) { ierr = ISCreateStride(PETSC_COMM_SELF,n_D,0,1,&tis);CHKERRQ(ierr); ierr = VecScatterCreate(pcbddc->vec1_R,tis,pcis->vec1_D,(IS)0,&pcbddc->R_to_D);CHKERRQ(ierr); ierr = ISDestroy(&tis);CHKERRQ(ierr); } } PetscFunctionReturn(0); } PetscErrorCode PCBDDCSetUpLocalSolvers(PC pc, PetscBool dirichlet, PetscBool neumann) { PC_BDDC *pcbddc = (PC_BDDC*)pc->data; PC_IS *pcis = (PC_IS*)pc->data; PC pc_temp; Mat A_RR; MatReuse reuse; PetscScalar m_one = -1.0; PetscReal value; PetscInt n_D,n_R; PetscBool check_corr[2],issbaij; PetscErrorCode ierr; /* prefixes stuff */ char dir_prefix[256],neu_prefix[256],str_level[16]; size_t len; PetscFunctionBegin; /* compute prefixes */ ierr = PetscStrcpy(dir_prefix,"");CHKERRQ(ierr); ierr = PetscStrcpy(neu_prefix,"");CHKERRQ(ierr); if (!pcbddc->current_level) { ierr = PetscStrcpy(dir_prefix,((PetscObject)pc)->prefix);CHKERRQ(ierr); ierr = PetscStrcpy(neu_prefix,((PetscObject)pc)->prefix);CHKERRQ(ierr); ierr = PetscStrcat(dir_prefix,"pc_bddc_dirichlet_");CHKERRQ(ierr); ierr = PetscStrcat(neu_prefix,"pc_bddc_neumann_");CHKERRQ(ierr); } else { ierr = PetscStrcpy(str_level,"");CHKERRQ(ierr); sprintf(str_level,"l%d_",(int)(pcbddc->current_level)); ierr = PetscStrlen(((PetscObject)pc)->prefix,&len);CHKERRQ(ierr); len -= 15; /* remove "pc_bddc_coarse_" */ if (pcbddc->current_level>1) len -= 3; /* remove "lX_" with X level number */ if (pcbddc->current_level>10) len -= 1; /* remove another char from level number */ ierr = PetscStrncpy(dir_prefix,((PetscObject)pc)->prefix,len+1);CHKERRQ(ierr); ierr = PetscStrncpy(neu_prefix,((PetscObject)pc)->prefix,len+1);CHKERRQ(ierr); ierr = PetscStrcat(dir_prefix,"pc_bddc_dirichlet_");CHKERRQ(ierr); ierr = PetscStrcat(neu_prefix,"pc_bddc_neumann_");CHKERRQ(ierr); ierr = PetscStrcat(dir_prefix,str_level);CHKERRQ(ierr); ierr = PetscStrcat(neu_prefix,str_level);CHKERRQ(ierr); } /* DIRICHLET PROBLEM */ if (dirichlet) { PCBDDCSubSchurs sub_schurs = pcbddc->sub_schurs; if (pcbddc->benign_n && !pcbddc->benign_change_explicit) { if (!sub_schurs || !sub_schurs->reuse_solver) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not yet implemented\n"); if (pcbddc->dbg_flag) { Mat A_IIn; ierr = PCBDDCBenignProject(pc,pcis->is_I_local,pcis->is_I_local,&A_IIn);CHKERRQ(ierr); ierr = MatDestroy(&pcis->A_II);CHKERRQ(ierr); pcis->A_II = A_IIn; } } if (pcbddc->local_mat->symmetric_set) { ierr = MatSetOption(pcis->A_II,MAT_SYMMETRIC,pcbddc->local_mat->symmetric_set);CHKERRQ(ierr); } /* Matrix for Dirichlet problem is pcis->A_II */ n_D = pcis->n - pcis->n_B; if (!pcbddc->ksp_D) { /* create object if not yet build */ ierr = KSPCreate(PETSC_COMM_SELF,&pcbddc->ksp_D);CHKERRQ(ierr); ierr = PetscObjectIncrementTabLevel((PetscObject)pcbddc->ksp_D,(PetscObject)pc,1);CHKERRQ(ierr); /* default */ ierr = KSPSetType(pcbddc->ksp_D,KSPPREONLY);CHKERRQ(ierr); ierr = KSPSetOptionsPrefix(pcbddc->ksp_D,dir_prefix);CHKERRQ(ierr); ierr = PetscObjectTypeCompare((PetscObject)pcis->A_II,MATSEQSBAIJ,&issbaij);CHKERRQ(ierr); ierr = KSPGetPC(pcbddc->ksp_D,&pc_temp);CHKERRQ(ierr); if (issbaij) { ierr = PCSetType(pc_temp,PCCHOLESKY);CHKERRQ(ierr); } else { ierr = PCSetType(pc_temp,PCLU);CHKERRQ(ierr); } /* Allow user's customization */ ierr = KSPSetFromOptions(pcbddc->ksp_D);CHKERRQ(ierr); ierr = PCFactorSetReuseFill(pc_temp,PETSC_TRUE);CHKERRQ(ierr); } ierr = KSPSetOperators(pcbddc->ksp_D,pcis->A_II,pcis->A_II);CHKERRQ(ierr); if (sub_schurs && sub_schurs->reuse_solver) { PCBDDCReuseSolvers reuse_solver = sub_schurs->reuse_solver; ierr = KSPSetPC(pcbddc->ksp_D,reuse_solver->interior_solver);CHKERRQ(ierr); } /* umfpack interface has a bug when matrix dimension is zero. TODO solve from umfpack interface */ if (!n_D) { ierr = KSPGetPC(pcbddc->ksp_D,&pc_temp);CHKERRQ(ierr); ierr = PCSetType(pc_temp,PCNONE);CHKERRQ(ierr); } /* Set Up KSP for Dirichlet problem of BDDC */ ierr = KSPSetUp(pcbddc->ksp_D);CHKERRQ(ierr); /* set ksp_D into pcis data */ ierr = KSPDestroy(&pcis->ksp_D);CHKERRQ(ierr); ierr = PetscObjectReference((PetscObject)pcbddc->ksp_D);CHKERRQ(ierr); pcis->ksp_D = pcbddc->ksp_D; } /* NEUMANN PROBLEM */ A_RR = 0; if (neumann) { PCBDDCSubSchurs sub_schurs = pcbddc->sub_schurs; PetscInt ibs,mbs; PetscBool issbaij; Mat_IS* matis = (Mat_IS*)pc->pmat->data; /* Matrix for Neumann problem is A_RR -> we need to create/reuse it at this point */ ierr = ISGetSize(pcbddc->is_R_local,&n_R);CHKERRQ(ierr); if (pcbddc->ksp_R) { /* already created ksp */ PetscInt nn_R; ierr = KSPGetOperators(pcbddc->ksp_R,NULL,&A_RR);CHKERRQ(ierr); ierr = PetscObjectReference((PetscObject)A_RR);CHKERRQ(ierr); ierr = MatGetSize(A_RR,&nn_R,NULL);CHKERRQ(ierr); if (nn_R != n_R) { /* old ksp is not reusable, so reset it */ ierr = KSPReset(pcbddc->ksp_R);CHKERRQ(ierr); ierr = MatDestroy(&A_RR);CHKERRQ(ierr); reuse = MAT_INITIAL_MATRIX; } else { /* same sizes, but nonzero pattern depend on primal vertices so it can be changed */ if (pcbddc->new_primal_space_local) { /* we are not sure the matrix will have the same nonzero pattern */ ierr = MatDestroy(&A_RR);CHKERRQ(ierr); reuse = MAT_INITIAL_MATRIX; } else { /* safe to reuse the matrix */ reuse = MAT_REUSE_MATRIX; } } /* last check */ if (pc->flag == DIFFERENT_NONZERO_PATTERN) { ierr = MatDestroy(&A_RR);CHKERRQ(ierr); reuse = MAT_INITIAL_MATRIX; } } else { /* first time, so we need to create the matrix */ reuse = MAT_INITIAL_MATRIX; } /* convert pcbddc->local_mat if needed later in PCBDDCSetUpCorrection */ ierr = MatGetBlockSize(pcbddc->local_mat,&mbs);CHKERRQ(ierr); ierr = ISGetBlockSize(pcbddc->is_R_local,&ibs);CHKERRQ(ierr); ierr = PetscObjectTypeCompare((PetscObject)pcbddc->local_mat,MATSEQSBAIJ,&issbaij);CHKERRQ(ierr); if (ibs != mbs) { /* need to convert to SEQAIJ to extract any submatrix with is_R_local */ if (matis->A == pcbddc->local_mat) { ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); ierr = MatConvert(matis->A,MATSEQAIJ,MAT_INITIAL_MATRIX,&pcbddc->local_mat);CHKERRQ(ierr); } else { ierr = MatConvert(pcbddc->local_mat,MATSEQAIJ,MAT_INPLACE_MATRIX,&pcbddc->local_mat);CHKERRQ(ierr); } } else if (issbaij) { /* need to convert to BAIJ to get offdiagonal blocks */ if (matis->A == pcbddc->local_mat) { ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); ierr = MatConvert(matis->A,MATSEQBAIJ,MAT_INITIAL_MATRIX,&pcbddc->local_mat);CHKERRQ(ierr); } else { ierr = MatConvert(pcbddc->local_mat,MATSEQBAIJ,MAT_INPLACE_MATRIX,&pcbddc->local_mat);CHKERRQ(ierr); } } /* extract A_RR */ if (sub_schurs && sub_schurs->reuse_solver) { PCBDDCReuseSolvers reuse_solver = sub_schurs->reuse_solver; if (pcbddc->dbg_flag) { /* we need A_RR to test the solver later */ ierr = MatDestroy(&A_RR);CHKERRQ(ierr); if (reuse_solver->benign_n) { /* we are not using the explicit change of basis on the pressures */ ierr = PCBDDCBenignProject(pc,pcbddc->is_R_local,pcbddc->is_R_local,&A_RR);CHKERRQ(ierr); } else { ierr = MatCreateSubMatrix(pcbddc->local_mat,pcbddc->is_R_local,pcbddc->is_R_local,MAT_INITIAL_MATRIX,&A_RR);CHKERRQ(ierr); } } else { ierr = MatDestroy(&A_RR);CHKERRQ(ierr); ierr = PCGetOperators(reuse_solver->correction_solver,&A_RR,NULL);CHKERRQ(ierr); ierr = PetscObjectReference((PetscObject)A_RR);CHKERRQ(ierr); } } else { /* we have to build the neumann solver, so we need to extract the relevant matrix */ ierr = MatCreateSubMatrix(pcbddc->local_mat,pcbddc->is_R_local,pcbddc->is_R_local,reuse,&A_RR);CHKERRQ(ierr); } if (pcbddc->local_mat->symmetric_set) { ierr = MatSetOption(A_RR,MAT_SYMMETRIC,pcbddc->local_mat->symmetric_set);CHKERRQ(ierr); } if (!pcbddc->ksp_R) { /* create object if not present */ ierr = KSPCreate(PETSC_COMM_SELF,&pcbddc->ksp_R);CHKERRQ(ierr); ierr = PetscObjectIncrementTabLevel((PetscObject)pcbddc->ksp_R,(PetscObject)pc,1);CHKERRQ(ierr); /* default */ ierr = KSPSetType(pcbddc->ksp_R,KSPPREONLY);CHKERRQ(ierr); ierr = KSPSetOptionsPrefix(pcbddc->ksp_R,neu_prefix);CHKERRQ(ierr); ierr = KSPGetPC(pcbddc->ksp_R,&pc_temp);CHKERRQ(ierr); ierr = PetscObjectTypeCompare((PetscObject)A_RR,MATSEQSBAIJ,&issbaij);CHKERRQ(ierr); if (issbaij) { ierr = PCSetType(pc_temp,PCCHOLESKY);CHKERRQ(ierr); } else { ierr = PCSetType(pc_temp,PCLU);CHKERRQ(ierr); } /* Allow user's customization */ ierr = KSPSetFromOptions(pcbddc->ksp_R);CHKERRQ(ierr); ierr = PCFactorSetReuseFill(pc_temp,PETSC_TRUE);CHKERRQ(ierr); } /* umfpack interface has a bug when matrix dimension is zero. TODO solve from umfpack interface */ if (!n_R) { ierr = KSPGetPC(pcbddc->ksp_R,&pc_temp);CHKERRQ(ierr); ierr = PCSetType(pc_temp,PCNONE);CHKERRQ(ierr); } ierr = KSPSetOperators(pcbddc->ksp_R,A_RR,A_RR);CHKERRQ(ierr); /* Reuse solver if it is present */ if (sub_schurs && sub_schurs->reuse_solver) { PCBDDCReuseSolvers reuse_solver = sub_schurs->reuse_solver; ierr = KSPSetPC(pcbddc->ksp_R,reuse_solver->correction_solver);CHKERRQ(ierr); } /* Set Up KSP for Neumann problem of BDDC */ ierr = KSPSetUp(pcbddc->ksp_R);CHKERRQ(ierr); } if (pcbddc->dbg_flag) { ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"--------------------------------------------------\n");CHKERRQ(ierr); } /* adapt Dirichlet and Neumann solvers if a nullspace correction has been requested */ check_corr[0] = check_corr[1] = PETSC_FALSE; if (pcbddc->NullSpace_corr[0]) { ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); } if (dirichlet && pcbddc->NullSpace_corr[0] && !pcbddc->switch_static) { check_corr[0] = PETSC_TRUE; ierr = PCBDDCNullSpaceAssembleCorrection(pc,PETSC_TRUE,pcbddc->NullSpace_corr[1]);CHKERRQ(ierr); } if (neumann && pcbddc->NullSpace_corr[2]) { check_corr[1] = PETSC_TRUE; ierr = PCBDDCNullSpaceAssembleCorrection(pc,PETSC_FALSE,pcbddc->NullSpace_corr[3]);CHKERRQ(ierr); } /* check Dirichlet and Neumann solvers */ if (pcbddc->dbg_flag) { if (dirichlet) { /* Dirichlet */ ierr = VecSetRandom(pcis->vec1_D,NULL);CHKERRQ(ierr); ierr = MatMult(pcis->A_II,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); ierr = KSPSolve(pcbddc->ksp_D,pcis->vec2_D,pcis->vec2_D);CHKERRQ(ierr); ierr = VecAXPY(pcis->vec1_D,m_one,pcis->vec2_D);CHKERRQ(ierr); ierr = VecNorm(pcis->vec1_D,NORM_INFINITY,&value);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d infinity error for Dirichlet solve (%s) = % 1.14e \n",PetscGlobalRank,((PetscObject)(pcbddc->ksp_D))->prefix,value);CHKERRQ(ierr); if (check_corr[0]) { ierr = PCBDDCNullSpaceCheckCorrection(pc,PETSC_TRUE);CHKERRQ(ierr); } ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); } if (neumann) { /* Neumann */ ierr = VecSetRandom(pcbddc->vec1_R,NULL);CHKERRQ(ierr); ierr = MatMult(A_RR,pcbddc->vec1_R,pcbddc->vec2_R);CHKERRQ(ierr); ierr = KSPSolve(pcbddc->ksp_R,pcbddc->vec2_R,pcbddc->vec2_R);CHKERRQ(ierr); ierr = VecAXPY(pcbddc->vec1_R,m_one,pcbddc->vec2_R);CHKERRQ(ierr); ierr = VecNorm(pcbddc->vec1_R,NORM_INFINITY,&value);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d infinity error for Neumann solve (%s) = % 1.14e\n",PetscGlobalRank,((PetscObject)(pcbddc->ksp_R))->prefix,value);CHKERRQ(ierr); if (check_corr[1]) { ierr = PCBDDCNullSpaceCheckCorrection(pc,PETSC_FALSE);CHKERRQ(ierr); } ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); } } /* free Neumann problem's matrix */ ierr = MatDestroy(&A_RR);CHKERRQ(ierr); PetscFunctionReturn(0); } static PetscErrorCode PCBDDCSolveSubstructureCorrection(PC pc, Vec inout_B, Vec inout_D, PetscBool applytranspose) { PetscErrorCode ierr; PC_BDDC* pcbddc = (PC_BDDC*)(pc->data); PCBDDCSubSchurs sub_schurs = pcbddc->sub_schurs; PetscBool reuse_solver = sub_schurs ? ( sub_schurs->reuse_solver ? PETSC_TRUE : PETSC_FALSE ) : PETSC_FALSE; PetscFunctionBegin; if (!reuse_solver) { ierr = VecSet(pcbddc->vec1_R,0.);CHKERRQ(ierr); } if (!pcbddc->switch_static) { if (applytranspose && pcbddc->local_auxmat1) { ierr = MatMultTranspose(pcbddc->local_auxmat2,inout_B,pcbddc->vec1_C);CHKERRQ(ierr); ierr = MatMultTransposeAdd(pcbddc->local_auxmat1,pcbddc->vec1_C,inout_B,inout_B);CHKERRQ(ierr); } if (!reuse_solver) { ierr = VecScatterBegin(pcbddc->R_to_B,inout_B,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_B,inout_B,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); } else { PCBDDCReuseSolvers reuse_solver = sub_schurs->reuse_solver; ierr = VecScatterBegin(reuse_solver->correction_scatter_B,inout_B,reuse_solver->rhs_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(reuse_solver->correction_scatter_B,inout_B,reuse_solver->rhs_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); } } else { ierr = VecScatterBegin(pcbddc->R_to_B,inout_B,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_B,inout_B,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->R_to_D,inout_D,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_D,inout_D,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); if (applytranspose && pcbddc->local_auxmat1) { ierr = MatMultTranspose(pcbddc->local_auxmat2,pcbddc->vec1_R,pcbddc->vec1_C);CHKERRQ(ierr); ierr = MatMultTransposeAdd(pcbddc->local_auxmat1,pcbddc->vec1_C,inout_B,inout_B);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->R_to_B,inout_B,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_B,inout_B,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); } } if (!reuse_solver || pcbddc->switch_static) { if (applytranspose) { ierr = KSPSolveTranspose(pcbddc->ksp_R,pcbddc->vec1_R,pcbddc->vec1_R);CHKERRQ(ierr); } else { ierr = KSPSolve(pcbddc->ksp_R,pcbddc->vec1_R,pcbddc->vec1_R);CHKERRQ(ierr); } } else { PCBDDCReuseSolvers reuse_solver = sub_schurs->reuse_solver; if (applytranspose) { ierr = MatFactorSolveSchurComplementTranspose(reuse_solver->F,reuse_solver->rhs_B,reuse_solver->sol_B);CHKERRQ(ierr); } else { ierr = MatFactorSolveSchurComplement(reuse_solver->F,reuse_solver->rhs_B,reuse_solver->sol_B);CHKERRQ(ierr); } } ierr = VecSet(inout_B,0.);CHKERRQ(ierr); if (!pcbddc->switch_static) { if (!reuse_solver) { ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec1_R,inout_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_B,pcbddc->vec1_R,inout_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); } else { PCBDDCReuseSolvers reuse_solver = sub_schurs->reuse_solver; ierr = VecScatterBegin(reuse_solver->correction_scatter_B,reuse_solver->sol_B,inout_B,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(reuse_solver->correction_scatter_B,reuse_solver->sol_B,inout_B,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); } if (!applytranspose && pcbddc->local_auxmat1) { ierr = MatMult(pcbddc->local_auxmat1,inout_B,pcbddc->vec1_C);CHKERRQ(ierr); ierr = MatMultAdd(pcbddc->local_auxmat2,pcbddc->vec1_C,inout_B,inout_B);CHKERRQ(ierr); } } else { ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec1_R,inout_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_B,pcbddc->vec1_R,inout_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->R_to_D,pcbddc->vec1_R,inout_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_D,pcbddc->vec1_R,inout_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); if (!applytranspose && pcbddc->local_auxmat1) { ierr = MatMult(pcbddc->local_auxmat1,inout_B,pcbddc->vec1_C);CHKERRQ(ierr); ierr = MatMultAdd(pcbddc->local_auxmat2,pcbddc->vec1_C,pcbddc->vec1_R,pcbddc->vec1_R);CHKERRQ(ierr); } ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec1_R,inout_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_B,pcbddc->vec1_R,inout_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->R_to_D,pcbddc->vec1_R,inout_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->R_to_D,pcbddc->vec1_R,inout_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); } PetscFunctionReturn(0); } /* parameter apply transpose determines if the interface preconditioner should be applied transposed or not */ PetscErrorCode PCBDDCApplyInterfacePreconditioner(PC pc, PetscBool applytranspose) { PetscErrorCode ierr; PC_BDDC* pcbddc = (PC_BDDC*)(pc->data); PC_IS* pcis = (PC_IS*) (pc->data); const PetscScalar zero = 0.0; PetscFunctionBegin; /* Application of PSI^T or PHI^T (depending on applytranspose, see comment above) */ if (!pcbddc->benign_apply_coarse_only) { if (applytranspose) { ierr = MatMultTranspose(pcbddc->coarse_phi_B,pcis->vec1_B,pcbddc->vec1_P);CHKERRQ(ierr); if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcbddc->coarse_phi_D,pcis->vec1_D,pcbddc->vec1_P,pcbddc->vec1_P);CHKERRQ(ierr); } } else { ierr = MatMultTranspose(pcbddc->coarse_psi_B,pcis->vec1_B,pcbddc->vec1_P);CHKERRQ(ierr); if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcbddc->coarse_psi_D,pcis->vec1_D,pcbddc->vec1_P,pcbddc->vec1_P);CHKERRQ(ierr); } } } else { ierr = VecSet(pcbddc->vec1_P,zero);CHKERRQ(ierr); } /* add p0 to the last value of vec1_P holding the coarse dof relative to p0 */ if (pcbddc->benign_n) { PetscScalar *array; PetscInt j; ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); for (j=0;jbenign_n;j++) array[pcbddc->local_primal_size-pcbddc->benign_n+j] += pcbddc->benign_p0[j]; ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); } /* start communications from local primal nodes to rhs of coarse solver */ ierr = VecSet(pcbddc->coarse_vec,zero);CHKERRQ(ierr); ierr = PCBDDCScatterCoarseDataBegin(pc,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = PCBDDCScatterCoarseDataEnd(pc,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); /* Coarse solution -> rhs and sol updated inside PCBDDCScattarCoarseDataBegin/End */ if (pcbddc->coarse_ksp) { Mat coarse_mat; Vec rhs,sol; MatNullSpace nullsp; PetscBool isbddc = PETSC_FALSE; if (pcbddc->benign_have_null) { PC coarse_pc; ierr = KSPGetPC(pcbddc->coarse_ksp,&coarse_pc);CHKERRQ(ierr); ierr = PetscObjectTypeCompare((PetscObject)coarse_pc,PCBDDC,&isbddc);CHKERRQ(ierr); /* we need to propagate to coarser levels the need for a possible benign correction */ if (isbddc && pcbddc->benign_apply_coarse_only && !pcbddc->benign_skip_correction) { PC_BDDC* coarsepcbddc = (PC_BDDC*)(coarse_pc->data); coarsepcbddc->benign_skip_correction = PETSC_FALSE; coarsepcbddc->benign_apply_coarse_only = PETSC_TRUE; } } ierr = KSPGetRhs(pcbddc->coarse_ksp,&rhs);CHKERRQ(ierr); ierr = KSPGetSolution(pcbddc->coarse_ksp,&sol);CHKERRQ(ierr); ierr = KSPGetOperators(pcbddc->coarse_ksp,&coarse_mat,NULL);CHKERRQ(ierr); ierr = MatGetNullSpace(coarse_mat,&nullsp);CHKERRQ(ierr); if (nullsp) { ierr = MatNullSpaceRemove(nullsp,rhs);CHKERRQ(ierr); } if (applytranspose) { if (pcbddc->benign_apply_coarse_only) SETERRQ(PetscObjectComm((PetscObject)pcbddc->coarse_ksp),PETSC_ERR_SUP,"Not yet implemented"); ierr = KSPSolveTranspose(pcbddc->coarse_ksp,rhs,sol);CHKERRQ(ierr); } else { if (pcbddc->benign_apply_coarse_only && isbddc) { /* need just to apply the coarse preconditioner during presolve */ PC coarse_pc; ierr = KSPGetPC(pcbddc->coarse_ksp,&coarse_pc);CHKERRQ(ierr); ierr = PCPreSolve(coarse_pc,pcbddc->coarse_ksp);CHKERRQ(ierr); ierr = PCBDDCBenignRemoveInterior(coarse_pc,rhs,sol);CHKERRQ(ierr); ierr = PCPostSolve(coarse_pc,pcbddc->coarse_ksp);CHKERRQ(ierr); } else { ierr = KSPSolve(pcbddc->coarse_ksp,rhs,sol);CHKERRQ(ierr); } } /* we don't need the benign correction at coarser levels anymore */ if (pcbddc->benign_have_null && isbddc) { PC coarse_pc; PC_BDDC* coarsepcbddc; ierr = KSPGetPC(pcbddc->coarse_ksp,&coarse_pc);CHKERRQ(ierr); coarsepcbddc = (PC_BDDC*)(coarse_pc->data); coarsepcbddc->benign_skip_correction = PETSC_TRUE; coarsepcbddc->benign_apply_coarse_only = PETSC_FALSE; } if (nullsp) { ierr = MatNullSpaceRemove(nullsp,sol);CHKERRQ(ierr); } } /* Local solution on R nodes */ if (pcis->n && !pcbddc->benign_apply_coarse_only) { ierr = PCBDDCSolveSubstructureCorrection(pc,pcis->vec1_B,pcis->vec1_D,applytranspose);CHKERRQ(ierr); } /* communications from coarse sol to local primal nodes */ ierr = PCBDDCScatterCoarseDataBegin(pc,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = PCBDDCScatterCoarseDataEnd(pc,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); /* Sum contributions from the two levels */ if (!pcbddc->benign_apply_coarse_only) { if (applytranspose) { ierr = MatMultAdd(pcbddc->coarse_psi_B,pcbddc->vec1_P,pcis->vec1_B,pcis->vec1_B);CHKERRQ(ierr); if (pcbddc->switch_static) { ierr = MatMultAdd(pcbddc->coarse_psi_D,pcbddc->vec1_P,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } } else { ierr = MatMultAdd(pcbddc->coarse_phi_B,pcbddc->vec1_P,pcis->vec1_B,pcis->vec1_B);CHKERRQ(ierr); if (pcbddc->switch_static) { ierr = MatMultAdd(pcbddc->coarse_phi_D,pcbddc->vec1_P,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } } /* store p0 */ if (pcbddc->benign_n) { PetscScalar *array; PetscInt j; ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); for (j=0;jbenign_n;j++) pcbddc->benign_p0[j] = array[pcbddc->local_primal_size-pcbddc->benign_n+j]; ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); } } else { /* expand the coarse solution */ if (applytranspose) { ierr = MatMult(pcbddc->coarse_psi_B,pcbddc->vec1_P,pcis->vec1_B);CHKERRQ(ierr); } else { ierr = MatMult(pcbddc->coarse_phi_B,pcbddc->vec1_P,pcis->vec1_B);CHKERRQ(ierr); } } PetscFunctionReturn(0); } PetscErrorCode PCBDDCScatterCoarseDataBegin(PC pc,InsertMode imode, ScatterMode smode) { PetscErrorCode ierr; PC_BDDC* pcbddc = (PC_BDDC*)(pc->data); PetscScalar *array; Vec from,to; PetscFunctionBegin; if (smode == SCATTER_REVERSE) { /* from global to local -> get data from coarse solution */ from = pcbddc->coarse_vec; to = pcbddc->vec1_P; if (pcbddc->coarse_ksp) { /* get array from coarse processes */ Vec tvec; ierr = KSPGetRhs(pcbddc->coarse_ksp,&tvec);CHKERRQ(ierr); ierr = VecResetArray(tvec);CHKERRQ(ierr); ierr = KSPGetSolution(pcbddc->coarse_ksp,&tvec);CHKERRQ(ierr); ierr = VecGetArray(tvec,&array);CHKERRQ(ierr); ierr = VecPlaceArray(from,array);CHKERRQ(ierr); ierr = VecRestoreArray(tvec,&array);CHKERRQ(ierr); } } else { /* from local to global -> put data in coarse right hand side */ from = pcbddc->vec1_P; to = pcbddc->coarse_vec; } ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,from,to,imode,smode);CHKERRQ(ierr); PetscFunctionReturn(0); } PetscErrorCode PCBDDCScatterCoarseDataEnd(PC pc, InsertMode imode, ScatterMode smode) { PetscErrorCode ierr; PC_BDDC* pcbddc = (PC_BDDC*)(pc->data); PetscScalar *array; Vec from,to; PetscFunctionBegin; if (smode == SCATTER_REVERSE) { /* from global to local -> get data from coarse solution */ from = pcbddc->coarse_vec; to = pcbddc->vec1_P; } else { /* from local to global -> put data in coarse right hand side */ from = pcbddc->vec1_P; to = pcbddc->coarse_vec; } ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,from,to,imode,smode);CHKERRQ(ierr); if (smode == SCATTER_FORWARD) { if (pcbddc->coarse_ksp) { /* get array from coarse processes */ Vec tvec; ierr = KSPGetRhs(pcbddc->coarse_ksp,&tvec);CHKERRQ(ierr); ierr = VecGetArray(to,&array);CHKERRQ(ierr); ierr = VecPlaceArray(tvec,array);CHKERRQ(ierr); ierr = VecRestoreArray(to,&array);CHKERRQ(ierr); } } else { if (pcbddc->coarse_ksp) { /* restore array of pcbddc->coarse_vec */ ierr = VecResetArray(from);CHKERRQ(ierr); } } PetscFunctionReturn(0); } /* uncomment for testing purposes */ /* #define PETSC_MISSING_LAPACK_GESVD 1 */ PetscErrorCode PCBDDCConstraintsSetUp(PC pc) { PetscErrorCode ierr; PC_IS* pcis = (PC_IS*)(pc->data); PC_BDDC* pcbddc = (PC_BDDC*)pc->data; Mat_IS* matis = (Mat_IS*)pc->pmat->data; /* one and zero */ PetscScalar one=1.0,zero=0.0; /* space to store constraints and their local indices */ PetscScalar *constraints_data; PetscInt *constraints_idxs,*constraints_idxs_B; PetscInt *constraints_idxs_ptr,*constraints_data_ptr; PetscInt *constraints_n; /* iterators */ PetscInt i,j,k,total_counts,total_counts_cc,cum; /* BLAS integers */ PetscBLASInt lwork,lierr; PetscBLASInt Blas_N,Blas_M,Blas_K,Blas_one=1; PetscBLASInt Blas_LDA,Blas_LDB,Blas_LDC; /* reuse */ PetscInt olocal_primal_size,olocal_primal_size_cc; PetscInt *olocal_primal_ref_node,*olocal_primal_ref_mult; /* change of basis */ PetscBool qr_needed; PetscBT change_basis,qr_needed_idx; /* auxiliary stuff */ PetscInt *nnz,*is_indices; PetscInt ncc; /* some quantities */ PetscInt n_vertices,total_primal_vertices,valid_constraints; PetscInt size_of_constraint,max_size_of_constraint=0,max_constraints,temp_constraints; PetscFunctionBegin; /* Destroy Mat objects computed previously */ ierr = MatDestroy(&pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->ConstraintMatrix);CHKERRQ(ierr); ierr = MatDestroy(&pcbddc->switch_static_change);CHKERRQ(ierr); /* save info on constraints from previous setup (if any) */ olocal_primal_size = pcbddc->local_primal_size; olocal_primal_size_cc = pcbddc->local_primal_size_cc; ierr = PetscMalloc2(olocal_primal_size_cc,&olocal_primal_ref_node,olocal_primal_size_cc,&olocal_primal_ref_mult);CHKERRQ(ierr); ierr = PetscMemcpy(olocal_primal_ref_node,pcbddc->local_primal_ref_node,olocal_primal_size_cc*sizeof(PetscInt));CHKERRQ(ierr); ierr = PetscMemcpy(olocal_primal_ref_mult,pcbddc->local_primal_ref_mult,olocal_primal_size_cc*sizeof(PetscInt));CHKERRQ(ierr); ierr = PetscFree2(pcbddc->local_primal_ref_node,pcbddc->local_primal_ref_mult);CHKERRQ(ierr); ierr = PetscFree(pcbddc->primal_indices_local_idxs);CHKERRQ(ierr); if (!pcbddc->adaptive_selection) { IS ISForVertices,*ISForFaces,*ISForEdges; MatNullSpace nearnullsp; const Vec *nearnullvecs; Vec *localnearnullsp; PetscScalar *array; PetscInt n_ISForFaces,n_ISForEdges,nnsp_size; PetscBool nnsp_has_cnst; /* LAPACK working arrays for SVD or POD */ PetscBool skip_lapack,boolforchange; PetscScalar *work; PetscReal *singular_vals; #if defined(PETSC_USE_COMPLEX) PetscReal *rwork; #endif #if defined(PETSC_MISSING_LAPACK_GESVD) PetscScalar *temp_basis,*correlation_mat; #else PetscBLASInt dummy_int=1; PetscScalar dummy_scalar=1.; #endif /* Get index sets for faces, edges and vertices from graph */ ierr = PCBDDCGraphGetCandidatesIS(pcbddc->mat_graph,&n_ISForFaces,&ISForFaces,&n_ISForEdges,&ISForEdges,&ISForVertices);CHKERRQ(ierr); /* print some info */ if (pcbddc->dbg_flag && (!pcbddc->sub_schurs || pcbddc->sub_schurs_rebuild)) { PetscInt nv; ierr = PCBDDCGraphASCIIView(pcbddc->mat_graph,pcbddc->dbg_flag,pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = ISGetSize(ISForVertices,&nv);CHKERRQ(ierr); ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"--------------------------------------------------------------\n");CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d got %02d local candidate vertices (%d)\n",PetscGlobalRank,nv,pcbddc->use_vertices);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d got %02d local candidate edges (%d)\n",PetscGlobalRank,n_ISForEdges,pcbddc->use_edges);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d got %02d local candidate faces (%d)\n",PetscGlobalRank,n_ISForFaces,pcbddc->use_faces);CHKERRQ(ierr); ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PetscViewerASCIIPopSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); } /* free unneeded index sets */ if (!pcbddc->use_vertices) { ierr = ISDestroy(&ISForVertices);CHKERRQ(ierr); } if (!pcbddc->use_edges) { for (i=0;iuse_faces) { for (i=0;ipmat,&nearnullsp);CHKERRQ(ierr); if (nearnullsp) { ierr = MatNullSpaceGetVecs(nearnullsp,&nnsp_has_cnst,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); /* remove any stored info */ ierr = MatNullSpaceDestroy(&pcbddc->onearnullspace);CHKERRQ(ierr); ierr = PetscFree(pcbddc->onearnullvecs_state);CHKERRQ(ierr); /* store information for BDDC solver reuse */ ierr = PetscObjectReference((PetscObject)nearnullsp);CHKERRQ(ierr); pcbddc->onearnullspace = nearnullsp; ierr = PetscMalloc1(nnsp_size,&pcbddc->onearnullvecs_state);CHKERRQ(ierr); for (i=0;ionearnullvecs_state[i]);CHKERRQ(ierr); } } else { /* if near null space is not provided BDDC uses constants by default */ nnsp_size = 0; nnsp_has_cnst = PETSC_TRUE; } /* get max number of constraints on a single cc */ max_constraints = nnsp_size; if (nnsp_has_cnst) max_constraints++; /* Evaluate maximum storage size needed by the procedure - Indices for connected component i stored at "constraints_idxs + constraints_idxs_ptr[i]" - Values for constraints on connected component i stored at "constraints_data + constraints_data_ptr[i]" There can be multiple constraints per connected component */ n_vertices = 0; if (ISForVertices) { ierr = ISGetSize(ISForVertices,&n_vertices);CHKERRQ(ierr); } ncc = n_vertices+n_ISForFaces+n_ISForEdges; ierr = PetscMalloc3(ncc+1,&constraints_idxs_ptr,ncc+1,&constraints_data_ptr,ncc,&constraints_n);CHKERRQ(ierr); total_counts = n_ISForFaces+n_ISForEdges; total_counts *= max_constraints; total_counts += n_vertices; ierr = PetscBTCreate(total_counts,&change_basis);CHKERRQ(ierr); total_counts = 0; max_size_of_constraint = 0; for (i=0;ivec1_N,&localnearnullsp[k]);CHKERRQ(ierr); ierr = VecScatterBegin(matis->rctx,nearnullvecs[k],localnearnullsp[k],INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(matis->rctx,nearnullvecs[k],localnearnullsp[k],INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); } /* whether or not to skip lapack calls */ skip_lapack = PETSC_TRUE; if (n_ISForFaces+n_ISForEdges && max_constraints > 1 && !pcbddc->use_nnsp_true) skip_lapack = PETSC_FALSE; /* First we issue queries to allocate optimal workspace for LAPACKgesvd (or LAPACKsyev if SVD is missing) */ if (!skip_lapack) { PetscScalar temp_work; #if defined(PETSC_MISSING_LAPACK_GESVD) /* Proper Orthogonal Decomposition (POD) using the snapshot method */ ierr = PetscMalloc1(max_constraints*max_constraints,&correlation_mat);CHKERRQ(ierr); ierr = PetscMalloc1(max_constraints,&singular_vals);CHKERRQ(ierr); ierr = PetscMalloc1(max_size_of_constraint*max_constraints,&temp_basis);CHKERRQ(ierr); #if defined(PETSC_USE_COMPLEX) ierr = PetscMalloc1(3*max_constraints,&rwork);CHKERRQ(ierr); #endif /* now we evaluate the optimal workspace using query with lwork=-1 */ ierr = PetscBLASIntCast(max_constraints,&Blas_N);CHKERRQ(ierr); ierr = PetscBLASIntCast(max_constraints,&Blas_LDA);CHKERRQ(ierr); lwork = -1; ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); #if !defined(PETSC_USE_COMPLEX) PetscStackCallBLAS("LAPACKsyev",LAPACKsyev_("V","U",&Blas_N,correlation_mat,&Blas_LDA,singular_vals,&temp_work,&lwork,&lierr)); #else PetscStackCallBLAS("LAPACKsyev",LAPACKsyev_("V","U",&Blas_N,correlation_mat,&Blas_LDA,singular_vals,&temp_work,&lwork,rwork,&lierr)); #endif ierr = PetscFPTrapPop();CHKERRQ(ierr); if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in query to SYEV Lapack routine %d",(int)lierr); #else /* on missing GESVD */ /* SVD */ PetscInt max_n,min_n; max_n = max_size_of_constraint; min_n = max_constraints; if (max_size_of_constraint < max_constraints) { min_n = max_size_of_constraint; max_n = max_constraints; } ierr = PetscMalloc1(min_n,&singular_vals);CHKERRQ(ierr); #if defined(PETSC_USE_COMPLEX) ierr = PetscMalloc1(5*min_n,&rwork);CHKERRQ(ierr); #endif /* now we evaluate the optimal workspace using query with lwork=-1 */ lwork = -1; ierr = PetscBLASIntCast(max_n,&Blas_M);CHKERRQ(ierr); ierr = PetscBLASIntCast(min_n,&Blas_N);CHKERRQ(ierr); ierr = PetscBLASIntCast(max_n,&Blas_LDA);CHKERRQ(ierr); ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); #if !defined(PETSC_USE_COMPLEX) PetscStackCallBLAS("LAPACKgesvd",LAPACKgesvd_("O","N",&Blas_M,&Blas_N,&constraints_data[0],&Blas_LDA,singular_vals,&dummy_scalar,&dummy_int,&dummy_scalar,&dummy_int,&temp_work,&lwork,&lierr)); #else PetscStackCallBLAS("LAPACKgesvd",LAPACKgesvd_("O","N",&Blas_M,&Blas_N,&constraints_data[0],&Blas_LDA,singular_vals,&dummy_scalar,&dummy_int,&dummy_scalar,&dummy_int,&temp_work,&lwork,rwork,&lierr)); #endif ierr = PetscFPTrapPop();CHKERRQ(ierr); if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in query to GESVD Lapack routine %d",(int)lierr); #endif /* on missing GESVD */ /* Allocate optimal workspace */ ierr = PetscBLASIntCast((PetscInt)PetscRealPart(temp_work),&lwork);CHKERRQ(ierr); ierr = PetscMalloc1(lwork,&work);CHKERRQ(ierr); } /* Now we can loop on constraining sets */ total_counts = 0; constraints_idxs_ptr[0] = 0; constraints_data_ptr[0] = 0; /* vertices */ if (n_vertices) { ierr = ISGetIndices(ISForVertices,(const PetscInt**)&is_indices);CHKERRQ(ierr); ierr = PetscMemcpy(constraints_idxs,is_indices,n_vertices*sizeof(PetscInt));CHKERRQ(ierr); for (i=0;iuse_change_of_basis; /* change or not the basis on the edge */ } else { used_is = ISForFaces[ncc-n_ISForEdges]; boolforchange = (PetscBool)(pcbddc->use_change_of_basis && pcbddc->use_change_on_faces); /* change or not the basis on the face */ } temp_constraints = 0; /* zero the number of constraints I have on this conn comp */ ierr = ISGetSize(used_is,&size_of_constraint);CHKERRQ(ierr); ierr = ISGetIndices(used_is,(const PetscInt**)&is_indices);CHKERRQ(ierr); /* change of basis should not be performed on local periodic nodes */ if (pcbddc->mat_graph->mirrors && pcbddc->mat_graph->mirrors[is_indices[0]]) boolforchange = PETSC_FALSE; if (nnsp_has_cnst) { PetscScalar quad_value; ierr = PetscMemcpy(constraints_idxs + constraints_idxs_ptr[total_counts_cc],is_indices,size_of_constraint*sizeof(PetscInt));CHKERRQ(ierr); idxs_copied = PETSC_TRUE; if (!pcbddc->use_nnsp_true) { quad_value = (PetscScalar)(1.0/PetscSqrtReal((PetscReal)size_of_constraint)); } else { quad_value = 1.0; } for (j=0;j 0.0) { /* keep indices and values */ temp_constraints++; total_counts++; if (!idxs_copied) { ierr = PetscMemcpy(constraints_idxs + constraints_idxs_ptr[total_counts_cc],is_indices,size_of_constraint*sizeof(PetscInt));CHKERRQ(ierr); idxs_copied = PETSC_TRUE; } } } ierr = ISRestoreIndices(used_is,(const PetscInt**)&is_indices);CHKERRQ(ierr); valid_constraints = temp_constraints; if (!pcbddc->use_nnsp_true && temp_constraints) { if (temp_constraints == 1) { /* just normalize the constraint */ PetscScalar norm,*ptr_to_data; ptr_to_data = &constraints_data[constraints_data_ptr[total_counts_cc]]; ierr = PetscBLASIntCast(size_of_constraint,&Blas_N);CHKERRQ(ierr); PetscStackCallBLAS("BLASdot",norm = BLASdot_(&Blas_N,ptr_to_data,&Blas_one,ptr_to_data,&Blas_one)); norm = 1.0/PetscSqrtReal(PetscRealPart(norm)); PetscStackCallBLAS("BLASscal",BLASscal_(&Blas_N,&norm,ptr_to_data,&Blas_one)); } else { /* perform SVD */ PetscReal tol = 1.0e-8; /* tolerance for retaining eigenmodes */ PetscScalar *ptr_to_data = &constraints_data[constraints_data_ptr[total_counts_cc]]; #if defined(PETSC_MISSING_LAPACK_GESVD) /* SVD: Y = U*S*V^H -> U (eigenvectors of Y*Y^H) = Y*V*(S)^\dag POD: Y^H*Y = V*D*V^H, D = S^H*S -> U = Y*V*D^(-1/2) -> When PETSC_USE_COMPLEX and PETSC_MISSING_LAPACK_GESVD are defined the constraints basis will differ (by a complex factor with absolute value equal to 1) from that computed using LAPACKgesvd -> This is due to a different computation of eigenvectors in LAPACKheev -> The quality of the POD-computed basis will be the same */ ierr = PetscMemzero(correlation_mat,temp_constraints*temp_constraints*sizeof(PetscScalar));CHKERRQ(ierr); /* Store upper triangular part of correlation matrix */ ierr = PetscBLASIntCast(size_of_constraint,&Blas_N);CHKERRQ(ierr); ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); for (j=0;j size_of_constraint) k = size_of_constraint; j = 0; while (j < k && singular_vals[k-j-1] < tol) j++; valid_constraints = k-j; total_counts = total_counts-temp_constraints+valid_constraints; #endif /* on missing GESVD */ } } /* update pointers information */ if (valid_constraints) { constraints_n[total_counts_cc] = valid_constraints; constraints_idxs_ptr[total_counts_cc+1] = constraints_idxs_ptr[total_counts_cc]+size_of_constraint; constraints_data_ptr[total_counts_cc+1] = constraints_data_ptr[total_counts_cc]+size_of_constraint*valid_constraints; /* set change_of_basis flag */ if (boolforchange) { PetscBTSet(change_basis,total_counts_cc); } total_counts_cc++; } } /* free workspace */ if (!skip_lapack) { ierr = PetscFree(work);CHKERRQ(ierr); #if defined(PETSC_USE_COMPLEX) ierr = PetscFree(rwork);CHKERRQ(ierr); #endif ierr = PetscFree(singular_vals);CHKERRQ(ierr); #if defined(PETSC_MISSING_LAPACK_GESVD) ierr = PetscFree(correlation_mat);CHKERRQ(ierr); ierr = PetscFree(temp_basis);CHKERRQ(ierr); #endif } for (k=0;ksub_schurs; total_counts = 0; n_vertices = 0; if (sub_schurs->is_vertices && pcbddc->use_vertices) { ierr = ISGetLocalSize(sub_schurs->is_vertices,&n_vertices);CHKERRQ(ierr); } max_constraints = 0; total_counts_cc = 0; for (i=0;in_subs+n_vertices;i++) { total_counts += pcbddc->adaptive_constraints_n[i]; if (pcbddc->adaptive_constraints_n[i]) total_counts_cc++; max_constraints = PetscMax(max_constraints,pcbddc->adaptive_constraints_n[i]); } constraints_idxs_ptr = pcbddc->adaptive_constraints_idxs_ptr; constraints_data_ptr = pcbddc->adaptive_constraints_data_ptr; constraints_idxs = pcbddc->adaptive_constraints_idxs; constraints_data = pcbddc->adaptive_constraints_data; /* constraints_n differs from pcbddc->adaptive_constraints_n */ ierr = PetscMalloc1(total_counts_cc,&constraints_n);CHKERRQ(ierr); total_counts_cc = 0; for (i=0;in_subs+n_vertices;i++) { if (pcbddc->adaptive_constraints_n[i]) { constraints_n[total_counts_cc++] = pcbddc->adaptive_constraints_n[i]; } } #if 0 printf("Found %d totals (%d)\n",total_counts_cc,total_counts); for (i=0;iadaptive_constraints_n[i]); } for (i=0;in_subs;i++) { PetscPrintf(PETSC_COMM_SELF,"[%d] sub %d, edge %d, n %d\n",PetscGlobalRank,i,(PetscBool)PetscBTLookup(sub_schurs->is_edge,i),pcbddc->adaptive_constraints_n[i+n_vertices]); } #endif max_size_of_constraint = 0; for (i=0;iuse_change_of_basis) { for (i=0;in_subs;i++) { if (PetscBTLookup(sub_schurs->is_edge,i) || pcbddc->use_change_on_faces) { ierr = PetscBTSet(change_basis,i+n_vertices);CHKERRQ(ierr); } } } } pcbddc->local_primal_size = total_counts; ierr = PetscMalloc1(pcbddc->local_primal_size+pcbddc->benign_n,&pcbddc->primal_indices_local_idxs);CHKERRQ(ierr); /* map constraints_idxs in boundary numbering */ ierr = ISGlobalToLocalMappingApply(pcis->BtoNmap,IS_GTOLM_DROP,constraints_idxs_ptr[total_counts_cc],constraints_idxs,&i,constraints_idxs_B);CHKERRQ(ierr); if (i != constraints_idxs_ptr[total_counts_cc]) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Error in boundary numbering for constraints indices %D != %D\n",constraints_idxs_ptr[total_counts_cc],i); /* Create constraint matrix */ ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->ConstraintMatrix);CHKERRQ(ierr); ierr = MatSetType(pcbddc->ConstraintMatrix,MATAIJ);CHKERRQ(ierr); ierr = MatSetSizes(pcbddc->ConstraintMatrix,pcbddc->local_primal_size,pcis->n,pcbddc->local_primal_size,pcis->n);CHKERRQ(ierr); /* find primal_dofs: subdomain corners plus dofs selected as primal after change of basis */ /* determine if a QR strategy is needed for change of basis */ qr_needed = PETSC_FALSE; ierr = PetscBTCreate(total_counts_cc,&qr_needed_idx);CHKERRQ(ierr); total_primal_vertices=0; pcbddc->local_primal_size_cc = 0; for (i=0;imat_graph->custom_minimal_size) { pcbddc->primal_indices_local_idxs[total_primal_vertices++] = constraints_idxs[constraints_idxs_ptr[i]]; pcbddc->local_primal_size_cc += 1; } else if (PetscBTLookup(change_basis,i)) { for (k=0;kprimal_indices_local_idxs[total_primal_vertices++] = constraints_idxs[constraints_idxs_ptr[i]+k]; } pcbddc->local_primal_size_cc += constraints_n[i]; if (constraints_n[i] > 1 || pcbddc->use_qr_single) { PetscBTSet(qr_needed_idx,i); qr_needed = PETSC_TRUE; } } else { pcbddc->local_primal_size_cc += 1; } } /* note that the local variable n_vertices used below stores the number of pointwise constraints */ pcbddc->n_vertices = total_primal_vertices; /* permute indices in order to have a sorted set of vertices */ ierr = PetscSortInt(total_primal_vertices,pcbddc->primal_indices_local_idxs);CHKERRQ(ierr); ierr = PetscMalloc2(pcbddc->local_primal_size_cc+pcbddc->benign_n,&pcbddc->local_primal_ref_node,pcbddc->local_primal_size_cc+pcbddc->benign_n,&pcbddc->local_primal_ref_mult);CHKERRQ(ierr); ierr = PetscMemcpy(pcbddc->local_primal_ref_node,pcbddc->primal_indices_local_idxs,total_primal_vertices*sizeof(PetscInt));CHKERRQ(ierr); for (i=0;ilocal_primal_ref_mult[i] = 1; /* nonzero structure of constraint matrix */ /* and get reference dof for local constraints */ ierr = PetscMalloc1(pcbddc->local_primal_size,&nnz);CHKERRQ(ierr); for (i=0;ilocal_primal_ref_node[cum] = constraints_idxs[constraints_idxs_ptr[i]]; pcbddc->local_primal_ref_mult[cum] = constraints_n[i]; cum++; size_of_constraint = constraints_idxs_ptr[i+1]-constraints_idxs_ptr[i]; for (k=0;kprimal_indices_local_idxs[total_counts++] = constraints_idxs[constraints_idxs_ptr[i]+k]; nnz[j+k] = size_of_constraint; } j += constraints_n[i]; } } ierr = MatSeqAIJSetPreallocation(pcbddc->ConstraintMatrix,0,nnz);CHKERRQ(ierr); ierr = PetscFree(nnz);CHKERRQ(ierr); /* set values in constraint matrix */ for (i=0;iConstraintMatrix,i,pcbddc->local_primal_ref_node[i],1.0,INSERT_VALUES);CHKERRQ(ierr); } total_counts = total_primal_vertices; for (i=n_vertices;iConstraintMatrix,1,&row,size_of_constraint,cols,vals,INSERT_VALUES);CHKERRQ(ierr); } total_counts += constraints_n[i]; } } /* assembling */ ierr = MatAssemblyBegin(pcbddc->ConstraintMatrix,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); ierr = MatAssemblyEnd(pcbddc->ConstraintMatrix,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); /* ierr = PetscViewerPushFormat(PETSC_VIEWER_STDOUT_SELF,PETSC_VIEWER_ASCII_MATLAB);CHKERRQ(ierr); ierr = MatView(pcbddc->ConstraintMatrix,(PetscViewer)0);CHKERRQ(ierr); ierr = PetscViewerPopFormat(PETSC_VIEWER_STDOUT_SELF);CHKERRQ(ierr); */ /* Create matrix for change of basis. We don't need it in case pcbddc->use_change_of_basis is FALSE */ if (pcbddc->use_change_of_basis) { /* dual and primal dofs on a single cc */ PetscInt dual_dofs,primal_dofs; /* working stuff for GEQRF */ PetscScalar *qr_basis,*qr_tau = NULL,*qr_work,lqr_work_t; PetscBLASInt lqr_work; /* working stuff for UNGQR */ PetscScalar *gqr_work,lgqr_work_t; PetscBLASInt lgqr_work; /* working stuff for TRTRS */ PetscScalar *trs_rhs; PetscBLASInt Blas_NRHS; /* pointers for values insertion into change of basis matrix */ PetscInt *start_rows,*start_cols; PetscScalar *start_vals; /* working stuff for values insertion */ PetscBT is_primal; PetscInt *aux_primal_numbering_B; /* matrix sizes */ PetscInt global_size,local_size; /* temporary change of basis */ Mat localChangeOfBasisMatrix; /* extra space for debugging */ PetscScalar *dbg_work; /* local temporary change of basis acts on local interfaces -> dimension is n_B x n_B */ ierr = MatCreate(PETSC_COMM_SELF,&localChangeOfBasisMatrix);CHKERRQ(ierr); ierr = MatSetType(localChangeOfBasisMatrix,MATAIJ);CHKERRQ(ierr); ierr = MatSetSizes(localChangeOfBasisMatrix,pcis->n,pcis->n,pcis->n,pcis->n);CHKERRQ(ierr); /* nonzeros for local mat */ ierr = PetscMalloc1(pcis->n,&nnz);CHKERRQ(ierr); if (!pcbddc->benign_change || pcbddc->fake_change) { for (i=0;in;i++) nnz[i]=1; } else { const PetscInt *ii; PetscInt n; PetscBool flg_row; ierr = MatGetRowIJ(pcbddc->benign_change,0,PETSC_FALSE,PETSC_FALSE,&n,&ii,NULL,&flg_row);CHKERRQ(ierr); for (i=0;ibenign_change,0,PETSC_FALSE,PETSC_FALSE,&n,&ii,NULL,&flg_row);CHKERRQ(ierr); } for (i=n_vertices;ibenign_change || pcbddc->fake_change) { for (i=0;in;i++) { ierr = MatSetValue(localChangeOfBasisMatrix,i,i,1.0,INSERT_VALUES);CHKERRQ(ierr); } } else { const PetscInt *ii,*jj; PetscScalar *aa; PetscInt n; PetscBool flg_row; ierr = MatGetRowIJ(pcbddc->benign_change,0,PETSC_FALSE,PETSC_FALSE,&n,&ii,&jj,&flg_row);CHKERRQ(ierr); ierr = MatSeqAIJGetArray(pcbddc->benign_change,&aa);CHKERRQ(ierr); for (i=0;ibenign_change,&aa);CHKERRQ(ierr); ierr = MatRestoreRowIJ(pcbddc->benign_change,0,PETSC_FALSE,PETSC_FALSE,&n,&ii,&jj,&flg_row);CHKERRQ(ierr); } if (pcbddc->dbg_flag) { ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"--------------------------------------------------------------\n");CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Checking change of basis computation for subdomain %04d\n",PetscGlobalRank);CHKERRQ(ierr); } /* Now we loop on the constraints which need a change of basis */ /* Change of basis matrix is evaluated similarly to the FIRST APPROACH in Klawonn and Widlund, Dual-primal FETI-DP methods for linear elasticity, (see Sect 6.2.1) Basic blocks of change of basis matrix T computed by - Using the following block transformation if there is only a primal dof on the cc (and -pc_bddc_use_qr_single is not specified) | 1 0 ... 0 s_1/S | | 0 1 ... 0 s_2/S | | ... | | 0 ... 1 s_{n-1}/S | | -s_1/s_n ... -s_{n-1}/s_n s_n/S | with S = \sum_{i=1}^n s_i^2 NOTE: in the above example, the primal dof is the last one of the edge in LOCAL ordering in the current implementation, the primal dof is the first one of the edge in GLOBAL ordering - QR decomposition of constraints otherwise */ if (qr_needed) { /* space to store Q */ ierr = PetscMalloc1(max_size_of_constraint*max_size_of_constraint,&qr_basis);CHKERRQ(ierr); /* array to store scaling factors for reflectors */ ierr = PetscMalloc1(max_constraints,&qr_tau);CHKERRQ(ierr); /* first we issue queries for optimal work */ ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_M);CHKERRQ(ierr); ierr = PetscBLASIntCast(max_constraints,&Blas_N);CHKERRQ(ierr); ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_LDA);CHKERRQ(ierr); lqr_work = -1; PetscStackCallBLAS("LAPACKgeqrf",LAPACKgeqrf_(&Blas_M,&Blas_N,qr_basis,&Blas_LDA,qr_tau,&lqr_work_t,&lqr_work,&lierr)); if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in query to GEQRF Lapack routine %d",(int)lierr); ierr = PetscBLASIntCast((PetscInt)PetscRealPart(lqr_work_t),&lqr_work);CHKERRQ(ierr); ierr = PetscMalloc1((PetscInt)PetscRealPart(lqr_work_t),&qr_work);CHKERRQ(ierr); lgqr_work = -1; ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_M);CHKERRQ(ierr); ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_N);CHKERRQ(ierr); ierr = PetscBLASIntCast(max_constraints,&Blas_K);CHKERRQ(ierr); ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_LDA);CHKERRQ(ierr); if (Blas_K>Blas_M) Blas_K=Blas_M; /* adjust just for computing optimal work */ PetscStackCallBLAS("LAPACKungqr",LAPACKungqr_(&Blas_M,&Blas_N,&Blas_K,qr_basis,&Blas_LDA,qr_tau,&lgqr_work_t,&lgqr_work,&lierr)); if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in query to UNGQR Lapack routine %d",(int)lierr); ierr = PetscBLASIntCast((PetscInt)PetscRealPart(lgqr_work_t),&lgqr_work);CHKERRQ(ierr); ierr = PetscMalloc1((PetscInt)PetscRealPart(lgqr_work_t),&gqr_work);CHKERRQ(ierr); /* array to store rhs and solution of triangular solver */ ierr = PetscMalloc1(max_constraints*max_constraints,&trs_rhs);CHKERRQ(ierr); /* allocating workspace for check */ if (pcbddc->dbg_flag) { ierr = PetscMalloc1(max_size_of_constraint*(max_constraints+max_size_of_constraint),&dbg_work);CHKERRQ(ierr); } } /* array to store whether a node is primal or not */ ierr = PetscBTCreate(pcis->n_B,&is_primal);CHKERRQ(ierr); ierr = PetscMalloc1(total_primal_vertices,&aux_primal_numbering_B);CHKERRQ(ierr); ierr = ISGlobalToLocalMappingApply(pcis->BtoNmap,IS_GTOLM_DROP,total_primal_vertices,pcbddc->local_primal_ref_node,&i,aux_primal_numbering_B);CHKERRQ(ierr); if (i != total_primal_vertices) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Error in boundary numbering for BDDC vertices! %D != %D\n",total_primal_vertices,i); for (i=0;idbg_flag) { ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Constraints %d: %d need a change of basis (size %d)\n",total_counts,primal_dofs,size_of_constraint);CHKERRQ(ierr); } if (PetscBTLookup(qr_needed_idx,total_counts)) { /* QR */ /* copy quadrature constraints for change of basis check */ if (pcbddc->dbg_flag) { ierr = PetscMemcpy(dbg_work,&constraints_data[constraints_data_ptr[total_counts]],size_of_constraint*primal_dofs*sizeof(PetscScalar));CHKERRQ(ierr); } /* copy temporary constraints into larger work vector (in order to store all columns of Q) */ ierr = PetscMemcpy(qr_basis,&constraints_data[constraints_data_ptr[total_counts]],size_of_constraint*primal_dofs*sizeof(PetscScalar));CHKERRQ(ierr); /* compute QR decomposition of constraints */ ierr = PetscBLASIntCast(size_of_constraint,&Blas_M);CHKERRQ(ierr); ierr = PetscBLASIntCast(primal_dofs,&Blas_N);CHKERRQ(ierr); ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr); ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); PetscStackCallBLAS("LAPACKgeqrf",LAPACKgeqrf_(&Blas_M,&Blas_N,qr_basis,&Blas_LDA,qr_tau,qr_work,&lqr_work,&lierr)); if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in GEQRF Lapack routine %d",(int)lierr); ierr = PetscFPTrapPop();CHKERRQ(ierr); /* explictly compute R^-T */ ierr = PetscMemzero(trs_rhs,primal_dofs*primal_dofs*sizeof(*trs_rhs));CHKERRQ(ierr); for (j=0;jdbg_flag) { PetscInt ii,jj; PetscBool valid_qr=PETSC_TRUE; ierr = PetscBLASIntCast(primal_dofs,&Blas_M);CHKERRQ(ierr); ierr = PetscBLASIntCast(size_of_constraint,&Blas_N);CHKERRQ(ierr); ierr = PetscBLASIntCast(size_of_constraint,&Blas_K);CHKERRQ(ierr); ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr); ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDB);CHKERRQ(ierr); ierr = PetscBLASIntCast(primal_dofs,&Blas_LDC);CHKERRQ(ierr); ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); PetscStackCallBLAS("BLASgemm",BLASgemm_("T","N",&Blas_M,&Blas_N,&Blas_K,&one,dbg_work,&Blas_LDA,qr_basis,&Blas_LDB,&zero,&dbg_work[size_of_constraint*primal_dofs],&Blas_LDC)); ierr = PetscFPTrapPop();CHKERRQ(ierr); for (jj=0;jj 1.e-12) valid_qr = PETSC_FALSE; if (ii == jj && PetscAbsScalar(dbg_work[size_of_constraint*primal_dofs+jj*primal_dofs+ii]-1.0) > 1.e-12) valid_qr = PETSC_FALSE; } } if (!valid_qr) { ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"\t-> wrong change of basis!\n");CHKERRQ(ierr); for (jj=0;jj 1.e-12) { PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"\tQr basis function %d is not orthogonal to constraint %d (%1.14e)!\n",jj,ii,PetscAbsScalar(dbg_work[size_of_constraint*primal_dofs+jj*primal_dofs+ii])); } if (ii == jj && PetscAbsScalar(dbg_work[size_of_constraint*primal_dofs+jj*primal_dofs+ii]-1.0) > 1.e-12) { PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"\tQr basis function %d is not unitary w.r.t constraint %d (%1.14e)!\n",jj,ii,PetscAbsScalar(dbg_work[size_of_constraint*primal_dofs+jj*primal_dofs+ii])); } } } } else { ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"\t-> right change of basis!\n");CHKERRQ(ierr); } } } else { /* simple transformation block */ PetscInt row,col; PetscScalar val,norm; ierr = PetscBLASIntCast(size_of_constraint,&Blas_N);CHKERRQ(ierr); PetscStackCallBLAS("BLASdot",norm = BLASdot_(&Blas_N,constraints_data+constraints_data_ptr[total_counts],&Blas_one,constraints_data+constraints_data_ptr[total_counts],&Blas_one)); for (j=0;jdbg_flag) { ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"\t-> using standard change of basis\n");CHKERRQ(ierr); } } } else { if (pcbddc->dbg_flag) { ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Constraint %d does not need a change of basis (size %d)\n",total_counts,size_of_constraint);CHKERRQ(ierr); } } } /* free workspace */ if (qr_needed) { if (pcbddc->dbg_flag) { ierr = PetscFree(dbg_work);CHKERRQ(ierr); } ierr = PetscFree(trs_rhs);CHKERRQ(ierr); ierr = PetscFree(qr_tau);CHKERRQ(ierr); ierr = PetscFree(qr_work);CHKERRQ(ierr); ierr = PetscFree(gqr_work);CHKERRQ(ierr); ierr = PetscFree(qr_basis);CHKERRQ(ierr); } ierr = PetscBTDestroy(&is_primal);CHKERRQ(ierr); ierr = MatAssemblyBegin(localChangeOfBasisMatrix,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); ierr = MatAssemblyEnd(localChangeOfBasisMatrix,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); /* assembling of global change of variable */ if (!pcbddc->fake_change) { Mat tmat; PetscInt bs; ierr = VecGetSize(pcis->vec1_global,&global_size);CHKERRQ(ierr); ierr = VecGetLocalSize(pcis->vec1_global,&local_size);CHKERRQ(ierr); ierr = MatDuplicate(pc->pmat,MAT_DO_NOT_COPY_VALUES,&tmat);CHKERRQ(ierr); ierr = MatISSetLocalMat(tmat,localChangeOfBasisMatrix);CHKERRQ(ierr); ierr = MatCreate(PetscObjectComm((PetscObject)pc),&pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); ierr = MatSetType(pcbddc->ChangeOfBasisMatrix,MATAIJ);CHKERRQ(ierr); ierr = MatGetBlockSize(pc->pmat,&bs);CHKERRQ(ierr); ierr = MatSetBlockSize(pcbddc->ChangeOfBasisMatrix,bs);CHKERRQ(ierr); ierr = MatSetSizes(pcbddc->ChangeOfBasisMatrix,local_size,local_size,global_size,global_size);CHKERRQ(ierr); ierr = MatISSetMPIXAIJPreallocation_Private(tmat,pcbddc->ChangeOfBasisMatrix,PETSC_TRUE);CHKERRQ(ierr); ierr = MatISGetMPIXAIJ(tmat,MAT_REUSE_MATRIX,&pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); ierr = MatDestroy(&tmat);CHKERRQ(ierr); ierr = VecSet(pcis->vec1_global,0.0);CHKERRQ(ierr); ierr = VecSet(pcis->vec1_N,1.0);CHKERRQ(ierr); ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,pcis->vec1_global,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,pcis->vec1_global,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecReciprocal(pcis->vec1_global);CHKERRQ(ierr); ierr = MatDiagonalScale(pcbddc->ChangeOfBasisMatrix,pcis->vec1_global,NULL);CHKERRQ(ierr); /* check */ if (pcbddc->dbg_flag) { PetscReal error; Vec x,x_change; ierr = VecDuplicate(pcis->vec1_global,&x);CHKERRQ(ierr); ierr = VecDuplicate(pcis->vec1_global,&x_change);CHKERRQ(ierr); ierr = VecSetRandom(x,NULL);CHKERRQ(ierr); ierr = VecCopy(x,pcis->vec1_global);CHKERRQ(ierr); ierr = VecScatterBegin(matis->rctx,x,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(matis->rctx,x,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = MatMult(localChangeOfBasisMatrix,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); ierr = VecScatterBegin(matis->rctx,pcis->vec2_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(matis->rctx,pcis->vec2_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcis->vec1_global,x_change);CHKERRQ(ierr); ierr = VecAXPY(x,-1.0,x_change);CHKERRQ(ierr); ierr = VecNorm(x,NORM_INFINITY,&error);CHKERRQ(ierr); if (error > PETSC_SMALL) { SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_PLIB,"Error global vs local change on N: %1.6e\n",error); } ierr = VecDestroy(&x);CHKERRQ(ierr); ierr = VecDestroy(&x_change);CHKERRQ(ierr); } /* adapt sub_schurs computed (if any) */ if (pcbddc->use_deluxe_scaling) { PCBDDCSubSchurs sub_schurs=pcbddc->sub_schurs; if (pcbddc->use_change_of_basis && pcbddc->adaptive_userdefined) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot mix automatic change of basis, adaptive selection and user-defined constraints");CHKERRQ(ierr); if (sub_schurs && sub_schurs->S_Ej_all) { Mat S_new,tmat; IS is_all_N,is_V_Sall = NULL; ierr = ISLocalToGlobalMappingApplyIS(pcis->BtoNmap,sub_schurs->is_Ej_all,&is_all_N);CHKERRQ(ierr); ierr = MatCreateSubMatrix(localChangeOfBasisMatrix,is_all_N,is_all_N,MAT_INITIAL_MATRIX,&tmat);CHKERRQ(ierr); if (pcbddc->deluxe_zerorows) { ISLocalToGlobalMapping NtoSall; IS is_V; ierr = ISCreateGeneral(PETSC_COMM_SELF,pcbddc->n_vertices,pcbddc->local_primal_ref_node,PETSC_COPY_VALUES,&is_V);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingCreateIS(is_all_N,&NtoSall);CHKERRQ(ierr); ierr = ISGlobalToLocalMappingApplyIS(NtoSall,IS_GTOLM_DROP,is_V,&is_V_Sall);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingDestroy(&NtoSall);CHKERRQ(ierr); ierr = ISDestroy(&is_V);CHKERRQ(ierr); } ierr = ISDestroy(&is_all_N);CHKERRQ(ierr); ierr = MatPtAP(sub_schurs->S_Ej_all,tmat,MAT_INITIAL_MATRIX,1.0,&S_new);CHKERRQ(ierr); ierr = MatDestroy(&sub_schurs->S_Ej_all);CHKERRQ(ierr); ierr = PetscObjectReference((PetscObject)S_new);CHKERRQ(ierr); if (pcbddc->deluxe_zerorows) { const PetscScalar *array; const PetscInt *idxs_V,*idxs_all; PetscInt i,n_V; ierr = MatZeroRowsColumnsIS(S_new,is_V_Sall,1.,NULL,NULL);CHKERRQ(ierr); ierr = ISGetLocalSize(is_V_Sall,&n_V);CHKERRQ(ierr); ierr = ISGetIndices(is_V_Sall,&idxs_V);CHKERRQ(ierr); ierr = ISGetIndices(sub_schurs->is_Ej_all,&idxs_all);CHKERRQ(ierr); ierr = VecGetArrayRead(pcis->D,&array);CHKERRQ(ierr); for (i=0;iD,&array);CHKERRQ(ierr); ierr = ISRestoreIndices(sub_schurs->is_Ej_all,&idxs_all);CHKERRQ(ierr); ierr = ISRestoreIndices(is_V_Sall,&idxs_V);CHKERRQ(ierr); } sub_schurs->S_Ej_all = S_new; ierr = MatDestroy(&S_new);CHKERRQ(ierr); if (sub_schurs->sum_S_Ej_all) { ierr = MatPtAP(sub_schurs->sum_S_Ej_all,tmat,MAT_INITIAL_MATRIX,1.0,&S_new);CHKERRQ(ierr); ierr = MatDestroy(&sub_schurs->sum_S_Ej_all);CHKERRQ(ierr); ierr = PetscObjectReference((PetscObject)S_new);CHKERRQ(ierr); if (pcbddc->deluxe_zerorows) { ierr = MatZeroRowsColumnsIS(S_new,is_V_Sall,1.,NULL,NULL);CHKERRQ(ierr); } sub_schurs->sum_S_Ej_all = S_new; ierr = MatDestroy(&S_new);CHKERRQ(ierr); } ierr = ISDestroy(&is_V_Sall);CHKERRQ(ierr); ierr = MatDestroy(&tmat);CHKERRQ(ierr); } /* destroy any change of basis context in sub_schurs */ if (sub_schurs && sub_schurs->change) { PetscInt i; for (i=0;in_subs;i++) { ierr = KSPDestroy(&sub_schurs->change[i]);CHKERRQ(ierr); } ierr = PetscFree(sub_schurs->change);CHKERRQ(ierr); } } if (pcbddc->switch_static) { /* need to save the local change */ pcbddc->switch_static_change = localChangeOfBasisMatrix; } else { ierr = MatDestroy(&localChangeOfBasisMatrix);CHKERRQ(ierr); } /* determine if any process has changed the pressures locally */ pcbddc->change_interior = pcbddc->benign_have_null; } else { /* fake change (get back change of basis into ConstraintMatrix and info on qr) */ ierr = MatDestroy(&pcbddc->ConstraintMatrix);CHKERRQ(ierr); pcbddc->ConstraintMatrix = localChangeOfBasisMatrix; pcbddc->use_qr_single = qr_needed; } } else if (pcbddc->user_ChangeOfBasisMatrix || pcbddc->benign_saddle_point) { if (!pcbddc->benign_have_null && pcbddc->user_ChangeOfBasisMatrix) { ierr = PetscObjectReference((PetscObject)pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); pcbddc->ChangeOfBasisMatrix = pcbddc->user_ChangeOfBasisMatrix; } else { Mat benign_global = NULL; if (pcbddc->benign_have_null) { Mat tmat; pcbddc->change_interior = PETSC_TRUE; ierr = VecSet(pcis->vec1_global,0.0);CHKERRQ(ierr); ierr = VecSet(pcis->vec1_N,1.0);CHKERRQ(ierr); ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,pcis->vec1_global,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,pcis->vec1_global,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecReciprocal(pcis->vec1_global);CHKERRQ(ierr); ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = MatDuplicate(pc->pmat,MAT_DO_NOT_COPY_VALUES,&tmat);CHKERRQ(ierr); if (pcbddc->benign_change) { Mat M; ierr = MatDuplicate(pcbddc->benign_change,MAT_COPY_VALUES,&M);CHKERRQ(ierr); ierr = MatDiagonalScale(M,pcis->vec1_N,NULL);CHKERRQ(ierr); ierr = MatISSetLocalMat(tmat,M);CHKERRQ(ierr); ierr = MatDestroy(&M);CHKERRQ(ierr); } else { Mat eye; PetscScalar *array; ierr = VecGetArray(pcis->vec1_N,&array);CHKERRQ(ierr); ierr = MatCreateSeqAIJ(PETSC_COMM_SELF,pcis->n,pcis->n,1,NULL,&eye);CHKERRQ(ierr); for (i=0;in;i++) { ierr = MatSetValue(eye,i,i,array[i],INSERT_VALUES);CHKERRQ(ierr); } ierr = VecRestoreArray(pcis->vec1_N,&array);CHKERRQ(ierr); ierr = MatAssemblyBegin(eye,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); ierr = MatAssemblyEnd(eye,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); ierr = MatISSetLocalMat(tmat,eye);CHKERRQ(ierr); ierr = MatDestroy(&eye);CHKERRQ(ierr); } ierr = MatISGetMPIXAIJ(tmat,MAT_INITIAL_MATRIX,&benign_global);CHKERRQ(ierr); ierr = MatDestroy(&tmat);CHKERRQ(ierr); } if (pcbddc->user_ChangeOfBasisMatrix) { ierr = MatMatMult(pcbddc->user_ChangeOfBasisMatrix,benign_global,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); ierr = MatDestroy(&benign_global);CHKERRQ(ierr); } else if (pcbddc->benign_have_null) { pcbddc->ChangeOfBasisMatrix = benign_global; } } if (pcbddc->switch_static && pcbddc->ChangeOfBasisMatrix) { /* need to save the local change */ IS is_global; const PetscInt *gidxs; ierr = ISLocalToGlobalMappingGetIndices(pc->pmat->rmap->mapping,&gidxs);CHKERRQ(ierr); ierr = ISCreateGeneral(PetscObjectComm((PetscObject)pc),pcis->n,gidxs,PETSC_COPY_VALUES,&is_global);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingRestoreIndices(pc->pmat->rmap->mapping,&gidxs);CHKERRQ(ierr); ierr = MatCreateSubMatrixUnsorted(pcbddc->ChangeOfBasisMatrix,is_global,is_global,&pcbddc->switch_static_change);CHKERRQ(ierr); ierr = ISDestroy(&is_global);CHKERRQ(ierr); } } if (!pcbddc->fake_change && pcbddc->ChangeOfBasisMatrix && !pcbddc->work_change) { ierr = VecDuplicate(pcis->vec1_global,&pcbddc->work_change);CHKERRQ(ierr); } if (!pcbddc->fake_change) { /* add pressure dofs to set of primal nodes for numbering purposes */ for (i=0;ibenign_n;i++) { pcbddc->local_primal_ref_node[pcbddc->local_primal_size_cc] = pcbddc->benign_p0_lidx[i]; pcbddc->primal_indices_local_idxs[pcbddc->local_primal_size] = pcbddc->benign_p0_lidx[i]; pcbddc->local_primal_ref_mult[pcbddc->local_primal_size_cc] = 1; pcbddc->local_primal_size_cc++; pcbddc->local_primal_size++; } /* check if a new primal space has been introduced (also take into account benign trick) */ pcbddc->new_primal_space_local = PETSC_TRUE; if (olocal_primal_size == pcbddc->local_primal_size) { ierr = PetscMemcmp(pcbddc->local_primal_ref_node,olocal_primal_ref_node,olocal_primal_size_cc*sizeof(PetscInt),&pcbddc->new_primal_space_local);CHKERRQ(ierr); pcbddc->new_primal_space_local = (PetscBool)(!pcbddc->new_primal_space_local); if (!pcbddc->new_primal_space_local) { ierr = PetscMemcmp(pcbddc->local_primal_ref_mult,olocal_primal_ref_mult,olocal_primal_size_cc*sizeof(PetscInt),&pcbddc->new_primal_space_local);CHKERRQ(ierr); pcbddc->new_primal_space_local = (PetscBool)(!pcbddc->new_primal_space_local); } } /* new_primal_space will be used for numbering of coarse dofs, so it should be the same across all subdomains */ ierr = MPIU_Allreduce(&pcbddc->new_primal_space_local,&pcbddc->new_primal_space,1,MPIU_BOOL,MPI_LOR,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); } ierr = PetscFree2(olocal_primal_ref_node,olocal_primal_ref_mult);CHKERRQ(ierr); /* flush dbg viewer */ if (pcbddc->dbg_flag) { ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); } /* free workspace */ ierr = PetscBTDestroy(&qr_needed_idx);CHKERRQ(ierr); ierr = PetscBTDestroy(&change_basis);CHKERRQ(ierr); if (!pcbddc->adaptive_selection) { ierr = PetscFree3(constraints_idxs_ptr,constraints_data_ptr,constraints_n);CHKERRQ(ierr); ierr = PetscFree3(constraints_data,constraints_idxs,constraints_idxs_B);CHKERRQ(ierr); } else { ierr = PetscFree5(pcbddc->adaptive_constraints_n, pcbddc->adaptive_constraints_idxs_ptr, pcbddc->adaptive_constraints_data_ptr, pcbddc->adaptive_constraints_idxs, pcbddc->adaptive_constraints_data);CHKERRQ(ierr); ierr = PetscFree(constraints_n);CHKERRQ(ierr); ierr = PetscFree(constraints_idxs_B);CHKERRQ(ierr); } PetscFunctionReturn(0); } PetscErrorCode PCBDDCAnalyzeInterface(PC pc) { ISLocalToGlobalMapping map; PC_BDDC *pcbddc = (PC_BDDC*)pc->data; Mat_IS *matis = (Mat_IS*)pc->pmat->data; PetscInt i,N; PetscBool rcsr = PETSC_FALSE; PetscErrorCode ierr; PetscFunctionBegin; if (pcbddc->recompute_topography) { pcbddc->graphanalyzed = PETSC_FALSE; /* Reset previously computed graph */ ierr = PCBDDCGraphReset(pcbddc->mat_graph);CHKERRQ(ierr); /* Init local Graph struct */ ierr = MatGetSize(pc->pmat,&N,NULL);CHKERRQ(ierr); ierr = MatGetLocalToGlobalMapping(pc->pmat,&map,NULL);CHKERRQ(ierr); ierr = PCBDDCGraphInit(pcbddc->mat_graph,map,N,pcbddc->graphmaxcount);CHKERRQ(ierr); if (pcbddc->user_primal_vertices_local && !pcbddc->user_primal_vertices) { ierr = PCBDDCConsistencyCheckIS(pc,MPI_LOR,&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); } /* Check validity of the csr graph passed in by the user */ if (pcbddc->mat_graph->nvtxs_csr && pcbddc->mat_graph->nvtxs_csr != pcbddc->mat_graph->nvtxs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Invalid size of local CSR graph! Found %d, expected %d\n",pcbddc->mat_graph->nvtxs_csr,pcbddc->mat_graph->nvtxs); /* Set default CSR adjacency of local dofs if not provided by the user with PCBDDCSetLocalAdjacencyGraph */ if (!pcbddc->mat_graph->xadj && pcbddc->use_local_adj) { PetscInt *xadj,*adjncy; PetscInt nvtxs; PetscBool flg_row=PETSC_FALSE; ierr = MatGetRowIJ(matis->A,0,PETSC_TRUE,PETSC_FALSE,&nvtxs,(const PetscInt**)&xadj,(const PetscInt**)&adjncy,&flg_row);CHKERRQ(ierr); if (flg_row) { ierr = PCBDDCSetLocalAdjacencyGraph(pc,nvtxs,xadj,adjncy,PETSC_COPY_VALUES);CHKERRQ(ierr); pcbddc->computed_rowadj = PETSC_TRUE; } ierr = MatRestoreRowIJ(matis->A,0,PETSC_TRUE,PETSC_FALSE,&nvtxs,(const PetscInt**)&xadj,(const PetscInt**)&adjncy,&flg_row);CHKERRQ(ierr); rcsr = PETSC_TRUE; } if (pcbddc->dbg_flag) { ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); } /* Setup of Graph */ pcbddc->mat_graph->commsizelimit = 0; /* don't use the COMM_SELF variant of the graph */ ierr = PCBDDCGraphSetUp(pcbddc->mat_graph,pcbddc->vertex_size,pcbddc->NeumannBoundariesLocal,pcbddc->DirichletBoundariesLocal,pcbddc->n_ISForDofsLocal,pcbddc->ISForDofsLocal,pcbddc->user_primal_vertices_local);CHKERRQ(ierr); /* attach info on disconnected subdomains if present */ if (pcbddc->n_local_subs) { PetscInt *local_subs; ierr = PetscMalloc1(N,&local_subs);CHKERRQ(ierr); for (i=0;in_local_subs;i++) { const PetscInt *idxs; PetscInt nl,j; ierr = ISGetLocalSize(pcbddc->local_subs[i],&nl);CHKERRQ(ierr); ierr = ISGetIndices(pcbddc->local_subs[i],&idxs);CHKERRQ(ierr); for (j=0;jlocal_subs[i],&idxs);CHKERRQ(ierr); } pcbddc->mat_graph->n_local_subs = pcbddc->n_local_subs; pcbddc->mat_graph->local_subs = local_subs; } } if (!pcbddc->graphanalyzed) { /* Graph's connected components analysis */ ierr = PCBDDCGraphComputeConnectedComponents(pcbddc->mat_graph);CHKERRQ(ierr); pcbddc->graphanalyzed = PETSC_TRUE; } if (rcsr) pcbddc->mat_graph->nvtxs_csr = 0; PetscFunctionReturn(0); } PetscErrorCode PCBDDCOrthonormalizeVecs(PetscInt n, Vec vecs[]) { PetscInt i,j; PetscScalar *alphas; PetscErrorCode ierr; PetscFunctionBegin; ierr = PetscMalloc1(n,&alphas);CHKERRQ(ierr); for (i=0;i just shift the matrix number of subdomains requested 1 -> send to master or first candidate in voids */ ierr = MatGetSize(mat,&N,NULL);CHKERRQ(ierr); if (active_procs < *n_subdomains || *n_subdomains == 1 || N <= *n_subdomains) { PetscInt issize,isidx,dest; if (*n_subdomains == 1) dest = 0; else dest = rank; if (im_active) { issize = 1; if (procs_candidates) { /* shift the pattern on non-active candidates (if any) */ isidx = procs_candidates[dest]; } else { isidx = dest; } } else { issize = 0; isidx = -1; } if (*n_subdomains != 1) *n_subdomains = active_procs; ierr = ISCreateGeneral(PetscObjectComm((PetscObject)mat),issize,&isidx,PETSC_COPY_VALUES,is_sends);CHKERRQ(ierr); ierr = PetscFree(procs_candidates);CHKERRQ(ierr); PetscFunctionReturn(0); } ierr = PetscOptionsGetBool(NULL,NULL,"-matis_partitioning_use_vwgt",&use_vwgt,NULL);CHKERRQ(ierr); ierr = PetscOptionsGetInt(NULL,NULL,"-matis_partitioning_threshold",&threshold,NULL);CHKERRQ(ierr); threshold = PetscMax(threshold,2); /* Get info on mapping */ ierr = ISLocalToGlobalMappingGetInfo(mat->rmap->mapping,&n_neighs,&neighs,&n_shared,&shared);CHKERRQ(ierr); /* build local CSR graph of subdomains' connectivity */ ierr = PetscMalloc1(2,&xadj);CHKERRQ(ierr); xadj[0] = 0; xadj[1] = PetscMax(n_neighs-1,0); ierr = PetscMalloc1(xadj[1],&adjncy);CHKERRQ(ierr); ierr = PetscMalloc1(xadj[1],&adjncy_wgt);CHKERRQ(ierr); ierr = PetscCalloc1(n,&count);CHKERRQ(ierr); for (i=1;irmap->mapping,&n_neighs,&neighs,&n_shared,&shared);CHKERRQ(ierr); ierr = PetscSortIntWithArray(xadj[1],adjncy,adjncy_wgt);CHKERRQ(ierr); ierr = PetscMalloc1(1,&ranks_send_to_idx);CHKERRQ(ierr); /* Restrict work on active processes only */ ierr = PetscMPIIntCast(im_active,&color);CHKERRQ(ierr); if (void_procs) { ierr = PetscSubcommCreate(PetscObjectComm((PetscObject)mat),&psubcomm);CHKERRQ(ierr); ierr = PetscSubcommSetNumber(psubcomm,2);CHKERRQ(ierr); /* 2 groups, active process and not active processes */ ierr = PetscSubcommSetTypeGeneral(psubcomm,color,rank);CHKERRQ(ierr); subcomm = PetscSubcommChild(psubcomm); } else { psubcomm = NULL; subcomm = PetscObjectComm((PetscObject)mat); } v_wgt = NULL; if (!color) { ierr = PetscFree(xadj);CHKERRQ(ierr); ierr = PetscFree(adjncy);CHKERRQ(ierr); ierr = PetscFree(adjncy_wgt);CHKERRQ(ierr); } else { Mat subdomain_adj; IS new_ranks,new_ranks_contig; MatPartitioning partitioner; PetscInt rstart=0,rend=0; PetscInt *is_indices,*oldranks; PetscMPIInt size; PetscBool aggregate; ierr = MPI_Comm_size(subcomm,&size);CHKERRQ(ierr); if (void_procs) { PetscInt prank = rank; ierr = PetscMalloc1(size,&oldranks);CHKERRQ(ierr); ierr = MPI_Allgather(&prank,1,MPIU_INT,oldranks,1,MPIU_INT,subcomm);CHKERRQ(ierr); for (i=0;i 0 && redprocs < size) ? PETSC_TRUE : PETSC_FALSE); if (aggregate) { /* TODO: all this part could be made more efficient */ PetscInt lrows,row,ncols,*cols; PetscMPIInt nrank; PetscScalar *vals; ierr = MPI_Comm_rank(subcomm,&nrank);CHKERRQ(ierr); lrows = 0; if (nrank 1) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Just 1 vector supported"); PetscValidHeaderSpecific(nnsp_vec[0],VEC_CLASSID,11); } /* further checks */ ierr = MatISGetLocalMat(mat,&local_mat);CHKERRQ(ierr); ierr = PetscObjectTypeCompare((PetscObject)local_mat,MATSEQDENSE,&isdense);CHKERRQ(ierr); if (!isdense) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Currently cannot subassemble MATIS when local matrix type is not of type SEQDENSE"); ierr = MatGetSize(local_mat,&rows,&cols);CHKERRQ(ierr); if (rows != cols) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Local MATIS matrices should be square"); if (reuse && *mat_n) { PetscInt mrows,mcols,mnrows,mncols; PetscValidHeaderSpecific(*mat_n,MAT_CLASSID,7); ierr = PetscObjectTypeCompare((PetscObject)*mat_n,MATIS,&ismatis);CHKERRQ(ierr); if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)*mat_n),PETSC_ERR_SUP,"Cannot reuse a matrix which is not of type MATIS"); ierr = MatGetSize(mat,&mrows,&mcols);CHKERRQ(ierr); ierr = MatGetSize(*mat_n,&mnrows,&mncols);CHKERRQ(ierr); if (mrows != mnrows) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse matrix! Wrong number of rows %D != %D",mrows,mnrows); if (mcols != mncols) SETERRQ2(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"Cannot reuse matrix! Wrong number of cols %D != %D",mcols,mncols); } ierr = MatGetBlockSize(local_mat,&bs);CHKERRQ(ierr); PetscValidLogicalCollectiveInt(mat,bs,0); /* prepare IS for sending if not provided */ if (!is_sends) { if (!n_subdomains) SETERRQ(PetscObjectComm((PetscObject)mat),PETSC_ERR_SUP,"You should specify either an IS or a target number of subdomains"); ierr = PCBDDCMatISGetSubassemblingPattern(mat,&n_subdomains,0,&is_sends_internal,NULL);CHKERRQ(ierr); } else { ierr = PetscObjectReference((PetscObject)is_sends);CHKERRQ(ierr); is_sends_internal = is_sends; } /* get comm */ ierr = PetscObjectGetComm((PetscObject)mat,&comm);CHKERRQ(ierr); /* compute number of sends */ ierr = ISGetLocalSize(is_sends_internal,&i);CHKERRQ(ierr); ierr = PetscMPIIntCast(i,&n_sends);CHKERRQ(ierr); /* compute number of receives */ ierr = MPI_Comm_size(comm,&commsize);CHKERRQ(ierr); ierr = PetscMalloc1(commsize,&iflags);CHKERRQ(ierr); ierr = PetscMemzero(iflags,commsize*sizeof(*iflags));CHKERRQ(ierr); ierr = ISGetIndices(is_sends_internal,&is_indices);CHKERRQ(ierr); for (i=0;irmap->mapping,&i);CHKERRQ(ierr); ierr = PetscMalloc1(i+2,&send_buffer_idxs);CHKERRQ(ierr); send_buffer_idxs[0] = (PetscInt)MATDENSE_PRIVATE; send_buffer_idxs[1] = i; ierr = ISLocalToGlobalMappingGetIndices(mat->rmap->mapping,(const PetscInt**)&ptr_idxs);CHKERRQ(ierr); ierr = PetscMemcpy(&send_buffer_idxs[2],ptr_idxs,i*sizeof(PetscInt));CHKERRQ(ierr); ierr = ISLocalToGlobalMappingRestoreIndices(mat->rmap->mapping,(const PetscInt**)&ptr_idxs);CHKERRQ(ierr); ierr = PetscMPIIntCast(i,&len);CHKERRQ(ierr); for (i=0;i type of incoming data and its size */ buf_size_idxs = 0; for (i=0;i PetscErrorCode PCBDDCSetUpCoarseSolver(PC pc,PetscScalar* coarse_submat_vals) { PC_BDDC *pcbddc = (PC_BDDC*)pc->data; PC_IS *pcis = (PC_IS*)pc->data; Mat coarse_mat,coarse_mat_is,coarse_submat_dense; Mat coarsedivudotp = NULL; Mat coarseG,t_coarse_mat_is; MatNullSpace CoarseNullSpace = NULL; ISLocalToGlobalMapping coarse_islg; IS coarse_is,*isarray; PetscInt i,im_active=-1,active_procs=-1; PetscInt nis,nisdofs,nisneu,nisvert; PC pc_temp; PCType coarse_pc_type; KSPType coarse_ksp_type; PetscBool multilevel_requested,multilevel_allowed; PetscBool isredundant,isbddc,isnn,coarse_reuse; PetscInt ncoarse,nedcfield; PetscBool compute_vecs = PETSC_FALSE; PetscScalar *array; MatReuse coarse_mat_reuse; PetscBool restr, full_restr, have_void; PetscMPIInt commsize; PetscErrorCode ierr; PetscFunctionBegin; /* Assign global numbering to coarse dofs */ if (pcbddc->new_primal_space || pcbddc->coarse_size == -1) { /* a new primal space is present or it is the first initialization, so recompute global numbering */ PetscInt ocoarse_size; compute_vecs = PETSC_TRUE; pcbddc->new_primal_space = PETSC_TRUE; ocoarse_size = pcbddc->coarse_size; ierr = PetscFree(pcbddc->global_primal_indices);CHKERRQ(ierr); ierr = PCBDDCComputePrimalNumbering(pc,&pcbddc->coarse_size,&pcbddc->global_primal_indices);CHKERRQ(ierr); /* see if we can avoid some work */ if (pcbddc->coarse_ksp) { /* coarse ksp has already been created */ /* if the coarse size is different or we are using adaptive selection, better to not reuse the coarse matrix */ if (ocoarse_size != pcbddc->coarse_size || pcbddc->adaptive_selection) { ierr = KSPReset(pcbddc->coarse_ksp);CHKERRQ(ierr); coarse_reuse = PETSC_FALSE; } else { /* we can safely reuse already computed coarse matrix */ coarse_reuse = PETSC_TRUE; } } else { /* there's no coarse ksp, so we need to create the coarse matrix too */ coarse_reuse = PETSC_FALSE; } /* reset any subassembling information */ if (!coarse_reuse || pcbddc->recompute_topography) { ierr = ISDestroy(&pcbddc->coarse_subassembling);CHKERRQ(ierr); } } else { /* primal space is unchanged, so we can reuse coarse matrix */ coarse_reuse = PETSC_TRUE; } /* assemble coarse matrix */ if (coarse_reuse && pcbddc->coarse_ksp) { ierr = KSPGetOperators(pcbddc->coarse_ksp,&coarse_mat,NULL);CHKERRQ(ierr); ierr = PetscObjectReference((PetscObject)coarse_mat);CHKERRQ(ierr); coarse_mat_reuse = MAT_REUSE_MATRIX; } else { coarse_mat = NULL; coarse_mat_reuse = MAT_INITIAL_MATRIX; } /* creates temporary l2gmap and IS for coarse indexes */ ierr = ISCreateGeneral(PetscObjectComm((PetscObject)pc),pcbddc->local_primal_size,pcbddc->global_primal_indices,PETSC_COPY_VALUES,&coarse_is);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingCreateIS(coarse_is,&coarse_islg);CHKERRQ(ierr); /* creates temporary MATIS object for coarse matrix */ ierr = MatCreateSeqDense(PETSC_COMM_SELF,pcbddc->local_primal_size,pcbddc->local_primal_size,NULL,&coarse_submat_dense);CHKERRQ(ierr); ierr = MatDenseGetArray(coarse_submat_dense,&array);CHKERRQ(ierr); ierr = PetscMemcpy(array,coarse_submat_vals,sizeof(*coarse_submat_vals)*pcbddc->local_primal_size*pcbddc->local_primal_size);CHKERRQ(ierr); ierr = MatDenseRestoreArray(coarse_submat_dense,&array);CHKERRQ(ierr); ierr = MatCreateIS(PetscObjectComm((PetscObject)pc),1,PETSC_DECIDE,PETSC_DECIDE,pcbddc->coarse_size,pcbddc->coarse_size,coarse_islg,NULL,&t_coarse_mat_is);CHKERRQ(ierr); ierr = MatISSetLocalMat(t_coarse_mat_is,coarse_submat_dense);CHKERRQ(ierr); ierr = MatAssemblyBegin(t_coarse_mat_is,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); ierr = MatAssemblyEnd(t_coarse_mat_is,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); ierr = MatDestroy(&coarse_submat_dense);CHKERRQ(ierr); /* count "active" (i.e. with positive local size) and "void" processes */ im_active = !!(pcis->n); ierr = MPIU_Allreduce(&im_active,&active_procs,1,MPIU_INT,MPI_SUM,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); /* determine number of processes partecipating to coarse solver and compute subassembling pattern */ /* restr : whether if we want to exclude senders (which are not receivers) from the subassembling pattern */ /* full_restr : just use the receivers from the subassembling pattern */ ierr = MPI_Comm_size(PetscObjectComm((PetscObject)pc),&commsize);CHKERRQ(ierr); coarse_mat_is = NULL; multilevel_allowed = PETSC_FALSE; multilevel_requested = PETSC_FALSE; pcbddc->coarse_eqs_per_proc = PetscMin(PetscMax(pcbddc->coarse_size,1),pcbddc->coarse_eqs_per_proc); if (pcbddc->current_level < pcbddc->max_levels) multilevel_requested = PETSC_TRUE; if (multilevel_requested) { ncoarse = active_procs/pcbddc->coarsening_ratio; restr = PETSC_FALSE; full_restr = PETSC_FALSE; } else { ncoarse = pcbddc->coarse_size/pcbddc->coarse_eqs_per_proc; restr = PETSC_TRUE; full_restr = PETSC_TRUE; } if (!pcbddc->coarse_size || commsize == 1) multilevel_allowed = multilevel_requested = restr = full_restr = PETSC_FALSE; ncoarse = PetscMax(1,ncoarse); if (!pcbddc->coarse_subassembling) { if (pcbddc->coarsening_ratio > 1) { if (multilevel_requested) { ierr = PCBDDCMatISGetSubassemblingPattern(pc->pmat,&ncoarse,pcbddc->coarse_adj_red,&pcbddc->coarse_subassembling,&have_void);CHKERRQ(ierr); } else { ierr = PCBDDCMatISGetSubassemblingPattern(t_coarse_mat_is,&ncoarse,pcbddc->coarse_adj_red,&pcbddc->coarse_subassembling,&have_void);CHKERRQ(ierr); } } else { PetscMPIInt rank; ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)pc),&rank);CHKERRQ(ierr); have_void = (active_procs == (PetscInt)commsize) ? PETSC_FALSE : PETSC_TRUE; ierr = ISCreateStride(PetscObjectComm((PetscObject)pc),1,rank,1,&pcbddc->coarse_subassembling);CHKERRQ(ierr); } } else { /* if a subassembling pattern exists, then we can reuse the coarse ksp and compute the number of process involved */ PetscInt psum; if (pcbddc->coarse_ksp) psum = 1; else psum = 0; ierr = MPIU_Allreduce(&psum,&ncoarse,1,MPIU_INT,MPI_SUM,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); if (ncoarse < commsize) have_void = PETSC_TRUE; } /* determine if we can go multilevel */ if (multilevel_requested) { if (ncoarse > 1) multilevel_allowed = PETSC_TRUE; /* found enough processes */ else restr = full_restr = PETSC_TRUE; /* 1 subdomain, use a direct solver */ } if (multilevel_allowed && have_void) restr = PETSC_TRUE; /* dump subassembling pattern */ if (pcbddc->dbg_flag && multilevel_allowed) { ierr = ISView(pcbddc->coarse_subassembling,pcbddc->dbg_viewer);CHKERRQ(ierr); } /* compute dofs splitting and neumann boundaries for coarse dofs */ nedcfield = -1; if (multilevel_allowed && !coarse_reuse && (pcbddc->n_ISForDofsLocal || pcbddc->NeumannBoundariesLocal || pcbddc->nedclocal)) { /* protects from unneded computations */ PetscInt *tidxs,*tidxs2,nout,tsize,i; const PetscInt *idxs; ISLocalToGlobalMapping tmap; /* create map between primal indices (in local representative ordering) and local primal numbering */ ierr = ISLocalToGlobalMappingCreate(PETSC_COMM_SELF,1,pcbddc->local_primal_size,pcbddc->primal_indices_local_idxs,PETSC_COPY_VALUES,&tmap);CHKERRQ(ierr); /* allocate space for temporary storage */ ierr = PetscMalloc1(pcbddc->local_primal_size,&tidxs);CHKERRQ(ierr); ierr = PetscMalloc1(pcbddc->local_primal_size,&tidxs2);CHKERRQ(ierr); /* allocate for IS array */ nisdofs = pcbddc->n_ISForDofsLocal; if (pcbddc->nedclocal) { if (pcbddc->nedfield > -1) { nedcfield = pcbddc->nedfield; } else { nedcfield = 0; if (nisdofs) SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_PLIB,"This should not happen (%d)",nisdofs); nisdofs = 1; } } nisneu = !!pcbddc->NeumannBoundariesLocal; nisvert = 0; /* nisvert is not used */ nis = nisdofs + nisneu + nisvert; ierr = PetscMalloc1(nis,&isarray);CHKERRQ(ierr); /* dofs splitting */ for (i=0;iISForDofsLocal[i],0);CHKERRQ(ierr); */ if (nedcfield != i) { ierr = ISGetLocalSize(pcbddc->ISForDofsLocal[i],&tsize);CHKERRQ(ierr); ierr = ISGetIndices(pcbddc->ISForDofsLocal[i],&idxs);CHKERRQ(ierr); ierr = ISGlobalToLocalMappingApply(tmap,IS_GTOLM_DROP,tsize,idxs,&nout,tidxs);CHKERRQ(ierr); ierr = ISRestoreIndices(pcbddc->ISForDofsLocal[i],&idxs);CHKERRQ(ierr); } else { ierr = ISGetLocalSize(pcbddc->nedclocal,&tsize);CHKERRQ(ierr); ierr = ISGetIndices(pcbddc->nedclocal,&idxs);CHKERRQ(ierr); ierr = ISGlobalToLocalMappingApply(tmap,IS_GTOLM_DROP,tsize,idxs,&nout,tidxs);CHKERRQ(ierr); if (tsize != nout) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Failed when mapping coarse nedelec field! %d != %d\n",tsize,nout); ierr = ISRestoreIndices(pcbddc->nedclocal,&idxs);CHKERRQ(ierr); } ierr = ISLocalToGlobalMappingApply(coarse_islg,nout,tidxs,tidxs2);CHKERRQ(ierr); ierr = ISCreateGeneral(PetscObjectComm((PetscObject)pc),nout,tidxs2,PETSC_COPY_VALUES,&isarray[i]);CHKERRQ(ierr); /* ierr = ISView(isarray[i],0);CHKERRQ(ierr); */ } /* neumann boundaries */ if (pcbddc->NeumannBoundariesLocal) { /* ierr = ISView(pcbddc->NeumannBoundariesLocal,0);CHKERRQ(ierr); */ ierr = ISGetLocalSize(pcbddc->NeumannBoundariesLocal,&tsize);CHKERRQ(ierr); ierr = ISGetIndices(pcbddc->NeumannBoundariesLocal,&idxs);CHKERRQ(ierr); ierr = ISGlobalToLocalMappingApply(tmap,IS_GTOLM_DROP,tsize,idxs,&nout,tidxs);CHKERRQ(ierr); ierr = ISRestoreIndices(pcbddc->NeumannBoundariesLocal,&idxs);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingApply(coarse_islg,nout,tidxs,tidxs2);CHKERRQ(ierr); ierr = ISCreateGeneral(PetscObjectComm((PetscObject)pc),nout,tidxs2,PETSC_COPY_VALUES,&isarray[nisdofs]);CHKERRQ(ierr); /* ierr = ISView(isarray[nisdofs],0);CHKERRQ(ierr); */ } /* free memory */ ierr = PetscFree(tidxs);CHKERRQ(ierr); ierr = PetscFree(tidxs2);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingDestroy(&tmap);CHKERRQ(ierr); } else { nis = 0; nisdofs = 0; nisneu = 0; nisvert = 0; isarray = NULL; } /* destroy no longer needed map */ ierr = ISLocalToGlobalMappingDestroy(&coarse_islg);CHKERRQ(ierr); /* subassemble */ if (multilevel_allowed) { Vec vp[1]; PetscInt nvecs = 0; PetscBool reuse,reuser; if (coarse_mat) reuse = PETSC_TRUE; else reuse = PETSC_FALSE; ierr = MPIU_Allreduce(&reuse,&reuser,1,MPIU_BOOL,MPI_LOR,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); vp[0] = NULL; if (pcbddc->benign_have_null) { /* propagate no-net-flux quadrature to coarser level */ ierr = VecCreate(PetscObjectComm((PetscObject)pc),&vp[0]);CHKERRQ(ierr); ierr = VecSetSizes(vp[0],pcbddc->local_primal_size,PETSC_DECIDE);CHKERRQ(ierr); ierr = VecSetType(vp[0],VECSTANDARD);CHKERRQ(ierr); nvecs = 1; if (pcbddc->divudotp) { Mat B,loc_divudotp; Vec v,p; IS dummy; PetscInt np; ierr = MatISGetLocalMat(pcbddc->divudotp,&loc_divudotp);CHKERRQ(ierr); ierr = MatGetSize(loc_divudotp,&np,NULL);CHKERRQ(ierr); ierr = ISCreateStride(PETSC_COMM_SELF,np,0,1,&dummy);CHKERRQ(ierr); ierr = MatCreateSubMatrix(loc_divudotp,dummy,pcis->is_B_local,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr); ierr = MatCreateVecs(B,&v,&p);CHKERRQ(ierr); ierr = VecSet(p,1.);CHKERRQ(ierr); ierr = MatMultTranspose(B,p,v);CHKERRQ(ierr); ierr = VecDestroy(&p);CHKERRQ(ierr); ierr = MatDestroy(&B);CHKERRQ(ierr); ierr = VecGetArray(vp[0],&array);CHKERRQ(ierr); ierr = VecPlaceArray(pcbddc->vec1_P,array);CHKERRQ(ierr); ierr = VecRestoreArray(vp[0],&array);CHKERRQ(ierr); ierr = MatMultTranspose(pcbddc->coarse_phi_B,v,pcbddc->vec1_P);CHKERRQ(ierr); ierr = VecResetArray(pcbddc->vec1_P);CHKERRQ(ierr); ierr = ISDestroy(&dummy);CHKERRQ(ierr); ierr = VecDestroy(&v);CHKERRQ(ierr); } } if (reuser) { ierr = PCBDDCMatISSubassemble(t_coarse_mat_is,pcbddc->coarse_subassembling,0,restr,full_restr,PETSC_TRUE,&coarse_mat,nis,isarray,nvecs,vp);CHKERRQ(ierr); } else { ierr = PCBDDCMatISSubassemble(t_coarse_mat_is,pcbddc->coarse_subassembling,0,restr,full_restr,PETSC_FALSE,&coarse_mat_is,nis,isarray,nvecs,vp);CHKERRQ(ierr); } if (vp[0]) { /* vp[0] could have been placed on a different set of processes */ PetscScalar *arraym,*arrayv; PetscInt nl; ierr = VecGetLocalSize(vp[0],&nl);CHKERRQ(ierr); ierr = MatCreateSeqDense(PETSC_COMM_SELF,1,nl,NULL,&coarsedivudotp);CHKERRQ(ierr); ierr = MatDenseGetArray(coarsedivudotp,&arraym);CHKERRQ(ierr); ierr = VecGetArray(vp[0],&arrayv);CHKERRQ(ierr); ierr = PetscMemcpy(arraym,arrayv,nl*sizeof(PetscScalar));CHKERRQ(ierr); ierr = VecRestoreArray(vp[0],&arrayv);CHKERRQ(ierr); ierr = MatDenseRestoreArray(coarsedivudotp,&arraym);CHKERRQ(ierr); ierr = VecDestroy(&vp[0]);CHKERRQ(ierr); } else { ierr = MatCreateSeqAIJ(PETSC_COMM_SELF,0,0,1,NULL,&coarsedivudotp);CHKERRQ(ierr); } } else { ierr = PCBDDCMatISSubassemble(t_coarse_mat_is,pcbddc->coarse_subassembling,0,restr,full_restr,PETSC_FALSE,&coarse_mat_is,0,NULL,0,NULL);CHKERRQ(ierr); } if (coarse_mat_is || coarse_mat) { PetscMPIInt size; ierr = MPI_Comm_size(PetscObjectComm((PetscObject)coarse_mat_is),&size);CHKERRQ(ierr); if (!multilevel_allowed) { ierr = MatISGetMPIXAIJ(coarse_mat_is,coarse_mat_reuse,&coarse_mat);CHKERRQ(ierr); } else { Mat A; /* if this matrix is present, it means we are not reusing the coarse matrix */ if (coarse_mat_is) { if (coarse_mat) SETERRQ(PetscObjectComm((PetscObject)coarse_mat_is),PETSC_ERR_PLIB,"This should not happen"); ierr = PetscObjectReference((PetscObject)coarse_mat_is);CHKERRQ(ierr); coarse_mat = coarse_mat_is; } /* be sure we don't have MatSeqDENSE as local mat */ ierr = MatISGetLocalMat(coarse_mat,&A);CHKERRQ(ierr); ierr = MatConvert(A,MATSEQAIJ,MAT_INPLACE_MATRIX,&A);CHKERRQ(ierr); } } ierr = MatDestroy(&t_coarse_mat_is);CHKERRQ(ierr); ierr = MatDestroy(&coarse_mat_is);CHKERRQ(ierr); /* create local to global scatters for coarse problem */ if (compute_vecs) { PetscInt lrows; ierr = VecDestroy(&pcbddc->coarse_vec);CHKERRQ(ierr); if (coarse_mat) { ierr = MatGetLocalSize(coarse_mat,&lrows,NULL);CHKERRQ(ierr); } else { lrows = 0; } ierr = VecCreate(PetscObjectComm((PetscObject)pc),&pcbddc->coarse_vec);CHKERRQ(ierr); ierr = VecSetSizes(pcbddc->coarse_vec,lrows,PETSC_DECIDE);CHKERRQ(ierr); ierr = VecSetType(pcbddc->coarse_vec,VECSTANDARD);CHKERRQ(ierr); ierr = VecScatterDestroy(&pcbddc->coarse_loc_to_glob);CHKERRQ(ierr); ierr = VecScatterCreate(pcbddc->vec1_P,NULL,pcbddc->coarse_vec,coarse_is,&pcbddc->coarse_loc_to_glob);CHKERRQ(ierr); } ierr = ISDestroy(&coarse_is);CHKERRQ(ierr); /* set defaults for coarse KSP and PC */ if (multilevel_allowed) { coarse_ksp_type = KSPRICHARDSON; coarse_pc_type = PCBDDC; } else { coarse_ksp_type = KSPPREONLY; coarse_pc_type = PCREDUNDANT; } /* print some info if requested */ if (pcbddc->dbg_flag) { if (!multilevel_allowed) { ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"--------------------------------------------------\n");CHKERRQ(ierr); if (multilevel_requested) { ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"Not enough active processes on level %d (active processes %d, coarsening ratio %d)\n",pcbddc->current_level,active_procs,pcbddc->coarsening_ratio);CHKERRQ(ierr); } else if (pcbddc->max_levels) { ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"Maximum number of requested levels reached (%d)\n",pcbddc->max_levels);CHKERRQ(ierr); } ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); } } /* communicate coarse discrete gradient */ coarseG = NULL; if (pcbddc->nedcG && multilevel_allowed) { MPI_Comm ccomm; if (coarse_mat) { ccomm = PetscObjectComm((PetscObject)coarse_mat); } else { ccomm = MPI_COMM_NULL; } ierr = MatMPIAIJRestrict(pcbddc->nedcG,ccomm,&coarseG);CHKERRQ(ierr); } /* create the coarse KSP object only once with defaults */ if (coarse_mat) { PetscViewer dbg_viewer = NULL; if (pcbddc->dbg_flag) { dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)coarse_mat)); ierr = PetscViewerASCIIAddTab(dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); } if (!pcbddc->coarse_ksp) { char prefix[256],str_level[16]; size_t len; ierr = KSPCreate(PetscObjectComm((PetscObject)coarse_mat),&pcbddc->coarse_ksp);CHKERRQ(ierr); ierr = KSPSetErrorIfNotConverged(pcbddc->coarse_ksp,pc->erroriffailure);CHKERRQ(ierr); ierr = PetscObjectIncrementTabLevel((PetscObject)pcbddc->coarse_ksp,(PetscObject)pc,1);CHKERRQ(ierr); ierr = KSPSetTolerances(pcbddc->coarse_ksp,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT,1);CHKERRQ(ierr); ierr = KSPSetOperators(pcbddc->coarse_ksp,coarse_mat,coarse_mat);CHKERRQ(ierr); ierr = KSPSetType(pcbddc->coarse_ksp,coarse_ksp_type);CHKERRQ(ierr); ierr = KSPSetNormType(pcbddc->coarse_ksp,KSP_NORM_NONE);CHKERRQ(ierr); ierr = KSPGetPC(pcbddc->coarse_ksp,&pc_temp);CHKERRQ(ierr); /* TODO is this logic correct? should check for coarse_mat type */ ierr = PCSetType(pc_temp,coarse_pc_type);CHKERRQ(ierr); /* prefix */ ierr = PetscStrcpy(prefix,"");CHKERRQ(ierr); ierr = PetscStrcpy(str_level,"");CHKERRQ(ierr); if (!pcbddc->current_level) { ierr = PetscStrcpy(prefix,((PetscObject)pc)->prefix);CHKERRQ(ierr); ierr = PetscStrcat(prefix,"pc_bddc_coarse_");CHKERRQ(ierr); } else { ierr = PetscStrlen(((PetscObject)pc)->prefix,&len);CHKERRQ(ierr); if (pcbddc->current_level>1) len -= 3; /* remove "lX_" with X level number */ if (pcbddc->current_level>10) len -= 1; /* remove another char from level number */ ierr = PetscStrncpy(prefix,((PetscObject)pc)->prefix,len+1);CHKERRQ(ierr); sprintf(str_level,"l%d_",(int)(pcbddc->current_level)); ierr = PetscStrcat(prefix,str_level);CHKERRQ(ierr); } ierr = KSPSetOptionsPrefix(pcbddc->coarse_ksp,prefix);CHKERRQ(ierr); /* propagate BDDC info to the next level (these are dummy calls if pc_temp is not of type PCBDDC) */ ierr = PCBDDCSetLevel(pc_temp,pcbddc->current_level+1);CHKERRQ(ierr); ierr = PCBDDCSetCoarseningRatio(pc_temp,pcbddc->coarsening_ratio);CHKERRQ(ierr); ierr = PCBDDCSetLevels(pc_temp,pcbddc->max_levels);CHKERRQ(ierr); /* allow user customization */ ierr = KSPSetFromOptions(pcbddc->coarse_ksp);CHKERRQ(ierr); } /* propagate BDDC info to the next level (these are dummy calls if pc_temp is not of type PCBDDC) */ ierr = KSPGetPC(pcbddc->coarse_ksp,&pc_temp);CHKERRQ(ierr); if (nisdofs) { ierr = PCBDDCSetDofsSplitting(pc_temp,nisdofs,isarray);CHKERRQ(ierr); for (i=0;idata; pcbddc_coarse->detect_disconnected = PETSC_TRUE; pcbddc_coarse->coarse_eqs_per_proc = pcbddc->coarse_eqs_per_proc; pcbddc_coarse->benign_saddle_point = pcbddc->benign_have_null; if (pcbddc_coarse->benign_saddle_point) { Mat coarsedivudotp_is; ISLocalToGlobalMapping l2gmap,rl2g,cl2g; IS row,col; const PetscInt *gidxs; PetscInt n,st,M,N; ierr = MatGetSize(coarsedivudotp,&n,NULL);CHKERRQ(ierr); ierr = MPI_Scan(&n,&st,1,MPIU_INT,MPI_SUM,PetscObjectComm((PetscObject)coarse_mat));CHKERRQ(ierr); st = st-n; ierr = ISCreateStride(PetscObjectComm((PetscObject)coarse_mat),1,st,1,&row);CHKERRQ(ierr); ierr = MatGetLocalToGlobalMapping(coarse_mat,&l2gmap,NULL);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingGetSize(l2gmap,&n);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingGetIndices(l2gmap,&gidxs);CHKERRQ(ierr); ierr = ISCreateGeneral(PetscObjectComm((PetscObject)coarse_mat),n,gidxs,PETSC_COPY_VALUES,&col);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingRestoreIndices(l2gmap,&gidxs);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingCreateIS(row,&rl2g);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingCreateIS(col,&cl2g);CHKERRQ(ierr); ierr = ISGetSize(row,&M);CHKERRQ(ierr); ierr = MatGetSize(coarse_mat,&N,NULL);CHKERRQ(ierr); ierr = ISDestroy(&row);CHKERRQ(ierr); ierr = ISDestroy(&col);CHKERRQ(ierr); ierr = MatCreate(PetscObjectComm((PetscObject)coarse_mat),&coarsedivudotp_is);CHKERRQ(ierr); ierr = MatSetType(coarsedivudotp_is,MATIS);CHKERRQ(ierr); ierr = MatSetSizes(coarsedivudotp_is,PETSC_DECIDE,PETSC_DECIDE,M,N);CHKERRQ(ierr); ierr = MatSetLocalToGlobalMapping(coarsedivudotp_is,rl2g,cl2g);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr); ierr = MatISSetLocalMat(coarsedivudotp_is,coarsedivudotp);CHKERRQ(ierr); ierr = MatDestroy(&coarsedivudotp);CHKERRQ(ierr); ierr = PCBDDCSetDivergenceMat(pc_temp,coarsedivudotp_is,PETSC_FALSE,NULL);CHKERRQ(ierr); ierr = MatDestroy(&coarsedivudotp_is);CHKERRQ(ierr); pcbddc_coarse->adaptive_userdefined = PETSC_TRUE; if (pcbddc->adaptive_threshold < 1.0) pcbddc_coarse->deluxe_zerorows = PETSC_TRUE; } } /* propagate symmetry info of coarse matrix */ ierr = MatSetOption(coarse_mat,MAT_STRUCTURALLY_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr); if (pc->pmat->symmetric_set) { ierr = MatSetOption(coarse_mat,MAT_SYMMETRIC,pc->pmat->symmetric);CHKERRQ(ierr); } if (pc->pmat->hermitian_set) { ierr = MatSetOption(coarse_mat,MAT_HERMITIAN,pc->pmat->hermitian);CHKERRQ(ierr); } if (pc->pmat->spd_set) { ierr = MatSetOption(coarse_mat,MAT_SPD,pc->pmat->spd);CHKERRQ(ierr); } if (pcbddc->benign_saddle_point && !pcbddc->benign_have_null) { ierr = MatSetOption(coarse_mat,MAT_SPD,PETSC_TRUE);CHKERRQ(ierr); } /* set operators */ ierr = KSPSetOperators(pcbddc->coarse_ksp,coarse_mat,coarse_mat);CHKERRQ(ierr); if (pcbddc->dbg_flag) { ierr = PetscViewerASCIISubtractTab(dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); } } ierr = MatDestroy(&coarseG);CHKERRQ(ierr); ierr = PetscFree(isarray);CHKERRQ(ierr); #if 0 { PetscViewer viewer; char filename[256]; sprintf(filename,"coarse_mat_level%d.m",pcbddc->current_level); ierr = PetscViewerASCIIOpen(PetscObjectComm((PetscObject)coarse_mat),filename,&viewer);CHKERRQ(ierr); ierr = PetscViewerPushFormat(viewer,PETSC_VIEWER_ASCII_MATLAB);CHKERRQ(ierr); ierr = MatView(coarse_mat,viewer);CHKERRQ(ierr); ierr = PetscViewerPopFormat(viewer);CHKERRQ(ierr); ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); } #endif if (pcbddc->coarse_ksp) { Vec crhs,csol; ierr = KSPGetSolution(pcbddc->coarse_ksp,&csol);CHKERRQ(ierr); ierr = KSPGetRhs(pcbddc->coarse_ksp,&crhs);CHKERRQ(ierr); if (!csol) { ierr = MatCreateVecs(coarse_mat,&((pcbddc->coarse_ksp)->vec_sol),NULL);CHKERRQ(ierr); } if (!crhs) { ierr = MatCreateVecs(coarse_mat,NULL,&((pcbddc->coarse_ksp)->vec_rhs));CHKERRQ(ierr); } } ierr = MatDestroy(&coarsedivudotp);CHKERRQ(ierr); /* compute null space for coarse solver if the benign trick has been requested */ if (pcbddc->benign_null) { ierr = VecSet(pcbddc->vec1_P,0.);CHKERRQ(ierr); for (i=0;ibenign_n;i++) { ierr = VecSetValue(pcbddc->vec1_P,pcbddc->local_primal_size-pcbddc->benign_n+i,1.0,INSERT_VALUES);CHKERRQ(ierr); } ierr = VecAssemblyBegin(pcbddc->vec1_P);CHKERRQ(ierr); ierr = VecAssemblyEnd(pcbddc->vec1_P);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->vec1_P,pcbddc->coarse_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->vec1_P,pcbddc->coarse_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); if (coarse_mat) { Vec nullv; PetscScalar *array,*array2; PetscInt nl; ierr = MatCreateVecs(coarse_mat,&nullv,NULL);CHKERRQ(ierr); ierr = VecGetLocalSize(nullv,&nl);CHKERRQ(ierr); ierr = VecGetArrayRead(pcbddc->coarse_vec,(const PetscScalar**)&array);CHKERRQ(ierr); ierr = VecGetArray(nullv,&array2);CHKERRQ(ierr); ierr = PetscMemcpy(array2,array,nl*sizeof(*array));CHKERRQ(ierr); ierr = VecRestoreArray(nullv,&array2);CHKERRQ(ierr); ierr = VecRestoreArrayRead(pcbddc->coarse_vec,(const PetscScalar**)&array);CHKERRQ(ierr); ierr = VecNormalize(nullv,NULL);CHKERRQ(ierr); ierr = MatNullSpaceCreate(PetscObjectComm((PetscObject)coarse_mat),PETSC_FALSE,1,&nullv,&CoarseNullSpace);CHKERRQ(ierr); ierr = VecDestroy(&nullv);CHKERRQ(ierr); } } if (pcbddc->coarse_ksp) { PetscBool ispreonly; if (CoarseNullSpace) { PetscBool isnull; ierr = MatNullSpaceTest(CoarseNullSpace,coarse_mat,&isnull);CHKERRQ(ierr); if (isnull) { ierr = MatSetNullSpace(coarse_mat,CoarseNullSpace);CHKERRQ(ierr); } /* TODO: add local nullspaces (if any) */ } /* setup coarse ksp */ ierr = KSPSetUp(pcbddc->coarse_ksp);CHKERRQ(ierr); /* Check coarse problem if in debug mode or if solving with an iterative method */ ierr = PetscObjectTypeCompare((PetscObject)pcbddc->coarse_ksp,KSPPREONLY,&ispreonly);CHKERRQ(ierr); if (pcbddc->dbg_flag || (!ispreonly && pcbddc->use_coarse_estimates) ) { KSP check_ksp; KSPType check_ksp_type; PC check_pc; Vec check_vec,coarse_vec; PetscReal abs_infty_error,infty_error,lambda_min=1.0,lambda_max=1.0; PetscInt its; PetscBool compute_eigs; PetscReal *eigs_r,*eigs_c; PetscInt neigs; const char *prefix; /* Create ksp object suitable for estimation of extreme eigenvalues */ ierr = KSPCreate(PetscObjectComm((PetscObject)pcbddc->coarse_ksp),&check_ksp);CHKERRQ(ierr); ierr = KSPSetErrorIfNotConverged(pcbddc->coarse_ksp,pc->erroriffailure);CHKERRQ(ierr); ierr = KSPSetOperators(check_ksp,coarse_mat,coarse_mat);CHKERRQ(ierr); ierr = KSPSetTolerances(check_ksp,1.e-12,1.e-12,PETSC_DEFAULT,pcbddc->coarse_size);CHKERRQ(ierr); /* prevent from setup unneeded object */ ierr = KSPGetPC(check_ksp,&check_pc);CHKERRQ(ierr); ierr = PCSetType(check_pc,PCNONE);CHKERRQ(ierr); if (ispreonly) { check_ksp_type = KSPPREONLY; compute_eigs = PETSC_FALSE; } else { check_ksp_type = KSPGMRES; compute_eigs = PETSC_TRUE; } ierr = KSPSetType(check_ksp,check_ksp_type);CHKERRQ(ierr); ierr = KSPSetComputeSingularValues(check_ksp,compute_eigs);CHKERRQ(ierr); ierr = KSPSetComputeEigenvalues(check_ksp,compute_eigs);CHKERRQ(ierr); ierr = KSPGMRESSetRestart(check_ksp,pcbddc->coarse_size+1);CHKERRQ(ierr); ierr = KSPGetOptionsPrefix(pcbddc->coarse_ksp,&prefix);CHKERRQ(ierr); ierr = KSPSetOptionsPrefix(check_ksp,prefix);CHKERRQ(ierr); ierr = KSPAppendOptionsPrefix(check_ksp,"check_");CHKERRQ(ierr); ierr = KSPSetFromOptions(check_ksp);CHKERRQ(ierr); ierr = KSPSetUp(check_ksp);CHKERRQ(ierr); ierr = KSPGetPC(pcbddc->coarse_ksp,&check_pc);CHKERRQ(ierr); ierr = KSPSetPC(check_ksp,check_pc);CHKERRQ(ierr); /* create random vec */ ierr = MatCreateVecs(coarse_mat,&coarse_vec,&check_vec);CHKERRQ(ierr); ierr = VecSetRandom(check_vec,NULL);CHKERRQ(ierr); ierr = MatMult(coarse_mat,check_vec,coarse_vec);CHKERRQ(ierr); /* solve coarse problem */ ierr = KSPSolve(check_ksp,coarse_vec,coarse_vec);CHKERRQ(ierr); /* set eigenvalue estimation if preonly has not been requested */ if (compute_eigs) { ierr = PetscMalloc1(pcbddc->coarse_size+1,&eigs_r);CHKERRQ(ierr); ierr = PetscMalloc1(pcbddc->coarse_size+1,&eigs_c);CHKERRQ(ierr); ierr = KSPComputeEigenvalues(check_ksp,pcbddc->coarse_size+1,eigs_r,eigs_c,&neigs);CHKERRQ(ierr); if (neigs) { lambda_max = eigs_r[neigs-1]; lambda_min = eigs_r[0]; if (pcbddc->use_coarse_estimates) { if (lambda_max>=lambda_min) { /* using PETSC_SMALL since lambda_max == lambda_min is not allowed by KSPChebyshevSetEigenvalues */ ierr = KSPChebyshevSetEigenvalues(pcbddc->coarse_ksp,lambda_max+PETSC_SMALL,lambda_min);CHKERRQ(ierr); ierr = KSPRichardsonSetScale(pcbddc->coarse_ksp,2.0/(lambda_max+lambda_min));CHKERRQ(ierr); } } } } /* check coarse problem residual error */ if (pcbddc->dbg_flag) { PetscViewer dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pcbddc->coarse_ksp)); ierr = PetscViewerASCIIAddTab(dbg_viewer,2*(pcbddc->current_level+1));CHKERRQ(ierr); ierr = VecAXPY(check_vec,-1.0,coarse_vec);CHKERRQ(ierr); ierr = VecNorm(check_vec,NORM_INFINITY,&infty_error);CHKERRQ(ierr); ierr = MatMult(coarse_mat,check_vec,coarse_vec);CHKERRQ(ierr); ierr = VecNorm(coarse_vec,NORM_INFINITY,&abs_infty_error);CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(dbg_viewer,"Coarse problem details (use estimates %d)\n",pcbddc->use_coarse_estimates);CHKERRQ(ierr); ierr = PetscObjectPrintClassNamePrefixType((PetscObject)(pcbddc->coarse_ksp),dbg_viewer);CHKERRQ(ierr); ierr = PetscObjectPrintClassNamePrefixType((PetscObject)(check_pc),dbg_viewer);CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(dbg_viewer,"Coarse problem exact infty_error : %1.6e\n",infty_error);CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(dbg_viewer,"Coarse problem residual infty_error: %1.6e\n",abs_infty_error);CHKERRQ(ierr); if (CoarseNullSpace) { ierr = PetscViewerASCIIPrintf(dbg_viewer,"Coarse problem is singular\n");CHKERRQ(ierr); } if (compute_eigs) { PetscReal lambda_max_s,lambda_min_s; KSPConvergedReason reason; ierr = KSPGetType(check_ksp,&check_ksp_type);CHKERRQ(ierr); ierr = KSPGetIterationNumber(check_ksp,&its);CHKERRQ(ierr); ierr = KSPGetConvergedReason(check_ksp,&reason);CHKERRQ(ierr); ierr = KSPComputeExtremeSingularValues(check_ksp,&lambda_max_s,&lambda_min_s);CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(dbg_viewer,"Coarse problem eigenvalues (estimated with %d iterations of %s, conv reason %d): %1.6e %1.6e (%1.6e %1.6e)\n",its,check_ksp_type,reason,lambda_min,lambda_max,lambda_min_s,lambda_max_s);CHKERRQ(ierr); for (i=0;icurrent_level+1));CHKERRQ(ierr); } ierr = VecDestroy(&check_vec);CHKERRQ(ierr); ierr = VecDestroy(&coarse_vec);CHKERRQ(ierr); ierr = KSPDestroy(&check_ksp);CHKERRQ(ierr); if (compute_eigs) { ierr = PetscFree(eigs_r);CHKERRQ(ierr); ierr = PetscFree(eigs_c);CHKERRQ(ierr); } } } ierr = MatNullSpaceDestroy(&CoarseNullSpace);CHKERRQ(ierr); /* print additional info */ if (pcbddc->dbg_flag) { /* waits until all processes reaches this point */ ierr = PetscBarrier((PetscObject)pc);CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"Coarse solver setup completed at level %d\n",pcbddc->current_level);CHKERRQ(ierr); ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); } /* free memory */ ierr = MatDestroy(&coarse_mat);CHKERRQ(ierr); PetscFunctionReturn(0); } PetscErrorCode PCBDDCComputePrimalNumbering(PC pc,PetscInt* coarse_size_n,PetscInt** local_primal_indices_n) { PC_BDDC* pcbddc = (PC_BDDC*)pc->data; PC_IS* pcis = (PC_IS*)pc->data; Mat_IS* matis = (Mat_IS*)pc->pmat->data; IS subset,subset_mult,subset_n; PetscInt local_size,coarse_size=0; PetscInt *local_primal_indices=NULL; const PetscInt *t_local_primal_indices; PetscErrorCode ierr; PetscFunctionBegin; /* Compute global number of coarse dofs */ if (pcbddc->local_primal_size && !pcbddc->local_primal_ref_node) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"BDDC ConstraintsSetUp should be called first"); ierr = ISCreateGeneral(PetscObjectComm((PetscObject)(pc->pmat)),pcbddc->local_primal_size_cc,pcbddc->local_primal_ref_node,PETSC_COPY_VALUES,&subset_n);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingApplyIS(pcis->mapping,subset_n,&subset);CHKERRQ(ierr); ierr = ISDestroy(&subset_n);CHKERRQ(ierr); ierr = ISCreateGeneral(PetscObjectComm((PetscObject)(pc->pmat)),pcbddc->local_primal_size_cc,pcbddc->local_primal_ref_mult,PETSC_COPY_VALUES,&subset_mult);CHKERRQ(ierr); ierr = ISRenumber(subset,subset_mult,&coarse_size,&subset_n);CHKERRQ(ierr); ierr = ISDestroy(&subset);CHKERRQ(ierr); ierr = ISDestroy(&subset_mult);CHKERRQ(ierr); ierr = ISGetLocalSize(subset_n,&local_size);CHKERRQ(ierr); if (local_size != pcbddc->local_primal_size) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Invalid number of local primal indices computed %D != %D",local_size,pcbddc->local_primal_size); ierr = PetscMalloc1(local_size,&local_primal_indices);CHKERRQ(ierr); ierr = ISGetIndices(subset_n,&t_local_primal_indices);CHKERRQ(ierr); ierr = PetscMemcpy(local_primal_indices,t_local_primal_indices,local_size*sizeof(PetscInt));CHKERRQ(ierr); ierr = ISRestoreIndices(subset_n,&t_local_primal_indices);CHKERRQ(ierr); ierr = ISDestroy(&subset_n);CHKERRQ(ierr); /* check numbering */ if (pcbddc->dbg_flag) { PetscScalar coarsesum,*array,*array2; PetscInt i; PetscBool set_error = PETSC_FALSE,set_error_reduced = PETSC_FALSE; ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"--------------------------------------------------\n");CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"Check coarse indices\n");CHKERRQ(ierr); ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); /* counter */ ierr = VecSet(pcis->vec1_global,0.0);CHKERRQ(ierr); ierr = VecSet(pcis->vec1_N,1.0);CHKERRQ(ierr); ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,pcis->vec1_global,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,pcis->vec1_global,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecSet(pcis->vec1_N,0.0);CHKERRQ(ierr); for (i=0;ilocal_primal_size;i++) { ierr = VecSetValue(pcis->vec1_N,pcbddc->primal_indices_local_idxs[i],1.0,INSERT_VALUES);CHKERRQ(ierr); } ierr = VecAssemblyBegin(pcis->vec1_N);CHKERRQ(ierr); ierr = VecAssemblyEnd(pcis->vec1_N);CHKERRQ(ierr); ierr = VecSet(pcis->vec1_global,0.0);CHKERRQ(ierr); ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,pcis->vec1_global,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,pcis->vec1_global,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecGetArray(pcis->vec1_N,&array);CHKERRQ(ierr); ierr = VecGetArray(pcis->vec2_N,&array2);CHKERRQ(ierr); for (i=0;in;i++) { if (array[i] != 0.0 && array[i] != array2[i]) { PetscInt owned = (PetscInt)PetscRealPart(array[i]),gi; PetscInt neigh = (PetscInt)PetscRealPart(array2[i]); set_error = PETSC_TRUE; ierr = ISLocalToGlobalMappingApply(pcis->mapping,1,&i,&gi);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d: local index %d (gid %d) owned by %d processes instead of %d!\n",PetscGlobalRank,i,gi,owned,neigh);CHKERRQ(ierr); } } ierr = VecRestoreArray(pcis->vec2_N,&array2);CHKERRQ(ierr); ierr = MPIU_Allreduce(&set_error,&set_error_reduced,1,MPIU_BOOL,MPI_LOR,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); for (i=0;in;i++) { if (PetscRealPart(array[i]) > 0.0) array[i] = 1.0/PetscRealPart(array[i]); } ierr = VecRestoreArray(pcis->vec1_N,&array);CHKERRQ(ierr); ierr = VecSet(pcis->vec1_global,0.0);CHKERRQ(ierr); ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,pcis->vec1_global,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,pcis->vec1_global,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecSum(pcis->vec1_global,&coarsesum);CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"Size of coarse problem is %d (%lf)\n",coarse_size,PetscRealPart(coarsesum));CHKERRQ(ierr); if (pcbddc->dbg_flag > 1 || set_error_reduced) { PetscInt *gidxs; ierr = PetscMalloc1(pcbddc->local_primal_size,&gidxs);CHKERRQ(ierr); ierr = ISLocalToGlobalMappingApply(pcis->mapping,pcbddc->local_primal_size,pcbddc->primal_indices_local_idxs,gidxs);CHKERRQ(ierr); ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"Distribution of local primal indices\n");CHKERRQ(ierr); ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d\n",PetscGlobalRank);CHKERRQ(ierr); for (i=0;ilocal_primal_size;i++) { ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"local_primal_indices[%d]=%d (%d,%d)\n",i,local_primal_indices[i],pcbddc->primal_indices_local_idxs[i],gidxs[i]);CHKERRQ(ierr); } ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PetscFree(gidxs);CHKERRQ(ierr); } ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); if (set_error_reduced) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_PLIB,"BDDC Numbering of coarse dofs failed"); } /* ierr = PetscPrintf(PetscObjectComm((PetscObject)pc),"Size of coarse problem is %d\n",coarse_size);CHKERRQ(ierr); */ /* get back data */ *coarse_size_n = coarse_size; *local_primal_indices_n = local_primal_indices; PetscFunctionReturn(0); } PetscErrorCode PCBDDCGlobalToLocal(VecScatter g2l_ctx,Vec gwork, Vec lwork, IS globalis, IS* localis) { IS localis_t; PetscInt i,lsize,*idxs,n; PetscScalar *vals; PetscErrorCode ierr; PetscFunctionBegin; /* get indices in local ordering exploiting local to global map */ ierr = ISGetLocalSize(globalis,&lsize);CHKERRQ(ierr); ierr = PetscMalloc1(lsize,&vals);CHKERRQ(ierr); for (i=0;i 0.5) { lsize++; } } ierr = PetscMalloc1(lsize,&idxs);CHKERRQ(ierr); for (i=0,lsize=0;i 0.5) { idxs[lsize++] = i; } } ierr = VecRestoreArrayRead(lwork,(const PetscScalar**)&vals);CHKERRQ(ierr); ierr = ISCreateGeneral(PetscObjectComm((PetscObject)gwork),lsize,idxs,PETSC_OWN_POINTER,&localis_t);CHKERRQ(ierr); *localis = localis_t; PetscFunctionReturn(0); } PetscErrorCode PCBDDCSetUpSubSchurs(PC pc) { PC_IS *pcis=(PC_IS*)pc->data; PC_BDDC *pcbddc=(PC_BDDC*)pc->data; PCBDDCSubSchurs sub_schurs=pcbddc->sub_schurs; Mat S_j; PetscInt *used_xadj,*used_adjncy; PetscBool free_used_adj; PetscErrorCode ierr; PetscFunctionBegin; /* decide the adjacency to be used for determining internal problems for local schur on subsets */ free_used_adj = PETSC_FALSE; if (pcbddc->sub_schurs_layers == -1) { used_xadj = NULL; used_adjncy = NULL; } else { if (pcbddc->sub_schurs_use_useradj && pcbddc->mat_graph->xadj) { used_xadj = pcbddc->mat_graph->xadj; used_adjncy = pcbddc->mat_graph->adjncy; } else if (pcbddc->computed_rowadj) { used_xadj = pcbddc->mat_graph->xadj; used_adjncy = pcbddc->mat_graph->adjncy; } else { PetscBool flg_row=PETSC_FALSE; const PetscInt *xadj,*adjncy; PetscInt nvtxs; ierr = MatGetRowIJ(pcbddc->local_mat,0,PETSC_TRUE,PETSC_FALSE,&nvtxs,&xadj,&adjncy,&flg_row);CHKERRQ(ierr); if (flg_row) { ierr = PetscMalloc2(nvtxs+1,&used_xadj,xadj[nvtxs],&used_adjncy);CHKERRQ(ierr); ierr = PetscMemcpy(used_xadj,xadj,(nvtxs+1)*sizeof(*xadj));CHKERRQ(ierr); ierr = PetscMemcpy(used_adjncy,adjncy,(xadj[nvtxs])*sizeof(*adjncy));CHKERRQ(ierr); free_used_adj = PETSC_TRUE; } else { pcbddc->sub_schurs_layers = -1; used_xadj = NULL; used_adjncy = NULL; } ierr = MatRestoreRowIJ(pcbddc->local_mat,0,PETSC_TRUE,PETSC_FALSE,&nvtxs,&xadj,&adjncy,&flg_row);CHKERRQ(ierr); } } /* setup sub_schurs data */ ierr = MatCreateSchurComplement(pcis->A_II,pcis->A_II,pcis->A_IB,pcis->A_BI,pcis->A_BB,&S_j);CHKERRQ(ierr); if (!sub_schurs->schur_explicit) { /* pcbddc->ksp_D up to date only if not using MatFactor with Schur complement support */ ierr = MatSchurComplementSetKSP(S_j,pcbddc->ksp_D);CHKERRQ(ierr); ierr = PCBDDCSubSchursSetUp(sub_schurs,NULL,S_j,PETSC_FALSE,used_xadj,used_adjncy,pcbddc->sub_schurs_layers,NULL,pcbddc->adaptive_selection,PETSC_FALSE,PETSC_FALSE,0,NULL,NULL,NULL,NULL);CHKERRQ(ierr); } else { PetscBool reuse_solvers = (PetscBool)!pcbddc->use_change_of_basis; PetscBool isseqaij,need_change = PETSC_FALSE; PetscInt benign_n; Mat change = NULL; Vec scaling = NULL; IS change_primal = NULL; if (!pcbddc->use_vertices && reuse_solvers) { PetscInt n_vertices; ierr = ISGetLocalSize(sub_schurs->is_vertices,&n_vertices);CHKERRQ(ierr); reuse_solvers = (PetscBool)!n_vertices; } ierr = PetscObjectTypeCompare((PetscObject)pcbddc->local_mat,MATSEQAIJ,&isseqaij);CHKERRQ(ierr); if (!isseqaij) { Mat_IS* matis = (Mat_IS*)pc->pmat->data; if (matis->A == pcbddc->local_mat) { ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); ierr = MatConvert(matis->A,MATSEQAIJ,MAT_INITIAL_MATRIX,&pcbddc->local_mat);CHKERRQ(ierr); } else { ierr = MatConvert(pcbddc->local_mat,MATSEQAIJ,MAT_INPLACE_MATRIX,&pcbddc->local_mat);CHKERRQ(ierr); } } if (!pcbddc->benign_change_explicit) { benign_n = pcbddc->benign_n; } else { benign_n = 0; } /* sub_schurs->change is a local object; instead, PCBDDCConstraintsSetUp and the quantities used in the test below are logically collective on pc. We need a global reduction to avoid possible deadlocks. We assume that sub_schurs->change is created once, and then reused for different solves, unless the topography has been recomputed */ if (pcbddc->adaptive_userdefined || (pcbddc->deluxe_zerorows && !pcbddc->use_change_of_basis)) { PetscBool have_loc_change = (PetscBool)(!!sub_schurs->change); ierr = MPIU_Allreduce(&have_loc_change,&need_change,1,MPIU_BOOL,MPI_LOR,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); need_change = (PetscBool)(!need_change); } /* If the user defines additional constraints, we import them here. We need to compute the change of basis according to the quadrature weights attached to pmat via MatSetNearNullSpace, and this could not be done (at the moment) without some hacking */ if (need_change) { PC_IS *pcisf; PC_BDDC *pcbddcf; PC pcf; if (pcbddc->sub_schurs_rebuild) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot compute change of basis with a different graph"); ierr = PCCreate(PetscObjectComm((PetscObject)pc),&pcf);CHKERRQ(ierr); ierr = PCSetOperators(pcf,pc->mat,pc->pmat);CHKERRQ(ierr); ierr = PCSetType(pcf,PCBDDC);CHKERRQ(ierr); /* hacks */ pcisf = (PC_IS*)pcf->data; pcisf->is_B_local = pcis->is_B_local; pcisf->vec1_N = pcis->vec1_N; pcisf->BtoNmap = pcis->BtoNmap; pcisf->n = pcis->n; pcisf->n_B = pcis->n_B; pcbddcf = (PC_BDDC*)pcf->data; ierr = PetscFree(pcbddcf->mat_graph);CHKERRQ(ierr); pcbddcf->mat_graph = pcbddc->mat_graph; pcbddcf->use_faces = PETSC_TRUE; pcbddcf->use_change_of_basis = PETSC_TRUE; pcbddcf->use_change_on_faces = PETSC_TRUE; pcbddcf->use_qr_single = PETSC_TRUE; pcbddcf->fake_change = PETSC_TRUE; /* setup constraints so that we can get information on primal vertices and change of basis (in local numbering) */ ierr = PCBDDCConstraintsSetUp(pcf);CHKERRQ(ierr); sub_schurs->change_with_qr = pcbddcf->use_qr_single; ierr = ISCreateGeneral(PETSC_COMM_SELF,pcbddcf->n_vertices,pcbddcf->local_primal_ref_node,PETSC_COPY_VALUES,&change_primal);CHKERRQ(ierr); change = pcbddcf->ConstraintMatrix; pcbddcf->ConstraintMatrix = NULL; /* free unneeded memory allocated in PCBDDCConstraintsSetUp */ ierr = PetscFree(pcbddcf->sub_schurs);CHKERRQ(ierr); ierr = MatNullSpaceDestroy(&pcbddcf->onearnullspace);CHKERRQ(ierr); ierr = PetscFree2(pcbddcf->local_primal_ref_node,pcbddcf->local_primal_ref_mult);CHKERRQ(ierr); ierr = PetscFree(pcbddcf->primal_indices_local_idxs);CHKERRQ(ierr); ierr = PetscFree(pcbddcf->onearnullvecs_state);CHKERRQ(ierr); ierr = PetscFree(pcf->data);CHKERRQ(ierr); pcf->ops->destroy = NULL; pcf->ops->reset = NULL; ierr = PCDestroy(&pcf);CHKERRQ(ierr); } if (!pcbddc->use_deluxe_scaling) scaling = pcis->D; ierr = PCBDDCSubSchursSetUp(sub_schurs,pcbddc->local_mat,S_j,pcbddc->sub_schurs_exact_schur,used_xadj,used_adjncy,pcbddc->sub_schurs_layers,scaling,pcbddc->adaptive_selection,reuse_solvers,pcbddc->benign_saddle_point,benign_n,pcbddc->benign_p0_lidx,pcbddc->benign_zerodiag_subs,change,change_primal);CHKERRQ(ierr); ierr = MatDestroy(&change);CHKERRQ(ierr); ierr = ISDestroy(&change_primal);CHKERRQ(ierr); } ierr = MatDestroy(&S_j);CHKERRQ(ierr); /* free adjacency */ if (free_used_adj) { ierr = PetscFree2(used_xadj,used_adjncy);CHKERRQ(ierr); } PetscFunctionReturn(0); } PetscErrorCode PCBDDCInitSubSchurs(PC pc) { PC_IS *pcis=(PC_IS*)pc->data; PC_BDDC *pcbddc=(PC_BDDC*)pc->data; PCBDDCGraph graph; PetscErrorCode ierr; PetscFunctionBegin; /* attach interface graph for determining subsets */ if (pcbddc->sub_schurs_rebuild) { /* in case rebuild has been requested, it uses a graph generated only by the neighbouring information */ IS verticesIS,verticescomm; PetscInt vsize,*idxs; ierr = PCBDDCGraphGetCandidatesIS(pcbddc->mat_graph,NULL,NULL,NULL,NULL,&verticesIS);CHKERRQ(ierr); ierr = ISGetSize(verticesIS,&vsize);CHKERRQ(ierr); ierr = ISGetIndices(verticesIS,(const PetscInt**)&idxs);CHKERRQ(ierr); ierr = ISCreateGeneral(PetscObjectComm((PetscObject)pc),vsize,idxs,PETSC_COPY_VALUES,&verticescomm);CHKERRQ(ierr); ierr = ISRestoreIndices(verticesIS,(const PetscInt**)&idxs);CHKERRQ(ierr); ierr = PCBDDCGraphRestoreCandidatesIS(pcbddc->mat_graph,NULL,NULL,NULL,NULL,&verticesIS);CHKERRQ(ierr); ierr = PCBDDCGraphCreate(&graph);CHKERRQ(ierr); ierr = PCBDDCGraphInit(graph,pcbddc->mat_graph->l2gmap,pcbddc->mat_graph->nvtxs_global,pcbddc->graphmaxcount);CHKERRQ(ierr); ierr = PCBDDCGraphSetUp(graph,pcbddc->mat_graph->custom_minimal_size,NULL,pcbddc->DirichletBoundariesLocal,0,NULL,verticescomm);CHKERRQ(ierr); ierr = ISDestroy(&verticescomm);CHKERRQ(ierr); ierr = PCBDDCGraphComputeConnectedComponents(graph);CHKERRQ(ierr); } else { graph = pcbddc->mat_graph; } /* print some info */ if (pcbddc->dbg_flag && !pcbddc->sub_schurs_rebuild) { IS vertices; PetscInt nv,nedges,nfaces; ierr = PCBDDCGraphASCIIView(graph,pcbddc->dbg_flag,pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PCBDDCGraphGetCandidatesIS(graph,&nfaces,NULL,&nedges,NULL,&vertices);CHKERRQ(ierr); ierr = ISGetSize(vertices,&nv);CHKERRQ(ierr); ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"--------------------------------------------------------------\n");CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d got %02d local candidate vertices (%d)\n",PetscGlobalRank,nv,pcbddc->use_vertices);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d got %02d local candidate edges (%d)\n",PetscGlobalRank,nedges,pcbddc->use_edges);CHKERRQ(ierr); ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d got %02d local candidate faces (%d)\n",PetscGlobalRank,nfaces,pcbddc->use_faces);CHKERRQ(ierr); ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PetscViewerASCIIPopSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); ierr = PCBDDCGraphRestoreCandidatesIS(graph,&nfaces,NULL,&nedges,NULL,&vertices);CHKERRQ(ierr); } /* sub_schurs init */ if (!pcbddc->sub_schurs) { ierr = PCBDDCSubSchursCreate(&pcbddc->sub_schurs);CHKERRQ(ierr); } ierr = PCBDDCSubSchursInit(pcbddc->sub_schurs,pcis->is_I_local,pcis->is_B_local,graph,pcis->BtoNmap,pcbddc->sub_schurs_rebuild);CHKERRQ(ierr); pcbddc->sub_schurs->prefix = ((PetscObject)pc)->prefix; /* free graph struct */ if (pcbddc->sub_schurs_rebuild) { ierr = PCBDDCGraphDestroy(&graph);CHKERRQ(ierr); } PetscFunctionReturn(0); } PetscErrorCode PCBDDCCheckOperator(PC pc) { PC_IS *pcis=(PC_IS*)pc->data; PC_BDDC *pcbddc=(PC_BDDC*)pc->data; PetscErrorCode ierr; PetscFunctionBegin; if (pcbddc->n_vertices == pcbddc->local_primal_size) { IS zerodiag = NULL; Mat S_j,B0_B=NULL; Vec dummy_vec=NULL,vec_check_B,vec_scale_P; PetscScalar *p0_check,*array,*array2; PetscReal norm; PetscInt i; /* B0 and B0_B */ if (zerodiag) { IS dummy; ierr = ISCreateStride(PETSC_COMM_SELF,pcbddc->benign_n,0,1,&dummy);CHKERRQ(ierr); ierr = MatCreateSubMatrix(pcbddc->benign_B0,dummy,pcis->is_B_local,MAT_INITIAL_MATRIX,&B0_B);CHKERRQ(ierr); ierr = MatCreateVecs(B0_B,NULL,&dummy_vec);CHKERRQ(ierr); ierr = ISDestroy(&dummy);CHKERRQ(ierr); } /* I need a primal vector to scale primal nodes since BDDC sums contibutions */ ierr = VecDuplicate(pcbddc->vec1_P,&vec_scale_P);CHKERRQ(ierr); ierr = VecSet(pcbddc->vec1_P,1.0);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->vec1_P,pcbddc->coarse_vec,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->vec1_P,pcbddc->coarse_vec,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,vec_scale_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,vec_scale_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecReciprocal(vec_scale_P);CHKERRQ(ierr); /* S_j */ ierr = MatCreateSchurComplement(pcis->A_II,pcis->A_II,pcis->A_IB,pcis->A_BI,pcis->A_BB,&S_j);CHKERRQ(ierr); ierr = MatSchurComplementSetKSP(S_j,pcbddc->ksp_D);CHKERRQ(ierr); /* mimic vector in \widetilde{W}_\Gamma */ ierr = VecSetRandom(pcis->vec1_N,NULL);CHKERRQ(ierr); /* continuous in primal space */ ierr = VecSetRandom(pcbddc->coarse_vec,NULL);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); ierr = PetscCalloc1(pcbddc->benign_n,&p0_check);CHKERRQ(ierr); for (i=0;ibenign_n;i++) p0_check[i] = array[pcbddc->local_primal_size-pcbddc->benign_n+i]; ierr = VecSetValues(pcis->vec1_N,pcbddc->local_primal_size,pcbddc->local_primal_ref_node,array,INSERT_VALUES);CHKERRQ(ierr); ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); ierr = VecAssemblyBegin(pcis->vec1_N);CHKERRQ(ierr); ierr = VecAssemblyEnd(pcis->vec1_N);CHKERRQ(ierr); ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_N,pcis->vec2_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_N,pcis->vec2_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecDuplicate(pcis->vec2_B,&vec_check_B);CHKERRQ(ierr); ierr = VecCopy(pcis->vec2_B,vec_check_B);CHKERRQ(ierr); /* assemble rhs for coarse problem */ /* widetilde{S}_\Gamma w_\Gamma + \widetilde{B0}^T_B p0 */ /* local with Schur */ ierr = MatMult(S_j,pcis->vec2_B,pcis->vec1_B);CHKERRQ(ierr); if (zerodiag) { ierr = VecGetArray(dummy_vec,&array);CHKERRQ(ierr); for (i=0;ibenign_n;i++) array[i] = p0_check[i]; ierr = VecRestoreArray(dummy_vec,&array);CHKERRQ(ierr); ierr = MatMultTransposeAdd(B0_B,dummy_vec,pcis->vec1_B,pcis->vec1_B);CHKERRQ(ierr); } /* sum on primal nodes the local contributions */ ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecGetArray(pcis->vec1_N,&array);CHKERRQ(ierr); ierr = VecGetArray(pcbddc->vec1_P,&array2);CHKERRQ(ierr); for (i=0;ilocal_primal_size;i++) array2[i] = array[pcbddc->local_primal_ref_node[i]]; ierr = VecRestoreArray(pcbddc->vec1_P,&array2);CHKERRQ(ierr); ierr = VecRestoreArray(pcis->vec1_N,&array);CHKERRQ(ierr); ierr = VecSet(pcbddc->coarse_vec,0.);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->vec1_P,pcbddc->coarse_vec,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->vec1_P,pcbddc->coarse_vec,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); /* scale primal nodes (BDDC sums contibutions) */ ierr = VecPointwiseMult(pcbddc->vec1_P,vec_scale_P,pcbddc->vec1_P);CHKERRQ(ierr); ierr = VecSetValues(pcis->vec1_N,pcbddc->local_primal_size,pcbddc->local_primal_ref_node,array,INSERT_VALUES);CHKERRQ(ierr); ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); ierr = VecAssemblyBegin(pcis->vec1_N);CHKERRQ(ierr); ierr = VecAssemblyEnd(pcis->vec1_N);CHKERRQ(ierr); ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); /* global: \widetilde{B0}_B w_\Gamma */ if (zerodiag) { ierr = MatMult(B0_B,pcis->vec2_B,dummy_vec);CHKERRQ(ierr); ierr = VecGetArray(dummy_vec,&array);CHKERRQ(ierr); for (i=0;ibenign_n;i++) pcbddc->benign_p0[i] = array[i]; ierr = VecRestoreArray(dummy_vec,&array);CHKERRQ(ierr); } /* BDDC */ ierr = VecSet(pcis->vec1_D,0.);CHKERRQ(ierr); ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); ierr = VecCopy(pcis->vec1_B,pcis->vec2_B);CHKERRQ(ierr); ierr = VecAXPY(pcis->vec1_B,-1.0,vec_check_B);CHKERRQ(ierr); ierr = VecNorm(pcis->vec1_B,NORM_INFINITY,&norm);CHKERRQ(ierr); PetscPrintf(PETSC_COMM_SELF,"[%d] BDDC local error is %1.4e\n",PetscGlobalRank,norm); for (i=0;ibenign_n;i++) { PetscPrintf(PETSC_COMM_SELF,"[%d] BDDC p0[%d] error is %1.4e\n",PetscGlobalRank,i,PetscAbsScalar(pcbddc->benign_p0[i]-p0_check[i])); } ierr = PetscFree(p0_check);CHKERRQ(ierr); ierr = VecDestroy(&vec_scale_P);CHKERRQ(ierr); ierr = VecDestroy(&vec_check_B);CHKERRQ(ierr); ierr = VecDestroy(&dummy_vec);CHKERRQ(ierr); ierr = MatDestroy(&S_j);CHKERRQ(ierr); ierr = MatDestroy(&B0_B);CHKERRQ(ierr); } PetscFunctionReturn(0); } #include <../src/mat/impls/aij/mpi/mpiaij.h> PetscErrorCode MatMPIAIJRestrict(Mat A, MPI_Comm ccomm, Mat *B) { Mat At; IS rows; PetscInt rst,ren; PetscErrorCode ierr; PetscLayout rmap; PetscFunctionBegin; rst = ren = 0; if (ccomm != MPI_COMM_NULL) { ierr = PetscLayoutCreate(ccomm,&rmap);CHKERRQ(ierr); ierr = PetscLayoutSetSize(rmap,A->rmap->N);CHKERRQ(ierr); ierr = PetscLayoutSetBlockSize(rmap,1);CHKERRQ(ierr); ierr = PetscLayoutSetUp(rmap);CHKERRQ(ierr); ierr = PetscLayoutGetRange(rmap,&rst,&ren);CHKERRQ(ierr); } ierr = ISCreateStride(PetscObjectComm((PetscObject)A),ren-rst,rst,1,&rows);CHKERRQ(ierr); ierr = MatCreateSubMatrix(A,rows,NULL,MAT_INITIAL_MATRIX,&At);CHKERRQ(ierr); ierr = ISDestroy(&rows);CHKERRQ(ierr); if (ccomm != MPI_COMM_NULL) { Mat_MPIAIJ *a,*b; IS from,to; Vec gvec; PetscInt lsize; ierr = MatCreate(ccomm,B);CHKERRQ(ierr); ierr = MatSetSizes(*B,ren-rst,PETSC_DECIDE,PETSC_DECIDE,At->cmap->N);CHKERRQ(ierr); ierr = MatSetType(*B,MATAIJ);CHKERRQ(ierr); ierr = PetscLayoutDestroy(&((*B)->rmap));CHKERRQ(ierr); ierr = PetscLayoutSetUp((*B)->cmap);CHKERRQ(ierr); a = (Mat_MPIAIJ*)At->data; b = (Mat_MPIAIJ*)(*B)->data; ierr = MPI_Comm_size(ccomm,&b->size);CHKERRQ(ierr); ierr = MPI_Comm_rank(ccomm,&b->rank);CHKERRQ(ierr); ierr = PetscObjectReference((PetscObject)a->A);CHKERRQ(ierr); ierr = PetscObjectReference((PetscObject)a->B);CHKERRQ(ierr); b->A = a->A; b->B = a->B; b->donotstash = a->donotstash; b->roworiented = a->roworiented; b->rowindices = 0; b->rowvalues = 0; b->getrowactive = PETSC_FALSE; (*B)->rmap = rmap; (*B)->factortype = A->factortype; (*B)->assembled = PETSC_TRUE; (*B)->insertmode = NOT_SET_VALUES; (*B)->preallocated = PETSC_TRUE; if (a->colmap) { #if defined(PETSC_USE_CTABLE) ierr = PetscTableCreateCopy(a->colmap,&b->colmap);CHKERRQ(ierr); #else ierr = PetscMalloc1(At->cmap->N,&b->colmap);CHKERRQ(ierr); ierr = PetscLogObjectMemory((PetscObject)*B,At->cmap->N*sizeof(PetscInt));CHKERRQ(ierr); ierr = PetscMemcpy(b->colmap,a->colmap,At->cmap->N*sizeof(PetscInt));CHKERRQ(ierr); #endif } else b->colmap = 0; if (a->garray) { PetscInt len; len = a->B->cmap->n; ierr = PetscMalloc1(len+1,&b->garray);CHKERRQ(ierr); ierr = PetscLogObjectMemory((PetscObject)(*B),len*sizeof(PetscInt));CHKERRQ(ierr); if (len) { ierr = PetscMemcpy(b->garray,a->garray,len*sizeof(PetscInt));CHKERRQ(ierr); } } else b->garray = 0; ierr = PetscObjectReference((PetscObject)a->lvec);CHKERRQ(ierr); b->lvec = a->lvec; ierr = PetscLogObjectParent((PetscObject)*B,(PetscObject)b->lvec);CHKERRQ(ierr); /* cannot use VecScatterCopy */ ierr = VecGetLocalSize(b->lvec,&lsize);CHKERRQ(ierr); ierr = ISCreateGeneral(ccomm,lsize,b->garray,PETSC_USE_POINTER,&from);CHKERRQ(ierr); ierr = ISCreateStride(PETSC_COMM_SELF,lsize,0,1,&to);CHKERRQ(ierr); ierr = MatCreateVecs(*B,&gvec,NULL);CHKERRQ(ierr); ierr = VecScatterCreate(gvec,from,b->lvec,to,&b->Mvctx);CHKERRQ(ierr); ierr = PetscLogObjectParent((PetscObject)*B,(PetscObject)b->Mvctx);CHKERRQ(ierr); ierr = ISDestroy(&from);CHKERRQ(ierr); ierr = ISDestroy(&to);CHKERRQ(ierr); ierr = VecDestroy(&gvec);CHKERRQ(ierr); } ierr = MatDestroy(&At);CHKERRQ(ierr); PetscFunctionReturn(0); }