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 78fa24674SBarry Smith /* 88fa24674SBarry Smith Computes an ordering to get most of the large numerical values in the lower triangular part of the matrix 9312e372aSBarry Smith 10312e372aSBarry Smith This code does not work and is not called anywhere. It would be registered with MatOrderingRegisterAll() 118fa24674SBarry Smith */ 1219fd82e9SBarry Smith PetscErrorCode MatGetOrdering_Flow_SeqAIJ(Mat mat,MatOrderingType type,IS *irow,IS *icol) 133b2fbd54SBarry Smith { 148fa24674SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)mat->data; 158fa24674SBarry Smith PetscErrorCode ierr; 16889a8dedSBarry Smith PetscInt i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order; 178fa24674SBarry Smith const PetscInt *ai = a->i, *aj = a->j; 188fa24674SBarry Smith const PetscScalar *aa = a->a; 19ace3abfcSBarry Smith PetscBool *done; 20889a8dedSBarry Smith PetscReal best,past = 0,future; 213a40ed3dSBarry Smith 228fa24674SBarry Smith PetscFunctionBegin; 238fa24674SBarry Smith /* pick initial row */ 248fa24674SBarry Smith best = -1; 258fa24674SBarry Smith for (i=0; i<n; i++) { 2675567043SBarry Smith future = 0.0; 278fa24674SBarry Smith for (j=ai[i]; j<ai[i+1]; j++) { 28db4deed7SKarl Rupp if (aj[j] != i) future += PetscAbsScalar(aa[j]); 29db4deed7SKarl Rupp else past = PetscAbsScalar(aa[j]); 308fa24674SBarry Smith } 318fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 328fa24674SBarry Smith if (past/future > best) { 338fa24674SBarry Smith best = past/future; 348fa24674SBarry Smith current = i; 358fa24674SBarry Smith } 368fa24674SBarry Smith } 378fa24674SBarry Smith 38785e854fSJed Brown ierr = PetscMalloc1(n,&done);CHKERRQ(ierr); 39ace3abfcSBarry Smith ierr = PetscMemzero(done,n*sizeof(PetscBool));CHKERRQ(ierr); 40785e854fSJed Brown ierr = PetscMalloc1(n,&order);CHKERRQ(ierr); 418fa24674SBarry Smith order[0] = current; 428fa24674SBarry Smith for (i=0; i<n-1; i++) { 438fa24674SBarry Smith done[current] = PETSC_TRUE; 448fa24674SBarry Smith best = -1; 458fa24674SBarry Smith /* loop over all neighbors of current pivot */ 468fa24674SBarry Smith for (j=ai[current]; j<ai[current+1]; j++) { 478fa24674SBarry Smith jj = aj[j]; 488fa24674SBarry Smith if (done[jj]) continue; 498fa24674SBarry Smith /* loop over columns of potential next row computing weights for below and above diagonal */ 508fa24674SBarry Smith past = future = 0.0; 518fa24674SBarry Smith for (k=ai[jj]; k<ai[jj+1]; k++) { 528fa24674SBarry Smith kk = aj[k]; 538fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[k]); 548fa24674SBarry Smith else if (kk != jj) future += PetscAbsScalar(aa[k]); 558fa24674SBarry Smith } 568fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 578fa24674SBarry Smith if (past/future > best) { 588fa24674SBarry Smith best = past/future; 598fa24674SBarry Smith newcurrent = jj; 608fa24674SBarry Smith } 618fa24674SBarry Smith } 628fa24674SBarry Smith if (best == -1) { /* no neighbors to select from so select best of all that remain */ 638fa24674SBarry Smith best = -1; 648fa24674SBarry Smith for (k=0; k<n; k++) { 658fa24674SBarry Smith if (done[k]) continue; 6675567043SBarry Smith future = 0.0; 6775567043SBarry Smith past = 0.0; 688fa24674SBarry Smith for (j=ai[k]; j<ai[k+1]; j++) { 698fa24674SBarry Smith kk = aj[j]; 708fa24674SBarry Smith if (done[kk]) past += PetscAbsScalar(aa[j]); 718fa24674SBarry Smith else if (kk != k) future += PetscAbsScalar(aa[j]); 728fa24674SBarry Smith } 738fa24674SBarry Smith if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */ 748fa24674SBarry Smith if (past/future > best) { 758fa24674SBarry Smith best = past/future; 768fa24674SBarry Smith newcurrent = k; 778fa24674SBarry Smith } 788fa24674SBarry Smith } 798fa24674SBarry Smith } 80e32f2f54SBarry Smith if (current == newcurrent) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"newcurrent cannot be current"); 818fa24674SBarry Smith current = newcurrent; 828fa24674SBarry Smith order[i+1] = current; 838fa24674SBarry Smith } 8470b3c8c7SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,order,PETSC_COPY_VALUES,irow);CHKERRQ(ierr); 858fa24674SBarry Smith *icol = *irow; 868fa24674SBarry Smith ierr = PetscObjectReference((PetscObject)*irow);CHKERRQ(ierr); 878fa24674SBarry Smith ierr = PetscFree(done);CHKERRQ(ierr); 888fa24674SBarry Smith ierr = PetscFree(order);CHKERRQ(ierr); 89d64ed03dSBarry Smith PetscFunctionReturn(0); 903b2fbd54SBarry Smith } 913b2fbd54SBarry Smith 92cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B) 93b24902e0SBarry Smith { 94d0f46423SBarry Smith PetscInt n = A->rmap->n; 95b24902e0SBarry Smith PetscErrorCode ierr; 96b24902e0SBarry Smith 97b24902e0SBarry Smith PetscFunctionBegin; 98891bceaeSHong Zhang #if defined(PETSC_USE_COMPLEX) 99891bceaeSHong Zhang if (A->hermitian && (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported"); 100891bceaeSHong Zhang #endif 101ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr); 102b24902e0SBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 103599ef60dSHong Zhang if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT) { 104b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr); 1052205254eSKarl Rupp 10635233d99SShri Abhyankar (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ; 10735233d99SShri Abhyankar (*B)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqAIJ; 1082205254eSKarl Rupp 10933d57670SJed Brown ierr = MatSetBlockSizesFromMats(*B,A,A);CHKERRQ(ierr); 110b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 111b24902e0SBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 1120298fd71SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 1132205254eSKarl Rupp 11435233d99SShri Abhyankar (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqAIJ; 11535233d99SShri Abhyankar (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ; 116e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported"); 117d5f3da31SBarry Smith (*B)->factortype = ftype; 11800c67f3bSHong Zhang 11900c67f3bSHong Zhang ierr = PetscFree((*B)->solvertype);CHKERRQ(ierr); 12000c67f3bSHong Zhang ierr = PetscStrallocpy(MATSOLVERPETSC,&(*B)->solvertype);CHKERRQ(ierr); 121b24902e0SBarry Smith PetscFunctionReturn(0); 122b24902e0SBarry Smith } 123b24902e0SBarry Smith 124ad04f41aSHong Zhang PetscErrorCode MatLUFactorSymbolic_SeqAIJ_inplace(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 125289bc588SBarry Smith { 126416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 127289bc588SBarry Smith IS isicol; 1286849ba73SBarry Smith PetscErrorCode ierr; 1295d0c19d7SBarry Smith const PetscInt *r,*ic; 1305d0c19d7SBarry Smith PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j; 1311393bc97SHong Zhang PetscInt *bi,*bj,*ajtmp; 1321393bc97SHong Zhang PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 1339e046878SBarry Smith PetscReal f; 1344875ba38SHong Zhang PetscInt nlnk,*lnk,k,**bi_ptr; 1350298fd71SBarry Smith PetscFreeSpaceList free_space=NULL,current_space=NULL; 1361393bc97SHong Zhang PetscBT lnkbt; 137ece542f9SHong Zhang PetscBool missing; 138289bc588SBarry Smith 1393a40ed3dSBarry Smith PetscFunctionBegin; 140e32f2f54SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square"); 141ece542f9SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 142ece542f9SHong Zhang if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 143ece542f9SHong Zhang 1444c49b128SBarry Smith ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 145ac121b3dSBarry Smith ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 146ac121b3dSBarry Smith ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 147289bc588SBarry Smith 148289bc588SBarry Smith /* get new row pointers */ 149854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&bi);CHKERRQ(ierr); 1501393bc97SHong Zhang bi[0] = 0; 1511393bc97SHong Zhang 1521393bc97SHong Zhang /* bdiag is location of diagonal in factor */ 153854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr); 1541393bc97SHong Zhang bdiag[0] = 0; 1551393bc97SHong Zhang 1561393bc97SHong Zhang /* linked list for storing column indices of the active row */ 1571393bc97SHong Zhang nlnk = n + 1; 1581393bc97SHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1591393bc97SHong Zhang 160dcca6d9dSJed Brown ierr = PetscMalloc2(n+1,&bi_ptr,n+1,&im);CHKERRQ(ierr); 1611393bc97SHong Zhang 1621393bc97SHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 163d3d32019SBarry Smith f = info->fill; 1642f18eb33SBarry Smith ierr = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr); 1651393bc97SHong Zhang current_space = free_space; 166289bc588SBarry Smith 167289bc588SBarry Smith for (i=0; i<n; i++) { 1681393bc97SHong Zhang /* copy previous fill into linked list */ 169289bc588SBarry Smith nzi = 0; 1701393bc97SHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 1711393bc97SHong Zhang ajtmp = aj + ai[r[i]]; 172afcc9904SHong Zhang ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 1731393bc97SHong Zhang nzi += nlnk; 1741393bc97SHong Zhang 1751393bc97SHong Zhang /* add pivot rows into linked list */ 1761393bc97SHong Zhang row = lnk[n]; 1771393bc97SHong Zhang while (row < i) { 178d1170560SHong Zhang nzbd = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */ 179d1170560SHong Zhang ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 180d1170560SHong Zhang ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 1811393bc97SHong Zhang nzi += nlnk; 1821393bc97SHong Zhang row = lnk[row]; 183289bc588SBarry Smith } 1841393bc97SHong Zhang bi[i+1] = bi[i] + nzi; 1851393bc97SHong Zhang im[i] = nzi; 1861393bc97SHong Zhang 1871393bc97SHong Zhang /* mark bdiag */ 1881393bc97SHong Zhang nzbd = 0; 1891393bc97SHong Zhang nnz = nzi; 1901393bc97SHong Zhang k = lnk[n]; 1911393bc97SHong Zhang while (nnz-- && k < i) { 1921393bc97SHong Zhang nzbd++; 1931393bc97SHong Zhang k = lnk[k]; 1941393bc97SHong Zhang } 1951393bc97SHong Zhang bdiag[i] = bi[i] + nzbd; 1961393bc97SHong Zhang 1971393bc97SHong Zhang /* if free space is not available, make more free space */ 1981393bc97SHong Zhang if (current_space->local_remaining<nzi) { 199f91af8c7SBarry Smith nnz = PetscIntMultTruncate(n - i,nzi); /* estimated and max additional space needed */ 200a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 2011393bc97SHong Zhang reallocs++; 2021393bc97SHong Zhang } 2031393bc97SHong Zhang 2041393bc97SHong Zhang /* copy data into free space, then initialize lnk */ 2051393bc97SHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 2062205254eSKarl Rupp 2071393bc97SHong Zhang bi_ptr[i] = current_space->array; 2081393bc97SHong Zhang current_space->array += nzi; 2091393bc97SHong Zhang current_space->local_used += nzi; 2101393bc97SHong Zhang current_space->local_remaining -= nzi; 211289bc588SBarry Smith } 2126cf91177SBarry Smith #if defined(PETSC_USE_INFO) 213464e8d44SSatish Balay if (ai[n] != 0) { 2141393bc97SHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 21557622a8eSBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr); 21657622a8eSBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr); 21757622a8eSBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g);\n",(double)af);CHKERRQ(ierr); 218ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 21905bf559cSSatish Balay } else { 220ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 22105bf559cSSatish Balay } 22263ba0a88SBarry Smith #endif 2232fb3ed76SBarry Smith 224898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 225898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2261987afe7SBarry Smith 2271393bc97SHong Zhang /* destroy list of free space and other temporary array(s) */ 228854ce69bSBarry Smith ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr); 229a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); 2301393bc97SHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 23135baf27eSSatish Balay ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 232289bc588SBarry Smith 233289bc588SBarry Smith /* put together the new matrix */ 2340298fd71SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 2353bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr); 236719d5645SBarry Smith b = (Mat_SeqAIJ*)(B)->data; 2372205254eSKarl Rupp 238e6b907acSBarry Smith b->free_a = PETSC_TRUE; 239e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 2407c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 2412205254eSKarl Rupp 242854ce69bSBarry Smith ierr = PetscMalloc1(bi[n]+1,&b->a);CHKERRQ(ierr); 2431393bc97SHong Zhang b->j = bj; 2441393bc97SHong Zhang b->i = bi; 2451393bc97SHong Zhang b->diag = bdiag; 246416022c9SBarry Smith b->ilen = 0; 247416022c9SBarry Smith b->imax = 0; 248416022c9SBarry Smith b->row = isrow; 249416022c9SBarry Smith b->col = iscol; 250c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 251c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 25282bf6240SBarry Smith b->icol = isicol; 253854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr); 2541393bc97SHong Zhang 2551393bc97SHong Zhang /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 2563bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 2571393bc97SHong Zhang b->maxnz = b->nz = bi[n]; 2587fd98bd6SLois Curfman McInnes 259d5f3da31SBarry Smith (B)->factortype = MAT_FACTOR_LU; 260719d5645SBarry Smith (B)->info.factor_mallocs = reallocs; 261719d5645SBarry Smith (B)->info.fill_ratio_given = f; 262703deb49SSatish Balay 2639f7d0b68SHong Zhang if (ai[n]) { 264719d5645SBarry Smith (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 265eea2913aSSatish Balay } else { 266719d5645SBarry Smith (B)->info.fill_ratio_needed = 0.0; 267eea2913aSSatish Balay } 268ad04f41aSHong Zhang (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 269f2344125SBarry Smith if (a->inode.size) { 270f2344125SBarry Smith (B)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 271f2344125SBarry Smith } 2723a40ed3dSBarry Smith PetscFunctionReturn(0); 273289bc588SBarry Smith } 2741393bc97SHong Zhang 27535233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 276ce3d78c0SShri Abhyankar { 277ce3d78c0SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 278ce3d78c0SShri Abhyankar IS isicol; 279ce3d78c0SShri Abhyankar PetscErrorCode ierr; 2808758e1faSBarry Smith const PetscInt *r,*ic,*ai=a->i,*aj=a->j,*ajtmp; 2818758e1faSBarry Smith PetscInt i,n=A->rmap->n; 2828758e1faSBarry Smith PetscInt *bi,*bj; 283ce3d78c0SShri Abhyankar PetscInt *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im; 284ce3d78c0SShri Abhyankar PetscReal f; 285ce3d78c0SShri Abhyankar PetscInt nlnk,*lnk,k,**bi_ptr; 2860298fd71SBarry Smith PetscFreeSpaceList free_space=NULL,current_space=NULL; 287ce3d78c0SShri Abhyankar PetscBT lnkbt; 288ece542f9SHong Zhang PetscBool missing; 289ce3d78c0SShri Abhyankar 290ce3d78c0SShri Abhyankar PetscFunctionBegin; 291e32f2f54SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square"); 292ece542f9SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 293ece542f9SHong Zhang if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 294ece542f9SHong Zhang 295ce3d78c0SShri Abhyankar ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 296ce3d78c0SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 297ce3d78c0SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 298ce3d78c0SShri Abhyankar 2991bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 300854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&bi);CHKERRQ(ierr); 301854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr); 302a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 3039b48462bSShri Abhyankar 3049b48462bSShri Abhyankar /* linked list for storing column indices of the active row */ 3059b48462bSShri Abhyankar nlnk = n + 1; 3069b48462bSShri Abhyankar ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3079b48462bSShri Abhyankar 308dcca6d9dSJed Brown ierr = PetscMalloc2(n+1,&bi_ptr,n+1,&im);CHKERRQ(ierr); 3099b48462bSShri Abhyankar 3109b48462bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 3119b48462bSShri Abhyankar f = info->fill; 3122f18eb33SBarry Smith ierr = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr); 3139b48462bSShri Abhyankar current_space = free_space; 3149b48462bSShri Abhyankar 3159b48462bSShri Abhyankar for (i=0; i<n; i++) { 3169b48462bSShri Abhyankar /* copy previous fill into linked list */ 3179b48462bSShri Abhyankar nzi = 0; 3189b48462bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 3199b48462bSShri Abhyankar ajtmp = aj + ai[r[i]]; 3209b48462bSShri Abhyankar ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3219b48462bSShri Abhyankar nzi += nlnk; 3229b48462bSShri Abhyankar 3239b48462bSShri Abhyankar /* add pivot rows into linked list */ 3249b48462bSShri Abhyankar row = lnk[n]; 3259b48462bSShri Abhyankar while (row < i) { 3269b48462bSShri Abhyankar nzbd = bdiag[row] + 1; /* num of entries in the row with column index <= row */ 3279b48462bSShri Abhyankar ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */ 3289b48462bSShri Abhyankar ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr); 3299b48462bSShri Abhyankar nzi += nlnk; 3309b48462bSShri Abhyankar row = lnk[row]; 3319b48462bSShri Abhyankar } 3329b48462bSShri Abhyankar bi[i+1] = bi[i] + nzi; 3339b48462bSShri Abhyankar im[i] = nzi; 3349b48462bSShri Abhyankar 3359b48462bSShri Abhyankar /* mark bdiag */ 3369b48462bSShri Abhyankar nzbd = 0; 3379b48462bSShri Abhyankar nnz = nzi; 3389b48462bSShri Abhyankar k = lnk[n]; 3399b48462bSShri Abhyankar while (nnz-- && k < i) { 3409b48462bSShri Abhyankar nzbd++; 3419b48462bSShri Abhyankar k = lnk[k]; 3429b48462bSShri Abhyankar } 3439b48462bSShri Abhyankar bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */ 3449b48462bSShri Abhyankar 3459b48462bSShri Abhyankar /* if free space is not available, make more free space */ 3469b48462bSShri Abhyankar if (current_space->local_remaining<nzi) { 3472f18eb33SBarry Smith /* estimated additional space needed */ 3481df4278eSBarry Smith nnz = PetscIntMultTruncate(2,PetscIntMultTruncate(n-1,nzi)); 3499b48462bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 3509b48462bSShri Abhyankar reallocs++; 3519b48462bSShri Abhyankar } 3529b48462bSShri Abhyankar 3539b48462bSShri Abhyankar /* copy data into free space, then initialize lnk */ 3549b48462bSShri Abhyankar ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 3552205254eSKarl Rupp 3569b48462bSShri Abhyankar bi_ptr[i] = current_space->array; 3579b48462bSShri Abhyankar current_space->array += nzi; 3589b48462bSShri Abhyankar current_space->local_used += nzi; 3599b48462bSShri Abhyankar current_space->local_remaining -= nzi; 3609b48462bSShri Abhyankar } 3619b48462bSShri Abhyankar 3629b48462bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3639b48462bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3649b48462bSShri Abhyankar 3659263d837SHong Zhang /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 366854ce69bSBarry Smith ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr); 3672ce24eb6SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 3689b48462bSShri Abhyankar ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 3699b48462bSShri Abhyankar ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr); 3709b48462bSShri Abhyankar 3719b48462bSShri Abhyankar /* put together the new matrix */ 3720298fd71SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 3733bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr); 3749b48462bSShri Abhyankar b = (Mat_SeqAIJ*)(B)->data; 3752205254eSKarl Rupp 3769b48462bSShri Abhyankar b->free_a = PETSC_TRUE; 3779b48462bSShri Abhyankar b->free_ij = PETSC_TRUE; 3789b48462bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 3792205254eSKarl Rupp 380854ce69bSBarry Smith ierr = PetscMalloc1(bdiag[0]+1,&b->a);CHKERRQ(ierr); 3812205254eSKarl Rupp 3829b48462bSShri Abhyankar b->j = bj; 3839b48462bSShri Abhyankar b->i = bi; 3849b48462bSShri Abhyankar b->diag = bdiag; 3859b48462bSShri Abhyankar b->ilen = 0; 3869b48462bSShri Abhyankar b->imax = 0; 3879b48462bSShri Abhyankar b->row = isrow; 3889b48462bSShri Abhyankar b->col = iscol; 3899b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 3909b48462bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 3919b48462bSShri Abhyankar b->icol = isicol; 392854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr); 3939b48462bSShri Abhyankar 3949b48462bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. Allocate solve_work, new a, new j */ 3953bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 3969b48462bSShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 3972205254eSKarl Rupp 398d5f3da31SBarry Smith B->factortype = MAT_FACTOR_LU; 399f284f12aSHong Zhang B->info.factor_mallocs = reallocs; 400f284f12aSHong Zhang B->info.fill_ratio_given = f; 4019b48462bSShri Abhyankar 4029b48462bSShri Abhyankar if (ai[n]) { 403f284f12aSHong Zhang B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 4049b48462bSShri Abhyankar } else { 405f284f12aSHong Zhang B->info.fill_ratio_needed = 0.0; 4069b48462bSShri Abhyankar } 4079263d837SHong Zhang #if defined(PETSC_USE_INFO) 4089263d837SHong Zhang if (ai[n] != 0) { 4099263d837SHong Zhang PetscReal af = B->info.fill_ratio_needed; 41057622a8eSBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr); 41157622a8eSBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr); 41257622a8eSBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g);\n",(double)af);CHKERRQ(ierr); 4139263d837SHong Zhang ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 4149263d837SHong Zhang } else { 4159263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr); 4169263d837SHong Zhang } 4179263d837SHong Zhang #endif 41835233d99SShri Abhyankar B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 419f2344125SBarry Smith if (a->inode.size) { 420d3ac4fa3SBarry Smith B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 421d3ac4fa3SBarry Smith } 422a02bda8eSBarry Smith ierr = MatSeqAIJCheckInode_FactorLU(B);CHKERRQ(ierr); 4239b48462bSShri Abhyankar PetscFunctionReturn(0); 4249b48462bSShri Abhyankar } 4259b48462bSShri Abhyankar 4265b5bc046SBarry Smith /* 4275b5bc046SBarry Smith Trouble in factorization, should we dump the original matrix? 4285b5bc046SBarry Smith */ 4295b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A) 4305b5bc046SBarry Smith { 4315b5bc046SBarry Smith PetscErrorCode ierr; 432ace3abfcSBarry Smith PetscBool flg = PETSC_FALSE; 4335b5bc046SBarry Smith 4345b5bc046SBarry Smith PetscFunctionBegin; 435c5929fdfSBarry Smith ierr = PetscOptionsGetBool(((PetscObject)A)->options,NULL,"-mat_factor_dump_on_error",&flg,NULL);CHKERRQ(ierr); 4365b5bc046SBarry Smith if (flg) { 4375b5bc046SBarry Smith PetscViewer viewer; 4385b5bc046SBarry Smith char filename[PETSC_MAX_PATH_LEN]; 4395b5bc046SBarry Smith 4405b5bc046SBarry Smith ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr); 441ce94432eSBarry Smith ierr = PetscViewerBinaryOpen(PetscObjectComm((PetscObject)A),filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr); 4425b5bc046SBarry Smith ierr = MatView(A,viewer);CHKERRQ(ierr); 4436bf464f9SBarry Smith ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr); 4445b5bc046SBarry Smith } 4455b5bc046SBarry Smith PetscFunctionReturn(0); 4465b5bc046SBarry Smith } 4475b5bc046SBarry Smith 44835233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 449c9c72f8fSHong Zhang { 450c9c72f8fSHong Zhang Mat C =B; 451c9c72f8fSHong Zhang Mat_SeqAIJ *a =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data; 452c9c72f8fSHong Zhang IS isrow = b->row,isicol = b->icol; 453c9c72f8fSHong Zhang PetscErrorCode ierr; 454c9c72f8fSHong Zhang const PetscInt *r,*ic,*ics; 455bbd65245SShri Abhyankar const PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag; 456bbd65245SShri Abhyankar PetscInt i,j,k,nz,nzL,row,*pj; 457bbd65245SShri Abhyankar const PetscInt *ajtmp,*bjtmp; 458bbd65245SShri Abhyankar MatScalar *rtmp,*pc,multiplier,*pv; 459bbd65245SShri Abhyankar const MatScalar *aa=a->a,*v; 460ace3abfcSBarry Smith PetscBool row_identity,col_identity; 461d90ac83dSHong Zhang FactorShiftCtx sctx; 4624f81c4b7SBarry Smith const PetscInt *ddiag; 463d4ad06f7SHong Zhang PetscReal rs; 464d4ad06f7SHong Zhang MatScalar d; 465d4ad06f7SHong Zhang 466c9c72f8fSHong Zhang PetscFunctionBegin; 467d6e5152cSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 468d90ac83dSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 469d4ad06f7SHong Zhang 470f4db908eSBarry Smith if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 471d4ad06f7SHong Zhang ddiag = a->diag; 472d4ad06f7SHong Zhang sctx.shift_top = info->zeropivot; 473d4ad06f7SHong Zhang for (i=0; i<n; i++) { 474d4ad06f7SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 475d4ad06f7SHong Zhang d = (aa)[ddiag[i]]; 476d4ad06f7SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 477d4ad06f7SHong Zhang v = aa+ai[i]; 478d4ad06f7SHong Zhang nz = ai[i+1] - ai[i]; 4792205254eSKarl Rupp for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]); 480d4ad06f7SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 481d4ad06f7SHong Zhang } 482d4ad06f7SHong Zhang sctx.shift_top *= 1.1; 483d4ad06f7SHong Zhang sctx.nshift_max = 5; 484d4ad06f7SHong Zhang sctx.shift_lo = 0.; 485d4ad06f7SHong Zhang sctx.shift_hi = 1.; 486d4ad06f7SHong Zhang } 487d4ad06f7SHong Zhang 488c9c72f8fSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 489c9c72f8fSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 490854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr); 491c9c72f8fSHong Zhang ics = ic; 492c9c72f8fSHong Zhang 493d4ad06f7SHong Zhang do { 49407b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 495c9c72f8fSHong Zhang for (i=0; i<n; i++) { 496c9c72f8fSHong Zhang /* zero rtmp */ 497c9c72f8fSHong Zhang /* L part */ 498c9c72f8fSHong Zhang nz = bi[i+1] - bi[i]; 499c9c72f8fSHong Zhang bjtmp = bj + bi[i]; 500c9c72f8fSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 501c9c72f8fSHong Zhang 502c9c72f8fSHong Zhang /* U part */ 50362fbd6afSShri Abhyankar nz = bdiag[i]-bdiag[i+1]; 50462fbd6afSShri Abhyankar bjtmp = bj + bdiag[i+1]+1; 505813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 506813a1f2bSShri Abhyankar 507813a1f2bSShri Abhyankar /* load in initial (unfactored row) */ 508813a1f2bSShri Abhyankar nz = ai[r[i]+1] - ai[r[i]]; 509813a1f2bSShri Abhyankar ajtmp = aj + ai[r[i]]; 510813a1f2bSShri Abhyankar v = aa + ai[r[i]]; 511813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 512813a1f2bSShri Abhyankar rtmp[ics[ajtmp[j]]] = v[j]; 513813a1f2bSShri Abhyankar } 514813a1f2bSShri Abhyankar /* ZeropivotApply() */ 51598a93161SHong Zhang rtmp[i] += sctx.shift_amount; /* shift the diagonal of the matrix */ 516813a1f2bSShri Abhyankar 517813a1f2bSShri Abhyankar /* elimination */ 518813a1f2bSShri Abhyankar bjtmp = bj + bi[i]; 519813a1f2bSShri Abhyankar row = *bjtmp++; 520813a1f2bSShri Abhyankar nzL = bi[i+1] - bi[i]; 521f268cba8SShri Abhyankar for (k=0; k < nzL; k++) { 522813a1f2bSShri Abhyankar pc = rtmp + row; 523813a1f2bSShri Abhyankar if (*pc != 0.0) { 524813a1f2bSShri Abhyankar pv = b->a + bdiag[row]; 525813a1f2bSShri Abhyankar multiplier = *pc * (*pv); 526813a1f2bSShri Abhyankar *pc = multiplier; 5272205254eSKarl Rupp 52862fbd6afSShri Abhyankar pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 52962fbd6afSShri Abhyankar pv = b->a + bdiag[row+1]+1; 53062fbd6afSShri Abhyankar nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 5312205254eSKarl Rupp 532813a1f2bSShri Abhyankar for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 53328b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 534813a1f2bSShri Abhyankar } 535f268cba8SShri Abhyankar row = *bjtmp++; 536813a1f2bSShri Abhyankar } 537813a1f2bSShri Abhyankar 538813a1f2bSShri Abhyankar /* finished row so stick it into b->a */ 539813a1f2bSShri Abhyankar rs = 0.0; 540813a1f2bSShri Abhyankar /* L part */ 541813a1f2bSShri Abhyankar pv = b->a + bi[i]; 542813a1f2bSShri Abhyankar pj = b->j + bi[i]; 543813a1f2bSShri Abhyankar nz = bi[i+1] - bi[i]; 544813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 545813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 546813a1f2bSShri Abhyankar } 547813a1f2bSShri Abhyankar 548813a1f2bSShri Abhyankar /* U part */ 54962fbd6afSShri Abhyankar pv = b->a + bdiag[i+1]+1; 55062fbd6afSShri Abhyankar pj = b->j + bdiag[i+1]+1; 55162fbd6afSShri Abhyankar nz = bdiag[i] - bdiag[i+1]-1; 552813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 553813a1f2bSShri Abhyankar pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]); 554813a1f2bSShri Abhyankar } 555813a1f2bSShri Abhyankar 556813a1f2bSShri Abhyankar sctx.rs = rs; 557813a1f2bSShri Abhyankar sctx.pv = rtmp[i]; 5584cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr); 55907b50cabSHong Zhang if (sctx.newshift) break; /* break for-loop */ 56007b50cabSHong Zhang rtmp[i] = sctx.pv; /* sctx.pv might be updated in the case of MAT_SHIFT_INBLOCKS */ 561813a1f2bSShri Abhyankar 562813a1f2bSShri Abhyankar /* Mark diagonal and invert diagonal for simplier triangular solves */ 563813a1f2bSShri Abhyankar pv = b->a + bdiag[i]; 564813a1f2bSShri Abhyankar *pv = 1.0/rtmp[i]; 565813a1f2bSShri Abhyankar 566813a1f2bSShri Abhyankar } /* endof for (i=0; i<n; i++) { */ 567813a1f2bSShri Abhyankar 5688ff23777SHong Zhang /* MatPivotRefine() */ 56907b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 570813a1f2bSShri Abhyankar /* 571813a1f2bSShri Abhyankar * if no shift in this attempt & shifting & started shifting & can refine, 572813a1f2bSShri Abhyankar * then try lower shift 573813a1f2bSShri Abhyankar */ 574813a1f2bSShri Abhyankar sctx.shift_hi = sctx.shift_fraction; 575813a1f2bSShri Abhyankar sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 576813a1f2bSShri Abhyankar sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 57707b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 578813a1f2bSShri Abhyankar sctx.nshift++; 579813a1f2bSShri Abhyankar } 58007b50cabSHong Zhang } while (sctx.newshift); 581813a1f2bSShri Abhyankar 582813a1f2bSShri Abhyankar ierr = PetscFree(rtmp);CHKERRQ(ierr); 583813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 584813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 585d3ac4fa3SBarry Smith 586813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 587813a1f2bSShri Abhyankar ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 588abb87a52SBarry Smith if (b->inode.size) { 589abb87a52SBarry Smith C->ops->solve = MatSolve_SeqAIJ_Inode; 590abb87a52SBarry Smith } else if (row_identity && col_identity) { 59135233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 592813a1f2bSShri Abhyankar } else { 59335233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ; 594813a1f2bSShri Abhyankar } 59535233d99SShri Abhyankar C->ops->solveadd = MatSolveAdd_SeqAIJ; 59635233d99SShri Abhyankar C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 59735233d99SShri Abhyankar C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 59835233d99SShri Abhyankar C->ops->matsolve = MatMatSolve_SeqAIJ; 599813a1f2bSShri Abhyankar C->assembled = PETSC_TRUE; 600813a1f2bSShri Abhyankar C->preallocated = PETSC_TRUE; 6012205254eSKarl Rupp 602813a1f2bSShri Abhyankar ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 603813a1f2bSShri Abhyankar 60414d2772eSHong Zhang /* MatShiftView(A,info,&sctx) */ 605813a1f2bSShri Abhyankar if (sctx.nshift) { 606f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 60757622a8eSBarry 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,(double)sctx.shift_amount,(double)sctx.shift_fraction,(double)sctx.shift_top);CHKERRQ(ierr); 608f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 60957622a8eSBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr); 610f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) { 61157622a8eSBarry Smith ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %g\n",sctx.nshift,(double)info->shiftamount);CHKERRQ(ierr); 612813a1f2bSShri Abhyankar } 613813a1f2bSShri Abhyankar } 614813a1f2bSShri Abhyankar PetscFunctionReturn(0); 615813a1f2bSShri Abhyankar } 616813a1f2bSShri Abhyankar 617ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info) 618289bc588SBarry Smith { 619719d5645SBarry Smith Mat C =B; 620416022c9SBarry Smith Mat_SeqAIJ *a =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data; 62182bf6240SBarry Smith IS isrow = b->row,isicol = b->icol; 6226849ba73SBarry Smith PetscErrorCode ierr; 6235d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 624d278edc7SBarry Smith PetscInt nz,row,i,j,n=A->rmap->n,diag; 625d278edc7SBarry Smith const PetscInt *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 626d278edc7SBarry Smith const PetscInt *ajtmp,*bjtmp,*diag_offset = b->diag,*pj; 627d278edc7SBarry Smith MatScalar *pv,*rtmp,*pc,multiplier,d; 628d278edc7SBarry Smith const MatScalar *v,*aa=a->a; 62933f9893dSHong Zhang PetscReal rs=0.0; 630d90ac83dSHong Zhang FactorShiftCtx sctx; 6314f81c4b7SBarry Smith const PetscInt *ddiag; 632ace3abfcSBarry Smith PetscBool row_identity, col_identity; 633289bc588SBarry Smith 6343a40ed3dSBarry Smith PetscFunctionBegin; 635504a3fadSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 636504a3fadSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 637ada7143aSSatish Balay 638f4db908eSBarry Smith if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 63942898933SHong Zhang ddiag = a->diag; 6409f7d0b68SHong Zhang sctx.shift_top = info->zeropivot; 6416cc28720Svictorle for (i=0; i<n; i++) { 6429f95998fSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 6439f7d0b68SHong Zhang d = (aa)[ddiag[i]]; 6441a890ab1SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 6459f7d0b68SHong Zhang v = aa+ai[i]; 6469f7d0b68SHong Zhang nz = ai[i+1] - ai[i]; 6472205254eSKarl Rupp for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]); 6481a890ab1SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 6496cc28720Svictorle } 650118f5789SBarry Smith sctx.shift_top *= 1.1; 651d2276718SHong Zhang sctx.nshift_max = 5; 652d2276718SHong Zhang sctx.shift_lo = 0.; 653d2276718SHong Zhang sctx.shift_hi = 1.; 654a0a356efSHong Zhang } 655a0a356efSHong Zhang 656504a3fadSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 657504a3fadSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 658854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr); 659504a3fadSHong Zhang ics = ic; 660504a3fadSHong Zhang 661085256b3SBarry Smith do { 66207b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 663289bc588SBarry Smith for (i=0; i<n; i++) { 664b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 665b3bf805bSHong Zhang bjtmp = bj + bi[i]; 6669f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 667289bc588SBarry Smith 668289bc588SBarry Smith /* load in initial (unfactored row) */ 6699f7d0b68SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 6709f7d0b68SHong Zhang ajtmp = aj + ai[r[i]]; 6719f7d0b68SHong Zhang v = aa + ai[r[i]]; 672085256b3SBarry Smith for (j=0; j<nz; j++) { 673b3bf805bSHong Zhang rtmp[ics[ajtmp[j]]] = v[j]; 674085256b3SBarry Smith } 675d2276718SHong Zhang rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 676289bc588SBarry Smith 677b3bf805bSHong Zhang row = *bjtmp++; 678f2582111SSatish Balay while (row < i) { 6798c37ef55SBarry Smith pc = rtmp + row; 6808c37ef55SBarry Smith if (*pc != 0.0) { 681010a20caSHong Zhang pv = b->a + diag_offset[row]; 682010a20caSHong Zhang pj = b->j + diag_offset[row] + 1; 68335aab85fSBarry Smith multiplier = *pc / *pv++; 6848c37ef55SBarry Smith *pc = multiplier; 685b3bf805bSHong Zhang nz = bi[row+1] - diag_offset[row] - 1; 6869f7d0b68SHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 68728b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 688289bc588SBarry Smith } 689b3bf805bSHong Zhang row = *bjtmp++; 690289bc588SBarry Smith } 691416022c9SBarry Smith /* finished row so stick it into b->a */ 692b3bf805bSHong Zhang pv = b->a + bi[i]; 693b3bf805bSHong Zhang pj = b->j + bi[i]; 694b3bf805bSHong Zhang nz = bi[i+1] - bi[i]; 695b3bf805bSHong Zhang diag = diag_offset[i] - bi[i]; 696e57ff13aSBarry Smith rs = 0.0; 697d3d32019SBarry Smith for (j=0; j<nz; j++) { 6989f7d0b68SHong Zhang pv[j] = rtmp[pj[j]]; 6999f7d0b68SHong Zhang rs += PetscAbsScalar(pv[j]); 700d3d32019SBarry Smith } 701e57ff13aSBarry Smith rs -= PetscAbsScalar(pv[diag]); 702d2276718SHong Zhang 703d2276718SHong Zhang sctx.rs = rs; 704d2276718SHong Zhang sctx.pv = pv[diag]; 7054cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr); 70607b50cabSHong Zhang if (sctx.newshift) break; 707504a3fadSHong Zhang pv[diag] = sctx.pv; 70835aab85fSBarry Smith } 709d2276718SHong Zhang 71007b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 7116cc28720Svictorle /* 7126c037d1bSvictorle * if no shift in this attempt & shifting & started shifting & can refine, 7136cc28720Svictorle * then try lower shift 7146cc28720Svictorle */ 7150481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 7160481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 7170481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 71807b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 719d2276718SHong Zhang sctx.nshift++; 7206cc28720Svictorle } 72107b50cabSHong Zhang } while (sctx.newshift); 7220f11f9aeSSatish Balay 72335aab85fSBarry Smith /* invert diagonal entries for simplier triangular solves */ 72435aab85fSBarry Smith for (i=0; i<n; i++) { 725010a20caSHong Zhang b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]]; 72635aab85fSBarry Smith } 727606d414cSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 72878b31e54SBarry Smith ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 72978b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 730d3ac4fa3SBarry Smith 7318d8c7c43SBarry Smith ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 7328d8c7c43SBarry Smith ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 7338d8c7c43SBarry Smith if (row_identity && col_identity) { 734ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_inplace; 7358d8c7c43SBarry Smith } else { 736ad04f41aSHong Zhang C->ops->solve = MatSolve_SeqAIJ_inplace; 737db4efbfdSBarry Smith } 738ad04f41aSHong Zhang C->ops->solveadd = MatSolveAdd_SeqAIJ_inplace; 739ad04f41aSHong Zhang C->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 740ad04f41aSHong Zhang C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 741ad04f41aSHong Zhang C->ops->matsolve = MatMatSolve_SeqAIJ_inplace; 7422205254eSKarl Rupp 743c456f294SBarry Smith C->assembled = PETSC_TRUE; 7445c9eb25fSBarry Smith C->preallocated = PETSC_TRUE; 7452205254eSKarl Rupp 746d0f46423SBarry Smith ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 747d2276718SHong Zhang if (sctx.nshift) { 748f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 74957622a8eSBarry 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,(double)sctx.shift_amount,(double)sctx.shift_fraction,(double)sctx.shift_top);CHKERRQ(ierr); 750f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 75157622a8eSBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr); 7526cc28720Svictorle } 7530697b6caSHong Zhang } 754d3ac4fa3SBarry Smith (C)->ops->solve = MatSolve_SeqAIJ_inplace; 755d3ac4fa3SBarry Smith (C)->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 7562205254eSKarl Rupp 757a02bda8eSBarry Smith ierr = MatSeqAIJCheckInode(C);CHKERRQ(ierr); 7583a40ed3dSBarry Smith PetscFunctionReturn(0); 759289bc588SBarry Smith } 7600889b5dcSvictorle 761137fb511SHong Zhang /* 762137fb511SHong Zhang This routine implements inplace ILU(0) with row or/and column permutations. 763137fb511SHong Zhang Input: 764137fb511SHong Zhang A - original matrix 765137fb511SHong Zhang Output; 766137fb511SHong Zhang A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i] 767137fb511SHong Zhang a->j (col index) is permuted by the inverse of colperm, then sorted 768137fb511SHong Zhang a->a reordered accordingly with a->j 769137fb511SHong Zhang a->diag (ptr to diagonal elements) is updated. 770137fb511SHong Zhang */ 7710481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info) 772137fb511SHong Zhang { 773b051339dSHong Zhang Mat_SeqAIJ *a =(Mat_SeqAIJ*)A->data; 774b051339dSHong Zhang IS isrow = a->row,isicol = a->icol; 775137fb511SHong Zhang PetscErrorCode ierr; 7765d0c19d7SBarry Smith const PetscInt *r,*ic,*ics; 7775d0c19d7SBarry Smith PetscInt i,j,n=A->rmap->n,*ai=a->i,*aj=a->j; 7785d0c19d7SBarry Smith PetscInt *ajtmp,nz,row; 779b051339dSHong Zhang PetscInt *diag = a->diag,nbdiag,*pj; 780a77337e4SBarry Smith PetscScalar *rtmp,*pc,multiplier,d; 781504a3fadSHong Zhang MatScalar *pv,*v; 782137fb511SHong Zhang PetscReal rs; 783d90ac83dSHong Zhang FactorShiftCtx sctx; 784504a3fadSHong Zhang const MatScalar *aa=a->a,*vtmp; 785137fb511SHong Zhang 786137fb511SHong Zhang PetscFunctionBegin; 787e32f2f54SBarry Smith if (A != B) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address"); 788504a3fadSHong Zhang 789504a3fadSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 790504a3fadSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 791504a3fadSHong Zhang 792504a3fadSHong Zhang if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 793504a3fadSHong Zhang const PetscInt *ddiag = a->diag; 794504a3fadSHong Zhang sctx.shift_top = info->zeropivot; 795504a3fadSHong Zhang for (i=0; i<n; i++) { 796504a3fadSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 797504a3fadSHong Zhang d = (aa)[ddiag[i]]; 798504a3fadSHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 799504a3fadSHong Zhang vtmp = aa+ai[i]; 800504a3fadSHong Zhang nz = ai[i+1] - ai[i]; 8012205254eSKarl Rupp for (j=0; j<nz; j++) rs += PetscAbsScalar(vtmp[j]); 802504a3fadSHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 803504a3fadSHong Zhang } 804504a3fadSHong Zhang sctx.shift_top *= 1.1; 805504a3fadSHong Zhang sctx.nshift_max = 5; 806504a3fadSHong Zhang sctx.shift_lo = 0.; 807504a3fadSHong Zhang sctx.shift_hi = 1.; 808504a3fadSHong Zhang } 809504a3fadSHong Zhang 810137fb511SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 811137fb511SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 812854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr); 813137fb511SHong Zhang ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 814b051339dSHong Zhang ics = ic; 815137fb511SHong Zhang 816504a3fadSHong Zhang #if defined(MV) 81775567043SBarry Smith sctx.shift_top = 0.; 818e60cf9a0SBarry Smith sctx.nshift_max = 0; 81975567043SBarry Smith sctx.shift_lo = 0.; 82075567043SBarry Smith sctx.shift_hi = 0.; 82175567043SBarry Smith sctx.shift_fraction = 0.; 822137fb511SHong Zhang 823f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 82475567043SBarry Smith sctx.shift_top = 0.; 825137fb511SHong Zhang for (i=0; i<n; i++) { 826137fb511SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 827b051339dSHong Zhang d = (a->a)[diag[i]]; 828137fb511SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 829b051339dSHong Zhang v = a->a+ai[i]; 830b051339dSHong Zhang nz = ai[i+1] - ai[i]; 8312205254eSKarl Rupp for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]); 832137fb511SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 833137fb511SHong Zhang } 834137fb511SHong Zhang if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot; 835137fb511SHong Zhang sctx.shift_top *= 1.1; 836137fb511SHong Zhang sctx.nshift_max = 5; 837137fb511SHong Zhang sctx.shift_lo = 0.; 838137fb511SHong Zhang sctx.shift_hi = 1.; 839137fb511SHong Zhang } 840137fb511SHong Zhang 84175567043SBarry Smith sctx.shift_amount = 0.; 842137fb511SHong Zhang sctx.nshift = 0; 843504a3fadSHong Zhang #endif 844504a3fadSHong Zhang 845137fb511SHong Zhang do { 84607b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 847137fb511SHong Zhang for (i=0; i<n; i++) { 848b051339dSHong Zhang /* load in initial unfactored row */ 849b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 850b051339dSHong Zhang ajtmp = aj + ai[r[i]]; 851b051339dSHong Zhang v = a->a + ai[r[i]]; 852137fb511SHong Zhang /* sort permuted ajtmp and values v accordingly */ 853b051339dSHong Zhang for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]]; 854137fb511SHong Zhang ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr); 855137fb511SHong Zhang 856b051339dSHong Zhang diag[r[i]] = ai[r[i]]; 857137fb511SHong Zhang for (j=0; j<nz; j++) { 858137fb511SHong Zhang rtmp[ajtmp[j]] = v[j]; 859b051339dSHong Zhang if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */ 860137fb511SHong Zhang } 861137fb511SHong Zhang rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */ 862137fb511SHong Zhang 863b051339dSHong Zhang row = *ajtmp++; 864137fb511SHong Zhang while (row < i) { 865137fb511SHong Zhang pc = rtmp + row; 866137fb511SHong Zhang if (*pc != 0.0) { 867b051339dSHong Zhang pv = a->a + diag[r[row]]; 868b051339dSHong Zhang pj = aj + diag[r[row]] + 1; 869137fb511SHong Zhang 870137fb511SHong Zhang multiplier = *pc / *pv++; 871137fb511SHong Zhang *pc = multiplier; 872b051339dSHong Zhang nz = ai[r[row]+1] - diag[r[row]] - 1; 873b051339dSHong Zhang for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j]; 87428b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 875137fb511SHong Zhang } 876b051339dSHong Zhang row = *ajtmp++; 877137fb511SHong Zhang } 878b051339dSHong Zhang /* finished row so overwrite it onto a->a */ 879b051339dSHong Zhang pv = a->a + ai[r[i]]; 880b051339dSHong Zhang pj = aj + ai[r[i]]; 881b051339dSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 882b051339dSHong Zhang nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */ 883137fb511SHong Zhang 884137fb511SHong Zhang rs = 0.0; 885137fb511SHong Zhang for (j=0; j<nz; j++) { 886b051339dSHong Zhang pv[j] = rtmp[pj[j]]; 887b051339dSHong Zhang if (j != nbdiag) rs += PetscAbsScalar(pv[j]); 888137fb511SHong Zhang } 889137fb511SHong Zhang 890137fb511SHong Zhang sctx.rs = rs; 891b051339dSHong Zhang sctx.pv = pv[nbdiag]; 8924cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr); 89307b50cabSHong Zhang if (sctx.newshift) break; 894504a3fadSHong Zhang pv[nbdiag] = sctx.pv; 895137fb511SHong Zhang } 896137fb511SHong Zhang 89707b50cabSHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 898137fb511SHong Zhang /* 899137fb511SHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 900137fb511SHong Zhang * then try lower shift 901137fb511SHong Zhang */ 9020481f469SBarry Smith sctx.shift_hi = sctx.shift_fraction; 9030481f469SBarry Smith sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 9040481f469SBarry Smith sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 90507b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 906137fb511SHong Zhang sctx.nshift++; 907137fb511SHong Zhang } 90807b50cabSHong Zhang } while (sctx.newshift); 909137fb511SHong Zhang 910137fb511SHong Zhang /* invert diagonal entries for simplier triangular solves */ 911137fb511SHong Zhang for (i=0; i<n; i++) { 912b051339dSHong Zhang a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]]; 913137fb511SHong Zhang } 914137fb511SHong Zhang 915137fb511SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 916137fb511SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 917137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 9182205254eSKarl Rupp 919b051339dSHong Zhang A->ops->solve = MatSolve_SeqAIJ_InplaceWithPerm; 920ad04f41aSHong Zhang A->ops->solveadd = MatSolveAdd_SeqAIJ_inplace; 921ad04f41aSHong Zhang A->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 922ad04f41aSHong Zhang A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 9232205254eSKarl Rupp 924b051339dSHong Zhang A->assembled = PETSC_TRUE; 9255c9eb25fSBarry Smith A->preallocated = PETSC_TRUE; 9262205254eSKarl Rupp 927d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr); 928137fb511SHong Zhang if (sctx.nshift) { 929f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 93057622a8eSBarry 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,(double)sctx.shift_amount,(double)sctx.shift_fraction,(double)sctx.shift_top);CHKERRQ(ierr); 931f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 93257622a8eSBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr); 933137fb511SHong Zhang } 934137fb511SHong Zhang } 935137fb511SHong Zhang PetscFunctionReturn(0); 936137fb511SHong Zhang } 937137fb511SHong Zhang 9380a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9390481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info) 940da3a660dSBarry Smith { 941dfbe8321SBarry Smith PetscErrorCode ierr; 942416022c9SBarry Smith Mat C; 943416022c9SBarry Smith 9443a40ed3dSBarry Smith PetscFunctionBegin; 9452692d6eeSBarry Smith ierr = MatGetFactor(A,MATSOLVERPETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr); 946719d5645SBarry Smith ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr); 947719d5645SBarry Smith ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr); 9482205254eSKarl Rupp 949db4efbfdSBarry Smith A->ops->solve = C->ops->solve; 950db4efbfdSBarry Smith A->ops->solvetranspose = C->ops->solvetranspose; 9512205254eSKarl Rupp 95228be2f97SBarry Smith ierr = MatHeaderMerge(A,&C);CHKERRQ(ierr); 9533bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)A,(PetscObject)((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr); 9543a40ed3dSBarry Smith PetscFunctionReturn(0); 955da3a660dSBarry Smith } 9560a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */ 9571d098868SHong Zhang 958889a8dedSBarry Smith 959ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx) 9608c37ef55SBarry Smith { 961416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 962416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 9636849ba73SBarry Smith PetscErrorCode ierr; 9645d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 9655d0c19d7SBarry Smith PetscInt nz; 9665d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 967dd6ea824SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 968d9fead3dSBarry Smith const PetscScalar *b; 969dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 9708c37ef55SBarry Smith 9713a40ed3dSBarry Smith PetscFunctionBegin; 9723a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 97391bf9adeSBarry Smith 9743649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 975*fdc842d1SBarry Smith ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr); 976416022c9SBarry Smith tmp = a->solve_work; 9778c37ef55SBarry Smith 9783b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 9793b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 9808c37ef55SBarry Smith 9818c37ef55SBarry Smith /* forward solve the lower triangular */ 9828c37ef55SBarry Smith tmp[0] = b[*r++]; 983010a20caSHong Zhang tmps = tmp; 9848c37ef55SBarry Smith for (i=1; i<n; i++) { 985010a20caSHong Zhang v = aa + ai[i]; 986010a20caSHong Zhang vi = aj + ai[i]; 98753e7425aSBarry Smith nz = a->diag[i] - ai[i]; 98853e7425aSBarry Smith sum = b[*r++]; 989003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 9908c37ef55SBarry Smith tmp[i] = sum; 9918c37ef55SBarry Smith } 9928c37ef55SBarry Smith 9938c37ef55SBarry Smith /* backward solve the upper triangular */ 9948c37ef55SBarry Smith for (i=n-1; i>=0; i--) { 995010a20caSHong Zhang v = aa + a->diag[i] + 1; 996010a20caSHong Zhang vi = aj + a->diag[i] + 1; 997416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 9988c37ef55SBarry Smith sum = tmp[i]; 999003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1000010a20caSHong Zhang x[*c--] = tmp[i] = sum*aa[a->diag[i]]; 10018c37ef55SBarry Smith } 10028c37ef55SBarry Smith 10033b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10043b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10053649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1006*fdc842d1SBarry Smith ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr); 1007dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 10083a40ed3dSBarry Smith PetscFunctionReturn(0); 10098c37ef55SBarry Smith } 1010026e39d0SSatish Balay 1011ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X) 10122bebee5dSHong Zhang { 10132bebee5dSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10142bebee5dSHong Zhang IS iscol = a->col,isrow = a->row; 10152bebee5dSHong Zhang PetscErrorCode ierr; 10165d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 10175d0c19d7SBarry Smith PetscInt nz,neq; 10185d0c19d7SBarry Smith const PetscInt *rout,*cout,*r,*c; 10191683a169SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 10201683a169SBarry Smith const PetscScalar *aa = a->a,*v; 10211683a169SBarry Smith const PetscScalar *b; 1022ace3abfcSBarry Smith PetscBool bisdense,xisdense; 10232bebee5dSHong Zhang 10242bebee5dSHong Zhang PetscFunctionBegin; 10252bebee5dSHong Zhang if (!n) PetscFunctionReturn(0); 10262bebee5dSHong Zhang 1027251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1028e32f2f54SBarry Smith if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 1029251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1030e32f2f54SBarry Smith if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 1031db4efbfdSBarry Smith 10321683a169SBarry Smith ierr = MatDenseGetArrayRead(B,&b);CHKERRQ(ierr); 10338c778c55SBarry Smith ierr = MatDenseGetArray(X,&x);CHKERRQ(ierr); 10342bebee5dSHong Zhang 10352bebee5dSHong Zhang tmp = a->solve_work; 10362bebee5dSHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10372bebee5dSHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 10382bebee5dSHong Zhang 1039a36b22ccSSatish Balay for (neq=0; neq<B->cmap->n; neq++) { 10402bebee5dSHong Zhang /* forward solve the lower triangular */ 10412bebee5dSHong Zhang tmp[0] = b[r[0]]; 10422bebee5dSHong Zhang tmps = tmp; 10432bebee5dSHong Zhang for (i=1; i<n; i++) { 10442bebee5dSHong Zhang v = aa + ai[i]; 10452bebee5dSHong Zhang vi = aj + ai[i]; 10462bebee5dSHong Zhang nz = a->diag[i] - ai[i]; 10472bebee5dSHong Zhang sum = b[r[i]]; 1048003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10492bebee5dSHong Zhang tmp[i] = sum; 10502bebee5dSHong Zhang } 10512bebee5dSHong Zhang /* backward solve the upper triangular */ 10522bebee5dSHong Zhang for (i=n-1; i>=0; i--) { 10532bebee5dSHong Zhang v = aa + a->diag[i] + 1; 10542bebee5dSHong Zhang vi = aj + a->diag[i] + 1; 10552bebee5dSHong Zhang nz = ai[i+1] - a->diag[i] - 1; 10562bebee5dSHong Zhang sum = tmp[i]; 1057003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 10582bebee5dSHong Zhang x[c[i]] = tmp[i] = sum*aa[a->diag[i]]; 10592bebee5dSHong Zhang } 10602bebee5dSHong Zhang 10612bebee5dSHong Zhang b += n; 10622bebee5dSHong Zhang x += n; 10632bebee5dSHong Zhang } 10642bebee5dSHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 10652bebee5dSHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 10661683a169SBarry Smith ierr = MatDenseRestoreArrayRead(B,&b);CHKERRQ(ierr); 10678c778c55SBarry Smith ierr = MatDenseRestoreArray(X,&x);CHKERRQ(ierr); 1068dc0b31edSSatish Balay ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 10692bebee5dSHong Zhang PetscFunctionReturn(0); 10702bebee5dSHong Zhang } 10712bebee5dSHong Zhang 107235233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X) 10739bd0847aSShri Abhyankar { 10749bd0847aSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 10759bd0847aSShri Abhyankar IS iscol = a->col,isrow = a->row; 10769bd0847aSShri Abhyankar PetscErrorCode ierr; 10779bd0847aSShri Abhyankar PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 10789bd0847aSShri Abhyankar PetscInt nz,neq; 10799bd0847aSShri Abhyankar const PetscInt *rout,*cout,*r,*c; 10801683a169SBarry Smith PetscScalar *x,*tmp,sum; 10811683a169SBarry Smith const PetscScalar *b; 10821683a169SBarry Smith const PetscScalar *aa = a->a,*v; 1083ace3abfcSBarry Smith PetscBool bisdense,xisdense; 10849bd0847aSShri Abhyankar 10859bd0847aSShri Abhyankar PetscFunctionBegin; 10869bd0847aSShri Abhyankar if (!n) PetscFunctionReturn(0); 10879bd0847aSShri Abhyankar 1088251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr); 1089e32f2f54SBarry Smith if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix"); 1090251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr); 1091e32f2f54SBarry Smith if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix"); 10929bd0847aSShri Abhyankar 10931683a169SBarry Smith ierr = MatDenseGetArrayRead(B,&b);CHKERRQ(ierr); 10948c778c55SBarry Smith ierr = MatDenseGetArray(X,&x);CHKERRQ(ierr); 10959bd0847aSShri Abhyankar 10969bd0847aSShri Abhyankar tmp = a->solve_work; 10979bd0847aSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 10989bd0847aSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 10999bd0847aSShri Abhyankar 11009bd0847aSShri Abhyankar for (neq=0; neq<B->cmap->n; neq++) { 11019bd0847aSShri Abhyankar /* forward solve the lower triangular */ 11029bd0847aSShri Abhyankar tmp[0] = b[r[0]]; 11039bd0847aSShri Abhyankar v = aa; 11049bd0847aSShri Abhyankar vi = aj; 11059bd0847aSShri Abhyankar for (i=1; i<n; i++) { 11069bd0847aSShri Abhyankar nz = ai[i+1] - ai[i]; 11079bd0847aSShri Abhyankar sum = b[r[i]]; 11089bd0847aSShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 11099bd0847aSShri Abhyankar tmp[i] = sum; 11109bd0847aSShri Abhyankar v += nz; vi += nz; 11119bd0847aSShri Abhyankar } 11129bd0847aSShri Abhyankar 11139bd0847aSShri Abhyankar /* backward solve the upper triangular */ 11149bd0847aSShri Abhyankar for (i=n-1; i>=0; i--) { 11159bd0847aSShri Abhyankar v = aa + adiag[i+1]+1; 11169bd0847aSShri Abhyankar vi = aj + adiag[i+1]+1; 11179bd0847aSShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 11189bd0847aSShri Abhyankar sum = tmp[i]; 11199bd0847aSShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 11209bd0847aSShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 11219bd0847aSShri Abhyankar } 11229bd0847aSShri Abhyankar 11239bd0847aSShri Abhyankar b += n; 11249bd0847aSShri Abhyankar x += n; 11259bd0847aSShri Abhyankar } 11269bd0847aSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 11279bd0847aSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 11281683a169SBarry Smith ierr = MatDenseRestoreArrayRead(B,&b);CHKERRQ(ierr); 11298c778c55SBarry Smith ierr = MatDenseRestoreArray(X,&x);CHKERRQ(ierr); 11309bd0847aSShri Abhyankar ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr); 11319bd0847aSShri Abhyankar PetscFunctionReturn(0); 11329bd0847aSShri Abhyankar } 11339bd0847aSShri Abhyankar 1134137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx) 1135137fb511SHong Zhang { 1136137fb511SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1137137fb511SHong Zhang IS iscol = a->col,isrow = a->row; 1138137fb511SHong Zhang PetscErrorCode ierr; 11395d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 11405d0c19d7SBarry Smith PetscInt i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j; 11415d0c19d7SBarry Smith PetscInt nz,row; 1142*fdc842d1SBarry Smith PetscScalar *x,*tmp,*tmps,sum; 1143*fdc842d1SBarry Smith const PetscScalar *b; 1144dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1145137fb511SHong Zhang 1146137fb511SHong Zhang PetscFunctionBegin; 1147137fb511SHong Zhang if (!n) PetscFunctionReturn(0); 1148137fb511SHong Zhang 1149*fdc842d1SBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1150*fdc842d1SBarry Smith ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr); 1151137fb511SHong Zhang tmp = a->solve_work; 1152137fb511SHong Zhang 1153137fb511SHong Zhang ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1154137fb511SHong Zhang ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1155137fb511SHong Zhang 1156137fb511SHong Zhang /* forward solve the lower triangular */ 1157137fb511SHong Zhang tmp[0] = b[*r++]; 1158137fb511SHong Zhang tmps = tmp; 1159137fb511SHong Zhang for (row=1; row<n; row++) { 1160137fb511SHong Zhang i = rout[row]; /* permuted row */ 1161137fb511SHong Zhang v = aa + ai[i]; 1162137fb511SHong Zhang vi = aj + ai[i]; 1163137fb511SHong Zhang nz = a->diag[i] - ai[i]; 1164137fb511SHong Zhang sum = b[*r++]; 1165003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1166137fb511SHong Zhang tmp[row] = sum; 1167137fb511SHong Zhang } 1168137fb511SHong Zhang 1169137fb511SHong Zhang /* backward solve the upper triangular */ 1170137fb511SHong Zhang for (row=n-1; row>=0; row--) { 1171137fb511SHong Zhang i = rout[row]; /* permuted row */ 1172137fb511SHong Zhang v = aa + a->diag[i] + 1; 1173137fb511SHong Zhang vi = aj + a->diag[i] + 1; 1174137fb511SHong Zhang nz = ai[i+1] - a->diag[i] - 1; 1175137fb511SHong Zhang sum = tmp[row]; 1176003131ecSBarry Smith PetscSparseDenseMinusDot(sum,tmps,v,vi,nz); 1177137fb511SHong Zhang x[*c--] = tmp[row] = sum*aa[a->diag[i]]; 1178137fb511SHong Zhang } 1179137fb511SHong Zhang 1180137fb511SHong Zhang ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1181137fb511SHong Zhang ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1182*fdc842d1SBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1183*fdc842d1SBarry Smith ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr); 1184dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1185137fb511SHong Zhang PetscFunctionReturn(0); 1186137fb511SHong Zhang } 1187137fb511SHong Zhang 1188930ae53cSBarry Smith /* ----------------------------------------------------------- */ 1189c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/fsolve.h> 1190ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx) 1191930ae53cSBarry Smith { 1192930ae53cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 11936849ba73SBarry Smith PetscErrorCode ierr; 1194d0f46423SBarry Smith PetscInt n = A->rmap->n; 1195b377110cSBarry Smith const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag; 1196356650c2SBarry Smith PetscScalar *x; 1197356650c2SBarry Smith const PetscScalar *b; 1198dd6ea824SBarry Smith const MatScalar *aa = a->a; 1199aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 1200356650c2SBarry Smith PetscInt adiag_i,i,nz,ai_i; 1201b377110cSBarry Smith const PetscInt *vi; 1202dd6ea824SBarry Smith const MatScalar *v; 1203dd6ea824SBarry Smith PetscScalar sum; 1204d85afc42SSatish Balay #endif 1205930ae53cSBarry Smith 12063a40ed3dSBarry Smith PetscFunctionBegin; 12073a40ed3dSBarry Smith if (!n) PetscFunctionReturn(0); 1208930ae53cSBarry Smith 12093649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1210*fdc842d1SBarry Smith ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr); 1211930ae53cSBarry Smith 1212aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ) 12131c4feecaSSatish Balay fortransolveaij_(&n,x,ai,aj,adiag,aa,b); 12141c4feecaSSatish Balay #else 1215930ae53cSBarry Smith /* forward solve the lower triangular */ 1216930ae53cSBarry Smith x[0] = b[0]; 1217930ae53cSBarry Smith for (i=1; i<n; i++) { 1218930ae53cSBarry Smith ai_i = ai[i]; 1219930ae53cSBarry Smith v = aa + ai_i; 1220930ae53cSBarry Smith vi = aj + ai_i; 1221930ae53cSBarry Smith nz = adiag[i] - ai_i; 1222930ae53cSBarry Smith sum = b[i]; 1223003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1224930ae53cSBarry Smith x[i] = sum; 1225930ae53cSBarry Smith } 1226930ae53cSBarry Smith 1227930ae53cSBarry Smith /* backward solve the upper triangular */ 1228930ae53cSBarry Smith for (i=n-1; i>=0; i--) { 1229930ae53cSBarry Smith adiag_i = adiag[i]; 1230d708749eSBarry Smith v = aa + adiag_i + 1; 1231d708749eSBarry Smith vi = aj + adiag_i + 1; 1232930ae53cSBarry Smith nz = ai[i+1] - adiag_i - 1; 1233930ae53cSBarry Smith sum = x[i]; 1234003131ecSBarry Smith PetscSparseDenseMinusDot(sum,x,v,vi,nz); 1235930ae53cSBarry Smith x[i] = sum*aa[adiag_i]; 1236930ae53cSBarry Smith } 12371c4feecaSSatish Balay #endif 1238dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 12393649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1240*fdc842d1SBarry Smith ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr); 12413a40ed3dSBarry Smith PetscFunctionReturn(0); 1242930ae53cSBarry Smith } 1243930ae53cSBarry Smith 1244ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx) 1245da3a660dSBarry Smith { 1246416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1247416022c9SBarry Smith IS iscol = a->col,isrow = a->row; 12486849ba73SBarry Smith PetscErrorCode ierr; 124904fbf559SBarry Smith PetscInt i, n = A->rmap->n,j; 12505d0c19d7SBarry Smith PetscInt nz; 125104fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j; 125204fbf559SBarry Smith PetscScalar *x,*tmp,sum; 125304fbf559SBarry Smith const PetscScalar *b; 1254dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 1255da3a660dSBarry Smith 12563a40ed3dSBarry Smith PetscFunctionBegin; 125778b31e54SBarry Smith if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 1258da3a660dSBarry Smith 12593649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 12601ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1261416022c9SBarry Smith tmp = a->solve_work; 1262da3a660dSBarry Smith 12633b2fbd54SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 12643b2fbd54SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 1265da3a660dSBarry Smith 1266da3a660dSBarry Smith /* forward solve the lower triangular */ 1267da3a660dSBarry Smith tmp[0] = b[*r++]; 1268da3a660dSBarry Smith for (i=1; i<n; i++) { 1269010a20caSHong Zhang v = aa + ai[i]; 1270010a20caSHong Zhang vi = aj + ai[i]; 1271416022c9SBarry Smith nz = a->diag[i] - ai[i]; 1272da3a660dSBarry Smith sum = b[*r++]; 127304fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1274da3a660dSBarry Smith tmp[i] = sum; 1275da3a660dSBarry Smith } 1276da3a660dSBarry Smith 1277da3a660dSBarry Smith /* backward solve the upper triangular */ 1278da3a660dSBarry Smith for (i=n-1; i>=0; i--) { 1279010a20caSHong Zhang v = aa + a->diag[i] + 1; 1280010a20caSHong Zhang vi = aj + a->diag[i] + 1; 1281416022c9SBarry Smith nz = ai[i+1] - a->diag[i] - 1; 1282da3a660dSBarry Smith sum = tmp[i]; 128304fbf559SBarry Smith for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 1284010a20caSHong Zhang tmp[i] = sum*aa[a->diag[i]]; 1285da3a660dSBarry Smith x[*c--] += tmp[i]; 1286da3a660dSBarry Smith } 1287da3a660dSBarry Smith 12883b2fbd54SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 12893b2fbd54SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 12903649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 12911ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1292dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 12933a40ed3dSBarry Smith PetscFunctionReturn(0); 1294da3a660dSBarry Smith } 129504fbf559SBarry Smith 129635233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx) 12973c0119dfSShri Abhyankar { 12983c0119dfSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 12993c0119dfSShri Abhyankar IS iscol = a->col,isrow = a->row; 13003c0119dfSShri Abhyankar PetscErrorCode ierr; 13013c0119dfSShri Abhyankar PetscInt i, n = A->rmap->n,j; 13023c0119dfSShri Abhyankar PetscInt nz; 13033c0119dfSShri Abhyankar const PetscInt *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag; 13043c0119dfSShri Abhyankar PetscScalar *x,*tmp,sum; 13053c0119dfSShri Abhyankar const PetscScalar *b; 13063c0119dfSShri Abhyankar const MatScalar *aa = a->a,*v; 13073c0119dfSShri Abhyankar 13083c0119dfSShri Abhyankar PetscFunctionBegin; 13093c0119dfSShri Abhyankar if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);} 13103c0119dfSShri Abhyankar 13113649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 13123c0119dfSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 13133c0119dfSShri Abhyankar tmp = a->solve_work; 13143c0119dfSShri Abhyankar 13153c0119dfSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13163c0119dfSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 13173c0119dfSShri Abhyankar 13183c0119dfSShri Abhyankar /* forward solve the lower triangular */ 13193c0119dfSShri Abhyankar tmp[0] = b[r[0]]; 13203c0119dfSShri Abhyankar v = aa; 13213c0119dfSShri Abhyankar vi = aj; 13223c0119dfSShri Abhyankar for (i=1; i<n; i++) { 13233c0119dfSShri Abhyankar nz = ai[i+1] - ai[i]; 13243c0119dfSShri Abhyankar sum = b[r[i]]; 13253c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 13263c0119dfSShri Abhyankar tmp[i] = sum; 13272205254eSKarl Rupp v += nz; 13282205254eSKarl Rupp vi += nz; 13293c0119dfSShri Abhyankar } 13303c0119dfSShri Abhyankar 13313c0119dfSShri Abhyankar /* backward solve the upper triangular */ 13323c0119dfSShri Abhyankar v = aa + adiag[n-1]; 13333c0119dfSShri Abhyankar vi = aj + adiag[n-1]; 13343c0119dfSShri Abhyankar for (i=n-1; i>=0; i--) { 13353c0119dfSShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 13363c0119dfSShri Abhyankar sum = tmp[i]; 13373c0119dfSShri Abhyankar for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]]; 13383c0119dfSShri Abhyankar tmp[i] = sum*v[nz]; 13393c0119dfSShri Abhyankar x[c[i]] += tmp[i]; 13403c0119dfSShri Abhyankar v += nz+1; vi += nz+1; 13413c0119dfSShri Abhyankar } 13423c0119dfSShri Abhyankar 13433c0119dfSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13443c0119dfSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 13453649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 13463c0119dfSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 13473c0119dfSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 13483c0119dfSShri Abhyankar PetscFunctionReturn(0); 13493c0119dfSShri Abhyankar } 13503c0119dfSShri Abhyankar 1351ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx) 1352da3a660dSBarry Smith { 1353416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 13542235e667SBarry Smith IS iscol = a->col,isrow = a->row; 13556849ba73SBarry Smith PetscErrorCode ierr; 135604fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 135704fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 135804fbf559SBarry Smith PetscInt nz; 135904fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1360dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 136104fbf559SBarry Smith const PetscScalar *b; 1362da3a660dSBarry Smith 13633a40ed3dSBarry Smith PetscFunctionBegin; 13643649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1365*fdc842d1SBarry Smith ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr); 1366416022c9SBarry Smith tmp = a->solve_work; 1367da3a660dSBarry Smith 13682235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 13692235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1370da3a660dSBarry Smith 1371da3a660dSBarry Smith /* copy the b into temp work space according to permutation */ 13722235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1373da3a660dSBarry Smith 1374da3a660dSBarry Smith /* forward solve the U^T */ 1375da3a660dSBarry Smith for (i=0; i<n; i++) { 1376010a20caSHong Zhang v = aa + diag[i]; 1377010a20caSHong Zhang vi = aj + diag[i] + 1; 1378f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 1379c38d4ed2SBarry Smith s1 = tmp[i]; 1380ef66eb69SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 138104fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1382c38d4ed2SBarry Smith tmp[i] = s1; 1383da3a660dSBarry Smith } 1384da3a660dSBarry Smith 1385da3a660dSBarry Smith /* backward solve the L^T */ 1386da3a660dSBarry Smith for (i=n-1; i>=0; i--) { 1387010a20caSHong Zhang v = aa + diag[i] - 1; 1388010a20caSHong Zhang vi = aj + diag[i] - 1; 1389f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 1390f1af5d2fSBarry Smith s1 = tmp[i]; 139104fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 1392da3a660dSBarry Smith } 1393da3a660dSBarry Smith 1394da3a660dSBarry Smith /* copy tmp into x according to permutation */ 1395591294e4SBarry Smith for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1396da3a660dSBarry Smith 13972235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 13982235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 13993649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1400*fdc842d1SBarry Smith ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr); 1401da3a660dSBarry Smith 1402dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 14033a40ed3dSBarry Smith PetscFunctionReturn(0); 1404da3a660dSBarry Smith } 1405da3a660dSBarry Smith 140635233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx) 1407d1fa9404SShri Abhyankar { 1408d1fa9404SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1409d1fa9404SShri Abhyankar IS iscol = a->col,isrow = a->row; 1410d1fa9404SShri Abhyankar PetscErrorCode ierr; 1411d1fa9404SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1412d1fa9404SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1413d1fa9404SShri Abhyankar PetscInt nz; 1414d1fa9404SShri Abhyankar PetscScalar *x,*tmp,s1; 1415d1fa9404SShri Abhyankar const MatScalar *aa = a->a,*v; 1416d1fa9404SShri Abhyankar const PetscScalar *b; 1417d1fa9404SShri Abhyankar 1418d1fa9404SShri Abhyankar PetscFunctionBegin; 14193649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1420*fdc842d1SBarry Smith ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr); 1421d1fa9404SShri Abhyankar tmp = a->solve_work; 1422d1fa9404SShri Abhyankar 1423d1fa9404SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1424d1fa9404SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1425d1fa9404SShri Abhyankar 1426d1fa9404SShri Abhyankar /* copy the b into temp work space according to permutation */ 1427d1fa9404SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1428d1fa9404SShri Abhyankar 1429d1fa9404SShri Abhyankar /* forward solve the U^T */ 1430d1fa9404SShri Abhyankar for (i=0; i<n; i++) { 1431d1fa9404SShri Abhyankar v = aa + adiag[i+1] + 1; 1432d1fa9404SShri Abhyankar vi = aj + adiag[i+1] + 1; 1433d1fa9404SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1434d1fa9404SShri Abhyankar s1 = tmp[i]; 1435d1fa9404SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1436d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1437d1fa9404SShri Abhyankar tmp[i] = s1; 1438d1fa9404SShri Abhyankar } 1439d1fa9404SShri Abhyankar 1440d1fa9404SShri Abhyankar /* backward solve the L^T */ 1441d1fa9404SShri Abhyankar for (i=n-1; i>=0; i--) { 1442d1fa9404SShri Abhyankar v = aa + ai[i]; 1443d1fa9404SShri Abhyankar vi = aj + ai[i]; 1444d1fa9404SShri Abhyankar nz = ai[i+1] - ai[i]; 1445d1fa9404SShri Abhyankar s1 = tmp[i]; 1446d1fa9404SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1447d1fa9404SShri Abhyankar } 1448d1fa9404SShri Abhyankar 1449d1fa9404SShri Abhyankar /* copy tmp into x according to permutation */ 1450d1fa9404SShri Abhyankar for (i=0; i<n; i++) x[r[i]] = tmp[i]; 1451d1fa9404SShri Abhyankar 1452d1fa9404SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1453d1fa9404SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 14543649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1455*fdc842d1SBarry Smith ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr); 1456d1fa9404SShri Abhyankar 1457d1fa9404SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1458d1fa9404SShri Abhyankar PetscFunctionReturn(0); 1459d1fa9404SShri Abhyankar } 1460d1fa9404SShri Abhyankar 1461ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx) 1462da3a660dSBarry Smith { 1463416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 14642235e667SBarry Smith IS iscol = a->col,isrow = a->row; 14656849ba73SBarry Smith PetscErrorCode ierr; 146604fbf559SBarry Smith const PetscInt *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi; 146704fbf559SBarry Smith PetscInt i,n = A->rmap->n,j; 146804fbf559SBarry Smith PetscInt nz; 146904fbf559SBarry Smith PetscScalar *x,*tmp,s1; 1470dd6ea824SBarry Smith const MatScalar *aa = a->a,*v; 147104fbf559SBarry Smith const PetscScalar *b; 14726abc6512SBarry Smith 14733a40ed3dSBarry Smith PetscFunctionBegin; 147423bc6035SMatthew Knepley if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 14753649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 14761ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1477416022c9SBarry Smith tmp = a->solve_work; 14786abc6512SBarry Smith 14792235e667SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 14802235e667SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 14816abc6512SBarry Smith 14826abc6512SBarry Smith /* copy the b into temp work space according to permutation */ 14832235e667SBarry Smith for (i=0; i<n; i++) tmp[i] = b[c[i]]; 14846abc6512SBarry Smith 14856abc6512SBarry Smith /* forward solve the U^T */ 14866abc6512SBarry Smith for (i=0; i<n; i++) { 1487010a20caSHong Zhang v = aa + diag[i]; 1488010a20caSHong Zhang vi = aj + diag[i] + 1; 1489f1af5d2fSBarry Smith nz = ai[i+1] - diag[i] - 1; 149004fbf559SBarry Smith s1 = tmp[i]; 149104fbf559SBarry Smith s1 *= (*v++); /* multiply by inverse of diagonal entry */ 149204fbf559SBarry Smith for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 149304fbf559SBarry Smith tmp[i] = s1; 14946abc6512SBarry Smith } 14956abc6512SBarry Smith 14966abc6512SBarry Smith /* backward solve the L^T */ 14976abc6512SBarry Smith for (i=n-1; i>=0; i--) { 1498010a20caSHong Zhang v = aa + diag[i] - 1; 1499010a20caSHong Zhang vi = aj + diag[i] - 1; 1500f1af5d2fSBarry Smith nz = diag[i] - ai[i]; 150104fbf559SBarry Smith s1 = tmp[i]; 150204fbf559SBarry Smith for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j]; 15036abc6512SBarry Smith } 15046abc6512SBarry Smith 15056abc6512SBarry Smith /* copy tmp into x according to permutation */ 15066abc6512SBarry Smith for (i=0; i<n; i++) x[r[i]] += tmp[i]; 15076abc6512SBarry Smith 15082235e667SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 15092235e667SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 15103649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 15111ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 15126abc6512SBarry Smith 151304fbf559SBarry Smith ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 15143a40ed3dSBarry Smith PetscFunctionReturn(0); 1515da3a660dSBarry Smith } 151604fbf559SBarry Smith 151735233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx) 1518533e6011SShri Abhyankar { 1519533e6011SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1520533e6011SShri Abhyankar IS iscol = a->col,isrow = a->row; 1521533e6011SShri Abhyankar PetscErrorCode ierr; 1522533e6011SShri Abhyankar const PetscInt *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi; 1523533e6011SShri Abhyankar PetscInt i,n = A->rmap->n,j; 1524533e6011SShri Abhyankar PetscInt nz; 1525533e6011SShri Abhyankar PetscScalar *x,*tmp,s1; 1526533e6011SShri Abhyankar const MatScalar *aa = a->a,*v; 1527533e6011SShri Abhyankar const PetscScalar *b; 1528533e6011SShri Abhyankar 1529533e6011SShri Abhyankar PetscFunctionBegin; 1530533e6011SShri Abhyankar if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);} 15313649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 1532533e6011SShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1533533e6011SShri Abhyankar tmp = a->solve_work; 1534533e6011SShri Abhyankar 1535533e6011SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1536533e6011SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1537533e6011SShri Abhyankar 1538533e6011SShri Abhyankar /* copy the b into temp work space according to permutation */ 1539533e6011SShri Abhyankar for (i=0; i<n; i++) tmp[i] = b[c[i]]; 1540533e6011SShri Abhyankar 1541533e6011SShri Abhyankar /* forward solve the U^T */ 1542533e6011SShri Abhyankar for (i=0; i<n; i++) { 1543533e6011SShri Abhyankar v = aa + adiag[i+1] + 1; 1544533e6011SShri Abhyankar vi = aj + adiag[i+1] + 1; 1545533e6011SShri Abhyankar nz = adiag[i] - adiag[i+1] - 1; 1546533e6011SShri Abhyankar s1 = tmp[i]; 1547533e6011SShri Abhyankar s1 *= v[nz]; /* multiply by inverse of diagonal entry */ 1548533e6011SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1549533e6011SShri Abhyankar tmp[i] = s1; 1550533e6011SShri Abhyankar } 1551533e6011SShri Abhyankar 1552533e6011SShri Abhyankar 1553533e6011SShri Abhyankar /* backward solve the L^T */ 1554533e6011SShri Abhyankar for (i=n-1; i>=0; i--) { 1555533e6011SShri Abhyankar v = aa + ai[i]; 1556533e6011SShri Abhyankar vi = aj + ai[i]; 1557533e6011SShri Abhyankar nz = ai[i+1] - ai[i]; 1558533e6011SShri Abhyankar s1 = tmp[i]; 1559c5e3b2a3SShri Abhyankar for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j]; 1560533e6011SShri Abhyankar } 1561533e6011SShri Abhyankar 1562533e6011SShri Abhyankar /* copy tmp into x according to permutation */ 1563533e6011SShri Abhyankar for (i=0; i<n; i++) x[r[i]] += tmp[i]; 1564533e6011SShri Abhyankar 1565533e6011SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1566533e6011SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 15673649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1568533e6011SShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1569533e6011SShri Abhyankar 1570533e6011SShri Abhyankar ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr); 1571533e6011SShri Abhyankar PetscFunctionReturn(0); 1572533e6011SShri Abhyankar } 1573533e6011SShri Abhyankar 15749e25ed09SBarry Smith /* ----------------------------------------------------------------*/ 1575b47f5a65SHong Zhang 15768fc3a347SHong Zhang /* 15778a76ed4fSHong Zhang ilu() under revised new data structure. 1578813a1f2bSShri Abhyankar Factored arrays bj and ba are stored as 1579813a1f2bSShri Abhyankar L(0,:), L(1,:), ...,L(n-1,:), U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:) 1580813a1f2bSShri Abhyankar 1581813a1f2bSShri Abhyankar bi=fact->i is an array of size n+1, in which 1582813a1f2bSShri Abhyankar bi+ 1583baabb860SHong Zhang bi[i]: points to 1st entry of L(i,:),i=0,...,n-1 1584baabb860SHong Zhang bi[n]: points to L(n-1,n-1)+1 1585baabb860SHong Zhang 1586813a1f2bSShri Abhyankar bdiag=fact->diag is an array of size n+1,in which 1587baabb860SHong Zhang bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1 1588a24f213cSHong Zhang bdiag[n]: points to entry of U(n-1,0)-1 1589813a1f2bSShri Abhyankar 1590813a1f2bSShri Abhyankar U(i,:) contains bdiag[i] as its last entry, i.e., 1591813a1f2bSShri Abhyankar U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 1592813a1f2bSShri Abhyankar */ 159335233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1594813a1f2bSShri Abhyankar { 1595813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1596813a1f2bSShri Abhyankar PetscErrorCode ierr; 1597bbd65245SShri Abhyankar const PetscInt n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag; 1598bbd65245SShri Abhyankar PetscInt i,j,k=0,nz,*bi,*bj,*bdiag; 15991df811f5SHong Zhang IS isicol; 1600813a1f2bSShri Abhyankar 1601813a1f2bSShri Abhyankar PetscFunctionBegin; 16021df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 1603813a1f2bSShri Abhyankar ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr); 1604813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 1605813a1f2bSShri Abhyankar 1606813a1f2bSShri Abhyankar /* allocate matrix arrays for new data structure */ 1607dcca6d9dSJed Brown ierr = PetscMalloc3(ai[n]+1,&b->a,ai[n]+1,&b->j,n+1,&b->i);CHKERRQ(ierr); 16083bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 16092205254eSKarl Rupp 1610813a1f2bSShri Abhyankar b->singlemalloc = PETSC_TRUE; 1611813a1f2bSShri Abhyankar if (!b->diag) { 1612854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&b->diag);CHKERRQ(ierr); 16133bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr); 1614813a1f2bSShri Abhyankar } 1615813a1f2bSShri Abhyankar bdiag = b->diag; 1616813a1f2bSShri Abhyankar 1617813a1f2bSShri Abhyankar if (n > 0) { 1618813a1f2bSShri Abhyankar ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr); 1619813a1f2bSShri Abhyankar } 1620813a1f2bSShri Abhyankar 1621813a1f2bSShri Abhyankar /* set bi and bj with new data structure */ 1622813a1f2bSShri Abhyankar bi = b->i; 1623813a1f2bSShri Abhyankar bj = b->j; 1624813a1f2bSShri Abhyankar 1625813a1f2bSShri Abhyankar /* L part */ 1626e60cf9a0SBarry Smith bi[0] = 0; 1627813a1f2bSShri Abhyankar for (i=0; i<n; i++) { 1628813a1f2bSShri Abhyankar nz = adiag[i] - ai[i]; 1629813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nz; 1630813a1f2bSShri Abhyankar aj = a->j + ai[i]; 1631813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 1632bbd65245SShri Abhyankar /* *bj = aj[j]; bj++; */ 1633bbd65245SShri Abhyankar bj[k++] = aj[j]; 1634813a1f2bSShri Abhyankar } 1635813a1f2bSShri Abhyankar } 1636813a1f2bSShri Abhyankar 1637813a1f2bSShri Abhyankar /* U part */ 163862fbd6afSShri Abhyankar bdiag[n] = bi[n]-1; 1639813a1f2bSShri Abhyankar for (i=n-1; i>=0; i--) { 1640813a1f2bSShri Abhyankar nz = ai[i+1] - adiag[i] - 1; 1641813a1f2bSShri Abhyankar aj = a->j + adiag[i] + 1; 1642813a1f2bSShri Abhyankar for (j=0; j<nz; j++) { 1643bbd65245SShri Abhyankar /* *bj = aj[j]; bj++; */ 1644bbd65245SShri Abhyankar bj[k++] = aj[j]; 1645813a1f2bSShri Abhyankar } 1646813a1f2bSShri Abhyankar /* diag[i] */ 1647bbd65245SShri Abhyankar /* *bj = i; bj++; */ 1648bbd65245SShri Abhyankar bj[k++] = i; 1649a24f213cSHong Zhang bdiag[i] = bdiag[i+1] + nz + 1; 1650813a1f2bSShri Abhyankar } 16511df811f5SHong Zhang 1652d5f3da31SBarry Smith fact->factortype = MAT_FACTOR_ILU; 16531df811f5SHong Zhang fact->info.factor_mallocs = 0; 16541df811f5SHong Zhang fact->info.fill_ratio_given = info->fill; 16551df811f5SHong Zhang fact->info.fill_ratio_needed = 1.0; 165635233d99SShri Abhyankar fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1657a02bda8eSBarry Smith ierr = MatSeqAIJCheckInode_FactorLU(fact);CHKERRQ(ierr); 16581df811f5SHong Zhang 16591df811f5SHong Zhang b = (Mat_SeqAIJ*)(fact)->data; 16601df811f5SHong Zhang b->row = isrow; 16611df811f5SHong Zhang b->col = iscol; 16621df811f5SHong Zhang b->icol = isicol; 1663854ce69bSBarry Smith ierr = PetscMalloc1(fact->rmap->n+1,&b->solve_work);CHKERRQ(ierr); 16641df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 16651df811f5SHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1666813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1667813a1f2bSShri Abhyankar } 1668813a1f2bSShri Abhyankar 166935233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 1670813a1f2bSShri Abhyankar { 1671813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 1672813a1f2bSShri Abhyankar IS isicol; 1673813a1f2bSShri Abhyankar PetscErrorCode ierr; 1674813a1f2bSShri Abhyankar const PetscInt *r,*ic; 16751df811f5SHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j; 1676813a1f2bSShri Abhyankar PetscInt *bi,*cols,nnz,*cols_lvl; 1677813a1f2bSShri Abhyankar PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 1678813a1f2bSShri Abhyankar PetscInt i,levels,diagonal_fill; 1679035f4963SHong Zhang PetscBool col_identity,row_identity,missing; 1680813a1f2bSShri Abhyankar PetscReal f; 16810298fd71SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=NULL; 1682813a1f2bSShri Abhyankar PetscBT lnkbt; 1683813a1f2bSShri Abhyankar PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 16840298fd71SBarry Smith PetscFreeSpaceList free_space =NULL,current_space=NULL; 16850298fd71SBarry Smith PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL; 1686813a1f2bSShri Abhyankar 1687813a1f2bSShri Abhyankar PetscFunctionBegin; 1688035f4963SHong Zhang 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); 1689035f4963SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 1690035f4963SHong Zhang if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 1691ad04f41aSHong Zhang 1692813a1f2bSShri Abhyankar levels = (PetscInt)info->levels; 1693813a1f2bSShri Abhyankar ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 1694813a1f2bSShri Abhyankar ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 1695813a1f2bSShri Abhyankar if (!levels && row_identity && col_identity) { 1696813a1f2bSShri Abhyankar /* special case: ilu(0) with natural ordering */ 169735233d99SShri Abhyankar ierr = MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info);CHKERRQ(ierr); 1698124a3351SHong Zhang if (a->inode.size) { 1699124a3351SHong Zhang fact->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 1700124a3351SHong Zhang } 1701813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1702813a1f2bSShri Abhyankar } 1703813a1f2bSShri Abhyankar 17041df811f5SHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 1705813a1f2bSShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1706813a1f2bSShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 1707813a1f2bSShri Abhyankar 17081bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 1709854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&bi);CHKERRQ(ierr); 1710854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr); 1711e60cf9a0SBarry Smith bi[0] = bdiag[0] = 0; 1712dcca6d9dSJed Brown ierr = PetscMalloc2(n,&bj_ptr,n,&bjlvl_ptr);CHKERRQ(ierr); 1713813a1f2bSShri Abhyankar 1714813a1f2bSShri Abhyankar /* create a linked list for storing column indices of the active row */ 1715813a1f2bSShri Abhyankar nlnk = n + 1; 1716813a1f2bSShri Abhyankar ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1717813a1f2bSShri Abhyankar 1718813a1f2bSShri Abhyankar /* initial FreeSpace size is f*(ai[n]+1) */ 17191df811f5SHong Zhang f = info->fill; 17201df811f5SHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 17212f18eb33SBarry Smith ierr = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr); 1722813a1f2bSShri Abhyankar current_space = free_space; 17232f18eb33SBarry Smith ierr = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space_lvl);CHKERRQ(ierr); 1724813a1f2bSShri Abhyankar current_space_lvl = free_space_lvl; 1725813a1f2bSShri Abhyankar for (i=0; i<n; i++) { 1726813a1f2bSShri Abhyankar nzi = 0; 1727813a1f2bSShri Abhyankar /* copy current row into linked list */ 1728813a1f2bSShri Abhyankar nnz = ai[r[i]+1] - ai[r[i]]; 1729e32f2f54SBarry 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); 1730813a1f2bSShri Abhyankar cols = aj + ai[r[i]]; 1731813a1f2bSShri Abhyankar lnk[i] = -1; /* marker to indicate if diagonal exists */ 1732813a1f2bSShri Abhyankar ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 1733813a1f2bSShri Abhyankar nzi += nlnk; 1734813a1f2bSShri Abhyankar 1735813a1f2bSShri Abhyankar /* make sure diagonal entry is included */ 1736813a1f2bSShri Abhyankar if (diagonal_fill && lnk[i] == -1) { 1737813a1f2bSShri Abhyankar fm = n; 1738813a1f2bSShri Abhyankar while (lnk[fm] < i) fm = lnk[fm]; 1739813a1f2bSShri Abhyankar lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 1740813a1f2bSShri Abhyankar lnk[fm] = i; 1741813a1f2bSShri Abhyankar lnk_lvl[i] = 0; 1742813a1f2bSShri Abhyankar nzi++; dcount++; 1743813a1f2bSShri Abhyankar } 1744813a1f2bSShri Abhyankar 1745813a1f2bSShri Abhyankar /* add pivot rows into the active row */ 1746813a1f2bSShri Abhyankar nzbd = 0; 1747813a1f2bSShri Abhyankar prow = lnk[n]; 1748813a1f2bSShri Abhyankar while (prow < i) { 1749813a1f2bSShri Abhyankar nnz = bdiag[prow]; 1750813a1f2bSShri Abhyankar cols = bj_ptr[prow] + nnz + 1; 1751813a1f2bSShri Abhyankar cols_lvl = bjlvl_ptr[prow] + nnz + 1; 1752813a1f2bSShri Abhyankar nnz = bi[prow+1] - bi[prow] - nnz - 1; 1753813a1f2bSShri Abhyankar ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 1754813a1f2bSShri Abhyankar nzi += nlnk; 1755813a1f2bSShri Abhyankar prow = lnk[prow]; 1756813a1f2bSShri Abhyankar nzbd++; 1757813a1f2bSShri Abhyankar } 1758813a1f2bSShri Abhyankar bdiag[i] = nzbd; 1759813a1f2bSShri Abhyankar bi[i+1] = bi[i] + nzi; 1760813a1f2bSShri Abhyankar /* if free space is not available, make more free space */ 1761813a1f2bSShri Abhyankar if (current_space->local_remaining<nzi) { 1762f91af8c7SBarry Smith nnz = PetscIntMultTruncate(2,PetscIntMultTruncate(nzi,n - i)); /* estimated and max additional space needed */ 1763813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1764813a1f2bSShri Abhyankar ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 1765813a1f2bSShri Abhyankar reallocs++; 1766813a1f2bSShri Abhyankar } 1767813a1f2bSShri Abhyankar 1768813a1f2bSShri Abhyankar /* copy data into free_space and free_space_lvl, then initialize lnk */ 1769813a1f2bSShri Abhyankar ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 1770813a1f2bSShri Abhyankar bj_ptr[i] = current_space->array; 1771813a1f2bSShri Abhyankar bjlvl_ptr[i] = current_space_lvl->array; 1772813a1f2bSShri Abhyankar 1773813a1f2bSShri Abhyankar /* make sure the active row i has diagonal entry */ 177465e19b50SBarry 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); 1775813a1f2bSShri Abhyankar 1776813a1f2bSShri Abhyankar current_space->array += nzi; 1777813a1f2bSShri Abhyankar current_space->local_used += nzi; 1778813a1f2bSShri Abhyankar current_space->local_remaining -= nzi; 1779813a1f2bSShri Abhyankar current_space_lvl->array += nzi; 1780813a1f2bSShri Abhyankar current_space_lvl->local_used += nzi; 1781813a1f2bSShri Abhyankar current_space_lvl->local_remaining -= nzi; 1782813a1f2bSShri Abhyankar } 1783813a1f2bSShri Abhyankar 1784813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1785813a1f2bSShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 1786813a1f2bSShri Abhyankar /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */ 1787854ce69bSBarry Smith ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr); 17882ce24eb6SHong Zhang ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); 1789813a1f2bSShri Abhyankar 1790813a1f2bSShri Abhyankar ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1791813a1f2bSShri Abhyankar ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 17920e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 1793813a1f2bSShri Abhyankar 1794813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO) 1795813a1f2bSShri Abhyankar { 1796aef85c9fSShri Abhyankar PetscReal af = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 179757622a8eSBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr); 179857622a8eSBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr); 179957622a8eSBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%g);\n",(double)af);CHKERRQ(ierr); 1800813a1f2bSShri Abhyankar ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1801813a1f2bSShri Abhyankar if (diagonal_fill) { 1802955c1f14SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals\n",dcount);CHKERRQ(ierr); 1803813a1f2bSShri Abhyankar } 1804813a1f2bSShri Abhyankar } 1805813a1f2bSShri Abhyankar #endif 1806813a1f2bSShri Abhyankar /* put together the new matrix */ 18070298fd71SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 18083bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)fact,(PetscObject)isicol);CHKERRQ(ierr); 1809813a1f2bSShri Abhyankar b = (Mat_SeqAIJ*)(fact)->data; 18102205254eSKarl Rupp 1811813a1f2bSShri Abhyankar b->free_a = PETSC_TRUE; 1812813a1f2bSShri Abhyankar b->free_ij = PETSC_TRUE; 1813813a1f2bSShri Abhyankar b->singlemalloc = PETSC_FALSE; 18142205254eSKarl Rupp 1815854ce69bSBarry Smith ierr = PetscMalloc1(bdiag[0]+1,&b->a);CHKERRQ(ierr); 18162205254eSKarl Rupp 1817813a1f2bSShri Abhyankar b->j = bj; 1818813a1f2bSShri Abhyankar b->i = bi; 1819813a1f2bSShri Abhyankar b->diag = bdiag; 1820813a1f2bSShri Abhyankar b->ilen = 0; 1821813a1f2bSShri Abhyankar b->imax = 0; 1822813a1f2bSShri Abhyankar b->row = isrow; 1823813a1f2bSShri Abhyankar b->col = iscol; 1824813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1825813a1f2bSShri Abhyankar ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 1826813a1f2bSShri Abhyankar b->icol = isicol; 18272205254eSKarl Rupp 1828854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr); 1829813a1f2bSShri Abhyankar /* In b structure: Free imax, ilen, old a, old j. 1830813a1f2bSShri Abhyankar Allocate bdiag, solve_work, new a, new j */ 18313bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 1832f268cba8SShri Abhyankar b->maxnz = b->nz = bdiag[0]+1; 18332205254eSKarl Rupp 1834813a1f2bSShri Abhyankar (fact)->info.factor_mallocs = reallocs; 1835813a1f2bSShri Abhyankar (fact)->info.fill_ratio_given = f; 1836f268cba8SShri Abhyankar (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]); 183735233d99SShri Abhyankar (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ; 1838f2344125SBarry Smith if (a->inode.size) { 1839f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode; 1840f2344125SBarry Smith } 1841a02bda8eSBarry Smith ierr = MatSeqAIJCheckInode_FactorLU(fact);CHKERRQ(ierr); 1842813a1f2bSShri Abhyankar PetscFunctionReturn(0); 1843813a1f2bSShri Abhyankar } 1844813a1f2bSShri Abhyankar 1845ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info) 18466516239fSHong Zhang { 18476516239fSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b; 18486516239fSHong Zhang IS isicol; 18496516239fSHong Zhang PetscErrorCode ierr; 18506516239fSHong Zhang const PetscInt *r,*ic; 1851ece542f9SHong Zhang PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j; 18526516239fSHong Zhang PetscInt *bi,*cols,nnz,*cols_lvl; 18536516239fSHong Zhang PetscInt *bdiag,prow,fm,nzbd,reallocs=0,dcount=0; 18546516239fSHong Zhang PetscInt i,levels,diagonal_fill; 1855ace3abfcSBarry Smith PetscBool col_identity,row_identity; 18566516239fSHong Zhang PetscReal f; 18570298fd71SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=NULL; 18586516239fSHong Zhang PetscBT lnkbt; 18596516239fSHong Zhang PetscInt nzi,*bj,**bj_ptr,**bjlvl_ptr; 18600298fd71SBarry Smith PetscFreeSpaceList free_space =NULL,current_space=NULL; 18610298fd71SBarry Smith PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL; 1862ace3abfcSBarry Smith PetscBool missing; 18636516239fSHong Zhang 18646516239fSHong Zhang PetscFunctionBegin; 1865e32f2f54SBarry 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); 1866ece542f9SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 1867ece542f9SHong Zhang if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 1868ece542f9SHong Zhang 18696516239fSHong Zhang f = info->fill; 18706516239fSHong Zhang levels = (PetscInt)info->levels; 18716516239fSHong Zhang diagonal_fill = (PetscInt)info->diagonal_fill; 18722205254eSKarl Rupp 18736516239fSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 18746516239fSHong Zhang 18756516239fSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 18766516239fSHong Zhang ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr); 18776516239fSHong Zhang if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */ 1878f77e22a1SHong Zhang ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr); 18792205254eSKarl Rupp 1880ad04f41aSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 1881f2344125SBarry Smith if (a->inode.size) { 1882f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 1883f2344125SBarry Smith } 1884d5f3da31SBarry Smith fact->factortype = MAT_FACTOR_ILU; 1885719d5645SBarry Smith (fact)->info.factor_mallocs = 0; 1886719d5645SBarry Smith (fact)->info.fill_ratio_given = info->fill; 1887719d5645SBarry Smith (fact)->info.fill_ratio_needed = 1.0; 18882205254eSKarl Rupp 1889719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 189008480c60SBarry Smith b->row = isrow; 189108480c60SBarry Smith b->col = iscol; 189282bf6240SBarry Smith b->icol = isicol; 1893854ce69bSBarry Smith ierr = PetscMalloc1((fact)->rmap->n+1,&b->solve_work);CHKERRQ(ierr); 1894c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 1895c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 18963a40ed3dSBarry Smith PetscFunctionReturn(0); 189708480c60SBarry Smith } 189808480c60SBarry Smith 1899898183ecSLois Curfman McInnes ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1900898183ecSLois Curfman McInnes ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 19019e25ed09SBarry Smith 19021bfa06eaSJed Brown /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */ 1903854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&bi);CHKERRQ(ierr); 1904854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr); 1905a6df321fSShri Abhyankar bi[0] = bdiag[0] = 0; 19066cf997b0SBarry Smith 1907dcca6d9dSJed Brown ierr = PetscMalloc2(n,&bj_ptr,n,&bjlvl_ptr);CHKERRQ(ierr); 19085a8e39fbSHong Zhang 19095a8e39fbSHong Zhang /* create a linked list for storing column indices of the active row */ 19105a8e39fbSHong Zhang nlnk = n + 1; 19115a8e39fbSHong Zhang ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19125a8e39fbSHong Zhang 19135a8e39fbSHong Zhang /* initial FreeSpace size is f*(ai[n]+1) */ 19142f18eb33SBarry Smith ierr = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space);CHKERRQ(ierr); 19155a8e39fbSHong Zhang current_space = free_space; 19162f18eb33SBarry Smith ierr = PetscFreeSpaceGet(PetscRealIntMultTruncate(f,ai[n]+1),&free_space_lvl);CHKERRQ(ierr); 19175a8e39fbSHong Zhang current_space_lvl = free_space_lvl; 19185a8e39fbSHong Zhang 19195a8e39fbSHong Zhang for (i=0; i<n; i++) { 19205a8e39fbSHong Zhang nzi = 0; 19216cf997b0SBarry Smith /* copy current row into linked list */ 19225a8e39fbSHong Zhang nnz = ai[r[i]+1] - ai[r[i]]; 1923e32f2f54SBarry 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); 19245a8e39fbSHong Zhang cols = aj + ai[r[i]]; 19255a8e39fbSHong Zhang lnk[i] = -1; /* marker to indicate if diagonal exists */ 19265a8e39fbSHong Zhang ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 19275a8e39fbSHong Zhang nzi += nlnk; 19286cf997b0SBarry Smith 19296cf997b0SBarry Smith /* make sure diagonal entry is included */ 19305a8e39fbSHong Zhang if (diagonal_fill && lnk[i] == -1) { 19316cf997b0SBarry Smith fm = n; 19325a8e39fbSHong Zhang while (lnk[fm] < i) fm = lnk[fm]; 19335a8e39fbSHong Zhang lnk[i] = lnk[fm]; /* insert diagonal into linked list */ 19345a8e39fbSHong Zhang lnk[fm] = i; 19355a8e39fbSHong Zhang lnk_lvl[i] = 0; 19365a8e39fbSHong Zhang nzi++; dcount++; 19376cf997b0SBarry Smith } 19386cf997b0SBarry Smith 19395a8e39fbSHong Zhang /* add pivot rows into the active row */ 19405a8e39fbSHong Zhang nzbd = 0; 19415a8e39fbSHong Zhang prow = lnk[n]; 19425a8e39fbSHong Zhang while (prow < i) { 19435a8e39fbSHong Zhang nnz = bdiag[prow]; 19445a8e39fbSHong Zhang cols = bj_ptr[prow] + nnz + 1; 19455a8e39fbSHong Zhang cols_lvl = bjlvl_ptr[prow] + nnz + 1; 19465a8e39fbSHong Zhang nnz = bi[prow+1] - bi[prow] - nnz - 1; 19475a8e39fbSHong Zhang ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr); 19485a8e39fbSHong Zhang nzi += nlnk; 19495a8e39fbSHong Zhang prow = lnk[prow]; 19505a8e39fbSHong Zhang nzbd++; 195156beaf04SBarry Smith } 19525a8e39fbSHong Zhang bdiag[i] = nzbd; 19535a8e39fbSHong Zhang bi[i+1] = bi[i] + nzi; 1954ecf371e4SBarry Smith 19555a8e39fbSHong Zhang /* if free space is not available, make more free space */ 19565a8e39fbSHong Zhang if (current_space->local_remaining<nzi) { 1957f91af8c7SBarry Smith nnz = PetscIntMultTruncate(nzi,n - i); /* estimated and max additional space needed */ 1958a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space);CHKERRQ(ierr); 1959a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(nnz,¤t_space_lvl);CHKERRQ(ierr); 19605a8e39fbSHong Zhang reallocs++; 19615a8e39fbSHong Zhang } 1962ecf371e4SBarry Smith 19635a8e39fbSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 19645a8e39fbSHong Zhang ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 19655a8e39fbSHong Zhang bj_ptr[i] = current_space->array; 19665a8e39fbSHong Zhang bjlvl_ptr[i] = current_space_lvl->array; 19675a8e39fbSHong Zhang 19685a8e39fbSHong Zhang /* make sure the active row i has diagonal entry */ 196965e19b50SBarry 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); 19705a8e39fbSHong Zhang 19715a8e39fbSHong Zhang current_space->array += nzi; 19725a8e39fbSHong Zhang current_space->local_used += nzi; 19735a8e39fbSHong Zhang current_space->local_remaining -= nzi; 19745a8e39fbSHong Zhang current_space_lvl->array += nzi; 19755a8e39fbSHong Zhang current_space_lvl->local_used += nzi; 19765a8e39fbSHong Zhang current_space_lvl->local_remaining -= nzi; 19779e25ed09SBarry Smith } 19785a8e39fbSHong Zhang 1979898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 1980898183ecSLois Curfman McInnes ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 19815a8e39fbSHong Zhang 19825a8e39fbSHong Zhang /* destroy list of free space and other temporary arrays */ 1983854ce69bSBarry Smith ierr = PetscMalloc1(bi[n]+1,&bj);CHKERRQ(ierr); 19846516239fSHong Zhang ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */ 19855a8e39fbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 1986a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 19870e83c824SBarry Smith ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr); 19889e25ed09SBarry Smith 19896cf91177SBarry Smith #if defined(PETSC_USE_INFO) 1990f64065bbSBarry Smith { 19915a8e39fbSHong Zhang PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]); 199257622a8eSBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)f,(double)af);CHKERRQ(ierr); 199357622a8eSBarry Smith ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr); 199457622a8eSBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%g);\n",(double)af);CHKERRQ(ierr); 1995ae15b995SBarry Smith ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr); 1996335d9088SBarry Smith if (diagonal_fill) { 1997955c1f14SBarry Smith ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals\n",dcount);CHKERRQ(ierr); 1998335d9088SBarry Smith } 1999f64065bbSBarry Smith } 200063ba0a88SBarry Smith #endif 2001bbb6d6a8SBarry Smith 20029e25ed09SBarry Smith /* put together the new matrix */ 20030298fd71SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 20043bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)fact,(PetscObject)isicol);CHKERRQ(ierr); 2005719d5645SBarry Smith b = (Mat_SeqAIJ*)(fact)->data; 20062205254eSKarl Rupp 2007e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2008e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 20097c922b88SBarry Smith b->singlemalloc = PETSC_FALSE; 20102205254eSKarl Rupp 2011785e854fSJed Brown ierr = PetscMalloc1(bi[n],&b->a);CHKERRQ(ierr); 20125a8e39fbSHong Zhang b->j = bj; 20135a8e39fbSHong Zhang b->i = bi; 20145a8e39fbSHong Zhang for (i=0; i<n; i++) bdiag[i] += bi[i]; 20155a8e39fbSHong Zhang b->diag = bdiag; 2016416022c9SBarry Smith b->ilen = 0; 2017416022c9SBarry Smith b->imax = 0; 2018416022c9SBarry Smith b->row = isrow; 2019416022c9SBarry Smith b->col = iscol; 2020c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 2021c38d4ed2SBarry Smith ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 202282bf6240SBarry Smith b->icol = isicol; 2023854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr); 2024416022c9SBarry Smith /* In b structure: Free imax, ilen, old a, old j. 20255a8e39fbSHong Zhang Allocate bdiag, solve_work, new a, new j */ 20263bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr); 20275a8e39fbSHong Zhang b->maxnz = b->nz = bi[n]; 20282205254eSKarl Rupp 2029719d5645SBarry Smith (fact)->info.factor_mallocs = reallocs; 2030719d5645SBarry Smith (fact)->info.fill_ratio_given = f; 2031719d5645SBarry Smith (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]); 2032ad04f41aSHong Zhang (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_inplace; 2033f2344125SBarry Smith if (a->inode.size) { 2034f2344125SBarry Smith (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace; 2035d3ac4fa3SBarry Smith } 20363a40ed3dSBarry Smith PetscFunctionReturn(0); 20379e25ed09SBarry Smith } 2038d5d45c9bSBarry Smith 203935233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info) 20405f44c854SHong Zhang { 20415f44c854SHong Zhang Mat C = B; 20425f44c854SHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 20435f44c854SHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 20445f44c854SHong Zhang IS ip=b->row,iip = b->icol; 20455f44c854SHong Zhang PetscErrorCode ierr; 20465f44c854SHong Zhang const PetscInt *rip,*riip; 20475f44c854SHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp; 20485f44c854SHong Zhang PetscInt *ai=a->i,*aj=a->j; 20495f44c854SHong Zhang PetscInt k,jmin,jmax,*c2r,*il,col,nexti,ili,nz; 20505f44c854SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 2051ace3abfcSBarry Smith PetscBool perm_identity; 2052d90ac83dSHong Zhang FactorShiftCtx sctx; 205398a93161SHong Zhang PetscReal rs; 205498a93161SHong Zhang MatScalar d,*v; 205598a93161SHong Zhang 20565f44c854SHong Zhang PetscFunctionBegin; 205798a93161SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 2058d90ac83dSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 205998a93161SHong Zhang 2060f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 206198a93161SHong Zhang sctx.shift_top = info->zeropivot; 206298a93161SHong Zhang for (i=0; i<mbs; i++) { 206398a93161SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 206498a93161SHong Zhang d = (aa)[a->diag[i]]; 206598a93161SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 206698a93161SHong Zhang v = aa+ai[i]; 206798a93161SHong Zhang nz = ai[i+1] - ai[i]; 20682205254eSKarl Rupp for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]); 206998a93161SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 207098a93161SHong Zhang } 207198a93161SHong Zhang sctx.shift_top *= 1.1; 207298a93161SHong Zhang sctx.nshift_max = 5; 207398a93161SHong Zhang sctx.shift_lo = 0.; 207498a93161SHong Zhang sctx.shift_hi = 1.; 207598a93161SHong Zhang } 20765f44c854SHong Zhang 20775f44c854SHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 20785f44c854SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 20795f44c854SHong Zhang 20805f44c854SHong Zhang /* allocate working arrays 20815f44c854SHong Zhang c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col 20825f44c854SHong 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 20835f44c854SHong Zhang */ 2084dcca6d9dSJed Brown ierr = PetscMalloc3(mbs,&rtmp,mbs,&il,mbs,&c2r);CHKERRQ(ierr); 20855f44c854SHong Zhang 208698a93161SHong Zhang do { 208707b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 208898a93161SHong Zhang 2089c5546cabSHong Zhang for (i=0; i<mbs; i++) c2r[i] = mbs; 20902e987568SSatish Balay if (mbs) il[0] = 0; 20915f44c854SHong Zhang 20925f44c854SHong Zhang for (k = 0; k<mbs; k++) { 20935f44c854SHong Zhang /* zero rtmp */ 20945f44c854SHong Zhang nz = bi[k+1] - bi[k]; 20955f44c854SHong Zhang bjtmp = bj + bi[k]; 20965f44c854SHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 20975f44c854SHong Zhang 20985f44c854SHong Zhang /* load in initial unfactored row */ 20995f44c854SHong Zhang bval = ba + bi[k]; 21005f44c854SHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 21015f44c854SHong Zhang for (j = jmin; j < jmax; j++) { 21025f44c854SHong Zhang col = riip[aj[j]]; 21035f44c854SHong Zhang if (col >= k) { /* only take upper triangular entry */ 21045f44c854SHong Zhang rtmp[col] = aa[j]; 21055f44c854SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 21065f44c854SHong Zhang } 21075f44c854SHong Zhang } 210898a93161SHong Zhang /* shift the diagonal of the matrix: ZeropivotApply() */ 210998a93161SHong Zhang rtmp[k] += sctx.shift_amount; /* shift the diagonal of the matrix */ 21105f44c854SHong Zhang 21115f44c854SHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 21125f44c854SHong Zhang dk = rtmp[k]; 21135f44c854SHong Zhang i = c2r[k]; /* first row to be added to k_th row */ 21145f44c854SHong Zhang 21155f44c854SHong Zhang while (i < k) { 21165f44c854SHong Zhang nexti = c2r[i]; /* next row to be added to k_th row */ 21175f44c854SHong Zhang 21185f44c854SHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 21195f44c854SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 21205f44c854SHong Zhang uikdi = -ba[ili]*ba[bdiag[i]]; /* diagonal(k) */ 21215f44c854SHong Zhang dk += uikdi*ba[ili]; /* update diag[k] */ 21225f44c854SHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 21235f44c854SHong Zhang 21245f44c854SHong Zhang /* add multiple of row i to k-th row */ 21255f44c854SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 21265f44c854SHong Zhang if (jmin < jmax) { 21275f44c854SHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 21285f44c854SHong Zhang /* update il and c2r for row i */ 21295f44c854SHong Zhang il[i] = jmin; 21305f44c854SHong Zhang j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i; 21315f44c854SHong Zhang } 21325f44c854SHong Zhang i = nexti; 21335f44c854SHong Zhang } 21345f44c854SHong Zhang 21355f44c854SHong Zhang /* copy data into U(k,:) */ 213698a93161SHong Zhang rs = 0.0; 21375f44c854SHong Zhang jmin = bi[k]; jmax = bi[k+1]-1; 21385f44c854SHong Zhang if (jmin < jmax) { 21395f44c854SHong Zhang for (j=jmin; j<jmax; j++) { 214098a93161SHong Zhang col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]); 21415f44c854SHong Zhang } 21425f44c854SHong Zhang /* add the k-th row into il and c2r */ 21435f44c854SHong Zhang il[k] = jmin; 21445f44c854SHong Zhang i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k; 21455f44c854SHong Zhang } 214698a93161SHong Zhang 214798a93161SHong Zhang /* MatPivotCheck() */ 214898a93161SHong Zhang sctx.rs = rs; 214998a93161SHong Zhang sctx.pv = dk; 21504cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr); 215107b50cabSHong Zhang if (sctx.newshift) break; 215298a93161SHong Zhang dk = sctx.pv; 215398a93161SHong Zhang 215498a93161SHong Zhang ba[bdiag[k]] = 1.0/dk; /* U(k,k) */ 215598a93161SHong Zhang } 215607b50cabSHong Zhang } while (sctx.newshift); 21575f44c854SHong Zhang 21580e83c824SBarry Smith ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr); 21595f44c854SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 21605f44c854SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 21615f44c854SHong Zhang 21625f44c854SHong Zhang ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 21635f44c854SHong Zhang if (perm_identity) { 216435233d99SShri Abhyankar B->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 216535233d99SShri Abhyankar B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 21662169352eSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering; 21672169352eSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering; 21685f44c854SHong Zhang } else { 216935233d99SShri Abhyankar B->ops->solve = MatSolve_SeqSBAIJ_1; 217035233d99SShri Abhyankar B->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 217135233d99SShri Abhyankar B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1; 217235233d99SShri Abhyankar B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1; 21735f44c854SHong Zhang } 21745f44c854SHong Zhang 21755f44c854SHong Zhang C->assembled = PETSC_TRUE; 21765f44c854SHong Zhang C->preallocated = PETSC_TRUE; 21772205254eSKarl Rupp 21785f44c854SHong Zhang ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 217998a93161SHong Zhang 218098a93161SHong Zhang /* MatPivotView() */ 218198a93161SHong Zhang if (sctx.nshift) { 2182f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 218357622a8eSBarry 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,(double)sctx.shift_amount,(double)sctx.shift_fraction,(double)sctx.shift_top);CHKERRQ(ierr); 2184f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 218557622a8eSBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr); 2186f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) { 218757622a8eSBarry Smith ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %g\n",sctx.nshift,(double)info->shiftamount);CHKERRQ(ierr); 218898a93161SHong Zhang } 218998a93161SHong Zhang } 21905f44c854SHong Zhang PetscFunctionReturn(0); 21915f44c854SHong Zhang } 21925f44c854SHong Zhang 21930a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info) 2194a6175056SHong Zhang { 2195719d5645SBarry Smith Mat C = B; 21960968510aSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 21972ed133dbSHong Zhang Mat_SeqSBAIJ *b=(Mat_SeqSBAIJ*)C->data; 21989bfd6278SHong Zhang IS ip=b->row,iip = b->icol; 21992ed133dbSHong Zhang PetscErrorCode ierr; 22005d0c19d7SBarry Smith const PetscInt *rip,*riip; 2201c5546cabSHong Zhang PetscInt i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp; 22022ed133dbSHong Zhang PetscInt *ai=a->i,*aj=a->j; 22032ed133dbSHong Zhang PetscInt k,jmin,jmax,*jl,*il,col,nexti,ili,nz; 2204622e2028SHong Zhang MatScalar *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi; 2205ace3abfcSBarry Smith PetscBool perm_identity; 22060e95ead3SHong Zhang FactorShiftCtx sctx; 22070e95ead3SHong Zhang PetscReal rs; 22080e95ead3SHong Zhang MatScalar d,*v; 2209d5d45c9bSBarry Smith 2210a6175056SHong Zhang PetscFunctionBegin; 22110e95ead3SHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 22120e95ead3SHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 22130e95ead3SHong Zhang 22140e95ead3SHong Zhang if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 22150e95ead3SHong Zhang sctx.shift_top = info->zeropivot; 22160e95ead3SHong Zhang for (i=0; i<mbs; i++) { 22170e95ead3SHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 22180e95ead3SHong Zhang d = (aa)[a->diag[i]]; 22190e95ead3SHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 22200e95ead3SHong Zhang v = aa+ai[i]; 22210e95ead3SHong Zhang nz = ai[i+1] - ai[i]; 22222205254eSKarl Rupp for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]); 22230e95ead3SHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 22240e95ead3SHong Zhang } 22250e95ead3SHong Zhang sctx.shift_top *= 1.1; 22260e95ead3SHong Zhang sctx.nshift_max = 5; 22270e95ead3SHong Zhang sctx.shift_lo = 0.; 22280e95ead3SHong Zhang sctx.shift_hi = 1.; 22290e95ead3SHong Zhang } 2230ee45ca4aSHong Zhang 22312ed133dbSHong Zhang ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 22329bfd6278SHong Zhang ierr = ISGetIndices(iip,&riip);CHKERRQ(ierr); 22332ed133dbSHong Zhang 22342ed133dbSHong Zhang /* initialization */ 2235dcca6d9dSJed Brown ierr = PetscMalloc3(mbs,&rtmp,mbs,&il,mbs,&jl);CHKERRQ(ierr); 22360e95ead3SHong Zhang 22372ed133dbSHong Zhang do { 223807b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 22390e95ead3SHong Zhang 2240c5546cabSHong Zhang for (i=0; i<mbs; i++) jl[i] = mbs; 2241c5546cabSHong Zhang il[0] = 0; 22422ed133dbSHong Zhang 22432ed133dbSHong Zhang for (k = 0; k<mbs; k++) { 2244c5546cabSHong Zhang /* zero rtmp */ 2245c5546cabSHong Zhang nz = bi[k+1] - bi[k]; 2246c5546cabSHong Zhang bjtmp = bj + bi[k]; 2247c5546cabSHong Zhang for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0; 2248c5546cabSHong Zhang 2249c05c3958SHong Zhang bval = ba + bi[k]; 22502ed133dbSHong Zhang /* initialize k-th row by the perm[k]-th row of A */ 22512ed133dbSHong Zhang jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 22522ed133dbSHong Zhang for (j = jmin; j < jmax; j++) { 22539bfd6278SHong Zhang col = riip[aj[j]]; 22542ed133dbSHong Zhang if (col >= k) { /* only take upper triangular entry */ 22552ed133dbSHong Zhang rtmp[col] = aa[j]; 2256c05c3958SHong Zhang *bval++ = 0.0; /* for in-place factorization */ 22572ed133dbSHong Zhang } 22582ed133dbSHong Zhang } 225939e3d630SHong Zhang /* shift the diagonal of the matrix */ 2260540703acSHong Zhang if (sctx.nshift) rtmp[k] += sctx.shift_amount; 22612ed133dbSHong Zhang 22622ed133dbSHong Zhang /* modify k-th row by adding in those rows i with U(i,k)!=0 */ 22632ed133dbSHong Zhang dk = rtmp[k]; 22642ed133dbSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 22652ed133dbSHong Zhang 22662ed133dbSHong Zhang while (i < k) { 22672ed133dbSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 22682ed133dbSHong Zhang 22692ed133dbSHong Zhang /* compute multiplier, update diag(k) and U(i,k) */ 22702ed133dbSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 22712ed133dbSHong Zhang uikdi = -ba[ili]*ba[bi[i]]; /* diagonal(k) */ 22722ed133dbSHong Zhang dk += uikdi*ba[ili]; 22732ed133dbSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 22742ed133dbSHong Zhang 22752ed133dbSHong Zhang /* add multiple of row i to k-th row */ 22762ed133dbSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 22772ed133dbSHong Zhang if (jmin < jmax) { 22782ed133dbSHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 22792ed133dbSHong Zhang /* update il and jl for row i */ 22802ed133dbSHong Zhang il[i] = jmin; 22812ed133dbSHong Zhang j = bj[jmin]; jl[i] = jl[j]; jl[j] = i; 22822ed133dbSHong Zhang } 22832ed133dbSHong Zhang i = nexti; 22842ed133dbSHong Zhang } 22852ed133dbSHong Zhang 22860697b6caSHong Zhang /* shift the diagonals when zero pivot is detected */ 22870697b6caSHong Zhang /* compute rs=sum of abs(off-diagonal) */ 22880697b6caSHong Zhang rs = 0.0; 22893c9e8598SHong Zhang jmin = bi[k]+1; 22903c9e8598SHong Zhang nz = bi[k+1] - jmin; 22913c9e8598SHong Zhang bcol = bj + jmin; 229204fbf559SBarry Smith for (j=0; j<nz; j++) { 229304fbf559SBarry Smith rs += PetscAbsScalar(rtmp[bcol[j]]); 22943c9e8598SHong Zhang } 2295540703acSHong Zhang 2296540703acSHong Zhang sctx.rs = rs; 2297540703acSHong Zhang sctx.pv = dk; 22984cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,k);CHKERRQ(ierr); 229907b50cabSHong Zhang if (sctx.newshift) break; 23000e95ead3SHong Zhang dk = sctx.pv; 23013c9e8598SHong Zhang 23023c9e8598SHong Zhang /* copy data into U(k,:) */ 230339e3d630SHong Zhang ba[bi[k]] = 1.0/dk; /* U(k,k) */ 230439e3d630SHong Zhang jmin = bi[k]+1; jmax = bi[k+1]; 230539e3d630SHong Zhang if (jmin < jmax) { 230639e3d630SHong Zhang for (j=jmin; j<jmax; j++) { 2307c6ae9bfcSHong Zhang col = bj[j]; ba[j] = rtmp[col]; 23083c9e8598SHong Zhang } 230939e3d630SHong Zhang /* add the k-th row into il and jl */ 23103c9e8598SHong Zhang il[k] = jmin; 23113c9e8598SHong Zhang i = bj[jmin]; jl[k] = jl[i]; jl[i] = k; 23123c9e8598SHong Zhang } 23133c9e8598SHong Zhang } 231407b50cabSHong Zhang } while (sctx.newshift); 23150e95ead3SHong Zhang 23160e83c824SBarry Smith ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr); 231739e3d630SHong Zhang ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 23189bfd6278SHong Zhang ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr); 2319db4efbfdSBarry Smith 2320db4efbfdSBarry Smith ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr); 2321db4efbfdSBarry Smith if (perm_identity) { 23220a3c351aSHong Zhang B->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23230a3c351aSHong Zhang B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23240a3c351aSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 23250a3c351aSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace; 2326db4efbfdSBarry Smith } else { 23270a3c351aSHong Zhang B->ops->solve = MatSolve_SeqSBAIJ_1_inplace; 23280a3c351aSHong Zhang B->ops->solvetranspose = MatSolve_SeqSBAIJ_1_inplace; 23290a3c351aSHong Zhang B->ops->forwardsolve = MatForwardSolve_SeqSBAIJ_1_inplace; 23300a3c351aSHong Zhang B->ops->backwardsolve = MatBackwardSolve_SeqSBAIJ_1_inplace; 2331db4efbfdSBarry Smith } 2332db4efbfdSBarry Smith 23333c9e8598SHong Zhang C->assembled = PETSC_TRUE; 23343c9e8598SHong Zhang C->preallocated = PETSC_TRUE; 23352205254eSKarl Rupp 2336d0f46423SBarry Smith ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr); 2337540703acSHong Zhang if (sctx.nshift) { 2338f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 233957622a8eSBarry Smith ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr); 2340f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 234157622a8eSBarry Smith ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr); 23420697b6caSHong Zhang } 23433c9e8598SHong Zhang } 23443c9e8598SHong Zhang PetscFunctionReturn(0); 23453c9e8598SHong Zhang } 2346a6175056SHong Zhang 23478a76ed4fSHong Zhang /* 23488a76ed4fSHong Zhang icc() under revised new data structure. 23498a76ed4fSHong Zhang Factored arrays bj and ba are stored as 23508a76ed4fSHong Zhang U(0,:),...,U(i,:),U(n-1,:) 23518a76ed4fSHong Zhang 23528a76ed4fSHong Zhang ui=fact->i is an array of size n+1, in which 23538a76ed4fSHong Zhang ui+ 23548a76ed4fSHong Zhang ui[i]: points to 1st entry of U(i,:),i=0,...,n-1 23558a76ed4fSHong Zhang ui[n]: points to U(n-1,n-1)+1 23568a76ed4fSHong Zhang 23578a76ed4fSHong Zhang udiag=fact->diag is an array of size n,in which 23588a76ed4fSHong Zhang udiag[i]: points to diagonal of U(i,:), i=0,...,n-1 23598a76ed4fSHong Zhang 23608a76ed4fSHong Zhang U(i,:) contains udiag[i] as its last entry, i.e., 23618a76ed4fSHong Zhang U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i]) 23628a76ed4fSHong Zhang */ 23638a76ed4fSHong Zhang 236435233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2365a6175056SHong Zhang { 23660968510aSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2367ed59401aSHong Zhang Mat_SeqSBAIJ *b; 2368dfbe8321SBarry Smith PetscErrorCode ierr; 2369ace3abfcSBarry Smith PetscBool perm_identity,missing; 23705f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 23715d0c19d7SBarry Smith const PetscInt *rip,*riip; 2372ed59401aSHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 23730298fd71SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=NULL,d; 23745a8e39fbSHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 2375ed59401aSHong Zhang PetscReal fill =info->fill,levels=info->levels; 23760298fd71SBarry Smith PetscFreeSpaceList free_space =NULL,current_space=NULL; 23770298fd71SBarry Smith PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL; 23787a48dd6fSHong Zhang PetscBT lnkbt; 2379b635d36bSHong Zhang IS iperm; 2380a6175056SHong Zhang 2381a6175056SHong Zhang PetscFunctionBegin; 2382e32f2f54SBarry 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); 238358ebbce7SBarry Smith ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 2384e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 2385653a6975SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 2386b635d36bSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 23877a48dd6fSHong Zhang 2388854ce69bSBarry Smith ierr = PetscMalloc1(am+1,&ui);CHKERRQ(ierr); 2389854ce69bSBarry Smith ierr = PetscMalloc1(am+1,&udiag);CHKERRQ(ierr); 239039e3d630SHong Zhang ui[0] = 0; 239139e3d630SHong Zhang 23928a76ed4fSHong Zhang /* ICC(0) without matrix ordering: simply rearrange column indices */ 2393622e2028SHong Zhang if (!levels && perm_identity) { 23945f44c854SHong Zhang for (i=0; i<am; i++) { 2395c5546cabSHong Zhang ncols = ai[i+1] - a->diag[i]; 2396c5546cabSHong Zhang ui[i+1] = ui[i] + ncols; 2397c5546cabSHong Zhang udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */ 2398dc1e2a3fSHong Zhang } 2399854ce69bSBarry Smith ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr); 2400dc1e2a3fSHong Zhang cols = uj; 2401dc1e2a3fSHong Zhang for (i=0; i<am; i++) { 24025f44c854SHong Zhang aj = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */ 2403dc1e2a3fSHong Zhang ncols = ai[i+1] - a->diag[i] -1; 24045f44c854SHong Zhang for (j=0; j<ncols; j++) *cols++ = aj[j]; 24055f44c854SHong Zhang *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */ 24065f44c854SHong Zhang } 24075f44c854SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 24085f44c854SHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 24095f44c854SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 24105f44c854SHong Zhang 24115f44c854SHong Zhang /* initialization */ 2412854ce69bSBarry Smith ierr = PetscMalloc1(am+1,&ajtmp);CHKERRQ(ierr); 24135f44c854SHong Zhang 24145f44c854SHong Zhang /* jl: linked list for storing indices of the pivot rows 24155f44c854SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 2416dcca6d9dSJed Brown ierr = PetscMalloc4(am,&uj_ptr,am,&uj_lvl_ptr,am,&jl,am,&il);CHKERRQ(ierr); 24175f44c854SHong Zhang for (i=0; i<am; i++) { 24185f44c854SHong Zhang jl[i] = am; il[i] = 0; 24195f44c854SHong Zhang } 24205f44c854SHong Zhang 24215f44c854SHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 24225f44c854SHong Zhang nlnk = am + 1; 24235f44c854SHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 24245f44c854SHong Zhang 242595b5ac22SHong Zhang /* initial FreeSpace size is fill*(ai[am]+am)/2 */ 24262f18eb33SBarry Smith ierr = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space);CHKERRQ(ierr); 24275f44c854SHong Zhang current_space = free_space; 24282f18eb33SBarry Smith ierr = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space_lvl);CHKERRQ(ierr); 24295f44c854SHong Zhang current_space_lvl = free_space_lvl; 24305f44c854SHong Zhang 24315f44c854SHong Zhang for (k=0; k<am; k++) { /* for each active row k */ 24325f44c854SHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 24335f44c854SHong Zhang nzk = 0; 24345f44c854SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2435e32f2f54SBarry 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); 24365f44c854SHong Zhang ncols_upper = 0; 24375f44c854SHong Zhang for (j=0; j<ncols; j++) { 24385f44c854SHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 24395f44c854SHong Zhang if (riip[i] >= k) { /* only take upper triangular entry */ 24405f44c854SHong Zhang ajtmp[ncols_upper] = i; 24415f44c854SHong Zhang ncols_upper++; 24425f44c854SHong Zhang } 24435f44c854SHong Zhang } 24445f44c854SHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 24455f44c854SHong Zhang nzk += nlnk; 24465f44c854SHong Zhang 24475f44c854SHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 24485f44c854SHong Zhang prow = jl[k]; /* 1st pivot row */ 24495f44c854SHong Zhang 24505f44c854SHong Zhang while (prow < k) { 24515f44c854SHong Zhang nextprow = jl[prow]; 24525f44c854SHong Zhang 24535f44c854SHong Zhang /* merge prow into k-th row */ 24545f44c854SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 24555f44c854SHong Zhang jmax = ui[prow+1]; 24565f44c854SHong Zhang ncols = jmax-jmin; 24575f44c854SHong Zhang i = jmin - ui[prow]; 24585f44c854SHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 24595f44c854SHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 24605f44c854SHong Zhang j = *(uj - 1); 24615f44c854SHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 24625f44c854SHong Zhang nzk += nlnk; 24635f44c854SHong Zhang 24645f44c854SHong Zhang /* update il and jl for prow */ 24655f44c854SHong Zhang if (jmin < jmax) { 24665f44c854SHong Zhang il[prow] = jmin; 24675f44c854SHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 24685f44c854SHong Zhang } 24695f44c854SHong Zhang prow = nextprow; 24705f44c854SHong Zhang } 24715f44c854SHong Zhang 24725f44c854SHong Zhang /* if free space is not available, make more free space */ 24735f44c854SHong Zhang if (current_space->local_remaining<nzk) { 24745f44c854SHong Zhang i = am - k + 1; /* num of unfactored rows */ 2475f91af8c7SBarry Smith i = PetscIntMultTruncate(i,PetscMin(nzk, i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 24765f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 24775f44c854SHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 24785f44c854SHong Zhang reallocs++; 24795f44c854SHong Zhang } 24805f44c854SHong Zhang 24815f44c854SHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 2482e32f2f54SBarry Smith if (nzk == 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 24835f44c854SHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 24845f44c854SHong Zhang 24855f44c854SHong Zhang /* add the k-th row into il and jl */ 24865f44c854SHong Zhang if (nzk > 1) { 24875f44c854SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 24885f44c854SHong Zhang jl[k] = jl[i]; jl[i] = k; 24895f44c854SHong Zhang il[k] = ui[k] + 1; 24905f44c854SHong Zhang } 24915f44c854SHong Zhang uj_ptr[k] = current_space->array; 24925f44c854SHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 24935f44c854SHong Zhang 24945f44c854SHong Zhang current_space->array += nzk; 24955f44c854SHong Zhang current_space->local_used += nzk; 24965f44c854SHong Zhang current_space->local_remaining -= nzk; 24975f44c854SHong Zhang 24985f44c854SHong Zhang current_space_lvl->array += nzk; 24995f44c854SHong Zhang current_space_lvl->local_used += nzk; 25005f44c854SHong Zhang current_space_lvl->local_remaining -= nzk; 25015f44c854SHong Zhang 25025f44c854SHong Zhang ui[k+1] = ui[k] + nzk; 25035f44c854SHong Zhang } 25045f44c854SHong Zhang 25055f44c854SHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 25065f44c854SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 25070e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 25085f44c854SHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 25095f44c854SHong Zhang 25109263d837SHong Zhang /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */ 2511854ce69bSBarry Smith ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr); 251235233d99SShri Abhyankar ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */ 25135f44c854SHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 25145f44c854SHong Zhang ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 25155f44c854SHong Zhang 25165f44c854SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 25175f44c854SHong Zhang 25185f44c854SHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 25195f44c854SHong Zhang b = (Mat_SeqSBAIJ*)(fact)->data; 25205f44c854SHong Zhang b->singlemalloc = PETSC_FALSE; 25212205254eSKarl Rupp 2522854ce69bSBarry Smith ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr); 25232205254eSKarl Rupp 25245f44c854SHong Zhang b->j = uj; 25255f44c854SHong Zhang b->i = ui; 25265f44c854SHong Zhang b->diag = udiag; 25277f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 25285f44c854SHong Zhang b->ilen = 0; 25295f44c854SHong Zhang b->imax = 0; 25305f44c854SHong Zhang b->row = perm; 25315f44c854SHong Zhang b->col = perm; 25325f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25335f44c854SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 25345f44c854SHong Zhang b->icol = iperm; 25355f44c854SHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 25362205254eSKarl Rupp 2537854ce69bSBarry Smith ierr = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr); 25383bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 25392205254eSKarl Rupp 25405f44c854SHong Zhang b->maxnz = b->nz = ui[am]; 25415f44c854SHong Zhang b->free_a = PETSC_TRUE; 25425f44c854SHong Zhang b->free_ij = PETSC_TRUE; 25435f44c854SHong Zhang 2544f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2545f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 25465f44c854SHong Zhang if (ai[am] != 0) { 25476a69fef8SHong Zhang /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */ 254895b5ac22SHong Zhang fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am); 25495f44c854SHong Zhang } else { 2550f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 25515f44c854SHong Zhang } 25529263d837SHong Zhang #if defined(PETSC_USE_INFO) 25539263d837SHong Zhang if (ai[am] != 0) { 25549263d837SHong Zhang PetscReal af = fact->info.fill_ratio_needed; 255557622a8eSBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr); 255657622a8eSBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr); 255757622a8eSBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr); 25589263d837SHong Zhang } else { 25599263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 25609263d837SHong Zhang } 25619263d837SHong Zhang #endif 256235233d99SShri Abhyankar fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 25635f44c854SHong Zhang PetscFunctionReturn(0); 25645f44c854SHong Zhang } 25655f44c854SHong Zhang 2566ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 25675f44c854SHong Zhang { 25685f44c854SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 25695f44c854SHong Zhang Mat_SeqSBAIJ *b; 25705f44c854SHong Zhang PetscErrorCode ierr; 2571ace3abfcSBarry Smith PetscBool perm_identity,missing; 25725f44c854SHong Zhang PetscInt reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag; 25735f44c854SHong Zhang const PetscInt *rip,*riip; 25745f44c854SHong Zhang PetscInt jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow; 25750298fd71SBarry Smith PetscInt nlnk,*lnk,*lnk_lvl=NULL,d; 25765f44c854SHong Zhang PetscInt ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr; 25775f44c854SHong Zhang PetscReal fill =info->fill,levels=info->levels; 25780298fd71SBarry Smith PetscFreeSpaceList free_space =NULL,current_space=NULL; 25790298fd71SBarry Smith PetscFreeSpaceList free_space_lvl=NULL,current_space_lvl=NULL; 25805f44c854SHong Zhang PetscBT lnkbt; 25815f44c854SHong Zhang IS iperm; 25825f44c854SHong Zhang 25835f44c854SHong Zhang PetscFunctionBegin; 2584e32f2f54SBarry 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); 25855f44c854SHong Zhang ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr); 2586e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d); 25875f44c854SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 25885f44c854SHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 25895f44c854SHong Zhang 2590854ce69bSBarry Smith ierr = PetscMalloc1(am+1,&ui);CHKERRQ(ierr); 2591854ce69bSBarry Smith ierr = PetscMalloc1(am+1,&udiag);CHKERRQ(ierr); 25925f44c854SHong Zhang ui[0] = 0; 25935f44c854SHong Zhang 25945f44c854SHong Zhang /* ICC(0) without matrix ordering: simply copies fill pattern */ 25955f44c854SHong Zhang if (!levels && perm_identity) { 25965f44c854SHong Zhang 25975f44c854SHong Zhang for (i=0; i<am; i++) { 25985f44c854SHong Zhang ui[i+1] = ui[i] + ai[i+1] - a->diag[i]; 25995f44c854SHong Zhang udiag[i] = ui[i]; 2600ed59401aSHong Zhang } 2601854ce69bSBarry Smith ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr); 260239e3d630SHong Zhang cols = uj; 2603ed59401aSHong Zhang for (i=0; i<am; i++) { 2604ed59401aSHong Zhang aj = a->j + a->diag[i]; 260539e3d630SHong Zhang ncols = ui[i+1] - ui[i]; 260639e3d630SHong Zhang for (j=0; j<ncols; j++) *cols++ = *aj++; 2607ed59401aSHong Zhang } 260839e3d630SHong Zhang } else { /* case: levels>0 || (levels=0 && !perm_identity) */ 2609b635d36bSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 2610b635d36bSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 2611b635d36bSHong Zhang 26127a48dd6fSHong Zhang /* initialization */ 2613854ce69bSBarry Smith ierr = PetscMalloc1(am+1,&ajtmp);CHKERRQ(ierr); 26147a48dd6fSHong Zhang 26157a48dd6fSHong Zhang /* jl: linked list for storing indices of the pivot rows 26167a48dd6fSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 2617dcca6d9dSJed Brown ierr = PetscMalloc4(am,&uj_ptr,am,&uj_lvl_ptr,am,&jl,am,&il);CHKERRQ(ierr); 26187a48dd6fSHong Zhang for (i=0; i<am; i++) { 26197a48dd6fSHong Zhang jl[i] = am; il[i] = 0; 26207a48dd6fSHong Zhang } 26217a48dd6fSHong Zhang 26227a48dd6fSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 26237a48dd6fSHong Zhang nlnk = am + 1; 26242ed133dbSHong Zhang ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 26257a48dd6fSHong Zhang 26267a48dd6fSHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 26272f18eb33SBarry Smith ierr = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space);CHKERRQ(ierr); 26287a48dd6fSHong Zhang current_space = free_space; 26292f18eb33SBarry Smith ierr = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space_lvl);CHKERRQ(ierr); 26307a48dd6fSHong Zhang current_space_lvl = free_space_lvl; 26317a48dd6fSHong Zhang 26327a48dd6fSHong Zhang for (k=0; k<am; k++) { /* for each active row k */ 26337a48dd6fSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 26347a48dd6fSHong Zhang nzk = 0; 2635622e2028SHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2636e32f2f54SBarry 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); 2637622e2028SHong Zhang ncols_upper = 0; 2638622e2028SHong Zhang for (j=0; j<ncols; j++) { 2639b635d36bSHong Zhang i = *(aj + ai[rip[k]] + j); /* unpermuted column index */ 2640b635d36bSHong Zhang if (riip[i] >= k) { /* only take upper triangular entry */ 26415a8e39fbSHong Zhang ajtmp[ncols_upper] = i; 2642622e2028SHong Zhang ncols_upper++; 2643622e2028SHong Zhang } 2644622e2028SHong Zhang } 2645b635d36bSHong Zhang ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr); 26467a48dd6fSHong Zhang nzk += nlnk; 26477a48dd6fSHong Zhang 26487a48dd6fSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 26497a48dd6fSHong Zhang prow = jl[k]; /* 1st pivot row */ 26507a48dd6fSHong Zhang 26517a48dd6fSHong Zhang while (prow < k) { 26527a48dd6fSHong Zhang nextprow = jl[prow]; 26537a48dd6fSHong Zhang 26547a48dd6fSHong Zhang /* merge prow into k-th row */ 26557a48dd6fSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 26567a48dd6fSHong Zhang jmax = ui[prow+1]; 26577a48dd6fSHong Zhang ncols = jmax-jmin; 2658ed59401aSHong Zhang i = jmin - ui[prow]; 2659ed59401aSHong Zhang cols = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 26605a8e39fbSHong Zhang uj = uj_lvl_ptr[prow] + i; /* levels of cols */ 26615a8e39fbSHong Zhang j = *(uj - 1); 26625a8e39fbSHong Zhang ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr); 26637a48dd6fSHong Zhang nzk += nlnk; 26647a48dd6fSHong Zhang 26657a48dd6fSHong Zhang /* update il and jl for prow */ 26667a48dd6fSHong Zhang if (jmin < jmax) { 26677a48dd6fSHong Zhang il[prow] = jmin; 26687a48dd6fSHong Zhang j = *cols; jl[prow] = jl[j]; jl[j] = prow; 26697a48dd6fSHong Zhang } 26707a48dd6fSHong Zhang prow = nextprow; 26717a48dd6fSHong Zhang } 26727a48dd6fSHong Zhang 26737a48dd6fSHong Zhang /* if free space is not available, make more free space */ 26747a48dd6fSHong Zhang if (current_space->local_remaining<nzk) { 26757a48dd6fSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2676f91af8c7SBarry Smith i = PetscIntMultTruncate(i,PetscMin(nzk, i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 2677a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 2678a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space_lvl);CHKERRQ(ierr); 26797a48dd6fSHong Zhang reallocs++; 26807a48dd6fSHong Zhang } 26817a48dd6fSHong Zhang 26827a48dd6fSHong Zhang /* copy data into free_space and free_space_lvl, then initialize lnk */ 268365e19b50SBarry Smith if (!nzk) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k); 26842ed133dbSHong Zhang ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr); 26857a48dd6fSHong Zhang 26867a48dd6fSHong Zhang /* add the k-th row into il and jl */ 268739e3d630SHong Zhang if (nzk > 1) { 26887a48dd6fSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 26897a48dd6fSHong Zhang jl[k] = jl[i]; jl[i] = k; 26907a48dd6fSHong Zhang il[k] = ui[k] + 1; 26917a48dd6fSHong Zhang } 26927a48dd6fSHong Zhang uj_ptr[k] = current_space->array; 26937a48dd6fSHong Zhang uj_lvl_ptr[k] = current_space_lvl->array; 26947a48dd6fSHong Zhang 26957a48dd6fSHong Zhang current_space->array += nzk; 26967a48dd6fSHong Zhang current_space->local_used += nzk; 26977a48dd6fSHong Zhang current_space->local_remaining -= nzk; 26987a48dd6fSHong Zhang 26997a48dd6fSHong Zhang current_space_lvl->array += nzk; 27007a48dd6fSHong Zhang current_space_lvl->local_used += nzk; 27017a48dd6fSHong Zhang current_space_lvl->local_remaining -= nzk; 27027a48dd6fSHong Zhang 27037a48dd6fSHong Zhang ui[k+1] = ui[k] + nzk; 27047a48dd6fSHong Zhang } 27057a48dd6fSHong Zhang 27066cf91177SBarry Smith #if defined(PETSC_USE_INFO) 27077a48dd6fSHong Zhang if (ai[am] != 0) { 270839e3d630SHong Zhang PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]); 270957622a8eSBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr); 271057622a8eSBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr); 271157622a8eSBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr); 27127a48dd6fSHong Zhang } else { 2713ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 27147a48dd6fSHong Zhang } 271563ba0a88SBarry Smith #endif 27167a48dd6fSHong Zhang 27177a48dd6fSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 2718b635d36bSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 27190e83c824SBarry Smith ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr); 27205a8e39fbSHong Zhang ierr = PetscFree(ajtmp);CHKERRQ(ierr); 27217a48dd6fSHong Zhang 27227a48dd6fSHong Zhang /* destroy list of free space and other temporary array(s) */ 2723854ce69bSBarry Smith ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr); 2724a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 27252ed133dbSHong Zhang ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 2726a1a86e44SBarry Smith ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr); 27277a48dd6fSHong Zhang 272839e3d630SHong Zhang } /* end of case: levels>0 || (levels=0 && !perm_identity) */ 272939e3d630SHong Zhang 27307a48dd6fSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 27317a48dd6fSHong Zhang 2732f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 27337a48dd6fSHong Zhang b->singlemalloc = PETSC_FALSE; 27342205254eSKarl Rupp 2735854ce69bSBarry Smith ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr); 27362205254eSKarl Rupp 27377a48dd6fSHong Zhang b->j = uj; 27387a48dd6fSHong Zhang b->i = ui; 27395f44c854SHong Zhang b->diag = udiag; 27407f53bb6cSHong Zhang b->free_diag = PETSC_TRUE; 27417a48dd6fSHong Zhang b->ilen = 0; 27427a48dd6fSHong Zhang b->imax = 0; 27437a48dd6fSHong Zhang b->row = perm; 2744b635d36bSHong Zhang b->col = perm; 27452205254eSKarl Rupp 2746b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 2747b635d36bSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 27482205254eSKarl Rupp 2749b635d36bSHong Zhang b->icol = iperm; 27507a48dd6fSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 2751854ce69bSBarry Smith ierr = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr); 27523bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 27537a48dd6fSHong Zhang b->maxnz = b->nz = ui[am]; 2754e6b907acSBarry Smith b->free_a = PETSC_TRUE; 2755e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 27567a48dd6fSHong Zhang 2757f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2758f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 27597a48dd6fSHong Zhang if (ai[am] != 0) { 2760f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 27617a48dd6fSHong Zhang } else { 2762f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 27637a48dd6fSHong Zhang } 27640a3c351aSHong Zhang fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace; 2765a6175056SHong Zhang PetscFunctionReturn(0); 2766a6175056SHong Zhang } 2767d5d45c9bSBarry Smith 276835233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 27690be760fbSHong Zhang { 27700be760fbSHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 27710be760fbSHong Zhang Mat_SeqSBAIJ *b; 27720be760fbSHong Zhang PetscErrorCode ierr; 27739186b105SHong Zhang PetscBool perm_identity,missing; 27740be760fbSHong Zhang PetscReal fill = info->fill; 27750be760fbSHong Zhang const PetscInt *rip,*riip; 27760be760fbSHong Zhang PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 27770be760fbSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 27780be760fbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag; 27790298fd71SBarry Smith PetscFreeSpaceList free_space=NULL,current_space=NULL; 27800be760fbSHong Zhang PetscBT lnkbt; 27810be760fbSHong Zhang IS iperm; 27820be760fbSHong Zhang 27830be760fbSHong Zhang PetscFunctionBegin; 2784e32f2f54SBarry 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); 27859186b105SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 27869186b105SHong Zhang if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 27879186b105SHong Zhang 27880be760fbSHong Zhang /* check whether perm is the identity mapping */ 27890be760fbSHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 27900be760fbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 27910be760fbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 27920be760fbSHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 27930be760fbSHong Zhang 27940be760fbSHong Zhang /* initialization */ 2795854ce69bSBarry Smith ierr = PetscMalloc1(am+1,&ui);CHKERRQ(ierr); 2796854ce69bSBarry Smith ierr = PetscMalloc1(am+1,&udiag);CHKERRQ(ierr); 27970be760fbSHong Zhang ui[0] = 0; 27980be760fbSHong Zhang 27990be760fbSHong Zhang /* jl: linked list for storing indices of the pivot rows 28000be760fbSHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 2801dcca6d9dSJed Brown ierr = PetscMalloc4(am,&ui_ptr,am,&jl,am,&il,am,&cols);CHKERRQ(ierr); 28020be760fbSHong Zhang for (i=0; i<am; i++) { 28030be760fbSHong Zhang jl[i] = am; il[i] = 0; 28040be760fbSHong Zhang } 28050be760fbSHong Zhang 28060be760fbSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 28070be760fbSHong Zhang nlnk = am + 1; 28080be760fbSHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28090be760fbSHong Zhang 281095b5ac22SHong Zhang /* initial FreeSpace size is fill*(ai[am]+am)/2 */ 28112f18eb33SBarry Smith ierr = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,(ai[am]+am)/2),&free_space);CHKERRQ(ierr); 28120be760fbSHong Zhang current_space = free_space; 28130be760fbSHong Zhang 28140be760fbSHong Zhang for (k=0; k<am; k++) { /* for each active row k */ 28150be760fbSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 28160be760fbSHong Zhang nzk = 0; 28170be760fbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2818e32f2f54SBarry 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); 28190be760fbSHong Zhang ncols_upper = 0; 28200be760fbSHong Zhang for (j=0; j<ncols; j++) { 28210be760fbSHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 28220be760fbSHong Zhang if (i >= k) { /* only take upper triangular entry */ 28230be760fbSHong Zhang cols[ncols_upper] = i; 28240be760fbSHong Zhang ncols_upper++; 28250be760fbSHong Zhang } 28260be760fbSHong Zhang } 28270be760fbSHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28280be760fbSHong Zhang nzk += nlnk; 28290be760fbSHong Zhang 28300be760fbSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 28310be760fbSHong Zhang prow = jl[k]; /* 1st pivot row */ 28320be760fbSHong Zhang 28330be760fbSHong Zhang while (prow < k) { 28340be760fbSHong Zhang nextprow = jl[prow]; 28350be760fbSHong Zhang /* merge prow into k-th row */ 28360be760fbSHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 28370be760fbSHong Zhang jmax = ui[prow+1]; 28380be760fbSHong Zhang ncols = jmax-jmin; 28390be760fbSHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 28400be760fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 28410be760fbSHong Zhang nzk += nlnk; 28420be760fbSHong Zhang 28430be760fbSHong Zhang /* update il and jl for prow */ 28440be760fbSHong Zhang if (jmin < jmax) { 28450be760fbSHong Zhang il[prow] = jmin; 28462205254eSKarl Rupp j = *uj_ptr; 28472205254eSKarl Rupp jl[prow] = jl[j]; 28482205254eSKarl Rupp jl[j] = prow; 28490be760fbSHong Zhang } 28500be760fbSHong Zhang prow = nextprow; 28510be760fbSHong Zhang } 28520be760fbSHong Zhang 28530be760fbSHong Zhang /* if free space is not available, make more free space */ 28540be760fbSHong Zhang if (current_space->local_remaining<nzk) { 28550be760fbSHong Zhang i = am - k + 1; /* num of unfactored rows */ 2856f91af8c7SBarry Smith i = PetscIntMultTruncate(i,PetscMin(nzk,i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 28570be760fbSHong Zhang ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 28580be760fbSHong Zhang reallocs++; 28590be760fbSHong Zhang } 28600be760fbSHong Zhang 28610be760fbSHong Zhang /* copy data into free space, then initialize lnk */ 28620be760fbSHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 28630be760fbSHong Zhang 28640be760fbSHong Zhang /* add the k-th row into il and jl */ 28657b056e98SHong Zhang if (nzk > 1) { 28660be760fbSHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 28670be760fbSHong Zhang jl[k] = jl[i]; jl[i] = k; 28680be760fbSHong Zhang il[k] = ui[k] + 1; 28690be760fbSHong Zhang } 28700be760fbSHong Zhang ui_ptr[k] = current_space->array; 28712205254eSKarl Rupp 28720be760fbSHong Zhang current_space->array += nzk; 28730be760fbSHong Zhang current_space->local_used += nzk; 28740be760fbSHong Zhang current_space->local_remaining -= nzk; 28750be760fbSHong Zhang 28760be760fbSHong Zhang ui[k+1] = ui[k] + nzk; 28770be760fbSHong Zhang } 28780be760fbSHong Zhang 28790be760fbSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 28800be760fbSHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 28810be760fbSHong Zhang ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 28820be760fbSHong Zhang 28839263d837SHong Zhang /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */ 2884854ce69bSBarry Smith ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr); 288535233d99SShri Abhyankar ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */ 28860be760fbSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 28870be760fbSHong Zhang 28880be760fbSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 28890be760fbSHong Zhang 2890f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 28910be760fbSHong Zhang b->singlemalloc = PETSC_FALSE; 28920be760fbSHong Zhang b->free_a = PETSC_TRUE; 28930be760fbSHong Zhang b->free_ij = PETSC_TRUE; 28942205254eSKarl Rupp 2895854ce69bSBarry Smith ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr); 28962205254eSKarl Rupp 28970be760fbSHong Zhang b->j = uj; 28980be760fbSHong Zhang b->i = ui; 28990be760fbSHong Zhang b->diag = udiag; 29000be760fbSHong Zhang b->free_diag = PETSC_TRUE; 29010be760fbSHong Zhang b->ilen = 0; 29020be760fbSHong Zhang b->imax = 0; 29030be760fbSHong Zhang b->row = perm; 29040be760fbSHong Zhang b->col = perm; 290526fbe8dcSKarl Rupp 29060be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29070be760fbSHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 29082205254eSKarl Rupp 29090be760fbSHong Zhang b->icol = iperm; 29100be760fbSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 291126fbe8dcSKarl Rupp 2912854ce69bSBarry Smith ierr = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr); 29133bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 29142205254eSKarl Rupp 29150be760fbSHong Zhang b->maxnz = b->nz = ui[am]; 29160be760fbSHong Zhang 2917f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 2918f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 29190be760fbSHong Zhang if (ai[am] != 0) { 29206a69fef8SHong Zhang /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */ 292195b5ac22SHong Zhang fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am); 29220be760fbSHong Zhang } else { 2923f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 29240be760fbSHong Zhang } 29259263d837SHong Zhang #if defined(PETSC_USE_INFO) 29269263d837SHong Zhang if (ai[am] != 0) { 29279263d837SHong Zhang PetscReal af = fact->info.fill_ratio_needed; 292857622a8eSBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr); 292957622a8eSBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr); 293057622a8eSBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr); 29319263d837SHong Zhang } else { 29329263d837SHong Zhang ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 29339263d837SHong Zhang } 29349263d837SHong Zhang #endif 293535233d99SShri Abhyankar fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ; 29360be760fbSHong Zhang PetscFunctionReturn(0); 29370be760fbSHong Zhang } 29380be760fbSHong Zhang 2939ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info) 2940f76d2b81SHong Zhang { 2941f76d2b81SHong Zhang Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 294210c27e3dSHong Zhang Mat_SeqSBAIJ *b; 2943dfbe8321SBarry Smith PetscErrorCode ierr; 29449186b105SHong Zhang PetscBool perm_identity,missing; 294510c27e3dSHong Zhang PetscReal fill = info->fill; 29465d0c19d7SBarry Smith const PetscInt *rip,*riip; 29475d0c19d7SBarry Smith PetscInt i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow; 294810c27e3dSHong Zhang PetscInt *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow; 29492ed133dbSHong Zhang PetscInt nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr; 29500298fd71SBarry Smith PetscFreeSpaceList free_space=NULL,current_space=NULL; 295110c27e3dSHong Zhang PetscBT lnkbt; 29522ed133dbSHong Zhang IS iperm; 2953f76d2b81SHong Zhang 2954f76d2b81SHong Zhang PetscFunctionBegin; 2955e32f2f54SBarry 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); 29569186b105SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 29579186b105SHong Zhang if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 29589186b105SHong Zhang 29592ed133dbSHong Zhang /* check whether perm is the identity mapping */ 2960f76d2b81SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 29612ed133dbSHong Zhang ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr); 29622ed133dbSHong Zhang ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr); 29639bfd6278SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 296410c27e3dSHong Zhang 296510c27e3dSHong Zhang /* initialization */ 2966854ce69bSBarry Smith ierr = PetscMalloc1(am+1,&ui);CHKERRQ(ierr); 296710c27e3dSHong Zhang ui[0] = 0; 296810c27e3dSHong Zhang 296910c27e3dSHong Zhang /* jl: linked list for storing indices of the pivot rows 29701393bc97SHong Zhang il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */ 2971dcca6d9dSJed Brown ierr = PetscMalloc4(am,&ui_ptr,am,&jl,am,&il,am,&cols);CHKERRQ(ierr); 29721393bc97SHong Zhang for (i=0; i<am; i++) { 29731393bc97SHong Zhang jl[i] = am; il[i] = 0; 2974f76d2b81SHong Zhang } 297510c27e3dSHong Zhang 297610c27e3dSHong Zhang /* create and initialize a linked list for storing column indices of the active row k */ 29771393bc97SHong Zhang nlnk = am + 1; 29781393bc97SHong Zhang ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 297910c27e3dSHong Zhang 29801393bc97SHong Zhang /* initial FreeSpace size is fill*(ai[am]+1) */ 29812f18eb33SBarry Smith ierr = PetscFreeSpaceGet(PetscRealIntMultTruncate(fill,ai[am]+1),&free_space);CHKERRQ(ierr); 298210c27e3dSHong Zhang current_space = free_space; 298310c27e3dSHong Zhang 29841393bc97SHong Zhang for (k=0; k<am; k++) { /* for each active row k */ 298510c27e3dSHong Zhang /* initialize lnk by the column indices of row rip[k] of A */ 298610c27e3dSHong Zhang nzk = 0; 29872ed133dbSHong Zhang ncols = ai[rip[k]+1] - ai[rip[k]]; 2988e32f2f54SBarry 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); 29892ed133dbSHong Zhang ncols_upper = 0; 29902ed133dbSHong Zhang for (j=0; j<ncols; j++) { 29919bfd6278SHong Zhang i = riip[*(aj + ai[rip[k]] + j)]; 29922ed133dbSHong Zhang if (i >= k) { /* only take upper triangular entry */ 29932ed133dbSHong Zhang cols[ncols_upper] = i; 29942ed133dbSHong Zhang ncols_upper++; 29952ed133dbSHong Zhang } 29962ed133dbSHong Zhang } 29971393bc97SHong Zhang ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 299810c27e3dSHong Zhang nzk += nlnk; 299910c27e3dSHong Zhang 300010c27e3dSHong Zhang /* update lnk by computing fill-in for each pivot row to be merged in */ 300110c27e3dSHong Zhang prow = jl[k]; /* 1st pivot row */ 300210c27e3dSHong Zhang 300310c27e3dSHong Zhang while (prow < k) { 300410c27e3dSHong Zhang nextprow = jl[prow]; 300510c27e3dSHong Zhang /* merge prow into k-th row */ 30061393bc97SHong Zhang jmin = il[prow] + 1; /* index of the 2nd nzero entry in U(prow,k:am-1) */ 300710c27e3dSHong Zhang jmax = ui[prow+1]; 300810c27e3dSHong Zhang ncols = jmax-jmin; 30091393bc97SHong Zhang uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */ 30105a8e39fbSHong Zhang ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr); 301110c27e3dSHong Zhang nzk += nlnk; 301210c27e3dSHong Zhang 301310c27e3dSHong Zhang /* update il and jl for prow */ 301410c27e3dSHong Zhang if (jmin < jmax) { 301510c27e3dSHong Zhang il[prow] = jmin; 30162ed133dbSHong Zhang j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow; 301710c27e3dSHong Zhang } 301810c27e3dSHong Zhang prow = nextprow; 301910c27e3dSHong Zhang } 302010c27e3dSHong Zhang 302110c27e3dSHong Zhang /* if free space is not available, make more free space */ 302210c27e3dSHong Zhang if (current_space->local_remaining<nzk) { 30231393bc97SHong Zhang i = am - k + 1; /* num of unfactored rows */ 302410c27e3dSHong Zhang i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */ 3025a1a86e44SBarry Smith ierr = PetscFreeSpaceGet(i,¤t_space);CHKERRQ(ierr); 302610c27e3dSHong Zhang reallocs++; 302710c27e3dSHong Zhang } 302810c27e3dSHong Zhang 302910c27e3dSHong Zhang /* copy data into free space, then initialize lnk */ 30301393bc97SHong Zhang ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr); 303110c27e3dSHong Zhang 303210c27e3dSHong Zhang /* add the k-th row into il and jl */ 303310c27e3dSHong Zhang if (nzk-1 > 0) { 30341393bc97SHong Zhang i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */ 303510c27e3dSHong Zhang jl[k] = jl[i]; jl[i] = k; 303610c27e3dSHong Zhang il[k] = ui[k] + 1; 303710c27e3dSHong Zhang } 30382ed133dbSHong Zhang ui_ptr[k] = current_space->array; 30392205254eSKarl Rupp 304010c27e3dSHong Zhang current_space->array += nzk; 304110c27e3dSHong Zhang current_space->local_used += nzk; 304210c27e3dSHong Zhang current_space->local_remaining -= nzk; 304310c27e3dSHong Zhang 304410c27e3dSHong Zhang ui[k+1] = ui[k] + nzk; 304510c27e3dSHong Zhang } 304610c27e3dSHong Zhang 30476cf91177SBarry Smith #if defined(PETSC_USE_INFO) 30481393bc97SHong Zhang if (ai[am] != 0) { 30491393bc97SHong Zhang PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]); 305057622a8eSBarry Smith ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %g needed %g\n",reallocs,(double)fill,(double)af);CHKERRQ(ierr); 305157622a8eSBarry Smith ierr = PetscInfo1(A,"Run with -pc_factor_fill %g or use \n",(double)af);CHKERRQ(ierr); 305257622a8eSBarry Smith ierr = PetscInfo1(A,"PCFactorSetFill(pc,%g) for best performance.\n",(double)af);CHKERRQ(ierr); 305310c27e3dSHong Zhang } else { 3054ae15b995SBarry Smith ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr); 305510c27e3dSHong Zhang } 305663ba0a88SBarry Smith #endif 305710c27e3dSHong Zhang 305810c27e3dSHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 30599bfd6278SHong Zhang ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr); 30600e83c824SBarry Smith ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr); 306110c27e3dSHong Zhang 306210c27e3dSHong Zhang /* destroy list of free space and other temporary array(s) */ 3063854ce69bSBarry Smith ierr = PetscMalloc1(ui[am]+1,&uj);CHKERRQ(ierr); 3064a1a86e44SBarry Smith ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr); 306510c27e3dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 306610c27e3dSHong Zhang 306710c27e3dSHong Zhang /* put together the new matrix in MATSEQSBAIJ format */ 306810c27e3dSHong Zhang 3069f284f12aSHong Zhang b = (Mat_SeqSBAIJ*)fact->data; 307010c27e3dSHong Zhang b->singlemalloc = PETSC_FALSE; 3071e6b907acSBarry Smith b->free_a = PETSC_TRUE; 3072e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 30732205254eSKarl Rupp 3074854ce69bSBarry Smith ierr = PetscMalloc1(ui[am]+1,&b->a);CHKERRQ(ierr); 30752205254eSKarl Rupp 307610c27e3dSHong Zhang b->j = uj; 307710c27e3dSHong Zhang b->i = ui; 307810c27e3dSHong Zhang b->diag = 0; 307910c27e3dSHong Zhang b->ilen = 0; 308010c27e3dSHong Zhang b->imax = 0; 308110c27e3dSHong Zhang b->row = perm; 30829bfd6278SHong Zhang b->col = perm; 30832205254eSKarl Rupp 30849bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 30859bfd6278SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 30862205254eSKarl Rupp 30879bfd6278SHong Zhang b->icol = iperm; 308810c27e3dSHong Zhang b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */ 30892205254eSKarl Rupp 3090854ce69bSBarry Smith ierr = PetscMalloc1(am+1,&b->solve_work);CHKERRQ(ierr); 30913bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 30921393bc97SHong Zhang b->maxnz = b->nz = ui[am]; 309310c27e3dSHong Zhang 3094f284f12aSHong Zhang fact->info.factor_mallocs = reallocs; 3095f284f12aSHong Zhang fact->info.fill_ratio_given = fill; 30961393bc97SHong Zhang if (ai[am] != 0) { 3097f284f12aSHong Zhang fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]); 309810c27e3dSHong Zhang } else { 3099f284f12aSHong Zhang fact->info.fill_ratio_needed = 0.0; 310010c27e3dSHong Zhang } 31010a3c351aSHong Zhang fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace; 3102f76d2b81SHong Zhang PetscFunctionReturn(0); 3103f76d2b81SHong Zhang } 3104599ef60dSHong Zhang 310535233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx) 3106813a1f2bSShri Abhyankar { 3107813a1f2bSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3108813a1f2bSShri Abhyankar PetscErrorCode ierr; 3109813a1f2bSShri Abhyankar PetscInt n = A->rmap->n; 3110813a1f2bSShri Abhyankar const PetscInt *ai = a->i,*aj = a->j,*adiag = a->diag,*vi; 3111813a1f2bSShri Abhyankar PetscScalar *x,sum; 3112813a1f2bSShri Abhyankar const PetscScalar *b; 3113813a1f2bSShri Abhyankar const MatScalar *aa = a->a,*v; 3114813a1f2bSShri Abhyankar PetscInt i,nz; 3115813a1f2bSShri Abhyankar 3116813a1f2bSShri Abhyankar PetscFunctionBegin; 3117813a1f2bSShri Abhyankar if (!n) PetscFunctionReturn(0); 3118813a1f2bSShri Abhyankar 31193649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 3120*fdc842d1SBarry Smith ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr); 3121813a1f2bSShri Abhyankar 3122813a1f2bSShri Abhyankar /* forward solve the lower triangular */ 3123813a1f2bSShri Abhyankar x[0] = b[0]; 3124813a1f2bSShri Abhyankar v = aa; 3125813a1f2bSShri Abhyankar vi = aj; 3126813a1f2bSShri Abhyankar for (i=1; i<n; i++) { 3127813a1f2bSShri Abhyankar nz = ai[i+1] - ai[i]; 3128813a1f2bSShri Abhyankar sum = b[i]; 3129813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 3130813a1f2bSShri Abhyankar v += nz; 3131813a1f2bSShri Abhyankar vi += nz; 3132813a1f2bSShri Abhyankar x[i] = sum; 3133813a1f2bSShri Abhyankar } 3134813a1f2bSShri Abhyankar 3135813a1f2bSShri Abhyankar /* backward solve the upper triangular */ 313662fbd6afSShri Abhyankar for (i=n-1; i>=0; i--) { 31373c0119dfSShri Abhyankar v = aa + adiag[i+1] + 1; 31383c0119dfSShri Abhyankar vi = aj + adiag[i+1] + 1; 3139813a1f2bSShri Abhyankar nz = adiag[i] - adiag[i+1]-1; 3140813a1f2bSShri Abhyankar sum = x[i]; 3141813a1f2bSShri Abhyankar PetscSparseDenseMinusDot(sum,x,v,vi,nz); 31423c0119dfSShri Abhyankar x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */ 3143813a1f2bSShri Abhyankar } 3144813a1f2bSShri Abhyankar 3145813a1f2bSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 31463649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3147*fdc842d1SBarry Smith ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr); 3148813a1f2bSShri Abhyankar PetscFunctionReturn(0); 3149813a1f2bSShri Abhyankar } 3150813a1f2bSShri Abhyankar 315135233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx) 3152f268cba8SShri Abhyankar { 3153f268cba8SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3154f268cba8SShri Abhyankar IS iscol = a->col,isrow = a->row; 3155f268cba8SShri Abhyankar PetscErrorCode ierr; 3156f268cba8SShri Abhyankar PetscInt i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz; 3157f268cba8SShri Abhyankar const PetscInt *rout,*cout,*r,*c; 3158f268cba8SShri Abhyankar PetscScalar *x,*tmp,sum; 3159f268cba8SShri Abhyankar const PetscScalar *b; 3160f268cba8SShri Abhyankar const MatScalar *aa = a->a,*v; 3161f268cba8SShri Abhyankar 3162f268cba8SShri Abhyankar PetscFunctionBegin; 3163f268cba8SShri Abhyankar if (!n) PetscFunctionReturn(0); 3164f268cba8SShri Abhyankar 31653649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 3166*fdc842d1SBarry Smith ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr); 3167f268cba8SShri Abhyankar tmp = a->solve_work; 3168f268cba8SShri Abhyankar 3169f268cba8SShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 3170f268cba8SShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 3171f268cba8SShri Abhyankar 3172f268cba8SShri Abhyankar /* forward solve the lower triangular */ 3173f268cba8SShri Abhyankar tmp[0] = b[r[0]]; 3174f268cba8SShri Abhyankar v = aa; 3175f268cba8SShri Abhyankar vi = aj; 3176f268cba8SShri Abhyankar for (i=1; i<n; i++) { 3177f268cba8SShri Abhyankar nz = ai[i+1] - ai[i]; 3178f268cba8SShri Abhyankar sum = b[r[i]]; 3179f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3180f268cba8SShri Abhyankar tmp[i] = sum; 3181f268cba8SShri Abhyankar v += nz; vi += nz; 3182f268cba8SShri Abhyankar } 3183f268cba8SShri Abhyankar 3184f268cba8SShri Abhyankar /* backward solve the upper triangular */ 3185f268cba8SShri Abhyankar for (i=n-1; i>=0; i--) { 31863c0119dfSShri Abhyankar v = aa + adiag[i+1]+1; 31873c0119dfSShri Abhyankar vi = aj + adiag[i+1]+1; 3188f268cba8SShri Abhyankar nz = adiag[i]-adiag[i+1]-1; 3189f268cba8SShri Abhyankar sum = tmp[i]; 3190f268cba8SShri Abhyankar PetscSparseDenseMinusDot(sum,tmp,v,vi,nz); 3191f268cba8SShri Abhyankar x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */ 3192f268cba8SShri Abhyankar } 3193f268cba8SShri Abhyankar 3194f268cba8SShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 3195f268cba8SShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 31963649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3197*fdc842d1SBarry Smith ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr); 3198f268cba8SShri Abhyankar ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr); 3199f268cba8SShri Abhyankar PetscFunctionReturn(0); 3200f268cba8SShri Abhyankar } 3201f268cba8SShri Abhyankar 3202fe97e370SBarry Smith /* 3203bebe2cf6SSatish Balay This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer separate functions in the matrix function table for dt factors 3204fe97e370SBarry Smith */ 32051d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact) 32061d098868SHong Zhang { 32071d098868SHong Zhang Mat B = *fact; 3208599ef60dSHong Zhang Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data,*b; 3209599ef60dSHong Zhang IS isicol; 3210599ef60dSHong Zhang PetscErrorCode ierr; 32111d098868SHong Zhang const PetscInt *r,*ic; 32121d098868SHong Zhang PetscInt i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag; 3213dc3a2fd3SHong Zhang PetscInt *bi,*bj,*bdiag,*bdiag_rev; 3214f61a948aSLisandro Dalcin PetscInt row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au; 32151d098868SHong Zhang PetscInt nlnk,*lnk; 32161d098868SHong Zhang PetscBT lnkbt; 3217ace3abfcSBarry Smith PetscBool row_identity,icol_identity; 32188aee2decSHong Zhang MatScalar *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp; 32191d098868SHong Zhang const PetscInt *ics; 32201d098868SHong Zhang PetscInt j,nz,*pj,*bjtmp,k,ncut,*jtmp; 3221a2ea699eSBarry Smith PetscReal dt =info->dt,shift=info->shiftamount; 3222f61a948aSLisandro Dalcin PetscInt dtcount=(PetscInt)info->dtcount,nnz_max; 3223ace3abfcSBarry Smith PetscBool missing; 3224599ef60dSHong Zhang 3225599ef60dSHong Zhang PetscFunctionBegin; 3226f61a948aSLisandro Dalcin if (dt == PETSC_DEFAULT) dt = 0.005; 3227f61a948aSLisandro Dalcin if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax); 3228f61a948aSLisandro Dalcin 32291d098868SHong Zhang /* ------- symbolic factorization, can be reused ---------*/ 32301d098868SHong Zhang ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr); 3231e32f2f54SBarry Smith if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i); 32321d098868SHong Zhang adiag=a->diag; 3233599ef60dSHong Zhang 3234599ef60dSHong Zhang ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr); 3235599ef60dSHong Zhang 3236599ef60dSHong Zhang /* bdiag is location of diagonal in factor */ 3237854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&bdiag);CHKERRQ(ierr); /* becomes b->diag */ 3238854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&bdiag_rev);CHKERRQ(ierr); /* temporary */ 3239599ef60dSHong Zhang 32401d098868SHong Zhang /* allocate row pointers bi */ 3241854ce69bSBarry Smith ierr = PetscMalloc1(2*n+2,&bi);CHKERRQ(ierr); 32421d098868SHong Zhang 32431d098868SHong Zhang /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */ 3244393d3a68SHong Zhang if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */ 32451d098868SHong Zhang nnz_max = ai[n]+2*n*dtcount+2; 32468fc3a347SHong Zhang 3247854ce69bSBarry Smith ierr = PetscMalloc1(nnz_max+1,&bj);CHKERRQ(ierr); 3248854ce69bSBarry Smith ierr = PetscMalloc1(nnz_max+1,&ba);CHKERRQ(ierr); 3249599ef60dSHong Zhang 3250599ef60dSHong Zhang /* put together the new matrix */ 32510298fd71SBarry Smith ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr); 32523bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)isicol);CHKERRQ(ierr); 3253f284f12aSHong Zhang b = (Mat_SeqAIJ*)B->data; 32542205254eSKarl Rupp 3255599ef60dSHong Zhang b->free_a = PETSC_TRUE; 3256599ef60dSHong Zhang b->free_ij = PETSC_TRUE; 3257599ef60dSHong Zhang b->singlemalloc = PETSC_FALSE; 32582205254eSKarl Rupp 3259599ef60dSHong Zhang b->a = ba; 3260599ef60dSHong Zhang b->j = bj; 3261599ef60dSHong Zhang b->i = bi; 3262599ef60dSHong Zhang b->diag = bdiag; 3263599ef60dSHong Zhang b->ilen = 0; 3264599ef60dSHong Zhang b->imax = 0; 3265599ef60dSHong Zhang b->row = isrow; 3266599ef60dSHong Zhang b->col = iscol; 3267599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr); 3268599ef60dSHong Zhang ierr = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr); 3269599ef60dSHong Zhang b->icol = isicol; 3270599ef60dSHong Zhang 3271854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&b->solve_work);CHKERRQ(ierr); 32723bb1ff40SBarry Smith ierr = PetscLogObjectMemory((PetscObject)B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr); 32731d098868SHong Zhang b->maxnz = nnz_max; 3274599ef60dSHong Zhang 3275d5f3da31SBarry Smith B->factortype = MAT_FACTOR_ILUDT; 3276f284f12aSHong Zhang B->info.factor_mallocs = 0; 3277f284f12aSHong Zhang B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]); 32781d098868SHong Zhang /* ------- end of symbolic factorization ---------*/ 3279599ef60dSHong Zhang 3280599ef60dSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 3281599ef60dSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3282599ef60dSHong Zhang ics = ic; 3283599ef60dSHong Zhang 3284599ef60dSHong Zhang /* linked list for storing column indices of the active row */ 3285599ef60dSHong Zhang nlnk = n + 1; 3286599ef60dSHong Zhang ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 32871d098868SHong Zhang 32881d098868SHong Zhang /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */ 3289dcca6d9dSJed Brown ierr = PetscMalloc2(n,&im,n,&jtmp);CHKERRQ(ierr); 32901d098868SHong Zhang /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */ 3291dcca6d9dSJed Brown ierr = PetscMalloc2(n,&rtmp,n,&vtmp);CHKERRQ(ierr); 32920e83c824SBarry Smith ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr); 3293599ef60dSHong Zhang 3294599ef60dSHong Zhang bi[0] = 0; 32958fc3a347SHong Zhang bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */ 3296dc3a2fd3SHong Zhang bdiag_rev[n] = bdiag[0]; 32978fc3a347SHong Zhang bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */ 3298599ef60dSHong Zhang for (i=0; i<n; i++) { 3299599ef60dSHong Zhang /* copy initial fill into linked list */ 33008369b28dSHong Zhang nzi = ai[r[i]+1] - ai[r[i]]; 3301e32f2f54SBarry 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); 33028369b28dSHong Zhang nzi_al = adiag[r[i]] - ai[r[i]]; 33038369b28dSHong Zhang nzi_au = ai[r[i]+1] - adiag[r[i]] -1; 3304599ef60dSHong Zhang ajtmp = aj + ai[r[i]]; 33058369b28dSHong Zhang ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr); 3306599ef60dSHong Zhang 3307599ef60dSHong Zhang /* load in initial (unfactored row) */ 33081d098868SHong Zhang aatmp = a->a + ai[r[i]]; 33098369b28dSHong Zhang for (j=0; j<nzi; j++) { 33101d098868SHong Zhang rtmp[ics[*ajtmp++]] = *aatmp++; 3311599ef60dSHong Zhang } 3312599ef60dSHong Zhang 3313599ef60dSHong Zhang /* add pivot rows into linked list */ 3314599ef60dSHong Zhang row = lnk[n]; 3315599ef60dSHong Zhang while (row < i) { 33161d098868SHong Zhang nzi_bl = bi[row+1] - bi[row] + 1; 33171d098868SHong Zhang bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */ 33181d098868SHong Zhang ierr = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr); 3319599ef60dSHong Zhang nzi += nlnk; 3320599ef60dSHong Zhang row = lnk[row]; 3321599ef60dSHong Zhang } 3322599ef60dSHong Zhang 33238369b28dSHong Zhang /* copy data from lnk into jtmp, then initialize lnk */ 33248369b28dSHong Zhang ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr); 3325599ef60dSHong Zhang 3326599ef60dSHong Zhang /* numerical factorization */ 33278369b28dSHong Zhang bjtmp = jtmp; 3328599ef60dSHong Zhang row = *bjtmp++; /* 1st pivot row */ 3329599ef60dSHong Zhang while (row < i) { 3330599ef60dSHong Zhang pc = rtmp + row; 33313c2a7987SHong Zhang pv = ba + bdiag[row]; /* 1./(diag of the pivot row) */ 33323c2a7987SHong Zhang multiplier = (*pc) * (*pv); 3333599ef60dSHong Zhang *pc = multiplier; 33343c2a7987SHong Zhang if (PetscAbsScalar(*pc) > dt) { /* apply tolerance dropping rule */ 33351d098868SHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 33361d098868SHong Zhang pv = ba + bdiag[row+1] + 1; 33371d098868SHong Zhang /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */ 33381d098868SHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 33391d098868SHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 334028b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 3341599ef60dSHong Zhang } 3342599ef60dSHong Zhang row = *bjtmp++; 3343599ef60dSHong Zhang } 3344599ef60dSHong Zhang 33458369b28dSHong Zhang /* copy sparse rtmp into contiguous vtmp; separate L and U part */ 33468aee2decSHong Zhang diag_tmp = rtmp[i]; /* save diagonal value - may not needed?? */ 33478369b28dSHong Zhang nzi_bl = 0; j = 0; 33488369b28dSHong Zhang while (jtmp[j] < i) { /* Note: jtmp is sorted */ 33498aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 33508369b28dSHong Zhang nzi_bl++; j++; 33518369b28dSHong Zhang } 33528369b28dSHong Zhang nzi_bu = nzi - nzi_bl -1; 33538369b28dSHong Zhang while (j < nzi) { 33548aee2decSHong Zhang vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0; 33558369b28dSHong Zhang j++; 33568369b28dSHong Zhang } 33578369b28dSHong Zhang 3358599ef60dSHong Zhang bjtmp = bj + bi[i]; 3359599ef60dSHong Zhang batmp = ba + bi[i]; 33608369b28dSHong Zhang /* apply level dropping rule to L part */ 33618369b28dSHong Zhang ncut = nzi_al + dtcount; 33628369b28dSHong Zhang if (ncut < nzi_bl) { 33638369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr); 33648369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr); 3365599ef60dSHong Zhang } else { 33668369b28dSHong Zhang ncut = nzi_bl; 33678369b28dSHong Zhang } 33688369b28dSHong Zhang for (j=0; j<ncut; j++) { 33698369b28dSHong Zhang bjtmp[j] = jtmp[j]; 33708369b28dSHong Zhang batmp[j] = vtmp[j]; 33716da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */ 33728369b28dSHong Zhang } 33731d098868SHong Zhang bi[i+1] = bi[i] + ncut; 33748369b28dSHong Zhang nzi = ncut + 1; 33758369b28dSHong Zhang 33768369b28dSHong Zhang /* apply level dropping rule to U part */ 33778369b28dSHong Zhang ncut = nzi_au + dtcount; 33788369b28dSHong Zhang if (ncut < nzi_bu) { 33798369b28dSHong Zhang ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr); 33808369b28dSHong Zhang ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr); 33818369b28dSHong Zhang } else { 33828369b28dSHong Zhang ncut = nzi_bu; 33838369b28dSHong Zhang } 33848369b28dSHong Zhang nzi += ncut; 33851d098868SHong Zhang 33861d098868SHong Zhang /* mark bdiagonal */ 33871d098868SHong Zhang bdiag[i+1] = bdiag[i] - (ncut + 1); 3388dc3a2fd3SHong Zhang bdiag_rev[n-i-1] = bdiag[i+1]; 33898fc3a347SHong Zhang bi[2*n - i] = bi[2*n - i +1] - (ncut + 1); 33901d098868SHong Zhang bjtmp = bj + bdiag[i]; 33911d098868SHong Zhang batmp = ba + bdiag[i]; 33921d098868SHong Zhang *bjtmp = i; 33938aee2decSHong Zhang *batmp = diag_tmp; /* rtmp[i]; */ 3394c9c72f8fSHong Zhang if (*batmp == 0.0) { 3395c9c72f8fSHong Zhang *batmp = dt+shift; 3396bd1bc851SBarry Smith /* printf(" row %d add shift %g\n",i,shift); */ 3397c9c72f8fSHong Zhang } 3398b78a477dSHong Zhang *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */ 33996da515a1SHong Zhang /* printf(" (%d,%g),",*bjtmp,*batmp); */ 34001d098868SHong Zhang 34011d098868SHong Zhang bjtmp = bj + bdiag[i+1]+1; 34021d098868SHong Zhang batmp = ba + bdiag[i+1]+1; 34038369b28dSHong Zhang for (k=0; k<ncut; k++) { 34041d098868SHong Zhang bjtmp[k] = jtmp[nzi_bl+1+k]; 34051d098868SHong Zhang batmp[k] = vtmp[nzi_bl+1+k]; 34066da515a1SHong Zhang /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */ 3407599ef60dSHong Zhang } 34086da515a1SHong Zhang /* printf("\n"); */ 3409599ef60dSHong Zhang 3410599ef60dSHong Zhang im[i] = nzi; /* used by PetscLLAddSortedLU() */ 3411599ef60dSHong Zhang /* 34121d098868SHong Zhang printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]); 34138369b28dSHong Zhang printf(" ----------------------------\n"); 3414599ef60dSHong Zhang */ 34158369b28dSHong Zhang } /* for (i=0; i<n; i++) */ 34161d098868SHong Zhang /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */ 3417e32f2f54SBarry 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]); 3418599ef60dSHong Zhang 3419599ef60dSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3420599ef60dSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 3421599ef60dSHong Zhang 3422599ef60dSHong Zhang ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr); 34230e83c824SBarry Smith ierr = PetscFree2(im,jtmp);CHKERRQ(ierr); 34240e83c824SBarry Smith ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr); 34251bfa06eaSJed Brown ierr = PetscFree(bdiag_rev);CHKERRQ(ierr); 3426599ef60dSHong Zhang 3427599ef60dSHong Zhang ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr); 34281d098868SHong Zhang b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n]; 3429599ef60dSHong Zhang 3430599ef60dSHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 3431599ef60dSHong Zhang ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr); 3432599ef60dSHong Zhang if (row_identity && icol_identity) { 343335233d99SShri Abhyankar B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 3434599ef60dSHong Zhang } else { 343535233d99SShri Abhyankar B->ops->solve = MatSolve_SeqAIJ; 3436599ef60dSHong Zhang } 3437599ef60dSHong Zhang 34381d098868SHong Zhang B->ops->solveadd = 0; 34391d098868SHong Zhang B->ops->solvetranspose = 0; 34401d098868SHong Zhang B->ops->solvetransposeadd = 0; 34411d098868SHong Zhang B->ops->matsolve = 0; 3442599ef60dSHong Zhang B->assembled = PETSC_TRUE; 3443599ef60dSHong Zhang B->preallocated = PETSC_TRUE; 3444599ef60dSHong Zhang PetscFunctionReturn(0); 3445599ef60dSHong Zhang } 3446599ef60dSHong Zhang 34473c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */ 3448fe97e370SBarry Smith /* 3449bebe2cf6SSatish Balay This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer separate functions in the matrix function table for dt factors 3450fe97e370SBarry Smith */ 3451fe97e370SBarry Smith 34527087cfbeSBarry Smith PetscErrorCode MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info) 34533c2a7987SHong Zhang { 34543c2a7987SHong Zhang PetscErrorCode ierr; 34553c2a7987SHong Zhang 34563c2a7987SHong Zhang PetscFunctionBegin; 34573c2a7987SHong Zhang ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr); 34583c2a7987SHong Zhang PetscFunctionReturn(0); 34593c2a7987SHong Zhang } 34603c2a7987SHong Zhang 34613c2a7987SHong Zhang /* 34623c2a7987SHong Zhang same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors 34633c2a7987SHong Zhang - intend to replace existing MatLUFactorNumeric_SeqAIJ() 34643c2a7987SHong Zhang */ 3465fe97e370SBarry Smith /* 3466bebe2cf6SSatish Balay This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer separate functions in the matrix function table for dt factors 3467fe97e370SBarry Smith */ 3468fe97e370SBarry Smith 34697087cfbeSBarry Smith PetscErrorCode MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info) 34703c2a7987SHong Zhang { 34713c2a7987SHong Zhang Mat C =fact; 34723c2a7987SHong Zhang Mat_SeqAIJ *a =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data; 34733c2a7987SHong Zhang IS isrow = b->row,isicol = b->icol; 34743c2a7987SHong Zhang PetscErrorCode ierr; 34753c2a7987SHong Zhang const PetscInt *r,*ic,*ics; 3476b78a477dSHong Zhang PetscInt i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j; 3477b78a477dSHong Zhang PetscInt *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj; 34783c2a7987SHong Zhang MatScalar *rtmp,*pc,multiplier,*v,*pv,*aa=a->a; 3479182b8fbaSHong Zhang PetscReal dt=info->dt,shift=info->shiftamount; 3480ace3abfcSBarry Smith PetscBool row_identity, col_identity; 34813c2a7987SHong Zhang 34823c2a7987SHong Zhang PetscFunctionBegin; 34833c2a7987SHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 34843c2a7987SHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 3485854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&rtmp);CHKERRQ(ierr); 34863c2a7987SHong Zhang ics = ic; 34873c2a7987SHong Zhang 34883c2a7987SHong Zhang for (i=0; i<n; i++) { 3489b78a477dSHong Zhang /* initialize rtmp array */ 3490b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of nozeros in L(i,:) */ 34913c2a7987SHong Zhang bjtmp = bj + bi[i]; 3492b78a477dSHong Zhang for (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0; 3493b78a477dSHong Zhang rtmp[i] = 0.0; 3494b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */ 3495b78a477dSHong Zhang bjtmp = bj + bdiag[i+1] + 1; 3496b78a477dSHong Zhang for (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0; 34973c2a7987SHong Zhang 3498b78a477dSHong Zhang /* load in initial unfactored row of A */ 34996da515a1SHong Zhang /* printf("row %d\n",i); */ 35003c2a7987SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 35013c2a7987SHong Zhang ajtmp = aj + ai[r[i]]; 35023c2a7987SHong Zhang v = aa + ai[r[i]]; 35033c2a7987SHong Zhang for (j=0; j<nz; j++) { 3504b78a477dSHong Zhang rtmp[ics[*ajtmp++]] = v[j]; 35056da515a1SHong Zhang /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */ 35063c2a7987SHong Zhang } 35076da515a1SHong Zhang /* printf("\n"); */ 35083c2a7987SHong Zhang 3509b78a477dSHong Zhang /* numerical factorization */ 3510b78a477dSHong Zhang bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */ 3511b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; /* num of entries in L(i,:) */ 3512b78a477dSHong Zhang k = 0; 3513b78a477dSHong Zhang while (k < nzl) { 35143c2a7987SHong Zhang row = *bjtmp++; 35156da515a1SHong Zhang /* printf(" prow %d\n",row); */ 35163c2a7987SHong Zhang pc = rtmp + row; 3517b78a477dSHong Zhang pv = b->a + bdiag[row]; /* 1./(diag of the pivot row) */ 3518b78a477dSHong Zhang multiplier = (*pc) * (*pv); 35193c2a7987SHong Zhang *pc = multiplier; 3520b78a477dSHong Zhang if (PetscAbsScalar(multiplier) > dt) { 3521b78a477dSHong Zhang pj = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */ 3522b78a477dSHong Zhang pv = b->a + bdiag[row+1] + 1; 3523b78a477dSHong Zhang nz = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */ 3524b78a477dSHong Zhang for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++); 352528b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 35263c2a7987SHong Zhang } 3527b78a477dSHong Zhang k++; 35283c2a7987SHong Zhang } 35293c2a7987SHong Zhang 3530b78a477dSHong Zhang /* finished row so stick it into b->a */ 3531b78a477dSHong Zhang /* L-part */ 3532b78a477dSHong Zhang pv = b->a + bi[i]; 3533b78a477dSHong Zhang pj = bj + bi[i]; 3534b78a477dSHong Zhang nzl = bi[i+1] - bi[i]; 3535b78a477dSHong Zhang for (j=0; j<nzl; j++) { 3536b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 35376da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 35383c2a7987SHong Zhang } 3539b78a477dSHong Zhang 3540b78a477dSHong Zhang /* diagonal: invert diagonal entries for simplier triangular solves */ 3541b78a477dSHong Zhang if (rtmp[i] == 0.0) rtmp[i] = dt+shift; 3542b78a477dSHong Zhang b->a[bdiag[i]] = 1.0/rtmp[i]; 35436da515a1SHong Zhang /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */ 3544b78a477dSHong Zhang 3545b78a477dSHong Zhang /* U-part */ 3546b78a477dSHong Zhang pv = b->a + bdiag[i+1] + 1; 3547b78a477dSHong Zhang pj = bj + bdiag[i+1] + 1; 3548b78a477dSHong Zhang nzu = bdiag[i] - bdiag[i+1] - 1; 3549b78a477dSHong Zhang for (j=0; j<nzu; j++) { 3550b78a477dSHong Zhang pv[j] = rtmp[pj[j]]; 35516da515a1SHong Zhang /* printf(" (%d,%g),",pj[j],pv[j]); */ 3552b78a477dSHong Zhang } 35536da515a1SHong Zhang /* printf("\n"); */ 3554b78a477dSHong Zhang } 3555b78a477dSHong Zhang 35563c2a7987SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 35573c2a7987SHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 35583c2a7987SHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 3559b78a477dSHong Zhang 35603c2a7987SHong Zhang ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr); 35613c2a7987SHong Zhang ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr); 35623c2a7987SHong Zhang if (row_identity && col_identity) { 356335233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering; 35643c2a7987SHong Zhang } else { 356535233d99SShri Abhyankar C->ops->solve = MatSolve_SeqAIJ; 35663c2a7987SHong Zhang } 3567b78a477dSHong Zhang C->ops->solveadd = 0; 3568b78a477dSHong Zhang C->ops->solvetranspose = 0; 3569b78a477dSHong Zhang C->ops->solvetransposeadd = 0; 3570b78a477dSHong Zhang C->ops->matsolve = 0; 35713c2a7987SHong Zhang C->assembled = PETSC_TRUE; 35723c2a7987SHong Zhang C->preallocated = PETSC_TRUE; 35732205254eSKarl Rupp 35743c2a7987SHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 35753c2a7987SHong Zhang PetscFunctionReturn(0); 35763c2a7987SHong Zhang } 3577