xref: /petsc/src/mat/impls/aij/seq/inode.c (revision ec710b6acee97ecb7c6d9b176ebf365d3891e033)
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 
8470880abSPatrick Sanan static PetscErrorCode MatCreateColInode_Private(Mat A,PetscInt *size,PetscInt **ns)
94c1414c8SBarry Smith {
104c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
114c1414c8SBarry Smith   PetscErrorCode ierr;
124c1414c8SBarry Smith   PetscInt       i,count,m,n,min_mn,*ns_row,*ns_col;
134c1414c8SBarry Smith 
144c1414c8SBarry Smith   PetscFunctionBegin;
15d0f46423SBarry Smith   n = A->cmap->n;
16d0f46423SBarry Smith   m = A->rmap->n;
17b215bc84SStefano Zampini   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
184c1414c8SBarry Smith   ns_row = a->inode.size;
194c1414c8SBarry Smith 
204c1414c8SBarry Smith   min_mn = (m < n) ? m : n;
214c1414c8SBarry Smith   if (!ns) {
224c1414c8SBarry Smith     for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) ;
234c1414c8SBarry Smith     for (; count+1 < n; count++,i++) ;
244c1414c8SBarry Smith     if (count < n)  {
254c1414c8SBarry Smith       i++;
264c1414c8SBarry Smith     }
274c1414c8SBarry Smith     *size = i;
284c1414c8SBarry Smith     PetscFunctionReturn(0);
294c1414c8SBarry Smith   }
30854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&ns_col);CHKERRQ(ierr);
314c1414c8SBarry Smith 
324c1414c8SBarry Smith   /* Use the same row structure wherever feasible. */
334c1414c8SBarry Smith   for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) {
344c1414c8SBarry Smith     ns_col[i] = ns_row[i];
354c1414c8SBarry Smith   }
364c1414c8SBarry Smith 
374c1414c8SBarry Smith   /* if m < n; pad up the remainder with inode_limit */
384c1414c8SBarry Smith   for (; count+1 < n; count++,i++) {
394c1414c8SBarry Smith     ns_col[i] = 1;
404c1414c8SBarry Smith   }
414c1414c8SBarry Smith   /* The last node is the odd ball. padd it up with the remaining rows; */
424c1414c8SBarry Smith   if (count < n) {
434c1414c8SBarry Smith     ns_col[i] = n - count;
444c1414c8SBarry Smith     i++;
454c1414c8SBarry Smith   } else if (count > n) {
464c1414c8SBarry Smith     /* Adjust for the over estimation */
474c1414c8SBarry Smith     ns_col[i-1] += n - count;
484c1414c8SBarry Smith   }
494c1414c8SBarry Smith   *size = i;
504c1414c8SBarry Smith   *ns   = ns_col;
514c1414c8SBarry Smith   PetscFunctionReturn(0);
524c1414c8SBarry Smith }
534c1414c8SBarry Smith 
544c1414c8SBarry Smith 
554c1414c8SBarry Smith /*
564c1414c8SBarry Smith       This builds symmetric version of nonzero structure,
574c1414c8SBarry Smith */
581a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Symmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift)
594c1414c8SBarry Smith {
604c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
614c1414c8SBarry Smith   PetscErrorCode ierr;
628758e1faSBarry Smith   PetscInt       *work,*ia,*ja,nz,nslim_row,nslim_col,m,row,col,n;
638758e1faSBarry Smith   PetscInt       *tns,*tvc,*ns_row = a->inode.size,*ns_col,nsz,i1,i2;
648758e1faSBarry Smith   const PetscInt *j,*jmax,*ai= a->i,*aj = a->j;
654c1414c8SBarry Smith 
664c1414c8SBarry Smith   PetscFunctionBegin;
674c1414c8SBarry Smith   nslim_row = a->inode.node_count;
68d0f46423SBarry Smith   m         = A->rmap->n;
69d0f46423SBarry Smith   n         = A->cmap->n;
70e32f2f54SBarry Smith   if (m != n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatGetRowIJ_SeqAIJ_Inode_Symmetric: Matrix should be square");
71b215bc84SStefano Zampini   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
724c1414c8SBarry Smith 
734c1414c8SBarry Smith   /* Use the row_inode as column_inode */
744c1414c8SBarry Smith   nslim_col = nslim_row;
754c1414c8SBarry Smith   ns_col    = ns_row;
764c1414c8SBarry Smith 
774c1414c8SBarry Smith   /* allocate space for reformated inode structure */
78580bdb30SBarry Smith   ierr = PetscMalloc2(nslim_col+1,&tns,n+1,&tvc);CHKERRQ(ierr);
794c1414c8SBarry Smith   for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1]+ ns_row[i1];
804c1414c8SBarry Smith 
814c1414c8SBarry Smith   for (i1=0,col=0; i1<nslim_col; ++i1) {
824c1414c8SBarry Smith     nsz = ns_col[i1];
832205254eSKarl Rupp     for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1;
844c1414c8SBarry Smith   }
854c1414c8SBarry Smith   /* allocate space for row pointers */
86580bdb30SBarry Smith   ierr = PetscCalloc1(nslim_row+1,&ia);CHKERRQ(ierr);
874c1414c8SBarry Smith   *iia = ia;
88854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_row+1,&work);CHKERRQ(ierr);
894c1414c8SBarry Smith 
904c1414c8SBarry Smith   /* determine the number of columns in each row */
914c1414c8SBarry Smith   ia[0] = oshift;
924c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
934c1414c8SBarry Smith 
944c1414c8SBarry Smith     j    = aj + ai[row] + ishift;
954c1414c8SBarry Smith     jmax = aj + ai[row+1] + ishift;
9683fed2edSSatish Balay     if (j==jmax) continue; /* empty row */
974c1414c8SBarry Smith     col  = *j++ + ishift;
984c1414c8SBarry Smith     i2   = tvc[col];
994c1414c8SBarry Smith     while (i2<i1 && j<jmax) { /* 1.[-xx-d-xx--] 2.[-xx-------],off-diagonal elemets */
1004c1414c8SBarry Smith       ia[i1+1]++;
1014c1414c8SBarry Smith       ia[i2+1]++;
1024c1414c8SBarry Smith       i2++;                     /* Start col of next node */
10390d2dec7SBarry Smith       while ((j<jmax) && ((col=*j+ishift)<tns[i2])) ++j;
1044c1414c8SBarry Smith       i2 = tvc[col];
1054c1414c8SBarry Smith     }
1064c1414c8SBarry Smith     if (i2 == i1) ia[i2+1]++;    /* now the diagonal element */
1074c1414c8SBarry Smith   }
1084c1414c8SBarry Smith 
1094c1414c8SBarry Smith   /* shift ia[i] to point to next row */
1104c1414c8SBarry Smith   for (i1=1; i1<nslim_row+1; i1++) {
1114c1414c8SBarry Smith     row        = ia[i1-1];
1124c1414c8SBarry Smith     ia[i1]    += row;
1134c1414c8SBarry Smith     work[i1-1] = row - oshift;
1144c1414c8SBarry Smith   }
1154c1414c8SBarry Smith 
1164c1414c8SBarry Smith   /* allocate space for column pointers */
1174c1414c8SBarry Smith   nz   = ia[nslim_row] + (!ishift);
118785e854fSJed Brown   ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr);
1194c1414c8SBarry Smith   *jja = ja;
1204c1414c8SBarry Smith 
1214c1414c8SBarry Smith   /* loop over lower triangular part putting into ja */
1224c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row += ns_row[i1],i1++) {
1234c1414c8SBarry Smith     j    = aj + ai[row] + ishift;
1244c1414c8SBarry Smith     jmax = aj + ai[row+1] + ishift;
12583fed2edSSatish Balay     if (j==jmax) continue; /* empty row */
1264c1414c8SBarry Smith     col  = *j++ + ishift;
1274c1414c8SBarry Smith     i2   = tvc[col];
1284c1414c8SBarry Smith     while (i2<i1 && j<jmax) {
1294c1414c8SBarry Smith       ja[work[i2]++] = i1 + oshift;
1304c1414c8SBarry Smith       ja[work[i1]++] = i2 + oshift;
1314c1414c8SBarry Smith       ++i2;
13290d2dec7SBarry Smith       while ((j<jmax) && ((col=*j+ishift)< tns[i2])) ++j; /* Skip rest col indices in this node */
1334c1414c8SBarry Smith       i2 = tvc[col];
1344c1414c8SBarry Smith     }
1354c1414c8SBarry Smith     if (i2 == i1) ja[work[i1]++] = i2 + oshift;
1364c1414c8SBarry Smith 
1374c1414c8SBarry Smith   }
1384c1414c8SBarry Smith   ierr = PetscFree(work);CHKERRQ(ierr);
139580bdb30SBarry Smith   ierr = PetscFree2(tns,tvc);CHKERRQ(ierr);
1404c1414c8SBarry Smith   PetscFunctionReturn(0);
1414c1414c8SBarry Smith }
1424c1414c8SBarry Smith 
1434c1414c8SBarry Smith /*
1444c1414c8SBarry Smith       This builds nonsymmetric version of nonzero structure,
1454c1414c8SBarry Smith */
1461a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift)
1474c1414c8SBarry Smith {
1484c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
1494c1414c8SBarry Smith   PetscErrorCode ierr;
1508758e1faSBarry Smith   PetscInt       *work,*ia,*ja,nz,nslim_row,n,row,col,*ns_col,nslim_col;
1518758e1faSBarry Smith   PetscInt       *tns,*tvc,nsz,i1,i2;
1528758e1faSBarry Smith   const PetscInt *j,*ai= a->i,*aj = a->j,*ns_row = a->inode.size;
1534c1414c8SBarry Smith 
1544c1414c8SBarry Smith   PetscFunctionBegin;
155b215bc84SStefano Zampini   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
1564c1414c8SBarry Smith   nslim_row = a->inode.node_count;
157d0f46423SBarry Smith   n         = A->cmap->n;
1584c1414c8SBarry Smith 
1594c1414c8SBarry Smith   /* Create The column_inode for this matrix */
160470880abSPatrick Sanan   ierr = MatCreateColInode_Private(A,&nslim_col,&ns_col);CHKERRQ(ierr);
1614c1414c8SBarry Smith 
1624c1414c8SBarry Smith   /* allocate space for reformated column_inode structure */
163580bdb30SBarry Smith   ierr = PetscMalloc2(nslim_col +1,&tns,n + 1,&tvc);CHKERRQ(ierr);
1644c1414c8SBarry Smith   for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1];
1654c1414c8SBarry Smith 
1664c1414c8SBarry Smith   for (i1=0,col=0; i1<nslim_col; ++i1) {
1674c1414c8SBarry Smith     nsz = ns_col[i1];
1682205254eSKarl Rupp     for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1;
1694c1414c8SBarry Smith   }
1704c1414c8SBarry Smith   /* allocate space for row pointers */
171580bdb30SBarry Smith   ierr = PetscCalloc1(nslim_row+1,&ia);CHKERRQ(ierr);
1724c1414c8SBarry Smith   *iia = ia;
173854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_row+1,&work);CHKERRQ(ierr);
1744c1414c8SBarry Smith 
1754c1414c8SBarry Smith   /* determine the number of columns in each row */
1764c1414c8SBarry Smith   ia[0] = oshift;
1774c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
1784c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
17983fed2edSSatish Balay     nz  = ai[row+1] - ai[row];
18083fed2edSSatish Balay     if (!nz) continue; /* empty row */
1814c1414c8SBarry Smith     col = *j++ + ishift;
1824c1414c8SBarry Smith     i2  = tvc[col];
1834c1414c8SBarry Smith     while (nz-- > 0) {           /* off-diagonal elemets */
1844c1414c8SBarry Smith       ia[i1+1]++;
1854c1414c8SBarry Smith       i2++;                     /* Start col of next node */
186a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
1874c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
1884c1414c8SBarry Smith     }
1894c1414c8SBarry Smith   }
1904c1414c8SBarry Smith 
1914c1414c8SBarry Smith   /* shift ia[i] to point to next row */
1924c1414c8SBarry Smith   for (i1=1; i1<nslim_row+1; i1++) {
1934c1414c8SBarry Smith     row        = ia[i1-1];
1944c1414c8SBarry Smith     ia[i1]    += row;
1954c1414c8SBarry Smith     work[i1-1] = row - oshift;
1964c1414c8SBarry Smith   }
1974c1414c8SBarry Smith 
1984c1414c8SBarry Smith   /* allocate space for column pointers */
1994c1414c8SBarry Smith   nz   = ia[nslim_row] + (!ishift);
200785e854fSJed Brown   ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr);
2014c1414c8SBarry Smith   *jja = ja;
2024c1414c8SBarry Smith 
2034c1414c8SBarry Smith   /* loop over matrix putting into ja */
2044c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
2054c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
20683fed2edSSatish Balay     nz  = ai[row+1] - ai[row];
20783fed2edSSatish Balay     if (!nz) continue; /* empty row */
2084c1414c8SBarry Smith     col = *j++ + ishift;
2094c1414c8SBarry Smith     i2  = tvc[col];
2104c1414c8SBarry Smith     while (nz-- > 0) {
2114c1414c8SBarry Smith       ja[work[i1]++] = i2 + oshift;
2124c1414c8SBarry Smith       ++i2;
213a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
2144c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
2154c1414c8SBarry Smith     }
2164c1414c8SBarry Smith   }
2174c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
2184c1414c8SBarry Smith   ierr = PetscFree(work);CHKERRQ(ierr);
219580bdb30SBarry Smith   ierr = PetscFree2(tns,tvc);CHKERRQ(ierr);
2204c1414c8SBarry Smith   PetscFunctionReturn(0);
2214c1414c8SBarry Smith }
2224c1414c8SBarry Smith 
2231a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
2244c1414c8SBarry Smith {
2254c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
2264c1414c8SBarry Smith   PetscErrorCode ierr;
2274c1414c8SBarry Smith 
2284c1414c8SBarry Smith   PetscFunctionBegin;
2294c1414c8SBarry Smith   *n = a->inode.node_count;
2304c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
2318f7157efSSatish Balay   if (!blockcompressed) {
232feb237baSPierre Jolivet     ierr = MatGetRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);
2338f7157efSSatish Balay   } else if (symmetric) {
2344108e4d5SBarry Smith     ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
2354c1414c8SBarry Smith   } else {
2364108e4d5SBarry Smith     ierr = MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
2374c1414c8SBarry Smith   }
2384c1414c8SBarry Smith   PetscFunctionReturn(0);
2394c1414c8SBarry Smith }
2404c1414c8SBarry Smith 
2411a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
2424c1414c8SBarry Smith {
2434c1414c8SBarry Smith   PetscErrorCode ierr;
2444c1414c8SBarry Smith 
2454c1414c8SBarry Smith   PetscFunctionBegin;
2464c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
2478f7157efSSatish Balay 
2488f7157efSSatish Balay   if (!blockcompressed) {
249feb237baSPierre Jolivet     ierr = MatRestoreRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);
2508f7157efSSatish Balay   } else {
2514c1414c8SBarry Smith     ierr = PetscFree(*ia);CHKERRQ(ierr);
2524c1414c8SBarry Smith     ierr = PetscFree(*ja);CHKERRQ(ierr);
2538f7157efSSatish Balay   }
2544c1414c8SBarry Smith   PetscFunctionReturn(0);
2554c1414c8SBarry Smith }
2564c1414c8SBarry Smith 
2574c1414c8SBarry Smith /* ----------------------------------------------------------- */
2584c1414c8SBarry Smith 
2591a83f524SJed Brown static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift)
2604c1414c8SBarry Smith {
2614c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
2624c1414c8SBarry Smith   PetscErrorCode ierr;
2634c1414c8SBarry Smith   PetscInt       *work,*ia,*ja,*j,nz,nslim_row, n,row,col,*ns_col,nslim_col;
2644c1414c8SBarry Smith   PetscInt       *tns,*tvc,*ns_row = a->inode.size,nsz,i1,i2,*ai= a->i,*aj = a->j;
2654c1414c8SBarry Smith 
2664c1414c8SBarry Smith   PetscFunctionBegin;
267b215bc84SStefano Zampini   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
2684c1414c8SBarry Smith   nslim_row = a->inode.node_count;
269d0f46423SBarry Smith   n         = A->cmap->n;
2704c1414c8SBarry Smith 
2714c1414c8SBarry Smith   /* Create The column_inode for this matrix */
272470880abSPatrick Sanan   ierr = MatCreateColInode_Private(A,&nslim_col,&ns_col);CHKERRQ(ierr);
2734c1414c8SBarry Smith 
2744c1414c8SBarry Smith   /* allocate space for reformated column_inode structure */
275580bdb30SBarry Smith   ierr = PetscMalloc2(nslim_col + 1,&tns,n + 1,&tvc);CHKERRQ(ierr);
2764c1414c8SBarry Smith   for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1];
2774c1414c8SBarry Smith 
2784c1414c8SBarry Smith   for (i1=0,col=0; i1<nslim_col; ++i1) {
2794c1414c8SBarry Smith     nsz = ns_col[i1];
2802205254eSKarl Rupp     for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1;
2814c1414c8SBarry Smith   }
2824c1414c8SBarry Smith   /* allocate space for column pointers */
283580bdb30SBarry Smith   ierr = PetscCalloc1(nslim_col+1,&ia);CHKERRQ(ierr);
2844c1414c8SBarry Smith   *iia = ia;
285854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_col+1,&work);CHKERRQ(ierr);
2864c1414c8SBarry Smith 
2874c1414c8SBarry Smith   /* determine the number of columns in each row */
2884c1414c8SBarry Smith   ia[0] = oshift;
2894c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
2904c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
2914c1414c8SBarry Smith     col = *j++ + ishift;
2924c1414c8SBarry Smith     i2  = tvc[col];
2934c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
2944c1414c8SBarry Smith     while (nz-- > 0) {           /* off-diagonal elemets */
2954c1414c8SBarry Smith       /* ia[i1+1]++; */
2964c1414c8SBarry Smith       ia[i2+1]++;
2974c1414c8SBarry Smith       i2++;
298a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
2994c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
3004c1414c8SBarry Smith     }
3014c1414c8SBarry Smith   }
3024c1414c8SBarry Smith 
3034c1414c8SBarry Smith   /* shift ia[i] to point to next col */
3044c1414c8SBarry Smith   for (i1=1; i1<nslim_col+1; i1++) {
3054c1414c8SBarry Smith     col        = ia[i1-1];
3064c1414c8SBarry Smith     ia[i1]    += col;
3074c1414c8SBarry Smith     work[i1-1] = col - oshift;
3084c1414c8SBarry Smith   }
3094c1414c8SBarry Smith 
3104c1414c8SBarry Smith   /* allocate space for column pointers */
3114c1414c8SBarry Smith   nz   = ia[nslim_col] + (!ishift);
312785e854fSJed Brown   ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr);
3134c1414c8SBarry Smith   *jja = ja;
3144c1414c8SBarry Smith 
3154c1414c8SBarry Smith   /* loop over matrix putting into ja */
3164c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
3174c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
3184c1414c8SBarry Smith     col = *j++ + ishift;
3194c1414c8SBarry Smith     i2  = tvc[col];
3204c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
3214c1414c8SBarry Smith     while (nz-- > 0) {
3224c1414c8SBarry Smith       /* ja[work[i1]++] = i2 + oshift; */
3234c1414c8SBarry Smith       ja[work[i2]++] = i1 + oshift;
3244c1414c8SBarry Smith       i2++;
325a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
3264c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
3274c1414c8SBarry Smith     }
3284c1414c8SBarry Smith   }
3294c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
3304c1414c8SBarry Smith   ierr = PetscFree(work);CHKERRQ(ierr);
331580bdb30SBarry Smith   ierr = PetscFree2(tns,tvc);CHKERRQ(ierr);
3324c1414c8SBarry Smith   PetscFunctionReturn(0);
3334c1414c8SBarry Smith }
3344c1414c8SBarry Smith 
3351a83f524SJed Brown static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
3364c1414c8SBarry Smith {
3374c1414c8SBarry Smith   PetscErrorCode ierr;
3384c1414c8SBarry Smith 
3394c1414c8SBarry Smith   PetscFunctionBegin;
340470880abSPatrick Sanan   ierr = MatCreateColInode_Private(A,n,NULL);CHKERRQ(ierr);
3414c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
3424c1414c8SBarry Smith 
3438f7157efSSatish Balay   if (!blockcompressed) {
344feb237baSPierre Jolivet     ierr = MatGetColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);
3458f7157efSSatish Balay   } else if (symmetric) {
3464c1414c8SBarry Smith     /* Since the indices are symmetric it does'nt matter */
3474108e4d5SBarry Smith     ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
3484c1414c8SBarry Smith   } else {
3494108e4d5SBarry Smith     ierr = MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
3504c1414c8SBarry Smith   }
3514c1414c8SBarry Smith   PetscFunctionReturn(0);
3524c1414c8SBarry Smith }
3534c1414c8SBarry Smith 
3541a83f524SJed Brown static PetscErrorCode MatRestoreColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
3554c1414c8SBarry Smith {
3564c1414c8SBarry Smith   PetscErrorCode ierr;
3574c1414c8SBarry Smith 
3584c1414c8SBarry Smith   PetscFunctionBegin;
3594c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
3608f7157efSSatish Balay   if (!blockcompressed) {
361feb237baSPierre Jolivet     ierr = MatRestoreColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);
3628f7157efSSatish Balay   } else {
3634c1414c8SBarry Smith     ierr = PetscFree(*ia);CHKERRQ(ierr);
3644c1414c8SBarry Smith     ierr = PetscFree(*ja);CHKERRQ(ierr);
3658f7157efSSatish Balay   }
3664c1414c8SBarry Smith   PetscFunctionReturn(0);
3674c1414c8SBarry Smith }
3684c1414c8SBarry Smith 
3694c1414c8SBarry Smith /* ----------------------------------------------------------- */
3704c1414c8SBarry Smith 
371b215bc84SStefano Zampini PetscErrorCode MatMult_SeqAIJ_Inode(Mat A,Vec xx,Vec yy)
3724c1414c8SBarry Smith {
3734c1414c8SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3744c1414c8SBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5,tmp0,tmp1;
375d9fead3dSBarry Smith   PetscScalar       *y;
376dd6ea824SBarry Smith   const PetscScalar *x;
377dd6ea824SBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5;
3784c1414c8SBarry Smith   PetscErrorCode    ierr;
3798758e1faSBarry Smith   PetscInt          i1,i2,n,i,row,node_max,nsz,sz,nonzerorow=0;
3808758e1faSBarry Smith   const PetscInt    *idx,*ns,*ii;
3814c1414c8SBarry Smith 
3824c1414c8SBarry Smith #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
3834c1414c8SBarry Smith #pragma disjoint(*x,*y,*v1,*v2,*v3,*v4,*v5)
3844c1414c8SBarry Smith #endif
3854c1414c8SBarry Smith 
3864c1414c8SBarry Smith   PetscFunctionBegin;
387e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
3884c1414c8SBarry Smith   node_max = a->inode.node_count;
3894c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
3903649974fSBarry Smith   ierr     = VecGetArrayRead(xx,&x);CHKERRQ(ierr);
3914c1414c8SBarry Smith   ierr     = VecGetArray(yy,&y);CHKERRQ(ierr);
3924c1414c8SBarry Smith   idx      = a->j;
3934c1414c8SBarry Smith   v1       = a->a;
3944c1414c8SBarry Smith   ii       = a->i;
3954c1414c8SBarry Smith 
3964c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i) {
3974c1414c8SBarry Smith     nsz         = ns[i];
3984c1414c8SBarry Smith     n           = ii[1] - ii[0];
39998c9bda7SSatish Balay     nonzerorow += (n>0)*nsz;
4004c1414c8SBarry Smith     ii         += nsz;
40150d8bf02SJed Brown     PetscPrefetchBlock(idx+nsz*n,n,0,PETSC_PREFETCH_HINT_NTA);    /* Prefetch the indices for the block row after the current one */
40250d8bf02SJed Brown     PetscPrefetchBlock(v1+nsz*n,nsz*n,0,PETSC_PREFETCH_HINT_NTA); /* Prefetch the values for the block row after the current one  */
4034c1414c8SBarry Smith     sz = n;                     /* No of non zeros in this row */
4044c1414c8SBarry Smith                                 /* Switch on the size of Node */
4054c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
4064c1414c8SBarry Smith     case 1:
40775567043SBarry Smith       sum1 = 0.;
4084c1414c8SBarry Smith 
4094c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4104c1414c8SBarry Smith         i1    = idx[0];         /* The instructions are ordered to */
4114c1414c8SBarry Smith         i2    = idx[1];         /* make the compiler's job easy */
4124c1414c8SBarry Smith         idx  += 2;
4134c1414c8SBarry Smith         tmp0  = x[i1];
4144c1414c8SBarry Smith         tmp1  = x[i2];
4154c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4164c1414c8SBarry Smith       }
4174c1414c8SBarry Smith 
4184c1414c8SBarry Smith       if (n == sz-1) {          /* Take care of the last nonzero  */
4194c1414c8SBarry Smith         tmp0  = x[*idx++];
4204c1414c8SBarry Smith         sum1 += *v1++ *tmp0;
4214c1414c8SBarry Smith       }
4224c1414c8SBarry Smith       y[row++]=sum1;
4234c1414c8SBarry Smith       break;
4244c1414c8SBarry Smith     case 2:
42575567043SBarry Smith       sum1 = 0.;
42675567043SBarry Smith       sum2 = 0.;
4274c1414c8SBarry Smith       v2   = v1 + n;
4284c1414c8SBarry Smith 
4294c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4304c1414c8SBarry Smith         i1    = idx[0];
4314c1414c8SBarry Smith         i2    = idx[1];
4324c1414c8SBarry Smith         idx  += 2;
4334c1414c8SBarry Smith         tmp0  = x[i1];
4344c1414c8SBarry Smith         tmp1  = x[i2];
4354c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4364c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
4374c1414c8SBarry Smith       }
4384c1414c8SBarry Smith       if (n == sz-1) {
4394c1414c8SBarry Smith         tmp0  = x[*idx++];
4404c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
4414c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
4424c1414c8SBarry Smith       }
4434c1414c8SBarry Smith       y[row++]=sum1;
4444c1414c8SBarry Smith       y[row++]=sum2;
4454c1414c8SBarry Smith       v1      =v2;              /* Since the next block to be processed starts there*/
4464c1414c8SBarry Smith       idx    +=sz;
4474c1414c8SBarry Smith       break;
4484c1414c8SBarry Smith     case 3:
44975567043SBarry Smith       sum1 = 0.;
45075567043SBarry Smith       sum2 = 0.;
45175567043SBarry Smith       sum3 = 0.;
4524c1414c8SBarry Smith       v2   = v1 + n;
4534c1414c8SBarry Smith       v3   = v2 + n;
4544c1414c8SBarry Smith 
4554c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4564c1414c8SBarry Smith         i1    = idx[0];
4574c1414c8SBarry Smith         i2    = idx[1];
4584c1414c8SBarry Smith         idx  += 2;
4594c1414c8SBarry Smith         tmp0  = x[i1];
4604c1414c8SBarry Smith         tmp1  = x[i2];
4614c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4624c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
4634c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
4644c1414c8SBarry Smith       }
4654c1414c8SBarry Smith       if (n == sz-1) {
4664c1414c8SBarry Smith         tmp0  = x[*idx++];
4674c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
4684c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
4694c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
4704c1414c8SBarry Smith       }
4714c1414c8SBarry Smith       y[row++]=sum1;
4724c1414c8SBarry Smith       y[row++]=sum2;
4734c1414c8SBarry Smith       y[row++]=sum3;
4744c1414c8SBarry Smith       v1      =v3;              /* Since the next block to be processed starts there*/
4754c1414c8SBarry Smith       idx    +=2*sz;
4764c1414c8SBarry Smith       break;
4774c1414c8SBarry Smith     case 4:
47875567043SBarry Smith       sum1 = 0.;
47975567043SBarry Smith       sum2 = 0.;
48075567043SBarry Smith       sum3 = 0.;
48175567043SBarry Smith       sum4 = 0.;
4824c1414c8SBarry Smith       v2   = v1 + n;
4834c1414c8SBarry Smith       v3   = v2 + n;
4844c1414c8SBarry Smith       v4   = v3 + n;
4854c1414c8SBarry Smith 
4864c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4874c1414c8SBarry Smith         i1    = idx[0];
4884c1414c8SBarry Smith         i2    = idx[1];
4894c1414c8SBarry Smith         idx  += 2;
4904c1414c8SBarry Smith         tmp0  = x[i1];
4914c1414c8SBarry Smith         tmp1  = x[i2];
4924c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
4934c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
4944c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
4954c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
4964c1414c8SBarry Smith       }
4974c1414c8SBarry Smith       if (n == sz-1) {
4984c1414c8SBarry Smith         tmp0  = x[*idx++];
4994c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
5004c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
5014c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
5024c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
5034c1414c8SBarry Smith       }
5044c1414c8SBarry Smith       y[row++]=sum1;
5054c1414c8SBarry Smith       y[row++]=sum2;
5064c1414c8SBarry Smith       y[row++]=sum3;
5074c1414c8SBarry Smith       y[row++]=sum4;
5084c1414c8SBarry Smith       v1      =v4;              /* Since the next block to be processed starts there*/
5094c1414c8SBarry Smith       idx    +=3*sz;
5104c1414c8SBarry Smith       break;
5114c1414c8SBarry Smith     case 5:
51275567043SBarry Smith       sum1 = 0.;
51375567043SBarry Smith       sum2 = 0.;
51475567043SBarry Smith       sum3 = 0.;
51575567043SBarry Smith       sum4 = 0.;
51675567043SBarry Smith       sum5 = 0.;
5174c1414c8SBarry Smith       v2   = v1 + n;
5184c1414c8SBarry Smith       v3   = v2 + n;
5194c1414c8SBarry Smith       v4   = v3 + n;
5204c1414c8SBarry Smith       v5   = v4 + n;
5214c1414c8SBarry Smith 
5224c1414c8SBarry Smith       for (n = 0; n<sz-1; n+=2) {
5234c1414c8SBarry Smith         i1    = idx[0];
5244c1414c8SBarry Smith         i2    = idx[1];
5254c1414c8SBarry Smith         idx  += 2;
5264c1414c8SBarry Smith         tmp0  = x[i1];
5274c1414c8SBarry Smith         tmp1  = x[i2];
5284c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
5294c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
5304c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
5314c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
5324c1414c8SBarry Smith         sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2;
5334c1414c8SBarry Smith       }
5344c1414c8SBarry Smith       if (n == sz-1) {
5354c1414c8SBarry Smith         tmp0  = x[*idx++];
5364c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
5374c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
5384c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
5394c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
5404c1414c8SBarry Smith         sum5 += *v5++ * tmp0;
5414c1414c8SBarry Smith       }
5424c1414c8SBarry Smith       y[row++]=sum1;
5434c1414c8SBarry Smith       y[row++]=sum2;
5444c1414c8SBarry Smith       y[row++]=sum3;
5454c1414c8SBarry Smith       y[row++]=sum4;
5464c1414c8SBarry Smith       y[row++]=sum5;
5474c1414c8SBarry Smith       v1      =v5;       /* Since the next block to be processed starts there */
5484c1414c8SBarry Smith       idx    +=4*sz;
5494c1414c8SBarry Smith       break;
5504c1414c8SBarry Smith     default:
551e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported");
5524c1414c8SBarry Smith     }
5534c1414c8SBarry Smith   }
5543649974fSBarry Smith   ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr);
5554c1414c8SBarry Smith   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
556dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr);
5574c1414c8SBarry Smith   PetscFunctionReturn(0);
5584c1414c8SBarry Smith }
5594c1414c8SBarry Smith /* ----------------------------------------------------------- */
5604108e4d5SBarry Smith /* Almost same code as the MatMult_SeqAIJ_Inode() */
561b215bc84SStefano Zampini PetscErrorCode MatMultAdd_SeqAIJ_Inode(Mat A,Vec xx,Vec zz,Vec yy)
5624c1414c8SBarry Smith {
5634c1414c8SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
5644c1414c8SBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5,tmp0,tmp1;
5658758e1faSBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5;
5668758e1faSBarry Smith   const PetscScalar *x;
5678758e1faSBarry Smith   PetscScalar       *y,*z,*zt;
5684c1414c8SBarry Smith   PetscErrorCode    ierr;
5698758e1faSBarry Smith   PetscInt          i1,i2,n,i,row,node_max,nsz,sz;
5708758e1faSBarry Smith   const PetscInt    *idx,*ns,*ii;
5714c1414c8SBarry Smith 
5724c1414c8SBarry Smith   PetscFunctionBegin;
573e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
5744c1414c8SBarry Smith   node_max = a->inode.node_count;
5754c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
5762205254eSKarl Rupp 
5778758e1faSBarry Smith   ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr);
578f95e332cSBarry Smith   ierr = VecGetArrayPair(zz,yy,&z,&y);CHKERRQ(ierr);
5794c1414c8SBarry Smith   zt = z;
5804c1414c8SBarry Smith 
5814c1414c8SBarry Smith   idx = a->j;
5824c1414c8SBarry Smith   v1  = a->a;
5834c1414c8SBarry Smith   ii  = a->i;
5844c1414c8SBarry Smith 
5854c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i) {
5864c1414c8SBarry Smith     nsz = ns[i];
5874c1414c8SBarry Smith     n   = ii[1] - ii[0];
5884c1414c8SBarry Smith     ii += nsz;
5894c1414c8SBarry Smith     sz  = n;                    /* No of non zeros in this row */
5904c1414c8SBarry Smith                                 /* Switch on the size of Node */
5914c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
5924c1414c8SBarry Smith     case 1:
5934c1414c8SBarry Smith       sum1 = *zt++;
5944c1414c8SBarry Smith 
5954c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
5964c1414c8SBarry Smith         i1    = idx[0];         /* The instructions are ordered to */
5974c1414c8SBarry Smith         i2    = idx[1];         /* make the compiler's job easy */
5984c1414c8SBarry Smith         idx  += 2;
5994c1414c8SBarry Smith         tmp0  = x[i1];
6004c1414c8SBarry Smith         tmp1  = x[i2];
6014c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6024c1414c8SBarry Smith       }
6034c1414c8SBarry Smith 
6044c1414c8SBarry Smith       if (n   == sz-1) {          /* Take care of the last nonzero  */
6054c1414c8SBarry Smith         tmp0  = x[*idx++];
6064c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6074c1414c8SBarry Smith       }
6084c1414c8SBarry Smith       y[row++]=sum1;
6094c1414c8SBarry Smith       break;
6104c1414c8SBarry Smith     case 2:
6114c1414c8SBarry Smith       sum1 = *zt++;
6124c1414c8SBarry Smith       sum2 = *zt++;
6134c1414c8SBarry Smith       v2   = v1 + n;
6144c1414c8SBarry Smith 
6154c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6164c1414c8SBarry Smith         i1    = idx[0];
6174c1414c8SBarry Smith         i2    = idx[1];
6184c1414c8SBarry Smith         idx  += 2;
6194c1414c8SBarry Smith         tmp0  = x[i1];
6204c1414c8SBarry Smith         tmp1  = x[i2];
6214c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6224c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
6234c1414c8SBarry Smith       }
6244c1414c8SBarry Smith       if (n   == sz-1) {
6254c1414c8SBarry Smith         tmp0  = x[*idx++];
6264c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6274c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
6284c1414c8SBarry Smith       }
6294c1414c8SBarry Smith       y[row++]=sum1;
6304c1414c8SBarry Smith       y[row++]=sum2;
6314c1414c8SBarry Smith       v1      =v2;              /* Since the next block to be processed starts there*/
6324c1414c8SBarry Smith       idx    +=sz;
6334c1414c8SBarry Smith       break;
6344c1414c8SBarry Smith     case 3:
6354c1414c8SBarry Smith       sum1 = *zt++;
6364c1414c8SBarry Smith       sum2 = *zt++;
6374c1414c8SBarry Smith       sum3 = *zt++;
6384c1414c8SBarry Smith       v2   = v1 + n;
6394c1414c8SBarry Smith       v3   = v2 + n;
6404c1414c8SBarry Smith 
6414c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6424c1414c8SBarry Smith         i1    = idx[0];
6434c1414c8SBarry Smith         i2    = idx[1];
6444c1414c8SBarry Smith         idx  += 2;
6454c1414c8SBarry Smith         tmp0  = x[i1];
6464c1414c8SBarry Smith         tmp1  = x[i2];
6474c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6484c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
6494c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
6504c1414c8SBarry Smith       }
6514c1414c8SBarry Smith       if (n == sz-1) {
6524c1414c8SBarry Smith         tmp0  = x[*idx++];
6534c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6544c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
6554c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
6564c1414c8SBarry Smith       }
6574c1414c8SBarry Smith       y[row++]=sum1;
6584c1414c8SBarry Smith       y[row++]=sum2;
6594c1414c8SBarry Smith       y[row++]=sum3;
6604c1414c8SBarry Smith       v1      =v3;              /* Since the next block to be processed starts there*/
6614c1414c8SBarry Smith       idx    +=2*sz;
6624c1414c8SBarry Smith       break;
6634c1414c8SBarry Smith     case 4:
6644c1414c8SBarry Smith       sum1 = *zt++;
6654c1414c8SBarry Smith       sum2 = *zt++;
6664c1414c8SBarry Smith       sum3 = *zt++;
6674c1414c8SBarry Smith       sum4 = *zt++;
6684c1414c8SBarry Smith       v2   = v1 + n;
6694c1414c8SBarry Smith       v3   = v2 + n;
6704c1414c8SBarry Smith       v4   = v3 + n;
6714c1414c8SBarry Smith 
6724c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6734c1414c8SBarry Smith         i1    = idx[0];
6744c1414c8SBarry Smith         i2    = idx[1];
6754c1414c8SBarry Smith         idx  += 2;
6764c1414c8SBarry Smith         tmp0  = x[i1];
6774c1414c8SBarry Smith         tmp1  = x[i2];
6784c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
6794c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
6804c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
6814c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
6824c1414c8SBarry Smith       }
6834c1414c8SBarry Smith       if (n == sz-1) {
6844c1414c8SBarry Smith         tmp0  = x[*idx++];
6854c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6864c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
6874c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
6884c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
6894c1414c8SBarry Smith       }
6904c1414c8SBarry Smith       y[row++]=sum1;
6914c1414c8SBarry Smith       y[row++]=sum2;
6924c1414c8SBarry Smith       y[row++]=sum3;
6934c1414c8SBarry Smith       y[row++]=sum4;
6944c1414c8SBarry Smith       v1      =v4;              /* Since the next block to be processed starts there*/
6954c1414c8SBarry Smith       idx    +=3*sz;
6964c1414c8SBarry Smith       break;
6974c1414c8SBarry Smith     case 5:
6984c1414c8SBarry Smith       sum1 = *zt++;
6994c1414c8SBarry Smith       sum2 = *zt++;
7004c1414c8SBarry Smith       sum3 = *zt++;
7014c1414c8SBarry Smith       sum4 = *zt++;
7024c1414c8SBarry Smith       sum5 = *zt++;
7034c1414c8SBarry Smith       v2   = v1 + n;
7044c1414c8SBarry Smith       v3   = v2 + n;
7054c1414c8SBarry Smith       v4   = v3 + n;
7064c1414c8SBarry Smith       v5   = v4 + n;
7074c1414c8SBarry Smith 
7084c1414c8SBarry Smith       for (n = 0; n<sz-1; n+=2) {
7094c1414c8SBarry Smith         i1    = idx[0];
7104c1414c8SBarry Smith         i2    = idx[1];
7114c1414c8SBarry Smith         idx  += 2;
7124c1414c8SBarry Smith         tmp0  = x[i1];
7134c1414c8SBarry Smith         tmp1  = x[i2];
7144c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
7154c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
7164c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
7174c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
7184c1414c8SBarry Smith         sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2;
7194c1414c8SBarry Smith       }
7204c1414c8SBarry Smith       if (n   == sz-1) {
7214c1414c8SBarry Smith         tmp0  = x[*idx++];
7224c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
7234c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
7244c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
7254c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
7264c1414c8SBarry Smith         sum5 += *v5++ * tmp0;
7274c1414c8SBarry Smith       }
7284c1414c8SBarry Smith       y[row++]=sum1;
7294c1414c8SBarry Smith       y[row++]=sum2;
7304c1414c8SBarry Smith       y[row++]=sum3;
7314c1414c8SBarry Smith       y[row++]=sum4;
7324c1414c8SBarry Smith       y[row++]=sum5;
7334c1414c8SBarry Smith       v1      =v5;       /* Since the next block to be processed starts there */
7344c1414c8SBarry Smith       idx    +=4*sz;
7354c1414c8SBarry Smith       break;
7364c1414c8SBarry Smith     default:
737e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported");
7384c1414c8SBarry Smith     }
7394c1414c8SBarry Smith   }
7408758e1faSBarry Smith   ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr);
741f95e332cSBarry Smith   ierr = VecRestoreArrayPair(zz,yy,&z,&y);CHKERRQ(ierr);
742dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
7434c1414c8SBarry Smith   PetscFunctionReturn(0);
7444c1414c8SBarry Smith }
7454c1414c8SBarry Smith 
7464c1414c8SBarry Smith /* ----------------------------------------------------------- */
747ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_Inode_inplace(Mat A,Vec bb,Vec xx)
7484c1414c8SBarry Smith {
7494c1414c8SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
7504c1414c8SBarry Smith   IS                iscol = a->col,isrow = a->row;
7514c1414c8SBarry Smith   PetscErrorCode    ierr;
7525d0c19d7SBarry Smith   const PetscInt    *r,*c,*rout,*cout;
7538758e1faSBarry Smith   PetscInt          i,j,n = A->rmap->n,nz;
7548758e1faSBarry Smith   PetscInt          node_max,*ns,row,nsz,aii,i0,i1;
7558758e1faSBarry Smith   const PetscInt    *ai = a->i,*a_j = a->j,*vi,*ad,*aj;
756d9fead3dSBarry Smith   PetscScalar       *x,*tmp,*tmps,tmp0,tmp1;
757d9fead3dSBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5;
758dd6ea824SBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa;
759dd6ea824SBarry Smith   const PetscScalar *b;
7604c1414c8SBarry Smith 
7614c1414c8SBarry Smith   PetscFunctionBegin;
762e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
7634c1414c8SBarry Smith   node_max = a->inode.node_count;
7644c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
7654c1414c8SBarry Smith 
7663649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
767fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
7684c1414c8SBarry Smith   tmp  = a->solve_work;
7694c1414c8SBarry Smith 
7704c1414c8SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
7714c1414c8SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
7724c1414c8SBarry Smith 
7734c1414c8SBarry Smith   /* forward solve the lower triangular */
7744c1414c8SBarry Smith   tmps = tmp;
7754c1414c8SBarry Smith   aa   = a_a;
7764c1414c8SBarry Smith   aj   = a_j;
7774c1414c8SBarry Smith   ad   = a->diag;
7784c1414c8SBarry Smith 
7794c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i) {
7804c1414c8SBarry Smith     nsz = ns[i];
7814c1414c8SBarry Smith     aii = ai[row];
7824c1414c8SBarry Smith     v1  = aa + aii;
7834c1414c8SBarry Smith     vi  = aj + aii;
7844c1414c8SBarry Smith     nz  = ad[row]- aii;
78526549573SJed Brown     if (i < node_max-1) {
78626549573SJed Brown       /* Prefetch the block after the current one, the prefetch itself can't cause a memory error,
78726549573SJed Brown       * but our indexing to determine it's size could. */
78850d8bf02SJed Brown       PetscPrefetchBlock(aj+ai[row+nsz],ad[row+nsz]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); /* indices */
78926549573SJed Brown       /* In my tests, it seems to be better to fetch entire rows instead of just the lower-triangular part */
79050d8bf02SJed Brown       PetscPrefetchBlock(aa+ai[row+nsz],ad[row+nsz+ns[i+1]-1]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA);
79126549573SJed 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); */
79226549573SJed Brown     }
7934c1414c8SBarry Smith 
7944c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
7954c1414c8SBarry Smith     case 1:
7964c1414c8SBarry Smith       sum1 = b[*r++];
7974c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
7984c1414c8SBarry Smith         i0    = vi[0];
7994c1414c8SBarry Smith         i1    = vi[1];
8004c1414c8SBarry Smith         vi   +=2;
8014c1414c8SBarry Smith         tmp0  = tmps[i0];
8024c1414c8SBarry Smith         tmp1  = tmps[i1];
8034c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8044c1414c8SBarry Smith       }
8054c1414c8SBarry Smith       if (j == nz-1) {
8064c1414c8SBarry Smith         tmp0  = tmps[*vi++];
8074c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8084c1414c8SBarry Smith       }
8094c1414c8SBarry Smith       tmp[row++]=sum1;
8104c1414c8SBarry Smith       break;
8114c1414c8SBarry Smith     case 2:
8124c1414c8SBarry Smith       sum1 = b[*r++];
8134c1414c8SBarry Smith       sum2 = b[*r++];
8144c1414c8SBarry Smith       v2   = aa + ai[row+1];
8154c1414c8SBarry Smith 
8164c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
8174c1414c8SBarry Smith         i0    = vi[0];
8184c1414c8SBarry Smith         i1    = vi[1];
8194c1414c8SBarry Smith         vi   +=2;
8204c1414c8SBarry Smith         tmp0  = tmps[i0];
8214c1414c8SBarry Smith         tmp1  = tmps[i1];
8224c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8234c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
8244c1414c8SBarry Smith       }
8254c1414c8SBarry Smith       if (j == nz-1) {
8264c1414c8SBarry Smith         tmp0  = tmps[*vi++];
8274c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8284c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
8294c1414c8SBarry Smith       }
8304c1414c8SBarry Smith       sum2     -= *v2++ *sum1;
8314c1414c8SBarry Smith       tmp[row++]=sum1;
8324c1414c8SBarry Smith       tmp[row++]=sum2;
8334c1414c8SBarry Smith       break;
8344c1414c8SBarry Smith     case 3:
8354c1414c8SBarry Smith       sum1 = b[*r++];
8364c1414c8SBarry Smith       sum2 = b[*r++];
8374c1414c8SBarry Smith       sum3 = b[*r++];
8384c1414c8SBarry Smith       v2   = aa + ai[row+1];
8394c1414c8SBarry Smith       v3   = aa + ai[row+2];
8404c1414c8SBarry Smith 
8414c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
8424c1414c8SBarry Smith         i0    = vi[0];
8434c1414c8SBarry Smith         i1    = vi[1];
8444c1414c8SBarry Smith         vi   +=2;
8454c1414c8SBarry Smith         tmp0  = tmps[i0];
8464c1414c8SBarry Smith         tmp1  = tmps[i1];
8474c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8484c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
8494c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
8504c1414c8SBarry Smith       }
8514c1414c8SBarry Smith       if (j == nz-1) {
8524c1414c8SBarry Smith         tmp0  = tmps[*vi++];
8534c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8544c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
8554c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
8564c1414c8SBarry Smith       }
8574c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
8584c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
8594c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
8602205254eSKarl Rupp 
8614c1414c8SBarry Smith       tmp[row++]=sum1;
8624c1414c8SBarry Smith       tmp[row++]=sum2;
8634c1414c8SBarry Smith       tmp[row++]=sum3;
8644c1414c8SBarry Smith       break;
8654c1414c8SBarry Smith 
8664c1414c8SBarry Smith     case 4:
8674c1414c8SBarry Smith       sum1 = b[*r++];
8684c1414c8SBarry Smith       sum2 = b[*r++];
8694c1414c8SBarry Smith       sum3 = b[*r++];
8704c1414c8SBarry Smith       sum4 = b[*r++];
8714c1414c8SBarry Smith       v2   = aa + ai[row+1];
8724c1414c8SBarry Smith       v3   = aa + ai[row+2];
8734c1414c8SBarry Smith       v4   = aa + ai[row+3];
8744c1414c8SBarry Smith 
8754c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
8764c1414c8SBarry Smith         i0    = vi[0];
8774c1414c8SBarry Smith         i1    = vi[1];
8784c1414c8SBarry Smith         vi   +=2;
8794c1414c8SBarry Smith         tmp0  = tmps[i0];
8804c1414c8SBarry Smith         tmp1  = tmps[i1];
8814c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8824c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
8834c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
8844c1414c8SBarry Smith         sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
8854c1414c8SBarry Smith       }
8864c1414c8SBarry Smith       if (j == nz-1) {
8874c1414c8SBarry Smith         tmp0  = tmps[*vi++];
8884c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8894c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
8904c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
8914c1414c8SBarry Smith         sum4 -= *v4++ *tmp0;
8924c1414c8SBarry Smith       }
8934c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
8944c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
8954c1414c8SBarry Smith       sum4 -= *v4++ * sum1;
8964c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
8974c1414c8SBarry Smith       sum4 -= *v4++ * sum2;
8984c1414c8SBarry Smith       sum4 -= *v4++ * sum3;
8994c1414c8SBarry Smith 
9004c1414c8SBarry Smith       tmp[row++]=sum1;
9014c1414c8SBarry Smith       tmp[row++]=sum2;
9024c1414c8SBarry Smith       tmp[row++]=sum3;
9034c1414c8SBarry Smith       tmp[row++]=sum4;
9044c1414c8SBarry Smith       break;
9054c1414c8SBarry Smith     case 5:
9064c1414c8SBarry Smith       sum1 = b[*r++];
9074c1414c8SBarry Smith       sum2 = b[*r++];
9084c1414c8SBarry Smith       sum3 = b[*r++];
9094c1414c8SBarry Smith       sum4 = b[*r++];
9104c1414c8SBarry Smith       sum5 = b[*r++];
9114c1414c8SBarry Smith       v2   = aa + ai[row+1];
9124c1414c8SBarry Smith       v3   = aa + ai[row+2];
9134c1414c8SBarry Smith       v4   = aa + ai[row+3];
9144c1414c8SBarry Smith       v5   = aa + ai[row+4];
9154c1414c8SBarry Smith 
9164c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
9174c1414c8SBarry Smith         i0    = vi[0];
9184c1414c8SBarry Smith         i1    = vi[1];
9194c1414c8SBarry Smith         vi   +=2;
9204c1414c8SBarry Smith         tmp0  = tmps[i0];
9214c1414c8SBarry Smith         tmp1  = tmps[i1];
9224c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
9234c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
9244c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
9254c1414c8SBarry Smith         sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
9264c1414c8SBarry Smith         sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
9274c1414c8SBarry Smith       }
9284c1414c8SBarry Smith       if (j == nz-1) {
9294c1414c8SBarry Smith         tmp0  = tmps[*vi++];
9304c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
9314c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
9324c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
9334c1414c8SBarry Smith         sum4 -= *v4++ *tmp0;
9344c1414c8SBarry Smith         sum5 -= *v5++ *tmp0;
9354c1414c8SBarry Smith       }
9364c1414c8SBarry Smith 
9374c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
9384c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
9394c1414c8SBarry Smith       sum4 -= *v4++ * sum1;
9404c1414c8SBarry Smith       sum5 -= *v5++ * sum1;
9414c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
9424c1414c8SBarry Smith       sum4 -= *v4++ * sum2;
9434c1414c8SBarry Smith       sum5 -= *v5++ * sum2;
9444c1414c8SBarry Smith       sum4 -= *v4++ * sum3;
9454c1414c8SBarry Smith       sum5 -= *v5++ * sum3;
9464c1414c8SBarry Smith       sum5 -= *v5++ * sum4;
9474c1414c8SBarry Smith 
9484c1414c8SBarry Smith       tmp[row++]=sum1;
9494c1414c8SBarry Smith       tmp[row++]=sum2;
9504c1414c8SBarry Smith       tmp[row++]=sum3;
9514c1414c8SBarry Smith       tmp[row++]=sum4;
9524c1414c8SBarry Smith       tmp[row++]=sum5;
9534c1414c8SBarry Smith       break;
9544c1414c8SBarry Smith     default:
955e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
9564c1414c8SBarry Smith     }
9574c1414c8SBarry Smith   }
9584c1414c8SBarry Smith   /* backward solve the upper triangular */
9594c1414c8SBarry Smith   for (i=node_max -1,row = n-1; i>=0; i--) {
9604c1414c8SBarry Smith     nsz = ns[i];
9614c1414c8SBarry Smith     aii = ai[row+1] -1;
9624c1414c8SBarry Smith     v1  = aa + aii;
9634c1414c8SBarry Smith     vi  = aj + aii;
9644c1414c8SBarry Smith     nz  = aii- ad[row];
9654c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
9664c1414c8SBarry Smith     case 1:
9674c1414c8SBarry Smith       sum1 = tmp[row];
9684c1414c8SBarry Smith 
9694c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
9704c1414c8SBarry Smith         vi   -=2;
9714c1414c8SBarry Smith         i0    = vi[2];
9724c1414c8SBarry Smith         i1    = vi[1];
9734c1414c8SBarry Smith         tmp0  = tmps[i0];
9744c1414c8SBarry Smith         tmp1  = tmps[i1];
9754c1414c8SBarry Smith         v1   -= 2;
9764c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
9774c1414c8SBarry Smith       }
9784c1414c8SBarry Smith       if (j==1) {
9794c1414c8SBarry Smith         tmp0  = tmps[*vi--];
9804c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
9814c1414c8SBarry Smith       }
9824c1414c8SBarry Smith       x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
9834c1414c8SBarry Smith       break;
9844c1414c8SBarry Smith     case 2:
9854c1414c8SBarry Smith       sum1 = tmp[row];
9864c1414c8SBarry Smith       sum2 = tmp[row -1];
9874c1414c8SBarry Smith       v2   = aa + ai[row]-1;
9884c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
9894c1414c8SBarry Smith         vi   -=2;
9904c1414c8SBarry Smith         i0    = vi[2];
9914c1414c8SBarry Smith         i1    = vi[1];
9924c1414c8SBarry Smith         tmp0  = tmps[i0];
9934c1414c8SBarry Smith         tmp1  = tmps[i1];
9944c1414c8SBarry Smith         v1   -= 2;
9954c1414c8SBarry Smith         v2   -= 2;
9964c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
9974c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
9984c1414c8SBarry Smith       }
9994c1414c8SBarry Smith       if (j==1) {
10004c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10014c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10024c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10034c1414c8SBarry Smith       }
10044c1414c8SBarry Smith 
10054c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10064c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10074c1414c8SBarry Smith       x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
10084c1414c8SBarry Smith       break;
10094c1414c8SBarry Smith     case 3:
10104c1414c8SBarry Smith       sum1 = tmp[row];
10114c1414c8SBarry Smith       sum2 = tmp[row -1];
10124c1414c8SBarry Smith       sum3 = tmp[row -2];
10134c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10144c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
10154c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
10164c1414c8SBarry Smith         vi   -=2;
10174c1414c8SBarry Smith         i0    = vi[2];
10184c1414c8SBarry Smith         i1    = vi[1];
10194c1414c8SBarry Smith         tmp0  = tmps[i0];
10204c1414c8SBarry Smith         tmp1  = tmps[i1];
10214c1414c8SBarry Smith         v1   -= 2;
10224c1414c8SBarry Smith         v2   -= 2;
10234c1414c8SBarry Smith         v3   -= 2;
10244c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10254c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10264c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
10274c1414c8SBarry Smith       }
10284c1414c8SBarry Smith       if (j==1) {
10294c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10304c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10314c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10324c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
10334c1414c8SBarry Smith       }
10344c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10354c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10364c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10374c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
10384c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10394c1414c8SBarry Smith       x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
10404c1414c8SBarry Smith 
10414c1414c8SBarry Smith       break;
10424c1414c8SBarry Smith     case 4:
10434c1414c8SBarry Smith       sum1 = tmp[row];
10444c1414c8SBarry Smith       sum2 = tmp[row -1];
10454c1414c8SBarry Smith       sum3 = tmp[row -2];
10464c1414c8SBarry Smith       sum4 = tmp[row -3];
10474c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10484c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
10494c1414c8SBarry Smith       v4   = aa + ai[row -2]-1;
10504c1414c8SBarry Smith 
10514c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
10524c1414c8SBarry Smith         vi   -=2;
10534c1414c8SBarry Smith         i0    = vi[2];
10544c1414c8SBarry Smith         i1    = vi[1];
10554c1414c8SBarry Smith         tmp0  = tmps[i0];
10564c1414c8SBarry Smith         tmp1  = tmps[i1];
10574c1414c8SBarry Smith         v1   -= 2;
10584c1414c8SBarry Smith         v2   -= 2;
10594c1414c8SBarry Smith         v3   -= 2;
10604c1414c8SBarry Smith         v4   -= 2;
10614c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10624c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10634c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
10644c1414c8SBarry Smith         sum4 -= v4[2] * tmp0 + v4[1] * tmp1;
10654c1414c8SBarry Smith       }
10664c1414c8SBarry Smith       if (j==1) {
10674c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10684c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10694c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10704c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
10714c1414c8SBarry Smith         sum4 -= *v4-- * tmp0;
10724c1414c8SBarry Smith       }
10734c1414c8SBarry Smith 
10744c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10754c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10764c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10774c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
10784c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
10794c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10804c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
10814c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
10824c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
10834c1414c8SBarry Smith       x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--;
10844c1414c8SBarry Smith       break;
10854c1414c8SBarry Smith     case 5:
10864c1414c8SBarry Smith       sum1 = tmp[row];
10874c1414c8SBarry Smith       sum2 = tmp[row -1];
10884c1414c8SBarry Smith       sum3 = tmp[row -2];
10894c1414c8SBarry Smith       sum4 = tmp[row -3];
10904c1414c8SBarry Smith       sum5 = tmp[row -4];
10914c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10924c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
10934c1414c8SBarry Smith       v4   = aa + ai[row -2]-1;
10944c1414c8SBarry Smith       v5   = aa + ai[row -3]-1;
10954c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
10964c1414c8SBarry Smith         vi   -= 2;
10974c1414c8SBarry Smith         i0    = vi[2];
10984c1414c8SBarry Smith         i1    = vi[1];
10994c1414c8SBarry Smith         tmp0  = tmps[i0];
11004c1414c8SBarry Smith         tmp1  = tmps[i1];
11014c1414c8SBarry Smith         v1   -= 2;
11024c1414c8SBarry Smith         v2   -= 2;
11034c1414c8SBarry Smith         v3   -= 2;
11044c1414c8SBarry Smith         v4   -= 2;
11054c1414c8SBarry Smith         v5   -= 2;
11064c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
11074c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
11084c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
11094c1414c8SBarry Smith         sum4 -= v4[2] * tmp0 + v4[1] * tmp1;
11104c1414c8SBarry Smith         sum5 -= v5[2] * tmp0 + v5[1] * tmp1;
11114c1414c8SBarry Smith       }
11124c1414c8SBarry Smith       if (j==1) {
11134c1414c8SBarry Smith         tmp0  = tmps[*vi--];
11144c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
11154c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
11164c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
11174c1414c8SBarry Smith         sum4 -= *v4-- * tmp0;
11184c1414c8SBarry Smith         sum5 -= *v5-- * tmp0;
11194c1414c8SBarry Smith       }
11204c1414c8SBarry Smith 
11214c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
11224c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
11234c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11244c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11254c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11264c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
11274c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11284c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11294c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11304c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
11314c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11324c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11334c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--;
11344c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11354c1414c8SBarry Smith       x[*c--] = tmp[row] = sum5*a_a[ad[row]]; row--;
11364c1414c8SBarry Smith       break;
11374c1414c8SBarry Smith     default:
1138e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
11394c1414c8SBarry Smith     }
11404c1414c8SBarry Smith   }
11414c1414c8SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11424c1414c8SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
11433649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1144fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
1145dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
11464c1414c8SBarry Smith   PetscFunctionReturn(0);
11474c1414c8SBarry Smith }
11484c1414c8SBarry Smith 
11499a625307SHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_Inode(Mat B,Mat A,const MatFactorInfo *info)
115028f1b45aSHong Zhang {
115128f1b45aSHong Zhang   Mat             C     =B;
115228f1b45aSHong Zhang   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
115328f1b45aSHong Zhang   IS              isrow = b->row,isicol = b->icol;
115428f1b45aSHong Zhang   PetscErrorCode  ierr;
115528f1b45aSHong Zhang   const PetscInt  *r,*ic,*ics;
115628f1b45aSHong Zhang   const PetscInt  n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag;
115728f1b45aSHong Zhang   PetscInt        i,j,k,nz,nzL,row,*pj;
115828f1b45aSHong Zhang   const PetscInt  *ajtmp,*bjtmp;
11599877982aSShri Abhyankar   MatScalar       *pc,*pc1,*pc2,*pc3,*pc4,mul1,mul2,mul3,mul4,*pv,*rtmp1,*rtmp2,*rtmp3,*rtmp4;
11609877982aSShri Abhyankar   const MatScalar *aa=a->a,*v,*v1,*v2,*v3,*v4;
116128f1b45aSHong Zhang   FactorShiftCtx  sctx;
11624f81c4b7SBarry Smith   const PetscInt  *ddiag;
116328f1b45aSHong Zhang   PetscReal       rs;
116428f1b45aSHong Zhang   MatScalar       d;
11654f81c4b7SBarry Smith   PetscInt        inod,nodesz,node_max,col;
11664f81c4b7SBarry Smith   const PetscInt  *ns;
116707b50cabSHong Zhang   PetscInt        *tmp_vec1,*tmp_vec2,*nsmap;
11680e95ead3SHong Zhang 
116928f1b45aSHong Zhang   PetscFunctionBegin;
117028f1b45aSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
117128f1b45aSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
117228f1b45aSHong Zhang 
1173f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
117428f1b45aSHong Zhang     ddiag          = a->diag;
117528f1b45aSHong Zhang     sctx.shift_top = info->zeropivot;
117628f1b45aSHong Zhang     for (i=0; i<n; i++) {
117728f1b45aSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
117828f1b45aSHong Zhang       d  = (aa)[ddiag[i]];
117928f1b45aSHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
118028f1b45aSHong Zhang       v  = aa+ai[i];
118128f1b45aSHong Zhang       nz = ai[i+1] - ai[i];
11822205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
118328f1b45aSHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
118428f1b45aSHong Zhang     }
118528f1b45aSHong Zhang     sctx.shift_top *= 1.1;
118628f1b45aSHong Zhang     sctx.nshift_max = 5;
118728f1b45aSHong Zhang     sctx.shift_lo   = 0.;
118828f1b45aSHong Zhang     sctx.shift_hi   = 1.;
118928f1b45aSHong Zhang   }
119028f1b45aSHong Zhang 
119128f1b45aSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
119228f1b45aSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
119368785679SHong Zhang 
11941795a4d1SJed Brown   ierr  = PetscCalloc4(n,&rtmp1,n,&rtmp2,n,&rtmp3,n,&rtmp4);CHKERRQ(ierr);
119528f1b45aSHong Zhang   ics   = ic;
119628f1b45aSHong Zhang 
119728f1b45aSHong Zhang   node_max = a->inode.node_count;
119828f1b45aSHong Zhang   ns       = a->inode.size;
1199e7e72b3dSBarry Smith   if (!ns) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix without inode information");
120028f1b45aSHong Zhang 
12019877982aSShri Abhyankar   /* If max inode size > 4, split it into two inodes.*/
120268785679SHong Zhang   /* also map the inode sizes according to the ordering */
1203854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&tmp_vec1);CHKERRQ(ierr);
120468785679SHong Zhang   for (i=0,j=0; i<node_max; ++i,++j) {
1205b1550197SShri Abhyankar     if (ns[i] > 4) {
1206048b5e81SShri Abhyankar       tmp_vec1[j] = 4;
120768785679SHong Zhang       ++j;
120868785679SHong Zhang       tmp_vec1[j] = ns[i] - tmp_vec1[j-1];
120968785679SHong Zhang     } else {
121068785679SHong Zhang       tmp_vec1[j] = ns[i];
121168785679SHong Zhang     }
121268785679SHong Zhang   }
121368785679SHong Zhang   /* Use the correct node_max */
121468785679SHong Zhang   node_max = j;
121568785679SHong Zhang 
121668785679SHong Zhang   /* Now reorder the inode info based on mat re-ordering info */
121768785679SHong Zhang   /* First create a row -> inode_size_array_index map */
1218854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&nsmap);CHKERRQ(ierr);
1219854ce69bSBarry Smith   ierr = PetscMalloc1(node_max+1,&tmp_vec2);CHKERRQ(ierr);
122068785679SHong Zhang   for (i=0,row=0; i<node_max; i++) {
122168785679SHong Zhang     nodesz = tmp_vec1[i];
122268785679SHong Zhang     for (j=0; j<nodesz; j++,row++) {
122368785679SHong Zhang       nsmap[row] = i;
122468785679SHong Zhang     }
122568785679SHong Zhang   }
122668785679SHong Zhang   /* Using nsmap, create a reordered ns structure */
122768785679SHong Zhang   for (i=0,j=0; i< node_max; i++) {
122868785679SHong Zhang     nodesz      = tmp_vec1[nsmap[r[j]]];     /* here the reordered row_no is in r[] */
122968785679SHong Zhang     tmp_vec2[i] = nodesz;
123068785679SHong Zhang     j          += nodesz;
123168785679SHong Zhang   }
123268785679SHong Zhang   ierr = PetscFree(nsmap);CHKERRQ(ierr);
123368785679SHong Zhang   ierr = PetscFree(tmp_vec1);CHKERRQ(ierr);
1234b89f182dSHong Zhang 
123568785679SHong Zhang   /* Now use the correct ns */
123668785679SHong Zhang   ns = tmp_vec2;
123768785679SHong Zhang 
123828f1b45aSHong Zhang   do {
123907b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
124028f1b45aSHong Zhang     /* Now loop over each block-row, and do the factorization */
124128f1b45aSHong Zhang     for (inod=0,i=0; inod<node_max; inod++) { /* i: row index; inod: inode index */
124228f1b45aSHong Zhang       nodesz = ns[inod];
124328f1b45aSHong Zhang 
124428f1b45aSHong Zhang       switch (nodesz) {
124528f1b45aSHong Zhang       case 1:
124668785679SHong Zhang         /*----------*/
1247b89f182dSHong Zhang         /* zero rtmp1 */
124828f1b45aSHong Zhang         /* L part */
124928f1b45aSHong Zhang         nz    = bi[i+1] - bi[i];
125028f1b45aSHong Zhang         bjtmp = bj + bi[i];
1251b89f182dSHong Zhang         for (j=0; j<nz; j++) rtmp1[bjtmp[j]] = 0.0;
125228f1b45aSHong Zhang 
125328f1b45aSHong Zhang         /* U part */
125428f1b45aSHong Zhang         nz    = bdiag[i]-bdiag[i+1];
125528f1b45aSHong Zhang         bjtmp = bj + bdiag[i+1]+1;
1256b89f182dSHong Zhang         for (j=0; j<nz; j++) rtmp1[bjtmp[j]] = 0.0;
125728f1b45aSHong Zhang 
125828f1b45aSHong Zhang         /* load in initial (unfactored row) */
125928f1b45aSHong Zhang         nz    = ai[r[i]+1] - ai[r[i]];
126028f1b45aSHong Zhang         ajtmp = aj + ai[r[i]];
126128f1b45aSHong Zhang         v     = aa + ai[r[i]];
12622205254eSKarl Rupp         for (j=0; j<nz; j++) rtmp1[ics[ajtmp[j]]] = v[j];
12632205254eSKarl Rupp 
126428f1b45aSHong Zhang         /* ZeropivotApply() */
1265b89f182dSHong Zhang         rtmp1[i] += sctx.shift_amount;  /* shift the diagonal of the matrix */
126628f1b45aSHong Zhang 
126728f1b45aSHong Zhang         /* elimination */
126828f1b45aSHong Zhang         bjtmp = bj + bi[i];
126928f1b45aSHong Zhang         row   = *bjtmp++;
127028f1b45aSHong Zhang         nzL   = bi[i+1] - bi[i];
127128f1b45aSHong Zhang         for (k=0; k < nzL; k++) {
1272b89f182dSHong Zhang           pc = rtmp1 + row;
127328f1b45aSHong Zhang           if (*pc != 0.0) {
127428f1b45aSHong Zhang             pv   = b->a + bdiag[row];
1275b89f182dSHong Zhang             mul1 = *pc * (*pv);
1276b89f182dSHong Zhang             *pc  = mul1;
127728f1b45aSHong Zhang             pj   = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
127828f1b45aSHong Zhang             pv   = b->a + bdiag[row+1]+1;
127928f1b45aSHong Zhang             nz   = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
1280b89f182dSHong Zhang             for (j=0; j<nz; j++) rtmp1[pj[j]] -= mul1 * pv[j];
1281ca0c957dSBarry Smith             ierr = PetscLogFlops(1+2.0*nz);CHKERRQ(ierr);
128228f1b45aSHong Zhang           }
128328f1b45aSHong Zhang           row = *bjtmp++;
128428f1b45aSHong Zhang         }
128528f1b45aSHong Zhang 
128628f1b45aSHong Zhang         /* finished row so stick it into b->a */
128728f1b45aSHong Zhang         rs = 0.0;
128828f1b45aSHong Zhang         /* L part */
128928f1b45aSHong Zhang         pv = b->a + bi[i];
129028f1b45aSHong Zhang         pj = b->j + bi[i];
129128f1b45aSHong Zhang         nz = bi[i+1] - bi[i];
129228f1b45aSHong Zhang         for (j=0; j<nz; j++) {
1293b89f182dSHong Zhang           pv[j] = rtmp1[pj[j]]; rs += PetscAbsScalar(pv[j]);
129428f1b45aSHong Zhang         }
129528f1b45aSHong Zhang 
129628f1b45aSHong Zhang         /* U part */
129728f1b45aSHong Zhang         pv = b->a + bdiag[i+1]+1;
129828f1b45aSHong Zhang         pj = b->j + bdiag[i+1]+1;
129928f1b45aSHong Zhang         nz = bdiag[i] - bdiag[i+1]-1;
130028f1b45aSHong Zhang         for (j=0; j<nz; j++) {
1301b89f182dSHong Zhang           pv[j] = rtmp1[pj[j]]; rs += PetscAbsScalar(pv[j]);
130228f1b45aSHong Zhang         }
130328f1b45aSHong Zhang 
1304b89f182dSHong Zhang         /* Check zero pivot */
130528f1b45aSHong Zhang         sctx.rs = rs;
1306b89f182dSHong Zhang         sctx.pv = rtmp1[i];
13074cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
130807b50cabSHong Zhang         if (sctx.newshift) break;
130928f1b45aSHong Zhang 
131028f1b45aSHong Zhang         /* Mark diagonal and invert diagonal for simplier triangular solves */
131128f1b45aSHong Zhang         pv  = b->a + bdiag[i];
1312b89f182dSHong Zhang         *pv = 1.0/sctx.pv; /* sctx.pv = rtmp1[i]+shiftamount if shifttype==MAT_SHIFT_INBLOCKS */
131328f1b45aSHong Zhang         break;
131428f1b45aSHong Zhang 
131528f1b45aSHong Zhang       case 2:
131628f1b45aSHong Zhang         /*----------*/
1317b89f182dSHong Zhang         /* zero rtmp1 and rtmp2 */
131828f1b45aSHong Zhang         /* L part */
131928f1b45aSHong Zhang         nz    = bi[i+1] - bi[i];
132028f1b45aSHong Zhang         bjtmp = bj + bi[i];
132128f1b45aSHong Zhang         for  (j=0; j<nz; j++) {
132268785679SHong Zhang           col        = bjtmp[j];
1323b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0;
132428f1b45aSHong Zhang         }
132528f1b45aSHong Zhang 
132628f1b45aSHong Zhang         /* U part */
132728f1b45aSHong Zhang         nz    = bdiag[i]-bdiag[i+1];
132828f1b45aSHong Zhang         bjtmp = bj + bdiag[i+1]+1;
132928f1b45aSHong Zhang         for  (j=0; j<nz; j++) {
133068785679SHong Zhang           col        = bjtmp[j];
1331b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0;
133228f1b45aSHong Zhang         }
133328f1b45aSHong Zhang 
133428f1b45aSHong Zhang         /* load in initial (unfactored row) */
133528f1b45aSHong Zhang         nz    = ai[r[i]+1] - ai[r[i]];
133628f1b45aSHong Zhang         ajtmp = aj + ai[r[i]];
133728f1b45aSHong Zhang         v1    = aa + ai[r[i]]; v2 = aa + ai[r[i]+1];
133828f1b45aSHong Zhang         for (j=0; j<nz; j++) {
133968785679SHong Zhang           col        = ics[ajtmp[j]];
1340b89f182dSHong Zhang           rtmp1[col] = v1[j]; rtmp2[col] = v2[j];
134128f1b45aSHong Zhang         }
134228f1b45aSHong Zhang         /* ZeropivotApply(): shift the diagonal of the matrix  */
1343b89f182dSHong Zhang         rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount;
134428f1b45aSHong Zhang 
134528f1b45aSHong Zhang         /* elimination */
134628f1b45aSHong Zhang         bjtmp = bj + bi[i];
134728f1b45aSHong Zhang         row   = *bjtmp++; /* pivot row */
134828f1b45aSHong Zhang         nzL   = bi[i+1] - bi[i];
134928f1b45aSHong Zhang         for (k=0; k < nzL; k++) {
1350b89f182dSHong Zhang           pc1 = rtmp1 + row;
1351b89f182dSHong Zhang           pc2 = rtmp2 + row;
135228f1b45aSHong Zhang           if (*pc1 != 0.0 || *pc2 != 0.0) {
135328f1b45aSHong Zhang             pv   = b->a + bdiag[row];
1354b89f182dSHong Zhang             mul1 = *pc1*(*pv); mul2 = *pc2*(*pv);
1355b89f182dSHong Zhang             *pc1 = mul1;       *pc2 = mul2;
135628f1b45aSHong Zhang 
135728f1b45aSHong Zhang             pj = b->j + bdiag[row+1]+1;     /* beginning of U(row,:) */
135828f1b45aSHong Zhang             pv = b->a + bdiag[row+1]+1;
135928f1b45aSHong Zhang             nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
136028f1b45aSHong Zhang             for (j=0; j<nz; j++) {
136168785679SHong Zhang               col         = pj[j];
1362b89f182dSHong Zhang               rtmp1[col] -= mul1 * pv[j];
1363b89f182dSHong Zhang               rtmp2[col] -= mul2 * pv[j];
136428f1b45aSHong Zhang             }
1365ca0c957dSBarry Smith             ierr = PetscLogFlops(2+4.0*nz);CHKERRQ(ierr);
136628f1b45aSHong Zhang           }
136728f1b45aSHong Zhang           row = *bjtmp++;
136828f1b45aSHong Zhang         }
136928f1b45aSHong Zhang 
1370b89f182dSHong Zhang         /* finished row i; check zero pivot, then stick row i into b->a */
137128f1b45aSHong Zhang         rs = 0.0;
137228f1b45aSHong Zhang         /* L part */
1373b89f182dSHong Zhang         pc1 = b->a + bi[i];
137428f1b45aSHong Zhang         pj  = b->j + bi[i];
137528f1b45aSHong Zhang         nz  = bi[i+1] - bi[i];
137628f1b45aSHong Zhang         for (j=0; j<nz; j++) {
137768785679SHong Zhang           col    = pj[j];
13780e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
137928f1b45aSHong Zhang         }
138028f1b45aSHong Zhang         /* U part */
1381b89f182dSHong Zhang         pc1 = b->a + bdiag[i+1]+1;
138228f1b45aSHong Zhang         pj  = b->j + bdiag[i+1]+1;
13830e7a5c2bSHong Zhang         nz  = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */
138428f1b45aSHong Zhang         for (j=0; j<nz; j++) {
138568785679SHong Zhang           col    = pj[j];
13860e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
138728f1b45aSHong Zhang         }
138828f1b45aSHong Zhang 
138928f1b45aSHong Zhang         sctx.rs = rs;
1390b89f182dSHong Zhang         sctx.pv = rtmp1[i];
13914cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
139207b50cabSHong Zhang         if (sctx.newshift) break;
1393b89f182dSHong Zhang         pc1  = b->a + bdiag[i]; /* Mark diagonal */
1394b89f182dSHong Zhang         *pc1 = 1.0/sctx.pv;
1395b89f182dSHong Zhang 
1396b89f182dSHong Zhang         /* Now take care of diagonal 2x2 block. */
1397b89f182dSHong Zhang         pc2 = rtmp2 + i;
1398b89f182dSHong Zhang         if (*pc2 != 0.0) {
1399b89f182dSHong Zhang           mul1 = (*pc2)*(*pc1); /* *pc1=diag[i] is inverted! */
1400b89f182dSHong Zhang           *pc2 = mul1;          /* insert L entry */
1401b89f182dSHong Zhang           pj   = b->j + bdiag[i+1]+1;   /* beginning of U(i,:) */
1402b89f182dSHong Zhang           nz   = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */
1403b89f182dSHong Zhang           for (j=0; j<nz; j++) {
1404b89f182dSHong Zhang             col = pj[j]; rtmp2[col] -= mul1 * rtmp1[col];
140528f1b45aSHong Zhang           }
1406ca0c957dSBarry Smith           ierr = PetscLogFlops(1+2.0*nz);CHKERRQ(ierr);
1407b89f182dSHong Zhang         }
1408b89f182dSHong Zhang 
1409b89f182dSHong Zhang         /* finished row i+1; check zero pivot, then stick row i+1 into b->a */
1410b89f182dSHong Zhang         rs = 0.0;
1411b89f182dSHong Zhang         /* L part */
1412b89f182dSHong Zhang         pc2 = b->a + bi[i+1];
1413b89f182dSHong Zhang         pj  = b->j + bi[i+1];
1414b89f182dSHong Zhang         nz  = bi[i+2] - bi[i+1];
1415b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1416b89f182dSHong Zhang           col    = pj[j];
14170e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1418b89f182dSHong Zhang         }
1419b89f182dSHong Zhang         /* U part */
1420b89f182dSHong Zhang         pc2 = b->a + bdiag[i+2]+1;
14210e7a5c2bSHong Zhang         pj  = b->j + bdiag[i+2]+1;
14220e7a5c2bSHong Zhang         nz  = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */
1423b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1424b89f182dSHong Zhang           col    = pj[j];
14250e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1426b89f182dSHong Zhang         }
1427b89f182dSHong Zhang 
142828f1b45aSHong Zhang         sctx.rs = rs;
1429b89f182dSHong Zhang         sctx.pv = rtmp2[i+1];
14304cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i+1);CHKERRQ(ierr);
143107b50cabSHong Zhang         if (sctx.newshift) break;
143228f1b45aSHong Zhang         pc2  = b->a + bdiag[i+1];
1433b89f182dSHong Zhang         *pc2 = 1.0/sctx.pv;
143428f1b45aSHong Zhang         break;
1435b89f182dSHong Zhang 
143668785679SHong Zhang       case 3:
143768785679SHong Zhang         /*----------*/
143868785679SHong Zhang         /* zero rtmp */
143968785679SHong Zhang         /* L part */
144068785679SHong Zhang         nz    = bi[i+1] - bi[i];
144168785679SHong Zhang         bjtmp = bj + bi[i];
144268785679SHong Zhang         for  (j=0; j<nz; j++) {
144368785679SHong Zhang           col        = bjtmp[j];
1444b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;
144568785679SHong Zhang         }
144668785679SHong Zhang 
144768785679SHong Zhang         /* U part */
144868785679SHong Zhang         nz    = bdiag[i]-bdiag[i+1];
144968785679SHong Zhang         bjtmp = bj + bdiag[i+1]+1;
145068785679SHong Zhang         for  (j=0; j<nz; j++) {
145168785679SHong Zhang           col        = bjtmp[j];
1452b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;
145368785679SHong Zhang         }
145468785679SHong Zhang 
145568785679SHong Zhang         /* load in initial (unfactored row) */
145668785679SHong Zhang         nz    = ai[r[i]+1] - ai[r[i]];
145768785679SHong Zhang         ajtmp = aj + ai[r[i]];
145868785679SHong Zhang         v1    = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; v3 = aa + ai[r[i]+2];
145968785679SHong Zhang         for (j=0; j<nz; j++) {
146068785679SHong Zhang           col        = ics[ajtmp[j]];
1461b89f182dSHong Zhang           rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; rtmp3[col] = v3[j];
146268785679SHong Zhang         }
146368785679SHong Zhang         /* ZeropivotApply(): shift the diagonal of the matrix  */
1464b89f182dSHong Zhang         rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; rtmp3[i+2] += sctx.shift_amount;
146568785679SHong Zhang 
146668785679SHong Zhang         /* elimination */
146768785679SHong Zhang         bjtmp = bj + bi[i];
146868785679SHong Zhang         row   = *bjtmp++; /* pivot row */
146968785679SHong Zhang         nzL   = bi[i+1] - bi[i];
147068785679SHong Zhang         for (k=0; k < nzL; k++) {
1471b89f182dSHong Zhang           pc1 = rtmp1 + row;
1472b89f182dSHong Zhang           pc2 = rtmp2 + row;
1473b89f182dSHong Zhang           pc3 = rtmp3 + row;
147468785679SHong Zhang           if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 != 0.0) {
147568785679SHong Zhang             pv   = b->a + bdiag[row];
1476b89f182dSHong Zhang             mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); mul3 = *pc3*(*pv);
1477b89f182dSHong Zhang             *pc1 = mul1; *pc2 = mul2; *pc3 = mul3;
147868785679SHong Zhang 
147968785679SHong Zhang             pj = b->j + bdiag[row+1]+1;     /* beginning of U(row,:) */
148068785679SHong Zhang             pv = b->a + bdiag[row+1]+1;
148168785679SHong Zhang             nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
148268785679SHong Zhang             for (j=0; j<nz; j++) {
148368785679SHong Zhang               col         = pj[j];
1484b89f182dSHong Zhang               rtmp1[col] -= mul1 * pv[j];
1485b89f182dSHong Zhang               rtmp2[col] -= mul2 * pv[j];
1486b89f182dSHong Zhang               rtmp3[col] -= mul3 * pv[j];
148768785679SHong Zhang             }
1488ca0c957dSBarry Smith             ierr = PetscLogFlops(3+6.0*nz);CHKERRQ(ierr);
148968785679SHong Zhang           }
149068785679SHong Zhang           row = *bjtmp++;
149168785679SHong Zhang         }
149268785679SHong Zhang 
1493b89f182dSHong Zhang         /* finished row i; check zero pivot, then stick row i into b->a */
1494b89f182dSHong Zhang         rs = 0.0;
1495b89f182dSHong Zhang         /* L part */
1496b89f182dSHong Zhang         pc1 = b->a + bi[i];
1497b89f182dSHong Zhang         pj  = b->j + bi[i];
1498b89f182dSHong Zhang         nz  = bi[i+1] - bi[i];
1499b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1500b89f182dSHong Zhang           col    = pj[j];
15010e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
1502b89f182dSHong Zhang         }
1503b89f182dSHong Zhang         /* U part */
1504b89f182dSHong Zhang         pc1 = b->a + bdiag[i+1]+1;
1505b89f182dSHong Zhang         pj  = b->j + bdiag[i+1]+1;
15060e7a5c2bSHong Zhang         nz  = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */
1507b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1508b89f182dSHong Zhang           col    = pj[j];
15090e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
1510b89f182dSHong Zhang         }
151168785679SHong Zhang 
1512b89f182dSHong Zhang         sctx.rs = rs;
1513b89f182dSHong Zhang         sctx.pv = rtmp1[i];
15144cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
151507b50cabSHong Zhang         if (sctx.newshift) break;
1516b89f182dSHong Zhang         pc1  = b->a + bdiag[i]; /* Mark diag[i] */
1517b89f182dSHong Zhang         *pc1 = 1.0/sctx.pv;
1518b89f182dSHong Zhang 
1519b89f182dSHong Zhang         /* Now take care of 1st column of diagonal 3x3 block. */
1520b89f182dSHong Zhang         pc2 = rtmp2 + i;
1521b89f182dSHong Zhang         pc3 = rtmp3 + i;
1522b89f182dSHong Zhang         if (*pc2 != 0.0 || *pc3 != 0.0) {
1523b89f182dSHong Zhang           mul2 = (*pc2)*(*pc1); *pc2 = mul2;
1524b89f182dSHong Zhang           mul3 = (*pc3)*(*pc1); *pc3 = mul3;
152568785679SHong Zhang           pj   = b->j + bdiag[i+1]+1;   /* beginning of U(i,:) */
152668785679SHong Zhang           nz   = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */
152768785679SHong Zhang           for (j=0; j<nz; j++) {
152868785679SHong Zhang             col         = pj[j];
1529b89f182dSHong Zhang             rtmp2[col] -= mul2 * rtmp1[col];
1530b89f182dSHong Zhang             rtmp3[col] -= mul3 * rtmp1[col];
153168785679SHong Zhang           }
1532ca0c957dSBarry Smith           ierr = PetscLogFlops(2+4.0*nz);CHKERRQ(ierr);
153368785679SHong Zhang         }
153468785679SHong Zhang 
1535b89f182dSHong Zhang         /* finished row i+1; check zero pivot, then stick row i+1 into b->a */
1536b89f182dSHong Zhang         rs = 0.0;
1537b89f182dSHong Zhang         /* L part */
1538b89f182dSHong Zhang         pc2 = b->a + bi[i+1];
1539b89f182dSHong Zhang         pj  = b->j + bi[i+1];
1540b89f182dSHong Zhang         nz  = bi[i+2] - bi[i+1];
1541b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1542b89f182dSHong Zhang           col    = pj[j];
15430e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1544b89f182dSHong Zhang         }
1545b89f182dSHong Zhang         /* U part */
1546b89f182dSHong Zhang         pc2 = b->a + bdiag[i+2]+1;
15470e7a5c2bSHong Zhang         pj  = b->j + bdiag[i+2]+1;
15480e7a5c2bSHong Zhang         nz  = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */
1549b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1550b89f182dSHong Zhang           col    = pj[j];
15510e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1552b89f182dSHong Zhang         }
1553b89f182dSHong Zhang 
1554b89f182dSHong Zhang         sctx.rs = rs;
1555b89f182dSHong Zhang         sctx.pv = rtmp2[i+1];
15564cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i+1);CHKERRQ(ierr);
155707b50cabSHong Zhang         if (sctx.newshift) break;
1558b89f182dSHong Zhang         pc2  = b->a + bdiag[i+1];
1559b89f182dSHong Zhang         *pc2 = 1.0/sctx.pv; /* Mark diag[i+1] */
1560b89f182dSHong Zhang 
1561b89f182dSHong Zhang         /* Now take care of 2nd column of diagonal 3x3 block. */
1562b89f182dSHong Zhang         pc3 = rtmp3 + i+1;
156368785679SHong Zhang         if (*pc3 != 0.0) {
1564b89f182dSHong Zhang           mul3 = (*pc3)*(*pc2); *pc3 = mul3;
156568785679SHong Zhang           pj   = b->j + bdiag[i+2]+1;     /* beginning of U(i+1,:) */
156668785679SHong Zhang           nz   = bdiag[i+1]-bdiag[i+2]-1; /* num of entries in U(i+1,:) excluding diag */
156768785679SHong Zhang           for (j=0; j<nz; j++) {
156868785679SHong Zhang             col         = pj[j];
1569b89f182dSHong Zhang             rtmp3[col] -= mul3 * rtmp2[col];
157068785679SHong Zhang           }
1571ca0c957dSBarry Smith           ierr = PetscLogFlops(1+2.0*nz);CHKERRQ(ierr);
157268785679SHong Zhang         }
157368785679SHong Zhang 
1574b89f182dSHong Zhang         /* finished i+2; check zero pivot, then stick row i+2 into b->a */
157568785679SHong Zhang         rs = 0.0;
157668785679SHong Zhang         /* L part */
1577b89f182dSHong Zhang         pc3 = b->a + bi[i+2];
1578b89f182dSHong Zhang         pj  = b->j + bi[i+2];
1579b89f182dSHong Zhang         nz  = bi[i+3] - bi[i+2];
158068785679SHong Zhang         for (j=0; j<nz; j++) {
158168785679SHong Zhang           col    = pj[j];
15820e7a5c2bSHong Zhang           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
158368785679SHong Zhang         }
158468785679SHong Zhang         /* U part */
1585b89f182dSHong Zhang         pc3 = b->a + bdiag[i+3]+1;
15860e7a5c2bSHong Zhang         pj  = b->j + bdiag[i+3]+1;
15870e7a5c2bSHong Zhang         nz  = bdiag[i+2] - bdiag[i+3] - 1; /* exclude diagonal */
158868785679SHong Zhang         for (j=0; j<nz; j++) {
158968785679SHong Zhang           col    = pj[j];
15900e7a5c2bSHong Zhang           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
159168785679SHong Zhang         }
159268785679SHong Zhang 
159368785679SHong Zhang         sctx.rs = rs;
1594b89f182dSHong Zhang         sctx.pv = rtmp3[i+2];
15954cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i+2);CHKERRQ(ierr);
159607b50cabSHong Zhang         if (sctx.newshift) break;
159768785679SHong Zhang         pc3  = b->a + bdiag[i+2];
1598b89f182dSHong Zhang         *pc3 = 1.0/sctx.pv; /* Mark diag[i+2] */
159968785679SHong Zhang         break;
16009877982aSShri Abhyankar       case 4:
16019877982aSShri Abhyankar         /*----------*/
16029877982aSShri Abhyankar         /* zero rtmp */
16039877982aSShri Abhyankar         /* L part */
16049877982aSShri Abhyankar         nz    = bi[i+1] - bi[i];
16059877982aSShri Abhyankar         bjtmp = bj + bi[i];
16069877982aSShri Abhyankar         for  (j=0; j<nz; j++) {
16079877982aSShri Abhyankar           col        = bjtmp[j];
16089877982aSShri Abhyankar           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;rtmp4[col] = 0.0;
16099877982aSShri Abhyankar         }
16109877982aSShri Abhyankar 
16119877982aSShri Abhyankar         /* U part */
16129877982aSShri Abhyankar         nz    = bdiag[i]-bdiag[i+1];
16139877982aSShri Abhyankar         bjtmp = bj + bdiag[i+1]+1;
16149877982aSShri Abhyankar         for  (j=0; j<nz; j++) {
16159877982aSShri Abhyankar           col        = bjtmp[j];
16169877982aSShri Abhyankar           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0; rtmp4[col] = 0.0;
16179877982aSShri Abhyankar         }
16189877982aSShri Abhyankar 
16199877982aSShri Abhyankar         /* load in initial (unfactored row) */
16209877982aSShri Abhyankar         nz    = ai[r[i]+1] - ai[r[i]];
16219877982aSShri Abhyankar         ajtmp = aj + ai[r[i]];
16229877982aSShri Abhyankar         v1    = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; v3 = aa + ai[r[i]+2]; v4 = aa + ai[r[i]+3];
16239877982aSShri Abhyankar         for (j=0; j<nz; j++) {
16249877982aSShri Abhyankar           col        = ics[ajtmp[j]];
16259877982aSShri Abhyankar           rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; rtmp3[col] = v3[j]; rtmp4[col] = v4[j];
16269877982aSShri Abhyankar         }
16279877982aSShri Abhyankar         /* ZeropivotApply(): shift the diagonal of the matrix  */
16289877982aSShri Abhyankar         rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; rtmp3[i+2] += sctx.shift_amount; rtmp4[i+3] += sctx.shift_amount;
16299877982aSShri Abhyankar 
16309877982aSShri Abhyankar         /* elimination */
16319877982aSShri Abhyankar         bjtmp = bj + bi[i];
16329877982aSShri Abhyankar         row   = *bjtmp++; /* pivot row */
16339877982aSShri Abhyankar         nzL   = bi[i+1] - bi[i];
16349877982aSShri Abhyankar         for (k=0; k < nzL; k++) {
16359877982aSShri Abhyankar           pc1 = rtmp1 + row;
16369877982aSShri Abhyankar           pc2 = rtmp2 + row;
16379877982aSShri Abhyankar           pc3 = rtmp3 + row;
16389877982aSShri Abhyankar           pc4 = rtmp4 + row;
16399877982aSShri Abhyankar           if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 != 0.0 || *pc4 != 0.0) {
16409877982aSShri Abhyankar             pv   = b->a + bdiag[row];
16419877982aSShri Abhyankar             mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); mul3 = *pc3*(*pv); mul4 = *pc4*(*pv);
16429877982aSShri Abhyankar             *pc1 = mul1; *pc2 = mul2; *pc3 = mul3; *pc4 = mul4;
16439877982aSShri Abhyankar 
16449877982aSShri Abhyankar             pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
16459877982aSShri Abhyankar             pv = b->a + bdiag[row+1]+1;
16469877982aSShri Abhyankar             nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
16479877982aSShri Abhyankar             for (j=0; j<nz; j++) {
16489877982aSShri Abhyankar               col         = pj[j];
16499877982aSShri Abhyankar               rtmp1[col] -= mul1 * pv[j];
16509877982aSShri Abhyankar               rtmp2[col] -= mul2 * pv[j];
16519877982aSShri Abhyankar               rtmp3[col] -= mul3 * pv[j];
16529877982aSShri Abhyankar               rtmp4[col] -= mul4 * pv[j];
16539877982aSShri Abhyankar             }
1654ca0c957dSBarry Smith             ierr = PetscLogFlops(4+8.0*nz);CHKERRQ(ierr);
16559877982aSShri Abhyankar           }
16569877982aSShri Abhyankar           row = *bjtmp++;
16579877982aSShri Abhyankar         }
16589877982aSShri Abhyankar 
16599877982aSShri Abhyankar         /* finished row i; check zero pivot, then stick row i into b->a */
16609877982aSShri Abhyankar         rs = 0.0;
16619877982aSShri Abhyankar         /* L part */
16629877982aSShri Abhyankar         pc1 = b->a + bi[i];
16639877982aSShri Abhyankar         pj  = b->j + bi[i];
16649877982aSShri Abhyankar         nz  = bi[i+1] - bi[i];
16659877982aSShri Abhyankar         for (j=0; j<nz; j++) {
16669877982aSShri Abhyankar           col    = pj[j];
16679877982aSShri Abhyankar           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
16689877982aSShri Abhyankar         }
16699877982aSShri Abhyankar         /* U part */
16709877982aSShri Abhyankar         pc1 = b->a + bdiag[i+1]+1;
16719877982aSShri Abhyankar         pj  = b->j + bdiag[i+1]+1;
16729877982aSShri Abhyankar         nz  = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */
16739877982aSShri Abhyankar         for (j=0; j<nz; j++) {
16749877982aSShri Abhyankar           col    = pj[j];
16759877982aSShri Abhyankar           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
16769877982aSShri Abhyankar         }
16779877982aSShri Abhyankar 
16789877982aSShri Abhyankar         sctx.rs = rs;
16799877982aSShri Abhyankar         sctx.pv = rtmp1[i];
16804cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
168107b50cabSHong Zhang         if (sctx.newshift) break;
16829877982aSShri Abhyankar         pc1  = b->a + bdiag[i]; /* Mark diag[i] */
16839877982aSShri Abhyankar         *pc1 = 1.0/sctx.pv;
16849877982aSShri Abhyankar 
16859877982aSShri Abhyankar         /* Now take care of 1st column of diagonal 4x4 block. */
16869877982aSShri Abhyankar         pc2 = rtmp2 + i;
16879877982aSShri Abhyankar         pc3 = rtmp3 + i;
16889877982aSShri Abhyankar         pc4 = rtmp4 + i;
16899877982aSShri Abhyankar         if (*pc2 != 0.0 || *pc3 != 0.0 || *pc4 != 0.0) {
16909877982aSShri Abhyankar           mul2 = (*pc2)*(*pc1); *pc2 = mul2;
16919877982aSShri Abhyankar           mul3 = (*pc3)*(*pc1); *pc3 = mul3;
16929877982aSShri Abhyankar           mul4 = (*pc4)*(*pc1); *pc4 = mul4;
16939877982aSShri Abhyankar           pj   = b->j + bdiag[i+1]+1;   /* beginning of U(i,:) */
16949877982aSShri Abhyankar           nz   = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */
16959877982aSShri Abhyankar           for (j=0; j<nz; j++) {
16969877982aSShri Abhyankar             col         = pj[j];
16979877982aSShri Abhyankar             rtmp2[col] -= mul2 * rtmp1[col];
16989877982aSShri Abhyankar             rtmp3[col] -= mul3 * rtmp1[col];
16999877982aSShri Abhyankar             rtmp4[col] -= mul4 * rtmp1[col];
17009877982aSShri Abhyankar           }
1701ca0c957dSBarry Smith           ierr = PetscLogFlops(3+6.0*nz);CHKERRQ(ierr);
17029877982aSShri Abhyankar         }
17039877982aSShri Abhyankar 
17049877982aSShri Abhyankar         /* finished row i+1; check zero pivot, then stick row i+1 into b->a */
17059877982aSShri Abhyankar         rs = 0.0;
17069877982aSShri Abhyankar         /* L part */
17079877982aSShri Abhyankar         pc2 = b->a + bi[i+1];
17089877982aSShri Abhyankar         pj  = b->j + bi[i+1];
17099877982aSShri Abhyankar         nz  = bi[i+2] - bi[i+1];
17109877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17119877982aSShri Abhyankar           col    = pj[j];
17129877982aSShri Abhyankar           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
17139877982aSShri Abhyankar         }
17149877982aSShri Abhyankar         /* U part */
17159877982aSShri Abhyankar         pc2 = b->a + bdiag[i+2]+1;
17169877982aSShri Abhyankar         pj  = b->j + bdiag[i+2]+1;
17179877982aSShri Abhyankar         nz  = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */
17189877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17199877982aSShri Abhyankar           col    = pj[j];
17209877982aSShri Abhyankar           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
17219877982aSShri Abhyankar         }
17229877982aSShri Abhyankar 
17239877982aSShri Abhyankar         sctx.rs = rs;
17249877982aSShri Abhyankar         sctx.pv = rtmp2[i+1];
17254cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i+1);CHKERRQ(ierr);
172607b50cabSHong Zhang         if (sctx.newshift) break;
17279877982aSShri Abhyankar         pc2  = b->a + bdiag[i+1];
17289877982aSShri Abhyankar         *pc2 = 1.0/sctx.pv; /* Mark diag[i+1] */
17299877982aSShri Abhyankar 
17309877982aSShri Abhyankar         /* Now take care of 2nd column of diagonal 4x4 block. */
17319877982aSShri Abhyankar         pc3 = rtmp3 + i+1;
17329877982aSShri Abhyankar         pc4 = rtmp4 + i+1;
17339877982aSShri Abhyankar         if (*pc3 != 0.0 || *pc4 != 0.0) {
17349877982aSShri Abhyankar           mul3 = (*pc3)*(*pc2); *pc3 = mul3;
17359877982aSShri Abhyankar           mul4 = (*pc4)*(*pc2); *pc4 = mul4;
17369877982aSShri Abhyankar           pj   = b->j + bdiag[i+2]+1;     /* beginning of U(i+1,:) */
17379877982aSShri Abhyankar           nz   = bdiag[i+1]-bdiag[i+2]-1; /* num of entries in U(i+1,:) excluding diag */
17389877982aSShri Abhyankar           for (j=0; j<nz; j++) {
17399877982aSShri Abhyankar             col         = pj[j];
17409877982aSShri Abhyankar             rtmp3[col] -= mul3 * rtmp2[col];
17419877982aSShri Abhyankar             rtmp4[col] -= mul4 * rtmp2[col];
17429877982aSShri Abhyankar           }
1743ca0c957dSBarry Smith           ierr = PetscLogFlops(4.0*nz);CHKERRQ(ierr);
17449877982aSShri Abhyankar         }
17459877982aSShri Abhyankar 
17469877982aSShri Abhyankar         /* finished i+2; check zero pivot, then stick row i+2 into b->a */
17479877982aSShri Abhyankar         rs = 0.0;
17489877982aSShri Abhyankar         /* L part */
17499877982aSShri Abhyankar         pc3 = b->a + bi[i+2];
17509877982aSShri Abhyankar         pj  = b->j + bi[i+2];
17519877982aSShri Abhyankar         nz  = bi[i+3] - bi[i+2];
17529877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17539877982aSShri Abhyankar           col    = pj[j];
17549877982aSShri Abhyankar           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
17559877982aSShri Abhyankar         }
17569877982aSShri Abhyankar         /* U part */
17579877982aSShri Abhyankar         pc3 = b->a + bdiag[i+3]+1;
17589877982aSShri Abhyankar         pj  = b->j + bdiag[i+3]+1;
17599877982aSShri Abhyankar         nz  = bdiag[i+2] - bdiag[i+3] - 1; /* exclude diagonal */
17609877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17619877982aSShri Abhyankar           col    = pj[j];
17629877982aSShri Abhyankar           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
17639877982aSShri Abhyankar         }
17649877982aSShri Abhyankar 
17659877982aSShri Abhyankar         sctx.rs = rs;
17669877982aSShri Abhyankar         sctx.pv = rtmp3[i+2];
17674cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i+2);CHKERRQ(ierr);
176807b50cabSHong Zhang         if (sctx.newshift) break;
17699877982aSShri Abhyankar         pc3  = b->a + bdiag[i+2];
17709877982aSShri Abhyankar         *pc3 = 1.0/sctx.pv; /* Mark diag[i+2] */
17719877982aSShri Abhyankar 
17729877982aSShri Abhyankar         /* Now take care of 3rd column of diagonal 4x4 block. */
17739877982aSShri Abhyankar         pc4 = rtmp4 + i+2;
17749877982aSShri Abhyankar         if (*pc4 != 0.0) {
17759877982aSShri Abhyankar           mul4 = (*pc4)*(*pc3); *pc4 = mul4;
17769877982aSShri Abhyankar           pj   = b->j + bdiag[i+3]+1;     /* beginning of U(i+2,:) */
17779877982aSShri Abhyankar           nz   = bdiag[i+2]-bdiag[i+3]-1; /* num of entries in U(i+2,:) excluding diag */
17789877982aSShri Abhyankar           for (j=0; j<nz; j++) {
17799877982aSShri Abhyankar             col         = pj[j];
17809877982aSShri Abhyankar             rtmp4[col] -= mul4 * rtmp3[col];
17819877982aSShri Abhyankar           }
1782ca0c957dSBarry Smith           ierr = PetscLogFlops(1+2.0*nz);CHKERRQ(ierr);
17839877982aSShri Abhyankar         }
17849877982aSShri Abhyankar 
17859877982aSShri Abhyankar         /* finished i+3; check zero pivot, then stick row i+3 into b->a */
17869877982aSShri Abhyankar         rs = 0.0;
17879877982aSShri Abhyankar         /* L part */
17889877982aSShri Abhyankar         pc4 = b->a + bi[i+3];
17899877982aSShri Abhyankar         pj  = b->j + bi[i+3];
17909877982aSShri Abhyankar         nz  = bi[i+4] - bi[i+3];
17919877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17929877982aSShri Abhyankar           col    = pj[j];
17939877982aSShri Abhyankar           pc4[j] = rtmp4[col]; rs += PetscAbsScalar(pc4[j]);
17949877982aSShri Abhyankar         }
17959877982aSShri Abhyankar         /* U part */
17969877982aSShri Abhyankar         pc4 = b->a + bdiag[i+4]+1;
17979877982aSShri Abhyankar         pj  = b->j + bdiag[i+4]+1;
17989877982aSShri Abhyankar         nz  = bdiag[i+3] - bdiag[i+4] - 1; /* exclude diagonal */
17999877982aSShri Abhyankar         for (j=0; j<nz; j++) {
18009877982aSShri Abhyankar           col    = pj[j];
18019877982aSShri Abhyankar           pc4[j] = rtmp4[col]; rs += PetscAbsScalar(pc4[j]);
18029877982aSShri Abhyankar         }
18039877982aSShri Abhyankar 
18049877982aSShri Abhyankar         sctx.rs = rs;
18059877982aSShri Abhyankar         sctx.pv = rtmp4[i+3];
18064cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i+3);CHKERRQ(ierr);
180707b50cabSHong Zhang         if (sctx.newshift) break;
18089877982aSShri Abhyankar         pc4  = b->a + bdiag[i+3];
18099877982aSShri Abhyankar         *pc4 = 1.0/sctx.pv; /* Mark diag[i+3] */
18109877982aSShri Abhyankar         break;
181168785679SHong Zhang 
181228f1b45aSHong Zhang       default:
1813e32f2f54SBarry Smith         SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Node size not yet supported \n");
181428f1b45aSHong Zhang       }
1815c2b86aeeSHong Zhang       if (sctx.newshift) break; /* break for (inod=0,i=0; inod<node_max; inod++) */
181628f1b45aSHong Zhang       i += nodesz;                 /* Update the row */
181768785679SHong Zhang     }
181828f1b45aSHong Zhang 
181928f1b45aSHong Zhang     /* MatPivotRefine() */
182007b50cabSHong Zhang     if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
182128f1b45aSHong Zhang       /*
182228f1b45aSHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
182328f1b45aSHong Zhang        * then try lower shift
182428f1b45aSHong Zhang        */
182528f1b45aSHong Zhang       sctx.shift_hi       = sctx.shift_fraction;
182628f1b45aSHong Zhang       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
182728f1b45aSHong Zhang       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
182807b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
182928f1b45aSHong Zhang       sctx.nshift++;
183028f1b45aSHong Zhang     }
183107b50cabSHong Zhang   } while (sctx.newshift);
183228f1b45aSHong Zhang 
1833eb9baa12SBarry Smith   ierr = PetscFree4(rtmp1,rtmp2,rtmp3,rtmp4);CHKERRQ(ierr);
183468785679SHong Zhang   ierr = PetscFree(tmp_vec2);CHKERRQ(ierr);
183528f1b45aSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
183628f1b45aSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
183728f1b45aSHong Zhang 
1838abb87a52SBarry Smith   if (b->inode.size) {
1839abb87a52SBarry Smith     C->ops->solve           = MatSolve_SeqAIJ_Inode;
1840abb87a52SBarry Smith   } else {
1841d3ac4fa3SBarry Smith     C->ops->solve           = MatSolve_SeqAIJ;
1842abb87a52SBarry Smith   }
184328f1b45aSHong Zhang   C->ops->solveadd          = MatSolveAdd_SeqAIJ;
184428f1b45aSHong Zhang   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ;
184528f1b45aSHong Zhang   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ;
184628f1b45aSHong Zhang   C->ops->matsolve          = MatMatSolve_SeqAIJ;
184728f1b45aSHong Zhang   C->assembled              = PETSC_TRUE;
184828f1b45aSHong Zhang   C->preallocated           = PETSC_TRUE;
18492205254eSKarl Rupp 
185028f1b45aSHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
185128f1b45aSHong Zhang 
185228f1b45aSHong Zhang   /* MatShiftView(A,info,&sctx) */
185328f1b45aSHong Zhang   if (sctx.nshift) {
1854f4db908eSBarry Smith     if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) {
185557622a8eSBarry 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);
1856f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
185757622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
1858f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) {
185957622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %g\n",sctx.nshift,(double)info->shiftamount);CHKERRQ(ierr);
186028f1b45aSHong Zhang     }
186128f1b45aSHong Zhang   }
186228f1b45aSHong Zhang   PetscFunctionReturn(0);
186328f1b45aSHong Zhang }
1864628f99d7SShri Abhyankar 
18659a625307SHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_Inode_inplace(Mat B,Mat A,const MatFactorInfo *info)
1866628f99d7SShri Abhyankar {
1867628f99d7SShri Abhyankar   Mat             C     = B;
1868628f99d7SShri Abhyankar   Mat_SeqAIJ      *a    = (Mat_SeqAIJ*)A->data,*b = (Mat_SeqAIJ*)C->data;
1869628f99d7SShri Abhyankar   IS              iscol = b->col,isrow = b->row,isicol = b->icol;
1870628f99d7SShri Abhyankar   PetscErrorCode  ierr;
1871628f99d7SShri Abhyankar   const PetscInt  *r,*ic,*c,*ics;
1872628f99d7SShri Abhyankar   PetscInt        n   = A->rmap->n,*bi = b->i;
1873628f99d7SShri Abhyankar   PetscInt        *bj = b->j,*nbj=b->j +1,*ajtmp,*bjtmp,nz,nz_tmp,row,prow;
18748758e1faSBarry Smith   PetscInt        i,j,idx,*bd = b->diag,node_max,nodesz;
18758758e1faSBarry Smith   PetscInt        *ai = a->i,*aj = a->j;
1876628f99d7SShri Abhyankar   PetscInt        *ns,*tmp_vec1,*tmp_vec2,*nsmap,*pj;
1877628f99d7SShri Abhyankar   PetscScalar     mul1,mul2,mul3,tmp;
1878628f99d7SShri Abhyankar   MatScalar       *pc1,*pc2,*pc3,*ba = b->a,*pv,*rtmp11,*rtmp22,*rtmp33;
1879628f99d7SShri Abhyankar   const MatScalar *v1,*v2,*v3,*aa = a->a,*rtmp1;
1880628f99d7SShri Abhyankar   PetscReal       rs=0.0;
1881628f99d7SShri Abhyankar   FactorShiftCtx  sctx;
1882628f99d7SShri Abhyankar 
1883628f99d7SShri Abhyankar   PetscFunctionBegin;
1884628f99d7SShri Abhyankar   sctx.shift_top      = 0;
1885628f99d7SShri Abhyankar   sctx.nshift_max     = 0;
1886628f99d7SShri Abhyankar   sctx.shift_lo       = 0;
1887628f99d7SShri Abhyankar   sctx.shift_hi       = 0;
1888628f99d7SShri Abhyankar   sctx.shift_fraction = 0;
1889628f99d7SShri Abhyankar 
1890628f99d7SShri Abhyankar   /* if both shift schemes are chosen by user, only use info->shiftpd */
1891f4db908eSBarry Smith   if (info->shifttype==(PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
1892628f99d7SShri Abhyankar     sctx.shift_top = 0;
1893628f99d7SShri Abhyankar     for (i=0; i<n; i++) {
1894628f99d7SShri Abhyankar       /* calculate rs = sum(|aij|)-RealPart(aii), amt of shift needed for this row */
1895628f99d7SShri Abhyankar       rs    = 0.0;
1896628f99d7SShri Abhyankar       ajtmp = aj + ai[i];
1897628f99d7SShri Abhyankar       rtmp1 = aa + ai[i];
1898628f99d7SShri Abhyankar       nz    = ai[i+1] - ai[i];
1899628f99d7SShri Abhyankar       for (j=0; j<nz; j++) {
1900628f99d7SShri Abhyankar         if (*ajtmp != i) {
1901628f99d7SShri Abhyankar           rs += PetscAbsScalar(*rtmp1++);
1902628f99d7SShri Abhyankar         } else {
1903628f99d7SShri Abhyankar           rs -= PetscRealPart(*rtmp1++);
1904628f99d7SShri Abhyankar         }
1905628f99d7SShri Abhyankar         ajtmp++;
1906628f99d7SShri Abhyankar       }
1907628f99d7SShri Abhyankar       if (rs>sctx.shift_top) sctx.shift_top = rs;
1908628f99d7SShri Abhyankar     }
1909628f99d7SShri Abhyankar     if (sctx.shift_top == 0.0) sctx.shift_top += 1.e-12;
1910628f99d7SShri Abhyankar     sctx.shift_top *= 1.1;
1911628f99d7SShri Abhyankar     sctx.nshift_max = 5;
1912628f99d7SShri Abhyankar     sctx.shift_lo   = 0.;
1913628f99d7SShri Abhyankar     sctx.shift_hi   = 1.;
1914628f99d7SShri Abhyankar   }
1915628f99d7SShri Abhyankar   sctx.shift_amount = 0;
1916628f99d7SShri Abhyankar   sctx.nshift       = 0;
1917628f99d7SShri Abhyankar 
1918628f99d7SShri Abhyankar   ierr   = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1919628f99d7SShri Abhyankar   ierr   = ISGetIndices(iscol,&c);CHKERRQ(ierr);
1920628f99d7SShri Abhyankar   ierr   = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
19211795a4d1SJed Brown   ierr   = PetscCalloc3(n,&rtmp11,n,&rtmp22,n,&rtmp33);CHKERRQ(ierr);
1922628f99d7SShri Abhyankar   ics    = ic;
1923628f99d7SShri Abhyankar 
1924628f99d7SShri Abhyankar   node_max = a->inode.node_count;
1925628f99d7SShri Abhyankar   ns       = a->inode.size;
1926e7e72b3dSBarry Smith   if (!ns) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix without inode information");
1927628f99d7SShri Abhyankar 
1928628f99d7SShri Abhyankar   /* If max inode size > 3, split it into two inodes.*/
1929628f99d7SShri Abhyankar   /* also map the inode sizes according to the ordering */
1930854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&tmp_vec1);CHKERRQ(ierr);
1931628f99d7SShri Abhyankar   for (i=0,j=0; i<node_max; ++i,++j) {
1932628f99d7SShri Abhyankar     if (ns[i]>3) {
1933628f99d7SShri Abhyankar       tmp_vec1[j] = ns[i]/2; /* Assuming ns[i] < =5  */
1934628f99d7SShri Abhyankar       ++j;
1935628f99d7SShri Abhyankar       tmp_vec1[j] = ns[i] - tmp_vec1[j-1];
1936628f99d7SShri Abhyankar     } else {
1937628f99d7SShri Abhyankar       tmp_vec1[j] = ns[i];
1938628f99d7SShri Abhyankar     }
1939628f99d7SShri Abhyankar   }
1940628f99d7SShri Abhyankar   /* Use the correct node_max */
1941628f99d7SShri Abhyankar   node_max = j;
1942628f99d7SShri Abhyankar 
1943628f99d7SShri Abhyankar   /* Now reorder the inode info based on mat re-ordering info */
1944628f99d7SShri Abhyankar   /* First create a row -> inode_size_array_index map */
1945071fcb05SBarry Smith   ierr = PetscMalloc2(n+1,&nsmap,node_max+1,&tmp_vec2);CHKERRQ(ierr);
1946628f99d7SShri Abhyankar   for (i=0,row=0; i<node_max; i++) {
1947628f99d7SShri Abhyankar     nodesz = tmp_vec1[i];
1948628f99d7SShri Abhyankar     for (j=0; j<nodesz; j++,row++) {
1949628f99d7SShri Abhyankar       nsmap[row] = i;
1950628f99d7SShri Abhyankar     }
1951628f99d7SShri Abhyankar   }
1952628f99d7SShri Abhyankar   /* Using nsmap, create a reordered ns structure */
1953628f99d7SShri Abhyankar   for (i=0,j=0; i< node_max; i++) {
1954628f99d7SShri Abhyankar     nodesz      = tmp_vec1[nsmap[r[j]]];     /* here the reordered row_no is in r[] */
1955628f99d7SShri Abhyankar     tmp_vec2[i] = nodesz;
1956628f99d7SShri Abhyankar     j          += nodesz;
1957628f99d7SShri Abhyankar   }
1958071fcb05SBarry Smith   ierr = PetscFree2(nsmap,tmp_vec1);CHKERRQ(ierr);
1959628f99d7SShri Abhyankar   /* Now use the correct ns */
1960628f99d7SShri Abhyankar   ns = tmp_vec2;
1961628f99d7SShri Abhyankar 
1962628f99d7SShri Abhyankar   do {
196307b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
1964628f99d7SShri Abhyankar     /* Now loop over each block-row, and do the factorization */
1965628f99d7SShri Abhyankar     for (i=0,row=0; i<node_max; i++) {
1966628f99d7SShri Abhyankar       nodesz = ns[i];
1967628f99d7SShri Abhyankar       nz     = bi[row+1] - bi[row];
1968628f99d7SShri Abhyankar       bjtmp  = bj + bi[row];
1969628f99d7SShri Abhyankar 
1970628f99d7SShri Abhyankar       switch (nodesz) {
1971628f99d7SShri Abhyankar       case 1:
1972628f99d7SShri Abhyankar         for  (j=0; j<nz; j++) {
1973628f99d7SShri Abhyankar           idx         = bjtmp[j];
1974628f99d7SShri Abhyankar           rtmp11[idx] = 0.0;
1975628f99d7SShri Abhyankar         }
1976628f99d7SShri Abhyankar 
1977628f99d7SShri Abhyankar         /* load in initial (unfactored row) */
1978628f99d7SShri Abhyankar         idx    = r[row];
1979628f99d7SShri Abhyankar         nz_tmp = ai[idx+1] - ai[idx];
1980628f99d7SShri Abhyankar         ajtmp  = aj + ai[idx];
1981628f99d7SShri Abhyankar         v1     = aa + ai[idx];
1982628f99d7SShri Abhyankar 
1983628f99d7SShri Abhyankar         for (j=0; j<nz_tmp; j++) {
1984628f99d7SShri Abhyankar           idx         = ics[ajtmp[j]];
1985628f99d7SShri Abhyankar           rtmp11[idx] = v1[j];
1986628f99d7SShri Abhyankar         }
1987628f99d7SShri Abhyankar         rtmp11[ics[r[row]]] += sctx.shift_amount;
1988628f99d7SShri Abhyankar 
1989628f99d7SShri Abhyankar         prow = *bjtmp++;
1990628f99d7SShri Abhyankar         while (prow < row) {
1991628f99d7SShri Abhyankar           pc1 = rtmp11 + prow;
1992628f99d7SShri Abhyankar           if (*pc1 != 0.0) {
1993628f99d7SShri Abhyankar             pv     = ba + bd[prow];
1994628f99d7SShri Abhyankar             pj     = nbj + bd[prow];
1995628f99d7SShri Abhyankar             mul1   = *pc1 * *pv++;
1996628f99d7SShri Abhyankar             *pc1   = mul1;
1997628f99d7SShri Abhyankar             nz_tmp = bi[prow+1] - bd[prow] - 1;
1998ca0c957dSBarry Smith             ierr   = PetscLogFlops(1+2.0*nz_tmp);CHKERRQ(ierr);
1999628f99d7SShri Abhyankar             for (j=0; j<nz_tmp; j++) {
2000628f99d7SShri Abhyankar               tmp          = pv[j];
2001628f99d7SShri Abhyankar               idx          = pj[j];
2002628f99d7SShri Abhyankar               rtmp11[idx] -= mul1 * tmp;
2003628f99d7SShri Abhyankar             }
2004628f99d7SShri Abhyankar           }
2005628f99d7SShri Abhyankar           prow = *bjtmp++;
2006628f99d7SShri Abhyankar         }
2007628f99d7SShri Abhyankar         pj  = bj + bi[row];
2008628f99d7SShri Abhyankar         pc1 = ba + bi[row];
2009628f99d7SShri Abhyankar 
2010628f99d7SShri Abhyankar         sctx.pv     = rtmp11[row];
2011628f99d7SShri Abhyankar         rtmp11[row] = 1.0/rtmp11[row]; /* invert diag */
2012628f99d7SShri Abhyankar         rs          = 0.0;
2013628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2014628f99d7SShri Abhyankar           idx    = pj[j];
2015628f99d7SShri Abhyankar           pc1[j] = rtmp11[idx]; /* rtmp11 -> ba */
2016628f99d7SShri Abhyankar           if (idx != row) rs += PetscAbsScalar(pc1[j]);
2017628f99d7SShri Abhyankar         }
2018628f99d7SShri Abhyankar         sctx.rs = rs;
20194cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,row);CHKERRQ(ierr);
202007b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2021628f99d7SShri Abhyankar         break;
2022628f99d7SShri Abhyankar 
2023628f99d7SShri Abhyankar       case 2:
2024628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2025628f99d7SShri Abhyankar           idx         = bjtmp[j];
2026628f99d7SShri Abhyankar           rtmp11[idx] = 0.0;
2027628f99d7SShri Abhyankar           rtmp22[idx] = 0.0;
2028628f99d7SShri Abhyankar         }
2029628f99d7SShri Abhyankar 
2030628f99d7SShri Abhyankar         /* load in initial (unfactored row) */
2031628f99d7SShri Abhyankar         idx    = r[row];
2032628f99d7SShri Abhyankar         nz_tmp = ai[idx+1] - ai[idx];
2033628f99d7SShri Abhyankar         ajtmp  = aj + ai[idx];
2034628f99d7SShri Abhyankar         v1     = aa + ai[idx];
2035628f99d7SShri Abhyankar         v2     = aa + ai[idx+1];
2036628f99d7SShri Abhyankar         for (j=0; j<nz_tmp; j++) {
2037628f99d7SShri Abhyankar           idx         = ics[ajtmp[j]];
2038628f99d7SShri Abhyankar           rtmp11[idx] = v1[j];
2039628f99d7SShri Abhyankar           rtmp22[idx] = v2[j];
2040628f99d7SShri Abhyankar         }
2041628f99d7SShri Abhyankar         rtmp11[ics[r[row]]]   += sctx.shift_amount;
2042628f99d7SShri Abhyankar         rtmp22[ics[r[row+1]]] += sctx.shift_amount;
2043628f99d7SShri Abhyankar 
2044628f99d7SShri Abhyankar         prow = *bjtmp++;
2045628f99d7SShri Abhyankar         while (prow < row) {
2046628f99d7SShri Abhyankar           pc1 = rtmp11 + prow;
2047628f99d7SShri Abhyankar           pc2 = rtmp22 + prow;
2048628f99d7SShri Abhyankar           if (*pc1 != 0.0 || *pc2 != 0.0) {
2049628f99d7SShri Abhyankar             pv   = ba + bd[prow];
2050628f99d7SShri Abhyankar             pj   = nbj + bd[prow];
2051628f99d7SShri Abhyankar             mul1 = *pc1 * *pv;
2052628f99d7SShri Abhyankar             mul2 = *pc2 * *pv;
2053628f99d7SShri Abhyankar             ++pv;
2054628f99d7SShri Abhyankar             *pc1 = mul1;
2055628f99d7SShri Abhyankar             *pc2 = mul2;
2056628f99d7SShri Abhyankar 
2057628f99d7SShri Abhyankar             nz_tmp = bi[prow+1] - bd[prow] - 1;
2058628f99d7SShri Abhyankar             for (j=0; j<nz_tmp; j++) {
2059628f99d7SShri Abhyankar               tmp          = pv[j];
2060628f99d7SShri Abhyankar               idx          = pj[j];
2061628f99d7SShri Abhyankar               rtmp11[idx] -= mul1 * tmp;
2062628f99d7SShri Abhyankar               rtmp22[idx] -= mul2 * tmp;
2063628f99d7SShri Abhyankar             }
2064ca0c957dSBarry Smith             ierr = PetscLogFlops(2+4.0*nz_tmp);CHKERRQ(ierr);
2065628f99d7SShri Abhyankar           }
2066628f99d7SShri Abhyankar           prow = *bjtmp++;
2067628f99d7SShri Abhyankar         }
2068628f99d7SShri Abhyankar 
2069628f99d7SShri Abhyankar         /* Now take care of diagonal 2x2 block. Note: prow = row here */
2070628f99d7SShri Abhyankar         pc1 = rtmp11 + prow;
2071628f99d7SShri Abhyankar         pc2 = rtmp22 + prow;
2072628f99d7SShri Abhyankar 
2073628f99d7SShri Abhyankar         sctx.pv = *pc1;
2074628f99d7SShri Abhyankar         pj      = bj + bi[prow];
2075628f99d7SShri Abhyankar         rs      = 0.0;
2076628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2077628f99d7SShri Abhyankar           idx = pj[j];
2078628f99d7SShri Abhyankar           if (idx != prow) rs += PetscAbsScalar(rtmp11[idx]);
2079628f99d7SShri Abhyankar         }
2080628f99d7SShri Abhyankar         sctx.rs = rs;
20814cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,row);CHKERRQ(ierr);
208207b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2083628f99d7SShri Abhyankar 
2084628f99d7SShri Abhyankar         if (*pc2 != 0.0) {
2085628f99d7SShri Abhyankar           pj     = nbj + bd[prow];
2086628f99d7SShri Abhyankar           mul2   = (*pc2)/(*pc1); /* since diag is not yet inverted.*/
2087628f99d7SShri Abhyankar           *pc2   = mul2;
2088628f99d7SShri Abhyankar           nz_tmp = bi[prow+1] - bd[prow] - 1;
2089628f99d7SShri Abhyankar           for (j=0; j<nz_tmp; j++) {
2090628f99d7SShri Abhyankar             idx          = pj[j];
2091628f99d7SShri Abhyankar             tmp          = rtmp11[idx];
2092628f99d7SShri Abhyankar             rtmp22[idx] -= mul2 * tmp;
2093628f99d7SShri Abhyankar           }
2094ca0c957dSBarry Smith           ierr = PetscLogFlops(1+2.0*nz_tmp);CHKERRQ(ierr);
2095628f99d7SShri Abhyankar         }
2096628f99d7SShri Abhyankar 
2097628f99d7SShri Abhyankar         pj  = bj + bi[row];
2098628f99d7SShri Abhyankar         pc1 = ba + bi[row];
2099628f99d7SShri Abhyankar         pc2 = ba + bi[row+1];
2100628f99d7SShri Abhyankar 
2101628f99d7SShri Abhyankar         sctx.pv       = rtmp22[row+1];
2102628f99d7SShri Abhyankar         rs            = 0.0;
2103628f99d7SShri Abhyankar         rtmp11[row]   = 1.0/rtmp11[row];
2104628f99d7SShri Abhyankar         rtmp22[row+1] = 1.0/rtmp22[row+1];
2105628f99d7SShri Abhyankar         /* copy row entries from dense representation to sparse */
2106628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2107628f99d7SShri Abhyankar           idx    = pj[j];
2108628f99d7SShri Abhyankar           pc1[j] = rtmp11[idx];
2109628f99d7SShri Abhyankar           pc2[j] = rtmp22[idx];
2110628f99d7SShri Abhyankar           if (idx != row+1) rs += PetscAbsScalar(pc2[j]);
2111628f99d7SShri Abhyankar         }
2112628f99d7SShri Abhyankar         sctx.rs = rs;
21134cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,row+1);CHKERRQ(ierr);
211407b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2115628f99d7SShri Abhyankar         break;
2116628f99d7SShri Abhyankar 
2117628f99d7SShri Abhyankar       case 3:
2118628f99d7SShri Abhyankar         for  (j=0; j<nz; j++) {
2119628f99d7SShri Abhyankar           idx         = bjtmp[j];
2120628f99d7SShri Abhyankar           rtmp11[idx] = 0.0;
2121628f99d7SShri Abhyankar           rtmp22[idx] = 0.0;
2122628f99d7SShri Abhyankar           rtmp33[idx] = 0.0;
2123628f99d7SShri Abhyankar         }
2124628f99d7SShri Abhyankar         /* copy the nonzeros for the 3 rows from sparse representation to dense in rtmp*[] */
2125628f99d7SShri Abhyankar         idx    = r[row];
2126628f99d7SShri Abhyankar         nz_tmp = ai[idx+1] - ai[idx];
2127628f99d7SShri Abhyankar         ajtmp  = aj + ai[idx];
2128628f99d7SShri Abhyankar         v1     = aa + ai[idx];
2129628f99d7SShri Abhyankar         v2     = aa + ai[idx+1];
2130628f99d7SShri Abhyankar         v3     = aa + ai[idx+2];
2131628f99d7SShri Abhyankar         for (j=0; j<nz_tmp; j++) {
2132628f99d7SShri Abhyankar           idx         = ics[ajtmp[j]];
2133628f99d7SShri Abhyankar           rtmp11[idx] = v1[j];
2134628f99d7SShri Abhyankar           rtmp22[idx] = v2[j];
2135628f99d7SShri Abhyankar           rtmp33[idx] = v3[j];
2136628f99d7SShri Abhyankar         }
2137628f99d7SShri Abhyankar         rtmp11[ics[r[row]]]   += sctx.shift_amount;
2138628f99d7SShri Abhyankar         rtmp22[ics[r[row+1]]] += sctx.shift_amount;
2139628f99d7SShri Abhyankar         rtmp33[ics[r[row+2]]] += sctx.shift_amount;
2140628f99d7SShri Abhyankar 
2141628f99d7SShri Abhyankar         /* loop over all pivot row blocks above this row block */
2142628f99d7SShri Abhyankar         prow = *bjtmp++;
2143628f99d7SShri Abhyankar         while (prow < row) {
2144628f99d7SShri Abhyankar           pc1 = rtmp11 + prow;
2145628f99d7SShri Abhyankar           pc2 = rtmp22 + prow;
2146628f99d7SShri Abhyankar           pc3 = rtmp33 + prow;
2147628f99d7SShri Abhyankar           if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 !=0.0) {
2148628f99d7SShri Abhyankar             pv   = ba  + bd[prow];
2149628f99d7SShri Abhyankar             pj   = nbj + bd[prow];
2150628f99d7SShri Abhyankar             mul1 = *pc1 * *pv;
2151628f99d7SShri Abhyankar             mul2 = *pc2 * *pv;
2152628f99d7SShri Abhyankar             mul3 = *pc3 * *pv;
2153628f99d7SShri Abhyankar             ++pv;
2154628f99d7SShri Abhyankar             *pc1 = mul1;
2155628f99d7SShri Abhyankar             *pc2 = mul2;
2156628f99d7SShri Abhyankar             *pc3 = mul3;
2157628f99d7SShri Abhyankar 
2158628f99d7SShri Abhyankar             nz_tmp = bi[prow+1] - bd[prow] - 1;
2159628f99d7SShri Abhyankar             /* update this row based on pivot row */
2160628f99d7SShri Abhyankar             for (j=0; j<nz_tmp; j++) {
2161628f99d7SShri Abhyankar               tmp          = pv[j];
2162628f99d7SShri Abhyankar               idx          = pj[j];
2163628f99d7SShri Abhyankar               rtmp11[idx] -= mul1 * tmp;
2164628f99d7SShri Abhyankar               rtmp22[idx] -= mul2 * tmp;
2165628f99d7SShri Abhyankar               rtmp33[idx] -= mul3 * tmp;
2166628f99d7SShri Abhyankar             }
2167ca0c957dSBarry Smith             ierr = PetscLogFlops(3+6.0*nz_tmp);CHKERRQ(ierr);
2168628f99d7SShri Abhyankar           }
2169628f99d7SShri Abhyankar           prow = *bjtmp++;
2170628f99d7SShri Abhyankar         }
2171628f99d7SShri Abhyankar 
2172628f99d7SShri Abhyankar         /* Now take care of diagonal 3x3 block in this set of rows */
2173628f99d7SShri Abhyankar         /* note: prow = row here */
2174628f99d7SShri Abhyankar         pc1 = rtmp11 + prow;
2175628f99d7SShri Abhyankar         pc2 = rtmp22 + prow;
2176628f99d7SShri Abhyankar         pc3 = rtmp33 + prow;
2177628f99d7SShri Abhyankar 
2178628f99d7SShri Abhyankar         sctx.pv = *pc1;
2179628f99d7SShri Abhyankar         pj      = bj + bi[prow];
2180628f99d7SShri Abhyankar         rs      = 0.0;
2181628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2182628f99d7SShri Abhyankar           idx = pj[j];
2183628f99d7SShri Abhyankar           if (idx != row) rs += PetscAbsScalar(rtmp11[idx]);
2184628f99d7SShri Abhyankar         }
2185628f99d7SShri Abhyankar         sctx.rs = rs;
21864cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,row);CHKERRQ(ierr);
218707b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2188628f99d7SShri Abhyankar 
2189628f99d7SShri Abhyankar         if (*pc2 != 0.0 || *pc3 != 0.0) {
2190628f99d7SShri Abhyankar           mul2   = (*pc2)/(*pc1);
2191628f99d7SShri Abhyankar           mul3   = (*pc3)/(*pc1);
2192628f99d7SShri Abhyankar           *pc2   = mul2;
2193628f99d7SShri Abhyankar           *pc3   = mul3;
2194628f99d7SShri Abhyankar           nz_tmp = bi[prow+1] - bd[prow] - 1;
2195628f99d7SShri Abhyankar           pj     = nbj + bd[prow];
2196628f99d7SShri Abhyankar           for (j=0; j<nz_tmp; j++) {
2197628f99d7SShri Abhyankar             idx          = pj[j];
2198628f99d7SShri Abhyankar             tmp          = rtmp11[idx];
2199628f99d7SShri Abhyankar             rtmp22[idx] -= mul2 * tmp;
2200628f99d7SShri Abhyankar             rtmp33[idx] -= mul3 * tmp;
2201628f99d7SShri Abhyankar           }
2202ca0c957dSBarry Smith           ierr = PetscLogFlops(2+4.0*nz_tmp);CHKERRQ(ierr);
2203628f99d7SShri Abhyankar         }
2204628f99d7SShri Abhyankar         ++prow;
2205628f99d7SShri Abhyankar 
2206628f99d7SShri Abhyankar         pc2     = rtmp22 + prow;
2207628f99d7SShri Abhyankar         pc3     = rtmp33 + prow;
2208628f99d7SShri Abhyankar         sctx.pv = *pc2;
2209628f99d7SShri Abhyankar         pj      = bj + bi[prow];
2210628f99d7SShri Abhyankar         rs      = 0.0;
2211628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2212628f99d7SShri Abhyankar           idx = pj[j];
2213628f99d7SShri Abhyankar           if (idx != prow) rs += PetscAbsScalar(rtmp22[idx]);
2214628f99d7SShri Abhyankar         }
2215628f99d7SShri Abhyankar         sctx.rs = rs;
22164cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,row+1);CHKERRQ(ierr);
221707b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2218628f99d7SShri Abhyankar 
2219628f99d7SShri Abhyankar         if (*pc3 != 0.0) {
2220628f99d7SShri Abhyankar           mul3   = (*pc3)/(*pc2);
2221628f99d7SShri Abhyankar           *pc3   = mul3;
2222628f99d7SShri Abhyankar           pj     = nbj + bd[prow];
2223628f99d7SShri Abhyankar           nz_tmp = bi[prow+1] - bd[prow] - 1;
2224628f99d7SShri Abhyankar           for (j=0; j<nz_tmp; j++) {
2225628f99d7SShri Abhyankar             idx          = pj[j];
2226628f99d7SShri Abhyankar             tmp          = rtmp22[idx];
2227628f99d7SShri Abhyankar             rtmp33[idx] -= mul3 * tmp;
2228628f99d7SShri Abhyankar           }
2229ca0c957dSBarry Smith           ierr = PetscLogFlops(1+2.0*nz_tmp);CHKERRQ(ierr);
2230628f99d7SShri Abhyankar         }
2231628f99d7SShri Abhyankar 
2232628f99d7SShri Abhyankar         pj  = bj + bi[row];
2233628f99d7SShri Abhyankar         pc1 = ba + bi[row];
2234628f99d7SShri Abhyankar         pc2 = ba + bi[row+1];
2235628f99d7SShri Abhyankar         pc3 = ba + bi[row+2];
2236628f99d7SShri Abhyankar 
2237628f99d7SShri Abhyankar         sctx.pv       = rtmp33[row+2];
2238628f99d7SShri Abhyankar         rs            = 0.0;
2239628f99d7SShri Abhyankar         rtmp11[row]   = 1.0/rtmp11[row];
2240628f99d7SShri Abhyankar         rtmp22[row+1] = 1.0/rtmp22[row+1];
2241628f99d7SShri Abhyankar         rtmp33[row+2] = 1.0/rtmp33[row+2];
2242628f99d7SShri Abhyankar         /* copy row entries from dense representation to sparse */
2243628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2244628f99d7SShri Abhyankar           idx    = pj[j];
2245628f99d7SShri Abhyankar           pc1[j] = rtmp11[idx];
2246628f99d7SShri Abhyankar           pc2[j] = rtmp22[idx];
2247628f99d7SShri Abhyankar           pc3[j] = rtmp33[idx];
2248628f99d7SShri Abhyankar           if (idx != row+2) rs += PetscAbsScalar(pc3[j]);
2249628f99d7SShri Abhyankar         }
2250628f99d7SShri Abhyankar 
2251628f99d7SShri Abhyankar         sctx.rs = rs;
22524cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,row+2);CHKERRQ(ierr);
225307b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2254628f99d7SShri Abhyankar         break;
2255628f99d7SShri Abhyankar 
2256628f99d7SShri Abhyankar       default:
2257e32f2f54SBarry Smith         SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Node size not yet supported \n");
2258628f99d7SShri Abhyankar       }
2259628f99d7SShri Abhyankar       row += nodesz;                 /* Update the row */
2260628f99d7SShri Abhyankar     }
2261628f99d7SShri Abhyankar endofwhile:;
226207b50cabSHong Zhang   } while (sctx.newshift);
2263eb9baa12SBarry Smith   ierr = PetscFree3(rtmp11,rtmp22,rtmp33);CHKERRQ(ierr);
2264628f99d7SShri Abhyankar   ierr = PetscFree(tmp_vec2);CHKERRQ(ierr);
2265628f99d7SShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2266628f99d7SShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
2267628f99d7SShri Abhyankar   ierr = ISRestoreIndices(iscol,&c);CHKERRQ(ierr);
22682205254eSKarl Rupp 
2269d3ac4fa3SBarry Smith   (B)->ops->solve = MatSolve_SeqAIJ_inplace;
2270628f99d7SShri Abhyankar   /* do not set solve add, since MatSolve_Inode + Add is faster */
2271628f99d7SShri Abhyankar   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
2272628f99d7SShri Abhyankar   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
2273628f99d7SShri Abhyankar   C->assembled              = PETSC_TRUE;
2274628f99d7SShri Abhyankar   C->preallocated           = PETSC_TRUE;
2275628f99d7SShri Abhyankar   if (sctx.nshift) {
2276f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
227757622a8eSBarry 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);
2278f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
227957622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
2280628f99d7SShri Abhyankar     }
2281628f99d7SShri Abhyankar   }
2282628f99d7SShri Abhyankar   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
2283a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode(C);CHKERRQ(ierr);
2284628f99d7SShri Abhyankar   PetscFunctionReturn(0);
2285628f99d7SShri Abhyankar }
2286628f99d7SShri Abhyankar 
2287628f99d7SShri Abhyankar 
2288019b515eSShri Abhyankar /* ----------------------------------------------------------- */
228935233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_Inode(Mat A,Vec bb,Vec xx)
2290019b515eSShri Abhyankar {
2291019b515eSShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
2292019b515eSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
2293019b515eSShri Abhyankar   PetscErrorCode    ierr;
2294019b515eSShri Abhyankar   const PetscInt    *r,*c,*rout,*cout;
22958758e1faSBarry Smith   PetscInt          i,j,n = A->rmap->n;
22968758e1faSBarry Smith   PetscInt          node_max,row,nsz,aii,i0,i1,nz;
22978758e1faSBarry Smith   const PetscInt    *ai = a->i,*a_j = a->j,*ns,*vi,*ad,*aj;
2298019b515eSShri Abhyankar   PetscScalar       *x,*tmp,*tmps,tmp0,tmp1;
2299019b515eSShri Abhyankar   PetscScalar       sum1,sum2,sum3,sum4,sum5;
2300019b515eSShri Abhyankar   const MatScalar   *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa;
2301019b515eSShri Abhyankar   const PetscScalar *b;
2302019b515eSShri Abhyankar 
2303019b515eSShri Abhyankar   PetscFunctionBegin;
2304e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
2305019b515eSShri Abhyankar   node_max = a->inode.node_count;
2306019b515eSShri Abhyankar   ns       = a->inode.size;     /* Node Size array */
2307019b515eSShri Abhyankar 
23083649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
2309fdc842d1SBarry Smith   ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr);
2310019b515eSShri Abhyankar   tmp  = a->solve_work;
2311019b515eSShri Abhyankar 
2312019b515eSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
2313019b515eSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
2314019b515eSShri Abhyankar 
2315019b515eSShri Abhyankar   /* forward solve the lower triangular */
2316019b515eSShri Abhyankar   tmps = tmp;
2317019b515eSShri Abhyankar   aa   = a_a;
2318019b515eSShri Abhyankar   aj   = a_j;
2319019b515eSShri Abhyankar   ad   = a->diag;
2320019b515eSShri Abhyankar 
2321019b515eSShri Abhyankar   for (i = 0,row = 0; i< node_max; ++i) {
2322019b515eSShri Abhyankar     nsz = ns[i];
2323019b515eSShri Abhyankar     aii = ai[row];
2324019b515eSShri Abhyankar     v1  = aa + aii;
2325019b515eSShri Abhyankar     vi  = aj + aii;
2326019b515eSShri Abhyankar     nz  = ai[row+1]- ai[row];
2327019b515eSShri Abhyankar 
232898991853SShri Abhyankar     if (i < node_max-1) {
232998991853SShri Abhyankar       /* Prefetch the indices for the next block */
233050d8bf02SJed Brown       PetscPrefetchBlock(aj+ai[row+nsz],ai[row+nsz+1]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); /* indices */
233198991853SShri Abhyankar       /* Prefetch the data for the next block */
233250d8bf02SJed Brown       PetscPrefetchBlock(aa+ai[row+nsz],ai[row+nsz+ns[i+1]]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA);
233398991853SShri Abhyankar     }
233498991853SShri Abhyankar 
2335019b515eSShri Abhyankar     switch (nsz) {               /* Each loop in 'case' is unrolled */
2336019b515eSShri Abhyankar     case 1:
2337019b515eSShri Abhyankar       sum1 = b[r[row]];
2338019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2339019b515eSShri Abhyankar         i0    = vi[j];
2340019b515eSShri Abhyankar         i1    = vi[j+1];
2341019b515eSShri Abhyankar         tmp0  = tmps[i0];
2342019b515eSShri Abhyankar         tmp1  = tmps[i1];
2343019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0 + v1[j+1]*tmp1;
2344019b515eSShri Abhyankar       }
2345019b515eSShri Abhyankar       if (j == nz-1) {
2346019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2347019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0;
2348019b515eSShri Abhyankar       }
2349019b515eSShri Abhyankar       tmp[row++]=sum1;
2350019b515eSShri Abhyankar       break;
2351019b515eSShri Abhyankar     case 2:
2352019b515eSShri Abhyankar       sum1 = b[r[row]];
2353019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2354019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2355019b515eSShri Abhyankar 
2356019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2357019b515eSShri Abhyankar         i0    = vi[j];
2358019b515eSShri Abhyankar         i1    = vi[j+1];
2359019b515eSShri Abhyankar         tmp0  = tmps[i0];
2360019b515eSShri Abhyankar         tmp1  = tmps[i1];
2361019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2362019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2363019b515eSShri Abhyankar       }
2364019b515eSShri Abhyankar       if (j == nz-1) {
2365019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2366019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2367019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2368019b515eSShri Abhyankar       }
2369019b515eSShri Abhyankar       sum2     -= v2[nz] * sum1;
2370019b515eSShri Abhyankar       tmp[row++]=sum1;
2371019b515eSShri Abhyankar       tmp[row++]=sum2;
2372019b515eSShri Abhyankar       break;
2373019b515eSShri Abhyankar     case 3:
2374019b515eSShri Abhyankar       sum1 = b[r[row]];
2375019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2376019b515eSShri Abhyankar       sum3 = b[r[row+2]];
2377019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2378019b515eSShri Abhyankar       v3   = aa + ai[row+2];
2379019b515eSShri Abhyankar 
2380019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2381019b515eSShri Abhyankar         i0    = vi[j];
2382019b515eSShri Abhyankar         i1    = vi[j+1];
2383019b515eSShri Abhyankar         tmp0  = tmps[i0];
2384019b515eSShri Abhyankar         tmp1  = tmps[i1];
2385019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2386019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2387019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
2388019b515eSShri Abhyankar       }
2389019b515eSShri Abhyankar       if (j == nz-1) {
2390019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2391019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2392019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2393019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
2394019b515eSShri Abhyankar       }
2395019b515eSShri Abhyankar       sum2     -= v2[nz] * sum1;
2396019b515eSShri Abhyankar       sum3     -= v3[nz] * sum1;
2397019b515eSShri Abhyankar       sum3     -= v3[nz+1] * sum2;
2398019b515eSShri Abhyankar       tmp[row++]=sum1;
2399019b515eSShri Abhyankar       tmp[row++]=sum2;
2400019b515eSShri Abhyankar       tmp[row++]=sum3;
2401019b515eSShri Abhyankar       break;
2402019b515eSShri Abhyankar 
2403019b515eSShri Abhyankar     case 4:
2404019b515eSShri Abhyankar       sum1 = b[r[row]];
2405019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2406019b515eSShri Abhyankar       sum3 = b[r[row+2]];
2407019b515eSShri Abhyankar       sum4 = b[r[row+3]];
2408019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2409019b515eSShri Abhyankar       v3   = aa + ai[row+2];
2410019b515eSShri Abhyankar       v4   = aa + ai[row+3];
2411019b515eSShri Abhyankar 
2412019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2413019b515eSShri Abhyankar         i0    = vi[j];
2414019b515eSShri Abhyankar         i1    = vi[j+1];
2415019b515eSShri Abhyankar         tmp0  = tmps[i0];
2416019b515eSShri Abhyankar         tmp1  = tmps[i1];
2417019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2418019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2419019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
2420019b515eSShri Abhyankar         sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1;
2421019b515eSShri Abhyankar       }
2422019b515eSShri Abhyankar       if (j == nz-1) {
2423019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2424019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2425019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2426019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
2427019b515eSShri Abhyankar         sum4 -= v4[j] *tmp0;
2428019b515eSShri Abhyankar       }
2429019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
2430019b515eSShri Abhyankar       sum3 -= v3[nz] * sum1;
2431019b515eSShri Abhyankar       sum4 -= v4[nz] * sum1;
2432019b515eSShri Abhyankar       sum3 -= v3[nz+1] * sum2;
2433019b515eSShri Abhyankar       sum4 -= v4[nz+1] * sum2;
2434019b515eSShri Abhyankar       sum4 -= v4[nz+2] * sum3;
2435019b515eSShri Abhyankar 
2436019b515eSShri Abhyankar       tmp[row++]=sum1;
2437019b515eSShri Abhyankar       tmp[row++]=sum2;
2438019b515eSShri Abhyankar       tmp[row++]=sum3;
2439019b515eSShri Abhyankar       tmp[row++]=sum4;
2440019b515eSShri Abhyankar       break;
2441019b515eSShri Abhyankar     case 5:
2442019b515eSShri Abhyankar       sum1 = b[r[row]];
2443019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2444019b515eSShri Abhyankar       sum3 = b[r[row+2]];
2445019b515eSShri Abhyankar       sum4 = b[r[row+3]];
2446019b515eSShri Abhyankar       sum5 = b[r[row+4]];
2447019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2448019b515eSShri Abhyankar       v3   = aa + ai[row+2];
2449019b515eSShri Abhyankar       v4   = aa + ai[row+3];
2450019b515eSShri Abhyankar       v5   = aa + ai[row+4];
2451019b515eSShri Abhyankar 
2452019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2453019b515eSShri Abhyankar         i0    = vi[j];
2454019b515eSShri Abhyankar         i1    = vi[j+1];
2455019b515eSShri Abhyankar         tmp0  = tmps[i0];
2456019b515eSShri Abhyankar         tmp1  = tmps[i1];
2457019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2458019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2459019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
2460019b515eSShri Abhyankar         sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1;
2461019b515eSShri Abhyankar         sum5 -= v5[j] * tmp0 + v5[j+1] * tmp1;
2462019b515eSShri Abhyankar       }
2463019b515eSShri Abhyankar       if (j == nz-1) {
2464019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2465019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2466019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2467019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
2468019b515eSShri Abhyankar         sum4 -= v4[j] *tmp0;
2469019b515eSShri Abhyankar         sum5 -= v5[j] *tmp0;
2470019b515eSShri Abhyankar       }
2471019b515eSShri Abhyankar 
2472019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
2473019b515eSShri Abhyankar       sum3 -= v3[nz] * sum1;
2474019b515eSShri Abhyankar       sum4 -= v4[nz] * sum1;
2475019b515eSShri Abhyankar       sum5 -= v5[nz] * sum1;
2476019b515eSShri Abhyankar       sum3 -= v3[nz+1] * sum2;
2477019b515eSShri Abhyankar       sum4 -= v4[nz+1] * sum2;
2478019b515eSShri Abhyankar       sum5 -= v5[nz+1] * sum2;
2479019b515eSShri Abhyankar       sum4 -= v4[nz+2] * sum3;
2480019b515eSShri Abhyankar       sum5 -= v5[nz+2] * sum3;
2481019b515eSShri Abhyankar       sum5 -= v5[nz+3] * sum4;
2482019b515eSShri Abhyankar 
2483019b515eSShri Abhyankar       tmp[row++]=sum1;
2484019b515eSShri Abhyankar       tmp[row++]=sum2;
2485019b515eSShri Abhyankar       tmp[row++]=sum3;
2486019b515eSShri Abhyankar       tmp[row++]=sum4;
2487019b515eSShri Abhyankar       tmp[row++]=sum5;
2488019b515eSShri Abhyankar       break;
2489019b515eSShri Abhyankar     default:
2490e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
2491019b515eSShri Abhyankar     }
2492019b515eSShri Abhyankar   }
2493019b515eSShri Abhyankar   /* backward solve the upper triangular */
2494019b515eSShri Abhyankar   for (i=node_max -1,row = n-1; i>=0; i--) {
2495019b515eSShri Abhyankar     nsz = ns[i];
2496019b515eSShri Abhyankar     aii = ad[row+1] + 1;
2497019b515eSShri Abhyankar     v1  = aa + aii;
2498019b515eSShri Abhyankar     vi  = aj + aii;
2499019b515eSShri Abhyankar     nz  = ad[row]- ad[row+1] - 1;
250098991853SShri Abhyankar 
250198991853SShri Abhyankar     if (i > 0) {
250298991853SShri Abhyankar       /* Prefetch the indices for the next block */
250350d8bf02SJed Brown       PetscPrefetchBlock(aj+ad[row-nsz+1]+1,ad[row-nsz]-ad[row-nsz+1],0,PETSC_PREFETCH_HINT_NTA);
250498991853SShri Abhyankar       /* Prefetch the data for the next block */
250550d8bf02SJed Brown       PetscPrefetchBlock(aa+ad[row-nsz+1]+1,ad[row-nsz-ns[i-1]+1]-ad[row-nsz+1],0,PETSC_PREFETCH_HINT_NTA);
250698991853SShri Abhyankar     }
250798991853SShri Abhyankar 
2508019b515eSShri Abhyankar     switch (nsz) {               /* Each loop in 'case' is unrolled */
2509019b515eSShri Abhyankar     case 1:
2510019b515eSShri Abhyankar       sum1 = tmp[row];
2511019b515eSShri Abhyankar 
2512019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2513019b515eSShri Abhyankar         i0    = vi[j];
2514019b515eSShri Abhyankar         i1    = vi[j+1];
2515019b515eSShri Abhyankar         tmp0  = tmps[i0];
2516019b515eSShri Abhyankar         tmp1  = tmps[i1];
2517019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2518019b515eSShri Abhyankar       }
2519019b515eSShri Abhyankar       if (j == nz-1) {
2520019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2521019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0;
2522019b515eSShri Abhyankar       }
2523019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2524019b515eSShri Abhyankar       break;
2525019b515eSShri Abhyankar     case 2:
2526019b515eSShri Abhyankar       sum1 = tmp[row];
2527019b515eSShri Abhyankar       sum2 = tmp[row-1];
2528019b515eSShri Abhyankar       v2   = aa + ad[row] + 1;
2529019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2530019b515eSShri Abhyankar         i0    = vi[j];
2531019b515eSShri Abhyankar         i1    = vi[j+1];
2532019b515eSShri Abhyankar         tmp0  = tmps[i0];
2533019b515eSShri Abhyankar         tmp1  = tmps[i1];
2534019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2535019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2536019b515eSShri Abhyankar       }
2537019b515eSShri Abhyankar       if (j == nz-1) {
2538019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2539019b515eSShri Abhyankar         sum1 -= v1[j]* tmp0;
2540019b515eSShri Abhyankar         sum2 -= v2[j+1]* tmp0;
2541019b515eSShri Abhyankar       }
2542019b515eSShri Abhyankar 
2543019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2544019b515eSShri Abhyankar       sum2     -= v2[0] * tmp0;
2545019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2546019b515eSShri Abhyankar       break;
2547019b515eSShri Abhyankar     case 3:
2548019b515eSShri Abhyankar       sum1 = tmp[row];
2549019b515eSShri Abhyankar       sum2 = tmp[row -1];
2550019b515eSShri Abhyankar       sum3 = tmp[row -2];
2551019b515eSShri Abhyankar       v2   = aa + ad[row] + 1;
2552019b515eSShri Abhyankar       v3   = aa + ad[row -1] + 1;
2553019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2554019b515eSShri Abhyankar         i0    = vi[j];
2555019b515eSShri Abhyankar         i1    = vi[j+1];
2556019b515eSShri Abhyankar         tmp0  = tmps[i0];
2557019b515eSShri Abhyankar         tmp1  = tmps[i1];
2558019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2559019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2560019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
2561019b515eSShri Abhyankar       }
2562019b515eSShri Abhyankar       if (j== nz-1) {
2563019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2564019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
2565019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
2566019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
2567019b515eSShri Abhyankar       }
2568019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2569019b515eSShri Abhyankar       sum2     -= v2[0]* tmp0;
2570019b515eSShri Abhyankar       sum3     -= v3[1] * tmp0;
2571019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2572019b515eSShri Abhyankar       sum3     -= v3[0]* tmp0;
2573019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
2574019b515eSShri Abhyankar 
2575019b515eSShri Abhyankar       break;
2576019b515eSShri Abhyankar     case 4:
2577019b515eSShri Abhyankar       sum1 = tmp[row];
2578019b515eSShri Abhyankar       sum2 = tmp[row -1];
2579019b515eSShri Abhyankar       sum3 = tmp[row -2];
2580019b515eSShri Abhyankar       sum4 = tmp[row -3];
2581019b515eSShri Abhyankar       v2   = aa + ad[row]+1;
2582019b515eSShri Abhyankar       v3   = aa + ad[row -1]+1;
2583019b515eSShri Abhyankar       v4   = aa + ad[row -2]+1;
2584019b515eSShri Abhyankar 
2585019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2586019b515eSShri Abhyankar         i0    = vi[j];
2587019b515eSShri Abhyankar         i1    = vi[j+1];
2588019b515eSShri Abhyankar         tmp0  = tmps[i0];
2589019b515eSShri Abhyankar         tmp1  = tmps[i1];
2590019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0   + v1[j+1] * tmp1;
2591019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2592019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
2593019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1;
2594019b515eSShri Abhyankar       }
2595019b515eSShri Abhyankar       if (j== nz-1) {
2596019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2597019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
2598019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
2599019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
2600019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0;
2601019b515eSShri Abhyankar       }
2602019b515eSShri Abhyankar 
2603019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2604019b515eSShri Abhyankar       sum2     -= v2[0] * tmp0;
2605019b515eSShri Abhyankar       sum3     -= v3[1] * tmp0;
2606019b515eSShri Abhyankar       sum4     -= v4[2] * tmp0;
2607019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2608019b515eSShri Abhyankar       sum3     -= v3[0] * tmp0;
2609019b515eSShri Abhyankar       sum4     -= v4[1] * tmp0;
2610019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
2611019b515eSShri Abhyankar       sum4     -= v4[0] * tmp0;
2612019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--;
2613019b515eSShri Abhyankar       break;
2614019b515eSShri Abhyankar     case 5:
2615019b515eSShri Abhyankar       sum1 = tmp[row];
2616019b515eSShri Abhyankar       sum2 = tmp[row -1];
2617019b515eSShri Abhyankar       sum3 = tmp[row -2];
2618019b515eSShri Abhyankar       sum4 = tmp[row -3];
2619019b515eSShri Abhyankar       sum5 = tmp[row -4];
2620019b515eSShri Abhyankar       v2   = aa + ad[row]+1;
2621019b515eSShri Abhyankar       v3   = aa + ad[row -1]+1;
2622019b515eSShri Abhyankar       v4   = aa + ad[row -2]+1;
2623019b515eSShri Abhyankar       v5   = aa + ad[row -3]+1;
2624019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2625019b515eSShri Abhyankar         i0    = vi[j];
2626019b515eSShri Abhyankar         i1    = vi[j+1];
2627019b515eSShri Abhyankar         tmp0  = tmps[i0];
2628019b515eSShri Abhyankar         tmp1  = tmps[i1];
2629019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2630019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2631019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
2632019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1;
2633019b515eSShri Abhyankar         sum5 -= v5[j+4] * tmp0 + v5[j+5] * tmp1;
2634019b515eSShri Abhyankar       }
2635019b515eSShri Abhyankar       if (j==nz-1) {
2636019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2637019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
2638019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
2639019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
2640019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0;
2641019b515eSShri Abhyankar         sum5 -= v5[j+4] * tmp0;
2642019b515eSShri Abhyankar       }
2643019b515eSShri Abhyankar 
2644019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2645019b515eSShri Abhyankar       sum2     -= v2[0] * tmp0;
2646019b515eSShri Abhyankar       sum3     -= v3[1] * tmp0;
2647019b515eSShri Abhyankar       sum4     -= v4[2] * tmp0;
2648019b515eSShri Abhyankar       sum5     -= v5[3] * tmp0;
2649019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2650019b515eSShri Abhyankar       sum3     -= v3[0] * tmp0;
2651019b515eSShri Abhyankar       sum4     -= v4[1] * tmp0;
2652019b515eSShri Abhyankar       sum5     -= v5[2] * tmp0;
2653019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
2654019b515eSShri Abhyankar       sum4     -= v4[0] * tmp0;
2655019b515eSShri Abhyankar       sum5     -= v5[1] * tmp0;
2656019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--;
2657019b515eSShri Abhyankar       sum5     -= v5[0] * tmp0;
2658019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum5*v5[nz+4]; row--;
2659019b515eSShri Abhyankar       break;
2660019b515eSShri Abhyankar     default:
2661e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
2662019b515eSShri Abhyankar     }
2663019b515eSShri Abhyankar   }
2664019b515eSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
2665019b515eSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
26663649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
2667fdc842d1SBarry Smith   ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr);
2668019b515eSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
2669019b515eSShri Abhyankar   PetscFunctionReturn(0);
2670019b515eSShri Abhyankar }
2671019b515eSShri Abhyankar 
2672019b515eSShri Abhyankar 
26734c1414c8SBarry Smith /*
26744c1414c8SBarry Smith      Makes a longer coloring[] array and calls the usual code with that
26754c1414c8SBarry Smith */
26764108e4d5SBarry Smith PetscErrorCode MatColoringPatch_SeqAIJ_Inode(Mat mat,PetscInt ncolors,PetscInt nin,ISColoringValue coloring[],ISColoring *iscoloring)
26774c1414c8SBarry Smith {
26784c1414c8SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)mat->data;
26794c1414c8SBarry Smith   PetscErrorCode  ierr;
2680d0f46423SBarry Smith   PetscInt        n = mat->cmap->n,m = a->inode.node_count,j,*ns = a->inode.size,row;
26814c1414c8SBarry Smith   PetscInt        *colorused,i;
26824c1414c8SBarry Smith   ISColoringValue *newcolor;
26834c1414c8SBarry Smith 
26844c1414c8SBarry Smith   PetscFunctionBegin;
2685b215bc84SStefano Zampini   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
2686854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&newcolor);CHKERRQ(ierr);
26874c1414c8SBarry Smith   /* loop over inodes, marking a color for each column*/
26884c1414c8SBarry Smith   row = 0;
26894c1414c8SBarry Smith   for (i=0; i<m; i++) {
26904c1414c8SBarry Smith     for (j=0; j<ns[i]; j++) {
26914c1414c8SBarry Smith       newcolor[row++] = coloring[i] + j*ncolors;
26924c1414c8SBarry Smith     }
26934c1414c8SBarry Smith   }
26944c1414c8SBarry Smith 
26954c1414c8SBarry Smith   /* eliminate unneeded colors */
26961795a4d1SJed Brown   ierr = PetscCalloc1(5*ncolors,&colorused);CHKERRQ(ierr);
26974c1414c8SBarry Smith   for (i=0; i<n; i++) {
26984c1414c8SBarry Smith     colorused[newcolor[i]] = 1;
26994c1414c8SBarry Smith   }
27004c1414c8SBarry Smith 
27014c1414c8SBarry Smith   for (i=1; i<5*ncolors; i++) {
27024c1414c8SBarry Smith     colorused[i] += colorused[i-1];
27034c1414c8SBarry Smith   }
27044c1414c8SBarry Smith   ncolors = colorused[5*ncolors-1];
27054c1414c8SBarry Smith   for (i=0; i<n; i++) {
270657815681SSatish Balay     newcolor[i] = colorused[newcolor[i]]-1;
27074c1414c8SBarry Smith   }
27084c1414c8SBarry Smith   ierr = PetscFree(colorused);CHKERRQ(ierr);
2709aaf3ff59SMatthew G. Knepley   ierr = ISColoringCreate(PetscObjectComm((PetscObject)mat),ncolors,n,newcolor,PETSC_OWN_POINTER,iscoloring);CHKERRQ(ierr);
27104c1414c8SBarry Smith   ierr = PetscFree(coloring);CHKERRQ(ierr);
27114c1414c8SBarry Smith   PetscFunctionReturn(0);
27124c1414c8SBarry Smith }
27134c1414c8SBarry Smith 
2714af0996ceSBarry Smith #include <petsc/private/kernels/blockinvert.h>
27152af78befSBarry Smith 
27164108e4d5SBarry Smith PetscErrorCode MatSOR_SeqAIJ_Inode(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx)
27172af78befSBarry Smith {
27182af78befSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
27197aaeff0aSMatthew G. Knepley   PetscScalar       sum1 = 0.0,sum2 = 0.0,sum3 = 0.0,sum4 = 0.0,sum5 = 0.0,tmp0,tmp1,tmp2,tmp3;
27205850ef23SBarry Smith   MatScalar         *ibdiag,*bdiag,work[25],*t;
2721a8b09249SBarry Smith   PetscScalar       *x,tmp4,tmp5,x1,x2,x3,x4,x5;
27227aaeff0aSMatthew G. Knepley   const MatScalar   *v = a->a,*v1 = NULL,*v2 = NULL,*v3 = NULL,*v4 = NULL,*v5 = NULL;
27235850ef23SBarry Smith   const PetscScalar *xb, *b;
27247b6c816cSBarry Smith   PetscReal         zeropivot = 100.*PETSC_MACHINE_EPSILON, shift = 0.0;
27252af78befSBarry Smith   PetscErrorCode    ierr;
27268758e1faSBarry Smith   PetscInt          n,m = a->inode.node_count,cnt = 0,i,j,row,i1,i2;
27278758e1faSBarry Smith   PetscInt          sz,k,ipvt[5];
27287b6c816cSBarry Smith   PetscBool         allowzeropivot,zeropivotdetected;
27298758e1faSBarry Smith   const PetscInt    *sizes = a->inode.size,*idx,*diag = a->diag,*ii = a->i;
27302af78befSBarry Smith 
27312af78befSBarry Smith   PetscFunctionBegin;
2732a455e926SHong Zhang   allowzeropivot = PetscNot(A->erroriffailure);
2733b215bc84SStefano Zampini   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
2734e32f2f54SBarry Smith   if (omega != 1.0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for omega != 1.0; use -mat_no_inode");
2735e32f2f54SBarry Smith   if (fshift != 0.0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for fshift != 0.0; use -mat_no_inode");
27362af78befSBarry Smith 
273771f1c65dSBarry Smith   if (!a->inode.ibdiagvalid) {
27382af78befSBarry Smith     if (!a->inode.ibdiag) {
27392af78befSBarry Smith       /* calculate space needed for diagonal blocks */
27402af78befSBarry Smith       for (i=0; i<m; i++) {
27412af78befSBarry Smith         cnt += sizes[i]*sizes[i];
27422af78befSBarry Smith       }
2743f0d39aaaSBarry Smith       a->inode.bdiagsize = cnt;
27442205254eSKarl Rupp 
2745dcca6d9dSJed Brown       ierr = PetscMalloc3(cnt,&a->inode.ibdiag,cnt,&a->inode.bdiag,A->rmap->n,&a->inode.ssor_work);CHKERRQ(ierr);
274671f1c65dSBarry Smith     }
274771f1c65dSBarry Smith 
274871f1c65dSBarry Smith     /* copy over the diagonal blocks and invert them */
27492af78befSBarry Smith     ibdiag = a->inode.ibdiag;
27502af78befSBarry Smith     bdiag  = a->inode.bdiag;
27512af78befSBarry Smith     cnt    = 0;
27522af78befSBarry Smith     for (i=0, row = 0; i<m; i++) {
27532af78befSBarry Smith       for (j=0; j<sizes[i]; j++) {
2754f0d39aaaSBarry Smith         for (k=0; k<sizes[i]; k++) {
2755f0d39aaaSBarry Smith           bdiag[cnt+k*sizes[i]+j] = v[diag[row+j] - j + k];
2756f0d39aaaSBarry Smith         }
27572af78befSBarry Smith       }
2758580bdb30SBarry Smith       ierr = PetscArraycpy(ibdiag+cnt,bdiag+cnt,sizes[i]*sizes[i]);CHKERRQ(ierr);
27592af78befSBarry Smith 
27602af78befSBarry Smith       switch (sizes[i]) {
27612af78befSBarry Smith       case 1:
27622af78befSBarry Smith         /* Create matrix data structure */
27638e0e2a9aSHong Zhang         if (PetscAbsScalar(ibdiag[cnt]) < zeropivot) {
27648e0e2a9aSHong Zhang           if (allowzeropivot) {
27657b6c816cSBarry Smith             A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT;
27667b6c816cSBarry Smith             A->factorerror_zeropivot_value = PetscAbsScalar(ibdiag[cnt]);
27677b6c816cSBarry Smith             A->factorerror_zeropivot_row   = row;
27688e0e2a9aSHong Zhang             ierr = PetscInfo1(A,"Zero pivot, row %D\n",row);CHKERRQ(ierr);
27698e0e2a9aSHong Zhang           } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot on row %D",row);
27708e0e2a9aSHong Zhang         }
277164c62002SMatthew Knepley         ibdiag[cnt] = 1.0/ibdiag[cnt];
27722af78befSBarry Smith         break;
27732af78befSBarry Smith       case 2:
2774a455e926SHong Zhang         ierr = PetscKernel_A_gets_inverse_A_2(ibdiag+cnt,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr);
27757b6c816cSBarry Smith         if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT;
27762af78befSBarry Smith         break;
27772af78befSBarry Smith       case 3:
2778a455e926SHong Zhang         ierr = PetscKernel_A_gets_inverse_A_3(ibdiag+cnt,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr);
27797b6c816cSBarry Smith         if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT;
27802af78befSBarry Smith         break;
27812af78befSBarry Smith       case 4:
2782a455e926SHong Zhang         ierr = PetscKernel_A_gets_inverse_A_4(ibdiag+cnt,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr);
27837b6c816cSBarry Smith         if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT;
27842af78befSBarry Smith         break;
27852af78befSBarry Smith       case 5:
2786a455e926SHong Zhang         ierr = PetscKernel_A_gets_inverse_A_5(ibdiag+cnt,ipvt,work,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr);
27877b6c816cSBarry Smith         if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT;
27882af78befSBarry Smith         break;
27892af78befSBarry Smith       default:
2790e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
27912af78befSBarry Smith       }
27922af78befSBarry Smith       cnt += sizes[i]*sizes[i];
27932af78befSBarry Smith       row += sizes[i];
27942af78befSBarry Smith     }
279571f1c65dSBarry Smith     a->inode.ibdiagvalid = PETSC_TRUE;
27962af78befSBarry Smith   }
27972af78befSBarry Smith   ibdiag = a->inode.ibdiag;
27982af78befSBarry Smith   bdiag  = a->inode.bdiag;
27995850ef23SBarry Smith   t      = a->inode.ssor_work;
28002af78befSBarry Smith 
28012af78befSBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
28023649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
28035850ef23SBarry Smith   /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */
28045850ef23SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
28052af78befSBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) {
28062af78befSBarry Smith 
28078862d2efSBarry Smith       for (i=0, row=0; i<m; i++) {
28088862d2efSBarry Smith         sz  = diag[row] - ii[row];
28098862d2efSBarry Smith         v1  = a->a + ii[row];
28108862d2efSBarry Smith         idx = a->j + ii[row];
28118862d2efSBarry Smith 
28124108e4d5SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
28138862d2efSBarry Smith         switch (sizes[i]) {
28148862d2efSBarry Smith         case 1:
28158862d2efSBarry Smith 
28168862d2efSBarry Smith           sum1 = b[row];
28178862d2efSBarry Smith           for (n = 0; n<sz-1; n+=2) {
28188862d2efSBarry Smith             i1    = idx[0];
28198862d2efSBarry Smith             i2    = idx[1];
28208862d2efSBarry Smith             idx  += 2;
28218862d2efSBarry Smith             tmp0  = x[i1];
28228862d2efSBarry Smith             tmp1  = x[i2];
28238862d2efSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
28248862d2efSBarry Smith           }
28258862d2efSBarry Smith 
28268862d2efSBarry Smith           if (n == sz-1) {
2827f0d39aaaSBarry Smith             tmp0  = x[*idx];
2828f0d39aaaSBarry Smith             sum1 -= *v1 * tmp0;
28298862d2efSBarry Smith           }
28305850ef23SBarry Smith           t[row]   = sum1;
28318862d2efSBarry Smith           x[row++] = sum1*(*ibdiag++);
28328862d2efSBarry Smith           break;
2833f0d39aaaSBarry Smith         case 2:
2834f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2835f0d39aaaSBarry Smith           sum1 = b[row];
2836f0d39aaaSBarry Smith           sum2 = b[row+1];
2837f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2838f0d39aaaSBarry Smith             i1    = idx[0];
2839f0d39aaaSBarry Smith             i2    = idx[1];
2840f0d39aaaSBarry Smith             idx  += 2;
2841f0d39aaaSBarry Smith             tmp0  = x[i1];
2842f0d39aaaSBarry Smith             tmp1  = x[i2];
2843f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2844f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2845f0d39aaaSBarry Smith           }
2846f0d39aaaSBarry Smith 
2847f0d39aaaSBarry Smith           if (n == sz-1) {
2848f0d39aaaSBarry Smith             tmp0  = x[*idx];
2849f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2850f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2851f0d39aaaSBarry Smith           }
28525850ef23SBarry Smith           t[row]   = sum1;
28535850ef23SBarry Smith           t[row+1] = sum2;
2854f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[2];
2855f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[3];
2856f0d39aaaSBarry Smith           ibdiag  += 4;
2857f0d39aaaSBarry Smith           break;
2858f0d39aaaSBarry Smith         case 3:
2859f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2860f0d39aaaSBarry Smith           v3   = a->a + ii[row+2];
2861f0d39aaaSBarry Smith           sum1 = b[row];
2862f0d39aaaSBarry Smith           sum2 = b[row+1];
2863f0d39aaaSBarry Smith           sum3 = b[row+2];
2864f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2865f0d39aaaSBarry Smith             i1    = idx[0];
2866f0d39aaaSBarry Smith             i2    = idx[1];
2867f0d39aaaSBarry Smith             idx  += 2;
2868f0d39aaaSBarry Smith             tmp0  = x[i1];
2869f0d39aaaSBarry Smith             tmp1  = x[i2];
2870f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2871f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2872f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2873f0d39aaaSBarry Smith           }
2874f0d39aaaSBarry Smith 
2875f0d39aaaSBarry Smith           if (n == sz-1) {
2876f0d39aaaSBarry Smith             tmp0  = x[*idx];
2877f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2878f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2879f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0;
2880f0d39aaaSBarry Smith           }
28815850ef23SBarry Smith           t[row]   = sum1;
28825850ef23SBarry Smith           t[row+1] = sum2;
28835850ef23SBarry Smith           t[row+2] = sum3;
2884f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
2885f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
2886f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
2887f0d39aaaSBarry Smith           ibdiag  += 9;
2888f0d39aaaSBarry Smith           break;
2889f0d39aaaSBarry Smith         case 4:
2890f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2891f0d39aaaSBarry Smith           v3   = a->a + ii[row+2];
2892f0d39aaaSBarry Smith           v4   = a->a + ii[row+3];
2893f0d39aaaSBarry Smith           sum1 = b[row];
2894f0d39aaaSBarry Smith           sum2 = b[row+1];
2895f0d39aaaSBarry Smith           sum3 = b[row+2];
2896f0d39aaaSBarry Smith           sum4 = b[row+3];
2897f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2898f0d39aaaSBarry Smith             i1    = idx[0];
2899f0d39aaaSBarry Smith             i2    = idx[1];
2900f0d39aaaSBarry Smith             idx  += 2;
2901f0d39aaaSBarry Smith             tmp0  = x[i1];
2902f0d39aaaSBarry Smith             tmp1  = x[i2];
2903f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2904f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2905f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2906f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
2907f0d39aaaSBarry Smith           }
2908f0d39aaaSBarry Smith 
2909f0d39aaaSBarry Smith           if (n == sz-1) {
2910f0d39aaaSBarry Smith             tmp0  = x[*idx];
2911f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2912f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2913f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0;
2914f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0;
2915f0d39aaaSBarry Smith           }
29165850ef23SBarry Smith           t[row]   = sum1;
29175850ef23SBarry Smith           t[row+1] = sum2;
29185850ef23SBarry Smith           t[row+2] = sum3;
29195850ef23SBarry Smith           t[row+3] = sum4;
2920f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
2921f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
2922f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
2923f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
2924f0d39aaaSBarry Smith           ibdiag  += 16;
2925f0d39aaaSBarry Smith           break;
2926f0d39aaaSBarry Smith         case 5:
2927f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2928f0d39aaaSBarry Smith           v3   = a->a + ii[row+2];
2929f0d39aaaSBarry Smith           v4   = a->a + ii[row+3];
2930f0d39aaaSBarry Smith           v5   = a->a + ii[row+4];
2931f0d39aaaSBarry Smith           sum1 = b[row];
2932f0d39aaaSBarry Smith           sum2 = b[row+1];
2933f0d39aaaSBarry Smith           sum3 = b[row+2];
2934f0d39aaaSBarry Smith           sum4 = b[row+3];
2935f0d39aaaSBarry Smith           sum5 = b[row+4];
2936f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2937f0d39aaaSBarry Smith             i1    = idx[0];
2938f0d39aaaSBarry Smith             i2    = idx[1];
2939f0d39aaaSBarry Smith             idx  += 2;
2940f0d39aaaSBarry Smith             tmp0  = x[i1];
2941f0d39aaaSBarry Smith             tmp1  = x[i2];
2942f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2943f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2944f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2945f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
2946f0d39aaaSBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
2947f0d39aaaSBarry Smith           }
2948f0d39aaaSBarry Smith 
2949f0d39aaaSBarry Smith           if (n == sz-1) {
2950f0d39aaaSBarry Smith             tmp0  = x[*idx];
2951f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2952f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2953f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0;
2954f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0;
2955f0d39aaaSBarry Smith             sum5 -= v5[0] * tmp0;
2956f0d39aaaSBarry Smith           }
29575850ef23SBarry Smith           t[row]   = sum1;
29585850ef23SBarry Smith           t[row+1] = sum2;
29595850ef23SBarry Smith           t[row+2] = sum3;
29605850ef23SBarry Smith           t[row+3] = sum4;
29615850ef23SBarry Smith           t[row+4] = sum5;
2962f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
2963f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
2964f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
2965f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
2966f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
2967f0d39aaaSBarry Smith           ibdiag  += 25;
2968f0d39aaaSBarry Smith           break;
29698862d2efSBarry Smith         default:
2970e32f2f54SBarry Smith           SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
29718862d2efSBarry Smith         }
29722af78befSBarry Smith       }
29732af78befSBarry Smith 
29745850ef23SBarry Smith       xb   = t;
29752af78befSBarry Smith       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
29762af78befSBarry Smith     } else xb = b;
29772af78befSBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) {
29782af78befSBarry Smith 
2979f0d39aaaSBarry Smith       ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
2980d0f46423SBarry Smith       for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
2981f0d39aaaSBarry Smith         ibdiag -= sizes[i]*sizes[i];
29828862d2efSBarry Smith         sz      = ii[row+1] - diag[row] - 1;
29838862d2efSBarry Smith         v1      = a->a + diag[row] + 1;
29848862d2efSBarry Smith         idx     = a->j + diag[row] + 1;
29852af78befSBarry Smith 
29864108e4d5SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
29878862d2efSBarry Smith         switch (sizes[i]) {
29888862d2efSBarry Smith         case 1:
29898862d2efSBarry Smith 
29908862d2efSBarry Smith           sum1 = xb[row];
29918862d2efSBarry Smith           for (n = 0; n<sz-1; n+=2) {
29928862d2efSBarry Smith             i1    = idx[0];
29938862d2efSBarry Smith             i2    = idx[1];
29948862d2efSBarry Smith             idx  += 2;
29958862d2efSBarry Smith             tmp0  = x[i1];
29968862d2efSBarry Smith             tmp1  = x[i2];
29978862d2efSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
29988862d2efSBarry Smith           }
29998862d2efSBarry Smith 
30008862d2efSBarry Smith           if (n == sz-1) {
3001f0d39aaaSBarry Smith             tmp0  = x[*idx];
3002f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
30038862d2efSBarry Smith           }
3004f0d39aaaSBarry Smith           x[row--] = sum1*(*ibdiag);
3005f0d39aaaSBarry Smith           break;
3006f0d39aaaSBarry Smith 
3007f0d39aaaSBarry Smith         case 2:
3008f0d39aaaSBarry Smith 
3009f0d39aaaSBarry Smith           sum1 = xb[row];
3010f0d39aaaSBarry Smith           sum2 = xb[row-1];
3011f0d39aaaSBarry Smith           /* note that sum1 is associated with the second of the two rows */
3012f0d39aaaSBarry Smith           v2 = a->a + diag[row-1] + 2;
3013f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3014f0d39aaaSBarry Smith             i1    = idx[0];
3015f0d39aaaSBarry Smith             i2    = idx[1];
3016f0d39aaaSBarry Smith             idx  += 2;
3017f0d39aaaSBarry Smith             tmp0  = x[i1];
3018f0d39aaaSBarry Smith             tmp1  = x[i2];
3019f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3020f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3021f0d39aaaSBarry Smith           }
3022f0d39aaaSBarry Smith 
3023f0d39aaaSBarry Smith           if (n == sz-1) {
3024f0d39aaaSBarry Smith             tmp0  = x[*idx];
3025f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3026f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3027f0d39aaaSBarry Smith           }
3028f0d39aaaSBarry Smith           x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3];
3029f0d39aaaSBarry Smith           x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2];
3030f0d39aaaSBarry Smith           break;
3031f0d39aaaSBarry Smith         case 3:
3032f0d39aaaSBarry Smith 
3033f0d39aaaSBarry Smith           sum1 = xb[row];
3034f0d39aaaSBarry Smith           sum2 = xb[row-1];
3035f0d39aaaSBarry Smith           sum3 = xb[row-2];
3036f0d39aaaSBarry Smith           v2   = a->a + diag[row-1] + 2;
3037f0d39aaaSBarry Smith           v3   = a->a + diag[row-2] + 3;
3038f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3039f0d39aaaSBarry Smith             i1    = idx[0];
3040f0d39aaaSBarry Smith             i2    = idx[1];
3041f0d39aaaSBarry Smith             idx  += 2;
3042f0d39aaaSBarry Smith             tmp0  = x[i1];
3043f0d39aaaSBarry Smith             tmp1  = x[i2];
3044f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3045f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3046f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3047f0d39aaaSBarry Smith           }
3048f0d39aaaSBarry Smith 
3049f0d39aaaSBarry Smith           if (n == sz-1) {
3050f0d39aaaSBarry Smith             tmp0  = x[*idx];
3051f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3052f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3053f0d39aaaSBarry Smith             sum3 -= *v3*tmp0;
3054f0d39aaaSBarry Smith           }
3055f0d39aaaSBarry Smith           x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3056f0d39aaaSBarry Smith           x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3057f0d39aaaSBarry Smith           x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3058f0d39aaaSBarry Smith           break;
3059f0d39aaaSBarry Smith         case 4:
3060f0d39aaaSBarry Smith 
3061f0d39aaaSBarry Smith           sum1 = xb[row];
3062f0d39aaaSBarry Smith           sum2 = xb[row-1];
3063f0d39aaaSBarry Smith           sum3 = xb[row-2];
3064f0d39aaaSBarry Smith           sum4 = xb[row-3];
3065f0d39aaaSBarry Smith           v2   = a->a + diag[row-1] + 2;
3066f0d39aaaSBarry Smith           v3   = a->a + diag[row-2] + 3;
3067f0d39aaaSBarry Smith           v4   = a->a + diag[row-3] + 4;
3068f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3069f0d39aaaSBarry Smith             i1    = idx[0];
3070f0d39aaaSBarry Smith             i2    = idx[1];
3071f0d39aaaSBarry Smith             idx  += 2;
3072f0d39aaaSBarry Smith             tmp0  = x[i1];
3073f0d39aaaSBarry Smith             tmp1  = x[i2];
3074f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3075f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3076f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3077f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3078f0d39aaaSBarry Smith           }
3079f0d39aaaSBarry Smith 
3080f0d39aaaSBarry Smith           if (n == sz-1) {
3081f0d39aaaSBarry Smith             tmp0  = x[*idx];
3082f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3083f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3084f0d39aaaSBarry Smith             sum3 -= *v3*tmp0;
3085f0d39aaaSBarry Smith             sum4 -= *v4*tmp0;
3086f0d39aaaSBarry Smith           }
3087f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3088f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3089f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3090f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3091f0d39aaaSBarry Smith           break;
3092f0d39aaaSBarry Smith         case 5:
3093f0d39aaaSBarry Smith 
3094f0d39aaaSBarry Smith           sum1 = xb[row];
3095f0d39aaaSBarry Smith           sum2 = xb[row-1];
3096f0d39aaaSBarry Smith           sum3 = xb[row-2];
3097f0d39aaaSBarry Smith           sum4 = xb[row-3];
3098f0d39aaaSBarry Smith           sum5 = xb[row-4];
3099f0d39aaaSBarry Smith           v2   = a->a + diag[row-1] + 2;
3100f0d39aaaSBarry Smith           v3   = a->a + diag[row-2] + 3;
3101f0d39aaaSBarry Smith           v4   = a->a + diag[row-3] + 4;
3102f0d39aaaSBarry Smith           v5   = a->a + diag[row-4] + 5;
3103f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3104f0d39aaaSBarry Smith             i1    = idx[0];
3105f0d39aaaSBarry Smith             i2    = idx[1];
3106f0d39aaaSBarry Smith             idx  += 2;
3107f0d39aaaSBarry Smith             tmp0  = x[i1];
3108f0d39aaaSBarry Smith             tmp1  = x[i2];
3109f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3110f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3111f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3112f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3113f0d39aaaSBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
3114f0d39aaaSBarry Smith           }
3115f0d39aaaSBarry Smith 
3116f0d39aaaSBarry Smith           if (n == sz-1) {
3117f0d39aaaSBarry Smith             tmp0  = x[*idx];
3118f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3119f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3120f0d39aaaSBarry Smith             sum3 -= *v3*tmp0;
3121f0d39aaaSBarry Smith             sum4 -= *v4*tmp0;
3122f0d39aaaSBarry Smith             sum5 -= *v5*tmp0;
3123f0d39aaaSBarry Smith           }
3124f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
3125f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
3126f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
3127f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
3128f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
31298862d2efSBarry Smith           break;
31308862d2efSBarry Smith         default:
3131e32f2f54SBarry Smith           SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
31328862d2efSBarry Smith         }
31332af78befSBarry Smith       }
31342af78befSBarry Smith 
31352af78befSBarry Smith       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
31362af78befSBarry Smith     }
31372af78befSBarry Smith     its--;
31385850ef23SBarry Smith   }
31395850ef23SBarry Smith   while (its--) {
31405850ef23SBarry Smith 
31415850ef23SBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) {
3142d876e2b0SMark Adams       for (i=0, row=0, ibdiag = a->inode.ibdiag;
3143d876e2b0SMark Adams            i<m;
3144d876e2b0SMark Adams            row += sizes[i], ibdiag += sizes[i]*sizes[i], i++) {
31455850ef23SBarry Smith 
3146d876e2b0SMark Adams         sz  = diag[row] - ii[row];
31475850ef23SBarry Smith         v1  = a->a + ii[row];
31485850ef23SBarry Smith         idx = a->j + ii[row];
31495850ef23SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
31505850ef23SBarry Smith         switch (sizes[i]) {
31515850ef23SBarry Smith         case 1:
31525850ef23SBarry Smith           sum1 = b[row];
31535850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
31545850ef23SBarry Smith             i1    = idx[0];
31555850ef23SBarry Smith             i2    = idx[1];
31565850ef23SBarry Smith             idx  += 2;
31575850ef23SBarry Smith             tmp0  = x[i1];
31585850ef23SBarry Smith             tmp1  = x[i2];
31595850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
31605850ef23SBarry Smith           }
31615850ef23SBarry Smith           if (n == sz-1) {
3162d876e2b0SMark Adams             tmp0  = x[*idx++];
3163d876e2b0SMark Adams             sum1 -= *v1 * tmp0;
3164d876e2b0SMark Adams             v1++;
3165d876e2b0SMark Adams           }
3166d876e2b0SMark Adams           t[row]   = sum1;
3167d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 1;
3168d876e2b0SMark Adams           idx     = a->j + diag[row] + 1;
3169d876e2b0SMark Adams           v1 += 1;
3170d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3171d876e2b0SMark Adams             i1    = idx[0];
3172d876e2b0SMark Adams             i2    = idx[1];
3173d876e2b0SMark Adams             idx  += 2;
3174d876e2b0SMark Adams             tmp0  = x[i1];
3175d876e2b0SMark Adams             tmp1  = x[i2];
3176d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3177d876e2b0SMark Adams           }
3178d876e2b0SMark Adams           if (n == sz-1) {
3179d876e2b0SMark Adams             tmp0  = x[*idx++];
31805850ef23SBarry Smith             sum1 -= *v1 * tmp0;
31815850ef23SBarry Smith           }
31825850ef23SBarry Smith           /* in MatSOR_SeqAIJ this line would be
31835850ef23SBarry Smith            *
31845850ef23SBarry Smith            * x[row] = (1-omega)*x[row]+(sum1+(*bdiag++)*x[row])*(*ibdiag++);
31855850ef23SBarry Smith            *
31865850ef23SBarry Smith            * but omega == 1, so this becomes
31875850ef23SBarry Smith            *
3188d876e2b0SMark Adams            * x[row] = sum1*(*ibdiag++);
31895850ef23SBarry Smith            *
31905850ef23SBarry Smith            */
3191d876e2b0SMark Adams           x[row] = sum1*(*ibdiag);
31925850ef23SBarry Smith           break;
31935850ef23SBarry Smith         case 2:
31945850ef23SBarry Smith           v2   = a->a + ii[row+1];
31955850ef23SBarry Smith           sum1 = b[row];
31965850ef23SBarry Smith           sum2 = b[row+1];
31975850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
31985850ef23SBarry Smith             i1    = idx[0];
31995850ef23SBarry Smith             i2    = idx[1];
32005850ef23SBarry Smith             idx  += 2;
32015850ef23SBarry Smith             tmp0  = x[i1];
32025850ef23SBarry Smith             tmp1  = x[i2];
32035850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
32045850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
32055850ef23SBarry Smith           }
3206d876e2b0SMark Adams           if (n == sz-1) {
3207d876e2b0SMark Adams             tmp0  = x[*idx++];
3208d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3209d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3210d876e2b0SMark Adams             v1++; v2++;
3211d876e2b0SMark Adams           }
3212d876e2b0SMark Adams           t[row]   = sum1;
3213d876e2b0SMark Adams           t[row+1] = sum2;
3214d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 2;
3215d876e2b0SMark Adams           idx     = a->j + diag[row] + 2;
3216d876e2b0SMark Adams           v1 += 2;
3217d876e2b0SMark Adams           v2 += 2;
3218d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3219d876e2b0SMark Adams             i1    = idx[0];
3220d876e2b0SMark Adams             i2    = idx[1];
3221d876e2b0SMark Adams             idx  += 2;
3222d876e2b0SMark Adams             tmp0  = x[i1];
3223d876e2b0SMark Adams             tmp1  = x[i2];
3224d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3225d876e2b0SMark Adams             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3226d876e2b0SMark Adams           }
32275850ef23SBarry Smith           if (n == sz-1) {
32285850ef23SBarry Smith             tmp0  = x[*idx];
32295850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
32305850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
32315850ef23SBarry Smith           }
3232d876e2b0SMark Adams           x[row] = sum1*ibdiag[0] + sum2*ibdiag[2];
3233d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3];
32345850ef23SBarry Smith           break;
32355850ef23SBarry Smith         case 3:
32365850ef23SBarry Smith           v2   = a->a + ii[row+1];
32375850ef23SBarry Smith           v3   = a->a + ii[row+2];
32385850ef23SBarry Smith           sum1 = b[row];
32395850ef23SBarry Smith           sum2 = b[row+1];
32405850ef23SBarry Smith           sum3 = b[row+2];
32415850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
32425850ef23SBarry Smith             i1    = idx[0];
32435850ef23SBarry Smith             i2    = idx[1];
32445850ef23SBarry Smith             idx  += 2;
32455850ef23SBarry Smith             tmp0  = x[i1];
32465850ef23SBarry Smith             tmp1  = x[i2];
32475850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
32485850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
32495850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
32505850ef23SBarry Smith           }
3251d876e2b0SMark Adams           if (n == sz-1) {
3252d876e2b0SMark Adams             tmp0  = x[*idx++];
3253d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3254d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3255d876e2b0SMark Adams             sum3 -= v3[0] * tmp0;
3256d876e2b0SMark Adams             v1++; v2++; v3++;
3257d876e2b0SMark Adams           }
3258d876e2b0SMark Adams           t[row]   = sum1;
3259d876e2b0SMark Adams           t[row+1] = sum2;
3260d876e2b0SMark Adams           t[row+2] = sum3;
3261d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 3;
3262d876e2b0SMark Adams           idx     = a->j + diag[row] + 3;
3263d876e2b0SMark Adams           v1 += 3;
3264d876e2b0SMark Adams           v2 += 3;
3265d876e2b0SMark Adams           v3 += 3;
3266d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3267d876e2b0SMark Adams             i1    = idx[0];
3268d876e2b0SMark Adams             i2    = idx[1];
3269d876e2b0SMark Adams             idx  += 2;
3270d876e2b0SMark Adams             tmp0  = x[i1];
3271d876e2b0SMark Adams             tmp1  = x[i2];
3272d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3273d876e2b0SMark Adams             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3274d876e2b0SMark Adams             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3275d876e2b0SMark Adams           }
32765850ef23SBarry Smith           if (n == sz-1) {
32775850ef23SBarry Smith             tmp0  = x[*idx];
32785850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
32795850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
32805850ef23SBarry Smith             sum3 -= v3[0] * tmp0;
32815850ef23SBarry Smith           }
3282d876e2b0SMark Adams           x[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
3283d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
3284d876e2b0SMark Adams           x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
32855850ef23SBarry Smith           break;
32865850ef23SBarry Smith         case 4:
32875850ef23SBarry Smith           v2   = a->a + ii[row+1];
32885850ef23SBarry Smith           v3   = a->a + ii[row+2];
32895850ef23SBarry Smith           v4   = a->a + ii[row+3];
32905850ef23SBarry Smith           sum1 = b[row];
32915850ef23SBarry Smith           sum2 = b[row+1];
32925850ef23SBarry Smith           sum3 = b[row+2];
32935850ef23SBarry Smith           sum4 = b[row+3];
32945850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
32955850ef23SBarry Smith             i1    = idx[0];
32965850ef23SBarry Smith             i2    = idx[1];
32975850ef23SBarry Smith             idx  += 2;
32985850ef23SBarry Smith             tmp0  = x[i1];
32995850ef23SBarry Smith             tmp1  = x[i2];
33005850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
33015850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
33025850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
33035850ef23SBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
33045850ef23SBarry Smith           }
3305d876e2b0SMark Adams           if (n == sz-1) {
3306d876e2b0SMark Adams             tmp0  = x[*idx++];
3307d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3308d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3309d876e2b0SMark Adams             sum3 -= v3[0] * tmp0;
3310d876e2b0SMark Adams             sum4 -= v4[0] * tmp0;
3311d876e2b0SMark Adams             v1++; v2++; v3++; v4++;
3312d876e2b0SMark Adams           }
3313d876e2b0SMark Adams           t[row]   = sum1;
3314d876e2b0SMark Adams           t[row+1] = sum2;
3315d876e2b0SMark Adams           t[row+2] = sum3;
3316d876e2b0SMark Adams           t[row+3] = sum4;
3317d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 4;
3318d876e2b0SMark Adams           idx     = a->j + diag[row] + 4;
3319d876e2b0SMark Adams           v1 += 4;
3320d876e2b0SMark Adams           v2 += 4;
3321d876e2b0SMark Adams           v3 += 4;
3322d876e2b0SMark Adams           v4 += 4;
3323d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3324d876e2b0SMark Adams             i1    = idx[0];
3325d876e2b0SMark Adams             i2    = idx[1];
3326d876e2b0SMark Adams             idx  += 2;
3327d876e2b0SMark Adams             tmp0  = x[i1];
3328d876e2b0SMark Adams             tmp1  = x[i2];
3329d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3330d876e2b0SMark Adams             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3331d876e2b0SMark Adams             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3332d876e2b0SMark Adams             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3333d876e2b0SMark Adams           }
33345850ef23SBarry Smith           if (n == sz-1) {
33355850ef23SBarry Smith             tmp0  = x[*idx];
33365850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
33375850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
33385850ef23SBarry Smith             sum3 -= v3[0] * tmp0;
33395850ef23SBarry Smith             sum4 -= v4[0] * tmp0;
33405850ef23SBarry Smith           }
3341d876e2b0SMark Adams           x[row] =   sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
3342d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
3343d876e2b0SMark Adams           x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
3344d876e2b0SMark Adams           x[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
33455850ef23SBarry Smith           break;
33465850ef23SBarry Smith         case 5:
33475850ef23SBarry Smith           v2   = a->a + ii[row+1];
33485850ef23SBarry Smith           v3   = a->a + ii[row+2];
33495850ef23SBarry Smith           v4   = a->a + ii[row+3];
33505850ef23SBarry Smith           v5   = a->a + ii[row+4];
33515850ef23SBarry Smith           sum1 = b[row];
33525850ef23SBarry Smith           sum2 = b[row+1];
33535850ef23SBarry Smith           sum3 = b[row+2];
33545850ef23SBarry Smith           sum4 = b[row+3];
33555850ef23SBarry Smith           sum5 = b[row+4];
33565850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
33575850ef23SBarry Smith             i1    = idx[0];
33585850ef23SBarry Smith             i2    = idx[1];
33595850ef23SBarry Smith             idx  += 2;
33605850ef23SBarry Smith             tmp0  = x[i1];
33615850ef23SBarry Smith             tmp1  = x[i2];
33625850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
33635850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
33645850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
33655850ef23SBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
33665850ef23SBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
33675850ef23SBarry Smith           }
33685850ef23SBarry Smith           if (n == sz-1) {
3369d876e2b0SMark Adams             tmp0  = x[*idx++];
33705850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
33715850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
33725850ef23SBarry Smith             sum3 -= v3[0] * tmp0;
33735850ef23SBarry Smith             sum4 -= v4[0] * tmp0;
33745850ef23SBarry Smith             sum5 -= v5[0] * tmp0;
3375d876e2b0SMark Adams             v1++; v2++; v3++; v4++; v5++;
33765850ef23SBarry Smith           }
3377d876e2b0SMark Adams           t[row]   = sum1;
3378d876e2b0SMark Adams           t[row+1] = sum2;
3379d876e2b0SMark Adams           t[row+2] = sum3;
3380d876e2b0SMark Adams           t[row+3] = sum4;
3381d876e2b0SMark Adams           t[row+4] = sum5;
3382d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 5;
3383d876e2b0SMark Adams           idx     = a->j + diag[row] + 5;
3384d876e2b0SMark Adams           v1 += 5;
3385d876e2b0SMark Adams           v2 += 5;
3386d876e2b0SMark Adams           v3 += 5;
3387d876e2b0SMark Adams           v4 += 5;
3388d876e2b0SMark Adams           v5 += 5;
33895850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
33905850ef23SBarry Smith             i1    = idx[0];
33915850ef23SBarry Smith             i2    = idx[1];
33925850ef23SBarry Smith             idx  += 2;
33935850ef23SBarry Smith             tmp0  = x[i1];
33945850ef23SBarry Smith             tmp1  = x[i2];
33955850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
33965850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
33975850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
33985850ef23SBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
33995850ef23SBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
34005850ef23SBarry Smith           }
34015850ef23SBarry Smith           if (n == sz-1) {
34025850ef23SBarry Smith             tmp0  = x[*idx];
3403d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3404d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3405d876e2b0SMark Adams             sum3 -= v3[0] * tmp0;
3406d876e2b0SMark Adams             sum4 -= v4[0] * tmp0;
3407d876e2b0SMark Adams             sum5 -= v5[0] * tmp0;
34085850ef23SBarry Smith           }
3409d876e2b0SMark Adams           x[row]   = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
3410d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
3411d876e2b0SMark Adams           x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
3412d876e2b0SMark Adams           x[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
3413d876e2b0SMark Adams           x[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
3414d876e2b0SMark Adams           break;
3415d876e2b0SMark Adams         default:
3416d876e2b0SMark Adams           SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
3417d876e2b0SMark Adams         }
3418d876e2b0SMark Adams       }
3419d876e2b0SMark Adams       xb   = t;
3420d876e2b0SMark Adams       ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);  /* undercounts diag inverse */
3421d876e2b0SMark Adams     } else xb = b;
3422d876e2b0SMark Adams 
3423d876e2b0SMark Adams     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) {
3424d876e2b0SMark Adams 
3425d876e2b0SMark Adams       ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
3426d876e2b0SMark Adams       for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
3427d876e2b0SMark Adams         ibdiag -= sizes[i]*sizes[i];
3428d876e2b0SMark Adams 
3429d876e2b0SMark Adams         /* set RHS */
3430d876e2b0SMark Adams         if (xb == b) {
3431d876e2b0SMark Adams           /* whole (old way) */
3432d876e2b0SMark Adams           sz      = ii[row+1] - ii[row];
3433d876e2b0SMark Adams           idx     = a->j + ii[row];
3434d876e2b0SMark Adams           switch (sizes[i]) {
3435d876e2b0SMark Adams           case 5:
3436d876e2b0SMark Adams             v5      = a->a + ii[row-4];
3437d876e2b0SMark Adams           case 4: /* fall through */
3438d876e2b0SMark Adams             v4      = a->a + ii[row-3];
3439d876e2b0SMark Adams           case 3:
3440d876e2b0SMark Adams             v3      = a->a + ii[row-2];
3441d876e2b0SMark Adams           case 2:
3442d876e2b0SMark Adams             v2      = a->a + ii[row-1];
3443d876e2b0SMark Adams           case 1:
3444d876e2b0SMark Adams             v1      = a->a + ii[row];
3445d876e2b0SMark Adams             break;
3446d876e2b0SMark Adams           default:
3447d876e2b0SMark Adams             SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
3448d876e2b0SMark Adams           }
3449d876e2b0SMark Adams         } else {
3450d876e2b0SMark Adams           /* upper, no diag */
3451d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 1;
3452d876e2b0SMark Adams           idx     = a->j + diag[row] + 1;
3453d876e2b0SMark Adams           switch (sizes[i]) {
3454d876e2b0SMark Adams           case 5:
3455d876e2b0SMark Adams             v5      = a->a + diag[row-4] + 5;
3456d876e2b0SMark Adams           case 4: /* fall through */
3457d876e2b0SMark Adams             v4      = a->a + diag[row-3] + 4;
3458d876e2b0SMark Adams           case 3:
3459d876e2b0SMark Adams             v3      = a->a + diag[row-2] + 3;
3460d876e2b0SMark Adams           case 2:
3461d876e2b0SMark Adams             v2      = a->a + diag[row-1] + 2;
3462d876e2b0SMark Adams           case 1:
3463d876e2b0SMark Adams             v1      = a->a + diag[row] + 1;
3464d876e2b0SMark Adams           }
3465d876e2b0SMark Adams         }
3466d876e2b0SMark Adams         /* set sum */
3467d876e2b0SMark Adams         switch (sizes[i]) {
3468d876e2b0SMark Adams         case 5:
3469d876e2b0SMark Adams           sum5 = xb[row-4];
3470d876e2b0SMark Adams         case 4: /* fall through */
3471d876e2b0SMark Adams           sum4 = xb[row-3];
3472d876e2b0SMark Adams         case 3:
3473d876e2b0SMark Adams           sum3 = xb[row-2];
3474d876e2b0SMark Adams         case 2:
3475d876e2b0SMark Adams           sum2 = xb[row-1];
3476d876e2b0SMark Adams         case 1:
3477d876e2b0SMark Adams           /* note that sum1 is associated with the last row */
3478d876e2b0SMark Adams           sum1 = xb[row];
3479d876e2b0SMark Adams         }
3480d876e2b0SMark Adams         /* do sums */
3481d876e2b0SMark Adams         for (n = 0; n<sz-1; n+=2) {
3482d876e2b0SMark Adams             i1    = idx[0];
3483d876e2b0SMark Adams             i2    = idx[1];
3484d876e2b0SMark Adams             idx  += 2;
3485d876e2b0SMark Adams             tmp0  = x[i1];
3486d876e2b0SMark Adams             tmp1  = x[i2];
3487d876e2b0SMark Adams             switch (sizes[i]) {
3488d876e2b0SMark Adams             case 5:
3489d876e2b0SMark Adams               sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
3490d876e2b0SMark Adams             case 4: /* fall through */
3491d876e2b0SMark Adams               sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3492d876e2b0SMark Adams             case 3:
3493d876e2b0SMark Adams               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3494d876e2b0SMark Adams             case 2:
3495d876e2b0SMark Adams               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3496d876e2b0SMark Adams             case 1:
3497d876e2b0SMark Adams               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3498d876e2b0SMark Adams             }
3499d876e2b0SMark Adams         }
3500d876e2b0SMark Adams         /* ragged edge */
3501d876e2b0SMark Adams         if (n == sz-1) {
3502d876e2b0SMark Adams           tmp0  = x[*idx];
3503d876e2b0SMark Adams           switch (sizes[i]) {
3504d876e2b0SMark Adams           case 5:
3505d876e2b0SMark Adams             sum5 -= *v5*tmp0;
3506d876e2b0SMark Adams           case 4: /* fall through */
3507d876e2b0SMark Adams             sum4 -= *v4*tmp0;
3508d876e2b0SMark Adams           case 3:
3509d876e2b0SMark Adams             sum3 -= *v3*tmp0;
3510d876e2b0SMark Adams           case 2:
3511d876e2b0SMark Adams             sum2 -= *v2*tmp0;
3512d876e2b0SMark Adams           case 1:
3513d876e2b0SMark Adams             sum1 -= *v1*tmp0;
3514d876e2b0SMark Adams           }
3515d876e2b0SMark Adams         }
3516d876e2b0SMark Adams         /* update */
3517d876e2b0SMark Adams         if (xb == b) {
3518d876e2b0SMark Adams           /* whole (old way) w/ diag */
3519d876e2b0SMark Adams           switch (sizes[i]) {
3520d876e2b0SMark Adams           case 5:
35215850ef23SBarry Smith             x[row--] += sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
35225850ef23SBarry Smith             x[row--] += sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
35235850ef23SBarry Smith             x[row--] += sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
35245850ef23SBarry Smith             x[row--] += sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
35255850ef23SBarry Smith             x[row--] += sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
35265850ef23SBarry Smith             break;
3527d876e2b0SMark Adams           case 4:
3528d876e2b0SMark Adams             x[row--] += sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3529d876e2b0SMark Adams             x[row--] += sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3530d876e2b0SMark Adams             x[row--] += sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3531d876e2b0SMark Adams             x[row--] += sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3532d876e2b0SMark Adams             break;
3533d876e2b0SMark Adams           case 3:
3534d876e2b0SMark Adams             x[row--] += sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3535d876e2b0SMark Adams             x[row--] += sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3536d876e2b0SMark Adams             x[row--] += sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3537d876e2b0SMark Adams             break;
3538d876e2b0SMark Adams           case 2:
3539d876e2b0SMark Adams             x[row--] += sum2*ibdiag[1] + sum1*ibdiag[3];
3540d876e2b0SMark Adams             x[row--] += sum2*ibdiag[0] + sum1*ibdiag[2];
3541d876e2b0SMark Adams             break;
3542d876e2b0SMark Adams           case 1:
3543d876e2b0SMark Adams             x[row--] += sum1*(*ibdiag);
3544d876e2b0SMark Adams             break;
3545d876e2b0SMark Adams           }
3546d876e2b0SMark Adams         } else {
3547d876e2b0SMark Adams           /* no diag so set =  */
3548d876e2b0SMark Adams           switch (sizes[i]) {
3549d876e2b0SMark Adams           case 5:
3550d876e2b0SMark Adams             x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
3551d876e2b0SMark Adams             x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
3552d876e2b0SMark Adams             x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
3553d876e2b0SMark Adams             x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
3554d876e2b0SMark Adams             x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
3555d876e2b0SMark Adams             break;
3556d876e2b0SMark Adams           case 4:
3557d876e2b0SMark Adams             x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3558d876e2b0SMark Adams             x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3559d876e2b0SMark Adams             x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3560d876e2b0SMark Adams             x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3561d876e2b0SMark Adams             break;
3562d876e2b0SMark Adams           case 3:
3563d876e2b0SMark Adams             x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3564d876e2b0SMark Adams             x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3565d876e2b0SMark Adams             x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3566d876e2b0SMark Adams             break;
3567d876e2b0SMark Adams           case 2:
3568d876e2b0SMark Adams             x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3];
3569d876e2b0SMark Adams             x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2];
3570d876e2b0SMark Adams             break;
3571d876e2b0SMark Adams           case 1:
3572d876e2b0SMark Adams             x[row--] = sum1*(*ibdiag);
3573d876e2b0SMark Adams             break;
35745850ef23SBarry Smith           }
35755850ef23SBarry Smith         }
3576d876e2b0SMark Adams       }
3577d876e2b0SMark Adams       if (xb == b) {
35785850ef23SBarry Smith         ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
3579d876e2b0SMark Adams       } else {
3580d876e2b0SMark Adams         ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper, undercounts diag inverse */
3581d876e2b0SMark Adams       }
35825850ef23SBarry Smith     }
35832af78befSBarry Smith   }
358489c6957cSBarry Smith   if (flag & SOR_EISENSTAT) {
358589c6957cSBarry Smith     /*
358689c6957cSBarry Smith           Apply  (U + D)^-1  where D is now the block diagonal
358789c6957cSBarry Smith     */
358889c6957cSBarry Smith     ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
358989c6957cSBarry Smith     for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
359089c6957cSBarry Smith       ibdiag -= sizes[i]*sizes[i];
359189c6957cSBarry Smith       sz      = ii[row+1] - diag[row] - 1;
359289c6957cSBarry Smith       v1      = a->a + diag[row] + 1;
359389c6957cSBarry Smith       idx     = a->j + diag[row] + 1;
35944108e4d5SBarry Smith       /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
359589c6957cSBarry Smith       switch (sizes[i]) {
359689c6957cSBarry Smith       case 1:
359789c6957cSBarry Smith 
359889c6957cSBarry Smith         sum1 = b[row];
359989c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
360089c6957cSBarry Smith           i1    = idx[0];
360189c6957cSBarry Smith           i2    = idx[1];
360289c6957cSBarry Smith           idx  += 2;
360389c6957cSBarry Smith           tmp0  = x[i1];
360489c6957cSBarry Smith           tmp1  = x[i2];
360589c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
360689c6957cSBarry Smith         }
360789c6957cSBarry Smith 
360889c6957cSBarry Smith         if (n == sz-1) {
360989c6957cSBarry Smith           tmp0  = x[*idx];
361089c6957cSBarry Smith           sum1 -= *v1*tmp0;
361189c6957cSBarry Smith         }
3612938d4eb3SBarry Smith         x[row] = sum1*(*ibdiag);row--;
361389c6957cSBarry Smith         break;
361489c6957cSBarry Smith 
361589c6957cSBarry Smith       case 2:
361689c6957cSBarry Smith 
361789c6957cSBarry Smith         sum1 = b[row];
361889c6957cSBarry Smith         sum2 = b[row-1];
361989c6957cSBarry Smith         /* note that sum1 is associated with the second of the two rows */
362089c6957cSBarry Smith         v2 = a->a + diag[row-1] + 2;
362189c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
362289c6957cSBarry Smith           i1    = idx[0];
362389c6957cSBarry Smith           i2    = idx[1];
362489c6957cSBarry Smith           idx  += 2;
362589c6957cSBarry Smith           tmp0  = x[i1];
362689c6957cSBarry Smith           tmp1  = x[i2];
362789c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
362889c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
362989c6957cSBarry Smith         }
363089c6957cSBarry Smith 
363189c6957cSBarry Smith         if (n == sz-1) {
363289c6957cSBarry Smith           tmp0  = x[*idx];
363389c6957cSBarry Smith           sum1 -= *v1*tmp0;
363489c6957cSBarry Smith           sum2 -= *v2*tmp0;
363589c6957cSBarry Smith         }
3636938d4eb3SBarry Smith         x[row]   = sum2*ibdiag[1] + sum1*ibdiag[3];
3637938d4eb3SBarry Smith         x[row-1] = sum2*ibdiag[0] + sum1*ibdiag[2];
3638938d4eb3SBarry Smith         row     -= 2;
363989c6957cSBarry Smith         break;
364089c6957cSBarry Smith       case 3:
364189c6957cSBarry Smith 
364289c6957cSBarry Smith         sum1 = b[row];
364389c6957cSBarry Smith         sum2 = b[row-1];
364489c6957cSBarry Smith         sum3 = b[row-2];
364589c6957cSBarry Smith         v2   = a->a + diag[row-1] + 2;
364689c6957cSBarry Smith         v3   = a->a + diag[row-2] + 3;
364789c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
364889c6957cSBarry Smith           i1    = idx[0];
364989c6957cSBarry Smith           i2    = idx[1];
365089c6957cSBarry Smith           idx  += 2;
365189c6957cSBarry Smith           tmp0  = x[i1];
365289c6957cSBarry Smith           tmp1  = x[i2];
365389c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
365489c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
365589c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
365689c6957cSBarry Smith         }
365789c6957cSBarry Smith 
365889c6957cSBarry Smith         if (n == sz-1) {
365989c6957cSBarry Smith           tmp0  = x[*idx];
366089c6957cSBarry Smith           sum1 -= *v1*tmp0;
366189c6957cSBarry Smith           sum2 -= *v2*tmp0;
366289c6957cSBarry Smith           sum3 -= *v3*tmp0;
366389c6957cSBarry Smith         }
3664938d4eb3SBarry Smith         x[row]   = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3665938d4eb3SBarry Smith         x[row-1] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3666938d4eb3SBarry Smith         x[row-2] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3667938d4eb3SBarry Smith         row     -= 3;
366889c6957cSBarry Smith         break;
366989c6957cSBarry Smith       case 4:
367089c6957cSBarry Smith 
367189c6957cSBarry Smith         sum1 = b[row];
367289c6957cSBarry Smith         sum2 = b[row-1];
367389c6957cSBarry Smith         sum3 = b[row-2];
367489c6957cSBarry Smith         sum4 = b[row-3];
367589c6957cSBarry Smith         v2   = a->a + diag[row-1] + 2;
367689c6957cSBarry Smith         v3   = a->a + diag[row-2] + 3;
367789c6957cSBarry Smith         v4   = a->a + diag[row-3] + 4;
367889c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
367989c6957cSBarry Smith           i1    = idx[0];
368089c6957cSBarry Smith           i2    = idx[1];
368189c6957cSBarry Smith           idx  += 2;
368289c6957cSBarry Smith           tmp0  = x[i1];
368389c6957cSBarry Smith           tmp1  = x[i2];
368489c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
368589c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
368689c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
368789c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
368889c6957cSBarry Smith         }
368989c6957cSBarry Smith 
369089c6957cSBarry Smith         if (n == sz-1) {
369189c6957cSBarry Smith           tmp0  = x[*idx];
369289c6957cSBarry Smith           sum1 -= *v1*tmp0;
369389c6957cSBarry Smith           sum2 -= *v2*tmp0;
369489c6957cSBarry Smith           sum3 -= *v3*tmp0;
369589c6957cSBarry Smith           sum4 -= *v4*tmp0;
369689c6957cSBarry Smith         }
3697938d4eb3SBarry Smith         x[row]   = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3698938d4eb3SBarry Smith         x[row-1] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3699938d4eb3SBarry Smith         x[row-2] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3700938d4eb3SBarry Smith         x[row-3] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3701938d4eb3SBarry Smith         row     -= 4;
370289c6957cSBarry Smith         break;
370389c6957cSBarry Smith       case 5:
370489c6957cSBarry Smith 
370589c6957cSBarry Smith         sum1 = b[row];
370689c6957cSBarry Smith         sum2 = b[row-1];
370789c6957cSBarry Smith         sum3 = b[row-2];
370889c6957cSBarry Smith         sum4 = b[row-3];
370989c6957cSBarry Smith         sum5 = b[row-4];
371089c6957cSBarry Smith         v2   = a->a + diag[row-1] + 2;
371189c6957cSBarry Smith         v3   = a->a + diag[row-2] + 3;
371289c6957cSBarry Smith         v4   = a->a + diag[row-3] + 4;
371389c6957cSBarry Smith         v5   = a->a + diag[row-4] + 5;
371489c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
371589c6957cSBarry Smith           i1    = idx[0];
371689c6957cSBarry Smith           i2    = idx[1];
371789c6957cSBarry Smith           idx  += 2;
371889c6957cSBarry Smith           tmp0  = x[i1];
371989c6957cSBarry Smith           tmp1  = x[i2];
372089c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
372189c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
372289c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
372389c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
372489c6957cSBarry Smith           sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
372589c6957cSBarry Smith         }
372689c6957cSBarry Smith 
372789c6957cSBarry Smith         if (n == sz-1) {
372889c6957cSBarry Smith           tmp0  = x[*idx];
372989c6957cSBarry Smith           sum1 -= *v1*tmp0;
373089c6957cSBarry Smith           sum2 -= *v2*tmp0;
373189c6957cSBarry Smith           sum3 -= *v3*tmp0;
373289c6957cSBarry Smith           sum4 -= *v4*tmp0;
373389c6957cSBarry Smith           sum5 -= *v5*tmp0;
373489c6957cSBarry Smith         }
3735938d4eb3SBarry Smith         x[row]   = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
3736938d4eb3SBarry Smith         x[row-1] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
3737938d4eb3SBarry Smith         x[row-2] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
3738938d4eb3SBarry Smith         x[row-3] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
3739938d4eb3SBarry Smith         x[row-4] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
3740938d4eb3SBarry Smith         row     -= 5;
374189c6957cSBarry Smith         break;
374289c6957cSBarry Smith       default:
3743e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
374489c6957cSBarry Smith       }
374589c6957cSBarry Smith     }
374689c6957cSBarry Smith     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
374789c6957cSBarry Smith 
374889c6957cSBarry Smith     /*
374989c6957cSBarry Smith            t = b - D x    where D is the block diagonal
375089c6957cSBarry Smith     */
375189c6957cSBarry Smith     cnt = 0;
375289c6957cSBarry Smith     for (i=0, row=0; i<m; i++) {
375389c6957cSBarry Smith       switch (sizes[i]) {
375489c6957cSBarry Smith       case 1:
375589c6957cSBarry Smith         t[row] = b[row] - bdiag[cnt++]*x[row]; row++;
375689c6957cSBarry Smith         break;
375789c6957cSBarry Smith       case 2:
375889c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1];
375989c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+2];
376089c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+3];
376189c6957cSBarry Smith         t[row]   = b[row] - tmp1;
376289c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2; row += 2;
376389c6957cSBarry Smith         cnt     += 4;
376489c6957cSBarry Smith         break;
376589c6957cSBarry Smith       case 3:
376689c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1]; x3 = x[row+2];
376789c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6];
376889c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7];
376989c6957cSBarry Smith         tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8];
377089c6957cSBarry Smith         t[row]   = b[row] - tmp1;
377189c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2;
377289c6957cSBarry Smith         t[row+2] = b[row+2] - tmp3; row += 3;
377389c6957cSBarry Smith         cnt     += 9;
377489c6957cSBarry Smith         break;
377589c6957cSBarry Smith       case 4:
377689c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3];
377789c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12];
377889c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13];
377989c6957cSBarry Smith         tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14];
378089c6957cSBarry Smith         tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15];
378189c6957cSBarry Smith         t[row]   = b[row] - tmp1;
378289c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2;
378389c6957cSBarry Smith         t[row+2] = b[row+2] - tmp3;
378489c6957cSBarry Smith         t[row+3] = b[row+3] - tmp4; row += 4;
378589c6957cSBarry Smith         cnt     += 16;
378689c6957cSBarry Smith         break;
378789c6957cSBarry Smith       case 5:
378889c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; x5 = x[row+4];
378989c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20];
379089c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21];
379189c6957cSBarry Smith         tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22];
379289c6957cSBarry Smith         tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23];
379389c6957cSBarry Smith         tmp5     = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24];
379489c6957cSBarry Smith         t[row]   = b[row] - tmp1;
379589c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2;
379689c6957cSBarry Smith         t[row+2] = b[row+2] - tmp3;
379789c6957cSBarry Smith         t[row+3] = b[row+3] - tmp4;
379889c6957cSBarry Smith         t[row+4] = b[row+4] - tmp5;row += 5;
379989c6957cSBarry Smith         cnt     += 25;
380089c6957cSBarry Smith         break;
380189c6957cSBarry Smith       default:
3802e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
380389c6957cSBarry Smith       }
380489c6957cSBarry Smith     }
380589c6957cSBarry Smith     ierr = PetscLogFlops(m);CHKERRQ(ierr);
380689c6957cSBarry Smith 
380789c6957cSBarry Smith 
380889c6957cSBarry Smith 
380989c6957cSBarry Smith     /*
381089c6957cSBarry Smith           Apply (L + D)^-1 where D is the block diagonal
381189c6957cSBarry Smith     */
381289c6957cSBarry Smith     for (i=0, row=0; i<m; i++) {
381389c6957cSBarry Smith       sz  = diag[row] - ii[row];
381489c6957cSBarry Smith       v1  = a->a + ii[row];
381589c6957cSBarry Smith       idx = a->j + ii[row];
38164108e4d5SBarry Smith       /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
381789c6957cSBarry Smith       switch (sizes[i]) {
381889c6957cSBarry Smith       case 1:
381989c6957cSBarry Smith 
382089c6957cSBarry Smith         sum1 = t[row];
382189c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
382289c6957cSBarry Smith           i1    = idx[0];
382389c6957cSBarry Smith           i2    = idx[1];
382489c6957cSBarry Smith           idx  += 2;
382589c6957cSBarry Smith           tmp0  = t[i1];
382689c6957cSBarry Smith           tmp1  = t[i2];
382789c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
382889c6957cSBarry Smith         }
382989c6957cSBarry Smith 
383089c6957cSBarry Smith         if (n == sz-1) {
383189c6957cSBarry Smith           tmp0  = t[*idx];
383289c6957cSBarry Smith           sum1 -= *v1 * tmp0;
383389c6957cSBarry Smith         }
383489c6957cSBarry Smith         x[row] += t[row] = sum1*(*ibdiag++); row++;
383589c6957cSBarry Smith         break;
383689c6957cSBarry Smith       case 2:
383789c6957cSBarry Smith         v2   = a->a + ii[row+1];
383889c6957cSBarry Smith         sum1 = t[row];
383989c6957cSBarry Smith         sum2 = t[row+1];
384089c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
384189c6957cSBarry Smith           i1    = idx[0];
384289c6957cSBarry Smith           i2    = idx[1];
384389c6957cSBarry Smith           idx  += 2;
384489c6957cSBarry Smith           tmp0  = t[i1];
384589c6957cSBarry Smith           tmp1  = t[i2];
384689c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
384789c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
384889c6957cSBarry Smith         }
384989c6957cSBarry Smith 
385089c6957cSBarry Smith         if (n == sz-1) {
385189c6957cSBarry Smith           tmp0  = t[*idx];
385289c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
385389c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
385489c6957cSBarry Smith         }
385589c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[2];
385689c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3];
385789c6957cSBarry Smith         ibdiag   += 4; row += 2;
385889c6957cSBarry Smith         break;
385989c6957cSBarry Smith       case 3:
386089c6957cSBarry Smith         v2   = a->a + ii[row+1];
386189c6957cSBarry Smith         v3   = a->a + ii[row+2];
386289c6957cSBarry Smith         sum1 = t[row];
386389c6957cSBarry Smith         sum2 = t[row+1];
386489c6957cSBarry Smith         sum3 = t[row+2];
386589c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
386689c6957cSBarry Smith           i1    = idx[0];
386789c6957cSBarry Smith           i2    = idx[1];
386889c6957cSBarry Smith           idx  += 2;
386989c6957cSBarry Smith           tmp0  = t[i1];
387089c6957cSBarry Smith           tmp1  = t[i2];
387189c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
387289c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
387389c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
387489c6957cSBarry Smith         }
387589c6957cSBarry Smith 
387689c6957cSBarry Smith         if (n == sz-1) {
387789c6957cSBarry Smith           tmp0  = t[*idx];
387889c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
387989c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
388089c6957cSBarry Smith           sum3 -= v3[0] * tmp0;
388189c6957cSBarry Smith         }
388289c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
388389c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
388489c6957cSBarry Smith         x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
388589c6957cSBarry Smith         ibdiag   += 9; row += 3;
388689c6957cSBarry Smith         break;
388789c6957cSBarry Smith       case 4:
388889c6957cSBarry Smith         v2   = a->a + ii[row+1];
388989c6957cSBarry Smith         v3   = a->a + ii[row+2];
389089c6957cSBarry Smith         v4   = a->a + ii[row+3];
389189c6957cSBarry Smith         sum1 = t[row];
389289c6957cSBarry Smith         sum2 = t[row+1];
389389c6957cSBarry Smith         sum3 = t[row+2];
389489c6957cSBarry Smith         sum4 = t[row+3];
389589c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
389689c6957cSBarry Smith           i1    = idx[0];
389789c6957cSBarry Smith           i2    = idx[1];
389889c6957cSBarry Smith           idx  += 2;
389989c6957cSBarry Smith           tmp0  = t[i1];
390089c6957cSBarry Smith           tmp1  = t[i2];
390189c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
390289c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
390389c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
390489c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
390589c6957cSBarry Smith         }
390689c6957cSBarry Smith 
390789c6957cSBarry Smith         if (n == sz-1) {
390889c6957cSBarry Smith           tmp0  = t[*idx];
390989c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
391089c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
391189c6957cSBarry Smith           sum3 -= v3[0] * tmp0;
391289c6957cSBarry Smith           sum4 -= v4[0] * tmp0;
391389c6957cSBarry Smith         }
391489c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
391589c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
391689c6957cSBarry Smith         x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
391789c6957cSBarry Smith         x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
391889c6957cSBarry Smith         ibdiag   += 16; row += 4;
391989c6957cSBarry Smith         break;
392089c6957cSBarry Smith       case 5:
392189c6957cSBarry Smith         v2   = a->a + ii[row+1];
392289c6957cSBarry Smith         v3   = a->a + ii[row+2];
392389c6957cSBarry Smith         v4   = a->a + ii[row+3];
392489c6957cSBarry Smith         v5   = a->a + ii[row+4];
392589c6957cSBarry Smith         sum1 = t[row];
392689c6957cSBarry Smith         sum2 = t[row+1];
392789c6957cSBarry Smith         sum3 = t[row+2];
392889c6957cSBarry Smith         sum4 = t[row+3];
392989c6957cSBarry Smith         sum5 = t[row+4];
393089c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
393189c6957cSBarry Smith           i1    = idx[0];
393289c6957cSBarry Smith           i2    = idx[1];
393389c6957cSBarry Smith           idx  += 2;
393489c6957cSBarry Smith           tmp0  = t[i1];
393589c6957cSBarry Smith           tmp1  = t[i2];
393689c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
393789c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
393889c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
393989c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
394089c6957cSBarry Smith           sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
394189c6957cSBarry Smith         }
394289c6957cSBarry Smith 
394389c6957cSBarry Smith         if (n == sz-1) {
394489c6957cSBarry Smith           tmp0  = t[*idx];
394589c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
394689c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
394789c6957cSBarry Smith           sum3 -= v3[0] * tmp0;
394889c6957cSBarry Smith           sum4 -= v4[0] * tmp0;
394989c6957cSBarry Smith           sum5 -= v5[0] * tmp0;
395089c6957cSBarry Smith         }
395189c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
395289c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
395389c6957cSBarry Smith         x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
395489c6957cSBarry Smith         x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
395589c6957cSBarry Smith         x[row+4] += t[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
395689c6957cSBarry Smith         ibdiag   += 25; row += 5;
395789c6957cSBarry Smith         break;
395889c6957cSBarry Smith       default:
3959e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
396089c6957cSBarry Smith       }
396189c6957cSBarry Smith     }
396289c6957cSBarry Smith     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
39635850ef23SBarry Smith   }
39642af78befSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
39653649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
3966365a8a9eSBarry Smith   PetscFunctionReturn(0);
39672af78befSBarry Smith }
39682af78befSBarry Smith 
39694108e4d5SBarry Smith PetscErrorCode MatMultDiagonalBlock_SeqAIJ_Inode(Mat A,Vec bb,Vec xx)
397089c6957cSBarry Smith {
397189c6957cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
397289c6957cSBarry Smith   PetscScalar       *x,tmp1,tmp2,tmp3,tmp4,tmp5,x1,x2,x3,x4,x5;
397389c6957cSBarry Smith   const MatScalar   *bdiag = a->inode.bdiag;
397489c6957cSBarry Smith   const PetscScalar *b;
397589c6957cSBarry Smith   PetscErrorCode    ierr;
397689c6957cSBarry Smith   PetscInt          m      = a->inode.node_count,cnt = 0,i,row;
397789c6957cSBarry Smith   const PetscInt    *sizes = a->inode.size;
39782af78befSBarry Smith 
397989c6957cSBarry Smith   PetscFunctionBegin;
3980b215bc84SStefano Zampini   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
398189c6957cSBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
39823649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
398389c6957cSBarry Smith   cnt  = 0;
398489c6957cSBarry Smith   for (i=0, row=0; i<m; i++) {
398589c6957cSBarry Smith     switch (sizes[i]) {
398689c6957cSBarry Smith     case 1:
398789c6957cSBarry Smith       x[row] = b[row]*bdiag[cnt++];row++;
398889c6957cSBarry Smith       break;
398989c6957cSBarry Smith     case 2:
399089c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1];
399189c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+2];
399289c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+3];
399389c6957cSBarry Smith       x[row++] = tmp1;
399489c6957cSBarry Smith       x[row++] = tmp2;
399589c6957cSBarry Smith       cnt     += 4;
399689c6957cSBarry Smith       break;
399789c6957cSBarry Smith     case 3:
399889c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1]; x3 = b[row+2];
399989c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6];
400089c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7];
400189c6957cSBarry Smith       tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8];
400289c6957cSBarry Smith       x[row++] = tmp1;
400389c6957cSBarry Smith       x[row++] = tmp2;
400489c6957cSBarry Smith       x[row++] = tmp3;
400589c6957cSBarry Smith       cnt     += 9;
400689c6957cSBarry Smith       break;
400789c6957cSBarry Smith     case 4:
400889c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3];
400989c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12];
401089c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13];
401189c6957cSBarry Smith       tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14];
401289c6957cSBarry Smith       tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15];
401389c6957cSBarry Smith       x[row++] = tmp1;
401489c6957cSBarry Smith       x[row++] = tmp2;
401589c6957cSBarry Smith       x[row++] = tmp3;
401689c6957cSBarry Smith       x[row++] = tmp4;
401789c6957cSBarry Smith       cnt     += 16;
401889c6957cSBarry Smith       break;
401989c6957cSBarry Smith     case 5:
402089c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3]; x5 = b[row+4];
402189c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20];
402289c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21];
402389c6957cSBarry Smith       tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22];
402489c6957cSBarry Smith       tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23];
402589c6957cSBarry Smith       tmp5     = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24];
402689c6957cSBarry Smith       x[row++] = tmp1;
402789c6957cSBarry Smith       x[row++] = tmp2;
402889c6957cSBarry Smith       x[row++] = tmp3;
402989c6957cSBarry Smith       x[row++] = tmp4;
403089c6957cSBarry Smith       x[row++] = tmp5;
403189c6957cSBarry Smith       cnt     += 25;
403289c6957cSBarry Smith       break;
403389c6957cSBarry Smith     default:
4034e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
403589c6957cSBarry Smith     }
403689c6957cSBarry Smith   }
4037ca0c957dSBarry Smith   ierr = PetscLogFlops(2.0*cnt);CHKERRQ(ierr);
403889c6957cSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
40393649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
404089c6957cSBarry Smith   PetscFunctionReturn(0);
404189c6957cSBarry Smith }
404289c6957cSBarry Smith 
4043b215bc84SStefano Zampini static PetscErrorCode MatSeqAIJ_Inode_ResetOps(Mat A)
4044b215bc84SStefano Zampini {
4045b215bc84SStefano Zampini   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
4046b215bc84SStefano Zampini 
4047b215bc84SStefano Zampini   PetscFunctionBegin;
4048b215bc84SStefano Zampini   a->inode.node_count       = 0;
4049b215bc84SStefano Zampini   a->inode.use              = PETSC_FALSE;
4050b215bc84SStefano Zampini   a->inode.checked          = PETSC_FALSE;
4051b215bc84SStefano Zampini   a->inode.mat_nonzerostate = -1;
4052b215bc84SStefano Zampini   A->ops->getrowij          = MatGetRowIJ_SeqAIJ;
4053b215bc84SStefano Zampini   A->ops->restorerowij      = MatRestoreRowIJ_SeqAIJ;
4054b215bc84SStefano Zampini   A->ops->getcolumnij       = MatGetColumnIJ_SeqAIJ;
4055b215bc84SStefano Zampini   A->ops->restorecolumnij   = MatRestoreColumnIJ_SeqAIJ;
4056b215bc84SStefano Zampini   A->ops->coloringpatch     = NULL;
4057b215bc84SStefano Zampini   A->ops->multdiagonalblock = NULL;
4058b215bc84SStefano Zampini   if (A->factortype) {
4059b215bc84SStefano Zampini     A->ops->solve = MatSolve_SeqAIJ_inplace;
4060b215bc84SStefano Zampini   }
4061b215bc84SStefano Zampini   PetscFunctionReturn(0);
4062b215bc84SStefano Zampini }
4063b215bc84SStefano Zampini 
40644c1414c8SBarry Smith /*
40654c1414c8SBarry Smith     samestructure indicates that the matrix has not changed its nonzero structure so we
40664c1414c8SBarry Smith     do not need to recompute the inodes
40674c1414c8SBarry Smith */
4068a02bda8eSBarry Smith PetscErrorCode MatSeqAIJCheckInode(Mat A)
40694c1414c8SBarry Smith {
40704c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
40714c1414c8SBarry Smith   PetscErrorCode ierr;
40728758e1faSBarry Smith   PetscInt       i,j,m,nzx,nzy,*ns,node_count,blk_size;
4073ace3abfcSBarry Smith   PetscBool      flag;
40748758e1faSBarry Smith   const PetscInt *idx,*idy,*ii;
40754c1414c8SBarry Smith 
40764c1414c8SBarry Smith   PetscFunctionBegin;
4077b215bc84SStefano Zampini   if (!a->inode.use) {
4078b215bc84SStefano Zampini     ierr = MatSeqAIJ_Inode_ResetOps(A);CHKERRQ(ierr);
4079b215bc84SStefano Zampini     ierr = PetscFree(a->inode.size);CHKERRQ(ierr);
4080b215bc84SStefano Zampini     PetscFunctionReturn(0);
4081b215bc84SStefano Zampini   }
4082a02bda8eSBarry Smith   if (a->inode.checked && A->nonzerostate == a->inode.mat_nonzerostate) PetscFunctionReturn(0);
40834c1414c8SBarry Smith 
4084d0f46423SBarry Smith   m = A->rmap->n;
4085b215bc84SStefano Zampini   if (!a->inode.size) { ierr = PetscMalloc1(m+1,&a->inode.size);CHKERRQ(ierr); }
4086b215bc84SStefano Zampini   ns = a->inode.size;
40874c1414c8SBarry Smith 
40884c1414c8SBarry Smith   i          = 0;
40894c1414c8SBarry Smith   node_count = 0;
40904c1414c8SBarry Smith   idx        = a->j;
40914c1414c8SBarry Smith   ii         = a->i;
40924c1414c8SBarry Smith   while (i < m) {                /* For each row */
40934c1414c8SBarry Smith     nzx = ii[i+1] - ii[i];       /* Number of nonzeros */
40944c1414c8SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
40954c1414c8SBarry Smith     for (j=i+1,idy=idx,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
40964c1414c8SBarry Smith       nzy = ii[j+1] - ii[j];     /* Same number of nonzeros */
40974c1414c8SBarry Smith       if (nzy != nzx) break;
40984c1414c8SBarry Smith       idy += nzx;              /* Same nonzero pattern */
4099580bdb30SBarry Smith       ierr = PetscArraycmp(idx,idy,nzx,&flag);CHKERRQ(ierr);
41004c1414c8SBarry Smith       if (!flag) break;
41014c1414c8SBarry Smith     }
41024c1414c8SBarry Smith     ns[node_count++] = blk_size;
41034c1414c8SBarry Smith     idx             += blk_size*nzx;
41044c1414c8SBarry Smith     i                = j;
41054c1414c8SBarry Smith   }
41062cb58ee3SKarl Rupp 
41074c1414c8SBarry Smith   /* If not enough inodes found,, do not use inode version of the routines */
4108be6adb11SBarry Smith   if (!m || node_count > .8*m) {
4109b215bc84SStefano Zampini     ierr = MatSeqAIJ_Inode_ResetOps(A);CHKERRQ(ierr);
4110b215bc84SStefano Zampini     ierr = PetscFree(a->inode.size);CHKERRQ(ierr);
41114c1414c8SBarry Smith     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
41124c1414c8SBarry Smith   } else {
4113d5f3da31SBarry Smith     if (!A->factortype) {
4114375a6242SBarry Smith       A->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode;
4115375a6242SBarry Smith       if (A->rmap->n == A->cmap->n) {
41164108e4d5SBarry Smith         A->ops->getrowij        = MatGetRowIJ_SeqAIJ_Inode;
41174108e4d5SBarry Smith         A->ops->restorerowij    = MatRestoreRowIJ_SeqAIJ_Inode;
41184108e4d5SBarry Smith         A->ops->getcolumnij     = MatGetColumnIJ_SeqAIJ_Inode;
41194108e4d5SBarry Smith         A->ops->restorecolumnij = MatRestoreColumnIJ_SeqAIJ_Inode;
41204108e4d5SBarry Smith         A->ops->coloringpatch   = MatColoringPatch_SeqAIJ_Inode;
4121375a6242SBarry Smith       }
4122d3ac4fa3SBarry Smith     } else {
4123d3ac4fa3SBarry Smith       A->ops->solve = MatSolve_SeqAIJ_Inode_inplace;
4124d3ac4fa3SBarry Smith     }
41254c1414c8SBarry Smith     a->inode.node_count = node_count;
41264c1414c8SBarry Smith     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
41274c1414c8SBarry Smith   }
4128be6adb11SBarry Smith   a->inode.checked          = PETSC_TRUE;
4129a02bda8eSBarry Smith   a->inode.mat_nonzerostate = A->nonzerostate;
41304c1414c8SBarry Smith   PetscFunctionReturn(0);
41314c1414c8SBarry Smith }
41324c1414c8SBarry Smith 
4133150f0143SBarry Smith PetscErrorCode MatDuplicate_SeqAIJ_Inode(Mat A,MatDuplicateOption cpvalues,Mat *C)
4134150f0143SBarry Smith {
4135150f0143SBarry Smith   Mat            B =*C;
4136150f0143SBarry Smith   Mat_SeqAIJ     *c=(Mat_SeqAIJ*)B->data,*a=(Mat_SeqAIJ*)A->data;
4137150f0143SBarry Smith   PetscErrorCode ierr;
4138150f0143SBarry Smith   PetscInt       m=A->rmap->n;
4139150f0143SBarry Smith 
4140150f0143SBarry Smith   PetscFunctionBegin;
4141150f0143SBarry Smith   c->inode.use              = a->inode.use;
4142150f0143SBarry Smith   c->inode.limit            = a->inode.limit;
4143150f0143SBarry Smith   c->inode.max_limit        = a->inode.max_limit;
4144*ec710b6aSStefano Zampini   c->inode.checked          = PETSC_FALSE;
4145*ec710b6aSStefano Zampini   c->inode.size             = NULL;
4146*ec710b6aSStefano Zampini   c->inode.node_count       = 0;
4147*ec710b6aSStefano Zampini   c->inode.ibdiagvalid      = PETSC_FALSE;
4148*ec710b6aSStefano Zampini   c->inode.ibdiag           = NULL;
4149*ec710b6aSStefano Zampini   c->inode.bdiag            = NULL;
4150*ec710b6aSStefano Zampini   c->inode.mat_nonzerostate = -1;
4151b215bc84SStefano Zampini   if (a->inode.use) {
4152*ec710b6aSStefano Zampini     if (a->inode.checked && a->inode.size) {
4153b215bc84SStefano Zampini       ierr = PetscMalloc1(m+1,&c->inode.size);CHKERRQ(ierr);
4154580bdb30SBarry Smith       ierr = PetscArraycpy(c->inode.size,a->inode.size,m+1);CHKERRQ(ierr);
4155*ec710b6aSStefano Zampini 
4156*ec710b6aSStefano Zampini       c->inode.checked          = PETSC_TRUE;
4157*ec710b6aSStefano Zampini       c->inode.node_count       = a->inode.node_count;
4158*ec710b6aSStefano Zampini       c->inode.mat_nonzerostate = (*C)->nonzerostate;
4159*ec710b6aSStefano Zampini     }
4160a02bda8eSBarry Smith     /* note the table of functions below should match that in MatSeqAIJCheckInode() */
41612c451681SBarry Smith     if (!B->factortype) {
41622c451681SBarry Smith       B->ops->getrowij          = MatGetRowIJ_SeqAIJ_Inode;
41632c451681SBarry Smith       B->ops->restorerowij      = MatRestoreRowIJ_SeqAIJ_Inode;
41642c451681SBarry Smith       B->ops->getcolumnij       = MatGetColumnIJ_SeqAIJ_Inode;
41652c451681SBarry Smith       B->ops->restorecolumnij   = MatRestoreColumnIJ_SeqAIJ_Inode;
41662c451681SBarry Smith       B->ops->coloringpatch     = MatColoringPatch_SeqAIJ_Inode;
41672c451681SBarry Smith       B->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode;
4168150f0143SBarry Smith     } else {
41692c451681SBarry Smith       B->ops->solve = MatSolve_SeqAIJ_Inode_inplace;
4170150f0143SBarry Smith     }
4171150f0143SBarry Smith   }
4172150f0143SBarry Smith   PetscFunctionReturn(0);
4173150f0143SBarry Smith }
4174150f0143SBarry Smith 
41758758e1faSBarry 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)
417617454e89SShri Abhyankar {
41778758e1faSBarry Smith   PetscInt       k;
41788758e1faSBarry Smith   const PetscInt *vi;
41796e111a19SKarl Rupp 
418017454e89SShri Abhyankar   PetscFunctionBegin;
418117454e89SShri Abhyankar   vi = aj + ai[row];
418217454e89SShri Abhyankar   for (k=0; k<nzl; k++) cols[k] = vi[k];
418317454e89SShri Abhyankar   vi        = aj + adiag[row];
418417454e89SShri Abhyankar   cols[nzl] = vi[0];
418517454e89SShri Abhyankar   vi        = aj + adiag[row+1]+1;
418617454e89SShri Abhyankar   for (k=0; k<nzu; k++) cols[nzl+1+k] = vi[k];
418717454e89SShri Abhyankar   PetscFunctionReturn(0);
418817454e89SShri Abhyankar }
41896936b636SHong Zhang /*
4190a02bda8eSBarry Smith    MatSeqAIJCheckInode_FactorLU - Check Inode for factored seqaij matrix.
4191a02bda8eSBarry Smith    Modified from MatSeqAIJCheckInode().
41926936b636SHong Zhang 
41936936b636SHong Zhang    Input Parameters:
4194abb87a52SBarry Smith .  Mat A - ILU or LU matrix factor
4195abb87a52SBarry Smith 
41966936b636SHong Zhang */
4197a02bda8eSBarry Smith PetscErrorCode MatSeqAIJCheckInode_FactorLU(Mat A)
4198019b515eSShri Abhyankar {
4199019b515eSShri Abhyankar   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
4200019b515eSShri Abhyankar   PetscErrorCode ierr;
4201019b515eSShri Abhyankar   PetscInt       i,j,m,nzl1,nzu1,nzl2,nzu2,nzx,nzy,node_count,blk_size;
42028758e1faSBarry Smith   PetscInt       *cols1,*cols2,*ns;
42038758e1faSBarry Smith   const PetscInt *ai = a->i,*aj = a->j, *adiag = a->diag;
4204ace3abfcSBarry Smith   PetscBool      flag;
4205019b515eSShri Abhyankar 
4206019b515eSShri Abhyankar   PetscFunctionBegin;
4207019b515eSShri Abhyankar   if (!a->inode.use)    PetscFunctionReturn(0);
4208abb87a52SBarry Smith   if (a->inode.checked) PetscFunctionReturn(0);
4209019b515eSShri Abhyankar 
4210019b515eSShri Abhyankar   m = A->rmap->n;
42112205254eSKarl Rupp   if (a->inode.size) ns = a->inode.size;
42122205254eSKarl Rupp   else {
4213854ce69bSBarry Smith     ierr = PetscMalloc1(m+1,&ns);CHKERRQ(ierr);
42142205254eSKarl Rupp   }
4215019b515eSShri Abhyankar 
4216019b515eSShri Abhyankar   i          = 0;
4217019b515eSShri Abhyankar   node_count = 0;
4218dcca6d9dSJed Brown   ierr = PetscMalloc2(m,&cols1,m,&cols2);CHKERRQ(ierr);
4219019b515eSShri Abhyankar   while (i < m) {                /* For each row */
4220019b515eSShri Abhyankar     nzl1 = ai[i+1] - ai[i];       /* Number of nonzeros in L */
4221019b515eSShri Abhyankar     nzu1 = adiag[i] - adiag[i+1] - 1; /* Number of nonzeros in U excluding diagonal*/
4222019b515eSShri Abhyankar     nzx  = nzl1 + nzu1 + 1;
4223019b515eSShri Abhyankar     MatGetRow_FactoredLU(cols1,nzl1,nzu1,nzx,ai,aj,adiag,i);
4224019b515eSShri Abhyankar 
4225019b515eSShri Abhyankar     /* Limits the number of elements in a node to 'a->inode.limit' */
4226019b515eSShri Abhyankar     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
4227019b515eSShri Abhyankar       nzl2 = ai[j+1] - ai[j];
4228019b515eSShri Abhyankar       nzu2 = adiag[j] - adiag[j+1] - 1;
4229019b515eSShri Abhyankar       nzy  = nzl2 + nzu2 + 1;
4230019b515eSShri Abhyankar       if (nzy != nzx) break;
423117454e89SShri Abhyankar       ierr = MatGetRow_FactoredLU(cols2,nzl2,nzu2,nzy,ai,aj,adiag,j);CHKERRQ(ierr);
4232580bdb30SBarry Smith       ierr = PetscArraycmp(cols1,cols2,nzx,&flag);CHKERRQ(ierr);
42338758e1faSBarry Smith       if (!flag) break;
4234019b515eSShri Abhyankar     }
4235019b515eSShri Abhyankar     ns[node_count++] = blk_size;
4236019b515eSShri Abhyankar     i                = j;
4237019b515eSShri Abhyankar   }
42388758e1faSBarry Smith   ierr             = PetscFree2(cols1,cols2);CHKERRQ(ierr);
4239019b515eSShri Abhyankar   /* If not enough inodes found,, do not use inode version of the routines */
4240be6adb11SBarry Smith   if (!m || node_count > .8*m) {
4241019b515eSShri Abhyankar     ierr = PetscFree(ns);CHKERRQ(ierr);
42422205254eSKarl Rupp 
4243019b515eSShri Abhyankar     a->inode.node_count = 0;
42440298fd71SBarry Smith     a->inode.size       = NULL;
4245019b515eSShri Abhyankar     a->inode.use        = PETSC_FALSE;
42462205254eSKarl Rupp 
4247019b515eSShri Abhyankar     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
4248019b515eSShri Abhyankar   } else {
4249f4259b30SLisandro Dalcin     A->ops->mult              = NULL;
4250f4259b30SLisandro Dalcin     A->ops->sor               = NULL;
4251f4259b30SLisandro Dalcin     A->ops->multadd           = NULL;
4252f4259b30SLisandro Dalcin     A->ops->getrowij          = NULL;
4253f4259b30SLisandro Dalcin     A->ops->restorerowij      = NULL;
4254f4259b30SLisandro Dalcin     A->ops->getcolumnij       = NULL;
4255f4259b30SLisandro Dalcin     A->ops->restorecolumnij   = NULL;
4256f4259b30SLisandro Dalcin     A->ops->coloringpatch     = NULL;
4257f4259b30SLisandro Dalcin     A->ops->multdiagonalblock = NULL;
4258019b515eSShri Abhyankar     a->inode.node_count       = node_count;
4259019b515eSShri Abhyankar     a->inode.size             = ns;
42602205254eSKarl Rupp 
4261019b515eSShri Abhyankar     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
4262019b515eSShri Abhyankar   }
4263be6adb11SBarry Smith   a->inode.checked = PETSC_TRUE;
4264019b515eSShri Abhyankar   PetscFunctionReturn(0);
4265019b515eSShri Abhyankar }
4266019b515eSShri Abhyankar 
4267acf2f550SJed Brown PetscErrorCode MatSeqAIJInvalidateDiagonal_Inode(Mat A)
4268acf2f550SJed Brown {
4269acf2f550SJed Brown   Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data;
4270acf2f550SJed Brown 
4271acf2f550SJed Brown   PetscFunctionBegin;
4272acf2f550SJed Brown   a->inode.ibdiagvalid = PETSC_FALSE;
4273acf2f550SJed Brown   PetscFunctionReturn(0);
4274acf2f550SJed Brown }
4275acf2f550SJed Brown 
42764c1414c8SBarry Smith /*
42774c1414c8SBarry Smith      This is really ugly. if inodes are used this replaces the
42784c1414c8SBarry Smith   permutations with ones that correspond to rows/cols of the matrix
42794c1414c8SBarry Smith   rather then inode blocks
42804c1414c8SBarry Smith */
42817087cfbeSBarry Smith PetscErrorCode  MatInodeAdjustForInodes(Mat A,IS *rperm,IS *cperm)
42824c1414c8SBarry Smith {
42834ac538c5SBarry Smith   PetscErrorCode ierr;
42844c1414c8SBarry Smith 
42854c1414c8SBarry Smith   PetscFunctionBegin;
42864ac538c5SBarry Smith   ierr = PetscTryMethod(A,"MatInodeAdjustForInodes_C",(Mat,IS*,IS*),(A,rperm,cperm));CHKERRQ(ierr);
42874c1414c8SBarry Smith   PetscFunctionReturn(0);
42884c1414c8SBarry Smith }
42894c1414c8SBarry Smith 
42907087cfbeSBarry Smith PetscErrorCode  MatInodeAdjustForInodes_SeqAIJ_Inode(Mat A,IS *rperm,IS *cperm)
42914c1414c8SBarry Smith {
42924c1414c8SBarry Smith   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
42934c1414c8SBarry Smith   PetscErrorCode ierr;
42945d0c19d7SBarry Smith   PetscInt       m = A->rmap->n,n = A->cmap->n,i,j,nslim_row = a->inode.node_count;
42955d0c19d7SBarry Smith   const PetscInt *ridx,*cidx;
42964c1414c8SBarry Smith   PetscInt       row,col,*permr,*permc,*ns_row =  a->inode.size,*tns,start_val,end_val,indx;
42974c1414c8SBarry Smith   PetscInt       nslim_col,*ns_col;
42984c1414c8SBarry Smith   IS             ris = *rperm,cis = *cperm;
42994c1414c8SBarry Smith 
43004c1414c8SBarry Smith   PetscFunctionBegin;
43014c1414c8SBarry Smith   if (!a->inode.size) PetscFunctionReturn(0); /* no inodes so return */
43024c1414c8SBarry Smith   if (a->inode.node_count == m) PetscFunctionReturn(0); /* all inodes are of size 1 */
43034c1414c8SBarry Smith 
4304470880abSPatrick Sanan   ierr = MatCreateColInode_Private(A,&nslim_col,&ns_col);CHKERRQ(ierr);
4305854ce69bSBarry Smith   ierr = PetscMalloc1(((nslim_row>nslim_col) ? nslim_row : nslim_col)+1,&tns);CHKERRQ(ierr);
4306dcca6d9dSJed Brown   ierr = PetscMalloc2(m,&permr,n,&permc);CHKERRQ(ierr);
43074c1414c8SBarry Smith 
43084c1414c8SBarry Smith   ierr = ISGetIndices(ris,&ridx);CHKERRQ(ierr);
43094c1414c8SBarry Smith   ierr = ISGetIndices(cis,&cidx);CHKERRQ(ierr);
43104c1414c8SBarry Smith 
43114c1414c8SBarry Smith   /* Form the inode structure for the rows of permuted matric using inv perm*/
43124c1414c8SBarry Smith   for (i=0,tns[0]=0; i<nslim_row; ++i) tns[i+1] = tns[i] + ns_row[i];
43134c1414c8SBarry Smith 
43144c1414c8SBarry Smith   /* Construct the permutations for rows*/
43154c1414c8SBarry Smith   for (i=0,row = 0; i<nslim_row; ++i) {
43164c1414c8SBarry Smith     indx      = ridx[i];
43174c1414c8SBarry Smith     start_val = tns[indx];
43184c1414c8SBarry Smith     end_val   = tns[indx + 1];
43194c1414c8SBarry Smith     for (j=start_val; j<end_val; ++j,++row) permr[row]= j;
43204c1414c8SBarry Smith   }
43214c1414c8SBarry Smith 
43224c1414c8SBarry Smith   /* Form the inode structure for the columns of permuted matrix using inv perm*/
43234c1414c8SBarry Smith   for (i=0,tns[0]=0; i<nslim_col; ++i) tns[i+1] = tns[i] + ns_col[i];
43244c1414c8SBarry Smith 
43254c1414c8SBarry Smith   /* Construct permutations for columns */
43264c1414c8SBarry Smith   for (i=0,col=0; i<nslim_col; ++i) {
43274c1414c8SBarry Smith     indx      = cidx[i];
43284c1414c8SBarry Smith     start_val = tns[indx];
43294c1414c8SBarry Smith     end_val   = tns[indx + 1];
43304c1414c8SBarry Smith     for (j = start_val; j<end_val; ++j,++col) permc[col]= j;
43314c1414c8SBarry Smith   }
43324c1414c8SBarry Smith 
433370b3c8c7SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permr,PETSC_COPY_VALUES,rperm);CHKERRQ(ierr);
43340e83c824SBarry Smith   ierr = ISSetPermutation(*rperm);CHKERRQ(ierr);
433570b3c8c7SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permc,PETSC_COPY_VALUES,cperm);CHKERRQ(ierr);
43360e83c824SBarry Smith   ierr = ISSetPermutation(*cperm);CHKERRQ(ierr);
43374c1414c8SBarry Smith 
43384c1414c8SBarry Smith   ierr = ISRestoreIndices(ris,&ridx);CHKERRQ(ierr);
43394c1414c8SBarry Smith   ierr = ISRestoreIndices(cis,&cidx);CHKERRQ(ierr);
43404c1414c8SBarry Smith 
43414c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
43420e83c824SBarry Smith   ierr = PetscFree2(permr,permc);CHKERRQ(ierr);
43436bf464f9SBarry Smith   ierr = ISDestroy(&cis);CHKERRQ(ierr);
43446bf464f9SBarry Smith   ierr = ISDestroy(&ris);CHKERRQ(ierr);
43454c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
43464c1414c8SBarry Smith   PetscFunctionReturn(0);
43474c1414c8SBarry Smith }
43484c1414c8SBarry Smith 
43494c1414c8SBarry Smith /*@C
43504c1414c8SBarry Smith    MatInodeGetInodeSizes - Returns the inode information of the Inode matrix.
43514c1414c8SBarry Smith 
43523f9fe445SBarry Smith    Not Collective
43534c1414c8SBarry Smith 
43544c1414c8SBarry Smith    Input Parameter:
43554c1414c8SBarry Smith .  A - the Inode matrix or matrix derived from the Inode class -- e.g., SeqAIJ
43564c1414c8SBarry Smith 
43574c1414c8SBarry Smith    Output Parameter:
43584c1414c8SBarry Smith +  node_count - no of inodes present in the matrix.
43594c1414c8SBarry Smith .  sizes      - an array of size node_count,with sizes of each inode.
43604c1414c8SBarry Smith -  limit      - the max size used to generate the inodes.
43614c1414c8SBarry Smith 
43624c1414c8SBarry Smith    Level: advanced
43634c1414c8SBarry Smith 
436495452b02SPatrick Sanan    Notes:
436595452b02SPatrick Sanan     This routine returns some internal storage information
43664c1414c8SBarry Smith    of the matrix, it is intended to be used by advanced users.
43674c1414c8SBarry Smith    It should be called after the matrix is assembled.
43684c1414c8SBarry Smith    The contents of the sizes[] array should not be changed.
43690298fd71SBarry Smith    NULL may be passed for information not requested.
43704c1414c8SBarry Smith 
43714c1414c8SBarry Smith .seealso: MatGetInfo()
43724c1414c8SBarry Smith @*/
43737087cfbeSBarry Smith PetscErrorCode  MatInodeGetInodeSizes(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit)
43744c1414c8SBarry Smith {
43754c1414c8SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscInt*,PetscInt*[],PetscInt*);
43764c1414c8SBarry Smith 
43774c1414c8SBarry Smith   PetscFunctionBegin;
4378e32f2f54SBarry Smith   if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
43790005d66cSJed Brown   ierr = PetscObjectQueryFunction((PetscObject)A,"MatInodeGetInodeSizes_C",&f);CHKERRQ(ierr);
43804c1414c8SBarry Smith   if (f) {
43814c1414c8SBarry Smith     ierr = (*f)(A,node_count,sizes,limit);CHKERRQ(ierr);
43824c1414c8SBarry Smith   }
43834c1414c8SBarry Smith   PetscFunctionReturn(0);
43844c1414c8SBarry Smith }
43854c1414c8SBarry Smith 
43867087cfbeSBarry Smith PetscErrorCode  MatInodeGetInodeSizes_SeqAIJ_Inode(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit)
43874c1414c8SBarry Smith {
43884c1414c8SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
43894c1414c8SBarry Smith 
43904c1414c8SBarry Smith   PetscFunctionBegin;
43914c1414c8SBarry Smith   if (node_count) *node_count = a->inode.node_count;
43924c1414c8SBarry Smith   if (sizes)      *sizes      = a->inode.size;
43934c1414c8SBarry Smith   if (limit)      *limit      = a->inode.limit;
43944c1414c8SBarry Smith   PetscFunctionReturn(0);
43954c1414c8SBarry Smith }
4396