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++) { 2975567043SBarry Smith future = 0.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 = PetscMemzero(done,n*sizeof(PetscTruth));CHKERRQ(ierr); 420e83c824SBarry Smith ierr = PetscMalloc(n*sizeof(PetscInt),&order);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; 6875567043SBarry Smith future = 0.0; 6975567043SBarry Smith past = 0.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 } 1569b48462bSShri Abhyankar 157ce3d78c0SShri Abhyankar 158d0f46423SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square"); 1594c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 160ac121b3dSBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 161ac121b3dSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 162289bc588SBarry Smith 163289bc588SBarry Smith /* get new row pointers */ 1641393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 1651393bc97SHong Zhang bi[0] = 0; 1661393bc97SHong Zhang 1671393bc97SHong Zhang /* bdiag is location of diagonal in factor */ 1681393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1691393bc97SHong Zhang bdiag[0] = 0; 1701393bc97SHong Zhang 1711393bc97SHong Zhang /* linked list for storing column indices of the active row */ 1721393bc97SHong Zhang nlnk = n + 1; 1731393bc97SHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1741393bc97SHong Zhang 17535baf27eSSatish Balay ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 1761393bc97SHong Zhang 1771393bc97SHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 178d3d32019SBarry Smith f = info->fill; 179a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1801393bc97SHong Zhang current_space = free_space; 181289bc588SBarry Smith 182289bc588SBarry Smith for (i=0; i<n; i++) { 1831393bc97SHong Zhang /* copy previous fill into linked list */ 184289bc588SBarry Smith nzi = 0; 1851393bc97SHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 1863b4a8b6dSBarry 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); 1871393bc97SHong Zhang ajtmp = aj + ai[r[i]]; 188afcc9904SHong Zhang ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1891393bc97SHong Zhang nzi += nlnk; 1901393bc97SHong Zhang 1911393bc97SHong Zhang /* add pivot rows into linked list */ 1921393bc97SHong Zhang row = lnk[n]; 1931393bc97SHong Zhang while (row < i) { 194d1170560SHong Zhang nzbd = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */ 195d1170560SHong Zhang ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 196d1170560SHong Zhang ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 1971393bc97SHong Zhang nzi += nlnk; 1981393bc97SHong Zhang row = lnk[row]; 199289bc588SBarry Smith } 2001393bc97SHong Zhang bi[i+1] = bi[i] + nzi; 2011393bc97SHong Zhang im[i] = nzi; 2021393bc97SHong Zhang 2031393bc97SHong Zhang /* mark bdiag */ 2041393bc97SHong Zhang nzbd = 0; 2051393bc97SHong Zhang nnz = nzi; 2061393bc97SHong Zhang k = lnk[n]; 2071393bc97SHong Zhang while (nnz-- && k < i){ 2081393bc97SHong Zhang nzbd++; 2091393bc97SHong Zhang k = lnk[k]; 2101393bc97SHong Zhang } 2111393bc97SHong Zhang bdiag[i] = bi[i] + nzbd; 2121393bc97SHong Zhang 2131393bc97SHong Zhang /* if free space is not available, make more free space */ 2141393bc97SHong Zhang if (current_space->local_remaining<nzi) { 2151393bc97SHong Zhang nnz = (n - i)*nzi; /* estimated and max additional space needed */ 216a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2171393bc97SHong Zhang reallocs++; 2181393bc97SHong Zhang } 2191393bc97SHong Zhang 2201393bc97SHong Zhang /* copy data into free space, then initialize lnk */ 2211393bc97SHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 2221393bc97SHong Zhang bi_ptr[i] = current_space->array; 2231393bc97SHong Zhang current_space->array += nzi; 2241393bc97SHong Zhang current_space->local_used += nzi; 2251393bc97SHong Zhang current_space->local_remaining -= nzi; 226289bc588SBarry Smith } 2276cf91177SBarry Smith #if defined(PETSC_USE_INFO) 228464e8d44SSatish Balay if (ai[n] != 0) { 2291393bc97SHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 230ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 231ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 232ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 233ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 23405bf559cSSatish Balay } else { 235ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 23605bf559cSSatish Balay } 23763ba0a88SBarry Smith #endif 2382fb3ed76SBarry Smith 239898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 240898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2411987afe7SBarry Smith 2421393bc97SHong Zhang /* destroy list of free space and other temporary array(s) */ 2431393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 244a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); 2451393bc97SHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 24635baf27eSSatish Balay ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 247289bc588SBarry Smith 248289bc588SBarry Smith /* put together the new matrix */ 249719d5645SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 250719d5645SBarry Smith ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 251719d5645SBarry Smith b = (Mat_SeqAIJ*)(B)->data; 252e6b907acSBarry Smith b->free_a = PETSC_TRUE; 253e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 2547c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 2551393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 2561393bc97SHong Zhang b->j = bj; 2571393bc97SHong Zhang b->i = bi; 2581393bc97SHong Zhang b->diag = bdiag; 259416022c9SBarry Smith b->ilen = 0; 260416022c9SBarry Smith b->imax = 0; 261416022c9SBarry Smith b->row = isrow; 262416022c9SBarry Smith b->col = iscol; 263c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 264c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 26582bf6240SBarry Smith b->icol = isicol; 26687828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2671393bc97SHong Zhang 2681393bc97SHong Zhang /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 269719d5645SBarry Smith ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 2701393bc97SHong Zhang b->maxnz = b->nz = bi[n] ; 2717fd98bd6SLois Curfman McInnes 272719d5645SBarry Smith (B)->factor = MAT_FACTOR_LU; 273719d5645SBarry Smith (B)->info.factor_mallocs = reallocs; 274719d5645SBarry Smith (B)->info.fill_ratio_given = f; 275703deb49SSatish Balay 2769f7d0b68SHong Zhang if (ai[n]) { 277719d5645SBarry Smith (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 278eea2913aSSatish Balay } else { 279719d5645SBarry Smith (B)->info.fill_ratio_needed = 0.0; 280eea2913aSSatish Balay } 281719d5645SBarry Smith (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 282719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqAIJ; 283719d5645SBarry Smith (B)->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 284db4efbfdSBarry Smith /* switch to inodes if appropriate */ 2854108e4d5SBarry Smith ierr = MatLUFactorSymbolic_SeqAIJ_Inode(B,A,isrow,iscol,info);CHKERRQ(ierr); 2863a40ed3dSBarry Smith PetscFunctionReturn(0); 287289bc588SBarry Smith } 2881393bc97SHong Zhang 289ce3d78c0SShri Abhyankar #undef __FUNCT__ 290ce3d78c0SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_newdatastruct" 291ce3d78c0SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ_newdatastruct(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 292ce3d78c0SShri Abhyankar { 293ce3d78c0SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 294ce3d78c0SShri Abhyankar IS isicol; 295ce3d78c0SShri Abhyankar PetscErrorCode ierr; 296ce3d78c0SShri Abhyankar const PetscInt *r,*ic; 297ce3d78c0SShri Abhyankar PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 298ce3d78c0SShri Abhyankar PetscInt *bi,*bj,*ajtmp; 299ce3d78c0SShri Abhyankar PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 300ce3d78c0SShri Abhyankar PetscReal f; 301ce3d78c0SShri Abhyankar PetscInt nlnk,*lnk,k,**bi_ptr; 302ce3d78c0SShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 303ce3d78c0SShri Abhyankar PetscBT lnkbt; 304ce3d78c0SShri Abhyankar 305ce3d78c0SShri Abhyankar PetscFunctionBegin; 306ce3d78c0SShri Abhyankar if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square"); 307ce3d78c0SShri Abhyankar ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 308ce3d78c0SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 309ce3d78c0SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 310ce3d78c0SShri Abhyankar 3111bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 3121bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 3131bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 314a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 3159b48462bSShri Abhyankar 3169b48462bSShri Abhyankar /* linked list for storing column indices of the active row */ 3179b48462bSShri Abhyankar nlnk = n + 1; 3189b48462bSShri Abhyankar ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3199b48462bSShri Abhyankar 3209b48462bSShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 3219b48462bSShri Abhyankar 3229b48462bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 3239b48462bSShri Abhyankar f = info->fill; 3249b48462bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 3259b48462bSShri Abhyankar current_space = free_space; 3269b48462bSShri Abhyankar 3279b48462bSShri Abhyankar for (i=0; i<n; i++) { 3289b48462bSShri Abhyankar /* copy previous fill into linked list */ 3299b48462bSShri Abhyankar nzi = 0; 3309b48462bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 3319b48462bSShri 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); 3329b48462bSShri Abhyankar ajtmp = aj + ai[r[i]]; 3339b48462bSShri Abhyankar ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3349b48462bSShri Abhyankar nzi += nlnk; 3359b48462bSShri Abhyankar 3369b48462bSShri Abhyankar /* add pivot rows into linked list */ 3379b48462bSShri Abhyankar row = lnk[n]; 3389b48462bSShri Abhyankar while (row < i){ 3399b48462bSShri Abhyankar nzbd = bdiag[row] + 1; /* num of entries in the row with column index <= row */ 3409b48462bSShri Abhyankar ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 3419b48462bSShri Abhyankar ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 3429b48462bSShri Abhyankar nzi += nlnk; 3439b48462bSShri Abhyankar row = lnk[row]; 3449b48462bSShri Abhyankar } 3459b48462bSShri Abhyankar bi[i+1] = bi[i] + nzi; 3469b48462bSShri Abhyankar im[i] = nzi; 3479b48462bSShri Abhyankar 3489b48462bSShri Abhyankar /* mark bdiag */ 3499b48462bSShri Abhyankar nzbd = 0; 3509b48462bSShri Abhyankar nnz = nzi; 3519b48462bSShri Abhyankar k = lnk[n]; 3529b48462bSShri Abhyankar while (nnz-- && k < i){ 3539b48462bSShri Abhyankar nzbd++; 3549b48462bSShri Abhyankar k = lnk[k]; 3559b48462bSShri Abhyankar } 3569b48462bSShri Abhyankar bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */ 3579b48462bSShri Abhyankar 3589b48462bSShri Abhyankar /* if free space is not available, make more free space */ 3599b48462bSShri Abhyankar if (current_space->local_remaining<nzi) { 3609b48462bSShri Abhyankar nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */ 3619b48462bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 3629b48462bSShri Abhyankar reallocs++; 3639b48462bSShri Abhyankar } 3649b48462bSShri Abhyankar 3659b48462bSShri Abhyankar /* copy data into free space, then initialize lnk */ 3669b48462bSShri Abhyankar ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 3679b48462bSShri Abhyankar bi_ptr[i] = current_space->array; 3689b48462bSShri Abhyankar current_space->array += nzi; 3699b48462bSShri Abhyankar current_space->local_used += nzi; 3709b48462bSShri Abhyankar current_space->local_remaining -= nzi; 3719b48462bSShri Abhyankar } 3729b48462bSShri Abhyankar #if defined(PETSC_USE_INFO) 3739b48462bSShri Abhyankar if (ai[n] != 0) { 3749b48462bSShri Abhyankar PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 3759b48462bSShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 3769b48462bSShri Abhyankar ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 3779b48462bSShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 3789b48462bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 3799b48462bSShri Abhyankar } else { 3809b48462bSShri Abhyankar ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 3819b48462bSShri Abhyankar } 3829b48462bSShri Abhyankar #endif 3839b48462bSShri Abhyankar 3849b48462bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3859b48462bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3869b48462bSShri Abhyankar 3879b48462bSShri Abhyankar /* destroy list of free space and other temporary array(s) */ 3889b48462bSShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 3899b48462bSShri Abhyankar ierr = PetscFreeSpaceContiguous_LU_v2(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 3909b48462bSShri Abhyankar ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 3919b48462bSShri Abhyankar ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 3929b48462bSShri Abhyankar 3939b48462bSShri Abhyankar /* put together the new matrix */ 3949b48462bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 3959b48462bSShri Abhyankar ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 3969b48462bSShri Abhyankar b = (Mat_SeqAIJ*)(B)->data; 3979b48462bSShri Abhyankar b->free_a = PETSC_TRUE; 3989b48462bSShri Abhyankar b->free_ij = PETSC_TRUE; 3999b48462bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 4009b48462bSShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 4019b48462bSShri Abhyankar b->j = bj; 4029b48462bSShri Abhyankar b->i = bi; 4039b48462bSShri Abhyankar b->diag = bdiag; 4049b48462bSShri Abhyankar b->ilen = 0; 4059b48462bSShri Abhyankar b->imax = 0; 4069b48462bSShri Abhyankar b->row = isrow; 4079b48462bSShri Abhyankar b->col = iscol; 4089b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 4099b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 4109b48462bSShri Abhyankar b->icol = isicol; 4119b48462bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 4129b48462bSShri Abhyankar 4139b48462bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 4149b48462bSShri Abhyankar ierr = PetscLogObjectMemory(B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 4159b48462bSShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 4169b48462bSShri Abhyankar (B)->factor = MAT_FACTOR_LU; 4179b48462bSShri Abhyankar (B)->info.factor_mallocs = reallocs; 4189b48462bSShri Abhyankar (B)->info.fill_ratio_given = f; 4199b48462bSShri Abhyankar 4209b48462bSShri Abhyankar if (ai[n]) { 4219b48462bSShri Abhyankar (B)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 4229b48462bSShri Abhyankar } else { 4239b48462bSShri Abhyankar (B)->info.fill_ratio_needed = 0.0; 4249b48462bSShri Abhyankar } 425b4f5729fSShri Abhyankar (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 4269b48462bSShri Abhyankar PetscFunctionReturn(0); 4279b48462bSShri Abhyankar } 4289b48462bSShri Abhyankar 4295b5bc046SBarry Smith /* 4305b5bc046SBarry Smith Trouble in factorization, should we dump the original matrix? 4315b5bc046SBarry Smith */ 4325b5bc046SBarry Smith #undef __FUNCT__ 4335b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix" 4345b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A) 4355b5bc046SBarry Smith { 4365b5bc046SBarry Smith PetscErrorCode ierr; 43790d69ab7SBarry Smith PetscTruth flg = PETSC_FALSE; 4385b5bc046SBarry Smith 4395b5bc046SBarry Smith PetscFunctionBegin; 44090d69ab7SBarry Smith ierr = PetscOptionsGetTruth(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr); 4415b5bc046SBarry Smith if (flg) { 4425b5bc046SBarry Smith PetscViewer viewer; 4435b5bc046SBarry Smith char filename[PETSC_MAX_PATH_LEN]; 4445b5bc046SBarry Smith 4455b5bc046SBarry Smith ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr); 4467adad957SLisandro Dalcin ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 4475b5bc046SBarry Smith ierr = MatView(A,viewer);CHKERRQ(ierr); 4485b5bc046SBarry Smith ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr); 4495b5bc046SBarry Smith } 4505b5bc046SBarry Smith PetscFunctionReturn(0); 4515b5bc046SBarry Smith } 4525b5bc046SBarry Smith 4534108e4d5SBarry Smith extern PetscErrorCode MatSolve_SeqAIJ_Inode(Mat,Vec,Vec); 454db4efbfdSBarry Smith 455c9c72f8fSHong Zhang #undef __FUNCT__ 456c9c72f8fSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_newdatastruct" 457c9c72f8fSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info) 458c9c72f8fSHong Zhang { 459c9c72f8fSHong Zhang Mat C=B; 460c9c72f8fSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 461c9c72f8fSHong Zhang IS isrow = b->row,isicol = b->icol; 462c9c72f8fSHong Zhang PetscErrorCode ierr; 463c9c72f8fSHong Zhang const PetscInt *r,*ic,*ics; 464c9c72f8fSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 465c9c72f8fSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzL,row,*bdiag=b->diag,*pj; 466c9c72f8fSHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 467c9c72f8fSHong Zhang PetscTruth row_identity,col_identity; 468c9c72f8fSHong Zhang 469d4ad06f7SHong Zhang LUShift_Ctx sctx; 470d4ad06f7SHong Zhang PetscInt *ddiag,newshift; 471d4ad06f7SHong Zhang PetscReal rs; 472d4ad06f7SHong Zhang MatScalar d; 473d4ad06f7SHong Zhang 474c9c72f8fSHong Zhang PetscFunctionBegin; 475d6e5152cSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 476d6e5152cSHong Zhang ierr = PetscMemzero(&sctx,sizeof(LUShift_Ctx));CHKERRQ(ierr); 477d4ad06f7SHong Zhang 478d4ad06f7SHong Zhang /* if both shift schemes are chosen by user, only use info->shiftpd */ 479d4ad06f7SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 480d4ad06f7SHong Zhang ddiag = a->diag; 481d4ad06f7SHong Zhang sctx.shift_top = info->zeropivot; 482d4ad06f7SHong Zhang for (i=0; i<n; i++) { 483d4ad06f7SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 484d4ad06f7SHong Zhang d = (aa)[ddiag[i]]; 485d4ad06f7SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 486d4ad06f7SHong Zhang v = aa+ai[i]; 487d4ad06f7SHong Zhang nz = ai[i+1] - ai[i]; 488d4ad06f7SHong Zhang for (j=0; j<nz; j++) 489d4ad06f7SHong Zhang rs += PetscAbsScalar(v[j]); 490d4ad06f7SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 491d4ad06f7SHong Zhang } 492d4ad06f7SHong Zhang sctx.shift_top *= 1.1; 493d4ad06f7SHong Zhang sctx.nshift_max = 5; 494d4ad06f7SHong Zhang sctx.shift_lo = 0.; 495d4ad06f7SHong Zhang sctx.shift_hi = 1.; 496d4ad06f7SHong Zhang } 497d4ad06f7SHong Zhang 498c9c72f8fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 499c9c72f8fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 500c9c72f8fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 501c9c72f8fSHong Zhang ics = ic; 502c9c72f8fSHong Zhang 503d4ad06f7SHong Zhang do { 504d4ad06f7SHong Zhang sctx.lushift = PETSC_FALSE; 505c9c72f8fSHong Zhang for (i=0; i<n; i++){ 506c9c72f8fSHong Zhang /* zero rtmp */ 507c9c72f8fSHong Zhang /* L part */ 508c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 509c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 510c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 511c9c72f8fSHong Zhang 512c9c72f8fSHong Zhang /* U part */ 51362fbd6afSShri Abhyankar nz = bdiag[i]-bdiag[i+1]; 51462fbd6afSShri Abhyankar bjtmp = bj + bdiag[i+1]+1; 515813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 516813a1f2bSShri Abhyankar 517813a1f2bSShri Abhyankar /* load in initial (unfactored row) */ 518813a1f2bSShri Abhyankar nz = ai[r[i]+1] - ai[r[i]]; 519813a1f2bSShri Abhyankar ajtmp = aj + ai[r[i]]; 520813a1f2bSShri Abhyankar v = aa + ai[r[i]]; 521813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 522813a1f2bSShri Abhyankar rtmp[ics[ajtmp[j]]] = v[j]; 523813a1f2bSShri Abhyankar } 524813a1f2bSShri Abhyankar /* ZeropivotApply() */ 52598a93161SHong Zhang rtmp[i] += sctx.shift_amount; /* shift the diagonal of the matrix */ 526813a1f2bSShri Abhyankar 527813a1f2bSShri Abhyankar /* elimination */ 528813a1f2bSShri Abhyankar bjtmp = bj + bi[i]; 529813a1f2bSShri Abhyankar row = *bjtmp++; 530813a1f2bSShri Abhyankar nzL = bi[i+1] - bi[i]; 531f268cba8SShri Abhyankar for(k=0; k < nzL;k++) { 532813a1f2bSShri Abhyankar pc = rtmp + row; 533813a1f2bSShri Abhyankar if (*pc != 0.0) { 534813a1f2bSShri Abhyankar pv = b->a + bdiag[row]; 535813a1f2bSShri Abhyankar multiplier = *pc * (*pv); 536813a1f2bSShri Abhyankar *pc = multiplier; 53762fbd6afSShri Abhyankar pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 53862fbd6afSShri Abhyankar pv = b->a + bdiag[row+1]+1; 53962fbd6afSShri Abhyankar nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 540813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 541813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 542813a1f2bSShri Abhyankar } 543f268cba8SShri Abhyankar row = *bjtmp++; 544813a1f2bSShri Abhyankar } 545813a1f2bSShri Abhyankar 546813a1f2bSShri Abhyankar /* finished row so stick it into b->a */ 547813a1f2bSShri Abhyankar rs = 0.0; 548813a1f2bSShri Abhyankar /* L part */ 549813a1f2bSShri Abhyankar pv = b->a + bi[i] ; 550813a1f2bSShri Abhyankar pj = b->j + bi[i] ; 551813a1f2bSShri Abhyankar nz = bi[i+1] - bi[i]; 552813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 553813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 554813a1f2bSShri Abhyankar } 555813a1f2bSShri Abhyankar 556813a1f2bSShri Abhyankar /* U part */ 55762fbd6afSShri Abhyankar pv = b->a + bdiag[i+1]+1; 55862fbd6afSShri Abhyankar pj = b->j + bdiag[i+1]+1; 55962fbd6afSShri Abhyankar nz = bdiag[i] - bdiag[i+1]-1; 560813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 561813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 562813a1f2bSShri Abhyankar } 563813a1f2bSShri Abhyankar 5648ff23777SHong Zhang /* MatPivotCheck() */ 565813a1f2bSShri Abhyankar sctx.rs = rs; 566813a1f2bSShri Abhyankar sctx.pv = rtmp[i]; 5678ff23777SHong Zhang /* ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); */ 5688ff23777SHong Zhang if (info->shiftnz){ 5698ff23777SHong Zhang ierr = MatPivotCheck_nz(info,sctx,i,newshift);CHKERRQ(ierr); 5708ff23777SHong Zhang } else if (info->shiftpd){ 5718ff23777SHong Zhang ierr = MatPivotCheck_pd(info,sctx,i,newshift);CHKERRQ(ierr); 5728ff23777SHong Zhang } else if (info->shiftinblocks){ 5738ff23777SHong Zhang ierr = MatPivotCheck_inblocks(info,sctx,i,newshift);CHKERRQ(ierr); 5748ff23777SHong Zhang } else { 5758ff23777SHong Zhang ierr = MatPivotCheck_none(info,sctx,i,newshift);CHKERRQ(ierr); 5768ff23777SHong Zhang } 5778ff23777SHong Zhang rtmp[i] = sctx.pv; 578813a1f2bSShri Abhyankar if (newshift == 1) break; 579813a1f2bSShri Abhyankar 580813a1f2bSShri Abhyankar /* Mark diagonal and invert diagonal for simplier triangular solves */ 581813a1f2bSShri Abhyankar pv = b->a + bdiag[i]; 582813a1f2bSShri Abhyankar *pv = 1.0/rtmp[i]; 583813a1f2bSShri Abhyankar 584813a1f2bSShri Abhyankar } /* endof for (i=0; i<n; i++){ */ 585813a1f2bSShri Abhyankar 5868ff23777SHong Zhang /* MatPivotRefine() */ 587813a1f2bSShri Abhyankar if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){ 588813a1f2bSShri Abhyankar /* 589813a1f2bSShri Abhyankar * if no shift in this attempt & shifting & started shifting & can refine, 590813a1f2bSShri Abhyankar * then try lower shift 591813a1f2bSShri Abhyankar */ 592813a1f2bSShri Abhyankar sctx.shift_hi = sctx.shift_fraction; 593813a1f2bSShri Abhyankar sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 594813a1f2bSShri Abhyankar sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 595813a1f2bSShri Abhyankar sctx.lushift = PETSC_TRUE; 596813a1f2bSShri Abhyankar sctx.nshift++; 597813a1f2bSShri Abhyankar } 598813a1f2bSShri Abhyankar } while (sctx.lushift); 599813a1f2bSShri Abhyankar 600813a1f2bSShri Abhyankar ierr = PetscFree(rtmp);CHKERRQ(ierr); 601813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 602813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 603813a1f2bSShri Abhyankar 604813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 605813a1f2bSShri Abhyankar ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 606813a1f2bSShri Abhyankar if (row_identity && col_identity) { 6074395751fSShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct; 608813a1f2bSShri Abhyankar } else { 6094395751fSShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_newdatastruct; 610813a1f2bSShri Abhyankar } 611813a1f2bSShri Abhyankar 6123c0119dfSShri Abhyankar C->ops->solveadd = MatSolveAdd_SeqAIJ_newdatastruct; 6134395751fSShri Abhyankar C->ops->solvetranspose = MatSolveTranspose_SeqAIJ_newdatastruct; 614533e6011SShri Abhyankar C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_newdatastruct; 615813a1f2bSShri Abhyankar C->ops->matsolve = 0; 616813a1f2bSShri Abhyankar C->assembled = PETSC_TRUE; 617813a1f2bSShri Abhyankar C->preallocated = PETSC_TRUE; 618813a1f2bSShri Abhyankar ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 619813a1f2bSShri Abhyankar 6208ff23777SHong Zhang /* MatPivotView() */ 621813a1f2bSShri Abhyankar if (sctx.nshift){ 622813a1f2bSShri Abhyankar if (info->shiftpd) { 623813a1f2bSShri Abhyankar 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); 624813a1f2bSShri Abhyankar } else if (info->shiftnz) { 625813a1f2bSShri Abhyankar ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 6268ff23777SHong Zhang } else if (info->shiftinblocks){ 6278ff23777SHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftinblocks);CHKERRQ(ierr); 628813a1f2bSShri Abhyankar } 629813a1f2bSShri Abhyankar } 630813a1f2bSShri Abhyankar PetscFunctionReturn(0); 631813a1f2bSShri Abhyankar } 632813a1f2bSShri Abhyankar 6334a2ae208SSatish Balay #undef __FUNCT__ 6344a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ" 6350481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 636289bc588SBarry Smith { 637719d5645SBarry Smith Mat C=B; 638416022c9SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 63982bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 6406849ba73SBarry Smith PetscErrorCode ierr; 6415d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 642d278edc7SBarry Smith PetscInt nz,row,i,j,n=A->rmap->n,diag; 643d278edc7SBarry Smith const PetscInt *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 644d278edc7SBarry Smith const PetscInt *ajtmp,*bjtmp,*diag_offset = b->diag,*pj; 645d278edc7SBarry Smith MatScalar *pv,*rtmp,*pc,multiplier,d; 646d278edc7SBarry Smith const MatScalar *v,*aa=a->a; 64733f9893dSHong Zhang PetscReal rs=0.0; 648b3bf805bSHong Zhang LUShift_Ctx sctx; 64942898933SHong Zhang PetscInt newshift,*ddiag; 650289bc588SBarry Smith 6513a40ed3dSBarry Smith PetscFunctionBegin; 65278b31e54SBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 65378b31e54SBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 654b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 6559f7d0b68SHong Zhang ics = ic; 656289bc588SBarry Smith 657d4ad06f7SHong Zhang /* initialize shift context sctx */ 658d4ad06f7SHong Zhang sctx.nshift = 0; 659ada7143aSSatish Balay sctx.nshift_max = 0; 660d4ad06f7SHong Zhang sctx.shift_top = 0.0; 661d4ad06f7SHong Zhang sctx.shift_lo = 0.0; 662d4ad06f7SHong Zhang sctx.shift_hi = 0.0; 663d4ad06f7SHong Zhang sctx.shift_fraction = 0.0; 664d4ad06f7SHong Zhang sctx.shift_amount = 0.0; 665ada7143aSSatish Balay 666118f5789SBarry Smith /* if both shift schemes are chosen by user, only use info->shiftpd */ 6671a890ab1SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 66842898933SHong Zhang ddiag = a->diag; 6699f7d0b68SHong Zhang sctx.shift_top = info->zeropivot; 6706cc28720Svictorle for (i=0; i<n; i++) { 6719f95998fSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 6729f7d0b68SHong Zhang d = (aa)[ddiag[i]]; 6731a890ab1SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 6749f7d0b68SHong Zhang v = aa+ai[i]; 6759f7d0b68SHong Zhang nz = ai[i+1] - ai[i]; 676e24cfe64SHong Zhang for (j=0; j<nz; j++) 6771a890ab1SHong Zhang rs += PetscAbsScalar(v[j]); 6781a890ab1SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 6796cc28720Svictorle } 680118f5789SBarry Smith sctx.shift_top *= 1.1; 681d2276718SHong Zhang sctx.nshift_max = 5; 682d2276718SHong Zhang sctx.shift_lo = 0.; 683d2276718SHong Zhang sctx.shift_hi = 1.; 684a0a356efSHong Zhang } 685a0a356efSHong Zhang 686085256b3SBarry Smith do { 687d2276718SHong Zhang sctx.lushift = PETSC_FALSE; 688289bc588SBarry Smith for (i=0; i<n; i++){ 689b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 690b3bf805bSHong Zhang bjtmp = bj + bi[i]; 6919f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 692289bc588SBarry Smith 693289bc588SBarry Smith /* load in initial (unfactored row) */ 6949f7d0b68SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 6959f7d0b68SHong Zhang ajtmp = aj + ai[r[i]]; 6969f7d0b68SHong Zhang v = aa + ai[r[i]]; 697085256b3SBarry Smith for (j=0; j<nz; j++) { 698b3bf805bSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 699085256b3SBarry Smith } 700d2276718SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 701bd1bc851SBarry Smith /* if (sctx.shift_amount > 0.0) printf("row %d, shift %g\n",i,sctx.shift_amount); */ 702289bc588SBarry Smith 703b3bf805bSHong Zhang row = *bjtmp++; 704f2582111SSatish Balay while (row < i) { 7058c37ef55SBarry Smith pc = rtmp + row; 7068c37ef55SBarry Smith if (*pc != 0.0) { 707010a20caSHong Zhang pv = b->a + diag_offset[row]; 708010a20caSHong Zhang pj = b->j + diag_offset[row] + 1; 70935aab85fSBarry Smith multiplier = *pc / *pv++; 7108c37ef55SBarry Smith *pc = multiplier; 711b3bf805bSHong Zhang nz = bi[row+1] - diag_offset[row] - 1; 7129f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 713dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 714289bc588SBarry Smith } 715b3bf805bSHong Zhang row = *bjtmp++; 716289bc588SBarry Smith } 717416022c9SBarry Smith /* finished row so stick it into b->a */ 718b3bf805bSHong Zhang pv = b->a + bi[i] ; 719b3bf805bSHong Zhang pj = b->j + bi[i] ; 720b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 721b3bf805bSHong Zhang diag = diag_offset[i] - bi[i]; 722e57ff13aSBarry Smith rs = 0.0; 723d3d32019SBarry Smith for (j=0; j<nz; j++) { 7249f7d0b68SHong Zhang pv[j] = rtmp[pj[j]]; 7259f7d0b68SHong Zhang rs += PetscAbsScalar(pv[j]); 726d3d32019SBarry Smith } 727e57ff13aSBarry Smith rs -= PetscAbsScalar(pv[diag]); 728d2276718SHong Zhang 729b3bf805bSHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 730d2276718SHong Zhang sctx.rs = rs; 731d2276718SHong Zhang sctx.pv = pv[diag]; 732c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 7335b5bc046SBarry Smith if (newshift == 1) break; 73435aab85fSBarry Smith } 735d2276718SHong Zhang 7360481f469SBarry Smith if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 7376cc28720Svictorle /* 7386c037d1bSvictorle * if no shift in this attempt & shifting & started shifting & can refine, 7396cc28720Svictorle * then try lower shift 7406cc28720Svictorle */ 7410481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 7420481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 7430481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 744d2276718SHong Zhang sctx.lushift = PETSC_TRUE; 745d2276718SHong Zhang sctx.nshift++; 7466cc28720Svictorle } 747d2276718SHong Zhang } while (sctx.lushift); 7480f11f9aeSSatish Balay 74935aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 75035aab85fSBarry Smith for (i=0; i<n; i++) { 751010a20caSHong Zhang b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]]; 75235aab85fSBarry Smith } 753606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 75478b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 75578b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 756db4efbfdSBarry Smith if (b->inode.use) { 7574108e4d5SBarry Smith C->ops->solve = MatSolve_SeqAIJ_Inode; 758db4efbfdSBarry Smith } else { 7598d8c7c43SBarry Smith PetscTruth row_identity, col_identity; 7608d8c7c43SBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 7618d8c7c43SBarry Smith ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 7628d8c7c43SBarry Smith if (row_identity && col_identity) { 7638d8c7c43SBarry Smith C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 7648d8c7c43SBarry Smith } else { 765db4efbfdSBarry Smith C->ops->solve = MatSolve_SeqAIJ; 766db4efbfdSBarry Smith } 7678d8c7c43SBarry Smith } 768db4efbfdSBarry Smith C->ops->solveadd = MatSolveAdd_SeqAIJ; 769db4efbfdSBarry Smith C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 770db4efbfdSBarry Smith C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 771de26497bSBarry Smith C->ops->matsolve = MatMatSolve_SeqAIJ; 772c456f294SBarry Smith C->assembled = PETSC_TRUE; 7735c9eb25fSBarry Smith C->preallocated = PETSC_TRUE; 774d0f46423SBarry Smith ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 775d2276718SHong Zhang if (sctx.nshift){ 776e738e422SBarry Smith if (info->shiftpd) { 7770481f469SBarry 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); 778e738e422SBarry Smith } else if (info->shiftnz) { 779e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 7806cc28720Svictorle } 7810697b6caSHong Zhang } 7823a40ed3dSBarry Smith PetscFunctionReturn(0); 783289bc588SBarry Smith } 7840889b5dcSvictorle 785137fb511SHong Zhang /* 786137fb511SHong Zhang This routine implements inplace ILU(0) with row or/and column permutations. 787137fb511SHong Zhang Input: 788137fb511SHong Zhang A - original matrix 789137fb511SHong Zhang Output; 790137fb511SHong Zhang A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i] 791137fb511SHong Zhang a->j (col index) is permuted by the inverse of colperm, then sorted 792137fb511SHong Zhang a->a reordered accordingly with a->j 793137fb511SHong Zhang a->diag (ptr to diagonal elements) is updated. 794137fb511SHong Zhang */ 795137fb511SHong Zhang #undef __FUNCT__ 796137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm" 7970481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info) 798137fb511SHong Zhang { 799b051339dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 800b051339dSHong Zhang IS isrow = a->row,isicol = a->icol; 801137fb511SHong Zhang PetscErrorCode ierr; 8025d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 8035d0c19d7SBarry Smith PetscInt i,j,n=A->rmap->n,*ai=a->i,*aj=a->j; 8045d0c19d7SBarry Smith PetscInt *ajtmp,nz,row; 805b051339dSHong Zhang PetscInt *diag = a->diag,nbdiag,*pj; 806a77337e4SBarry Smith PetscScalar *rtmp,*pc,multiplier,d; 807a77337e4SBarry Smith MatScalar *v,*pv; 808137fb511SHong Zhang PetscReal rs; 809137fb511SHong Zhang LUShift_Ctx sctx; 810b051339dSHong Zhang PetscInt newshift; 811137fb511SHong Zhang 812137fb511SHong Zhang PetscFunctionBegin; 813719d5645SBarry Smith if (A != B) SETERRQ(PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address"); 814137fb511SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 815137fb511SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 816137fb511SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr); 817137fb511SHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 818b051339dSHong Zhang ics = ic; 819137fb511SHong Zhang 82075567043SBarry Smith sctx.shift_top = 0.; 821*e60cf9a0SBarry Smith sctx.nshift_max = 0; 82275567043SBarry Smith sctx.shift_lo = 0.; 82375567043SBarry Smith sctx.shift_hi = 0.; 82475567043SBarry Smith sctx.shift_fraction = 0.; 825137fb511SHong Zhang 826137fb511SHong Zhang /* if both shift schemes are chosen by user, only use info->shiftpd */ 827137fb511SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 82875567043SBarry Smith sctx.shift_top = 0.; 829137fb511SHong Zhang for (i=0; i<n; i++) { 830137fb511SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 831b051339dSHong Zhang d = (a->a)[diag[i]]; 832137fb511SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 833b051339dSHong Zhang v = a->a+ai[i]; 834b051339dSHong Zhang nz = ai[i+1] - ai[i]; 835137fb511SHong Zhang for (j=0; j<nz; j++) 836137fb511SHong Zhang rs += PetscAbsScalar(v[j]); 837137fb511SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 838137fb511SHong Zhang } 839137fb511SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 840137fb511SHong Zhang sctx.shift_top *= 1.1; 841137fb511SHong Zhang sctx.nshift_max = 5; 842137fb511SHong Zhang sctx.shift_lo = 0.; 843137fb511SHong Zhang sctx.shift_hi = 1.; 844137fb511SHong Zhang } 845137fb511SHong Zhang 84675567043SBarry Smith sctx.shift_amount = 0.; 847137fb511SHong Zhang sctx.nshift = 0; 848137fb511SHong Zhang do { 849137fb511SHong Zhang sctx.lushift = PETSC_FALSE; 850137fb511SHong Zhang for (i=0; i<n; i++){ 851b051339dSHong Zhang /* load in initial unfactored row */ 852b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 853b051339dSHong Zhang ajtmp = aj + ai[r[i]]; 854b051339dSHong Zhang v = a->a + ai[r[i]]; 855137fb511SHong Zhang /* sort permuted ajtmp and values v accordingly */ 856b051339dSHong Zhang for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]]; 857137fb511SHong Zhang ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr); 858137fb511SHong Zhang 859b051339dSHong Zhang diag[r[i]] = ai[r[i]]; 860137fb511SHong Zhang for (j=0; j<nz; j++) { 861137fb511SHong Zhang rtmp[ajtmp[j]] = v[j]; 862b051339dSHong Zhang if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */ 863137fb511SHong Zhang } 864137fb511SHong Zhang rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 865137fb511SHong Zhang 866b051339dSHong Zhang row = *ajtmp++; 867137fb511SHong Zhang while (row < i) { 868137fb511SHong Zhang pc = rtmp + row; 869137fb511SHong Zhang if (*pc != 0.0) { 870b051339dSHong Zhang pv = a->a + diag[r[row]]; 871b051339dSHong Zhang pj = aj + diag[r[row]] + 1; 872137fb511SHong Zhang 873137fb511SHong Zhang multiplier = *pc / *pv++; 874137fb511SHong Zhang *pc = multiplier; 875b051339dSHong Zhang nz = ai[r[row]+1] - diag[r[row]] - 1; 876b051339dSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 877dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 878137fb511SHong Zhang } 879b051339dSHong Zhang row = *ajtmp++; 880137fb511SHong Zhang } 881b051339dSHong Zhang /* finished row so overwrite it onto a->a */ 882b051339dSHong Zhang pv = a->a + ai[r[i]] ; 883b051339dSHong Zhang pj = aj + ai[r[i]] ; 884b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 885b051339dSHong Zhang nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */ 886137fb511SHong Zhang 887137fb511SHong Zhang rs = 0.0; 888137fb511SHong Zhang for (j=0; j<nz; j++) { 889b051339dSHong Zhang pv[j] = rtmp[pj[j]]; 890b051339dSHong Zhang if (j != nbdiag) rs += PetscAbsScalar(pv[j]); 891137fb511SHong Zhang } 892137fb511SHong Zhang 893137fb511SHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 894137fb511SHong Zhang sctx.rs = rs; 895b051339dSHong Zhang sctx.pv = pv[nbdiag]; 896c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 897137fb511SHong Zhang if (newshift == 1) break; 898137fb511SHong Zhang } 899137fb511SHong Zhang 9000481f469SBarry Smith if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 901137fb511SHong Zhang /* 902137fb511SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 903137fb511SHong Zhang * then try lower shift 904137fb511SHong Zhang */ 9050481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 9060481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 9070481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 908137fb511SHong Zhang sctx.lushift = PETSC_TRUE; 909137fb511SHong Zhang sctx.nshift++; 910137fb511SHong Zhang } 911137fb511SHong Zhang } while (sctx.lushift); 912137fb511SHong Zhang 913137fb511SHong Zhang /* invert diagonal entries for simplier triangular solves */ 914137fb511SHong Zhang for (i=0; i<n; i++) { 915b051339dSHong Zhang a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]]; 916137fb511SHong Zhang } 917137fb511SHong Zhang 918137fb511SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 919137fb511SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 920137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 921b051339dSHong Zhang A->ops->solve = MatSolve_SeqAIJ_InplaceWithPerm; 922db4efbfdSBarry Smith A->ops->solveadd = MatSolveAdd_SeqAIJ; 923db4efbfdSBarry Smith A->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 924db4efbfdSBarry Smith A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 925b051339dSHong Zhang A->assembled = PETSC_TRUE; 9265c9eb25fSBarry Smith A->preallocated = PETSC_TRUE; 927d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr); 928137fb511SHong Zhang if (sctx.nshift){ 929e738e422SBarry Smith if (info->shiftpd) { 9300481f469SBarry 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); 931e738e422SBarry Smith } else if (info->shiftnz) { 932e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 933137fb511SHong Zhang } 934137fb511SHong Zhang } 935137fb511SHong Zhang PetscFunctionReturn(0); 936137fb511SHong Zhang } 937137fb511SHong Zhang 9380a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9394a2ae208SSatish Balay #undef __FUNCT__ 9404a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ" 9410481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info) 942da3a660dSBarry Smith { 943dfbe8321SBarry Smith PetscErrorCode ierr; 944416022c9SBarry Smith Mat C; 945416022c9SBarry Smith 9463a40ed3dSBarry Smith PetscFunctionBegin; 94743244d56SBarry Smith ierr = MatGetFactor(A,MAT_SOLVER_PETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr); 948719d5645SBarry Smith ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr); 949719d5645SBarry Smith ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr); 950db4efbfdSBarry Smith A->ops->solve = C->ops->solve; 951db4efbfdSBarry Smith A->ops->solvetranspose = C->ops->solvetranspose; 952273d9f13SBarry Smith ierr = MatHeaderCopy(A,C);CHKERRQ(ierr); 95352e6d16bSBarry Smith ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr); 9543a40ed3dSBarry Smith PetscFunctionReturn(0); 955da3a660dSBarry Smith } 9560a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9571d098868SHong Zhang 958889a8dedSBarry Smith 9594a2ae208SSatish Balay #undef __FUNCT__ 9604a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ" 961dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 9628c37ef55SBarry Smith { 963416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 964416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 9656849ba73SBarry Smith PetscErrorCode ierr; 9665d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 9675d0c19d7SBarry Smith PetscInt nz; 9685d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 969dd6ea824SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 970d9fead3dSBarry Smith const PetscScalar *b; 971dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 9728c37ef55SBarry Smith 9733a40ed3dSBarry Smith PetscFunctionBegin; 9743a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 97591bf9adeSBarry Smith 976d9fead3dSBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 9771ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 978416022c9SBarry Smith tmp = a->solve_work; 9798c37ef55SBarry Smith 9803b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 9813b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 9828c37ef55SBarry Smith 9838c37ef55SBarry Smith /* forward solve the lower triangular */ 9848c37ef55SBarry Smith tmp[0] = b[*r++]; 985010a20caSHong Zhang tmps = tmp; 9868c37ef55SBarry Smith for (i=1; i<n; i++) { 987010a20caSHong Zhang v = aa + ai[i] ; 988010a20caSHong Zhang vi = aj + ai[i] ; 98953e7425aSBarry Smith nz = a->diag[i] - ai[i]; 99053e7425aSBarry Smith sum = b[*r++]; 991003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 9928c37ef55SBarry Smith tmp[i] = sum; 9938c37ef55SBarry Smith } 9948c37ef55SBarry Smith 9958c37ef55SBarry Smith /* backward solve the upper triangular */ 9968c37ef55SBarry Smith for (i=n-1; i>=0; i--){ 997010a20caSHong Zhang v = aa + a->diag[i] + 1; 998010a20caSHong Zhang vi = aj + a->diag[i] + 1; 999416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 10008c37ef55SBarry Smith sum = tmp[i]; 1001003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1002010a20caSHong Zhang x[*c--] = tmp[i] = sum*aa[a->diag[i]]; 10038c37ef55SBarry Smith } 10048c37ef55SBarry Smith 10053b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10063b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1007d9fead3dSBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 10081ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1009dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 10103a40ed3dSBarry Smith PetscFunctionReturn(0); 10118c37ef55SBarry Smith } 1012026e39d0SSatish Balay 10132bebee5dSHong Zhang #undef __FUNCT__ 10142bebee5dSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ" 10152bebee5dSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X) 10162bebee5dSHong Zhang { 10172bebee5dSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10182bebee5dSHong Zhang IS iscol = a->col,isrow = a->row; 10192bebee5dSHong Zhang PetscErrorCode ierr; 10205d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10215d0c19d7SBarry Smith PetscInt nz,neq; 10225d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 1023dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1024dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1025db4efbfdSBarry Smith PetscTruth bisdense,xisdense; 10262bebee5dSHong Zhang 10272bebee5dSHong Zhang PetscFunctionBegin; 10282bebee5dSHong Zhang if (!n) PetscFunctionReturn(0); 10292bebee5dSHong Zhang 1030db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1031db4efbfdSBarry Smith if (!bisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 1032db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1033db4efbfdSBarry Smith if (!xisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 1034db4efbfdSBarry Smith 10352bebee5dSHong Zhang ierr = MatGetArray(B,&b);CHKERRQ(ierr); 10362bebee5dSHong Zhang ierr = MatGetArray(X,&x);CHKERRQ(ierr); 10372bebee5dSHong Zhang 10382bebee5dSHong Zhang tmp = a->solve_work; 10392bebee5dSHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10402bebee5dSHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 10412bebee5dSHong Zhang 1042a36b22ccSSatish Balay for (neq=0; neq<B->cmap->n; neq++){ 10432bebee5dSHong Zhang /* forward solve the lower triangular */ 10442bebee5dSHong Zhang tmp[0] = b[r[0]]; 10452bebee5dSHong Zhang tmps = tmp; 10462bebee5dSHong Zhang for (i=1; i<n; i++) { 10472bebee5dSHong Zhang v = aa + ai[i] ; 10482bebee5dSHong Zhang vi = aj + ai[i] ; 10492bebee5dSHong Zhang nz = a->diag[i] - ai[i]; 10502bebee5dSHong Zhang sum = b[r[i]]; 1051003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10522bebee5dSHong Zhang tmp[i] = sum; 10532bebee5dSHong Zhang } 10542bebee5dSHong Zhang /* backward solve the upper triangular */ 10552bebee5dSHong Zhang for (i=n-1; i>=0; i--){ 10562bebee5dSHong Zhang v = aa + a->diag[i] + 1; 10572bebee5dSHong Zhang vi = aj + a->diag[i] + 1; 10582bebee5dSHong Zhang nz = ai[i+1] - a->diag[i] - 1; 10592bebee5dSHong Zhang sum = tmp[i]; 1060003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10612bebee5dSHong Zhang x[c[i]] = tmp[i] = sum*aa[a->diag[i]]; 10622bebee5dSHong Zhang } 10632bebee5dSHong Zhang 10642bebee5dSHong Zhang b += n; 10652bebee5dSHong Zhang x += n; 10662bebee5dSHong Zhang } 10672bebee5dSHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10682bebee5dSHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10692bebee5dSHong Zhang ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 10702bebee5dSHong Zhang ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 1071dc0b31edSSatish Balay ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 10722bebee5dSHong Zhang PetscFunctionReturn(0); 10732bebee5dSHong Zhang } 10742bebee5dSHong Zhang 1075137fb511SHong Zhang #undef __FUNCT__ 1076137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm" 1077137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx) 1078137fb511SHong Zhang { 1079137fb511SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1080137fb511SHong Zhang IS iscol = a->col,isrow = a->row; 1081137fb511SHong Zhang PetscErrorCode ierr; 10825d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 10835d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10845d0c19d7SBarry Smith PetscInt nz,row; 1085dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1086dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1087137fb511SHong Zhang 1088137fb511SHong Zhang PetscFunctionBegin; 1089137fb511SHong Zhang if (!n) PetscFunctionReturn(0); 1090137fb511SHong Zhang 1091137fb511SHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 1092137fb511SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1093137fb511SHong Zhang tmp = a->solve_work; 1094137fb511SHong Zhang 1095137fb511SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1096137fb511SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1097137fb511SHong Zhang 1098137fb511SHong Zhang /* forward solve the lower triangular */ 1099137fb511SHong Zhang tmp[0] = b[*r++]; 1100137fb511SHong Zhang tmps = tmp; 1101137fb511SHong Zhang for (row=1; row<n; row++) { 1102137fb511SHong Zhang i = rout[row]; /* permuted row */ 1103137fb511SHong Zhang v = aa + ai[i] ; 1104137fb511SHong Zhang vi = aj + ai[i] ; 1105137fb511SHong Zhang nz = a->diag[i] - ai[i]; 1106137fb511SHong Zhang sum = b[*r++]; 1107003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1108137fb511SHong Zhang tmp[row] = sum; 1109137fb511SHong Zhang } 1110137fb511SHong Zhang 1111137fb511SHong Zhang /* backward solve the upper triangular */ 1112137fb511SHong Zhang for (row=n-1; row>=0; row--){ 1113137fb511SHong Zhang i = rout[row]; /* permuted row */ 1114137fb511SHong Zhang v = aa + a->diag[i] + 1; 1115137fb511SHong Zhang vi = aj + a->diag[i] + 1; 1116137fb511SHong Zhang nz = ai[i+1] - a->diag[i] - 1; 1117137fb511SHong Zhang sum = tmp[row]; 1118003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1119137fb511SHong Zhang x[*c--] = tmp[row] = sum*aa[a->diag[i]]; 1120137fb511SHong Zhang } 1121137fb511SHong Zhang 1122137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1123137fb511SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1124137fb511SHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 1125137fb511SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1126dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1127137fb511SHong Zhang PetscFunctionReturn(0); 1128137fb511SHong Zhang } 1129137fb511SHong Zhang 1130930ae53cSBarry Smith /* ----------------------------------------------------------- */ 1131a4005a5dSBarry Smith #include "../src/mat/impls/aij/seq/ftn-kernels/fsolve.h" 11324a2ae208SSatish Balay #undef __FUNCT__ 11334a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering" 1134dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 1135930ae53cSBarry Smith { 1136930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 11376849ba73SBarry Smith PetscErrorCode ierr; 1138d0f46423SBarry Smith PetscInt n = A->rmap->n; 1139b377110cSBarry Smith const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag; 1140356650c2SBarry Smith PetscScalar *x; 1141356650c2SBarry Smith const PetscScalar *b; 1142dd6ea824SBarry Smith const MatScalar *aa = a->a; 1143aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 1144356650c2SBarry Smith PetscInt adiag_i,i,nz,ai_i; 1145b377110cSBarry Smith const PetscInt *vi; 1146dd6ea824SBarry Smith const MatScalar *v; 1147dd6ea824SBarry Smith PetscScalar sum; 1148d85afc42SSatish Balay #endif 1149930ae53cSBarry Smith 11503a40ed3dSBarry Smith PetscFunctionBegin; 11513a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 1152930ae53cSBarry Smith 1153356650c2SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 11541ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1155930ae53cSBarry Smith 1156aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 11571c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 11581c4feecaSSatish Balay #else 1159930ae53cSBarry Smith /* forward solve the lower triangular */ 1160930ae53cSBarry Smith x[0] = b[0]; 1161930ae53cSBarry Smith for (i=1; i<n; i++) { 1162930ae53cSBarry Smith ai_i = ai[i]; 1163930ae53cSBarry Smith v = aa + ai_i; 1164930ae53cSBarry Smith vi = aj + ai_i; 1165930ae53cSBarry Smith nz = adiag[i] - ai_i; 1166930ae53cSBarry Smith sum = b[i]; 1167003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1168930ae53cSBarry Smith x[i] = sum; 1169930ae53cSBarry Smith } 1170930ae53cSBarry Smith 1171930ae53cSBarry Smith /* backward solve the upper triangular */ 1172930ae53cSBarry Smith for (i=n-1; i>=0; i--){ 1173930ae53cSBarry Smith adiag_i = adiag[i]; 1174d708749eSBarry Smith v = aa + adiag_i + 1; 1175d708749eSBarry Smith vi = aj + adiag_i + 1; 1176930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 1177930ae53cSBarry Smith sum = x[i]; 1178003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1179930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 1180930ae53cSBarry Smith } 11811c4feecaSSatish Balay #endif 1182dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1183356650c2SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 11841ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 11853a40ed3dSBarry Smith PetscFunctionReturn(0); 1186930ae53cSBarry Smith } 1187930ae53cSBarry Smith 11884a2ae208SSatish Balay #undef __FUNCT__ 11894a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqAIJ" 1190dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 1191da3a660dSBarry Smith { 1192416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1193416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 11946849ba73SBarry Smith PetscErrorCode ierr; 119504fbf559SBarry Smith PetscInt i, n = A->rmap->n,j; 11965d0c19d7SBarry Smith PetscInt nz; 119704fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j; 119804fbf559SBarry Smith PetscScalar *x,*tmp,sum; 119904fbf559SBarry Smith const PetscScalar *b; 1200dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1201da3a660dSBarry Smith 12023a40ed3dSBarry Smith PetscFunctionBegin; 120378b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 1204da3a660dSBarry Smith 120504fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12061ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1207416022c9SBarry Smith tmp = a->solve_work; 1208da3a660dSBarry Smith 12093b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12103b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1211da3a660dSBarry Smith 1212da3a660dSBarry Smith /* forward solve the lower triangular */ 1213da3a660dSBarry Smith tmp[0] = b[*r++]; 1214da3a660dSBarry Smith for (i=1; i<n; i++) { 1215010a20caSHong Zhang v = aa + ai[i] ; 1216010a20caSHong Zhang vi = aj + ai[i] ; 1217416022c9SBarry Smith nz = a->diag[i] - ai[i]; 1218da3a660dSBarry Smith sum = b[*r++]; 121904fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1220da3a660dSBarry Smith tmp[i] = sum; 1221da3a660dSBarry Smith } 1222da3a660dSBarry Smith 1223da3a660dSBarry Smith /* backward solve the upper triangular */ 1224da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1225010a20caSHong Zhang v = aa + a->diag[i] + 1; 1226010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1227416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 1228da3a660dSBarry Smith sum = tmp[i]; 122904fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1230010a20caSHong Zhang tmp[i] = sum*aa[a->diag[i]]; 1231da3a660dSBarry Smith x[*c--] += tmp[i]; 1232da3a660dSBarry Smith } 1233da3a660dSBarry Smith 12343b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 12353b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 123604fbf559SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12371ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1238dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1239898183ecSLois Curfman McInnes 12403a40ed3dSBarry Smith PetscFunctionReturn(0); 1241da3a660dSBarry Smith } 124204fbf559SBarry Smith 12434a2ae208SSatish Balay #undef __FUNCT__ 12443c0119dfSShri Abhyankar #define __FUNCT__ "MatSolveAdd_SeqAIJ_newdatastruct" 12453c0119dfSShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec yy,Vec xx) 12463c0119dfSShri Abhyankar { 12473c0119dfSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 12483c0119dfSShri Abhyankar IS iscol = a->col,isrow = a->row; 12493c0119dfSShri Abhyankar PetscErrorCode ierr; 12503c0119dfSShri Abhyankar PetscInt i, n = A->rmap->n,j; 12513c0119dfSShri Abhyankar PetscInt nz; 12523c0119dfSShri Abhyankar const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 12533c0119dfSShri Abhyankar PetscScalar *x,*tmp,sum; 12543c0119dfSShri Abhyankar const PetscScalar *b; 12553c0119dfSShri Abhyankar const MatScalar *aa = a->a,*v; 12563c0119dfSShri Abhyankar 12573c0119dfSShri Abhyankar PetscFunctionBegin; 12583c0119dfSShri Abhyankar if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 12593c0119dfSShri Abhyankar 12603c0119dfSShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12613c0119dfSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 12623c0119dfSShri Abhyankar tmp = a->solve_work; 12633c0119dfSShri Abhyankar 12643c0119dfSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12653c0119dfSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 12663c0119dfSShri Abhyankar 12673c0119dfSShri Abhyankar /* forward solve the lower triangular */ 12683c0119dfSShri Abhyankar tmp[0] = b[r[0]]; 12693c0119dfSShri Abhyankar v = aa; 12703c0119dfSShri Abhyankar vi = aj; 12713c0119dfSShri Abhyankar for (i=1; i<n; i++) { 12723c0119dfSShri Abhyankar nz = ai[i+1] - ai[i]; 12733c0119dfSShri Abhyankar sum = b[r[i]]; 12743c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 12753c0119dfSShri Abhyankar tmp[i] = sum; 12763c0119dfSShri Abhyankar v += nz; vi += nz; 12773c0119dfSShri Abhyankar } 12783c0119dfSShri Abhyankar 12793c0119dfSShri Abhyankar /* backward solve the upper triangular */ 12803c0119dfSShri Abhyankar v = aa + adiag[n-1]; 12813c0119dfSShri Abhyankar vi = aj + adiag[n-1]; 12823c0119dfSShri Abhyankar for (i=n-1; i>=0; i--){ 12833c0119dfSShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 12843c0119dfSShri Abhyankar sum = tmp[i]; 12853c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 12863c0119dfSShri Abhyankar tmp[i] = sum*v[nz]; 12873c0119dfSShri Abhyankar x[c[i]] += tmp[i]; 12883c0119dfSShri Abhyankar v += nz+1; vi += nz+1; 12893c0119dfSShri Abhyankar } 12903c0119dfSShri Abhyankar 12913c0119dfSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 12923c0119dfSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 12933c0119dfSShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12943c0119dfSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 12953c0119dfSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 12963c0119dfSShri Abhyankar 12973c0119dfSShri Abhyankar PetscFunctionReturn(0); 12983c0119dfSShri Abhyankar } 12993c0119dfSShri Abhyankar 13003c0119dfSShri Abhyankar #undef __FUNCT__ 13014a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqAIJ" 1302dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 1303da3a660dSBarry Smith { 1304416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13052235e667SBarry Smith IS iscol = a->col,isrow = a->row; 13066849ba73SBarry Smith PetscErrorCode ierr; 130704fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 130804fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 130904fbf559SBarry Smith PetscInt nz; 131004fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1311dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 131204fbf559SBarry Smith const PetscScalar *b; 1313da3a660dSBarry Smith 13143a40ed3dSBarry Smith PetscFunctionBegin; 131504fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 13161ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1317416022c9SBarry Smith tmp = a->solve_work; 1318da3a660dSBarry Smith 13192235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13202235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1321da3a660dSBarry Smith 1322da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 13232235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1324da3a660dSBarry Smith 1325da3a660dSBarry Smith /* forward solve the U^T */ 1326da3a660dSBarry Smith for (i=0; i<n; i++) { 1327010a20caSHong Zhang v = aa + diag[i] ; 1328010a20caSHong Zhang vi = aj + diag[i] + 1; 1329f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 1330c38d4ed2SBarry Smith s1 = tmp[i]; 1331ef66eb69SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 133204fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1333c38d4ed2SBarry Smith tmp[i] = s1; 1334da3a660dSBarry Smith } 1335da3a660dSBarry Smith 1336da3a660dSBarry Smith /* backward solve the L^T */ 1337da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1338010a20caSHong Zhang v = aa + diag[i] - 1 ; 1339010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1340f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 1341f1af5d2fSBarry Smith s1 = tmp[i]; 134204fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 1343da3a660dSBarry Smith } 1344da3a660dSBarry Smith 1345da3a660dSBarry Smith /* copy tmp into x according to permutation */ 1346591294e4SBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1347da3a660dSBarry Smith 13482235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13492235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 135004fbf559SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 13511ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1352da3a660dSBarry Smith 1353dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 13543a40ed3dSBarry Smith PetscFunctionReturn(0); 1355da3a660dSBarry Smith } 1356da3a660dSBarry Smith 1357d1fa9404SShri Abhyankar #undef __FUNCT__ 13584395751fSShri Abhyankar #define __FUNCT__ "MatSolveTranspose_SeqAIJ_newdatastruct" 13594395751fSShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec xx) 1360d1fa9404SShri Abhyankar { 1361d1fa9404SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1362d1fa9404SShri Abhyankar IS iscol = a->col,isrow = a->row; 1363d1fa9404SShri Abhyankar PetscErrorCode ierr; 1364d1fa9404SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1365d1fa9404SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1366d1fa9404SShri Abhyankar PetscInt nz; 1367d1fa9404SShri Abhyankar PetscScalar *x,*tmp,s1; 1368d1fa9404SShri Abhyankar const MatScalar *aa = a->a,*v; 1369d1fa9404SShri Abhyankar const PetscScalar *b; 1370d1fa9404SShri Abhyankar 1371d1fa9404SShri Abhyankar PetscFunctionBegin; 1372d1fa9404SShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 1373d1fa9404SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1374d1fa9404SShri Abhyankar tmp = a->solve_work; 1375d1fa9404SShri Abhyankar 1376d1fa9404SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1377d1fa9404SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1378d1fa9404SShri Abhyankar 1379d1fa9404SShri Abhyankar /* copy the b into temp work space according to permutation */ 1380d1fa9404SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1381d1fa9404SShri Abhyankar 1382d1fa9404SShri Abhyankar /* forward solve the U^T */ 1383d1fa9404SShri Abhyankar for (i=0; i<n; i++) { 1384d1fa9404SShri Abhyankar v = aa + adiag[i+1] + 1; 1385d1fa9404SShri Abhyankar vi = aj + adiag[i+1] + 1; 1386d1fa9404SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1387d1fa9404SShri Abhyankar s1 = tmp[i]; 1388d1fa9404SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1389d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1390d1fa9404SShri Abhyankar tmp[i] = s1; 1391d1fa9404SShri Abhyankar } 1392d1fa9404SShri Abhyankar 1393d1fa9404SShri Abhyankar /* backward solve the L^T */ 1394d1fa9404SShri Abhyankar for (i=n-1; i>=0; i--){ 1395d1fa9404SShri Abhyankar v = aa + ai[i]; 1396d1fa9404SShri Abhyankar vi = aj + ai[i]; 1397d1fa9404SShri Abhyankar nz = ai[i+1] - ai[i]; 1398d1fa9404SShri Abhyankar s1 = tmp[i]; 1399d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1400d1fa9404SShri Abhyankar } 1401d1fa9404SShri Abhyankar 1402d1fa9404SShri Abhyankar /* copy tmp into x according to permutation */ 1403d1fa9404SShri Abhyankar for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1404d1fa9404SShri Abhyankar 1405d1fa9404SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1406d1fa9404SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1407d1fa9404SShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 1408d1fa9404SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1409d1fa9404SShri Abhyankar 1410d1fa9404SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1411d1fa9404SShri Abhyankar PetscFunctionReturn(0); 1412d1fa9404SShri Abhyankar } 1413d1fa9404SShri Abhyankar 14144a2ae208SSatish Balay #undef __FUNCT__ 14154a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ" 1416dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 1417da3a660dSBarry Smith { 1418416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 14192235e667SBarry Smith IS iscol = a->col,isrow = a->row; 14206849ba73SBarry Smith PetscErrorCode ierr; 142104fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 142204fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 142304fbf559SBarry Smith PetscInt nz; 142404fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1425dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 142604fbf559SBarry Smith const PetscScalar *b; 14276abc6512SBarry Smith 14283a40ed3dSBarry Smith PetscFunctionBegin; 142923bc6035SMatthew Knepley if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 143004fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 14311ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1432416022c9SBarry Smith tmp = a->solve_work; 14336abc6512SBarry Smith 14342235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 14352235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 14366abc6512SBarry Smith 14376abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 14382235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 14396abc6512SBarry Smith 14406abc6512SBarry Smith /* forward solve the U^T */ 14416abc6512SBarry Smith for (i=0; i<n; i++) { 1442010a20caSHong Zhang v = aa + diag[i] ; 1443010a20caSHong Zhang vi = aj + diag[i] + 1; 1444f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 144504fbf559SBarry Smith s1 = tmp[i]; 144604fbf559SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 144704fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 144804fbf559SBarry Smith tmp[i] = s1; 14496abc6512SBarry Smith } 14506abc6512SBarry Smith 14516abc6512SBarry Smith /* backward solve the L^T */ 14526abc6512SBarry Smith for (i=n-1; i>=0; i--){ 1453010a20caSHong Zhang v = aa + diag[i] - 1 ; 1454010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1455f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 145604fbf559SBarry Smith s1 = tmp[i]; 145704fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 14586abc6512SBarry Smith } 14596abc6512SBarry Smith 14606abc6512SBarry Smith /* copy tmp into x according to permutation */ 14616abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 14626abc6512SBarry Smith 14632235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 14642235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 146504fbf559SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 14661ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 14676abc6512SBarry Smith 146804fbf559SBarry Smith ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 14693a40ed3dSBarry Smith PetscFunctionReturn(0); 1470da3a660dSBarry Smith } 147104fbf559SBarry Smith 1472533e6011SShri Abhyankar #undef __FUNCT__ 1473533e6011SShri Abhyankar #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ_newdatastruct" 1474533e6011SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec zz,Vec xx) 1475533e6011SShri Abhyankar { 1476533e6011SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1477533e6011SShri Abhyankar IS iscol = a->col,isrow = a->row; 1478533e6011SShri Abhyankar PetscErrorCode ierr; 1479533e6011SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1480533e6011SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1481533e6011SShri Abhyankar PetscInt nz; 1482533e6011SShri Abhyankar PetscScalar *x,*tmp,s1; 1483533e6011SShri Abhyankar const MatScalar *aa = a->a,*v; 1484533e6011SShri Abhyankar const PetscScalar *b; 1485533e6011SShri Abhyankar 1486533e6011SShri Abhyankar PetscFunctionBegin; 1487533e6011SShri Abhyankar if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 1488533e6011SShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 1489533e6011SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1490533e6011SShri Abhyankar tmp = a->solve_work; 1491533e6011SShri Abhyankar 1492533e6011SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1493533e6011SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1494533e6011SShri Abhyankar 1495533e6011SShri Abhyankar /* copy the b into temp work space according to permutation */ 1496533e6011SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1497533e6011SShri Abhyankar 1498533e6011SShri Abhyankar /* forward solve the U^T */ 1499533e6011SShri Abhyankar for (i=0; i<n; i++) { 1500533e6011SShri Abhyankar v = aa + adiag[i+1] + 1; 1501533e6011SShri Abhyankar vi = aj + adiag[i+1] + 1; 1502533e6011SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1503533e6011SShri Abhyankar s1 = tmp[i]; 1504533e6011SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1505533e6011SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1506533e6011SShri Abhyankar tmp[i] = s1; 1507533e6011SShri Abhyankar } 1508533e6011SShri Abhyankar 1509533e6011SShri Abhyankar 1510533e6011SShri Abhyankar /* backward solve the L^T */ 1511533e6011SShri Abhyankar for (i=n-1; i>=0; i--){ 1512533e6011SShri Abhyankar v = aa + ai[i] ; 1513533e6011SShri Abhyankar vi = aj + ai[i]; 1514533e6011SShri Abhyankar nz = ai[i+1] - ai[i]; 1515533e6011SShri Abhyankar s1 = tmp[i]; 1516c5e3b2a3SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1517533e6011SShri Abhyankar } 1518533e6011SShri Abhyankar 1519533e6011SShri Abhyankar /* copy tmp into x according to permutation */ 1520533e6011SShri Abhyankar for (i=0; i<n; i++) x[r[i]] += tmp[i]; 1521533e6011SShri Abhyankar 1522533e6011SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1523533e6011SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1524533e6011SShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 1525533e6011SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1526533e6011SShri Abhyankar 1527533e6011SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1528533e6011SShri Abhyankar PetscFunctionReturn(0); 1529533e6011SShri Abhyankar } 1530533e6011SShri Abhyankar 15319e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 15323c5fc038SBarry Smith EXTERN PetscErrorCode Mat_CheckInode(Mat,PetscTruth); 1533f77e22a1SHong Zhang EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscTruth); 153408480c60SBarry Smith 15358fc3a347SHong Zhang /* 15368a76ed4fSHong Zhang ilu() under revised new data structure. 1537813a1f2bSShri Abhyankar Factored arrays bj and ba are stored as 1538813a1f2bSShri Abhyankar L(0,:), L(1,:), ...,L(n-1,:), U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:) 1539813a1f2bSShri Abhyankar 1540813a1f2bSShri Abhyankar bi=fact->i is an array of size n+1, in which 1541813a1f2bSShri Abhyankar bi+ 1542baabb860SHong Zhang bi[i]: points to 1st entry of L(i,:),i=0,...,n-1 1543baabb860SHong Zhang bi[n]: points to L(n-1,n-1)+1 1544baabb860SHong Zhang 1545813a1f2bSShri Abhyankar bdiag=fact->diag is an array of size n+1,in which 1546baabb860SHong Zhang bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1 1547a24f213cSHong Zhang bdiag[n]: points to entry of U(n-1,0)-1 1548813a1f2bSShri Abhyankar 1549813a1f2bSShri Abhyankar U(i,:) contains bdiag[i] as its last entry, i.e., 1550813a1f2bSShri Abhyankar U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 1551813a1f2bSShri Abhyankar */ 1552813a1f2bSShri Abhyankar #undef __FUNCT__ 1553b4f5729fSShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct" 1554b4f5729fSShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1555813a1f2bSShri Abhyankar { 1556813a1f2bSShri Abhyankar 1557813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1558813a1f2bSShri Abhyankar PetscErrorCode ierr; 1559813a1f2bSShri Abhyankar PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag; 1560a24f213cSHong Zhang PetscInt i,j,nz,*bi,*bj,*bdiag; 15611df811f5SHong Zhang PetscTruth missing; 15621df811f5SHong Zhang IS isicol; 1563813a1f2bSShri Abhyankar 1564813a1f2bSShri Abhyankar PetscFunctionBegin; 15651df811f5SHong 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); 15661df811f5SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 15671df811f5SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 15681df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 15691df811f5SHong Zhang 1570813a1f2bSShri Abhyankar ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr); 1571813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1572813a1f2bSShri Abhyankar 1573813a1f2bSShri Abhyankar /* allocate matrix arrays for new data structure */ 1574813a1f2bSShri Abhyankar ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,n+1,PetscInt,&b->i);CHKERRQ(ierr); 1575813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1576813a1f2bSShri Abhyankar b->singlemalloc = PETSC_TRUE; 1577813a1f2bSShri Abhyankar if (!b->diag){ 1578813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr); 1579813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1580813a1f2bSShri Abhyankar } 1581813a1f2bSShri Abhyankar bdiag = b->diag; 1582813a1f2bSShri Abhyankar 1583813a1f2bSShri Abhyankar if (n > 0) { 1584813a1f2bSShri Abhyankar ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr); 1585813a1f2bSShri Abhyankar } 1586813a1f2bSShri Abhyankar 1587813a1f2bSShri Abhyankar /* set bi and bj with new data structure */ 1588813a1f2bSShri Abhyankar bi = b->i; 1589813a1f2bSShri Abhyankar bj = b->j; 1590813a1f2bSShri Abhyankar 1591813a1f2bSShri Abhyankar /* L part */ 1592*e60cf9a0SBarry Smith bi[0] = 0; 1593813a1f2bSShri Abhyankar for (i=0; i<n; i++){ 1594813a1f2bSShri Abhyankar nz = adiag[i] - ai[i]; 1595813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nz; 1596813a1f2bSShri Abhyankar aj = a->j + ai[i]; 1597813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 1598813a1f2bSShri Abhyankar *bj = aj[j]; bj++; 1599813a1f2bSShri Abhyankar } 1600813a1f2bSShri Abhyankar } 1601813a1f2bSShri Abhyankar 1602813a1f2bSShri Abhyankar /* U part */ 160362fbd6afSShri Abhyankar bdiag[n] = bi[n]-1; 1604813a1f2bSShri Abhyankar for (i=n-1; i>=0; i--){ 1605813a1f2bSShri Abhyankar nz = ai[i+1] - adiag[i] - 1; 1606813a1f2bSShri Abhyankar aj = a->j + adiag[i] + 1; 1607813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 1608813a1f2bSShri Abhyankar *bj = aj[j]; bj++; 1609813a1f2bSShri Abhyankar } 1610813a1f2bSShri Abhyankar /* diag[i] */ 1611813a1f2bSShri Abhyankar *bj = i; bj++; 1612a24f213cSHong Zhang bdiag[i] = bdiag[i+1] + nz + 1; 1613813a1f2bSShri Abhyankar } 16141df811f5SHong Zhang 16151df811f5SHong Zhang fact->factor = MAT_FACTOR_ILU; 16161df811f5SHong Zhang fact->info.factor_mallocs = 0; 16171df811f5SHong Zhang fact->info.fill_ratio_given = info->fill; 16181df811f5SHong Zhang fact->info.fill_ratio_needed = 1.0; 16191df811f5SHong Zhang fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 16201df811f5SHong Zhang 16211df811f5SHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 16221df811f5SHong Zhang b->row = isrow; 16231df811f5SHong Zhang b->col = iscol; 16241df811f5SHong Zhang b->icol = isicol; 16251df811f5SHong Zhang ierr = PetscMalloc((fact->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 16261df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 16271df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1628813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1629813a1f2bSShri Abhyankar } 1630813a1f2bSShri Abhyankar 1631813a1f2bSShri Abhyankar #undef __FUNCT__ 1632b4f5729fSShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_newdatastruct" 1633b4f5729fSShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1634813a1f2bSShri Abhyankar { 1635813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1636813a1f2bSShri Abhyankar IS isicol; 1637813a1f2bSShri Abhyankar PetscErrorCode ierr; 1638813a1f2bSShri Abhyankar const PetscInt *r,*ic; 16391df811f5SHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j; 1640813a1f2bSShri Abhyankar PetscInt *bi,*cols,nnz,*cols_lvl; 1641813a1f2bSShri Abhyankar PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 1642813a1f2bSShri Abhyankar PetscInt i,levels,diagonal_fill; 1643813a1f2bSShri Abhyankar PetscTruth col_identity,row_identity; 1644813a1f2bSShri Abhyankar PetscReal f; 1645813a1f2bSShri Abhyankar PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 1646813a1f2bSShri Abhyankar PetscBT lnkbt; 1647813a1f2bSShri Abhyankar PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 1648813a1f2bSShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1649813a1f2bSShri Abhyankar PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 1650813a1f2bSShri Abhyankar 1651813a1f2bSShri Abhyankar PetscFunctionBegin; 16521df811f5SHong Zhang /* printf("MatILUFactorSymbolic_SeqAIJ_newdatastruct ...\n"); */ 1653813a1f2bSShri Abhyankar levels = (PetscInt)info->levels; 1654813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 1655813a1f2bSShri Abhyankar ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 1656813a1f2bSShri Abhyankar 1657813a1f2bSShri Abhyankar if (!levels && row_identity && col_identity) { 1658813a1f2bSShri Abhyankar /* special case: ilu(0) with natural ordering */ 1659b4f5729fSShri Abhyankar ierr = MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1660813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1661813a1f2bSShri Abhyankar } 1662813a1f2bSShri Abhyankar 16631df811f5SHong 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); 16641df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 1665813a1f2bSShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1666813a1f2bSShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 1667813a1f2bSShri Abhyankar 16681bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 16691bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 16701bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1671*e60cf9a0SBarry Smith bi[0] = bdiag[0] = 0; 1672813a1f2bSShri Abhyankar 16730e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr); 1674813a1f2bSShri Abhyankar 1675813a1f2bSShri Abhyankar /* create a linked list for storing column indices of the active row */ 1676813a1f2bSShri Abhyankar nlnk = n + 1; 1677813a1f2bSShri Abhyankar ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1678813a1f2bSShri Abhyankar 1679813a1f2bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 16801df811f5SHong Zhang f = info->fill; 16811df811f5SHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 1682813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1683813a1f2bSShri Abhyankar current_space = free_space; 1684813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 1685813a1f2bSShri Abhyankar current_space_lvl = free_space_lvl; 1686813a1f2bSShri Abhyankar 1687813a1f2bSShri Abhyankar for (i=0; i<n; i++) { 1688813a1f2bSShri Abhyankar nzi = 0; 1689813a1f2bSShri Abhyankar /* copy current row into linked list */ 1690813a1f2bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 1691813a1f2bSShri 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); 1692813a1f2bSShri Abhyankar cols = aj + ai[r[i]]; 1693813a1f2bSShri Abhyankar lnk[i] = -1; /* marker to indicate if diagonal exists */ 1694813a1f2bSShri Abhyankar ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1695813a1f2bSShri Abhyankar nzi += nlnk; 1696813a1f2bSShri Abhyankar 1697813a1f2bSShri Abhyankar /* make sure diagonal entry is included */ 1698813a1f2bSShri Abhyankar if (diagonal_fill && lnk[i] == -1) { 1699813a1f2bSShri Abhyankar fm = n; 1700813a1f2bSShri Abhyankar while (lnk[fm] < i) fm = lnk[fm]; 1701813a1f2bSShri Abhyankar lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 1702813a1f2bSShri Abhyankar lnk[fm] = i; 1703813a1f2bSShri Abhyankar lnk_lvl[i] = 0; 1704813a1f2bSShri Abhyankar nzi++; dcount++; 1705813a1f2bSShri Abhyankar } 1706813a1f2bSShri Abhyankar 1707813a1f2bSShri Abhyankar /* add pivot rows into the active row */ 1708813a1f2bSShri Abhyankar nzbd = 0; 1709813a1f2bSShri Abhyankar prow = lnk[n]; 1710813a1f2bSShri Abhyankar while (prow < i) { 1711813a1f2bSShri Abhyankar nnz = bdiag[prow]; 1712813a1f2bSShri Abhyankar cols = bj_ptr[prow] + nnz + 1; 1713813a1f2bSShri Abhyankar cols_lvl = bjlvl_ptr[prow] + nnz + 1; 1714813a1f2bSShri Abhyankar nnz = bi[prow+1] - bi[prow] - nnz - 1; 1715813a1f2bSShri Abhyankar ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 1716813a1f2bSShri Abhyankar nzi += nlnk; 1717813a1f2bSShri Abhyankar prow = lnk[prow]; 1718813a1f2bSShri Abhyankar nzbd++; 1719813a1f2bSShri Abhyankar } 1720813a1f2bSShri Abhyankar bdiag[i] = nzbd; 1721813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nzi; 1722813a1f2bSShri Abhyankar 1723813a1f2bSShri Abhyankar /* if free space is not available, make more free space */ 1724813a1f2bSShri Abhyankar if (current_space->local_remaining<nzi) { 1725813a1f2bSShri Abhyankar nnz = 2*nzi*(n - i); /* estimated and max additional space needed */ 1726813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1727813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 1728813a1f2bSShri Abhyankar reallocs++; 1729813a1f2bSShri Abhyankar } 1730813a1f2bSShri Abhyankar 1731813a1f2bSShri Abhyankar /* copy data into free_space and free_space_lvl, then initialize lnk */ 1732813a1f2bSShri Abhyankar ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 1733813a1f2bSShri Abhyankar bj_ptr[i] = current_space->array; 1734813a1f2bSShri Abhyankar bjlvl_ptr[i] = current_space_lvl->array; 1735813a1f2bSShri Abhyankar 1736813a1f2bSShri Abhyankar /* make sure the active row i has diagonal entry */ 1737813a1f2bSShri Abhyankar if (*(bj_ptr[i]+bdiag[i]) != i) { 1738813a1f2bSShri Abhyankar SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 1739813a1f2bSShri Abhyankar try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 1740813a1f2bSShri Abhyankar } 1741813a1f2bSShri Abhyankar 1742813a1f2bSShri Abhyankar current_space->array += nzi; 1743813a1f2bSShri Abhyankar current_space->local_used += nzi; 1744813a1f2bSShri Abhyankar current_space->local_remaining -= nzi; 1745813a1f2bSShri Abhyankar current_space_lvl->array += nzi; 1746813a1f2bSShri Abhyankar current_space_lvl->local_used += nzi; 1747813a1f2bSShri Abhyankar current_space_lvl->local_remaining -= nzi; 1748813a1f2bSShri Abhyankar } 1749813a1f2bSShri Abhyankar 1750813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1751813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 1752813a1f2bSShri Abhyankar 1753813a1f2bSShri Abhyankar /* destroy list of free space and other temporary arrays */ 1754813a1f2bSShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 1755813a1f2bSShri Abhyankar 1756813a1f2bSShri Abhyankar /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 1757f268cba8SShri Abhyankar ierr = PetscFreeSpaceContiguous_LU_v2(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 1758813a1f2bSShri Abhyankar 1759813a1f2bSShri Abhyankar ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1760813a1f2bSShri Abhyankar ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 17610e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 1762813a1f2bSShri Abhyankar 1763813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO) 1764813a1f2bSShri Abhyankar { 1765813a1f2bSShri Abhyankar PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1766813a1f2bSShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 1767813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1768813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 1769813a1f2bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1770813a1f2bSShri Abhyankar if (diagonal_fill) { 1771813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 1772813a1f2bSShri Abhyankar } 1773813a1f2bSShri Abhyankar } 1774813a1f2bSShri Abhyankar #endif 1775813a1f2bSShri Abhyankar 1776813a1f2bSShri Abhyankar /* put together the new matrix */ 1777813a1f2bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1778813a1f2bSShri Abhyankar ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 1779813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1780813a1f2bSShri Abhyankar b->free_a = PETSC_TRUE; 1781813a1f2bSShri Abhyankar b->free_ij = PETSC_TRUE; 1782813a1f2bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 1783f268cba8SShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 1784813a1f2bSShri Abhyankar b->j = bj; 1785813a1f2bSShri Abhyankar b->i = bi; 1786813a1f2bSShri Abhyankar b->diag = bdiag; 1787813a1f2bSShri Abhyankar b->ilen = 0; 1788813a1f2bSShri Abhyankar b->imax = 0; 1789813a1f2bSShri Abhyankar b->row = isrow; 1790813a1f2bSShri Abhyankar b->col = iscol; 1791813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1792813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1793813a1f2bSShri Abhyankar b->icol = isicol; 1794813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1795813a1f2bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. 1796813a1f2bSShri Abhyankar Allocate bdiag, solve_work, new a, new j */ 1797f268cba8SShri Abhyankar ierr = PetscLogObjectMemory(fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 1798f268cba8SShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 1799813a1f2bSShri Abhyankar (fact)->info.factor_mallocs = reallocs; 1800813a1f2bSShri Abhyankar (fact)->info.fill_ratio_given = f; 1801f268cba8SShri Abhyankar (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 1802b4f5729fSShri Abhyankar (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 18034108e4d5SBarry Smith /* ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */ 1804813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1805813a1f2bSShri Abhyankar } 1806813a1f2bSShri Abhyankar 18076516239fSHong Zhang #undef __FUNCT__ 18086516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ" 18096516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 18106516239fSHong Zhang { 18116516239fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 18126516239fSHong Zhang IS isicol; 18136516239fSHong Zhang PetscErrorCode ierr; 18146516239fSHong Zhang const PetscInt *r,*ic; 18156516239fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 18166516239fSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 18176516239fSHong Zhang PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 18186516239fSHong Zhang PetscInt i,levels,diagonal_fill; 18196516239fSHong Zhang PetscTruth col_identity,row_identity; 18206516239fSHong Zhang PetscReal f; 18216516239fSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 18226516239fSHong Zhang PetscBT lnkbt; 18236516239fSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 18246516239fSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 18256516239fSHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 18266516239fSHong Zhang PetscTruth missing; 1827b4f5729fSShri Abhyankar PetscTruth newdatastruct=PETSC_FALSE; 18286516239fSHong Zhang 18296516239fSHong Zhang PetscFunctionBegin; 18306516239fSHong Zhang ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 18316516239fSHong Zhang if(newdatastruct){ 18326516239fSHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr); 18336516239fSHong Zhang PetscFunctionReturn(0); 18346516239fSHong Zhang } 18356516239fSHong Zhang 18366516239fSHong 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); 18376516239fSHong Zhang f = info->fill; 18386516239fSHong Zhang levels = (PetscInt)info->levels; 18396516239fSHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 18406516239fSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 18416516239fSHong Zhang 18426516239fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 18436516239fSHong Zhang ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 18446516239fSHong Zhang if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */ 1845f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr); 1846c9c72f8fSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1847c9c72f8fSHong Zhang 1848719d5645SBarry Smith fact->factor = MAT_FACTOR_ILU; 1849719d5645SBarry Smith (fact)->info.factor_mallocs = 0; 1850719d5645SBarry Smith (fact)->info.fill_ratio_given = info->fill; 1851719d5645SBarry Smith (fact)->info.fill_ratio_needed = 1.0; 1852719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 18538ef3462fSBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 185409f38230SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 185508480c60SBarry Smith b->row = isrow; 185608480c60SBarry Smith b->col = iscol; 185782bf6240SBarry Smith b->icol = isicol; 1858719d5645SBarry Smith ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1859c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1860c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 18614108e4d5SBarry Smith ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); 18623a40ed3dSBarry Smith PetscFunctionReturn(0); 186308480c60SBarry Smith } 186408480c60SBarry Smith 1865898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1866898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 18679e25ed09SBarry Smith 18681bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 18691bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 18701bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1871a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 18726cf997b0SBarry Smith 18730e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr); 18745a8e39fbSHong Zhang 18755a8e39fbSHong Zhang /* create a linked list for storing column indices of the active row */ 18765a8e39fbSHong Zhang nlnk = n + 1; 18775a8e39fbSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 18785a8e39fbSHong Zhang 18795a8e39fbSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 1880a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 18815a8e39fbSHong Zhang current_space = free_space; 1882a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 18835a8e39fbSHong Zhang current_space_lvl = free_space_lvl; 18845a8e39fbSHong Zhang 18855a8e39fbSHong Zhang for (i=0; i<n; i++) { 18865a8e39fbSHong Zhang nzi = 0; 18876cf997b0SBarry Smith /* copy current row into linked list */ 18885a8e39fbSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 18893b4a8b6dSBarry 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); 18905a8e39fbSHong Zhang cols = aj + ai[r[i]]; 18915a8e39fbSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 18925a8e39fbSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 18935a8e39fbSHong Zhang nzi += nlnk; 18946cf997b0SBarry Smith 18956cf997b0SBarry Smith /* make sure diagonal entry is included */ 18965a8e39fbSHong Zhang if (diagonal_fill && lnk[i] == -1) { 18976cf997b0SBarry Smith fm = n; 18985a8e39fbSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 18995a8e39fbSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 19005a8e39fbSHong Zhang lnk[fm] = i; 19015a8e39fbSHong Zhang lnk_lvl[i] = 0; 19025a8e39fbSHong Zhang nzi++; dcount++; 19036cf997b0SBarry Smith } 19046cf997b0SBarry Smith 19055a8e39fbSHong Zhang /* add pivot rows into the active row */ 19065a8e39fbSHong Zhang nzbd = 0; 19075a8e39fbSHong Zhang prow = lnk[n]; 19085a8e39fbSHong Zhang while (prow < i) { 19095a8e39fbSHong Zhang nnz = bdiag[prow]; 19105a8e39fbSHong Zhang cols = bj_ptr[prow] + nnz + 1; 19115a8e39fbSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 19125a8e39fbSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 19135a8e39fbSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 19145a8e39fbSHong Zhang nzi += nlnk; 19155a8e39fbSHong Zhang prow = lnk[prow]; 19165a8e39fbSHong Zhang nzbd++; 191756beaf04SBarry Smith } 19185a8e39fbSHong Zhang bdiag[i] = nzbd; 19195a8e39fbSHong Zhang bi[i+1] = bi[i] + nzi; 1920ecf371e4SBarry Smith 19215a8e39fbSHong Zhang /* if free space is not available, make more free space */ 19225a8e39fbSHong Zhang if (current_space->local_remaining<nzi) { 19235a8e39fbSHong Zhang nnz = nzi*(n - i); /* estimated and max additional space needed */ 1924a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1925a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 19265a8e39fbSHong Zhang reallocs++; 19275a8e39fbSHong Zhang } 1928ecf371e4SBarry Smith 19295a8e39fbSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 19305a8e39fbSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 19315a8e39fbSHong Zhang bj_ptr[i] = current_space->array; 19325a8e39fbSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 19335a8e39fbSHong Zhang 19345a8e39fbSHong Zhang /* make sure the active row i has diagonal entry */ 19355a8e39fbSHong Zhang if (*(bj_ptr[i]+bdiag[i]) != i) { 193677431f27SBarry Smith SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 1937d7ee0231SBarry Smith try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 19386cf997b0SBarry Smith } 19395a8e39fbSHong Zhang 19405a8e39fbSHong Zhang current_space->array += nzi; 19415a8e39fbSHong Zhang current_space->local_used += nzi; 19425a8e39fbSHong Zhang current_space->local_remaining -= nzi; 19435a8e39fbSHong Zhang current_space_lvl->array += nzi; 19445a8e39fbSHong Zhang current_space_lvl->local_used += nzi; 19455a8e39fbSHong Zhang current_space_lvl->local_remaining -= nzi; 19469e25ed09SBarry Smith } 19475a8e39fbSHong Zhang 1948898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1949898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 19505a8e39fbSHong Zhang 19515a8e39fbSHong Zhang /* destroy list of free space and other temporary arrays */ 19525a8e39fbSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 19536516239fSHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */ 19545a8e39fbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1955a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 19560e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 19579e25ed09SBarry Smith 19586cf91177SBarry Smith #if defined(PETSC_USE_INFO) 1959f64065bbSBarry Smith { 19605a8e39fbSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1961ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 1962ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1963ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 1964ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1965335d9088SBarry Smith if (diagonal_fill) { 1966ae15b995SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 1967335d9088SBarry Smith } 1968f64065bbSBarry Smith } 196963ba0a88SBarry Smith #endif 1970bbb6d6a8SBarry Smith 19719e25ed09SBarry Smith /* put together the new matrix */ 197207dbf95fSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1973719d5645SBarry Smith ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 1974719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 1975e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1976e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 19777c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 19780e83c824SBarry Smith ierr = PetscMalloc(bi[n]*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 19795a8e39fbSHong Zhang b->j = bj; 19805a8e39fbSHong Zhang b->i = bi; 19815a8e39fbSHong Zhang for (i=0; i<n; i++) bdiag[i] += bi[i]; 19825a8e39fbSHong Zhang b->diag = bdiag; 1983416022c9SBarry Smith b->ilen = 0; 1984416022c9SBarry Smith b->imax = 0; 1985416022c9SBarry Smith b->row = isrow; 1986416022c9SBarry Smith b->col = iscol; 1987c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1988c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 198982bf6240SBarry Smith b->icol = isicol; 199087828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1991416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 19925a8e39fbSHong Zhang Allocate bdiag, solve_work, new a, new j */ 1993719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 19945a8e39fbSHong Zhang b->maxnz = b->nz = bi[n] ; 1995719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 1996719d5645SBarry Smith (fact)->info.fill_ratio_given = f; 1997719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1998719d5645SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 19994108e4d5SBarry Smith ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); 20003a40ed3dSBarry Smith PetscFunctionReturn(0); 20019e25ed09SBarry Smith } 2002d5d45c9bSBarry Smith 2003a6175056SHong Zhang #undef __FUNCT__ 20045f44c854SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_newdatastruct" 20055f44c854SHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info) 20065f44c854SHong Zhang { 20075f44c854SHong Zhang Mat C = B; 20085f44c854SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 20095f44c854SHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 20105f44c854SHong Zhang IS ip=b->row,iip = b->icol; 20115f44c854SHong Zhang PetscErrorCode ierr; 20125f44c854SHong Zhang const PetscInt *rip,*riip; 20135f44c854SHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp; 20145f44c854SHong Zhang PetscInt *ai=a->i,*aj=a->j; 20155f44c854SHong Zhang PetscInt k,jmin,jmax,*c2r,*il,col,nexti,ili,nz; 20165f44c854SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 20175f44c854SHong Zhang PetscTruth perm_identity; 20185f44c854SHong Zhang 201998a93161SHong Zhang LUShift_Ctx sctx; 202098a93161SHong Zhang PetscInt newshift; 202198a93161SHong Zhang PetscReal rs; 202298a93161SHong Zhang MatScalar d,*v; 202398a93161SHong Zhang 20245f44c854SHong Zhang PetscFunctionBegin; 202598a93161SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 202698a93161SHong Zhang ierr = PetscMemzero(&sctx,sizeof(LUShift_Ctx));CHKERRQ(ierr); 202798a93161SHong Zhang 202898a93161SHong Zhang /* if both shift schemes are chosen by user, only use info->shiftpd */ 202998a93161SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 203098a93161SHong Zhang sctx.shift_top = info->zeropivot; 203198a93161SHong Zhang for (i=0; i<mbs; i++) { 203298a93161SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 203398a93161SHong Zhang d = (aa)[a->diag[i]]; 203498a93161SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 203598a93161SHong Zhang v = aa+ai[i]; 203698a93161SHong Zhang nz = ai[i+1] - ai[i]; 203798a93161SHong Zhang for (j=0; j<nz; j++) 203898a93161SHong Zhang rs += PetscAbsScalar(v[j]); 203998a93161SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 204098a93161SHong Zhang } 204198a93161SHong Zhang sctx.shift_top *= 1.1; 204298a93161SHong Zhang sctx.nshift_max = 5; 204398a93161SHong Zhang sctx.shift_lo = 0.; 204498a93161SHong Zhang sctx.shift_hi = 1.; 204598a93161SHong Zhang } 20465f44c854SHong Zhang 20475f44c854SHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 20485f44c854SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 20495f44c854SHong Zhang 20505f44c854SHong Zhang /* allocate working arrays 20515f44c854SHong Zhang c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col 20525f44c854SHong 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 20535f44c854SHong Zhang */ 20540e83c824SBarry Smith ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&c2r);CHKERRQ(ierr); 20555f44c854SHong Zhang 205698a93161SHong Zhang do { 205798a93161SHong Zhang sctx.lushift = PETSC_FALSE; 205898a93161SHong Zhang 2059c5546cabSHong Zhang for (i=0; i<mbs; i++) c2r[i] = mbs; 20605f44c854SHong Zhang il[0] = 0; 20615f44c854SHong Zhang 20625f44c854SHong Zhang for (k = 0; k<mbs; k++){ 20635f44c854SHong Zhang /* zero rtmp */ 20645f44c854SHong Zhang nz = bi[k+1] - bi[k]; 20655f44c854SHong Zhang bjtmp = bj + bi[k]; 20665f44c854SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 20675f44c854SHong Zhang 20685f44c854SHong Zhang /* load in initial unfactored row */ 20695f44c854SHong Zhang bval = ba + bi[k]; 20705f44c854SHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 20715f44c854SHong Zhang for (j = jmin; j < jmax; j++){ 20725f44c854SHong Zhang col = riip[aj[j]]; 20735f44c854SHong Zhang if (col >= k){ /* only take upper triangular entry */ 20745f44c854SHong Zhang rtmp[col] = aa[j]; 20755f44c854SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 20765f44c854SHong Zhang } 20775f44c854SHong Zhang } 207898a93161SHong Zhang /* shift the diagonal of the matrix: ZeropivotApply() */ 207998a93161SHong Zhang rtmp[k] += sctx.shift_amount; /* shift the diagonal of the matrix */ 20805f44c854SHong Zhang 20815f44c854SHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 20825f44c854SHong Zhang dk = rtmp[k]; 20835f44c854SHong Zhang i = c2r[k]; /* first row to be added to k_th row */ 20845f44c854SHong Zhang 20855f44c854SHong Zhang while (i < k){ 20865f44c854SHong Zhang nexti = c2r[i]; /* next row to be added to k_th row */ 20875f44c854SHong Zhang 20885f44c854SHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 20895f44c854SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 20905f44c854SHong Zhang uikdi = - ba[ili]*ba[bdiag[i]]; /* diagonal(k) */ 20915f44c854SHong Zhang dk += uikdi*ba[ili]; /* update diag[k] */ 20925f44c854SHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 20935f44c854SHong Zhang 20945f44c854SHong Zhang /* add multiple of row i to k-th row */ 20955f44c854SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 20965f44c854SHong Zhang if (jmin < jmax){ 20975f44c854SHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 20985f44c854SHong Zhang /* update il and c2r for row i */ 20995f44c854SHong Zhang il[i] = jmin; 21005f44c854SHong Zhang j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i; 21015f44c854SHong Zhang } 21025f44c854SHong Zhang i = nexti; 21035f44c854SHong Zhang } 21045f44c854SHong Zhang 21055f44c854SHong Zhang /* copy data into U(k,:) */ 210698a93161SHong Zhang rs = 0.0; 21075f44c854SHong Zhang jmin = bi[k]; jmax = bi[k+1]-1; 21085f44c854SHong Zhang if (jmin < jmax) { 21095f44c854SHong Zhang for (j=jmin; j<jmax; j++){ 211098a93161SHong Zhang col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]); 21115f44c854SHong Zhang } 21125f44c854SHong Zhang /* add the k-th row into il and c2r */ 21135f44c854SHong Zhang il[k] = jmin; 21145f44c854SHong Zhang i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k; 21155f44c854SHong Zhang } 211698a93161SHong Zhang 211798a93161SHong Zhang /* MatPivotCheck() */ 211898a93161SHong Zhang sctx.rs = rs; 211998a93161SHong Zhang sctx.pv = dk; 212098a93161SHong Zhang if (info->shiftnz){ 212198a93161SHong Zhang ierr = MatPivotCheck_nz(info,sctx,k,newshift);CHKERRQ(ierr); 212298a93161SHong Zhang } else if (info->shiftpd){ 212398a93161SHong Zhang ierr = MatPivotCheck_pd(info,sctx,k,newshift);CHKERRQ(ierr); 212498a93161SHong Zhang } else if (info->shiftinblocks){ 212598a93161SHong Zhang ierr = MatPivotCheck_inblocks(info,sctx,k,newshift);CHKERRQ(ierr); 212698a93161SHong Zhang } else { 212798a93161SHong Zhang ierr = MatPivotCheck_none(info,sctx,k,newshift);CHKERRQ(ierr); 21285f44c854SHong Zhang } 212998a93161SHong Zhang dk = sctx.pv; 213098a93161SHong Zhang if (newshift == 1) break; 213198a93161SHong Zhang 213298a93161SHong Zhang ba[bdiag[k]] = 1.0/dk; /* U(k,k) */ 213398a93161SHong Zhang } 213498a93161SHong Zhang } while (sctx.lushift); 21355f44c854SHong Zhang 21360e83c824SBarry Smith ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr); 21375f44c854SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 21385f44c854SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 21395f44c854SHong Zhang 21405f44c854SHong Zhang ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 21415f44c854SHong Zhang if (perm_identity){ 21425f44c854SHong Zhang (B)->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering_newdatastruct; 21435f44c854SHong Zhang (B)->ops->solvetranspose = 0; 21445f44c854SHong Zhang (B)->ops->forwardsolve = 0; 21455f44c854SHong Zhang (B)->ops->backwardsolve = 0; 21465f44c854SHong Zhang } else { 2147783ef271SHong Zhang (B)->ops->solve = MatSolve_SeqSBAIJ_1_newdatastruct; 21485f44c854SHong Zhang (B)->ops->solvetranspose = 0; 21495f44c854SHong Zhang (B)->ops->forwardsolve = 0; 21505f44c854SHong Zhang (B)->ops->backwardsolve = 0; 21515f44c854SHong Zhang } 21525f44c854SHong Zhang 21535f44c854SHong Zhang C->assembled = PETSC_TRUE; 21545f44c854SHong Zhang C->preallocated = PETSC_TRUE; 21555f44c854SHong Zhang ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 215698a93161SHong Zhang 215798a93161SHong Zhang /* MatPivotView() */ 215898a93161SHong Zhang if (sctx.nshift){ 215998a93161SHong Zhang if (info->shiftpd) { 216098a93161SHong 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); 216198a93161SHong Zhang } else if (info->shiftnz) { 216298a93161SHong Zhang ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 216398a93161SHong Zhang } else if (info->shiftinblocks){ 216498a93161SHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftinblocks);CHKERRQ(ierr); 216598a93161SHong Zhang } 216698a93161SHong Zhang } 21675f44c854SHong Zhang PetscFunctionReturn(0); 21685f44c854SHong Zhang } 21695f44c854SHong Zhang 21705f44c854SHong Zhang #undef __FUNCT__ 2171a6175056SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ" 21720481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 2173a6175056SHong Zhang { 2174719d5645SBarry Smith Mat C = B; 21750968510aSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 21762ed133dbSHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 21779bfd6278SHong Zhang IS ip=b->row,iip = b->icol; 21782ed133dbSHong Zhang PetscErrorCode ierr; 21795d0c19d7SBarry Smith const PetscInt *rip,*riip; 2180c5546cabSHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp; 21812ed133dbSHong Zhang PetscInt *ai=a->i,*aj=a->j; 21822ed133dbSHong Zhang PetscInt k,jmin,jmax,*jl,*il,col,nexti,ili,nz; 2183622e2028SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 2184540703acSHong Zhang PetscReal zeropivot,rs,shiftnz; 2185fbf22428SSatish Balay PetscReal shiftpd; 2186540703acSHong Zhang ChShift_Ctx sctx; 2187540703acSHong Zhang PetscInt newshift; 2188db4efbfdSBarry Smith PetscTruth perm_identity; 2189d5d45c9bSBarry Smith 2190a6175056SHong Zhang PetscFunctionBegin; 2191540703acSHong Zhang shiftnz = info->shiftnz; 2192540703acSHong Zhang shiftpd = info->shiftpd; 2193ee45ca4aSHong Zhang zeropivot = info->zeropivot; 2194ee45ca4aSHong Zhang 21952ed133dbSHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 21969bfd6278SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 21972ed133dbSHong Zhang 21982ed133dbSHong Zhang /* initialization */ 21990e83c824SBarry Smith ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&jl);CHKERRQ(ierr); 2200540703acSHong Zhang sctx.shift_amount = 0; 2201540703acSHong Zhang sctx.nshift = 0; 22022ed133dbSHong Zhang do { 2203540703acSHong Zhang sctx.chshift = PETSC_FALSE; 2204c5546cabSHong Zhang for (i=0; i<mbs; i++) jl[i] = mbs; 2205c5546cabSHong Zhang il[0] = 0; 22062ed133dbSHong Zhang 22072ed133dbSHong Zhang for (k = 0; k<mbs; k++){ 2208c5546cabSHong Zhang /* zero rtmp */ 2209c5546cabSHong Zhang nz = bi[k+1] - bi[k]; 2210c5546cabSHong Zhang bjtmp = bj + bi[k]; 2211c5546cabSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 2212c5546cabSHong Zhang 2213c05c3958SHong Zhang bval = ba + bi[k]; 22142ed133dbSHong Zhang /* initialize k-th row by the perm[k]-th row of A */ 22152ed133dbSHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 22162ed133dbSHong Zhang for (j = jmin; j < jmax; j++){ 22179bfd6278SHong Zhang col = riip[aj[j]]; 22182ed133dbSHong Zhang if (col >= k){ /* only take upper triangular entry */ 22192ed133dbSHong Zhang rtmp[col] = aa[j]; 2220c05c3958SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 22212ed133dbSHong Zhang } 22222ed133dbSHong Zhang } 222339e3d630SHong Zhang /* shift the diagonal of the matrix */ 2224540703acSHong Zhang if (sctx.nshift) rtmp[k] += sctx.shift_amount; 22252ed133dbSHong Zhang 22262ed133dbSHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 22272ed133dbSHong Zhang dk = rtmp[k]; 22282ed133dbSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 22292ed133dbSHong Zhang 22302ed133dbSHong Zhang while (i < k){ 22312ed133dbSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 22322ed133dbSHong Zhang 22332ed133dbSHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 22342ed133dbSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 22352ed133dbSHong Zhang uikdi = - ba[ili]*ba[bi[i]]; /* diagonal(k) */ 22362ed133dbSHong Zhang dk += uikdi*ba[ili]; 22372ed133dbSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 22382ed133dbSHong Zhang 22392ed133dbSHong Zhang /* add multiple of row i to k-th row */ 22402ed133dbSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 22412ed133dbSHong Zhang if (jmin < jmax){ 22422ed133dbSHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 22432ed133dbSHong Zhang /* update il and jl for row i */ 22442ed133dbSHong Zhang il[i] = jmin; 22452ed133dbSHong Zhang j = bj[jmin]; jl[i] = jl[j]; jl[j] = i; 22462ed133dbSHong Zhang } 22472ed133dbSHong Zhang i = nexti; 22482ed133dbSHong Zhang } 22492ed133dbSHong Zhang 22500697b6caSHong Zhang /* shift the diagonals when zero pivot is detected */ 22510697b6caSHong Zhang /* compute rs=sum of abs(off-diagonal) */ 22520697b6caSHong Zhang rs = 0.0; 22533c9e8598SHong Zhang jmin = bi[k]+1; 22543c9e8598SHong Zhang nz = bi[k+1] - jmin; 22553c9e8598SHong Zhang bcol = bj + jmin; 225604fbf559SBarry Smith for (j=0; j<nz; j++) { 225704fbf559SBarry Smith rs += PetscAbsScalar(rtmp[bcol[j]]); 22583c9e8598SHong Zhang } 2259540703acSHong Zhang 2260540703acSHong Zhang sctx.rs = rs; 2261540703acSHong Zhang sctx.pv = dk; 22625b5bc046SBarry Smith ierr = MatCholeskyCheckShift_inline(info,sctx,k,newshift);CHKERRQ(ierr); 2263117f1891Stmunson 2264117f1891Stmunson if (newshift == 1) { 2265117f1891Stmunson if (!sctx.shift_amount) { 2266117f1891Stmunson sctx.shift_amount = 1e-5; 2267117f1891Stmunson } 2268117f1891Stmunson break; 2269117f1891Stmunson } 22703c9e8598SHong Zhang 22713c9e8598SHong Zhang /* copy data into U(k,:) */ 227239e3d630SHong Zhang ba[bi[k]] = 1.0/dk; /* U(k,k) */ 227339e3d630SHong Zhang jmin = bi[k]+1; jmax = bi[k+1]; 227439e3d630SHong Zhang if (jmin < jmax) { 227539e3d630SHong Zhang for (j=jmin; j<jmax; j++){ 2276c6ae9bfcSHong Zhang col = bj[j]; ba[j] = rtmp[col]; 22773c9e8598SHong Zhang } 227839e3d630SHong Zhang /* add the k-th row into il and jl */ 22793c9e8598SHong Zhang il[k] = jmin; 22803c9e8598SHong Zhang i = bj[jmin]; jl[k] = jl[i]; jl[i] = k; 22813c9e8598SHong Zhang } 22823c9e8598SHong Zhang } 2283540703acSHong Zhang } while (sctx.chshift); 22840e83c824SBarry Smith ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr); 228539e3d630SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 22869bfd6278SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 2287db4efbfdSBarry Smith 2288db4efbfdSBarry Smith ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 2289db4efbfdSBarry Smith if (perm_identity){ 2290719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 2291719d5645SBarry Smith (B)->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 2292719d5645SBarry Smith (B)->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering; 2293719d5645SBarry Smith (B)->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering; 2294db4efbfdSBarry Smith } else { 2295719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqSBAIJ_1; 2296719d5645SBarry Smith (B)->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 2297719d5645SBarry Smith (B)->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1; 2298719d5645SBarry Smith (B)->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1; 2299db4efbfdSBarry Smith } 2300db4efbfdSBarry Smith 23013c9e8598SHong Zhang C->assembled = PETSC_TRUE; 23023c9e8598SHong Zhang C->preallocated = PETSC_TRUE; 2303d0f46423SBarry Smith ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 2304540703acSHong Zhang if (sctx.nshift){ 2305540703acSHong Zhang if (shiftnz) { 23061e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2307540703acSHong Zhang } else if (shiftpd) { 23081e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 23090697b6caSHong Zhang } 23103c9e8598SHong Zhang } 23113c9e8598SHong Zhang PetscFunctionReturn(0); 23123c9e8598SHong Zhang } 2313a6175056SHong Zhang 23148a76ed4fSHong Zhang /* 23158a76ed4fSHong Zhang icc() under revised new data structure. 23168a76ed4fSHong Zhang Factored arrays bj and ba are stored as 23178a76ed4fSHong Zhang U(0,:),...,U(i,:),U(n-1,:) 23188a76ed4fSHong Zhang 23198a76ed4fSHong Zhang ui=fact->i is an array of size n+1, in which 23208a76ed4fSHong Zhang ui+ 23218a76ed4fSHong Zhang ui[i]: points to 1st entry of U(i,:),i=0,...,n-1 23228a76ed4fSHong Zhang ui[n]: points to U(n-1,n-1)+1 23238a76ed4fSHong Zhang 23248a76ed4fSHong Zhang udiag=fact->diag is an array of size n,in which 23258a76ed4fSHong Zhang udiag[i]: points to diagonal of U(i,:), i=0,...,n-1 23268a76ed4fSHong Zhang 23278a76ed4fSHong Zhang U(i,:) contains udiag[i] as its last entry, i.e., 23288a76ed4fSHong Zhang U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 23298a76ed4fSHong Zhang */ 23308a76ed4fSHong Zhang 2331a6175056SHong Zhang #undef __FUNCT__ 23325f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_newdatastruct" 23335f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2334a6175056SHong Zhang { 23350968510aSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2336ed59401aSHong Zhang Mat_SeqSBAIJ *b; 2337dfbe8321SBarry Smith PetscErrorCode ierr; 233858ebbce7SBarry Smith PetscTruth perm_identity,missing; 23395f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 23405d0c19d7SBarry Smith const PetscInt *rip,*riip; 2341ed59401aSHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 234258ebbce7SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 23435a8e39fbSHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 2344ed59401aSHong Zhang PetscReal fill=info->fill,levels=info->levels; 2345a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 2346a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 23477a48dd6fSHong Zhang PetscBT lnkbt; 2348b635d36bSHong Zhang IS iperm; 2349a6175056SHong Zhang 2350a6175056SHong Zhang PetscFunctionBegin; 2351d0f46423SBarry 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); 235258ebbce7SBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 235358ebbce7SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 2354653a6975SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 2355b635d36bSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 23567a48dd6fSHong Zhang 235739e3d630SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 23585f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 235939e3d630SHong Zhang ui[0] = 0; 236039e3d630SHong Zhang 23618a76ed4fSHong Zhang /* ICC(0) without matrix ordering: simply rearrange column indices */ 2362622e2028SHong Zhang if (!levels && perm_identity) { 23635f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 23645f44c854SHong Zhang cols = uj; 23655f44c854SHong Zhang for (i=0; i<am; i++) { 2366c5546cabSHong Zhang ncols = ai[i+1] - a->diag[i]; 2367c5546cabSHong Zhang ui[i+1] = ui[i] + ncols; 2368c5546cabSHong Zhang udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */ 2369c5546cabSHong Zhang 23705f44c854SHong Zhang aj = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */ 2371c5546cabSHong Zhang ncols--; /* exclude diagonal */ 23725f44c854SHong Zhang for (j=0; j<ncols; j++) *cols++ = aj[j]; 23735f44c854SHong Zhang *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */ 23745f44c854SHong Zhang } 23755f44c854SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 23765f44c854SHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 23775f44c854SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 23785f44c854SHong Zhang 23795f44c854SHong Zhang /* initialization */ 23805f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 23815f44c854SHong Zhang 23825f44c854SHong Zhang /* jl: linked list for storing indices of the pivot rows 23835f44c854SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 23840e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr); 23855f44c854SHong Zhang for (i=0; i<am; i++){ 23865f44c854SHong Zhang jl[i] = am; il[i] = 0; 23875f44c854SHong Zhang } 23885f44c854SHong Zhang 23895f44c854SHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 23905f44c854SHong Zhang nlnk = am + 1; 23915f44c854SHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 23925f44c854SHong Zhang 23935f44c854SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 23945f44c854SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 23955f44c854SHong Zhang current_space = free_space; 23965f44c854SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 23975f44c854SHong Zhang current_space_lvl = free_space_lvl; 23985f44c854SHong Zhang 23995f44c854SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 24005f44c854SHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 24015f44c854SHong Zhang nzk = 0; 24025f44c854SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 24035f44c854SHong 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); 24045f44c854SHong Zhang ncols_upper = 0; 24055f44c854SHong Zhang for (j=0; j<ncols; j++){ 24065f44c854SHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 24075f44c854SHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 24085f44c854SHong Zhang ajtmp[ncols_upper] = i; 24095f44c854SHong Zhang ncols_upper++; 24105f44c854SHong Zhang } 24115f44c854SHong Zhang } 24125f44c854SHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 24135f44c854SHong Zhang nzk += nlnk; 24145f44c854SHong Zhang 24155f44c854SHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 24165f44c854SHong Zhang prow = jl[k]; /* 1st pivot row */ 24175f44c854SHong Zhang 24185f44c854SHong Zhang while (prow < k){ 24195f44c854SHong Zhang nextprow = jl[prow]; 24205f44c854SHong Zhang 24215f44c854SHong Zhang /* merge prow into k-th row */ 24225f44c854SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 24235f44c854SHong Zhang jmax = ui[prow+1]; 24245f44c854SHong Zhang ncols = jmax-jmin; 24255f44c854SHong Zhang i = jmin - ui[prow]; 24265f44c854SHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 24275f44c854SHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 24285f44c854SHong Zhang j = *(uj - 1); 24295f44c854SHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 24305f44c854SHong Zhang nzk += nlnk; 24315f44c854SHong Zhang 24325f44c854SHong Zhang /* update il and jl for prow */ 24335f44c854SHong Zhang if (jmin < jmax){ 24345f44c854SHong Zhang il[prow] = jmin; 24355f44c854SHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 24365f44c854SHong Zhang } 24375f44c854SHong Zhang prow = nextprow; 24385f44c854SHong Zhang } 24395f44c854SHong Zhang 24405f44c854SHong Zhang /* if free space is not available, make more free space */ 24415f44c854SHong Zhang if (current_space->local_remaining<nzk) { 24425f44c854SHong Zhang i = am - k + 1; /* num of unfactored rows */ 2443c6ae9bfcSHong Zhang i *= PetscMin(nzk, i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */ 24445f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 24455f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 24465f44c854SHong Zhang reallocs++; 24475f44c854SHong Zhang } 24485f44c854SHong Zhang 24495f44c854SHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 24505f44c854SHong Zhang if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 24515f44c854SHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 24525f44c854SHong Zhang 24535f44c854SHong Zhang /* add the k-th row into il and jl */ 24545f44c854SHong Zhang if (nzk > 1){ 24555f44c854SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 24565f44c854SHong Zhang jl[k] = jl[i]; jl[i] = k; 24575f44c854SHong Zhang il[k] = ui[k] + 1; 24585f44c854SHong Zhang } 24595f44c854SHong Zhang uj_ptr[k] = current_space->array; 24605f44c854SHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 24615f44c854SHong Zhang 24625f44c854SHong Zhang current_space->array += nzk; 24635f44c854SHong Zhang current_space->local_used += nzk; 24645f44c854SHong Zhang current_space->local_remaining -= nzk; 24655f44c854SHong Zhang 24665f44c854SHong Zhang current_space_lvl->array += nzk; 24675f44c854SHong Zhang current_space_lvl->local_used += nzk; 24685f44c854SHong Zhang current_space_lvl->local_remaining -= nzk; 24695f44c854SHong Zhang 24705f44c854SHong Zhang ui[k+1] = ui[k] + nzk; 24715f44c854SHong Zhang } 24725f44c854SHong Zhang 24735f44c854SHong Zhang #if defined(PETSC_USE_INFO) 24745f44c854SHong Zhang if (ai[am] != 0) { 24755f44c854SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 24765f44c854SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 24775f44c854SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 24785f44c854SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 24795f44c854SHong Zhang } else { 24805f44c854SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 24815f44c854SHong Zhang } 24825f44c854SHong Zhang #endif 24835f44c854SHong Zhang 24845f44c854SHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 24855f44c854SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 24860e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 24875f44c854SHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 24885f44c854SHong Zhang 24895f44c854SHong Zhang /* destroy list of free space and other temporary array(s) */ 24905f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2491c5546cabSHong Zhang ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor in newdatastruct */ 24925f44c854SHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 24935f44c854SHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 24945f44c854SHong Zhang 24955f44c854SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 24965f44c854SHong Zhang 24975f44c854SHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 24985f44c854SHong Zhang 24995f44c854SHong Zhang b = (Mat_SeqSBAIJ*)(fact)->data; 25005f44c854SHong Zhang b->singlemalloc = PETSC_FALSE; 25015f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 25025f44c854SHong Zhang b->j = uj; 25035f44c854SHong Zhang b->i = ui; 25045f44c854SHong Zhang b->diag = udiag; 25057f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 25065f44c854SHong Zhang b->ilen = 0; 25075f44c854SHong Zhang b->imax = 0; 25085f44c854SHong Zhang b->row = perm; 25095f44c854SHong Zhang b->col = perm; 25105f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25115f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25125f44c854SHong Zhang b->icol = iperm; 25135f44c854SHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 25145f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 25155f44c854SHong Zhang ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 25165f44c854SHong Zhang b->maxnz = b->nz = ui[am]; 25175f44c854SHong Zhang b->free_a = PETSC_TRUE; 25185f44c854SHong Zhang b->free_ij = PETSC_TRUE; 25195f44c854SHong Zhang 25205f44c854SHong Zhang (fact)->info.factor_mallocs = reallocs; 25215f44c854SHong Zhang (fact)->info.fill_ratio_given = fill; 25225f44c854SHong Zhang if (ai[am] != 0) { 25235f44c854SHong Zhang (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 25245f44c854SHong Zhang } else { 25255f44c854SHong Zhang (fact)->info.fill_ratio_needed = 0.0; 25265f44c854SHong Zhang } 25275f44c854SHong Zhang (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_newdatastruct; 25285f44c854SHong Zhang PetscFunctionReturn(0); 25295f44c854SHong Zhang } 25305f44c854SHong Zhang 25315f44c854SHong Zhang #undef __FUNCT__ 25325f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ" 25335f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 25345f44c854SHong Zhang { 25355f44c854SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 25365f44c854SHong Zhang Mat_SeqSBAIJ *b; 25375f44c854SHong Zhang PetscErrorCode ierr; 25385f44c854SHong Zhang PetscTruth perm_identity,missing; 25395f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 25405f44c854SHong Zhang const PetscInt *rip,*riip; 25415f44c854SHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 25425f44c854SHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 25435f44c854SHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 25445f44c854SHong Zhang PetscReal fill=info->fill,levels=info->levels; 25455f44c854SHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 25465f44c854SHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 25475f44c854SHong Zhang PetscBT lnkbt; 25485f44c854SHong Zhang IS iperm; 25495f44c854SHong Zhang PetscTruth newdatastruct=PETSC_FALSE; 25505f44c854SHong Zhang 25515f44c854SHong Zhang PetscFunctionBegin; 25525f44c854SHong Zhang ierr = PetscOptionsGetTruth(PETSC_NULL,"-icc_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 25535f44c854SHong Zhang if(newdatastruct){ 25545f44c854SHong Zhang ierr = MatICCFactorSymbolic_SeqAIJ_newdatastruct(fact,A,perm,info);CHKERRQ(ierr); 25555f44c854SHong Zhang PetscFunctionReturn(0); 25565f44c854SHong Zhang } 25575f44c854SHong Zhang 25585f44c854SHong 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); 25595f44c854SHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 25605f44c854SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 25615f44c854SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 25625f44c854SHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 25635f44c854SHong Zhang 25645f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 25655f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 25665f44c854SHong Zhang ui[0] = 0; 25675f44c854SHong Zhang 25685f44c854SHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 25695f44c854SHong Zhang if (!levels && perm_identity) { 25705f44c854SHong Zhang 25715f44c854SHong Zhang for (i=0; i<am; i++) { 25725f44c854SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 25735f44c854SHong Zhang udiag[i] = ui[i]; 2574ed59401aSHong Zhang } 257539e3d630SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 257639e3d630SHong Zhang cols = uj; 2577ed59401aSHong Zhang for (i=0; i<am; i++) { 2578ed59401aSHong Zhang aj = a->j + a->diag[i]; 257939e3d630SHong Zhang ncols = ui[i+1] - ui[i]; 258039e3d630SHong Zhang for (j=0; j<ncols; j++) *cols++ = *aj++; 2581ed59401aSHong Zhang } 258239e3d630SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 2583b635d36bSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 2584b635d36bSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 2585b635d36bSHong Zhang 25867a48dd6fSHong Zhang /* initialization */ 25875a8e39fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 25887a48dd6fSHong Zhang 25897a48dd6fSHong Zhang /* jl: linked list for storing indices of the pivot rows 25907a48dd6fSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 25910e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr); 25927a48dd6fSHong Zhang for (i=0; i<am; i++){ 25937a48dd6fSHong Zhang jl[i] = am; il[i] = 0; 25947a48dd6fSHong Zhang } 25957a48dd6fSHong Zhang 25967a48dd6fSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 25977a48dd6fSHong Zhang nlnk = am + 1; 25982ed133dbSHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 25997a48dd6fSHong Zhang 26007a48dd6fSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2601a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 26027a48dd6fSHong Zhang current_space = free_space; 2603a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 26047a48dd6fSHong Zhang current_space_lvl = free_space_lvl; 26057a48dd6fSHong Zhang 26067a48dd6fSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 26077a48dd6fSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 26087a48dd6fSHong Zhang nzk = 0; 2609622e2028SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 26103b4a8b6dSBarry 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); 2611622e2028SHong Zhang ncols_upper = 0; 2612622e2028SHong Zhang for (j=0; j<ncols; j++){ 2613b635d36bSHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 2614b635d36bSHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 26155a8e39fbSHong Zhang ajtmp[ncols_upper] = i; 2616622e2028SHong Zhang ncols_upper++; 2617622e2028SHong Zhang } 2618622e2028SHong Zhang } 2619b635d36bSHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 26207a48dd6fSHong Zhang nzk += nlnk; 26217a48dd6fSHong Zhang 26227a48dd6fSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 26237a48dd6fSHong Zhang prow = jl[k]; /* 1st pivot row */ 26247a48dd6fSHong Zhang 26257a48dd6fSHong Zhang while (prow < k){ 26267a48dd6fSHong Zhang nextprow = jl[prow]; 26277a48dd6fSHong Zhang 26287a48dd6fSHong Zhang /* merge prow into k-th row */ 26297a48dd6fSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 26307a48dd6fSHong Zhang jmax = ui[prow+1]; 26317a48dd6fSHong Zhang ncols = jmax-jmin; 2632ed59401aSHong Zhang i = jmin - ui[prow]; 2633ed59401aSHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 26345a8e39fbSHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 26355a8e39fbSHong Zhang j = *(uj - 1); 26365a8e39fbSHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 26377a48dd6fSHong Zhang nzk += nlnk; 26387a48dd6fSHong Zhang 26397a48dd6fSHong Zhang /* update il and jl for prow */ 26407a48dd6fSHong Zhang if (jmin < jmax){ 26417a48dd6fSHong Zhang il[prow] = jmin; 26427a48dd6fSHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 26437a48dd6fSHong Zhang } 26447a48dd6fSHong Zhang prow = nextprow; 26457a48dd6fSHong Zhang } 26467a48dd6fSHong Zhang 26477a48dd6fSHong Zhang /* if free space is not available, make more free space */ 26487a48dd6fSHong Zhang if (current_space->local_remaining<nzk) { 26497a48dd6fSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2650fd3f55cbSMatthew Knepley i *= PetscMin(nzk, (i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2651a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 2652a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 26537a48dd6fSHong Zhang reallocs++; 26547a48dd6fSHong Zhang } 26557a48dd6fSHong Zhang 26567a48dd6fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 2657b635d36bSHong Zhang if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 26582ed133dbSHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 26597a48dd6fSHong Zhang 26607a48dd6fSHong Zhang /* add the k-th row into il and jl */ 266139e3d630SHong Zhang if (nzk > 1){ 26627a48dd6fSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 26637a48dd6fSHong Zhang jl[k] = jl[i]; jl[i] = k; 26647a48dd6fSHong Zhang il[k] = ui[k] + 1; 26657a48dd6fSHong Zhang } 26667a48dd6fSHong Zhang uj_ptr[k] = current_space->array; 26677a48dd6fSHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 26687a48dd6fSHong Zhang 26697a48dd6fSHong Zhang current_space->array += nzk; 26707a48dd6fSHong Zhang current_space->local_used += nzk; 26717a48dd6fSHong Zhang current_space->local_remaining -= nzk; 26727a48dd6fSHong Zhang 26737a48dd6fSHong Zhang current_space_lvl->array += nzk; 26747a48dd6fSHong Zhang current_space_lvl->local_used += nzk; 26757a48dd6fSHong Zhang current_space_lvl->local_remaining -= nzk; 26767a48dd6fSHong Zhang 26777a48dd6fSHong Zhang ui[k+1] = ui[k] + nzk; 26787a48dd6fSHong Zhang } 26797a48dd6fSHong Zhang 26806cf91177SBarry Smith #if defined(PETSC_USE_INFO) 26817a48dd6fSHong Zhang if (ai[am] != 0) { 268239e3d630SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 2683ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2684ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2685ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 26867a48dd6fSHong Zhang } else { 2687ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 26887a48dd6fSHong Zhang } 268963ba0a88SBarry Smith #endif 26907a48dd6fSHong Zhang 26917a48dd6fSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 2692b635d36bSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 26930e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 26945a8e39fbSHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 26957a48dd6fSHong Zhang 26967a48dd6fSHong Zhang /* destroy list of free space and other temporary array(s) */ 26977a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2698a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 26992ed133dbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2700a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 27017a48dd6fSHong Zhang 270239e3d630SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 270339e3d630SHong Zhang 27047a48dd6fSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 27057a48dd6fSHong Zhang 2706719d5645SBarry Smith b = (Mat_SeqSBAIJ*)(fact)->data; 27077a48dd6fSHong Zhang b->singlemalloc = PETSC_FALSE; 27087a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 27097a48dd6fSHong Zhang b->j = uj; 27107a48dd6fSHong Zhang b->i = ui; 27115f44c854SHong Zhang b->diag = udiag; 27127f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 27137a48dd6fSHong Zhang b->ilen = 0; 27147a48dd6fSHong Zhang b->imax = 0; 27157a48dd6fSHong Zhang b->row = perm; 2716b635d36bSHong Zhang b->col = perm; 2717b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2718b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2719b635d36bSHong Zhang b->icol = iperm; 27207a48dd6fSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 27217a48dd6fSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2722719d5645SBarry Smith ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 27237a48dd6fSHong Zhang b->maxnz = b->nz = ui[am]; 2724e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2725e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 27267a48dd6fSHong Zhang 2727719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2728719d5645SBarry Smith (fact)->info.fill_ratio_given = fill; 27297a48dd6fSHong Zhang if (ai[am] != 0) { 2730719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 27317a48dd6fSHong Zhang } else { 2732719d5645SBarry Smith (fact)->info.fill_ratio_needed = 0.0; 27337a48dd6fSHong Zhang } 2734719d5645SBarry Smith (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 2735a6175056SHong Zhang PetscFunctionReturn(0); 2736a6175056SHong Zhang } 2737d5d45c9bSBarry Smith 27380be760fbSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 27390be760fbSHong Zhang { 27400be760fbSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 27410be760fbSHong Zhang Mat_SeqSBAIJ *b; 27420be760fbSHong Zhang PetscErrorCode ierr; 27430be760fbSHong Zhang PetscTruth perm_identity; 27440be760fbSHong Zhang PetscReal fill = info->fill; 27450be760fbSHong Zhang const PetscInt *rip,*riip; 27460be760fbSHong Zhang PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 27470be760fbSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 27480be760fbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag; 27490be760fbSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 27500be760fbSHong Zhang PetscBT lnkbt; 27510be760fbSHong Zhang IS iperm; 27520be760fbSHong Zhang 27530be760fbSHong Zhang PetscFunctionBegin; 27540be760fbSHong 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); 27550be760fbSHong Zhang /* check whether perm is the identity mapping */ 27560be760fbSHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 27570be760fbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 27580be760fbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 27590be760fbSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 27600be760fbSHong Zhang 27610be760fbSHong Zhang /* initialization */ 27620be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 27630be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 27640be760fbSHong Zhang ui[0] = 0; 27650be760fbSHong Zhang 27660be760fbSHong Zhang /* jl: linked list for storing indices of the pivot rows 27670be760fbSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 27680be760fbSHong Zhang ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr); 27690be760fbSHong Zhang for (i=0; i<am; i++){ 27700be760fbSHong Zhang jl[i] = am; il[i] = 0; 27710be760fbSHong Zhang } 27720be760fbSHong Zhang 27730be760fbSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 27740be760fbSHong Zhang nlnk = am + 1; 27750be760fbSHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 27760be760fbSHong Zhang 27770be760fbSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 27780be760fbSHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 27790be760fbSHong Zhang current_space = free_space; 27800be760fbSHong Zhang 27810be760fbSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 27820be760fbSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 27830be760fbSHong Zhang nzk = 0; 27840be760fbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 27850be760fbSHong 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); 27860be760fbSHong Zhang ncols_upper = 0; 27870be760fbSHong Zhang for (j=0; j<ncols; j++){ 27880be760fbSHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 27890be760fbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 27900be760fbSHong Zhang cols[ncols_upper] = i; 27910be760fbSHong Zhang ncols_upper++; 27920be760fbSHong Zhang } 27930be760fbSHong Zhang } 27940be760fbSHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 27950be760fbSHong Zhang nzk += nlnk; 27960be760fbSHong Zhang 27970be760fbSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 27980be760fbSHong Zhang prow = jl[k]; /* 1st pivot row */ 27990be760fbSHong Zhang 28000be760fbSHong Zhang while (prow < k){ 28010be760fbSHong Zhang nextprow = jl[prow]; 28020be760fbSHong Zhang /* merge prow into k-th row */ 28030be760fbSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 28040be760fbSHong Zhang jmax = ui[prow+1]; 28050be760fbSHong Zhang ncols = jmax-jmin; 28060be760fbSHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 28070be760fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28080be760fbSHong Zhang nzk += nlnk; 28090be760fbSHong Zhang 28100be760fbSHong Zhang /* update il and jl for prow */ 28110be760fbSHong Zhang if (jmin < jmax){ 28120be760fbSHong Zhang il[prow] = jmin; 28130be760fbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 28140be760fbSHong Zhang } 28150be760fbSHong Zhang prow = nextprow; 28160be760fbSHong Zhang } 28170be760fbSHong Zhang 28180be760fbSHong Zhang /* if free space is not available, make more free space */ 28190be760fbSHong Zhang if (current_space->local_remaining<nzk) { 28200be760fbSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2821c6ae9bfcSHong Zhang i *= PetscMin(nzk,i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */ 28220be760fbSHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 28230be760fbSHong Zhang reallocs++; 28240be760fbSHong Zhang } 28250be760fbSHong Zhang 28260be760fbSHong Zhang /* copy data into free space, then initialize lnk */ 28270be760fbSHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 28280be760fbSHong Zhang 28290be760fbSHong Zhang /* add the k-th row into il and jl */ 28300be760fbSHong Zhang if (nzk-1 > 0){ 28310be760fbSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 28320be760fbSHong Zhang jl[k] = jl[i]; jl[i] = k; 28330be760fbSHong Zhang il[k] = ui[k] + 1; 28340be760fbSHong Zhang } 28350be760fbSHong Zhang ui_ptr[k] = current_space->array; 28360be760fbSHong Zhang current_space->array += nzk; 28370be760fbSHong Zhang current_space->local_used += nzk; 28380be760fbSHong Zhang current_space->local_remaining -= nzk; 28390be760fbSHong Zhang 28400be760fbSHong Zhang ui[k+1] = ui[k] + nzk; 28410be760fbSHong Zhang } 28420be760fbSHong Zhang 28430be760fbSHong Zhang #if defined(PETSC_USE_INFO) 28440be760fbSHong Zhang if (ai[am] != 0) { 28450be760fbSHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 28460be760fbSHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 28470be760fbSHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 28480be760fbSHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 28490be760fbSHong Zhang } else { 28500be760fbSHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 28510be760fbSHong Zhang } 28520be760fbSHong Zhang #endif 28530be760fbSHong Zhang 28540be760fbSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 28550be760fbSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 28560be760fbSHong Zhang ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 28570be760fbSHong Zhang 28580be760fbSHong Zhang /* destroy list of free space and other temporary array(s) */ 28590be760fbSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 28600be760fbSHong Zhang ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor in newdatastruct */ 28610be760fbSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 28620be760fbSHong Zhang 28630be760fbSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 28640be760fbSHong Zhang 28650be760fbSHong Zhang b = (Mat_SeqSBAIJ*)(fact)->data; 28660be760fbSHong Zhang b->singlemalloc = PETSC_FALSE; 28670be760fbSHong Zhang b->free_a = PETSC_TRUE; 28680be760fbSHong Zhang b->free_ij = PETSC_TRUE; 28690be760fbSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 28700be760fbSHong Zhang b->j = uj; 28710be760fbSHong Zhang b->i = ui; 28720be760fbSHong Zhang b->diag = udiag; 28730be760fbSHong Zhang b->free_diag = PETSC_TRUE; 28740be760fbSHong Zhang b->ilen = 0; 28750be760fbSHong Zhang b->imax = 0; 28760be760fbSHong Zhang b->row = perm; 28770be760fbSHong Zhang b->col = perm; 28780be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 28790be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 28800be760fbSHong Zhang b->icol = iperm; 28810be760fbSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 28820be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 28830be760fbSHong Zhang ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 28840be760fbSHong Zhang b->maxnz = b->nz = ui[am]; 28850be760fbSHong Zhang 28860be760fbSHong Zhang (fact)->info.factor_mallocs = reallocs; 28870be760fbSHong Zhang (fact)->info.fill_ratio_given = fill; 28880be760fbSHong Zhang if (ai[am] != 0) { 28890be760fbSHong Zhang (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 28900be760fbSHong Zhang } else { 28910be760fbSHong Zhang (fact)->info.fill_ratio_needed = 0.0; 28920be760fbSHong Zhang } 28930be760fbSHong Zhang (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_newdatastruct; 28940be760fbSHong Zhang PetscFunctionReturn(0); 28950be760fbSHong Zhang } 28960be760fbSHong Zhang 2897f76d2b81SHong Zhang #undef __FUNCT__ 2898f76d2b81SHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ" 28990481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2900f76d2b81SHong Zhang { 2901f76d2b81SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 290210c27e3dSHong Zhang Mat_SeqSBAIJ *b; 2903dfbe8321SBarry Smith PetscErrorCode ierr; 2904f76d2b81SHong Zhang PetscTruth perm_identity; 290510c27e3dSHong Zhang PetscReal fill = info->fill; 29065d0c19d7SBarry Smith const PetscInt *rip,*riip; 29075d0c19d7SBarry Smith PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 290810c27e3dSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 29092ed133dbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr; 2910a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 291110c27e3dSHong Zhang PetscBT lnkbt; 29122ed133dbSHong Zhang IS iperm; 29130be760fbSHong Zhang PetscTruth newdatastruct=PETSC_FALSE; 2914f76d2b81SHong Zhang 2915f76d2b81SHong Zhang PetscFunctionBegin; 29160be760fbSHong Zhang ierr = PetscOptionsGetTruth(PETSC_NULL,"-cholesky_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 29170be760fbSHong Zhang if(newdatastruct){ 29180be760fbSHong Zhang ierr = MatCholeskyFactorSymbolic_SeqAIJ_newdatastruct(fact,A,perm,info);CHKERRQ(ierr); 29190be760fbSHong Zhang PetscFunctionReturn(0); 29200be760fbSHong Zhang } 29210be760fbSHong Zhang 2922d0f46423SBarry 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); 29232ed133dbSHong Zhang /* check whether perm is the identity mapping */ 2924f76d2b81SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 29252ed133dbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 29262ed133dbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 29279bfd6278SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 292810c27e3dSHong Zhang 292910c27e3dSHong Zhang /* initialization */ 29301393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 293110c27e3dSHong Zhang ui[0] = 0; 293210c27e3dSHong Zhang 293310c27e3dSHong Zhang /* jl: linked list for storing indices of the pivot rows 29341393bc97SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 29350e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr); 29361393bc97SHong Zhang for (i=0; i<am; i++){ 29371393bc97SHong Zhang jl[i] = am; il[i] = 0; 2938f76d2b81SHong Zhang } 293910c27e3dSHong Zhang 294010c27e3dSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 29411393bc97SHong Zhang nlnk = am + 1; 29421393bc97SHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 294310c27e3dSHong Zhang 29441393bc97SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2945a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 294610c27e3dSHong Zhang current_space = free_space; 294710c27e3dSHong Zhang 29481393bc97SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 294910c27e3dSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 295010c27e3dSHong Zhang nzk = 0; 29512ed133dbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 29523b4a8b6dSBarry 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); 29532ed133dbSHong Zhang ncols_upper = 0; 29542ed133dbSHong Zhang for (j=0; j<ncols; j++){ 29559bfd6278SHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 29562ed133dbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 29572ed133dbSHong Zhang cols[ncols_upper] = i; 29582ed133dbSHong Zhang ncols_upper++; 29592ed133dbSHong Zhang } 29602ed133dbSHong Zhang } 29611393bc97SHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 296210c27e3dSHong Zhang nzk += nlnk; 296310c27e3dSHong Zhang 296410c27e3dSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 296510c27e3dSHong Zhang prow = jl[k]; /* 1st pivot row */ 296610c27e3dSHong Zhang 296710c27e3dSHong Zhang while (prow < k){ 296810c27e3dSHong Zhang nextprow = jl[prow]; 296910c27e3dSHong Zhang /* merge prow into k-th row */ 29701393bc97SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 297110c27e3dSHong Zhang jmax = ui[prow+1]; 297210c27e3dSHong Zhang ncols = jmax-jmin; 29731393bc97SHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 29745a8e39fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 297510c27e3dSHong Zhang nzk += nlnk; 297610c27e3dSHong Zhang 297710c27e3dSHong Zhang /* update il and jl for prow */ 297810c27e3dSHong Zhang if (jmin < jmax){ 297910c27e3dSHong Zhang il[prow] = jmin; 29802ed133dbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 298110c27e3dSHong Zhang } 298210c27e3dSHong Zhang prow = nextprow; 298310c27e3dSHong Zhang } 298410c27e3dSHong Zhang 298510c27e3dSHong Zhang /* if free space is not available, make more free space */ 298610c27e3dSHong Zhang if (current_space->local_remaining<nzk) { 29871393bc97SHong Zhang i = am - k + 1; /* num of unfactored rows */ 298810c27e3dSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2989a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 299010c27e3dSHong Zhang reallocs++; 299110c27e3dSHong Zhang } 299210c27e3dSHong Zhang 299310c27e3dSHong Zhang /* copy data into free space, then initialize lnk */ 29941393bc97SHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 299510c27e3dSHong Zhang 299610c27e3dSHong Zhang /* add the k-th row into il and jl */ 299710c27e3dSHong Zhang if (nzk-1 > 0){ 29981393bc97SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 299910c27e3dSHong Zhang jl[k] = jl[i]; jl[i] = k; 300010c27e3dSHong Zhang il[k] = ui[k] + 1; 300110c27e3dSHong Zhang } 30022ed133dbSHong Zhang ui_ptr[k] = current_space->array; 300310c27e3dSHong Zhang current_space->array += nzk; 300410c27e3dSHong Zhang current_space->local_used += nzk; 300510c27e3dSHong Zhang current_space->local_remaining -= nzk; 300610c27e3dSHong Zhang 300710c27e3dSHong Zhang ui[k+1] = ui[k] + nzk; 300810c27e3dSHong Zhang } 300910c27e3dSHong Zhang 30106cf91177SBarry Smith #if defined(PETSC_USE_INFO) 30111393bc97SHong Zhang if (ai[am] != 0) { 30121393bc97SHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 3013ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 3014ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 3015ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 301610c27e3dSHong Zhang } else { 3017ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 301810c27e3dSHong Zhang } 301963ba0a88SBarry Smith #endif 302010c27e3dSHong Zhang 302110c27e3dSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 30229bfd6278SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 30230e83c824SBarry Smith ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 302410c27e3dSHong Zhang 302510c27e3dSHong Zhang /* destroy list of free space and other temporary array(s) */ 30261393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 3027a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 302810c27e3dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 302910c27e3dSHong Zhang 303010c27e3dSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 303110c27e3dSHong Zhang 3032719d5645SBarry Smith b = (Mat_SeqSBAIJ*)(fact)->data; 303310c27e3dSHong Zhang b->singlemalloc = PETSC_FALSE; 3034e6b907acSBarry Smith b->free_a = PETSC_TRUE; 3035e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 30361393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 303710c27e3dSHong Zhang b->j = uj; 303810c27e3dSHong Zhang b->i = ui; 303910c27e3dSHong Zhang b->diag = 0; 304010c27e3dSHong Zhang b->ilen = 0; 304110c27e3dSHong Zhang b->imax = 0; 304210c27e3dSHong Zhang b->row = perm; 30439bfd6278SHong Zhang b->col = perm; 30449bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 30459bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 30469bfd6278SHong Zhang b->icol = iperm; 304710c27e3dSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 30481393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3049719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 30501393bc97SHong Zhang b->maxnz = b->nz = ui[am]; 305110c27e3dSHong Zhang 3052719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 3053719d5645SBarry Smith (fact)->info.fill_ratio_given = fill; 30541393bc97SHong Zhang if (ai[am] != 0) { 3055719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 305610c27e3dSHong Zhang } else { 3057719d5645SBarry Smith (fact)->info.fill_ratio_needed = 0.0; 305810c27e3dSHong Zhang } 3059719d5645SBarry Smith (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 3060f76d2b81SHong Zhang PetscFunctionReturn(0); 3061f76d2b81SHong Zhang } 3062599ef60dSHong Zhang 30631d098868SHong Zhang #undef __FUNCT__ 30644395751fSShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_newdatastruct" 30654395751fSShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat A,Vec bb,Vec xx) 3066813a1f2bSShri Abhyankar { 3067813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3068813a1f2bSShri Abhyankar PetscErrorCode ierr; 3069813a1f2bSShri Abhyankar PetscInt n = A->rmap->n; 3070813a1f2bSShri Abhyankar const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag,*vi; 3071813a1f2bSShri Abhyankar PetscScalar *x,sum; 3072813a1f2bSShri Abhyankar const PetscScalar *b; 3073813a1f2bSShri Abhyankar const MatScalar *aa = a->a,*v; 3074813a1f2bSShri Abhyankar PetscInt i,nz; 3075813a1f2bSShri Abhyankar 3076813a1f2bSShri Abhyankar PetscFunctionBegin; 3077813a1f2bSShri Abhyankar if (!n) PetscFunctionReturn(0); 3078813a1f2bSShri Abhyankar 3079813a1f2bSShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 3080813a1f2bSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3081813a1f2bSShri Abhyankar 3082813a1f2bSShri Abhyankar /* forward solve the lower triangular */ 3083813a1f2bSShri Abhyankar x[0] = b[0]; 3084813a1f2bSShri Abhyankar v = aa; 3085813a1f2bSShri Abhyankar vi = aj; 3086813a1f2bSShri Abhyankar for (i=1; i<n; i++) { 3087813a1f2bSShri Abhyankar nz = ai[i+1] - ai[i]; 3088813a1f2bSShri Abhyankar sum = b[i]; 3089813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 3090813a1f2bSShri Abhyankar v += nz; 3091813a1f2bSShri Abhyankar vi += nz; 3092813a1f2bSShri Abhyankar x[i] = sum; 3093813a1f2bSShri Abhyankar } 3094813a1f2bSShri Abhyankar 3095813a1f2bSShri Abhyankar /* backward solve the upper triangular */ 309662fbd6afSShri Abhyankar for (i=n-1; i>=0; i--){ 30973c0119dfSShri Abhyankar v = aa + adiag[i+1] + 1; 30983c0119dfSShri Abhyankar vi = aj + adiag[i+1] + 1; 3099813a1f2bSShri Abhyankar nz = adiag[i] - adiag[i+1]-1; 3100813a1f2bSShri Abhyankar sum = x[i]; 3101813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 31023c0119dfSShri Abhyankar x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */ 3103813a1f2bSShri Abhyankar } 3104813a1f2bSShri Abhyankar 3105813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 3106813a1f2bSShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 3107813a1f2bSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3108813a1f2bSShri Abhyankar PetscFunctionReturn(0); 3109813a1f2bSShri Abhyankar } 3110813a1f2bSShri Abhyankar 31111d098868SHong Zhang #undef __FUNCT__ 31124395751fSShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_newdatastruct" 31134395751fSShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec xx) 3114f268cba8SShri Abhyankar { 3115f268cba8SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3116f268cba8SShri Abhyankar IS iscol = a->col,isrow = a->row; 3117f268cba8SShri Abhyankar PetscErrorCode ierr; 3118f268cba8SShri Abhyankar PetscInt i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz; 3119f268cba8SShri Abhyankar const PetscInt *rout,*cout,*r,*c; 3120f268cba8SShri Abhyankar PetscScalar *x,*tmp,sum; 3121f268cba8SShri Abhyankar const PetscScalar *b; 3122f268cba8SShri Abhyankar const MatScalar *aa = a->a,*v; 3123f268cba8SShri Abhyankar 3124f268cba8SShri Abhyankar PetscFunctionBegin; 3125f268cba8SShri Abhyankar if (!n) PetscFunctionReturn(0); 3126f268cba8SShri Abhyankar 3127f268cba8SShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 3128f268cba8SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3129f268cba8SShri Abhyankar tmp = a->solve_work; 3130f268cba8SShri Abhyankar 3131f268cba8SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 3132f268cba8SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 3133f268cba8SShri Abhyankar 3134f268cba8SShri Abhyankar /* forward solve the lower triangular */ 3135f268cba8SShri Abhyankar tmp[0] = b[r[0]]; 3136f268cba8SShri Abhyankar v = aa; 3137f268cba8SShri Abhyankar vi = aj; 3138f268cba8SShri Abhyankar for (i=1; i<n; i++) { 3139f268cba8SShri Abhyankar nz = ai[i+1] - ai[i]; 3140f268cba8SShri Abhyankar sum = b[r[i]]; 3141f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3142f268cba8SShri Abhyankar tmp[i] = sum; 3143f268cba8SShri Abhyankar v += nz; vi += nz; 3144f268cba8SShri Abhyankar } 3145f268cba8SShri Abhyankar 3146f268cba8SShri Abhyankar /* backward solve the upper triangular */ 3147f268cba8SShri Abhyankar for (i=n-1; i>=0; i--){ 31483c0119dfSShri Abhyankar v = aa + adiag[i+1]+1; 31493c0119dfSShri Abhyankar vi = aj + adiag[i+1]+1; 3150f268cba8SShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 3151f268cba8SShri Abhyankar sum = tmp[i]; 3152f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3153f268cba8SShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 3154f268cba8SShri Abhyankar } 3155f268cba8SShri Abhyankar 3156f268cba8SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 3157f268cba8SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 3158f268cba8SShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 3159f268cba8SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3160f268cba8SShri Abhyankar ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 3161f268cba8SShri Abhyankar PetscFunctionReturn(0); 3162f268cba8SShri Abhyankar } 3163f268cba8SShri Abhyankar 3164f268cba8SShri Abhyankar #undef __FUNCT__ 31651d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ" 31661d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact) 31671d098868SHong Zhang { 31681d098868SHong Zhang Mat B = *fact; 3169599ef60dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b; 3170599ef60dSHong Zhang IS isicol; 3171599ef60dSHong Zhang PetscErrorCode ierr; 31721d098868SHong Zhang const PetscInt *r,*ic; 31731d098868SHong Zhang PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag; 3174dc3a2fd3SHong Zhang PetscInt *bi,*bj,*bdiag,*bdiag_rev; 3175f61a948aSLisandro Dalcin PetscInt row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au; 31761d098868SHong Zhang PetscInt nlnk,*lnk; 31771d098868SHong Zhang PetscBT lnkbt; 31781d098868SHong Zhang PetscTruth row_identity,icol_identity,both_identity; 31798aee2decSHong Zhang MatScalar *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp; 31801d098868SHong Zhang const PetscInt *ics; 31811d098868SHong Zhang PetscInt j,nz,*pj,*bjtmp,k,ncut,*jtmp; 3182f61a948aSLisandro Dalcin PetscReal dt=info->dt,dtcol=info->dtcol,shift=info->shiftinblocks; 3183f61a948aSLisandro Dalcin PetscInt dtcount=(PetscInt)info->dtcount,nnz_max; 3184599ef60dSHong Zhang PetscTruth missing; 3185599ef60dSHong Zhang 3186599ef60dSHong Zhang PetscFunctionBegin; 3187f61a948aSLisandro Dalcin 3188f61a948aSLisandro Dalcin if (dt == PETSC_DEFAULT) dt = 0.005; 3189f61a948aSLisandro Dalcin if (dtcol == PETSC_DEFAULT) dtcol = 0.01; /* XXX unused! */ 3190f61a948aSLisandro Dalcin if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax); 3191f61a948aSLisandro Dalcin 31921d098868SHong Zhang /* ------- symbolic factorization, can be reused ---------*/ 31931d098868SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 31941d098868SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 31951d098868SHong Zhang adiag=a->diag; 3196599ef60dSHong Zhang 3197599ef60dSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 3198599ef60dSHong Zhang 3199599ef60dSHong Zhang /* bdiag is location of diagonal in factor */ 32001bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); /* becomes b->diag */ 32011bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag_rev);CHKERRQ(ierr); /* temporary */ 3202599ef60dSHong Zhang 32031d098868SHong Zhang /* allocate row pointers bi */ 32048fc3a347SHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 32051d098868SHong Zhang 32061d098868SHong Zhang /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */ 3207393d3a68SHong Zhang if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */ 32081d098868SHong Zhang nnz_max = ai[n]+2*n*dtcount+2; 32098fc3a347SHong Zhang 32108fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 32118fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr); 3212599ef60dSHong Zhang 3213599ef60dSHong Zhang /* put together the new matrix */ 3214599ef60dSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 3215599ef60dSHong Zhang ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 3216599ef60dSHong Zhang b = (Mat_SeqAIJ*)(B)->data; 3217599ef60dSHong Zhang b->free_a = PETSC_TRUE; 3218599ef60dSHong Zhang b->free_ij = PETSC_TRUE; 3219599ef60dSHong Zhang b->singlemalloc = PETSC_FALSE; 3220599ef60dSHong Zhang b->a = ba; 3221599ef60dSHong Zhang b->j = bj; 3222599ef60dSHong Zhang b->i = bi; 3223599ef60dSHong Zhang b->diag = bdiag; 3224599ef60dSHong Zhang b->ilen = 0; 3225599ef60dSHong Zhang b->imax = 0; 3226599ef60dSHong Zhang b->row = isrow; 3227599ef60dSHong Zhang b->col = iscol; 3228599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 3229599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 3230599ef60dSHong Zhang b->icol = isicol; 3231599ef60dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3232599ef60dSHong Zhang 32331d098868SHong Zhang ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 32341d098868SHong Zhang b->maxnz = nnz_max; 3235599ef60dSHong Zhang 3236599ef60dSHong Zhang (B)->factor = MAT_FACTOR_ILUDT; 3237599ef60dSHong Zhang (B)->info.factor_mallocs = 0; 32381d098868SHong Zhang (B)->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]); 32391d098868SHong Zhang CHKMEMQ; 32401d098868SHong Zhang /* ------- end of symbolic factorization ---------*/ 3241599ef60dSHong Zhang 3242599ef60dSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 3243599ef60dSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3244599ef60dSHong Zhang ics = ic; 3245599ef60dSHong Zhang 3246599ef60dSHong Zhang /* linked list for storing column indices of the active row */ 3247599ef60dSHong Zhang nlnk = n + 1; 3248599ef60dSHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 32491d098868SHong Zhang 32501d098868SHong Zhang /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */ 32510e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt,&im,n,PetscInt,&jtmp);CHKERRQ(ierr); 32521d098868SHong Zhang /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */ 32530e83c824SBarry Smith ierr = PetscMalloc2(n,MatScalar,&rtmp,n,MatScalar,&vtmp);CHKERRQ(ierr); 32540e83c824SBarry Smith ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr); 3255599ef60dSHong Zhang 3256599ef60dSHong Zhang bi[0] = 0; 32578fc3a347SHong Zhang bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */ 3258dc3a2fd3SHong Zhang bdiag_rev[n] = bdiag[0]; 32598fc3a347SHong Zhang bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */ 3260599ef60dSHong Zhang for (i=0; i<n; i++) { 3261599ef60dSHong Zhang /* copy initial fill into linked list */ 3262599ef60dSHong Zhang nzi = 0; /* nonzeros for active row i */ 32638369b28dSHong Zhang nzi = ai[r[i]+1] - ai[r[i]]; 32648369b28dSHong 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); 32658369b28dSHong Zhang nzi_al = adiag[r[i]] - ai[r[i]]; 32668369b28dSHong Zhang nzi_au = ai[r[i]+1] - adiag[r[i]] -1; 3267599ef60dSHong Zhang ajtmp = aj + ai[r[i]]; 32688369b28dSHong Zhang ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3269599ef60dSHong Zhang 3270599ef60dSHong Zhang /* load in initial (unfactored row) */ 32711d098868SHong Zhang aatmp = a->a + ai[r[i]]; 32728369b28dSHong Zhang for (j=0; j<nzi; j++) { 32731d098868SHong Zhang rtmp[ics[*ajtmp++]] = *aatmp++; 3274599ef60dSHong Zhang } 3275599ef60dSHong Zhang 3276599ef60dSHong Zhang /* add pivot rows into linked list */ 3277599ef60dSHong Zhang row = lnk[n]; 3278599ef60dSHong Zhang while (row < i ) { 32791d098868SHong Zhang nzi_bl = bi[row+1] - bi[row] + 1; 32801d098868SHong Zhang bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */ 32811d098868SHong Zhang ierr = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr); 3282599ef60dSHong Zhang nzi += nlnk; 3283599ef60dSHong Zhang row = lnk[row]; 3284599ef60dSHong Zhang } 3285599ef60dSHong Zhang 32868369b28dSHong Zhang /* copy data from lnk into jtmp, then initialize lnk */ 32878369b28dSHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr); 3288599ef60dSHong Zhang 3289599ef60dSHong Zhang /* numerical factorization */ 32908369b28dSHong Zhang bjtmp = jtmp; 3291599ef60dSHong Zhang row = *bjtmp++; /* 1st pivot row */ 3292599ef60dSHong Zhang while ( row < i ) { 3293599ef60dSHong Zhang pc = rtmp + row; 32943c2a7987SHong Zhang pv = ba + bdiag[row]; /* 1./(diag of the pivot row) */ 32953c2a7987SHong Zhang multiplier = (*pc) * (*pv); 3296599ef60dSHong Zhang *pc = multiplier; 32973c2a7987SHong Zhang if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */ 32981d098868SHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 32991d098868SHong Zhang pv = ba + bdiag[row+1] + 1; 33001d098868SHong Zhang /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */ 33011d098868SHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 33021d098868SHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 3303dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 3304599ef60dSHong Zhang } 3305599ef60dSHong Zhang row = *bjtmp++; 3306599ef60dSHong Zhang } 3307599ef60dSHong Zhang 33088369b28dSHong Zhang /* copy sparse rtmp into contiguous vtmp; separate L and U part */ 33098aee2decSHong Zhang diag_tmp = rtmp[i]; /* save diagonal value - may not needed?? */ 33108369b28dSHong Zhang nzi_bl = 0; j = 0; 33118369b28dSHong Zhang while (jtmp[j] < i){ /* Note: jtmp is sorted */ 33128aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 33138369b28dSHong Zhang nzi_bl++; j++; 33148369b28dSHong Zhang } 33158369b28dSHong Zhang nzi_bu = nzi - nzi_bl -1; 33168369b28dSHong Zhang while (j < nzi){ 33178aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 33188369b28dSHong Zhang j++; 33198369b28dSHong Zhang } 33208369b28dSHong Zhang 3321599ef60dSHong Zhang bjtmp = bj + bi[i]; 3322599ef60dSHong Zhang batmp = ba + bi[i]; 33238369b28dSHong Zhang /* apply level dropping rule to L part */ 33248369b28dSHong Zhang ncut = nzi_al + dtcount; 33258369b28dSHong Zhang if (ncut < nzi_bl){ 33268369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr); 33278369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr); 3328599ef60dSHong Zhang } else { 33298369b28dSHong Zhang ncut = nzi_bl; 33308369b28dSHong Zhang } 33318369b28dSHong Zhang for (j=0; j<ncut; j++){ 33328369b28dSHong Zhang bjtmp[j] = jtmp[j]; 33338369b28dSHong Zhang batmp[j] = vtmp[j]; 33346da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */ 33358369b28dSHong Zhang } 33361d098868SHong Zhang bi[i+1] = bi[i] + ncut; 33378369b28dSHong Zhang nzi = ncut + 1; 33388369b28dSHong Zhang 33398369b28dSHong Zhang /* apply level dropping rule to U part */ 33408369b28dSHong Zhang ncut = nzi_au + dtcount; 33418369b28dSHong Zhang if (ncut < nzi_bu){ 33428369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr); 33438369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr); 33448369b28dSHong Zhang } else { 33458369b28dSHong Zhang ncut = nzi_bu; 33468369b28dSHong Zhang } 33478369b28dSHong Zhang nzi += ncut; 33481d098868SHong Zhang 33491d098868SHong Zhang /* mark bdiagonal */ 33501d098868SHong Zhang bdiag[i+1] = bdiag[i] - (ncut + 1); 3351dc3a2fd3SHong Zhang bdiag_rev[n-i-1] = bdiag[i+1]; 33528fc3a347SHong Zhang bi[2*n - i] = bi[2*n - i +1] - (ncut + 1); 33531d098868SHong Zhang bjtmp = bj + bdiag[i]; 33541d098868SHong Zhang batmp = ba + bdiag[i]; 33551d098868SHong Zhang *bjtmp = i; 33568aee2decSHong Zhang *batmp = diag_tmp; /* rtmp[i]; */ 3357c9c72f8fSHong Zhang if (*batmp == 0.0) { 3358c9c72f8fSHong Zhang *batmp = dt+shift; 3359bd1bc851SBarry Smith /* printf(" row %d add shift %g\n",i,shift); */ 3360c9c72f8fSHong Zhang } 3361b78a477dSHong Zhang *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */ 33626da515a1SHong Zhang /* printf(" (%d,%g),",*bjtmp,*batmp); */ 33631d098868SHong Zhang 33641d098868SHong Zhang bjtmp = bj + bdiag[i+1]+1; 33651d098868SHong Zhang batmp = ba + bdiag[i+1]+1; 33668369b28dSHong Zhang for (k=0; k<ncut; k++){ 33671d098868SHong Zhang bjtmp[k] = jtmp[nzi_bl+1+k]; 33681d098868SHong Zhang batmp[k] = vtmp[nzi_bl+1+k]; 33696da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */ 3370599ef60dSHong Zhang } 33716da515a1SHong Zhang /* printf("\n"); */ 3372599ef60dSHong Zhang 3373599ef60dSHong Zhang im[i] = nzi; /* used by PetscLLAddSortedLU() */ 3374599ef60dSHong Zhang /* 33751d098868SHong Zhang printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]); 33768369b28dSHong Zhang printf(" ----------------------------\n"); 3377599ef60dSHong Zhang */ 33788369b28dSHong Zhang } /* for (i=0; i<n; i++) */ 33791d098868SHong Zhang /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */ 33801d098868SHong 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]); 3381599ef60dSHong Zhang 3382599ef60dSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3383599ef60dSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3384599ef60dSHong Zhang 3385599ef60dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 33860e83c824SBarry Smith ierr = PetscFree2(im,jtmp);CHKERRQ(ierr); 33870e83c824SBarry Smith ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr); 33881bfa06eaSJed Brown ierr = PetscFree(bdiag_rev);CHKERRQ(ierr); 3389599ef60dSHong Zhang 3390599ef60dSHong Zhang ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr); 33911d098868SHong Zhang b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n]; 3392599ef60dSHong Zhang 3393599ef60dSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 3394599ef60dSHong Zhang ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr); 3395599ef60dSHong Zhang both_identity = (PetscTruth) (row_identity && icol_identity); 3396599ef60dSHong Zhang if (row_identity && icol_identity) { 3397894a2336SHong Zhang B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct; 3398599ef60dSHong Zhang } else { 3399894a2336SHong Zhang B->ops->solve = MatSolve_SeqAIJ_newdatastruct; 3400599ef60dSHong Zhang } 3401599ef60dSHong Zhang 3402b78a477dSHong Zhang B->ops->lufactorsymbolic = MatILUDTFactorSymbolic_SeqAIJ; 3403b78a477dSHong Zhang B->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ; 34041d098868SHong Zhang B->ops->solveadd = 0; 34051d098868SHong Zhang B->ops->solvetranspose = 0; 34061d098868SHong Zhang B->ops->solvetransposeadd = 0; 34071d098868SHong Zhang B->ops->matsolve = 0; 3408599ef60dSHong Zhang B->assembled = PETSC_TRUE; 3409599ef60dSHong Zhang B->preallocated = PETSC_TRUE; 3410599ef60dSHong Zhang PetscFunctionReturn(0); 3411599ef60dSHong Zhang } 3412599ef60dSHong Zhang 34133c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */ 34143c2a7987SHong Zhang #undef __FUNCT__ 34153c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ" 34163c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info) 34173c2a7987SHong Zhang { 34183c2a7987SHong Zhang PetscErrorCode ierr; 34193c2a7987SHong Zhang 34203c2a7987SHong Zhang PetscFunctionBegin; 34213c2a7987SHong Zhang ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr); 34223c2a7987SHong Zhang 34233c2a7987SHong Zhang fact->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ; 34243c2a7987SHong Zhang PetscFunctionReturn(0); 34253c2a7987SHong Zhang } 34263c2a7987SHong Zhang 34273c2a7987SHong Zhang /* 34283c2a7987SHong Zhang same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors 34293c2a7987SHong Zhang - intend to replace existing MatLUFactorNumeric_SeqAIJ() 34303c2a7987SHong Zhang */ 34313c2a7987SHong Zhang #undef __FUNCT__ 34323c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ" 34333c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info) 34343c2a7987SHong Zhang { 34353c2a7987SHong Zhang Mat C=fact; 34363c2a7987SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 34373c2a7987SHong Zhang IS isrow = b->row,isicol = b->icol; 34383c2a7987SHong Zhang PetscErrorCode ierr; 34393c2a7987SHong Zhang const PetscInt *r,*ic,*ics; 3440b78a477dSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 3441b78a477dSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj; 34423c2a7987SHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 3443b78a477dSHong Zhang PetscReal dt=info->dt,shift=info->shiftinblocks; 3444b78a477dSHong Zhang PetscTruth row_identity, col_identity; 34453c2a7987SHong Zhang 34463c2a7987SHong Zhang PetscFunctionBegin; 34473c2a7987SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 34483c2a7987SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3449b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 34503c2a7987SHong Zhang ics = ic; 34513c2a7987SHong Zhang 34523c2a7987SHong Zhang for (i=0; i<n; i++){ 3453b78a477dSHong Zhang /* initialize rtmp array */ 3454b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of nozeros in L(i,:) */ 34553c2a7987SHong Zhang bjtmp = bj + bi[i]; 3456b78a477dSHong Zhang for (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0; 3457b78a477dSHong Zhang rtmp[i] = 0.0; 3458b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */ 3459b78a477dSHong Zhang bjtmp = bj + bdiag[i+1] + 1; 3460b78a477dSHong Zhang for (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0; 34613c2a7987SHong Zhang 3462b78a477dSHong Zhang /* load in initial unfactored row of A */ 34636da515a1SHong Zhang /* printf("row %d\n",i); */ 34643c2a7987SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 34653c2a7987SHong Zhang ajtmp = aj + ai[r[i]]; 34663c2a7987SHong Zhang v = aa + ai[r[i]]; 34673c2a7987SHong Zhang for (j=0; j<nz; j++) { 3468b78a477dSHong Zhang rtmp[ics[*ajtmp++]] = v[j]; 34696da515a1SHong Zhang /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */ 34703c2a7987SHong Zhang } 34716da515a1SHong Zhang /* printf("\n"); */ 34723c2a7987SHong Zhang 3473b78a477dSHong Zhang /* numerical factorization */ 3474b78a477dSHong Zhang bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */ 3475b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of entries in L(i,:) */ 3476b78a477dSHong Zhang k = 0; 3477b78a477dSHong Zhang while (k < nzl){ 34783c2a7987SHong Zhang row = *bjtmp++; 34796da515a1SHong Zhang /* printf(" prow %d\n",row); */ 34803c2a7987SHong Zhang pc = rtmp + row; 3481b78a477dSHong Zhang pv = b->a + bdiag[row]; /* 1./(diag of the pivot row) */ 3482b78a477dSHong Zhang multiplier = (*pc) * (*pv); 34833c2a7987SHong Zhang *pc = multiplier; 3484b78a477dSHong Zhang if (PetscAbsScalar(multiplier) > dt){ 3485b78a477dSHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 3486b78a477dSHong Zhang pv = b->a + bdiag[row+1] + 1; 3487b78a477dSHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 3488b78a477dSHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 34896da515a1SHong Zhang /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */ 34903c2a7987SHong Zhang } 3491b78a477dSHong Zhang k++; 34923c2a7987SHong Zhang } 34933c2a7987SHong Zhang 3494b78a477dSHong Zhang /* finished row so stick it into b->a */ 3495b78a477dSHong Zhang /* L-part */ 3496b78a477dSHong Zhang pv = b->a + bi[i] ; 3497b78a477dSHong Zhang pj = bj + bi[i] ; 3498b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; 3499b78a477dSHong Zhang for (j=0; j<nzl; j++) { 3500b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 35016da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 35023c2a7987SHong Zhang } 3503b78a477dSHong Zhang 3504b78a477dSHong Zhang /* diagonal: invert diagonal entries for simplier triangular solves */ 3505b78a477dSHong Zhang if (rtmp[i] == 0.0) rtmp[i] = dt+shift; 3506b78a477dSHong Zhang b->a[bdiag[i]] = 1.0/rtmp[i]; 35076da515a1SHong Zhang /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */ 3508b78a477dSHong Zhang 3509b78a477dSHong Zhang /* U-part */ 3510b78a477dSHong Zhang pv = b->a + bdiag[i+1] + 1; 3511b78a477dSHong Zhang pj = bj + bdiag[i+1] + 1; 3512b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1] - 1; 3513b78a477dSHong Zhang for (j=0; j<nzu; j++) { 3514b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 35156da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 3516b78a477dSHong Zhang } 35176da515a1SHong Zhang /* printf("\n"); */ 3518b78a477dSHong Zhang } 3519b78a477dSHong Zhang 35203c2a7987SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 35213c2a7987SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 35223c2a7987SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3523b78a477dSHong Zhang 35243c2a7987SHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 35253c2a7987SHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 35263c2a7987SHong Zhang if (row_identity && col_identity) { 3527894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct; 35283c2a7987SHong Zhang } else { 3529894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_newdatastruct; 35303c2a7987SHong Zhang } 3531b78a477dSHong Zhang C->ops->solveadd = 0; 3532b78a477dSHong Zhang C->ops->solvetranspose = 0; 3533b78a477dSHong Zhang C->ops->solvetransposeadd = 0; 3534b78a477dSHong Zhang C->ops->matsolve = 0; 35353c2a7987SHong Zhang C->assembled = PETSC_TRUE; 35363c2a7987SHong Zhang C->preallocated = PETSC_TRUE; 35373c2a7987SHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 35383c2a7987SHong Zhang PetscFunctionReturn(0); 35393c2a7987SHong Zhang } 3540