1be1d678aSKris Buschelman #define PETSCMAT_DLL 2289bc588SBarry Smith 3b377110cSBarry Smith 47c4f633dSBarry Smith #include "../src/mat/impls/aij/seq/aij.h" 5783ef271SHong Zhang #include "../src/mat/impls/sbaij/seq/sbaij.h" 61393bc97SHong Zhang #include "petscbt.h" 77c4f633dSBarry Smith #include "../src/mat/utils/freespace.h" 83b2fbd54SBarry Smith 9f6a26d55SMatthew Knepley EXTERN_C_BEGIN 104a2ae208SSatish Balay #undef __FUNCT__ 114a2ae208SSatish Balay #define __FUNCT__ "MatOrdering_Flow_SeqAIJ" 128fa24674SBarry Smith /* 138fa24674SBarry Smith Computes an ordering to get most of the large numerical values in the lower triangular part of the matrix 148fa24674SBarry Smith */ 15dfbe8321SBarry Smith PetscErrorCode MatOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol) 163b2fbd54SBarry Smith { 178fa24674SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)mat->data; 188fa24674SBarry Smith PetscErrorCode ierr; 19889a8dedSBarry Smith PetscInt i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order; 208fa24674SBarry Smith const PetscInt *ai = a->i, *aj = a->j; 218fa24674SBarry Smith const PetscScalar *aa = a->a; 228fa24674SBarry Smith PetscTruth *done; 23889a8dedSBarry Smith PetscReal best,past = 0,future; 243a40ed3dSBarry Smith 258fa24674SBarry Smith PetscFunctionBegin; 268fa24674SBarry Smith /* pick initial row */ 278fa24674SBarry Smith best = -1; 288fa24674SBarry Smith for (i=0; i<n; i++) { 2975567043SBarry Smith future = 0.0; 308fa24674SBarry Smith for (j=ai[i]; j<ai[i+1]; j++) { 318fa24674SBarry Smith if (aj[j] != i) future += PetscAbsScalar(aa[j]); else past = PetscAbsScalar(aa[j]); 328fa24674SBarry Smith } 338fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 348fa24674SBarry Smith if (past/future > best) { 358fa24674SBarry Smith best = past/future; 368fa24674SBarry Smith current = i; 378fa24674SBarry Smith } 388fa24674SBarry Smith } 398fa24674SBarry Smith 408fa24674SBarry Smith ierr = PetscMalloc(n*sizeof(PetscTruth),&done);CHKERRQ(ierr); 418fa24674SBarry Smith ierr = PetscMemzero(done,n*sizeof(PetscTruth));CHKERRQ(ierr); 420e83c824SBarry Smith ierr = PetscMalloc(n*sizeof(PetscInt),&order);CHKERRQ(ierr); 438fa24674SBarry Smith order[0] = current; 448fa24674SBarry Smith for (i=0; i<n-1; i++) { 458fa24674SBarry Smith done[current] = PETSC_TRUE; 468fa24674SBarry Smith best = -1; 478fa24674SBarry Smith /* loop over all neighbors of current pivot */ 488fa24674SBarry Smith for (j=ai[current]; j<ai[current+1]; j++) { 498fa24674SBarry Smith jj = aj[j]; 508fa24674SBarry Smith if (done[jj]) continue; 518fa24674SBarry Smith /* loop over columns of potential next row computing weights for below and above diagonal */ 528fa24674SBarry Smith past = future = 0.0; 538fa24674SBarry Smith for (k=ai[jj]; k<ai[jj+1]; k++) { 548fa24674SBarry Smith kk = aj[k]; 558fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[k]); 568fa24674SBarry Smith else if (kk != jj) future += PetscAbsScalar(aa[k]); 578fa24674SBarry Smith } 588fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 598fa24674SBarry Smith if (past/future > best) { 608fa24674SBarry Smith best = past/future; 618fa24674SBarry Smith newcurrent = jj; 628fa24674SBarry Smith } 638fa24674SBarry Smith } 648fa24674SBarry Smith if (best == -1) { /* no neighbors to select from so select best of all that remain */ 658fa24674SBarry Smith best = -1; 668fa24674SBarry Smith for (k=0; k<n; k++) { 678fa24674SBarry Smith if (done[k]) continue; 6875567043SBarry Smith future = 0.0; 6975567043SBarry Smith past = 0.0; 708fa24674SBarry Smith for (j=ai[k]; j<ai[k+1]; j++) { 718fa24674SBarry Smith kk = aj[j]; 728fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[j]); 738fa24674SBarry Smith else if (kk != k) future += PetscAbsScalar(aa[j]); 748fa24674SBarry Smith } 758fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 768fa24674SBarry Smith if (past/future > best) { 778fa24674SBarry Smith best = past/future; 788fa24674SBarry Smith newcurrent = k; 798fa24674SBarry Smith } 808fa24674SBarry Smith } 818fa24674SBarry Smith } 828fa24674SBarry Smith if (current == newcurrent) SETERRQ(PETSC_ERR_PLIB,"newcurrent cannot be current"); 838fa24674SBarry Smith current = newcurrent; 848fa24674SBarry Smith order[i+1] = current; 858fa24674SBarry Smith } 868fa24674SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,order,irow);CHKERRQ(ierr); 878fa24674SBarry Smith *icol = *irow; 888fa24674SBarry Smith ierr = PetscObjectReference((PetscObject)*irow);CHKERRQ(ierr); 898fa24674SBarry Smith ierr = PetscFree(done);CHKERRQ(ierr); 908fa24674SBarry Smith ierr = PetscFree(order);CHKERRQ(ierr); 91d64ed03dSBarry Smith PetscFunctionReturn(0); 923b2fbd54SBarry Smith } 93f6a26d55SMatthew Knepley EXTERN_C_END 943b2fbd54SBarry Smith 95e631078cSBarry Smith EXTERN_C_BEGIN 964a2ae208SSatish Balay #undef __FUNCT__ 97db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc" 98db4efbfdSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscTruth *flg) 99db4efbfdSBarry Smith { 100db4efbfdSBarry Smith PetscFunctionBegin; 101db4efbfdSBarry Smith *flg = PETSC_TRUE; 102db4efbfdSBarry Smith PetscFunctionReturn(0); 103db4efbfdSBarry Smith } 104db4efbfdSBarry Smith EXTERN_C_END 105db4efbfdSBarry Smith 106db4efbfdSBarry Smith EXTERN_C_BEGIN 107db4efbfdSBarry Smith #undef __FUNCT__ 108b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc" 109b24902e0SBarry Smith PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B) 110b24902e0SBarry Smith { 111d0f46423SBarry Smith PetscInt n = A->rmap->n; 112b24902e0SBarry Smith PetscErrorCode ierr; 113b24902e0SBarry Smith 114b24902e0SBarry Smith PetscFunctionBegin; 115b24902e0SBarry Smith ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr); 116b24902e0SBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 117599ef60dSHong Zhang if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT){ 118b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr); 11935233d99SShri Abhyankar (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ; 12035233d99SShri Abhyankar (*B)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqAIJ; 121b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 122b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 123b24902e0SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 12435233d99SShri Abhyankar (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqAIJ; 12535233d99SShri Abhyankar (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ; 126b24902e0SBarry Smith } else SETERRQ(PETSC_ERR_SUP,"Factor type not supported"); 127db4efbfdSBarry Smith (*B)->factor = ftype; 128b24902e0SBarry Smith PetscFunctionReturn(0); 129b24902e0SBarry Smith } 130e631078cSBarry Smith EXTERN_C_END 131b24902e0SBarry Smith 132b24902e0SBarry Smith #undef __FUNCT__ 133ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_inplace" 134ad04f41aSHong Zhang PetscErrorCode MatLUFactorSymbolic_SeqAIJ_inplace(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 135289bc588SBarry Smith { 136416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 137289bc588SBarry Smith IS isicol; 1386849ba73SBarry Smith PetscErrorCode ierr; 1395d0c19d7SBarry Smith const PetscInt *r,*ic; 1405d0c19d7SBarry Smith PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 1411393bc97SHong Zhang PetscInt *bi,*bj,*ajtmp; 1421393bc97SHong Zhang PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 1439e046878SBarry Smith PetscReal f; 1444875ba38SHong Zhang PetscInt nlnk,*lnk,k,**bi_ptr; 145a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1461393bc97SHong Zhang PetscBT lnkbt; 147289bc588SBarry Smith 1483a40ed3dSBarry Smith PetscFunctionBegin; 149d0f46423SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square"); 1504c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 151ac121b3dSBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 152ac121b3dSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 153289bc588SBarry Smith 154289bc588SBarry Smith /* get new row pointers */ 1551393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 1561393bc97SHong Zhang bi[0] = 0; 1571393bc97SHong Zhang 1581393bc97SHong Zhang /* bdiag is location of diagonal in factor */ 1591393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1601393bc97SHong Zhang bdiag[0] = 0; 1611393bc97SHong Zhang 1621393bc97SHong Zhang /* linked list for storing column indices of the active row */ 1631393bc97SHong Zhang nlnk = n + 1; 1641393bc97SHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1651393bc97SHong Zhang 16635baf27eSSatish Balay ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 1671393bc97SHong Zhang 1681393bc97SHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 169d3d32019SBarry Smith f = info->fill; 170a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1711393bc97SHong Zhang current_space = free_space; 172289bc588SBarry Smith 173289bc588SBarry Smith for (i=0; i<n; i++) { 1741393bc97SHong Zhang /* copy previous fill into linked list */ 175289bc588SBarry Smith nzi = 0; 1761393bc97SHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 1773b4a8b6dSBarry 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); 1781393bc97SHong Zhang ajtmp = aj + ai[r[i]]; 179afcc9904SHong Zhang ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1801393bc97SHong Zhang nzi += nlnk; 1811393bc97SHong Zhang 1821393bc97SHong Zhang /* add pivot rows into linked list */ 1831393bc97SHong Zhang row = lnk[n]; 1841393bc97SHong Zhang while (row < i) { 185d1170560SHong Zhang nzbd = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */ 186d1170560SHong Zhang ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 187d1170560SHong Zhang ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 1881393bc97SHong Zhang nzi += nlnk; 1891393bc97SHong Zhang row = lnk[row]; 190289bc588SBarry Smith } 1911393bc97SHong Zhang bi[i+1] = bi[i] + nzi; 1921393bc97SHong Zhang im[i] = nzi; 1931393bc97SHong Zhang 1941393bc97SHong Zhang /* mark bdiag */ 1951393bc97SHong Zhang nzbd = 0; 1961393bc97SHong Zhang nnz = nzi; 1971393bc97SHong Zhang k = lnk[n]; 1981393bc97SHong Zhang while (nnz-- && k < i){ 1991393bc97SHong Zhang nzbd++; 2001393bc97SHong Zhang k = lnk[k]; 2011393bc97SHong Zhang } 2021393bc97SHong Zhang bdiag[i] = bi[i] + nzbd; 2031393bc97SHong Zhang 2041393bc97SHong Zhang /* if free space is not available, make more free space */ 2051393bc97SHong Zhang if (current_space->local_remaining<nzi) { 2061393bc97SHong Zhang nnz = (n - i)*nzi; /* estimated and max additional space needed */ 207a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2081393bc97SHong Zhang reallocs++; 2091393bc97SHong Zhang } 2101393bc97SHong Zhang 2111393bc97SHong Zhang /* copy data into free space, then initialize lnk */ 2121393bc97SHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 2131393bc97SHong Zhang bi_ptr[i] = current_space->array; 2141393bc97SHong Zhang current_space->array += nzi; 2151393bc97SHong Zhang current_space->local_used += nzi; 2161393bc97SHong Zhang current_space->local_remaining -= nzi; 217289bc588SBarry Smith } 2186cf91177SBarry Smith #if defined(PETSC_USE_INFO) 219464e8d44SSatish Balay if (ai[n] != 0) { 2201393bc97SHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 221ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 222ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 223ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 224ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 22505bf559cSSatish Balay } else { 226ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 22705bf559cSSatish Balay } 22863ba0a88SBarry Smith #endif 2292fb3ed76SBarry Smith 230898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 231898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2321987afe7SBarry Smith 2331393bc97SHong Zhang /* destroy list of free space and other temporary array(s) */ 2341393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 235a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); 2361393bc97SHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 23735baf27eSSatish Balay ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 238289bc588SBarry Smith 239289bc588SBarry Smith /* put together the new matrix */ 240719d5645SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 241719d5645SBarry Smith ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 242719d5645SBarry Smith b = (Mat_SeqAIJ*)(B)->data; 243e6b907acSBarry Smith b->free_a = PETSC_TRUE; 244e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 2457c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 2461393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 2471393bc97SHong Zhang b->j = bj; 2481393bc97SHong Zhang b->i = bi; 2491393bc97SHong Zhang b->diag = bdiag; 250416022c9SBarry Smith b->ilen = 0; 251416022c9SBarry Smith b->imax = 0; 252416022c9SBarry Smith b->row = isrow; 253416022c9SBarry Smith b->col = iscol; 254c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 255c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 25682bf6240SBarry Smith b->icol = isicol; 25787828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2581393bc97SHong Zhang 2591393bc97SHong Zhang /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 260719d5645SBarry Smith ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 2611393bc97SHong Zhang b->maxnz = b->nz = bi[n] ; 2627fd98bd6SLois Curfman McInnes 263719d5645SBarry Smith (B)->factor = MAT_FACTOR_LU; 264719d5645SBarry Smith (B)->info.factor_mallocs = reallocs; 265719d5645SBarry Smith (B)->info.fill_ratio_given = f; 266703deb49SSatish Balay 2679f7d0b68SHong Zhang if (ai[n]) { 268719d5645SBarry Smith (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 269eea2913aSSatish Balay } else { 270719d5645SBarry Smith (B)->info.fill_ratio_needed = 0.0; 271eea2913aSSatish Balay } 272ad04f41aSHong Zhang (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 273ad04f41aSHong Zhang (B)->ops->solve = MatSolve_SeqAIJ_inplace; 274ad04f41aSHong Zhang (B)->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 275db4efbfdSBarry Smith /* switch to inodes if appropriate */ 2764108e4d5SBarry Smith ierr = MatLUFactorSymbolic_SeqAIJ_Inode(B,A,isrow,iscol,info);CHKERRQ(ierr); 2773a40ed3dSBarry Smith PetscFunctionReturn(0); 278289bc588SBarry Smith } 2791393bc97SHong Zhang 280ce3d78c0SShri Abhyankar #undef __FUNCT__ 28135233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ" 28235233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 283ce3d78c0SShri Abhyankar { 284ce3d78c0SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 285ce3d78c0SShri Abhyankar IS isicol; 286ce3d78c0SShri Abhyankar PetscErrorCode ierr; 287ce3d78c0SShri Abhyankar const PetscInt *r,*ic; 288ce3d78c0SShri Abhyankar PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 289ce3d78c0SShri Abhyankar PetscInt *bi,*bj,*ajtmp; 290ce3d78c0SShri Abhyankar PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 291ce3d78c0SShri Abhyankar PetscReal f; 292ce3d78c0SShri Abhyankar PetscInt nlnk,*lnk,k,**bi_ptr; 293ce3d78c0SShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 294ce3d78c0SShri Abhyankar PetscBT lnkbt; 295ce3d78c0SShri Abhyankar 296ce3d78c0SShri Abhyankar PetscFunctionBegin; 297ce3d78c0SShri Abhyankar if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square"); 298ce3d78c0SShri Abhyankar ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 299ce3d78c0SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 300ce3d78c0SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 301ce3d78c0SShri Abhyankar 3021bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 3031bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 3041bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 305a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 3069b48462bSShri Abhyankar 3079b48462bSShri Abhyankar /* linked list for storing column indices of the active row */ 3089b48462bSShri Abhyankar nlnk = n + 1; 3099b48462bSShri Abhyankar ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3109b48462bSShri Abhyankar 3119b48462bSShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 3129b48462bSShri Abhyankar 3139b48462bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 3149b48462bSShri Abhyankar f = info->fill; 3159b48462bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 3169b48462bSShri Abhyankar current_space = free_space; 3179b48462bSShri Abhyankar 3189b48462bSShri Abhyankar for (i=0; i<n; i++) { 3199b48462bSShri Abhyankar /* copy previous fill into linked list */ 3209b48462bSShri Abhyankar nzi = 0; 3219b48462bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 3229b48462bSShri 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); 3239b48462bSShri Abhyankar ajtmp = aj + ai[r[i]]; 3249b48462bSShri Abhyankar ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3259b48462bSShri Abhyankar nzi += nlnk; 3269b48462bSShri Abhyankar 3279b48462bSShri Abhyankar /* add pivot rows into linked list */ 3289b48462bSShri Abhyankar row = lnk[n]; 3299b48462bSShri Abhyankar while (row < i){ 3309b48462bSShri Abhyankar nzbd = bdiag[row] + 1; /* num of entries in the row with column index <= row */ 3319b48462bSShri Abhyankar ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 3329b48462bSShri Abhyankar ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 3339b48462bSShri Abhyankar nzi += nlnk; 3349b48462bSShri Abhyankar row = lnk[row]; 3359b48462bSShri Abhyankar } 3369b48462bSShri Abhyankar bi[i+1] = bi[i] + nzi; 3379b48462bSShri Abhyankar im[i] = nzi; 3389b48462bSShri Abhyankar 3399b48462bSShri Abhyankar /* mark bdiag */ 3409b48462bSShri Abhyankar nzbd = 0; 3419b48462bSShri Abhyankar nnz = nzi; 3429b48462bSShri Abhyankar k = lnk[n]; 3439b48462bSShri Abhyankar while (nnz-- && k < i){ 3449b48462bSShri Abhyankar nzbd++; 3459b48462bSShri Abhyankar k = lnk[k]; 3469b48462bSShri Abhyankar } 3479b48462bSShri Abhyankar bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */ 3489b48462bSShri Abhyankar 3499b48462bSShri Abhyankar /* if free space is not available, make more free space */ 3509b48462bSShri Abhyankar if (current_space->local_remaining<nzi) { 3519b48462bSShri Abhyankar nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */ 3529b48462bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 3539b48462bSShri Abhyankar reallocs++; 3549b48462bSShri Abhyankar } 3559b48462bSShri Abhyankar 3569b48462bSShri Abhyankar /* copy data into free space, then initialize lnk */ 3579b48462bSShri Abhyankar ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 3589b48462bSShri Abhyankar bi_ptr[i] = current_space->array; 3599b48462bSShri Abhyankar current_space->array += nzi; 3609b48462bSShri Abhyankar current_space->local_used += nzi; 3619b48462bSShri Abhyankar current_space->local_remaining -= nzi; 3629b48462bSShri Abhyankar } 3639b48462bSShri Abhyankar #if defined(PETSC_USE_INFO) 3649b48462bSShri Abhyankar if (ai[n] != 0) { 3659b48462bSShri Abhyankar PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 3669b48462bSShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 3679b48462bSShri Abhyankar ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 3689b48462bSShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 3699b48462bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 3709b48462bSShri Abhyankar } else { 3719b48462bSShri Abhyankar ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 3729b48462bSShri Abhyankar } 3739b48462bSShri Abhyankar #endif 3749b48462bSShri Abhyankar 3759b48462bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3769b48462bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3779b48462bSShri Abhyankar 3789b48462bSShri Abhyankar /* destroy list of free space and other temporary array(s) */ 3799b48462bSShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 3802ce24eb6SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 3819b48462bSShri Abhyankar ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 3829b48462bSShri Abhyankar ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 3839b48462bSShri Abhyankar 3849b48462bSShri Abhyankar /* put together the new matrix */ 3859b48462bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 3869b48462bSShri Abhyankar ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 3879b48462bSShri Abhyankar b = (Mat_SeqAIJ*)(B)->data; 3889b48462bSShri Abhyankar b->free_a = PETSC_TRUE; 3899b48462bSShri Abhyankar b->free_ij = PETSC_TRUE; 3909b48462bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 3919b48462bSShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 3929b48462bSShri Abhyankar b->j = bj; 3939b48462bSShri Abhyankar b->i = bi; 3949b48462bSShri Abhyankar b->diag = bdiag; 3959b48462bSShri Abhyankar b->ilen = 0; 3969b48462bSShri Abhyankar b->imax = 0; 3979b48462bSShri Abhyankar b->row = isrow; 3989b48462bSShri Abhyankar b->col = iscol; 3999b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 4009b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 4019b48462bSShri Abhyankar b->icol = isicol; 4029b48462bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 4039b48462bSShri Abhyankar 4049b48462bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 4059b48462bSShri Abhyankar ierr = PetscLogObjectMemory(B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 4069b48462bSShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 407f284f12aSHong Zhang B->factor = MAT_FACTOR_LU; 408f284f12aSHong Zhang B->info.factor_mallocs = reallocs; 409f284f12aSHong Zhang B->info.fill_ratio_given = f; 4109b48462bSShri Abhyankar 4119b48462bSShri Abhyankar if (ai[n]) { 412f284f12aSHong Zhang B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 4139b48462bSShri Abhyankar } else { 414f284f12aSHong Zhang B->info.fill_ratio_needed = 0.0; 4159b48462bSShri Abhyankar } 41635233d99SShri Abhyankar B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 417b47f5a65SHong Zhang /* switch to inodes if appropriate */ 418019b515eSShri Abhyankar ierr = Mat_CheckInode_FactorLU(B,PETSC_FALSE);CHKERRQ(ierr); 4199b48462bSShri Abhyankar PetscFunctionReturn(0); 4209b48462bSShri Abhyankar } 4219b48462bSShri Abhyankar 4225b5bc046SBarry Smith /* 4235b5bc046SBarry Smith Trouble in factorization, should we dump the original matrix? 4245b5bc046SBarry Smith */ 4255b5bc046SBarry Smith #undef __FUNCT__ 4265b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix" 4275b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A) 4285b5bc046SBarry Smith { 4295b5bc046SBarry Smith PetscErrorCode ierr; 43090d69ab7SBarry Smith PetscTruth flg = PETSC_FALSE; 4315b5bc046SBarry Smith 4325b5bc046SBarry Smith PetscFunctionBegin; 43390d69ab7SBarry Smith ierr = PetscOptionsGetTruth(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr); 4345b5bc046SBarry Smith if (flg) { 4355b5bc046SBarry Smith PetscViewer viewer; 4365b5bc046SBarry Smith char filename[PETSC_MAX_PATH_LEN]; 4375b5bc046SBarry Smith 4385b5bc046SBarry Smith ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr); 4397adad957SLisandro Dalcin ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 4405b5bc046SBarry Smith ierr = MatView(A,viewer);CHKERRQ(ierr); 4415b5bc046SBarry Smith ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr); 4425b5bc046SBarry Smith } 4435b5bc046SBarry Smith PetscFunctionReturn(0); 4445b5bc046SBarry Smith } 4455b5bc046SBarry Smith 446c9c72f8fSHong Zhang #undef __FUNCT__ 44735233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ" 44835233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 449c9c72f8fSHong Zhang { 450c9c72f8fSHong Zhang Mat C=B; 451c9c72f8fSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 452c9c72f8fSHong Zhang IS isrow = b->row,isicol = b->icol; 453c9c72f8fSHong Zhang PetscErrorCode ierr; 454c9c72f8fSHong Zhang const PetscInt *r,*ic,*ics; 455bbd65245SShri Abhyankar const PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag; 456bbd65245SShri Abhyankar PetscInt i,j,k,nz,nzL,row,*pj; 457bbd65245SShri Abhyankar const PetscInt *ajtmp,*bjtmp; 458bbd65245SShri Abhyankar MatScalar *rtmp,*pc,multiplier,*pv; 459bbd65245SShri Abhyankar const MatScalar *aa=a->a,*v; 460c9c72f8fSHong Zhang PetscTruth row_identity,col_identity; 461c9c72f8fSHong Zhang 462d90ac83dSHong Zhang FactorShiftCtx sctx; 463bbd65245SShri Abhyankar PetscInt *ddiag; 464d4ad06f7SHong Zhang PetscReal rs; 465d4ad06f7SHong Zhang MatScalar d; 466d4ad06f7SHong Zhang 467c9c72f8fSHong Zhang PetscFunctionBegin; 468d6e5152cSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 469d90ac83dSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 470d4ad06f7SHong Zhang 471915743fcSHong Zhang if (info->shifttype == MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 472d4ad06f7SHong Zhang ddiag = a->diag; 473d4ad06f7SHong Zhang sctx.shift_top = info->zeropivot; 474d4ad06f7SHong Zhang for (i=0; i<n; i++) { 475d4ad06f7SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 476d4ad06f7SHong Zhang d = (aa)[ddiag[i]]; 477d4ad06f7SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 478d4ad06f7SHong Zhang v = aa+ai[i]; 479d4ad06f7SHong Zhang nz = ai[i+1] - ai[i]; 480d4ad06f7SHong Zhang for (j=0; j<nz; j++) 481d4ad06f7SHong Zhang rs += PetscAbsScalar(v[j]); 482d4ad06f7SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 483d4ad06f7SHong Zhang } 484d4ad06f7SHong Zhang sctx.shift_top *= 1.1; 485d4ad06f7SHong Zhang sctx.nshift_max = 5; 486d4ad06f7SHong Zhang sctx.shift_lo = 0.; 487d4ad06f7SHong Zhang sctx.shift_hi = 1.; 488d4ad06f7SHong Zhang } 489d4ad06f7SHong Zhang 490c9c72f8fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 491c9c72f8fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 492c9c72f8fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 493c9c72f8fSHong Zhang ics = ic; 494c9c72f8fSHong Zhang 495d4ad06f7SHong Zhang do { 496d90ac83dSHong Zhang sctx.useshift = PETSC_FALSE; 497c9c72f8fSHong Zhang for (i=0; i<n; i++){ 498c9c72f8fSHong Zhang /* zero rtmp */ 499c9c72f8fSHong Zhang /* L part */ 500c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 501c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 502c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 503c9c72f8fSHong Zhang 504c9c72f8fSHong Zhang /* U part */ 50562fbd6afSShri Abhyankar nz = bdiag[i]-bdiag[i+1]; 50662fbd6afSShri Abhyankar bjtmp = bj + bdiag[i+1]+1; 507813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 508813a1f2bSShri Abhyankar 509813a1f2bSShri Abhyankar /* load in initial (unfactored row) */ 510813a1f2bSShri Abhyankar nz = ai[r[i]+1] - ai[r[i]]; 511813a1f2bSShri Abhyankar ajtmp = aj + ai[r[i]]; 512813a1f2bSShri Abhyankar v = aa + ai[r[i]]; 513813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 514813a1f2bSShri Abhyankar rtmp[ics[ajtmp[j]]] = v[j]; 515813a1f2bSShri Abhyankar } 516813a1f2bSShri Abhyankar /* ZeropivotApply() */ 51798a93161SHong Zhang rtmp[i] += sctx.shift_amount; /* shift the diagonal of the matrix */ 518813a1f2bSShri Abhyankar 519813a1f2bSShri Abhyankar /* elimination */ 520813a1f2bSShri Abhyankar bjtmp = bj + bi[i]; 521813a1f2bSShri Abhyankar row = *bjtmp++; 522813a1f2bSShri Abhyankar nzL = bi[i+1] - bi[i]; 523f268cba8SShri Abhyankar for(k=0; k < nzL;k++) { 524813a1f2bSShri Abhyankar pc = rtmp + row; 525813a1f2bSShri Abhyankar if (*pc != 0.0) { 526813a1f2bSShri Abhyankar pv = b->a + bdiag[row]; 527813a1f2bSShri Abhyankar multiplier = *pc * (*pv); 528813a1f2bSShri Abhyankar *pc = multiplier; 52962fbd6afSShri Abhyankar pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 53062fbd6afSShri Abhyankar pv = b->a + bdiag[row+1]+1; 53162fbd6afSShri Abhyankar nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 532813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 533813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 534813a1f2bSShri Abhyankar } 535f268cba8SShri Abhyankar row = *bjtmp++; 536813a1f2bSShri Abhyankar } 537813a1f2bSShri Abhyankar 538813a1f2bSShri Abhyankar /* finished row so stick it into b->a */ 539813a1f2bSShri Abhyankar rs = 0.0; 540813a1f2bSShri Abhyankar /* L part */ 541813a1f2bSShri Abhyankar pv = b->a + bi[i] ; 542813a1f2bSShri Abhyankar pj = b->j + bi[i] ; 543813a1f2bSShri Abhyankar nz = bi[i+1] - bi[i]; 544813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 545813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 546813a1f2bSShri Abhyankar } 547813a1f2bSShri Abhyankar 548813a1f2bSShri Abhyankar /* U part */ 54962fbd6afSShri Abhyankar pv = b->a + bdiag[i+1]+1; 55062fbd6afSShri Abhyankar pj = b->j + bdiag[i+1]+1; 55162fbd6afSShri Abhyankar nz = bdiag[i] - bdiag[i+1]-1; 552813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 553813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 554813a1f2bSShri Abhyankar } 555813a1f2bSShri Abhyankar 5568ff23777SHong Zhang /* MatPivotCheck() */ 557813a1f2bSShri Abhyankar sctx.rs = rs; 558813a1f2bSShri Abhyankar sctx.pv = rtmp[i]; 559915743fcSHong Zhang if (info->shifttype == MAT_SHIFT_NONZERO){ 560bbd65245SShri Abhyankar ierr = MatPivotCheck_nz(info,sctx,i);CHKERRQ(ierr); 561915743fcSHong Zhang } else if (info->shifttype == MAT_SHIFT_POSITIVE_DEFINITE){ 562bbd65245SShri Abhyankar ierr = MatPivotCheck_pd(info,sctx,i);CHKERRQ(ierr); 563915743fcSHong Zhang } else if (info->shifttype == MAT_SHIFT_INBLOCKS){ 564bbd65245SShri Abhyankar ierr = MatPivotCheck_inblocks(info,sctx,i);CHKERRQ(ierr); 5658ff23777SHong Zhang } else { 566bbd65245SShri Abhyankar ierr = MatPivotCheck_none(info,sctx,i);CHKERRQ(ierr); 5678ff23777SHong Zhang } 5688ff23777SHong Zhang rtmp[i] = sctx.pv; 569813a1f2bSShri Abhyankar 570813a1f2bSShri Abhyankar /* Mark diagonal and invert diagonal for simplier triangular solves */ 571813a1f2bSShri Abhyankar pv = b->a + bdiag[i]; 572813a1f2bSShri Abhyankar *pv = 1.0/rtmp[i]; 573813a1f2bSShri Abhyankar 574813a1f2bSShri Abhyankar } /* endof for (i=0; i<n; i++){ */ 575813a1f2bSShri Abhyankar 5768ff23777SHong Zhang /* MatPivotRefine() */ 577915743fcSHong Zhang if (info->shifttype == MAT_SHIFT_POSITIVE_DEFINITE && !sctx.useshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){ 578813a1f2bSShri Abhyankar /* 579813a1f2bSShri Abhyankar * if no shift in this attempt & shifting & started shifting & can refine, 580813a1f2bSShri Abhyankar * then try lower shift 581813a1f2bSShri Abhyankar */ 582813a1f2bSShri Abhyankar sctx.shift_hi = sctx.shift_fraction; 583813a1f2bSShri Abhyankar sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 584813a1f2bSShri Abhyankar sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 585d90ac83dSHong Zhang sctx.useshift = PETSC_TRUE; 586813a1f2bSShri Abhyankar sctx.nshift++; 587813a1f2bSShri Abhyankar } 588d90ac83dSHong Zhang } while (sctx.useshift); 589813a1f2bSShri Abhyankar 590813a1f2bSShri Abhyankar ierr = PetscFree(rtmp);CHKERRQ(ierr); 591813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 592813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 593b47f5a65SHong Zhang if (b->inode.use){ 59435233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_Inode; 595b47f5a65SHong Zhang } else { 596813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 597813a1f2bSShri Abhyankar ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 598813a1f2bSShri Abhyankar if (row_identity && col_identity) { 59935233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 600813a1f2bSShri Abhyankar } else { 60135233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ; 602813a1f2bSShri Abhyankar } 603b47f5a65SHong Zhang } 60435233d99SShri Abhyankar C->ops->solveadd = MatSolveAdd_SeqAIJ; 60535233d99SShri Abhyankar C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 60635233d99SShri Abhyankar C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 60735233d99SShri Abhyankar C->ops->matsolve = MatMatSolve_SeqAIJ; 608813a1f2bSShri Abhyankar C->assembled = PETSC_TRUE; 609813a1f2bSShri Abhyankar C->preallocated = PETSC_TRUE; 610813a1f2bSShri Abhyankar ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 611813a1f2bSShri Abhyankar 6128ff23777SHong Zhang /* MatPivotView() */ 613813a1f2bSShri Abhyankar if (sctx.nshift){ 614915743fcSHong Zhang if (info->shifttype == MAT_SHIFT_POSITIVE_DEFINITE) { 615813a1f2bSShri 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); 616915743fcSHong Zhang } else if (info->shifttype == MAT_SHIFT_NONZERO) { 617813a1f2bSShri Abhyankar ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 618915743fcSHong Zhang } else if (info->shifttype == MAT_SHIFT_INBLOCKS){ 619915743fcSHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr); 620813a1f2bSShri Abhyankar } 621813a1f2bSShri Abhyankar } 622813a1f2bSShri Abhyankar PetscFunctionReturn(0); 623813a1f2bSShri Abhyankar } 624813a1f2bSShri Abhyankar 6254a2ae208SSatish Balay #undef __FUNCT__ 626ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_inplace" 627ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info) 628289bc588SBarry Smith { 629719d5645SBarry Smith Mat C=B; 630416022c9SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 63182bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 6326849ba73SBarry Smith PetscErrorCode ierr; 6335d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 634d278edc7SBarry Smith PetscInt nz,row,i,j,n=A->rmap->n,diag; 635d278edc7SBarry Smith const PetscInt *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 636d278edc7SBarry Smith const PetscInt *ajtmp,*bjtmp,*diag_offset = b->diag,*pj; 637d278edc7SBarry Smith MatScalar *pv,*rtmp,*pc,multiplier,d; 638d278edc7SBarry Smith const MatScalar *v,*aa=a->a; 63933f9893dSHong Zhang PetscReal rs=0.0; 640d90ac83dSHong Zhang FactorShiftCtx sctx; 64142898933SHong Zhang PetscInt newshift,*ddiag; 642289bc588SBarry Smith 6433a40ed3dSBarry Smith PetscFunctionBegin; 64478b31e54SBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 64578b31e54SBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 646b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 6479f7d0b68SHong Zhang ics = ic; 648289bc588SBarry Smith 649d4ad06f7SHong Zhang /* initialize shift context sctx */ 650d4ad06f7SHong Zhang sctx.nshift = 0; 651ada7143aSSatish Balay sctx.nshift_max = 0; 652d4ad06f7SHong Zhang sctx.shift_top = 0.0; 653d4ad06f7SHong Zhang sctx.shift_lo = 0.0; 654d4ad06f7SHong Zhang sctx.shift_hi = 0.0; 655d4ad06f7SHong Zhang sctx.shift_fraction = 0.0; 656d4ad06f7SHong Zhang sctx.shift_amount = 0.0; 657ada7143aSSatish Balay 658*b2a402fcSHong Zhang if (info->shifttype == MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 65942898933SHong Zhang ddiag = a->diag; 6609f7d0b68SHong Zhang sctx.shift_top = info->zeropivot; 6616cc28720Svictorle for (i=0; i<n; i++) { 6629f95998fSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 6639f7d0b68SHong Zhang d = (aa)[ddiag[i]]; 6641a890ab1SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 6659f7d0b68SHong Zhang v = aa+ai[i]; 6669f7d0b68SHong Zhang nz = ai[i+1] - ai[i]; 667e24cfe64SHong Zhang for (j=0; j<nz; j++) 6681a890ab1SHong Zhang rs += PetscAbsScalar(v[j]); 6691a890ab1SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 6706cc28720Svictorle } 671118f5789SBarry Smith sctx.shift_top *= 1.1; 672d2276718SHong Zhang sctx.nshift_max = 5; 673d2276718SHong Zhang sctx.shift_lo = 0.; 674d2276718SHong Zhang sctx.shift_hi = 1.; 675a0a356efSHong Zhang } 676a0a356efSHong Zhang 677085256b3SBarry Smith do { 678d90ac83dSHong Zhang sctx.useshift = PETSC_FALSE; 679289bc588SBarry Smith for (i=0; i<n; i++){ 680b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 681b3bf805bSHong Zhang bjtmp = bj + bi[i]; 6829f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 683289bc588SBarry Smith 684289bc588SBarry Smith /* load in initial (unfactored row) */ 6859f7d0b68SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 6869f7d0b68SHong Zhang ajtmp = aj + ai[r[i]]; 6879f7d0b68SHong Zhang v = aa + ai[r[i]]; 688085256b3SBarry Smith for (j=0; j<nz; j++) { 689b3bf805bSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 690085256b3SBarry Smith } 691d2276718SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 692bd1bc851SBarry Smith /* if (sctx.shift_amount > 0.0) printf("row %d, shift %g\n",i,sctx.shift_amount); */ 693289bc588SBarry Smith 694b3bf805bSHong Zhang row = *bjtmp++; 695f2582111SSatish Balay while (row < i) { 6968c37ef55SBarry Smith pc = rtmp + row; 6978c37ef55SBarry Smith if (*pc != 0.0) { 698010a20caSHong Zhang pv = b->a + diag_offset[row]; 699010a20caSHong Zhang pj = b->j + diag_offset[row] + 1; 70035aab85fSBarry Smith multiplier = *pc / *pv++; 7018c37ef55SBarry Smith *pc = multiplier; 702b3bf805bSHong Zhang nz = bi[row+1] - diag_offset[row] - 1; 7039f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 704dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 705289bc588SBarry Smith } 706b3bf805bSHong Zhang row = *bjtmp++; 707289bc588SBarry Smith } 708416022c9SBarry Smith /* finished row so stick it into b->a */ 709b3bf805bSHong Zhang pv = b->a + bi[i] ; 710b3bf805bSHong Zhang pj = b->j + bi[i] ; 711b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 712b3bf805bSHong Zhang diag = diag_offset[i] - bi[i]; 713e57ff13aSBarry Smith rs = 0.0; 714d3d32019SBarry Smith for (j=0; j<nz; j++) { 7159f7d0b68SHong Zhang pv[j] = rtmp[pj[j]]; 7169f7d0b68SHong Zhang rs += PetscAbsScalar(pv[j]); 717d3d32019SBarry Smith } 718e57ff13aSBarry Smith rs -= PetscAbsScalar(pv[diag]); 719d2276718SHong Zhang 720b3bf805bSHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 721d2276718SHong Zhang sctx.rs = rs; 722d2276718SHong Zhang sctx.pv = pv[diag]; 723c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 7245b5bc046SBarry Smith if (newshift == 1) break; 72535aab85fSBarry Smith } 726d2276718SHong Zhang 727*b2a402fcSHong Zhang if (info->shifttype == MAT_SHIFT_POSITIVE_DEFINITE && !sctx.useshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 7286cc28720Svictorle /* 7296c037d1bSvictorle * if no shift in this attempt & shifting & started shifting & can refine, 7306cc28720Svictorle * then try lower shift 7316cc28720Svictorle */ 7320481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 7330481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 7340481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 735d90ac83dSHong Zhang sctx.useshift = PETSC_TRUE; 736d2276718SHong Zhang sctx.nshift++; 7376cc28720Svictorle } 738d90ac83dSHong Zhang } while (sctx.useshift); 7390f11f9aeSSatish Balay 74035aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 74135aab85fSBarry Smith for (i=0; i<n; i++) { 742010a20caSHong Zhang b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]]; 74335aab85fSBarry Smith } 744606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 74578b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 74678b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 747db4efbfdSBarry Smith if (b->inode.use) { 748ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_Inode_inplace; 749db4efbfdSBarry Smith } else { 7508d8c7c43SBarry Smith PetscTruth row_identity, col_identity; 7518d8c7c43SBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 7528d8c7c43SBarry Smith ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 7538d8c7c43SBarry Smith if (row_identity && col_identity) { 754ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_inplace; 7558d8c7c43SBarry Smith } else { 756ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_inplace; 757db4efbfdSBarry Smith } 7588d8c7c43SBarry Smith } 759ad04f41aSHong Zhang C->ops->solveadd = MatSolveAdd_SeqAIJ_inplace; 760ad04f41aSHong Zhang C->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 761ad04f41aSHong Zhang C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 762ad04f41aSHong Zhang C->ops->matsolve = MatMatSolve_SeqAIJ_inplace; 763c456f294SBarry Smith C->assembled = PETSC_TRUE; 7645c9eb25fSBarry Smith C->preallocated = PETSC_TRUE; 765d0f46423SBarry Smith ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 766d2276718SHong Zhang if (sctx.nshift){ 767*b2a402fcSHong Zhang if (info->shifttype == MAT_SHIFT_POSITIVE_DEFINITE) { 7680481f469SBarry 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); 769*b2a402fcSHong Zhang } else if (info->shifttype == MAT_SHIFT_NONZERO) { 770e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 7716cc28720Svictorle } 7720697b6caSHong Zhang } 7733a40ed3dSBarry Smith PetscFunctionReturn(0); 774289bc588SBarry Smith } 7750889b5dcSvictorle 776137fb511SHong Zhang /* 777137fb511SHong Zhang This routine implements inplace ILU(0) with row or/and column permutations. 778137fb511SHong Zhang Input: 779137fb511SHong Zhang A - original matrix 780137fb511SHong Zhang Output; 781137fb511SHong Zhang A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i] 782137fb511SHong Zhang a->j (col index) is permuted by the inverse of colperm, then sorted 783137fb511SHong Zhang a->a reordered accordingly with a->j 784137fb511SHong Zhang a->diag (ptr to diagonal elements) is updated. 785137fb511SHong Zhang */ 786137fb511SHong Zhang #undef __FUNCT__ 787137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm" 7880481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info) 789137fb511SHong Zhang { 790b051339dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 791b051339dSHong Zhang IS isrow = a->row,isicol = a->icol; 792137fb511SHong Zhang PetscErrorCode ierr; 7935d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 7945d0c19d7SBarry Smith PetscInt i,j,n=A->rmap->n,*ai=a->i,*aj=a->j; 7955d0c19d7SBarry Smith PetscInt *ajtmp,nz,row; 796b051339dSHong Zhang PetscInt *diag = a->diag,nbdiag,*pj; 797a77337e4SBarry Smith PetscScalar *rtmp,*pc,multiplier,d; 798a77337e4SBarry Smith MatScalar *v,*pv; 799137fb511SHong Zhang PetscReal rs; 800d90ac83dSHong Zhang FactorShiftCtx sctx; 801b051339dSHong Zhang PetscInt newshift; 802137fb511SHong Zhang 803137fb511SHong Zhang PetscFunctionBegin; 804719d5645SBarry Smith if (A != B) SETERRQ(PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address"); 805137fb511SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 806137fb511SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 807137fb511SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr); 808137fb511SHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 809b051339dSHong Zhang ics = ic; 810137fb511SHong Zhang 81175567043SBarry Smith sctx.shift_top = 0.; 812e60cf9a0SBarry Smith sctx.nshift_max = 0; 81375567043SBarry Smith sctx.shift_lo = 0.; 81475567043SBarry Smith sctx.shift_hi = 0.; 81575567043SBarry Smith sctx.shift_fraction = 0.; 816137fb511SHong Zhang 817*b2a402fcSHong Zhang if (info->shifttype == MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 81875567043SBarry Smith sctx.shift_top = 0.; 819137fb511SHong Zhang for (i=0; i<n; i++) { 820137fb511SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 821b051339dSHong Zhang d = (a->a)[diag[i]]; 822137fb511SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 823b051339dSHong Zhang v = a->a+ai[i]; 824b051339dSHong Zhang nz = ai[i+1] - ai[i]; 825137fb511SHong Zhang for (j=0; j<nz; j++) 826137fb511SHong Zhang rs += PetscAbsScalar(v[j]); 827137fb511SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 828137fb511SHong Zhang } 829137fb511SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 830137fb511SHong Zhang sctx.shift_top *= 1.1; 831137fb511SHong Zhang sctx.nshift_max = 5; 832137fb511SHong Zhang sctx.shift_lo = 0.; 833137fb511SHong Zhang sctx.shift_hi = 1.; 834137fb511SHong Zhang } 835137fb511SHong Zhang 83675567043SBarry Smith sctx.shift_amount = 0.; 837137fb511SHong Zhang sctx.nshift = 0; 838137fb511SHong Zhang do { 839d90ac83dSHong Zhang sctx.useshift = PETSC_FALSE; 840137fb511SHong Zhang for (i=0; i<n; i++){ 841b051339dSHong Zhang /* load in initial unfactored row */ 842b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 843b051339dSHong Zhang ajtmp = aj + ai[r[i]]; 844b051339dSHong Zhang v = a->a + ai[r[i]]; 845137fb511SHong Zhang /* sort permuted ajtmp and values v accordingly */ 846b051339dSHong Zhang for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]]; 847137fb511SHong Zhang ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr); 848137fb511SHong Zhang 849b051339dSHong Zhang diag[r[i]] = ai[r[i]]; 850137fb511SHong Zhang for (j=0; j<nz; j++) { 851137fb511SHong Zhang rtmp[ajtmp[j]] = v[j]; 852b051339dSHong Zhang if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */ 853137fb511SHong Zhang } 854137fb511SHong Zhang rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 855137fb511SHong Zhang 856b051339dSHong Zhang row = *ajtmp++; 857137fb511SHong Zhang while (row < i) { 858137fb511SHong Zhang pc = rtmp + row; 859137fb511SHong Zhang if (*pc != 0.0) { 860b051339dSHong Zhang pv = a->a + diag[r[row]]; 861b051339dSHong Zhang pj = aj + diag[r[row]] + 1; 862137fb511SHong Zhang 863137fb511SHong Zhang multiplier = *pc / *pv++; 864137fb511SHong Zhang *pc = multiplier; 865b051339dSHong Zhang nz = ai[r[row]+1] - diag[r[row]] - 1; 866b051339dSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 867dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 868137fb511SHong Zhang } 869b051339dSHong Zhang row = *ajtmp++; 870137fb511SHong Zhang } 871b051339dSHong Zhang /* finished row so overwrite it onto a->a */ 872b051339dSHong Zhang pv = a->a + ai[r[i]] ; 873b051339dSHong Zhang pj = aj + ai[r[i]] ; 874b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 875b051339dSHong Zhang nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */ 876137fb511SHong Zhang 877137fb511SHong Zhang rs = 0.0; 878137fb511SHong Zhang for (j=0; j<nz; j++) { 879b051339dSHong Zhang pv[j] = rtmp[pj[j]]; 880b051339dSHong Zhang if (j != nbdiag) rs += PetscAbsScalar(pv[j]); 881137fb511SHong Zhang } 882137fb511SHong Zhang 883137fb511SHong Zhang /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */ 884137fb511SHong Zhang sctx.rs = rs; 885b051339dSHong Zhang sctx.pv = pv[nbdiag]; 886c6b1f410SHong Zhang ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); 887137fb511SHong Zhang if (newshift == 1) break; 888137fb511SHong Zhang } 889137fb511SHong Zhang 890*b2a402fcSHong Zhang if (info->shifttype == MAT_SHIFT_POSITIVE_DEFINITE && !sctx.useshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 891137fb511SHong Zhang /* 892137fb511SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 893137fb511SHong Zhang * then try lower shift 894137fb511SHong Zhang */ 8950481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 8960481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 8970481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 898d90ac83dSHong Zhang sctx.useshift = PETSC_TRUE; 899137fb511SHong Zhang sctx.nshift++; 900137fb511SHong Zhang } 901d90ac83dSHong Zhang } while (sctx.useshift); 902137fb511SHong Zhang 903137fb511SHong Zhang /* invert diagonal entries for simplier triangular solves */ 904137fb511SHong Zhang for (i=0; i<n; i++) { 905b051339dSHong Zhang a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]]; 906137fb511SHong Zhang } 907137fb511SHong Zhang 908137fb511SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 909137fb511SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 910137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 911b051339dSHong Zhang A->ops->solve = MatSolve_SeqAIJ_InplaceWithPerm; 912ad04f41aSHong Zhang A->ops->solveadd = MatSolveAdd_SeqAIJ_inplace; 913ad04f41aSHong Zhang A->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 914ad04f41aSHong Zhang A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 915b051339dSHong Zhang A->assembled = PETSC_TRUE; 9165c9eb25fSBarry Smith A->preallocated = PETSC_TRUE; 917d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr); 918137fb511SHong Zhang if (sctx.nshift){ 919*b2a402fcSHong Zhang if (info->shifttype == MAT_SHIFT_POSITIVE_DEFINITE) { 9200481f469SBarry 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); 921*b2a402fcSHong Zhang } else if (info->shifttype == MAT_SHIFT_NONZERO) { 922e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 923137fb511SHong Zhang } 924137fb511SHong Zhang } 925137fb511SHong Zhang PetscFunctionReturn(0); 926137fb511SHong Zhang } 927137fb511SHong Zhang 9280a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9294a2ae208SSatish Balay #undef __FUNCT__ 9304a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ" 9310481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info) 932da3a660dSBarry Smith { 933dfbe8321SBarry Smith PetscErrorCode ierr; 934416022c9SBarry Smith Mat C; 935416022c9SBarry Smith 9363a40ed3dSBarry Smith PetscFunctionBegin; 93743244d56SBarry Smith ierr = MatGetFactor(A,MAT_SOLVER_PETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr); 938719d5645SBarry Smith ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr); 939719d5645SBarry Smith ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr); 940db4efbfdSBarry Smith A->ops->solve = C->ops->solve; 941db4efbfdSBarry Smith A->ops->solvetranspose = C->ops->solvetranspose; 942273d9f13SBarry Smith ierr = MatHeaderCopy(A,C);CHKERRQ(ierr); 94352e6d16bSBarry Smith ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr); 9443a40ed3dSBarry Smith PetscFunctionReturn(0); 945da3a660dSBarry Smith } 9460a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9471d098868SHong Zhang 948889a8dedSBarry Smith 9494a2ae208SSatish Balay #undef __FUNCT__ 950ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_inplace" 951ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx) 9528c37ef55SBarry Smith { 953416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 954416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 9556849ba73SBarry Smith PetscErrorCode ierr; 9565d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 9575d0c19d7SBarry Smith PetscInt nz; 9585d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 959dd6ea824SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 960d9fead3dSBarry Smith const PetscScalar *b; 961dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 9628c37ef55SBarry Smith 9633a40ed3dSBarry Smith PetscFunctionBegin; 9643a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 96591bf9adeSBarry Smith 966d9fead3dSBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 9671ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 968416022c9SBarry Smith tmp = a->solve_work; 9698c37ef55SBarry Smith 9703b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 9713b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 9728c37ef55SBarry Smith 9738c37ef55SBarry Smith /* forward solve the lower triangular */ 9748c37ef55SBarry Smith tmp[0] = b[*r++]; 975010a20caSHong Zhang tmps = tmp; 9768c37ef55SBarry Smith for (i=1; i<n; i++) { 977010a20caSHong Zhang v = aa + ai[i] ; 978010a20caSHong Zhang vi = aj + ai[i] ; 97953e7425aSBarry Smith nz = a->diag[i] - ai[i]; 98053e7425aSBarry Smith sum = b[*r++]; 981003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 9828c37ef55SBarry Smith tmp[i] = sum; 9838c37ef55SBarry Smith } 9848c37ef55SBarry Smith 9858c37ef55SBarry Smith /* backward solve the upper triangular */ 9868c37ef55SBarry Smith for (i=n-1; i>=0; i--){ 987010a20caSHong Zhang v = aa + a->diag[i] + 1; 988010a20caSHong Zhang vi = aj + a->diag[i] + 1; 989416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 9908c37ef55SBarry Smith sum = tmp[i]; 991003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 992010a20caSHong Zhang x[*c--] = tmp[i] = sum*aa[a->diag[i]]; 9938c37ef55SBarry Smith } 9948c37ef55SBarry Smith 9953b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 9963b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 997d9fead3dSBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 9981ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 999dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 10003a40ed3dSBarry Smith PetscFunctionReturn(0); 10018c37ef55SBarry Smith } 1002026e39d0SSatish Balay 10032bebee5dSHong Zhang #undef __FUNCT__ 1004ad04f41aSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ_inplace" 1005ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X) 10062bebee5dSHong Zhang { 10072bebee5dSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10082bebee5dSHong Zhang IS iscol = a->col,isrow = a->row; 10092bebee5dSHong Zhang PetscErrorCode ierr; 10105d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10115d0c19d7SBarry Smith PetscInt nz,neq; 10125d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 1013dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1014dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1015db4efbfdSBarry Smith PetscTruth bisdense,xisdense; 10162bebee5dSHong Zhang 10172bebee5dSHong Zhang PetscFunctionBegin; 10182bebee5dSHong Zhang if (!n) PetscFunctionReturn(0); 10192bebee5dSHong Zhang 1020db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1021db4efbfdSBarry Smith if (!bisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 1022db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1023db4efbfdSBarry Smith if (!xisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 1024db4efbfdSBarry Smith 10252bebee5dSHong Zhang ierr = MatGetArray(B,&b);CHKERRQ(ierr); 10262bebee5dSHong Zhang ierr = MatGetArray(X,&x);CHKERRQ(ierr); 10272bebee5dSHong Zhang 10282bebee5dSHong Zhang tmp = a->solve_work; 10292bebee5dSHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10302bebee5dSHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 10312bebee5dSHong Zhang 1032a36b22ccSSatish Balay for (neq=0; neq<B->cmap->n; neq++){ 10332bebee5dSHong Zhang /* forward solve the lower triangular */ 10342bebee5dSHong Zhang tmp[0] = b[r[0]]; 10352bebee5dSHong Zhang tmps = tmp; 10362bebee5dSHong Zhang for (i=1; i<n; i++) { 10372bebee5dSHong Zhang v = aa + ai[i] ; 10382bebee5dSHong Zhang vi = aj + ai[i] ; 10392bebee5dSHong Zhang nz = a->diag[i] - ai[i]; 10402bebee5dSHong Zhang sum = b[r[i]]; 1041003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10422bebee5dSHong Zhang tmp[i] = sum; 10432bebee5dSHong Zhang } 10442bebee5dSHong Zhang /* backward solve the upper triangular */ 10452bebee5dSHong Zhang for (i=n-1; i>=0; i--){ 10462bebee5dSHong Zhang v = aa + a->diag[i] + 1; 10472bebee5dSHong Zhang vi = aj + a->diag[i] + 1; 10482bebee5dSHong Zhang nz = ai[i+1] - a->diag[i] - 1; 10492bebee5dSHong Zhang sum = tmp[i]; 1050003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10512bebee5dSHong Zhang x[c[i]] = tmp[i] = sum*aa[a->diag[i]]; 10522bebee5dSHong Zhang } 10532bebee5dSHong Zhang 10542bebee5dSHong Zhang b += n; 10552bebee5dSHong Zhang x += n; 10562bebee5dSHong Zhang } 10572bebee5dSHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10582bebee5dSHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10592bebee5dSHong Zhang ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 10602bebee5dSHong Zhang ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 1061dc0b31edSSatish Balay ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 10622bebee5dSHong Zhang PetscFunctionReturn(0); 10632bebee5dSHong Zhang } 10642bebee5dSHong Zhang 1065137fb511SHong Zhang #undef __FUNCT__ 106635233d99SShri Abhyankar #define __FUNCT__ "MatMatSolve_SeqAIJ" 106735233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X) 10689bd0847aSShri Abhyankar { 10699bd0847aSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10709bd0847aSShri Abhyankar IS iscol = a->col,isrow = a->row; 10719bd0847aSShri Abhyankar PetscErrorCode ierr; 10729bd0847aSShri Abhyankar PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 10739bd0847aSShri Abhyankar PetscInt nz,neq; 10749bd0847aSShri Abhyankar const PetscInt *rout,*cout,*r,*c; 10759bd0847aSShri Abhyankar PetscScalar *x,*b,*tmp,sum; 10769bd0847aSShri Abhyankar const MatScalar *aa = a->a,*v; 10779bd0847aSShri Abhyankar PetscTruth bisdense,xisdense; 10789bd0847aSShri Abhyankar 10799bd0847aSShri Abhyankar PetscFunctionBegin; 10809bd0847aSShri Abhyankar if (!n) PetscFunctionReturn(0); 10819bd0847aSShri Abhyankar 10829bd0847aSShri Abhyankar ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 10839bd0847aSShri Abhyankar if (!bisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 10849bd0847aSShri Abhyankar ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 10859bd0847aSShri Abhyankar if (!xisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 10869bd0847aSShri Abhyankar 10879bd0847aSShri Abhyankar ierr = MatGetArray(B,&b);CHKERRQ(ierr); 10889bd0847aSShri Abhyankar ierr = MatGetArray(X,&x);CHKERRQ(ierr); 10899bd0847aSShri Abhyankar 10909bd0847aSShri Abhyankar tmp = a->solve_work; 10919bd0847aSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10929bd0847aSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 10939bd0847aSShri Abhyankar 10949bd0847aSShri Abhyankar for (neq=0; neq<B->cmap->n; neq++){ 10959bd0847aSShri Abhyankar /* forward solve the lower triangular */ 10969bd0847aSShri Abhyankar tmp[0] = b[r[0]]; 10979bd0847aSShri Abhyankar v = aa; 10989bd0847aSShri Abhyankar vi = aj; 10999bd0847aSShri Abhyankar for (i=1; i<n; i++) { 11009bd0847aSShri Abhyankar nz = ai[i+1] - ai[i]; 11019bd0847aSShri Abhyankar sum = b[r[i]]; 11029bd0847aSShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 11039bd0847aSShri Abhyankar tmp[i] = sum; 11049bd0847aSShri Abhyankar v += nz; vi += nz; 11059bd0847aSShri Abhyankar } 11069bd0847aSShri Abhyankar 11079bd0847aSShri Abhyankar /* backward solve the upper triangular */ 11089bd0847aSShri Abhyankar for (i=n-1; i>=0; i--){ 11099bd0847aSShri Abhyankar v = aa + adiag[i+1]+1; 11109bd0847aSShri Abhyankar vi = aj + adiag[i+1]+1; 11119bd0847aSShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 11129bd0847aSShri Abhyankar sum = tmp[i]; 11139bd0847aSShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 11149bd0847aSShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 11159bd0847aSShri Abhyankar } 11169bd0847aSShri Abhyankar 11179bd0847aSShri Abhyankar b += n; 11189bd0847aSShri Abhyankar x += n; 11199bd0847aSShri Abhyankar } 11209bd0847aSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 11219bd0847aSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 11229bd0847aSShri Abhyankar ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 11239bd0847aSShri Abhyankar ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 11249bd0847aSShri Abhyankar ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 11259bd0847aSShri Abhyankar PetscFunctionReturn(0); 11269bd0847aSShri Abhyankar } 11279bd0847aSShri Abhyankar 11289bd0847aSShri Abhyankar #undef __FUNCT__ 1129137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm" 1130137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx) 1131137fb511SHong Zhang { 1132137fb511SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1133137fb511SHong Zhang IS iscol = a->col,isrow = a->row; 1134137fb511SHong Zhang PetscErrorCode ierr; 11355d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 11365d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 11375d0c19d7SBarry Smith PetscInt nz,row; 1138dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1139dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1140137fb511SHong Zhang 1141137fb511SHong Zhang PetscFunctionBegin; 1142137fb511SHong Zhang if (!n) PetscFunctionReturn(0); 1143137fb511SHong Zhang 1144137fb511SHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 1145137fb511SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1146137fb511SHong Zhang tmp = a->solve_work; 1147137fb511SHong Zhang 1148137fb511SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1149137fb511SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1150137fb511SHong Zhang 1151137fb511SHong Zhang /* forward solve the lower triangular */ 1152137fb511SHong Zhang tmp[0] = b[*r++]; 1153137fb511SHong Zhang tmps = tmp; 1154137fb511SHong Zhang for (row=1; row<n; row++) { 1155137fb511SHong Zhang i = rout[row]; /* permuted row */ 1156137fb511SHong Zhang v = aa + ai[i] ; 1157137fb511SHong Zhang vi = aj + ai[i] ; 1158137fb511SHong Zhang nz = a->diag[i] - ai[i]; 1159137fb511SHong Zhang sum = b[*r++]; 1160003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1161137fb511SHong Zhang tmp[row] = sum; 1162137fb511SHong Zhang } 1163137fb511SHong Zhang 1164137fb511SHong Zhang /* backward solve the upper triangular */ 1165137fb511SHong Zhang for (row=n-1; row>=0; row--){ 1166137fb511SHong Zhang i = rout[row]; /* permuted row */ 1167137fb511SHong Zhang v = aa + a->diag[i] + 1; 1168137fb511SHong Zhang vi = aj + a->diag[i] + 1; 1169137fb511SHong Zhang nz = ai[i+1] - a->diag[i] - 1; 1170137fb511SHong Zhang sum = tmp[row]; 1171003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1172137fb511SHong Zhang x[*c--] = tmp[row] = sum*aa[a->diag[i]]; 1173137fb511SHong Zhang } 1174137fb511SHong Zhang 1175137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1176137fb511SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1177137fb511SHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 1178137fb511SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1179dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1180137fb511SHong Zhang PetscFunctionReturn(0); 1181137fb511SHong Zhang } 1182137fb511SHong Zhang 1183930ae53cSBarry Smith /* ----------------------------------------------------------- */ 1184a4005a5dSBarry Smith #include "../src/mat/impls/aij/seq/ftn-kernels/fsolve.h" 11854a2ae208SSatish Balay #undef __FUNCT__ 1186ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_inplace" 1187ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx) 1188930ae53cSBarry Smith { 1189930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 11906849ba73SBarry Smith PetscErrorCode ierr; 1191d0f46423SBarry Smith PetscInt n = A->rmap->n; 1192b377110cSBarry Smith const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag; 1193356650c2SBarry Smith PetscScalar *x; 1194356650c2SBarry Smith const PetscScalar *b; 1195dd6ea824SBarry Smith const MatScalar *aa = a->a; 1196aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 1197356650c2SBarry Smith PetscInt adiag_i,i,nz,ai_i; 1198b377110cSBarry Smith const PetscInt *vi; 1199dd6ea824SBarry Smith const MatScalar *v; 1200dd6ea824SBarry Smith PetscScalar sum; 1201d85afc42SSatish Balay #endif 1202930ae53cSBarry Smith 12033a40ed3dSBarry Smith PetscFunctionBegin; 12043a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 1205930ae53cSBarry Smith 1206356650c2SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12071ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1208930ae53cSBarry Smith 1209aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 12101c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 12111c4feecaSSatish Balay #else 1212930ae53cSBarry Smith /* forward solve the lower triangular */ 1213930ae53cSBarry Smith x[0] = b[0]; 1214930ae53cSBarry Smith for (i=1; i<n; i++) { 1215930ae53cSBarry Smith ai_i = ai[i]; 1216930ae53cSBarry Smith v = aa + ai_i; 1217930ae53cSBarry Smith vi = aj + ai_i; 1218930ae53cSBarry Smith nz = adiag[i] - ai_i; 1219930ae53cSBarry Smith sum = b[i]; 1220003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1221930ae53cSBarry Smith x[i] = sum; 1222930ae53cSBarry Smith } 1223930ae53cSBarry Smith 1224930ae53cSBarry Smith /* backward solve the upper triangular */ 1225930ae53cSBarry Smith for (i=n-1; i>=0; i--){ 1226930ae53cSBarry Smith adiag_i = adiag[i]; 1227d708749eSBarry Smith v = aa + adiag_i + 1; 1228d708749eSBarry Smith vi = aj + adiag_i + 1; 1229930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 1230930ae53cSBarry Smith sum = x[i]; 1231003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1232930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 1233930ae53cSBarry Smith } 12341c4feecaSSatish Balay #endif 1235dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1236356650c2SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12371ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 12383a40ed3dSBarry Smith PetscFunctionReturn(0); 1239930ae53cSBarry Smith } 1240930ae53cSBarry Smith 12414a2ae208SSatish Balay #undef __FUNCT__ 1242ad04f41aSHong Zhang #define __FUNCT__ "MatSolveAdd_SeqAIJ_inplace" 1243ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx) 1244da3a660dSBarry Smith { 1245416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1246416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 12476849ba73SBarry Smith PetscErrorCode ierr; 124804fbf559SBarry Smith PetscInt i, n = A->rmap->n,j; 12495d0c19d7SBarry Smith PetscInt nz; 125004fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j; 125104fbf559SBarry Smith PetscScalar *x,*tmp,sum; 125204fbf559SBarry Smith const PetscScalar *b; 1253dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1254da3a660dSBarry Smith 12553a40ed3dSBarry Smith PetscFunctionBegin; 125678b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 1257da3a660dSBarry Smith 125804fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12591ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1260416022c9SBarry Smith tmp = a->solve_work; 1261da3a660dSBarry Smith 12623b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12633b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1264da3a660dSBarry Smith 1265da3a660dSBarry Smith /* forward solve the lower triangular */ 1266da3a660dSBarry Smith tmp[0] = b[*r++]; 1267da3a660dSBarry Smith for (i=1; i<n; i++) { 1268010a20caSHong Zhang v = aa + ai[i] ; 1269010a20caSHong Zhang vi = aj + ai[i] ; 1270416022c9SBarry Smith nz = a->diag[i] - ai[i]; 1271da3a660dSBarry Smith sum = b[*r++]; 127204fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1273da3a660dSBarry Smith tmp[i] = sum; 1274da3a660dSBarry Smith } 1275da3a660dSBarry Smith 1276da3a660dSBarry Smith /* backward solve the upper triangular */ 1277da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1278010a20caSHong Zhang v = aa + a->diag[i] + 1; 1279010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1280416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 1281da3a660dSBarry Smith sum = tmp[i]; 128204fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1283010a20caSHong Zhang tmp[i] = sum*aa[a->diag[i]]; 1284da3a660dSBarry Smith x[*c--] += tmp[i]; 1285da3a660dSBarry Smith } 1286da3a660dSBarry Smith 12873b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 12883b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 128904fbf559SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 12901ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1291dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1292898183ecSLois Curfman McInnes 12933a40ed3dSBarry Smith PetscFunctionReturn(0); 1294da3a660dSBarry Smith } 129504fbf559SBarry Smith 12964a2ae208SSatish Balay #undef __FUNCT__ 129735233d99SShri Abhyankar #define __FUNCT__ "MatSolveAdd_SeqAIJ" 129835233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 12993c0119dfSShri Abhyankar { 13003c0119dfSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13013c0119dfSShri Abhyankar IS iscol = a->col,isrow = a->row; 13023c0119dfSShri Abhyankar PetscErrorCode ierr; 13033c0119dfSShri Abhyankar PetscInt i, n = A->rmap->n,j; 13043c0119dfSShri Abhyankar PetscInt nz; 13053c0119dfSShri Abhyankar const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 13063c0119dfSShri Abhyankar PetscScalar *x,*tmp,sum; 13073c0119dfSShri Abhyankar const PetscScalar *b; 13083c0119dfSShri Abhyankar const MatScalar *aa = a->a,*v; 13093c0119dfSShri Abhyankar 13103c0119dfSShri Abhyankar PetscFunctionBegin; 13113c0119dfSShri Abhyankar if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 13123c0119dfSShri Abhyankar 13133c0119dfSShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 13143c0119dfSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 13153c0119dfSShri Abhyankar tmp = a->solve_work; 13163c0119dfSShri Abhyankar 13173c0119dfSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13183c0119dfSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 13193c0119dfSShri Abhyankar 13203c0119dfSShri Abhyankar /* forward solve the lower triangular */ 13213c0119dfSShri Abhyankar tmp[0] = b[r[0]]; 13223c0119dfSShri Abhyankar v = aa; 13233c0119dfSShri Abhyankar vi = aj; 13243c0119dfSShri Abhyankar for (i=1; i<n; i++) { 13253c0119dfSShri Abhyankar nz = ai[i+1] - ai[i]; 13263c0119dfSShri Abhyankar sum = b[r[i]]; 13273c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 13283c0119dfSShri Abhyankar tmp[i] = sum; 13293c0119dfSShri Abhyankar v += nz; vi += nz; 13303c0119dfSShri Abhyankar } 13313c0119dfSShri Abhyankar 13323c0119dfSShri Abhyankar /* backward solve the upper triangular */ 13333c0119dfSShri Abhyankar v = aa + adiag[n-1]; 13343c0119dfSShri Abhyankar vi = aj + adiag[n-1]; 13353c0119dfSShri Abhyankar for (i=n-1; i>=0; i--){ 13363c0119dfSShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 13373c0119dfSShri Abhyankar sum = tmp[i]; 13383c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 13393c0119dfSShri Abhyankar tmp[i] = sum*v[nz]; 13403c0119dfSShri Abhyankar x[c[i]] += tmp[i]; 13413c0119dfSShri Abhyankar v += nz+1; vi += nz+1; 13423c0119dfSShri Abhyankar } 13433c0119dfSShri Abhyankar 13443c0119dfSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13453c0119dfSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 13463c0119dfSShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 13473c0119dfSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 13483c0119dfSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 13493c0119dfSShri Abhyankar 13503c0119dfSShri Abhyankar PetscFunctionReturn(0); 13513c0119dfSShri Abhyankar } 13523c0119dfSShri Abhyankar 13533c0119dfSShri Abhyankar #undef __FUNCT__ 1354ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTranspose_SeqAIJ_inplace" 1355ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx) 1356da3a660dSBarry Smith { 1357416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13582235e667SBarry Smith IS iscol = a->col,isrow = a->row; 13596849ba73SBarry Smith PetscErrorCode ierr; 136004fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 136104fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 136204fbf559SBarry Smith PetscInt nz; 136304fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1364dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 136504fbf559SBarry Smith const PetscScalar *b; 1366da3a660dSBarry Smith 13673a40ed3dSBarry Smith PetscFunctionBegin; 136804fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 13691ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1370416022c9SBarry Smith tmp = a->solve_work; 1371da3a660dSBarry Smith 13722235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13732235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1374da3a660dSBarry Smith 1375da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 13762235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1377da3a660dSBarry Smith 1378da3a660dSBarry Smith /* forward solve the U^T */ 1379da3a660dSBarry Smith for (i=0; i<n; i++) { 1380010a20caSHong Zhang v = aa + diag[i] ; 1381010a20caSHong Zhang vi = aj + diag[i] + 1; 1382f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 1383c38d4ed2SBarry Smith s1 = tmp[i]; 1384ef66eb69SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 138504fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1386c38d4ed2SBarry Smith tmp[i] = s1; 1387da3a660dSBarry Smith } 1388da3a660dSBarry Smith 1389da3a660dSBarry Smith /* backward solve the L^T */ 1390da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1391010a20caSHong Zhang v = aa + diag[i] - 1 ; 1392010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1393f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 1394f1af5d2fSBarry Smith s1 = tmp[i]; 139504fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 1396da3a660dSBarry Smith } 1397da3a660dSBarry Smith 1398da3a660dSBarry Smith /* copy tmp into x according to permutation */ 1399591294e4SBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1400da3a660dSBarry Smith 14012235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 14022235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 140304fbf559SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 14041ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1405da3a660dSBarry Smith 1406dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 14073a40ed3dSBarry Smith PetscFunctionReturn(0); 1408da3a660dSBarry Smith } 1409da3a660dSBarry Smith 1410d1fa9404SShri Abhyankar #undef __FUNCT__ 141135233d99SShri Abhyankar #define __FUNCT__ "MatSolveTranspose_SeqAIJ" 141235233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 1413d1fa9404SShri Abhyankar { 1414d1fa9404SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1415d1fa9404SShri Abhyankar IS iscol = a->col,isrow = a->row; 1416d1fa9404SShri Abhyankar PetscErrorCode ierr; 1417d1fa9404SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1418d1fa9404SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1419d1fa9404SShri Abhyankar PetscInt nz; 1420d1fa9404SShri Abhyankar PetscScalar *x,*tmp,s1; 1421d1fa9404SShri Abhyankar const MatScalar *aa = a->a,*v; 1422d1fa9404SShri Abhyankar const PetscScalar *b; 1423d1fa9404SShri Abhyankar 1424d1fa9404SShri Abhyankar PetscFunctionBegin; 1425d1fa9404SShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 1426d1fa9404SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1427d1fa9404SShri Abhyankar tmp = a->solve_work; 1428d1fa9404SShri Abhyankar 1429d1fa9404SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1430d1fa9404SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1431d1fa9404SShri Abhyankar 1432d1fa9404SShri Abhyankar /* copy the b into temp work space according to permutation */ 1433d1fa9404SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1434d1fa9404SShri Abhyankar 1435d1fa9404SShri Abhyankar /* forward solve the U^T */ 1436d1fa9404SShri Abhyankar for (i=0; i<n; i++) { 1437d1fa9404SShri Abhyankar v = aa + adiag[i+1] + 1; 1438d1fa9404SShri Abhyankar vi = aj + adiag[i+1] + 1; 1439d1fa9404SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1440d1fa9404SShri Abhyankar s1 = tmp[i]; 1441d1fa9404SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1442d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1443d1fa9404SShri Abhyankar tmp[i] = s1; 1444d1fa9404SShri Abhyankar } 1445d1fa9404SShri Abhyankar 1446d1fa9404SShri Abhyankar /* backward solve the L^T */ 1447d1fa9404SShri Abhyankar for (i=n-1; i>=0; i--){ 1448d1fa9404SShri Abhyankar v = aa + ai[i]; 1449d1fa9404SShri Abhyankar vi = aj + ai[i]; 1450d1fa9404SShri Abhyankar nz = ai[i+1] - ai[i]; 1451d1fa9404SShri Abhyankar s1 = tmp[i]; 1452d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1453d1fa9404SShri Abhyankar } 1454d1fa9404SShri Abhyankar 1455d1fa9404SShri Abhyankar /* copy tmp into x according to permutation */ 1456d1fa9404SShri Abhyankar for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1457d1fa9404SShri Abhyankar 1458d1fa9404SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1459d1fa9404SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1460d1fa9404SShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 1461d1fa9404SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1462d1fa9404SShri Abhyankar 1463d1fa9404SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1464d1fa9404SShri Abhyankar PetscFunctionReturn(0); 1465d1fa9404SShri Abhyankar } 1466d1fa9404SShri Abhyankar 14674a2ae208SSatish Balay #undef __FUNCT__ 1468ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ_inplace" 1469ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx) 1470da3a660dSBarry Smith { 1471416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 14722235e667SBarry Smith IS iscol = a->col,isrow = a->row; 14736849ba73SBarry Smith PetscErrorCode ierr; 147404fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 147504fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 147604fbf559SBarry Smith PetscInt nz; 147704fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1478dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 147904fbf559SBarry Smith const PetscScalar *b; 14806abc6512SBarry Smith 14813a40ed3dSBarry Smith PetscFunctionBegin; 148223bc6035SMatthew Knepley if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 148304fbf559SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 14841ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1485416022c9SBarry Smith tmp = a->solve_work; 14866abc6512SBarry Smith 14872235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 14882235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 14896abc6512SBarry Smith 14906abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 14912235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 14926abc6512SBarry Smith 14936abc6512SBarry Smith /* forward solve the U^T */ 14946abc6512SBarry Smith for (i=0; i<n; i++) { 1495010a20caSHong Zhang v = aa + diag[i] ; 1496010a20caSHong Zhang vi = aj + diag[i] + 1; 1497f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 149804fbf559SBarry Smith s1 = tmp[i]; 149904fbf559SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 150004fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 150104fbf559SBarry Smith tmp[i] = s1; 15026abc6512SBarry Smith } 15036abc6512SBarry Smith 15046abc6512SBarry Smith /* backward solve the L^T */ 15056abc6512SBarry Smith for (i=n-1; i>=0; i--){ 1506010a20caSHong Zhang v = aa + diag[i] - 1 ; 1507010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1508f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 150904fbf559SBarry Smith s1 = tmp[i]; 151004fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 15116abc6512SBarry Smith } 15126abc6512SBarry Smith 15136abc6512SBarry Smith /* copy tmp into x according to permutation */ 15146abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 15156abc6512SBarry Smith 15162235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 15172235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 151804fbf559SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 15191ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 15206abc6512SBarry Smith 152104fbf559SBarry Smith ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 15223a40ed3dSBarry Smith PetscFunctionReturn(0); 1523da3a660dSBarry Smith } 152404fbf559SBarry Smith 1525533e6011SShri Abhyankar #undef __FUNCT__ 152635233d99SShri Abhyankar #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ" 152735233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 1528533e6011SShri Abhyankar { 1529533e6011SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1530533e6011SShri Abhyankar IS iscol = a->col,isrow = a->row; 1531533e6011SShri Abhyankar PetscErrorCode ierr; 1532533e6011SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1533533e6011SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1534533e6011SShri Abhyankar PetscInt nz; 1535533e6011SShri Abhyankar PetscScalar *x,*tmp,s1; 1536533e6011SShri Abhyankar const MatScalar *aa = a->a,*v; 1537533e6011SShri Abhyankar const PetscScalar *b; 1538533e6011SShri Abhyankar 1539533e6011SShri Abhyankar PetscFunctionBegin; 1540533e6011SShri Abhyankar if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 1541533e6011SShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 1542533e6011SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1543533e6011SShri Abhyankar tmp = a->solve_work; 1544533e6011SShri Abhyankar 1545533e6011SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1546533e6011SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1547533e6011SShri Abhyankar 1548533e6011SShri Abhyankar /* copy the b into temp work space according to permutation */ 1549533e6011SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1550533e6011SShri Abhyankar 1551533e6011SShri Abhyankar /* forward solve the U^T */ 1552533e6011SShri Abhyankar for (i=0; i<n; i++) { 1553533e6011SShri Abhyankar v = aa + adiag[i+1] + 1; 1554533e6011SShri Abhyankar vi = aj + adiag[i+1] + 1; 1555533e6011SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1556533e6011SShri Abhyankar s1 = tmp[i]; 1557533e6011SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1558533e6011SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1559533e6011SShri Abhyankar tmp[i] = s1; 1560533e6011SShri Abhyankar } 1561533e6011SShri Abhyankar 1562533e6011SShri Abhyankar 1563533e6011SShri Abhyankar /* backward solve the L^T */ 1564533e6011SShri Abhyankar for (i=n-1; i>=0; i--){ 1565533e6011SShri Abhyankar v = aa + ai[i] ; 1566533e6011SShri Abhyankar vi = aj + ai[i]; 1567533e6011SShri Abhyankar nz = ai[i+1] - ai[i]; 1568533e6011SShri Abhyankar s1 = tmp[i]; 1569c5e3b2a3SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1570533e6011SShri Abhyankar } 1571533e6011SShri Abhyankar 1572533e6011SShri Abhyankar /* copy tmp into x according to permutation */ 1573533e6011SShri Abhyankar for (i=0; i<n; i++) x[r[i]] += tmp[i]; 1574533e6011SShri Abhyankar 1575533e6011SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1576533e6011SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1577533e6011SShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 1578533e6011SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1579533e6011SShri Abhyankar 1580533e6011SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1581533e6011SShri Abhyankar PetscFunctionReturn(0); 1582533e6011SShri Abhyankar } 1583533e6011SShri Abhyankar 15849e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 1585b47f5a65SHong Zhang 1586f77e22a1SHong Zhang EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscTruth); 158708480c60SBarry Smith 15888fc3a347SHong Zhang /* 15898a76ed4fSHong Zhang ilu() under revised new data structure. 1590813a1f2bSShri Abhyankar Factored arrays bj and ba are stored as 1591813a1f2bSShri Abhyankar L(0,:), L(1,:), ...,L(n-1,:), U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:) 1592813a1f2bSShri Abhyankar 1593813a1f2bSShri Abhyankar bi=fact->i is an array of size n+1, in which 1594813a1f2bSShri Abhyankar bi+ 1595baabb860SHong Zhang bi[i]: points to 1st entry of L(i,:),i=0,...,n-1 1596baabb860SHong Zhang bi[n]: points to L(n-1,n-1)+1 1597baabb860SHong Zhang 1598813a1f2bSShri Abhyankar bdiag=fact->diag is an array of size n+1,in which 1599baabb860SHong Zhang bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1 1600a24f213cSHong Zhang bdiag[n]: points to entry of U(n-1,0)-1 1601813a1f2bSShri Abhyankar 1602813a1f2bSShri Abhyankar U(i,:) contains bdiag[i] as its last entry, i.e., 1603813a1f2bSShri Abhyankar U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 1604813a1f2bSShri Abhyankar */ 1605813a1f2bSShri Abhyankar #undef __FUNCT__ 160635233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0" 160735233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1608813a1f2bSShri Abhyankar { 1609813a1f2bSShri Abhyankar 1610813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1611813a1f2bSShri Abhyankar PetscErrorCode ierr; 1612bbd65245SShri Abhyankar const PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag; 1613bbd65245SShri Abhyankar PetscInt i,j,k=0,nz,*bi,*bj,*bdiag; 16141df811f5SHong Zhang PetscTruth missing; 16151df811f5SHong Zhang IS isicol; 1616813a1f2bSShri Abhyankar 1617813a1f2bSShri Abhyankar PetscFunctionBegin; 16181df811f5SHong 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); 16191df811f5SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 16201df811f5SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 16211df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 16221df811f5SHong Zhang 1623813a1f2bSShri Abhyankar ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr); 1624813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1625813a1f2bSShri Abhyankar 1626813a1f2bSShri Abhyankar /* allocate matrix arrays for new data structure */ 1627813a1f2bSShri Abhyankar ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,n+1,PetscInt,&b->i);CHKERRQ(ierr); 1628813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1629813a1f2bSShri Abhyankar b->singlemalloc = PETSC_TRUE; 1630813a1f2bSShri Abhyankar if (!b->diag){ 1631813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr); 1632813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1633813a1f2bSShri Abhyankar } 1634813a1f2bSShri Abhyankar bdiag = b->diag; 1635813a1f2bSShri Abhyankar 1636813a1f2bSShri Abhyankar if (n > 0) { 1637813a1f2bSShri Abhyankar ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr); 1638813a1f2bSShri Abhyankar } 1639813a1f2bSShri Abhyankar 1640813a1f2bSShri Abhyankar /* set bi and bj with new data structure */ 1641813a1f2bSShri Abhyankar bi = b->i; 1642813a1f2bSShri Abhyankar bj = b->j; 1643813a1f2bSShri Abhyankar 1644813a1f2bSShri Abhyankar /* L part */ 1645e60cf9a0SBarry Smith bi[0] = 0; 1646813a1f2bSShri Abhyankar for (i=0; i<n; i++){ 1647813a1f2bSShri Abhyankar nz = adiag[i] - ai[i]; 1648813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nz; 1649813a1f2bSShri Abhyankar aj = a->j + ai[i]; 1650813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 1651bbd65245SShri Abhyankar /* *bj = aj[j]; bj++; */ 1652bbd65245SShri Abhyankar bj[k++] = aj[j]; 1653813a1f2bSShri Abhyankar } 1654813a1f2bSShri Abhyankar } 1655813a1f2bSShri Abhyankar 1656813a1f2bSShri Abhyankar /* U part */ 165762fbd6afSShri Abhyankar bdiag[n] = bi[n]-1; 1658813a1f2bSShri Abhyankar for (i=n-1; i>=0; i--){ 1659813a1f2bSShri Abhyankar nz = ai[i+1] - adiag[i] - 1; 1660813a1f2bSShri Abhyankar aj = a->j + adiag[i] + 1; 1661813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 1662bbd65245SShri Abhyankar /* *bj = aj[j]; bj++; */ 1663bbd65245SShri Abhyankar bj[k++] = aj[j]; 1664813a1f2bSShri Abhyankar } 1665813a1f2bSShri Abhyankar /* diag[i] */ 1666bbd65245SShri Abhyankar /* *bj = i; bj++; */ 1667bbd65245SShri Abhyankar bj[k++] = i; 1668a24f213cSHong Zhang bdiag[i] = bdiag[i+1] + nz + 1; 1669813a1f2bSShri Abhyankar } 16701df811f5SHong Zhang 16711df811f5SHong Zhang fact->factor = MAT_FACTOR_ILU; 16721df811f5SHong Zhang fact->info.factor_mallocs = 0; 16731df811f5SHong Zhang fact->info.fill_ratio_given = info->fill; 16741df811f5SHong Zhang fact->info.fill_ratio_needed = 1.0; 167535233d99SShri Abhyankar fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 16761df811f5SHong Zhang 16771df811f5SHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 16781df811f5SHong Zhang b->row = isrow; 16791df811f5SHong Zhang b->col = iscol; 16801df811f5SHong Zhang b->icol = isicol; 16811df811f5SHong Zhang ierr = PetscMalloc((fact->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 16821df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 16831df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1684813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1685813a1f2bSShri Abhyankar } 1686813a1f2bSShri Abhyankar 1687813a1f2bSShri Abhyankar #undef __FUNCT__ 168835233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ" 168935233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1690813a1f2bSShri Abhyankar { 1691813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1692813a1f2bSShri Abhyankar IS isicol; 1693813a1f2bSShri Abhyankar PetscErrorCode ierr; 1694813a1f2bSShri Abhyankar const PetscInt *r,*ic; 16951df811f5SHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j; 1696813a1f2bSShri Abhyankar PetscInt *bi,*cols,nnz,*cols_lvl; 1697813a1f2bSShri Abhyankar PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 1698813a1f2bSShri Abhyankar PetscInt i,levels,diagonal_fill; 1699813a1f2bSShri Abhyankar PetscTruth col_identity,row_identity; 1700813a1f2bSShri Abhyankar PetscReal f; 1701813a1f2bSShri Abhyankar PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 1702813a1f2bSShri Abhyankar PetscBT lnkbt; 1703813a1f2bSShri Abhyankar PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 1704813a1f2bSShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1705813a1f2bSShri Abhyankar PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 1706813a1f2bSShri Abhyankar 1707813a1f2bSShri Abhyankar PetscFunctionBegin; 17082169352eSHong Zhang /* // Testing new data structure for MatSolve() 17092169352eSHong Zhang PetscTruth olddatastruct=PETSC_FALSE 171035233d99SShri Abhyankar ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr); 171135233d99SShri Abhyankar if(olddatastruct){ 1712ad04f41aSHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_inplace(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1713ad04f41aSHong Zhang PetscFunctionReturn(0); 1714ad04f41aSHong Zhang } 17152169352eSHong Zhang */ 1716ad04f41aSHong Zhang 1717813a1f2bSShri Abhyankar levels = (PetscInt)info->levels; 1718813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 1719813a1f2bSShri Abhyankar ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 1720813a1f2bSShri Abhyankar 1721813a1f2bSShri Abhyankar if (!levels && row_identity && col_identity) { 1722813a1f2bSShri Abhyankar /* special case: ilu(0) with natural ordering */ 172335233d99SShri Abhyankar ierr = MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1724019b515eSShri Abhyankar ierr = Mat_CheckInode_FactorLU(fact,PETSC_FALSE);CHKERRQ(ierr); 1725813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1726813a1f2bSShri Abhyankar } 1727813a1f2bSShri Abhyankar 17281df811f5SHong 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); 17291df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 1730813a1f2bSShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1731813a1f2bSShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 1732813a1f2bSShri Abhyankar 17331bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 17341bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 17351bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1736e60cf9a0SBarry Smith bi[0] = bdiag[0] = 0; 1737813a1f2bSShri Abhyankar 17380e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr); 1739813a1f2bSShri Abhyankar 1740813a1f2bSShri Abhyankar /* create a linked list for storing column indices of the active row */ 1741813a1f2bSShri Abhyankar nlnk = n + 1; 1742813a1f2bSShri Abhyankar ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1743813a1f2bSShri Abhyankar 1744813a1f2bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 17451df811f5SHong Zhang f = info->fill; 17461df811f5SHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 1747813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1748813a1f2bSShri Abhyankar current_space = free_space; 1749813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 1750813a1f2bSShri Abhyankar current_space_lvl = free_space_lvl; 1751813a1f2bSShri Abhyankar 1752813a1f2bSShri Abhyankar for (i=0; i<n; i++) { 1753813a1f2bSShri Abhyankar nzi = 0; 1754813a1f2bSShri Abhyankar /* copy current row into linked list */ 1755813a1f2bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 1756813a1f2bSShri 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); 1757813a1f2bSShri Abhyankar cols = aj + ai[r[i]]; 1758813a1f2bSShri Abhyankar lnk[i] = -1; /* marker to indicate if diagonal exists */ 1759813a1f2bSShri Abhyankar ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1760813a1f2bSShri Abhyankar nzi += nlnk; 1761813a1f2bSShri Abhyankar 1762813a1f2bSShri Abhyankar /* make sure diagonal entry is included */ 1763813a1f2bSShri Abhyankar if (diagonal_fill && lnk[i] == -1) { 1764813a1f2bSShri Abhyankar fm = n; 1765813a1f2bSShri Abhyankar while (lnk[fm] < i) fm = lnk[fm]; 1766813a1f2bSShri Abhyankar lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 1767813a1f2bSShri Abhyankar lnk[fm] = i; 1768813a1f2bSShri Abhyankar lnk_lvl[i] = 0; 1769813a1f2bSShri Abhyankar nzi++; dcount++; 1770813a1f2bSShri Abhyankar } 1771813a1f2bSShri Abhyankar 1772813a1f2bSShri Abhyankar /* add pivot rows into the active row */ 1773813a1f2bSShri Abhyankar nzbd = 0; 1774813a1f2bSShri Abhyankar prow = lnk[n]; 1775813a1f2bSShri Abhyankar while (prow < i) { 1776813a1f2bSShri Abhyankar nnz = bdiag[prow]; 1777813a1f2bSShri Abhyankar cols = bj_ptr[prow] + nnz + 1; 1778813a1f2bSShri Abhyankar cols_lvl = bjlvl_ptr[prow] + nnz + 1; 1779813a1f2bSShri Abhyankar nnz = bi[prow+1] - bi[prow] - nnz - 1; 1780813a1f2bSShri Abhyankar ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 1781813a1f2bSShri Abhyankar nzi += nlnk; 1782813a1f2bSShri Abhyankar prow = lnk[prow]; 1783813a1f2bSShri Abhyankar nzbd++; 1784813a1f2bSShri Abhyankar } 1785813a1f2bSShri Abhyankar bdiag[i] = nzbd; 1786813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nzi; 1787813a1f2bSShri Abhyankar 1788813a1f2bSShri Abhyankar /* if free space is not available, make more free space */ 1789813a1f2bSShri Abhyankar if (current_space->local_remaining<nzi) { 1790813a1f2bSShri Abhyankar nnz = 2*nzi*(n - i); /* estimated and max additional space needed */ 1791813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1792813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 1793813a1f2bSShri Abhyankar reallocs++; 1794813a1f2bSShri Abhyankar } 1795813a1f2bSShri Abhyankar 1796813a1f2bSShri Abhyankar /* copy data into free_space and free_space_lvl, then initialize lnk */ 1797813a1f2bSShri Abhyankar ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 1798813a1f2bSShri Abhyankar bj_ptr[i] = current_space->array; 1799813a1f2bSShri Abhyankar bjlvl_ptr[i] = current_space_lvl->array; 1800813a1f2bSShri Abhyankar 1801813a1f2bSShri Abhyankar /* make sure the active row i has diagonal entry */ 1802813a1f2bSShri Abhyankar if (*(bj_ptr[i]+bdiag[i]) != i) { 1803813a1f2bSShri Abhyankar SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 1804813a1f2bSShri Abhyankar try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 1805813a1f2bSShri Abhyankar } 1806813a1f2bSShri Abhyankar 1807813a1f2bSShri Abhyankar current_space->array += nzi; 1808813a1f2bSShri Abhyankar current_space->local_used += nzi; 1809813a1f2bSShri Abhyankar current_space->local_remaining -= nzi; 1810813a1f2bSShri Abhyankar current_space_lvl->array += nzi; 1811813a1f2bSShri Abhyankar current_space_lvl->local_used += nzi; 1812813a1f2bSShri Abhyankar current_space_lvl->local_remaining -= nzi; 1813813a1f2bSShri Abhyankar } 1814813a1f2bSShri Abhyankar 1815813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1816813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 1817813a1f2bSShri Abhyankar 1818813a1f2bSShri Abhyankar /* destroy list of free space and other temporary arrays */ 1819813a1f2bSShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 1820813a1f2bSShri Abhyankar 1821813a1f2bSShri Abhyankar /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 18222ce24eb6SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 1823813a1f2bSShri Abhyankar 1824813a1f2bSShri Abhyankar ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1825813a1f2bSShri Abhyankar ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 18260e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 1827813a1f2bSShri Abhyankar 1828813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO) 1829813a1f2bSShri Abhyankar { 1830813a1f2bSShri Abhyankar PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 1831813a1f2bSShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 1832813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1833813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 1834813a1f2bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1835813a1f2bSShri Abhyankar if (diagonal_fill) { 1836813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 1837813a1f2bSShri Abhyankar } 1838813a1f2bSShri Abhyankar } 1839813a1f2bSShri Abhyankar #endif 1840813a1f2bSShri Abhyankar 1841813a1f2bSShri Abhyankar /* put together the new matrix */ 1842813a1f2bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1843813a1f2bSShri Abhyankar ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 1844813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1845813a1f2bSShri Abhyankar b->free_a = PETSC_TRUE; 1846813a1f2bSShri Abhyankar b->free_ij = PETSC_TRUE; 1847813a1f2bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 1848f268cba8SShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 1849813a1f2bSShri Abhyankar b->j = bj; 1850813a1f2bSShri Abhyankar b->i = bi; 1851813a1f2bSShri Abhyankar b->diag = bdiag; 1852813a1f2bSShri Abhyankar b->ilen = 0; 1853813a1f2bSShri Abhyankar b->imax = 0; 1854813a1f2bSShri Abhyankar b->row = isrow; 1855813a1f2bSShri Abhyankar b->col = iscol; 1856813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1857813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1858813a1f2bSShri Abhyankar b->icol = isicol; 1859813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1860813a1f2bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. 1861813a1f2bSShri Abhyankar Allocate bdiag, solve_work, new a, new j */ 1862f268cba8SShri Abhyankar ierr = PetscLogObjectMemory(fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 1863f268cba8SShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 1864813a1f2bSShri Abhyankar (fact)->info.factor_mallocs = reallocs; 1865813a1f2bSShri Abhyankar (fact)->info.fill_ratio_given = f; 1866f268cba8SShri Abhyankar (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 186735233d99SShri Abhyankar (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1868019b515eSShri Abhyankar ierr = Mat_CheckInode_FactorLU(fact,PETSC_FALSE);CHKERRQ(ierr); 1869813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1870813a1f2bSShri Abhyankar } 1871813a1f2bSShri Abhyankar 18726516239fSHong Zhang #undef __FUNCT__ 1873ad04f41aSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_inplace" 1874ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 18756516239fSHong Zhang { 18766516239fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 18776516239fSHong Zhang IS isicol; 18786516239fSHong Zhang PetscErrorCode ierr; 18796516239fSHong Zhang const PetscInt *r,*ic; 18806516239fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 18816516239fSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 18826516239fSHong Zhang PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 18836516239fSHong Zhang PetscInt i,levels,diagonal_fill; 18846516239fSHong Zhang PetscTruth col_identity,row_identity; 18856516239fSHong Zhang PetscReal f; 18866516239fSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 18876516239fSHong Zhang PetscBT lnkbt; 18886516239fSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 18896516239fSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 18906516239fSHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 18916516239fSHong Zhang PetscTruth missing; 18926516239fSHong Zhang 18936516239fSHong Zhang PetscFunctionBegin; 18946516239fSHong 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); 18956516239fSHong Zhang f = info->fill; 18966516239fSHong Zhang levels = (PetscInt)info->levels; 18976516239fSHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 18986516239fSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 18996516239fSHong Zhang 19006516239fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 19016516239fSHong Zhang ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 19026516239fSHong Zhang if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */ 1903f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr); 1904ad04f41aSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 1905c9c72f8fSHong Zhang 1906719d5645SBarry Smith fact->factor = MAT_FACTOR_ILU; 1907719d5645SBarry Smith (fact)->info.factor_mallocs = 0; 1908719d5645SBarry Smith (fact)->info.fill_ratio_given = info->fill; 1909719d5645SBarry Smith (fact)->info.fill_ratio_needed = 1.0; 1910719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 19118ef3462fSBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 191209f38230SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 191308480c60SBarry Smith b->row = isrow; 191408480c60SBarry Smith b->col = iscol; 191582bf6240SBarry Smith b->icol = isicol; 1916719d5645SBarry Smith ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1917c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1918c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 19194108e4d5SBarry Smith ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); 19203a40ed3dSBarry Smith PetscFunctionReturn(0); 192108480c60SBarry Smith } 192208480c60SBarry Smith 1923898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1924898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 19259e25ed09SBarry Smith 19261bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 19271bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 19281bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1929a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 19306cf997b0SBarry Smith 19310e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr); 19325a8e39fbSHong Zhang 19335a8e39fbSHong Zhang /* create a linked list for storing column indices of the active row */ 19345a8e39fbSHong Zhang nlnk = n + 1; 19355a8e39fbSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19365a8e39fbSHong Zhang 19375a8e39fbSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 1938a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 19395a8e39fbSHong Zhang current_space = free_space; 1940a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 19415a8e39fbSHong Zhang current_space_lvl = free_space_lvl; 19425a8e39fbSHong Zhang 19435a8e39fbSHong Zhang for (i=0; i<n; i++) { 19445a8e39fbSHong Zhang nzi = 0; 19456cf997b0SBarry Smith /* copy current row into linked list */ 19465a8e39fbSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 19473b4a8b6dSBarry 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); 19485a8e39fbSHong Zhang cols = aj + ai[r[i]]; 19495a8e39fbSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 19505a8e39fbSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19515a8e39fbSHong Zhang nzi += nlnk; 19526cf997b0SBarry Smith 19536cf997b0SBarry Smith /* make sure diagonal entry is included */ 19545a8e39fbSHong Zhang if (diagonal_fill && lnk[i] == -1) { 19556cf997b0SBarry Smith fm = n; 19565a8e39fbSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 19575a8e39fbSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 19585a8e39fbSHong Zhang lnk[fm] = i; 19595a8e39fbSHong Zhang lnk_lvl[i] = 0; 19605a8e39fbSHong Zhang nzi++; dcount++; 19616cf997b0SBarry Smith } 19626cf997b0SBarry Smith 19635a8e39fbSHong Zhang /* add pivot rows into the active row */ 19645a8e39fbSHong Zhang nzbd = 0; 19655a8e39fbSHong Zhang prow = lnk[n]; 19665a8e39fbSHong Zhang while (prow < i) { 19675a8e39fbSHong Zhang nnz = bdiag[prow]; 19685a8e39fbSHong Zhang cols = bj_ptr[prow] + nnz + 1; 19695a8e39fbSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 19705a8e39fbSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 19715a8e39fbSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 19725a8e39fbSHong Zhang nzi += nlnk; 19735a8e39fbSHong Zhang prow = lnk[prow]; 19745a8e39fbSHong Zhang nzbd++; 197556beaf04SBarry Smith } 19765a8e39fbSHong Zhang bdiag[i] = nzbd; 19775a8e39fbSHong Zhang bi[i+1] = bi[i] + nzi; 1978ecf371e4SBarry Smith 19795a8e39fbSHong Zhang /* if free space is not available, make more free space */ 19805a8e39fbSHong Zhang if (current_space->local_remaining<nzi) { 19815a8e39fbSHong Zhang nnz = nzi*(n - i); /* estimated and max additional space needed */ 1982a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1983a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 19845a8e39fbSHong Zhang reallocs++; 19855a8e39fbSHong Zhang } 1986ecf371e4SBarry Smith 19875a8e39fbSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 19885a8e39fbSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 19895a8e39fbSHong Zhang bj_ptr[i] = current_space->array; 19905a8e39fbSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 19915a8e39fbSHong Zhang 19925a8e39fbSHong Zhang /* make sure the active row i has diagonal entry */ 19935a8e39fbSHong Zhang if (*(bj_ptr[i]+bdiag[i]) != i) { 199477431f27SBarry Smith SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\ 1995d7ee0231SBarry Smith try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 19966cf997b0SBarry Smith } 19975a8e39fbSHong Zhang 19985a8e39fbSHong Zhang current_space->array += nzi; 19995a8e39fbSHong Zhang current_space->local_used += nzi; 20005a8e39fbSHong Zhang current_space->local_remaining -= nzi; 20015a8e39fbSHong Zhang current_space_lvl->array += nzi; 20025a8e39fbSHong Zhang current_space_lvl->local_used += nzi; 20035a8e39fbSHong Zhang current_space_lvl->local_remaining -= nzi; 20049e25ed09SBarry Smith } 20055a8e39fbSHong Zhang 2006898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 2007898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 20085a8e39fbSHong Zhang 20095a8e39fbSHong Zhang /* destroy list of free space and other temporary arrays */ 20105a8e39fbSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 20116516239fSHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */ 20125a8e39fbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2013a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 20140e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 20159e25ed09SBarry Smith 20166cf91177SBarry Smith #if defined(PETSC_USE_INFO) 2017f64065bbSBarry Smith { 20185a8e39fbSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2019ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 2020ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2021ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 2022ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 2023335d9088SBarry Smith if (diagonal_fill) { 2024ae15b995SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 2025335d9088SBarry Smith } 2026f64065bbSBarry Smith } 202763ba0a88SBarry Smith #endif 2028bbb6d6a8SBarry Smith 20299e25ed09SBarry Smith /* put together the new matrix */ 203007dbf95fSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 2031719d5645SBarry Smith ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 2032719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 2033e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2034e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 20357c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 20360e83c824SBarry Smith ierr = PetscMalloc(bi[n]*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 20375a8e39fbSHong Zhang b->j = bj; 20385a8e39fbSHong Zhang b->i = bi; 20395a8e39fbSHong Zhang for (i=0; i<n; i++) bdiag[i] += bi[i]; 20405a8e39fbSHong Zhang b->diag = bdiag; 2041416022c9SBarry Smith b->ilen = 0; 2042416022c9SBarry Smith b->imax = 0; 2043416022c9SBarry Smith b->row = isrow; 2044416022c9SBarry Smith b->col = iscol; 2045c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2046c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 204782bf6240SBarry Smith b->icol = isicol; 204887828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2049416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 20505a8e39fbSHong Zhang Allocate bdiag, solve_work, new a, new j */ 2051719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 20525a8e39fbSHong Zhang b->maxnz = b->nz = bi[n] ; 2053719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2054719d5645SBarry Smith (fact)->info.fill_ratio_given = f; 2055719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2056ad04f41aSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 20574108e4d5SBarry Smith ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); 20583a40ed3dSBarry Smith PetscFunctionReturn(0); 20599e25ed09SBarry Smith } 2060d5d45c9bSBarry Smith 2061a6175056SHong Zhang #undef __FUNCT__ 206235233d99SShri Abhyankar #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ" 206335233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 20645f44c854SHong Zhang { 20655f44c854SHong Zhang Mat C = B; 20665f44c854SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 20675f44c854SHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 20685f44c854SHong Zhang IS ip=b->row,iip = b->icol; 20695f44c854SHong Zhang PetscErrorCode ierr; 20705f44c854SHong Zhang const PetscInt *rip,*riip; 20715f44c854SHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp; 20725f44c854SHong Zhang PetscInt *ai=a->i,*aj=a->j; 20735f44c854SHong Zhang PetscInt k,jmin,jmax,*c2r,*il,col,nexti,ili,nz; 20745f44c854SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 20755f44c854SHong Zhang PetscTruth perm_identity; 20765f44c854SHong Zhang 2077d90ac83dSHong Zhang FactorShiftCtx sctx; 207898a93161SHong Zhang PetscReal rs; 207998a93161SHong Zhang MatScalar d,*v; 208098a93161SHong Zhang 20815f44c854SHong Zhang PetscFunctionBegin; 208298a93161SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 2083d90ac83dSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 208498a93161SHong Zhang 2085*b2a402fcSHong Zhang if (info->shifttype == MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 208698a93161SHong Zhang sctx.shift_top = info->zeropivot; 208798a93161SHong Zhang for (i=0; i<mbs; i++) { 208898a93161SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 208998a93161SHong Zhang d = (aa)[a->diag[i]]; 209098a93161SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 209198a93161SHong Zhang v = aa+ai[i]; 209298a93161SHong Zhang nz = ai[i+1] - ai[i]; 209398a93161SHong Zhang for (j=0; j<nz; j++) 209498a93161SHong Zhang rs += PetscAbsScalar(v[j]); 209598a93161SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 209698a93161SHong Zhang } 209798a93161SHong Zhang sctx.shift_top *= 1.1; 209898a93161SHong Zhang sctx.nshift_max = 5; 209998a93161SHong Zhang sctx.shift_lo = 0.; 210098a93161SHong Zhang sctx.shift_hi = 1.; 210198a93161SHong Zhang } 21025f44c854SHong Zhang 21035f44c854SHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 21045f44c854SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 21055f44c854SHong Zhang 21065f44c854SHong Zhang /* allocate working arrays 21075f44c854SHong Zhang c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col 21085f44c854SHong 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 21095f44c854SHong Zhang */ 21100e83c824SBarry Smith ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&c2r);CHKERRQ(ierr); 21115f44c854SHong Zhang 211298a93161SHong Zhang do { 2113d90ac83dSHong Zhang sctx.useshift = PETSC_FALSE; 211498a93161SHong Zhang 2115c5546cabSHong Zhang for (i=0; i<mbs; i++) c2r[i] = mbs; 21165f44c854SHong Zhang il[0] = 0; 21175f44c854SHong Zhang 21185f44c854SHong Zhang for (k = 0; k<mbs; k++){ 21195f44c854SHong Zhang /* zero rtmp */ 21205f44c854SHong Zhang nz = bi[k+1] - bi[k]; 21215f44c854SHong Zhang bjtmp = bj + bi[k]; 21225f44c854SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 21235f44c854SHong Zhang 21245f44c854SHong Zhang /* load in initial unfactored row */ 21255f44c854SHong Zhang bval = ba + bi[k]; 21265f44c854SHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 21275f44c854SHong Zhang for (j = jmin; j < jmax; j++){ 21285f44c854SHong Zhang col = riip[aj[j]]; 21295f44c854SHong Zhang if (col >= k){ /* only take upper triangular entry */ 21305f44c854SHong Zhang rtmp[col] = aa[j]; 21315f44c854SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 21325f44c854SHong Zhang } 21335f44c854SHong Zhang } 213498a93161SHong Zhang /* shift the diagonal of the matrix: ZeropivotApply() */ 213598a93161SHong Zhang rtmp[k] += sctx.shift_amount; /* shift the diagonal of the matrix */ 21365f44c854SHong Zhang 21375f44c854SHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 21385f44c854SHong Zhang dk = rtmp[k]; 21395f44c854SHong Zhang i = c2r[k]; /* first row to be added to k_th row */ 21405f44c854SHong Zhang 21415f44c854SHong Zhang while (i < k){ 21425f44c854SHong Zhang nexti = c2r[i]; /* next row to be added to k_th row */ 21435f44c854SHong Zhang 21445f44c854SHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 21455f44c854SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 21465f44c854SHong Zhang uikdi = - ba[ili]*ba[bdiag[i]]; /* diagonal(k) */ 21475f44c854SHong Zhang dk += uikdi*ba[ili]; /* update diag[k] */ 21485f44c854SHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 21495f44c854SHong Zhang 21505f44c854SHong Zhang /* add multiple of row i to k-th row */ 21515f44c854SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 21525f44c854SHong Zhang if (jmin < jmax){ 21535f44c854SHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 21545f44c854SHong Zhang /* update il and c2r for row i */ 21555f44c854SHong Zhang il[i] = jmin; 21565f44c854SHong Zhang j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i; 21575f44c854SHong Zhang } 21585f44c854SHong Zhang i = nexti; 21595f44c854SHong Zhang } 21605f44c854SHong Zhang 21615f44c854SHong Zhang /* copy data into U(k,:) */ 216298a93161SHong Zhang rs = 0.0; 21635f44c854SHong Zhang jmin = bi[k]; jmax = bi[k+1]-1; 21645f44c854SHong Zhang if (jmin < jmax) { 21655f44c854SHong Zhang for (j=jmin; j<jmax; j++){ 216698a93161SHong Zhang col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]); 21675f44c854SHong Zhang } 21685f44c854SHong Zhang /* add the k-th row into il and c2r */ 21695f44c854SHong Zhang il[k] = jmin; 21705f44c854SHong Zhang i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k; 21715f44c854SHong Zhang } 217298a93161SHong Zhang 217398a93161SHong Zhang /* MatPivotCheck() */ 217498a93161SHong Zhang sctx.rs = rs; 217598a93161SHong Zhang sctx.pv = dk; 2176*b2a402fcSHong Zhang if (info->shifttype == MAT_SHIFT_NONZERO){ 2177bbd65245SShri Abhyankar ierr = MatPivotCheck_nz(info,sctx,k);CHKERRQ(ierr); 2178*b2a402fcSHong Zhang } else if (info->shifttype == MAT_SHIFT_POSITIVE_DEFINITE){ 2179bbd65245SShri Abhyankar ierr = MatPivotCheck_pd(info,sctx,k);CHKERRQ(ierr); 2180*b2a402fcSHong Zhang } else if (info->shifttype == MAT_SHIFT_INBLOCKS){ 2181bbd65245SShri Abhyankar ierr = MatPivotCheck_inblocks(info,sctx,k);CHKERRQ(ierr); 218298a93161SHong Zhang } else { 2183bbd65245SShri Abhyankar ierr = MatPivotCheck_none(info,sctx,k);CHKERRQ(ierr); 21845f44c854SHong Zhang } 218598a93161SHong Zhang dk = sctx.pv; 218698a93161SHong Zhang 218798a93161SHong Zhang ba[bdiag[k]] = 1.0/dk; /* U(k,k) */ 218898a93161SHong Zhang } 2189d90ac83dSHong Zhang } while (sctx.useshift); 21905f44c854SHong Zhang 21910e83c824SBarry Smith ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr); 21925f44c854SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 21935f44c854SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 21945f44c854SHong Zhang 21955f44c854SHong Zhang ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 21965f44c854SHong Zhang if (perm_identity){ 219735233d99SShri Abhyankar B->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 219835233d99SShri Abhyankar B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 21992169352eSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering; 22002169352eSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering; 22015f44c854SHong Zhang } else { 220235233d99SShri Abhyankar B->ops->solve = MatSolve_SeqSBAIJ_1; 220335233d99SShri Abhyankar B->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 220435233d99SShri Abhyankar B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1; 220535233d99SShri Abhyankar B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1; 22065f44c854SHong Zhang } 22075f44c854SHong Zhang 22085f44c854SHong Zhang C->assembled = PETSC_TRUE; 22095f44c854SHong Zhang C->preallocated = PETSC_TRUE; 22105f44c854SHong Zhang ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 221198a93161SHong Zhang 221298a93161SHong Zhang /* MatPivotView() */ 221398a93161SHong Zhang if (sctx.nshift){ 2214*b2a402fcSHong Zhang if (info->shifttype == MAT_SHIFT_POSITIVE_DEFINITE) { 221598a93161SHong Zhang ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr); 2216*b2a402fcSHong Zhang } else if (info->shifttype == MAT_SHIFT_NONZERO) { 221798a93161SHong Zhang ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2218*b2a402fcSHong Zhang } else if (info->shifttype == MAT_SHIFT_INBLOCKS){ 2219*b2a402fcSHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr); 222098a93161SHong Zhang } 222198a93161SHong Zhang } 22225f44c854SHong Zhang PetscFunctionReturn(0); 22235f44c854SHong Zhang } 22245f44c854SHong Zhang 22255f44c854SHong Zhang #undef __FUNCT__ 22260a3c351aSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_inplace" 22270a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info) 2228a6175056SHong Zhang { 2229719d5645SBarry Smith Mat C = B; 22300968510aSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 22312ed133dbSHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 22329bfd6278SHong Zhang IS ip=b->row,iip = b->icol; 22332ed133dbSHong Zhang PetscErrorCode ierr; 22345d0c19d7SBarry Smith const PetscInt *rip,*riip; 2235c5546cabSHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp; 22362ed133dbSHong Zhang PetscInt *ai=a->i,*aj=a->j; 22372ed133dbSHong Zhang PetscInt k,jmin,jmax,*jl,*il,col,nexti,ili,nz; 2238622e2028SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 2239540703acSHong Zhang PetscReal zeropivot,rs,shiftnz; 2240fbf22428SSatish Balay PetscReal shiftpd; 2241540703acSHong Zhang ChShift_Ctx sctx; 2242540703acSHong Zhang PetscInt newshift; 2243db4efbfdSBarry Smith PetscTruth perm_identity; 2244d5d45c9bSBarry Smith 2245a6175056SHong Zhang PetscFunctionBegin; 2246ee45ca4aSHong Zhang zeropivot = info->zeropivot; 2247ee45ca4aSHong Zhang 22482ed133dbSHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 22499bfd6278SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 22502ed133dbSHong Zhang 22512ed133dbSHong Zhang /* initialization */ 22520e83c824SBarry Smith ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&jl);CHKERRQ(ierr); 2253540703acSHong Zhang sctx.shift_amount = 0; 2254540703acSHong Zhang sctx.nshift = 0; 22552ed133dbSHong Zhang do { 2256540703acSHong Zhang sctx.chshift = PETSC_FALSE; 2257c5546cabSHong Zhang for (i=0; i<mbs; i++) jl[i] = mbs; 2258c5546cabSHong Zhang il[0] = 0; 22592ed133dbSHong Zhang 22602ed133dbSHong Zhang for (k = 0; k<mbs; k++){ 2261c5546cabSHong Zhang /* zero rtmp */ 2262c5546cabSHong Zhang nz = bi[k+1] - bi[k]; 2263c5546cabSHong Zhang bjtmp = bj + bi[k]; 2264c5546cabSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 2265c5546cabSHong Zhang 2266c05c3958SHong Zhang bval = ba + bi[k]; 22672ed133dbSHong Zhang /* initialize k-th row by the perm[k]-th row of A */ 22682ed133dbSHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 22692ed133dbSHong Zhang for (j = jmin; j < jmax; j++){ 22709bfd6278SHong Zhang col = riip[aj[j]]; 22712ed133dbSHong Zhang if (col >= k){ /* only take upper triangular entry */ 22722ed133dbSHong Zhang rtmp[col] = aa[j]; 2273c05c3958SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 22742ed133dbSHong Zhang } 22752ed133dbSHong Zhang } 227639e3d630SHong Zhang /* shift the diagonal of the matrix */ 2277540703acSHong Zhang if (sctx.nshift) rtmp[k] += sctx.shift_amount; 22782ed133dbSHong Zhang 22792ed133dbSHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 22802ed133dbSHong Zhang dk = rtmp[k]; 22812ed133dbSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 22822ed133dbSHong Zhang 22832ed133dbSHong Zhang while (i < k){ 22842ed133dbSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 22852ed133dbSHong Zhang 22862ed133dbSHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 22872ed133dbSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 22882ed133dbSHong Zhang uikdi = - ba[ili]*ba[bi[i]]; /* diagonal(k) */ 22892ed133dbSHong Zhang dk += uikdi*ba[ili]; 22902ed133dbSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 22912ed133dbSHong Zhang 22922ed133dbSHong Zhang /* add multiple of row i to k-th row */ 22932ed133dbSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 22942ed133dbSHong Zhang if (jmin < jmax){ 22952ed133dbSHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 22962ed133dbSHong Zhang /* update il and jl for row i */ 22972ed133dbSHong Zhang il[i] = jmin; 22982ed133dbSHong Zhang j = bj[jmin]; jl[i] = jl[j]; jl[j] = i; 22992ed133dbSHong Zhang } 23002ed133dbSHong Zhang i = nexti; 23012ed133dbSHong Zhang } 23022ed133dbSHong Zhang 23030697b6caSHong Zhang /* shift the diagonals when zero pivot is detected */ 23040697b6caSHong Zhang /* compute rs=sum of abs(off-diagonal) */ 23050697b6caSHong Zhang rs = 0.0; 23063c9e8598SHong Zhang jmin = bi[k]+1; 23073c9e8598SHong Zhang nz = bi[k+1] - jmin; 23083c9e8598SHong Zhang bcol = bj + jmin; 230904fbf559SBarry Smith for (j=0; j<nz; j++) { 231004fbf559SBarry Smith rs += PetscAbsScalar(rtmp[bcol[j]]); 23113c9e8598SHong Zhang } 2312540703acSHong Zhang 2313540703acSHong Zhang sctx.rs = rs; 2314540703acSHong Zhang sctx.pv = dk; 23155b5bc046SBarry Smith ierr = MatCholeskyCheckShift_inline(info,sctx,k,newshift);CHKERRQ(ierr); 2316117f1891Stmunson 2317117f1891Stmunson if (newshift == 1) { 2318117f1891Stmunson if (!sctx.shift_amount) { 2319117f1891Stmunson sctx.shift_amount = 1e-5; 2320117f1891Stmunson } 2321117f1891Stmunson break; 2322117f1891Stmunson } 23233c9e8598SHong Zhang 23243c9e8598SHong Zhang /* copy data into U(k,:) */ 232539e3d630SHong Zhang ba[bi[k]] = 1.0/dk; /* U(k,k) */ 232639e3d630SHong Zhang jmin = bi[k]+1; jmax = bi[k+1]; 232739e3d630SHong Zhang if (jmin < jmax) { 232839e3d630SHong Zhang for (j=jmin; j<jmax; j++){ 2329c6ae9bfcSHong Zhang col = bj[j]; ba[j] = rtmp[col]; 23303c9e8598SHong Zhang } 233139e3d630SHong Zhang /* add the k-th row into il and jl */ 23323c9e8598SHong Zhang il[k] = jmin; 23333c9e8598SHong Zhang i = bj[jmin]; jl[k] = jl[i]; jl[i] = k; 23343c9e8598SHong Zhang } 23353c9e8598SHong Zhang } 2336540703acSHong Zhang } while (sctx.chshift); 23370e83c824SBarry Smith ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr); 233839e3d630SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 23399bfd6278SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 2340db4efbfdSBarry Smith 2341db4efbfdSBarry Smith ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 2342db4efbfdSBarry Smith if (perm_identity){ 23430a3c351aSHong Zhang B->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23440a3c351aSHong Zhang B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23450a3c351aSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23460a3c351aSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 2347db4efbfdSBarry Smith } else { 23480a3c351aSHong Zhang B->ops->solve = MatSolve_SeqSBAIJ_1_inplace; 23490a3c351aSHong Zhang B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_inplace; 23500a3c351aSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_inplace; 23510a3c351aSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_inplace; 2352db4efbfdSBarry Smith } 2353db4efbfdSBarry Smith 23543c9e8598SHong Zhang C->assembled = PETSC_TRUE; 23553c9e8598SHong Zhang C->preallocated = PETSC_TRUE; 2356d0f46423SBarry Smith ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 2357540703acSHong Zhang if (sctx.nshift){ 2358540703acSHong Zhang if (shiftnz) { 23591e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2360540703acSHong Zhang } else if (shiftpd) { 23611e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 23620697b6caSHong Zhang } 23633c9e8598SHong Zhang } 23643c9e8598SHong Zhang PetscFunctionReturn(0); 23653c9e8598SHong Zhang } 2366a6175056SHong Zhang 23678a76ed4fSHong Zhang /* 23688a76ed4fSHong Zhang icc() under revised new data structure. 23698a76ed4fSHong Zhang Factored arrays bj and ba are stored as 23708a76ed4fSHong Zhang U(0,:),...,U(i,:),U(n-1,:) 23718a76ed4fSHong Zhang 23728a76ed4fSHong Zhang ui=fact->i is an array of size n+1, in which 23738a76ed4fSHong Zhang ui+ 23748a76ed4fSHong Zhang ui[i]: points to 1st entry of U(i,:),i=0,...,n-1 23758a76ed4fSHong Zhang ui[n]: points to U(n-1,n-1)+1 23768a76ed4fSHong Zhang 23778a76ed4fSHong Zhang udiag=fact->diag is an array of size n,in which 23788a76ed4fSHong Zhang udiag[i]: points to diagonal of U(i,:), i=0,...,n-1 23798a76ed4fSHong Zhang 23808a76ed4fSHong Zhang U(i,:) contains udiag[i] as its last entry, i.e., 23818a76ed4fSHong Zhang U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 23828a76ed4fSHong Zhang */ 23838a76ed4fSHong Zhang 2384a6175056SHong Zhang #undef __FUNCT__ 238535233d99SShri Abhyankar #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ" 238635233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2387a6175056SHong Zhang { 23880968510aSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2389ed59401aSHong Zhang Mat_SeqSBAIJ *b; 2390dfbe8321SBarry Smith PetscErrorCode ierr; 239158ebbce7SBarry Smith PetscTruth perm_identity,missing; 23925f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 23935d0c19d7SBarry Smith const PetscInt *rip,*riip; 2394ed59401aSHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 239558ebbce7SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 23965a8e39fbSHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 2397ed59401aSHong Zhang PetscReal fill=info->fill,levels=info->levels; 2398a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 2399a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 24007a48dd6fSHong Zhang PetscBT lnkbt; 2401b635d36bSHong Zhang IS iperm; 2402a6175056SHong Zhang 2403a6175056SHong Zhang PetscFunctionBegin; 2404d0f46423SBarry 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); 240558ebbce7SBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 240658ebbce7SBarry Smith if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 2407653a6975SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 2408b635d36bSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 24097a48dd6fSHong Zhang 241039e3d630SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 24115f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 241239e3d630SHong Zhang ui[0] = 0; 241339e3d630SHong Zhang 24148a76ed4fSHong Zhang /* ICC(0) without matrix ordering: simply rearrange column indices */ 2415622e2028SHong Zhang if (!levels && perm_identity) { 24165f44c854SHong Zhang for (i=0; i<am; i++) { 2417c5546cabSHong Zhang ncols = ai[i+1] - a->diag[i]; 2418c5546cabSHong Zhang ui[i+1] = ui[i] + ncols; 2419c5546cabSHong Zhang udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */ 2420dc1e2a3fSHong Zhang } 2421dc1e2a3fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2422dc1e2a3fSHong Zhang cols = uj; 2423dc1e2a3fSHong Zhang for (i=0; i<am; i++) { 24245f44c854SHong Zhang aj = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */ 2425dc1e2a3fSHong Zhang ncols = ai[i+1] - a->diag[i] -1; 24265f44c854SHong Zhang for (j=0; j<ncols; j++) *cols++ = aj[j]; 24275f44c854SHong Zhang *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */ 24285f44c854SHong Zhang } 24295f44c854SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 24305f44c854SHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 24315f44c854SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 24325f44c854SHong Zhang 24335f44c854SHong Zhang /* initialization */ 24345f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 24355f44c854SHong Zhang 24365f44c854SHong Zhang /* jl: linked list for storing indices of the pivot rows 24375f44c854SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 24380e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr); 24395f44c854SHong Zhang for (i=0; i<am; i++){ 24405f44c854SHong Zhang jl[i] = am; il[i] = 0; 24415f44c854SHong Zhang } 24425f44c854SHong Zhang 24435f44c854SHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 24445f44c854SHong Zhang nlnk = am + 1; 24455f44c854SHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 24465f44c854SHong Zhang 24475f44c854SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 24485f44c854SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 24495f44c854SHong Zhang current_space = free_space; 24505f44c854SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 24515f44c854SHong Zhang current_space_lvl = free_space_lvl; 24525f44c854SHong Zhang 24535f44c854SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 24545f44c854SHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 24555f44c854SHong Zhang nzk = 0; 24565f44c854SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 24575f44c854SHong 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); 24585f44c854SHong Zhang ncols_upper = 0; 24595f44c854SHong Zhang for (j=0; j<ncols; j++){ 24605f44c854SHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 24615f44c854SHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 24625f44c854SHong Zhang ajtmp[ncols_upper] = i; 24635f44c854SHong Zhang ncols_upper++; 24645f44c854SHong Zhang } 24655f44c854SHong Zhang } 24665f44c854SHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 24675f44c854SHong Zhang nzk += nlnk; 24685f44c854SHong Zhang 24695f44c854SHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 24705f44c854SHong Zhang prow = jl[k]; /* 1st pivot row */ 24715f44c854SHong Zhang 24725f44c854SHong Zhang while (prow < k){ 24735f44c854SHong Zhang nextprow = jl[prow]; 24745f44c854SHong Zhang 24755f44c854SHong Zhang /* merge prow into k-th row */ 24765f44c854SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 24775f44c854SHong Zhang jmax = ui[prow+1]; 24785f44c854SHong Zhang ncols = jmax-jmin; 24795f44c854SHong Zhang i = jmin - ui[prow]; 24805f44c854SHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 24815f44c854SHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 24825f44c854SHong Zhang j = *(uj - 1); 24835f44c854SHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 24845f44c854SHong Zhang nzk += nlnk; 24855f44c854SHong Zhang 24865f44c854SHong Zhang /* update il and jl for prow */ 24875f44c854SHong Zhang if (jmin < jmax){ 24885f44c854SHong Zhang il[prow] = jmin; 24895f44c854SHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 24905f44c854SHong Zhang } 24915f44c854SHong Zhang prow = nextprow; 24925f44c854SHong Zhang } 24935f44c854SHong Zhang 24945f44c854SHong Zhang /* if free space is not available, make more free space */ 24955f44c854SHong Zhang if (current_space->local_remaining<nzk) { 24965f44c854SHong Zhang i = am - k + 1; /* num of unfactored rows */ 2497c6ae9bfcSHong Zhang i *= PetscMin(nzk, i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */ 24985f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 24995f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 25005f44c854SHong Zhang reallocs++; 25015f44c854SHong Zhang } 25025f44c854SHong Zhang 25035f44c854SHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 25045f44c854SHong Zhang if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 25055f44c854SHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 25065f44c854SHong Zhang 25075f44c854SHong Zhang /* add the k-th row into il and jl */ 25085f44c854SHong Zhang if (nzk > 1){ 25095f44c854SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 25105f44c854SHong Zhang jl[k] = jl[i]; jl[i] = k; 25115f44c854SHong Zhang il[k] = ui[k] + 1; 25125f44c854SHong Zhang } 25135f44c854SHong Zhang uj_ptr[k] = current_space->array; 25145f44c854SHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 25155f44c854SHong Zhang 25165f44c854SHong Zhang current_space->array += nzk; 25175f44c854SHong Zhang current_space->local_used += nzk; 25185f44c854SHong Zhang current_space->local_remaining -= nzk; 25195f44c854SHong Zhang 25205f44c854SHong Zhang current_space_lvl->array += nzk; 25215f44c854SHong Zhang current_space_lvl->local_used += nzk; 25225f44c854SHong Zhang current_space_lvl->local_remaining -= nzk; 25235f44c854SHong Zhang 25245f44c854SHong Zhang ui[k+1] = ui[k] + nzk; 25255f44c854SHong Zhang } 25265f44c854SHong Zhang 25275f44c854SHong Zhang #if defined(PETSC_USE_INFO) 25285f44c854SHong Zhang if (ai[am] != 0) { 25295f44c854SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 25305f44c854SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 25315f44c854SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 25325f44c854SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 25335f44c854SHong Zhang } else { 25345f44c854SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 25355f44c854SHong Zhang } 25365f44c854SHong Zhang #endif 25375f44c854SHong Zhang 25385f44c854SHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 25395f44c854SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 25400e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 25415f44c854SHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 25425f44c854SHong Zhang 25435f44c854SHong Zhang /* destroy list of free space and other temporary array(s) */ 25445f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 254535233d99SShri Abhyankar ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */ 25465f44c854SHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 25475f44c854SHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 25485f44c854SHong Zhang 25495f44c854SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 25505f44c854SHong Zhang 25515f44c854SHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 25525f44c854SHong Zhang b = (Mat_SeqSBAIJ*)(fact)->data; 25535f44c854SHong Zhang b->singlemalloc = PETSC_FALSE; 25545f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 25555f44c854SHong Zhang b->j = uj; 25565f44c854SHong Zhang b->i = ui; 25575f44c854SHong Zhang b->diag = udiag; 25587f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 25595f44c854SHong Zhang b->ilen = 0; 25605f44c854SHong Zhang b->imax = 0; 25615f44c854SHong Zhang b->row = perm; 25625f44c854SHong Zhang b->col = perm; 25635f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25645f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25655f44c854SHong Zhang b->icol = iperm; 25665f44c854SHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 25675f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2568d595f711SHong Zhang ierr = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 25695f44c854SHong Zhang b->maxnz = b->nz = ui[am]; 25705f44c854SHong Zhang b->free_a = PETSC_TRUE; 25715f44c854SHong Zhang b->free_ij = PETSC_TRUE; 25725f44c854SHong Zhang 2573f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2574f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 25755f44c854SHong Zhang if (ai[am] != 0) { 2576f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 25775f44c854SHong Zhang } else { 2578f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 25795f44c854SHong Zhang } 258035233d99SShri Abhyankar fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 25815f44c854SHong Zhang PetscFunctionReturn(0); 25825f44c854SHong Zhang } 25835f44c854SHong Zhang 25845f44c854SHong Zhang #undef __FUNCT__ 2585ad04f41aSHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_inplace" 2586ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 25875f44c854SHong Zhang { 25885f44c854SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 25895f44c854SHong Zhang Mat_SeqSBAIJ *b; 25905f44c854SHong Zhang PetscErrorCode ierr; 25915f44c854SHong Zhang PetscTruth perm_identity,missing; 25925f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 25935f44c854SHong Zhang const PetscInt *rip,*riip; 25945f44c854SHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 25955f44c854SHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 25965f44c854SHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 25975f44c854SHong Zhang PetscReal fill=info->fill,levels=info->levels; 25985f44c854SHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 25995f44c854SHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 26005f44c854SHong Zhang PetscBT lnkbt; 26015f44c854SHong Zhang IS iperm; 26025f44c854SHong Zhang 26035f44c854SHong Zhang PetscFunctionBegin; 26045f44c854SHong 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); 26055f44c854SHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 26065f44c854SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 26075f44c854SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 26085f44c854SHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 26095f44c854SHong Zhang 26105f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 26115f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 26125f44c854SHong Zhang ui[0] = 0; 26135f44c854SHong Zhang 26145f44c854SHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 26155f44c854SHong Zhang if (!levels && perm_identity) { 26165f44c854SHong Zhang 26175f44c854SHong Zhang for (i=0; i<am; i++) { 26185f44c854SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 26195f44c854SHong Zhang udiag[i] = ui[i]; 2620ed59401aSHong Zhang } 262139e3d630SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 262239e3d630SHong Zhang cols = uj; 2623ed59401aSHong Zhang for (i=0; i<am; i++) { 2624ed59401aSHong Zhang aj = a->j + a->diag[i]; 262539e3d630SHong Zhang ncols = ui[i+1] - ui[i]; 262639e3d630SHong Zhang for (j=0; j<ncols; j++) *cols++ = *aj++; 2627ed59401aSHong Zhang } 262839e3d630SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 2629b635d36bSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 2630b635d36bSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 2631b635d36bSHong Zhang 26327a48dd6fSHong Zhang /* initialization */ 26335a8e39fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 26347a48dd6fSHong Zhang 26357a48dd6fSHong Zhang /* jl: linked list for storing indices of the pivot rows 26367a48dd6fSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 26370e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr); 26387a48dd6fSHong Zhang for (i=0; i<am; i++){ 26397a48dd6fSHong Zhang jl[i] = am; il[i] = 0; 26407a48dd6fSHong Zhang } 26417a48dd6fSHong Zhang 26427a48dd6fSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 26437a48dd6fSHong Zhang nlnk = am + 1; 26442ed133dbSHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 26457a48dd6fSHong Zhang 26467a48dd6fSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2647a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 26487a48dd6fSHong Zhang current_space = free_space; 2649a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 26507a48dd6fSHong Zhang current_space_lvl = free_space_lvl; 26517a48dd6fSHong Zhang 26527a48dd6fSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 26537a48dd6fSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 26547a48dd6fSHong Zhang nzk = 0; 2655622e2028SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 26563b4a8b6dSBarry 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); 2657622e2028SHong Zhang ncols_upper = 0; 2658622e2028SHong Zhang for (j=0; j<ncols; j++){ 2659b635d36bSHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 2660b635d36bSHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 26615a8e39fbSHong Zhang ajtmp[ncols_upper] = i; 2662622e2028SHong Zhang ncols_upper++; 2663622e2028SHong Zhang } 2664622e2028SHong Zhang } 2665b635d36bSHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 26667a48dd6fSHong Zhang nzk += nlnk; 26677a48dd6fSHong Zhang 26687a48dd6fSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 26697a48dd6fSHong Zhang prow = jl[k]; /* 1st pivot row */ 26707a48dd6fSHong Zhang 26717a48dd6fSHong Zhang while (prow < k){ 26727a48dd6fSHong Zhang nextprow = jl[prow]; 26737a48dd6fSHong Zhang 26747a48dd6fSHong Zhang /* merge prow into k-th row */ 26757a48dd6fSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 26767a48dd6fSHong Zhang jmax = ui[prow+1]; 26777a48dd6fSHong Zhang ncols = jmax-jmin; 2678ed59401aSHong Zhang i = jmin - ui[prow]; 2679ed59401aSHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 26805a8e39fbSHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 26815a8e39fbSHong Zhang j = *(uj - 1); 26825a8e39fbSHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 26837a48dd6fSHong Zhang nzk += nlnk; 26847a48dd6fSHong Zhang 26857a48dd6fSHong Zhang /* update il and jl for prow */ 26867a48dd6fSHong Zhang if (jmin < jmax){ 26877a48dd6fSHong Zhang il[prow] = jmin; 26887a48dd6fSHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 26897a48dd6fSHong Zhang } 26907a48dd6fSHong Zhang prow = nextprow; 26917a48dd6fSHong Zhang } 26927a48dd6fSHong Zhang 26937a48dd6fSHong Zhang /* if free space is not available, make more free space */ 26947a48dd6fSHong Zhang if (current_space->local_remaining<nzk) { 26957a48dd6fSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2696fd3f55cbSMatthew Knepley i *= PetscMin(nzk, (i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2697a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 2698a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 26997a48dd6fSHong Zhang reallocs++; 27007a48dd6fSHong Zhang } 27017a48dd6fSHong Zhang 27027a48dd6fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 2703b635d36bSHong Zhang if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 27042ed133dbSHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 27057a48dd6fSHong Zhang 27067a48dd6fSHong Zhang /* add the k-th row into il and jl */ 270739e3d630SHong Zhang if (nzk > 1){ 27087a48dd6fSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 27097a48dd6fSHong Zhang jl[k] = jl[i]; jl[i] = k; 27107a48dd6fSHong Zhang il[k] = ui[k] + 1; 27117a48dd6fSHong Zhang } 27127a48dd6fSHong Zhang uj_ptr[k] = current_space->array; 27137a48dd6fSHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 27147a48dd6fSHong Zhang 27157a48dd6fSHong Zhang current_space->array += nzk; 27167a48dd6fSHong Zhang current_space->local_used += nzk; 27177a48dd6fSHong Zhang current_space->local_remaining -= nzk; 27187a48dd6fSHong Zhang 27197a48dd6fSHong Zhang current_space_lvl->array += nzk; 27207a48dd6fSHong Zhang current_space_lvl->local_used += nzk; 27217a48dd6fSHong Zhang current_space_lvl->local_remaining -= nzk; 27227a48dd6fSHong Zhang 27237a48dd6fSHong Zhang ui[k+1] = ui[k] + nzk; 27247a48dd6fSHong Zhang } 27257a48dd6fSHong Zhang 27266cf91177SBarry Smith #if defined(PETSC_USE_INFO) 27277a48dd6fSHong Zhang if (ai[am] != 0) { 272839e3d630SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 2729ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2730ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2731ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 27327a48dd6fSHong Zhang } else { 2733ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 27347a48dd6fSHong Zhang } 273563ba0a88SBarry Smith #endif 27367a48dd6fSHong Zhang 27377a48dd6fSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 2738b635d36bSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 27390e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 27405a8e39fbSHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 27417a48dd6fSHong Zhang 27427a48dd6fSHong Zhang /* destroy list of free space and other temporary array(s) */ 27437a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2744a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 27452ed133dbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2746a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 27477a48dd6fSHong Zhang 274839e3d630SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 274939e3d630SHong Zhang 27507a48dd6fSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 27517a48dd6fSHong Zhang 2752f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 27537a48dd6fSHong Zhang b->singlemalloc = PETSC_FALSE; 27547a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 27557a48dd6fSHong Zhang b->j = uj; 27567a48dd6fSHong Zhang b->i = ui; 27575f44c854SHong Zhang b->diag = udiag; 27587f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 27597a48dd6fSHong Zhang b->ilen = 0; 27607a48dd6fSHong Zhang b->imax = 0; 27617a48dd6fSHong Zhang b->row = perm; 2762b635d36bSHong Zhang b->col = perm; 2763b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2764b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2765b635d36bSHong Zhang b->icol = iperm; 27667a48dd6fSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 27677a48dd6fSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2768f284f12aSHong Zhang ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 27697a48dd6fSHong Zhang b->maxnz = b->nz = ui[am]; 2770e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2771e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 27727a48dd6fSHong Zhang 2773f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2774f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 27757a48dd6fSHong Zhang if (ai[am] != 0) { 2776f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 27777a48dd6fSHong Zhang } else { 2778f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 27797a48dd6fSHong Zhang } 27800a3c351aSHong Zhang fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace; 2781a6175056SHong Zhang PetscFunctionReturn(0); 2782a6175056SHong Zhang } 2783d5d45c9bSBarry Smith 278435233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 27850be760fbSHong Zhang { 27860be760fbSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 27870be760fbSHong Zhang Mat_SeqSBAIJ *b; 27880be760fbSHong Zhang PetscErrorCode ierr; 27890be760fbSHong Zhang PetscTruth perm_identity; 27900be760fbSHong Zhang PetscReal fill = info->fill; 27910be760fbSHong Zhang const PetscInt *rip,*riip; 27920be760fbSHong Zhang PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 27930be760fbSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 27940be760fbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag; 27950be760fbSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 27960be760fbSHong Zhang PetscBT lnkbt; 27970be760fbSHong Zhang IS iperm; 27980be760fbSHong Zhang 27990be760fbSHong Zhang PetscFunctionBegin; 28000be760fbSHong 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); 28010be760fbSHong Zhang /* check whether perm is the identity mapping */ 28020be760fbSHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 28030be760fbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 28040be760fbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 28050be760fbSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 28060be760fbSHong Zhang 28070be760fbSHong Zhang /* initialization */ 28080be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 28090be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 28100be760fbSHong Zhang ui[0] = 0; 28110be760fbSHong Zhang 28120be760fbSHong Zhang /* jl: linked list for storing indices of the pivot rows 28130be760fbSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 28140be760fbSHong Zhang ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr); 28150be760fbSHong Zhang for (i=0; i<am; i++){ 28160be760fbSHong Zhang jl[i] = am; il[i] = 0; 28170be760fbSHong Zhang } 28180be760fbSHong Zhang 28190be760fbSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 28200be760fbSHong Zhang nlnk = am + 1; 28210be760fbSHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28220be760fbSHong Zhang 28230be760fbSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 28240be760fbSHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 28250be760fbSHong Zhang current_space = free_space; 28260be760fbSHong Zhang 28270be760fbSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 28280be760fbSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 28290be760fbSHong Zhang nzk = 0; 28300be760fbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 28310be760fbSHong 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); 28320be760fbSHong Zhang ncols_upper = 0; 28330be760fbSHong Zhang for (j=0; j<ncols; j++){ 28340be760fbSHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 28350be760fbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 28360be760fbSHong Zhang cols[ncols_upper] = i; 28370be760fbSHong Zhang ncols_upper++; 28380be760fbSHong Zhang } 28390be760fbSHong Zhang } 28400be760fbSHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28410be760fbSHong Zhang nzk += nlnk; 28420be760fbSHong Zhang 28430be760fbSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 28440be760fbSHong Zhang prow = jl[k]; /* 1st pivot row */ 28450be760fbSHong Zhang 28460be760fbSHong Zhang while (prow < k){ 28470be760fbSHong Zhang nextprow = jl[prow]; 28480be760fbSHong Zhang /* merge prow into k-th row */ 28490be760fbSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 28500be760fbSHong Zhang jmax = ui[prow+1]; 28510be760fbSHong Zhang ncols = jmax-jmin; 28520be760fbSHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 28530be760fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28540be760fbSHong Zhang nzk += nlnk; 28550be760fbSHong Zhang 28560be760fbSHong Zhang /* update il and jl for prow */ 28570be760fbSHong Zhang if (jmin < jmax){ 28580be760fbSHong Zhang il[prow] = jmin; 28590be760fbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 28600be760fbSHong Zhang } 28610be760fbSHong Zhang prow = nextprow; 28620be760fbSHong Zhang } 28630be760fbSHong Zhang 28640be760fbSHong Zhang /* if free space is not available, make more free space */ 28650be760fbSHong Zhang if (current_space->local_remaining<nzk) { 28660be760fbSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2867c6ae9bfcSHong Zhang i *= PetscMin(nzk,i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */ 28680be760fbSHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 28690be760fbSHong Zhang reallocs++; 28700be760fbSHong Zhang } 28710be760fbSHong Zhang 28720be760fbSHong Zhang /* copy data into free space, then initialize lnk */ 28730be760fbSHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 28740be760fbSHong Zhang 28750be760fbSHong Zhang /* add the k-th row into il and jl */ 28767b056e98SHong Zhang if (nzk > 1){ 28770be760fbSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 28780be760fbSHong Zhang jl[k] = jl[i]; jl[i] = k; 28790be760fbSHong Zhang il[k] = ui[k] + 1; 28800be760fbSHong Zhang } 28810be760fbSHong Zhang ui_ptr[k] = current_space->array; 28820be760fbSHong Zhang current_space->array += nzk; 28830be760fbSHong Zhang current_space->local_used += nzk; 28840be760fbSHong Zhang current_space->local_remaining -= nzk; 28850be760fbSHong Zhang 28860be760fbSHong Zhang ui[k+1] = ui[k] + nzk; 28870be760fbSHong Zhang } 28880be760fbSHong Zhang 28890be760fbSHong Zhang #if defined(PETSC_USE_INFO) 28900be760fbSHong Zhang if (ai[am] != 0) { 28910be760fbSHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 28920be760fbSHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 28930be760fbSHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 28940be760fbSHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 28950be760fbSHong Zhang } else { 28960be760fbSHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 28970be760fbSHong Zhang } 28980be760fbSHong Zhang #endif 28990be760fbSHong Zhang 29000be760fbSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 29010be760fbSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 29020be760fbSHong Zhang ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 29030be760fbSHong Zhang 29040be760fbSHong Zhang /* destroy list of free space and other temporary array(s) */ 29050be760fbSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 290635233d99SShri Abhyankar ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */ 29070be760fbSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 29080be760fbSHong Zhang 29090be760fbSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 29100be760fbSHong Zhang 2911f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 29120be760fbSHong Zhang b->singlemalloc = PETSC_FALSE; 29130be760fbSHong Zhang b->free_a = PETSC_TRUE; 29140be760fbSHong Zhang b->free_ij = PETSC_TRUE; 29150be760fbSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 29160be760fbSHong Zhang b->j = uj; 29170be760fbSHong Zhang b->i = ui; 29180be760fbSHong Zhang b->diag = udiag; 29190be760fbSHong Zhang b->free_diag = PETSC_TRUE; 29200be760fbSHong Zhang b->ilen = 0; 29210be760fbSHong Zhang b->imax = 0; 29220be760fbSHong Zhang b->row = perm; 29230be760fbSHong Zhang b->col = perm; 29240be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29250be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29260be760fbSHong Zhang b->icol = iperm; 29270be760fbSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 29280be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 29290860e08cSHong Zhang ierr = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 29300be760fbSHong Zhang b->maxnz = b->nz = ui[am]; 29310be760fbSHong Zhang 2932f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2933f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 29340be760fbSHong Zhang if (ai[am] != 0) { 2935f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 29360be760fbSHong Zhang } else { 2937f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 29380be760fbSHong Zhang } 293935233d99SShri Abhyankar fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 29400be760fbSHong Zhang PetscFunctionReturn(0); 29410be760fbSHong Zhang } 29420be760fbSHong Zhang 2943f76d2b81SHong Zhang #undef __FUNCT__ 2944ad04f41aSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ_inplace" 2945ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2946f76d2b81SHong Zhang { 2947f76d2b81SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 294810c27e3dSHong Zhang Mat_SeqSBAIJ *b; 2949dfbe8321SBarry Smith PetscErrorCode ierr; 2950f76d2b81SHong Zhang PetscTruth perm_identity; 295110c27e3dSHong Zhang PetscReal fill = info->fill; 29525d0c19d7SBarry Smith const PetscInt *rip,*riip; 29535d0c19d7SBarry Smith PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 295410c27e3dSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 29552ed133dbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr; 2956a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 295710c27e3dSHong Zhang PetscBT lnkbt; 29582ed133dbSHong Zhang IS iperm; 2959f76d2b81SHong Zhang 2960f76d2b81SHong Zhang PetscFunctionBegin; 2961d0f46423SBarry 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); 29622ed133dbSHong Zhang /* check whether perm is the identity mapping */ 2963f76d2b81SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 29642ed133dbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 29652ed133dbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 29669bfd6278SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 296710c27e3dSHong Zhang 296810c27e3dSHong Zhang /* initialization */ 29691393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 297010c27e3dSHong Zhang ui[0] = 0; 297110c27e3dSHong Zhang 297210c27e3dSHong Zhang /* jl: linked list for storing indices of the pivot rows 29731393bc97SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 29740e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr); 29751393bc97SHong Zhang for (i=0; i<am; i++){ 29761393bc97SHong Zhang jl[i] = am; il[i] = 0; 2977f76d2b81SHong Zhang } 297810c27e3dSHong Zhang 297910c27e3dSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 29801393bc97SHong Zhang nlnk = am + 1; 29811393bc97SHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 298210c27e3dSHong Zhang 29831393bc97SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2984a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 298510c27e3dSHong Zhang current_space = free_space; 298610c27e3dSHong Zhang 29871393bc97SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 298810c27e3dSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 298910c27e3dSHong Zhang nzk = 0; 29902ed133dbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 29913b4a8b6dSBarry 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); 29922ed133dbSHong Zhang ncols_upper = 0; 29932ed133dbSHong Zhang for (j=0; j<ncols; j++){ 29949bfd6278SHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 29952ed133dbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 29962ed133dbSHong Zhang cols[ncols_upper] = i; 29972ed133dbSHong Zhang ncols_upper++; 29982ed133dbSHong Zhang } 29992ed133dbSHong Zhang } 30001393bc97SHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 300110c27e3dSHong Zhang nzk += nlnk; 300210c27e3dSHong Zhang 300310c27e3dSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 300410c27e3dSHong Zhang prow = jl[k]; /* 1st pivot row */ 300510c27e3dSHong Zhang 300610c27e3dSHong Zhang while (prow < k){ 300710c27e3dSHong Zhang nextprow = jl[prow]; 300810c27e3dSHong Zhang /* merge prow into k-th row */ 30091393bc97SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 301010c27e3dSHong Zhang jmax = ui[prow+1]; 301110c27e3dSHong Zhang ncols = jmax-jmin; 30121393bc97SHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 30135a8e39fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 301410c27e3dSHong Zhang nzk += nlnk; 301510c27e3dSHong Zhang 301610c27e3dSHong Zhang /* update il and jl for prow */ 301710c27e3dSHong Zhang if (jmin < jmax){ 301810c27e3dSHong Zhang il[prow] = jmin; 30192ed133dbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 302010c27e3dSHong Zhang } 302110c27e3dSHong Zhang prow = nextprow; 302210c27e3dSHong Zhang } 302310c27e3dSHong Zhang 302410c27e3dSHong Zhang /* if free space is not available, make more free space */ 302510c27e3dSHong Zhang if (current_space->local_remaining<nzk) { 30261393bc97SHong Zhang i = am - k + 1; /* num of unfactored rows */ 302710c27e3dSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 3028a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 302910c27e3dSHong Zhang reallocs++; 303010c27e3dSHong Zhang } 303110c27e3dSHong Zhang 303210c27e3dSHong Zhang /* copy data into free space, then initialize lnk */ 30331393bc97SHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 303410c27e3dSHong Zhang 303510c27e3dSHong Zhang /* add the k-th row into il and jl */ 303610c27e3dSHong Zhang if (nzk-1 > 0){ 30371393bc97SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 303810c27e3dSHong Zhang jl[k] = jl[i]; jl[i] = k; 303910c27e3dSHong Zhang il[k] = ui[k] + 1; 304010c27e3dSHong Zhang } 30412ed133dbSHong Zhang ui_ptr[k] = current_space->array; 304210c27e3dSHong Zhang current_space->array += nzk; 304310c27e3dSHong Zhang current_space->local_used += nzk; 304410c27e3dSHong Zhang current_space->local_remaining -= nzk; 304510c27e3dSHong Zhang 304610c27e3dSHong Zhang ui[k+1] = ui[k] + nzk; 304710c27e3dSHong Zhang } 304810c27e3dSHong Zhang 30496cf91177SBarry Smith #if defined(PETSC_USE_INFO) 30501393bc97SHong Zhang if (ai[am] != 0) { 30511393bc97SHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 3052ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 3053ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 3054ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 305510c27e3dSHong Zhang } else { 3056ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 305710c27e3dSHong Zhang } 305863ba0a88SBarry Smith #endif 305910c27e3dSHong Zhang 306010c27e3dSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 30619bfd6278SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 30620e83c824SBarry Smith ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 306310c27e3dSHong Zhang 306410c27e3dSHong Zhang /* destroy list of free space and other temporary array(s) */ 30651393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 3066a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 306710c27e3dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 306810c27e3dSHong Zhang 306910c27e3dSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 307010c27e3dSHong Zhang 3071f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 307210c27e3dSHong Zhang b->singlemalloc = PETSC_FALSE; 3073e6b907acSBarry Smith b->free_a = PETSC_TRUE; 3074e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 30751393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 307610c27e3dSHong Zhang b->j = uj; 307710c27e3dSHong Zhang b->i = ui; 307810c27e3dSHong Zhang b->diag = 0; 307910c27e3dSHong Zhang b->ilen = 0; 308010c27e3dSHong Zhang b->imax = 0; 308110c27e3dSHong Zhang b->row = perm; 30829bfd6278SHong Zhang b->col = perm; 30839bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 30849bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 30859bfd6278SHong Zhang b->icol = iperm; 308610c27e3dSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 30871393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3088719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 30891393bc97SHong Zhang b->maxnz = b->nz = ui[am]; 309010c27e3dSHong Zhang 3091f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 3092f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 30931393bc97SHong Zhang if (ai[am] != 0) { 3094f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 309510c27e3dSHong Zhang } else { 3096f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 309710c27e3dSHong Zhang } 30980a3c351aSHong Zhang fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace; 3099f76d2b81SHong Zhang PetscFunctionReturn(0); 3100f76d2b81SHong Zhang } 3101599ef60dSHong Zhang 31021d098868SHong Zhang #undef __FUNCT__ 310335233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering" 310435233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 3105813a1f2bSShri Abhyankar { 3106813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3107813a1f2bSShri Abhyankar PetscErrorCode ierr; 3108813a1f2bSShri Abhyankar PetscInt n = A->rmap->n; 3109813a1f2bSShri Abhyankar const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag,*vi; 3110813a1f2bSShri Abhyankar PetscScalar *x,sum; 3111813a1f2bSShri Abhyankar const PetscScalar *b; 3112813a1f2bSShri Abhyankar const MatScalar *aa = a->a,*v; 3113813a1f2bSShri Abhyankar PetscInt i,nz; 3114813a1f2bSShri Abhyankar 3115813a1f2bSShri Abhyankar PetscFunctionBegin; 3116813a1f2bSShri Abhyankar if (!n) PetscFunctionReturn(0); 3117813a1f2bSShri Abhyankar 3118813a1f2bSShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 3119813a1f2bSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3120813a1f2bSShri Abhyankar 3121813a1f2bSShri Abhyankar /* forward solve the lower triangular */ 3122813a1f2bSShri Abhyankar x[0] = b[0]; 3123813a1f2bSShri Abhyankar v = aa; 3124813a1f2bSShri Abhyankar vi = aj; 3125813a1f2bSShri Abhyankar for (i=1; i<n; i++) { 3126813a1f2bSShri Abhyankar nz = ai[i+1] - ai[i]; 3127813a1f2bSShri Abhyankar sum = b[i]; 3128813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 3129813a1f2bSShri Abhyankar v += nz; 3130813a1f2bSShri Abhyankar vi += nz; 3131813a1f2bSShri Abhyankar x[i] = sum; 3132813a1f2bSShri Abhyankar } 3133813a1f2bSShri Abhyankar 3134813a1f2bSShri Abhyankar /* backward solve the upper triangular */ 313562fbd6afSShri Abhyankar for (i=n-1; i>=0; i--){ 31363c0119dfSShri Abhyankar v = aa + adiag[i+1] + 1; 31373c0119dfSShri Abhyankar vi = aj + adiag[i+1] + 1; 3138813a1f2bSShri Abhyankar nz = adiag[i] - adiag[i+1]-1; 3139813a1f2bSShri Abhyankar sum = x[i]; 3140813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 31413c0119dfSShri Abhyankar x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */ 3142813a1f2bSShri Abhyankar } 3143813a1f2bSShri Abhyankar 3144813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 3145813a1f2bSShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 3146813a1f2bSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3147813a1f2bSShri Abhyankar PetscFunctionReturn(0); 3148813a1f2bSShri Abhyankar } 3149813a1f2bSShri Abhyankar 31501d098868SHong Zhang #undef __FUNCT__ 315135233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ" 315235233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 3153f268cba8SShri Abhyankar { 3154f268cba8SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3155f268cba8SShri Abhyankar IS iscol = a->col,isrow = a->row; 3156f268cba8SShri Abhyankar PetscErrorCode ierr; 3157f268cba8SShri Abhyankar PetscInt i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz; 3158f268cba8SShri Abhyankar const PetscInt *rout,*cout,*r,*c; 3159f268cba8SShri Abhyankar PetscScalar *x,*tmp,sum; 3160f268cba8SShri Abhyankar const PetscScalar *b; 3161f268cba8SShri Abhyankar const MatScalar *aa = a->a,*v; 3162f268cba8SShri Abhyankar 3163f268cba8SShri Abhyankar PetscFunctionBegin; 3164f268cba8SShri Abhyankar if (!n) PetscFunctionReturn(0); 3165f268cba8SShri Abhyankar 3166f268cba8SShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 3167f268cba8SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3168f268cba8SShri Abhyankar tmp = a->solve_work; 3169f268cba8SShri Abhyankar 3170f268cba8SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 3171f268cba8SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 3172f268cba8SShri Abhyankar 3173f268cba8SShri Abhyankar /* forward solve the lower triangular */ 3174f268cba8SShri Abhyankar tmp[0] = b[r[0]]; 3175f268cba8SShri Abhyankar v = aa; 3176f268cba8SShri Abhyankar vi = aj; 3177f268cba8SShri Abhyankar for (i=1; i<n; i++) { 3178f268cba8SShri Abhyankar nz = ai[i+1] - ai[i]; 3179f268cba8SShri Abhyankar sum = b[r[i]]; 3180f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3181f268cba8SShri Abhyankar tmp[i] = sum; 3182f268cba8SShri Abhyankar v += nz; vi += nz; 3183f268cba8SShri Abhyankar } 3184f268cba8SShri Abhyankar 3185f268cba8SShri Abhyankar /* backward solve the upper triangular */ 3186f268cba8SShri Abhyankar for (i=n-1; i>=0; i--){ 31873c0119dfSShri Abhyankar v = aa + adiag[i+1]+1; 31883c0119dfSShri Abhyankar vi = aj + adiag[i+1]+1; 3189f268cba8SShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 3190f268cba8SShri Abhyankar sum = tmp[i]; 3191f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3192f268cba8SShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 3193f268cba8SShri Abhyankar } 3194f268cba8SShri Abhyankar 3195f268cba8SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 3196f268cba8SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 3197f268cba8SShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 3198f268cba8SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3199f268cba8SShri Abhyankar ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 3200f268cba8SShri Abhyankar PetscFunctionReturn(0); 3201f268cba8SShri Abhyankar } 3202f268cba8SShri Abhyankar 3203f268cba8SShri Abhyankar #undef __FUNCT__ 32041d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ" 3205fe97e370SBarry Smith /* 3206fe97e370SBarry Smith This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer seperate functions in the matrix function table for dt factors 3207fe97e370SBarry Smith */ 32081d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact) 32091d098868SHong Zhang { 32101d098868SHong Zhang Mat B = *fact; 3211599ef60dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b; 3212599ef60dSHong Zhang IS isicol; 3213599ef60dSHong Zhang PetscErrorCode ierr; 32141d098868SHong Zhang const PetscInt *r,*ic; 32151d098868SHong Zhang PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag; 3216dc3a2fd3SHong Zhang PetscInt *bi,*bj,*bdiag,*bdiag_rev; 3217f61a948aSLisandro Dalcin PetscInt row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au; 32181d098868SHong Zhang PetscInt nlnk,*lnk; 32191d098868SHong Zhang PetscBT lnkbt; 32201d098868SHong Zhang PetscTruth row_identity,icol_identity,both_identity; 32218aee2decSHong Zhang MatScalar *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp; 32221d098868SHong Zhang const PetscInt *ics; 32231d098868SHong Zhang PetscInt j,nz,*pj,*bjtmp,k,ncut,*jtmp; 3224182b8fbaSHong Zhang PetscReal dt=info->dt,dtcol=info->dtcol,shift=info->shiftamount; 3225f61a948aSLisandro Dalcin PetscInt dtcount=(PetscInt)info->dtcount,nnz_max; 3226599ef60dSHong Zhang PetscTruth missing; 3227599ef60dSHong Zhang 3228599ef60dSHong Zhang PetscFunctionBegin; 3229f61a948aSLisandro Dalcin 3230f61a948aSLisandro Dalcin if (dt == PETSC_DEFAULT) dt = 0.005; 3231f61a948aSLisandro Dalcin if (dtcol == PETSC_DEFAULT) dtcol = 0.01; /* XXX unused! */ 3232f61a948aSLisandro Dalcin if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax); 3233f61a948aSLisandro Dalcin 32341d098868SHong Zhang /* ------- symbolic factorization, can be reused ---------*/ 32351d098868SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 32361d098868SHong Zhang if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 32371d098868SHong Zhang adiag=a->diag; 3238599ef60dSHong Zhang 3239599ef60dSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 3240599ef60dSHong Zhang 3241599ef60dSHong Zhang /* bdiag is location of diagonal in factor */ 32421bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); /* becomes b->diag */ 32431bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag_rev);CHKERRQ(ierr); /* temporary */ 3244599ef60dSHong Zhang 32451d098868SHong Zhang /* allocate row pointers bi */ 32468fc3a347SHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 32471d098868SHong Zhang 32481d098868SHong Zhang /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */ 3249393d3a68SHong Zhang if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */ 32501d098868SHong Zhang nnz_max = ai[n]+2*n*dtcount+2; 32518fc3a347SHong Zhang 32528fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 32538fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr); 3254599ef60dSHong Zhang 3255599ef60dSHong Zhang /* put together the new matrix */ 3256599ef60dSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 3257599ef60dSHong Zhang ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 3258f284f12aSHong Zhang b = (Mat_SeqAIJ*)B->data; 3259599ef60dSHong Zhang b->free_a = PETSC_TRUE; 3260599ef60dSHong Zhang b->free_ij = PETSC_TRUE; 3261599ef60dSHong Zhang b->singlemalloc = PETSC_FALSE; 3262599ef60dSHong Zhang b->a = ba; 3263599ef60dSHong Zhang b->j = bj; 3264599ef60dSHong Zhang b->i = bi; 3265599ef60dSHong Zhang b->diag = bdiag; 3266599ef60dSHong Zhang b->ilen = 0; 3267599ef60dSHong Zhang b->imax = 0; 3268599ef60dSHong Zhang b->row = isrow; 3269599ef60dSHong Zhang b->col = iscol; 3270599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 3271599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 3272599ef60dSHong Zhang b->icol = isicol; 3273599ef60dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3274599ef60dSHong Zhang 32751d098868SHong Zhang ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 32761d098868SHong Zhang b->maxnz = nnz_max; 3277599ef60dSHong Zhang 3278f284f12aSHong Zhang B->factor = MAT_FACTOR_ILUDT; 3279f284f12aSHong Zhang B->info.factor_mallocs = 0; 3280f284f12aSHong Zhang B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]); 32811d098868SHong Zhang CHKMEMQ; 32821d098868SHong Zhang /* ------- end of symbolic factorization ---------*/ 3283599ef60dSHong Zhang 3284599ef60dSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 3285599ef60dSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3286599ef60dSHong Zhang ics = ic; 3287599ef60dSHong Zhang 3288599ef60dSHong Zhang /* linked list for storing column indices of the active row */ 3289599ef60dSHong Zhang nlnk = n + 1; 3290599ef60dSHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 32911d098868SHong Zhang 32921d098868SHong Zhang /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */ 32930e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt,&im,n,PetscInt,&jtmp);CHKERRQ(ierr); 32941d098868SHong Zhang /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */ 32950e83c824SBarry Smith ierr = PetscMalloc2(n,MatScalar,&rtmp,n,MatScalar,&vtmp);CHKERRQ(ierr); 32960e83c824SBarry Smith ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr); 3297599ef60dSHong Zhang 3298599ef60dSHong Zhang bi[0] = 0; 32998fc3a347SHong Zhang bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */ 3300dc3a2fd3SHong Zhang bdiag_rev[n] = bdiag[0]; 33018fc3a347SHong Zhang bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */ 3302599ef60dSHong Zhang for (i=0; i<n; i++) { 3303599ef60dSHong Zhang /* copy initial fill into linked list */ 3304599ef60dSHong Zhang nzi = 0; /* nonzeros for active row i */ 33058369b28dSHong Zhang nzi = ai[r[i]+1] - ai[r[i]]; 33068369b28dSHong 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); 33078369b28dSHong Zhang nzi_al = adiag[r[i]] - ai[r[i]]; 33088369b28dSHong Zhang nzi_au = ai[r[i]+1] - adiag[r[i]] -1; 3309599ef60dSHong Zhang ajtmp = aj + ai[r[i]]; 33108369b28dSHong Zhang ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3311599ef60dSHong Zhang 3312599ef60dSHong Zhang /* load in initial (unfactored row) */ 33131d098868SHong Zhang aatmp = a->a + ai[r[i]]; 33148369b28dSHong Zhang for (j=0; j<nzi; j++) { 33151d098868SHong Zhang rtmp[ics[*ajtmp++]] = *aatmp++; 3316599ef60dSHong Zhang } 3317599ef60dSHong Zhang 3318599ef60dSHong Zhang /* add pivot rows into linked list */ 3319599ef60dSHong Zhang row = lnk[n]; 3320599ef60dSHong Zhang while (row < i ) { 33211d098868SHong Zhang nzi_bl = bi[row+1] - bi[row] + 1; 33221d098868SHong Zhang bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */ 33231d098868SHong Zhang ierr = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr); 3324599ef60dSHong Zhang nzi += nlnk; 3325599ef60dSHong Zhang row = lnk[row]; 3326599ef60dSHong Zhang } 3327599ef60dSHong Zhang 33288369b28dSHong Zhang /* copy data from lnk into jtmp, then initialize lnk */ 33298369b28dSHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr); 3330599ef60dSHong Zhang 3331599ef60dSHong Zhang /* numerical factorization */ 33328369b28dSHong Zhang bjtmp = jtmp; 3333599ef60dSHong Zhang row = *bjtmp++; /* 1st pivot row */ 3334599ef60dSHong Zhang while ( row < i ) { 3335599ef60dSHong Zhang pc = rtmp + row; 33363c2a7987SHong Zhang pv = ba + bdiag[row]; /* 1./(diag of the pivot row) */ 33373c2a7987SHong Zhang multiplier = (*pc) * (*pv); 3338599ef60dSHong Zhang *pc = multiplier; 33393c2a7987SHong Zhang if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */ 33401d098868SHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 33411d098868SHong Zhang pv = ba + bdiag[row+1] + 1; 33421d098868SHong Zhang /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */ 33431d098868SHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 33441d098868SHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 3345dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 3346599ef60dSHong Zhang } 3347599ef60dSHong Zhang row = *bjtmp++; 3348599ef60dSHong Zhang } 3349599ef60dSHong Zhang 33508369b28dSHong Zhang /* copy sparse rtmp into contiguous vtmp; separate L and U part */ 33518aee2decSHong Zhang diag_tmp = rtmp[i]; /* save diagonal value - may not needed?? */ 33528369b28dSHong Zhang nzi_bl = 0; j = 0; 33538369b28dSHong Zhang while (jtmp[j] < i){ /* Note: jtmp is sorted */ 33548aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 33558369b28dSHong Zhang nzi_bl++; j++; 33568369b28dSHong Zhang } 33578369b28dSHong Zhang nzi_bu = nzi - nzi_bl -1; 33588369b28dSHong Zhang while (j < nzi){ 33598aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 33608369b28dSHong Zhang j++; 33618369b28dSHong Zhang } 33628369b28dSHong Zhang 3363599ef60dSHong Zhang bjtmp = bj + bi[i]; 3364599ef60dSHong Zhang batmp = ba + bi[i]; 33658369b28dSHong Zhang /* apply level dropping rule to L part */ 33668369b28dSHong Zhang ncut = nzi_al + dtcount; 33678369b28dSHong Zhang if (ncut < nzi_bl){ 33688369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr); 33698369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr); 3370599ef60dSHong Zhang } else { 33718369b28dSHong Zhang ncut = nzi_bl; 33728369b28dSHong Zhang } 33738369b28dSHong Zhang for (j=0; j<ncut; j++){ 33748369b28dSHong Zhang bjtmp[j] = jtmp[j]; 33758369b28dSHong Zhang batmp[j] = vtmp[j]; 33766da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */ 33778369b28dSHong Zhang } 33781d098868SHong Zhang bi[i+1] = bi[i] + ncut; 33798369b28dSHong Zhang nzi = ncut + 1; 33808369b28dSHong Zhang 33818369b28dSHong Zhang /* apply level dropping rule to U part */ 33828369b28dSHong Zhang ncut = nzi_au + dtcount; 33838369b28dSHong Zhang if (ncut < nzi_bu){ 33848369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr); 33858369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr); 33868369b28dSHong Zhang } else { 33878369b28dSHong Zhang ncut = nzi_bu; 33888369b28dSHong Zhang } 33898369b28dSHong Zhang nzi += ncut; 33901d098868SHong Zhang 33911d098868SHong Zhang /* mark bdiagonal */ 33921d098868SHong Zhang bdiag[i+1] = bdiag[i] - (ncut + 1); 3393dc3a2fd3SHong Zhang bdiag_rev[n-i-1] = bdiag[i+1]; 33948fc3a347SHong Zhang bi[2*n - i] = bi[2*n - i +1] - (ncut + 1); 33951d098868SHong Zhang bjtmp = bj + bdiag[i]; 33961d098868SHong Zhang batmp = ba + bdiag[i]; 33971d098868SHong Zhang *bjtmp = i; 33988aee2decSHong Zhang *batmp = diag_tmp; /* rtmp[i]; */ 3399c9c72f8fSHong Zhang if (*batmp == 0.0) { 3400c9c72f8fSHong Zhang *batmp = dt+shift; 3401bd1bc851SBarry Smith /* printf(" row %d add shift %g\n",i,shift); */ 3402c9c72f8fSHong Zhang } 3403b78a477dSHong Zhang *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */ 34046da515a1SHong Zhang /* printf(" (%d,%g),",*bjtmp,*batmp); */ 34051d098868SHong Zhang 34061d098868SHong Zhang bjtmp = bj + bdiag[i+1]+1; 34071d098868SHong Zhang batmp = ba + bdiag[i+1]+1; 34088369b28dSHong Zhang for (k=0; k<ncut; k++){ 34091d098868SHong Zhang bjtmp[k] = jtmp[nzi_bl+1+k]; 34101d098868SHong Zhang batmp[k] = vtmp[nzi_bl+1+k]; 34116da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */ 3412599ef60dSHong Zhang } 34136da515a1SHong Zhang /* printf("\n"); */ 3414599ef60dSHong Zhang 3415599ef60dSHong Zhang im[i] = nzi; /* used by PetscLLAddSortedLU() */ 3416599ef60dSHong Zhang /* 34171d098868SHong Zhang printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]); 34188369b28dSHong Zhang printf(" ----------------------------\n"); 3419599ef60dSHong Zhang */ 34208369b28dSHong Zhang } /* for (i=0; i<n; i++) */ 34211d098868SHong Zhang /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */ 34221d098868SHong 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]); 3423599ef60dSHong Zhang 3424599ef60dSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3425599ef60dSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3426599ef60dSHong Zhang 3427599ef60dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 34280e83c824SBarry Smith ierr = PetscFree2(im,jtmp);CHKERRQ(ierr); 34290e83c824SBarry Smith ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr); 34301bfa06eaSJed Brown ierr = PetscFree(bdiag_rev);CHKERRQ(ierr); 3431599ef60dSHong Zhang 3432599ef60dSHong Zhang ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr); 34331d098868SHong Zhang b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n]; 3434599ef60dSHong Zhang 3435599ef60dSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 3436599ef60dSHong Zhang ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr); 3437599ef60dSHong Zhang both_identity = (PetscTruth) (row_identity && icol_identity); 3438599ef60dSHong Zhang if (row_identity && icol_identity) { 343935233d99SShri Abhyankar B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 3440599ef60dSHong Zhang } else { 344135233d99SShri Abhyankar B->ops->solve = MatSolve_SeqAIJ; 3442599ef60dSHong Zhang } 3443599ef60dSHong Zhang 34441d098868SHong Zhang B->ops->solveadd = 0; 34451d098868SHong Zhang B->ops->solvetranspose = 0; 34461d098868SHong Zhang B->ops->solvetransposeadd = 0; 34471d098868SHong Zhang B->ops->matsolve = 0; 3448599ef60dSHong Zhang B->assembled = PETSC_TRUE; 3449599ef60dSHong Zhang B->preallocated = PETSC_TRUE; 3450599ef60dSHong Zhang PetscFunctionReturn(0); 3451599ef60dSHong Zhang } 3452599ef60dSHong Zhang 34533c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */ 34543c2a7987SHong Zhang #undef __FUNCT__ 34553c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ" 3456fe97e370SBarry Smith /* 3457fe97e370SBarry Smith This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer seperate functions in the matrix function table for dt factors 3458fe97e370SBarry Smith */ 3459fe97e370SBarry Smith 34603c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info) 34613c2a7987SHong Zhang { 34623c2a7987SHong Zhang PetscErrorCode ierr; 34633c2a7987SHong Zhang 34643c2a7987SHong Zhang PetscFunctionBegin; 34653c2a7987SHong Zhang ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr); 34663c2a7987SHong Zhang PetscFunctionReturn(0); 34673c2a7987SHong Zhang } 34683c2a7987SHong Zhang 34693c2a7987SHong Zhang /* 34703c2a7987SHong Zhang same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors 34713c2a7987SHong Zhang - intend to replace existing MatLUFactorNumeric_SeqAIJ() 34723c2a7987SHong Zhang */ 34733c2a7987SHong Zhang #undef __FUNCT__ 34743c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ" 3475fe97e370SBarry Smith /* 3476fe97e370SBarry Smith This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer seperate functions in the matrix function table for dt factors 3477fe97e370SBarry Smith */ 3478fe97e370SBarry Smith 34793c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info) 34803c2a7987SHong Zhang { 34813c2a7987SHong Zhang Mat C=fact; 34823c2a7987SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 34833c2a7987SHong Zhang IS isrow = b->row,isicol = b->icol; 34843c2a7987SHong Zhang PetscErrorCode ierr; 34853c2a7987SHong Zhang const PetscInt *r,*ic,*ics; 3486b78a477dSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 3487b78a477dSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj; 34883c2a7987SHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 3489182b8fbaSHong Zhang PetscReal dt=info->dt,shift=info->shiftamount; 3490b78a477dSHong Zhang PetscTruth row_identity, col_identity; 34913c2a7987SHong Zhang 34923c2a7987SHong Zhang PetscFunctionBegin; 34933c2a7987SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 34943c2a7987SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3495b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 34963c2a7987SHong Zhang ics = ic; 34973c2a7987SHong Zhang 34983c2a7987SHong Zhang for (i=0; i<n; i++){ 3499b78a477dSHong Zhang /* initialize rtmp array */ 3500b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of nozeros in L(i,:) */ 35013c2a7987SHong Zhang bjtmp = bj + bi[i]; 3502b78a477dSHong Zhang for (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0; 3503b78a477dSHong Zhang rtmp[i] = 0.0; 3504b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */ 3505b78a477dSHong Zhang bjtmp = bj + bdiag[i+1] + 1; 3506b78a477dSHong Zhang for (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0; 35073c2a7987SHong Zhang 3508b78a477dSHong Zhang /* load in initial unfactored row of A */ 35096da515a1SHong Zhang /* printf("row %d\n",i); */ 35103c2a7987SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 35113c2a7987SHong Zhang ajtmp = aj + ai[r[i]]; 35123c2a7987SHong Zhang v = aa + ai[r[i]]; 35133c2a7987SHong Zhang for (j=0; j<nz; j++) { 3514b78a477dSHong Zhang rtmp[ics[*ajtmp++]] = v[j]; 35156da515a1SHong Zhang /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */ 35163c2a7987SHong Zhang } 35176da515a1SHong Zhang /* printf("\n"); */ 35183c2a7987SHong Zhang 3519b78a477dSHong Zhang /* numerical factorization */ 3520b78a477dSHong Zhang bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */ 3521b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of entries in L(i,:) */ 3522b78a477dSHong Zhang k = 0; 3523b78a477dSHong Zhang while (k < nzl){ 35243c2a7987SHong Zhang row = *bjtmp++; 35256da515a1SHong Zhang /* printf(" prow %d\n",row); */ 35263c2a7987SHong Zhang pc = rtmp + row; 3527b78a477dSHong Zhang pv = b->a + bdiag[row]; /* 1./(diag of the pivot row) */ 3528b78a477dSHong Zhang multiplier = (*pc) * (*pv); 35293c2a7987SHong Zhang *pc = multiplier; 3530b78a477dSHong Zhang if (PetscAbsScalar(multiplier) > dt){ 3531b78a477dSHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 3532b78a477dSHong Zhang pv = b->a + bdiag[row+1] + 1; 3533b78a477dSHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 3534b78a477dSHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 35356da515a1SHong Zhang /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */ 35363c2a7987SHong Zhang } 3537b78a477dSHong Zhang k++; 35383c2a7987SHong Zhang } 35393c2a7987SHong Zhang 3540b78a477dSHong Zhang /* finished row so stick it into b->a */ 3541b78a477dSHong Zhang /* L-part */ 3542b78a477dSHong Zhang pv = b->a + bi[i] ; 3543b78a477dSHong Zhang pj = bj + bi[i] ; 3544b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; 3545b78a477dSHong Zhang for (j=0; j<nzl; j++) { 3546b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 35476da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 35483c2a7987SHong Zhang } 3549b78a477dSHong Zhang 3550b78a477dSHong Zhang /* diagonal: invert diagonal entries for simplier triangular solves */ 3551b78a477dSHong Zhang if (rtmp[i] == 0.0) rtmp[i] = dt+shift; 3552b78a477dSHong Zhang b->a[bdiag[i]] = 1.0/rtmp[i]; 35536da515a1SHong Zhang /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */ 3554b78a477dSHong Zhang 3555b78a477dSHong Zhang /* U-part */ 3556b78a477dSHong Zhang pv = b->a + bdiag[i+1] + 1; 3557b78a477dSHong Zhang pj = bj + bdiag[i+1] + 1; 3558b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1] - 1; 3559b78a477dSHong Zhang for (j=0; j<nzu; j++) { 3560b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 35616da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 3562b78a477dSHong Zhang } 35636da515a1SHong Zhang /* printf("\n"); */ 3564b78a477dSHong Zhang } 3565b78a477dSHong Zhang 35663c2a7987SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 35673c2a7987SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 35683c2a7987SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3569b78a477dSHong Zhang 35703c2a7987SHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 35713c2a7987SHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 35723c2a7987SHong Zhang if (row_identity && col_identity) { 357335233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 35743c2a7987SHong Zhang } else { 357535233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ; 35763c2a7987SHong Zhang } 3577b78a477dSHong Zhang C->ops->solveadd = 0; 3578b78a477dSHong Zhang C->ops->solvetranspose = 0; 3579b78a477dSHong Zhang C->ops->solvetransposeadd = 0; 3580b78a477dSHong Zhang C->ops->matsolve = 0; 35813c2a7987SHong Zhang C->assembled = PETSC_TRUE; 35823c2a7987SHong Zhang C->preallocated = PETSC_TRUE; 35833c2a7987SHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 35843c2a7987SHong Zhang PetscFunctionReturn(0); 35853c2a7987SHong Zhang } 3586