1be1d678aSKris Buschelman #define PETSCMAT_DLL 2289bc588SBarry Smith 3b377110cSBarry Smith 47c4f633dSBarry Smith #include "../src/mat/impls/aij/seq/aij.h" 5783ef271SHong Zhang #include "../src/mat/impls/sbaij/seq/sbaij.h" 61393bc97SHong Zhang #include "petscbt.h" 77c4f633dSBarry Smith #include "../src/mat/utils/freespace.h" 83b2fbd54SBarry Smith 9f6a26d55SMatthew Knepley EXTERN_C_BEGIN 104a2ae208SSatish Balay #undef __FUNCT__ 114a2ae208SSatish Balay #define __FUNCT__ "MatOrdering_Flow_SeqAIJ" 128fa24674SBarry Smith /* 138fa24674SBarry Smith Computes an ordering to get most of the large numerical values in the lower triangular part of the matrix 148fa24674SBarry Smith */ 15dfbe8321SBarry Smith PetscErrorCode MatOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol) 163b2fbd54SBarry Smith { 178fa24674SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)mat->data; 188fa24674SBarry Smith PetscErrorCode ierr; 19889a8dedSBarry Smith PetscInt i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order; 208fa24674SBarry Smith const PetscInt *ai = a->i, *aj = a->j; 218fa24674SBarry Smith const PetscScalar *aa = a->a; 228fa24674SBarry Smith PetscTruth *done; 23889a8dedSBarry Smith PetscReal best,past = 0,future; 243a40ed3dSBarry Smith 258fa24674SBarry Smith PetscFunctionBegin; 268fa24674SBarry Smith /* pick initial row */ 278fa24674SBarry Smith best = -1; 288fa24674SBarry Smith for (i=0; i<n; i++) { 298fa24674SBarry Smith future = 0; 308fa24674SBarry Smith for (j=ai[i]; j<ai[i+1]; j++) { 318fa24674SBarry Smith if (aj[j] != i) future += PetscAbsScalar(aa[j]); else past = PetscAbsScalar(aa[j]); 328fa24674SBarry Smith } 338fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 348fa24674SBarry Smith if (past/future > best) { 358fa24674SBarry Smith best = past/future; 368fa24674SBarry Smith current = i; 378fa24674SBarry Smith } 388fa24674SBarry Smith } 398fa24674SBarry Smith 408fa24674SBarry Smith ierr = PetscMalloc(n*sizeof(PetscTruth),&done);CHKERRQ(ierr); 418fa24674SBarry Smith ierr = 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; 688fa24674SBarry Smith future = 0; 698fa24674SBarry Smith past = 0; 708fa24674SBarry Smith for (j=ai[k]; j<ai[k+1]; j++) { 718fa24674SBarry Smith kk = aj[j]; 728fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[j]); 738fa24674SBarry Smith else if (kk != k) future += PetscAbsScalar(aa[j]); 748fa24674SBarry Smith } 758fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 768fa24674SBarry Smith if (past/future > best) { 778fa24674SBarry Smith best = past/future; 788fa24674SBarry Smith newcurrent = k; 798fa24674SBarry Smith } 808fa24674SBarry Smith } 818fa24674SBarry Smith } 828fa24674SBarry Smith if (current == newcurrent) SETERRQ(PETSC_ERR_PLIB,"newcurrent cannot be current"); 838fa24674SBarry Smith current = newcurrent; 848fa24674SBarry Smith order[i+1] = current; 858fa24674SBarry Smith } 868fa24674SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,order,irow);CHKERRQ(ierr); 878fa24674SBarry Smith *icol = *irow; 888fa24674SBarry Smith ierr = PetscObjectReference((PetscObject)*irow);CHKERRQ(ierr); 898fa24674SBarry Smith ierr = PetscFree(done);CHKERRQ(ierr); 908fa24674SBarry Smith ierr = PetscFree(order);CHKERRQ(ierr); 91d64ed03dSBarry Smith PetscFunctionReturn(0); 923b2fbd54SBarry Smith } 93f6a26d55SMatthew Knepley EXTERN_C_END 943b2fbd54SBarry Smith 95e631078cSBarry Smith EXTERN_C_BEGIN 964a2ae208SSatish Balay #undef __FUNCT__ 97db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc" 98db4efbfdSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscTruth *flg) 99db4efbfdSBarry Smith { 100db4efbfdSBarry Smith PetscFunctionBegin; 101db4efbfdSBarry Smith *flg = PETSC_TRUE; 102db4efbfdSBarry Smith PetscFunctionReturn(0); 103db4efbfdSBarry Smith } 104db4efbfdSBarry Smith EXTERN_C_END 105db4efbfdSBarry Smith 106db4efbfdSBarry Smith EXTERN_C_BEGIN 107db4efbfdSBarry Smith #undef __FUNCT__ 108b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc" 109b24902e0SBarry Smith PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B) 110b24902e0SBarry Smith { 111d0f46423SBarry Smith PetscInt n = A->rmap->n; 112b24902e0SBarry Smith PetscErrorCode ierr; 113b24902e0SBarry Smith 114b24902e0SBarry Smith PetscFunctionBegin; 115b24902e0SBarry Smith ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr); 116b24902e0SBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 117599ef60dSHong Zhang if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT){ 118b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr); 119db4efbfdSBarry Smith (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ; 12035bd34faSBarry Smith (*B)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqAIJ; 1211da05e5aSHong Zhang (*B)->ops->iludtfactor = MatILUDTFactor_SeqAIJ; 122b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 123b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 124b24902e0SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1255c9eb25fSBarry Smith (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqAIJ; 1265c9eb25fSBarry Smith (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ; 127b24902e0SBarry Smith } else SETERRQ(PETSC_ERR_SUP,"Factor type not supported"); 128db4efbfdSBarry Smith (*B)->factor = ftype; 129b24902e0SBarry Smith PetscFunctionReturn(0); 130b24902e0SBarry Smith } 131e631078cSBarry Smith EXTERN_C_END 132b24902e0SBarry Smith 133b24902e0SBarry Smith #undef __FUNCT__ 134b9617806SBarry Smith #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ" 1350481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 136289bc588SBarry Smith { 137416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 138289bc588SBarry Smith IS isicol; 1396849ba73SBarry Smith PetscErrorCode ierr; 1405d0c19d7SBarry Smith const PetscInt *r,*ic; 1415d0c19d7SBarry Smith PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 1421393bc97SHong Zhang PetscInt *bi,*bj,*ajtmp; 1431393bc97SHong Zhang PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 1449e046878SBarry Smith PetscReal f; 1454875ba38SHong Zhang PetscInt nlnk,*lnk,k,**bi_ptr; 146a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1471393bc97SHong Zhang PetscBT lnkbt; 148ce3d78c0SShri Abhyankar PetscTruth newdatastruct= PETSC_FALSE; 149289bc588SBarry Smith 1503a40ed3dSBarry Smith PetscFunctionBegin; 151ce3d78c0SShri Abhyankar ierr = PetscOptionsGetTruth(PETSC_NULL,"-lu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 152ce3d78c0SShri Abhyankar if(newdatastruct){ 153ce3d78c0SShri Abhyankar ierr = MatLUFactorSymbolic_SeqAIJ_newdatastruct(B,A,isrow,iscol,info);CHKERRQ(ierr); 154ce3d78c0SShri Abhyankar PetscFunctionReturn(0); 155ce3d78c0SShri Abhyankar } 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 311*a6df321fSShri Abhyankar /* get new row and diagonal pointers */ 312*a6df321fSShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt,&bi,n+1,PetscInt,&bdiag);CHKERRQ(ierr); 313*a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 3149b48462bSShri Abhyankar 3159b48462bSShri Abhyankar /* linked list for storing column indices of the active row */ 3169b48462bSShri Abhyankar nlnk = n + 1; 3179b48462bSShri Abhyankar ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3189b48462bSShri Abhyankar 3199b48462bSShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 3209b48462bSShri Abhyankar 3219b48462bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 3229b48462bSShri Abhyankar f = info->fill; 3239b48462bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 3249b48462bSShri Abhyankar current_space = free_space; 3259b48462bSShri Abhyankar 3269b48462bSShri Abhyankar for (i=0; i<n; i++) { 3279b48462bSShri Abhyankar /* copy previous fill into linked list */ 3289b48462bSShri Abhyankar nzi = 0; 3299b48462bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 3309b48462bSShri 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); 3319b48462bSShri Abhyankar ajtmp = aj + ai[r[i]]; 3329b48462bSShri Abhyankar ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3339b48462bSShri Abhyankar nzi += nlnk; 3349b48462bSShri Abhyankar 3359b48462bSShri Abhyankar /* add pivot rows into linked list */ 3369b48462bSShri Abhyankar row = lnk[n]; 3379b48462bSShri Abhyankar while (row < i){ 3389b48462bSShri Abhyankar nzbd = bdiag[row] + 1; /* num of entries in the row with column index <= row */ 3399b48462bSShri Abhyankar ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 3409b48462bSShri Abhyankar ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 3419b48462bSShri Abhyankar nzi += nlnk; 3429b48462bSShri Abhyankar row = lnk[row]; 3439b48462bSShri Abhyankar } 3449b48462bSShri Abhyankar bi[i+1] = bi[i] + nzi; 3459b48462bSShri Abhyankar im[i] = nzi; 3469b48462bSShri Abhyankar 3479b48462bSShri Abhyankar /* mark bdiag */ 3489b48462bSShri Abhyankar nzbd = 0; 3499b48462bSShri Abhyankar nnz = nzi; 3509b48462bSShri Abhyankar k = lnk[n]; 3519b48462bSShri Abhyankar while (nnz-- && k < i){ 3529b48462bSShri Abhyankar nzbd++; 3539b48462bSShri Abhyankar k = lnk[k]; 3549b48462bSShri Abhyankar } 3559b48462bSShri Abhyankar bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */ 3569b48462bSShri Abhyankar 3579b48462bSShri Abhyankar /* if free space is not available, make more free space */ 3589b48462bSShri Abhyankar if (current_space->local_remaining<nzi) { 3599b48462bSShri Abhyankar nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */ 3609b48462bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 3619b48462bSShri Abhyankar reallocs++; 3629b48462bSShri Abhyankar } 3639b48462bSShri Abhyankar 3649b48462bSShri Abhyankar /* copy data into free space, then initialize lnk */ 3659b48462bSShri Abhyankar ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 3669b48462bSShri Abhyankar bi_ptr[i] = current_space->array; 3679b48462bSShri Abhyankar current_space->array += nzi; 3689b48462bSShri Abhyankar current_space->local_used += nzi; 3699b48462bSShri Abhyankar current_space->local_remaining -= nzi; 3709b48462bSShri Abhyankar } 3719b48462bSShri Abhyankar #if defined(PETSC_USE_INFO) 3729b48462bSShri Abhyankar if (ai[n] != 0) { 3739b48462bSShri Abhyankar PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 3749b48462bSShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 3759b48462bSShri Abhyankar ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 3769b48462bSShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 3779b48462bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 3789b48462bSShri Abhyankar } else { 3799b48462bSShri Abhyankar ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 3809b48462bSShri Abhyankar } 3819b48462bSShri Abhyankar #endif 3829b48462bSShri Abhyankar 3839b48462bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3849b48462bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3859b48462bSShri Abhyankar 3869b48462bSShri Abhyankar /* destroy list of free space and other temporary array(s) */ 3879b48462bSShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 3889b48462bSShri Abhyankar ierr = PetscFreeSpaceContiguous_LU_v2(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 3899b48462bSShri Abhyankar ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 3909b48462bSShri Abhyankar ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 3919b48462bSShri Abhyankar 3929b48462bSShri Abhyankar /* put together the new matrix */ 3939b48462bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 3949b48462bSShri Abhyankar ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 3959b48462bSShri Abhyankar b = (Mat_SeqAIJ*)(B)->data; 3969b48462bSShri Abhyankar b->free_a = PETSC_TRUE; 3979b48462bSShri Abhyankar b->free_ij = PETSC_TRUE; 3989b48462bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 3999b48462bSShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 4009b48462bSShri Abhyankar b->j = bj; 4019b48462bSShri Abhyankar b->i = bi; 4029b48462bSShri Abhyankar b->diag = bdiag; 4039b48462bSShri Abhyankar b->ilen = 0; 4049b48462bSShri Abhyankar b->imax = 0; 4059b48462bSShri Abhyankar b->row = isrow; 4069b48462bSShri Abhyankar b->col = iscol; 4079b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 4089b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 4099b48462bSShri Abhyankar b->icol = isicol; 4109b48462bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 4119b48462bSShri Abhyankar 4129b48462bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 4139b48462bSShri Abhyankar ierr = PetscLogObjectMemory(B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 4149b48462bSShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 4159b48462bSShri Abhyankar (B)->factor = MAT_FACTOR_LU; 4169b48462bSShri Abhyankar (B)->info.factor_mallocs = reallocs; 4179b48462bSShri Abhyankar (B)->info.fill_ratio_given = f; 4189b48462bSShri Abhyankar 4199b48462bSShri Abhyankar if (ai[n]) { 4209b48462bSShri Abhyankar (B)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 4219b48462bSShri Abhyankar } else { 4229b48462bSShri Abhyankar (B)->info.fill_ratio_needed = 0.0; 4239b48462bSShri Abhyankar } 424b4f5729fSShri Abhyankar (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 4259b48462bSShri Abhyankar PetscFunctionReturn(0); 4269b48462bSShri Abhyankar } 4279b48462bSShri Abhyankar 4285b5bc046SBarry Smith /* 4295b5bc046SBarry Smith Trouble in factorization, should we dump the original matrix? 4305b5bc046SBarry Smith */ 4315b5bc046SBarry Smith #undef __FUNCT__ 4325b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix" 4335b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A) 4345b5bc046SBarry Smith { 4355b5bc046SBarry Smith PetscErrorCode ierr; 43690d69ab7SBarry Smith PetscTruth flg = PETSC_FALSE; 4375b5bc046SBarry Smith 4385b5bc046SBarry Smith PetscFunctionBegin; 43990d69ab7SBarry Smith ierr = PetscOptionsGetTruth(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr); 4405b5bc046SBarry Smith if (flg) { 4415b5bc046SBarry Smith PetscViewer viewer; 4425b5bc046SBarry Smith char filename[PETSC_MAX_PATH_LEN]; 4435b5bc046SBarry Smith 4445b5bc046SBarry Smith ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr); 4457adad957SLisandro Dalcin ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 4465b5bc046SBarry Smith ierr = MatView(A,viewer);CHKERRQ(ierr); 4475b5bc046SBarry Smith ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr); 4485b5bc046SBarry Smith } 4495b5bc046SBarry Smith PetscFunctionReturn(0); 4505b5bc046SBarry Smith } 4515b5bc046SBarry Smith 4524108e4d5SBarry Smith extern PetscErrorCode MatSolve_SeqAIJ_Inode(Mat,Vec,Vec); 453db4efbfdSBarry Smith 454c9c72f8fSHong Zhang #undef __FUNCT__ 455c9c72f8fSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_newdatastruct" 456c9c72f8fSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info) 457c9c72f8fSHong Zhang { 458c9c72f8fSHong Zhang Mat C=B; 459c9c72f8fSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 460c9c72f8fSHong Zhang IS isrow = b->row,isicol = b->icol; 461c9c72f8fSHong Zhang PetscErrorCode ierr; 462c9c72f8fSHong Zhang const PetscInt *r,*ic,*ics; 463c9c72f8fSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 464c9c72f8fSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzL,row,*bdiag=b->diag,*pj; 465c9c72f8fSHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 466c9c72f8fSHong Zhang PetscTruth row_identity,col_identity; 467c9c72f8fSHong Zhang 468d4ad06f7SHong Zhang LUShift_Ctx sctx; 469d4ad06f7SHong Zhang PetscInt *ddiag,newshift; 470d4ad06f7SHong Zhang PetscReal rs; 471d4ad06f7SHong Zhang MatScalar d; 472d4ad06f7SHong Zhang 473c9c72f8fSHong Zhang PetscFunctionBegin; 474d6e5152cSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 475d6e5152cSHong Zhang ierr = PetscMemzero(&sctx,sizeof(LUShift_Ctx));CHKERRQ(ierr); 476d4ad06f7SHong Zhang 477d4ad06f7SHong Zhang /* if both shift schemes are chosen by user, only use info->shiftpd */ 478d4ad06f7SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 479d4ad06f7SHong Zhang ddiag = a->diag; 480d4ad06f7SHong Zhang sctx.shift_top = info->zeropivot; 481d4ad06f7SHong Zhang for (i=0; i<n; i++) { 482d4ad06f7SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 483d4ad06f7SHong Zhang d = (aa)[ddiag[i]]; 484d4ad06f7SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 485d4ad06f7SHong Zhang v = aa+ai[i]; 486d4ad06f7SHong Zhang nz = ai[i+1] - ai[i]; 487d4ad06f7SHong Zhang for (j=0; j<nz; j++) 488d4ad06f7SHong Zhang rs += PetscAbsScalar(v[j]); 489d4ad06f7SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 490d4ad06f7SHong Zhang } 491d4ad06f7SHong Zhang sctx.shift_top *= 1.1; 492d4ad06f7SHong Zhang sctx.nshift_max = 5; 493d4ad06f7SHong Zhang sctx.shift_lo = 0.; 494d4ad06f7SHong Zhang sctx.shift_hi = 1.; 495d4ad06f7SHong Zhang } 496d4ad06f7SHong Zhang 497c9c72f8fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 498c9c72f8fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 499c9c72f8fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 500c9c72f8fSHong Zhang ics = ic; 501c9c72f8fSHong Zhang 502d4ad06f7SHong Zhang do { 503d4ad06f7SHong Zhang sctx.lushift = PETSC_FALSE; 504c9c72f8fSHong Zhang for (i=0; i<n; i++){ 505c9c72f8fSHong Zhang /* zero rtmp */ 506c9c72f8fSHong Zhang /* L part */ 507c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 508c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 509c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 510c9c72f8fSHong Zhang 511c9c72f8fSHong Zhang /* U part */ 51262fbd6afSShri Abhyankar nz = bdiag[i]-bdiag[i+1]; 51362fbd6afSShri Abhyankar bjtmp = bj + bdiag[i+1]+1; 514813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 515813a1f2bSShri Abhyankar 516813a1f2bSShri Abhyankar /* load in initial (unfactored row) */ 517813a1f2bSShri Abhyankar nz = ai[r[i]+1] - ai[r[i]]; 518813a1f2bSShri Abhyankar ajtmp = aj + ai[r[i]]; 519813a1f2bSShri Abhyankar v = aa + ai[r[i]]; 520813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 521813a1f2bSShri Abhyankar rtmp[ics[ajtmp[j]]] = v[j]; 522813a1f2bSShri Abhyankar } 523813a1f2bSShri Abhyankar /* ZeropivotApply() */ 524813a1f2bSShri Abhyankar rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 525813a1f2bSShri Abhyankar 526813a1f2bSShri Abhyankar /* elimination */ 527813a1f2bSShri Abhyankar bjtmp = bj + bi[i]; 528813a1f2bSShri Abhyankar row = *bjtmp++; 529813a1f2bSShri Abhyankar nzL = bi[i+1] - bi[i]; 530f268cba8SShri Abhyankar for(k=0; k < nzL;k++) { 531813a1f2bSShri Abhyankar pc = rtmp + row; 532813a1f2bSShri Abhyankar if (*pc != 0.0) { 533813a1f2bSShri Abhyankar pv = b->a + bdiag[row]; 534813a1f2bSShri Abhyankar multiplier = *pc * (*pv); 535813a1f2bSShri Abhyankar *pc = multiplier; 53662fbd6afSShri Abhyankar pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 53762fbd6afSShri Abhyankar pv = b->a + bdiag[row+1]+1; 53862fbd6afSShri Abhyankar nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 539813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 540813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 541813a1f2bSShri Abhyankar } 542f268cba8SShri Abhyankar row = *bjtmp++; 543813a1f2bSShri Abhyankar } 544813a1f2bSShri Abhyankar 545813a1f2bSShri Abhyankar /* finished row so stick it into b->a */ 546813a1f2bSShri Abhyankar rs = 0.0; 547813a1f2bSShri Abhyankar /* L part */ 548813a1f2bSShri Abhyankar pv = b->a + bi[i] ; 549813a1f2bSShri Abhyankar pj = b->j + bi[i] ; 550813a1f2bSShri Abhyankar nz = bi[i+1] - bi[i]; 551813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 552813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 553813a1f2bSShri Abhyankar } 554813a1f2bSShri Abhyankar 555813a1f2bSShri Abhyankar /* U part */ 55662fbd6afSShri Abhyankar pv = b->a + bdiag[i+1]+1; 55762fbd6afSShri Abhyankar pj = b->j + bdiag[i+1]+1; 55862fbd6afSShri Abhyankar nz = bdiag[i] - bdiag[i+1]-1; 559813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 560813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 561813a1f2bSShri Abhyankar } 562813a1f2bSShri Abhyankar 5638ff23777SHong Zhang /* MatPivotCheck() */ 564813a1f2bSShri Abhyankar sctx.rs = rs; 565813a1f2bSShri Abhyankar sctx.pv = rtmp[i]; 5668ff23777SHong Zhang /* ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); */ 5678ff23777SHong Zhang if (info->shiftnz){ 5688ff23777SHong Zhang ierr = MatPivotCheck_nz(info,sctx,i,newshift);CHKERRQ(ierr); 5698ff23777SHong Zhang } else if (info->shiftpd){ 5708ff23777SHong Zhang ierr = MatPivotCheck_pd(info,sctx,i,newshift);CHKERRQ(ierr); 5718ff23777SHong Zhang } else if (info->shiftinblocks){ 5728ff23777SHong Zhang ierr = MatPivotCheck_inblocks(info,sctx,i,newshift);CHKERRQ(ierr); 5738ff23777SHong Zhang } else { 5748ff23777SHong Zhang ierr = MatPivotCheck_none(info,sctx,i,newshift);CHKERRQ(ierr); 5758ff23777SHong Zhang } 5768ff23777SHong Zhang rtmp[i] = sctx.pv; 577813a1f2bSShri Abhyankar if (newshift == 1) break; 578813a1f2bSShri Abhyankar 579813a1f2bSShri Abhyankar /* Mark diagonal and invert diagonal for simplier triangular solves */ 580813a1f2bSShri Abhyankar pv = b->a + bdiag[i]; 581813a1f2bSShri Abhyankar *pv = 1.0/rtmp[i]; 582813a1f2bSShri Abhyankar 583813a1f2bSShri Abhyankar } /* endof for (i=0; i<n; i++){ */ 584813a1f2bSShri Abhyankar 5858ff23777SHong Zhang /* MatPivotRefine() */ 586813a1f2bSShri Abhyankar if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){ 587813a1f2bSShri Abhyankar /* 588813a1f2bSShri Abhyankar * if no shift in this attempt & shifting & started shifting & can refine, 589813a1f2bSShri Abhyankar * then try lower shift 590813a1f2bSShri Abhyankar */ 591813a1f2bSShri Abhyankar sctx.shift_hi = sctx.shift_fraction; 592813a1f2bSShri Abhyankar sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 593813a1f2bSShri Abhyankar sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 594813a1f2bSShri Abhyankar sctx.lushift = PETSC_TRUE; 595813a1f2bSShri Abhyankar sctx.nshift++; 596813a1f2bSShri Abhyankar } 597813a1f2bSShri Abhyankar } while (sctx.lushift); 598813a1f2bSShri Abhyankar 599813a1f2bSShri Abhyankar ierr = PetscFree(rtmp);CHKERRQ(ierr); 600813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 601813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 602813a1f2bSShri Abhyankar 603813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 604813a1f2bSShri Abhyankar ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 605813a1f2bSShri Abhyankar if (row_identity && col_identity) { 60662fbd6afSShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct_v2; 607813a1f2bSShri Abhyankar } else { 608f268cba8SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_newdatastruct_v2; 609813a1f2bSShri Abhyankar } 610813a1f2bSShri Abhyankar 611813a1f2bSShri Abhyankar C->ops->solveadd = 0; 612813a1f2bSShri Abhyankar C->ops->solvetranspose = 0; 613813a1f2bSShri Abhyankar C->ops->solvetransposeadd = 0; 614813a1f2bSShri Abhyankar C->ops->matsolve = 0; 615813a1f2bSShri Abhyankar C->assembled = PETSC_TRUE; 616813a1f2bSShri Abhyankar C->preallocated = PETSC_TRUE; 617813a1f2bSShri Abhyankar ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 618813a1f2bSShri Abhyankar 6198ff23777SHong Zhang /* MatPivotView() */ 620813a1f2bSShri Abhyankar if (sctx.nshift){ 621813a1f2bSShri Abhyankar if (info->shiftpd) { 622813a1f2bSShri 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); 623813a1f2bSShri Abhyankar } else if (info->shiftnz) { 624813a1f2bSShri Abhyankar ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 6258ff23777SHong Zhang } else if (info->shiftinblocks){ 6268ff23777SHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftinblocks);CHKERRQ(ierr); 627813a1f2bSShri Abhyankar } 628813a1f2bSShri Abhyankar } 629813a1f2bSShri Abhyankar PetscFunctionReturn(0); 630813a1f2bSShri Abhyankar } 631813a1f2bSShri Abhyankar 6324a2ae208SSatish Balay #undef __FUNCT__ 6334a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ" 6340481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 635289bc588SBarry Smith { 636719d5645SBarry Smith Mat C=B; 637416022c9SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 63882bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 6396849ba73SBarry Smith PetscErrorCode ierr; 6405d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 641d278edc7SBarry Smith PetscInt nz,row,i,j,n=A->rmap->n,diag; 642d278edc7SBarry Smith const PetscInt *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 643d278edc7SBarry Smith const PetscInt *ajtmp,*bjtmp,*diag_offset = b->diag,*pj; 644d278edc7SBarry Smith MatScalar *pv,*rtmp,*pc,multiplier,d; 645d278edc7SBarry Smith const MatScalar *v,*aa=a->a; 64633f9893dSHong Zhang PetscReal rs=0.0; 647b3bf805bSHong Zhang LUShift_Ctx sctx; 64842898933SHong Zhang PetscInt newshift,*ddiag; 649289bc588SBarry Smith 6503a40ed3dSBarry Smith PetscFunctionBegin; 65178b31e54SBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 65278b31e54SBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 653b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 6549f7d0b68SHong Zhang ics = ic; 655289bc588SBarry Smith 656d4ad06f7SHong Zhang /* initialize shift context sctx */ 657d4ad06f7SHong Zhang sctx.nshift = 0; 658ada7143aSSatish Balay sctx.nshift_max = 0; 659d4ad06f7SHong Zhang sctx.shift_top = 0.0; 660d4ad06f7SHong Zhang sctx.shift_lo = 0.0; 661d4ad06f7SHong Zhang sctx.shift_hi = 0.0; 662d4ad06f7SHong Zhang sctx.shift_fraction = 0.0; 663d4ad06f7SHong Zhang sctx.shift_amount = 0.0; 664ada7143aSSatish Balay 665118f5789SBarry Smith /* if both shift schemes are chosen by user, only use info->shiftpd */ 6661a890ab1SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 66742898933SHong Zhang ddiag = a->diag; 6689f7d0b68SHong Zhang sctx.shift_top = info->zeropivot; 6696cc28720Svictorle for (i=0; i<n; i++) { 6709f95998fSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 6719f7d0b68SHong Zhang d = (aa)[ddiag[i]]; 6721a890ab1SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 6739f7d0b68SHong Zhang v = aa+ai[i]; 6749f7d0b68SHong Zhang nz = ai[i+1] - ai[i]; 675e24cfe64SHong Zhang for (j=0; j<nz; j++) 6761a890ab1SHong Zhang rs += PetscAbsScalar(v[j]); 6771a890ab1SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 6786cc28720Svictorle } 679118f5789SBarry Smith sctx.shift_top *= 1.1; 680d2276718SHong Zhang sctx.nshift_max = 5; 681d2276718SHong Zhang sctx.shift_lo = 0.; 682d2276718SHong Zhang sctx.shift_hi = 1.; 683a0a356efSHong Zhang } 684a0a356efSHong Zhang 685085256b3SBarry Smith do { 686d2276718SHong Zhang sctx.lushift = PETSC_FALSE; 687289bc588SBarry Smith for (i=0; i<n; i++){ 688b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 689b3bf805bSHong Zhang bjtmp = bj + bi[i]; 6909f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 691289bc588SBarry Smith 692289bc588SBarry Smith /* load in initial (unfactored row) */ 6939f7d0b68SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 6949f7d0b68SHong Zhang ajtmp = aj + ai[r[i]]; 6959f7d0b68SHong Zhang v = aa + ai[r[i]]; 696085256b3SBarry Smith for (j=0; j<nz; j++) { 697b3bf805bSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 698085256b3SBarry Smith } 699d2276718SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 700bd1bc851SBarry Smith /* if (sctx.shift_amount > 0.0) printf("row %d, shift %g\n",i,sctx.shift_amount); */ 701289bc588SBarry Smith 702b3bf805bSHong Zhang row = *bjtmp++; 703f2582111SSatish Balay while (row < i) { 7048c37ef55SBarry Smith pc = rtmp + row; 7058c37ef55SBarry Smith if (*pc != 0.0) { 706010a20caSHong Zhang pv = b->a + diag_offset[row]; 707010a20caSHong Zhang pj = b->j + diag_offset[row] + 1; 70835aab85fSBarry Smith multiplier = *pc / *pv++; 7098c37ef55SBarry Smith *pc = multiplier; 710b3bf805bSHong Zhang nz = bi[row+1] - diag_offset[row] - 1; 7119f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 712dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 713289bc588SBarry Smith } 714b3bf805bSHong Zhang row = *bjtmp++; 715289bc588SBarry Smith } 716416022c9SBarry Smith /* finished row so stick it into b->a */ 717b3bf805bSHong Zhang pv = b->a + bi[i] ; 718b3bf805bSHong Zhang pj = b->j + bi[i] ; 719b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 720b3bf805bSHong Zhang diag = diag_offset[i] - bi[i]; 721e57ff13aSBarry Smith rs = 0.0; 722d3d32019SBarry Smith for (j=0; j<nz; j++) { 7239f7d0b68SHong Zhang pv[j] = rtmp[pj[j]]; 7249f7d0b68SHong Zhang rs += PetscAbsScalar(pv[j]); 725d3d32019SBarry Smith } 726e57ff13aSBarry Smith rs -= PetscAbsScalar(pv[diag]); 727d2276718SHong Zhang 728b3bf805bSHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 729d2276718SHong Zhang sctx.rs = rs; 730d2276718SHong Zhang sctx.pv = pv[diag]; 731c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 7325b5bc046SBarry Smith if (newshift == 1) break; 73335aab85fSBarry Smith } 734d2276718SHong Zhang 7350481f469SBarry Smith if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 7366cc28720Svictorle /* 7376c037d1bSvictorle * if no shift in this attempt & shifting & started shifting & can refine, 7386cc28720Svictorle * then try lower shift 7396cc28720Svictorle */ 7400481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 7410481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 7420481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 743d2276718SHong Zhang sctx.lushift = PETSC_TRUE; 744d2276718SHong Zhang sctx.nshift++; 7456cc28720Svictorle } 746d2276718SHong Zhang } while (sctx.lushift); 7470f11f9aeSSatish Balay 74835aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 74935aab85fSBarry Smith for (i=0; i<n; i++) { 750010a20caSHong Zhang b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]]; 75135aab85fSBarry Smith } 752606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 75378b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 75478b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 755db4efbfdSBarry Smith if (b->inode.use) { 7564108e4d5SBarry Smith C->ops->solve = MatSolve_SeqAIJ_Inode; 757db4efbfdSBarry Smith } else { 7588d8c7c43SBarry Smith PetscTruth row_identity, col_identity; 7598d8c7c43SBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 7608d8c7c43SBarry Smith ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 7618d8c7c43SBarry Smith if (row_identity && col_identity) { 7628d8c7c43SBarry Smith C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 7638d8c7c43SBarry Smith } else { 764db4efbfdSBarry Smith C->ops->solve = MatSolve_SeqAIJ; 765db4efbfdSBarry Smith } 7668d8c7c43SBarry Smith } 767db4efbfdSBarry Smith C->ops->solveadd = MatSolveAdd_SeqAIJ; 768db4efbfdSBarry Smith C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 769db4efbfdSBarry Smith C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 770de26497bSBarry Smith C->ops->matsolve = MatMatSolve_SeqAIJ; 771c456f294SBarry Smith C->assembled = PETSC_TRUE; 7725c9eb25fSBarry Smith C->preallocated = PETSC_TRUE; 773d0f46423SBarry Smith ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 774d2276718SHong Zhang if (sctx.nshift){ 775e738e422SBarry Smith if (info->shiftpd) { 7760481f469SBarry 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); 777e738e422SBarry Smith } else if (info->shiftnz) { 778e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 7796cc28720Svictorle } 7800697b6caSHong Zhang } 7813a40ed3dSBarry Smith PetscFunctionReturn(0); 782289bc588SBarry Smith } 7830889b5dcSvictorle 784137fb511SHong Zhang /* 785137fb511SHong Zhang This routine implements inplace ILU(0) with row or/and column permutations. 786137fb511SHong Zhang Input: 787137fb511SHong Zhang A - original matrix 788137fb511SHong Zhang Output; 789137fb511SHong Zhang A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i] 790137fb511SHong Zhang a->j (col index) is permuted by the inverse of colperm, then sorted 791137fb511SHong Zhang a->a reordered accordingly with a->j 792137fb511SHong Zhang a->diag (ptr to diagonal elements) is updated. 793137fb511SHong Zhang */ 794137fb511SHong Zhang #undef __FUNCT__ 795137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm" 7960481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info) 797137fb511SHong Zhang { 798b051339dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 799b051339dSHong Zhang IS isrow = a->row,isicol = a->icol; 800137fb511SHong Zhang PetscErrorCode ierr; 8015d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 8025d0c19d7SBarry Smith PetscInt i,j,n=A->rmap->n,*ai=a->i,*aj=a->j; 8035d0c19d7SBarry Smith PetscInt *ajtmp,nz,row; 804b051339dSHong Zhang PetscInt *diag = a->diag,nbdiag,*pj; 805a77337e4SBarry Smith PetscScalar *rtmp,*pc,multiplier,d; 806a77337e4SBarry Smith MatScalar *v,*pv; 807137fb511SHong Zhang PetscReal rs; 808137fb511SHong Zhang LUShift_Ctx sctx; 809b051339dSHong Zhang PetscInt newshift; 810137fb511SHong Zhang 811137fb511SHong Zhang PetscFunctionBegin; 812719d5645SBarry Smith if (A != B) SETERRQ(PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address"); 813137fb511SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 814137fb511SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 815137fb511SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr); 816137fb511SHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 817b051339dSHong Zhang ics = ic; 818137fb511SHong Zhang 819137fb511SHong Zhang sctx.shift_top = 0; 820137fb511SHong Zhang sctx.nshift_max = 0; 821137fb511SHong Zhang sctx.shift_lo = 0; 822137fb511SHong Zhang sctx.shift_hi = 0; 8230481f469SBarry Smith sctx.shift_fraction = 0; 824137fb511SHong Zhang 825137fb511SHong Zhang /* if both shift schemes are chosen by user, only use info->shiftpd */ 826137fb511SHong Zhang if (info->shiftpd) { /* set sctx.shift_top=max{rs} */ 827137fb511SHong Zhang sctx.shift_top = 0; 828137fb511SHong Zhang for (i=0; i<n; i++) { 829137fb511SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 830b051339dSHong Zhang d = (a->a)[diag[i]]; 831137fb511SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 832b051339dSHong Zhang v = a->a+ai[i]; 833b051339dSHong Zhang nz = ai[i+1] - ai[i]; 834137fb511SHong Zhang for (j=0; j<nz; j++) 835137fb511SHong Zhang rs += PetscAbsScalar(v[j]); 836137fb511SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 837137fb511SHong Zhang } 838137fb511SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 839137fb511SHong Zhang sctx.shift_top *= 1.1; 840137fb511SHong Zhang sctx.nshift_max = 5; 841137fb511SHong Zhang sctx.shift_lo = 0.; 842137fb511SHong Zhang sctx.shift_hi = 1.; 843137fb511SHong Zhang } 844137fb511SHong Zhang 845137fb511SHong Zhang sctx.shift_amount = 0; 846137fb511SHong Zhang sctx.nshift = 0; 847137fb511SHong Zhang do { 848137fb511SHong Zhang sctx.lushift = PETSC_FALSE; 849137fb511SHong Zhang for (i=0; i<n; i++){ 850b051339dSHong Zhang /* load in initial unfactored row */ 851b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 852b051339dSHong Zhang ajtmp = aj + ai[r[i]]; 853b051339dSHong Zhang v = a->a + ai[r[i]]; 854137fb511SHong Zhang /* sort permuted ajtmp and values v accordingly */ 855b051339dSHong Zhang for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]]; 856137fb511SHong Zhang ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr); 857137fb511SHong Zhang 858b051339dSHong Zhang diag[r[i]] = ai[r[i]]; 859137fb511SHong Zhang for (j=0; j<nz; j++) { 860137fb511SHong Zhang rtmp[ajtmp[j]] = v[j]; 861b051339dSHong Zhang if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */ 862137fb511SHong Zhang } 863137fb511SHong Zhang rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 864137fb511SHong Zhang 865b051339dSHong Zhang row = *ajtmp++; 866137fb511SHong Zhang while (row < i) { 867137fb511SHong Zhang pc = rtmp + row; 868137fb511SHong Zhang if (*pc != 0.0) { 869b051339dSHong Zhang pv = a->a + diag[r[row]]; 870b051339dSHong Zhang pj = aj + diag[r[row]] + 1; 871137fb511SHong Zhang 872137fb511SHong Zhang multiplier = *pc / *pv++; 873137fb511SHong Zhang *pc = multiplier; 874b051339dSHong Zhang nz = ai[r[row]+1] - diag[r[row]] - 1; 875b051339dSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 876dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 877137fb511SHong Zhang } 878b051339dSHong Zhang row = *ajtmp++; 879137fb511SHong Zhang } 880b051339dSHong Zhang /* finished row so overwrite it onto a->a */ 881b051339dSHong Zhang pv = a->a + ai[r[i]] ; 882b051339dSHong Zhang pj = aj + ai[r[i]] ; 883b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 884b051339dSHong Zhang nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */ 885137fb511SHong Zhang 886137fb511SHong Zhang rs = 0.0; 887137fb511SHong Zhang for (j=0; j<nz; j++) { 888b051339dSHong Zhang pv[j] = rtmp[pj[j]]; 889b051339dSHong Zhang if (j != nbdiag) rs += PetscAbsScalar(pv[j]); 890137fb511SHong Zhang } 891137fb511SHong Zhang 892137fb511SHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 893137fb511SHong Zhang sctx.rs = rs; 894b051339dSHong Zhang sctx.pv = pv[nbdiag]; 895c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 896137fb511SHong Zhang if (newshift == 1) break; 897137fb511SHong Zhang } 898137fb511SHong Zhang 8990481f469SBarry Smith if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 900137fb511SHong Zhang /* 901137fb511SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 902137fb511SHong Zhang * then try lower shift 903137fb511SHong Zhang */ 9040481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 9050481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 9060481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 907137fb511SHong Zhang sctx.lushift = PETSC_TRUE; 908137fb511SHong Zhang sctx.nshift++; 909137fb511SHong Zhang } 910137fb511SHong Zhang } while (sctx.lushift); 911137fb511SHong Zhang 912137fb511SHong Zhang /* invert diagonal entries for simplier triangular solves */ 913137fb511SHong Zhang for (i=0; i<n; i++) { 914b051339dSHong Zhang a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]]; 915137fb511SHong Zhang } 916137fb511SHong Zhang 917137fb511SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 918137fb511SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 919137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 920b051339dSHong Zhang A->ops->solve = MatSolve_SeqAIJ_InplaceWithPerm; 921db4efbfdSBarry Smith A->ops->solveadd = MatSolveAdd_SeqAIJ; 922db4efbfdSBarry Smith A->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 923db4efbfdSBarry Smith A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 924b051339dSHong Zhang A->assembled = PETSC_TRUE; 9255c9eb25fSBarry Smith A->preallocated = PETSC_TRUE; 926d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr); 927137fb511SHong Zhang if (sctx.nshift){ 928e738e422SBarry Smith if (info->shiftpd) { 9290481f469SBarry 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); 930e738e422SBarry Smith } else if (info->shiftnz) { 931e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 932137fb511SHong Zhang } 933137fb511SHong Zhang } 934137fb511SHong Zhang PetscFunctionReturn(0); 935137fb511SHong Zhang } 936137fb511SHong Zhang 9370a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9384a2ae208SSatish Balay #undef __FUNCT__ 9394a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ" 9400481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info) 941da3a660dSBarry Smith { 942dfbe8321SBarry Smith PetscErrorCode ierr; 943416022c9SBarry Smith Mat C; 944416022c9SBarry Smith 9453a40ed3dSBarry Smith PetscFunctionBegin; 94643244d56SBarry Smith ierr = MatGetFactor(A,MAT_SOLVER_PETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr); 947719d5645SBarry Smith ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr); 948719d5645SBarry Smith ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr); 949db4efbfdSBarry Smith A->ops->solve = C->ops->solve; 950db4efbfdSBarry Smith A->ops->solvetranspose = C->ops->solvetranspose; 951273d9f13SBarry Smith ierr = MatHeaderCopy(A,C);CHKERRQ(ierr); 95252e6d16bSBarry Smith ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr); 9533a40ed3dSBarry Smith PetscFunctionReturn(0); 954da3a660dSBarry Smith } 9550a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9561d098868SHong Zhang 957889a8dedSBarry Smith 9584a2ae208SSatish Balay #undef __FUNCT__ 9594a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ" 960dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 9618c37ef55SBarry Smith { 962416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 963416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 9646849ba73SBarry Smith PetscErrorCode ierr; 9655d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 9665d0c19d7SBarry Smith PetscInt nz; 9675d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 968dd6ea824SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 969d9fead3dSBarry Smith const PetscScalar *b; 970dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 9718c37ef55SBarry Smith 9723a40ed3dSBarry Smith PetscFunctionBegin; 9733a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 97491bf9adeSBarry Smith 975d9fead3dSBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 9761ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 977416022c9SBarry Smith tmp = a->solve_work; 9788c37ef55SBarry Smith 9793b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 9803b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 9818c37ef55SBarry Smith 9828c37ef55SBarry Smith /* forward solve the lower triangular */ 9838c37ef55SBarry Smith tmp[0] = b[*r++]; 984010a20caSHong Zhang tmps = tmp; 9858c37ef55SBarry Smith for (i=1; i<n; i++) { 986010a20caSHong Zhang v = aa + ai[i] ; 987010a20caSHong Zhang vi = aj + ai[i] ; 98853e7425aSBarry Smith nz = a->diag[i] - ai[i]; 98953e7425aSBarry Smith sum = b[*r++]; 990003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 9918c37ef55SBarry Smith tmp[i] = sum; 9928c37ef55SBarry Smith } 9938c37ef55SBarry Smith 9948c37ef55SBarry Smith /* backward solve the upper triangular */ 9958c37ef55SBarry Smith for (i=n-1; i>=0; i--){ 996010a20caSHong Zhang v = aa + a->diag[i] + 1; 997010a20caSHong Zhang vi = aj + a->diag[i] + 1; 998416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 9998c37ef55SBarry Smith sum = tmp[i]; 1000003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1001010a20caSHong Zhang x[*c--] = tmp[i] = sum*aa[a->diag[i]]; 10028c37ef55SBarry Smith } 10038c37ef55SBarry Smith 10043b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10053b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1006d9fead3dSBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 10071ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1008dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 10093a40ed3dSBarry Smith PetscFunctionReturn(0); 10108c37ef55SBarry Smith } 1011026e39d0SSatish Balay 10122bebee5dSHong Zhang #undef __FUNCT__ 10132bebee5dSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ" 10142bebee5dSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X) 10152bebee5dSHong Zhang { 10162bebee5dSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10172bebee5dSHong Zhang IS iscol = a->col,isrow = a->row; 10182bebee5dSHong Zhang PetscErrorCode ierr; 10195d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10205d0c19d7SBarry Smith PetscInt nz,neq; 10215d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 1022dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1023dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1024db4efbfdSBarry Smith PetscTruth bisdense,xisdense; 10252bebee5dSHong Zhang 10262bebee5dSHong Zhang PetscFunctionBegin; 10272bebee5dSHong Zhang if (!n) PetscFunctionReturn(0); 10282bebee5dSHong Zhang 1029db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1030db4efbfdSBarry Smith if (!bisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 1031db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1032db4efbfdSBarry Smith if (!xisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 1033db4efbfdSBarry Smith 10342bebee5dSHong Zhang ierr = MatGetArray(B,&b);CHKERRQ(ierr); 10352bebee5dSHong Zhang ierr = MatGetArray(X,&x);CHKERRQ(ierr); 10362bebee5dSHong Zhang 10372bebee5dSHong Zhang tmp = a->solve_work; 10382bebee5dSHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10392bebee5dSHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 10402bebee5dSHong Zhang 1041a36b22ccSSatish Balay for (neq=0; neq<B->cmap->n; neq++){ 10422bebee5dSHong Zhang /* forward solve the lower triangular */ 10432bebee5dSHong Zhang tmp[0] = b[r[0]]; 10442bebee5dSHong Zhang tmps = tmp; 10452bebee5dSHong Zhang for (i=1; i<n; i++) { 10462bebee5dSHong Zhang v = aa + ai[i] ; 10472bebee5dSHong Zhang vi = aj + ai[i] ; 10482bebee5dSHong Zhang nz = a->diag[i] - ai[i]; 10492bebee5dSHong Zhang sum = b[r[i]]; 1050003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10512bebee5dSHong Zhang tmp[i] = sum; 10522bebee5dSHong Zhang } 10532bebee5dSHong Zhang /* backward solve the upper triangular */ 10542bebee5dSHong Zhang for (i=n-1; i>=0; i--){ 10552bebee5dSHong Zhang v = aa + a->diag[i] + 1; 10562bebee5dSHong Zhang vi = aj + a->diag[i] + 1; 10572bebee5dSHong Zhang nz = ai[i+1] - a->diag[i] - 1; 10582bebee5dSHong Zhang sum = tmp[i]; 1059003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10602bebee5dSHong Zhang x[c[i]] = tmp[i] = sum*aa[a->diag[i]]; 10612bebee5dSHong Zhang } 10622bebee5dSHong Zhang 10632bebee5dSHong Zhang b += n; 10642bebee5dSHong Zhang x += n; 10652bebee5dSHong Zhang } 10662bebee5dSHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10672bebee5dSHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10682bebee5dSHong Zhang ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 10692bebee5dSHong Zhang ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 1070dc0b31edSSatish Balay ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 10712bebee5dSHong Zhang PetscFunctionReturn(0); 10722bebee5dSHong Zhang } 10732bebee5dSHong Zhang 1074137fb511SHong Zhang #undef __FUNCT__ 1075137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm" 1076137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx) 1077137fb511SHong Zhang { 1078137fb511SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1079137fb511SHong Zhang IS iscol = a->col,isrow = a->row; 1080137fb511SHong Zhang PetscErrorCode ierr; 10815d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 10825d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10835d0c19d7SBarry Smith PetscInt nz,row; 1084dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1085dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1086137fb511SHong Zhang 1087137fb511SHong Zhang PetscFunctionBegin; 1088137fb511SHong Zhang if (!n) PetscFunctionReturn(0); 1089137fb511SHong Zhang 1090137fb511SHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 1091137fb511SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1092137fb511SHong Zhang tmp = a->solve_work; 1093137fb511SHong Zhang 1094137fb511SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1095137fb511SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1096137fb511SHong Zhang 1097137fb511SHong Zhang /* forward solve the lower triangular */ 1098137fb511SHong Zhang tmp[0] = b[*r++]; 1099137fb511SHong Zhang tmps = tmp; 1100137fb511SHong Zhang for (row=1; row<n; row++) { 1101137fb511SHong Zhang i = rout[row]; /* permuted row */ 1102137fb511SHong Zhang v = aa + ai[i] ; 1103137fb511SHong Zhang vi = aj + ai[i] ; 1104137fb511SHong Zhang nz = a->diag[i] - ai[i]; 1105137fb511SHong Zhang sum = b[*r++]; 1106003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1107137fb511SHong Zhang tmp[row] = sum; 1108137fb511SHong Zhang } 1109137fb511SHong Zhang 1110137fb511SHong Zhang /* backward solve the upper triangular */ 1111137fb511SHong Zhang for (row=n-1; row>=0; row--){ 1112137fb511SHong Zhang i = rout[row]; /* permuted row */ 1113137fb511SHong Zhang v = aa + a->diag[i] + 1; 1114137fb511SHong Zhang vi = aj + a->diag[i] + 1; 1115137fb511SHong Zhang nz = ai[i+1] - a->diag[i] - 1; 1116137fb511SHong Zhang sum = tmp[row]; 1117003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1118137fb511SHong Zhang x[*c--] = tmp[row] = sum*aa[a->diag[i]]; 1119137fb511SHong Zhang } 1120137fb511SHong Zhang 1121137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1122137fb511SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1123137fb511SHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 1124137fb511SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1125dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1126137fb511SHong Zhang PetscFunctionReturn(0); 1127137fb511SHong Zhang } 1128137fb511SHong Zhang 1129930ae53cSBarry Smith /* ----------------------------------------------------------- */ 1130a4005a5dSBarry Smith #include "../src/mat/impls/aij/seq/ftn-kernels/fsolve.h" 11314a2ae208SSatish Balay #undef __FUNCT__ 11324a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering" 1133dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 1134930ae53cSBarry Smith { 1135930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 11366849ba73SBarry Smith PetscErrorCode ierr; 1137d0f46423SBarry Smith PetscInt n = A->rmap->n; 1138b377110cSBarry Smith const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag; 1139356650c2SBarry Smith PetscScalar *x; 1140356650c2SBarry Smith const PetscScalar *b; 1141dd6ea824SBarry Smith const MatScalar *aa = a->a; 1142aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 1143356650c2SBarry Smith PetscInt adiag_i,i,nz,ai_i; 1144b377110cSBarry Smith const PetscInt *vi; 1145dd6ea824SBarry Smith const MatScalar *v; 1146dd6ea824SBarry Smith PetscScalar sum; 1147d85afc42SSatish Balay #endif 1148930ae53cSBarry Smith 11493a40ed3dSBarry Smith PetscFunctionBegin; 11503a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 1151930ae53cSBarry Smith 1152356650c2SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 11531ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1154930ae53cSBarry Smith 1155aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 11561c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 11571c4feecaSSatish Balay #else 1158930ae53cSBarry Smith /* forward solve the lower triangular */ 1159930ae53cSBarry Smith x[0] = b[0]; 1160930ae53cSBarry Smith for (i=1; i<n; i++) { 1161930ae53cSBarry Smith ai_i = ai[i]; 1162930ae53cSBarry Smith v = aa + ai_i; 1163930ae53cSBarry Smith vi = aj + ai_i; 1164930ae53cSBarry Smith nz = adiag[i] - ai_i; 1165930ae53cSBarry Smith sum = b[i]; 1166003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1167930ae53cSBarry Smith x[i] = sum; 1168930ae53cSBarry Smith } 1169930ae53cSBarry Smith 1170930ae53cSBarry Smith /* backward solve the upper triangular */ 1171930ae53cSBarry Smith for (i=n-1; i>=0; i--){ 1172930ae53cSBarry Smith adiag_i = adiag[i]; 1173d708749eSBarry Smith v = aa + adiag_i + 1; 1174d708749eSBarry Smith vi = aj + adiag_i + 1; 1175930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 1176930ae53cSBarry Smith sum = x[i]; 1177003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1178930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 1179930ae53cSBarry Smith } 11801c4feecaSSatish Balay #endif 1181dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1182356650c2SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 11831ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 11843a40ed3dSBarry Smith PetscFunctionReturn(0); 1185930ae53cSBarry Smith } 1186930ae53cSBarry Smith 11874a2ae208SSatish Balay #undef __FUNCT__ 11884a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqAIJ" 1189dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 1190da3a660dSBarry Smith { 1191416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1192416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 11936849ba73SBarry Smith PetscErrorCode ierr; 119404fbf559SBarry Smith PetscInt i, n = A->rmap->n,j; 11955d0c19d7SBarry Smith PetscInt nz; 119604fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j; 119704fbf559SBarry Smith PetscScalar *x,*tmp,sum; 119804fbf559SBarry Smith const PetscScalar *b; 1199dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1200da3a660dSBarry Smith 12013a40ed3dSBarry Smith PetscFunctionBegin; 120278b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 1203da3a660dSBarry Smith 120404fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12051ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1206416022c9SBarry Smith tmp = a->solve_work; 1207da3a660dSBarry Smith 12083b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12093b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1210da3a660dSBarry Smith 1211da3a660dSBarry Smith /* forward solve the lower triangular */ 1212da3a660dSBarry Smith tmp[0] = b[*r++]; 1213da3a660dSBarry Smith for (i=1; i<n; i++) { 1214010a20caSHong Zhang v = aa + ai[i] ; 1215010a20caSHong Zhang vi = aj + ai[i] ; 1216416022c9SBarry Smith nz = a->diag[i] - ai[i]; 1217da3a660dSBarry Smith sum = b[*r++]; 121804fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1219da3a660dSBarry Smith tmp[i] = sum; 1220da3a660dSBarry Smith } 1221da3a660dSBarry Smith 1222da3a660dSBarry Smith /* backward solve the upper triangular */ 1223da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1224010a20caSHong Zhang v = aa + a->diag[i] + 1; 1225010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1226416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 1227da3a660dSBarry Smith sum = tmp[i]; 122804fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1229010a20caSHong Zhang tmp[i] = sum*aa[a->diag[i]]; 1230da3a660dSBarry Smith x[*c--] += tmp[i]; 1231da3a660dSBarry Smith } 1232da3a660dSBarry Smith 12333b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 12343b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 123504fbf559SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12361ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1237dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1238898183ecSLois Curfman McInnes 12393a40ed3dSBarry Smith PetscFunctionReturn(0); 1240da3a660dSBarry Smith } 124104fbf559SBarry Smith 12424a2ae208SSatish Balay #undef __FUNCT__ 12434a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqAIJ" 1244dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 1245da3a660dSBarry Smith { 1246416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 12472235e667SBarry Smith IS iscol = a->col,isrow = a->row; 12486849ba73SBarry Smith PetscErrorCode ierr; 124904fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 125004fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 125104fbf559SBarry Smith PetscInt nz; 125204fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1253dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 125404fbf559SBarry Smith const PetscScalar *b; 1255da3a660dSBarry Smith 12563a40ed3dSBarry Smith PetscFunctionBegin; 125704fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12581ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1259416022c9SBarry Smith tmp = a->solve_work; 1260da3a660dSBarry Smith 12612235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12622235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1263da3a660dSBarry Smith 1264da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 12652235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1266da3a660dSBarry Smith 1267da3a660dSBarry Smith /* forward solve the U^T */ 1268da3a660dSBarry Smith for (i=0; i<n; i++) { 1269010a20caSHong Zhang v = aa + diag[i] ; 1270010a20caSHong Zhang vi = aj + diag[i] + 1; 1271f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 1272c38d4ed2SBarry Smith s1 = tmp[i]; 1273ef66eb69SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 127404fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1275c38d4ed2SBarry Smith tmp[i] = s1; 1276da3a660dSBarry Smith } 1277da3a660dSBarry Smith 1278da3a660dSBarry Smith /* backward solve the L^T */ 1279da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1280010a20caSHong Zhang v = aa + diag[i] - 1 ; 1281010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1282f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 1283f1af5d2fSBarry Smith s1 = tmp[i]; 128404fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 1285da3a660dSBarry Smith } 1286da3a660dSBarry Smith 1287da3a660dSBarry Smith /* copy tmp into x according to permutation */ 1288591294e4SBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1289da3a660dSBarry Smith 12902235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 12912235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 129204fbf559SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12931ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1294da3a660dSBarry Smith 1295dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 12963a40ed3dSBarry Smith PetscFunctionReturn(0); 1297da3a660dSBarry Smith } 1298da3a660dSBarry Smith 12994a2ae208SSatish Balay #undef __FUNCT__ 13004a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ" 1301dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 1302da3a660dSBarry Smith { 1303416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13042235e667SBarry Smith IS iscol = a->col,isrow = a->row; 13056849ba73SBarry Smith PetscErrorCode ierr; 130604fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 130704fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 130804fbf559SBarry Smith PetscInt nz; 130904fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1310dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 131104fbf559SBarry Smith const PetscScalar *b; 13126abc6512SBarry Smith 13133a40ed3dSBarry Smith PetscFunctionBegin; 131423bc6035SMatthew Knepley if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 131504fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 13161ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1317416022c9SBarry Smith tmp = a->solve_work; 13186abc6512SBarry Smith 13192235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13202235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 13216abc6512SBarry Smith 13226abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 13232235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 13246abc6512SBarry Smith 13256abc6512SBarry Smith /* forward solve the U^T */ 13266abc6512SBarry 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; 133004fbf559SBarry Smith s1 = tmp[i]; 133104fbf559SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 133204fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 133304fbf559SBarry Smith tmp[i] = s1; 13346abc6512SBarry Smith } 13356abc6512SBarry Smith 13366abc6512SBarry Smith /* backward solve the L^T */ 13376abc6512SBarry 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]; 134104fbf559SBarry Smith s1 = tmp[i]; 134204fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 13436abc6512SBarry Smith } 13446abc6512SBarry Smith 13456abc6512SBarry Smith /* copy tmp into x according to permutation */ 13466abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 13476abc6512SBarry 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); 13526abc6512SBarry Smith 135304fbf559SBarry Smith ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 13543a40ed3dSBarry Smith PetscFunctionReturn(0); 1355da3a660dSBarry Smith } 135604fbf559SBarry Smith 13579e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 13583c5fc038SBarry Smith EXTERN PetscErrorCode Mat_CheckInode(Mat,PetscTruth); 1359f77e22a1SHong Zhang EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscTruth); 136008480c60SBarry Smith 13618fc3a347SHong Zhang /* 13628a76ed4fSHong Zhang ilu() under revised new data structure. 1363813a1f2bSShri Abhyankar Factored arrays bj and ba are stored as 1364813a1f2bSShri Abhyankar L(0,:), L(1,:), ...,L(n-1,:), U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:) 1365813a1f2bSShri Abhyankar 1366813a1f2bSShri Abhyankar bi=fact->i is an array of size n+1, in which 1367813a1f2bSShri Abhyankar bi+ 1368baabb860SHong Zhang bi[i]: points to 1st entry of L(i,:),i=0,...,n-1 1369baabb860SHong Zhang bi[n]: points to L(n-1,n-1)+1 1370baabb860SHong Zhang 1371813a1f2bSShri Abhyankar bdiag=fact->diag is an array of size n+1,in which 1372baabb860SHong Zhang bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1 1373baabb860SHong Zhang bdiag[n]: points to diagonal of U(n-1,0)-1 1374813a1f2bSShri Abhyankar 1375813a1f2bSShri Abhyankar U(i,:) contains bdiag[i] as its last entry, i.e., 1376813a1f2bSShri Abhyankar U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 1377813a1f2bSShri Abhyankar */ 1378813a1f2bSShri Abhyankar #undef __FUNCT__ 1379b4f5729fSShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct" 1380b4f5729fSShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1381813a1f2bSShri Abhyankar { 1382813a1f2bSShri Abhyankar 1383813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1384813a1f2bSShri Abhyankar PetscErrorCode ierr; 1385813a1f2bSShri Abhyankar PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag; 1386813a1f2bSShri Abhyankar PetscInt i,j,nz,*bi,*bj,*bdiag,bi_temp; 13871df811f5SHong Zhang PetscTruth missing; 13881df811f5SHong Zhang IS isicol; 1389813a1f2bSShri Abhyankar 1390813a1f2bSShri Abhyankar PetscFunctionBegin; 13911df811f5SHong Zhang /* printf("MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct ...\n"); */ 13921df811f5SHong 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); 13931df811f5SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 13941df811f5SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 13951df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 13961df811f5SHong Zhang 1397813a1f2bSShri Abhyankar ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr); 1398813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1399813a1f2bSShri Abhyankar 1400813a1f2bSShri Abhyankar /* allocate matrix arrays for new data structure */ 1401813a1f2bSShri Abhyankar ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,n+1,PetscInt,&b->i);CHKERRQ(ierr); 1402813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1403813a1f2bSShri Abhyankar b->singlemalloc = PETSC_TRUE; 1404813a1f2bSShri Abhyankar if (!b->diag){ 1405813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr); 1406813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1407813a1f2bSShri Abhyankar } 1408813a1f2bSShri Abhyankar bdiag = b->diag; 1409813a1f2bSShri Abhyankar 1410813a1f2bSShri Abhyankar if (n > 0) { 1411813a1f2bSShri Abhyankar ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr); 1412813a1f2bSShri Abhyankar } 1413813a1f2bSShri Abhyankar 1414813a1f2bSShri Abhyankar /* set bi and bj with new data structure */ 1415813a1f2bSShri Abhyankar bi = b->i; 1416813a1f2bSShri Abhyankar bj = b->j; 1417813a1f2bSShri Abhyankar 1418813a1f2bSShri Abhyankar /* L part */ 1419813a1f2bSShri Abhyankar bi[0] = 0; 1420813a1f2bSShri Abhyankar for (i=0; i<n; i++){ 1421813a1f2bSShri Abhyankar nz = adiag[i] - ai[i]; 1422813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nz; 1423813a1f2bSShri Abhyankar aj = a->j + ai[i]; 1424813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 1425813a1f2bSShri Abhyankar *bj = aj[j]; bj++; 1426813a1f2bSShri Abhyankar } 1427813a1f2bSShri Abhyankar } 1428813a1f2bSShri Abhyankar 1429813a1f2bSShri Abhyankar /* U part */ 1430813a1f2bSShri Abhyankar bi_temp = bi[n]; 143162fbd6afSShri Abhyankar bdiag[n] = bi[n]-1; 1432813a1f2bSShri Abhyankar for (i=n-1; i>=0; i--){ 1433813a1f2bSShri Abhyankar nz = ai[i+1] - adiag[i] - 1; 1434813a1f2bSShri Abhyankar bi_temp = bi_temp + nz + 1; 1435813a1f2bSShri Abhyankar aj = a->j + adiag[i] + 1; 1436813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 1437813a1f2bSShri Abhyankar *bj = aj[j]; bj++; 1438813a1f2bSShri Abhyankar } 1439813a1f2bSShri Abhyankar /* diag[i] */ 1440813a1f2bSShri Abhyankar *bj = i; bj++; 1441813a1f2bSShri Abhyankar bdiag[i] = bi_temp - 1; 1442813a1f2bSShri Abhyankar } 14431df811f5SHong Zhang 14441df811f5SHong Zhang fact->factor = MAT_FACTOR_ILU; 14451df811f5SHong Zhang fact->info.factor_mallocs = 0; 14461df811f5SHong Zhang fact->info.fill_ratio_given = info->fill; 14471df811f5SHong Zhang fact->info.fill_ratio_needed = 1.0; 14481df811f5SHong Zhang fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 14491df811f5SHong Zhang 14501df811f5SHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 14511df811f5SHong Zhang b->row = isrow; 14521df811f5SHong Zhang b->col = iscol; 14531df811f5SHong Zhang b->icol = isicol; 14541df811f5SHong Zhang ierr = PetscMalloc((fact->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 14551df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 14561df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1457813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1458813a1f2bSShri Abhyankar } 1459813a1f2bSShri Abhyankar 1460813a1f2bSShri Abhyankar #undef __FUNCT__ 1461b4f5729fSShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_newdatastruct" 1462b4f5729fSShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1463813a1f2bSShri Abhyankar { 1464813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1465813a1f2bSShri Abhyankar IS isicol; 1466813a1f2bSShri Abhyankar PetscErrorCode ierr; 1467813a1f2bSShri Abhyankar const PetscInt *r,*ic; 14681df811f5SHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j; 1469813a1f2bSShri Abhyankar PetscInt *bi,*cols,nnz,*cols_lvl; 1470813a1f2bSShri Abhyankar PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 1471813a1f2bSShri Abhyankar PetscInt i,levels,diagonal_fill; 1472813a1f2bSShri Abhyankar PetscTruth col_identity,row_identity; 1473813a1f2bSShri Abhyankar PetscReal f; 1474813a1f2bSShri Abhyankar PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 1475813a1f2bSShri Abhyankar PetscBT lnkbt; 1476813a1f2bSShri Abhyankar PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 1477813a1f2bSShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1478813a1f2bSShri Abhyankar PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 1479813a1f2bSShri Abhyankar 1480813a1f2bSShri Abhyankar PetscFunctionBegin; 14811df811f5SHong Zhang /* printf("MatILUFactorSymbolic_SeqAIJ_newdatastruct ...\n"); */ 1482813a1f2bSShri Abhyankar levels = (PetscInt)info->levels; 1483813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 1484813a1f2bSShri Abhyankar ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 1485813a1f2bSShri Abhyankar 1486813a1f2bSShri Abhyankar if (!levels && row_identity && col_identity) { 1487813a1f2bSShri Abhyankar /* special case: ilu(0) with natural ordering */ 1488b4f5729fSShri Abhyankar ierr = MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1489813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1490813a1f2bSShri Abhyankar } 1491813a1f2bSShri Abhyankar 14921df811f5SHong 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); 14931df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 1494813a1f2bSShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1495813a1f2bSShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 1496813a1f2bSShri Abhyankar 1497*a6df321fSShri Abhyankar /* get new row and diagonal pointers */ 1498*a6df321fSShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt,&bi,n+1,PetscInt,&bdiag);CHKERRQ(ierr); 1499*a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 1500813a1f2bSShri Abhyankar 15010e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr); 1502813a1f2bSShri Abhyankar 1503813a1f2bSShri Abhyankar /* create a linked list for storing column indices of the active row */ 1504813a1f2bSShri Abhyankar nlnk = n + 1; 1505813a1f2bSShri Abhyankar ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1506813a1f2bSShri Abhyankar 1507813a1f2bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 15081df811f5SHong Zhang f = info->fill; 15091df811f5SHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 1510813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1511813a1f2bSShri Abhyankar current_space = free_space; 1512813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 1513813a1f2bSShri Abhyankar current_space_lvl = free_space_lvl; 1514813a1f2bSShri Abhyankar 1515813a1f2bSShri Abhyankar for (i=0; i<n; i++) { 1516813a1f2bSShri Abhyankar nzi = 0; 1517813a1f2bSShri Abhyankar /* copy current row into linked list */ 1518813a1f2bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 1519813a1f2bSShri 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); 1520813a1f2bSShri Abhyankar cols = aj + ai[r[i]]; 1521813a1f2bSShri Abhyankar lnk[i] = -1; /* marker to indicate if diagonal exists */ 1522813a1f2bSShri Abhyankar ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1523813a1f2bSShri Abhyankar nzi += nlnk; 1524813a1f2bSShri Abhyankar 1525813a1f2bSShri Abhyankar /* make sure diagonal entry is included */ 1526813a1f2bSShri Abhyankar if (diagonal_fill && lnk[i] == -1) { 1527813a1f2bSShri Abhyankar fm = n; 1528813a1f2bSShri Abhyankar while (lnk[fm] < i) fm = lnk[fm]; 1529813a1f2bSShri Abhyankar lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 1530813a1f2bSShri Abhyankar lnk[fm] = i; 1531813a1f2bSShri Abhyankar lnk_lvl[i] = 0; 1532813a1f2bSShri Abhyankar nzi++; dcount++; 1533813a1f2bSShri Abhyankar } 1534813a1f2bSShri Abhyankar 1535813a1f2bSShri Abhyankar /* add pivot rows into the active row */ 1536813a1f2bSShri Abhyankar nzbd = 0; 1537813a1f2bSShri Abhyankar prow = lnk[n]; 1538813a1f2bSShri Abhyankar while (prow < i) { 1539813a1f2bSShri Abhyankar nnz = bdiag[prow]; 1540813a1f2bSShri Abhyankar cols = bj_ptr[prow] + nnz + 1; 1541813a1f2bSShri Abhyankar cols_lvl = bjlvl_ptr[prow] + nnz + 1; 1542813a1f2bSShri Abhyankar nnz = bi[prow+1] - bi[prow] - nnz - 1; 1543813a1f2bSShri Abhyankar ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 1544813a1f2bSShri Abhyankar nzi += nlnk; 1545813a1f2bSShri Abhyankar prow = lnk[prow]; 1546813a1f2bSShri Abhyankar nzbd++; 1547813a1f2bSShri Abhyankar } 1548813a1f2bSShri Abhyankar bdiag[i] = nzbd; 1549813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nzi; 1550813a1f2bSShri Abhyankar 1551813a1f2bSShri Abhyankar /* if free space is not available, make more free space */ 1552813a1f2bSShri Abhyankar if (current_space->local_remaining<nzi) { 1553813a1f2bSShri Abhyankar nnz = 2*nzi*(n - i); /* estimated and max additional space needed */ 1554813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1555813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 1556813a1f2bSShri Abhyankar reallocs++; 1557813a1f2bSShri Abhyankar } 1558813a1f2bSShri Abhyankar 1559813a1f2bSShri Abhyankar /* copy data into free_space and free_space_lvl, then initialize lnk */ 1560813a1f2bSShri Abhyankar ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 1561813a1f2bSShri Abhyankar bj_ptr[i] = current_space->array; 1562813a1f2bSShri Abhyankar bjlvl_ptr[i] = current_space_lvl->array; 1563813a1f2bSShri Abhyankar 1564813a1f2bSShri Abhyankar /* make sure the active row i has diagonal entry */ 1565813a1f2bSShri Abhyankar if (*(bj_ptr[i]+bdiag[i]) != i) { 1566813a1f2bSShri Abhyankar SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 1567813a1f2bSShri Abhyankar try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 1568813a1f2bSShri Abhyankar } 1569813a1f2bSShri Abhyankar 1570813a1f2bSShri Abhyankar current_space->array += nzi; 1571813a1f2bSShri Abhyankar current_space->local_used += nzi; 1572813a1f2bSShri Abhyankar current_space->local_remaining -= nzi; 1573813a1f2bSShri Abhyankar current_space_lvl->array += nzi; 1574813a1f2bSShri Abhyankar current_space_lvl->local_used += nzi; 1575813a1f2bSShri Abhyankar current_space_lvl->local_remaining -= nzi; 1576813a1f2bSShri Abhyankar } 1577813a1f2bSShri Abhyankar 1578813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1579813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 1580813a1f2bSShri Abhyankar 1581813a1f2bSShri Abhyankar /* destroy list of free space and other temporary arrays */ 1582813a1f2bSShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 1583813a1f2bSShri Abhyankar 1584813a1f2bSShri Abhyankar /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 1585f268cba8SShri Abhyankar ierr = PetscFreeSpaceContiguous_LU_v2(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 1586813a1f2bSShri Abhyankar 1587813a1f2bSShri Abhyankar ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1588813a1f2bSShri Abhyankar ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 15890e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 1590813a1f2bSShri Abhyankar 1591813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO) 1592813a1f2bSShri Abhyankar { 1593813a1f2bSShri Abhyankar PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1594813a1f2bSShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 1595813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1596813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 1597813a1f2bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1598813a1f2bSShri Abhyankar if (diagonal_fill) { 1599813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 1600813a1f2bSShri Abhyankar } 1601813a1f2bSShri Abhyankar } 1602813a1f2bSShri Abhyankar #endif 1603813a1f2bSShri Abhyankar 1604813a1f2bSShri Abhyankar /* put together the new matrix */ 1605813a1f2bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1606813a1f2bSShri Abhyankar ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 1607813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1608813a1f2bSShri Abhyankar b->free_a = PETSC_TRUE; 1609813a1f2bSShri Abhyankar b->free_ij = PETSC_TRUE; 1610813a1f2bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 1611f268cba8SShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 1612813a1f2bSShri Abhyankar b->j = bj; 1613813a1f2bSShri Abhyankar b->i = bi; 1614813a1f2bSShri Abhyankar b->diag = bdiag; 1615813a1f2bSShri Abhyankar b->ilen = 0; 1616813a1f2bSShri Abhyankar b->imax = 0; 1617813a1f2bSShri Abhyankar b->row = isrow; 1618813a1f2bSShri Abhyankar b->col = iscol; 1619813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1620813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1621813a1f2bSShri Abhyankar b->icol = isicol; 1622813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1623813a1f2bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. 1624813a1f2bSShri Abhyankar Allocate bdiag, solve_work, new a, new j */ 1625f268cba8SShri Abhyankar ierr = PetscLogObjectMemory(fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 1626f268cba8SShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 1627813a1f2bSShri Abhyankar (fact)->info.factor_mallocs = reallocs; 1628813a1f2bSShri Abhyankar (fact)->info.fill_ratio_given = f; 1629f268cba8SShri Abhyankar (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 1630b4f5729fSShri Abhyankar (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct; 16314108e4d5SBarry Smith /* ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */ 1632813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1633813a1f2bSShri Abhyankar } 1634813a1f2bSShri Abhyankar 16356516239fSHong Zhang #undef __FUNCT__ 16366516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ" 16376516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 16386516239fSHong Zhang { 16396516239fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 16406516239fSHong Zhang IS isicol; 16416516239fSHong Zhang PetscErrorCode ierr; 16426516239fSHong Zhang const PetscInt *r,*ic; 16436516239fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 16446516239fSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 16456516239fSHong Zhang PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 16466516239fSHong Zhang PetscInt i,levels,diagonal_fill; 16476516239fSHong Zhang PetscTruth col_identity,row_identity; 16486516239fSHong Zhang PetscReal f; 16496516239fSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 16506516239fSHong Zhang PetscBT lnkbt; 16516516239fSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 16526516239fSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 16536516239fSHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 16546516239fSHong Zhang PetscTruth missing; 1655b4f5729fSShri Abhyankar PetscTruth newdatastruct=PETSC_FALSE; 16566516239fSHong Zhang 16576516239fSHong Zhang PetscFunctionBegin; 16586516239fSHong Zhang ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 16596516239fSHong Zhang if(newdatastruct){ 16606516239fSHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr); 16616516239fSHong Zhang PetscFunctionReturn(0); 16626516239fSHong Zhang } 16636516239fSHong Zhang 16646516239fSHong 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); 16656516239fSHong Zhang f = info->fill; 16666516239fSHong Zhang levels = (PetscInt)info->levels; 16676516239fSHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 16686516239fSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 16696516239fSHong Zhang 16706516239fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 16716516239fSHong Zhang ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 16726516239fSHong Zhang if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */ 1673f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr); 1674c9c72f8fSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1675c9c72f8fSHong Zhang 1676719d5645SBarry Smith fact->factor = MAT_FACTOR_ILU; 1677719d5645SBarry Smith (fact)->info.factor_mallocs = 0; 1678719d5645SBarry Smith (fact)->info.fill_ratio_given = info->fill; 1679719d5645SBarry Smith (fact)->info.fill_ratio_needed = 1.0; 1680719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 16818ef3462fSBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 168209f38230SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 168308480c60SBarry Smith b->row = isrow; 168408480c60SBarry Smith b->col = iscol; 168582bf6240SBarry Smith b->icol = isicol; 1686719d5645SBarry Smith ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1687c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1688c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 16894108e4d5SBarry Smith ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); 16903a40ed3dSBarry Smith PetscFunctionReturn(0); 169108480c60SBarry Smith } 169208480c60SBarry Smith 1693898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1694898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 16959e25ed09SBarry Smith 1696*a6df321fSShri Abhyankar /* get new row and diagonal pointers */ 1697*a6df321fSShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt,&bi,n+1,PetscInt,&bdiag);CHKERRQ(ierr); 1698*a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 16996cf997b0SBarry Smith 17000e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr); 17015a8e39fbSHong Zhang 17025a8e39fbSHong Zhang /* create a linked list for storing column indices of the active row */ 17035a8e39fbSHong Zhang nlnk = n + 1; 17045a8e39fbSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 17055a8e39fbSHong Zhang 17065a8e39fbSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 1707a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 17085a8e39fbSHong Zhang current_space = free_space; 1709a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 17105a8e39fbSHong Zhang current_space_lvl = free_space_lvl; 17115a8e39fbSHong Zhang 17125a8e39fbSHong Zhang for (i=0; i<n; i++) { 17135a8e39fbSHong Zhang nzi = 0; 17146cf997b0SBarry Smith /* copy current row into linked list */ 17155a8e39fbSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 17163b4a8b6dSBarry 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); 17175a8e39fbSHong Zhang cols = aj + ai[r[i]]; 17185a8e39fbSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 17195a8e39fbSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 17205a8e39fbSHong Zhang nzi += nlnk; 17216cf997b0SBarry Smith 17226cf997b0SBarry Smith /* make sure diagonal entry is included */ 17235a8e39fbSHong Zhang if (diagonal_fill && lnk[i] == -1) { 17246cf997b0SBarry Smith fm = n; 17255a8e39fbSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 17265a8e39fbSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 17275a8e39fbSHong Zhang lnk[fm] = i; 17285a8e39fbSHong Zhang lnk_lvl[i] = 0; 17295a8e39fbSHong Zhang nzi++; dcount++; 17306cf997b0SBarry Smith } 17316cf997b0SBarry Smith 17325a8e39fbSHong Zhang /* add pivot rows into the active row */ 17335a8e39fbSHong Zhang nzbd = 0; 17345a8e39fbSHong Zhang prow = lnk[n]; 17355a8e39fbSHong Zhang while (prow < i) { 17365a8e39fbSHong Zhang nnz = bdiag[prow]; 17375a8e39fbSHong Zhang cols = bj_ptr[prow] + nnz + 1; 17385a8e39fbSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 17395a8e39fbSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 17405a8e39fbSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 17415a8e39fbSHong Zhang nzi += nlnk; 17425a8e39fbSHong Zhang prow = lnk[prow]; 17435a8e39fbSHong Zhang nzbd++; 174456beaf04SBarry Smith } 17455a8e39fbSHong Zhang bdiag[i] = nzbd; 17465a8e39fbSHong Zhang bi[i+1] = bi[i] + nzi; 1747ecf371e4SBarry Smith 17485a8e39fbSHong Zhang /* if free space is not available, make more free space */ 17495a8e39fbSHong Zhang if (current_space->local_remaining<nzi) { 17505a8e39fbSHong Zhang nnz = nzi*(n - i); /* estimated and max additional space needed */ 1751a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1752a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 17535a8e39fbSHong Zhang reallocs++; 17545a8e39fbSHong Zhang } 1755ecf371e4SBarry Smith 17565a8e39fbSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 17575a8e39fbSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 17585a8e39fbSHong Zhang bj_ptr[i] = current_space->array; 17595a8e39fbSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 17605a8e39fbSHong Zhang 17615a8e39fbSHong Zhang /* make sure the active row i has diagonal entry */ 17625a8e39fbSHong Zhang if (*(bj_ptr[i]+bdiag[i]) != i) { 176377431f27SBarry Smith SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 1764d7ee0231SBarry Smith try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 17656cf997b0SBarry Smith } 17665a8e39fbSHong Zhang 17675a8e39fbSHong Zhang current_space->array += nzi; 17685a8e39fbSHong Zhang current_space->local_used += nzi; 17695a8e39fbSHong Zhang current_space->local_remaining -= nzi; 17705a8e39fbSHong Zhang current_space_lvl->array += nzi; 17715a8e39fbSHong Zhang current_space_lvl->local_used += nzi; 17725a8e39fbSHong Zhang current_space_lvl->local_remaining -= nzi; 17739e25ed09SBarry Smith } 17745a8e39fbSHong Zhang 1775898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1776898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 17775a8e39fbSHong Zhang 17785a8e39fbSHong Zhang /* destroy list of free space and other temporary arrays */ 17795a8e39fbSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 17806516239fSHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */ 17815a8e39fbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1782a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 17830e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 17849e25ed09SBarry Smith 17856cf91177SBarry Smith #if defined(PETSC_USE_INFO) 1786f64065bbSBarry Smith { 17875a8e39fbSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1788ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 1789ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1790ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 1791ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1792335d9088SBarry Smith if (diagonal_fill) { 1793ae15b995SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 1794335d9088SBarry Smith } 1795f64065bbSBarry Smith } 179663ba0a88SBarry Smith #endif 1797bbb6d6a8SBarry Smith 17989e25ed09SBarry Smith /* put together the new matrix */ 179907dbf95fSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1800719d5645SBarry Smith ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 1801719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 1802e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1803e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 18047c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 18050e83c824SBarry Smith ierr = PetscMalloc(bi[n]*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 18065a8e39fbSHong Zhang b->j = bj; 18075a8e39fbSHong Zhang b->i = bi; 18085a8e39fbSHong Zhang for (i=0; i<n; i++) bdiag[i] += bi[i]; 18095a8e39fbSHong Zhang b->diag = bdiag; 1810416022c9SBarry Smith b->ilen = 0; 1811416022c9SBarry Smith b->imax = 0; 1812416022c9SBarry Smith b->row = isrow; 1813416022c9SBarry Smith b->col = iscol; 1814c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1815c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 181682bf6240SBarry Smith b->icol = isicol; 181787828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1818416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 18195a8e39fbSHong Zhang Allocate bdiag, solve_work, new a, new j */ 1820719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 18215a8e39fbSHong Zhang b->maxnz = b->nz = bi[n] ; 1822719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 1823719d5645SBarry Smith (fact)->info.fill_ratio_given = f; 1824719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1825719d5645SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 18264108e4d5SBarry Smith ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); 18273a40ed3dSBarry Smith PetscFunctionReturn(0); 18289e25ed09SBarry Smith } 1829d5d45c9bSBarry Smith 1830a6175056SHong Zhang #undef __FUNCT__ 18315f44c854SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_newdatastruct" 18325f44c854SHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info) 18335f44c854SHong Zhang { 18345f44c854SHong Zhang Mat C = B; 18355f44c854SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 18365f44c854SHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 18375f44c854SHong Zhang IS ip=b->row,iip = b->icol; 18385f44c854SHong Zhang PetscErrorCode ierr; 18395f44c854SHong Zhang const PetscInt *rip,*riip; 18405f44c854SHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp; 18415f44c854SHong Zhang PetscInt *ai=a->i,*aj=a->j; 18425f44c854SHong Zhang PetscInt k,jmin,jmax,*c2r,*il,col,nexti,ili,nz; 18435f44c854SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 18445f44c854SHong Zhang PetscReal zeropivot,shiftnz; 18455f44c854SHong Zhang PetscReal shiftpd; 18465f44c854SHong Zhang PetscTruth perm_identity; 18475f44c854SHong Zhang 18485f44c854SHong Zhang PetscFunctionBegin; 18495f44c854SHong Zhang 18505f44c854SHong Zhang shiftnz = info->shiftnz; 18515f44c854SHong Zhang shiftpd = info->shiftpd; 18525f44c854SHong Zhang zeropivot = info->zeropivot; 18535f44c854SHong Zhang 18545f44c854SHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 18555f44c854SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 18565f44c854SHong Zhang 18575f44c854SHong Zhang /* allocate working arrays 18585f44c854SHong Zhang c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col 18595f44c854SHong 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 18605f44c854SHong Zhang */ 18610e83c824SBarry Smith ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&c2r);CHKERRQ(ierr); 18625f44c854SHong Zhang 18635f44c854SHong Zhang for (i=0; i<mbs; i++) { 18645f44c854SHong Zhang c2r[i] = mbs; 18655f44c854SHong Zhang } 18665f44c854SHong Zhang il[0] = 0; 18675f44c854SHong Zhang 18685f44c854SHong Zhang for (k = 0; k<mbs; k++){ 18695f44c854SHong Zhang /* zero rtmp */ 18705f44c854SHong Zhang nz = bi[k+1] - bi[k]; 18715f44c854SHong Zhang bjtmp = bj + bi[k]; 18725f44c854SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 18735f44c854SHong Zhang 18745f44c854SHong Zhang /* load in initial unfactored row */ 18755f44c854SHong Zhang bval = ba + bi[k]; 18765f44c854SHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 18775f44c854SHong Zhang for (j = jmin; j < jmax; j++){ 18785f44c854SHong Zhang col = riip[aj[j]]; 18795f44c854SHong Zhang if (col >= k){ /* only take upper triangular entry */ 18805f44c854SHong Zhang rtmp[col] = aa[j]; 18815f44c854SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 18825f44c854SHong Zhang } 18835f44c854SHong Zhang } 18845f44c854SHong Zhang /* shift the diagonal of the matrix */ 18855f44c854SHong Zhang 18865f44c854SHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 18875f44c854SHong Zhang dk = rtmp[k]; 18885f44c854SHong Zhang i = c2r[k]; /* first row to be added to k_th row */ 18895f44c854SHong Zhang 18905f44c854SHong Zhang while (i < k){ 18915f44c854SHong Zhang nexti = c2r[i]; /* next row to be added to k_th row */ 18925f44c854SHong Zhang 18935f44c854SHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 18945f44c854SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 18955f44c854SHong Zhang uikdi = - ba[ili]*ba[bdiag[i]]; /* diagonal(k) */ 18965f44c854SHong Zhang dk += uikdi*ba[ili]; /* update diag[k] */ 18975f44c854SHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 18985f44c854SHong Zhang 18995f44c854SHong Zhang /* add multiple of row i to k-th row */ 19005f44c854SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 19015f44c854SHong Zhang if (jmin < jmax){ 19025f44c854SHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 19035f44c854SHong Zhang /* update il and c2r for row i */ 19045f44c854SHong Zhang il[i] = jmin; 19055f44c854SHong Zhang j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i; 19065f44c854SHong Zhang } 19075f44c854SHong Zhang i = nexti; 19085f44c854SHong Zhang } 19095f44c854SHong Zhang 19105f44c854SHong Zhang /* copy data into U(k,:) */ 19115f44c854SHong Zhang ba[bdiag[k]] = 1.0/dk; /* U(k,k) */ 19125f44c854SHong Zhang jmin = bi[k]; jmax = bi[k+1]-1; 19135f44c854SHong Zhang if (jmin < jmax) { 19145f44c854SHong Zhang for (j=jmin; j<jmax; j++){ 19155f44c854SHong Zhang col = bj[j]; ba[j] = rtmp[col]; 19165f44c854SHong Zhang } 19175f44c854SHong Zhang /* add the k-th row into il and c2r */ 19185f44c854SHong Zhang il[k] = jmin; 19195f44c854SHong Zhang i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k; 19205f44c854SHong Zhang } 19215f44c854SHong Zhang } 19225f44c854SHong Zhang 19230e83c824SBarry Smith ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr); 19245f44c854SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 19255f44c854SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 19265f44c854SHong Zhang 19275f44c854SHong Zhang ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 19285f44c854SHong Zhang if (perm_identity){ 19295f44c854SHong Zhang (B)->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering_newdatastruct; 19305f44c854SHong Zhang (B)->ops->solvetranspose = 0; 19315f44c854SHong Zhang (B)->ops->forwardsolve = 0; 19325f44c854SHong Zhang (B)->ops->backwardsolve = 0; 19335f44c854SHong Zhang } else { 1934783ef271SHong Zhang (B)->ops->solve = MatSolve_SeqSBAIJ_1_newdatastruct; 19355f44c854SHong Zhang (B)->ops->solvetranspose = 0; 19365f44c854SHong Zhang (B)->ops->forwardsolve = 0; 19375f44c854SHong Zhang (B)->ops->backwardsolve = 0; 19385f44c854SHong Zhang } 19395f44c854SHong Zhang 19405f44c854SHong Zhang C->assembled = PETSC_TRUE; 19415f44c854SHong Zhang C->preallocated = PETSC_TRUE; 19425f44c854SHong Zhang ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 19435f44c854SHong Zhang PetscFunctionReturn(0); 19445f44c854SHong Zhang } 19455f44c854SHong Zhang 19465f44c854SHong Zhang #undef __FUNCT__ 1947a6175056SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ" 19480481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 1949a6175056SHong Zhang { 1950719d5645SBarry Smith Mat C = B; 19510968510aSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 19522ed133dbSHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 19539bfd6278SHong Zhang IS ip=b->row,iip = b->icol; 19542ed133dbSHong Zhang PetscErrorCode ierr; 19555d0c19d7SBarry Smith const PetscInt *rip,*riip; 19565d0c19d7SBarry Smith PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol; 19572ed133dbSHong Zhang PetscInt *ai=a->i,*aj=a->j; 19582ed133dbSHong Zhang PetscInt k,jmin,jmax,*jl,*il,col,nexti,ili,nz; 1959622e2028SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 1960540703acSHong Zhang PetscReal zeropivot,rs,shiftnz; 1961fbf22428SSatish Balay PetscReal shiftpd; 1962540703acSHong Zhang ChShift_Ctx sctx; 1963540703acSHong Zhang PetscInt newshift; 1964db4efbfdSBarry Smith PetscTruth perm_identity; 1965d5d45c9bSBarry Smith 1966a6175056SHong Zhang PetscFunctionBegin; 1967540703acSHong Zhang shiftnz = info->shiftnz; 1968540703acSHong Zhang shiftpd = info->shiftpd; 1969ee45ca4aSHong Zhang zeropivot = info->zeropivot; 1970ee45ca4aSHong Zhang 19712ed133dbSHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 19729bfd6278SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 19732ed133dbSHong Zhang 19742ed133dbSHong Zhang /* initialization */ 19750e83c824SBarry Smith ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&jl);CHKERRQ(ierr); 1976540703acSHong Zhang sctx.shift_amount = 0; 1977540703acSHong Zhang sctx.nshift = 0; 19782ed133dbSHong Zhang do { 1979540703acSHong Zhang sctx.chshift = PETSC_FALSE; 19802ed133dbSHong Zhang for (i=0; i<mbs; i++) { 19812ed133dbSHong Zhang rtmp[i] = 0.0; jl[i] = mbs; il[0] = 0; 1982a6175056SHong Zhang } 19832ed133dbSHong Zhang 19842ed133dbSHong Zhang for (k = 0; k<mbs; k++){ 1985c05c3958SHong Zhang bval = ba + bi[k]; 19862ed133dbSHong Zhang /* initialize k-th row by the perm[k]-th row of A */ 19872ed133dbSHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 19882ed133dbSHong Zhang for (j = jmin; j < jmax; j++){ 19899bfd6278SHong Zhang col = riip[aj[j]]; 19902ed133dbSHong Zhang if (col >= k){ /* only take upper triangular entry */ 19912ed133dbSHong Zhang rtmp[col] = aa[j]; 1992c05c3958SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 19932ed133dbSHong Zhang } 19942ed133dbSHong Zhang } 199539e3d630SHong Zhang /* shift the diagonal of the matrix */ 1996540703acSHong Zhang if (sctx.nshift) rtmp[k] += sctx.shift_amount; 19972ed133dbSHong Zhang 19982ed133dbSHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 19992ed133dbSHong Zhang dk = rtmp[k]; 20002ed133dbSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 20012ed133dbSHong Zhang 20022ed133dbSHong Zhang while (i < k){ 20032ed133dbSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 20042ed133dbSHong Zhang 20052ed133dbSHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 20062ed133dbSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 20072ed133dbSHong Zhang uikdi = - ba[ili]*ba[bi[i]]; /* diagonal(k) */ 20082ed133dbSHong Zhang dk += uikdi*ba[ili]; 20092ed133dbSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 20102ed133dbSHong Zhang 20112ed133dbSHong Zhang /* add multiple of row i to k-th row */ 20122ed133dbSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 20132ed133dbSHong Zhang if (jmin < jmax){ 20142ed133dbSHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 20152ed133dbSHong Zhang /* update il and jl for row i */ 20162ed133dbSHong Zhang il[i] = jmin; 20172ed133dbSHong Zhang j = bj[jmin]; jl[i] = jl[j]; jl[j] = i; 20182ed133dbSHong Zhang } 20192ed133dbSHong Zhang i = nexti; 20202ed133dbSHong Zhang } 20212ed133dbSHong Zhang 20220697b6caSHong Zhang /* shift the diagonals when zero pivot is detected */ 20230697b6caSHong Zhang /* compute rs=sum of abs(off-diagonal) */ 20240697b6caSHong Zhang rs = 0.0; 20253c9e8598SHong Zhang jmin = bi[k]+1; 20263c9e8598SHong Zhang nz = bi[k+1] - jmin; 20273c9e8598SHong Zhang bcol = bj + jmin; 202804fbf559SBarry Smith for (j=0; j<nz; j++) { 202904fbf559SBarry Smith rs += PetscAbsScalar(rtmp[bcol[j]]); 20303c9e8598SHong Zhang } 2031540703acSHong Zhang 2032540703acSHong Zhang sctx.rs = rs; 2033540703acSHong Zhang sctx.pv = dk; 20345b5bc046SBarry Smith ierr = MatCholeskyCheckShift_inline(info,sctx,k,newshift);CHKERRQ(ierr); 2035117f1891Stmunson 2036117f1891Stmunson if (newshift == 1) { 2037117f1891Stmunson if (!sctx.shift_amount) { 2038117f1891Stmunson sctx.shift_amount = 1e-5; 2039117f1891Stmunson } 2040117f1891Stmunson break; 2041117f1891Stmunson } 20423c9e8598SHong Zhang 20433c9e8598SHong Zhang /* copy data into U(k,:) */ 204439e3d630SHong Zhang ba[bi[k]] = 1.0/dk; /* U(k,k) */ 204539e3d630SHong Zhang jmin = bi[k]+1; jmax = bi[k+1]; 204639e3d630SHong Zhang if (jmin < jmax) { 204739e3d630SHong Zhang for (j=jmin; j<jmax; j++){ 204839e3d630SHong Zhang col = bj[j]; ba[j] = rtmp[col]; rtmp[col] = 0.0; 20493c9e8598SHong Zhang } 205039e3d630SHong Zhang /* add the k-th row into il and jl */ 20513c9e8598SHong Zhang il[k] = jmin; 20523c9e8598SHong Zhang i = bj[jmin]; jl[k] = jl[i]; jl[i] = k; 20533c9e8598SHong Zhang } 20543c9e8598SHong Zhang } 2055540703acSHong Zhang } while (sctx.chshift); 20560e83c824SBarry Smith ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr); 205739e3d630SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 20589bfd6278SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 2059db4efbfdSBarry Smith 2060db4efbfdSBarry Smith ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 2061db4efbfdSBarry Smith if (perm_identity){ 2062719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 2063719d5645SBarry Smith (B)->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 2064719d5645SBarry Smith (B)->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering; 2065719d5645SBarry Smith (B)->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering; 2066db4efbfdSBarry Smith } else { 2067719d5645SBarry Smith (B)->ops->solve = MatSolve_SeqSBAIJ_1; 2068719d5645SBarry Smith (B)->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 2069719d5645SBarry Smith (B)->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1; 2070719d5645SBarry Smith (B)->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1; 2071db4efbfdSBarry Smith } 2072db4efbfdSBarry Smith 20733c9e8598SHong Zhang C->assembled = PETSC_TRUE; 20743c9e8598SHong Zhang C->preallocated = PETSC_TRUE; 2075d0f46423SBarry Smith ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 2076540703acSHong Zhang if (sctx.nshift){ 2077540703acSHong Zhang if (shiftnz) { 20781e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2079540703acSHong Zhang } else if (shiftpd) { 20801e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 20810697b6caSHong Zhang } 20823c9e8598SHong Zhang } 20833c9e8598SHong Zhang PetscFunctionReturn(0); 20843c9e8598SHong Zhang } 2085a6175056SHong Zhang 20868a76ed4fSHong Zhang /* 20878a76ed4fSHong Zhang icc() under revised new data structure. 20888a76ed4fSHong Zhang Factored arrays bj and ba are stored as 20898a76ed4fSHong Zhang U(0,:),...,U(i,:),U(n-1,:) 20908a76ed4fSHong Zhang 20918a76ed4fSHong Zhang ui=fact->i is an array of size n+1, in which 20928a76ed4fSHong Zhang ui+ 20938a76ed4fSHong Zhang ui[i]: points to 1st entry of U(i,:),i=0,...,n-1 20948a76ed4fSHong Zhang ui[n]: points to U(n-1,n-1)+1 20958a76ed4fSHong Zhang 20968a76ed4fSHong Zhang udiag=fact->diag is an array of size n,in which 20978a76ed4fSHong Zhang udiag[i]: points to diagonal of U(i,:), i=0,...,n-1 20988a76ed4fSHong Zhang 20998a76ed4fSHong Zhang U(i,:) contains udiag[i] as its last entry, i.e., 21008a76ed4fSHong Zhang U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 21018a76ed4fSHong Zhang */ 21028a76ed4fSHong Zhang 2103a6175056SHong Zhang #undef __FUNCT__ 21045f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_newdatastruct" 21055f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2106a6175056SHong Zhang { 21070968510aSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2108ed59401aSHong Zhang Mat_SeqSBAIJ *b; 2109dfbe8321SBarry Smith PetscErrorCode ierr; 211058ebbce7SBarry Smith PetscTruth perm_identity,missing; 21115f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 21125d0c19d7SBarry Smith const PetscInt *rip,*riip; 2113ed59401aSHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 211458ebbce7SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 21155a8e39fbSHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 2116ed59401aSHong Zhang PetscReal fill=info->fill,levels=info->levels; 2117a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 2118a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 21197a48dd6fSHong Zhang PetscBT lnkbt; 2120b635d36bSHong Zhang IS iperm; 2121a6175056SHong Zhang 2122a6175056SHong Zhang PetscFunctionBegin; 2123d0f46423SBarry 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); 212458ebbce7SBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 212558ebbce7SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 2126653a6975SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 2127b635d36bSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 21287a48dd6fSHong Zhang 212939e3d630SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 21305f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 213139e3d630SHong Zhang ui[0] = 0; 213239e3d630SHong Zhang 21338a76ed4fSHong Zhang /* ICC(0) without matrix ordering: simply rearrange column indices */ 2134622e2028SHong Zhang if (!levels && perm_identity) { 213558ebbce7SBarry Smith 2136ed59401aSHong Zhang for (i=0; i<am; i++) { 213739e3d630SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 21385f44c854SHong Zhang udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */ 21395f44c854SHong Zhang } 21405f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 21415f44c854SHong Zhang cols = uj; 21425f44c854SHong Zhang for (i=0; i<am; i++) { 21435f44c854SHong Zhang aj = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */ 21445f44c854SHong Zhang ncols = ui[i+1] - ui[i] - 1; 21455f44c854SHong Zhang for (j=0; j<ncols; j++) *cols++ = aj[j]; 21465f44c854SHong Zhang *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */ 21475f44c854SHong Zhang } 21485f44c854SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 21498a76ed4fSHong Zhang if (levels) SETERRQ(PETSC_ERR_ARG_WRONG,"not done yet"); 21505f44c854SHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 21515f44c854SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 21525f44c854SHong Zhang 21535f44c854SHong Zhang /* initialization */ 21545f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 21555f44c854SHong Zhang 21565f44c854SHong Zhang /* jl: linked list for storing indices of the pivot rows 21575f44c854SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 21580e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr); 21595f44c854SHong Zhang for (i=0; i<am; i++){ 21605f44c854SHong Zhang jl[i] = am; il[i] = 0; 21615f44c854SHong Zhang } 21625f44c854SHong Zhang 21635f44c854SHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 21645f44c854SHong Zhang nlnk = am + 1; 21655f44c854SHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 21665f44c854SHong Zhang 21675f44c854SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 21685f44c854SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 21695f44c854SHong Zhang current_space = free_space; 21705f44c854SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 21715f44c854SHong Zhang current_space_lvl = free_space_lvl; 21725f44c854SHong Zhang 21735f44c854SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 21745f44c854SHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 21755f44c854SHong Zhang nzk = 0; 21765f44c854SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 21775f44c854SHong 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); 21785f44c854SHong Zhang ncols_upper = 0; 21795f44c854SHong Zhang for (j=0; j<ncols; j++){ 21805f44c854SHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 21815f44c854SHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 21825f44c854SHong Zhang ajtmp[ncols_upper] = i; 21835f44c854SHong Zhang ncols_upper++; 21845f44c854SHong Zhang } 21855f44c854SHong Zhang } 21865f44c854SHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 21875f44c854SHong Zhang nzk += nlnk; 21885f44c854SHong Zhang 21895f44c854SHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 21905f44c854SHong Zhang prow = jl[k]; /* 1st pivot row */ 21915f44c854SHong Zhang 21925f44c854SHong Zhang while (prow < k){ 21935f44c854SHong Zhang nextprow = jl[prow]; 21945f44c854SHong Zhang 21955f44c854SHong Zhang /* merge prow into k-th row */ 21965f44c854SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 21975f44c854SHong Zhang jmax = ui[prow+1]; 21985f44c854SHong Zhang ncols = jmax-jmin; 21995f44c854SHong Zhang i = jmin - ui[prow]; 22005f44c854SHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 22015f44c854SHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 22025f44c854SHong Zhang j = *(uj - 1); 22035f44c854SHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 22045f44c854SHong Zhang nzk += nlnk; 22055f44c854SHong Zhang 22065f44c854SHong Zhang /* update il and jl for prow */ 22075f44c854SHong Zhang if (jmin < jmax){ 22085f44c854SHong Zhang il[prow] = jmin; 22095f44c854SHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 22105f44c854SHong Zhang } 22115f44c854SHong Zhang prow = nextprow; 22125f44c854SHong Zhang } 22135f44c854SHong Zhang 22145f44c854SHong Zhang /* if free space is not available, make more free space */ 22155f44c854SHong Zhang if (current_space->local_remaining<nzk) { 22165f44c854SHong Zhang i = am - k + 1; /* num of unfactored rows */ 22175f44c854SHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 22185f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 22195f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 22205f44c854SHong Zhang reallocs++; 22215f44c854SHong Zhang } 22225f44c854SHong Zhang 22235f44c854SHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 22245f44c854SHong Zhang if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 22255f44c854SHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 22265f44c854SHong Zhang 22275f44c854SHong Zhang /* add the k-th row into il and jl */ 22285f44c854SHong Zhang if (nzk > 1){ 22295f44c854SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 22305f44c854SHong Zhang jl[k] = jl[i]; jl[i] = k; 22315f44c854SHong Zhang il[k] = ui[k] + 1; 22325f44c854SHong Zhang } 22335f44c854SHong Zhang uj_ptr[k] = current_space->array; 22345f44c854SHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 22355f44c854SHong Zhang 22365f44c854SHong Zhang current_space->array += nzk; 22375f44c854SHong Zhang current_space->local_used += nzk; 22385f44c854SHong Zhang current_space->local_remaining -= nzk; 22395f44c854SHong Zhang 22405f44c854SHong Zhang current_space_lvl->array += nzk; 22415f44c854SHong Zhang current_space_lvl->local_used += nzk; 22425f44c854SHong Zhang current_space_lvl->local_remaining -= nzk; 22435f44c854SHong Zhang 22445f44c854SHong Zhang ui[k+1] = ui[k] + nzk; 22455f44c854SHong Zhang } 22465f44c854SHong Zhang 22475f44c854SHong Zhang #if defined(PETSC_USE_INFO) 22485f44c854SHong Zhang if (ai[am] != 0) { 22495f44c854SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 22505f44c854SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 22515f44c854SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 22525f44c854SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 22535f44c854SHong Zhang } else { 22545f44c854SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 22555f44c854SHong Zhang } 22565f44c854SHong Zhang #endif 22575f44c854SHong Zhang 22585f44c854SHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 22595f44c854SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 22600e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 22615f44c854SHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 22625f44c854SHong Zhang 22635f44c854SHong Zhang /* destroy list of free space and other temporary array(s) */ 22645f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2265783ef271SHong Zhang ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); 22665f44c854SHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 22675f44c854SHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 22685f44c854SHong Zhang 22695f44c854SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 22705f44c854SHong Zhang 22715f44c854SHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 22725f44c854SHong Zhang 22735f44c854SHong Zhang b = (Mat_SeqSBAIJ*)(fact)->data; 22745f44c854SHong Zhang b->singlemalloc = PETSC_FALSE; 22755f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 22765f44c854SHong Zhang b->j = uj; 22775f44c854SHong Zhang b->i = ui; 22785f44c854SHong Zhang b->diag = udiag; 22797f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 22805f44c854SHong Zhang b->ilen = 0; 22815f44c854SHong Zhang b->imax = 0; 22825f44c854SHong Zhang b->row = perm; 22835f44c854SHong Zhang b->col = perm; 22845f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 22855f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 22865f44c854SHong Zhang b->icol = iperm; 22875f44c854SHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 22885f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 22895f44c854SHong Zhang ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 22905f44c854SHong Zhang b->maxnz = b->nz = ui[am]; 22915f44c854SHong Zhang b->free_a = PETSC_TRUE; 22925f44c854SHong Zhang b->free_ij = PETSC_TRUE; 22935f44c854SHong Zhang 22945f44c854SHong Zhang (fact)->info.factor_mallocs = reallocs; 22955f44c854SHong Zhang (fact)->info.fill_ratio_given = fill; 22965f44c854SHong Zhang if (ai[am] != 0) { 22975f44c854SHong Zhang (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 22985f44c854SHong Zhang } else { 22995f44c854SHong Zhang (fact)->info.fill_ratio_needed = 0.0; 23005f44c854SHong Zhang } 23015f44c854SHong Zhang (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_newdatastruct; 23025f44c854SHong Zhang PetscFunctionReturn(0); 23035f44c854SHong Zhang } 23045f44c854SHong Zhang 23055f44c854SHong Zhang #undef __FUNCT__ 23065f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ" 23075f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 23085f44c854SHong Zhang { 23095f44c854SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 23105f44c854SHong Zhang Mat_SeqSBAIJ *b; 23115f44c854SHong Zhang PetscErrorCode ierr; 23125f44c854SHong Zhang PetscTruth perm_identity,missing; 23135f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 23145f44c854SHong Zhang const PetscInt *rip,*riip; 23155f44c854SHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 23165f44c854SHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 23175f44c854SHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 23185f44c854SHong Zhang PetscReal fill=info->fill,levels=info->levels; 23195f44c854SHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 23205f44c854SHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 23215f44c854SHong Zhang PetscBT lnkbt; 23225f44c854SHong Zhang IS iperm; 23235f44c854SHong Zhang PetscTruth newdatastruct=PETSC_FALSE; 23245f44c854SHong Zhang 23255f44c854SHong Zhang PetscFunctionBegin; 23265f44c854SHong Zhang ierr = PetscOptionsGetTruth(PETSC_NULL,"-icc_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr); 23275f44c854SHong Zhang if(newdatastruct){ 23285f44c854SHong Zhang ierr = MatICCFactorSymbolic_SeqAIJ_newdatastruct(fact,A,perm,info);CHKERRQ(ierr); 23295f44c854SHong Zhang PetscFunctionReturn(0); 23305f44c854SHong Zhang } 23315f44c854SHong Zhang 23325f44c854SHong 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); 23335f44c854SHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 23345f44c854SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 23355f44c854SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 23365f44c854SHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 23375f44c854SHong Zhang 23385f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 23395f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 23405f44c854SHong Zhang ui[0] = 0; 23415f44c854SHong Zhang 23425f44c854SHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 23435f44c854SHong Zhang if (!levels && perm_identity) { 23445f44c854SHong Zhang 23455f44c854SHong Zhang for (i=0; i<am; i++) { 23465f44c854SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 23475f44c854SHong Zhang udiag[i] = ui[i]; 2348ed59401aSHong Zhang } 234939e3d630SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 235039e3d630SHong Zhang cols = uj; 2351ed59401aSHong Zhang for (i=0; i<am; i++) { 2352ed59401aSHong Zhang aj = a->j + a->diag[i]; 235339e3d630SHong Zhang ncols = ui[i+1] - ui[i]; 235439e3d630SHong Zhang for (j=0; j<ncols; j++) *cols++ = *aj++; 2355ed59401aSHong Zhang } 235639e3d630SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 2357b635d36bSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 2358b635d36bSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 2359b635d36bSHong Zhang 23607a48dd6fSHong Zhang /* initialization */ 23615a8e39fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 23627a48dd6fSHong Zhang 23637a48dd6fSHong Zhang /* jl: linked list for storing indices of the pivot rows 23647a48dd6fSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 23650e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr); 23667a48dd6fSHong Zhang for (i=0; i<am; i++){ 23677a48dd6fSHong Zhang jl[i] = am; il[i] = 0; 23687a48dd6fSHong Zhang } 23697a48dd6fSHong Zhang 23707a48dd6fSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 23717a48dd6fSHong Zhang nlnk = am + 1; 23722ed133dbSHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 23737a48dd6fSHong Zhang 23747a48dd6fSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2375a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 23767a48dd6fSHong Zhang current_space = free_space; 2377a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 23787a48dd6fSHong Zhang current_space_lvl = free_space_lvl; 23797a48dd6fSHong Zhang 23807a48dd6fSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 23817a48dd6fSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 23827a48dd6fSHong Zhang nzk = 0; 2383622e2028SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 23843b4a8b6dSBarry 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); 2385622e2028SHong Zhang ncols_upper = 0; 2386622e2028SHong Zhang for (j=0; j<ncols; j++){ 2387b635d36bSHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 2388b635d36bSHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 23895a8e39fbSHong Zhang ajtmp[ncols_upper] = i; 2390622e2028SHong Zhang ncols_upper++; 2391622e2028SHong Zhang } 2392622e2028SHong Zhang } 2393b635d36bSHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 23947a48dd6fSHong Zhang nzk += nlnk; 23957a48dd6fSHong Zhang 23967a48dd6fSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 23977a48dd6fSHong Zhang prow = jl[k]; /* 1st pivot row */ 23987a48dd6fSHong Zhang 23997a48dd6fSHong Zhang while (prow < k){ 24007a48dd6fSHong Zhang nextprow = jl[prow]; 24017a48dd6fSHong Zhang 24027a48dd6fSHong Zhang /* merge prow into k-th row */ 24037a48dd6fSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 24047a48dd6fSHong Zhang jmax = ui[prow+1]; 24057a48dd6fSHong Zhang ncols = jmax-jmin; 2406ed59401aSHong Zhang i = jmin - ui[prow]; 2407ed59401aSHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 24085a8e39fbSHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 24095a8e39fbSHong Zhang j = *(uj - 1); 24105a8e39fbSHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 24117a48dd6fSHong Zhang nzk += nlnk; 24127a48dd6fSHong Zhang 24137a48dd6fSHong Zhang /* update il and jl for prow */ 24147a48dd6fSHong Zhang if (jmin < jmax){ 24157a48dd6fSHong Zhang il[prow] = jmin; 24167a48dd6fSHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 24177a48dd6fSHong Zhang } 24187a48dd6fSHong Zhang prow = nextprow; 24197a48dd6fSHong Zhang } 24207a48dd6fSHong Zhang 24217a48dd6fSHong Zhang /* if free space is not available, make more free space */ 24227a48dd6fSHong Zhang if (current_space->local_remaining<nzk) { 24237a48dd6fSHong Zhang i = am - k + 1; /* num of unfactored rows */ 24247a48dd6fSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2425a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 2426a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 24277a48dd6fSHong Zhang reallocs++; 24287a48dd6fSHong Zhang } 24297a48dd6fSHong Zhang 24307a48dd6fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 2431b635d36bSHong Zhang if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 24322ed133dbSHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 24337a48dd6fSHong Zhang 24347a48dd6fSHong Zhang /* add the k-th row into il and jl */ 243539e3d630SHong Zhang if (nzk > 1){ 24367a48dd6fSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 24377a48dd6fSHong Zhang jl[k] = jl[i]; jl[i] = k; 24387a48dd6fSHong Zhang il[k] = ui[k] + 1; 24397a48dd6fSHong Zhang } 24407a48dd6fSHong Zhang uj_ptr[k] = current_space->array; 24417a48dd6fSHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 24427a48dd6fSHong Zhang 24437a48dd6fSHong Zhang current_space->array += nzk; 24447a48dd6fSHong Zhang current_space->local_used += nzk; 24457a48dd6fSHong Zhang current_space->local_remaining -= nzk; 24467a48dd6fSHong Zhang 24477a48dd6fSHong Zhang current_space_lvl->array += nzk; 24487a48dd6fSHong Zhang current_space_lvl->local_used += nzk; 24497a48dd6fSHong Zhang current_space_lvl->local_remaining -= nzk; 24507a48dd6fSHong Zhang 24517a48dd6fSHong Zhang ui[k+1] = ui[k] + nzk; 24527a48dd6fSHong Zhang } 24537a48dd6fSHong Zhang 24546cf91177SBarry Smith #if defined(PETSC_USE_INFO) 24557a48dd6fSHong Zhang if (ai[am] != 0) { 245639e3d630SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 2457ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2458ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2459ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 24607a48dd6fSHong Zhang } else { 2461ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 24627a48dd6fSHong Zhang } 246363ba0a88SBarry Smith #endif 24647a48dd6fSHong Zhang 24657a48dd6fSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 2466b635d36bSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 24670e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 24685a8e39fbSHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 24697a48dd6fSHong Zhang 24707a48dd6fSHong Zhang /* destroy list of free space and other temporary array(s) */ 24717a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2472a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 24732ed133dbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2474a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 24757a48dd6fSHong Zhang 247639e3d630SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 247739e3d630SHong Zhang 24787a48dd6fSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 24797a48dd6fSHong Zhang 2480719d5645SBarry Smith b = (Mat_SeqSBAIJ*)(fact)->data; 24817a48dd6fSHong Zhang b->singlemalloc = PETSC_FALSE; 24827a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 24837a48dd6fSHong Zhang b->j = uj; 24847a48dd6fSHong Zhang b->i = ui; 24855f44c854SHong Zhang b->diag = udiag; 24867f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 24877a48dd6fSHong Zhang b->ilen = 0; 24887a48dd6fSHong Zhang b->imax = 0; 24897a48dd6fSHong Zhang b->row = perm; 2490b635d36bSHong Zhang b->col = perm; 2491b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2492b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2493b635d36bSHong Zhang b->icol = iperm; 24947a48dd6fSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 24957a48dd6fSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2496719d5645SBarry Smith ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 24977a48dd6fSHong Zhang b->maxnz = b->nz = ui[am]; 2498e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2499e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 25007a48dd6fSHong Zhang 2501719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2502719d5645SBarry Smith (fact)->info.fill_ratio_given = fill; 25037a48dd6fSHong Zhang if (ai[am] != 0) { 2504719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 25057a48dd6fSHong Zhang } else { 2506719d5645SBarry Smith (fact)->info.fill_ratio_needed = 0.0; 25077a48dd6fSHong Zhang } 2508719d5645SBarry Smith (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 2509a6175056SHong Zhang PetscFunctionReturn(0); 2510a6175056SHong Zhang } 2511d5d45c9bSBarry Smith 2512f76d2b81SHong Zhang #undef __FUNCT__ 2513f76d2b81SHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ" 25140481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2515f76d2b81SHong Zhang { 2516f76d2b81SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 251710c27e3dSHong Zhang Mat_SeqSBAIJ *b; 2518dfbe8321SBarry Smith PetscErrorCode ierr; 2519f76d2b81SHong Zhang PetscTruth perm_identity; 252010c27e3dSHong Zhang PetscReal fill = info->fill; 25215d0c19d7SBarry Smith const PetscInt *rip,*riip; 25225d0c19d7SBarry Smith PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 252310c27e3dSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 25242ed133dbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr; 2525a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 252610c27e3dSHong Zhang PetscBT lnkbt; 25272ed133dbSHong Zhang IS iperm; 2528f76d2b81SHong Zhang 2529f76d2b81SHong Zhang PetscFunctionBegin; 2530d0f46423SBarry 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); 25312ed133dbSHong Zhang /* check whether perm is the identity mapping */ 2532f76d2b81SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 25332ed133dbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 25342ed133dbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 25359bfd6278SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 253610c27e3dSHong Zhang 253710c27e3dSHong Zhang /* initialization */ 25381393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 253910c27e3dSHong Zhang ui[0] = 0; 254010c27e3dSHong Zhang 254110c27e3dSHong Zhang /* jl: linked list for storing indices of the pivot rows 25421393bc97SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 25430e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr); 25441393bc97SHong Zhang for (i=0; i<am; i++){ 25451393bc97SHong Zhang jl[i] = am; il[i] = 0; 2546f76d2b81SHong Zhang } 254710c27e3dSHong Zhang 254810c27e3dSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 25491393bc97SHong Zhang nlnk = am + 1; 25501393bc97SHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 255110c27e3dSHong Zhang 25521393bc97SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2553a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 255410c27e3dSHong Zhang current_space = free_space; 255510c27e3dSHong Zhang 25561393bc97SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 255710c27e3dSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 255810c27e3dSHong Zhang nzk = 0; 25592ed133dbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 25603b4a8b6dSBarry 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); 25612ed133dbSHong Zhang ncols_upper = 0; 25622ed133dbSHong Zhang for (j=0; j<ncols; j++){ 25639bfd6278SHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 25642ed133dbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 25652ed133dbSHong Zhang cols[ncols_upper] = i; 25662ed133dbSHong Zhang ncols_upper++; 25672ed133dbSHong Zhang } 25682ed133dbSHong Zhang } 25691393bc97SHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 257010c27e3dSHong Zhang nzk += nlnk; 257110c27e3dSHong Zhang 257210c27e3dSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 257310c27e3dSHong Zhang prow = jl[k]; /* 1st pivot row */ 257410c27e3dSHong Zhang 257510c27e3dSHong Zhang while (prow < k){ 257610c27e3dSHong Zhang nextprow = jl[prow]; 257710c27e3dSHong Zhang /* merge prow into k-th row */ 25781393bc97SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 257910c27e3dSHong Zhang jmax = ui[prow+1]; 258010c27e3dSHong Zhang ncols = jmax-jmin; 25811393bc97SHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 25825a8e39fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 258310c27e3dSHong Zhang nzk += nlnk; 258410c27e3dSHong Zhang 258510c27e3dSHong Zhang /* update il and jl for prow */ 258610c27e3dSHong Zhang if (jmin < jmax){ 258710c27e3dSHong Zhang il[prow] = jmin; 25882ed133dbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 258910c27e3dSHong Zhang } 259010c27e3dSHong Zhang prow = nextprow; 259110c27e3dSHong Zhang } 259210c27e3dSHong Zhang 259310c27e3dSHong Zhang /* if free space is not available, make more free space */ 259410c27e3dSHong Zhang if (current_space->local_remaining<nzk) { 25951393bc97SHong Zhang i = am - k + 1; /* num of unfactored rows */ 259610c27e3dSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2597a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 259810c27e3dSHong Zhang reallocs++; 259910c27e3dSHong Zhang } 260010c27e3dSHong Zhang 260110c27e3dSHong Zhang /* copy data into free space, then initialize lnk */ 26021393bc97SHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 260310c27e3dSHong Zhang 260410c27e3dSHong Zhang /* add the k-th row into il and jl */ 260510c27e3dSHong Zhang if (nzk-1 > 0){ 26061393bc97SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 260710c27e3dSHong Zhang jl[k] = jl[i]; jl[i] = k; 260810c27e3dSHong Zhang il[k] = ui[k] + 1; 260910c27e3dSHong Zhang } 26102ed133dbSHong Zhang ui_ptr[k] = current_space->array; 261110c27e3dSHong Zhang current_space->array += nzk; 261210c27e3dSHong Zhang current_space->local_used += nzk; 261310c27e3dSHong Zhang current_space->local_remaining -= nzk; 261410c27e3dSHong Zhang 261510c27e3dSHong Zhang ui[k+1] = ui[k] + nzk; 261610c27e3dSHong Zhang } 261710c27e3dSHong Zhang 26186cf91177SBarry Smith #if defined(PETSC_USE_INFO) 26191393bc97SHong Zhang if (ai[am] != 0) { 26201393bc97SHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 2621ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2622ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2623ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 262410c27e3dSHong Zhang } else { 2625ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 262610c27e3dSHong Zhang } 262763ba0a88SBarry Smith #endif 262810c27e3dSHong Zhang 262910c27e3dSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 26309bfd6278SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 26310e83c824SBarry Smith ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 263210c27e3dSHong Zhang 263310c27e3dSHong Zhang /* destroy list of free space and other temporary array(s) */ 26341393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2635a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 263610c27e3dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 263710c27e3dSHong Zhang 263810c27e3dSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 263910c27e3dSHong Zhang 2640719d5645SBarry Smith b = (Mat_SeqSBAIJ*)(fact)->data; 264110c27e3dSHong Zhang b->singlemalloc = PETSC_FALSE; 2642e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2643e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 26441393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 264510c27e3dSHong Zhang b->j = uj; 264610c27e3dSHong Zhang b->i = ui; 264710c27e3dSHong Zhang b->diag = 0; 264810c27e3dSHong Zhang b->ilen = 0; 264910c27e3dSHong Zhang b->imax = 0; 265010c27e3dSHong Zhang b->row = perm; 26519bfd6278SHong Zhang b->col = perm; 26529bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 26539bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 26549bfd6278SHong Zhang b->icol = iperm; 265510c27e3dSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 26561393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2657719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 26581393bc97SHong Zhang b->maxnz = b->nz = ui[am]; 265910c27e3dSHong Zhang 2660719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2661719d5645SBarry Smith (fact)->info.fill_ratio_given = fill; 26621393bc97SHong Zhang if (ai[am] != 0) { 2663719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 266410c27e3dSHong Zhang } else { 2665719d5645SBarry Smith (fact)->info.fill_ratio_needed = 0.0; 266610c27e3dSHong Zhang } 2667719d5645SBarry Smith (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 2668f76d2b81SHong Zhang PetscFunctionReturn(0); 2669f76d2b81SHong Zhang } 2670599ef60dSHong Zhang 2671599ef60dSHong Zhang #undef __FUNCT__ 2672894a2336SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_newdatastruct" 2673894a2336SHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat A,Vec bb,Vec xx) 2674599ef60dSHong Zhang { 26751d098868SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 26761d098868SHong Zhang PetscErrorCode ierr; 26771d098868SHong Zhang PetscInt n = A->rmap->n; 26788fc3a347SHong Zhang const PetscInt *ai = a->i,*aj = a->j,*vi; 26791d098868SHong Zhang PetscScalar *x,sum; 26801d098868SHong Zhang const PetscScalar *b; 26811d098868SHong Zhang const MatScalar *aa = a->a,*v; 26828fc3a347SHong Zhang PetscInt i,nz; 26831d098868SHong Zhang 26841d098868SHong Zhang PetscFunctionBegin; 26851d098868SHong Zhang if (!n) PetscFunctionReturn(0); 26861d098868SHong Zhang 26871d098868SHong Zhang ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 26881d098868SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 26891d098868SHong Zhang 26901d098868SHong Zhang /* forward solve the lower triangular */ 26911d098868SHong Zhang x[0] = b[0]; 26921d098868SHong Zhang v = aa; 26931d098868SHong Zhang vi = aj; 26941d098868SHong Zhang for (i=1; i<n; i++) { 26951d098868SHong Zhang nz = ai[i+1] - ai[i]; 26961d098868SHong Zhang sum = b[i]; 2697003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 2698e0e43300SBarry Smith v += nz; 2699e0e43300SBarry Smith vi += nz; 27001d098868SHong Zhang x[i] = sum; 27011d098868SHong Zhang } 27021d098868SHong Zhang 27031d098868SHong Zhang /* backward solve the upper triangular */ 27048fc3a347SHong Zhang v = aa + ai[n+1]; 27058fc3a347SHong Zhang vi = aj + ai[n+1]; 27061d098868SHong Zhang for (i=n-1; i>=0; i--){ 27078fc3a347SHong Zhang nz = ai[2*n-i +1] - ai[2*n-i]-1; 27081d098868SHong Zhang sum = x[i]; 2709003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 2710e0e43300SBarry Smith v += nz; 27118fc3a347SHong Zhang vi += nz; vi++; 2712fe31b70dSHong Zhang x[i] = *v++ *sum; /* x[i]=aa[adiag[i]]*sum; v++; */ 27131d098868SHong Zhang } 27141d098868SHong Zhang 27151d098868SHong Zhang ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 27161d098868SHong Zhang ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 27171d098868SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 27181d098868SHong Zhang PetscFunctionReturn(0); 27191d098868SHong Zhang } 27201d098868SHong Zhang 27211d098868SHong Zhang #undef __FUNCT__ 2722813a1f2bSShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_newdatastruct_v2" 2723813a1f2bSShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct_v2(Mat A,Vec bb,Vec xx) 2724813a1f2bSShri Abhyankar { 2725813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2726813a1f2bSShri Abhyankar PetscErrorCode ierr; 2727813a1f2bSShri Abhyankar PetscInt n = A->rmap->n; 2728813a1f2bSShri Abhyankar const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag,*vi; 2729813a1f2bSShri Abhyankar PetscScalar *x,sum; 2730813a1f2bSShri Abhyankar const PetscScalar *b; 2731813a1f2bSShri Abhyankar const MatScalar *aa = a->a,*v; 2732813a1f2bSShri Abhyankar PetscInt i,nz; 2733813a1f2bSShri Abhyankar 2734813a1f2bSShri Abhyankar PetscFunctionBegin; 2735813a1f2bSShri Abhyankar if (!n) PetscFunctionReturn(0); 2736813a1f2bSShri Abhyankar 2737813a1f2bSShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 2738813a1f2bSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2739813a1f2bSShri Abhyankar 2740813a1f2bSShri Abhyankar /* forward solve the lower triangular */ 2741813a1f2bSShri Abhyankar x[0] = b[0]; 2742813a1f2bSShri Abhyankar v = aa; 2743813a1f2bSShri Abhyankar vi = aj; 2744813a1f2bSShri Abhyankar for (i=1; i<n; i++) { 2745813a1f2bSShri Abhyankar nz = ai[i+1] - ai[i]; 2746813a1f2bSShri Abhyankar sum = b[i]; 2747813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 2748813a1f2bSShri Abhyankar v += nz; 2749813a1f2bSShri Abhyankar vi += nz; 2750813a1f2bSShri Abhyankar x[i] = sum; 2751813a1f2bSShri Abhyankar } 2752813a1f2bSShri Abhyankar 2753813a1f2bSShri Abhyankar /* backward solve the upper triangular */ 2754813a1f2bSShri Abhyankar /* v = aa + ai[n+1]; 2755813a1f2bSShri Abhyankar vi = aj + ai[n+1]; */ 2756813a1f2bSShri Abhyankar v = aa + adiag[n-1]; 2757813a1f2bSShri Abhyankar vi = aj + adiag[n-1]; 275862fbd6afSShri Abhyankar for (i=n-1; i>=0; i--){ 2759813a1f2bSShri Abhyankar nz = adiag[i] - adiag[i+1]-1; 2760813a1f2bSShri Abhyankar sum = x[i]; 2761813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 2762813a1f2bSShri Abhyankar v += nz; 2763813a1f2bSShri Abhyankar vi += nz; vi++; 2764813a1f2bSShri Abhyankar x[i] = *v++ *sum; /* x[i]=aa[adiag[i]]*sum; v++; */ 2765813a1f2bSShri Abhyankar } 2766813a1f2bSShri Abhyankar 2767813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 2768813a1f2bSShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 2769813a1f2bSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2770813a1f2bSShri Abhyankar PetscFunctionReturn(0); 2771813a1f2bSShri Abhyankar } 2772813a1f2bSShri Abhyankar 2773813a1f2bSShri Abhyankar #undef __FUNCT__ 2774894a2336SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_newdatastruct" 2775894a2336SHong Zhang PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec xx) 27761d098868SHong Zhang { 27771d098868SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 27781d098868SHong Zhang IS iscol = a->col,isrow = a->row; 27791d098868SHong Zhang PetscErrorCode ierr; 2780894a2336SHong Zhang PetscInt i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,nz,k; 27811d098868SHong Zhang const PetscInt *rout,*cout,*r,*c; 2782894a2336SHong Zhang PetscScalar *x,*tmp,*tmps,sum; 27831d098868SHong Zhang const PetscScalar *b; 27841d098868SHong Zhang const MatScalar *aa = a->a,*v; 27851d098868SHong Zhang 27861d098868SHong Zhang PetscFunctionBegin; 27871d098868SHong Zhang if (!n) PetscFunctionReturn(0); 27881d098868SHong Zhang 27891d098868SHong Zhang ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 27901d098868SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 27911d098868SHong Zhang tmp = a->solve_work; 27921d098868SHong Zhang 27931d098868SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 2794894a2336SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 27951d098868SHong Zhang 27961d098868SHong Zhang /* forward solve the lower triangular */ 27971d098868SHong Zhang tmp[0] = b[*r++]; 27981d098868SHong Zhang tmps = tmp; 27991d098868SHong Zhang v = aa; 28001d098868SHong Zhang vi = aj; 28011d098868SHong Zhang for (i=1; i<n; i++) { 28021d098868SHong Zhang nz = ai[i+1] - ai[i]; 2803894a2336SHong Zhang sum = b[*r++]; 2804894a2336SHong Zhang PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 2805894a2336SHong Zhang tmp[i] = sum; 28061d098868SHong Zhang v += nz; vi += nz; 28071d098868SHong Zhang } 28081d098868SHong Zhang 28091d098868SHong Zhang /* backward solve the upper triangular */ 2810894a2336SHong Zhang k = n+1; 2811894a2336SHong Zhang v = aa + ai[k]; /* 1st entry of U(n-1,:) */ 2812894a2336SHong Zhang vi = aj + ai[k]; 28131d098868SHong Zhang for (i=n-1; i>=0; i--){ 2814894a2336SHong Zhang k = 2*n-i; 2815894a2336SHong Zhang nz = ai[k +1] - ai[k] - 1; 2816894a2336SHong Zhang sum = tmp[i]; 2817894a2336SHong Zhang PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 2818894a2336SHong Zhang x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 28191d098868SHong Zhang v += nz+1; vi += nz+1; 28201d098868SHong Zhang } 28211d098868SHong Zhang 28221d098868SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 28231d098868SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 28241d098868SHong Zhang ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 28251d098868SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 28261d098868SHong Zhang ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 28271d098868SHong Zhang PetscFunctionReturn(0); 28281d098868SHong Zhang } 28291d098868SHong Zhang 28301d098868SHong Zhang #undef __FUNCT__ 2831f268cba8SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_newdatastruct_v2" 2832f268cba8SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_newdatastruct_v2(Mat A,Vec bb,Vec xx) 2833f268cba8SShri Abhyankar { 2834f268cba8SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2835f268cba8SShri Abhyankar IS iscol = a->col,isrow = a->row; 2836f268cba8SShri Abhyankar PetscErrorCode ierr; 2837f268cba8SShri Abhyankar PetscInt i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz; 2838f268cba8SShri Abhyankar const PetscInt *rout,*cout,*r,*c; 2839f268cba8SShri Abhyankar PetscScalar *x,*tmp,sum; 2840f268cba8SShri Abhyankar const PetscScalar *b; 2841f268cba8SShri Abhyankar const MatScalar *aa = a->a,*v; 2842f268cba8SShri Abhyankar 2843f268cba8SShri Abhyankar PetscFunctionBegin; 2844f268cba8SShri Abhyankar if (!n) PetscFunctionReturn(0); 2845f268cba8SShri Abhyankar 2846f268cba8SShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 2847f268cba8SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2848f268cba8SShri Abhyankar tmp = a->solve_work; 2849f268cba8SShri Abhyankar 2850f268cba8SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 2851f268cba8SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 2852f268cba8SShri Abhyankar 2853f268cba8SShri Abhyankar /* forward solve the lower triangular */ 2854f268cba8SShri Abhyankar /* tmp[0] = b[*r++]; */ 2855f268cba8SShri Abhyankar tmp[0] = b[r[0]]; 2856f268cba8SShri Abhyankar v = aa; 2857f268cba8SShri Abhyankar vi = aj; 2858f268cba8SShri Abhyankar for (i=1; i<n; i++) { 2859f268cba8SShri Abhyankar nz = ai[i+1] - ai[i]; 2860f268cba8SShri Abhyankar sum = b[r[i]]; 2861f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 2862f268cba8SShri Abhyankar tmp[i] = sum; 2863f268cba8SShri Abhyankar v += nz; vi += nz; 2864f268cba8SShri Abhyankar } 2865f268cba8SShri Abhyankar 2866f268cba8SShri Abhyankar /* backward solve the upper triangular */ 2867f268cba8SShri Abhyankar /* v = aa + ai[k]; *//* 1st entry of U(n-1,:) */ 2868f268cba8SShri Abhyankar /* vi = aj + ai[k]; */ 2869f268cba8SShri Abhyankar v = aa + adiag[n-1]; 2870f268cba8SShri Abhyankar vi = aj + adiag[n-1]; 2871f268cba8SShri Abhyankar for (i=n-1; i>=0; i--){ 2872f268cba8SShri Abhyankar /* nz = ai[k +1] - ai[k] - 1;*/ 2873f268cba8SShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 2874f268cba8SShri Abhyankar sum = tmp[i]; 2875f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 2876f268cba8SShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 2877f268cba8SShri Abhyankar v += nz+1; vi += nz+1; 2878f268cba8SShri Abhyankar } 2879f268cba8SShri Abhyankar 2880f268cba8SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 2881f268cba8SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 2882f268cba8SShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 2883f268cba8SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2884f268cba8SShri Abhyankar ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 2885f268cba8SShri Abhyankar PetscFunctionReturn(0); 2886f268cba8SShri Abhyankar } 2887f268cba8SShri Abhyankar 2888f268cba8SShri Abhyankar #undef __FUNCT__ 28891d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ" 28901d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact) 28911d098868SHong Zhang { 28921d098868SHong Zhang Mat B = *fact; 2893599ef60dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b; 2894599ef60dSHong Zhang IS isicol; 2895599ef60dSHong Zhang PetscErrorCode ierr; 28961d098868SHong Zhang const PetscInt *r,*ic; 28971d098868SHong Zhang PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag; 2898dc3a2fd3SHong Zhang PetscInt *bi,*bj,*bdiag,*bdiag_rev; 2899f61a948aSLisandro Dalcin PetscInt row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au; 29001d098868SHong Zhang PetscInt nlnk,*lnk; 29011d098868SHong Zhang PetscBT lnkbt; 29021d098868SHong Zhang PetscTruth row_identity,icol_identity,both_identity; 29038aee2decSHong Zhang MatScalar *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp; 29041d098868SHong Zhang const PetscInt *ics; 29051d098868SHong Zhang PetscInt j,nz,*pj,*bjtmp,k,ncut,*jtmp; 2906f61a948aSLisandro Dalcin PetscReal dt=info->dt,dtcol=info->dtcol,shift=info->shiftinblocks; 2907f61a948aSLisandro Dalcin PetscInt dtcount=(PetscInt)info->dtcount,nnz_max; 2908599ef60dSHong Zhang PetscTruth missing; 2909599ef60dSHong Zhang 2910599ef60dSHong Zhang PetscFunctionBegin; 2911f61a948aSLisandro Dalcin 2912f61a948aSLisandro Dalcin if (dt == PETSC_DEFAULT) dt = 0.005; 2913f61a948aSLisandro Dalcin if (dtcol == PETSC_DEFAULT) dtcol = 0.01; /* XXX unused! */ 2914f61a948aSLisandro Dalcin if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax); 2915f61a948aSLisandro Dalcin 29161d098868SHong Zhang /* ------- symbolic factorization, can be reused ---------*/ 29171d098868SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 29181d098868SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 29191d098868SHong Zhang adiag=a->diag; 2920599ef60dSHong Zhang 2921599ef60dSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 2922599ef60dSHong Zhang 2923599ef60dSHong Zhang /* bdiag is location of diagonal in factor */ 2924*a6df321fSShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt,&bdiag,n+1,PetscInt,&bdiag_rev);CHKERRQ(ierr); 2925599ef60dSHong Zhang 29261d098868SHong Zhang /* allocate row pointers bi */ 29278fc3a347SHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 29281d098868SHong Zhang 29291d098868SHong Zhang /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */ 2930393d3a68SHong Zhang if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */ 29311d098868SHong Zhang nnz_max = ai[n]+2*n*dtcount+2; 29328fc3a347SHong Zhang 29338fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 29348fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr); 2935599ef60dSHong Zhang 2936599ef60dSHong Zhang /* put together the new matrix */ 2937599ef60dSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 2938599ef60dSHong Zhang ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 2939599ef60dSHong Zhang b = (Mat_SeqAIJ*)(B)->data; 2940599ef60dSHong Zhang b->free_a = PETSC_TRUE; 2941599ef60dSHong Zhang b->free_ij = PETSC_TRUE; 2942599ef60dSHong Zhang b->singlemalloc = PETSC_FALSE; 2943599ef60dSHong Zhang b->a = ba; 2944599ef60dSHong Zhang b->j = bj; 2945599ef60dSHong Zhang b->i = bi; 2946599ef60dSHong Zhang b->diag = bdiag; 2947599ef60dSHong Zhang b->ilen = 0; 2948599ef60dSHong Zhang b->imax = 0; 2949599ef60dSHong Zhang b->row = isrow; 2950599ef60dSHong Zhang b->col = iscol; 2951599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2952599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 2953599ef60dSHong Zhang b->icol = isicol; 2954599ef60dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2955599ef60dSHong Zhang 29561d098868SHong Zhang ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 29571d098868SHong Zhang b->maxnz = nnz_max; 2958599ef60dSHong Zhang 2959599ef60dSHong Zhang (B)->factor = MAT_FACTOR_ILUDT; 2960599ef60dSHong Zhang (B)->info.factor_mallocs = 0; 29611d098868SHong Zhang (B)->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]); 29621d098868SHong Zhang CHKMEMQ; 29631d098868SHong Zhang /* ------- end of symbolic factorization ---------*/ 2964599ef60dSHong Zhang 2965599ef60dSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 2966599ef60dSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 2967599ef60dSHong Zhang ics = ic; 2968599ef60dSHong Zhang 2969599ef60dSHong Zhang /* linked list for storing column indices of the active row */ 2970599ef60dSHong Zhang nlnk = n + 1; 2971599ef60dSHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 29721d098868SHong Zhang 29731d098868SHong Zhang /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */ 29740e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt,&im,n,PetscInt,&jtmp);CHKERRQ(ierr); 29751d098868SHong Zhang /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */ 29760e83c824SBarry Smith ierr = PetscMalloc2(n,MatScalar,&rtmp,n,MatScalar,&vtmp);CHKERRQ(ierr); 29770e83c824SBarry Smith ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr); 2978599ef60dSHong Zhang 2979599ef60dSHong Zhang bi[0] = 0; 29808fc3a347SHong Zhang bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */ 2981dc3a2fd3SHong Zhang bdiag_rev[n] = bdiag[0]; 29828fc3a347SHong Zhang bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */ 2983599ef60dSHong Zhang for (i=0; i<n; i++) { 2984599ef60dSHong Zhang /* copy initial fill into linked list */ 2985599ef60dSHong Zhang nzi = 0; /* nonzeros for active row i */ 29868369b28dSHong Zhang nzi = ai[r[i]+1] - ai[r[i]]; 29878369b28dSHong 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); 29888369b28dSHong Zhang nzi_al = adiag[r[i]] - ai[r[i]]; 29898369b28dSHong Zhang nzi_au = ai[r[i]+1] - adiag[r[i]] -1; 2990599ef60dSHong Zhang ajtmp = aj + ai[r[i]]; 29918369b28dSHong Zhang ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 2992599ef60dSHong Zhang 2993599ef60dSHong Zhang /* load in initial (unfactored row) */ 29941d098868SHong Zhang aatmp = a->a + ai[r[i]]; 29958369b28dSHong Zhang for (j=0; j<nzi; j++) { 29961d098868SHong Zhang rtmp[ics[*ajtmp++]] = *aatmp++; 2997599ef60dSHong Zhang } 2998599ef60dSHong Zhang 2999599ef60dSHong Zhang /* add pivot rows into linked list */ 3000599ef60dSHong Zhang row = lnk[n]; 3001599ef60dSHong Zhang while (row < i ) { 30021d098868SHong Zhang nzi_bl = bi[row+1] - bi[row] + 1; 30031d098868SHong Zhang bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */ 30041d098868SHong Zhang ierr = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr); 3005599ef60dSHong Zhang nzi += nlnk; 3006599ef60dSHong Zhang row = lnk[row]; 3007599ef60dSHong Zhang } 3008599ef60dSHong Zhang 30098369b28dSHong Zhang /* copy data from lnk into jtmp, then initialize lnk */ 30108369b28dSHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr); 3011599ef60dSHong Zhang 3012599ef60dSHong Zhang /* numerical factorization */ 30138369b28dSHong Zhang bjtmp = jtmp; 3014599ef60dSHong Zhang row = *bjtmp++; /* 1st pivot row */ 3015599ef60dSHong Zhang while ( row < i ) { 3016599ef60dSHong Zhang pc = rtmp + row; 30173c2a7987SHong Zhang pv = ba + bdiag[row]; /* 1./(diag of the pivot row) */ 30183c2a7987SHong Zhang multiplier = (*pc) * (*pv); 3019599ef60dSHong Zhang *pc = multiplier; 30203c2a7987SHong Zhang if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */ 30211d098868SHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 30221d098868SHong Zhang pv = ba + bdiag[row+1] + 1; 30231d098868SHong Zhang /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */ 30241d098868SHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 30251d098868SHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 3026dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 3027599ef60dSHong Zhang } 3028599ef60dSHong Zhang row = *bjtmp++; 3029599ef60dSHong Zhang } 3030599ef60dSHong Zhang 30318369b28dSHong Zhang /* copy sparse rtmp into contiguous vtmp; separate L and U part */ 30328aee2decSHong Zhang diag_tmp = rtmp[i]; /* save diagonal value - may not needed?? */ 30338369b28dSHong Zhang nzi_bl = 0; j = 0; 30348369b28dSHong Zhang while (jtmp[j] < i){ /* Note: jtmp is sorted */ 30358aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 30368369b28dSHong Zhang nzi_bl++; j++; 30378369b28dSHong Zhang } 30388369b28dSHong Zhang nzi_bu = nzi - nzi_bl -1; 30398369b28dSHong Zhang while (j < nzi){ 30408aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 30418369b28dSHong Zhang j++; 30428369b28dSHong Zhang } 30438369b28dSHong Zhang 3044599ef60dSHong Zhang bjtmp = bj + bi[i]; 3045599ef60dSHong Zhang batmp = ba + bi[i]; 30468369b28dSHong Zhang /* apply level dropping rule to L part */ 30478369b28dSHong Zhang ncut = nzi_al + dtcount; 30488369b28dSHong Zhang if (ncut < nzi_bl){ 30498369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr); 30508369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr); 3051599ef60dSHong Zhang } else { 30528369b28dSHong Zhang ncut = nzi_bl; 30538369b28dSHong Zhang } 30548369b28dSHong Zhang for (j=0; j<ncut; j++){ 30558369b28dSHong Zhang bjtmp[j] = jtmp[j]; 30568369b28dSHong Zhang batmp[j] = vtmp[j]; 30576da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */ 30588369b28dSHong Zhang } 30591d098868SHong Zhang bi[i+1] = bi[i] + ncut; 30608369b28dSHong Zhang nzi = ncut + 1; 30618369b28dSHong Zhang 30628369b28dSHong Zhang /* apply level dropping rule to U part */ 30638369b28dSHong Zhang ncut = nzi_au + dtcount; 30648369b28dSHong Zhang if (ncut < nzi_bu){ 30658369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr); 30668369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr); 30678369b28dSHong Zhang } else { 30688369b28dSHong Zhang ncut = nzi_bu; 30698369b28dSHong Zhang } 30708369b28dSHong Zhang nzi += ncut; 30711d098868SHong Zhang 30721d098868SHong Zhang /* mark bdiagonal */ 30731d098868SHong Zhang bdiag[i+1] = bdiag[i] - (ncut + 1); 3074dc3a2fd3SHong Zhang bdiag_rev[n-i-1] = bdiag[i+1]; 30758fc3a347SHong Zhang bi[2*n - i] = bi[2*n - i +1] - (ncut + 1); 30761d098868SHong Zhang bjtmp = bj + bdiag[i]; 30771d098868SHong Zhang batmp = ba + bdiag[i]; 30781d098868SHong Zhang *bjtmp = i; 30798aee2decSHong Zhang *batmp = diag_tmp; /* rtmp[i]; */ 3080c9c72f8fSHong Zhang if (*batmp == 0.0) { 3081c9c72f8fSHong Zhang *batmp = dt+shift; 3082bd1bc851SBarry Smith /* printf(" row %d add shift %g\n",i,shift); */ 3083c9c72f8fSHong Zhang } 3084b78a477dSHong Zhang *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */ 30856da515a1SHong Zhang /* printf(" (%d,%g),",*bjtmp,*batmp); */ 30861d098868SHong Zhang 30871d098868SHong Zhang bjtmp = bj + bdiag[i+1]+1; 30881d098868SHong Zhang batmp = ba + bdiag[i+1]+1; 30898369b28dSHong Zhang for (k=0; k<ncut; k++){ 30901d098868SHong Zhang bjtmp[k] = jtmp[nzi_bl+1+k]; 30911d098868SHong Zhang batmp[k] = vtmp[nzi_bl+1+k]; 30926da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */ 3093599ef60dSHong Zhang } 30946da515a1SHong Zhang /* printf("\n"); */ 3095599ef60dSHong Zhang 3096599ef60dSHong Zhang im[i] = nzi; /* used by PetscLLAddSortedLU() */ 3097599ef60dSHong Zhang /* 30981d098868SHong Zhang printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]); 30998369b28dSHong Zhang printf(" ----------------------------\n"); 3100599ef60dSHong Zhang */ 31018369b28dSHong Zhang } /* for (i=0; i<n; i++) */ 31021d098868SHong Zhang /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */ 31031d098868SHong 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]); 3104599ef60dSHong Zhang 3105599ef60dSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3106599ef60dSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3107599ef60dSHong Zhang 3108599ef60dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 31090e83c824SBarry Smith ierr = PetscFree2(im,jtmp);CHKERRQ(ierr); 31100e83c824SBarry Smith ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr); 3111599ef60dSHong Zhang 3112599ef60dSHong Zhang ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr); 31131d098868SHong Zhang b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n]; 3114599ef60dSHong Zhang 3115599ef60dSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 3116599ef60dSHong Zhang ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr); 3117599ef60dSHong Zhang both_identity = (PetscTruth) (row_identity && icol_identity); 3118599ef60dSHong Zhang if (row_identity && icol_identity) { 3119894a2336SHong Zhang B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct; 3120599ef60dSHong Zhang } else { 3121894a2336SHong Zhang B->ops->solve = MatSolve_SeqAIJ_newdatastruct; 3122599ef60dSHong Zhang } 3123599ef60dSHong Zhang 3124b78a477dSHong Zhang B->ops->lufactorsymbolic = MatILUDTFactorSymbolic_SeqAIJ; 3125b78a477dSHong Zhang B->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ; 31261d098868SHong Zhang B->ops->solveadd = 0; 31271d098868SHong Zhang B->ops->solvetranspose = 0; 31281d098868SHong Zhang B->ops->solvetransposeadd = 0; 31291d098868SHong Zhang B->ops->matsolve = 0; 3130599ef60dSHong Zhang B->assembled = PETSC_TRUE; 3131599ef60dSHong Zhang B->preallocated = PETSC_TRUE; 3132599ef60dSHong Zhang PetscFunctionReturn(0); 3133599ef60dSHong Zhang } 3134599ef60dSHong Zhang 31353c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */ 31363c2a7987SHong Zhang #undef __FUNCT__ 31373c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ" 31383c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info) 31393c2a7987SHong Zhang { 31403c2a7987SHong Zhang PetscErrorCode ierr; 31413c2a7987SHong Zhang 31423c2a7987SHong Zhang PetscFunctionBegin; 31433c2a7987SHong Zhang ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr); 31443c2a7987SHong Zhang 31453c2a7987SHong Zhang fact->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ; 31463c2a7987SHong Zhang PetscFunctionReturn(0); 31473c2a7987SHong Zhang } 31483c2a7987SHong Zhang 31493c2a7987SHong Zhang /* 31503c2a7987SHong Zhang same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors 31513c2a7987SHong Zhang - intend to replace existing MatLUFactorNumeric_SeqAIJ() 31523c2a7987SHong Zhang */ 31533c2a7987SHong Zhang #undef __FUNCT__ 31543c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ" 31553c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info) 31563c2a7987SHong Zhang { 31573c2a7987SHong Zhang Mat C=fact; 31583c2a7987SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 31593c2a7987SHong Zhang IS isrow = b->row,isicol = b->icol; 31603c2a7987SHong Zhang PetscErrorCode ierr; 31613c2a7987SHong Zhang const PetscInt *r,*ic,*ics; 3162b78a477dSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 3163b78a477dSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj; 31643c2a7987SHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 3165b78a477dSHong Zhang PetscReal dt=info->dt,shift=info->shiftinblocks; 3166b78a477dSHong Zhang PetscTruth row_identity, col_identity; 31673c2a7987SHong Zhang 31683c2a7987SHong Zhang PetscFunctionBegin; 31693c2a7987SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 31703c2a7987SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3171b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 31723c2a7987SHong Zhang ics = ic; 31733c2a7987SHong Zhang 31743c2a7987SHong Zhang for (i=0; i<n; i++){ 3175b78a477dSHong Zhang /* initialize rtmp array */ 3176b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of nozeros in L(i,:) */ 31773c2a7987SHong Zhang bjtmp = bj + bi[i]; 3178b78a477dSHong Zhang for (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0; 3179b78a477dSHong Zhang rtmp[i] = 0.0; 3180b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */ 3181b78a477dSHong Zhang bjtmp = bj + bdiag[i+1] + 1; 3182b78a477dSHong Zhang for (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0; 31833c2a7987SHong Zhang 3184b78a477dSHong Zhang /* load in initial unfactored row of A */ 31856da515a1SHong Zhang /* printf("row %d\n",i); */ 31863c2a7987SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 31873c2a7987SHong Zhang ajtmp = aj + ai[r[i]]; 31883c2a7987SHong Zhang v = aa + ai[r[i]]; 31893c2a7987SHong Zhang for (j=0; j<nz; j++) { 3190b78a477dSHong Zhang rtmp[ics[*ajtmp++]] = v[j]; 31916da515a1SHong Zhang /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */ 31923c2a7987SHong Zhang } 31936da515a1SHong Zhang /* printf("\n"); */ 31943c2a7987SHong Zhang 3195b78a477dSHong Zhang /* numerical factorization */ 3196b78a477dSHong Zhang bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */ 3197b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of entries in L(i,:) */ 3198b78a477dSHong Zhang k = 0; 3199b78a477dSHong Zhang while (k < nzl){ 32003c2a7987SHong Zhang row = *bjtmp++; 32016da515a1SHong Zhang /* printf(" prow %d\n",row); */ 32023c2a7987SHong Zhang pc = rtmp + row; 3203b78a477dSHong Zhang pv = b->a + bdiag[row]; /* 1./(diag of the pivot row) */ 3204b78a477dSHong Zhang multiplier = (*pc) * (*pv); 32053c2a7987SHong Zhang *pc = multiplier; 3206b78a477dSHong Zhang if (PetscAbsScalar(multiplier) > dt){ 3207b78a477dSHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 3208b78a477dSHong Zhang pv = b->a + bdiag[row+1] + 1; 3209b78a477dSHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 3210b78a477dSHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 32116da515a1SHong Zhang /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */ 32123c2a7987SHong Zhang } 3213b78a477dSHong Zhang k++; 32143c2a7987SHong Zhang } 32153c2a7987SHong Zhang 3216b78a477dSHong Zhang /* finished row so stick it into b->a */ 3217b78a477dSHong Zhang /* L-part */ 3218b78a477dSHong Zhang pv = b->a + bi[i] ; 3219b78a477dSHong Zhang pj = bj + bi[i] ; 3220b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; 3221b78a477dSHong Zhang for (j=0; j<nzl; j++) { 3222b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 32236da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 32243c2a7987SHong Zhang } 3225b78a477dSHong Zhang 3226b78a477dSHong Zhang /* diagonal: invert diagonal entries for simplier triangular solves */ 3227b78a477dSHong Zhang if (rtmp[i] == 0.0) rtmp[i] = dt+shift; 3228b78a477dSHong Zhang b->a[bdiag[i]] = 1.0/rtmp[i]; 32296da515a1SHong Zhang /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */ 3230b78a477dSHong Zhang 3231b78a477dSHong Zhang /* U-part */ 3232b78a477dSHong Zhang pv = b->a + bdiag[i+1] + 1; 3233b78a477dSHong Zhang pj = bj + bdiag[i+1] + 1; 3234b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1] - 1; 3235b78a477dSHong Zhang for (j=0; j<nzu; j++) { 3236b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 32376da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 3238b78a477dSHong Zhang } 32396da515a1SHong Zhang /* printf("\n"); */ 3240b78a477dSHong Zhang } 3241b78a477dSHong Zhang 32423c2a7987SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 32433c2a7987SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 32443c2a7987SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3245b78a477dSHong Zhang 32463c2a7987SHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 32473c2a7987SHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 32483c2a7987SHong Zhang if (row_identity && col_identity) { 3249894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct; 32503c2a7987SHong Zhang } else { 3251894a2336SHong Zhang C->ops->solve = MatSolve_SeqAIJ_newdatastruct; 32523c2a7987SHong Zhang } 3253b78a477dSHong Zhang C->ops->solveadd = 0; 3254b78a477dSHong Zhang C->ops->solvetranspose = 0; 3255b78a477dSHong Zhang C->ops->solvetransposeadd = 0; 3256b78a477dSHong Zhang C->ops->matsolve = 0; 32573c2a7987SHong Zhang C->assembled = PETSC_TRUE; 32583c2a7987SHong Zhang C->preallocated = PETSC_TRUE; 32593c2a7987SHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 32603c2a7987SHong Zhang PetscFunctionReturn(0); 32613c2a7987SHong Zhang } 3262