1*7dda7e10SBarry Smith /*$Id: aijfact.c,v 1.141 2000/01/11 21:00:37 bsmith Exp bsmith $*/ 2289bc588SBarry Smith 370f55243SBarry Smith #include "src/mat/impls/aij/seq/aij.h" 490f02eecSBarry Smith #include "src/vec/vecimpl.h" 51c4feecaSSatish Balay #include "src/inline/dot.h" 63b2fbd54SBarry Smith 75615d1e5SSatish Balay #undef __FUNC__ 891e9ee9fSBarry Smith #define __FUNC__ "MatOrdering_Flow_SeqAIJ" 991e9ee9fSBarry Smith int MatOrdering_Flow_SeqAIJ(Mat mat,MatOrderingType type,IS *irow,IS *icol) 103b2fbd54SBarry Smith { 113a40ed3dSBarry Smith PetscFunctionBegin; 123a40ed3dSBarry Smith 13e3372554SBarry Smith SETERRQ(PETSC_ERR_SUP,0,"Code not written"); 14aa482453SBarry Smith #if !defined(PETSC_USE_DEBUG) 15d64ed03dSBarry Smith PetscFunctionReturn(0); 16d64ed03dSBarry Smith #endif 173b2fbd54SBarry Smith } 183b2fbd54SBarry Smith 1986bacbd2SBarry Smith 2086bacbd2SBarry Smith extern int MatMarkDiagonal_SeqAIJ(Mat); 2186bacbd2SBarry Smith extern int Mat_AIJ_CheckInode(Mat); 2286bacbd2SBarry Smith 235bd2ddc7SBarry Smith extern int SPARSEKIT2dperm(int*,Scalar*,int*,int*,Scalar*,int*,int*,int*,int*,int*); 24329f5518SBarry Smith extern int SPARSEKIT2ilutp(int*,Scalar*,int*,int*,int*,PetscReal*,PetscReal*,int*,Scalar*,int*,int*,int*,Scalar*,int*,int*,int*); 255bd2ddc7SBarry Smith extern int SPARSEKIT2msrcsr(int*,Scalar*,int*,Scalar*,int*,int*,Scalar*,int*); 2686bacbd2SBarry Smith 2786bacbd2SBarry Smith #undef __FUNC__ 2886bacbd2SBarry Smith #define __FUNC__ "MatILUDTFactor_SeqAIJ" 2986bacbd2SBarry Smith /* ------------------------------------------------------------ 3086bacbd2SBarry Smith 3186bacbd2SBarry Smith This interface was contribed by Tony Caola 3286bacbd2SBarry Smith 3386bacbd2SBarry Smith This routine is an interface to the pivoting drop-tolerance 345bd2ddc7SBarry Smith ILU routine written by Yousef Saad (saad@cs.umn.edu) as part of 355bd2ddc7SBarry Smith SPARSEKIT2. 365bd2ddc7SBarry Smith 375bd2ddc7SBarry Smith The SPARSEKIT2 routines used here are covered by the GNU 385bd2ddc7SBarry Smith copyright; see the file gnu in this directory. 3986bacbd2SBarry Smith 4086bacbd2SBarry Smith Thanks to Prof. Saad, Dr. Hysom, and Dr. Smith for their 4186bacbd2SBarry Smith help in getting this routine ironed out. 4286bacbd2SBarry Smith 435bd2ddc7SBarry Smith The major drawback to this routine is that if info->fill is 445bd2ddc7SBarry Smith not large enough it fails rather than allocating more space; 455bd2ddc7SBarry Smith this can be fixed by hacking/improving the f2c version of 465bd2ddc7SBarry Smith Yousef Saad's code. 4786bacbd2SBarry Smith 485bd2ddc7SBarry Smith ishift = 0, for indices start at 1 495bd2ddc7SBarry Smith ishift = 1, for indices starting at 0 5086bacbd2SBarry Smith ------------------------------------------------------------ 5186bacbd2SBarry Smith */ 5286bacbd2SBarry Smith 5386bacbd2SBarry Smith int MatILUDTFactor_SeqAIJ(Mat A,MatILUInfo *info,IS isrow,IS iscol,Mat *fact) 5486bacbd2SBarry Smith { 5586bacbd2SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 5607d2056aSBarry Smith IS iscolf,isicol,isirow; 5786bacbd2SBarry Smith PetscTruth reorder; 585bd2ddc7SBarry Smith int *c,*r,*ic,ierr,i,n = a->m; 59329f5518SBarry Smith int *old_i = a->i,*old_j = a->j,*new_i,*old_i2 = 0,*old_j2 = 0,*new_j; 60313ae024SBarry Smith int *ordcol,*iwk,*iperm,*jw; 615bd2ddc7SBarry Smith int ishift = !a->indexshift; 62b19713cbSBarry Smith int jmax,lfill,job,*o_i,*o_j; 63329f5518SBarry Smith Scalar *old_a = a->a,*w,*new_a,*old_a2 = 0,*wk,*o_a; 64329f5518SBarry Smith PetscReal permtol,af; 6586bacbd2SBarry Smith 6686bacbd2SBarry Smith PetscFunctionBegin; 6786bacbd2SBarry Smith 6886bacbd2SBarry Smith if (info->dt == PETSC_DEFAULT) info->dt = .005; 6986bacbd2SBarry Smith if (info->dtcount == PETSC_DEFAULT) info->dtcount = (int)(1.5*a->rmax); 7086bacbd2SBarry Smith if (info->dtcol == PETSC_DEFAULT) info->dtcol = .01; 716faa4630SBarry Smith if (info->fill == PETSC_DEFAULT) info->fill = ((double)(n*info->dtcount))/a->nz; 726faa4630SBarry Smith lfill = (int)(info->dtcount/2.0); 736faa4630SBarry Smith jmax = (int)(info->fill*a->nz); 7486bacbd2SBarry Smith permtol = info->dtcol; 7586bacbd2SBarry Smith 7686bacbd2SBarry Smith 7786bacbd2SBarry Smith /* ------------------------------------------------------------ 7886bacbd2SBarry Smith If reorder=.TRUE., then the original matrix has to be 7986bacbd2SBarry Smith reordered to reflect the user selected ordering scheme, and 8086bacbd2SBarry Smith then de-reordered so it is in it's original format. 8186bacbd2SBarry Smith Because Saad's dperm() is NOT in place, we have to copy 8286bacbd2SBarry Smith the original matrix and allocate more storage. . . 8386bacbd2SBarry Smith ------------------------------------------------------------ 8486bacbd2SBarry Smith */ 8586bacbd2SBarry Smith 8686bacbd2SBarry Smith /* set reorder to true if either isrow or iscol is not identity */ 8786bacbd2SBarry Smith ierr = ISIdentity(isrow,&reorder);CHKERRQ(ierr); 8886bacbd2SBarry Smith if (reorder) {ierr = ISIdentity(iscol,&reorder);CHKERRQ(ierr);} 8986bacbd2SBarry Smith reorder = PetscNot(reorder); 9086bacbd2SBarry Smith 9186bacbd2SBarry Smith 9286bacbd2SBarry Smith /* storage for ilu factor */ 9386bacbd2SBarry Smith new_i = (int*) PetscMalloc((n+1)*sizeof(int));CHKPTRQ(new_i); 9486bacbd2SBarry Smith new_j = (int*) PetscMalloc(jmax*sizeof(int));CHKPTRQ(new_j); 9586bacbd2SBarry Smith new_a = (Scalar*)PetscMalloc(jmax*sizeof(Scalar));CHKPTRQ(new_a); 9686bacbd2SBarry Smith 9786bacbd2SBarry Smith ordcol = (int*)PetscMalloc(n*sizeof(int));CHKPTRQ(ordcol); 9886bacbd2SBarry Smith 9986bacbd2SBarry Smith /* ------------------------------------------------------------ 10086bacbd2SBarry Smith Make sure that everything is Fortran formatted (1-Based) 10186bacbd2SBarry Smith ------------------------------------------------------------ 10286bacbd2SBarry Smith */ 103b19713cbSBarry Smith if (ishift) { 104b19713cbSBarry Smith for (i=old_i[0];i<old_i[n];i++) { 105b19713cbSBarry Smith old_j[i]++; 10686bacbd2SBarry Smith } 107b19713cbSBarry Smith for(i=0;i<n+1;i++) { 108b19713cbSBarry Smith old_i[i]++; 109b19713cbSBarry Smith }; 11086bacbd2SBarry Smith }; 11186bacbd2SBarry Smith 11286bacbd2SBarry Smith if (reorder) { 113c0c2c81eSBarry Smith ierr = ISGetIndices(iscol,&c); CHKERRQ(ierr); 114c0c2c81eSBarry Smith ierr = ISGetIndices(isrow,&r); CHKERRQ(ierr); 115c0c2c81eSBarry Smith for(i=0;i<n;i++) { 116c0c2c81eSBarry Smith r[i] = r[i]+1; 117c0c2c81eSBarry Smith c[i] = c[i]+1; 118c0c2c81eSBarry Smith } 119313ae024SBarry Smith old_i2 = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(old_i2); 120313ae024SBarry Smith old_j2 = (int*)PetscMalloc((old_i[n]-old_i[0]+1)*sizeof(int));CHKPTRQ(old_j2); 121313ae024SBarry Smith old_a2 = (Scalar*)PetscMalloc((old_i[n]-old_i[0]+1)*sizeof(Scalar));CHKPTRQ(old_a2); 1225bd2ddc7SBarry Smith job = 3; SPARSEKIT2dperm(&n,old_a,old_j,old_i,old_a2,old_j2,old_i2,r,c,&job); 123c0c2c81eSBarry Smith for (i=0;i<n;i++) { 124c0c2c81eSBarry Smith r[i] = r[i]-1; 125c0c2c81eSBarry Smith c[i] = c[i]-1; 126c0c2c81eSBarry Smith } 127c0c2c81eSBarry Smith ierr = ISRestoreIndices(iscol,&c);CHKERRQ(ierr); 128c0c2c81eSBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 129b19713cbSBarry Smith o_a = old_a2; 130b19713cbSBarry Smith o_j = old_j2; 131b19713cbSBarry Smith o_i = old_i2; 132b19713cbSBarry Smith } else { 133b19713cbSBarry Smith o_a = old_a; 134b19713cbSBarry Smith o_j = old_j; 135b19713cbSBarry Smith o_i = old_i; 13686bacbd2SBarry Smith } 13786bacbd2SBarry Smith 13886bacbd2SBarry Smith /* ------------------------------------------------------------ 13986bacbd2SBarry Smith Call Saad's ilutp() routine to generate the factorization 14086bacbd2SBarry Smith ------------------------------------------------------------ 14186bacbd2SBarry Smith */ 14286bacbd2SBarry Smith 14386bacbd2SBarry Smith iperm = (int*) PetscMalloc(2*n*sizeof(int));CHKPTRQ(iperm); 14486bacbd2SBarry Smith jw = (int*) PetscMalloc(2*n*sizeof(int));CHKPTRQ(jw); 14586bacbd2SBarry Smith w = (Scalar*)PetscMalloc(n*sizeof(Scalar));CHKPTRQ(w); 14686bacbd2SBarry Smith 1475bd2ddc7SBarry Smith SPARSEKIT2ilutp(&n,o_a,o_j,o_i,&lfill,&info->dt,&permtol,&n,new_a,new_j,new_i,&jmax,w,jw,iperm,&ierr); 14886bacbd2SBarry Smith if (ierr) { 14986bacbd2SBarry Smith switch (ierr) { 15086bacbd2SBarry Smith case -3: SETERRQ1(1,1,"ilutp(), matrix U overflows, need larger info->fill value %d",jmax); 15186bacbd2SBarry Smith case -2: SETERRQ1(1,1,"ilutp(), matrix L overflows, need larger info->fill value %d",jmax); 15286bacbd2SBarry Smith case -5: SETERRQ(1,1,"ilutp(), zero row encountered"); 15386bacbd2SBarry Smith case -1: SETERRQ(1,1,"ilutp(), input matrix may be wrong"); 15486bacbd2SBarry Smith case -4: SETERRQ1(1,1,"ilutp(), illegal info->fill value %d",jmax); 15586bacbd2SBarry Smith default: SETERRQ1(1,1,"ilutp(), zero pivot detected on row %d",ierr); 15686bacbd2SBarry Smith } 15786bacbd2SBarry Smith } 15886bacbd2SBarry Smith 15986bacbd2SBarry Smith ierr = PetscFree(w);CHKERRQ(ierr); 16086bacbd2SBarry Smith ierr = PetscFree(jw);CHKERRQ(ierr); 16186bacbd2SBarry Smith 16286bacbd2SBarry Smith /* ------------------------------------------------------------ 16386bacbd2SBarry Smith Saad's routine gives the result in Modified Sparse Row (msr) 16486bacbd2SBarry Smith Convert to Compressed Sparse Row format (csr) 16586bacbd2SBarry Smith ------------------------------------------------------------ 16686bacbd2SBarry Smith */ 16786bacbd2SBarry Smith 16886bacbd2SBarry Smith wk = (Scalar*)PetscMalloc(n*sizeof(Scalar));CHKPTRQ(wk); 16986bacbd2SBarry Smith iwk = (int*) PetscMalloc((n+1)*sizeof(int));CHKPTRQ(iwk); 17086bacbd2SBarry Smith 1715bd2ddc7SBarry Smith SPARSEKIT2msrcsr(&n,new_a,new_j,new_a,new_j,new_i,wk,iwk); 17286bacbd2SBarry Smith 17386bacbd2SBarry Smith ierr = PetscFree(iwk);CHKERRQ(ierr); 17486bacbd2SBarry Smith ierr = PetscFree(wk);CHKERRQ(ierr); 17586bacbd2SBarry Smith 17686bacbd2SBarry Smith if (reorder) { 17786bacbd2SBarry Smith ierr = PetscFree(old_a2);CHKERRQ(ierr); 17886bacbd2SBarry Smith ierr = PetscFree(old_j2);CHKERRQ(ierr); 17986bacbd2SBarry Smith ierr = PetscFree(old_i2);CHKERRQ(ierr); 180313ae024SBarry Smith } else { 181b19713cbSBarry Smith /* fix permutation of old_j that the factorization introduced */ 182b19713cbSBarry Smith for (i=old_i[0]; i<=old_i[n]; i++) { 183b19713cbSBarry Smith old_j[i-1] = iperm[old_j[i-1]-1]; 184b19713cbSBarry Smith } 185b19713cbSBarry Smith } 186b19713cbSBarry Smith 187b801d0f9SBarry Smith /* get rid of the shift to indices starting at 1 */ 188b19713cbSBarry Smith if (ishift) { 18986bacbd2SBarry Smith for (i=0; i<n+1; i++) { 190b19713cbSBarry Smith old_i[i]--; 191b19713cbSBarry Smith } 192b19713cbSBarry Smith for (i=old_i[0];i<old_i[n];i++) { 193b19713cbSBarry Smith old_j[i]--; 194b19713cbSBarry Smith } 19586bacbd2SBarry Smith } 19686bacbd2SBarry Smith 197b19713cbSBarry Smith /* Make the factored matrix 0-based */ 198b19713cbSBarry Smith if (ishift) { 19986bacbd2SBarry Smith for (i=0; i<n+1; i++) { 200b19713cbSBarry Smith new_i[i]--; 201b19713cbSBarry Smith } 202b19713cbSBarry Smith for (i=new_i[0];i<new_i[n];i++) { 203b19713cbSBarry Smith new_j[i]--; 204b19713cbSBarry Smith } 20586bacbd2SBarry Smith } 20686bacbd2SBarry Smith 20786bacbd2SBarry Smith /*-- due to the pivoting, we need to reorder iscol to correctly --*/ 20886bacbd2SBarry Smith /*-- permute the right-hand-side and solution vectors --*/ 209a3a5ab83SBarry Smith ierr = ISInvertPermutation(iscol,&isicol);CHKERRQ(ierr); 210a3a5ab83SBarry Smith ierr = ISInvertPermutation(isrow,&isirow);CHKERRQ(ierr); 211c0c2c81eSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 21286bacbd2SBarry Smith for(i=0; i<n; i++) { 21386bacbd2SBarry Smith ordcol[i] = ic[iperm[i]-1]; 21486bacbd2SBarry Smith }; 21586bacbd2SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 21607d2056aSBarry Smith ierr = ISDestroy(isicol);CHKERRQ(ierr); 21786bacbd2SBarry Smith 218c0c2c81eSBarry Smith ierr = PetscFree(iperm);CHKERRQ(ierr); 219c0c2c81eSBarry Smith 220329f5518SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,ordcol,&iscolf);CHKERRQ(ierr); 22107d2056aSBarry Smith ierr = PetscFree(ordcol);CHKERRQ(ierr); 22286bacbd2SBarry Smith 22386bacbd2SBarry Smith /*----- put together the new matrix -----*/ 22486bacbd2SBarry Smith 22586bacbd2SBarry Smith ierr = MatCreateSeqAIJ(A->comm,n,n,0,PETSC_NULL,fact);CHKERRQ(ierr); 22686bacbd2SBarry Smith (*fact)->factor = FACTOR_LU; 22786bacbd2SBarry Smith (*fact)->assembled = PETSC_TRUE; 22886bacbd2SBarry Smith 22986bacbd2SBarry Smith b = (Mat_SeqAIJ*)(*fact)->data; 23086bacbd2SBarry Smith ierr = PetscFree(b->imax);CHKERRQ(ierr); 23186bacbd2SBarry Smith b->sorted = PETSC_FALSE; 23207d2056aSBarry Smith b->singlemalloc = PETSC_FALSE; 233b19713cbSBarry Smith /* the next line frees the default space generated by the MatCreate() */ 23486bacbd2SBarry Smith ierr = PetscFree(b->a);CHKERRQ(ierr); 23586bacbd2SBarry Smith ierr = PetscFree(b->ilen);CHKERRQ(ierr); 23686bacbd2SBarry Smith b->a = new_a; 23786bacbd2SBarry Smith b->j = new_j; 23886bacbd2SBarry Smith b->i = new_i; 23986bacbd2SBarry Smith b->ilen = 0; 24086bacbd2SBarry Smith b->imax = 0; 2411f9e874aSBarry Smith /* I am not sure why these are the inverses of the row and column permutations; but the other way is NO GOOD */ 242313ae024SBarry Smith b->row = isirow; 24386bacbd2SBarry Smith b->col = iscolf; 24486bacbd2SBarry Smith b->solve_work = (Scalar*)PetscMalloc((n+1)*sizeof(Scalar));CHKPTRQ(b->solve_work); 24586bacbd2SBarry Smith b->maxnz = b->nz = new_i[n]; 24686bacbd2SBarry Smith b->indexshift = a->indexshift; 24786bacbd2SBarry Smith ierr = MatMarkDiagonal_SeqAIJ(*fact);CHKERRQ(ierr); 24886bacbd2SBarry Smith (*fact)->info.factor_mallocs = 0; 24986bacbd2SBarry Smith 25086bacbd2SBarry Smith ierr = MatMarkDiagonal_SeqAIJ(A);CHKERRQ(ierr); 25186bacbd2SBarry Smith 25286bacbd2SBarry Smith /* check out for identical nodes. If found, use inode functions */ 25386bacbd2SBarry Smith ierr = Mat_AIJ_CheckInode(*fact);CHKERRQ(ierr); 25486bacbd2SBarry Smith 255431e710aSBarry Smith af = ((double)b->nz)/((double)a->nz) + .001; 256431e710aSBarry Smith PLogInfo(A,"MatILUDTFactor_SeqAIJ:Fill ratio:given %g needed %g\n",info->fill,af); 257431e710aSBarry Smith PLogInfo(A,"MatILUDTFactor_SeqAIJ:Run with -pc_ilu_fill %g or use \n",af); 258431e710aSBarry Smith PLogInfo(A,"MatILUDTFactor_SeqAIJ:PCILUSetFill(pc,%g);\n",af); 259431e710aSBarry Smith PLogInfo(A,"MatILUDTFactor_SeqAIJ:for best performance.\n"); 260431e710aSBarry Smith 26186bacbd2SBarry Smith PetscFunctionReturn(0); 26286bacbd2SBarry Smith } 26386bacbd2SBarry Smith 264289bc588SBarry Smith /* 265289bc588SBarry Smith Factorization code for AIJ format. 266289bc588SBarry Smith */ 2675615d1e5SSatish Balay #undef __FUNC__ 2685615d1e5SSatish Balay #define __FUNC__ "MatLUFactorSymbolic_SeqAIJ" 269329f5518SBarry Smith int MatLUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,PetscReal f,Mat *B) 270289bc588SBarry Smith { 271416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 272289bc588SBarry Smith IS isicol; 273416022c9SBarry Smith int *r,*ic,ierr,i,n = a->m,*ai = a->i,*aj = a->j; 274416022c9SBarry Smith int *ainew,*ajnew,jmax,*fill,*ajtmp,nz,shift = a->indexshift; 27591bf9adeSBarry Smith int *idnew,idx,row,m,fm,nnz,nzi,realloc = 0,nzbd,*im; 276289bc588SBarry Smith 2773a40ed3dSBarry Smith PetscFunctionBegin; 278d3cbd50cSLois Curfman McInnes PetscValidHeaderSpecific(isrow,IS_COOKIE); 279d3cbd50cSLois Curfman McInnes PetscValidHeaderSpecific(iscol,IS_COOKIE); 280f1af5d2fSBarry Smith if (A->M != A->N) SETERRQ(PETSC_ERR_ARG_WRONG,0,"matrix must be square"); 2813b2fbd54SBarry Smith 28278b31e54SBarry Smith ierr = ISInvertPermutation(iscol,&isicol);CHKERRQ(ierr); 283ac121b3dSBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 284ac121b3dSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 285289bc588SBarry Smith 286289bc588SBarry Smith /* get new row pointers */ 2870452661fSBarry Smith ainew = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(ainew); 288dbb450caSBarry Smith ainew[0] = -shift; 289289bc588SBarry Smith /* don't know how many column pointers are needed so estimate */ 290dbb450caSBarry Smith jmax = (int)(f*ai[n]+(!shift)); 2910452661fSBarry Smith ajnew = (int*)PetscMalloc((jmax)*sizeof(int));CHKPTRQ(ajnew); 292289bc588SBarry Smith /* fill is a linked list of nonzeros in active row */ 2930452661fSBarry Smith fill = (int*)PetscMalloc((2*n+1)*sizeof(int));CHKPTRQ(fill); 2942fb3ed76SBarry Smith im = fill + n + 1; 295289bc588SBarry Smith /* idnew is location of diagonal in factor */ 2960452661fSBarry Smith idnew = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(idnew); 297dbb450caSBarry Smith idnew[0] = -shift; 298289bc588SBarry Smith 299289bc588SBarry Smith for (i=0; i<n; i++) { 300289bc588SBarry Smith /* first copy previous fill into linked list */ 301289bc588SBarry Smith nnz = nz = ai[r[i]+1] - ai[r[i]]; 3026b96ef82SSatish Balay if (!nz) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,1,"Empty row in matrix"); 303dbb450caSBarry Smith ajtmp = aj + ai[r[i]] + shift; 304289bc588SBarry Smith fill[n] = n; 305289bc588SBarry Smith while (nz--) { 306289bc588SBarry Smith fm = n; 307dbb450caSBarry Smith idx = ic[*ajtmp++ + shift]; 308289bc588SBarry Smith do { 309289bc588SBarry Smith m = fm; 310289bc588SBarry Smith fm = fill[m]; 311289bc588SBarry Smith } while (fm < idx); 312289bc588SBarry Smith fill[m] = idx; 313289bc588SBarry Smith fill[idx] = fm; 314289bc588SBarry Smith } 315289bc588SBarry Smith row = fill[n]; 316289bc588SBarry Smith while (row < i) { 317dbb450caSBarry Smith ajtmp = ajnew + idnew[row] + (!shift); 3182fb3ed76SBarry Smith nzbd = 1 + idnew[row] - ainew[row]; 3192fb3ed76SBarry Smith nz = im[row] - nzbd; 320289bc588SBarry Smith fm = row; 3212fb3ed76SBarry Smith while (nz-- > 0) { 322dbb450caSBarry Smith idx = *ajtmp++ + shift; 3232fb3ed76SBarry Smith nzbd++; 3242fb3ed76SBarry Smith if (idx == i) im[row] = nzbd; 325289bc588SBarry Smith do { 326289bc588SBarry Smith m = fm; 327289bc588SBarry Smith fm = fill[m]; 328289bc588SBarry Smith } while (fm < idx); 329289bc588SBarry Smith if (fm != idx) { 330289bc588SBarry Smith fill[m] = idx; 331289bc588SBarry Smith fill[idx] = fm; 332289bc588SBarry Smith fm = idx; 333289bc588SBarry Smith nnz++; 334289bc588SBarry Smith } 335289bc588SBarry Smith } 336289bc588SBarry Smith row = fill[row]; 337289bc588SBarry Smith } 338289bc588SBarry Smith /* copy new filled row into permanent storage */ 339289bc588SBarry Smith ainew[i+1] = ainew[i] + nnz; 340496e697eSBarry Smith if (ainew[i+1] > jmax) { 341ecf371e4SBarry Smith 342ecf371e4SBarry Smith /* estimate how much additional space we will need */ 343ecf371e4SBarry Smith /* use the strategy suggested by David Hysom <hysom@perch-t.icase.edu> */ 344ecf371e4SBarry Smith /* just double the memory each time */ 345ecf371e4SBarry Smith int maxadd = jmax; 346ecf371e4SBarry Smith /* maxadd = (int)((f*(ai[n]+(!shift))*(n-i+5))/n); */ 347bbb6d6a8SBarry Smith if (maxadd < nnz) maxadd = (n-i)*(nnz+1); 3482fb3ed76SBarry Smith jmax += maxadd; 349ecf371e4SBarry Smith 350ecf371e4SBarry Smith /* allocate a longer ajnew */ 3510452661fSBarry Smith ajtmp = (int*)PetscMalloc(jmax*sizeof(int));CHKPTRQ(ajtmp); 352549d3d68SSatish Balay ierr = PetscMemcpy(ajtmp,ajnew,(ainew[i]+shift)*sizeof(int));CHKERRQ(ierr); 353606d414cSSatish Balay ierr = PetscFree(ajnew);CHKERRQ(ierr); 354289bc588SBarry Smith ajnew = ajtmp; 3552fb3ed76SBarry Smith realloc++; /* count how many times we realloc */ 356289bc588SBarry Smith } 357dbb450caSBarry Smith ajtmp = ajnew + ainew[i] + shift; 358289bc588SBarry Smith fm = fill[n]; 359289bc588SBarry Smith nzi = 0; 3602fb3ed76SBarry Smith im[i] = nnz; 361289bc588SBarry Smith while (nnz--) { 362289bc588SBarry Smith if (fm < i) nzi++; 363dbb450caSBarry Smith *ajtmp++ = fm - shift; 364289bc588SBarry Smith fm = fill[fm]; 365289bc588SBarry Smith } 366289bc588SBarry Smith idnew[i] = ainew[i] + nzi; 367289bc588SBarry Smith } 368464e8d44SSatish Balay if (ai[n] != 0) { 369329f5518SBarry Smith PetscReal af = ((PetscReal)ainew[n])/((PetscReal)ai[n]); 370c38d4ed2SBarry Smith PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:Reallocs %d Fill ratio:given %g needed %g\n",realloc,f,af); 371981c4779SBarry Smith PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:Run with -pc_lu_fill %g or use \n",af); 372981c4779SBarry Smith PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:PCLUSetFill(pc,%g);\n",af); 373981c4779SBarry Smith PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:for best performance.\n"); 37405bf559cSSatish Balay } else { 375981c4779SBarry Smith PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ: Empty matrix\n"); 37605bf559cSSatish Balay } 3772fb3ed76SBarry Smith 378898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 379898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3801987afe7SBarry Smith 381606d414cSSatish Balay ierr = PetscFree(fill);CHKERRQ(ierr); 382289bc588SBarry Smith 383289bc588SBarry Smith /* put together the new matrix */ 384b4fd4287SBarry Smith ierr = MatCreateSeqAIJ(A->comm,n,n,0,PETSC_NULL,B);CHKERRQ(ierr); 3851987afe7SBarry Smith PLogObjectParent(*B,isicol); 386416022c9SBarry Smith b = (Mat_SeqAIJ*)(*B)->data; 387606d414cSSatish Balay ierr = PetscFree(b->imax);CHKERRQ(ierr); 3887c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 389e8d4e0b9SBarry Smith /* the next line frees the default space generated by the Create() */ 390606d414cSSatish Balay ierr = PetscFree(b->a);CHKERRQ(ierr); 391606d414cSSatish Balay ierr = PetscFree(b->ilen);CHKERRQ(ierr); 39291bf9adeSBarry Smith b->a = (Scalar*)PetscMalloc((ainew[n]+shift+1)*sizeof(Scalar));CHKPTRQ(b->a); 393416022c9SBarry Smith b->j = ajnew; 394416022c9SBarry Smith b->i = ainew; 395416022c9SBarry Smith b->diag = idnew; 396416022c9SBarry Smith b->ilen = 0; 397416022c9SBarry Smith b->imax = 0; 398416022c9SBarry Smith b->row = isrow; 399416022c9SBarry Smith b->col = iscol; 400c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 401c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 40282bf6240SBarry Smith b->icol = isicol; 40391bf9adeSBarry Smith b->solve_work = (Scalar*)PetscMalloc((n+1)*sizeof(Scalar));CHKPTRQ(b->solve_work); 404416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 4057fd98bd6SLois Curfman McInnes Allocate idnew, solve_work, new a, new j */ 406416022c9SBarry Smith PLogObjectMemory(*B,(ainew[n]+shift-n)*(sizeof(int)+sizeof(Scalar))); 407416022c9SBarry Smith b->maxnz = b->nz = ainew[n] + shift; 4087fd98bd6SLois Curfman McInnes 409329f5518SBarry Smith (*B)->factor = FACTOR_LU; 410ae068f58SLois Curfman McInnes (*B)->info.factor_mallocs = realloc; 411ae068f58SLois Curfman McInnes (*B)->info.fill_ratio_given = f; 412*7dda7e10SBarry Smith (*B)->ops->lufactornumeric = A->ops->lufactornumeric; /* Use Inode variant ONLY if A has inodes */ 413703deb49SSatish Balay 414e93ccc4dSBarry Smith if (ai[n] != 0) { 415329f5518SBarry Smith (*B)->info.fill_ratio_needed = ((PetscReal)ainew[n])/((PetscReal)ai[n]); 416eea2913aSSatish Balay } else { 417eea2913aSSatish Balay (*B)->info.fill_ratio_needed = 0.0; 418eea2913aSSatish Balay } 4193a40ed3dSBarry Smith PetscFunctionReturn(0); 420289bc588SBarry Smith } 4210a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 422184914b5SBarry Smith extern int Mat_AIJ_CheckInode(Mat); 42341c01911SSatish Balay 4245615d1e5SSatish Balay #undef __FUNC__ 4255615d1e5SSatish Balay #define __FUNC__ "MatLUFactorNumeric_SeqAIJ" 426416022c9SBarry Smith int MatLUFactorNumeric_SeqAIJ(Mat A,Mat *B) 427289bc588SBarry Smith { 428416022c9SBarry Smith Mat C = *B; 429416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b = (Mat_SeqAIJ *)C->data; 43082bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 431416022c9SBarry Smith int *r,*ic,ierr,i,j,n = a->m,*ai = b->i,*aj = b->j; 4321987afe7SBarry Smith int *ajtmpold,*ajtmp,nz,row,*ics,shift = a->indexshift; 433f2582111SSatish Balay int *diag_offset = b->diag,diag,k; 434329f5518SBarry Smith int preserve_row_sums = (int)a->ilu_preserve_row_sums,*pj; 435329f5518SBarry Smith Scalar *rtmp,*v,*pc,multiplier,sum,inner_sum,*rowsums = 0,*pv,*rtmps,*u_values; 436329f5518SBarry Smith PetscReal ssum; 437289bc588SBarry Smith 4383a40ed3dSBarry Smith PetscFunctionBegin; 43982bf6240SBarry Smith 44078b31e54SBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 44178b31e54SBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 4420452661fSBarry Smith rtmp = (Scalar*)PetscMalloc((n+1)*sizeof(Scalar));CHKPTRQ(rtmp); 443549d3d68SSatish Balay ierr = PetscMemzero(rtmp,(n+1)*sizeof(Scalar));CHKERRQ(ierr); 4440cf60383SBarry Smith rtmps = rtmp + shift; ics = ic + shift; 445289bc588SBarry Smith 4466cf997b0SBarry Smith /* precalculate row sums */ 44735aab85fSBarry Smith if (preserve_row_sums) { 44835aab85fSBarry Smith rowsums = (Scalar*)PetscMalloc(n*sizeof(Scalar));CHKPTRQ(rowsums); 44935aab85fSBarry Smith for (i=0; i<n; i++) { 45035aab85fSBarry Smith nz = a->i[r[i]+1] - a->i[r[i]]; 45135aab85fSBarry Smith v = a->a + a->i[r[i]] + shift; 45235aab85fSBarry Smith sum = 0.0; 45335aab85fSBarry Smith for (j=0; j<nz; j++) sum += v[j]; 45435aab85fSBarry Smith rowsums[i] = sum; 45535aab85fSBarry Smith } 45635aab85fSBarry Smith } 45735aab85fSBarry Smith 458289bc588SBarry Smith for (i=0; i<n; i++) { 459289bc588SBarry Smith nz = ai[i+1] - ai[i]; 460dbb450caSBarry Smith ajtmp = aj + ai[i] + shift; 46148e94559SBarry Smith for (j=0; j<nz; j++) rtmps[ajtmp[j]] = 0.0; 462289bc588SBarry Smith 463289bc588SBarry Smith /* load in initial (unfactored row) */ 464416022c9SBarry Smith nz = a->i[r[i]+1] - a->i[r[i]]; 465416022c9SBarry Smith ajtmpold = a->j + a->i[r[i]] + shift; 466416022c9SBarry Smith v = a->a + a->i[r[i]] + shift; 4670cf60383SBarry Smith for (j=0; j<nz; j++) rtmp[ics[ajtmpold[j]]] = v[j]; 468289bc588SBarry Smith 469dbb450caSBarry Smith row = *ajtmp++ + shift; 470f2582111SSatish Balay while (row < i) { 4718c37ef55SBarry Smith pc = rtmp + row; 4728c37ef55SBarry Smith if (*pc != 0.0) { 4731987afe7SBarry Smith pv = b->a + diag_offset[row] + shift; 4741987afe7SBarry Smith pj = b->j + diag_offset[row] + (!shift); 47535aab85fSBarry Smith multiplier = *pc / *pv++; 4768c37ef55SBarry Smith *pc = multiplier; 477f2582111SSatish Balay nz = ai[row+1] - diag_offset[row] - 1; 478f2582111SSatish Balay for (j=0; j<nz; j++) rtmps[pj[j]] -= multiplier * pv[j]; 479f2582111SSatish Balay PLogFlops(2*nz); 480289bc588SBarry Smith } 481dbb450caSBarry Smith row = *ajtmp++ + shift; 482289bc588SBarry Smith } 483416022c9SBarry Smith /* finished row so stick it into b->a */ 484416022c9SBarry Smith pv = b->a + ai[i] + shift; 485416022c9SBarry Smith pj = b->j + ai[i] + shift; 4868c37ef55SBarry Smith nz = ai[i+1] - ai[i]; 487416022c9SBarry Smith for (j=0; j<nz; j++) {pv[j] = rtmps[pj[j]];} 48835aab85fSBarry Smith diag = diag_offset[i] - ai[i]; 48935aab85fSBarry Smith /* 49035aab85fSBarry Smith Possibly adjust diagonal entry on current row to force 49135aab85fSBarry Smith LU matrix to have same row sum as initial matrix. 49235aab85fSBarry Smith */ 493d708749eSBarry Smith if (pv[diag] == 0.0) { 4946cf997b0SBarry Smith SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,0,"Zero pivot row %d",i); 495d708749eSBarry Smith } 49635aab85fSBarry Smith if (preserve_row_sums) { 49735aab85fSBarry Smith pj = b->j + ai[i] + shift; 49835aab85fSBarry Smith sum = rowsums[i]; 49935aab85fSBarry Smith for (j=0; j<diag; j++) { 50035aab85fSBarry Smith u_values = b->a + diag_offset[pj[j]] + shift; 50135aab85fSBarry Smith nz = ai[pj[j]+1] - diag_offset[pj[j]]; 50235aab85fSBarry Smith inner_sum = 0.0; 50335aab85fSBarry Smith for (k=0; k<nz; k++) { 50435aab85fSBarry Smith inner_sum += u_values[k]; 50535aab85fSBarry Smith } 50635aab85fSBarry Smith sum -= pv[j]*inner_sum; 50735aab85fSBarry Smith 50835aab85fSBarry Smith } 50935aab85fSBarry Smith nz = ai[i+1] - diag_offset[i] - 1; 51035aab85fSBarry Smith u_values = b->a + diag_offset[i] + 1 + shift; 51135aab85fSBarry Smith for (k=0; k<nz; k++) { 51235aab85fSBarry Smith sum -= u_values[k]; 51335aab85fSBarry Smith } 51435aab85fSBarry Smith ssum = PetscAbsScalar(sum/pv[diag]); 51535aab85fSBarry Smith if (ssum < 1000. && ssum > .001) pv[diag] = sum; 51635aab85fSBarry Smith } 51735aab85fSBarry Smith /* check pivot entry for current row */ 5188c37ef55SBarry Smith } 5190f11f9aeSSatish Balay 52035aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 52135aab85fSBarry Smith for (i=0; i<n; i++) { 52235aab85fSBarry Smith b->a[diag_offset[i]+shift] = 1.0/b->a[diag_offset[i]+shift]; 52335aab85fSBarry Smith } 52435aab85fSBarry Smith 525606d414cSSatish Balay if (preserve_row_sums) {ierr = PetscFree(rowsums);CHKERRQ(ierr);} 526606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 52778b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 52878b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 529416022c9SBarry Smith C->factor = FACTOR_LU; 53041c01911SSatish Balay ierr = Mat_AIJ_CheckInode(C);CHKERRQ(ierr); 531*7dda7e10SBarry Smith (*B)->ops->lufactornumeric = A->ops->lufactornumeric; /* Use Inode variant ONLY if A has inodes */ 532c456f294SBarry Smith C->assembled = PETSC_TRUE; 533416022c9SBarry Smith PLogFlops(b->n); 5343a40ed3dSBarry Smith PetscFunctionReturn(0); 535289bc588SBarry Smith } 5360a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 5375615d1e5SSatish Balay #undef __FUNC__ 5385615d1e5SSatish Balay #define __FUNC__ "MatLUFactor_SeqAIJ" 539329f5518SBarry Smith int MatLUFactor_SeqAIJ(Mat A,IS row,IS col,PetscReal f) 540da3a660dSBarry Smith { 541416022c9SBarry Smith Mat_SeqAIJ *mat = (Mat_SeqAIJ*)A->data; 54286bacbd2SBarry Smith int ierr,refct; 543416022c9SBarry Smith Mat C; 544f830108cSBarry Smith PetscOps *Abops; 5450a6ffc59SBarry Smith MatOps Aops; 546416022c9SBarry Smith 5473a40ed3dSBarry Smith PetscFunctionBegin; 548f2582111SSatish Balay ierr = MatLUFactorSymbolic(A,row,col,f,&C);CHKERRQ(ierr); 549f2582111SSatish Balay ierr = MatLUFactorNumeric(A,&C);CHKERRQ(ierr); 550da3a660dSBarry Smith 551da3a660dSBarry Smith /* free all the data structures from mat */ 552606d414cSSatish Balay ierr = PetscFree(mat->a);CHKERRQ(ierr); 553606d414cSSatish Balay if (!mat->singlemalloc) { 554606d414cSSatish Balay ierr = PetscFree(mat->i);CHKERRQ(ierr); 555606d414cSSatish Balay ierr = PetscFree(mat->j);CHKERRQ(ierr); 556606d414cSSatish Balay } 557606d414cSSatish Balay if (mat->diag) {ierr = PetscFree(mat->diag);CHKERRQ(ierr);} 558606d414cSSatish Balay if (mat->ilen) {ierr = PetscFree(mat->ilen);CHKERRQ(ierr);} 559606d414cSSatish Balay if (mat->imax) {ierr = PetscFree(mat->imax);CHKERRQ(ierr);} 560606d414cSSatish Balay if (mat->solve_work) {ierr = PetscFree(mat->solve_work);CHKERRQ(ierr);} 561606d414cSSatish Balay if (mat->inode.size) {ierr = PetscFree(mat->inode.size);CHKERRQ(ierr);} 5620f0aacb9SBarry Smith if (mat->icol) {ierr = ISDestroy(mat->icol);CHKERRQ(ierr);} 563606d414cSSatish Balay ierr = PetscFree(mat);CHKERRQ(ierr); 564da3a660dSBarry Smith 56517642b18SBarry Smith ierr = MapDestroy(A->rmap);CHKERRQ(ierr); 56617642b18SBarry Smith ierr = MapDestroy(A->cmap);CHKERRQ(ierr); 56717642b18SBarry Smith 568f830108cSBarry Smith /* 569f830108cSBarry Smith This is horrible, horrible code. We need to keep the 570f830108cSBarry Smith A pointers for the bops and ops but copy everything 571f830108cSBarry Smith else from C. 572f830108cSBarry Smith */ 573f830108cSBarry Smith Abops = A->bops; 574f830108cSBarry Smith Aops = A->ops; 57586bacbd2SBarry Smith refct = A->refct; 576549d3d68SSatish Balay ierr = PetscMemcpy(A,C,sizeof(struct _p_Mat));CHKERRQ(ierr); 577451c4af8SSatish Balay mat = (Mat_SeqAIJ*)A->data; 578451c4af8SSatish Balay PLogObjectParent(A,mat->icol); 579451c4af8SSatish Balay 580f830108cSBarry Smith A->bops = Abops; 581f830108cSBarry Smith A->ops = Aops; 582bef8e0ddSBarry Smith A->qlist = 0; 58386bacbd2SBarry Smith A->refct = refct; 584c38d4ed2SBarry Smith /* copy over the type_name and name */ 585c38d4ed2SBarry Smith ierr = PetscStrallocpy(C->type_name,&A->type_name);CHKERRQ(ierr); 586c38d4ed2SBarry Smith ierr = PetscStrallocpy(C->name,&A->name);CHKERRQ(ierr); 587f830108cSBarry Smith 5880452661fSBarry Smith PetscHeaderDestroy(C); 589c38d4ed2SBarry Smith 5903a40ed3dSBarry Smith PetscFunctionReturn(0); 591da3a660dSBarry Smith } 5920a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 5935615d1e5SSatish Balay #undef __FUNC__ 5945615d1e5SSatish Balay #define __FUNC__ "MatSolve_SeqAIJ" 595416022c9SBarry Smith int MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 5968c37ef55SBarry Smith { 597416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 598416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 599416022c9SBarry Smith int *r,*c,ierr,i, n = a->m,*vi,*ai = a->i,*aj = a->j; 6003b2fbd54SBarry Smith int nz,shift = a->indexshift,*rout,*cout; 601416022c9SBarry Smith Scalar *x,*b,*tmp,*tmps,*aa = a->a,sum,*v; 6028c37ef55SBarry Smith 6033a40ed3dSBarry Smith PetscFunctionBegin; 6043a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 60591bf9adeSBarry Smith 606e1311b90SBarry Smith ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 607e1311b90SBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 608416022c9SBarry Smith tmp = a->solve_work; 6098c37ef55SBarry Smith 6103b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 6113b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 6128c37ef55SBarry Smith 6138c37ef55SBarry Smith /* forward solve the lower triangular */ 6148c37ef55SBarry Smith tmp[0] = b[*r++]; 6150cf60383SBarry Smith tmps = tmp + shift; 6168c37ef55SBarry Smith for (i=1; i<n; i++) { 617dbb450caSBarry Smith v = aa + ai[i] + shift; 618dbb450caSBarry Smith vi = aj + ai[i] + shift; 619416022c9SBarry Smith nz = a->diag[i] - ai[i]; 6208c37ef55SBarry Smith sum = b[*r++]; 6210cf60383SBarry Smith while (nz--) sum -= *v++ * tmps[*vi++]; 6228c37ef55SBarry Smith tmp[i] = sum; 6238c37ef55SBarry Smith } 6248c37ef55SBarry Smith 6258c37ef55SBarry Smith /* backward solve the upper triangular */ 6268c37ef55SBarry Smith for (i=n-1; i>=0; i--){ 627416022c9SBarry Smith v = aa + a->diag[i] + (!shift); 628416022c9SBarry Smith vi = aj + a->diag[i] + (!shift); 629416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 6308c37ef55SBarry Smith sum = tmp[i]; 6310cf60383SBarry Smith while (nz--) sum -= *v++ * tmps[*vi++]; 632416022c9SBarry Smith x[*c--] = tmp[i] = sum*aa[a->diag[i]+shift]; 6338c37ef55SBarry Smith } 6348c37ef55SBarry Smith 6353b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 6363b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 637cb5b572fSBarry Smith ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 638cb5b572fSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 639416022c9SBarry Smith PLogFlops(2*a->nz - a->n); 6403a40ed3dSBarry Smith PetscFunctionReturn(0); 6418c37ef55SBarry Smith } 642026e39d0SSatish Balay 643930ae53cSBarry Smith /* ----------------------------------------------------------- */ 644930ae53cSBarry Smith #undef __FUNC__ 645930ae53cSBarry Smith #define __FUNC__ "MatSolve_SeqAIJ_NaturalOrdering" 646930ae53cSBarry Smith int MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 647930ae53cSBarry Smith { 648930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 649d85afc42SSatish Balay int n = a->m,*ai = a->i,*aj = a->j,*adiag = a->diag,ierr; 650d85afc42SSatish Balay Scalar *x,*b,*aa = a->a,sum; 651aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 652d85afc42SSatish Balay int adiag_i,i,*vi,nz,ai_i; 653d85afc42SSatish Balay Scalar *v; 654d85afc42SSatish Balay #endif 655930ae53cSBarry Smith 6563a40ed3dSBarry Smith PetscFunctionBegin; 6573a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 658930ae53cSBarry Smith if (a->indexshift) { 6593a40ed3dSBarry Smith ierr = MatSolve_SeqAIJ(A,bb,xx);CHKERRQ(ierr); 6603a40ed3dSBarry Smith PetscFunctionReturn(0); 661930ae53cSBarry Smith } 662930ae53cSBarry Smith 663e1311b90SBarry Smith ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 664e1311b90SBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 665930ae53cSBarry Smith 666aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 6671c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 6681c4feecaSSatish Balay #else 669930ae53cSBarry Smith /* forward solve the lower triangular */ 670930ae53cSBarry Smith x[0] = b[0]; 671930ae53cSBarry Smith for (i=1; i<n; i++) { 672930ae53cSBarry Smith ai_i = ai[i]; 673930ae53cSBarry Smith v = aa + ai_i; 674930ae53cSBarry Smith vi = aj + ai_i; 675930ae53cSBarry Smith nz = adiag[i] - ai_i; 676930ae53cSBarry Smith sum = b[i]; 677930ae53cSBarry Smith while (nz--) sum -= *v++ * x[*vi++]; 678930ae53cSBarry Smith x[i] = sum; 679930ae53cSBarry Smith } 680930ae53cSBarry Smith 681930ae53cSBarry Smith /* backward solve the upper triangular */ 682930ae53cSBarry Smith for (i=n-1; i>=0; i--){ 683930ae53cSBarry Smith adiag_i = adiag[i]; 684d708749eSBarry Smith v = aa + adiag_i + 1; 685d708749eSBarry Smith vi = aj + adiag_i + 1; 686930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 687930ae53cSBarry Smith sum = x[i]; 688930ae53cSBarry Smith while (nz--) sum -= *v++ * x[*vi++]; 689930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 690930ae53cSBarry Smith } 6911c4feecaSSatish Balay #endif 692930ae53cSBarry Smith PLogFlops(2*a->nz - a->n); 693cb5b572fSBarry Smith ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 694cb5b572fSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 6953a40ed3dSBarry Smith PetscFunctionReturn(0); 696930ae53cSBarry Smith } 697930ae53cSBarry Smith 6985615d1e5SSatish Balay #undef __FUNC__ 6995615d1e5SSatish Balay #define __FUNC__ "MatSolveAdd_SeqAIJ" 700416022c9SBarry Smith int MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 701da3a660dSBarry Smith { 702416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 703416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 704416022c9SBarry Smith int *r,*c,ierr,i, n = a->m,*vi,*ai = a->i,*aj = a->j; 7053b2fbd54SBarry Smith int nz,shift = a->indexshift,*rout,*cout; 706416022c9SBarry Smith Scalar *x,*b,*tmp,*aa = a->a,sum,*v; 707da3a660dSBarry Smith 7083a40ed3dSBarry Smith PetscFunctionBegin; 70978b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 710da3a660dSBarry Smith 711e1311b90SBarry Smith ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 712e1311b90SBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 713416022c9SBarry Smith tmp = a->solve_work; 714da3a660dSBarry Smith 7153b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 7163b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 717da3a660dSBarry Smith 718da3a660dSBarry Smith /* forward solve the lower triangular */ 719da3a660dSBarry Smith tmp[0] = b[*r++]; 720da3a660dSBarry Smith for (i=1; i<n; i++) { 721dbb450caSBarry Smith v = aa + ai[i] + shift; 722dbb450caSBarry Smith vi = aj + ai[i] + shift; 723416022c9SBarry Smith nz = a->diag[i] - ai[i]; 724da3a660dSBarry Smith sum = b[*r++]; 725dbb450caSBarry Smith while (nz--) sum -= *v++ * tmp[*vi++ + shift]; 726da3a660dSBarry Smith tmp[i] = sum; 727da3a660dSBarry Smith } 728da3a660dSBarry Smith 729da3a660dSBarry Smith /* backward solve the upper triangular */ 730da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 731416022c9SBarry Smith v = aa + a->diag[i] + (!shift); 732416022c9SBarry Smith vi = aj + a->diag[i] + (!shift); 733416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 734da3a660dSBarry Smith sum = tmp[i]; 735dbb450caSBarry Smith while (nz--) sum -= *v++ * tmp[*vi++ + shift]; 736416022c9SBarry Smith tmp[i] = sum*aa[a->diag[i]+shift]; 737da3a660dSBarry Smith x[*c--] += tmp[i]; 738da3a660dSBarry Smith } 739da3a660dSBarry Smith 7403b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 7413b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 742cb5b572fSBarry Smith ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 743cb5b572fSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 744416022c9SBarry Smith PLogFlops(2*a->nz); 745898183ecSLois Curfman McInnes 7463a40ed3dSBarry Smith PetscFunctionReturn(0); 747da3a660dSBarry Smith } 748da3a660dSBarry Smith /* -------------------------------------------------------------------*/ 7495615d1e5SSatish Balay #undef __FUNC__ 7507c922b88SBarry Smith #define __FUNC__ "MatSolveTranspose_SeqAIJ" 7517c922b88SBarry Smith int MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 752da3a660dSBarry Smith { 753416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 7542235e667SBarry Smith IS iscol = a->col,isrow = a->row; 755416022c9SBarry Smith int *r,*c,ierr,i,n = a->m,*vi,*ai = a->i,*aj = a->j; 756f1af5d2fSBarry Smith int nz,shift = a->indexshift,*rout,*cout,*diag = a->diag; 757f1af5d2fSBarry Smith Scalar *x,*b,*tmp,*aa = a->a,*v,s1; 758da3a660dSBarry Smith 7593a40ed3dSBarry Smith PetscFunctionBegin; 760e1311b90SBarry Smith ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 761e1311b90SBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 762416022c9SBarry Smith tmp = a->solve_work; 763da3a660dSBarry Smith 7642235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 7652235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 766da3a660dSBarry Smith 767da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 7682235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 769da3a660dSBarry Smith 770da3a660dSBarry Smith /* forward solve the U^T */ 771da3a660dSBarry Smith for (i=0; i<n; i++) { 772f1af5d2fSBarry Smith v = aa + diag[i] + shift; 773f1af5d2fSBarry Smith vi = aj + diag[i] + (!shift); 774f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 775c38d4ed2SBarry Smith s1 = tmp[i]; 776c38d4ed2SBarry Smith s1 *= *(v++); /* multiply by inverse of diagonal entry */ 777da3a660dSBarry Smith while (nz--) { 778f1af5d2fSBarry Smith tmp[*vi++ + shift] -= (*v++)*s1; 779da3a660dSBarry Smith } 780c38d4ed2SBarry Smith tmp[i] = s1; 781da3a660dSBarry Smith } 782da3a660dSBarry Smith 783da3a660dSBarry Smith /* backward solve the L^T */ 784da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 785f1af5d2fSBarry Smith v = aa + diag[i] - 1 + shift; 786f1af5d2fSBarry Smith vi = aj + diag[i] - 1 + shift; 787f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 788f1af5d2fSBarry Smith s1 = tmp[i]; 789da3a660dSBarry Smith while (nz--) { 790f1af5d2fSBarry Smith tmp[*vi-- + shift] -= (*v--)*s1; 791da3a660dSBarry Smith } 792da3a660dSBarry Smith } 793da3a660dSBarry Smith 794da3a660dSBarry Smith /* copy tmp into x according to permutation */ 795da3a660dSBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 796da3a660dSBarry Smith 7972235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 7982235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 799cb5b572fSBarry Smith ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 800cb5b572fSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 801da3a660dSBarry Smith 802416022c9SBarry Smith PLogFlops(2*a->nz-a->n); 8033a40ed3dSBarry Smith PetscFunctionReturn(0); 804da3a660dSBarry Smith } 805da3a660dSBarry Smith 8065615d1e5SSatish Balay #undef __FUNC__ 8077c922b88SBarry Smith #define __FUNC__ "MatSolveTransposeAdd_SeqAIJ" 8087c922b88SBarry Smith int MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 809da3a660dSBarry Smith { 810416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 8112235e667SBarry Smith IS iscol = a->col,isrow = a->row; 812416022c9SBarry Smith int *r,*c,ierr,i,n = a->m,*vi,*ai = a->i,*aj = a->j; 813f1af5d2fSBarry Smith int nz,shift = a->indexshift,*rout,*cout,*diag = a->diag; 814416022c9SBarry Smith Scalar *x,*b,*tmp,*aa = a->a,*v; 8156abc6512SBarry Smith 8163a40ed3dSBarry Smith PetscFunctionBegin; 8176abc6512SBarry Smith if (zz != xx) VecCopy(zz,xx); 8186abc6512SBarry Smith 819e1311b90SBarry Smith ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 820e1311b90SBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 821416022c9SBarry Smith tmp = a->solve_work; 8226abc6512SBarry Smith 8232235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 8242235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 8256abc6512SBarry Smith 8266abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 8272235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 8286abc6512SBarry Smith 8296abc6512SBarry Smith /* forward solve the U^T */ 8306abc6512SBarry Smith for (i=0; i<n; i++) { 831f1af5d2fSBarry Smith v = aa + diag[i] + shift; 832f1af5d2fSBarry Smith vi = aj + diag[i] + (!shift); 833f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 8346abc6512SBarry Smith tmp[i] *= *v++; 8356abc6512SBarry Smith while (nz--) { 836dbb450caSBarry Smith tmp[*vi++ + shift] -= (*v++)*tmp[i]; 8376abc6512SBarry Smith } 8386abc6512SBarry Smith } 8396abc6512SBarry Smith 8406abc6512SBarry Smith /* backward solve the L^T */ 8416abc6512SBarry Smith for (i=n-1; i>=0; i--){ 842f1af5d2fSBarry Smith v = aa + diag[i] - 1 + shift; 843f1af5d2fSBarry Smith vi = aj + diag[i] - 1 + shift; 844f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 8456abc6512SBarry Smith while (nz--) { 846dbb450caSBarry Smith tmp[*vi-- + shift] -= (*v--)*tmp[i]; 8476abc6512SBarry Smith } 8486abc6512SBarry Smith } 8496abc6512SBarry Smith 8506abc6512SBarry Smith /* copy tmp into x according to permutation */ 8516abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 8526abc6512SBarry Smith 8532235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 8542235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 855cb5b572fSBarry Smith ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 856cb5b572fSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 8576abc6512SBarry Smith 858416022c9SBarry Smith PLogFlops(2*a->nz); 8593a40ed3dSBarry Smith PetscFunctionReturn(0); 860da3a660dSBarry Smith } 8619e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 8627c922b88SBarry Smith extern int MatMissingDiagonal_SeqAIJ(Mat); 86308480c60SBarry Smith 8645615d1e5SSatish Balay #undef __FUNC__ 8655615d1e5SSatish Balay #define __FUNC__ "MatILUFactorSymbolic_SeqAIJ" 8666cf997b0SBarry Smith int MatILUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,MatILUInfo *info,Mat *fact) 8679e25ed09SBarry Smith { 868416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 8699e25ed09SBarry Smith IS isicol; 870416022c9SBarry Smith int *r,*ic,ierr,prow,n = a->m,*ai = a->i,*aj = a->j; 87156beaf04SBarry Smith int *ainew,*ajnew,jmax,*fill,*xi,nz,*im,*ajfill,*flev; 872335d9088SBarry Smith int *dloc,idx,row,m,fm,nzf,nzi,len, realloc = 0,dcount = 0; 8736cf997b0SBarry Smith int incrlev,nnz,i,shift = a->indexshift,levels,diagonal_fill; 87477c4ece6SBarry Smith PetscTruth col_identity,row_identity; 875329f5518SBarry Smith PetscReal f; 8769e25ed09SBarry Smith 8773a40ed3dSBarry Smith PetscFunctionBegin; 8786cf997b0SBarry Smith if (info) { 8796cf997b0SBarry Smith f = info->fill; 8800cd17293SBarry Smith levels = (int)info->levels; 8810cd17293SBarry Smith diagonal_fill = (int)info->diagonal_fill; 8826cf997b0SBarry Smith } else { 8836cf997b0SBarry Smith f = 1.0; 8846cf997b0SBarry Smith levels = 0; 8856cf997b0SBarry Smith diagonal_fill = 0; 8866cf997b0SBarry Smith } 88782bf6240SBarry Smith ierr = ISInvertPermutation(iscol,&isicol);CHKERRQ(ierr); 88882bf6240SBarry Smith 88908480c60SBarry Smith /* special case that simply copies fill pattern */ 890be0abb6dSBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 891be0abb6dSBarry Smith ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 89286bacbd2SBarry Smith if (!levels && row_identity && col_identity) { 8932e8a6d31SBarry Smith ierr = MatDuplicate_SeqAIJ(A,MAT_DO_NOT_COPY_VALUES,fact);CHKERRQ(ierr); 89408480c60SBarry Smith (*fact)->factor = FACTOR_LU; 89508480c60SBarry Smith b = (Mat_SeqAIJ*)(*fact)->data; 89608480c60SBarry Smith if (!b->diag) { 8977c922b88SBarry Smith ierr = MatMarkDiagonal_SeqAIJ(*fact);CHKERRQ(ierr); 89808480c60SBarry Smith } 8997c922b88SBarry Smith ierr = MatMissingDiagonal_SeqAIJ(*fact);CHKERRQ(ierr); 90008480c60SBarry Smith b->row = isrow; 90108480c60SBarry Smith b->col = iscol; 90282bf6240SBarry Smith b->icol = isicol; 9030452661fSBarry Smith b->solve_work = (Scalar*)PetscMalloc((b->m+1)*sizeof(Scalar));CHKPTRQ(b->solve_work); 904f830108cSBarry Smith (*fact)->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 905c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 906c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 9073a40ed3dSBarry Smith PetscFunctionReturn(0); 90808480c60SBarry Smith } 90908480c60SBarry Smith 910898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 911898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 9129e25ed09SBarry Smith 9139e25ed09SBarry Smith /* get new row pointers */ 9140452661fSBarry Smith ainew = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(ainew); 915dbb450caSBarry Smith ainew[0] = -shift; 9169e25ed09SBarry Smith /* don't know how many column pointers are needed so estimate */ 917dbb450caSBarry Smith jmax = (int)(f*(ai[n]+!shift)); 9180452661fSBarry Smith ajnew = (int*)PetscMalloc((jmax)*sizeof(int));CHKPTRQ(ajnew); 9199e25ed09SBarry Smith /* ajfill is level of fill for each fill entry */ 9200452661fSBarry Smith ajfill = (int*)PetscMalloc((jmax)*sizeof(int));CHKPTRQ(ajfill); 9219e25ed09SBarry Smith /* fill is a linked list of nonzeros in active row */ 9220452661fSBarry Smith fill = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(fill); 92356beaf04SBarry Smith /* im is level for each filled value */ 9240452661fSBarry Smith im = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(im); 92556beaf04SBarry Smith /* dloc is location of diagonal in factor */ 9260452661fSBarry Smith dloc = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(dloc); 92756beaf04SBarry Smith dloc[0] = 0; 92856beaf04SBarry Smith for (prow=0; prow<n; prow++) { 9296cf997b0SBarry Smith 9306cf997b0SBarry Smith /* copy current row into linked list */ 93156beaf04SBarry Smith nzf = nz = ai[r[prow]+1] - ai[r[prow]]; 932385f7a74SSatish Balay if (!nz) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,1,"Empty row in matrix"); 933dbb450caSBarry Smith xi = aj + ai[r[prow]] + shift; 9349e25ed09SBarry Smith fill[n] = n; 935435faa5fSBarry Smith fill[prow] = -1; /* marker to indicate if diagonal exists */ 9369e25ed09SBarry Smith while (nz--) { 9379e25ed09SBarry Smith fm = n; 938dbb450caSBarry Smith idx = ic[*xi++ + shift]; 9399e25ed09SBarry Smith do { 9409e25ed09SBarry Smith m = fm; 9419e25ed09SBarry Smith fm = fill[m]; 9429e25ed09SBarry Smith } while (fm < idx); 9439e25ed09SBarry Smith fill[m] = idx; 9449e25ed09SBarry Smith fill[idx] = fm; 94556beaf04SBarry Smith im[idx] = 0; 9469e25ed09SBarry Smith } 9476cf997b0SBarry Smith 9486cf997b0SBarry Smith /* make sure diagonal entry is included */ 949435faa5fSBarry Smith if (diagonal_fill && fill[prow] == -1) { 9506cf997b0SBarry Smith fm = n; 951435faa5fSBarry Smith while (fill[fm] < prow) fm = fill[fm]; 952435faa5fSBarry Smith fill[prow] = fill[fm]; /* insert diagonal into linked list */ 953435faa5fSBarry Smith fill[fm] = prow; 9546cf997b0SBarry Smith im[prow] = 0; 9556cf997b0SBarry Smith nzf++; 956335d9088SBarry Smith dcount++; 9576cf997b0SBarry Smith } 9586cf997b0SBarry Smith 95956beaf04SBarry Smith nzi = 0; 9609e25ed09SBarry Smith row = fill[n]; 96156beaf04SBarry Smith while (row < prow) { 96256beaf04SBarry Smith incrlev = im[row] + 1; 96356beaf04SBarry Smith nz = dloc[row]; 9646cf997b0SBarry Smith xi = ajnew + ainew[row] + shift + nz + 1; 965dbb450caSBarry Smith flev = ajfill + ainew[row] + shift + nz + 1; 966416022c9SBarry Smith nnz = ainew[row+1] - ainew[row] - nz - 1; 96756beaf04SBarry Smith fm = row; 96856beaf04SBarry Smith while (nnz-- > 0) { 969dbb450caSBarry Smith idx = *xi++ + shift; 97056beaf04SBarry Smith if (*flev + incrlev > levels) { 97156beaf04SBarry Smith flev++; 97256beaf04SBarry Smith continue; 97356beaf04SBarry Smith } 97456beaf04SBarry Smith do { 9759e25ed09SBarry Smith m = fm; 9769e25ed09SBarry Smith fm = fill[m]; 97756beaf04SBarry Smith } while (fm < idx); 9789e25ed09SBarry Smith if (fm != idx) { 97956beaf04SBarry Smith im[idx] = *flev + incrlev; 9809e25ed09SBarry Smith fill[m] = idx; 9819e25ed09SBarry Smith fill[idx] = fm; 9829e25ed09SBarry Smith fm = idx; 98356beaf04SBarry Smith nzf++; 984ecf371e4SBarry Smith } else { 98556beaf04SBarry Smith if (im[idx] > *flev + incrlev) im[idx] = *flev+incrlev; 9869e25ed09SBarry Smith } 98756beaf04SBarry Smith flev++; 9889e25ed09SBarry Smith } 9899e25ed09SBarry Smith row = fill[row]; 99056beaf04SBarry Smith nzi++; 9919e25ed09SBarry Smith } 9929e25ed09SBarry Smith /* copy new filled row into permanent storage */ 99356beaf04SBarry Smith ainew[prow+1] = ainew[prow] + nzf; 994d7e8b826SBarry Smith if (ainew[prow+1] > jmax-shift) { 995ecf371e4SBarry Smith 996ecf371e4SBarry Smith /* estimate how much additional space we will need */ 997ecf371e4SBarry Smith /* use the strategy suggested by David Hysom <hysom@perch-t.icase.edu> */ 998ecf371e4SBarry Smith /* just double the memory each time */ 999ecf371e4SBarry Smith /* maxadd = (int)((f*(ai[n]+!shift)*(n-prow+5))/n); */ 1000ecf371e4SBarry Smith int maxadd = jmax; 100156beaf04SBarry Smith if (maxadd < nzf) maxadd = (n-prow)*(nzf+1); 1002bbb6d6a8SBarry Smith jmax += maxadd; 1003ecf371e4SBarry Smith 1004ecf371e4SBarry Smith /* allocate a longer ajnew and ajfill */ 10050452661fSBarry Smith xi = (int*)PetscMalloc(jmax*sizeof(int));CHKPTRQ(xi); 1006549d3d68SSatish Balay ierr = PetscMemcpy(xi,ajnew,(ainew[prow]+shift)*sizeof(int));CHKERRQ(ierr); 1007606d414cSSatish Balay ierr = PetscFree(ajnew);CHKERRQ(ierr); 100856beaf04SBarry Smith ajnew = xi; 10090452661fSBarry Smith xi = (int*)PetscMalloc(jmax*sizeof(int));CHKPTRQ(xi); 1010549d3d68SSatish Balay ierr = PetscMemcpy(xi,ajfill,(ainew[prow]+shift)*sizeof(int));CHKERRQ(ierr); 1011606d414cSSatish Balay ierr = PetscFree(ajfill);CHKERRQ(ierr); 101256beaf04SBarry Smith ajfill = xi; 1013ecf371e4SBarry Smith realloc++; /* count how many times we realloc */ 10149e25ed09SBarry Smith } 1015dbb450caSBarry Smith xi = ajnew + ainew[prow] + shift; 1016dbb450caSBarry Smith flev = ajfill + ainew[prow] + shift; 101756beaf04SBarry Smith dloc[prow] = nzi; 10189e25ed09SBarry Smith fm = fill[n]; 101956beaf04SBarry Smith while (nzf--) { 1020dbb450caSBarry Smith *xi++ = fm - shift; 102156beaf04SBarry Smith *flev++ = im[fm]; 10229e25ed09SBarry Smith fm = fill[fm]; 10239e25ed09SBarry Smith } 10246cf997b0SBarry Smith /* make sure row has diagonal entry */ 10256cf997b0SBarry Smith if (ajnew[ainew[prow]+shift+dloc[prow]]+shift != prow) { 10266cf997b0SBarry Smith SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,1,"Row %d has missing diagonal in factored matrix\n\ 10276cf997b0SBarry Smith try running with -pc_ilu_nonzeros_along_diagonal or -pc_ilu_diagonal_fill",prow); 10286cf997b0SBarry Smith } 10299e25ed09SBarry Smith } 1030606d414cSSatish Balay ierr = PetscFree(ajfill);CHKERRQ(ierr); 1031898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1032898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 1033606d414cSSatish Balay ierr = PetscFree(fill);CHKERRQ(ierr); 1034606d414cSSatish Balay ierr = PetscFree(im);CHKERRQ(ierr); 10359e25ed09SBarry Smith 1036f64065bbSBarry Smith { 1037329f5518SBarry Smith PetscReal af = ((PetscReal)ainew[n])/((PetscReal)ai[n]); 1038c38d4ed2SBarry Smith PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Reallocs %d Fill ratio:given %g needed %g\n",realloc,f,af); 1039981c4779SBarry Smith PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Run with -pc_ilu_fill %g or use \n",af); 1040981c4779SBarry Smith PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:PCILUSetFill(pc,%g);\n",af); 1041981c4779SBarry Smith PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:for best performance.\n"); 1042335d9088SBarry Smith if (diagonal_fill) { 1043335d9088SBarry Smith PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Detected and replace %d missing diagonals",dcount); 1044335d9088SBarry Smith } 1045f64065bbSBarry Smith } 1046bbb6d6a8SBarry Smith 10479e25ed09SBarry Smith /* put together the new matrix */ 1048b4fd4287SBarry Smith ierr = MatCreateSeqAIJ(A->comm,n,n,0,PETSC_NULL,fact);CHKERRQ(ierr); 1049fa6007c9SSatish Balay PLogObjectParent(*fact,isicol); 1050416022c9SBarry Smith b = (Mat_SeqAIJ*)(*fact)->data; 1051606d414cSSatish Balay ierr = PetscFree(b->imax);CHKERRQ(ierr); 10527c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 1053dbb450caSBarry Smith len = (ainew[n] + shift)*sizeof(Scalar); 10549e25ed09SBarry Smith /* the next line frees the default space generated by the Create() */ 1055606d414cSSatish Balay ierr = PetscFree(b->a);CHKERRQ(ierr); 1056606d414cSSatish Balay ierr = PetscFree(b->ilen);CHKERRQ(ierr); 10576b96ef82SSatish Balay b->a = (Scalar*)PetscMalloc(len+1);CHKPTRQ(b->a); 1058416022c9SBarry Smith b->j = ajnew; 1059416022c9SBarry Smith b->i = ainew; 106056beaf04SBarry Smith for (i=0; i<n; i++) dloc[i] += ainew[i]; 1061416022c9SBarry Smith b->diag = dloc; 1062416022c9SBarry Smith b->ilen = 0; 1063416022c9SBarry Smith b->imax = 0; 1064416022c9SBarry Smith b->row = isrow; 1065416022c9SBarry Smith b->col = iscol; 1066c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1067c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 106882bf6240SBarry Smith b->icol = isicol; 106982bf6240SBarry Smith b->solve_work = (Scalar*)PetscMalloc((n+1)*sizeof(Scalar));CHKPTRQ(b->solve_work); 1070416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 107156beaf04SBarry Smith Allocate dloc, solve_work, new a, new j */ 1072dbb450caSBarry Smith PLogObjectMemory(*fact,(ainew[n]+shift-n) * (sizeof(int)+sizeof(Scalar))); 1073416022c9SBarry Smith b->maxnz = b->nz = ainew[n] + shift; 10749e25ed09SBarry Smith (*fact)->factor = FACTOR_LU; 1075ae068f58SLois Curfman McInnes 1076ae068f58SLois Curfman McInnes (*fact)->info.factor_mallocs = realloc; 1077ae068f58SLois Curfman McInnes (*fact)->info.fill_ratio_given = f; 1078329f5518SBarry Smith (*fact)->info.fill_ratio_needed = ((PetscReal)ainew[n])/((PetscReal)ai[prow]); 1079329f5518SBarry Smith (*fact)->factor = FACTOR_LU; 1080ae068f58SLois Curfman McInnes 10813a40ed3dSBarry Smith PetscFunctionReturn(0); 10829e25ed09SBarry Smith } 1083d5d45c9bSBarry Smith 1084d5d45c9bSBarry Smith 1085d5d45c9bSBarry Smith 1086d5d45c9bSBarry Smith 1087