1be1d678aSKris Buschelman #define PETSCMAT_DLL 2289bc588SBarry Smith 3b377110cSBarry Smith 47c4f633dSBarry Smith #include "../src/mat/impls/aij/seq/aij.h" 5783ef271SHong Zhang #include "../src/mat/impls/sbaij/seq/sbaij.h" 61393bc97SHong Zhang #include "petscbt.h" 77c4f633dSBarry Smith #include "../src/mat/utils/freespace.h" 83b2fbd54SBarry Smith 9f6a26d55SMatthew Knepley EXTERN_C_BEGIN 104a2ae208SSatish Balay #undef __FUNCT__ 114a2ae208SSatish Balay #define __FUNCT__ "MatOrdering_Flow_SeqAIJ" 128fa24674SBarry Smith /* 138fa24674SBarry Smith Computes an ordering to get most of the large numerical values in the lower triangular part of the matrix 148fa24674SBarry Smith */ 15dfbe8321SBarry Smith PetscErrorCode MatOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol) 163b2fbd54SBarry Smith { 178fa24674SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)mat->data; 188fa24674SBarry Smith PetscErrorCode ierr; 19889a8dedSBarry Smith PetscInt i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order; 208fa24674SBarry Smith const PetscInt *ai = a->i, *aj = a->j; 218fa24674SBarry Smith const PetscScalar *aa = a->a; 228fa24674SBarry Smith PetscTruth *done; 23889a8dedSBarry Smith PetscReal best,past = 0,future; 243a40ed3dSBarry Smith 258fa24674SBarry Smith PetscFunctionBegin; 268fa24674SBarry Smith /* pick initial row */ 278fa24674SBarry Smith best = -1; 288fa24674SBarry Smith for (i=0; i<n; i++) { 298fa24674SBarry Smith future = 0; 308fa24674SBarry Smith for (j=ai[i]; j<ai[i+1]; j++) { 318fa24674SBarry Smith if (aj[j] != i) future += PetscAbsScalar(aa[j]); else past = PetscAbsScalar(aa[j]); 328fa24674SBarry Smith } 338fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 348fa24674SBarry Smith if (past/future > best) { 358fa24674SBarry Smith best = past/future; 368fa24674SBarry Smith current = i; 378fa24674SBarry Smith } 388fa24674SBarry Smith } 398fa24674SBarry Smith 408fa24674SBarry Smith ierr = PetscMalloc(n*sizeof(PetscTruth),&done);CHKERRQ(ierr); 418fa24674SBarry Smith ierr = PetscMalloc(n*sizeof(PetscInt),&order);CHKERRQ(ierr); 428fa24674SBarry Smith ierr = PetscMemzero(done,n*sizeof(PetscTruth));CHKERRQ(ierr); 438fa24674SBarry Smith order[0] = current; 448fa24674SBarry Smith for (i=0; i<n-1; i++) { 458fa24674SBarry Smith done[current] = PETSC_TRUE; 468fa24674SBarry Smith best = -1; 478fa24674SBarry Smith /* loop over all neighbors of current pivot */ 488fa24674SBarry Smith for (j=ai[current]; j<ai[current+1]; j++) { 498fa24674SBarry Smith jj = aj[j]; 508fa24674SBarry Smith if (done[jj]) continue; 518fa24674SBarry Smith /* loop over columns of potential next row computing weights for below and above diagonal */ 528fa24674SBarry Smith past = future = 0.0; 538fa24674SBarry Smith for (k=ai[jj]; k<ai[jj+1]; k++) { 548fa24674SBarry Smith kk = aj[k]; 558fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[k]); 568fa24674SBarry Smith else if (kk != jj) future += PetscAbsScalar(aa[k]); 578fa24674SBarry Smith } 588fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 598fa24674SBarry Smith if (past/future > best) { 608fa24674SBarry Smith best = past/future; 618fa24674SBarry Smith newcurrent = jj; 628fa24674SBarry Smith } 638fa24674SBarry Smith } 648fa24674SBarry Smith if (best == -1) { /* no neighbors to select from so select best of all that remain */ 658fa24674SBarry Smith best = -1; 668fa24674SBarry Smith for (k=0; k<n; k++) { 678fa24674SBarry Smith if (done[k]) continue; 688fa24674SBarry Smith future = 0; 698fa24674SBarry Smith past = 0; 708fa24674SBarry Smith for (j=ai[k]; j<ai[k+1]; j++) { 718fa24674SBarry Smith kk = aj[j]; 728fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[j]); 738fa24674SBarry Smith else if (kk != k) future += PetscAbsScalar(aa[j]); 748fa24674SBarry Smith } 758fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 768fa24674SBarry Smith if (past/future > best) { 778fa24674SBarry Smith best = past/future; 788fa24674SBarry Smith newcurrent = k; 798fa24674SBarry Smith } 808fa24674SBarry Smith } 818fa24674SBarry Smith } 828fa24674SBarry Smith if (current == newcurrent) SETERRQ(PETSC_ERR_PLIB,"newcurrent cannot be current"); 838fa24674SBarry Smith current = newcurrent; 848fa24674SBarry Smith order[i+1] = current; 858fa24674SBarry Smith } 868fa24674SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,order,irow);CHKERRQ(ierr); 878fa24674SBarry Smith *icol = *irow; 888fa24674SBarry Smith ierr = PetscObjectReference((PetscObject)*irow);CHKERRQ(ierr); 898fa24674SBarry Smith ierr = PetscFree(done);CHKERRQ(ierr); 908fa24674SBarry Smith ierr = PetscFree(order);CHKERRQ(ierr); 91d64ed03dSBarry Smith PetscFunctionReturn(0); 923b2fbd54SBarry Smith } 93f6a26d55SMatthew Knepley EXTERN_C_END 943b2fbd54SBarry Smith 95e631078cSBarry Smith EXTERN_C_BEGIN 964a2ae208SSatish Balay #undef __FUNCT__ 97db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc" 98db4efbfdSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscTruth *flg) 99db4efbfdSBarry Smith { 100db4efbfdSBarry Smith PetscFunctionBegin; 101db4efbfdSBarry Smith *flg = PETSC_TRUE; 102db4efbfdSBarry Smith PetscFunctionReturn(0); 103db4efbfdSBarry Smith } 104db4efbfdSBarry Smith EXTERN_C_END 105db4efbfdSBarry Smith 106db4efbfdSBarry Smith EXTERN_C_BEGIN 107db4efbfdSBarry Smith #undef __FUNCT__ 108b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc" 109b24902e0SBarry Smith PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B) 110b24902e0SBarry Smith { 111d0f46423SBarry Smith PetscInt n = A->rmap->n; 112b24902e0SBarry Smith PetscErrorCode ierr; 113b24902e0SBarry Smith 114b24902e0SBarry Smith PetscFunctionBegin; 115b24902e0SBarry Smith ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr); 116b24902e0SBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 117599ef60dSHong Zhang if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT){ 118b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr); 119db4efbfdSBarry Smith (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ; 12035bd34faSBarry Smith (*B)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqAIJ; 1211da05e5aSHong Zhang (*B)->ops->iludtfactor = MatILUDTFactor_SeqAIJ; 122b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 123b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 124b24902e0SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1255c9eb25fSBarry Smith (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqAIJ; 1265c9eb25fSBarry Smith (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ; 127b24902e0SBarry Smith } else SETERRQ(PETSC_ERR_SUP,"Factor type not supported"); 128db4efbfdSBarry Smith (*B)->factor = ftype; 129b24902e0SBarry Smith PetscFunctionReturn(0); 130b24902e0SBarry Smith } 131e631078cSBarry Smith EXTERN_C_END 132b24902e0SBarry Smith 133b24902e0SBarry Smith #undef __FUNCT__ 134b9617806SBarry Smith #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ" 1350481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 136289bc588SBarry Smith { 137416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 138289bc588SBarry Smith IS isicol; 1396849ba73SBarry Smith PetscErrorCode ierr; 1405d0c19d7SBarry Smith const PetscInt *r,*ic; 1415d0c19d7SBarry Smith PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 1421393bc97SHong Zhang PetscInt *bi,*bj,*ajtmp; 1431393bc97SHong Zhang PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 1449e046878SBarry Smith PetscReal f; 1454875ba38SHong Zhang PetscInt nlnk,*lnk,k,**bi_ptr; 146a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1471393bc97SHong Zhang PetscBT lnkbt; 148ce3d78c0SShri Abhyankar PetscTruth newdatastruct=PETSC_FALSE; 149289bc588SBarry Smith 1503a40ed3dSBarry Smith PetscFunctionBegin; 151ce3d78c0SShri Abhyankar ierr = PetscOptionsGetTruth(PETSC_NULL,"-lu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 152ce3d78c0SShri Abhyankar if(newdatastruct){ 153ce3d78c0SShri Abhyankar ierr = MatLUFactorSymbolic_SeqAIJ_newdatastruct(B,A,isrow,iscol,info);CHKERRQ(ierr); 154ce3d78c0SShri Abhyankar PetscFunctionReturn(0); 155ce3d78c0SShri Abhyankar } 156ce3d78c0SShri Abhyankar 157d0f46423SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square"); 1584c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 159ac121b3dSBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 160ac121b3dSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 161289bc588SBarry Smith 162289bc588SBarry Smith /* get new row pointers */ 1631393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 1641393bc97SHong Zhang bi[0] = 0; 1651393bc97SHong Zhang 1661393bc97SHong Zhang /* bdiag is location of diagonal in factor */ 1671393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1681393bc97SHong Zhang bdiag[0] = 0; 1691393bc97SHong Zhang 1701393bc97SHong Zhang /* linked list for storing column indices of the active row */ 1711393bc97SHong Zhang nlnk = n + 1; 1721393bc97SHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1731393bc97SHong Zhang 17435baf27eSSatish Balay ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 1751393bc97SHong Zhang 1761393bc97SHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 177d3d32019SBarry Smith f = info->fill; 178a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1791393bc97SHong Zhang current_space = free_space; 180289bc588SBarry Smith 181289bc588SBarry Smith for (i=0; i<n; i++) { 1821393bc97SHong Zhang /* copy previous fill into linked list */ 183289bc588SBarry Smith nzi = 0; 1841393bc97SHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 1853b4a8b6dSBarry Smith if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 1861393bc97SHong Zhang ajtmp = aj + ai[r[i]]; 187afcc9904SHong Zhang ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1881393bc97SHong Zhang nzi += nlnk; 1891393bc97SHong Zhang 1901393bc97SHong Zhang /* add pivot rows into linked list */ 1911393bc97SHong Zhang row = lnk[n]; 1921393bc97SHong Zhang while (row < i) { 193d1170560SHong Zhang nzbd = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */ 194d1170560SHong Zhang ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 195d1170560SHong Zhang ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 1961393bc97SHong Zhang nzi += nlnk; 1971393bc97SHong Zhang row = lnk[row]; 198289bc588SBarry Smith } 1991393bc97SHong Zhang bi[i+1] = bi[i] + nzi; 2001393bc97SHong Zhang im[i] = nzi; 2011393bc97SHong Zhang 2021393bc97SHong Zhang /* mark bdiag */ 2031393bc97SHong Zhang nzbd = 0; 2041393bc97SHong Zhang nnz = nzi; 2051393bc97SHong Zhang k = lnk[n]; 2061393bc97SHong Zhang while (nnz-- && k < i){ 2071393bc97SHong Zhang nzbd++; 2081393bc97SHong Zhang k = lnk[k]; 2091393bc97SHong Zhang } 2101393bc97SHong Zhang bdiag[i] = bi[i] + nzbd; 2111393bc97SHong Zhang 2121393bc97SHong Zhang /* if free space is not available, make more free space */ 2131393bc97SHong Zhang if (current_space->local_remaining<nzi) { 2141393bc97SHong Zhang nnz = (n - i)*nzi; /* estimated and max additional space needed */ 215a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2161393bc97SHong Zhang reallocs++; 2171393bc97SHong Zhang } 2181393bc97SHong Zhang 2191393bc97SHong Zhang /* copy data into free space, then initialize lnk */ 2201393bc97SHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 2211393bc97SHong Zhang bi_ptr[i] = current_space->array; 2221393bc97SHong Zhang current_space->array += nzi; 2231393bc97SHong Zhang current_space->local_used += nzi; 2241393bc97SHong Zhang current_space->local_remaining -= nzi; 225289bc588SBarry Smith } 2266cf91177SBarry Smith #if defined(PETSC_USE_INFO) 227464e8d44SSatish Balay if (ai[n] != 0) { 2281393bc97SHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 229ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 230ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 231ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 232ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 23305bf559cSSatish Balay } else { 234ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 23505bf559cSSatish Balay } 23663ba0a88SBarry Smith #endif 2372fb3ed76SBarry Smith 238898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 239898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2401987afe7SBarry Smith 2411393bc97SHong Zhang /* destroy list of free space and other temporary array(s) */ 2421393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 243a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); 2441393bc97SHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 24535baf27eSSatish Balay ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 246289bc588SBarry Smith 247289bc588SBarry Smith /* put together the new matrix */ 248719d5645SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 249719d5645SBarry Smith ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 250719d5645SBarry Smith b = (Mat_SeqAIJ*)(B)->data; 251e6b907acSBarry Smith b->free_a = PETSC_TRUE; 252e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 2537c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 2541393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 2551393bc97SHong Zhang b->j = bj; 2561393bc97SHong Zhang b->i = bi; 2571393bc97SHong Zhang b->diag = bdiag; 258416022c9SBarry Smith b->ilen = 0; 259416022c9SBarry Smith b->imax = 0; 260416022c9SBarry Smith b->row = isrow; 261416022c9SBarry Smith b->col = iscol; 262c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 263c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 26482bf6240SBarry Smith b->icol = isicol; 26587828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2661393bc97SHong Zhang 2671393bc97SHong Zhang /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 268719d5645SBarry Smith ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 2691393bc97SHong Zhang b->maxnz = b->nz = bi[n] ; 2707fd98bd6SLois Curfman McInnes 271719d5645SBarry Smith (B)->factor = MAT_FACTOR_LU; 272719d5645SBarry Smith (B)->info.factor_mallocs = reallocs; 273719d5645SBarry Smith (B)->info.fill_ratio_given = f; 274703deb49SSatish Balay 2759f7d0b68SHong Zhang if (ai[n]) { 276719d5645SBarry Smith (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 277eea2913aSSatish Balay } else { 278719d5645SBarry Smith (B)->info.fill_ratio_needed = 0.0; 279eea2913aSSatish Balay } 280719d5645SBarry Smith (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 281719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqAIJ; 282719d5645SBarry Smith (B)->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 283db4efbfdSBarry Smith /* switch to inodes if appropriate */ 284719d5645SBarry Smith ierr = MatLUFactorSymbolic_Inode(B,A,isrow,iscol,info);CHKERRQ(ierr); 2853a40ed3dSBarry Smith PetscFunctionReturn(0); 286289bc588SBarry Smith } 2871393bc97SHong Zhang 288ce3d78c0SShri Abhyankar #undef __FUNCT__ 289ce3d78c0SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_newdatastruct" 290ce3d78c0SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ_newdatastruct(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 291ce3d78c0SShri Abhyankar { 292ce3d78c0SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 293ce3d78c0SShri Abhyankar IS isicol; 294ce3d78c0SShri Abhyankar PetscErrorCode ierr; 295ce3d78c0SShri Abhyankar const PetscInt *r,*ic; 296ce3d78c0SShri Abhyankar PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 297ce3d78c0SShri Abhyankar PetscInt *bi,*bj,*ajtmp; 298ce3d78c0SShri Abhyankar PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 299ce3d78c0SShri Abhyankar PetscReal f; 300ce3d78c0SShri Abhyankar PetscInt nlnk,*lnk,k,**bi_ptr; 301ce3d78c0SShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 302ce3d78c0SShri Abhyankar PetscBT lnkbt; 303ce3d78c0SShri Abhyankar 304ce3d78c0SShri Abhyankar PetscFunctionBegin; 305ce3d78c0SShri Abhyankar if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square"); 306ce3d78c0SShri Abhyankar ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 307ce3d78c0SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 308ce3d78c0SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 309ce3d78c0SShri Abhyankar 310ce3d78c0SShri Abhyankar /* get new row pointers */ 311ce3d78c0SShri Abhyankar ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 312ce3d78c0SShri Abhyankar bi[0] = 0; 313ce3d78c0SShri Abhyankar 314ce3d78c0SShri Abhyankar /* bdiag is location of diagonal in factor */ 315ce3d78c0SShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 316ce3d78c0SShri Abhyankar bdiag[0] = 0; 317ce3d78c0SShri Abhyankar 318ce3d78c0SShri Abhyankar /* linked list for storing column indices of the active row */ 319ce3d78c0SShri Abhyankar nlnk = n + 1; 320ce3d78c0SShri Abhyankar ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 321ce3d78c0SShri Abhyankar 322ce3d78c0SShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 323ce3d78c0SShri Abhyankar 324ce3d78c0SShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 325ce3d78c0SShri Abhyankar f = info->fill; 326ce3d78c0SShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 327ce3d78c0SShri Abhyankar current_space = free_space; 328ce3d78c0SShri Abhyankar 329ce3d78c0SShri Abhyankar for (i=0; i<n; i++) { 330ce3d78c0SShri Abhyankar /* copy previous fill into linked list */ 331ce3d78c0SShri Abhyankar nzi = 0; 332ce3d78c0SShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 333ce3d78c0SShri Abhyankar if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 334ce3d78c0SShri Abhyankar ajtmp = aj + ai[r[i]]; 335ce3d78c0SShri Abhyankar ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 336ce3d78c0SShri Abhyankar nzi += nlnk; 337ce3d78c0SShri Abhyankar 338ce3d78c0SShri Abhyankar /* add pivot rows into linked list */ 339ce3d78c0SShri Abhyankar row = lnk[n]; 340ce3d78c0SShri Abhyankar while (row < i){ 34170319c84SHong Zhang nzbd = bdiag[row] + 1; /* num of entries in the row with column index <= row */ 342ce3d78c0SShri Abhyankar ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 343ce3d78c0SShri Abhyankar ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 344ce3d78c0SShri Abhyankar nzi += nlnk; 345ce3d78c0SShri Abhyankar row = lnk[row]; 346ce3d78c0SShri Abhyankar } 347ce3d78c0SShri Abhyankar bi[i+1] = bi[i] + nzi; 348ce3d78c0SShri Abhyankar im[i] = nzi; 349ce3d78c0SShri Abhyankar 350ce3d78c0SShri Abhyankar /* mark bdiag */ 351ce3d78c0SShri Abhyankar nzbd = 0; 352ce3d78c0SShri Abhyankar nnz = nzi; 353ce3d78c0SShri Abhyankar k = lnk[n]; 354ce3d78c0SShri Abhyankar while (nnz-- && k < i){ 355ce3d78c0SShri Abhyankar nzbd++; 356ce3d78c0SShri Abhyankar k = lnk[k]; 357ce3d78c0SShri Abhyankar } 358783ef271SHong Zhang bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */ 359ce3d78c0SShri Abhyankar 360ce3d78c0SShri Abhyankar /* if free space is not available, make more free space */ 361ce3d78c0SShri Abhyankar if (current_space->local_remaining<nzi) { 362ce3d78c0SShri Abhyankar nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */ 363ce3d78c0SShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 364ce3d78c0SShri Abhyankar reallocs++; 365ce3d78c0SShri Abhyankar } 366ce3d78c0SShri Abhyankar 367ce3d78c0SShri Abhyankar /* copy data into free space, then initialize lnk */ 368ce3d78c0SShri Abhyankar ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 369ce3d78c0SShri Abhyankar bi_ptr[i] = current_space->array; 370ce3d78c0SShri Abhyankar current_space->array += nzi; 371ce3d78c0SShri Abhyankar current_space->local_used += nzi; 372ce3d78c0SShri Abhyankar current_space->local_remaining -= nzi; 373ce3d78c0SShri Abhyankar } 374ce3d78c0SShri Abhyankar #if defined(PETSC_USE_INFO) 375ce3d78c0SShri Abhyankar if (ai[n] != 0) { 376ce3d78c0SShri Abhyankar PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 377ce3d78c0SShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 378ce3d78c0SShri Abhyankar ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 379ce3d78c0SShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 380ce3d78c0SShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 381ce3d78c0SShri Abhyankar } else { 382ce3d78c0SShri Abhyankar ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 383ce3d78c0SShri Abhyankar } 384ce3d78c0SShri Abhyankar #endif 385ce3d78c0SShri Abhyankar 386ce3d78c0SShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 387ce3d78c0SShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 388ce3d78c0SShri Abhyankar 389ce3d78c0SShri Abhyankar /* destroy list of free space and other temporary array(s) */ 390ce3d78c0SShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 391783ef271SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 392ce3d78c0SShri Abhyankar ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 393ce3d78c0SShri Abhyankar ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 394ce3d78c0SShri Abhyankar 395ce3d78c0SShri Abhyankar /* put together the new matrix */ 396ce3d78c0SShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 397ce3d78c0SShri Abhyankar ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 398ce3d78c0SShri Abhyankar b = (Mat_SeqAIJ*)(B)->data; 399ce3d78c0SShri Abhyankar b->free_a = PETSC_TRUE; 400ce3d78c0SShri Abhyankar b->free_ij = PETSC_TRUE; 401ce3d78c0SShri Abhyankar b->singlemalloc = PETSC_FALSE; 402ce3d78c0SShri Abhyankar ierr = PetscMalloc((bi[2*n+1])*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 403ce3d78c0SShri Abhyankar b->j = bj; 404ce3d78c0SShri Abhyankar b->i = bi; 405ce3d78c0SShri Abhyankar b->diag = bdiag; 406ce3d78c0SShri Abhyankar b->ilen = 0; 407ce3d78c0SShri Abhyankar b->imax = 0; 408ce3d78c0SShri Abhyankar b->row = isrow; 409ce3d78c0SShri Abhyankar b->col = iscol; 410ce3d78c0SShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 411ce3d78c0SShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 412ce3d78c0SShri Abhyankar b->icol = isicol; 413ce3d78c0SShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 414ce3d78c0SShri Abhyankar 415ce3d78c0SShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 416ce3d78c0SShri Abhyankar ierr = PetscLogObjectMemory(B,bi[2*n+1]*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 417ce3d78c0SShri Abhyankar b->maxnz = b->nz = bi[2*n+1] ; 418ce3d78c0SShri Abhyankar 419ce3d78c0SShri Abhyankar (B)->factor = MAT_FACTOR_LU; 420ce3d78c0SShri Abhyankar (B)->info.factor_mallocs = reallocs; 421ce3d78c0SShri Abhyankar (B)->info.fill_ratio_given = f; 422ce3d78c0SShri Abhyankar 423ce3d78c0SShri Abhyankar if (ai[n]) { 424894a2336SHong Zhang (B)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]); 425ce3d78c0SShri Abhyankar } else { 426ce3d78c0SShri Abhyankar (B)->info.fill_ratio_needed = 0.0; 427ce3d78c0SShri Abhyankar } 428ce3d78c0SShri Abhyankar (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 429ce3d78c0SShri Abhyankar PetscFunctionReturn(0); 430ce3d78c0SShri Abhyankar } 431ce3d78c0SShri Abhyankar 4325b5bc046SBarry Smith /* 4335b5bc046SBarry Smith Trouble in factorization, should we dump the original matrix? 4345b5bc046SBarry Smith */ 4355b5bc046SBarry Smith #undef __FUNCT__ 4365b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix" 4375b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A) 4385b5bc046SBarry Smith { 4395b5bc046SBarry Smith PetscErrorCode ierr; 44090d69ab7SBarry Smith PetscTruth flg = PETSC_FALSE; 4415b5bc046SBarry Smith 4425b5bc046SBarry Smith PetscFunctionBegin; 44390d69ab7SBarry Smith ierr = PetscOptionsGetTruth(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr); 4445b5bc046SBarry Smith if (flg) { 4455b5bc046SBarry Smith PetscViewer viewer; 4465b5bc046SBarry Smith char filename[PETSC_MAX_PATH_LEN]; 4475b5bc046SBarry Smith 4485b5bc046SBarry Smith ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr); 4497adad957SLisandro Dalcin ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 4505b5bc046SBarry Smith ierr = MatView(A,viewer);CHKERRQ(ierr); 4515b5bc046SBarry Smith ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr); 4525b5bc046SBarry Smith } 4535b5bc046SBarry Smith PetscFunctionReturn(0); 4545b5bc046SBarry Smith } 4555b5bc046SBarry Smith 456db4efbfdSBarry Smith extern PetscErrorCode MatSolve_Inode(Mat,Vec,Vec); 457db4efbfdSBarry Smith 4580a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 459894a2336SHong Zhang extern PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat,Vec,Vec); 460894a2336SHong Zhang extern PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat,Vec,Vec); 461c9c72f8fSHong Zhang 462c9c72f8fSHong Zhang #undef __FUNCT__ 463c9c72f8fSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_newdatastruct" 464c9c72f8fSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info) 465c9c72f8fSHong Zhang { 466c9c72f8fSHong Zhang Mat C=B; 467c9c72f8fSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 468c9c72f8fSHong Zhang IS isrow = b->row,isicol = b->icol; 469c9c72f8fSHong Zhang PetscErrorCode ierr; 470c9c72f8fSHong Zhang const PetscInt *r,*ic,*ics; 471c9c72f8fSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 472c9c72f8fSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzL,row,*bdiag=b->diag,*pj; 473c9c72f8fSHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 474c9c72f8fSHong Zhang PetscTruth row_identity,col_identity; 475c9c72f8fSHong Zhang 476*d4ad06f7SHong Zhang LUShift_Ctx sctx; 477*d4ad06f7SHong Zhang PetscInt *ddiag,newshift; 478*d4ad06f7SHong Zhang PetscReal rs; 479*d4ad06f7SHong Zhang MatScalar d; 480*d4ad06f7SHong Zhang 481c9c72f8fSHong Zhang PetscFunctionBegin; 482*d4ad06f7SHong Zhang /* ZeropivotSetUp(): initialize shift context sctx */ 483*d4ad06f7SHong Zhang sctx.nshift = 0; 484*d4ad06f7SHong Zhang sctx.nshift_max = 0; 485*d4ad06f7SHong Zhang sctx.shift_top = 0.0; 486*d4ad06f7SHong Zhang sctx.shift_lo = 0.0; 487*d4ad06f7SHong Zhang sctx.shift_hi = 0.0; 488*d4ad06f7SHong Zhang sctx.shift_fraction = 0.0; 489*d4ad06f7SHong Zhang sctx.shift_amount = 0.0; 490*d4ad06f7SHong Zhang 491*d4ad06f7SHong Zhang /* if both shift schemes are chosen by user, only use info->shiftpd */ 492*d4ad06f7SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 493*d4ad06f7SHong Zhang ddiag = a->diag; 494*d4ad06f7SHong Zhang sctx.shift_top = info->zeropivot; 495*d4ad06f7SHong Zhang for (i=0; i<n; i++) { 496*d4ad06f7SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 497*d4ad06f7SHong Zhang d = (aa)[ddiag[i]]; 498*d4ad06f7SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 499*d4ad06f7SHong Zhang v = aa+ai[i]; 500*d4ad06f7SHong Zhang nz = ai[i+1] - ai[i]; 501*d4ad06f7SHong Zhang for (j=0; j<nz; j++) 502*d4ad06f7SHong Zhang rs += PetscAbsScalar(v[j]); 503*d4ad06f7SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 504*d4ad06f7SHong Zhang } 505*d4ad06f7SHong Zhang sctx.shift_top *= 1.1; 506*d4ad06f7SHong Zhang sctx.nshift_max = 5; 507*d4ad06f7SHong Zhang sctx.shift_lo = 0.; 508*d4ad06f7SHong Zhang sctx.shift_hi = 1.; 509*d4ad06f7SHong Zhang } 510*d4ad06f7SHong Zhang 511c9c72f8fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 512c9c72f8fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 513c9c72f8fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 514c9c72f8fSHong Zhang ics = ic; 515c9c72f8fSHong Zhang 516*d4ad06f7SHong Zhang do { 517*d4ad06f7SHong Zhang sctx.lushift = PETSC_FALSE; 518c9c72f8fSHong Zhang for (i=0; i<n; i++){ 519c9c72f8fSHong Zhang /* zero rtmp */ 520c9c72f8fSHong Zhang /* L part */ 521c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 522c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 523c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 524c9c72f8fSHong Zhang 525c9c72f8fSHong Zhang /* U part */ 5268fc3a347SHong Zhang nz = bi[2*n-i+1] - bi[2*n-i]; 5278fc3a347SHong Zhang bjtmp = bj + bi[2*n-i]; 528c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 529c9c72f8fSHong Zhang 530c9c72f8fSHong Zhang /* load in initial (unfactored row) */ 531c9c72f8fSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 532c9c72f8fSHong Zhang ajtmp = aj + ai[r[i]]; 533c9c72f8fSHong Zhang v = aa + ai[r[i]]; 534c9c72f8fSHong Zhang for (j=0; j<nz; j++) { 535c9c72f8fSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 536c9c72f8fSHong Zhang } 537*d4ad06f7SHong Zhang /* ZeropivotApply() */ 538*d4ad06f7SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 539c9c72f8fSHong Zhang 540c9c72f8fSHong Zhang /* elimination */ 541c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 542c9c72f8fSHong Zhang row = *bjtmp++; 543c9c72f8fSHong Zhang nzL = bi[i+1] - bi[i]; 544c9c72f8fSHong Zhang k = 0; 545c9c72f8fSHong Zhang while (k < nzL) { 546c9c72f8fSHong Zhang pc = rtmp + row; 547c9c72f8fSHong Zhang if (*pc != 0.0) { 548c9c72f8fSHong Zhang pv = b->a + bdiag[row]; 549c9c72f8fSHong Zhang multiplier = *pc * (*pv); 550c9c72f8fSHong Zhang *pc = multiplier; 5518fc3a347SHong Zhang pj = b->j + bi[2*n-row]; /* begining of U(row,:) */ 5528fc3a347SHong Zhang pv = b->a + bi[2*n-row]; 5538fc3a347SHong Zhang nz = bi[2*n-row+1] - bi[2*n-row] - 1; /* num of entries in U(row,:), excluding diag */ 554c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 555c9c72f8fSHong Zhang ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 556c9c72f8fSHong Zhang } 557c9c72f8fSHong Zhang row = *bjtmp++; k++; 558c9c72f8fSHong Zhang } 559c9c72f8fSHong Zhang 560c9c72f8fSHong Zhang /* finished row so stick it into b->a */ 561*d4ad06f7SHong Zhang rs = 0.0; 562c9c72f8fSHong Zhang /* L part */ 563c9c72f8fSHong Zhang pv = b->a + bi[i] ; 564c9c72f8fSHong Zhang pj = b->j + bi[i] ; 565c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 566c9c72f8fSHong Zhang for (j=0; j<nz; j++) { 567*d4ad06f7SHong Zhang pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 568c9c72f8fSHong Zhang } 569c9c72f8fSHong Zhang 570c9c72f8fSHong Zhang /* U part */ 5718fc3a347SHong Zhang pv = b->a + bi[2*n-i]; 5728fc3a347SHong Zhang pj = b->j + bi[2*n-i]; 5738fc3a347SHong Zhang nz = bi[2*n-i+1] - bi[2*n-i] - 1; 574*d4ad06f7SHong Zhang for (j=0; j<nz; j++) { 575*d4ad06f7SHong Zhang pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 576c9c72f8fSHong Zhang } 577*d4ad06f7SHong Zhang 578*d4ad06f7SHong Zhang /* ZeropivotCheck() */ 579*d4ad06f7SHong Zhang sctx.rs = rs; 580*d4ad06f7SHong Zhang sctx.pv = rtmp[i]; 581*d4ad06f7SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 582*d4ad06f7SHong Zhang if (newshift == 1) break; 583*d4ad06f7SHong Zhang 584*d4ad06f7SHong Zhang /* Mark diagonal and invert diagonal for simplier triangular solves */ 585*d4ad06f7SHong Zhang pv = b->a + bdiag[i]; 586*d4ad06f7SHong Zhang *pv = 1.0/rtmp[i]; 587*d4ad06f7SHong Zhang 588*d4ad06f7SHong Zhang } /* endof for (i=0; i<n; i++){ */ 589*d4ad06f7SHong Zhang 590*d4ad06f7SHong Zhang /* ZeropivotRefine() */ 591*d4ad06f7SHong Zhang if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){ 592*d4ad06f7SHong Zhang /* 593*d4ad06f7SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 594*d4ad06f7SHong Zhang * then try lower shift 595*d4ad06f7SHong Zhang */ 596*d4ad06f7SHong Zhang sctx.shift_hi = sctx.shift_fraction; 597*d4ad06f7SHong Zhang sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 598*d4ad06f7SHong Zhang sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 599*d4ad06f7SHong Zhang sctx.lushift = PETSC_TRUE; 600*d4ad06f7SHong Zhang sctx.nshift++; 601*d4ad06f7SHong Zhang } 602*d4ad06f7SHong Zhang } while (sctx.lushift); 603*d4ad06f7SHong Zhang 604c9c72f8fSHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 605c9c72f8fSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 606c9c72f8fSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 607c9c72f8fSHong Zhang 608c9c72f8fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 609c9c72f8fSHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 610c9c72f8fSHong Zhang if (row_identity && col_identity) { 611894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct; 612c9c72f8fSHong Zhang } else { 613894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_newdatastruct; 614c9c72f8fSHong Zhang } 615c9c72f8fSHong Zhang 616c9c72f8fSHong Zhang C->ops->solveadd = 0; 617c9c72f8fSHong Zhang C->ops->solvetranspose = 0; 618c9c72f8fSHong Zhang C->ops->solvetransposeadd = 0; 619c9c72f8fSHong Zhang C->ops->matsolve = 0; 620c9c72f8fSHong Zhang C->assembled = PETSC_TRUE; 621c9c72f8fSHong Zhang C->preallocated = PETSC_TRUE; 622c9c72f8fSHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 623*d4ad06f7SHong Zhang 624*d4ad06f7SHong Zhang /* ZeropivotView() */ 625*d4ad06f7SHong Zhang if (sctx.nshift){ 626*d4ad06f7SHong Zhang if (info->shiftpd) { 627*d4ad06f7SHong Zhang 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,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr); 628*d4ad06f7SHong Zhang } else if (info->shiftnz) { 629*d4ad06f7SHong Zhang ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 630*d4ad06f7SHong Zhang } 631*d4ad06f7SHong Zhang } 632c9c72f8fSHong Zhang PetscFunctionReturn(0); 633c9c72f8fSHong Zhang } 634c9c72f8fSHong Zhang 6354a2ae208SSatish Balay #undef __FUNCT__ 6364a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ" 6370481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 638289bc588SBarry Smith { 639719d5645SBarry Smith Mat C=B; 640416022c9SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 64182bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 6426849ba73SBarry Smith PetscErrorCode ierr; 6435d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 644d278edc7SBarry Smith PetscInt nz,row,i,j,n=A->rmap->n,diag; 645d278edc7SBarry Smith const PetscInt *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 646d278edc7SBarry Smith const PetscInt *ajtmp,*bjtmp,*diag_offset = b->diag,*pj; 647d278edc7SBarry Smith MatScalar *pv,*rtmp,*pc,multiplier,d; 648d278edc7SBarry Smith const MatScalar *v,*aa=a->a; 64933f9893dSHong Zhang PetscReal rs=0.0; 650b3bf805bSHong Zhang LUShift_Ctx sctx; 65142898933SHong Zhang PetscInt newshift,*ddiag; 652289bc588SBarry Smith 6533a40ed3dSBarry Smith PetscFunctionBegin; 65478b31e54SBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 65578b31e54SBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 656b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 6579f7d0b68SHong Zhang ics = ic; 658289bc588SBarry Smith 659*d4ad06f7SHong Zhang /* initialize shift context sctx */ 660*d4ad06f7SHong Zhang sctx.nshift = 0; 661ada7143aSSatish Balay sctx.nshift_max = 0; 662*d4ad06f7SHong Zhang sctx.shift_top = 0.0; 663*d4ad06f7SHong Zhang sctx.shift_lo = 0.0; 664*d4ad06f7SHong Zhang sctx.shift_hi = 0.0; 665*d4ad06f7SHong Zhang sctx.shift_fraction = 0.0; 666*d4ad06f7SHong Zhang sctx.shift_amount = 0.0; 667ada7143aSSatish Balay 668118f5789SBarry Smith /* if both shift schemes are chosen by user, only use info->shiftpd */ 6691a890ab1SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 67042898933SHong Zhang ddiag = a->diag; 6719f7d0b68SHong Zhang sctx.shift_top = info->zeropivot; 6726cc28720Svictorle for (i=0; i<n; i++) { 6739f95998fSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 6749f7d0b68SHong Zhang d = (aa)[ddiag[i]]; 6751a890ab1SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 6769f7d0b68SHong Zhang v = aa+ai[i]; 6779f7d0b68SHong Zhang nz = ai[i+1] - ai[i]; 678e24cfe64SHong Zhang for (j=0; j<nz; j++) 6791a890ab1SHong Zhang rs += PetscAbsScalar(v[j]); 6801a890ab1SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 6816cc28720Svictorle } 682118f5789SBarry Smith sctx.shift_top *= 1.1; 683d2276718SHong Zhang sctx.nshift_max = 5; 684d2276718SHong Zhang sctx.shift_lo = 0.; 685d2276718SHong Zhang sctx.shift_hi = 1.; 686a0a356efSHong Zhang } 687a0a356efSHong Zhang 688085256b3SBarry Smith do { 689d2276718SHong Zhang sctx.lushift = PETSC_FALSE; 690289bc588SBarry Smith for (i=0; i<n; i++){ 691b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 692b3bf805bSHong Zhang bjtmp = bj + bi[i]; 6939f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 694289bc588SBarry Smith 695289bc588SBarry Smith /* load in initial (unfactored row) */ 6969f7d0b68SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 6979f7d0b68SHong Zhang ajtmp = aj + ai[r[i]]; 6989f7d0b68SHong Zhang v = aa + ai[r[i]]; 699085256b3SBarry Smith for (j=0; j<nz; j++) { 700b3bf805bSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 701085256b3SBarry Smith } 702d2276718SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 703bd1bc851SBarry Smith /* if (sctx.shift_amount > 0.0) printf("row %d, shift %g\n",i,sctx.shift_amount); */ 704289bc588SBarry Smith 705b3bf805bSHong Zhang row = *bjtmp++; 706f2582111SSatish Balay while (row < i) { 7078c37ef55SBarry Smith pc = rtmp + row; 7088c37ef55SBarry Smith if (*pc != 0.0) { 709010a20caSHong Zhang pv = b->a + diag_offset[row]; 710010a20caSHong Zhang pj = b->j + diag_offset[row] + 1; 71135aab85fSBarry Smith multiplier = *pc / *pv++; 7128c37ef55SBarry Smith *pc = multiplier; 713b3bf805bSHong Zhang nz = bi[row+1] - diag_offset[row] - 1; 7149f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 715dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 716289bc588SBarry Smith } 717b3bf805bSHong Zhang row = *bjtmp++; 718289bc588SBarry Smith } 719416022c9SBarry Smith /* finished row so stick it into b->a */ 720b3bf805bSHong Zhang pv = b->a + bi[i] ; 721b3bf805bSHong Zhang pj = b->j + bi[i] ; 722b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 723b3bf805bSHong Zhang diag = diag_offset[i] - bi[i]; 724e57ff13aSBarry Smith rs = 0.0; 725d3d32019SBarry Smith for (j=0; j<nz; j++) { 7269f7d0b68SHong Zhang pv[j] = rtmp[pj[j]]; 7279f7d0b68SHong Zhang rs += PetscAbsScalar(pv[j]); 728d3d32019SBarry Smith } 729e57ff13aSBarry Smith rs -= PetscAbsScalar(pv[diag]); 730d2276718SHong Zhang 731b3bf805bSHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 732d2276718SHong Zhang sctx.rs = rs; 733d2276718SHong Zhang sctx.pv = pv[diag]; 734c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 7355b5bc046SBarry Smith if (newshift == 1) break; 73635aab85fSBarry Smith } 737d2276718SHong Zhang 7380481f469SBarry Smith if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 7396cc28720Svictorle /* 7406c037d1bSvictorle * if no shift in this attempt & shifting & started shifting & can refine, 7416cc28720Svictorle * then try lower shift 7426cc28720Svictorle */ 7430481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 7440481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 7450481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 746d2276718SHong Zhang sctx.lushift = PETSC_TRUE; 747d2276718SHong Zhang sctx.nshift++; 7486cc28720Svictorle } 749d2276718SHong Zhang } while (sctx.lushift); 7500f11f9aeSSatish Balay 75135aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 75235aab85fSBarry Smith for (i=0; i<n; i++) { 753010a20caSHong Zhang b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]]; 75435aab85fSBarry Smith } 755606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 75678b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 75778b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 758db4efbfdSBarry Smith if (b->inode.use) { 759db4efbfdSBarry Smith C->ops->solve = MatSolve_Inode; 760db4efbfdSBarry Smith } else { 7618d8c7c43SBarry Smith PetscTruth row_identity, col_identity; 7628d8c7c43SBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 7638d8c7c43SBarry Smith ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 7648d8c7c43SBarry Smith if (row_identity && col_identity) { 7658d8c7c43SBarry Smith C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 7668d8c7c43SBarry Smith } else { 767db4efbfdSBarry Smith C->ops->solve = MatSolve_SeqAIJ; 768db4efbfdSBarry Smith } 7698d8c7c43SBarry Smith } 770db4efbfdSBarry Smith C->ops->solveadd = MatSolveAdd_SeqAIJ; 771db4efbfdSBarry Smith C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 772db4efbfdSBarry Smith C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 773de26497bSBarry Smith C->ops->matsolve = MatMatSolve_SeqAIJ; 774c456f294SBarry Smith C->assembled = PETSC_TRUE; 7755c9eb25fSBarry Smith C->preallocated = PETSC_TRUE; 776d0f46423SBarry Smith ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 777d2276718SHong Zhang if (sctx.nshift){ 778e738e422SBarry Smith if (info->shiftpd) { 7790481f469SBarry 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,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr); 780e738e422SBarry Smith } else if (info->shiftnz) { 781e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 7826cc28720Svictorle } 7830697b6caSHong Zhang } 7843a40ed3dSBarry Smith PetscFunctionReturn(0); 785289bc588SBarry Smith } 7860889b5dcSvictorle 787137fb511SHong Zhang /* 788137fb511SHong Zhang This routine implements inplace ILU(0) with row or/and column permutations. 789137fb511SHong Zhang Input: 790137fb511SHong Zhang A - original matrix 791137fb511SHong Zhang Output; 792137fb511SHong Zhang A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i] 793137fb511SHong Zhang a->j (col index) is permuted by the inverse of colperm, then sorted 794137fb511SHong Zhang a->a reordered accordingly with a->j 795137fb511SHong Zhang a->diag (ptr to diagonal elements) is updated. 796137fb511SHong Zhang */ 797137fb511SHong Zhang #undef __FUNCT__ 798137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm" 7990481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info) 800137fb511SHong Zhang { 801b051339dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 802b051339dSHong Zhang IS isrow = a->row,isicol = a->icol; 803137fb511SHong Zhang PetscErrorCode ierr; 8045d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 8055d0c19d7SBarry Smith PetscInt i,j,n=A->rmap->n,*ai=a->i,*aj=a->j; 8065d0c19d7SBarry Smith PetscInt *ajtmp,nz,row; 807b051339dSHong Zhang PetscInt *diag = a->diag,nbdiag,*pj; 808a77337e4SBarry Smith PetscScalar *rtmp,*pc,multiplier,d; 809a77337e4SBarry Smith MatScalar *v,*pv; 810137fb511SHong Zhang PetscReal rs; 811137fb511SHong Zhang LUShift_Ctx sctx; 812b051339dSHong Zhang PetscInt newshift; 813137fb511SHong Zhang 814137fb511SHong Zhang PetscFunctionBegin; 815719d5645SBarry Smith if (A != B) SETERRQ(PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address"); 816137fb511SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 817137fb511SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 818137fb511SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr); 819137fb511SHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 820b051339dSHong Zhang ics = ic; 821137fb511SHong Zhang 822137fb511SHong Zhang sctx.shift_top = 0; 823137fb511SHong Zhang sctx.nshift_max = 0; 824137fb511SHong Zhang sctx.shift_lo = 0; 825137fb511SHong Zhang sctx.shift_hi = 0; 8260481f469SBarry Smith sctx.shift_fraction = 0; 827137fb511SHong Zhang 828137fb511SHong Zhang /* if both shift schemes are chosen by user, only use info->shiftpd */ 829137fb511SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 830137fb511SHong Zhang sctx.shift_top = 0; 831137fb511SHong Zhang for (i=0; i<n; i++) { 832137fb511SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 833b051339dSHong Zhang d = (a->a)[diag[i]]; 834137fb511SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 835b051339dSHong Zhang v = a->a+ai[i]; 836b051339dSHong Zhang nz = ai[i+1] - ai[i]; 837137fb511SHong Zhang for (j=0; j<nz; j++) 838137fb511SHong Zhang rs += PetscAbsScalar(v[j]); 839137fb511SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 840137fb511SHong Zhang } 841137fb511SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 842137fb511SHong Zhang sctx.shift_top *= 1.1; 843137fb511SHong Zhang sctx.nshift_max = 5; 844137fb511SHong Zhang sctx.shift_lo = 0.; 845137fb511SHong Zhang sctx.shift_hi = 1.; 846137fb511SHong Zhang } 847137fb511SHong Zhang 848137fb511SHong Zhang sctx.shift_amount = 0; 849137fb511SHong Zhang sctx.nshift = 0; 850137fb511SHong Zhang do { 851137fb511SHong Zhang sctx.lushift = PETSC_FALSE; 852137fb511SHong Zhang for (i=0; i<n; i++){ 853b051339dSHong Zhang /* load in initial unfactored row */ 854b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 855b051339dSHong Zhang ajtmp = aj + ai[r[i]]; 856b051339dSHong Zhang v = a->a + ai[r[i]]; 857137fb511SHong Zhang /* sort permuted ajtmp and values v accordingly */ 858b051339dSHong Zhang for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]]; 859137fb511SHong Zhang ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr); 860137fb511SHong Zhang 861b051339dSHong Zhang diag[r[i]] = ai[r[i]]; 862137fb511SHong Zhang for (j=0; j<nz; j++) { 863137fb511SHong Zhang rtmp[ajtmp[j]] = v[j]; 864b051339dSHong Zhang if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */ 865137fb511SHong Zhang } 866137fb511SHong Zhang rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 867137fb511SHong Zhang 868b051339dSHong Zhang row = *ajtmp++; 869137fb511SHong Zhang while (row < i) { 870137fb511SHong Zhang pc = rtmp + row; 871137fb511SHong Zhang if (*pc != 0.0) { 872b051339dSHong Zhang pv = a->a + diag[r[row]]; 873b051339dSHong Zhang pj = aj + diag[r[row]] + 1; 874137fb511SHong Zhang 875137fb511SHong Zhang multiplier = *pc / *pv++; 876137fb511SHong Zhang *pc = multiplier; 877b051339dSHong Zhang nz = ai[r[row]+1] - diag[r[row]] - 1; 878b051339dSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 879dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 880137fb511SHong Zhang } 881b051339dSHong Zhang row = *ajtmp++; 882137fb511SHong Zhang } 883b051339dSHong Zhang /* finished row so overwrite it onto a->a */ 884b051339dSHong Zhang pv = a->a + ai[r[i]] ; 885b051339dSHong Zhang pj = aj + ai[r[i]] ; 886b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 887b051339dSHong Zhang nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */ 888137fb511SHong Zhang 889137fb511SHong Zhang rs = 0.0; 890137fb511SHong Zhang for (j=0; j<nz; j++) { 891b051339dSHong Zhang pv[j] = rtmp[pj[j]]; 892b051339dSHong Zhang if (j != nbdiag) rs += PetscAbsScalar(pv[j]); 893137fb511SHong Zhang } 894137fb511SHong Zhang 895137fb511SHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 896137fb511SHong Zhang sctx.rs = rs; 897b051339dSHong Zhang sctx.pv = pv[nbdiag]; 898c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 899137fb511SHong Zhang if (newshift == 1) break; 900137fb511SHong Zhang } 901137fb511SHong Zhang 9020481f469SBarry Smith if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 903137fb511SHong Zhang /* 904137fb511SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 905137fb511SHong Zhang * then try lower shift 906137fb511SHong Zhang */ 9070481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 9080481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 9090481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 910137fb511SHong Zhang sctx.lushift = PETSC_TRUE; 911137fb511SHong Zhang sctx.nshift++; 912137fb511SHong Zhang } 913137fb511SHong Zhang } while (sctx.lushift); 914137fb511SHong Zhang 915137fb511SHong Zhang /* invert diagonal entries for simplier triangular solves */ 916137fb511SHong Zhang for (i=0; i<n; i++) { 917b051339dSHong Zhang a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]]; 918137fb511SHong Zhang } 919137fb511SHong Zhang 920137fb511SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 921137fb511SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 922137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 923b051339dSHong Zhang A->ops->solve = MatSolve_SeqAIJ_InplaceWithPerm; 924db4efbfdSBarry Smith A->ops->solveadd = MatSolveAdd_SeqAIJ; 925db4efbfdSBarry Smith A->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 926db4efbfdSBarry Smith A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 927b051339dSHong Zhang A->assembled = PETSC_TRUE; 9285c9eb25fSBarry Smith A->preallocated = PETSC_TRUE; 929d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr); 930137fb511SHong Zhang if (sctx.nshift){ 931e738e422SBarry Smith if (info->shiftpd) { 9320481f469SBarry 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,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr); 933e738e422SBarry Smith } else if (info->shiftnz) { 934e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 935137fb511SHong Zhang } 936137fb511SHong Zhang } 937137fb511SHong Zhang PetscFunctionReturn(0); 938137fb511SHong Zhang } 939137fb511SHong Zhang 9400a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9414a2ae208SSatish Balay #undef __FUNCT__ 9424a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ" 9430481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info) 944da3a660dSBarry Smith { 945dfbe8321SBarry Smith PetscErrorCode ierr; 946416022c9SBarry Smith Mat C; 947416022c9SBarry Smith 9483a40ed3dSBarry Smith PetscFunctionBegin; 94943244d56SBarry Smith ierr = MatGetFactor(A,MAT_SOLVER_PETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr); 950719d5645SBarry Smith ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr); 951719d5645SBarry Smith ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr); 952db4efbfdSBarry Smith A->ops->solve = C->ops->solve; 953db4efbfdSBarry Smith A->ops->solvetranspose = C->ops->solvetranspose; 954273d9f13SBarry Smith ierr = MatHeaderCopy(A,C);CHKERRQ(ierr); 95552e6d16bSBarry Smith ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr); 9563a40ed3dSBarry Smith PetscFunctionReturn(0); 957da3a660dSBarry Smith } 9580a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9591d098868SHong Zhang 960889a8dedSBarry Smith 9614a2ae208SSatish Balay #undef __FUNCT__ 9624a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ" 963dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 9648c37ef55SBarry Smith { 965416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 966416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 9676849ba73SBarry Smith PetscErrorCode ierr; 9685d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 9695d0c19d7SBarry Smith PetscInt nz; 9705d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 971dd6ea824SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 972d9fead3dSBarry Smith const PetscScalar *b; 973dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 9748c37ef55SBarry Smith 9753a40ed3dSBarry Smith PetscFunctionBegin; 9763a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 97791bf9adeSBarry Smith 978d9fead3dSBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 9791ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 980416022c9SBarry Smith tmp = a->solve_work; 9818c37ef55SBarry Smith 9823b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 9833b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 9848c37ef55SBarry Smith 9858c37ef55SBarry Smith /* forward solve the lower triangular */ 9868c37ef55SBarry Smith tmp[0] = b[*r++]; 987010a20caSHong Zhang tmps = tmp; 9888c37ef55SBarry Smith for (i=1; i<n; i++) { 989010a20caSHong Zhang v = aa + ai[i] ; 990010a20caSHong Zhang vi = aj + ai[i] ; 99153e7425aSBarry Smith nz = a->diag[i] - ai[i]; 99253e7425aSBarry Smith sum = b[*r++]; 993003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 9948c37ef55SBarry Smith tmp[i] = sum; 9958c37ef55SBarry Smith } 9968c37ef55SBarry Smith 9978c37ef55SBarry Smith /* backward solve the upper triangular */ 9988c37ef55SBarry Smith for (i=n-1; i>=0; i--){ 999010a20caSHong Zhang v = aa + a->diag[i] + 1; 1000010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1001416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 10028c37ef55SBarry Smith sum = tmp[i]; 1003003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1004010a20caSHong Zhang x[*c--] = tmp[i] = sum*aa[a->diag[i]]; 10058c37ef55SBarry Smith } 10068c37ef55SBarry Smith 10073b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10083b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1009d9fead3dSBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 10101ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1011dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 10123a40ed3dSBarry Smith PetscFunctionReturn(0); 10138c37ef55SBarry Smith } 1014026e39d0SSatish Balay 10152bebee5dSHong Zhang #undef __FUNCT__ 10162bebee5dSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ" 10172bebee5dSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X) 10182bebee5dSHong Zhang { 10192bebee5dSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10202bebee5dSHong Zhang IS iscol = a->col,isrow = a->row; 10212bebee5dSHong Zhang PetscErrorCode ierr; 10225d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10235d0c19d7SBarry Smith PetscInt nz,neq; 10245d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 1025dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1026dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1027db4efbfdSBarry Smith PetscTruth bisdense,xisdense; 10282bebee5dSHong Zhang 10292bebee5dSHong Zhang PetscFunctionBegin; 10302bebee5dSHong Zhang if (!n) PetscFunctionReturn(0); 10312bebee5dSHong Zhang 1032db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1033db4efbfdSBarry Smith if (!bisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 1034db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1035db4efbfdSBarry Smith if (!xisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 1036db4efbfdSBarry Smith 10372bebee5dSHong Zhang ierr = MatGetArray(B,&b);CHKERRQ(ierr); 10382bebee5dSHong Zhang ierr = MatGetArray(X,&x);CHKERRQ(ierr); 10392bebee5dSHong Zhang 10402bebee5dSHong Zhang tmp = a->solve_work; 10412bebee5dSHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10422bebee5dSHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 10432bebee5dSHong Zhang 1044a36b22ccSSatish Balay for (neq=0; neq<B->cmap->n; neq++){ 10452bebee5dSHong Zhang /* forward solve the lower triangular */ 10462bebee5dSHong Zhang tmp[0] = b[r[0]]; 10472bebee5dSHong Zhang tmps = tmp; 10482bebee5dSHong Zhang for (i=1; i<n; i++) { 10492bebee5dSHong Zhang v = aa + ai[i] ; 10502bebee5dSHong Zhang vi = aj + ai[i] ; 10512bebee5dSHong Zhang nz = a->diag[i] - ai[i]; 10522bebee5dSHong Zhang sum = b[r[i]]; 1053003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10542bebee5dSHong Zhang tmp[i] = sum; 10552bebee5dSHong Zhang } 10562bebee5dSHong Zhang /* backward solve the upper triangular */ 10572bebee5dSHong Zhang for (i=n-1; i>=0; i--){ 10582bebee5dSHong Zhang v = aa + a->diag[i] + 1; 10592bebee5dSHong Zhang vi = aj + a->diag[i] + 1; 10602bebee5dSHong Zhang nz = ai[i+1] - a->diag[i] - 1; 10612bebee5dSHong Zhang sum = tmp[i]; 1062003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10632bebee5dSHong Zhang x[c[i]] = tmp[i] = sum*aa[a->diag[i]]; 10642bebee5dSHong Zhang } 10652bebee5dSHong Zhang 10662bebee5dSHong Zhang b += n; 10672bebee5dSHong Zhang x += n; 10682bebee5dSHong Zhang } 10692bebee5dSHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10702bebee5dSHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10712bebee5dSHong Zhang ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 10722bebee5dSHong Zhang ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 1073dc0b31edSSatish Balay ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 10742bebee5dSHong Zhang PetscFunctionReturn(0); 10752bebee5dSHong Zhang } 10762bebee5dSHong Zhang 1077137fb511SHong Zhang #undef __FUNCT__ 1078137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm" 1079137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx) 1080137fb511SHong Zhang { 1081137fb511SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1082137fb511SHong Zhang IS iscol = a->col,isrow = a->row; 1083137fb511SHong Zhang PetscErrorCode ierr; 10845d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 10855d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10865d0c19d7SBarry Smith PetscInt nz,row; 1087dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1088dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1089137fb511SHong Zhang 1090137fb511SHong Zhang PetscFunctionBegin; 1091137fb511SHong Zhang if (!n) PetscFunctionReturn(0); 1092137fb511SHong Zhang 1093137fb511SHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 1094137fb511SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1095137fb511SHong Zhang tmp = a->solve_work; 1096137fb511SHong Zhang 1097137fb511SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1098137fb511SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1099137fb511SHong Zhang 1100137fb511SHong Zhang /* forward solve the lower triangular */ 1101137fb511SHong Zhang tmp[0] = b[*r++]; 1102137fb511SHong Zhang tmps = tmp; 1103137fb511SHong Zhang for (row=1; row<n; row++) { 1104137fb511SHong Zhang i = rout[row]; /* permuted row */ 1105137fb511SHong Zhang v = aa + ai[i] ; 1106137fb511SHong Zhang vi = aj + ai[i] ; 1107137fb511SHong Zhang nz = a->diag[i] - ai[i]; 1108137fb511SHong Zhang sum = b[*r++]; 1109003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1110137fb511SHong Zhang tmp[row] = sum; 1111137fb511SHong Zhang } 1112137fb511SHong Zhang 1113137fb511SHong Zhang /* backward solve the upper triangular */ 1114137fb511SHong Zhang for (row=n-1; row>=0; row--){ 1115137fb511SHong Zhang i = rout[row]; /* permuted row */ 1116137fb511SHong Zhang v = aa + a->diag[i] + 1; 1117137fb511SHong Zhang vi = aj + a->diag[i] + 1; 1118137fb511SHong Zhang nz = ai[i+1] - a->diag[i] - 1; 1119137fb511SHong Zhang sum = tmp[row]; 1120003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1121137fb511SHong Zhang x[*c--] = tmp[row] = sum*aa[a->diag[i]]; 1122137fb511SHong Zhang } 1123137fb511SHong Zhang 1124137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1125137fb511SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1126137fb511SHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 1127137fb511SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1128dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1129137fb511SHong Zhang PetscFunctionReturn(0); 1130137fb511SHong Zhang } 1131137fb511SHong Zhang 1132930ae53cSBarry Smith /* ----------------------------------------------------------- */ 1133a4005a5dSBarry Smith #include "../src/mat/impls/aij/seq/ftn-kernels/fsolve.h" 11344a2ae208SSatish Balay #undef __FUNCT__ 11354a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering" 1136dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 1137930ae53cSBarry Smith { 1138930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 11396849ba73SBarry Smith PetscErrorCode ierr; 1140d0f46423SBarry Smith PetscInt n = A->rmap->n; 1141b377110cSBarry Smith const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag; 1142356650c2SBarry Smith PetscScalar *x; 1143356650c2SBarry Smith const PetscScalar *b; 1144dd6ea824SBarry Smith const MatScalar *aa = a->a; 1145aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 1146356650c2SBarry Smith PetscInt adiag_i,i,nz,ai_i; 1147b377110cSBarry Smith const PetscInt *vi; 1148dd6ea824SBarry Smith const MatScalar *v; 1149dd6ea824SBarry Smith PetscScalar sum; 1150d85afc42SSatish Balay #endif 1151930ae53cSBarry Smith 11523a40ed3dSBarry Smith PetscFunctionBegin; 11533a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 1154930ae53cSBarry Smith 1155356650c2SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 11561ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1157930ae53cSBarry Smith 1158aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 11591c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 11601c4feecaSSatish Balay #else 1161930ae53cSBarry Smith /* forward solve the lower triangular */ 1162930ae53cSBarry Smith x[0] = b[0]; 1163930ae53cSBarry Smith for (i=1; i<n; i++) { 1164930ae53cSBarry Smith ai_i = ai[i]; 1165930ae53cSBarry Smith v = aa + ai_i; 1166930ae53cSBarry Smith vi = aj + ai_i; 1167930ae53cSBarry Smith nz = adiag[i] - ai_i; 1168930ae53cSBarry Smith sum = b[i]; 1169003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1170930ae53cSBarry Smith x[i] = sum; 1171930ae53cSBarry Smith } 1172930ae53cSBarry Smith 1173930ae53cSBarry Smith /* backward solve the upper triangular */ 1174930ae53cSBarry Smith for (i=n-1; i>=0; i--){ 1175930ae53cSBarry Smith adiag_i = adiag[i]; 1176d708749eSBarry Smith v = aa + adiag_i + 1; 1177d708749eSBarry Smith vi = aj + adiag_i + 1; 1178930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 1179930ae53cSBarry Smith sum = x[i]; 1180003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1181930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 1182930ae53cSBarry Smith } 11831c4feecaSSatish Balay #endif 1184dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1185356650c2SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 11861ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 11873a40ed3dSBarry Smith PetscFunctionReturn(0); 1188930ae53cSBarry Smith } 1189930ae53cSBarry Smith 11904a2ae208SSatish Balay #undef __FUNCT__ 11914a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqAIJ" 1192dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 1193da3a660dSBarry Smith { 1194416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1195416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 11966849ba73SBarry Smith PetscErrorCode ierr; 119704fbf559SBarry Smith PetscInt i, n = A->rmap->n,j; 11985d0c19d7SBarry Smith PetscInt nz; 119904fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j; 120004fbf559SBarry Smith PetscScalar *x,*tmp,sum; 120104fbf559SBarry Smith const PetscScalar *b; 1202dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1203da3a660dSBarry Smith 12043a40ed3dSBarry Smith PetscFunctionBegin; 120578b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 1206da3a660dSBarry Smith 120704fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12081ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1209416022c9SBarry Smith tmp = a->solve_work; 1210da3a660dSBarry Smith 12113b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12123b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1213da3a660dSBarry Smith 1214da3a660dSBarry Smith /* forward solve the lower triangular */ 1215da3a660dSBarry Smith tmp[0] = b[*r++]; 1216da3a660dSBarry Smith for (i=1; i<n; i++) { 1217010a20caSHong Zhang v = aa + ai[i] ; 1218010a20caSHong Zhang vi = aj + ai[i] ; 1219416022c9SBarry Smith nz = a->diag[i] - ai[i]; 1220da3a660dSBarry Smith sum = b[*r++]; 122104fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1222da3a660dSBarry Smith tmp[i] = sum; 1223da3a660dSBarry Smith } 1224da3a660dSBarry Smith 1225da3a660dSBarry Smith /* backward solve the upper triangular */ 1226da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1227010a20caSHong Zhang v = aa + a->diag[i] + 1; 1228010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1229416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 1230da3a660dSBarry Smith sum = tmp[i]; 123104fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1232010a20caSHong Zhang tmp[i] = sum*aa[a->diag[i]]; 1233da3a660dSBarry Smith x[*c--] += tmp[i]; 1234da3a660dSBarry Smith } 1235da3a660dSBarry Smith 12363b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 12373b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 123804fbf559SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12391ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1240dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1241898183ecSLois Curfman McInnes 12423a40ed3dSBarry Smith PetscFunctionReturn(0); 1243da3a660dSBarry Smith } 124404fbf559SBarry Smith 12454a2ae208SSatish Balay #undef __FUNCT__ 12464a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqAIJ" 1247dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 1248da3a660dSBarry Smith { 1249416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 12502235e667SBarry Smith IS iscol = a->col,isrow = a->row; 12516849ba73SBarry Smith PetscErrorCode ierr; 125204fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 125304fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 125404fbf559SBarry Smith PetscInt nz; 125504fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1256dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 125704fbf559SBarry Smith const PetscScalar *b; 1258da3a660dSBarry Smith 12593a40ed3dSBarry Smith PetscFunctionBegin; 126004fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12611ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1262416022c9SBarry Smith tmp = a->solve_work; 1263da3a660dSBarry Smith 12642235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12652235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1266da3a660dSBarry Smith 1267da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 12682235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1269da3a660dSBarry Smith 1270da3a660dSBarry Smith /* forward solve the U^T */ 1271da3a660dSBarry Smith for (i=0; i<n; i++) { 1272010a20caSHong Zhang v = aa + diag[i] ; 1273010a20caSHong Zhang vi = aj + diag[i] + 1; 1274f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 1275c38d4ed2SBarry Smith s1 = tmp[i]; 1276ef66eb69SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 127704fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1278c38d4ed2SBarry Smith tmp[i] = s1; 1279da3a660dSBarry Smith } 1280da3a660dSBarry Smith 1281da3a660dSBarry Smith /* backward solve the L^T */ 1282da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1283010a20caSHong Zhang v = aa + diag[i] - 1 ; 1284010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1285f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 1286f1af5d2fSBarry Smith s1 = tmp[i]; 128704fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 1288da3a660dSBarry Smith } 1289da3a660dSBarry Smith 1290da3a660dSBarry Smith /* copy tmp into x according to permutation */ 1291591294e4SBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1292da3a660dSBarry Smith 12932235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 12942235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 129504fbf559SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12961ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1297da3a660dSBarry Smith 1298dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 12993a40ed3dSBarry Smith PetscFunctionReturn(0); 1300da3a660dSBarry Smith } 1301da3a660dSBarry Smith 13024a2ae208SSatish Balay #undef __FUNCT__ 13034a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ" 1304dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 1305da3a660dSBarry Smith { 1306416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13072235e667SBarry Smith IS iscol = a->col,isrow = a->row; 13086849ba73SBarry Smith PetscErrorCode ierr; 130904fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 131004fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 131104fbf559SBarry Smith PetscInt nz; 131204fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1313dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 131404fbf559SBarry Smith const PetscScalar *b; 13156abc6512SBarry Smith 13163a40ed3dSBarry Smith PetscFunctionBegin; 131723bc6035SMatthew Knepley if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 131804fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 13191ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1320416022c9SBarry Smith tmp = a->solve_work; 13216abc6512SBarry Smith 13222235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13232235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 13246abc6512SBarry Smith 13256abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 13262235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 13276abc6512SBarry Smith 13286abc6512SBarry Smith /* forward solve the U^T */ 13296abc6512SBarry Smith for (i=0; i<n; i++) { 1330010a20caSHong Zhang v = aa + diag[i] ; 1331010a20caSHong Zhang vi = aj + diag[i] + 1; 1332f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 133304fbf559SBarry Smith s1 = tmp[i]; 133404fbf559SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 133504fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 133604fbf559SBarry Smith tmp[i] = s1; 13376abc6512SBarry Smith } 13386abc6512SBarry Smith 13396abc6512SBarry Smith /* backward solve the L^T */ 13406abc6512SBarry Smith for (i=n-1; i>=0; i--){ 1341010a20caSHong Zhang v = aa + diag[i] - 1 ; 1342010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1343f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 134404fbf559SBarry Smith s1 = tmp[i]; 134504fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 13466abc6512SBarry Smith } 13476abc6512SBarry Smith 13486abc6512SBarry Smith /* copy tmp into x according to permutation */ 13496abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 13506abc6512SBarry Smith 13512235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13522235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 135304fbf559SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 13541ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 13556abc6512SBarry Smith 135604fbf559SBarry Smith ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 13573a40ed3dSBarry Smith PetscFunctionReturn(0); 1358da3a660dSBarry Smith } 135904fbf559SBarry Smith 13609e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 13613c5fc038SBarry Smith EXTERN PetscErrorCode Mat_CheckInode(Mat,PetscTruth); 1362f77e22a1SHong Zhang EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscTruth); 136308480c60SBarry Smith 13648fc3a347SHong Zhang /* 13658fc3a347SHong Zhang ilu(0) with natural ordering under new data structure. 13668fc3a347SHong Zhang Factored arrays bj and ba are stored as 13678fc3a347SHong Zhang L(0,:), L(1,:), ...,L(n-1,:), U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:) 13688fc3a347SHong Zhang 13698fc3a347SHong Zhang bi=fact->i is an array of size 2n+2, in which 13708fc3a347SHong Zhang bi+ 13718fc3a347SHong Zhang bi[i] -> 1st entry of L(i,:),i=0,...,i-1 1372f77e22a1SHong Zhang bi[n] -> points to L(n-1,:)+1 13738fc3a347SHong Zhang bi[n+1] -> 1st entry of U(n-1,:) 13748fc3a347SHong Zhang bi[2n-i] -> 1st entry of U(i,:) 13758fc3a347SHong Zhang bi[2n-i+1] -> end of U(i,:)+1, the 1st entry of U(i-1,:) 1376f77e22a1SHong Zhang bi[2n] -> 1st entry of U(0,:) 1377f77e22a1SHong Zhang bi[2n+1] -> points to U(0,:)+1 13788fc3a347SHong Zhang 13798fc3a347SHong Zhang U(i,:) contains diag[i] as its last entry, i.e., 13808fc3a347SHong Zhang U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 13818fc3a347SHong Zhang */ 1382c9c72f8fSHong Zhang #undef __FUNCT__ 13838fc3a347SHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct" 13848fc3a347SHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1385c9c72f8fSHong Zhang { 1386c9c72f8fSHong Zhang 1387c9c72f8fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1388c9c72f8fSHong Zhang PetscErrorCode ierr; 1389c9c72f8fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag; 13908fc3a347SHong Zhang PetscInt i,j,nz,*bi,*bj,*bdiag; 1391c9c72f8fSHong Zhang 1392c9c72f8fSHong Zhang PetscFunctionBegin; 1393f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr); 1394c9c72f8fSHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 1395c9c72f8fSHong Zhang 1396f77e22a1SHong Zhang /* allocate matrix arrays for new data structure */ 1397f77e22a1SHong Zhang ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,2*n+2,PetscInt,&b->i);CHKERRQ(ierr); 1398f77e22a1SHong Zhang ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(2*n+2)*sizeof(PetscInt));CHKERRQ(ierr); 1399f77e22a1SHong Zhang b->singlemalloc = PETSC_TRUE; 1400f77e22a1SHong Zhang if (!b->diag){ 14018fc3a347SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr); 1402f77e22a1SHong Zhang } 1403c9c72f8fSHong Zhang bdiag = b->diag; 1404c9c72f8fSHong Zhang 1405c9c72f8fSHong Zhang if (n > 0) { 140616a2bf60SHong Zhang ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr); 1407c9c72f8fSHong Zhang } 1408c9c72f8fSHong Zhang 1409c9c72f8fSHong Zhang /* set bi and bj with new data structure */ 1410c9c72f8fSHong Zhang bi = b->i; 1411c9c72f8fSHong Zhang bj = b->j; 1412c9c72f8fSHong Zhang 1413c9c72f8fSHong Zhang /* L part */ 1414c9c72f8fSHong Zhang bi[0] = 0; 1415c9c72f8fSHong Zhang for (i=0; i<n; i++){ 1416c9c72f8fSHong Zhang nz = adiag[i] - ai[i]; 1417c9c72f8fSHong Zhang bi[i+1] = bi[i] + nz; 1418c9c72f8fSHong Zhang aj = a->j + ai[i]; 1419c9c72f8fSHong Zhang for (j=0; j<nz; j++){ 1420c9c72f8fSHong Zhang *bj = aj[j]; bj++; 1421c9c72f8fSHong Zhang } 1422c9c72f8fSHong Zhang } 1423c9c72f8fSHong Zhang 1424c9c72f8fSHong Zhang /* U part */ 14258fc3a347SHong Zhang bi[n+1] = bi[n]; 1426c9c72f8fSHong Zhang for (i=n-1; i>=0; i--){ 1427c9c72f8fSHong Zhang nz = ai[i+1] - adiag[i] - 1; 14288fc3a347SHong Zhang bi[2*n-i+1] = bi[2*n-i] + nz + 1; 1429c9c72f8fSHong Zhang aj = a->j + adiag[i] + 1; 1430c9c72f8fSHong Zhang for (j=0; j<nz; j++){ 1431c9c72f8fSHong Zhang *bj = aj[j]; bj++; 1432c9c72f8fSHong Zhang } 1433c9c72f8fSHong Zhang /* diag[i] */ 1434c9c72f8fSHong Zhang *bj = i; bj++; 14358fc3a347SHong Zhang bdiag[i] = bi[2*n-i+1]-1; 1436c9c72f8fSHong Zhang } 1437c9c72f8fSHong Zhang PetscFunctionReturn(0); 1438c9c72f8fSHong Zhang } 1439c9c72f8fSHong Zhang 14404a2ae208SSatish Balay #undef __FUNCT__ 14416516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_newdatastruct" 14426516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 14439e25ed09SBarry Smith { 1444416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 14459e25ed09SBarry Smith IS isicol; 14466849ba73SBarry Smith PetscErrorCode ierr; 14475d0c19d7SBarry Smith const PetscInt *r,*ic; 14485d0c19d7SBarry Smith PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 14495a8e39fbSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 1450052f0c41SBarry Smith PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 14515a8e39fbSHong Zhang PetscInt i,levels,diagonal_fill; 145277c4ece6SBarry Smith PetscTruth col_identity,row_identity; 1453329f5518SBarry Smith PetscReal f; 14545a8e39fbSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 14555a8e39fbSHong Zhang PetscBT lnkbt; 14565a8e39fbSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 1457a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1458a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 145909f38230SBarry Smith PetscTruth missing; 14609e25ed09SBarry Smith 14613a40ed3dSBarry Smith PetscFunctionBegin; 1462d0f46423SBarry 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); 14636cf997b0SBarry Smith f = info->fill; 146438baddfdSBarry Smith levels = (PetscInt)info->levels; 146538baddfdSBarry Smith diagonal_fill = (PetscInt)info->diagonal_fill; 14664c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 146782bf6240SBarry Smith 1468be0abb6dSBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 1469be0abb6dSBarry Smith ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 1470c9c72f8fSHong Zhang 14716516239fSHong Zhang if (!levels && row_identity && col_identity) { 14726516239fSHong Zhang /* special case: ilu(0) with natural ordering */ 14738fc3a347SHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1474c9c72f8fSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 14756516239fSHong Zhang 14766516239fSHong Zhang fact->factor = MAT_FACTOR_ILU; 14776516239fSHong Zhang (fact)->info.factor_mallocs = 0; 14786516239fSHong Zhang (fact)->info.fill_ratio_given = info->fill; 14796516239fSHong Zhang (fact)->info.fill_ratio_needed = 1.0; 14806516239fSHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 14816516239fSHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 14826516239fSHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 14836516239fSHong Zhang b->row = isrow; 14846516239fSHong Zhang b->col = iscol; 14856516239fSHong Zhang b->icol = isicol; 14866516239fSHong Zhang ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 14876516239fSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 14886516239fSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 14896516239fSHong Zhang /* ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */ 14906516239fSHong Zhang PetscFunctionReturn(0); 14916516239fSHong Zhang } 14926516239fSHong Zhang 14936516239fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 14946516239fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 14956516239fSHong Zhang 14966516239fSHong Zhang /* get new row pointers */ 14976516239fSHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 14986516239fSHong Zhang bi[0] = 0; 14996516239fSHong Zhang /* bdiag is location of diagonal in factor */ 15006516239fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 15016516239fSHong Zhang bdiag[0] = 0; 15026516239fSHong Zhang 15036516239fSHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr); 15046516239fSHong Zhang bjlvl_ptr = (PetscInt**)(bj_ptr + n); 15056516239fSHong Zhang 15066516239fSHong Zhang /* create a linked list for storing column indices of the active row */ 15076516239fSHong Zhang nlnk = n + 1; 15086516239fSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 15096516239fSHong Zhang 15106516239fSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 15116516239fSHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 15126516239fSHong Zhang current_space = free_space; 15136516239fSHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 15146516239fSHong Zhang current_space_lvl = free_space_lvl; 15156516239fSHong Zhang 15166516239fSHong Zhang for (i=0; i<n; i++) { 15176516239fSHong Zhang nzi = 0; 15186516239fSHong Zhang /* copy current row into linked list */ 15196516239fSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 15206516239fSHong Zhang if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 15216516239fSHong Zhang cols = aj + ai[r[i]]; 15226516239fSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 15236516239fSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 15246516239fSHong Zhang nzi += nlnk; 15256516239fSHong Zhang 15266516239fSHong Zhang /* make sure diagonal entry is included */ 15276516239fSHong Zhang if (diagonal_fill && lnk[i] == -1) { 15286516239fSHong Zhang fm = n; 15296516239fSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 15306516239fSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 15316516239fSHong Zhang lnk[fm] = i; 15326516239fSHong Zhang lnk_lvl[i] = 0; 15336516239fSHong Zhang nzi++; dcount++; 15346516239fSHong Zhang } 15356516239fSHong Zhang 15366516239fSHong Zhang /* add pivot rows into the active row */ 15376516239fSHong Zhang nzbd = 0; 15386516239fSHong Zhang prow = lnk[n]; 15396516239fSHong Zhang while (prow < i) { 15406516239fSHong Zhang nnz = bdiag[prow]; 15416516239fSHong Zhang cols = bj_ptr[prow] + nnz + 1; 15426516239fSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 15436516239fSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 15446516239fSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 15456516239fSHong Zhang nzi += nlnk; 15466516239fSHong Zhang prow = lnk[prow]; 15476516239fSHong Zhang nzbd++; 15486516239fSHong Zhang } 15496516239fSHong Zhang bdiag[i] = nzbd; 15506516239fSHong Zhang bi[i+1] = bi[i] + nzi; 15516516239fSHong Zhang 15526516239fSHong Zhang /* if free space is not available, make more free space */ 15536516239fSHong Zhang if (current_space->local_remaining<nzi) { 15546516239fSHong Zhang nnz = 2*nzi*(n - i); /* estimated and max additional space needed */ 15556516239fSHong Zhang ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 15566516239fSHong Zhang ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 15576516239fSHong Zhang reallocs++; 15586516239fSHong Zhang } 15596516239fSHong Zhang 15606516239fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 15616516239fSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 15626516239fSHong Zhang bj_ptr[i] = current_space->array; 15636516239fSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 15646516239fSHong Zhang 15656516239fSHong Zhang /* make sure the active row i has diagonal entry */ 15666516239fSHong Zhang if (*(bj_ptr[i]+bdiag[i]) != i) { 15676516239fSHong Zhang SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 15686516239fSHong Zhang try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 15696516239fSHong Zhang } 15706516239fSHong Zhang 15716516239fSHong Zhang current_space->array += nzi; 15726516239fSHong Zhang current_space->local_used += nzi; 15736516239fSHong Zhang current_space->local_remaining -= nzi; 15746516239fSHong Zhang current_space_lvl->array += nzi; 15756516239fSHong Zhang current_space_lvl->local_used += nzi; 15766516239fSHong Zhang current_space_lvl->local_remaining -= nzi; 15776516239fSHong Zhang } 15786516239fSHong Zhang 15796516239fSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 15806516239fSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 15816516239fSHong Zhang 15826516239fSHong Zhang /* destroy list of free space and other temporary arrays */ 15836516239fSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 15846516239fSHong Zhang 15856516239fSHong Zhang /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 1586783ef271SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 15876516239fSHong Zhang 15886516239fSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 15896516239fSHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 15906516239fSHong Zhang ierr = PetscFree(bj_ptr);CHKERRQ(ierr); 15916516239fSHong Zhang 15926516239fSHong Zhang #if defined(PETSC_USE_INFO) 15936516239fSHong Zhang { 15946516239fSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 15956516239fSHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 15966516239fSHong Zhang ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 15976516239fSHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 15986516239fSHong Zhang ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 15996516239fSHong Zhang if (diagonal_fill) { 16006516239fSHong Zhang ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 16016516239fSHong Zhang } 16026516239fSHong Zhang } 16036516239fSHong Zhang #endif 16046516239fSHong Zhang 16056516239fSHong Zhang /* put together the new matrix */ 16066516239fSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 16076516239fSHong Zhang ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 16086516239fSHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 16096516239fSHong Zhang b->free_a = PETSC_TRUE; 16106516239fSHong Zhang b->free_ij = PETSC_TRUE; 16116516239fSHong Zhang b->singlemalloc = PETSC_FALSE; 16126516239fSHong Zhang ierr = PetscMalloc( (bi[2*n+1] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 16136516239fSHong Zhang b->j = bj; 16146516239fSHong Zhang b->i = bi; 16156516239fSHong Zhang b->diag = bdiag; 16166516239fSHong Zhang b->ilen = 0; 16176516239fSHong Zhang b->imax = 0; 16186516239fSHong Zhang b->row = isrow; 16196516239fSHong Zhang b->col = iscol; 16206516239fSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 16216516239fSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 16226516239fSHong Zhang b->icol = isicol; 16236516239fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 16246516239fSHong Zhang /* In b structure: Free imax, ilen, old a, old j. 16256516239fSHong Zhang Allocate bdiag, solve_work, new a, new j */ 16266516239fSHong Zhang ierr = PetscLogObjectMemory(fact,bi[2*n+1] * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 16276516239fSHong Zhang b->maxnz = b->nz = bi[2*n+1] ; 16286516239fSHong Zhang (fact)->info.factor_mallocs = reallocs; 16296516239fSHong Zhang (fact)->info.fill_ratio_given = f; 16306516239fSHong Zhang (fact)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]); 16316516239fSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 16326516239fSHong Zhang /* ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */ 16336516239fSHong Zhang PetscFunctionReturn(0); 16346516239fSHong Zhang } 16356516239fSHong Zhang 16366516239fSHong Zhang #undef __FUNCT__ 16376516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ" 16386516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 16396516239fSHong Zhang { 16406516239fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 16416516239fSHong Zhang IS isicol; 16426516239fSHong Zhang PetscErrorCode ierr; 16436516239fSHong Zhang const PetscInt *r,*ic; 16446516239fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 16456516239fSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 16466516239fSHong Zhang PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 16476516239fSHong Zhang PetscInt i,levels,diagonal_fill; 16486516239fSHong Zhang PetscTruth col_identity,row_identity; 16496516239fSHong Zhang PetscReal f; 16506516239fSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 16516516239fSHong Zhang PetscBT lnkbt; 16526516239fSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 16536516239fSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 16546516239fSHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 16556516239fSHong Zhang PetscTruth missing; 16566516239fSHong Zhang PetscTruth newdatastruct=PETSC_FALSE; 16576516239fSHong Zhang 16586516239fSHong Zhang PetscFunctionBegin; 16596516239fSHong Zhang ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 16606516239fSHong Zhang if (newdatastruct){ 16616516239fSHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr); 16626516239fSHong Zhang PetscFunctionReturn(0); 16636516239fSHong Zhang } 16646516239fSHong Zhang 16656516239fSHong Zhang 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); 16666516239fSHong Zhang f = info->fill; 16676516239fSHong Zhang levels = (PetscInt)info->levels; 16686516239fSHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 16696516239fSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 16706516239fSHong Zhang 16716516239fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 16726516239fSHong Zhang ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 16736516239fSHong Zhang if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */ 1674f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr); 1675c9c72f8fSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1676c9c72f8fSHong Zhang 1677719d5645SBarry Smith fact->factor = MAT_FACTOR_ILU; 1678719d5645SBarry Smith (fact)->info.factor_mallocs = 0; 1679719d5645SBarry Smith (fact)->info.fill_ratio_given = info->fill; 1680719d5645SBarry Smith (fact)->info.fill_ratio_needed = 1.0; 1681719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 16828ef3462fSBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 168309f38230SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 168408480c60SBarry Smith b->row = isrow; 168508480c60SBarry Smith b->col = iscol; 168682bf6240SBarry Smith b->icol = isicol; 1687719d5645SBarry Smith ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1688c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1689c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1690719d5645SBarry Smith ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); 16913a40ed3dSBarry Smith PetscFunctionReturn(0); 169208480c60SBarry Smith } 169308480c60SBarry Smith 1694898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1695898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 16969e25ed09SBarry Smith 16979e25ed09SBarry Smith /* get new row pointers */ 16985a8e39fbSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 16995a8e39fbSHong Zhang bi[0] = 0; 17005a8e39fbSHong Zhang /* bdiag is location of diagonal in factor */ 17015a8e39fbSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 17025a8e39fbSHong Zhang bdiag[0] = 0; 17036cf997b0SBarry Smith 17045a8e39fbSHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr); 17055a8e39fbSHong Zhang bjlvl_ptr = (PetscInt**)(bj_ptr + n); 17065a8e39fbSHong Zhang 17075a8e39fbSHong Zhang /* create a linked list for storing column indices of the active row */ 17085a8e39fbSHong Zhang nlnk = n + 1; 17095a8e39fbSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 17105a8e39fbSHong Zhang 17115a8e39fbSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 1712a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 17135a8e39fbSHong Zhang current_space = free_space; 1714a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 17155a8e39fbSHong Zhang current_space_lvl = free_space_lvl; 17165a8e39fbSHong Zhang 17175a8e39fbSHong Zhang for (i=0; i<n; i++) { 17185a8e39fbSHong Zhang nzi = 0; 17196cf997b0SBarry Smith /* copy current row into linked list */ 17205a8e39fbSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 17213b4a8b6dSBarry Smith if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 17225a8e39fbSHong Zhang cols = aj + ai[r[i]]; 17235a8e39fbSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 17245a8e39fbSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 17255a8e39fbSHong Zhang nzi += nlnk; 17266cf997b0SBarry Smith 17276cf997b0SBarry Smith /* make sure diagonal entry is included */ 17285a8e39fbSHong Zhang if (diagonal_fill && lnk[i] == -1) { 17296cf997b0SBarry Smith fm = n; 17305a8e39fbSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 17315a8e39fbSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 17325a8e39fbSHong Zhang lnk[fm] = i; 17335a8e39fbSHong Zhang lnk_lvl[i] = 0; 17345a8e39fbSHong Zhang nzi++; dcount++; 17356cf997b0SBarry Smith } 17366cf997b0SBarry Smith 17375a8e39fbSHong Zhang /* add pivot rows into the active row */ 17385a8e39fbSHong Zhang nzbd = 0; 17395a8e39fbSHong Zhang prow = lnk[n]; 17405a8e39fbSHong Zhang while (prow < i) { 17415a8e39fbSHong Zhang nnz = bdiag[prow]; 17425a8e39fbSHong Zhang cols = bj_ptr[prow] + nnz + 1; 17435a8e39fbSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 17445a8e39fbSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 17455a8e39fbSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 17465a8e39fbSHong Zhang nzi += nlnk; 17475a8e39fbSHong Zhang prow = lnk[prow]; 17485a8e39fbSHong Zhang nzbd++; 174956beaf04SBarry Smith } 17505a8e39fbSHong Zhang bdiag[i] = nzbd; 17515a8e39fbSHong Zhang bi[i+1] = bi[i] + nzi; 1752ecf371e4SBarry Smith 17535a8e39fbSHong Zhang /* if free space is not available, make more free space */ 17545a8e39fbSHong Zhang if (current_space->local_remaining<nzi) { 17555a8e39fbSHong Zhang nnz = nzi*(n - i); /* estimated and max additional space needed */ 1756a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1757a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 17585a8e39fbSHong Zhang reallocs++; 17595a8e39fbSHong Zhang } 1760ecf371e4SBarry Smith 17615a8e39fbSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 17625a8e39fbSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 17635a8e39fbSHong Zhang bj_ptr[i] = current_space->array; 17645a8e39fbSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 17655a8e39fbSHong Zhang 17665a8e39fbSHong Zhang /* make sure the active row i has diagonal entry */ 17675a8e39fbSHong Zhang if (*(bj_ptr[i]+bdiag[i]) != i) { 176877431f27SBarry Smith SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 1769d7ee0231SBarry Smith try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 17706cf997b0SBarry Smith } 17715a8e39fbSHong Zhang 17725a8e39fbSHong Zhang current_space->array += nzi; 17735a8e39fbSHong Zhang current_space->local_used += nzi; 17745a8e39fbSHong Zhang current_space->local_remaining -= nzi; 17755a8e39fbSHong Zhang current_space_lvl->array += nzi; 17765a8e39fbSHong Zhang current_space_lvl->local_used += nzi; 17775a8e39fbSHong Zhang current_space_lvl->local_remaining -= nzi; 17789e25ed09SBarry Smith } 17795a8e39fbSHong Zhang 1780898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1781898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 17825a8e39fbSHong Zhang 17835a8e39fbSHong Zhang /* destroy list of free space and other temporary arrays */ 17845a8e39fbSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 17856516239fSHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */ 17865a8e39fbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1787a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 17885a8e39fbSHong Zhang ierr = PetscFree(bj_ptr);CHKERRQ(ierr); 17899e25ed09SBarry Smith 17906cf91177SBarry Smith #if defined(PETSC_USE_INFO) 1791f64065bbSBarry Smith { 17925a8e39fbSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1793ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 1794ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1795ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 1796ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1797335d9088SBarry Smith if (diagonal_fill) { 1798ae15b995SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 1799335d9088SBarry Smith } 1800f64065bbSBarry Smith } 180163ba0a88SBarry Smith #endif 1802bbb6d6a8SBarry Smith 18039e25ed09SBarry Smith /* put together the new matrix */ 180407dbf95fSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1805719d5645SBarry Smith ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 1806719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 1807e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1808e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 18097c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 1810052f0c41SBarry Smith ierr = PetscMalloc( (bi[n] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 18115a8e39fbSHong Zhang b->j = bj; 18125a8e39fbSHong Zhang b->i = bi; 18135a8e39fbSHong Zhang for (i=0; i<n; i++) bdiag[i] += bi[i]; 18145a8e39fbSHong Zhang b->diag = bdiag; 1815416022c9SBarry Smith b->ilen = 0; 1816416022c9SBarry Smith b->imax = 0; 1817416022c9SBarry Smith b->row = isrow; 1818416022c9SBarry Smith b->col = iscol; 1819c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1820c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 182182bf6240SBarry Smith b->icol = isicol; 182287828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1823416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 18245a8e39fbSHong Zhang Allocate bdiag, solve_work, new a, new j */ 1825719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 18265a8e39fbSHong Zhang b->maxnz = b->nz = bi[n] ; 1827719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 1828719d5645SBarry Smith (fact)->info.fill_ratio_given = f; 1829719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1830719d5645SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1831719d5645SBarry Smith ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); 18323a40ed3dSBarry Smith PetscFunctionReturn(0); 18339e25ed09SBarry Smith } 1834d5d45c9bSBarry Smith 1835a6175056SHong Zhang #undef __FUNCT__ 18365f44c854SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_newdatastruct" 18375f44c854SHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info) 18385f44c854SHong Zhang { 18395f44c854SHong Zhang Mat C = B; 18405f44c854SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 18415f44c854SHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 18425f44c854SHong Zhang IS ip=b->row,iip = b->icol; 18435f44c854SHong Zhang PetscErrorCode ierr; 18445f44c854SHong Zhang const PetscInt *rip,*riip; 18455f44c854SHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp; 18465f44c854SHong Zhang PetscInt *ai=a->i,*aj=a->j; 18475f44c854SHong Zhang PetscInt k,jmin,jmax,*c2r,*il,col,nexti,ili,nz; 18485f44c854SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 18495f44c854SHong Zhang PetscReal zeropivot,shiftnz; 18505f44c854SHong Zhang PetscReal shiftpd; 18515f44c854SHong Zhang PetscTruth perm_identity; 18525f44c854SHong Zhang 18535f44c854SHong Zhang PetscFunctionBegin; 18545f44c854SHong Zhang 18555f44c854SHong Zhang shiftnz = info->shiftnz; 18565f44c854SHong Zhang shiftpd = info->shiftpd; 18575f44c854SHong Zhang zeropivot = info->zeropivot; 18585f44c854SHong Zhang 18595f44c854SHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 18605f44c854SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 18615f44c854SHong Zhang 18625f44c854SHong Zhang /* allocate working arrays 18635f44c854SHong Zhang c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col 18645f44c854SHong Zhang il: for active k row, il[i] gives the index of the 1st nonzero entry in U[i,k:n-1] in bj and ba arrays 18655f44c854SHong Zhang */ 18665f44c854SHong Zhang nz = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar); 18675f44c854SHong Zhang ierr = PetscMalloc(nz,&il);CHKERRQ(ierr); 18685f44c854SHong Zhang c2r = il + mbs; 18695f44c854SHong Zhang rtmp = (MatScalar*)(c2r + mbs); 18705f44c854SHong Zhang 18715f44c854SHong Zhang for (i=0; i<mbs; i++) { 18725f44c854SHong Zhang c2r[i] = mbs; 18735f44c854SHong Zhang } 18745f44c854SHong Zhang il[0] = 0; 18755f44c854SHong Zhang 18765f44c854SHong Zhang for (k = 0; k<mbs; k++){ 18775f44c854SHong Zhang /* zero rtmp */ 18785f44c854SHong Zhang nz = bi[k+1] - bi[k]; 18795f44c854SHong Zhang bjtmp = bj + bi[k]; 18805f44c854SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 18815f44c854SHong Zhang 18825f44c854SHong Zhang /* load in initial unfactored row */ 18835f44c854SHong Zhang bval = ba + bi[k]; 18845f44c854SHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 18855f44c854SHong Zhang for (j = jmin; j < jmax; j++){ 18865f44c854SHong Zhang col = riip[aj[j]]; 18875f44c854SHong Zhang if (col >= k){ /* only take upper triangular entry */ 18885f44c854SHong Zhang rtmp[col] = aa[j]; 18895f44c854SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 18905f44c854SHong Zhang } 18915f44c854SHong Zhang } 18925f44c854SHong Zhang /* shift the diagonal of the matrix */ 18935f44c854SHong Zhang 18945f44c854SHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 18955f44c854SHong Zhang dk = rtmp[k]; 18965f44c854SHong Zhang i = c2r[k]; /* first row to be added to k_th row */ 18975f44c854SHong Zhang 18985f44c854SHong Zhang while (i < k){ 18995f44c854SHong Zhang nexti = c2r[i]; /* next row to be added to k_th row */ 19005f44c854SHong Zhang 19015f44c854SHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 19025f44c854SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 19035f44c854SHong Zhang uikdi = - ba[ili]*ba[bdiag[i]]; /* diagonal(k) */ 19045f44c854SHong Zhang dk += uikdi*ba[ili]; /* update diag[k] */ 19055f44c854SHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 19065f44c854SHong Zhang 19075f44c854SHong Zhang /* add multiple of row i to k-th row */ 19085f44c854SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 19095f44c854SHong Zhang if (jmin < jmax){ 19105f44c854SHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 19115f44c854SHong Zhang /* update il and c2r for row i */ 19125f44c854SHong Zhang il[i] = jmin; 19135f44c854SHong Zhang j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i; 19145f44c854SHong Zhang } 19155f44c854SHong Zhang i = nexti; 19165f44c854SHong Zhang } 19175f44c854SHong Zhang 19185f44c854SHong Zhang /* copy data into U(k,:) */ 19195f44c854SHong Zhang ba[bdiag[k]] = 1.0/dk; /* U(k,k) */ 19205f44c854SHong Zhang jmin = bi[k]; jmax = bi[k+1]-1; 19215f44c854SHong Zhang if (jmin < jmax) { 19225f44c854SHong Zhang for (j=jmin; j<jmax; j++){ 19235f44c854SHong Zhang col = bj[j]; ba[j] = rtmp[col]; 19245f44c854SHong Zhang } 19255f44c854SHong Zhang /* add the k-th row into il and c2r */ 19265f44c854SHong Zhang il[k] = jmin; 19275f44c854SHong Zhang i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k; 19285f44c854SHong Zhang } 19295f44c854SHong Zhang } 19305f44c854SHong Zhang 19315f44c854SHong Zhang ierr = PetscFree(il);CHKERRQ(ierr); 19325f44c854SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 19335f44c854SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 19345f44c854SHong Zhang 19355f44c854SHong Zhang ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 19365f44c854SHong Zhang if (perm_identity){ 19375f44c854SHong Zhang (B)->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering_newdatastruct; 19385f44c854SHong Zhang (B)->ops->solvetranspose = 0; 19395f44c854SHong Zhang (B)->ops->forwardsolve = 0; 19405f44c854SHong Zhang (B)->ops->backwardsolve = 0; 19415f44c854SHong Zhang } else { 1942783ef271SHong Zhang (B)->ops->solve = MatSolve_SeqSBAIJ_1_newdatastruct; 19435f44c854SHong Zhang (B)->ops->solvetranspose = 0; 19445f44c854SHong Zhang (B)->ops->forwardsolve = 0; 19455f44c854SHong Zhang (B)->ops->backwardsolve = 0; 19465f44c854SHong Zhang } 19475f44c854SHong Zhang 19485f44c854SHong Zhang C->assembled = PETSC_TRUE; 19495f44c854SHong Zhang C->preallocated = PETSC_TRUE; 19505f44c854SHong Zhang ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 19515f44c854SHong Zhang PetscFunctionReturn(0); 19525f44c854SHong Zhang } 19535f44c854SHong Zhang 19545f44c854SHong Zhang #undef __FUNCT__ 1955a6175056SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ" 19560481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 1957a6175056SHong Zhang { 1958719d5645SBarry Smith Mat C = B; 19590968510aSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 19602ed133dbSHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 19619bfd6278SHong Zhang IS ip=b->row,iip = b->icol; 19622ed133dbSHong Zhang PetscErrorCode ierr; 19635d0c19d7SBarry Smith const PetscInt *rip,*riip; 19645d0c19d7SBarry Smith PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol; 19652ed133dbSHong Zhang PetscInt *ai=a->i,*aj=a->j; 19662ed133dbSHong Zhang PetscInt k,jmin,jmax,*jl,*il,col,nexti,ili,nz; 1967622e2028SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 1968540703acSHong Zhang PetscReal zeropivot,rs,shiftnz; 1969fbf22428SSatish Balay PetscReal shiftpd; 1970540703acSHong Zhang ChShift_Ctx sctx; 1971540703acSHong Zhang PetscInt newshift; 1972db4efbfdSBarry Smith PetscTruth perm_identity; 1973d5d45c9bSBarry Smith 1974a6175056SHong Zhang PetscFunctionBegin; 1975540703acSHong Zhang shiftnz = info->shiftnz; 1976540703acSHong Zhang shiftpd = info->shiftpd; 1977ee45ca4aSHong Zhang zeropivot = info->zeropivot; 1978ee45ca4aSHong Zhang 19792ed133dbSHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 19809bfd6278SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 19812ed133dbSHong Zhang 19822ed133dbSHong Zhang /* initialization */ 19832ed133dbSHong Zhang nz = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar); 19842ed133dbSHong Zhang ierr = PetscMalloc(nz,&il);CHKERRQ(ierr); 19852ed133dbSHong Zhang jl = il + mbs; 19862ed133dbSHong Zhang rtmp = (MatScalar*)(jl + mbs); 19872ed133dbSHong Zhang 1988540703acSHong Zhang sctx.shift_amount = 0; 1989540703acSHong Zhang sctx.nshift = 0; 19902ed133dbSHong Zhang do { 1991540703acSHong Zhang sctx.chshift = PETSC_FALSE; 19922ed133dbSHong Zhang for (i=0; i<mbs; i++) { 19932ed133dbSHong Zhang rtmp[i] = 0.0; jl[i] = mbs; il[0] = 0; 1994a6175056SHong Zhang } 19952ed133dbSHong Zhang 19962ed133dbSHong Zhang for (k = 0; k<mbs; k++){ 1997c05c3958SHong Zhang bval = ba + bi[k]; 19982ed133dbSHong Zhang /* initialize k-th row by the perm[k]-th row of A */ 19992ed133dbSHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 20002ed133dbSHong Zhang for (j = jmin; j < jmax; j++){ 20019bfd6278SHong Zhang col = riip[aj[j]]; 20022ed133dbSHong Zhang if (col >= k){ /* only take upper triangular entry */ 20032ed133dbSHong Zhang rtmp[col] = aa[j]; 2004c05c3958SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 20052ed133dbSHong Zhang } 20062ed133dbSHong Zhang } 200739e3d630SHong Zhang /* shift the diagonal of the matrix */ 2008540703acSHong Zhang if (sctx.nshift) rtmp[k] += sctx.shift_amount; 20092ed133dbSHong Zhang 20102ed133dbSHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 20112ed133dbSHong Zhang dk = rtmp[k]; 20122ed133dbSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 20132ed133dbSHong Zhang 20142ed133dbSHong Zhang while (i < k){ 20152ed133dbSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 20162ed133dbSHong Zhang 20172ed133dbSHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 20182ed133dbSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 20192ed133dbSHong Zhang uikdi = - ba[ili]*ba[bi[i]]; /* diagonal(k) */ 20202ed133dbSHong Zhang dk += uikdi*ba[ili]; 20212ed133dbSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 20222ed133dbSHong Zhang 20232ed133dbSHong Zhang /* add multiple of row i to k-th row */ 20242ed133dbSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 20252ed133dbSHong Zhang if (jmin < jmax){ 20262ed133dbSHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 20272ed133dbSHong Zhang /* update il and jl for row i */ 20282ed133dbSHong Zhang il[i] = jmin; 20292ed133dbSHong Zhang j = bj[jmin]; jl[i] = jl[j]; jl[j] = i; 20302ed133dbSHong Zhang } 20312ed133dbSHong Zhang i = nexti; 20322ed133dbSHong Zhang } 20332ed133dbSHong Zhang 20340697b6caSHong Zhang /* shift the diagonals when zero pivot is detected */ 20350697b6caSHong Zhang /* compute rs=sum of abs(off-diagonal) */ 20360697b6caSHong Zhang rs = 0.0; 20373c9e8598SHong Zhang jmin = bi[k]+1; 20383c9e8598SHong Zhang nz = bi[k+1] - jmin; 20393c9e8598SHong Zhang bcol = bj + jmin; 204004fbf559SBarry Smith for (j=0; j<nz; j++) { 204104fbf559SBarry Smith rs += PetscAbsScalar(rtmp[bcol[j]]); 20423c9e8598SHong Zhang } 2043540703acSHong Zhang 2044540703acSHong Zhang sctx.rs = rs; 2045540703acSHong Zhang sctx.pv = dk; 20465b5bc046SBarry Smith ierr = MatCholeskyCheckShift_inline(info,sctx,k,newshift);CHKERRQ(ierr); 2047117f1891Stmunson 2048117f1891Stmunson if (newshift == 1) { 2049117f1891Stmunson if (!sctx.shift_amount) { 2050117f1891Stmunson sctx.shift_amount = 1e-5; 2051117f1891Stmunson } 2052117f1891Stmunson break; 2053117f1891Stmunson } 20543c9e8598SHong Zhang 20553c9e8598SHong Zhang /* copy data into U(k,:) */ 205639e3d630SHong Zhang ba[bi[k]] = 1.0/dk; /* U(k,k) */ 205739e3d630SHong Zhang jmin = bi[k]+1; jmax = bi[k+1]; 205839e3d630SHong Zhang if (jmin < jmax) { 205939e3d630SHong Zhang for (j=jmin; j<jmax; j++){ 206039e3d630SHong Zhang col = bj[j]; ba[j] = rtmp[col]; rtmp[col] = 0.0; 20613c9e8598SHong Zhang } 206239e3d630SHong Zhang /* add the k-th row into il and jl */ 20633c9e8598SHong Zhang il[k] = jmin; 20643c9e8598SHong Zhang i = bj[jmin]; jl[k] = jl[i]; jl[i] = k; 20653c9e8598SHong Zhang } 20663c9e8598SHong Zhang } 2067540703acSHong Zhang } while (sctx.chshift); 20683c9e8598SHong Zhang ierr = PetscFree(il);CHKERRQ(ierr); 20693c9e8598SHong Zhang 207039e3d630SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 20719bfd6278SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 2072db4efbfdSBarry Smith 2073db4efbfdSBarry Smith ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 2074db4efbfdSBarry Smith if (perm_identity){ 2075719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 2076719d5645SBarry Smith (B)->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 2077719d5645SBarry Smith (B)->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering; 2078719d5645SBarry Smith (B)->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering; 2079db4efbfdSBarry Smith } else { 2080719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqSBAIJ_1; 2081719d5645SBarry Smith (B)->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 2082719d5645SBarry Smith (B)->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1; 2083719d5645SBarry Smith (B)->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1; 2084db4efbfdSBarry Smith } 2085db4efbfdSBarry Smith 20863c9e8598SHong Zhang C->assembled = PETSC_TRUE; 20873c9e8598SHong Zhang C->preallocated = PETSC_TRUE; 2088d0f46423SBarry Smith ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 2089540703acSHong Zhang if (sctx.nshift){ 2090540703acSHong Zhang if (shiftnz) { 20911e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2092540703acSHong Zhang } else if (shiftpd) { 20931e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 20940697b6caSHong Zhang } 20953c9e8598SHong Zhang } 20963c9e8598SHong Zhang PetscFunctionReturn(0); 20973c9e8598SHong Zhang } 2098a6175056SHong Zhang 2099a6175056SHong Zhang #undef __FUNCT__ 21005f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_newdatastruct" 21015f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2102a6175056SHong Zhang { 21030968510aSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2104ed59401aSHong Zhang Mat_SeqSBAIJ *b; 2105dfbe8321SBarry Smith PetscErrorCode ierr; 210658ebbce7SBarry Smith PetscTruth perm_identity,missing; 21075f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 21085d0c19d7SBarry Smith const PetscInt *rip,*riip; 2109ed59401aSHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 211058ebbce7SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 21115a8e39fbSHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 2112ed59401aSHong Zhang PetscReal fill=info->fill,levels=info->levels; 2113a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 2114a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 21157a48dd6fSHong Zhang PetscBT lnkbt; 2116b635d36bSHong Zhang IS iperm; 2117a6175056SHong Zhang 2118a6175056SHong Zhang PetscFunctionBegin; 2119d0f46423SBarry 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); 212058ebbce7SBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 212158ebbce7SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 2122653a6975SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 2123b635d36bSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 21247a48dd6fSHong Zhang 212539e3d630SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 21265f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 212739e3d630SHong Zhang ui[0] = 0; 212839e3d630SHong Zhang 2129b635d36bSHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 2130622e2028SHong Zhang if (!levels && perm_identity) { 213158ebbce7SBarry Smith 2132ed59401aSHong Zhang for (i=0; i<am; i++) { 213339e3d630SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 21345f44c854SHong Zhang udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */ 21355f44c854SHong Zhang } 21365f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 21375f44c854SHong Zhang cols = uj; 21385f44c854SHong Zhang for (i=0; i<am; i++) { 21395f44c854SHong Zhang aj = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */ 21405f44c854SHong Zhang ncols = ui[i+1] - ui[i] - 1; 21415f44c854SHong Zhang for (j=0; j<ncols; j++) *cols++ = aj[j]; 21425f44c854SHong Zhang *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */ 21435f44c854SHong Zhang } 21445f44c854SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 2145783ef271SHong Zhang 21465f44c854SHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 21475f44c854SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 21485f44c854SHong Zhang 21495f44c854SHong Zhang /* initialization */ 21505f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 21515f44c854SHong Zhang 21525f44c854SHong Zhang /* jl: linked list for storing indices of the pivot rows 21535f44c854SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 21545f44c854SHong Zhang ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr); 21555f44c854SHong Zhang il = jl + am; 21565f44c854SHong Zhang uj_ptr = (PetscInt**)(il + am); 21575f44c854SHong Zhang uj_lvl_ptr = (PetscInt**)(uj_ptr + am); 21585f44c854SHong Zhang for (i=0; i<am; i++){ 21595f44c854SHong Zhang jl[i] = am; il[i] = 0; 21605f44c854SHong Zhang } 21615f44c854SHong Zhang 21625f44c854SHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 21635f44c854SHong Zhang nlnk = am + 1; 21645f44c854SHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 21655f44c854SHong Zhang 21665f44c854SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 21675f44c854SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 21685f44c854SHong Zhang current_space = free_space; 21695f44c854SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 21705f44c854SHong Zhang current_space_lvl = free_space_lvl; 21715f44c854SHong Zhang 21725f44c854SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 21735f44c854SHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 21745f44c854SHong Zhang nzk = 0; 21755f44c854SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 21765f44c854SHong Zhang if (!ncols) SETERRQ2(PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 21775f44c854SHong Zhang ncols_upper = 0; 21785f44c854SHong Zhang for (j=0; j<ncols; j++){ 21795f44c854SHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 21805f44c854SHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 21815f44c854SHong Zhang ajtmp[ncols_upper] = i; 21825f44c854SHong Zhang ncols_upper++; 21835f44c854SHong Zhang } 21845f44c854SHong Zhang } 21855f44c854SHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 21865f44c854SHong Zhang nzk += nlnk; 21875f44c854SHong Zhang 21885f44c854SHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 21895f44c854SHong Zhang prow = jl[k]; /* 1st pivot row */ 21905f44c854SHong Zhang 21915f44c854SHong Zhang while (prow < k){ 21925f44c854SHong Zhang nextprow = jl[prow]; 21935f44c854SHong Zhang 21945f44c854SHong Zhang /* merge prow into k-th row */ 21955f44c854SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 21965f44c854SHong Zhang jmax = ui[prow+1]; 21975f44c854SHong Zhang ncols = jmax-jmin; 21985f44c854SHong Zhang i = jmin - ui[prow]; 21995f44c854SHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 22005f44c854SHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 22015f44c854SHong Zhang j = *(uj - 1); 22025f44c854SHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 22035f44c854SHong Zhang nzk += nlnk; 22045f44c854SHong Zhang 22055f44c854SHong Zhang /* update il and jl for prow */ 22065f44c854SHong Zhang if (jmin < jmax){ 22075f44c854SHong Zhang il[prow] = jmin; 22085f44c854SHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 22095f44c854SHong Zhang } 22105f44c854SHong Zhang prow = nextprow; 22115f44c854SHong Zhang } 22125f44c854SHong Zhang 22135f44c854SHong Zhang /* if free space is not available, make more free space */ 22145f44c854SHong Zhang if (current_space->local_remaining<nzk) { 22155f44c854SHong Zhang i = am - k + 1; /* num of unfactored rows */ 22165f44c854SHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 22175f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 22185f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 22195f44c854SHong Zhang reallocs++; 22205f44c854SHong Zhang } 22215f44c854SHong Zhang 22225f44c854SHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 22235f44c854SHong Zhang if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 22245f44c854SHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 22255f44c854SHong Zhang 22265f44c854SHong Zhang /* add the k-th row into il and jl */ 22275f44c854SHong Zhang if (nzk > 1){ 22285f44c854SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 22295f44c854SHong Zhang jl[k] = jl[i]; jl[i] = k; 22305f44c854SHong Zhang il[k] = ui[k] + 1; 22315f44c854SHong Zhang } 22325f44c854SHong Zhang uj_ptr[k] = current_space->array; 22335f44c854SHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 22345f44c854SHong Zhang 22355f44c854SHong Zhang current_space->array += nzk; 22365f44c854SHong Zhang current_space->local_used += nzk; 22375f44c854SHong Zhang current_space->local_remaining -= nzk; 22385f44c854SHong Zhang 22395f44c854SHong Zhang current_space_lvl->array += nzk; 22405f44c854SHong Zhang current_space_lvl->local_used += nzk; 22415f44c854SHong Zhang current_space_lvl->local_remaining -= nzk; 22425f44c854SHong Zhang 22435f44c854SHong Zhang ui[k+1] = ui[k] + nzk; 22445f44c854SHong Zhang } 22455f44c854SHong Zhang 22465f44c854SHong Zhang #if defined(PETSC_USE_INFO) 22475f44c854SHong Zhang if (ai[am] != 0) { 22485f44c854SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 22495f44c854SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 22505f44c854SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 22515f44c854SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 22525f44c854SHong Zhang } else { 22535f44c854SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 22545f44c854SHong Zhang } 22555f44c854SHong Zhang #endif 22565f44c854SHong Zhang 22575f44c854SHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 22585f44c854SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 22595f44c854SHong Zhang ierr = PetscFree(jl);CHKERRQ(ierr); 22605f44c854SHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 22615f44c854SHong Zhang 22625f44c854SHong Zhang /* destroy list of free space and other temporary array(s) */ 22635f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2264783ef271SHong Zhang ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); 22655f44c854SHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 22665f44c854SHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 22675f44c854SHong Zhang 22685f44c854SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 22695f44c854SHong Zhang 22705f44c854SHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 22715f44c854SHong Zhang 22725f44c854SHong Zhang b = (Mat_SeqSBAIJ*)(fact)->data; 22735f44c854SHong Zhang b->singlemalloc = PETSC_FALSE; 22745f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 22755f44c854SHong Zhang b->j = uj; 22765f44c854SHong Zhang b->i = ui; 22775f44c854SHong Zhang b->diag = udiag; 22787f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 22795f44c854SHong Zhang b->ilen = 0; 22805f44c854SHong Zhang b->imax = 0; 22815f44c854SHong Zhang b->row = perm; 22825f44c854SHong Zhang b->col = perm; 22835f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 22845f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 22855f44c854SHong Zhang b->icol = iperm; 22865f44c854SHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 22875f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 22885f44c854SHong Zhang ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 22895f44c854SHong Zhang b->maxnz = b->nz = ui[am]; 22905f44c854SHong Zhang b->free_a = PETSC_TRUE; 22915f44c854SHong Zhang b->free_ij = PETSC_TRUE; 22925f44c854SHong Zhang 22935f44c854SHong Zhang (fact)->info.factor_mallocs = reallocs; 22945f44c854SHong Zhang (fact)->info.fill_ratio_given = fill; 22955f44c854SHong Zhang if (ai[am] != 0) { 22965f44c854SHong Zhang (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 22975f44c854SHong Zhang } else { 22985f44c854SHong Zhang (fact)->info.fill_ratio_needed = 0.0; 22995f44c854SHong Zhang } 23005f44c854SHong Zhang (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_newdatastruct; 23015f44c854SHong Zhang PetscFunctionReturn(0); 23025f44c854SHong Zhang } 23035f44c854SHong Zhang 23045f44c854SHong Zhang #undef __FUNCT__ 23055f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ" 23065f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 23075f44c854SHong Zhang { 23085f44c854SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 23095f44c854SHong Zhang Mat_SeqSBAIJ *b; 23105f44c854SHong Zhang PetscErrorCode ierr; 23115f44c854SHong Zhang PetscTruth perm_identity,missing; 23125f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 23135f44c854SHong Zhang const PetscInt *rip,*riip; 23145f44c854SHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 23155f44c854SHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 23165f44c854SHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 23175f44c854SHong Zhang PetscReal fill=info->fill,levels=info->levels; 23185f44c854SHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 23195f44c854SHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 23205f44c854SHong Zhang PetscBT lnkbt; 23215f44c854SHong Zhang IS iperm; 23225f44c854SHong Zhang PetscTruth newdatastruct=PETSC_FALSE; 23235f44c854SHong Zhang 23245f44c854SHong Zhang PetscFunctionBegin; 23255f44c854SHong Zhang ierr = PetscOptionsGetTruth(PETSC_NULL,"-icc_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 23265f44c854SHong Zhang if(newdatastruct){ 23275f44c854SHong Zhang ierr = MatICCFactorSymbolic_SeqAIJ_newdatastruct(fact,A,perm,info);CHKERRQ(ierr); 23285f44c854SHong Zhang PetscFunctionReturn(0); 23295f44c854SHong Zhang } 23305f44c854SHong Zhang 23315f44c854SHong Zhang 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); 23325f44c854SHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 23335f44c854SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 23345f44c854SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 23355f44c854SHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 23365f44c854SHong Zhang 23375f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 23385f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 23395f44c854SHong Zhang ui[0] = 0; 23405f44c854SHong Zhang 23415f44c854SHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 23425f44c854SHong Zhang if (!levels && perm_identity) { 23435f44c854SHong Zhang 23445f44c854SHong Zhang for (i=0; i<am; i++) { 23455f44c854SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 23465f44c854SHong Zhang udiag[i] = ui[i]; 2347ed59401aSHong Zhang } 234839e3d630SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 234939e3d630SHong Zhang cols = uj; 2350ed59401aSHong Zhang for (i=0; i<am; i++) { 2351ed59401aSHong Zhang aj = a->j + a->diag[i]; 235239e3d630SHong Zhang ncols = ui[i+1] - ui[i]; 235339e3d630SHong Zhang for (j=0; j<ncols; j++) *cols++ = *aj++; 2354ed59401aSHong Zhang } 235539e3d630SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 2356b635d36bSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 2357b635d36bSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 2358b635d36bSHong Zhang 23597a48dd6fSHong Zhang /* initialization */ 23605a8e39fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 23617a48dd6fSHong Zhang 23627a48dd6fSHong Zhang /* jl: linked list for storing indices of the pivot rows 23637a48dd6fSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 23641393bc97SHong Zhang ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr); 23657a48dd6fSHong Zhang il = jl + am; 23667a48dd6fSHong Zhang uj_ptr = (PetscInt**)(il + am); 23677a48dd6fSHong Zhang uj_lvl_ptr = (PetscInt**)(uj_ptr + am); 23687a48dd6fSHong Zhang for (i=0; i<am; i++){ 23697a48dd6fSHong Zhang jl[i] = am; il[i] = 0; 23707a48dd6fSHong Zhang } 23717a48dd6fSHong Zhang 23727a48dd6fSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 23737a48dd6fSHong Zhang nlnk = am + 1; 23742ed133dbSHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 23757a48dd6fSHong Zhang 23767a48dd6fSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2377a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 23787a48dd6fSHong Zhang current_space = free_space; 2379a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 23807a48dd6fSHong Zhang current_space_lvl = free_space_lvl; 23817a48dd6fSHong Zhang 23827a48dd6fSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 23837a48dd6fSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 23847a48dd6fSHong Zhang nzk = 0; 2385622e2028SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 23863b4a8b6dSBarry Smith if (!ncols) SETERRQ2(PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 2387622e2028SHong Zhang ncols_upper = 0; 2388622e2028SHong Zhang for (j=0; j<ncols; j++){ 2389b635d36bSHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 2390b635d36bSHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 23915a8e39fbSHong Zhang ajtmp[ncols_upper] = i; 2392622e2028SHong Zhang ncols_upper++; 2393622e2028SHong Zhang } 2394622e2028SHong Zhang } 2395b635d36bSHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 23967a48dd6fSHong Zhang nzk += nlnk; 23977a48dd6fSHong Zhang 23987a48dd6fSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 23997a48dd6fSHong Zhang prow = jl[k]; /* 1st pivot row */ 24007a48dd6fSHong Zhang 24017a48dd6fSHong Zhang while (prow < k){ 24027a48dd6fSHong Zhang nextprow = jl[prow]; 24037a48dd6fSHong Zhang 24047a48dd6fSHong Zhang /* merge prow into k-th row */ 24057a48dd6fSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 24067a48dd6fSHong Zhang jmax = ui[prow+1]; 24077a48dd6fSHong Zhang ncols = jmax-jmin; 2408ed59401aSHong Zhang i = jmin - ui[prow]; 2409ed59401aSHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 24105a8e39fbSHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 24115a8e39fbSHong Zhang j = *(uj - 1); 24125a8e39fbSHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 24137a48dd6fSHong Zhang nzk += nlnk; 24147a48dd6fSHong Zhang 24157a48dd6fSHong Zhang /* update il and jl for prow */ 24167a48dd6fSHong Zhang if (jmin < jmax){ 24177a48dd6fSHong Zhang il[prow] = jmin; 24187a48dd6fSHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 24197a48dd6fSHong Zhang } 24207a48dd6fSHong Zhang prow = nextprow; 24217a48dd6fSHong Zhang } 24227a48dd6fSHong Zhang 24237a48dd6fSHong Zhang /* if free space is not available, make more free space */ 24247a48dd6fSHong Zhang if (current_space->local_remaining<nzk) { 24257a48dd6fSHong Zhang i = am - k + 1; /* num of unfactored rows */ 24267a48dd6fSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2427a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 2428a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 24297a48dd6fSHong Zhang reallocs++; 24307a48dd6fSHong Zhang } 24317a48dd6fSHong Zhang 24327a48dd6fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 2433b635d36bSHong Zhang if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 24342ed133dbSHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 24357a48dd6fSHong Zhang 24367a48dd6fSHong Zhang /* add the k-th row into il and jl */ 243739e3d630SHong Zhang if (nzk > 1){ 24387a48dd6fSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 24397a48dd6fSHong Zhang jl[k] = jl[i]; jl[i] = k; 24407a48dd6fSHong Zhang il[k] = ui[k] + 1; 24417a48dd6fSHong Zhang } 24427a48dd6fSHong Zhang uj_ptr[k] = current_space->array; 24437a48dd6fSHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 24447a48dd6fSHong Zhang 24457a48dd6fSHong Zhang current_space->array += nzk; 24467a48dd6fSHong Zhang current_space->local_used += nzk; 24477a48dd6fSHong Zhang current_space->local_remaining -= nzk; 24487a48dd6fSHong Zhang 24497a48dd6fSHong Zhang current_space_lvl->array += nzk; 24507a48dd6fSHong Zhang current_space_lvl->local_used += nzk; 24517a48dd6fSHong Zhang current_space_lvl->local_remaining -= nzk; 24527a48dd6fSHong Zhang 24537a48dd6fSHong Zhang ui[k+1] = ui[k] + nzk; 24547a48dd6fSHong Zhang } 24557a48dd6fSHong Zhang 24566cf91177SBarry Smith #if defined(PETSC_USE_INFO) 24577a48dd6fSHong Zhang if (ai[am] != 0) { 245839e3d630SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 2459ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2460ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2461ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 24627a48dd6fSHong Zhang } else { 2463ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 24647a48dd6fSHong Zhang } 246563ba0a88SBarry Smith #endif 24667a48dd6fSHong Zhang 24677a48dd6fSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 2468b635d36bSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 24697a48dd6fSHong Zhang ierr = PetscFree(jl);CHKERRQ(ierr); 24705a8e39fbSHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 24717a48dd6fSHong Zhang 24727a48dd6fSHong Zhang /* destroy list of free space and other temporary array(s) */ 24737a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2474a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 24752ed133dbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2476a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 24777a48dd6fSHong Zhang 247839e3d630SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 247939e3d630SHong Zhang 24807a48dd6fSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 24817a48dd6fSHong Zhang 2482719d5645SBarry Smith b = (Mat_SeqSBAIJ*)(fact)->data; 24837a48dd6fSHong Zhang b->singlemalloc = PETSC_FALSE; 24847a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 24857a48dd6fSHong Zhang b->j = uj; 24867a48dd6fSHong Zhang b->i = ui; 24875f44c854SHong Zhang b->diag = udiag; 24887f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 24897a48dd6fSHong Zhang b->ilen = 0; 24907a48dd6fSHong Zhang b->imax = 0; 24917a48dd6fSHong Zhang b->row = perm; 2492b635d36bSHong Zhang b->col = perm; 2493b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2494b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2495b635d36bSHong Zhang b->icol = iperm; 24967a48dd6fSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 24977a48dd6fSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2498719d5645SBarry Smith ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 24997a48dd6fSHong Zhang b->maxnz = b->nz = ui[am]; 2500e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2501e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 25027a48dd6fSHong Zhang 2503719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2504719d5645SBarry Smith (fact)->info.fill_ratio_given = fill; 25057a48dd6fSHong Zhang if (ai[am] != 0) { 2506719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 25077a48dd6fSHong Zhang } else { 2508719d5645SBarry Smith (fact)->info.fill_ratio_needed = 0.0; 25097a48dd6fSHong Zhang } 2510719d5645SBarry Smith (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 2511a6175056SHong Zhang PetscFunctionReturn(0); 2512a6175056SHong Zhang } 2513d5d45c9bSBarry Smith 2514f76d2b81SHong Zhang #undef __FUNCT__ 2515f76d2b81SHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ" 25160481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2517f76d2b81SHong Zhang { 2518f76d2b81SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 251910c27e3dSHong Zhang Mat_SeqSBAIJ *b; 2520dfbe8321SBarry Smith PetscErrorCode ierr; 2521f76d2b81SHong Zhang PetscTruth perm_identity; 252210c27e3dSHong Zhang PetscReal fill = info->fill; 25235d0c19d7SBarry Smith const PetscInt *rip,*riip; 25245d0c19d7SBarry Smith PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 252510c27e3dSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 25262ed133dbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr; 2527a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 252810c27e3dSHong Zhang PetscBT lnkbt; 25292ed133dbSHong Zhang IS iperm; 2530f76d2b81SHong Zhang 2531f76d2b81SHong Zhang PetscFunctionBegin; 2532d0f46423SBarry 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); 25332ed133dbSHong Zhang /* check whether perm is the identity mapping */ 2534f76d2b81SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 25352ed133dbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 25362ed133dbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 25379bfd6278SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 253810c27e3dSHong Zhang 253910c27e3dSHong Zhang /* initialization */ 25401393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 254110c27e3dSHong Zhang ui[0] = 0; 254210c27e3dSHong Zhang 254310c27e3dSHong Zhang /* jl: linked list for storing indices of the pivot rows 25441393bc97SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 25451393bc97SHong Zhang ierr = PetscMalloc((3*am+1)*sizeof(PetscInt)+am*sizeof(PetscInt**),&jl);CHKERRQ(ierr); 25461393bc97SHong Zhang il = jl + am; 25471393bc97SHong Zhang cols = il + am; 25481393bc97SHong Zhang ui_ptr = (PetscInt**)(cols + am); 25491393bc97SHong Zhang for (i=0; i<am; i++){ 25501393bc97SHong Zhang jl[i] = am; il[i] = 0; 2551f76d2b81SHong Zhang } 255210c27e3dSHong Zhang 255310c27e3dSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 25541393bc97SHong Zhang nlnk = am + 1; 25551393bc97SHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 255610c27e3dSHong Zhang 25571393bc97SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2558a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 255910c27e3dSHong Zhang current_space = free_space; 256010c27e3dSHong Zhang 25611393bc97SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 256210c27e3dSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 256310c27e3dSHong Zhang nzk = 0; 25642ed133dbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 25653b4a8b6dSBarry Smith if (!ncols) SETERRQ2(PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 25662ed133dbSHong Zhang ncols_upper = 0; 25672ed133dbSHong Zhang for (j=0; j<ncols; j++){ 25689bfd6278SHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 25692ed133dbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 25702ed133dbSHong Zhang cols[ncols_upper] = i; 25712ed133dbSHong Zhang ncols_upper++; 25722ed133dbSHong Zhang } 25732ed133dbSHong Zhang } 25741393bc97SHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 257510c27e3dSHong Zhang nzk += nlnk; 257610c27e3dSHong Zhang 257710c27e3dSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 257810c27e3dSHong Zhang prow = jl[k]; /* 1st pivot row */ 257910c27e3dSHong Zhang 258010c27e3dSHong Zhang while (prow < k){ 258110c27e3dSHong Zhang nextprow = jl[prow]; 258210c27e3dSHong Zhang /* merge prow into k-th row */ 25831393bc97SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 258410c27e3dSHong Zhang jmax = ui[prow+1]; 258510c27e3dSHong Zhang ncols = jmax-jmin; 25861393bc97SHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 25875a8e39fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 258810c27e3dSHong Zhang nzk += nlnk; 258910c27e3dSHong Zhang 259010c27e3dSHong Zhang /* update il and jl for prow */ 259110c27e3dSHong Zhang if (jmin < jmax){ 259210c27e3dSHong Zhang il[prow] = jmin; 25932ed133dbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 259410c27e3dSHong Zhang } 259510c27e3dSHong Zhang prow = nextprow; 259610c27e3dSHong Zhang } 259710c27e3dSHong Zhang 259810c27e3dSHong Zhang /* if free space is not available, make more free space */ 259910c27e3dSHong Zhang if (current_space->local_remaining<nzk) { 26001393bc97SHong Zhang i = am - k + 1; /* num of unfactored rows */ 260110c27e3dSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2602a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 260310c27e3dSHong Zhang reallocs++; 260410c27e3dSHong Zhang } 260510c27e3dSHong Zhang 260610c27e3dSHong Zhang /* copy data into free space, then initialize lnk */ 26071393bc97SHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 260810c27e3dSHong Zhang 260910c27e3dSHong Zhang /* add the k-th row into il and jl */ 261010c27e3dSHong Zhang if (nzk-1 > 0){ 26111393bc97SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 261210c27e3dSHong Zhang jl[k] = jl[i]; jl[i] = k; 261310c27e3dSHong Zhang il[k] = ui[k] + 1; 261410c27e3dSHong Zhang } 26152ed133dbSHong Zhang ui_ptr[k] = current_space->array; 261610c27e3dSHong Zhang current_space->array += nzk; 261710c27e3dSHong Zhang current_space->local_used += nzk; 261810c27e3dSHong Zhang current_space->local_remaining -= nzk; 261910c27e3dSHong Zhang 262010c27e3dSHong Zhang ui[k+1] = ui[k] + nzk; 262110c27e3dSHong Zhang } 262210c27e3dSHong Zhang 26236cf91177SBarry Smith #if defined(PETSC_USE_INFO) 26241393bc97SHong Zhang if (ai[am] != 0) { 26251393bc97SHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 2626ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2627ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2628ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 262910c27e3dSHong Zhang } else { 2630ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 263110c27e3dSHong Zhang } 263263ba0a88SBarry Smith #endif 263310c27e3dSHong Zhang 263410c27e3dSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 26359bfd6278SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 263610c27e3dSHong Zhang ierr = PetscFree(jl);CHKERRQ(ierr); 263710c27e3dSHong Zhang 263810c27e3dSHong Zhang /* destroy list of free space and other temporary array(s) */ 26391393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2640a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 264110c27e3dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 264210c27e3dSHong Zhang 264310c27e3dSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 264410c27e3dSHong Zhang 2645719d5645SBarry Smith b = (Mat_SeqSBAIJ*)(fact)->data; 264610c27e3dSHong Zhang b->singlemalloc = PETSC_FALSE; 2647e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2648e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 26491393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 265010c27e3dSHong Zhang b->j = uj; 265110c27e3dSHong Zhang b->i = ui; 265210c27e3dSHong Zhang b->diag = 0; 265310c27e3dSHong Zhang b->ilen = 0; 265410c27e3dSHong Zhang b->imax = 0; 265510c27e3dSHong Zhang b->row = perm; 26569bfd6278SHong Zhang b->col = perm; 26579bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 26589bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 26599bfd6278SHong Zhang b->icol = iperm; 266010c27e3dSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 26611393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2662719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 26631393bc97SHong Zhang b->maxnz = b->nz = ui[am]; 266410c27e3dSHong Zhang 2665719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2666719d5645SBarry Smith (fact)->info.fill_ratio_given = fill; 26671393bc97SHong Zhang if (ai[am] != 0) { 2668719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 266910c27e3dSHong Zhang } else { 2670719d5645SBarry Smith (fact)->info.fill_ratio_needed = 0.0; 267110c27e3dSHong Zhang } 2672719d5645SBarry Smith (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 2673f76d2b81SHong Zhang PetscFunctionReturn(0); 2674f76d2b81SHong Zhang } 2675599ef60dSHong Zhang 2676599ef60dSHong Zhang #undef __FUNCT__ 2677894a2336SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_newdatastruct" 2678894a2336SHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat A,Vec bb,Vec xx) 2679599ef60dSHong Zhang { 26801d098868SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 26811d098868SHong Zhang PetscErrorCode ierr; 26821d098868SHong Zhang PetscInt n = A->rmap->n; 26838fc3a347SHong Zhang const PetscInt *ai = a->i,*aj = a->j,*vi; 26841d098868SHong Zhang PetscScalar *x,sum; 26851d098868SHong Zhang const PetscScalar *b; 26861d098868SHong Zhang const MatScalar *aa = a->a,*v; 26878fc3a347SHong Zhang PetscInt i,nz; 26881d098868SHong Zhang 26891d098868SHong Zhang PetscFunctionBegin; 26901d098868SHong Zhang if (!n) PetscFunctionReturn(0); 26911d098868SHong Zhang 26921d098868SHong Zhang ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 26931d098868SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 26941d098868SHong Zhang 26951d098868SHong Zhang /* forward solve the lower triangular */ 26961d098868SHong Zhang x[0] = b[0]; 26971d098868SHong Zhang v = aa; 26981d098868SHong Zhang vi = aj; 26991d098868SHong Zhang for (i=1; i<n; i++) { 27001d098868SHong Zhang nz = ai[i+1] - ai[i]; 27011d098868SHong Zhang sum = b[i]; 2702003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 2703e0e43300SBarry Smith v += nz; 2704e0e43300SBarry Smith vi += nz; 27051d098868SHong Zhang x[i] = sum; 27061d098868SHong Zhang } 27071d098868SHong Zhang 27081d098868SHong Zhang /* backward solve the upper triangular */ 27098fc3a347SHong Zhang v = aa + ai[n+1]; 27108fc3a347SHong Zhang vi = aj + ai[n+1]; 27111d098868SHong Zhang for (i=n-1; i>=0; i--){ 27128fc3a347SHong Zhang nz = ai[2*n-i +1] - ai[2*n-i]-1; 27131d098868SHong Zhang sum = x[i]; 2714003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 2715e0e43300SBarry Smith v += nz; 27168fc3a347SHong Zhang vi += nz; vi++; 2717fe31b70dSHong Zhang x[i] = *v++ *sum; /* x[i]=aa[adiag[i]]*sum; v++; */ 27181d098868SHong Zhang } 27191d098868SHong Zhang 27201d098868SHong Zhang ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 27211d098868SHong Zhang ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 27221d098868SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 27231d098868SHong Zhang PetscFunctionReturn(0); 27241d098868SHong Zhang } 27251d098868SHong Zhang 27261d098868SHong Zhang #undef __FUNCT__ 2727894a2336SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_newdatastruct" 2728894a2336SHong Zhang PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec xx) 27291d098868SHong Zhang { 27301d098868SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 27311d098868SHong Zhang IS iscol = a->col,isrow = a->row; 27321d098868SHong Zhang PetscErrorCode ierr; 2733894a2336SHong Zhang PetscInt i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,nz,k; 27341d098868SHong Zhang const PetscInt *rout,*cout,*r,*c; 2735894a2336SHong Zhang PetscScalar *x,*tmp,*tmps,sum; 27361d098868SHong Zhang const PetscScalar *b; 27371d098868SHong Zhang const MatScalar *aa = a->a,*v; 27381d098868SHong Zhang 27391d098868SHong Zhang PetscFunctionBegin; 27401d098868SHong Zhang if (!n) PetscFunctionReturn(0); 27411d098868SHong Zhang 27421d098868SHong Zhang ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 27431d098868SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 27441d098868SHong Zhang tmp = a->solve_work; 27451d098868SHong Zhang 27461d098868SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 2747894a2336SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 27481d098868SHong Zhang 27491d098868SHong Zhang /* forward solve the lower triangular */ 27501d098868SHong Zhang tmp[0] = b[*r++]; 27511d098868SHong Zhang tmps = tmp; 27521d098868SHong Zhang v = aa; 27531d098868SHong Zhang vi = aj; 27541d098868SHong Zhang for (i=1; i<n; i++) { 27551d098868SHong Zhang nz = ai[i+1] - ai[i]; 2756894a2336SHong Zhang sum = b[*r++]; 2757894a2336SHong Zhang PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 2758894a2336SHong Zhang tmp[i] = sum; 27591d098868SHong Zhang v += nz; vi += nz; 27601d098868SHong Zhang } 27611d098868SHong Zhang 27621d098868SHong Zhang /* backward solve the upper triangular */ 2763894a2336SHong Zhang k = n+1; 2764894a2336SHong Zhang v = aa + ai[k]; /* 1st entry of U(n-1,:) */ 2765894a2336SHong Zhang vi = aj + ai[k]; 27661d098868SHong Zhang for (i=n-1; i>=0; i--){ 2767894a2336SHong Zhang k = 2*n-i; 2768894a2336SHong Zhang nz = ai[k +1] - ai[k] - 1; 2769894a2336SHong Zhang sum = tmp[i]; 2770894a2336SHong Zhang PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 2771894a2336SHong Zhang x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 27721d098868SHong Zhang v += nz+1; vi += nz+1; 27731d098868SHong Zhang } 27741d098868SHong Zhang 27751d098868SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 27761d098868SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 27771d098868SHong Zhang ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 27781d098868SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 27791d098868SHong Zhang ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 27801d098868SHong Zhang PetscFunctionReturn(0); 27811d098868SHong Zhang } 27821d098868SHong Zhang 27831d098868SHong Zhang #undef __FUNCT__ 27841d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ" 27851d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact) 27861d098868SHong Zhang { 27871d098868SHong Zhang Mat B = *fact; 2788599ef60dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b; 2789599ef60dSHong Zhang IS isicol; 2790599ef60dSHong Zhang PetscErrorCode ierr; 27911d098868SHong Zhang const PetscInt *r,*ic; 27921d098868SHong Zhang PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag; 2793dc3a2fd3SHong Zhang PetscInt *bi,*bj,*bdiag,*bdiag_rev; 2794f61a948aSLisandro Dalcin PetscInt row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au; 27951d098868SHong Zhang PetscInt nlnk,*lnk; 27961d098868SHong Zhang PetscBT lnkbt; 27971d098868SHong Zhang PetscTruth row_identity,icol_identity,both_identity; 27988aee2decSHong Zhang MatScalar *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp; 27991d098868SHong Zhang const PetscInt *ics; 28001d098868SHong Zhang PetscInt j,nz,*pj,*bjtmp,k,ncut,*jtmp; 2801f61a948aSLisandro Dalcin PetscReal dt=info->dt,dtcol=info->dtcol,shift=info->shiftinblocks; 2802f61a948aSLisandro Dalcin PetscInt dtcount=(PetscInt)info->dtcount,nnz_max; 2803599ef60dSHong Zhang PetscTruth missing; 2804599ef60dSHong Zhang 2805599ef60dSHong Zhang PetscFunctionBegin; 2806f61a948aSLisandro Dalcin 2807f61a948aSLisandro Dalcin if (dt == PETSC_DEFAULT) dt = 0.005; 2808f61a948aSLisandro Dalcin if (dtcol == PETSC_DEFAULT) dtcol = 0.01; /* XXX unused! */ 2809f61a948aSLisandro Dalcin if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax); 2810f61a948aSLisandro Dalcin 28111d098868SHong Zhang /* ------- symbolic factorization, can be reused ---------*/ 28121d098868SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 28131d098868SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 28141d098868SHong Zhang adiag=a->diag; 2815599ef60dSHong Zhang 2816599ef60dSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 2817599ef60dSHong Zhang 2818599ef60dSHong Zhang /* bdiag is location of diagonal in factor */ 2819a6226a1aSHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 2820a6226a1aSHong Zhang bdiag_rev = bdiag + n+1; 2821599ef60dSHong Zhang 28221d098868SHong Zhang /* allocate row pointers bi */ 28238fc3a347SHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 28241d098868SHong Zhang 28251d098868SHong Zhang /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */ 2826393d3a68SHong Zhang if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */ 28271d098868SHong Zhang nnz_max = ai[n]+2*n*dtcount+2; 28288fc3a347SHong Zhang 28298fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 28308fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr); 2831599ef60dSHong Zhang 2832599ef60dSHong Zhang /* put together the new matrix */ 2833599ef60dSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 2834599ef60dSHong Zhang ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 2835599ef60dSHong Zhang b = (Mat_SeqAIJ*)(B)->data; 2836599ef60dSHong Zhang b->free_a = PETSC_TRUE; 2837599ef60dSHong Zhang b->free_ij = PETSC_TRUE; 2838599ef60dSHong Zhang b->singlemalloc = PETSC_FALSE; 2839599ef60dSHong Zhang b->a = ba; 2840599ef60dSHong Zhang b->j = bj; 2841599ef60dSHong Zhang b->i = bi; 2842599ef60dSHong Zhang b->diag = bdiag; 2843599ef60dSHong Zhang b->ilen = 0; 2844599ef60dSHong Zhang b->imax = 0; 2845599ef60dSHong Zhang b->row = isrow; 2846599ef60dSHong Zhang b->col = iscol; 2847599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2848599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 2849599ef60dSHong Zhang b->icol = isicol; 2850599ef60dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2851599ef60dSHong Zhang 28521d098868SHong Zhang ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 28531d098868SHong Zhang b->maxnz = nnz_max; 2854599ef60dSHong Zhang 2855599ef60dSHong Zhang (B)->factor = MAT_FACTOR_ILUDT; 2856599ef60dSHong Zhang (B)->info.factor_mallocs = 0; 28571d098868SHong Zhang (B)->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]); 28581d098868SHong Zhang CHKMEMQ; 28591d098868SHong Zhang /* ------- end of symbolic factorization ---------*/ 2860599ef60dSHong Zhang 2861599ef60dSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 2862599ef60dSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 2863599ef60dSHong Zhang ics = ic; 2864599ef60dSHong Zhang 2865599ef60dSHong Zhang /* linked list for storing column indices of the active row */ 2866599ef60dSHong Zhang nlnk = n + 1; 2867599ef60dSHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28681d098868SHong Zhang 28691d098868SHong Zhang /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */ 28701d098868SHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(PetscInt),&im);CHKERRQ(ierr); 28711d098868SHong Zhang jtmp = im + n; 28721d098868SHong Zhang /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */ 28738369b28dSHong Zhang ierr = PetscMalloc((2*n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 28748aee2decSHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 28758369b28dSHong Zhang vtmp = rtmp + n; 2876599ef60dSHong Zhang 2877599ef60dSHong Zhang bi[0] = 0; 28788fc3a347SHong Zhang bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */ 2879dc3a2fd3SHong Zhang bdiag_rev[n] = bdiag[0]; 28808fc3a347SHong Zhang bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */ 2881599ef60dSHong Zhang for (i=0; i<n; i++) { 2882599ef60dSHong Zhang /* copy initial fill into linked list */ 2883599ef60dSHong Zhang nzi = 0; /* nonzeros for active row i */ 28848369b28dSHong Zhang nzi = ai[r[i]+1] - ai[r[i]]; 28858369b28dSHong Zhang if (!nzi) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 28868369b28dSHong Zhang nzi_al = adiag[r[i]] - ai[r[i]]; 28878369b28dSHong Zhang nzi_au = ai[r[i]+1] - adiag[r[i]] -1; 2888599ef60dSHong Zhang ajtmp = aj + ai[r[i]]; 28898369b28dSHong Zhang ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 2890599ef60dSHong Zhang 2891599ef60dSHong Zhang /* load in initial (unfactored row) */ 28921d098868SHong Zhang aatmp = a->a + ai[r[i]]; 28938369b28dSHong Zhang for (j=0; j<nzi; j++) { 28941d098868SHong Zhang rtmp[ics[*ajtmp++]] = *aatmp++; 2895599ef60dSHong Zhang } 2896599ef60dSHong Zhang 2897599ef60dSHong Zhang /* add pivot rows into linked list */ 2898599ef60dSHong Zhang row = lnk[n]; 2899599ef60dSHong Zhang while (row < i ) { 29001d098868SHong Zhang nzi_bl = bi[row+1] - bi[row] + 1; 29011d098868SHong Zhang bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */ 29021d098868SHong Zhang ierr = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr); 2903599ef60dSHong Zhang nzi += nlnk; 2904599ef60dSHong Zhang row = lnk[row]; 2905599ef60dSHong Zhang } 2906599ef60dSHong Zhang 29078369b28dSHong Zhang /* copy data from lnk into jtmp, then initialize lnk */ 29088369b28dSHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr); 2909599ef60dSHong Zhang 2910599ef60dSHong Zhang /* numerical factorization */ 29118369b28dSHong Zhang bjtmp = jtmp; 2912599ef60dSHong Zhang row = *bjtmp++; /* 1st pivot row */ 2913599ef60dSHong Zhang while ( row < i ) { 2914599ef60dSHong Zhang pc = rtmp + row; 29153c2a7987SHong Zhang pv = ba + bdiag[row]; /* 1./(diag of the pivot row) */ 29163c2a7987SHong Zhang multiplier = (*pc) * (*pv); 2917599ef60dSHong Zhang *pc = multiplier; 29183c2a7987SHong Zhang if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */ 29191d098868SHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 29201d098868SHong Zhang pv = ba + bdiag[row+1] + 1; 29211d098868SHong Zhang /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */ 29221d098868SHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 29231d098868SHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 2924dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 2925599ef60dSHong Zhang } 2926599ef60dSHong Zhang row = *bjtmp++; 2927599ef60dSHong Zhang } 2928599ef60dSHong Zhang 29298369b28dSHong Zhang /* copy sparse rtmp into contiguous vtmp; separate L and U part */ 29308aee2decSHong Zhang diag_tmp = rtmp[i]; /* save diagonal value - may not needed?? */ 29318369b28dSHong Zhang nzi_bl = 0; j = 0; 29328369b28dSHong Zhang while (jtmp[j] < i){ /* Note: jtmp is sorted */ 29338aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 29348369b28dSHong Zhang nzi_bl++; j++; 29358369b28dSHong Zhang } 29368369b28dSHong Zhang nzi_bu = nzi - nzi_bl -1; 29378369b28dSHong Zhang while (j < nzi){ 29388aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 29398369b28dSHong Zhang j++; 29408369b28dSHong Zhang } 29418369b28dSHong Zhang 2942599ef60dSHong Zhang bjtmp = bj + bi[i]; 2943599ef60dSHong Zhang batmp = ba + bi[i]; 29448369b28dSHong Zhang /* apply level dropping rule to L part */ 29458369b28dSHong Zhang ncut = nzi_al + dtcount; 29468369b28dSHong Zhang if (ncut < nzi_bl){ 29478369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr); 29488369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr); 2949599ef60dSHong Zhang } else { 29508369b28dSHong Zhang ncut = nzi_bl; 29518369b28dSHong Zhang } 29528369b28dSHong Zhang for (j=0; j<ncut; j++){ 29538369b28dSHong Zhang bjtmp[j] = jtmp[j]; 29548369b28dSHong Zhang batmp[j] = vtmp[j]; 29556da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */ 29568369b28dSHong Zhang } 29571d098868SHong Zhang bi[i+1] = bi[i] + ncut; 29588369b28dSHong Zhang nzi = ncut + 1; 29598369b28dSHong Zhang 29608369b28dSHong Zhang /* apply level dropping rule to U part */ 29618369b28dSHong Zhang ncut = nzi_au + dtcount; 29628369b28dSHong Zhang if (ncut < nzi_bu){ 29638369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr); 29648369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr); 29658369b28dSHong Zhang } else { 29668369b28dSHong Zhang ncut = nzi_bu; 29678369b28dSHong Zhang } 29688369b28dSHong Zhang nzi += ncut; 29691d098868SHong Zhang 29701d098868SHong Zhang /* mark bdiagonal */ 29711d098868SHong Zhang bdiag[i+1] = bdiag[i] - (ncut + 1); 2972dc3a2fd3SHong Zhang bdiag_rev[n-i-1] = bdiag[i+1]; 29738fc3a347SHong Zhang bi[2*n - i] = bi[2*n - i +1] - (ncut + 1); 29741d098868SHong Zhang bjtmp = bj + bdiag[i]; 29751d098868SHong Zhang batmp = ba + bdiag[i]; 29761d098868SHong Zhang *bjtmp = i; 29778aee2decSHong Zhang *batmp = diag_tmp; /* rtmp[i]; */ 2978c9c72f8fSHong Zhang if (*batmp == 0.0) { 2979c9c72f8fSHong Zhang *batmp = dt+shift; 2980bd1bc851SBarry Smith /* printf(" row %d add shift %g\n",i,shift); */ 2981c9c72f8fSHong Zhang } 2982b78a477dSHong Zhang *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */ 29836da515a1SHong Zhang /* printf(" (%d,%g),",*bjtmp,*batmp); */ 29841d098868SHong Zhang 29851d098868SHong Zhang bjtmp = bj + bdiag[i+1]+1; 29861d098868SHong Zhang batmp = ba + bdiag[i+1]+1; 29878369b28dSHong Zhang for (k=0; k<ncut; k++){ 29881d098868SHong Zhang bjtmp[k] = jtmp[nzi_bl+1+k]; 29891d098868SHong Zhang batmp[k] = vtmp[nzi_bl+1+k]; 29906da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */ 2991599ef60dSHong Zhang } 29926da515a1SHong Zhang /* printf("\n"); */ 2993599ef60dSHong Zhang 2994599ef60dSHong Zhang im[i] = nzi; /* used by PetscLLAddSortedLU() */ 2995599ef60dSHong Zhang /* 29961d098868SHong Zhang printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]); 29978369b28dSHong Zhang printf(" ----------------------------\n"); 2998599ef60dSHong Zhang */ 29998369b28dSHong Zhang } /* for (i=0; i<n; i++) */ 30001d098868SHong Zhang /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */ 30011d098868SHong Zhang if (bi[n] >= bdiag[n]) SETERRQ2(PETSC_ERR_ARG_SIZ,"end of L array %d cannot >= the beginning of U array %d",bi[n],bdiag[n]); 3002599ef60dSHong Zhang 3003599ef60dSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3004599ef60dSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3005599ef60dSHong Zhang 3006599ef60dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 3007599ef60dSHong Zhang ierr = PetscFree(im);CHKERRQ(ierr); 3008599ef60dSHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 3009599ef60dSHong Zhang 3010599ef60dSHong Zhang ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr); 30111d098868SHong Zhang b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n]; 3012599ef60dSHong Zhang 3013599ef60dSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 3014599ef60dSHong Zhang ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr); 3015599ef60dSHong Zhang both_identity = (PetscTruth) (row_identity && icol_identity); 3016599ef60dSHong Zhang if (row_identity && icol_identity) { 3017894a2336SHong Zhang B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct; 3018599ef60dSHong Zhang } else { 3019894a2336SHong Zhang B->ops->solve = MatSolve_SeqAIJ_newdatastruct; 3020599ef60dSHong Zhang } 3021599ef60dSHong Zhang 3022b78a477dSHong Zhang B->ops->lufactorsymbolic = MatILUDTFactorSymbolic_SeqAIJ; 3023b78a477dSHong Zhang B->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ; 30241d098868SHong Zhang B->ops->solveadd = 0; 30251d098868SHong Zhang B->ops->solvetranspose = 0; 30261d098868SHong Zhang B->ops->solvetransposeadd = 0; 30271d098868SHong Zhang B->ops->matsolve = 0; 3028599ef60dSHong Zhang B->assembled = PETSC_TRUE; 3029599ef60dSHong Zhang B->preallocated = PETSC_TRUE; 3030599ef60dSHong Zhang PetscFunctionReturn(0); 3031599ef60dSHong Zhang } 3032599ef60dSHong Zhang 30333c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */ 30343c2a7987SHong Zhang #undef __FUNCT__ 30353c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ" 30363c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info) 30373c2a7987SHong Zhang { 30383c2a7987SHong Zhang PetscErrorCode ierr; 30393c2a7987SHong Zhang 30403c2a7987SHong Zhang PetscFunctionBegin; 30413c2a7987SHong Zhang ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr); 30423c2a7987SHong Zhang 30433c2a7987SHong Zhang fact->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ; 30443c2a7987SHong Zhang PetscFunctionReturn(0); 30453c2a7987SHong Zhang } 30463c2a7987SHong Zhang 30473c2a7987SHong Zhang /* 30483c2a7987SHong Zhang same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors 30493c2a7987SHong Zhang - intend to replace existing MatLUFactorNumeric_SeqAIJ() 30503c2a7987SHong Zhang */ 30513c2a7987SHong Zhang #undef __FUNCT__ 30523c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ" 30533c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info) 30543c2a7987SHong Zhang { 30553c2a7987SHong Zhang Mat C=fact; 30563c2a7987SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 30573c2a7987SHong Zhang IS isrow = b->row,isicol = b->icol; 30583c2a7987SHong Zhang PetscErrorCode ierr; 30593c2a7987SHong Zhang const PetscInt *r,*ic,*ics; 3060b78a477dSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 3061b78a477dSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj; 30623c2a7987SHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 3063b78a477dSHong Zhang PetscReal dt=info->dt,shift=info->shiftinblocks; 3064b78a477dSHong Zhang PetscTruth row_identity, col_identity; 30653c2a7987SHong Zhang 30663c2a7987SHong Zhang PetscFunctionBegin; 30673c2a7987SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 30683c2a7987SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3069b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 30703c2a7987SHong Zhang ics = ic; 30713c2a7987SHong Zhang 30723c2a7987SHong Zhang for (i=0; i<n; i++){ 3073b78a477dSHong Zhang /* initialize rtmp array */ 3074b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of nozeros in L(i,:) */ 30753c2a7987SHong Zhang bjtmp = bj + bi[i]; 3076b78a477dSHong Zhang for (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0; 3077b78a477dSHong Zhang rtmp[i] = 0.0; 3078b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */ 3079b78a477dSHong Zhang bjtmp = bj + bdiag[i+1] + 1; 3080b78a477dSHong Zhang for (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0; 30813c2a7987SHong Zhang 3082b78a477dSHong Zhang /* load in initial unfactored row of A */ 30836da515a1SHong Zhang /* printf("row %d\n",i); */ 30843c2a7987SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 30853c2a7987SHong Zhang ajtmp = aj + ai[r[i]]; 30863c2a7987SHong Zhang v = aa + ai[r[i]]; 30873c2a7987SHong Zhang for (j=0; j<nz; j++) { 3088b78a477dSHong Zhang rtmp[ics[*ajtmp++]] = v[j]; 30896da515a1SHong Zhang /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */ 30903c2a7987SHong Zhang } 30916da515a1SHong Zhang /* printf("\n"); */ 30923c2a7987SHong Zhang 3093b78a477dSHong Zhang /* numerical factorization */ 3094b78a477dSHong Zhang bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */ 3095b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of entries in L(i,:) */ 3096b78a477dSHong Zhang k = 0; 3097b78a477dSHong Zhang while (k < nzl){ 30983c2a7987SHong Zhang row = *bjtmp++; 30996da515a1SHong Zhang /* printf(" prow %d\n",row); */ 31003c2a7987SHong Zhang pc = rtmp + row; 3101b78a477dSHong Zhang pv = b->a + bdiag[row]; /* 1./(diag of the pivot row) */ 3102b78a477dSHong Zhang multiplier = (*pc) * (*pv); 31033c2a7987SHong Zhang *pc = multiplier; 3104b78a477dSHong Zhang if (PetscAbsScalar(multiplier) > dt){ 3105b78a477dSHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 3106b78a477dSHong Zhang pv = b->a + bdiag[row+1] + 1; 3107b78a477dSHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 3108b78a477dSHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 31096da515a1SHong Zhang /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */ 31103c2a7987SHong Zhang } 3111b78a477dSHong Zhang k++; 31123c2a7987SHong Zhang } 31133c2a7987SHong Zhang 3114b78a477dSHong Zhang /* finished row so stick it into b->a */ 3115b78a477dSHong Zhang /* L-part */ 3116b78a477dSHong Zhang pv = b->a + bi[i] ; 3117b78a477dSHong Zhang pj = bj + bi[i] ; 3118b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; 3119b78a477dSHong Zhang for (j=0; j<nzl; j++) { 3120b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 31216da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 31223c2a7987SHong Zhang } 3123b78a477dSHong Zhang 3124b78a477dSHong Zhang /* diagonal: invert diagonal entries for simplier triangular solves */ 3125b78a477dSHong Zhang if (rtmp[i] == 0.0) rtmp[i] = dt+shift; 3126b78a477dSHong Zhang b->a[bdiag[i]] = 1.0/rtmp[i]; 31276da515a1SHong Zhang /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */ 3128b78a477dSHong Zhang 3129b78a477dSHong Zhang /* U-part */ 3130b78a477dSHong Zhang pv = b->a + bdiag[i+1] + 1; 3131b78a477dSHong Zhang pj = bj + bdiag[i+1] + 1; 3132b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1] - 1; 3133b78a477dSHong Zhang for (j=0; j<nzu; j++) { 3134b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 31356da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 3136b78a477dSHong Zhang } 31376da515a1SHong Zhang /* printf("\n"); */ 3138b78a477dSHong Zhang } 3139b78a477dSHong Zhang 31403c2a7987SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 31413c2a7987SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 31423c2a7987SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3143b78a477dSHong Zhang 31443c2a7987SHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 31453c2a7987SHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 31463c2a7987SHong Zhang if (row_identity && col_identity) { 3147894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct; 31483c2a7987SHong Zhang } else { 3149894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_newdatastruct; 31503c2a7987SHong Zhang } 3151b78a477dSHong Zhang C->ops->solveadd = 0; 3152b78a477dSHong Zhang C->ops->solvetranspose = 0; 3153b78a477dSHong Zhang C->ops->solvetransposeadd = 0; 3154b78a477dSHong Zhang C->ops->matsolve = 0; 31553c2a7987SHong Zhang C->assembled = PETSC_TRUE; 31563c2a7987SHong Zhang C->preallocated = PETSC_TRUE; 31573c2a7987SHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 31583c2a7987SHong Zhang PetscFunctionReturn(0); 31593c2a7987SHong Zhang } 3160