1b377110cSBarry Smith 2c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/aij.h> 3c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/sbaij.h> 4c6db04a5SJed Brown #include <petscbt.h> 5c6db04a5SJed Brown #include <../src/mat/utils/freespace.h> 63b2fbd54SBarry Smith 7f6a26d55SMatthew Knepley EXTERN_C_BEGIN 84a2ae208SSatish Balay #undef __FUNCT__ 9a7a54a73SBarry Smith #define __FUNCT__ "MatGetOrdering_Flow_SeqAIJ" 108fa24674SBarry Smith /* 118fa24674SBarry Smith Computes an ordering to get most of the large numerical values in the lower triangular part of the matrix 12312e372aSBarry Smith 13312e372aSBarry Smith This code does not work and is not called anywhere. It would be registered with MatOrderingRegisterAll() 148fa24674SBarry Smith */ 15a7a54a73SBarry Smith PetscErrorCode MatGetOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol) 163b2fbd54SBarry Smith { 178fa24674SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)mat->data; 188fa24674SBarry Smith PetscErrorCode ierr; 19889a8dedSBarry Smith PetscInt i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order; 208fa24674SBarry Smith const PetscInt *ai = a->i, *aj = a->j; 218fa24674SBarry Smith const PetscScalar *aa = a->a; 22ace3abfcSBarry Smith PetscBool *done; 23889a8dedSBarry Smith PetscReal best,past = 0,future; 243a40ed3dSBarry Smith 258fa24674SBarry Smith PetscFunctionBegin; 268fa24674SBarry Smith /* pick initial row */ 278fa24674SBarry Smith best = -1; 288fa24674SBarry Smith for (i=0; i<n; i++) { 2975567043SBarry Smith future = 0.0; 308fa24674SBarry Smith for (j=ai[i]; j<ai[i+1]; j++) { 318fa24674SBarry Smith if (aj[j] != i) future += PetscAbsScalar(aa[j]); else past = PetscAbsScalar(aa[j]); 328fa24674SBarry Smith } 338fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 348fa24674SBarry Smith if (past/future > best) { 358fa24674SBarry Smith best = past/future; 368fa24674SBarry Smith current = i; 378fa24674SBarry Smith } 388fa24674SBarry Smith } 398fa24674SBarry Smith 40ace3abfcSBarry Smith ierr = PetscMalloc(n*sizeof(PetscBool),&done);CHKERRQ(ierr); 41ace3abfcSBarry Smith ierr = PetscMemzero(done,n*sizeof(PetscBool));CHKERRQ(ierr); 420e83c824SBarry Smith ierr = PetscMalloc(n*sizeof(PetscInt),&order);CHKERRQ(ierr); 438fa24674SBarry Smith order[0] = current; 448fa24674SBarry Smith for (i=0; i<n-1; i++) { 458fa24674SBarry Smith done[current] = PETSC_TRUE; 468fa24674SBarry Smith best = -1; 478fa24674SBarry Smith /* loop over all neighbors of current pivot */ 488fa24674SBarry Smith for (j=ai[current]; j<ai[current+1]; j++) { 498fa24674SBarry Smith jj = aj[j]; 508fa24674SBarry Smith if (done[jj]) continue; 518fa24674SBarry Smith /* loop over columns of potential next row computing weights for below and above diagonal */ 528fa24674SBarry Smith past = future = 0.0; 538fa24674SBarry Smith for (k=ai[jj]; k<ai[jj+1]; k++) { 548fa24674SBarry Smith kk = aj[k]; 558fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[k]); 568fa24674SBarry Smith else if (kk != jj) future += PetscAbsScalar(aa[k]); 578fa24674SBarry Smith } 588fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 598fa24674SBarry Smith if (past/future > best) { 608fa24674SBarry Smith best = past/future; 618fa24674SBarry Smith newcurrent = jj; 628fa24674SBarry Smith } 638fa24674SBarry Smith } 648fa24674SBarry Smith if (best == -1) { /* no neighbors to select from so select best of all that remain */ 658fa24674SBarry Smith best = -1; 668fa24674SBarry Smith for (k=0; k<n; k++) { 678fa24674SBarry Smith if (done[k]) continue; 6875567043SBarry Smith future = 0.0; 6975567043SBarry Smith past = 0.0; 708fa24674SBarry Smith for (j=ai[k]; j<ai[k+1]; j++) { 718fa24674SBarry Smith kk = aj[j]; 728fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[j]); 738fa24674SBarry Smith else if (kk != k) future += PetscAbsScalar(aa[j]); 748fa24674SBarry Smith } 758fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 768fa24674SBarry Smith if (past/future > best) { 778fa24674SBarry Smith best = past/future; 788fa24674SBarry Smith newcurrent = k; 798fa24674SBarry Smith } 808fa24674SBarry Smith } 818fa24674SBarry Smith } 82e32f2f54SBarry Smith if (current == newcurrent) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"newcurrent cannot be current"); 838fa24674SBarry Smith current = newcurrent; 848fa24674SBarry Smith order[i+1] = current; 858fa24674SBarry Smith } 8670b3c8c7SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,order,PETSC_COPY_VALUES,irow);CHKERRQ(ierr); 878fa24674SBarry Smith *icol = *irow; 888fa24674SBarry Smith ierr = PetscObjectReference((PetscObject)*irow);CHKERRQ(ierr); 898fa24674SBarry Smith ierr = PetscFree(done);CHKERRQ(ierr); 908fa24674SBarry Smith ierr = PetscFree(order);CHKERRQ(ierr); 91d64ed03dSBarry Smith PetscFunctionReturn(0); 923b2fbd54SBarry Smith } 93f6a26d55SMatthew Knepley EXTERN_C_END 943b2fbd54SBarry Smith 95e631078cSBarry Smith EXTERN_C_BEGIN 964a2ae208SSatish Balay #undef __FUNCT__ 97db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc" 98ace3abfcSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscBool *flg) 99db4efbfdSBarry Smith { 100db4efbfdSBarry Smith PetscFunctionBegin; 101db4efbfdSBarry Smith *flg = PETSC_TRUE; 102db4efbfdSBarry Smith PetscFunctionReturn(0); 103db4efbfdSBarry Smith } 104db4efbfdSBarry Smith EXTERN_C_END 105db4efbfdSBarry Smith 106db4efbfdSBarry Smith EXTERN_C_BEGIN 107db4efbfdSBarry Smith #undef __FUNCT__ 108b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc" 109b24902e0SBarry Smith PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B) 110b24902e0SBarry Smith { 111d0f46423SBarry Smith PetscInt n = A->rmap->n; 112b24902e0SBarry Smith PetscErrorCode ierr; 113b24902e0SBarry Smith 114b24902e0SBarry Smith PetscFunctionBegin; 115*891bceaeSHong Zhang #if defined(PETSC_USE_COMPLEX) 116*891bceaeSHong Zhang if (A->hermitian && (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC))SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported"); 117*891bceaeSHong Zhang #endif 118b24902e0SBarry Smith ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr); 119b24902e0SBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 120599ef60dSHong Zhang if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT){ 121b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr); 12235233d99SShri Abhyankar (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ; 12335233d99SShri Abhyankar (*B)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqAIJ; 124e0527adcSJed Brown ierr = MatSetBlockSizes(*B,A->rmap->bs,A->cmap->bs);CHKERRQ(ierr); 125b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 126b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 127b24902e0SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 12835233d99SShri Abhyankar (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqAIJ; 12935233d99SShri Abhyankar (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ; 130e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported"); 131d5f3da31SBarry Smith (*B)->factortype = ftype; 132b24902e0SBarry Smith PetscFunctionReturn(0); 133b24902e0SBarry Smith } 134e631078cSBarry Smith EXTERN_C_END 135b24902e0SBarry Smith 136b24902e0SBarry Smith #undef __FUNCT__ 137ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_inplace" 138ad04f41aSHong Zhang PetscErrorCode MatLUFactorSymbolic_SeqAIJ_inplace(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 139289bc588SBarry Smith { 140416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 141289bc588SBarry Smith IS isicol; 1426849ba73SBarry Smith PetscErrorCode ierr; 1435d0c19d7SBarry Smith const PetscInt *r,*ic; 1445d0c19d7SBarry Smith PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 1451393bc97SHong Zhang PetscInt *bi,*bj,*ajtmp; 1461393bc97SHong Zhang PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 1479e046878SBarry Smith PetscReal f; 1484875ba38SHong Zhang PetscInt nlnk,*lnk,k,**bi_ptr; 149a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1501393bc97SHong Zhang PetscBT lnkbt; 151289bc588SBarry Smith 1523a40ed3dSBarry Smith PetscFunctionBegin; 153e32f2f54SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square"); 1544c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 155ac121b3dSBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 156ac121b3dSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 157289bc588SBarry Smith 158289bc588SBarry Smith /* get new row pointers */ 1591393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 1601393bc97SHong Zhang bi[0] = 0; 1611393bc97SHong Zhang 1621393bc97SHong Zhang /* bdiag is location of diagonal in factor */ 1631393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1641393bc97SHong Zhang bdiag[0] = 0; 1651393bc97SHong Zhang 1661393bc97SHong Zhang /* linked list for storing column indices of the active row */ 1671393bc97SHong Zhang nlnk = n + 1; 1681393bc97SHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1691393bc97SHong Zhang 17035baf27eSSatish Balay ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 1711393bc97SHong Zhang 1721393bc97SHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 173d3d32019SBarry Smith f = info->fill; 174a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1751393bc97SHong Zhang current_space = free_space; 176289bc588SBarry Smith 177289bc588SBarry Smith for (i=0; i<n; i++) { 1781393bc97SHong Zhang /* copy previous fill into linked list */ 179289bc588SBarry Smith nzi = 0; 1801393bc97SHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 181e32f2f54SBarry 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); 1821393bc97SHong Zhang ajtmp = aj + ai[r[i]]; 183afcc9904SHong Zhang ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1841393bc97SHong Zhang nzi += nlnk; 1851393bc97SHong Zhang 1861393bc97SHong Zhang /* add pivot rows into linked list */ 1871393bc97SHong Zhang row = lnk[n]; 1881393bc97SHong Zhang while (row < i) { 189d1170560SHong Zhang nzbd = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */ 190d1170560SHong Zhang ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 191d1170560SHong Zhang ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 1921393bc97SHong Zhang nzi += nlnk; 1931393bc97SHong Zhang row = lnk[row]; 194289bc588SBarry Smith } 1951393bc97SHong Zhang bi[i+1] = bi[i] + nzi; 1961393bc97SHong Zhang im[i] = nzi; 1971393bc97SHong Zhang 1981393bc97SHong Zhang /* mark bdiag */ 1991393bc97SHong Zhang nzbd = 0; 2001393bc97SHong Zhang nnz = nzi; 2011393bc97SHong Zhang k = lnk[n]; 2021393bc97SHong Zhang while (nnz-- && k < i){ 2031393bc97SHong Zhang nzbd++; 2041393bc97SHong Zhang k = lnk[k]; 2051393bc97SHong Zhang } 2061393bc97SHong Zhang bdiag[i] = bi[i] + nzbd; 2071393bc97SHong Zhang 2081393bc97SHong Zhang /* if free space is not available, make more free space */ 2091393bc97SHong Zhang if (current_space->local_remaining<nzi) { 2101393bc97SHong Zhang nnz = (n - i)*nzi; /* estimated and max additional space needed */ 211a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2121393bc97SHong Zhang reallocs++; 2131393bc97SHong Zhang } 2141393bc97SHong Zhang 2151393bc97SHong Zhang /* copy data into free space, then initialize lnk */ 2161393bc97SHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 2171393bc97SHong Zhang bi_ptr[i] = current_space->array; 2181393bc97SHong Zhang current_space->array += nzi; 2191393bc97SHong Zhang current_space->local_used += nzi; 2201393bc97SHong Zhang current_space->local_remaining -= nzi; 221289bc588SBarry Smith } 2226cf91177SBarry Smith #if defined(PETSC_USE_INFO) 223464e8d44SSatish Balay if (ai[n] != 0) { 2241393bc97SHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 225ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 226ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 227ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 228ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 22905bf559cSSatish Balay } else { 230ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 23105bf559cSSatish Balay } 23263ba0a88SBarry Smith #endif 2332fb3ed76SBarry Smith 234898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 235898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2361987afe7SBarry Smith 2371393bc97SHong Zhang /* destroy list of free space and other temporary array(s) */ 2381393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 239a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); 2401393bc97SHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 24135baf27eSSatish Balay ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 242289bc588SBarry Smith 243289bc588SBarry Smith /* put together the new matrix */ 244719d5645SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 245719d5645SBarry Smith ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 246719d5645SBarry Smith b = (Mat_SeqAIJ*)(B)->data; 247e6b907acSBarry Smith b->free_a = PETSC_TRUE; 248e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 2497c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 2501393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 2511393bc97SHong Zhang b->j = bj; 2521393bc97SHong Zhang b->i = bi; 2531393bc97SHong Zhang b->diag = bdiag; 254416022c9SBarry Smith b->ilen = 0; 255416022c9SBarry Smith b->imax = 0; 256416022c9SBarry Smith b->row = isrow; 257416022c9SBarry Smith b->col = iscol; 258c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 259c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 26082bf6240SBarry Smith b->icol = isicol; 26187828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2621393bc97SHong Zhang 2631393bc97SHong Zhang /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 264719d5645SBarry Smith ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 2651393bc97SHong Zhang b->maxnz = b->nz = bi[n] ; 2667fd98bd6SLois Curfman McInnes 267d5f3da31SBarry Smith (B)->factortype = MAT_FACTOR_LU; 268719d5645SBarry Smith (B)->info.factor_mallocs = reallocs; 269719d5645SBarry Smith (B)->info.fill_ratio_given = f; 270703deb49SSatish Balay 2719f7d0b68SHong Zhang if (ai[n]) { 272719d5645SBarry Smith (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 273eea2913aSSatish Balay } else { 274719d5645SBarry Smith (B)->info.fill_ratio_needed = 0.0; 275eea2913aSSatish Balay } 276ad04f41aSHong Zhang (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 277f2344125SBarry Smith if (a->inode.size) { 278f2344125SBarry Smith (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 279f2344125SBarry Smith } 2803a40ed3dSBarry Smith PetscFunctionReturn(0); 281289bc588SBarry Smith } 2821393bc97SHong Zhang 283ce3d78c0SShri Abhyankar #undef __FUNCT__ 28435233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ" 28535233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 286ce3d78c0SShri Abhyankar { 287ce3d78c0SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 288ce3d78c0SShri Abhyankar IS isicol; 289ce3d78c0SShri Abhyankar PetscErrorCode ierr; 290ce3d78c0SShri Abhyankar const PetscInt *r,*ic; 291ce3d78c0SShri Abhyankar PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 292ce3d78c0SShri Abhyankar PetscInt *bi,*bj,*ajtmp; 293ce3d78c0SShri Abhyankar PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 294ce3d78c0SShri Abhyankar PetscReal f; 295ce3d78c0SShri Abhyankar PetscInt nlnk,*lnk,k,**bi_ptr; 296ce3d78c0SShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 297ce3d78c0SShri Abhyankar PetscBT lnkbt; 298ce3d78c0SShri Abhyankar 299ce3d78c0SShri Abhyankar PetscFunctionBegin; 3003898ab1dSSatish Balay /* Uncomment the oldatastruct part only while testing new data structure for MatSolve() */ 301a4be958cSHong Zhang /* 3021a26ff17SHong Zhang PetscBool olddatastruct=PETSC_FALSE; 303acfcf0e5SJed Brown ierr = PetscOptionsGetBool(PETSC_NULL,"-lu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr); 304628f99d7SShri Abhyankar if(olddatastruct){ 305628f99d7SShri Abhyankar ierr = MatLUFactorSymbolic_SeqAIJ_inplace(B,A,isrow,iscol,info);CHKERRQ(ierr); 306628f99d7SShri Abhyankar PetscFunctionReturn(0); 307628f99d7SShri Abhyankar } 308a4be958cSHong Zhang */ 309e32f2f54SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square"); 310ce3d78c0SShri Abhyankar ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 311ce3d78c0SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 312ce3d78c0SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 313ce3d78c0SShri Abhyankar 3141bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 3151bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 3161bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 317a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 3189b48462bSShri Abhyankar 3199b48462bSShri Abhyankar /* linked list for storing column indices of the active row */ 3209b48462bSShri Abhyankar nlnk = n + 1; 3219b48462bSShri Abhyankar ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3229b48462bSShri Abhyankar 3239b48462bSShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 3249b48462bSShri Abhyankar 3259b48462bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 3269b48462bSShri Abhyankar f = info->fill; 3279b48462bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 3289b48462bSShri Abhyankar current_space = free_space; 3299b48462bSShri Abhyankar 3309b48462bSShri Abhyankar for (i=0; i<n; i++) { 3319b48462bSShri Abhyankar /* copy previous fill into linked list */ 3329b48462bSShri Abhyankar nzi = 0; 3339b48462bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 334e32f2f54SBarry 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); 3359b48462bSShri Abhyankar ajtmp = aj + ai[r[i]]; 3369b48462bSShri Abhyankar ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3379b48462bSShri Abhyankar nzi += nlnk; 3389b48462bSShri Abhyankar 3399b48462bSShri Abhyankar /* add pivot rows into linked list */ 3409b48462bSShri Abhyankar row = lnk[n]; 3419b48462bSShri Abhyankar while (row < i){ 3429b48462bSShri Abhyankar nzbd = bdiag[row] + 1; /* num of entries in the row with column index <= row */ 3439b48462bSShri Abhyankar ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 3449b48462bSShri Abhyankar ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 3459b48462bSShri Abhyankar nzi += nlnk; 3469b48462bSShri Abhyankar row = lnk[row]; 3479b48462bSShri Abhyankar } 3489b48462bSShri Abhyankar bi[i+1] = bi[i] + nzi; 3499b48462bSShri Abhyankar im[i] = nzi; 3509b48462bSShri Abhyankar 3519b48462bSShri Abhyankar /* mark bdiag */ 3529b48462bSShri Abhyankar nzbd = 0; 3539b48462bSShri Abhyankar nnz = nzi; 3549b48462bSShri Abhyankar k = lnk[n]; 3559b48462bSShri Abhyankar while (nnz-- && k < i){ 3569b48462bSShri Abhyankar nzbd++; 3579b48462bSShri Abhyankar k = lnk[k]; 3589b48462bSShri Abhyankar } 3599b48462bSShri Abhyankar bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */ 3609b48462bSShri Abhyankar 3619b48462bSShri Abhyankar /* if free space is not available, make more free space */ 3629b48462bSShri Abhyankar if (current_space->local_remaining<nzi) { 3639b48462bSShri Abhyankar nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */ 3649b48462bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 3659b48462bSShri Abhyankar reallocs++; 3669b48462bSShri Abhyankar } 3679b48462bSShri Abhyankar 3689b48462bSShri Abhyankar /* copy data into free space, then initialize lnk */ 3699b48462bSShri Abhyankar ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 3709b48462bSShri Abhyankar bi_ptr[i] = current_space->array; 3719b48462bSShri Abhyankar current_space->array += nzi; 3729b48462bSShri Abhyankar current_space->local_used += nzi; 3739b48462bSShri Abhyankar current_space->local_remaining -= nzi; 3749b48462bSShri Abhyankar } 3759b48462bSShri Abhyankar 3769b48462bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3779b48462bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3789b48462bSShri Abhyankar 3799263d837SHong Zhang /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 3809b48462bSShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 3812ce24eb6SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 3829b48462bSShri Abhyankar ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 3839b48462bSShri Abhyankar ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 3849b48462bSShri Abhyankar 3859b48462bSShri Abhyankar /* put together the new matrix */ 3869b48462bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 3879b48462bSShri Abhyankar ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 3889b48462bSShri Abhyankar b = (Mat_SeqAIJ*)(B)->data; 3899b48462bSShri Abhyankar b->free_a = PETSC_TRUE; 3909b48462bSShri Abhyankar b->free_ij = PETSC_TRUE; 3919b48462bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 3929b48462bSShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 3939b48462bSShri Abhyankar b->j = bj; 3949b48462bSShri Abhyankar b->i = bi; 3959b48462bSShri Abhyankar b->diag = bdiag; 3969b48462bSShri Abhyankar b->ilen = 0; 3979b48462bSShri Abhyankar b->imax = 0; 3989b48462bSShri Abhyankar b->row = isrow; 3999b48462bSShri Abhyankar b->col = iscol; 4009b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 4019b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 4029b48462bSShri Abhyankar b->icol = isicol; 4039b48462bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 4049b48462bSShri Abhyankar 4059b48462bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 4069b48462bSShri Abhyankar ierr = PetscLogObjectMemory(B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 4079b48462bSShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 408d5f3da31SBarry Smith B->factortype = MAT_FACTOR_LU; 409f284f12aSHong Zhang B->info.factor_mallocs = reallocs; 410f284f12aSHong Zhang B->info.fill_ratio_given = f; 4119b48462bSShri Abhyankar 4129b48462bSShri Abhyankar if (ai[n]) { 413f284f12aSHong Zhang B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 4149b48462bSShri Abhyankar } else { 415f284f12aSHong Zhang B->info.fill_ratio_needed = 0.0; 4169b48462bSShri Abhyankar } 4179263d837SHong Zhang #if defined(PETSC_USE_INFO) 4189263d837SHong Zhang if (ai[n] != 0) { 4199263d837SHong Zhang PetscReal af = B->info.fill_ratio_needed; 4209263d837SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 4219263d837SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 4229263d837SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 4239263d837SHong Zhang ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 4249263d837SHong Zhang } else { 4259263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 4269263d837SHong Zhang } 4279263d837SHong Zhang #endif 42835233d99SShri Abhyankar B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 429f2344125SBarry Smith if (a->inode.size) { 430d3ac4fa3SBarry Smith B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 431d3ac4fa3SBarry Smith } 4329b48462bSShri Abhyankar PetscFunctionReturn(0); 4339b48462bSShri Abhyankar } 4349b48462bSShri Abhyankar 4355b5bc046SBarry Smith /* 4365b5bc046SBarry Smith Trouble in factorization, should we dump the original matrix? 4375b5bc046SBarry Smith */ 4385b5bc046SBarry Smith #undef __FUNCT__ 4395b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix" 4405b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A) 4415b5bc046SBarry Smith { 4425b5bc046SBarry Smith PetscErrorCode ierr; 443ace3abfcSBarry Smith PetscBool flg = PETSC_FALSE; 4445b5bc046SBarry Smith 4455b5bc046SBarry Smith PetscFunctionBegin; 446acfcf0e5SJed Brown ierr = PetscOptionsGetBool(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr); 4475b5bc046SBarry Smith if (flg) { 4485b5bc046SBarry Smith PetscViewer viewer; 4495b5bc046SBarry Smith char filename[PETSC_MAX_PATH_LEN]; 4505b5bc046SBarry Smith 4515b5bc046SBarry Smith ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr); 4527adad957SLisandro Dalcin ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 4535b5bc046SBarry Smith ierr = MatView(A,viewer);CHKERRQ(ierr); 4546bf464f9SBarry Smith ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 4555b5bc046SBarry Smith } 4565b5bc046SBarry Smith PetscFunctionReturn(0); 4575b5bc046SBarry Smith } 4585b5bc046SBarry Smith 459c9c72f8fSHong Zhang #undef __FUNCT__ 46035233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ" 46135233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 462c9c72f8fSHong Zhang { 463c9c72f8fSHong Zhang Mat C=B; 464c9c72f8fSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 465c9c72f8fSHong Zhang IS isrow = b->row,isicol = b->icol; 466c9c72f8fSHong Zhang PetscErrorCode ierr; 467c9c72f8fSHong Zhang const PetscInt *r,*ic,*ics; 468bbd65245SShri Abhyankar const PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag; 469bbd65245SShri Abhyankar PetscInt i,j,k,nz,nzL,row,*pj; 470bbd65245SShri Abhyankar const PetscInt *ajtmp,*bjtmp; 471bbd65245SShri Abhyankar MatScalar *rtmp,*pc,multiplier,*pv; 472bbd65245SShri Abhyankar const MatScalar *aa=a->a,*v; 473ace3abfcSBarry Smith PetscBool row_identity,col_identity; 474d90ac83dSHong Zhang FactorShiftCtx sctx; 4754f81c4b7SBarry Smith const PetscInt *ddiag; 476d4ad06f7SHong Zhang PetscReal rs; 477d4ad06f7SHong Zhang MatScalar d; 478d4ad06f7SHong Zhang 479c9c72f8fSHong Zhang PetscFunctionBegin; 480d6e5152cSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 481d90ac83dSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 482d4ad06f7SHong Zhang 483f4db908eSBarry Smith if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 484d4ad06f7SHong Zhang ddiag = a->diag; 485d4ad06f7SHong Zhang sctx.shift_top = info->zeropivot; 486d4ad06f7SHong Zhang for (i=0; i<n; i++) { 487d4ad06f7SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 488d4ad06f7SHong Zhang d = (aa)[ddiag[i]]; 489d4ad06f7SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 490d4ad06f7SHong Zhang v = aa+ai[i]; 491d4ad06f7SHong Zhang nz = ai[i+1] - ai[i]; 492d4ad06f7SHong Zhang for (j=0; j<nz; j++) 493d4ad06f7SHong Zhang rs += PetscAbsScalar(v[j]); 494d4ad06f7SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 495d4ad06f7SHong Zhang } 496d4ad06f7SHong Zhang sctx.shift_top *= 1.1; 497d4ad06f7SHong Zhang sctx.nshift_max = 5; 498d4ad06f7SHong Zhang sctx.shift_lo = 0.; 499d4ad06f7SHong Zhang sctx.shift_hi = 1.; 500d4ad06f7SHong Zhang } 501d4ad06f7SHong Zhang 502c9c72f8fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 503c9c72f8fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 504c9c72f8fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 505c9c72f8fSHong Zhang ics = ic; 506c9c72f8fSHong Zhang 507d4ad06f7SHong Zhang do { 50807b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 509c9c72f8fSHong Zhang for (i=0; i<n; i++){ 510c9c72f8fSHong Zhang /* zero rtmp */ 511c9c72f8fSHong Zhang /* L part */ 512c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 513c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 514c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 515c9c72f8fSHong Zhang 516c9c72f8fSHong Zhang /* U part */ 51762fbd6afSShri Abhyankar nz = bdiag[i]-bdiag[i+1]; 51862fbd6afSShri Abhyankar bjtmp = bj + bdiag[i+1]+1; 519813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 520813a1f2bSShri Abhyankar 521813a1f2bSShri Abhyankar /* load in initial (unfactored row) */ 522813a1f2bSShri Abhyankar nz = ai[r[i]+1] - ai[r[i]]; 523813a1f2bSShri Abhyankar ajtmp = aj + ai[r[i]]; 524813a1f2bSShri Abhyankar v = aa + ai[r[i]]; 525813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 526813a1f2bSShri Abhyankar rtmp[ics[ajtmp[j]]] = v[j]; 527813a1f2bSShri Abhyankar } 528813a1f2bSShri Abhyankar /* ZeropivotApply() */ 52998a93161SHong Zhang rtmp[i] += sctx.shift_amount; /* shift the diagonal of the matrix */ 530813a1f2bSShri Abhyankar 531813a1f2bSShri Abhyankar /* elimination */ 532813a1f2bSShri Abhyankar bjtmp = bj + bi[i]; 533813a1f2bSShri Abhyankar row = *bjtmp++; 534813a1f2bSShri Abhyankar nzL = bi[i+1] - bi[i]; 535f268cba8SShri Abhyankar for(k=0; k < nzL;k++) { 536813a1f2bSShri Abhyankar pc = rtmp + row; 537813a1f2bSShri Abhyankar if (*pc != 0.0) { 538813a1f2bSShri Abhyankar pv = b->a + bdiag[row]; 539813a1f2bSShri Abhyankar multiplier = *pc * (*pv); 540813a1f2bSShri Abhyankar *pc = multiplier; 54162fbd6afSShri Abhyankar pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 54262fbd6afSShri Abhyankar pv = b->a + bdiag[row+1]+1; 54362fbd6afSShri Abhyankar nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 544813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 54528b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 546813a1f2bSShri Abhyankar } 547f268cba8SShri Abhyankar row = *bjtmp++; 548813a1f2bSShri Abhyankar } 549813a1f2bSShri Abhyankar 550813a1f2bSShri Abhyankar /* finished row so stick it into b->a */ 551813a1f2bSShri Abhyankar rs = 0.0; 552813a1f2bSShri Abhyankar /* L part */ 553813a1f2bSShri Abhyankar pv = b->a + bi[i] ; 554813a1f2bSShri Abhyankar pj = b->j + bi[i] ; 555813a1f2bSShri Abhyankar nz = bi[i+1] - bi[i]; 556813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 557813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 558813a1f2bSShri Abhyankar } 559813a1f2bSShri Abhyankar 560813a1f2bSShri Abhyankar /* U part */ 56162fbd6afSShri Abhyankar pv = b->a + bdiag[i+1]+1; 56262fbd6afSShri Abhyankar pj = b->j + bdiag[i+1]+1; 56362fbd6afSShri Abhyankar nz = bdiag[i] - bdiag[i+1]-1; 564813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 565813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 566813a1f2bSShri Abhyankar } 567813a1f2bSShri Abhyankar 568813a1f2bSShri Abhyankar sctx.rs = rs; 569813a1f2bSShri Abhyankar sctx.pv = rtmp[i]; 570d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 57107b50cabSHong Zhang if(sctx.newshift) break; /* break for-loop */ 57207b50cabSHong Zhang rtmp[i] = sctx.pv; /* sctx.pv might be updated in the case of MAT_SHIFT_INBLOCKS */ 573813a1f2bSShri Abhyankar 574813a1f2bSShri Abhyankar /* Mark diagonal and invert diagonal for simplier triangular solves */ 575813a1f2bSShri Abhyankar pv = b->a + bdiag[i]; 576813a1f2bSShri Abhyankar *pv = 1.0/rtmp[i]; 577813a1f2bSShri Abhyankar 578813a1f2bSShri Abhyankar } /* endof for (i=0; i<n; i++){ */ 579813a1f2bSShri Abhyankar 5808ff23777SHong Zhang /* MatPivotRefine() */ 58107b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){ 582813a1f2bSShri Abhyankar /* 583813a1f2bSShri Abhyankar * if no shift in this attempt & shifting & started shifting & can refine, 584813a1f2bSShri Abhyankar * then try lower shift 585813a1f2bSShri Abhyankar */ 586813a1f2bSShri Abhyankar sctx.shift_hi = sctx.shift_fraction; 587813a1f2bSShri Abhyankar sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 588813a1f2bSShri Abhyankar sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 58907b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 590813a1f2bSShri Abhyankar sctx.nshift++; 591813a1f2bSShri Abhyankar } 59207b50cabSHong Zhang } while (sctx.newshift); 593813a1f2bSShri Abhyankar 594813a1f2bSShri Abhyankar ierr = PetscFree(rtmp);CHKERRQ(ierr); 595813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 596813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 597d3ac4fa3SBarry Smith 598813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 599813a1f2bSShri Abhyankar ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 600813a1f2bSShri Abhyankar if (row_identity && col_identity) { 60135233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 602813a1f2bSShri Abhyankar } else { 60335233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ; 604813a1f2bSShri Abhyankar } 60535233d99SShri Abhyankar C->ops->solveadd = MatSolveAdd_SeqAIJ; 60635233d99SShri Abhyankar C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 60735233d99SShri Abhyankar C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 60835233d99SShri Abhyankar C->ops->matsolve = MatMatSolve_SeqAIJ; 609813a1f2bSShri Abhyankar C->assembled = PETSC_TRUE; 610813a1f2bSShri Abhyankar C->preallocated = PETSC_TRUE; 611813a1f2bSShri Abhyankar ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 612813a1f2bSShri Abhyankar 61314d2772eSHong Zhang /* MatShiftView(A,info,&sctx) */ 614813a1f2bSShri Abhyankar if (sctx.nshift){ 615f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 616813a1f2bSShri 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); 617f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 618813a1f2bSShri Abhyankar ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 619f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){ 620915743fcSHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr); 621813a1f2bSShri Abhyankar } 622813a1f2bSShri Abhyankar } 623d3ac4fa3SBarry Smith ierr = Mat_CheckInode_FactorLU(C,PETSC_FALSE);CHKERRQ(ierr); 624813a1f2bSShri Abhyankar PetscFunctionReturn(0); 625813a1f2bSShri Abhyankar } 626813a1f2bSShri Abhyankar 6274a2ae208SSatish Balay #undef __FUNCT__ 628ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_inplace" 629ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info) 630289bc588SBarry Smith { 631719d5645SBarry Smith Mat C=B; 632416022c9SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 63382bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 6346849ba73SBarry Smith PetscErrorCode ierr; 6355d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 636d278edc7SBarry Smith PetscInt nz,row,i,j,n=A->rmap->n,diag; 637d278edc7SBarry Smith const PetscInt *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 638d278edc7SBarry Smith const PetscInt *ajtmp,*bjtmp,*diag_offset = b->diag,*pj; 639d278edc7SBarry Smith MatScalar *pv,*rtmp,*pc,multiplier,d; 640d278edc7SBarry Smith const MatScalar *v,*aa=a->a; 64133f9893dSHong Zhang PetscReal rs=0.0; 642d90ac83dSHong Zhang FactorShiftCtx sctx; 6434f81c4b7SBarry Smith const PetscInt *ddiag; 644ace3abfcSBarry Smith PetscBool row_identity, col_identity; 645289bc588SBarry Smith 6463a40ed3dSBarry Smith PetscFunctionBegin; 647504a3fadSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 648504a3fadSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 649ada7143aSSatish Balay 650f4db908eSBarry Smith if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 65142898933SHong Zhang ddiag = a->diag; 6529f7d0b68SHong Zhang sctx.shift_top = info->zeropivot; 6536cc28720Svictorle for (i=0; i<n; i++) { 6549f95998fSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 6559f7d0b68SHong Zhang d = (aa)[ddiag[i]]; 6561a890ab1SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 6579f7d0b68SHong Zhang v = aa+ai[i]; 6589f7d0b68SHong Zhang nz = ai[i+1] - ai[i]; 659e24cfe64SHong Zhang for (j=0; j<nz; j++) 6601a890ab1SHong Zhang rs += PetscAbsScalar(v[j]); 6611a890ab1SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 6626cc28720Svictorle } 663118f5789SBarry Smith sctx.shift_top *= 1.1; 664d2276718SHong Zhang sctx.nshift_max = 5; 665d2276718SHong Zhang sctx.shift_lo = 0.; 666d2276718SHong Zhang sctx.shift_hi = 1.; 667a0a356efSHong Zhang } 668a0a356efSHong Zhang 669504a3fadSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 670504a3fadSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 671504a3fadSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 672504a3fadSHong Zhang ics = ic; 673504a3fadSHong Zhang 674085256b3SBarry Smith do { 67507b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 676289bc588SBarry Smith for (i=0; i<n; i++){ 677b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 678b3bf805bSHong Zhang bjtmp = bj + bi[i]; 6799f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 680289bc588SBarry Smith 681289bc588SBarry Smith /* load in initial (unfactored row) */ 6829f7d0b68SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 6839f7d0b68SHong Zhang ajtmp = aj + ai[r[i]]; 6849f7d0b68SHong Zhang v = aa + ai[r[i]]; 685085256b3SBarry Smith for (j=0; j<nz; j++) { 686b3bf805bSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 687085256b3SBarry Smith } 688d2276718SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 689289bc588SBarry Smith 690b3bf805bSHong Zhang row = *bjtmp++; 691f2582111SSatish Balay while (row < i) { 6928c37ef55SBarry Smith pc = rtmp + row; 6938c37ef55SBarry Smith if (*pc != 0.0) { 694010a20caSHong Zhang pv = b->a + diag_offset[row]; 695010a20caSHong Zhang pj = b->j + diag_offset[row] + 1; 69635aab85fSBarry Smith multiplier = *pc / *pv++; 6978c37ef55SBarry Smith *pc = multiplier; 698b3bf805bSHong Zhang nz = bi[row+1] - diag_offset[row] - 1; 6999f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 70028b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 701289bc588SBarry Smith } 702b3bf805bSHong Zhang row = *bjtmp++; 703289bc588SBarry Smith } 704416022c9SBarry Smith /* finished row so stick it into b->a */ 705b3bf805bSHong Zhang pv = b->a + bi[i] ; 706b3bf805bSHong Zhang pj = b->j + bi[i] ; 707b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 708b3bf805bSHong Zhang diag = diag_offset[i] - bi[i]; 709e57ff13aSBarry Smith rs = 0.0; 710d3d32019SBarry Smith for (j=0; j<nz; j++) { 7119f7d0b68SHong Zhang pv[j] = rtmp[pj[j]]; 7129f7d0b68SHong Zhang rs += PetscAbsScalar(pv[j]); 713d3d32019SBarry Smith } 714e57ff13aSBarry Smith rs -= PetscAbsScalar(pv[diag]); 715d2276718SHong Zhang 716d2276718SHong Zhang sctx.rs = rs; 717d2276718SHong Zhang sctx.pv = pv[diag]; 718d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 71907b50cabSHong Zhang if (sctx.newshift) break; 720504a3fadSHong Zhang pv[diag] = sctx.pv; 72135aab85fSBarry Smith } 722d2276718SHong Zhang 72307b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 7246cc28720Svictorle /* 7256c037d1bSvictorle * if no shift in this attempt & shifting & started shifting & can refine, 7266cc28720Svictorle * then try lower shift 7276cc28720Svictorle */ 7280481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 7290481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 7300481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 73107b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 732d2276718SHong Zhang sctx.nshift++; 7336cc28720Svictorle } 73407b50cabSHong Zhang } while (sctx.newshift); 7350f11f9aeSSatish Balay 73635aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 73735aab85fSBarry Smith for (i=0; i<n; i++) { 738010a20caSHong Zhang b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]]; 73935aab85fSBarry Smith } 740606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 74178b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 74278b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 743d3ac4fa3SBarry Smith 7448d8c7c43SBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 7458d8c7c43SBarry Smith ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 7468d8c7c43SBarry Smith if (row_identity && col_identity) { 747ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_inplace; 7488d8c7c43SBarry Smith } else { 749ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_inplace; 750db4efbfdSBarry Smith } 751ad04f41aSHong Zhang C->ops->solveadd = MatSolveAdd_SeqAIJ_inplace; 752ad04f41aSHong Zhang C->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 753ad04f41aSHong Zhang C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 754ad04f41aSHong Zhang C->ops->matsolve = MatMatSolve_SeqAIJ_inplace; 755c456f294SBarry Smith C->assembled = PETSC_TRUE; 7565c9eb25fSBarry Smith C->preallocated = PETSC_TRUE; 757d0f46423SBarry Smith ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 758d2276718SHong Zhang if (sctx.nshift){ 759f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 7600481f469SBarry 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); 761f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 762e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 7636cc28720Svictorle } 7640697b6caSHong Zhang } 765d3ac4fa3SBarry Smith (C)->ops->solve = MatSolve_SeqAIJ_inplace; 766d3ac4fa3SBarry Smith (C)->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 767d3ac4fa3SBarry Smith ierr = Mat_CheckInode(C,PETSC_FALSE);CHKERRQ(ierr); 7683a40ed3dSBarry Smith PetscFunctionReturn(0); 769289bc588SBarry Smith } 7700889b5dcSvictorle 771137fb511SHong Zhang /* 772137fb511SHong Zhang This routine implements inplace ILU(0) with row or/and column permutations. 773137fb511SHong Zhang Input: 774137fb511SHong Zhang A - original matrix 775137fb511SHong Zhang Output; 776137fb511SHong Zhang A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i] 777137fb511SHong Zhang a->j (col index) is permuted by the inverse of colperm, then sorted 778137fb511SHong Zhang a->a reordered accordingly with a->j 779137fb511SHong Zhang a->diag (ptr to diagonal elements) is updated. 780137fb511SHong Zhang */ 781137fb511SHong Zhang #undef __FUNCT__ 782137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm" 7830481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info) 784137fb511SHong Zhang { 785b051339dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 786b051339dSHong Zhang IS isrow = a->row,isicol = a->icol; 787137fb511SHong Zhang PetscErrorCode ierr; 7885d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 7895d0c19d7SBarry Smith PetscInt i,j,n=A->rmap->n,*ai=a->i,*aj=a->j; 7905d0c19d7SBarry Smith PetscInt *ajtmp,nz,row; 791b051339dSHong Zhang PetscInt *diag = a->diag,nbdiag,*pj; 792a77337e4SBarry Smith PetscScalar *rtmp,*pc,multiplier,d; 793504a3fadSHong Zhang MatScalar *pv,*v; 794137fb511SHong Zhang PetscReal rs; 795d90ac83dSHong Zhang FactorShiftCtx sctx; 796504a3fadSHong Zhang const MatScalar *aa=a->a,*vtmp; 797137fb511SHong Zhang 798137fb511SHong Zhang PetscFunctionBegin; 799e32f2f54SBarry Smith if (A != B) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address"); 800504a3fadSHong Zhang 801504a3fadSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 802504a3fadSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 803504a3fadSHong Zhang 804504a3fadSHong Zhang if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 805504a3fadSHong Zhang const PetscInt *ddiag = a->diag; 806504a3fadSHong Zhang sctx.shift_top = info->zeropivot; 807504a3fadSHong Zhang for (i=0; i<n; i++) { 808504a3fadSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 809504a3fadSHong Zhang d = (aa)[ddiag[i]]; 810504a3fadSHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 811504a3fadSHong Zhang vtmp = aa+ai[i]; 812504a3fadSHong Zhang nz = ai[i+1] - ai[i]; 813504a3fadSHong Zhang for (j=0; j<nz; j++) 814504a3fadSHong Zhang rs += PetscAbsScalar(vtmp[j]); 815504a3fadSHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 816504a3fadSHong Zhang } 817504a3fadSHong Zhang sctx.shift_top *= 1.1; 818504a3fadSHong Zhang sctx.nshift_max = 5; 819504a3fadSHong Zhang sctx.shift_lo = 0.; 820504a3fadSHong Zhang sctx.shift_hi = 1.; 821504a3fadSHong Zhang } 822504a3fadSHong Zhang 823137fb511SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 824137fb511SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 825137fb511SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr); 826137fb511SHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 827b051339dSHong Zhang ics = ic; 828137fb511SHong Zhang 829504a3fadSHong Zhang #if defined(MV) 83075567043SBarry Smith sctx.shift_top = 0.; 831e60cf9a0SBarry Smith sctx.nshift_max = 0; 83275567043SBarry Smith sctx.shift_lo = 0.; 83375567043SBarry Smith sctx.shift_hi = 0.; 83475567043SBarry Smith sctx.shift_fraction = 0.; 835137fb511SHong Zhang 836f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 83775567043SBarry Smith sctx.shift_top = 0.; 838137fb511SHong Zhang for (i=0; i<n; i++) { 839137fb511SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 840b051339dSHong Zhang d = (a->a)[diag[i]]; 841137fb511SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 842b051339dSHong Zhang v = a->a+ai[i]; 843b051339dSHong Zhang nz = ai[i+1] - ai[i]; 844137fb511SHong Zhang for (j=0; j<nz; j++) 845137fb511SHong Zhang rs += PetscAbsScalar(v[j]); 846137fb511SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 847137fb511SHong Zhang } 848137fb511SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 849137fb511SHong Zhang sctx.shift_top *= 1.1; 850137fb511SHong Zhang sctx.nshift_max = 5; 851137fb511SHong Zhang sctx.shift_lo = 0.; 852137fb511SHong Zhang sctx.shift_hi = 1.; 853137fb511SHong Zhang } 854137fb511SHong Zhang 85575567043SBarry Smith sctx.shift_amount = 0.; 856137fb511SHong Zhang sctx.nshift = 0; 857504a3fadSHong Zhang #endif 858504a3fadSHong Zhang 859137fb511SHong Zhang do { 86007b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 861137fb511SHong Zhang for (i=0; i<n; i++){ 862b051339dSHong Zhang /* load in initial unfactored row */ 863b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 864b051339dSHong Zhang ajtmp = aj + ai[r[i]]; 865b051339dSHong Zhang v = a->a + ai[r[i]]; 866137fb511SHong Zhang /* sort permuted ajtmp and values v accordingly */ 867b051339dSHong Zhang for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]]; 868137fb511SHong Zhang ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr); 869137fb511SHong Zhang 870b051339dSHong Zhang diag[r[i]] = ai[r[i]]; 871137fb511SHong Zhang for (j=0; j<nz; j++) { 872137fb511SHong Zhang rtmp[ajtmp[j]] = v[j]; 873b051339dSHong Zhang if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */ 874137fb511SHong Zhang } 875137fb511SHong Zhang rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 876137fb511SHong Zhang 877b051339dSHong Zhang row = *ajtmp++; 878137fb511SHong Zhang while (row < i) { 879137fb511SHong Zhang pc = rtmp + row; 880137fb511SHong Zhang if (*pc != 0.0) { 881b051339dSHong Zhang pv = a->a + diag[r[row]]; 882b051339dSHong Zhang pj = aj + diag[r[row]] + 1; 883137fb511SHong Zhang 884137fb511SHong Zhang multiplier = *pc / *pv++; 885137fb511SHong Zhang *pc = multiplier; 886b051339dSHong Zhang nz = ai[r[row]+1] - diag[r[row]] - 1; 887b051339dSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 88828b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 889137fb511SHong Zhang } 890b051339dSHong Zhang row = *ajtmp++; 891137fb511SHong Zhang } 892b051339dSHong Zhang /* finished row so overwrite it onto a->a */ 893b051339dSHong Zhang pv = a->a + ai[r[i]] ; 894b051339dSHong Zhang pj = aj + ai[r[i]] ; 895b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 896b051339dSHong Zhang nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */ 897137fb511SHong Zhang 898137fb511SHong Zhang rs = 0.0; 899137fb511SHong Zhang for (j=0; j<nz; j++) { 900b051339dSHong Zhang pv[j] = rtmp[pj[j]]; 901b051339dSHong Zhang if (j != nbdiag) rs += PetscAbsScalar(pv[j]); 902137fb511SHong Zhang } 903137fb511SHong Zhang 904137fb511SHong Zhang sctx.rs = rs; 905b051339dSHong Zhang sctx.pv = pv[nbdiag]; 906d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 90707b50cabSHong Zhang if (sctx.newshift) break; 908504a3fadSHong Zhang pv[nbdiag] = sctx.pv; 909137fb511SHong Zhang } 910137fb511SHong Zhang 91107b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 912137fb511SHong Zhang /* 913137fb511SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 914137fb511SHong Zhang * then try lower shift 915137fb511SHong Zhang */ 9160481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 9170481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 9180481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 91907b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 920137fb511SHong Zhang sctx.nshift++; 921137fb511SHong Zhang } 92207b50cabSHong Zhang } while (sctx.newshift); 923137fb511SHong Zhang 924137fb511SHong Zhang /* invert diagonal entries for simplier triangular solves */ 925137fb511SHong Zhang for (i=0; i<n; i++) { 926b051339dSHong Zhang a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]]; 927137fb511SHong Zhang } 928137fb511SHong Zhang 929137fb511SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 930137fb511SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 931137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 932b051339dSHong Zhang A->ops->solve = MatSolve_SeqAIJ_InplaceWithPerm; 933ad04f41aSHong Zhang A->ops->solveadd = MatSolveAdd_SeqAIJ_inplace; 934ad04f41aSHong Zhang A->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 935ad04f41aSHong Zhang A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 936b051339dSHong Zhang A->assembled = PETSC_TRUE; 9375c9eb25fSBarry Smith A->preallocated = PETSC_TRUE; 938d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr); 939137fb511SHong Zhang if (sctx.nshift){ 940f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 9410481f469SBarry 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); 942f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 943e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 944137fb511SHong Zhang } 945137fb511SHong Zhang } 946137fb511SHong Zhang PetscFunctionReturn(0); 947137fb511SHong Zhang } 948137fb511SHong Zhang 9490a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9504a2ae208SSatish Balay #undef __FUNCT__ 9514a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ" 9520481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info) 953da3a660dSBarry Smith { 954dfbe8321SBarry Smith PetscErrorCode ierr; 955416022c9SBarry Smith Mat C; 956416022c9SBarry Smith 9573a40ed3dSBarry Smith PetscFunctionBegin; 9582692d6eeSBarry Smith ierr = MatGetFactor(A,MATSOLVERPETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr); 959719d5645SBarry Smith ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr); 960719d5645SBarry Smith ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr); 961db4efbfdSBarry Smith A->ops->solve = C->ops->solve; 962db4efbfdSBarry Smith A->ops->solvetranspose = C->ops->solvetranspose; 963eb6b5d47SBarry Smith ierr = MatHeaderMerge(A,C);CHKERRQ(ierr); 96452e6d16bSBarry Smith ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr); 9653a40ed3dSBarry Smith PetscFunctionReturn(0); 966da3a660dSBarry Smith } 9670a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9681d098868SHong Zhang 969889a8dedSBarry Smith 9704a2ae208SSatish Balay #undef __FUNCT__ 971ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_inplace" 972ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx) 9738c37ef55SBarry Smith { 974416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 975416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 9766849ba73SBarry Smith PetscErrorCode ierr; 9775d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 9785d0c19d7SBarry Smith PetscInt nz; 9795d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 980dd6ea824SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 981d9fead3dSBarry Smith const PetscScalar *b; 982dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 9838c37ef55SBarry Smith 9843a40ed3dSBarry Smith PetscFunctionBegin; 9853a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 98691bf9adeSBarry Smith 9873649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 9881ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 989416022c9SBarry Smith tmp = a->solve_work; 9908c37ef55SBarry Smith 9913b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 9923b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 9938c37ef55SBarry Smith 9948c37ef55SBarry Smith /* forward solve the lower triangular */ 9958c37ef55SBarry Smith tmp[0] = b[*r++]; 996010a20caSHong Zhang tmps = tmp; 9978c37ef55SBarry Smith for (i=1; i<n; i++) { 998010a20caSHong Zhang v = aa + ai[i] ; 999010a20caSHong Zhang vi = aj + ai[i] ; 100053e7425aSBarry Smith nz = a->diag[i] - ai[i]; 100153e7425aSBarry Smith sum = b[*r++]; 1002003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10038c37ef55SBarry Smith tmp[i] = sum; 10048c37ef55SBarry Smith } 10058c37ef55SBarry Smith 10068c37ef55SBarry Smith /* backward solve the upper triangular */ 10078c37ef55SBarry Smith for (i=n-1; i>=0; i--){ 1008010a20caSHong Zhang v = aa + a->diag[i] + 1; 1009010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1010416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 10118c37ef55SBarry Smith sum = tmp[i]; 1012003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1013010a20caSHong Zhang x[*c--] = tmp[i] = sum*aa[a->diag[i]]; 10148c37ef55SBarry Smith } 10158c37ef55SBarry Smith 10163b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10173b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10183649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 10191ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1020dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 10213a40ed3dSBarry Smith PetscFunctionReturn(0); 10228c37ef55SBarry Smith } 1023026e39d0SSatish Balay 10242bebee5dSHong Zhang #undef __FUNCT__ 1025ad04f41aSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ_inplace" 1026ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X) 10272bebee5dSHong Zhang { 10282bebee5dSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10292bebee5dSHong Zhang IS iscol = a->col,isrow = a->row; 10302bebee5dSHong Zhang PetscErrorCode ierr; 10315d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10325d0c19d7SBarry Smith PetscInt nz,neq; 10335d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 1034dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1035dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1036ace3abfcSBarry Smith PetscBool bisdense,xisdense; 10372bebee5dSHong Zhang 10382bebee5dSHong Zhang PetscFunctionBegin; 10392bebee5dSHong Zhang if (!n) PetscFunctionReturn(0); 10402bebee5dSHong Zhang 1041251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1042e32f2f54SBarry Smith if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 1043251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1044e32f2f54SBarry Smith if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 1045db4efbfdSBarry Smith 10462bebee5dSHong Zhang ierr = MatGetArray(B,&b);CHKERRQ(ierr); 10472bebee5dSHong Zhang ierr = MatGetArray(X,&x);CHKERRQ(ierr); 10482bebee5dSHong Zhang 10492bebee5dSHong Zhang tmp = a->solve_work; 10502bebee5dSHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10512bebee5dSHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 10522bebee5dSHong Zhang 1053a36b22ccSSatish Balay for (neq=0; neq<B->cmap->n; neq++){ 10542bebee5dSHong Zhang /* forward solve the lower triangular */ 10552bebee5dSHong Zhang tmp[0] = b[r[0]]; 10562bebee5dSHong Zhang tmps = tmp; 10572bebee5dSHong Zhang for (i=1; i<n; i++) { 10582bebee5dSHong Zhang v = aa + ai[i] ; 10592bebee5dSHong Zhang vi = aj + ai[i] ; 10602bebee5dSHong Zhang nz = a->diag[i] - ai[i]; 10612bebee5dSHong Zhang sum = b[r[i]]; 1062003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10632bebee5dSHong Zhang tmp[i] = sum; 10642bebee5dSHong Zhang } 10652bebee5dSHong Zhang /* backward solve the upper triangular */ 10662bebee5dSHong Zhang for (i=n-1; i>=0; i--){ 10672bebee5dSHong Zhang v = aa + a->diag[i] + 1; 10682bebee5dSHong Zhang vi = aj + a->diag[i] + 1; 10692bebee5dSHong Zhang nz = ai[i+1] - a->diag[i] - 1; 10702bebee5dSHong Zhang sum = tmp[i]; 1071003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10722bebee5dSHong Zhang x[c[i]] = tmp[i] = sum*aa[a->diag[i]]; 10732bebee5dSHong Zhang } 10742bebee5dSHong Zhang 10752bebee5dSHong Zhang b += n; 10762bebee5dSHong Zhang x += n; 10772bebee5dSHong Zhang } 10782bebee5dSHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10792bebee5dSHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10802bebee5dSHong Zhang ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 10812bebee5dSHong Zhang ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 1082dc0b31edSSatish Balay ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 10832bebee5dSHong Zhang PetscFunctionReturn(0); 10842bebee5dSHong Zhang } 10852bebee5dSHong Zhang 1086137fb511SHong Zhang #undef __FUNCT__ 108735233d99SShri Abhyankar #define __FUNCT__ "MatMatSolve_SeqAIJ" 108835233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X) 10899bd0847aSShri Abhyankar { 10909bd0847aSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10919bd0847aSShri Abhyankar IS iscol = a->col,isrow = a->row; 10929bd0847aSShri Abhyankar PetscErrorCode ierr; 10939bd0847aSShri Abhyankar PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 10949bd0847aSShri Abhyankar PetscInt nz,neq; 10959bd0847aSShri Abhyankar const PetscInt *rout,*cout,*r,*c; 10969bd0847aSShri Abhyankar PetscScalar *x,*b,*tmp,sum; 10979bd0847aSShri Abhyankar const MatScalar *aa = a->a,*v; 1098ace3abfcSBarry Smith PetscBool bisdense,xisdense; 10999bd0847aSShri Abhyankar 11009bd0847aSShri Abhyankar PetscFunctionBegin; 11019bd0847aSShri Abhyankar if (!n) PetscFunctionReturn(0); 11029bd0847aSShri Abhyankar 1103251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1104e32f2f54SBarry Smith if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 1105251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1106e32f2f54SBarry Smith if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 11079bd0847aSShri Abhyankar 11089bd0847aSShri Abhyankar ierr = MatGetArray(B,&b);CHKERRQ(ierr); 11099bd0847aSShri Abhyankar ierr = MatGetArray(X,&x);CHKERRQ(ierr); 11109bd0847aSShri Abhyankar 11119bd0847aSShri Abhyankar tmp = a->solve_work; 11129bd0847aSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 11139bd0847aSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 11149bd0847aSShri Abhyankar 11159bd0847aSShri Abhyankar for (neq=0; neq<B->cmap->n; neq++){ 11169bd0847aSShri Abhyankar /* forward solve the lower triangular */ 11179bd0847aSShri Abhyankar tmp[0] = b[r[0]]; 11189bd0847aSShri Abhyankar v = aa; 11199bd0847aSShri Abhyankar vi = aj; 11209bd0847aSShri Abhyankar for (i=1; i<n; i++) { 11219bd0847aSShri Abhyankar nz = ai[i+1] - ai[i]; 11229bd0847aSShri Abhyankar sum = b[r[i]]; 11239bd0847aSShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 11249bd0847aSShri Abhyankar tmp[i] = sum; 11259bd0847aSShri Abhyankar v += nz; vi += nz; 11269bd0847aSShri Abhyankar } 11279bd0847aSShri Abhyankar 11289bd0847aSShri Abhyankar /* backward solve the upper triangular */ 11299bd0847aSShri Abhyankar for (i=n-1; i>=0; i--){ 11309bd0847aSShri Abhyankar v = aa + adiag[i+1]+1; 11319bd0847aSShri Abhyankar vi = aj + adiag[i+1]+1; 11329bd0847aSShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 11339bd0847aSShri Abhyankar sum = tmp[i]; 11349bd0847aSShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 11359bd0847aSShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 11369bd0847aSShri Abhyankar } 11379bd0847aSShri Abhyankar 11389bd0847aSShri Abhyankar b += n; 11399bd0847aSShri Abhyankar x += n; 11409bd0847aSShri Abhyankar } 11419bd0847aSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 11429bd0847aSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 11439bd0847aSShri Abhyankar ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 11449bd0847aSShri Abhyankar ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 11459bd0847aSShri Abhyankar ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 11469bd0847aSShri Abhyankar PetscFunctionReturn(0); 11479bd0847aSShri Abhyankar } 11489bd0847aSShri Abhyankar 11499bd0847aSShri Abhyankar #undef __FUNCT__ 1150137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm" 1151137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx) 1152137fb511SHong Zhang { 1153137fb511SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1154137fb511SHong Zhang IS iscol = a->col,isrow = a->row; 1155137fb511SHong Zhang PetscErrorCode ierr; 11565d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 11575d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 11585d0c19d7SBarry Smith PetscInt nz,row; 1159dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1160dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1161137fb511SHong Zhang 1162137fb511SHong Zhang PetscFunctionBegin; 1163137fb511SHong Zhang if (!n) PetscFunctionReturn(0); 1164137fb511SHong Zhang 1165137fb511SHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 1166137fb511SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1167137fb511SHong Zhang tmp = a->solve_work; 1168137fb511SHong Zhang 1169137fb511SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1170137fb511SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1171137fb511SHong Zhang 1172137fb511SHong Zhang /* forward solve the lower triangular */ 1173137fb511SHong Zhang tmp[0] = b[*r++]; 1174137fb511SHong Zhang tmps = tmp; 1175137fb511SHong Zhang for (row=1; row<n; row++) { 1176137fb511SHong Zhang i = rout[row]; /* permuted row */ 1177137fb511SHong Zhang v = aa + ai[i] ; 1178137fb511SHong Zhang vi = aj + ai[i] ; 1179137fb511SHong Zhang nz = a->diag[i] - ai[i]; 1180137fb511SHong Zhang sum = b[*r++]; 1181003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1182137fb511SHong Zhang tmp[row] = sum; 1183137fb511SHong Zhang } 1184137fb511SHong Zhang 1185137fb511SHong Zhang /* backward solve the upper triangular */ 1186137fb511SHong Zhang for (row=n-1; row>=0; row--){ 1187137fb511SHong Zhang i = rout[row]; /* permuted row */ 1188137fb511SHong Zhang v = aa + a->diag[i] + 1; 1189137fb511SHong Zhang vi = aj + a->diag[i] + 1; 1190137fb511SHong Zhang nz = ai[i+1] - a->diag[i] - 1; 1191137fb511SHong Zhang sum = tmp[row]; 1192003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1193137fb511SHong Zhang x[*c--] = tmp[row] = sum*aa[a->diag[i]]; 1194137fb511SHong Zhang } 1195137fb511SHong Zhang 1196137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1197137fb511SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1198137fb511SHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 1199137fb511SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1200dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1201137fb511SHong Zhang PetscFunctionReturn(0); 1202137fb511SHong Zhang } 1203137fb511SHong Zhang 1204930ae53cSBarry Smith /* ----------------------------------------------------------- */ 1205c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/fsolve.h> 12064a2ae208SSatish Balay #undef __FUNCT__ 1207ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_inplace" 1208ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx) 1209930ae53cSBarry Smith { 1210930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 12116849ba73SBarry Smith PetscErrorCode ierr; 1212d0f46423SBarry Smith PetscInt n = A->rmap->n; 1213b377110cSBarry Smith const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag; 1214356650c2SBarry Smith PetscScalar *x; 1215356650c2SBarry Smith const PetscScalar *b; 1216dd6ea824SBarry Smith const MatScalar *aa = a->a; 1217aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 1218356650c2SBarry Smith PetscInt adiag_i,i,nz,ai_i; 1219b377110cSBarry Smith const PetscInt *vi; 1220dd6ea824SBarry Smith const MatScalar *v; 1221dd6ea824SBarry Smith PetscScalar sum; 1222d85afc42SSatish Balay #endif 1223930ae53cSBarry Smith 12243a40ed3dSBarry Smith PetscFunctionBegin; 12253a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 1226930ae53cSBarry Smith 12273649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 12281ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1229930ae53cSBarry Smith 1230aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 12311c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 12321c4feecaSSatish Balay #else 1233930ae53cSBarry Smith /* forward solve the lower triangular */ 1234930ae53cSBarry Smith x[0] = b[0]; 1235930ae53cSBarry Smith for (i=1; i<n; i++) { 1236930ae53cSBarry Smith ai_i = ai[i]; 1237930ae53cSBarry Smith v = aa + ai_i; 1238930ae53cSBarry Smith vi = aj + ai_i; 1239930ae53cSBarry Smith nz = adiag[i] - ai_i; 1240930ae53cSBarry Smith sum = b[i]; 1241003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1242930ae53cSBarry Smith x[i] = sum; 1243930ae53cSBarry Smith } 1244930ae53cSBarry Smith 1245930ae53cSBarry Smith /* backward solve the upper triangular */ 1246930ae53cSBarry Smith for (i=n-1; i>=0; i--){ 1247930ae53cSBarry Smith adiag_i = adiag[i]; 1248d708749eSBarry Smith v = aa + adiag_i + 1; 1249d708749eSBarry Smith vi = aj + adiag_i + 1; 1250930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 1251930ae53cSBarry Smith sum = x[i]; 1252003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1253930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 1254930ae53cSBarry Smith } 12551c4feecaSSatish Balay #endif 1256dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 12573649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 12581ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 12593a40ed3dSBarry Smith PetscFunctionReturn(0); 1260930ae53cSBarry Smith } 1261930ae53cSBarry Smith 12624a2ae208SSatish Balay #undef __FUNCT__ 1263ad04f41aSHong Zhang #define __FUNCT__ "MatSolveAdd_SeqAIJ_inplace" 1264ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx) 1265da3a660dSBarry Smith { 1266416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1267416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 12686849ba73SBarry Smith PetscErrorCode ierr; 126904fbf559SBarry Smith PetscInt i, n = A->rmap->n,j; 12705d0c19d7SBarry Smith PetscInt nz; 127104fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j; 127204fbf559SBarry Smith PetscScalar *x,*tmp,sum; 127304fbf559SBarry Smith const PetscScalar *b; 1274dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1275da3a660dSBarry Smith 12763a40ed3dSBarry Smith PetscFunctionBegin; 127778b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 1278da3a660dSBarry Smith 12793649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 12801ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1281416022c9SBarry Smith tmp = a->solve_work; 1282da3a660dSBarry Smith 12833b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12843b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1285da3a660dSBarry Smith 1286da3a660dSBarry Smith /* forward solve the lower triangular */ 1287da3a660dSBarry Smith tmp[0] = b[*r++]; 1288da3a660dSBarry Smith for (i=1; i<n; i++) { 1289010a20caSHong Zhang v = aa + ai[i] ; 1290010a20caSHong Zhang vi = aj + ai[i] ; 1291416022c9SBarry Smith nz = a->diag[i] - ai[i]; 1292da3a660dSBarry Smith sum = b[*r++]; 129304fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1294da3a660dSBarry Smith tmp[i] = sum; 1295da3a660dSBarry Smith } 1296da3a660dSBarry Smith 1297da3a660dSBarry Smith /* backward solve the upper triangular */ 1298da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1299010a20caSHong Zhang v = aa + a->diag[i] + 1; 1300010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1301416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 1302da3a660dSBarry Smith sum = tmp[i]; 130304fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1304010a20caSHong Zhang tmp[i] = sum*aa[a->diag[i]]; 1305da3a660dSBarry Smith x[*c--] += tmp[i]; 1306da3a660dSBarry Smith } 1307da3a660dSBarry Smith 13083b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13093b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 13103649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 13111ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1312dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1313898183ecSLois Curfman McInnes 13143a40ed3dSBarry Smith PetscFunctionReturn(0); 1315da3a660dSBarry Smith } 131604fbf559SBarry Smith 13174a2ae208SSatish Balay #undef __FUNCT__ 131835233d99SShri Abhyankar #define __FUNCT__ "MatSolveAdd_SeqAIJ" 131935233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 13203c0119dfSShri Abhyankar { 13213c0119dfSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13223c0119dfSShri Abhyankar IS iscol = a->col,isrow = a->row; 13233c0119dfSShri Abhyankar PetscErrorCode ierr; 13243c0119dfSShri Abhyankar PetscInt i, n = A->rmap->n,j; 13253c0119dfSShri Abhyankar PetscInt nz; 13263c0119dfSShri Abhyankar const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 13273c0119dfSShri Abhyankar PetscScalar *x,*tmp,sum; 13283c0119dfSShri Abhyankar const PetscScalar *b; 13293c0119dfSShri Abhyankar const MatScalar *aa = a->a,*v; 13303c0119dfSShri Abhyankar 13313c0119dfSShri Abhyankar PetscFunctionBegin; 13323c0119dfSShri Abhyankar if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 13333c0119dfSShri Abhyankar 13343649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 13353c0119dfSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 13363c0119dfSShri Abhyankar tmp = a->solve_work; 13373c0119dfSShri Abhyankar 13383c0119dfSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13393c0119dfSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 13403c0119dfSShri Abhyankar 13413c0119dfSShri Abhyankar /* forward solve the lower triangular */ 13423c0119dfSShri Abhyankar tmp[0] = b[r[0]]; 13433c0119dfSShri Abhyankar v = aa; 13443c0119dfSShri Abhyankar vi = aj; 13453c0119dfSShri Abhyankar for (i=1; i<n; i++) { 13463c0119dfSShri Abhyankar nz = ai[i+1] - ai[i]; 13473c0119dfSShri Abhyankar sum = b[r[i]]; 13483c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 13493c0119dfSShri Abhyankar tmp[i] = sum; 13503c0119dfSShri Abhyankar v += nz; vi += nz; 13513c0119dfSShri Abhyankar } 13523c0119dfSShri Abhyankar 13533c0119dfSShri Abhyankar /* backward solve the upper triangular */ 13543c0119dfSShri Abhyankar v = aa + adiag[n-1]; 13553c0119dfSShri Abhyankar vi = aj + adiag[n-1]; 13563c0119dfSShri Abhyankar for (i=n-1; i>=0; i--){ 13573c0119dfSShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 13583c0119dfSShri Abhyankar sum = tmp[i]; 13593c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 13603c0119dfSShri Abhyankar tmp[i] = sum*v[nz]; 13613c0119dfSShri Abhyankar x[c[i]] += tmp[i]; 13623c0119dfSShri Abhyankar v += nz+1; vi += nz+1; 13633c0119dfSShri Abhyankar } 13643c0119dfSShri Abhyankar 13653c0119dfSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13663c0119dfSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 13673649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 13683c0119dfSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 13693c0119dfSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 13703c0119dfSShri Abhyankar 13713c0119dfSShri Abhyankar PetscFunctionReturn(0); 13723c0119dfSShri Abhyankar } 13733c0119dfSShri Abhyankar 13743c0119dfSShri Abhyankar #undef __FUNCT__ 1375ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTranspose_SeqAIJ_inplace" 1376ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx) 1377da3a660dSBarry Smith { 1378416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13792235e667SBarry Smith IS iscol = a->col,isrow = a->row; 13806849ba73SBarry Smith PetscErrorCode ierr; 138104fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 138204fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 138304fbf559SBarry Smith PetscInt nz; 138404fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1385dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 138604fbf559SBarry Smith const PetscScalar *b; 1387da3a660dSBarry Smith 13883a40ed3dSBarry Smith PetscFunctionBegin; 13893649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 13901ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1391416022c9SBarry Smith tmp = a->solve_work; 1392da3a660dSBarry Smith 13932235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13942235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1395da3a660dSBarry Smith 1396da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 13972235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1398da3a660dSBarry Smith 1399da3a660dSBarry Smith /* forward solve the U^T */ 1400da3a660dSBarry Smith for (i=0; i<n; i++) { 1401010a20caSHong Zhang v = aa + diag[i] ; 1402010a20caSHong Zhang vi = aj + diag[i] + 1; 1403f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 1404c38d4ed2SBarry Smith s1 = tmp[i]; 1405ef66eb69SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 140604fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1407c38d4ed2SBarry Smith tmp[i] = s1; 1408da3a660dSBarry Smith } 1409da3a660dSBarry Smith 1410da3a660dSBarry Smith /* backward solve the L^T */ 1411da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1412010a20caSHong Zhang v = aa + diag[i] - 1 ; 1413010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1414f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 1415f1af5d2fSBarry Smith s1 = tmp[i]; 141604fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 1417da3a660dSBarry Smith } 1418da3a660dSBarry Smith 1419da3a660dSBarry Smith /* copy tmp into x according to permutation */ 1420591294e4SBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1421da3a660dSBarry Smith 14222235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 14232235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 14243649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 14251ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1426da3a660dSBarry Smith 1427dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 14283a40ed3dSBarry Smith PetscFunctionReturn(0); 1429da3a660dSBarry Smith } 1430da3a660dSBarry Smith 1431d1fa9404SShri Abhyankar #undef __FUNCT__ 143235233d99SShri Abhyankar #define __FUNCT__ "MatSolveTranspose_SeqAIJ" 143335233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 1434d1fa9404SShri Abhyankar { 1435d1fa9404SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1436d1fa9404SShri Abhyankar IS iscol = a->col,isrow = a->row; 1437d1fa9404SShri Abhyankar PetscErrorCode ierr; 1438d1fa9404SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1439d1fa9404SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1440d1fa9404SShri Abhyankar PetscInt nz; 1441d1fa9404SShri Abhyankar PetscScalar *x,*tmp,s1; 1442d1fa9404SShri Abhyankar const MatScalar *aa = a->a,*v; 1443d1fa9404SShri Abhyankar const PetscScalar *b; 1444d1fa9404SShri Abhyankar 1445d1fa9404SShri Abhyankar PetscFunctionBegin; 14463649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1447d1fa9404SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1448d1fa9404SShri Abhyankar tmp = a->solve_work; 1449d1fa9404SShri Abhyankar 1450d1fa9404SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1451d1fa9404SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1452d1fa9404SShri Abhyankar 1453d1fa9404SShri Abhyankar /* copy the b into temp work space according to permutation */ 1454d1fa9404SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1455d1fa9404SShri Abhyankar 1456d1fa9404SShri Abhyankar /* forward solve the U^T */ 1457d1fa9404SShri Abhyankar for (i=0; i<n; i++) { 1458d1fa9404SShri Abhyankar v = aa + adiag[i+1] + 1; 1459d1fa9404SShri Abhyankar vi = aj + adiag[i+1] + 1; 1460d1fa9404SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1461d1fa9404SShri Abhyankar s1 = tmp[i]; 1462d1fa9404SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1463d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1464d1fa9404SShri Abhyankar tmp[i] = s1; 1465d1fa9404SShri Abhyankar } 1466d1fa9404SShri Abhyankar 1467d1fa9404SShri Abhyankar /* backward solve the L^T */ 1468d1fa9404SShri Abhyankar for (i=n-1; i>=0; i--){ 1469d1fa9404SShri Abhyankar v = aa + ai[i]; 1470d1fa9404SShri Abhyankar vi = aj + ai[i]; 1471d1fa9404SShri Abhyankar nz = ai[i+1] - ai[i]; 1472d1fa9404SShri Abhyankar s1 = tmp[i]; 1473d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1474d1fa9404SShri Abhyankar } 1475d1fa9404SShri Abhyankar 1476d1fa9404SShri Abhyankar /* copy tmp into x according to permutation */ 1477d1fa9404SShri Abhyankar for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1478d1fa9404SShri Abhyankar 1479d1fa9404SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1480d1fa9404SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 14813649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1482d1fa9404SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1483d1fa9404SShri Abhyankar 1484d1fa9404SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1485d1fa9404SShri Abhyankar PetscFunctionReturn(0); 1486d1fa9404SShri Abhyankar } 1487d1fa9404SShri Abhyankar 14884a2ae208SSatish Balay #undef __FUNCT__ 1489ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ_inplace" 1490ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx) 1491da3a660dSBarry Smith { 1492416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 14932235e667SBarry Smith IS iscol = a->col,isrow = a->row; 14946849ba73SBarry Smith PetscErrorCode ierr; 149504fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 149604fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 149704fbf559SBarry Smith PetscInt nz; 149804fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1499dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 150004fbf559SBarry Smith const PetscScalar *b; 15016abc6512SBarry Smith 15023a40ed3dSBarry Smith PetscFunctionBegin; 150323bc6035SMatthew Knepley if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 15043649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 15051ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1506416022c9SBarry Smith tmp = a->solve_work; 15076abc6512SBarry Smith 15082235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 15092235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 15106abc6512SBarry Smith 15116abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 15122235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 15136abc6512SBarry Smith 15146abc6512SBarry Smith /* forward solve the U^T */ 15156abc6512SBarry Smith for (i=0; i<n; i++) { 1516010a20caSHong Zhang v = aa + diag[i] ; 1517010a20caSHong Zhang vi = aj + diag[i] + 1; 1518f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 151904fbf559SBarry Smith s1 = tmp[i]; 152004fbf559SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 152104fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 152204fbf559SBarry Smith tmp[i] = s1; 15236abc6512SBarry Smith } 15246abc6512SBarry Smith 15256abc6512SBarry Smith /* backward solve the L^T */ 15266abc6512SBarry Smith for (i=n-1; i>=0; i--){ 1527010a20caSHong Zhang v = aa + diag[i] - 1 ; 1528010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1529f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 153004fbf559SBarry Smith s1 = tmp[i]; 153104fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 15326abc6512SBarry Smith } 15336abc6512SBarry Smith 15346abc6512SBarry Smith /* copy tmp into x according to permutation */ 15356abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 15366abc6512SBarry Smith 15372235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 15382235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 15393649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 15401ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 15416abc6512SBarry Smith 154204fbf559SBarry Smith ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 15433a40ed3dSBarry Smith PetscFunctionReturn(0); 1544da3a660dSBarry Smith } 154504fbf559SBarry Smith 1546533e6011SShri Abhyankar #undef __FUNCT__ 154735233d99SShri Abhyankar #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ" 154835233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 1549533e6011SShri Abhyankar { 1550533e6011SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1551533e6011SShri Abhyankar IS iscol = a->col,isrow = a->row; 1552533e6011SShri Abhyankar PetscErrorCode ierr; 1553533e6011SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1554533e6011SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1555533e6011SShri Abhyankar PetscInt nz; 1556533e6011SShri Abhyankar PetscScalar *x,*tmp,s1; 1557533e6011SShri Abhyankar const MatScalar *aa = a->a,*v; 1558533e6011SShri Abhyankar const PetscScalar *b; 1559533e6011SShri Abhyankar 1560533e6011SShri Abhyankar PetscFunctionBegin; 1561533e6011SShri Abhyankar if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 15623649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1563533e6011SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1564533e6011SShri Abhyankar tmp = a->solve_work; 1565533e6011SShri Abhyankar 1566533e6011SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1567533e6011SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1568533e6011SShri Abhyankar 1569533e6011SShri Abhyankar /* copy the b into temp work space according to permutation */ 1570533e6011SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1571533e6011SShri Abhyankar 1572533e6011SShri Abhyankar /* forward solve the U^T */ 1573533e6011SShri Abhyankar for (i=0; i<n; i++) { 1574533e6011SShri Abhyankar v = aa + adiag[i+1] + 1; 1575533e6011SShri Abhyankar vi = aj + adiag[i+1] + 1; 1576533e6011SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1577533e6011SShri Abhyankar s1 = tmp[i]; 1578533e6011SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1579533e6011SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1580533e6011SShri Abhyankar tmp[i] = s1; 1581533e6011SShri Abhyankar } 1582533e6011SShri Abhyankar 1583533e6011SShri Abhyankar 1584533e6011SShri Abhyankar /* backward solve the L^T */ 1585533e6011SShri Abhyankar for (i=n-1; i>=0; i--){ 1586533e6011SShri Abhyankar v = aa + ai[i] ; 1587533e6011SShri Abhyankar vi = aj + ai[i]; 1588533e6011SShri Abhyankar nz = ai[i+1] - ai[i]; 1589533e6011SShri Abhyankar s1 = tmp[i]; 1590c5e3b2a3SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1591533e6011SShri Abhyankar } 1592533e6011SShri Abhyankar 1593533e6011SShri Abhyankar /* copy tmp into x according to permutation */ 1594533e6011SShri Abhyankar for (i=0; i<n; i++) x[r[i]] += tmp[i]; 1595533e6011SShri Abhyankar 1596533e6011SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1597533e6011SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 15983649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1599533e6011SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1600533e6011SShri Abhyankar 1601533e6011SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1602533e6011SShri Abhyankar PetscFunctionReturn(0); 1603533e6011SShri Abhyankar } 1604533e6011SShri Abhyankar 16059e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 1606b47f5a65SHong Zhang 160709573ac7SBarry Smith extern PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscBool ); 160808480c60SBarry Smith 16098fc3a347SHong Zhang /* 16108a76ed4fSHong Zhang ilu() under revised new data structure. 1611813a1f2bSShri Abhyankar Factored arrays bj and ba are stored as 1612813a1f2bSShri Abhyankar L(0,:), L(1,:), ...,L(n-1,:), U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:) 1613813a1f2bSShri Abhyankar 1614813a1f2bSShri Abhyankar bi=fact->i is an array of size n+1, in which 1615813a1f2bSShri Abhyankar bi+ 1616baabb860SHong Zhang bi[i]: points to 1st entry of L(i,:),i=0,...,n-1 1617baabb860SHong Zhang bi[n]: points to L(n-1,n-1)+1 1618baabb860SHong Zhang 1619813a1f2bSShri Abhyankar bdiag=fact->diag is an array of size n+1,in which 1620baabb860SHong Zhang bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1 1621a24f213cSHong Zhang bdiag[n]: points to entry of U(n-1,0)-1 1622813a1f2bSShri Abhyankar 1623813a1f2bSShri Abhyankar U(i,:) contains bdiag[i] as its last entry, i.e., 1624813a1f2bSShri Abhyankar U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 1625813a1f2bSShri Abhyankar */ 1626813a1f2bSShri Abhyankar #undef __FUNCT__ 162735233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0" 162835233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1629813a1f2bSShri Abhyankar { 1630813a1f2bSShri Abhyankar 1631813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1632813a1f2bSShri Abhyankar PetscErrorCode ierr; 1633bbd65245SShri Abhyankar const PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag; 1634bbd65245SShri Abhyankar PetscInt i,j,k=0,nz,*bi,*bj,*bdiag; 1635ace3abfcSBarry Smith PetscBool missing; 16361df811f5SHong Zhang IS isicol; 1637813a1f2bSShri Abhyankar 1638813a1f2bSShri Abhyankar PetscFunctionBegin; 1639e32f2f54SBarry 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); 16401df811f5SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 1641e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 16421df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 16431df811f5SHong Zhang 1644813a1f2bSShri Abhyankar ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr); 1645813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1646813a1f2bSShri Abhyankar 1647813a1f2bSShri Abhyankar /* allocate matrix arrays for new data structure */ 1648813a1f2bSShri Abhyankar ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,n+1,PetscInt,&b->i);CHKERRQ(ierr); 1649813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1650813a1f2bSShri Abhyankar b->singlemalloc = PETSC_TRUE; 1651813a1f2bSShri Abhyankar if (!b->diag){ 1652813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr); 1653813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1654813a1f2bSShri Abhyankar } 1655813a1f2bSShri Abhyankar bdiag = b->diag; 1656813a1f2bSShri Abhyankar 1657813a1f2bSShri Abhyankar if (n > 0) { 1658813a1f2bSShri Abhyankar ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr); 1659813a1f2bSShri Abhyankar } 1660813a1f2bSShri Abhyankar 1661813a1f2bSShri Abhyankar /* set bi and bj with new data structure */ 1662813a1f2bSShri Abhyankar bi = b->i; 1663813a1f2bSShri Abhyankar bj = b->j; 1664813a1f2bSShri Abhyankar 1665813a1f2bSShri Abhyankar /* L part */ 1666e60cf9a0SBarry Smith bi[0] = 0; 1667813a1f2bSShri Abhyankar for (i=0; i<n; i++){ 1668813a1f2bSShri Abhyankar nz = adiag[i] - ai[i]; 1669813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nz; 1670813a1f2bSShri Abhyankar aj = a->j + ai[i]; 1671813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 1672bbd65245SShri Abhyankar /* *bj = aj[j]; bj++; */ 1673bbd65245SShri Abhyankar bj[k++] = aj[j]; 1674813a1f2bSShri Abhyankar } 1675813a1f2bSShri Abhyankar } 1676813a1f2bSShri Abhyankar 1677813a1f2bSShri Abhyankar /* U part */ 167862fbd6afSShri Abhyankar bdiag[n] = bi[n]-1; 1679813a1f2bSShri Abhyankar for (i=n-1; i>=0; i--){ 1680813a1f2bSShri Abhyankar nz = ai[i+1] - adiag[i] - 1; 1681813a1f2bSShri Abhyankar aj = a->j + adiag[i] + 1; 1682813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 1683bbd65245SShri Abhyankar /* *bj = aj[j]; bj++; */ 1684bbd65245SShri Abhyankar bj[k++] = aj[j]; 1685813a1f2bSShri Abhyankar } 1686813a1f2bSShri Abhyankar /* diag[i] */ 1687bbd65245SShri Abhyankar /* *bj = i; bj++; */ 1688bbd65245SShri Abhyankar bj[k++] = i; 1689a24f213cSHong Zhang bdiag[i] = bdiag[i+1] + nz + 1; 1690813a1f2bSShri Abhyankar } 16911df811f5SHong Zhang 1692d5f3da31SBarry Smith fact->factortype = MAT_FACTOR_ILU; 16931df811f5SHong Zhang fact->info.factor_mallocs = 0; 16941df811f5SHong Zhang fact->info.fill_ratio_given = info->fill; 16951df811f5SHong Zhang fact->info.fill_ratio_needed = 1.0; 169635233d99SShri Abhyankar fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 16971df811f5SHong Zhang 16981df811f5SHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 16991df811f5SHong Zhang b->row = isrow; 17001df811f5SHong Zhang b->col = iscol; 17011df811f5SHong Zhang b->icol = isicol; 17021df811f5SHong Zhang ierr = PetscMalloc((fact->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 17031df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 17041df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1705813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1706813a1f2bSShri Abhyankar } 1707813a1f2bSShri Abhyankar 1708813a1f2bSShri Abhyankar #undef __FUNCT__ 170935233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ" 171035233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1711813a1f2bSShri Abhyankar { 1712813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1713813a1f2bSShri Abhyankar IS isicol; 1714813a1f2bSShri Abhyankar PetscErrorCode ierr; 1715813a1f2bSShri Abhyankar const PetscInt *r,*ic; 17161df811f5SHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j; 1717813a1f2bSShri Abhyankar PetscInt *bi,*cols,nnz,*cols_lvl; 1718813a1f2bSShri Abhyankar PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 1719813a1f2bSShri Abhyankar PetscInt i,levels,diagonal_fill; 1720ace3abfcSBarry Smith PetscBool col_identity,row_identity; 1721813a1f2bSShri Abhyankar PetscReal f; 1722813a1f2bSShri Abhyankar PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 1723813a1f2bSShri Abhyankar PetscBT lnkbt; 1724813a1f2bSShri Abhyankar PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 1725813a1f2bSShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1726813a1f2bSShri Abhyankar PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 1727813a1f2bSShri Abhyankar 1728813a1f2bSShri Abhyankar PetscFunctionBegin; 17293898ab1dSSatish Balay /* Uncomment the old data struct part only while testing new data structure for MatSolve() */ 1730124a3351SHong Zhang /* 1731ace3abfcSBarry Smith PetscBool olddatastruct=PETSC_FALSE; 1732acfcf0e5SJed Brown ierr = PetscOptionsGetBool(PETSC_NULL,"-ilu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr); 173335233d99SShri Abhyankar if(olddatastruct){ 1734ad04f41aSHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_inplace(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1735ad04f41aSHong Zhang PetscFunctionReturn(0); 1736ad04f41aSHong Zhang } 1737124a3351SHong Zhang */ 1738ad04f41aSHong Zhang 1739813a1f2bSShri Abhyankar levels = (PetscInt)info->levels; 1740813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 1741813a1f2bSShri Abhyankar ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 1742813a1f2bSShri Abhyankar 1743813a1f2bSShri Abhyankar if (!levels && row_identity && col_identity) { 1744813a1f2bSShri Abhyankar /* special case: ilu(0) with natural ordering */ 174535233d99SShri Abhyankar ierr = MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1746124a3351SHong Zhang if (a->inode.size) { 1747124a3351SHong Zhang fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 1748124a3351SHong Zhang } 1749813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1750813a1f2bSShri Abhyankar } 1751813a1f2bSShri Abhyankar 1752e32f2f54SBarry 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); 17531df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 1754813a1f2bSShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1755813a1f2bSShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 1756813a1f2bSShri Abhyankar 17571bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 17581bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 17591bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1760e60cf9a0SBarry Smith bi[0] = bdiag[0] = 0; 1761813a1f2bSShri Abhyankar 17620e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr); 1763813a1f2bSShri Abhyankar 1764813a1f2bSShri Abhyankar /* create a linked list for storing column indices of the active row */ 1765813a1f2bSShri Abhyankar nlnk = n + 1; 1766813a1f2bSShri Abhyankar ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1767813a1f2bSShri Abhyankar 1768813a1f2bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 17691df811f5SHong Zhang f = info->fill; 17701df811f5SHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 1771813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1772813a1f2bSShri Abhyankar current_space = free_space; 1773813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 1774813a1f2bSShri Abhyankar current_space_lvl = free_space_lvl; 1775813a1f2bSShri Abhyankar 1776813a1f2bSShri Abhyankar for (i=0; i<n; i++) { 1777813a1f2bSShri Abhyankar nzi = 0; 1778813a1f2bSShri Abhyankar /* copy current row into linked list */ 1779813a1f2bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 1780e32f2f54SBarry 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); 1781813a1f2bSShri Abhyankar cols = aj + ai[r[i]]; 1782813a1f2bSShri Abhyankar lnk[i] = -1; /* marker to indicate if diagonal exists */ 1783813a1f2bSShri Abhyankar ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1784813a1f2bSShri Abhyankar nzi += nlnk; 1785813a1f2bSShri Abhyankar 1786813a1f2bSShri Abhyankar /* make sure diagonal entry is included */ 1787813a1f2bSShri Abhyankar if (diagonal_fill && lnk[i] == -1) { 1788813a1f2bSShri Abhyankar fm = n; 1789813a1f2bSShri Abhyankar while (lnk[fm] < i) fm = lnk[fm]; 1790813a1f2bSShri Abhyankar lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 1791813a1f2bSShri Abhyankar lnk[fm] = i; 1792813a1f2bSShri Abhyankar lnk_lvl[i] = 0; 1793813a1f2bSShri Abhyankar nzi++; dcount++; 1794813a1f2bSShri Abhyankar } 1795813a1f2bSShri Abhyankar 1796813a1f2bSShri Abhyankar /* add pivot rows into the active row */ 1797813a1f2bSShri Abhyankar nzbd = 0; 1798813a1f2bSShri Abhyankar prow = lnk[n]; 1799813a1f2bSShri Abhyankar while (prow < i) { 1800813a1f2bSShri Abhyankar nnz = bdiag[prow]; 1801813a1f2bSShri Abhyankar cols = bj_ptr[prow] + nnz + 1; 1802813a1f2bSShri Abhyankar cols_lvl = bjlvl_ptr[prow] + nnz + 1; 1803813a1f2bSShri Abhyankar nnz = bi[prow+1] - bi[prow] - nnz - 1; 1804813a1f2bSShri Abhyankar ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 1805813a1f2bSShri Abhyankar nzi += nlnk; 1806813a1f2bSShri Abhyankar prow = lnk[prow]; 1807813a1f2bSShri Abhyankar nzbd++; 1808813a1f2bSShri Abhyankar } 1809813a1f2bSShri Abhyankar bdiag[i] = nzbd; 1810813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nzi; 1811813a1f2bSShri Abhyankar 1812813a1f2bSShri Abhyankar /* if free space is not available, make more free space */ 1813813a1f2bSShri Abhyankar if (current_space->local_remaining<nzi) { 1814813a1f2bSShri Abhyankar nnz = 2*nzi*(n - i); /* estimated and max additional space needed */ 1815813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1816813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 1817813a1f2bSShri Abhyankar reallocs++; 1818813a1f2bSShri Abhyankar } 1819813a1f2bSShri Abhyankar 1820813a1f2bSShri Abhyankar /* copy data into free_space and free_space_lvl, then initialize lnk */ 1821813a1f2bSShri Abhyankar ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 1822813a1f2bSShri Abhyankar bj_ptr[i] = current_space->array; 1823813a1f2bSShri Abhyankar bjlvl_ptr[i] = current_space_lvl->array; 1824813a1f2bSShri Abhyankar 1825813a1f2bSShri Abhyankar /* make sure the active row i has diagonal entry */ 182665e19b50SBarry 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); 1827813a1f2bSShri Abhyankar 1828813a1f2bSShri Abhyankar current_space->array += nzi; 1829813a1f2bSShri Abhyankar current_space->local_used += nzi; 1830813a1f2bSShri Abhyankar current_space->local_remaining -= nzi; 1831813a1f2bSShri Abhyankar current_space_lvl->array += nzi; 1832813a1f2bSShri Abhyankar current_space_lvl->local_used += nzi; 1833813a1f2bSShri Abhyankar current_space_lvl->local_remaining -= nzi; 1834813a1f2bSShri Abhyankar } 1835813a1f2bSShri Abhyankar 1836813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1837813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 1838813a1f2bSShri Abhyankar 1839813a1f2bSShri Abhyankar /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 18409263d837SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 18412ce24eb6SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 1842813a1f2bSShri Abhyankar 1843813a1f2bSShri Abhyankar ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1844813a1f2bSShri Abhyankar ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 18450e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 1846813a1f2bSShri Abhyankar 1847813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO) 1848813a1f2bSShri Abhyankar { 1849aef85c9fSShri Abhyankar PetscReal af = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 1850813a1f2bSShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 1851813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1852813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 1853813a1f2bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1854813a1f2bSShri Abhyankar if (diagonal_fill) { 1855813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 1856813a1f2bSShri Abhyankar } 1857813a1f2bSShri Abhyankar } 1858813a1f2bSShri Abhyankar #endif 1859813a1f2bSShri Abhyankar 1860813a1f2bSShri Abhyankar /* put together the new matrix */ 1861813a1f2bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1862813a1f2bSShri Abhyankar ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 1863813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1864813a1f2bSShri Abhyankar b->free_a = PETSC_TRUE; 1865813a1f2bSShri Abhyankar b->free_ij = PETSC_TRUE; 1866813a1f2bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 1867f268cba8SShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 1868813a1f2bSShri Abhyankar b->j = bj; 1869813a1f2bSShri Abhyankar b->i = bi; 1870813a1f2bSShri Abhyankar b->diag = bdiag; 1871813a1f2bSShri Abhyankar b->ilen = 0; 1872813a1f2bSShri Abhyankar b->imax = 0; 1873813a1f2bSShri Abhyankar b->row = isrow; 1874813a1f2bSShri Abhyankar b->col = iscol; 1875813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1876813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1877813a1f2bSShri Abhyankar b->icol = isicol; 1878813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1879813a1f2bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. 1880813a1f2bSShri Abhyankar Allocate bdiag, solve_work, new a, new j */ 1881f268cba8SShri Abhyankar ierr = PetscLogObjectMemory(fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 1882f268cba8SShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 1883813a1f2bSShri Abhyankar (fact)->info.factor_mallocs = reallocs; 1884813a1f2bSShri Abhyankar (fact)->info.fill_ratio_given = f; 1885f268cba8SShri Abhyankar (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 188635233d99SShri Abhyankar (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1887f2344125SBarry Smith if (a->inode.size) { 1888f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 1889f2344125SBarry Smith } 1890813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1891813a1f2bSShri Abhyankar } 1892813a1f2bSShri Abhyankar 18936516239fSHong Zhang #undef __FUNCT__ 1894ad04f41aSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_inplace" 1895ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 18966516239fSHong Zhang { 18976516239fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 18986516239fSHong Zhang IS isicol; 18996516239fSHong Zhang PetscErrorCode ierr; 19006516239fSHong Zhang const PetscInt *r,*ic; 19016516239fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 19026516239fSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 19036516239fSHong Zhang PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 19046516239fSHong Zhang PetscInt i,levels,diagonal_fill; 1905ace3abfcSBarry Smith PetscBool col_identity,row_identity; 19066516239fSHong Zhang PetscReal f; 19076516239fSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 19086516239fSHong Zhang PetscBT lnkbt; 19096516239fSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 19106516239fSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 19116516239fSHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 1912ace3abfcSBarry Smith PetscBool missing; 19136516239fSHong Zhang 19146516239fSHong Zhang PetscFunctionBegin; 1915e32f2f54SBarry 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); 19166516239fSHong Zhang f = info->fill; 19176516239fSHong Zhang levels = (PetscInt)info->levels; 19186516239fSHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 19196516239fSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 19206516239fSHong Zhang 19216516239fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 19226516239fSHong Zhang ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 19236516239fSHong Zhang if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */ 1924f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr); 1925ad04f41aSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 1926f2344125SBarry Smith if (a->inode.size) { 1927f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 1928f2344125SBarry Smith } 1929d5f3da31SBarry Smith fact->factortype = MAT_FACTOR_ILU; 1930719d5645SBarry Smith (fact)->info.factor_mallocs = 0; 1931719d5645SBarry Smith (fact)->info.fill_ratio_given = info->fill; 1932719d5645SBarry Smith (fact)->info.fill_ratio_needed = 1.0; 1933719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 19348ef3462fSBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 1935e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 193608480c60SBarry Smith b->row = isrow; 193708480c60SBarry Smith b->col = iscol; 193882bf6240SBarry Smith b->icol = isicol; 1939719d5645SBarry Smith ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1940c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1941c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 19423a40ed3dSBarry Smith PetscFunctionReturn(0); 194308480c60SBarry Smith } 194408480c60SBarry Smith 1945898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1946898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 19479e25ed09SBarry Smith 19481bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 19491bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 19501bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1951a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 19526cf997b0SBarry Smith 19530e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr); 19545a8e39fbSHong Zhang 19555a8e39fbSHong Zhang /* create a linked list for storing column indices of the active row */ 19565a8e39fbSHong Zhang nlnk = n + 1; 19575a8e39fbSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19585a8e39fbSHong Zhang 19595a8e39fbSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 1960a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 19615a8e39fbSHong Zhang current_space = free_space; 1962a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 19635a8e39fbSHong Zhang current_space_lvl = free_space_lvl; 19645a8e39fbSHong Zhang 19655a8e39fbSHong Zhang for (i=0; i<n; i++) { 19665a8e39fbSHong Zhang nzi = 0; 19676cf997b0SBarry Smith /* copy current row into linked list */ 19685a8e39fbSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 1969e32f2f54SBarry 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); 19705a8e39fbSHong Zhang cols = aj + ai[r[i]]; 19715a8e39fbSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 19725a8e39fbSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19735a8e39fbSHong Zhang nzi += nlnk; 19746cf997b0SBarry Smith 19756cf997b0SBarry Smith /* make sure diagonal entry is included */ 19765a8e39fbSHong Zhang if (diagonal_fill && lnk[i] == -1) { 19776cf997b0SBarry Smith fm = n; 19785a8e39fbSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 19795a8e39fbSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 19805a8e39fbSHong Zhang lnk[fm] = i; 19815a8e39fbSHong Zhang lnk_lvl[i] = 0; 19825a8e39fbSHong Zhang nzi++; dcount++; 19836cf997b0SBarry Smith } 19846cf997b0SBarry Smith 19855a8e39fbSHong Zhang /* add pivot rows into the active row */ 19865a8e39fbSHong Zhang nzbd = 0; 19875a8e39fbSHong Zhang prow = lnk[n]; 19885a8e39fbSHong Zhang while (prow < i) { 19895a8e39fbSHong Zhang nnz = bdiag[prow]; 19905a8e39fbSHong Zhang cols = bj_ptr[prow] + nnz + 1; 19915a8e39fbSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 19925a8e39fbSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 19935a8e39fbSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 19945a8e39fbSHong Zhang nzi += nlnk; 19955a8e39fbSHong Zhang prow = lnk[prow]; 19965a8e39fbSHong Zhang nzbd++; 199756beaf04SBarry Smith } 19985a8e39fbSHong Zhang bdiag[i] = nzbd; 19995a8e39fbSHong Zhang bi[i+1] = bi[i] + nzi; 2000ecf371e4SBarry Smith 20015a8e39fbSHong Zhang /* if free space is not available, make more free space */ 20025a8e39fbSHong Zhang if (current_space->local_remaining<nzi) { 20035a8e39fbSHong Zhang nnz = nzi*(n - i); /* estimated and max additional space needed */ 2004a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2005a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 20065a8e39fbSHong Zhang reallocs++; 20075a8e39fbSHong Zhang } 2008ecf371e4SBarry Smith 20095a8e39fbSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 20105a8e39fbSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 20115a8e39fbSHong Zhang bj_ptr[i] = current_space->array; 20125a8e39fbSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 20135a8e39fbSHong Zhang 20145a8e39fbSHong Zhang /* make sure the active row i has diagonal entry */ 201565e19b50SBarry 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); 20165a8e39fbSHong Zhang 20175a8e39fbSHong Zhang current_space->array += nzi; 20185a8e39fbSHong Zhang current_space->local_used += nzi; 20195a8e39fbSHong Zhang current_space->local_remaining -= nzi; 20205a8e39fbSHong Zhang current_space_lvl->array += nzi; 20215a8e39fbSHong Zhang current_space_lvl->local_used += nzi; 20225a8e39fbSHong Zhang current_space_lvl->local_remaining -= nzi; 20239e25ed09SBarry Smith } 20245a8e39fbSHong Zhang 2025898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 2026898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 20275a8e39fbSHong Zhang 20285a8e39fbSHong Zhang /* destroy list of free space and other temporary arrays */ 20295a8e39fbSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 20306516239fSHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */ 20315a8e39fbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2032a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 20330e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 20349e25ed09SBarry Smith 20356cf91177SBarry Smith #if defined(PETSC_USE_INFO) 2036f64065bbSBarry Smith { 20375a8e39fbSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2038ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 2039ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2040ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 2041ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 2042335d9088SBarry Smith if (diagonal_fill) { 2043ae15b995SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 2044335d9088SBarry Smith } 2045f64065bbSBarry Smith } 204663ba0a88SBarry Smith #endif 2047bbb6d6a8SBarry Smith 20489e25ed09SBarry Smith /* put together the new matrix */ 204907dbf95fSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 2050719d5645SBarry Smith ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 2051719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 2052e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2053e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 20547c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 20550e83c824SBarry Smith ierr = PetscMalloc(bi[n]*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 20565a8e39fbSHong Zhang b->j = bj; 20575a8e39fbSHong Zhang b->i = bi; 20585a8e39fbSHong Zhang for (i=0; i<n; i++) bdiag[i] += bi[i]; 20595a8e39fbSHong Zhang b->diag = bdiag; 2060416022c9SBarry Smith b->ilen = 0; 2061416022c9SBarry Smith b->imax = 0; 2062416022c9SBarry Smith b->row = isrow; 2063416022c9SBarry Smith b->col = iscol; 2064c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2065c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 206682bf6240SBarry Smith b->icol = isicol; 206787828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2068416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 20695a8e39fbSHong Zhang Allocate bdiag, solve_work, new a, new j */ 2070719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 20715a8e39fbSHong Zhang b->maxnz = b->nz = bi[n] ; 2072719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2073719d5645SBarry Smith (fact)->info.fill_ratio_given = f; 2074719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2075ad04f41aSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 2076f2344125SBarry Smith if (a->inode.size) { 2077f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 2078d3ac4fa3SBarry Smith } 20793a40ed3dSBarry Smith PetscFunctionReturn(0); 20809e25ed09SBarry Smith } 2081d5d45c9bSBarry Smith 2082a6175056SHong Zhang #undef __FUNCT__ 208335233d99SShri Abhyankar #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ" 208435233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 20855f44c854SHong Zhang { 20865f44c854SHong Zhang Mat C = B; 20875f44c854SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 20885f44c854SHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 20895f44c854SHong Zhang IS ip=b->row,iip = b->icol; 20905f44c854SHong Zhang PetscErrorCode ierr; 20915f44c854SHong Zhang const PetscInt *rip,*riip; 20925f44c854SHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp; 20935f44c854SHong Zhang PetscInt *ai=a->i,*aj=a->j; 20945f44c854SHong Zhang PetscInt k,jmin,jmax,*c2r,*il,col,nexti,ili,nz; 20955f44c854SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 2096ace3abfcSBarry Smith PetscBool perm_identity; 2097d90ac83dSHong Zhang FactorShiftCtx sctx; 209898a93161SHong Zhang PetscReal rs; 209998a93161SHong Zhang MatScalar d,*v; 210098a93161SHong Zhang 21015f44c854SHong Zhang PetscFunctionBegin; 210298a93161SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 2103d90ac83dSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 210498a93161SHong Zhang 2105f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 210698a93161SHong Zhang sctx.shift_top = info->zeropivot; 210798a93161SHong Zhang for (i=0; i<mbs; i++) { 210898a93161SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 210998a93161SHong Zhang d = (aa)[a->diag[i]]; 211098a93161SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 211198a93161SHong Zhang v = aa+ai[i]; 211298a93161SHong Zhang nz = ai[i+1] - ai[i]; 211398a93161SHong Zhang for (j=0; j<nz; j++) 211498a93161SHong Zhang rs += PetscAbsScalar(v[j]); 211598a93161SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 211698a93161SHong Zhang } 211798a93161SHong Zhang sctx.shift_top *= 1.1; 211898a93161SHong Zhang sctx.nshift_max = 5; 211998a93161SHong Zhang sctx.shift_lo = 0.; 212098a93161SHong Zhang sctx.shift_hi = 1.; 212198a93161SHong Zhang } 21225f44c854SHong Zhang 21235f44c854SHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 21245f44c854SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 21255f44c854SHong Zhang 21265f44c854SHong Zhang /* allocate working arrays 21275f44c854SHong Zhang c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col 21285f44c854SHong 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 21295f44c854SHong Zhang */ 21300e83c824SBarry Smith ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&c2r);CHKERRQ(ierr); 21315f44c854SHong Zhang 213298a93161SHong Zhang do { 213307b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 213498a93161SHong Zhang 2135c5546cabSHong Zhang for (i=0; i<mbs; i++) c2r[i] = mbs; 21365f44c854SHong Zhang il[0] = 0; 21375f44c854SHong Zhang 21385f44c854SHong Zhang for (k = 0; k<mbs; k++){ 21395f44c854SHong Zhang /* zero rtmp */ 21405f44c854SHong Zhang nz = bi[k+1] - bi[k]; 21415f44c854SHong Zhang bjtmp = bj + bi[k]; 21425f44c854SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 21435f44c854SHong Zhang 21445f44c854SHong Zhang /* load in initial unfactored row */ 21455f44c854SHong Zhang bval = ba + bi[k]; 21465f44c854SHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 21475f44c854SHong Zhang for (j = jmin; j < jmax; j++){ 21485f44c854SHong Zhang col = riip[aj[j]]; 21495f44c854SHong Zhang if (col >= k){ /* only take upper triangular entry */ 21505f44c854SHong Zhang rtmp[col] = aa[j]; 21515f44c854SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 21525f44c854SHong Zhang } 21535f44c854SHong Zhang } 215498a93161SHong Zhang /* shift the diagonal of the matrix: ZeropivotApply() */ 215598a93161SHong Zhang rtmp[k] += sctx.shift_amount; /* shift the diagonal of the matrix */ 21565f44c854SHong Zhang 21575f44c854SHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 21585f44c854SHong Zhang dk = rtmp[k]; 21595f44c854SHong Zhang i = c2r[k]; /* first row to be added to k_th row */ 21605f44c854SHong Zhang 21615f44c854SHong Zhang while (i < k){ 21625f44c854SHong Zhang nexti = c2r[i]; /* next row to be added to k_th row */ 21635f44c854SHong Zhang 21645f44c854SHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 21655f44c854SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 21665f44c854SHong Zhang uikdi = - ba[ili]*ba[bdiag[i]]; /* diagonal(k) */ 21675f44c854SHong Zhang dk += uikdi*ba[ili]; /* update diag[k] */ 21685f44c854SHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 21695f44c854SHong Zhang 21705f44c854SHong Zhang /* add multiple of row i to k-th row */ 21715f44c854SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 21725f44c854SHong Zhang if (jmin < jmax){ 21735f44c854SHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 21745f44c854SHong Zhang /* update il and c2r for row i */ 21755f44c854SHong Zhang il[i] = jmin; 21765f44c854SHong Zhang j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i; 21775f44c854SHong Zhang } 21785f44c854SHong Zhang i = nexti; 21795f44c854SHong Zhang } 21805f44c854SHong Zhang 21815f44c854SHong Zhang /* copy data into U(k,:) */ 218298a93161SHong Zhang rs = 0.0; 21835f44c854SHong Zhang jmin = bi[k]; jmax = bi[k+1]-1; 21845f44c854SHong Zhang if (jmin < jmax) { 21855f44c854SHong Zhang for (j=jmin; j<jmax; j++){ 218698a93161SHong Zhang col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]); 21875f44c854SHong Zhang } 21885f44c854SHong Zhang /* add the k-th row into il and c2r */ 21895f44c854SHong Zhang il[k] = jmin; 21905f44c854SHong Zhang i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k; 21915f44c854SHong Zhang } 219298a93161SHong Zhang 219398a93161SHong Zhang /* MatPivotCheck() */ 219498a93161SHong Zhang sctx.rs = rs; 219598a93161SHong Zhang sctx.pv = dk; 2196d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 219707b50cabSHong Zhang if(sctx.newshift) break; 219898a93161SHong Zhang dk = sctx.pv; 219998a93161SHong Zhang 220098a93161SHong Zhang ba[bdiag[k]] = 1.0/dk; /* U(k,k) */ 220198a93161SHong Zhang } 220207b50cabSHong Zhang } while (sctx.newshift); 22035f44c854SHong Zhang 22040e83c824SBarry Smith ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr); 22055f44c854SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 22065f44c854SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 22075f44c854SHong Zhang 22085f44c854SHong Zhang ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 22095f44c854SHong Zhang if (perm_identity){ 221035233d99SShri Abhyankar B->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 221135233d99SShri Abhyankar B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 22122169352eSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering; 22132169352eSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering; 22145f44c854SHong Zhang } else { 221535233d99SShri Abhyankar B->ops->solve = MatSolve_SeqSBAIJ_1; 221635233d99SShri Abhyankar B->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 221735233d99SShri Abhyankar B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1; 221835233d99SShri Abhyankar B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1; 22195f44c854SHong Zhang } 22205f44c854SHong Zhang 22215f44c854SHong Zhang C->assembled = PETSC_TRUE; 22225f44c854SHong Zhang C->preallocated = PETSC_TRUE; 22235f44c854SHong Zhang ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 222498a93161SHong Zhang 222598a93161SHong Zhang /* MatPivotView() */ 222698a93161SHong Zhang if (sctx.nshift){ 2227f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 222898a93161SHong 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); 2229f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 223098a93161SHong Zhang ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2231f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){ 2232b2a402fcSHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr); 223398a93161SHong Zhang } 223498a93161SHong Zhang } 22355f44c854SHong Zhang PetscFunctionReturn(0); 22365f44c854SHong Zhang } 22375f44c854SHong Zhang 22385f44c854SHong Zhang #undef __FUNCT__ 22390a3c351aSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_inplace" 22400a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info) 2241a6175056SHong Zhang { 2242719d5645SBarry Smith Mat C = B; 22430968510aSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 22442ed133dbSHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 22459bfd6278SHong Zhang IS ip=b->row,iip = b->icol; 22462ed133dbSHong Zhang PetscErrorCode ierr; 22475d0c19d7SBarry Smith const PetscInt *rip,*riip; 2248c5546cabSHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp; 22492ed133dbSHong Zhang PetscInt *ai=a->i,*aj=a->j; 22502ed133dbSHong Zhang PetscInt k,jmin,jmax,*jl,*il,col,nexti,ili,nz; 2251622e2028SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 2252ace3abfcSBarry Smith PetscBool perm_identity; 22530e95ead3SHong Zhang FactorShiftCtx sctx; 22540e95ead3SHong Zhang PetscReal rs; 22550e95ead3SHong Zhang MatScalar d,*v; 2256d5d45c9bSBarry Smith 2257a6175056SHong Zhang PetscFunctionBegin; 22580e95ead3SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 22590e95ead3SHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 22600e95ead3SHong Zhang 22610e95ead3SHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 22620e95ead3SHong Zhang sctx.shift_top = info->zeropivot; 22630e95ead3SHong Zhang for (i=0; i<mbs; i++) { 22640e95ead3SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 22650e95ead3SHong Zhang d = (aa)[a->diag[i]]; 22660e95ead3SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 22670e95ead3SHong Zhang v = aa+ai[i]; 22680e95ead3SHong Zhang nz = ai[i+1] - ai[i]; 22690e95ead3SHong Zhang for (j=0; j<nz; j++) 22700e95ead3SHong Zhang rs += PetscAbsScalar(v[j]); 22710e95ead3SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 22720e95ead3SHong Zhang } 22730e95ead3SHong Zhang sctx.shift_top *= 1.1; 22740e95ead3SHong Zhang sctx.nshift_max = 5; 22750e95ead3SHong Zhang sctx.shift_lo = 0.; 22760e95ead3SHong Zhang sctx.shift_hi = 1.; 22770e95ead3SHong Zhang } 2278ee45ca4aSHong Zhang 22792ed133dbSHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 22809bfd6278SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 22812ed133dbSHong Zhang 22822ed133dbSHong Zhang /* initialization */ 22830e83c824SBarry Smith ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&jl);CHKERRQ(ierr); 22840e95ead3SHong Zhang 22852ed133dbSHong Zhang do { 228607b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 22870e95ead3SHong Zhang 2288c5546cabSHong Zhang for (i=0; i<mbs; i++) jl[i] = mbs; 2289c5546cabSHong Zhang il[0] = 0; 22902ed133dbSHong Zhang 22912ed133dbSHong Zhang for (k = 0; k<mbs; k++){ 2292c5546cabSHong Zhang /* zero rtmp */ 2293c5546cabSHong Zhang nz = bi[k+1] - bi[k]; 2294c5546cabSHong Zhang bjtmp = bj + bi[k]; 2295c5546cabSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 2296c5546cabSHong Zhang 2297c05c3958SHong Zhang bval = ba + bi[k]; 22982ed133dbSHong Zhang /* initialize k-th row by the perm[k]-th row of A */ 22992ed133dbSHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 23002ed133dbSHong Zhang for (j = jmin; j < jmax; j++){ 23019bfd6278SHong Zhang col = riip[aj[j]]; 23022ed133dbSHong Zhang if (col >= k){ /* only take upper triangular entry */ 23032ed133dbSHong Zhang rtmp[col] = aa[j]; 2304c05c3958SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 23052ed133dbSHong Zhang } 23062ed133dbSHong Zhang } 230739e3d630SHong Zhang /* shift the diagonal of the matrix */ 2308540703acSHong Zhang if (sctx.nshift) rtmp[k] += sctx.shift_amount; 23092ed133dbSHong Zhang 23102ed133dbSHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 23112ed133dbSHong Zhang dk = rtmp[k]; 23122ed133dbSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 23132ed133dbSHong Zhang 23142ed133dbSHong Zhang while (i < k){ 23152ed133dbSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 23162ed133dbSHong Zhang 23172ed133dbSHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 23182ed133dbSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 23192ed133dbSHong Zhang uikdi = - ba[ili]*ba[bi[i]]; /* diagonal(k) */ 23202ed133dbSHong Zhang dk += uikdi*ba[ili]; 23212ed133dbSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 23222ed133dbSHong Zhang 23232ed133dbSHong Zhang /* add multiple of row i to k-th row */ 23242ed133dbSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 23252ed133dbSHong Zhang if (jmin < jmax){ 23262ed133dbSHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 23272ed133dbSHong Zhang /* update il and jl for row i */ 23282ed133dbSHong Zhang il[i] = jmin; 23292ed133dbSHong Zhang j = bj[jmin]; jl[i] = jl[j]; jl[j] = i; 23302ed133dbSHong Zhang } 23312ed133dbSHong Zhang i = nexti; 23322ed133dbSHong Zhang } 23332ed133dbSHong Zhang 23340697b6caSHong Zhang /* shift the diagonals when zero pivot is detected */ 23350697b6caSHong Zhang /* compute rs=sum of abs(off-diagonal) */ 23360697b6caSHong Zhang rs = 0.0; 23373c9e8598SHong Zhang jmin = bi[k]+1; 23383c9e8598SHong Zhang nz = bi[k+1] - jmin; 23393c9e8598SHong Zhang bcol = bj + jmin; 234004fbf559SBarry Smith for (j=0; j<nz; j++) { 234104fbf559SBarry Smith rs += PetscAbsScalar(rtmp[bcol[j]]); 23423c9e8598SHong Zhang } 2343540703acSHong Zhang 2344540703acSHong Zhang sctx.rs = rs; 2345540703acSHong Zhang sctx.pv = dk; 2346d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,k);CHKERRQ(ierr); 234707b50cabSHong Zhang if (sctx.newshift) break; 23480e95ead3SHong Zhang dk = sctx.pv; 23493c9e8598SHong Zhang 23503c9e8598SHong Zhang /* copy data into U(k,:) */ 235139e3d630SHong Zhang ba[bi[k]] = 1.0/dk; /* U(k,k) */ 235239e3d630SHong Zhang jmin = bi[k]+1; jmax = bi[k+1]; 235339e3d630SHong Zhang if (jmin < jmax) { 235439e3d630SHong Zhang for (j=jmin; j<jmax; j++){ 2355c6ae9bfcSHong Zhang col = bj[j]; ba[j] = rtmp[col]; 23563c9e8598SHong Zhang } 235739e3d630SHong Zhang /* add the k-th row into il and jl */ 23583c9e8598SHong Zhang il[k] = jmin; 23593c9e8598SHong Zhang i = bj[jmin]; jl[k] = jl[i]; jl[i] = k; 23603c9e8598SHong Zhang } 23613c9e8598SHong Zhang } 236207b50cabSHong Zhang } while (sctx.newshift); 23630e95ead3SHong Zhang 23640e83c824SBarry Smith ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr); 236539e3d630SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 23669bfd6278SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 2367db4efbfdSBarry Smith 2368db4efbfdSBarry Smith ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 2369db4efbfdSBarry Smith if (perm_identity){ 23700a3c351aSHong Zhang B->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23710a3c351aSHong Zhang B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23720a3c351aSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23730a3c351aSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 2374db4efbfdSBarry Smith } else { 23750a3c351aSHong Zhang B->ops->solve = MatSolve_SeqSBAIJ_1_inplace; 23760a3c351aSHong Zhang B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_inplace; 23770a3c351aSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_inplace; 23780a3c351aSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_inplace; 2379db4efbfdSBarry Smith } 2380db4efbfdSBarry Smith 23813c9e8598SHong Zhang C->assembled = PETSC_TRUE; 23823c9e8598SHong Zhang C->preallocated = PETSC_TRUE; 2383d0f46423SBarry Smith ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 2384540703acSHong Zhang if (sctx.nshift){ 2385f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 23861e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2387f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 23881e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 23890697b6caSHong Zhang } 23903c9e8598SHong Zhang } 23913c9e8598SHong Zhang PetscFunctionReturn(0); 23923c9e8598SHong Zhang } 2393a6175056SHong Zhang 23948a76ed4fSHong Zhang /* 23958a76ed4fSHong Zhang icc() under revised new data structure. 23968a76ed4fSHong Zhang Factored arrays bj and ba are stored as 23978a76ed4fSHong Zhang U(0,:),...,U(i,:),U(n-1,:) 23988a76ed4fSHong Zhang 23998a76ed4fSHong Zhang ui=fact->i is an array of size n+1, in which 24008a76ed4fSHong Zhang ui+ 24018a76ed4fSHong Zhang ui[i]: points to 1st entry of U(i,:),i=0,...,n-1 24028a76ed4fSHong Zhang ui[n]: points to U(n-1,n-1)+1 24038a76ed4fSHong Zhang 24048a76ed4fSHong Zhang udiag=fact->diag is an array of size n,in which 24058a76ed4fSHong Zhang udiag[i]: points to diagonal of U(i,:), i=0,...,n-1 24068a76ed4fSHong Zhang 24078a76ed4fSHong Zhang U(i,:) contains udiag[i] as its last entry, i.e., 24088a76ed4fSHong Zhang U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 24098a76ed4fSHong Zhang */ 24108a76ed4fSHong Zhang 2411a6175056SHong Zhang #undef __FUNCT__ 241235233d99SShri Abhyankar #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ" 241335233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2414a6175056SHong Zhang { 24150968510aSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2416ed59401aSHong Zhang Mat_SeqSBAIJ *b; 2417dfbe8321SBarry Smith PetscErrorCode ierr; 2418ace3abfcSBarry Smith PetscBool perm_identity,missing; 24195f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 24205d0c19d7SBarry Smith const PetscInt *rip,*riip; 2421ed59401aSHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 242258ebbce7SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 24235a8e39fbSHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 2424ed59401aSHong Zhang PetscReal fill=info->fill,levels=info->levels; 2425a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 2426a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 24277a48dd6fSHong Zhang PetscBT lnkbt; 2428b635d36bSHong Zhang IS iperm; 2429a6175056SHong Zhang 2430a6175056SHong Zhang PetscFunctionBegin; 2431e32f2f54SBarry 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); 243258ebbce7SBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 2433e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 2434653a6975SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 2435b635d36bSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 24367a48dd6fSHong Zhang 243739e3d630SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 24385f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 243939e3d630SHong Zhang ui[0] = 0; 244039e3d630SHong Zhang 24418a76ed4fSHong Zhang /* ICC(0) without matrix ordering: simply rearrange column indices */ 2442622e2028SHong Zhang if (!levels && perm_identity) { 24435f44c854SHong Zhang for (i=0; i<am; i++) { 2444c5546cabSHong Zhang ncols = ai[i+1] - a->diag[i]; 2445c5546cabSHong Zhang ui[i+1] = ui[i] + ncols; 2446c5546cabSHong Zhang udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */ 2447dc1e2a3fSHong Zhang } 2448dc1e2a3fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2449dc1e2a3fSHong Zhang cols = uj; 2450dc1e2a3fSHong Zhang for (i=0; i<am; i++) { 24515f44c854SHong Zhang aj = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */ 2452dc1e2a3fSHong Zhang ncols = ai[i+1] - a->diag[i] -1; 24535f44c854SHong Zhang for (j=0; j<ncols; j++) *cols++ = aj[j]; 24545f44c854SHong Zhang *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */ 24555f44c854SHong Zhang } 24565f44c854SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 24575f44c854SHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 24585f44c854SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 24595f44c854SHong Zhang 24605f44c854SHong Zhang /* initialization */ 24615f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 24625f44c854SHong Zhang 24635f44c854SHong Zhang /* jl: linked list for storing indices of the pivot rows 24645f44c854SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 24650e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr); 24665f44c854SHong Zhang for (i=0; i<am; i++){ 24675f44c854SHong Zhang jl[i] = am; il[i] = 0; 24685f44c854SHong Zhang } 24695f44c854SHong Zhang 24705f44c854SHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 24715f44c854SHong Zhang nlnk = am + 1; 24725f44c854SHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 24735f44c854SHong Zhang 247495b5ac22SHong Zhang /* initial FreeSpace size is fill*(ai[am]+am)/2 */ 247595b5ac22SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space);CHKERRQ(ierr); 24765f44c854SHong Zhang current_space = free_space; 247795b5ac22SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space_lvl);CHKERRQ(ierr); 24785f44c854SHong Zhang current_space_lvl = free_space_lvl; 24795f44c854SHong Zhang 24805f44c854SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 24815f44c854SHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 24825f44c854SHong Zhang nzk = 0; 24835f44c854SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2484e32f2f54SBarry 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); 24855f44c854SHong Zhang ncols_upper = 0; 24865f44c854SHong Zhang for (j=0; j<ncols; j++){ 24875f44c854SHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 24885f44c854SHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 24895f44c854SHong Zhang ajtmp[ncols_upper] = i; 24905f44c854SHong Zhang ncols_upper++; 24915f44c854SHong Zhang } 24925f44c854SHong Zhang } 24935f44c854SHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 24945f44c854SHong Zhang nzk += nlnk; 24955f44c854SHong Zhang 24965f44c854SHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 24975f44c854SHong Zhang prow = jl[k]; /* 1st pivot row */ 24985f44c854SHong Zhang 24995f44c854SHong Zhang while (prow < k){ 25005f44c854SHong Zhang nextprow = jl[prow]; 25015f44c854SHong Zhang 25025f44c854SHong Zhang /* merge prow into k-th row */ 25035f44c854SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 25045f44c854SHong Zhang jmax = ui[prow+1]; 25055f44c854SHong Zhang ncols = jmax-jmin; 25065f44c854SHong Zhang i = jmin - ui[prow]; 25075f44c854SHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 25085f44c854SHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 25095f44c854SHong Zhang j = *(uj - 1); 25105f44c854SHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 25115f44c854SHong Zhang nzk += nlnk; 25125f44c854SHong Zhang 25135f44c854SHong Zhang /* update il and jl for prow */ 25145f44c854SHong Zhang if (jmin < jmax){ 25155f44c854SHong Zhang il[prow] = jmin; 25165f44c854SHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 25175f44c854SHong Zhang } 25185f44c854SHong Zhang prow = nextprow; 25195f44c854SHong Zhang } 25205f44c854SHong Zhang 25215f44c854SHong Zhang /* if free space is not available, make more free space */ 25225f44c854SHong Zhang if (current_space->local_remaining<nzk) { 25235f44c854SHong Zhang i = am - k + 1; /* num of unfactored rows */ 2524c6ae9bfcSHong Zhang i *= PetscMin(nzk, i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */ 25255f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 25265f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 25275f44c854SHong Zhang reallocs++; 25285f44c854SHong Zhang } 25295f44c854SHong Zhang 25305f44c854SHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 2531e32f2f54SBarry Smith if (nzk == 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 25325f44c854SHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 25335f44c854SHong Zhang 25345f44c854SHong Zhang /* add the k-th row into il and jl */ 25355f44c854SHong Zhang if (nzk > 1){ 25365f44c854SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 25375f44c854SHong Zhang jl[k] = jl[i]; jl[i] = k; 25385f44c854SHong Zhang il[k] = ui[k] + 1; 25395f44c854SHong Zhang } 25405f44c854SHong Zhang uj_ptr[k] = current_space->array; 25415f44c854SHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 25425f44c854SHong Zhang 25435f44c854SHong Zhang current_space->array += nzk; 25445f44c854SHong Zhang current_space->local_used += nzk; 25455f44c854SHong Zhang current_space->local_remaining -= nzk; 25465f44c854SHong Zhang 25475f44c854SHong Zhang current_space_lvl->array += nzk; 25485f44c854SHong Zhang current_space_lvl->local_used += nzk; 25495f44c854SHong Zhang current_space_lvl->local_remaining -= nzk; 25505f44c854SHong Zhang 25515f44c854SHong Zhang ui[k+1] = ui[k] + nzk; 25525f44c854SHong Zhang } 25535f44c854SHong Zhang 25545f44c854SHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 25555f44c854SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 25560e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 25575f44c854SHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 25585f44c854SHong Zhang 25599263d837SHong Zhang /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */ 25605f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 256135233d99SShri Abhyankar ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */ 25625f44c854SHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 25635f44c854SHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 25645f44c854SHong Zhang 25655f44c854SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 25665f44c854SHong Zhang 25675f44c854SHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 25685f44c854SHong Zhang b = (Mat_SeqSBAIJ*)(fact)->data; 25695f44c854SHong Zhang b->singlemalloc = PETSC_FALSE; 25705f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 25715f44c854SHong Zhang b->j = uj; 25725f44c854SHong Zhang b->i = ui; 25735f44c854SHong Zhang b->diag = udiag; 25747f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 25755f44c854SHong Zhang b->ilen = 0; 25765f44c854SHong Zhang b->imax = 0; 25775f44c854SHong Zhang b->row = perm; 25785f44c854SHong Zhang b->col = perm; 25795f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25805f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25815f44c854SHong Zhang b->icol = iperm; 25825f44c854SHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 25835f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2584d595f711SHong Zhang ierr = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 25855f44c854SHong Zhang b->maxnz = b->nz = ui[am]; 25865f44c854SHong Zhang b->free_a = PETSC_TRUE; 25875f44c854SHong Zhang b->free_ij = PETSC_TRUE; 25885f44c854SHong Zhang 2589f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2590f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 25915f44c854SHong Zhang if (ai[am] != 0) { 25926a69fef8SHong Zhang /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */ 259395b5ac22SHong Zhang fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am); 25945f44c854SHong Zhang } else { 2595f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 25965f44c854SHong Zhang } 25979263d837SHong Zhang #if defined(PETSC_USE_INFO) 25989263d837SHong Zhang if (ai[am] != 0) { 25999263d837SHong Zhang PetscReal af = fact->info.fill_ratio_needed; 26009263d837SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 26019263d837SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 26029263d837SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 26039263d837SHong Zhang } else { 26049263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 26059263d837SHong Zhang } 26069263d837SHong Zhang #endif 260735233d99SShri Abhyankar fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 26085f44c854SHong Zhang PetscFunctionReturn(0); 26095f44c854SHong Zhang } 26105f44c854SHong Zhang 26115f44c854SHong Zhang #undef __FUNCT__ 2612ad04f41aSHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_inplace" 2613ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 26145f44c854SHong Zhang { 26155f44c854SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 26165f44c854SHong Zhang Mat_SeqSBAIJ *b; 26175f44c854SHong Zhang PetscErrorCode ierr; 2618ace3abfcSBarry Smith PetscBool perm_identity,missing; 26195f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 26205f44c854SHong Zhang const PetscInt *rip,*riip; 26215f44c854SHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 26225f44c854SHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 26235f44c854SHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 26245f44c854SHong Zhang PetscReal fill=info->fill,levels=info->levels; 26255f44c854SHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 26265f44c854SHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 26275f44c854SHong Zhang PetscBT lnkbt; 26285f44c854SHong Zhang IS iperm; 26295f44c854SHong Zhang 26305f44c854SHong Zhang PetscFunctionBegin; 2631e32f2f54SBarry 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); 26325f44c854SHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 2633e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 26345f44c854SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 26355f44c854SHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 26365f44c854SHong Zhang 26375f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 26385f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 26395f44c854SHong Zhang ui[0] = 0; 26405f44c854SHong Zhang 26415f44c854SHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 26425f44c854SHong Zhang if (!levels && perm_identity) { 26435f44c854SHong Zhang 26445f44c854SHong Zhang for (i=0; i<am; i++) { 26455f44c854SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 26465f44c854SHong Zhang udiag[i] = ui[i]; 2647ed59401aSHong Zhang } 264839e3d630SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 264939e3d630SHong Zhang cols = uj; 2650ed59401aSHong Zhang for (i=0; i<am; i++) { 2651ed59401aSHong Zhang aj = a->j + a->diag[i]; 265239e3d630SHong Zhang ncols = ui[i+1] - ui[i]; 265339e3d630SHong Zhang for (j=0; j<ncols; j++) *cols++ = *aj++; 2654ed59401aSHong Zhang } 265539e3d630SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 2656b635d36bSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 2657b635d36bSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 2658b635d36bSHong Zhang 26597a48dd6fSHong Zhang /* initialization */ 26605a8e39fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 26617a48dd6fSHong Zhang 26627a48dd6fSHong Zhang /* jl: linked list for storing indices of the pivot rows 26637a48dd6fSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 26640e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr); 26657a48dd6fSHong Zhang for (i=0; i<am; i++){ 26667a48dd6fSHong Zhang jl[i] = am; il[i] = 0; 26677a48dd6fSHong Zhang } 26687a48dd6fSHong Zhang 26697a48dd6fSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 26707a48dd6fSHong Zhang nlnk = am + 1; 26712ed133dbSHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 26727a48dd6fSHong Zhang 26737a48dd6fSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2674a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 26757a48dd6fSHong Zhang current_space = free_space; 2676a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 26777a48dd6fSHong Zhang current_space_lvl = free_space_lvl; 26787a48dd6fSHong Zhang 26797a48dd6fSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 26807a48dd6fSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 26817a48dd6fSHong Zhang nzk = 0; 2682622e2028SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2683e32f2f54SBarry 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); 2684622e2028SHong Zhang ncols_upper = 0; 2685622e2028SHong Zhang for (j=0; j<ncols; j++){ 2686b635d36bSHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 2687b635d36bSHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 26885a8e39fbSHong Zhang ajtmp[ncols_upper] = i; 2689622e2028SHong Zhang ncols_upper++; 2690622e2028SHong Zhang } 2691622e2028SHong Zhang } 2692b635d36bSHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 26937a48dd6fSHong Zhang nzk += nlnk; 26947a48dd6fSHong Zhang 26957a48dd6fSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 26967a48dd6fSHong Zhang prow = jl[k]; /* 1st pivot row */ 26977a48dd6fSHong Zhang 26987a48dd6fSHong Zhang while (prow < k){ 26997a48dd6fSHong Zhang nextprow = jl[prow]; 27007a48dd6fSHong Zhang 27017a48dd6fSHong Zhang /* merge prow into k-th row */ 27027a48dd6fSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 27037a48dd6fSHong Zhang jmax = ui[prow+1]; 27047a48dd6fSHong Zhang ncols = jmax-jmin; 2705ed59401aSHong Zhang i = jmin - ui[prow]; 2706ed59401aSHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 27075a8e39fbSHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 27085a8e39fbSHong Zhang j = *(uj - 1); 27095a8e39fbSHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 27107a48dd6fSHong Zhang nzk += nlnk; 27117a48dd6fSHong Zhang 27127a48dd6fSHong Zhang /* update il and jl for prow */ 27137a48dd6fSHong Zhang if (jmin < jmax){ 27147a48dd6fSHong Zhang il[prow] = jmin; 27157a48dd6fSHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 27167a48dd6fSHong Zhang } 27177a48dd6fSHong Zhang prow = nextprow; 27187a48dd6fSHong Zhang } 27197a48dd6fSHong Zhang 27207a48dd6fSHong Zhang /* if free space is not available, make more free space */ 27217a48dd6fSHong Zhang if (current_space->local_remaining<nzk) { 27227a48dd6fSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2723fd3f55cbSMatthew Knepley i *= PetscMin(nzk, (i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2724a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 2725a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 27267a48dd6fSHong Zhang reallocs++; 27277a48dd6fSHong Zhang } 27287a48dd6fSHong Zhang 27297a48dd6fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 273065e19b50SBarry Smith if (!nzk) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 27312ed133dbSHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 27327a48dd6fSHong Zhang 27337a48dd6fSHong Zhang /* add the k-th row into il and jl */ 273439e3d630SHong Zhang if (nzk > 1){ 27357a48dd6fSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 27367a48dd6fSHong Zhang jl[k] = jl[i]; jl[i] = k; 27377a48dd6fSHong Zhang il[k] = ui[k] + 1; 27387a48dd6fSHong Zhang } 27397a48dd6fSHong Zhang uj_ptr[k] = current_space->array; 27407a48dd6fSHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 27417a48dd6fSHong Zhang 27427a48dd6fSHong Zhang current_space->array += nzk; 27437a48dd6fSHong Zhang current_space->local_used += nzk; 27447a48dd6fSHong Zhang current_space->local_remaining -= nzk; 27457a48dd6fSHong Zhang 27467a48dd6fSHong Zhang current_space_lvl->array += nzk; 27477a48dd6fSHong Zhang current_space_lvl->local_used += nzk; 27487a48dd6fSHong Zhang current_space_lvl->local_remaining -= nzk; 27497a48dd6fSHong Zhang 27507a48dd6fSHong Zhang ui[k+1] = ui[k] + nzk; 27517a48dd6fSHong Zhang } 27527a48dd6fSHong Zhang 27536cf91177SBarry Smith #if defined(PETSC_USE_INFO) 27547a48dd6fSHong Zhang if (ai[am] != 0) { 275539e3d630SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 2756ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2757ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2758ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 27597a48dd6fSHong Zhang } else { 2760ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 27617a48dd6fSHong Zhang } 276263ba0a88SBarry Smith #endif 27637a48dd6fSHong Zhang 27647a48dd6fSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 2765b635d36bSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 27660e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 27675a8e39fbSHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 27687a48dd6fSHong Zhang 27697a48dd6fSHong Zhang /* destroy list of free space and other temporary array(s) */ 27707a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2771a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 27722ed133dbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2773a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 27747a48dd6fSHong Zhang 277539e3d630SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 277639e3d630SHong Zhang 27777a48dd6fSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 27787a48dd6fSHong Zhang 2779f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 27807a48dd6fSHong Zhang b->singlemalloc = PETSC_FALSE; 27817a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 27827a48dd6fSHong Zhang b->j = uj; 27837a48dd6fSHong Zhang b->i = ui; 27845f44c854SHong Zhang b->diag = udiag; 27857f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 27867a48dd6fSHong Zhang b->ilen = 0; 27877a48dd6fSHong Zhang b->imax = 0; 27887a48dd6fSHong Zhang b->row = perm; 2789b635d36bSHong Zhang b->col = perm; 2790b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2791b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2792b635d36bSHong Zhang b->icol = iperm; 27937a48dd6fSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 27947a48dd6fSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2795f284f12aSHong Zhang ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 27967a48dd6fSHong Zhang b->maxnz = b->nz = ui[am]; 2797e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2798e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 27997a48dd6fSHong Zhang 2800f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2801f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 28027a48dd6fSHong Zhang if (ai[am] != 0) { 2803f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 28047a48dd6fSHong Zhang } else { 2805f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 28067a48dd6fSHong Zhang } 28070a3c351aSHong Zhang fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace; 2808a6175056SHong Zhang PetscFunctionReturn(0); 2809a6175056SHong Zhang } 2810d5d45c9bSBarry Smith 2811177d4faaSHong Zhang #undef __FUNCT__ 2812177d4faaSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ" 281335233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 28140be760fbSHong Zhang { 28150be760fbSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 28160be760fbSHong Zhang Mat_SeqSBAIJ *b; 28170be760fbSHong Zhang PetscErrorCode ierr; 2818ace3abfcSBarry Smith PetscBool perm_identity; 28190be760fbSHong Zhang PetscReal fill = info->fill; 28200be760fbSHong Zhang const PetscInt *rip,*riip; 28210be760fbSHong Zhang PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 28220be760fbSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 28230be760fbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag; 28240be760fbSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 28250be760fbSHong Zhang PetscBT lnkbt; 28260be760fbSHong Zhang IS iperm; 28270be760fbSHong Zhang 28280be760fbSHong Zhang PetscFunctionBegin; 2829e32f2f54SBarry 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); 28300be760fbSHong Zhang /* check whether perm is the identity mapping */ 28310be760fbSHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 28320be760fbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 28330be760fbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 28340be760fbSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 28350be760fbSHong Zhang 28360be760fbSHong Zhang /* initialization */ 28370be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 28380be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 28390be760fbSHong Zhang ui[0] = 0; 28400be760fbSHong Zhang 28410be760fbSHong Zhang /* jl: linked list for storing indices of the pivot rows 28420be760fbSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 28430be760fbSHong Zhang ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr); 28440be760fbSHong Zhang for (i=0; i<am; i++){ 28450be760fbSHong Zhang jl[i] = am; il[i] = 0; 28460be760fbSHong Zhang } 28470be760fbSHong Zhang 28480be760fbSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 28490be760fbSHong Zhang nlnk = am + 1; 28500be760fbSHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28510be760fbSHong Zhang 285295b5ac22SHong Zhang /* initial FreeSpace size is fill*(ai[am]+am)/2 */ 285395b5ac22SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space);CHKERRQ(ierr); 28540be760fbSHong Zhang current_space = free_space; 28550be760fbSHong Zhang 28560be760fbSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 28570be760fbSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 28580be760fbSHong Zhang nzk = 0; 28590be760fbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2860e32f2f54SBarry 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); 28610be760fbSHong Zhang ncols_upper = 0; 28620be760fbSHong Zhang for (j=0; j<ncols; j++){ 28630be760fbSHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 28640be760fbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 28650be760fbSHong Zhang cols[ncols_upper] = i; 28660be760fbSHong Zhang ncols_upper++; 28670be760fbSHong Zhang } 28680be760fbSHong Zhang } 28690be760fbSHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28700be760fbSHong Zhang nzk += nlnk; 28710be760fbSHong Zhang 28720be760fbSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 28730be760fbSHong Zhang prow = jl[k]; /* 1st pivot row */ 28740be760fbSHong Zhang 28750be760fbSHong Zhang while (prow < k){ 28760be760fbSHong Zhang nextprow = jl[prow]; 28770be760fbSHong Zhang /* merge prow into k-th row */ 28780be760fbSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 28790be760fbSHong Zhang jmax = ui[prow+1]; 28800be760fbSHong Zhang ncols = jmax-jmin; 28810be760fbSHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 28820be760fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28830be760fbSHong Zhang nzk += nlnk; 28840be760fbSHong Zhang 28850be760fbSHong Zhang /* update il and jl for prow */ 28860be760fbSHong Zhang if (jmin < jmax){ 28870be760fbSHong Zhang il[prow] = jmin; 28880be760fbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 28890be760fbSHong Zhang } 28900be760fbSHong Zhang prow = nextprow; 28910be760fbSHong Zhang } 28920be760fbSHong Zhang 28930be760fbSHong Zhang /* if free space is not available, make more free space */ 28940be760fbSHong Zhang if (current_space->local_remaining<nzk) { 28950be760fbSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2896c6ae9bfcSHong Zhang i *= PetscMin(nzk,i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */ 28970be760fbSHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 28980be760fbSHong Zhang reallocs++; 28990be760fbSHong Zhang } 29000be760fbSHong Zhang 29010be760fbSHong Zhang /* copy data into free space, then initialize lnk */ 29020be760fbSHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 29030be760fbSHong Zhang 29040be760fbSHong Zhang /* add the k-th row into il and jl */ 29057b056e98SHong Zhang if (nzk > 1){ 29060be760fbSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 29070be760fbSHong Zhang jl[k] = jl[i]; jl[i] = k; 29080be760fbSHong Zhang il[k] = ui[k] + 1; 29090be760fbSHong Zhang } 29100be760fbSHong Zhang ui_ptr[k] = current_space->array; 29110be760fbSHong Zhang current_space->array += nzk; 29120be760fbSHong Zhang current_space->local_used += nzk; 29130be760fbSHong Zhang current_space->local_remaining -= nzk; 29140be760fbSHong Zhang 29150be760fbSHong Zhang ui[k+1] = ui[k] + nzk; 29160be760fbSHong Zhang } 29170be760fbSHong Zhang 29180be760fbSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 29190be760fbSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 29200be760fbSHong Zhang ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 29210be760fbSHong Zhang 29229263d837SHong Zhang /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */ 29230be760fbSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 292435233d99SShri Abhyankar ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */ 29250be760fbSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 29260be760fbSHong Zhang 29270be760fbSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 29280be760fbSHong Zhang 2929f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 29300be760fbSHong Zhang b->singlemalloc = PETSC_FALSE; 29310be760fbSHong Zhang b->free_a = PETSC_TRUE; 29320be760fbSHong Zhang b->free_ij = PETSC_TRUE; 29330be760fbSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 29340be760fbSHong Zhang b->j = uj; 29350be760fbSHong Zhang b->i = ui; 29360be760fbSHong Zhang b->diag = udiag; 29370be760fbSHong Zhang b->free_diag = PETSC_TRUE; 29380be760fbSHong Zhang b->ilen = 0; 29390be760fbSHong Zhang b->imax = 0; 29400be760fbSHong Zhang b->row = perm; 29410be760fbSHong Zhang b->col = perm; 29420be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29430be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29440be760fbSHong Zhang b->icol = iperm; 29450be760fbSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 29460be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 29470860e08cSHong Zhang ierr = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 29480be760fbSHong Zhang b->maxnz = b->nz = ui[am]; 29490be760fbSHong Zhang 2950f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2951f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 29520be760fbSHong Zhang if (ai[am] != 0) { 29536a69fef8SHong Zhang /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */ 295495b5ac22SHong Zhang fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am); 29550be760fbSHong Zhang } else { 2956f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 29570be760fbSHong Zhang } 29589263d837SHong Zhang #if defined(PETSC_USE_INFO) 29599263d837SHong Zhang if (ai[am] != 0) { 29609263d837SHong Zhang PetscReal af = fact->info.fill_ratio_needed; 29619263d837SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 29629263d837SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 29639263d837SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 29649263d837SHong Zhang } else { 29659263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 29669263d837SHong Zhang } 29679263d837SHong Zhang #endif 296835233d99SShri Abhyankar fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 29690be760fbSHong Zhang PetscFunctionReturn(0); 29700be760fbSHong Zhang } 29710be760fbSHong Zhang 2972f76d2b81SHong Zhang #undef __FUNCT__ 2973ad04f41aSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ_inplace" 2974ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2975f76d2b81SHong Zhang { 2976f76d2b81SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 297710c27e3dSHong Zhang Mat_SeqSBAIJ *b; 2978dfbe8321SBarry Smith PetscErrorCode ierr; 2979ace3abfcSBarry Smith PetscBool perm_identity; 298010c27e3dSHong Zhang PetscReal fill = info->fill; 29815d0c19d7SBarry Smith const PetscInt *rip,*riip; 29825d0c19d7SBarry Smith PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 298310c27e3dSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 29842ed133dbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr; 2985a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 298610c27e3dSHong Zhang PetscBT lnkbt; 29872ed133dbSHong Zhang IS iperm; 2988f76d2b81SHong Zhang 2989f76d2b81SHong Zhang PetscFunctionBegin; 2990e32f2f54SBarry 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); 29912ed133dbSHong Zhang /* check whether perm is the identity mapping */ 2992f76d2b81SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 29932ed133dbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 29942ed133dbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 29959bfd6278SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 299610c27e3dSHong Zhang 299710c27e3dSHong Zhang /* initialization */ 29981393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 299910c27e3dSHong Zhang ui[0] = 0; 300010c27e3dSHong Zhang 300110c27e3dSHong Zhang /* jl: linked list for storing indices of the pivot rows 30021393bc97SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 30030e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr); 30041393bc97SHong Zhang for (i=0; i<am; i++){ 30051393bc97SHong Zhang jl[i] = am; il[i] = 0; 3006f76d2b81SHong Zhang } 300710c27e3dSHong Zhang 300810c27e3dSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 30091393bc97SHong Zhang nlnk = am + 1; 30101393bc97SHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 301110c27e3dSHong Zhang 30121393bc97SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 3013a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 301410c27e3dSHong Zhang current_space = free_space; 301510c27e3dSHong Zhang 30161393bc97SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 301710c27e3dSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 301810c27e3dSHong Zhang nzk = 0; 30192ed133dbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 3020e32f2f54SBarry 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); 30212ed133dbSHong Zhang ncols_upper = 0; 30222ed133dbSHong Zhang for (j=0; j<ncols; j++){ 30239bfd6278SHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 30242ed133dbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 30252ed133dbSHong Zhang cols[ncols_upper] = i; 30262ed133dbSHong Zhang ncols_upper++; 30272ed133dbSHong Zhang } 30282ed133dbSHong Zhang } 30291393bc97SHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 303010c27e3dSHong Zhang nzk += nlnk; 303110c27e3dSHong Zhang 303210c27e3dSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 303310c27e3dSHong Zhang prow = jl[k]; /* 1st pivot row */ 303410c27e3dSHong Zhang 303510c27e3dSHong Zhang while (prow < k){ 303610c27e3dSHong Zhang nextprow = jl[prow]; 303710c27e3dSHong Zhang /* merge prow into k-th row */ 30381393bc97SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 303910c27e3dSHong Zhang jmax = ui[prow+1]; 304010c27e3dSHong Zhang ncols = jmax-jmin; 30411393bc97SHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 30425a8e39fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 304310c27e3dSHong Zhang nzk += nlnk; 304410c27e3dSHong Zhang 304510c27e3dSHong Zhang /* update il and jl for prow */ 304610c27e3dSHong Zhang if (jmin < jmax){ 304710c27e3dSHong Zhang il[prow] = jmin; 30482ed133dbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 304910c27e3dSHong Zhang } 305010c27e3dSHong Zhang prow = nextprow; 305110c27e3dSHong Zhang } 305210c27e3dSHong Zhang 305310c27e3dSHong Zhang /* if free space is not available, make more free space */ 305410c27e3dSHong Zhang if (current_space->local_remaining<nzk) { 30551393bc97SHong Zhang i = am - k + 1; /* num of unfactored rows */ 305610c27e3dSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 3057a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 305810c27e3dSHong Zhang reallocs++; 305910c27e3dSHong Zhang } 306010c27e3dSHong Zhang 306110c27e3dSHong Zhang /* copy data into free space, then initialize lnk */ 30621393bc97SHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 306310c27e3dSHong Zhang 306410c27e3dSHong Zhang /* add the k-th row into il and jl */ 306510c27e3dSHong Zhang if (nzk-1 > 0){ 30661393bc97SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 306710c27e3dSHong Zhang jl[k] = jl[i]; jl[i] = k; 306810c27e3dSHong Zhang il[k] = ui[k] + 1; 306910c27e3dSHong Zhang } 30702ed133dbSHong Zhang ui_ptr[k] = current_space->array; 307110c27e3dSHong Zhang current_space->array += nzk; 307210c27e3dSHong Zhang current_space->local_used += nzk; 307310c27e3dSHong Zhang current_space->local_remaining -= nzk; 307410c27e3dSHong Zhang 307510c27e3dSHong Zhang ui[k+1] = ui[k] + nzk; 307610c27e3dSHong Zhang } 307710c27e3dSHong Zhang 30786cf91177SBarry Smith #if defined(PETSC_USE_INFO) 30791393bc97SHong Zhang if (ai[am] != 0) { 30801393bc97SHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 3081ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 3082ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 3083ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 308410c27e3dSHong Zhang } else { 3085ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 308610c27e3dSHong Zhang } 308763ba0a88SBarry Smith #endif 308810c27e3dSHong Zhang 308910c27e3dSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 30909bfd6278SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 30910e83c824SBarry Smith ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 309210c27e3dSHong Zhang 309310c27e3dSHong Zhang /* destroy list of free space and other temporary array(s) */ 30941393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 3095a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 309610c27e3dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 309710c27e3dSHong Zhang 309810c27e3dSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 309910c27e3dSHong Zhang 3100f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 310110c27e3dSHong Zhang b->singlemalloc = PETSC_FALSE; 3102e6b907acSBarry Smith b->free_a = PETSC_TRUE; 3103e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 31041393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 310510c27e3dSHong Zhang b->j = uj; 310610c27e3dSHong Zhang b->i = ui; 310710c27e3dSHong Zhang b->diag = 0; 310810c27e3dSHong Zhang b->ilen = 0; 310910c27e3dSHong Zhang b->imax = 0; 311010c27e3dSHong Zhang b->row = perm; 31119bfd6278SHong Zhang b->col = perm; 31129bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 31139bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 31149bfd6278SHong Zhang b->icol = iperm; 311510c27e3dSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 31161393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3117719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 31181393bc97SHong Zhang b->maxnz = b->nz = ui[am]; 311910c27e3dSHong Zhang 3120f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 3121f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 31221393bc97SHong Zhang if (ai[am] != 0) { 3123f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 312410c27e3dSHong Zhang } else { 3125f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 312610c27e3dSHong Zhang } 31270a3c351aSHong Zhang fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace; 3128f76d2b81SHong Zhang PetscFunctionReturn(0); 3129f76d2b81SHong Zhang } 3130599ef60dSHong Zhang 31311d098868SHong Zhang #undef __FUNCT__ 313235233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering" 313335233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 3134813a1f2bSShri Abhyankar { 3135813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3136813a1f2bSShri Abhyankar PetscErrorCode ierr; 3137813a1f2bSShri Abhyankar PetscInt n = A->rmap->n; 3138813a1f2bSShri Abhyankar const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag,*vi; 3139813a1f2bSShri Abhyankar PetscScalar *x,sum; 3140813a1f2bSShri Abhyankar const PetscScalar *b; 3141813a1f2bSShri Abhyankar const MatScalar *aa = a->a,*v; 3142813a1f2bSShri Abhyankar PetscInt i,nz; 3143813a1f2bSShri Abhyankar 3144813a1f2bSShri Abhyankar PetscFunctionBegin; 3145813a1f2bSShri Abhyankar if (!n) PetscFunctionReturn(0); 3146813a1f2bSShri Abhyankar 31473649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 3148813a1f2bSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3149813a1f2bSShri Abhyankar 3150813a1f2bSShri Abhyankar /* forward solve the lower triangular */ 3151813a1f2bSShri Abhyankar x[0] = b[0]; 3152813a1f2bSShri Abhyankar v = aa; 3153813a1f2bSShri Abhyankar vi = aj; 3154813a1f2bSShri Abhyankar for (i=1; i<n; i++) { 3155813a1f2bSShri Abhyankar nz = ai[i+1] - ai[i]; 3156813a1f2bSShri Abhyankar sum = b[i]; 3157813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 3158813a1f2bSShri Abhyankar v += nz; 3159813a1f2bSShri Abhyankar vi += nz; 3160813a1f2bSShri Abhyankar x[i] = sum; 3161813a1f2bSShri Abhyankar } 3162813a1f2bSShri Abhyankar 3163813a1f2bSShri Abhyankar /* backward solve the upper triangular */ 316462fbd6afSShri Abhyankar for (i=n-1; i>=0; i--){ 31653c0119dfSShri Abhyankar v = aa + adiag[i+1] + 1; 31663c0119dfSShri Abhyankar vi = aj + adiag[i+1] + 1; 3167813a1f2bSShri Abhyankar nz = adiag[i] - adiag[i+1]-1; 3168813a1f2bSShri Abhyankar sum = x[i]; 3169813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 31703c0119dfSShri Abhyankar x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */ 3171813a1f2bSShri Abhyankar } 3172813a1f2bSShri Abhyankar 3173813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 31743649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3175813a1f2bSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3176813a1f2bSShri Abhyankar PetscFunctionReturn(0); 3177813a1f2bSShri Abhyankar } 3178813a1f2bSShri Abhyankar 31791d098868SHong Zhang #undef __FUNCT__ 318035233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ" 318135233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 3182f268cba8SShri Abhyankar { 3183f268cba8SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3184f268cba8SShri Abhyankar IS iscol = a->col,isrow = a->row; 3185f268cba8SShri Abhyankar PetscErrorCode ierr; 3186f268cba8SShri Abhyankar PetscInt i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz; 3187f268cba8SShri Abhyankar const PetscInt *rout,*cout,*r,*c; 3188f268cba8SShri Abhyankar PetscScalar *x,*tmp,sum; 3189f268cba8SShri Abhyankar const PetscScalar *b; 3190f268cba8SShri Abhyankar const MatScalar *aa = a->a,*v; 3191f268cba8SShri Abhyankar 3192f268cba8SShri Abhyankar PetscFunctionBegin; 3193f268cba8SShri Abhyankar if (!n) PetscFunctionReturn(0); 3194f268cba8SShri Abhyankar 31953649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 3196f268cba8SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3197f268cba8SShri Abhyankar tmp = a->solve_work; 3198f268cba8SShri Abhyankar 3199f268cba8SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 3200f268cba8SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 3201f268cba8SShri Abhyankar 3202f268cba8SShri Abhyankar /* forward solve the lower triangular */ 3203f268cba8SShri Abhyankar tmp[0] = b[r[0]]; 3204f268cba8SShri Abhyankar v = aa; 3205f268cba8SShri Abhyankar vi = aj; 3206f268cba8SShri Abhyankar for (i=1; i<n; i++) { 3207f268cba8SShri Abhyankar nz = ai[i+1] - ai[i]; 3208f268cba8SShri Abhyankar sum = b[r[i]]; 3209f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3210f268cba8SShri Abhyankar tmp[i] = sum; 3211f268cba8SShri Abhyankar v += nz; vi += nz; 3212f268cba8SShri Abhyankar } 3213f268cba8SShri Abhyankar 3214f268cba8SShri Abhyankar /* backward solve the upper triangular */ 3215f268cba8SShri Abhyankar for (i=n-1; i>=0; i--){ 32163c0119dfSShri Abhyankar v = aa + adiag[i+1]+1; 32173c0119dfSShri Abhyankar vi = aj + adiag[i+1]+1; 3218f268cba8SShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 3219f268cba8SShri Abhyankar sum = tmp[i]; 3220f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3221f268cba8SShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 3222f268cba8SShri Abhyankar } 3223f268cba8SShri Abhyankar 3224f268cba8SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 3225f268cba8SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 32263649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3227f268cba8SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3228f268cba8SShri Abhyankar ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 3229f268cba8SShri Abhyankar PetscFunctionReturn(0); 3230f268cba8SShri Abhyankar } 3231f268cba8SShri Abhyankar 3232f268cba8SShri Abhyankar #undef __FUNCT__ 32331d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ" 3234fe97e370SBarry Smith /* 3235fe97e370SBarry 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 3236fe97e370SBarry Smith */ 32371d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact) 32381d098868SHong Zhang { 32391d098868SHong Zhang Mat B = *fact; 3240599ef60dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b; 3241599ef60dSHong Zhang IS isicol; 3242599ef60dSHong Zhang PetscErrorCode ierr; 32431d098868SHong Zhang const PetscInt *r,*ic; 32441d098868SHong Zhang PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag; 3245dc3a2fd3SHong Zhang PetscInt *bi,*bj,*bdiag,*bdiag_rev; 3246f61a948aSLisandro Dalcin PetscInt row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au; 32471d098868SHong Zhang PetscInt nlnk,*lnk; 32481d098868SHong Zhang PetscBT lnkbt; 3249ace3abfcSBarry Smith PetscBool row_identity,icol_identity; 32508aee2decSHong Zhang MatScalar *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp; 32511d098868SHong Zhang const PetscInt *ics; 32521d098868SHong Zhang PetscInt j,nz,*pj,*bjtmp,k,ncut,*jtmp; 3253182b8fbaSHong Zhang PetscReal dt=info->dt,dtcol=info->dtcol,shift=info->shiftamount; 3254f61a948aSLisandro Dalcin PetscInt dtcount=(PetscInt)info->dtcount,nnz_max; 3255ace3abfcSBarry Smith PetscBool missing; 3256599ef60dSHong Zhang 3257599ef60dSHong Zhang PetscFunctionBegin; 3258f61a948aSLisandro Dalcin 3259f61a948aSLisandro Dalcin if (dt == PETSC_DEFAULT) dt = 0.005; 3260f61a948aSLisandro Dalcin if (dtcol == PETSC_DEFAULT) dtcol = 0.01; /* XXX unused! */ 3261f61a948aSLisandro Dalcin if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax); 3262f61a948aSLisandro Dalcin 32631d098868SHong Zhang /* ------- symbolic factorization, can be reused ---------*/ 32641d098868SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 3265e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 32661d098868SHong Zhang adiag=a->diag; 3267599ef60dSHong Zhang 3268599ef60dSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 3269599ef60dSHong Zhang 3270599ef60dSHong Zhang /* bdiag is location of diagonal in factor */ 32711bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); /* becomes b->diag */ 32721bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag_rev);CHKERRQ(ierr); /* temporary */ 3273599ef60dSHong Zhang 32741d098868SHong Zhang /* allocate row pointers bi */ 32758fc3a347SHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 32761d098868SHong Zhang 32771d098868SHong Zhang /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */ 3278393d3a68SHong Zhang if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */ 32791d098868SHong Zhang nnz_max = ai[n]+2*n*dtcount+2; 32808fc3a347SHong Zhang 32818fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 32828fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr); 3283599ef60dSHong Zhang 3284599ef60dSHong Zhang /* put together the new matrix */ 3285599ef60dSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 3286599ef60dSHong Zhang ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 3287f284f12aSHong Zhang b = (Mat_SeqAIJ*)B->data; 3288599ef60dSHong Zhang b->free_a = PETSC_TRUE; 3289599ef60dSHong Zhang b->free_ij = PETSC_TRUE; 3290599ef60dSHong Zhang b->singlemalloc = PETSC_FALSE; 3291599ef60dSHong Zhang b->a = ba; 3292599ef60dSHong Zhang b->j = bj; 3293599ef60dSHong Zhang b->i = bi; 3294599ef60dSHong Zhang b->diag = bdiag; 3295599ef60dSHong Zhang b->ilen = 0; 3296599ef60dSHong Zhang b->imax = 0; 3297599ef60dSHong Zhang b->row = isrow; 3298599ef60dSHong Zhang b->col = iscol; 3299599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 3300599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 3301599ef60dSHong Zhang b->icol = isicol; 3302599ef60dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3303599ef60dSHong Zhang 33041d098868SHong Zhang ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 33051d098868SHong Zhang b->maxnz = nnz_max; 3306599ef60dSHong Zhang 3307d5f3da31SBarry Smith B->factortype = MAT_FACTOR_ILUDT; 3308f284f12aSHong Zhang B->info.factor_mallocs = 0; 3309f284f12aSHong Zhang B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]); 33101d098868SHong Zhang CHKMEMQ; 33111d098868SHong Zhang /* ------- end of symbolic factorization ---------*/ 3312599ef60dSHong Zhang 3313599ef60dSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 3314599ef60dSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3315599ef60dSHong Zhang ics = ic; 3316599ef60dSHong Zhang 3317599ef60dSHong Zhang /* linked list for storing column indices of the active row */ 3318599ef60dSHong Zhang nlnk = n + 1; 3319599ef60dSHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 33201d098868SHong Zhang 33211d098868SHong Zhang /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */ 33220e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt,&im,n,PetscInt,&jtmp);CHKERRQ(ierr); 33231d098868SHong Zhang /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */ 33240e83c824SBarry Smith ierr = PetscMalloc2(n,MatScalar,&rtmp,n,MatScalar,&vtmp);CHKERRQ(ierr); 33250e83c824SBarry Smith ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr); 3326599ef60dSHong Zhang 3327599ef60dSHong Zhang bi[0] = 0; 33288fc3a347SHong Zhang bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */ 3329dc3a2fd3SHong Zhang bdiag_rev[n] = bdiag[0]; 33308fc3a347SHong Zhang bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */ 3331599ef60dSHong Zhang for (i=0; i<n; i++) { 3332599ef60dSHong Zhang /* copy initial fill into linked list */ 3333599ef60dSHong Zhang nzi = 0; /* nonzeros for active row i */ 33348369b28dSHong Zhang nzi = ai[r[i]+1] - ai[r[i]]; 3335e32f2f54SBarry 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); 33368369b28dSHong Zhang nzi_al = adiag[r[i]] - ai[r[i]]; 33378369b28dSHong Zhang nzi_au = ai[r[i]+1] - adiag[r[i]] -1; 3338599ef60dSHong Zhang ajtmp = aj + ai[r[i]]; 33398369b28dSHong Zhang ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3340599ef60dSHong Zhang 3341599ef60dSHong Zhang /* load in initial (unfactored row) */ 33421d098868SHong Zhang aatmp = a->a + ai[r[i]]; 33438369b28dSHong Zhang for (j=0; j<nzi; j++) { 33441d098868SHong Zhang rtmp[ics[*ajtmp++]] = *aatmp++; 3345599ef60dSHong Zhang } 3346599ef60dSHong Zhang 3347599ef60dSHong Zhang /* add pivot rows into linked list */ 3348599ef60dSHong Zhang row = lnk[n]; 3349599ef60dSHong Zhang while (row < i ) { 33501d098868SHong Zhang nzi_bl = bi[row+1] - bi[row] + 1; 33511d098868SHong Zhang bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */ 33521d098868SHong Zhang ierr = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr); 3353599ef60dSHong Zhang nzi += nlnk; 3354599ef60dSHong Zhang row = lnk[row]; 3355599ef60dSHong Zhang } 3356599ef60dSHong Zhang 33578369b28dSHong Zhang /* copy data from lnk into jtmp, then initialize lnk */ 33588369b28dSHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr); 3359599ef60dSHong Zhang 3360599ef60dSHong Zhang /* numerical factorization */ 33618369b28dSHong Zhang bjtmp = jtmp; 3362599ef60dSHong Zhang row = *bjtmp++; /* 1st pivot row */ 3363599ef60dSHong Zhang while ( row < i ) { 3364599ef60dSHong Zhang pc = rtmp + row; 33653c2a7987SHong Zhang pv = ba + bdiag[row]; /* 1./(diag of the pivot row) */ 33663c2a7987SHong Zhang multiplier = (*pc) * (*pv); 3367599ef60dSHong Zhang *pc = multiplier; 33683c2a7987SHong Zhang if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */ 33691d098868SHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 33701d098868SHong Zhang pv = ba + bdiag[row+1] + 1; 33711d098868SHong Zhang /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */ 33721d098868SHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 33731d098868SHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 337428b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 3375599ef60dSHong Zhang } 3376599ef60dSHong Zhang row = *bjtmp++; 3377599ef60dSHong Zhang } 3378599ef60dSHong Zhang 33798369b28dSHong Zhang /* copy sparse rtmp into contiguous vtmp; separate L and U part */ 33808aee2decSHong Zhang diag_tmp = rtmp[i]; /* save diagonal value - may not needed?? */ 33818369b28dSHong Zhang nzi_bl = 0; j = 0; 33828369b28dSHong Zhang while (jtmp[j] < i){ /* Note: jtmp is sorted */ 33838aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 33848369b28dSHong Zhang nzi_bl++; j++; 33858369b28dSHong Zhang } 33868369b28dSHong Zhang nzi_bu = nzi - nzi_bl -1; 33878369b28dSHong Zhang while (j < nzi){ 33888aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 33898369b28dSHong Zhang j++; 33908369b28dSHong Zhang } 33918369b28dSHong Zhang 3392599ef60dSHong Zhang bjtmp = bj + bi[i]; 3393599ef60dSHong Zhang batmp = ba + bi[i]; 33948369b28dSHong Zhang /* apply level dropping rule to L part */ 33958369b28dSHong Zhang ncut = nzi_al + dtcount; 33968369b28dSHong Zhang if (ncut < nzi_bl){ 33978369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr); 33988369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr); 3399599ef60dSHong Zhang } else { 34008369b28dSHong Zhang ncut = nzi_bl; 34018369b28dSHong Zhang } 34028369b28dSHong Zhang for (j=0; j<ncut; j++){ 34038369b28dSHong Zhang bjtmp[j] = jtmp[j]; 34048369b28dSHong Zhang batmp[j] = vtmp[j]; 34056da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */ 34068369b28dSHong Zhang } 34071d098868SHong Zhang bi[i+1] = bi[i] + ncut; 34088369b28dSHong Zhang nzi = ncut + 1; 34098369b28dSHong Zhang 34108369b28dSHong Zhang /* apply level dropping rule to U part */ 34118369b28dSHong Zhang ncut = nzi_au + dtcount; 34128369b28dSHong Zhang if (ncut < nzi_bu){ 34138369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr); 34148369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr); 34158369b28dSHong Zhang } else { 34168369b28dSHong Zhang ncut = nzi_bu; 34178369b28dSHong Zhang } 34188369b28dSHong Zhang nzi += ncut; 34191d098868SHong Zhang 34201d098868SHong Zhang /* mark bdiagonal */ 34211d098868SHong Zhang bdiag[i+1] = bdiag[i] - (ncut + 1); 3422dc3a2fd3SHong Zhang bdiag_rev[n-i-1] = bdiag[i+1]; 34238fc3a347SHong Zhang bi[2*n - i] = bi[2*n - i +1] - (ncut + 1); 34241d098868SHong Zhang bjtmp = bj + bdiag[i]; 34251d098868SHong Zhang batmp = ba + bdiag[i]; 34261d098868SHong Zhang *bjtmp = i; 34278aee2decSHong Zhang *batmp = diag_tmp; /* rtmp[i]; */ 3428c9c72f8fSHong Zhang if (*batmp == 0.0) { 3429c9c72f8fSHong Zhang *batmp = dt+shift; 3430bd1bc851SBarry Smith /* printf(" row %d add shift %g\n",i,shift); */ 3431c9c72f8fSHong Zhang } 3432b78a477dSHong Zhang *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */ 34336da515a1SHong Zhang /* printf(" (%d,%g),",*bjtmp,*batmp); */ 34341d098868SHong Zhang 34351d098868SHong Zhang bjtmp = bj + bdiag[i+1]+1; 34361d098868SHong Zhang batmp = ba + bdiag[i+1]+1; 34378369b28dSHong Zhang for (k=0; k<ncut; k++){ 34381d098868SHong Zhang bjtmp[k] = jtmp[nzi_bl+1+k]; 34391d098868SHong Zhang batmp[k] = vtmp[nzi_bl+1+k]; 34406da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */ 3441599ef60dSHong Zhang } 34426da515a1SHong Zhang /* printf("\n"); */ 3443599ef60dSHong Zhang 3444599ef60dSHong Zhang im[i] = nzi; /* used by PetscLLAddSortedLU() */ 3445599ef60dSHong Zhang /* 34461d098868SHong Zhang printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]); 34478369b28dSHong Zhang printf(" ----------------------------\n"); 3448599ef60dSHong Zhang */ 34498369b28dSHong Zhang } /* for (i=0; i<n; i++) */ 34501d098868SHong Zhang /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */ 3451e32f2f54SBarry 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]); 3452599ef60dSHong Zhang 3453599ef60dSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3454599ef60dSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3455599ef60dSHong Zhang 3456599ef60dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 34570e83c824SBarry Smith ierr = PetscFree2(im,jtmp);CHKERRQ(ierr); 34580e83c824SBarry Smith ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr); 34591bfa06eaSJed Brown ierr = PetscFree(bdiag_rev);CHKERRQ(ierr); 3460599ef60dSHong Zhang 3461599ef60dSHong Zhang ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr); 34621d098868SHong Zhang b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n]; 3463599ef60dSHong Zhang 3464599ef60dSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 3465599ef60dSHong Zhang ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr); 3466599ef60dSHong Zhang if (row_identity && icol_identity) { 346735233d99SShri Abhyankar B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 3468599ef60dSHong Zhang } else { 346935233d99SShri Abhyankar B->ops->solve = MatSolve_SeqAIJ; 3470599ef60dSHong Zhang } 3471599ef60dSHong Zhang 34721d098868SHong Zhang B->ops->solveadd = 0; 34731d098868SHong Zhang B->ops->solvetranspose = 0; 34741d098868SHong Zhang B->ops->solvetransposeadd = 0; 34751d098868SHong Zhang B->ops->matsolve = 0; 3476599ef60dSHong Zhang B->assembled = PETSC_TRUE; 3477599ef60dSHong Zhang B->preallocated = PETSC_TRUE; 3478599ef60dSHong Zhang PetscFunctionReturn(0); 3479599ef60dSHong Zhang } 3480599ef60dSHong Zhang 34813c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */ 34823c2a7987SHong Zhang #undef __FUNCT__ 34833c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ" 3484fe97e370SBarry Smith /* 3485fe97e370SBarry 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 3486fe97e370SBarry Smith */ 3487fe97e370SBarry Smith 34887087cfbeSBarry Smith PetscErrorCode MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info) 34893c2a7987SHong Zhang { 34903c2a7987SHong Zhang PetscErrorCode ierr; 34913c2a7987SHong Zhang 34923c2a7987SHong Zhang PetscFunctionBegin; 34933c2a7987SHong Zhang ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr); 34943c2a7987SHong Zhang PetscFunctionReturn(0); 34953c2a7987SHong Zhang } 34963c2a7987SHong Zhang 34973c2a7987SHong Zhang /* 34983c2a7987SHong Zhang same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors 34993c2a7987SHong Zhang - intend to replace existing MatLUFactorNumeric_SeqAIJ() 35003c2a7987SHong Zhang */ 35013c2a7987SHong Zhang #undef __FUNCT__ 35023c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ" 3503fe97e370SBarry Smith /* 3504fe97e370SBarry 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 3505fe97e370SBarry Smith */ 3506fe97e370SBarry Smith 35077087cfbeSBarry Smith PetscErrorCode MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info) 35083c2a7987SHong Zhang { 35093c2a7987SHong Zhang Mat C=fact; 35103c2a7987SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 35113c2a7987SHong Zhang IS isrow = b->row,isicol = b->icol; 35123c2a7987SHong Zhang PetscErrorCode ierr; 35133c2a7987SHong Zhang const PetscInt *r,*ic,*ics; 3514b78a477dSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 3515b78a477dSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj; 35163c2a7987SHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 3517182b8fbaSHong Zhang PetscReal dt=info->dt,shift=info->shiftamount; 3518ace3abfcSBarry Smith PetscBool row_identity, col_identity; 35193c2a7987SHong Zhang 35203c2a7987SHong Zhang PetscFunctionBegin; 35213c2a7987SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 35223c2a7987SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3523b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 35243c2a7987SHong Zhang ics = ic; 35253c2a7987SHong Zhang 35263c2a7987SHong Zhang for (i=0; i<n; i++){ 3527b78a477dSHong Zhang /* initialize rtmp array */ 3528b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of nozeros in L(i,:) */ 35293c2a7987SHong Zhang bjtmp = bj + bi[i]; 3530b78a477dSHong Zhang for (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0; 3531b78a477dSHong Zhang rtmp[i] = 0.0; 3532b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */ 3533b78a477dSHong Zhang bjtmp = bj + bdiag[i+1] + 1; 3534b78a477dSHong Zhang for (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0; 35353c2a7987SHong Zhang 3536b78a477dSHong Zhang /* load in initial unfactored row of A */ 35376da515a1SHong Zhang /* printf("row %d\n",i); */ 35383c2a7987SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 35393c2a7987SHong Zhang ajtmp = aj + ai[r[i]]; 35403c2a7987SHong Zhang v = aa + ai[r[i]]; 35413c2a7987SHong Zhang for (j=0; j<nz; j++) { 3542b78a477dSHong Zhang rtmp[ics[*ajtmp++]] = v[j]; 35436da515a1SHong Zhang /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */ 35443c2a7987SHong Zhang } 35456da515a1SHong Zhang /* printf("\n"); */ 35463c2a7987SHong Zhang 3547b78a477dSHong Zhang /* numerical factorization */ 3548b78a477dSHong Zhang bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */ 3549b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of entries in L(i,:) */ 3550b78a477dSHong Zhang k = 0; 3551b78a477dSHong Zhang while (k < nzl){ 35523c2a7987SHong Zhang row = *bjtmp++; 35536da515a1SHong Zhang /* printf(" prow %d\n",row); */ 35543c2a7987SHong Zhang pc = rtmp + row; 3555b78a477dSHong Zhang pv = b->a + bdiag[row]; /* 1./(diag of the pivot row) */ 3556b78a477dSHong Zhang multiplier = (*pc) * (*pv); 35573c2a7987SHong Zhang *pc = multiplier; 3558b78a477dSHong Zhang if (PetscAbsScalar(multiplier) > dt){ 3559b78a477dSHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 3560b78a477dSHong Zhang pv = b->a + bdiag[row+1] + 1; 3561b78a477dSHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 3562b78a477dSHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 356328b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 35643c2a7987SHong Zhang } 3565b78a477dSHong Zhang k++; 35663c2a7987SHong Zhang } 35673c2a7987SHong Zhang 3568b78a477dSHong Zhang /* finished row so stick it into b->a */ 3569b78a477dSHong Zhang /* L-part */ 3570b78a477dSHong Zhang pv = b->a + bi[i] ; 3571b78a477dSHong Zhang pj = bj + bi[i] ; 3572b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; 3573b78a477dSHong Zhang for (j=0; j<nzl; j++) { 3574b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 35756da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 35763c2a7987SHong Zhang } 3577b78a477dSHong Zhang 3578b78a477dSHong Zhang /* diagonal: invert diagonal entries for simplier triangular solves */ 3579b78a477dSHong Zhang if (rtmp[i] == 0.0) rtmp[i] = dt+shift; 3580b78a477dSHong Zhang b->a[bdiag[i]] = 1.0/rtmp[i]; 35816da515a1SHong Zhang /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */ 3582b78a477dSHong Zhang 3583b78a477dSHong Zhang /* U-part */ 3584b78a477dSHong Zhang pv = b->a + bdiag[i+1] + 1; 3585b78a477dSHong Zhang pj = bj + bdiag[i+1] + 1; 3586b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1] - 1; 3587b78a477dSHong Zhang for (j=0; j<nzu; j++) { 3588b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 35896da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 3590b78a477dSHong Zhang } 35916da515a1SHong Zhang /* printf("\n"); */ 3592b78a477dSHong Zhang } 3593b78a477dSHong Zhang 35943c2a7987SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 35953c2a7987SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 35963c2a7987SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3597b78a477dSHong Zhang 35983c2a7987SHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 35993c2a7987SHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 36003c2a7987SHong Zhang if (row_identity && col_identity) { 360135233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 36023c2a7987SHong Zhang } else { 360335233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ; 36043c2a7987SHong Zhang } 3605b78a477dSHong Zhang C->ops->solveadd = 0; 3606b78a477dSHong Zhang C->ops->solvetranspose = 0; 3607b78a477dSHong Zhang C->ops->solvetransposeadd = 0; 3608b78a477dSHong Zhang C->ops->matsolve = 0; 36093c2a7987SHong Zhang C->assembled = PETSC_TRUE; 36103c2a7987SHong Zhang C->preallocated = PETSC_TRUE; 36113c2a7987SHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 36123c2a7987SHong Zhang PetscFunctionReturn(0); 36133c2a7987SHong Zhang } 3614