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 74a2ae208SSatish Balay #undef __FUNCT__ 8a7a54a73SBarry Smith #define __FUNCT__ "MatGetOrdering_Flow_SeqAIJ" 98fa24674SBarry Smith /* 108fa24674SBarry Smith Computes an ordering to get most of the large numerical values in the lower triangular part of the matrix 11312e372aSBarry Smith 12312e372aSBarry Smith This code does not work and is not called anywhere. It would be registered with MatOrderingRegisterAll() 138fa24674SBarry Smith */ 1419fd82e9SBarry Smith PetscErrorCode MatGetOrdering_Flow_SeqAIJ(Mat mat,MatOrderingType type,IS *irow,IS *icol) 153b2fbd54SBarry Smith { 168fa24674SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)mat->data; 178fa24674SBarry Smith PetscErrorCode ierr; 18889a8dedSBarry Smith PetscInt i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order; 198fa24674SBarry Smith const PetscInt *ai = a->i, *aj = a->j; 208fa24674SBarry Smith const PetscScalar *aa = a->a; 21ace3abfcSBarry Smith PetscBool *done; 22889a8dedSBarry Smith PetscReal best,past = 0,future; 233a40ed3dSBarry Smith 248fa24674SBarry Smith PetscFunctionBegin; 258fa24674SBarry Smith /* pick initial row */ 268fa24674SBarry Smith best = -1; 278fa24674SBarry Smith for (i=0; i<n; i++) { 2875567043SBarry Smith future = 0.0; 298fa24674SBarry Smith for (j=ai[i]; j<ai[i+1]; j++) { 30db4deed7SKarl Rupp if (aj[j] != i) future += PetscAbsScalar(aa[j]); 31db4deed7SKarl Rupp 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 40*785e854fSJed Brown ierr = PetscMalloc1(n,&done);CHKERRQ(ierr); 41ace3abfcSBarry Smith ierr = PetscMemzero(done,n*sizeof(PetscBool));CHKERRQ(ierr); 42*785e854fSJed Brown ierr = PetscMalloc1(n,&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 } 933b2fbd54SBarry Smith 944a2ae208SSatish Balay #undef __FUNCT__ 95db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc" 96ace3abfcSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscBool *flg) 97db4efbfdSBarry Smith { 98db4efbfdSBarry Smith PetscFunctionBegin; 99db4efbfdSBarry Smith *flg = PETSC_TRUE; 100db4efbfdSBarry Smith PetscFunctionReturn(0); 101db4efbfdSBarry Smith } 102db4efbfdSBarry Smith 103db4efbfdSBarry Smith #undef __FUNCT__ 104b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc" 1058cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B) 106b24902e0SBarry Smith { 107d0f46423SBarry Smith PetscInt n = A->rmap->n; 108b24902e0SBarry Smith PetscErrorCode ierr; 109b24902e0SBarry Smith 110b24902e0SBarry Smith PetscFunctionBegin; 111891bceaeSHong Zhang #if defined(PETSC_USE_COMPLEX) 112891bceaeSHong Zhang if (A->hermitian && (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported"); 113891bceaeSHong Zhang #endif 114ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr); 115b24902e0SBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 116599ef60dSHong Zhang if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT) { 117b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr); 1182205254eSKarl Rupp 11935233d99SShri Abhyankar (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ; 12035233d99SShri Abhyankar (*B)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqAIJ; 1212205254eSKarl Rupp 122e0527adcSJed Brown ierr = MatSetBlockSizes(*B,A->rmap->bs,A->cmap->bs);CHKERRQ(ierr); 123b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 124b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 1250298fd71SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 1262205254eSKarl Rupp 12735233d99SShri Abhyankar (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqAIJ; 12835233d99SShri Abhyankar (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ; 129e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported"); 130d5f3da31SBarry Smith (*B)->factortype = ftype; 131b24902e0SBarry Smith PetscFunctionReturn(0); 132b24902e0SBarry Smith } 133b24902e0SBarry Smith 134b24902e0SBarry Smith #undef __FUNCT__ 135ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_inplace" 136ad04f41aSHong Zhang PetscErrorCode MatLUFactorSymbolic_SeqAIJ_inplace(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 137289bc588SBarry Smith { 138416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 139289bc588SBarry Smith IS isicol; 1406849ba73SBarry Smith PetscErrorCode ierr; 1415d0c19d7SBarry Smith const PetscInt *r,*ic; 1425d0c19d7SBarry Smith PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 1431393bc97SHong Zhang PetscInt *bi,*bj,*ajtmp; 1441393bc97SHong Zhang PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 1459e046878SBarry Smith PetscReal f; 1464875ba38SHong Zhang PetscInt nlnk,*lnk,k,**bi_ptr; 1470298fd71SBarry Smith PetscFreeSpaceList free_space=NULL,current_space=NULL; 1481393bc97SHong Zhang PetscBT lnkbt; 149289bc588SBarry Smith 1503a40ed3dSBarry Smith PetscFunctionBegin; 151e32f2f54SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square"); 1524c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 153ac121b3dSBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 154ac121b3dSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 155289bc588SBarry Smith 156289bc588SBarry Smith /* get new row pointers */ 157*785e854fSJed Brown ierr = PetscMalloc1((n+1),&bi);CHKERRQ(ierr); 1581393bc97SHong Zhang bi[0] = 0; 1591393bc97SHong Zhang 1601393bc97SHong Zhang /* bdiag is location of diagonal in factor */ 161*785e854fSJed Brown ierr = PetscMalloc1((n+1),&bdiag);CHKERRQ(ierr); 1621393bc97SHong Zhang bdiag[0] = 0; 1631393bc97SHong Zhang 1641393bc97SHong Zhang /* linked list for storing column indices of the active row */ 1651393bc97SHong Zhang nlnk = n + 1; 1661393bc97SHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1671393bc97SHong Zhang 168dcca6d9dSJed Brown ierr = PetscMalloc2(n+1,&bi_ptr,n+1,&im);CHKERRQ(ierr); 1691393bc97SHong Zhang 1701393bc97SHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 171d3d32019SBarry Smith f = info->fill; 172a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1731393bc97SHong Zhang current_space = free_space; 174289bc588SBarry Smith 175289bc588SBarry Smith for (i=0; i<n; i++) { 1761393bc97SHong Zhang /* copy previous fill into linked list */ 177289bc588SBarry Smith nzi = 0; 1781393bc97SHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 179e32f2f54SBarry Smith if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i); 1801393bc97SHong Zhang ajtmp = aj + ai[r[i]]; 181afcc9904SHong Zhang ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1821393bc97SHong Zhang nzi += nlnk; 1831393bc97SHong Zhang 1841393bc97SHong Zhang /* add pivot rows into linked list */ 1851393bc97SHong Zhang row = lnk[n]; 1861393bc97SHong Zhang while (row < i) { 187d1170560SHong Zhang nzbd = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */ 188d1170560SHong Zhang ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 189d1170560SHong Zhang ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 1901393bc97SHong Zhang nzi += nlnk; 1911393bc97SHong Zhang row = lnk[row]; 192289bc588SBarry Smith } 1931393bc97SHong Zhang bi[i+1] = bi[i] + nzi; 1941393bc97SHong Zhang im[i] = nzi; 1951393bc97SHong Zhang 1961393bc97SHong Zhang /* mark bdiag */ 1971393bc97SHong Zhang nzbd = 0; 1981393bc97SHong Zhang nnz = nzi; 1991393bc97SHong Zhang k = lnk[n]; 2001393bc97SHong Zhang while (nnz-- && k < i) { 2011393bc97SHong Zhang nzbd++; 2021393bc97SHong Zhang k = lnk[k]; 2031393bc97SHong Zhang } 2041393bc97SHong Zhang bdiag[i] = bi[i] + nzbd; 2051393bc97SHong Zhang 2061393bc97SHong Zhang /* if free space is not available, make more free space */ 2071393bc97SHong Zhang if (current_space->local_remaining<nzi) { 2081393bc97SHong Zhang nnz = (n - i)*nzi; /* estimated and max additional space needed */ 209a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2101393bc97SHong Zhang reallocs++; 2111393bc97SHong Zhang } 2121393bc97SHong Zhang 2131393bc97SHong Zhang /* copy data into free space, then initialize lnk */ 2141393bc97SHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 2152205254eSKarl Rupp 2161393bc97SHong Zhang bi_ptr[i] = current_space->array; 2171393bc97SHong Zhang current_space->array += nzi; 2181393bc97SHong Zhang current_space->local_used += nzi; 2191393bc97SHong Zhang current_space->local_remaining -= nzi; 220289bc588SBarry Smith } 2216cf91177SBarry Smith #if defined(PETSC_USE_INFO) 222464e8d44SSatish Balay if (ai[n] != 0) { 2231393bc97SHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 224ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 225ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 226ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 227ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 22805bf559cSSatish Balay } else { 229ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 23005bf559cSSatish Balay } 23163ba0a88SBarry Smith #endif 2322fb3ed76SBarry Smith 233898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 234898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2351987afe7SBarry Smith 2361393bc97SHong Zhang /* destroy list of free space and other temporary array(s) */ 237*785e854fSJed Brown ierr = PetscMalloc1((bi[n]+1),&bj);CHKERRQ(ierr); 238a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); 2391393bc97SHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 24035baf27eSSatish Balay ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 241289bc588SBarry Smith 242289bc588SBarry Smith /* put together the new matrix */ 2430298fd71SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 2443bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr); 245719d5645SBarry Smith b = (Mat_SeqAIJ*)(B)->data; 2462205254eSKarl Rupp 247e6b907acSBarry Smith b->free_a = PETSC_TRUE; 248e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 2497c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 2502205254eSKarl Rupp 251*785e854fSJed Brown ierr = PetscMalloc1((bi[n]+1),&b->a);CHKERRQ(ierr); 2521393bc97SHong Zhang b->j = bj; 2531393bc97SHong Zhang b->i = bi; 2541393bc97SHong Zhang b->diag = bdiag; 255416022c9SBarry Smith b->ilen = 0; 256416022c9SBarry Smith b->imax = 0; 257416022c9SBarry Smith b->row = isrow; 258416022c9SBarry Smith b->col = iscol; 259c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 260c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 26182bf6240SBarry Smith b->icol = isicol; 262*785e854fSJed Brown ierr = PetscMalloc1((n+1),&b->solve_work);CHKERRQ(ierr); 2631393bc97SHong Zhang 2641393bc97SHong Zhang /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 2653bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 2661393bc97SHong Zhang b->maxnz = b->nz = bi[n]; 2677fd98bd6SLois Curfman McInnes 268d5f3da31SBarry Smith (B)->factortype = MAT_FACTOR_LU; 269719d5645SBarry Smith (B)->info.factor_mallocs = reallocs; 270719d5645SBarry Smith (B)->info.fill_ratio_given = f; 271703deb49SSatish Balay 2729f7d0b68SHong Zhang if (ai[n]) { 273719d5645SBarry Smith (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 274eea2913aSSatish Balay } else { 275719d5645SBarry Smith (B)->info.fill_ratio_needed = 0.0; 276eea2913aSSatish Balay } 277ad04f41aSHong Zhang (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 278f2344125SBarry Smith if (a->inode.size) { 279f2344125SBarry Smith (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 280f2344125SBarry Smith } 2813a40ed3dSBarry Smith PetscFunctionReturn(0); 282289bc588SBarry Smith } 2831393bc97SHong Zhang 284ce3d78c0SShri Abhyankar #undef __FUNCT__ 28535233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ" 28635233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 287ce3d78c0SShri Abhyankar { 288ce3d78c0SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 289ce3d78c0SShri Abhyankar IS isicol; 290ce3d78c0SShri Abhyankar PetscErrorCode ierr; 2918758e1faSBarry Smith const PetscInt *r,*ic,*ai=a->i,*aj=a->j,*ajtmp; 2928758e1faSBarry Smith PetscInt i,n=A->rmap->n; 2938758e1faSBarry Smith PetscInt *bi,*bj; 294ce3d78c0SShri Abhyankar PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 295ce3d78c0SShri Abhyankar PetscReal f; 296ce3d78c0SShri Abhyankar PetscInt nlnk,*lnk,k,**bi_ptr; 2970298fd71SBarry Smith PetscFreeSpaceList free_space=NULL,current_space=NULL; 298ce3d78c0SShri Abhyankar PetscBT lnkbt; 299ce3d78c0SShri Abhyankar 300ce3d78c0SShri Abhyankar PetscFunctionBegin; 3013898ab1dSSatish Balay /* Uncomment the oldatastruct part only while testing new data structure for MatSolve() */ 302a4be958cSHong Zhang /* 3031a26ff17SHong Zhang PetscBool olddatastruct=PETSC_FALSE; 3040298fd71SBarry Smith ierr = PetscOptionsGetBool(NULL,"-lu_old",&olddatastruct,NULL);CHKERRQ(ierr); 305628f99d7SShri Abhyankar if (olddatastruct) { 306628f99d7SShri Abhyankar ierr = MatLUFactorSymbolic_SeqAIJ_inplace(B,A,isrow,iscol,info);CHKERRQ(ierr); 307628f99d7SShri Abhyankar PetscFunctionReturn(0); 308628f99d7SShri Abhyankar } 309a4be958cSHong Zhang */ 310e32f2f54SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square"); 311ce3d78c0SShri Abhyankar ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 312ce3d78c0SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 313ce3d78c0SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 314ce3d78c0SShri Abhyankar 3151bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 316*785e854fSJed Brown ierr = PetscMalloc1((n+1),&bi);CHKERRQ(ierr); 317*785e854fSJed Brown ierr = PetscMalloc1((n+1),&bdiag);CHKERRQ(ierr); 318a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 3199b48462bSShri Abhyankar 3209b48462bSShri Abhyankar /* linked list for storing column indices of the active row */ 3219b48462bSShri Abhyankar nlnk = n + 1; 3229b48462bSShri Abhyankar ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3239b48462bSShri Abhyankar 324dcca6d9dSJed Brown ierr = PetscMalloc2(n+1,&bi_ptr,n+1,&im);CHKERRQ(ierr); 3259b48462bSShri Abhyankar 3269b48462bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 3279b48462bSShri Abhyankar f = info->fill; 3289b48462bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 3299b48462bSShri Abhyankar current_space = free_space; 3309b48462bSShri Abhyankar 3319b48462bSShri Abhyankar for (i=0; i<n; i++) { 3329b48462bSShri Abhyankar /* copy previous fill into linked list */ 3339b48462bSShri Abhyankar nzi = 0; 3349b48462bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 335e32f2f54SBarry 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); 3369b48462bSShri Abhyankar ajtmp = aj + ai[r[i]]; 3379b48462bSShri Abhyankar ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3389b48462bSShri Abhyankar nzi += nlnk; 3399b48462bSShri Abhyankar 3409b48462bSShri Abhyankar /* add pivot rows into linked list */ 3419b48462bSShri Abhyankar row = lnk[n]; 3429b48462bSShri Abhyankar while (row < i) { 3439b48462bSShri Abhyankar nzbd = bdiag[row] + 1; /* num of entries in the row with column index <= row */ 3449b48462bSShri Abhyankar ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 3459b48462bSShri Abhyankar ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 3469b48462bSShri Abhyankar nzi += nlnk; 3479b48462bSShri Abhyankar row = lnk[row]; 3489b48462bSShri Abhyankar } 3499b48462bSShri Abhyankar bi[i+1] = bi[i] + nzi; 3509b48462bSShri Abhyankar im[i] = nzi; 3519b48462bSShri Abhyankar 3529b48462bSShri Abhyankar /* mark bdiag */ 3539b48462bSShri Abhyankar nzbd = 0; 3549b48462bSShri Abhyankar nnz = nzi; 3559b48462bSShri Abhyankar k = lnk[n]; 3569b48462bSShri Abhyankar while (nnz-- && k < i) { 3579b48462bSShri Abhyankar nzbd++; 3589b48462bSShri Abhyankar k = lnk[k]; 3599b48462bSShri Abhyankar } 3609b48462bSShri Abhyankar bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */ 3619b48462bSShri Abhyankar 3629b48462bSShri Abhyankar /* if free space is not available, make more free space */ 3639b48462bSShri Abhyankar if (current_space->local_remaining<nzi) { 3649b48462bSShri Abhyankar nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */ 3659b48462bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 3669b48462bSShri Abhyankar reallocs++; 3679b48462bSShri Abhyankar } 3689b48462bSShri Abhyankar 3699b48462bSShri Abhyankar /* copy data into free space, then initialize lnk */ 3709b48462bSShri Abhyankar ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 3712205254eSKarl Rupp 3729b48462bSShri Abhyankar bi_ptr[i] = current_space->array; 3739b48462bSShri Abhyankar current_space->array += nzi; 3749b48462bSShri Abhyankar current_space->local_used += nzi; 3759b48462bSShri Abhyankar current_space->local_remaining -= nzi; 3769b48462bSShri Abhyankar } 3779b48462bSShri Abhyankar 3789b48462bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3799b48462bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3809b48462bSShri Abhyankar 3819263d837SHong Zhang /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 382*785e854fSJed Brown ierr = PetscMalloc1((bi[n]+1),&bj);CHKERRQ(ierr); 3832ce24eb6SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 3849b48462bSShri Abhyankar ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 3859b48462bSShri Abhyankar ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 3869b48462bSShri Abhyankar 3879b48462bSShri Abhyankar /* put together the new matrix */ 3880298fd71SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 3893bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr); 3909b48462bSShri Abhyankar b = (Mat_SeqAIJ*)(B)->data; 3912205254eSKarl Rupp 3929b48462bSShri Abhyankar b->free_a = PETSC_TRUE; 3939b48462bSShri Abhyankar b->free_ij = PETSC_TRUE; 3949b48462bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 3952205254eSKarl Rupp 396*785e854fSJed Brown ierr = PetscMalloc1((bdiag[0]+1),&b->a);CHKERRQ(ierr); 3972205254eSKarl Rupp 3989b48462bSShri Abhyankar b->j = bj; 3999b48462bSShri Abhyankar b->i = bi; 4009b48462bSShri Abhyankar b->diag = bdiag; 4019b48462bSShri Abhyankar b->ilen = 0; 4029b48462bSShri Abhyankar b->imax = 0; 4039b48462bSShri Abhyankar b->row = isrow; 4049b48462bSShri Abhyankar b->col = iscol; 4059b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 4069b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 4079b48462bSShri Abhyankar b->icol = isicol; 408*785e854fSJed Brown ierr = PetscMalloc1((n+1),&b->solve_work);CHKERRQ(ierr); 4099b48462bSShri Abhyankar 4109b48462bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 4113bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 4129b48462bSShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 4132205254eSKarl Rupp 414d5f3da31SBarry Smith B->factortype = MAT_FACTOR_LU; 415f284f12aSHong Zhang B->info.factor_mallocs = reallocs; 416f284f12aSHong Zhang B->info.fill_ratio_given = f; 4179b48462bSShri Abhyankar 4189b48462bSShri Abhyankar if (ai[n]) { 419f284f12aSHong Zhang B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 4209b48462bSShri Abhyankar } else { 421f284f12aSHong Zhang B->info.fill_ratio_needed = 0.0; 4229b48462bSShri Abhyankar } 4239263d837SHong Zhang #if defined(PETSC_USE_INFO) 4249263d837SHong Zhang if (ai[n] != 0) { 4259263d837SHong Zhang PetscReal af = B->info.fill_ratio_needed; 4269263d837SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 4279263d837SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 4289263d837SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr); 4299263d837SHong Zhang ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 4309263d837SHong Zhang } else { 4319263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 4329263d837SHong Zhang } 4339263d837SHong Zhang #endif 43435233d99SShri Abhyankar B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 435f2344125SBarry Smith if (a->inode.size) { 436d3ac4fa3SBarry Smith B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 437d3ac4fa3SBarry Smith } 438abb87a52SBarry Smith ierr = Mat_CheckInode_FactorLU(B);CHKERRQ(ierr); 4399b48462bSShri Abhyankar PetscFunctionReturn(0); 4409b48462bSShri Abhyankar } 4419b48462bSShri Abhyankar 4425b5bc046SBarry Smith /* 4435b5bc046SBarry Smith Trouble in factorization, should we dump the original matrix? 4445b5bc046SBarry Smith */ 4455b5bc046SBarry Smith #undef __FUNCT__ 4465b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix" 4475b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A) 4485b5bc046SBarry Smith { 4495b5bc046SBarry Smith PetscErrorCode ierr; 450ace3abfcSBarry Smith PetscBool flg = PETSC_FALSE; 4515b5bc046SBarry Smith 4525b5bc046SBarry Smith PetscFunctionBegin; 4530298fd71SBarry Smith ierr = PetscOptionsGetBool(NULL,"-mat_factor_dump_on_error",&flg,NULL);CHKERRQ(ierr); 4545b5bc046SBarry Smith if (flg) { 4555b5bc046SBarry Smith PetscViewer viewer; 4565b5bc046SBarry Smith char filename[PETSC_MAX_PATH_LEN]; 4575b5bc046SBarry Smith 4585b5bc046SBarry Smith ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr); 459ce94432eSBarry Smith ierr = PetscViewerBinaryOpen(PetscObjectComm((PetscObject)A),filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 4605b5bc046SBarry Smith ierr = MatView(A,viewer);CHKERRQ(ierr); 4616bf464f9SBarry Smith ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 4625b5bc046SBarry Smith } 4635b5bc046SBarry Smith PetscFunctionReturn(0); 4645b5bc046SBarry Smith } 4655b5bc046SBarry Smith 466c9c72f8fSHong Zhang #undef __FUNCT__ 46735233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ" 46835233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 469c9c72f8fSHong Zhang { 470c9c72f8fSHong Zhang Mat C =B; 471c9c72f8fSHong Zhang Mat_SeqAIJ *a =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data; 472c9c72f8fSHong Zhang IS isrow = b->row,isicol = b->icol; 473c9c72f8fSHong Zhang PetscErrorCode ierr; 474c9c72f8fSHong Zhang const PetscInt *r,*ic,*ics; 475bbd65245SShri Abhyankar const PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag; 476bbd65245SShri Abhyankar PetscInt i,j,k,nz,nzL,row,*pj; 477bbd65245SShri Abhyankar const PetscInt *ajtmp,*bjtmp; 478bbd65245SShri Abhyankar MatScalar *rtmp,*pc,multiplier,*pv; 479bbd65245SShri Abhyankar const MatScalar *aa=a->a,*v; 480ace3abfcSBarry Smith PetscBool row_identity,col_identity; 481d90ac83dSHong Zhang FactorShiftCtx sctx; 4824f81c4b7SBarry Smith const PetscInt *ddiag; 483d4ad06f7SHong Zhang PetscReal rs; 484d4ad06f7SHong Zhang MatScalar d; 485d4ad06f7SHong Zhang 486c9c72f8fSHong Zhang PetscFunctionBegin; 487d6e5152cSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 488d90ac83dSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 489d4ad06f7SHong Zhang 490f4db908eSBarry Smith if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 491d4ad06f7SHong Zhang ddiag = a->diag; 492d4ad06f7SHong Zhang sctx.shift_top = info->zeropivot; 493d4ad06f7SHong Zhang for (i=0; i<n; i++) { 494d4ad06f7SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 495d4ad06f7SHong Zhang d = (aa)[ddiag[i]]; 496d4ad06f7SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 497d4ad06f7SHong Zhang v = aa+ai[i]; 498d4ad06f7SHong Zhang nz = ai[i+1] - ai[i]; 4992205254eSKarl Rupp for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]); 500d4ad06f7SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 501d4ad06f7SHong Zhang } 502d4ad06f7SHong Zhang sctx.shift_top *= 1.1; 503d4ad06f7SHong Zhang sctx.nshift_max = 5; 504d4ad06f7SHong Zhang sctx.shift_lo = 0.; 505d4ad06f7SHong Zhang sctx.shift_hi = 1.; 506d4ad06f7SHong Zhang } 507d4ad06f7SHong Zhang 508c9c72f8fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 509c9c72f8fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 510*785e854fSJed Brown ierr = PetscMalloc1((n+1),&rtmp);CHKERRQ(ierr); 511c9c72f8fSHong Zhang ics = ic; 512c9c72f8fSHong Zhang 513d4ad06f7SHong Zhang do { 51407b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 515c9c72f8fSHong Zhang for (i=0; i<n; i++) { 516c9c72f8fSHong Zhang /* zero rtmp */ 517c9c72f8fSHong Zhang /* L part */ 518c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 519c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 520c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 521c9c72f8fSHong Zhang 522c9c72f8fSHong Zhang /* U part */ 52362fbd6afSShri Abhyankar nz = bdiag[i]-bdiag[i+1]; 52462fbd6afSShri Abhyankar bjtmp = bj + bdiag[i+1]+1; 525813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 526813a1f2bSShri Abhyankar 527813a1f2bSShri Abhyankar /* load in initial (unfactored row) */ 528813a1f2bSShri Abhyankar nz = ai[r[i]+1] - ai[r[i]]; 529813a1f2bSShri Abhyankar ajtmp = aj + ai[r[i]]; 530813a1f2bSShri Abhyankar v = aa + ai[r[i]]; 531813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 532813a1f2bSShri Abhyankar rtmp[ics[ajtmp[j]]] = v[j]; 533813a1f2bSShri Abhyankar } 534813a1f2bSShri Abhyankar /* ZeropivotApply() */ 53598a93161SHong Zhang rtmp[i] += sctx.shift_amount; /* shift the diagonal of the matrix */ 536813a1f2bSShri Abhyankar 537813a1f2bSShri Abhyankar /* elimination */ 538813a1f2bSShri Abhyankar bjtmp = bj + bi[i]; 539813a1f2bSShri Abhyankar row = *bjtmp++; 540813a1f2bSShri Abhyankar nzL = bi[i+1] - bi[i]; 541f268cba8SShri Abhyankar for (k=0; k < nzL; k++) { 542813a1f2bSShri Abhyankar pc = rtmp + row; 543813a1f2bSShri Abhyankar if (*pc != 0.0) { 544813a1f2bSShri Abhyankar pv = b->a + bdiag[row]; 545813a1f2bSShri Abhyankar multiplier = *pc * (*pv); 546813a1f2bSShri Abhyankar *pc = multiplier; 5472205254eSKarl Rupp 54862fbd6afSShri Abhyankar pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 54962fbd6afSShri Abhyankar pv = b->a + bdiag[row+1]+1; 55062fbd6afSShri Abhyankar nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 5512205254eSKarl Rupp 552813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 55328b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 554813a1f2bSShri Abhyankar } 555f268cba8SShri Abhyankar row = *bjtmp++; 556813a1f2bSShri Abhyankar } 557813a1f2bSShri Abhyankar 558813a1f2bSShri Abhyankar /* finished row so stick it into b->a */ 559813a1f2bSShri Abhyankar rs = 0.0; 560813a1f2bSShri Abhyankar /* L part */ 561813a1f2bSShri Abhyankar pv = b->a + bi[i]; 562813a1f2bSShri Abhyankar pj = b->j + bi[i]; 563813a1f2bSShri Abhyankar nz = bi[i+1] - bi[i]; 564813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 565813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 566813a1f2bSShri Abhyankar } 567813a1f2bSShri Abhyankar 568813a1f2bSShri Abhyankar /* U part */ 56962fbd6afSShri Abhyankar pv = b->a + bdiag[i+1]+1; 57062fbd6afSShri Abhyankar pj = b->j + bdiag[i+1]+1; 57162fbd6afSShri Abhyankar nz = bdiag[i] - bdiag[i+1]-1; 572813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 573813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 574813a1f2bSShri Abhyankar } 575813a1f2bSShri Abhyankar 576813a1f2bSShri Abhyankar sctx.rs = rs; 577813a1f2bSShri Abhyankar sctx.pv = rtmp[i]; 578d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 57907b50cabSHong Zhang if (sctx.newshift) break; /* break for-loop */ 58007b50cabSHong Zhang rtmp[i] = sctx.pv; /* sctx.pv might be updated in the case of MAT_SHIFT_INBLOCKS */ 581813a1f2bSShri Abhyankar 582813a1f2bSShri Abhyankar /* Mark diagonal and invert diagonal for simplier triangular solves */ 583813a1f2bSShri Abhyankar pv = b->a + bdiag[i]; 584813a1f2bSShri Abhyankar *pv = 1.0/rtmp[i]; 585813a1f2bSShri Abhyankar 586813a1f2bSShri Abhyankar } /* endof for (i=0; i<n; i++) { */ 587813a1f2bSShri Abhyankar 5888ff23777SHong Zhang /* MatPivotRefine() */ 58907b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 590813a1f2bSShri Abhyankar /* 591813a1f2bSShri Abhyankar * if no shift in this attempt & shifting & started shifting & can refine, 592813a1f2bSShri Abhyankar * then try lower shift 593813a1f2bSShri Abhyankar */ 594813a1f2bSShri Abhyankar sctx.shift_hi = sctx.shift_fraction; 595813a1f2bSShri Abhyankar sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 596813a1f2bSShri Abhyankar sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 59707b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 598813a1f2bSShri Abhyankar sctx.nshift++; 599813a1f2bSShri Abhyankar } 60007b50cabSHong Zhang } while (sctx.newshift); 601813a1f2bSShri Abhyankar 602813a1f2bSShri Abhyankar ierr = PetscFree(rtmp);CHKERRQ(ierr); 603813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 604813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 605d3ac4fa3SBarry Smith 606813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 607813a1f2bSShri Abhyankar ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 608abb87a52SBarry Smith if (b->inode.size) { 609abb87a52SBarry Smith C->ops->solve = MatSolve_SeqAIJ_Inode; 610abb87a52SBarry Smith } else if (row_identity && col_identity) { 61135233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 612813a1f2bSShri Abhyankar } else { 61335233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ; 614813a1f2bSShri Abhyankar } 61535233d99SShri Abhyankar C->ops->solveadd = MatSolveAdd_SeqAIJ; 61635233d99SShri Abhyankar C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 61735233d99SShri Abhyankar C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 61835233d99SShri Abhyankar C->ops->matsolve = MatMatSolve_SeqAIJ; 619813a1f2bSShri Abhyankar C->assembled = PETSC_TRUE; 620813a1f2bSShri Abhyankar C->preallocated = PETSC_TRUE; 6212205254eSKarl Rupp 622813a1f2bSShri Abhyankar ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 623813a1f2bSShri Abhyankar 62414d2772eSHong Zhang /* MatShiftView(A,info,&sctx) */ 625813a1f2bSShri Abhyankar if (sctx.nshift) { 626f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 627813a1f2bSShri 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); 628f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 629813a1f2bSShri Abhyankar ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 630f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) { 631915743fcSHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr); 632813a1f2bSShri Abhyankar } 633813a1f2bSShri Abhyankar } 634813a1f2bSShri Abhyankar PetscFunctionReturn(0); 635813a1f2bSShri Abhyankar } 636813a1f2bSShri Abhyankar 6374a2ae208SSatish Balay #undef __FUNCT__ 638ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_inplace" 639ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info) 640289bc588SBarry Smith { 641719d5645SBarry Smith Mat C =B; 642416022c9SBarry Smith Mat_SeqAIJ *a =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data; 64382bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 6446849ba73SBarry Smith PetscErrorCode ierr; 6455d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 646d278edc7SBarry Smith PetscInt nz,row,i,j,n=A->rmap->n,diag; 647d278edc7SBarry Smith const PetscInt *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 648d278edc7SBarry Smith const PetscInt *ajtmp,*bjtmp,*diag_offset = b->diag,*pj; 649d278edc7SBarry Smith MatScalar *pv,*rtmp,*pc,multiplier,d; 650d278edc7SBarry Smith const MatScalar *v,*aa=a->a; 65133f9893dSHong Zhang PetscReal rs=0.0; 652d90ac83dSHong Zhang FactorShiftCtx sctx; 6534f81c4b7SBarry Smith const PetscInt *ddiag; 654ace3abfcSBarry Smith PetscBool row_identity, col_identity; 655289bc588SBarry Smith 6563a40ed3dSBarry Smith PetscFunctionBegin; 657504a3fadSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 658504a3fadSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 659ada7143aSSatish Balay 660f4db908eSBarry Smith if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 66142898933SHong Zhang ddiag = a->diag; 6629f7d0b68SHong Zhang sctx.shift_top = info->zeropivot; 6636cc28720Svictorle for (i=0; i<n; i++) { 6649f95998fSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 6659f7d0b68SHong Zhang d = (aa)[ddiag[i]]; 6661a890ab1SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 6679f7d0b68SHong Zhang v = aa+ai[i]; 6689f7d0b68SHong Zhang nz = ai[i+1] - ai[i]; 6692205254eSKarl Rupp for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]); 6701a890ab1SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 6716cc28720Svictorle } 672118f5789SBarry Smith sctx.shift_top *= 1.1; 673d2276718SHong Zhang sctx.nshift_max = 5; 674d2276718SHong Zhang sctx.shift_lo = 0.; 675d2276718SHong Zhang sctx.shift_hi = 1.; 676a0a356efSHong Zhang } 677a0a356efSHong Zhang 678504a3fadSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 679504a3fadSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 680*785e854fSJed Brown ierr = PetscMalloc1((n+1),&rtmp);CHKERRQ(ierr); 681504a3fadSHong Zhang ics = ic; 682504a3fadSHong Zhang 683085256b3SBarry Smith do { 68407b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 685289bc588SBarry Smith for (i=0; i<n; i++) { 686b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 687b3bf805bSHong Zhang bjtmp = bj + bi[i]; 6889f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 689289bc588SBarry Smith 690289bc588SBarry Smith /* load in initial (unfactored row) */ 6919f7d0b68SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 6929f7d0b68SHong Zhang ajtmp = aj + ai[r[i]]; 6939f7d0b68SHong Zhang v = aa + ai[r[i]]; 694085256b3SBarry Smith for (j=0; j<nz; j++) { 695b3bf805bSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 696085256b3SBarry Smith } 697d2276718SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 698289bc588SBarry Smith 699b3bf805bSHong Zhang row = *bjtmp++; 700f2582111SSatish Balay while (row < i) { 7018c37ef55SBarry Smith pc = rtmp + row; 7028c37ef55SBarry Smith if (*pc != 0.0) { 703010a20caSHong Zhang pv = b->a + diag_offset[row]; 704010a20caSHong Zhang pj = b->j + diag_offset[row] + 1; 70535aab85fSBarry Smith multiplier = *pc / *pv++; 7068c37ef55SBarry Smith *pc = multiplier; 707b3bf805bSHong Zhang nz = bi[row+1] - diag_offset[row] - 1; 7089f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 70928b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 710289bc588SBarry Smith } 711b3bf805bSHong Zhang row = *bjtmp++; 712289bc588SBarry Smith } 713416022c9SBarry Smith /* finished row so stick it into b->a */ 714b3bf805bSHong Zhang pv = b->a + bi[i]; 715b3bf805bSHong Zhang pj = b->j + bi[i]; 716b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 717b3bf805bSHong Zhang diag = diag_offset[i] - bi[i]; 718e57ff13aSBarry Smith rs = 0.0; 719d3d32019SBarry Smith for (j=0; j<nz; j++) { 7209f7d0b68SHong Zhang pv[j] = rtmp[pj[j]]; 7219f7d0b68SHong Zhang rs += PetscAbsScalar(pv[j]); 722d3d32019SBarry Smith } 723e57ff13aSBarry Smith rs -= PetscAbsScalar(pv[diag]); 724d2276718SHong Zhang 725d2276718SHong Zhang sctx.rs = rs; 726d2276718SHong Zhang sctx.pv = pv[diag]; 727d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 72807b50cabSHong Zhang if (sctx.newshift) break; 729504a3fadSHong Zhang pv[diag] = sctx.pv; 73035aab85fSBarry Smith } 731d2276718SHong Zhang 73207b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 7336cc28720Svictorle /* 7346c037d1bSvictorle * if no shift in this attempt & shifting & started shifting & can refine, 7356cc28720Svictorle * then try lower shift 7366cc28720Svictorle */ 7370481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 7380481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 7390481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 74007b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 741d2276718SHong Zhang sctx.nshift++; 7426cc28720Svictorle } 74307b50cabSHong Zhang } while (sctx.newshift); 7440f11f9aeSSatish Balay 74535aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 74635aab85fSBarry Smith for (i=0; i<n; i++) { 747010a20caSHong Zhang b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]]; 74835aab85fSBarry Smith } 749606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 75078b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 75178b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 752d3ac4fa3SBarry Smith 7538d8c7c43SBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 7548d8c7c43SBarry Smith ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 7558d8c7c43SBarry Smith if (row_identity && col_identity) { 756ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_inplace; 7578d8c7c43SBarry Smith } else { 758ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_inplace; 759db4efbfdSBarry Smith } 760ad04f41aSHong Zhang C->ops->solveadd = MatSolveAdd_SeqAIJ_inplace; 761ad04f41aSHong Zhang C->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 762ad04f41aSHong Zhang C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 763ad04f41aSHong Zhang C->ops->matsolve = MatMatSolve_SeqAIJ_inplace; 7642205254eSKarl Rupp 765c456f294SBarry Smith C->assembled = PETSC_TRUE; 7665c9eb25fSBarry Smith C->preallocated = PETSC_TRUE; 7672205254eSKarl Rupp 768d0f46423SBarry Smith ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 769d2276718SHong Zhang if (sctx.nshift) { 770f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 7710481f469SBarry 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); 772f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 773e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 7746cc28720Svictorle } 7750697b6caSHong Zhang } 776d3ac4fa3SBarry Smith (C)->ops->solve = MatSolve_SeqAIJ_inplace; 777d3ac4fa3SBarry Smith (C)->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 7782205254eSKarl Rupp 779d3ac4fa3SBarry Smith ierr = Mat_CheckInode(C,PETSC_FALSE);CHKERRQ(ierr); 7803a40ed3dSBarry Smith PetscFunctionReturn(0); 781289bc588SBarry Smith } 7820889b5dcSvictorle 783137fb511SHong Zhang /* 784137fb511SHong Zhang This routine implements inplace ILU(0) with row or/and column permutations. 785137fb511SHong Zhang Input: 786137fb511SHong Zhang A - original matrix 787137fb511SHong Zhang Output; 788137fb511SHong Zhang A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i] 789137fb511SHong Zhang a->j (col index) is permuted by the inverse of colperm, then sorted 790137fb511SHong Zhang a->a reordered accordingly with a->j 791137fb511SHong Zhang a->diag (ptr to diagonal elements) is updated. 792137fb511SHong Zhang */ 793137fb511SHong Zhang #undef __FUNCT__ 794137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm" 7950481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info) 796137fb511SHong Zhang { 797b051339dSHong Zhang Mat_SeqAIJ *a =(Mat_SeqAIJ*)A->data; 798b051339dSHong Zhang IS isrow = a->row,isicol = a->icol; 799137fb511SHong Zhang PetscErrorCode ierr; 8005d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 8015d0c19d7SBarry Smith PetscInt i,j,n=A->rmap->n,*ai=a->i,*aj=a->j; 8025d0c19d7SBarry Smith PetscInt *ajtmp,nz,row; 803b051339dSHong Zhang PetscInt *diag = a->diag,nbdiag,*pj; 804a77337e4SBarry Smith PetscScalar *rtmp,*pc,multiplier,d; 805504a3fadSHong Zhang MatScalar *pv,*v; 806137fb511SHong Zhang PetscReal rs; 807d90ac83dSHong Zhang FactorShiftCtx sctx; 808504a3fadSHong Zhang const MatScalar *aa=a->a,*vtmp; 809137fb511SHong Zhang 810137fb511SHong Zhang PetscFunctionBegin; 811e32f2f54SBarry Smith if (A != B) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address"); 812504a3fadSHong Zhang 813504a3fadSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 814504a3fadSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 815504a3fadSHong Zhang 816504a3fadSHong Zhang if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 817504a3fadSHong Zhang const PetscInt *ddiag = a->diag; 818504a3fadSHong Zhang sctx.shift_top = info->zeropivot; 819504a3fadSHong Zhang for (i=0; i<n; i++) { 820504a3fadSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 821504a3fadSHong Zhang d = (aa)[ddiag[i]]; 822504a3fadSHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 823504a3fadSHong Zhang vtmp = aa+ai[i]; 824504a3fadSHong Zhang nz = ai[i+1] - ai[i]; 8252205254eSKarl Rupp for (j=0; j<nz; j++) rs += PetscAbsScalar(vtmp[j]); 826504a3fadSHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 827504a3fadSHong Zhang } 828504a3fadSHong Zhang sctx.shift_top *= 1.1; 829504a3fadSHong Zhang sctx.nshift_max = 5; 830504a3fadSHong Zhang sctx.shift_lo = 0.; 831504a3fadSHong Zhang sctx.shift_hi = 1.; 832504a3fadSHong Zhang } 833504a3fadSHong Zhang 834137fb511SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 835137fb511SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 836*785e854fSJed Brown ierr = PetscMalloc1((n+1),&rtmp);CHKERRQ(ierr); 837137fb511SHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 838b051339dSHong Zhang ics = ic; 839137fb511SHong Zhang 840504a3fadSHong Zhang #if defined(MV) 84175567043SBarry Smith sctx.shift_top = 0.; 842e60cf9a0SBarry Smith sctx.nshift_max = 0; 84375567043SBarry Smith sctx.shift_lo = 0.; 84475567043SBarry Smith sctx.shift_hi = 0.; 84575567043SBarry Smith sctx.shift_fraction = 0.; 846137fb511SHong Zhang 847f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 84875567043SBarry Smith sctx.shift_top = 0.; 849137fb511SHong Zhang for (i=0; i<n; i++) { 850137fb511SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 851b051339dSHong Zhang d = (a->a)[diag[i]]; 852137fb511SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 853b051339dSHong Zhang v = a->a+ai[i]; 854b051339dSHong Zhang nz = ai[i+1] - ai[i]; 8552205254eSKarl Rupp for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]); 856137fb511SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 857137fb511SHong Zhang } 858137fb511SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 859137fb511SHong Zhang sctx.shift_top *= 1.1; 860137fb511SHong Zhang sctx.nshift_max = 5; 861137fb511SHong Zhang sctx.shift_lo = 0.; 862137fb511SHong Zhang sctx.shift_hi = 1.; 863137fb511SHong Zhang } 864137fb511SHong Zhang 86575567043SBarry Smith sctx.shift_amount = 0.; 866137fb511SHong Zhang sctx.nshift = 0; 867504a3fadSHong Zhang #endif 868504a3fadSHong Zhang 869137fb511SHong Zhang do { 87007b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 871137fb511SHong Zhang for (i=0; i<n; i++) { 872b051339dSHong Zhang /* load in initial unfactored row */ 873b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 874b051339dSHong Zhang ajtmp = aj + ai[r[i]]; 875b051339dSHong Zhang v = a->a + ai[r[i]]; 876137fb511SHong Zhang /* sort permuted ajtmp and values v accordingly */ 877b051339dSHong Zhang for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]]; 878137fb511SHong Zhang ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr); 879137fb511SHong Zhang 880b051339dSHong Zhang diag[r[i]] = ai[r[i]]; 881137fb511SHong Zhang for (j=0; j<nz; j++) { 882137fb511SHong Zhang rtmp[ajtmp[j]] = v[j]; 883b051339dSHong Zhang if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */ 884137fb511SHong Zhang } 885137fb511SHong Zhang rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 886137fb511SHong Zhang 887b051339dSHong Zhang row = *ajtmp++; 888137fb511SHong Zhang while (row < i) { 889137fb511SHong Zhang pc = rtmp + row; 890137fb511SHong Zhang if (*pc != 0.0) { 891b051339dSHong Zhang pv = a->a + diag[r[row]]; 892b051339dSHong Zhang pj = aj + diag[r[row]] + 1; 893137fb511SHong Zhang 894137fb511SHong Zhang multiplier = *pc / *pv++; 895137fb511SHong Zhang *pc = multiplier; 896b051339dSHong Zhang nz = ai[r[row]+1] - diag[r[row]] - 1; 897b051339dSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 89828b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 899137fb511SHong Zhang } 900b051339dSHong Zhang row = *ajtmp++; 901137fb511SHong Zhang } 902b051339dSHong Zhang /* finished row so overwrite it onto a->a */ 903b051339dSHong Zhang pv = a->a + ai[r[i]]; 904b051339dSHong Zhang pj = aj + ai[r[i]]; 905b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 906b051339dSHong Zhang nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */ 907137fb511SHong Zhang 908137fb511SHong Zhang rs = 0.0; 909137fb511SHong Zhang for (j=0; j<nz; j++) { 910b051339dSHong Zhang pv[j] = rtmp[pj[j]]; 911b051339dSHong Zhang if (j != nbdiag) rs += PetscAbsScalar(pv[j]); 912137fb511SHong Zhang } 913137fb511SHong Zhang 914137fb511SHong Zhang sctx.rs = rs; 915b051339dSHong Zhang sctx.pv = pv[nbdiag]; 916d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 91707b50cabSHong Zhang if (sctx.newshift) break; 918504a3fadSHong Zhang pv[nbdiag] = sctx.pv; 919137fb511SHong Zhang } 920137fb511SHong Zhang 92107b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 922137fb511SHong Zhang /* 923137fb511SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 924137fb511SHong Zhang * then try lower shift 925137fb511SHong Zhang */ 9260481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 9270481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 9280481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 92907b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 930137fb511SHong Zhang sctx.nshift++; 931137fb511SHong Zhang } 93207b50cabSHong Zhang } while (sctx.newshift); 933137fb511SHong Zhang 934137fb511SHong Zhang /* invert diagonal entries for simplier triangular solves */ 935137fb511SHong Zhang for (i=0; i<n; i++) { 936b051339dSHong Zhang a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]]; 937137fb511SHong Zhang } 938137fb511SHong Zhang 939137fb511SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 940137fb511SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 941137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 9422205254eSKarl Rupp 943b051339dSHong Zhang A->ops->solve = MatSolve_SeqAIJ_InplaceWithPerm; 944ad04f41aSHong Zhang A->ops->solveadd = MatSolveAdd_SeqAIJ_inplace; 945ad04f41aSHong Zhang A->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 946ad04f41aSHong Zhang A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 9472205254eSKarl Rupp 948b051339dSHong Zhang A->assembled = PETSC_TRUE; 9495c9eb25fSBarry Smith A->preallocated = PETSC_TRUE; 9502205254eSKarl Rupp 951d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr); 952137fb511SHong Zhang if (sctx.nshift) { 953f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 9540481f469SBarry 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); 955f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 956e738e422SBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 957137fb511SHong Zhang } 958137fb511SHong Zhang } 959137fb511SHong Zhang PetscFunctionReturn(0); 960137fb511SHong Zhang } 961137fb511SHong Zhang 9620a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9634a2ae208SSatish Balay #undef __FUNCT__ 9644a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ" 9650481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info) 966da3a660dSBarry Smith { 967dfbe8321SBarry Smith PetscErrorCode ierr; 968416022c9SBarry Smith Mat C; 969416022c9SBarry Smith 9703a40ed3dSBarry Smith PetscFunctionBegin; 9712692d6eeSBarry Smith ierr = MatGetFactor(A,MATSOLVERPETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr); 972719d5645SBarry Smith ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr); 973719d5645SBarry Smith ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr); 9742205254eSKarl Rupp 975db4efbfdSBarry Smith A->ops->solve = C->ops->solve; 976db4efbfdSBarry Smith A->ops->solvetranspose = C->ops->solvetranspose; 9772205254eSKarl Rupp 978eb6b5d47SBarry Smith ierr = MatHeaderMerge(A,C);CHKERRQ(ierr); 9793bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)A,(PetscObject)((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr); 9803a40ed3dSBarry Smith PetscFunctionReturn(0); 981da3a660dSBarry Smith } 9820a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9831d098868SHong Zhang 984889a8dedSBarry Smith 9854a2ae208SSatish Balay #undef __FUNCT__ 986ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_inplace" 987ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx) 9888c37ef55SBarry Smith { 989416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 990416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 9916849ba73SBarry Smith PetscErrorCode ierr; 9925d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 9935d0c19d7SBarry Smith PetscInt nz; 9945d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 995dd6ea824SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 996d9fead3dSBarry Smith const PetscScalar *b; 997dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 9988c37ef55SBarry Smith 9993a40ed3dSBarry Smith PetscFunctionBegin; 10003a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 100191bf9adeSBarry Smith 10023649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 10031ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1004416022c9SBarry Smith tmp = a->solve_work; 10058c37ef55SBarry Smith 10063b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10073b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 10088c37ef55SBarry Smith 10098c37ef55SBarry Smith /* forward solve the lower triangular */ 10108c37ef55SBarry Smith tmp[0] = b[*r++]; 1011010a20caSHong Zhang tmps = tmp; 10128c37ef55SBarry Smith for (i=1; i<n; i++) { 1013010a20caSHong Zhang v = aa + ai[i]; 1014010a20caSHong Zhang vi = aj + ai[i]; 101553e7425aSBarry Smith nz = a->diag[i] - ai[i]; 101653e7425aSBarry Smith sum = b[*r++]; 1017003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10188c37ef55SBarry Smith tmp[i] = sum; 10198c37ef55SBarry Smith } 10208c37ef55SBarry Smith 10218c37ef55SBarry Smith /* backward solve the upper triangular */ 10228c37ef55SBarry Smith for (i=n-1; i>=0; i--) { 1023010a20caSHong Zhang v = aa + a->diag[i] + 1; 1024010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1025416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 10268c37ef55SBarry Smith sum = tmp[i]; 1027003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1028010a20caSHong Zhang x[*c--] = tmp[i] = sum*aa[a->diag[i]]; 10298c37ef55SBarry Smith } 10308c37ef55SBarry Smith 10313b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10323b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10333649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 10341ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1035dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 10363a40ed3dSBarry Smith PetscFunctionReturn(0); 10378c37ef55SBarry Smith } 1038026e39d0SSatish Balay 10392bebee5dSHong Zhang #undef __FUNCT__ 1040ad04f41aSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ_inplace" 1041ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X) 10422bebee5dSHong Zhang { 10432bebee5dSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10442bebee5dSHong Zhang IS iscol = a->col,isrow = a->row; 10452bebee5dSHong Zhang PetscErrorCode ierr; 10465d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10475d0c19d7SBarry Smith PetscInt nz,neq; 10485d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 1049dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1050dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1051ace3abfcSBarry Smith PetscBool bisdense,xisdense; 10522bebee5dSHong Zhang 10532bebee5dSHong Zhang PetscFunctionBegin; 10542bebee5dSHong Zhang if (!n) PetscFunctionReturn(0); 10552bebee5dSHong Zhang 1056251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1057e32f2f54SBarry Smith if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 1058251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1059e32f2f54SBarry Smith if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 1060db4efbfdSBarry Smith 10618c778c55SBarry Smith ierr = MatDenseGetArray(B,&b);CHKERRQ(ierr); 10628c778c55SBarry Smith ierr = MatDenseGetArray(X,&x);CHKERRQ(ierr); 10632bebee5dSHong Zhang 10642bebee5dSHong Zhang tmp = a->solve_work; 10652bebee5dSHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10662bebee5dSHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 10672bebee5dSHong Zhang 1068a36b22ccSSatish Balay for (neq=0; neq<B->cmap->n; neq++) { 10692bebee5dSHong Zhang /* forward solve the lower triangular */ 10702bebee5dSHong Zhang tmp[0] = b[r[0]]; 10712bebee5dSHong Zhang tmps = tmp; 10722bebee5dSHong Zhang for (i=1; i<n; i++) { 10732bebee5dSHong Zhang v = aa + ai[i]; 10742bebee5dSHong Zhang vi = aj + ai[i]; 10752bebee5dSHong Zhang nz = a->diag[i] - ai[i]; 10762bebee5dSHong Zhang sum = b[r[i]]; 1077003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10782bebee5dSHong Zhang tmp[i] = sum; 10792bebee5dSHong Zhang } 10802bebee5dSHong Zhang /* backward solve the upper triangular */ 10812bebee5dSHong Zhang for (i=n-1; i>=0; i--) { 10822bebee5dSHong Zhang v = aa + a->diag[i] + 1; 10832bebee5dSHong Zhang vi = aj + a->diag[i] + 1; 10842bebee5dSHong Zhang nz = ai[i+1] - a->diag[i] - 1; 10852bebee5dSHong Zhang sum = tmp[i]; 1086003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10872bebee5dSHong Zhang x[c[i]] = tmp[i] = sum*aa[a->diag[i]]; 10882bebee5dSHong Zhang } 10892bebee5dSHong Zhang 10902bebee5dSHong Zhang b += n; 10912bebee5dSHong Zhang x += n; 10922bebee5dSHong Zhang } 10932bebee5dSHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10942bebee5dSHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10958c778c55SBarry Smith ierr = MatDenseRestoreArray(B,&b);CHKERRQ(ierr); 10968c778c55SBarry Smith ierr = MatDenseRestoreArray(X,&x);CHKERRQ(ierr); 1097dc0b31edSSatish Balay ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 10982bebee5dSHong Zhang PetscFunctionReturn(0); 10992bebee5dSHong Zhang } 11002bebee5dSHong Zhang 1101137fb511SHong Zhang #undef __FUNCT__ 110235233d99SShri Abhyankar #define __FUNCT__ "MatMatSolve_SeqAIJ" 110335233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X) 11049bd0847aSShri Abhyankar { 11059bd0847aSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 11069bd0847aSShri Abhyankar IS iscol = a->col,isrow = a->row; 11079bd0847aSShri Abhyankar PetscErrorCode ierr; 11089bd0847aSShri Abhyankar PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 11099bd0847aSShri Abhyankar PetscInt nz,neq; 11109bd0847aSShri Abhyankar const PetscInt *rout,*cout,*r,*c; 11119bd0847aSShri Abhyankar PetscScalar *x,*b,*tmp,sum; 11129bd0847aSShri Abhyankar const MatScalar *aa = a->a,*v; 1113ace3abfcSBarry Smith PetscBool bisdense,xisdense; 11149bd0847aSShri Abhyankar 11159bd0847aSShri Abhyankar PetscFunctionBegin; 11169bd0847aSShri Abhyankar if (!n) PetscFunctionReturn(0); 11179bd0847aSShri Abhyankar 1118251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1119e32f2f54SBarry Smith if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 1120251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1121e32f2f54SBarry Smith if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 11229bd0847aSShri Abhyankar 11238c778c55SBarry Smith ierr = MatDenseGetArray(B,&b);CHKERRQ(ierr); 11248c778c55SBarry Smith ierr = MatDenseGetArray(X,&x);CHKERRQ(ierr); 11259bd0847aSShri Abhyankar 11269bd0847aSShri Abhyankar tmp = a->solve_work; 11279bd0847aSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 11289bd0847aSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 11299bd0847aSShri Abhyankar 11309bd0847aSShri Abhyankar for (neq=0; neq<B->cmap->n; neq++) { 11319bd0847aSShri Abhyankar /* forward solve the lower triangular */ 11329bd0847aSShri Abhyankar tmp[0] = b[r[0]]; 11339bd0847aSShri Abhyankar v = aa; 11349bd0847aSShri Abhyankar vi = aj; 11359bd0847aSShri Abhyankar for (i=1; i<n; i++) { 11369bd0847aSShri Abhyankar nz = ai[i+1] - ai[i]; 11379bd0847aSShri Abhyankar sum = b[r[i]]; 11389bd0847aSShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 11399bd0847aSShri Abhyankar tmp[i] = sum; 11409bd0847aSShri Abhyankar v += nz; vi += nz; 11419bd0847aSShri Abhyankar } 11429bd0847aSShri Abhyankar 11439bd0847aSShri Abhyankar /* backward solve the upper triangular */ 11449bd0847aSShri Abhyankar for (i=n-1; i>=0; i--) { 11459bd0847aSShri Abhyankar v = aa + adiag[i+1]+1; 11469bd0847aSShri Abhyankar vi = aj + adiag[i+1]+1; 11479bd0847aSShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 11489bd0847aSShri Abhyankar sum = tmp[i]; 11499bd0847aSShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 11509bd0847aSShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 11519bd0847aSShri Abhyankar } 11529bd0847aSShri Abhyankar 11539bd0847aSShri Abhyankar b += n; 11549bd0847aSShri Abhyankar x += n; 11559bd0847aSShri Abhyankar } 11569bd0847aSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 11579bd0847aSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 11588c778c55SBarry Smith ierr = MatDenseRestoreArray(B,&b);CHKERRQ(ierr); 11598c778c55SBarry Smith ierr = MatDenseRestoreArray(X,&x);CHKERRQ(ierr); 11609bd0847aSShri Abhyankar ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 11619bd0847aSShri Abhyankar PetscFunctionReturn(0); 11629bd0847aSShri Abhyankar } 11639bd0847aSShri Abhyankar 11649bd0847aSShri Abhyankar #undef __FUNCT__ 1165137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm" 1166137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx) 1167137fb511SHong Zhang { 1168137fb511SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1169137fb511SHong Zhang IS iscol = a->col,isrow = a->row; 1170137fb511SHong Zhang PetscErrorCode ierr; 11715d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 11725d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 11735d0c19d7SBarry Smith PetscInt nz,row; 1174dd6ea824SBarry Smith PetscScalar *x,*b,*tmp,*tmps,sum; 1175dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1176137fb511SHong Zhang 1177137fb511SHong Zhang PetscFunctionBegin; 1178137fb511SHong Zhang if (!n) PetscFunctionReturn(0); 1179137fb511SHong Zhang 1180137fb511SHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 1181137fb511SHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1182137fb511SHong Zhang tmp = a->solve_work; 1183137fb511SHong Zhang 1184137fb511SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1185137fb511SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1186137fb511SHong Zhang 1187137fb511SHong Zhang /* forward solve the lower triangular */ 1188137fb511SHong Zhang tmp[0] = b[*r++]; 1189137fb511SHong Zhang tmps = tmp; 1190137fb511SHong Zhang for (row=1; row<n; row++) { 1191137fb511SHong Zhang i = rout[row]; /* permuted row */ 1192137fb511SHong Zhang v = aa + ai[i]; 1193137fb511SHong Zhang vi = aj + ai[i]; 1194137fb511SHong Zhang nz = a->diag[i] - ai[i]; 1195137fb511SHong Zhang sum = b[*r++]; 1196003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1197137fb511SHong Zhang tmp[row] = sum; 1198137fb511SHong Zhang } 1199137fb511SHong Zhang 1200137fb511SHong Zhang /* backward solve the upper triangular */ 1201137fb511SHong Zhang for (row=n-1; row>=0; row--) { 1202137fb511SHong Zhang i = rout[row]; /* permuted row */ 1203137fb511SHong Zhang v = aa + a->diag[i] + 1; 1204137fb511SHong Zhang vi = aj + a->diag[i] + 1; 1205137fb511SHong Zhang nz = ai[i+1] - a->diag[i] - 1; 1206137fb511SHong Zhang sum = tmp[row]; 1207003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1208137fb511SHong Zhang x[*c--] = tmp[row] = sum*aa[a->diag[i]]; 1209137fb511SHong Zhang } 1210137fb511SHong Zhang 1211137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1212137fb511SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1213137fb511SHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 1214137fb511SHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1215dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1216137fb511SHong Zhang PetscFunctionReturn(0); 1217137fb511SHong Zhang } 1218137fb511SHong Zhang 1219930ae53cSBarry Smith /* ----------------------------------------------------------- */ 1220c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/fsolve.h> 12214a2ae208SSatish Balay #undef __FUNCT__ 1222ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_inplace" 1223ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx) 1224930ae53cSBarry Smith { 1225930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 12266849ba73SBarry Smith PetscErrorCode ierr; 1227d0f46423SBarry Smith PetscInt n = A->rmap->n; 1228b377110cSBarry Smith const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag; 1229356650c2SBarry Smith PetscScalar *x; 1230356650c2SBarry Smith const PetscScalar *b; 1231dd6ea824SBarry Smith const MatScalar *aa = a->a; 1232aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 1233356650c2SBarry Smith PetscInt adiag_i,i,nz,ai_i; 1234b377110cSBarry Smith const PetscInt *vi; 1235dd6ea824SBarry Smith const MatScalar *v; 1236dd6ea824SBarry Smith PetscScalar sum; 1237d85afc42SSatish Balay #endif 1238930ae53cSBarry Smith 12393a40ed3dSBarry Smith PetscFunctionBegin; 12403a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 1241930ae53cSBarry Smith 12423649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 12431ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1244930ae53cSBarry Smith 1245aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 12461c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 12471c4feecaSSatish Balay #else 1248930ae53cSBarry Smith /* forward solve the lower triangular */ 1249930ae53cSBarry Smith x[0] = b[0]; 1250930ae53cSBarry Smith for (i=1; i<n; i++) { 1251930ae53cSBarry Smith ai_i = ai[i]; 1252930ae53cSBarry Smith v = aa + ai_i; 1253930ae53cSBarry Smith vi = aj + ai_i; 1254930ae53cSBarry Smith nz = adiag[i] - ai_i; 1255930ae53cSBarry Smith sum = b[i]; 1256003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1257930ae53cSBarry Smith x[i] = sum; 1258930ae53cSBarry Smith } 1259930ae53cSBarry Smith 1260930ae53cSBarry Smith /* backward solve the upper triangular */ 1261930ae53cSBarry Smith for (i=n-1; i>=0; i--) { 1262930ae53cSBarry Smith adiag_i = adiag[i]; 1263d708749eSBarry Smith v = aa + adiag_i + 1; 1264d708749eSBarry Smith vi = aj + adiag_i + 1; 1265930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 1266930ae53cSBarry Smith sum = x[i]; 1267003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1268930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 1269930ae53cSBarry Smith } 12701c4feecaSSatish Balay #endif 1271dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 12723649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 12731ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 12743a40ed3dSBarry Smith PetscFunctionReturn(0); 1275930ae53cSBarry Smith } 1276930ae53cSBarry Smith 12774a2ae208SSatish Balay #undef __FUNCT__ 1278ad04f41aSHong Zhang #define __FUNCT__ "MatSolveAdd_SeqAIJ_inplace" 1279ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx) 1280da3a660dSBarry Smith { 1281416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1282416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 12836849ba73SBarry Smith PetscErrorCode ierr; 128404fbf559SBarry Smith PetscInt i, n = A->rmap->n,j; 12855d0c19d7SBarry Smith PetscInt nz; 128604fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j; 128704fbf559SBarry Smith PetscScalar *x,*tmp,sum; 128804fbf559SBarry Smith const PetscScalar *b; 1289dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1290da3a660dSBarry Smith 12913a40ed3dSBarry Smith PetscFunctionBegin; 129278b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 1293da3a660dSBarry Smith 12943649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 12951ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1296416022c9SBarry Smith tmp = a->solve_work; 1297da3a660dSBarry Smith 12983b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12993b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1300da3a660dSBarry Smith 1301da3a660dSBarry Smith /* forward solve the lower triangular */ 1302da3a660dSBarry Smith tmp[0] = b[*r++]; 1303da3a660dSBarry Smith for (i=1; i<n; i++) { 1304010a20caSHong Zhang v = aa + ai[i]; 1305010a20caSHong Zhang vi = aj + ai[i]; 1306416022c9SBarry Smith nz = a->diag[i] - ai[i]; 1307da3a660dSBarry Smith sum = b[*r++]; 130804fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1309da3a660dSBarry Smith tmp[i] = sum; 1310da3a660dSBarry Smith } 1311da3a660dSBarry Smith 1312da3a660dSBarry Smith /* backward solve the upper triangular */ 1313da3a660dSBarry Smith for (i=n-1; i>=0; i--) { 1314010a20caSHong Zhang v = aa + a->diag[i] + 1; 1315010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1316416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 1317da3a660dSBarry Smith sum = tmp[i]; 131804fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1319010a20caSHong Zhang tmp[i] = sum*aa[a->diag[i]]; 1320da3a660dSBarry Smith x[*c--] += tmp[i]; 1321da3a660dSBarry Smith } 1322da3a660dSBarry Smith 13233b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13243b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 13253649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 13261ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1327dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 13283a40ed3dSBarry Smith PetscFunctionReturn(0); 1329da3a660dSBarry Smith } 133004fbf559SBarry Smith 13314a2ae208SSatish Balay #undef __FUNCT__ 133235233d99SShri Abhyankar #define __FUNCT__ "MatSolveAdd_SeqAIJ" 133335233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 13343c0119dfSShri Abhyankar { 13353c0119dfSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13363c0119dfSShri Abhyankar IS iscol = a->col,isrow = a->row; 13373c0119dfSShri Abhyankar PetscErrorCode ierr; 13383c0119dfSShri Abhyankar PetscInt i, n = A->rmap->n,j; 13393c0119dfSShri Abhyankar PetscInt nz; 13403c0119dfSShri Abhyankar const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 13413c0119dfSShri Abhyankar PetscScalar *x,*tmp,sum; 13423c0119dfSShri Abhyankar const PetscScalar *b; 13433c0119dfSShri Abhyankar const MatScalar *aa = a->a,*v; 13443c0119dfSShri Abhyankar 13453c0119dfSShri Abhyankar PetscFunctionBegin; 13463c0119dfSShri Abhyankar if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 13473c0119dfSShri Abhyankar 13483649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 13493c0119dfSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 13503c0119dfSShri Abhyankar tmp = a->solve_work; 13513c0119dfSShri Abhyankar 13523c0119dfSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13533c0119dfSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 13543c0119dfSShri Abhyankar 13553c0119dfSShri Abhyankar /* forward solve the lower triangular */ 13563c0119dfSShri Abhyankar tmp[0] = b[r[0]]; 13573c0119dfSShri Abhyankar v = aa; 13583c0119dfSShri Abhyankar vi = aj; 13593c0119dfSShri Abhyankar for (i=1; i<n; i++) { 13603c0119dfSShri Abhyankar nz = ai[i+1] - ai[i]; 13613c0119dfSShri Abhyankar sum = b[r[i]]; 13623c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 13633c0119dfSShri Abhyankar tmp[i] = sum; 13642205254eSKarl Rupp v += nz; 13652205254eSKarl Rupp vi += nz; 13663c0119dfSShri Abhyankar } 13673c0119dfSShri Abhyankar 13683c0119dfSShri Abhyankar /* backward solve the upper triangular */ 13693c0119dfSShri Abhyankar v = aa + adiag[n-1]; 13703c0119dfSShri Abhyankar vi = aj + adiag[n-1]; 13713c0119dfSShri Abhyankar for (i=n-1; i>=0; i--) { 13723c0119dfSShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 13733c0119dfSShri Abhyankar sum = tmp[i]; 13743c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 13753c0119dfSShri Abhyankar tmp[i] = sum*v[nz]; 13763c0119dfSShri Abhyankar x[c[i]] += tmp[i]; 13773c0119dfSShri Abhyankar v += nz+1; vi += nz+1; 13783c0119dfSShri Abhyankar } 13793c0119dfSShri Abhyankar 13803c0119dfSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13813c0119dfSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 13823649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 13833c0119dfSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 13843c0119dfSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 13853c0119dfSShri Abhyankar PetscFunctionReturn(0); 13863c0119dfSShri Abhyankar } 13873c0119dfSShri Abhyankar 13883c0119dfSShri Abhyankar #undef __FUNCT__ 1389ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTranspose_SeqAIJ_inplace" 1390ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx) 1391da3a660dSBarry Smith { 1392416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13932235e667SBarry Smith IS iscol = a->col,isrow = a->row; 13946849ba73SBarry Smith PetscErrorCode ierr; 139504fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 139604fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 139704fbf559SBarry Smith PetscInt nz; 139804fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1399dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 140004fbf559SBarry Smith const PetscScalar *b; 1401da3a660dSBarry Smith 14023a40ed3dSBarry Smith PetscFunctionBegin; 14033649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 14041ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1405416022c9SBarry Smith tmp = a->solve_work; 1406da3a660dSBarry Smith 14072235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 14082235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1409da3a660dSBarry Smith 1410da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 14112235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1412da3a660dSBarry Smith 1413da3a660dSBarry Smith /* forward solve the U^T */ 1414da3a660dSBarry Smith for (i=0; i<n; i++) { 1415010a20caSHong Zhang v = aa + diag[i]; 1416010a20caSHong Zhang vi = aj + diag[i] + 1; 1417f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 1418c38d4ed2SBarry Smith s1 = tmp[i]; 1419ef66eb69SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 142004fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1421c38d4ed2SBarry Smith tmp[i] = s1; 1422da3a660dSBarry Smith } 1423da3a660dSBarry Smith 1424da3a660dSBarry Smith /* backward solve the L^T */ 1425da3a660dSBarry Smith for (i=n-1; i>=0; i--) { 1426010a20caSHong Zhang v = aa + diag[i] - 1; 1427010a20caSHong Zhang vi = aj + diag[i] - 1; 1428f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 1429f1af5d2fSBarry Smith s1 = tmp[i]; 143004fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 1431da3a660dSBarry Smith } 1432da3a660dSBarry Smith 1433da3a660dSBarry Smith /* copy tmp into x according to permutation */ 1434591294e4SBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1435da3a660dSBarry Smith 14362235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 14372235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 14383649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 14391ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1440da3a660dSBarry Smith 1441dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 14423a40ed3dSBarry Smith PetscFunctionReturn(0); 1443da3a660dSBarry Smith } 1444da3a660dSBarry Smith 1445d1fa9404SShri Abhyankar #undef __FUNCT__ 144635233d99SShri Abhyankar #define __FUNCT__ "MatSolveTranspose_SeqAIJ" 144735233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 1448d1fa9404SShri Abhyankar { 1449d1fa9404SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1450d1fa9404SShri Abhyankar IS iscol = a->col,isrow = a->row; 1451d1fa9404SShri Abhyankar PetscErrorCode ierr; 1452d1fa9404SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1453d1fa9404SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1454d1fa9404SShri Abhyankar PetscInt nz; 1455d1fa9404SShri Abhyankar PetscScalar *x,*tmp,s1; 1456d1fa9404SShri Abhyankar const MatScalar *aa = a->a,*v; 1457d1fa9404SShri Abhyankar const PetscScalar *b; 1458d1fa9404SShri Abhyankar 1459d1fa9404SShri Abhyankar PetscFunctionBegin; 14603649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1461d1fa9404SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1462d1fa9404SShri Abhyankar tmp = a->solve_work; 1463d1fa9404SShri Abhyankar 1464d1fa9404SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1465d1fa9404SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1466d1fa9404SShri Abhyankar 1467d1fa9404SShri Abhyankar /* copy the b into temp work space according to permutation */ 1468d1fa9404SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1469d1fa9404SShri Abhyankar 1470d1fa9404SShri Abhyankar /* forward solve the U^T */ 1471d1fa9404SShri Abhyankar for (i=0; i<n; i++) { 1472d1fa9404SShri Abhyankar v = aa + adiag[i+1] + 1; 1473d1fa9404SShri Abhyankar vi = aj + adiag[i+1] + 1; 1474d1fa9404SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1475d1fa9404SShri Abhyankar s1 = tmp[i]; 1476d1fa9404SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1477d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1478d1fa9404SShri Abhyankar tmp[i] = s1; 1479d1fa9404SShri Abhyankar } 1480d1fa9404SShri Abhyankar 1481d1fa9404SShri Abhyankar /* backward solve the L^T */ 1482d1fa9404SShri Abhyankar for (i=n-1; i>=0; i--) { 1483d1fa9404SShri Abhyankar v = aa + ai[i]; 1484d1fa9404SShri Abhyankar vi = aj + ai[i]; 1485d1fa9404SShri Abhyankar nz = ai[i+1] - ai[i]; 1486d1fa9404SShri Abhyankar s1 = tmp[i]; 1487d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1488d1fa9404SShri Abhyankar } 1489d1fa9404SShri Abhyankar 1490d1fa9404SShri Abhyankar /* copy tmp into x according to permutation */ 1491d1fa9404SShri Abhyankar for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1492d1fa9404SShri Abhyankar 1493d1fa9404SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1494d1fa9404SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 14953649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1496d1fa9404SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1497d1fa9404SShri Abhyankar 1498d1fa9404SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1499d1fa9404SShri Abhyankar PetscFunctionReturn(0); 1500d1fa9404SShri Abhyankar } 1501d1fa9404SShri Abhyankar 15024a2ae208SSatish Balay #undef __FUNCT__ 1503ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ_inplace" 1504ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx) 1505da3a660dSBarry Smith { 1506416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 15072235e667SBarry Smith IS iscol = a->col,isrow = a->row; 15086849ba73SBarry Smith PetscErrorCode ierr; 150904fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 151004fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 151104fbf559SBarry Smith PetscInt nz; 151204fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1513dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 151404fbf559SBarry Smith const PetscScalar *b; 15156abc6512SBarry Smith 15163a40ed3dSBarry Smith PetscFunctionBegin; 151723bc6035SMatthew Knepley if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 15183649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 15191ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1520416022c9SBarry Smith tmp = a->solve_work; 15216abc6512SBarry Smith 15222235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 15232235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 15246abc6512SBarry Smith 15256abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 15262235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 15276abc6512SBarry Smith 15286abc6512SBarry Smith /* forward solve the U^T */ 15296abc6512SBarry Smith for (i=0; i<n; i++) { 1530010a20caSHong Zhang v = aa + diag[i]; 1531010a20caSHong Zhang vi = aj + diag[i] + 1; 1532f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 153304fbf559SBarry Smith s1 = tmp[i]; 153404fbf559SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 153504fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 153604fbf559SBarry Smith tmp[i] = s1; 15376abc6512SBarry Smith } 15386abc6512SBarry Smith 15396abc6512SBarry Smith /* backward solve the L^T */ 15406abc6512SBarry Smith for (i=n-1; i>=0; i--) { 1541010a20caSHong Zhang v = aa + diag[i] - 1; 1542010a20caSHong Zhang vi = aj + diag[i] - 1; 1543f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 154404fbf559SBarry Smith s1 = tmp[i]; 154504fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 15466abc6512SBarry Smith } 15476abc6512SBarry Smith 15486abc6512SBarry Smith /* copy tmp into x according to permutation */ 15496abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 15506abc6512SBarry Smith 15512235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 15522235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 15533649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 15541ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 15556abc6512SBarry Smith 155604fbf559SBarry Smith ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 15573a40ed3dSBarry Smith PetscFunctionReturn(0); 1558da3a660dSBarry Smith } 155904fbf559SBarry Smith 1560533e6011SShri Abhyankar #undef __FUNCT__ 156135233d99SShri Abhyankar #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ" 156235233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 1563533e6011SShri Abhyankar { 1564533e6011SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1565533e6011SShri Abhyankar IS iscol = a->col,isrow = a->row; 1566533e6011SShri Abhyankar PetscErrorCode ierr; 1567533e6011SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1568533e6011SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1569533e6011SShri Abhyankar PetscInt nz; 1570533e6011SShri Abhyankar PetscScalar *x,*tmp,s1; 1571533e6011SShri Abhyankar const MatScalar *aa = a->a,*v; 1572533e6011SShri Abhyankar const PetscScalar *b; 1573533e6011SShri Abhyankar 1574533e6011SShri Abhyankar PetscFunctionBegin; 1575533e6011SShri Abhyankar if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 15763649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1577533e6011SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1578533e6011SShri Abhyankar tmp = a->solve_work; 1579533e6011SShri Abhyankar 1580533e6011SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1581533e6011SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1582533e6011SShri Abhyankar 1583533e6011SShri Abhyankar /* copy the b into temp work space according to permutation */ 1584533e6011SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1585533e6011SShri Abhyankar 1586533e6011SShri Abhyankar /* forward solve the U^T */ 1587533e6011SShri Abhyankar for (i=0; i<n; i++) { 1588533e6011SShri Abhyankar v = aa + adiag[i+1] + 1; 1589533e6011SShri Abhyankar vi = aj + adiag[i+1] + 1; 1590533e6011SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1591533e6011SShri Abhyankar s1 = tmp[i]; 1592533e6011SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1593533e6011SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1594533e6011SShri Abhyankar tmp[i] = s1; 1595533e6011SShri Abhyankar } 1596533e6011SShri Abhyankar 1597533e6011SShri Abhyankar 1598533e6011SShri Abhyankar /* backward solve the L^T */ 1599533e6011SShri Abhyankar for (i=n-1; i>=0; i--) { 1600533e6011SShri Abhyankar v = aa + ai[i]; 1601533e6011SShri Abhyankar vi = aj + ai[i]; 1602533e6011SShri Abhyankar nz = ai[i+1] - ai[i]; 1603533e6011SShri Abhyankar s1 = tmp[i]; 1604c5e3b2a3SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1605533e6011SShri Abhyankar } 1606533e6011SShri Abhyankar 1607533e6011SShri Abhyankar /* copy tmp into x according to permutation */ 1608533e6011SShri Abhyankar for (i=0; i<n; i++) x[r[i]] += tmp[i]; 1609533e6011SShri Abhyankar 1610533e6011SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1611533e6011SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 16123649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1613533e6011SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1614533e6011SShri Abhyankar 1615533e6011SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1616533e6011SShri Abhyankar PetscFunctionReturn(0); 1617533e6011SShri Abhyankar } 1618533e6011SShri Abhyankar 16199e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 1620b47f5a65SHong Zhang 162109573ac7SBarry Smith extern PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscBool); 162208480c60SBarry Smith 16238fc3a347SHong Zhang /* 16248a76ed4fSHong Zhang ilu() under revised new data structure. 1625813a1f2bSShri Abhyankar Factored arrays bj and ba are stored as 1626813a1f2bSShri Abhyankar L(0,:), L(1,:), ...,L(n-1,:), U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:) 1627813a1f2bSShri Abhyankar 1628813a1f2bSShri Abhyankar bi=fact->i is an array of size n+1, in which 1629813a1f2bSShri Abhyankar bi+ 1630baabb860SHong Zhang bi[i]: points to 1st entry of L(i,:),i=0,...,n-1 1631baabb860SHong Zhang bi[n]: points to L(n-1,n-1)+1 1632baabb860SHong Zhang 1633813a1f2bSShri Abhyankar bdiag=fact->diag is an array of size n+1,in which 1634baabb860SHong Zhang bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1 1635a24f213cSHong Zhang bdiag[n]: points to entry of U(n-1,0)-1 1636813a1f2bSShri Abhyankar 1637813a1f2bSShri Abhyankar U(i,:) contains bdiag[i] as its last entry, i.e., 1638813a1f2bSShri Abhyankar U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 1639813a1f2bSShri Abhyankar */ 1640813a1f2bSShri Abhyankar #undef __FUNCT__ 164135233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0" 164235233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1643813a1f2bSShri Abhyankar { 1644813a1f2bSShri Abhyankar 1645813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1646813a1f2bSShri Abhyankar PetscErrorCode ierr; 1647bbd65245SShri Abhyankar const PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag; 1648bbd65245SShri Abhyankar PetscInt i,j,k=0,nz,*bi,*bj,*bdiag; 1649ace3abfcSBarry Smith PetscBool missing; 16501df811f5SHong Zhang IS isicol; 1651813a1f2bSShri Abhyankar 1652813a1f2bSShri Abhyankar PetscFunctionBegin; 1653e32f2f54SBarry 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); 16541df811f5SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 1655e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 16561df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 1657813a1f2bSShri Abhyankar ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr); 1658813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1659813a1f2bSShri Abhyankar 1660813a1f2bSShri Abhyankar /* allocate matrix arrays for new data structure */ 1661dcca6d9dSJed Brown ierr = PetscMalloc3(ai[n]+1,&b->a,ai[n]+1,&b->j,n+1,&b->i);CHKERRQ(ierr); 16623bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 16632205254eSKarl Rupp 1664813a1f2bSShri Abhyankar b->singlemalloc = PETSC_TRUE; 1665813a1f2bSShri Abhyankar if (!b->diag) { 1666*785e854fSJed Brown ierr = PetscMalloc1((n+1),&b->diag);CHKERRQ(ierr); 16673bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1668813a1f2bSShri Abhyankar } 1669813a1f2bSShri Abhyankar bdiag = b->diag; 1670813a1f2bSShri Abhyankar 1671813a1f2bSShri Abhyankar if (n > 0) { 1672813a1f2bSShri Abhyankar ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr); 1673813a1f2bSShri Abhyankar } 1674813a1f2bSShri Abhyankar 1675813a1f2bSShri Abhyankar /* set bi and bj with new data structure */ 1676813a1f2bSShri Abhyankar bi = b->i; 1677813a1f2bSShri Abhyankar bj = b->j; 1678813a1f2bSShri Abhyankar 1679813a1f2bSShri Abhyankar /* L part */ 1680e60cf9a0SBarry Smith bi[0] = 0; 1681813a1f2bSShri Abhyankar for (i=0; i<n; i++) { 1682813a1f2bSShri Abhyankar nz = adiag[i] - ai[i]; 1683813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nz; 1684813a1f2bSShri Abhyankar aj = a->j + ai[i]; 1685813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 1686bbd65245SShri Abhyankar /* *bj = aj[j]; bj++; */ 1687bbd65245SShri Abhyankar bj[k++] = aj[j]; 1688813a1f2bSShri Abhyankar } 1689813a1f2bSShri Abhyankar } 1690813a1f2bSShri Abhyankar 1691813a1f2bSShri Abhyankar /* U part */ 169262fbd6afSShri Abhyankar bdiag[n] = bi[n]-1; 1693813a1f2bSShri Abhyankar for (i=n-1; i>=0; i--) { 1694813a1f2bSShri Abhyankar nz = ai[i+1] - adiag[i] - 1; 1695813a1f2bSShri Abhyankar aj = a->j + adiag[i] + 1; 1696813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 1697bbd65245SShri Abhyankar /* *bj = aj[j]; bj++; */ 1698bbd65245SShri Abhyankar bj[k++] = aj[j]; 1699813a1f2bSShri Abhyankar } 1700813a1f2bSShri Abhyankar /* diag[i] */ 1701bbd65245SShri Abhyankar /* *bj = i; bj++; */ 1702bbd65245SShri Abhyankar bj[k++] = i; 1703a24f213cSHong Zhang bdiag[i] = bdiag[i+1] + nz + 1; 1704813a1f2bSShri Abhyankar } 17051df811f5SHong Zhang 1706d5f3da31SBarry Smith fact->factortype = MAT_FACTOR_ILU; 17071df811f5SHong Zhang fact->info.factor_mallocs = 0; 17081df811f5SHong Zhang fact->info.fill_ratio_given = info->fill; 17091df811f5SHong Zhang fact->info.fill_ratio_needed = 1.0; 171035233d99SShri Abhyankar fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1711abb87a52SBarry Smith ierr = Mat_CheckInode_FactorLU(fact);CHKERRQ(ierr); 17121df811f5SHong Zhang 17131df811f5SHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 17141df811f5SHong Zhang b->row = isrow; 17151df811f5SHong Zhang b->col = iscol; 17161df811f5SHong Zhang b->icol = isicol; 1717*785e854fSJed Brown ierr = PetscMalloc1((fact->rmap->n+1),&b->solve_work);CHKERRQ(ierr); 17181df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 17191df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1720813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1721813a1f2bSShri Abhyankar } 1722813a1f2bSShri Abhyankar 1723813a1f2bSShri Abhyankar #undef __FUNCT__ 172435233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ" 172535233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1726813a1f2bSShri Abhyankar { 1727813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1728813a1f2bSShri Abhyankar IS isicol; 1729813a1f2bSShri Abhyankar PetscErrorCode ierr; 1730813a1f2bSShri Abhyankar const PetscInt *r,*ic; 17311df811f5SHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j; 1732813a1f2bSShri Abhyankar PetscInt *bi,*cols,nnz,*cols_lvl; 1733813a1f2bSShri Abhyankar PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 1734813a1f2bSShri Abhyankar PetscInt i,levels,diagonal_fill; 1735ace3abfcSBarry Smith PetscBool col_identity,row_identity; 1736813a1f2bSShri Abhyankar PetscReal f; 17370298fd71SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=NULL; 1738813a1f2bSShri Abhyankar PetscBT lnkbt; 1739813a1f2bSShri Abhyankar PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 17400298fd71SBarry Smith PetscFreeSpaceList free_space =NULL,current_space=NULL; 17410298fd71SBarry Smith PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL; 1742813a1f2bSShri Abhyankar 1743813a1f2bSShri Abhyankar PetscFunctionBegin; 17443898ab1dSSatish Balay /* Uncomment the old data struct part only while testing new data structure for MatSolve() */ 1745124a3351SHong Zhang /* 1746ace3abfcSBarry Smith PetscBool olddatastruct=PETSC_FALSE; 17470298fd71SBarry Smith ierr = PetscOptionsGetBool(NULL,"-ilu_old",&olddatastruct,NULL);CHKERRQ(ierr); 174835233d99SShri Abhyankar if (olddatastruct) { 1749ad04f41aSHong Zhang ierr = MatILUFactorSymbolic_SeqAIJ_inplace(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1750ad04f41aSHong Zhang PetscFunctionReturn(0); 1751ad04f41aSHong Zhang } 1752124a3351SHong Zhang */ 1753ad04f41aSHong Zhang 1754813a1f2bSShri Abhyankar levels = (PetscInt)info->levels; 1755813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 1756813a1f2bSShri Abhyankar ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 1757813a1f2bSShri Abhyankar if (!levels && row_identity && col_identity) { 1758813a1f2bSShri Abhyankar /* special case: ilu(0) with natural ordering */ 175935233d99SShri Abhyankar ierr = MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1760124a3351SHong Zhang if (a->inode.size) { 1761124a3351SHong Zhang fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 1762124a3351SHong Zhang } 1763813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1764813a1f2bSShri Abhyankar } 1765813a1f2bSShri Abhyankar 1766e32f2f54SBarry 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); 17671df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 1768813a1f2bSShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1769813a1f2bSShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 1770813a1f2bSShri Abhyankar 17711bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 1772*785e854fSJed Brown ierr = PetscMalloc1((n+1),&bi);CHKERRQ(ierr); 1773*785e854fSJed Brown ierr = PetscMalloc1((n+1),&bdiag);CHKERRQ(ierr); 1774e60cf9a0SBarry Smith bi[0] = bdiag[0] = 0; 1775dcca6d9dSJed Brown ierr = PetscMalloc2(n,&bj_ptr,n,&bjlvl_ptr);CHKERRQ(ierr); 1776813a1f2bSShri Abhyankar 1777813a1f2bSShri Abhyankar /* create a linked list for storing column indices of the active row */ 1778813a1f2bSShri Abhyankar nlnk = n + 1; 1779813a1f2bSShri Abhyankar ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1780813a1f2bSShri Abhyankar 1781813a1f2bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 17821df811f5SHong Zhang f = info->fill; 17831df811f5SHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 1784813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 1785813a1f2bSShri Abhyankar current_space = free_space; 1786813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 1787813a1f2bSShri Abhyankar current_space_lvl = free_space_lvl; 1788813a1f2bSShri Abhyankar for (i=0; i<n; i++) { 1789813a1f2bSShri Abhyankar nzi = 0; 1790813a1f2bSShri Abhyankar /* copy current row into linked list */ 1791813a1f2bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 1792e32f2f54SBarry 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); 1793813a1f2bSShri Abhyankar cols = aj + ai[r[i]]; 1794813a1f2bSShri Abhyankar lnk[i] = -1; /* marker to indicate if diagonal exists */ 1795813a1f2bSShri Abhyankar ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1796813a1f2bSShri Abhyankar nzi += nlnk; 1797813a1f2bSShri Abhyankar 1798813a1f2bSShri Abhyankar /* make sure diagonal entry is included */ 1799813a1f2bSShri Abhyankar if (diagonal_fill && lnk[i] == -1) { 1800813a1f2bSShri Abhyankar fm = n; 1801813a1f2bSShri Abhyankar while (lnk[fm] < i) fm = lnk[fm]; 1802813a1f2bSShri Abhyankar lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 1803813a1f2bSShri Abhyankar lnk[fm] = i; 1804813a1f2bSShri Abhyankar lnk_lvl[i] = 0; 1805813a1f2bSShri Abhyankar nzi++; dcount++; 1806813a1f2bSShri Abhyankar } 1807813a1f2bSShri Abhyankar 1808813a1f2bSShri Abhyankar /* add pivot rows into the active row */ 1809813a1f2bSShri Abhyankar nzbd = 0; 1810813a1f2bSShri Abhyankar prow = lnk[n]; 1811813a1f2bSShri Abhyankar while (prow < i) { 1812813a1f2bSShri Abhyankar nnz = bdiag[prow]; 1813813a1f2bSShri Abhyankar cols = bj_ptr[prow] + nnz + 1; 1814813a1f2bSShri Abhyankar cols_lvl = bjlvl_ptr[prow] + nnz + 1; 1815813a1f2bSShri Abhyankar nnz = bi[prow+1] - bi[prow] - nnz - 1; 1816813a1f2bSShri Abhyankar ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 1817813a1f2bSShri Abhyankar nzi += nlnk; 1818813a1f2bSShri Abhyankar prow = lnk[prow]; 1819813a1f2bSShri Abhyankar nzbd++; 1820813a1f2bSShri Abhyankar } 1821813a1f2bSShri Abhyankar bdiag[i] = nzbd; 1822813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nzi; 1823813a1f2bSShri Abhyankar /* if free space is not available, make more free space */ 1824813a1f2bSShri Abhyankar if (current_space->local_remaining<nzi) { 1825813a1f2bSShri Abhyankar nnz = 2*nzi*(n - i); /* estimated and max additional space needed */ 1826813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1827813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 1828813a1f2bSShri Abhyankar reallocs++; 1829813a1f2bSShri Abhyankar } 1830813a1f2bSShri Abhyankar 1831813a1f2bSShri Abhyankar /* copy data into free_space and free_space_lvl, then initialize lnk */ 1832813a1f2bSShri Abhyankar ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 1833813a1f2bSShri Abhyankar bj_ptr[i] = current_space->array; 1834813a1f2bSShri Abhyankar bjlvl_ptr[i] = current_space_lvl->array; 1835813a1f2bSShri Abhyankar 1836813a1f2bSShri Abhyankar /* make sure the active row i has diagonal entry */ 183765e19b50SBarry 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); 1838813a1f2bSShri Abhyankar 1839813a1f2bSShri Abhyankar current_space->array += nzi; 1840813a1f2bSShri Abhyankar current_space->local_used += nzi; 1841813a1f2bSShri Abhyankar current_space->local_remaining -= nzi; 1842813a1f2bSShri Abhyankar current_space_lvl->array += nzi; 1843813a1f2bSShri Abhyankar current_space_lvl->local_used += nzi; 1844813a1f2bSShri Abhyankar current_space_lvl->local_remaining -= nzi; 1845813a1f2bSShri Abhyankar } 1846813a1f2bSShri Abhyankar 1847813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1848813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 1849813a1f2bSShri Abhyankar /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 1850*785e854fSJed Brown ierr = PetscMalloc1((bi[n]+1),&bj);CHKERRQ(ierr); 18512ce24eb6SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 1852813a1f2bSShri Abhyankar 1853813a1f2bSShri Abhyankar ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1854813a1f2bSShri Abhyankar ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 18550e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 1856813a1f2bSShri Abhyankar 1857813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO) 1858813a1f2bSShri Abhyankar { 1859aef85c9fSShri Abhyankar PetscReal af = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 1860813a1f2bSShri Abhyankar ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 1861813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 1862813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 1863813a1f2bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1864813a1f2bSShri Abhyankar if (diagonal_fill) { 1865813a1f2bSShri Abhyankar ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 1866813a1f2bSShri Abhyankar } 1867813a1f2bSShri Abhyankar } 1868813a1f2bSShri Abhyankar #endif 1869813a1f2bSShri Abhyankar /* put together the new matrix */ 18700298fd71SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 18713bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)fact,(PetscObject)isicol);CHKERRQ(ierr); 1872813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 18732205254eSKarl Rupp 1874813a1f2bSShri Abhyankar b->free_a = PETSC_TRUE; 1875813a1f2bSShri Abhyankar b->free_ij = PETSC_TRUE; 1876813a1f2bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 18772205254eSKarl Rupp 1878*785e854fSJed Brown ierr = PetscMalloc1((bdiag[0]+1),&b->a);CHKERRQ(ierr); 18792205254eSKarl Rupp 1880813a1f2bSShri Abhyankar b->j = bj; 1881813a1f2bSShri Abhyankar b->i = bi; 1882813a1f2bSShri Abhyankar b->diag = bdiag; 1883813a1f2bSShri Abhyankar b->ilen = 0; 1884813a1f2bSShri Abhyankar b->imax = 0; 1885813a1f2bSShri Abhyankar b->row = isrow; 1886813a1f2bSShri Abhyankar b->col = iscol; 1887813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1888813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1889813a1f2bSShri Abhyankar b->icol = isicol; 18902205254eSKarl Rupp 1891*785e854fSJed Brown ierr = PetscMalloc1((n+1),&b->solve_work);CHKERRQ(ierr); 1892813a1f2bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. 1893813a1f2bSShri Abhyankar Allocate bdiag, solve_work, new a, new j */ 18943bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 1895f268cba8SShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 18962205254eSKarl Rupp 1897813a1f2bSShri Abhyankar (fact)->info.factor_mallocs = reallocs; 1898813a1f2bSShri Abhyankar (fact)->info.fill_ratio_given = f; 1899f268cba8SShri Abhyankar (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 190035233d99SShri Abhyankar (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1901f2344125SBarry Smith if (a->inode.size) { 1902f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 1903f2344125SBarry Smith } 1904abb87a52SBarry Smith ierr = Mat_CheckInode_FactorLU(fact);CHKERRQ(ierr); 1905813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1906813a1f2bSShri Abhyankar } 1907813a1f2bSShri Abhyankar 19086516239fSHong Zhang #undef __FUNCT__ 1909ad04f41aSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_inplace" 1910ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 19116516239fSHong Zhang { 19126516239fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 19136516239fSHong Zhang IS isicol; 19146516239fSHong Zhang PetscErrorCode ierr; 19156516239fSHong Zhang const PetscInt *r,*ic; 19166516239fSHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,d; 19176516239fSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 19186516239fSHong Zhang PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 19196516239fSHong Zhang PetscInt i,levels,diagonal_fill; 1920ace3abfcSBarry Smith PetscBool col_identity,row_identity; 19216516239fSHong Zhang PetscReal f; 19220298fd71SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=NULL; 19236516239fSHong Zhang PetscBT lnkbt; 19246516239fSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 19250298fd71SBarry Smith PetscFreeSpaceList free_space =NULL,current_space=NULL; 19260298fd71SBarry Smith PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL; 1927ace3abfcSBarry Smith PetscBool missing; 19286516239fSHong Zhang 19296516239fSHong Zhang PetscFunctionBegin; 1930e32f2f54SBarry 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); 19316516239fSHong Zhang f = info->fill; 19326516239fSHong Zhang levels = (PetscInt)info->levels; 19336516239fSHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 19342205254eSKarl Rupp 19356516239fSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 19366516239fSHong Zhang 19376516239fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 19386516239fSHong Zhang ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 19396516239fSHong Zhang if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */ 1940f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr); 19412205254eSKarl Rupp 1942ad04f41aSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 1943f2344125SBarry Smith if (a->inode.size) { 1944f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 1945f2344125SBarry Smith } 1946d5f3da31SBarry Smith fact->factortype = MAT_FACTOR_ILU; 1947719d5645SBarry Smith (fact)->info.factor_mallocs = 0; 1948719d5645SBarry Smith (fact)->info.fill_ratio_given = info->fill; 1949719d5645SBarry Smith (fact)->info.fill_ratio_needed = 1.0; 19502205254eSKarl Rupp 1951719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 19528ef3462fSBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 1953e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 195408480c60SBarry Smith b->row = isrow; 195508480c60SBarry Smith b->col = iscol; 195682bf6240SBarry Smith b->icol = isicol; 1957*785e854fSJed Brown ierr = PetscMalloc1(((fact)->rmap->n+1),&b->solve_work);CHKERRQ(ierr); 1958c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1959c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 19603a40ed3dSBarry Smith PetscFunctionReturn(0); 196108480c60SBarry Smith } 196208480c60SBarry Smith 1963898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1964898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 19659e25ed09SBarry Smith 19661bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 1967*785e854fSJed Brown ierr = PetscMalloc1((n+1),&bi);CHKERRQ(ierr); 1968*785e854fSJed Brown ierr = PetscMalloc1((n+1),&bdiag);CHKERRQ(ierr); 1969a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 19706cf997b0SBarry Smith 1971dcca6d9dSJed Brown ierr = PetscMalloc2(n,&bj_ptr,n,&bjlvl_ptr);CHKERRQ(ierr); 19725a8e39fbSHong Zhang 19735a8e39fbSHong Zhang /* create a linked list for storing column indices of the active row */ 19745a8e39fbSHong Zhang nlnk = n + 1; 19755a8e39fbSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19765a8e39fbSHong Zhang 19775a8e39fbSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 1978a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr); 19795a8e39fbSHong Zhang current_space = free_space; 1980a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr); 19815a8e39fbSHong Zhang current_space_lvl = free_space_lvl; 19825a8e39fbSHong Zhang 19835a8e39fbSHong Zhang for (i=0; i<n; i++) { 19845a8e39fbSHong Zhang nzi = 0; 19856cf997b0SBarry Smith /* copy current row into linked list */ 19865a8e39fbSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 1987e32f2f54SBarry 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); 19885a8e39fbSHong Zhang cols = aj + ai[r[i]]; 19895a8e39fbSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 19905a8e39fbSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19915a8e39fbSHong Zhang nzi += nlnk; 19926cf997b0SBarry Smith 19936cf997b0SBarry Smith /* make sure diagonal entry is included */ 19945a8e39fbSHong Zhang if (diagonal_fill && lnk[i] == -1) { 19956cf997b0SBarry Smith fm = n; 19965a8e39fbSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 19975a8e39fbSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 19985a8e39fbSHong Zhang lnk[fm] = i; 19995a8e39fbSHong Zhang lnk_lvl[i] = 0; 20005a8e39fbSHong Zhang nzi++; dcount++; 20016cf997b0SBarry Smith } 20026cf997b0SBarry Smith 20035a8e39fbSHong Zhang /* add pivot rows into the active row */ 20045a8e39fbSHong Zhang nzbd = 0; 20055a8e39fbSHong Zhang prow = lnk[n]; 20065a8e39fbSHong Zhang while (prow < i) { 20075a8e39fbSHong Zhang nnz = bdiag[prow]; 20085a8e39fbSHong Zhang cols = bj_ptr[prow] + nnz + 1; 20095a8e39fbSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 20105a8e39fbSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 20115a8e39fbSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 20125a8e39fbSHong Zhang nzi += nlnk; 20135a8e39fbSHong Zhang prow = lnk[prow]; 20145a8e39fbSHong Zhang nzbd++; 201556beaf04SBarry Smith } 20165a8e39fbSHong Zhang bdiag[i] = nzbd; 20175a8e39fbSHong Zhang bi[i+1] = bi[i] + nzi; 2018ecf371e4SBarry Smith 20195a8e39fbSHong Zhang /* if free space is not available, make more free space */ 20205a8e39fbSHong Zhang if (current_space->local_remaining<nzi) { 20215a8e39fbSHong Zhang nnz = nzi*(n - i); /* estimated and max additional space needed */ 2022a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2023a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 20245a8e39fbSHong Zhang reallocs++; 20255a8e39fbSHong Zhang } 2026ecf371e4SBarry Smith 20275a8e39fbSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 20285a8e39fbSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 20295a8e39fbSHong Zhang bj_ptr[i] = current_space->array; 20305a8e39fbSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 20315a8e39fbSHong Zhang 20325a8e39fbSHong Zhang /* make sure the active row i has diagonal entry */ 203365e19b50SBarry 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); 20345a8e39fbSHong Zhang 20355a8e39fbSHong Zhang current_space->array += nzi; 20365a8e39fbSHong Zhang current_space->local_used += nzi; 20375a8e39fbSHong Zhang current_space->local_remaining -= nzi; 20385a8e39fbSHong Zhang current_space_lvl->array += nzi; 20395a8e39fbSHong Zhang current_space_lvl->local_used += nzi; 20405a8e39fbSHong Zhang current_space_lvl->local_remaining -= nzi; 20419e25ed09SBarry Smith } 20425a8e39fbSHong Zhang 2043898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 2044898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 20455a8e39fbSHong Zhang 20465a8e39fbSHong Zhang /* destroy list of free space and other temporary arrays */ 2047*785e854fSJed Brown ierr = PetscMalloc1((bi[n]+1),&bj);CHKERRQ(ierr); 20486516239fSHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */ 20495a8e39fbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2050a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 20510e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 20529e25ed09SBarry Smith 20536cf91177SBarry Smith #if defined(PETSC_USE_INFO) 2054f64065bbSBarry Smith { 20555a8e39fbSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2056ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr); 2057ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2058ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr); 2059ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 2060335d9088SBarry Smith if (diagonal_fill) { 2061ae15b995SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr); 2062335d9088SBarry Smith } 2063f64065bbSBarry Smith } 206463ba0a88SBarry Smith #endif 2065bbb6d6a8SBarry Smith 20669e25ed09SBarry Smith /* put together the new matrix */ 20670298fd71SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 20683bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)fact,(PetscObject)isicol);CHKERRQ(ierr); 2069719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 20702205254eSKarl Rupp 2071e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2072e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 20737c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 20742205254eSKarl Rupp 2075*785e854fSJed Brown ierr = PetscMalloc1(bi[n],&b->a);CHKERRQ(ierr); 20765a8e39fbSHong Zhang b->j = bj; 20775a8e39fbSHong Zhang b->i = bi; 20785a8e39fbSHong Zhang for (i=0; i<n; i++) bdiag[i] += bi[i]; 20795a8e39fbSHong Zhang b->diag = bdiag; 2080416022c9SBarry Smith b->ilen = 0; 2081416022c9SBarry Smith b->imax = 0; 2082416022c9SBarry Smith b->row = isrow; 2083416022c9SBarry Smith b->col = iscol; 2084c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2085c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 208682bf6240SBarry Smith b->icol = isicol; 2087*785e854fSJed Brown ierr = PetscMalloc1((n+1),&b->solve_work);CHKERRQ(ierr); 2088416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 20895a8e39fbSHong Zhang Allocate bdiag, solve_work, new a, new j */ 20903bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 20915a8e39fbSHong Zhang b->maxnz = b->nz = bi[n]; 20922205254eSKarl Rupp 2093719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2094719d5645SBarry Smith (fact)->info.fill_ratio_given = f; 2095719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2096ad04f41aSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 2097f2344125SBarry Smith if (a->inode.size) { 2098f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 2099d3ac4fa3SBarry Smith } 21003a40ed3dSBarry Smith PetscFunctionReturn(0); 21019e25ed09SBarry Smith } 2102d5d45c9bSBarry Smith 2103a6175056SHong Zhang #undef __FUNCT__ 210435233d99SShri Abhyankar #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ" 210535233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 21065f44c854SHong Zhang { 21075f44c854SHong Zhang Mat C = B; 21085f44c854SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 21095f44c854SHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 21105f44c854SHong Zhang IS ip=b->row,iip = b->icol; 21115f44c854SHong Zhang PetscErrorCode ierr; 21125f44c854SHong Zhang const PetscInt *rip,*riip; 21135f44c854SHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp; 21145f44c854SHong Zhang PetscInt *ai=a->i,*aj=a->j; 21155f44c854SHong Zhang PetscInt k,jmin,jmax,*c2r,*il,col,nexti,ili,nz; 21165f44c854SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 2117ace3abfcSBarry Smith PetscBool perm_identity; 2118d90ac83dSHong Zhang FactorShiftCtx sctx; 211998a93161SHong Zhang PetscReal rs; 212098a93161SHong Zhang MatScalar d,*v; 212198a93161SHong Zhang 21225f44c854SHong Zhang PetscFunctionBegin; 212398a93161SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 2124d90ac83dSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 212598a93161SHong Zhang 2126f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 212798a93161SHong Zhang sctx.shift_top = info->zeropivot; 212898a93161SHong Zhang for (i=0; i<mbs; i++) { 212998a93161SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 213098a93161SHong Zhang d = (aa)[a->diag[i]]; 213198a93161SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 213298a93161SHong Zhang v = aa+ai[i]; 213398a93161SHong Zhang nz = ai[i+1] - ai[i]; 21342205254eSKarl Rupp for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]); 213598a93161SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 213698a93161SHong Zhang } 213798a93161SHong Zhang sctx.shift_top *= 1.1; 213898a93161SHong Zhang sctx.nshift_max = 5; 213998a93161SHong Zhang sctx.shift_lo = 0.; 214098a93161SHong Zhang sctx.shift_hi = 1.; 214198a93161SHong Zhang } 21425f44c854SHong Zhang 21435f44c854SHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 21445f44c854SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 21455f44c854SHong Zhang 21465f44c854SHong Zhang /* allocate working arrays 21475f44c854SHong Zhang c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col 21485f44c854SHong 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 21495f44c854SHong Zhang */ 2150dcca6d9dSJed Brown ierr = PetscMalloc3(mbs,&rtmp,mbs,&il,mbs,&c2r);CHKERRQ(ierr); 21515f44c854SHong Zhang 215298a93161SHong Zhang do { 215307b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 215498a93161SHong Zhang 2155c5546cabSHong Zhang for (i=0; i<mbs; i++) c2r[i] = mbs; 21562e987568SSatish Balay if (mbs) il[0] = 0; 21575f44c854SHong Zhang 21585f44c854SHong Zhang for (k = 0; k<mbs; k++) { 21595f44c854SHong Zhang /* zero rtmp */ 21605f44c854SHong Zhang nz = bi[k+1] - bi[k]; 21615f44c854SHong Zhang bjtmp = bj + bi[k]; 21625f44c854SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 21635f44c854SHong Zhang 21645f44c854SHong Zhang /* load in initial unfactored row */ 21655f44c854SHong Zhang bval = ba + bi[k]; 21665f44c854SHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 21675f44c854SHong Zhang for (j = jmin; j < jmax; j++) { 21685f44c854SHong Zhang col = riip[aj[j]]; 21695f44c854SHong Zhang if (col >= k) { /* only take upper triangular entry */ 21705f44c854SHong Zhang rtmp[col] = aa[j]; 21715f44c854SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 21725f44c854SHong Zhang } 21735f44c854SHong Zhang } 217498a93161SHong Zhang /* shift the diagonal of the matrix: ZeropivotApply() */ 217598a93161SHong Zhang rtmp[k] += sctx.shift_amount; /* shift the diagonal of the matrix */ 21765f44c854SHong Zhang 21775f44c854SHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 21785f44c854SHong Zhang dk = rtmp[k]; 21795f44c854SHong Zhang i = c2r[k]; /* first row to be added to k_th row */ 21805f44c854SHong Zhang 21815f44c854SHong Zhang while (i < k) { 21825f44c854SHong Zhang nexti = c2r[i]; /* next row to be added to k_th row */ 21835f44c854SHong Zhang 21845f44c854SHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 21855f44c854SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 21865f44c854SHong Zhang uikdi = -ba[ili]*ba[bdiag[i]]; /* diagonal(k) */ 21875f44c854SHong Zhang dk += uikdi*ba[ili]; /* update diag[k] */ 21885f44c854SHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 21895f44c854SHong Zhang 21905f44c854SHong Zhang /* add multiple of row i to k-th row */ 21915f44c854SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 21925f44c854SHong Zhang if (jmin < jmax) { 21935f44c854SHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 21945f44c854SHong Zhang /* update il and c2r for row i */ 21955f44c854SHong Zhang il[i] = jmin; 21965f44c854SHong Zhang j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i; 21975f44c854SHong Zhang } 21985f44c854SHong Zhang i = nexti; 21995f44c854SHong Zhang } 22005f44c854SHong Zhang 22015f44c854SHong Zhang /* copy data into U(k,:) */ 220298a93161SHong Zhang rs = 0.0; 22035f44c854SHong Zhang jmin = bi[k]; jmax = bi[k+1]-1; 22045f44c854SHong Zhang if (jmin < jmax) { 22055f44c854SHong Zhang for (j=jmin; j<jmax; j++) { 220698a93161SHong Zhang col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]); 22075f44c854SHong Zhang } 22085f44c854SHong Zhang /* add the k-th row into il and c2r */ 22095f44c854SHong Zhang il[k] = jmin; 22105f44c854SHong Zhang i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k; 22115f44c854SHong Zhang } 221298a93161SHong Zhang 221398a93161SHong Zhang /* MatPivotCheck() */ 221498a93161SHong Zhang sctx.rs = rs; 221598a93161SHong Zhang sctx.pv = dk; 2216d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 221707b50cabSHong Zhang if (sctx.newshift) break; 221898a93161SHong Zhang dk = sctx.pv; 221998a93161SHong Zhang 222098a93161SHong Zhang ba[bdiag[k]] = 1.0/dk; /* U(k,k) */ 222198a93161SHong Zhang } 222207b50cabSHong Zhang } while (sctx.newshift); 22235f44c854SHong Zhang 22240e83c824SBarry Smith ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr); 22255f44c854SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 22265f44c854SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 22275f44c854SHong Zhang 22285f44c854SHong Zhang ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 22295f44c854SHong Zhang if (perm_identity) { 223035233d99SShri Abhyankar B->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 223135233d99SShri Abhyankar B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 22322169352eSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering; 22332169352eSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering; 22345f44c854SHong Zhang } else { 223535233d99SShri Abhyankar B->ops->solve = MatSolve_SeqSBAIJ_1; 223635233d99SShri Abhyankar B->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 223735233d99SShri Abhyankar B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1; 223835233d99SShri Abhyankar B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1; 22395f44c854SHong Zhang } 22405f44c854SHong Zhang 22415f44c854SHong Zhang C->assembled = PETSC_TRUE; 22425f44c854SHong Zhang C->preallocated = PETSC_TRUE; 22432205254eSKarl Rupp 22445f44c854SHong Zhang ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 224598a93161SHong Zhang 224698a93161SHong Zhang /* MatPivotView() */ 224798a93161SHong Zhang if (sctx.nshift) { 2248f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 224998a93161SHong 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); 2250f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 225198a93161SHong Zhang ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2252f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) { 2253b2a402fcSHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr); 225498a93161SHong Zhang } 225598a93161SHong Zhang } 22565f44c854SHong Zhang PetscFunctionReturn(0); 22575f44c854SHong Zhang } 22585f44c854SHong Zhang 22595f44c854SHong Zhang #undef __FUNCT__ 22600a3c351aSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_inplace" 22610a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info) 2262a6175056SHong Zhang { 2263719d5645SBarry Smith Mat C = B; 22640968510aSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 22652ed133dbSHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 22669bfd6278SHong Zhang IS ip=b->row,iip = b->icol; 22672ed133dbSHong Zhang PetscErrorCode ierr; 22685d0c19d7SBarry Smith const PetscInt *rip,*riip; 2269c5546cabSHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp; 22702ed133dbSHong Zhang PetscInt *ai=a->i,*aj=a->j; 22712ed133dbSHong Zhang PetscInt k,jmin,jmax,*jl,*il,col,nexti,ili,nz; 2272622e2028SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 2273ace3abfcSBarry Smith PetscBool perm_identity; 22740e95ead3SHong Zhang FactorShiftCtx sctx; 22750e95ead3SHong Zhang PetscReal rs; 22760e95ead3SHong Zhang MatScalar d,*v; 2277d5d45c9bSBarry Smith 2278a6175056SHong Zhang PetscFunctionBegin; 22790e95ead3SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 22800e95ead3SHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 22810e95ead3SHong Zhang 22820e95ead3SHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 22830e95ead3SHong Zhang sctx.shift_top = info->zeropivot; 22840e95ead3SHong Zhang for (i=0; i<mbs; i++) { 22850e95ead3SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 22860e95ead3SHong Zhang d = (aa)[a->diag[i]]; 22870e95ead3SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 22880e95ead3SHong Zhang v = aa+ai[i]; 22890e95ead3SHong Zhang nz = ai[i+1] - ai[i]; 22902205254eSKarl Rupp for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]); 22910e95ead3SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 22920e95ead3SHong Zhang } 22930e95ead3SHong Zhang sctx.shift_top *= 1.1; 22940e95ead3SHong Zhang sctx.nshift_max = 5; 22950e95ead3SHong Zhang sctx.shift_lo = 0.; 22960e95ead3SHong Zhang sctx.shift_hi = 1.; 22970e95ead3SHong Zhang } 2298ee45ca4aSHong Zhang 22992ed133dbSHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 23009bfd6278SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 23012ed133dbSHong Zhang 23022ed133dbSHong Zhang /* initialization */ 2303dcca6d9dSJed Brown ierr = PetscMalloc3(mbs,&rtmp,mbs,&il,mbs,&jl);CHKERRQ(ierr); 23040e95ead3SHong Zhang 23052ed133dbSHong Zhang do { 230607b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 23070e95ead3SHong Zhang 2308c5546cabSHong Zhang for (i=0; i<mbs; i++) jl[i] = mbs; 2309c5546cabSHong Zhang il[0] = 0; 23102ed133dbSHong Zhang 23112ed133dbSHong Zhang for (k = 0; k<mbs; k++) { 2312c5546cabSHong Zhang /* zero rtmp */ 2313c5546cabSHong Zhang nz = bi[k+1] - bi[k]; 2314c5546cabSHong Zhang bjtmp = bj + bi[k]; 2315c5546cabSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 2316c5546cabSHong Zhang 2317c05c3958SHong Zhang bval = ba + bi[k]; 23182ed133dbSHong Zhang /* initialize k-th row by the perm[k]-th row of A */ 23192ed133dbSHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 23202ed133dbSHong Zhang for (j = jmin; j < jmax; j++) { 23219bfd6278SHong Zhang col = riip[aj[j]]; 23222ed133dbSHong Zhang if (col >= k) { /* only take upper triangular entry */ 23232ed133dbSHong Zhang rtmp[col] = aa[j]; 2324c05c3958SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 23252ed133dbSHong Zhang } 23262ed133dbSHong Zhang } 232739e3d630SHong Zhang /* shift the diagonal of the matrix */ 2328540703acSHong Zhang if (sctx.nshift) rtmp[k] += sctx.shift_amount; 23292ed133dbSHong Zhang 23302ed133dbSHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 23312ed133dbSHong Zhang dk = rtmp[k]; 23322ed133dbSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 23332ed133dbSHong Zhang 23342ed133dbSHong Zhang while (i < k) { 23352ed133dbSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 23362ed133dbSHong Zhang 23372ed133dbSHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 23382ed133dbSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 23392ed133dbSHong Zhang uikdi = -ba[ili]*ba[bi[i]]; /* diagonal(k) */ 23402ed133dbSHong Zhang dk += uikdi*ba[ili]; 23412ed133dbSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 23422ed133dbSHong Zhang 23432ed133dbSHong Zhang /* add multiple of row i to k-th row */ 23442ed133dbSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 23452ed133dbSHong Zhang if (jmin < jmax) { 23462ed133dbSHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 23472ed133dbSHong Zhang /* update il and jl for row i */ 23482ed133dbSHong Zhang il[i] = jmin; 23492ed133dbSHong Zhang j = bj[jmin]; jl[i] = jl[j]; jl[j] = i; 23502ed133dbSHong Zhang } 23512ed133dbSHong Zhang i = nexti; 23522ed133dbSHong Zhang } 23532ed133dbSHong Zhang 23540697b6caSHong Zhang /* shift the diagonals when zero pivot is detected */ 23550697b6caSHong Zhang /* compute rs=sum of abs(off-diagonal) */ 23560697b6caSHong Zhang rs = 0.0; 23573c9e8598SHong Zhang jmin = bi[k]+1; 23583c9e8598SHong Zhang nz = bi[k+1] - jmin; 23593c9e8598SHong Zhang bcol = bj + jmin; 236004fbf559SBarry Smith for (j=0; j<nz; j++) { 236104fbf559SBarry Smith rs += PetscAbsScalar(rtmp[bcol[j]]); 23623c9e8598SHong Zhang } 2363540703acSHong Zhang 2364540703acSHong Zhang sctx.rs = rs; 2365540703acSHong Zhang sctx.pv = dk; 2366d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,k);CHKERRQ(ierr); 236707b50cabSHong Zhang if (sctx.newshift) break; 23680e95ead3SHong Zhang dk = sctx.pv; 23693c9e8598SHong Zhang 23703c9e8598SHong Zhang /* copy data into U(k,:) */ 237139e3d630SHong Zhang ba[bi[k]] = 1.0/dk; /* U(k,k) */ 237239e3d630SHong Zhang jmin = bi[k]+1; jmax = bi[k+1]; 237339e3d630SHong Zhang if (jmin < jmax) { 237439e3d630SHong Zhang for (j=jmin; j<jmax; j++) { 2375c6ae9bfcSHong Zhang col = bj[j]; ba[j] = rtmp[col]; 23763c9e8598SHong Zhang } 237739e3d630SHong Zhang /* add the k-th row into il and jl */ 23783c9e8598SHong Zhang il[k] = jmin; 23793c9e8598SHong Zhang i = bj[jmin]; jl[k] = jl[i]; jl[i] = k; 23803c9e8598SHong Zhang } 23813c9e8598SHong Zhang } 238207b50cabSHong Zhang } while (sctx.newshift); 23830e95ead3SHong Zhang 23840e83c824SBarry Smith ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr); 238539e3d630SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 23869bfd6278SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 2387db4efbfdSBarry Smith 2388db4efbfdSBarry Smith ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 2389db4efbfdSBarry Smith if (perm_identity) { 23900a3c351aSHong Zhang B->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23910a3c351aSHong Zhang B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23920a3c351aSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23930a3c351aSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 2394db4efbfdSBarry Smith } else { 23950a3c351aSHong Zhang B->ops->solve = MatSolve_SeqSBAIJ_1_inplace; 23960a3c351aSHong Zhang B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_inplace; 23970a3c351aSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_inplace; 23980a3c351aSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_inplace; 2399db4efbfdSBarry Smith } 2400db4efbfdSBarry Smith 24013c9e8598SHong Zhang C->assembled = PETSC_TRUE; 24023c9e8598SHong Zhang C->preallocated = PETSC_TRUE; 24032205254eSKarl Rupp 2404d0f46423SBarry Smith ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 2405540703acSHong Zhang if (sctx.nshift) { 2406f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 24071e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2408f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 24091e2582c4SBarry Smith ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 24100697b6caSHong Zhang } 24113c9e8598SHong Zhang } 24123c9e8598SHong Zhang PetscFunctionReturn(0); 24133c9e8598SHong Zhang } 2414a6175056SHong Zhang 24158a76ed4fSHong Zhang /* 24168a76ed4fSHong Zhang icc() under revised new data structure. 24178a76ed4fSHong Zhang Factored arrays bj and ba are stored as 24188a76ed4fSHong Zhang U(0,:),...,U(i,:),U(n-1,:) 24198a76ed4fSHong Zhang 24208a76ed4fSHong Zhang ui=fact->i is an array of size n+1, in which 24218a76ed4fSHong Zhang ui+ 24228a76ed4fSHong Zhang ui[i]: points to 1st entry of U(i,:),i=0,...,n-1 24238a76ed4fSHong Zhang ui[n]: points to U(n-1,n-1)+1 24248a76ed4fSHong Zhang 24258a76ed4fSHong Zhang udiag=fact->diag is an array of size n,in which 24268a76ed4fSHong Zhang udiag[i]: points to diagonal of U(i,:), i=0,...,n-1 24278a76ed4fSHong Zhang 24288a76ed4fSHong Zhang U(i,:) contains udiag[i] as its last entry, i.e., 24298a76ed4fSHong Zhang U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 24308a76ed4fSHong Zhang */ 24318a76ed4fSHong Zhang 2432a6175056SHong Zhang #undef __FUNCT__ 243335233d99SShri Abhyankar #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ" 243435233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2435a6175056SHong Zhang { 24360968510aSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2437ed59401aSHong Zhang Mat_SeqSBAIJ *b; 2438dfbe8321SBarry Smith PetscErrorCode ierr; 2439ace3abfcSBarry Smith PetscBool perm_identity,missing; 24405f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 24415d0c19d7SBarry Smith const PetscInt *rip,*riip; 2442ed59401aSHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 24430298fd71SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=NULL,d; 24445a8e39fbSHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 2445ed59401aSHong Zhang PetscReal fill =info->fill,levels=info->levels; 24460298fd71SBarry Smith PetscFreeSpaceList free_space =NULL,current_space=NULL; 24470298fd71SBarry Smith PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL; 24487a48dd6fSHong Zhang PetscBT lnkbt; 2449b635d36bSHong Zhang IS iperm; 2450a6175056SHong Zhang 2451a6175056SHong Zhang PetscFunctionBegin; 2452e32f2f54SBarry 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); 245358ebbce7SBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 2454e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 2455653a6975SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 2456b635d36bSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 24577a48dd6fSHong Zhang 2458*785e854fSJed Brown ierr = PetscMalloc1((am+1),&ui);CHKERRQ(ierr); 2459*785e854fSJed Brown ierr = PetscMalloc1((am+1),&udiag);CHKERRQ(ierr); 246039e3d630SHong Zhang ui[0] = 0; 246139e3d630SHong Zhang 24628a76ed4fSHong Zhang /* ICC(0) without matrix ordering: simply rearrange column indices */ 2463622e2028SHong Zhang if (!levels && perm_identity) { 24645f44c854SHong Zhang for (i=0; i<am; i++) { 2465c5546cabSHong Zhang ncols = ai[i+1] - a->diag[i]; 2466c5546cabSHong Zhang ui[i+1] = ui[i] + ncols; 2467c5546cabSHong Zhang udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */ 2468dc1e2a3fSHong Zhang } 2469*785e854fSJed Brown ierr = PetscMalloc1((ui[am]+1),&uj);CHKERRQ(ierr); 2470dc1e2a3fSHong Zhang cols = uj; 2471dc1e2a3fSHong Zhang for (i=0; i<am; i++) { 24725f44c854SHong Zhang aj = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */ 2473dc1e2a3fSHong Zhang ncols = ai[i+1] - a->diag[i] -1; 24745f44c854SHong Zhang for (j=0; j<ncols; j++) *cols++ = aj[j]; 24755f44c854SHong Zhang *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */ 24765f44c854SHong Zhang } 24775f44c854SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 24785f44c854SHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 24795f44c854SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 24805f44c854SHong Zhang 24815f44c854SHong Zhang /* initialization */ 2482*785e854fSJed Brown ierr = PetscMalloc1((am+1),&ajtmp);CHKERRQ(ierr); 24835f44c854SHong Zhang 24845f44c854SHong Zhang /* jl: linked list for storing indices of the pivot rows 24855f44c854SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 2486dcca6d9dSJed Brown ierr = PetscMalloc4(am,&uj_ptr,am,&uj_lvl_ptr,am,&jl,am,&il);CHKERRQ(ierr); 24875f44c854SHong Zhang for (i=0; i<am; i++) { 24885f44c854SHong Zhang jl[i] = am; il[i] = 0; 24895f44c854SHong Zhang } 24905f44c854SHong Zhang 24915f44c854SHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 24925f44c854SHong Zhang nlnk = am + 1; 24935f44c854SHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 24945f44c854SHong Zhang 249595b5ac22SHong Zhang /* initial FreeSpace size is fill*(ai[am]+am)/2 */ 249695b5ac22SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space);CHKERRQ(ierr); 24975f44c854SHong Zhang current_space = free_space; 249895b5ac22SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space_lvl);CHKERRQ(ierr); 24995f44c854SHong Zhang current_space_lvl = free_space_lvl; 25005f44c854SHong Zhang 25015f44c854SHong Zhang for (k=0; k<am; k++) { /* for each active row k */ 25025f44c854SHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 25035f44c854SHong Zhang nzk = 0; 25045f44c854SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2505e32f2f54SBarry 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); 25065f44c854SHong Zhang ncols_upper = 0; 25075f44c854SHong Zhang for (j=0; j<ncols; j++) { 25085f44c854SHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 25095f44c854SHong Zhang if (riip[i] >= k) { /* only take upper triangular entry */ 25105f44c854SHong Zhang ajtmp[ncols_upper] = i; 25115f44c854SHong Zhang ncols_upper++; 25125f44c854SHong Zhang } 25135f44c854SHong Zhang } 25145f44c854SHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 25155f44c854SHong Zhang nzk += nlnk; 25165f44c854SHong Zhang 25175f44c854SHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 25185f44c854SHong Zhang prow = jl[k]; /* 1st pivot row */ 25195f44c854SHong Zhang 25205f44c854SHong Zhang while (prow < k) { 25215f44c854SHong Zhang nextprow = jl[prow]; 25225f44c854SHong Zhang 25235f44c854SHong Zhang /* merge prow into k-th row */ 25245f44c854SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 25255f44c854SHong Zhang jmax = ui[prow+1]; 25265f44c854SHong Zhang ncols = jmax-jmin; 25275f44c854SHong Zhang i = jmin - ui[prow]; 25285f44c854SHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 25295f44c854SHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 25305f44c854SHong Zhang j = *(uj - 1); 25315f44c854SHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 25325f44c854SHong Zhang nzk += nlnk; 25335f44c854SHong Zhang 25345f44c854SHong Zhang /* update il and jl for prow */ 25355f44c854SHong Zhang if (jmin < jmax) { 25365f44c854SHong Zhang il[prow] = jmin; 25375f44c854SHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 25385f44c854SHong Zhang } 25395f44c854SHong Zhang prow = nextprow; 25405f44c854SHong Zhang } 25415f44c854SHong Zhang 25425f44c854SHong Zhang /* if free space is not available, make more free space */ 25435f44c854SHong Zhang if (current_space->local_remaining<nzk) { 25445f44c854SHong Zhang i = am - k + 1; /* num of unfactored rows */ 2545c6ae9bfcSHong Zhang i *= PetscMin(nzk, i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */ 25465f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 25475f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 25485f44c854SHong Zhang reallocs++; 25495f44c854SHong Zhang } 25505f44c854SHong Zhang 25515f44c854SHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 2552e32f2f54SBarry Smith if (nzk == 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 25535f44c854SHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 25545f44c854SHong Zhang 25555f44c854SHong Zhang /* add the k-th row into il and jl */ 25565f44c854SHong Zhang if (nzk > 1) { 25575f44c854SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 25585f44c854SHong Zhang jl[k] = jl[i]; jl[i] = k; 25595f44c854SHong Zhang il[k] = ui[k] + 1; 25605f44c854SHong Zhang } 25615f44c854SHong Zhang uj_ptr[k] = current_space->array; 25625f44c854SHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 25635f44c854SHong Zhang 25645f44c854SHong Zhang current_space->array += nzk; 25655f44c854SHong Zhang current_space->local_used += nzk; 25665f44c854SHong Zhang current_space->local_remaining -= nzk; 25675f44c854SHong Zhang 25685f44c854SHong Zhang current_space_lvl->array += nzk; 25695f44c854SHong Zhang current_space_lvl->local_used += nzk; 25705f44c854SHong Zhang current_space_lvl->local_remaining -= nzk; 25715f44c854SHong Zhang 25725f44c854SHong Zhang ui[k+1] = ui[k] + nzk; 25735f44c854SHong Zhang } 25745f44c854SHong Zhang 25755f44c854SHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 25765f44c854SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 25770e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 25785f44c854SHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 25795f44c854SHong Zhang 25809263d837SHong Zhang /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */ 2581*785e854fSJed Brown ierr = PetscMalloc1((ui[am]+1),&uj);CHKERRQ(ierr); 258235233d99SShri Abhyankar ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */ 25835f44c854SHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 25845f44c854SHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 25855f44c854SHong Zhang 25865f44c854SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 25875f44c854SHong Zhang 25885f44c854SHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 25895f44c854SHong Zhang b = (Mat_SeqSBAIJ*)(fact)->data; 25905f44c854SHong Zhang b->singlemalloc = PETSC_FALSE; 25912205254eSKarl Rupp 2592*785e854fSJed Brown ierr = PetscMalloc1((ui[am]+1),&b->a);CHKERRQ(ierr); 25932205254eSKarl Rupp 25945f44c854SHong Zhang b->j = uj; 25955f44c854SHong Zhang b->i = ui; 25965f44c854SHong Zhang b->diag = udiag; 25977f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 25985f44c854SHong Zhang b->ilen = 0; 25995f44c854SHong Zhang b->imax = 0; 26005f44c854SHong Zhang b->row = perm; 26015f44c854SHong Zhang b->col = perm; 26025f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 26035f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 26045f44c854SHong Zhang b->icol = iperm; 26055f44c854SHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 26062205254eSKarl Rupp 2607*785e854fSJed Brown ierr = PetscMalloc1((am+1),&b->solve_work);CHKERRQ(ierr); 26083bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 26092205254eSKarl Rupp 26105f44c854SHong Zhang b->maxnz = b->nz = ui[am]; 26115f44c854SHong Zhang b->free_a = PETSC_TRUE; 26125f44c854SHong Zhang b->free_ij = PETSC_TRUE; 26135f44c854SHong Zhang 2614f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2615f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 26165f44c854SHong Zhang if (ai[am] != 0) { 26176a69fef8SHong Zhang /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */ 261895b5ac22SHong Zhang fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am); 26195f44c854SHong Zhang } else { 2620f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 26215f44c854SHong Zhang } 26229263d837SHong Zhang #if defined(PETSC_USE_INFO) 26239263d837SHong Zhang if (ai[am] != 0) { 26249263d837SHong Zhang PetscReal af = fact->info.fill_ratio_needed; 26259263d837SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 26269263d837SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 26279263d837SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 26289263d837SHong Zhang } else { 26299263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 26309263d837SHong Zhang } 26319263d837SHong Zhang #endif 263235233d99SShri Abhyankar fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 26335f44c854SHong Zhang PetscFunctionReturn(0); 26345f44c854SHong Zhang } 26355f44c854SHong Zhang 26365f44c854SHong Zhang #undef __FUNCT__ 2637ad04f41aSHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_inplace" 2638ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 26395f44c854SHong Zhang { 26405f44c854SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 26415f44c854SHong Zhang Mat_SeqSBAIJ *b; 26425f44c854SHong Zhang PetscErrorCode ierr; 2643ace3abfcSBarry Smith PetscBool perm_identity,missing; 26445f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 26455f44c854SHong Zhang const PetscInt *rip,*riip; 26465f44c854SHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 26470298fd71SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=NULL,d; 26485f44c854SHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 26495f44c854SHong Zhang PetscReal fill =info->fill,levels=info->levels; 26500298fd71SBarry Smith PetscFreeSpaceList free_space =NULL,current_space=NULL; 26510298fd71SBarry Smith PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL; 26525f44c854SHong Zhang PetscBT lnkbt; 26535f44c854SHong Zhang IS iperm; 26545f44c854SHong Zhang 26555f44c854SHong Zhang PetscFunctionBegin; 2656e32f2f54SBarry 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); 26575f44c854SHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 2658e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 26595f44c854SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 26605f44c854SHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 26615f44c854SHong Zhang 2662*785e854fSJed Brown ierr = PetscMalloc1((am+1),&ui);CHKERRQ(ierr); 2663*785e854fSJed Brown ierr = PetscMalloc1((am+1),&udiag);CHKERRQ(ierr); 26645f44c854SHong Zhang ui[0] = 0; 26655f44c854SHong Zhang 26665f44c854SHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 26675f44c854SHong Zhang if (!levels && perm_identity) { 26685f44c854SHong Zhang 26695f44c854SHong Zhang for (i=0; i<am; i++) { 26705f44c854SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 26715f44c854SHong Zhang udiag[i] = ui[i]; 2672ed59401aSHong Zhang } 2673*785e854fSJed Brown ierr = PetscMalloc1((ui[am]+1),&uj);CHKERRQ(ierr); 267439e3d630SHong Zhang cols = uj; 2675ed59401aSHong Zhang for (i=0; i<am; i++) { 2676ed59401aSHong Zhang aj = a->j + a->diag[i]; 267739e3d630SHong Zhang ncols = ui[i+1] - ui[i]; 267839e3d630SHong Zhang for (j=0; j<ncols; j++) *cols++ = *aj++; 2679ed59401aSHong Zhang } 268039e3d630SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 2681b635d36bSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 2682b635d36bSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 2683b635d36bSHong Zhang 26847a48dd6fSHong Zhang /* initialization */ 2685*785e854fSJed Brown ierr = PetscMalloc1((am+1),&ajtmp);CHKERRQ(ierr); 26867a48dd6fSHong Zhang 26877a48dd6fSHong Zhang /* jl: linked list for storing indices of the pivot rows 26887a48dd6fSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 2689dcca6d9dSJed Brown ierr = PetscMalloc4(am,&uj_ptr,am,&uj_lvl_ptr,am,&jl,am,&il);CHKERRQ(ierr); 26907a48dd6fSHong Zhang for (i=0; i<am; i++) { 26917a48dd6fSHong Zhang jl[i] = am; il[i] = 0; 26927a48dd6fSHong Zhang } 26937a48dd6fSHong Zhang 26947a48dd6fSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 26957a48dd6fSHong Zhang nlnk = am + 1; 26962ed133dbSHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 26977a48dd6fSHong Zhang 26987a48dd6fSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 2699a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 27007a48dd6fSHong Zhang current_space = free_space; 2701a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr); 27027a48dd6fSHong Zhang current_space_lvl = free_space_lvl; 27037a48dd6fSHong Zhang 27047a48dd6fSHong Zhang for (k=0; k<am; k++) { /* for each active row k */ 27057a48dd6fSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 27067a48dd6fSHong Zhang nzk = 0; 2707622e2028SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2708e32f2f54SBarry 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); 2709622e2028SHong Zhang ncols_upper = 0; 2710622e2028SHong Zhang for (j=0; j<ncols; j++) { 2711b635d36bSHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 2712b635d36bSHong Zhang if (riip[i] >= k) { /* only take upper triangular entry */ 27135a8e39fbSHong Zhang ajtmp[ncols_upper] = i; 2714622e2028SHong Zhang ncols_upper++; 2715622e2028SHong Zhang } 2716622e2028SHong Zhang } 2717b635d36bSHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 27187a48dd6fSHong Zhang nzk += nlnk; 27197a48dd6fSHong Zhang 27207a48dd6fSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 27217a48dd6fSHong Zhang prow = jl[k]; /* 1st pivot row */ 27227a48dd6fSHong Zhang 27237a48dd6fSHong Zhang while (prow < k) { 27247a48dd6fSHong Zhang nextprow = jl[prow]; 27257a48dd6fSHong Zhang 27267a48dd6fSHong Zhang /* merge prow into k-th row */ 27277a48dd6fSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 27287a48dd6fSHong Zhang jmax = ui[prow+1]; 27297a48dd6fSHong Zhang ncols = jmax-jmin; 2730ed59401aSHong Zhang i = jmin - ui[prow]; 2731ed59401aSHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 27325a8e39fbSHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 27335a8e39fbSHong Zhang j = *(uj - 1); 27345a8e39fbSHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 27357a48dd6fSHong Zhang nzk += nlnk; 27367a48dd6fSHong Zhang 27377a48dd6fSHong Zhang /* update il and jl for prow */ 27387a48dd6fSHong Zhang if (jmin < jmax) { 27397a48dd6fSHong Zhang il[prow] = jmin; 27407a48dd6fSHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 27417a48dd6fSHong Zhang } 27427a48dd6fSHong Zhang prow = nextprow; 27437a48dd6fSHong Zhang } 27447a48dd6fSHong Zhang 27457a48dd6fSHong Zhang /* if free space is not available, make more free space */ 27467a48dd6fSHong Zhang if (current_space->local_remaining<nzk) { 27477a48dd6fSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2748fd3f55cbSMatthew Knepley i *= PetscMin(nzk, (i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2749a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 2750a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 27517a48dd6fSHong Zhang reallocs++; 27527a48dd6fSHong Zhang } 27537a48dd6fSHong Zhang 27547a48dd6fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 275565e19b50SBarry Smith if (!nzk) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 27562ed133dbSHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 27577a48dd6fSHong Zhang 27587a48dd6fSHong Zhang /* add the k-th row into il and jl */ 275939e3d630SHong Zhang if (nzk > 1) { 27607a48dd6fSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 27617a48dd6fSHong Zhang jl[k] = jl[i]; jl[i] = k; 27627a48dd6fSHong Zhang il[k] = ui[k] + 1; 27637a48dd6fSHong Zhang } 27647a48dd6fSHong Zhang uj_ptr[k] = current_space->array; 27657a48dd6fSHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 27667a48dd6fSHong Zhang 27677a48dd6fSHong Zhang current_space->array += nzk; 27687a48dd6fSHong Zhang current_space->local_used += nzk; 27697a48dd6fSHong Zhang current_space->local_remaining -= nzk; 27707a48dd6fSHong Zhang 27717a48dd6fSHong Zhang current_space_lvl->array += nzk; 27727a48dd6fSHong Zhang current_space_lvl->local_used += nzk; 27737a48dd6fSHong Zhang current_space_lvl->local_remaining -= nzk; 27747a48dd6fSHong Zhang 27757a48dd6fSHong Zhang ui[k+1] = ui[k] + nzk; 27767a48dd6fSHong Zhang } 27777a48dd6fSHong Zhang 27786cf91177SBarry Smith #if defined(PETSC_USE_INFO) 27797a48dd6fSHong Zhang if (ai[am] != 0) { 278039e3d630SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 2781ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 2782ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 2783ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 27847a48dd6fSHong Zhang } else { 2785ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 27867a48dd6fSHong Zhang } 278763ba0a88SBarry Smith #endif 27887a48dd6fSHong Zhang 27897a48dd6fSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 2790b635d36bSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 27910e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 27925a8e39fbSHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 27937a48dd6fSHong Zhang 27947a48dd6fSHong Zhang /* destroy list of free space and other temporary array(s) */ 2795*785e854fSJed Brown ierr = PetscMalloc1((ui[am]+1),&uj);CHKERRQ(ierr); 2796a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 27972ed133dbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2798a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 27997a48dd6fSHong Zhang 280039e3d630SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 280139e3d630SHong Zhang 28027a48dd6fSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 28037a48dd6fSHong Zhang 2804f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 28057a48dd6fSHong Zhang b->singlemalloc = PETSC_FALSE; 28062205254eSKarl Rupp 2807*785e854fSJed Brown ierr = PetscMalloc1((ui[am]+1),&b->a);CHKERRQ(ierr); 28082205254eSKarl Rupp 28097a48dd6fSHong Zhang b->j = uj; 28107a48dd6fSHong Zhang b->i = ui; 28115f44c854SHong Zhang b->diag = udiag; 28127f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 28137a48dd6fSHong Zhang b->ilen = 0; 28147a48dd6fSHong Zhang b->imax = 0; 28157a48dd6fSHong Zhang b->row = perm; 2816b635d36bSHong Zhang b->col = perm; 28172205254eSKarl Rupp 2818b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2819b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 28202205254eSKarl Rupp 2821b635d36bSHong Zhang b->icol = iperm; 28227a48dd6fSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 2823*785e854fSJed Brown ierr = PetscMalloc1((am+1),&b->solve_work);CHKERRQ(ierr); 28243bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 28257a48dd6fSHong Zhang b->maxnz = b->nz = ui[am]; 2826e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2827e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 28287a48dd6fSHong Zhang 2829f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2830f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 28317a48dd6fSHong Zhang if (ai[am] != 0) { 2832f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 28337a48dd6fSHong Zhang } else { 2834f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 28357a48dd6fSHong Zhang } 28360a3c351aSHong Zhang fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace; 2837a6175056SHong Zhang PetscFunctionReturn(0); 2838a6175056SHong Zhang } 2839d5d45c9bSBarry Smith 2840177d4faaSHong Zhang #undef __FUNCT__ 2841177d4faaSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ" 284235233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 28430be760fbSHong Zhang { 28440be760fbSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 28450be760fbSHong Zhang Mat_SeqSBAIJ *b; 28460be760fbSHong Zhang PetscErrorCode ierr; 2847ace3abfcSBarry Smith PetscBool perm_identity; 28480be760fbSHong Zhang PetscReal fill = info->fill; 28490be760fbSHong Zhang const PetscInt *rip,*riip; 28500be760fbSHong Zhang PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 28510be760fbSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 28520be760fbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag; 28530298fd71SBarry Smith PetscFreeSpaceList free_space=NULL,current_space=NULL; 28540be760fbSHong Zhang PetscBT lnkbt; 28550be760fbSHong Zhang IS iperm; 28560be760fbSHong Zhang 28570be760fbSHong Zhang PetscFunctionBegin; 2858e32f2f54SBarry 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); 28590be760fbSHong Zhang /* check whether perm is the identity mapping */ 28600be760fbSHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 28610be760fbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 28620be760fbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 28630be760fbSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 28640be760fbSHong Zhang 28650be760fbSHong Zhang /* initialization */ 2866*785e854fSJed Brown ierr = PetscMalloc1((am+1),&ui);CHKERRQ(ierr); 2867*785e854fSJed Brown ierr = PetscMalloc1((am+1),&udiag);CHKERRQ(ierr); 28680be760fbSHong Zhang ui[0] = 0; 28690be760fbSHong Zhang 28700be760fbSHong Zhang /* jl: linked list for storing indices of the pivot rows 28710be760fbSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 2872dcca6d9dSJed Brown ierr = PetscMalloc4(am,&ui_ptr,am,&jl,am,&il,am,&cols);CHKERRQ(ierr); 28730be760fbSHong Zhang for (i=0; i<am; i++) { 28740be760fbSHong Zhang jl[i] = am; il[i] = 0; 28750be760fbSHong Zhang } 28760be760fbSHong Zhang 28770be760fbSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 28780be760fbSHong Zhang nlnk = am + 1; 28790be760fbSHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28800be760fbSHong Zhang 288195b5ac22SHong Zhang /* initial FreeSpace size is fill*(ai[am]+am)/2 */ 288295b5ac22SHong Zhang ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space);CHKERRQ(ierr); 28830be760fbSHong Zhang current_space = free_space; 28840be760fbSHong Zhang 28850be760fbSHong Zhang for (k=0; k<am; k++) { /* for each active row k */ 28860be760fbSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 28870be760fbSHong Zhang nzk = 0; 28880be760fbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2889e32f2f54SBarry 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); 28900be760fbSHong Zhang ncols_upper = 0; 28910be760fbSHong Zhang for (j=0; j<ncols; j++) { 28920be760fbSHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 28930be760fbSHong Zhang if (i >= k) { /* only take upper triangular entry */ 28940be760fbSHong Zhang cols[ncols_upper] = i; 28950be760fbSHong Zhang ncols_upper++; 28960be760fbSHong Zhang } 28970be760fbSHong Zhang } 28980be760fbSHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28990be760fbSHong Zhang nzk += nlnk; 29000be760fbSHong Zhang 29010be760fbSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 29020be760fbSHong Zhang prow = jl[k]; /* 1st pivot row */ 29030be760fbSHong Zhang 29040be760fbSHong Zhang while (prow < k) { 29050be760fbSHong Zhang nextprow = jl[prow]; 29060be760fbSHong Zhang /* merge prow into k-th row */ 29070be760fbSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 29080be760fbSHong Zhang jmax = ui[prow+1]; 29090be760fbSHong Zhang ncols = jmax-jmin; 29100be760fbSHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 29110be760fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 29120be760fbSHong Zhang nzk += nlnk; 29130be760fbSHong Zhang 29140be760fbSHong Zhang /* update il and jl for prow */ 29150be760fbSHong Zhang if (jmin < jmax) { 29160be760fbSHong Zhang il[prow] = jmin; 29172205254eSKarl Rupp j = *uj_ptr; 29182205254eSKarl Rupp jl[prow] = jl[j]; 29192205254eSKarl Rupp jl[j] = prow; 29200be760fbSHong Zhang } 29210be760fbSHong Zhang prow = nextprow; 29220be760fbSHong Zhang } 29230be760fbSHong Zhang 29240be760fbSHong Zhang /* if free space is not available, make more free space */ 29250be760fbSHong Zhang if (current_space->local_remaining<nzk) { 29260be760fbSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2927c6ae9bfcSHong Zhang i *= PetscMin(nzk,i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */ 29280be760fbSHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 29290be760fbSHong Zhang reallocs++; 29300be760fbSHong Zhang } 29310be760fbSHong Zhang 29320be760fbSHong Zhang /* copy data into free space, then initialize lnk */ 29330be760fbSHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 29340be760fbSHong Zhang 29350be760fbSHong Zhang /* add the k-th row into il and jl */ 29367b056e98SHong Zhang if (nzk > 1) { 29370be760fbSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 29380be760fbSHong Zhang jl[k] = jl[i]; jl[i] = k; 29390be760fbSHong Zhang il[k] = ui[k] + 1; 29400be760fbSHong Zhang } 29410be760fbSHong Zhang ui_ptr[k] = current_space->array; 29422205254eSKarl Rupp 29430be760fbSHong Zhang current_space->array += nzk; 29440be760fbSHong Zhang current_space->local_used += nzk; 29450be760fbSHong Zhang current_space->local_remaining -= nzk; 29460be760fbSHong Zhang 29470be760fbSHong Zhang ui[k+1] = ui[k] + nzk; 29480be760fbSHong Zhang } 29490be760fbSHong Zhang 29500be760fbSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 29510be760fbSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 29520be760fbSHong Zhang ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 29530be760fbSHong Zhang 29549263d837SHong Zhang /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */ 2955*785e854fSJed Brown ierr = PetscMalloc1((ui[am]+1),&uj);CHKERRQ(ierr); 295635233d99SShri Abhyankar ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */ 29570be760fbSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 29580be760fbSHong Zhang 29590be760fbSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 29600be760fbSHong Zhang 2961f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 29620be760fbSHong Zhang b->singlemalloc = PETSC_FALSE; 29630be760fbSHong Zhang b->free_a = PETSC_TRUE; 29640be760fbSHong Zhang b->free_ij = PETSC_TRUE; 29652205254eSKarl Rupp 2966*785e854fSJed Brown ierr = PetscMalloc1((ui[am]+1),&b->a);CHKERRQ(ierr); 29672205254eSKarl Rupp 29680be760fbSHong Zhang b->j = uj; 29690be760fbSHong Zhang b->i = ui; 29700be760fbSHong Zhang b->diag = udiag; 29710be760fbSHong Zhang b->free_diag = PETSC_TRUE; 29720be760fbSHong Zhang b->ilen = 0; 29730be760fbSHong Zhang b->imax = 0; 29740be760fbSHong Zhang b->row = perm; 29750be760fbSHong Zhang b->col = perm; 297626fbe8dcSKarl Rupp 29770be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29780be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29792205254eSKarl Rupp 29800be760fbSHong Zhang b->icol = iperm; 29810be760fbSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 298226fbe8dcSKarl Rupp 2983*785e854fSJed Brown ierr = PetscMalloc1((am+1),&b->solve_work);CHKERRQ(ierr); 29843bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 29852205254eSKarl Rupp 29860be760fbSHong Zhang b->maxnz = b->nz = ui[am]; 29870be760fbSHong Zhang 2988f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2989f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 29900be760fbSHong Zhang if (ai[am] != 0) { 29916a69fef8SHong Zhang /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */ 299295b5ac22SHong Zhang fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am); 29930be760fbSHong Zhang } else { 2994f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 29950be760fbSHong Zhang } 29969263d837SHong Zhang #if defined(PETSC_USE_INFO) 29979263d837SHong Zhang if (ai[am] != 0) { 29989263d837SHong Zhang PetscReal af = fact->info.fill_ratio_needed; 29999263d837SHong Zhang ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 30009263d837SHong Zhang ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 30019263d837SHong Zhang ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 30029263d837SHong Zhang } else { 30039263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 30049263d837SHong Zhang } 30059263d837SHong Zhang #endif 300635233d99SShri Abhyankar fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 30070be760fbSHong Zhang PetscFunctionReturn(0); 30080be760fbSHong Zhang } 30090be760fbSHong Zhang 3010f76d2b81SHong Zhang #undef __FUNCT__ 3011ad04f41aSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ_inplace" 3012ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 3013f76d2b81SHong Zhang { 3014f76d2b81SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 301510c27e3dSHong Zhang Mat_SeqSBAIJ *b; 3016dfbe8321SBarry Smith PetscErrorCode ierr; 3017ace3abfcSBarry Smith PetscBool perm_identity; 301810c27e3dSHong Zhang PetscReal fill = info->fill; 30195d0c19d7SBarry Smith const PetscInt *rip,*riip; 30205d0c19d7SBarry Smith PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 302110c27e3dSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 30222ed133dbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr; 30230298fd71SBarry Smith PetscFreeSpaceList free_space=NULL,current_space=NULL; 302410c27e3dSHong Zhang PetscBT lnkbt; 30252ed133dbSHong Zhang IS iperm; 3026f76d2b81SHong Zhang 3027f76d2b81SHong Zhang PetscFunctionBegin; 3028e32f2f54SBarry 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); 30292ed133dbSHong Zhang /* check whether perm is the identity mapping */ 3030f76d2b81SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 30312ed133dbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 30322ed133dbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 30339bfd6278SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 303410c27e3dSHong Zhang 303510c27e3dSHong Zhang /* initialization */ 3036*785e854fSJed Brown ierr = PetscMalloc1((am+1),&ui);CHKERRQ(ierr); 303710c27e3dSHong Zhang ui[0] = 0; 303810c27e3dSHong Zhang 303910c27e3dSHong Zhang /* jl: linked list for storing indices of the pivot rows 30401393bc97SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 3041dcca6d9dSJed Brown ierr = PetscMalloc4(am,&ui_ptr,am,&jl,am,&il,am,&cols);CHKERRQ(ierr); 30421393bc97SHong Zhang for (i=0; i<am; i++) { 30431393bc97SHong Zhang jl[i] = am; il[i] = 0; 3044f76d2b81SHong Zhang } 304510c27e3dSHong Zhang 304610c27e3dSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 30471393bc97SHong Zhang nlnk = am + 1; 30481393bc97SHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 304910c27e3dSHong Zhang 30501393bc97SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 3051a1a86e44SBarry Smith ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr); 305210c27e3dSHong Zhang current_space = free_space; 305310c27e3dSHong Zhang 30541393bc97SHong Zhang for (k=0; k<am; k++) { /* for each active row k */ 305510c27e3dSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 305610c27e3dSHong Zhang nzk = 0; 30572ed133dbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 3058e32f2f54SBarry 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); 30592ed133dbSHong Zhang ncols_upper = 0; 30602ed133dbSHong Zhang for (j=0; j<ncols; j++) { 30619bfd6278SHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 30622ed133dbSHong Zhang if (i >= k) { /* only take upper triangular entry */ 30632ed133dbSHong Zhang cols[ncols_upper] = i; 30642ed133dbSHong Zhang ncols_upper++; 30652ed133dbSHong Zhang } 30662ed133dbSHong Zhang } 30671393bc97SHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 306810c27e3dSHong Zhang nzk += nlnk; 306910c27e3dSHong Zhang 307010c27e3dSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 307110c27e3dSHong Zhang prow = jl[k]; /* 1st pivot row */ 307210c27e3dSHong Zhang 307310c27e3dSHong Zhang while (prow < k) { 307410c27e3dSHong Zhang nextprow = jl[prow]; 307510c27e3dSHong Zhang /* merge prow into k-th row */ 30761393bc97SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 307710c27e3dSHong Zhang jmax = ui[prow+1]; 307810c27e3dSHong Zhang ncols = jmax-jmin; 30791393bc97SHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 30805a8e39fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 308110c27e3dSHong Zhang nzk += nlnk; 308210c27e3dSHong Zhang 308310c27e3dSHong Zhang /* update il and jl for prow */ 308410c27e3dSHong Zhang if (jmin < jmax) { 308510c27e3dSHong Zhang il[prow] = jmin; 30862ed133dbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 308710c27e3dSHong Zhang } 308810c27e3dSHong Zhang prow = nextprow; 308910c27e3dSHong Zhang } 309010c27e3dSHong Zhang 309110c27e3dSHong Zhang /* if free space is not available, make more free space */ 309210c27e3dSHong Zhang if (current_space->local_remaining<nzk) { 30931393bc97SHong Zhang i = am - k + 1; /* num of unfactored rows */ 309410c27e3dSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 3095a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 309610c27e3dSHong Zhang reallocs++; 309710c27e3dSHong Zhang } 309810c27e3dSHong Zhang 309910c27e3dSHong Zhang /* copy data into free space, then initialize lnk */ 31001393bc97SHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 310110c27e3dSHong Zhang 310210c27e3dSHong Zhang /* add the k-th row into il and jl */ 310310c27e3dSHong Zhang if (nzk-1 > 0) { 31041393bc97SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 310510c27e3dSHong Zhang jl[k] = jl[i]; jl[i] = k; 310610c27e3dSHong Zhang il[k] = ui[k] + 1; 310710c27e3dSHong Zhang } 31082ed133dbSHong Zhang ui_ptr[k] = current_space->array; 31092205254eSKarl Rupp 311010c27e3dSHong Zhang current_space->array += nzk; 311110c27e3dSHong Zhang current_space->local_used += nzk; 311210c27e3dSHong Zhang current_space->local_remaining -= nzk; 311310c27e3dSHong Zhang 311410c27e3dSHong Zhang ui[k+1] = ui[k] + nzk; 311510c27e3dSHong Zhang } 311610c27e3dSHong Zhang 31176cf91177SBarry Smith #if defined(PETSC_USE_INFO) 31181393bc97SHong Zhang if (ai[am] != 0) { 31191393bc97SHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 3120ae15b995SBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr); 3121ae15b995SBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr); 3122ae15b995SBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr); 312310c27e3dSHong Zhang } else { 3124ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 312510c27e3dSHong Zhang } 312663ba0a88SBarry Smith #endif 312710c27e3dSHong Zhang 312810c27e3dSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 31299bfd6278SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 31300e83c824SBarry Smith ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 313110c27e3dSHong Zhang 313210c27e3dSHong Zhang /* destroy list of free space and other temporary array(s) */ 3133*785e854fSJed Brown ierr = PetscMalloc1((ui[am]+1),&uj);CHKERRQ(ierr); 3134a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 313510c27e3dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 313610c27e3dSHong Zhang 313710c27e3dSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 313810c27e3dSHong Zhang 3139f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 314010c27e3dSHong Zhang b->singlemalloc = PETSC_FALSE; 3141e6b907acSBarry Smith b->free_a = PETSC_TRUE; 3142e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 31432205254eSKarl Rupp 3144*785e854fSJed Brown ierr = PetscMalloc1((ui[am]+1),&b->a);CHKERRQ(ierr); 31452205254eSKarl Rupp 314610c27e3dSHong Zhang b->j = uj; 314710c27e3dSHong Zhang b->i = ui; 314810c27e3dSHong Zhang b->diag = 0; 314910c27e3dSHong Zhang b->ilen = 0; 315010c27e3dSHong Zhang b->imax = 0; 315110c27e3dSHong Zhang b->row = perm; 31529bfd6278SHong Zhang b->col = perm; 31532205254eSKarl Rupp 31549bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 31559bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 31562205254eSKarl Rupp 31579bfd6278SHong Zhang b->icol = iperm; 315810c27e3dSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 31592205254eSKarl Rupp 3160*785e854fSJed Brown ierr = PetscMalloc1((am+1),&b->solve_work);CHKERRQ(ierr); 31613bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 31621393bc97SHong Zhang b->maxnz = b->nz = ui[am]; 316310c27e3dSHong Zhang 3164f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 3165f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 31661393bc97SHong Zhang if (ai[am] != 0) { 3167f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 316810c27e3dSHong Zhang } else { 3169f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 317010c27e3dSHong Zhang } 31710a3c351aSHong Zhang fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace; 3172f76d2b81SHong Zhang PetscFunctionReturn(0); 3173f76d2b81SHong Zhang } 3174599ef60dSHong Zhang 31751d098868SHong Zhang #undef __FUNCT__ 317635233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering" 317735233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 3178813a1f2bSShri Abhyankar { 3179813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3180813a1f2bSShri Abhyankar PetscErrorCode ierr; 3181813a1f2bSShri Abhyankar PetscInt n = A->rmap->n; 3182813a1f2bSShri Abhyankar const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag,*vi; 3183813a1f2bSShri Abhyankar PetscScalar *x,sum; 3184813a1f2bSShri Abhyankar const PetscScalar *b; 3185813a1f2bSShri Abhyankar const MatScalar *aa = a->a,*v; 3186813a1f2bSShri Abhyankar PetscInt i,nz; 3187813a1f2bSShri Abhyankar 3188813a1f2bSShri Abhyankar PetscFunctionBegin; 3189813a1f2bSShri Abhyankar if (!n) PetscFunctionReturn(0); 3190813a1f2bSShri Abhyankar 31913649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 3192813a1f2bSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3193813a1f2bSShri Abhyankar 3194813a1f2bSShri Abhyankar /* forward solve the lower triangular */ 3195813a1f2bSShri Abhyankar x[0] = b[0]; 3196813a1f2bSShri Abhyankar v = aa; 3197813a1f2bSShri Abhyankar vi = aj; 3198813a1f2bSShri Abhyankar for (i=1; i<n; i++) { 3199813a1f2bSShri Abhyankar nz = ai[i+1] - ai[i]; 3200813a1f2bSShri Abhyankar sum = b[i]; 3201813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 3202813a1f2bSShri Abhyankar v += nz; 3203813a1f2bSShri Abhyankar vi += nz; 3204813a1f2bSShri Abhyankar x[i] = sum; 3205813a1f2bSShri Abhyankar } 3206813a1f2bSShri Abhyankar 3207813a1f2bSShri Abhyankar /* backward solve the upper triangular */ 320862fbd6afSShri Abhyankar for (i=n-1; i>=0; i--) { 32093c0119dfSShri Abhyankar v = aa + adiag[i+1] + 1; 32103c0119dfSShri Abhyankar vi = aj + adiag[i+1] + 1; 3211813a1f2bSShri Abhyankar nz = adiag[i] - adiag[i+1]-1; 3212813a1f2bSShri Abhyankar sum = x[i]; 3213813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 32143c0119dfSShri Abhyankar x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */ 3215813a1f2bSShri Abhyankar } 3216813a1f2bSShri Abhyankar 3217813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 32183649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3219813a1f2bSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3220813a1f2bSShri Abhyankar PetscFunctionReturn(0); 3221813a1f2bSShri Abhyankar } 3222813a1f2bSShri Abhyankar 32231d098868SHong Zhang #undef __FUNCT__ 322435233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ" 322535233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 3226f268cba8SShri Abhyankar { 3227f268cba8SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3228f268cba8SShri Abhyankar IS iscol = a->col,isrow = a->row; 3229f268cba8SShri Abhyankar PetscErrorCode ierr; 3230f268cba8SShri Abhyankar PetscInt i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz; 3231f268cba8SShri Abhyankar const PetscInt *rout,*cout,*r,*c; 3232f268cba8SShri Abhyankar PetscScalar *x,*tmp,sum; 3233f268cba8SShri Abhyankar const PetscScalar *b; 3234f268cba8SShri Abhyankar const MatScalar *aa = a->a,*v; 3235f268cba8SShri Abhyankar 3236f268cba8SShri Abhyankar PetscFunctionBegin; 3237f268cba8SShri Abhyankar if (!n) PetscFunctionReturn(0); 3238f268cba8SShri Abhyankar 32393649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 3240f268cba8SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3241f268cba8SShri Abhyankar tmp = a->solve_work; 3242f268cba8SShri Abhyankar 3243f268cba8SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 3244f268cba8SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 3245f268cba8SShri Abhyankar 3246f268cba8SShri Abhyankar /* forward solve the lower triangular */ 3247f268cba8SShri Abhyankar tmp[0] = b[r[0]]; 3248f268cba8SShri Abhyankar v = aa; 3249f268cba8SShri Abhyankar vi = aj; 3250f268cba8SShri Abhyankar for (i=1; i<n; i++) { 3251f268cba8SShri Abhyankar nz = ai[i+1] - ai[i]; 3252f268cba8SShri Abhyankar sum = b[r[i]]; 3253f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3254f268cba8SShri Abhyankar tmp[i] = sum; 3255f268cba8SShri Abhyankar v += nz; vi += nz; 3256f268cba8SShri Abhyankar } 3257f268cba8SShri Abhyankar 3258f268cba8SShri Abhyankar /* backward solve the upper triangular */ 3259f268cba8SShri Abhyankar for (i=n-1; i>=0; i--) { 32603c0119dfSShri Abhyankar v = aa + adiag[i+1]+1; 32613c0119dfSShri Abhyankar vi = aj + adiag[i+1]+1; 3262f268cba8SShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 3263f268cba8SShri Abhyankar sum = tmp[i]; 3264f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3265f268cba8SShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 3266f268cba8SShri Abhyankar } 3267f268cba8SShri Abhyankar 3268f268cba8SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 3269f268cba8SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 32703649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3271f268cba8SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3272f268cba8SShri Abhyankar ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 3273f268cba8SShri Abhyankar PetscFunctionReturn(0); 3274f268cba8SShri Abhyankar } 3275f268cba8SShri Abhyankar 3276f268cba8SShri Abhyankar #undef __FUNCT__ 32771d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ" 3278fe97e370SBarry Smith /* 3279fe97e370SBarry 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 3280fe97e370SBarry Smith */ 32811d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact) 32821d098868SHong Zhang { 32831d098868SHong Zhang Mat B = *fact; 3284599ef60dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b; 3285599ef60dSHong Zhang IS isicol; 3286599ef60dSHong Zhang PetscErrorCode ierr; 32871d098868SHong Zhang const PetscInt *r,*ic; 32881d098868SHong Zhang PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag; 3289dc3a2fd3SHong Zhang PetscInt *bi,*bj,*bdiag,*bdiag_rev; 3290f61a948aSLisandro Dalcin PetscInt row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au; 32911d098868SHong Zhang PetscInt nlnk,*lnk; 32921d098868SHong Zhang PetscBT lnkbt; 3293ace3abfcSBarry Smith PetscBool row_identity,icol_identity; 32948aee2decSHong Zhang MatScalar *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp; 32951d098868SHong Zhang const PetscInt *ics; 32961d098868SHong Zhang PetscInt j,nz,*pj,*bjtmp,k,ncut,*jtmp; 3297a2ea699eSBarry Smith PetscReal dt =info->dt,shift=info->shiftamount; 3298f61a948aSLisandro Dalcin PetscInt dtcount=(PetscInt)info->dtcount,nnz_max; 3299ace3abfcSBarry Smith PetscBool missing; 3300599ef60dSHong Zhang 3301599ef60dSHong Zhang PetscFunctionBegin; 3302f61a948aSLisandro Dalcin if (dt == PETSC_DEFAULT) dt = 0.005; 3303f61a948aSLisandro Dalcin if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax); 3304f61a948aSLisandro Dalcin 33051d098868SHong Zhang /* ------- symbolic factorization, can be reused ---------*/ 33061d098868SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 3307e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 33081d098868SHong Zhang adiag=a->diag; 3309599ef60dSHong Zhang 3310599ef60dSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 3311599ef60dSHong Zhang 3312599ef60dSHong Zhang /* bdiag is location of diagonal in factor */ 3313*785e854fSJed Brown ierr = PetscMalloc1((n+1),&bdiag);CHKERRQ(ierr); /* becomes b->diag */ 3314*785e854fSJed Brown ierr = PetscMalloc1((n+1),&bdiag_rev);CHKERRQ(ierr); /* temporary */ 3315599ef60dSHong Zhang 33161d098868SHong Zhang /* allocate row pointers bi */ 3317*785e854fSJed Brown ierr = PetscMalloc1((2*n+2),&bi);CHKERRQ(ierr); 33181d098868SHong Zhang 33191d098868SHong Zhang /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */ 3320393d3a68SHong Zhang if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */ 33211d098868SHong Zhang nnz_max = ai[n]+2*n*dtcount+2; 33228fc3a347SHong Zhang 3323*785e854fSJed Brown ierr = PetscMalloc1((nnz_max+1),&bj);CHKERRQ(ierr); 3324*785e854fSJed Brown ierr = PetscMalloc1((nnz_max+1),&ba);CHKERRQ(ierr); 3325599ef60dSHong Zhang 3326599ef60dSHong Zhang /* put together the new matrix */ 33270298fd71SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 33283bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr); 3329f284f12aSHong Zhang b = (Mat_SeqAIJ*)B->data; 33302205254eSKarl Rupp 3331599ef60dSHong Zhang b->free_a = PETSC_TRUE; 3332599ef60dSHong Zhang b->free_ij = PETSC_TRUE; 3333599ef60dSHong Zhang b->singlemalloc = PETSC_FALSE; 33342205254eSKarl Rupp 3335599ef60dSHong Zhang b->a = ba; 3336599ef60dSHong Zhang b->j = bj; 3337599ef60dSHong Zhang b->i = bi; 3338599ef60dSHong Zhang b->diag = bdiag; 3339599ef60dSHong Zhang b->ilen = 0; 3340599ef60dSHong Zhang b->imax = 0; 3341599ef60dSHong Zhang b->row = isrow; 3342599ef60dSHong Zhang b->col = iscol; 3343599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 3344599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 3345599ef60dSHong Zhang b->icol = isicol; 3346599ef60dSHong Zhang 3347*785e854fSJed Brown ierr = PetscMalloc1((n+1),&b->solve_work);CHKERRQ(ierr); 33483bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 33491d098868SHong Zhang b->maxnz = nnz_max; 3350599ef60dSHong Zhang 3351d5f3da31SBarry Smith B->factortype = MAT_FACTOR_ILUDT; 3352f284f12aSHong Zhang B->info.factor_mallocs = 0; 3353f284f12aSHong Zhang B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]); 33541d098868SHong Zhang /* ------- end of symbolic factorization ---------*/ 3355599ef60dSHong Zhang 3356599ef60dSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 3357599ef60dSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3358599ef60dSHong Zhang ics = ic; 3359599ef60dSHong Zhang 3360599ef60dSHong Zhang /* linked list for storing column indices of the active row */ 3361599ef60dSHong Zhang nlnk = n + 1; 3362599ef60dSHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 33631d098868SHong Zhang 33641d098868SHong Zhang /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */ 3365dcca6d9dSJed Brown ierr = PetscMalloc2(n,&im,n,&jtmp);CHKERRQ(ierr); 33661d098868SHong Zhang /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */ 3367dcca6d9dSJed Brown ierr = PetscMalloc2(n,&rtmp,n,&vtmp);CHKERRQ(ierr); 33680e83c824SBarry Smith ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr); 3369599ef60dSHong Zhang 3370599ef60dSHong Zhang bi[0] = 0; 33718fc3a347SHong Zhang bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */ 3372dc3a2fd3SHong Zhang bdiag_rev[n] = bdiag[0]; 33738fc3a347SHong Zhang bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */ 3374599ef60dSHong Zhang for (i=0; i<n; i++) { 3375599ef60dSHong Zhang /* copy initial fill into linked list */ 33768369b28dSHong Zhang nzi = ai[r[i]+1] - ai[r[i]]; 3377e32f2f54SBarry 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); 33788369b28dSHong Zhang nzi_al = adiag[r[i]] - ai[r[i]]; 33798369b28dSHong Zhang nzi_au = ai[r[i]+1] - adiag[r[i]] -1; 3380599ef60dSHong Zhang ajtmp = aj + ai[r[i]]; 33818369b28dSHong Zhang ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3382599ef60dSHong Zhang 3383599ef60dSHong Zhang /* load in initial (unfactored row) */ 33841d098868SHong Zhang aatmp = a->a + ai[r[i]]; 33858369b28dSHong Zhang for (j=0; j<nzi; j++) { 33861d098868SHong Zhang rtmp[ics[*ajtmp++]] = *aatmp++; 3387599ef60dSHong Zhang } 3388599ef60dSHong Zhang 3389599ef60dSHong Zhang /* add pivot rows into linked list */ 3390599ef60dSHong Zhang row = lnk[n]; 3391599ef60dSHong Zhang while (row < i) { 33921d098868SHong Zhang nzi_bl = bi[row+1] - bi[row] + 1; 33931d098868SHong Zhang bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */ 33941d098868SHong Zhang ierr = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr); 3395599ef60dSHong Zhang nzi += nlnk; 3396599ef60dSHong Zhang row = lnk[row]; 3397599ef60dSHong Zhang } 3398599ef60dSHong Zhang 33998369b28dSHong Zhang /* copy data from lnk into jtmp, then initialize lnk */ 34008369b28dSHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr); 3401599ef60dSHong Zhang 3402599ef60dSHong Zhang /* numerical factorization */ 34038369b28dSHong Zhang bjtmp = jtmp; 3404599ef60dSHong Zhang row = *bjtmp++; /* 1st pivot row */ 3405599ef60dSHong Zhang while (row < i) { 3406599ef60dSHong Zhang pc = rtmp + row; 34073c2a7987SHong Zhang pv = ba + bdiag[row]; /* 1./(diag of the pivot row) */ 34083c2a7987SHong Zhang multiplier = (*pc) * (*pv); 3409599ef60dSHong Zhang *pc = multiplier; 34103c2a7987SHong Zhang if (PetscAbsScalar(*pc) > dt) { /* apply tolerance dropping rule */ 34111d098868SHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 34121d098868SHong Zhang pv = ba + bdiag[row+1] + 1; 34131d098868SHong Zhang /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */ 34141d098868SHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 34151d098868SHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 341628b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 3417599ef60dSHong Zhang } 3418599ef60dSHong Zhang row = *bjtmp++; 3419599ef60dSHong Zhang } 3420599ef60dSHong Zhang 34218369b28dSHong Zhang /* copy sparse rtmp into contiguous vtmp; separate L and U part */ 34228aee2decSHong Zhang diag_tmp = rtmp[i]; /* save diagonal value - may not needed?? */ 34238369b28dSHong Zhang nzi_bl = 0; j = 0; 34248369b28dSHong Zhang while (jtmp[j] < i) { /* Note: jtmp is sorted */ 34258aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 34268369b28dSHong Zhang nzi_bl++; j++; 34278369b28dSHong Zhang } 34288369b28dSHong Zhang nzi_bu = nzi - nzi_bl -1; 34298369b28dSHong Zhang while (j < nzi) { 34308aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 34318369b28dSHong Zhang j++; 34328369b28dSHong Zhang } 34338369b28dSHong Zhang 3434599ef60dSHong Zhang bjtmp = bj + bi[i]; 3435599ef60dSHong Zhang batmp = ba + bi[i]; 34368369b28dSHong Zhang /* apply level dropping rule to L part */ 34378369b28dSHong Zhang ncut = nzi_al + dtcount; 34388369b28dSHong Zhang if (ncut < nzi_bl) { 34398369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr); 34408369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr); 3441599ef60dSHong Zhang } else { 34428369b28dSHong Zhang ncut = nzi_bl; 34438369b28dSHong Zhang } 34448369b28dSHong Zhang for (j=0; j<ncut; j++) { 34458369b28dSHong Zhang bjtmp[j] = jtmp[j]; 34468369b28dSHong Zhang batmp[j] = vtmp[j]; 34476da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */ 34488369b28dSHong Zhang } 34491d098868SHong Zhang bi[i+1] = bi[i] + ncut; 34508369b28dSHong Zhang nzi = ncut + 1; 34518369b28dSHong Zhang 34528369b28dSHong Zhang /* apply level dropping rule to U part */ 34538369b28dSHong Zhang ncut = nzi_au + dtcount; 34548369b28dSHong Zhang if (ncut < nzi_bu) { 34558369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr); 34568369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr); 34578369b28dSHong Zhang } else { 34588369b28dSHong Zhang ncut = nzi_bu; 34598369b28dSHong Zhang } 34608369b28dSHong Zhang nzi += ncut; 34611d098868SHong Zhang 34621d098868SHong Zhang /* mark bdiagonal */ 34631d098868SHong Zhang bdiag[i+1] = bdiag[i] - (ncut + 1); 3464dc3a2fd3SHong Zhang bdiag_rev[n-i-1] = bdiag[i+1]; 34658fc3a347SHong Zhang bi[2*n - i] = bi[2*n - i +1] - (ncut + 1); 34661d098868SHong Zhang bjtmp = bj + bdiag[i]; 34671d098868SHong Zhang batmp = ba + bdiag[i]; 34681d098868SHong Zhang *bjtmp = i; 34698aee2decSHong Zhang *batmp = diag_tmp; /* rtmp[i]; */ 3470c9c72f8fSHong Zhang if (*batmp == 0.0) { 3471c9c72f8fSHong Zhang *batmp = dt+shift; 3472bd1bc851SBarry Smith /* printf(" row %d add shift %g\n",i,shift); */ 3473c9c72f8fSHong Zhang } 3474b78a477dSHong Zhang *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */ 34756da515a1SHong Zhang /* printf(" (%d,%g),",*bjtmp,*batmp); */ 34761d098868SHong Zhang 34771d098868SHong Zhang bjtmp = bj + bdiag[i+1]+1; 34781d098868SHong Zhang batmp = ba + bdiag[i+1]+1; 34798369b28dSHong Zhang for (k=0; k<ncut; k++) { 34801d098868SHong Zhang bjtmp[k] = jtmp[nzi_bl+1+k]; 34811d098868SHong Zhang batmp[k] = vtmp[nzi_bl+1+k]; 34826da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */ 3483599ef60dSHong Zhang } 34846da515a1SHong Zhang /* printf("\n"); */ 3485599ef60dSHong Zhang 3486599ef60dSHong Zhang im[i] = nzi; /* used by PetscLLAddSortedLU() */ 3487599ef60dSHong Zhang /* 34881d098868SHong Zhang printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]); 34898369b28dSHong Zhang printf(" ----------------------------\n"); 3490599ef60dSHong Zhang */ 34918369b28dSHong Zhang } /* for (i=0; i<n; i++) */ 34921d098868SHong Zhang /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */ 3493e32f2f54SBarry 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]); 3494599ef60dSHong Zhang 3495599ef60dSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3496599ef60dSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3497599ef60dSHong Zhang 3498599ef60dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 34990e83c824SBarry Smith ierr = PetscFree2(im,jtmp);CHKERRQ(ierr); 35000e83c824SBarry Smith ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr); 35011bfa06eaSJed Brown ierr = PetscFree(bdiag_rev);CHKERRQ(ierr); 3502599ef60dSHong Zhang 3503599ef60dSHong Zhang ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr); 35041d098868SHong Zhang b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n]; 3505599ef60dSHong Zhang 3506599ef60dSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 3507599ef60dSHong Zhang ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr); 3508599ef60dSHong Zhang if (row_identity && icol_identity) { 350935233d99SShri Abhyankar B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 3510599ef60dSHong Zhang } else { 351135233d99SShri Abhyankar B->ops->solve = MatSolve_SeqAIJ; 3512599ef60dSHong Zhang } 3513599ef60dSHong Zhang 35141d098868SHong Zhang B->ops->solveadd = 0; 35151d098868SHong Zhang B->ops->solvetranspose = 0; 35161d098868SHong Zhang B->ops->solvetransposeadd = 0; 35171d098868SHong Zhang B->ops->matsolve = 0; 3518599ef60dSHong Zhang B->assembled = PETSC_TRUE; 3519599ef60dSHong Zhang B->preallocated = PETSC_TRUE; 3520599ef60dSHong Zhang PetscFunctionReturn(0); 3521599ef60dSHong Zhang } 3522599ef60dSHong Zhang 35233c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */ 35243c2a7987SHong Zhang #undef __FUNCT__ 35253c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ" 3526fe97e370SBarry Smith /* 3527fe97e370SBarry 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 3528fe97e370SBarry Smith */ 3529fe97e370SBarry Smith 35307087cfbeSBarry Smith PetscErrorCode MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info) 35313c2a7987SHong Zhang { 35323c2a7987SHong Zhang PetscErrorCode ierr; 35333c2a7987SHong Zhang 35343c2a7987SHong Zhang PetscFunctionBegin; 35353c2a7987SHong Zhang ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr); 35363c2a7987SHong Zhang PetscFunctionReturn(0); 35373c2a7987SHong Zhang } 35383c2a7987SHong Zhang 35393c2a7987SHong Zhang /* 35403c2a7987SHong Zhang same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors 35413c2a7987SHong Zhang - intend to replace existing MatLUFactorNumeric_SeqAIJ() 35423c2a7987SHong Zhang */ 35433c2a7987SHong Zhang #undef __FUNCT__ 35443c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ" 3545fe97e370SBarry Smith /* 3546fe97e370SBarry 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 3547fe97e370SBarry Smith */ 3548fe97e370SBarry Smith 35497087cfbeSBarry Smith PetscErrorCode MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info) 35503c2a7987SHong Zhang { 35513c2a7987SHong Zhang Mat C =fact; 35523c2a7987SHong Zhang Mat_SeqAIJ *a =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data; 35533c2a7987SHong Zhang IS isrow = b->row,isicol = b->icol; 35543c2a7987SHong Zhang PetscErrorCode ierr; 35553c2a7987SHong Zhang const PetscInt *r,*ic,*ics; 3556b78a477dSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 3557b78a477dSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj; 35583c2a7987SHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 3559182b8fbaSHong Zhang PetscReal dt=info->dt,shift=info->shiftamount; 3560ace3abfcSBarry Smith PetscBool row_identity, col_identity; 35613c2a7987SHong Zhang 35623c2a7987SHong Zhang PetscFunctionBegin; 35633c2a7987SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 35643c2a7987SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3565*785e854fSJed Brown ierr = PetscMalloc1((n+1),&rtmp);CHKERRQ(ierr); 35663c2a7987SHong Zhang ics = ic; 35673c2a7987SHong Zhang 35683c2a7987SHong Zhang for (i=0; i<n; i++) { 3569b78a477dSHong Zhang /* initialize rtmp array */ 3570b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of nozeros in L(i,:) */ 35713c2a7987SHong Zhang bjtmp = bj + bi[i]; 3572b78a477dSHong Zhang for (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0; 3573b78a477dSHong Zhang rtmp[i] = 0.0; 3574b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */ 3575b78a477dSHong Zhang bjtmp = bj + bdiag[i+1] + 1; 3576b78a477dSHong Zhang for (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0; 35773c2a7987SHong Zhang 3578b78a477dSHong Zhang /* load in initial unfactored row of A */ 35796da515a1SHong Zhang /* printf("row %d\n",i); */ 35803c2a7987SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 35813c2a7987SHong Zhang ajtmp = aj + ai[r[i]]; 35823c2a7987SHong Zhang v = aa + ai[r[i]]; 35833c2a7987SHong Zhang for (j=0; j<nz; j++) { 3584b78a477dSHong Zhang rtmp[ics[*ajtmp++]] = v[j]; 35856da515a1SHong Zhang /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */ 35863c2a7987SHong Zhang } 35876da515a1SHong Zhang /* printf("\n"); */ 35883c2a7987SHong Zhang 3589b78a477dSHong Zhang /* numerical factorization */ 3590b78a477dSHong Zhang bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */ 3591b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of entries in L(i,:) */ 3592b78a477dSHong Zhang k = 0; 3593b78a477dSHong Zhang while (k < nzl) { 35943c2a7987SHong Zhang row = *bjtmp++; 35956da515a1SHong Zhang /* printf(" prow %d\n",row); */ 35963c2a7987SHong Zhang pc = rtmp + row; 3597b78a477dSHong Zhang pv = b->a + bdiag[row]; /* 1./(diag of the pivot row) */ 3598b78a477dSHong Zhang multiplier = (*pc) * (*pv); 35993c2a7987SHong Zhang *pc = multiplier; 3600b78a477dSHong Zhang if (PetscAbsScalar(multiplier) > dt) { 3601b78a477dSHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 3602b78a477dSHong Zhang pv = b->a + bdiag[row+1] + 1; 3603b78a477dSHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 3604b78a477dSHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 360528b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 36063c2a7987SHong Zhang } 3607b78a477dSHong Zhang k++; 36083c2a7987SHong Zhang } 36093c2a7987SHong Zhang 3610b78a477dSHong Zhang /* finished row so stick it into b->a */ 3611b78a477dSHong Zhang /* L-part */ 3612b78a477dSHong Zhang pv = b->a + bi[i]; 3613b78a477dSHong Zhang pj = bj + bi[i]; 3614b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; 3615b78a477dSHong Zhang for (j=0; j<nzl; j++) { 3616b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 36176da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 36183c2a7987SHong Zhang } 3619b78a477dSHong Zhang 3620b78a477dSHong Zhang /* diagonal: invert diagonal entries for simplier triangular solves */ 3621b78a477dSHong Zhang if (rtmp[i] == 0.0) rtmp[i] = dt+shift; 3622b78a477dSHong Zhang b->a[bdiag[i]] = 1.0/rtmp[i]; 36236da515a1SHong Zhang /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */ 3624b78a477dSHong Zhang 3625b78a477dSHong Zhang /* U-part */ 3626b78a477dSHong Zhang pv = b->a + bdiag[i+1] + 1; 3627b78a477dSHong Zhang pj = bj + bdiag[i+1] + 1; 3628b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1] - 1; 3629b78a477dSHong Zhang for (j=0; j<nzu; j++) { 3630b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 36316da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 3632b78a477dSHong Zhang } 36336da515a1SHong Zhang /* printf("\n"); */ 3634b78a477dSHong Zhang } 3635b78a477dSHong Zhang 36363c2a7987SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 36373c2a7987SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 36383c2a7987SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3639b78a477dSHong Zhang 36403c2a7987SHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 36413c2a7987SHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 36423c2a7987SHong Zhang if (row_identity && col_identity) { 364335233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 36443c2a7987SHong Zhang } else { 364535233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ; 36463c2a7987SHong Zhang } 3647b78a477dSHong Zhang C->ops->solveadd = 0; 3648b78a477dSHong Zhang C->ops->solvetranspose = 0; 3649b78a477dSHong Zhang C->ops->solvetransposeadd = 0; 3650b78a477dSHong Zhang C->ops->matsolve = 0; 36513c2a7987SHong Zhang C->assembled = PETSC_TRUE; 36523c2a7987SHong Zhang C->preallocated = PETSC_TRUE; 36532205254eSKarl Rupp 36543c2a7987SHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 36553c2a7987SHong Zhang PetscFunctionReturn(0); 36563c2a7987SHong Zhang } 3657