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; 115b24902e0SBarry Smith ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr); 116b24902e0SBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 117599ef60dSHong Zhang if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT){ 118b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr); 11935233d99SShri Abhyankar (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ; 12035233d99SShri Abhyankar (*B)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqAIJ; 121b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 122b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 123b24902e0SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 12435233d99SShri Abhyankar (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqAIJ; 12535233d99SShri Abhyankar (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ; 126e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported"); 127d5f3da31SBarry Smith (*B)->factortype = ftype; 128b24902e0SBarry Smith PetscFunctionReturn(0); 129b24902e0SBarry Smith } 130e631078cSBarry Smith EXTERN_C_END 131b24902e0SBarry Smith 132b24902e0SBarry Smith #undef __FUNCT__ 133ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_inplace" 134ad04f41aSHong Zhang PetscErrorCode MatLUFactorSymbolic_SeqAIJ_inplace(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 135289bc588SBarry Smith { 136416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 137289bc588SBarry Smith IS isicol; 1386849ba73SBarry Smith PetscErrorCode ierr; 1395d0c19d7SBarry Smith const PetscInt *r,*ic; 1405d0c19d7SBarry Smith PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 1411393bc97SHong Zhang PetscInt *bi,*bj,*ajtmp; 1421393bc97SHong Zhang PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 1439e046878SBarry Smith PetscReal f; 1444875ba38SHong Zhang PetscInt nlnk,*lnk,k,**bi_ptr; 145a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1461393bc97SHong Zhang PetscBT lnkbt; 147289bc588SBarry Smith 1483a40ed3dSBarry Smith PetscFunctionBegin; 149e32f2f54SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square"); 1504c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 151ac121b3dSBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 152ac121b3dSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 153289bc588SBarry Smith 154289bc588SBarry Smith /* get new row pointers */ 1551393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 1561393bc97SHong Zhang bi[0] = 0; 1571393bc97SHong Zhang 1581393bc97SHong Zhang /* bdiag is location of diagonal in factor */ 1591393bc97SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1601393bc97SHong Zhang bdiag[0] = 0; 1611393bc97SHong Zhang 1621393bc97SHong Zhang /* linked list for storing column indices of the active row */ 1631393bc97SHong Zhang nlnk = n + 1; 1641393bc97SHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1651393bc97SHong Zhang 16635baf27eSSatish Balay ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 1671393bc97SHong Zhang 1681393bc97SHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 169d3d32019SBarry Smith f = info->fill; 170a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1711393bc97SHong Zhang current_space = free_space; 172289bc588SBarry Smith 173289bc588SBarry Smith for (i=0; i<n; i++) { 1741393bc97SHong Zhang /* copy previous fill into linked list */ 175289bc588SBarry Smith nzi = 0; 1761393bc97SHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 177e32f2f54SBarry 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); 1781393bc97SHong Zhang ajtmp = aj + ai[r[i]]; 179afcc9904SHong Zhang ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1801393bc97SHong Zhang nzi += nlnk; 1811393bc97SHong Zhang 1821393bc97SHong Zhang /* add pivot rows into linked list */ 1831393bc97SHong Zhang row = lnk[n]; 1841393bc97SHong Zhang while (row < i) { 185d1170560SHong Zhang nzbd = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */ 186d1170560SHong Zhang ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 187d1170560SHong Zhang ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 1881393bc97SHong Zhang nzi += nlnk; 1891393bc97SHong Zhang row = lnk[row]; 190289bc588SBarry Smith } 1911393bc97SHong Zhang bi[i+1] = bi[i] + nzi; 1921393bc97SHong Zhang im[i] = nzi; 1931393bc97SHong Zhang 1941393bc97SHong Zhang /* mark bdiag */ 1951393bc97SHong Zhang nzbd = 0; 1961393bc97SHong Zhang nnz = nzi; 1971393bc97SHong Zhang k = lnk[n]; 1981393bc97SHong Zhang while (nnz-- && k < i){ 1991393bc97SHong Zhang nzbd++; 2001393bc97SHong Zhang k = lnk[k]; 2011393bc97SHong Zhang } 2021393bc97SHong Zhang bdiag[i] = bi[i] + nzbd; 2031393bc97SHong Zhang 2041393bc97SHong Zhang /* if free space is not available, make more free space */ 2051393bc97SHong Zhang if (current_space->local_remaining<nzi) { 2061393bc97SHong Zhang nnz = (n - i)*nzi; /* estimated and max additional space needed */ 207a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2081393bc97SHong Zhang reallocs++; 2091393bc97SHong Zhang } 2101393bc97SHong Zhang 2111393bc97SHong Zhang /* copy data into free space, then initialize lnk */ 2121393bc97SHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 2131393bc97SHong Zhang bi_ptr[i] = current_space->array; 2141393bc97SHong Zhang current_space->array += nzi; 2151393bc97SHong Zhang current_space->local_used += nzi; 2161393bc97SHong Zhang current_space->local_remaining -= nzi; 217289bc588SBarry Smith } 2186cf91177SBarry Smith #if defined(PETSC_USE_INFO) 219464e8d44SSatish Balay if (ai[n] != 0) { 2201393bc97SHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 221ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 222ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 223ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 224ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 22505bf559cSSatish Balay } else { 226ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 22705bf559cSSatish Balay } 22863ba0a88SBarry Smith #endif 2292fb3ed76SBarry Smith 230898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 231898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2321987afe7SBarry Smith 2331393bc97SHong Zhang /* destroy list of free space and other temporary array(s) */ 2341393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 235a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); 2361393bc97SHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 23735baf27eSSatish Balay ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 238289bc588SBarry Smith 239289bc588SBarry Smith /* put together the new matrix */ 240719d5645SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 241719d5645SBarry Smith ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 242719d5645SBarry Smith b = (Mat_SeqAIJ*)(B)->data; 243e6b907acSBarry Smith b->free_a = PETSC_TRUE; 244e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 2457c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 2461393bc97SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 2471393bc97SHong Zhang b->j = bj; 2481393bc97SHong Zhang b->i = bi; 2491393bc97SHong Zhang b->diag = bdiag; 250416022c9SBarry Smith b->ilen = 0; 251416022c9SBarry Smith b->imax = 0; 252416022c9SBarry Smith b->row = isrow; 253416022c9SBarry Smith b->col = iscol; 254c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 255c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 25682bf6240SBarry Smith b->icol = isicol; 25787828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2581393bc97SHong Zhang 2591393bc97SHong Zhang /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 260719d5645SBarry Smith ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 2611393bc97SHong Zhang b->maxnz = b->nz = bi[n] ; 2627fd98bd6SLois Curfman McInnes 263d5f3da31SBarry Smith (B)->factortype = MAT_FACTOR_LU; 264719d5645SBarry Smith (B)->info.factor_mallocs = reallocs; 265719d5645SBarry Smith (B)->info.fill_ratio_given = f; 266703deb49SSatish Balay 2679f7d0b68SHong Zhang if (ai[n]) { 268719d5645SBarry Smith (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 269eea2913aSSatish Balay } else { 270719d5645SBarry Smith (B)->info.fill_ratio_needed = 0.0; 271eea2913aSSatish Balay } 272ad04f41aSHong Zhang (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 273f2344125SBarry Smith if (a->inode.size) { 274f2344125SBarry Smith (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 275f2344125SBarry Smith } 2763a40ed3dSBarry Smith PetscFunctionReturn(0); 277289bc588SBarry Smith } 2781393bc97SHong Zhang 279ce3d78c0SShri Abhyankar #undef __FUNCT__ 28035233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ" 28135233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 282ce3d78c0SShri Abhyankar { 283ce3d78c0SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 284ce3d78c0SShri Abhyankar IS isicol; 285ce3d78c0SShri Abhyankar PetscErrorCode ierr; 286ce3d78c0SShri Abhyankar const PetscInt *r,*ic; 287ce3d78c0SShri Abhyankar PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 288ce3d78c0SShri Abhyankar PetscInt *bi,*bj,*ajtmp; 289ce3d78c0SShri Abhyankar PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 290ce3d78c0SShri Abhyankar PetscReal f; 291ce3d78c0SShri Abhyankar PetscInt nlnk,*lnk,k,**bi_ptr; 292ce3d78c0SShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 293ce3d78c0SShri Abhyankar PetscBT lnkbt; 294ce3d78c0SShri Abhyankar 295ce3d78c0SShri Abhyankar PetscFunctionBegin; 2963898ab1dSSatish Balay /* Uncomment the oldatastruct part only while testing new data structure for MatSolve() */ 297a4be958cSHong Zhang /* 298*1a26ff17SHong Zhang PetscBool olddatastruct=PETSC_FALSE; 299acfcf0e5SJed Brown ierr = PetscOptionsGetBool(PETSC_NULL,"-lu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr); 300628f99d7SShri Abhyankar if(olddatastruct){ 301628f99d7SShri Abhyankar ierr = MatLUFactorSymbolic_SeqAIJ_inplace(B,A,isrow,iscol,info);CHKERRQ(ierr); 302628f99d7SShri Abhyankar PetscFunctionReturn(0); 303628f99d7SShri Abhyankar } 304a4be958cSHong Zhang */ 305e32f2f54SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square"); 306ce3d78c0SShri Abhyankar ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 307ce3d78c0SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 308ce3d78c0SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 309ce3d78c0SShri Abhyankar 3101bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 3111bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 3121bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 313a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 3149b48462bSShri Abhyankar 3159b48462bSShri Abhyankar /* linked list for storing column indices of the active row */ 3169b48462bSShri Abhyankar nlnk = n + 1; 3179b48462bSShri Abhyankar ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3189b48462bSShri Abhyankar 3199b48462bSShri Abhyankar ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr); 3209b48462bSShri Abhyankar 3219b48462bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 3229b48462bSShri Abhyankar f = info->fill; 3239b48462bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 3249b48462bSShri Abhyankar current_space = free_space; 3259b48462bSShri Abhyankar 3269b48462bSShri Abhyankar for (i=0; i<n; i++) { 3279b48462bSShri Abhyankar /* copy previous fill into linked list */ 3289b48462bSShri Abhyankar nzi = 0; 3299b48462bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 330e32f2f54SBarry 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); 3319b48462bSShri Abhyankar ajtmp = aj + ai[r[i]]; 3329b48462bSShri Abhyankar ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3339b48462bSShri Abhyankar nzi += nlnk; 3349b48462bSShri Abhyankar 3359b48462bSShri Abhyankar /* add pivot rows into linked list */ 3369b48462bSShri Abhyankar row = lnk[n]; 3379b48462bSShri Abhyankar while (row < i){ 3389b48462bSShri Abhyankar nzbd = bdiag[row] + 1; /* num of entries in the row with column index <= row */ 3399b48462bSShri Abhyankar ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 3409b48462bSShri Abhyankar ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 3419b48462bSShri Abhyankar nzi += nlnk; 3429b48462bSShri Abhyankar row = lnk[row]; 3439b48462bSShri Abhyankar } 3449b48462bSShri Abhyankar bi[i+1] = bi[i] + nzi; 3459b48462bSShri Abhyankar im[i] = nzi; 3469b48462bSShri Abhyankar 3479b48462bSShri Abhyankar /* mark bdiag */ 3489b48462bSShri Abhyankar nzbd = 0; 3499b48462bSShri Abhyankar nnz = nzi; 3509b48462bSShri Abhyankar k = lnk[n]; 3519b48462bSShri Abhyankar while (nnz-- && k < i){ 3529b48462bSShri Abhyankar nzbd++; 3539b48462bSShri Abhyankar k = lnk[k]; 3549b48462bSShri Abhyankar } 3559b48462bSShri Abhyankar bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */ 3569b48462bSShri Abhyankar 3579b48462bSShri Abhyankar /* if free space is not available, make more free space */ 3589b48462bSShri Abhyankar if (current_space->local_remaining<nzi) { 3599b48462bSShri Abhyankar nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */ 3609b48462bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 3619b48462bSShri Abhyankar reallocs++; 3629b48462bSShri Abhyankar } 3639b48462bSShri Abhyankar 3649b48462bSShri Abhyankar /* copy data into free space, then initialize lnk */ 3659b48462bSShri Abhyankar ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 3669b48462bSShri Abhyankar bi_ptr[i] = current_space->array; 3679b48462bSShri Abhyankar current_space->array += nzi; 3689b48462bSShri Abhyankar current_space->local_used += nzi; 3699b48462bSShri Abhyankar current_space->local_remaining -= nzi; 3709b48462bSShri Abhyankar } 3719b48462bSShri Abhyankar 3729b48462bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3739b48462bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3749b48462bSShri Abhyankar 3759263d837SHong Zhang /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 3769b48462bSShri Abhyankar ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 3772ce24eb6SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 3789b48462bSShri Abhyankar ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 3799b48462bSShri Abhyankar ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 3809b48462bSShri Abhyankar 3819b48462bSShri Abhyankar /* put together the new matrix */ 3829b48462bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 3839b48462bSShri Abhyankar ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 3849b48462bSShri Abhyankar b = (Mat_SeqAIJ*)(B)->data; 3859b48462bSShri Abhyankar b->free_a = PETSC_TRUE; 3869b48462bSShri Abhyankar b->free_ij = PETSC_TRUE; 3879b48462bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 3889b48462bSShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 3899b48462bSShri Abhyankar b->j = bj; 3909b48462bSShri Abhyankar b->i = bi; 3919b48462bSShri Abhyankar b->diag = bdiag; 3929b48462bSShri Abhyankar b->ilen = 0; 3939b48462bSShri Abhyankar b->imax = 0; 3949b48462bSShri Abhyankar b->row = isrow; 3959b48462bSShri Abhyankar b->col = iscol; 3969b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 3979b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 3989b48462bSShri Abhyankar b->icol = isicol; 3999b48462bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 4009b48462bSShri Abhyankar 4019b48462bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 4029b48462bSShri Abhyankar ierr = PetscLogObjectMemory(B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 4039b48462bSShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 404d5f3da31SBarry Smith B->factortype = MAT_FACTOR_LU; 405f284f12aSHong Zhang B->info.factor_mallocs = reallocs; 406f284f12aSHong Zhang B->info.fill_ratio_given = f; 4079b48462bSShri Abhyankar 4089b48462bSShri Abhyankar if (ai[n]) { 409f284f12aSHong Zhang B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 4109b48462bSShri Abhyankar } else { 411f284f12aSHong Zhang B->info.fill_ratio_needed = 0.0; 4129b48462bSShri Abhyankar } 4139263d837SHong Zhang #if defined(PETSC_USE_INFO) 4149263d837SHong Zhang if (ai[n] != 0) { 4159263d837SHong Zhang PetscReal af = B->info.fill_ratio_needed; 4169263d837SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 4179263d837SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 4189263d837SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 4199263d837SHong Zhang ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 4209263d837SHong Zhang } else { 4219263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 4229263d837SHong Zhang } 4239263d837SHong Zhang #endif 42435233d99SShri Abhyankar B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 425f2344125SBarry Smith if (a->inode.size) { 426d3ac4fa3SBarry Smith B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 427d3ac4fa3SBarry Smith } 4289b48462bSShri Abhyankar PetscFunctionReturn(0); 4299b48462bSShri Abhyankar } 4309b48462bSShri Abhyankar 4315b5bc046SBarry Smith /* 4325b5bc046SBarry Smith Trouble in factorization, should we dump the original matrix? 4335b5bc046SBarry Smith */ 4345b5bc046SBarry Smith #undef __FUNCT__ 4355b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix" 4365b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A) 4375b5bc046SBarry Smith { 4385b5bc046SBarry Smith PetscErrorCode ierr; 439ace3abfcSBarry Smith PetscBool flg = PETSC_FALSE; 4405b5bc046SBarry Smith 4415b5bc046SBarry Smith PetscFunctionBegin; 442acfcf0e5SJed Brown ierr = PetscOptionsGetBool(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr); 4435b5bc046SBarry Smith if (flg) { 4445b5bc046SBarry Smith PetscViewer viewer; 4455b5bc046SBarry Smith char filename[PETSC_MAX_PATH_LEN]; 4465b5bc046SBarry Smith 4475b5bc046SBarry Smith ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr); 4487adad957SLisandro Dalcin ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 4495b5bc046SBarry Smith ierr = MatView(A,viewer);CHKERRQ(ierr); 4506bf464f9SBarry Smith ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 4515b5bc046SBarry Smith } 4525b5bc046SBarry Smith PetscFunctionReturn(0); 4535b5bc046SBarry Smith } 4545b5bc046SBarry Smith 455c9c72f8fSHong Zhang #undef __FUNCT__ 45635233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ" 45735233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 458c9c72f8fSHong Zhang { 459c9c72f8fSHong Zhang Mat C=B; 460c9c72f8fSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 461c9c72f8fSHong Zhang IS isrow = b->row,isicol = b->icol; 462c9c72f8fSHong Zhang PetscErrorCode ierr; 463c9c72f8fSHong Zhang const PetscInt *r,*ic,*ics; 464bbd65245SShri Abhyankar const PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag; 465bbd65245SShri Abhyankar PetscInt i,j,k,nz,nzL,row,*pj; 466bbd65245SShri Abhyankar const PetscInt *ajtmp,*bjtmp; 467bbd65245SShri Abhyankar MatScalar *rtmp,*pc,multiplier,*pv; 468bbd65245SShri Abhyankar const MatScalar *aa=a->a,*v; 469ace3abfcSBarry Smith PetscBool row_identity,col_identity; 470d90ac83dSHong Zhang FactorShiftCtx sctx; 4714f81c4b7SBarry Smith const PetscInt *ddiag; 472d4ad06f7SHong Zhang PetscReal rs; 473d4ad06f7SHong Zhang MatScalar d; 474d4ad06f7SHong Zhang 475c9c72f8fSHong Zhang PetscFunctionBegin; 476d6e5152cSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 477d90ac83dSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 478d4ad06f7SHong Zhang 479f4db908eSBarry Smith if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 480d4ad06f7SHong Zhang ddiag = a->diag; 481d4ad06f7SHong Zhang sctx.shift_top = info->zeropivot; 482d4ad06f7SHong Zhang for (i=0; i<n; i++) { 483d4ad06f7SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 484d4ad06f7SHong Zhang d = (aa)[ddiag[i]]; 485d4ad06f7SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 486d4ad06f7SHong Zhang v = aa+ai[i]; 487d4ad06f7SHong Zhang nz = ai[i+1] - ai[i]; 488d4ad06f7SHong Zhang for (j=0; j<nz; j++) 489d4ad06f7SHong Zhang rs += PetscAbsScalar(v[j]); 490d4ad06f7SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 491d4ad06f7SHong Zhang } 492d4ad06f7SHong Zhang sctx.shift_top *= 1.1; 493d4ad06f7SHong Zhang sctx.nshift_max = 5; 494d4ad06f7SHong Zhang sctx.shift_lo = 0.; 495d4ad06f7SHong Zhang sctx.shift_hi = 1.; 496d4ad06f7SHong Zhang } 497d4ad06f7SHong Zhang 498c9c72f8fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 499c9c72f8fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 500c9c72f8fSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 501c9c72f8fSHong Zhang ics = ic; 502c9c72f8fSHong Zhang 503d4ad06f7SHong Zhang do { 50407b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 505c9c72f8fSHong Zhang for (i=0; i<n; i++){ 506c9c72f8fSHong Zhang /* zero rtmp */ 507c9c72f8fSHong Zhang /* L part */ 508c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 509c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 510c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 511c9c72f8fSHong Zhang 512c9c72f8fSHong Zhang /* U part */ 51362fbd6afSShri Abhyankar nz = bdiag[i]-bdiag[i+1]; 51462fbd6afSShri Abhyankar bjtmp = bj + bdiag[i+1]+1; 515813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 516813a1f2bSShri Abhyankar 517813a1f2bSShri Abhyankar /* load in initial (unfactored row) */ 518813a1f2bSShri Abhyankar nz = ai[r[i]+1] - ai[r[i]]; 519813a1f2bSShri Abhyankar ajtmp = aj + ai[r[i]]; 520813a1f2bSShri Abhyankar v = aa + ai[r[i]]; 521813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 522813a1f2bSShri Abhyankar rtmp[ics[ajtmp[j]]] = v[j]; 523813a1f2bSShri Abhyankar } 524813a1f2bSShri Abhyankar /* ZeropivotApply() */ 52598a93161SHong Zhang rtmp[i] += sctx.shift_amount; /* shift the diagonal of the matrix */ 526813a1f2bSShri Abhyankar 527813a1f2bSShri Abhyankar /* elimination */ 528813a1f2bSShri Abhyankar bjtmp = bj + bi[i]; 529813a1f2bSShri Abhyankar row = *bjtmp++; 530813a1f2bSShri Abhyankar nzL = bi[i+1] - bi[i]; 531f268cba8SShri Abhyankar for(k=0; k < nzL;k++) { 532813a1f2bSShri Abhyankar pc = rtmp + row; 533813a1f2bSShri Abhyankar if (*pc != 0.0) { 534813a1f2bSShri Abhyankar pv = b->a + bdiag[row]; 535813a1f2bSShri Abhyankar multiplier = *pc * (*pv); 536813a1f2bSShri Abhyankar *pc = multiplier; 53762fbd6afSShri Abhyankar pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 53862fbd6afSShri Abhyankar pv = b->a + bdiag[row+1]+1; 53962fbd6afSShri Abhyankar nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 540813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 541813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 542813a1f2bSShri Abhyankar } 543f268cba8SShri Abhyankar row = *bjtmp++; 544813a1f2bSShri Abhyankar } 545813a1f2bSShri Abhyankar 546813a1f2bSShri Abhyankar /* finished row so stick it into b->a */ 547813a1f2bSShri Abhyankar rs = 0.0; 548813a1f2bSShri Abhyankar /* L part */ 549813a1f2bSShri Abhyankar pv = b->a + bi[i] ; 550813a1f2bSShri Abhyankar pj = b->j + bi[i] ; 551813a1f2bSShri Abhyankar nz = bi[i+1] - bi[i]; 552813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 553813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 554813a1f2bSShri Abhyankar } 555813a1f2bSShri Abhyankar 556813a1f2bSShri Abhyankar /* U part */ 55762fbd6afSShri Abhyankar pv = b->a + bdiag[i+1]+1; 55862fbd6afSShri Abhyankar pj = b->j + bdiag[i+1]+1; 55962fbd6afSShri Abhyankar nz = bdiag[i] - bdiag[i+1]-1; 560813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 561813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 562813a1f2bSShri Abhyankar } 563813a1f2bSShri Abhyankar 564813a1f2bSShri Abhyankar sctx.rs = rs; 565813a1f2bSShri Abhyankar sctx.pv = rtmp[i]; 566d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 56707b50cabSHong Zhang if(sctx.newshift) break; /* break for-loop */ 56807b50cabSHong Zhang rtmp[i] = sctx.pv; /* sctx.pv might be updated in the case of MAT_SHIFT_INBLOCKS */ 569813a1f2bSShri Abhyankar 570813a1f2bSShri Abhyankar /* Mark diagonal and invert diagonal for simplier triangular solves */ 571813a1f2bSShri Abhyankar pv = b->a + bdiag[i]; 572813a1f2bSShri Abhyankar *pv = 1.0/rtmp[i]; 573813a1f2bSShri Abhyankar 574813a1f2bSShri Abhyankar } /* endof for (i=0; i<n; i++){ */ 575813a1f2bSShri Abhyankar 5768ff23777SHong Zhang /* MatPivotRefine() */ 57707b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){ 578813a1f2bSShri Abhyankar /* 579813a1f2bSShri Abhyankar * if no shift in this attempt & shifting & started shifting & can refine, 580813a1f2bSShri Abhyankar * then try lower shift 581813a1f2bSShri Abhyankar */ 582813a1f2bSShri Abhyankar sctx.shift_hi = sctx.shift_fraction; 583813a1f2bSShri Abhyankar sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 584813a1f2bSShri Abhyankar sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 58507b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 586813a1f2bSShri Abhyankar sctx.nshift++; 587813a1f2bSShri Abhyankar } 58807b50cabSHong Zhang } while (sctx.newshift); 589813a1f2bSShri Abhyankar 590813a1f2bSShri Abhyankar ierr = PetscFree(rtmp);CHKERRQ(ierr); 591813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 592813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 593d3ac4fa3SBarry Smith 594813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 595813a1f2bSShri Abhyankar ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 596813a1f2bSShri Abhyankar if (row_identity && col_identity) { 59735233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 598813a1f2bSShri Abhyankar } else { 59935233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ; 600813a1f2bSShri Abhyankar } 60135233d99SShri Abhyankar C->ops->solveadd = MatSolveAdd_SeqAIJ; 60235233d99SShri Abhyankar C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 60335233d99SShri Abhyankar C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 60435233d99SShri Abhyankar C->ops->matsolve = MatMatSolve_SeqAIJ; 605813a1f2bSShri Abhyankar C->assembled = PETSC_TRUE; 606813a1f2bSShri Abhyankar C->preallocated = PETSC_TRUE; 607813a1f2bSShri Abhyankar ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 608813a1f2bSShri Abhyankar 60914d2772eSHong Zhang /* MatShiftView(A,info,&sctx) */ 610813a1f2bSShri Abhyankar if (sctx.nshift){ 611f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 612813a1f2bSShri 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); 613f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 614813a1f2bSShri Abhyankar ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 615f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){ 616915743fcSHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr); 617813a1f2bSShri Abhyankar } 618813a1f2bSShri Abhyankar } 619d3ac4fa3SBarry Smith ierr = Mat_CheckInode_FactorLU(C,PETSC_FALSE);CHKERRQ(ierr); 620813a1f2bSShri Abhyankar PetscFunctionReturn(0); 621813a1f2bSShri Abhyankar } 622813a1f2bSShri Abhyankar 6234a2ae208SSatish Balay #undef __FUNCT__ 624ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_inplace" 625ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info) 626289bc588SBarry Smith { 627719d5645SBarry Smith Mat C=B; 628416022c9SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 62982bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 6306849ba73SBarry Smith PetscErrorCode ierr; 6315d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 632d278edc7SBarry Smith PetscInt nz,row,i,j,n=A->rmap->n,diag; 633d278edc7SBarry Smith const PetscInt *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 634d278edc7SBarry Smith const PetscInt *ajtmp,*bjtmp,*diag_offset = b->diag,*pj; 635d278edc7SBarry Smith MatScalar *pv,*rtmp,*pc,multiplier,d; 636d278edc7SBarry Smith const MatScalar *v,*aa=a->a; 63733f9893dSHong Zhang PetscReal rs=0.0; 638d90ac83dSHong Zhang FactorShiftCtx sctx; 6394f81c4b7SBarry Smith const PetscInt *ddiag; 640ace3abfcSBarry Smith PetscBool row_identity, col_identity; 641289bc588SBarry Smith 6423a40ed3dSBarry Smith PetscFunctionBegin; 643504a3fadSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 644504a3fadSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 645ada7143aSSatish Balay 646f4db908eSBarry Smith if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 64742898933SHong Zhang ddiag = a->diag; 6489f7d0b68SHong Zhang sctx.shift_top = info->zeropivot; 6496cc28720Svictorle for (i=0; i<n; i++) { 6509f95998fSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 6519f7d0b68SHong Zhang d = (aa)[ddiag[i]]; 6521a890ab1SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 6539f7d0b68SHong Zhang v = aa+ai[i]; 6549f7d0b68SHong Zhang nz = ai[i+1] - ai[i]; 655e24cfe64SHong Zhang for (j=0; j<nz; j++) 6561a890ab1SHong Zhang rs += PetscAbsScalar(v[j]); 6571a890ab1SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 6586cc28720Svictorle } 659118f5789SBarry Smith sctx.shift_top *= 1.1; 660d2276718SHong Zhang sctx.nshift_max = 5; 661d2276718SHong Zhang sctx.shift_lo = 0.; 662d2276718SHong Zhang sctx.shift_hi = 1.; 663a0a356efSHong Zhang } 664a0a356efSHong Zhang 665504a3fadSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 666504a3fadSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 667504a3fadSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 668504a3fadSHong Zhang ics = ic; 669504a3fadSHong Zhang 670085256b3SBarry Smith do { 67107b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 672289bc588SBarry Smith for (i=0; i<n; i++){ 673b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 674b3bf805bSHong Zhang bjtmp = bj + bi[i]; 6759f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 676289bc588SBarry Smith 677289bc588SBarry Smith /* load in initial (unfactored row) */ 6789f7d0b68SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 6799f7d0b68SHong Zhang ajtmp = aj + ai[r[i]]; 6809f7d0b68SHong Zhang v = aa + ai[r[i]]; 681085256b3SBarry Smith for (j=0; j<nz; j++) { 682b3bf805bSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 683085256b3SBarry Smith } 684d2276718SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 685289bc588SBarry Smith 686b3bf805bSHong Zhang row = *bjtmp++; 687f2582111SSatish Balay while (row < i) { 6888c37ef55SBarry Smith pc = rtmp + row; 6898c37ef55SBarry Smith if (*pc != 0.0) { 690010a20caSHong Zhang pv = b->a + diag_offset[row]; 691010a20caSHong Zhang pj = b->j + diag_offset[row] + 1; 69235aab85fSBarry Smith multiplier = *pc / *pv++; 6938c37ef55SBarry Smith *pc = multiplier; 694b3bf805bSHong Zhang nz = bi[row+1] - diag_offset[row] - 1; 6959f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 696dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 697289bc588SBarry Smith } 698b3bf805bSHong Zhang row = *bjtmp++; 699289bc588SBarry Smith } 700416022c9SBarry Smith /* finished row so stick it into b->a */ 701b3bf805bSHong Zhang pv = b->a + bi[i] ; 702b3bf805bSHong Zhang pj = b->j + bi[i] ; 703b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 704b3bf805bSHong Zhang diag = diag_offset[i] - bi[i]; 705e57ff13aSBarry Smith rs = 0.0; 706d3d32019SBarry Smith for (j=0; j<nz; j++) { 7079f7d0b68SHong Zhang pv[j] = rtmp[pj[j]]; 7089f7d0b68SHong Zhang rs += PetscAbsScalar(pv[j]); 709d3d32019SBarry Smith } 710e57ff13aSBarry Smith rs -= PetscAbsScalar(pv[diag]); 711d2276718SHong Zhang 712d2276718SHong Zhang sctx.rs = rs; 713d2276718SHong Zhang sctx.pv = pv[diag]; 714d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 71507b50cabSHong Zhang if (sctx.newshift) break; 716504a3fadSHong Zhang pv[diag] = sctx.pv; 71735aab85fSBarry Smith } 718d2276718SHong Zhang 71907b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 7206cc28720Svictorle /* 7216c037d1bSvictorle * if no shift in this attempt & shifting & started shifting & can refine, 7226cc28720Svictorle * then try lower shift 7236cc28720Svictorle */ 7240481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 7250481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 7260481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 72707b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 728d2276718SHong Zhang sctx.nshift++; 7296cc28720Svictorle } 73007b50cabSHong Zhang } while (sctx.newshift); 7310f11f9aeSSatish Balay 73235aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 73335aab85fSBarry Smith for (i=0; i<n; i++) { 734010a20caSHong Zhang b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]]; 73535aab85fSBarry Smith } 736606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 73778b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 73878b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 739d3ac4fa3SBarry Smith 7408d8c7c43SBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 7418d8c7c43SBarry Smith ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 7428d8c7c43SBarry Smith if (row_identity && col_identity) { 743ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_inplace; 7448d8c7c43SBarry Smith } else { 745ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_inplace; 746db4efbfdSBarry Smith } 747ad04f41aSHong Zhang C->ops->solveadd = MatSolveAdd_SeqAIJ_inplace; 748ad04f41aSHong Zhang C->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 749ad04f41aSHong Zhang C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 750ad04f41aSHong Zhang C->ops->matsolve = MatMatSolve_SeqAIJ_inplace; 751c456f294SBarry Smith C->assembled = PETSC_TRUE; 7525c9eb25fSBarry Smith C->preallocated = PETSC_TRUE; 753d0f46423SBarry Smith ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 754d2276718SHong Zhang if (sctx.nshift){ 755f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 7560481f469SBarry 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); 757f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 758e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 7596cc28720Svictorle } 7600697b6caSHong Zhang } 761d3ac4fa3SBarry Smith (C)->ops->solve = MatSolve_SeqAIJ_inplace; 762d3ac4fa3SBarry Smith (C)->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 763d3ac4fa3SBarry Smith ierr = Mat_CheckInode(C,PETSC_FALSE);CHKERRQ(ierr); 7643a40ed3dSBarry Smith PetscFunctionReturn(0); 765289bc588SBarry Smith } 7660889b5dcSvictorle 767137fb511SHong Zhang /* 768137fb511SHong Zhang This routine implements inplace ILU(0) with row or/and column permutations. 769137fb511SHong Zhang Input: 770137fb511SHong Zhang A - original matrix 771137fb511SHong Zhang Output; 772137fb511SHong Zhang A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i] 773137fb511SHong Zhang a->j (col index) is permuted by the inverse of colperm, then sorted 774137fb511SHong Zhang a->a reordered accordingly with a->j 775137fb511SHong Zhang a->diag (ptr to diagonal elements) is updated. 776137fb511SHong Zhang */ 777137fb511SHong Zhang #undef __FUNCT__ 778137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm" 7790481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info) 780137fb511SHong Zhang { 781b051339dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 782b051339dSHong Zhang IS isrow = a->row,isicol = a->icol; 783137fb511SHong Zhang PetscErrorCode ierr; 7845d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 7855d0c19d7SBarry Smith PetscInt i,j,n=A->rmap->n,*ai=a->i,*aj=a->j; 7865d0c19d7SBarry Smith PetscInt *ajtmp,nz,row; 787b051339dSHong Zhang PetscInt *diag = a->diag,nbdiag,*pj; 788a77337e4SBarry Smith PetscScalar *rtmp,*pc,multiplier,d; 789504a3fadSHong Zhang MatScalar *pv,*v; 790137fb511SHong Zhang PetscReal rs; 791d90ac83dSHong Zhang FactorShiftCtx sctx; 792504a3fadSHong Zhang const MatScalar *aa=a->a,*vtmp; 793137fb511SHong Zhang 794137fb511SHong Zhang PetscFunctionBegin; 795e32f2f54SBarry Smith if (A != B) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address"); 796504a3fadSHong Zhang 797504a3fadSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 798504a3fadSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 799504a3fadSHong Zhang 800504a3fadSHong Zhang if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 801504a3fadSHong Zhang const PetscInt *ddiag = a->diag; 802504a3fadSHong Zhang sctx.shift_top = info->zeropivot; 803504a3fadSHong Zhang for (i=0; i<n; i++) { 804504a3fadSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 805504a3fadSHong Zhang d = (aa)[ddiag[i]]; 806504a3fadSHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 807504a3fadSHong Zhang vtmp = aa+ai[i]; 808504a3fadSHong Zhang nz = ai[i+1] - ai[i]; 809504a3fadSHong Zhang for (j=0; j<nz; j++) 810504a3fadSHong Zhang rs += PetscAbsScalar(vtmp[j]); 811504a3fadSHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 812504a3fadSHong Zhang } 813504a3fadSHong Zhang sctx.shift_top *= 1.1; 814504a3fadSHong Zhang sctx.nshift_max = 5; 815504a3fadSHong Zhang sctx.shift_lo = 0.; 816504a3fadSHong Zhang sctx.shift_hi = 1.; 817504a3fadSHong Zhang } 818504a3fadSHong Zhang 819137fb511SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 820137fb511SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 821137fb511SHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr); 822137fb511SHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 823b051339dSHong Zhang ics = ic; 824137fb511SHong Zhang 825504a3fadSHong Zhang #if defined(MV) 82675567043SBarry Smith sctx.shift_top = 0.; 827e60cf9a0SBarry Smith sctx.nshift_max = 0; 82875567043SBarry Smith sctx.shift_lo = 0.; 82975567043SBarry Smith sctx.shift_hi = 0.; 83075567043SBarry Smith sctx.shift_fraction = 0.; 831137fb511SHong Zhang 832f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 83375567043SBarry Smith sctx.shift_top = 0.; 834137fb511SHong Zhang for (i=0; i<n; i++) { 835137fb511SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 836b051339dSHong Zhang d = (a->a)[diag[i]]; 837137fb511SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 838b051339dSHong Zhang v = a->a+ai[i]; 839b051339dSHong Zhang nz = ai[i+1] - ai[i]; 840137fb511SHong Zhang for (j=0; j<nz; j++) 841137fb511SHong Zhang rs += PetscAbsScalar(v[j]); 842137fb511SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 843137fb511SHong Zhang } 844137fb511SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 845137fb511SHong Zhang sctx.shift_top *= 1.1; 846137fb511SHong Zhang sctx.nshift_max = 5; 847137fb511SHong Zhang sctx.shift_lo = 0.; 848137fb511SHong Zhang sctx.shift_hi = 1.; 849137fb511SHong Zhang } 850137fb511SHong Zhang 85175567043SBarry Smith sctx.shift_amount = 0.; 852137fb511SHong Zhang sctx.nshift = 0; 853504a3fadSHong Zhang #endif 854504a3fadSHong Zhang 855137fb511SHong Zhang do { 85607b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 857137fb511SHong Zhang for (i=0; i<n; i++){ 858b051339dSHong Zhang /* load in initial unfactored row */ 859b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 860b051339dSHong Zhang ajtmp = aj + ai[r[i]]; 861b051339dSHong Zhang v = a->a + ai[r[i]]; 862137fb511SHong Zhang /* sort permuted ajtmp and values v accordingly */ 863b051339dSHong Zhang for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]]; 864137fb511SHong Zhang ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr); 865137fb511SHong Zhang 866b051339dSHong Zhang diag[r[i]] = ai[r[i]]; 867137fb511SHong Zhang for (j=0; j<nz; j++) { 868137fb511SHong Zhang rtmp[ajtmp[j]] = v[j]; 869b051339dSHong Zhang if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */ 870137fb511SHong Zhang } 871137fb511SHong Zhang rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 872137fb511SHong Zhang 873b051339dSHong Zhang row = *ajtmp++; 874137fb511SHong Zhang while (row < i) { 875137fb511SHong Zhang pc = rtmp + row; 876137fb511SHong Zhang if (*pc != 0.0) { 877b051339dSHong Zhang pv = a->a + diag[r[row]]; 878b051339dSHong Zhang pj = aj + diag[r[row]] + 1; 879137fb511SHong Zhang 880137fb511SHong Zhang multiplier = *pc / *pv++; 881137fb511SHong Zhang *pc = multiplier; 882b051339dSHong Zhang nz = ai[r[row]+1] - diag[r[row]] - 1; 883b051339dSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 884dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 885137fb511SHong Zhang } 886b051339dSHong Zhang row = *ajtmp++; 887137fb511SHong Zhang } 888b051339dSHong Zhang /* finished row so overwrite it onto a->a */ 889b051339dSHong Zhang pv = a->a + ai[r[i]] ; 890b051339dSHong Zhang pj = aj + ai[r[i]] ; 891b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 892b051339dSHong Zhang nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */ 893137fb511SHong Zhang 894137fb511SHong Zhang rs = 0.0; 895137fb511SHong Zhang for (j=0; j<nz; j++) { 896b051339dSHong Zhang pv[j] = rtmp[pj[j]]; 897b051339dSHong Zhang if (j != nbdiag) rs += PetscAbsScalar(pv[j]); 898137fb511SHong Zhang } 899137fb511SHong Zhang 900137fb511SHong Zhang sctx.rs = rs; 901b051339dSHong Zhang sctx.pv = pv[nbdiag]; 902d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 90307b50cabSHong Zhang if (sctx.newshift) break; 904504a3fadSHong Zhang pv[nbdiag] = sctx.pv; 905137fb511SHong Zhang } 906137fb511SHong Zhang 90707b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 908137fb511SHong Zhang /* 909137fb511SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 910137fb511SHong Zhang * then try lower shift 911137fb511SHong Zhang */ 9120481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 9130481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 9140481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 91507b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 916137fb511SHong Zhang sctx.nshift++; 917137fb511SHong Zhang } 91807b50cabSHong Zhang } while (sctx.newshift); 919137fb511SHong Zhang 920137fb511SHong Zhang /* invert diagonal entries for simplier triangular solves */ 921137fb511SHong Zhang for (i=0; i<n; i++) { 922b051339dSHong Zhang a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]]; 923137fb511SHong Zhang } 924137fb511SHong Zhang 925137fb511SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 926137fb511SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 927137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 928b051339dSHong Zhang A->ops->solve = MatSolve_SeqAIJ_InplaceWithPerm; 929ad04f41aSHong Zhang A->ops->solveadd = MatSolveAdd_SeqAIJ_inplace; 930ad04f41aSHong Zhang A->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 931ad04f41aSHong Zhang A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 932b051339dSHong Zhang A->assembled = PETSC_TRUE; 9335c9eb25fSBarry Smith A->preallocated = PETSC_TRUE; 934d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr); 935137fb511SHong Zhang if (sctx.nshift){ 936f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 9370481f469SBarry 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); 938f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 939e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 940137fb511SHong Zhang } 941137fb511SHong Zhang } 942137fb511SHong Zhang PetscFunctionReturn(0); 943137fb511SHong Zhang } 944137fb511SHong Zhang 9450a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9464a2ae208SSatish Balay #undef __FUNCT__ 9474a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ" 9480481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info) 949da3a660dSBarry Smith { 950dfbe8321SBarry Smith PetscErrorCode ierr; 951416022c9SBarry Smith Mat C; 952416022c9SBarry Smith 9533a40ed3dSBarry Smith PetscFunctionBegin; 9542692d6eeSBarry Smith ierr = MatGetFactor(A,MATSOLVERPETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr); 955719d5645SBarry Smith ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr); 956719d5645SBarry Smith ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr); 957db4efbfdSBarry Smith A->ops->solve = C->ops->solve; 958db4efbfdSBarry Smith A->ops->solvetranspose = C->ops->solvetranspose; 959eb6b5d47SBarry Smith ierr = MatHeaderMerge(A,C);CHKERRQ(ierr); 96052e6d16bSBarry Smith ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr); 9613a40ed3dSBarry Smith PetscFunctionReturn(0); 962da3a660dSBarry Smith } 9630a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9641d098868SHong Zhang 965889a8dedSBarry Smith 9664a2ae208SSatish Balay #undef __FUNCT__ 967ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_inplace" 968ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx) 9698c37ef55SBarry Smith { 970416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 971416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 9726849ba73SBarry Smith PetscErrorCode ierr; 9735d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 9745d0c19d7SBarry Smith PetscInt nz; 9755d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 976dd6ea824SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 977d9fead3dSBarry Smith const PetscScalar *b; 978dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 9798c37ef55SBarry Smith 9803a40ed3dSBarry Smith PetscFunctionBegin; 9813a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 98291bf9adeSBarry Smith 9833649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 9841ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 985416022c9SBarry Smith tmp = a->solve_work; 9868c37ef55SBarry Smith 9873b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 9883b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 9898c37ef55SBarry Smith 9908c37ef55SBarry Smith /* forward solve the lower triangular */ 9918c37ef55SBarry Smith tmp[0] = b[*r++]; 992010a20caSHong Zhang tmps = tmp; 9938c37ef55SBarry Smith for (i=1; i<n; i++) { 994010a20caSHong Zhang v = aa + ai[i] ; 995010a20caSHong Zhang vi = aj + ai[i] ; 99653e7425aSBarry Smith nz = a->diag[i] - ai[i]; 99753e7425aSBarry Smith sum = b[*r++]; 998003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 9998c37ef55SBarry Smith tmp[i] = sum; 10008c37ef55SBarry Smith } 10018c37ef55SBarry Smith 10028c37ef55SBarry Smith /* backward solve the upper triangular */ 10038c37ef55SBarry Smith for (i=n-1; i>=0; i--){ 1004010a20caSHong Zhang v = aa + a->diag[i] + 1; 1005010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1006416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 10078c37ef55SBarry Smith sum = tmp[i]; 1008003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1009010a20caSHong Zhang x[*c--] = tmp[i] = sum*aa[a->diag[i]]; 10108c37ef55SBarry Smith } 10118c37ef55SBarry Smith 10123b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10133b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10143649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 10151ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1016dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 10173a40ed3dSBarry Smith PetscFunctionReturn(0); 10188c37ef55SBarry Smith } 1019026e39d0SSatish Balay 10202bebee5dSHong Zhang #undef __FUNCT__ 1021ad04f41aSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ_inplace" 1022ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X) 10232bebee5dSHong Zhang { 10242bebee5dSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10252bebee5dSHong Zhang IS iscol = a->col,isrow = a->row; 10262bebee5dSHong Zhang PetscErrorCode ierr; 10275d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10285d0c19d7SBarry Smith PetscInt nz,neq; 10295d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 1030dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1031dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1032ace3abfcSBarry Smith PetscBool bisdense,xisdense; 10332bebee5dSHong Zhang 10342bebee5dSHong Zhang PetscFunctionBegin; 10352bebee5dSHong Zhang if (!n) PetscFunctionReturn(0); 10362bebee5dSHong Zhang 1037db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1038e32f2f54SBarry Smith if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 1039db4efbfdSBarry Smith ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1040e32f2f54SBarry Smith if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 1041db4efbfdSBarry Smith 10422bebee5dSHong Zhang ierr = MatGetArray(B,&b);CHKERRQ(ierr); 10432bebee5dSHong Zhang ierr = MatGetArray(X,&x);CHKERRQ(ierr); 10442bebee5dSHong Zhang 10452bebee5dSHong Zhang tmp = a->solve_work; 10462bebee5dSHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10472bebee5dSHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 10482bebee5dSHong Zhang 1049a36b22ccSSatish Balay for (neq=0; neq<B->cmap->n; neq++){ 10502bebee5dSHong Zhang /* forward solve the lower triangular */ 10512bebee5dSHong Zhang tmp[0] = b[r[0]]; 10522bebee5dSHong Zhang tmps = tmp; 10532bebee5dSHong Zhang for (i=1; i<n; i++) { 10542bebee5dSHong Zhang v = aa + ai[i] ; 10552bebee5dSHong Zhang vi = aj + ai[i] ; 10562bebee5dSHong Zhang nz = a->diag[i] - ai[i]; 10572bebee5dSHong Zhang sum = b[r[i]]; 1058003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10592bebee5dSHong Zhang tmp[i] = sum; 10602bebee5dSHong Zhang } 10612bebee5dSHong Zhang /* backward solve the upper triangular */ 10622bebee5dSHong Zhang for (i=n-1; i>=0; i--){ 10632bebee5dSHong Zhang v = aa + a->diag[i] + 1; 10642bebee5dSHong Zhang vi = aj + a->diag[i] + 1; 10652bebee5dSHong Zhang nz = ai[i+1] - a->diag[i] - 1; 10662bebee5dSHong Zhang sum = tmp[i]; 1067003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10682bebee5dSHong Zhang x[c[i]] = tmp[i] = sum*aa[a->diag[i]]; 10692bebee5dSHong Zhang } 10702bebee5dSHong Zhang 10712bebee5dSHong Zhang b += n; 10722bebee5dSHong Zhang x += n; 10732bebee5dSHong Zhang } 10742bebee5dSHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10752bebee5dSHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10762bebee5dSHong Zhang ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 10772bebee5dSHong Zhang ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 1078dc0b31edSSatish Balay ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 10792bebee5dSHong Zhang PetscFunctionReturn(0); 10802bebee5dSHong Zhang } 10812bebee5dSHong Zhang 1082137fb511SHong Zhang #undef __FUNCT__ 108335233d99SShri Abhyankar #define __FUNCT__ "MatMatSolve_SeqAIJ" 108435233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X) 10859bd0847aSShri Abhyankar { 10869bd0847aSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10879bd0847aSShri Abhyankar IS iscol = a->col,isrow = a->row; 10889bd0847aSShri Abhyankar PetscErrorCode ierr; 10899bd0847aSShri Abhyankar PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 10909bd0847aSShri Abhyankar PetscInt nz,neq; 10919bd0847aSShri Abhyankar const PetscInt *rout,*cout,*r,*c; 10929bd0847aSShri Abhyankar PetscScalar *x,*b,*tmp,sum; 10939bd0847aSShri Abhyankar const MatScalar *aa = a->a,*v; 1094ace3abfcSBarry Smith PetscBool bisdense,xisdense; 10959bd0847aSShri Abhyankar 10969bd0847aSShri Abhyankar PetscFunctionBegin; 10979bd0847aSShri Abhyankar if (!n) PetscFunctionReturn(0); 10989bd0847aSShri Abhyankar 10999bd0847aSShri Abhyankar ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1100e32f2f54SBarry Smith if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 11019bd0847aSShri Abhyankar ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1102e32f2f54SBarry Smith if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 11039bd0847aSShri Abhyankar 11049bd0847aSShri Abhyankar ierr = MatGetArray(B,&b);CHKERRQ(ierr); 11059bd0847aSShri Abhyankar ierr = MatGetArray(X,&x);CHKERRQ(ierr); 11069bd0847aSShri Abhyankar 11079bd0847aSShri Abhyankar tmp = a->solve_work; 11089bd0847aSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 11099bd0847aSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 11109bd0847aSShri Abhyankar 11119bd0847aSShri Abhyankar for (neq=0; neq<B->cmap->n; neq++){ 11129bd0847aSShri Abhyankar /* forward solve the lower triangular */ 11139bd0847aSShri Abhyankar tmp[0] = b[r[0]]; 11149bd0847aSShri Abhyankar v = aa; 11159bd0847aSShri Abhyankar vi = aj; 11169bd0847aSShri Abhyankar for (i=1; i<n; i++) { 11179bd0847aSShri Abhyankar nz = ai[i+1] - ai[i]; 11189bd0847aSShri Abhyankar sum = b[r[i]]; 11199bd0847aSShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 11209bd0847aSShri Abhyankar tmp[i] = sum; 11219bd0847aSShri Abhyankar v += nz; vi += nz; 11229bd0847aSShri Abhyankar } 11239bd0847aSShri Abhyankar 11249bd0847aSShri Abhyankar /* backward solve the upper triangular */ 11259bd0847aSShri Abhyankar for (i=n-1; i>=0; i--){ 11269bd0847aSShri Abhyankar v = aa + adiag[i+1]+1; 11279bd0847aSShri Abhyankar vi = aj + adiag[i+1]+1; 11289bd0847aSShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 11299bd0847aSShri Abhyankar sum = tmp[i]; 11309bd0847aSShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 11319bd0847aSShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 11329bd0847aSShri Abhyankar } 11339bd0847aSShri Abhyankar 11349bd0847aSShri Abhyankar b += n; 11359bd0847aSShri Abhyankar x += n; 11369bd0847aSShri Abhyankar } 11379bd0847aSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 11389bd0847aSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 11399bd0847aSShri Abhyankar ierr = MatRestoreArray(B,&b);CHKERRQ(ierr); 11409bd0847aSShri Abhyankar ierr = MatRestoreArray(X,&x);CHKERRQ(ierr); 11419bd0847aSShri Abhyankar ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 11429bd0847aSShri Abhyankar PetscFunctionReturn(0); 11439bd0847aSShri Abhyankar } 11449bd0847aSShri Abhyankar 11459bd0847aSShri Abhyankar #undef __FUNCT__ 1146137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm" 1147137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx) 1148137fb511SHong Zhang { 1149137fb511SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1150137fb511SHong Zhang IS iscol = a->col,isrow = a->row; 1151137fb511SHong Zhang PetscErrorCode ierr; 11525d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 11535d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 11545d0c19d7SBarry Smith PetscInt nz,row; 1155dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1156dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1157137fb511SHong Zhang 1158137fb511SHong Zhang PetscFunctionBegin; 1159137fb511SHong Zhang if (!n) PetscFunctionReturn(0); 1160137fb511SHong Zhang 1161137fb511SHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 1162137fb511SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1163137fb511SHong Zhang tmp = a->solve_work; 1164137fb511SHong Zhang 1165137fb511SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1166137fb511SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1167137fb511SHong Zhang 1168137fb511SHong Zhang /* forward solve the lower triangular */ 1169137fb511SHong Zhang tmp[0] = b[*r++]; 1170137fb511SHong Zhang tmps = tmp; 1171137fb511SHong Zhang for (row=1; row<n; row++) { 1172137fb511SHong Zhang i = rout[row]; /* permuted row */ 1173137fb511SHong Zhang v = aa + ai[i] ; 1174137fb511SHong Zhang vi = aj + ai[i] ; 1175137fb511SHong Zhang nz = a->diag[i] - ai[i]; 1176137fb511SHong Zhang sum = b[*r++]; 1177003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1178137fb511SHong Zhang tmp[row] = sum; 1179137fb511SHong Zhang } 1180137fb511SHong Zhang 1181137fb511SHong Zhang /* backward solve the upper triangular */ 1182137fb511SHong Zhang for (row=n-1; row>=0; row--){ 1183137fb511SHong Zhang i = rout[row]; /* permuted row */ 1184137fb511SHong Zhang v = aa + a->diag[i] + 1; 1185137fb511SHong Zhang vi = aj + a->diag[i] + 1; 1186137fb511SHong Zhang nz = ai[i+1] - a->diag[i] - 1; 1187137fb511SHong Zhang sum = tmp[row]; 1188003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1189137fb511SHong Zhang x[*c--] = tmp[row] = sum*aa[a->diag[i]]; 1190137fb511SHong Zhang } 1191137fb511SHong Zhang 1192137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1193137fb511SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1194137fb511SHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 1195137fb511SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1196dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1197137fb511SHong Zhang PetscFunctionReturn(0); 1198137fb511SHong Zhang } 1199137fb511SHong Zhang 1200930ae53cSBarry Smith /* ----------------------------------------------------------- */ 1201c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/fsolve.h> 12024a2ae208SSatish Balay #undef __FUNCT__ 1203ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_inplace" 1204ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx) 1205930ae53cSBarry Smith { 1206930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 12076849ba73SBarry Smith PetscErrorCode ierr; 1208d0f46423SBarry Smith PetscInt n = A->rmap->n; 1209b377110cSBarry Smith const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag; 1210356650c2SBarry Smith PetscScalar *x; 1211356650c2SBarry Smith const PetscScalar *b; 1212dd6ea824SBarry Smith const MatScalar *aa = a->a; 1213aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 1214356650c2SBarry Smith PetscInt adiag_i,i,nz,ai_i; 1215b377110cSBarry Smith const PetscInt *vi; 1216dd6ea824SBarry Smith const MatScalar *v; 1217dd6ea824SBarry Smith PetscScalar sum; 1218d85afc42SSatish Balay #endif 1219930ae53cSBarry Smith 12203a40ed3dSBarry Smith PetscFunctionBegin; 12213a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 1222930ae53cSBarry Smith 12233649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 12241ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1225930ae53cSBarry Smith 1226aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 12271c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 12281c4feecaSSatish Balay #else 1229930ae53cSBarry Smith /* forward solve the lower triangular */ 1230930ae53cSBarry Smith x[0] = b[0]; 1231930ae53cSBarry Smith for (i=1; i<n; i++) { 1232930ae53cSBarry Smith ai_i = ai[i]; 1233930ae53cSBarry Smith v = aa + ai_i; 1234930ae53cSBarry Smith vi = aj + ai_i; 1235930ae53cSBarry Smith nz = adiag[i] - ai_i; 1236930ae53cSBarry Smith sum = b[i]; 1237003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1238930ae53cSBarry Smith x[i] = sum; 1239930ae53cSBarry Smith } 1240930ae53cSBarry Smith 1241930ae53cSBarry Smith /* backward solve the upper triangular */ 1242930ae53cSBarry Smith for (i=n-1; i>=0; i--){ 1243930ae53cSBarry Smith adiag_i = adiag[i]; 1244d708749eSBarry Smith v = aa + adiag_i + 1; 1245d708749eSBarry Smith vi = aj + adiag_i + 1; 1246930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 1247930ae53cSBarry Smith sum = x[i]; 1248003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1249930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 1250930ae53cSBarry Smith } 12511c4feecaSSatish Balay #endif 1252dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 12533649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 12541ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 12553a40ed3dSBarry Smith PetscFunctionReturn(0); 1256930ae53cSBarry Smith } 1257930ae53cSBarry Smith 12584a2ae208SSatish Balay #undef __FUNCT__ 1259ad04f41aSHong Zhang #define __FUNCT__ "MatSolveAdd_SeqAIJ_inplace" 1260ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx) 1261da3a660dSBarry Smith { 1262416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1263416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 12646849ba73SBarry Smith PetscErrorCode ierr; 126504fbf559SBarry Smith PetscInt i, n = A->rmap->n,j; 12665d0c19d7SBarry Smith PetscInt nz; 126704fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j; 126804fbf559SBarry Smith PetscScalar *x,*tmp,sum; 126904fbf559SBarry Smith const PetscScalar *b; 1270dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1271da3a660dSBarry Smith 12723a40ed3dSBarry Smith PetscFunctionBegin; 127378b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 1274da3a660dSBarry Smith 12753649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 12761ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1277416022c9SBarry Smith tmp = a->solve_work; 1278da3a660dSBarry Smith 12793b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12803b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1281da3a660dSBarry Smith 1282da3a660dSBarry Smith /* forward solve the lower triangular */ 1283da3a660dSBarry Smith tmp[0] = b[*r++]; 1284da3a660dSBarry Smith for (i=1; i<n; i++) { 1285010a20caSHong Zhang v = aa + ai[i] ; 1286010a20caSHong Zhang vi = aj + ai[i] ; 1287416022c9SBarry Smith nz = a->diag[i] - ai[i]; 1288da3a660dSBarry Smith sum = b[*r++]; 128904fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1290da3a660dSBarry Smith tmp[i] = sum; 1291da3a660dSBarry Smith } 1292da3a660dSBarry Smith 1293da3a660dSBarry Smith /* backward solve the upper triangular */ 1294da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1295010a20caSHong Zhang v = aa + a->diag[i] + 1; 1296010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1297416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 1298da3a660dSBarry Smith sum = tmp[i]; 129904fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1300010a20caSHong Zhang tmp[i] = sum*aa[a->diag[i]]; 1301da3a660dSBarry Smith x[*c--] += tmp[i]; 1302da3a660dSBarry Smith } 1303da3a660dSBarry Smith 13043b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13053b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 13063649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 13071ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1308dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 1309898183ecSLois Curfman McInnes 13103a40ed3dSBarry Smith PetscFunctionReturn(0); 1311da3a660dSBarry Smith } 131204fbf559SBarry Smith 13134a2ae208SSatish Balay #undef __FUNCT__ 131435233d99SShri Abhyankar #define __FUNCT__ "MatSolveAdd_SeqAIJ" 131535233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 13163c0119dfSShri Abhyankar { 13173c0119dfSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13183c0119dfSShri Abhyankar IS iscol = a->col,isrow = a->row; 13193c0119dfSShri Abhyankar PetscErrorCode ierr; 13203c0119dfSShri Abhyankar PetscInt i, n = A->rmap->n,j; 13213c0119dfSShri Abhyankar PetscInt nz; 13223c0119dfSShri Abhyankar const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 13233c0119dfSShri Abhyankar PetscScalar *x,*tmp,sum; 13243c0119dfSShri Abhyankar const PetscScalar *b; 13253c0119dfSShri Abhyankar const MatScalar *aa = a->a,*v; 13263c0119dfSShri Abhyankar 13273c0119dfSShri Abhyankar PetscFunctionBegin; 13283c0119dfSShri Abhyankar if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 13293c0119dfSShri Abhyankar 13303649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 13313c0119dfSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 13323c0119dfSShri Abhyankar tmp = a->solve_work; 13333c0119dfSShri Abhyankar 13343c0119dfSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13353c0119dfSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 13363c0119dfSShri Abhyankar 13373c0119dfSShri Abhyankar /* forward solve the lower triangular */ 13383c0119dfSShri Abhyankar tmp[0] = b[r[0]]; 13393c0119dfSShri Abhyankar v = aa; 13403c0119dfSShri Abhyankar vi = aj; 13413c0119dfSShri Abhyankar for (i=1; i<n; i++) { 13423c0119dfSShri Abhyankar nz = ai[i+1] - ai[i]; 13433c0119dfSShri Abhyankar sum = b[r[i]]; 13443c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 13453c0119dfSShri Abhyankar tmp[i] = sum; 13463c0119dfSShri Abhyankar v += nz; vi += nz; 13473c0119dfSShri Abhyankar } 13483c0119dfSShri Abhyankar 13493c0119dfSShri Abhyankar /* backward solve the upper triangular */ 13503c0119dfSShri Abhyankar v = aa + adiag[n-1]; 13513c0119dfSShri Abhyankar vi = aj + adiag[n-1]; 13523c0119dfSShri Abhyankar for (i=n-1; i>=0; i--){ 13533c0119dfSShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 13543c0119dfSShri Abhyankar sum = tmp[i]; 13553c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 13563c0119dfSShri Abhyankar tmp[i] = sum*v[nz]; 13573c0119dfSShri Abhyankar x[c[i]] += tmp[i]; 13583c0119dfSShri Abhyankar v += nz+1; vi += nz+1; 13593c0119dfSShri Abhyankar } 13603c0119dfSShri Abhyankar 13613c0119dfSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13623c0119dfSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 13633649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 13643c0119dfSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 13653c0119dfSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 13663c0119dfSShri Abhyankar 13673c0119dfSShri Abhyankar PetscFunctionReturn(0); 13683c0119dfSShri Abhyankar } 13693c0119dfSShri Abhyankar 13703c0119dfSShri Abhyankar #undef __FUNCT__ 1371ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTranspose_SeqAIJ_inplace" 1372ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx) 1373da3a660dSBarry Smith { 1374416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13752235e667SBarry Smith IS iscol = a->col,isrow = a->row; 13766849ba73SBarry Smith PetscErrorCode ierr; 137704fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 137804fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 137904fbf559SBarry Smith PetscInt nz; 138004fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1381dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 138204fbf559SBarry Smith const PetscScalar *b; 1383da3a660dSBarry Smith 13843a40ed3dSBarry Smith PetscFunctionBegin; 13853649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 13861ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1387416022c9SBarry Smith tmp = a->solve_work; 1388da3a660dSBarry Smith 13892235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13902235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1391da3a660dSBarry Smith 1392da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 13932235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1394da3a660dSBarry Smith 1395da3a660dSBarry Smith /* forward solve the U^T */ 1396da3a660dSBarry Smith for (i=0; i<n; i++) { 1397010a20caSHong Zhang v = aa + diag[i] ; 1398010a20caSHong Zhang vi = aj + diag[i] + 1; 1399f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 1400c38d4ed2SBarry Smith s1 = tmp[i]; 1401ef66eb69SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 140204fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1403c38d4ed2SBarry Smith tmp[i] = s1; 1404da3a660dSBarry Smith } 1405da3a660dSBarry Smith 1406da3a660dSBarry Smith /* backward solve the L^T */ 1407da3a660dSBarry Smith for (i=n-1; i>=0; i--){ 1408010a20caSHong Zhang v = aa + diag[i] - 1 ; 1409010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1410f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 1411f1af5d2fSBarry Smith s1 = tmp[i]; 141204fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 1413da3a660dSBarry Smith } 1414da3a660dSBarry Smith 1415da3a660dSBarry Smith /* copy tmp into x according to permutation */ 1416591294e4SBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1417da3a660dSBarry Smith 14182235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 14192235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 14203649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 14211ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1422da3a660dSBarry Smith 1423dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 14243a40ed3dSBarry Smith PetscFunctionReturn(0); 1425da3a660dSBarry Smith } 1426da3a660dSBarry Smith 1427d1fa9404SShri Abhyankar #undef __FUNCT__ 142835233d99SShri Abhyankar #define __FUNCT__ "MatSolveTranspose_SeqAIJ" 142935233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 1430d1fa9404SShri Abhyankar { 1431d1fa9404SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1432d1fa9404SShri Abhyankar IS iscol = a->col,isrow = a->row; 1433d1fa9404SShri Abhyankar PetscErrorCode ierr; 1434d1fa9404SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1435d1fa9404SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1436d1fa9404SShri Abhyankar PetscInt nz; 1437d1fa9404SShri Abhyankar PetscScalar *x,*tmp,s1; 1438d1fa9404SShri Abhyankar const MatScalar *aa = a->a,*v; 1439d1fa9404SShri Abhyankar const PetscScalar *b; 1440d1fa9404SShri Abhyankar 1441d1fa9404SShri Abhyankar PetscFunctionBegin; 14423649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1443d1fa9404SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1444d1fa9404SShri Abhyankar tmp = a->solve_work; 1445d1fa9404SShri Abhyankar 1446d1fa9404SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1447d1fa9404SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1448d1fa9404SShri Abhyankar 1449d1fa9404SShri Abhyankar /* copy the b into temp work space according to permutation */ 1450d1fa9404SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1451d1fa9404SShri Abhyankar 1452d1fa9404SShri Abhyankar /* forward solve the U^T */ 1453d1fa9404SShri Abhyankar for (i=0; i<n; i++) { 1454d1fa9404SShri Abhyankar v = aa + adiag[i+1] + 1; 1455d1fa9404SShri Abhyankar vi = aj + adiag[i+1] + 1; 1456d1fa9404SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1457d1fa9404SShri Abhyankar s1 = tmp[i]; 1458d1fa9404SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1459d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1460d1fa9404SShri Abhyankar tmp[i] = s1; 1461d1fa9404SShri Abhyankar } 1462d1fa9404SShri Abhyankar 1463d1fa9404SShri Abhyankar /* backward solve the L^T */ 1464d1fa9404SShri Abhyankar for (i=n-1; i>=0; i--){ 1465d1fa9404SShri Abhyankar v = aa + ai[i]; 1466d1fa9404SShri Abhyankar vi = aj + ai[i]; 1467d1fa9404SShri Abhyankar nz = ai[i+1] - ai[i]; 1468d1fa9404SShri Abhyankar s1 = tmp[i]; 1469d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1470d1fa9404SShri Abhyankar } 1471d1fa9404SShri Abhyankar 1472d1fa9404SShri Abhyankar /* copy tmp into x according to permutation */ 1473d1fa9404SShri Abhyankar for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1474d1fa9404SShri Abhyankar 1475d1fa9404SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1476d1fa9404SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 14773649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1478d1fa9404SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1479d1fa9404SShri Abhyankar 1480d1fa9404SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1481d1fa9404SShri Abhyankar PetscFunctionReturn(0); 1482d1fa9404SShri Abhyankar } 1483d1fa9404SShri Abhyankar 14844a2ae208SSatish Balay #undef __FUNCT__ 1485ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ_inplace" 1486ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx) 1487da3a660dSBarry Smith { 1488416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 14892235e667SBarry Smith IS iscol = a->col,isrow = a->row; 14906849ba73SBarry Smith PetscErrorCode ierr; 149104fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 149204fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 149304fbf559SBarry Smith PetscInt nz; 149404fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1495dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 149604fbf559SBarry Smith const PetscScalar *b; 14976abc6512SBarry Smith 14983a40ed3dSBarry Smith PetscFunctionBegin; 149923bc6035SMatthew Knepley if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 15003649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 15011ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1502416022c9SBarry Smith tmp = a->solve_work; 15036abc6512SBarry Smith 15042235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 15052235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 15066abc6512SBarry Smith 15076abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 15082235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 15096abc6512SBarry Smith 15106abc6512SBarry Smith /* forward solve the U^T */ 15116abc6512SBarry Smith for (i=0; i<n; i++) { 1512010a20caSHong Zhang v = aa + diag[i] ; 1513010a20caSHong Zhang vi = aj + diag[i] + 1; 1514f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 151504fbf559SBarry Smith s1 = tmp[i]; 151604fbf559SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 151704fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 151804fbf559SBarry Smith tmp[i] = s1; 15196abc6512SBarry Smith } 15206abc6512SBarry Smith 15216abc6512SBarry Smith /* backward solve the L^T */ 15226abc6512SBarry Smith for (i=n-1; i>=0; i--){ 1523010a20caSHong Zhang v = aa + diag[i] - 1 ; 1524010a20caSHong Zhang vi = aj + diag[i] - 1 ; 1525f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 152604fbf559SBarry Smith s1 = tmp[i]; 152704fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 15286abc6512SBarry Smith } 15296abc6512SBarry Smith 15306abc6512SBarry Smith /* copy tmp into x according to permutation */ 15316abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 15326abc6512SBarry Smith 15332235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 15342235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 15353649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 15361ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 15376abc6512SBarry Smith 153804fbf559SBarry Smith ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 15393a40ed3dSBarry Smith PetscFunctionReturn(0); 1540da3a660dSBarry Smith } 154104fbf559SBarry Smith 1542533e6011SShri Abhyankar #undef __FUNCT__ 154335233d99SShri Abhyankar #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ" 154435233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 1545533e6011SShri Abhyankar { 1546533e6011SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1547533e6011SShri Abhyankar IS iscol = a->col,isrow = a->row; 1548533e6011SShri Abhyankar PetscErrorCode ierr; 1549533e6011SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1550533e6011SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1551533e6011SShri Abhyankar PetscInt nz; 1552533e6011SShri Abhyankar PetscScalar *x,*tmp,s1; 1553533e6011SShri Abhyankar const MatScalar *aa = a->a,*v; 1554533e6011SShri Abhyankar const PetscScalar *b; 1555533e6011SShri Abhyankar 1556533e6011SShri Abhyankar PetscFunctionBegin; 1557533e6011SShri Abhyankar if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 15583649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1559533e6011SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1560533e6011SShri Abhyankar tmp = a->solve_work; 1561533e6011SShri Abhyankar 1562533e6011SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1563533e6011SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1564533e6011SShri Abhyankar 1565533e6011SShri Abhyankar /* copy the b into temp work space according to permutation */ 1566533e6011SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1567533e6011SShri Abhyankar 1568533e6011SShri Abhyankar /* forward solve the U^T */ 1569533e6011SShri Abhyankar for (i=0; i<n; i++) { 1570533e6011SShri Abhyankar v = aa + adiag[i+1] + 1; 1571533e6011SShri Abhyankar vi = aj + adiag[i+1] + 1; 1572533e6011SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1573533e6011SShri Abhyankar s1 = tmp[i]; 1574533e6011SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1575533e6011SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1576533e6011SShri Abhyankar tmp[i] = s1; 1577533e6011SShri Abhyankar } 1578533e6011SShri Abhyankar 1579533e6011SShri Abhyankar 1580533e6011SShri Abhyankar /* backward solve the L^T */ 1581533e6011SShri Abhyankar for (i=n-1; i>=0; i--){ 1582533e6011SShri Abhyankar v = aa + ai[i] ; 1583533e6011SShri Abhyankar vi = aj + ai[i]; 1584533e6011SShri Abhyankar nz = ai[i+1] - ai[i]; 1585533e6011SShri Abhyankar s1 = tmp[i]; 1586c5e3b2a3SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1587533e6011SShri Abhyankar } 1588533e6011SShri Abhyankar 1589533e6011SShri Abhyankar /* copy tmp into x according to permutation */ 1590533e6011SShri Abhyankar for (i=0; i<n; i++) x[r[i]] += tmp[i]; 1591533e6011SShri Abhyankar 1592533e6011SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1593533e6011SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 15943649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1595533e6011SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1596533e6011SShri Abhyankar 1597533e6011SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1598533e6011SShri Abhyankar PetscFunctionReturn(0); 1599533e6011SShri Abhyankar } 1600533e6011SShri Abhyankar 16019e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 1602b47f5a65SHong Zhang 160309573ac7SBarry Smith extern PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscBool ); 160408480c60SBarry Smith 16058fc3a347SHong Zhang /* 16068a76ed4fSHong Zhang ilu() under revised new data structure. 1607813a1f2bSShri Abhyankar Factored arrays bj and ba are stored as 1608813a1f2bSShri Abhyankar L(0,:), L(1,:), ...,L(n-1,:), U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:) 1609813a1f2bSShri Abhyankar 1610813a1f2bSShri Abhyankar bi=fact->i is an array of size n+1, in which 1611813a1f2bSShri Abhyankar bi+ 1612baabb860SHong Zhang bi[i]: points to 1st entry of L(i,:),i=0,...,n-1 1613baabb860SHong Zhang bi[n]: points to L(n-1,n-1)+1 1614baabb860SHong Zhang 1615813a1f2bSShri Abhyankar bdiag=fact->diag is an array of size n+1,in which 1616baabb860SHong Zhang bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1 1617a24f213cSHong Zhang bdiag[n]: points to entry of U(n-1,0)-1 1618813a1f2bSShri Abhyankar 1619813a1f2bSShri Abhyankar U(i,:) contains bdiag[i] as its last entry, i.e., 1620813a1f2bSShri Abhyankar U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 1621813a1f2bSShri Abhyankar */ 1622813a1f2bSShri Abhyankar #undef __FUNCT__ 162335233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0" 162435233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1625813a1f2bSShri Abhyankar { 1626813a1f2bSShri Abhyankar 1627813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1628813a1f2bSShri Abhyankar PetscErrorCode ierr; 1629bbd65245SShri Abhyankar const PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag; 1630bbd65245SShri Abhyankar PetscInt i,j,k=0,nz,*bi,*bj,*bdiag; 1631ace3abfcSBarry Smith PetscBool missing; 16321df811f5SHong Zhang IS isicol; 1633813a1f2bSShri Abhyankar 1634813a1f2bSShri Abhyankar PetscFunctionBegin; 1635e32f2f54SBarry 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); 16361df811f5SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 1637e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 16381df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 16391df811f5SHong Zhang 1640813a1f2bSShri Abhyankar ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr); 1641813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1642813a1f2bSShri Abhyankar 1643813a1f2bSShri Abhyankar /* allocate matrix arrays for new data structure */ 1644813a1f2bSShri Abhyankar ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,n+1,PetscInt,&b->i);CHKERRQ(ierr); 1645813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1646813a1f2bSShri Abhyankar b->singlemalloc = PETSC_TRUE; 1647813a1f2bSShri Abhyankar if (!b->diag){ 1648813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr); 1649813a1f2bSShri Abhyankar ierr = PetscLogObjectMemory(fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1650813a1f2bSShri Abhyankar } 1651813a1f2bSShri Abhyankar bdiag = b->diag; 1652813a1f2bSShri Abhyankar 1653813a1f2bSShri Abhyankar if (n > 0) { 1654813a1f2bSShri Abhyankar ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr); 1655813a1f2bSShri Abhyankar } 1656813a1f2bSShri Abhyankar 1657813a1f2bSShri Abhyankar /* set bi and bj with new data structure */ 1658813a1f2bSShri Abhyankar bi = b->i; 1659813a1f2bSShri Abhyankar bj = b->j; 1660813a1f2bSShri Abhyankar 1661813a1f2bSShri Abhyankar /* L part */ 1662e60cf9a0SBarry Smith bi[0] = 0; 1663813a1f2bSShri Abhyankar for (i=0; i<n; i++){ 1664813a1f2bSShri Abhyankar nz = adiag[i] - ai[i]; 1665813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nz; 1666813a1f2bSShri Abhyankar aj = a->j + ai[i]; 1667813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 1668bbd65245SShri Abhyankar /* *bj = aj[j]; bj++; */ 1669bbd65245SShri Abhyankar bj[k++] = aj[j]; 1670813a1f2bSShri Abhyankar } 1671813a1f2bSShri Abhyankar } 1672813a1f2bSShri Abhyankar 1673813a1f2bSShri Abhyankar /* U part */ 167462fbd6afSShri Abhyankar bdiag[n] = bi[n]-1; 1675813a1f2bSShri Abhyankar for (i=n-1; i>=0; i--){ 1676813a1f2bSShri Abhyankar nz = ai[i+1] - adiag[i] - 1; 1677813a1f2bSShri Abhyankar aj = a->j + adiag[i] + 1; 1678813a1f2bSShri Abhyankar for (j=0; j<nz; j++){ 1679bbd65245SShri Abhyankar /* *bj = aj[j]; bj++; */ 1680bbd65245SShri Abhyankar bj[k++] = aj[j]; 1681813a1f2bSShri Abhyankar } 1682813a1f2bSShri Abhyankar /* diag[i] */ 1683bbd65245SShri Abhyankar /* *bj = i; bj++; */ 1684bbd65245SShri Abhyankar bj[k++] = i; 1685a24f213cSHong Zhang bdiag[i] = bdiag[i+1] + nz + 1; 1686813a1f2bSShri Abhyankar } 16871df811f5SHong Zhang 1688d5f3da31SBarry Smith fact->factortype = MAT_FACTOR_ILU; 16891df811f5SHong Zhang fact->info.factor_mallocs = 0; 16901df811f5SHong Zhang fact->info.fill_ratio_given = info->fill; 16911df811f5SHong Zhang fact->info.fill_ratio_needed = 1.0; 169235233d99SShri Abhyankar fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 16931df811f5SHong Zhang 16941df811f5SHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 16951df811f5SHong Zhang b->row = isrow; 16961df811f5SHong Zhang b->col = iscol; 16971df811f5SHong Zhang b->icol = isicol; 16981df811f5SHong Zhang ierr = PetscMalloc((fact->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 16991df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 17001df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1701813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1702813a1f2bSShri Abhyankar } 1703813a1f2bSShri Abhyankar 1704813a1f2bSShri Abhyankar #undef __FUNCT__ 170535233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ" 170635233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1707813a1f2bSShri Abhyankar { 1708813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1709813a1f2bSShri Abhyankar IS isicol; 1710813a1f2bSShri Abhyankar PetscErrorCode ierr; 1711813a1f2bSShri Abhyankar const PetscInt *r,*ic; 17121df811f5SHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j; 1713813a1f2bSShri Abhyankar PetscInt *bi,*cols,nnz,*cols_lvl; 1714813a1f2bSShri Abhyankar PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 1715813a1f2bSShri Abhyankar PetscInt i,levels,diagonal_fill; 1716ace3abfcSBarry Smith PetscBool col_identity,row_identity; 1717813a1f2bSShri Abhyankar PetscReal f; 1718813a1f2bSShri Abhyankar PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 1719813a1f2bSShri Abhyankar PetscBT lnkbt; 1720813a1f2bSShri Abhyankar PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 1721813a1f2bSShri Abhyankar PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 1722813a1f2bSShri Abhyankar PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 1723813a1f2bSShri Abhyankar 1724813a1f2bSShri Abhyankar PetscFunctionBegin; 17253898ab1dSSatish Balay /* Uncomment the old data struct part only while testing new data structure for MatSolve() */ 1726124a3351SHong Zhang /* 1727ace3abfcSBarry Smith PetscBool olddatastruct=PETSC_FALSE; 1728acfcf0e5SJed Brown ierr = PetscOptionsGetBool(PETSC_NULL,"-ilu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr); 172935233d99SShri Abhyankar if(olddatastruct){ 1730ad04f41aSHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_inplace(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1731ad04f41aSHong Zhang PetscFunctionReturn(0); 1732ad04f41aSHong Zhang } 1733124a3351SHong Zhang */ 1734ad04f41aSHong Zhang 1735813a1f2bSShri Abhyankar levels = (PetscInt)info->levels; 1736813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 1737813a1f2bSShri Abhyankar ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 1738813a1f2bSShri Abhyankar 1739813a1f2bSShri Abhyankar if (!levels && row_identity && col_identity) { 1740813a1f2bSShri Abhyankar /* special case: ilu(0) with natural ordering */ 174135233d99SShri Abhyankar ierr = MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1742124a3351SHong Zhang if (a->inode.size) { 1743124a3351SHong Zhang fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 1744124a3351SHong Zhang } 1745813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1746813a1f2bSShri Abhyankar } 1747813a1f2bSShri Abhyankar 1748e32f2f54SBarry 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); 17491df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 1750813a1f2bSShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1751813a1f2bSShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 1752813a1f2bSShri Abhyankar 17531bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 17541bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 17551bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1756e60cf9a0SBarry Smith bi[0] = bdiag[0] = 0; 1757813a1f2bSShri Abhyankar 17580e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr); 1759813a1f2bSShri Abhyankar 1760813a1f2bSShri Abhyankar /* create a linked list for storing column indices of the active row */ 1761813a1f2bSShri Abhyankar nlnk = n + 1; 1762813a1f2bSShri Abhyankar ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1763813a1f2bSShri Abhyankar 1764813a1f2bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 17651df811f5SHong Zhang f = info->fill; 17661df811f5SHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 1767813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1768813a1f2bSShri Abhyankar current_space = free_space; 1769813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 1770813a1f2bSShri Abhyankar current_space_lvl = free_space_lvl; 1771813a1f2bSShri Abhyankar 1772813a1f2bSShri Abhyankar for (i=0; i<n; i++) { 1773813a1f2bSShri Abhyankar nzi = 0; 1774813a1f2bSShri Abhyankar /* copy current row into linked list */ 1775813a1f2bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 1776e32f2f54SBarry 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); 1777813a1f2bSShri Abhyankar cols = aj + ai[r[i]]; 1778813a1f2bSShri Abhyankar lnk[i] = -1; /* marker to indicate if diagonal exists */ 1779813a1f2bSShri Abhyankar ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1780813a1f2bSShri Abhyankar nzi += nlnk; 1781813a1f2bSShri Abhyankar 1782813a1f2bSShri Abhyankar /* make sure diagonal entry is included */ 1783813a1f2bSShri Abhyankar if (diagonal_fill && lnk[i] == -1) { 1784813a1f2bSShri Abhyankar fm = n; 1785813a1f2bSShri Abhyankar while (lnk[fm] < i) fm = lnk[fm]; 1786813a1f2bSShri Abhyankar lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 1787813a1f2bSShri Abhyankar lnk[fm] = i; 1788813a1f2bSShri Abhyankar lnk_lvl[i] = 0; 1789813a1f2bSShri Abhyankar nzi++; dcount++; 1790813a1f2bSShri Abhyankar } 1791813a1f2bSShri Abhyankar 1792813a1f2bSShri Abhyankar /* add pivot rows into the active row */ 1793813a1f2bSShri Abhyankar nzbd = 0; 1794813a1f2bSShri Abhyankar prow = lnk[n]; 1795813a1f2bSShri Abhyankar while (prow < i) { 1796813a1f2bSShri Abhyankar nnz = bdiag[prow]; 1797813a1f2bSShri Abhyankar cols = bj_ptr[prow] + nnz + 1; 1798813a1f2bSShri Abhyankar cols_lvl = bjlvl_ptr[prow] + nnz + 1; 1799813a1f2bSShri Abhyankar nnz = bi[prow+1] - bi[prow] - nnz - 1; 1800813a1f2bSShri Abhyankar ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 1801813a1f2bSShri Abhyankar nzi += nlnk; 1802813a1f2bSShri Abhyankar prow = lnk[prow]; 1803813a1f2bSShri Abhyankar nzbd++; 1804813a1f2bSShri Abhyankar } 1805813a1f2bSShri Abhyankar bdiag[i] = nzbd; 1806813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nzi; 1807813a1f2bSShri Abhyankar 1808813a1f2bSShri Abhyankar /* if free space is not available, make more free space */ 1809813a1f2bSShri Abhyankar if (current_space->local_remaining<nzi) { 1810813a1f2bSShri Abhyankar nnz = 2*nzi*(n - i); /* estimated and max additional space needed */ 1811813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1812813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 1813813a1f2bSShri Abhyankar reallocs++; 1814813a1f2bSShri Abhyankar } 1815813a1f2bSShri Abhyankar 1816813a1f2bSShri Abhyankar /* copy data into free_space and free_space_lvl, then initialize lnk */ 1817813a1f2bSShri Abhyankar ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 1818813a1f2bSShri Abhyankar bj_ptr[i] = current_space->array; 1819813a1f2bSShri Abhyankar bjlvl_ptr[i] = current_space_lvl->array; 1820813a1f2bSShri Abhyankar 1821813a1f2bSShri Abhyankar /* make sure the active row i has diagonal entry */ 182265e19b50SBarry 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); 1823813a1f2bSShri Abhyankar 1824813a1f2bSShri Abhyankar current_space->array += nzi; 1825813a1f2bSShri Abhyankar current_space->local_used += nzi; 1826813a1f2bSShri Abhyankar current_space->local_remaining -= nzi; 1827813a1f2bSShri Abhyankar current_space_lvl->array += nzi; 1828813a1f2bSShri Abhyankar current_space_lvl->local_used += nzi; 1829813a1f2bSShri Abhyankar current_space_lvl->local_remaining -= nzi; 1830813a1f2bSShri Abhyankar } 1831813a1f2bSShri Abhyankar 1832813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1833813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 1834813a1f2bSShri Abhyankar 1835813a1f2bSShri Abhyankar /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 18369263d837SHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 18372ce24eb6SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 1838813a1f2bSShri Abhyankar 1839813a1f2bSShri Abhyankar ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1840813a1f2bSShri Abhyankar ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 18410e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 1842813a1f2bSShri Abhyankar 1843813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO) 1844813a1f2bSShri Abhyankar { 1845aef85c9fSShri Abhyankar PetscReal af = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 1846813a1f2bSShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 1847813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1848813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 1849813a1f2bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1850813a1f2bSShri Abhyankar if (diagonal_fill) { 1851813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 1852813a1f2bSShri Abhyankar } 1853813a1f2bSShri Abhyankar } 1854813a1f2bSShri Abhyankar #endif 1855813a1f2bSShri Abhyankar 1856813a1f2bSShri Abhyankar /* put together the new matrix */ 1857813a1f2bSShri Abhyankar ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 1858813a1f2bSShri Abhyankar ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 1859813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1860813a1f2bSShri Abhyankar b->free_a = PETSC_TRUE; 1861813a1f2bSShri Abhyankar b->free_ij = PETSC_TRUE; 1862813a1f2bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 1863f268cba8SShri Abhyankar ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 1864813a1f2bSShri Abhyankar b->j = bj; 1865813a1f2bSShri Abhyankar b->i = bi; 1866813a1f2bSShri Abhyankar b->diag = bdiag; 1867813a1f2bSShri Abhyankar b->ilen = 0; 1868813a1f2bSShri Abhyankar b->imax = 0; 1869813a1f2bSShri Abhyankar b->row = isrow; 1870813a1f2bSShri Abhyankar b->col = iscol; 1871813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1872813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1873813a1f2bSShri Abhyankar b->icol = isicol; 1874813a1f2bSShri Abhyankar ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1875813a1f2bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. 1876813a1f2bSShri Abhyankar Allocate bdiag, solve_work, new a, new j */ 1877f268cba8SShri Abhyankar ierr = PetscLogObjectMemory(fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 1878f268cba8SShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 1879813a1f2bSShri Abhyankar (fact)->info.factor_mallocs = reallocs; 1880813a1f2bSShri Abhyankar (fact)->info.fill_ratio_given = f; 1881f268cba8SShri Abhyankar (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 188235233d99SShri Abhyankar (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1883f2344125SBarry Smith if (a->inode.size) { 1884f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 1885f2344125SBarry Smith } 1886813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1887813a1f2bSShri Abhyankar } 1888813a1f2bSShri Abhyankar 18896516239fSHong Zhang #undef __FUNCT__ 1890ad04f41aSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_inplace" 1891ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 18926516239fSHong Zhang { 18936516239fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 18946516239fSHong Zhang IS isicol; 18956516239fSHong Zhang PetscErrorCode ierr; 18966516239fSHong Zhang const PetscInt *r,*ic; 18976516239fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 18986516239fSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 18996516239fSHong Zhang PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 19006516239fSHong Zhang PetscInt i,levels,diagonal_fill; 1901ace3abfcSBarry Smith PetscBool col_identity,row_identity; 19026516239fSHong Zhang PetscReal f; 19036516239fSHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL; 19046516239fSHong Zhang PetscBT lnkbt; 19056516239fSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 19066516239fSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 19076516239fSHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 1908ace3abfcSBarry Smith PetscBool missing; 19096516239fSHong Zhang 19106516239fSHong Zhang PetscFunctionBegin; 1911e32f2f54SBarry 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); 19126516239fSHong Zhang f = info->fill; 19136516239fSHong Zhang levels = (PetscInt)info->levels; 19146516239fSHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 19156516239fSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 19166516239fSHong Zhang 19176516239fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 19186516239fSHong Zhang ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 19196516239fSHong Zhang if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */ 1920f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr); 1921ad04f41aSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 1922f2344125SBarry Smith if (a->inode.size) { 1923f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 1924f2344125SBarry Smith } 1925d5f3da31SBarry Smith fact->factortype = MAT_FACTOR_ILU; 1926719d5645SBarry Smith (fact)->info.factor_mallocs = 0; 1927719d5645SBarry Smith (fact)->info.fill_ratio_given = info->fill; 1928719d5645SBarry Smith (fact)->info.fill_ratio_needed = 1.0; 1929719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 19308ef3462fSBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 1931e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 193208480c60SBarry Smith b->row = isrow; 193308480c60SBarry Smith b->col = iscol; 193482bf6240SBarry Smith b->icol = isicol; 1935719d5645SBarry Smith ierr = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 1936c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1937c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 19383a40ed3dSBarry Smith PetscFunctionReturn(0); 193908480c60SBarry Smith } 194008480c60SBarry Smith 1941898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1942898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 19439e25ed09SBarry Smith 19441bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 19451bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 19461bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); 1947a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 19486cf997b0SBarry Smith 19490e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr); 19505a8e39fbSHong Zhang 19515a8e39fbSHong Zhang /* create a linked list for storing column indices of the active row */ 19525a8e39fbSHong Zhang nlnk = n + 1; 19535a8e39fbSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19545a8e39fbSHong Zhang 19555a8e39fbSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 1956a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 19575a8e39fbSHong Zhang current_space = free_space; 1958a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 19595a8e39fbSHong Zhang current_space_lvl = free_space_lvl; 19605a8e39fbSHong Zhang 19615a8e39fbSHong Zhang for (i=0; i<n; i++) { 19625a8e39fbSHong Zhang nzi = 0; 19636cf997b0SBarry Smith /* copy current row into linked list */ 19645a8e39fbSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 1965e32f2f54SBarry 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); 19665a8e39fbSHong Zhang cols = aj + ai[r[i]]; 19675a8e39fbSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 19685a8e39fbSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19695a8e39fbSHong Zhang nzi += nlnk; 19706cf997b0SBarry Smith 19716cf997b0SBarry Smith /* make sure diagonal entry is included */ 19725a8e39fbSHong Zhang if (diagonal_fill && lnk[i] == -1) { 19736cf997b0SBarry Smith fm = n; 19745a8e39fbSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 19755a8e39fbSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 19765a8e39fbSHong Zhang lnk[fm] = i; 19775a8e39fbSHong Zhang lnk_lvl[i] = 0; 19785a8e39fbSHong Zhang nzi++; dcount++; 19796cf997b0SBarry Smith } 19806cf997b0SBarry Smith 19815a8e39fbSHong Zhang /* add pivot rows into the active row */ 19825a8e39fbSHong Zhang nzbd = 0; 19835a8e39fbSHong Zhang prow = lnk[n]; 19845a8e39fbSHong Zhang while (prow < i) { 19855a8e39fbSHong Zhang nnz = bdiag[prow]; 19865a8e39fbSHong Zhang cols = bj_ptr[prow] + nnz + 1; 19875a8e39fbSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 19885a8e39fbSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 19895a8e39fbSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 19905a8e39fbSHong Zhang nzi += nlnk; 19915a8e39fbSHong Zhang prow = lnk[prow]; 19925a8e39fbSHong Zhang nzbd++; 199356beaf04SBarry Smith } 19945a8e39fbSHong Zhang bdiag[i] = nzbd; 19955a8e39fbSHong Zhang bi[i+1] = bi[i] + nzi; 1996ecf371e4SBarry Smith 19975a8e39fbSHong Zhang /* if free space is not available, make more free space */ 19985a8e39fbSHong Zhang if (current_space->local_remaining<nzi) { 19995a8e39fbSHong Zhang nnz = nzi*(n - i); /* estimated and max additional space needed */ 2000a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2001a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 20025a8e39fbSHong Zhang reallocs++; 20035a8e39fbSHong Zhang } 2004ecf371e4SBarry Smith 20055a8e39fbSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 20065a8e39fbSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 20075a8e39fbSHong Zhang bj_ptr[i] = current_space->array; 20085a8e39fbSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 20095a8e39fbSHong Zhang 20105a8e39fbSHong Zhang /* make sure the active row i has diagonal entry */ 201165e19b50SBarry 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); 20125a8e39fbSHong Zhang 20135a8e39fbSHong Zhang current_space->array += nzi; 20145a8e39fbSHong Zhang current_space->local_used += nzi; 20155a8e39fbSHong Zhang current_space->local_remaining -= nzi; 20165a8e39fbSHong Zhang current_space_lvl->array += nzi; 20175a8e39fbSHong Zhang current_space_lvl->local_used += nzi; 20185a8e39fbSHong Zhang current_space_lvl->local_remaining -= nzi; 20199e25ed09SBarry Smith } 20205a8e39fbSHong Zhang 2021898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 2022898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 20235a8e39fbSHong Zhang 20245a8e39fbSHong Zhang /* destroy list of free space and other temporary arrays */ 20255a8e39fbSHong Zhang ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 20266516239fSHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */ 20275a8e39fbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2028a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 20290e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 20309e25ed09SBarry Smith 20316cf91177SBarry Smith #if defined(PETSC_USE_INFO) 2032f64065bbSBarry Smith { 20335a8e39fbSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2034ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 2035ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2036ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 2037ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 2038335d9088SBarry Smith if (diagonal_fill) { 2039ae15b995SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 2040335d9088SBarry Smith } 2041f64065bbSBarry Smith } 204263ba0a88SBarry Smith #endif 2043bbb6d6a8SBarry Smith 20449e25ed09SBarry Smith /* put together the new matrix */ 204507dbf95fSBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 2046719d5645SBarry Smith ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr); 2047719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 2048e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2049e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 20507c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 20510e83c824SBarry Smith ierr = PetscMalloc(bi[n]*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); 20525a8e39fbSHong Zhang b->j = bj; 20535a8e39fbSHong Zhang b->i = bi; 20545a8e39fbSHong Zhang for (i=0; i<n; i++) bdiag[i] += bi[i]; 20555a8e39fbSHong Zhang b->diag = bdiag; 2056416022c9SBarry Smith b->ilen = 0; 2057416022c9SBarry Smith b->imax = 0; 2058416022c9SBarry Smith b->row = isrow; 2059416022c9SBarry Smith b->col = iscol; 2060c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2061c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 206282bf6240SBarry Smith b->icol = isicol; 206387828ca2SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2064416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 20655a8e39fbSHong Zhang Allocate bdiag, solve_work, new a, new j */ 2066719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 20675a8e39fbSHong Zhang b->maxnz = b->nz = bi[n] ; 2068719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2069719d5645SBarry Smith (fact)->info.fill_ratio_given = f; 2070719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2071ad04f41aSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 2072f2344125SBarry Smith if (a->inode.size) { 2073f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 2074d3ac4fa3SBarry Smith } 20753a40ed3dSBarry Smith PetscFunctionReturn(0); 20769e25ed09SBarry Smith } 2077d5d45c9bSBarry Smith 2078a6175056SHong Zhang #undef __FUNCT__ 207935233d99SShri Abhyankar #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ" 208035233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 20815f44c854SHong Zhang { 20825f44c854SHong Zhang Mat C = B; 20835f44c854SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 20845f44c854SHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 20855f44c854SHong Zhang IS ip=b->row,iip = b->icol; 20865f44c854SHong Zhang PetscErrorCode ierr; 20875f44c854SHong Zhang const PetscInt *rip,*riip; 20885f44c854SHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp; 20895f44c854SHong Zhang PetscInt *ai=a->i,*aj=a->j; 20905f44c854SHong Zhang PetscInt k,jmin,jmax,*c2r,*il,col,nexti,ili,nz; 20915f44c854SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 2092ace3abfcSBarry Smith PetscBool perm_identity; 2093d90ac83dSHong Zhang FactorShiftCtx sctx; 209498a93161SHong Zhang PetscReal rs; 209598a93161SHong Zhang MatScalar d,*v; 209698a93161SHong Zhang 20975f44c854SHong Zhang PetscFunctionBegin; 209898a93161SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 2099d90ac83dSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 210098a93161SHong Zhang 2101f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 210298a93161SHong Zhang sctx.shift_top = info->zeropivot; 210398a93161SHong Zhang for (i=0; i<mbs; i++) { 210498a93161SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 210598a93161SHong Zhang d = (aa)[a->diag[i]]; 210698a93161SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 210798a93161SHong Zhang v = aa+ai[i]; 210898a93161SHong Zhang nz = ai[i+1] - ai[i]; 210998a93161SHong Zhang for (j=0; j<nz; j++) 211098a93161SHong Zhang rs += PetscAbsScalar(v[j]); 211198a93161SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 211298a93161SHong Zhang } 211398a93161SHong Zhang sctx.shift_top *= 1.1; 211498a93161SHong Zhang sctx.nshift_max = 5; 211598a93161SHong Zhang sctx.shift_lo = 0.; 211698a93161SHong Zhang sctx.shift_hi = 1.; 211798a93161SHong Zhang } 21185f44c854SHong Zhang 21195f44c854SHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 21205f44c854SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 21215f44c854SHong Zhang 21225f44c854SHong Zhang /* allocate working arrays 21235f44c854SHong Zhang c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col 21245f44c854SHong 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 21255f44c854SHong Zhang */ 21260e83c824SBarry Smith ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&c2r);CHKERRQ(ierr); 21275f44c854SHong Zhang 212898a93161SHong Zhang do { 212907b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 213098a93161SHong Zhang 2131c5546cabSHong Zhang for (i=0; i<mbs; i++) c2r[i] = mbs; 21325f44c854SHong Zhang il[0] = 0; 21335f44c854SHong Zhang 21345f44c854SHong Zhang for (k = 0; k<mbs; k++){ 21355f44c854SHong Zhang /* zero rtmp */ 21365f44c854SHong Zhang nz = bi[k+1] - bi[k]; 21375f44c854SHong Zhang bjtmp = bj + bi[k]; 21385f44c854SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 21395f44c854SHong Zhang 21405f44c854SHong Zhang /* load in initial unfactored row */ 21415f44c854SHong Zhang bval = ba + bi[k]; 21425f44c854SHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 21435f44c854SHong Zhang for (j = jmin; j < jmax; j++){ 21445f44c854SHong Zhang col = riip[aj[j]]; 21455f44c854SHong Zhang if (col >= k){ /* only take upper triangular entry */ 21465f44c854SHong Zhang rtmp[col] = aa[j]; 21475f44c854SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 21485f44c854SHong Zhang } 21495f44c854SHong Zhang } 215098a93161SHong Zhang /* shift the diagonal of the matrix: ZeropivotApply() */ 215198a93161SHong Zhang rtmp[k] += sctx.shift_amount; /* shift the diagonal of the matrix */ 21525f44c854SHong Zhang 21535f44c854SHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 21545f44c854SHong Zhang dk = rtmp[k]; 21555f44c854SHong Zhang i = c2r[k]; /* first row to be added to k_th row */ 21565f44c854SHong Zhang 21575f44c854SHong Zhang while (i < k){ 21585f44c854SHong Zhang nexti = c2r[i]; /* next row to be added to k_th row */ 21595f44c854SHong Zhang 21605f44c854SHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 21615f44c854SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 21625f44c854SHong Zhang uikdi = - ba[ili]*ba[bdiag[i]]; /* diagonal(k) */ 21635f44c854SHong Zhang dk += uikdi*ba[ili]; /* update diag[k] */ 21645f44c854SHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 21655f44c854SHong Zhang 21665f44c854SHong Zhang /* add multiple of row i to k-th row */ 21675f44c854SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 21685f44c854SHong Zhang if (jmin < jmax){ 21695f44c854SHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 21705f44c854SHong Zhang /* update il and c2r for row i */ 21715f44c854SHong Zhang il[i] = jmin; 21725f44c854SHong Zhang j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i; 21735f44c854SHong Zhang } 21745f44c854SHong Zhang i = nexti; 21755f44c854SHong Zhang } 21765f44c854SHong Zhang 21775f44c854SHong Zhang /* copy data into U(k,:) */ 217898a93161SHong Zhang rs = 0.0; 21795f44c854SHong Zhang jmin = bi[k]; jmax = bi[k+1]-1; 21805f44c854SHong Zhang if (jmin < jmax) { 21815f44c854SHong Zhang for (j=jmin; j<jmax; j++){ 218298a93161SHong Zhang col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]); 21835f44c854SHong Zhang } 21845f44c854SHong Zhang /* add the k-th row into il and c2r */ 21855f44c854SHong Zhang il[k] = jmin; 21865f44c854SHong Zhang i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k; 21875f44c854SHong Zhang } 218898a93161SHong Zhang 218998a93161SHong Zhang /* MatPivotCheck() */ 219098a93161SHong Zhang sctx.rs = rs; 219198a93161SHong Zhang sctx.pv = dk; 2192d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 219307b50cabSHong Zhang if(sctx.newshift) break; 219498a93161SHong Zhang dk = sctx.pv; 219598a93161SHong Zhang 219698a93161SHong Zhang ba[bdiag[k]] = 1.0/dk; /* U(k,k) */ 219798a93161SHong Zhang } 219807b50cabSHong Zhang } while (sctx.newshift); 21995f44c854SHong Zhang 22000e83c824SBarry Smith ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr); 22015f44c854SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 22025f44c854SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 22035f44c854SHong Zhang 22045f44c854SHong Zhang ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 22055f44c854SHong Zhang if (perm_identity){ 220635233d99SShri Abhyankar B->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 220735233d99SShri Abhyankar B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 22082169352eSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering; 22092169352eSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering; 22105f44c854SHong Zhang } else { 221135233d99SShri Abhyankar B->ops->solve = MatSolve_SeqSBAIJ_1; 221235233d99SShri Abhyankar B->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 221335233d99SShri Abhyankar B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1; 221435233d99SShri Abhyankar B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1; 22155f44c854SHong Zhang } 22165f44c854SHong Zhang 22175f44c854SHong Zhang C->assembled = PETSC_TRUE; 22185f44c854SHong Zhang C->preallocated = PETSC_TRUE; 22195f44c854SHong Zhang ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 222098a93161SHong Zhang 222198a93161SHong Zhang /* MatPivotView() */ 222298a93161SHong Zhang if (sctx.nshift){ 2223f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 222498a93161SHong 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); 2225f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 222698a93161SHong Zhang ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2227f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){ 2228b2a402fcSHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr); 222998a93161SHong Zhang } 223098a93161SHong Zhang } 22315f44c854SHong Zhang PetscFunctionReturn(0); 22325f44c854SHong Zhang } 22335f44c854SHong Zhang 22345f44c854SHong Zhang #undef __FUNCT__ 22350a3c351aSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_inplace" 22360a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info) 2237a6175056SHong Zhang { 2238719d5645SBarry Smith Mat C = B; 22390968510aSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 22402ed133dbSHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 22419bfd6278SHong Zhang IS ip=b->row,iip = b->icol; 22422ed133dbSHong Zhang PetscErrorCode ierr; 22435d0c19d7SBarry Smith const PetscInt *rip,*riip; 2244c5546cabSHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp; 22452ed133dbSHong Zhang PetscInt *ai=a->i,*aj=a->j; 22462ed133dbSHong Zhang PetscInt k,jmin,jmax,*jl,*il,col,nexti,ili,nz; 2247622e2028SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 2248ace3abfcSBarry Smith PetscBool perm_identity; 22490e95ead3SHong Zhang FactorShiftCtx sctx; 22500e95ead3SHong Zhang PetscReal rs; 22510e95ead3SHong Zhang MatScalar d,*v; 2252d5d45c9bSBarry Smith 2253a6175056SHong Zhang PetscFunctionBegin; 22540e95ead3SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 22550e95ead3SHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 22560e95ead3SHong Zhang 22570e95ead3SHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 22580e95ead3SHong Zhang sctx.shift_top = info->zeropivot; 22590e95ead3SHong Zhang for (i=0; i<mbs; i++) { 22600e95ead3SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 22610e95ead3SHong Zhang d = (aa)[a->diag[i]]; 22620e95ead3SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 22630e95ead3SHong Zhang v = aa+ai[i]; 22640e95ead3SHong Zhang nz = ai[i+1] - ai[i]; 22650e95ead3SHong Zhang for (j=0; j<nz; j++) 22660e95ead3SHong Zhang rs += PetscAbsScalar(v[j]); 22670e95ead3SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 22680e95ead3SHong Zhang } 22690e95ead3SHong Zhang sctx.shift_top *= 1.1; 22700e95ead3SHong Zhang sctx.nshift_max = 5; 22710e95ead3SHong Zhang sctx.shift_lo = 0.; 22720e95ead3SHong Zhang sctx.shift_hi = 1.; 22730e95ead3SHong Zhang } 2274ee45ca4aSHong Zhang 22752ed133dbSHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 22769bfd6278SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 22772ed133dbSHong Zhang 22782ed133dbSHong Zhang /* initialization */ 22790e83c824SBarry Smith ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&jl);CHKERRQ(ierr); 22800e95ead3SHong Zhang 22812ed133dbSHong Zhang do { 228207b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 22830e95ead3SHong Zhang 2284c5546cabSHong Zhang for (i=0; i<mbs; i++) jl[i] = mbs; 2285c5546cabSHong Zhang il[0] = 0; 22862ed133dbSHong Zhang 22872ed133dbSHong Zhang for (k = 0; k<mbs; k++){ 2288c5546cabSHong Zhang /* zero rtmp */ 2289c5546cabSHong Zhang nz = bi[k+1] - bi[k]; 2290c5546cabSHong Zhang bjtmp = bj + bi[k]; 2291c5546cabSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 2292c5546cabSHong Zhang 2293c05c3958SHong Zhang bval = ba + bi[k]; 22942ed133dbSHong Zhang /* initialize k-th row by the perm[k]-th row of A */ 22952ed133dbSHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 22962ed133dbSHong Zhang for (j = jmin; j < jmax; j++){ 22979bfd6278SHong Zhang col = riip[aj[j]]; 22982ed133dbSHong Zhang if (col >= k){ /* only take upper triangular entry */ 22992ed133dbSHong Zhang rtmp[col] = aa[j]; 2300c05c3958SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 23012ed133dbSHong Zhang } 23022ed133dbSHong Zhang } 230339e3d630SHong Zhang /* shift the diagonal of the matrix */ 2304540703acSHong Zhang if (sctx.nshift) rtmp[k] += sctx.shift_amount; 23052ed133dbSHong Zhang 23062ed133dbSHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 23072ed133dbSHong Zhang dk = rtmp[k]; 23082ed133dbSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 23092ed133dbSHong Zhang 23102ed133dbSHong Zhang while (i < k){ 23112ed133dbSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 23122ed133dbSHong Zhang 23132ed133dbSHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 23142ed133dbSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 23152ed133dbSHong Zhang uikdi = - ba[ili]*ba[bi[i]]; /* diagonal(k) */ 23162ed133dbSHong Zhang dk += uikdi*ba[ili]; 23172ed133dbSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 23182ed133dbSHong Zhang 23192ed133dbSHong Zhang /* add multiple of row i to k-th row */ 23202ed133dbSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 23212ed133dbSHong Zhang if (jmin < jmax){ 23222ed133dbSHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 23232ed133dbSHong Zhang /* update il and jl for row i */ 23242ed133dbSHong Zhang il[i] = jmin; 23252ed133dbSHong Zhang j = bj[jmin]; jl[i] = jl[j]; jl[j] = i; 23262ed133dbSHong Zhang } 23272ed133dbSHong Zhang i = nexti; 23282ed133dbSHong Zhang } 23292ed133dbSHong Zhang 23300697b6caSHong Zhang /* shift the diagonals when zero pivot is detected */ 23310697b6caSHong Zhang /* compute rs=sum of abs(off-diagonal) */ 23320697b6caSHong Zhang rs = 0.0; 23333c9e8598SHong Zhang jmin = bi[k]+1; 23343c9e8598SHong Zhang nz = bi[k+1] - jmin; 23353c9e8598SHong Zhang bcol = bj + jmin; 233604fbf559SBarry Smith for (j=0; j<nz; j++) { 233704fbf559SBarry Smith rs += PetscAbsScalar(rtmp[bcol[j]]); 23383c9e8598SHong Zhang } 2339540703acSHong Zhang 2340540703acSHong Zhang sctx.rs = rs; 2341540703acSHong Zhang sctx.pv = dk; 2342d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,k);CHKERRQ(ierr); 234307b50cabSHong Zhang if (sctx.newshift) break; 23440e95ead3SHong Zhang dk = sctx.pv; 23453c9e8598SHong Zhang 23463c9e8598SHong Zhang /* copy data into U(k,:) */ 234739e3d630SHong Zhang ba[bi[k]] = 1.0/dk; /* U(k,k) */ 234839e3d630SHong Zhang jmin = bi[k]+1; jmax = bi[k+1]; 234939e3d630SHong Zhang if (jmin < jmax) { 235039e3d630SHong Zhang for (j=jmin; j<jmax; j++){ 2351c6ae9bfcSHong Zhang col = bj[j]; ba[j] = rtmp[col]; 23523c9e8598SHong Zhang } 235339e3d630SHong Zhang /* add the k-th row into il and jl */ 23543c9e8598SHong Zhang il[k] = jmin; 23553c9e8598SHong Zhang i = bj[jmin]; jl[k] = jl[i]; jl[i] = k; 23563c9e8598SHong Zhang } 23573c9e8598SHong Zhang } 235807b50cabSHong Zhang } while (sctx.newshift); 23590e95ead3SHong Zhang 23600e83c824SBarry Smith ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr); 236139e3d630SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 23629bfd6278SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 2363db4efbfdSBarry Smith 2364db4efbfdSBarry Smith ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 2365db4efbfdSBarry Smith if (perm_identity){ 23660a3c351aSHong Zhang B->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23670a3c351aSHong Zhang B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23680a3c351aSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23690a3c351aSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 2370db4efbfdSBarry Smith } else { 23710a3c351aSHong Zhang B->ops->solve = MatSolve_SeqSBAIJ_1_inplace; 23720a3c351aSHong Zhang B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_inplace; 23730a3c351aSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_inplace; 23740a3c351aSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_inplace; 2375db4efbfdSBarry Smith } 2376db4efbfdSBarry Smith 23773c9e8598SHong Zhang C->assembled = PETSC_TRUE; 23783c9e8598SHong Zhang C->preallocated = PETSC_TRUE; 2379d0f46423SBarry Smith ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 2380540703acSHong Zhang if (sctx.nshift){ 2381f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 23821e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2383f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 23841e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 23850697b6caSHong Zhang } 23863c9e8598SHong Zhang } 23873c9e8598SHong Zhang PetscFunctionReturn(0); 23883c9e8598SHong Zhang } 2389a6175056SHong Zhang 23908a76ed4fSHong Zhang /* 23918a76ed4fSHong Zhang icc() under revised new data structure. 23928a76ed4fSHong Zhang Factored arrays bj and ba are stored as 23938a76ed4fSHong Zhang U(0,:),...,U(i,:),U(n-1,:) 23948a76ed4fSHong Zhang 23958a76ed4fSHong Zhang ui=fact->i is an array of size n+1, in which 23968a76ed4fSHong Zhang ui+ 23978a76ed4fSHong Zhang ui[i]: points to 1st entry of U(i,:),i=0,...,n-1 23988a76ed4fSHong Zhang ui[n]: points to U(n-1,n-1)+1 23998a76ed4fSHong Zhang 24008a76ed4fSHong Zhang udiag=fact->diag is an array of size n,in which 24018a76ed4fSHong Zhang udiag[i]: points to diagonal of U(i,:), i=0,...,n-1 24028a76ed4fSHong Zhang 24038a76ed4fSHong Zhang U(i,:) contains udiag[i] as its last entry, i.e., 24048a76ed4fSHong Zhang U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 24058a76ed4fSHong Zhang */ 24068a76ed4fSHong Zhang 2407a6175056SHong Zhang #undef __FUNCT__ 240835233d99SShri Abhyankar #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ" 240935233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2410a6175056SHong Zhang { 24110968510aSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2412ed59401aSHong Zhang Mat_SeqSBAIJ *b; 2413dfbe8321SBarry Smith PetscErrorCode ierr; 2414ace3abfcSBarry Smith PetscBool perm_identity,missing; 24155f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 24165d0c19d7SBarry Smith const PetscInt *rip,*riip; 2417ed59401aSHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 241858ebbce7SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 24195a8e39fbSHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 2420ed59401aSHong Zhang PetscReal fill=info->fill,levels=info->levels; 2421a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 2422a1a86e44SBarry Smith PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 24237a48dd6fSHong Zhang PetscBT lnkbt; 2424b635d36bSHong Zhang IS iperm; 2425a6175056SHong Zhang 2426a6175056SHong Zhang PetscFunctionBegin; 2427e32f2f54SBarry 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); 242858ebbce7SBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 2429e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 2430653a6975SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 2431b635d36bSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 24327a48dd6fSHong Zhang 243339e3d630SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 24345f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 243539e3d630SHong Zhang ui[0] = 0; 243639e3d630SHong Zhang 24378a76ed4fSHong Zhang /* ICC(0) without matrix ordering: simply rearrange column indices */ 2438622e2028SHong Zhang if (!levels && perm_identity) { 24395f44c854SHong Zhang for (i=0; i<am; i++) { 2440c5546cabSHong Zhang ncols = ai[i+1] - a->diag[i]; 2441c5546cabSHong Zhang ui[i+1] = ui[i] + ncols; 2442c5546cabSHong Zhang udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */ 2443dc1e2a3fSHong Zhang } 2444dc1e2a3fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2445dc1e2a3fSHong Zhang cols = uj; 2446dc1e2a3fSHong Zhang for (i=0; i<am; i++) { 24475f44c854SHong Zhang aj = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */ 2448dc1e2a3fSHong Zhang ncols = ai[i+1] - a->diag[i] -1; 24495f44c854SHong Zhang for (j=0; j<ncols; j++) *cols++ = aj[j]; 24505f44c854SHong Zhang *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */ 24515f44c854SHong Zhang } 24525f44c854SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 24535f44c854SHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 24545f44c854SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 24555f44c854SHong Zhang 24565f44c854SHong Zhang /* initialization */ 24575f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 24585f44c854SHong Zhang 24595f44c854SHong Zhang /* jl: linked list for storing indices of the pivot rows 24605f44c854SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 24610e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr); 24625f44c854SHong Zhang for (i=0; i<am; i++){ 24635f44c854SHong Zhang jl[i] = am; il[i] = 0; 24645f44c854SHong Zhang } 24655f44c854SHong Zhang 24665f44c854SHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 24675f44c854SHong Zhang nlnk = am + 1; 24685f44c854SHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 24695f44c854SHong Zhang 247095b5ac22SHong Zhang /* initial FreeSpace size is fill*(ai[am]+am)/2 */ 247195b5ac22SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space);CHKERRQ(ierr); 24725f44c854SHong Zhang current_space = free_space; 247395b5ac22SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space_lvl);CHKERRQ(ierr); 24745f44c854SHong Zhang current_space_lvl = free_space_lvl; 24755f44c854SHong Zhang 24765f44c854SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 24775f44c854SHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 24785f44c854SHong Zhang nzk = 0; 24795f44c854SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2480e32f2f54SBarry 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); 24815f44c854SHong Zhang ncols_upper = 0; 24825f44c854SHong Zhang for (j=0; j<ncols; j++){ 24835f44c854SHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 24845f44c854SHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 24855f44c854SHong Zhang ajtmp[ncols_upper] = i; 24865f44c854SHong Zhang ncols_upper++; 24875f44c854SHong Zhang } 24885f44c854SHong Zhang } 24895f44c854SHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 24905f44c854SHong Zhang nzk += nlnk; 24915f44c854SHong Zhang 24925f44c854SHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 24935f44c854SHong Zhang prow = jl[k]; /* 1st pivot row */ 24945f44c854SHong Zhang 24955f44c854SHong Zhang while (prow < k){ 24965f44c854SHong Zhang nextprow = jl[prow]; 24975f44c854SHong Zhang 24985f44c854SHong Zhang /* merge prow into k-th row */ 24995f44c854SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 25005f44c854SHong Zhang jmax = ui[prow+1]; 25015f44c854SHong Zhang ncols = jmax-jmin; 25025f44c854SHong Zhang i = jmin - ui[prow]; 25035f44c854SHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 25045f44c854SHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 25055f44c854SHong Zhang j = *(uj - 1); 25065f44c854SHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 25075f44c854SHong Zhang nzk += nlnk; 25085f44c854SHong Zhang 25095f44c854SHong Zhang /* update il and jl for prow */ 25105f44c854SHong Zhang if (jmin < jmax){ 25115f44c854SHong Zhang il[prow] = jmin; 25125f44c854SHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 25135f44c854SHong Zhang } 25145f44c854SHong Zhang prow = nextprow; 25155f44c854SHong Zhang } 25165f44c854SHong Zhang 25175f44c854SHong Zhang /* if free space is not available, make more free space */ 25185f44c854SHong Zhang if (current_space->local_remaining<nzk) { 25195f44c854SHong Zhang i = am - k + 1; /* num of unfactored rows */ 2520c6ae9bfcSHong Zhang i *= PetscMin(nzk, i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */ 25215f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 25225f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 25235f44c854SHong Zhang reallocs++; 25245f44c854SHong Zhang } 25255f44c854SHong Zhang 25265f44c854SHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 2527e32f2f54SBarry Smith if (nzk == 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 25285f44c854SHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 25295f44c854SHong Zhang 25305f44c854SHong Zhang /* add the k-th row into il and jl */ 25315f44c854SHong Zhang if (nzk > 1){ 25325f44c854SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 25335f44c854SHong Zhang jl[k] = jl[i]; jl[i] = k; 25345f44c854SHong Zhang il[k] = ui[k] + 1; 25355f44c854SHong Zhang } 25365f44c854SHong Zhang uj_ptr[k] = current_space->array; 25375f44c854SHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 25385f44c854SHong Zhang 25395f44c854SHong Zhang current_space->array += nzk; 25405f44c854SHong Zhang current_space->local_used += nzk; 25415f44c854SHong Zhang current_space->local_remaining -= nzk; 25425f44c854SHong Zhang 25435f44c854SHong Zhang current_space_lvl->array += nzk; 25445f44c854SHong Zhang current_space_lvl->local_used += nzk; 25455f44c854SHong Zhang current_space_lvl->local_remaining -= nzk; 25465f44c854SHong Zhang 25475f44c854SHong Zhang ui[k+1] = ui[k] + nzk; 25485f44c854SHong Zhang } 25495f44c854SHong Zhang 25505f44c854SHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 25515f44c854SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 25520e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 25535f44c854SHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 25545f44c854SHong Zhang 25559263d837SHong Zhang /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */ 25565f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 255735233d99SShri Abhyankar ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */ 25585f44c854SHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 25595f44c854SHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 25605f44c854SHong Zhang 25615f44c854SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 25625f44c854SHong Zhang 25635f44c854SHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 25645f44c854SHong Zhang b = (Mat_SeqSBAIJ*)(fact)->data; 25655f44c854SHong Zhang b->singlemalloc = PETSC_FALSE; 25665f44c854SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 25675f44c854SHong Zhang b->j = uj; 25685f44c854SHong Zhang b->i = ui; 25695f44c854SHong Zhang b->diag = udiag; 25707f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 25715f44c854SHong Zhang b->ilen = 0; 25725f44c854SHong Zhang b->imax = 0; 25735f44c854SHong Zhang b->row = perm; 25745f44c854SHong Zhang b->col = perm; 25755f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25765f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25775f44c854SHong Zhang b->icol = iperm; 25785f44c854SHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 25795f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2580d595f711SHong Zhang ierr = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 25815f44c854SHong Zhang b->maxnz = b->nz = ui[am]; 25825f44c854SHong Zhang b->free_a = PETSC_TRUE; 25835f44c854SHong Zhang b->free_ij = PETSC_TRUE; 25845f44c854SHong Zhang 2585f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2586f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 25875f44c854SHong Zhang if (ai[am] != 0) { 25886a69fef8SHong Zhang /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */ 258995b5ac22SHong Zhang fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am); 25905f44c854SHong Zhang } else { 2591f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 25925f44c854SHong Zhang } 25939263d837SHong Zhang #if defined(PETSC_USE_INFO) 25949263d837SHong Zhang if (ai[am] != 0) { 25959263d837SHong Zhang PetscReal af = fact->info.fill_ratio_needed; 25969263d837SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 25979263d837SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 25989263d837SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 25999263d837SHong Zhang } else { 26009263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 26019263d837SHong Zhang } 26029263d837SHong Zhang #endif 260335233d99SShri Abhyankar fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 26045f44c854SHong Zhang PetscFunctionReturn(0); 26055f44c854SHong Zhang } 26065f44c854SHong Zhang 26075f44c854SHong Zhang #undef __FUNCT__ 2608ad04f41aSHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_inplace" 2609ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 26105f44c854SHong Zhang { 26115f44c854SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 26125f44c854SHong Zhang Mat_SeqSBAIJ *b; 26135f44c854SHong Zhang PetscErrorCode ierr; 2614ace3abfcSBarry Smith PetscBool perm_identity,missing; 26155f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 26165f44c854SHong Zhang const PetscInt *rip,*riip; 26175f44c854SHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 26185f44c854SHong Zhang PetscInt nlnk,*lnk,*lnk_lvl=PETSC_NULL,d; 26195f44c854SHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 26205f44c854SHong Zhang PetscReal fill=info->fill,levels=info->levels; 26215f44c854SHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 26225f44c854SHong Zhang PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL; 26235f44c854SHong Zhang PetscBT lnkbt; 26245f44c854SHong Zhang IS iperm; 26255f44c854SHong Zhang 26265f44c854SHong Zhang PetscFunctionBegin; 2627e32f2f54SBarry 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); 26285f44c854SHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 2629e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 26305f44c854SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 26315f44c854SHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 26325f44c854SHong Zhang 26335f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 26345f44c854SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 26355f44c854SHong Zhang ui[0] = 0; 26365f44c854SHong Zhang 26375f44c854SHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 26385f44c854SHong Zhang if (!levels && perm_identity) { 26395f44c854SHong Zhang 26405f44c854SHong Zhang for (i=0; i<am; i++) { 26415f44c854SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 26425f44c854SHong Zhang udiag[i] = ui[i]; 2643ed59401aSHong Zhang } 264439e3d630SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 264539e3d630SHong Zhang cols = uj; 2646ed59401aSHong Zhang for (i=0; i<am; i++) { 2647ed59401aSHong Zhang aj = a->j + a->diag[i]; 264839e3d630SHong Zhang ncols = ui[i+1] - ui[i]; 264939e3d630SHong Zhang for (j=0; j<ncols; j++) *cols++ = *aj++; 2650ed59401aSHong Zhang } 265139e3d630SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 2652b635d36bSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 2653b635d36bSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 2654b635d36bSHong Zhang 26557a48dd6fSHong Zhang /* initialization */ 26565a8e39fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr); 26577a48dd6fSHong Zhang 26587a48dd6fSHong Zhang /* jl: linked list for storing indices of the pivot rows 26597a48dd6fSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 26600e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr); 26617a48dd6fSHong Zhang for (i=0; i<am; i++){ 26627a48dd6fSHong Zhang jl[i] = am; il[i] = 0; 26637a48dd6fSHong Zhang } 26647a48dd6fSHong Zhang 26657a48dd6fSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 26667a48dd6fSHong Zhang nlnk = am + 1; 26672ed133dbSHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 26687a48dd6fSHong Zhang 26697a48dd6fSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2670a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 26717a48dd6fSHong Zhang current_space = free_space; 2672a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 26737a48dd6fSHong Zhang current_space_lvl = free_space_lvl; 26747a48dd6fSHong Zhang 26757a48dd6fSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 26767a48dd6fSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 26777a48dd6fSHong Zhang nzk = 0; 2678622e2028SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2679e32f2f54SBarry 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); 2680622e2028SHong Zhang ncols_upper = 0; 2681622e2028SHong Zhang for (j=0; j<ncols; j++){ 2682b635d36bSHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 2683b635d36bSHong Zhang if (riip[i] >= k){ /* only take upper triangular entry */ 26845a8e39fbSHong Zhang ajtmp[ncols_upper] = i; 2685622e2028SHong Zhang ncols_upper++; 2686622e2028SHong Zhang } 2687622e2028SHong Zhang } 2688b635d36bSHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 26897a48dd6fSHong Zhang nzk += nlnk; 26907a48dd6fSHong Zhang 26917a48dd6fSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 26927a48dd6fSHong Zhang prow = jl[k]; /* 1st pivot row */ 26937a48dd6fSHong Zhang 26947a48dd6fSHong Zhang while (prow < k){ 26957a48dd6fSHong Zhang nextprow = jl[prow]; 26967a48dd6fSHong Zhang 26977a48dd6fSHong Zhang /* merge prow into k-th row */ 26987a48dd6fSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 26997a48dd6fSHong Zhang jmax = ui[prow+1]; 27007a48dd6fSHong Zhang ncols = jmax-jmin; 2701ed59401aSHong Zhang i = jmin - ui[prow]; 2702ed59401aSHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 27035a8e39fbSHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 27045a8e39fbSHong Zhang j = *(uj - 1); 27055a8e39fbSHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 27067a48dd6fSHong Zhang nzk += nlnk; 27077a48dd6fSHong Zhang 27087a48dd6fSHong Zhang /* update il and jl for prow */ 27097a48dd6fSHong Zhang if (jmin < jmax){ 27107a48dd6fSHong Zhang il[prow] = jmin; 27117a48dd6fSHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 27127a48dd6fSHong Zhang } 27137a48dd6fSHong Zhang prow = nextprow; 27147a48dd6fSHong Zhang } 27157a48dd6fSHong Zhang 27167a48dd6fSHong Zhang /* if free space is not available, make more free space */ 27177a48dd6fSHong Zhang if (current_space->local_remaining<nzk) { 27187a48dd6fSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2719fd3f55cbSMatthew Knepley i *= PetscMin(nzk, (i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2720a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 2721a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 27227a48dd6fSHong Zhang reallocs++; 27237a48dd6fSHong Zhang } 27247a48dd6fSHong Zhang 27257a48dd6fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 272665e19b50SBarry Smith if (!nzk) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 27272ed133dbSHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 27287a48dd6fSHong Zhang 27297a48dd6fSHong Zhang /* add the k-th row into il and jl */ 273039e3d630SHong Zhang if (nzk > 1){ 27317a48dd6fSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 27327a48dd6fSHong Zhang jl[k] = jl[i]; jl[i] = k; 27337a48dd6fSHong Zhang il[k] = ui[k] + 1; 27347a48dd6fSHong Zhang } 27357a48dd6fSHong Zhang uj_ptr[k] = current_space->array; 27367a48dd6fSHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 27377a48dd6fSHong Zhang 27387a48dd6fSHong Zhang current_space->array += nzk; 27397a48dd6fSHong Zhang current_space->local_used += nzk; 27407a48dd6fSHong Zhang current_space->local_remaining -= nzk; 27417a48dd6fSHong Zhang 27427a48dd6fSHong Zhang current_space_lvl->array += nzk; 27437a48dd6fSHong Zhang current_space_lvl->local_used += nzk; 27447a48dd6fSHong Zhang current_space_lvl->local_remaining -= nzk; 27457a48dd6fSHong Zhang 27467a48dd6fSHong Zhang ui[k+1] = ui[k] + nzk; 27477a48dd6fSHong Zhang } 27487a48dd6fSHong Zhang 27496cf91177SBarry Smith #if defined(PETSC_USE_INFO) 27507a48dd6fSHong Zhang if (ai[am] != 0) { 275139e3d630SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 2752ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2753ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2754ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 27557a48dd6fSHong Zhang } else { 2756ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 27577a48dd6fSHong Zhang } 275863ba0a88SBarry Smith #endif 27597a48dd6fSHong Zhang 27607a48dd6fSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 2761b635d36bSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 27620e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 27635a8e39fbSHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 27647a48dd6fSHong Zhang 27657a48dd6fSHong Zhang /* destroy list of free space and other temporary array(s) */ 27667a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 2767a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 27682ed133dbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2769a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 27707a48dd6fSHong Zhang 277139e3d630SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 277239e3d630SHong Zhang 27737a48dd6fSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 27747a48dd6fSHong Zhang 2775f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 27767a48dd6fSHong Zhang b->singlemalloc = PETSC_FALSE; 27777a48dd6fSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 27787a48dd6fSHong Zhang b->j = uj; 27797a48dd6fSHong Zhang b->i = ui; 27805f44c854SHong Zhang b->diag = udiag; 27817f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 27827a48dd6fSHong Zhang b->ilen = 0; 27837a48dd6fSHong Zhang b->imax = 0; 27847a48dd6fSHong Zhang b->row = perm; 2785b635d36bSHong Zhang b->col = perm; 2786b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2787b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2788b635d36bSHong Zhang b->icol = iperm; 27897a48dd6fSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 27907a48dd6fSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 2791f284f12aSHong Zhang ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 27927a48dd6fSHong Zhang b->maxnz = b->nz = ui[am]; 2793e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2794e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 27957a48dd6fSHong Zhang 2796f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2797f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 27987a48dd6fSHong Zhang if (ai[am] != 0) { 2799f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 28007a48dd6fSHong Zhang } else { 2801f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 28027a48dd6fSHong Zhang } 28030a3c351aSHong Zhang fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace; 2804a6175056SHong Zhang PetscFunctionReturn(0); 2805a6175056SHong Zhang } 2806d5d45c9bSBarry Smith 2807177d4faaSHong Zhang #undef __FUNCT__ 2808177d4faaSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ" 280935233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 28100be760fbSHong Zhang { 28110be760fbSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 28120be760fbSHong Zhang Mat_SeqSBAIJ *b; 28130be760fbSHong Zhang PetscErrorCode ierr; 2814ace3abfcSBarry Smith PetscBool perm_identity; 28150be760fbSHong Zhang PetscReal fill = info->fill; 28160be760fbSHong Zhang const PetscInt *rip,*riip; 28170be760fbSHong Zhang PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 28180be760fbSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 28190be760fbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag; 28200be760fbSHong Zhang PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 28210be760fbSHong Zhang PetscBT lnkbt; 28220be760fbSHong Zhang IS iperm; 28230be760fbSHong Zhang 28240be760fbSHong Zhang PetscFunctionBegin; 2825e32f2f54SBarry 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); 28260be760fbSHong Zhang /* check whether perm is the identity mapping */ 28270be760fbSHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 28280be760fbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 28290be760fbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 28300be760fbSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 28310be760fbSHong Zhang 28320be760fbSHong Zhang /* initialization */ 28330be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 28340be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr); 28350be760fbSHong Zhang ui[0] = 0; 28360be760fbSHong Zhang 28370be760fbSHong Zhang /* jl: linked list for storing indices of the pivot rows 28380be760fbSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 28390be760fbSHong Zhang ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr); 28400be760fbSHong Zhang for (i=0; i<am; i++){ 28410be760fbSHong Zhang jl[i] = am; il[i] = 0; 28420be760fbSHong Zhang } 28430be760fbSHong Zhang 28440be760fbSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 28450be760fbSHong Zhang nlnk = am + 1; 28460be760fbSHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28470be760fbSHong Zhang 284895b5ac22SHong Zhang /* initial FreeSpace size is fill*(ai[am]+am)/2 */ 284995b5ac22SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space);CHKERRQ(ierr); 28500be760fbSHong Zhang current_space = free_space; 28510be760fbSHong Zhang 28520be760fbSHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 28530be760fbSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 28540be760fbSHong Zhang nzk = 0; 28550be760fbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2856e32f2f54SBarry 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); 28570be760fbSHong Zhang ncols_upper = 0; 28580be760fbSHong Zhang for (j=0; j<ncols; j++){ 28590be760fbSHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 28600be760fbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 28610be760fbSHong Zhang cols[ncols_upper] = i; 28620be760fbSHong Zhang ncols_upper++; 28630be760fbSHong Zhang } 28640be760fbSHong Zhang } 28650be760fbSHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28660be760fbSHong Zhang nzk += nlnk; 28670be760fbSHong Zhang 28680be760fbSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 28690be760fbSHong Zhang prow = jl[k]; /* 1st pivot row */ 28700be760fbSHong Zhang 28710be760fbSHong Zhang while (prow < k){ 28720be760fbSHong Zhang nextprow = jl[prow]; 28730be760fbSHong Zhang /* merge prow into k-th row */ 28740be760fbSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 28750be760fbSHong Zhang jmax = ui[prow+1]; 28760be760fbSHong Zhang ncols = jmax-jmin; 28770be760fbSHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 28780be760fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28790be760fbSHong Zhang nzk += nlnk; 28800be760fbSHong Zhang 28810be760fbSHong Zhang /* update il and jl for prow */ 28820be760fbSHong Zhang if (jmin < jmax){ 28830be760fbSHong Zhang il[prow] = jmin; 28840be760fbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 28850be760fbSHong Zhang } 28860be760fbSHong Zhang prow = nextprow; 28870be760fbSHong Zhang } 28880be760fbSHong Zhang 28890be760fbSHong Zhang /* if free space is not available, make more free space */ 28900be760fbSHong Zhang if (current_space->local_remaining<nzk) { 28910be760fbSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2892c6ae9bfcSHong Zhang i *= PetscMin(nzk,i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */ 28930be760fbSHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 28940be760fbSHong Zhang reallocs++; 28950be760fbSHong Zhang } 28960be760fbSHong Zhang 28970be760fbSHong Zhang /* copy data into free space, then initialize lnk */ 28980be760fbSHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 28990be760fbSHong Zhang 29000be760fbSHong Zhang /* add the k-th row into il and jl */ 29017b056e98SHong Zhang if (nzk > 1){ 29020be760fbSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 29030be760fbSHong Zhang jl[k] = jl[i]; jl[i] = k; 29040be760fbSHong Zhang il[k] = ui[k] + 1; 29050be760fbSHong Zhang } 29060be760fbSHong Zhang ui_ptr[k] = current_space->array; 29070be760fbSHong Zhang current_space->array += nzk; 29080be760fbSHong Zhang current_space->local_used += nzk; 29090be760fbSHong Zhang current_space->local_remaining -= nzk; 29100be760fbSHong Zhang 29110be760fbSHong Zhang ui[k+1] = ui[k] + nzk; 29120be760fbSHong Zhang } 29130be760fbSHong Zhang 29140be760fbSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 29150be760fbSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 29160be760fbSHong Zhang ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 29170be760fbSHong Zhang 29189263d837SHong Zhang /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */ 29190be760fbSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 292035233d99SShri Abhyankar ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */ 29210be760fbSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 29220be760fbSHong Zhang 29230be760fbSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 29240be760fbSHong Zhang 2925f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 29260be760fbSHong Zhang b->singlemalloc = PETSC_FALSE; 29270be760fbSHong Zhang b->free_a = PETSC_TRUE; 29280be760fbSHong Zhang b->free_ij = PETSC_TRUE; 29290be760fbSHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 29300be760fbSHong Zhang b->j = uj; 29310be760fbSHong Zhang b->i = ui; 29320be760fbSHong Zhang b->diag = udiag; 29330be760fbSHong Zhang b->free_diag = PETSC_TRUE; 29340be760fbSHong Zhang b->ilen = 0; 29350be760fbSHong Zhang b->imax = 0; 29360be760fbSHong Zhang b->row = perm; 29370be760fbSHong Zhang b->col = perm; 29380be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29390be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29400be760fbSHong Zhang b->icol = iperm; 29410be760fbSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 29420be760fbSHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 29430860e08cSHong Zhang ierr = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 29440be760fbSHong Zhang b->maxnz = b->nz = ui[am]; 29450be760fbSHong Zhang 2946f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2947f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 29480be760fbSHong Zhang if (ai[am] != 0) { 29496a69fef8SHong Zhang /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */ 295095b5ac22SHong Zhang fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am); 29510be760fbSHong Zhang } else { 2952f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 29530be760fbSHong Zhang } 29549263d837SHong Zhang #if defined(PETSC_USE_INFO) 29559263d837SHong Zhang if (ai[am] != 0) { 29569263d837SHong Zhang PetscReal af = fact->info.fill_ratio_needed; 29579263d837SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 29589263d837SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 29599263d837SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 29609263d837SHong Zhang } else { 29619263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 29629263d837SHong Zhang } 29639263d837SHong Zhang #endif 296435233d99SShri Abhyankar fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 29650be760fbSHong Zhang PetscFunctionReturn(0); 29660be760fbSHong Zhang } 29670be760fbSHong Zhang 2968f76d2b81SHong Zhang #undef __FUNCT__ 2969ad04f41aSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ_inplace" 2970ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2971f76d2b81SHong Zhang { 2972f76d2b81SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 297310c27e3dSHong Zhang Mat_SeqSBAIJ *b; 2974dfbe8321SBarry Smith PetscErrorCode ierr; 2975ace3abfcSBarry Smith PetscBool perm_identity; 297610c27e3dSHong Zhang PetscReal fill = info->fill; 29775d0c19d7SBarry Smith const PetscInt *rip,*riip; 29785d0c19d7SBarry Smith PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 297910c27e3dSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 29802ed133dbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr; 2981a1a86e44SBarry Smith PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL; 298210c27e3dSHong Zhang PetscBT lnkbt; 29832ed133dbSHong Zhang IS iperm; 2984f76d2b81SHong Zhang 2985f76d2b81SHong Zhang PetscFunctionBegin; 2986e32f2f54SBarry 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); 29872ed133dbSHong Zhang /* check whether perm is the identity mapping */ 2988f76d2b81SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 29892ed133dbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 29902ed133dbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 29919bfd6278SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 299210c27e3dSHong Zhang 299310c27e3dSHong Zhang /* initialization */ 29941393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr); 299510c27e3dSHong Zhang ui[0] = 0; 299610c27e3dSHong Zhang 299710c27e3dSHong Zhang /* jl: linked list for storing indices of the pivot rows 29981393bc97SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 29990e83c824SBarry Smith ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr); 30001393bc97SHong Zhang for (i=0; i<am; i++){ 30011393bc97SHong Zhang jl[i] = am; il[i] = 0; 3002f76d2b81SHong Zhang } 300310c27e3dSHong Zhang 300410c27e3dSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 30051393bc97SHong Zhang nlnk = am + 1; 30061393bc97SHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 300710c27e3dSHong Zhang 30081393bc97SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 3009a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 301010c27e3dSHong Zhang current_space = free_space; 301110c27e3dSHong Zhang 30121393bc97SHong Zhang for (k=0; k<am; k++){ /* for each active row k */ 301310c27e3dSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 301410c27e3dSHong Zhang nzk = 0; 30152ed133dbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 3016e32f2f54SBarry 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); 30172ed133dbSHong Zhang ncols_upper = 0; 30182ed133dbSHong Zhang for (j=0; j<ncols; j++){ 30199bfd6278SHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 30202ed133dbSHong Zhang if (i >= k){ /* only take upper triangular entry */ 30212ed133dbSHong Zhang cols[ncols_upper] = i; 30222ed133dbSHong Zhang ncols_upper++; 30232ed133dbSHong Zhang } 30242ed133dbSHong Zhang } 30251393bc97SHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 302610c27e3dSHong Zhang nzk += nlnk; 302710c27e3dSHong Zhang 302810c27e3dSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 302910c27e3dSHong Zhang prow = jl[k]; /* 1st pivot row */ 303010c27e3dSHong Zhang 303110c27e3dSHong Zhang while (prow < k){ 303210c27e3dSHong Zhang nextprow = jl[prow]; 303310c27e3dSHong Zhang /* merge prow into k-th row */ 30341393bc97SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 303510c27e3dSHong Zhang jmax = ui[prow+1]; 303610c27e3dSHong Zhang ncols = jmax-jmin; 30371393bc97SHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 30385a8e39fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 303910c27e3dSHong Zhang nzk += nlnk; 304010c27e3dSHong Zhang 304110c27e3dSHong Zhang /* update il and jl for prow */ 304210c27e3dSHong Zhang if (jmin < jmax){ 304310c27e3dSHong Zhang il[prow] = jmin; 30442ed133dbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 304510c27e3dSHong Zhang } 304610c27e3dSHong Zhang prow = nextprow; 304710c27e3dSHong Zhang } 304810c27e3dSHong Zhang 304910c27e3dSHong Zhang /* if free space is not available, make more free space */ 305010c27e3dSHong Zhang if (current_space->local_remaining<nzk) { 30511393bc97SHong Zhang i = am - k + 1; /* num of unfactored rows */ 305210c27e3dSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 3053a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 305410c27e3dSHong Zhang reallocs++; 305510c27e3dSHong Zhang } 305610c27e3dSHong Zhang 305710c27e3dSHong Zhang /* copy data into free space, then initialize lnk */ 30581393bc97SHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 305910c27e3dSHong Zhang 306010c27e3dSHong Zhang /* add the k-th row into il and jl */ 306110c27e3dSHong Zhang if (nzk-1 > 0){ 30621393bc97SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 306310c27e3dSHong Zhang jl[k] = jl[i]; jl[i] = k; 306410c27e3dSHong Zhang il[k] = ui[k] + 1; 306510c27e3dSHong Zhang } 30662ed133dbSHong Zhang ui_ptr[k] = current_space->array; 306710c27e3dSHong Zhang current_space->array += nzk; 306810c27e3dSHong Zhang current_space->local_used += nzk; 306910c27e3dSHong Zhang current_space->local_remaining -= nzk; 307010c27e3dSHong Zhang 307110c27e3dSHong Zhang ui[k+1] = ui[k] + nzk; 307210c27e3dSHong Zhang } 307310c27e3dSHong Zhang 30746cf91177SBarry Smith #if defined(PETSC_USE_INFO) 30751393bc97SHong Zhang if (ai[am] != 0) { 30761393bc97SHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 3077ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 3078ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 3079ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 308010c27e3dSHong Zhang } else { 3081ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 308210c27e3dSHong Zhang } 308363ba0a88SBarry Smith #endif 308410c27e3dSHong Zhang 308510c27e3dSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 30869bfd6278SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 30870e83c824SBarry Smith ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 308810c27e3dSHong Zhang 308910c27e3dSHong Zhang /* destroy list of free space and other temporary array(s) */ 30901393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr); 3091a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 309210c27e3dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 309310c27e3dSHong Zhang 309410c27e3dSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 309510c27e3dSHong Zhang 3096f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 309710c27e3dSHong Zhang b->singlemalloc = PETSC_FALSE; 3098e6b907acSBarry Smith b->free_a = PETSC_TRUE; 3099e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 31001393bc97SHong Zhang ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr); 310110c27e3dSHong Zhang b->j = uj; 310210c27e3dSHong Zhang b->i = ui; 310310c27e3dSHong Zhang b->diag = 0; 310410c27e3dSHong Zhang b->ilen = 0; 310510c27e3dSHong Zhang b->imax = 0; 310610c27e3dSHong Zhang b->row = perm; 31079bfd6278SHong Zhang b->col = perm; 31089bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 31099bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 31109bfd6278SHong Zhang b->icol = iperm; 311110c27e3dSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 31121393bc97SHong Zhang ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3113719d5645SBarry Smith ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 31141393bc97SHong Zhang b->maxnz = b->nz = ui[am]; 311510c27e3dSHong Zhang 3116f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 3117f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 31181393bc97SHong Zhang if (ai[am] != 0) { 3119f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 312010c27e3dSHong Zhang } else { 3121f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 312210c27e3dSHong Zhang } 31230a3c351aSHong Zhang fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace; 3124f76d2b81SHong Zhang PetscFunctionReturn(0); 3125f76d2b81SHong Zhang } 3126599ef60dSHong Zhang 31271d098868SHong Zhang #undef __FUNCT__ 312835233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering" 312935233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 3130813a1f2bSShri Abhyankar { 3131813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3132813a1f2bSShri Abhyankar PetscErrorCode ierr; 3133813a1f2bSShri Abhyankar PetscInt n = A->rmap->n; 3134813a1f2bSShri Abhyankar const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag,*vi; 3135813a1f2bSShri Abhyankar PetscScalar *x,sum; 3136813a1f2bSShri Abhyankar const PetscScalar *b; 3137813a1f2bSShri Abhyankar const MatScalar *aa = a->a,*v; 3138813a1f2bSShri Abhyankar PetscInt i,nz; 3139813a1f2bSShri Abhyankar 3140813a1f2bSShri Abhyankar PetscFunctionBegin; 3141813a1f2bSShri Abhyankar if (!n) PetscFunctionReturn(0); 3142813a1f2bSShri Abhyankar 31433649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 3144813a1f2bSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3145813a1f2bSShri Abhyankar 3146813a1f2bSShri Abhyankar /* forward solve the lower triangular */ 3147813a1f2bSShri Abhyankar x[0] = b[0]; 3148813a1f2bSShri Abhyankar v = aa; 3149813a1f2bSShri Abhyankar vi = aj; 3150813a1f2bSShri Abhyankar for (i=1; i<n; i++) { 3151813a1f2bSShri Abhyankar nz = ai[i+1] - ai[i]; 3152813a1f2bSShri Abhyankar sum = b[i]; 3153813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 3154813a1f2bSShri Abhyankar v += nz; 3155813a1f2bSShri Abhyankar vi += nz; 3156813a1f2bSShri Abhyankar x[i] = sum; 3157813a1f2bSShri Abhyankar } 3158813a1f2bSShri Abhyankar 3159813a1f2bSShri Abhyankar /* backward solve the upper triangular */ 316062fbd6afSShri Abhyankar for (i=n-1; i>=0; i--){ 31613c0119dfSShri Abhyankar v = aa + adiag[i+1] + 1; 31623c0119dfSShri Abhyankar vi = aj + adiag[i+1] + 1; 3163813a1f2bSShri Abhyankar nz = adiag[i] - adiag[i+1]-1; 3164813a1f2bSShri Abhyankar sum = x[i]; 3165813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 31663c0119dfSShri Abhyankar x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */ 3167813a1f2bSShri Abhyankar } 3168813a1f2bSShri Abhyankar 3169813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 31703649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3171813a1f2bSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3172813a1f2bSShri Abhyankar PetscFunctionReturn(0); 3173813a1f2bSShri Abhyankar } 3174813a1f2bSShri Abhyankar 31751d098868SHong Zhang #undef __FUNCT__ 317635233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ" 317735233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 3178f268cba8SShri Abhyankar { 3179f268cba8SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3180f268cba8SShri Abhyankar IS iscol = a->col,isrow = a->row; 3181f268cba8SShri Abhyankar PetscErrorCode ierr; 3182f268cba8SShri Abhyankar PetscInt i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz; 3183f268cba8SShri Abhyankar const PetscInt *rout,*cout,*r,*c; 3184f268cba8SShri Abhyankar PetscScalar *x,*tmp,sum; 3185f268cba8SShri Abhyankar const PetscScalar *b; 3186f268cba8SShri Abhyankar const MatScalar *aa = a->a,*v; 3187f268cba8SShri Abhyankar 3188f268cba8SShri Abhyankar PetscFunctionBegin; 3189f268cba8SShri Abhyankar if (!n) PetscFunctionReturn(0); 3190f268cba8SShri Abhyankar 31913649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 3192f268cba8SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3193f268cba8SShri Abhyankar tmp = a->solve_work; 3194f268cba8SShri Abhyankar 3195f268cba8SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 3196f268cba8SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 3197f268cba8SShri Abhyankar 3198f268cba8SShri Abhyankar /* forward solve the lower triangular */ 3199f268cba8SShri Abhyankar tmp[0] = b[r[0]]; 3200f268cba8SShri Abhyankar v = aa; 3201f268cba8SShri Abhyankar vi = aj; 3202f268cba8SShri Abhyankar for (i=1; i<n; i++) { 3203f268cba8SShri Abhyankar nz = ai[i+1] - ai[i]; 3204f268cba8SShri Abhyankar sum = b[r[i]]; 3205f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3206f268cba8SShri Abhyankar tmp[i] = sum; 3207f268cba8SShri Abhyankar v += nz; vi += nz; 3208f268cba8SShri Abhyankar } 3209f268cba8SShri Abhyankar 3210f268cba8SShri Abhyankar /* backward solve the upper triangular */ 3211f268cba8SShri Abhyankar for (i=n-1; i>=0; i--){ 32123c0119dfSShri Abhyankar v = aa + adiag[i+1]+1; 32133c0119dfSShri Abhyankar vi = aj + adiag[i+1]+1; 3214f268cba8SShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 3215f268cba8SShri Abhyankar sum = tmp[i]; 3216f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3217f268cba8SShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 3218f268cba8SShri Abhyankar } 3219f268cba8SShri Abhyankar 3220f268cba8SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 3221f268cba8SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 32223649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3223f268cba8SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3224f268cba8SShri Abhyankar ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 3225f268cba8SShri Abhyankar PetscFunctionReturn(0); 3226f268cba8SShri Abhyankar } 3227f268cba8SShri Abhyankar 3228f268cba8SShri Abhyankar #undef __FUNCT__ 32291d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ" 3230fe97e370SBarry Smith /* 3231fe97e370SBarry 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 3232fe97e370SBarry Smith */ 32331d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact) 32341d098868SHong Zhang { 32351d098868SHong Zhang Mat B = *fact; 3236599ef60dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b; 3237599ef60dSHong Zhang IS isicol; 3238599ef60dSHong Zhang PetscErrorCode ierr; 32391d098868SHong Zhang const PetscInt *r,*ic; 32401d098868SHong Zhang PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag; 3241dc3a2fd3SHong Zhang PetscInt *bi,*bj,*bdiag,*bdiag_rev; 3242f61a948aSLisandro Dalcin PetscInt row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au; 32431d098868SHong Zhang PetscInt nlnk,*lnk; 32441d098868SHong Zhang PetscBT lnkbt; 3245ace3abfcSBarry Smith PetscBool row_identity,icol_identity; 32468aee2decSHong Zhang MatScalar *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp; 32471d098868SHong Zhang const PetscInt *ics; 32481d098868SHong Zhang PetscInt j,nz,*pj,*bjtmp,k,ncut,*jtmp; 3249182b8fbaSHong Zhang PetscReal dt=info->dt,dtcol=info->dtcol,shift=info->shiftamount; 3250f61a948aSLisandro Dalcin PetscInt dtcount=(PetscInt)info->dtcount,nnz_max; 3251ace3abfcSBarry Smith PetscBool missing; 3252599ef60dSHong Zhang 3253599ef60dSHong Zhang PetscFunctionBegin; 3254f61a948aSLisandro Dalcin 3255f61a948aSLisandro Dalcin if (dt == PETSC_DEFAULT) dt = 0.005; 3256f61a948aSLisandro Dalcin if (dtcol == PETSC_DEFAULT) dtcol = 0.01; /* XXX unused! */ 3257f61a948aSLisandro Dalcin if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax); 3258f61a948aSLisandro Dalcin 32591d098868SHong Zhang /* ------- symbolic factorization, can be reused ---------*/ 32601d098868SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 3261e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 32621d098868SHong Zhang adiag=a->diag; 3263599ef60dSHong Zhang 3264599ef60dSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 3265599ef60dSHong Zhang 3266599ef60dSHong Zhang /* bdiag is location of diagonal in factor */ 32671bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr); /* becomes b->diag */ 32681bfa06eaSJed Brown ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag_rev);CHKERRQ(ierr); /* temporary */ 3269599ef60dSHong Zhang 32701d098868SHong Zhang /* allocate row pointers bi */ 32718fc3a347SHong Zhang ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); 32721d098868SHong Zhang 32731d098868SHong Zhang /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */ 3274393d3a68SHong Zhang if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */ 32751d098868SHong Zhang nnz_max = ai[n]+2*n*dtcount+2; 32768fc3a347SHong Zhang 32778fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr); 32788fc3a347SHong Zhang ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr); 3279599ef60dSHong Zhang 3280599ef60dSHong Zhang /* put together the new matrix */ 3281599ef60dSHong Zhang ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 3282599ef60dSHong Zhang ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr); 3283f284f12aSHong Zhang b = (Mat_SeqAIJ*)B->data; 3284599ef60dSHong Zhang b->free_a = PETSC_TRUE; 3285599ef60dSHong Zhang b->free_ij = PETSC_TRUE; 3286599ef60dSHong Zhang b->singlemalloc = PETSC_FALSE; 3287599ef60dSHong Zhang b->a = ba; 3288599ef60dSHong Zhang b->j = bj; 3289599ef60dSHong Zhang b->i = bi; 3290599ef60dSHong Zhang b->diag = bdiag; 3291599ef60dSHong Zhang b->ilen = 0; 3292599ef60dSHong Zhang b->imax = 0; 3293599ef60dSHong Zhang b->row = isrow; 3294599ef60dSHong Zhang b->col = iscol; 3295599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 3296599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 3297599ef60dSHong Zhang b->icol = isicol; 3298599ef60dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 3299599ef60dSHong Zhang 33001d098868SHong Zhang ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 33011d098868SHong Zhang b->maxnz = nnz_max; 3302599ef60dSHong Zhang 3303d5f3da31SBarry Smith B->factortype = MAT_FACTOR_ILUDT; 3304f284f12aSHong Zhang B->info.factor_mallocs = 0; 3305f284f12aSHong Zhang B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]); 33061d098868SHong Zhang CHKMEMQ; 33071d098868SHong Zhang /* ------- end of symbolic factorization ---------*/ 3308599ef60dSHong Zhang 3309599ef60dSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 3310599ef60dSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3311599ef60dSHong Zhang ics = ic; 3312599ef60dSHong Zhang 3313599ef60dSHong Zhang /* linked list for storing column indices of the active row */ 3314599ef60dSHong Zhang nlnk = n + 1; 3315599ef60dSHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 33161d098868SHong Zhang 33171d098868SHong Zhang /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */ 33180e83c824SBarry Smith ierr = PetscMalloc2(n,PetscInt,&im,n,PetscInt,&jtmp);CHKERRQ(ierr); 33191d098868SHong Zhang /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */ 33200e83c824SBarry Smith ierr = PetscMalloc2(n,MatScalar,&rtmp,n,MatScalar,&vtmp);CHKERRQ(ierr); 33210e83c824SBarry Smith ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr); 3322599ef60dSHong Zhang 3323599ef60dSHong Zhang bi[0] = 0; 33248fc3a347SHong Zhang bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */ 3325dc3a2fd3SHong Zhang bdiag_rev[n] = bdiag[0]; 33268fc3a347SHong Zhang bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */ 3327599ef60dSHong Zhang for (i=0; i<n; i++) { 3328599ef60dSHong Zhang /* copy initial fill into linked list */ 3329599ef60dSHong Zhang nzi = 0; /* nonzeros for active row i */ 33308369b28dSHong Zhang nzi = ai[r[i]+1] - ai[r[i]]; 3331e32f2f54SBarry 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); 33328369b28dSHong Zhang nzi_al = adiag[r[i]] - ai[r[i]]; 33338369b28dSHong Zhang nzi_au = ai[r[i]+1] - adiag[r[i]] -1; 3334599ef60dSHong Zhang ajtmp = aj + ai[r[i]]; 33358369b28dSHong Zhang ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3336599ef60dSHong Zhang 3337599ef60dSHong Zhang /* load in initial (unfactored row) */ 33381d098868SHong Zhang aatmp = a->a + ai[r[i]]; 33398369b28dSHong Zhang for (j=0; j<nzi; j++) { 33401d098868SHong Zhang rtmp[ics[*ajtmp++]] = *aatmp++; 3341599ef60dSHong Zhang } 3342599ef60dSHong Zhang 3343599ef60dSHong Zhang /* add pivot rows into linked list */ 3344599ef60dSHong Zhang row = lnk[n]; 3345599ef60dSHong Zhang while (row < i ) { 33461d098868SHong Zhang nzi_bl = bi[row+1] - bi[row] + 1; 33471d098868SHong Zhang bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */ 33481d098868SHong Zhang ierr = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr); 3349599ef60dSHong Zhang nzi += nlnk; 3350599ef60dSHong Zhang row = lnk[row]; 3351599ef60dSHong Zhang } 3352599ef60dSHong Zhang 33538369b28dSHong Zhang /* copy data from lnk into jtmp, then initialize lnk */ 33548369b28dSHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr); 3355599ef60dSHong Zhang 3356599ef60dSHong Zhang /* numerical factorization */ 33578369b28dSHong Zhang bjtmp = jtmp; 3358599ef60dSHong Zhang row = *bjtmp++; /* 1st pivot row */ 3359599ef60dSHong Zhang while ( row < i ) { 3360599ef60dSHong Zhang pc = rtmp + row; 33613c2a7987SHong Zhang pv = ba + bdiag[row]; /* 1./(diag of the pivot row) */ 33623c2a7987SHong Zhang multiplier = (*pc) * (*pv); 3363599ef60dSHong Zhang *pc = multiplier; 33643c2a7987SHong Zhang if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */ 33651d098868SHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 33661d098868SHong Zhang pv = ba + bdiag[row+1] + 1; 33671d098868SHong Zhang /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */ 33681d098868SHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 33691d098868SHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 3370dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 3371599ef60dSHong Zhang } 3372599ef60dSHong Zhang row = *bjtmp++; 3373599ef60dSHong Zhang } 3374599ef60dSHong Zhang 33758369b28dSHong Zhang /* copy sparse rtmp into contiguous vtmp; separate L and U part */ 33768aee2decSHong Zhang diag_tmp = rtmp[i]; /* save diagonal value - may not needed?? */ 33778369b28dSHong Zhang nzi_bl = 0; j = 0; 33788369b28dSHong Zhang while (jtmp[j] < i){ /* Note: jtmp is sorted */ 33798aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 33808369b28dSHong Zhang nzi_bl++; j++; 33818369b28dSHong Zhang } 33828369b28dSHong Zhang nzi_bu = nzi - nzi_bl -1; 33838369b28dSHong Zhang while (j < nzi){ 33848aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 33858369b28dSHong Zhang j++; 33868369b28dSHong Zhang } 33878369b28dSHong Zhang 3388599ef60dSHong Zhang bjtmp = bj + bi[i]; 3389599ef60dSHong Zhang batmp = ba + bi[i]; 33908369b28dSHong Zhang /* apply level dropping rule to L part */ 33918369b28dSHong Zhang ncut = nzi_al + dtcount; 33928369b28dSHong Zhang if (ncut < nzi_bl){ 33938369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr); 33948369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr); 3395599ef60dSHong Zhang } else { 33968369b28dSHong Zhang ncut = nzi_bl; 33978369b28dSHong Zhang } 33988369b28dSHong Zhang for (j=0; j<ncut; j++){ 33998369b28dSHong Zhang bjtmp[j] = jtmp[j]; 34008369b28dSHong Zhang batmp[j] = vtmp[j]; 34016da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */ 34028369b28dSHong Zhang } 34031d098868SHong Zhang bi[i+1] = bi[i] + ncut; 34048369b28dSHong Zhang nzi = ncut + 1; 34058369b28dSHong Zhang 34068369b28dSHong Zhang /* apply level dropping rule to U part */ 34078369b28dSHong Zhang ncut = nzi_au + dtcount; 34088369b28dSHong Zhang if (ncut < nzi_bu){ 34098369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr); 34108369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr); 34118369b28dSHong Zhang } else { 34128369b28dSHong Zhang ncut = nzi_bu; 34138369b28dSHong Zhang } 34148369b28dSHong Zhang nzi += ncut; 34151d098868SHong Zhang 34161d098868SHong Zhang /* mark bdiagonal */ 34171d098868SHong Zhang bdiag[i+1] = bdiag[i] - (ncut + 1); 3418dc3a2fd3SHong Zhang bdiag_rev[n-i-1] = bdiag[i+1]; 34198fc3a347SHong Zhang bi[2*n - i] = bi[2*n - i +1] - (ncut + 1); 34201d098868SHong Zhang bjtmp = bj + bdiag[i]; 34211d098868SHong Zhang batmp = ba + bdiag[i]; 34221d098868SHong Zhang *bjtmp = i; 34238aee2decSHong Zhang *batmp = diag_tmp; /* rtmp[i]; */ 3424c9c72f8fSHong Zhang if (*batmp == 0.0) { 3425c9c72f8fSHong Zhang *batmp = dt+shift; 3426bd1bc851SBarry Smith /* printf(" row %d add shift %g\n",i,shift); */ 3427c9c72f8fSHong Zhang } 3428b78a477dSHong Zhang *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */ 34296da515a1SHong Zhang /* printf(" (%d,%g),",*bjtmp,*batmp); */ 34301d098868SHong Zhang 34311d098868SHong Zhang bjtmp = bj + bdiag[i+1]+1; 34321d098868SHong Zhang batmp = ba + bdiag[i+1]+1; 34338369b28dSHong Zhang for (k=0; k<ncut; k++){ 34341d098868SHong Zhang bjtmp[k] = jtmp[nzi_bl+1+k]; 34351d098868SHong Zhang batmp[k] = vtmp[nzi_bl+1+k]; 34366da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */ 3437599ef60dSHong Zhang } 34386da515a1SHong Zhang /* printf("\n"); */ 3439599ef60dSHong Zhang 3440599ef60dSHong Zhang im[i] = nzi; /* used by PetscLLAddSortedLU() */ 3441599ef60dSHong Zhang /* 34421d098868SHong Zhang printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]); 34438369b28dSHong Zhang printf(" ----------------------------\n"); 3444599ef60dSHong Zhang */ 34458369b28dSHong Zhang } /* for (i=0; i<n; i++) */ 34461d098868SHong Zhang /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */ 3447e32f2f54SBarry 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]); 3448599ef60dSHong Zhang 3449599ef60dSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3450599ef60dSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3451599ef60dSHong Zhang 3452599ef60dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 34530e83c824SBarry Smith ierr = PetscFree2(im,jtmp);CHKERRQ(ierr); 34540e83c824SBarry Smith ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr); 34551bfa06eaSJed Brown ierr = PetscFree(bdiag_rev);CHKERRQ(ierr); 3456599ef60dSHong Zhang 3457599ef60dSHong Zhang ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr); 34581d098868SHong Zhang b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n]; 3459599ef60dSHong Zhang 3460599ef60dSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 3461599ef60dSHong Zhang ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr); 3462599ef60dSHong Zhang if (row_identity && icol_identity) { 346335233d99SShri Abhyankar B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 3464599ef60dSHong Zhang } else { 346535233d99SShri Abhyankar B->ops->solve = MatSolve_SeqAIJ; 3466599ef60dSHong Zhang } 3467599ef60dSHong Zhang 34681d098868SHong Zhang B->ops->solveadd = 0; 34691d098868SHong Zhang B->ops->solvetranspose = 0; 34701d098868SHong Zhang B->ops->solvetransposeadd = 0; 34711d098868SHong Zhang B->ops->matsolve = 0; 3472599ef60dSHong Zhang B->assembled = PETSC_TRUE; 3473599ef60dSHong Zhang B->preallocated = PETSC_TRUE; 3474599ef60dSHong Zhang PetscFunctionReturn(0); 3475599ef60dSHong Zhang } 3476599ef60dSHong Zhang 34773c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */ 34783c2a7987SHong Zhang #undef __FUNCT__ 34793c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ" 3480fe97e370SBarry Smith /* 3481fe97e370SBarry 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 3482fe97e370SBarry Smith */ 3483fe97e370SBarry Smith 34847087cfbeSBarry Smith PetscErrorCode MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info) 34853c2a7987SHong Zhang { 34863c2a7987SHong Zhang PetscErrorCode ierr; 34873c2a7987SHong Zhang 34883c2a7987SHong Zhang PetscFunctionBegin; 34893c2a7987SHong Zhang ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr); 34903c2a7987SHong Zhang PetscFunctionReturn(0); 34913c2a7987SHong Zhang } 34923c2a7987SHong Zhang 34933c2a7987SHong Zhang /* 34943c2a7987SHong Zhang same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors 34953c2a7987SHong Zhang - intend to replace existing MatLUFactorNumeric_SeqAIJ() 34963c2a7987SHong Zhang */ 34973c2a7987SHong Zhang #undef __FUNCT__ 34983c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ" 3499fe97e370SBarry Smith /* 3500fe97e370SBarry 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 3501fe97e370SBarry Smith */ 3502fe97e370SBarry Smith 35037087cfbeSBarry Smith PetscErrorCode MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info) 35043c2a7987SHong Zhang { 35053c2a7987SHong Zhang Mat C=fact; 35063c2a7987SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data; 35073c2a7987SHong Zhang IS isrow = b->row,isicol = b->icol; 35083c2a7987SHong Zhang PetscErrorCode ierr; 35093c2a7987SHong Zhang const PetscInt *r,*ic,*ics; 3510b78a477dSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 3511b78a477dSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj; 35123c2a7987SHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 3513182b8fbaSHong Zhang PetscReal dt=info->dt,shift=info->shiftamount; 3514ace3abfcSBarry Smith PetscBool row_identity, col_identity; 35153c2a7987SHong Zhang 35163c2a7987SHong Zhang PetscFunctionBegin; 35173c2a7987SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 35183c2a7987SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3519b78a477dSHong Zhang ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 35203c2a7987SHong Zhang ics = ic; 35213c2a7987SHong Zhang 35223c2a7987SHong Zhang for (i=0; i<n; i++){ 3523b78a477dSHong Zhang /* initialize rtmp array */ 3524b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of nozeros in L(i,:) */ 35253c2a7987SHong Zhang bjtmp = bj + bi[i]; 3526b78a477dSHong Zhang for (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0; 3527b78a477dSHong Zhang rtmp[i] = 0.0; 3528b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */ 3529b78a477dSHong Zhang bjtmp = bj + bdiag[i+1] + 1; 3530b78a477dSHong Zhang for (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0; 35313c2a7987SHong Zhang 3532b78a477dSHong Zhang /* load in initial unfactored row of A */ 35336da515a1SHong Zhang /* printf("row %d\n",i); */ 35343c2a7987SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 35353c2a7987SHong Zhang ajtmp = aj + ai[r[i]]; 35363c2a7987SHong Zhang v = aa + ai[r[i]]; 35373c2a7987SHong Zhang for (j=0; j<nz; j++) { 3538b78a477dSHong Zhang rtmp[ics[*ajtmp++]] = v[j]; 35396da515a1SHong Zhang /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */ 35403c2a7987SHong Zhang } 35416da515a1SHong Zhang /* printf("\n"); */ 35423c2a7987SHong Zhang 3543b78a477dSHong Zhang /* numerical factorization */ 3544b78a477dSHong Zhang bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */ 3545b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of entries in L(i,:) */ 3546b78a477dSHong Zhang k = 0; 3547b78a477dSHong Zhang while (k < nzl){ 35483c2a7987SHong Zhang row = *bjtmp++; 35496da515a1SHong Zhang /* printf(" prow %d\n",row); */ 35503c2a7987SHong Zhang pc = rtmp + row; 3551b78a477dSHong Zhang pv = b->a + bdiag[row]; /* 1./(diag of the pivot row) */ 3552b78a477dSHong Zhang multiplier = (*pc) * (*pv); 35533c2a7987SHong Zhang *pc = multiplier; 3554b78a477dSHong Zhang if (PetscAbsScalar(multiplier) > dt){ 3555b78a477dSHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 3556b78a477dSHong Zhang pv = b->a + bdiag[row+1] + 1; 3557b78a477dSHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 3558b78a477dSHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 35596da515a1SHong Zhang /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */ 35603c2a7987SHong Zhang } 3561b78a477dSHong Zhang k++; 35623c2a7987SHong Zhang } 35633c2a7987SHong Zhang 3564b78a477dSHong Zhang /* finished row so stick it into b->a */ 3565b78a477dSHong Zhang /* L-part */ 3566b78a477dSHong Zhang pv = b->a + bi[i] ; 3567b78a477dSHong Zhang pj = bj + bi[i] ; 3568b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; 3569b78a477dSHong Zhang for (j=0; j<nzl; j++) { 3570b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 35716da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 35723c2a7987SHong Zhang } 3573b78a477dSHong Zhang 3574b78a477dSHong Zhang /* diagonal: invert diagonal entries for simplier triangular solves */ 3575b78a477dSHong Zhang if (rtmp[i] == 0.0) rtmp[i] = dt+shift; 3576b78a477dSHong Zhang b->a[bdiag[i]] = 1.0/rtmp[i]; 35776da515a1SHong Zhang /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */ 3578b78a477dSHong Zhang 3579b78a477dSHong Zhang /* U-part */ 3580b78a477dSHong Zhang pv = b->a + bdiag[i+1] + 1; 3581b78a477dSHong Zhang pj = bj + bdiag[i+1] + 1; 3582b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1] - 1; 3583b78a477dSHong Zhang for (j=0; j<nzu; j++) { 3584b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 35856da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 3586b78a477dSHong Zhang } 35876da515a1SHong Zhang /* printf("\n"); */ 3588b78a477dSHong Zhang } 3589b78a477dSHong Zhang 35903c2a7987SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 35913c2a7987SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 35923c2a7987SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3593b78a477dSHong Zhang 35943c2a7987SHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 35953c2a7987SHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 35963c2a7987SHong Zhang if (row_identity && col_identity) { 359735233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 35983c2a7987SHong Zhang } else { 359935233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ; 36003c2a7987SHong Zhang } 3601b78a477dSHong Zhang C->ops->solveadd = 0; 3602b78a477dSHong Zhang C->ops->solvetranspose = 0; 3603b78a477dSHong Zhang C->ops->solvetransposeadd = 0; 3604b78a477dSHong Zhang C->ops->matsolve = 0; 36053c2a7987SHong Zhang C->assembled = PETSC_TRUE; 36063c2a7987SHong Zhang C->preallocated = PETSC_TRUE; 36073c2a7987SHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 36083c2a7987SHong Zhang PetscFunctionReturn(0); 36093c2a7987SHong Zhang } 3610