xref: /petsc/src/mat/impls/aij/seq/inode.c (revision 9aa7eafd167052bfb398e13ee2edaf1f84a0393a)
14c1414c8SBarry Smith 
24c1414c8SBarry Smith /*
34c1414c8SBarry Smith   This file provides high performance routines for the Inode format (compressed sparse row)
44c1414c8SBarry Smith   by taking advantage of rows with identical nonzero structure (I-nodes).
54c1414c8SBarry Smith */
6c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/aij.h>
74c1414c8SBarry Smith 
84c1414c8SBarry Smith #undef __FUNCT__
94c1414c8SBarry Smith #define __FUNCT__ "Mat_CreateColInode"
104c1414c8SBarry Smith static PetscErrorCode Mat_CreateColInode(Mat A,PetscInt *size,PetscInt **ns)
114c1414c8SBarry Smith {
124c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
134c1414c8SBarry Smith   PetscErrorCode ierr;
144c1414c8SBarry Smith   PetscInt       i,count,m,n,min_mn,*ns_row,*ns_col;
154c1414c8SBarry Smith 
164c1414c8SBarry Smith   PetscFunctionBegin;
17d0f46423SBarry Smith   n      = A->cmap->n;
18d0f46423SBarry Smith   m      = A->rmap->n;
194c1414c8SBarry Smith   ns_row = a->inode.size;
204c1414c8SBarry Smith 
214c1414c8SBarry Smith   min_mn = (m < n) ? m : n;
224c1414c8SBarry Smith   if (!ns) {
234c1414c8SBarry Smith     for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) ;
244c1414c8SBarry Smith     for (; count+1 < n; count++,i++) ;
254c1414c8SBarry Smith     if (count < n)  {
264c1414c8SBarry Smith       i++;
274c1414c8SBarry Smith     }
284c1414c8SBarry Smith     *size = i;
294c1414c8SBarry Smith     PetscFunctionReturn(0);
304c1414c8SBarry Smith   }
31854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&ns_col);CHKERRQ(ierr);
324c1414c8SBarry Smith 
334c1414c8SBarry Smith   /* Use the same row structure wherever feasible. */
344c1414c8SBarry Smith   for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) {
354c1414c8SBarry Smith     ns_col[i] = ns_row[i];
364c1414c8SBarry Smith   }
374c1414c8SBarry Smith 
384c1414c8SBarry Smith   /* if m < n; pad up the remainder with inode_limit */
394c1414c8SBarry Smith   for (; count+1 < n; count++,i++) {
404c1414c8SBarry Smith     ns_col[i] = 1;
414c1414c8SBarry Smith   }
424c1414c8SBarry Smith   /* The last node is the odd ball. padd it up with the remaining rows; */
434c1414c8SBarry Smith   if (count < n) {
444c1414c8SBarry Smith     ns_col[i] = n - count;
454c1414c8SBarry Smith     i++;
464c1414c8SBarry Smith   } else if (count > n) {
474c1414c8SBarry Smith     /* Adjust for the over estimation */
484c1414c8SBarry Smith     ns_col[i-1] += n - count;
494c1414c8SBarry Smith   }
504c1414c8SBarry Smith   *size = i;
514c1414c8SBarry Smith   *ns   = ns_col;
524c1414c8SBarry Smith   PetscFunctionReturn(0);
534c1414c8SBarry Smith }
544c1414c8SBarry Smith 
554c1414c8SBarry Smith 
564c1414c8SBarry Smith /*
574c1414c8SBarry Smith       This builds symmetric version of nonzero structure,
584c1414c8SBarry Smith */
594c1414c8SBarry Smith #undef __FUNCT__
604108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode_Symmetric"
611a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Symmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift)
624c1414c8SBarry Smith {
634c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
644c1414c8SBarry Smith   PetscErrorCode ierr;
658758e1faSBarry Smith   PetscInt       *work,*ia,*ja,nz,nslim_row,nslim_col,m,row,col,n;
668758e1faSBarry Smith   PetscInt       *tns,*tvc,*ns_row = a->inode.size,*ns_col,nsz,i1,i2;
678758e1faSBarry Smith   const PetscInt *j,*jmax,*ai= a->i,*aj = a->j;
684c1414c8SBarry Smith 
694c1414c8SBarry Smith   PetscFunctionBegin;
704c1414c8SBarry Smith   nslim_row = a->inode.node_count;
71d0f46423SBarry Smith   m         = A->rmap->n;
72d0f46423SBarry Smith   n         = A->cmap->n;
73e32f2f54SBarry Smith   if (m != n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatGetRowIJ_SeqAIJ_Inode_Symmetric: Matrix should be square");
744c1414c8SBarry Smith 
754c1414c8SBarry Smith   /* Use the row_inode as column_inode */
764c1414c8SBarry Smith   nslim_col = nslim_row;
774c1414c8SBarry Smith   ns_col    = ns_row;
784c1414c8SBarry Smith 
794c1414c8SBarry Smith   /* allocate space for reformated inode structure */
80854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_col+1,&tns);CHKERRQ(ierr);
81854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&tvc);CHKERRQ(ierr);
824c1414c8SBarry Smith   for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1]+ ns_row[i1];
834c1414c8SBarry Smith 
844c1414c8SBarry Smith   for (i1=0,col=0; i1<nslim_col; ++i1) {
854c1414c8SBarry Smith     nsz = ns_col[i1];
862205254eSKarl Rupp     for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1;
874c1414c8SBarry Smith   }
884c1414c8SBarry Smith   /* allocate space for row pointers */
89854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_row+1,&ia);CHKERRQ(ierr);
904c1414c8SBarry Smith   *iia = ia;
914c1414c8SBarry Smith   ierr = PetscMemzero(ia,(nslim_row+1)*sizeof(PetscInt));CHKERRQ(ierr);
92854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_row+1,&work);CHKERRQ(ierr);
934c1414c8SBarry Smith 
944c1414c8SBarry Smith   /* determine the number of columns in each row */
954c1414c8SBarry Smith   ia[0] = oshift;
964c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
974c1414c8SBarry Smith 
984c1414c8SBarry Smith     j    = aj + ai[row] + ishift;
994c1414c8SBarry Smith     jmax = aj + ai[row+1] + ishift;
1004c1414c8SBarry Smith     col  = *j++ + ishift;
1014c1414c8SBarry Smith     i2   = tvc[col];
1024c1414c8SBarry Smith     while (i2<i1 && j<jmax) { /* 1.[-xx-d-xx--] 2.[-xx-------],off-diagonal elemets */
1034c1414c8SBarry Smith       ia[i1+1]++;
1044c1414c8SBarry Smith       ia[i2+1]++;
1054c1414c8SBarry Smith       i2++;                     /* Start col of next node */
10690d2dec7SBarry Smith       while ((j<jmax) && ((col=*j+ishift)<tns[i2])) ++j;
1074c1414c8SBarry Smith       i2 = tvc[col];
1084c1414c8SBarry Smith     }
1094c1414c8SBarry Smith     if (i2 == i1) ia[i2+1]++;    /* now the diagonal element */
1104c1414c8SBarry Smith   }
1114c1414c8SBarry Smith 
1124c1414c8SBarry Smith   /* shift ia[i] to point to next row */
1134c1414c8SBarry Smith   for (i1=1; i1<nslim_row+1; i1++) {
1144c1414c8SBarry Smith     row        = ia[i1-1];
1154c1414c8SBarry Smith     ia[i1]    += row;
1164c1414c8SBarry Smith     work[i1-1] = row - oshift;
1174c1414c8SBarry Smith   }
1184c1414c8SBarry Smith 
1194c1414c8SBarry Smith   /* allocate space for column pointers */
1204c1414c8SBarry Smith   nz   = ia[nslim_row] + (!ishift);
121785e854fSJed Brown   ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr);
1224c1414c8SBarry Smith   *jja = ja;
1234c1414c8SBarry Smith 
1244c1414c8SBarry Smith   /* loop over lower triangular part putting into ja */
1254c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row += ns_row[i1],i1++) {
1264c1414c8SBarry Smith     j    = aj + ai[row] + ishift;
1274c1414c8SBarry Smith     jmax = aj + ai[row+1] + ishift;
1284c1414c8SBarry Smith     col  = *j++ + ishift;
1294c1414c8SBarry Smith     i2   = tvc[col];
1304c1414c8SBarry Smith     while (i2<i1 && j<jmax) {
1314c1414c8SBarry Smith       ja[work[i2]++] = i1 + oshift;
1324c1414c8SBarry Smith       ja[work[i1]++] = i2 + oshift;
1334c1414c8SBarry Smith       ++i2;
13490d2dec7SBarry Smith       while ((j<jmax) && ((col=*j+ishift)< tns[i2])) ++j; /* Skip rest col indices in this node */
1354c1414c8SBarry Smith       i2 = tvc[col];
1364c1414c8SBarry Smith     }
1374c1414c8SBarry Smith     if (i2 == i1) ja[work[i1]++] = i2 + oshift;
1384c1414c8SBarry Smith 
1394c1414c8SBarry Smith   }
1404c1414c8SBarry Smith   ierr = PetscFree(work);CHKERRQ(ierr);
1414c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
1424c1414c8SBarry Smith   ierr = PetscFree(tvc);CHKERRQ(ierr);
1434c1414c8SBarry Smith   PetscFunctionReturn(0);
1444c1414c8SBarry Smith }
1454c1414c8SBarry Smith 
1464c1414c8SBarry Smith /*
1474c1414c8SBarry Smith       This builds nonsymmetric version of nonzero structure,
1484c1414c8SBarry Smith */
1494c1414c8SBarry Smith #undef __FUNCT__
1504108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric"
1511a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift)
1524c1414c8SBarry Smith {
1534c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
1544c1414c8SBarry Smith   PetscErrorCode ierr;
1558758e1faSBarry Smith   PetscInt       *work,*ia,*ja,nz,nslim_row,n,row,col,*ns_col,nslim_col;
1568758e1faSBarry Smith   PetscInt       *tns,*tvc,nsz,i1,i2;
1578758e1faSBarry Smith   const PetscInt *j,*ai= a->i,*aj = a->j,*ns_row = a->inode.size;
1584c1414c8SBarry Smith 
1594c1414c8SBarry Smith   PetscFunctionBegin;
1604c1414c8SBarry Smith   nslim_row = a->inode.node_count;
161d0f46423SBarry Smith   n         = A->cmap->n;
1624c1414c8SBarry Smith 
1634c1414c8SBarry Smith   /* Create The column_inode for this matrix */
1644c1414c8SBarry Smith   ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr);
1654c1414c8SBarry Smith 
1664c1414c8SBarry Smith   /* allocate space for reformated column_inode structure */
167854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_col +1,&tns);CHKERRQ(ierr);
168854ce69bSBarry Smith   ierr = PetscMalloc1(n + 1,&tvc);CHKERRQ(ierr);
1694c1414c8SBarry Smith   for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1];
1704c1414c8SBarry Smith 
1714c1414c8SBarry Smith   for (i1=0,col=0; i1<nslim_col; ++i1) {
1724c1414c8SBarry Smith     nsz = ns_col[i1];
1732205254eSKarl Rupp     for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1;
1744c1414c8SBarry Smith   }
1754c1414c8SBarry Smith   /* allocate space for row pointers */
176854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_row+1,&ia);CHKERRQ(ierr);
1774c1414c8SBarry Smith   *iia = ia;
1784c1414c8SBarry Smith   ierr = PetscMemzero(ia,(nslim_row+1)*sizeof(PetscInt));CHKERRQ(ierr);
179854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_row+1,&work);CHKERRQ(ierr);
1804c1414c8SBarry Smith 
1814c1414c8SBarry Smith   /* determine the number of columns in each row */
1824c1414c8SBarry Smith   ia[0] = oshift;
1834c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
1844c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
1854c1414c8SBarry Smith     col = *j++ + ishift;
1864c1414c8SBarry Smith     i2  = tvc[col];
1874c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
1884c1414c8SBarry Smith     while (nz-- > 0) {           /* off-diagonal elemets */
1894c1414c8SBarry Smith       ia[i1+1]++;
1904c1414c8SBarry Smith       i2++;                     /* Start col of next node */
191a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
1924c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
1934c1414c8SBarry Smith     }
1944c1414c8SBarry Smith   }
1954c1414c8SBarry Smith 
1964c1414c8SBarry Smith   /* shift ia[i] to point to next row */
1974c1414c8SBarry Smith   for (i1=1; i1<nslim_row+1; i1++) {
1984c1414c8SBarry Smith     row        = ia[i1-1];
1994c1414c8SBarry Smith     ia[i1]    += row;
2004c1414c8SBarry Smith     work[i1-1] = row - oshift;
2014c1414c8SBarry Smith   }
2024c1414c8SBarry Smith 
2034c1414c8SBarry Smith   /* allocate space for column pointers */
2044c1414c8SBarry Smith   nz   = ia[nslim_row] + (!ishift);
205785e854fSJed Brown   ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr);
2064c1414c8SBarry Smith   *jja = ja;
2074c1414c8SBarry Smith 
2084c1414c8SBarry Smith   /* loop over matrix putting into ja */
2094c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
2104c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
2114c1414c8SBarry Smith     col = *j++ + ishift;
2124c1414c8SBarry Smith     i2  = tvc[col];
2134c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
2144c1414c8SBarry Smith     while (nz-- > 0) {
2154c1414c8SBarry Smith       ja[work[i1]++] = i2 + oshift;
2164c1414c8SBarry Smith       ++i2;
217a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
2184c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
2194c1414c8SBarry Smith     }
2204c1414c8SBarry Smith   }
2214c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
2224c1414c8SBarry Smith   ierr = PetscFree(work);CHKERRQ(ierr);
2234c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
2244c1414c8SBarry Smith   ierr = PetscFree(tvc);CHKERRQ(ierr);
2254c1414c8SBarry Smith   PetscFunctionReturn(0);
2264c1414c8SBarry Smith }
2274c1414c8SBarry Smith 
2284c1414c8SBarry Smith #undef __FUNCT__
2294108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode"
2301a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
2314c1414c8SBarry Smith {
2324c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
2334c1414c8SBarry Smith   PetscErrorCode ierr;
2344c1414c8SBarry Smith 
2354c1414c8SBarry Smith   PetscFunctionBegin;
2364c1414c8SBarry Smith   *n = a->inode.node_count;
2374c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
2388f7157efSSatish Balay   if (!blockcompressed) {
2398f7157efSSatish Balay     ierr = MatGetRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);;
2408f7157efSSatish Balay   } else if (symmetric) {
2414108e4d5SBarry Smith     ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
2424c1414c8SBarry Smith   } else {
2434108e4d5SBarry Smith     ierr = MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
2444c1414c8SBarry Smith   }
2454c1414c8SBarry Smith   PetscFunctionReturn(0);
2464c1414c8SBarry Smith }
2474c1414c8SBarry Smith 
2484c1414c8SBarry Smith #undef __FUNCT__
2494108e4d5SBarry Smith #define __FUNCT__ "MatRestoreRowIJ_SeqAIJ_Inode"
2501a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
2514c1414c8SBarry Smith {
2524c1414c8SBarry Smith   PetscErrorCode ierr;
2534c1414c8SBarry Smith 
2544c1414c8SBarry Smith   PetscFunctionBegin;
2554c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
2568f7157efSSatish Balay 
2578f7157efSSatish Balay   if (!blockcompressed) {
2588f7157efSSatish Balay     ierr = MatRestoreRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);;
2598f7157efSSatish Balay   } else {
2604c1414c8SBarry Smith     ierr = PetscFree(*ia);CHKERRQ(ierr);
2614c1414c8SBarry Smith     ierr = PetscFree(*ja);CHKERRQ(ierr);
2628f7157efSSatish Balay   }
2634c1414c8SBarry Smith   PetscFunctionReturn(0);
2644c1414c8SBarry Smith }
2654c1414c8SBarry Smith 
2664c1414c8SBarry Smith /* ----------------------------------------------------------- */
2674c1414c8SBarry Smith 
2684c1414c8SBarry Smith #undef __FUNCT__
2694108e4d5SBarry Smith #define __FUNCT__ "MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric"
2701a83f524SJed Brown static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift)
2714c1414c8SBarry Smith {
2724c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
2734c1414c8SBarry Smith   PetscErrorCode ierr;
2744c1414c8SBarry Smith   PetscInt       *work,*ia,*ja,*j,nz,nslim_row, n,row,col,*ns_col,nslim_col;
2754c1414c8SBarry Smith   PetscInt       *tns,*tvc,*ns_row = a->inode.size,nsz,i1,i2,*ai= a->i,*aj = a->j;
2764c1414c8SBarry Smith 
2774c1414c8SBarry Smith   PetscFunctionBegin;
2784c1414c8SBarry Smith   nslim_row = a->inode.node_count;
279d0f46423SBarry Smith   n         = A->cmap->n;
2804c1414c8SBarry Smith 
2814c1414c8SBarry Smith   /* Create The column_inode for this matrix */
2824c1414c8SBarry Smith   ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr);
2834c1414c8SBarry Smith 
2844c1414c8SBarry Smith   /* allocate space for reformated column_inode structure */
285854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_col + 1,&tns);CHKERRQ(ierr);
286854ce69bSBarry Smith   ierr = PetscMalloc1(n + 1,&tvc);CHKERRQ(ierr);
2874c1414c8SBarry Smith   for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1];
2884c1414c8SBarry Smith 
2894c1414c8SBarry Smith   for (i1=0,col=0; i1<nslim_col; ++i1) {
2904c1414c8SBarry Smith     nsz = ns_col[i1];
2912205254eSKarl Rupp     for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1;
2924c1414c8SBarry Smith   }
2934c1414c8SBarry Smith   /* allocate space for column pointers */
294854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_col+1,&ia);CHKERRQ(ierr);
2954c1414c8SBarry Smith   *iia = ia;
2964c1414c8SBarry Smith   ierr = PetscMemzero(ia,(nslim_col+1)*sizeof(PetscInt));CHKERRQ(ierr);
297854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_col+1,&work);CHKERRQ(ierr);
2984c1414c8SBarry Smith 
2994c1414c8SBarry Smith   /* determine the number of columns in each row */
3004c1414c8SBarry Smith   ia[0] = oshift;
3014c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
3024c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
3034c1414c8SBarry Smith     col = *j++ + ishift;
3044c1414c8SBarry Smith     i2  = tvc[col];
3054c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
3064c1414c8SBarry Smith     while (nz-- > 0) {           /* off-diagonal elemets */
3074c1414c8SBarry Smith       /* ia[i1+1]++; */
3084c1414c8SBarry Smith       ia[i2+1]++;
3094c1414c8SBarry Smith       i2++;
310a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
3114c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
3124c1414c8SBarry Smith     }
3134c1414c8SBarry Smith   }
3144c1414c8SBarry Smith 
3154c1414c8SBarry Smith   /* shift ia[i] to point to next col */
3164c1414c8SBarry Smith   for (i1=1; i1<nslim_col+1; i1++) {
3174c1414c8SBarry Smith     col        = ia[i1-1];
3184c1414c8SBarry Smith     ia[i1]    += col;
3194c1414c8SBarry Smith     work[i1-1] = col - oshift;
3204c1414c8SBarry Smith   }
3214c1414c8SBarry Smith 
3224c1414c8SBarry Smith   /* allocate space for column pointers */
3234c1414c8SBarry Smith   nz   = ia[nslim_col] + (!ishift);
324785e854fSJed Brown   ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr);
3254c1414c8SBarry Smith   *jja = ja;
3264c1414c8SBarry Smith 
3274c1414c8SBarry Smith   /* loop over matrix putting into ja */
3284c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
3294c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
3304c1414c8SBarry Smith     col = *j++ + ishift;
3314c1414c8SBarry Smith     i2  = tvc[col];
3324c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
3334c1414c8SBarry Smith     while (nz-- > 0) {
3344c1414c8SBarry Smith       /* ja[work[i1]++] = i2 + oshift; */
3354c1414c8SBarry Smith       ja[work[i2]++] = i1 + oshift;
3364c1414c8SBarry Smith       i2++;
337a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
3384c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
3394c1414c8SBarry Smith     }
3404c1414c8SBarry Smith   }
3414c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
3424c1414c8SBarry Smith   ierr = PetscFree(work);CHKERRQ(ierr);
3434c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
3444c1414c8SBarry Smith   ierr = PetscFree(tvc);CHKERRQ(ierr);
3454c1414c8SBarry Smith   PetscFunctionReturn(0);
3464c1414c8SBarry Smith }
3474c1414c8SBarry Smith 
3484c1414c8SBarry Smith #undef __FUNCT__
3494108e4d5SBarry Smith #define __FUNCT__ "MatGetColumnIJ_SeqAIJ_Inode"
3501a83f524SJed Brown static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
3514c1414c8SBarry Smith {
3524c1414c8SBarry Smith   PetscErrorCode ierr;
3534c1414c8SBarry Smith 
3544c1414c8SBarry Smith   PetscFunctionBegin;
3550298fd71SBarry Smith   ierr = Mat_CreateColInode(A,n,NULL);CHKERRQ(ierr);
3564c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
3574c1414c8SBarry Smith 
3588f7157efSSatish Balay   if (!blockcompressed) {
3598f7157efSSatish Balay     ierr = MatGetColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);;
3608f7157efSSatish Balay   } else if (symmetric) {
3614c1414c8SBarry Smith     /* Since the indices are symmetric it does'nt matter */
3624108e4d5SBarry Smith     ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
3634c1414c8SBarry Smith   } else {
3644108e4d5SBarry Smith     ierr = MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
3654c1414c8SBarry Smith   }
3664c1414c8SBarry Smith   PetscFunctionReturn(0);
3674c1414c8SBarry Smith }
3684c1414c8SBarry Smith 
3694c1414c8SBarry Smith #undef __FUNCT__
3704108e4d5SBarry Smith #define __FUNCT__ "MatRestoreColumnIJ_SeqAIJ_Inode"
3711a83f524SJed Brown static PetscErrorCode MatRestoreColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
3724c1414c8SBarry Smith {
3734c1414c8SBarry Smith   PetscErrorCode ierr;
3744c1414c8SBarry Smith 
3754c1414c8SBarry Smith   PetscFunctionBegin;
3764c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
3778f7157efSSatish Balay   if (!blockcompressed) {
3788f7157efSSatish Balay     ierr = MatRestoreColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);;
3798f7157efSSatish Balay   } else {
3804c1414c8SBarry Smith     ierr = PetscFree(*ia);CHKERRQ(ierr);
3814c1414c8SBarry Smith     ierr = PetscFree(*ja);CHKERRQ(ierr);
3828f7157efSSatish Balay   }
3834c1414c8SBarry Smith   PetscFunctionReturn(0);
3844c1414c8SBarry Smith }
3854c1414c8SBarry Smith 
3864c1414c8SBarry Smith /* ----------------------------------------------------------- */
3874c1414c8SBarry Smith 
3884c1414c8SBarry Smith #undef __FUNCT__
3894108e4d5SBarry Smith #define __FUNCT__ "MatMult_SeqAIJ_Inode"
3904108e4d5SBarry Smith static PetscErrorCode MatMult_SeqAIJ_Inode(Mat A,Vec xx,Vec yy)
3914c1414c8SBarry Smith {
3924c1414c8SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3934c1414c8SBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5,tmp0,tmp1;
394d9fead3dSBarry Smith   PetscScalar       *y;
395dd6ea824SBarry Smith   const PetscScalar *x;
396dd6ea824SBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5;
3974c1414c8SBarry Smith   PetscErrorCode    ierr;
3988758e1faSBarry Smith   PetscInt          i1,i2,n,i,row,node_max,nsz,sz,nonzerorow=0;
3998758e1faSBarry Smith   const PetscInt    *idx,*ns,*ii;
4004c1414c8SBarry Smith 
4014c1414c8SBarry Smith #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
4024c1414c8SBarry Smith #pragma disjoint(*x,*y,*v1,*v2,*v3,*v4,*v5)
4034c1414c8SBarry Smith #endif
4044c1414c8SBarry Smith 
4054c1414c8SBarry Smith   PetscFunctionBegin;
406e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
4074c1414c8SBarry Smith   node_max = a->inode.node_count;
4084c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
4093649974fSBarry Smith   ierr     = VecGetArrayRead(xx,&x);CHKERRQ(ierr);
4104c1414c8SBarry Smith   ierr     = VecGetArray(yy,&y);CHKERRQ(ierr);
4114c1414c8SBarry Smith   idx      = a->j;
4124c1414c8SBarry Smith   v1       = a->a;
4134c1414c8SBarry Smith   ii       = a->i;
4144c1414c8SBarry Smith 
4154c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i) {
4164c1414c8SBarry Smith     nsz         = ns[i];
4174c1414c8SBarry Smith     n           = ii[1] - ii[0];
41898c9bda7SSatish Balay     nonzerorow += (n>0)*nsz;
4194c1414c8SBarry Smith     ii         += nsz;
42050d8bf02SJed Brown     PetscPrefetchBlock(idx+nsz*n,n,0,PETSC_PREFETCH_HINT_NTA);    /* Prefetch the indices for the block row after the current one */
42150d8bf02SJed Brown     PetscPrefetchBlock(v1+nsz*n,nsz*n,0,PETSC_PREFETCH_HINT_NTA); /* Prefetch the values for the block row after the current one  */
4224c1414c8SBarry Smith     sz = n;                     /* No of non zeros in this row */
4234c1414c8SBarry Smith                                 /* Switch on the size of Node */
4244c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
4254c1414c8SBarry Smith     case 1:
42675567043SBarry Smith       sum1 = 0.;
4274c1414c8SBarry Smith 
4284c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4294c1414c8SBarry Smith         i1    = idx[0];         /* The instructions are ordered to */
4304c1414c8SBarry Smith         i2    = idx[1];         /* make the compiler's job easy */
4314c1414c8SBarry Smith         idx  += 2;
4324c1414c8SBarry Smith         tmp0  = x[i1];
4334c1414c8SBarry Smith         tmp1  = x[i2];
4344c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4354c1414c8SBarry Smith       }
4364c1414c8SBarry Smith 
4374c1414c8SBarry Smith       if (n == sz-1) {          /* Take care of the last nonzero  */
4384c1414c8SBarry Smith         tmp0  = x[*idx++];
4394c1414c8SBarry Smith         sum1 += *v1++ *tmp0;
4404c1414c8SBarry Smith       }
4414c1414c8SBarry Smith       y[row++]=sum1;
4424c1414c8SBarry Smith       break;
4434c1414c8SBarry Smith     case 2:
44475567043SBarry Smith       sum1 = 0.;
44575567043SBarry Smith       sum2 = 0.;
4464c1414c8SBarry Smith       v2   = v1 + n;
4474c1414c8SBarry Smith 
4484c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4494c1414c8SBarry Smith         i1    = idx[0];
4504c1414c8SBarry Smith         i2    = idx[1];
4514c1414c8SBarry Smith         idx  += 2;
4524c1414c8SBarry Smith         tmp0  = x[i1];
4534c1414c8SBarry Smith         tmp1  = x[i2];
4544c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4554c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
4564c1414c8SBarry Smith       }
4574c1414c8SBarry Smith       if (n == sz-1) {
4584c1414c8SBarry Smith         tmp0  = x[*idx++];
4594c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
4604c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
4614c1414c8SBarry Smith       }
4624c1414c8SBarry Smith       y[row++]=sum1;
4634c1414c8SBarry Smith       y[row++]=sum2;
4644c1414c8SBarry Smith       v1      =v2;              /* Since the next block to be processed starts there*/
4654c1414c8SBarry Smith       idx    +=sz;
4664c1414c8SBarry Smith       break;
4674c1414c8SBarry Smith     case 3:
46875567043SBarry Smith       sum1 = 0.;
46975567043SBarry Smith       sum2 = 0.;
47075567043SBarry Smith       sum3 = 0.;
4714c1414c8SBarry Smith       v2   = v1 + n;
4724c1414c8SBarry Smith       v3   = v2 + n;
4734c1414c8SBarry Smith 
4744c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4754c1414c8SBarry Smith         i1    = idx[0];
4764c1414c8SBarry Smith         i2    = idx[1];
4774c1414c8SBarry Smith         idx  += 2;
4784c1414c8SBarry Smith         tmp0  = x[i1];
4794c1414c8SBarry Smith         tmp1  = x[i2];
4804c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4814c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
4824c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
4834c1414c8SBarry Smith       }
4844c1414c8SBarry Smith       if (n == sz-1) {
4854c1414c8SBarry Smith         tmp0  = x[*idx++];
4864c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
4874c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
4884c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
4894c1414c8SBarry Smith       }
4904c1414c8SBarry Smith       y[row++]=sum1;
4914c1414c8SBarry Smith       y[row++]=sum2;
4924c1414c8SBarry Smith       y[row++]=sum3;
4934c1414c8SBarry Smith       v1      =v3;              /* Since the next block to be processed starts there*/
4944c1414c8SBarry Smith       idx    +=2*sz;
4954c1414c8SBarry Smith       break;
4964c1414c8SBarry Smith     case 4:
49775567043SBarry Smith       sum1 = 0.;
49875567043SBarry Smith       sum2 = 0.;
49975567043SBarry Smith       sum3 = 0.;
50075567043SBarry Smith       sum4 = 0.;
5014c1414c8SBarry Smith       v2   = v1 + n;
5024c1414c8SBarry Smith       v3   = v2 + n;
5034c1414c8SBarry Smith       v4   = v3 + n;
5044c1414c8SBarry Smith 
5054c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
5064c1414c8SBarry Smith         i1    = idx[0];
5074c1414c8SBarry Smith         i2    = idx[1];
5084c1414c8SBarry Smith         idx  += 2;
5094c1414c8SBarry Smith         tmp0  = x[i1];
5104c1414c8SBarry Smith         tmp1  = x[i2];
5114c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
5124c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
5134c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
5144c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
5154c1414c8SBarry Smith       }
5164c1414c8SBarry Smith       if (n == sz-1) {
5174c1414c8SBarry Smith         tmp0  = x[*idx++];
5184c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
5194c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
5204c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
5214c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
5224c1414c8SBarry Smith       }
5234c1414c8SBarry Smith       y[row++]=sum1;
5244c1414c8SBarry Smith       y[row++]=sum2;
5254c1414c8SBarry Smith       y[row++]=sum3;
5264c1414c8SBarry Smith       y[row++]=sum4;
5274c1414c8SBarry Smith       v1      =v4;              /* Since the next block to be processed starts there*/
5284c1414c8SBarry Smith       idx    +=3*sz;
5294c1414c8SBarry Smith       break;
5304c1414c8SBarry Smith     case 5:
53175567043SBarry Smith       sum1 = 0.;
53275567043SBarry Smith       sum2 = 0.;
53375567043SBarry Smith       sum3 = 0.;
53475567043SBarry Smith       sum4 = 0.;
53575567043SBarry Smith       sum5 = 0.;
5364c1414c8SBarry Smith       v2   = v1 + n;
5374c1414c8SBarry Smith       v3   = v2 + n;
5384c1414c8SBarry Smith       v4   = v3 + n;
5394c1414c8SBarry Smith       v5   = v4 + n;
5404c1414c8SBarry Smith 
5414c1414c8SBarry Smith       for (n = 0; n<sz-1; n+=2) {
5424c1414c8SBarry Smith         i1    = idx[0];
5434c1414c8SBarry Smith         i2    = idx[1];
5444c1414c8SBarry Smith         idx  += 2;
5454c1414c8SBarry Smith         tmp0  = x[i1];
5464c1414c8SBarry Smith         tmp1  = x[i2];
5474c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
5484c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
5494c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
5504c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
5514c1414c8SBarry Smith         sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2;
5524c1414c8SBarry Smith       }
5534c1414c8SBarry Smith       if (n == sz-1) {
5544c1414c8SBarry Smith         tmp0  = x[*idx++];
5554c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
5564c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
5574c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
5584c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
5594c1414c8SBarry Smith         sum5 += *v5++ * tmp0;
5604c1414c8SBarry Smith       }
5614c1414c8SBarry Smith       y[row++]=sum1;
5624c1414c8SBarry Smith       y[row++]=sum2;
5634c1414c8SBarry Smith       y[row++]=sum3;
5644c1414c8SBarry Smith       y[row++]=sum4;
5654c1414c8SBarry Smith       y[row++]=sum5;
5664c1414c8SBarry Smith       v1      =v5;       /* Since the next block to be processed starts there */
5674c1414c8SBarry Smith       idx    +=4*sz;
5684c1414c8SBarry Smith       break;
5694c1414c8SBarry Smith     default:
570e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported");
5714c1414c8SBarry Smith     }
5724c1414c8SBarry Smith   }
5733649974fSBarry Smith   ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr);
5744c1414c8SBarry Smith   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
575dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr);
5764c1414c8SBarry Smith   PetscFunctionReturn(0);
5774c1414c8SBarry Smith }
5784c1414c8SBarry Smith /* ----------------------------------------------------------- */
5794108e4d5SBarry Smith /* Almost same code as the MatMult_SeqAIJ_Inode() */
5804c1414c8SBarry Smith #undef __FUNCT__
5814108e4d5SBarry Smith #define __FUNCT__ "MatMultAdd_SeqAIJ_Inode"
5824108e4d5SBarry Smith static PetscErrorCode MatMultAdd_SeqAIJ_Inode(Mat A,Vec xx,Vec zz,Vec yy)
5834c1414c8SBarry Smith {
5844c1414c8SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
5854c1414c8SBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5,tmp0,tmp1;
5868758e1faSBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5;
5878758e1faSBarry Smith   const PetscScalar *x;
5888758e1faSBarry Smith   PetscScalar       *y,*z,*zt;
5894c1414c8SBarry Smith   PetscErrorCode    ierr;
5908758e1faSBarry Smith   PetscInt          i1,i2,n,i,row,node_max,nsz,sz;
5918758e1faSBarry Smith   const PetscInt    *idx,*ns,*ii;
5924c1414c8SBarry Smith 
5934c1414c8SBarry Smith   PetscFunctionBegin;
594e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
5954c1414c8SBarry Smith   node_max = a->inode.node_count;
5964c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
5972205254eSKarl Rupp 
5988758e1faSBarry Smith   ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr);
599f95e332cSBarry Smith   ierr = VecGetArrayPair(zz,yy,&z,&y);CHKERRQ(ierr);
6004c1414c8SBarry Smith   zt = z;
6014c1414c8SBarry Smith 
6024c1414c8SBarry Smith   idx = a->j;
6034c1414c8SBarry Smith   v1  = a->a;
6044c1414c8SBarry Smith   ii  = a->i;
6054c1414c8SBarry Smith 
6064c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i) {
6074c1414c8SBarry Smith     nsz = ns[i];
6084c1414c8SBarry Smith     n   = ii[1] - ii[0];
6094c1414c8SBarry Smith     ii += nsz;
6104c1414c8SBarry Smith     sz  = n;                    /* No of non zeros in this row */
6114c1414c8SBarry Smith                                 /* Switch on the size of Node */
6124c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
6134c1414c8SBarry Smith     case 1:
6144c1414c8SBarry Smith       sum1 = *zt++;
6154c1414c8SBarry Smith 
6164c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6174c1414c8SBarry Smith         i1    = idx[0];         /* The instructions are ordered to */
6184c1414c8SBarry Smith         i2    = idx[1];         /* make the compiler's job easy */
6194c1414c8SBarry Smith         idx  += 2;
6204c1414c8SBarry Smith         tmp0  = x[i1];
6214c1414c8SBarry Smith         tmp1  = x[i2];
6224c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6234c1414c8SBarry Smith       }
6244c1414c8SBarry Smith 
6254c1414c8SBarry Smith       if (n   == sz-1) {          /* Take care of the last nonzero  */
6264c1414c8SBarry Smith         tmp0  = x[*idx++];
6274c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6284c1414c8SBarry Smith       }
6294c1414c8SBarry Smith       y[row++]=sum1;
6304c1414c8SBarry Smith       break;
6314c1414c8SBarry Smith     case 2:
6324c1414c8SBarry Smith       sum1 = *zt++;
6334c1414c8SBarry Smith       sum2 = *zt++;
6344c1414c8SBarry Smith       v2   = v1 + n;
6354c1414c8SBarry Smith 
6364c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6374c1414c8SBarry Smith         i1    = idx[0];
6384c1414c8SBarry Smith         i2    = idx[1];
6394c1414c8SBarry Smith         idx  += 2;
6404c1414c8SBarry Smith         tmp0  = x[i1];
6414c1414c8SBarry Smith         tmp1  = x[i2];
6424c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6434c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
6444c1414c8SBarry Smith       }
6454c1414c8SBarry Smith       if (n   == sz-1) {
6464c1414c8SBarry Smith         tmp0  = x[*idx++];
6474c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6484c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
6494c1414c8SBarry Smith       }
6504c1414c8SBarry Smith       y[row++]=sum1;
6514c1414c8SBarry Smith       y[row++]=sum2;
6524c1414c8SBarry Smith       v1      =v2;              /* Since the next block to be processed starts there*/
6534c1414c8SBarry Smith       idx    +=sz;
6544c1414c8SBarry Smith       break;
6554c1414c8SBarry Smith     case 3:
6564c1414c8SBarry Smith       sum1 = *zt++;
6574c1414c8SBarry Smith       sum2 = *zt++;
6584c1414c8SBarry Smith       sum3 = *zt++;
6594c1414c8SBarry Smith       v2   = v1 + n;
6604c1414c8SBarry Smith       v3   = v2 + n;
6614c1414c8SBarry Smith 
6624c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6634c1414c8SBarry Smith         i1    = idx[0];
6644c1414c8SBarry Smith         i2    = idx[1];
6654c1414c8SBarry Smith         idx  += 2;
6664c1414c8SBarry Smith         tmp0  = x[i1];
6674c1414c8SBarry Smith         tmp1  = x[i2];
6684c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6694c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
6704c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
6714c1414c8SBarry Smith       }
6724c1414c8SBarry Smith       if (n == sz-1) {
6734c1414c8SBarry Smith         tmp0  = x[*idx++];
6744c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6754c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
6764c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
6774c1414c8SBarry Smith       }
6784c1414c8SBarry Smith       y[row++]=sum1;
6794c1414c8SBarry Smith       y[row++]=sum2;
6804c1414c8SBarry Smith       y[row++]=sum3;
6814c1414c8SBarry Smith       v1      =v3;              /* Since the next block to be processed starts there*/
6824c1414c8SBarry Smith       idx    +=2*sz;
6834c1414c8SBarry Smith       break;
6844c1414c8SBarry Smith     case 4:
6854c1414c8SBarry Smith       sum1 = *zt++;
6864c1414c8SBarry Smith       sum2 = *zt++;
6874c1414c8SBarry Smith       sum3 = *zt++;
6884c1414c8SBarry Smith       sum4 = *zt++;
6894c1414c8SBarry Smith       v2   = v1 + n;
6904c1414c8SBarry Smith       v3   = v2 + n;
6914c1414c8SBarry Smith       v4   = v3 + n;
6924c1414c8SBarry Smith 
6934c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6944c1414c8SBarry Smith         i1    = idx[0];
6954c1414c8SBarry Smith         i2    = idx[1];
6964c1414c8SBarry Smith         idx  += 2;
6974c1414c8SBarry Smith         tmp0  = x[i1];
6984c1414c8SBarry Smith         tmp1  = x[i2];
6994c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
7004c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
7014c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
7024c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
7034c1414c8SBarry Smith       }
7044c1414c8SBarry Smith       if (n == sz-1) {
7054c1414c8SBarry Smith         tmp0  = x[*idx++];
7064c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
7074c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
7084c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
7094c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
7104c1414c8SBarry Smith       }
7114c1414c8SBarry Smith       y[row++]=sum1;
7124c1414c8SBarry Smith       y[row++]=sum2;
7134c1414c8SBarry Smith       y[row++]=sum3;
7144c1414c8SBarry Smith       y[row++]=sum4;
7154c1414c8SBarry Smith       v1      =v4;              /* Since the next block to be processed starts there*/
7164c1414c8SBarry Smith       idx    +=3*sz;
7174c1414c8SBarry Smith       break;
7184c1414c8SBarry Smith     case 5:
7194c1414c8SBarry Smith       sum1 = *zt++;
7204c1414c8SBarry Smith       sum2 = *zt++;
7214c1414c8SBarry Smith       sum3 = *zt++;
7224c1414c8SBarry Smith       sum4 = *zt++;
7234c1414c8SBarry Smith       sum5 = *zt++;
7244c1414c8SBarry Smith       v2   = v1 + n;
7254c1414c8SBarry Smith       v3   = v2 + n;
7264c1414c8SBarry Smith       v4   = v3 + n;
7274c1414c8SBarry Smith       v5   = v4 + n;
7284c1414c8SBarry Smith 
7294c1414c8SBarry Smith       for (n = 0; n<sz-1; n+=2) {
7304c1414c8SBarry Smith         i1    = idx[0];
7314c1414c8SBarry Smith         i2    = idx[1];
7324c1414c8SBarry Smith         idx  += 2;
7334c1414c8SBarry Smith         tmp0  = x[i1];
7344c1414c8SBarry Smith         tmp1  = x[i2];
7354c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
7364c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
7374c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
7384c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
7394c1414c8SBarry Smith         sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2;
7404c1414c8SBarry Smith       }
7414c1414c8SBarry Smith       if (n   == sz-1) {
7424c1414c8SBarry Smith         tmp0  = x[*idx++];
7434c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
7444c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
7454c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
7464c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
7474c1414c8SBarry Smith         sum5 += *v5++ * tmp0;
7484c1414c8SBarry Smith       }
7494c1414c8SBarry Smith       y[row++]=sum1;
7504c1414c8SBarry Smith       y[row++]=sum2;
7514c1414c8SBarry Smith       y[row++]=sum3;
7524c1414c8SBarry Smith       y[row++]=sum4;
7534c1414c8SBarry Smith       y[row++]=sum5;
7544c1414c8SBarry Smith       v1      =v5;       /* Since the next block to be processed starts there */
7554c1414c8SBarry Smith       idx    +=4*sz;
7564c1414c8SBarry Smith       break;
7574c1414c8SBarry Smith     default:
758e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported");
7594c1414c8SBarry Smith     }
7604c1414c8SBarry Smith   }
7618758e1faSBarry Smith   ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr);
762f95e332cSBarry Smith   ierr = VecRestoreArrayPair(zz,yy,&z,&y);CHKERRQ(ierr);
763dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
7644c1414c8SBarry Smith   PetscFunctionReturn(0);
7654c1414c8SBarry Smith }
7664c1414c8SBarry Smith 
7674c1414c8SBarry Smith /* ----------------------------------------------------------- */
7684c1414c8SBarry Smith #undef __FUNCT__
769ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_Inode_inplace"
770ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_Inode_inplace(Mat A,Vec bb,Vec xx)
7714c1414c8SBarry Smith {
7724c1414c8SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
7734c1414c8SBarry Smith   IS                iscol = a->col,isrow = a->row;
7744c1414c8SBarry Smith   PetscErrorCode    ierr;
7755d0c19d7SBarry Smith   const PetscInt    *r,*c,*rout,*cout;
7768758e1faSBarry Smith   PetscInt          i,j,n = A->rmap->n,nz;
7778758e1faSBarry Smith   PetscInt          node_max,*ns,row,nsz,aii,i0,i1;
7788758e1faSBarry Smith   const PetscInt    *ai = a->i,*a_j = a->j,*vi,*ad,*aj;
779d9fead3dSBarry Smith   PetscScalar       *x,*tmp,*tmps,tmp0,tmp1;
780d9fead3dSBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5;
781dd6ea824SBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa;
782dd6ea824SBarry Smith   const PetscScalar *b;
7834c1414c8SBarry Smith 
7844c1414c8SBarry Smith   PetscFunctionBegin;
785e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
7864c1414c8SBarry Smith   node_max = a->inode.node_count;
7874c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
7884c1414c8SBarry Smith 
7893649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
7904c1414c8SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
7914c1414c8SBarry Smith   tmp  = a->solve_work;
7924c1414c8SBarry Smith 
7934c1414c8SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
7944c1414c8SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
7954c1414c8SBarry Smith 
7964c1414c8SBarry Smith   /* forward solve the lower triangular */
7974c1414c8SBarry Smith   tmps = tmp;
7984c1414c8SBarry Smith   aa   = a_a;
7994c1414c8SBarry Smith   aj   = a_j;
8004c1414c8SBarry Smith   ad   = a->diag;
8014c1414c8SBarry Smith 
8024c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i) {
8034c1414c8SBarry Smith     nsz = ns[i];
8044c1414c8SBarry Smith     aii = ai[row];
8054c1414c8SBarry Smith     v1  = aa + aii;
8064c1414c8SBarry Smith     vi  = aj + aii;
8074c1414c8SBarry Smith     nz  = ad[row]- aii;
80826549573SJed Brown     if (i < node_max-1) {
80926549573SJed Brown       /* Prefetch the block after the current one, the prefetch itself can't cause a memory error,
81026549573SJed Brown       * but our indexing to determine it's size could. */
81150d8bf02SJed Brown       PetscPrefetchBlock(aj+ai[row+nsz],ad[row+nsz]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); /* indices */
81226549573SJed Brown       /* In my tests, it seems to be better to fetch entire rows instead of just the lower-triangular part */
81350d8bf02SJed Brown       PetscPrefetchBlock(aa+ai[row+nsz],ad[row+nsz+ns[i+1]-1]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA);
81426549573SJed Brown       /* for (j=0; j<ns[i+1]; j++) PetscPrefetchBlock(aa+ai[row+nsz+j],ad[row+nsz+j]-ai[row+nsz+j],0,0); */
81526549573SJed Brown     }
8164c1414c8SBarry Smith 
8174c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
8184c1414c8SBarry Smith     case 1:
8194c1414c8SBarry Smith       sum1 = b[*r++];
8204c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
8214c1414c8SBarry Smith         i0    = vi[0];
8224c1414c8SBarry Smith         i1    = vi[1];
8234c1414c8SBarry Smith         vi   +=2;
8244c1414c8SBarry Smith         tmp0  = tmps[i0];
8254c1414c8SBarry Smith         tmp1  = tmps[i1];
8264c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8274c1414c8SBarry Smith       }
8284c1414c8SBarry Smith       if (j == nz-1) {
8294c1414c8SBarry Smith         tmp0  = tmps[*vi++];
8304c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8314c1414c8SBarry Smith       }
8324c1414c8SBarry Smith       tmp[row++]=sum1;
8334c1414c8SBarry Smith       break;
8344c1414c8SBarry Smith     case 2:
8354c1414c8SBarry Smith       sum1 = b[*r++];
8364c1414c8SBarry Smith       sum2 = b[*r++];
8374c1414c8SBarry Smith       v2   = aa + ai[row+1];
8384c1414c8SBarry Smith 
8394c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
8404c1414c8SBarry Smith         i0    = vi[0];
8414c1414c8SBarry Smith         i1    = vi[1];
8424c1414c8SBarry Smith         vi   +=2;
8434c1414c8SBarry Smith         tmp0  = tmps[i0];
8444c1414c8SBarry Smith         tmp1  = tmps[i1];
8454c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8464c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
8474c1414c8SBarry Smith       }
8484c1414c8SBarry Smith       if (j == nz-1) {
8494c1414c8SBarry Smith         tmp0  = tmps[*vi++];
8504c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8514c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
8524c1414c8SBarry Smith       }
8534c1414c8SBarry Smith       sum2     -= *v2++ *sum1;
8544c1414c8SBarry Smith       tmp[row++]=sum1;
8554c1414c8SBarry Smith       tmp[row++]=sum2;
8564c1414c8SBarry Smith       break;
8574c1414c8SBarry Smith     case 3:
8584c1414c8SBarry Smith       sum1 = b[*r++];
8594c1414c8SBarry Smith       sum2 = b[*r++];
8604c1414c8SBarry Smith       sum3 = b[*r++];
8614c1414c8SBarry Smith       v2   = aa + ai[row+1];
8624c1414c8SBarry Smith       v3   = aa + ai[row+2];
8634c1414c8SBarry Smith 
8644c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
8654c1414c8SBarry Smith         i0    = vi[0];
8664c1414c8SBarry Smith         i1    = vi[1];
8674c1414c8SBarry Smith         vi   +=2;
8684c1414c8SBarry Smith         tmp0  = tmps[i0];
8694c1414c8SBarry Smith         tmp1  = tmps[i1];
8704c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8714c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
8724c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
8734c1414c8SBarry Smith       }
8744c1414c8SBarry Smith       if (j == nz-1) {
8754c1414c8SBarry Smith         tmp0  = tmps[*vi++];
8764c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8774c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
8784c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
8794c1414c8SBarry Smith       }
8804c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
8814c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
8824c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
8832205254eSKarl Rupp 
8844c1414c8SBarry Smith       tmp[row++]=sum1;
8854c1414c8SBarry Smith       tmp[row++]=sum2;
8864c1414c8SBarry Smith       tmp[row++]=sum3;
8874c1414c8SBarry Smith       break;
8884c1414c8SBarry Smith 
8894c1414c8SBarry Smith     case 4:
8904c1414c8SBarry Smith       sum1 = b[*r++];
8914c1414c8SBarry Smith       sum2 = b[*r++];
8924c1414c8SBarry Smith       sum3 = b[*r++];
8934c1414c8SBarry Smith       sum4 = b[*r++];
8944c1414c8SBarry Smith       v2   = aa + ai[row+1];
8954c1414c8SBarry Smith       v3   = aa + ai[row+2];
8964c1414c8SBarry Smith       v4   = aa + ai[row+3];
8974c1414c8SBarry Smith 
8984c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
8994c1414c8SBarry Smith         i0    = vi[0];
9004c1414c8SBarry Smith         i1    = vi[1];
9014c1414c8SBarry Smith         vi   +=2;
9024c1414c8SBarry Smith         tmp0  = tmps[i0];
9034c1414c8SBarry Smith         tmp1  = tmps[i1];
9044c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
9054c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
9064c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
9074c1414c8SBarry Smith         sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
9084c1414c8SBarry Smith       }
9094c1414c8SBarry Smith       if (j == nz-1) {
9104c1414c8SBarry Smith         tmp0  = tmps[*vi++];
9114c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
9124c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
9134c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
9144c1414c8SBarry Smith         sum4 -= *v4++ *tmp0;
9154c1414c8SBarry Smith       }
9164c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
9174c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
9184c1414c8SBarry Smith       sum4 -= *v4++ * sum1;
9194c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
9204c1414c8SBarry Smith       sum4 -= *v4++ * sum2;
9214c1414c8SBarry Smith       sum4 -= *v4++ * sum3;
9224c1414c8SBarry Smith 
9234c1414c8SBarry Smith       tmp[row++]=sum1;
9244c1414c8SBarry Smith       tmp[row++]=sum2;
9254c1414c8SBarry Smith       tmp[row++]=sum3;
9264c1414c8SBarry Smith       tmp[row++]=sum4;
9274c1414c8SBarry Smith       break;
9284c1414c8SBarry Smith     case 5:
9294c1414c8SBarry Smith       sum1 = b[*r++];
9304c1414c8SBarry Smith       sum2 = b[*r++];
9314c1414c8SBarry Smith       sum3 = b[*r++];
9324c1414c8SBarry Smith       sum4 = b[*r++];
9334c1414c8SBarry Smith       sum5 = b[*r++];
9344c1414c8SBarry Smith       v2   = aa + ai[row+1];
9354c1414c8SBarry Smith       v3   = aa + ai[row+2];
9364c1414c8SBarry Smith       v4   = aa + ai[row+3];
9374c1414c8SBarry Smith       v5   = aa + ai[row+4];
9384c1414c8SBarry Smith 
9394c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
9404c1414c8SBarry Smith         i0    = vi[0];
9414c1414c8SBarry Smith         i1    = vi[1];
9424c1414c8SBarry Smith         vi   +=2;
9434c1414c8SBarry Smith         tmp0  = tmps[i0];
9444c1414c8SBarry Smith         tmp1  = tmps[i1];
9454c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
9464c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
9474c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
9484c1414c8SBarry Smith         sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
9494c1414c8SBarry Smith         sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
9504c1414c8SBarry Smith       }
9514c1414c8SBarry Smith       if (j == nz-1) {
9524c1414c8SBarry Smith         tmp0  = tmps[*vi++];
9534c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
9544c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
9554c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
9564c1414c8SBarry Smith         sum4 -= *v4++ *tmp0;
9574c1414c8SBarry Smith         sum5 -= *v5++ *tmp0;
9584c1414c8SBarry Smith       }
9594c1414c8SBarry Smith 
9604c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
9614c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
9624c1414c8SBarry Smith       sum4 -= *v4++ * sum1;
9634c1414c8SBarry Smith       sum5 -= *v5++ * sum1;
9644c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
9654c1414c8SBarry Smith       sum4 -= *v4++ * sum2;
9664c1414c8SBarry Smith       sum5 -= *v5++ * sum2;
9674c1414c8SBarry Smith       sum4 -= *v4++ * sum3;
9684c1414c8SBarry Smith       sum5 -= *v5++ * sum3;
9694c1414c8SBarry Smith       sum5 -= *v5++ * sum4;
9704c1414c8SBarry Smith 
9714c1414c8SBarry Smith       tmp[row++]=sum1;
9724c1414c8SBarry Smith       tmp[row++]=sum2;
9734c1414c8SBarry Smith       tmp[row++]=sum3;
9744c1414c8SBarry Smith       tmp[row++]=sum4;
9754c1414c8SBarry Smith       tmp[row++]=sum5;
9764c1414c8SBarry Smith       break;
9774c1414c8SBarry Smith     default:
978e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
9794c1414c8SBarry Smith     }
9804c1414c8SBarry Smith   }
9814c1414c8SBarry Smith   /* backward solve the upper triangular */
9824c1414c8SBarry Smith   for (i=node_max -1,row = n-1; i>=0; i--) {
9834c1414c8SBarry Smith     nsz = ns[i];
9844c1414c8SBarry Smith     aii = ai[row+1] -1;
9854c1414c8SBarry Smith     v1  = aa + aii;
9864c1414c8SBarry Smith     vi  = aj + aii;
9874c1414c8SBarry Smith     nz  = aii- ad[row];
9884c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
9894c1414c8SBarry Smith     case 1:
9904c1414c8SBarry Smith       sum1 = tmp[row];
9914c1414c8SBarry Smith 
9924c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
9934c1414c8SBarry Smith         vi   -=2;
9944c1414c8SBarry Smith         i0    = vi[2];
9954c1414c8SBarry Smith         i1    = vi[1];
9964c1414c8SBarry Smith         tmp0  = tmps[i0];
9974c1414c8SBarry Smith         tmp1  = tmps[i1];
9984c1414c8SBarry Smith         v1   -= 2;
9994c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10004c1414c8SBarry Smith       }
10014c1414c8SBarry Smith       if (j==1) {
10024c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10034c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10044c1414c8SBarry Smith       }
10054c1414c8SBarry Smith       x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10064c1414c8SBarry Smith       break;
10074c1414c8SBarry Smith     case 2:
10084c1414c8SBarry Smith       sum1 = tmp[row];
10094c1414c8SBarry Smith       sum2 = tmp[row -1];
10104c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10114c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
10124c1414c8SBarry Smith         vi   -=2;
10134c1414c8SBarry Smith         i0    = vi[2];
10144c1414c8SBarry Smith         i1    = vi[1];
10154c1414c8SBarry Smith         tmp0  = tmps[i0];
10164c1414c8SBarry Smith         tmp1  = tmps[i1];
10174c1414c8SBarry Smith         v1   -= 2;
10184c1414c8SBarry Smith         v2   -= 2;
10194c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10204c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10214c1414c8SBarry Smith       }
10224c1414c8SBarry Smith       if (j==1) {
10234c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10244c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10254c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10264c1414c8SBarry Smith       }
10274c1414c8SBarry Smith 
10284c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10294c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10304c1414c8SBarry Smith       x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
10314c1414c8SBarry Smith       break;
10324c1414c8SBarry Smith     case 3:
10334c1414c8SBarry Smith       sum1 = tmp[row];
10344c1414c8SBarry Smith       sum2 = tmp[row -1];
10354c1414c8SBarry Smith       sum3 = tmp[row -2];
10364c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10374c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
10384c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
10394c1414c8SBarry Smith         vi   -=2;
10404c1414c8SBarry Smith         i0    = vi[2];
10414c1414c8SBarry Smith         i1    = vi[1];
10424c1414c8SBarry Smith         tmp0  = tmps[i0];
10434c1414c8SBarry Smith         tmp1  = tmps[i1];
10444c1414c8SBarry Smith         v1   -= 2;
10454c1414c8SBarry Smith         v2   -= 2;
10464c1414c8SBarry Smith         v3   -= 2;
10474c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10484c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10494c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
10504c1414c8SBarry Smith       }
10514c1414c8SBarry Smith       if (j==1) {
10524c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10534c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10544c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10554c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
10564c1414c8SBarry Smith       }
10574c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10584c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10594c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10604c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
10614c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10624c1414c8SBarry Smith       x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
10634c1414c8SBarry Smith 
10644c1414c8SBarry Smith       break;
10654c1414c8SBarry Smith     case 4:
10664c1414c8SBarry Smith       sum1 = tmp[row];
10674c1414c8SBarry Smith       sum2 = tmp[row -1];
10684c1414c8SBarry Smith       sum3 = tmp[row -2];
10694c1414c8SBarry Smith       sum4 = tmp[row -3];
10704c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10714c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
10724c1414c8SBarry Smith       v4   = aa + ai[row -2]-1;
10734c1414c8SBarry Smith 
10744c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
10754c1414c8SBarry Smith         vi   -=2;
10764c1414c8SBarry Smith         i0    = vi[2];
10774c1414c8SBarry Smith         i1    = vi[1];
10784c1414c8SBarry Smith         tmp0  = tmps[i0];
10794c1414c8SBarry Smith         tmp1  = tmps[i1];
10804c1414c8SBarry Smith         v1   -= 2;
10814c1414c8SBarry Smith         v2   -= 2;
10824c1414c8SBarry Smith         v3   -= 2;
10834c1414c8SBarry Smith         v4   -= 2;
10844c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10854c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10864c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
10874c1414c8SBarry Smith         sum4 -= v4[2] * tmp0 + v4[1] * tmp1;
10884c1414c8SBarry Smith       }
10894c1414c8SBarry Smith       if (j==1) {
10904c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10914c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10924c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10934c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
10944c1414c8SBarry Smith         sum4 -= *v4-- * tmp0;
10954c1414c8SBarry Smith       }
10964c1414c8SBarry Smith 
10974c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10984c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10994c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11004c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11014c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
11024c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11034c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11044c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
11054c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11064c1414c8SBarry Smith       x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--;
11074c1414c8SBarry Smith       break;
11084c1414c8SBarry Smith     case 5:
11094c1414c8SBarry Smith       sum1 = tmp[row];
11104c1414c8SBarry Smith       sum2 = tmp[row -1];
11114c1414c8SBarry Smith       sum3 = tmp[row -2];
11124c1414c8SBarry Smith       sum4 = tmp[row -3];
11134c1414c8SBarry Smith       sum5 = tmp[row -4];
11144c1414c8SBarry Smith       v2   = aa + ai[row]-1;
11154c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
11164c1414c8SBarry Smith       v4   = aa + ai[row -2]-1;
11174c1414c8SBarry Smith       v5   = aa + ai[row -3]-1;
11184c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
11194c1414c8SBarry Smith         vi   -= 2;
11204c1414c8SBarry Smith         i0    = vi[2];
11214c1414c8SBarry Smith         i1    = vi[1];
11224c1414c8SBarry Smith         tmp0  = tmps[i0];
11234c1414c8SBarry Smith         tmp1  = tmps[i1];
11244c1414c8SBarry Smith         v1   -= 2;
11254c1414c8SBarry Smith         v2   -= 2;
11264c1414c8SBarry Smith         v3   -= 2;
11274c1414c8SBarry Smith         v4   -= 2;
11284c1414c8SBarry Smith         v5   -= 2;
11294c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
11304c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
11314c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
11324c1414c8SBarry Smith         sum4 -= v4[2] * tmp0 + v4[1] * tmp1;
11334c1414c8SBarry Smith         sum5 -= v5[2] * tmp0 + v5[1] * tmp1;
11344c1414c8SBarry Smith       }
11354c1414c8SBarry Smith       if (j==1) {
11364c1414c8SBarry Smith         tmp0  = tmps[*vi--];
11374c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
11384c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
11394c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
11404c1414c8SBarry Smith         sum4 -= *v4-- * tmp0;
11414c1414c8SBarry Smith         sum5 -= *v5-- * tmp0;
11424c1414c8SBarry Smith       }
11434c1414c8SBarry Smith 
11444c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
11454c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
11464c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11474c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11484c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11494c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
11504c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11514c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11524c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11534c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
11544c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11554c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11564c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--;
11574c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11584c1414c8SBarry Smith       x[*c--] = tmp[row] = sum5*a_a[ad[row]]; row--;
11594c1414c8SBarry Smith       break;
11604c1414c8SBarry Smith     default:
1161e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
11624c1414c8SBarry Smith     }
11634c1414c8SBarry Smith   }
11644c1414c8SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11654c1414c8SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
11663649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
11674c1414c8SBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1168dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
11694c1414c8SBarry Smith   PetscFunctionReturn(0);
11704c1414c8SBarry Smith }
11714c1414c8SBarry Smith 
117228f1b45aSHong Zhang #undef __FUNCT__
117328f1b45aSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_Inode"
1174*9aa7eafdSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_Inode(Mat B,Mat A,MatFactorInfo *info)
117528f1b45aSHong Zhang {
117628f1b45aSHong Zhang   Mat             C     =B;
117728f1b45aSHong Zhang   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
117828f1b45aSHong Zhang   IS              isrow = b->row,isicol = b->icol;
117928f1b45aSHong Zhang   PetscErrorCode  ierr;
118028f1b45aSHong Zhang   const PetscInt  *r,*ic,*ics;
118128f1b45aSHong Zhang   const PetscInt  n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag;
118228f1b45aSHong Zhang   PetscInt        i,j,k,nz,nzL,row,*pj;
118328f1b45aSHong Zhang   const PetscInt  *ajtmp,*bjtmp;
11849877982aSShri Abhyankar   MatScalar       *pc,*pc1,*pc2,*pc3,*pc4,mul1,mul2,mul3,mul4,*pv,*rtmp1,*rtmp2,*rtmp3,*rtmp4;
11859877982aSShri Abhyankar   const MatScalar *aa=a->a,*v,*v1,*v2,*v3,*v4;
118628f1b45aSHong Zhang   FactorShiftCtx  sctx;
11874f81c4b7SBarry Smith   const PetscInt  *ddiag;
118828f1b45aSHong Zhang   PetscReal       rs;
118928f1b45aSHong Zhang   MatScalar       d;
11904f81c4b7SBarry Smith   PetscInt        inod,nodesz,node_max,col;
11914f81c4b7SBarry Smith   const PetscInt  *ns;
119207b50cabSHong Zhang   PetscInt        *tmp_vec1,*tmp_vec2,*nsmap;
11930e95ead3SHong Zhang 
119428f1b45aSHong Zhang   PetscFunctionBegin;
119528f1b45aSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
119628f1b45aSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
119728f1b45aSHong Zhang 
1198f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
119928f1b45aSHong Zhang     ddiag          = a->diag;
120028f1b45aSHong Zhang     sctx.shift_top = info->zeropivot;
120128f1b45aSHong Zhang     for (i=0; i<n; i++) {
120228f1b45aSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
120328f1b45aSHong Zhang       d  = (aa)[ddiag[i]];
120428f1b45aSHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
120528f1b45aSHong Zhang       v  = aa+ai[i];
120628f1b45aSHong Zhang       nz = ai[i+1] - ai[i];
12072205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
120828f1b45aSHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
120928f1b45aSHong Zhang     }
121028f1b45aSHong Zhang     sctx.shift_top *= 1.1;
121128f1b45aSHong Zhang     sctx.nshift_max = 5;
121228f1b45aSHong Zhang     sctx.shift_lo   = 0.;
121328f1b45aSHong Zhang     sctx.shift_hi   = 1.;
121428f1b45aSHong Zhang   }
121528f1b45aSHong Zhang 
121628f1b45aSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
121728f1b45aSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
121868785679SHong Zhang 
12191795a4d1SJed Brown   ierr  = PetscCalloc4(n,&rtmp1,n,&rtmp2,n,&rtmp3,n,&rtmp4);CHKERRQ(ierr);
122028f1b45aSHong Zhang   ics   = ic;
122128f1b45aSHong Zhang 
122228f1b45aSHong Zhang   node_max = a->inode.node_count;
122328f1b45aSHong Zhang   ns       = a->inode.size;
1224e7e72b3dSBarry Smith   if (!ns) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix without inode information");
122528f1b45aSHong Zhang 
12269877982aSShri Abhyankar   /* If max inode size > 4, split it into two inodes.*/
122768785679SHong Zhang   /* also map the inode sizes according to the ordering */
1228854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&tmp_vec1);CHKERRQ(ierr);
122968785679SHong Zhang   for (i=0,j=0; i<node_max; ++i,++j) {
1230b1550197SShri Abhyankar     if (ns[i] > 4) {
1231048b5e81SShri Abhyankar       tmp_vec1[j] = 4;
123268785679SHong Zhang       ++j;
123368785679SHong Zhang       tmp_vec1[j] = ns[i] - tmp_vec1[j-1];
123468785679SHong Zhang     } else {
123568785679SHong Zhang       tmp_vec1[j] = ns[i];
123668785679SHong Zhang     }
123768785679SHong Zhang   }
123868785679SHong Zhang   /* Use the correct node_max */
123968785679SHong Zhang   node_max = j;
124068785679SHong Zhang 
124168785679SHong Zhang   /* Now reorder the inode info based on mat re-ordering info */
124268785679SHong Zhang   /* First create a row -> inode_size_array_index map */
1243854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&nsmap);CHKERRQ(ierr);
1244854ce69bSBarry Smith   ierr = PetscMalloc1(node_max+1,&tmp_vec2);CHKERRQ(ierr);
124568785679SHong Zhang   for (i=0,row=0; i<node_max; i++) {
124668785679SHong Zhang     nodesz = tmp_vec1[i];
124768785679SHong Zhang     for (j=0; j<nodesz; j++,row++) {
124868785679SHong Zhang       nsmap[row] = i;
124968785679SHong Zhang     }
125068785679SHong Zhang   }
125168785679SHong Zhang   /* Using nsmap, create a reordered ns structure */
125268785679SHong Zhang   for (i=0,j=0; i< node_max; i++) {
125368785679SHong Zhang     nodesz      = tmp_vec1[nsmap[r[j]]];     /* here the reordered row_no is in r[] */
125468785679SHong Zhang     tmp_vec2[i] = nodesz;
125568785679SHong Zhang     j          += nodesz;
125668785679SHong Zhang   }
125768785679SHong Zhang   ierr = PetscFree(nsmap);CHKERRQ(ierr);
125868785679SHong Zhang   ierr = PetscFree(tmp_vec1);CHKERRQ(ierr);
1259b89f182dSHong Zhang 
126068785679SHong Zhang   /* Now use the correct ns */
126168785679SHong Zhang   ns = tmp_vec2;
126268785679SHong Zhang 
126328f1b45aSHong Zhang   do {
126407b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
126528f1b45aSHong Zhang     /* Now loop over each block-row, and do the factorization */
126628f1b45aSHong Zhang     for (inod=0,i=0; inod<node_max; inod++) { /* i: row index; inod: inode index */
126728f1b45aSHong Zhang       nodesz = ns[inod];
126828f1b45aSHong Zhang 
126928f1b45aSHong Zhang       switch (nodesz) {
127028f1b45aSHong Zhang       case 1:
127168785679SHong Zhang         /*----------*/
1272b89f182dSHong Zhang         /* zero rtmp1 */
127328f1b45aSHong Zhang         /* L part */
127428f1b45aSHong Zhang         nz    = bi[i+1] - bi[i];
127528f1b45aSHong Zhang         bjtmp = bj + bi[i];
1276b89f182dSHong Zhang         for (j=0; j<nz; j++) rtmp1[bjtmp[j]] = 0.0;
127728f1b45aSHong Zhang 
127828f1b45aSHong Zhang         /* U part */
127928f1b45aSHong Zhang         nz    = bdiag[i]-bdiag[i+1];
128028f1b45aSHong Zhang         bjtmp = bj + bdiag[i+1]+1;
1281b89f182dSHong Zhang         for (j=0; j<nz; j++) rtmp1[bjtmp[j]] = 0.0;
128228f1b45aSHong Zhang 
128328f1b45aSHong Zhang         /* load in initial (unfactored row) */
128428f1b45aSHong Zhang         nz    = ai[r[i]+1] - ai[r[i]];
128528f1b45aSHong Zhang         ajtmp = aj + ai[r[i]];
128628f1b45aSHong Zhang         v     = aa + ai[r[i]];
12872205254eSKarl Rupp         for (j=0; j<nz; j++) rtmp1[ics[ajtmp[j]]] = v[j];
12882205254eSKarl Rupp 
128928f1b45aSHong Zhang         /* ZeropivotApply() */
1290b89f182dSHong Zhang         rtmp1[i] += sctx.shift_amount;  /* shift the diagonal of the matrix */
129128f1b45aSHong Zhang 
129228f1b45aSHong Zhang         /* elimination */
129328f1b45aSHong Zhang         bjtmp = bj + bi[i];
129428f1b45aSHong Zhang         row   = *bjtmp++;
129528f1b45aSHong Zhang         nzL   = bi[i+1] - bi[i];
129628f1b45aSHong Zhang         for (k=0; k < nzL; k++) {
1297b89f182dSHong Zhang           pc = rtmp1 + row;
129828f1b45aSHong Zhang           if (*pc != 0.0) {
129928f1b45aSHong Zhang             pv   = b->a + bdiag[row];
1300b89f182dSHong Zhang             mul1 = *pc * (*pv);
1301b89f182dSHong Zhang             *pc  = mul1;
130228f1b45aSHong Zhang             pj   = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
130328f1b45aSHong Zhang             pv   = b->a + bdiag[row+1]+1;
130428f1b45aSHong Zhang             nz   = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
1305b89f182dSHong Zhang             for (j=0; j<nz; j++) rtmp1[pj[j]] -= mul1 * pv[j];
130628b1a77fSLisandro Dalcin             ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
130728f1b45aSHong Zhang           }
130828f1b45aSHong Zhang           row = *bjtmp++;
130928f1b45aSHong Zhang         }
131028f1b45aSHong Zhang 
131128f1b45aSHong Zhang         /* finished row so stick it into b->a */
131228f1b45aSHong Zhang         rs = 0.0;
131328f1b45aSHong Zhang         /* L part */
131428f1b45aSHong Zhang         pv = b->a + bi[i];
131528f1b45aSHong Zhang         pj = b->j + bi[i];
131628f1b45aSHong Zhang         nz = bi[i+1] - bi[i];
131728f1b45aSHong Zhang         for (j=0; j<nz; j++) {
1318b89f182dSHong Zhang           pv[j] = rtmp1[pj[j]]; rs += PetscAbsScalar(pv[j]);
131928f1b45aSHong Zhang         }
132028f1b45aSHong Zhang 
132128f1b45aSHong Zhang         /* U part */
132228f1b45aSHong Zhang         pv = b->a + bdiag[i+1]+1;
132328f1b45aSHong Zhang         pj = b->j + bdiag[i+1]+1;
132428f1b45aSHong Zhang         nz = bdiag[i] - bdiag[i+1]-1;
132528f1b45aSHong Zhang         for (j=0; j<nz; j++) {
1326b89f182dSHong Zhang           pv[j] = rtmp1[pj[j]]; rs += PetscAbsScalar(pv[j]);
132728f1b45aSHong Zhang         }
132828f1b45aSHong Zhang 
1329b89f182dSHong Zhang         /* Check zero pivot */
133028f1b45aSHong Zhang         sctx.rs = rs;
1331b89f182dSHong Zhang         sctx.pv = rtmp1[i];
1332d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
133307b50cabSHong Zhang         if (sctx.newshift) break;
133428f1b45aSHong Zhang 
133528f1b45aSHong Zhang         /* Mark diagonal and invert diagonal for simplier triangular solves */
133628f1b45aSHong Zhang         pv  = b->a + bdiag[i];
1337b89f182dSHong Zhang         *pv = 1.0/sctx.pv; /* sctx.pv = rtmp1[i]+shiftamount if shifttype==MAT_SHIFT_INBLOCKS */
133828f1b45aSHong Zhang         break;
133928f1b45aSHong Zhang 
134028f1b45aSHong Zhang       case 2:
134128f1b45aSHong Zhang         /*----------*/
1342b89f182dSHong Zhang         /* zero rtmp1 and rtmp2 */
134328f1b45aSHong Zhang         /* L part */
134428f1b45aSHong Zhang         nz    = bi[i+1] - bi[i];
134528f1b45aSHong Zhang         bjtmp = bj + bi[i];
134628f1b45aSHong Zhang         for  (j=0; j<nz; j++) {
134768785679SHong Zhang           col        = bjtmp[j];
1348b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0;
134928f1b45aSHong Zhang         }
135028f1b45aSHong Zhang 
135128f1b45aSHong Zhang         /* U part */
135228f1b45aSHong Zhang         nz    = bdiag[i]-bdiag[i+1];
135328f1b45aSHong Zhang         bjtmp = bj + bdiag[i+1]+1;
135428f1b45aSHong Zhang         for  (j=0; j<nz; j++) {
135568785679SHong Zhang           col        = bjtmp[j];
1356b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0;
135728f1b45aSHong Zhang         }
135828f1b45aSHong Zhang 
135928f1b45aSHong Zhang         /* load in initial (unfactored row) */
136028f1b45aSHong Zhang         nz    = ai[r[i]+1] - ai[r[i]];
136128f1b45aSHong Zhang         ajtmp = aj + ai[r[i]];
136228f1b45aSHong Zhang         v1    = aa + ai[r[i]]; v2 = aa + ai[r[i]+1];
136328f1b45aSHong Zhang         for (j=0; j<nz; j++) {
136468785679SHong Zhang           col        = ics[ajtmp[j]];
1365b89f182dSHong Zhang           rtmp1[col] = v1[j]; rtmp2[col] = v2[j];
136628f1b45aSHong Zhang         }
136728f1b45aSHong Zhang         /* ZeropivotApply(): shift the diagonal of the matrix  */
1368b89f182dSHong Zhang         rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount;
136928f1b45aSHong Zhang 
137028f1b45aSHong Zhang         /* elimination */
137128f1b45aSHong Zhang         bjtmp = bj + bi[i];
137228f1b45aSHong Zhang         row   = *bjtmp++; /* pivot row */
137328f1b45aSHong Zhang         nzL   = bi[i+1] - bi[i];
137428f1b45aSHong Zhang         for (k=0; k < nzL; k++) {
1375b89f182dSHong Zhang           pc1 = rtmp1 + row;
1376b89f182dSHong Zhang           pc2 = rtmp2 + row;
137728f1b45aSHong Zhang           if (*pc1 != 0.0 || *pc2 != 0.0) {
137828f1b45aSHong Zhang             pv   = b->a + bdiag[row];
1379b89f182dSHong Zhang             mul1 = *pc1*(*pv); mul2 = *pc2*(*pv);
1380b89f182dSHong Zhang             *pc1 = mul1;       *pc2 = mul2;
138128f1b45aSHong Zhang 
138228f1b45aSHong Zhang             pj = b->j + bdiag[row+1]+1;     /* beginning of U(row,:) */
138328f1b45aSHong Zhang             pv = b->a + bdiag[row+1]+1;
138428f1b45aSHong Zhang             nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
138528f1b45aSHong Zhang             for (j=0; j<nz; j++) {
138668785679SHong Zhang               col         = pj[j];
1387b89f182dSHong Zhang               rtmp1[col] -= mul1 * pv[j];
1388b89f182dSHong Zhang               rtmp2[col] -= mul2 * pv[j];
138928f1b45aSHong Zhang             }
139028b1a77fSLisandro Dalcin             ierr = PetscLogFlops(2+4*nz);CHKERRQ(ierr);
139128f1b45aSHong Zhang           }
139228f1b45aSHong Zhang           row = *bjtmp++;
139328f1b45aSHong Zhang         }
139428f1b45aSHong Zhang 
1395b89f182dSHong Zhang         /* finished row i; check zero pivot, then stick row i into b->a */
139628f1b45aSHong Zhang         rs = 0.0;
139728f1b45aSHong Zhang         /* L part */
1398b89f182dSHong Zhang         pc1 = b->a + bi[i];
139928f1b45aSHong Zhang         pj  = b->j + bi[i];
140028f1b45aSHong Zhang         nz  = bi[i+1] - bi[i];
140128f1b45aSHong Zhang         for (j=0; j<nz; j++) {
140268785679SHong Zhang           col    = pj[j];
14030e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
140428f1b45aSHong Zhang         }
140528f1b45aSHong Zhang         /* U part */
1406b89f182dSHong Zhang         pc1 = b->a + bdiag[i+1]+1;
140728f1b45aSHong Zhang         pj  = b->j + bdiag[i+1]+1;
14080e7a5c2bSHong Zhang         nz  = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */
140928f1b45aSHong Zhang         for (j=0; j<nz; j++) {
141068785679SHong Zhang           col    = pj[j];
14110e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
141228f1b45aSHong Zhang         }
141328f1b45aSHong Zhang 
141428f1b45aSHong Zhang         sctx.rs = rs;
1415b89f182dSHong Zhang         sctx.pv = rtmp1[i];
1416d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
141707b50cabSHong Zhang         if (sctx.newshift) break;
1418b89f182dSHong Zhang         pc1  = b->a + bdiag[i]; /* Mark diagonal */
1419b89f182dSHong Zhang         *pc1 = 1.0/sctx.pv;
1420b89f182dSHong Zhang 
1421b89f182dSHong Zhang         /* Now take care of diagonal 2x2 block. */
1422b89f182dSHong Zhang         pc2 = rtmp2 + i;
1423b89f182dSHong Zhang         if (*pc2 != 0.0) {
1424b89f182dSHong Zhang           mul1 = (*pc2)*(*pc1); /* *pc1=diag[i] is inverted! */
1425b89f182dSHong Zhang           *pc2 = mul1;          /* insert L entry */
1426b89f182dSHong Zhang           pj   = b->j + bdiag[i+1]+1;   /* beginning of U(i,:) */
1427b89f182dSHong Zhang           nz   = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */
1428b89f182dSHong Zhang           for (j=0; j<nz; j++) {
1429b89f182dSHong Zhang             col = pj[j]; rtmp2[col] -= mul1 * rtmp1[col];
143028f1b45aSHong Zhang           }
143128b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
1432b89f182dSHong Zhang         }
1433b89f182dSHong Zhang 
1434b89f182dSHong Zhang         /* finished row i+1; check zero pivot, then stick row i+1 into b->a */
1435b89f182dSHong Zhang         rs = 0.0;
1436b89f182dSHong Zhang         /* L part */
1437b89f182dSHong Zhang         pc2 = b->a + bi[i+1];
1438b89f182dSHong Zhang         pj  = b->j + bi[i+1];
1439b89f182dSHong Zhang         nz  = bi[i+2] - bi[i+1];
1440b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1441b89f182dSHong Zhang           col    = pj[j];
14420e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1443b89f182dSHong Zhang         }
1444b89f182dSHong Zhang         /* U part */
1445b89f182dSHong Zhang         pc2 = b->a + bdiag[i+2]+1;
14460e7a5c2bSHong Zhang         pj  = b->j + bdiag[i+2]+1;
14470e7a5c2bSHong Zhang         nz  = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */
1448b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1449b89f182dSHong Zhang           col    = pj[j];
14500e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1451b89f182dSHong Zhang         }
1452b89f182dSHong Zhang 
145328f1b45aSHong Zhang         sctx.rs = rs;
1454b89f182dSHong Zhang         sctx.pv = rtmp2[i+1];
1455d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i+1);CHKERRQ(ierr);
145607b50cabSHong Zhang         if (sctx.newshift) break;
145728f1b45aSHong Zhang         pc2  = b->a + bdiag[i+1];
1458b89f182dSHong Zhang         *pc2 = 1.0/sctx.pv;
145928f1b45aSHong Zhang         break;
1460b89f182dSHong Zhang 
146168785679SHong Zhang       case 3:
146268785679SHong Zhang         /*----------*/
146368785679SHong Zhang         /* zero rtmp */
146468785679SHong Zhang         /* L part */
146568785679SHong Zhang         nz    = bi[i+1] - bi[i];
146668785679SHong Zhang         bjtmp = bj + bi[i];
146768785679SHong Zhang         for  (j=0; j<nz; j++) {
146868785679SHong Zhang           col        = bjtmp[j];
1469b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;
147068785679SHong Zhang         }
147168785679SHong Zhang 
147268785679SHong Zhang         /* U part */
147368785679SHong Zhang         nz    = bdiag[i]-bdiag[i+1];
147468785679SHong Zhang         bjtmp = bj + bdiag[i+1]+1;
147568785679SHong Zhang         for  (j=0; j<nz; j++) {
147668785679SHong Zhang           col        = bjtmp[j];
1477b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;
147868785679SHong Zhang         }
147968785679SHong Zhang 
148068785679SHong Zhang         /* load in initial (unfactored row) */
148168785679SHong Zhang         nz    = ai[r[i]+1] - ai[r[i]];
148268785679SHong Zhang         ajtmp = aj + ai[r[i]];
148368785679SHong Zhang         v1    = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; v3 = aa + ai[r[i]+2];
148468785679SHong Zhang         for (j=0; j<nz; j++) {
148568785679SHong Zhang           col        = ics[ajtmp[j]];
1486b89f182dSHong Zhang           rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; rtmp3[col] = v3[j];
148768785679SHong Zhang         }
148868785679SHong Zhang         /* ZeropivotApply(): shift the diagonal of the matrix  */
1489b89f182dSHong Zhang         rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; rtmp3[i+2] += sctx.shift_amount;
149068785679SHong Zhang 
149168785679SHong Zhang         /* elimination */
149268785679SHong Zhang         bjtmp = bj + bi[i];
149368785679SHong Zhang         row   = *bjtmp++; /* pivot row */
149468785679SHong Zhang         nzL   = bi[i+1] - bi[i];
149568785679SHong Zhang         for (k=0; k < nzL; k++) {
1496b89f182dSHong Zhang           pc1 = rtmp1 + row;
1497b89f182dSHong Zhang           pc2 = rtmp2 + row;
1498b89f182dSHong Zhang           pc3 = rtmp3 + row;
149968785679SHong Zhang           if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 != 0.0) {
150068785679SHong Zhang             pv   = b->a + bdiag[row];
1501b89f182dSHong Zhang             mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); mul3 = *pc3*(*pv);
1502b89f182dSHong Zhang             *pc1 = mul1; *pc2 = mul2; *pc3 = mul3;
150368785679SHong Zhang 
150468785679SHong Zhang             pj = b->j + bdiag[row+1]+1;     /* beginning of U(row,:) */
150568785679SHong Zhang             pv = b->a + bdiag[row+1]+1;
150668785679SHong Zhang             nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
150768785679SHong Zhang             for (j=0; j<nz; j++) {
150868785679SHong Zhang               col         = pj[j];
1509b89f182dSHong Zhang               rtmp1[col] -= mul1 * pv[j];
1510b89f182dSHong Zhang               rtmp2[col] -= mul2 * pv[j];
1511b89f182dSHong Zhang               rtmp3[col] -= mul3 * pv[j];
151268785679SHong Zhang             }
151328b1a77fSLisandro Dalcin             ierr = PetscLogFlops(3+6*nz);CHKERRQ(ierr);
151468785679SHong Zhang           }
151568785679SHong Zhang           row = *bjtmp++;
151668785679SHong Zhang         }
151768785679SHong Zhang 
1518b89f182dSHong Zhang         /* finished row i; check zero pivot, then stick row i into b->a */
1519b89f182dSHong Zhang         rs = 0.0;
1520b89f182dSHong Zhang         /* L part */
1521b89f182dSHong Zhang         pc1 = b->a + bi[i];
1522b89f182dSHong Zhang         pj  = b->j + bi[i];
1523b89f182dSHong Zhang         nz  = bi[i+1] - bi[i];
1524b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1525b89f182dSHong Zhang           col    = pj[j];
15260e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
1527b89f182dSHong Zhang         }
1528b89f182dSHong Zhang         /* U part */
1529b89f182dSHong Zhang         pc1 = b->a + bdiag[i+1]+1;
1530b89f182dSHong Zhang         pj  = b->j + bdiag[i+1]+1;
15310e7a5c2bSHong Zhang         nz  = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */
1532b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1533b89f182dSHong Zhang           col    = pj[j];
15340e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
1535b89f182dSHong Zhang         }
153668785679SHong Zhang 
1537b89f182dSHong Zhang         sctx.rs = rs;
1538b89f182dSHong Zhang         sctx.pv = rtmp1[i];
1539d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
154007b50cabSHong Zhang         if (sctx.newshift) break;
1541b89f182dSHong Zhang         pc1  = b->a + bdiag[i]; /* Mark diag[i] */
1542b89f182dSHong Zhang         *pc1 = 1.0/sctx.pv;
1543b89f182dSHong Zhang 
1544b89f182dSHong Zhang         /* Now take care of 1st column of diagonal 3x3 block. */
1545b89f182dSHong Zhang         pc2 = rtmp2 + i;
1546b89f182dSHong Zhang         pc3 = rtmp3 + i;
1547b89f182dSHong Zhang         if (*pc2 != 0.0 || *pc3 != 0.0) {
1548b89f182dSHong Zhang           mul2 = (*pc2)*(*pc1); *pc2 = mul2;
1549b89f182dSHong Zhang           mul3 = (*pc3)*(*pc1); *pc3 = mul3;
155068785679SHong Zhang           pj   = b->j + bdiag[i+1]+1;   /* beginning of U(i,:) */
155168785679SHong Zhang           nz   = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */
155268785679SHong Zhang           for (j=0; j<nz; j++) {
155368785679SHong Zhang             col         = pj[j];
1554b89f182dSHong Zhang             rtmp2[col] -= mul2 * rtmp1[col];
1555b89f182dSHong Zhang             rtmp3[col] -= mul3 * rtmp1[col];
155668785679SHong Zhang           }
155728b1a77fSLisandro Dalcin           ierr = PetscLogFlops(2+4*nz);CHKERRQ(ierr);
155868785679SHong Zhang         }
155968785679SHong Zhang 
1560b89f182dSHong Zhang         /* finished row i+1; check zero pivot, then stick row i+1 into b->a */
1561b89f182dSHong Zhang         rs = 0.0;
1562b89f182dSHong Zhang         /* L part */
1563b89f182dSHong Zhang         pc2 = b->a + bi[i+1];
1564b89f182dSHong Zhang         pj  = b->j + bi[i+1];
1565b89f182dSHong Zhang         nz  = bi[i+2] - bi[i+1];
1566b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1567b89f182dSHong Zhang           col    = pj[j];
15680e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1569b89f182dSHong Zhang         }
1570b89f182dSHong Zhang         /* U part */
1571b89f182dSHong Zhang         pc2 = b->a + bdiag[i+2]+1;
15720e7a5c2bSHong Zhang         pj  = b->j + bdiag[i+2]+1;
15730e7a5c2bSHong Zhang         nz  = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */
1574b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1575b89f182dSHong Zhang           col    = pj[j];
15760e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1577b89f182dSHong Zhang         }
1578b89f182dSHong Zhang 
1579b89f182dSHong Zhang         sctx.rs = rs;
1580b89f182dSHong Zhang         sctx.pv = rtmp2[i+1];
1581d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i+1);CHKERRQ(ierr);
158207b50cabSHong Zhang         if (sctx.newshift) break;
1583b89f182dSHong Zhang         pc2  = b->a + bdiag[i+1];
1584b89f182dSHong Zhang         *pc2 = 1.0/sctx.pv; /* Mark diag[i+1] */
1585b89f182dSHong Zhang 
1586b89f182dSHong Zhang         /* Now take care of 2nd column of diagonal 3x3 block. */
1587b89f182dSHong Zhang         pc3 = rtmp3 + i+1;
158868785679SHong Zhang         if (*pc3 != 0.0) {
1589b89f182dSHong Zhang           mul3 = (*pc3)*(*pc2); *pc3 = mul3;
159068785679SHong Zhang           pj   = b->j + bdiag[i+2]+1;     /* beginning of U(i+1,:) */
159168785679SHong Zhang           nz   = bdiag[i+1]-bdiag[i+2]-1; /* num of entries in U(i+1,:) excluding diag */
159268785679SHong Zhang           for (j=0; j<nz; j++) {
159368785679SHong Zhang             col         = pj[j];
1594b89f182dSHong Zhang             rtmp3[col] -= mul3 * rtmp2[col];
159568785679SHong Zhang           }
159628b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
159768785679SHong Zhang         }
159868785679SHong Zhang 
1599b89f182dSHong Zhang         /* finished i+2; check zero pivot, then stick row i+2 into b->a */
160068785679SHong Zhang         rs = 0.0;
160168785679SHong Zhang         /* L part */
1602b89f182dSHong Zhang         pc3 = b->a + bi[i+2];
1603b89f182dSHong Zhang         pj  = b->j + bi[i+2];
1604b89f182dSHong Zhang         nz  = bi[i+3] - bi[i+2];
160568785679SHong Zhang         for (j=0; j<nz; j++) {
160668785679SHong Zhang           col    = pj[j];
16070e7a5c2bSHong Zhang           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
160868785679SHong Zhang         }
160968785679SHong Zhang         /* U part */
1610b89f182dSHong Zhang         pc3 = b->a + bdiag[i+3]+1;
16110e7a5c2bSHong Zhang         pj  = b->j + bdiag[i+3]+1;
16120e7a5c2bSHong Zhang         nz  = bdiag[i+2] - bdiag[i+3] - 1; /* exclude diagonal */
161368785679SHong Zhang         for (j=0; j<nz; j++) {
161468785679SHong Zhang           col    = pj[j];
16150e7a5c2bSHong Zhang           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
161668785679SHong Zhang         }
161768785679SHong Zhang 
161868785679SHong Zhang         sctx.rs = rs;
1619b89f182dSHong Zhang         sctx.pv = rtmp3[i+2];
1620d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i+2);CHKERRQ(ierr);
162107b50cabSHong Zhang         if (sctx.newshift) break;
162268785679SHong Zhang         pc3  = b->a + bdiag[i+2];
1623b89f182dSHong Zhang         *pc3 = 1.0/sctx.pv; /* Mark diag[i+2] */
162468785679SHong Zhang         break;
16259877982aSShri Abhyankar       case 4:
16269877982aSShri Abhyankar         /*----------*/
16279877982aSShri Abhyankar         /* zero rtmp */
16289877982aSShri Abhyankar         /* L part */
16299877982aSShri Abhyankar         nz    = bi[i+1] - bi[i];
16309877982aSShri Abhyankar         bjtmp = bj + bi[i];
16319877982aSShri Abhyankar         for  (j=0; j<nz; j++) {
16329877982aSShri Abhyankar           col        = bjtmp[j];
16339877982aSShri Abhyankar           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;rtmp4[col] = 0.0;
16349877982aSShri Abhyankar         }
16359877982aSShri Abhyankar 
16369877982aSShri Abhyankar         /* U part */
16379877982aSShri Abhyankar         nz    = bdiag[i]-bdiag[i+1];
16389877982aSShri Abhyankar         bjtmp = bj + bdiag[i+1]+1;
16399877982aSShri Abhyankar         for  (j=0; j<nz; j++) {
16409877982aSShri Abhyankar           col        = bjtmp[j];
16419877982aSShri Abhyankar           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0; rtmp4[col] = 0.0;
16429877982aSShri Abhyankar         }
16439877982aSShri Abhyankar 
16449877982aSShri Abhyankar         /* load in initial (unfactored row) */
16459877982aSShri Abhyankar         nz    = ai[r[i]+1] - ai[r[i]];
16469877982aSShri Abhyankar         ajtmp = aj + ai[r[i]];
16479877982aSShri Abhyankar         v1    = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; v3 = aa + ai[r[i]+2]; v4 = aa + ai[r[i]+3];
16489877982aSShri Abhyankar         for (j=0; j<nz; j++) {
16499877982aSShri Abhyankar           col        = ics[ajtmp[j]];
16509877982aSShri Abhyankar           rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; rtmp3[col] = v3[j]; rtmp4[col] = v4[j];
16519877982aSShri Abhyankar         }
16529877982aSShri Abhyankar         /* ZeropivotApply(): shift the diagonal of the matrix  */
16539877982aSShri Abhyankar         rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; rtmp3[i+2] += sctx.shift_amount; rtmp4[i+3] += sctx.shift_amount;
16549877982aSShri Abhyankar 
16559877982aSShri Abhyankar         /* elimination */
16569877982aSShri Abhyankar         bjtmp = bj + bi[i];
16579877982aSShri Abhyankar         row   = *bjtmp++; /* pivot row */
16589877982aSShri Abhyankar         nzL   = bi[i+1] - bi[i];
16599877982aSShri Abhyankar         for (k=0; k < nzL; k++) {
16609877982aSShri Abhyankar           pc1 = rtmp1 + row;
16619877982aSShri Abhyankar           pc2 = rtmp2 + row;
16629877982aSShri Abhyankar           pc3 = rtmp3 + row;
16639877982aSShri Abhyankar           pc4 = rtmp4 + row;
16649877982aSShri Abhyankar           if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 != 0.0 || *pc4 != 0.0) {
16659877982aSShri Abhyankar             pv   = b->a + bdiag[row];
16669877982aSShri Abhyankar             mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); mul3 = *pc3*(*pv); mul4 = *pc4*(*pv);
16679877982aSShri Abhyankar             *pc1 = mul1; *pc2 = mul2; *pc3 = mul3; *pc4 = mul4;
16689877982aSShri Abhyankar 
16699877982aSShri Abhyankar             pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
16709877982aSShri Abhyankar             pv = b->a + bdiag[row+1]+1;
16719877982aSShri Abhyankar             nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
16729877982aSShri Abhyankar             for (j=0; j<nz; j++) {
16739877982aSShri Abhyankar               col         = pj[j];
16749877982aSShri Abhyankar               rtmp1[col] -= mul1 * pv[j];
16759877982aSShri Abhyankar               rtmp2[col] -= mul2 * pv[j];
16769877982aSShri Abhyankar               rtmp3[col] -= mul3 * pv[j];
16779877982aSShri Abhyankar               rtmp4[col] -= mul4 * pv[j];
16789877982aSShri Abhyankar             }
167928b1a77fSLisandro Dalcin             ierr = PetscLogFlops(4+8*nz);CHKERRQ(ierr);
16809877982aSShri Abhyankar           }
16819877982aSShri Abhyankar           row = *bjtmp++;
16829877982aSShri Abhyankar         }
16839877982aSShri Abhyankar 
16849877982aSShri Abhyankar         /* finished row i; check zero pivot, then stick row i into b->a */
16859877982aSShri Abhyankar         rs = 0.0;
16869877982aSShri Abhyankar         /* L part */
16879877982aSShri Abhyankar         pc1 = b->a + bi[i];
16889877982aSShri Abhyankar         pj  = b->j + bi[i];
16899877982aSShri Abhyankar         nz  = bi[i+1] - bi[i];
16909877982aSShri Abhyankar         for (j=0; j<nz; j++) {
16919877982aSShri Abhyankar           col    = pj[j];
16929877982aSShri Abhyankar           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
16939877982aSShri Abhyankar         }
16949877982aSShri Abhyankar         /* U part */
16959877982aSShri Abhyankar         pc1 = b->a + bdiag[i+1]+1;
16969877982aSShri Abhyankar         pj  = b->j + bdiag[i+1]+1;
16979877982aSShri Abhyankar         nz  = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */
16989877982aSShri Abhyankar         for (j=0; j<nz; j++) {
16999877982aSShri Abhyankar           col    = pj[j];
17009877982aSShri Abhyankar           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
17019877982aSShri Abhyankar         }
17029877982aSShri Abhyankar 
17039877982aSShri Abhyankar         sctx.rs = rs;
17049877982aSShri Abhyankar         sctx.pv = rtmp1[i];
1705d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
170607b50cabSHong Zhang         if (sctx.newshift) break;
17079877982aSShri Abhyankar         pc1  = b->a + bdiag[i]; /* Mark diag[i] */
17089877982aSShri Abhyankar         *pc1 = 1.0/sctx.pv;
17099877982aSShri Abhyankar 
17109877982aSShri Abhyankar         /* Now take care of 1st column of diagonal 4x4 block. */
17119877982aSShri Abhyankar         pc2 = rtmp2 + i;
17129877982aSShri Abhyankar         pc3 = rtmp3 + i;
17139877982aSShri Abhyankar         pc4 = rtmp4 + i;
17149877982aSShri Abhyankar         if (*pc2 != 0.0 || *pc3 != 0.0 || *pc4 != 0.0) {
17159877982aSShri Abhyankar           mul2 = (*pc2)*(*pc1); *pc2 = mul2;
17169877982aSShri Abhyankar           mul3 = (*pc3)*(*pc1); *pc3 = mul3;
17179877982aSShri Abhyankar           mul4 = (*pc4)*(*pc1); *pc4 = mul4;
17189877982aSShri Abhyankar           pj   = b->j + bdiag[i+1]+1;   /* beginning of U(i,:) */
17199877982aSShri Abhyankar           nz   = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */
17209877982aSShri Abhyankar           for (j=0; j<nz; j++) {
17219877982aSShri Abhyankar             col         = pj[j];
17229877982aSShri Abhyankar             rtmp2[col] -= mul2 * rtmp1[col];
17239877982aSShri Abhyankar             rtmp3[col] -= mul3 * rtmp1[col];
17249877982aSShri Abhyankar             rtmp4[col] -= mul4 * rtmp1[col];
17259877982aSShri Abhyankar           }
172628b1a77fSLisandro Dalcin           ierr = PetscLogFlops(3+6*nz);CHKERRQ(ierr);
17279877982aSShri Abhyankar         }
17289877982aSShri Abhyankar 
17299877982aSShri Abhyankar         /* finished row i+1; check zero pivot, then stick row i+1 into b->a */
17309877982aSShri Abhyankar         rs = 0.0;
17319877982aSShri Abhyankar         /* L part */
17329877982aSShri Abhyankar         pc2 = b->a + bi[i+1];
17339877982aSShri Abhyankar         pj  = b->j + bi[i+1];
17349877982aSShri Abhyankar         nz  = bi[i+2] - bi[i+1];
17359877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17369877982aSShri Abhyankar           col    = pj[j];
17379877982aSShri Abhyankar           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
17389877982aSShri Abhyankar         }
17399877982aSShri Abhyankar         /* U part */
17409877982aSShri Abhyankar         pc2 = b->a + bdiag[i+2]+1;
17419877982aSShri Abhyankar         pj  = b->j + bdiag[i+2]+1;
17429877982aSShri Abhyankar         nz  = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */
17439877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17449877982aSShri Abhyankar           col    = pj[j];
17459877982aSShri Abhyankar           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
17469877982aSShri Abhyankar         }
17479877982aSShri Abhyankar 
17489877982aSShri Abhyankar         sctx.rs = rs;
17499877982aSShri Abhyankar         sctx.pv = rtmp2[i+1];
1750d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i+1);CHKERRQ(ierr);
175107b50cabSHong Zhang         if (sctx.newshift) break;
17529877982aSShri Abhyankar         pc2  = b->a + bdiag[i+1];
17539877982aSShri Abhyankar         *pc2 = 1.0/sctx.pv; /* Mark diag[i+1] */
17549877982aSShri Abhyankar 
17559877982aSShri Abhyankar         /* Now take care of 2nd column of diagonal 4x4 block. */
17569877982aSShri Abhyankar         pc3 = rtmp3 + i+1;
17579877982aSShri Abhyankar         pc4 = rtmp4 + i+1;
17589877982aSShri Abhyankar         if (*pc3 != 0.0 || *pc4 != 0.0) {
17599877982aSShri Abhyankar           mul3 = (*pc3)*(*pc2); *pc3 = mul3;
17609877982aSShri Abhyankar           mul4 = (*pc4)*(*pc2); *pc4 = mul4;
17619877982aSShri Abhyankar           pj   = b->j + bdiag[i+2]+1;     /* beginning of U(i+1,:) */
17629877982aSShri Abhyankar           nz   = bdiag[i+1]-bdiag[i+2]-1; /* num of entries in U(i+1,:) excluding diag */
17639877982aSShri Abhyankar           for (j=0; j<nz; j++) {
17649877982aSShri Abhyankar             col         = pj[j];
17659877982aSShri Abhyankar             rtmp3[col] -= mul3 * rtmp2[col];
17669877982aSShri Abhyankar             rtmp4[col] -= mul4 * rtmp2[col];
17679877982aSShri Abhyankar           }
17689877982aSShri Abhyankar           ierr = PetscLogFlops(4*nz);CHKERRQ(ierr);
17699877982aSShri Abhyankar         }
17709877982aSShri Abhyankar 
17719877982aSShri Abhyankar         /* finished i+2; check zero pivot, then stick row i+2 into b->a */
17729877982aSShri Abhyankar         rs = 0.0;
17739877982aSShri Abhyankar         /* L part */
17749877982aSShri Abhyankar         pc3 = b->a + bi[i+2];
17759877982aSShri Abhyankar         pj  = b->j + bi[i+2];
17769877982aSShri Abhyankar         nz  = bi[i+3] - bi[i+2];
17779877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17789877982aSShri Abhyankar           col    = pj[j];
17799877982aSShri Abhyankar           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
17809877982aSShri Abhyankar         }
17819877982aSShri Abhyankar         /* U part */
17829877982aSShri Abhyankar         pc3 = b->a + bdiag[i+3]+1;
17839877982aSShri Abhyankar         pj  = b->j + bdiag[i+3]+1;
17849877982aSShri Abhyankar         nz  = bdiag[i+2] - bdiag[i+3] - 1; /* exclude diagonal */
17859877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17869877982aSShri Abhyankar           col    = pj[j];
17879877982aSShri Abhyankar           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
17889877982aSShri Abhyankar         }
17899877982aSShri Abhyankar 
17909877982aSShri Abhyankar         sctx.rs = rs;
17919877982aSShri Abhyankar         sctx.pv = rtmp3[i+2];
1792d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i+2);CHKERRQ(ierr);
179307b50cabSHong Zhang         if (sctx.newshift) break;
17949877982aSShri Abhyankar         pc3  = b->a + bdiag[i+2];
17959877982aSShri Abhyankar         *pc3 = 1.0/sctx.pv; /* Mark diag[i+2] */
17969877982aSShri Abhyankar 
17979877982aSShri Abhyankar         /* Now take care of 3rd column of diagonal 4x4 block. */
17989877982aSShri Abhyankar         pc4 = rtmp4 + i+2;
17999877982aSShri Abhyankar         if (*pc4 != 0.0) {
18009877982aSShri Abhyankar           mul4 = (*pc4)*(*pc3); *pc4 = mul4;
18019877982aSShri Abhyankar           pj   = b->j + bdiag[i+3]+1;     /* beginning of U(i+2,:) */
18029877982aSShri Abhyankar           nz   = bdiag[i+2]-bdiag[i+3]-1; /* num of entries in U(i+2,:) excluding diag */
18039877982aSShri Abhyankar           for (j=0; j<nz; j++) {
18049877982aSShri Abhyankar             col         = pj[j];
18059877982aSShri Abhyankar             rtmp4[col] -= mul4 * rtmp3[col];
18069877982aSShri Abhyankar           }
180728b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
18089877982aSShri Abhyankar         }
18099877982aSShri Abhyankar 
18109877982aSShri Abhyankar         /* finished i+3; check zero pivot, then stick row i+3 into b->a */
18119877982aSShri Abhyankar         rs = 0.0;
18129877982aSShri Abhyankar         /* L part */
18139877982aSShri Abhyankar         pc4 = b->a + bi[i+3];
18149877982aSShri Abhyankar         pj  = b->j + bi[i+3];
18159877982aSShri Abhyankar         nz  = bi[i+4] - bi[i+3];
18169877982aSShri Abhyankar         for (j=0; j<nz; j++) {
18179877982aSShri Abhyankar           col    = pj[j];
18189877982aSShri Abhyankar           pc4[j] = rtmp4[col]; rs += PetscAbsScalar(pc4[j]);
18199877982aSShri Abhyankar         }
18209877982aSShri Abhyankar         /* U part */
18219877982aSShri Abhyankar         pc4 = b->a + bdiag[i+4]+1;
18229877982aSShri Abhyankar         pj  = b->j + bdiag[i+4]+1;
18239877982aSShri Abhyankar         nz  = bdiag[i+3] - bdiag[i+4] - 1; /* exclude diagonal */
18249877982aSShri Abhyankar         for (j=0; j<nz; j++) {
18259877982aSShri Abhyankar           col    = pj[j];
18269877982aSShri Abhyankar           pc4[j] = rtmp4[col]; rs += PetscAbsScalar(pc4[j]);
18279877982aSShri Abhyankar         }
18289877982aSShri Abhyankar 
18299877982aSShri Abhyankar         sctx.rs = rs;
18309877982aSShri Abhyankar         sctx.pv = rtmp4[i+3];
1831d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i+3);CHKERRQ(ierr);
183207b50cabSHong Zhang         if (sctx.newshift) break;
18339877982aSShri Abhyankar         pc4  = b->a + bdiag[i+3];
18349877982aSShri Abhyankar         *pc4 = 1.0/sctx.pv; /* Mark diag[i+3] */
18359877982aSShri Abhyankar         break;
183668785679SHong Zhang 
183728f1b45aSHong Zhang       default:
1838e32f2f54SBarry Smith         SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Node size not yet supported \n");
183928f1b45aSHong Zhang       }
1840c2b86aeeSHong Zhang       if (sctx.newshift) break; /* break for (inod=0,i=0; inod<node_max; inod++) */
184128f1b45aSHong Zhang       i += nodesz;                 /* Update the row */
184268785679SHong Zhang     }
184328f1b45aSHong Zhang 
184428f1b45aSHong Zhang     /* MatPivotRefine() */
184507b50cabSHong Zhang     if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
184628f1b45aSHong Zhang       /*
184728f1b45aSHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
184828f1b45aSHong Zhang        * then try lower shift
184928f1b45aSHong Zhang        */
185028f1b45aSHong Zhang       sctx.shift_hi       = sctx.shift_fraction;
185128f1b45aSHong Zhang       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
185228f1b45aSHong Zhang       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
185307b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
185428f1b45aSHong Zhang       sctx.nshift++;
185528f1b45aSHong Zhang     }
185607b50cabSHong Zhang   } while (sctx.newshift);
185728f1b45aSHong Zhang 
1858eb9baa12SBarry Smith   ierr = PetscFree4(rtmp1,rtmp2,rtmp3,rtmp4);CHKERRQ(ierr);
185968785679SHong Zhang   ierr = PetscFree(tmp_vec2);CHKERRQ(ierr);
186028f1b45aSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
186128f1b45aSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
186228f1b45aSHong Zhang 
1863abb87a52SBarry Smith   if (b->inode.size) {
1864abb87a52SBarry Smith     C->ops->solve           = MatSolve_SeqAIJ_Inode;
1865abb87a52SBarry Smith   } else {
1866d3ac4fa3SBarry Smith     C->ops->solve           = MatSolve_SeqAIJ;
1867abb87a52SBarry Smith   }
186828f1b45aSHong Zhang   C->ops->solveadd          = MatSolveAdd_SeqAIJ;
186928f1b45aSHong Zhang   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ;
187028f1b45aSHong Zhang   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ;
187128f1b45aSHong Zhang   C->ops->matsolve          = MatMatSolve_SeqAIJ;
187228f1b45aSHong Zhang   C->assembled              = PETSC_TRUE;
187328f1b45aSHong Zhang   C->preallocated           = PETSC_TRUE;
18742205254eSKarl Rupp 
187528f1b45aSHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
187628f1b45aSHong Zhang 
187728f1b45aSHong Zhang   /* MatShiftView(A,info,&sctx) */
187828f1b45aSHong Zhang   if (sctx.nshift) {
1879f4db908eSBarry Smith     if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) {
188057622a8eSBarry 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);
1881f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
188257622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
1883f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) {
188457622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %g\n",sctx.nshift,(double)info->shiftamount);CHKERRQ(ierr);
188528f1b45aSHong Zhang     }
188628f1b45aSHong Zhang   }
188728f1b45aSHong Zhang   PetscFunctionReturn(0);
188828f1b45aSHong Zhang }
1889628f99d7SShri Abhyankar 
1890628f99d7SShri Abhyankar #undef __FUNCT__
1891628f99d7SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_Inode_inplace"
1892*9aa7eafdSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_Inode_inplace(Mat B,Mat A,MatFactorInfo *info)
1893628f99d7SShri Abhyankar {
1894628f99d7SShri Abhyankar   Mat             C     = B;
1895628f99d7SShri Abhyankar   Mat_SeqAIJ      *a    = (Mat_SeqAIJ*)A->data,*b = (Mat_SeqAIJ*)C->data;
1896628f99d7SShri Abhyankar   IS              iscol = b->col,isrow = b->row,isicol = b->icol;
1897628f99d7SShri Abhyankar   PetscErrorCode  ierr;
1898628f99d7SShri Abhyankar   const PetscInt  *r,*ic,*c,*ics;
1899628f99d7SShri Abhyankar   PetscInt        n   = A->rmap->n,*bi = b->i;
1900628f99d7SShri Abhyankar   PetscInt        *bj = b->j,*nbj=b->j +1,*ajtmp,*bjtmp,nz,nz_tmp,row,prow;
19018758e1faSBarry Smith   PetscInt        i,j,idx,*bd = b->diag,node_max,nodesz;
19028758e1faSBarry Smith   PetscInt        *ai = a->i,*aj = a->j;
1903628f99d7SShri Abhyankar   PetscInt        *ns,*tmp_vec1,*tmp_vec2,*nsmap,*pj;
1904628f99d7SShri Abhyankar   PetscScalar     mul1,mul2,mul3,tmp;
1905628f99d7SShri Abhyankar   MatScalar       *pc1,*pc2,*pc3,*ba = b->a,*pv,*rtmp11,*rtmp22,*rtmp33;
1906628f99d7SShri Abhyankar   const MatScalar *v1,*v2,*v3,*aa = a->a,*rtmp1;
1907628f99d7SShri Abhyankar   PetscReal       rs=0.0;
1908628f99d7SShri Abhyankar   FactorShiftCtx  sctx;
1909628f99d7SShri Abhyankar 
1910628f99d7SShri Abhyankar   PetscFunctionBegin;
1911628f99d7SShri Abhyankar   sctx.shift_top      = 0;
1912628f99d7SShri Abhyankar   sctx.nshift_max     = 0;
1913628f99d7SShri Abhyankar   sctx.shift_lo       = 0;
1914628f99d7SShri Abhyankar   sctx.shift_hi       = 0;
1915628f99d7SShri Abhyankar   sctx.shift_fraction = 0;
1916628f99d7SShri Abhyankar 
1917628f99d7SShri Abhyankar   /* if both shift schemes are chosen by user, only use info->shiftpd */
1918f4db908eSBarry Smith   if (info->shifttype==(PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
1919628f99d7SShri Abhyankar     sctx.shift_top = 0;
1920628f99d7SShri Abhyankar     for (i=0; i<n; i++) {
1921628f99d7SShri Abhyankar       /* calculate rs = sum(|aij|)-RealPart(aii), amt of shift needed for this row */
1922628f99d7SShri Abhyankar       rs    = 0.0;
1923628f99d7SShri Abhyankar       ajtmp = aj + ai[i];
1924628f99d7SShri Abhyankar       rtmp1 = aa + ai[i];
1925628f99d7SShri Abhyankar       nz    = ai[i+1] - ai[i];
1926628f99d7SShri Abhyankar       for (j=0; j<nz; j++) {
1927628f99d7SShri Abhyankar         if (*ajtmp != i) {
1928628f99d7SShri Abhyankar           rs += PetscAbsScalar(*rtmp1++);
1929628f99d7SShri Abhyankar         } else {
1930628f99d7SShri Abhyankar           rs -= PetscRealPart(*rtmp1++);
1931628f99d7SShri Abhyankar         }
1932628f99d7SShri Abhyankar         ajtmp++;
1933628f99d7SShri Abhyankar       }
1934628f99d7SShri Abhyankar       if (rs>sctx.shift_top) sctx.shift_top = rs;
1935628f99d7SShri Abhyankar     }
1936628f99d7SShri Abhyankar     if (sctx.shift_top == 0.0) sctx.shift_top += 1.e-12;
1937628f99d7SShri Abhyankar     sctx.shift_top *= 1.1;
1938628f99d7SShri Abhyankar     sctx.nshift_max = 5;
1939628f99d7SShri Abhyankar     sctx.shift_lo   = 0.;
1940628f99d7SShri Abhyankar     sctx.shift_hi   = 1.;
1941628f99d7SShri Abhyankar   }
1942628f99d7SShri Abhyankar   sctx.shift_amount = 0;
1943628f99d7SShri Abhyankar   sctx.nshift       = 0;
1944628f99d7SShri Abhyankar 
1945628f99d7SShri Abhyankar   ierr   = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1946628f99d7SShri Abhyankar   ierr   = ISGetIndices(iscol,&c);CHKERRQ(ierr);
1947628f99d7SShri Abhyankar   ierr   = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
19481795a4d1SJed Brown   ierr   = PetscCalloc3(n,&rtmp11,n,&rtmp22,n,&rtmp33);CHKERRQ(ierr);
1949628f99d7SShri Abhyankar   ics    = ic;
1950628f99d7SShri Abhyankar 
1951628f99d7SShri Abhyankar   node_max = a->inode.node_count;
1952628f99d7SShri Abhyankar   ns       = a->inode.size;
1953e7e72b3dSBarry Smith   if (!ns) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix without inode information");
1954628f99d7SShri Abhyankar 
1955628f99d7SShri Abhyankar   /* If max inode size > 3, split it into two inodes.*/
1956628f99d7SShri Abhyankar   /* also map the inode sizes according to the ordering */
1957854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&tmp_vec1);CHKERRQ(ierr);
1958628f99d7SShri Abhyankar   for (i=0,j=0; i<node_max; ++i,++j) {
1959628f99d7SShri Abhyankar     if (ns[i]>3) {
1960628f99d7SShri Abhyankar       tmp_vec1[j] = ns[i]/2; /* Assuming ns[i] < =5  */
1961628f99d7SShri Abhyankar       ++j;
1962628f99d7SShri Abhyankar       tmp_vec1[j] = ns[i] - tmp_vec1[j-1];
1963628f99d7SShri Abhyankar     } else {
1964628f99d7SShri Abhyankar       tmp_vec1[j] = ns[i];
1965628f99d7SShri Abhyankar     }
1966628f99d7SShri Abhyankar   }
1967628f99d7SShri Abhyankar   /* Use the correct node_max */
1968628f99d7SShri Abhyankar   node_max = j;
1969628f99d7SShri Abhyankar 
1970628f99d7SShri Abhyankar   /* Now reorder the inode info based on mat re-ordering info */
1971628f99d7SShri Abhyankar   /* First create a row -> inode_size_array_index map */
1972854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&nsmap);CHKERRQ(ierr);
1973854ce69bSBarry Smith   ierr = PetscMalloc1(node_max+1,&tmp_vec2);CHKERRQ(ierr);
1974628f99d7SShri Abhyankar   for (i=0,row=0; i<node_max; i++) {
1975628f99d7SShri Abhyankar     nodesz = tmp_vec1[i];
1976628f99d7SShri Abhyankar     for (j=0; j<nodesz; j++,row++) {
1977628f99d7SShri Abhyankar       nsmap[row] = i;
1978628f99d7SShri Abhyankar     }
1979628f99d7SShri Abhyankar   }
1980628f99d7SShri Abhyankar   /* Using nsmap, create a reordered ns structure */
1981628f99d7SShri Abhyankar   for (i=0,j=0; i< node_max; i++) {
1982628f99d7SShri Abhyankar     nodesz      = tmp_vec1[nsmap[r[j]]];     /* here the reordered row_no is in r[] */
1983628f99d7SShri Abhyankar     tmp_vec2[i] = nodesz;
1984628f99d7SShri Abhyankar     j          += nodesz;
1985628f99d7SShri Abhyankar   }
1986628f99d7SShri Abhyankar   ierr = PetscFree(nsmap);CHKERRQ(ierr);
1987628f99d7SShri Abhyankar   ierr = PetscFree(tmp_vec1);CHKERRQ(ierr);
1988628f99d7SShri Abhyankar   /* Now use the correct ns */
1989628f99d7SShri Abhyankar   ns = tmp_vec2;
1990628f99d7SShri Abhyankar 
1991628f99d7SShri Abhyankar   do {
199207b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
1993628f99d7SShri Abhyankar     /* Now loop over each block-row, and do the factorization */
1994628f99d7SShri Abhyankar     for (i=0,row=0; i<node_max; i++) {
1995628f99d7SShri Abhyankar       nodesz = ns[i];
1996628f99d7SShri Abhyankar       nz     = bi[row+1] - bi[row];
1997628f99d7SShri Abhyankar       bjtmp  = bj + bi[row];
1998628f99d7SShri Abhyankar 
1999628f99d7SShri Abhyankar       switch (nodesz) {
2000628f99d7SShri Abhyankar       case 1:
2001628f99d7SShri Abhyankar         for  (j=0; j<nz; j++) {
2002628f99d7SShri Abhyankar           idx         = bjtmp[j];
2003628f99d7SShri Abhyankar           rtmp11[idx] = 0.0;
2004628f99d7SShri Abhyankar         }
2005628f99d7SShri Abhyankar 
2006628f99d7SShri Abhyankar         /* load in initial (unfactored row) */
2007628f99d7SShri Abhyankar         idx    = r[row];
2008628f99d7SShri Abhyankar         nz_tmp = ai[idx+1] - ai[idx];
2009628f99d7SShri Abhyankar         ajtmp  = aj + ai[idx];
2010628f99d7SShri Abhyankar         v1     = aa + ai[idx];
2011628f99d7SShri Abhyankar 
2012628f99d7SShri Abhyankar         for (j=0; j<nz_tmp; j++) {
2013628f99d7SShri Abhyankar           idx         = ics[ajtmp[j]];
2014628f99d7SShri Abhyankar           rtmp11[idx] = v1[j];
2015628f99d7SShri Abhyankar         }
2016628f99d7SShri Abhyankar         rtmp11[ics[r[row]]] += sctx.shift_amount;
2017628f99d7SShri Abhyankar 
2018628f99d7SShri Abhyankar         prow = *bjtmp++;
2019628f99d7SShri Abhyankar         while (prow < row) {
2020628f99d7SShri Abhyankar           pc1 = rtmp11 + prow;
2021628f99d7SShri Abhyankar           if (*pc1 != 0.0) {
2022628f99d7SShri Abhyankar             pv     = ba + bd[prow];
2023628f99d7SShri Abhyankar             pj     = nbj + bd[prow];
2024628f99d7SShri Abhyankar             mul1   = *pc1 * *pv++;
2025628f99d7SShri Abhyankar             *pc1   = mul1;
2026628f99d7SShri Abhyankar             nz_tmp = bi[prow+1] - bd[prow] - 1;
202728b1a77fSLisandro Dalcin             ierr   = PetscLogFlops(1+2*nz_tmp);CHKERRQ(ierr);
2028628f99d7SShri Abhyankar             for (j=0; j<nz_tmp; j++) {
2029628f99d7SShri Abhyankar               tmp          = pv[j];
2030628f99d7SShri Abhyankar               idx          = pj[j];
2031628f99d7SShri Abhyankar               rtmp11[idx] -= mul1 * tmp;
2032628f99d7SShri Abhyankar             }
2033628f99d7SShri Abhyankar           }
2034628f99d7SShri Abhyankar           prow = *bjtmp++;
2035628f99d7SShri Abhyankar         }
2036628f99d7SShri Abhyankar         pj  = bj + bi[row];
2037628f99d7SShri Abhyankar         pc1 = ba + bi[row];
2038628f99d7SShri Abhyankar 
2039628f99d7SShri Abhyankar         sctx.pv     = rtmp11[row];
2040628f99d7SShri Abhyankar         rtmp11[row] = 1.0/rtmp11[row]; /* invert diag */
2041628f99d7SShri Abhyankar         rs          = 0.0;
2042628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2043628f99d7SShri Abhyankar           idx    = pj[j];
2044628f99d7SShri Abhyankar           pc1[j] = rtmp11[idx]; /* rtmp11 -> ba */
2045628f99d7SShri Abhyankar           if (idx != row) rs += PetscAbsScalar(pc1[j]);
2046628f99d7SShri Abhyankar         }
2047628f99d7SShri Abhyankar         sctx.rs = rs;
2048d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,row);CHKERRQ(ierr);
204907b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2050628f99d7SShri Abhyankar         break;
2051628f99d7SShri Abhyankar 
2052628f99d7SShri Abhyankar       case 2:
2053628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2054628f99d7SShri Abhyankar           idx         = bjtmp[j];
2055628f99d7SShri Abhyankar           rtmp11[idx] = 0.0;
2056628f99d7SShri Abhyankar           rtmp22[idx] = 0.0;
2057628f99d7SShri Abhyankar         }
2058628f99d7SShri Abhyankar 
2059628f99d7SShri Abhyankar         /* load in initial (unfactored row) */
2060628f99d7SShri Abhyankar         idx    = r[row];
2061628f99d7SShri Abhyankar         nz_tmp = ai[idx+1] - ai[idx];
2062628f99d7SShri Abhyankar         ajtmp  = aj + ai[idx];
2063628f99d7SShri Abhyankar         v1     = aa + ai[idx];
2064628f99d7SShri Abhyankar         v2     = aa + ai[idx+1];
2065628f99d7SShri Abhyankar         for (j=0; j<nz_tmp; j++) {
2066628f99d7SShri Abhyankar           idx         = ics[ajtmp[j]];
2067628f99d7SShri Abhyankar           rtmp11[idx] = v1[j];
2068628f99d7SShri Abhyankar           rtmp22[idx] = v2[j];
2069628f99d7SShri Abhyankar         }
2070628f99d7SShri Abhyankar         rtmp11[ics[r[row]]]   += sctx.shift_amount;
2071628f99d7SShri Abhyankar         rtmp22[ics[r[row+1]]] += sctx.shift_amount;
2072628f99d7SShri Abhyankar 
2073628f99d7SShri Abhyankar         prow = *bjtmp++;
2074628f99d7SShri Abhyankar         while (prow < row) {
2075628f99d7SShri Abhyankar           pc1 = rtmp11 + prow;
2076628f99d7SShri Abhyankar           pc2 = rtmp22 + prow;
2077628f99d7SShri Abhyankar           if (*pc1 != 0.0 || *pc2 != 0.0) {
2078628f99d7SShri Abhyankar             pv   = ba + bd[prow];
2079628f99d7SShri Abhyankar             pj   = nbj + bd[prow];
2080628f99d7SShri Abhyankar             mul1 = *pc1 * *pv;
2081628f99d7SShri Abhyankar             mul2 = *pc2 * *pv;
2082628f99d7SShri Abhyankar             ++pv;
2083628f99d7SShri Abhyankar             *pc1 = mul1;
2084628f99d7SShri Abhyankar             *pc2 = mul2;
2085628f99d7SShri Abhyankar 
2086628f99d7SShri Abhyankar             nz_tmp = bi[prow+1] - bd[prow] - 1;
2087628f99d7SShri Abhyankar             for (j=0; j<nz_tmp; j++) {
2088628f99d7SShri Abhyankar               tmp          = pv[j];
2089628f99d7SShri Abhyankar               idx          = pj[j];
2090628f99d7SShri Abhyankar               rtmp11[idx] -= mul1 * tmp;
2091628f99d7SShri Abhyankar               rtmp22[idx] -= mul2 * tmp;
2092628f99d7SShri Abhyankar             }
209328b1a77fSLisandro Dalcin             ierr = PetscLogFlops(2+4*nz_tmp);CHKERRQ(ierr);
2094628f99d7SShri Abhyankar           }
2095628f99d7SShri Abhyankar           prow = *bjtmp++;
2096628f99d7SShri Abhyankar         }
2097628f99d7SShri Abhyankar 
2098628f99d7SShri Abhyankar         /* Now take care of diagonal 2x2 block. Note: prow = row here */
2099628f99d7SShri Abhyankar         pc1 = rtmp11 + prow;
2100628f99d7SShri Abhyankar         pc2 = rtmp22 + prow;
2101628f99d7SShri Abhyankar 
2102628f99d7SShri Abhyankar         sctx.pv = *pc1;
2103628f99d7SShri Abhyankar         pj      = bj + bi[prow];
2104628f99d7SShri Abhyankar         rs      = 0.0;
2105628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2106628f99d7SShri Abhyankar           idx = pj[j];
2107628f99d7SShri Abhyankar           if (idx != prow) rs += PetscAbsScalar(rtmp11[idx]);
2108628f99d7SShri Abhyankar         }
2109628f99d7SShri Abhyankar         sctx.rs = rs;
2110d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,row);CHKERRQ(ierr);
211107b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2112628f99d7SShri Abhyankar 
2113628f99d7SShri Abhyankar         if (*pc2 != 0.0) {
2114628f99d7SShri Abhyankar           pj     = nbj + bd[prow];
2115628f99d7SShri Abhyankar           mul2   = (*pc2)/(*pc1); /* since diag is not yet inverted.*/
2116628f99d7SShri Abhyankar           *pc2   = mul2;
2117628f99d7SShri Abhyankar           nz_tmp = bi[prow+1] - bd[prow] - 1;
2118628f99d7SShri Abhyankar           for (j=0; j<nz_tmp; j++) {
2119628f99d7SShri Abhyankar             idx          = pj[j];
2120628f99d7SShri Abhyankar             tmp          = rtmp11[idx];
2121628f99d7SShri Abhyankar             rtmp22[idx] -= mul2 * tmp;
2122628f99d7SShri Abhyankar           }
212328b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz_tmp);CHKERRQ(ierr);
2124628f99d7SShri Abhyankar         }
2125628f99d7SShri Abhyankar 
2126628f99d7SShri Abhyankar         pj  = bj + bi[row];
2127628f99d7SShri Abhyankar         pc1 = ba + bi[row];
2128628f99d7SShri Abhyankar         pc2 = ba + bi[row+1];
2129628f99d7SShri Abhyankar 
2130628f99d7SShri Abhyankar         sctx.pv       = rtmp22[row+1];
2131628f99d7SShri Abhyankar         rs            = 0.0;
2132628f99d7SShri Abhyankar         rtmp11[row]   = 1.0/rtmp11[row];
2133628f99d7SShri Abhyankar         rtmp22[row+1] = 1.0/rtmp22[row+1];
2134628f99d7SShri Abhyankar         /* copy row entries from dense representation to sparse */
2135628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2136628f99d7SShri Abhyankar           idx    = pj[j];
2137628f99d7SShri Abhyankar           pc1[j] = rtmp11[idx];
2138628f99d7SShri Abhyankar           pc2[j] = rtmp22[idx];
2139628f99d7SShri Abhyankar           if (idx != row+1) rs += PetscAbsScalar(pc2[j]);
2140628f99d7SShri Abhyankar         }
2141628f99d7SShri Abhyankar         sctx.rs = rs;
2142d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,row+1);CHKERRQ(ierr);
214307b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2144628f99d7SShri Abhyankar         break;
2145628f99d7SShri Abhyankar 
2146628f99d7SShri Abhyankar       case 3:
2147628f99d7SShri Abhyankar         for  (j=0; j<nz; j++) {
2148628f99d7SShri Abhyankar           idx         = bjtmp[j];
2149628f99d7SShri Abhyankar           rtmp11[idx] = 0.0;
2150628f99d7SShri Abhyankar           rtmp22[idx] = 0.0;
2151628f99d7SShri Abhyankar           rtmp33[idx] = 0.0;
2152628f99d7SShri Abhyankar         }
2153628f99d7SShri Abhyankar         /* copy the nonzeros for the 3 rows from sparse representation to dense in rtmp*[] */
2154628f99d7SShri Abhyankar         idx    = r[row];
2155628f99d7SShri Abhyankar         nz_tmp = ai[idx+1] - ai[idx];
2156628f99d7SShri Abhyankar         ajtmp  = aj + ai[idx];
2157628f99d7SShri Abhyankar         v1     = aa + ai[idx];
2158628f99d7SShri Abhyankar         v2     = aa + ai[idx+1];
2159628f99d7SShri Abhyankar         v3     = aa + ai[idx+2];
2160628f99d7SShri Abhyankar         for (j=0; j<nz_tmp; j++) {
2161628f99d7SShri Abhyankar           idx         = ics[ajtmp[j]];
2162628f99d7SShri Abhyankar           rtmp11[idx] = v1[j];
2163628f99d7SShri Abhyankar           rtmp22[idx] = v2[j];
2164628f99d7SShri Abhyankar           rtmp33[idx] = v3[j];
2165628f99d7SShri Abhyankar         }
2166628f99d7SShri Abhyankar         rtmp11[ics[r[row]]]   += sctx.shift_amount;
2167628f99d7SShri Abhyankar         rtmp22[ics[r[row+1]]] += sctx.shift_amount;
2168628f99d7SShri Abhyankar         rtmp33[ics[r[row+2]]] += sctx.shift_amount;
2169628f99d7SShri Abhyankar 
2170628f99d7SShri Abhyankar         /* loop over all pivot row blocks above this row block */
2171628f99d7SShri Abhyankar         prow = *bjtmp++;
2172628f99d7SShri Abhyankar         while (prow < row) {
2173628f99d7SShri Abhyankar           pc1 = rtmp11 + prow;
2174628f99d7SShri Abhyankar           pc2 = rtmp22 + prow;
2175628f99d7SShri Abhyankar           pc3 = rtmp33 + prow;
2176628f99d7SShri Abhyankar           if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 !=0.0) {
2177628f99d7SShri Abhyankar             pv   = ba  + bd[prow];
2178628f99d7SShri Abhyankar             pj   = nbj + bd[prow];
2179628f99d7SShri Abhyankar             mul1 = *pc1 * *pv;
2180628f99d7SShri Abhyankar             mul2 = *pc2 * *pv;
2181628f99d7SShri Abhyankar             mul3 = *pc3 * *pv;
2182628f99d7SShri Abhyankar             ++pv;
2183628f99d7SShri Abhyankar             *pc1 = mul1;
2184628f99d7SShri Abhyankar             *pc2 = mul2;
2185628f99d7SShri Abhyankar             *pc3 = mul3;
2186628f99d7SShri Abhyankar 
2187628f99d7SShri Abhyankar             nz_tmp = bi[prow+1] - bd[prow] - 1;
2188628f99d7SShri Abhyankar             /* update this row based on pivot row */
2189628f99d7SShri Abhyankar             for (j=0; j<nz_tmp; j++) {
2190628f99d7SShri Abhyankar               tmp          = pv[j];
2191628f99d7SShri Abhyankar               idx          = pj[j];
2192628f99d7SShri Abhyankar               rtmp11[idx] -= mul1 * tmp;
2193628f99d7SShri Abhyankar               rtmp22[idx] -= mul2 * tmp;
2194628f99d7SShri Abhyankar               rtmp33[idx] -= mul3 * tmp;
2195628f99d7SShri Abhyankar             }
219628b1a77fSLisandro Dalcin             ierr = PetscLogFlops(3+6*nz_tmp);CHKERRQ(ierr);
2197628f99d7SShri Abhyankar           }
2198628f99d7SShri Abhyankar           prow = *bjtmp++;
2199628f99d7SShri Abhyankar         }
2200628f99d7SShri Abhyankar 
2201628f99d7SShri Abhyankar         /* Now take care of diagonal 3x3 block in this set of rows */
2202628f99d7SShri Abhyankar         /* note: prow = row here */
2203628f99d7SShri Abhyankar         pc1 = rtmp11 + prow;
2204628f99d7SShri Abhyankar         pc2 = rtmp22 + prow;
2205628f99d7SShri Abhyankar         pc3 = rtmp33 + prow;
2206628f99d7SShri Abhyankar 
2207628f99d7SShri Abhyankar         sctx.pv = *pc1;
2208628f99d7SShri Abhyankar         pj      = bj + bi[prow];
2209628f99d7SShri Abhyankar         rs      = 0.0;
2210628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2211628f99d7SShri Abhyankar           idx = pj[j];
2212628f99d7SShri Abhyankar           if (idx != row) rs += PetscAbsScalar(rtmp11[idx]);
2213628f99d7SShri Abhyankar         }
2214628f99d7SShri Abhyankar         sctx.rs = rs;
2215d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,row);CHKERRQ(ierr);
221607b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2217628f99d7SShri Abhyankar 
2218628f99d7SShri Abhyankar         if (*pc2 != 0.0 || *pc3 != 0.0) {
2219628f99d7SShri Abhyankar           mul2   = (*pc2)/(*pc1);
2220628f99d7SShri Abhyankar           mul3   = (*pc3)/(*pc1);
2221628f99d7SShri Abhyankar           *pc2   = mul2;
2222628f99d7SShri Abhyankar           *pc3   = mul3;
2223628f99d7SShri Abhyankar           nz_tmp = bi[prow+1] - bd[prow] - 1;
2224628f99d7SShri Abhyankar           pj     = nbj + bd[prow];
2225628f99d7SShri Abhyankar           for (j=0; j<nz_tmp; j++) {
2226628f99d7SShri Abhyankar             idx          = pj[j];
2227628f99d7SShri Abhyankar             tmp          = rtmp11[idx];
2228628f99d7SShri Abhyankar             rtmp22[idx] -= mul2 * tmp;
2229628f99d7SShri Abhyankar             rtmp33[idx] -= mul3 * tmp;
2230628f99d7SShri Abhyankar           }
223128b1a77fSLisandro Dalcin           ierr = PetscLogFlops(2+4*nz_tmp);CHKERRQ(ierr);
2232628f99d7SShri Abhyankar         }
2233628f99d7SShri Abhyankar         ++prow;
2234628f99d7SShri Abhyankar 
2235628f99d7SShri Abhyankar         pc2     = rtmp22 + prow;
2236628f99d7SShri Abhyankar         pc3     = rtmp33 + prow;
2237628f99d7SShri Abhyankar         sctx.pv = *pc2;
2238628f99d7SShri Abhyankar         pj      = bj + bi[prow];
2239628f99d7SShri Abhyankar         rs      = 0.0;
2240628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2241628f99d7SShri Abhyankar           idx = pj[j];
2242628f99d7SShri Abhyankar           if (idx != prow) rs += PetscAbsScalar(rtmp22[idx]);
2243628f99d7SShri Abhyankar         }
2244628f99d7SShri Abhyankar         sctx.rs = rs;
2245d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,row+1);CHKERRQ(ierr);
224607b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2247628f99d7SShri Abhyankar 
2248628f99d7SShri Abhyankar         if (*pc3 != 0.0) {
2249628f99d7SShri Abhyankar           mul3   = (*pc3)/(*pc2);
2250628f99d7SShri Abhyankar           *pc3   = mul3;
2251628f99d7SShri Abhyankar           pj     = nbj + bd[prow];
2252628f99d7SShri Abhyankar           nz_tmp = bi[prow+1] - bd[prow] - 1;
2253628f99d7SShri Abhyankar           for (j=0; j<nz_tmp; j++) {
2254628f99d7SShri Abhyankar             idx          = pj[j];
2255628f99d7SShri Abhyankar             tmp          = rtmp22[idx];
2256628f99d7SShri Abhyankar             rtmp33[idx] -= mul3 * tmp;
2257628f99d7SShri Abhyankar           }
225828b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz_tmp);CHKERRQ(ierr);
2259628f99d7SShri Abhyankar         }
2260628f99d7SShri Abhyankar 
2261628f99d7SShri Abhyankar         pj  = bj + bi[row];
2262628f99d7SShri Abhyankar         pc1 = ba + bi[row];
2263628f99d7SShri Abhyankar         pc2 = ba + bi[row+1];
2264628f99d7SShri Abhyankar         pc3 = ba + bi[row+2];
2265628f99d7SShri Abhyankar 
2266628f99d7SShri Abhyankar         sctx.pv       = rtmp33[row+2];
2267628f99d7SShri Abhyankar         rs            = 0.0;
2268628f99d7SShri Abhyankar         rtmp11[row]   = 1.0/rtmp11[row];
2269628f99d7SShri Abhyankar         rtmp22[row+1] = 1.0/rtmp22[row+1];
2270628f99d7SShri Abhyankar         rtmp33[row+2] = 1.0/rtmp33[row+2];
2271628f99d7SShri Abhyankar         /* copy row entries from dense representation to sparse */
2272628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2273628f99d7SShri Abhyankar           idx    = pj[j];
2274628f99d7SShri Abhyankar           pc1[j] = rtmp11[idx];
2275628f99d7SShri Abhyankar           pc2[j] = rtmp22[idx];
2276628f99d7SShri Abhyankar           pc3[j] = rtmp33[idx];
2277628f99d7SShri Abhyankar           if (idx != row+2) rs += PetscAbsScalar(pc3[j]);
2278628f99d7SShri Abhyankar         }
2279628f99d7SShri Abhyankar 
2280628f99d7SShri Abhyankar         sctx.rs = rs;
2281d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,row+2);CHKERRQ(ierr);
228207b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2283628f99d7SShri Abhyankar         break;
2284628f99d7SShri Abhyankar 
2285628f99d7SShri Abhyankar       default:
2286e32f2f54SBarry Smith         SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Node size not yet supported \n");
2287628f99d7SShri Abhyankar       }
2288628f99d7SShri Abhyankar       row += nodesz;                 /* Update the row */
2289628f99d7SShri Abhyankar     }
2290628f99d7SShri Abhyankar endofwhile:;
229107b50cabSHong Zhang   } while (sctx.newshift);
2292eb9baa12SBarry Smith   ierr = PetscFree3(rtmp11,rtmp22,rtmp33);CHKERRQ(ierr);
2293628f99d7SShri Abhyankar   ierr = PetscFree(tmp_vec2);CHKERRQ(ierr);
2294628f99d7SShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2295628f99d7SShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
2296628f99d7SShri Abhyankar   ierr = ISRestoreIndices(iscol,&c);CHKERRQ(ierr);
22972205254eSKarl Rupp 
2298d3ac4fa3SBarry Smith   (B)->ops->solve = MatSolve_SeqAIJ_inplace;
2299628f99d7SShri Abhyankar   /* do not set solve add, since MatSolve_Inode + Add is faster */
2300628f99d7SShri Abhyankar   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
2301628f99d7SShri Abhyankar   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
2302628f99d7SShri Abhyankar   C->assembled              = PETSC_TRUE;
2303628f99d7SShri Abhyankar   C->preallocated           = PETSC_TRUE;
2304628f99d7SShri Abhyankar   if (sctx.nshift) {
2305f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
230657622a8eSBarry 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);
2307f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
230857622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
2309628f99d7SShri Abhyankar     }
2310628f99d7SShri Abhyankar   }
2311628f99d7SShri Abhyankar   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
2312a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode(C);CHKERRQ(ierr);
2313628f99d7SShri Abhyankar   PetscFunctionReturn(0);
2314628f99d7SShri Abhyankar }
2315628f99d7SShri Abhyankar 
2316628f99d7SShri Abhyankar 
2317019b515eSShri Abhyankar /* ----------------------------------------------------------- */
2318019b515eSShri Abhyankar #undef __FUNCT__
231935233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_Inode"
232035233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_Inode(Mat A,Vec bb,Vec xx)
2321019b515eSShri Abhyankar {
2322019b515eSShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
2323019b515eSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
2324019b515eSShri Abhyankar   PetscErrorCode    ierr;
2325019b515eSShri Abhyankar   const PetscInt    *r,*c,*rout,*cout;
23268758e1faSBarry Smith   PetscInt          i,j,n = A->rmap->n;
23278758e1faSBarry Smith   PetscInt          node_max,row,nsz,aii,i0,i1,nz;
23288758e1faSBarry Smith   const PetscInt    *ai = a->i,*a_j = a->j,*ns,*vi,*ad,*aj;
2329019b515eSShri Abhyankar   PetscScalar       *x,*tmp,*tmps,tmp0,tmp1;
2330019b515eSShri Abhyankar   PetscScalar       sum1,sum2,sum3,sum4,sum5;
2331019b515eSShri Abhyankar   const MatScalar   *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa;
2332019b515eSShri Abhyankar   const PetscScalar *b;
2333019b515eSShri Abhyankar 
2334019b515eSShri Abhyankar   PetscFunctionBegin;
2335e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
2336019b515eSShri Abhyankar   node_max = a->inode.node_count;
2337019b515eSShri Abhyankar   ns       = a->inode.size;     /* Node Size array */
2338019b515eSShri Abhyankar 
23393649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
2340019b515eSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2341019b515eSShri Abhyankar   tmp  = a->solve_work;
2342019b515eSShri Abhyankar 
2343019b515eSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
2344019b515eSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
2345019b515eSShri Abhyankar 
2346019b515eSShri Abhyankar   /* forward solve the lower triangular */
2347019b515eSShri Abhyankar   tmps = tmp;
2348019b515eSShri Abhyankar   aa   = a_a;
2349019b515eSShri Abhyankar   aj   = a_j;
2350019b515eSShri Abhyankar   ad   = a->diag;
2351019b515eSShri Abhyankar 
2352019b515eSShri Abhyankar   for (i = 0,row = 0; i< node_max; ++i) {
2353019b515eSShri Abhyankar     nsz = ns[i];
2354019b515eSShri Abhyankar     aii = ai[row];
2355019b515eSShri Abhyankar     v1  = aa + aii;
2356019b515eSShri Abhyankar     vi  = aj + aii;
2357019b515eSShri Abhyankar     nz  = ai[row+1]- ai[row];
2358019b515eSShri Abhyankar 
235998991853SShri Abhyankar     if (i < node_max-1) {
236098991853SShri Abhyankar       /* Prefetch the indices for the next block */
236150d8bf02SJed Brown       PetscPrefetchBlock(aj+ai[row+nsz],ai[row+nsz+1]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); /* indices */
236298991853SShri Abhyankar       /* Prefetch the data for the next block */
236350d8bf02SJed Brown       PetscPrefetchBlock(aa+ai[row+nsz],ai[row+nsz+ns[i+1]]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA);
236498991853SShri Abhyankar     }
236598991853SShri Abhyankar 
2366019b515eSShri Abhyankar     switch (nsz) {               /* Each loop in 'case' is unrolled */
2367019b515eSShri Abhyankar     case 1:
2368019b515eSShri Abhyankar       sum1 = b[r[row]];
2369019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2370019b515eSShri Abhyankar         i0    = vi[j];
2371019b515eSShri Abhyankar         i1    = vi[j+1];
2372019b515eSShri Abhyankar         tmp0  = tmps[i0];
2373019b515eSShri Abhyankar         tmp1  = tmps[i1];
2374019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0 + v1[j+1]*tmp1;
2375019b515eSShri Abhyankar       }
2376019b515eSShri Abhyankar       if (j == nz-1) {
2377019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2378019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0;
2379019b515eSShri Abhyankar       }
2380019b515eSShri Abhyankar       tmp[row++]=sum1;
2381019b515eSShri Abhyankar       break;
2382019b515eSShri Abhyankar     case 2:
2383019b515eSShri Abhyankar       sum1 = b[r[row]];
2384019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2385019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2386019b515eSShri Abhyankar 
2387019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2388019b515eSShri Abhyankar         i0    = vi[j];
2389019b515eSShri Abhyankar         i1    = vi[j+1];
2390019b515eSShri Abhyankar         tmp0  = tmps[i0];
2391019b515eSShri Abhyankar         tmp1  = tmps[i1];
2392019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2393019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2394019b515eSShri Abhyankar       }
2395019b515eSShri Abhyankar       if (j == nz-1) {
2396019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2397019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2398019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2399019b515eSShri Abhyankar       }
2400019b515eSShri Abhyankar       sum2     -= v2[nz] * sum1;
2401019b515eSShri Abhyankar       tmp[row++]=sum1;
2402019b515eSShri Abhyankar       tmp[row++]=sum2;
2403019b515eSShri Abhyankar       break;
2404019b515eSShri Abhyankar     case 3:
2405019b515eSShri Abhyankar       sum1 = b[r[row]];
2406019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2407019b515eSShri Abhyankar       sum3 = b[r[row+2]];
2408019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2409019b515eSShri Abhyankar       v3   = aa + ai[row+2];
2410019b515eSShri Abhyankar 
2411019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2412019b515eSShri Abhyankar         i0    = vi[j];
2413019b515eSShri Abhyankar         i1    = vi[j+1];
2414019b515eSShri Abhyankar         tmp0  = tmps[i0];
2415019b515eSShri Abhyankar         tmp1  = tmps[i1];
2416019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2417019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2418019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
2419019b515eSShri Abhyankar       }
2420019b515eSShri Abhyankar       if (j == nz-1) {
2421019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2422019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2423019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2424019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
2425019b515eSShri Abhyankar       }
2426019b515eSShri Abhyankar       sum2     -= v2[nz] * sum1;
2427019b515eSShri Abhyankar       sum3     -= v3[nz] * sum1;
2428019b515eSShri Abhyankar       sum3     -= v3[nz+1] * sum2;
2429019b515eSShri Abhyankar       tmp[row++]=sum1;
2430019b515eSShri Abhyankar       tmp[row++]=sum2;
2431019b515eSShri Abhyankar       tmp[row++]=sum3;
2432019b515eSShri Abhyankar       break;
2433019b515eSShri Abhyankar 
2434019b515eSShri Abhyankar     case 4:
2435019b515eSShri Abhyankar       sum1 = b[r[row]];
2436019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2437019b515eSShri Abhyankar       sum3 = b[r[row+2]];
2438019b515eSShri Abhyankar       sum4 = b[r[row+3]];
2439019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2440019b515eSShri Abhyankar       v3   = aa + ai[row+2];
2441019b515eSShri Abhyankar       v4   = aa + ai[row+3];
2442019b515eSShri Abhyankar 
2443019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2444019b515eSShri Abhyankar         i0    = vi[j];
2445019b515eSShri Abhyankar         i1    = vi[j+1];
2446019b515eSShri Abhyankar         tmp0  = tmps[i0];
2447019b515eSShri Abhyankar         tmp1  = tmps[i1];
2448019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2449019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2450019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
2451019b515eSShri Abhyankar         sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1;
2452019b515eSShri Abhyankar       }
2453019b515eSShri Abhyankar       if (j == nz-1) {
2454019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2455019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2456019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2457019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
2458019b515eSShri Abhyankar         sum4 -= v4[j] *tmp0;
2459019b515eSShri Abhyankar       }
2460019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
2461019b515eSShri Abhyankar       sum3 -= v3[nz] * sum1;
2462019b515eSShri Abhyankar       sum4 -= v4[nz] * sum1;
2463019b515eSShri Abhyankar       sum3 -= v3[nz+1] * sum2;
2464019b515eSShri Abhyankar       sum4 -= v4[nz+1] * sum2;
2465019b515eSShri Abhyankar       sum4 -= v4[nz+2] * sum3;
2466019b515eSShri Abhyankar 
2467019b515eSShri Abhyankar       tmp[row++]=sum1;
2468019b515eSShri Abhyankar       tmp[row++]=sum2;
2469019b515eSShri Abhyankar       tmp[row++]=sum3;
2470019b515eSShri Abhyankar       tmp[row++]=sum4;
2471019b515eSShri Abhyankar       break;
2472019b515eSShri Abhyankar     case 5:
2473019b515eSShri Abhyankar       sum1 = b[r[row]];
2474019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2475019b515eSShri Abhyankar       sum3 = b[r[row+2]];
2476019b515eSShri Abhyankar       sum4 = b[r[row+3]];
2477019b515eSShri Abhyankar       sum5 = b[r[row+4]];
2478019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2479019b515eSShri Abhyankar       v3   = aa + ai[row+2];
2480019b515eSShri Abhyankar       v4   = aa + ai[row+3];
2481019b515eSShri Abhyankar       v5   = aa + ai[row+4];
2482019b515eSShri Abhyankar 
2483019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2484019b515eSShri Abhyankar         i0    = vi[j];
2485019b515eSShri Abhyankar         i1    = vi[j+1];
2486019b515eSShri Abhyankar         tmp0  = tmps[i0];
2487019b515eSShri Abhyankar         tmp1  = tmps[i1];
2488019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2489019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2490019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
2491019b515eSShri Abhyankar         sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1;
2492019b515eSShri Abhyankar         sum5 -= v5[j] * tmp0 + v5[j+1] * tmp1;
2493019b515eSShri Abhyankar       }
2494019b515eSShri Abhyankar       if (j == nz-1) {
2495019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2496019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2497019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2498019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
2499019b515eSShri Abhyankar         sum4 -= v4[j] *tmp0;
2500019b515eSShri Abhyankar         sum5 -= v5[j] *tmp0;
2501019b515eSShri Abhyankar       }
2502019b515eSShri Abhyankar 
2503019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
2504019b515eSShri Abhyankar       sum3 -= v3[nz] * sum1;
2505019b515eSShri Abhyankar       sum4 -= v4[nz] * sum1;
2506019b515eSShri Abhyankar       sum5 -= v5[nz] * sum1;
2507019b515eSShri Abhyankar       sum3 -= v3[nz+1] * sum2;
2508019b515eSShri Abhyankar       sum4 -= v4[nz+1] * sum2;
2509019b515eSShri Abhyankar       sum5 -= v5[nz+1] * sum2;
2510019b515eSShri Abhyankar       sum4 -= v4[nz+2] * sum3;
2511019b515eSShri Abhyankar       sum5 -= v5[nz+2] * sum3;
2512019b515eSShri Abhyankar       sum5 -= v5[nz+3] * sum4;
2513019b515eSShri Abhyankar 
2514019b515eSShri Abhyankar       tmp[row++]=sum1;
2515019b515eSShri Abhyankar       tmp[row++]=sum2;
2516019b515eSShri Abhyankar       tmp[row++]=sum3;
2517019b515eSShri Abhyankar       tmp[row++]=sum4;
2518019b515eSShri Abhyankar       tmp[row++]=sum5;
2519019b515eSShri Abhyankar       break;
2520019b515eSShri Abhyankar     default:
2521e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
2522019b515eSShri Abhyankar     }
2523019b515eSShri Abhyankar   }
2524019b515eSShri Abhyankar   /* backward solve the upper triangular */
2525019b515eSShri Abhyankar   for (i=node_max -1,row = n-1; i>=0; i--) {
2526019b515eSShri Abhyankar     nsz = ns[i];
2527019b515eSShri Abhyankar     aii = ad[row+1] + 1;
2528019b515eSShri Abhyankar     v1  = aa + aii;
2529019b515eSShri Abhyankar     vi  = aj + aii;
2530019b515eSShri Abhyankar     nz  = ad[row]- ad[row+1] - 1;
253198991853SShri Abhyankar 
253298991853SShri Abhyankar     if (i > 0) {
253398991853SShri Abhyankar       /* Prefetch the indices for the next block */
253450d8bf02SJed Brown       PetscPrefetchBlock(aj+ad[row-nsz+1]+1,ad[row-nsz]-ad[row-nsz+1],0,PETSC_PREFETCH_HINT_NTA);
253598991853SShri Abhyankar       /* Prefetch the data for the next block */
253650d8bf02SJed Brown       PetscPrefetchBlock(aa+ad[row-nsz+1]+1,ad[row-nsz-ns[i-1]+1]-ad[row-nsz+1],0,PETSC_PREFETCH_HINT_NTA);
253798991853SShri Abhyankar     }
253898991853SShri Abhyankar 
2539019b515eSShri Abhyankar     switch (nsz) {               /* Each loop in 'case' is unrolled */
2540019b515eSShri Abhyankar     case 1:
2541019b515eSShri Abhyankar       sum1 = tmp[row];
2542019b515eSShri Abhyankar 
2543019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2544019b515eSShri Abhyankar         i0    = vi[j];
2545019b515eSShri Abhyankar         i1    = vi[j+1];
2546019b515eSShri Abhyankar         tmp0  = tmps[i0];
2547019b515eSShri Abhyankar         tmp1  = tmps[i1];
2548019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2549019b515eSShri Abhyankar       }
2550019b515eSShri Abhyankar       if (j == nz-1) {
2551019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2552019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0;
2553019b515eSShri Abhyankar       }
2554019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2555019b515eSShri Abhyankar       break;
2556019b515eSShri Abhyankar     case 2:
2557019b515eSShri Abhyankar       sum1 = tmp[row];
2558019b515eSShri Abhyankar       sum2 = tmp[row-1];
2559019b515eSShri Abhyankar       v2   = aa + ad[row] + 1;
2560019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2561019b515eSShri Abhyankar         i0    = vi[j];
2562019b515eSShri Abhyankar         i1    = vi[j+1];
2563019b515eSShri Abhyankar         tmp0  = tmps[i0];
2564019b515eSShri Abhyankar         tmp1  = tmps[i1];
2565019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2566019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2567019b515eSShri Abhyankar       }
2568019b515eSShri Abhyankar       if (j == nz-1) {
2569019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2570019b515eSShri Abhyankar         sum1 -= v1[j]* tmp0;
2571019b515eSShri Abhyankar         sum2 -= v2[j+1]* tmp0;
2572019b515eSShri Abhyankar       }
2573019b515eSShri Abhyankar 
2574019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2575019b515eSShri Abhyankar       sum2     -= v2[0] * tmp0;
2576019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2577019b515eSShri Abhyankar       break;
2578019b515eSShri Abhyankar     case 3:
2579019b515eSShri Abhyankar       sum1 = tmp[row];
2580019b515eSShri Abhyankar       sum2 = tmp[row -1];
2581019b515eSShri Abhyankar       sum3 = tmp[row -2];
2582019b515eSShri Abhyankar       v2   = aa + ad[row] + 1;
2583019b515eSShri Abhyankar       v3   = aa + ad[row -1] + 1;
2584019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2585019b515eSShri Abhyankar         i0    = vi[j];
2586019b515eSShri Abhyankar         i1    = vi[j+1];
2587019b515eSShri Abhyankar         tmp0  = tmps[i0];
2588019b515eSShri Abhyankar         tmp1  = tmps[i1];
2589019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2590019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2591019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
2592019b515eSShri Abhyankar       }
2593019b515eSShri Abhyankar       if (j== nz-1) {
2594019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2595019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
2596019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
2597019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
2598019b515eSShri Abhyankar       }
2599019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2600019b515eSShri Abhyankar       sum2     -= v2[0]* tmp0;
2601019b515eSShri Abhyankar       sum3     -= v3[1] * tmp0;
2602019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2603019b515eSShri Abhyankar       sum3     -= v3[0]* tmp0;
2604019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
2605019b515eSShri Abhyankar 
2606019b515eSShri Abhyankar       break;
2607019b515eSShri Abhyankar     case 4:
2608019b515eSShri Abhyankar       sum1 = tmp[row];
2609019b515eSShri Abhyankar       sum2 = tmp[row -1];
2610019b515eSShri Abhyankar       sum3 = tmp[row -2];
2611019b515eSShri Abhyankar       sum4 = tmp[row -3];
2612019b515eSShri Abhyankar       v2   = aa + ad[row]+1;
2613019b515eSShri Abhyankar       v3   = aa + ad[row -1]+1;
2614019b515eSShri Abhyankar       v4   = aa + ad[row -2]+1;
2615019b515eSShri Abhyankar 
2616019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2617019b515eSShri Abhyankar         i0    = vi[j];
2618019b515eSShri Abhyankar         i1    = vi[j+1];
2619019b515eSShri Abhyankar         tmp0  = tmps[i0];
2620019b515eSShri Abhyankar         tmp1  = tmps[i1];
2621019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0   + v1[j+1] * tmp1;
2622019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2623019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
2624019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1;
2625019b515eSShri Abhyankar       }
2626019b515eSShri Abhyankar       if (j== nz-1) {
2627019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2628019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
2629019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
2630019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
2631019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0;
2632019b515eSShri Abhyankar       }
2633019b515eSShri Abhyankar 
2634019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2635019b515eSShri Abhyankar       sum2     -= v2[0] * tmp0;
2636019b515eSShri Abhyankar       sum3     -= v3[1] * tmp0;
2637019b515eSShri Abhyankar       sum4     -= v4[2] * tmp0;
2638019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2639019b515eSShri Abhyankar       sum3     -= v3[0] * tmp0;
2640019b515eSShri Abhyankar       sum4     -= v4[1] * tmp0;
2641019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
2642019b515eSShri Abhyankar       sum4     -= v4[0] * tmp0;
2643019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--;
2644019b515eSShri Abhyankar       break;
2645019b515eSShri Abhyankar     case 5:
2646019b515eSShri Abhyankar       sum1 = tmp[row];
2647019b515eSShri Abhyankar       sum2 = tmp[row -1];
2648019b515eSShri Abhyankar       sum3 = tmp[row -2];
2649019b515eSShri Abhyankar       sum4 = tmp[row -3];
2650019b515eSShri Abhyankar       sum5 = tmp[row -4];
2651019b515eSShri Abhyankar       v2   = aa + ad[row]+1;
2652019b515eSShri Abhyankar       v3   = aa + ad[row -1]+1;
2653019b515eSShri Abhyankar       v4   = aa + ad[row -2]+1;
2654019b515eSShri Abhyankar       v5   = aa + ad[row -3]+1;
2655019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2656019b515eSShri Abhyankar         i0    = vi[j];
2657019b515eSShri Abhyankar         i1    = vi[j+1];
2658019b515eSShri Abhyankar         tmp0  = tmps[i0];
2659019b515eSShri Abhyankar         tmp1  = tmps[i1];
2660019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2661019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2662019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
2663019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1;
2664019b515eSShri Abhyankar         sum5 -= v5[j+4] * tmp0 + v5[j+5] * tmp1;
2665019b515eSShri Abhyankar       }
2666019b515eSShri Abhyankar       if (j==nz-1) {
2667019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2668019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
2669019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
2670019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
2671019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0;
2672019b515eSShri Abhyankar         sum5 -= v5[j+4] * tmp0;
2673019b515eSShri Abhyankar       }
2674019b515eSShri Abhyankar 
2675019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2676019b515eSShri Abhyankar       sum2     -= v2[0] * tmp0;
2677019b515eSShri Abhyankar       sum3     -= v3[1] * tmp0;
2678019b515eSShri Abhyankar       sum4     -= v4[2] * tmp0;
2679019b515eSShri Abhyankar       sum5     -= v5[3] * tmp0;
2680019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2681019b515eSShri Abhyankar       sum3     -= v3[0] * tmp0;
2682019b515eSShri Abhyankar       sum4     -= v4[1] * tmp0;
2683019b515eSShri Abhyankar       sum5     -= v5[2] * tmp0;
2684019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
2685019b515eSShri Abhyankar       sum4     -= v4[0] * tmp0;
2686019b515eSShri Abhyankar       sum5     -= v5[1] * tmp0;
2687019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--;
2688019b515eSShri Abhyankar       sum5     -= v5[0] * tmp0;
2689019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum5*v5[nz+4]; row--;
2690019b515eSShri Abhyankar       break;
2691019b515eSShri Abhyankar     default:
2692e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
2693019b515eSShri Abhyankar     }
2694019b515eSShri Abhyankar   }
2695019b515eSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
2696019b515eSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
26973649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
2698019b515eSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
2699019b515eSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
2700019b515eSShri Abhyankar   PetscFunctionReturn(0);
2701019b515eSShri Abhyankar }
2702019b515eSShri Abhyankar 
2703019b515eSShri Abhyankar 
27044c1414c8SBarry Smith /*
27054c1414c8SBarry Smith      Makes a longer coloring[] array and calls the usual code with that
27064c1414c8SBarry Smith */
27074c1414c8SBarry Smith #undef __FUNCT__
27084108e4d5SBarry Smith #define __FUNCT__ "MatColoringPatch_SeqAIJ_Inode"
27094108e4d5SBarry Smith PetscErrorCode MatColoringPatch_SeqAIJ_Inode(Mat mat,PetscInt ncolors,PetscInt nin,ISColoringValue coloring[],ISColoring *iscoloring)
27104c1414c8SBarry Smith {
27114c1414c8SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)mat->data;
27124c1414c8SBarry Smith   PetscErrorCode  ierr;
2713d0f46423SBarry Smith   PetscInt        n = mat->cmap->n,m = a->inode.node_count,j,*ns = a->inode.size,row;
27144c1414c8SBarry Smith   PetscInt        *colorused,i;
27154c1414c8SBarry Smith   ISColoringValue *newcolor;
27164c1414c8SBarry Smith 
27174c1414c8SBarry Smith   PetscFunctionBegin;
2718854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&newcolor);CHKERRQ(ierr);
27194c1414c8SBarry Smith   /* loop over inodes, marking a color for each column*/
27204c1414c8SBarry Smith   row = 0;
27214c1414c8SBarry Smith   for (i=0; i<m; i++) {
27224c1414c8SBarry Smith     for (j=0; j<ns[i]; j++) {
27234c1414c8SBarry Smith       newcolor[row++] = coloring[i] + j*ncolors;
27244c1414c8SBarry Smith     }
27254c1414c8SBarry Smith   }
27264c1414c8SBarry Smith 
27274c1414c8SBarry Smith   /* eliminate unneeded colors */
27281795a4d1SJed Brown   ierr = PetscCalloc1(5*ncolors,&colorused);CHKERRQ(ierr);
27294c1414c8SBarry Smith   for (i=0; i<n; i++) {
27304c1414c8SBarry Smith     colorused[newcolor[i]] = 1;
27314c1414c8SBarry Smith   }
27324c1414c8SBarry Smith 
27334c1414c8SBarry Smith   for (i=1; i<5*ncolors; i++) {
27344c1414c8SBarry Smith     colorused[i] += colorused[i-1];
27354c1414c8SBarry Smith   }
27364c1414c8SBarry Smith   ncolors = colorused[5*ncolors-1];
27374c1414c8SBarry Smith   for (i=0; i<n; i++) {
273857815681SSatish Balay     newcolor[i] = colorused[newcolor[i]]-1;
27394c1414c8SBarry Smith   }
27404c1414c8SBarry Smith   ierr = PetscFree(colorused);CHKERRQ(ierr);
2741aaf3ff59SMatthew G. Knepley   ierr = ISColoringCreate(PetscObjectComm((PetscObject)mat),ncolors,n,newcolor,PETSC_OWN_POINTER,iscoloring);CHKERRQ(ierr);
27424c1414c8SBarry Smith   ierr = PetscFree(coloring);CHKERRQ(ierr);
27434c1414c8SBarry Smith   PetscFunctionReturn(0);
27444c1414c8SBarry Smith }
27454c1414c8SBarry Smith 
2746af0996ceSBarry Smith #include <petsc/private/kernels/blockinvert.h>
27472af78befSBarry Smith 
27482af78befSBarry Smith #undef __FUNCT__
27494108e4d5SBarry Smith #define __FUNCT__ "MatSOR_SeqAIJ_Inode"
27504108e4d5SBarry Smith PetscErrorCode MatSOR_SeqAIJ_Inode(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx)
27512af78befSBarry Smith {
27522af78befSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
27537aaeff0aSMatthew G. Knepley   PetscScalar       sum1 = 0.0,sum2 = 0.0,sum3 = 0.0,sum4 = 0.0,sum5 = 0.0,tmp0,tmp1,tmp2,tmp3;
27545850ef23SBarry Smith   MatScalar         *ibdiag,*bdiag,work[25],*t;
2755a8b09249SBarry Smith   PetscScalar       *x,tmp4,tmp5,x1,x2,x3,x4,x5;
27567aaeff0aSMatthew G. Knepley   const MatScalar   *v = a->a,*v1 = NULL,*v2 = NULL,*v3 = NULL,*v4 = NULL,*v5 = NULL;
27575850ef23SBarry Smith   const PetscScalar *xb, *b;
275862bba022SBarry Smith   PetscReal         zeropivot = 1.0e-15, shift = 0.0;
27592af78befSBarry Smith   PetscErrorCode    ierr;
27608758e1faSBarry Smith   PetscInt          n,m = a->inode.node_count,cnt = 0,i,j,row,i1,i2;
27618758e1faSBarry Smith   PetscInt          sz,k,ipvt[5];
27628758e1faSBarry Smith   const PetscInt    *sizes = a->inode.size,*idx,*diag = a->diag,*ii = a->i;
27632af78befSBarry Smith 
27642af78befSBarry Smith   PetscFunctionBegin;
2765e32f2f54SBarry Smith   if (omega != 1.0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for omega != 1.0; use -mat_no_inode");
2766422a814eSBarry Smith   if (fshift == -1.0) fshift = 0.0; /* negative fshift indicates do not error on zero diagonal; this code never errors on zero diagonal */
2767e32f2f54SBarry Smith   if (fshift != 0.0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for fshift != 0.0; use -mat_no_inode");
27682af78befSBarry Smith 
276971f1c65dSBarry Smith   if (!a->inode.ibdiagvalid) {
27702af78befSBarry Smith     if (!a->inode.ibdiag) {
27712af78befSBarry Smith       /* calculate space needed for diagonal blocks */
27722af78befSBarry Smith       for (i=0; i<m; i++) {
27732af78befSBarry Smith         cnt += sizes[i]*sizes[i];
27742af78befSBarry Smith       }
2775f0d39aaaSBarry Smith       a->inode.bdiagsize = cnt;
27762205254eSKarl Rupp 
2777dcca6d9dSJed Brown       ierr = PetscMalloc3(cnt,&a->inode.ibdiag,cnt,&a->inode.bdiag,A->rmap->n,&a->inode.ssor_work);CHKERRQ(ierr);
277871f1c65dSBarry Smith     }
277971f1c65dSBarry Smith 
278071f1c65dSBarry Smith     /* copy over the diagonal blocks and invert them */
27812af78befSBarry Smith     ibdiag = a->inode.ibdiag;
27822af78befSBarry Smith     bdiag  = a->inode.bdiag;
27832af78befSBarry Smith     cnt    = 0;
27842af78befSBarry Smith     for (i=0, row = 0; i<m; i++) {
27852af78befSBarry Smith       for (j=0; j<sizes[i]; j++) {
2786f0d39aaaSBarry Smith         for (k=0; k<sizes[i]; k++) {
2787f0d39aaaSBarry Smith           bdiag[cnt+k*sizes[i]+j] = v[diag[row+j] - j + k];
2788f0d39aaaSBarry Smith         }
27892af78befSBarry Smith       }
2790a77337e4SBarry Smith       ierr = PetscMemcpy(ibdiag+cnt,bdiag+cnt,sizes[i]*sizes[i]*sizeof(MatScalar));CHKERRQ(ierr);
27912af78befSBarry Smith 
27922af78befSBarry Smith       switch (sizes[i]) {
27932af78befSBarry Smith       case 1:
27942af78befSBarry Smith         /* Create matrix data structure */
2795e32f2f54SBarry Smith         if (PetscAbsScalar(ibdiag[cnt]) < zeropivot) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot on row %D",row);
279664c62002SMatthew Knepley         ibdiag[cnt] = 1.0/ibdiag[cnt];
27972af78befSBarry Smith         break;
27982af78befSBarry Smith       case 2:
279996b95a6bSBarry Smith         ierr = PetscKernel_A_gets_inverse_A_2(ibdiag+cnt,shift);CHKERRQ(ierr);
28002af78befSBarry Smith         break;
28012af78befSBarry Smith       case 3:
280296b95a6bSBarry Smith         ierr = PetscKernel_A_gets_inverse_A_3(ibdiag+cnt,shift);CHKERRQ(ierr);
28032af78befSBarry Smith         break;
28042af78befSBarry Smith       case 4:
280596b95a6bSBarry Smith         ierr = PetscKernel_A_gets_inverse_A_4(ibdiag+cnt,shift);CHKERRQ(ierr);
28062af78befSBarry Smith         break;
28072af78befSBarry Smith       case 5:
280896b95a6bSBarry Smith         ierr = PetscKernel_A_gets_inverse_A_5(ibdiag+cnt,ipvt,work,shift);CHKERRQ(ierr);
28092af78befSBarry Smith         break;
28102af78befSBarry Smith       default:
2811e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
28122af78befSBarry Smith       }
28132af78befSBarry Smith       cnt += sizes[i]*sizes[i];
28142af78befSBarry Smith       row += sizes[i];
28152af78befSBarry Smith     }
281671f1c65dSBarry Smith     a->inode.ibdiagvalid = PETSC_TRUE;
28172af78befSBarry Smith   }
28182af78befSBarry Smith   ibdiag = a->inode.ibdiag;
28192af78befSBarry Smith   bdiag  = a->inode.bdiag;
28205850ef23SBarry Smith   t      = a->inode.ssor_work;
28212af78befSBarry Smith 
28222af78befSBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
28233649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
28245850ef23SBarry Smith   /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */
28255850ef23SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
28262af78befSBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) {
28272af78befSBarry Smith 
28288862d2efSBarry Smith       for (i=0, row=0; i<m; i++) {
28298862d2efSBarry Smith         sz  = diag[row] - ii[row];
28308862d2efSBarry Smith         v1  = a->a + ii[row];
28318862d2efSBarry Smith         idx = a->j + ii[row];
28328862d2efSBarry Smith 
28334108e4d5SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
28348862d2efSBarry Smith         switch (sizes[i]) {
28358862d2efSBarry Smith         case 1:
28368862d2efSBarry Smith 
28378862d2efSBarry Smith           sum1 = b[row];
28388862d2efSBarry Smith           for (n = 0; n<sz-1; n+=2) {
28398862d2efSBarry Smith             i1    = idx[0];
28408862d2efSBarry Smith             i2    = idx[1];
28418862d2efSBarry Smith             idx  += 2;
28428862d2efSBarry Smith             tmp0  = x[i1];
28438862d2efSBarry Smith             tmp1  = x[i2];
28448862d2efSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
28458862d2efSBarry Smith           }
28468862d2efSBarry Smith 
28478862d2efSBarry Smith           if (n == sz-1) {
2848f0d39aaaSBarry Smith             tmp0  = x[*idx];
2849f0d39aaaSBarry Smith             sum1 -= *v1 * tmp0;
28508862d2efSBarry Smith           }
28515850ef23SBarry Smith           t[row]   = sum1;
28528862d2efSBarry Smith           x[row++] = sum1*(*ibdiag++);
28538862d2efSBarry Smith           break;
2854f0d39aaaSBarry Smith         case 2:
2855f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2856f0d39aaaSBarry Smith           sum1 = b[row];
2857f0d39aaaSBarry Smith           sum2 = b[row+1];
2858f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2859f0d39aaaSBarry Smith             i1    = idx[0];
2860f0d39aaaSBarry Smith             i2    = idx[1];
2861f0d39aaaSBarry Smith             idx  += 2;
2862f0d39aaaSBarry Smith             tmp0  = x[i1];
2863f0d39aaaSBarry Smith             tmp1  = x[i2];
2864f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2865f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2866f0d39aaaSBarry Smith           }
2867f0d39aaaSBarry Smith 
2868f0d39aaaSBarry Smith           if (n == sz-1) {
2869f0d39aaaSBarry Smith             tmp0  = x[*idx];
2870f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2871f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2872f0d39aaaSBarry Smith           }
28735850ef23SBarry Smith           t[row]   = sum1;
28745850ef23SBarry Smith           t[row+1] = sum2;
2875f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[2];
2876f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[3];
2877f0d39aaaSBarry Smith           ibdiag  += 4;
2878f0d39aaaSBarry Smith           break;
2879f0d39aaaSBarry Smith         case 3:
2880f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2881f0d39aaaSBarry Smith           v3   = a->a + ii[row+2];
2882f0d39aaaSBarry Smith           sum1 = b[row];
2883f0d39aaaSBarry Smith           sum2 = b[row+1];
2884f0d39aaaSBarry Smith           sum3 = b[row+2];
2885f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2886f0d39aaaSBarry Smith             i1    = idx[0];
2887f0d39aaaSBarry Smith             i2    = idx[1];
2888f0d39aaaSBarry Smith             idx  += 2;
2889f0d39aaaSBarry Smith             tmp0  = x[i1];
2890f0d39aaaSBarry Smith             tmp1  = x[i2];
2891f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2892f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2893f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2894f0d39aaaSBarry Smith           }
2895f0d39aaaSBarry Smith 
2896f0d39aaaSBarry Smith           if (n == sz-1) {
2897f0d39aaaSBarry Smith             tmp0  = x[*idx];
2898f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2899f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2900f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0;
2901f0d39aaaSBarry Smith           }
29025850ef23SBarry Smith           t[row]   = sum1;
29035850ef23SBarry Smith           t[row+1] = sum2;
29045850ef23SBarry Smith           t[row+2] = sum3;
2905f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
2906f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
2907f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
2908f0d39aaaSBarry Smith           ibdiag  += 9;
2909f0d39aaaSBarry Smith           break;
2910f0d39aaaSBarry Smith         case 4:
2911f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2912f0d39aaaSBarry Smith           v3   = a->a + ii[row+2];
2913f0d39aaaSBarry Smith           v4   = a->a + ii[row+3];
2914f0d39aaaSBarry Smith           sum1 = b[row];
2915f0d39aaaSBarry Smith           sum2 = b[row+1];
2916f0d39aaaSBarry Smith           sum3 = b[row+2];
2917f0d39aaaSBarry Smith           sum4 = b[row+3];
2918f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2919f0d39aaaSBarry Smith             i1    = idx[0];
2920f0d39aaaSBarry Smith             i2    = idx[1];
2921f0d39aaaSBarry Smith             idx  += 2;
2922f0d39aaaSBarry Smith             tmp0  = x[i1];
2923f0d39aaaSBarry Smith             tmp1  = x[i2];
2924f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2925f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2926f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2927f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
2928f0d39aaaSBarry Smith           }
2929f0d39aaaSBarry Smith 
2930f0d39aaaSBarry Smith           if (n == sz-1) {
2931f0d39aaaSBarry Smith             tmp0  = x[*idx];
2932f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2933f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2934f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0;
2935f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0;
2936f0d39aaaSBarry Smith           }
29375850ef23SBarry Smith           t[row]   = sum1;
29385850ef23SBarry Smith           t[row+1] = sum2;
29395850ef23SBarry Smith           t[row+2] = sum3;
29405850ef23SBarry Smith           t[row+3] = sum4;
2941f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
2942f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
2943f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
2944f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
2945f0d39aaaSBarry Smith           ibdiag  += 16;
2946f0d39aaaSBarry Smith           break;
2947f0d39aaaSBarry Smith         case 5:
2948f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2949f0d39aaaSBarry Smith           v3   = a->a + ii[row+2];
2950f0d39aaaSBarry Smith           v4   = a->a + ii[row+3];
2951f0d39aaaSBarry Smith           v5   = a->a + ii[row+4];
2952f0d39aaaSBarry Smith           sum1 = b[row];
2953f0d39aaaSBarry Smith           sum2 = b[row+1];
2954f0d39aaaSBarry Smith           sum3 = b[row+2];
2955f0d39aaaSBarry Smith           sum4 = b[row+3];
2956f0d39aaaSBarry Smith           sum5 = b[row+4];
2957f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2958f0d39aaaSBarry Smith             i1    = idx[0];
2959f0d39aaaSBarry Smith             i2    = idx[1];
2960f0d39aaaSBarry Smith             idx  += 2;
2961f0d39aaaSBarry Smith             tmp0  = x[i1];
2962f0d39aaaSBarry Smith             tmp1  = x[i2];
2963f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2964f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2965f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2966f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
2967f0d39aaaSBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
2968f0d39aaaSBarry Smith           }
2969f0d39aaaSBarry Smith 
2970f0d39aaaSBarry Smith           if (n == sz-1) {
2971f0d39aaaSBarry Smith             tmp0  = x[*idx];
2972f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2973f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2974f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0;
2975f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0;
2976f0d39aaaSBarry Smith             sum5 -= v5[0] * tmp0;
2977f0d39aaaSBarry Smith           }
29785850ef23SBarry Smith           t[row]   = sum1;
29795850ef23SBarry Smith           t[row+1] = sum2;
29805850ef23SBarry Smith           t[row+2] = sum3;
29815850ef23SBarry Smith           t[row+3] = sum4;
29825850ef23SBarry Smith           t[row+4] = sum5;
2983f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
2984f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
2985f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
2986f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
2987f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
2988f0d39aaaSBarry Smith           ibdiag  += 25;
2989f0d39aaaSBarry Smith           break;
29908862d2efSBarry Smith         default:
2991e32f2f54SBarry Smith           SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
29928862d2efSBarry Smith         }
29932af78befSBarry Smith       }
29942af78befSBarry Smith 
29955850ef23SBarry Smith       xb   = t;
29962af78befSBarry Smith       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
29972af78befSBarry Smith     } else xb = b;
29982af78befSBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) {
29992af78befSBarry Smith 
3000f0d39aaaSBarry Smith       ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
3001d0f46423SBarry Smith       for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
3002f0d39aaaSBarry Smith         ibdiag -= sizes[i]*sizes[i];
30038862d2efSBarry Smith         sz      = ii[row+1] - diag[row] - 1;
30048862d2efSBarry Smith         v1      = a->a + diag[row] + 1;
30058862d2efSBarry Smith         idx     = a->j + diag[row] + 1;
30062af78befSBarry Smith 
30074108e4d5SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
30088862d2efSBarry Smith         switch (sizes[i]) {
30098862d2efSBarry Smith         case 1:
30108862d2efSBarry Smith 
30118862d2efSBarry Smith           sum1 = xb[row];
30128862d2efSBarry Smith           for (n = 0; n<sz-1; n+=2) {
30138862d2efSBarry Smith             i1    = idx[0];
30148862d2efSBarry Smith             i2    = idx[1];
30158862d2efSBarry Smith             idx  += 2;
30168862d2efSBarry Smith             tmp0  = x[i1];
30178862d2efSBarry Smith             tmp1  = x[i2];
30188862d2efSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
30198862d2efSBarry Smith           }
30208862d2efSBarry Smith 
30218862d2efSBarry Smith           if (n == sz-1) {
3022f0d39aaaSBarry Smith             tmp0  = x[*idx];
3023f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
30248862d2efSBarry Smith           }
3025f0d39aaaSBarry Smith           x[row--] = sum1*(*ibdiag);
3026f0d39aaaSBarry Smith           break;
3027f0d39aaaSBarry Smith 
3028f0d39aaaSBarry Smith         case 2:
3029f0d39aaaSBarry Smith 
3030f0d39aaaSBarry Smith           sum1 = xb[row];
3031f0d39aaaSBarry Smith           sum2 = xb[row-1];
3032f0d39aaaSBarry Smith           /* note that sum1 is associated with the second of the two rows */
3033f0d39aaaSBarry Smith           v2 = a->a + diag[row-1] + 2;
3034f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3035f0d39aaaSBarry Smith             i1    = idx[0];
3036f0d39aaaSBarry Smith             i2    = idx[1];
3037f0d39aaaSBarry Smith             idx  += 2;
3038f0d39aaaSBarry Smith             tmp0  = x[i1];
3039f0d39aaaSBarry Smith             tmp1  = x[i2];
3040f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3041f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3042f0d39aaaSBarry Smith           }
3043f0d39aaaSBarry Smith 
3044f0d39aaaSBarry Smith           if (n == sz-1) {
3045f0d39aaaSBarry Smith             tmp0  = x[*idx];
3046f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3047f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3048f0d39aaaSBarry Smith           }
3049f0d39aaaSBarry Smith           x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3];
3050f0d39aaaSBarry Smith           x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2];
3051f0d39aaaSBarry Smith           break;
3052f0d39aaaSBarry Smith         case 3:
3053f0d39aaaSBarry Smith 
3054f0d39aaaSBarry Smith           sum1 = xb[row];
3055f0d39aaaSBarry Smith           sum2 = xb[row-1];
3056f0d39aaaSBarry Smith           sum3 = xb[row-2];
3057f0d39aaaSBarry Smith           v2   = a->a + diag[row-1] + 2;
3058f0d39aaaSBarry Smith           v3   = a->a + diag[row-2] + 3;
3059f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3060f0d39aaaSBarry Smith             i1    = idx[0];
3061f0d39aaaSBarry Smith             i2    = idx[1];
3062f0d39aaaSBarry Smith             idx  += 2;
3063f0d39aaaSBarry Smith             tmp0  = x[i1];
3064f0d39aaaSBarry Smith             tmp1  = x[i2];
3065f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3066f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3067f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3068f0d39aaaSBarry Smith           }
3069f0d39aaaSBarry Smith 
3070f0d39aaaSBarry Smith           if (n == sz-1) {
3071f0d39aaaSBarry Smith             tmp0  = x[*idx];
3072f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3073f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3074f0d39aaaSBarry Smith             sum3 -= *v3*tmp0;
3075f0d39aaaSBarry Smith           }
3076f0d39aaaSBarry Smith           x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3077f0d39aaaSBarry Smith           x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3078f0d39aaaSBarry Smith           x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3079f0d39aaaSBarry Smith           break;
3080f0d39aaaSBarry Smith         case 4:
3081f0d39aaaSBarry Smith 
3082f0d39aaaSBarry Smith           sum1 = xb[row];
3083f0d39aaaSBarry Smith           sum2 = xb[row-1];
3084f0d39aaaSBarry Smith           sum3 = xb[row-2];
3085f0d39aaaSBarry Smith           sum4 = xb[row-3];
3086f0d39aaaSBarry Smith           v2   = a->a + diag[row-1] + 2;
3087f0d39aaaSBarry Smith           v3   = a->a + diag[row-2] + 3;
3088f0d39aaaSBarry Smith           v4   = a->a + diag[row-3] + 4;
3089f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3090f0d39aaaSBarry Smith             i1    = idx[0];
3091f0d39aaaSBarry Smith             i2    = idx[1];
3092f0d39aaaSBarry Smith             idx  += 2;
3093f0d39aaaSBarry Smith             tmp0  = x[i1];
3094f0d39aaaSBarry Smith             tmp1  = x[i2];
3095f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3096f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3097f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3098f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3099f0d39aaaSBarry Smith           }
3100f0d39aaaSBarry Smith 
3101f0d39aaaSBarry Smith           if (n == sz-1) {
3102f0d39aaaSBarry Smith             tmp0  = x[*idx];
3103f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3104f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3105f0d39aaaSBarry Smith             sum3 -= *v3*tmp0;
3106f0d39aaaSBarry Smith             sum4 -= *v4*tmp0;
3107f0d39aaaSBarry Smith           }
3108f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3109f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3110f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3111f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3112f0d39aaaSBarry Smith           break;
3113f0d39aaaSBarry Smith         case 5:
3114f0d39aaaSBarry Smith 
3115f0d39aaaSBarry Smith           sum1 = xb[row];
3116f0d39aaaSBarry Smith           sum2 = xb[row-1];
3117f0d39aaaSBarry Smith           sum3 = xb[row-2];
3118f0d39aaaSBarry Smith           sum4 = xb[row-3];
3119f0d39aaaSBarry Smith           sum5 = xb[row-4];
3120f0d39aaaSBarry Smith           v2   = a->a + diag[row-1] + 2;
3121f0d39aaaSBarry Smith           v3   = a->a + diag[row-2] + 3;
3122f0d39aaaSBarry Smith           v4   = a->a + diag[row-3] + 4;
3123f0d39aaaSBarry Smith           v5   = a->a + diag[row-4] + 5;
3124f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3125f0d39aaaSBarry Smith             i1    = idx[0];
3126f0d39aaaSBarry Smith             i2    = idx[1];
3127f0d39aaaSBarry Smith             idx  += 2;
3128f0d39aaaSBarry Smith             tmp0  = x[i1];
3129f0d39aaaSBarry Smith             tmp1  = x[i2];
3130f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3131f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3132f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3133f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3134f0d39aaaSBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
3135f0d39aaaSBarry Smith           }
3136f0d39aaaSBarry Smith 
3137f0d39aaaSBarry Smith           if (n == sz-1) {
3138f0d39aaaSBarry Smith             tmp0  = x[*idx];
3139f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3140f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3141f0d39aaaSBarry Smith             sum3 -= *v3*tmp0;
3142f0d39aaaSBarry Smith             sum4 -= *v4*tmp0;
3143f0d39aaaSBarry Smith             sum5 -= *v5*tmp0;
3144f0d39aaaSBarry Smith           }
3145f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
3146f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
3147f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
3148f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
3149f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
31508862d2efSBarry Smith           break;
31518862d2efSBarry Smith         default:
3152e32f2f54SBarry Smith           SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
31538862d2efSBarry Smith         }
31542af78befSBarry Smith       }
31552af78befSBarry Smith 
31562af78befSBarry Smith       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
31572af78befSBarry Smith     }
31582af78befSBarry Smith     its--;
31595850ef23SBarry Smith   }
31605850ef23SBarry Smith   while (its--) {
31615850ef23SBarry Smith 
31625850ef23SBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) {
3163d876e2b0SMark Adams       for (i=0, row=0, ibdiag = a->inode.ibdiag;
3164d876e2b0SMark Adams            i<m;
3165d876e2b0SMark Adams            row += sizes[i], ibdiag += sizes[i]*sizes[i], i++) {
31665850ef23SBarry Smith 
3167d876e2b0SMark Adams         sz  = diag[row] - ii[row];
31685850ef23SBarry Smith         v1  = a->a + ii[row];
31695850ef23SBarry Smith         idx = a->j + ii[row];
31705850ef23SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
31715850ef23SBarry Smith         switch (sizes[i]) {
31725850ef23SBarry Smith         case 1:
31735850ef23SBarry Smith           sum1 = b[row];
31745850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
31755850ef23SBarry Smith             i1    = idx[0];
31765850ef23SBarry Smith             i2    = idx[1];
31775850ef23SBarry Smith             idx  += 2;
31785850ef23SBarry Smith             tmp0  = x[i1];
31795850ef23SBarry Smith             tmp1  = x[i2];
31805850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
31815850ef23SBarry Smith           }
31825850ef23SBarry Smith           if (n == sz-1) {
3183d876e2b0SMark Adams             tmp0  = x[*idx++];
3184d876e2b0SMark Adams             sum1 -= *v1 * tmp0;
3185d876e2b0SMark Adams             v1++;
3186d876e2b0SMark Adams           }
3187d876e2b0SMark Adams           t[row]   = sum1;
3188d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 1;
3189d876e2b0SMark Adams           idx     = a->j + diag[row] + 1;
3190d876e2b0SMark Adams           v1 += 1;
3191d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3192d876e2b0SMark Adams             i1    = idx[0];
3193d876e2b0SMark Adams             i2    = idx[1];
3194d876e2b0SMark Adams             idx  += 2;
3195d876e2b0SMark Adams             tmp0  = x[i1];
3196d876e2b0SMark Adams             tmp1  = x[i2];
3197d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3198d876e2b0SMark Adams           }
3199d876e2b0SMark Adams           if (n == sz-1) {
3200d876e2b0SMark Adams             tmp0  = x[*idx++];
32015850ef23SBarry Smith             sum1 -= *v1 * tmp0;
32025850ef23SBarry Smith           }
32035850ef23SBarry Smith           /* in MatSOR_SeqAIJ this line would be
32045850ef23SBarry Smith            *
32055850ef23SBarry Smith            * x[row] = (1-omega)*x[row]+(sum1+(*bdiag++)*x[row])*(*ibdiag++);
32065850ef23SBarry Smith            *
32075850ef23SBarry Smith            * but omega == 1, so this becomes
32085850ef23SBarry Smith            *
3209d876e2b0SMark Adams            * x[row] = sum1*(*ibdiag++);
32105850ef23SBarry Smith            *
32115850ef23SBarry Smith            */
3212d876e2b0SMark Adams           x[row] = sum1*(*ibdiag);
32135850ef23SBarry Smith           break;
32145850ef23SBarry Smith         case 2:
32155850ef23SBarry Smith           v2   = a->a + ii[row+1];
32165850ef23SBarry Smith           sum1 = b[row];
32175850ef23SBarry Smith           sum2 = b[row+1];
32185850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
32195850ef23SBarry Smith             i1    = idx[0];
32205850ef23SBarry Smith             i2    = idx[1];
32215850ef23SBarry Smith             idx  += 2;
32225850ef23SBarry Smith             tmp0  = x[i1];
32235850ef23SBarry Smith             tmp1  = x[i2];
32245850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
32255850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
32265850ef23SBarry Smith           }
3227d876e2b0SMark Adams           if (n == sz-1) {
3228d876e2b0SMark Adams             tmp0  = x[*idx++];
3229d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3230d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3231d876e2b0SMark Adams             v1++; v2++;
3232d876e2b0SMark Adams           }
3233d876e2b0SMark Adams           t[row]   = sum1;
3234d876e2b0SMark Adams           t[row+1] = sum2;
3235d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 2;
3236d876e2b0SMark Adams           idx     = a->j + diag[row] + 2;
3237d876e2b0SMark Adams           v1 += 2;
3238d876e2b0SMark Adams           v2 += 2;
3239d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3240d876e2b0SMark Adams             i1    = idx[0];
3241d876e2b0SMark Adams             i2    = idx[1];
3242d876e2b0SMark Adams             idx  += 2;
3243d876e2b0SMark Adams             tmp0  = x[i1];
3244d876e2b0SMark Adams             tmp1  = x[i2];
3245d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3246d876e2b0SMark Adams             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3247d876e2b0SMark Adams           }
32485850ef23SBarry Smith           if (n == sz-1) {
32495850ef23SBarry Smith             tmp0  = x[*idx];
32505850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
32515850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
32525850ef23SBarry Smith           }
3253d876e2b0SMark Adams           x[row] = sum1*ibdiag[0] + sum2*ibdiag[2];
3254d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3];
32555850ef23SBarry Smith           break;
32565850ef23SBarry Smith         case 3:
32575850ef23SBarry Smith           v2   = a->a + ii[row+1];
32585850ef23SBarry Smith           v3   = a->a + ii[row+2];
32595850ef23SBarry Smith           sum1 = b[row];
32605850ef23SBarry Smith           sum2 = b[row+1];
32615850ef23SBarry Smith           sum3 = b[row+2];
32625850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
32635850ef23SBarry Smith             i1    = idx[0];
32645850ef23SBarry Smith             i2    = idx[1];
32655850ef23SBarry Smith             idx  += 2;
32665850ef23SBarry Smith             tmp0  = x[i1];
32675850ef23SBarry Smith             tmp1  = x[i2];
32685850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
32695850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
32705850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
32715850ef23SBarry Smith           }
3272d876e2b0SMark Adams           if (n == sz-1) {
3273d876e2b0SMark Adams             tmp0  = x[*idx++];
3274d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3275d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3276d876e2b0SMark Adams             sum3 -= v3[0] * tmp0;
3277d876e2b0SMark Adams             v1++; v2++; v3++;
3278d876e2b0SMark Adams           }
3279d876e2b0SMark Adams           t[row]   = sum1;
3280d876e2b0SMark Adams           t[row+1] = sum2;
3281d876e2b0SMark Adams           t[row+2] = sum3;
3282d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 3;
3283d876e2b0SMark Adams           idx     = a->j + diag[row] + 3;
3284d876e2b0SMark Adams           v1 += 3;
3285d876e2b0SMark Adams           v2 += 3;
3286d876e2b0SMark Adams           v3 += 3;
3287d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3288d876e2b0SMark Adams             i1    = idx[0];
3289d876e2b0SMark Adams             i2    = idx[1];
3290d876e2b0SMark Adams             idx  += 2;
3291d876e2b0SMark Adams             tmp0  = x[i1];
3292d876e2b0SMark Adams             tmp1  = x[i2];
3293d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3294d876e2b0SMark Adams             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3295d876e2b0SMark Adams             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3296d876e2b0SMark Adams           }
32975850ef23SBarry Smith           if (n == sz-1) {
32985850ef23SBarry Smith             tmp0  = x[*idx];
32995850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
33005850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
33015850ef23SBarry Smith             sum3 -= v3[0] * tmp0;
33025850ef23SBarry Smith           }
3303d876e2b0SMark Adams           x[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
3304d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
3305d876e2b0SMark Adams           x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
33065850ef23SBarry Smith           break;
33075850ef23SBarry Smith         case 4:
33085850ef23SBarry Smith           v2   = a->a + ii[row+1];
33095850ef23SBarry Smith           v3   = a->a + ii[row+2];
33105850ef23SBarry Smith           v4   = a->a + ii[row+3];
33115850ef23SBarry Smith           sum1 = b[row];
33125850ef23SBarry Smith           sum2 = b[row+1];
33135850ef23SBarry Smith           sum3 = b[row+2];
33145850ef23SBarry Smith           sum4 = b[row+3];
33155850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
33165850ef23SBarry Smith             i1    = idx[0];
33175850ef23SBarry Smith             i2    = idx[1];
33185850ef23SBarry Smith             idx  += 2;
33195850ef23SBarry Smith             tmp0  = x[i1];
33205850ef23SBarry Smith             tmp1  = x[i2];
33215850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
33225850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
33235850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
33245850ef23SBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
33255850ef23SBarry Smith           }
3326d876e2b0SMark Adams           if (n == sz-1) {
3327d876e2b0SMark Adams             tmp0  = x[*idx++];
3328d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3329d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3330d876e2b0SMark Adams             sum3 -= v3[0] * tmp0;
3331d876e2b0SMark Adams             sum4 -= v4[0] * tmp0;
3332d876e2b0SMark Adams             v1++; v2++; v3++; v4++;
3333d876e2b0SMark Adams           }
3334d876e2b0SMark Adams           t[row]   = sum1;
3335d876e2b0SMark Adams           t[row+1] = sum2;
3336d876e2b0SMark Adams           t[row+2] = sum3;
3337d876e2b0SMark Adams           t[row+3] = sum4;
3338d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 4;
3339d876e2b0SMark Adams           idx     = a->j + diag[row] + 4;
3340d876e2b0SMark Adams           v1 += 4;
3341d876e2b0SMark Adams           v2 += 4;
3342d876e2b0SMark Adams           v3 += 4;
3343d876e2b0SMark Adams           v4 += 4;
3344d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3345d876e2b0SMark Adams             i1    = idx[0];
3346d876e2b0SMark Adams             i2    = idx[1];
3347d876e2b0SMark Adams             idx  += 2;
3348d876e2b0SMark Adams             tmp0  = x[i1];
3349d876e2b0SMark Adams             tmp1  = x[i2];
3350d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3351d876e2b0SMark Adams             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3352d876e2b0SMark Adams             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3353d876e2b0SMark Adams             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3354d876e2b0SMark Adams           }
33555850ef23SBarry Smith           if (n == sz-1) {
33565850ef23SBarry Smith             tmp0  = x[*idx];
33575850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
33585850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
33595850ef23SBarry Smith             sum3 -= v3[0] * tmp0;
33605850ef23SBarry Smith             sum4 -= v4[0] * tmp0;
33615850ef23SBarry Smith           }
3362d876e2b0SMark Adams           x[row] =   sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
3363d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
3364d876e2b0SMark Adams           x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
3365d876e2b0SMark Adams           x[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
33665850ef23SBarry Smith           break;
33675850ef23SBarry Smith         case 5:
33685850ef23SBarry Smith           v2   = a->a + ii[row+1];
33695850ef23SBarry Smith           v3   = a->a + ii[row+2];
33705850ef23SBarry Smith           v4   = a->a + ii[row+3];
33715850ef23SBarry Smith           v5   = a->a + ii[row+4];
33725850ef23SBarry Smith           sum1 = b[row];
33735850ef23SBarry Smith           sum2 = b[row+1];
33745850ef23SBarry Smith           sum3 = b[row+2];
33755850ef23SBarry Smith           sum4 = b[row+3];
33765850ef23SBarry Smith           sum5 = b[row+4];
33775850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
33785850ef23SBarry Smith             i1    = idx[0];
33795850ef23SBarry Smith             i2    = idx[1];
33805850ef23SBarry Smith             idx  += 2;
33815850ef23SBarry Smith             tmp0  = x[i1];
33825850ef23SBarry Smith             tmp1  = x[i2];
33835850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
33845850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
33855850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
33865850ef23SBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
33875850ef23SBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
33885850ef23SBarry Smith           }
33895850ef23SBarry Smith           if (n == sz-1) {
3390d876e2b0SMark Adams             tmp0  = x[*idx++];
33915850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
33925850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
33935850ef23SBarry Smith             sum3 -= v3[0] * tmp0;
33945850ef23SBarry Smith             sum4 -= v4[0] * tmp0;
33955850ef23SBarry Smith             sum5 -= v5[0] * tmp0;
3396d876e2b0SMark Adams             v1++; v2++; v3++; v4++; v5++;
33975850ef23SBarry Smith           }
3398d876e2b0SMark Adams           t[row]   = sum1;
3399d876e2b0SMark Adams           t[row+1] = sum2;
3400d876e2b0SMark Adams           t[row+2] = sum3;
3401d876e2b0SMark Adams           t[row+3] = sum4;
3402d876e2b0SMark Adams           t[row+4] = sum5;
3403d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 5;
3404d876e2b0SMark Adams           idx     = a->j + diag[row] + 5;
3405d876e2b0SMark Adams           v1 += 5;
3406d876e2b0SMark Adams           v2 += 5;
3407d876e2b0SMark Adams           v3 += 5;
3408d876e2b0SMark Adams           v4 += 5;
3409d876e2b0SMark Adams           v5 += 5;
34105850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
34115850ef23SBarry Smith             i1    = idx[0];
34125850ef23SBarry Smith             i2    = idx[1];
34135850ef23SBarry Smith             idx  += 2;
34145850ef23SBarry Smith             tmp0  = x[i1];
34155850ef23SBarry Smith             tmp1  = x[i2];
34165850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
34175850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
34185850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
34195850ef23SBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
34205850ef23SBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
34215850ef23SBarry Smith           }
34225850ef23SBarry Smith           if (n == sz-1) {
34235850ef23SBarry Smith             tmp0  = x[*idx];
3424d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3425d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3426d876e2b0SMark Adams             sum3 -= v3[0] * tmp0;
3427d876e2b0SMark Adams             sum4 -= v4[0] * tmp0;
3428d876e2b0SMark Adams             sum5 -= v5[0] * tmp0;
34295850ef23SBarry Smith           }
3430d876e2b0SMark Adams           x[row]   = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
3431d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
3432d876e2b0SMark Adams           x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
3433d876e2b0SMark Adams           x[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
3434d876e2b0SMark Adams           x[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
3435d876e2b0SMark Adams           break;
3436d876e2b0SMark Adams         default:
3437d876e2b0SMark Adams           SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
3438d876e2b0SMark Adams         }
3439d876e2b0SMark Adams       }
3440d876e2b0SMark Adams       xb   = t;
3441d876e2b0SMark Adams       ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);  /* undercounts diag inverse */
3442d876e2b0SMark Adams     } else xb = b;
3443d876e2b0SMark Adams 
3444d876e2b0SMark Adams     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) {
3445d876e2b0SMark Adams 
3446d876e2b0SMark Adams       ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
3447d876e2b0SMark Adams       for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
3448d876e2b0SMark Adams         ibdiag -= sizes[i]*sizes[i];
3449d876e2b0SMark Adams 
3450d876e2b0SMark Adams         /* set RHS */
3451d876e2b0SMark Adams         if (xb == b) {
3452d876e2b0SMark Adams           /* whole (old way) */
3453d876e2b0SMark Adams           sz      = ii[row+1] - ii[row];
3454d876e2b0SMark Adams           idx     = a->j + ii[row];
3455d876e2b0SMark Adams           switch (sizes[i]) {
3456d876e2b0SMark Adams           case 5:
3457d876e2b0SMark Adams             v5      = a->a + ii[row-4];
3458d876e2b0SMark Adams           case 4: /* fall through */
3459d876e2b0SMark Adams             v4      = a->a + ii[row-3];
3460d876e2b0SMark Adams           case 3:
3461d876e2b0SMark Adams             v3      = a->a + ii[row-2];
3462d876e2b0SMark Adams           case 2:
3463d876e2b0SMark Adams             v2      = a->a + ii[row-1];
3464d876e2b0SMark Adams           case 1:
3465d876e2b0SMark Adams             v1      = a->a + ii[row];
3466d876e2b0SMark Adams             break;
3467d876e2b0SMark Adams           default:
3468d876e2b0SMark Adams             SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
3469d876e2b0SMark Adams           }
3470d876e2b0SMark Adams         } else {
3471d876e2b0SMark Adams           /* upper, no diag */
3472d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 1;
3473d876e2b0SMark Adams           idx     = a->j + diag[row] + 1;
3474d876e2b0SMark Adams           switch (sizes[i]) {
3475d876e2b0SMark Adams           case 5:
3476d876e2b0SMark Adams             v5      = a->a + diag[row-4] + 5;
3477d876e2b0SMark Adams           case 4: /* fall through */
3478d876e2b0SMark Adams             v4      = a->a + diag[row-3] + 4;
3479d876e2b0SMark Adams           case 3:
3480d876e2b0SMark Adams             v3      = a->a + diag[row-2] + 3;
3481d876e2b0SMark Adams           case 2:
3482d876e2b0SMark Adams             v2      = a->a + diag[row-1] + 2;
3483d876e2b0SMark Adams           case 1:
3484d876e2b0SMark Adams             v1      = a->a + diag[row] + 1;
3485d876e2b0SMark Adams           }
3486d876e2b0SMark Adams         }
3487d876e2b0SMark Adams         /* set sum */
3488d876e2b0SMark Adams         switch (sizes[i]) {
3489d876e2b0SMark Adams         case 5:
3490d876e2b0SMark Adams           sum5 = xb[row-4];
3491d876e2b0SMark Adams         case 4: /* fall through */
3492d876e2b0SMark Adams           sum4 = xb[row-3];
3493d876e2b0SMark Adams         case 3:
3494d876e2b0SMark Adams           sum3 = xb[row-2];
3495d876e2b0SMark Adams         case 2:
3496d876e2b0SMark Adams           sum2 = xb[row-1];
3497d876e2b0SMark Adams         case 1:
3498d876e2b0SMark Adams           /* note that sum1 is associated with the last row */
3499d876e2b0SMark Adams           sum1 = xb[row];
3500d876e2b0SMark Adams         }
3501d876e2b0SMark Adams         /* do sums */
3502d876e2b0SMark Adams         for (n = 0; n<sz-1; n+=2) {
3503d876e2b0SMark Adams             i1    = idx[0];
3504d876e2b0SMark Adams             i2    = idx[1];
3505d876e2b0SMark Adams             idx  += 2;
3506d876e2b0SMark Adams             tmp0  = x[i1];
3507d876e2b0SMark Adams             tmp1  = x[i2];
3508d876e2b0SMark Adams             switch (sizes[i]) {
3509d876e2b0SMark Adams             case 5:
3510d876e2b0SMark Adams               sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
3511d876e2b0SMark Adams             case 4: /* fall through */
3512d876e2b0SMark Adams               sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3513d876e2b0SMark Adams             case 3:
3514d876e2b0SMark Adams               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3515d876e2b0SMark Adams             case 2:
3516d876e2b0SMark Adams               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3517d876e2b0SMark Adams             case 1:
3518d876e2b0SMark Adams               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3519d876e2b0SMark Adams             }
3520d876e2b0SMark Adams         }
3521d876e2b0SMark Adams         /* ragged edge */
3522d876e2b0SMark Adams         if (n == sz-1) {
3523d876e2b0SMark Adams           tmp0  = x[*idx];
3524d876e2b0SMark Adams           switch (sizes[i]) {
3525d876e2b0SMark Adams           case 5:
3526d876e2b0SMark Adams             sum5 -= *v5*tmp0;
3527d876e2b0SMark Adams           case 4: /* fall through */
3528d876e2b0SMark Adams             sum4 -= *v4*tmp0;
3529d876e2b0SMark Adams           case 3:
3530d876e2b0SMark Adams             sum3 -= *v3*tmp0;
3531d876e2b0SMark Adams           case 2:
3532d876e2b0SMark Adams             sum2 -= *v2*tmp0;
3533d876e2b0SMark Adams           case 1:
3534d876e2b0SMark Adams             sum1 -= *v1*tmp0;
3535d876e2b0SMark Adams           }
3536d876e2b0SMark Adams         }
3537d876e2b0SMark Adams         /* update */
3538d876e2b0SMark Adams         if (xb == b) {
3539d876e2b0SMark Adams           /* whole (old way) w/ diag */
3540d876e2b0SMark Adams           switch (sizes[i]) {
3541d876e2b0SMark Adams           case 5:
35425850ef23SBarry Smith             x[row--] += sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
35435850ef23SBarry Smith             x[row--] += sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
35445850ef23SBarry Smith             x[row--] += sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
35455850ef23SBarry Smith             x[row--] += sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
35465850ef23SBarry Smith             x[row--] += sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
35475850ef23SBarry Smith             break;
3548d876e2b0SMark Adams           case 4:
3549d876e2b0SMark Adams             x[row--] += sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3550d876e2b0SMark Adams             x[row--] += sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3551d876e2b0SMark Adams             x[row--] += sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3552d876e2b0SMark Adams             x[row--] += sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3553d876e2b0SMark Adams             break;
3554d876e2b0SMark Adams           case 3:
3555d876e2b0SMark Adams             x[row--] += sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3556d876e2b0SMark Adams             x[row--] += sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3557d876e2b0SMark Adams             x[row--] += sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3558d876e2b0SMark Adams             break;
3559d876e2b0SMark Adams           case 2:
3560d876e2b0SMark Adams             x[row--] += sum2*ibdiag[1] + sum1*ibdiag[3];
3561d876e2b0SMark Adams             x[row--] += sum2*ibdiag[0] + sum1*ibdiag[2];
3562d876e2b0SMark Adams             break;
3563d876e2b0SMark Adams           case 1:
3564d876e2b0SMark Adams             x[row--] += sum1*(*ibdiag);
3565d876e2b0SMark Adams             break;
3566d876e2b0SMark Adams           }
3567d876e2b0SMark Adams         } else {
3568d876e2b0SMark Adams           /* no diag so set =  */
3569d876e2b0SMark Adams           switch (sizes[i]) {
3570d876e2b0SMark Adams           case 5:
3571d876e2b0SMark Adams             x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
3572d876e2b0SMark Adams             x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
3573d876e2b0SMark Adams             x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
3574d876e2b0SMark Adams             x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
3575d876e2b0SMark Adams             x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
3576d876e2b0SMark Adams             break;
3577d876e2b0SMark Adams           case 4:
3578d876e2b0SMark Adams             x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3579d876e2b0SMark Adams             x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3580d876e2b0SMark Adams             x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3581d876e2b0SMark Adams             x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3582d876e2b0SMark Adams             break;
3583d876e2b0SMark Adams           case 3:
3584d876e2b0SMark Adams             x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3585d876e2b0SMark Adams             x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3586d876e2b0SMark Adams             x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3587d876e2b0SMark Adams             break;
3588d876e2b0SMark Adams           case 2:
3589d876e2b0SMark Adams             x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3];
3590d876e2b0SMark Adams             x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2];
3591d876e2b0SMark Adams             break;
3592d876e2b0SMark Adams           case 1:
3593d876e2b0SMark Adams             x[row--] = sum1*(*ibdiag);
3594d876e2b0SMark Adams             break;
35955850ef23SBarry Smith           }
35965850ef23SBarry Smith         }
3597d876e2b0SMark Adams       }
3598d876e2b0SMark Adams       if (xb == b) {
35995850ef23SBarry Smith         ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
3600d876e2b0SMark Adams       } else {
3601d876e2b0SMark Adams         ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper, undercounts diag inverse */
3602d876e2b0SMark Adams       }
36035850ef23SBarry Smith     }
36042af78befSBarry Smith   }
360589c6957cSBarry Smith   if (flag & SOR_EISENSTAT) {
360689c6957cSBarry Smith     /*
360789c6957cSBarry Smith           Apply  (U + D)^-1  where D is now the block diagonal
360889c6957cSBarry Smith     */
360989c6957cSBarry Smith     ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
361089c6957cSBarry Smith     for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
361189c6957cSBarry Smith       ibdiag -= sizes[i]*sizes[i];
361289c6957cSBarry Smith       sz      = ii[row+1] - diag[row] - 1;
361389c6957cSBarry Smith       v1      = a->a + diag[row] + 1;
361489c6957cSBarry Smith       idx     = a->j + diag[row] + 1;
36154108e4d5SBarry Smith       /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
361689c6957cSBarry Smith       switch (sizes[i]) {
361789c6957cSBarry Smith       case 1:
361889c6957cSBarry Smith 
361989c6957cSBarry Smith         sum1 = b[row];
362089c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
362189c6957cSBarry Smith           i1    = idx[0];
362289c6957cSBarry Smith           i2    = idx[1];
362389c6957cSBarry Smith           idx  += 2;
362489c6957cSBarry Smith           tmp0  = x[i1];
362589c6957cSBarry Smith           tmp1  = x[i2];
362689c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
362789c6957cSBarry Smith         }
362889c6957cSBarry Smith 
362989c6957cSBarry Smith         if (n == sz-1) {
363089c6957cSBarry Smith           tmp0  = x[*idx];
363189c6957cSBarry Smith           sum1 -= *v1*tmp0;
363289c6957cSBarry Smith         }
3633938d4eb3SBarry Smith         x[row] = sum1*(*ibdiag);row--;
363489c6957cSBarry Smith         break;
363589c6957cSBarry Smith 
363689c6957cSBarry Smith       case 2:
363789c6957cSBarry Smith 
363889c6957cSBarry Smith         sum1 = b[row];
363989c6957cSBarry Smith         sum2 = b[row-1];
364089c6957cSBarry Smith         /* note that sum1 is associated with the second of the two rows */
364189c6957cSBarry Smith         v2 = a->a + diag[row-1] + 2;
364289c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
364389c6957cSBarry Smith           i1    = idx[0];
364489c6957cSBarry Smith           i2    = idx[1];
364589c6957cSBarry Smith           idx  += 2;
364689c6957cSBarry Smith           tmp0  = x[i1];
364789c6957cSBarry Smith           tmp1  = x[i2];
364889c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
364989c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
365089c6957cSBarry Smith         }
365189c6957cSBarry Smith 
365289c6957cSBarry Smith         if (n == sz-1) {
365389c6957cSBarry Smith           tmp0  = x[*idx];
365489c6957cSBarry Smith           sum1 -= *v1*tmp0;
365589c6957cSBarry Smith           sum2 -= *v2*tmp0;
365689c6957cSBarry Smith         }
3657938d4eb3SBarry Smith         x[row]   = sum2*ibdiag[1] + sum1*ibdiag[3];
3658938d4eb3SBarry Smith         x[row-1] = sum2*ibdiag[0] + sum1*ibdiag[2];
3659938d4eb3SBarry Smith         row     -= 2;
366089c6957cSBarry Smith         break;
366189c6957cSBarry Smith       case 3:
366289c6957cSBarry Smith 
366389c6957cSBarry Smith         sum1 = b[row];
366489c6957cSBarry Smith         sum2 = b[row-1];
366589c6957cSBarry Smith         sum3 = b[row-2];
366689c6957cSBarry Smith         v2   = a->a + diag[row-1] + 2;
366789c6957cSBarry Smith         v3   = a->a + diag[row-2] + 3;
366889c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
366989c6957cSBarry Smith           i1    = idx[0];
367089c6957cSBarry Smith           i2    = idx[1];
367189c6957cSBarry Smith           idx  += 2;
367289c6957cSBarry Smith           tmp0  = x[i1];
367389c6957cSBarry Smith           tmp1  = x[i2];
367489c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
367589c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
367689c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
367789c6957cSBarry Smith         }
367889c6957cSBarry Smith 
367989c6957cSBarry Smith         if (n == sz-1) {
368089c6957cSBarry Smith           tmp0  = x[*idx];
368189c6957cSBarry Smith           sum1 -= *v1*tmp0;
368289c6957cSBarry Smith           sum2 -= *v2*tmp0;
368389c6957cSBarry Smith           sum3 -= *v3*tmp0;
368489c6957cSBarry Smith         }
3685938d4eb3SBarry Smith         x[row]   = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3686938d4eb3SBarry Smith         x[row-1] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3687938d4eb3SBarry Smith         x[row-2] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3688938d4eb3SBarry Smith         row     -= 3;
368989c6957cSBarry Smith         break;
369089c6957cSBarry Smith       case 4:
369189c6957cSBarry Smith 
369289c6957cSBarry Smith         sum1 = b[row];
369389c6957cSBarry Smith         sum2 = b[row-1];
369489c6957cSBarry Smith         sum3 = b[row-2];
369589c6957cSBarry Smith         sum4 = b[row-3];
369689c6957cSBarry Smith         v2   = a->a + diag[row-1] + 2;
369789c6957cSBarry Smith         v3   = a->a + diag[row-2] + 3;
369889c6957cSBarry Smith         v4   = a->a + diag[row-3] + 4;
369989c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
370089c6957cSBarry Smith           i1    = idx[0];
370189c6957cSBarry Smith           i2    = idx[1];
370289c6957cSBarry Smith           idx  += 2;
370389c6957cSBarry Smith           tmp0  = x[i1];
370489c6957cSBarry Smith           tmp1  = x[i2];
370589c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
370689c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
370789c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
370889c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
370989c6957cSBarry Smith         }
371089c6957cSBarry Smith 
371189c6957cSBarry Smith         if (n == sz-1) {
371289c6957cSBarry Smith           tmp0  = x[*idx];
371389c6957cSBarry Smith           sum1 -= *v1*tmp0;
371489c6957cSBarry Smith           sum2 -= *v2*tmp0;
371589c6957cSBarry Smith           sum3 -= *v3*tmp0;
371689c6957cSBarry Smith           sum4 -= *v4*tmp0;
371789c6957cSBarry Smith         }
3718938d4eb3SBarry Smith         x[row]   = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3719938d4eb3SBarry Smith         x[row-1] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3720938d4eb3SBarry Smith         x[row-2] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3721938d4eb3SBarry Smith         x[row-3] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3722938d4eb3SBarry Smith         row     -= 4;
372389c6957cSBarry Smith         break;
372489c6957cSBarry Smith       case 5:
372589c6957cSBarry Smith 
372689c6957cSBarry Smith         sum1 = b[row];
372789c6957cSBarry Smith         sum2 = b[row-1];
372889c6957cSBarry Smith         sum3 = b[row-2];
372989c6957cSBarry Smith         sum4 = b[row-3];
373089c6957cSBarry Smith         sum5 = b[row-4];
373189c6957cSBarry Smith         v2   = a->a + diag[row-1] + 2;
373289c6957cSBarry Smith         v3   = a->a + diag[row-2] + 3;
373389c6957cSBarry Smith         v4   = a->a + diag[row-3] + 4;
373489c6957cSBarry Smith         v5   = a->a + diag[row-4] + 5;
373589c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
373689c6957cSBarry Smith           i1    = idx[0];
373789c6957cSBarry Smith           i2    = idx[1];
373889c6957cSBarry Smith           idx  += 2;
373989c6957cSBarry Smith           tmp0  = x[i1];
374089c6957cSBarry Smith           tmp1  = x[i2];
374189c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
374289c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
374389c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
374489c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
374589c6957cSBarry Smith           sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
374689c6957cSBarry Smith         }
374789c6957cSBarry Smith 
374889c6957cSBarry Smith         if (n == sz-1) {
374989c6957cSBarry Smith           tmp0  = x[*idx];
375089c6957cSBarry Smith           sum1 -= *v1*tmp0;
375189c6957cSBarry Smith           sum2 -= *v2*tmp0;
375289c6957cSBarry Smith           sum3 -= *v3*tmp0;
375389c6957cSBarry Smith           sum4 -= *v4*tmp0;
375489c6957cSBarry Smith           sum5 -= *v5*tmp0;
375589c6957cSBarry Smith         }
3756938d4eb3SBarry Smith         x[row]   = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
3757938d4eb3SBarry Smith         x[row-1] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
3758938d4eb3SBarry Smith         x[row-2] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
3759938d4eb3SBarry Smith         x[row-3] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
3760938d4eb3SBarry Smith         x[row-4] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
3761938d4eb3SBarry Smith         row     -= 5;
376289c6957cSBarry Smith         break;
376389c6957cSBarry Smith       default:
3764e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
376589c6957cSBarry Smith       }
376689c6957cSBarry Smith     }
376789c6957cSBarry Smith     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
376889c6957cSBarry Smith 
376989c6957cSBarry Smith     /*
377089c6957cSBarry Smith            t = b - D x    where D is the block diagonal
377189c6957cSBarry Smith     */
377289c6957cSBarry Smith     cnt = 0;
377389c6957cSBarry Smith     for (i=0, row=0; i<m; i++) {
377489c6957cSBarry Smith       switch (sizes[i]) {
377589c6957cSBarry Smith       case 1:
377689c6957cSBarry Smith         t[row] = b[row] - bdiag[cnt++]*x[row]; row++;
377789c6957cSBarry Smith         break;
377889c6957cSBarry Smith       case 2:
377989c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1];
378089c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+2];
378189c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+3];
378289c6957cSBarry Smith         t[row]   = b[row] - tmp1;
378389c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2; row += 2;
378489c6957cSBarry Smith         cnt     += 4;
378589c6957cSBarry Smith         break;
378689c6957cSBarry Smith       case 3:
378789c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1]; x3 = x[row+2];
378889c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6];
378989c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7];
379089c6957cSBarry Smith         tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8];
379189c6957cSBarry Smith         t[row]   = b[row] - tmp1;
379289c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2;
379389c6957cSBarry Smith         t[row+2] = b[row+2] - tmp3; row += 3;
379489c6957cSBarry Smith         cnt     += 9;
379589c6957cSBarry Smith         break;
379689c6957cSBarry Smith       case 4:
379789c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3];
379889c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12];
379989c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13];
380089c6957cSBarry Smith         tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14];
380189c6957cSBarry Smith         tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15];
380289c6957cSBarry Smith         t[row]   = b[row] - tmp1;
380389c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2;
380489c6957cSBarry Smith         t[row+2] = b[row+2] - tmp3;
380589c6957cSBarry Smith         t[row+3] = b[row+3] - tmp4; row += 4;
380689c6957cSBarry Smith         cnt     += 16;
380789c6957cSBarry Smith         break;
380889c6957cSBarry Smith       case 5:
380989c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; x5 = x[row+4];
381089c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20];
381189c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21];
381289c6957cSBarry Smith         tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22];
381389c6957cSBarry Smith         tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23];
381489c6957cSBarry Smith         tmp5     = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24];
381589c6957cSBarry Smith         t[row]   = b[row] - tmp1;
381689c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2;
381789c6957cSBarry Smith         t[row+2] = b[row+2] - tmp3;
381889c6957cSBarry Smith         t[row+3] = b[row+3] - tmp4;
381989c6957cSBarry Smith         t[row+4] = b[row+4] - tmp5;row += 5;
382089c6957cSBarry Smith         cnt     += 25;
382189c6957cSBarry Smith         break;
382289c6957cSBarry Smith       default:
3823e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
382489c6957cSBarry Smith       }
382589c6957cSBarry Smith     }
382689c6957cSBarry Smith     ierr = PetscLogFlops(m);CHKERRQ(ierr);
382789c6957cSBarry Smith 
382889c6957cSBarry Smith 
382989c6957cSBarry Smith 
383089c6957cSBarry Smith     /*
383189c6957cSBarry Smith           Apply (L + D)^-1 where D is the block diagonal
383289c6957cSBarry Smith     */
383389c6957cSBarry Smith     for (i=0, row=0; i<m; i++) {
383489c6957cSBarry Smith       sz  = diag[row] - ii[row];
383589c6957cSBarry Smith       v1  = a->a + ii[row];
383689c6957cSBarry Smith       idx = a->j + ii[row];
38374108e4d5SBarry Smith       /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
383889c6957cSBarry Smith       switch (sizes[i]) {
383989c6957cSBarry Smith       case 1:
384089c6957cSBarry Smith 
384189c6957cSBarry Smith         sum1 = t[row];
384289c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
384389c6957cSBarry Smith           i1    = idx[0];
384489c6957cSBarry Smith           i2    = idx[1];
384589c6957cSBarry Smith           idx  += 2;
384689c6957cSBarry Smith           tmp0  = t[i1];
384789c6957cSBarry Smith           tmp1  = t[i2];
384889c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
384989c6957cSBarry Smith         }
385089c6957cSBarry Smith 
385189c6957cSBarry Smith         if (n == sz-1) {
385289c6957cSBarry Smith           tmp0  = t[*idx];
385389c6957cSBarry Smith           sum1 -= *v1 * tmp0;
385489c6957cSBarry Smith         }
385589c6957cSBarry Smith         x[row] += t[row] = sum1*(*ibdiag++); row++;
385689c6957cSBarry Smith         break;
385789c6957cSBarry Smith       case 2:
385889c6957cSBarry Smith         v2   = a->a + ii[row+1];
385989c6957cSBarry Smith         sum1 = t[row];
386089c6957cSBarry Smith         sum2 = t[row+1];
386189c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
386289c6957cSBarry Smith           i1    = idx[0];
386389c6957cSBarry Smith           i2    = idx[1];
386489c6957cSBarry Smith           idx  += 2;
386589c6957cSBarry Smith           tmp0  = t[i1];
386689c6957cSBarry Smith           tmp1  = t[i2];
386789c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
386889c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
386989c6957cSBarry Smith         }
387089c6957cSBarry Smith 
387189c6957cSBarry Smith         if (n == sz-1) {
387289c6957cSBarry Smith           tmp0  = t[*idx];
387389c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
387489c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
387589c6957cSBarry Smith         }
387689c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[2];
387789c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3];
387889c6957cSBarry Smith         ibdiag   += 4; row += 2;
387989c6957cSBarry Smith         break;
388089c6957cSBarry Smith       case 3:
388189c6957cSBarry Smith         v2   = a->a + ii[row+1];
388289c6957cSBarry Smith         v3   = a->a + ii[row+2];
388389c6957cSBarry Smith         sum1 = t[row];
388489c6957cSBarry Smith         sum2 = t[row+1];
388589c6957cSBarry Smith         sum3 = t[row+2];
388689c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
388789c6957cSBarry Smith           i1    = idx[0];
388889c6957cSBarry Smith           i2    = idx[1];
388989c6957cSBarry Smith           idx  += 2;
389089c6957cSBarry Smith           tmp0  = t[i1];
389189c6957cSBarry Smith           tmp1  = t[i2];
389289c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
389389c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
389489c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
389589c6957cSBarry Smith         }
389689c6957cSBarry Smith 
389789c6957cSBarry Smith         if (n == sz-1) {
389889c6957cSBarry Smith           tmp0  = t[*idx];
389989c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
390089c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
390189c6957cSBarry Smith           sum3 -= v3[0] * tmp0;
390289c6957cSBarry Smith         }
390389c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
390489c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
390589c6957cSBarry Smith         x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
390689c6957cSBarry Smith         ibdiag   += 9; row += 3;
390789c6957cSBarry Smith         break;
390889c6957cSBarry Smith       case 4:
390989c6957cSBarry Smith         v2   = a->a + ii[row+1];
391089c6957cSBarry Smith         v3   = a->a + ii[row+2];
391189c6957cSBarry Smith         v4   = a->a + ii[row+3];
391289c6957cSBarry Smith         sum1 = t[row];
391389c6957cSBarry Smith         sum2 = t[row+1];
391489c6957cSBarry Smith         sum3 = t[row+2];
391589c6957cSBarry Smith         sum4 = t[row+3];
391689c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
391789c6957cSBarry Smith           i1    = idx[0];
391889c6957cSBarry Smith           i2    = idx[1];
391989c6957cSBarry Smith           idx  += 2;
392089c6957cSBarry Smith           tmp0  = t[i1];
392189c6957cSBarry Smith           tmp1  = t[i2];
392289c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
392389c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
392489c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
392589c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
392689c6957cSBarry Smith         }
392789c6957cSBarry Smith 
392889c6957cSBarry Smith         if (n == sz-1) {
392989c6957cSBarry Smith           tmp0  = t[*idx];
393089c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
393189c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
393289c6957cSBarry Smith           sum3 -= v3[0] * tmp0;
393389c6957cSBarry Smith           sum4 -= v4[0] * tmp0;
393489c6957cSBarry Smith         }
393589c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
393689c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
393789c6957cSBarry Smith         x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
393889c6957cSBarry Smith         x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
393989c6957cSBarry Smith         ibdiag   += 16; row += 4;
394089c6957cSBarry Smith         break;
394189c6957cSBarry Smith       case 5:
394289c6957cSBarry Smith         v2   = a->a + ii[row+1];
394389c6957cSBarry Smith         v3   = a->a + ii[row+2];
394489c6957cSBarry Smith         v4   = a->a + ii[row+3];
394589c6957cSBarry Smith         v5   = a->a + ii[row+4];
394689c6957cSBarry Smith         sum1 = t[row];
394789c6957cSBarry Smith         sum2 = t[row+1];
394889c6957cSBarry Smith         sum3 = t[row+2];
394989c6957cSBarry Smith         sum4 = t[row+3];
395089c6957cSBarry Smith         sum5 = t[row+4];
395189c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
395289c6957cSBarry Smith           i1    = idx[0];
395389c6957cSBarry Smith           i2    = idx[1];
395489c6957cSBarry Smith           idx  += 2;
395589c6957cSBarry Smith           tmp0  = t[i1];
395689c6957cSBarry Smith           tmp1  = t[i2];
395789c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
395889c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
395989c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
396089c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
396189c6957cSBarry Smith           sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
396289c6957cSBarry Smith         }
396389c6957cSBarry Smith 
396489c6957cSBarry Smith         if (n == sz-1) {
396589c6957cSBarry Smith           tmp0  = t[*idx];
396689c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
396789c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
396889c6957cSBarry Smith           sum3 -= v3[0] * tmp0;
396989c6957cSBarry Smith           sum4 -= v4[0] * tmp0;
397089c6957cSBarry Smith           sum5 -= v5[0] * tmp0;
397189c6957cSBarry Smith         }
397289c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
397389c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
397489c6957cSBarry Smith         x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
397589c6957cSBarry Smith         x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
397689c6957cSBarry Smith         x[row+4] += t[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
397789c6957cSBarry Smith         ibdiag   += 25; row += 5;
397889c6957cSBarry Smith         break;
397989c6957cSBarry Smith       default:
3980e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
398189c6957cSBarry Smith       }
398289c6957cSBarry Smith     }
398389c6957cSBarry Smith     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
39845850ef23SBarry Smith   }
39852af78befSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
39863649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
3987365a8a9eSBarry Smith   PetscFunctionReturn(0);
39882af78befSBarry Smith }
39892af78befSBarry Smith 
399089c6957cSBarry Smith #undef __FUNCT__
39914108e4d5SBarry Smith #define __FUNCT__ "MatMultDiagonalBlock_SeqAIJ_Inode"
39924108e4d5SBarry Smith PetscErrorCode MatMultDiagonalBlock_SeqAIJ_Inode(Mat A,Vec bb,Vec xx)
399389c6957cSBarry Smith {
399489c6957cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
399589c6957cSBarry Smith   PetscScalar       *x,tmp1,tmp2,tmp3,tmp4,tmp5,x1,x2,x3,x4,x5;
399689c6957cSBarry Smith   const MatScalar   *bdiag = a->inode.bdiag;
399789c6957cSBarry Smith   const PetscScalar *b;
399889c6957cSBarry Smith   PetscErrorCode    ierr;
399989c6957cSBarry Smith   PetscInt          m      = a->inode.node_count,cnt = 0,i,row;
400089c6957cSBarry Smith   const PetscInt    *sizes = a->inode.size;
40012af78befSBarry Smith 
400289c6957cSBarry Smith   PetscFunctionBegin;
400389c6957cSBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
40043649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
400589c6957cSBarry Smith   cnt  = 0;
400689c6957cSBarry Smith   for (i=0, row=0; i<m; i++) {
400789c6957cSBarry Smith     switch (sizes[i]) {
400889c6957cSBarry Smith     case 1:
400989c6957cSBarry Smith       x[row] = b[row]*bdiag[cnt++];row++;
401089c6957cSBarry Smith       break;
401189c6957cSBarry Smith     case 2:
401289c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1];
401389c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+2];
401489c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+3];
401589c6957cSBarry Smith       x[row++] = tmp1;
401689c6957cSBarry Smith       x[row++] = tmp2;
401789c6957cSBarry Smith       cnt     += 4;
401889c6957cSBarry Smith       break;
401989c6957cSBarry Smith     case 3:
402089c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1]; x3 = b[row+2];
402189c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6];
402289c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7];
402389c6957cSBarry Smith       tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8];
402489c6957cSBarry Smith       x[row++] = tmp1;
402589c6957cSBarry Smith       x[row++] = tmp2;
402689c6957cSBarry Smith       x[row++] = tmp3;
402789c6957cSBarry Smith       cnt     += 9;
402889c6957cSBarry Smith       break;
402989c6957cSBarry Smith     case 4:
403089c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3];
403189c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12];
403289c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13];
403389c6957cSBarry Smith       tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14];
403489c6957cSBarry Smith       tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15];
403589c6957cSBarry Smith       x[row++] = tmp1;
403689c6957cSBarry Smith       x[row++] = tmp2;
403789c6957cSBarry Smith       x[row++] = tmp3;
403889c6957cSBarry Smith       x[row++] = tmp4;
403989c6957cSBarry Smith       cnt     += 16;
404089c6957cSBarry Smith       break;
404189c6957cSBarry Smith     case 5:
404289c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3]; x5 = b[row+4];
404389c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20];
404489c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21];
404589c6957cSBarry Smith       tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22];
404689c6957cSBarry Smith       tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23];
404789c6957cSBarry Smith       tmp5     = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24];
404889c6957cSBarry Smith       x[row++] = tmp1;
404989c6957cSBarry Smith       x[row++] = tmp2;
405089c6957cSBarry Smith       x[row++] = tmp3;
405189c6957cSBarry Smith       x[row++] = tmp4;
405289c6957cSBarry Smith       x[row++] = tmp5;
405389c6957cSBarry Smith       cnt     += 25;
405489c6957cSBarry Smith       break;
405589c6957cSBarry Smith     default:
4056e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
405789c6957cSBarry Smith     }
405889c6957cSBarry Smith   }
405989c6957cSBarry Smith   ierr = PetscLogFlops(2*cnt);CHKERRQ(ierr);
406089c6957cSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
40613649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
406289c6957cSBarry Smith   PetscFunctionReturn(0);
406389c6957cSBarry Smith }
406489c6957cSBarry Smith 
40654c1414c8SBarry Smith /*
40664c1414c8SBarry Smith     samestructure indicates that the matrix has not changed its nonzero structure so we
40674c1414c8SBarry Smith     do not need to recompute the inodes
40684c1414c8SBarry Smith */
40694c1414c8SBarry Smith #undef __FUNCT__
4070a02bda8eSBarry Smith #define __FUNCT__ "MatSeqAIJCheckInode"
4071a02bda8eSBarry Smith PetscErrorCode MatSeqAIJCheckInode(Mat A)
40724c1414c8SBarry Smith {
40734c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
40744c1414c8SBarry Smith   PetscErrorCode ierr;
40758758e1faSBarry Smith   PetscInt       i,j,m,nzx,nzy,*ns,node_count,blk_size;
4076ace3abfcSBarry Smith   PetscBool      flag;
40778758e1faSBarry Smith   const PetscInt *idx,*idy,*ii;
40784c1414c8SBarry Smith 
40794c1414c8SBarry Smith   PetscFunctionBegin;
4080d74fe821SBarry Smith   if (!a->inode.use) PetscFunctionReturn(0);
4081a02bda8eSBarry Smith   if (a->inode.checked && A->nonzerostate == a->inode.mat_nonzerostate) PetscFunctionReturn(0);
40824c1414c8SBarry Smith 
4083d0f46423SBarry Smith   m = A->rmap->n;
40842205254eSKarl Rupp   if (a->inode.size) ns = a->inode.size;
40852205254eSKarl Rupp   else {
4086854ce69bSBarry Smith     ierr = PetscMalloc1(m+1,&ns);CHKERRQ(ierr);
40872205254eSKarl Rupp   }
40884c1414c8SBarry Smith 
40894c1414c8SBarry Smith   i          = 0;
40904c1414c8SBarry Smith   node_count = 0;
40914c1414c8SBarry Smith   idx        = a->j;
40924c1414c8SBarry Smith   ii         = a->i;
40934c1414c8SBarry Smith   while (i < m) {                /* For each row */
40944c1414c8SBarry Smith     nzx = ii[i+1] - ii[i];       /* Number of nonzeros */
40954c1414c8SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
40964c1414c8SBarry Smith     for (j=i+1,idy=idx,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
40974c1414c8SBarry Smith       nzy = ii[j+1] - ii[j];     /* Same number of nonzeros */
40984c1414c8SBarry Smith       if (nzy != nzx) break;
40994c1414c8SBarry Smith       idy += nzx;              /* Same nonzero pattern */
41004c1414c8SBarry Smith       ierr = PetscMemcmp(idx,idy,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr);
41014c1414c8SBarry Smith       if (!flag) break;
41024c1414c8SBarry Smith     }
41034c1414c8SBarry Smith     ns[node_count++] = blk_size;
41044c1414c8SBarry Smith     idx             += blk_size*nzx;
41054c1414c8SBarry Smith     i                = j;
41064c1414c8SBarry Smith   }
41074c1414c8SBarry Smith   /* If not enough inodes found,, do not use inode version of the routines */
4108be6adb11SBarry Smith   if (!m || node_count > .8*m) {
41094c1414c8SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
41102205254eSKarl Rupp 
41114c1414c8SBarry Smith     a->inode.node_count       = 0;
41120298fd71SBarry Smith     a->inode.size             = NULL;
41134c1414c8SBarry Smith     a->inode.use              = PETSC_FALSE;
4114ded6e084SBarry Smith     A->ops->mult              = MatMult_SeqAIJ;
4115ded6e084SBarry Smith     A->ops->sor               = MatSOR_SeqAIJ;
4116ded6e084SBarry Smith     A->ops->multadd           = MatMultAdd_SeqAIJ;
4117ded6e084SBarry Smith     A->ops->getrowij          = MatGetRowIJ_SeqAIJ;
4118ded6e084SBarry Smith     A->ops->restorerowij      = MatRestoreRowIJ_SeqAIJ;
4119ded6e084SBarry Smith     A->ops->getcolumnij       = MatGetColumnIJ_SeqAIJ;
4120ded6e084SBarry Smith     A->ops->restorecolumnij   = MatRestoreColumnIJ_SeqAIJ;
4121ded6e084SBarry Smith     A->ops->coloringpatch     = 0;
4122ded6e084SBarry Smith     A->ops->multdiagonalblock = 0;
41232205254eSKarl Rupp 
41244c1414c8SBarry Smith     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
41254c1414c8SBarry Smith   } else {
4126d5f3da31SBarry Smith     if (!A->factortype) {
41274108e4d5SBarry Smith       A->ops->mult              = MatMult_SeqAIJ_Inode;
41284108e4d5SBarry Smith       A->ops->sor               = MatSOR_SeqAIJ_Inode;
41294108e4d5SBarry Smith       A->ops->multadd           = MatMultAdd_SeqAIJ_Inode;
4130375a6242SBarry Smith       A->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode;
4131375a6242SBarry Smith       if (A->rmap->n == A->cmap->n) {
41324108e4d5SBarry Smith         A->ops->getrowij          = MatGetRowIJ_SeqAIJ_Inode;
41334108e4d5SBarry Smith         A->ops->restorerowij      = MatRestoreRowIJ_SeqAIJ_Inode;
41344108e4d5SBarry Smith         A->ops->getcolumnij       = MatGetColumnIJ_SeqAIJ_Inode;
41354108e4d5SBarry Smith         A->ops->restorecolumnij   = MatRestoreColumnIJ_SeqAIJ_Inode;
41364108e4d5SBarry Smith         A->ops->coloringpatch     = MatColoringPatch_SeqAIJ_Inode;
4137375a6242SBarry Smith       }
4138d3ac4fa3SBarry Smith     } else {
4139d3ac4fa3SBarry Smith       A->ops->solve = MatSolve_SeqAIJ_Inode_inplace;
4140d3ac4fa3SBarry Smith     }
41414c1414c8SBarry Smith     a->inode.node_count = node_count;
41424c1414c8SBarry Smith     a->inode.size       = ns;
41434c1414c8SBarry Smith     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
41444c1414c8SBarry Smith   }
4145be6adb11SBarry Smith   a->inode.checked          = PETSC_TRUE;
4146a02bda8eSBarry Smith   a->inode.mat_nonzerostate = A->nonzerostate;
41474c1414c8SBarry Smith   PetscFunctionReturn(0);
41484c1414c8SBarry Smith }
41494c1414c8SBarry Smith 
415017454e89SShri Abhyankar #undef __FUNCT__
4151150f0143SBarry Smith #define __FUNCT__ "MatDuplicate_SeqAIJ_Inode"
4152150f0143SBarry Smith PetscErrorCode MatDuplicate_SeqAIJ_Inode(Mat A,MatDuplicateOption cpvalues,Mat *C)
4153150f0143SBarry Smith {
4154150f0143SBarry Smith   Mat            B =*C;
4155150f0143SBarry Smith   Mat_SeqAIJ     *c=(Mat_SeqAIJ*)B->data,*a=(Mat_SeqAIJ*)A->data;
4156150f0143SBarry Smith   PetscErrorCode ierr;
4157150f0143SBarry Smith   PetscInt       m=A->rmap->n;
4158150f0143SBarry Smith 
4159150f0143SBarry Smith   PetscFunctionBegin;
4160150f0143SBarry Smith   c->inode.use       = a->inode.use;
4161150f0143SBarry Smith   c->inode.limit     = a->inode.limit;
4162150f0143SBarry Smith   c->inode.max_limit = a->inode.max_limit;
4163150f0143SBarry Smith   if (a->inode.size) {
4164854ce69bSBarry Smith     ierr                = PetscMalloc1(m+1,&c->inode.size);CHKERRQ(ierr);
4165150f0143SBarry Smith     c->inode.node_count = a->inode.node_count;
4166150f0143SBarry Smith     ierr                = PetscMemcpy(c->inode.size,a->inode.size,(m+1)*sizeof(PetscInt));CHKERRQ(ierr);
4167a02bda8eSBarry Smith     /* note the table of functions below should match that in MatSeqAIJCheckInode() */
41682c451681SBarry Smith     if (!B->factortype) {
41692c451681SBarry Smith       B->ops->mult              = MatMult_SeqAIJ_Inode;
41702c451681SBarry Smith       B->ops->sor               = MatSOR_SeqAIJ_Inode;
41712c451681SBarry Smith       B->ops->multadd           = MatMultAdd_SeqAIJ_Inode;
41722c451681SBarry Smith       B->ops->getrowij          = MatGetRowIJ_SeqAIJ_Inode;
41732c451681SBarry Smith       B->ops->restorerowij      = MatRestoreRowIJ_SeqAIJ_Inode;
41742c451681SBarry Smith       B->ops->getcolumnij       = MatGetColumnIJ_SeqAIJ_Inode;
41752c451681SBarry Smith       B->ops->restorecolumnij   = MatRestoreColumnIJ_SeqAIJ_Inode;
41762c451681SBarry Smith       B->ops->coloringpatch     = MatColoringPatch_SeqAIJ_Inode;
41772c451681SBarry Smith       B->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode;
4178150f0143SBarry Smith     } else {
41792c451681SBarry Smith       B->ops->solve = MatSolve_SeqAIJ_Inode_inplace;
4180150f0143SBarry Smith     }
4181150f0143SBarry Smith   } else {
4182150f0143SBarry Smith     c->inode.size       = 0;
4183150f0143SBarry Smith     c->inode.node_count = 0;
4184150f0143SBarry Smith   }
4185150f0143SBarry Smith   c->inode.ibdiagvalid = PETSC_FALSE;
4186150f0143SBarry Smith   c->inode.ibdiag      = 0;
4187150f0143SBarry Smith   c->inode.bdiag       = 0;
4188150f0143SBarry Smith   PetscFunctionReturn(0);
4189150f0143SBarry Smith }
4190150f0143SBarry Smith 
4191150f0143SBarry Smith #undef __FUNCT__
419217454e89SShri Abhyankar #define __FUNCT__ "MatGetRow_FactoredLU"
41938758e1faSBarry Smith PETSC_STATIC_INLINE PetscErrorCode MatGetRow_FactoredLU(PetscInt *cols,PetscInt nzl,PetscInt nzu,PetscInt nz,const PetscInt *ai,const PetscInt *aj,const PetscInt *adiag,PetscInt row)
419417454e89SShri Abhyankar {
41958758e1faSBarry Smith   PetscInt       k;
41968758e1faSBarry Smith   const PetscInt *vi;
41976e111a19SKarl Rupp 
419817454e89SShri Abhyankar   PetscFunctionBegin;
419917454e89SShri Abhyankar   vi = aj + ai[row];
420017454e89SShri Abhyankar   for (k=0; k<nzl; k++) cols[k] = vi[k];
420117454e89SShri Abhyankar   vi        = aj + adiag[row];
420217454e89SShri Abhyankar   cols[nzl] = vi[0];
420317454e89SShri Abhyankar   vi        = aj + adiag[row+1]+1;
420417454e89SShri Abhyankar   for (k=0; k<nzu; k++) cols[nzl+1+k] = vi[k];
420517454e89SShri Abhyankar   PetscFunctionReturn(0);
420617454e89SShri Abhyankar }
42076936b636SHong Zhang /*
4208a02bda8eSBarry Smith    MatSeqAIJCheckInode_FactorLU - Check Inode for factored seqaij matrix.
4209a02bda8eSBarry Smith    Modified from MatSeqAIJCheckInode().
42106936b636SHong Zhang 
42116936b636SHong Zhang    Input Parameters:
4212abb87a52SBarry Smith .  Mat A - ILU or LU matrix factor
4213abb87a52SBarry Smith 
42146936b636SHong Zhang */
4215019b515eSShri Abhyankar #undef __FUNCT__
4216a02bda8eSBarry Smith #define __FUNCT__ "MatSeqAIJCheckInode_FactorLU"
4217a02bda8eSBarry Smith PetscErrorCode MatSeqAIJCheckInode_FactorLU(Mat A)
4218019b515eSShri Abhyankar {
4219019b515eSShri Abhyankar   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
4220019b515eSShri Abhyankar   PetscErrorCode ierr;
4221019b515eSShri Abhyankar   PetscInt       i,j,m,nzl1,nzu1,nzl2,nzu2,nzx,nzy,node_count,blk_size;
42228758e1faSBarry Smith   PetscInt       *cols1,*cols2,*ns;
42238758e1faSBarry Smith   const PetscInt *ai = a->i,*aj = a->j, *adiag = a->diag;
4224ace3abfcSBarry Smith   PetscBool      flag;
4225019b515eSShri Abhyankar 
4226019b515eSShri Abhyankar   PetscFunctionBegin;
4227019b515eSShri Abhyankar   if (!a->inode.use)    PetscFunctionReturn(0);
4228abb87a52SBarry Smith   if (a->inode.checked) PetscFunctionReturn(0);
4229019b515eSShri Abhyankar 
4230019b515eSShri Abhyankar   m = A->rmap->n;
42312205254eSKarl Rupp   if (a->inode.size) ns = a->inode.size;
42322205254eSKarl Rupp   else {
4233854ce69bSBarry Smith     ierr = PetscMalloc1(m+1,&ns);CHKERRQ(ierr);
42342205254eSKarl Rupp   }
4235019b515eSShri Abhyankar 
4236019b515eSShri Abhyankar   i          = 0;
4237019b515eSShri Abhyankar   node_count = 0;
4238dcca6d9dSJed Brown   ierr = PetscMalloc2(m,&cols1,m,&cols2);CHKERRQ(ierr);
4239019b515eSShri Abhyankar   while (i < m) {                /* For each row */
4240019b515eSShri Abhyankar     nzl1 = ai[i+1] - ai[i];       /* Number of nonzeros in L */
4241019b515eSShri Abhyankar     nzu1 = adiag[i] - adiag[i+1] - 1; /* Number of nonzeros in U excluding diagonal*/
4242019b515eSShri Abhyankar     nzx  = nzl1 + nzu1 + 1;
4243019b515eSShri Abhyankar     MatGetRow_FactoredLU(cols1,nzl1,nzu1,nzx,ai,aj,adiag,i);
4244019b515eSShri Abhyankar 
4245019b515eSShri Abhyankar     /* Limits the number of elements in a node to 'a->inode.limit' */
4246019b515eSShri Abhyankar     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
4247019b515eSShri Abhyankar       nzl2 = ai[j+1] - ai[j];
4248019b515eSShri Abhyankar       nzu2 = adiag[j] - adiag[j+1] - 1;
4249019b515eSShri Abhyankar       nzy  = nzl2 + nzu2 + 1;
4250019b515eSShri Abhyankar       if (nzy != nzx) break;
425117454e89SShri Abhyankar       ierr = MatGetRow_FactoredLU(cols2,nzl2,nzu2,nzy,ai,aj,adiag,j);CHKERRQ(ierr);
4252019b515eSShri Abhyankar       ierr = PetscMemcmp(cols1,cols2,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr);
42538758e1faSBarry Smith       if (!flag) break;
4254019b515eSShri Abhyankar     }
4255019b515eSShri Abhyankar     ns[node_count++] = blk_size;
4256019b515eSShri Abhyankar     i                = j;
4257019b515eSShri Abhyankar   }
42588758e1faSBarry Smith   ierr             = PetscFree2(cols1,cols2);CHKERRQ(ierr);
4259019b515eSShri Abhyankar   /* If not enough inodes found,, do not use inode version of the routines */
4260be6adb11SBarry Smith   if (!m || node_count > .8*m) {
4261019b515eSShri Abhyankar     ierr = PetscFree(ns);CHKERRQ(ierr);
42622205254eSKarl Rupp 
4263019b515eSShri Abhyankar     a->inode.node_count = 0;
42640298fd71SBarry Smith     a->inode.size       = NULL;
4265019b515eSShri Abhyankar     a->inode.use        = PETSC_FALSE;
42662205254eSKarl Rupp 
4267019b515eSShri Abhyankar     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
4268019b515eSShri Abhyankar   } else {
4269019b515eSShri Abhyankar     A->ops->mult              = 0;
4270019b515eSShri Abhyankar     A->ops->sor               = 0;
4271019b515eSShri Abhyankar     A->ops->multadd           = 0;
4272019b515eSShri Abhyankar     A->ops->getrowij          = 0;
4273019b515eSShri Abhyankar     A->ops->restorerowij      = 0;
4274019b515eSShri Abhyankar     A->ops->getcolumnij       = 0;
4275019b515eSShri Abhyankar     A->ops->restorecolumnij   = 0;
4276019b515eSShri Abhyankar     A->ops->coloringpatch     = 0;
4277019b515eSShri Abhyankar     A->ops->multdiagonalblock = 0;
4278019b515eSShri Abhyankar     a->inode.node_count       = node_count;
4279019b515eSShri Abhyankar     a->inode.size             = ns;
42802205254eSKarl Rupp 
4281019b515eSShri Abhyankar     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
4282019b515eSShri Abhyankar   }
4283be6adb11SBarry Smith   a->inode.checked = PETSC_TRUE;
4284019b515eSShri Abhyankar   PetscFunctionReturn(0);
4285019b515eSShri Abhyankar }
4286019b515eSShri Abhyankar 
4287acf2f550SJed Brown #undef __FUNCT__
4288acf2f550SJed Brown #define __FUNCT__ "MatSeqAIJInvalidateDiagonal_Inode"
4289acf2f550SJed Brown PetscErrorCode MatSeqAIJInvalidateDiagonal_Inode(Mat A)
4290acf2f550SJed Brown {
4291acf2f550SJed Brown   Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data;
4292acf2f550SJed Brown 
4293acf2f550SJed Brown   PetscFunctionBegin;
4294acf2f550SJed Brown   a->inode.ibdiagvalid = PETSC_FALSE;
4295acf2f550SJed Brown   PetscFunctionReturn(0);
4296acf2f550SJed Brown }
4297acf2f550SJed Brown 
42984c1414c8SBarry Smith /*
42994c1414c8SBarry Smith      This is really ugly. if inodes are used this replaces the
43004c1414c8SBarry Smith   permutations with ones that correspond to rows/cols of the matrix
43014c1414c8SBarry Smith   rather then inode blocks
43024c1414c8SBarry Smith */
43034c1414c8SBarry Smith #undef __FUNCT__
43044c1414c8SBarry Smith #define __FUNCT__ "MatInodeAdjustForInodes"
43057087cfbeSBarry Smith PetscErrorCode  MatInodeAdjustForInodes(Mat A,IS *rperm,IS *cperm)
43064c1414c8SBarry Smith {
43074ac538c5SBarry Smith   PetscErrorCode ierr;
43084c1414c8SBarry Smith 
43094c1414c8SBarry Smith   PetscFunctionBegin;
43104ac538c5SBarry Smith   ierr = PetscTryMethod(A,"MatInodeAdjustForInodes_C",(Mat,IS*,IS*),(A,rperm,cperm));CHKERRQ(ierr);
43114c1414c8SBarry Smith   PetscFunctionReturn(0);
43124c1414c8SBarry Smith }
43134c1414c8SBarry Smith 
43144c1414c8SBarry Smith #undef __FUNCT__
431553c77d0aSJed Brown #define __FUNCT__ "MatInodeAdjustForInodes_SeqAIJ_Inode"
43167087cfbeSBarry Smith PetscErrorCode  MatInodeAdjustForInodes_SeqAIJ_Inode(Mat A,IS *rperm,IS *cperm)
43174c1414c8SBarry Smith {
43184c1414c8SBarry Smith   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
43194c1414c8SBarry Smith   PetscErrorCode ierr;
43205d0c19d7SBarry Smith   PetscInt       m = A->rmap->n,n = A->cmap->n,i,j,nslim_row = a->inode.node_count;
43215d0c19d7SBarry Smith   const PetscInt *ridx,*cidx;
43224c1414c8SBarry Smith   PetscInt       row,col,*permr,*permc,*ns_row =  a->inode.size,*tns,start_val,end_val,indx;
43234c1414c8SBarry Smith   PetscInt       nslim_col,*ns_col;
43244c1414c8SBarry Smith   IS             ris = *rperm,cis = *cperm;
43254c1414c8SBarry Smith 
43264c1414c8SBarry Smith   PetscFunctionBegin;
43274c1414c8SBarry Smith   if (!a->inode.size) PetscFunctionReturn(0); /* no inodes so return */
43284c1414c8SBarry Smith   if (a->inode.node_count == m) PetscFunctionReturn(0); /* all inodes are of size 1 */
43294c1414c8SBarry Smith 
43304c1414c8SBarry Smith   ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr);
4331854ce69bSBarry Smith   ierr = PetscMalloc1(((nslim_row>nslim_col) ? nslim_row : nslim_col)+1,&tns);CHKERRQ(ierr);
4332dcca6d9dSJed Brown   ierr = PetscMalloc2(m,&permr,n,&permc);CHKERRQ(ierr);
43334c1414c8SBarry Smith 
43344c1414c8SBarry Smith   ierr = ISGetIndices(ris,&ridx);CHKERRQ(ierr);
43354c1414c8SBarry Smith   ierr = ISGetIndices(cis,&cidx);CHKERRQ(ierr);
43364c1414c8SBarry Smith 
43374c1414c8SBarry Smith   /* Form the inode structure for the rows of permuted matric using inv perm*/
43384c1414c8SBarry Smith   for (i=0,tns[0]=0; i<nslim_row; ++i) tns[i+1] = tns[i] + ns_row[i];
43394c1414c8SBarry Smith 
43404c1414c8SBarry Smith   /* Construct the permutations for rows*/
43414c1414c8SBarry Smith   for (i=0,row = 0; i<nslim_row; ++i) {
43424c1414c8SBarry Smith     indx      = ridx[i];
43434c1414c8SBarry Smith     start_val = tns[indx];
43444c1414c8SBarry Smith     end_val   = tns[indx + 1];
43454c1414c8SBarry Smith     for (j=start_val; j<end_val; ++j,++row) permr[row]= j;
43464c1414c8SBarry Smith   }
43474c1414c8SBarry Smith 
43484c1414c8SBarry Smith   /* Form the inode structure for the columns of permuted matrix using inv perm*/
43494c1414c8SBarry Smith   for (i=0,tns[0]=0; i<nslim_col; ++i) tns[i+1] = tns[i] + ns_col[i];
43504c1414c8SBarry Smith 
43514c1414c8SBarry Smith   /* Construct permutations for columns */
43524c1414c8SBarry Smith   for (i=0,col=0; i<nslim_col; ++i) {
43534c1414c8SBarry Smith     indx      = cidx[i];
43544c1414c8SBarry Smith     start_val = tns[indx];
43554c1414c8SBarry Smith     end_val   = tns[indx + 1];
43564c1414c8SBarry Smith     for (j = start_val; j<end_val; ++j,++col) permc[col]= j;
43574c1414c8SBarry Smith   }
43584c1414c8SBarry Smith 
435970b3c8c7SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permr,PETSC_COPY_VALUES,rperm);CHKERRQ(ierr);
43600e83c824SBarry Smith   ierr = ISSetPermutation(*rperm);CHKERRQ(ierr);
436170b3c8c7SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permc,PETSC_COPY_VALUES,cperm);CHKERRQ(ierr);
43620e83c824SBarry Smith   ierr = ISSetPermutation(*cperm);CHKERRQ(ierr);
43634c1414c8SBarry Smith 
43644c1414c8SBarry Smith   ierr = ISRestoreIndices(ris,&ridx);CHKERRQ(ierr);
43654c1414c8SBarry Smith   ierr = ISRestoreIndices(cis,&cidx);CHKERRQ(ierr);
43664c1414c8SBarry Smith 
43674c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
43680e83c824SBarry Smith   ierr = PetscFree2(permr,permc);CHKERRQ(ierr);
43696bf464f9SBarry Smith   ierr = ISDestroy(&cis);CHKERRQ(ierr);
43706bf464f9SBarry Smith   ierr = ISDestroy(&ris);CHKERRQ(ierr);
43714c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
43724c1414c8SBarry Smith   PetscFunctionReturn(0);
43734c1414c8SBarry Smith }
43744c1414c8SBarry Smith 
43754c1414c8SBarry Smith #undef __FUNCT__
43764c1414c8SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes"
43774c1414c8SBarry Smith /*@C
43784c1414c8SBarry Smith    MatInodeGetInodeSizes - Returns the inode information of the Inode matrix.
43794c1414c8SBarry Smith 
43803f9fe445SBarry Smith    Not Collective
43814c1414c8SBarry Smith 
43824c1414c8SBarry Smith    Input Parameter:
43834c1414c8SBarry Smith .  A - the Inode matrix or matrix derived from the Inode class -- e.g., SeqAIJ
43844c1414c8SBarry Smith 
43854c1414c8SBarry Smith    Output Parameter:
43864c1414c8SBarry Smith +  node_count - no of inodes present in the matrix.
43874c1414c8SBarry Smith .  sizes      - an array of size node_count,with sizes of each inode.
43884c1414c8SBarry Smith -  limit      - the max size used to generate the inodes.
43894c1414c8SBarry Smith 
43904c1414c8SBarry Smith    Level: advanced
43914c1414c8SBarry Smith 
43924c1414c8SBarry Smith    Notes: This routine returns some internal storage information
43934c1414c8SBarry Smith    of the matrix, it is intended to be used by advanced users.
43944c1414c8SBarry Smith    It should be called after the matrix is assembled.
43954c1414c8SBarry Smith    The contents of the sizes[] array should not be changed.
43960298fd71SBarry Smith    NULL may be passed for information not requested.
43974c1414c8SBarry Smith 
43984c1414c8SBarry Smith .keywords: matrix, seqaij, get, inode
43994c1414c8SBarry Smith 
44004c1414c8SBarry Smith .seealso: MatGetInfo()
44014c1414c8SBarry Smith @*/
44027087cfbeSBarry Smith PetscErrorCode  MatInodeGetInodeSizes(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit)
44034c1414c8SBarry Smith {
44044c1414c8SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscInt*,PetscInt*[],PetscInt*);
44054c1414c8SBarry Smith 
44064c1414c8SBarry Smith   PetscFunctionBegin;
4407e32f2f54SBarry Smith   if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
44080005d66cSJed Brown   ierr = PetscObjectQueryFunction((PetscObject)A,"MatInodeGetInodeSizes_C",&f);CHKERRQ(ierr);
44094c1414c8SBarry Smith   if (f) {
44104c1414c8SBarry Smith     ierr = (*f)(A,node_count,sizes,limit);CHKERRQ(ierr);
44114c1414c8SBarry Smith   }
44124c1414c8SBarry Smith   PetscFunctionReturn(0);
44134c1414c8SBarry Smith }
44144c1414c8SBarry Smith 
44154c1414c8SBarry Smith #undef __FUNCT__
44164108e4d5SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes_SeqAIJ_Inode"
44177087cfbeSBarry Smith PetscErrorCode  MatInodeGetInodeSizes_SeqAIJ_Inode(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit)
44184c1414c8SBarry Smith {
44194c1414c8SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
44204c1414c8SBarry Smith 
44214c1414c8SBarry Smith   PetscFunctionBegin;
44224c1414c8SBarry Smith   if (node_count) *node_count = a->inode.node_count;
44234c1414c8SBarry Smith   if (sizes)      *sizes      = a->inode.size;
44244c1414c8SBarry Smith   if (limit)      *limit      = a->inode.limit;
44254c1414c8SBarry Smith   PetscFunctionReturn(0);
44264c1414c8SBarry Smith }
4427