1be1d678aSKris Buschelman #define PETSCMAT_DLL 2289bc588SBarry Smith 370f55243SBarry Smith #include "src/mat/impls/aij/seq/aij.h" 41c4feecaSSatish Balay #include "src/inline/dot.h" 5ed480e8bSBarry Smith #include "src/inline/spops.h" 61393bc97SHong Zhang #include "petscbt.h" 71393bc97SHong Zhang #include "src/mat/utils/freespace.h" 83b2fbd54SBarry Smith 94a2ae208SSatish Balay #undef __FUNCT__ 104a2ae208SSatish Balay #define __FUNCT__ "MatOrdering_Flow_SeqAIJ" 11dfbe8321SBarry Smith PetscErrorCode MatOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol) 123b2fbd54SBarry Smith { 133a40ed3dSBarry Smith PetscFunctionBegin; 143a40ed3dSBarry Smith 1529bbc08cSBarry Smith SETERRQ(PETSC_ERR_SUP,"Code not written"); 16aa482453SBarry Smith #if !defined(PETSC_USE_DEBUG) 17d64ed03dSBarry Smith PetscFunctionReturn(0); 18d64ed03dSBarry Smith #endif 193b2fbd54SBarry Smith } 203b2fbd54SBarry Smith 2186bacbd2SBarry Smith 22e34fafa9SBarry Smith #if !defined(PETSC_AVOID_GNUCOPYRIGHT_CODE) 23a77337e4SBarry Smith EXTERN PetscErrorCode SPARSEKIT2dperm(PetscInt*,MatScalar*,PetscInt*,PetscInt*,MatScalar*,PetscInt*,PetscInt*,PetscInt*,PetscInt*,PetscInt*); 24a77337e4SBarry Smith EXTERN PetscErrorCode SPARSEKIT2ilutp(PetscInt*,MatScalar*,PetscInt*,PetscInt*,PetscInt*,PetscReal,PetscReal*,PetscInt*,MatScalar*,PetscInt*,PetscInt*,PetscInt*,MatScalar*,PetscInt*,PetscInt*,PetscErrorCode*); 25a77337e4SBarry Smith EXTERN PetscErrorCode SPARSEKIT2msrcsr(PetscInt*,MatScalar*,PetscInt*,MatScalar*,PetscInt*,PetscInt*,MatScalar*,PetscInt*); 26e34fafa9SBarry Smith #endif 2786bacbd2SBarry Smith 284a2ae208SSatish Balay #undef __FUNCT__ 294a2ae208SSatish Balay #define __FUNCT__ "MatILUDTFactor_SeqAIJ" 3086bacbd2SBarry Smith /* ------------------------------------------------------------ 3186bacbd2SBarry Smith 3286bacbd2SBarry Smith This interface was contribed by Tony Caola 3386bacbd2SBarry Smith 3486bacbd2SBarry Smith This routine is an interface to the pivoting drop-tolerance 355bd2ddc7SBarry Smith ILU routine written by Yousef Saad (saad@cs.umn.edu) as part of 365bd2ddc7SBarry Smith SPARSEKIT2. 375bd2ddc7SBarry Smith 385bd2ddc7SBarry Smith The SPARSEKIT2 routines used here are covered by the GNU 395bd2ddc7SBarry Smith copyright; see the file gnu in this directory. 4086bacbd2SBarry Smith 4186bacbd2SBarry Smith Thanks to Prof. Saad, Dr. Hysom, and Dr. Smith for their 4286bacbd2SBarry Smith help in getting this routine ironed out. 4386bacbd2SBarry Smith 445bd2ddc7SBarry Smith The major drawback to this routine is that if info->fill is 455bd2ddc7SBarry Smith not large enough it fails rather than allocating more space; 465bd2ddc7SBarry Smith this can be fixed by hacking/improving the f2c version of 475bd2ddc7SBarry Smith Yousef Saad's code. 4886bacbd2SBarry Smith 4986bacbd2SBarry Smith ------------------------------------------------------------ 5086bacbd2SBarry Smith */ 517529efd4SKris Buschelman PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,MatFactorInfo *info,Mat *fact) 5286bacbd2SBarry Smith { 5360ec86bdSBarry Smith #if defined(PETSC_AVOID_GNUCOPYRIGHT_CODE) 5460ec86bdSBarry Smith PetscFunctionBegin; 55e005ede5SBarry Smith SETERRQ(PETSC_ERR_SUP_SYS,"This distribution does not include GNU Copyright code\n\ 5660ec86bdSBarry Smith You can obtain the drop tolerance routines by installing PETSc from\n\ 5760ec86bdSBarry Smith www.mcs.anl.gov/petsc\n"); 5860ec86bdSBarry Smith #else 5986bacbd2SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 6007d2056aSBarry Smith IS iscolf,isicol,isirow; 6186bacbd2SBarry Smith PetscTruth reorder; 629cc1f4e3SBarry Smith PetscErrorCode ierr,sierr; 63d0f46423SBarry Smith PetscInt *c,*r,*ic,i,n = A->rmap->n; 6438baddfdSBarry Smith PetscInt *old_i = a->i,*old_j = a->j,*new_i,*old_i2 = 0,*old_j2 = 0,*new_j; 6538baddfdSBarry Smith PetscInt *ordcol,*iwk,*iperm,*jw; 6638baddfdSBarry Smith PetscInt jmax,lfill,job,*o_i,*o_j; 6754f21887SBarry Smith MatScalar *old_a = a->a,*w,*new_a,*old_a2 = 0,*wk,*o_a; 6854a8161fSBarry Smith PetscReal af; 6986bacbd2SBarry Smith 7086bacbd2SBarry Smith PetscFunctionBegin; 7186bacbd2SBarry Smith 7286bacbd2SBarry Smith if (info->dt == PETSC_DEFAULT) info->dt = .005; 7338baddfdSBarry Smith if (info->dtcount == PETSC_DEFAULT) info->dtcount = (PetscInt)(1.5*a->rmax); 7486bacbd2SBarry Smith if (info->dtcol == PETSC_DEFAULT) info->dtcol = .01; 7533259db3SBarry Smith if (info->fill == PETSC_DEFAULT) info->fill = ((double)(n*(info->dtcount+1)))/a->nz; 7638baddfdSBarry Smith lfill = (PetscInt)(info->dtcount/2.0); 7738baddfdSBarry Smith jmax = (PetscInt)(info->fill*a->nz); 7886bacbd2SBarry Smith 7986bacbd2SBarry Smith 8086bacbd2SBarry Smith /* ------------------------------------------------------------ 8186bacbd2SBarry Smith If reorder=.TRUE., then the original matrix has to be 8286bacbd2SBarry Smith reordered to reflect the user selected ordering scheme, and 8386bacbd2SBarry Smith then de-reordered so it is in it's original format. 8486bacbd2SBarry Smith Because Saad's dperm() is NOT in place, we have to copy 8586bacbd2SBarry Smith the original matrix and allocate more storage. . . 8686bacbd2SBarry Smith ------------------------------------------------------------ 8786bacbd2SBarry Smith */ 8886bacbd2SBarry Smith 8986bacbd2SBarry Smith /* set reorder to true if either isrow or iscol is not identity */ 9086bacbd2SBarry Smith ierr = ISIdentity(isrow,&reorder);CHKERRQ(ierr); 9186bacbd2SBarry Smith if (reorder) {ierr = ISIdentity(iscol,&reorder);CHKERRQ(ierr);} 9286bacbd2SBarry Smith reorder = PetscNot(reorder); 9386bacbd2SBarry Smith 9486bacbd2SBarry Smith 9586bacbd2SBarry Smith /* storage for ilu factor */ 9638baddfdSBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscInt),&new_i);CHKERRQ(ierr); 9738baddfdSBarry Smith ierr = PetscMalloc(jmax*sizeof(PetscInt),&new_j);CHKERRQ(ierr); 98a77337e4SBarry Smith ierr = PetscMalloc(jmax*sizeof(MatScalar),&new_a);CHKERRQ(ierr); 9938baddfdSBarry Smith ierr = PetscMalloc(n*sizeof(PetscInt),&ordcol);CHKERRQ(ierr); 10086bacbd2SBarry Smith 10186bacbd2SBarry Smith /* ------------------------------------------------------------ 10286bacbd2SBarry Smith Make sure that everything is Fortran formatted (1-Based) 10386bacbd2SBarry Smith ------------------------------------------------------------ 10486bacbd2SBarry Smith */ 105b19713cbSBarry Smith for (i=old_i[0];i<old_i[n];i++) { 106b19713cbSBarry Smith old_j[i]++; 10786bacbd2SBarry Smith } 108b19713cbSBarry Smith for(i=0;i<n+1;i++) { 109b19713cbSBarry Smith old_i[i]++; 110b19713cbSBarry Smith }; 111010a20caSHong Zhang 11286bacbd2SBarry Smith 11386bacbd2SBarry Smith if (reorder) { 114c0c2c81eSBarry Smith ierr = ISGetIndices(iscol,&c);CHKERRQ(ierr); 115c0c2c81eSBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 116c0c2c81eSBarry Smith for(i=0;i<n;i++) { 117c0c2c81eSBarry Smith r[i] = r[i]+1; 118c0c2c81eSBarry Smith c[i] = c[i]+1; 119c0c2c81eSBarry Smith } 12038baddfdSBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscInt),&old_i2);CHKERRQ(ierr); 12138baddfdSBarry Smith ierr = PetscMalloc((old_i[n]-old_i[0]+1)*sizeof(PetscInt),&old_j2);CHKERRQ(ierr); 122a77337e4SBarry Smith ierr = PetscMalloc((old_i[n]-old_i[0]+1)*sizeof(MatScalar),&old_a2);CHKERRQ(ierr); 1235bd2ddc7SBarry Smith job = 3; SPARSEKIT2dperm(&n,old_a,old_j,old_i,old_a2,old_j2,old_i2,r,c,&job); 124c0c2c81eSBarry Smith for (i=0;i<n;i++) { 125c0c2c81eSBarry Smith r[i] = r[i]-1; 126c0c2c81eSBarry Smith c[i] = c[i]-1; 127c0c2c81eSBarry Smith } 128c0c2c81eSBarry Smith ierr = ISRestoreIndices(iscol,&c);CHKERRQ(ierr); 129c0c2c81eSBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 130b19713cbSBarry Smith o_a = old_a2; 131b19713cbSBarry Smith o_j = old_j2; 132b19713cbSBarry Smith o_i = old_i2; 133b19713cbSBarry Smith } else { 134b19713cbSBarry Smith o_a = old_a; 135b19713cbSBarry Smith o_j = old_j; 136b19713cbSBarry Smith o_i = old_i; 13786bacbd2SBarry Smith } 13886bacbd2SBarry Smith 13986bacbd2SBarry Smith /* ------------------------------------------------------------ 14086bacbd2SBarry Smith Call Saad's ilutp() routine to generate the factorization 14186bacbd2SBarry Smith ------------------------------------------------------------ 14286bacbd2SBarry Smith */ 14386bacbd2SBarry Smith 14438baddfdSBarry Smith ierr = PetscMalloc(2*n*sizeof(PetscInt),&iperm);CHKERRQ(ierr); 14538baddfdSBarry Smith ierr = PetscMalloc(2*n*sizeof(PetscInt),&jw);CHKERRQ(ierr); 14687828ca2SBarry Smith ierr = PetscMalloc(n*sizeof(PetscScalar),&w);CHKERRQ(ierr); 14786bacbd2SBarry Smith 14854a8161fSBarry Smith SPARSEKIT2ilutp(&n,o_a,o_j,o_i,&lfill,(PetscReal)info->dt,&info->dtcol,&n,new_a,new_j,new_i,&jmax,w,jw,iperm,&sierr); 1496849ba73SBarry Smith if (sierr) { 1506849ba73SBarry Smith switch (sierr) { 151a83599f4SBarry Smith case -3: SETERRQ2(PETSC_ERR_LIB,"ilutp(), matrix U overflows, need larger info->fill current fill %G space allocated %D",info->fill,jmax); 152a83599f4SBarry Smith case -2: SETERRQ2(PETSC_ERR_LIB,"ilutp(), matrix L overflows, need larger info->fill current fill %G space allocated %D",info->fill,jmax); 153e005ede5SBarry Smith case -5: SETERRQ(PETSC_ERR_LIB,"ilutp(), zero row encountered"); 154e005ede5SBarry Smith case -1: SETERRQ(PETSC_ERR_LIB,"ilutp(), input matrix may be wrong"); 15577431f27SBarry Smith case -4: SETERRQ1(PETSC_ERR_LIB,"ilutp(), illegal info->fill value %D",jmax); 15677431f27SBarry Smith default: SETERRQ1(PETSC_ERR_LIB,"ilutp(), zero pivot detected on row %D",sierr); 15786bacbd2SBarry Smith } 15886bacbd2SBarry Smith } 15986bacbd2SBarry Smith 16086bacbd2SBarry Smith ierr = PetscFree(w);CHKERRQ(ierr); 16186bacbd2SBarry Smith ierr = PetscFree(jw);CHKERRQ(ierr); 16286bacbd2SBarry Smith 16386bacbd2SBarry Smith /* ------------------------------------------------------------ 16486bacbd2SBarry Smith Saad's routine gives the result in Modified Sparse Row (msr) 16586bacbd2SBarry Smith Convert to Compressed Sparse Row format (csr) 16686bacbd2SBarry Smith ------------------------------------------------------------ 16786bacbd2SBarry Smith */ 16886bacbd2SBarry Smith 16987828ca2SBarry Smith ierr = PetscMalloc(n*sizeof(PetscScalar),&wk);CHKERRQ(ierr); 17038baddfdSBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscInt),&iwk);CHKERRQ(ierr); 17186bacbd2SBarry Smith 1725bd2ddc7SBarry Smith SPARSEKIT2msrcsr(&n,new_a,new_j,new_a,new_j,new_i,wk,iwk); 17386bacbd2SBarry Smith 17486bacbd2SBarry Smith ierr = PetscFree(iwk);CHKERRQ(ierr); 17586bacbd2SBarry Smith ierr = PetscFree(wk);CHKERRQ(ierr); 17686bacbd2SBarry Smith 17786bacbd2SBarry Smith if (reorder) { 17886bacbd2SBarry Smith ierr = PetscFree(old_a2);CHKERRQ(ierr); 17986bacbd2SBarry Smith ierr = PetscFree(old_j2);CHKERRQ(ierr); 18086bacbd2SBarry Smith ierr = PetscFree(old_i2);CHKERRQ(ierr); 181313ae024SBarry Smith } else { 182b19713cbSBarry Smith /* fix permutation of old_j that the factorization introduced */ 183f170005cSBarry Smith for (i=old_i[0]; i<old_i[n]; i++) { 184b19713cbSBarry Smith old_j[i-1] = iperm[old_j[i-1]-1]; 185b19713cbSBarry Smith } 186b19713cbSBarry Smith } 187b19713cbSBarry Smith 188b801d0f9SBarry Smith /* get rid of the shift to indices starting at 1 */ 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 196b19713cbSBarry Smith /* Make the factored matrix 0-based */ 19786bacbd2SBarry Smith for (i=0; i<n+1; i++) { 198b19713cbSBarry Smith new_i[i]--; 199b19713cbSBarry Smith } 200b19713cbSBarry Smith for (i=new_i[0];i<new_i[n];i++) { 201b19713cbSBarry Smith new_j[i]--; 202b19713cbSBarry Smith } 20386bacbd2SBarry Smith 20486bacbd2SBarry Smith /*-- due to the pivoting, we need to reorder iscol to correctly --*/ 20586bacbd2SBarry Smith /*-- permute the right-hand-side and solution vectors --*/ 2064c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 2074c49b128SBarry Smith ierr = ISInvertPermutation(isrow,PETSC_DECIDE,&isirow);CHKERRQ(ierr); 208c0c2c81eSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 20986bacbd2SBarry Smith for(i=0; i<n; i++) { 21086bacbd2SBarry Smith ordcol[i] = ic[iperm[i]-1]; 21186bacbd2SBarry Smith }; 21286bacbd2SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 21307d2056aSBarry Smith ierr = ISDestroy(isicol);CHKERRQ(ierr); 21486bacbd2SBarry Smith 215c0c2c81eSBarry Smith ierr = PetscFree(iperm);CHKERRQ(ierr); 216c0c2c81eSBarry Smith 217329f5518SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,ordcol,&iscolf);CHKERRQ(ierr); 21807d2056aSBarry Smith ierr = PetscFree(ordcol);CHKERRQ(ierr); 21986bacbd2SBarry Smith 22086bacbd2SBarry Smith /*----- put together the new matrix -----*/ 22186bacbd2SBarry Smith 2227adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,fact);CHKERRQ(ierr); 223f69a0ea3SMatthew Knepley ierr = MatSetSizes(*fact,n,n,n,n);CHKERRQ(ierr); 2247adad957SLisandro Dalcin ierr = MatSetType(*fact,((PetscObject)A)->type_name);CHKERRQ(ierr); 225ab93d7beSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(*fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 2265c9eb25fSBarry Smith (*fact)->factor = MAT_FACTOR_LU; 22786bacbd2SBarry Smith (*fact)->assembled = PETSC_TRUE; 22886bacbd2SBarry Smith 22986bacbd2SBarry Smith b = (Mat_SeqAIJ*)(*fact)->data; 230e6b907acSBarry Smith b->free_a = PETSC_TRUE; 231e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 23207d2056aSBarry Smith b->singlemalloc = PETSC_FALSE; 23386bacbd2SBarry Smith b->a = new_a; 23486bacbd2SBarry Smith b->j = new_j; 23586bacbd2SBarry Smith b->i = new_i; 23686bacbd2SBarry Smith b->ilen = 0; 23786bacbd2SBarry Smith b->imax = 0; 2381f9e874aSBarry Smith /* I am not sure why these are the inverses of the row and column permutations; but the other way is NO GOOD */ 239313ae024SBarry Smith b->row = isirow; 24086bacbd2SBarry Smith b->col = iscolf; 24187828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 24286bacbd2SBarry Smith b->maxnz = b->nz = new_i[n]; 24386bacbd2SBarry Smith ierr = MatMarkDiagonal_SeqAIJ(*fact);CHKERRQ(ierr); 24486bacbd2SBarry Smith (*fact)->info.factor_mallocs = 0; 24586bacbd2SBarry Smith 246431e710aSBarry Smith af = ((double)b->nz)/((double)a->nz) + .001; 247ae15b995SBarry Smith ierr = PetscInfo2(A,"Fill ratio:given %G needed %G\n",info->fill,af);CHKERRQ(ierr); 248ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 249ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 250ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 251431e710aSBarry Smith 2527529efd4SKris Buschelman ierr = MatILUDTFactor_Inode(A,isrow,iscol,info,fact);CHKERRQ(ierr); 25371c2f376SKris Buschelman 25486bacbd2SBarry Smith PetscFunctionReturn(0); 25560ec86bdSBarry Smith #endif 25686bacbd2SBarry Smith } 25786bacbd2SBarry Smith 258e631078cSBarry Smith EXTERN_C_BEGIN 2594a2ae208SSatish Balay #undef __FUNCT__ 260*db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc" 261*db4efbfdSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscTruth *flg) 262*db4efbfdSBarry Smith { 263*db4efbfdSBarry Smith PetscFunctionBegin; 264*db4efbfdSBarry Smith *flg = PETSC_TRUE; 265*db4efbfdSBarry Smith PetscFunctionReturn(0); 266*db4efbfdSBarry Smith } 267*db4efbfdSBarry Smith EXTERN_C_END 268*db4efbfdSBarry Smith 269*db4efbfdSBarry Smith EXTERN_C_BEGIN 270*db4efbfdSBarry Smith #undef __FUNCT__ 271b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc" 272b24902e0SBarry Smith PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B) 273b24902e0SBarry Smith { 274d0f46423SBarry Smith PetscInt n = A->rmap->n; 275b24902e0SBarry Smith PetscErrorCode ierr; 276b24902e0SBarry Smith 277b24902e0SBarry Smith PetscFunctionBegin; 278b24902e0SBarry Smith ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr); 279b24902e0SBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 280b24902e0SBarry Smith if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU) { 281b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr); 282*db4efbfdSBarry Smith (*B)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqAIJ; 283*db4efbfdSBarry Smith (*B)->ops->ilufactorsymbolic= MatILUFactorSymbolic_SeqAIJ; 284b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 285b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 286b24902e0SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 2875c9eb25fSBarry Smith (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqAIJ; 2885c9eb25fSBarry Smith (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ; 289b24902e0SBarry Smith } else SETERRQ(PETSC_ERR_SUP,"Factor type not supported"); 290*db4efbfdSBarry Smith (*B)->factor = ftype; 291b24902e0SBarry Smith PetscFunctionReturn(0); 292b24902e0SBarry Smith } 293e631078cSBarry Smith EXTERN_C_END 294b24902e0SBarry Smith 295b24902e0SBarry Smith #undef __FUNCT__ 296b9617806SBarry Smith #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ" 297dfbe8321SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,MatFactorInfo *info,Mat *B) 298289bc588SBarry Smith { 299416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 300289bc588SBarry Smith IS isicol; 3016849ba73SBarry Smith PetscErrorCode ierr; 302d0f46423SBarry Smith PetscInt *r,*ic,i,n=A->rmap->n,*ai=a->i,*aj=a->j; 3031393bc97SHong Zhang PetscInt *bi,*bj,*ajtmp; 3041393bc97SHong Zhang PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 3059e046878SBarry Smith PetscReal f; 3064875ba38SHong Zhang PetscInt nlnk,*lnk,k,**bi_ptr; 307a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 3081393bc97SHong Zhang PetscBT lnkbt; 309289bc588SBarry Smith 3103a40ed3dSBarry Smith PetscFunctionBegin; 311d0f46423SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square"); 3124c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 313ac121b3dSBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 314ac121b3dSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 315289bc588SBarry Smith 316289bc588SBarry Smith /* get new row pointers */ 3171393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 3181393bc97SHong Zhang bi[0] = 0; 3191393bc97SHong Zhang 3201393bc97SHong Zhang /* bdiag is location of diagonal in factor */ 3211393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 3221393bc97SHong Zhang bdiag[0] = 0; 3231393bc97SHong Zhang 3241393bc97SHong Zhang /* linked list for storing column indices of the active row */ 3251393bc97SHong Zhang nlnk = n + 1; 3261393bc97SHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3271393bc97SHong Zhang 32835baf27eSSatish Balay ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 3291393bc97SHong Zhang 3301393bc97SHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 331d3d32019SBarry Smith f = info->fill; 332a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 3331393bc97SHong Zhang current_space = free_space; 334289bc588SBarry Smith 335289bc588SBarry Smith for (i=0; i<n; i++) { 3361393bc97SHong Zhang /* copy previous fill into linked list */ 337289bc588SBarry Smith nzi = 0; 3381393bc97SHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 3391393bc97SHong Zhang if (!nnz) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix"); 3401393bc97SHong Zhang ajtmp = aj + ai[r[i]]; 341afcc9904SHong Zhang ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3421393bc97SHong Zhang nzi += nlnk; 3431393bc97SHong Zhang 3441393bc97SHong Zhang /* add pivot rows into linked list */ 3451393bc97SHong Zhang row = lnk[n]; 3461393bc97SHong Zhang while (row < i) { 347d1170560SHong Zhang nzbd = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */ 348d1170560SHong Zhang ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 349d1170560SHong Zhang ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 3501393bc97SHong Zhang nzi += nlnk; 3511393bc97SHong Zhang row = lnk[row]; 352289bc588SBarry Smith } 3531393bc97SHong Zhang bi[i+1] = bi[i] + nzi; 3541393bc97SHong Zhang im[i] = nzi; 3551393bc97SHong Zhang 3561393bc97SHong Zhang /* mark bdiag */ 3571393bc97SHong Zhang nzbd = 0; 3581393bc97SHong Zhang nnz = nzi; 3591393bc97SHong Zhang k = lnk[n]; 3601393bc97SHong Zhang while (nnz-- && k < i){ 3611393bc97SHong Zhang nzbd++; 3621393bc97SHong Zhang k = lnk[k]; 3631393bc97SHong Zhang } 3641393bc97SHong Zhang bdiag[i] = bi[i] + nzbd; 3651393bc97SHong Zhang 3661393bc97SHong Zhang /* if free space is not available, make more free space */ 3671393bc97SHong Zhang if (current_space->local_remaining<nzi) { 3681393bc97SHong Zhang nnz = (n - i)*nzi; /* estimated and max additional space needed */ 369a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 3701393bc97SHong Zhang reallocs++; 3711393bc97SHong Zhang } 3721393bc97SHong Zhang 3731393bc97SHong Zhang /* copy data into free space, then initialize lnk */ 3741393bc97SHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 3751393bc97SHong Zhang bi_ptr[i] = current_space->array; 3761393bc97SHong Zhang current_space->array += nzi; 3771393bc97SHong Zhang current_space->local_used += nzi; 3781393bc97SHong Zhang current_space->local_remaining -= nzi; 379289bc588SBarry Smith } 3806cf91177SBarry Smith #if defined(PETSC_USE_INFO) 381464e8d44SSatish Balay if (ai[n] != 0) { 3821393bc97SHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 383ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 384ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 385ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 386ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 38705bf559cSSatish Balay } else { 388ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 38905bf559cSSatish Balay } 39063ba0a88SBarry Smith #endif 3912fb3ed76SBarry Smith 392898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 393898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3941987afe7SBarry Smith 3951393bc97SHong Zhang /* destroy list of free space and other temporary array(s) */ 3961393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 397a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); 3981393bc97SHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 39935baf27eSSatish Balay ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 400289bc588SBarry Smith 401289bc588SBarry Smith /* put together the new matrix */ 4025c9eb25fSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(*B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 40352e6d16bSBarry Smith ierr = PetscLogObjectParent(*B,isicol);CHKERRQ(ierr); 404416022c9SBarry Smith b = (Mat_SeqAIJ*)(*B)->data; 405e6b907acSBarry Smith b->free_a = PETSC_TRUE; 406e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 4077c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 4081393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 4091393bc97SHong Zhang b->j = bj; 4101393bc97SHong Zhang b->i = bi; 4111393bc97SHong Zhang b->diag = bdiag; 412416022c9SBarry Smith b->ilen = 0; 413416022c9SBarry Smith b->imax = 0; 414416022c9SBarry Smith b->row = isrow; 415416022c9SBarry Smith b->col = iscol; 416c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 417c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 41882bf6240SBarry Smith b->icol = isicol; 41987828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 4201393bc97SHong Zhang 4211393bc97SHong Zhang /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 4221393bc97SHong Zhang ierr = PetscLogObjectMemory(*B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 4231393bc97SHong Zhang b->maxnz = b->nz = bi[n] ; 4247fd98bd6SLois Curfman McInnes 4255c9eb25fSBarry Smith (*B)->factor = MAT_FACTOR_LU; 426418422e8SSatish Balay (*B)->info.factor_mallocs = reallocs; 427ae068f58SLois Curfman McInnes (*B)->info.fill_ratio_given = f; 428703deb49SSatish Balay 429e93ccc4dSBarry Smith if (ai[n] != 0) { 4301393bc97SHong Zhang (*B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 431eea2913aSSatish Balay } else { 432eea2913aSSatish Balay (*B)->info.fill_ratio_needed = 0.0; 433eea2913aSSatish Balay } 434*db4efbfdSBarry Smith (*B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 435*db4efbfdSBarry Smith (*B)->ops->solve = MatSolve_SeqAIJ; 436*db4efbfdSBarry Smith (*B)->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 437*db4efbfdSBarry Smith /* switch to inodes if appropriate */ 4384846f1f5SKris Buschelman ierr = MatLUFactorSymbolic_Inode(A,isrow,iscol,info,B);CHKERRQ(ierr); 4393a40ed3dSBarry Smith PetscFunctionReturn(0); 440289bc588SBarry Smith } 4411393bc97SHong Zhang 4425b5bc046SBarry Smith /* 4435b5bc046SBarry Smith Trouble in factorization, should we dump the original matrix? 4445b5bc046SBarry Smith */ 4455b5bc046SBarry Smith #undef __FUNCT__ 4465b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix" 4475b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A) 4485b5bc046SBarry Smith { 4495b5bc046SBarry Smith PetscErrorCode ierr; 4505b5bc046SBarry Smith PetscTruth flg; 4515b5bc046SBarry Smith 4525b5bc046SBarry Smith PetscFunctionBegin; 4535b5bc046SBarry Smith ierr = PetscOptionsHasName(PETSC_NULL,"-mat_factor_dump_on_error",&flg);CHKERRQ(ierr); 4545b5bc046SBarry Smith if (flg) { 4555b5bc046SBarry Smith PetscViewer viewer; 4565b5bc046SBarry Smith char filename[PETSC_MAX_PATH_LEN]; 4575b5bc046SBarry Smith 4585b5bc046SBarry Smith ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr); 4597adad957SLisandro Dalcin ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 4605b5bc046SBarry Smith ierr = MatView(A,viewer);CHKERRQ(ierr); 4615b5bc046SBarry Smith ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr); 4625b5bc046SBarry Smith } 4635b5bc046SBarry Smith PetscFunctionReturn(0); 4645b5bc046SBarry Smith } 4655b5bc046SBarry Smith 466*db4efbfdSBarry Smith extern PetscErrorCode MatSolve_Inode(Mat,Vec,Vec); 467*db4efbfdSBarry Smith 4680a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 4694a2ae208SSatish Balay #undef __FUNCT__ 4704a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ" 471af281ebdSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat A,MatFactorInfo *info,Mat *B) 472289bc588SBarry Smith { 473416022c9SBarry Smith Mat C=*B; 474416022c9SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 47582bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 4766849ba73SBarry Smith PetscErrorCode ierr; 477d0f46423SBarry Smith PetscInt *r,*ic,i,j,n=A->rmap->n,*bi=b->i,*bj=b->j; 478b3bf805bSHong Zhang PetscInt *ajtmp,*bjtmp,nz,row,*ics; 479d2276718SHong Zhang PetscInt *diag_offset = b->diag,diag,*pj; 48054f21887SBarry Smith PetscScalar *rtmp,*pc,multiplier,*rtmps; 48154f21887SBarry Smith MatScalar *v,*pv; 4826a7c8fc2SHong Zhang PetscScalar d; 4836a7c8fc2SHong Zhang PetscReal rs; 484b3bf805bSHong Zhang LUShift_Ctx sctx; 48542898933SHong Zhang PetscInt newshift,*ddiag; 486289bc588SBarry Smith 4873a40ed3dSBarry Smith PetscFunctionBegin; 48878b31e54SBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 48978b31e54SBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 49087828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr); 49187828ca2SBarry Smith ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 492010a20caSHong Zhang rtmps = rtmp; ics = ic; 493289bc588SBarry Smith 494ada7143aSSatish Balay sctx.shift_top = 0; 495ada7143aSSatish Balay sctx.nshift_max = 0; 496ada7143aSSatish Balay sctx.shift_lo = 0; 497ada7143aSSatish Balay sctx.shift_hi = 0; 498ada7143aSSatish Balay 499118f5789SBarry Smith /* if both shift schemes are chosen by user, only use info->shiftpd */ 500118f5789SBarry Smith if (info->shiftpd && info->shiftnz) info->shiftnz = 0.0; 5011a890ab1SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 5029deb78dcSHong Zhang PetscInt *aai = a->i; 50342898933SHong Zhang ddiag = a->diag; 504118f5789SBarry Smith sctx.shift_top = 0; 5056cc28720Svictorle for (i=0; i<n; i++) { 5069f95998fSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 5079f95998fSHong Zhang d = (a->a)[ddiag[i]]; 5081a890ab1SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 509010a20caSHong Zhang v = a->a+aai[i]; 510e24cfe64SHong Zhang nz = aai[i+1] - aai[i]; 511e24cfe64SHong Zhang for (j=0; j<nz; j++) 5121a890ab1SHong Zhang rs += PetscAbsScalar(v[j]); 5131a890ab1SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 5146cc28720Svictorle } 515c3af1dc1SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 516118f5789SBarry Smith sctx.shift_top *= 1.1; 517d2276718SHong Zhang sctx.nshift_max = 5; 518d2276718SHong Zhang sctx.shift_lo = 0.; 519d2276718SHong Zhang sctx.shift_hi = 1.; 520a0a356efSHong Zhang } 521a0a356efSHong Zhang 522a0a356efSHong Zhang sctx.shift_amount = 0; 523a0a356efSHong Zhang sctx.nshift = 0; 524085256b3SBarry Smith do { 525d2276718SHong Zhang sctx.lushift = PETSC_FALSE; 526289bc588SBarry Smith for (i=0; i<n; i++){ 527b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 528b3bf805bSHong Zhang bjtmp = bj + bi[i]; 529b3bf805bSHong Zhang for (j=0; j<nz; j++) rtmps[bjtmp[j]] = 0.0; 530289bc588SBarry Smith 531289bc588SBarry Smith /* load in initial (unfactored row) */ 532416022c9SBarry Smith nz = a->i[r[i]+1] - a->i[r[i]]; 533b3bf805bSHong Zhang ajtmp = a->j + a->i[r[i]]; 534010a20caSHong Zhang v = a->a + a->i[r[i]]; 535085256b3SBarry Smith for (j=0; j<nz; j++) { 536b3bf805bSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 537085256b3SBarry Smith } 538d2276718SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 539289bc588SBarry Smith 540b3bf805bSHong Zhang row = *bjtmp++; 541f2582111SSatish Balay while (row < i) { 5428c37ef55SBarry Smith pc = rtmp + row; 5438c37ef55SBarry Smith if (*pc != 0.0) { 544010a20caSHong Zhang pv = b->a + diag_offset[row]; 545010a20caSHong Zhang pj = b->j + diag_offset[row] + 1; 54635aab85fSBarry Smith multiplier = *pc / *pv++; 5478c37ef55SBarry Smith *pc = multiplier; 548b3bf805bSHong Zhang nz = bi[row+1] - diag_offset[row] - 1; 549f2582111SSatish Balay for (j=0; j<nz; j++) rtmps[pj[j]] -= multiplier * pv[j]; 550efee365bSSatish Balay ierr = PetscLogFlops(2*nz);CHKERRQ(ierr); 551289bc588SBarry Smith } 552b3bf805bSHong Zhang row = *bjtmp++; 553289bc588SBarry Smith } 554416022c9SBarry Smith /* finished row so stick it into b->a */ 555b3bf805bSHong Zhang pv = b->a + bi[i] ; 556b3bf805bSHong Zhang pj = b->j + bi[i] ; 557b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 558b3bf805bSHong Zhang diag = diag_offset[i] - bi[i]; 559d3d32019SBarry Smith rs = 0.0; 560d3d32019SBarry Smith for (j=0; j<nz; j++) { 561d3d32019SBarry Smith pv[j] = rtmps[pj[j]]; 562d3d32019SBarry Smith if (j != diag) rs += PetscAbsScalar(pv[j]); 563d3d32019SBarry Smith } 564d2276718SHong Zhang 565b3bf805bSHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 566d2276718SHong Zhang sctx.rs = rs; 567d2276718SHong Zhang sctx.pv = pv[diag]; 568c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 5695b5bc046SBarry Smith if (newshift == 1) break; 57035aab85fSBarry Smith } 571d2276718SHong Zhang 572118f5789SBarry Smith if (info->shiftpd && !sctx.lushift && info->shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 5736cc28720Svictorle /* 5746c037d1bSvictorle * if no shift in this attempt & shifting & started shifting & can refine, 5756cc28720Svictorle * then try lower shift 5766cc28720Svictorle */ 577d2276718SHong Zhang sctx.shift_hi = info->shift_fraction; 578d2276718SHong Zhang info->shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 579d2276718SHong Zhang sctx.shift_amount = info->shift_fraction * sctx.shift_top; 580d2276718SHong Zhang sctx.lushift = PETSC_TRUE; 581d2276718SHong Zhang sctx.nshift++; 5826cc28720Svictorle } 583d2276718SHong Zhang } while (sctx.lushift); 5840f11f9aeSSatish Balay 58535aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 58635aab85fSBarry Smith for (i=0; i<n; i++) { 587010a20caSHong Zhang b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]]; 58835aab85fSBarry Smith } 589606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 59078b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 59178b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 592*db4efbfdSBarry Smith if (b->inode.use) { 593*db4efbfdSBarry Smith C->ops->solve = MatSolve_Inode; 594*db4efbfdSBarry Smith } else { 595*db4efbfdSBarry Smith C->ops->solve = MatSolve_SeqAIJ; 596*db4efbfdSBarry Smith } 597*db4efbfdSBarry Smith C->ops->solveadd = MatSolveAdd_SeqAIJ; 598*db4efbfdSBarry Smith C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 599*db4efbfdSBarry Smith C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 600c456f294SBarry Smith C->assembled = PETSC_TRUE; 6015c9eb25fSBarry Smith C->preallocated = PETSC_TRUE; 602d0f46423SBarry Smith ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 603d2276718SHong Zhang if (sctx.nshift){ 604118f5789SBarry Smith if (info->shiftnz) { 6051e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 606118f5789SBarry Smith } else if (info->shiftpd) { 6071e2582c4SBarry Smith ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,info->shift_fraction,sctx.shift_top);CHKERRQ(ierr); 6086cc28720Svictorle } 6090697b6caSHong Zhang } 6103a40ed3dSBarry Smith PetscFunctionReturn(0); 611289bc588SBarry Smith } 6120889b5dcSvictorle 613137fb511SHong Zhang /* 614137fb511SHong Zhang This routine implements inplace ILU(0) with row or/and column permutations. 615137fb511SHong Zhang Input: 616137fb511SHong Zhang A - original matrix 617137fb511SHong Zhang Output; 618137fb511SHong Zhang A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i] 619137fb511SHong Zhang a->j (col index) is permuted by the inverse of colperm, then sorted 620137fb511SHong Zhang a->a reordered accordingly with a->j 621137fb511SHong Zhang a->diag (ptr to diagonal elements) is updated. 622137fb511SHong Zhang */ 623137fb511SHong Zhang #undef __FUNCT__ 624137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm" 625137fb511SHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat A,MatFactorInfo *info,Mat *B) 626137fb511SHong Zhang { 627b051339dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 628b051339dSHong Zhang IS isrow = a->row,isicol = a->icol; 629137fb511SHong Zhang PetscErrorCode ierr; 630d0f46423SBarry Smith PetscInt *r,*ic,i,j,n=A->rmap->n,*ai=a->i,*aj=a->j; 631b051339dSHong Zhang PetscInt *ajtmp,nz,row,*ics; 632b051339dSHong Zhang PetscInt *diag = a->diag,nbdiag,*pj; 633a77337e4SBarry Smith PetscScalar *rtmp,*pc,multiplier,d; 634a77337e4SBarry Smith MatScalar *v,*pv; 635137fb511SHong Zhang PetscReal rs; 636137fb511SHong Zhang LUShift_Ctx sctx; 637b051339dSHong Zhang PetscInt newshift; 638137fb511SHong Zhang 639137fb511SHong Zhang PetscFunctionBegin; 640b051339dSHong Zhang if (A != *B) SETERRQ(PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address"); 641137fb511SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 642137fb511SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 643137fb511SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr); 644137fb511SHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 645b051339dSHong Zhang ics = ic; 646137fb511SHong Zhang 647137fb511SHong Zhang sctx.shift_top = 0; 648137fb511SHong Zhang sctx.nshift_max = 0; 649137fb511SHong Zhang sctx.shift_lo = 0; 650137fb511SHong Zhang sctx.shift_hi = 0; 651137fb511SHong Zhang 652137fb511SHong Zhang /* if both shift schemes are chosen by user, only use info->shiftpd */ 653137fb511SHong Zhang if (info->shiftpd && info->shiftnz) info->shiftnz = 0.0; 654137fb511SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 655137fb511SHong Zhang sctx.shift_top = 0; 656137fb511SHong Zhang for (i=0; i<n; i++) { 657137fb511SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 658b051339dSHong Zhang d = (a->a)[diag[i]]; 659137fb511SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 660b051339dSHong Zhang v = a->a+ai[i]; 661b051339dSHong Zhang nz = ai[i+1] - ai[i]; 662137fb511SHong Zhang for (j=0; j<nz; j++) 663137fb511SHong Zhang rs += PetscAbsScalar(v[j]); 664137fb511SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 665137fb511SHong Zhang } 666137fb511SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 667137fb511SHong Zhang sctx.shift_top *= 1.1; 668137fb511SHong Zhang sctx.nshift_max = 5; 669137fb511SHong Zhang sctx.shift_lo = 0.; 670137fb511SHong Zhang sctx.shift_hi = 1.; 671137fb511SHong Zhang } 672137fb511SHong Zhang 673137fb511SHong Zhang sctx.shift_amount = 0; 674137fb511SHong Zhang sctx.nshift = 0; 675137fb511SHong Zhang do { 676137fb511SHong Zhang sctx.lushift = PETSC_FALSE; 677137fb511SHong Zhang for (i=0; i<n; i++){ 678b051339dSHong Zhang /* load in initial unfactored row */ 679b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 680b051339dSHong Zhang ajtmp = aj + ai[r[i]]; 681b051339dSHong Zhang v = a->a + ai[r[i]]; 682137fb511SHong Zhang /* sort permuted ajtmp and values v accordingly */ 683b051339dSHong Zhang for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]]; 684137fb511SHong Zhang ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr); 685137fb511SHong Zhang 686b051339dSHong Zhang diag[r[i]] = ai[r[i]]; 687137fb511SHong Zhang for (j=0; j<nz; j++) { 688137fb511SHong Zhang rtmp[ajtmp[j]] = v[j]; 689b051339dSHong Zhang if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */ 690137fb511SHong Zhang } 691137fb511SHong Zhang rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 692137fb511SHong Zhang 693b051339dSHong Zhang row = *ajtmp++; 694137fb511SHong Zhang while (row < i) { 695137fb511SHong Zhang pc = rtmp + row; 696137fb511SHong Zhang if (*pc != 0.0) { 697b051339dSHong Zhang pv = a->a + diag[r[row]]; 698b051339dSHong Zhang pj = aj + diag[r[row]] + 1; 699137fb511SHong Zhang 700137fb511SHong Zhang multiplier = *pc / *pv++; 701137fb511SHong Zhang *pc = multiplier; 702b051339dSHong Zhang nz = ai[r[row]+1] - diag[r[row]] - 1; 703b051339dSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 704137fb511SHong Zhang ierr = PetscLogFlops(2*nz);CHKERRQ(ierr); 705137fb511SHong Zhang } 706b051339dSHong Zhang row = *ajtmp++; 707137fb511SHong Zhang } 708b051339dSHong Zhang /* finished row so overwrite it onto a->a */ 709b051339dSHong Zhang pv = a->a + ai[r[i]] ; 710b051339dSHong Zhang pj = aj + ai[r[i]] ; 711b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 712b051339dSHong Zhang nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */ 713137fb511SHong Zhang 714137fb511SHong Zhang rs = 0.0; 715137fb511SHong Zhang for (j=0; j<nz; j++) { 716b051339dSHong Zhang pv[j] = rtmp[pj[j]]; 717b051339dSHong Zhang if (j != nbdiag) rs += PetscAbsScalar(pv[j]); 718137fb511SHong Zhang } 719137fb511SHong Zhang 720137fb511SHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 721137fb511SHong Zhang sctx.rs = rs; 722b051339dSHong Zhang sctx.pv = pv[nbdiag]; 723c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 724137fb511SHong Zhang if (newshift == 1) break; 725137fb511SHong Zhang } 726137fb511SHong Zhang 727137fb511SHong Zhang if (info->shiftpd && !sctx.lushift && info->shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 728137fb511SHong Zhang /* 729137fb511SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 730137fb511SHong Zhang * then try lower shift 731137fb511SHong Zhang */ 732137fb511SHong Zhang sctx.shift_hi = info->shift_fraction; 733137fb511SHong Zhang info->shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 734137fb511SHong Zhang sctx.shift_amount = info->shift_fraction * sctx.shift_top; 735137fb511SHong Zhang sctx.lushift = PETSC_TRUE; 736137fb511SHong Zhang sctx.nshift++; 737137fb511SHong Zhang } 738137fb511SHong Zhang } while (sctx.lushift); 739137fb511SHong Zhang 740137fb511SHong Zhang /* invert diagonal entries for simplier triangular solves */ 741137fb511SHong Zhang for (i=0; i<n; i++) { 742b051339dSHong Zhang a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]]; 743137fb511SHong Zhang } 744137fb511SHong Zhang 745137fb511SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 746137fb511SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 747137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 748b051339dSHong Zhang A->ops->solve = MatSolve_SeqAIJ_InplaceWithPerm; 749*db4efbfdSBarry Smith A->ops->solveadd = MatSolveAdd_SeqAIJ; 750*db4efbfdSBarry Smith A->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 751*db4efbfdSBarry Smith A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 752b051339dSHong Zhang A->assembled = PETSC_TRUE; 7535c9eb25fSBarry Smith A->preallocated = PETSC_TRUE; 754d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr); 755137fb511SHong Zhang if (sctx.nshift){ 756137fb511SHong Zhang if (info->shiftnz) { 7571e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 758137fb511SHong Zhang } else if (info->shiftpd) { 7591e2582c4SBarry Smith ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,info->shift_fraction,sctx.shift_top);CHKERRQ(ierr); 760137fb511SHong Zhang } 761137fb511SHong Zhang } 762137fb511SHong Zhang PetscFunctionReturn(0); 763137fb511SHong Zhang } 764137fb511SHong Zhang 7650a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 7664a2ae208SSatish Balay #undef __FUNCT__ 7674a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ" 768dfbe8321SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,MatFactorInfo *info) 769da3a660dSBarry Smith { 770dfbe8321SBarry Smith PetscErrorCode ierr; 771416022c9SBarry Smith Mat C; 772416022c9SBarry Smith 7733a40ed3dSBarry Smith PetscFunctionBegin; 77443244d56SBarry Smith ierr = MatGetFactor(A,MAT_SOLVER_PETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr); 7759e046878SBarry Smith ierr = MatLUFactorSymbolic(A,row,col,info,&C);CHKERRQ(ierr); 776af281ebdSHong Zhang ierr = MatLUFactorNumeric(A,info,&C);CHKERRQ(ierr); 777*db4efbfdSBarry Smith A->ops->solve = C->ops->solve; 778*db4efbfdSBarry Smith A->ops->solvetranspose = C->ops->solvetranspose; 779273d9f13SBarry Smith ierr = MatHeaderCopy(A,C);CHKERRQ(ierr); 78052e6d16bSBarry Smith ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr); 7813a40ed3dSBarry Smith PetscFunctionReturn(0); 782da3a660dSBarry Smith } 7830a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 7844a2ae208SSatish Balay #undef __FUNCT__ 7854a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ" 786dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 7878c37ef55SBarry Smith { 788416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 789416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 7906849ba73SBarry Smith PetscErrorCode ierr; 791d0f46423SBarry Smith PetscInt *r,*c,i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 79238baddfdSBarry Smith PetscInt nz,*rout,*cout; 793dd6ea824SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 794d9fead3dSBarry Smith const PetscScalar *b; 795dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 7968c37ef55SBarry Smith 7973a40ed3dSBarry Smith PetscFunctionBegin; 7983a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 79991bf9adeSBarry Smith 800d9fead3dSBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 8011ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 802416022c9SBarry Smith tmp = a->solve_work; 8038c37ef55SBarry Smith 8043b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 8053b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 8068c37ef55SBarry Smith 8078c37ef55SBarry Smith /* forward solve the lower triangular */ 8088c37ef55SBarry Smith tmp[0] = b[*r++]; 809010a20caSHong Zhang tmps = tmp; 8108c37ef55SBarry Smith for (i=1; i<n; i++) { 811010a20caSHong Zhang v = aa + ai[i] ; 812010a20caSHong Zhang vi = aj + ai[i] ; 81353e7425aSBarry Smith nz = a->diag[i] - ai[i]; 81453e7425aSBarry Smith sum = b[*r++]; 815ed480e8bSBarry Smith SPARSEDENSEMDOT(sum,tmps,v,vi,nz); 8168c37ef55SBarry Smith tmp[i] = sum; 8178c37ef55SBarry Smith } 8188c37ef55SBarry Smith 8198c37ef55SBarry Smith /* backward solve the upper triangular */ 8208c37ef55SBarry Smith for (i=n-1; i>=0; i--){ 821010a20caSHong Zhang v = aa + a->diag[i] + 1; 822010a20caSHong Zhang vi = aj + a->diag[i] + 1; 823416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 8248c37ef55SBarry Smith sum = tmp[i]; 825ed480e8bSBarry Smith SPARSEDENSEMDOT(sum,tmps,v,vi,nz); 826010a20caSHong Zhang x[*c--] = tmp[i] = sum*aa[a->diag[i]]; 8278c37ef55SBarry Smith } 8288c37ef55SBarry Smith 8293b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 8303b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 831d9fead3dSBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 8321ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 833d0f46423SBarry Smith ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 8343a40ed3dSBarry Smith PetscFunctionReturn(0); 8358c37ef55SBarry Smith } 836026e39d0SSatish Balay 8372bebee5dSHong Zhang #undef __FUNCT__ 8382bebee5dSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ" 8392bebee5dSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X) 8402bebee5dSHong Zhang { 8412bebee5dSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 8422bebee5dSHong Zhang IS iscol = a->col,isrow = a->row; 8432bebee5dSHong Zhang PetscErrorCode ierr; 844d0f46423SBarry Smith PetscInt *r,*c,i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 8452bebee5dSHong Zhang PetscInt nz,*rout,*cout,neq; 846dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 847dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 848*db4efbfdSBarry Smith PetscTruth bisdense,xisdense; 8492bebee5dSHong Zhang 8502bebee5dSHong Zhang PetscFunctionBegin; 8512bebee5dSHong Zhang if (!n) PetscFunctionReturn(0); 8522bebee5dSHong Zhang 853*db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 854*db4efbfdSBarry Smith if (!bisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 855*db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 856*db4efbfdSBarry Smith if (!xisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 857*db4efbfdSBarry Smith 8582bebee5dSHong Zhang ierr = MatGetArray(B,&b);CHKERRQ(ierr); 8592bebee5dSHong Zhang ierr = MatGetArray(X,&x);CHKERRQ(ierr); 8602bebee5dSHong Zhang 8612bebee5dSHong Zhang tmp = a->solve_work; 8622bebee5dSHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 8632bebee5dSHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 8642bebee5dSHong Zhang 8652bebee5dSHong Zhang for (neq=0; neq<n; neq++){ 8662bebee5dSHong Zhang /* forward solve the lower triangular */ 8672bebee5dSHong Zhang tmp[0] = b[r[0]]; 8682bebee5dSHong Zhang tmps = tmp; 8692bebee5dSHong Zhang for (i=1; i<n; i++) { 8702bebee5dSHong Zhang v = aa + ai[i] ; 8712bebee5dSHong Zhang vi = aj + ai[i] ; 8722bebee5dSHong Zhang nz = a->diag[i] - ai[i]; 8732bebee5dSHong Zhang sum = b[r[i]]; 8742bebee5dSHong Zhang SPARSEDENSEMDOT(sum,tmps,v,vi,nz); 8752bebee5dSHong Zhang tmp[i] = sum; 8762bebee5dSHong Zhang } 8772bebee5dSHong Zhang /* backward solve the upper triangular */ 8782bebee5dSHong Zhang for (i=n-1; i>=0; i--){ 8792bebee5dSHong Zhang v = aa + a->diag[i] + 1; 8802bebee5dSHong Zhang vi = aj + a->diag[i] + 1; 8812bebee5dSHong Zhang nz = ai[i+1] - a->diag[i] - 1; 8822bebee5dSHong Zhang sum = tmp[i]; 8832bebee5dSHong Zhang SPARSEDENSEMDOT(sum,tmps,v,vi,nz); 8842bebee5dSHong Zhang x[c[i]] = tmp[i] = sum*aa[a->diag[i]]; 8852bebee5dSHong Zhang } 8862bebee5dSHong Zhang 8872bebee5dSHong Zhang b += n; 8882bebee5dSHong Zhang x += n; 8892bebee5dSHong Zhang } 8902bebee5dSHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 8912bebee5dSHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 8922bebee5dSHong Zhang ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 8932bebee5dSHong Zhang ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 8942bebee5dSHong Zhang ierr = PetscLogFlops(n*(2*a->nz - n));CHKERRQ(ierr); 8952bebee5dSHong Zhang PetscFunctionReturn(0); 8962bebee5dSHong Zhang } 8972bebee5dSHong Zhang 898137fb511SHong Zhang #undef __FUNCT__ 899137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm" 900137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx) 901137fb511SHong Zhang { 902137fb511SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 903137fb511SHong Zhang IS iscol = a->col,isrow = a->row; 904137fb511SHong Zhang PetscErrorCode ierr; 905d0f46423SBarry Smith PetscInt *r,*c,i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 906137fb511SHong Zhang PetscInt nz,*rout,*cout,row; 907dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 908dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 909137fb511SHong Zhang 910137fb511SHong Zhang PetscFunctionBegin; 911137fb511SHong Zhang if (!n) PetscFunctionReturn(0); 912137fb511SHong Zhang 913137fb511SHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 914137fb511SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 915137fb511SHong Zhang tmp = a->solve_work; 916137fb511SHong Zhang 917137fb511SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 918137fb511SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 919137fb511SHong Zhang 920137fb511SHong Zhang /* forward solve the lower triangular */ 921137fb511SHong Zhang tmp[0] = b[*r++]; 922137fb511SHong Zhang tmps = tmp; 923137fb511SHong Zhang for (row=1; row<n; row++) { 924137fb511SHong Zhang i = rout[row]; /* permuted row */ 925137fb511SHong Zhang v = aa + ai[i] ; 926137fb511SHong Zhang vi = aj + ai[i] ; 927137fb511SHong Zhang nz = a->diag[i] - ai[i]; 928137fb511SHong Zhang sum = b[*r++]; 929137fb511SHong Zhang SPARSEDENSEMDOT(sum,tmps,v,vi,nz); 930137fb511SHong Zhang tmp[row] = sum; 931137fb511SHong Zhang } 932137fb511SHong Zhang 933137fb511SHong Zhang /* backward solve the upper triangular */ 934137fb511SHong Zhang for (row=n-1; row>=0; row--){ 935137fb511SHong Zhang i = rout[row]; /* permuted row */ 936137fb511SHong Zhang v = aa + a->diag[i] + 1; 937137fb511SHong Zhang vi = aj + a->diag[i] + 1; 938137fb511SHong Zhang nz = ai[i+1] - a->diag[i] - 1; 939137fb511SHong Zhang sum = tmp[row]; 940137fb511SHong Zhang SPARSEDENSEMDOT(sum,tmps,v,vi,nz); 941137fb511SHong Zhang x[*c--] = tmp[row] = sum*aa[a->diag[i]]; 942137fb511SHong Zhang } 943137fb511SHong Zhang 944137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 945137fb511SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 946137fb511SHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 947137fb511SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 948d0f46423SBarry Smith ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 949137fb511SHong Zhang PetscFunctionReturn(0); 950137fb511SHong Zhang } 951137fb511SHong Zhang 952930ae53cSBarry Smith /* ----------------------------------------------------------- */ 9534a2ae208SSatish Balay #undef __FUNCT__ 9544a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering" 955dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 956930ae53cSBarry Smith { 957930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 9586849ba73SBarry Smith PetscErrorCode ierr; 959d0f46423SBarry Smith PetscInt n = A->rmap->n; 960356650c2SBarry Smith const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag,*vi; 961356650c2SBarry Smith PetscScalar *x; 962356650c2SBarry Smith const PetscScalar *b; 963dd6ea824SBarry Smith const MatScalar *aa = a->a; 964aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 965356650c2SBarry Smith PetscInt adiag_i,i,nz,ai_i; 966dd6ea824SBarry Smith const MatScalar *v; 967dd6ea824SBarry Smith PetscScalar sum; 968d85afc42SSatish Balay #endif 969930ae53cSBarry Smith 9703a40ed3dSBarry Smith PetscFunctionBegin; 9713a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 972930ae53cSBarry Smith 973356650c2SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 9741ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 975930ae53cSBarry Smith 976aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 9771c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 9781c4feecaSSatish Balay #else 979930ae53cSBarry Smith /* forward solve the lower triangular */ 980930ae53cSBarry Smith x[0] = b[0]; 981930ae53cSBarry Smith for (i=1; i<n; i++) { 982930ae53cSBarry Smith ai_i = ai[i]; 983930ae53cSBarry Smith v = aa + ai_i; 984930ae53cSBarry Smith vi = aj + ai_i; 985930ae53cSBarry Smith nz = adiag[i] - ai_i; 986930ae53cSBarry Smith sum = b[i]; 987930ae53cSBarry Smith while (nz--) sum -= *v++ * x[*vi++]; 988930ae53cSBarry Smith x[i] = sum; 989930ae53cSBarry Smith } 990930ae53cSBarry Smith 991930ae53cSBarry Smith /* backward solve the upper triangular */ 992930ae53cSBarry Smith for (i=n-1; i>=0; i--){ 993930ae53cSBarry Smith adiag_i = adiag[i]; 994d708749eSBarry Smith v = aa + adiag_i + 1; 995d708749eSBarry Smith vi = aj + adiag_i + 1; 996930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 997930ae53cSBarry Smith sum = x[i]; 998930ae53cSBarry Smith while (nz--) sum -= *v++ * x[*vi++]; 999930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 1000930ae53cSBarry Smith } 10011c4feecaSSatish Balay #endif 1002d0f46423SBarry Smith ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 1003356650c2SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 10041ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 10053a40ed3dSBarry Smith PetscFunctionReturn(0); 1006930ae53cSBarry Smith } 1007930ae53cSBarry Smith 10084a2ae208SSatish Balay #undef __FUNCT__ 10094a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqAIJ" 1010dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 1011da3a660dSBarry Smith { 1012416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1013416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 10146849ba73SBarry Smith PetscErrorCode ierr; 1015d0f46423SBarry Smith PetscInt *r,*c,i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 101638baddfdSBarry Smith PetscInt nz,*rout,*cout; 1017dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,sum; 1018dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1019da3a660dSBarry Smith 10203a40ed3dSBarry Smith PetscFunctionBegin; 102178b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 1022da3a660dSBarry Smith 10231ebc52fbSHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 10241ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1025416022c9SBarry Smith tmp = a->solve_work; 1026da3a660dSBarry Smith 10273b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10283b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1029da3a660dSBarry Smith 1030da3a660dSBarry Smith /* forward solve the lower triangular */ 1031da3a660dSBarry Smith tmp[0] = b[*r++]; 1032da3a660dSBarry Smith for (i=1; i<n; i++) { 1033010a20caSHong Zhang v = aa + ai[i] ; 1034010a20caSHong Zhang vi = aj + ai[i] ; 1035416022c9SBarry Smith nz = a->diag[i] - ai[i]; 1036da3a660dSBarry Smith sum = b[*r++]; 1037010a20caSHong Zhang while (nz--) sum -= *v++ * tmp[*vi++ ]; 1038da3a660dSBarry Smith tmp[i] = sum; 1039da3a660dSBarry Smith } 1040da3a660dSBarry Smith 1041da3a660dSBarry Smith /* backward solve the upper triangular */ 1042da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1043010a20caSHong Zhang v = aa + a->diag[i] + 1; 1044010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1045416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 1046da3a660dSBarry Smith sum = tmp[i]; 1047010a20caSHong Zhang while (nz--) sum -= *v++ * tmp[*vi++ ]; 1048010a20caSHong Zhang tmp[i] = sum*aa[a->diag[i]]; 1049da3a660dSBarry Smith x[*c--] += tmp[i]; 1050da3a660dSBarry Smith } 1051da3a660dSBarry Smith 10523b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10533b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10541ebc52fbSHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 10551ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1056efee365bSSatish Balay ierr = PetscLogFlops(2*a->nz);CHKERRQ(ierr); 1057898183ecSLois Curfman McInnes 10583a40ed3dSBarry Smith PetscFunctionReturn(0); 1059da3a660dSBarry Smith } 1060da3a660dSBarry Smith /* -------------------------------------------------------------------*/ 10614a2ae208SSatish Balay #undef __FUNCT__ 10624a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqAIJ" 1063dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 1064da3a660dSBarry Smith { 1065416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10662235e667SBarry Smith IS iscol = a->col,isrow = a->row; 10676849ba73SBarry Smith PetscErrorCode ierr; 1068d0f46423SBarry Smith PetscInt *r,*c,i,n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 106938baddfdSBarry Smith PetscInt nz,*rout,*cout,*diag = a->diag; 1070dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,s1; 1071dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1072da3a660dSBarry Smith 10733a40ed3dSBarry Smith PetscFunctionBegin; 10741ebc52fbSHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 10751ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1076416022c9SBarry Smith tmp = a->solve_work; 1077da3a660dSBarry Smith 10782235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10792235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1080da3a660dSBarry Smith 1081da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 10822235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1083da3a660dSBarry Smith 1084da3a660dSBarry Smith /* forward solve the U^T */ 1085da3a660dSBarry Smith for (i=0; i<n; i++) { 1086010a20caSHong Zhang v = aa + diag[i] ; 1087010a20caSHong Zhang vi = aj + diag[i] + 1; 1088f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 1089c38d4ed2SBarry Smith s1 = tmp[i]; 1090ef66eb69SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 1091da3a660dSBarry Smith while (nz--) { 1092010a20caSHong Zhang tmp[*vi++ ] -= (*v++)*s1; 1093da3a660dSBarry Smith } 1094c38d4ed2SBarry Smith tmp[i] = s1; 1095da3a660dSBarry Smith } 1096da3a660dSBarry Smith 1097da3a660dSBarry Smith /* backward solve the L^T */ 1098da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1099010a20caSHong Zhang v = aa + diag[i] - 1 ; 1100010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1101f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 1102f1af5d2fSBarry Smith s1 = tmp[i]; 1103da3a660dSBarry Smith while (nz--) { 1104010a20caSHong Zhang tmp[*vi-- ] -= (*v--)*s1; 1105da3a660dSBarry Smith } 1106da3a660dSBarry Smith } 1107da3a660dSBarry Smith 1108da3a660dSBarry Smith /* copy tmp into x according to permutation */ 1109da3a660dSBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1110da3a660dSBarry Smith 11112235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 11122235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 11131ebc52fbSHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 11141ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1115da3a660dSBarry Smith 1116d0f46423SBarry Smith ierr = PetscLogFlops(2*a->nz-A->cmap->n);CHKERRQ(ierr); 11173a40ed3dSBarry Smith PetscFunctionReturn(0); 1118da3a660dSBarry Smith } 1119da3a660dSBarry Smith 11204a2ae208SSatish Balay #undef __FUNCT__ 11214a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ" 1122dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 1123da3a660dSBarry Smith { 1124416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 11252235e667SBarry Smith IS iscol = a->col,isrow = a->row; 11266849ba73SBarry Smith PetscErrorCode ierr; 1127d0f46423SBarry Smith PetscInt *r,*c,i,n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 112838baddfdSBarry Smith PetscInt nz,*rout,*cout,*diag = a->diag; 1129dd6ea824SBarry Smith PetscScalar *x,*b,*tmp; 1130dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 11316abc6512SBarry Smith 11323a40ed3dSBarry Smith PetscFunctionBegin; 113323bc6035SMatthew Knepley if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 11346abc6512SBarry Smith 11351ebc52fbSHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 11361ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1137416022c9SBarry Smith tmp = a->solve_work; 11386abc6512SBarry Smith 11392235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 11402235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 11416abc6512SBarry Smith 11426abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 11432235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 11446abc6512SBarry Smith 11456abc6512SBarry Smith /* forward solve the U^T */ 11466abc6512SBarry Smith for (i=0; i<n; i++) { 1147010a20caSHong Zhang v = aa + diag[i] ; 1148010a20caSHong Zhang vi = aj + diag[i] + 1; 1149f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 11506abc6512SBarry Smith tmp[i] *= *v++; 11516abc6512SBarry Smith while (nz--) { 1152010a20caSHong Zhang tmp[*vi++ ] -= (*v++)*tmp[i]; 11536abc6512SBarry Smith } 11546abc6512SBarry Smith } 11556abc6512SBarry Smith 11566abc6512SBarry Smith /* backward solve the L^T */ 11576abc6512SBarry Smith for (i=n-1; i>=0; i--){ 1158010a20caSHong Zhang v = aa + diag[i] - 1 ; 1159010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1160f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 11616abc6512SBarry Smith while (nz--) { 1162010a20caSHong Zhang tmp[*vi-- ] -= (*v--)*tmp[i]; 11636abc6512SBarry Smith } 11646abc6512SBarry Smith } 11656abc6512SBarry Smith 11666abc6512SBarry Smith /* copy tmp into x according to permutation */ 11676abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 11686abc6512SBarry Smith 11692235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 11702235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 11711ebc52fbSHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 11721ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 11736abc6512SBarry Smith 1174efee365bSSatish Balay ierr = PetscLogFlops(2*a->nz);CHKERRQ(ierr); 11753a40ed3dSBarry Smith PetscFunctionReturn(0); 1176da3a660dSBarry Smith } 11779e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 11783c5fc038SBarry Smith EXTERN PetscErrorCode Mat_CheckInode(Mat,PetscTruth); 1179b24902e0SBarry Smith EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,MatDuplicateOption,Mat*); 118008480c60SBarry Smith 11814a2ae208SSatish Balay #undef __FUNCT__ 11824a2ae208SSatish Balay #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ" 1183dfbe8321SBarry Smith PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,MatFactorInfo *info,Mat *fact) 11849e25ed09SBarry Smith { 1185416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 11869e25ed09SBarry Smith IS isicol; 11876849ba73SBarry Smith PetscErrorCode ierr; 1188d0f46423SBarry Smith PetscInt *r,*ic,n=A->rmap->n,*ai=a->i,*aj=a->j,d; 11895a8e39fbSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 11905a8e39fbSHong Zhang PetscInt *bdiag,prow,fm,nzbd,len, reallocs=0,dcount=0; 11915a8e39fbSHong Zhang PetscInt i,levels,diagonal_fill; 119277c4ece6SBarry Smith PetscTruth col_identity,row_identity; 1193329f5518SBarry Smith PetscReal f; 11945a8e39fbSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 11955a8e39fbSHong Zhang PetscBT lnkbt; 11965a8e39fbSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 1197a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1198a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 119909f38230SBarry Smith PetscTruth missing; 12009e25ed09SBarry Smith 12013a40ed3dSBarry Smith PetscFunctionBegin; 1202d0f46423SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 12036cf997b0SBarry Smith f = info->fill; 120438baddfdSBarry Smith levels = (PetscInt)info->levels; 120538baddfdSBarry Smith diagonal_fill = (PetscInt)info->diagonal_fill; 12064c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 120782bf6240SBarry Smith 120808480c60SBarry Smith /* special case that simply copies fill pattern */ 1209be0abb6dSBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 1210be0abb6dSBarry Smith ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 121186bacbd2SBarry Smith if (!levels && row_identity && col_identity) { 1212b24902e0SBarry Smith ierr = MatDuplicateNoCreate_SeqAIJ(A,MAT_DO_NOT_COPY_VALUES,fact);CHKERRQ(ierr); 12135c9eb25fSBarry Smith (*fact)->factor = MAT_FACTOR_LU; 1214f3a39becSBarry Smith (*fact)->info.factor_mallocs = 0; 1215f3a39becSBarry Smith (*fact)->info.fill_ratio_given = info->fill; 1216f3a39becSBarry Smith (*fact)->info.fill_ratio_needed = 1.0; 121708480c60SBarry Smith b = (Mat_SeqAIJ*)(*fact)->data; 12188ef3462fSBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 121909f38230SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 122008480c60SBarry Smith b->row = isrow; 122108480c60SBarry Smith b->col = iscol; 122282bf6240SBarry Smith b->icol = isicol; 1223d0f46423SBarry Smith ierr = PetscMalloc(((*fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1224c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1225c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1226*db4efbfdSBarry Smith (*fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1227*db4efbfdSBarry Smith ierr = MatILUFactorSymbolic_Inode(A,isrow,iscol,info,fact);CHKERRQ(ierr); 1228*db4efbfdSBarry Smith (*fact)->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 12293a40ed3dSBarry Smith PetscFunctionReturn(0); 123008480c60SBarry Smith } 123108480c60SBarry Smith 1232898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1233898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 12349e25ed09SBarry Smith 12359e25ed09SBarry Smith /* get new row pointers */ 12365a8e39fbSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 12375a8e39fbSHong Zhang bi[0] = 0; 12385a8e39fbSHong Zhang /* bdiag is location of diagonal in factor */ 12395a8e39fbSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 12405a8e39fbSHong Zhang bdiag[0] = 0; 12416cf997b0SBarry Smith 12425a8e39fbSHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr); 12435a8e39fbSHong Zhang bjlvl_ptr = (PetscInt**)(bj_ptr + n); 12445a8e39fbSHong Zhang 12455a8e39fbSHong Zhang /* create a linked list for storing column indices of the active row */ 12465a8e39fbSHong Zhang nlnk = n + 1; 12475a8e39fbSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 12485a8e39fbSHong Zhang 12495a8e39fbSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 1250a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 12515a8e39fbSHong Zhang current_space = free_space; 1252a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 12535a8e39fbSHong Zhang current_space_lvl = free_space_lvl; 12545a8e39fbSHong Zhang 12555a8e39fbSHong Zhang for (i=0; i<n; i++) { 12565a8e39fbSHong Zhang nzi = 0; 12576cf997b0SBarry Smith /* copy current row into linked list */ 12585a8e39fbSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 12595a8e39fbSHong Zhang if (!nnz) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix"); 12605a8e39fbSHong Zhang cols = aj + ai[r[i]]; 12615a8e39fbSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 12625a8e39fbSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 12635a8e39fbSHong Zhang nzi += nlnk; 12646cf997b0SBarry Smith 12656cf997b0SBarry Smith /* make sure diagonal entry is included */ 12665a8e39fbSHong Zhang if (diagonal_fill && lnk[i] == -1) { 12676cf997b0SBarry Smith fm = n; 12685a8e39fbSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 12695a8e39fbSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 12705a8e39fbSHong Zhang lnk[fm] = i; 12715a8e39fbSHong Zhang lnk_lvl[i] = 0; 12725a8e39fbSHong Zhang nzi++; dcount++; 12736cf997b0SBarry Smith } 12746cf997b0SBarry Smith 12755a8e39fbSHong Zhang /* add pivot rows into the active row */ 12765a8e39fbSHong Zhang nzbd = 0; 12775a8e39fbSHong Zhang prow = lnk[n]; 12785a8e39fbSHong Zhang while (prow < i) { 12795a8e39fbSHong Zhang nnz = bdiag[prow]; 12805a8e39fbSHong Zhang cols = bj_ptr[prow] + nnz + 1; 12815a8e39fbSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 12825a8e39fbSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 12835a8e39fbSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 12845a8e39fbSHong Zhang nzi += nlnk; 12855a8e39fbSHong Zhang prow = lnk[prow]; 12865a8e39fbSHong Zhang nzbd++; 128756beaf04SBarry Smith } 12885a8e39fbSHong Zhang bdiag[i] = nzbd; 12895a8e39fbSHong Zhang bi[i+1] = bi[i] + nzi; 1290ecf371e4SBarry Smith 12915a8e39fbSHong Zhang /* if free space is not available, make more free space */ 12925a8e39fbSHong Zhang if (current_space->local_remaining<nzi) { 12935a8e39fbSHong Zhang nnz = nzi*(n - i); /* estimated and max additional space needed */ 1294a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1295a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 12965a8e39fbSHong Zhang reallocs++; 12975a8e39fbSHong Zhang } 1298ecf371e4SBarry Smith 12995a8e39fbSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 13005a8e39fbSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 13015a8e39fbSHong Zhang bj_ptr[i] = current_space->array; 13025a8e39fbSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 13035a8e39fbSHong Zhang 13045a8e39fbSHong Zhang /* make sure the active row i has diagonal entry */ 13055a8e39fbSHong Zhang if (*(bj_ptr[i]+bdiag[i]) != i) { 130677431f27SBarry Smith SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 1307d7ee0231SBarry Smith try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 13086cf997b0SBarry Smith } 13095a8e39fbSHong Zhang 13105a8e39fbSHong Zhang current_space->array += nzi; 13115a8e39fbSHong Zhang current_space->local_used += nzi; 13125a8e39fbSHong Zhang current_space->local_remaining -= nzi; 13135a8e39fbSHong Zhang current_space_lvl->array += nzi; 13145a8e39fbSHong Zhang current_space_lvl->local_used += nzi; 13155a8e39fbSHong Zhang current_space_lvl->local_remaining -= nzi; 13169e25ed09SBarry Smith } 13175a8e39fbSHong Zhang 1318898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1319898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 13205a8e39fbSHong Zhang 13215a8e39fbSHong Zhang /* destroy list of free space and other temporary arrays */ 13225a8e39fbSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 1323a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); 13245a8e39fbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1325a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 13265a8e39fbSHong Zhang ierr = PetscFree(bj_ptr);CHKERRQ(ierr); 13279e25ed09SBarry Smith 13286cf91177SBarry Smith #if defined(PETSC_USE_INFO) 1329f64065bbSBarry Smith { 13305a8e39fbSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1331ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 1332ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1333ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 1334ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1335335d9088SBarry Smith if (diagonal_fill) { 1336ae15b995SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 1337335d9088SBarry Smith } 1338f64065bbSBarry Smith } 133963ba0a88SBarry Smith #endif 1340bbb6d6a8SBarry Smith 13419e25ed09SBarry Smith /* put together the new matrix */ 134252e6d16bSBarry Smith ierr = PetscLogObjectParent(*fact,isicol);CHKERRQ(ierr); 1343416022c9SBarry Smith b = (Mat_SeqAIJ*)(*fact)->data; 1344e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1345e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 13467c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 13475a8e39fbSHong Zhang len = (bi[n] )*sizeof(PetscScalar); 1348b0a32e0cSBarry Smith ierr = PetscMalloc(len+1,&b->a);CHKERRQ(ierr); 13495a8e39fbSHong Zhang b->j = bj; 13505a8e39fbSHong Zhang b->i = bi; 13515a8e39fbSHong Zhang for (i=0; i<n; i++) bdiag[i] += bi[i]; 13525a8e39fbSHong Zhang b->diag = bdiag; 1353416022c9SBarry Smith b->ilen = 0; 1354416022c9SBarry Smith b->imax = 0; 1355416022c9SBarry Smith b->row = isrow; 1356416022c9SBarry Smith b->col = iscol; 1357c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1358c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 135982bf6240SBarry Smith b->icol = isicol; 136087828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1361416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 13625a8e39fbSHong Zhang Allocate bdiag, solve_work, new a, new j */ 13635a8e39fbSHong Zhang ierr = PetscLogObjectMemory(*fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 13645a8e39fbSHong Zhang b->maxnz = b->nz = bi[n] ; 13655c9eb25fSBarry Smith (*fact)->factor = MAT_FACTOR_LU; 1366418422e8SSatish Balay (*fact)->info.factor_mallocs = reallocs; 1367ae068f58SLois Curfman McInnes (*fact)->info.fill_ratio_given = f; 13685a8e39fbSHong Zhang (*fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1369*db4efbfdSBarry Smith (*fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 137054e71613SHong Zhang ierr = MatILUFactorSymbolic_Inode(A,isrow,iscol,info,fact);CHKERRQ(ierr); 13713a40ed3dSBarry Smith PetscFunctionReturn(0); 13729e25ed09SBarry Smith } 1373d5d45c9bSBarry Smith 13743c9e8598SHong Zhang #include "src/mat/impls/sbaij/seq/sbaij.h" 1375a6175056SHong Zhang #undef __FUNCT__ 1376a6175056SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ" 1377af281ebdSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat A,MatFactorInfo *info,Mat *B) 1378a6175056SHong Zhang { 13792ed133dbSHong Zhang Mat C = *B; 13800968510aSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 13812ed133dbSHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 13829bfd6278SHong Zhang IS ip=b->row,iip = b->icol; 13832ed133dbSHong Zhang PetscErrorCode ierr; 1384d0f46423SBarry Smith PetscInt *rip,*riip,i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol; 13852ed133dbSHong Zhang PetscInt *ai=a->i,*aj=a->j; 13862ed133dbSHong Zhang PetscInt k,jmin,jmax,*jl,*il,col,nexti,ili,nz; 1387622e2028SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 1388540703acSHong Zhang PetscReal zeropivot,rs,shiftnz; 1389fbf22428SSatish Balay PetscReal shiftpd; 1390540703acSHong Zhang ChShift_Ctx sctx; 1391540703acSHong Zhang PetscInt newshift; 1392*db4efbfdSBarry Smith PetscTruth perm_identity; 1393d5d45c9bSBarry Smith 1394a6175056SHong Zhang PetscFunctionBegin; 1395117f1891Stmunson 1396540703acSHong Zhang shiftnz = info->shiftnz; 1397540703acSHong Zhang shiftpd = info->shiftpd; 1398ee45ca4aSHong Zhang zeropivot = info->zeropivot; 1399ee45ca4aSHong Zhang 14002ed133dbSHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 14019bfd6278SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 14022ed133dbSHong Zhang 14032ed133dbSHong Zhang /* initialization */ 14042ed133dbSHong Zhang nz = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar); 14052ed133dbSHong Zhang ierr = PetscMalloc(nz,&il);CHKERRQ(ierr); 14062ed133dbSHong Zhang jl = il + mbs; 14072ed133dbSHong Zhang rtmp = (MatScalar*)(jl + mbs); 14082ed133dbSHong Zhang 1409540703acSHong Zhang sctx.shift_amount = 0; 1410540703acSHong Zhang sctx.nshift = 0; 14112ed133dbSHong Zhang do { 1412540703acSHong Zhang sctx.chshift = PETSC_FALSE; 14132ed133dbSHong Zhang for (i=0; i<mbs; i++) { 14142ed133dbSHong Zhang rtmp[i] = 0.0; jl[i] = mbs; il[0] = 0; 1415a6175056SHong Zhang } 14162ed133dbSHong Zhang 14172ed133dbSHong Zhang for (k = 0; k<mbs; k++){ 1418c05c3958SHong Zhang bval = ba + bi[k]; 14192ed133dbSHong Zhang /* initialize k-th row by the perm[k]-th row of A */ 14202ed133dbSHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 14212ed133dbSHong Zhang for (j = jmin; j < jmax; j++){ 14229bfd6278SHong Zhang col = riip[aj[j]]; 14232ed133dbSHong Zhang if (col >= k){ /* only take upper triangular entry */ 14242ed133dbSHong Zhang rtmp[col] = aa[j]; 1425c05c3958SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 14262ed133dbSHong Zhang } 14272ed133dbSHong Zhang } 142839e3d630SHong Zhang /* shift the diagonal of the matrix */ 1429540703acSHong Zhang if (sctx.nshift) rtmp[k] += sctx.shift_amount; 14302ed133dbSHong Zhang 14312ed133dbSHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 14322ed133dbSHong Zhang dk = rtmp[k]; 14332ed133dbSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 14342ed133dbSHong Zhang 14352ed133dbSHong Zhang while (i < k){ 14362ed133dbSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 14372ed133dbSHong Zhang 14382ed133dbSHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 14392ed133dbSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 14402ed133dbSHong Zhang uikdi = - ba[ili]*ba[bi[i]]; /* diagonal(k) */ 14412ed133dbSHong Zhang dk += uikdi*ba[ili]; 14422ed133dbSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 14432ed133dbSHong Zhang 14442ed133dbSHong Zhang /* add multiple of row i to k-th row */ 14452ed133dbSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 14462ed133dbSHong Zhang if (jmin < jmax){ 14472ed133dbSHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 14482ed133dbSHong Zhang /* update il and jl for row i */ 14492ed133dbSHong Zhang il[i] = jmin; 14502ed133dbSHong Zhang j = bj[jmin]; jl[i] = jl[j]; jl[j] = i; 14512ed133dbSHong Zhang } 14522ed133dbSHong Zhang i = nexti; 14532ed133dbSHong Zhang } 14542ed133dbSHong Zhang 14550697b6caSHong Zhang /* shift the diagonals when zero pivot is detected */ 14560697b6caSHong Zhang /* compute rs=sum of abs(off-diagonal) */ 14570697b6caSHong Zhang rs = 0.0; 14583c9e8598SHong Zhang jmin = bi[k]+1; 14593c9e8598SHong Zhang nz = bi[k+1] - jmin; 14603c9e8598SHong Zhang bcol = bj + jmin; 14613c9e8598SHong Zhang while (nz--){ 146289f845c8SHong Zhang rs += PetscAbsScalar(rtmp[*bcol]); 146389f845c8SHong Zhang bcol++; 14643c9e8598SHong Zhang } 1465540703acSHong Zhang 1466540703acSHong Zhang sctx.rs = rs; 1467540703acSHong Zhang sctx.pv = dk; 14685b5bc046SBarry Smith ierr = MatCholeskyCheckShift_inline(info,sctx,k,newshift);CHKERRQ(ierr); 1469117f1891Stmunson 1470117f1891Stmunson if (newshift == 1) { 1471117f1891Stmunson if (!sctx.shift_amount) { 1472117f1891Stmunson sctx.shift_amount = 1e-5; 1473117f1891Stmunson } 1474117f1891Stmunson break; 1475117f1891Stmunson } 14763c9e8598SHong Zhang 14773c9e8598SHong Zhang /* copy data into U(k,:) */ 147839e3d630SHong Zhang ba[bi[k]] = 1.0/dk; /* U(k,k) */ 147939e3d630SHong Zhang jmin = bi[k]+1; jmax = bi[k+1]; 148039e3d630SHong Zhang if (jmin < jmax) { 148139e3d630SHong Zhang for (j=jmin; j<jmax; j++){ 148239e3d630SHong Zhang col = bj[j]; ba[j] = rtmp[col]; rtmp[col] = 0.0; 14833c9e8598SHong Zhang } 148439e3d630SHong Zhang /* add the k-th row into il and jl */ 14853c9e8598SHong Zhang il[k] = jmin; 14863c9e8598SHong Zhang i = bj[jmin]; jl[k] = jl[i]; jl[i] = k; 14873c9e8598SHong Zhang } 14883c9e8598SHong Zhang } 1489540703acSHong Zhang } while (sctx.chshift); 14903c9e8598SHong Zhang ierr = PetscFree(il);CHKERRQ(ierr); 14913c9e8598SHong Zhang 149239e3d630SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 14939bfd6278SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 1494*db4efbfdSBarry Smith 1495*db4efbfdSBarry Smith ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 1496*db4efbfdSBarry Smith if (perm_identity){ 1497*db4efbfdSBarry Smith (*B)->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 1498*db4efbfdSBarry Smith (*B)->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 1499*db4efbfdSBarry Smith (*B)->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering; 1500*db4efbfdSBarry Smith (*B)->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering; 1501*db4efbfdSBarry Smith } else { 1502*db4efbfdSBarry Smith (*B)->ops->solve = MatSolve_SeqSBAIJ_1; 1503*db4efbfdSBarry Smith (*B)->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 1504*db4efbfdSBarry Smith (*B)->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1; 1505*db4efbfdSBarry Smith (*B)->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1; 1506*db4efbfdSBarry Smith } 1507*db4efbfdSBarry Smith 15083c9e8598SHong Zhang C->assembled = PETSC_TRUE; 15093c9e8598SHong Zhang C->preallocated = PETSC_TRUE; 1510d0f46423SBarry Smith ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 1511540703acSHong Zhang if (sctx.nshift){ 1512540703acSHong Zhang if (shiftnz) { 15131e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 1514540703acSHong Zhang } else if (shiftpd) { 15151e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 15160697b6caSHong Zhang } 15173c9e8598SHong Zhang } 15183c9e8598SHong Zhang PetscFunctionReturn(0); 15193c9e8598SHong Zhang } 1520a6175056SHong Zhang 1521a6175056SHong Zhang #undef __FUNCT__ 1522a6175056SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ" 1523dfbe8321SBarry Smith PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat A,IS perm,MatFactorInfo *info,Mat *fact) 1524a6175056SHong Zhang { 15250968510aSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1526ed59401aSHong Zhang Mat_SeqSBAIJ *b; 1527dfbe8321SBarry Smith PetscErrorCode ierr; 152858ebbce7SBarry Smith PetscTruth perm_identity,missing; 1529d0f46423SBarry Smith PetscInt reallocs=0,*rip,*riip,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui; 1530ed59401aSHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 153158ebbce7SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 15325a8e39fbSHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 1533ed59401aSHong Zhang PetscReal fill=info->fill,levels=info->levels; 1534a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1535a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 15367a48dd6fSHong Zhang PetscBT lnkbt; 1537b635d36bSHong Zhang IS iperm; 1538a6175056SHong Zhang 1539a6175056SHong Zhang PetscFunctionBegin; 1540d0f46423SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 154158ebbce7SBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 154258ebbce7SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 1543653a6975SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 1544b635d36bSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 15457a48dd6fSHong Zhang 154639e3d630SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 154739e3d630SHong Zhang ui[0] = 0; 154839e3d630SHong Zhang 1549b635d36bSHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 1550622e2028SHong Zhang if (!levels && perm_identity) { 155158ebbce7SBarry Smith 1552ed59401aSHong Zhang for (i=0; i<am; i++) { 155339e3d630SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 1554ed59401aSHong Zhang } 155539e3d630SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 155639e3d630SHong Zhang cols = uj; 1557ed59401aSHong Zhang for (i=0; i<am; i++) { 1558ed59401aSHong Zhang aj = a->j + a->diag[i]; 155939e3d630SHong Zhang ncols = ui[i+1] - ui[i]; 156039e3d630SHong Zhang for (j=0; j<ncols; j++) *cols++ = *aj++; 1561ed59401aSHong Zhang } 156239e3d630SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 1563b635d36bSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 1564b635d36bSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 1565b635d36bSHong Zhang 15667a48dd6fSHong Zhang /* initialization */ 15675a8e39fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 15687a48dd6fSHong Zhang 15697a48dd6fSHong Zhang /* jl: linked list for storing indices of the pivot rows 15707a48dd6fSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 15711393bc97SHong Zhang ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr); 15727a48dd6fSHong Zhang il = jl + am; 15737a48dd6fSHong Zhang uj_ptr = (PetscInt**)(il + am); 15747a48dd6fSHong Zhang uj_lvl_ptr = (PetscInt**)(uj_ptr + am); 15757a48dd6fSHong Zhang for (i=0; i<am; i++){ 15767a48dd6fSHong Zhang jl[i] = am; il[i] = 0; 15777a48dd6fSHong Zhang } 15787a48dd6fSHong Zhang 15797a48dd6fSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 15807a48dd6fSHong Zhang nlnk = am + 1; 15812ed133dbSHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 15827a48dd6fSHong Zhang 15837a48dd6fSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 1584a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 15857a48dd6fSHong Zhang current_space = free_space; 1586a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 15877a48dd6fSHong Zhang current_space_lvl = free_space_lvl; 15887a48dd6fSHong Zhang 15897a48dd6fSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 15907a48dd6fSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 15917a48dd6fSHong Zhang nzk = 0; 1592622e2028SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 1593b635d36bSHong Zhang if (!ncols) SETERRQ(PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix"); 1594622e2028SHong Zhang ncols_upper = 0; 1595622e2028SHong Zhang for (j=0; j<ncols; j++){ 1596b635d36bSHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 1597b635d36bSHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 15985a8e39fbSHong Zhang ajtmp[ncols_upper] = i; 1599622e2028SHong Zhang ncols_upper++; 1600622e2028SHong Zhang } 1601622e2028SHong Zhang } 1602b635d36bSHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 16037a48dd6fSHong Zhang nzk += nlnk; 16047a48dd6fSHong Zhang 16057a48dd6fSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 16067a48dd6fSHong Zhang prow = jl[k]; /* 1st pivot row */ 16077a48dd6fSHong Zhang 16087a48dd6fSHong Zhang while (prow < k){ 16097a48dd6fSHong Zhang nextprow = jl[prow]; 16107a48dd6fSHong Zhang 16117a48dd6fSHong Zhang /* merge prow into k-th row */ 16127a48dd6fSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 16137a48dd6fSHong Zhang jmax = ui[prow+1]; 16147a48dd6fSHong Zhang ncols = jmax-jmin; 1615ed59401aSHong Zhang i = jmin - ui[prow]; 1616ed59401aSHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 16175a8e39fbSHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 16185a8e39fbSHong Zhang j = *(uj - 1); 16195a8e39fbSHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 16207a48dd6fSHong Zhang nzk += nlnk; 16217a48dd6fSHong Zhang 16227a48dd6fSHong Zhang /* update il and jl for prow */ 16237a48dd6fSHong Zhang if (jmin < jmax){ 16247a48dd6fSHong Zhang il[prow] = jmin; 16257a48dd6fSHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 16267a48dd6fSHong Zhang } 16277a48dd6fSHong Zhang prow = nextprow; 16287a48dd6fSHong Zhang } 16297a48dd6fSHong Zhang 16307a48dd6fSHong Zhang /* if free space is not available, make more free space */ 16317a48dd6fSHong Zhang if (current_space->local_remaining<nzk) { 16327a48dd6fSHong Zhang i = am - k + 1; /* num of unfactored rows */ 16337a48dd6fSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 1634a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 1635a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 16367a48dd6fSHong Zhang reallocs++; 16377a48dd6fSHong Zhang } 16387a48dd6fSHong Zhang 16397a48dd6fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 1640b635d36bSHong Zhang if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 16412ed133dbSHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 16427a48dd6fSHong Zhang 16437a48dd6fSHong Zhang /* add the k-th row into il and jl */ 164439e3d630SHong Zhang if (nzk > 1){ 16457a48dd6fSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 16467a48dd6fSHong Zhang jl[k] = jl[i]; jl[i] = k; 16477a48dd6fSHong Zhang il[k] = ui[k] + 1; 16487a48dd6fSHong Zhang } 16497a48dd6fSHong Zhang uj_ptr[k] = current_space->array; 16507a48dd6fSHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 16517a48dd6fSHong Zhang 16527a48dd6fSHong Zhang current_space->array += nzk; 16537a48dd6fSHong Zhang current_space->local_used += nzk; 16547a48dd6fSHong Zhang current_space->local_remaining -= nzk; 16557a48dd6fSHong Zhang 16567a48dd6fSHong Zhang current_space_lvl->array += nzk; 16577a48dd6fSHong Zhang current_space_lvl->local_used += nzk; 16587a48dd6fSHong Zhang current_space_lvl->local_remaining -= nzk; 16597a48dd6fSHong Zhang 16607a48dd6fSHong Zhang ui[k+1] = ui[k] + nzk; 16617a48dd6fSHong Zhang } 16627a48dd6fSHong Zhang 16636cf91177SBarry Smith #if defined(PETSC_USE_INFO) 16647a48dd6fSHong Zhang if (ai[am] != 0) { 166539e3d630SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 1666ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 1667ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1668ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 16697a48dd6fSHong Zhang } else { 1670ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 16717a48dd6fSHong Zhang } 167263ba0a88SBarry Smith #endif 16737a48dd6fSHong Zhang 16747a48dd6fSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 1675b635d36bSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 16767a48dd6fSHong Zhang ierr = PetscFree(jl);CHKERRQ(ierr); 16775a8e39fbSHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 16787a48dd6fSHong Zhang 16797a48dd6fSHong Zhang /* destroy list of free space and other temporary array(s) */ 16807a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 1681a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 16822ed133dbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1683a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 16847a48dd6fSHong Zhang 168539e3d630SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 168639e3d630SHong Zhang 16877a48dd6fSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 16887a48dd6fSHong Zhang 1689b24902e0SBarry Smith b = (Mat_SeqSBAIJ*)(*fact)->data; 16907a48dd6fSHong Zhang b->singlemalloc = PETSC_FALSE; 16917a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 16927a48dd6fSHong Zhang b->j = uj; 16937a48dd6fSHong Zhang b->i = ui; 16947a48dd6fSHong Zhang b->diag = 0; 16957a48dd6fSHong Zhang b->ilen = 0; 16967a48dd6fSHong Zhang b->imax = 0; 16977a48dd6fSHong Zhang b->row = perm; 1698b635d36bSHong Zhang b->col = perm; 1699b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 1700b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 1701b635d36bSHong Zhang b->icol = iperm; 17027a48dd6fSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 17037a48dd6fSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1704b24902e0SBarry Smith ierr = PetscLogObjectMemory((*fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 17057a48dd6fSHong Zhang b->maxnz = b->nz = ui[am]; 1706e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1707e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 17087a48dd6fSHong Zhang 1709b24902e0SBarry Smith (*fact)->info.factor_mallocs = reallocs; 1710b24902e0SBarry Smith (*fact)->info.fill_ratio_given = fill; 17117a48dd6fSHong Zhang if (ai[am] != 0) { 1712b24902e0SBarry Smith (*fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 17137a48dd6fSHong Zhang } else { 1714b24902e0SBarry Smith (*fact)->info.fill_ratio_needed = 0.0; 17157a48dd6fSHong Zhang } 171639e3d630SHong Zhang (*fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 1717a6175056SHong Zhang PetscFunctionReturn(0); 1718a6175056SHong Zhang } 1719d5d45c9bSBarry Smith 1720f76d2b81SHong Zhang #undef __FUNCT__ 1721f76d2b81SHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ" 1722dfbe8321SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat A,IS perm,MatFactorInfo *info,Mat *fact) 1723f76d2b81SHong Zhang { 1724f76d2b81SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 172510c27e3dSHong Zhang Mat_SeqSBAIJ *b; 1726dfbe8321SBarry Smith PetscErrorCode ierr; 1727f76d2b81SHong Zhang PetscTruth perm_identity; 172810c27e3dSHong Zhang PetscReal fill = info->fill; 1729d0f46423SBarry Smith PetscInt *rip,*riip,i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 173010c27e3dSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 17312ed133dbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr; 1732a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 173310c27e3dSHong Zhang PetscBT lnkbt; 17342ed133dbSHong Zhang IS iperm; 1735f76d2b81SHong Zhang 1736f76d2b81SHong Zhang PetscFunctionBegin; 1737d0f46423SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 17382ed133dbSHong Zhang /* check whether perm is the identity mapping */ 1739f76d2b81SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 17402ed133dbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 17412ed133dbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 17429bfd6278SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 174310c27e3dSHong Zhang 174410c27e3dSHong Zhang /* initialization */ 17451393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 174610c27e3dSHong Zhang ui[0] = 0; 174710c27e3dSHong Zhang 174810c27e3dSHong Zhang /* jl: linked list for storing indices of the pivot rows 17491393bc97SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 17501393bc97SHong Zhang ierr = PetscMalloc((3*am+1)*sizeof(PetscInt)+am*sizeof(PetscInt**),&jl);CHKERRQ(ierr); 17511393bc97SHong Zhang il = jl + am; 17521393bc97SHong Zhang cols = il + am; 17531393bc97SHong Zhang ui_ptr = (PetscInt**)(cols + am); 17541393bc97SHong Zhang for (i=0; i<am; i++){ 17551393bc97SHong Zhang jl[i] = am; il[i] = 0; 1756f76d2b81SHong Zhang } 175710c27e3dSHong Zhang 175810c27e3dSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 17591393bc97SHong Zhang nlnk = am + 1; 17601393bc97SHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 176110c27e3dSHong Zhang 17621393bc97SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 1763a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 176410c27e3dSHong Zhang current_space = free_space; 176510c27e3dSHong Zhang 17661393bc97SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 176710c27e3dSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 176810c27e3dSHong Zhang nzk = 0; 17692ed133dbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 17709bfd6278SHong Zhang if (!ncols) SETERRQ(PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix"); 17712ed133dbSHong Zhang ncols_upper = 0; 17722ed133dbSHong Zhang for (j=0; j<ncols; j++){ 17739bfd6278SHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 17742ed133dbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 17752ed133dbSHong Zhang cols[ncols_upper] = i; 17762ed133dbSHong Zhang ncols_upper++; 17772ed133dbSHong Zhang } 17782ed133dbSHong Zhang } 17791393bc97SHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 178010c27e3dSHong Zhang nzk += nlnk; 178110c27e3dSHong Zhang 178210c27e3dSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 178310c27e3dSHong Zhang prow = jl[k]; /* 1st pivot row */ 178410c27e3dSHong Zhang 178510c27e3dSHong Zhang while (prow < k){ 178610c27e3dSHong Zhang nextprow = jl[prow]; 178710c27e3dSHong Zhang /* merge prow into k-th row */ 17881393bc97SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 178910c27e3dSHong Zhang jmax = ui[prow+1]; 179010c27e3dSHong Zhang ncols = jmax-jmin; 17911393bc97SHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 17925a8e39fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 179310c27e3dSHong Zhang nzk += nlnk; 179410c27e3dSHong Zhang 179510c27e3dSHong Zhang /* update il and jl for prow */ 179610c27e3dSHong Zhang if (jmin < jmax){ 179710c27e3dSHong Zhang il[prow] = jmin; 17982ed133dbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 179910c27e3dSHong Zhang } 180010c27e3dSHong Zhang prow = nextprow; 180110c27e3dSHong Zhang } 180210c27e3dSHong Zhang 180310c27e3dSHong Zhang /* if free space is not available, make more free space */ 180410c27e3dSHong Zhang if (current_space->local_remaining<nzk) { 18051393bc97SHong Zhang i = am - k + 1; /* num of unfactored rows */ 180610c27e3dSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 1807a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 180810c27e3dSHong Zhang reallocs++; 180910c27e3dSHong Zhang } 181010c27e3dSHong Zhang 181110c27e3dSHong Zhang /* copy data into free space, then initialize lnk */ 18121393bc97SHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 181310c27e3dSHong Zhang 181410c27e3dSHong Zhang /* add the k-th row into il and jl */ 181510c27e3dSHong Zhang if (nzk-1 > 0){ 18161393bc97SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 181710c27e3dSHong Zhang jl[k] = jl[i]; jl[i] = k; 181810c27e3dSHong Zhang il[k] = ui[k] + 1; 181910c27e3dSHong Zhang } 18202ed133dbSHong Zhang ui_ptr[k] = current_space->array; 182110c27e3dSHong Zhang current_space->array += nzk; 182210c27e3dSHong Zhang current_space->local_used += nzk; 182310c27e3dSHong Zhang current_space->local_remaining -= nzk; 182410c27e3dSHong Zhang 182510c27e3dSHong Zhang ui[k+1] = ui[k] + nzk; 182610c27e3dSHong Zhang } 182710c27e3dSHong Zhang 18286cf91177SBarry Smith #if defined(PETSC_USE_INFO) 18291393bc97SHong Zhang if (ai[am] != 0) { 18301393bc97SHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 1831ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 1832ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1833ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 183410c27e3dSHong Zhang } else { 1835ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 183610c27e3dSHong Zhang } 183763ba0a88SBarry Smith #endif 183810c27e3dSHong Zhang 183910c27e3dSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 18409bfd6278SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 184110c27e3dSHong Zhang ierr = PetscFree(jl);CHKERRQ(ierr); 184210c27e3dSHong Zhang 184310c27e3dSHong Zhang /* destroy list of free space and other temporary array(s) */ 18441393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 1845a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 184610c27e3dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 184710c27e3dSHong Zhang 184810c27e3dSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 184910c27e3dSHong Zhang 1850b24902e0SBarry Smith b = (Mat_SeqSBAIJ*)(*fact)->data; 185110c27e3dSHong Zhang b->singlemalloc = PETSC_FALSE; 1852e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1853e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 18541393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 185510c27e3dSHong Zhang b->j = uj; 185610c27e3dSHong Zhang b->i = ui; 185710c27e3dSHong Zhang b->diag = 0; 185810c27e3dSHong Zhang b->ilen = 0; 185910c27e3dSHong Zhang b->imax = 0; 186010c27e3dSHong Zhang b->row = perm; 18619bfd6278SHong Zhang b->col = perm; 18629bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 18639bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 18649bfd6278SHong Zhang b->icol = iperm; 186510c27e3dSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 18661393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1867b24902e0SBarry Smith ierr = PetscLogObjectMemory(*fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 18681393bc97SHong Zhang b->maxnz = b->nz = ui[am]; 186910c27e3dSHong Zhang 1870b24902e0SBarry Smith (*fact)->info.factor_mallocs = reallocs; 1871b24902e0SBarry Smith (*fact)->info.fill_ratio_given = fill; 18721393bc97SHong Zhang if (ai[am] != 0) { 1873b24902e0SBarry Smith (*fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 187410c27e3dSHong Zhang } else { 1875b24902e0SBarry Smith (*fact)->info.fill_ratio_needed = 0.0; 187610c27e3dSHong Zhang } 187739e3d630SHong Zhang (*fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 1878f76d2b81SHong Zhang PetscFunctionReturn(0); 1879f76d2b81SHong Zhang } 1880