173f4d377SMatthew Knepley /*$Id: sbaijfact.c,v 1.61 2001/08/06 21:15:47 bsmith Exp $*/ 249b5e25fSSatish Balay /* Using Modified Sparse Row (MSR) storage. 349b5e25fSSatish Balay See page 85, "Iterative Methods ..." by Saad. */ 449b5e25fSSatish Balay 549b5e25fSSatish Balay /* 6effa298cSHong Zhang Symbolic U^T*D*U factorization for SBAIJ format. Modified from SSF of YSMP. 749b5e25fSSatish Balay */ 849b5e25fSSatish Balay #include "src/mat/impls/baij/seq/baij.h" 93a7fca6bSBarry Smith #include "src/mat/impls/sbaij/seq/sbaij.h" 1049b5e25fSSatish Balay #include "src/vec/vecimpl.h" 1149b5e25fSSatish Balay #include "src/inline/ilu.h" 1249a6740bSHong Zhang #include "include/petscis.h" 1349b5e25fSSatish Balay 1487fe1012SHong Zhang /* Use Modified Sparse Row storage for u and ju, see Sasd pp.85 */ 154a2ae208SSatish Balay #undef __FUNCT__ 164a2ae208SSatish Balay #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqSBAIJ" 1777f73120SHong Zhang int MatCholeskyFactorSymbolic_SeqSBAIJ(Mat A,IS perm,PetscReal f,Mat *B) 1849b5e25fSSatish Balay { 1949b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data,*b; 20cb718733SHong Zhang int *rip,ierr,i,mbs = a->mbs,*ai,*aj; 21066653e3SSatish Balay int *jutmp,bs = a->bs,bs2=a->bs2; 2287fe1012SHong Zhang int m,realloc = 0,prow; 23d60e5362SHong Zhang int *jl,*q,jmin,jmax,juidx,nzk,qm,*iu,*ju,k,j,vj,umax,maxadd; 24671cb588SHong Zhang PetscTruth perm_identity; 2549b5e25fSSatish Balay 2649b5e25fSSatish Balay PetscFunctionBegin; 272d007305SHong Zhang 28cb718733SHong Zhang /* check whether perm is the identity mapping */ 29671cb588SHong Zhang ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr); 30671cb588SHong Zhang if (!perm_identity) a->permute = PETSC_TRUE; 312d007305SHong Zhang 32671cb588SHong Zhang ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr); 33671cb588SHong Zhang 34671cb588SHong Zhang if (perm_identity){ /* without permutation */ 352d007305SHong Zhang ai = a->i; aj = a->j; 362d007305SHong Zhang } else { /* non-trivial permutation */ 37323b833fSBarry Smith ierr = MatReorderingSeqSBAIJ(A,perm);CHKERRQ(ierr); 382d007305SHong Zhang ai = a->inew; aj = a->jnew; 392d007305SHong Zhang } 402d007305SHong Zhang 4149b5e25fSSatish Balay /* initialization */ 42b0a32e0cSBarry Smith ierr = PetscMalloc((mbs+1)*sizeof(int),&iu);CHKERRQ(ierr); 4349b5e25fSSatish Balay umax = (int)(f*ai[mbs] + 1); umax += mbs + 1; 4482502324SSatish Balay ierr = PetscMalloc(umax*sizeof(int),&ju);CHKERRQ(ierr); 4549b5e25fSSatish Balay iu[0] = mbs+1; 4687fe1012SHong Zhang juidx = mbs + 1; /* index for ju */ 4787fe1012SHong Zhang ierr = PetscMalloc(2*mbs*sizeof(int),&jl);CHKERRQ(ierr); /* linked list for pivot row */ 4887fe1012SHong Zhang q = jl + mbs; /* linked list for col index */ 4949b5e25fSSatish Balay for (i=0; i<mbs; i++){ 5087fe1012SHong Zhang jl[i] = mbs; 5187fe1012SHong Zhang q[i] = 0; 5249b5e25fSSatish Balay } 5349b5e25fSSatish Balay 5449b5e25fSSatish Balay /* for each row k */ 5549b5e25fSSatish Balay for (k=0; k<mbs; k++){ 5649b5e25fSSatish Balay nzk = 0; /* num. of nz blocks in k-th block row with diagonal block excluded */ 5749b5e25fSSatish Balay q[k] = mbs; 5849b5e25fSSatish Balay /* initialize nonzero structure of k-th row to row rip[k] of A */ 5949b5e25fSSatish Balay jmin = ai[rip[k]]; 6049b5e25fSSatish Balay jmax = ai[rip[k]+1]; 6149b5e25fSSatish Balay for (j=jmin; j<jmax; j++){ 62cb718733SHong Zhang vj = rip[aj[j]]; /* col. value */ 6349b5e25fSSatish Balay if(vj > k){ 6449b5e25fSSatish Balay qm = k; 6549b5e25fSSatish Balay do { 6649b5e25fSSatish Balay m = qm; qm = q[m]; 6749b5e25fSSatish Balay } while(qm < vj); 6849b5e25fSSatish Balay if (qm == vj) { 693078e140SBarry Smith SETERRQ(1," error: duplicate entry in A\n"); 7049b5e25fSSatish Balay } 7149b5e25fSSatish Balay nzk++; 7249b5e25fSSatish Balay q[m] = vj; 7349b5e25fSSatish Balay q[vj] = qm; 7449b5e25fSSatish Balay } /* if(vj > k) */ 7549b5e25fSSatish Balay } /* for (j=jmin; j<jmax; j++) */ 7649b5e25fSSatish Balay 7749b5e25fSSatish Balay /* modify nonzero structure of k-th row by computing fill-in 7849b5e25fSSatish Balay for each row i to be merged in */ 7987fe1012SHong Zhang prow = k; 8087fe1012SHong Zhang prow = jl[prow]; /* next pivot row (== mbs for symbolic factorization) */ 8187fe1012SHong Zhang 8287fe1012SHong Zhang while (prow < k){ 8387fe1012SHong Zhang /* merge row prow into k-th row */ 8487fe1012SHong Zhang jmin = iu[prow] + 1; jmax = iu[prow+1]; 8549b5e25fSSatish Balay qm = k; 8687fe1012SHong Zhang for (j=jmin; j<jmax; j++){ 8749b5e25fSSatish Balay vj = ju[j]; 8849b5e25fSSatish Balay do { 8949b5e25fSSatish Balay m = qm; qm = q[m]; 9049b5e25fSSatish Balay } while (qm < vj); 9149b5e25fSSatish Balay if (qm != vj){ 9249b5e25fSSatish Balay nzk++; q[m] = vj; q[vj] = qm; qm = vj; 9349b5e25fSSatish Balay } 9449b5e25fSSatish Balay } 9587fe1012SHong Zhang prow = jl[prow]; /* next pivot row */ 9649b5e25fSSatish Balay } 9749b5e25fSSatish Balay 9849b5e25fSSatish Balay /* add k to row list for first nonzero element in k-th row */ 9949b5e25fSSatish Balay if (nzk > 0){ 10049b5e25fSSatish Balay i = q[k]; /* col value of first nonzero element in U(k, k+1:mbs-1) */ 10149b5e25fSSatish Balay jl[k] = jl[i]; jl[i] = k; 10249b5e25fSSatish Balay } 10387fe1012SHong Zhang iu[k+1] = iu[k] + nzk; 10449b5e25fSSatish Balay 10549b5e25fSSatish Balay /* allocate more space to ju if needed */ 1063078e140SBarry Smith if (iu[k+1] > umax) { 10749b5e25fSSatish Balay /* estimate how much additional space we will need */ 10849b5e25fSSatish Balay /* use the strategy suggested by David Hysom <hysom@perch-t.icase.edu> */ 10949b5e25fSSatish Balay /* just double the memory each time */ 11049b5e25fSSatish Balay maxadd = umax; 11149b5e25fSSatish Balay if (maxadd < nzk) maxadd = (mbs-k)*(nzk+1)/2; 11249b5e25fSSatish Balay umax += maxadd; 11349b5e25fSSatish Balay 1149f9cb213SHong Zhang /* allocate a longer ju */ 11582502324SSatish Balay ierr = PetscMalloc(umax*sizeof(int),&jutmp);CHKERRQ(ierr); 11649b5e25fSSatish Balay ierr = PetscMemcpy(jutmp,ju,iu[k]*sizeof(int));CHKERRQ(ierr); 11749b5e25fSSatish Balay ierr = PetscFree(ju);CHKERRQ(ierr); 1189f9cb213SHong Zhang ju = jutmp; 11949b5e25fSSatish Balay realloc++; /* count how many times we realloc */ 12049b5e25fSSatish Balay } 12149b5e25fSSatish Balay 12249b5e25fSSatish Balay /* save nonzero structure of k-th row in ju */ 12349b5e25fSSatish Balay i=k; 12487fe1012SHong Zhang while (nzk --) { 12549b5e25fSSatish Balay i = q[i]; 12687fe1012SHong Zhang ju[juidx++] = i; 12749b5e25fSSatish Balay } 12877f73120SHong Zhang } 12949b5e25fSSatish Balay 13049b5e25fSSatish Balay if (ai[mbs] != 0) { 13149b5e25fSSatish Balay PetscReal af = ((PetscReal)iu[mbs])/((PetscReal)ai[mbs]); 132b0a32e0cSBarry Smith PetscLogInfo(A,"MatCholeskyFactorSymbolic_SeqSBAIJ:Reallocs %d Fill ratio:given %g needed %g\n",realloc,f,af); 1333078e140SBarry Smith PetscLogInfo(A,"MatCholeskyFactorSymbolic_SeqSBAIJ:Run with -pc_cholesky_fill %g or use \n",af); 1343078e140SBarry Smith PetscLogInfo(A,"MatCholeskyFactorSymbolic_SeqSBAIJ:PCCholeskySetFill(pc,%g);\n",af); 135b0a32e0cSBarry Smith PetscLogInfo(A,"MatCholeskyFactorSymbolic_SeqSBAIJ:for best performance.\n"); 13649b5e25fSSatish Balay } else { 137b0a32e0cSBarry Smith PetscLogInfo(A,"MatCholeskyFactorSymbolic_SeqSBAIJ:Empty matrix.\n"); 13849b5e25fSSatish Balay } 13949b5e25fSSatish Balay 14077f73120SHong Zhang ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr); 14187fe1012SHong Zhang /* ierr = PetscFree(q);CHKERRQ(ierr); */ 14249b5e25fSSatish Balay ierr = PetscFree(jl);CHKERRQ(ierr); 14349b5e25fSSatish Balay 14449b5e25fSSatish Balay /* put together the new matrix */ 14549b5e25fSSatish Balay ierr = MatCreateSeqSBAIJ(A->comm,bs,bs*mbs,bs*mbs,0,PETSC_NULL,B);CHKERRQ(ierr); 146b0a32e0cSBarry Smith /* PetscLogObjectParent(*B,iperm); */ 14749b5e25fSSatish Balay b = (Mat_SeqSBAIJ*)(*B)->data; 14849b5e25fSSatish Balay ierr = PetscFree(b->imax);CHKERRQ(ierr); 14949b5e25fSSatish Balay b->singlemalloc = PETSC_FALSE; 15049b5e25fSSatish Balay /* the next line frees the default space generated by the Create() */ 15149b5e25fSSatish Balay ierr = PetscFree(b->a);CHKERRQ(ierr); 15249b5e25fSSatish Balay ierr = PetscFree(b->ilen);CHKERRQ(ierr); 15382502324SSatish Balay ierr = PetscMalloc((iu[mbs]+1)*sizeof(MatScalar)*bs2,&b->a);CHKERRQ(ierr); 15449b5e25fSSatish Balay b->j = ju; 15549b5e25fSSatish Balay b->i = iu; 15649b5e25fSSatish Balay b->diag = 0; 15749b5e25fSSatish Balay b->ilen = 0; 15849b5e25fSSatish Balay b->imax = 0; 15977f73120SHong Zhang b->row = perm; 160bcd9e38bSBarry Smith b->pivotinblocks = PETSC_FALSE; /* need to get from MatCholeskyInfo */ 16177f73120SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 162cb718733SHong Zhang b->icol = perm; 163cb718733SHong Zhang ierr = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr); 16487828ca2SBarry Smith ierr = PetscMalloc((bs*mbs+bs)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr); 16549b5e25fSSatish Balay /* In b structure: Free imax, ilen, old a, old j. 16649b5e25fSSatish Balay Allocate idnew, solve_work, new a, new j */ 167b0a32e0cSBarry Smith PetscLogObjectMemory(*B,(iu[mbs]-mbs)*(sizeof(int)+sizeof(MatScalar))); 16849b5e25fSSatish Balay b->s_maxnz = b->s_nz = iu[mbs]; 16949b5e25fSSatish Balay 1705c0bcdfcSHong Zhang (*B)->factor = FACTOR_CHOLESKY; 17149b5e25fSSatish Balay (*B)->info.factor_mallocs = realloc; 17249b5e25fSSatish Balay (*B)->info.fill_ratio_given = f; 17349b5e25fSSatish Balay if (ai[mbs] != 0) { 17449b5e25fSSatish Balay (*B)->info.fill_ratio_needed = ((PetscReal)iu[mbs])/((PetscReal)ai[mbs]); 17549b5e25fSSatish Balay } else { 17649b5e25fSSatish Balay (*B)->info.fill_ratio_needed = 0.0; 17749b5e25fSSatish Balay } 1780aa0ce06SHong Zhang 179671cb588SHong Zhang if (perm_identity){ 180671cb588SHong Zhang switch (bs) { 181671cb588SHong Zhang case 1: 182671cb588SHong Zhang (*B)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering; 183671cb588SHong Zhang (*B)->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 1844d101231SSatish Balay PetscLogInfo(A,"MatICCFactorSymbolic_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=1\n"); 185671cb588SHong Zhang break; 186671cb588SHong Zhang case 2: 187671cb588SHong Zhang (*B)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering; 188671cb588SHong Zhang (*B)->ops->solve = MatSolve_SeqSBAIJ_2_NaturalOrdering; 1894d101231SSatish Balay PetscLogInfo(A,"MatICCFactorSymbolic_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=2\n"); 190671cb588SHong Zhang break; 191671cb588SHong Zhang case 3: 192671cb588SHong Zhang (*B)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering; 193671cb588SHong Zhang (*B)->ops->solve = MatSolve_SeqSBAIJ_3_NaturalOrdering; 1944d101231SSatish Balay PetscLogInfo(A,"MatICCFactorSymbolic_SeqSBAIJ:sing special in-place natural ordering factor and solve BS=3\n"); 195671cb588SHong Zhang break; 196671cb588SHong Zhang case 4: 197671cb588SHong Zhang (*B)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering; 198671cb588SHong Zhang (*B)->ops->solve = MatSolve_SeqSBAIJ_4_NaturalOrdering; 1994d101231SSatish Balay PetscLogInfo(A,"MatICCFactorSymbolic_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=4\n"); 200671cb588SHong Zhang break; 201671cb588SHong Zhang case 5: 202671cb588SHong Zhang (*B)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering; 203671cb588SHong Zhang (*B)->ops->solve = MatSolve_SeqSBAIJ_5_NaturalOrdering; 2044d101231SSatish Balay PetscLogInfo(A,"MatICCFactorSymbolic_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=5\n"); 205671cb588SHong Zhang break; 206671cb588SHong Zhang case 6: 207671cb588SHong Zhang (*B)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering; 208671cb588SHong Zhang (*B)->ops->solve = MatSolve_SeqSBAIJ_6_NaturalOrdering; 2094d101231SSatish Balay PetscLogInfo(A,"MatICCFactorSymbolic_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=6\n"); 210671cb588SHong Zhang break; 211671cb588SHong Zhang case 7: 212671cb588SHong Zhang (*B)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering; 213671cb588SHong Zhang (*B)->ops->solve = MatSolve_SeqSBAIJ_7_NaturalOrdering; 2144d101231SSatish Balay PetscLogInfo(A,"MatICCFactorSymbolic_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=7\n"); 215671cb588SHong Zhang break; 216671cb588SHong Zhang default: 217671cb588SHong Zhang (*B)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering; 218671cb588SHong Zhang (*B)->ops->solve = MatSolve_SeqSBAIJ_N_NaturalOrdering; 2194d101231SSatish Balay PetscLogInfo(A,"MatICCFactorSymbolic_SeqSBAIJ:Using special in-place natural ordering factor and solve BS>7\n"); 220671cb588SHong Zhang break; 221671cb588SHong Zhang } 222671cb588SHong Zhang } 223671cb588SHong Zhang 22449b5e25fSSatish Balay PetscFunctionReturn(0); 22549b5e25fSSatish Balay } 22649b5e25fSSatish Balay 22735d8aa7fSBarry Smith 2284a2ae208SSatish Balay #undef __FUNCT__ 2294a2ae208SSatish Balay #define __FUNCT__ "MatCholeskyFactorNumeric_SeqSBAIJ_N" 2306f9a564bSHong Zhang int MatCholeskyFactorNumeric_SeqSBAIJ_N(Mat A,Mat *B) 23149b5e25fSSatish Balay { 23249b5e25fSSatish Balay Mat C = *B; 2334c16a6a6SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data,*b = (Mat_SeqSBAIJ *)C->data; 2344c16a6a6SHong Zhang IS perm = b->row; 2354c16a6a6SHong Zhang int *perm_ptr,ierr,i,j,mbs=a->mbs,*bi=b->i,*bj=b->j; 2364c16a6a6SHong Zhang int *ai,*aj,*a2anew,k,k1,jmin,jmax,*jl,*il,vj,nexti,ili; 2374c16a6a6SHong Zhang int bs=a->bs,bs2 = a->bs2; 2384c16a6a6SHong Zhang MatScalar *ba = b->a,*aa,*ap,*dk,*uik; 2394c16a6a6SHong Zhang MatScalar *u,*diag,*rtmp,*rtmp_ptr; 24028de702eSHong Zhang MatScalar *work; 2414c16a6a6SHong Zhang int *pivots; 2424c16a6a6SHong Zhang 2434c16a6a6SHong Zhang PetscFunctionBegin; 2449706f043SHong Zhang 2454c16a6a6SHong Zhang /* initialization */ 24682502324SSatish Balay ierr = PetscMalloc(bs2*mbs*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 2474c16a6a6SHong Zhang ierr = PetscMemzero(rtmp,bs2*mbs*sizeof(MatScalar));CHKERRQ(ierr); 24882502324SSatish Balay ierr = PetscMalloc(2*mbs*sizeof(int),&il);CHKERRQ(ierr); 24928de702eSHong Zhang jl = il + mbs; 2504c16a6a6SHong Zhang for (i=0; i<mbs; i++) { 2514c16a6a6SHong Zhang jl[i] = mbs; il[0] = 0; 2524c16a6a6SHong Zhang } 253b0a32e0cSBarry Smith ierr = PetscMalloc((2*bs2+bs)*sizeof(MatScalar),&dk);CHKERRQ(ierr); 25428de702eSHong Zhang uik = dk + bs2; 25528de702eSHong Zhang work = uik + bs2; 25682502324SSatish Balay ierr = PetscMalloc(bs*sizeof(int),&pivots);CHKERRQ(ierr); 2574c16a6a6SHong Zhang 2584c16a6a6SHong Zhang ierr = ISGetIndices(perm,&perm_ptr);CHKERRQ(ierr); 2594c16a6a6SHong Zhang 2604c16a6a6SHong Zhang /* check permutation */ 2614c16a6a6SHong Zhang if (!a->permute){ 2624c16a6a6SHong Zhang ai = a->i; aj = a->j; aa = a->a; 2634c16a6a6SHong Zhang } else { 2644c16a6a6SHong Zhang ai = a->inew; aj = a->jnew; 26582502324SSatish Balay ierr = PetscMalloc(bs2*ai[mbs]*sizeof(MatScalar),&aa);CHKERRQ(ierr); 2664c16a6a6SHong Zhang ierr = PetscMemcpy(aa,a->a,bs2*ai[mbs]*sizeof(MatScalar));CHKERRQ(ierr); 26782502324SSatish Balay ierr = PetscMalloc(ai[mbs]*sizeof(int),&a2anew);CHKERRQ(ierr); 2684c16a6a6SHong Zhang ierr = PetscMemcpy(a2anew,a->a2anew,(ai[mbs])*sizeof(int));CHKERRQ(ierr); 2694c16a6a6SHong Zhang 2704c16a6a6SHong Zhang for (i=0; i<mbs; i++){ 2714c16a6a6SHong Zhang jmin = ai[i]; jmax = ai[i+1]; 2724c16a6a6SHong Zhang for (j=jmin; j<jmax; j++){ 2734c16a6a6SHong Zhang while (a2anew[j] != j){ 2744c16a6a6SHong Zhang k = a2anew[j]; a2anew[j] = a2anew[k]; a2anew[k] = k; 2754c16a6a6SHong Zhang for (k1=0; k1<bs2; k1++){ 2764c16a6a6SHong Zhang dk[k1] = aa[k*bs2+k1]; 2774c16a6a6SHong Zhang aa[k*bs2+k1] = aa[j*bs2+k1]; 2784c16a6a6SHong Zhang aa[j*bs2+k1] = dk[k1]; 2794c16a6a6SHong Zhang } 2804c16a6a6SHong Zhang } 2814c16a6a6SHong Zhang /* transform columnoriented blocks that lie in the lower triangle to roworiented blocks */ 2824c16a6a6SHong Zhang if (i > aj[j]){ 2834c16a6a6SHong Zhang /* printf("change orientation, row: %d, col: %d\n",i,aj[j]); */ 2844c16a6a6SHong Zhang ap = aa + j*bs2; /* ptr to the beginning of j-th block of aa */ 2854c16a6a6SHong Zhang for (k=0; k<bs2; k++) dk[k] = ap[k]; /* dk <- j-th block of aa */ 2864c16a6a6SHong Zhang for (k=0; k<bs; k++){ /* j-th block of aa <- dk^T */ 2874c16a6a6SHong Zhang for (k1=0; k1<bs; k1++) *ap++ = dk[k + bs*k1]; 2884c16a6a6SHong Zhang } 2894c16a6a6SHong Zhang } 2904c16a6a6SHong Zhang } 2914c16a6a6SHong Zhang } 292323b833fSBarry Smith ierr = PetscFree(a2anew);CHKERRQ(ierr); 2934c16a6a6SHong Zhang } 2944c16a6a6SHong Zhang 2954c16a6a6SHong Zhang /* for each row k */ 2964c16a6a6SHong Zhang for (k = 0; k<mbs; k++){ 2974c16a6a6SHong Zhang 2984c16a6a6SHong Zhang /*initialize k-th row with elements nonzero in row perm(k) of A */ 2994c16a6a6SHong Zhang jmin = ai[perm_ptr[k]]; jmax = ai[perm_ptr[k]+1]; 300*057f5ba7SHong Zhang 3014c16a6a6SHong Zhang ap = aa + jmin*bs2; 3024c16a6a6SHong Zhang for (j = jmin; j < jmax; j++){ 3034c16a6a6SHong Zhang vj = perm_ptr[aj[j]]; /* block col. index */ 3044c16a6a6SHong Zhang rtmp_ptr = rtmp + vj*bs2; 3054c16a6a6SHong Zhang for (i=0; i<bs2; i++) *rtmp_ptr++ = *ap++; 3064c16a6a6SHong Zhang } 3074c16a6a6SHong Zhang 3084c16a6a6SHong Zhang /* modify k-th row by adding in those rows i with U(i,k) != 0 */ 3094c16a6a6SHong Zhang ierr = PetscMemcpy(dk,rtmp+k*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr); 3104c16a6a6SHong Zhang i = jl[k]; /* first row to be added to k_th row */ 3114c16a6a6SHong Zhang 312*057f5ba7SHong Zhang while (i < k){ 3134c16a6a6SHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 3144c16a6a6SHong Zhang 3154c16a6a6SHong Zhang /* compute multiplier */ 3164c16a6a6SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 3174c16a6a6SHong Zhang 3184c16a6a6SHong Zhang /* uik = -inv(Di)*U_bar(i,k) */ 3194c16a6a6SHong Zhang diag = ba + i*bs2; 3204c16a6a6SHong Zhang u = ba + ili*bs2; 3214c16a6a6SHong Zhang ierr = PetscMemzero(uik,bs2*sizeof(MatScalar));CHKERRQ(ierr); 3224c16a6a6SHong Zhang Kernel_A_gets_A_minus_B_times_C(bs,uik,diag,u); 3234c16a6a6SHong Zhang 3244c16a6a6SHong Zhang /* update D(k) += -U(i,k)^T * U_bar(i,k) */ 3254c16a6a6SHong Zhang Kernel_A_gets_A_plus_Btranspose_times_C(bs,dk,uik,u); 3264c16a6a6SHong Zhang 3274c16a6a6SHong Zhang /* update -U(i,k) */ 3284c16a6a6SHong Zhang ierr = PetscMemcpy(ba+ili*bs2,uik,bs2*sizeof(MatScalar));CHKERRQ(ierr); 3294c16a6a6SHong Zhang 3304c16a6a6SHong Zhang /* add multiple of row i to k-th row ... */ 3314c16a6a6SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 3324c16a6a6SHong Zhang if (jmin < jmax){ 3334c16a6a6SHong Zhang for (j=jmin; j<jmax; j++) { 3344c16a6a6SHong Zhang /* rtmp += -U(i,k)^T * U_bar(i,j) */ 3354c16a6a6SHong Zhang rtmp_ptr = rtmp + bj[j]*bs2; 3364c16a6a6SHong Zhang u = ba + j*bs2; 3374c16a6a6SHong Zhang Kernel_A_gets_A_plus_Btranspose_times_C(bs,rtmp_ptr,uik,u); 3384c16a6a6SHong Zhang } 3394c16a6a6SHong Zhang 3404c16a6a6SHong Zhang /* ... add i to row list for next nonzero entry */ 3414c16a6a6SHong Zhang il[i] = jmin; /* update il(i) in column k+1, ... mbs-1 */ 3424c16a6a6SHong Zhang j = bj[jmin]; 3434c16a6a6SHong Zhang jl[i] = jl[j]; jl[j] = i; /* update jl */ 3444c16a6a6SHong Zhang } 3454c16a6a6SHong Zhang i = nexti; 3464c16a6a6SHong Zhang } 3474c16a6a6SHong Zhang 3484c16a6a6SHong Zhang /* save nonzero entries in k-th row of U ... */ 3494c16a6a6SHong Zhang 3504c16a6a6SHong Zhang /* invert diagonal block */ 3514c16a6a6SHong Zhang diag = ba+k*bs2; 3524c16a6a6SHong Zhang ierr = PetscMemcpy(diag,dk,bs2*sizeof(MatScalar));CHKERRQ(ierr); 3534c16a6a6SHong Zhang Kernel_A_gets_inverse_A(bs,diag,pivots,work); 3544c16a6a6SHong Zhang 3554c16a6a6SHong Zhang jmin = bi[k]; jmax = bi[k+1]; 3564c16a6a6SHong Zhang if (jmin < jmax) { 3574c16a6a6SHong Zhang for (j=jmin; j<jmax; j++){ 3584c16a6a6SHong Zhang vj = bj[j]; /* block col. index of U */ 3594c16a6a6SHong Zhang u = ba + j*bs2; 3604c16a6a6SHong Zhang rtmp_ptr = rtmp + vj*bs2; 3614c16a6a6SHong Zhang for (k1=0; k1<bs2; k1++){ 3624c16a6a6SHong Zhang *u++ = *rtmp_ptr; 3634c16a6a6SHong Zhang *rtmp_ptr++ = 0.0; 3644c16a6a6SHong Zhang } 3654c16a6a6SHong Zhang } 3664c16a6a6SHong Zhang 3674c16a6a6SHong Zhang /* ... add k to row list for first nonzero entry in k-th row */ 3684c16a6a6SHong Zhang il[k] = jmin; 3694c16a6a6SHong Zhang i = bj[jmin]; 3704c16a6a6SHong Zhang jl[k] = jl[i]; jl[i] = k; 3714c16a6a6SHong Zhang } 3724c16a6a6SHong Zhang } 3734c16a6a6SHong Zhang 3744c16a6a6SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 3754c16a6a6SHong Zhang ierr = PetscFree(il);CHKERRQ(ierr); 3764c16a6a6SHong Zhang ierr = PetscFree(dk);CHKERRQ(ierr); 3774c16a6a6SHong Zhang ierr = PetscFree(pivots);CHKERRQ(ierr); 3784c16a6a6SHong Zhang if (a->permute){ 3794c16a6a6SHong Zhang ierr = PetscFree(aa);CHKERRQ(ierr); 3804c16a6a6SHong Zhang } 3814c16a6a6SHong Zhang 3824c16a6a6SHong Zhang ierr = ISRestoreIndices(perm,&perm_ptr);CHKERRQ(ierr); 3834c16a6a6SHong Zhang C->factor = FACTOR_CHOLESKY; 3844c16a6a6SHong Zhang C->assembled = PETSC_TRUE; 3854c16a6a6SHong Zhang C->preallocated = PETSC_TRUE; 386b0a32e0cSBarry Smith PetscLogFlops(1.3333*bs*bs2*b->mbs); /* from inverting diagonal blocks */ 3874c16a6a6SHong Zhang PetscFunctionReturn(0); 3884c16a6a6SHong Zhang } 389d4edadd4SHong Zhang 3904a2ae208SSatish Balay #undef __FUNCT__ 3914a2ae208SSatish Balay #define __FUNCT__ "MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering" 392671cb588SHong Zhang int MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering(Mat A,Mat *B) 393671cb588SHong Zhang { 394671cb588SHong Zhang Mat C = *B; 395671cb588SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data,*b = (Mat_SeqSBAIJ *)C->data; 396671cb588SHong Zhang int ierr,i,j,mbs=a->mbs,*bi=b->i,*bj=b->j; 3971ad1de41SHong Zhang int *ai,*aj,k,k1,jmin,jmax,*jl,*il,vj,nexti,ili; 398671cb588SHong Zhang int bs=a->bs,bs2 = a->bs2; 399671cb588SHong Zhang MatScalar *ba = b->a,*aa,*ap,*dk,*uik; 400671cb588SHong Zhang MatScalar *u,*diag,*rtmp,*rtmp_ptr; 40128de702eSHong Zhang MatScalar *work; 402671cb588SHong Zhang int *pivots; 403671cb588SHong Zhang 404671cb588SHong Zhang PetscFunctionBegin; 405671cb588SHong Zhang 406671cb588SHong Zhang /* initialization */ 407671cb588SHong Zhang 40882502324SSatish Balay ierr = PetscMalloc(bs2*mbs*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 409671cb588SHong Zhang ierr = PetscMemzero(rtmp,bs2*mbs*sizeof(MatScalar));CHKERRQ(ierr); 41082502324SSatish Balay ierr = PetscMalloc(2*mbs*sizeof(int),&il);CHKERRQ(ierr); 41128de702eSHong Zhang jl = il + mbs; 412671cb588SHong Zhang for (i=0; i<mbs; i++) { 413671cb588SHong Zhang jl[i] = mbs; il[0] = 0; 414671cb588SHong Zhang } 415b0a32e0cSBarry Smith ierr = PetscMalloc((2*bs2+bs)*sizeof(MatScalar),&dk);CHKERRQ(ierr); 41628de702eSHong Zhang uik = dk + bs2; 41728de702eSHong Zhang work = uik + bs2; 41882502324SSatish Balay ierr = PetscMalloc(bs*sizeof(int),&pivots);CHKERRQ(ierr); 419671cb588SHong Zhang 420671cb588SHong Zhang ai = a->i; aj = a->j; aa = a->a; 421671cb588SHong Zhang 422671cb588SHong Zhang /* for each row k */ 423671cb588SHong Zhang for (k = 0; k<mbs; k++){ 424671cb588SHong Zhang 425671cb588SHong Zhang /*initialize k-th row with elements nonzero in row k of A */ 426671cb588SHong Zhang jmin = ai[k]; jmax = ai[k+1]; 427671cb588SHong Zhang ap = aa + jmin*bs2; 428671cb588SHong Zhang for (j = jmin; j < jmax; j++){ 429671cb588SHong Zhang vj = aj[j]; /* block col. index */ 430671cb588SHong Zhang rtmp_ptr = rtmp + vj*bs2; 431671cb588SHong Zhang for (i=0; i<bs2; i++) *rtmp_ptr++ = *ap++; 432671cb588SHong Zhang } 433671cb588SHong Zhang 434671cb588SHong Zhang /* modify k-th row by adding in those rows i with U(i,k) != 0 */ 435671cb588SHong Zhang ierr = PetscMemcpy(dk,rtmp+k*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr); 436671cb588SHong Zhang i = jl[k]; /* first row to be added to k_th row */ 437671cb588SHong Zhang 438*057f5ba7SHong Zhang while (i < k){ 439671cb588SHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 440671cb588SHong Zhang 441671cb588SHong Zhang /* compute multiplier */ 442671cb588SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 443671cb588SHong Zhang 444671cb588SHong Zhang /* uik = -inv(Di)*U_bar(i,k) */ 445671cb588SHong Zhang diag = ba + i*bs2; 446671cb588SHong Zhang u = ba + ili*bs2; 447671cb588SHong Zhang ierr = PetscMemzero(uik,bs2*sizeof(MatScalar));CHKERRQ(ierr); 448671cb588SHong Zhang Kernel_A_gets_A_minus_B_times_C(bs,uik,diag,u); 449671cb588SHong Zhang 450671cb588SHong Zhang /* update D(k) += -U(i,k)^T * U_bar(i,k) */ 451671cb588SHong Zhang Kernel_A_gets_A_plus_Btranspose_times_C(bs,dk,uik,u); 452671cb588SHong Zhang 453671cb588SHong Zhang /* update -U(i,k) */ 454671cb588SHong Zhang ierr = PetscMemcpy(ba+ili*bs2,uik,bs2*sizeof(MatScalar));CHKERRQ(ierr); 455671cb588SHong Zhang 456671cb588SHong Zhang /* add multiple of row i to k-th row ... */ 457671cb588SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 458671cb588SHong Zhang if (jmin < jmax){ 459671cb588SHong Zhang for (j=jmin; j<jmax; j++) { 460671cb588SHong Zhang /* rtmp += -U(i,k)^T * U_bar(i,j) */ 461671cb588SHong Zhang rtmp_ptr = rtmp + bj[j]*bs2; 462671cb588SHong Zhang u = ba + j*bs2; 463671cb588SHong Zhang Kernel_A_gets_A_plus_Btranspose_times_C(bs,rtmp_ptr,uik,u); 464671cb588SHong Zhang } 465671cb588SHong Zhang 466671cb588SHong Zhang /* ... add i to row list for next nonzero entry */ 467671cb588SHong Zhang il[i] = jmin; /* update il(i) in column k+1, ... mbs-1 */ 468671cb588SHong Zhang j = bj[jmin]; 469671cb588SHong Zhang jl[i] = jl[j]; jl[j] = i; /* update jl */ 470671cb588SHong Zhang } 471671cb588SHong Zhang i = nexti; 472671cb588SHong Zhang } 473671cb588SHong Zhang 474671cb588SHong Zhang /* save nonzero entries in k-th row of U ... */ 475671cb588SHong Zhang 476671cb588SHong Zhang /* invert diagonal block */ 477671cb588SHong Zhang diag = ba+k*bs2; 478671cb588SHong Zhang ierr = PetscMemcpy(diag,dk,bs2*sizeof(MatScalar));CHKERRQ(ierr); 479671cb588SHong Zhang Kernel_A_gets_inverse_A(bs,diag,pivots,work); 480671cb588SHong Zhang 481671cb588SHong Zhang jmin = bi[k]; jmax = bi[k+1]; 482671cb588SHong Zhang if (jmin < jmax) { 483671cb588SHong Zhang for (j=jmin; j<jmax; j++){ 484671cb588SHong Zhang vj = bj[j]; /* block col. index of U */ 485671cb588SHong Zhang u = ba + j*bs2; 486671cb588SHong Zhang rtmp_ptr = rtmp + vj*bs2; 487671cb588SHong Zhang for (k1=0; k1<bs2; k1++){ 488671cb588SHong Zhang *u++ = *rtmp_ptr; 489671cb588SHong Zhang *rtmp_ptr++ = 0.0; 490671cb588SHong Zhang } 491671cb588SHong Zhang } 492671cb588SHong Zhang 493671cb588SHong Zhang /* ... add k to row list for first nonzero entry in k-th row */ 494671cb588SHong Zhang il[k] = jmin; 495671cb588SHong Zhang i = bj[jmin]; 496671cb588SHong Zhang jl[k] = jl[i]; jl[i] = k; 497671cb588SHong Zhang } 498671cb588SHong Zhang } 499671cb588SHong Zhang 500671cb588SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 501671cb588SHong Zhang ierr = PetscFree(il);CHKERRQ(ierr); 502671cb588SHong Zhang ierr = PetscFree(dk);CHKERRQ(ierr); 503671cb588SHong Zhang ierr = PetscFree(pivots);CHKERRQ(ierr); 504671cb588SHong Zhang 505671cb588SHong Zhang C->factor = FACTOR_CHOLESKY; 506671cb588SHong Zhang C->assembled = PETSC_TRUE; 507671cb588SHong Zhang C->preallocated = PETSC_TRUE; 508b0a32e0cSBarry Smith PetscLogFlops(1.3333*bs*bs2*b->mbs); /* from inverting diagonal blocks */ 509671cb588SHong Zhang PetscFunctionReturn(0); 510671cb588SHong Zhang } 511671cb588SHong Zhang 51249b5e25fSSatish Balay /* 513fcf159c0SHong Zhang Numeric U^T*D*U factorization for SBAIJ format. Modified from SNF of YSMP. 514cc0c071aSHong Zhang Version for blocks 2 by 2. 51549b5e25fSSatish Balay */ 5164a2ae208SSatish Balay #undef __FUNCT__ 5174a2ae208SSatish Balay #define __FUNCT__ "MatCholeskyFactorNumeric_SeqSBAIJ_2" 5186f9a564bSHong Zhang int MatCholeskyFactorNumeric_SeqSBAIJ_2(Mat A,Mat *B) 51949b5e25fSSatish Balay { 52049b5e25fSSatish Balay Mat C = *B; 52191602c66SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data,*b = (Mat_SeqSBAIJ *)C->data; 522cc0c071aSHong Zhang IS perm = b->row; 523cc0c071aSHong Zhang int *perm_ptr,ierr,i,j,mbs=a->mbs,*bi=b->i,*bj=b->j; 524cc0c071aSHong Zhang int *ai,*aj,*a2anew,k,k1,jmin,jmax,*jl,*il,vj,nexti,ili; 525a1723e09SHong Zhang MatScalar *ba = b->a,*aa,*ap,*dk,*uik; 526cc0c071aSHong Zhang MatScalar *u,*diag,*rtmp,*rtmp_ptr; 52749b5e25fSSatish Balay 52849b5e25fSSatish Balay PetscFunctionBegin; 529671cb588SHong Zhang 53091602c66SHong Zhang /* initialization */ 53191602c66SHong Zhang /* il and jl record the first nonzero element in each row of the accessing 53291602c66SHong Zhang window U(0:k, k:mbs-1). 53391602c66SHong Zhang jl: list of rows to be added to uneliminated rows 53491602c66SHong Zhang i>= k: jl(i) is the first row to be added to row i 53591602c66SHong Zhang i< k: jl(i) is the row following row i in some list of rows 53691602c66SHong Zhang jl(i) = mbs indicates the end of a list 53791602c66SHong Zhang il(i): points to the first nonzero element in columns k,...,mbs-1 of 53891602c66SHong Zhang row i of U */ 53982502324SSatish Balay ierr = PetscMalloc(4*mbs*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 540cc0c071aSHong Zhang ierr = PetscMemzero(rtmp,4*mbs*sizeof(MatScalar));CHKERRQ(ierr); 54182502324SSatish Balay ierr = PetscMalloc(2*mbs*sizeof(int),&il);CHKERRQ(ierr); 54228de702eSHong Zhang jl = il + mbs; 54391602c66SHong Zhang for (i=0; i<mbs; i++) { 5443845f261SHong Zhang jl[i] = mbs; il[0] = 0; 54591602c66SHong Zhang } 54682502324SSatish Balay ierr = PetscMalloc(8*sizeof(MatScalar),&dk);CHKERRQ(ierr); 54728de702eSHong Zhang uik = dk + 4; 548cc0c071aSHong Zhang ierr = ISGetIndices(perm,&perm_ptr);CHKERRQ(ierr); 549cc0c071aSHong Zhang 550cc0c071aSHong Zhang /* check permutation */ 551cc0c071aSHong Zhang if (!a->permute){ 552cc0c071aSHong Zhang ai = a->i; aj = a->j; aa = a->a; 553cc0c071aSHong Zhang } else { 554cc0c071aSHong Zhang ai = a->inew; aj = a->jnew; 55582502324SSatish Balay ierr = PetscMalloc(4*ai[mbs]*sizeof(MatScalar),&aa);CHKERRQ(ierr); 556cc0c071aSHong Zhang ierr = PetscMemcpy(aa,a->a,4*ai[mbs]*sizeof(MatScalar));CHKERRQ(ierr); 55782502324SSatish Balay ierr = PetscMalloc(ai[mbs]*sizeof(int),&a2anew);CHKERRQ(ierr); 558cc0c071aSHong Zhang ierr = PetscMemcpy(a2anew,a->a2anew,(ai[mbs])*sizeof(int));CHKERRQ(ierr); 559cc0c071aSHong Zhang 560cc0c071aSHong Zhang for (i=0; i<mbs; i++){ 561cc0c071aSHong Zhang jmin = ai[i]; jmax = ai[i+1]; 562cc0c071aSHong Zhang for (j=jmin; j<jmax; j++){ 563cc0c071aSHong Zhang while (a2anew[j] != j){ 564cc0c071aSHong Zhang k = a2anew[j]; a2anew[j] = a2anew[k]; a2anew[k] = k; 565cc0c071aSHong Zhang for (k1=0; k1<4; k1++){ 566cc0c071aSHong Zhang dk[k1] = aa[k*4+k1]; 567cc0c071aSHong Zhang aa[k*4+k1] = aa[j*4+k1]; 568cc0c071aSHong Zhang aa[j*4+k1] = dk[k1]; 569cc0c071aSHong Zhang } 570cc0c071aSHong Zhang } 571cc0c071aSHong Zhang /* transform columnoriented blocks that lie in the lower triangle to roworiented blocks */ 572cc0c071aSHong Zhang if (i > aj[j]){ 573a1723e09SHong Zhang /* printf("change orientation, row: %d, col: %d\n",i,aj[j]); */ 574cc0c071aSHong Zhang ap = aa + j*4; /* ptr to the beginning of the block */ 575cc0c071aSHong Zhang dk[1] = ap[1]; /* swap ap[1] and ap[2] */ 576cc0c071aSHong Zhang ap[1] = ap[2]; 577cc0c071aSHong Zhang ap[2] = dk[1]; 578cc0c071aSHong Zhang } 579cc0c071aSHong Zhang } 580cc0c071aSHong Zhang } 581ac355199SBarry Smith ierr = PetscFree(a2anew);CHKERRQ(ierr); 582cc0c071aSHong Zhang } 5833845f261SHong Zhang 58491602c66SHong Zhang /* for each row k */ 58591602c66SHong Zhang for (k = 0; k<mbs; k++){ 58691602c66SHong Zhang 58791602c66SHong Zhang /*initialize k-th row with elements nonzero in row perm(k) of A */ 588cc0c071aSHong Zhang jmin = ai[perm_ptr[k]]; jmax = ai[perm_ptr[k]+1]; 589cc0c071aSHong Zhang ap = aa + jmin*4; 59091602c66SHong Zhang for (j = jmin; j < jmax; j++){ 591cc0c071aSHong Zhang vj = perm_ptr[aj[j]]; /* block col. index */ 592cc0c071aSHong Zhang rtmp_ptr = rtmp + vj*4; 593cc0c071aSHong Zhang for (i=0; i<4; i++) *rtmp_ptr++ = *ap++; 59491602c66SHong Zhang } 59591602c66SHong Zhang 59691602c66SHong Zhang /* modify k-th row by adding in those rows i with U(i,k) != 0 */ 597cc0c071aSHong Zhang ierr = PetscMemcpy(dk,rtmp+k*4,4*sizeof(MatScalar));CHKERRQ(ierr); 59891602c66SHong Zhang i = jl[k]; /* first row to be added to k_th row */ 59991602c66SHong Zhang 600*057f5ba7SHong Zhang while (i < k){ 60191602c66SHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 60291602c66SHong Zhang 6033845f261SHong Zhang /* compute multiplier */ 60491602c66SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 6053845f261SHong Zhang 6063845f261SHong Zhang /* uik = -inv(Di)*U_bar(i,k): - ba[ili]*ba[i] */ 607cc0c071aSHong Zhang diag = ba + i*4; 608cc0c071aSHong Zhang u = ba + ili*4; 609cc0c071aSHong Zhang uik[0] = -(diag[0]*u[0] + diag[2]*u[1]); 610cc0c071aSHong Zhang uik[1] = -(diag[1]*u[0] + diag[3]*u[1]); 611cc0c071aSHong Zhang uik[2] = -(diag[0]*u[2] + diag[2]*u[3]); 612cc0c071aSHong Zhang uik[3] = -(diag[1]*u[2] + diag[3]*u[3]); 6133845f261SHong Zhang 6143845f261SHong Zhang /* update D(k) += -U(i,k)^T * U_bar(i,k): dk += uik*ba[ili] */ 615cc0c071aSHong Zhang dk[0] += uik[0]*u[0] + uik[1]*u[1]; 616cc0c071aSHong Zhang dk[1] += uik[2]*u[0] + uik[3]*u[1]; 617cc0c071aSHong Zhang dk[2] += uik[0]*u[2] + uik[1]*u[3]; 618cc0c071aSHong Zhang dk[3] += uik[2]*u[2] + uik[3]*u[3]; 6193845f261SHong Zhang 6203845f261SHong Zhang /* update -U(i,k): ba[ili] = uik */ 621cc0c071aSHong Zhang ierr = PetscMemcpy(ba+ili*4,uik,4*sizeof(MatScalar));CHKERRQ(ierr); 62291602c66SHong Zhang 62391602c66SHong Zhang /* add multiple of row i to k-th row ... */ 62491602c66SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 62591602c66SHong Zhang if (jmin < jmax){ 6263845f261SHong Zhang for (j=jmin; j<jmax; j++) { 6273845f261SHong Zhang /* rtmp += -U(i,k)^T * U_bar(i,j): rtmp[bj[j]] += uik*ba[j]; */ 628cc0c071aSHong Zhang rtmp_ptr = rtmp + bj[j]*4; 629cc0c071aSHong Zhang u = ba + j*4; 630cc0c071aSHong Zhang rtmp_ptr[0] += uik[0]*u[0] + uik[1]*u[1]; 631cc0c071aSHong Zhang rtmp_ptr[1] += uik[2]*u[0] + uik[3]*u[1]; 632cc0c071aSHong Zhang rtmp_ptr[2] += uik[0]*u[2] + uik[1]*u[3]; 633cc0c071aSHong Zhang rtmp_ptr[3] += uik[2]*u[2] + uik[3]*u[3]; 6343845f261SHong Zhang } 6353845f261SHong Zhang 63691602c66SHong Zhang /* ... add i to row list for next nonzero entry */ 63791602c66SHong Zhang il[i] = jmin; /* update il(i) in column k+1, ... mbs-1 */ 63891602c66SHong Zhang j = bj[jmin]; 63991602c66SHong Zhang jl[i] = jl[j]; jl[j] = i; /* update jl */ 64091602c66SHong Zhang } 641a1723e09SHong Zhang i = nexti; 64291602c66SHong Zhang } 643cc0c071aSHong Zhang 64491602c66SHong Zhang /* save nonzero entries in k-th row of U ... */ 6453845f261SHong Zhang 646cc0c071aSHong Zhang /* invert diagonal block */ 647cc0c071aSHong Zhang diag = ba+k*4; 648cc0c071aSHong Zhang ierr = PetscMemcpy(diag,dk,4*sizeof(MatScalar));CHKERRQ(ierr); 6493845f261SHong Zhang ierr = Kernel_A_gets_inverse_A_2(diag);CHKERRQ(ierr); 6503845f261SHong Zhang 65191602c66SHong Zhang jmin = bi[k]; jmax = bi[k+1]; 65291602c66SHong Zhang if (jmin < jmax) { 65391602c66SHong Zhang for (j=jmin; j<jmax; j++){ 654cc0c071aSHong Zhang vj = bj[j]; /* block col. index of U */ 655cc0c071aSHong Zhang u = ba + j*4; 656cc0c071aSHong Zhang rtmp_ptr = rtmp + vj*4; 657cc0c071aSHong Zhang for (k1=0; k1<4; k1++){ 658cc0c071aSHong Zhang *u++ = *rtmp_ptr; 659cc0c071aSHong Zhang *rtmp_ptr++ = 0.0; 660cc0c071aSHong Zhang } 661cc0c071aSHong Zhang } 6623845f261SHong Zhang 66391602c66SHong Zhang /* ... add k to row list for first nonzero entry in k-th row */ 66491602c66SHong Zhang il[k] = jmin; 66591602c66SHong Zhang i = bj[jmin]; 66691602c66SHong Zhang jl[k] = jl[i]; jl[i] = k; 66791602c66SHong Zhang } 66891602c66SHong Zhang } 6693845f261SHong Zhang 67049b5e25fSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 67191602c66SHong Zhang ierr = PetscFree(il);CHKERRQ(ierr); 6723845f261SHong Zhang ierr = PetscFree(dk);CHKERRQ(ierr); 67391602c66SHong Zhang if (a->permute) { 67491602c66SHong Zhang ierr = PetscFree(aa);CHKERRQ(ierr); 67591602c66SHong Zhang } 676cc0c071aSHong Zhang ierr = ISRestoreIndices(perm,&perm_ptr);CHKERRQ(ierr); 67791602c66SHong Zhang C->factor = FACTOR_CHOLESKY; 67849b5e25fSSatish Balay C->assembled = PETSC_TRUE; 6795c0bcdfcSHong Zhang C->preallocated = PETSC_TRUE; 680b0a32e0cSBarry Smith PetscLogFlops(1.3333*8*b->mbs); /* from inverting diagonal blocks */ 68149b5e25fSSatish Balay PetscFunctionReturn(0); 68249b5e25fSSatish Balay } 68391602c66SHong Zhang 68449b5e25fSSatish Balay /* 68549b5e25fSSatish Balay Version for when blocks are 2 by 2 Using natural ordering 68649b5e25fSSatish Balay */ 6874a2ae208SSatish Balay #undef __FUNCT__ 6884a2ae208SSatish Balay #define __FUNCT__ "MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering" 6896f9a564bSHong Zhang int MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering(Mat A,Mat *B) 69049b5e25fSSatish Balay { 69149b5e25fSSatish Balay Mat C = *B; 692ab27746eSHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data,*b = (Mat_SeqSBAIJ *)C->data; 693ab27746eSHong Zhang int ierr,i,j,mbs=a->mbs,*bi=b->i,*bj=b->j; 69471149678SHong Zhang int *ai,*aj,k,k1,jmin,jmax,*jl,*il,vj,nexti,ili; 695ab27746eSHong Zhang MatScalar *ba = b->a,*aa,*ap,*dk,*uik; 696ab27746eSHong Zhang MatScalar *u,*diag,*rtmp,*rtmp_ptr; 69749b5e25fSSatish Balay 69849b5e25fSSatish Balay PetscFunctionBegin; 699671cb588SHong Zhang 700ab27746eSHong Zhang /* initialization */ 701ab27746eSHong Zhang /* il and jl record the first nonzero element in each row of the accessing 702ab27746eSHong Zhang window U(0:k, k:mbs-1). 703ab27746eSHong Zhang jl: list of rows to be added to uneliminated rows 704ab27746eSHong Zhang i>= k: jl(i) is the first row to be added to row i 705ab27746eSHong Zhang i< k: jl(i) is the row following row i in some list of rows 706ab27746eSHong Zhang jl(i) = mbs indicates the end of a list 707ab27746eSHong Zhang il(i): points to the first nonzero element in columns k,...,mbs-1 of 708ab27746eSHong Zhang row i of U */ 70982502324SSatish Balay ierr = PetscMalloc(4*mbs*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 710ab27746eSHong Zhang ierr = PetscMemzero(rtmp,4*mbs*sizeof(MatScalar));CHKERRQ(ierr); 71182502324SSatish Balay ierr = PetscMalloc(2*mbs*sizeof(int),&il);CHKERRQ(ierr); 71228de702eSHong Zhang jl = il + mbs; 713ab27746eSHong Zhang for (i=0; i<mbs; i++) { 714ab27746eSHong Zhang jl[i] = mbs; il[0] = 0; 71549b5e25fSSatish Balay } 71682502324SSatish Balay ierr = PetscMalloc(8*sizeof(MatScalar),&dk);CHKERRQ(ierr); 71728de702eSHong Zhang uik = dk + 4; 718ab27746eSHong Zhang 719ab27746eSHong Zhang ai = a->i; aj = a->j; aa = a->a; 720ab27746eSHong Zhang 721ab27746eSHong Zhang /* for each row k */ 722ab27746eSHong Zhang for (k = 0; k<mbs; k++){ 723ab27746eSHong Zhang 724ab27746eSHong Zhang /*initialize k-th row with elements nonzero in row k of A */ 725ab27746eSHong Zhang jmin = ai[k]; jmax = ai[k+1]; 726ab27746eSHong Zhang ap = aa + jmin*4; 727ab27746eSHong Zhang for (j = jmin; j < jmax; j++){ 728ab27746eSHong Zhang vj = aj[j]; /* block col. index */ 729ab27746eSHong Zhang rtmp_ptr = rtmp + vj*4; 730ab27746eSHong Zhang for (i=0; i<4; i++) *rtmp_ptr++ = *ap++; 73149b5e25fSSatish Balay } 732ab27746eSHong Zhang 733ab27746eSHong Zhang /* modify k-th row by adding in those rows i with U(i,k) != 0 */ 734ab27746eSHong Zhang ierr = PetscMemcpy(dk,rtmp+k*4,4*sizeof(MatScalar));CHKERRQ(ierr); 735ab27746eSHong Zhang i = jl[k]; /* first row to be added to k_th row */ 736ab27746eSHong Zhang 737*057f5ba7SHong Zhang while (i < k){ 738ab27746eSHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 739ab27746eSHong Zhang 740ab27746eSHong Zhang /* compute multiplier */ 741ab27746eSHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 742ab27746eSHong Zhang 743ab27746eSHong Zhang /* uik = -inv(Di)*U_bar(i,k): - ba[ili]*ba[i] */ 744ab27746eSHong Zhang diag = ba + i*4; 745ab27746eSHong Zhang u = ba + ili*4; 746ab27746eSHong Zhang uik[0] = -(diag[0]*u[0] + diag[2]*u[1]); 747ab27746eSHong Zhang uik[1] = -(diag[1]*u[0] + diag[3]*u[1]); 748ab27746eSHong Zhang uik[2] = -(diag[0]*u[2] + diag[2]*u[3]); 749ab27746eSHong Zhang uik[3] = -(diag[1]*u[2] + diag[3]*u[3]); 750ab27746eSHong Zhang 751ab27746eSHong Zhang /* update D(k) += -U(i,k)^T * U_bar(i,k): dk += uik*ba[ili] */ 752ab27746eSHong Zhang dk[0] += uik[0]*u[0] + uik[1]*u[1]; 753ab27746eSHong Zhang dk[1] += uik[2]*u[0] + uik[3]*u[1]; 754ab27746eSHong Zhang dk[2] += uik[0]*u[2] + uik[1]*u[3]; 755ab27746eSHong Zhang dk[3] += uik[2]*u[2] + uik[3]*u[3]; 756ab27746eSHong Zhang 757ab27746eSHong Zhang /* update -U(i,k): ba[ili] = uik */ 758ab27746eSHong Zhang ierr = PetscMemcpy(ba+ili*4,uik,4*sizeof(MatScalar));CHKERRQ(ierr); 759ab27746eSHong Zhang 760ab27746eSHong Zhang /* add multiple of row i to k-th row ... */ 761ab27746eSHong Zhang jmin = ili + 1; jmax = bi[i+1]; 762ab27746eSHong Zhang if (jmin < jmax){ 763ab27746eSHong Zhang for (j=jmin; j<jmax; j++) { 764ab27746eSHong Zhang /* rtmp += -U(i,k)^T * U_bar(i,j): rtmp[bj[j]] += uik*ba[j]; */ 765ab27746eSHong Zhang rtmp_ptr = rtmp + bj[j]*4; 766ab27746eSHong Zhang u = ba + j*4; 767ab27746eSHong Zhang rtmp_ptr[0] += uik[0]*u[0] + uik[1]*u[1]; 768ab27746eSHong Zhang rtmp_ptr[1] += uik[2]*u[0] + uik[3]*u[1]; 769ab27746eSHong Zhang rtmp_ptr[2] += uik[0]*u[2] + uik[1]*u[3]; 770ab27746eSHong Zhang rtmp_ptr[3] += uik[2]*u[2] + uik[3]*u[3]; 77149b5e25fSSatish Balay } 772ab27746eSHong Zhang 773ab27746eSHong Zhang /* ... add i to row list for next nonzero entry */ 774ab27746eSHong Zhang il[i] = jmin; /* update il(i) in column k+1, ... mbs-1 */ 775ab27746eSHong Zhang j = bj[jmin]; 776ab27746eSHong Zhang jl[i] = jl[j]; jl[j] = i; /* update jl */ 77749b5e25fSSatish Balay } 778ab27746eSHong Zhang i = nexti; 77949b5e25fSSatish Balay } 780ab27746eSHong Zhang 781ab27746eSHong Zhang /* save nonzero entries in k-th row of U ... */ 782ab27746eSHong Zhang 78349b5e25fSSatish Balay /* invert diagonal block */ 784ab27746eSHong Zhang diag = ba+k*4; 785ab27746eSHong Zhang ierr = PetscMemcpy(diag,dk,4*sizeof(MatScalar));CHKERRQ(ierr); 786ab27746eSHong Zhang ierr = Kernel_A_gets_inverse_A_2(diag);CHKERRQ(ierr); 787ab27746eSHong Zhang 788ab27746eSHong Zhang jmin = bi[k]; jmax = bi[k+1]; 789ab27746eSHong Zhang if (jmin < jmax) { 790ab27746eSHong Zhang for (j=jmin; j<jmax; j++){ 791ab27746eSHong Zhang vj = bj[j]; /* block col. index of U */ 792ab27746eSHong Zhang u = ba + j*4; 793ab27746eSHong Zhang rtmp_ptr = rtmp + vj*4; 794ab27746eSHong Zhang for (k1=0; k1<4; k1++){ 795ab27746eSHong Zhang *u++ = *rtmp_ptr; 796ab27746eSHong Zhang *rtmp_ptr++ = 0.0; 797ab27746eSHong Zhang } 798ab27746eSHong Zhang } 799ab27746eSHong Zhang 800ab27746eSHong Zhang /* ... add k to row list for first nonzero entry in k-th row */ 801ab27746eSHong Zhang il[k] = jmin; 802ab27746eSHong Zhang i = bj[jmin]; 803ab27746eSHong Zhang jl[k] = jl[i]; jl[i] = k; 804ab27746eSHong Zhang } 80549b5e25fSSatish Balay } 80649b5e25fSSatish Balay 80749b5e25fSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 808ab27746eSHong Zhang ierr = PetscFree(il);CHKERRQ(ierr); 809ab27746eSHong Zhang ierr = PetscFree(dk);CHKERRQ(ierr); 810ab27746eSHong Zhang 811ab27746eSHong Zhang C->factor = FACTOR_CHOLESKY; 81249b5e25fSSatish Balay C->assembled = PETSC_TRUE; 8135c0bcdfcSHong Zhang C->preallocated = PETSC_TRUE; 814b0a32e0cSBarry Smith PetscLogFlops(1.3333*8*b->mbs); /* from inverting diagonal blocks */ 81549b5e25fSSatish Balay PetscFunctionReturn(0); 81649b5e25fSSatish Balay } 81749b5e25fSSatish Balay 81849b5e25fSSatish Balay /* 8195c0bcdfcSHong Zhang Numeric U^T*D*U factorization for SBAIJ format. Modified from SNF of YSMP. 82091602c66SHong Zhang Version for blocks are 1 by 1. 82149b5e25fSSatish Balay */ 8224a2ae208SSatish Balay #undef __FUNCT__ 8234a2ae208SSatish Balay #define __FUNCT__ "MatCholeskyFactorNumeric_SeqSBAIJ_1" 8246f9a564bSHong Zhang int MatCholeskyFactorNumeric_SeqSBAIJ_1(Mat A,Mat *B) 82549b5e25fSSatish Balay { 82649b5e25fSSatish Balay Mat C = *B; 82749b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data,*b = (Mat_SeqSBAIJ *)C->data; 82849b5e25fSSatish Balay IS ip = b->row; 829351d0355SHong Zhang int *rip,ierr,i,j,mbs = a->mbs,*bi = b->i,*bj = b->j; 830cb718733SHong Zhang int *ai,*aj,*r; 831ab27746eSHong Zhang int k,jmin,jmax,*jl,*il,vj,nexti,ili; 832066653e3SSatish Balay MatScalar *rtmp; 8332d007305SHong Zhang MatScalar *ba = b->a,*aa,ak; 83449b5e25fSSatish Balay MatScalar dk,uikdi; 83549b5e25fSSatish Balay 83649b5e25fSSatish Balay PetscFunctionBegin; 83749b5e25fSSatish Balay ierr = ISGetIndices(ip,&rip);CHKERRQ(ierr); 838cb718733SHong Zhang if (!a->permute){ 8392d007305SHong Zhang ai = a->i; aj = a->j; aa = a->a; 8402d007305SHong Zhang } else { 8412d007305SHong Zhang ai = a->inew; aj = a->jnew; 84282502324SSatish Balay ierr = PetscMalloc(ai[mbs]*sizeof(MatScalar),&aa);CHKERRQ(ierr); 843cb718733SHong Zhang ierr = PetscMemcpy(aa,a->a,ai[mbs]*sizeof(MatScalar));CHKERRQ(ierr); 84482502324SSatish Balay ierr = PetscMalloc(ai[mbs]*sizeof(int),&r);CHKERRQ(ierr); 8452d007305SHong Zhang ierr= PetscMemcpy(r,a->a2anew,(ai[mbs])*sizeof(int));CHKERRQ(ierr); 8462d007305SHong Zhang 8472d007305SHong Zhang jmin = ai[0]; jmax = ai[mbs]; 8482d007305SHong Zhang for (j=jmin; j<jmax; j++){ 849cb718733SHong Zhang while (r[j] != j){ 8502d007305SHong Zhang k = r[j]; r[j] = r[k]; r[k] = k; 8512d007305SHong Zhang ak = aa[k]; aa[k] = aa[j]; aa[j] = ak; 8522d007305SHong Zhang } 8532d007305SHong Zhang } 854ac355199SBarry Smith ierr = PetscFree(r);CHKERRQ(ierr); 8552d007305SHong Zhang } 8562d007305SHong Zhang 85791602c66SHong Zhang /* initialization */ 85849b5e25fSSatish Balay /* il and jl record the first nonzero element in each row of the accessing 85949b5e25fSSatish Balay window U(0:k, k:mbs-1). 86049b5e25fSSatish Balay jl: list of rows to be added to uneliminated rows 86149b5e25fSSatish Balay i>= k: jl(i) is the first row to be added to row i 86249b5e25fSSatish Balay i< k: jl(i) is the row following row i in some list of rows 86349b5e25fSSatish Balay jl(i) = mbs indicates the end of a list 86449b5e25fSSatish Balay il(i): points to the first nonzero element in columns k,...,mbs-1 of 86549b5e25fSSatish Balay row i of U */ 86682502324SSatish Balay ierr = PetscMalloc(mbs*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 86782502324SSatish Balay ierr = PetscMalloc(2*mbs*sizeof(int),&il);CHKERRQ(ierr); 86828de702eSHong Zhang jl = il + mbs; 86949b5e25fSSatish Balay for (i=0; i<mbs; i++) { 87049b5e25fSSatish Balay rtmp[i] = 0.0; jl[i] = mbs; il[0] = 0; 87149b5e25fSSatish Balay } 87249b5e25fSSatish Balay 87391602c66SHong Zhang /* for each row k */ 87449b5e25fSSatish Balay for (k = 0; k<mbs; k++){ 87549b5e25fSSatish Balay 87691602c66SHong Zhang /*initialize k-th row with elements nonzero in row perm(k) of A */ 87749b5e25fSSatish Balay jmin = ai[rip[k]]; jmax = ai[rip[k]+1]; 878*057f5ba7SHong Zhang 87949b5e25fSSatish Balay for (j = jmin; j < jmax; j++){ 880351d0355SHong Zhang vj = rip[aj[j]]; 881ab27746eSHong Zhang rtmp[vj] = aa[j]; 88249b5e25fSSatish Balay } 88349b5e25fSSatish Balay 88491602c66SHong Zhang /* modify k-th row by adding in those rows i with U(i,k) != 0 */ 88549b5e25fSSatish Balay dk = rtmp[k]; 88649b5e25fSSatish Balay i = jl[k]; /* first row to be added to k_th row */ 88749b5e25fSSatish Balay 888*057f5ba7SHong Zhang while (i < k){ 88949b5e25fSSatish Balay nexti = jl[i]; /* next row to be added to k_th row */ 89049b5e25fSSatish Balay 89191602c66SHong Zhang /* compute multiplier, update D(k) and U(i,k) */ 89249b5e25fSSatish Balay ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 89349b5e25fSSatish Balay uikdi = - ba[ili]*ba[i]; 89449b5e25fSSatish Balay dk += uikdi*ba[ili]; 895658e7b3eSHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 89649b5e25fSSatish Balay 89791602c66SHong Zhang /* add multiple of row i to k-th row ... */ 89849b5e25fSSatish Balay jmin = ili + 1; jmax = bi[i+1]; 89949b5e25fSSatish Balay if (jmin < jmax){ 90049b5e25fSSatish Balay for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 90191602c66SHong Zhang /* ... add i to row list for next nonzero entry */ 90249b5e25fSSatish Balay il[i] = jmin; /* update il(i) in column k+1, ... mbs-1 */ 90349b5e25fSSatish Balay j = bj[jmin]; 90449b5e25fSSatish Balay jl[i] = jl[j]; jl[j] = i; /* update jl */ 90549b5e25fSSatish Balay } 906ab27746eSHong Zhang i = nexti; 90749b5e25fSSatish Balay } 90849b5e25fSSatish Balay 90991602c66SHong Zhang /* check for zero pivot and save diagoanl element */ 91049b5e25fSSatish Balay if (dk == 0.0){ 91129bbc08cSBarry Smith SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot"); 912671cb588SHong Zhang /* 9131223c01bSHong Zhang } else if (PetscRealPart(dk) < 0.0){ 9141223c01bSHong Zhang SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Negative pivot: d[%d] = %g\n",k,dk); 915671cb588SHong Zhang */ 91649b5e25fSSatish Balay } 91749b5e25fSSatish Balay 91891602c66SHong Zhang /* save nonzero entries in k-th row of U ... */ 919f3dd7b2dSKris Buschelman ba[k] = 1.0/dk; 92049b5e25fSSatish Balay jmin = bi[k]; jmax = bi[k+1]; 92149b5e25fSSatish Balay if (jmin < jmax) { 92249b5e25fSSatish Balay for (j=jmin; j<jmax; j++){ 923cc0c071aSHong Zhang vj = bj[j]; ba[j] = rtmp[vj]; rtmp[vj] = 0.0; 92449b5e25fSSatish Balay } 92591602c66SHong Zhang /* ... add k to row list for first nonzero entry in k-th row */ 92649b5e25fSSatish Balay il[k] = jmin; 92749b5e25fSSatish Balay i = bj[jmin]; 92849b5e25fSSatish Balay jl[k] = jl[i]; jl[i] = k; 92949b5e25fSSatish Balay } 93049b5e25fSSatish Balay } 93149b5e25fSSatish Balay 93249b5e25fSSatish Balay ierr = PetscFree(rtmp);CHKERRQ(ierr); 93349b5e25fSSatish Balay ierr = PetscFree(il);CHKERRQ(ierr); 934cb718733SHong Zhang if (a->permute){ 935cb718733SHong Zhang ierr = PetscFree(aa);CHKERRQ(ierr); 936cb718733SHong Zhang } 93749b5e25fSSatish Balay 93849b5e25fSSatish Balay ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr); 9398be8c0c7SHong Zhang C->factor = FACTOR_CHOLESKY; 94049b5e25fSSatish Balay C->assembled = PETSC_TRUE; 9415c0bcdfcSHong Zhang C->preallocated = PETSC_TRUE; 942b0a32e0cSBarry Smith PetscLogFlops(b->mbs); 94349b5e25fSSatish Balay PetscFunctionReturn(0); 94449b5e25fSSatish Balay } 94549b5e25fSSatish Balay 946671cb588SHong Zhang /* 947671cb588SHong Zhang Version for when blocks are 1 by 1 Using natural ordering 948671cb588SHong Zhang */ 9494a2ae208SSatish Balay #undef __FUNCT__ 9504a2ae208SSatish Balay #define __FUNCT__ "MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering" 951671cb588SHong Zhang int MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering(Mat A,Mat *B) 952671cb588SHong Zhang { 953671cb588SHong Zhang Mat C = *B; 954671cb588SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data,*b = (Mat_SeqSBAIJ *)C->data; 955671cb588SHong Zhang int ierr,i,j,mbs = a->mbs,*bi = b->i,*bj = b->j; 9561ad1de41SHong Zhang int *ai,*aj; 957671cb588SHong Zhang int k,jmin,jmax,*jl,*il,vj,nexti,ili; 9581ad1de41SHong Zhang MatScalar *rtmp,*ba = b->a,*aa,dk,uikdi; 959671cb588SHong Zhang 960671cb588SHong Zhang PetscFunctionBegin; 961671cb588SHong Zhang 962671cb588SHong Zhang /* initialization */ 963671cb588SHong Zhang /* il and jl record the first nonzero element in each row of the accessing 964671cb588SHong Zhang window U(0:k, k:mbs-1). 965671cb588SHong Zhang jl: list of rows to be added to uneliminated rows 966671cb588SHong Zhang i>= k: jl(i) is the first row to be added to row i 967671cb588SHong Zhang i< k: jl(i) is the row following row i in some list of rows 968671cb588SHong Zhang jl(i) = mbs indicates the end of a list 969671cb588SHong Zhang il(i): points to the first nonzero element in columns k,...,mbs-1 of 970671cb588SHong Zhang row i of U */ 97182502324SSatish Balay ierr = PetscMalloc(mbs*sizeof(MatScalar),&rtmp);CHKERRQ(ierr); 97282502324SSatish Balay ierr = PetscMalloc(2*mbs*sizeof(int),&il);CHKERRQ(ierr); 97328de702eSHong Zhang jl = il + mbs; 974671cb588SHong Zhang for (i=0; i<mbs; i++) { 975671cb588SHong Zhang rtmp[i] = 0.0; jl[i] = mbs; il[0] = 0; 976671cb588SHong Zhang } 977671cb588SHong Zhang 978671cb588SHong Zhang ai = a->i; aj = a->j; aa = a->a; 979671cb588SHong Zhang 980671cb588SHong Zhang /* for each row k */ 981671cb588SHong Zhang for (k = 0; k<mbs; k++){ 982671cb588SHong Zhang 983671cb588SHong Zhang /*initialize k-th row with elements nonzero in row perm(k) of A */ 984671cb588SHong Zhang jmin = ai[k]; jmax = ai[k+1]; 985*057f5ba7SHong Zhang 986671cb588SHong Zhang for (j = jmin; j < jmax; j++){ 987671cb588SHong Zhang vj = aj[j]; 988671cb588SHong Zhang rtmp[vj] = aa[j]; 989671cb588SHong Zhang } 990671cb588SHong Zhang 991671cb588SHong Zhang /* modify k-th row by adding in those rows i with U(i,k) != 0 */ 992671cb588SHong Zhang dk = rtmp[k]; 993671cb588SHong Zhang i = jl[k]; /* first row to be added to k_th row */ 994671cb588SHong Zhang 995*057f5ba7SHong Zhang while (i < k){ 996671cb588SHong Zhang nexti = jl[i]; /* next row to be added to k_th row */ 997671cb588SHong Zhang 998671cb588SHong Zhang /* compute multiplier, update D(k) and U(i,k) */ 999671cb588SHong Zhang ili = il[i]; /* index of first nonzero element in U(i,k:bms-1) */ 1000671cb588SHong Zhang uikdi = - ba[ili]*ba[i]; 1001671cb588SHong Zhang dk += uikdi*ba[ili]; 1002671cb588SHong Zhang ba[ili] = uikdi; /* -U(i,k) */ 1003671cb588SHong Zhang 1004671cb588SHong Zhang /* add multiple of row i to k-th row ... */ 1005671cb588SHong Zhang jmin = ili + 1; jmax = bi[i+1]; 1006671cb588SHong Zhang if (jmin < jmax){ 1007671cb588SHong Zhang for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j]; 1008671cb588SHong Zhang /* ... add i to row list for next nonzero entry */ 1009671cb588SHong Zhang il[i] = jmin; /* update il(i) in column k+1, ... mbs-1 */ 1010671cb588SHong Zhang j = bj[jmin]; 1011671cb588SHong Zhang jl[i] = jl[j]; jl[j] = i; /* update jl */ 1012671cb588SHong Zhang } 1013671cb588SHong Zhang i = nexti; 1014671cb588SHong Zhang } 1015671cb588SHong Zhang 1016671cb588SHong Zhang /* check for zero pivot and save diagoanl element */ 1017671cb588SHong Zhang if (dk == 0.0){ 1018671cb588SHong Zhang SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot"); 1019671cb588SHong Zhang /* 1020671cb588SHong Zhang } else if (PetscRealPart(dk) < 0){ 10211223c01bSHong Zhang SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Negative pivot: d[%d] = %g\n",k,dk); 1022671cb588SHong Zhang */ 1023671cb588SHong Zhang } 1024671cb588SHong Zhang 1025671cb588SHong Zhang /* save nonzero entries in k-th row of U ... */ 1026671cb588SHong Zhang ba[k] = 1.0/dk; 1027671cb588SHong Zhang jmin = bi[k]; jmax = bi[k+1]; 1028671cb588SHong Zhang if (jmin < jmax) { 1029671cb588SHong Zhang for (j=jmin; j<jmax; j++){ 1030671cb588SHong Zhang vj = bj[j]; ba[j] = rtmp[vj]; rtmp[vj] = 0.0; 1031671cb588SHong Zhang } 1032671cb588SHong Zhang /* ... add k to row list for first nonzero entry in k-th row */ 1033671cb588SHong Zhang il[k] = jmin; 1034671cb588SHong Zhang i = bj[jmin]; 1035671cb588SHong Zhang jl[k] = jl[i]; jl[i] = k; 1036671cb588SHong Zhang } 1037671cb588SHong Zhang } 1038671cb588SHong Zhang 1039671cb588SHong Zhang ierr = PetscFree(rtmp);CHKERRQ(ierr); 1040671cb588SHong Zhang ierr = PetscFree(il);CHKERRQ(ierr); 1041671cb588SHong Zhang 1042671cb588SHong Zhang C->factor = FACTOR_CHOLESKY; 1043671cb588SHong Zhang C->assembled = PETSC_TRUE; 1044671cb588SHong Zhang C->preallocated = PETSC_TRUE; 1045b0a32e0cSBarry Smith PetscLogFlops(b->mbs); 1046671cb588SHong Zhang PetscFunctionReturn(0); 1047671cb588SHong Zhang } 1048671cb588SHong Zhang 10494a2ae208SSatish Balay #undef __FUNCT__ 10504a2ae208SSatish Balay #define __FUNCT__ "MatCholeskyFactor_SeqSBAIJ" 1051b8b23757SHong Zhang int MatCholeskyFactor_SeqSBAIJ(Mat A,IS perm,PetscReal f) 105249b5e25fSSatish Balay { 10534445ddedSHong Zhang int ierr; 105449b5e25fSSatish Balay Mat C; 105549b5e25fSSatish Balay 105649b5e25fSSatish Balay PetscFunctionBegin; 1057b8b23757SHong Zhang ierr = MatCholeskyFactorSymbolic(A,perm,f,&C);CHKERRQ(ierr); 1058a4ada70bSHong Zhang ierr = MatCholeskyFactorNumeric(A,&C);CHKERRQ(ierr); 10594445ddedSHong Zhang ierr = MatHeaderCopy(A,C);CHKERRQ(ierr); 106049b5e25fSSatish Balay PetscFunctionReturn(0); 106149b5e25fSSatish Balay } 106249b5e25fSSatish Balay 106349b5e25fSSatish Balay 1064