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__ 11a7a54a73SBarry Smith #define __FUNCT__ "MatGetOrdering_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 14312e372aSBarry Smith 15312e372aSBarry Smith This code does not work and is not called anywhere. It would be registered with MatOrderingRegisterAll() 168fa24674SBarry Smith */ 17a7a54a73SBarry Smith PetscErrorCode MatGetOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol) 183b2fbd54SBarry Smith { 198fa24674SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)mat->data; 208fa24674SBarry Smith PetscErrorCode ierr; 21889a8dedSBarry Smith PetscInt i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order; 228fa24674SBarry Smith const PetscInt *ai = a->i, *aj = a->j; 238fa24674SBarry Smith const PetscScalar *aa = a->a; 24ace3abfcSBarry Smith PetscBool *done; 25889a8dedSBarry Smith PetscReal best,past = 0,future; 263a40ed3dSBarry Smith 278fa24674SBarry Smith PetscFunctionBegin; 288fa24674SBarry Smith /* pick initial row */ 298fa24674SBarry Smith best = -1; 308fa24674SBarry Smith for (i=0; i<n; i++) { 3175567043SBarry Smith future = 0.0; 328fa24674SBarry Smith for (j=ai[i]; j<ai[i+1]; j++) { 338fa24674SBarry Smith if (aj[j] != i) future += PetscAbsScalar(aa[j]); else past = PetscAbsScalar(aa[j]); 348fa24674SBarry Smith } 358fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 368fa24674SBarry Smith if (past/future > best) { 378fa24674SBarry Smith best = past/future; 388fa24674SBarry Smith current = i; 398fa24674SBarry Smith } 408fa24674SBarry Smith } 418fa24674SBarry Smith 42ace3abfcSBarry Smith ierr = PetscMalloc(n*sizeof(PetscBool ),&done);CHKERRQ(ierr); 43ace3abfcSBarry Smith ierr = PetscMemzero(done,n*sizeof(PetscBool ));CHKERRQ(ierr); 440e83c824SBarry Smith ierr = PetscMalloc(n*sizeof(PetscInt),&order);CHKERRQ(ierr); 458fa24674SBarry Smith order[0] = current; 468fa24674SBarry Smith for (i=0; i<n-1; i++) { 478fa24674SBarry Smith done[current] = PETSC_TRUE; 488fa24674SBarry Smith best = -1; 498fa24674SBarry Smith /* loop over all neighbors of current pivot */ 508fa24674SBarry Smith for (j=ai[current]; j<ai[current+1]; j++) { 518fa24674SBarry Smith jj = aj[j]; 528fa24674SBarry Smith if (done[jj]) continue; 538fa24674SBarry Smith /* loop over columns of potential next row computing weights for below and above diagonal */ 548fa24674SBarry Smith past = future = 0.0; 558fa24674SBarry Smith for (k=ai[jj]; k<ai[jj+1]; k++) { 568fa24674SBarry Smith kk = aj[k]; 578fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[k]); 588fa24674SBarry Smith else if (kk != jj) future += PetscAbsScalar(aa[k]); 598fa24674SBarry Smith } 608fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 618fa24674SBarry Smith if (past/future > best) { 628fa24674SBarry Smith best = past/future; 638fa24674SBarry Smith newcurrent = jj; 648fa24674SBarry Smith } 658fa24674SBarry Smith } 668fa24674SBarry Smith if (best == -1) { /* no neighbors to select from so select best of all that remain */ 678fa24674SBarry Smith best = -1; 688fa24674SBarry Smith for (k=0; k<n; k++) { 698fa24674SBarry Smith if (done[k]) continue; 7075567043SBarry Smith future = 0.0; 7175567043SBarry Smith past = 0.0; 728fa24674SBarry Smith for (j=ai[k]; j<ai[k+1]; j++) { 738fa24674SBarry Smith kk = aj[j]; 748fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[j]); 758fa24674SBarry Smith else if (kk != k) future += PetscAbsScalar(aa[j]); 768fa24674SBarry Smith } 778fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 788fa24674SBarry Smith if (past/future > best) { 798fa24674SBarry Smith best = past/future; 808fa24674SBarry Smith newcurrent = k; 818fa24674SBarry Smith } 828fa24674SBarry Smith } 838fa24674SBarry Smith } 84e32f2f54SBarry Smith if (current == newcurrent) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"newcurrent cannot be current"); 858fa24674SBarry Smith current = newcurrent; 868fa24674SBarry Smith order[i+1] = current; 878fa24674SBarry Smith } 8870b3c8c7SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,order,PETSC_COPY_VALUES,irow);CHKERRQ(ierr); 898fa24674SBarry Smith *icol = *irow; 908fa24674SBarry Smith ierr = PetscObjectReference((PetscObject)*irow);CHKERRQ(ierr); 918fa24674SBarry Smith ierr = PetscFree(done);CHKERRQ(ierr); 928fa24674SBarry Smith ierr = PetscFree(order);CHKERRQ(ierr); 93d64ed03dSBarry Smith PetscFunctionReturn(0); 943b2fbd54SBarry Smith } 95f6a26d55SMatthew Knepley EXTERN_C_END 963b2fbd54SBarry Smith 97e631078cSBarry Smith EXTERN_C_BEGIN 984a2ae208SSatish Balay #undef __FUNCT__ 99db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc" 100ace3abfcSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscBool *flg) 101db4efbfdSBarry Smith { 102db4efbfdSBarry Smith PetscFunctionBegin; 103db4efbfdSBarry Smith *flg = PETSC_TRUE; 104db4efbfdSBarry Smith PetscFunctionReturn(0); 105db4efbfdSBarry Smith } 106db4efbfdSBarry Smith EXTERN_C_END 107db4efbfdSBarry Smith 108db4efbfdSBarry Smith EXTERN_C_BEGIN 109db4efbfdSBarry Smith #undef __FUNCT__ 110b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc" 111b24902e0SBarry Smith PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B) 112b24902e0SBarry Smith { 113d0f46423SBarry Smith PetscInt n = A->rmap->n; 114b24902e0SBarry Smith PetscErrorCode ierr; 115b24902e0SBarry Smith 116b24902e0SBarry Smith PetscFunctionBegin; 117b24902e0SBarry Smith ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr); 118b24902e0SBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 119599ef60dSHong Zhang if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT){ 120b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr); 12135233d99SShri Abhyankar (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ; 12235233d99SShri Abhyankar (*B)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqAIJ; 123b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 124b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 125b24902e0SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 12635233d99SShri Abhyankar (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqAIJ; 12735233d99SShri Abhyankar (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ; 128e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported"); 129d5f3da31SBarry Smith (*B)->factortype = ftype; 130b24902e0SBarry Smith PetscFunctionReturn(0); 131b24902e0SBarry Smith } 132e631078cSBarry Smith EXTERN_C_END 133b24902e0SBarry Smith 134b24902e0SBarry Smith #undef __FUNCT__ 135ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_inplace" 136ad04f41aSHong Zhang PetscErrorCode MatLUFactorSymbolic_SeqAIJ_inplace(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 137289bc588SBarry Smith { 138416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 139289bc588SBarry Smith IS isicol; 1406849ba73SBarry Smith PetscErrorCode ierr; 1415d0c19d7SBarry Smith const PetscInt *r,*ic; 1425d0c19d7SBarry Smith PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 1431393bc97SHong Zhang PetscInt *bi,*bj,*ajtmp; 1441393bc97SHong Zhang PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 1459e046878SBarry Smith PetscReal f; 1464875ba38SHong Zhang PetscInt nlnk,*lnk,k,**bi_ptr; 147a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1481393bc97SHong Zhang PetscBT lnkbt; 149289bc588SBarry Smith 1503a40ed3dSBarry Smith PetscFunctionBegin; 151e32f2f54SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square"); 1524c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 153ac121b3dSBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 154ac121b3dSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 155289bc588SBarry Smith 156289bc588SBarry Smith /* get new row pointers */ 1571393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 1581393bc97SHong Zhang bi[0] = 0; 1591393bc97SHong Zhang 1601393bc97SHong Zhang /* bdiag is location of diagonal in factor */ 1611393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1621393bc97SHong Zhang bdiag[0] = 0; 1631393bc97SHong Zhang 1641393bc97SHong Zhang /* linked list for storing column indices of the active row */ 1651393bc97SHong Zhang nlnk = n + 1; 1661393bc97SHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1671393bc97SHong Zhang 16835baf27eSSatish Balay ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 1691393bc97SHong Zhang 1701393bc97SHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 171d3d32019SBarry Smith f = info->fill; 172a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1731393bc97SHong Zhang current_space = free_space; 174289bc588SBarry Smith 175289bc588SBarry Smith for (i=0; i<n; i++) { 1761393bc97SHong Zhang /* copy previous fill into linked list */ 177289bc588SBarry Smith nzi = 0; 1781393bc97SHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 179e32f2f54SBarry Smith if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 1801393bc97SHong Zhang ajtmp = aj + ai[r[i]]; 181afcc9904SHong Zhang ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1821393bc97SHong Zhang nzi += nlnk; 1831393bc97SHong Zhang 1841393bc97SHong Zhang /* add pivot rows into linked list */ 1851393bc97SHong Zhang row = lnk[n]; 1861393bc97SHong Zhang while (row < i) { 187d1170560SHong Zhang nzbd = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */ 188d1170560SHong Zhang ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 189d1170560SHong Zhang ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 1901393bc97SHong Zhang nzi += nlnk; 1911393bc97SHong Zhang row = lnk[row]; 192289bc588SBarry Smith } 1931393bc97SHong Zhang bi[i+1] = bi[i] + nzi; 1941393bc97SHong Zhang im[i] = nzi; 1951393bc97SHong Zhang 1961393bc97SHong Zhang /* mark bdiag */ 1971393bc97SHong Zhang nzbd = 0; 1981393bc97SHong Zhang nnz = nzi; 1991393bc97SHong Zhang k = lnk[n]; 2001393bc97SHong Zhang while (nnz-- && k < i){ 2011393bc97SHong Zhang nzbd++; 2021393bc97SHong Zhang k = lnk[k]; 2031393bc97SHong Zhang } 2041393bc97SHong Zhang bdiag[i] = bi[i] + nzbd; 2051393bc97SHong Zhang 2061393bc97SHong Zhang /* if free space is not available, make more free space */ 2071393bc97SHong Zhang if (current_space->local_remaining<nzi) { 2081393bc97SHong Zhang nnz = (n - i)*nzi; /* estimated and max additional space needed */ 209a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2101393bc97SHong Zhang reallocs++; 2111393bc97SHong Zhang } 2121393bc97SHong Zhang 2131393bc97SHong Zhang /* copy data into free space, then initialize lnk */ 2141393bc97SHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 2151393bc97SHong Zhang bi_ptr[i] = current_space->array; 2161393bc97SHong Zhang current_space->array += nzi; 2171393bc97SHong Zhang current_space->local_used += nzi; 2181393bc97SHong Zhang current_space->local_remaining -= nzi; 219289bc588SBarry Smith } 2206cf91177SBarry Smith #if defined(PETSC_USE_INFO) 221464e8d44SSatish Balay if (ai[n] != 0) { 2221393bc97SHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 223ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 224ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 225ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 226ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 22705bf559cSSatish Balay } else { 228ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 22905bf559cSSatish Balay } 23063ba0a88SBarry Smith #endif 2312fb3ed76SBarry Smith 232898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 233898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2341987afe7SBarry Smith 2351393bc97SHong Zhang /* destroy list of free space and other temporary array(s) */ 2361393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 237a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); 2381393bc97SHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 23935baf27eSSatish Balay ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 240289bc588SBarry Smith 241289bc588SBarry Smith /* put together the new matrix */ 242719d5645SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 243719d5645SBarry Smith ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 244719d5645SBarry Smith b = (Mat_SeqAIJ*)(B)->data; 245e6b907acSBarry Smith b->free_a = PETSC_TRUE; 246e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 2477c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 2481393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 2491393bc97SHong Zhang b->j = bj; 2501393bc97SHong Zhang b->i = bi; 2511393bc97SHong Zhang b->diag = bdiag; 252416022c9SBarry Smith b->ilen = 0; 253416022c9SBarry Smith b->imax = 0; 254416022c9SBarry Smith b->row = isrow; 255416022c9SBarry Smith b->col = iscol; 256c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 257c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 25882bf6240SBarry Smith b->icol = isicol; 25987828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2601393bc97SHong Zhang 2611393bc97SHong Zhang /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 262719d5645SBarry Smith ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 2631393bc97SHong Zhang b->maxnz = b->nz = bi[n] ; 2647fd98bd6SLois Curfman McInnes 265d5f3da31SBarry Smith (B)->factortype = MAT_FACTOR_LU; 266719d5645SBarry Smith (B)->info.factor_mallocs = reallocs; 267719d5645SBarry Smith (B)->info.fill_ratio_given = f; 268703deb49SSatish Balay 2699f7d0b68SHong Zhang if (ai[n]) { 270719d5645SBarry Smith (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 271eea2913aSSatish Balay } else { 272719d5645SBarry Smith (B)->info.fill_ratio_needed = 0.0; 273eea2913aSSatish Balay } 274ad04f41aSHong Zhang (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 275f2344125SBarry Smith if (a->inode.size) { 276f2344125SBarry Smith (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 277f2344125SBarry Smith } 2783a40ed3dSBarry Smith PetscFunctionReturn(0); 279289bc588SBarry Smith } 2801393bc97SHong Zhang 281ce3d78c0SShri Abhyankar #undef __FUNCT__ 28235233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ" 28335233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 284ce3d78c0SShri Abhyankar { 285ce3d78c0SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 286ce3d78c0SShri Abhyankar IS isicol; 287ce3d78c0SShri Abhyankar PetscErrorCode ierr; 288ce3d78c0SShri Abhyankar const PetscInt *r,*ic; 289ce3d78c0SShri Abhyankar PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 290ce3d78c0SShri Abhyankar PetscInt *bi,*bj,*ajtmp; 291ce3d78c0SShri Abhyankar PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 292ce3d78c0SShri Abhyankar PetscReal f; 293ce3d78c0SShri Abhyankar PetscInt nlnk,*lnk,k,**bi_ptr; 294ce3d78c0SShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 295ce3d78c0SShri Abhyankar PetscBT lnkbt; 296ace3abfcSBarry Smith PetscBool olddatastruct=PETSC_FALSE; 297ce3d78c0SShri Abhyankar 298ce3d78c0SShri Abhyankar PetscFunctionBegin; 2993898ab1dSSatish Balay /* Uncomment the oldatastruct part only while testing new data structure for MatSolve() */ 300*acfcf0e5SJed Brown ierr = PetscOptionsGetBool(PETSC_NULL,"-lu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr); 301628f99d7SShri Abhyankar if(olddatastruct){ 302628f99d7SShri Abhyankar ierr = MatLUFactorSymbolic_SeqAIJ_inplace(B,A,isrow,iscol,info);CHKERRQ(ierr); 303628f99d7SShri Abhyankar PetscFunctionReturn(0); 304628f99d7SShri Abhyankar } 305628f99d7SShri Abhyankar 306e32f2f54SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square"); 307ce3d78c0SShri Abhyankar ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 308ce3d78c0SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 309ce3d78c0SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 310ce3d78c0SShri Abhyankar 3111bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 3121bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 3131bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 314a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 3159b48462bSShri Abhyankar 3169b48462bSShri Abhyankar /* linked list for storing column indices of the active row */ 3179b48462bSShri Abhyankar nlnk = n + 1; 3189b48462bSShri Abhyankar ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3199b48462bSShri Abhyankar 3209b48462bSShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 3219b48462bSShri Abhyankar 3229b48462bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 3239b48462bSShri Abhyankar f = info->fill; 3249b48462bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 3259b48462bSShri Abhyankar current_space = free_space; 3269b48462bSShri Abhyankar 3279b48462bSShri Abhyankar for (i=0; i<n; i++) { 3289b48462bSShri Abhyankar /* copy previous fill into linked list */ 3299b48462bSShri Abhyankar nzi = 0; 3309b48462bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 331e32f2f54SBarry Smith if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 3329b48462bSShri Abhyankar ajtmp = aj + ai[r[i]]; 3339b48462bSShri Abhyankar ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3349b48462bSShri Abhyankar nzi += nlnk; 3359b48462bSShri Abhyankar 3369b48462bSShri Abhyankar /* add pivot rows into linked list */ 3379b48462bSShri Abhyankar row = lnk[n]; 3389b48462bSShri Abhyankar while (row < i){ 3399b48462bSShri Abhyankar nzbd = bdiag[row] + 1; /* num of entries in the row with column index <= row */ 3409b48462bSShri Abhyankar ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 3419b48462bSShri Abhyankar ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 3429b48462bSShri Abhyankar nzi += nlnk; 3439b48462bSShri Abhyankar row = lnk[row]; 3449b48462bSShri Abhyankar } 3459b48462bSShri Abhyankar bi[i+1] = bi[i] + nzi; 3469b48462bSShri Abhyankar im[i] = nzi; 3479b48462bSShri Abhyankar 3489b48462bSShri Abhyankar /* mark bdiag */ 3499b48462bSShri Abhyankar nzbd = 0; 3509b48462bSShri Abhyankar nnz = nzi; 3519b48462bSShri Abhyankar k = lnk[n]; 3529b48462bSShri Abhyankar while (nnz-- && k < i){ 3539b48462bSShri Abhyankar nzbd++; 3549b48462bSShri Abhyankar k = lnk[k]; 3559b48462bSShri Abhyankar } 3569b48462bSShri Abhyankar bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */ 3579b48462bSShri Abhyankar 3589b48462bSShri Abhyankar /* if free space is not available, make more free space */ 3599b48462bSShri Abhyankar if (current_space->local_remaining<nzi) { 3609b48462bSShri Abhyankar nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */ 3619b48462bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 3629b48462bSShri Abhyankar reallocs++; 3639b48462bSShri Abhyankar } 3649b48462bSShri Abhyankar 3659b48462bSShri Abhyankar /* copy data into free space, then initialize lnk */ 3669b48462bSShri Abhyankar ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 3679b48462bSShri Abhyankar bi_ptr[i] = current_space->array; 3689b48462bSShri Abhyankar current_space->array += nzi; 3699b48462bSShri Abhyankar current_space->local_used += nzi; 3709b48462bSShri Abhyankar current_space->local_remaining -= nzi; 3719b48462bSShri Abhyankar } 3729b48462bSShri Abhyankar 3739b48462bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3749b48462bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3759b48462bSShri Abhyankar 3769263d837SHong Zhang /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 3779b48462bSShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 3782ce24eb6SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 3799b48462bSShri Abhyankar ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 3809b48462bSShri Abhyankar ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 3819b48462bSShri Abhyankar 3829b48462bSShri Abhyankar /* put together the new matrix */ 3839b48462bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 3849b48462bSShri Abhyankar ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 3859b48462bSShri Abhyankar b = (Mat_SeqAIJ*)(B)->data; 3869b48462bSShri Abhyankar b->free_a = PETSC_TRUE; 3879b48462bSShri Abhyankar b->free_ij = PETSC_TRUE; 3889b48462bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 3899b48462bSShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 3909b48462bSShri Abhyankar b->j = bj; 3919b48462bSShri Abhyankar b->i = bi; 3929b48462bSShri Abhyankar b->diag = bdiag; 3939b48462bSShri Abhyankar b->ilen = 0; 3949b48462bSShri Abhyankar b->imax = 0; 3959b48462bSShri Abhyankar b->row = isrow; 3969b48462bSShri Abhyankar b->col = iscol; 3979b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 3989b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 3999b48462bSShri Abhyankar b->icol = isicol; 4009b48462bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 4019b48462bSShri Abhyankar 4029b48462bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 4039b48462bSShri Abhyankar ierr = PetscLogObjectMemory(B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 4049b48462bSShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 405d5f3da31SBarry Smith B->factortype = MAT_FACTOR_LU; 406f284f12aSHong Zhang B->info.factor_mallocs = reallocs; 407f284f12aSHong Zhang B->info.fill_ratio_given = f; 4089b48462bSShri Abhyankar 4099b48462bSShri Abhyankar if (ai[n]) { 410f284f12aSHong Zhang B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 4119b48462bSShri Abhyankar } else { 412f284f12aSHong Zhang B->info.fill_ratio_needed = 0.0; 4139b48462bSShri Abhyankar } 4149263d837SHong Zhang #if defined(PETSC_USE_INFO) 4159263d837SHong Zhang if (ai[n] != 0) { 4169263d837SHong Zhang PetscReal af = B->info.fill_ratio_needed; 4179263d837SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 4189263d837SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 4199263d837SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 4209263d837SHong Zhang ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 4219263d837SHong Zhang } else { 4229263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 4239263d837SHong Zhang } 4249263d837SHong Zhang #endif 42535233d99SShri Abhyankar B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 426f2344125SBarry Smith if (a->inode.size) { 427d3ac4fa3SBarry Smith B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 428d3ac4fa3SBarry Smith } 4299b48462bSShri Abhyankar PetscFunctionReturn(0); 4309b48462bSShri Abhyankar } 4319b48462bSShri Abhyankar 4325b5bc046SBarry Smith /* 4335b5bc046SBarry Smith Trouble in factorization, should we dump the original matrix? 4345b5bc046SBarry Smith */ 4355b5bc046SBarry Smith #undef __FUNCT__ 4365b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix" 4375b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A) 4385b5bc046SBarry Smith { 4395b5bc046SBarry Smith PetscErrorCode ierr; 440ace3abfcSBarry Smith PetscBool flg = PETSC_FALSE; 4415b5bc046SBarry Smith 4425b5bc046SBarry Smith PetscFunctionBegin; 443*acfcf0e5SJed Brown ierr = PetscOptionsGetBool(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr); 4445b5bc046SBarry Smith if (flg) { 4455b5bc046SBarry Smith PetscViewer viewer; 4465b5bc046SBarry Smith char filename[PETSC_MAX_PATH_LEN]; 4475b5bc046SBarry Smith 4485b5bc046SBarry Smith ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr); 4497adad957SLisandro Dalcin ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 4505b5bc046SBarry Smith ierr = MatView(A,viewer);CHKERRQ(ierr); 4515b5bc046SBarry Smith ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr); 4525b5bc046SBarry Smith } 4535b5bc046SBarry Smith PetscFunctionReturn(0); 4545b5bc046SBarry Smith } 4555b5bc046SBarry Smith 456c9c72f8fSHong Zhang #undef __FUNCT__ 45735233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ" 45835233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 459c9c72f8fSHong Zhang { 460c9c72f8fSHong Zhang Mat C=B; 461c9c72f8fSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 462c9c72f8fSHong Zhang IS isrow = b->row,isicol = b->icol; 463c9c72f8fSHong Zhang PetscErrorCode ierr; 464c9c72f8fSHong Zhang const PetscInt *r,*ic,*ics; 465bbd65245SShri Abhyankar const PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag; 466bbd65245SShri Abhyankar PetscInt i,j,k,nz,nzL,row,*pj; 467bbd65245SShri Abhyankar const PetscInt *ajtmp,*bjtmp; 468bbd65245SShri Abhyankar MatScalar *rtmp,*pc,multiplier,*pv; 469bbd65245SShri Abhyankar const MatScalar *aa=a->a,*v; 470ace3abfcSBarry Smith PetscBool row_identity,col_identity; 471d90ac83dSHong Zhang FactorShiftCtx sctx; 4724f81c4b7SBarry Smith const PetscInt *ddiag; 473d4ad06f7SHong Zhang PetscReal rs; 474d4ad06f7SHong Zhang MatScalar d; 475d4ad06f7SHong Zhang 476c9c72f8fSHong Zhang PetscFunctionBegin; 477d6e5152cSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 478d90ac83dSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 479d4ad06f7SHong Zhang 480f4db908eSBarry Smith if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 481d4ad06f7SHong Zhang ddiag = a->diag; 482d4ad06f7SHong Zhang sctx.shift_top = info->zeropivot; 483d4ad06f7SHong Zhang for (i=0; i<n; i++) { 484d4ad06f7SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 485d4ad06f7SHong Zhang d = (aa)[ddiag[i]]; 486d4ad06f7SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 487d4ad06f7SHong Zhang v = aa+ai[i]; 488d4ad06f7SHong Zhang nz = ai[i+1] - ai[i]; 489d4ad06f7SHong Zhang for (j=0; j<nz; j++) 490d4ad06f7SHong Zhang rs += PetscAbsScalar(v[j]); 491d4ad06f7SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 492d4ad06f7SHong Zhang } 493d4ad06f7SHong Zhang sctx.shift_top *= 1.1; 494d4ad06f7SHong Zhang sctx.nshift_max = 5; 495d4ad06f7SHong Zhang sctx.shift_lo = 0.; 496d4ad06f7SHong Zhang sctx.shift_hi = 1.; 497d4ad06f7SHong Zhang } 498d4ad06f7SHong Zhang 499c9c72f8fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 500c9c72f8fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 501c9c72f8fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 502c9c72f8fSHong Zhang ics = ic; 503c9c72f8fSHong Zhang 504d4ad06f7SHong Zhang do { 50507b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 506c9c72f8fSHong Zhang for (i=0; i<n; i++){ 507c9c72f8fSHong Zhang /* zero rtmp */ 508c9c72f8fSHong Zhang /* L part */ 509c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 510c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 511c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 512c9c72f8fSHong Zhang 513c9c72f8fSHong Zhang /* U part */ 51462fbd6afSShri Abhyankar nz = bdiag[i]-bdiag[i+1]; 51562fbd6afSShri Abhyankar bjtmp = bj + bdiag[i+1]+1; 516813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 517813a1f2bSShri Abhyankar 518813a1f2bSShri Abhyankar /* load in initial (unfactored row) */ 519813a1f2bSShri Abhyankar nz = ai[r[i]+1] - ai[r[i]]; 520813a1f2bSShri Abhyankar ajtmp = aj + ai[r[i]]; 521813a1f2bSShri Abhyankar v = aa + ai[r[i]]; 522813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 523813a1f2bSShri Abhyankar rtmp[ics[ajtmp[j]]] = v[j]; 524813a1f2bSShri Abhyankar } 525813a1f2bSShri Abhyankar /* ZeropivotApply() */ 52698a93161SHong Zhang rtmp[i] += sctx.shift_amount; /* shift the diagonal of the matrix */ 527813a1f2bSShri Abhyankar 528813a1f2bSShri Abhyankar /* elimination */ 529813a1f2bSShri Abhyankar bjtmp = bj + bi[i]; 530813a1f2bSShri Abhyankar row = *bjtmp++; 531813a1f2bSShri Abhyankar nzL = bi[i+1] - bi[i]; 532f268cba8SShri Abhyankar for(k=0; k < nzL;k++) { 533813a1f2bSShri Abhyankar pc = rtmp + row; 534813a1f2bSShri Abhyankar if (*pc != 0.0) { 535813a1f2bSShri Abhyankar pv = b->a + bdiag[row]; 536813a1f2bSShri Abhyankar multiplier = *pc * (*pv); 537813a1f2bSShri Abhyankar *pc = multiplier; 53862fbd6afSShri Abhyankar pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 53962fbd6afSShri Abhyankar pv = b->a + bdiag[row+1]+1; 54062fbd6afSShri Abhyankar nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 541813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 542813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 543813a1f2bSShri Abhyankar } 544f268cba8SShri Abhyankar row = *bjtmp++; 545813a1f2bSShri Abhyankar } 546813a1f2bSShri Abhyankar 547813a1f2bSShri Abhyankar /* finished row so stick it into b->a */ 548813a1f2bSShri Abhyankar rs = 0.0; 549813a1f2bSShri Abhyankar /* L part */ 550813a1f2bSShri Abhyankar pv = b->a + bi[i] ; 551813a1f2bSShri Abhyankar pj = b->j + bi[i] ; 552813a1f2bSShri Abhyankar nz = bi[i+1] - bi[i]; 553813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 554813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 555813a1f2bSShri Abhyankar } 556813a1f2bSShri Abhyankar 557813a1f2bSShri Abhyankar /* U part */ 55862fbd6afSShri Abhyankar pv = b->a + bdiag[i+1]+1; 55962fbd6afSShri Abhyankar pj = b->j + bdiag[i+1]+1; 56062fbd6afSShri Abhyankar nz = bdiag[i] - bdiag[i+1]-1; 561813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 562813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 563813a1f2bSShri Abhyankar } 564813a1f2bSShri Abhyankar 565813a1f2bSShri Abhyankar sctx.rs = rs; 566813a1f2bSShri Abhyankar sctx.pv = rtmp[i]; 56707b50cabSHong Zhang ierr = MatPivotCheck(info,&sctx,i);CHKERRQ(ierr); 56807b50cabSHong Zhang if(sctx.newshift) break; /* break for-loop */ 56907b50cabSHong Zhang rtmp[i] = sctx.pv; /* sctx.pv might be updated in the case of MAT_SHIFT_INBLOCKS */ 570813a1f2bSShri Abhyankar 571813a1f2bSShri Abhyankar /* Mark diagonal and invert diagonal for simplier triangular solves */ 572813a1f2bSShri Abhyankar pv = b->a + bdiag[i]; 573813a1f2bSShri Abhyankar *pv = 1.0/rtmp[i]; 574813a1f2bSShri Abhyankar 575813a1f2bSShri Abhyankar } /* endof for (i=0; i<n; i++){ */ 576813a1f2bSShri Abhyankar 5778ff23777SHong Zhang /* MatPivotRefine() */ 57807b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){ 579813a1f2bSShri Abhyankar /* 580813a1f2bSShri Abhyankar * if no shift in this attempt & shifting & started shifting & can refine, 581813a1f2bSShri Abhyankar * then try lower shift 582813a1f2bSShri Abhyankar */ 583813a1f2bSShri Abhyankar sctx.shift_hi = sctx.shift_fraction; 584813a1f2bSShri Abhyankar sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 585813a1f2bSShri Abhyankar sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 58607b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 587813a1f2bSShri Abhyankar sctx.nshift++; 588813a1f2bSShri Abhyankar } 58907b50cabSHong Zhang } while (sctx.newshift); 590813a1f2bSShri Abhyankar 591813a1f2bSShri Abhyankar ierr = PetscFree(rtmp);CHKERRQ(ierr); 592813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 593813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 594d3ac4fa3SBarry Smith 595813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 596813a1f2bSShri Abhyankar ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 597813a1f2bSShri Abhyankar if (row_identity && col_identity) { 59835233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 599813a1f2bSShri Abhyankar } else { 60035233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ; 601813a1f2bSShri Abhyankar } 60235233d99SShri Abhyankar C->ops->solveadd = MatSolveAdd_SeqAIJ; 60335233d99SShri Abhyankar C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 60435233d99SShri Abhyankar C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 60535233d99SShri Abhyankar C->ops->matsolve = MatMatSolve_SeqAIJ; 606813a1f2bSShri Abhyankar C->assembled = PETSC_TRUE; 607813a1f2bSShri Abhyankar C->preallocated = PETSC_TRUE; 608813a1f2bSShri Abhyankar ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 609813a1f2bSShri Abhyankar 61014d2772eSHong Zhang /* MatShiftView(A,info,&sctx) */ 611813a1f2bSShri Abhyankar if (sctx.nshift){ 612f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 613813a1f2bSShri 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); 614f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 615813a1f2bSShri Abhyankar ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 616f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){ 617915743fcSHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr); 618813a1f2bSShri Abhyankar } 619813a1f2bSShri Abhyankar } 620d3ac4fa3SBarry Smith ierr = Mat_CheckInode_FactorLU(C,PETSC_FALSE);CHKERRQ(ierr); 621813a1f2bSShri Abhyankar PetscFunctionReturn(0); 622813a1f2bSShri Abhyankar } 623813a1f2bSShri Abhyankar 6244a2ae208SSatish Balay #undef __FUNCT__ 625ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_inplace" 626ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info) 627289bc588SBarry Smith { 628719d5645SBarry Smith Mat C=B; 629416022c9SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 63082bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 6316849ba73SBarry Smith PetscErrorCode ierr; 6325d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 633d278edc7SBarry Smith PetscInt nz,row,i,j,n=A->rmap->n,diag; 634d278edc7SBarry Smith const PetscInt *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 635d278edc7SBarry Smith const PetscInt *ajtmp,*bjtmp,*diag_offset = b->diag,*pj; 636d278edc7SBarry Smith MatScalar *pv,*rtmp,*pc,multiplier,d; 637d278edc7SBarry Smith const MatScalar *v,*aa=a->a; 63833f9893dSHong Zhang PetscReal rs=0.0; 639d90ac83dSHong Zhang FactorShiftCtx sctx; 6404f81c4b7SBarry Smith const PetscInt *ddiag; 641ace3abfcSBarry Smith PetscBool row_identity, col_identity; 642289bc588SBarry Smith 6433a40ed3dSBarry Smith PetscFunctionBegin; 644504a3fadSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 645504a3fadSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 646ada7143aSSatish Balay 647f4db908eSBarry Smith if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 64842898933SHong Zhang ddiag = a->diag; 6499f7d0b68SHong Zhang sctx.shift_top = info->zeropivot; 6506cc28720Svictorle for (i=0; i<n; i++) { 6519f95998fSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 6529f7d0b68SHong Zhang d = (aa)[ddiag[i]]; 6531a890ab1SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 6549f7d0b68SHong Zhang v = aa+ai[i]; 6559f7d0b68SHong Zhang nz = ai[i+1] - ai[i]; 656e24cfe64SHong Zhang for (j=0; j<nz; j++) 6571a890ab1SHong Zhang rs += PetscAbsScalar(v[j]); 6581a890ab1SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 6596cc28720Svictorle } 660118f5789SBarry Smith sctx.shift_top *= 1.1; 661d2276718SHong Zhang sctx.nshift_max = 5; 662d2276718SHong Zhang sctx.shift_lo = 0.; 663d2276718SHong Zhang sctx.shift_hi = 1.; 664a0a356efSHong Zhang } 665a0a356efSHong Zhang 666504a3fadSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 667504a3fadSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 668504a3fadSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 669504a3fadSHong Zhang ics = ic; 670504a3fadSHong Zhang 671085256b3SBarry Smith do { 67207b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 673289bc588SBarry Smith for (i=0; i<n; i++){ 674b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 675b3bf805bSHong Zhang bjtmp = bj + bi[i]; 6769f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 677289bc588SBarry Smith 678289bc588SBarry Smith /* load in initial (unfactored row) */ 6799f7d0b68SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 6809f7d0b68SHong Zhang ajtmp = aj + ai[r[i]]; 6819f7d0b68SHong Zhang v = aa + ai[r[i]]; 682085256b3SBarry Smith for (j=0; j<nz; j++) { 683b3bf805bSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 684085256b3SBarry Smith } 685d2276718SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 686289bc588SBarry Smith 687b3bf805bSHong Zhang row = *bjtmp++; 688f2582111SSatish Balay while (row < i) { 6898c37ef55SBarry Smith pc = rtmp + row; 6908c37ef55SBarry Smith if (*pc != 0.0) { 691010a20caSHong Zhang pv = b->a + diag_offset[row]; 692010a20caSHong Zhang pj = b->j + diag_offset[row] + 1; 69335aab85fSBarry Smith multiplier = *pc / *pv++; 6948c37ef55SBarry Smith *pc = multiplier; 695b3bf805bSHong Zhang nz = bi[row+1] - diag_offset[row] - 1; 6969f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 697dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 698289bc588SBarry Smith } 699b3bf805bSHong Zhang row = *bjtmp++; 700289bc588SBarry Smith } 701416022c9SBarry Smith /* finished row so stick it into b->a */ 702b3bf805bSHong Zhang pv = b->a + bi[i] ; 703b3bf805bSHong Zhang pj = b->j + bi[i] ; 704b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 705b3bf805bSHong Zhang diag = diag_offset[i] - bi[i]; 706e57ff13aSBarry Smith rs = 0.0; 707d3d32019SBarry Smith for (j=0; j<nz; j++) { 7089f7d0b68SHong Zhang pv[j] = rtmp[pj[j]]; 7099f7d0b68SHong Zhang rs += PetscAbsScalar(pv[j]); 710d3d32019SBarry Smith } 711e57ff13aSBarry Smith rs -= PetscAbsScalar(pv[diag]); 712d2276718SHong Zhang 713d2276718SHong Zhang sctx.rs = rs; 714d2276718SHong Zhang sctx.pv = pv[diag]; 71507b50cabSHong Zhang ierr = MatPivotCheck(info,&sctx,i);CHKERRQ(ierr); 71607b50cabSHong Zhang if (sctx.newshift) break; 717504a3fadSHong Zhang pv[diag] = sctx.pv; 71835aab85fSBarry Smith } 719d2276718SHong Zhang 72007b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 7216cc28720Svictorle /* 7226c037d1bSvictorle * if no shift in this attempt & shifting & started shifting & can refine, 7236cc28720Svictorle * then try lower shift 7246cc28720Svictorle */ 7250481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 7260481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 7270481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 72807b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 729d2276718SHong Zhang sctx.nshift++; 7306cc28720Svictorle } 73107b50cabSHong Zhang } while (sctx.newshift); 7320f11f9aeSSatish Balay 73335aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 73435aab85fSBarry Smith for (i=0; i<n; i++) { 735010a20caSHong Zhang b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]]; 73635aab85fSBarry Smith } 737606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 73878b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 73978b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 740d3ac4fa3SBarry Smith 7418d8c7c43SBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 7428d8c7c43SBarry Smith ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 7438d8c7c43SBarry Smith if (row_identity && col_identity) { 744ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_inplace; 7458d8c7c43SBarry Smith } else { 746ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_inplace; 747db4efbfdSBarry Smith } 748ad04f41aSHong Zhang C->ops->solveadd = MatSolveAdd_SeqAIJ_inplace; 749ad04f41aSHong Zhang C->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 750ad04f41aSHong Zhang C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 751ad04f41aSHong Zhang C->ops->matsolve = MatMatSolve_SeqAIJ_inplace; 752c456f294SBarry Smith C->assembled = PETSC_TRUE; 7535c9eb25fSBarry Smith C->preallocated = PETSC_TRUE; 754d0f46423SBarry Smith ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 755d2276718SHong Zhang if (sctx.nshift){ 756f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 7570481f469SBarry 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); 758f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 759e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 7606cc28720Svictorle } 7610697b6caSHong Zhang } 762d3ac4fa3SBarry Smith (C)->ops->solve = MatSolve_SeqAIJ_inplace; 763d3ac4fa3SBarry Smith (C)->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 764d3ac4fa3SBarry Smith ierr = Mat_CheckInode(C,PETSC_FALSE);CHKERRQ(ierr); 7653a40ed3dSBarry Smith PetscFunctionReturn(0); 766289bc588SBarry Smith } 7670889b5dcSvictorle 768137fb511SHong Zhang /* 769137fb511SHong Zhang This routine implements inplace ILU(0) with row or/and column permutations. 770137fb511SHong Zhang Input: 771137fb511SHong Zhang A - original matrix 772137fb511SHong Zhang Output; 773137fb511SHong Zhang A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i] 774137fb511SHong Zhang a->j (col index) is permuted by the inverse of colperm, then sorted 775137fb511SHong Zhang a->a reordered accordingly with a->j 776137fb511SHong Zhang a->diag (ptr to diagonal elements) is updated. 777137fb511SHong Zhang */ 778137fb511SHong Zhang #undef __FUNCT__ 779137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm" 7800481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info) 781137fb511SHong Zhang { 782b051339dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 783b051339dSHong Zhang IS isrow = a->row,isicol = a->icol; 784137fb511SHong Zhang PetscErrorCode ierr; 7855d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 7865d0c19d7SBarry Smith PetscInt i,j,n=A->rmap->n,*ai=a->i,*aj=a->j; 7875d0c19d7SBarry Smith PetscInt *ajtmp,nz,row; 788b051339dSHong Zhang PetscInt *diag = a->diag,nbdiag,*pj; 789a77337e4SBarry Smith PetscScalar *rtmp,*pc,multiplier,d; 790504a3fadSHong Zhang MatScalar *pv,*v; 791137fb511SHong Zhang PetscReal rs; 792d90ac83dSHong Zhang FactorShiftCtx sctx; 793504a3fadSHong Zhang const MatScalar *aa=a->a,*vtmp; 794137fb511SHong Zhang 795137fb511SHong Zhang PetscFunctionBegin; 796e32f2f54SBarry Smith if (A != B) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address"); 797504a3fadSHong Zhang 798504a3fadSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 799504a3fadSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 800504a3fadSHong Zhang 801504a3fadSHong Zhang if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 802504a3fadSHong Zhang const PetscInt *ddiag = a->diag; 803504a3fadSHong Zhang sctx.shift_top = info->zeropivot; 804504a3fadSHong Zhang for (i=0; i<n; i++) { 805504a3fadSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 806504a3fadSHong Zhang d = (aa)[ddiag[i]]; 807504a3fadSHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 808504a3fadSHong Zhang vtmp = aa+ai[i]; 809504a3fadSHong Zhang nz = ai[i+1] - ai[i]; 810504a3fadSHong Zhang for (j=0; j<nz; j++) 811504a3fadSHong Zhang rs += PetscAbsScalar(vtmp[j]); 812504a3fadSHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 813504a3fadSHong Zhang } 814504a3fadSHong Zhang sctx.shift_top *= 1.1; 815504a3fadSHong Zhang sctx.nshift_max = 5; 816504a3fadSHong Zhang sctx.shift_lo = 0.; 817504a3fadSHong Zhang sctx.shift_hi = 1.; 818504a3fadSHong Zhang } 819504a3fadSHong Zhang 820137fb511SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 821137fb511SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 822137fb511SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr); 823137fb511SHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 824b051339dSHong Zhang ics = ic; 825137fb511SHong Zhang 826504a3fadSHong Zhang #if defined(MV) 82775567043SBarry Smith sctx.shift_top = 0.; 828e60cf9a0SBarry Smith sctx.nshift_max = 0; 82975567043SBarry Smith sctx.shift_lo = 0.; 83075567043SBarry Smith sctx.shift_hi = 0.; 83175567043SBarry Smith sctx.shift_fraction = 0.; 832137fb511SHong Zhang 833f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 83475567043SBarry Smith sctx.shift_top = 0.; 835137fb511SHong Zhang for (i=0; i<n; i++) { 836137fb511SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 837b051339dSHong Zhang d = (a->a)[diag[i]]; 838137fb511SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 839b051339dSHong Zhang v = a->a+ai[i]; 840b051339dSHong Zhang nz = ai[i+1] - ai[i]; 841137fb511SHong Zhang for (j=0; j<nz; j++) 842137fb511SHong Zhang rs += PetscAbsScalar(v[j]); 843137fb511SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 844137fb511SHong Zhang } 845137fb511SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 846137fb511SHong Zhang sctx.shift_top *= 1.1; 847137fb511SHong Zhang sctx.nshift_max = 5; 848137fb511SHong Zhang sctx.shift_lo = 0.; 849137fb511SHong Zhang sctx.shift_hi = 1.; 850137fb511SHong Zhang } 851137fb511SHong Zhang 85275567043SBarry Smith sctx.shift_amount = 0.; 853137fb511SHong Zhang sctx.nshift = 0; 854504a3fadSHong Zhang #endif 855504a3fadSHong Zhang 856137fb511SHong Zhang do { 85707b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 858137fb511SHong Zhang for (i=0; i<n; i++){ 859b051339dSHong Zhang /* load in initial unfactored row */ 860b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 861b051339dSHong Zhang ajtmp = aj + ai[r[i]]; 862b051339dSHong Zhang v = a->a + ai[r[i]]; 863137fb511SHong Zhang /* sort permuted ajtmp and values v accordingly */ 864b051339dSHong Zhang for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]]; 865137fb511SHong Zhang ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr); 866137fb511SHong Zhang 867b051339dSHong Zhang diag[r[i]] = ai[r[i]]; 868137fb511SHong Zhang for (j=0; j<nz; j++) { 869137fb511SHong Zhang rtmp[ajtmp[j]] = v[j]; 870b051339dSHong Zhang if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */ 871137fb511SHong Zhang } 872137fb511SHong Zhang rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 873137fb511SHong Zhang 874b051339dSHong Zhang row = *ajtmp++; 875137fb511SHong Zhang while (row < i) { 876137fb511SHong Zhang pc = rtmp + row; 877137fb511SHong Zhang if (*pc != 0.0) { 878b051339dSHong Zhang pv = a->a + diag[r[row]]; 879b051339dSHong Zhang pj = aj + diag[r[row]] + 1; 880137fb511SHong Zhang 881137fb511SHong Zhang multiplier = *pc / *pv++; 882137fb511SHong Zhang *pc = multiplier; 883b051339dSHong Zhang nz = ai[r[row]+1] - diag[r[row]] - 1; 884b051339dSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 885dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 886137fb511SHong Zhang } 887b051339dSHong Zhang row = *ajtmp++; 888137fb511SHong Zhang } 889b051339dSHong Zhang /* finished row so overwrite it onto a->a */ 890b051339dSHong Zhang pv = a->a + ai[r[i]] ; 891b051339dSHong Zhang pj = aj + ai[r[i]] ; 892b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 893b051339dSHong Zhang nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */ 894137fb511SHong Zhang 895137fb511SHong Zhang rs = 0.0; 896137fb511SHong Zhang for (j=0; j<nz; j++) { 897b051339dSHong Zhang pv[j] = rtmp[pj[j]]; 898b051339dSHong Zhang if (j != nbdiag) rs += PetscAbsScalar(pv[j]); 899137fb511SHong Zhang } 900137fb511SHong Zhang 901137fb511SHong Zhang sctx.rs = rs; 902b051339dSHong Zhang sctx.pv = pv[nbdiag]; 90307b50cabSHong Zhang ierr = MatPivotCheck(info,&sctx,i);CHKERRQ(ierr); 90407b50cabSHong Zhang if (sctx.newshift) break; 905504a3fadSHong Zhang pv[nbdiag] = sctx.pv; 906137fb511SHong Zhang } 907137fb511SHong Zhang 90807b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 909137fb511SHong Zhang /* 910137fb511SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 911137fb511SHong Zhang * then try lower shift 912137fb511SHong Zhang */ 9130481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 9140481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 9150481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 91607b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 917137fb511SHong Zhang sctx.nshift++; 918137fb511SHong Zhang } 91907b50cabSHong Zhang } while (sctx.newshift); 920137fb511SHong Zhang 921137fb511SHong Zhang /* invert diagonal entries for simplier triangular solves */ 922137fb511SHong Zhang for (i=0; i<n; i++) { 923b051339dSHong Zhang a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]]; 924137fb511SHong Zhang } 925137fb511SHong Zhang 926137fb511SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 927137fb511SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 928137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 929b051339dSHong Zhang A->ops->solve = MatSolve_SeqAIJ_InplaceWithPerm; 930ad04f41aSHong Zhang A->ops->solveadd = MatSolveAdd_SeqAIJ_inplace; 931ad04f41aSHong Zhang A->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 932ad04f41aSHong Zhang A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 933b051339dSHong Zhang A->assembled = PETSC_TRUE; 9345c9eb25fSBarry Smith A->preallocated = PETSC_TRUE; 935d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr); 936137fb511SHong Zhang if (sctx.nshift){ 937f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 9380481f469SBarry 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); 939f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 940e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 941137fb511SHong Zhang } 942137fb511SHong Zhang } 943137fb511SHong Zhang PetscFunctionReturn(0); 944137fb511SHong Zhang } 945137fb511SHong Zhang 9460a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9474a2ae208SSatish Balay #undef __FUNCT__ 9484a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ" 9490481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info) 950da3a660dSBarry Smith { 951dfbe8321SBarry Smith PetscErrorCode ierr; 952416022c9SBarry Smith Mat C; 953416022c9SBarry Smith 9543a40ed3dSBarry Smith PetscFunctionBegin; 9552692d6eeSBarry Smith ierr = MatGetFactor(A,MATSOLVERPETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr); 956719d5645SBarry Smith ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr); 957719d5645SBarry Smith ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr); 958db4efbfdSBarry Smith A->ops->solve = C->ops->solve; 959db4efbfdSBarry Smith A->ops->solvetranspose = C->ops->solvetranspose; 960eb6b5d47SBarry Smith ierr = MatHeaderMerge(A,C);CHKERRQ(ierr); 96152e6d16bSBarry Smith ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr); 9623a40ed3dSBarry Smith PetscFunctionReturn(0); 963da3a660dSBarry Smith } 9640a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9651d098868SHong Zhang 966889a8dedSBarry Smith 9674a2ae208SSatish Balay #undef __FUNCT__ 968ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_inplace" 969ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx) 9708c37ef55SBarry Smith { 971416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 972416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 9736849ba73SBarry Smith PetscErrorCode ierr; 9745d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 9755d0c19d7SBarry Smith PetscInt nz; 9765d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 977dd6ea824SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 978d9fead3dSBarry Smith const PetscScalar *b; 979dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 9808c37ef55SBarry Smith 9813a40ed3dSBarry Smith PetscFunctionBegin; 9823a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 98391bf9adeSBarry Smith 9843649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 9851ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 986416022c9SBarry Smith tmp = a->solve_work; 9878c37ef55SBarry Smith 9883b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 9893b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 9908c37ef55SBarry Smith 9918c37ef55SBarry Smith /* forward solve the lower triangular */ 9928c37ef55SBarry Smith tmp[0] = b[*r++]; 993010a20caSHong Zhang tmps = tmp; 9948c37ef55SBarry Smith for (i=1; i<n; i++) { 995010a20caSHong Zhang v = aa + ai[i] ; 996010a20caSHong Zhang vi = aj + ai[i] ; 99753e7425aSBarry Smith nz = a->diag[i] - ai[i]; 99853e7425aSBarry Smith sum = b[*r++]; 999003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10008c37ef55SBarry Smith tmp[i] = sum; 10018c37ef55SBarry Smith } 10028c37ef55SBarry Smith 10038c37ef55SBarry Smith /* backward solve the upper triangular */ 10048c37ef55SBarry Smith for (i=n-1; i>=0; i--){ 1005010a20caSHong Zhang v = aa + a->diag[i] + 1; 1006010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1007416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 10088c37ef55SBarry Smith sum = tmp[i]; 1009003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1010010a20caSHong Zhang x[*c--] = tmp[i] = sum*aa[a->diag[i]]; 10118c37ef55SBarry Smith } 10128c37ef55SBarry Smith 10133b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10143b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10153649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 10161ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1017dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 10183a40ed3dSBarry Smith PetscFunctionReturn(0); 10198c37ef55SBarry Smith } 1020026e39d0SSatish Balay 10212bebee5dSHong Zhang #undef __FUNCT__ 1022ad04f41aSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ_inplace" 1023ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X) 10242bebee5dSHong Zhang { 10252bebee5dSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10262bebee5dSHong Zhang IS iscol = a->col,isrow = a->row; 10272bebee5dSHong Zhang PetscErrorCode ierr; 10285d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10295d0c19d7SBarry Smith PetscInt nz,neq; 10305d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 1031dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1032dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1033ace3abfcSBarry Smith PetscBool bisdense,xisdense; 10342bebee5dSHong Zhang 10352bebee5dSHong Zhang PetscFunctionBegin; 10362bebee5dSHong Zhang if (!n) PetscFunctionReturn(0); 10372bebee5dSHong Zhang 1038db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1039e32f2f54SBarry Smith if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 1040db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1041e32f2f54SBarry Smith if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 1042db4efbfdSBarry Smith 10432bebee5dSHong Zhang ierr = MatGetArray(B,&b);CHKERRQ(ierr); 10442bebee5dSHong Zhang ierr = MatGetArray(X,&x);CHKERRQ(ierr); 10452bebee5dSHong Zhang 10462bebee5dSHong Zhang tmp = a->solve_work; 10472bebee5dSHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10482bebee5dSHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 10492bebee5dSHong Zhang 1050a36b22ccSSatish Balay for (neq=0; neq<B->cmap->n; neq++){ 10512bebee5dSHong Zhang /* forward solve the lower triangular */ 10522bebee5dSHong Zhang tmp[0] = b[r[0]]; 10532bebee5dSHong Zhang tmps = tmp; 10542bebee5dSHong Zhang for (i=1; i<n; i++) { 10552bebee5dSHong Zhang v = aa + ai[i] ; 10562bebee5dSHong Zhang vi = aj + ai[i] ; 10572bebee5dSHong Zhang nz = a->diag[i] - ai[i]; 10582bebee5dSHong Zhang sum = b[r[i]]; 1059003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10602bebee5dSHong Zhang tmp[i] = sum; 10612bebee5dSHong Zhang } 10622bebee5dSHong Zhang /* backward solve the upper triangular */ 10632bebee5dSHong Zhang for (i=n-1; i>=0; i--){ 10642bebee5dSHong Zhang v = aa + a->diag[i] + 1; 10652bebee5dSHong Zhang vi = aj + a->diag[i] + 1; 10662bebee5dSHong Zhang nz = ai[i+1] - a->diag[i] - 1; 10672bebee5dSHong Zhang sum = tmp[i]; 1068003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10692bebee5dSHong Zhang x[c[i]] = tmp[i] = sum*aa[a->diag[i]]; 10702bebee5dSHong Zhang } 10712bebee5dSHong Zhang 10722bebee5dSHong Zhang b += n; 10732bebee5dSHong Zhang x += n; 10742bebee5dSHong Zhang } 10752bebee5dSHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10762bebee5dSHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10772bebee5dSHong Zhang ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 10782bebee5dSHong Zhang ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 1079dc0b31edSSatish Balay ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 10802bebee5dSHong Zhang PetscFunctionReturn(0); 10812bebee5dSHong Zhang } 10822bebee5dSHong Zhang 1083137fb511SHong Zhang #undef __FUNCT__ 108435233d99SShri Abhyankar #define __FUNCT__ "MatMatSolve_SeqAIJ" 108535233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X) 10869bd0847aSShri Abhyankar { 10879bd0847aSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10889bd0847aSShri Abhyankar IS iscol = a->col,isrow = a->row; 10899bd0847aSShri Abhyankar PetscErrorCode ierr; 10909bd0847aSShri Abhyankar PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 10919bd0847aSShri Abhyankar PetscInt nz,neq; 10929bd0847aSShri Abhyankar const PetscInt *rout,*cout,*r,*c; 10939bd0847aSShri Abhyankar PetscScalar *x,*b,*tmp,sum; 10949bd0847aSShri Abhyankar const MatScalar *aa = a->a,*v; 1095ace3abfcSBarry Smith PetscBool bisdense,xisdense; 10969bd0847aSShri Abhyankar 10979bd0847aSShri Abhyankar PetscFunctionBegin; 10989bd0847aSShri Abhyankar if (!n) PetscFunctionReturn(0); 10999bd0847aSShri Abhyankar 11009bd0847aSShri Abhyankar ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1101e32f2f54SBarry Smith if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 11029bd0847aSShri Abhyankar ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1103e32f2f54SBarry Smith if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 11049bd0847aSShri Abhyankar 11059bd0847aSShri Abhyankar ierr = MatGetArray(B,&b);CHKERRQ(ierr); 11069bd0847aSShri Abhyankar ierr = MatGetArray(X,&x);CHKERRQ(ierr); 11079bd0847aSShri Abhyankar 11089bd0847aSShri Abhyankar tmp = a->solve_work; 11099bd0847aSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 11109bd0847aSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 11119bd0847aSShri Abhyankar 11129bd0847aSShri Abhyankar for (neq=0; neq<B->cmap->n; neq++){ 11139bd0847aSShri Abhyankar /* forward solve the lower triangular */ 11149bd0847aSShri Abhyankar tmp[0] = b[r[0]]; 11159bd0847aSShri Abhyankar v = aa; 11169bd0847aSShri Abhyankar vi = aj; 11179bd0847aSShri Abhyankar for (i=1; i<n; i++) { 11189bd0847aSShri Abhyankar nz = ai[i+1] - ai[i]; 11199bd0847aSShri Abhyankar sum = b[r[i]]; 11209bd0847aSShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 11219bd0847aSShri Abhyankar tmp[i] = sum; 11229bd0847aSShri Abhyankar v += nz; vi += nz; 11239bd0847aSShri Abhyankar } 11249bd0847aSShri Abhyankar 11259bd0847aSShri Abhyankar /* backward solve the upper triangular */ 11269bd0847aSShri Abhyankar for (i=n-1; i>=0; i--){ 11279bd0847aSShri Abhyankar v = aa + adiag[i+1]+1; 11289bd0847aSShri Abhyankar vi = aj + adiag[i+1]+1; 11299bd0847aSShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 11309bd0847aSShri Abhyankar sum = tmp[i]; 11319bd0847aSShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 11329bd0847aSShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 11339bd0847aSShri Abhyankar } 11349bd0847aSShri Abhyankar 11359bd0847aSShri Abhyankar b += n; 11369bd0847aSShri Abhyankar x += n; 11379bd0847aSShri Abhyankar } 11389bd0847aSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 11399bd0847aSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 11409bd0847aSShri Abhyankar ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 11419bd0847aSShri Abhyankar ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 11429bd0847aSShri Abhyankar ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 11439bd0847aSShri Abhyankar PetscFunctionReturn(0); 11449bd0847aSShri Abhyankar } 11459bd0847aSShri Abhyankar 11469bd0847aSShri Abhyankar #undef __FUNCT__ 1147137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm" 1148137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx) 1149137fb511SHong Zhang { 1150137fb511SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1151137fb511SHong Zhang IS iscol = a->col,isrow = a->row; 1152137fb511SHong Zhang PetscErrorCode ierr; 11535d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 11545d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 11555d0c19d7SBarry Smith PetscInt nz,row; 1156dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1157dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1158137fb511SHong Zhang 1159137fb511SHong Zhang PetscFunctionBegin; 1160137fb511SHong Zhang if (!n) PetscFunctionReturn(0); 1161137fb511SHong Zhang 1162137fb511SHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 1163137fb511SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1164137fb511SHong Zhang tmp = a->solve_work; 1165137fb511SHong Zhang 1166137fb511SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1167137fb511SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1168137fb511SHong Zhang 1169137fb511SHong Zhang /* forward solve the lower triangular */ 1170137fb511SHong Zhang tmp[0] = b[*r++]; 1171137fb511SHong Zhang tmps = tmp; 1172137fb511SHong Zhang for (row=1; row<n; row++) { 1173137fb511SHong Zhang i = rout[row]; /* permuted row */ 1174137fb511SHong Zhang v = aa + ai[i] ; 1175137fb511SHong Zhang vi = aj + ai[i] ; 1176137fb511SHong Zhang nz = a->diag[i] - ai[i]; 1177137fb511SHong Zhang sum = b[*r++]; 1178003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1179137fb511SHong Zhang tmp[row] = sum; 1180137fb511SHong Zhang } 1181137fb511SHong Zhang 1182137fb511SHong Zhang /* backward solve the upper triangular */ 1183137fb511SHong Zhang for (row=n-1; row>=0; row--){ 1184137fb511SHong Zhang i = rout[row]; /* permuted row */ 1185137fb511SHong Zhang v = aa + a->diag[i] + 1; 1186137fb511SHong Zhang vi = aj + a->diag[i] + 1; 1187137fb511SHong Zhang nz = ai[i+1] - a->diag[i] - 1; 1188137fb511SHong Zhang sum = tmp[row]; 1189003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1190137fb511SHong Zhang x[*c--] = tmp[row] = sum*aa[a->diag[i]]; 1191137fb511SHong Zhang } 1192137fb511SHong Zhang 1193137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1194137fb511SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1195137fb511SHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 1196137fb511SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1197dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1198137fb511SHong Zhang PetscFunctionReturn(0); 1199137fb511SHong Zhang } 1200137fb511SHong Zhang 1201930ae53cSBarry Smith /* ----------------------------------------------------------- */ 1202a4005a5dSBarry Smith #include "../src/mat/impls/aij/seq/ftn-kernels/fsolve.h" 12034a2ae208SSatish Balay #undef __FUNCT__ 1204ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_inplace" 1205ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx) 1206930ae53cSBarry Smith { 1207930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 12086849ba73SBarry Smith PetscErrorCode ierr; 1209d0f46423SBarry Smith PetscInt n = A->rmap->n; 1210b377110cSBarry Smith const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag; 1211356650c2SBarry Smith PetscScalar *x; 1212356650c2SBarry Smith const PetscScalar *b; 1213dd6ea824SBarry Smith const MatScalar *aa = a->a; 1214aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 1215356650c2SBarry Smith PetscInt adiag_i,i,nz,ai_i; 1216b377110cSBarry Smith const PetscInt *vi; 1217dd6ea824SBarry Smith const MatScalar *v; 1218dd6ea824SBarry Smith PetscScalar sum; 1219d85afc42SSatish Balay #endif 1220930ae53cSBarry Smith 12213a40ed3dSBarry Smith PetscFunctionBegin; 12223a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 1223930ae53cSBarry Smith 12243649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 12251ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1226930ae53cSBarry Smith 1227aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 12281c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 12291c4feecaSSatish Balay #else 1230930ae53cSBarry Smith /* forward solve the lower triangular */ 1231930ae53cSBarry Smith x[0] = b[0]; 1232930ae53cSBarry Smith for (i=1; i<n; i++) { 1233930ae53cSBarry Smith ai_i = ai[i]; 1234930ae53cSBarry Smith v = aa + ai_i; 1235930ae53cSBarry Smith vi = aj + ai_i; 1236930ae53cSBarry Smith nz = adiag[i] - ai_i; 1237930ae53cSBarry Smith sum = b[i]; 1238003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1239930ae53cSBarry Smith x[i] = sum; 1240930ae53cSBarry Smith } 1241930ae53cSBarry Smith 1242930ae53cSBarry Smith /* backward solve the upper triangular */ 1243930ae53cSBarry Smith for (i=n-1; i>=0; i--){ 1244930ae53cSBarry Smith adiag_i = adiag[i]; 1245d708749eSBarry Smith v = aa + adiag_i + 1; 1246d708749eSBarry Smith vi = aj + adiag_i + 1; 1247930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 1248930ae53cSBarry Smith sum = x[i]; 1249003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1250930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 1251930ae53cSBarry Smith } 12521c4feecaSSatish Balay #endif 1253dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 12543649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 12551ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 12563a40ed3dSBarry Smith PetscFunctionReturn(0); 1257930ae53cSBarry Smith } 1258930ae53cSBarry Smith 12594a2ae208SSatish Balay #undef __FUNCT__ 1260ad04f41aSHong Zhang #define __FUNCT__ "MatSolveAdd_SeqAIJ_inplace" 1261ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx) 1262da3a660dSBarry Smith { 1263416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1264416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 12656849ba73SBarry Smith PetscErrorCode ierr; 126604fbf559SBarry Smith PetscInt i, n = A->rmap->n,j; 12675d0c19d7SBarry Smith PetscInt nz; 126804fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j; 126904fbf559SBarry Smith PetscScalar *x,*tmp,sum; 127004fbf559SBarry Smith const PetscScalar *b; 1271dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1272da3a660dSBarry Smith 12733a40ed3dSBarry Smith PetscFunctionBegin; 127478b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 1275da3a660dSBarry Smith 12763649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 12771ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1278416022c9SBarry Smith tmp = a->solve_work; 1279da3a660dSBarry Smith 12803b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12813b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1282da3a660dSBarry Smith 1283da3a660dSBarry Smith /* forward solve the lower triangular */ 1284da3a660dSBarry Smith tmp[0] = b[*r++]; 1285da3a660dSBarry Smith for (i=1; i<n; i++) { 1286010a20caSHong Zhang v = aa + ai[i] ; 1287010a20caSHong Zhang vi = aj + ai[i] ; 1288416022c9SBarry Smith nz = a->diag[i] - ai[i]; 1289da3a660dSBarry Smith sum = b[*r++]; 129004fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1291da3a660dSBarry Smith tmp[i] = sum; 1292da3a660dSBarry Smith } 1293da3a660dSBarry Smith 1294da3a660dSBarry Smith /* backward solve the upper triangular */ 1295da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1296010a20caSHong Zhang v = aa + a->diag[i] + 1; 1297010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1298416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 1299da3a660dSBarry Smith sum = tmp[i]; 130004fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1301010a20caSHong Zhang tmp[i] = sum*aa[a->diag[i]]; 1302da3a660dSBarry Smith x[*c--] += tmp[i]; 1303da3a660dSBarry Smith } 1304da3a660dSBarry Smith 13053b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13063b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 13073649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 13081ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1309dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1310898183ecSLois Curfman McInnes 13113a40ed3dSBarry Smith PetscFunctionReturn(0); 1312da3a660dSBarry Smith } 131304fbf559SBarry Smith 13144a2ae208SSatish Balay #undef __FUNCT__ 131535233d99SShri Abhyankar #define __FUNCT__ "MatSolveAdd_SeqAIJ" 131635233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 13173c0119dfSShri Abhyankar { 13183c0119dfSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13193c0119dfSShri Abhyankar IS iscol = a->col,isrow = a->row; 13203c0119dfSShri Abhyankar PetscErrorCode ierr; 13213c0119dfSShri Abhyankar PetscInt i, n = A->rmap->n,j; 13223c0119dfSShri Abhyankar PetscInt nz; 13233c0119dfSShri Abhyankar const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 13243c0119dfSShri Abhyankar PetscScalar *x,*tmp,sum; 13253c0119dfSShri Abhyankar const PetscScalar *b; 13263c0119dfSShri Abhyankar const MatScalar *aa = a->a,*v; 13273c0119dfSShri Abhyankar 13283c0119dfSShri Abhyankar PetscFunctionBegin; 13293c0119dfSShri Abhyankar if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 13303c0119dfSShri Abhyankar 13313649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 13323c0119dfSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 13333c0119dfSShri Abhyankar tmp = a->solve_work; 13343c0119dfSShri Abhyankar 13353c0119dfSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13363c0119dfSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 13373c0119dfSShri Abhyankar 13383c0119dfSShri Abhyankar /* forward solve the lower triangular */ 13393c0119dfSShri Abhyankar tmp[0] = b[r[0]]; 13403c0119dfSShri Abhyankar v = aa; 13413c0119dfSShri Abhyankar vi = aj; 13423c0119dfSShri Abhyankar for (i=1; i<n; i++) { 13433c0119dfSShri Abhyankar nz = ai[i+1] - ai[i]; 13443c0119dfSShri Abhyankar sum = b[r[i]]; 13453c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 13463c0119dfSShri Abhyankar tmp[i] = sum; 13473c0119dfSShri Abhyankar v += nz; vi += nz; 13483c0119dfSShri Abhyankar } 13493c0119dfSShri Abhyankar 13503c0119dfSShri Abhyankar /* backward solve the upper triangular */ 13513c0119dfSShri Abhyankar v = aa + adiag[n-1]; 13523c0119dfSShri Abhyankar vi = aj + adiag[n-1]; 13533c0119dfSShri Abhyankar for (i=n-1; i>=0; i--){ 13543c0119dfSShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 13553c0119dfSShri Abhyankar sum = tmp[i]; 13563c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 13573c0119dfSShri Abhyankar tmp[i] = sum*v[nz]; 13583c0119dfSShri Abhyankar x[c[i]] += tmp[i]; 13593c0119dfSShri Abhyankar v += nz+1; vi += nz+1; 13603c0119dfSShri Abhyankar } 13613c0119dfSShri Abhyankar 13623c0119dfSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13633c0119dfSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 13643649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 13653c0119dfSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 13663c0119dfSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 13673c0119dfSShri Abhyankar 13683c0119dfSShri Abhyankar PetscFunctionReturn(0); 13693c0119dfSShri Abhyankar } 13703c0119dfSShri Abhyankar 13713c0119dfSShri Abhyankar #undef __FUNCT__ 1372ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTranspose_SeqAIJ_inplace" 1373ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx) 1374da3a660dSBarry Smith { 1375416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13762235e667SBarry Smith IS iscol = a->col,isrow = a->row; 13776849ba73SBarry Smith PetscErrorCode ierr; 137804fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 137904fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 138004fbf559SBarry Smith PetscInt nz; 138104fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1382dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 138304fbf559SBarry Smith const PetscScalar *b; 1384da3a660dSBarry Smith 13853a40ed3dSBarry Smith PetscFunctionBegin; 13863649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 13871ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1388416022c9SBarry Smith tmp = a->solve_work; 1389da3a660dSBarry Smith 13902235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13912235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1392da3a660dSBarry Smith 1393da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 13942235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1395da3a660dSBarry Smith 1396da3a660dSBarry Smith /* forward solve the U^T */ 1397da3a660dSBarry Smith for (i=0; i<n; i++) { 1398010a20caSHong Zhang v = aa + diag[i] ; 1399010a20caSHong Zhang vi = aj + diag[i] + 1; 1400f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 1401c38d4ed2SBarry Smith s1 = tmp[i]; 1402ef66eb69SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 140304fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1404c38d4ed2SBarry Smith tmp[i] = s1; 1405da3a660dSBarry Smith } 1406da3a660dSBarry Smith 1407da3a660dSBarry Smith /* backward solve the L^T */ 1408da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1409010a20caSHong Zhang v = aa + diag[i] - 1 ; 1410010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1411f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 1412f1af5d2fSBarry Smith s1 = tmp[i]; 141304fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 1414da3a660dSBarry Smith } 1415da3a660dSBarry Smith 1416da3a660dSBarry Smith /* copy tmp into x according to permutation */ 1417591294e4SBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1418da3a660dSBarry Smith 14192235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 14202235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 14213649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 14221ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1423da3a660dSBarry Smith 1424dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 14253a40ed3dSBarry Smith PetscFunctionReturn(0); 1426da3a660dSBarry Smith } 1427da3a660dSBarry Smith 1428d1fa9404SShri Abhyankar #undef __FUNCT__ 142935233d99SShri Abhyankar #define __FUNCT__ "MatSolveTranspose_SeqAIJ" 143035233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 1431d1fa9404SShri Abhyankar { 1432d1fa9404SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1433d1fa9404SShri Abhyankar IS iscol = a->col,isrow = a->row; 1434d1fa9404SShri Abhyankar PetscErrorCode ierr; 1435d1fa9404SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1436d1fa9404SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1437d1fa9404SShri Abhyankar PetscInt nz; 1438d1fa9404SShri Abhyankar PetscScalar *x,*tmp,s1; 1439d1fa9404SShri Abhyankar const MatScalar *aa = a->a,*v; 1440d1fa9404SShri Abhyankar const PetscScalar *b; 1441d1fa9404SShri Abhyankar 1442d1fa9404SShri Abhyankar PetscFunctionBegin; 14433649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1444d1fa9404SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1445d1fa9404SShri Abhyankar tmp = a->solve_work; 1446d1fa9404SShri Abhyankar 1447d1fa9404SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1448d1fa9404SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1449d1fa9404SShri Abhyankar 1450d1fa9404SShri Abhyankar /* copy the b into temp work space according to permutation */ 1451d1fa9404SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1452d1fa9404SShri Abhyankar 1453d1fa9404SShri Abhyankar /* forward solve the U^T */ 1454d1fa9404SShri Abhyankar for (i=0; i<n; i++) { 1455d1fa9404SShri Abhyankar v = aa + adiag[i+1] + 1; 1456d1fa9404SShri Abhyankar vi = aj + adiag[i+1] + 1; 1457d1fa9404SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1458d1fa9404SShri Abhyankar s1 = tmp[i]; 1459d1fa9404SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1460d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1461d1fa9404SShri Abhyankar tmp[i] = s1; 1462d1fa9404SShri Abhyankar } 1463d1fa9404SShri Abhyankar 1464d1fa9404SShri Abhyankar /* backward solve the L^T */ 1465d1fa9404SShri Abhyankar for (i=n-1; i>=0; i--){ 1466d1fa9404SShri Abhyankar v = aa + ai[i]; 1467d1fa9404SShri Abhyankar vi = aj + ai[i]; 1468d1fa9404SShri Abhyankar nz = ai[i+1] - ai[i]; 1469d1fa9404SShri Abhyankar s1 = tmp[i]; 1470d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1471d1fa9404SShri Abhyankar } 1472d1fa9404SShri Abhyankar 1473d1fa9404SShri Abhyankar /* copy tmp into x according to permutation */ 1474d1fa9404SShri Abhyankar for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1475d1fa9404SShri Abhyankar 1476d1fa9404SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1477d1fa9404SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 14783649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1479d1fa9404SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1480d1fa9404SShri Abhyankar 1481d1fa9404SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1482d1fa9404SShri Abhyankar PetscFunctionReturn(0); 1483d1fa9404SShri Abhyankar } 1484d1fa9404SShri Abhyankar 14854a2ae208SSatish Balay #undef __FUNCT__ 1486ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ_inplace" 1487ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx) 1488da3a660dSBarry Smith { 1489416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 14902235e667SBarry Smith IS iscol = a->col,isrow = a->row; 14916849ba73SBarry Smith PetscErrorCode ierr; 149204fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 149304fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 149404fbf559SBarry Smith PetscInt nz; 149504fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1496dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 149704fbf559SBarry Smith const PetscScalar *b; 14986abc6512SBarry Smith 14993a40ed3dSBarry Smith PetscFunctionBegin; 150023bc6035SMatthew Knepley if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 15013649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 15021ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1503416022c9SBarry Smith tmp = a->solve_work; 15046abc6512SBarry Smith 15052235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 15062235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 15076abc6512SBarry Smith 15086abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 15092235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 15106abc6512SBarry Smith 15116abc6512SBarry Smith /* forward solve the U^T */ 15126abc6512SBarry Smith for (i=0; i<n; i++) { 1513010a20caSHong Zhang v = aa + diag[i] ; 1514010a20caSHong Zhang vi = aj + diag[i] + 1; 1515f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 151604fbf559SBarry Smith s1 = tmp[i]; 151704fbf559SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 151804fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 151904fbf559SBarry Smith tmp[i] = s1; 15206abc6512SBarry Smith } 15216abc6512SBarry Smith 15226abc6512SBarry Smith /* backward solve the L^T */ 15236abc6512SBarry Smith for (i=n-1; i>=0; i--){ 1524010a20caSHong Zhang v = aa + diag[i] - 1 ; 1525010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1526f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 152704fbf559SBarry Smith s1 = tmp[i]; 152804fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 15296abc6512SBarry Smith } 15306abc6512SBarry Smith 15316abc6512SBarry Smith /* copy tmp into x according to permutation */ 15326abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 15336abc6512SBarry Smith 15342235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 15352235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 15363649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 15371ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 15386abc6512SBarry Smith 153904fbf559SBarry Smith ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 15403a40ed3dSBarry Smith PetscFunctionReturn(0); 1541da3a660dSBarry Smith } 154204fbf559SBarry Smith 1543533e6011SShri Abhyankar #undef __FUNCT__ 154435233d99SShri Abhyankar #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ" 154535233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 1546533e6011SShri Abhyankar { 1547533e6011SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1548533e6011SShri Abhyankar IS iscol = a->col,isrow = a->row; 1549533e6011SShri Abhyankar PetscErrorCode ierr; 1550533e6011SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1551533e6011SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1552533e6011SShri Abhyankar PetscInt nz; 1553533e6011SShri Abhyankar PetscScalar *x,*tmp,s1; 1554533e6011SShri Abhyankar const MatScalar *aa = a->a,*v; 1555533e6011SShri Abhyankar const PetscScalar *b; 1556533e6011SShri Abhyankar 1557533e6011SShri Abhyankar PetscFunctionBegin; 1558533e6011SShri Abhyankar if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 15593649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1560533e6011SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1561533e6011SShri Abhyankar tmp = a->solve_work; 1562533e6011SShri Abhyankar 1563533e6011SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1564533e6011SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1565533e6011SShri Abhyankar 1566533e6011SShri Abhyankar /* copy the b into temp work space according to permutation */ 1567533e6011SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1568533e6011SShri Abhyankar 1569533e6011SShri Abhyankar /* forward solve the U^T */ 1570533e6011SShri Abhyankar for (i=0; i<n; i++) { 1571533e6011SShri Abhyankar v = aa + adiag[i+1] + 1; 1572533e6011SShri Abhyankar vi = aj + adiag[i+1] + 1; 1573533e6011SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1574533e6011SShri Abhyankar s1 = tmp[i]; 1575533e6011SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1576533e6011SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1577533e6011SShri Abhyankar tmp[i] = s1; 1578533e6011SShri Abhyankar } 1579533e6011SShri Abhyankar 1580533e6011SShri Abhyankar 1581533e6011SShri Abhyankar /* backward solve the L^T */ 1582533e6011SShri Abhyankar for (i=n-1; i>=0; i--){ 1583533e6011SShri Abhyankar v = aa + ai[i] ; 1584533e6011SShri Abhyankar vi = aj + ai[i]; 1585533e6011SShri Abhyankar nz = ai[i+1] - ai[i]; 1586533e6011SShri Abhyankar s1 = tmp[i]; 1587c5e3b2a3SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1588533e6011SShri Abhyankar } 1589533e6011SShri Abhyankar 1590533e6011SShri Abhyankar /* copy tmp into x according to permutation */ 1591533e6011SShri Abhyankar for (i=0; i<n; i++) x[r[i]] += tmp[i]; 1592533e6011SShri Abhyankar 1593533e6011SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1594533e6011SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 15953649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1596533e6011SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1597533e6011SShri Abhyankar 1598533e6011SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1599533e6011SShri Abhyankar PetscFunctionReturn(0); 1600533e6011SShri Abhyankar } 1601533e6011SShri Abhyankar 16029e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 1603b47f5a65SHong Zhang 1604ace3abfcSBarry Smith EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscBool ); 160508480c60SBarry Smith 16068fc3a347SHong Zhang /* 16078a76ed4fSHong Zhang ilu() under revised new data structure. 1608813a1f2bSShri Abhyankar Factored arrays bj and ba are stored as 1609813a1f2bSShri Abhyankar L(0,:), L(1,:), ...,L(n-1,:), U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:) 1610813a1f2bSShri Abhyankar 1611813a1f2bSShri Abhyankar bi=fact->i is an array of size n+1, in which 1612813a1f2bSShri Abhyankar bi+ 1613baabb860SHong Zhang bi[i]: points to 1st entry of L(i,:),i=0,...,n-1 1614baabb860SHong Zhang bi[n]: points to L(n-1,n-1)+1 1615baabb860SHong Zhang 1616813a1f2bSShri Abhyankar bdiag=fact->diag is an array of size n+1,in which 1617baabb860SHong Zhang bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1 1618a24f213cSHong Zhang bdiag[n]: points to entry of U(n-1,0)-1 1619813a1f2bSShri Abhyankar 1620813a1f2bSShri Abhyankar U(i,:) contains bdiag[i] as its last entry, i.e., 1621813a1f2bSShri Abhyankar U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 1622813a1f2bSShri Abhyankar */ 1623813a1f2bSShri Abhyankar #undef __FUNCT__ 162435233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0" 162535233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1626813a1f2bSShri Abhyankar { 1627813a1f2bSShri Abhyankar 1628813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1629813a1f2bSShri Abhyankar PetscErrorCode ierr; 1630bbd65245SShri Abhyankar const PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag; 1631bbd65245SShri Abhyankar PetscInt i,j,k=0,nz,*bi,*bj,*bdiag; 1632ace3abfcSBarry Smith PetscBool missing; 16331df811f5SHong Zhang IS isicol; 1634813a1f2bSShri Abhyankar 1635813a1f2bSShri Abhyankar PetscFunctionBegin; 1636e32f2f54SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 16371df811f5SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 1638e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 16391df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 16401df811f5SHong Zhang 1641813a1f2bSShri Abhyankar ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr); 1642813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1643813a1f2bSShri Abhyankar 1644813a1f2bSShri Abhyankar /* allocate matrix arrays for new data structure */ 1645813a1f2bSShri Abhyankar ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,n+1,PetscInt,&b->i);CHKERRQ(ierr); 1646813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1647813a1f2bSShri Abhyankar b->singlemalloc = PETSC_TRUE; 1648813a1f2bSShri Abhyankar if (!b->diag){ 1649813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr); 1650813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1651813a1f2bSShri Abhyankar } 1652813a1f2bSShri Abhyankar bdiag = b->diag; 1653813a1f2bSShri Abhyankar 1654813a1f2bSShri Abhyankar if (n > 0) { 1655813a1f2bSShri Abhyankar ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr); 1656813a1f2bSShri Abhyankar } 1657813a1f2bSShri Abhyankar 1658813a1f2bSShri Abhyankar /* set bi and bj with new data structure */ 1659813a1f2bSShri Abhyankar bi = b->i; 1660813a1f2bSShri Abhyankar bj = b->j; 1661813a1f2bSShri Abhyankar 1662813a1f2bSShri Abhyankar /* L part */ 1663e60cf9a0SBarry Smith bi[0] = 0; 1664813a1f2bSShri Abhyankar for (i=0; i<n; i++){ 1665813a1f2bSShri Abhyankar nz = adiag[i] - ai[i]; 1666813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nz; 1667813a1f2bSShri Abhyankar aj = a->j + ai[i]; 1668813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 1669bbd65245SShri Abhyankar /* *bj = aj[j]; bj++; */ 1670bbd65245SShri Abhyankar bj[k++] = aj[j]; 1671813a1f2bSShri Abhyankar } 1672813a1f2bSShri Abhyankar } 1673813a1f2bSShri Abhyankar 1674813a1f2bSShri Abhyankar /* U part */ 167562fbd6afSShri Abhyankar bdiag[n] = bi[n]-1; 1676813a1f2bSShri Abhyankar for (i=n-1; i>=0; i--){ 1677813a1f2bSShri Abhyankar nz = ai[i+1] - adiag[i] - 1; 1678813a1f2bSShri Abhyankar aj = a->j + adiag[i] + 1; 1679813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 1680bbd65245SShri Abhyankar /* *bj = aj[j]; bj++; */ 1681bbd65245SShri Abhyankar bj[k++] = aj[j]; 1682813a1f2bSShri Abhyankar } 1683813a1f2bSShri Abhyankar /* diag[i] */ 1684bbd65245SShri Abhyankar /* *bj = i; bj++; */ 1685bbd65245SShri Abhyankar bj[k++] = i; 1686a24f213cSHong Zhang bdiag[i] = bdiag[i+1] + nz + 1; 1687813a1f2bSShri Abhyankar } 16881df811f5SHong Zhang 1689d5f3da31SBarry Smith fact->factortype = MAT_FACTOR_ILU; 16901df811f5SHong Zhang fact->info.factor_mallocs = 0; 16911df811f5SHong Zhang fact->info.fill_ratio_given = info->fill; 16921df811f5SHong Zhang fact->info.fill_ratio_needed = 1.0; 169335233d99SShri Abhyankar fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 16941df811f5SHong Zhang 16951df811f5SHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 16961df811f5SHong Zhang b->row = isrow; 16971df811f5SHong Zhang b->col = iscol; 16981df811f5SHong Zhang b->icol = isicol; 16991df811f5SHong Zhang ierr = PetscMalloc((fact->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 17001df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 17011df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1702813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1703813a1f2bSShri Abhyankar } 1704813a1f2bSShri Abhyankar 1705813a1f2bSShri Abhyankar #undef __FUNCT__ 170635233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ" 170735233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1708813a1f2bSShri Abhyankar { 1709813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1710813a1f2bSShri Abhyankar IS isicol; 1711813a1f2bSShri Abhyankar PetscErrorCode ierr; 1712813a1f2bSShri Abhyankar const PetscInt *r,*ic; 17131df811f5SHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j; 1714813a1f2bSShri Abhyankar PetscInt *bi,*cols,nnz,*cols_lvl; 1715813a1f2bSShri Abhyankar PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 1716813a1f2bSShri Abhyankar PetscInt i,levels,diagonal_fill; 1717ace3abfcSBarry Smith PetscBool col_identity,row_identity; 1718813a1f2bSShri Abhyankar PetscReal f; 1719813a1f2bSShri Abhyankar PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 1720813a1f2bSShri Abhyankar PetscBT lnkbt; 1721813a1f2bSShri Abhyankar PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 1722813a1f2bSShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1723813a1f2bSShri Abhyankar PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 1724813a1f2bSShri Abhyankar 1725813a1f2bSShri Abhyankar PetscFunctionBegin; 17263898ab1dSSatish Balay /* Uncomment the old data struct part only while testing new data structure for MatSolve() */ 1727124a3351SHong Zhang /* 1728ace3abfcSBarry Smith PetscBool olddatastruct=PETSC_FALSE; 1729*acfcf0e5SJed Brown ierr = PetscOptionsGetBool(PETSC_NULL,"-ilu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr); 173035233d99SShri Abhyankar if(olddatastruct){ 1731ad04f41aSHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_inplace(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1732ad04f41aSHong Zhang PetscFunctionReturn(0); 1733ad04f41aSHong Zhang } 1734124a3351SHong Zhang */ 1735ad04f41aSHong Zhang 1736813a1f2bSShri Abhyankar levels = (PetscInt)info->levels; 1737813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 1738813a1f2bSShri Abhyankar ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 1739813a1f2bSShri Abhyankar 1740813a1f2bSShri Abhyankar if (!levels && row_identity && col_identity) { 1741813a1f2bSShri Abhyankar /* special case: ilu(0) with natural ordering */ 174235233d99SShri Abhyankar ierr = MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1743124a3351SHong Zhang if (a->inode.size) { 1744124a3351SHong Zhang fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 1745124a3351SHong Zhang } 1746813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1747813a1f2bSShri Abhyankar } 1748813a1f2bSShri Abhyankar 1749e32f2f54SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 17501df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 1751813a1f2bSShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1752813a1f2bSShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 1753813a1f2bSShri Abhyankar 17541bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 17551bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 17561bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1757e60cf9a0SBarry Smith bi[0] = bdiag[0] = 0; 1758813a1f2bSShri Abhyankar 17590e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr); 1760813a1f2bSShri Abhyankar 1761813a1f2bSShri Abhyankar /* create a linked list for storing column indices of the active row */ 1762813a1f2bSShri Abhyankar nlnk = n + 1; 1763813a1f2bSShri Abhyankar ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1764813a1f2bSShri Abhyankar 1765813a1f2bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 17661df811f5SHong Zhang f = info->fill; 17671df811f5SHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 1768813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1769813a1f2bSShri Abhyankar current_space = free_space; 1770813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 1771813a1f2bSShri Abhyankar current_space_lvl = free_space_lvl; 1772813a1f2bSShri Abhyankar 1773813a1f2bSShri Abhyankar for (i=0; i<n; i++) { 1774813a1f2bSShri Abhyankar nzi = 0; 1775813a1f2bSShri Abhyankar /* copy current row into linked list */ 1776813a1f2bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 1777e32f2f54SBarry Smith if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 1778813a1f2bSShri Abhyankar cols = aj + ai[r[i]]; 1779813a1f2bSShri Abhyankar lnk[i] = -1; /* marker to indicate if diagonal exists */ 1780813a1f2bSShri Abhyankar ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1781813a1f2bSShri Abhyankar nzi += nlnk; 1782813a1f2bSShri Abhyankar 1783813a1f2bSShri Abhyankar /* make sure diagonal entry is included */ 1784813a1f2bSShri Abhyankar if (diagonal_fill && lnk[i] == -1) { 1785813a1f2bSShri Abhyankar fm = n; 1786813a1f2bSShri Abhyankar while (lnk[fm] < i) fm = lnk[fm]; 1787813a1f2bSShri Abhyankar lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 1788813a1f2bSShri Abhyankar lnk[fm] = i; 1789813a1f2bSShri Abhyankar lnk_lvl[i] = 0; 1790813a1f2bSShri Abhyankar nzi++; dcount++; 1791813a1f2bSShri Abhyankar } 1792813a1f2bSShri Abhyankar 1793813a1f2bSShri Abhyankar /* add pivot rows into the active row */ 1794813a1f2bSShri Abhyankar nzbd = 0; 1795813a1f2bSShri Abhyankar prow = lnk[n]; 1796813a1f2bSShri Abhyankar while (prow < i) { 1797813a1f2bSShri Abhyankar nnz = bdiag[prow]; 1798813a1f2bSShri Abhyankar cols = bj_ptr[prow] + nnz + 1; 1799813a1f2bSShri Abhyankar cols_lvl = bjlvl_ptr[prow] + nnz + 1; 1800813a1f2bSShri Abhyankar nnz = bi[prow+1] - bi[prow] - nnz - 1; 1801813a1f2bSShri Abhyankar ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 1802813a1f2bSShri Abhyankar nzi += nlnk; 1803813a1f2bSShri Abhyankar prow = lnk[prow]; 1804813a1f2bSShri Abhyankar nzbd++; 1805813a1f2bSShri Abhyankar } 1806813a1f2bSShri Abhyankar bdiag[i] = nzbd; 1807813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nzi; 1808813a1f2bSShri Abhyankar 1809813a1f2bSShri Abhyankar /* if free space is not available, make more free space */ 1810813a1f2bSShri Abhyankar if (current_space->local_remaining<nzi) { 1811813a1f2bSShri Abhyankar nnz = 2*nzi*(n - i); /* estimated and max additional space needed */ 1812813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1813813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 1814813a1f2bSShri Abhyankar reallocs++; 1815813a1f2bSShri Abhyankar } 1816813a1f2bSShri Abhyankar 1817813a1f2bSShri Abhyankar /* copy data into free_space and free_space_lvl, then initialize lnk */ 1818813a1f2bSShri Abhyankar ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 1819813a1f2bSShri Abhyankar bj_ptr[i] = current_space->array; 1820813a1f2bSShri Abhyankar bjlvl_ptr[i] = current_space_lvl->array; 1821813a1f2bSShri Abhyankar 1822813a1f2bSShri Abhyankar /* make sure the active row i has diagonal entry */ 182365e19b50SBarry Smith if (*(bj_ptr[i]+bdiag[i]) != i) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\ntry running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 1824813a1f2bSShri Abhyankar 1825813a1f2bSShri Abhyankar current_space->array += nzi; 1826813a1f2bSShri Abhyankar current_space->local_used += nzi; 1827813a1f2bSShri Abhyankar current_space->local_remaining -= nzi; 1828813a1f2bSShri Abhyankar current_space_lvl->array += nzi; 1829813a1f2bSShri Abhyankar current_space_lvl->local_used += nzi; 1830813a1f2bSShri Abhyankar current_space_lvl->local_remaining -= nzi; 1831813a1f2bSShri Abhyankar } 1832813a1f2bSShri Abhyankar 1833813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1834813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 1835813a1f2bSShri Abhyankar 1836813a1f2bSShri Abhyankar /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 18379263d837SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 18382ce24eb6SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 1839813a1f2bSShri Abhyankar 1840813a1f2bSShri Abhyankar ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1841813a1f2bSShri Abhyankar ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 18420e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 1843813a1f2bSShri Abhyankar 1844813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO) 1845813a1f2bSShri Abhyankar { 1846aef85c9fSShri Abhyankar PetscReal af = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 1847813a1f2bSShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 1848813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1849813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 1850813a1f2bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1851813a1f2bSShri Abhyankar if (diagonal_fill) { 1852813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 1853813a1f2bSShri Abhyankar } 1854813a1f2bSShri Abhyankar } 1855813a1f2bSShri Abhyankar #endif 1856813a1f2bSShri Abhyankar 1857813a1f2bSShri Abhyankar /* put together the new matrix */ 1858813a1f2bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1859813a1f2bSShri Abhyankar ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 1860813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1861813a1f2bSShri Abhyankar b->free_a = PETSC_TRUE; 1862813a1f2bSShri Abhyankar b->free_ij = PETSC_TRUE; 1863813a1f2bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 1864f268cba8SShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 1865813a1f2bSShri Abhyankar b->j = bj; 1866813a1f2bSShri Abhyankar b->i = bi; 1867813a1f2bSShri Abhyankar b->diag = bdiag; 1868813a1f2bSShri Abhyankar b->ilen = 0; 1869813a1f2bSShri Abhyankar b->imax = 0; 1870813a1f2bSShri Abhyankar b->row = isrow; 1871813a1f2bSShri Abhyankar b->col = iscol; 1872813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1873813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1874813a1f2bSShri Abhyankar b->icol = isicol; 1875813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1876813a1f2bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. 1877813a1f2bSShri Abhyankar Allocate bdiag, solve_work, new a, new j */ 1878f268cba8SShri Abhyankar ierr = PetscLogObjectMemory(fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 1879f268cba8SShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 1880813a1f2bSShri Abhyankar (fact)->info.factor_mallocs = reallocs; 1881813a1f2bSShri Abhyankar (fact)->info.fill_ratio_given = f; 1882f268cba8SShri Abhyankar (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 188335233d99SShri Abhyankar (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1884f2344125SBarry Smith if (a->inode.size) { 1885f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 1886f2344125SBarry Smith } 1887813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1888813a1f2bSShri Abhyankar } 1889813a1f2bSShri Abhyankar 18906516239fSHong Zhang #undef __FUNCT__ 1891ad04f41aSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_inplace" 1892ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 18936516239fSHong Zhang { 18946516239fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 18956516239fSHong Zhang IS isicol; 18966516239fSHong Zhang PetscErrorCode ierr; 18976516239fSHong Zhang const PetscInt *r,*ic; 18986516239fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 18996516239fSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 19006516239fSHong Zhang PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 19016516239fSHong Zhang PetscInt i,levels,diagonal_fill; 1902ace3abfcSBarry Smith PetscBool col_identity,row_identity; 19036516239fSHong Zhang PetscReal f; 19046516239fSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 19056516239fSHong Zhang PetscBT lnkbt; 19066516239fSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 19076516239fSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 19086516239fSHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 1909ace3abfcSBarry Smith PetscBool missing; 19106516239fSHong Zhang 19116516239fSHong Zhang PetscFunctionBegin; 1912e32f2f54SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 19136516239fSHong Zhang f = info->fill; 19146516239fSHong Zhang levels = (PetscInt)info->levels; 19156516239fSHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 19166516239fSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 19176516239fSHong Zhang 19186516239fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 19196516239fSHong Zhang ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 19206516239fSHong Zhang if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */ 1921f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr); 1922ad04f41aSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 1923f2344125SBarry Smith if (a->inode.size) { 1924f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 1925f2344125SBarry Smith } 1926d5f3da31SBarry Smith fact->factortype = MAT_FACTOR_ILU; 1927719d5645SBarry Smith (fact)->info.factor_mallocs = 0; 1928719d5645SBarry Smith (fact)->info.fill_ratio_given = info->fill; 1929719d5645SBarry Smith (fact)->info.fill_ratio_needed = 1.0; 1930719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 19318ef3462fSBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 1932e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 193308480c60SBarry Smith b->row = isrow; 193408480c60SBarry Smith b->col = iscol; 193582bf6240SBarry Smith b->icol = isicol; 1936719d5645SBarry Smith ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1937c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1938c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 19393a40ed3dSBarry Smith PetscFunctionReturn(0); 194008480c60SBarry Smith } 194108480c60SBarry Smith 1942898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1943898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 19449e25ed09SBarry Smith 19451bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 19461bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 19471bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1948a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 19496cf997b0SBarry Smith 19500e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr); 19515a8e39fbSHong Zhang 19525a8e39fbSHong Zhang /* create a linked list for storing column indices of the active row */ 19535a8e39fbSHong Zhang nlnk = n + 1; 19545a8e39fbSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19555a8e39fbSHong Zhang 19565a8e39fbSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 1957a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 19585a8e39fbSHong Zhang current_space = free_space; 1959a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 19605a8e39fbSHong Zhang current_space_lvl = free_space_lvl; 19615a8e39fbSHong Zhang 19625a8e39fbSHong Zhang for (i=0; i<n; i++) { 19635a8e39fbSHong Zhang nzi = 0; 19646cf997b0SBarry Smith /* copy current row into linked list */ 19655a8e39fbSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 1966e32f2f54SBarry Smith if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 19675a8e39fbSHong Zhang cols = aj + ai[r[i]]; 19685a8e39fbSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 19695a8e39fbSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19705a8e39fbSHong Zhang nzi += nlnk; 19716cf997b0SBarry Smith 19726cf997b0SBarry Smith /* make sure diagonal entry is included */ 19735a8e39fbSHong Zhang if (diagonal_fill && lnk[i] == -1) { 19746cf997b0SBarry Smith fm = n; 19755a8e39fbSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 19765a8e39fbSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 19775a8e39fbSHong Zhang lnk[fm] = i; 19785a8e39fbSHong Zhang lnk_lvl[i] = 0; 19795a8e39fbSHong Zhang nzi++; dcount++; 19806cf997b0SBarry Smith } 19816cf997b0SBarry Smith 19825a8e39fbSHong Zhang /* add pivot rows into the active row */ 19835a8e39fbSHong Zhang nzbd = 0; 19845a8e39fbSHong Zhang prow = lnk[n]; 19855a8e39fbSHong Zhang while (prow < i) { 19865a8e39fbSHong Zhang nnz = bdiag[prow]; 19875a8e39fbSHong Zhang cols = bj_ptr[prow] + nnz + 1; 19885a8e39fbSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 19895a8e39fbSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 19905a8e39fbSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 19915a8e39fbSHong Zhang nzi += nlnk; 19925a8e39fbSHong Zhang prow = lnk[prow]; 19935a8e39fbSHong Zhang nzbd++; 199456beaf04SBarry Smith } 19955a8e39fbSHong Zhang bdiag[i] = nzbd; 19965a8e39fbSHong Zhang bi[i+1] = bi[i] + nzi; 1997ecf371e4SBarry Smith 19985a8e39fbSHong Zhang /* if free space is not available, make more free space */ 19995a8e39fbSHong Zhang if (current_space->local_remaining<nzi) { 20005a8e39fbSHong Zhang nnz = nzi*(n - i); /* estimated and max additional space needed */ 2001a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2002a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 20035a8e39fbSHong Zhang reallocs++; 20045a8e39fbSHong Zhang } 2005ecf371e4SBarry Smith 20065a8e39fbSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 20075a8e39fbSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 20085a8e39fbSHong Zhang bj_ptr[i] = current_space->array; 20095a8e39fbSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 20105a8e39fbSHong Zhang 20115a8e39fbSHong Zhang /* make sure the active row i has diagonal entry */ 201265e19b50SBarry Smith if (*(bj_ptr[i]+bdiag[i]) != i) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\ntry running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i); 20135a8e39fbSHong Zhang 20145a8e39fbSHong Zhang current_space->array += nzi; 20155a8e39fbSHong Zhang current_space->local_used += nzi; 20165a8e39fbSHong Zhang current_space->local_remaining -= nzi; 20175a8e39fbSHong Zhang current_space_lvl->array += nzi; 20185a8e39fbSHong Zhang current_space_lvl->local_used += nzi; 20195a8e39fbSHong Zhang current_space_lvl->local_remaining -= nzi; 20209e25ed09SBarry Smith } 20215a8e39fbSHong Zhang 2022898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 2023898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 20245a8e39fbSHong Zhang 20255a8e39fbSHong Zhang /* destroy list of free space and other temporary arrays */ 20265a8e39fbSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 20276516239fSHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */ 20285a8e39fbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2029a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 20300e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 20319e25ed09SBarry Smith 20326cf91177SBarry Smith #if defined(PETSC_USE_INFO) 2033f64065bbSBarry Smith { 20345a8e39fbSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2035ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 2036ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2037ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 2038ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 2039335d9088SBarry Smith if (diagonal_fill) { 2040ae15b995SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 2041335d9088SBarry Smith } 2042f64065bbSBarry Smith } 204363ba0a88SBarry Smith #endif 2044bbb6d6a8SBarry Smith 20459e25ed09SBarry Smith /* put together the new matrix */ 204607dbf95fSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 2047719d5645SBarry Smith ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 2048719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 2049e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2050e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 20517c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 20520e83c824SBarry Smith ierr = PetscMalloc(bi[n]*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 20535a8e39fbSHong Zhang b->j = bj; 20545a8e39fbSHong Zhang b->i = bi; 20555a8e39fbSHong Zhang for (i=0; i<n; i++) bdiag[i] += bi[i]; 20565a8e39fbSHong Zhang b->diag = bdiag; 2057416022c9SBarry Smith b->ilen = 0; 2058416022c9SBarry Smith b->imax = 0; 2059416022c9SBarry Smith b->row = isrow; 2060416022c9SBarry Smith b->col = iscol; 2061c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2062c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 206382bf6240SBarry Smith b->icol = isicol; 206487828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2065416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 20665a8e39fbSHong Zhang Allocate bdiag, solve_work, new a, new j */ 2067719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 20685a8e39fbSHong Zhang b->maxnz = b->nz = bi[n] ; 2069719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2070719d5645SBarry Smith (fact)->info.fill_ratio_given = f; 2071719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2072ad04f41aSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 2073f2344125SBarry Smith if (a->inode.size) { 2074f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 2075d3ac4fa3SBarry Smith } 20763a40ed3dSBarry Smith PetscFunctionReturn(0); 20779e25ed09SBarry Smith } 2078d5d45c9bSBarry Smith 2079a6175056SHong Zhang #undef __FUNCT__ 208035233d99SShri Abhyankar #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ" 208135233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 20825f44c854SHong Zhang { 20835f44c854SHong Zhang Mat C = B; 20845f44c854SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 20855f44c854SHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 20865f44c854SHong Zhang IS ip=b->row,iip = b->icol; 20875f44c854SHong Zhang PetscErrorCode ierr; 20885f44c854SHong Zhang const PetscInt *rip,*riip; 20895f44c854SHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp; 20905f44c854SHong Zhang PetscInt *ai=a->i,*aj=a->j; 20915f44c854SHong Zhang PetscInt k,jmin,jmax,*c2r,*il,col,nexti,ili,nz; 20925f44c854SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 2093ace3abfcSBarry Smith PetscBool perm_identity; 2094d90ac83dSHong Zhang FactorShiftCtx sctx; 209598a93161SHong Zhang PetscReal rs; 209698a93161SHong Zhang MatScalar d,*v; 209798a93161SHong Zhang 20985f44c854SHong Zhang PetscFunctionBegin; 209998a93161SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 2100d90ac83dSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 210198a93161SHong Zhang 2102f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 210398a93161SHong Zhang sctx.shift_top = info->zeropivot; 210498a93161SHong Zhang for (i=0; i<mbs; i++) { 210598a93161SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 210698a93161SHong Zhang d = (aa)[a->diag[i]]; 210798a93161SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 210898a93161SHong Zhang v = aa+ai[i]; 210998a93161SHong Zhang nz = ai[i+1] - ai[i]; 211098a93161SHong Zhang for (j=0; j<nz; j++) 211198a93161SHong Zhang rs += PetscAbsScalar(v[j]); 211298a93161SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 211398a93161SHong Zhang } 211498a93161SHong Zhang sctx.shift_top *= 1.1; 211598a93161SHong Zhang sctx.nshift_max = 5; 211698a93161SHong Zhang sctx.shift_lo = 0.; 211798a93161SHong Zhang sctx.shift_hi = 1.; 211898a93161SHong Zhang } 21195f44c854SHong Zhang 21205f44c854SHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 21215f44c854SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 21225f44c854SHong Zhang 21235f44c854SHong Zhang /* allocate working arrays 21245f44c854SHong Zhang c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col 21255f44c854SHong 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 21265f44c854SHong Zhang */ 21270e83c824SBarry Smith ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&c2r);CHKERRQ(ierr); 21285f44c854SHong Zhang 212998a93161SHong Zhang do { 213007b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 213198a93161SHong Zhang 2132c5546cabSHong Zhang for (i=0; i<mbs; i++) c2r[i] = mbs; 21335f44c854SHong Zhang il[0] = 0; 21345f44c854SHong Zhang 21355f44c854SHong Zhang for (k = 0; k<mbs; k++){ 21365f44c854SHong Zhang /* zero rtmp */ 21375f44c854SHong Zhang nz = bi[k+1] - bi[k]; 21385f44c854SHong Zhang bjtmp = bj + bi[k]; 21395f44c854SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 21405f44c854SHong Zhang 21415f44c854SHong Zhang /* load in initial unfactored row */ 21425f44c854SHong Zhang bval = ba + bi[k]; 21435f44c854SHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 21445f44c854SHong Zhang for (j = jmin; j < jmax; j++){ 21455f44c854SHong Zhang col = riip[aj[j]]; 21465f44c854SHong Zhang if (col >= k){ /* only take upper triangular entry */ 21475f44c854SHong Zhang rtmp[col] = aa[j]; 21485f44c854SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 21495f44c854SHong Zhang } 21505f44c854SHong Zhang } 215198a93161SHong Zhang /* shift the diagonal of the matrix: ZeropivotApply() */ 215298a93161SHong Zhang rtmp[k] += sctx.shift_amount; /* shift the diagonal of the matrix */ 21535f44c854SHong Zhang 21545f44c854SHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 21555f44c854SHong Zhang dk = rtmp[k]; 21565f44c854SHong Zhang i = c2r[k]; /* first row to be added to k_th row */ 21575f44c854SHong Zhang 21585f44c854SHong Zhang while (i < k){ 21595f44c854SHong Zhang nexti = c2r[i]; /* next row to be added to k_th row */ 21605f44c854SHong Zhang 21615f44c854SHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 21625f44c854SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 21635f44c854SHong Zhang uikdi = - ba[ili]*ba[bdiag[i]]; /* diagonal(k) */ 21645f44c854SHong Zhang dk += uikdi*ba[ili]; /* update diag[k] */ 21655f44c854SHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 21665f44c854SHong Zhang 21675f44c854SHong Zhang /* add multiple of row i to k-th row */ 21685f44c854SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 21695f44c854SHong Zhang if (jmin < jmax){ 21705f44c854SHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 21715f44c854SHong Zhang /* update il and c2r for row i */ 21725f44c854SHong Zhang il[i] = jmin; 21735f44c854SHong Zhang j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i; 21745f44c854SHong Zhang } 21755f44c854SHong Zhang i = nexti; 21765f44c854SHong Zhang } 21775f44c854SHong Zhang 21785f44c854SHong Zhang /* copy data into U(k,:) */ 217998a93161SHong Zhang rs = 0.0; 21805f44c854SHong Zhang jmin = bi[k]; jmax = bi[k+1]-1; 21815f44c854SHong Zhang if (jmin < jmax) { 21825f44c854SHong Zhang for (j=jmin; j<jmax; j++){ 218398a93161SHong Zhang col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]); 21845f44c854SHong Zhang } 21855f44c854SHong Zhang /* add the k-th row into il and c2r */ 21865f44c854SHong Zhang il[k] = jmin; 21875f44c854SHong Zhang i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k; 21885f44c854SHong Zhang } 218998a93161SHong Zhang 219098a93161SHong Zhang /* MatPivotCheck() */ 219198a93161SHong Zhang sctx.rs = rs; 219298a93161SHong Zhang sctx.pv = dk; 219307b50cabSHong Zhang ierr = MatPivotCheck(info,&sctx,i);CHKERRQ(ierr); 219407b50cabSHong Zhang if(sctx.newshift) break; 219598a93161SHong Zhang dk = sctx.pv; 219698a93161SHong Zhang 219798a93161SHong Zhang ba[bdiag[k]] = 1.0/dk; /* U(k,k) */ 219898a93161SHong Zhang } 219907b50cabSHong Zhang } while (sctx.newshift); 22005f44c854SHong Zhang 22010e83c824SBarry Smith ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr); 22025f44c854SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 22035f44c854SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 22045f44c854SHong Zhang 22055f44c854SHong Zhang ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 22065f44c854SHong Zhang if (perm_identity){ 220735233d99SShri Abhyankar B->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 220835233d99SShri Abhyankar B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 22092169352eSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering; 22102169352eSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering; 22115f44c854SHong Zhang } else { 221235233d99SShri Abhyankar B->ops->solve = MatSolve_SeqSBAIJ_1; 221335233d99SShri Abhyankar B->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 221435233d99SShri Abhyankar B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1; 221535233d99SShri Abhyankar B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1; 22165f44c854SHong Zhang } 22175f44c854SHong Zhang 22185f44c854SHong Zhang C->assembled = PETSC_TRUE; 22195f44c854SHong Zhang C->preallocated = PETSC_TRUE; 22205f44c854SHong Zhang ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 222198a93161SHong Zhang 222298a93161SHong Zhang /* MatPivotView() */ 222398a93161SHong Zhang if (sctx.nshift){ 2224f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 222598a93161SHong 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); 2226f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 222798a93161SHong Zhang ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2228f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){ 2229b2a402fcSHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr); 223098a93161SHong Zhang } 223198a93161SHong Zhang } 22325f44c854SHong Zhang PetscFunctionReturn(0); 22335f44c854SHong Zhang } 22345f44c854SHong Zhang 22355f44c854SHong Zhang #undef __FUNCT__ 22360a3c351aSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_inplace" 22370a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info) 2238a6175056SHong Zhang { 2239719d5645SBarry Smith Mat C = B; 22400968510aSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 22412ed133dbSHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 22429bfd6278SHong Zhang IS ip=b->row,iip = b->icol; 22432ed133dbSHong Zhang PetscErrorCode ierr; 22445d0c19d7SBarry Smith const PetscInt *rip,*riip; 2245c5546cabSHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp; 22462ed133dbSHong Zhang PetscInt *ai=a->i,*aj=a->j; 22472ed133dbSHong Zhang PetscInt k,jmin,jmax,*jl,*il,col,nexti,ili,nz; 2248622e2028SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 2249ace3abfcSBarry Smith PetscBool perm_identity; 22500e95ead3SHong Zhang FactorShiftCtx sctx; 22510e95ead3SHong Zhang PetscReal rs; 22520e95ead3SHong Zhang MatScalar d,*v; 2253d5d45c9bSBarry Smith 2254a6175056SHong Zhang PetscFunctionBegin; 22550e95ead3SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 22560e95ead3SHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 22570e95ead3SHong Zhang 22580e95ead3SHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 22590e95ead3SHong Zhang sctx.shift_top = info->zeropivot; 22600e95ead3SHong Zhang for (i=0; i<mbs; i++) { 22610e95ead3SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 22620e95ead3SHong Zhang d = (aa)[a->diag[i]]; 22630e95ead3SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 22640e95ead3SHong Zhang v = aa+ai[i]; 22650e95ead3SHong Zhang nz = ai[i+1] - ai[i]; 22660e95ead3SHong Zhang for (j=0; j<nz; j++) 22670e95ead3SHong Zhang rs += PetscAbsScalar(v[j]); 22680e95ead3SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 22690e95ead3SHong Zhang } 22700e95ead3SHong Zhang sctx.shift_top *= 1.1; 22710e95ead3SHong Zhang sctx.nshift_max = 5; 22720e95ead3SHong Zhang sctx.shift_lo = 0.; 22730e95ead3SHong Zhang sctx.shift_hi = 1.; 22740e95ead3SHong Zhang } 2275ee45ca4aSHong Zhang 22762ed133dbSHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 22779bfd6278SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 22782ed133dbSHong Zhang 22792ed133dbSHong Zhang /* initialization */ 22800e83c824SBarry Smith ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&jl);CHKERRQ(ierr); 22810e95ead3SHong Zhang 22822ed133dbSHong Zhang do { 228307b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 22840e95ead3SHong Zhang 2285c5546cabSHong Zhang for (i=0; i<mbs; i++) jl[i] = mbs; 2286c5546cabSHong Zhang il[0] = 0; 22872ed133dbSHong Zhang 22882ed133dbSHong Zhang for (k = 0; k<mbs; k++){ 2289c5546cabSHong Zhang /* zero rtmp */ 2290c5546cabSHong Zhang nz = bi[k+1] - bi[k]; 2291c5546cabSHong Zhang bjtmp = bj + bi[k]; 2292c5546cabSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 2293c5546cabSHong Zhang 2294c05c3958SHong Zhang bval = ba + bi[k]; 22952ed133dbSHong Zhang /* initialize k-th row by the perm[k]-th row of A */ 22962ed133dbSHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 22972ed133dbSHong Zhang for (j = jmin; j < jmax; j++){ 22989bfd6278SHong Zhang col = riip[aj[j]]; 22992ed133dbSHong Zhang if (col >= k){ /* only take upper triangular entry */ 23002ed133dbSHong Zhang rtmp[col] = aa[j]; 2301c05c3958SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 23022ed133dbSHong Zhang } 23032ed133dbSHong Zhang } 230439e3d630SHong Zhang /* shift the diagonal of the matrix */ 2305540703acSHong Zhang if (sctx.nshift) rtmp[k] += sctx.shift_amount; 23062ed133dbSHong Zhang 23072ed133dbSHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 23082ed133dbSHong Zhang dk = rtmp[k]; 23092ed133dbSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 23102ed133dbSHong Zhang 23112ed133dbSHong Zhang while (i < k){ 23122ed133dbSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 23132ed133dbSHong Zhang 23142ed133dbSHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 23152ed133dbSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 23162ed133dbSHong Zhang uikdi = - ba[ili]*ba[bi[i]]; /* diagonal(k) */ 23172ed133dbSHong Zhang dk += uikdi*ba[ili]; 23182ed133dbSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 23192ed133dbSHong Zhang 23202ed133dbSHong Zhang /* add multiple of row i to k-th row */ 23212ed133dbSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 23222ed133dbSHong Zhang if (jmin < jmax){ 23232ed133dbSHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 23242ed133dbSHong Zhang /* update il and jl for row i */ 23252ed133dbSHong Zhang il[i] = jmin; 23262ed133dbSHong Zhang j = bj[jmin]; jl[i] = jl[j]; jl[j] = i; 23272ed133dbSHong Zhang } 23282ed133dbSHong Zhang i = nexti; 23292ed133dbSHong Zhang } 23302ed133dbSHong Zhang 23310697b6caSHong Zhang /* shift the diagonals when zero pivot is detected */ 23320697b6caSHong Zhang /* compute rs=sum of abs(off-diagonal) */ 23330697b6caSHong Zhang rs = 0.0; 23343c9e8598SHong Zhang jmin = bi[k]+1; 23353c9e8598SHong Zhang nz = bi[k+1] - jmin; 23363c9e8598SHong Zhang bcol = bj + jmin; 233704fbf559SBarry Smith for (j=0; j<nz; j++) { 233804fbf559SBarry Smith rs += PetscAbsScalar(rtmp[bcol[j]]); 23393c9e8598SHong Zhang } 2340540703acSHong Zhang 2341540703acSHong Zhang sctx.rs = rs; 2342540703acSHong Zhang sctx.pv = dk; 234307b50cabSHong Zhang ierr = MatPivotCheck(info,&sctx,k);CHKERRQ(ierr); 234407b50cabSHong Zhang if (sctx.newshift) break; 23450e95ead3SHong Zhang dk = sctx.pv; 23463c9e8598SHong Zhang 23473c9e8598SHong Zhang /* copy data into U(k,:) */ 234839e3d630SHong Zhang ba[bi[k]] = 1.0/dk; /* U(k,k) */ 234939e3d630SHong Zhang jmin = bi[k]+1; jmax = bi[k+1]; 235039e3d630SHong Zhang if (jmin < jmax) { 235139e3d630SHong Zhang for (j=jmin; j<jmax; j++){ 2352c6ae9bfcSHong Zhang col = bj[j]; ba[j] = rtmp[col]; 23533c9e8598SHong Zhang } 235439e3d630SHong Zhang /* add the k-th row into il and jl */ 23553c9e8598SHong Zhang il[k] = jmin; 23563c9e8598SHong Zhang i = bj[jmin]; jl[k] = jl[i]; jl[i] = k; 23573c9e8598SHong Zhang } 23583c9e8598SHong Zhang } 235907b50cabSHong Zhang } while (sctx.newshift); 23600e95ead3SHong Zhang 23610e83c824SBarry Smith ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr); 236239e3d630SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 23639bfd6278SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 2364db4efbfdSBarry Smith 2365db4efbfdSBarry Smith ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 2366db4efbfdSBarry Smith if (perm_identity){ 23670a3c351aSHong Zhang B->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23680a3c351aSHong Zhang B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23690a3c351aSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23700a3c351aSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 2371db4efbfdSBarry Smith } else { 23720a3c351aSHong Zhang B->ops->solve = MatSolve_SeqSBAIJ_1_inplace; 23730a3c351aSHong Zhang B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_inplace; 23740a3c351aSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_inplace; 23750a3c351aSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_inplace; 2376db4efbfdSBarry Smith } 2377db4efbfdSBarry Smith 23783c9e8598SHong Zhang C->assembled = PETSC_TRUE; 23793c9e8598SHong Zhang C->preallocated = PETSC_TRUE; 2380d0f46423SBarry Smith ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 2381540703acSHong Zhang if (sctx.nshift){ 2382f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 23831e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2384f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 23851e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 23860697b6caSHong Zhang } 23873c9e8598SHong Zhang } 23883c9e8598SHong Zhang PetscFunctionReturn(0); 23893c9e8598SHong Zhang } 2390a6175056SHong Zhang 23918a76ed4fSHong Zhang /* 23928a76ed4fSHong Zhang icc() under revised new data structure. 23938a76ed4fSHong Zhang Factored arrays bj and ba are stored as 23948a76ed4fSHong Zhang U(0,:),...,U(i,:),U(n-1,:) 23958a76ed4fSHong Zhang 23968a76ed4fSHong Zhang ui=fact->i is an array of size n+1, in which 23978a76ed4fSHong Zhang ui+ 23988a76ed4fSHong Zhang ui[i]: points to 1st entry of U(i,:),i=0,...,n-1 23998a76ed4fSHong Zhang ui[n]: points to U(n-1,n-1)+1 24008a76ed4fSHong Zhang 24018a76ed4fSHong Zhang udiag=fact->diag is an array of size n,in which 24028a76ed4fSHong Zhang udiag[i]: points to diagonal of U(i,:), i=0,...,n-1 24038a76ed4fSHong Zhang 24048a76ed4fSHong Zhang U(i,:) contains udiag[i] as its last entry, i.e., 24058a76ed4fSHong Zhang U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 24068a76ed4fSHong Zhang */ 24078a76ed4fSHong Zhang 2408a6175056SHong Zhang #undef __FUNCT__ 240935233d99SShri Abhyankar #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ" 241035233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2411a6175056SHong Zhang { 24120968510aSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2413ed59401aSHong Zhang Mat_SeqSBAIJ *b; 2414dfbe8321SBarry Smith PetscErrorCode ierr; 2415ace3abfcSBarry Smith PetscBool perm_identity,missing; 24165f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 24175d0c19d7SBarry Smith const PetscInt *rip,*riip; 2418ed59401aSHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 241958ebbce7SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 24205a8e39fbSHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 2421ed59401aSHong Zhang PetscReal fill=info->fill,levels=info->levels; 2422a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 2423a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 24247a48dd6fSHong Zhang PetscBT lnkbt; 2425b635d36bSHong Zhang IS iperm; 2426a6175056SHong Zhang 2427a6175056SHong Zhang PetscFunctionBegin; 2428e32f2f54SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 242958ebbce7SBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 2430e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 2431653a6975SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 2432b635d36bSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 24337a48dd6fSHong Zhang 243439e3d630SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 24355f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 243639e3d630SHong Zhang ui[0] = 0; 243739e3d630SHong Zhang 24388a76ed4fSHong Zhang /* ICC(0) without matrix ordering: simply rearrange column indices */ 2439622e2028SHong Zhang if (!levels && perm_identity) { 24405f44c854SHong Zhang for (i=0; i<am; i++) { 2441c5546cabSHong Zhang ncols = ai[i+1] - a->diag[i]; 2442c5546cabSHong Zhang ui[i+1] = ui[i] + ncols; 2443c5546cabSHong Zhang udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */ 2444dc1e2a3fSHong Zhang } 2445dc1e2a3fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2446dc1e2a3fSHong Zhang cols = uj; 2447dc1e2a3fSHong Zhang for (i=0; i<am; i++) { 24485f44c854SHong Zhang aj = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */ 2449dc1e2a3fSHong Zhang ncols = ai[i+1] - a->diag[i] -1; 24505f44c854SHong Zhang for (j=0; j<ncols; j++) *cols++ = aj[j]; 24515f44c854SHong Zhang *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */ 24525f44c854SHong Zhang } 24535f44c854SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 24545f44c854SHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 24555f44c854SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 24565f44c854SHong Zhang 24575f44c854SHong Zhang /* initialization */ 24585f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 24595f44c854SHong Zhang 24605f44c854SHong Zhang /* jl: linked list for storing indices of the pivot rows 24615f44c854SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 24620e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr); 24635f44c854SHong Zhang for (i=0; i<am; i++){ 24645f44c854SHong Zhang jl[i] = am; il[i] = 0; 24655f44c854SHong Zhang } 24665f44c854SHong Zhang 24675f44c854SHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 24685f44c854SHong Zhang nlnk = am + 1; 24695f44c854SHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 24705f44c854SHong Zhang 247195b5ac22SHong Zhang /* initial FreeSpace size is fill*(ai[am]+am)/2 */ 247295b5ac22SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space);CHKERRQ(ierr); 24735f44c854SHong Zhang current_space = free_space; 247495b5ac22SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space_lvl);CHKERRQ(ierr); 24755f44c854SHong Zhang current_space_lvl = free_space_lvl; 24765f44c854SHong Zhang 24775f44c854SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 24785f44c854SHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 24795f44c854SHong Zhang nzk = 0; 24805f44c854SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2481e32f2f54SBarry Smith if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 24825f44c854SHong Zhang ncols_upper = 0; 24835f44c854SHong Zhang for (j=0; j<ncols; j++){ 24845f44c854SHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 24855f44c854SHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 24865f44c854SHong Zhang ajtmp[ncols_upper] = i; 24875f44c854SHong Zhang ncols_upper++; 24885f44c854SHong Zhang } 24895f44c854SHong Zhang } 24905f44c854SHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 24915f44c854SHong Zhang nzk += nlnk; 24925f44c854SHong Zhang 24935f44c854SHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 24945f44c854SHong Zhang prow = jl[k]; /* 1st pivot row */ 24955f44c854SHong Zhang 24965f44c854SHong Zhang while (prow < k){ 24975f44c854SHong Zhang nextprow = jl[prow]; 24985f44c854SHong Zhang 24995f44c854SHong Zhang /* merge prow into k-th row */ 25005f44c854SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 25015f44c854SHong Zhang jmax = ui[prow+1]; 25025f44c854SHong Zhang ncols = jmax-jmin; 25035f44c854SHong Zhang i = jmin - ui[prow]; 25045f44c854SHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 25055f44c854SHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 25065f44c854SHong Zhang j = *(uj - 1); 25075f44c854SHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 25085f44c854SHong Zhang nzk += nlnk; 25095f44c854SHong Zhang 25105f44c854SHong Zhang /* update il and jl for prow */ 25115f44c854SHong Zhang if (jmin < jmax){ 25125f44c854SHong Zhang il[prow] = jmin; 25135f44c854SHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 25145f44c854SHong Zhang } 25155f44c854SHong Zhang prow = nextprow; 25165f44c854SHong Zhang } 25175f44c854SHong Zhang 25185f44c854SHong Zhang /* if free space is not available, make more free space */ 25195f44c854SHong Zhang if (current_space->local_remaining<nzk) { 25205f44c854SHong Zhang i = am - k + 1; /* num of unfactored rows */ 2521c6ae9bfcSHong Zhang i *= PetscMin(nzk, i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */ 25225f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 25235f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 25245f44c854SHong Zhang reallocs++; 25255f44c854SHong Zhang } 25265f44c854SHong Zhang 25275f44c854SHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 2528e32f2f54SBarry Smith if (nzk == 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 25295f44c854SHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 25305f44c854SHong Zhang 25315f44c854SHong Zhang /* add the k-th row into il and jl */ 25325f44c854SHong Zhang if (nzk > 1){ 25335f44c854SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 25345f44c854SHong Zhang jl[k] = jl[i]; jl[i] = k; 25355f44c854SHong Zhang il[k] = ui[k] + 1; 25365f44c854SHong Zhang } 25375f44c854SHong Zhang uj_ptr[k] = current_space->array; 25385f44c854SHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 25395f44c854SHong Zhang 25405f44c854SHong Zhang current_space->array += nzk; 25415f44c854SHong Zhang current_space->local_used += nzk; 25425f44c854SHong Zhang current_space->local_remaining -= nzk; 25435f44c854SHong Zhang 25445f44c854SHong Zhang current_space_lvl->array += nzk; 25455f44c854SHong Zhang current_space_lvl->local_used += nzk; 25465f44c854SHong Zhang current_space_lvl->local_remaining -= nzk; 25475f44c854SHong Zhang 25485f44c854SHong Zhang ui[k+1] = ui[k] + nzk; 25495f44c854SHong Zhang } 25505f44c854SHong Zhang 25515f44c854SHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 25525f44c854SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 25530e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 25545f44c854SHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 25555f44c854SHong Zhang 25569263d837SHong Zhang /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */ 25575f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 255835233d99SShri Abhyankar ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */ 25595f44c854SHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 25605f44c854SHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 25615f44c854SHong Zhang 25625f44c854SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 25635f44c854SHong Zhang 25645f44c854SHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 25655f44c854SHong Zhang b = (Mat_SeqSBAIJ*)(fact)->data; 25665f44c854SHong Zhang b->singlemalloc = PETSC_FALSE; 25675f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 25685f44c854SHong Zhang b->j = uj; 25695f44c854SHong Zhang b->i = ui; 25705f44c854SHong Zhang b->diag = udiag; 25717f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 25725f44c854SHong Zhang b->ilen = 0; 25735f44c854SHong Zhang b->imax = 0; 25745f44c854SHong Zhang b->row = perm; 25755f44c854SHong Zhang b->col = perm; 25765f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25775f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25785f44c854SHong Zhang b->icol = iperm; 25795f44c854SHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 25805f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2581d595f711SHong Zhang ierr = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 25825f44c854SHong Zhang b->maxnz = b->nz = ui[am]; 25835f44c854SHong Zhang b->free_a = PETSC_TRUE; 25845f44c854SHong Zhang b->free_ij = PETSC_TRUE; 25855f44c854SHong Zhang 2586f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2587f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 25885f44c854SHong Zhang if (ai[am] != 0) { 25896a69fef8SHong Zhang /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */ 259095b5ac22SHong Zhang fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am); 25915f44c854SHong Zhang } else { 2592f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 25935f44c854SHong Zhang } 25949263d837SHong Zhang #if defined(PETSC_USE_INFO) 25959263d837SHong Zhang if (ai[am] != 0) { 25969263d837SHong Zhang PetscReal af = fact->info.fill_ratio_needed; 25979263d837SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 25989263d837SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 25999263d837SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 26009263d837SHong Zhang } else { 26019263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 26029263d837SHong Zhang } 26039263d837SHong Zhang #endif 260435233d99SShri Abhyankar fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 26055f44c854SHong Zhang PetscFunctionReturn(0); 26065f44c854SHong Zhang } 26075f44c854SHong Zhang 26085f44c854SHong Zhang #undef __FUNCT__ 2609ad04f41aSHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_inplace" 2610ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 26115f44c854SHong Zhang { 26125f44c854SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 26135f44c854SHong Zhang Mat_SeqSBAIJ *b; 26145f44c854SHong Zhang PetscErrorCode ierr; 2615ace3abfcSBarry Smith PetscBool perm_identity,missing; 26165f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 26175f44c854SHong Zhang const PetscInt *rip,*riip; 26185f44c854SHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 26195f44c854SHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 26205f44c854SHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 26215f44c854SHong Zhang PetscReal fill=info->fill,levels=info->levels; 26225f44c854SHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 26235f44c854SHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 26245f44c854SHong Zhang PetscBT lnkbt; 26255f44c854SHong Zhang IS iperm; 26265f44c854SHong Zhang 26275f44c854SHong Zhang PetscFunctionBegin; 2628e32f2f54SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 26295f44c854SHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 2630e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 26315f44c854SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 26325f44c854SHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 26335f44c854SHong Zhang 26345f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 26355f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 26365f44c854SHong Zhang ui[0] = 0; 26375f44c854SHong Zhang 26385f44c854SHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 26395f44c854SHong Zhang if (!levels && perm_identity) { 26405f44c854SHong Zhang 26415f44c854SHong Zhang for (i=0; i<am; i++) { 26425f44c854SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 26435f44c854SHong Zhang udiag[i] = ui[i]; 2644ed59401aSHong Zhang } 264539e3d630SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 264639e3d630SHong Zhang cols = uj; 2647ed59401aSHong Zhang for (i=0; i<am; i++) { 2648ed59401aSHong Zhang aj = a->j + a->diag[i]; 264939e3d630SHong Zhang ncols = ui[i+1] - ui[i]; 265039e3d630SHong Zhang for (j=0; j<ncols; j++) *cols++ = *aj++; 2651ed59401aSHong Zhang } 265239e3d630SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 2653b635d36bSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 2654b635d36bSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 2655b635d36bSHong Zhang 26567a48dd6fSHong Zhang /* initialization */ 26575a8e39fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 26587a48dd6fSHong Zhang 26597a48dd6fSHong Zhang /* jl: linked list for storing indices of the pivot rows 26607a48dd6fSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 26610e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr); 26627a48dd6fSHong Zhang for (i=0; i<am; i++){ 26637a48dd6fSHong Zhang jl[i] = am; il[i] = 0; 26647a48dd6fSHong Zhang } 26657a48dd6fSHong Zhang 26667a48dd6fSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 26677a48dd6fSHong Zhang nlnk = am + 1; 26682ed133dbSHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 26697a48dd6fSHong Zhang 26707a48dd6fSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2671a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 26727a48dd6fSHong Zhang current_space = free_space; 2673a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 26747a48dd6fSHong Zhang current_space_lvl = free_space_lvl; 26757a48dd6fSHong Zhang 26767a48dd6fSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 26777a48dd6fSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 26787a48dd6fSHong Zhang nzk = 0; 2679622e2028SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2680e32f2f54SBarry Smith if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 2681622e2028SHong Zhang ncols_upper = 0; 2682622e2028SHong Zhang for (j=0; j<ncols; j++){ 2683b635d36bSHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 2684b635d36bSHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 26855a8e39fbSHong Zhang ajtmp[ncols_upper] = i; 2686622e2028SHong Zhang ncols_upper++; 2687622e2028SHong Zhang } 2688622e2028SHong Zhang } 2689b635d36bSHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 26907a48dd6fSHong Zhang nzk += nlnk; 26917a48dd6fSHong Zhang 26927a48dd6fSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 26937a48dd6fSHong Zhang prow = jl[k]; /* 1st pivot row */ 26947a48dd6fSHong Zhang 26957a48dd6fSHong Zhang while (prow < k){ 26967a48dd6fSHong Zhang nextprow = jl[prow]; 26977a48dd6fSHong Zhang 26987a48dd6fSHong Zhang /* merge prow into k-th row */ 26997a48dd6fSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 27007a48dd6fSHong Zhang jmax = ui[prow+1]; 27017a48dd6fSHong Zhang ncols = jmax-jmin; 2702ed59401aSHong Zhang i = jmin - ui[prow]; 2703ed59401aSHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 27045a8e39fbSHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 27055a8e39fbSHong Zhang j = *(uj - 1); 27065a8e39fbSHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 27077a48dd6fSHong Zhang nzk += nlnk; 27087a48dd6fSHong Zhang 27097a48dd6fSHong Zhang /* update il and jl for prow */ 27107a48dd6fSHong Zhang if (jmin < jmax){ 27117a48dd6fSHong Zhang il[prow] = jmin; 27127a48dd6fSHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 27137a48dd6fSHong Zhang } 27147a48dd6fSHong Zhang prow = nextprow; 27157a48dd6fSHong Zhang } 27167a48dd6fSHong Zhang 27177a48dd6fSHong Zhang /* if free space is not available, make more free space */ 27187a48dd6fSHong Zhang if (current_space->local_remaining<nzk) { 27197a48dd6fSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2720fd3f55cbSMatthew Knepley i *= PetscMin(nzk, (i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2721a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 2722a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 27237a48dd6fSHong Zhang reallocs++; 27247a48dd6fSHong Zhang } 27257a48dd6fSHong Zhang 27267a48dd6fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 272765e19b50SBarry Smith if (!nzk) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 27282ed133dbSHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 27297a48dd6fSHong Zhang 27307a48dd6fSHong Zhang /* add the k-th row into il and jl */ 273139e3d630SHong Zhang if (nzk > 1){ 27327a48dd6fSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 27337a48dd6fSHong Zhang jl[k] = jl[i]; jl[i] = k; 27347a48dd6fSHong Zhang il[k] = ui[k] + 1; 27357a48dd6fSHong Zhang } 27367a48dd6fSHong Zhang uj_ptr[k] = current_space->array; 27377a48dd6fSHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 27387a48dd6fSHong Zhang 27397a48dd6fSHong Zhang current_space->array += nzk; 27407a48dd6fSHong Zhang current_space->local_used += nzk; 27417a48dd6fSHong Zhang current_space->local_remaining -= nzk; 27427a48dd6fSHong Zhang 27437a48dd6fSHong Zhang current_space_lvl->array += nzk; 27447a48dd6fSHong Zhang current_space_lvl->local_used += nzk; 27457a48dd6fSHong Zhang current_space_lvl->local_remaining -= nzk; 27467a48dd6fSHong Zhang 27477a48dd6fSHong Zhang ui[k+1] = ui[k] + nzk; 27487a48dd6fSHong Zhang } 27497a48dd6fSHong Zhang 27506cf91177SBarry Smith #if defined(PETSC_USE_INFO) 27517a48dd6fSHong Zhang if (ai[am] != 0) { 275239e3d630SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 2753ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2754ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2755ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 27567a48dd6fSHong Zhang } else { 2757ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 27587a48dd6fSHong Zhang } 275963ba0a88SBarry Smith #endif 27607a48dd6fSHong Zhang 27617a48dd6fSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 2762b635d36bSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 27630e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 27645a8e39fbSHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 27657a48dd6fSHong Zhang 27667a48dd6fSHong Zhang /* destroy list of free space and other temporary array(s) */ 27677a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2768a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 27692ed133dbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2770a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 27717a48dd6fSHong Zhang 277239e3d630SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 277339e3d630SHong Zhang 27747a48dd6fSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 27757a48dd6fSHong Zhang 2776f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 27777a48dd6fSHong Zhang b->singlemalloc = PETSC_FALSE; 27787a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 27797a48dd6fSHong Zhang b->j = uj; 27807a48dd6fSHong Zhang b->i = ui; 27815f44c854SHong Zhang b->diag = udiag; 27827f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 27837a48dd6fSHong Zhang b->ilen = 0; 27847a48dd6fSHong Zhang b->imax = 0; 27857a48dd6fSHong Zhang b->row = perm; 2786b635d36bSHong Zhang b->col = perm; 2787b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2788b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2789b635d36bSHong Zhang b->icol = iperm; 27907a48dd6fSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 27917a48dd6fSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2792f284f12aSHong Zhang ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 27937a48dd6fSHong Zhang b->maxnz = b->nz = ui[am]; 2794e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2795e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 27967a48dd6fSHong Zhang 2797f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2798f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 27997a48dd6fSHong Zhang if (ai[am] != 0) { 2800f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 28017a48dd6fSHong Zhang } else { 2802f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 28037a48dd6fSHong Zhang } 28040a3c351aSHong Zhang fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace; 2805a6175056SHong Zhang PetscFunctionReturn(0); 2806a6175056SHong Zhang } 2807d5d45c9bSBarry Smith 2808177d4faaSHong Zhang #undef __FUNCT__ 2809177d4faaSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ" 281035233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 28110be760fbSHong Zhang { 28120be760fbSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 28130be760fbSHong Zhang Mat_SeqSBAIJ *b; 28140be760fbSHong Zhang PetscErrorCode ierr; 2815ace3abfcSBarry Smith PetscBool perm_identity; 28160be760fbSHong Zhang PetscReal fill = info->fill; 28170be760fbSHong Zhang const PetscInt *rip,*riip; 28180be760fbSHong Zhang PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 28190be760fbSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 28200be760fbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag; 28210be760fbSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 28220be760fbSHong Zhang PetscBT lnkbt; 28230be760fbSHong Zhang IS iperm; 28240be760fbSHong Zhang 28250be760fbSHong Zhang PetscFunctionBegin; 2826e32f2f54SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 28270be760fbSHong Zhang /* check whether perm is the identity mapping */ 28280be760fbSHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 28290be760fbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 28300be760fbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 28310be760fbSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 28320be760fbSHong Zhang 28330be760fbSHong Zhang /* initialization */ 28340be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 28350be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 28360be760fbSHong Zhang ui[0] = 0; 28370be760fbSHong Zhang 28380be760fbSHong Zhang /* jl: linked list for storing indices of the pivot rows 28390be760fbSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 28400be760fbSHong Zhang ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr); 28410be760fbSHong Zhang for (i=0; i<am; i++){ 28420be760fbSHong Zhang jl[i] = am; il[i] = 0; 28430be760fbSHong Zhang } 28440be760fbSHong Zhang 28450be760fbSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 28460be760fbSHong Zhang nlnk = am + 1; 28470be760fbSHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28480be760fbSHong Zhang 284995b5ac22SHong Zhang /* initial FreeSpace size is fill*(ai[am]+am)/2 */ 285095b5ac22SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space);CHKERRQ(ierr); 28510be760fbSHong Zhang current_space = free_space; 28520be760fbSHong Zhang 28530be760fbSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 28540be760fbSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 28550be760fbSHong Zhang nzk = 0; 28560be760fbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2857e32f2f54SBarry Smith if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 28580be760fbSHong Zhang ncols_upper = 0; 28590be760fbSHong Zhang for (j=0; j<ncols; j++){ 28600be760fbSHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 28610be760fbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 28620be760fbSHong Zhang cols[ncols_upper] = i; 28630be760fbSHong Zhang ncols_upper++; 28640be760fbSHong Zhang } 28650be760fbSHong Zhang } 28660be760fbSHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28670be760fbSHong Zhang nzk += nlnk; 28680be760fbSHong Zhang 28690be760fbSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 28700be760fbSHong Zhang prow = jl[k]; /* 1st pivot row */ 28710be760fbSHong Zhang 28720be760fbSHong Zhang while (prow < k){ 28730be760fbSHong Zhang nextprow = jl[prow]; 28740be760fbSHong Zhang /* merge prow into k-th row */ 28750be760fbSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 28760be760fbSHong Zhang jmax = ui[prow+1]; 28770be760fbSHong Zhang ncols = jmax-jmin; 28780be760fbSHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 28790be760fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28800be760fbSHong Zhang nzk += nlnk; 28810be760fbSHong Zhang 28820be760fbSHong Zhang /* update il and jl for prow */ 28830be760fbSHong Zhang if (jmin < jmax){ 28840be760fbSHong Zhang il[prow] = jmin; 28850be760fbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 28860be760fbSHong Zhang } 28870be760fbSHong Zhang prow = nextprow; 28880be760fbSHong Zhang } 28890be760fbSHong Zhang 28900be760fbSHong Zhang /* if free space is not available, make more free space */ 28910be760fbSHong Zhang if (current_space->local_remaining<nzk) { 28920be760fbSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2893c6ae9bfcSHong Zhang i *= PetscMin(nzk,i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */ 28940be760fbSHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 28950be760fbSHong Zhang reallocs++; 28960be760fbSHong Zhang } 28970be760fbSHong Zhang 28980be760fbSHong Zhang /* copy data into free space, then initialize lnk */ 28990be760fbSHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 29000be760fbSHong Zhang 29010be760fbSHong Zhang /* add the k-th row into il and jl */ 29027b056e98SHong Zhang if (nzk > 1){ 29030be760fbSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 29040be760fbSHong Zhang jl[k] = jl[i]; jl[i] = k; 29050be760fbSHong Zhang il[k] = ui[k] + 1; 29060be760fbSHong Zhang } 29070be760fbSHong Zhang ui_ptr[k] = current_space->array; 29080be760fbSHong Zhang current_space->array += nzk; 29090be760fbSHong Zhang current_space->local_used += nzk; 29100be760fbSHong Zhang current_space->local_remaining -= nzk; 29110be760fbSHong Zhang 29120be760fbSHong Zhang ui[k+1] = ui[k] + nzk; 29130be760fbSHong Zhang } 29140be760fbSHong Zhang 29150be760fbSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 29160be760fbSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 29170be760fbSHong Zhang ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 29180be760fbSHong Zhang 29199263d837SHong Zhang /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */ 29200be760fbSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 292135233d99SShri Abhyankar ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */ 29220be760fbSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 29230be760fbSHong Zhang 29240be760fbSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 29250be760fbSHong Zhang 2926f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 29270be760fbSHong Zhang b->singlemalloc = PETSC_FALSE; 29280be760fbSHong Zhang b->free_a = PETSC_TRUE; 29290be760fbSHong Zhang b->free_ij = PETSC_TRUE; 29300be760fbSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 29310be760fbSHong Zhang b->j = uj; 29320be760fbSHong Zhang b->i = ui; 29330be760fbSHong Zhang b->diag = udiag; 29340be760fbSHong Zhang b->free_diag = PETSC_TRUE; 29350be760fbSHong Zhang b->ilen = 0; 29360be760fbSHong Zhang b->imax = 0; 29370be760fbSHong Zhang b->row = perm; 29380be760fbSHong Zhang b->col = perm; 29390be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29400be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29410be760fbSHong Zhang b->icol = iperm; 29420be760fbSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 29430be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 29440860e08cSHong Zhang ierr = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 29450be760fbSHong Zhang b->maxnz = b->nz = ui[am]; 29460be760fbSHong Zhang 2947f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2948f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 29490be760fbSHong Zhang if (ai[am] != 0) { 29506a69fef8SHong Zhang /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */ 295195b5ac22SHong Zhang fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am); 29520be760fbSHong Zhang } else { 2953f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 29540be760fbSHong Zhang } 29559263d837SHong Zhang #if defined(PETSC_USE_INFO) 29569263d837SHong Zhang if (ai[am] != 0) { 29579263d837SHong Zhang PetscReal af = fact->info.fill_ratio_needed; 29589263d837SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 29599263d837SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 29609263d837SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 29619263d837SHong Zhang } else { 29629263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 29639263d837SHong Zhang } 29649263d837SHong Zhang #endif 296535233d99SShri Abhyankar fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 29660be760fbSHong Zhang PetscFunctionReturn(0); 29670be760fbSHong Zhang } 29680be760fbSHong Zhang 2969f76d2b81SHong Zhang #undef __FUNCT__ 2970ad04f41aSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ_inplace" 2971ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2972f76d2b81SHong Zhang { 2973f76d2b81SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 297410c27e3dSHong Zhang Mat_SeqSBAIJ *b; 2975dfbe8321SBarry Smith PetscErrorCode ierr; 2976ace3abfcSBarry Smith PetscBool perm_identity; 297710c27e3dSHong Zhang PetscReal fill = info->fill; 29785d0c19d7SBarry Smith const PetscInt *rip,*riip; 29795d0c19d7SBarry Smith PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 298010c27e3dSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 29812ed133dbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr; 2982a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 298310c27e3dSHong Zhang PetscBT lnkbt; 29842ed133dbSHong Zhang IS iperm; 2985f76d2b81SHong Zhang 2986f76d2b81SHong Zhang PetscFunctionBegin; 2987e32f2f54SBarry Smith if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n); 29882ed133dbSHong Zhang /* check whether perm is the identity mapping */ 2989f76d2b81SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 29902ed133dbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 29912ed133dbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 29929bfd6278SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 299310c27e3dSHong Zhang 299410c27e3dSHong Zhang /* initialization */ 29951393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 299610c27e3dSHong Zhang ui[0] = 0; 299710c27e3dSHong Zhang 299810c27e3dSHong Zhang /* jl: linked list for storing indices of the pivot rows 29991393bc97SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 30000e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr); 30011393bc97SHong Zhang for (i=0; i<am; i++){ 30021393bc97SHong Zhang jl[i] = am; il[i] = 0; 3003f76d2b81SHong Zhang } 300410c27e3dSHong Zhang 300510c27e3dSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 30061393bc97SHong Zhang nlnk = am + 1; 30071393bc97SHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 300810c27e3dSHong Zhang 30091393bc97SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 3010a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 301110c27e3dSHong Zhang current_space = free_space; 301210c27e3dSHong Zhang 30131393bc97SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 301410c27e3dSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 301510c27e3dSHong Zhang nzk = 0; 30162ed133dbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 3017e32f2f54SBarry Smith if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k); 30182ed133dbSHong Zhang ncols_upper = 0; 30192ed133dbSHong Zhang for (j=0; j<ncols; j++){ 30209bfd6278SHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 30212ed133dbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 30222ed133dbSHong Zhang cols[ncols_upper] = i; 30232ed133dbSHong Zhang ncols_upper++; 30242ed133dbSHong Zhang } 30252ed133dbSHong Zhang } 30261393bc97SHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 302710c27e3dSHong Zhang nzk += nlnk; 302810c27e3dSHong Zhang 302910c27e3dSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 303010c27e3dSHong Zhang prow = jl[k]; /* 1st pivot row */ 303110c27e3dSHong Zhang 303210c27e3dSHong Zhang while (prow < k){ 303310c27e3dSHong Zhang nextprow = jl[prow]; 303410c27e3dSHong Zhang /* merge prow into k-th row */ 30351393bc97SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 303610c27e3dSHong Zhang jmax = ui[prow+1]; 303710c27e3dSHong Zhang ncols = jmax-jmin; 30381393bc97SHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 30395a8e39fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 304010c27e3dSHong Zhang nzk += nlnk; 304110c27e3dSHong Zhang 304210c27e3dSHong Zhang /* update il and jl for prow */ 304310c27e3dSHong Zhang if (jmin < jmax){ 304410c27e3dSHong Zhang il[prow] = jmin; 30452ed133dbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 304610c27e3dSHong Zhang } 304710c27e3dSHong Zhang prow = nextprow; 304810c27e3dSHong Zhang } 304910c27e3dSHong Zhang 305010c27e3dSHong Zhang /* if free space is not available, make more free space */ 305110c27e3dSHong Zhang if (current_space->local_remaining<nzk) { 30521393bc97SHong Zhang i = am - k + 1; /* num of unfactored rows */ 305310c27e3dSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 3054a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 305510c27e3dSHong Zhang reallocs++; 305610c27e3dSHong Zhang } 305710c27e3dSHong Zhang 305810c27e3dSHong Zhang /* copy data into free space, then initialize lnk */ 30591393bc97SHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 306010c27e3dSHong Zhang 306110c27e3dSHong Zhang /* add the k-th row into il and jl */ 306210c27e3dSHong Zhang if (nzk-1 > 0){ 30631393bc97SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 306410c27e3dSHong Zhang jl[k] = jl[i]; jl[i] = k; 306510c27e3dSHong Zhang il[k] = ui[k] + 1; 306610c27e3dSHong Zhang } 30672ed133dbSHong Zhang ui_ptr[k] = current_space->array; 306810c27e3dSHong Zhang current_space->array += nzk; 306910c27e3dSHong Zhang current_space->local_used += nzk; 307010c27e3dSHong Zhang current_space->local_remaining -= nzk; 307110c27e3dSHong Zhang 307210c27e3dSHong Zhang ui[k+1] = ui[k] + nzk; 307310c27e3dSHong Zhang } 307410c27e3dSHong Zhang 30756cf91177SBarry Smith #if defined(PETSC_USE_INFO) 30761393bc97SHong Zhang if (ai[am] != 0) { 30771393bc97SHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 3078ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 3079ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 3080ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 308110c27e3dSHong Zhang } else { 3082ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 308310c27e3dSHong Zhang } 308463ba0a88SBarry Smith #endif 308510c27e3dSHong Zhang 308610c27e3dSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 30879bfd6278SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 30880e83c824SBarry Smith ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 308910c27e3dSHong Zhang 309010c27e3dSHong Zhang /* destroy list of free space and other temporary array(s) */ 30911393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 3092a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 309310c27e3dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 309410c27e3dSHong Zhang 309510c27e3dSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 309610c27e3dSHong Zhang 3097f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 309810c27e3dSHong Zhang b->singlemalloc = PETSC_FALSE; 3099e6b907acSBarry Smith b->free_a = PETSC_TRUE; 3100e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 31011393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 310210c27e3dSHong Zhang b->j = uj; 310310c27e3dSHong Zhang b->i = ui; 310410c27e3dSHong Zhang b->diag = 0; 310510c27e3dSHong Zhang b->ilen = 0; 310610c27e3dSHong Zhang b->imax = 0; 310710c27e3dSHong Zhang b->row = perm; 31089bfd6278SHong Zhang b->col = perm; 31099bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 31109bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 31119bfd6278SHong Zhang b->icol = iperm; 311210c27e3dSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 31131393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3114719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 31151393bc97SHong Zhang b->maxnz = b->nz = ui[am]; 311610c27e3dSHong Zhang 3117f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 3118f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 31191393bc97SHong Zhang if (ai[am] != 0) { 3120f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 312110c27e3dSHong Zhang } else { 3122f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 312310c27e3dSHong Zhang } 31240a3c351aSHong Zhang fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace; 3125f76d2b81SHong Zhang PetscFunctionReturn(0); 3126f76d2b81SHong Zhang } 3127599ef60dSHong Zhang 31281d098868SHong Zhang #undef __FUNCT__ 312935233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering" 313035233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 3131813a1f2bSShri Abhyankar { 3132813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3133813a1f2bSShri Abhyankar PetscErrorCode ierr; 3134813a1f2bSShri Abhyankar PetscInt n = A->rmap->n; 3135813a1f2bSShri Abhyankar const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag,*vi; 3136813a1f2bSShri Abhyankar PetscScalar *x,sum; 3137813a1f2bSShri Abhyankar const PetscScalar *b; 3138813a1f2bSShri Abhyankar const MatScalar *aa = a->a,*v; 3139813a1f2bSShri Abhyankar PetscInt i,nz; 3140813a1f2bSShri Abhyankar 3141813a1f2bSShri Abhyankar PetscFunctionBegin; 3142813a1f2bSShri Abhyankar if (!n) PetscFunctionReturn(0); 3143813a1f2bSShri Abhyankar 31443649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 3145813a1f2bSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3146813a1f2bSShri Abhyankar 3147813a1f2bSShri Abhyankar /* forward solve the lower triangular */ 3148813a1f2bSShri Abhyankar x[0] = b[0]; 3149813a1f2bSShri Abhyankar v = aa; 3150813a1f2bSShri Abhyankar vi = aj; 3151813a1f2bSShri Abhyankar for (i=1; i<n; i++) { 3152813a1f2bSShri Abhyankar nz = ai[i+1] - ai[i]; 3153813a1f2bSShri Abhyankar sum = b[i]; 3154813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 3155813a1f2bSShri Abhyankar v += nz; 3156813a1f2bSShri Abhyankar vi += nz; 3157813a1f2bSShri Abhyankar x[i] = sum; 3158813a1f2bSShri Abhyankar } 3159813a1f2bSShri Abhyankar 3160813a1f2bSShri Abhyankar /* backward solve the upper triangular */ 316162fbd6afSShri Abhyankar for (i=n-1; i>=0; i--){ 31623c0119dfSShri Abhyankar v = aa + adiag[i+1] + 1; 31633c0119dfSShri Abhyankar vi = aj + adiag[i+1] + 1; 3164813a1f2bSShri Abhyankar nz = adiag[i] - adiag[i+1]-1; 3165813a1f2bSShri Abhyankar sum = x[i]; 3166813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 31673c0119dfSShri Abhyankar x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */ 3168813a1f2bSShri Abhyankar } 3169813a1f2bSShri Abhyankar 3170813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 31713649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3172813a1f2bSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3173813a1f2bSShri Abhyankar PetscFunctionReturn(0); 3174813a1f2bSShri Abhyankar } 3175813a1f2bSShri Abhyankar 31761d098868SHong Zhang #undef __FUNCT__ 317735233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ" 317835233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 3179f268cba8SShri Abhyankar { 3180f268cba8SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3181f268cba8SShri Abhyankar IS iscol = a->col,isrow = a->row; 3182f268cba8SShri Abhyankar PetscErrorCode ierr; 3183f268cba8SShri Abhyankar PetscInt i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz; 3184f268cba8SShri Abhyankar const PetscInt *rout,*cout,*r,*c; 3185f268cba8SShri Abhyankar PetscScalar *x,*tmp,sum; 3186f268cba8SShri Abhyankar const PetscScalar *b; 3187f268cba8SShri Abhyankar const MatScalar *aa = a->a,*v; 3188f268cba8SShri Abhyankar 3189f268cba8SShri Abhyankar PetscFunctionBegin; 3190f268cba8SShri Abhyankar if (!n) PetscFunctionReturn(0); 3191f268cba8SShri Abhyankar 31923649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 3193f268cba8SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3194f268cba8SShri Abhyankar tmp = a->solve_work; 3195f268cba8SShri Abhyankar 3196f268cba8SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 3197f268cba8SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 3198f268cba8SShri Abhyankar 3199f268cba8SShri Abhyankar /* forward solve the lower triangular */ 3200f268cba8SShri Abhyankar tmp[0] = b[r[0]]; 3201f268cba8SShri Abhyankar v = aa; 3202f268cba8SShri Abhyankar vi = aj; 3203f268cba8SShri Abhyankar for (i=1; i<n; i++) { 3204f268cba8SShri Abhyankar nz = ai[i+1] - ai[i]; 3205f268cba8SShri Abhyankar sum = b[r[i]]; 3206f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3207f268cba8SShri Abhyankar tmp[i] = sum; 3208f268cba8SShri Abhyankar v += nz; vi += nz; 3209f268cba8SShri Abhyankar } 3210f268cba8SShri Abhyankar 3211f268cba8SShri Abhyankar /* backward solve the upper triangular */ 3212f268cba8SShri Abhyankar for (i=n-1; i>=0; i--){ 32133c0119dfSShri Abhyankar v = aa + adiag[i+1]+1; 32143c0119dfSShri Abhyankar vi = aj + adiag[i+1]+1; 3215f268cba8SShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 3216f268cba8SShri Abhyankar sum = tmp[i]; 3217f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3218f268cba8SShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 3219f268cba8SShri Abhyankar } 3220f268cba8SShri Abhyankar 3221f268cba8SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 3222f268cba8SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 32233649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3224f268cba8SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3225f268cba8SShri Abhyankar ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 3226f268cba8SShri Abhyankar PetscFunctionReturn(0); 3227f268cba8SShri Abhyankar } 3228f268cba8SShri Abhyankar 3229f268cba8SShri Abhyankar #undef __FUNCT__ 32301d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ" 3231fe97e370SBarry Smith /* 3232fe97e370SBarry 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 3233fe97e370SBarry Smith */ 32341d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact) 32351d098868SHong Zhang { 32361d098868SHong Zhang Mat B = *fact; 3237599ef60dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b; 3238599ef60dSHong Zhang IS isicol; 3239599ef60dSHong Zhang PetscErrorCode ierr; 32401d098868SHong Zhang const PetscInt *r,*ic; 32411d098868SHong Zhang PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag; 3242dc3a2fd3SHong Zhang PetscInt *bi,*bj,*bdiag,*bdiag_rev; 3243f61a948aSLisandro Dalcin PetscInt row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au; 32441d098868SHong Zhang PetscInt nlnk,*lnk; 32451d098868SHong Zhang PetscBT lnkbt; 3246ace3abfcSBarry Smith PetscBool row_identity,icol_identity; 32478aee2decSHong Zhang MatScalar *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp; 32481d098868SHong Zhang const PetscInt *ics; 32491d098868SHong Zhang PetscInt j,nz,*pj,*bjtmp,k,ncut,*jtmp; 3250182b8fbaSHong Zhang PetscReal dt=info->dt,dtcol=info->dtcol,shift=info->shiftamount; 3251f61a948aSLisandro Dalcin PetscInt dtcount=(PetscInt)info->dtcount,nnz_max; 3252ace3abfcSBarry Smith PetscBool missing; 3253599ef60dSHong Zhang 3254599ef60dSHong Zhang PetscFunctionBegin; 3255f61a948aSLisandro Dalcin 3256f61a948aSLisandro Dalcin if (dt == PETSC_DEFAULT) dt = 0.005; 3257f61a948aSLisandro Dalcin if (dtcol == PETSC_DEFAULT) dtcol = 0.01; /* XXX unused! */ 3258f61a948aSLisandro Dalcin if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax); 3259f61a948aSLisandro Dalcin 32601d098868SHong Zhang /* ------- symbolic factorization, can be reused ---------*/ 32611d098868SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 3262e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 32631d098868SHong Zhang adiag=a->diag; 3264599ef60dSHong Zhang 3265599ef60dSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 3266599ef60dSHong Zhang 3267599ef60dSHong Zhang /* bdiag is location of diagonal in factor */ 32681bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); /* becomes b->diag */ 32691bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag_rev);CHKERRQ(ierr); /* temporary */ 3270599ef60dSHong Zhang 32711d098868SHong Zhang /* allocate row pointers bi */ 32728fc3a347SHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 32731d098868SHong Zhang 32741d098868SHong Zhang /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */ 3275393d3a68SHong Zhang if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */ 32761d098868SHong Zhang nnz_max = ai[n]+2*n*dtcount+2; 32778fc3a347SHong Zhang 32788fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 32798fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr); 3280599ef60dSHong Zhang 3281599ef60dSHong Zhang /* put together the new matrix */ 3282599ef60dSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 3283599ef60dSHong Zhang ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 3284f284f12aSHong Zhang b = (Mat_SeqAIJ*)B->data; 3285599ef60dSHong Zhang b->free_a = PETSC_TRUE; 3286599ef60dSHong Zhang b->free_ij = PETSC_TRUE; 3287599ef60dSHong Zhang b->singlemalloc = PETSC_FALSE; 3288599ef60dSHong Zhang b->a = ba; 3289599ef60dSHong Zhang b->j = bj; 3290599ef60dSHong Zhang b->i = bi; 3291599ef60dSHong Zhang b->diag = bdiag; 3292599ef60dSHong Zhang b->ilen = 0; 3293599ef60dSHong Zhang b->imax = 0; 3294599ef60dSHong Zhang b->row = isrow; 3295599ef60dSHong Zhang b->col = iscol; 3296599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 3297599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 3298599ef60dSHong Zhang b->icol = isicol; 3299599ef60dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3300599ef60dSHong Zhang 33011d098868SHong Zhang ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 33021d098868SHong Zhang b->maxnz = nnz_max; 3303599ef60dSHong Zhang 3304d5f3da31SBarry Smith B->factortype = MAT_FACTOR_ILUDT; 3305f284f12aSHong Zhang B->info.factor_mallocs = 0; 3306f284f12aSHong Zhang B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]); 33071d098868SHong Zhang CHKMEMQ; 33081d098868SHong Zhang /* ------- end of symbolic factorization ---------*/ 3309599ef60dSHong Zhang 3310599ef60dSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 3311599ef60dSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3312599ef60dSHong Zhang ics = ic; 3313599ef60dSHong Zhang 3314599ef60dSHong Zhang /* linked list for storing column indices of the active row */ 3315599ef60dSHong Zhang nlnk = n + 1; 3316599ef60dSHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 33171d098868SHong Zhang 33181d098868SHong Zhang /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */ 33190e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt,&im,n,PetscInt,&jtmp);CHKERRQ(ierr); 33201d098868SHong Zhang /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */ 33210e83c824SBarry Smith ierr = PetscMalloc2(n,MatScalar,&rtmp,n,MatScalar,&vtmp);CHKERRQ(ierr); 33220e83c824SBarry Smith ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr); 3323599ef60dSHong Zhang 3324599ef60dSHong Zhang bi[0] = 0; 33258fc3a347SHong Zhang bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */ 3326dc3a2fd3SHong Zhang bdiag_rev[n] = bdiag[0]; 33278fc3a347SHong Zhang bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */ 3328599ef60dSHong Zhang for (i=0; i<n; i++) { 3329599ef60dSHong Zhang /* copy initial fill into linked list */ 3330599ef60dSHong Zhang nzi = 0; /* nonzeros for active row i */ 33318369b28dSHong Zhang nzi = ai[r[i]+1] - ai[r[i]]; 3332e32f2f54SBarry Smith if (!nzi) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 33338369b28dSHong Zhang nzi_al = adiag[r[i]] - ai[r[i]]; 33348369b28dSHong Zhang nzi_au = ai[r[i]+1] - adiag[r[i]] -1; 3335599ef60dSHong Zhang ajtmp = aj + ai[r[i]]; 33368369b28dSHong Zhang ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3337599ef60dSHong Zhang 3338599ef60dSHong Zhang /* load in initial (unfactored row) */ 33391d098868SHong Zhang aatmp = a->a + ai[r[i]]; 33408369b28dSHong Zhang for (j=0; j<nzi; j++) { 33411d098868SHong Zhang rtmp[ics[*ajtmp++]] = *aatmp++; 3342599ef60dSHong Zhang } 3343599ef60dSHong Zhang 3344599ef60dSHong Zhang /* add pivot rows into linked list */ 3345599ef60dSHong Zhang row = lnk[n]; 3346599ef60dSHong Zhang while (row < i ) { 33471d098868SHong Zhang nzi_bl = bi[row+1] - bi[row] + 1; 33481d098868SHong Zhang bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */ 33491d098868SHong Zhang ierr = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr); 3350599ef60dSHong Zhang nzi += nlnk; 3351599ef60dSHong Zhang row = lnk[row]; 3352599ef60dSHong Zhang } 3353599ef60dSHong Zhang 33548369b28dSHong Zhang /* copy data from lnk into jtmp, then initialize lnk */ 33558369b28dSHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr); 3356599ef60dSHong Zhang 3357599ef60dSHong Zhang /* numerical factorization */ 33588369b28dSHong Zhang bjtmp = jtmp; 3359599ef60dSHong Zhang row = *bjtmp++; /* 1st pivot row */ 3360599ef60dSHong Zhang while ( row < i ) { 3361599ef60dSHong Zhang pc = rtmp + row; 33623c2a7987SHong Zhang pv = ba + bdiag[row]; /* 1./(diag of the pivot row) */ 33633c2a7987SHong Zhang multiplier = (*pc) * (*pv); 3364599ef60dSHong Zhang *pc = multiplier; 33653c2a7987SHong Zhang if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */ 33661d098868SHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 33671d098868SHong Zhang pv = ba + bdiag[row+1] + 1; 33681d098868SHong Zhang /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */ 33691d098868SHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 33701d098868SHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 3371dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 3372599ef60dSHong Zhang } 3373599ef60dSHong Zhang row = *bjtmp++; 3374599ef60dSHong Zhang } 3375599ef60dSHong Zhang 33768369b28dSHong Zhang /* copy sparse rtmp into contiguous vtmp; separate L and U part */ 33778aee2decSHong Zhang diag_tmp = rtmp[i]; /* save diagonal value - may not needed?? */ 33788369b28dSHong Zhang nzi_bl = 0; j = 0; 33798369b28dSHong Zhang while (jtmp[j] < i){ /* Note: jtmp is sorted */ 33808aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 33818369b28dSHong Zhang nzi_bl++; j++; 33828369b28dSHong Zhang } 33838369b28dSHong Zhang nzi_bu = nzi - nzi_bl -1; 33848369b28dSHong Zhang while (j < nzi){ 33858aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 33868369b28dSHong Zhang j++; 33878369b28dSHong Zhang } 33888369b28dSHong Zhang 3389599ef60dSHong Zhang bjtmp = bj + bi[i]; 3390599ef60dSHong Zhang batmp = ba + bi[i]; 33918369b28dSHong Zhang /* apply level dropping rule to L part */ 33928369b28dSHong Zhang ncut = nzi_al + dtcount; 33938369b28dSHong Zhang if (ncut < nzi_bl){ 33948369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr); 33958369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr); 3396599ef60dSHong Zhang } else { 33978369b28dSHong Zhang ncut = nzi_bl; 33988369b28dSHong Zhang } 33998369b28dSHong Zhang for (j=0; j<ncut; j++){ 34008369b28dSHong Zhang bjtmp[j] = jtmp[j]; 34018369b28dSHong Zhang batmp[j] = vtmp[j]; 34026da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */ 34038369b28dSHong Zhang } 34041d098868SHong Zhang bi[i+1] = bi[i] + ncut; 34058369b28dSHong Zhang nzi = ncut + 1; 34068369b28dSHong Zhang 34078369b28dSHong Zhang /* apply level dropping rule to U part */ 34088369b28dSHong Zhang ncut = nzi_au + dtcount; 34098369b28dSHong Zhang if (ncut < nzi_bu){ 34108369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr); 34118369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr); 34128369b28dSHong Zhang } else { 34138369b28dSHong Zhang ncut = nzi_bu; 34148369b28dSHong Zhang } 34158369b28dSHong Zhang nzi += ncut; 34161d098868SHong Zhang 34171d098868SHong Zhang /* mark bdiagonal */ 34181d098868SHong Zhang bdiag[i+1] = bdiag[i] - (ncut + 1); 3419dc3a2fd3SHong Zhang bdiag_rev[n-i-1] = bdiag[i+1]; 34208fc3a347SHong Zhang bi[2*n - i] = bi[2*n - i +1] - (ncut + 1); 34211d098868SHong Zhang bjtmp = bj + bdiag[i]; 34221d098868SHong Zhang batmp = ba + bdiag[i]; 34231d098868SHong Zhang *bjtmp = i; 34248aee2decSHong Zhang *batmp = diag_tmp; /* rtmp[i]; */ 3425c9c72f8fSHong Zhang if (*batmp == 0.0) { 3426c9c72f8fSHong Zhang *batmp = dt+shift; 3427bd1bc851SBarry Smith /* printf(" row %d add shift %g\n",i,shift); */ 3428c9c72f8fSHong Zhang } 3429b78a477dSHong Zhang *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */ 34306da515a1SHong Zhang /* printf(" (%d,%g),",*bjtmp,*batmp); */ 34311d098868SHong Zhang 34321d098868SHong Zhang bjtmp = bj + bdiag[i+1]+1; 34331d098868SHong Zhang batmp = ba + bdiag[i+1]+1; 34348369b28dSHong Zhang for (k=0; k<ncut; k++){ 34351d098868SHong Zhang bjtmp[k] = jtmp[nzi_bl+1+k]; 34361d098868SHong Zhang batmp[k] = vtmp[nzi_bl+1+k]; 34376da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */ 3438599ef60dSHong Zhang } 34396da515a1SHong Zhang /* printf("\n"); */ 3440599ef60dSHong Zhang 3441599ef60dSHong Zhang im[i] = nzi; /* used by PetscLLAddSortedLU() */ 3442599ef60dSHong Zhang /* 34431d098868SHong Zhang printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]); 34448369b28dSHong Zhang printf(" ----------------------------\n"); 3445599ef60dSHong Zhang */ 34468369b28dSHong Zhang } /* for (i=0; i<n; i++) */ 34471d098868SHong Zhang /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */ 3448e32f2f54SBarry Smith if (bi[n] >= bdiag[n]) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"end of L array %d cannot >= the beginning of U array %d",bi[n],bdiag[n]); 3449599ef60dSHong Zhang 3450599ef60dSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3451599ef60dSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3452599ef60dSHong Zhang 3453599ef60dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 34540e83c824SBarry Smith ierr = PetscFree2(im,jtmp);CHKERRQ(ierr); 34550e83c824SBarry Smith ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr); 34561bfa06eaSJed Brown ierr = PetscFree(bdiag_rev);CHKERRQ(ierr); 3457599ef60dSHong Zhang 3458599ef60dSHong Zhang ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr); 34591d098868SHong Zhang b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n]; 3460599ef60dSHong Zhang 3461599ef60dSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 3462599ef60dSHong Zhang ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr); 3463599ef60dSHong Zhang if (row_identity && icol_identity) { 346435233d99SShri Abhyankar B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 3465599ef60dSHong Zhang } else { 346635233d99SShri Abhyankar B->ops->solve = MatSolve_SeqAIJ; 3467599ef60dSHong Zhang } 3468599ef60dSHong Zhang 34691d098868SHong Zhang B->ops->solveadd = 0; 34701d098868SHong Zhang B->ops->solvetranspose = 0; 34711d098868SHong Zhang B->ops->solvetransposeadd = 0; 34721d098868SHong Zhang B->ops->matsolve = 0; 3473599ef60dSHong Zhang B->assembled = PETSC_TRUE; 3474599ef60dSHong Zhang B->preallocated = PETSC_TRUE; 3475599ef60dSHong Zhang PetscFunctionReturn(0); 3476599ef60dSHong Zhang } 3477599ef60dSHong Zhang 34783c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */ 34793c2a7987SHong Zhang #undef __FUNCT__ 34803c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ" 3481fe97e370SBarry Smith /* 3482fe97e370SBarry 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 3483fe97e370SBarry Smith */ 3484fe97e370SBarry Smith 34853c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info) 34863c2a7987SHong Zhang { 34873c2a7987SHong Zhang PetscErrorCode ierr; 34883c2a7987SHong Zhang 34893c2a7987SHong Zhang PetscFunctionBegin; 34903c2a7987SHong Zhang ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr); 34913c2a7987SHong Zhang PetscFunctionReturn(0); 34923c2a7987SHong Zhang } 34933c2a7987SHong Zhang 34943c2a7987SHong Zhang /* 34953c2a7987SHong Zhang same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors 34963c2a7987SHong Zhang - intend to replace existing MatLUFactorNumeric_SeqAIJ() 34973c2a7987SHong Zhang */ 34983c2a7987SHong Zhang #undef __FUNCT__ 34993c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ" 3500fe97e370SBarry Smith /* 3501fe97e370SBarry 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 3502fe97e370SBarry Smith */ 3503fe97e370SBarry Smith 35043c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info) 35053c2a7987SHong Zhang { 35063c2a7987SHong Zhang Mat C=fact; 35073c2a7987SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 35083c2a7987SHong Zhang IS isrow = b->row,isicol = b->icol; 35093c2a7987SHong Zhang PetscErrorCode ierr; 35103c2a7987SHong Zhang const PetscInt *r,*ic,*ics; 3511b78a477dSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 3512b78a477dSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj; 35133c2a7987SHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 3514182b8fbaSHong Zhang PetscReal dt=info->dt,shift=info->shiftamount; 3515ace3abfcSBarry Smith PetscBool row_identity, col_identity; 35163c2a7987SHong Zhang 35173c2a7987SHong Zhang PetscFunctionBegin; 35183c2a7987SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 35193c2a7987SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3520b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 35213c2a7987SHong Zhang ics = ic; 35223c2a7987SHong Zhang 35233c2a7987SHong Zhang for (i=0; i<n; i++){ 3524b78a477dSHong Zhang /* initialize rtmp array */ 3525b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of nozeros in L(i,:) */ 35263c2a7987SHong Zhang bjtmp = bj + bi[i]; 3527b78a477dSHong Zhang for (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0; 3528b78a477dSHong Zhang rtmp[i] = 0.0; 3529b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */ 3530b78a477dSHong Zhang bjtmp = bj + bdiag[i+1] + 1; 3531b78a477dSHong Zhang for (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0; 35323c2a7987SHong Zhang 3533b78a477dSHong Zhang /* load in initial unfactored row of A */ 35346da515a1SHong Zhang /* printf("row %d\n",i); */ 35353c2a7987SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 35363c2a7987SHong Zhang ajtmp = aj + ai[r[i]]; 35373c2a7987SHong Zhang v = aa + ai[r[i]]; 35383c2a7987SHong Zhang for (j=0; j<nz; j++) { 3539b78a477dSHong Zhang rtmp[ics[*ajtmp++]] = v[j]; 35406da515a1SHong Zhang /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */ 35413c2a7987SHong Zhang } 35426da515a1SHong Zhang /* printf("\n"); */ 35433c2a7987SHong Zhang 3544b78a477dSHong Zhang /* numerical factorization */ 3545b78a477dSHong Zhang bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */ 3546b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of entries in L(i,:) */ 3547b78a477dSHong Zhang k = 0; 3548b78a477dSHong Zhang while (k < nzl){ 35493c2a7987SHong Zhang row = *bjtmp++; 35506da515a1SHong Zhang /* printf(" prow %d\n",row); */ 35513c2a7987SHong Zhang pc = rtmp + row; 3552b78a477dSHong Zhang pv = b->a + bdiag[row]; /* 1./(diag of the pivot row) */ 3553b78a477dSHong Zhang multiplier = (*pc) * (*pv); 35543c2a7987SHong Zhang *pc = multiplier; 3555b78a477dSHong Zhang if (PetscAbsScalar(multiplier) > dt){ 3556b78a477dSHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 3557b78a477dSHong Zhang pv = b->a + bdiag[row+1] + 1; 3558b78a477dSHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 3559b78a477dSHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 35606da515a1SHong Zhang /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */ 35613c2a7987SHong Zhang } 3562b78a477dSHong Zhang k++; 35633c2a7987SHong Zhang } 35643c2a7987SHong Zhang 3565b78a477dSHong Zhang /* finished row so stick it into b->a */ 3566b78a477dSHong Zhang /* L-part */ 3567b78a477dSHong Zhang pv = b->a + bi[i] ; 3568b78a477dSHong Zhang pj = bj + bi[i] ; 3569b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; 3570b78a477dSHong Zhang for (j=0; j<nzl; j++) { 3571b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 35726da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 35733c2a7987SHong Zhang } 3574b78a477dSHong Zhang 3575b78a477dSHong Zhang /* diagonal: invert diagonal entries for simplier triangular solves */ 3576b78a477dSHong Zhang if (rtmp[i] == 0.0) rtmp[i] = dt+shift; 3577b78a477dSHong Zhang b->a[bdiag[i]] = 1.0/rtmp[i]; 35786da515a1SHong Zhang /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */ 3579b78a477dSHong Zhang 3580b78a477dSHong Zhang /* U-part */ 3581b78a477dSHong Zhang pv = b->a + bdiag[i+1] + 1; 3582b78a477dSHong Zhang pj = bj + bdiag[i+1] + 1; 3583b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1] - 1; 3584b78a477dSHong Zhang for (j=0; j<nzu; j++) { 3585b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 35866da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 3587b78a477dSHong Zhang } 35886da515a1SHong Zhang /* printf("\n"); */ 3589b78a477dSHong Zhang } 3590b78a477dSHong Zhang 35913c2a7987SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 35923c2a7987SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 35933c2a7987SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3594b78a477dSHong Zhang 35953c2a7987SHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 35963c2a7987SHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 35973c2a7987SHong Zhang if (row_identity && col_identity) { 359835233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 35993c2a7987SHong Zhang } else { 360035233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ; 36013c2a7987SHong Zhang } 3602b78a477dSHong Zhang C->ops->solveadd = 0; 3603b78a477dSHong Zhang C->ops->solvetranspose = 0; 3604b78a477dSHong Zhang C->ops->solvetransposeadd = 0; 3605b78a477dSHong Zhang C->ops->matsolve = 0; 36063c2a7987SHong Zhang C->assembled = PETSC_TRUE; 36073c2a7987SHong Zhang C->preallocated = PETSC_TRUE; 36083c2a7987SHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 36093c2a7987SHong Zhang PetscFunctionReturn(0); 36103c2a7987SHong Zhang } 3611