xref: /petsc/src/mat/impls/aij/seq/inode.c (revision 7b6c816cf7b5355c1764dbf81d50649d87d3fa60)
14c1414c8SBarry Smith 
24c1414c8SBarry Smith /*
34c1414c8SBarry Smith   This file provides high performance routines for the Inode format (compressed sparse row)
44c1414c8SBarry Smith   by taking advantage of rows with identical nonzero structure (I-nodes).
54c1414c8SBarry Smith */
6c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/aij.h>
74c1414c8SBarry Smith 
84c1414c8SBarry Smith #undef __FUNCT__
94c1414c8SBarry Smith #define __FUNCT__ "Mat_CreateColInode"
104c1414c8SBarry Smith static PetscErrorCode Mat_CreateColInode(Mat A,PetscInt *size,PetscInt **ns)
114c1414c8SBarry Smith {
124c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
134c1414c8SBarry Smith   PetscErrorCode ierr;
144c1414c8SBarry Smith   PetscInt       i,count,m,n,min_mn,*ns_row,*ns_col;
154c1414c8SBarry Smith 
164c1414c8SBarry Smith   PetscFunctionBegin;
17d0f46423SBarry Smith   n      = A->cmap->n;
18d0f46423SBarry Smith   m      = A->rmap->n;
194c1414c8SBarry Smith   ns_row = a->inode.size;
204c1414c8SBarry Smith 
214c1414c8SBarry Smith   min_mn = (m < n) ? m : n;
224c1414c8SBarry Smith   if (!ns) {
234c1414c8SBarry Smith     for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) ;
244c1414c8SBarry Smith     for (; count+1 < n; count++,i++) ;
254c1414c8SBarry Smith     if (count < n)  {
264c1414c8SBarry Smith       i++;
274c1414c8SBarry Smith     }
284c1414c8SBarry Smith     *size = i;
294c1414c8SBarry Smith     PetscFunctionReturn(0);
304c1414c8SBarry Smith   }
31854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&ns_col);CHKERRQ(ierr);
324c1414c8SBarry Smith 
334c1414c8SBarry Smith   /* Use the same row structure wherever feasible. */
344c1414c8SBarry Smith   for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) {
354c1414c8SBarry Smith     ns_col[i] = ns_row[i];
364c1414c8SBarry Smith   }
374c1414c8SBarry Smith 
384c1414c8SBarry Smith   /* if m < n; pad up the remainder with inode_limit */
394c1414c8SBarry Smith   for (; count+1 < n; count++,i++) {
404c1414c8SBarry Smith     ns_col[i] = 1;
414c1414c8SBarry Smith   }
424c1414c8SBarry Smith   /* The last node is the odd ball. padd it up with the remaining rows; */
434c1414c8SBarry Smith   if (count < n) {
444c1414c8SBarry Smith     ns_col[i] = n - count;
454c1414c8SBarry Smith     i++;
464c1414c8SBarry Smith   } else if (count > n) {
474c1414c8SBarry Smith     /* Adjust for the over estimation */
484c1414c8SBarry Smith     ns_col[i-1] += n - count;
494c1414c8SBarry Smith   }
504c1414c8SBarry Smith   *size = i;
514c1414c8SBarry Smith   *ns   = ns_col;
524c1414c8SBarry Smith   PetscFunctionReturn(0);
534c1414c8SBarry Smith }
544c1414c8SBarry Smith 
554c1414c8SBarry Smith 
564c1414c8SBarry Smith /*
574c1414c8SBarry Smith       This builds symmetric version of nonzero structure,
584c1414c8SBarry Smith */
594c1414c8SBarry Smith #undef __FUNCT__
604108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode_Symmetric"
611a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Symmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift)
624c1414c8SBarry Smith {
634c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
644c1414c8SBarry Smith   PetscErrorCode ierr;
658758e1faSBarry Smith   PetscInt       *work,*ia,*ja,nz,nslim_row,nslim_col,m,row,col,n;
668758e1faSBarry Smith   PetscInt       *tns,*tvc,*ns_row = a->inode.size,*ns_col,nsz,i1,i2;
678758e1faSBarry Smith   const PetscInt *j,*jmax,*ai= a->i,*aj = a->j;
684c1414c8SBarry Smith 
694c1414c8SBarry Smith   PetscFunctionBegin;
704c1414c8SBarry Smith   nslim_row = a->inode.node_count;
71d0f46423SBarry Smith   m         = A->rmap->n;
72d0f46423SBarry Smith   n         = A->cmap->n;
73e32f2f54SBarry Smith   if (m != n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatGetRowIJ_SeqAIJ_Inode_Symmetric: Matrix should be square");
744c1414c8SBarry Smith 
754c1414c8SBarry Smith   /* Use the row_inode as column_inode */
764c1414c8SBarry Smith   nslim_col = nslim_row;
774c1414c8SBarry Smith   ns_col    = ns_row;
784c1414c8SBarry Smith 
794c1414c8SBarry Smith   /* allocate space for reformated inode structure */
80854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_col+1,&tns);CHKERRQ(ierr);
81854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&tvc);CHKERRQ(ierr);
824c1414c8SBarry Smith   for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1]+ ns_row[i1];
834c1414c8SBarry Smith 
844c1414c8SBarry Smith   for (i1=0,col=0; i1<nslim_col; ++i1) {
854c1414c8SBarry Smith     nsz = ns_col[i1];
862205254eSKarl Rupp     for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1;
874c1414c8SBarry Smith   }
884c1414c8SBarry Smith   /* allocate space for row pointers */
89854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_row+1,&ia);CHKERRQ(ierr);
904c1414c8SBarry Smith   *iia = ia;
914c1414c8SBarry Smith   ierr = PetscMemzero(ia,(nslim_row+1)*sizeof(PetscInt));CHKERRQ(ierr);
92854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_row+1,&work);CHKERRQ(ierr);
934c1414c8SBarry Smith 
944c1414c8SBarry Smith   /* determine the number of columns in each row */
954c1414c8SBarry Smith   ia[0] = oshift;
964c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
974c1414c8SBarry Smith 
984c1414c8SBarry Smith     j    = aj + ai[row] + ishift;
994c1414c8SBarry Smith     jmax = aj + ai[row+1] + ishift;
10083fed2edSSatish Balay     if (j==jmax) continue; /* empty row */
1014c1414c8SBarry Smith     col  = *j++ + ishift;
1024c1414c8SBarry Smith     i2   = tvc[col];
1034c1414c8SBarry Smith     while (i2<i1 && j<jmax) { /* 1.[-xx-d-xx--] 2.[-xx-------],off-diagonal elemets */
1044c1414c8SBarry Smith       ia[i1+1]++;
1054c1414c8SBarry Smith       ia[i2+1]++;
1064c1414c8SBarry Smith       i2++;                     /* Start col of next node */
10790d2dec7SBarry Smith       while ((j<jmax) && ((col=*j+ishift)<tns[i2])) ++j;
1084c1414c8SBarry Smith       i2 = tvc[col];
1094c1414c8SBarry Smith     }
1104c1414c8SBarry Smith     if (i2 == i1) ia[i2+1]++;    /* now the diagonal element */
1114c1414c8SBarry Smith   }
1124c1414c8SBarry Smith 
1134c1414c8SBarry Smith   /* shift ia[i] to point to next row */
1144c1414c8SBarry Smith   for (i1=1; i1<nslim_row+1; i1++) {
1154c1414c8SBarry Smith     row        = ia[i1-1];
1164c1414c8SBarry Smith     ia[i1]    += row;
1174c1414c8SBarry Smith     work[i1-1] = row - oshift;
1184c1414c8SBarry Smith   }
1194c1414c8SBarry Smith 
1204c1414c8SBarry Smith   /* allocate space for column pointers */
1214c1414c8SBarry Smith   nz   = ia[nslim_row] + (!ishift);
122785e854fSJed Brown   ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr);
1234c1414c8SBarry Smith   *jja = ja;
1244c1414c8SBarry Smith 
1254c1414c8SBarry Smith   /* loop over lower triangular part putting into ja */
1264c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row += ns_row[i1],i1++) {
1274c1414c8SBarry Smith     j    = aj + ai[row] + ishift;
1284c1414c8SBarry Smith     jmax = aj + ai[row+1] + ishift;
12983fed2edSSatish Balay     if (j==jmax) continue; /* empty row */
1304c1414c8SBarry Smith     col  = *j++ + ishift;
1314c1414c8SBarry Smith     i2   = tvc[col];
1324c1414c8SBarry Smith     while (i2<i1 && j<jmax) {
1334c1414c8SBarry Smith       ja[work[i2]++] = i1 + oshift;
1344c1414c8SBarry Smith       ja[work[i1]++] = i2 + oshift;
1354c1414c8SBarry Smith       ++i2;
13690d2dec7SBarry Smith       while ((j<jmax) && ((col=*j+ishift)< tns[i2])) ++j; /* Skip rest col indices in this node */
1374c1414c8SBarry Smith       i2 = tvc[col];
1384c1414c8SBarry Smith     }
1394c1414c8SBarry Smith     if (i2 == i1) ja[work[i1]++] = i2 + oshift;
1404c1414c8SBarry Smith 
1414c1414c8SBarry Smith   }
1424c1414c8SBarry Smith   ierr = PetscFree(work);CHKERRQ(ierr);
1434c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
1444c1414c8SBarry Smith   ierr = PetscFree(tvc);CHKERRQ(ierr);
1454c1414c8SBarry Smith   PetscFunctionReturn(0);
1464c1414c8SBarry Smith }
1474c1414c8SBarry Smith 
1484c1414c8SBarry Smith /*
1494c1414c8SBarry Smith       This builds nonsymmetric version of nonzero structure,
1504c1414c8SBarry Smith */
1514c1414c8SBarry Smith #undef __FUNCT__
1524108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric"
1531a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift)
1544c1414c8SBarry Smith {
1554c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
1564c1414c8SBarry Smith   PetscErrorCode ierr;
1578758e1faSBarry Smith   PetscInt       *work,*ia,*ja,nz,nslim_row,n,row,col,*ns_col,nslim_col;
1588758e1faSBarry Smith   PetscInt       *tns,*tvc,nsz,i1,i2;
1598758e1faSBarry Smith   const PetscInt *j,*ai= a->i,*aj = a->j,*ns_row = a->inode.size;
1604c1414c8SBarry Smith 
1614c1414c8SBarry Smith   PetscFunctionBegin;
1624c1414c8SBarry Smith   nslim_row = a->inode.node_count;
163d0f46423SBarry Smith   n         = A->cmap->n;
1644c1414c8SBarry Smith 
1654c1414c8SBarry Smith   /* Create The column_inode for this matrix */
1664c1414c8SBarry Smith   ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr);
1674c1414c8SBarry Smith 
1684c1414c8SBarry Smith   /* allocate space for reformated column_inode structure */
169854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_col +1,&tns);CHKERRQ(ierr);
170854ce69bSBarry Smith   ierr = PetscMalloc1(n + 1,&tvc);CHKERRQ(ierr);
1714c1414c8SBarry Smith   for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1];
1724c1414c8SBarry Smith 
1734c1414c8SBarry Smith   for (i1=0,col=0; i1<nslim_col; ++i1) {
1744c1414c8SBarry Smith     nsz = ns_col[i1];
1752205254eSKarl Rupp     for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1;
1764c1414c8SBarry Smith   }
1774c1414c8SBarry Smith   /* allocate space for row pointers */
178854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_row+1,&ia);CHKERRQ(ierr);
1794c1414c8SBarry Smith   *iia = ia;
1804c1414c8SBarry Smith   ierr = PetscMemzero(ia,(nslim_row+1)*sizeof(PetscInt));CHKERRQ(ierr);
181854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_row+1,&work);CHKERRQ(ierr);
1824c1414c8SBarry Smith 
1834c1414c8SBarry Smith   /* determine the number of columns in each row */
1844c1414c8SBarry Smith   ia[0] = oshift;
1854c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
1864c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
18783fed2edSSatish Balay     nz  = ai[row+1] - ai[row];
18883fed2edSSatish Balay     if (!nz) continue; /* empty row */
1894c1414c8SBarry Smith     col = *j++ + ishift;
1904c1414c8SBarry Smith     i2  = tvc[col];
1914c1414c8SBarry Smith     while (nz-- > 0) {           /* off-diagonal elemets */
1924c1414c8SBarry Smith       ia[i1+1]++;
1934c1414c8SBarry Smith       i2++;                     /* Start col of next node */
194a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
1954c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
1964c1414c8SBarry Smith     }
1974c1414c8SBarry Smith   }
1984c1414c8SBarry Smith 
1994c1414c8SBarry Smith   /* shift ia[i] to point to next row */
2004c1414c8SBarry Smith   for (i1=1; i1<nslim_row+1; i1++) {
2014c1414c8SBarry Smith     row        = ia[i1-1];
2024c1414c8SBarry Smith     ia[i1]    += row;
2034c1414c8SBarry Smith     work[i1-1] = row - oshift;
2044c1414c8SBarry Smith   }
2054c1414c8SBarry Smith 
2064c1414c8SBarry Smith   /* allocate space for column pointers */
2074c1414c8SBarry Smith   nz   = ia[nslim_row] + (!ishift);
208785e854fSJed Brown   ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr);
2094c1414c8SBarry Smith   *jja = ja;
2104c1414c8SBarry Smith 
2114c1414c8SBarry Smith   /* loop over matrix putting into ja */
2124c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
2134c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
21483fed2edSSatish Balay     nz  = ai[row+1] - ai[row];
21583fed2edSSatish Balay     if (!nz) continue; /* empty row */
2164c1414c8SBarry Smith     col = *j++ + ishift;
2174c1414c8SBarry Smith     i2  = tvc[col];
2184c1414c8SBarry Smith     while (nz-- > 0) {
2194c1414c8SBarry Smith       ja[work[i1]++] = i2 + oshift;
2204c1414c8SBarry Smith       ++i2;
221a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
2224c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
2234c1414c8SBarry Smith     }
2244c1414c8SBarry Smith   }
2254c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
2264c1414c8SBarry Smith   ierr = PetscFree(work);CHKERRQ(ierr);
2274c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
2284c1414c8SBarry Smith   ierr = PetscFree(tvc);CHKERRQ(ierr);
2294c1414c8SBarry Smith   PetscFunctionReturn(0);
2304c1414c8SBarry Smith }
2314c1414c8SBarry Smith 
2324c1414c8SBarry Smith #undef __FUNCT__
2334108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode"
2341a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
2354c1414c8SBarry Smith {
2364c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
2374c1414c8SBarry Smith   PetscErrorCode ierr;
2384c1414c8SBarry Smith 
2394c1414c8SBarry Smith   PetscFunctionBegin;
2404c1414c8SBarry Smith   *n = a->inode.node_count;
2414c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
2428f7157efSSatish Balay   if (!blockcompressed) {
2438f7157efSSatish Balay     ierr = MatGetRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);;
2448f7157efSSatish Balay   } else if (symmetric) {
2454108e4d5SBarry Smith     ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
2464c1414c8SBarry Smith   } else {
2474108e4d5SBarry Smith     ierr = MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
2484c1414c8SBarry Smith   }
2494c1414c8SBarry Smith   PetscFunctionReturn(0);
2504c1414c8SBarry Smith }
2514c1414c8SBarry Smith 
2524c1414c8SBarry Smith #undef __FUNCT__
2534108e4d5SBarry Smith #define __FUNCT__ "MatRestoreRowIJ_SeqAIJ_Inode"
2541a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
2554c1414c8SBarry Smith {
2564c1414c8SBarry Smith   PetscErrorCode ierr;
2574c1414c8SBarry Smith 
2584c1414c8SBarry Smith   PetscFunctionBegin;
2594c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
2608f7157efSSatish Balay 
2618f7157efSSatish Balay   if (!blockcompressed) {
2628f7157efSSatish Balay     ierr = MatRestoreRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);;
2638f7157efSSatish Balay   } else {
2644c1414c8SBarry Smith     ierr = PetscFree(*ia);CHKERRQ(ierr);
2654c1414c8SBarry Smith     ierr = PetscFree(*ja);CHKERRQ(ierr);
2668f7157efSSatish Balay   }
2674c1414c8SBarry Smith   PetscFunctionReturn(0);
2684c1414c8SBarry Smith }
2694c1414c8SBarry Smith 
2704c1414c8SBarry Smith /* ----------------------------------------------------------- */
2714c1414c8SBarry Smith 
2724c1414c8SBarry Smith #undef __FUNCT__
2734108e4d5SBarry Smith #define __FUNCT__ "MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric"
2741a83f524SJed Brown static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift)
2754c1414c8SBarry Smith {
2764c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
2774c1414c8SBarry Smith   PetscErrorCode ierr;
2784c1414c8SBarry Smith   PetscInt       *work,*ia,*ja,*j,nz,nslim_row, n,row,col,*ns_col,nslim_col;
2794c1414c8SBarry Smith   PetscInt       *tns,*tvc,*ns_row = a->inode.size,nsz,i1,i2,*ai= a->i,*aj = a->j;
2804c1414c8SBarry Smith 
2814c1414c8SBarry Smith   PetscFunctionBegin;
2824c1414c8SBarry Smith   nslim_row = a->inode.node_count;
283d0f46423SBarry Smith   n         = A->cmap->n;
2844c1414c8SBarry Smith 
2854c1414c8SBarry Smith   /* Create The column_inode for this matrix */
2864c1414c8SBarry Smith   ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr);
2874c1414c8SBarry Smith 
2884c1414c8SBarry Smith   /* allocate space for reformated column_inode structure */
289854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_col + 1,&tns);CHKERRQ(ierr);
290854ce69bSBarry Smith   ierr = PetscMalloc1(n + 1,&tvc);CHKERRQ(ierr);
2914c1414c8SBarry Smith   for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1];
2924c1414c8SBarry Smith 
2934c1414c8SBarry Smith   for (i1=0,col=0; i1<nslim_col; ++i1) {
2944c1414c8SBarry Smith     nsz = ns_col[i1];
2952205254eSKarl Rupp     for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1;
2964c1414c8SBarry Smith   }
2974c1414c8SBarry Smith   /* allocate space for column pointers */
298854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_col+1,&ia);CHKERRQ(ierr);
2994c1414c8SBarry Smith   *iia = ia;
3004c1414c8SBarry Smith   ierr = PetscMemzero(ia,(nslim_col+1)*sizeof(PetscInt));CHKERRQ(ierr);
301854ce69bSBarry Smith   ierr = PetscMalloc1(nslim_col+1,&work);CHKERRQ(ierr);
3024c1414c8SBarry Smith 
3034c1414c8SBarry Smith   /* determine the number of columns in each row */
3044c1414c8SBarry Smith   ia[0] = oshift;
3054c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
3064c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
3074c1414c8SBarry Smith     col = *j++ + ishift;
3084c1414c8SBarry Smith     i2  = tvc[col];
3094c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
3104c1414c8SBarry Smith     while (nz-- > 0) {           /* off-diagonal elemets */
3114c1414c8SBarry Smith       /* ia[i1+1]++; */
3124c1414c8SBarry Smith       ia[i2+1]++;
3134c1414c8SBarry Smith       i2++;
314a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
3154c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
3164c1414c8SBarry Smith     }
3174c1414c8SBarry Smith   }
3184c1414c8SBarry Smith 
3194c1414c8SBarry Smith   /* shift ia[i] to point to next col */
3204c1414c8SBarry Smith   for (i1=1; i1<nslim_col+1; i1++) {
3214c1414c8SBarry Smith     col        = ia[i1-1];
3224c1414c8SBarry Smith     ia[i1]    += col;
3234c1414c8SBarry Smith     work[i1-1] = col - oshift;
3244c1414c8SBarry Smith   }
3254c1414c8SBarry Smith 
3264c1414c8SBarry Smith   /* allocate space for column pointers */
3274c1414c8SBarry Smith   nz   = ia[nslim_col] + (!ishift);
328785e854fSJed Brown   ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr);
3294c1414c8SBarry Smith   *jja = ja;
3304c1414c8SBarry Smith 
3314c1414c8SBarry Smith   /* loop over matrix putting into ja */
3324c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
3334c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
3344c1414c8SBarry Smith     col = *j++ + ishift;
3354c1414c8SBarry Smith     i2  = tvc[col];
3364c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
3374c1414c8SBarry Smith     while (nz-- > 0) {
3384c1414c8SBarry Smith       /* ja[work[i1]++] = i2 + oshift; */
3394c1414c8SBarry Smith       ja[work[i2]++] = i1 + oshift;
3404c1414c8SBarry Smith       i2++;
341a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
3424c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
3434c1414c8SBarry Smith     }
3444c1414c8SBarry Smith   }
3454c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
3464c1414c8SBarry Smith   ierr = PetscFree(work);CHKERRQ(ierr);
3474c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
3484c1414c8SBarry Smith   ierr = PetscFree(tvc);CHKERRQ(ierr);
3494c1414c8SBarry Smith   PetscFunctionReturn(0);
3504c1414c8SBarry Smith }
3514c1414c8SBarry Smith 
3524c1414c8SBarry Smith #undef __FUNCT__
3534108e4d5SBarry Smith #define __FUNCT__ "MatGetColumnIJ_SeqAIJ_Inode"
3541a83f524SJed Brown static PetscErrorCode MatGetColumnIJ_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;
3590298fd71SBarry Smith   ierr = Mat_CreateColInode(A,n,NULL);CHKERRQ(ierr);
3604c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
3614c1414c8SBarry Smith 
3628f7157efSSatish Balay   if (!blockcompressed) {
3638f7157efSSatish Balay     ierr = MatGetColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);;
3648f7157efSSatish Balay   } else if (symmetric) {
3654c1414c8SBarry Smith     /* Since the indices are symmetric it does'nt matter */
3664108e4d5SBarry Smith     ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
3674c1414c8SBarry Smith   } else {
3684108e4d5SBarry Smith     ierr = MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
3694c1414c8SBarry Smith   }
3704c1414c8SBarry Smith   PetscFunctionReturn(0);
3714c1414c8SBarry Smith }
3724c1414c8SBarry Smith 
3734c1414c8SBarry Smith #undef __FUNCT__
3744108e4d5SBarry Smith #define __FUNCT__ "MatRestoreColumnIJ_SeqAIJ_Inode"
3751a83f524SJed Brown static PetscErrorCode MatRestoreColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
3764c1414c8SBarry Smith {
3774c1414c8SBarry Smith   PetscErrorCode ierr;
3784c1414c8SBarry Smith 
3794c1414c8SBarry Smith   PetscFunctionBegin;
3804c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
3818f7157efSSatish Balay   if (!blockcompressed) {
3828f7157efSSatish Balay     ierr = MatRestoreColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);;
3838f7157efSSatish Balay   } else {
3844c1414c8SBarry Smith     ierr = PetscFree(*ia);CHKERRQ(ierr);
3854c1414c8SBarry Smith     ierr = PetscFree(*ja);CHKERRQ(ierr);
3868f7157efSSatish Balay   }
3874c1414c8SBarry Smith   PetscFunctionReturn(0);
3884c1414c8SBarry Smith }
3894c1414c8SBarry Smith 
3904c1414c8SBarry Smith /* ----------------------------------------------------------- */
3914c1414c8SBarry Smith 
3924c1414c8SBarry Smith #undef __FUNCT__
3934108e4d5SBarry Smith #define __FUNCT__ "MatMult_SeqAIJ_Inode"
3944108e4d5SBarry Smith static PetscErrorCode MatMult_SeqAIJ_Inode(Mat A,Vec xx,Vec yy)
3954c1414c8SBarry Smith {
3964c1414c8SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3974c1414c8SBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5,tmp0,tmp1;
398d9fead3dSBarry Smith   PetscScalar       *y;
399dd6ea824SBarry Smith   const PetscScalar *x;
400dd6ea824SBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5;
4014c1414c8SBarry Smith   PetscErrorCode    ierr;
4028758e1faSBarry Smith   PetscInt          i1,i2,n,i,row,node_max,nsz,sz,nonzerorow=0;
4038758e1faSBarry Smith   const PetscInt    *idx,*ns,*ii;
4044c1414c8SBarry Smith 
4054c1414c8SBarry Smith #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
4064c1414c8SBarry Smith #pragma disjoint(*x,*y,*v1,*v2,*v3,*v4,*v5)
4074c1414c8SBarry Smith #endif
4084c1414c8SBarry Smith 
4094c1414c8SBarry Smith   PetscFunctionBegin;
410e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
4114c1414c8SBarry Smith   node_max = a->inode.node_count;
4124c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
4133649974fSBarry Smith   ierr     = VecGetArrayRead(xx,&x);CHKERRQ(ierr);
4144c1414c8SBarry Smith   ierr     = VecGetArray(yy,&y);CHKERRQ(ierr);
4154c1414c8SBarry Smith   idx      = a->j;
4164c1414c8SBarry Smith   v1       = a->a;
4174c1414c8SBarry Smith   ii       = a->i;
4184c1414c8SBarry Smith 
4194c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i) {
4204c1414c8SBarry Smith     nsz         = ns[i];
4214c1414c8SBarry Smith     n           = ii[1] - ii[0];
42298c9bda7SSatish Balay     nonzerorow += (n>0)*nsz;
4234c1414c8SBarry Smith     ii         += nsz;
42450d8bf02SJed Brown     PetscPrefetchBlock(idx+nsz*n,n,0,PETSC_PREFETCH_HINT_NTA);    /* Prefetch the indices for the block row after the current one */
42550d8bf02SJed Brown     PetscPrefetchBlock(v1+nsz*n,nsz*n,0,PETSC_PREFETCH_HINT_NTA); /* Prefetch the values for the block row after the current one  */
4264c1414c8SBarry Smith     sz = n;                     /* No of non zeros in this row */
4274c1414c8SBarry Smith                                 /* Switch on the size of Node */
4284c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
4294c1414c8SBarry Smith     case 1:
43075567043SBarry Smith       sum1 = 0.;
4314c1414c8SBarry Smith 
4324c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4334c1414c8SBarry Smith         i1    = idx[0];         /* The instructions are ordered to */
4344c1414c8SBarry Smith         i2    = idx[1];         /* make the compiler's job easy */
4354c1414c8SBarry Smith         idx  += 2;
4364c1414c8SBarry Smith         tmp0  = x[i1];
4374c1414c8SBarry Smith         tmp1  = x[i2];
4384c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4394c1414c8SBarry Smith       }
4404c1414c8SBarry Smith 
4414c1414c8SBarry Smith       if (n == sz-1) {          /* Take care of the last nonzero  */
4424c1414c8SBarry Smith         tmp0  = x[*idx++];
4434c1414c8SBarry Smith         sum1 += *v1++ *tmp0;
4444c1414c8SBarry Smith       }
4454c1414c8SBarry Smith       y[row++]=sum1;
4464c1414c8SBarry Smith       break;
4474c1414c8SBarry Smith     case 2:
44875567043SBarry Smith       sum1 = 0.;
44975567043SBarry Smith       sum2 = 0.;
4504c1414c8SBarry Smith       v2   = v1 + n;
4514c1414c8SBarry Smith 
4524c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4534c1414c8SBarry Smith         i1    = idx[0];
4544c1414c8SBarry Smith         i2    = idx[1];
4554c1414c8SBarry Smith         idx  += 2;
4564c1414c8SBarry Smith         tmp0  = x[i1];
4574c1414c8SBarry Smith         tmp1  = x[i2];
4584c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4594c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
4604c1414c8SBarry Smith       }
4614c1414c8SBarry Smith       if (n == sz-1) {
4624c1414c8SBarry Smith         tmp0  = x[*idx++];
4634c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
4644c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
4654c1414c8SBarry Smith       }
4664c1414c8SBarry Smith       y[row++]=sum1;
4674c1414c8SBarry Smith       y[row++]=sum2;
4684c1414c8SBarry Smith       v1      =v2;              /* Since the next block to be processed starts there*/
4694c1414c8SBarry Smith       idx    +=sz;
4704c1414c8SBarry Smith       break;
4714c1414c8SBarry Smith     case 3:
47275567043SBarry Smith       sum1 = 0.;
47375567043SBarry Smith       sum2 = 0.;
47475567043SBarry Smith       sum3 = 0.;
4754c1414c8SBarry Smith       v2   = v1 + n;
4764c1414c8SBarry Smith       v3   = v2 + n;
4774c1414c8SBarry Smith 
4784c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4794c1414c8SBarry Smith         i1    = idx[0];
4804c1414c8SBarry Smith         i2    = idx[1];
4814c1414c8SBarry Smith         idx  += 2;
4824c1414c8SBarry Smith         tmp0  = x[i1];
4834c1414c8SBarry Smith         tmp1  = x[i2];
4844c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4854c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
4864c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
4874c1414c8SBarry Smith       }
4884c1414c8SBarry Smith       if (n == sz-1) {
4894c1414c8SBarry Smith         tmp0  = x[*idx++];
4904c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
4914c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
4924c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
4934c1414c8SBarry Smith       }
4944c1414c8SBarry Smith       y[row++]=sum1;
4954c1414c8SBarry Smith       y[row++]=sum2;
4964c1414c8SBarry Smith       y[row++]=sum3;
4974c1414c8SBarry Smith       v1      =v3;              /* Since the next block to be processed starts there*/
4984c1414c8SBarry Smith       idx    +=2*sz;
4994c1414c8SBarry Smith       break;
5004c1414c8SBarry Smith     case 4:
50175567043SBarry Smith       sum1 = 0.;
50275567043SBarry Smith       sum2 = 0.;
50375567043SBarry Smith       sum3 = 0.;
50475567043SBarry Smith       sum4 = 0.;
5054c1414c8SBarry Smith       v2   = v1 + n;
5064c1414c8SBarry Smith       v3   = v2 + n;
5074c1414c8SBarry Smith       v4   = v3 + n;
5084c1414c8SBarry Smith 
5094c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
5104c1414c8SBarry Smith         i1    = idx[0];
5114c1414c8SBarry Smith         i2    = idx[1];
5124c1414c8SBarry Smith         idx  += 2;
5134c1414c8SBarry Smith         tmp0  = x[i1];
5144c1414c8SBarry Smith         tmp1  = x[i2];
5154c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
5164c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
5174c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
5184c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
5194c1414c8SBarry Smith       }
5204c1414c8SBarry Smith       if (n == sz-1) {
5214c1414c8SBarry Smith         tmp0  = x[*idx++];
5224c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
5234c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
5244c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
5254c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
5264c1414c8SBarry Smith       }
5274c1414c8SBarry Smith       y[row++]=sum1;
5284c1414c8SBarry Smith       y[row++]=sum2;
5294c1414c8SBarry Smith       y[row++]=sum3;
5304c1414c8SBarry Smith       y[row++]=sum4;
5314c1414c8SBarry Smith       v1      =v4;              /* Since the next block to be processed starts there*/
5324c1414c8SBarry Smith       idx    +=3*sz;
5334c1414c8SBarry Smith       break;
5344c1414c8SBarry Smith     case 5:
53575567043SBarry Smith       sum1 = 0.;
53675567043SBarry Smith       sum2 = 0.;
53775567043SBarry Smith       sum3 = 0.;
53875567043SBarry Smith       sum4 = 0.;
53975567043SBarry Smith       sum5 = 0.;
5404c1414c8SBarry Smith       v2   = v1 + n;
5414c1414c8SBarry Smith       v3   = v2 + n;
5424c1414c8SBarry Smith       v4   = v3 + n;
5434c1414c8SBarry Smith       v5   = v4 + n;
5444c1414c8SBarry Smith 
5454c1414c8SBarry Smith       for (n = 0; n<sz-1; n+=2) {
5464c1414c8SBarry Smith         i1    = idx[0];
5474c1414c8SBarry Smith         i2    = idx[1];
5484c1414c8SBarry Smith         idx  += 2;
5494c1414c8SBarry Smith         tmp0  = x[i1];
5504c1414c8SBarry Smith         tmp1  = x[i2];
5514c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
5524c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
5534c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
5544c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
5554c1414c8SBarry Smith         sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2;
5564c1414c8SBarry Smith       }
5574c1414c8SBarry Smith       if (n == sz-1) {
5584c1414c8SBarry Smith         tmp0  = x[*idx++];
5594c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
5604c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
5614c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
5624c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
5634c1414c8SBarry Smith         sum5 += *v5++ * tmp0;
5644c1414c8SBarry Smith       }
5654c1414c8SBarry Smith       y[row++]=sum1;
5664c1414c8SBarry Smith       y[row++]=sum2;
5674c1414c8SBarry Smith       y[row++]=sum3;
5684c1414c8SBarry Smith       y[row++]=sum4;
5694c1414c8SBarry Smith       y[row++]=sum5;
5704c1414c8SBarry Smith       v1      =v5;       /* Since the next block to be processed starts there */
5714c1414c8SBarry Smith       idx    +=4*sz;
5724c1414c8SBarry Smith       break;
5734c1414c8SBarry Smith     default:
574e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported");
5754c1414c8SBarry Smith     }
5764c1414c8SBarry Smith   }
5773649974fSBarry Smith   ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr);
5784c1414c8SBarry Smith   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
579dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr);
5804c1414c8SBarry Smith   PetscFunctionReturn(0);
5814c1414c8SBarry Smith }
5824c1414c8SBarry Smith /* ----------------------------------------------------------- */
5834108e4d5SBarry Smith /* Almost same code as the MatMult_SeqAIJ_Inode() */
5844c1414c8SBarry Smith #undef __FUNCT__
5854108e4d5SBarry Smith #define __FUNCT__ "MatMultAdd_SeqAIJ_Inode"
5864108e4d5SBarry Smith static PetscErrorCode MatMultAdd_SeqAIJ_Inode(Mat A,Vec xx,Vec zz,Vec yy)
5874c1414c8SBarry Smith {
5884c1414c8SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
5894c1414c8SBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5,tmp0,tmp1;
5908758e1faSBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5;
5918758e1faSBarry Smith   const PetscScalar *x;
5928758e1faSBarry Smith   PetscScalar       *y,*z,*zt;
5934c1414c8SBarry Smith   PetscErrorCode    ierr;
5948758e1faSBarry Smith   PetscInt          i1,i2,n,i,row,node_max,nsz,sz;
5958758e1faSBarry Smith   const PetscInt    *idx,*ns,*ii;
5964c1414c8SBarry Smith 
5974c1414c8SBarry Smith   PetscFunctionBegin;
598e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
5994c1414c8SBarry Smith   node_max = a->inode.node_count;
6004c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
6012205254eSKarl Rupp 
6028758e1faSBarry Smith   ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr);
603f95e332cSBarry Smith   ierr = VecGetArrayPair(zz,yy,&z,&y);CHKERRQ(ierr);
6044c1414c8SBarry Smith   zt = z;
6054c1414c8SBarry Smith 
6064c1414c8SBarry Smith   idx = a->j;
6074c1414c8SBarry Smith   v1  = a->a;
6084c1414c8SBarry Smith   ii  = a->i;
6094c1414c8SBarry Smith 
6104c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i) {
6114c1414c8SBarry Smith     nsz = ns[i];
6124c1414c8SBarry Smith     n   = ii[1] - ii[0];
6134c1414c8SBarry Smith     ii += nsz;
6144c1414c8SBarry Smith     sz  = n;                    /* No of non zeros in this row */
6154c1414c8SBarry Smith                                 /* Switch on the size of Node */
6164c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
6174c1414c8SBarry Smith     case 1:
6184c1414c8SBarry Smith       sum1 = *zt++;
6194c1414c8SBarry Smith 
6204c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6214c1414c8SBarry Smith         i1    = idx[0];         /* The instructions are ordered to */
6224c1414c8SBarry Smith         i2    = idx[1];         /* make the compiler's job easy */
6234c1414c8SBarry Smith         idx  += 2;
6244c1414c8SBarry Smith         tmp0  = x[i1];
6254c1414c8SBarry Smith         tmp1  = x[i2];
6264c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6274c1414c8SBarry Smith       }
6284c1414c8SBarry Smith 
6294c1414c8SBarry Smith       if (n   == sz-1) {          /* Take care of the last nonzero  */
6304c1414c8SBarry Smith         tmp0  = x[*idx++];
6314c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6324c1414c8SBarry Smith       }
6334c1414c8SBarry Smith       y[row++]=sum1;
6344c1414c8SBarry Smith       break;
6354c1414c8SBarry Smith     case 2:
6364c1414c8SBarry Smith       sum1 = *zt++;
6374c1414c8SBarry Smith       sum2 = *zt++;
6384c1414c8SBarry Smith       v2   = v1 + n;
6394c1414c8SBarry Smith 
6404c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6414c1414c8SBarry Smith         i1    = idx[0];
6424c1414c8SBarry Smith         i2    = idx[1];
6434c1414c8SBarry Smith         idx  += 2;
6444c1414c8SBarry Smith         tmp0  = x[i1];
6454c1414c8SBarry Smith         tmp1  = x[i2];
6464c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6474c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
6484c1414c8SBarry Smith       }
6494c1414c8SBarry Smith       if (n   == sz-1) {
6504c1414c8SBarry Smith         tmp0  = x[*idx++];
6514c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6524c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
6534c1414c8SBarry Smith       }
6544c1414c8SBarry Smith       y[row++]=sum1;
6554c1414c8SBarry Smith       y[row++]=sum2;
6564c1414c8SBarry Smith       v1      =v2;              /* Since the next block to be processed starts there*/
6574c1414c8SBarry Smith       idx    +=sz;
6584c1414c8SBarry Smith       break;
6594c1414c8SBarry Smith     case 3:
6604c1414c8SBarry Smith       sum1 = *zt++;
6614c1414c8SBarry Smith       sum2 = *zt++;
6624c1414c8SBarry Smith       sum3 = *zt++;
6634c1414c8SBarry Smith       v2   = v1 + n;
6644c1414c8SBarry Smith       v3   = v2 + n;
6654c1414c8SBarry Smith 
6664c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6674c1414c8SBarry Smith         i1    = idx[0];
6684c1414c8SBarry Smith         i2    = idx[1];
6694c1414c8SBarry Smith         idx  += 2;
6704c1414c8SBarry Smith         tmp0  = x[i1];
6714c1414c8SBarry Smith         tmp1  = x[i2];
6724c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6734c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
6744c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
6754c1414c8SBarry Smith       }
6764c1414c8SBarry Smith       if (n == sz-1) {
6774c1414c8SBarry Smith         tmp0  = x[*idx++];
6784c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6794c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
6804c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
6814c1414c8SBarry Smith       }
6824c1414c8SBarry Smith       y[row++]=sum1;
6834c1414c8SBarry Smith       y[row++]=sum2;
6844c1414c8SBarry Smith       y[row++]=sum3;
6854c1414c8SBarry Smith       v1      =v3;              /* Since the next block to be processed starts there*/
6864c1414c8SBarry Smith       idx    +=2*sz;
6874c1414c8SBarry Smith       break;
6884c1414c8SBarry Smith     case 4:
6894c1414c8SBarry Smith       sum1 = *zt++;
6904c1414c8SBarry Smith       sum2 = *zt++;
6914c1414c8SBarry Smith       sum3 = *zt++;
6924c1414c8SBarry Smith       sum4 = *zt++;
6934c1414c8SBarry Smith       v2   = v1 + n;
6944c1414c8SBarry Smith       v3   = v2 + n;
6954c1414c8SBarry Smith       v4   = v3 + n;
6964c1414c8SBarry Smith 
6974c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6984c1414c8SBarry Smith         i1    = idx[0];
6994c1414c8SBarry Smith         i2    = idx[1];
7004c1414c8SBarry Smith         idx  += 2;
7014c1414c8SBarry Smith         tmp0  = x[i1];
7024c1414c8SBarry Smith         tmp1  = x[i2];
7034c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
7044c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
7054c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
7064c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
7074c1414c8SBarry Smith       }
7084c1414c8SBarry Smith       if (n == sz-1) {
7094c1414c8SBarry Smith         tmp0  = x[*idx++];
7104c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
7114c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
7124c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
7134c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
7144c1414c8SBarry Smith       }
7154c1414c8SBarry Smith       y[row++]=sum1;
7164c1414c8SBarry Smith       y[row++]=sum2;
7174c1414c8SBarry Smith       y[row++]=sum3;
7184c1414c8SBarry Smith       y[row++]=sum4;
7194c1414c8SBarry Smith       v1      =v4;              /* Since the next block to be processed starts there*/
7204c1414c8SBarry Smith       idx    +=3*sz;
7214c1414c8SBarry Smith       break;
7224c1414c8SBarry Smith     case 5:
7234c1414c8SBarry Smith       sum1 = *zt++;
7244c1414c8SBarry Smith       sum2 = *zt++;
7254c1414c8SBarry Smith       sum3 = *zt++;
7264c1414c8SBarry Smith       sum4 = *zt++;
7274c1414c8SBarry Smith       sum5 = *zt++;
7284c1414c8SBarry Smith       v2   = v1 + n;
7294c1414c8SBarry Smith       v3   = v2 + n;
7304c1414c8SBarry Smith       v4   = v3 + n;
7314c1414c8SBarry Smith       v5   = v4 + n;
7324c1414c8SBarry Smith 
7334c1414c8SBarry Smith       for (n = 0; n<sz-1; n+=2) {
7344c1414c8SBarry Smith         i1    = idx[0];
7354c1414c8SBarry Smith         i2    = idx[1];
7364c1414c8SBarry Smith         idx  += 2;
7374c1414c8SBarry Smith         tmp0  = x[i1];
7384c1414c8SBarry Smith         tmp1  = x[i2];
7394c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
7404c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
7414c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
7424c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
7434c1414c8SBarry Smith         sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2;
7444c1414c8SBarry Smith       }
7454c1414c8SBarry Smith       if (n   == sz-1) {
7464c1414c8SBarry Smith         tmp0  = x[*idx++];
7474c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
7484c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
7494c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
7504c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
7514c1414c8SBarry Smith         sum5 += *v5++ * tmp0;
7524c1414c8SBarry Smith       }
7534c1414c8SBarry Smith       y[row++]=sum1;
7544c1414c8SBarry Smith       y[row++]=sum2;
7554c1414c8SBarry Smith       y[row++]=sum3;
7564c1414c8SBarry Smith       y[row++]=sum4;
7574c1414c8SBarry Smith       y[row++]=sum5;
7584c1414c8SBarry Smith       v1      =v5;       /* Since the next block to be processed starts there */
7594c1414c8SBarry Smith       idx    +=4*sz;
7604c1414c8SBarry Smith       break;
7614c1414c8SBarry Smith     default:
762e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported");
7634c1414c8SBarry Smith     }
7644c1414c8SBarry Smith   }
7658758e1faSBarry Smith   ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr);
766f95e332cSBarry Smith   ierr = VecRestoreArrayPair(zz,yy,&z,&y);CHKERRQ(ierr);
767dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
7684c1414c8SBarry Smith   PetscFunctionReturn(0);
7694c1414c8SBarry Smith }
7704c1414c8SBarry Smith 
7714c1414c8SBarry Smith /* ----------------------------------------------------------- */
7724c1414c8SBarry Smith #undef __FUNCT__
773ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_Inode_inplace"
774ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_Inode_inplace(Mat A,Vec bb,Vec xx)
7754c1414c8SBarry Smith {
7764c1414c8SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
7774c1414c8SBarry Smith   IS                iscol = a->col,isrow = a->row;
7784c1414c8SBarry Smith   PetscErrorCode    ierr;
7795d0c19d7SBarry Smith   const PetscInt    *r,*c,*rout,*cout;
7808758e1faSBarry Smith   PetscInt          i,j,n = A->rmap->n,nz;
7818758e1faSBarry Smith   PetscInt          node_max,*ns,row,nsz,aii,i0,i1;
7828758e1faSBarry Smith   const PetscInt    *ai = a->i,*a_j = a->j,*vi,*ad,*aj;
783d9fead3dSBarry Smith   PetscScalar       *x,*tmp,*tmps,tmp0,tmp1;
784d9fead3dSBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5;
785dd6ea824SBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa;
786dd6ea824SBarry Smith   const PetscScalar *b;
7874c1414c8SBarry Smith 
7884c1414c8SBarry Smith   PetscFunctionBegin;
789e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
7904c1414c8SBarry Smith   node_max = a->inode.node_count;
7914c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
7924c1414c8SBarry Smith 
7933649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
7944c1414c8SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
7954c1414c8SBarry Smith   tmp  = a->solve_work;
7964c1414c8SBarry Smith 
7974c1414c8SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
7984c1414c8SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
7994c1414c8SBarry Smith 
8004c1414c8SBarry Smith   /* forward solve the lower triangular */
8014c1414c8SBarry Smith   tmps = tmp;
8024c1414c8SBarry Smith   aa   = a_a;
8034c1414c8SBarry Smith   aj   = a_j;
8044c1414c8SBarry Smith   ad   = a->diag;
8054c1414c8SBarry Smith 
8064c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i) {
8074c1414c8SBarry Smith     nsz = ns[i];
8084c1414c8SBarry Smith     aii = ai[row];
8094c1414c8SBarry Smith     v1  = aa + aii;
8104c1414c8SBarry Smith     vi  = aj + aii;
8114c1414c8SBarry Smith     nz  = ad[row]- aii;
81226549573SJed Brown     if (i < node_max-1) {
81326549573SJed Brown       /* Prefetch the block after the current one, the prefetch itself can't cause a memory error,
81426549573SJed Brown       * but our indexing to determine it's size could. */
81550d8bf02SJed Brown       PetscPrefetchBlock(aj+ai[row+nsz],ad[row+nsz]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); /* indices */
81626549573SJed Brown       /* In my tests, it seems to be better to fetch entire rows instead of just the lower-triangular part */
81750d8bf02SJed Brown       PetscPrefetchBlock(aa+ai[row+nsz],ad[row+nsz+ns[i+1]-1]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA);
81826549573SJed 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); */
81926549573SJed Brown     }
8204c1414c8SBarry Smith 
8214c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
8224c1414c8SBarry Smith     case 1:
8234c1414c8SBarry Smith       sum1 = b[*r++];
8244c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
8254c1414c8SBarry Smith         i0    = vi[0];
8264c1414c8SBarry Smith         i1    = vi[1];
8274c1414c8SBarry Smith         vi   +=2;
8284c1414c8SBarry Smith         tmp0  = tmps[i0];
8294c1414c8SBarry Smith         tmp1  = tmps[i1];
8304c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8314c1414c8SBarry Smith       }
8324c1414c8SBarry Smith       if (j == nz-1) {
8334c1414c8SBarry Smith         tmp0  = tmps[*vi++];
8344c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8354c1414c8SBarry Smith       }
8364c1414c8SBarry Smith       tmp[row++]=sum1;
8374c1414c8SBarry Smith       break;
8384c1414c8SBarry Smith     case 2:
8394c1414c8SBarry Smith       sum1 = b[*r++];
8404c1414c8SBarry Smith       sum2 = b[*r++];
8414c1414c8SBarry Smith       v2   = aa + ai[row+1];
8424c1414c8SBarry Smith 
8434c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
8444c1414c8SBarry Smith         i0    = vi[0];
8454c1414c8SBarry Smith         i1    = vi[1];
8464c1414c8SBarry Smith         vi   +=2;
8474c1414c8SBarry Smith         tmp0  = tmps[i0];
8484c1414c8SBarry Smith         tmp1  = tmps[i1];
8494c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8504c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
8514c1414c8SBarry Smith       }
8524c1414c8SBarry Smith       if (j == nz-1) {
8534c1414c8SBarry Smith         tmp0  = tmps[*vi++];
8544c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8554c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
8564c1414c8SBarry Smith       }
8574c1414c8SBarry Smith       sum2     -= *v2++ *sum1;
8584c1414c8SBarry Smith       tmp[row++]=sum1;
8594c1414c8SBarry Smith       tmp[row++]=sum2;
8604c1414c8SBarry Smith       break;
8614c1414c8SBarry Smith     case 3:
8624c1414c8SBarry Smith       sum1 = b[*r++];
8634c1414c8SBarry Smith       sum2 = b[*r++];
8644c1414c8SBarry Smith       sum3 = b[*r++];
8654c1414c8SBarry Smith       v2   = aa + ai[row+1];
8664c1414c8SBarry Smith       v3   = aa + ai[row+2];
8674c1414c8SBarry Smith 
8684c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
8694c1414c8SBarry Smith         i0    = vi[0];
8704c1414c8SBarry Smith         i1    = vi[1];
8714c1414c8SBarry Smith         vi   +=2;
8724c1414c8SBarry Smith         tmp0  = tmps[i0];
8734c1414c8SBarry Smith         tmp1  = tmps[i1];
8744c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8754c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
8764c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
8774c1414c8SBarry Smith       }
8784c1414c8SBarry Smith       if (j == nz-1) {
8794c1414c8SBarry Smith         tmp0  = tmps[*vi++];
8804c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8814c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
8824c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
8834c1414c8SBarry Smith       }
8844c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
8854c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
8864c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
8872205254eSKarl Rupp 
8884c1414c8SBarry Smith       tmp[row++]=sum1;
8894c1414c8SBarry Smith       tmp[row++]=sum2;
8904c1414c8SBarry Smith       tmp[row++]=sum3;
8914c1414c8SBarry Smith       break;
8924c1414c8SBarry Smith 
8934c1414c8SBarry Smith     case 4:
8944c1414c8SBarry Smith       sum1 = b[*r++];
8954c1414c8SBarry Smith       sum2 = b[*r++];
8964c1414c8SBarry Smith       sum3 = b[*r++];
8974c1414c8SBarry Smith       sum4 = b[*r++];
8984c1414c8SBarry Smith       v2   = aa + ai[row+1];
8994c1414c8SBarry Smith       v3   = aa + ai[row+2];
9004c1414c8SBarry Smith       v4   = aa + ai[row+3];
9014c1414c8SBarry Smith 
9024c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
9034c1414c8SBarry Smith         i0    = vi[0];
9044c1414c8SBarry Smith         i1    = vi[1];
9054c1414c8SBarry Smith         vi   +=2;
9064c1414c8SBarry Smith         tmp0  = tmps[i0];
9074c1414c8SBarry Smith         tmp1  = tmps[i1];
9084c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
9094c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
9104c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
9114c1414c8SBarry Smith         sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
9124c1414c8SBarry Smith       }
9134c1414c8SBarry Smith       if (j == nz-1) {
9144c1414c8SBarry Smith         tmp0  = tmps[*vi++];
9154c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
9164c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
9174c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
9184c1414c8SBarry Smith         sum4 -= *v4++ *tmp0;
9194c1414c8SBarry Smith       }
9204c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
9214c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
9224c1414c8SBarry Smith       sum4 -= *v4++ * sum1;
9234c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
9244c1414c8SBarry Smith       sum4 -= *v4++ * sum2;
9254c1414c8SBarry Smith       sum4 -= *v4++ * sum3;
9264c1414c8SBarry Smith 
9274c1414c8SBarry Smith       tmp[row++]=sum1;
9284c1414c8SBarry Smith       tmp[row++]=sum2;
9294c1414c8SBarry Smith       tmp[row++]=sum3;
9304c1414c8SBarry Smith       tmp[row++]=sum4;
9314c1414c8SBarry Smith       break;
9324c1414c8SBarry Smith     case 5:
9334c1414c8SBarry Smith       sum1 = b[*r++];
9344c1414c8SBarry Smith       sum2 = b[*r++];
9354c1414c8SBarry Smith       sum3 = b[*r++];
9364c1414c8SBarry Smith       sum4 = b[*r++];
9374c1414c8SBarry Smith       sum5 = b[*r++];
9384c1414c8SBarry Smith       v2   = aa + ai[row+1];
9394c1414c8SBarry Smith       v3   = aa + ai[row+2];
9404c1414c8SBarry Smith       v4   = aa + ai[row+3];
9414c1414c8SBarry Smith       v5   = aa + ai[row+4];
9424c1414c8SBarry Smith 
9434c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
9444c1414c8SBarry Smith         i0    = vi[0];
9454c1414c8SBarry Smith         i1    = vi[1];
9464c1414c8SBarry Smith         vi   +=2;
9474c1414c8SBarry Smith         tmp0  = tmps[i0];
9484c1414c8SBarry Smith         tmp1  = tmps[i1];
9494c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
9504c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
9514c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
9524c1414c8SBarry Smith         sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
9534c1414c8SBarry Smith         sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
9544c1414c8SBarry Smith       }
9554c1414c8SBarry Smith       if (j == nz-1) {
9564c1414c8SBarry Smith         tmp0  = tmps[*vi++];
9574c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
9584c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
9594c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
9604c1414c8SBarry Smith         sum4 -= *v4++ *tmp0;
9614c1414c8SBarry Smith         sum5 -= *v5++ *tmp0;
9624c1414c8SBarry Smith       }
9634c1414c8SBarry Smith 
9644c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
9654c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
9664c1414c8SBarry Smith       sum4 -= *v4++ * sum1;
9674c1414c8SBarry Smith       sum5 -= *v5++ * sum1;
9684c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
9694c1414c8SBarry Smith       sum4 -= *v4++ * sum2;
9704c1414c8SBarry Smith       sum5 -= *v5++ * sum2;
9714c1414c8SBarry Smith       sum4 -= *v4++ * sum3;
9724c1414c8SBarry Smith       sum5 -= *v5++ * sum3;
9734c1414c8SBarry Smith       sum5 -= *v5++ * sum4;
9744c1414c8SBarry Smith 
9754c1414c8SBarry Smith       tmp[row++]=sum1;
9764c1414c8SBarry Smith       tmp[row++]=sum2;
9774c1414c8SBarry Smith       tmp[row++]=sum3;
9784c1414c8SBarry Smith       tmp[row++]=sum4;
9794c1414c8SBarry Smith       tmp[row++]=sum5;
9804c1414c8SBarry Smith       break;
9814c1414c8SBarry Smith     default:
982e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
9834c1414c8SBarry Smith     }
9844c1414c8SBarry Smith   }
9854c1414c8SBarry Smith   /* backward solve the upper triangular */
9864c1414c8SBarry Smith   for (i=node_max -1,row = n-1; i>=0; i--) {
9874c1414c8SBarry Smith     nsz = ns[i];
9884c1414c8SBarry Smith     aii = ai[row+1] -1;
9894c1414c8SBarry Smith     v1  = aa + aii;
9904c1414c8SBarry Smith     vi  = aj + aii;
9914c1414c8SBarry Smith     nz  = aii- ad[row];
9924c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
9934c1414c8SBarry Smith     case 1:
9944c1414c8SBarry Smith       sum1 = tmp[row];
9954c1414c8SBarry Smith 
9964c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
9974c1414c8SBarry Smith         vi   -=2;
9984c1414c8SBarry Smith         i0    = vi[2];
9994c1414c8SBarry Smith         i1    = vi[1];
10004c1414c8SBarry Smith         tmp0  = tmps[i0];
10014c1414c8SBarry Smith         tmp1  = tmps[i1];
10024c1414c8SBarry Smith         v1   -= 2;
10034c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10044c1414c8SBarry Smith       }
10054c1414c8SBarry Smith       if (j==1) {
10064c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10074c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10084c1414c8SBarry Smith       }
10094c1414c8SBarry Smith       x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10104c1414c8SBarry Smith       break;
10114c1414c8SBarry Smith     case 2:
10124c1414c8SBarry Smith       sum1 = tmp[row];
10134c1414c8SBarry Smith       sum2 = tmp[row -1];
10144c1414c8SBarry Smith       v2   = aa + ai[row]-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         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10244c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10254c1414c8SBarry Smith       }
10264c1414c8SBarry Smith       if (j==1) {
10274c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10284c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10294c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10304c1414c8SBarry Smith       }
10314c1414c8SBarry Smith 
10324c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10334c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10344c1414c8SBarry Smith       x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
10354c1414c8SBarry Smith       break;
10364c1414c8SBarry Smith     case 3:
10374c1414c8SBarry Smith       sum1 = tmp[row];
10384c1414c8SBarry Smith       sum2 = tmp[row -1];
10394c1414c8SBarry Smith       sum3 = tmp[row -2];
10404c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10414c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
10424c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
10434c1414c8SBarry Smith         vi   -=2;
10444c1414c8SBarry Smith         i0    = vi[2];
10454c1414c8SBarry Smith         i1    = vi[1];
10464c1414c8SBarry Smith         tmp0  = tmps[i0];
10474c1414c8SBarry Smith         tmp1  = tmps[i1];
10484c1414c8SBarry Smith         v1   -= 2;
10494c1414c8SBarry Smith         v2   -= 2;
10504c1414c8SBarry Smith         v3   -= 2;
10514c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10524c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10534c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
10544c1414c8SBarry Smith       }
10554c1414c8SBarry Smith       if (j==1) {
10564c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10574c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10584c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10594c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
10604c1414c8SBarry Smith       }
10614c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10624c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10634c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10644c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
10654c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10664c1414c8SBarry Smith       x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
10674c1414c8SBarry Smith 
10684c1414c8SBarry Smith       break;
10694c1414c8SBarry Smith     case 4:
10704c1414c8SBarry Smith       sum1 = tmp[row];
10714c1414c8SBarry Smith       sum2 = tmp[row -1];
10724c1414c8SBarry Smith       sum3 = tmp[row -2];
10734c1414c8SBarry Smith       sum4 = tmp[row -3];
10744c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10754c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
10764c1414c8SBarry Smith       v4   = aa + ai[row -2]-1;
10774c1414c8SBarry Smith 
10784c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
10794c1414c8SBarry Smith         vi   -=2;
10804c1414c8SBarry Smith         i0    = vi[2];
10814c1414c8SBarry Smith         i1    = vi[1];
10824c1414c8SBarry Smith         tmp0  = tmps[i0];
10834c1414c8SBarry Smith         tmp1  = tmps[i1];
10844c1414c8SBarry Smith         v1   -= 2;
10854c1414c8SBarry Smith         v2   -= 2;
10864c1414c8SBarry Smith         v3   -= 2;
10874c1414c8SBarry Smith         v4   -= 2;
10884c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10894c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10904c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
10914c1414c8SBarry Smith         sum4 -= v4[2] * tmp0 + v4[1] * tmp1;
10924c1414c8SBarry Smith       }
10934c1414c8SBarry Smith       if (j==1) {
10944c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10954c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10964c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10974c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
10984c1414c8SBarry Smith         sum4 -= *v4-- * tmp0;
10994c1414c8SBarry Smith       }
11004c1414c8SBarry Smith 
11014c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
11024c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
11034c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11044c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11054c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
11064c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11074c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11084c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
11094c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11104c1414c8SBarry Smith       x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--;
11114c1414c8SBarry Smith       break;
11124c1414c8SBarry Smith     case 5:
11134c1414c8SBarry Smith       sum1 = tmp[row];
11144c1414c8SBarry Smith       sum2 = tmp[row -1];
11154c1414c8SBarry Smith       sum3 = tmp[row -2];
11164c1414c8SBarry Smith       sum4 = tmp[row -3];
11174c1414c8SBarry Smith       sum5 = tmp[row -4];
11184c1414c8SBarry Smith       v2   = aa + ai[row]-1;
11194c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
11204c1414c8SBarry Smith       v4   = aa + ai[row -2]-1;
11214c1414c8SBarry Smith       v5   = aa + ai[row -3]-1;
11224c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
11234c1414c8SBarry Smith         vi   -= 2;
11244c1414c8SBarry Smith         i0    = vi[2];
11254c1414c8SBarry Smith         i1    = vi[1];
11264c1414c8SBarry Smith         tmp0  = tmps[i0];
11274c1414c8SBarry Smith         tmp1  = tmps[i1];
11284c1414c8SBarry Smith         v1   -= 2;
11294c1414c8SBarry Smith         v2   -= 2;
11304c1414c8SBarry Smith         v3   -= 2;
11314c1414c8SBarry Smith         v4   -= 2;
11324c1414c8SBarry Smith         v5   -= 2;
11334c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
11344c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
11354c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
11364c1414c8SBarry Smith         sum4 -= v4[2] * tmp0 + v4[1] * tmp1;
11374c1414c8SBarry Smith         sum5 -= v5[2] * tmp0 + v5[1] * tmp1;
11384c1414c8SBarry Smith       }
11394c1414c8SBarry Smith       if (j==1) {
11404c1414c8SBarry Smith         tmp0  = tmps[*vi--];
11414c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
11424c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
11434c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
11444c1414c8SBarry Smith         sum4 -= *v4-- * tmp0;
11454c1414c8SBarry Smith         sum5 -= *v5-- * tmp0;
11464c1414c8SBarry Smith       }
11474c1414c8SBarry Smith 
11484c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
11494c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
11504c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11514c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11524c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11534c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
11544c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11554c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11564c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11574c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
11584c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11594c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11604c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--;
11614c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11624c1414c8SBarry Smith       x[*c--] = tmp[row] = sum5*a_a[ad[row]]; row--;
11634c1414c8SBarry Smith       break;
11644c1414c8SBarry Smith     default:
1165e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
11664c1414c8SBarry Smith     }
11674c1414c8SBarry Smith   }
11684c1414c8SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11694c1414c8SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
11703649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
11714c1414c8SBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1172dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
11734c1414c8SBarry Smith   PetscFunctionReturn(0);
11744c1414c8SBarry Smith }
11754c1414c8SBarry Smith 
117628f1b45aSHong Zhang #undef __FUNCT__
117728f1b45aSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_Inode"
11789a625307SHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_Inode(Mat B,Mat A,const MatFactorInfo *info)
117928f1b45aSHong Zhang {
118028f1b45aSHong Zhang   Mat             C     =B;
118128f1b45aSHong Zhang   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
118228f1b45aSHong Zhang   IS              isrow = b->row,isicol = b->icol;
118328f1b45aSHong Zhang   PetscErrorCode  ierr;
118428f1b45aSHong Zhang   const PetscInt  *r,*ic,*ics;
118528f1b45aSHong Zhang   const PetscInt  n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag;
118628f1b45aSHong Zhang   PetscInt        i,j,k,nz,nzL,row,*pj;
118728f1b45aSHong Zhang   const PetscInt  *ajtmp,*bjtmp;
11889877982aSShri Abhyankar   MatScalar       *pc,*pc1,*pc2,*pc3,*pc4,mul1,mul2,mul3,mul4,*pv,*rtmp1,*rtmp2,*rtmp3,*rtmp4;
11899877982aSShri Abhyankar   const MatScalar *aa=a->a,*v,*v1,*v2,*v3,*v4;
119028f1b45aSHong Zhang   FactorShiftCtx  sctx;
11914f81c4b7SBarry Smith   const PetscInt  *ddiag;
119228f1b45aSHong Zhang   PetscReal       rs;
119328f1b45aSHong Zhang   MatScalar       d;
11944f81c4b7SBarry Smith   PetscInt        inod,nodesz,node_max,col;
11954f81c4b7SBarry Smith   const PetscInt  *ns;
119607b50cabSHong Zhang   PetscInt        *tmp_vec1,*tmp_vec2,*nsmap;
11970e95ead3SHong Zhang 
119828f1b45aSHong Zhang   PetscFunctionBegin;
119928f1b45aSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
120028f1b45aSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
120128f1b45aSHong Zhang 
1202f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
120328f1b45aSHong Zhang     ddiag          = a->diag;
120428f1b45aSHong Zhang     sctx.shift_top = info->zeropivot;
120528f1b45aSHong Zhang     for (i=0; i<n; i++) {
120628f1b45aSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
120728f1b45aSHong Zhang       d  = (aa)[ddiag[i]];
120828f1b45aSHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
120928f1b45aSHong Zhang       v  = aa+ai[i];
121028f1b45aSHong Zhang       nz = ai[i+1] - ai[i];
12112205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
121228f1b45aSHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
121328f1b45aSHong Zhang     }
121428f1b45aSHong Zhang     sctx.shift_top *= 1.1;
121528f1b45aSHong Zhang     sctx.nshift_max = 5;
121628f1b45aSHong Zhang     sctx.shift_lo   = 0.;
121728f1b45aSHong Zhang     sctx.shift_hi   = 1.;
121828f1b45aSHong Zhang   }
121928f1b45aSHong Zhang 
122028f1b45aSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
122128f1b45aSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
122268785679SHong Zhang 
12231795a4d1SJed Brown   ierr  = PetscCalloc4(n,&rtmp1,n,&rtmp2,n,&rtmp3,n,&rtmp4);CHKERRQ(ierr);
122428f1b45aSHong Zhang   ics   = ic;
122528f1b45aSHong Zhang 
122628f1b45aSHong Zhang   node_max = a->inode.node_count;
122728f1b45aSHong Zhang   ns       = a->inode.size;
1228e7e72b3dSBarry Smith   if (!ns) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix without inode information");
122928f1b45aSHong Zhang 
12309877982aSShri Abhyankar   /* If max inode size > 4, split it into two inodes.*/
123168785679SHong Zhang   /* also map the inode sizes according to the ordering */
1232854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&tmp_vec1);CHKERRQ(ierr);
123368785679SHong Zhang   for (i=0,j=0; i<node_max; ++i,++j) {
1234b1550197SShri Abhyankar     if (ns[i] > 4) {
1235048b5e81SShri Abhyankar       tmp_vec1[j] = 4;
123668785679SHong Zhang       ++j;
123768785679SHong Zhang       tmp_vec1[j] = ns[i] - tmp_vec1[j-1];
123868785679SHong Zhang     } else {
123968785679SHong Zhang       tmp_vec1[j] = ns[i];
124068785679SHong Zhang     }
124168785679SHong Zhang   }
124268785679SHong Zhang   /* Use the correct node_max */
124368785679SHong Zhang   node_max = j;
124468785679SHong Zhang 
124568785679SHong Zhang   /* Now reorder the inode info based on mat re-ordering info */
124668785679SHong Zhang   /* First create a row -> inode_size_array_index map */
1247854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&nsmap);CHKERRQ(ierr);
1248854ce69bSBarry Smith   ierr = PetscMalloc1(node_max+1,&tmp_vec2);CHKERRQ(ierr);
124968785679SHong Zhang   for (i=0,row=0; i<node_max; i++) {
125068785679SHong Zhang     nodesz = tmp_vec1[i];
125168785679SHong Zhang     for (j=0; j<nodesz; j++,row++) {
125268785679SHong Zhang       nsmap[row] = i;
125368785679SHong Zhang     }
125468785679SHong Zhang   }
125568785679SHong Zhang   /* Using nsmap, create a reordered ns structure */
125668785679SHong Zhang   for (i=0,j=0; i< node_max; i++) {
125768785679SHong Zhang     nodesz      = tmp_vec1[nsmap[r[j]]];     /* here the reordered row_no is in r[] */
125868785679SHong Zhang     tmp_vec2[i] = nodesz;
125968785679SHong Zhang     j          += nodesz;
126068785679SHong Zhang   }
126168785679SHong Zhang   ierr = PetscFree(nsmap);CHKERRQ(ierr);
126268785679SHong Zhang   ierr = PetscFree(tmp_vec1);CHKERRQ(ierr);
1263b89f182dSHong Zhang 
126468785679SHong Zhang   /* Now use the correct ns */
126568785679SHong Zhang   ns = tmp_vec2;
126668785679SHong Zhang 
126728f1b45aSHong Zhang   do {
126807b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
126928f1b45aSHong Zhang     /* Now loop over each block-row, and do the factorization */
127028f1b45aSHong Zhang     for (inod=0,i=0; inod<node_max; inod++) { /* i: row index; inod: inode index */
127128f1b45aSHong Zhang       nodesz = ns[inod];
127228f1b45aSHong Zhang 
127328f1b45aSHong Zhang       switch (nodesz) {
127428f1b45aSHong Zhang       case 1:
127568785679SHong Zhang         /*----------*/
1276b89f182dSHong Zhang         /* zero rtmp1 */
127728f1b45aSHong Zhang         /* L part */
127828f1b45aSHong Zhang         nz    = bi[i+1] - bi[i];
127928f1b45aSHong Zhang         bjtmp = bj + bi[i];
1280b89f182dSHong Zhang         for (j=0; j<nz; j++) rtmp1[bjtmp[j]] = 0.0;
128128f1b45aSHong Zhang 
128228f1b45aSHong Zhang         /* U part */
128328f1b45aSHong Zhang         nz    = bdiag[i]-bdiag[i+1];
128428f1b45aSHong Zhang         bjtmp = bj + bdiag[i+1]+1;
1285b89f182dSHong Zhang         for (j=0; j<nz; j++) rtmp1[bjtmp[j]] = 0.0;
128628f1b45aSHong Zhang 
128728f1b45aSHong Zhang         /* load in initial (unfactored row) */
128828f1b45aSHong Zhang         nz    = ai[r[i]+1] - ai[r[i]];
128928f1b45aSHong Zhang         ajtmp = aj + ai[r[i]];
129028f1b45aSHong Zhang         v     = aa + ai[r[i]];
12912205254eSKarl Rupp         for (j=0; j<nz; j++) rtmp1[ics[ajtmp[j]]] = v[j];
12922205254eSKarl Rupp 
129328f1b45aSHong Zhang         /* ZeropivotApply() */
1294b89f182dSHong Zhang         rtmp1[i] += sctx.shift_amount;  /* shift the diagonal of the matrix */
129528f1b45aSHong Zhang 
129628f1b45aSHong Zhang         /* elimination */
129728f1b45aSHong Zhang         bjtmp = bj + bi[i];
129828f1b45aSHong Zhang         row   = *bjtmp++;
129928f1b45aSHong Zhang         nzL   = bi[i+1] - bi[i];
130028f1b45aSHong Zhang         for (k=0; k < nzL; k++) {
1301b89f182dSHong Zhang           pc = rtmp1 + row;
130228f1b45aSHong Zhang           if (*pc != 0.0) {
130328f1b45aSHong Zhang             pv   = b->a + bdiag[row];
1304b89f182dSHong Zhang             mul1 = *pc * (*pv);
1305b89f182dSHong Zhang             *pc  = mul1;
130628f1b45aSHong Zhang             pj   = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
130728f1b45aSHong Zhang             pv   = b->a + bdiag[row+1]+1;
130828f1b45aSHong Zhang             nz   = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
1309b89f182dSHong Zhang             for (j=0; j<nz; j++) rtmp1[pj[j]] -= mul1 * pv[j];
131028b1a77fSLisandro Dalcin             ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
131128f1b45aSHong Zhang           }
131228f1b45aSHong Zhang           row = *bjtmp++;
131328f1b45aSHong Zhang         }
131428f1b45aSHong Zhang 
131528f1b45aSHong Zhang         /* finished row so stick it into b->a */
131628f1b45aSHong Zhang         rs = 0.0;
131728f1b45aSHong Zhang         /* L part */
131828f1b45aSHong Zhang         pv = b->a + bi[i];
131928f1b45aSHong Zhang         pj = b->j + bi[i];
132028f1b45aSHong Zhang         nz = bi[i+1] - bi[i];
132128f1b45aSHong Zhang         for (j=0; j<nz; j++) {
1322b89f182dSHong Zhang           pv[j] = rtmp1[pj[j]]; rs += PetscAbsScalar(pv[j]);
132328f1b45aSHong Zhang         }
132428f1b45aSHong Zhang 
132528f1b45aSHong Zhang         /* U part */
132628f1b45aSHong Zhang         pv = b->a + bdiag[i+1]+1;
132728f1b45aSHong Zhang         pj = b->j + bdiag[i+1]+1;
132828f1b45aSHong Zhang         nz = bdiag[i] - bdiag[i+1]-1;
132928f1b45aSHong Zhang         for (j=0; j<nz; j++) {
1330b89f182dSHong Zhang           pv[j] = rtmp1[pj[j]]; rs += PetscAbsScalar(pv[j]);
133128f1b45aSHong Zhang         }
133228f1b45aSHong Zhang 
1333b89f182dSHong Zhang         /* Check zero pivot */
133428f1b45aSHong Zhang         sctx.rs = rs;
1335b89f182dSHong Zhang         sctx.pv = rtmp1[i];
13364cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
133707b50cabSHong Zhang         if (sctx.newshift) break;
133828f1b45aSHong Zhang 
133928f1b45aSHong Zhang         /* Mark diagonal and invert diagonal for simplier triangular solves */
134028f1b45aSHong Zhang         pv  = b->a + bdiag[i];
1341b89f182dSHong Zhang         *pv = 1.0/sctx.pv; /* sctx.pv = rtmp1[i]+shiftamount if shifttype==MAT_SHIFT_INBLOCKS */
134228f1b45aSHong Zhang         break;
134328f1b45aSHong Zhang 
134428f1b45aSHong Zhang       case 2:
134528f1b45aSHong Zhang         /*----------*/
1346b89f182dSHong Zhang         /* zero rtmp1 and rtmp2 */
134728f1b45aSHong Zhang         /* L part */
134828f1b45aSHong Zhang         nz    = bi[i+1] - bi[i];
134928f1b45aSHong Zhang         bjtmp = bj + bi[i];
135028f1b45aSHong Zhang         for  (j=0; j<nz; j++) {
135168785679SHong Zhang           col        = bjtmp[j];
1352b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0;
135328f1b45aSHong Zhang         }
135428f1b45aSHong Zhang 
135528f1b45aSHong Zhang         /* U part */
135628f1b45aSHong Zhang         nz    = bdiag[i]-bdiag[i+1];
135728f1b45aSHong Zhang         bjtmp = bj + bdiag[i+1]+1;
135828f1b45aSHong Zhang         for  (j=0; j<nz; j++) {
135968785679SHong Zhang           col        = bjtmp[j];
1360b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0;
136128f1b45aSHong Zhang         }
136228f1b45aSHong Zhang 
136328f1b45aSHong Zhang         /* load in initial (unfactored row) */
136428f1b45aSHong Zhang         nz    = ai[r[i]+1] - ai[r[i]];
136528f1b45aSHong Zhang         ajtmp = aj + ai[r[i]];
136628f1b45aSHong Zhang         v1    = aa + ai[r[i]]; v2 = aa + ai[r[i]+1];
136728f1b45aSHong Zhang         for (j=0; j<nz; j++) {
136868785679SHong Zhang           col        = ics[ajtmp[j]];
1369b89f182dSHong Zhang           rtmp1[col] = v1[j]; rtmp2[col] = v2[j];
137028f1b45aSHong Zhang         }
137128f1b45aSHong Zhang         /* ZeropivotApply(): shift the diagonal of the matrix  */
1372b89f182dSHong Zhang         rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount;
137328f1b45aSHong Zhang 
137428f1b45aSHong Zhang         /* elimination */
137528f1b45aSHong Zhang         bjtmp = bj + bi[i];
137628f1b45aSHong Zhang         row   = *bjtmp++; /* pivot row */
137728f1b45aSHong Zhang         nzL   = bi[i+1] - bi[i];
137828f1b45aSHong Zhang         for (k=0; k < nzL; k++) {
1379b89f182dSHong Zhang           pc1 = rtmp1 + row;
1380b89f182dSHong Zhang           pc2 = rtmp2 + row;
138128f1b45aSHong Zhang           if (*pc1 != 0.0 || *pc2 != 0.0) {
138228f1b45aSHong Zhang             pv   = b->a + bdiag[row];
1383b89f182dSHong Zhang             mul1 = *pc1*(*pv); mul2 = *pc2*(*pv);
1384b89f182dSHong Zhang             *pc1 = mul1;       *pc2 = mul2;
138528f1b45aSHong Zhang 
138628f1b45aSHong Zhang             pj = b->j + bdiag[row+1]+1;     /* beginning of U(row,:) */
138728f1b45aSHong Zhang             pv = b->a + bdiag[row+1]+1;
138828f1b45aSHong Zhang             nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
138928f1b45aSHong Zhang             for (j=0; j<nz; j++) {
139068785679SHong Zhang               col         = pj[j];
1391b89f182dSHong Zhang               rtmp1[col] -= mul1 * pv[j];
1392b89f182dSHong Zhang               rtmp2[col] -= mul2 * pv[j];
139328f1b45aSHong Zhang             }
139428b1a77fSLisandro Dalcin             ierr = PetscLogFlops(2+4*nz);CHKERRQ(ierr);
139528f1b45aSHong Zhang           }
139628f1b45aSHong Zhang           row = *bjtmp++;
139728f1b45aSHong Zhang         }
139828f1b45aSHong Zhang 
1399b89f182dSHong Zhang         /* finished row i; check zero pivot, then stick row i into b->a */
140028f1b45aSHong Zhang         rs = 0.0;
140128f1b45aSHong Zhang         /* L part */
1402b89f182dSHong Zhang         pc1 = b->a + bi[i];
140328f1b45aSHong Zhang         pj  = b->j + bi[i];
140428f1b45aSHong Zhang         nz  = bi[i+1] - bi[i];
140528f1b45aSHong Zhang         for (j=0; j<nz; j++) {
140668785679SHong Zhang           col    = pj[j];
14070e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
140828f1b45aSHong Zhang         }
140928f1b45aSHong Zhang         /* U part */
1410b89f182dSHong Zhang         pc1 = b->a + bdiag[i+1]+1;
141128f1b45aSHong Zhang         pj  = b->j + bdiag[i+1]+1;
14120e7a5c2bSHong Zhang         nz  = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */
141328f1b45aSHong Zhang         for (j=0; j<nz; j++) {
141468785679SHong Zhang           col    = pj[j];
14150e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
141628f1b45aSHong Zhang         }
141728f1b45aSHong Zhang 
141828f1b45aSHong Zhang         sctx.rs = rs;
1419b89f182dSHong Zhang         sctx.pv = rtmp1[i];
14204cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
142107b50cabSHong Zhang         if (sctx.newshift) break;
1422b89f182dSHong Zhang         pc1  = b->a + bdiag[i]; /* Mark diagonal */
1423b89f182dSHong Zhang         *pc1 = 1.0/sctx.pv;
1424b89f182dSHong Zhang 
1425b89f182dSHong Zhang         /* Now take care of diagonal 2x2 block. */
1426b89f182dSHong Zhang         pc2 = rtmp2 + i;
1427b89f182dSHong Zhang         if (*pc2 != 0.0) {
1428b89f182dSHong Zhang           mul1 = (*pc2)*(*pc1); /* *pc1=diag[i] is inverted! */
1429b89f182dSHong Zhang           *pc2 = mul1;          /* insert L entry */
1430b89f182dSHong Zhang           pj   = b->j + bdiag[i+1]+1;   /* beginning of U(i,:) */
1431b89f182dSHong Zhang           nz   = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */
1432b89f182dSHong Zhang           for (j=0; j<nz; j++) {
1433b89f182dSHong Zhang             col = pj[j]; rtmp2[col] -= mul1 * rtmp1[col];
143428f1b45aSHong Zhang           }
143528b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
1436b89f182dSHong Zhang         }
1437b89f182dSHong Zhang 
1438b89f182dSHong Zhang         /* finished row i+1; check zero pivot, then stick row i+1 into b->a */
1439b89f182dSHong Zhang         rs = 0.0;
1440b89f182dSHong Zhang         /* L part */
1441b89f182dSHong Zhang         pc2 = b->a + bi[i+1];
1442b89f182dSHong Zhang         pj  = b->j + bi[i+1];
1443b89f182dSHong Zhang         nz  = bi[i+2] - bi[i+1];
1444b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1445b89f182dSHong Zhang           col    = pj[j];
14460e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1447b89f182dSHong Zhang         }
1448b89f182dSHong Zhang         /* U part */
1449b89f182dSHong Zhang         pc2 = b->a + bdiag[i+2]+1;
14500e7a5c2bSHong Zhang         pj  = b->j + bdiag[i+2]+1;
14510e7a5c2bSHong Zhang         nz  = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */
1452b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1453b89f182dSHong Zhang           col    = pj[j];
14540e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1455b89f182dSHong Zhang         }
1456b89f182dSHong Zhang 
145728f1b45aSHong Zhang         sctx.rs = rs;
1458b89f182dSHong Zhang         sctx.pv = rtmp2[i+1];
14594cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i+1);CHKERRQ(ierr);
146007b50cabSHong Zhang         if (sctx.newshift) break;
146128f1b45aSHong Zhang         pc2  = b->a + bdiag[i+1];
1462b89f182dSHong Zhang         *pc2 = 1.0/sctx.pv;
146328f1b45aSHong Zhang         break;
1464b89f182dSHong Zhang 
146568785679SHong Zhang       case 3:
146668785679SHong Zhang         /*----------*/
146768785679SHong Zhang         /* zero rtmp */
146868785679SHong Zhang         /* L part */
146968785679SHong Zhang         nz    = bi[i+1] - bi[i];
147068785679SHong Zhang         bjtmp = bj + bi[i];
147168785679SHong Zhang         for  (j=0; j<nz; j++) {
147268785679SHong Zhang           col        = bjtmp[j];
1473b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;
147468785679SHong Zhang         }
147568785679SHong Zhang 
147668785679SHong Zhang         /* U part */
147768785679SHong Zhang         nz    = bdiag[i]-bdiag[i+1];
147868785679SHong Zhang         bjtmp = bj + bdiag[i+1]+1;
147968785679SHong Zhang         for  (j=0; j<nz; j++) {
148068785679SHong Zhang           col        = bjtmp[j];
1481b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;
148268785679SHong Zhang         }
148368785679SHong Zhang 
148468785679SHong Zhang         /* load in initial (unfactored row) */
148568785679SHong Zhang         nz    = ai[r[i]+1] - ai[r[i]];
148668785679SHong Zhang         ajtmp = aj + ai[r[i]];
148768785679SHong Zhang         v1    = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; v3 = aa + ai[r[i]+2];
148868785679SHong Zhang         for (j=0; j<nz; j++) {
148968785679SHong Zhang           col        = ics[ajtmp[j]];
1490b89f182dSHong Zhang           rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; rtmp3[col] = v3[j];
149168785679SHong Zhang         }
149268785679SHong Zhang         /* ZeropivotApply(): shift the diagonal of the matrix  */
1493b89f182dSHong Zhang         rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; rtmp3[i+2] += sctx.shift_amount;
149468785679SHong Zhang 
149568785679SHong Zhang         /* elimination */
149668785679SHong Zhang         bjtmp = bj + bi[i];
149768785679SHong Zhang         row   = *bjtmp++; /* pivot row */
149868785679SHong Zhang         nzL   = bi[i+1] - bi[i];
149968785679SHong Zhang         for (k=0; k < nzL; k++) {
1500b89f182dSHong Zhang           pc1 = rtmp1 + row;
1501b89f182dSHong Zhang           pc2 = rtmp2 + row;
1502b89f182dSHong Zhang           pc3 = rtmp3 + row;
150368785679SHong Zhang           if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 != 0.0) {
150468785679SHong Zhang             pv   = b->a + bdiag[row];
1505b89f182dSHong Zhang             mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); mul3 = *pc3*(*pv);
1506b89f182dSHong Zhang             *pc1 = mul1; *pc2 = mul2; *pc3 = mul3;
150768785679SHong Zhang 
150868785679SHong Zhang             pj = b->j + bdiag[row+1]+1;     /* beginning of U(row,:) */
150968785679SHong Zhang             pv = b->a + bdiag[row+1]+1;
151068785679SHong Zhang             nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
151168785679SHong Zhang             for (j=0; j<nz; j++) {
151268785679SHong Zhang               col         = pj[j];
1513b89f182dSHong Zhang               rtmp1[col] -= mul1 * pv[j];
1514b89f182dSHong Zhang               rtmp2[col] -= mul2 * pv[j];
1515b89f182dSHong Zhang               rtmp3[col] -= mul3 * pv[j];
151668785679SHong Zhang             }
151728b1a77fSLisandro Dalcin             ierr = PetscLogFlops(3+6*nz);CHKERRQ(ierr);
151868785679SHong Zhang           }
151968785679SHong Zhang           row = *bjtmp++;
152068785679SHong Zhang         }
152168785679SHong Zhang 
1522b89f182dSHong Zhang         /* finished row i; check zero pivot, then stick row i into b->a */
1523b89f182dSHong Zhang         rs = 0.0;
1524b89f182dSHong Zhang         /* L part */
1525b89f182dSHong Zhang         pc1 = b->a + bi[i];
1526b89f182dSHong Zhang         pj  = b->j + bi[i];
1527b89f182dSHong Zhang         nz  = bi[i+1] - bi[i];
1528b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1529b89f182dSHong Zhang           col    = pj[j];
15300e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
1531b89f182dSHong Zhang         }
1532b89f182dSHong Zhang         /* U part */
1533b89f182dSHong Zhang         pc1 = b->a + bdiag[i+1]+1;
1534b89f182dSHong Zhang         pj  = b->j + bdiag[i+1]+1;
15350e7a5c2bSHong Zhang         nz  = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */
1536b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1537b89f182dSHong Zhang           col    = pj[j];
15380e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
1539b89f182dSHong Zhang         }
154068785679SHong Zhang 
1541b89f182dSHong Zhang         sctx.rs = rs;
1542b89f182dSHong Zhang         sctx.pv = rtmp1[i];
15434cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
154407b50cabSHong Zhang         if (sctx.newshift) break;
1545b89f182dSHong Zhang         pc1  = b->a + bdiag[i]; /* Mark diag[i] */
1546b89f182dSHong Zhang         *pc1 = 1.0/sctx.pv;
1547b89f182dSHong Zhang 
1548b89f182dSHong Zhang         /* Now take care of 1st column of diagonal 3x3 block. */
1549b89f182dSHong Zhang         pc2 = rtmp2 + i;
1550b89f182dSHong Zhang         pc3 = rtmp3 + i;
1551b89f182dSHong Zhang         if (*pc2 != 0.0 || *pc3 != 0.0) {
1552b89f182dSHong Zhang           mul2 = (*pc2)*(*pc1); *pc2 = mul2;
1553b89f182dSHong Zhang           mul3 = (*pc3)*(*pc1); *pc3 = mul3;
155468785679SHong Zhang           pj   = b->j + bdiag[i+1]+1;   /* beginning of U(i,:) */
155568785679SHong Zhang           nz   = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */
155668785679SHong Zhang           for (j=0; j<nz; j++) {
155768785679SHong Zhang             col         = pj[j];
1558b89f182dSHong Zhang             rtmp2[col] -= mul2 * rtmp1[col];
1559b89f182dSHong Zhang             rtmp3[col] -= mul3 * rtmp1[col];
156068785679SHong Zhang           }
156128b1a77fSLisandro Dalcin           ierr = PetscLogFlops(2+4*nz);CHKERRQ(ierr);
156268785679SHong Zhang         }
156368785679SHong Zhang 
1564b89f182dSHong Zhang         /* finished row i+1; check zero pivot, then stick row i+1 into b->a */
1565b89f182dSHong Zhang         rs = 0.0;
1566b89f182dSHong Zhang         /* L part */
1567b89f182dSHong Zhang         pc2 = b->a + bi[i+1];
1568b89f182dSHong Zhang         pj  = b->j + bi[i+1];
1569b89f182dSHong Zhang         nz  = bi[i+2] - bi[i+1];
1570b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1571b89f182dSHong Zhang           col    = pj[j];
15720e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1573b89f182dSHong Zhang         }
1574b89f182dSHong Zhang         /* U part */
1575b89f182dSHong Zhang         pc2 = b->a + bdiag[i+2]+1;
15760e7a5c2bSHong Zhang         pj  = b->j + bdiag[i+2]+1;
15770e7a5c2bSHong Zhang         nz  = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */
1578b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1579b89f182dSHong Zhang           col    = pj[j];
15800e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1581b89f182dSHong Zhang         }
1582b89f182dSHong Zhang 
1583b89f182dSHong Zhang         sctx.rs = rs;
1584b89f182dSHong Zhang         sctx.pv = rtmp2[i+1];
15854cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i+1);CHKERRQ(ierr);
158607b50cabSHong Zhang         if (sctx.newshift) break;
1587b89f182dSHong Zhang         pc2  = b->a + bdiag[i+1];
1588b89f182dSHong Zhang         *pc2 = 1.0/sctx.pv; /* Mark diag[i+1] */
1589b89f182dSHong Zhang 
1590b89f182dSHong Zhang         /* Now take care of 2nd column of diagonal 3x3 block. */
1591b89f182dSHong Zhang         pc3 = rtmp3 + i+1;
159268785679SHong Zhang         if (*pc3 != 0.0) {
1593b89f182dSHong Zhang           mul3 = (*pc3)*(*pc2); *pc3 = mul3;
159468785679SHong Zhang           pj   = b->j + bdiag[i+2]+1;     /* beginning of U(i+1,:) */
159568785679SHong Zhang           nz   = bdiag[i+1]-bdiag[i+2]-1; /* num of entries in U(i+1,:) excluding diag */
159668785679SHong Zhang           for (j=0; j<nz; j++) {
159768785679SHong Zhang             col         = pj[j];
1598b89f182dSHong Zhang             rtmp3[col] -= mul3 * rtmp2[col];
159968785679SHong Zhang           }
160028b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
160168785679SHong Zhang         }
160268785679SHong Zhang 
1603b89f182dSHong Zhang         /* finished i+2; check zero pivot, then stick row i+2 into b->a */
160468785679SHong Zhang         rs = 0.0;
160568785679SHong Zhang         /* L part */
1606b89f182dSHong Zhang         pc3 = b->a + bi[i+2];
1607b89f182dSHong Zhang         pj  = b->j + bi[i+2];
1608b89f182dSHong Zhang         nz  = bi[i+3] - bi[i+2];
160968785679SHong Zhang         for (j=0; j<nz; j++) {
161068785679SHong Zhang           col    = pj[j];
16110e7a5c2bSHong Zhang           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
161268785679SHong Zhang         }
161368785679SHong Zhang         /* U part */
1614b89f182dSHong Zhang         pc3 = b->a + bdiag[i+3]+1;
16150e7a5c2bSHong Zhang         pj  = b->j + bdiag[i+3]+1;
16160e7a5c2bSHong Zhang         nz  = bdiag[i+2] - bdiag[i+3] - 1; /* exclude diagonal */
161768785679SHong Zhang         for (j=0; j<nz; j++) {
161868785679SHong Zhang           col    = pj[j];
16190e7a5c2bSHong Zhang           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
162068785679SHong Zhang         }
162168785679SHong Zhang 
162268785679SHong Zhang         sctx.rs = rs;
1623b89f182dSHong Zhang         sctx.pv = rtmp3[i+2];
16244cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i+2);CHKERRQ(ierr);
162507b50cabSHong Zhang         if (sctx.newshift) break;
162668785679SHong Zhang         pc3  = b->a + bdiag[i+2];
1627b89f182dSHong Zhang         *pc3 = 1.0/sctx.pv; /* Mark diag[i+2] */
162868785679SHong Zhang         break;
16299877982aSShri Abhyankar       case 4:
16309877982aSShri Abhyankar         /*----------*/
16319877982aSShri Abhyankar         /* zero rtmp */
16329877982aSShri Abhyankar         /* L part */
16339877982aSShri Abhyankar         nz    = bi[i+1] - bi[i];
16349877982aSShri Abhyankar         bjtmp = bj + bi[i];
16359877982aSShri Abhyankar         for  (j=0; j<nz; j++) {
16369877982aSShri Abhyankar           col        = bjtmp[j];
16379877982aSShri Abhyankar           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;rtmp4[col] = 0.0;
16389877982aSShri Abhyankar         }
16399877982aSShri Abhyankar 
16409877982aSShri Abhyankar         /* U part */
16419877982aSShri Abhyankar         nz    = bdiag[i]-bdiag[i+1];
16429877982aSShri Abhyankar         bjtmp = bj + bdiag[i+1]+1;
16439877982aSShri Abhyankar         for  (j=0; j<nz; j++) {
16449877982aSShri Abhyankar           col        = bjtmp[j];
16459877982aSShri Abhyankar           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0; rtmp4[col] = 0.0;
16469877982aSShri Abhyankar         }
16479877982aSShri Abhyankar 
16489877982aSShri Abhyankar         /* load in initial (unfactored row) */
16499877982aSShri Abhyankar         nz    = ai[r[i]+1] - ai[r[i]];
16509877982aSShri Abhyankar         ajtmp = aj + ai[r[i]];
16519877982aSShri Abhyankar         v1    = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; v3 = aa + ai[r[i]+2]; v4 = aa + ai[r[i]+3];
16529877982aSShri Abhyankar         for (j=0; j<nz; j++) {
16539877982aSShri Abhyankar           col        = ics[ajtmp[j]];
16549877982aSShri Abhyankar           rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; rtmp3[col] = v3[j]; rtmp4[col] = v4[j];
16559877982aSShri Abhyankar         }
16569877982aSShri Abhyankar         /* ZeropivotApply(): shift the diagonal of the matrix  */
16579877982aSShri Abhyankar         rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; rtmp3[i+2] += sctx.shift_amount; rtmp4[i+3] += sctx.shift_amount;
16589877982aSShri Abhyankar 
16599877982aSShri Abhyankar         /* elimination */
16609877982aSShri Abhyankar         bjtmp = bj + bi[i];
16619877982aSShri Abhyankar         row   = *bjtmp++; /* pivot row */
16629877982aSShri Abhyankar         nzL   = bi[i+1] - bi[i];
16639877982aSShri Abhyankar         for (k=0; k < nzL; k++) {
16649877982aSShri Abhyankar           pc1 = rtmp1 + row;
16659877982aSShri Abhyankar           pc2 = rtmp2 + row;
16669877982aSShri Abhyankar           pc3 = rtmp3 + row;
16679877982aSShri Abhyankar           pc4 = rtmp4 + row;
16689877982aSShri Abhyankar           if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 != 0.0 || *pc4 != 0.0) {
16699877982aSShri Abhyankar             pv   = b->a + bdiag[row];
16709877982aSShri Abhyankar             mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); mul3 = *pc3*(*pv); mul4 = *pc4*(*pv);
16719877982aSShri Abhyankar             *pc1 = mul1; *pc2 = mul2; *pc3 = mul3; *pc4 = mul4;
16729877982aSShri Abhyankar 
16739877982aSShri Abhyankar             pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
16749877982aSShri Abhyankar             pv = b->a + bdiag[row+1]+1;
16759877982aSShri Abhyankar             nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
16769877982aSShri Abhyankar             for (j=0; j<nz; j++) {
16779877982aSShri Abhyankar               col         = pj[j];
16789877982aSShri Abhyankar               rtmp1[col] -= mul1 * pv[j];
16799877982aSShri Abhyankar               rtmp2[col] -= mul2 * pv[j];
16809877982aSShri Abhyankar               rtmp3[col] -= mul3 * pv[j];
16819877982aSShri Abhyankar               rtmp4[col] -= mul4 * pv[j];
16829877982aSShri Abhyankar             }
168328b1a77fSLisandro Dalcin             ierr = PetscLogFlops(4+8*nz);CHKERRQ(ierr);
16849877982aSShri Abhyankar           }
16859877982aSShri Abhyankar           row = *bjtmp++;
16869877982aSShri Abhyankar         }
16879877982aSShri Abhyankar 
16889877982aSShri Abhyankar         /* finished row i; check zero pivot, then stick row i into b->a */
16899877982aSShri Abhyankar         rs = 0.0;
16909877982aSShri Abhyankar         /* L part */
16919877982aSShri Abhyankar         pc1 = b->a + bi[i];
16929877982aSShri Abhyankar         pj  = b->j + bi[i];
16939877982aSShri Abhyankar         nz  = bi[i+1] - bi[i];
16949877982aSShri Abhyankar         for (j=0; j<nz; j++) {
16959877982aSShri Abhyankar           col    = pj[j];
16969877982aSShri Abhyankar           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
16979877982aSShri Abhyankar         }
16989877982aSShri Abhyankar         /* U part */
16999877982aSShri Abhyankar         pc1 = b->a + bdiag[i+1]+1;
17009877982aSShri Abhyankar         pj  = b->j + bdiag[i+1]+1;
17019877982aSShri Abhyankar         nz  = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */
17029877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17039877982aSShri Abhyankar           col    = pj[j];
17049877982aSShri Abhyankar           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
17059877982aSShri Abhyankar         }
17069877982aSShri Abhyankar 
17079877982aSShri Abhyankar         sctx.rs = rs;
17089877982aSShri Abhyankar         sctx.pv = rtmp1[i];
17094cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr);
171007b50cabSHong Zhang         if (sctx.newshift) break;
17119877982aSShri Abhyankar         pc1  = b->a + bdiag[i]; /* Mark diag[i] */
17129877982aSShri Abhyankar         *pc1 = 1.0/sctx.pv;
17139877982aSShri Abhyankar 
17149877982aSShri Abhyankar         /* Now take care of 1st column of diagonal 4x4 block. */
17159877982aSShri Abhyankar         pc2 = rtmp2 + i;
17169877982aSShri Abhyankar         pc3 = rtmp3 + i;
17179877982aSShri Abhyankar         pc4 = rtmp4 + i;
17189877982aSShri Abhyankar         if (*pc2 != 0.0 || *pc3 != 0.0 || *pc4 != 0.0) {
17199877982aSShri Abhyankar           mul2 = (*pc2)*(*pc1); *pc2 = mul2;
17209877982aSShri Abhyankar           mul3 = (*pc3)*(*pc1); *pc3 = mul3;
17219877982aSShri Abhyankar           mul4 = (*pc4)*(*pc1); *pc4 = mul4;
17229877982aSShri Abhyankar           pj   = b->j + bdiag[i+1]+1;   /* beginning of U(i,:) */
17239877982aSShri Abhyankar           nz   = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */
17249877982aSShri Abhyankar           for (j=0; j<nz; j++) {
17259877982aSShri Abhyankar             col         = pj[j];
17269877982aSShri Abhyankar             rtmp2[col] -= mul2 * rtmp1[col];
17279877982aSShri Abhyankar             rtmp3[col] -= mul3 * rtmp1[col];
17289877982aSShri Abhyankar             rtmp4[col] -= mul4 * rtmp1[col];
17299877982aSShri Abhyankar           }
173028b1a77fSLisandro Dalcin           ierr = PetscLogFlops(3+6*nz);CHKERRQ(ierr);
17319877982aSShri Abhyankar         }
17329877982aSShri Abhyankar 
17339877982aSShri Abhyankar         /* finished row i+1; check zero pivot, then stick row i+1 into b->a */
17349877982aSShri Abhyankar         rs = 0.0;
17359877982aSShri Abhyankar         /* L part */
17369877982aSShri Abhyankar         pc2 = b->a + bi[i+1];
17379877982aSShri Abhyankar         pj  = b->j + bi[i+1];
17389877982aSShri Abhyankar         nz  = bi[i+2] - bi[i+1];
17399877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17409877982aSShri Abhyankar           col    = pj[j];
17419877982aSShri Abhyankar           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
17429877982aSShri Abhyankar         }
17439877982aSShri Abhyankar         /* U part */
17449877982aSShri Abhyankar         pc2 = b->a + bdiag[i+2]+1;
17459877982aSShri Abhyankar         pj  = b->j + bdiag[i+2]+1;
17469877982aSShri Abhyankar         nz  = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */
17479877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17489877982aSShri Abhyankar           col    = pj[j];
17499877982aSShri Abhyankar           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
17509877982aSShri Abhyankar         }
17519877982aSShri Abhyankar 
17529877982aSShri Abhyankar         sctx.rs = rs;
17539877982aSShri Abhyankar         sctx.pv = rtmp2[i+1];
17544cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i+1);CHKERRQ(ierr);
175507b50cabSHong Zhang         if (sctx.newshift) break;
17569877982aSShri Abhyankar         pc2  = b->a + bdiag[i+1];
17579877982aSShri Abhyankar         *pc2 = 1.0/sctx.pv; /* Mark diag[i+1] */
17589877982aSShri Abhyankar 
17599877982aSShri Abhyankar         /* Now take care of 2nd column of diagonal 4x4 block. */
17609877982aSShri Abhyankar         pc3 = rtmp3 + i+1;
17619877982aSShri Abhyankar         pc4 = rtmp4 + i+1;
17629877982aSShri Abhyankar         if (*pc3 != 0.0 || *pc4 != 0.0) {
17639877982aSShri Abhyankar           mul3 = (*pc3)*(*pc2); *pc3 = mul3;
17649877982aSShri Abhyankar           mul4 = (*pc4)*(*pc2); *pc4 = mul4;
17659877982aSShri Abhyankar           pj   = b->j + bdiag[i+2]+1;     /* beginning of U(i+1,:) */
17669877982aSShri Abhyankar           nz   = bdiag[i+1]-bdiag[i+2]-1; /* num of entries in U(i+1,:) excluding diag */
17679877982aSShri Abhyankar           for (j=0; j<nz; j++) {
17689877982aSShri Abhyankar             col         = pj[j];
17699877982aSShri Abhyankar             rtmp3[col] -= mul3 * rtmp2[col];
17709877982aSShri Abhyankar             rtmp4[col] -= mul4 * rtmp2[col];
17719877982aSShri Abhyankar           }
17729877982aSShri Abhyankar           ierr = PetscLogFlops(4*nz);CHKERRQ(ierr);
17739877982aSShri Abhyankar         }
17749877982aSShri Abhyankar 
17759877982aSShri Abhyankar         /* finished i+2; check zero pivot, then stick row i+2 into b->a */
17769877982aSShri Abhyankar         rs = 0.0;
17779877982aSShri Abhyankar         /* L part */
17789877982aSShri Abhyankar         pc3 = b->a + bi[i+2];
17799877982aSShri Abhyankar         pj  = b->j + bi[i+2];
17809877982aSShri Abhyankar         nz  = bi[i+3] - bi[i+2];
17819877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17829877982aSShri Abhyankar           col    = pj[j];
17839877982aSShri Abhyankar           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
17849877982aSShri Abhyankar         }
17859877982aSShri Abhyankar         /* U part */
17869877982aSShri Abhyankar         pc3 = b->a + bdiag[i+3]+1;
17879877982aSShri Abhyankar         pj  = b->j + bdiag[i+3]+1;
17889877982aSShri Abhyankar         nz  = bdiag[i+2] - bdiag[i+3] - 1; /* exclude diagonal */
17899877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17909877982aSShri Abhyankar           col    = pj[j];
17919877982aSShri Abhyankar           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
17929877982aSShri Abhyankar         }
17939877982aSShri Abhyankar 
17949877982aSShri Abhyankar         sctx.rs = rs;
17959877982aSShri Abhyankar         sctx.pv = rtmp3[i+2];
17964cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i+2);CHKERRQ(ierr);
179707b50cabSHong Zhang         if (sctx.newshift) break;
17989877982aSShri Abhyankar         pc3  = b->a + bdiag[i+2];
17999877982aSShri Abhyankar         *pc3 = 1.0/sctx.pv; /* Mark diag[i+2] */
18009877982aSShri Abhyankar 
18019877982aSShri Abhyankar         /* Now take care of 3rd column of diagonal 4x4 block. */
18029877982aSShri Abhyankar         pc4 = rtmp4 + i+2;
18039877982aSShri Abhyankar         if (*pc4 != 0.0) {
18049877982aSShri Abhyankar           mul4 = (*pc4)*(*pc3); *pc4 = mul4;
18059877982aSShri Abhyankar           pj   = b->j + bdiag[i+3]+1;     /* beginning of U(i+2,:) */
18069877982aSShri Abhyankar           nz   = bdiag[i+2]-bdiag[i+3]-1; /* num of entries in U(i+2,:) excluding diag */
18079877982aSShri Abhyankar           for (j=0; j<nz; j++) {
18089877982aSShri Abhyankar             col         = pj[j];
18099877982aSShri Abhyankar             rtmp4[col] -= mul4 * rtmp3[col];
18109877982aSShri Abhyankar           }
181128b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
18129877982aSShri Abhyankar         }
18139877982aSShri Abhyankar 
18149877982aSShri Abhyankar         /* finished i+3; check zero pivot, then stick row i+3 into b->a */
18159877982aSShri Abhyankar         rs = 0.0;
18169877982aSShri Abhyankar         /* L part */
18179877982aSShri Abhyankar         pc4 = b->a + bi[i+3];
18189877982aSShri Abhyankar         pj  = b->j + bi[i+3];
18199877982aSShri Abhyankar         nz  = bi[i+4] - bi[i+3];
18209877982aSShri Abhyankar         for (j=0; j<nz; j++) {
18219877982aSShri Abhyankar           col    = pj[j];
18229877982aSShri Abhyankar           pc4[j] = rtmp4[col]; rs += PetscAbsScalar(pc4[j]);
18239877982aSShri Abhyankar         }
18249877982aSShri Abhyankar         /* U part */
18259877982aSShri Abhyankar         pc4 = b->a + bdiag[i+4]+1;
18269877982aSShri Abhyankar         pj  = b->j + bdiag[i+4]+1;
18279877982aSShri Abhyankar         nz  = bdiag[i+3] - bdiag[i+4] - 1; /* exclude diagonal */
18289877982aSShri Abhyankar         for (j=0; j<nz; j++) {
18299877982aSShri Abhyankar           col    = pj[j];
18309877982aSShri Abhyankar           pc4[j] = rtmp4[col]; rs += PetscAbsScalar(pc4[j]);
18319877982aSShri Abhyankar         }
18329877982aSShri Abhyankar 
18339877982aSShri Abhyankar         sctx.rs = rs;
18349877982aSShri Abhyankar         sctx.pv = rtmp4[i+3];
18354cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,i+3);CHKERRQ(ierr);
183607b50cabSHong Zhang         if (sctx.newshift) break;
18379877982aSShri Abhyankar         pc4  = b->a + bdiag[i+3];
18389877982aSShri Abhyankar         *pc4 = 1.0/sctx.pv; /* Mark diag[i+3] */
18399877982aSShri Abhyankar         break;
184068785679SHong Zhang 
184128f1b45aSHong Zhang       default:
1842e32f2f54SBarry Smith         SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Node size not yet supported \n");
184328f1b45aSHong Zhang       }
1844c2b86aeeSHong Zhang       if (sctx.newshift) break; /* break for (inod=0,i=0; inod<node_max; inod++) */
184528f1b45aSHong Zhang       i += nodesz;                 /* Update the row */
184668785679SHong Zhang     }
184728f1b45aSHong Zhang 
184828f1b45aSHong Zhang     /* MatPivotRefine() */
184907b50cabSHong Zhang     if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
185028f1b45aSHong Zhang       /*
185128f1b45aSHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
185228f1b45aSHong Zhang        * then try lower shift
185328f1b45aSHong Zhang        */
185428f1b45aSHong Zhang       sctx.shift_hi       = sctx.shift_fraction;
185528f1b45aSHong Zhang       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
185628f1b45aSHong Zhang       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
185707b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
185828f1b45aSHong Zhang       sctx.nshift++;
185928f1b45aSHong Zhang     }
186007b50cabSHong Zhang   } while (sctx.newshift);
186128f1b45aSHong Zhang 
1862eb9baa12SBarry Smith   ierr = PetscFree4(rtmp1,rtmp2,rtmp3,rtmp4);CHKERRQ(ierr);
186368785679SHong Zhang   ierr = PetscFree(tmp_vec2);CHKERRQ(ierr);
186428f1b45aSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
186528f1b45aSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
186628f1b45aSHong Zhang 
1867abb87a52SBarry Smith   if (b->inode.size) {
1868abb87a52SBarry Smith     C->ops->solve           = MatSolve_SeqAIJ_Inode;
1869abb87a52SBarry Smith   } else {
1870d3ac4fa3SBarry Smith     C->ops->solve           = MatSolve_SeqAIJ;
1871abb87a52SBarry Smith   }
187228f1b45aSHong Zhang   C->ops->solveadd          = MatSolveAdd_SeqAIJ;
187328f1b45aSHong Zhang   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ;
187428f1b45aSHong Zhang   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ;
187528f1b45aSHong Zhang   C->ops->matsolve          = MatMatSolve_SeqAIJ;
187628f1b45aSHong Zhang   C->assembled              = PETSC_TRUE;
187728f1b45aSHong Zhang   C->preallocated           = PETSC_TRUE;
18782205254eSKarl Rupp 
187928f1b45aSHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
188028f1b45aSHong Zhang 
188128f1b45aSHong Zhang   /* MatShiftView(A,info,&sctx) */
188228f1b45aSHong Zhang   if (sctx.nshift) {
1883f4db908eSBarry Smith     if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) {
188457622a8eSBarry 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);
1885f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
188657622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
1887f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) {
188857622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %g\n",sctx.nshift,(double)info->shiftamount);CHKERRQ(ierr);
188928f1b45aSHong Zhang     }
189028f1b45aSHong Zhang   }
189128f1b45aSHong Zhang   PetscFunctionReturn(0);
189228f1b45aSHong Zhang }
1893628f99d7SShri Abhyankar 
1894628f99d7SShri Abhyankar #undef __FUNCT__
1895628f99d7SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_Inode_inplace"
18969a625307SHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_Inode_inplace(Mat B,Mat A,const MatFactorInfo *info)
1897628f99d7SShri Abhyankar {
1898628f99d7SShri Abhyankar   Mat             C     = B;
1899628f99d7SShri Abhyankar   Mat_SeqAIJ      *a    = (Mat_SeqAIJ*)A->data,*b = (Mat_SeqAIJ*)C->data;
1900628f99d7SShri Abhyankar   IS              iscol = b->col,isrow = b->row,isicol = b->icol;
1901628f99d7SShri Abhyankar   PetscErrorCode  ierr;
1902628f99d7SShri Abhyankar   const PetscInt  *r,*ic,*c,*ics;
1903628f99d7SShri Abhyankar   PetscInt        n   = A->rmap->n,*bi = b->i;
1904628f99d7SShri Abhyankar   PetscInt        *bj = b->j,*nbj=b->j +1,*ajtmp,*bjtmp,nz,nz_tmp,row,prow;
19058758e1faSBarry Smith   PetscInt        i,j,idx,*bd = b->diag,node_max,nodesz;
19068758e1faSBarry Smith   PetscInt        *ai = a->i,*aj = a->j;
1907628f99d7SShri Abhyankar   PetscInt        *ns,*tmp_vec1,*tmp_vec2,*nsmap,*pj;
1908628f99d7SShri Abhyankar   PetscScalar     mul1,mul2,mul3,tmp;
1909628f99d7SShri Abhyankar   MatScalar       *pc1,*pc2,*pc3,*ba = b->a,*pv,*rtmp11,*rtmp22,*rtmp33;
1910628f99d7SShri Abhyankar   const MatScalar *v1,*v2,*v3,*aa = a->a,*rtmp1;
1911628f99d7SShri Abhyankar   PetscReal       rs=0.0;
1912628f99d7SShri Abhyankar   FactorShiftCtx  sctx;
1913628f99d7SShri Abhyankar 
1914628f99d7SShri Abhyankar   PetscFunctionBegin;
1915628f99d7SShri Abhyankar   sctx.shift_top      = 0;
1916628f99d7SShri Abhyankar   sctx.nshift_max     = 0;
1917628f99d7SShri Abhyankar   sctx.shift_lo       = 0;
1918628f99d7SShri Abhyankar   sctx.shift_hi       = 0;
1919628f99d7SShri Abhyankar   sctx.shift_fraction = 0;
1920628f99d7SShri Abhyankar 
1921628f99d7SShri Abhyankar   /* if both shift schemes are chosen by user, only use info->shiftpd */
1922f4db908eSBarry Smith   if (info->shifttype==(PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
1923628f99d7SShri Abhyankar     sctx.shift_top = 0;
1924628f99d7SShri Abhyankar     for (i=0; i<n; i++) {
1925628f99d7SShri Abhyankar       /* calculate rs = sum(|aij|)-RealPart(aii), amt of shift needed for this row */
1926628f99d7SShri Abhyankar       rs    = 0.0;
1927628f99d7SShri Abhyankar       ajtmp = aj + ai[i];
1928628f99d7SShri Abhyankar       rtmp1 = aa + ai[i];
1929628f99d7SShri Abhyankar       nz    = ai[i+1] - ai[i];
1930628f99d7SShri Abhyankar       for (j=0; j<nz; j++) {
1931628f99d7SShri Abhyankar         if (*ajtmp != i) {
1932628f99d7SShri Abhyankar           rs += PetscAbsScalar(*rtmp1++);
1933628f99d7SShri Abhyankar         } else {
1934628f99d7SShri Abhyankar           rs -= PetscRealPart(*rtmp1++);
1935628f99d7SShri Abhyankar         }
1936628f99d7SShri Abhyankar         ajtmp++;
1937628f99d7SShri Abhyankar       }
1938628f99d7SShri Abhyankar       if (rs>sctx.shift_top) sctx.shift_top = rs;
1939628f99d7SShri Abhyankar     }
1940628f99d7SShri Abhyankar     if (sctx.shift_top == 0.0) sctx.shift_top += 1.e-12;
1941628f99d7SShri Abhyankar     sctx.shift_top *= 1.1;
1942628f99d7SShri Abhyankar     sctx.nshift_max = 5;
1943628f99d7SShri Abhyankar     sctx.shift_lo   = 0.;
1944628f99d7SShri Abhyankar     sctx.shift_hi   = 1.;
1945628f99d7SShri Abhyankar   }
1946628f99d7SShri Abhyankar   sctx.shift_amount = 0;
1947628f99d7SShri Abhyankar   sctx.nshift       = 0;
1948628f99d7SShri Abhyankar 
1949628f99d7SShri Abhyankar   ierr   = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1950628f99d7SShri Abhyankar   ierr   = ISGetIndices(iscol,&c);CHKERRQ(ierr);
1951628f99d7SShri Abhyankar   ierr   = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
19521795a4d1SJed Brown   ierr   = PetscCalloc3(n,&rtmp11,n,&rtmp22,n,&rtmp33);CHKERRQ(ierr);
1953628f99d7SShri Abhyankar   ics    = ic;
1954628f99d7SShri Abhyankar 
1955628f99d7SShri Abhyankar   node_max = a->inode.node_count;
1956628f99d7SShri Abhyankar   ns       = a->inode.size;
1957e7e72b3dSBarry Smith   if (!ns) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix without inode information");
1958628f99d7SShri Abhyankar 
1959628f99d7SShri Abhyankar   /* If max inode size > 3, split it into two inodes.*/
1960628f99d7SShri Abhyankar   /* also map the inode sizes according to the ordering */
1961854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&tmp_vec1);CHKERRQ(ierr);
1962628f99d7SShri Abhyankar   for (i=0,j=0; i<node_max; ++i,++j) {
1963628f99d7SShri Abhyankar     if (ns[i]>3) {
1964628f99d7SShri Abhyankar       tmp_vec1[j] = ns[i]/2; /* Assuming ns[i] < =5  */
1965628f99d7SShri Abhyankar       ++j;
1966628f99d7SShri Abhyankar       tmp_vec1[j] = ns[i] - tmp_vec1[j-1];
1967628f99d7SShri Abhyankar     } else {
1968628f99d7SShri Abhyankar       tmp_vec1[j] = ns[i];
1969628f99d7SShri Abhyankar     }
1970628f99d7SShri Abhyankar   }
1971628f99d7SShri Abhyankar   /* Use the correct node_max */
1972628f99d7SShri Abhyankar   node_max = j;
1973628f99d7SShri Abhyankar 
1974628f99d7SShri Abhyankar   /* Now reorder the inode info based on mat re-ordering info */
1975628f99d7SShri Abhyankar   /* First create a row -> inode_size_array_index map */
1976854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&nsmap);CHKERRQ(ierr);
1977854ce69bSBarry Smith   ierr = PetscMalloc1(node_max+1,&tmp_vec2);CHKERRQ(ierr);
1978628f99d7SShri Abhyankar   for (i=0,row=0; i<node_max; i++) {
1979628f99d7SShri Abhyankar     nodesz = tmp_vec1[i];
1980628f99d7SShri Abhyankar     for (j=0; j<nodesz; j++,row++) {
1981628f99d7SShri Abhyankar       nsmap[row] = i;
1982628f99d7SShri Abhyankar     }
1983628f99d7SShri Abhyankar   }
1984628f99d7SShri Abhyankar   /* Using nsmap, create a reordered ns structure */
1985628f99d7SShri Abhyankar   for (i=0,j=0; i< node_max; i++) {
1986628f99d7SShri Abhyankar     nodesz      = tmp_vec1[nsmap[r[j]]];     /* here the reordered row_no is in r[] */
1987628f99d7SShri Abhyankar     tmp_vec2[i] = nodesz;
1988628f99d7SShri Abhyankar     j          += nodesz;
1989628f99d7SShri Abhyankar   }
1990628f99d7SShri Abhyankar   ierr = PetscFree(nsmap);CHKERRQ(ierr);
1991628f99d7SShri Abhyankar   ierr = PetscFree(tmp_vec1);CHKERRQ(ierr);
1992628f99d7SShri Abhyankar   /* Now use the correct ns */
1993628f99d7SShri Abhyankar   ns = tmp_vec2;
1994628f99d7SShri Abhyankar 
1995628f99d7SShri Abhyankar   do {
199607b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
1997628f99d7SShri Abhyankar     /* Now loop over each block-row, and do the factorization */
1998628f99d7SShri Abhyankar     for (i=0,row=0; i<node_max; i++) {
1999628f99d7SShri Abhyankar       nodesz = ns[i];
2000628f99d7SShri Abhyankar       nz     = bi[row+1] - bi[row];
2001628f99d7SShri Abhyankar       bjtmp  = bj + bi[row];
2002628f99d7SShri Abhyankar 
2003628f99d7SShri Abhyankar       switch (nodesz) {
2004628f99d7SShri Abhyankar       case 1:
2005628f99d7SShri Abhyankar         for  (j=0; j<nz; j++) {
2006628f99d7SShri Abhyankar           idx         = bjtmp[j];
2007628f99d7SShri Abhyankar           rtmp11[idx] = 0.0;
2008628f99d7SShri Abhyankar         }
2009628f99d7SShri Abhyankar 
2010628f99d7SShri Abhyankar         /* load in initial (unfactored row) */
2011628f99d7SShri Abhyankar         idx    = r[row];
2012628f99d7SShri Abhyankar         nz_tmp = ai[idx+1] - ai[idx];
2013628f99d7SShri Abhyankar         ajtmp  = aj + ai[idx];
2014628f99d7SShri Abhyankar         v1     = aa + ai[idx];
2015628f99d7SShri Abhyankar 
2016628f99d7SShri Abhyankar         for (j=0; j<nz_tmp; j++) {
2017628f99d7SShri Abhyankar           idx         = ics[ajtmp[j]];
2018628f99d7SShri Abhyankar           rtmp11[idx] = v1[j];
2019628f99d7SShri Abhyankar         }
2020628f99d7SShri Abhyankar         rtmp11[ics[r[row]]] += sctx.shift_amount;
2021628f99d7SShri Abhyankar 
2022628f99d7SShri Abhyankar         prow = *bjtmp++;
2023628f99d7SShri Abhyankar         while (prow < row) {
2024628f99d7SShri Abhyankar           pc1 = rtmp11 + prow;
2025628f99d7SShri Abhyankar           if (*pc1 != 0.0) {
2026628f99d7SShri Abhyankar             pv     = ba + bd[prow];
2027628f99d7SShri Abhyankar             pj     = nbj + bd[prow];
2028628f99d7SShri Abhyankar             mul1   = *pc1 * *pv++;
2029628f99d7SShri Abhyankar             *pc1   = mul1;
2030628f99d7SShri Abhyankar             nz_tmp = bi[prow+1] - bd[prow] - 1;
203128b1a77fSLisandro Dalcin             ierr   = PetscLogFlops(1+2*nz_tmp);CHKERRQ(ierr);
2032628f99d7SShri Abhyankar             for (j=0; j<nz_tmp; j++) {
2033628f99d7SShri Abhyankar               tmp          = pv[j];
2034628f99d7SShri Abhyankar               idx          = pj[j];
2035628f99d7SShri Abhyankar               rtmp11[idx] -= mul1 * tmp;
2036628f99d7SShri Abhyankar             }
2037628f99d7SShri Abhyankar           }
2038628f99d7SShri Abhyankar           prow = *bjtmp++;
2039628f99d7SShri Abhyankar         }
2040628f99d7SShri Abhyankar         pj  = bj + bi[row];
2041628f99d7SShri Abhyankar         pc1 = ba + bi[row];
2042628f99d7SShri Abhyankar 
2043628f99d7SShri Abhyankar         sctx.pv     = rtmp11[row];
2044628f99d7SShri Abhyankar         rtmp11[row] = 1.0/rtmp11[row]; /* invert diag */
2045628f99d7SShri Abhyankar         rs          = 0.0;
2046628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2047628f99d7SShri Abhyankar           idx    = pj[j];
2048628f99d7SShri Abhyankar           pc1[j] = rtmp11[idx]; /* rtmp11 -> ba */
2049628f99d7SShri Abhyankar           if (idx != row) rs += PetscAbsScalar(pc1[j]);
2050628f99d7SShri Abhyankar         }
2051628f99d7SShri Abhyankar         sctx.rs = rs;
20524cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,row);CHKERRQ(ierr);
205307b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2054628f99d7SShri Abhyankar         break;
2055628f99d7SShri Abhyankar 
2056628f99d7SShri Abhyankar       case 2:
2057628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2058628f99d7SShri Abhyankar           idx         = bjtmp[j];
2059628f99d7SShri Abhyankar           rtmp11[idx] = 0.0;
2060628f99d7SShri Abhyankar           rtmp22[idx] = 0.0;
2061628f99d7SShri Abhyankar         }
2062628f99d7SShri Abhyankar 
2063628f99d7SShri Abhyankar         /* load in initial (unfactored row) */
2064628f99d7SShri Abhyankar         idx    = r[row];
2065628f99d7SShri Abhyankar         nz_tmp = ai[idx+1] - ai[idx];
2066628f99d7SShri Abhyankar         ajtmp  = aj + ai[idx];
2067628f99d7SShri Abhyankar         v1     = aa + ai[idx];
2068628f99d7SShri Abhyankar         v2     = aa + ai[idx+1];
2069628f99d7SShri Abhyankar         for (j=0; j<nz_tmp; j++) {
2070628f99d7SShri Abhyankar           idx         = ics[ajtmp[j]];
2071628f99d7SShri Abhyankar           rtmp11[idx] = v1[j];
2072628f99d7SShri Abhyankar           rtmp22[idx] = v2[j];
2073628f99d7SShri Abhyankar         }
2074628f99d7SShri Abhyankar         rtmp11[ics[r[row]]]   += sctx.shift_amount;
2075628f99d7SShri Abhyankar         rtmp22[ics[r[row+1]]] += sctx.shift_amount;
2076628f99d7SShri Abhyankar 
2077628f99d7SShri Abhyankar         prow = *bjtmp++;
2078628f99d7SShri Abhyankar         while (prow < row) {
2079628f99d7SShri Abhyankar           pc1 = rtmp11 + prow;
2080628f99d7SShri Abhyankar           pc2 = rtmp22 + prow;
2081628f99d7SShri Abhyankar           if (*pc1 != 0.0 || *pc2 != 0.0) {
2082628f99d7SShri Abhyankar             pv   = ba + bd[prow];
2083628f99d7SShri Abhyankar             pj   = nbj + bd[prow];
2084628f99d7SShri Abhyankar             mul1 = *pc1 * *pv;
2085628f99d7SShri Abhyankar             mul2 = *pc2 * *pv;
2086628f99d7SShri Abhyankar             ++pv;
2087628f99d7SShri Abhyankar             *pc1 = mul1;
2088628f99d7SShri Abhyankar             *pc2 = mul2;
2089628f99d7SShri Abhyankar 
2090628f99d7SShri Abhyankar             nz_tmp = bi[prow+1] - bd[prow] - 1;
2091628f99d7SShri Abhyankar             for (j=0; j<nz_tmp; j++) {
2092628f99d7SShri Abhyankar               tmp          = pv[j];
2093628f99d7SShri Abhyankar               idx          = pj[j];
2094628f99d7SShri Abhyankar               rtmp11[idx] -= mul1 * tmp;
2095628f99d7SShri Abhyankar               rtmp22[idx] -= mul2 * tmp;
2096628f99d7SShri Abhyankar             }
209728b1a77fSLisandro Dalcin             ierr = PetscLogFlops(2+4*nz_tmp);CHKERRQ(ierr);
2098628f99d7SShri Abhyankar           }
2099628f99d7SShri Abhyankar           prow = *bjtmp++;
2100628f99d7SShri Abhyankar         }
2101628f99d7SShri Abhyankar 
2102628f99d7SShri Abhyankar         /* Now take care of diagonal 2x2 block. Note: prow = row here */
2103628f99d7SShri Abhyankar         pc1 = rtmp11 + prow;
2104628f99d7SShri Abhyankar         pc2 = rtmp22 + prow;
2105628f99d7SShri Abhyankar 
2106628f99d7SShri Abhyankar         sctx.pv = *pc1;
2107628f99d7SShri Abhyankar         pj      = bj + bi[prow];
2108628f99d7SShri Abhyankar         rs      = 0.0;
2109628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2110628f99d7SShri Abhyankar           idx = pj[j];
2111628f99d7SShri Abhyankar           if (idx != prow) rs += PetscAbsScalar(rtmp11[idx]);
2112628f99d7SShri Abhyankar         }
2113628f99d7SShri Abhyankar         sctx.rs = rs;
21144cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,row);CHKERRQ(ierr);
211507b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2116628f99d7SShri Abhyankar 
2117628f99d7SShri Abhyankar         if (*pc2 != 0.0) {
2118628f99d7SShri Abhyankar           pj     = nbj + bd[prow];
2119628f99d7SShri Abhyankar           mul2   = (*pc2)/(*pc1); /* since diag is not yet inverted.*/
2120628f99d7SShri Abhyankar           *pc2   = mul2;
2121628f99d7SShri Abhyankar           nz_tmp = bi[prow+1] - bd[prow] - 1;
2122628f99d7SShri Abhyankar           for (j=0; j<nz_tmp; j++) {
2123628f99d7SShri Abhyankar             idx          = pj[j];
2124628f99d7SShri Abhyankar             tmp          = rtmp11[idx];
2125628f99d7SShri Abhyankar             rtmp22[idx] -= mul2 * tmp;
2126628f99d7SShri Abhyankar           }
212728b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz_tmp);CHKERRQ(ierr);
2128628f99d7SShri Abhyankar         }
2129628f99d7SShri Abhyankar 
2130628f99d7SShri Abhyankar         pj  = bj + bi[row];
2131628f99d7SShri Abhyankar         pc1 = ba + bi[row];
2132628f99d7SShri Abhyankar         pc2 = ba + bi[row+1];
2133628f99d7SShri Abhyankar 
2134628f99d7SShri Abhyankar         sctx.pv       = rtmp22[row+1];
2135628f99d7SShri Abhyankar         rs            = 0.0;
2136628f99d7SShri Abhyankar         rtmp11[row]   = 1.0/rtmp11[row];
2137628f99d7SShri Abhyankar         rtmp22[row+1] = 1.0/rtmp22[row+1];
2138628f99d7SShri Abhyankar         /* copy row entries from dense representation to sparse */
2139628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2140628f99d7SShri Abhyankar           idx    = pj[j];
2141628f99d7SShri Abhyankar           pc1[j] = rtmp11[idx];
2142628f99d7SShri Abhyankar           pc2[j] = rtmp22[idx];
2143628f99d7SShri Abhyankar           if (idx != row+1) rs += PetscAbsScalar(pc2[j]);
2144628f99d7SShri Abhyankar         }
2145628f99d7SShri Abhyankar         sctx.rs = rs;
21464cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,row+1);CHKERRQ(ierr);
214707b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2148628f99d7SShri Abhyankar         break;
2149628f99d7SShri Abhyankar 
2150628f99d7SShri Abhyankar       case 3:
2151628f99d7SShri Abhyankar         for  (j=0; j<nz; j++) {
2152628f99d7SShri Abhyankar           idx         = bjtmp[j];
2153628f99d7SShri Abhyankar           rtmp11[idx] = 0.0;
2154628f99d7SShri Abhyankar           rtmp22[idx] = 0.0;
2155628f99d7SShri Abhyankar           rtmp33[idx] = 0.0;
2156628f99d7SShri Abhyankar         }
2157628f99d7SShri Abhyankar         /* copy the nonzeros for the 3 rows from sparse representation to dense in rtmp*[] */
2158628f99d7SShri Abhyankar         idx    = r[row];
2159628f99d7SShri Abhyankar         nz_tmp = ai[idx+1] - ai[idx];
2160628f99d7SShri Abhyankar         ajtmp  = aj + ai[idx];
2161628f99d7SShri Abhyankar         v1     = aa + ai[idx];
2162628f99d7SShri Abhyankar         v2     = aa + ai[idx+1];
2163628f99d7SShri Abhyankar         v3     = aa + ai[idx+2];
2164628f99d7SShri Abhyankar         for (j=0; j<nz_tmp; j++) {
2165628f99d7SShri Abhyankar           idx         = ics[ajtmp[j]];
2166628f99d7SShri Abhyankar           rtmp11[idx] = v1[j];
2167628f99d7SShri Abhyankar           rtmp22[idx] = v2[j];
2168628f99d7SShri Abhyankar           rtmp33[idx] = v3[j];
2169628f99d7SShri Abhyankar         }
2170628f99d7SShri Abhyankar         rtmp11[ics[r[row]]]   += sctx.shift_amount;
2171628f99d7SShri Abhyankar         rtmp22[ics[r[row+1]]] += sctx.shift_amount;
2172628f99d7SShri Abhyankar         rtmp33[ics[r[row+2]]] += sctx.shift_amount;
2173628f99d7SShri Abhyankar 
2174628f99d7SShri Abhyankar         /* loop over all pivot row blocks above this row block */
2175628f99d7SShri Abhyankar         prow = *bjtmp++;
2176628f99d7SShri Abhyankar         while (prow < row) {
2177628f99d7SShri Abhyankar           pc1 = rtmp11 + prow;
2178628f99d7SShri Abhyankar           pc2 = rtmp22 + prow;
2179628f99d7SShri Abhyankar           pc3 = rtmp33 + prow;
2180628f99d7SShri Abhyankar           if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 !=0.0) {
2181628f99d7SShri Abhyankar             pv   = ba  + bd[prow];
2182628f99d7SShri Abhyankar             pj   = nbj + bd[prow];
2183628f99d7SShri Abhyankar             mul1 = *pc1 * *pv;
2184628f99d7SShri Abhyankar             mul2 = *pc2 * *pv;
2185628f99d7SShri Abhyankar             mul3 = *pc3 * *pv;
2186628f99d7SShri Abhyankar             ++pv;
2187628f99d7SShri Abhyankar             *pc1 = mul1;
2188628f99d7SShri Abhyankar             *pc2 = mul2;
2189628f99d7SShri Abhyankar             *pc3 = mul3;
2190628f99d7SShri Abhyankar 
2191628f99d7SShri Abhyankar             nz_tmp = bi[prow+1] - bd[prow] - 1;
2192628f99d7SShri Abhyankar             /* update this row based on pivot row */
2193628f99d7SShri Abhyankar             for (j=0; j<nz_tmp; j++) {
2194628f99d7SShri Abhyankar               tmp          = pv[j];
2195628f99d7SShri Abhyankar               idx          = pj[j];
2196628f99d7SShri Abhyankar               rtmp11[idx] -= mul1 * tmp;
2197628f99d7SShri Abhyankar               rtmp22[idx] -= mul2 * tmp;
2198628f99d7SShri Abhyankar               rtmp33[idx] -= mul3 * tmp;
2199628f99d7SShri Abhyankar             }
220028b1a77fSLisandro Dalcin             ierr = PetscLogFlops(3+6*nz_tmp);CHKERRQ(ierr);
2201628f99d7SShri Abhyankar           }
2202628f99d7SShri Abhyankar           prow = *bjtmp++;
2203628f99d7SShri Abhyankar         }
2204628f99d7SShri Abhyankar 
2205628f99d7SShri Abhyankar         /* Now take care of diagonal 3x3 block in this set of rows */
2206628f99d7SShri Abhyankar         /* note: prow = row here */
2207628f99d7SShri Abhyankar         pc1 = rtmp11 + prow;
2208628f99d7SShri Abhyankar         pc2 = rtmp22 + prow;
2209628f99d7SShri Abhyankar         pc3 = rtmp33 + prow;
2210628f99d7SShri Abhyankar 
2211628f99d7SShri Abhyankar         sctx.pv = *pc1;
2212628f99d7SShri Abhyankar         pj      = bj + bi[prow];
2213628f99d7SShri Abhyankar         rs      = 0.0;
2214628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2215628f99d7SShri Abhyankar           idx = pj[j];
2216628f99d7SShri Abhyankar           if (idx != row) rs += PetscAbsScalar(rtmp11[idx]);
2217628f99d7SShri Abhyankar         }
2218628f99d7SShri Abhyankar         sctx.rs = rs;
22194cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,row);CHKERRQ(ierr);
222007b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2221628f99d7SShri Abhyankar 
2222628f99d7SShri Abhyankar         if (*pc2 != 0.0 || *pc3 != 0.0) {
2223628f99d7SShri Abhyankar           mul2   = (*pc2)/(*pc1);
2224628f99d7SShri Abhyankar           mul3   = (*pc3)/(*pc1);
2225628f99d7SShri Abhyankar           *pc2   = mul2;
2226628f99d7SShri Abhyankar           *pc3   = mul3;
2227628f99d7SShri Abhyankar           nz_tmp = bi[prow+1] - bd[prow] - 1;
2228628f99d7SShri Abhyankar           pj     = nbj + bd[prow];
2229628f99d7SShri Abhyankar           for (j=0; j<nz_tmp; j++) {
2230628f99d7SShri Abhyankar             idx          = pj[j];
2231628f99d7SShri Abhyankar             tmp          = rtmp11[idx];
2232628f99d7SShri Abhyankar             rtmp22[idx] -= mul2 * tmp;
2233628f99d7SShri Abhyankar             rtmp33[idx] -= mul3 * tmp;
2234628f99d7SShri Abhyankar           }
223528b1a77fSLisandro Dalcin           ierr = PetscLogFlops(2+4*nz_tmp);CHKERRQ(ierr);
2236628f99d7SShri Abhyankar         }
2237628f99d7SShri Abhyankar         ++prow;
2238628f99d7SShri Abhyankar 
2239628f99d7SShri Abhyankar         pc2     = rtmp22 + prow;
2240628f99d7SShri Abhyankar         pc3     = rtmp33 + prow;
2241628f99d7SShri Abhyankar         sctx.pv = *pc2;
2242628f99d7SShri Abhyankar         pj      = bj + bi[prow];
2243628f99d7SShri Abhyankar         rs      = 0.0;
2244628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2245628f99d7SShri Abhyankar           idx = pj[j];
2246628f99d7SShri Abhyankar           if (idx != prow) rs += PetscAbsScalar(rtmp22[idx]);
2247628f99d7SShri Abhyankar         }
2248628f99d7SShri Abhyankar         sctx.rs = rs;
22494cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,row+1);CHKERRQ(ierr);
225007b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2251628f99d7SShri Abhyankar 
2252628f99d7SShri Abhyankar         if (*pc3 != 0.0) {
2253628f99d7SShri Abhyankar           mul3   = (*pc3)/(*pc2);
2254628f99d7SShri Abhyankar           *pc3   = mul3;
2255628f99d7SShri Abhyankar           pj     = nbj + bd[prow];
2256628f99d7SShri Abhyankar           nz_tmp = bi[prow+1] - bd[prow] - 1;
2257628f99d7SShri Abhyankar           for (j=0; j<nz_tmp; j++) {
2258628f99d7SShri Abhyankar             idx          = pj[j];
2259628f99d7SShri Abhyankar             tmp          = rtmp22[idx];
2260628f99d7SShri Abhyankar             rtmp33[idx] -= mul3 * tmp;
2261628f99d7SShri Abhyankar           }
226228b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz_tmp);CHKERRQ(ierr);
2263628f99d7SShri Abhyankar         }
2264628f99d7SShri Abhyankar 
2265628f99d7SShri Abhyankar         pj  = bj + bi[row];
2266628f99d7SShri Abhyankar         pc1 = ba + bi[row];
2267628f99d7SShri Abhyankar         pc2 = ba + bi[row+1];
2268628f99d7SShri Abhyankar         pc3 = ba + bi[row+2];
2269628f99d7SShri Abhyankar 
2270628f99d7SShri Abhyankar         sctx.pv       = rtmp33[row+2];
2271628f99d7SShri Abhyankar         rs            = 0.0;
2272628f99d7SShri Abhyankar         rtmp11[row]   = 1.0/rtmp11[row];
2273628f99d7SShri Abhyankar         rtmp22[row+1] = 1.0/rtmp22[row+1];
2274628f99d7SShri Abhyankar         rtmp33[row+2] = 1.0/rtmp33[row+2];
2275628f99d7SShri Abhyankar         /* copy row entries from dense representation to sparse */
2276628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2277628f99d7SShri Abhyankar           idx    = pj[j];
2278628f99d7SShri Abhyankar           pc1[j] = rtmp11[idx];
2279628f99d7SShri Abhyankar           pc2[j] = rtmp22[idx];
2280628f99d7SShri Abhyankar           pc3[j] = rtmp33[idx];
2281628f99d7SShri Abhyankar           if (idx != row+2) rs += PetscAbsScalar(pc3[j]);
2282628f99d7SShri Abhyankar         }
2283628f99d7SShri Abhyankar 
2284628f99d7SShri Abhyankar         sctx.rs = rs;
22854cccfbddSHong Zhang         ierr    = MatPivotCheck(B,A,info,&sctx,row+2);CHKERRQ(ierr);
228607b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2287628f99d7SShri Abhyankar         break;
2288628f99d7SShri Abhyankar 
2289628f99d7SShri Abhyankar       default:
2290e32f2f54SBarry Smith         SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Node size not yet supported \n");
2291628f99d7SShri Abhyankar       }
2292628f99d7SShri Abhyankar       row += nodesz;                 /* Update the row */
2293628f99d7SShri Abhyankar     }
2294628f99d7SShri Abhyankar endofwhile:;
229507b50cabSHong Zhang   } while (sctx.newshift);
2296eb9baa12SBarry Smith   ierr = PetscFree3(rtmp11,rtmp22,rtmp33);CHKERRQ(ierr);
2297628f99d7SShri Abhyankar   ierr = PetscFree(tmp_vec2);CHKERRQ(ierr);
2298628f99d7SShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2299628f99d7SShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
2300628f99d7SShri Abhyankar   ierr = ISRestoreIndices(iscol,&c);CHKERRQ(ierr);
23012205254eSKarl Rupp 
2302d3ac4fa3SBarry Smith   (B)->ops->solve = MatSolve_SeqAIJ_inplace;
2303628f99d7SShri Abhyankar   /* do not set solve add, since MatSolve_Inode + Add is faster */
2304628f99d7SShri Abhyankar   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
2305628f99d7SShri Abhyankar   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
2306628f99d7SShri Abhyankar   C->assembled              = PETSC_TRUE;
2307628f99d7SShri Abhyankar   C->preallocated           = PETSC_TRUE;
2308628f99d7SShri Abhyankar   if (sctx.nshift) {
2309f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
231057622a8eSBarry 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);
2311f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
231257622a8eSBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr);
2313628f99d7SShri Abhyankar     }
2314628f99d7SShri Abhyankar   }
2315628f99d7SShri Abhyankar   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
2316a02bda8eSBarry Smith   ierr = MatSeqAIJCheckInode(C);CHKERRQ(ierr);
2317628f99d7SShri Abhyankar   PetscFunctionReturn(0);
2318628f99d7SShri Abhyankar }
2319628f99d7SShri Abhyankar 
2320628f99d7SShri Abhyankar 
2321019b515eSShri Abhyankar /* ----------------------------------------------------------- */
2322019b515eSShri Abhyankar #undef __FUNCT__
232335233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_Inode"
232435233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_Inode(Mat A,Vec bb,Vec xx)
2325019b515eSShri Abhyankar {
2326019b515eSShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
2327019b515eSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
2328019b515eSShri Abhyankar   PetscErrorCode    ierr;
2329019b515eSShri Abhyankar   const PetscInt    *r,*c,*rout,*cout;
23308758e1faSBarry Smith   PetscInt          i,j,n = A->rmap->n;
23318758e1faSBarry Smith   PetscInt          node_max,row,nsz,aii,i0,i1,nz;
23328758e1faSBarry Smith   const PetscInt    *ai = a->i,*a_j = a->j,*ns,*vi,*ad,*aj;
2333019b515eSShri Abhyankar   PetscScalar       *x,*tmp,*tmps,tmp0,tmp1;
2334019b515eSShri Abhyankar   PetscScalar       sum1,sum2,sum3,sum4,sum5;
2335019b515eSShri Abhyankar   const MatScalar   *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa;
2336019b515eSShri Abhyankar   const PetscScalar *b;
2337019b515eSShri Abhyankar 
2338019b515eSShri Abhyankar   PetscFunctionBegin;
2339e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
2340019b515eSShri Abhyankar   node_max = a->inode.node_count;
2341019b515eSShri Abhyankar   ns       = a->inode.size;     /* Node Size array */
2342019b515eSShri Abhyankar 
23433649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
2344019b515eSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2345019b515eSShri Abhyankar   tmp  = a->solve_work;
2346019b515eSShri Abhyankar 
2347019b515eSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
2348019b515eSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
2349019b515eSShri Abhyankar 
2350019b515eSShri Abhyankar   /* forward solve the lower triangular */
2351019b515eSShri Abhyankar   tmps = tmp;
2352019b515eSShri Abhyankar   aa   = a_a;
2353019b515eSShri Abhyankar   aj   = a_j;
2354019b515eSShri Abhyankar   ad   = a->diag;
2355019b515eSShri Abhyankar 
2356019b515eSShri Abhyankar   for (i = 0,row = 0; i< node_max; ++i) {
2357019b515eSShri Abhyankar     nsz = ns[i];
2358019b515eSShri Abhyankar     aii = ai[row];
2359019b515eSShri Abhyankar     v1  = aa + aii;
2360019b515eSShri Abhyankar     vi  = aj + aii;
2361019b515eSShri Abhyankar     nz  = ai[row+1]- ai[row];
2362019b515eSShri Abhyankar 
236398991853SShri Abhyankar     if (i < node_max-1) {
236498991853SShri Abhyankar       /* Prefetch the indices for the next block */
236550d8bf02SJed Brown       PetscPrefetchBlock(aj+ai[row+nsz],ai[row+nsz+1]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); /* indices */
236698991853SShri Abhyankar       /* Prefetch the data for the next block */
236750d8bf02SJed Brown       PetscPrefetchBlock(aa+ai[row+nsz],ai[row+nsz+ns[i+1]]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA);
236898991853SShri Abhyankar     }
236998991853SShri Abhyankar 
2370019b515eSShri Abhyankar     switch (nsz) {               /* Each loop in 'case' is unrolled */
2371019b515eSShri Abhyankar     case 1:
2372019b515eSShri Abhyankar       sum1 = b[r[row]];
2373019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2374019b515eSShri Abhyankar         i0    = vi[j];
2375019b515eSShri Abhyankar         i1    = vi[j+1];
2376019b515eSShri Abhyankar         tmp0  = tmps[i0];
2377019b515eSShri Abhyankar         tmp1  = tmps[i1];
2378019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0 + v1[j+1]*tmp1;
2379019b515eSShri Abhyankar       }
2380019b515eSShri Abhyankar       if (j == nz-1) {
2381019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2382019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0;
2383019b515eSShri Abhyankar       }
2384019b515eSShri Abhyankar       tmp[row++]=sum1;
2385019b515eSShri Abhyankar       break;
2386019b515eSShri Abhyankar     case 2:
2387019b515eSShri Abhyankar       sum1 = b[r[row]];
2388019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2389019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2390019b515eSShri Abhyankar 
2391019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2392019b515eSShri Abhyankar         i0    = vi[j];
2393019b515eSShri Abhyankar         i1    = vi[j+1];
2394019b515eSShri Abhyankar         tmp0  = tmps[i0];
2395019b515eSShri Abhyankar         tmp1  = tmps[i1];
2396019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2397019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2398019b515eSShri Abhyankar       }
2399019b515eSShri Abhyankar       if (j == nz-1) {
2400019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2401019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2402019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2403019b515eSShri Abhyankar       }
2404019b515eSShri Abhyankar       sum2     -= v2[nz] * sum1;
2405019b515eSShri Abhyankar       tmp[row++]=sum1;
2406019b515eSShri Abhyankar       tmp[row++]=sum2;
2407019b515eSShri Abhyankar       break;
2408019b515eSShri Abhyankar     case 3:
2409019b515eSShri Abhyankar       sum1 = b[r[row]];
2410019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2411019b515eSShri Abhyankar       sum3 = b[r[row+2]];
2412019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2413019b515eSShri Abhyankar       v3   = aa + ai[row+2];
2414019b515eSShri Abhyankar 
2415019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2416019b515eSShri Abhyankar         i0    = vi[j];
2417019b515eSShri Abhyankar         i1    = vi[j+1];
2418019b515eSShri Abhyankar         tmp0  = tmps[i0];
2419019b515eSShri Abhyankar         tmp1  = tmps[i1];
2420019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2421019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2422019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
2423019b515eSShri Abhyankar       }
2424019b515eSShri Abhyankar       if (j == nz-1) {
2425019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2426019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2427019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2428019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
2429019b515eSShri Abhyankar       }
2430019b515eSShri Abhyankar       sum2     -= v2[nz] * sum1;
2431019b515eSShri Abhyankar       sum3     -= v3[nz] * sum1;
2432019b515eSShri Abhyankar       sum3     -= v3[nz+1] * sum2;
2433019b515eSShri Abhyankar       tmp[row++]=sum1;
2434019b515eSShri Abhyankar       tmp[row++]=sum2;
2435019b515eSShri Abhyankar       tmp[row++]=sum3;
2436019b515eSShri Abhyankar       break;
2437019b515eSShri Abhyankar 
2438019b515eSShri Abhyankar     case 4:
2439019b515eSShri Abhyankar       sum1 = b[r[row]];
2440019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2441019b515eSShri Abhyankar       sum3 = b[r[row+2]];
2442019b515eSShri Abhyankar       sum4 = b[r[row+3]];
2443019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2444019b515eSShri Abhyankar       v3   = aa + ai[row+2];
2445019b515eSShri Abhyankar       v4   = aa + ai[row+3];
2446019b515eSShri Abhyankar 
2447019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2448019b515eSShri Abhyankar         i0    = vi[j];
2449019b515eSShri Abhyankar         i1    = vi[j+1];
2450019b515eSShri Abhyankar         tmp0  = tmps[i0];
2451019b515eSShri Abhyankar         tmp1  = tmps[i1];
2452019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2453019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2454019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
2455019b515eSShri Abhyankar         sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1;
2456019b515eSShri Abhyankar       }
2457019b515eSShri Abhyankar       if (j == nz-1) {
2458019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2459019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2460019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2461019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
2462019b515eSShri Abhyankar         sum4 -= v4[j] *tmp0;
2463019b515eSShri Abhyankar       }
2464019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
2465019b515eSShri Abhyankar       sum3 -= v3[nz] * sum1;
2466019b515eSShri Abhyankar       sum4 -= v4[nz] * sum1;
2467019b515eSShri Abhyankar       sum3 -= v3[nz+1] * sum2;
2468019b515eSShri Abhyankar       sum4 -= v4[nz+1] * sum2;
2469019b515eSShri Abhyankar       sum4 -= v4[nz+2] * sum3;
2470019b515eSShri Abhyankar 
2471019b515eSShri Abhyankar       tmp[row++]=sum1;
2472019b515eSShri Abhyankar       tmp[row++]=sum2;
2473019b515eSShri Abhyankar       tmp[row++]=sum3;
2474019b515eSShri Abhyankar       tmp[row++]=sum4;
2475019b515eSShri Abhyankar       break;
2476019b515eSShri Abhyankar     case 5:
2477019b515eSShri Abhyankar       sum1 = b[r[row]];
2478019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2479019b515eSShri Abhyankar       sum3 = b[r[row+2]];
2480019b515eSShri Abhyankar       sum4 = b[r[row+3]];
2481019b515eSShri Abhyankar       sum5 = b[r[row+4]];
2482019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2483019b515eSShri Abhyankar       v3   = aa + ai[row+2];
2484019b515eSShri Abhyankar       v4   = aa + ai[row+3];
2485019b515eSShri Abhyankar       v5   = aa + ai[row+4];
2486019b515eSShri Abhyankar 
2487019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2488019b515eSShri Abhyankar         i0    = vi[j];
2489019b515eSShri Abhyankar         i1    = vi[j+1];
2490019b515eSShri Abhyankar         tmp0  = tmps[i0];
2491019b515eSShri Abhyankar         tmp1  = tmps[i1];
2492019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2493019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2494019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
2495019b515eSShri Abhyankar         sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1;
2496019b515eSShri Abhyankar         sum5 -= v5[j] * tmp0 + v5[j+1] * tmp1;
2497019b515eSShri Abhyankar       }
2498019b515eSShri Abhyankar       if (j == nz-1) {
2499019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2500019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2501019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2502019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
2503019b515eSShri Abhyankar         sum4 -= v4[j] *tmp0;
2504019b515eSShri Abhyankar         sum5 -= v5[j] *tmp0;
2505019b515eSShri Abhyankar       }
2506019b515eSShri Abhyankar 
2507019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
2508019b515eSShri Abhyankar       sum3 -= v3[nz] * sum1;
2509019b515eSShri Abhyankar       sum4 -= v4[nz] * sum1;
2510019b515eSShri Abhyankar       sum5 -= v5[nz] * sum1;
2511019b515eSShri Abhyankar       sum3 -= v3[nz+1] * sum2;
2512019b515eSShri Abhyankar       sum4 -= v4[nz+1] * sum2;
2513019b515eSShri Abhyankar       sum5 -= v5[nz+1] * sum2;
2514019b515eSShri Abhyankar       sum4 -= v4[nz+2] * sum3;
2515019b515eSShri Abhyankar       sum5 -= v5[nz+2] * sum3;
2516019b515eSShri Abhyankar       sum5 -= v5[nz+3] * sum4;
2517019b515eSShri Abhyankar 
2518019b515eSShri Abhyankar       tmp[row++]=sum1;
2519019b515eSShri Abhyankar       tmp[row++]=sum2;
2520019b515eSShri Abhyankar       tmp[row++]=sum3;
2521019b515eSShri Abhyankar       tmp[row++]=sum4;
2522019b515eSShri Abhyankar       tmp[row++]=sum5;
2523019b515eSShri Abhyankar       break;
2524019b515eSShri Abhyankar     default:
2525e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
2526019b515eSShri Abhyankar     }
2527019b515eSShri Abhyankar   }
2528019b515eSShri Abhyankar   /* backward solve the upper triangular */
2529019b515eSShri Abhyankar   for (i=node_max -1,row = n-1; i>=0; i--) {
2530019b515eSShri Abhyankar     nsz = ns[i];
2531019b515eSShri Abhyankar     aii = ad[row+1] + 1;
2532019b515eSShri Abhyankar     v1  = aa + aii;
2533019b515eSShri Abhyankar     vi  = aj + aii;
2534019b515eSShri Abhyankar     nz  = ad[row]- ad[row+1] - 1;
253598991853SShri Abhyankar 
253698991853SShri Abhyankar     if (i > 0) {
253798991853SShri Abhyankar       /* Prefetch the indices for the next block */
253850d8bf02SJed Brown       PetscPrefetchBlock(aj+ad[row-nsz+1]+1,ad[row-nsz]-ad[row-nsz+1],0,PETSC_PREFETCH_HINT_NTA);
253998991853SShri Abhyankar       /* Prefetch the data for the next block */
254050d8bf02SJed Brown       PetscPrefetchBlock(aa+ad[row-nsz+1]+1,ad[row-nsz-ns[i-1]+1]-ad[row-nsz+1],0,PETSC_PREFETCH_HINT_NTA);
254198991853SShri Abhyankar     }
254298991853SShri Abhyankar 
2543019b515eSShri Abhyankar     switch (nsz) {               /* Each loop in 'case' is unrolled */
2544019b515eSShri Abhyankar     case 1:
2545019b515eSShri Abhyankar       sum1 = tmp[row];
2546019b515eSShri Abhyankar 
2547019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2548019b515eSShri Abhyankar         i0    = vi[j];
2549019b515eSShri Abhyankar         i1    = vi[j+1];
2550019b515eSShri Abhyankar         tmp0  = tmps[i0];
2551019b515eSShri Abhyankar         tmp1  = tmps[i1];
2552019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2553019b515eSShri Abhyankar       }
2554019b515eSShri Abhyankar       if (j == nz-1) {
2555019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2556019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0;
2557019b515eSShri Abhyankar       }
2558019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2559019b515eSShri Abhyankar       break;
2560019b515eSShri Abhyankar     case 2:
2561019b515eSShri Abhyankar       sum1 = tmp[row];
2562019b515eSShri Abhyankar       sum2 = tmp[row-1];
2563019b515eSShri Abhyankar       v2   = aa + ad[row] + 1;
2564019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2565019b515eSShri Abhyankar         i0    = vi[j];
2566019b515eSShri Abhyankar         i1    = vi[j+1];
2567019b515eSShri Abhyankar         tmp0  = tmps[i0];
2568019b515eSShri Abhyankar         tmp1  = tmps[i1];
2569019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2570019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2571019b515eSShri Abhyankar       }
2572019b515eSShri Abhyankar       if (j == nz-1) {
2573019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2574019b515eSShri Abhyankar         sum1 -= v1[j]* tmp0;
2575019b515eSShri Abhyankar         sum2 -= v2[j+1]* tmp0;
2576019b515eSShri Abhyankar       }
2577019b515eSShri Abhyankar 
2578019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2579019b515eSShri Abhyankar       sum2     -= v2[0] * tmp0;
2580019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2581019b515eSShri Abhyankar       break;
2582019b515eSShri Abhyankar     case 3:
2583019b515eSShri Abhyankar       sum1 = tmp[row];
2584019b515eSShri Abhyankar       sum2 = tmp[row -1];
2585019b515eSShri Abhyankar       sum3 = tmp[row -2];
2586019b515eSShri Abhyankar       v2   = aa + ad[row] + 1;
2587019b515eSShri Abhyankar       v3   = aa + ad[row -1] + 1;
2588019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2589019b515eSShri Abhyankar         i0    = vi[j];
2590019b515eSShri Abhyankar         i1    = vi[j+1];
2591019b515eSShri Abhyankar         tmp0  = tmps[i0];
2592019b515eSShri Abhyankar         tmp1  = tmps[i1];
2593019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2594019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2595019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
2596019b515eSShri Abhyankar       }
2597019b515eSShri Abhyankar       if (j== nz-1) {
2598019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2599019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
2600019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
2601019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
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       tmp0      = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2607019b515eSShri Abhyankar       sum3     -= v3[0]* tmp0;
2608019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
2609019b515eSShri Abhyankar 
2610019b515eSShri Abhyankar       break;
2611019b515eSShri Abhyankar     case 4:
2612019b515eSShri Abhyankar       sum1 = tmp[row];
2613019b515eSShri Abhyankar       sum2 = tmp[row -1];
2614019b515eSShri Abhyankar       sum3 = tmp[row -2];
2615019b515eSShri Abhyankar       sum4 = tmp[row -3];
2616019b515eSShri Abhyankar       v2   = aa + ad[row]+1;
2617019b515eSShri Abhyankar       v3   = aa + ad[row -1]+1;
2618019b515eSShri Abhyankar       v4   = aa + ad[row -2]+1;
2619019b515eSShri Abhyankar 
2620019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2621019b515eSShri Abhyankar         i0    = vi[j];
2622019b515eSShri Abhyankar         i1    = vi[j+1];
2623019b515eSShri Abhyankar         tmp0  = tmps[i0];
2624019b515eSShri Abhyankar         tmp1  = tmps[i1];
2625019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0   + v1[j+1] * tmp1;
2626019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2627019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
2628019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1;
2629019b515eSShri Abhyankar       }
2630019b515eSShri Abhyankar       if (j== nz-1) {
2631019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2632019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
2633019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
2634019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
2635019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0;
2636019b515eSShri Abhyankar       }
2637019b515eSShri Abhyankar 
2638019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2639019b515eSShri Abhyankar       sum2     -= v2[0] * tmp0;
2640019b515eSShri Abhyankar       sum3     -= v3[1] * tmp0;
2641019b515eSShri Abhyankar       sum4     -= v4[2] * tmp0;
2642019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2643019b515eSShri Abhyankar       sum3     -= v3[0] * tmp0;
2644019b515eSShri Abhyankar       sum4     -= v4[1] * tmp0;
2645019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
2646019b515eSShri Abhyankar       sum4     -= v4[0] * tmp0;
2647019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--;
2648019b515eSShri Abhyankar       break;
2649019b515eSShri Abhyankar     case 5:
2650019b515eSShri Abhyankar       sum1 = tmp[row];
2651019b515eSShri Abhyankar       sum2 = tmp[row -1];
2652019b515eSShri Abhyankar       sum3 = tmp[row -2];
2653019b515eSShri Abhyankar       sum4 = tmp[row -3];
2654019b515eSShri Abhyankar       sum5 = tmp[row -4];
2655019b515eSShri Abhyankar       v2   = aa + ad[row]+1;
2656019b515eSShri Abhyankar       v3   = aa + ad[row -1]+1;
2657019b515eSShri Abhyankar       v4   = aa + ad[row -2]+1;
2658019b515eSShri Abhyankar       v5   = aa + ad[row -3]+1;
2659019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2660019b515eSShri Abhyankar         i0    = vi[j];
2661019b515eSShri Abhyankar         i1    = vi[j+1];
2662019b515eSShri Abhyankar         tmp0  = tmps[i0];
2663019b515eSShri Abhyankar         tmp1  = tmps[i1];
2664019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2665019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2666019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
2667019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1;
2668019b515eSShri Abhyankar         sum5 -= v5[j+4] * tmp0 + v5[j+5] * tmp1;
2669019b515eSShri Abhyankar       }
2670019b515eSShri Abhyankar       if (j==nz-1) {
2671019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2672019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
2673019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
2674019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
2675019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0;
2676019b515eSShri Abhyankar         sum5 -= v5[j+4] * tmp0;
2677019b515eSShri Abhyankar       }
2678019b515eSShri Abhyankar 
2679019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2680019b515eSShri Abhyankar       sum2     -= v2[0] * tmp0;
2681019b515eSShri Abhyankar       sum3     -= v3[1] * tmp0;
2682019b515eSShri Abhyankar       sum4     -= v4[2] * tmp0;
2683019b515eSShri Abhyankar       sum5     -= v5[3] * tmp0;
2684019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2685019b515eSShri Abhyankar       sum3     -= v3[0] * tmp0;
2686019b515eSShri Abhyankar       sum4     -= v4[1] * tmp0;
2687019b515eSShri Abhyankar       sum5     -= v5[2] * tmp0;
2688019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
2689019b515eSShri Abhyankar       sum4     -= v4[0] * tmp0;
2690019b515eSShri Abhyankar       sum5     -= v5[1] * tmp0;
2691019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--;
2692019b515eSShri Abhyankar       sum5     -= v5[0] * tmp0;
2693019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum5*v5[nz+4]; row--;
2694019b515eSShri Abhyankar       break;
2695019b515eSShri Abhyankar     default:
2696e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
2697019b515eSShri Abhyankar     }
2698019b515eSShri Abhyankar   }
2699019b515eSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
2700019b515eSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
27013649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
2702019b515eSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
2703019b515eSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
2704019b515eSShri Abhyankar   PetscFunctionReturn(0);
2705019b515eSShri Abhyankar }
2706019b515eSShri Abhyankar 
2707019b515eSShri Abhyankar 
27084c1414c8SBarry Smith /*
27094c1414c8SBarry Smith      Makes a longer coloring[] array and calls the usual code with that
27104c1414c8SBarry Smith */
27114c1414c8SBarry Smith #undef __FUNCT__
27124108e4d5SBarry Smith #define __FUNCT__ "MatColoringPatch_SeqAIJ_Inode"
27134108e4d5SBarry Smith PetscErrorCode MatColoringPatch_SeqAIJ_Inode(Mat mat,PetscInt ncolors,PetscInt nin,ISColoringValue coloring[],ISColoring *iscoloring)
27144c1414c8SBarry Smith {
27154c1414c8SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)mat->data;
27164c1414c8SBarry Smith   PetscErrorCode  ierr;
2717d0f46423SBarry Smith   PetscInt        n = mat->cmap->n,m = a->inode.node_count,j,*ns = a->inode.size,row;
27184c1414c8SBarry Smith   PetscInt        *colorused,i;
27194c1414c8SBarry Smith   ISColoringValue *newcolor;
27204c1414c8SBarry Smith 
27214c1414c8SBarry Smith   PetscFunctionBegin;
2722854ce69bSBarry Smith   ierr = PetscMalloc1(n+1,&newcolor);CHKERRQ(ierr);
27234c1414c8SBarry Smith   /* loop over inodes, marking a color for each column*/
27244c1414c8SBarry Smith   row = 0;
27254c1414c8SBarry Smith   for (i=0; i<m; i++) {
27264c1414c8SBarry Smith     for (j=0; j<ns[i]; j++) {
27274c1414c8SBarry Smith       newcolor[row++] = coloring[i] + j*ncolors;
27284c1414c8SBarry Smith     }
27294c1414c8SBarry Smith   }
27304c1414c8SBarry Smith 
27314c1414c8SBarry Smith   /* eliminate unneeded colors */
27321795a4d1SJed Brown   ierr = PetscCalloc1(5*ncolors,&colorused);CHKERRQ(ierr);
27334c1414c8SBarry Smith   for (i=0; i<n; i++) {
27344c1414c8SBarry Smith     colorused[newcolor[i]] = 1;
27354c1414c8SBarry Smith   }
27364c1414c8SBarry Smith 
27374c1414c8SBarry Smith   for (i=1; i<5*ncolors; i++) {
27384c1414c8SBarry Smith     colorused[i] += colorused[i-1];
27394c1414c8SBarry Smith   }
27404c1414c8SBarry Smith   ncolors = colorused[5*ncolors-1];
27414c1414c8SBarry Smith   for (i=0; i<n; i++) {
274257815681SSatish Balay     newcolor[i] = colorused[newcolor[i]]-1;
27434c1414c8SBarry Smith   }
27444c1414c8SBarry Smith   ierr = PetscFree(colorused);CHKERRQ(ierr);
2745aaf3ff59SMatthew G. Knepley   ierr = ISColoringCreate(PetscObjectComm((PetscObject)mat),ncolors,n,newcolor,PETSC_OWN_POINTER,iscoloring);CHKERRQ(ierr);
27464c1414c8SBarry Smith   ierr = PetscFree(coloring);CHKERRQ(ierr);
27474c1414c8SBarry Smith   PetscFunctionReturn(0);
27484c1414c8SBarry Smith }
27494c1414c8SBarry Smith 
2750af0996ceSBarry Smith #include <petsc/private/kernels/blockinvert.h>
27512af78befSBarry Smith 
27522af78befSBarry Smith #undef __FUNCT__
27534108e4d5SBarry Smith #define __FUNCT__ "MatSOR_SeqAIJ_Inode"
27544108e4d5SBarry Smith PetscErrorCode MatSOR_SeqAIJ_Inode(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx)
27552af78befSBarry Smith {
27562af78befSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
27577aaeff0aSMatthew G. Knepley   PetscScalar       sum1 = 0.0,sum2 = 0.0,sum3 = 0.0,sum4 = 0.0,sum5 = 0.0,tmp0,tmp1,tmp2,tmp3;
27585850ef23SBarry Smith   MatScalar         *ibdiag,*bdiag,work[25],*t;
2759a8b09249SBarry Smith   PetscScalar       *x,tmp4,tmp5,x1,x2,x3,x4,x5;
27607aaeff0aSMatthew G. Knepley   const MatScalar   *v = a->a,*v1 = NULL,*v2 = NULL,*v3 = NULL,*v4 = NULL,*v5 = NULL;
27615850ef23SBarry Smith   const PetscScalar *xb, *b;
2762*7b6c816cSBarry Smith   PetscReal         zeropivot = 100.*PETSC_MACHINE_EPSILON, shift = 0.0;
27632af78befSBarry Smith   PetscErrorCode    ierr;
27648758e1faSBarry Smith   PetscInt          n,m = a->inode.node_count,cnt = 0,i,j,row,i1,i2;
27658758e1faSBarry Smith   PetscInt          sz,k,ipvt[5];
2766*7b6c816cSBarry Smith   PetscBool         allowzeropivot,zeropivotdetected;
27678758e1faSBarry Smith   const PetscInt    *sizes = a->inode.size,*idx,*diag = a->diag,*ii = a->i;
27682af78befSBarry Smith 
27692af78befSBarry Smith   PetscFunctionBegin;
2770a455e926SHong Zhang   allowzeropivot = PetscNot(A->erroriffailure);
2771e32f2f54SBarry Smith   if (omega != 1.0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for omega != 1.0; use -mat_no_inode");
2772422a814eSBarry Smith   if (fshift == -1.0) fshift = 0.0; /* negative fshift indicates do not error on zero diagonal; this code never errors on zero diagonal */
2773e32f2f54SBarry Smith   if (fshift != 0.0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for fshift != 0.0; use -mat_no_inode");
27742af78befSBarry Smith 
277571f1c65dSBarry Smith   if (!a->inode.ibdiagvalid) {
27762af78befSBarry Smith     if (!a->inode.ibdiag) {
27772af78befSBarry Smith       /* calculate space needed for diagonal blocks */
27782af78befSBarry Smith       for (i=0; i<m; i++) {
27792af78befSBarry Smith         cnt += sizes[i]*sizes[i];
27802af78befSBarry Smith       }
2781f0d39aaaSBarry Smith       a->inode.bdiagsize = cnt;
27822205254eSKarl Rupp 
2783dcca6d9dSJed Brown       ierr = PetscMalloc3(cnt,&a->inode.ibdiag,cnt,&a->inode.bdiag,A->rmap->n,&a->inode.ssor_work);CHKERRQ(ierr);
278471f1c65dSBarry Smith     }
278571f1c65dSBarry Smith 
278671f1c65dSBarry Smith     /* copy over the diagonal blocks and invert them */
27872af78befSBarry Smith     ibdiag = a->inode.ibdiag;
27882af78befSBarry Smith     bdiag  = a->inode.bdiag;
27892af78befSBarry Smith     cnt    = 0;
27902af78befSBarry Smith     for (i=0, row = 0; i<m; i++) {
27912af78befSBarry Smith       for (j=0; j<sizes[i]; j++) {
2792f0d39aaaSBarry Smith         for (k=0; k<sizes[i]; k++) {
2793f0d39aaaSBarry Smith           bdiag[cnt+k*sizes[i]+j] = v[diag[row+j] - j + k];
2794f0d39aaaSBarry Smith         }
27952af78befSBarry Smith       }
2796a77337e4SBarry Smith       ierr = PetscMemcpy(ibdiag+cnt,bdiag+cnt,sizes[i]*sizes[i]*sizeof(MatScalar));CHKERRQ(ierr);
27972af78befSBarry Smith 
27982af78befSBarry Smith       switch (sizes[i]) {
27992af78befSBarry Smith       case 1:
28002af78befSBarry Smith         /* Create matrix data structure */
28018e0e2a9aSHong Zhang         if (PetscAbsScalar(ibdiag[cnt]) < zeropivot) {
28028e0e2a9aSHong Zhang           if (allowzeropivot) {
2803*7b6c816cSBarry Smith             A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT;
2804*7b6c816cSBarry Smith             A->factorerror_zeropivot_value = PetscAbsScalar(ibdiag[cnt]);
2805*7b6c816cSBarry Smith             A->factorerror_zeropivot_row   = row;
28068e0e2a9aSHong Zhang             ierr = PetscInfo1(A,"Zero pivot, row %D\n",row);CHKERRQ(ierr);
28078e0e2a9aSHong Zhang           } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot on row %D",row);
28088e0e2a9aSHong Zhang         }
280964c62002SMatthew Knepley         ibdiag[cnt] = 1.0/ibdiag[cnt];
28102af78befSBarry Smith         break;
28112af78befSBarry Smith       case 2:
2812a455e926SHong Zhang         ierr = PetscKernel_A_gets_inverse_A_2(ibdiag+cnt,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr);
2813*7b6c816cSBarry Smith         if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT;
28142af78befSBarry Smith         break;
28152af78befSBarry Smith       case 3:
2816a455e926SHong Zhang         ierr = PetscKernel_A_gets_inverse_A_3(ibdiag+cnt,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr);
2817*7b6c816cSBarry Smith         if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT;
28182af78befSBarry Smith         break;
28192af78befSBarry Smith       case 4:
2820a455e926SHong Zhang         ierr = PetscKernel_A_gets_inverse_A_4(ibdiag+cnt,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr);
2821*7b6c816cSBarry Smith         if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT;
28222af78befSBarry Smith         break;
28232af78befSBarry Smith       case 5:
2824a455e926SHong Zhang         ierr = PetscKernel_A_gets_inverse_A_5(ibdiag+cnt,ipvt,work,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr);
2825*7b6c816cSBarry Smith         if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT;
28262af78befSBarry Smith         break;
28272af78befSBarry Smith       default:
2828e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
28292af78befSBarry Smith       }
28302af78befSBarry Smith       cnt += sizes[i]*sizes[i];
28312af78befSBarry Smith       row += sizes[i];
28322af78befSBarry Smith     }
283371f1c65dSBarry Smith     a->inode.ibdiagvalid = PETSC_TRUE;
28342af78befSBarry Smith   }
28352af78befSBarry Smith   ibdiag = a->inode.ibdiag;
28362af78befSBarry Smith   bdiag  = a->inode.bdiag;
28375850ef23SBarry Smith   t      = a->inode.ssor_work;
28382af78befSBarry Smith 
28392af78befSBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
28403649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
28415850ef23SBarry Smith   /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */
28425850ef23SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
28432af78befSBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) {
28442af78befSBarry Smith 
28458862d2efSBarry Smith       for (i=0, row=0; i<m; i++) {
28468862d2efSBarry Smith         sz  = diag[row] - ii[row];
28478862d2efSBarry Smith         v1  = a->a + ii[row];
28488862d2efSBarry Smith         idx = a->j + ii[row];
28498862d2efSBarry Smith 
28504108e4d5SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
28518862d2efSBarry Smith         switch (sizes[i]) {
28528862d2efSBarry Smith         case 1:
28538862d2efSBarry Smith 
28548862d2efSBarry Smith           sum1 = b[row];
28558862d2efSBarry Smith           for (n = 0; n<sz-1; n+=2) {
28568862d2efSBarry Smith             i1    = idx[0];
28578862d2efSBarry Smith             i2    = idx[1];
28588862d2efSBarry Smith             idx  += 2;
28598862d2efSBarry Smith             tmp0  = x[i1];
28608862d2efSBarry Smith             tmp1  = x[i2];
28618862d2efSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
28628862d2efSBarry Smith           }
28638862d2efSBarry Smith 
28648862d2efSBarry Smith           if (n == sz-1) {
2865f0d39aaaSBarry Smith             tmp0  = x[*idx];
2866f0d39aaaSBarry Smith             sum1 -= *v1 * tmp0;
28678862d2efSBarry Smith           }
28685850ef23SBarry Smith           t[row]   = sum1;
28698862d2efSBarry Smith           x[row++] = sum1*(*ibdiag++);
28708862d2efSBarry Smith           break;
2871f0d39aaaSBarry Smith         case 2:
2872f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2873f0d39aaaSBarry Smith           sum1 = b[row];
2874f0d39aaaSBarry Smith           sum2 = b[row+1];
2875f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2876f0d39aaaSBarry Smith             i1    = idx[0];
2877f0d39aaaSBarry Smith             i2    = idx[1];
2878f0d39aaaSBarry Smith             idx  += 2;
2879f0d39aaaSBarry Smith             tmp0  = x[i1];
2880f0d39aaaSBarry Smith             tmp1  = x[i2];
2881f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2882f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2883f0d39aaaSBarry Smith           }
2884f0d39aaaSBarry Smith 
2885f0d39aaaSBarry Smith           if (n == sz-1) {
2886f0d39aaaSBarry Smith             tmp0  = x[*idx];
2887f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2888f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2889f0d39aaaSBarry Smith           }
28905850ef23SBarry Smith           t[row]   = sum1;
28915850ef23SBarry Smith           t[row+1] = sum2;
2892f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[2];
2893f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[3];
2894f0d39aaaSBarry Smith           ibdiag  += 4;
2895f0d39aaaSBarry Smith           break;
2896f0d39aaaSBarry Smith         case 3:
2897f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2898f0d39aaaSBarry Smith           v3   = a->a + ii[row+2];
2899f0d39aaaSBarry Smith           sum1 = b[row];
2900f0d39aaaSBarry Smith           sum2 = b[row+1];
2901f0d39aaaSBarry Smith           sum3 = b[row+2];
2902f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2903f0d39aaaSBarry Smith             i1    = idx[0];
2904f0d39aaaSBarry Smith             i2    = idx[1];
2905f0d39aaaSBarry Smith             idx  += 2;
2906f0d39aaaSBarry Smith             tmp0  = x[i1];
2907f0d39aaaSBarry Smith             tmp1  = x[i2];
2908f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2909f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2910f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2911f0d39aaaSBarry Smith           }
2912f0d39aaaSBarry Smith 
2913f0d39aaaSBarry Smith           if (n == sz-1) {
2914f0d39aaaSBarry Smith             tmp0  = x[*idx];
2915f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2916f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2917f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0;
2918f0d39aaaSBarry Smith           }
29195850ef23SBarry Smith           t[row]   = sum1;
29205850ef23SBarry Smith           t[row+1] = sum2;
29215850ef23SBarry Smith           t[row+2] = sum3;
2922f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
2923f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
2924f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
2925f0d39aaaSBarry Smith           ibdiag  += 9;
2926f0d39aaaSBarry Smith           break;
2927f0d39aaaSBarry Smith         case 4:
2928f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2929f0d39aaaSBarry Smith           v3   = a->a + ii[row+2];
2930f0d39aaaSBarry Smith           v4   = a->a + ii[row+3];
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           for (n = 0; n<sz-1; n+=2) {
2936f0d39aaaSBarry Smith             i1    = idx[0];
2937f0d39aaaSBarry Smith             i2    = idx[1];
2938f0d39aaaSBarry Smith             idx  += 2;
2939f0d39aaaSBarry Smith             tmp0  = x[i1];
2940f0d39aaaSBarry Smith             tmp1  = x[i2];
2941f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2942f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2943f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2944f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
2945f0d39aaaSBarry Smith           }
2946f0d39aaaSBarry Smith 
2947f0d39aaaSBarry Smith           if (n == sz-1) {
2948f0d39aaaSBarry Smith             tmp0  = x[*idx];
2949f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2950f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2951f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0;
2952f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0;
2953f0d39aaaSBarry Smith           }
29545850ef23SBarry Smith           t[row]   = sum1;
29555850ef23SBarry Smith           t[row+1] = sum2;
29565850ef23SBarry Smith           t[row+2] = sum3;
29575850ef23SBarry Smith           t[row+3] = sum4;
2958f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
2959f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
2960f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
2961f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
2962f0d39aaaSBarry Smith           ibdiag  += 16;
2963f0d39aaaSBarry Smith           break;
2964f0d39aaaSBarry Smith         case 5:
2965f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2966f0d39aaaSBarry Smith           v3   = a->a + ii[row+2];
2967f0d39aaaSBarry Smith           v4   = a->a + ii[row+3];
2968f0d39aaaSBarry Smith           v5   = a->a + ii[row+4];
2969f0d39aaaSBarry Smith           sum1 = b[row];
2970f0d39aaaSBarry Smith           sum2 = b[row+1];
2971f0d39aaaSBarry Smith           sum3 = b[row+2];
2972f0d39aaaSBarry Smith           sum4 = b[row+3];
2973f0d39aaaSBarry Smith           sum5 = b[row+4];
2974f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2975f0d39aaaSBarry Smith             i1    = idx[0];
2976f0d39aaaSBarry Smith             i2    = idx[1];
2977f0d39aaaSBarry Smith             idx  += 2;
2978f0d39aaaSBarry Smith             tmp0  = x[i1];
2979f0d39aaaSBarry Smith             tmp1  = x[i2];
2980f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2981f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2982f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2983f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
2984f0d39aaaSBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
2985f0d39aaaSBarry Smith           }
2986f0d39aaaSBarry Smith 
2987f0d39aaaSBarry Smith           if (n == sz-1) {
2988f0d39aaaSBarry Smith             tmp0  = x[*idx];
2989f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2990f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2991f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0;
2992f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0;
2993f0d39aaaSBarry Smith             sum5 -= v5[0] * tmp0;
2994f0d39aaaSBarry Smith           }
29955850ef23SBarry Smith           t[row]   = sum1;
29965850ef23SBarry Smith           t[row+1] = sum2;
29975850ef23SBarry Smith           t[row+2] = sum3;
29985850ef23SBarry Smith           t[row+3] = sum4;
29995850ef23SBarry Smith           t[row+4] = sum5;
3000f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
3001f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
3002f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
3003f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
3004f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
3005f0d39aaaSBarry Smith           ibdiag  += 25;
3006f0d39aaaSBarry Smith           break;
30078862d2efSBarry Smith         default:
3008e32f2f54SBarry Smith           SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
30098862d2efSBarry Smith         }
30102af78befSBarry Smith       }
30112af78befSBarry Smith 
30125850ef23SBarry Smith       xb   = t;
30132af78befSBarry Smith       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
30142af78befSBarry Smith     } else xb = b;
30152af78befSBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) {
30162af78befSBarry Smith 
3017f0d39aaaSBarry Smith       ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
3018d0f46423SBarry Smith       for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
3019f0d39aaaSBarry Smith         ibdiag -= sizes[i]*sizes[i];
30208862d2efSBarry Smith         sz      = ii[row+1] - diag[row] - 1;
30218862d2efSBarry Smith         v1      = a->a + diag[row] + 1;
30228862d2efSBarry Smith         idx     = a->j + diag[row] + 1;
30232af78befSBarry Smith 
30244108e4d5SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
30258862d2efSBarry Smith         switch (sizes[i]) {
30268862d2efSBarry Smith         case 1:
30278862d2efSBarry Smith 
30288862d2efSBarry Smith           sum1 = xb[row];
30298862d2efSBarry Smith           for (n = 0; n<sz-1; n+=2) {
30308862d2efSBarry Smith             i1    = idx[0];
30318862d2efSBarry Smith             i2    = idx[1];
30328862d2efSBarry Smith             idx  += 2;
30338862d2efSBarry Smith             tmp0  = x[i1];
30348862d2efSBarry Smith             tmp1  = x[i2];
30358862d2efSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
30368862d2efSBarry Smith           }
30378862d2efSBarry Smith 
30388862d2efSBarry Smith           if (n == sz-1) {
3039f0d39aaaSBarry Smith             tmp0  = x[*idx];
3040f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
30418862d2efSBarry Smith           }
3042f0d39aaaSBarry Smith           x[row--] = sum1*(*ibdiag);
3043f0d39aaaSBarry Smith           break;
3044f0d39aaaSBarry Smith 
3045f0d39aaaSBarry Smith         case 2:
3046f0d39aaaSBarry Smith 
3047f0d39aaaSBarry Smith           sum1 = xb[row];
3048f0d39aaaSBarry Smith           sum2 = xb[row-1];
3049f0d39aaaSBarry Smith           /* note that sum1 is associated with the second of the two rows */
3050f0d39aaaSBarry Smith           v2 = a->a + diag[row-1] + 2;
3051f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3052f0d39aaaSBarry Smith             i1    = idx[0];
3053f0d39aaaSBarry Smith             i2    = idx[1];
3054f0d39aaaSBarry Smith             idx  += 2;
3055f0d39aaaSBarry Smith             tmp0  = x[i1];
3056f0d39aaaSBarry Smith             tmp1  = x[i2];
3057f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3058f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3059f0d39aaaSBarry Smith           }
3060f0d39aaaSBarry Smith 
3061f0d39aaaSBarry Smith           if (n == sz-1) {
3062f0d39aaaSBarry Smith             tmp0  = x[*idx];
3063f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3064f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3065f0d39aaaSBarry Smith           }
3066f0d39aaaSBarry Smith           x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3];
3067f0d39aaaSBarry Smith           x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2];
3068f0d39aaaSBarry Smith           break;
3069f0d39aaaSBarry Smith         case 3:
3070f0d39aaaSBarry Smith 
3071f0d39aaaSBarry Smith           sum1 = xb[row];
3072f0d39aaaSBarry Smith           sum2 = xb[row-1];
3073f0d39aaaSBarry Smith           sum3 = xb[row-2];
3074f0d39aaaSBarry Smith           v2   = a->a + diag[row-1] + 2;
3075f0d39aaaSBarry Smith           v3   = a->a + diag[row-2] + 3;
3076f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3077f0d39aaaSBarry Smith             i1    = idx[0];
3078f0d39aaaSBarry Smith             i2    = idx[1];
3079f0d39aaaSBarry Smith             idx  += 2;
3080f0d39aaaSBarry Smith             tmp0  = x[i1];
3081f0d39aaaSBarry Smith             tmp1  = x[i2];
3082f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3083f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3084f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3085f0d39aaaSBarry Smith           }
3086f0d39aaaSBarry Smith 
3087f0d39aaaSBarry Smith           if (n == sz-1) {
3088f0d39aaaSBarry Smith             tmp0  = x[*idx];
3089f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3090f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3091f0d39aaaSBarry Smith             sum3 -= *v3*tmp0;
3092f0d39aaaSBarry Smith           }
3093f0d39aaaSBarry Smith           x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3094f0d39aaaSBarry Smith           x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3095f0d39aaaSBarry Smith           x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3096f0d39aaaSBarry Smith           break;
3097f0d39aaaSBarry Smith         case 4:
3098f0d39aaaSBarry Smith 
3099f0d39aaaSBarry Smith           sum1 = xb[row];
3100f0d39aaaSBarry Smith           sum2 = xb[row-1];
3101f0d39aaaSBarry Smith           sum3 = xb[row-2];
3102f0d39aaaSBarry Smith           sum4 = xb[row-3];
3103f0d39aaaSBarry Smith           v2   = a->a + diag[row-1] + 2;
3104f0d39aaaSBarry Smith           v3   = a->a + diag[row-2] + 3;
3105f0d39aaaSBarry Smith           v4   = a->a + diag[row-3] + 4;
3106f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3107f0d39aaaSBarry Smith             i1    = idx[0];
3108f0d39aaaSBarry Smith             i2    = idx[1];
3109f0d39aaaSBarry Smith             idx  += 2;
3110f0d39aaaSBarry Smith             tmp0  = x[i1];
3111f0d39aaaSBarry Smith             tmp1  = x[i2];
3112f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3113f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3114f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3115f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3116f0d39aaaSBarry Smith           }
3117f0d39aaaSBarry Smith 
3118f0d39aaaSBarry Smith           if (n == sz-1) {
3119f0d39aaaSBarry Smith             tmp0  = x[*idx];
3120f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3121f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3122f0d39aaaSBarry Smith             sum3 -= *v3*tmp0;
3123f0d39aaaSBarry Smith             sum4 -= *v4*tmp0;
3124f0d39aaaSBarry Smith           }
3125f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3126f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3127f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3128f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3129f0d39aaaSBarry Smith           break;
3130f0d39aaaSBarry Smith         case 5:
3131f0d39aaaSBarry Smith 
3132f0d39aaaSBarry Smith           sum1 = xb[row];
3133f0d39aaaSBarry Smith           sum2 = xb[row-1];
3134f0d39aaaSBarry Smith           sum3 = xb[row-2];
3135f0d39aaaSBarry Smith           sum4 = xb[row-3];
3136f0d39aaaSBarry Smith           sum5 = xb[row-4];
3137f0d39aaaSBarry Smith           v2   = a->a + diag[row-1] + 2;
3138f0d39aaaSBarry Smith           v3   = a->a + diag[row-2] + 3;
3139f0d39aaaSBarry Smith           v4   = a->a + diag[row-3] + 4;
3140f0d39aaaSBarry Smith           v5   = a->a + diag[row-4] + 5;
3141f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3142f0d39aaaSBarry Smith             i1    = idx[0];
3143f0d39aaaSBarry Smith             i2    = idx[1];
3144f0d39aaaSBarry Smith             idx  += 2;
3145f0d39aaaSBarry Smith             tmp0  = x[i1];
3146f0d39aaaSBarry Smith             tmp1  = x[i2];
3147f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3148f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3149f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3150f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3151f0d39aaaSBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
3152f0d39aaaSBarry Smith           }
3153f0d39aaaSBarry Smith 
3154f0d39aaaSBarry Smith           if (n == sz-1) {
3155f0d39aaaSBarry Smith             tmp0  = x[*idx];
3156f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3157f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3158f0d39aaaSBarry Smith             sum3 -= *v3*tmp0;
3159f0d39aaaSBarry Smith             sum4 -= *v4*tmp0;
3160f0d39aaaSBarry Smith             sum5 -= *v5*tmp0;
3161f0d39aaaSBarry Smith           }
3162f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
3163f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
3164f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
3165f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
3166f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
31678862d2efSBarry Smith           break;
31688862d2efSBarry Smith         default:
3169e32f2f54SBarry Smith           SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
31708862d2efSBarry Smith         }
31712af78befSBarry Smith       }
31722af78befSBarry Smith 
31732af78befSBarry Smith       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
31742af78befSBarry Smith     }
31752af78befSBarry Smith     its--;
31765850ef23SBarry Smith   }
31775850ef23SBarry Smith   while (its--) {
31785850ef23SBarry Smith 
31795850ef23SBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) {
3180d876e2b0SMark Adams       for (i=0, row=0, ibdiag = a->inode.ibdiag;
3181d876e2b0SMark Adams            i<m;
3182d876e2b0SMark Adams            row += sizes[i], ibdiag += sizes[i]*sizes[i], i++) {
31835850ef23SBarry Smith 
3184d876e2b0SMark Adams         sz  = diag[row] - ii[row];
31855850ef23SBarry Smith         v1  = a->a + ii[row];
31865850ef23SBarry Smith         idx = a->j + ii[row];
31875850ef23SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
31885850ef23SBarry Smith         switch (sizes[i]) {
31895850ef23SBarry Smith         case 1:
31905850ef23SBarry Smith           sum1 = b[row];
31915850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
31925850ef23SBarry Smith             i1    = idx[0];
31935850ef23SBarry Smith             i2    = idx[1];
31945850ef23SBarry Smith             idx  += 2;
31955850ef23SBarry Smith             tmp0  = x[i1];
31965850ef23SBarry Smith             tmp1  = x[i2];
31975850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
31985850ef23SBarry Smith           }
31995850ef23SBarry Smith           if (n == sz-1) {
3200d876e2b0SMark Adams             tmp0  = x[*idx++];
3201d876e2b0SMark Adams             sum1 -= *v1 * tmp0;
3202d876e2b0SMark Adams             v1++;
3203d876e2b0SMark Adams           }
3204d876e2b0SMark Adams           t[row]   = sum1;
3205d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 1;
3206d876e2b0SMark Adams           idx     = a->j + diag[row] + 1;
3207d876e2b0SMark Adams           v1 += 1;
3208d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3209d876e2b0SMark Adams             i1    = idx[0];
3210d876e2b0SMark Adams             i2    = idx[1];
3211d876e2b0SMark Adams             idx  += 2;
3212d876e2b0SMark Adams             tmp0  = x[i1];
3213d876e2b0SMark Adams             tmp1  = x[i2];
3214d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3215d876e2b0SMark Adams           }
3216d876e2b0SMark Adams           if (n == sz-1) {
3217d876e2b0SMark Adams             tmp0  = x[*idx++];
32185850ef23SBarry Smith             sum1 -= *v1 * tmp0;
32195850ef23SBarry Smith           }
32205850ef23SBarry Smith           /* in MatSOR_SeqAIJ this line would be
32215850ef23SBarry Smith            *
32225850ef23SBarry Smith            * x[row] = (1-omega)*x[row]+(sum1+(*bdiag++)*x[row])*(*ibdiag++);
32235850ef23SBarry Smith            *
32245850ef23SBarry Smith            * but omega == 1, so this becomes
32255850ef23SBarry Smith            *
3226d876e2b0SMark Adams            * x[row] = sum1*(*ibdiag++);
32275850ef23SBarry Smith            *
32285850ef23SBarry Smith            */
3229d876e2b0SMark Adams           x[row] = sum1*(*ibdiag);
32305850ef23SBarry Smith           break;
32315850ef23SBarry Smith         case 2:
32325850ef23SBarry Smith           v2   = a->a + ii[row+1];
32335850ef23SBarry Smith           sum1 = b[row];
32345850ef23SBarry Smith           sum2 = b[row+1];
32355850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
32365850ef23SBarry Smith             i1    = idx[0];
32375850ef23SBarry Smith             i2    = idx[1];
32385850ef23SBarry Smith             idx  += 2;
32395850ef23SBarry Smith             tmp0  = x[i1];
32405850ef23SBarry Smith             tmp1  = x[i2];
32415850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
32425850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
32435850ef23SBarry Smith           }
3244d876e2b0SMark Adams           if (n == sz-1) {
3245d876e2b0SMark Adams             tmp0  = x[*idx++];
3246d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3247d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3248d876e2b0SMark Adams             v1++; v2++;
3249d876e2b0SMark Adams           }
3250d876e2b0SMark Adams           t[row]   = sum1;
3251d876e2b0SMark Adams           t[row+1] = sum2;
3252d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 2;
3253d876e2b0SMark Adams           idx     = a->j + diag[row] + 2;
3254d876e2b0SMark Adams           v1 += 2;
3255d876e2b0SMark Adams           v2 += 2;
3256d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3257d876e2b0SMark Adams             i1    = idx[0];
3258d876e2b0SMark Adams             i2    = idx[1];
3259d876e2b0SMark Adams             idx  += 2;
3260d876e2b0SMark Adams             tmp0  = x[i1];
3261d876e2b0SMark Adams             tmp1  = x[i2];
3262d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3263d876e2b0SMark Adams             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3264d876e2b0SMark Adams           }
32655850ef23SBarry Smith           if (n == sz-1) {
32665850ef23SBarry Smith             tmp0  = x[*idx];
32675850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
32685850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
32695850ef23SBarry Smith           }
3270d876e2b0SMark Adams           x[row] = sum1*ibdiag[0] + sum2*ibdiag[2];
3271d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3];
32725850ef23SBarry Smith           break;
32735850ef23SBarry Smith         case 3:
32745850ef23SBarry Smith           v2   = a->a + ii[row+1];
32755850ef23SBarry Smith           v3   = a->a + ii[row+2];
32765850ef23SBarry Smith           sum1 = b[row];
32775850ef23SBarry Smith           sum2 = b[row+1];
32785850ef23SBarry Smith           sum3 = b[row+2];
32795850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
32805850ef23SBarry Smith             i1    = idx[0];
32815850ef23SBarry Smith             i2    = idx[1];
32825850ef23SBarry Smith             idx  += 2;
32835850ef23SBarry Smith             tmp0  = x[i1];
32845850ef23SBarry Smith             tmp1  = x[i2];
32855850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
32865850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
32875850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
32885850ef23SBarry Smith           }
3289d876e2b0SMark Adams           if (n == sz-1) {
3290d876e2b0SMark Adams             tmp0  = x[*idx++];
3291d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3292d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3293d876e2b0SMark Adams             sum3 -= v3[0] * tmp0;
3294d876e2b0SMark Adams             v1++; v2++; v3++;
3295d876e2b0SMark Adams           }
3296d876e2b0SMark Adams           t[row]   = sum1;
3297d876e2b0SMark Adams           t[row+1] = sum2;
3298d876e2b0SMark Adams           t[row+2] = sum3;
3299d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 3;
3300d876e2b0SMark Adams           idx     = a->j + diag[row] + 3;
3301d876e2b0SMark Adams           v1 += 3;
3302d876e2b0SMark Adams           v2 += 3;
3303d876e2b0SMark Adams           v3 += 3;
3304d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3305d876e2b0SMark Adams             i1    = idx[0];
3306d876e2b0SMark Adams             i2    = idx[1];
3307d876e2b0SMark Adams             idx  += 2;
3308d876e2b0SMark Adams             tmp0  = x[i1];
3309d876e2b0SMark Adams             tmp1  = x[i2];
3310d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3311d876e2b0SMark Adams             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3312d876e2b0SMark Adams             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3313d876e2b0SMark Adams           }
33145850ef23SBarry Smith           if (n == sz-1) {
33155850ef23SBarry Smith             tmp0  = x[*idx];
33165850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
33175850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
33185850ef23SBarry Smith             sum3 -= v3[0] * tmp0;
33195850ef23SBarry Smith           }
3320d876e2b0SMark Adams           x[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
3321d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
3322d876e2b0SMark Adams           x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
33235850ef23SBarry Smith           break;
33245850ef23SBarry Smith         case 4:
33255850ef23SBarry Smith           v2   = a->a + ii[row+1];
33265850ef23SBarry Smith           v3   = a->a + ii[row+2];
33275850ef23SBarry Smith           v4   = a->a + ii[row+3];
33285850ef23SBarry Smith           sum1 = b[row];
33295850ef23SBarry Smith           sum2 = b[row+1];
33305850ef23SBarry Smith           sum3 = b[row+2];
33315850ef23SBarry Smith           sum4 = b[row+3];
33325850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
33335850ef23SBarry Smith             i1    = idx[0];
33345850ef23SBarry Smith             i2    = idx[1];
33355850ef23SBarry Smith             idx  += 2;
33365850ef23SBarry Smith             tmp0  = x[i1];
33375850ef23SBarry Smith             tmp1  = x[i2];
33385850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
33395850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
33405850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
33415850ef23SBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
33425850ef23SBarry Smith           }
3343d876e2b0SMark Adams           if (n == sz-1) {
3344d876e2b0SMark Adams             tmp0  = x[*idx++];
3345d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3346d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3347d876e2b0SMark Adams             sum3 -= v3[0] * tmp0;
3348d876e2b0SMark Adams             sum4 -= v4[0] * tmp0;
3349d876e2b0SMark Adams             v1++; v2++; v3++; v4++;
3350d876e2b0SMark Adams           }
3351d876e2b0SMark Adams           t[row]   = sum1;
3352d876e2b0SMark Adams           t[row+1] = sum2;
3353d876e2b0SMark Adams           t[row+2] = sum3;
3354d876e2b0SMark Adams           t[row+3] = sum4;
3355d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 4;
3356d876e2b0SMark Adams           idx     = a->j + diag[row] + 4;
3357d876e2b0SMark Adams           v1 += 4;
3358d876e2b0SMark Adams           v2 += 4;
3359d876e2b0SMark Adams           v3 += 4;
3360d876e2b0SMark Adams           v4 += 4;
3361d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3362d876e2b0SMark Adams             i1    = idx[0];
3363d876e2b0SMark Adams             i2    = idx[1];
3364d876e2b0SMark Adams             idx  += 2;
3365d876e2b0SMark Adams             tmp0  = x[i1];
3366d876e2b0SMark Adams             tmp1  = x[i2];
3367d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3368d876e2b0SMark Adams             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3369d876e2b0SMark Adams             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3370d876e2b0SMark Adams             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3371d876e2b0SMark Adams           }
33725850ef23SBarry Smith           if (n == sz-1) {
33735850ef23SBarry Smith             tmp0  = x[*idx];
33745850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
33755850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
33765850ef23SBarry Smith             sum3 -= v3[0] * tmp0;
33775850ef23SBarry Smith             sum4 -= v4[0] * tmp0;
33785850ef23SBarry Smith           }
3379d876e2b0SMark Adams           x[row] =   sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
3380d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
3381d876e2b0SMark Adams           x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
3382d876e2b0SMark Adams           x[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
33835850ef23SBarry Smith           break;
33845850ef23SBarry Smith         case 5:
33855850ef23SBarry Smith           v2   = a->a + ii[row+1];
33865850ef23SBarry Smith           v3   = a->a + ii[row+2];
33875850ef23SBarry Smith           v4   = a->a + ii[row+3];
33885850ef23SBarry Smith           v5   = a->a + ii[row+4];
33895850ef23SBarry Smith           sum1 = b[row];
33905850ef23SBarry Smith           sum2 = b[row+1];
33915850ef23SBarry Smith           sum3 = b[row+2];
33925850ef23SBarry Smith           sum4 = b[row+3];
33935850ef23SBarry Smith           sum5 = b[row+4];
33945850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
33955850ef23SBarry Smith             i1    = idx[0];
33965850ef23SBarry Smith             i2    = idx[1];
33975850ef23SBarry Smith             idx  += 2;
33985850ef23SBarry Smith             tmp0  = x[i1];
33995850ef23SBarry Smith             tmp1  = x[i2];
34005850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
34015850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
34025850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
34035850ef23SBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
34045850ef23SBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
34055850ef23SBarry Smith           }
34065850ef23SBarry Smith           if (n == sz-1) {
3407d876e2b0SMark Adams             tmp0  = x[*idx++];
34085850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
34095850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
34105850ef23SBarry Smith             sum3 -= v3[0] * tmp0;
34115850ef23SBarry Smith             sum4 -= v4[0] * tmp0;
34125850ef23SBarry Smith             sum5 -= v5[0] * tmp0;
3413d876e2b0SMark Adams             v1++; v2++; v3++; v4++; v5++;
34145850ef23SBarry Smith           }
3415d876e2b0SMark Adams           t[row]   = sum1;
3416d876e2b0SMark Adams           t[row+1] = sum2;
3417d876e2b0SMark Adams           t[row+2] = sum3;
3418d876e2b0SMark Adams           t[row+3] = sum4;
3419d876e2b0SMark Adams           t[row+4] = sum5;
3420d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 5;
3421d876e2b0SMark Adams           idx     = a->j + diag[row] + 5;
3422d876e2b0SMark Adams           v1 += 5;
3423d876e2b0SMark Adams           v2 += 5;
3424d876e2b0SMark Adams           v3 += 5;
3425d876e2b0SMark Adams           v4 += 5;
3426d876e2b0SMark Adams           v5 += 5;
34275850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
34285850ef23SBarry Smith             i1    = idx[0];
34295850ef23SBarry Smith             i2    = idx[1];
34305850ef23SBarry Smith             idx  += 2;
34315850ef23SBarry Smith             tmp0  = x[i1];
34325850ef23SBarry Smith             tmp1  = x[i2];
34335850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
34345850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
34355850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
34365850ef23SBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
34375850ef23SBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
34385850ef23SBarry Smith           }
34395850ef23SBarry Smith           if (n == sz-1) {
34405850ef23SBarry Smith             tmp0  = x[*idx];
3441d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3442d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3443d876e2b0SMark Adams             sum3 -= v3[0] * tmp0;
3444d876e2b0SMark Adams             sum4 -= v4[0] * tmp0;
3445d876e2b0SMark Adams             sum5 -= v5[0] * tmp0;
34465850ef23SBarry Smith           }
3447d876e2b0SMark Adams           x[row]   = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
3448d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
3449d876e2b0SMark Adams           x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
3450d876e2b0SMark Adams           x[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
3451d876e2b0SMark Adams           x[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
3452d876e2b0SMark Adams           break;
3453d876e2b0SMark Adams         default:
3454d876e2b0SMark Adams           SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
3455d876e2b0SMark Adams         }
3456d876e2b0SMark Adams       }
3457d876e2b0SMark Adams       xb   = t;
3458d876e2b0SMark Adams       ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);  /* undercounts diag inverse */
3459d876e2b0SMark Adams     } else xb = b;
3460d876e2b0SMark Adams 
3461d876e2b0SMark Adams     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) {
3462d876e2b0SMark Adams 
3463d876e2b0SMark Adams       ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
3464d876e2b0SMark Adams       for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
3465d876e2b0SMark Adams         ibdiag -= sizes[i]*sizes[i];
3466d876e2b0SMark Adams 
3467d876e2b0SMark Adams         /* set RHS */
3468d876e2b0SMark Adams         if (xb == b) {
3469d876e2b0SMark Adams           /* whole (old way) */
3470d876e2b0SMark Adams           sz      = ii[row+1] - ii[row];
3471d876e2b0SMark Adams           idx     = a->j + ii[row];
3472d876e2b0SMark Adams           switch (sizes[i]) {
3473d876e2b0SMark Adams           case 5:
3474d876e2b0SMark Adams             v5      = a->a + ii[row-4];
3475d876e2b0SMark Adams           case 4: /* fall through */
3476d876e2b0SMark Adams             v4      = a->a + ii[row-3];
3477d876e2b0SMark Adams           case 3:
3478d876e2b0SMark Adams             v3      = a->a + ii[row-2];
3479d876e2b0SMark Adams           case 2:
3480d876e2b0SMark Adams             v2      = a->a + ii[row-1];
3481d876e2b0SMark Adams           case 1:
3482d876e2b0SMark Adams             v1      = a->a + ii[row];
3483d876e2b0SMark Adams             break;
3484d876e2b0SMark Adams           default:
3485d876e2b0SMark Adams             SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
3486d876e2b0SMark Adams           }
3487d876e2b0SMark Adams         } else {
3488d876e2b0SMark Adams           /* upper, no diag */
3489d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 1;
3490d876e2b0SMark Adams           idx     = a->j + diag[row] + 1;
3491d876e2b0SMark Adams           switch (sizes[i]) {
3492d876e2b0SMark Adams           case 5:
3493d876e2b0SMark Adams             v5      = a->a + diag[row-4] + 5;
3494d876e2b0SMark Adams           case 4: /* fall through */
3495d876e2b0SMark Adams             v4      = a->a + diag[row-3] + 4;
3496d876e2b0SMark Adams           case 3:
3497d876e2b0SMark Adams             v3      = a->a + diag[row-2] + 3;
3498d876e2b0SMark Adams           case 2:
3499d876e2b0SMark Adams             v2      = a->a + diag[row-1] + 2;
3500d876e2b0SMark Adams           case 1:
3501d876e2b0SMark Adams             v1      = a->a + diag[row] + 1;
3502d876e2b0SMark Adams           }
3503d876e2b0SMark Adams         }
3504d876e2b0SMark Adams         /* set sum */
3505d876e2b0SMark Adams         switch (sizes[i]) {
3506d876e2b0SMark Adams         case 5:
3507d876e2b0SMark Adams           sum5 = xb[row-4];
3508d876e2b0SMark Adams         case 4: /* fall through */
3509d876e2b0SMark Adams           sum4 = xb[row-3];
3510d876e2b0SMark Adams         case 3:
3511d876e2b0SMark Adams           sum3 = xb[row-2];
3512d876e2b0SMark Adams         case 2:
3513d876e2b0SMark Adams           sum2 = xb[row-1];
3514d876e2b0SMark Adams         case 1:
3515d876e2b0SMark Adams           /* note that sum1 is associated with the last row */
3516d876e2b0SMark Adams           sum1 = xb[row];
3517d876e2b0SMark Adams         }
3518d876e2b0SMark Adams         /* do sums */
3519d876e2b0SMark Adams         for (n = 0; n<sz-1; n+=2) {
3520d876e2b0SMark Adams             i1    = idx[0];
3521d876e2b0SMark Adams             i2    = idx[1];
3522d876e2b0SMark Adams             idx  += 2;
3523d876e2b0SMark Adams             tmp0  = x[i1];
3524d876e2b0SMark Adams             tmp1  = x[i2];
3525d876e2b0SMark Adams             switch (sizes[i]) {
3526d876e2b0SMark Adams             case 5:
3527d876e2b0SMark Adams               sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
3528d876e2b0SMark Adams             case 4: /* fall through */
3529d876e2b0SMark Adams               sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3530d876e2b0SMark Adams             case 3:
3531d876e2b0SMark Adams               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3532d876e2b0SMark Adams             case 2:
3533d876e2b0SMark Adams               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3534d876e2b0SMark Adams             case 1:
3535d876e2b0SMark Adams               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3536d876e2b0SMark Adams             }
3537d876e2b0SMark Adams         }
3538d876e2b0SMark Adams         /* ragged edge */
3539d876e2b0SMark Adams         if (n == sz-1) {
3540d876e2b0SMark Adams           tmp0  = x[*idx];
3541d876e2b0SMark Adams           switch (sizes[i]) {
3542d876e2b0SMark Adams           case 5:
3543d876e2b0SMark Adams             sum5 -= *v5*tmp0;
3544d876e2b0SMark Adams           case 4: /* fall through */
3545d876e2b0SMark Adams             sum4 -= *v4*tmp0;
3546d876e2b0SMark Adams           case 3:
3547d876e2b0SMark Adams             sum3 -= *v3*tmp0;
3548d876e2b0SMark Adams           case 2:
3549d876e2b0SMark Adams             sum2 -= *v2*tmp0;
3550d876e2b0SMark Adams           case 1:
3551d876e2b0SMark Adams             sum1 -= *v1*tmp0;
3552d876e2b0SMark Adams           }
3553d876e2b0SMark Adams         }
3554d876e2b0SMark Adams         /* update */
3555d876e2b0SMark Adams         if (xb == b) {
3556d876e2b0SMark Adams           /* whole (old way) w/ diag */
3557d876e2b0SMark Adams           switch (sizes[i]) {
3558d876e2b0SMark Adams           case 5:
35595850ef23SBarry Smith             x[row--] += sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
35605850ef23SBarry Smith             x[row--] += sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
35615850ef23SBarry Smith             x[row--] += sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
35625850ef23SBarry Smith             x[row--] += sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
35635850ef23SBarry Smith             x[row--] += sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
35645850ef23SBarry Smith             break;
3565d876e2b0SMark Adams           case 4:
3566d876e2b0SMark Adams             x[row--] += sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3567d876e2b0SMark Adams             x[row--] += sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3568d876e2b0SMark Adams             x[row--] += sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3569d876e2b0SMark Adams             x[row--] += sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3570d876e2b0SMark Adams             break;
3571d876e2b0SMark Adams           case 3:
3572d876e2b0SMark Adams             x[row--] += sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3573d876e2b0SMark Adams             x[row--] += sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3574d876e2b0SMark Adams             x[row--] += sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3575d876e2b0SMark Adams             break;
3576d876e2b0SMark Adams           case 2:
3577d876e2b0SMark Adams             x[row--] += sum2*ibdiag[1] + sum1*ibdiag[3];
3578d876e2b0SMark Adams             x[row--] += sum2*ibdiag[0] + sum1*ibdiag[2];
3579d876e2b0SMark Adams             break;
3580d876e2b0SMark Adams           case 1:
3581d876e2b0SMark Adams             x[row--] += sum1*(*ibdiag);
3582d876e2b0SMark Adams             break;
3583d876e2b0SMark Adams           }
3584d876e2b0SMark Adams         } else {
3585d876e2b0SMark Adams           /* no diag so set =  */
3586d876e2b0SMark Adams           switch (sizes[i]) {
3587d876e2b0SMark Adams           case 5:
3588d876e2b0SMark Adams             x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
3589d876e2b0SMark Adams             x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
3590d876e2b0SMark Adams             x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
3591d876e2b0SMark Adams             x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
3592d876e2b0SMark Adams             x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
3593d876e2b0SMark Adams             break;
3594d876e2b0SMark Adams           case 4:
3595d876e2b0SMark Adams             x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3596d876e2b0SMark Adams             x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3597d876e2b0SMark Adams             x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3598d876e2b0SMark Adams             x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3599d876e2b0SMark Adams             break;
3600d876e2b0SMark Adams           case 3:
3601d876e2b0SMark Adams             x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3602d876e2b0SMark Adams             x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3603d876e2b0SMark Adams             x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3604d876e2b0SMark Adams             break;
3605d876e2b0SMark Adams           case 2:
3606d876e2b0SMark Adams             x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3];
3607d876e2b0SMark Adams             x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2];
3608d876e2b0SMark Adams             break;
3609d876e2b0SMark Adams           case 1:
3610d876e2b0SMark Adams             x[row--] = sum1*(*ibdiag);
3611d876e2b0SMark Adams             break;
36125850ef23SBarry Smith           }
36135850ef23SBarry Smith         }
3614d876e2b0SMark Adams       }
3615d876e2b0SMark Adams       if (xb == b) {
36165850ef23SBarry Smith         ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
3617d876e2b0SMark Adams       } else {
3618d876e2b0SMark Adams         ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper, undercounts diag inverse */
3619d876e2b0SMark Adams       }
36205850ef23SBarry Smith     }
36212af78befSBarry Smith   }
362289c6957cSBarry Smith   if (flag & SOR_EISENSTAT) {
362389c6957cSBarry Smith     /*
362489c6957cSBarry Smith           Apply  (U + D)^-1  where D is now the block diagonal
362589c6957cSBarry Smith     */
362689c6957cSBarry Smith     ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
362789c6957cSBarry Smith     for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
362889c6957cSBarry Smith       ibdiag -= sizes[i]*sizes[i];
362989c6957cSBarry Smith       sz      = ii[row+1] - diag[row] - 1;
363089c6957cSBarry Smith       v1      = a->a + diag[row] + 1;
363189c6957cSBarry Smith       idx     = a->j + diag[row] + 1;
36324108e4d5SBarry Smith       /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
363389c6957cSBarry Smith       switch (sizes[i]) {
363489c6957cSBarry Smith       case 1:
363589c6957cSBarry Smith 
363689c6957cSBarry Smith         sum1 = b[row];
363789c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
363889c6957cSBarry Smith           i1    = idx[0];
363989c6957cSBarry Smith           i2    = idx[1];
364089c6957cSBarry Smith           idx  += 2;
364189c6957cSBarry Smith           tmp0  = x[i1];
364289c6957cSBarry Smith           tmp1  = x[i2];
364389c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
364489c6957cSBarry Smith         }
364589c6957cSBarry Smith 
364689c6957cSBarry Smith         if (n == sz-1) {
364789c6957cSBarry Smith           tmp0  = x[*idx];
364889c6957cSBarry Smith           sum1 -= *v1*tmp0;
364989c6957cSBarry Smith         }
3650938d4eb3SBarry Smith         x[row] = sum1*(*ibdiag);row--;
365189c6957cSBarry Smith         break;
365289c6957cSBarry Smith 
365389c6957cSBarry Smith       case 2:
365489c6957cSBarry Smith 
365589c6957cSBarry Smith         sum1 = b[row];
365689c6957cSBarry Smith         sum2 = b[row-1];
365789c6957cSBarry Smith         /* note that sum1 is associated with the second of the two rows */
365889c6957cSBarry Smith         v2 = a->a + diag[row-1] + 2;
365989c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
366089c6957cSBarry Smith           i1    = idx[0];
366189c6957cSBarry Smith           i2    = idx[1];
366289c6957cSBarry Smith           idx  += 2;
366389c6957cSBarry Smith           tmp0  = x[i1];
366489c6957cSBarry Smith           tmp1  = x[i2];
366589c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
366689c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
366789c6957cSBarry Smith         }
366889c6957cSBarry Smith 
366989c6957cSBarry Smith         if (n == sz-1) {
367089c6957cSBarry Smith           tmp0  = x[*idx];
367189c6957cSBarry Smith           sum1 -= *v1*tmp0;
367289c6957cSBarry Smith           sum2 -= *v2*tmp0;
367389c6957cSBarry Smith         }
3674938d4eb3SBarry Smith         x[row]   = sum2*ibdiag[1] + sum1*ibdiag[3];
3675938d4eb3SBarry Smith         x[row-1] = sum2*ibdiag[0] + sum1*ibdiag[2];
3676938d4eb3SBarry Smith         row     -= 2;
367789c6957cSBarry Smith         break;
367889c6957cSBarry Smith       case 3:
367989c6957cSBarry Smith 
368089c6957cSBarry Smith         sum1 = b[row];
368189c6957cSBarry Smith         sum2 = b[row-1];
368289c6957cSBarry Smith         sum3 = b[row-2];
368389c6957cSBarry Smith         v2   = a->a + diag[row-1] + 2;
368489c6957cSBarry Smith         v3   = a->a + diag[row-2] + 3;
368589c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
368689c6957cSBarry Smith           i1    = idx[0];
368789c6957cSBarry Smith           i2    = idx[1];
368889c6957cSBarry Smith           idx  += 2;
368989c6957cSBarry Smith           tmp0  = x[i1];
369089c6957cSBarry Smith           tmp1  = x[i2];
369189c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
369289c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
369389c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
369489c6957cSBarry Smith         }
369589c6957cSBarry Smith 
369689c6957cSBarry Smith         if (n == sz-1) {
369789c6957cSBarry Smith           tmp0  = x[*idx];
369889c6957cSBarry Smith           sum1 -= *v1*tmp0;
369989c6957cSBarry Smith           sum2 -= *v2*tmp0;
370089c6957cSBarry Smith           sum3 -= *v3*tmp0;
370189c6957cSBarry Smith         }
3702938d4eb3SBarry Smith         x[row]   = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3703938d4eb3SBarry Smith         x[row-1] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3704938d4eb3SBarry Smith         x[row-2] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3705938d4eb3SBarry Smith         row     -= 3;
370689c6957cSBarry Smith         break;
370789c6957cSBarry Smith       case 4:
370889c6957cSBarry Smith 
370989c6957cSBarry Smith         sum1 = b[row];
371089c6957cSBarry Smith         sum2 = b[row-1];
371189c6957cSBarry Smith         sum3 = b[row-2];
371289c6957cSBarry Smith         sum4 = b[row-3];
371389c6957cSBarry Smith         v2   = a->a + diag[row-1] + 2;
371489c6957cSBarry Smith         v3   = a->a + diag[row-2] + 3;
371589c6957cSBarry Smith         v4   = a->a + diag[row-3] + 4;
371689c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
371789c6957cSBarry Smith           i1    = idx[0];
371889c6957cSBarry Smith           i2    = idx[1];
371989c6957cSBarry Smith           idx  += 2;
372089c6957cSBarry Smith           tmp0  = x[i1];
372189c6957cSBarry Smith           tmp1  = x[i2];
372289c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
372389c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
372489c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
372589c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
372689c6957cSBarry Smith         }
372789c6957cSBarry Smith 
372889c6957cSBarry Smith         if (n == sz-1) {
372989c6957cSBarry Smith           tmp0  = x[*idx];
373089c6957cSBarry Smith           sum1 -= *v1*tmp0;
373189c6957cSBarry Smith           sum2 -= *v2*tmp0;
373289c6957cSBarry Smith           sum3 -= *v3*tmp0;
373389c6957cSBarry Smith           sum4 -= *v4*tmp0;
373489c6957cSBarry Smith         }
3735938d4eb3SBarry Smith         x[row]   = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3736938d4eb3SBarry Smith         x[row-1] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3737938d4eb3SBarry Smith         x[row-2] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3738938d4eb3SBarry Smith         x[row-3] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3739938d4eb3SBarry Smith         row     -= 4;
374089c6957cSBarry Smith         break;
374189c6957cSBarry Smith       case 5:
374289c6957cSBarry Smith 
374389c6957cSBarry Smith         sum1 = b[row];
374489c6957cSBarry Smith         sum2 = b[row-1];
374589c6957cSBarry Smith         sum3 = b[row-2];
374689c6957cSBarry Smith         sum4 = b[row-3];
374789c6957cSBarry Smith         sum5 = b[row-4];
374889c6957cSBarry Smith         v2   = a->a + diag[row-1] + 2;
374989c6957cSBarry Smith         v3   = a->a + diag[row-2] + 3;
375089c6957cSBarry Smith         v4   = a->a + diag[row-3] + 4;
375189c6957cSBarry Smith         v5   = a->a + diag[row-4] + 5;
375289c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
375389c6957cSBarry Smith           i1    = idx[0];
375489c6957cSBarry Smith           i2    = idx[1];
375589c6957cSBarry Smith           idx  += 2;
375689c6957cSBarry Smith           tmp0  = x[i1];
375789c6957cSBarry Smith           tmp1  = x[i2];
375889c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
375989c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
376089c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
376189c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
376289c6957cSBarry Smith           sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
376389c6957cSBarry Smith         }
376489c6957cSBarry Smith 
376589c6957cSBarry Smith         if (n == sz-1) {
376689c6957cSBarry Smith           tmp0  = x[*idx];
376789c6957cSBarry Smith           sum1 -= *v1*tmp0;
376889c6957cSBarry Smith           sum2 -= *v2*tmp0;
376989c6957cSBarry Smith           sum3 -= *v3*tmp0;
377089c6957cSBarry Smith           sum4 -= *v4*tmp0;
377189c6957cSBarry Smith           sum5 -= *v5*tmp0;
377289c6957cSBarry Smith         }
3773938d4eb3SBarry Smith         x[row]   = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
3774938d4eb3SBarry Smith         x[row-1] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
3775938d4eb3SBarry Smith         x[row-2] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
3776938d4eb3SBarry Smith         x[row-3] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
3777938d4eb3SBarry Smith         x[row-4] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
3778938d4eb3SBarry Smith         row     -= 5;
377989c6957cSBarry Smith         break;
378089c6957cSBarry Smith       default:
3781e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
378289c6957cSBarry Smith       }
378389c6957cSBarry Smith     }
378489c6957cSBarry Smith     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
378589c6957cSBarry Smith 
378689c6957cSBarry Smith     /*
378789c6957cSBarry Smith            t = b - D x    where D is the block diagonal
378889c6957cSBarry Smith     */
378989c6957cSBarry Smith     cnt = 0;
379089c6957cSBarry Smith     for (i=0, row=0; i<m; i++) {
379189c6957cSBarry Smith       switch (sizes[i]) {
379289c6957cSBarry Smith       case 1:
379389c6957cSBarry Smith         t[row] = b[row] - bdiag[cnt++]*x[row]; row++;
379489c6957cSBarry Smith         break;
379589c6957cSBarry Smith       case 2:
379689c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1];
379789c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+2];
379889c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+3];
379989c6957cSBarry Smith         t[row]   = b[row] - tmp1;
380089c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2; row += 2;
380189c6957cSBarry Smith         cnt     += 4;
380289c6957cSBarry Smith         break;
380389c6957cSBarry Smith       case 3:
380489c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1]; x3 = x[row+2];
380589c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6];
380689c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7];
380789c6957cSBarry Smith         tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8];
380889c6957cSBarry Smith         t[row]   = b[row] - tmp1;
380989c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2;
381089c6957cSBarry Smith         t[row+2] = b[row+2] - tmp3; row += 3;
381189c6957cSBarry Smith         cnt     += 9;
381289c6957cSBarry Smith         break;
381389c6957cSBarry Smith       case 4:
381489c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3];
381589c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12];
381689c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13];
381789c6957cSBarry Smith         tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14];
381889c6957cSBarry Smith         tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15];
381989c6957cSBarry Smith         t[row]   = b[row] - tmp1;
382089c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2;
382189c6957cSBarry Smith         t[row+2] = b[row+2] - tmp3;
382289c6957cSBarry Smith         t[row+3] = b[row+3] - tmp4; row += 4;
382389c6957cSBarry Smith         cnt     += 16;
382489c6957cSBarry Smith         break;
382589c6957cSBarry Smith       case 5:
382689c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; x5 = x[row+4];
382789c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20];
382889c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21];
382989c6957cSBarry Smith         tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22];
383089c6957cSBarry Smith         tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23];
383189c6957cSBarry Smith         tmp5     = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24];
383289c6957cSBarry Smith         t[row]   = b[row] - tmp1;
383389c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2;
383489c6957cSBarry Smith         t[row+2] = b[row+2] - tmp3;
383589c6957cSBarry Smith         t[row+3] = b[row+3] - tmp4;
383689c6957cSBarry Smith         t[row+4] = b[row+4] - tmp5;row += 5;
383789c6957cSBarry Smith         cnt     += 25;
383889c6957cSBarry Smith         break;
383989c6957cSBarry Smith       default:
3840e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
384189c6957cSBarry Smith       }
384289c6957cSBarry Smith     }
384389c6957cSBarry Smith     ierr = PetscLogFlops(m);CHKERRQ(ierr);
384489c6957cSBarry Smith 
384589c6957cSBarry Smith 
384689c6957cSBarry Smith 
384789c6957cSBarry Smith     /*
384889c6957cSBarry Smith           Apply (L + D)^-1 where D is the block diagonal
384989c6957cSBarry Smith     */
385089c6957cSBarry Smith     for (i=0, row=0; i<m; i++) {
385189c6957cSBarry Smith       sz  = diag[row] - ii[row];
385289c6957cSBarry Smith       v1  = a->a + ii[row];
385389c6957cSBarry Smith       idx = a->j + ii[row];
38544108e4d5SBarry Smith       /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
385589c6957cSBarry Smith       switch (sizes[i]) {
385689c6957cSBarry Smith       case 1:
385789c6957cSBarry Smith 
385889c6957cSBarry Smith         sum1 = t[row];
385989c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
386089c6957cSBarry Smith           i1    = idx[0];
386189c6957cSBarry Smith           i2    = idx[1];
386289c6957cSBarry Smith           idx  += 2;
386389c6957cSBarry Smith           tmp0  = t[i1];
386489c6957cSBarry Smith           tmp1  = t[i2];
386589c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
386689c6957cSBarry Smith         }
386789c6957cSBarry Smith 
386889c6957cSBarry Smith         if (n == sz-1) {
386989c6957cSBarry Smith           tmp0  = t[*idx];
387089c6957cSBarry Smith           sum1 -= *v1 * tmp0;
387189c6957cSBarry Smith         }
387289c6957cSBarry Smith         x[row] += t[row] = sum1*(*ibdiag++); row++;
387389c6957cSBarry Smith         break;
387489c6957cSBarry Smith       case 2:
387589c6957cSBarry Smith         v2   = a->a + ii[row+1];
387689c6957cSBarry Smith         sum1 = t[row];
387789c6957cSBarry Smith         sum2 = t[row+1];
387889c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
387989c6957cSBarry Smith           i1    = idx[0];
388089c6957cSBarry Smith           i2    = idx[1];
388189c6957cSBarry Smith           idx  += 2;
388289c6957cSBarry Smith           tmp0  = t[i1];
388389c6957cSBarry Smith           tmp1  = t[i2];
388489c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
388589c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
388689c6957cSBarry Smith         }
388789c6957cSBarry Smith 
388889c6957cSBarry Smith         if (n == sz-1) {
388989c6957cSBarry Smith           tmp0  = t[*idx];
389089c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
389189c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
389289c6957cSBarry Smith         }
389389c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[2];
389489c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3];
389589c6957cSBarry Smith         ibdiag   += 4; row += 2;
389689c6957cSBarry Smith         break;
389789c6957cSBarry Smith       case 3:
389889c6957cSBarry Smith         v2   = a->a + ii[row+1];
389989c6957cSBarry Smith         v3   = a->a + ii[row+2];
390089c6957cSBarry Smith         sum1 = t[row];
390189c6957cSBarry Smith         sum2 = t[row+1];
390289c6957cSBarry Smith         sum3 = t[row+2];
390389c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
390489c6957cSBarry Smith           i1    = idx[0];
390589c6957cSBarry Smith           i2    = idx[1];
390689c6957cSBarry Smith           idx  += 2;
390789c6957cSBarry Smith           tmp0  = t[i1];
390889c6957cSBarry Smith           tmp1  = t[i2];
390989c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
391089c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
391189c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
391289c6957cSBarry Smith         }
391389c6957cSBarry Smith 
391489c6957cSBarry Smith         if (n == sz-1) {
391589c6957cSBarry Smith           tmp0  = t[*idx];
391689c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
391789c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
391889c6957cSBarry Smith           sum3 -= v3[0] * tmp0;
391989c6957cSBarry Smith         }
392089c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
392189c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
392289c6957cSBarry Smith         x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
392389c6957cSBarry Smith         ibdiag   += 9; row += 3;
392489c6957cSBarry Smith         break;
392589c6957cSBarry Smith       case 4:
392689c6957cSBarry Smith         v2   = a->a + ii[row+1];
392789c6957cSBarry Smith         v3   = a->a + ii[row+2];
392889c6957cSBarry Smith         v4   = a->a + ii[row+3];
392989c6957cSBarry Smith         sum1 = t[row];
393089c6957cSBarry Smith         sum2 = t[row+1];
393189c6957cSBarry Smith         sum3 = t[row+2];
393289c6957cSBarry Smith         sum4 = t[row+3];
393389c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
393489c6957cSBarry Smith           i1    = idx[0];
393589c6957cSBarry Smith           i2    = idx[1];
393689c6957cSBarry Smith           idx  += 2;
393789c6957cSBarry Smith           tmp0  = t[i1];
393889c6957cSBarry Smith           tmp1  = t[i2];
393989c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
394089c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
394189c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
394289c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
394389c6957cSBarry Smith         }
394489c6957cSBarry Smith 
394589c6957cSBarry Smith         if (n == sz-1) {
394689c6957cSBarry Smith           tmp0  = t[*idx];
394789c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
394889c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
394989c6957cSBarry Smith           sum3 -= v3[0] * tmp0;
395089c6957cSBarry Smith           sum4 -= v4[0] * tmp0;
395189c6957cSBarry Smith         }
395289c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
395389c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
395489c6957cSBarry Smith         x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
395589c6957cSBarry Smith         x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
395689c6957cSBarry Smith         ibdiag   += 16; row += 4;
395789c6957cSBarry Smith         break;
395889c6957cSBarry Smith       case 5:
395989c6957cSBarry Smith         v2   = a->a + ii[row+1];
396089c6957cSBarry Smith         v3   = a->a + ii[row+2];
396189c6957cSBarry Smith         v4   = a->a + ii[row+3];
396289c6957cSBarry Smith         v5   = a->a + ii[row+4];
396389c6957cSBarry Smith         sum1 = t[row];
396489c6957cSBarry Smith         sum2 = t[row+1];
396589c6957cSBarry Smith         sum3 = t[row+2];
396689c6957cSBarry Smith         sum4 = t[row+3];
396789c6957cSBarry Smith         sum5 = t[row+4];
396889c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
396989c6957cSBarry Smith           i1    = idx[0];
397089c6957cSBarry Smith           i2    = idx[1];
397189c6957cSBarry Smith           idx  += 2;
397289c6957cSBarry Smith           tmp0  = t[i1];
397389c6957cSBarry Smith           tmp1  = t[i2];
397489c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
397589c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
397689c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
397789c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
397889c6957cSBarry Smith           sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
397989c6957cSBarry Smith         }
398089c6957cSBarry Smith 
398189c6957cSBarry Smith         if (n == sz-1) {
398289c6957cSBarry Smith           tmp0  = t[*idx];
398389c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
398489c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
398589c6957cSBarry Smith           sum3 -= v3[0] * tmp0;
398689c6957cSBarry Smith           sum4 -= v4[0] * tmp0;
398789c6957cSBarry Smith           sum5 -= v5[0] * tmp0;
398889c6957cSBarry Smith         }
398989c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
399089c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
399189c6957cSBarry Smith         x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
399289c6957cSBarry Smith         x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
399389c6957cSBarry Smith         x[row+4] += t[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
399489c6957cSBarry Smith         ibdiag   += 25; row += 5;
399589c6957cSBarry Smith         break;
399689c6957cSBarry Smith       default:
3997e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
399889c6957cSBarry Smith       }
399989c6957cSBarry Smith     }
400089c6957cSBarry Smith     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
40015850ef23SBarry Smith   }
40022af78befSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
40033649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
4004365a8a9eSBarry Smith   PetscFunctionReturn(0);
40052af78befSBarry Smith }
40062af78befSBarry Smith 
400789c6957cSBarry Smith #undef __FUNCT__
40084108e4d5SBarry Smith #define __FUNCT__ "MatMultDiagonalBlock_SeqAIJ_Inode"
40094108e4d5SBarry Smith PetscErrorCode MatMultDiagonalBlock_SeqAIJ_Inode(Mat A,Vec bb,Vec xx)
401089c6957cSBarry Smith {
401189c6957cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
401289c6957cSBarry Smith   PetscScalar       *x,tmp1,tmp2,tmp3,tmp4,tmp5,x1,x2,x3,x4,x5;
401389c6957cSBarry Smith   const MatScalar   *bdiag = a->inode.bdiag;
401489c6957cSBarry Smith   const PetscScalar *b;
401589c6957cSBarry Smith   PetscErrorCode    ierr;
401689c6957cSBarry Smith   PetscInt          m      = a->inode.node_count,cnt = 0,i,row;
401789c6957cSBarry Smith   const PetscInt    *sizes = a->inode.size;
40182af78befSBarry Smith 
401989c6957cSBarry Smith   PetscFunctionBegin;
402089c6957cSBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
40213649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
402289c6957cSBarry Smith   cnt  = 0;
402389c6957cSBarry Smith   for (i=0, row=0; i<m; i++) {
402489c6957cSBarry Smith     switch (sizes[i]) {
402589c6957cSBarry Smith     case 1:
402689c6957cSBarry Smith       x[row] = b[row]*bdiag[cnt++];row++;
402789c6957cSBarry Smith       break;
402889c6957cSBarry Smith     case 2:
402989c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1];
403089c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+2];
403189c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+3];
403289c6957cSBarry Smith       x[row++] = tmp1;
403389c6957cSBarry Smith       x[row++] = tmp2;
403489c6957cSBarry Smith       cnt     += 4;
403589c6957cSBarry Smith       break;
403689c6957cSBarry Smith     case 3:
403789c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1]; x3 = b[row+2];
403889c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6];
403989c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7];
404089c6957cSBarry Smith       tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8];
404189c6957cSBarry Smith       x[row++] = tmp1;
404289c6957cSBarry Smith       x[row++] = tmp2;
404389c6957cSBarry Smith       x[row++] = tmp3;
404489c6957cSBarry Smith       cnt     += 9;
404589c6957cSBarry Smith       break;
404689c6957cSBarry Smith     case 4:
404789c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3];
404889c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12];
404989c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13];
405089c6957cSBarry Smith       tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14];
405189c6957cSBarry Smith       tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15];
405289c6957cSBarry Smith       x[row++] = tmp1;
405389c6957cSBarry Smith       x[row++] = tmp2;
405489c6957cSBarry Smith       x[row++] = tmp3;
405589c6957cSBarry Smith       x[row++] = tmp4;
405689c6957cSBarry Smith       cnt     += 16;
405789c6957cSBarry Smith       break;
405889c6957cSBarry Smith     case 5:
405989c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3]; x5 = b[row+4];
406089c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20];
406189c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21];
406289c6957cSBarry Smith       tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22];
406389c6957cSBarry Smith       tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23];
406489c6957cSBarry Smith       tmp5     = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24];
406589c6957cSBarry Smith       x[row++] = tmp1;
406689c6957cSBarry Smith       x[row++] = tmp2;
406789c6957cSBarry Smith       x[row++] = tmp3;
406889c6957cSBarry Smith       x[row++] = tmp4;
406989c6957cSBarry Smith       x[row++] = tmp5;
407089c6957cSBarry Smith       cnt     += 25;
407189c6957cSBarry Smith       break;
407289c6957cSBarry Smith     default:
4073e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
407489c6957cSBarry Smith     }
407589c6957cSBarry Smith   }
407689c6957cSBarry Smith   ierr = PetscLogFlops(2*cnt);CHKERRQ(ierr);
407789c6957cSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
40783649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
407989c6957cSBarry Smith   PetscFunctionReturn(0);
408089c6957cSBarry Smith }
408189c6957cSBarry Smith 
40824c1414c8SBarry Smith /*
40834c1414c8SBarry Smith     samestructure indicates that the matrix has not changed its nonzero structure so we
40844c1414c8SBarry Smith     do not need to recompute the inodes
40854c1414c8SBarry Smith */
40864c1414c8SBarry Smith #undef __FUNCT__
4087a02bda8eSBarry Smith #define __FUNCT__ "MatSeqAIJCheckInode"
4088a02bda8eSBarry Smith PetscErrorCode MatSeqAIJCheckInode(Mat A)
40894c1414c8SBarry Smith {
40904c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
40914c1414c8SBarry Smith   PetscErrorCode ierr;
40928758e1faSBarry Smith   PetscInt       i,j,m,nzx,nzy,*ns,node_count,blk_size;
4093ace3abfcSBarry Smith   PetscBool      flag;
40948758e1faSBarry Smith   const PetscInt *idx,*idy,*ii;
40954c1414c8SBarry Smith 
40964c1414c8SBarry Smith   PetscFunctionBegin;
4097d74fe821SBarry Smith   if (!a->inode.use) PetscFunctionReturn(0);
4098a02bda8eSBarry Smith   if (a->inode.checked && A->nonzerostate == a->inode.mat_nonzerostate) PetscFunctionReturn(0);
40994c1414c8SBarry Smith 
4100d0f46423SBarry Smith   m = A->rmap->n;
41012205254eSKarl Rupp   if (a->inode.size) ns = a->inode.size;
41022205254eSKarl Rupp   else {
4103854ce69bSBarry Smith     ierr = PetscMalloc1(m+1,&ns);CHKERRQ(ierr);
41042205254eSKarl Rupp   }
41054c1414c8SBarry Smith 
41064c1414c8SBarry Smith   i          = 0;
41074c1414c8SBarry Smith   node_count = 0;
41084c1414c8SBarry Smith   idx        = a->j;
41094c1414c8SBarry Smith   ii         = a->i;
41104c1414c8SBarry Smith   while (i < m) {                /* For each row */
41114c1414c8SBarry Smith     nzx = ii[i+1] - ii[i];       /* Number of nonzeros */
41124c1414c8SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
41134c1414c8SBarry Smith     for (j=i+1,idy=idx,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
41144c1414c8SBarry Smith       nzy = ii[j+1] - ii[j];     /* Same number of nonzeros */
41154c1414c8SBarry Smith       if (nzy != nzx) break;
41164c1414c8SBarry Smith       idy += nzx;              /* Same nonzero pattern */
41174c1414c8SBarry Smith       ierr = PetscMemcmp(idx,idy,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr);
41184c1414c8SBarry Smith       if (!flag) break;
41194c1414c8SBarry Smith     }
41204c1414c8SBarry Smith     ns[node_count++] = blk_size;
41214c1414c8SBarry Smith     idx             += blk_size*nzx;
41224c1414c8SBarry Smith     i                = j;
41234c1414c8SBarry Smith   }
41244c1414c8SBarry Smith   /* If not enough inodes found,, do not use inode version of the routines */
4125be6adb11SBarry Smith   if (!m || node_count > .8*m) {
41264c1414c8SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
41272205254eSKarl Rupp 
41284c1414c8SBarry Smith     a->inode.node_count       = 0;
41290298fd71SBarry Smith     a->inode.size             = NULL;
41304c1414c8SBarry Smith     a->inode.use              = PETSC_FALSE;
4131ded6e084SBarry Smith     A->ops->mult              = MatMult_SeqAIJ;
4132ded6e084SBarry Smith     A->ops->sor               = MatSOR_SeqAIJ;
4133ded6e084SBarry Smith     A->ops->multadd           = MatMultAdd_SeqAIJ;
4134ded6e084SBarry Smith     A->ops->getrowij          = MatGetRowIJ_SeqAIJ;
4135ded6e084SBarry Smith     A->ops->restorerowij      = MatRestoreRowIJ_SeqAIJ;
4136ded6e084SBarry Smith     A->ops->getcolumnij       = MatGetColumnIJ_SeqAIJ;
4137ded6e084SBarry Smith     A->ops->restorecolumnij   = MatRestoreColumnIJ_SeqAIJ;
4138ded6e084SBarry Smith     A->ops->coloringpatch     = 0;
4139ded6e084SBarry Smith     A->ops->multdiagonalblock = 0;
41402205254eSKarl Rupp 
41414c1414c8SBarry Smith     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
41424c1414c8SBarry Smith   } else {
4143d5f3da31SBarry Smith     if (!A->factortype) {
41444108e4d5SBarry Smith       A->ops->mult              = MatMult_SeqAIJ_Inode;
41454108e4d5SBarry Smith       A->ops->sor               = MatSOR_SeqAIJ_Inode;
41464108e4d5SBarry Smith       A->ops->multadd           = MatMultAdd_SeqAIJ_Inode;
4147375a6242SBarry Smith       A->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode;
4148375a6242SBarry Smith       if (A->rmap->n == A->cmap->n) {
41494108e4d5SBarry Smith         A->ops->getrowij          = MatGetRowIJ_SeqAIJ_Inode;
41504108e4d5SBarry Smith         A->ops->restorerowij      = MatRestoreRowIJ_SeqAIJ_Inode;
41514108e4d5SBarry Smith         A->ops->getcolumnij       = MatGetColumnIJ_SeqAIJ_Inode;
41524108e4d5SBarry Smith         A->ops->restorecolumnij   = MatRestoreColumnIJ_SeqAIJ_Inode;
41534108e4d5SBarry Smith         A->ops->coloringpatch     = MatColoringPatch_SeqAIJ_Inode;
4154375a6242SBarry Smith       }
4155d3ac4fa3SBarry Smith     } else {
4156d3ac4fa3SBarry Smith       A->ops->solve = MatSolve_SeqAIJ_Inode_inplace;
4157d3ac4fa3SBarry Smith     }
41584c1414c8SBarry Smith     a->inode.node_count = node_count;
41594c1414c8SBarry Smith     a->inode.size       = ns;
41604c1414c8SBarry Smith     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
41614c1414c8SBarry Smith   }
4162be6adb11SBarry Smith   a->inode.checked          = PETSC_TRUE;
4163a02bda8eSBarry Smith   a->inode.mat_nonzerostate = A->nonzerostate;
41644c1414c8SBarry Smith   PetscFunctionReturn(0);
41654c1414c8SBarry Smith }
41664c1414c8SBarry Smith 
416717454e89SShri Abhyankar #undef __FUNCT__
4168150f0143SBarry Smith #define __FUNCT__ "MatDuplicate_SeqAIJ_Inode"
4169150f0143SBarry Smith PetscErrorCode MatDuplicate_SeqAIJ_Inode(Mat A,MatDuplicateOption cpvalues,Mat *C)
4170150f0143SBarry Smith {
4171150f0143SBarry Smith   Mat            B =*C;
4172150f0143SBarry Smith   Mat_SeqAIJ     *c=(Mat_SeqAIJ*)B->data,*a=(Mat_SeqAIJ*)A->data;
4173150f0143SBarry Smith   PetscErrorCode ierr;
4174150f0143SBarry Smith   PetscInt       m=A->rmap->n;
4175150f0143SBarry Smith 
4176150f0143SBarry Smith   PetscFunctionBegin;
4177150f0143SBarry Smith   c->inode.use       = a->inode.use;
4178150f0143SBarry Smith   c->inode.limit     = a->inode.limit;
4179150f0143SBarry Smith   c->inode.max_limit = a->inode.max_limit;
4180150f0143SBarry Smith   if (a->inode.size) {
4181854ce69bSBarry Smith     ierr                = PetscMalloc1(m+1,&c->inode.size);CHKERRQ(ierr);
4182150f0143SBarry Smith     c->inode.node_count = a->inode.node_count;
4183150f0143SBarry Smith     ierr                = PetscMemcpy(c->inode.size,a->inode.size,(m+1)*sizeof(PetscInt));CHKERRQ(ierr);
4184a02bda8eSBarry Smith     /* note the table of functions below should match that in MatSeqAIJCheckInode() */
41852c451681SBarry Smith     if (!B->factortype) {
41862c451681SBarry Smith       B->ops->mult              = MatMult_SeqAIJ_Inode;
41872c451681SBarry Smith       B->ops->sor               = MatSOR_SeqAIJ_Inode;
41882c451681SBarry Smith       B->ops->multadd           = MatMultAdd_SeqAIJ_Inode;
41892c451681SBarry Smith       B->ops->getrowij          = MatGetRowIJ_SeqAIJ_Inode;
41902c451681SBarry Smith       B->ops->restorerowij      = MatRestoreRowIJ_SeqAIJ_Inode;
41912c451681SBarry Smith       B->ops->getcolumnij       = MatGetColumnIJ_SeqAIJ_Inode;
41922c451681SBarry Smith       B->ops->restorecolumnij   = MatRestoreColumnIJ_SeqAIJ_Inode;
41932c451681SBarry Smith       B->ops->coloringpatch     = MatColoringPatch_SeqAIJ_Inode;
41942c451681SBarry Smith       B->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode;
4195150f0143SBarry Smith     } else {
41962c451681SBarry Smith       B->ops->solve = MatSolve_SeqAIJ_Inode_inplace;
4197150f0143SBarry Smith     }
4198150f0143SBarry Smith   } else {
4199150f0143SBarry Smith     c->inode.size       = 0;
4200150f0143SBarry Smith     c->inode.node_count = 0;
4201150f0143SBarry Smith   }
4202150f0143SBarry Smith   c->inode.ibdiagvalid = PETSC_FALSE;
4203150f0143SBarry Smith   c->inode.ibdiag      = 0;
4204150f0143SBarry Smith   c->inode.bdiag       = 0;
4205150f0143SBarry Smith   PetscFunctionReturn(0);
4206150f0143SBarry Smith }
4207150f0143SBarry Smith 
4208150f0143SBarry Smith #undef __FUNCT__
420917454e89SShri Abhyankar #define __FUNCT__ "MatGetRow_FactoredLU"
42108758e1faSBarry 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)
421117454e89SShri Abhyankar {
42128758e1faSBarry Smith   PetscInt       k;
42138758e1faSBarry Smith   const PetscInt *vi;
42146e111a19SKarl Rupp 
421517454e89SShri Abhyankar   PetscFunctionBegin;
421617454e89SShri Abhyankar   vi = aj + ai[row];
421717454e89SShri Abhyankar   for (k=0; k<nzl; k++) cols[k] = vi[k];
421817454e89SShri Abhyankar   vi        = aj + adiag[row];
421917454e89SShri Abhyankar   cols[nzl] = vi[0];
422017454e89SShri Abhyankar   vi        = aj + adiag[row+1]+1;
422117454e89SShri Abhyankar   for (k=0; k<nzu; k++) cols[nzl+1+k] = vi[k];
422217454e89SShri Abhyankar   PetscFunctionReturn(0);
422317454e89SShri Abhyankar }
42246936b636SHong Zhang /*
4225a02bda8eSBarry Smith    MatSeqAIJCheckInode_FactorLU - Check Inode for factored seqaij matrix.
4226a02bda8eSBarry Smith    Modified from MatSeqAIJCheckInode().
42276936b636SHong Zhang 
42286936b636SHong Zhang    Input Parameters:
4229abb87a52SBarry Smith .  Mat A - ILU or LU matrix factor
4230abb87a52SBarry Smith 
42316936b636SHong Zhang */
4232019b515eSShri Abhyankar #undef __FUNCT__
4233a02bda8eSBarry Smith #define __FUNCT__ "MatSeqAIJCheckInode_FactorLU"
4234a02bda8eSBarry Smith PetscErrorCode MatSeqAIJCheckInode_FactorLU(Mat A)
4235019b515eSShri Abhyankar {
4236019b515eSShri Abhyankar   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
4237019b515eSShri Abhyankar   PetscErrorCode ierr;
4238019b515eSShri Abhyankar   PetscInt       i,j,m,nzl1,nzu1,nzl2,nzu2,nzx,nzy,node_count,blk_size;
42398758e1faSBarry Smith   PetscInt       *cols1,*cols2,*ns;
42408758e1faSBarry Smith   const PetscInt *ai = a->i,*aj = a->j, *adiag = a->diag;
4241ace3abfcSBarry Smith   PetscBool      flag;
4242019b515eSShri Abhyankar 
4243019b515eSShri Abhyankar   PetscFunctionBegin;
4244019b515eSShri Abhyankar   if (!a->inode.use)    PetscFunctionReturn(0);
4245abb87a52SBarry Smith   if (a->inode.checked) PetscFunctionReturn(0);
4246019b515eSShri Abhyankar 
4247019b515eSShri Abhyankar   m = A->rmap->n;
42482205254eSKarl Rupp   if (a->inode.size) ns = a->inode.size;
42492205254eSKarl Rupp   else {
4250854ce69bSBarry Smith     ierr = PetscMalloc1(m+1,&ns);CHKERRQ(ierr);
42512205254eSKarl Rupp   }
4252019b515eSShri Abhyankar 
4253019b515eSShri Abhyankar   i          = 0;
4254019b515eSShri Abhyankar   node_count = 0;
4255dcca6d9dSJed Brown   ierr = PetscMalloc2(m,&cols1,m,&cols2);CHKERRQ(ierr);
4256019b515eSShri Abhyankar   while (i < m) {                /* For each row */
4257019b515eSShri Abhyankar     nzl1 = ai[i+1] - ai[i];       /* Number of nonzeros in L */
4258019b515eSShri Abhyankar     nzu1 = adiag[i] - adiag[i+1] - 1; /* Number of nonzeros in U excluding diagonal*/
4259019b515eSShri Abhyankar     nzx  = nzl1 + nzu1 + 1;
4260019b515eSShri Abhyankar     MatGetRow_FactoredLU(cols1,nzl1,nzu1,nzx,ai,aj,adiag,i);
4261019b515eSShri Abhyankar 
4262019b515eSShri Abhyankar     /* Limits the number of elements in a node to 'a->inode.limit' */
4263019b515eSShri Abhyankar     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
4264019b515eSShri Abhyankar       nzl2 = ai[j+1] - ai[j];
4265019b515eSShri Abhyankar       nzu2 = adiag[j] - adiag[j+1] - 1;
4266019b515eSShri Abhyankar       nzy  = nzl2 + nzu2 + 1;
4267019b515eSShri Abhyankar       if (nzy != nzx) break;
426817454e89SShri Abhyankar       ierr = MatGetRow_FactoredLU(cols2,nzl2,nzu2,nzy,ai,aj,adiag,j);CHKERRQ(ierr);
4269019b515eSShri Abhyankar       ierr = PetscMemcmp(cols1,cols2,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr);
42708758e1faSBarry Smith       if (!flag) break;
4271019b515eSShri Abhyankar     }
4272019b515eSShri Abhyankar     ns[node_count++] = blk_size;
4273019b515eSShri Abhyankar     i                = j;
4274019b515eSShri Abhyankar   }
42758758e1faSBarry Smith   ierr             = PetscFree2(cols1,cols2);CHKERRQ(ierr);
4276019b515eSShri Abhyankar   /* If not enough inodes found,, do not use inode version of the routines */
4277be6adb11SBarry Smith   if (!m || node_count > .8*m) {
4278019b515eSShri Abhyankar     ierr = PetscFree(ns);CHKERRQ(ierr);
42792205254eSKarl Rupp 
4280019b515eSShri Abhyankar     a->inode.node_count = 0;
42810298fd71SBarry Smith     a->inode.size       = NULL;
4282019b515eSShri Abhyankar     a->inode.use        = PETSC_FALSE;
42832205254eSKarl Rupp 
4284019b515eSShri Abhyankar     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
4285019b515eSShri Abhyankar   } else {
4286019b515eSShri Abhyankar     A->ops->mult              = 0;
4287019b515eSShri Abhyankar     A->ops->sor               = 0;
4288019b515eSShri Abhyankar     A->ops->multadd           = 0;
4289019b515eSShri Abhyankar     A->ops->getrowij          = 0;
4290019b515eSShri Abhyankar     A->ops->restorerowij      = 0;
4291019b515eSShri Abhyankar     A->ops->getcolumnij       = 0;
4292019b515eSShri Abhyankar     A->ops->restorecolumnij   = 0;
4293019b515eSShri Abhyankar     A->ops->coloringpatch     = 0;
4294019b515eSShri Abhyankar     A->ops->multdiagonalblock = 0;
4295019b515eSShri Abhyankar     a->inode.node_count       = node_count;
4296019b515eSShri Abhyankar     a->inode.size             = ns;
42972205254eSKarl Rupp 
4298019b515eSShri Abhyankar     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
4299019b515eSShri Abhyankar   }
4300be6adb11SBarry Smith   a->inode.checked = PETSC_TRUE;
4301019b515eSShri Abhyankar   PetscFunctionReturn(0);
4302019b515eSShri Abhyankar }
4303019b515eSShri Abhyankar 
4304acf2f550SJed Brown #undef __FUNCT__
4305acf2f550SJed Brown #define __FUNCT__ "MatSeqAIJInvalidateDiagonal_Inode"
4306acf2f550SJed Brown PetscErrorCode MatSeqAIJInvalidateDiagonal_Inode(Mat A)
4307acf2f550SJed Brown {
4308acf2f550SJed Brown   Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data;
4309acf2f550SJed Brown 
4310acf2f550SJed Brown   PetscFunctionBegin;
4311acf2f550SJed Brown   a->inode.ibdiagvalid = PETSC_FALSE;
4312acf2f550SJed Brown   PetscFunctionReturn(0);
4313acf2f550SJed Brown }
4314acf2f550SJed Brown 
43154c1414c8SBarry Smith /*
43164c1414c8SBarry Smith      This is really ugly. if inodes are used this replaces the
43174c1414c8SBarry Smith   permutations with ones that correspond to rows/cols of the matrix
43184c1414c8SBarry Smith   rather then inode blocks
43194c1414c8SBarry Smith */
43204c1414c8SBarry Smith #undef __FUNCT__
43214c1414c8SBarry Smith #define __FUNCT__ "MatInodeAdjustForInodes"
43227087cfbeSBarry Smith PetscErrorCode  MatInodeAdjustForInodes(Mat A,IS *rperm,IS *cperm)
43234c1414c8SBarry Smith {
43244ac538c5SBarry Smith   PetscErrorCode ierr;
43254c1414c8SBarry Smith 
43264c1414c8SBarry Smith   PetscFunctionBegin;
43274ac538c5SBarry Smith   ierr = PetscTryMethod(A,"MatInodeAdjustForInodes_C",(Mat,IS*,IS*),(A,rperm,cperm));CHKERRQ(ierr);
43284c1414c8SBarry Smith   PetscFunctionReturn(0);
43294c1414c8SBarry Smith }
43304c1414c8SBarry Smith 
43314c1414c8SBarry Smith #undef __FUNCT__
433253c77d0aSJed Brown #define __FUNCT__ "MatInodeAdjustForInodes_SeqAIJ_Inode"
43337087cfbeSBarry Smith PetscErrorCode  MatInodeAdjustForInodes_SeqAIJ_Inode(Mat A,IS *rperm,IS *cperm)
43344c1414c8SBarry Smith {
43354c1414c8SBarry Smith   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
43364c1414c8SBarry Smith   PetscErrorCode ierr;
43375d0c19d7SBarry Smith   PetscInt       m = A->rmap->n,n = A->cmap->n,i,j,nslim_row = a->inode.node_count;
43385d0c19d7SBarry Smith   const PetscInt *ridx,*cidx;
43394c1414c8SBarry Smith   PetscInt       row,col,*permr,*permc,*ns_row =  a->inode.size,*tns,start_val,end_val,indx;
43404c1414c8SBarry Smith   PetscInt       nslim_col,*ns_col;
43414c1414c8SBarry Smith   IS             ris = *rperm,cis = *cperm;
43424c1414c8SBarry Smith 
43434c1414c8SBarry Smith   PetscFunctionBegin;
43444c1414c8SBarry Smith   if (!a->inode.size) PetscFunctionReturn(0); /* no inodes so return */
43454c1414c8SBarry Smith   if (a->inode.node_count == m) PetscFunctionReturn(0); /* all inodes are of size 1 */
43464c1414c8SBarry Smith 
43474c1414c8SBarry Smith   ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr);
4348854ce69bSBarry Smith   ierr = PetscMalloc1(((nslim_row>nslim_col) ? nslim_row : nslim_col)+1,&tns);CHKERRQ(ierr);
4349dcca6d9dSJed Brown   ierr = PetscMalloc2(m,&permr,n,&permc);CHKERRQ(ierr);
43504c1414c8SBarry Smith 
43514c1414c8SBarry Smith   ierr = ISGetIndices(ris,&ridx);CHKERRQ(ierr);
43524c1414c8SBarry Smith   ierr = ISGetIndices(cis,&cidx);CHKERRQ(ierr);
43534c1414c8SBarry Smith 
43544c1414c8SBarry Smith   /* Form the inode structure for the rows of permuted matric using inv perm*/
43554c1414c8SBarry Smith   for (i=0,tns[0]=0; i<nslim_row; ++i) tns[i+1] = tns[i] + ns_row[i];
43564c1414c8SBarry Smith 
43574c1414c8SBarry Smith   /* Construct the permutations for rows*/
43584c1414c8SBarry Smith   for (i=0,row = 0; i<nslim_row; ++i) {
43594c1414c8SBarry Smith     indx      = ridx[i];
43604c1414c8SBarry Smith     start_val = tns[indx];
43614c1414c8SBarry Smith     end_val   = tns[indx + 1];
43624c1414c8SBarry Smith     for (j=start_val; j<end_val; ++j,++row) permr[row]= j;
43634c1414c8SBarry Smith   }
43644c1414c8SBarry Smith 
43654c1414c8SBarry Smith   /* Form the inode structure for the columns of permuted matrix using inv perm*/
43664c1414c8SBarry Smith   for (i=0,tns[0]=0; i<nslim_col; ++i) tns[i+1] = tns[i] + ns_col[i];
43674c1414c8SBarry Smith 
43684c1414c8SBarry Smith   /* Construct permutations for columns */
43694c1414c8SBarry Smith   for (i=0,col=0; i<nslim_col; ++i) {
43704c1414c8SBarry Smith     indx      = cidx[i];
43714c1414c8SBarry Smith     start_val = tns[indx];
43724c1414c8SBarry Smith     end_val   = tns[indx + 1];
43734c1414c8SBarry Smith     for (j = start_val; j<end_val; ++j,++col) permc[col]= j;
43744c1414c8SBarry Smith   }
43754c1414c8SBarry Smith 
437670b3c8c7SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permr,PETSC_COPY_VALUES,rperm);CHKERRQ(ierr);
43770e83c824SBarry Smith   ierr = ISSetPermutation(*rperm);CHKERRQ(ierr);
437870b3c8c7SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permc,PETSC_COPY_VALUES,cperm);CHKERRQ(ierr);
43790e83c824SBarry Smith   ierr = ISSetPermutation(*cperm);CHKERRQ(ierr);
43804c1414c8SBarry Smith 
43814c1414c8SBarry Smith   ierr = ISRestoreIndices(ris,&ridx);CHKERRQ(ierr);
43824c1414c8SBarry Smith   ierr = ISRestoreIndices(cis,&cidx);CHKERRQ(ierr);
43834c1414c8SBarry Smith 
43844c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
43850e83c824SBarry Smith   ierr = PetscFree2(permr,permc);CHKERRQ(ierr);
43866bf464f9SBarry Smith   ierr = ISDestroy(&cis);CHKERRQ(ierr);
43876bf464f9SBarry Smith   ierr = ISDestroy(&ris);CHKERRQ(ierr);
43884c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
43894c1414c8SBarry Smith   PetscFunctionReturn(0);
43904c1414c8SBarry Smith }
43914c1414c8SBarry Smith 
43924c1414c8SBarry Smith #undef __FUNCT__
43934c1414c8SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes"
43944c1414c8SBarry Smith /*@C
43954c1414c8SBarry Smith    MatInodeGetInodeSizes - Returns the inode information of the Inode matrix.
43964c1414c8SBarry Smith 
43973f9fe445SBarry Smith    Not Collective
43984c1414c8SBarry Smith 
43994c1414c8SBarry Smith    Input Parameter:
44004c1414c8SBarry Smith .  A - the Inode matrix or matrix derived from the Inode class -- e.g., SeqAIJ
44014c1414c8SBarry Smith 
44024c1414c8SBarry Smith    Output Parameter:
44034c1414c8SBarry Smith +  node_count - no of inodes present in the matrix.
44044c1414c8SBarry Smith .  sizes      - an array of size node_count,with sizes of each inode.
44054c1414c8SBarry Smith -  limit      - the max size used to generate the inodes.
44064c1414c8SBarry Smith 
44074c1414c8SBarry Smith    Level: advanced
44084c1414c8SBarry Smith 
44094c1414c8SBarry Smith    Notes: This routine returns some internal storage information
44104c1414c8SBarry Smith    of the matrix, it is intended to be used by advanced users.
44114c1414c8SBarry Smith    It should be called after the matrix is assembled.
44124c1414c8SBarry Smith    The contents of the sizes[] array should not be changed.
44130298fd71SBarry Smith    NULL may be passed for information not requested.
44144c1414c8SBarry Smith 
44154c1414c8SBarry Smith .keywords: matrix, seqaij, get, inode
44164c1414c8SBarry Smith 
44174c1414c8SBarry Smith .seealso: MatGetInfo()
44184c1414c8SBarry Smith @*/
44197087cfbeSBarry Smith PetscErrorCode  MatInodeGetInodeSizes(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit)
44204c1414c8SBarry Smith {
44214c1414c8SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscInt*,PetscInt*[],PetscInt*);
44224c1414c8SBarry Smith 
44234c1414c8SBarry Smith   PetscFunctionBegin;
4424e32f2f54SBarry Smith   if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
44250005d66cSJed Brown   ierr = PetscObjectQueryFunction((PetscObject)A,"MatInodeGetInodeSizes_C",&f);CHKERRQ(ierr);
44264c1414c8SBarry Smith   if (f) {
44274c1414c8SBarry Smith     ierr = (*f)(A,node_count,sizes,limit);CHKERRQ(ierr);
44284c1414c8SBarry Smith   }
44294c1414c8SBarry Smith   PetscFunctionReturn(0);
44304c1414c8SBarry Smith }
44314c1414c8SBarry Smith 
44324c1414c8SBarry Smith #undef __FUNCT__
44334108e4d5SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes_SeqAIJ_Inode"
44347087cfbeSBarry Smith PetscErrorCode  MatInodeGetInodeSizes_SeqAIJ_Inode(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit)
44354c1414c8SBarry Smith {
44364c1414c8SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
44374c1414c8SBarry Smith 
44384c1414c8SBarry Smith   PetscFunctionBegin;
44394c1414c8SBarry Smith   if (node_count) *node_count = a->inode.node_count;
44404c1414c8SBarry Smith   if (sizes)      *sizes      = a->inode.size;
44414c1414c8SBarry Smith   if (limit)      *limit      = a->inode.limit;
44424c1414c8SBarry Smith   PetscFunctionReturn(0);
44434c1414c8SBarry Smith }
4444