xref: /petsc/src/mat/impls/aij/seq/inode.c (revision 019b515eed920573838ad77cd5e92e3da9dcf8f5)
14c1414c8SBarry Smith #define PETSCMAT_DLL
24c1414c8SBarry Smith 
34c1414c8SBarry Smith /*
44c1414c8SBarry Smith   This file provides high performance routines for the Inode format (compressed sparse row)
54c1414c8SBarry Smith   by taking advantage of rows with identical nonzero structure (I-nodes).
64c1414c8SBarry Smith */
77c4f633dSBarry Smith #include "../src/mat/impls/aij/seq/aij.h"
84c1414c8SBarry Smith 
94c1414c8SBarry Smith #undef __FUNCT__
104c1414c8SBarry Smith #define __FUNCT__ "Mat_CreateColInode"
114c1414c8SBarry Smith static PetscErrorCode Mat_CreateColInode(Mat A,PetscInt* size,PetscInt ** ns)
124c1414c8SBarry Smith {
134c1414c8SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
144c1414c8SBarry Smith   PetscErrorCode ierr;
154c1414c8SBarry Smith   PetscInt       i,count,m,n,min_mn,*ns_row,*ns_col;
164c1414c8SBarry Smith 
174c1414c8SBarry Smith   PetscFunctionBegin;
18d0f46423SBarry Smith   n      = A->cmap->n;
19d0f46423SBarry Smith   m      = A->rmap->n;
204c1414c8SBarry Smith   ns_row = a->inode.size;
214c1414c8SBarry Smith 
224c1414c8SBarry Smith   min_mn = (m < n) ? m : n;
234c1414c8SBarry Smith   if (!ns) {
244c1414c8SBarry Smith     for (count=0,i=0; count<min_mn; count+=ns_row[i],i++);
254c1414c8SBarry Smith     for(; count+1 < n; count++,i++);
264c1414c8SBarry Smith     if (count < n)  {
274c1414c8SBarry Smith       i++;
284c1414c8SBarry Smith     }
294c1414c8SBarry Smith     *size = i;
304c1414c8SBarry Smith     PetscFunctionReturn(0);
314c1414c8SBarry Smith   }
324c1414c8SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&ns_col);CHKERRQ(ierr);
334c1414c8SBarry Smith 
344c1414c8SBarry Smith   /* Use the same row structure wherever feasible. */
354c1414c8SBarry Smith   for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) {
364c1414c8SBarry Smith     ns_col[i] = ns_row[i];
374c1414c8SBarry Smith   }
384c1414c8SBarry Smith 
394c1414c8SBarry Smith   /* if m < n; pad up the remainder with inode_limit */
404c1414c8SBarry Smith   for(; count+1 < n; count++,i++) {
414c1414c8SBarry Smith     ns_col[i] = 1;
424c1414c8SBarry Smith   }
434c1414c8SBarry Smith   /* The last node is the odd ball. padd it up with the remaining rows; */
444c1414c8SBarry Smith   if (count < n)  {
454c1414c8SBarry Smith     ns_col[i] = n - count;
464c1414c8SBarry Smith     i++;
474c1414c8SBarry Smith   } else if (count > n) {
484c1414c8SBarry Smith     /* Adjust for the over estimation */
494c1414c8SBarry Smith     ns_col[i-1] += n - count;
504c1414c8SBarry Smith   }
514c1414c8SBarry Smith   *size = i;
524c1414c8SBarry Smith   *ns   = ns_col;
534c1414c8SBarry Smith   PetscFunctionReturn(0);
544c1414c8SBarry Smith }
554c1414c8SBarry Smith 
564c1414c8SBarry Smith 
574c1414c8SBarry Smith /*
584c1414c8SBarry Smith       This builds symmetric version of nonzero structure,
594c1414c8SBarry Smith */
604c1414c8SBarry Smith #undef __FUNCT__
614108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode_Symmetric"
624108e4d5SBarry Smith static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Symmetric(Mat A,PetscInt *iia[],PetscInt *jja[],PetscInt ishift,PetscInt oshift)
634c1414c8SBarry Smith {
644c1414c8SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
654c1414c8SBarry Smith   PetscErrorCode ierr;
664c1414c8SBarry Smith   PetscInt       *work,*ia,*ja,*j,nz,nslim_row,nslim_col,m,row,col,*jmax,n;
674c1414c8SBarry Smith   PetscInt       *tns,*tvc,*ns_row = a->inode.size,*ns_col,nsz,i1,i2,*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;
734108e4d5SBarry Smith   if (m != n) SETERRQ(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 */
804c1414c8SBarry Smith   ierr = PetscMalloc((nslim_col+1)*sizeof(PetscInt),&tns);CHKERRQ(ierr);
814c1414c8SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&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];
864c1414c8SBarry Smith     for (i2=0; i2<nsz; ++i2,++col)
874c1414c8SBarry Smith       tvc[col] = i1;
884c1414c8SBarry Smith   }
894c1414c8SBarry Smith   /* allocate space for row pointers */
904c1414c8SBarry Smith   ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&ia);CHKERRQ(ierr);
914c1414c8SBarry Smith   *iia = ia;
924c1414c8SBarry Smith   ierr = PetscMemzero(ia,(nslim_row+1)*sizeof(PetscInt));CHKERRQ(ierr);
934c1414c8SBarry Smith   ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&work);CHKERRQ(ierr);
944c1414c8SBarry Smith 
954c1414c8SBarry Smith   /* determine the number of columns in each row */
964c1414c8SBarry Smith   ia[0] = oshift;
974c1414c8SBarry Smith   for (i1=0,row=0 ; i1<nslim_row; row+=ns_row[i1],i1++) {
984c1414c8SBarry Smith 
994c1414c8SBarry Smith     j    = aj + ai[row] + ishift;
1004c1414c8SBarry Smith     jmax = aj + ai[row+1] + ishift;
1014c1414c8SBarry Smith     i2   = 0;
1024c1414c8SBarry Smith     col  = *j++ + ishift;
1034c1414c8SBarry Smith     i2   = tvc[col];
1044c1414c8SBarry Smith     while (i2<i1 && j<jmax) { /* 1.[-xx-d-xx--] 2.[-xx-------],off-diagonal elemets */
1054c1414c8SBarry Smith       ia[i1+1]++;
1064c1414c8SBarry Smith       ia[i2+1]++;
1074c1414c8SBarry Smith       i2++;                     /* Start col of next node */
1084c1414c8SBarry Smith       while(((col=*j+ishift)<tns[i2]) && (j<jmax)) ++j;
1094c1414c8SBarry Smith       i2 = tvc[col];
1104c1414c8SBarry Smith     }
1114c1414c8SBarry Smith     if(i2 == i1) ia[i2+1]++;    /* now the diagonal element */
1124c1414c8SBarry Smith   }
1134c1414c8SBarry Smith 
1144c1414c8SBarry Smith   /* shift ia[i] to point to next row */
1154c1414c8SBarry Smith   for (i1=1; i1<nslim_row+1; i1++) {
1164c1414c8SBarry Smith     row        = ia[i1-1];
1174c1414c8SBarry Smith     ia[i1]    += row;
1184c1414c8SBarry Smith     work[i1-1] = row - oshift;
1194c1414c8SBarry Smith   }
1204c1414c8SBarry Smith 
1214c1414c8SBarry Smith   /* allocate space for column pointers */
1224c1414c8SBarry Smith   nz   = ia[nslim_row] + (!ishift);
1234c1414c8SBarry Smith   ierr = PetscMalloc(nz*sizeof(PetscInt),&ja);CHKERRQ(ierr);
1244c1414c8SBarry Smith   *jja = ja;
1254c1414c8SBarry Smith 
1264c1414c8SBarry Smith  /* loop over lower triangular part putting into ja */
1274c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row += ns_row[i1],i1++) {
1284c1414c8SBarry Smith     j    = aj + ai[row] + ishift;
1294c1414c8SBarry Smith     jmax = aj + ai[row+1] + ishift;
1304c1414c8SBarry Smith     i2   = 0;                     /* Col inode index */
1314c1414c8SBarry Smith     col  = *j++ + ishift;
1324c1414c8SBarry Smith     i2   = tvc[col];
1334c1414c8SBarry Smith     while (i2<i1 && j<jmax) {
1344c1414c8SBarry Smith       ja[work[i2]++] = i1 + oshift;
1354c1414c8SBarry Smith       ja[work[i1]++] = i2 + oshift;
1364c1414c8SBarry Smith       ++i2;
1374c1414c8SBarry Smith       while(((col=*j+ishift)< tns[i2])&&(j<jmax)) ++j; /* Skip rest col indices in this node */
1384c1414c8SBarry Smith       i2 = tvc[col];
1394c1414c8SBarry Smith     }
1404c1414c8SBarry Smith     if (i2 == i1) ja[work[i1]++] = i2 + oshift;
1414c1414c8SBarry Smith 
1424c1414c8SBarry Smith   }
1434c1414c8SBarry Smith   ierr = PetscFree(work);CHKERRQ(ierr);
1444c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
1454c1414c8SBarry Smith   ierr = PetscFree(tvc);CHKERRQ(ierr);
1464c1414c8SBarry Smith   PetscFunctionReturn(0);
1474c1414c8SBarry Smith }
1484c1414c8SBarry Smith 
1494c1414c8SBarry Smith /*
1504c1414c8SBarry Smith       This builds nonsymmetric version of nonzero structure,
1514c1414c8SBarry Smith */
1524c1414c8SBarry Smith #undef __FUNCT__
1534108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric"
1544108e4d5SBarry Smith static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,PetscInt *iia[],PetscInt *jja[],PetscInt ishift,PetscInt oshift)
1554c1414c8SBarry Smith {
1564c1414c8SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
1574c1414c8SBarry Smith   PetscErrorCode ierr;
1584c1414c8SBarry Smith   PetscInt       *work,*ia,*ja,*j,nz,nslim_row,n,row,col,*ns_col,nslim_col;
1594c1414c8SBarry Smith   PetscInt       *tns,*tvc,*ns_row = a->inode.size,nsz,i1,i2,*ai= a->i,*aj = a->j;
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 */
1694c1414c8SBarry Smith   ierr = PetscMalloc((nslim_col +1)*sizeof(PetscInt),&tns);CHKERRQ(ierr);
1704c1414c8SBarry Smith   ierr = PetscMalloc((n +1)*sizeof(PetscInt),&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];
1754c1414c8SBarry Smith     for (i2=0; i2<nsz; ++i2,++col)
1764c1414c8SBarry Smith       tvc[col] = i1;
1774c1414c8SBarry Smith   }
1784c1414c8SBarry Smith   /* allocate space for row pointers */
1794c1414c8SBarry Smith   ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&ia);CHKERRQ(ierr);
1804c1414c8SBarry Smith   *iia = ia;
1814c1414c8SBarry Smith   ierr = PetscMemzero(ia,(nslim_row+1)*sizeof(PetscInt));CHKERRQ(ierr);
1824c1414c8SBarry Smith   ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&work);CHKERRQ(ierr);
1834c1414c8SBarry Smith 
1844c1414c8SBarry Smith   /* determine the number of columns in each row */
1854c1414c8SBarry Smith   ia[0] = oshift;
1864c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
1874c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
1884c1414c8SBarry Smith     col = *j++ + ishift;
1894c1414c8SBarry Smith     i2  = tvc[col];
1904c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
1914c1414c8SBarry Smith     while (nz-- > 0) {           /* off-diagonal elemets */
1924c1414c8SBarry Smith       ia[i1+1]++;
1934c1414c8SBarry Smith       i2++;                     /* Start col of next node */
1944c1414c8SBarry Smith       while (((col = *j++ + ishift) < tns[i2]) && nz > 0) {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);
2084c1414c8SBarry Smith   ierr = PetscMalloc(nz*sizeof(PetscInt),&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;
2144c1414c8SBarry Smith     i2  = 0;                     /* Col inode index */
2154c1414c8SBarry Smith     col = *j++ + ishift;
2164c1414c8SBarry Smith     i2  = tvc[col];
2174c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
2184c1414c8SBarry Smith     while (nz-- > 0) {
2194c1414c8SBarry Smith       ja[work[i1]++] = i2 + oshift;
2204c1414c8SBarry Smith       ++i2;
2214c1414c8SBarry Smith       while(((col = *j++ + ishift) < tns[i2]) && nz > 0) {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"
2344108e4d5SBarry Smith static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscTruth symmetric,PetscTruth blockcompressed,PetscInt *n,PetscInt *ia[],PetscInt *ja[],PetscTruth *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"
2544108e4d5SBarry Smith static PetscErrorCode MatRestoreRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscTruth symmetric,PetscTruth blockcompressed,PetscInt *n,PetscInt *ia[],PetscInt *ja[],PetscTruth *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   }
2678f7157efSSatish Balay 
2684c1414c8SBarry Smith   PetscFunctionReturn(0);
2694c1414c8SBarry Smith }
2704c1414c8SBarry Smith 
2714c1414c8SBarry Smith /* ----------------------------------------------------------- */
2724c1414c8SBarry Smith 
2734c1414c8SBarry Smith #undef __FUNCT__
2744108e4d5SBarry Smith #define __FUNCT__ "MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric"
2754108e4d5SBarry Smith static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,PetscInt *iia[],PetscInt *jja[],PetscInt ishift,PetscInt oshift)
2764c1414c8SBarry Smith {
2774c1414c8SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
2784c1414c8SBarry Smith   PetscErrorCode ierr;
2794c1414c8SBarry Smith   PetscInt       *work,*ia,*ja,*j,nz,nslim_row, n,row,col,*ns_col,nslim_col;
2804c1414c8SBarry Smith   PetscInt       *tns,*tvc,*ns_row = a->inode.size,nsz,i1,i2,*ai= a->i,*aj = a->j;
2814c1414c8SBarry Smith 
2824c1414c8SBarry Smith   PetscFunctionBegin;
2834c1414c8SBarry Smith   nslim_row = a->inode.node_count;
284d0f46423SBarry Smith   n         = A->cmap->n;
2854c1414c8SBarry Smith 
2864c1414c8SBarry Smith   /* Create The column_inode for this matrix */
2874c1414c8SBarry Smith   ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr);
2884c1414c8SBarry Smith 
2894c1414c8SBarry Smith   /* allocate space for reformated column_inode structure */
2904c1414c8SBarry Smith   ierr = PetscMalloc((nslim_col + 1)*sizeof(PetscInt),&tns);CHKERRQ(ierr);
2914c1414c8SBarry Smith   ierr = PetscMalloc((n + 1)*sizeof(PetscInt),&tvc);CHKERRQ(ierr);
2924c1414c8SBarry Smith   for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1];
2934c1414c8SBarry Smith 
2944c1414c8SBarry Smith   for (i1=0,col=0; i1<nslim_col; ++i1){
2954c1414c8SBarry Smith     nsz = ns_col[i1];
2964c1414c8SBarry Smith     for (i2=0; i2<nsz; ++i2,++col)
2974c1414c8SBarry Smith       tvc[col] = i1;
2984c1414c8SBarry Smith   }
2994c1414c8SBarry Smith   /* allocate space for column pointers */
3004c1414c8SBarry Smith   ierr = PetscMalloc((nslim_col+1)*sizeof(PetscInt),&ia);CHKERRQ(ierr);
3014c1414c8SBarry Smith   *iia = ia;
3024c1414c8SBarry Smith   ierr = PetscMemzero(ia,(nslim_col+1)*sizeof(PetscInt));CHKERRQ(ierr);
3034c1414c8SBarry Smith   ierr = PetscMalloc((nslim_col+1)*sizeof(PetscInt),&work);CHKERRQ(ierr);
3044c1414c8SBarry Smith 
3054c1414c8SBarry Smith   /* determine the number of columns in each row */
3064c1414c8SBarry Smith   ia[0] = oshift;
3074c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
3084c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
3094c1414c8SBarry Smith     col = *j++ + ishift;
3104c1414c8SBarry Smith     i2  = tvc[col];
3114c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
3124c1414c8SBarry Smith     while (nz-- > 0) {           /* off-diagonal elemets */
3134c1414c8SBarry Smith       /* ia[i1+1]++; */
3144c1414c8SBarry Smith       ia[i2+1]++;
3154c1414c8SBarry Smith       i2++;
3164c1414c8SBarry Smith       while (((col = *j++ + ishift) < tns[i2]) && nz > 0) {nz--;}
3174c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
3184c1414c8SBarry Smith     }
3194c1414c8SBarry Smith   }
3204c1414c8SBarry Smith 
3214c1414c8SBarry Smith   /* shift ia[i] to point to next col */
3224c1414c8SBarry Smith   for (i1=1; i1<nslim_col+1; i1++) {
3234c1414c8SBarry Smith     col        = ia[i1-1];
3244c1414c8SBarry Smith     ia[i1]    += col;
3254c1414c8SBarry Smith     work[i1-1] = col - oshift;
3264c1414c8SBarry Smith   }
3274c1414c8SBarry Smith 
3284c1414c8SBarry Smith   /* allocate space for column pointers */
3294c1414c8SBarry Smith   nz   = ia[nslim_col] + (!ishift);
3304c1414c8SBarry Smith   ierr = PetscMalloc(nz*sizeof(PetscInt),&ja);CHKERRQ(ierr);
3314c1414c8SBarry Smith   *jja = ja;
3324c1414c8SBarry Smith 
3334c1414c8SBarry Smith  /* loop over matrix putting into ja */
3344c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
3354c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
3364c1414c8SBarry Smith     i2  = 0;                     /* Col inode index */
3374c1414c8SBarry Smith     col = *j++ + ishift;
3384c1414c8SBarry Smith     i2  = tvc[col];
3394c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
3404c1414c8SBarry Smith     while (nz-- > 0) {
3414c1414c8SBarry Smith       /* ja[work[i1]++] = i2 + oshift; */
3424c1414c8SBarry Smith       ja[work[i2]++] = i1 + oshift;
3434c1414c8SBarry Smith       i2++;
3444c1414c8SBarry Smith       while(((col = *j++ + ishift) < tns[i2]) && nz > 0) {nz--;}
3454c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
3464c1414c8SBarry Smith     }
3474c1414c8SBarry Smith   }
3484c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
3494c1414c8SBarry Smith   ierr = PetscFree(work);CHKERRQ(ierr);
3504c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
3514c1414c8SBarry Smith   ierr = PetscFree(tvc);CHKERRQ(ierr);
3524c1414c8SBarry Smith   PetscFunctionReturn(0);
3534c1414c8SBarry Smith }
3544c1414c8SBarry Smith 
3554c1414c8SBarry Smith #undef __FUNCT__
3564108e4d5SBarry Smith #define __FUNCT__ "MatGetColumnIJ_SeqAIJ_Inode"
3574108e4d5SBarry Smith static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscTruth symmetric,PetscTruth blockcompressed,PetscInt *n,PetscInt *ia[],PetscInt *ja[],PetscTruth *done)
3584c1414c8SBarry Smith {
3594c1414c8SBarry Smith   PetscErrorCode ierr;
3604c1414c8SBarry Smith 
3614c1414c8SBarry Smith   PetscFunctionBegin;
3624c1414c8SBarry Smith   ierr = Mat_CreateColInode(A,n,PETSC_NULL);CHKERRQ(ierr);
3634c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
3644c1414c8SBarry Smith 
3658f7157efSSatish Balay   if (!blockcompressed) {
3668f7157efSSatish Balay     ierr = MatGetColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);;
3678f7157efSSatish Balay   } else if (symmetric) {
3684c1414c8SBarry Smith     /* Since the indices are symmetric it does'nt matter */
3694108e4d5SBarry Smith     ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
3704c1414c8SBarry Smith   } else {
3714108e4d5SBarry Smith     ierr = MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
3724c1414c8SBarry Smith   }
3734c1414c8SBarry Smith   PetscFunctionReturn(0);
3744c1414c8SBarry Smith }
3754c1414c8SBarry Smith 
3764c1414c8SBarry Smith #undef __FUNCT__
3774108e4d5SBarry Smith #define __FUNCT__ "MatRestoreColumnIJ_SeqAIJ_Inode"
3784108e4d5SBarry Smith static PetscErrorCode MatRestoreColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscTruth symmetric,PetscTruth blockcompressed,PetscInt *n,PetscInt *ia[],PetscInt *ja[],PetscTruth *done)
3794c1414c8SBarry Smith {
3804c1414c8SBarry Smith   PetscErrorCode ierr;
3814c1414c8SBarry Smith 
3824c1414c8SBarry Smith   PetscFunctionBegin;
3834c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
3848f7157efSSatish Balay   if (!blockcompressed) {
3858f7157efSSatish Balay     ierr = MatRestoreColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);;
3868f7157efSSatish Balay   } else {
3874c1414c8SBarry Smith     ierr = PetscFree(*ia);CHKERRQ(ierr);
3884c1414c8SBarry Smith     ierr = PetscFree(*ja);CHKERRQ(ierr);
3898f7157efSSatish Balay   }
3904c1414c8SBarry Smith   PetscFunctionReturn(0);
3914c1414c8SBarry Smith }
3924c1414c8SBarry Smith 
3934c1414c8SBarry Smith /* ----------------------------------------------------------- */
3944c1414c8SBarry Smith 
3954c1414c8SBarry Smith #undef __FUNCT__
3964108e4d5SBarry Smith #define __FUNCT__ "MatMult_SeqAIJ_Inode"
3974108e4d5SBarry Smith static PetscErrorCode MatMult_SeqAIJ_Inode(Mat A,Vec xx,Vec yy)
3984c1414c8SBarry Smith {
3994c1414c8SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
4004c1414c8SBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5,tmp0,tmp1;
401d9fead3dSBarry Smith   PetscScalar       *y;
402dd6ea824SBarry Smith   const PetscScalar *x;
403dd6ea824SBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5;
4044c1414c8SBarry Smith   PetscErrorCode    ierr;
40598c9bda7SSatish Balay   PetscInt          *idx,i1,i2,n,i,row,node_max,*ns,*ii,nsz,sz,nonzerorow=0;
4064c1414c8SBarry Smith 
4074c1414c8SBarry Smith #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
4084c1414c8SBarry Smith #pragma disjoint(*x,*y,*v1,*v2,*v3,*v4,*v5)
4094c1414c8SBarry Smith #endif
4104c1414c8SBarry Smith 
4114c1414c8SBarry Smith   PetscFunctionBegin;
4124c1414c8SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_ERR_COR,"Missing Inode Structure");
4134c1414c8SBarry Smith   node_max = a->inode.node_count;
4144c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
415d9fead3dSBarry Smith   ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
4164c1414c8SBarry Smith   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
4174c1414c8SBarry Smith   idx  = a->j;
4184c1414c8SBarry Smith   v1   = a->a;
4194c1414c8SBarry Smith   ii   = a->i;
4204c1414c8SBarry Smith 
4214c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i){
4224c1414c8SBarry Smith     nsz  = ns[i];
4234c1414c8SBarry Smith     n    = ii[1] - ii[0];
42498c9bda7SSatish Balay     nonzerorow += (n>0)*nsz;
4254c1414c8SBarry Smith     ii  += nsz;
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 :
5744c1414c8SBarry Smith       SETERRQ(PETSC_ERR_COR,"Node size not yet supported");
5754c1414c8SBarry Smith     }
5764c1414c8SBarry Smith   }
577d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&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;
590dd6ea824SBarry Smith   MatScalar      *v1,*v2,*v3,*v4,*v5;
591dd6ea824SBarry Smith   PetscScalar    *x,*y,*z,*zt;
5924c1414c8SBarry Smith   PetscErrorCode ierr;
5934c1414c8SBarry Smith   PetscInt       *idx,i1,i2,n,i,row,node_max,*ns,*ii,nsz,sz;
5944c1414c8SBarry Smith 
5954c1414c8SBarry Smith   PetscFunctionBegin;
5964c1414c8SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_ERR_COR,"Missing Inode Structure");
5974c1414c8SBarry Smith   node_max = a->inode.node_count;
5984c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
5994c1414c8SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
6004c1414c8SBarry Smith   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
6014c1414c8SBarry Smith   if (zz != yy) {
6024c1414c8SBarry Smith     ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
6034c1414c8SBarry Smith   } else {
6044c1414c8SBarry Smith     z = y;
6054c1414c8SBarry Smith   }
6064c1414c8SBarry Smith   zt = z;
6074c1414c8SBarry Smith 
6084c1414c8SBarry Smith   idx  = a->j;
6094c1414c8SBarry Smith   v1   = a->a;
6104c1414c8SBarry Smith   ii   = a->i;
6114c1414c8SBarry Smith 
6124c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i){
6134c1414c8SBarry Smith     nsz  = ns[i];
6144c1414c8SBarry Smith     n    = ii[1] - ii[0];
6154c1414c8SBarry Smith     ii  += nsz;
6164c1414c8SBarry Smith     sz   = n;                   /* No of non zeros in this row */
6174c1414c8SBarry Smith                                 /* Switch on the size of Node */
6184c1414c8SBarry Smith     switch (nsz){               /* Each loop in 'case' is unrolled */
6194c1414c8SBarry Smith     case 1 :
6204c1414c8SBarry Smith       sum1  = *zt++;
6214c1414c8SBarry Smith 
6224c1414c8SBarry Smith       for(n = 0; n< sz-1; n+=2) {
6234c1414c8SBarry Smith         i1   = idx[0];          /* The instructions are ordered to */
6244c1414c8SBarry Smith         i2   = idx[1];          /* make the compiler's job easy */
6254c1414c8SBarry Smith         idx += 2;
6264c1414c8SBarry Smith         tmp0 = x[i1];
6274c1414c8SBarry Smith         tmp1 = x[i2];
6284c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6294c1414c8SBarry Smith        }
6304c1414c8SBarry Smith 
6314c1414c8SBarry Smith       if(n   == sz-1){          /* Take care of the last nonzero  */
6324c1414c8SBarry Smith         tmp0  = x[*idx++];
6334c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6344c1414c8SBarry Smith       }
6354c1414c8SBarry Smith       y[row++]=sum1;
6364c1414c8SBarry Smith       break;
6374c1414c8SBarry Smith     case 2:
6384c1414c8SBarry Smith       sum1  = *zt++;
6394c1414c8SBarry Smith       sum2  = *zt++;
6404c1414c8SBarry Smith       v2    = v1 + n;
6414c1414c8SBarry Smith 
6424c1414c8SBarry Smith       for(n = 0; n< sz-1; n+=2) {
6434c1414c8SBarry Smith         i1   = idx[0];
6444c1414c8SBarry Smith         i2   = idx[1];
6454c1414c8SBarry Smith         idx += 2;
6464c1414c8SBarry Smith         tmp0 = x[i1];
6474c1414c8SBarry Smith         tmp1 = x[i2];
6484c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6494c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
6504c1414c8SBarry Smith       }
6514c1414c8SBarry Smith       if(n   == sz-1){
6524c1414c8SBarry Smith         tmp0  = x[*idx++];
6534c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6544c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
6554c1414c8SBarry Smith       }
6564c1414c8SBarry Smith       y[row++]=sum1;
6574c1414c8SBarry Smith       y[row++]=sum2;
6584c1414c8SBarry Smith       v1      =v2;              /* Since the next block to be processed starts there*/
6594c1414c8SBarry Smith       idx    +=sz;
6604c1414c8SBarry Smith       break;
6614c1414c8SBarry Smith     case 3:
6624c1414c8SBarry Smith       sum1  = *zt++;
6634c1414c8SBarry Smith       sum2  = *zt++;
6644c1414c8SBarry Smith       sum3  = *zt++;
6654c1414c8SBarry Smith       v2    = v1 + n;
6664c1414c8SBarry Smith       v3    = v2 + n;
6674c1414c8SBarry Smith 
6684c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6694c1414c8SBarry Smith         i1   = idx[0];
6704c1414c8SBarry Smith         i2   = idx[1];
6714c1414c8SBarry Smith         idx += 2;
6724c1414c8SBarry Smith         tmp0 = x[i1];
6734c1414c8SBarry Smith         tmp1 = x[i2];
6744c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6754c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
6764c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
6774c1414c8SBarry Smith       }
6784c1414c8SBarry Smith       if (n == sz-1){
6794c1414c8SBarry Smith         tmp0  = x[*idx++];
6804c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6814c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
6824c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
6834c1414c8SBarry Smith       }
6844c1414c8SBarry Smith       y[row++]=sum1;
6854c1414c8SBarry Smith       y[row++]=sum2;
6864c1414c8SBarry Smith       y[row++]=sum3;
6874c1414c8SBarry Smith       v1       =v3;             /* Since the next block to be processed starts there*/
6884c1414c8SBarry Smith       idx     +=2*sz;
6894c1414c8SBarry Smith       break;
6904c1414c8SBarry Smith     case 4:
6914c1414c8SBarry Smith       sum1  = *zt++;
6924c1414c8SBarry Smith       sum2  = *zt++;
6934c1414c8SBarry Smith       sum3  = *zt++;
6944c1414c8SBarry Smith       sum4  = *zt++;
6954c1414c8SBarry Smith       v2    = v1 + n;
6964c1414c8SBarry Smith       v3    = v2 + n;
6974c1414c8SBarry Smith       v4    = v3 + n;
6984c1414c8SBarry Smith 
6994c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
7004c1414c8SBarry Smith         i1   = idx[0];
7014c1414c8SBarry Smith         i2   = idx[1];
7024c1414c8SBarry Smith         idx += 2;
7034c1414c8SBarry Smith         tmp0 = x[i1];
7044c1414c8SBarry Smith         tmp1 = x[i2];
7054c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
7064c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
7074c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
7084c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
7094c1414c8SBarry Smith       }
7104c1414c8SBarry Smith       if (n == sz-1){
7114c1414c8SBarry Smith         tmp0  = x[*idx++];
7124c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
7134c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
7144c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
7154c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
7164c1414c8SBarry Smith       }
7174c1414c8SBarry Smith       y[row++]=sum1;
7184c1414c8SBarry Smith       y[row++]=sum2;
7194c1414c8SBarry Smith       y[row++]=sum3;
7204c1414c8SBarry Smith       y[row++]=sum4;
7214c1414c8SBarry Smith       v1      =v4;              /* Since the next block to be processed starts there*/
7224c1414c8SBarry Smith       idx    +=3*sz;
7234c1414c8SBarry Smith       break;
7244c1414c8SBarry Smith     case 5:
7254c1414c8SBarry Smith       sum1  = *zt++;
7264c1414c8SBarry Smith       sum2  = *zt++;
7274c1414c8SBarry Smith       sum3  = *zt++;
7284c1414c8SBarry Smith       sum4  = *zt++;
7294c1414c8SBarry Smith       sum5  = *zt++;
7304c1414c8SBarry Smith       v2    = v1 + n;
7314c1414c8SBarry Smith       v3    = v2 + n;
7324c1414c8SBarry Smith       v4    = v3 + n;
7334c1414c8SBarry Smith       v5    = v4 + n;
7344c1414c8SBarry Smith 
7354c1414c8SBarry Smith       for (n = 0; n<sz-1; n+=2) {
7364c1414c8SBarry Smith         i1   = idx[0];
7374c1414c8SBarry Smith         i2   = idx[1];
7384c1414c8SBarry Smith         idx += 2;
7394c1414c8SBarry Smith         tmp0 = x[i1];
7404c1414c8SBarry Smith         tmp1 = x[i2];
7414c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
7424c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
7434c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
7444c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
7454c1414c8SBarry Smith         sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2;
7464c1414c8SBarry Smith       }
7474c1414c8SBarry Smith       if(n   == sz-1){
7484c1414c8SBarry Smith         tmp0  = x[*idx++];
7494c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
7504c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
7514c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
7524c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
7534c1414c8SBarry Smith         sum5 += *v5++ * tmp0;
7544c1414c8SBarry Smith       }
7554c1414c8SBarry Smith       y[row++]=sum1;
7564c1414c8SBarry Smith       y[row++]=sum2;
7574c1414c8SBarry Smith       y[row++]=sum3;
7584c1414c8SBarry Smith       y[row++]=sum4;
7594c1414c8SBarry Smith       y[row++]=sum5;
7604c1414c8SBarry Smith       v1      =v5;       /* Since the next block to be processed starts there */
7614c1414c8SBarry Smith       idx    +=4*sz;
7624c1414c8SBarry Smith       break;
7634c1414c8SBarry Smith     default :
7644c1414c8SBarry Smith       SETERRQ(PETSC_ERR_COR,"Node size not yet supported");
7654c1414c8SBarry Smith     }
7664c1414c8SBarry Smith   }
7674c1414c8SBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
7684c1414c8SBarry Smith   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
7694c1414c8SBarry Smith   if (zz != yy) {
7704c1414c8SBarry Smith     ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
7714c1414c8SBarry Smith   }
772dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
7734c1414c8SBarry Smith   PetscFunctionReturn(0);
7744c1414c8SBarry Smith }
7754c1414c8SBarry Smith 
7764c1414c8SBarry Smith /* ----------------------------------------------------------- */
7774c1414c8SBarry Smith #undef __FUNCT__
7784108e4d5SBarry Smith #define __FUNCT__ "MatSolve_SeqAIJ_Inode"
7794108e4d5SBarry Smith PetscErrorCode MatSolve_SeqAIJ_Inode(Mat A,Vec bb,Vec xx)
7804c1414c8SBarry Smith {
7814c1414c8SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
7824c1414c8SBarry Smith   IS                iscol = a->col,isrow = a->row;
7834c1414c8SBarry Smith   PetscErrorCode    ierr;
7845d0c19d7SBarry Smith   const PetscInt    *r,*c,*rout,*cout;
7855d0c19d7SBarry Smith   PetscInt          i,j,n = A->rmap->n,*ai = a->i,nz,*a_j = a->j;
7865d0c19d7SBarry Smith   PetscInt          node_max,*ns,row,nsz,aii,*vi,*ad,*aj,i0,i1;
787d9fead3dSBarry Smith   PetscScalar       *x,*tmp,*tmps,tmp0,tmp1;
788d9fead3dSBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5;
789dd6ea824SBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa;
790dd6ea824SBarry Smith   const PetscScalar *b;
7914c1414c8SBarry Smith 
7924c1414c8SBarry Smith   PetscFunctionBegin;
7934c1414c8SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_ERR_COR,"Missing Inode Structure");
7944c1414c8SBarry Smith   node_max = a->inode.node_count;
7954c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
7964c1414c8SBarry Smith 
797d9fead3dSBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
7984c1414c8SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
7994c1414c8SBarry Smith   tmp  = a->solve_work;
8004c1414c8SBarry Smith 
8014c1414c8SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
8024c1414c8SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
8034c1414c8SBarry Smith 
8044c1414c8SBarry Smith   /* forward solve the lower triangular */
8054c1414c8SBarry Smith   tmps = tmp ;
8064c1414c8SBarry Smith   aa   = a_a ;
8074c1414c8SBarry Smith   aj   = a_j ;
8084c1414c8SBarry Smith   ad   = a->diag;
8094c1414c8SBarry Smith 
8104c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i){
8114c1414c8SBarry Smith     nsz = ns[i];
8124c1414c8SBarry Smith     aii = ai[row];
8134c1414c8SBarry Smith     v1  = aa + aii;
8144c1414c8SBarry Smith     vi  = aj + aii;
8154c1414c8SBarry Smith     nz  = ad[row]- aii;
8164c1414c8SBarry Smith 
8174c1414c8SBarry Smith     switch (nsz){               /* Each loop in 'case' is unrolled */
8184c1414c8SBarry Smith     case 1 :
8194c1414c8SBarry Smith       sum1 = b[*r++];
8204c1414c8SBarry Smith       for(j=0; j<nz-1; j+=2){
8214c1414c8SBarry Smith         i0   = vi[0];
8224c1414c8SBarry Smith         i1   = vi[1];
8234c1414c8SBarry Smith         vi  +=2;
8244c1414c8SBarry Smith         tmp0 = tmps[i0];
8254c1414c8SBarry Smith         tmp1 = tmps[i1];
8264c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8274c1414c8SBarry Smith       }
8284c1414c8SBarry Smith       if(j == nz-1){
8294c1414c8SBarry Smith         tmp0 = tmps[*vi++];
8304c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8314c1414c8SBarry Smith       }
8324c1414c8SBarry Smith       tmp[row ++]=sum1;
8334c1414c8SBarry Smith       break;
8344c1414c8SBarry Smith     case 2:
8354c1414c8SBarry Smith       sum1 = b[*r++];
8364c1414c8SBarry Smith       sum2 = b[*r++];
8374c1414c8SBarry Smith       v2   = aa + ai[row+1];
8384c1414c8SBarry Smith 
8394c1414c8SBarry Smith       for(j=0; j<nz-1; j+=2){
8404c1414c8SBarry Smith         i0   = vi[0];
8414c1414c8SBarry Smith         i1   = vi[1];
8424c1414c8SBarry Smith         vi  +=2;
8434c1414c8SBarry Smith         tmp0 = tmps[i0];
8444c1414c8SBarry Smith         tmp1 = tmps[i1];
8454c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8464c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
8474c1414c8SBarry Smith       }
8484c1414c8SBarry Smith       if(j == nz-1){
8494c1414c8SBarry Smith         tmp0 = tmps[*vi++];
8504c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8514c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
8524c1414c8SBarry Smith       }
8534c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
8544c1414c8SBarry Smith       tmp[row ++]=sum1;
8554c1414c8SBarry Smith       tmp[row ++]=sum2;
8564c1414c8SBarry Smith       break;
8574c1414c8SBarry Smith     case 3:
8584c1414c8SBarry Smith       sum1 = b[*r++];
8594c1414c8SBarry Smith       sum2 = b[*r++];
8604c1414c8SBarry Smith       sum3 = b[*r++];
8614c1414c8SBarry Smith       v2   = aa + ai[row+1];
8624c1414c8SBarry Smith       v3   = aa + ai[row+2];
8634c1414c8SBarry Smith 
8644c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2){
8654c1414c8SBarry Smith         i0   = vi[0];
8664c1414c8SBarry Smith         i1   = vi[1];
8674c1414c8SBarry Smith         vi  +=2;
8684c1414c8SBarry Smith         tmp0 = tmps[i0];
8694c1414c8SBarry Smith         tmp1 = tmps[i1];
8704c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8714c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
8724c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
8734c1414c8SBarry Smith       }
8744c1414c8SBarry Smith       if (j == nz-1){
8754c1414c8SBarry Smith         tmp0 = tmps[*vi++];
8764c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8774c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
8784c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
8794c1414c8SBarry Smith       }
8804c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
8814c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
8824c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
8834c1414c8SBarry Smith       tmp[row ++]=sum1;
8844c1414c8SBarry Smith       tmp[row ++]=sum2;
8854c1414c8SBarry Smith       tmp[row ++]=sum3;
8864c1414c8SBarry Smith       break;
8874c1414c8SBarry Smith 
8884c1414c8SBarry Smith     case 4:
8894c1414c8SBarry Smith       sum1 = b[*r++];
8904c1414c8SBarry Smith       sum2 = b[*r++];
8914c1414c8SBarry Smith       sum3 = b[*r++];
8924c1414c8SBarry Smith       sum4 = b[*r++];
8934c1414c8SBarry Smith       v2   = aa + ai[row+1];
8944c1414c8SBarry Smith       v3   = aa + ai[row+2];
8954c1414c8SBarry Smith       v4   = aa + ai[row+3];
8964c1414c8SBarry Smith 
8974c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2){
8984c1414c8SBarry Smith         i0   = vi[0];
8994c1414c8SBarry Smith         i1   = vi[1];
9004c1414c8SBarry Smith         vi  +=2;
9014c1414c8SBarry Smith         tmp0 = tmps[i0];
9024c1414c8SBarry Smith         tmp1 = tmps[i1];
9034c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
9044c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
9054c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
9064c1414c8SBarry Smith         sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
9074c1414c8SBarry Smith       }
9084c1414c8SBarry Smith       if (j == nz-1){
9094c1414c8SBarry Smith         tmp0 = tmps[*vi++];
9104c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
9114c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
9124c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
9134c1414c8SBarry Smith         sum4 -= *v4++ *tmp0;
9144c1414c8SBarry Smith       }
9154c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
9164c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
9174c1414c8SBarry Smith       sum4 -= *v4++ * sum1;
9184c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
9194c1414c8SBarry Smith       sum4 -= *v4++ * sum2;
9204c1414c8SBarry Smith       sum4 -= *v4++ * sum3;
9214c1414c8SBarry Smith 
9224c1414c8SBarry Smith       tmp[row ++]=sum1;
9234c1414c8SBarry Smith       tmp[row ++]=sum2;
9244c1414c8SBarry Smith       tmp[row ++]=sum3;
9254c1414c8SBarry Smith       tmp[row ++]=sum4;
9264c1414c8SBarry Smith       break;
9274c1414c8SBarry Smith     case 5:
9284c1414c8SBarry Smith       sum1 = b[*r++];
9294c1414c8SBarry Smith       sum2 = b[*r++];
9304c1414c8SBarry Smith       sum3 = b[*r++];
9314c1414c8SBarry Smith       sum4 = b[*r++];
9324c1414c8SBarry Smith       sum5 = b[*r++];
9334c1414c8SBarry Smith       v2   = aa + ai[row+1];
9344c1414c8SBarry Smith       v3   = aa + ai[row+2];
9354c1414c8SBarry Smith       v4   = aa + ai[row+3];
9364c1414c8SBarry Smith       v5   = aa + ai[row+4];
9374c1414c8SBarry Smith 
9384c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2){
9394c1414c8SBarry Smith         i0   = vi[0];
9404c1414c8SBarry Smith         i1   = vi[1];
9414c1414c8SBarry Smith         vi  +=2;
9424c1414c8SBarry Smith         tmp0 = tmps[i0];
9434c1414c8SBarry Smith         tmp1 = tmps[i1];
9444c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
9454c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
9464c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
9474c1414c8SBarry Smith         sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
9484c1414c8SBarry Smith         sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
9494c1414c8SBarry Smith       }
9504c1414c8SBarry Smith       if (j == nz-1){
9514c1414c8SBarry Smith         tmp0 = tmps[*vi++];
9524c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
9534c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
9544c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
9554c1414c8SBarry Smith         sum4 -= *v4++ *tmp0;
9564c1414c8SBarry Smith         sum5 -= *v5++ *tmp0;
9574c1414c8SBarry Smith       }
9584c1414c8SBarry Smith 
9594c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
9604c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
9614c1414c8SBarry Smith       sum4 -= *v4++ * sum1;
9624c1414c8SBarry Smith       sum5 -= *v5++ * sum1;
9634c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
9644c1414c8SBarry Smith       sum4 -= *v4++ * sum2;
9654c1414c8SBarry Smith       sum5 -= *v5++ * sum2;
9664c1414c8SBarry Smith       sum4 -= *v4++ * sum3;
9674c1414c8SBarry Smith       sum5 -= *v5++ * sum3;
9684c1414c8SBarry Smith       sum5 -= *v5++ * sum4;
9694c1414c8SBarry Smith 
9704c1414c8SBarry Smith       tmp[row ++]=sum1;
9714c1414c8SBarry Smith       tmp[row ++]=sum2;
9724c1414c8SBarry Smith       tmp[row ++]=sum3;
9734c1414c8SBarry Smith       tmp[row ++]=sum4;
9744c1414c8SBarry Smith       tmp[row ++]=sum5;
9754c1414c8SBarry Smith       break;
9764c1414c8SBarry Smith     default:
9774c1414c8SBarry Smith       SETERRQ(PETSC_ERR_COR,"Node size not yet supported \n");
9784c1414c8SBarry Smith     }
9794c1414c8SBarry Smith   }
9804c1414c8SBarry Smith   /* backward solve the upper triangular */
9814c1414c8SBarry Smith   for (i=node_max -1 ,row = n-1 ; i>=0; i--){
9824c1414c8SBarry Smith     nsz = ns[i];
9834c1414c8SBarry Smith     aii = ai[row+1] -1;
9844c1414c8SBarry Smith     v1  = aa + aii;
9854c1414c8SBarry Smith     vi  = aj + aii;
9864c1414c8SBarry Smith     nz  = aii- ad[row];
9874c1414c8SBarry Smith     switch (nsz){               /* Each loop in 'case' is unrolled */
9884c1414c8SBarry Smith     case 1 :
9894c1414c8SBarry Smith       sum1 = tmp[row];
9904c1414c8SBarry Smith 
9914c1414c8SBarry Smith       for(j=nz ; j>1; j-=2){
9924c1414c8SBarry Smith         vi  -=2;
9934c1414c8SBarry Smith         i0   = vi[2];
9944c1414c8SBarry Smith         i1   = vi[1];
9954c1414c8SBarry Smith         tmp0 = tmps[i0];
9964c1414c8SBarry Smith         tmp1 = tmps[i1];
9974c1414c8SBarry Smith         v1   -= 2;
9984c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
9994c1414c8SBarry Smith       }
10004c1414c8SBarry Smith       if (j==1){
10014c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10024c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10034c1414c8SBarry Smith       }
10044c1414c8SBarry Smith       x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10054c1414c8SBarry Smith       break;
10064c1414c8SBarry Smith     case 2 :
10074c1414c8SBarry Smith       sum1 = tmp[row];
10084c1414c8SBarry Smith       sum2 = tmp[row -1];
10094c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10104c1414c8SBarry Smith       for (j=nz ; j>1; j-=2){
10114c1414c8SBarry Smith         vi  -=2;
10124c1414c8SBarry Smith         i0   = vi[2];
10134c1414c8SBarry Smith         i1   = vi[1];
10144c1414c8SBarry Smith         tmp0 = tmps[i0];
10154c1414c8SBarry Smith         tmp1 = tmps[i1];
10164c1414c8SBarry Smith         v1   -= 2;
10174c1414c8SBarry Smith         v2   -= 2;
10184c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10194c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10204c1414c8SBarry Smith       }
10214c1414c8SBarry Smith       if (j==1){
10224c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10234c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10244c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10254c1414c8SBarry Smith       }
10264c1414c8SBarry Smith 
10274c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10284c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10294c1414c8SBarry Smith       x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
10304c1414c8SBarry Smith       break;
10314c1414c8SBarry Smith     case 3 :
10324c1414c8SBarry Smith       sum1 = tmp[row];
10334c1414c8SBarry Smith       sum2 = tmp[row -1];
10344c1414c8SBarry Smith       sum3 = tmp[row -2];
10354c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10364c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
10374c1414c8SBarry Smith       for (j=nz ; j>1; j-=2){
10384c1414c8SBarry Smith         vi  -=2;
10394c1414c8SBarry Smith         i0   = vi[2];
10404c1414c8SBarry Smith         i1   = vi[1];
10414c1414c8SBarry Smith         tmp0 = tmps[i0];
10424c1414c8SBarry Smith         tmp1 = tmps[i1];
10434c1414c8SBarry Smith         v1   -= 2;
10444c1414c8SBarry Smith         v2   -= 2;
10454c1414c8SBarry Smith         v3   -= 2;
10464c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10474c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10484c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
10494c1414c8SBarry Smith       }
10504c1414c8SBarry Smith       if (j==1){
10514c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10524c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10534c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10544c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
10554c1414c8SBarry Smith       }
10564c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10574c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10584c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10594c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
10604c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10614c1414c8SBarry Smith       x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
10624c1414c8SBarry Smith 
10634c1414c8SBarry Smith       break;
10644c1414c8SBarry Smith     case 4 :
10654c1414c8SBarry Smith       sum1 = tmp[row];
10664c1414c8SBarry Smith       sum2 = tmp[row -1];
10674c1414c8SBarry Smith       sum3 = tmp[row -2];
10684c1414c8SBarry Smith       sum4 = tmp[row -3];
10694c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10704c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
10714c1414c8SBarry Smith       v4   = aa + ai[row -2]-1;
10724c1414c8SBarry Smith 
10734c1414c8SBarry Smith       for (j=nz ; j>1; j-=2){
10744c1414c8SBarry Smith         vi  -=2;
10754c1414c8SBarry Smith         i0   = vi[2];
10764c1414c8SBarry Smith         i1   = vi[1];
10774c1414c8SBarry Smith         tmp0 = tmps[i0];
10784c1414c8SBarry Smith         tmp1 = tmps[i1];
10794c1414c8SBarry Smith         v1  -= 2;
10804c1414c8SBarry Smith         v2  -= 2;
10814c1414c8SBarry Smith         v3  -= 2;
10824c1414c8SBarry Smith         v4  -= 2;
10834c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10844c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10854c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
10864c1414c8SBarry Smith         sum4 -= v4[2] * tmp0 + v4[1] * tmp1;
10874c1414c8SBarry Smith       }
10884c1414c8SBarry Smith       if (j==1){
10894c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10904c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10914c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10924c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
10934c1414c8SBarry Smith         sum4 -= *v4-- * tmp0;
10944c1414c8SBarry Smith       }
10954c1414c8SBarry Smith 
10964c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10974c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10984c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10994c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11004c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
11014c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11024c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11034c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
11044c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11054c1414c8SBarry Smith       x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--;
11064c1414c8SBarry Smith       break;
11074c1414c8SBarry Smith     case 5 :
11084c1414c8SBarry Smith       sum1 = tmp[row];
11094c1414c8SBarry Smith       sum2 = tmp[row -1];
11104c1414c8SBarry Smith       sum3 = tmp[row -2];
11114c1414c8SBarry Smith       sum4 = tmp[row -3];
11124c1414c8SBarry Smith       sum5 = tmp[row -4];
11134c1414c8SBarry Smith       v2   = aa + ai[row]-1;
11144c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
11154c1414c8SBarry Smith       v4   = aa + ai[row -2]-1;
11164c1414c8SBarry Smith       v5   = aa + ai[row -3]-1;
11174c1414c8SBarry Smith       for (j=nz ; j>1; j-=2){
11184c1414c8SBarry Smith         vi  -= 2;
11194c1414c8SBarry Smith         i0   = vi[2];
11204c1414c8SBarry Smith         i1   = vi[1];
11214c1414c8SBarry Smith         tmp0 = tmps[i0];
11224c1414c8SBarry Smith         tmp1 = tmps[i1];
11234c1414c8SBarry Smith         v1   -= 2;
11244c1414c8SBarry Smith         v2   -= 2;
11254c1414c8SBarry Smith         v3   -= 2;
11264c1414c8SBarry Smith         v4   -= 2;
11274c1414c8SBarry Smith         v5   -= 2;
11284c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
11294c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
11304c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
11314c1414c8SBarry Smith         sum4 -= v4[2] * tmp0 + v4[1] * tmp1;
11324c1414c8SBarry Smith         sum5 -= v5[2] * tmp0 + v5[1] * tmp1;
11334c1414c8SBarry Smith       }
11344c1414c8SBarry Smith       if (j==1){
11354c1414c8SBarry Smith         tmp0  = tmps[*vi--];
11364c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
11374c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
11384c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
11394c1414c8SBarry Smith         sum4 -= *v4-- * tmp0;
11404c1414c8SBarry Smith         sum5 -= *v5-- * tmp0;
11414c1414c8SBarry Smith       }
11424c1414c8SBarry Smith 
11434c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
11444c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
11454c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11464c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11474c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11484c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
11494c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11504c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11514c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11524c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
11534c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11544c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11554c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--;
11564c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11574c1414c8SBarry Smith       x[*c--] = tmp[row] = sum5*a_a[ad[row]]; row--;
11584c1414c8SBarry Smith       break;
11594c1414c8SBarry Smith     default:
11604c1414c8SBarry Smith       SETERRQ(PETSC_ERR_COR,"Node size not yet supported \n");
11614c1414c8SBarry Smith     }
11624c1414c8SBarry Smith   }
11634c1414c8SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11644c1414c8SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1165d9fead3dSBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
11664c1414c8SBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1167dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
11684c1414c8SBarry Smith   PetscFunctionReturn(0);
11694c1414c8SBarry Smith }
11704c1414c8SBarry Smith 
1171*019b515eSShri Abhyankar /* ----------------------------------------------------------- */
1172*019b515eSShri Abhyankar #undef __FUNCT__
1173*019b515eSShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_Inode_newdatastruct"
1174*019b515eSShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_Inode_newdatastruct(Mat A,Vec bb,Vec xx)
1175*019b515eSShri Abhyankar {
1176*019b515eSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1177*019b515eSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1178*019b515eSShri Abhyankar   PetscErrorCode    ierr;
1179*019b515eSShri Abhyankar   const PetscInt    *r,*c,*rout,*cout;
1180*019b515eSShri Abhyankar   PetscInt          i,j,n = A->rmap->n,*ai = a->i,nz,*a_j = a->j;
1181*019b515eSShri Abhyankar   PetscInt          node_max,*ns,row,nsz,aii,*vi,*ad,*aj,i0,i1;
1182*019b515eSShri Abhyankar   PetscScalar       *x,*tmp,*tmps,tmp0,tmp1;
1183*019b515eSShri Abhyankar   PetscScalar       sum1,sum2,sum3,sum4,sum5;
1184*019b515eSShri Abhyankar   const MatScalar   *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa;
1185*019b515eSShri Abhyankar   const PetscScalar *b;
1186*019b515eSShri Abhyankar 
1187*019b515eSShri Abhyankar   PetscFunctionBegin;
1188*019b515eSShri Abhyankar   if (!a->inode.size) SETERRQ(PETSC_ERR_COR,"Missing Inode Structure");
1189*019b515eSShri Abhyankar   node_max = a->inode.node_count;
1190*019b515eSShri Abhyankar   ns       = a->inode.size;     /* Node Size array */
1191*019b515eSShri Abhyankar 
1192*019b515eSShri Abhyankar   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
1193*019b515eSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1194*019b515eSShri Abhyankar   tmp  = a->solve_work;
1195*019b515eSShri Abhyankar 
1196*019b515eSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1197*019b515eSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1198*019b515eSShri Abhyankar 
1199*019b515eSShri Abhyankar   /* forward solve the lower triangular */
1200*019b515eSShri Abhyankar   tmps = tmp ;
1201*019b515eSShri Abhyankar   aa   = a_a ;
1202*019b515eSShri Abhyankar   aj   = a_j ;
1203*019b515eSShri Abhyankar   ad   = a->diag;
1204*019b515eSShri Abhyankar 
1205*019b515eSShri Abhyankar   for (i = 0,row = 0; i< node_max; ++i){
1206*019b515eSShri Abhyankar     nsz = ns[i];
1207*019b515eSShri Abhyankar     aii = ai[row];
1208*019b515eSShri Abhyankar     v1  = aa + aii;
1209*019b515eSShri Abhyankar     vi  = aj + aii;
1210*019b515eSShri Abhyankar     nz  = ai[row+1]- ai[row];
1211*019b515eSShri Abhyankar 
1212*019b515eSShri Abhyankar     switch (nsz){               /* Each loop in 'case' is unrolled */
1213*019b515eSShri Abhyankar     case 1 :
1214*019b515eSShri Abhyankar       sum1 = b[r[row]];
1215*019b515eSShri Abhyankar       for(j=0; j<nz-1; j+=2){
1216*019b515eSShri Abhyankar         i0   = vi[j];
1217*019b515eSShri Abhyankar         i1   = vi[j+1];
1218*019b515eSShri Abhyankar         tmp0 = tmps[i0];
1219*019b515eSShri Abhyankar         tmp1 = tmps[i1];
1220*019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0 + v1[j+1]*tmp1;
1221*019b515eSShri Abhyankar       }
1222*019b515eSShri Abhyankar       if(j == nz-1){
1223*019b515eSShri Abhyankar         tmp0 = tmps[vi[j]];
1224*019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0;
1225*019b515eSShri Abhyankar       }
1226*019b515eSShri Abhyankar       tmp[row++]=sum1;
1227*019b515eSShri Abhyankar       break;
1228*019b515eSShri Abhyankar     case 2:
1229*019b515eSShri Abhyankar       sum1 = b[r[row]];
1230*019b515eSShri Abhyankar       sum2 = b[r[row+1]];
1231*019b515eSShri Abhyankar       v2   = aa + ai[row+1];
1232*019b515eSShri Abhyankar 
1233*019b515eSShri Abhyankar       for(j=0; j<nz-1; j+=2){
1234*019b515eSShri Abhyankar         i0   = vi[j];
1235*019b515eSShri Abhyankar         i1   = vi[j+1];
1236*019b515eSShri Abhyankar         tmp0 = tmps[i0];
1237*019b515eSShri Abhyankar         tmp1 = tmps[i1];
1238*019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1239*019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
1240*019b515eSShri Abhyankar       }
1241*019b515eSShri Abhyankar       if(j == nz-1){
1242*019b515eSShri Abhyankar         tmp0 = tmps[vi[j]];
1243*019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
1244*019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
1245*019b515eSShri Abhyankar       }
1246*019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
1247*019b515eSShri Abhyankar       tmp[row ++]=sum1;
1248*019b515eSShri Abhyankar       tmp[row ++]=sum2;
1249*019b515eSShri Abhyankar       break;
1250*019b515eSShri Abhyankar     case 3:
1251*019b515eSShri Abhyankar       sum1 = b[r[row]];
1252*019b515eSShri Abhyankar       sum2 = b[r[row+1]];
1253*019b515eSShri Abhyankar       sum3 = b[r[row+2]];
1254*019b515eSShri Abhyankar       v2   = aa + ai[row+1];
1255*019b515eSShri Abhyankar       v3   = aa + ai[row+2];
1256*019b515eSShri Abhyankar 
1257*019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2){
1258*019b515eSShri Abhyankar         i0   = vi[j];
1259*019b515eSShri Abhyankar         i1   = vi[j+1];
1260*019b515eSShri Abhyankar         tmp0 = tmps[i0];
1261*019b515eSShri Abhyankar         tmp1 = tmps[i1];
1262*019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1263*019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
1264*019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
1265*019b515eSShri Abhyankar       }
1266*019b515eSShri Abhyankar       if (j == nz-1){
1267*019b515eSShri Abhyankar         tmp0 = tmps[vi[j]];
1268*019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
1269*019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
1270*019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
1271*019b515eSShri Abhyankar       }
1272*019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
1273*019b515eSShri Abhyankar       sum3 -= v3[nz] * sum1;
1274*019b515eSShri Abhyankar       sum3 -= v3[nz+1] * sum2;
1275*019b515eSShri Abhyankar       tmp[row ++]=sum1;
1276*019b515eSShri Abhyankar       tmp[row ++]=sum2;
1277*019b515eSShri Abhyankar       tmp[row ++]=sum3;
1278*019b515eSShri Abhyankar       break;
1279*019b515eSShri Abhyankar 
1280*019b515eSShri Abhyankar     case 4:
1281*019b515eSShri Abhyankar       sum1 = b[r[row]];
1282*019b515eSShri Abhyankar       sum2 = b[r[row+1]];
1283*019b515eSShri Abhyankar       sum3 = b[r[row+2]];
1284*019b515eSShri Abhyankar       sum4 = b[r[row+3]];
1285*019b515eSShri Abhyankar       v2   = aa + ai[row+1];
1286*019b515eSShri Abhyankar       v3   = aa + ai[row+2];
1287*019b515eSShri Abhyankar       v4   = aa + ai[row+3];
1288*019b515eSShri Abhyankar 
1289*019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2){
1290*019b515eSShri Abhyankar         i0   = vi[j];
1291*019b515eSShri Abhyankar         i1   = vi[j+1];
1292*019b515eSShri Abhyankar         tmp0 = tmps[i0];
1293*019b515eSShri Abhyankar         tmp1 = tmps[i1];
1294*019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1295*019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
1296*019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
1297*019b515eSShri Abhyankar         sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1;
1298*019b515eSShri Abhyankar       }
1299*019b515eSShri Abhyankar       if (j == nz-1){
1300*019b515eSShri Abhyankar         tmp0 = tmps[vi[j]];
1301*019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
1302*019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
1303*019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
1304*019b515eSShri Abhyankar         sum4 -= v4[j] *tmp0;
1305*019b515eSShri Abhyankar       }
1306*019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
1307*019b515eSShri Abhyankar       sum3 -= v3[nz] * sum1;
1308*019b515eSShri Abhyankar       sum4 -= v4[nz] * sum1;
1309*019b515eSShri Abhyankar       sum3 -= v3[nz+1] * sum2;
1310*019b515eSShri Abhyankar       sum4 -= v4[nz+1] * sum2;
1311*019b515eSShri Abhyankar       sum4 -= v4[nz+2] * sum3;
1312*019b515eSShri Abhyankar 
1313*019b515eSShri Abhyankar       tmp[row ++]=sum1;
1314*019b515eSShri Abhyankar       tmp[row ++]=sum2;
1315*019b515eSShri Abhyankar       tmp[row ++]=sum3;
1316*019b515eSShri Abhyankar       tmp[row ++]=sum4;
1317*019b515eSShri Abhyankar       break;
1318*019b515eSShri Abhyankar     case 5:
1319*019b515eSShri Abhyankar       sum1 = b[r[row]];
1320*019b515eSShri Abhyankar       sum2 = b[r[row+1]];
1321*019b515eSShri Abhyankar       sum3 = b[r[row+2]];
1322*019b515eSShri Abhyankar       sum4 = b[r[row+3]];
1323*019b515eSShri Abhyankar       sum5 = b[r[row+4]];
1324*019b515eSShri Abhyankar       v2   = aa + ai[row+1];
1325*019b515eSShri Abhyankar       v3   = aa + ai[row+2];
1326*019b515eSShri Abhyankar       v4   = aa + ai[row+3];
1327*019b515eSShri Abhyankar       v5   = aa + ai[row+4];
1328*019b515eSShri Abhyankar 
1329*019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2){
1330*019b515eSShri Abhyankar         i0   = vi[j];
1331*019b515eSShri Abhyankar         i1   = vi[j+1];
1332*019b515eSShri Abhyankar         tmp0 = tmps[i0];
1333*019b515eSShri Abhyankar         tmp1 = tmps[i1];
1334*019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1335*019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
1336*019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
1337*019b515eSShri Abhyankar         sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1;
1338*019b515eSShri Abhyankar         sum5 -= v5[j] * tmp0 + v5[j+1] * tmp1;
1339*019b515eSShri Abhyankar       }
1340*019b515eSShri Abhyankar       if (j == nz-1){
1341*019b515eSShri Abhyankar         tmp0 = tmps[vi[j]];
1342*019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
1343*019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
1344*019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
1345*019b515eSShri Abhyankar         sum4 -= v4[j] *tmp0;
1346*019b515eSShri Abhyankar         sum5 -= v5[j] *tmp0;
1347*019b515eSShri Abhyankar       }
1348*019b515eSShri Abhyankar 
1349*019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
1350*019b515eSShri Abhyankar       sum3 -= v3[nz] * sum1;
1351*019b515eSShri Abhyankar       sum4 -= v4[nz] * sum1;
1352*019b515eSShri Abhyankar       sum5 -= v5[nz] * sum1;
1353*019b515eSShri Abhyankar       sum3 -= v3[nz+1] * sum2;
1354*019b515eSShri Abhyankar       sum4 -= v4[nz+1] * sum2;
1355*019b515eSShri Abhyankar       sum5 -= v5[nz+1] * sum2;
1356*019b515eSShri Abhyankar       sum4 -= v4[nz+2] * sum3;
1357*019b515eSShri Abhyankar       sum5 -= v5[nz+2] * sum3;
1358*019b515eSShri Abhyankar       sum5 -= v5[nz+3] * sum4;
1359*019b515eSShri Abhyankar 
1360*019b515eSShri Abhyankar       tmp[row ++]=sum1;
1361*019b515eSShri Abhyankar       tmp[row ++]=sum2;
1362*019b515eSShri Abhyankar       tmp[row ++]=sum3;
1363*019b515eSShri Abhyankar       tmp[row ++]=sum4;
1364*019b515eSShri Abhyankar       tmp[row ++]=sum5;
1365*019b515eSShri Abhyankar       break;
1366*019b515eSShri Abhyankar     default:
1367*019b515eSShri Abhyankar       SETERRQ(PETSC_ERR_COR,"Node size not yet supported \n");
1368*019b515eSShri Abhyankar     }
1369*019b515eSShri Abhyankar   }
1370*019b515eSShri Abhyankar   /* backward solve the upper triangular */
1371*019b515eSShri Abhyankar   for (i=node_max -1 ,row = n-1 ; i>=0; i--){
1372*019b515eSShri Abhyankar     nsz = ns[i];
1373*019b515eSShri Abhyankar     aii = ad[row+1] + 1;
1374*019b515eSShri Abhyankar     v1  = aa + aii;
1375*019b515eSShri Abhyankar     vi  = aj + aii;
1376*019b515eSShri Abhyankar     nz  = ad[row]- ad[row+1] - 1;
1377*019b515eSShri Abhyankar     switch (nsz){               /* Each loop in 'case' is unrolled */
1378*019b515eSShri Abhyankar     case 1 :
1379*019b515eSShri Abhyankar       sum1 = tmp[row];
1380*019b515eSShri Abhyankar 
1381*019b515eSShri Abhyankar       for(j=0 ; j<nz-1; j+=2){
1382*019b515eSShri Abhyankar         i0   = vi[j];
1383*019b515eSShri Abhyankar         i1   = vi[j+1];
1384*019b515eSShri Abhyankar         tmp0 = tmps[i0];
1385*019b515eSShri Abhyankar         tmp1 = tmps[i1];
1386*019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1387*019b515eSShri Abhyankar       }
1388*019b515eSShri Abhyankar       if (j == nz-1){
1389*019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
1390*019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0;
1391*019b515eSShri Abhyankar       }
1392*019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
1393*019b515eSShri Abhyankar       break;
1394*019b515eSShri Abhyankar     case 2 :
1395*019b515eSShri Abhyankar       sum1 = tmp[row];
1396*019b515eSShri Abhyankar       sum2 = tmp[row-1];
1397*019b515eSShri Abhyankar       v2   = aa + ad[row] + 1;
1398*019b515eSShri Abhyankar       for (j=0 ; j<nz-1; j+=2){
1399*019b515eSShri Abhyankar         i0   = vi[j];
1400*019b515eSShri Abhyankar         i1   = vi[j+1];
1401*019b515eSShri Abhyankar         tmp0 = tmps[i0];
1402*019b515eSShri Abhyankar         tmp1 = tmps[i1];
1403*019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1404*019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
1405*019b515eSShri Abhyankar       }
1406*019b515eSShri Abhyankar       if (j == nz-1){
1407*019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
1408*019b515eSShri Abhyankar         sum1 -= v1[j]* tmp0;
1409*019b515eSShri Abhyankar         sum2 -= v2[j+1]* tmp0;
1410*019b515eSShri Abhyankar       }
1411*019b515eSShri Abhyankar 
1412*019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
1413*019b515eSShri Abhyankar       sum2   -= v2[0] * tmp0;
1414*019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
1415*019b515eSShri Abhyankar       break;
1416*019b515eSShri Abhyankar     case 3 :
1417*019b515eSShri Abhyankar       sum1 = tmp[row];
1418*019b515eSShri Abhyankar       sum2 = tmp[row -1];
1419*019b515eSShri Abhyankar       sum3 = tmp[row -2];
1420*019b515eSShri Abhyankar       v2   = aa + ad[row] + 1;
1421*019b515eSShri Abhyankar       v3   = aa + ad[row -1] + 1;
1422*019b515eSShri Abhyankar       for (j=0 ; j<nz-1; j+=2){
1423*019b515eSShri Abhyankar         i0   = vi[j];
1424*019b515eSShri Abhyankar         i1   = vi[j+1];
1425*019b515eSShri Abhyankar         tmp0 = tmps[i0];
1426*019b515eSShri Abhyankar         tmp1 = tmps[i1];
1427*019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1428*019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
1429*019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
1430*019b515eSShri Abhyankar       }
1431*019b515eSShri Abhyankar       if (j== nz-1){
1432*019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
1433*019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
1434*019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
1435*019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
1436*019b515eSShri Abhyankar       }
1437*019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
1438*019b515eSShri Abhyankar       sum2   -= v2[0]* tmp0;
1439*019b515eSShri Abhyankar       sum3   -= v3[1] * tmp0;
1440*019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
1441*019b515eSShri Abhyankar       sum3   -= v3[0]* tmp0;
1442*019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
1443*019b515eSShri Abhyankar 
1444*019b515eSShri Abhyankar       break;
1445*019b515eSShri Abhyankar     case 4 :
1446*019b515eSShri Abhyankar       sum1 = tmp[row];
1447*019b515eSShri Abhyankar       sum2 = tmp[row -1];
1448*019b515eSShri Abhyankar       sum3 = tmp[row -2];
1449*019b515eSShri Abhyankar       sum4 = tmp[row -3];
1450*019b515eSShri Abhyankar       v2   = aa + ad[row]+1;
1451*019b515eSShri Abhyankar       v3   = aa + ad[row -1]+1;
1452*019b515eSShri Abhyankar       v4   = aa + ad[row -2]+1;
1453*019b515eSShri Abhyankar 
1454*019b515eSShri Abhyankar       for (j=0 ; j<nz-1; j+=2){
1455*019b515eSShri Abhyankar         i0   = vi[j];
1456*019b515eSShri Abhyankar         i1   = vi[j+1];
1457*019b515eSShri Abhyankar         tmp0 = tmps[i0];
1458*019b515eSShri Abhyankar         tmp1 = tmps[i1];
1459*019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0   + v1[j+1] * tmp1;
1460*019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
1461*019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
1462*019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1;
1463*019b515eSShri Abhyankar       }
1464*019b515eSShri Abhyankar       if (j== nz-1){
1465*019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
1466*019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
1467*019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
1468*019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
1469*019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0;
1470*019b515eSShri Abhyankar       }
1471*019b515eSShri Abhyankar 
1472*019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
1473*019b515eSShri Abhyankar       sum2   -= v2[0] * tmp0;
1474*019b515eSShri Abhyankar       sum3   -= v3[1] * tmp0;
1475*019b515eSShri Abhyankar       sum4   -= v4[2] * tmp0;
1476*019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
1477*019b515eSShri Abhyankar       sum3   -= v3[0] * tmp0;
1478*019b515eSShri Abhyankar       sum4   -= v4[1] * tmp0;
1479*019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
1480*019b515eSShri Abhyankar       sum4   -= v4[0] * tmp0;
1481*019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--;
1482*019b515eSShri Abhyankar       break;
1483*019b515eSShri Abhyankar     case 5 :
1484*019b515eSShri Abhyankar       sum1 = tmp[row];
1485*019b515eSShri Abhyankar       sum2 = tmp[row -1];
1486*019b515eSShri Abhyankar       sum3 = tmp[row -2];
1487*019b515eSShri Abhyankar       sum4 = tmp[row -3];
1488*019b515eSShri Abhyankar       sum5 = tmp[row -4];
1489*019b515eSShri Abhyankar       v2   = aa + ad[row]+1;
1490*019b515eSShri Abhyankar       v3   = aa + ad[row -1]+1;
1491*019b515eSShri Abhyankar       v4   = aa + ad[row -2]+1;
1492*019b515eSShri Abhyankar       v5   = aa + ad[row -3]+1;
1493*019b515eSShri Abhyankar       for (j=0 ; j<nz-1; j+=2){
1494*019b515eSShri Abhyankar         i0   = vi[j];
1495*019b515eSShri Abhyankar         i1   = vi[j+1];
1496*019b515eSShri Abhyankar         tmp0 = tmps[i0];
1497*019b515eSShri Abhyankar         tmp1 = tmps[i1];
1498*019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1499*019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
1500*019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
1501*019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1;
1502*019b515eSShri Abhyankar         sum5 -= v5[j+4] * tmp0 + v5[j+5] * tmp1;
1503*019b515eSShri Abhyankar       }
1504*019b515eSShri Abhyankar       if (j==nz-1){
1505*019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
1506*019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
1507*019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
1508*019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
1509*019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0;
1510*019b515eSShri Abhyankar         sum5 -= v5[j+4] * tmp0;
1511*019b515eSShri Abhyankar       }
1512*019b515eSShri Abhyankar 
1513*019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
1514*019b515eSShri Abhyankar       sum2   -= v2[0] * tmp0;
1515*019b515eSShri Abhyankar       sum3   -= v3[1] * tmp0;
1516*019b515eSShri Abhyankar       sum4   -= v4[2] * tmp0;
1517*019b515eSShri Abhyankar       sum5   -= v5[3] * tmp0;
1518*019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
1519*019b515eSShri Abhyankar       sum3   -= v3[0] * tmp0;
1520*019b515eSShri Abhyankar       sum4   -= v4[1] * tmp0;
1521*019b515eSShri Abhyankar       sum5   -= v5[2] * tmp0;
1522*019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
1523*019b515eSShri Abhyankar       sum4   -= v4[0] * tmp0;
1524*019b515eSShri Abhyankar       sum5   -= v5[1] * tmp0;
1525*019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--;
1526*019b515eSShri Abhyankar       sum5   -= v5[0] * tmp0;
1527*019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum5*v5[nz+4]; row--;
1528*019b515eSShri Abhyankar       break;
1529*019b515eSShri Abhyankar     default:
1530*019b515eSShri Abhyankar       SETERRQ(PETSC_ERR_COR,"Node size not yet supported \n");
1531*019b515eSShri Abhyankar     }
1532*019b515eSShri Abhyankar   }
1533*019b515eSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1534*019b515eSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1535*019b515eSShri Abhyankar   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
1536*019b515eSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1537*019b515eSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1538*019b515eSShri Abhyankar   PetscFunctionReturn(0);
1539*019b515eSShri Abhyankar }
1540*019b515eSShri Abhyankar 
1541*019b515eSShri Abhyankar 
15424c1414c8SBarry Smith /*
15434c1414c8SBarry Smith      Makes a longer coloring[] array and calls the usual code with that
15444c1414c8SBarry Smith */
15454c1414c8SBarry Smith #undef __FUNCT__
15464108e4d5SBarry Smith #define __FUNCT__ "MatColoringPatch_SeqAIJ_Inode"
15474108e4d5SBarry Smith PetscErrorCode MatColoringPatch_SeqAIJ_Inode(Mat mat,PetscInt ncolors,PetscInt nin,ISColoringValue coloring[],ISColoring *iscoloring)
15484c1414c8SBarry Smith {
15494c1414c8SBarry Smith   Mat_SeqAIJ       *a = (Mat_SeqAIJ*)mat->data;
15504c1414c8SBarry Smith   PetscErrorCode  ierr;
1551d0f46423SBarry Smith   PetscInt        n = mat->cmap->n,m = a->inode.node_count,j,*ns = a->inode.size,row;
15524c1414c8SBarry Smith   PetscInt        *colorused,i;
15534c1414c8SBarry Smith   ISColoringValue *newcolor;
15544c1414c8SBarry Smith 
15554c1414c8SBarry Smith   PetscFunctionBegin;
15564c1414c8SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&newcolor);CHKERRQ(ierr);
15574c1414c8SBarry Smith   /* loop over inodes, marking a color for each column*/
15584c1414c8SBarry Smith   row = 0;
15594c1414c8SBarry Smith   for (i=0; i<m; i++){
15604c1414c8SBarry Smith     for (j=0; j<ns[i]; j++) {
15614c1414c8SBarry Smith       newcolor[row++] = coloring[i] + j*ncolors;
15624c1414c8SBarry Smith     }
15634c1414c8SBarry Smith   }
15644c1414c8SBarry Smith 
15654c1414c8SBarry Smith   /* eliminate unneeded colors */
15664c1414c8SBarry Smith   ierr = PetscMalloc(5*ncolors*sizeof(PetscInt),&colorused);CHKERRQ(ierr);
15674c1414c8SBarry Smith   ierr = PetscMemzero(colorused,5*ncolors*sizeof(PetscInt));CHKERRQ(ierr);
15684c1414c8SBarry Smith   for (i=0; i<n; i++) {
15694c1414c8SBarry Smith     colorused[newcolor[i]] = 1;
15704c1414c8SBarry Smith   }
15714c1414c8SBarry Smith 
15724c1414c8SBarry Smith   for (i=1; i<5*ncolors; i++) {
15734c1414c8SBarry Smith     colorused[i] += colorused[i-1];
15744c1414c8SBarry Smith   }
15754c1414c8SBarry Smith   ncolors = colorused[5*ncolors-1];
15764c1414c8SBarry Smith   for (i=0; i<n; i++) {
157757815681SSatish Balay     newcolor[i] = colorused[newcolor[i]]-1;
15784c1414c8SBarry Smith   }
15794c1414c8SBarry Smith   ierr = PetscFree(colorused);CHKERRQ(ierr);
15807adad957SLisandro Dalcin   ierr = ISColoringCreate(((PetscObject)mat)->comm,ncolors,n,newcolor,iscoloring);CHKERRQ(ierr);
15814c1414c8SBarry Smith   ierr = PetscFree(coloring);CHKERRQ(ierr);
15824c1414c8SBarry Smith   PetscFunctionReturn(0);
15834c1414c8SBarry Smith }
15844c1414c8SBarry Smith 
1585c60f0209SBarry Smith #include "../src/mat/blockinvert.h"
15862af78befSBarry Smith 
15872af78befSBarry Smith #undef __FUNCT__
15884108e4d5SBarry Smith #define __FUNCT__ "MatSOR_SeqAIJ_Inode"
15894108e4d5SBarry Smith PetscErrorCode MatSOR_SeqAIJ_Inode(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx)
15902af78befSBarry Smith {
15912af78befSBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
1592938d4eb3SBarry Smith   PetscScalar        sum1,sum2,sum3,sum4,sum5,tmp0,tmp1,tmp2,tmp3;
1593a77337e4SBarry Smith   MatScalar          *ibdiag,*bdiag;
1594938d4eb3SBarry Smith   PetscScalar        *x,*xb,tmp4,tmp5,x1,x2,x3,x4,x5;
1595dd6ea824SBarry Smith   const MatScalar    *v = a->a,*v1,*v2,*v3,*v4,*v5;
159662bba022SBarry Smith   PetscReal          zeropivot = 1.0e-15, shift = 0.0;
15972af78befSBarry Smith   PetscErrorCode     ierr;
1598f0d39aaaSBarry Smith   PetscInt           n,m = a->inode.node_count,*sizes = a->inode.size,cnt = 0,i,j,row,i1,i2;
1599f0d39aaaSBarry Smith   PetscInt           *idx,*diag = a->diag,*ii = a->i,sz,k;
16002af78befSBarry Smith 
16012af78befSBarry Smith   PetscFunctionBegin;
160271f1c65dSBarry Smith   if (omega != 1.0) SETERRQ(PETSC_ERR_SUP,"No support for omega != 1.0; use -mat_no_inode");
160371f1c65dSBarry Smith   if (fshift != 0.0) SETERRQ(PETSC_ERR_SUP,"No support for fshift != 0.0; use -mat_no_inode");
16043f6d85c9SBarry Smith   if (its > 1) {
16053f6d85c9SBarry Smith     /* switch to non-inode version */
160641f059aeSBarry Smith     ierr = MatSOR_SeqAIJ(A,bb,omega,flag,fshift,its,lits,xx);CHKERRQ(ierr);
16073f6d85c9SBarry Smith     PetscFunctionReturn(0);
16083f6d85c9SBarry Smith   }
16092af78befSBarry Smith 
161071f1c65dSBarry Smith   if (!a->inode.ibdiagvalid) {
16112af78befSBarry Smith     if (!a->inode.ibdiag) {
16122af78befSBarry Smith       /* calculate space needed for diagonal blocks */
16132af78befSBarry Smith       for (i=0; i<m; i++) {
16142af78befSBarry Smith 	cnt += sizes[i]*sizes[i];
16152af78befSBarry Smith       }
1616f0d39aaaSBarry Smith       a->inode.bdiagsize = cnt;
161789c6957cSBarry Smith       ierr   = PetscMalloc3(cnt,MatScalar,&a->inode.ibdiag,cnt,MatScalar,&a->inode.bdiag,A->rmap->n,MatScalar,&a->inode.ssor_work);CHKERRQ(ierr);
161871f1c65dSBarry Smith     }
161971f1c65dSBarry Smith 
162071f1c65dSBarry Smith     /* copy over the diagonal blocks and invert them */
16212af78befSBarry Smith     ibdiag = a->inode.ibdiag;
16222af78befSBarry Smith     bdiag  = a->inode.bdiag;
16232af78befSBarry Smith     cnt = 0;
16242af78befSBarry Smith     for (i=0, row = 0; i<m; i++) {
16252af78befSBarry Smith       for (j=0; j<sizes[i]; j++) {
1626f0d39aaaSBarry Smith         for (k=0; k<sizes[i]; k++) {
1627f0d39aaaSBarry Smith           bdiag[cnt+k*sizes[i]+j] = v[diag[row+j] - j + k];
1628f0d39aaaSBarry Smith         }
16292af78befSBarry Smith       }
1630a77337e4SBarry Smith       ierr = PetscMemcpy(ibdiag+cnt,bdiag+cnt,sizes[i]*sizes[i]*sizeof(MatScalar));CHKERRQ(ierr);
16312af78befSBarry Smith 
16322af78befSBarry Smith       switch(sizes[i]) {
16332af78befSBarry Smith         case 1:
16342af78befSBarry Smith           /* Create matrix data structure */
163564c62002SMatthew Knepley           if (PetscAbsScalar(ibdiag[cnt]) < zeropivot) SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot on row %D",row);
163664c62002SMatthew Knepley           ibdiag[cnt] = 1.0/ibdiag[cnt];
16372af78befSBarry Smith           break;
16382af78befSBarry Smith         case 2:
163962bba022SBarry Smith           ierr = Kernel_A_gets_inverse_A_2(ibdiag+cnt,shift);CHKERRQ(ierr);
16402af78befSBarry Smith           break;
16412af78befSBarry Smith         case 3:
164262bba022SBarry Smith           ierr = Kernel_A_gets_inverse_A_3(ibdiag+cnt,shift);CHKERRQ(ierr);
16432af78befSBarry Smith           break;
16442af78befSBarry Smith         case 4:
164562bba022SBarry Smith           ierr = Kernel_A_gets_inverse_A_4(ibdiag+cnt,shift);CHKERRQ(ierr);
16462af78befSBarry Smith           break;
16472af78befSBarry Smith         case 5:
164862bba022SBarry Smith           ierr = Kernel_A_gets_inverse_A_5(ibdiag+cnt,shift);CHKERRQ(ierr);
16492af78befSBarry Smith           break;
16502af78befSBarry Smith        default:
16512af78befSBarry Smith 	 SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
16522af78befSBarry Smith       }
16532af78befSBarry Smith       cnt += sizes[i]*sizes[i];
16542af78befSBarry Smith       row += sizes[i];
16552af78befSBarry Smith     }
165671f1c65dSBarry Smith     a->inode.ibdiagvalid = PETSC_TRUE;
16572af78befSBarry Smith   }
16582af78befSBarry Smith   ibdiag = a->inode.ibdiag;
16592af78befSBarry Smith   bdiag  = a->inode.bdiag;
16602af78befSBarry Smith 
1661938d4eb3SBarry Smith 
1662938d4eb3SBarry Smith   /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */
1663938d4eb3SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
1664938d4eb3SBarry Smith     PetscScalar *b;
16652af78befSBarry Smith     ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
16662af78befSBarry Smith     if (xx != bb) {
16672af78befSBarry Smith       ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
16682af78befSBarry Smith     } else {
16692af78befSBarry Smith       b = x;
16702af78befSBarry Smith     }
16712af78befSBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
16722af78befSBarry Smith 
16738862d2efSBarry Smith       for (i=0, row=0; i<m; i++) {
16748862d2efSBarry Smith         sz  = diag[row] - ii[row];
16758862d2efSBarry Smith         v1  = a->a + ii[row];
16768862d2efSBarry Smith         idx = a->j + ii[row];
16778862d2efSBarry Smith 
16784108e4d5SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
16798862d2efSBarry Smith         switch (sizes[i]){
16808862d2efSBarry Smith           case 1:
16818862d2efSBarry Smith 
16828862d2efSBarry Smith             sum1  = b[row];
16838862d2efSBarry Smith             for(n = 0; n<sz-1; n+=2) {
16848862d2efSBarry Smith               i1   = idx[0];
16858862d2efSBarry Smith               i2   = idx[1];
16868862d2efSBarry Smith               idx += 2;
16878862d2efSBarry Smith               tmp0 = x[i1];
16888862d2efSBarry Smith               tmp1 = x[i2];
16898862d2efSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
16908862d2efSBarry Smith             }
16918862d2efSBarry Smith 
16928862d2efSBarry Smith             if (n == sz-1){
1693f0d39aaaSBarry Smith               tmp0  = x[*idx];
1694f0d39aaaSBarry Smith               sum1 -= *v1 * tmp0;
16958862d2efSBarry Smith             }
16968862d2efSBarry Smith             x[row++] = sum1*(*ibdiag++);
16978862d2efSBarry Smith             break;
1698f0d39aaaSBarry Smith           case 2:
1699f0d39aaaSBarry Smith             v2    = a->a + ii[row+1];
1700f0d39aaaSBarry Smith             sum1  = b[row];
1701f0d39aaaSBarry Smith             sum2  = b[row+1];
1702f0d39aaaSBarry Smith             for(n = 0; n<sz-1; n+=2) {
1703f0d39aaaSBarry Smith               i1   = idx[0];
1704f0d39aaaSBarry Smith               i2   = idx[1];
1705f0d39aaaSBarry Smith               idx += 2;
1706f0d39aaaSBarry Smith               tmp0 = x[i1];
1707f0d39aaaSBarry Smith               tmp1 = x[i2];
1708f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
1709f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
1710f0d39aaaSBarry Smith             }
1711f0d39aaaSBarry Smith 
1712f0d39aaaSBarry Smith             if (n == sz-1){
1713f0d39aaaSBarry Smith               tmp0  = x[*idx];
1714f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0;
1715f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0;
1716f0d39aaaSBarry Smith             }
1717f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[0] + sum2*ibdiag[2];
1718f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[1] + sum2*ibdiag[3];
1719f0d39aaaSBarry Smith             ibdiag  += 4;
1720f0d39aaaSBarry Smith             break;
1721f0d39aaaSBarry Smith           case 3:
1722f0d39aaaSBarry Smith             v2    = a->a + ii[row+1];
1723f0d39aaaSBarry Smith             v3    = a->a + ii[row+2];
1724f0d39aaaSBarry Smith             sum1  = b[row];
1725f0d39aaaSBarry Smith             sum2  = b[row+1];
1726f0d39aaaSBarry Smith             sum3  = b[row+2];
1727f0d39aaaSBarry Smith             for(n = 0; n<sz-1; n+=2) {
1728f0d39aaaSBarry Smith               i1   = idx[0];
1729f0d39aaaSBarry Smith               i2   = idx[1];
1730f0d39aaaSBarry Smith               idx += 2;
1731f0d39aaaSBarry Smith               tmp0 = x[i1];
1732f0d39aaaSBarry Smith               tmp1 = x[i2];
1733f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
1734f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
1735f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
1736f0d39aaaSBarry Smith             }
1737f0d39aaaSBarry Smith 
1738f0d39aaaSBarry Smith             if (n == sz-1){
1739f0d39aaaSBarry Smith               tmp0  = x[*idx];
1740f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0;
1741f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0;
1742f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0;
1743f0d39aaaSBarry Smith             }
1744f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
1745f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
1746f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
1747f0d39aaaSBarry Smith             ibdiag  += 9;
1748f0d39aaaSBarry Smith             break;
1749f0d39aaaSBarry Smith           case 4:
1750f0d39aaaSBarry Smith             v2    = a->a + ii[row+1];
1751f0d39aaaSBarry Smith             v3    = a->a + ii[row+2];
1752f0d39aaaSBarry Smith             v4    = a->a + ii[row+3];
1753f0d39aaaSBarry Smith             sum1  = b[row];
1754f0d39aaaSBarry Smith             sum2  = b[row+1];
1755f0d39aaaSBarry Smith             sum3  = b[row+2];
1756f0d39aaaSBarry Smith             sum4  = b[row+3];
1757f0d39aaaSBarry Smith             for(n = 0; n<sz-1; n+=2) {
1758f0d39aaaSBarry Smith               i1   = idx[0];
1759f0d39aaaSBarry Smith               i2   = idx[1];
1760f0d39aaaSBarry Smith               idx += 2;
1761f0d39aaaSBarry Smith               tmp0 = x[i1];
1762f0d39aaaSBarry Smith               tmp1 = x[i2];
1763f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
1764f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
1765f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
1766f0d39aaaSBarry Smith               sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
1767f0d39aaaSBarry Smith             }
1768f0d39aaaSBarry Smith 
1769f0d39aaaSBarry Smith             if (n == sz-1){
1770f0d39aaaSBarry Smith               tmp0  = x[*idx];
1771f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0;
1772f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0;
1773f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0;
1774f0d39aaaSBarry Smith               sum4 -= v4[0] * tmp0;
1775f0d39aaaSBarry Smith             }
1776f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
1777f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
1778f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
1779f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
1780f0d39aaaSBarry Smith             ibdiag  += 16;
1781f0d39aaaSBarry Smith             break;
1782f0d39aaaSBarry Smith           case 5:
1783f0d39aaaSBarry Smith             v2    = a->a + ii[row+1];
1784f0d39aaaSBarry Smith             v3    = a->a + ii[row+2];
1785f0d39aaaSBarry Smith             v4    = a->a + ii[row+3];
1786f0d39aaaSBarry Smith             v5    = a->a + ii[row+4];
1787f0d39aaaSBarry Smith             sum1  = b[row];
1788f0d39aaaSBarry Smith             sum2  = b[row+1];
1789f0d39aaaSBarry Smith             sum3  = b[row+2];
1790f0d39aaaSBarry Smith             sum4  = b[row+3];
1791f0d39aaaSBarry Smith             sum5  = b[row+4];
1792f0d39aaaSBarry Smith             for(n = 0; n<sz-1; n+=2) {
1793f0d39aaaSBarry Smith               i1   = idx[0];
1794f0d39aaaSBarry Smith               i2   = idx[1];
1795f0d39aaaSBarry Smith               idx += 2;
1796f0d39aaaSBarry Smith               tmp0 = x[i1];
1797f0d39aaaSBarry Smith               tmp1 = x[i2];
1798f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
1799f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
1800f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
1801f0d39aaaSBarry Smith               sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
1802f0d39aaaSBarry Smith               sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
1803f0d39aaaSBarry Smith             }
1804f0d39aaaSBarry Smith 
1805f0d39aaaSBarry Smith             if (n == sz-1){
1806f0d39aaaSBarry Smith               tmp0  = x[*idx];
1807f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0;
1808f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0;
1809f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0;
1810f0d39aaaSBarry Smith               sum4 -= v4[0] * tmp0;
1811f0d39aaaSBarry Smith               sum5 -= v5[0] * tmp0;
1812f0d39aaaSBarry Smith             }
1813f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
1814f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
1815f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
1816f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
1817f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
1818f0d39aaaSBarry Smith             ibdiag  += 25;
1819f0d39aaaSBarry Smith             break;
18208862d2efSBarry Smith 	  default:
18218862d2efSBarry Smith    	    SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
18228862d2efSBarry Smith         }
18232af78befSBarry Smith       }
18242af78befSBarry Smith 
18252af78befSBarry Smith       xb = x;
18262af78befSBarry Smith       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
18272af78befSBarry Smith     } else xb = b;
18282af78befSBarry Smith     if ((flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) &&
18292af78befSBarry Smith         (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP)) {
18308862d2efSBarry Smith       cnt = 0;
18318862d2efSBarry Smith       for (i=0, row=0; i<m; i++) {
18322af78befSBarry Smith 
18338862d2efSBarry Smith         switch (sizes[i]){
18348862d2efSBarry Smith           case 1:
18358862d2efSBarry Smith             x[row++] *= bdiag[cnt++];
18368862d2efSBarry Smith             break;
1837f0d39aaaSBarry Smith           case 2:
1838f0d39aaaSBarry Smith             x1   = x[row]; x2 = x[row+1];
1839f0d39aaaSBarry Smith             tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2];
1840f0d39aaaSBarry Smith             tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3];
1841f0d39aaaSBarry Smith             x[row++] = tmp1;
1842f0d39aaaSBarry Smith             x[row++] = tmp2;
1843f0d39aaaSBarry Smith             cnt += 4;
1844f0d39aaaSBarry Smith             break;
1845f0d39aaaSBarry Smith           case 3:
1846f0d39aaaSBarry Smith             x1   = x[row]; x2 = x[row+1]; x3 = x[row+2];
1847f0d39aaaSBarry Smith             tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6];
1848f0d39aaaSBarry Smith             tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7];
1849f0d39aaaSBarry Smith             tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8];
1850f0d39aaaSBarry Smith             x[row++] = tmp1;
1851f0d39aaaSBarry Smith             x[row++] = tmp2;
1852f0d39aaaSBarry Smith             x[row++] = tmp3;
1853f0d39aaaSBarry Smith             cnt += 9;
1854f0d39aaaSBarry Smith             break;
1855f0d39aaaSBarry Smith           case 4:
1856f0d39aaaSBarry Smith             x1   = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3];
1857f0d39aaaSBarry Smith             tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12];
1858f0d39aaaSBarry Smith             tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13];
1859f0d39aaaSBarry Smith             tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14];
1860f0d39aaaSBarry Smith             tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15];
1861f0d39aaaSBarry Smith             x[row++] = tmp1;
1862f0d39aaaSBarry Smith             x[row++] = tmp2;
1863f0d39aaaSBarry Smith             x[row++] = tmp3;
1864f0d39aaaSBarry Smith             x[row++] = tmp4;
1865f0d39aaaSBarry Smith             cnt += 16;
1866f0d39aaaSBarry Smith             break;
1867f0d39aaaSBarry Smith           case 5:
1868f0d39aaaSBarry Smith             x1   = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; x5 = x[row+4];
1869f0d39aaaSBarry Smith             tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20];
1870f0d39aaaSBarry Smith             tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21];
1871f0d39aaaSBarry Smith             tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22];
1872f0d39aaaSBarry Smith             tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23];
1873f0d39aaaSBarry Smith             tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24];
1874f0d39aaaSBarry Smith             x[row++] = tmp1;
1875f0d39aaaSBarry Smith             x[row++] = tmp2;
1876f0d39aaaSBarry Smith             x[row++] = tmp3;
1877f0d39aaaSBarry Smith             x[row++] = tmp4;
1878f0d39aaaSBarry Smith             x[row++] = tmp5;
1879f0d39aaaSBarry Smith             cnt += 25;
1880f0d39aaaSBarry Smith             break;
18818862d2efSBarry Smith 	  default:
18828862d2efSBarry Smith    	    SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
18838862d2efSBarry Smith         }
18842af78befSBarry Smith       }
18852af78befSBarry Smith       ierr = PetscLogFlops(m);CHKERRQ(ierr);
18862af78befSBarry Smith     }
18872af78befSBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
18882af78befSBarry Smith 
1889f0d39aaaSBarry Smith       ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
1890d0f46423SBarry Smith       for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
1891f0d39aaaSBarry Smith         ibdiag -= sizes[i]*sizes[i];
18928862d2efSBarry Smith         sz      = ii[row+1] - diag[row] - 1;
18938862d2efSBarry Smith         v1      = a->a + diag[row] + 1;
18948862d2efSBarry Smith         idx     = a->j + diag[row] + 1;
18952af78befSBarry Smith 
18964108e4d5SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
18978862d2efSBarry Smith         switch (sizes[i]){
18988862d2efSBarry Smith           case 1:
18998862d2efSBarry Smith 
19008862d2efSBarry Smith             sum1  = xb[row];
19018862d2efSBarry Smith             for(n = 0; n<sz-1; n+=2) {
19028862d2efSBarry Smith               i1   = idx[0];
19038862d2efSBarry Smith               i2   = idx[1];
19048862d2efSBarry Smith               idx += 2;
19058862d2efSBarry Smith               tmp0 = x[i1];
19068862d2efSBarry Smith               tmp1 = x[i2];
19078862d2efSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
19088862d2efSBarry Smith             }
19098862d2efSBarry Smith 
19108862d2efSBarry Smith             if (n == sz-1){
1911f0d39aaaSBarry Smith               tmp0  = x[*idx];
1912f0d39aaaSBarry Smith               sum1 -= *v1*tmp0;
19138862d2efSBarry Smith             }
1914f0d39aaaSBarry Smith             x[row--] = sum1*(*ibdiag);
1915f0d39aaaSBarry Smith             break;
1916f0d39aaaSBarry Smith 
1917f0d39aaaSBarry Smith           case 2:
1918f0d39aaaSBarry Smith 
1919f0d39aaaSBarry Smith             sum1  = xb[row];
1920f0d39aaaSBarry Smith             sum2  = xb[row-1];
1921f0d39aaaSBarry Smith             /* note that sum1 is associated with the second of the two rows */
1922f0d39aaaSBarry Smith             v2    = a->a + diag[row-1] + 2;
1923f0d39aaaSBarry Smith             for(n = 0; n<sz-1; n+=2) {
1924f0d39aaaSBarry Smith               i1   = idx[0];
1925f0d39aaaSBarry Smith               i2   = idx[1];
1926f0d39aaaSBarry Smith               idx += 2;
1927f0d39aaaSBarry Smith               tmp0 = x[i1];
1928f0d39aaaSBarry Smith               tmp1 = x[i2];
1929f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
1930f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
1931f0d39aaaSBarry Smith             }
1932f0d39aaaSBarry Smith 
1933f0d39aaaSBarry Smith             if (n == sz-1){
1934f0d39aaaSBarry Smith               tmp0  = x[*idx];
1935f0d39aaaSBarry Smith               sum1 -= *v1*tmp0;
1936f0d39aaaSBarry Smith               sum2 -= *v2*tmp0;
1937f0d39aaaSBarry Smith             }
1938f0d39aaaSBarry Smith             x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3];
1939f0d39aaaSBarry Smith             x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2];
1940f0d39aaaSBarry Smith             break;
1941f0d39aaaSBarry Smith           case 3:
1942f0d39aaaSBarry Smith 
1943f0d39aaaSBarry Smith             sum1  = xb[row];
1944f0d39aaaSBarry Smith             sum2  = xb[row-1];
1945f0d39aaaSBarry Smith             sum3  = xb[row-2];
1946f0d39aaaSBarry Smith             v2    = a->a + diag[row-1] + 2;
1947f0d39aaaSBarry Smith             v3    = a->a + diag[row-2] + 3;
1948f0d39aaaSBarry Smith             for(n = 0; n<sz-1; n+=2) {
1949f0d39aaaSBarry Smith               i1   = idx[0];
1950f0d39aaaSBarry Smith               i2   = idx[1];
1951f0d39aaaSBarry Smith               idx += 2;
1952f0d39aaaSBarry Smith               tmp0 = x[i1];
1953f0d39aaaSBarry Smith               tmp1 = x[i2];
1954f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
1955f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
1956f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
1957f0d39aaaSBarry Smith             }
1958f0d39aaaSBarry Smith 
1959f0d39aaaSBarry Smith             if (n == sz-1){
1960f0d39aaaSBarry Smith               tmp0  = x[*idx];
1961f0d39aaaSBarry Smith               sum1 -= *v1*tmp0;
1962f0d39aaaSBarry Smith               sum2 -= *v2*tmp0;
1963f0d39aaaSBarry Smith               sum3 -= *v3*tmp0;
1964f0d39aaaSBarry Smith             }
1965f0d39aaaSBarry Smith             x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
1966f0d39aaaSBarry Smith             x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
1967f0d39aaaSBarry Smith             x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
1968f0d39aaaSBarry Smith             break;
1969f0d39aaaSBarry Smith           case 4:
1970f0d39aaaSBarry Smith 
1971f0d39aaaSBarry Smith             sum1  = xb[row];
1972f0d39aaaSBarry Smith             sum2  = xb[row-1];
1973f0d39aaaSBarry Smith             sum3  = xb[row-2];
1974f0d39aaaSBarry Smith             sum4  = xb[row-3];
1975f0d39aaaSBarry Smith             v2    = a->a + diag[row-1] + 2;
1976f0d39aaaSBarry Smith             v3    = a->a + diag[row-2] + 3;
1977f0d39aaaSBarry Smith             v4    = a->a + diag[row-3] + 4;
1978f0d39aaaSBarry Smith             for(n = 0; n<sz-1; n+=2) {
1979f0d39aaaSBarry Smith               i1   = idx[0];
1980f0d39aaaSBarry Smith               i2   = idx[1];
1981f0d39aaaSBarry Smith               idx += 2;
1982f0d39aaaSBarry Smith               tmp0 = x[i1];
1983f0d39aaaSBarry Smith               tmp1 = x[i2];
1984f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
1985f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
1986f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
1987f0d39aaaSBarry Smith               sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
1988f0d39aaaSBarry Smith             }
1989f0d39aaaSBarry Smith 
1990f0d39aaaSBarry Smith             if (n == sz-1){
1991f0d39aaaSBarry Smith               tmp0  = x[*idx];
1992f0d39aaaSBarry Smith               sum1 -= *v1*tmp0;
1993f0d39aaaSBarry Smith               sum2 -= *v2*tmp0;
1994f0d39aaaSBarry Smith               sum3 -= *v3*tmp0;
1995f0d39aaaSBarry Smith               sum4 -= *v4*tmp0;
1996f0d39aaaSBarry Smith             }
1997f0d39aaaSBarry Smith             x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
1998f0d39aaaSBarry Smith             x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
1999f0d39aaaSBarry Smith             x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
2000f0d39aaaSBarry Smith             x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
2001f0d39aaaSBarry Smith             break;
2002f0d39aaaSBarry Smith           case 5:
2003f0d39aaaSBarry Smith 
2004f0d39aaaSBarry Smith             sum1  = xb[row];
2005f0d39aaaSBarry Smith             sum2  = xb[row-1];
2006f0d39aaaSBarry Smith             sum3  = xb[row-2];
2007f0d39aaaSBarry Smith             sum4  = xb[row-3];
2008f0d39aaaSBarry Smith             sum5  = xb[row-4];
2009f0d39aaaSBarry Smith             v2    = a->a + diag[row-1] + 2;
2010f0d39aaaSBarry Smith             v3    = a->a + diag[row-2] + 3;
2011f0d39aaaSBarry Smith             v4    = a->a + diag[row-3] + 4;
2012f0d39aaaSBarry Smith             v5    = a->a + diag[row-4] + 5;
2013f0d39aaaSBarry Smith             for(n = 0; n<sz-1; n+=2) {
2014f0d39aaaSBarry Smith               i1   = idx[0];
2015f0d39aaaSBarry Smith               i2   = idx[1];
2016f0d39aaaSBarry Smith               idx += 2;
2017f0d39aaaSBarry Smith               tmp0 = x[i1];
2018f0d39aaaSBarry Smith               tmp1 = x[i2];
2019f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2020f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2021f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2022f0d39aaaSBarry Smith               sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
2023f0d39aaaSBarry Smith               sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
2024f0d39aaaSBarry Smith             }
2025f0d39aaaSBarry Smith 
2026f0d39aaaSBarry Smith             if (n == sz-1){
2027f0d39aaaSBarry Smith               tmp0  = x[*idx];
2028f0d39aaaSBarry Smith               sum1 -= *v1*tmp0;
2029f0d39aaaSBarry Smith               sum2 -= *v2*tmp0;
2030f0d39aaaSBarry Smith               sum3 -= *v3*tmp0;
2031f0d39aaaSBarry Smith               sum4 -= *v4*tmp0;
2032f0d39aaaSBarry Smith               sum5 -= *v5*tmp0;
2033f0d39aaaSBarry Smith             }
2034f0d39aaaSBarry Smith             x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
2035f0d39aaaSBarry Smith             x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
2036f0d39aaaSBarry Smith             x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
2037f0d39aaaSBarry Smith             x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
2038f0d39aaaSBarry Smith             x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
20398862d2efSBarry Smith             break;
20408862d2efSBarry Smith 	  default:
20418862d2efSBarry Smith    	    SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
20428862d2efSBarry Smith         }
20432af78befSBarry Smith       }
20442af78befSBarry Smith 
20452af78befSBarry Smith       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
20462af78befSBarry Smith     }
20472af78befSBarry Smith     its--;
2048938d4eb3SBarry Smith     ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
2049938d4eb3SBarry Smith     if (bb != xx) {ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);}
20502af78befSBarry Smith   }
205189c6957cSBarry Smith   if (flag & SOR_EISENSTAT) {
2052938d4eb3SBarry Smith     const PetscScalar *b;
20536cf45366SBarry Smith     MatScalar         *t = a->inode.ssor_work;
20546cf45366SBarry Smith 
2055938d4eb3SBarry Smith     ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2056938d4eb3SBarry Smith     ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
205789c6957cSBarry Smith     /*
205889c6957cSBarry Smith           Apply  (U + D)^-1  where D is now the block diagonal
205989c6957cSBarry Smith     */
206089c6957cSBarry Smith     ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
206189c6957cSBarry Smith     for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
206289c6957cSBarry Smith       ibdiag -= sizes[i]*sizes[i];
206389c6957cSBarry Smith       sz      = ii[row+1] - diag[row] - 1;
206489c6957cSBarry Smith       v1      = a->a + diag[row] + 1;
206589c6957cSBarry Smith       idx     = a->j + diag[row] + 1;
206689c6957cSBarry Smith       CHKMEMQ;
20674108e4d5SBarry Smith       /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
206889c6957cSBarry Smith       switch (sizes[i]){
206989c6957cSBarry Smith         case 1:
207089c6957cSBarry Smith 
207189c6957cSBarry Smith 	  sum1  = b[row];
207289c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
207389c6957cSBarry Smith 	    i1   = idx[0];
207489c6957cSBarry Smith 	    i2   = idx[1];
207589c6957cSBarry Smith 	    idx += 2;
207689c6957cSBarry Smith 	    tmp0 = x[i1];
207789c6957cSBarry Smith 	    tmp1 = x[i2];
207889c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
207989c6957cSBarry Smith 	  }
208089c6957cSBarry Smith 
208189c6957cSBarry Smith 	  if (n == sz-1){
208289c6957cSBarry Smith 	    tmp0  = x[*idx];
208389c6957cSBarry Smith 	    sum1 -= *v1*tmp0;
208489c6957cSBarry Smith 	  }
2085938d4eb3SBarry Smith 	  x[row] = sum1*(*ibdiag);row--;
208689c6957cSBarry Smith 	  break;
208789c6957cSBarry Smith 
208889c6957cSBarry Smith         case 2:
208989c6957cSBarry Smith 
209089c6957cSBarry Smith 	  sum1  = b[row];
209189c6957cSBarry Smith 	  sum2  = b[row-1];
209289c6957cSBarry Smith 	  /* note that sum1 is associated with the second of the two rows */
209389c6957cSBarry Smith 	  v2    = a->a + diag[row-1] + 2;
209489c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
209589c6957cSBarry Smith 	    i1   = idx[0];
209689c6957cSBarry Smith 	    i2   = idx[1];
209789c6957cSBarry Smith 	    idx += 2;
209889c6957cSBarry Smith 	    tmp0 = x[i1];
209989c6957cSBarry Smith 	    tmp1 = x[i2];
210089c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
210189c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
210289c6957cSBarry Smith 	  }
210389c6957cSBarry Smith 
210489c6957cSBarry Smith 	  if (n == sz-1){
210589c6957cSBarry Smith 	    tmp0  = x[*idx];
210689c6957cSBarry Smith 	    sum1 -= *v1*tmp0;
210789c6957cSBarry Smith 	    sum2 -= *v2*tmp0;
210889c6957cSBarry Smith 	  }
2109938d4eb3SBarry Smith 	  x[row] = sum2*ibdiag[1] + sum1*ibdiag[3];
2110938d4eb3SBarry Smith 	  x[row-1] = sum2*ibdiag[0] + sum1*ibdiag[2];
2111938d4eb3SBarry Smith           row -= 2;
211289c6957cSBarry Smith 	  break;
211389c6957cSBarry Smith         case 3:
211489c6957cSBarry Smith 
211589c6957cSBarry Smith 	  sum1  = b[row];
211689c6957cSBarry Smith 	  sum2  = b[row-1];
211789c6957cSBarry Smith 	  sum3  = b[row-2];
211889c6957cSBarry Smith 	  v2    = a->a + diag[row-1] + 2;
211989c6957cSBarry Smith 	  v3    = a->a + diag[row-2] + 3;
212089c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
212189c6957cSBarry Smith 	    i1   = idx[0];
212289c6957cSBarry Smith 	    i2   = idx[1];
212389c6957cSBarry Smith 	    idx += 2;
212489c6957cSBarry Smith 	    tmp0 = x[i1];
212589c6957cSBarry Smith 	    tmp1 = x[i2];
212689c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
212789c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
212889c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
212989c6957cSBarry Smith 	  }
213089c6957cSBarry Smith 
213189c6957cSBarry Smith 	  if (n == sz-1){
213289c6957cSBarry Smith 	    tmp0  = x[*idx];
213389c6957cSBarry Smith 	    sum1 -= *v1*tmp0;
213489c6957cSBarry Smith 	    sum2 -= *v2*tmp0;
213589c6957cSBarry Smith 	    sum3 -= *v3*tmp0;
213689c6957cSBarry Smith 	  }
2137938d4eb3SBarry Smith 	  x[row] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
2138938d4eb3SBarry Smith 	  x[row-1] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
2139938d4eb3SBarry Smith 	  x[row-2] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
2140938d4eb3SBarry Smith           row -= 3;
214189c6957cSBarry Smith 	  break;
214289c6957cSBarry Smith         case 4:
214389c6957cSBarry Smith 
214489c6957cSBarry Smith 	  sum1  = b[row];
214589c6957cSBarry Smith 	  sum2  = b[row-1];
214689c6957cSBarry Smith 	  sum3  = b[row-2];
214789c6957cSBarry Smith 	  sum4  = b[row-3];
214889c6957cSBarry Smith 	  v2    = a->a + diag[row-1] + 2;
214989c6957cSBarry Smith 	  v3    = a->a + diag[row-2] + 3;
215089c6957cSBarry Smith 	  v4    = a->a + diag[row-3] + 4;
215189c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
215289c6957cSBarry Smith 	    i1   = idx[0];
215389c6957cSBarry Smith 	    i2   = idx[1];
215489c6957cSBarry Smith 	    idx += 2;
215589c6957cSBarry Smith 	    tmp0 = x[i1];
215689c6957cSBarry Smith 	    tmp1 = x[i2];
215789c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
215889c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
215989c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
216089c6957cSBarry Smith 	    sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
216189c6957cSBarry Smith 	  }
216289c6957cSBarry Smith 
216389c6957cSBarry Smith 	  if (n == sz-1){
216489c6957cSBarry Smith 	    tmp0  = x[*idx];
216589c6957cSBarry Smith 	    sum1 -= *v1*tmp0;
216689c6957cSBarry Smith 	    sum2 -= *v2*tmp0;
216789c6957cSBarry Smith 	    sum3 -= *v3*tmp0;
216889c6957cSBarry Smith 	    sum4 -= *v4*tmp0;
216989c6957cSBarry Smith 	  }
2170938d4eb3SBarry Smith 	  x[row] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
2171938d4eb3SBarry Smith 	  x[row-1] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
2172938d4eb3SBarry Smith 	  x[row-2] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
2173938d4eb3SBarry Smith 	  x[row-3] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
2174938d4eb3SBarry Smith           row -= 4;
217589c6957cSBarry Smith 	  break;
217689c6957cSBarry Smith         case 5:
217789c6957cSBarry Smith 
217889c6957cSBarry Smith 	  sum1  = b[row];
217989c6957cSBarry Smith 	  sum2  = b[row-1];
218089c6957cSBarry Smith 	  sum3  = b[row-2];
218189c6957cSBarry Smith 	  sum4  = b[row-3];
218289c6957cSBarry Smith 	  sum5  = b[row-4];
218389c6957cSBarry Smith 	  v2    = a->a + diag[row-1] + 2;
218489c6957cSBarry Smith 	  v3    = a->a + diag[row-2] + 3;
218589c6957cSBarry Smith 	  v4    = a->a + diag[row-3] + 4;
218689c6957cSBarry Smith 	  v5    = a->a + diag[row-4] + 5;
218789c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
218889c6957cSBarry Smith 	    i1   = idx[0];
218989c6957cSBarry Smith 	    i2   = idx[1];
219089c6957cSBarry Smith 	    idx += 2;
219189c6957cSBarry Smith 	    tmp0 = x[i1];
219289c6957cSBarry Smith 	    tmp1 = x[i2];
219389c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
219489c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
219589c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
219689c6957cSBarry Smith 	    sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
219789c6957cSBarry Smith 	    sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
219889c6957cSBarry Smith 	  }
219989c6957cSBarry Smith 
220089c6957cSBarry Smith 	  if (n == sz-1){
220189c6957cSBarry Smith 	    tmp0  = x[*idx];
220289c6957cSBarry Smith 	    sum1 -= *v1*tmp0;
220389c6957cSBarry Smith 	    sum2 -= *v2*tmp0;
220489c6957cSBarry Smith 	    sum3 -= *v3*tmp0;
220589c6957cSBarry Smith 	    sum4 -= *v4*tmp0;
220689c6957cSBarry Smith 	    sum5 -= *v5*tmp0;
220789c6957cSBarry Smith 	  }
2208938d4eb3SBarry Smith 	  x[row] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
2209938d4eb3SBarry Smith 	  x[row-1] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
2210938d4eb3SBarry Smith 	  x[row-2] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
2211938d4eb3SBarry Smith 	  x[row-3] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
2212938d4eb3SBarry Smith 	  x[row-4] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
2213938d4eb3SBarry Smith           row -= 5;
221489c6957cSBarry Smith 	  break;
221589c6957cSBarry Smith         default:
221689c6957cSBarry Smith 	  SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
221789c6957cSBarry Smith       }
221889c6957cSBarry Smith       CHKMEMQ;
221989c6957cSBarry Smith     }
222089c6957cSBarry Smith     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
222189c6957cSBarry Smith 
222289c6957cSBarry Smith     /*
222389c6957cSBarry Smith            t = b - D x    where D is the block diagonal
222489c6957cSBarry Smith     */
222589c6957cSBarry Smith     cnt = 0;
222689c6957cSBarry Smith     for (i=0, row=0; i<m; i++) {
222789c6957cSBarry Smith       CHKMEMQ;
222889c6957cSBarry Smith       switch (sizes[i]){
222989c6957cSBarry Smith         case 1:
223089c6957cSBarry Smith 	  t[row] = b[row] - bdiag[cnt++]*x[row]; row++;
223189c6957cSBarry Smith 	  break;
223289c6957cSBarry Smith         case 2:
223389c6957cSBarry Smith 	  x1   = x[row]; x2 = x[row+1];
223489c6957cSBarry Smith 	  tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2];
223589c6957cSBarry Smith 	  tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3];
223689c6957cSBarry Smith 	  t[row]   = b[row] - tmp1;
223789c6957cSBarry Smith 	  t[row+1] = b[row+1] - tmp2; row += 2;
223889c6957cSBarry Smith 	  cnt += 4;
223989c6957cSBarry Smith 	  break;
224089c6957cSBarry Smith         case 3:
224189c6957cSBarry Smith 	  x1   = x[row]; x2 = x[row+1]; x3 = x[row+2];
224289c6957cSBarry Smith 	  tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6];
224389c6957cSBarry Smith 	  tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7];
224489c6957cSBarry Smith 	  tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8];
224589c6957cSBarry Smith 	  t[row] = b[row] - tmp1;
224689c6957cSBarry Smith 	  t[row+1] = b[row+1] - tmp2;
224789c6957cSBarry Smith 	  t[row+2] = b[row+2] - tmp3; row += 3;
224889c6957cSBarry Smith 	  cnt += 9;
224989c6957cSBarry Smith 	  break;
225089c6957cSBarry Smith         case 4:
225189c6957cSBarry Smith 	  x1   = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3];
225289c6957cSBarry Smith 	  tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12];
225389c6957cSBarry Smith 	  tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13];
225489c6957cSBarry Smith 	  tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14];
225589c6957cSBarry Smith 	  tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15];
225689c6957cSBarry Smith 	  t[row] = b[row] - tmp1;
225789c6957cSBarry Smith 	  t[row+1] = b[row+1] - tmp2;
225889c6957cSBarry Smith 	  t[row+2] = b[row+2] - tmp3;
225989c6957cSBarry Smith 	  t[row+3] = b[row+3] - tmp4; row += 4;
226089c6957cSBarry Smith 	  cnt += 16;
226189c6957cSBarry Smith 	  break;
226289c6957cSBarry Smith         case 5:
226389c6957cSBarry Smith 	  x1   = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; x5 = x[row+4];
226489c6957cSBarry Smith 	  tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20];
226589c6957cSBarry Smith 	  tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21];
226689c6957cSBarry Smith 	  tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22];
226789c6957cSBarry Smith 	  tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23];
226889c6957cSBarry Smith 	  tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24];
226989c6957cSBarry Smith 	  t[row] = b[row] - tmp1;
227089c6957cSBarry Smith 	  t[row+1] = b[row+1] - tmp2;
227189c6957cSBarry Smith 	  t[row+2] = b[row+2] - tmp3;
227289c6957cSBarry Smith 	  t[row+3] = b[row+3] - tmp4;
227389c6957cSBarry Smith 	  t[row+4] = b[row+4] - tmp5;row += 5;
227489c6957cSBarry Smith 	  cnt += 25;
227589c6957cSBarry Smith 	  break;
227689c6957cSBarry Smith         default:
227789c6957cSBarry Smith 	  SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
227889c6957cSBarry Smith       }
227989c6957cSBarry Smith       CHKMEMQ;
228089c6957cSBarry Smith     }
228189c6957cSBarry Smith     ierr = PetscLogFlops(m);CHKERRQ(ierr);
228289c6957cSBarry Smith 
228389c6957cSBarry Smith 
228489c6957cSBarry Smith 
228589c6957cSBarry Smith     /*
228689c6957cSBarry Smith           Apply (L + D)^-1 where D is the block diagonal
228789c6957cSBarry Smith     */
228889c6957cSBarry Smith     for (i=0, row=0; i<m; i++) {
228989c6957cSBarry Smith       sz  = diag[row] - ii[row];
229089c6957cSBarry Smith       v1  = a->a + ii[row];
229189c6957cSBarry Smith       idx = a->j + ii[row];
229289c6957cSBarry Smith       CHKMEMQ;
22934108e4d5SBarry Smith       /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
229489c6957cSBarry Smith       switch (sizes[i]){
229589c6957cSBarry Smith         case 1:
229689c6957cSBarry Smith 
229789c6957cSBarry Smith 	  sum1  = t[row];
229889c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
229989c6957cSBarry Smith 	    i1   = idx[0];
230089c6957cSBarry Smith 	    i2   = idx[1];
230189c6957cSBarry Smith 	    idx += 2;
230289c6957cSBarry Smith 	    tmp0 = t[i1];
230389c6957cSBarry Smith 	    tmp1 = t[i2];
230489c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
230589c6957cSBarry Smith 	  }
230689c6957cSBarry Smith 
230789c6957cSBarry Smith 	  if (n == sz-1){
230889c6957cSBarry Smith 	    tmp0  = t[*idx];
230989c6957cSBarry Smith 	    sum1 -= *v1 * tmp0;
231089c6957cSBarry Smith 	  }
231189c6957cSBarry Smith 	  x[row] += t[row] = sum1*(*ibdiag++); row++;
231289c6957cSBarry Smith 	  break;
231389c6957cSBarry Smith         case 2:
231489c6957cSBarry Smith 	  v2    = a->a + ii[row+1];
231589c6957cSBarry Smith 	  sum1  = t[row];
231689c6957cSBarry Smith 	  sum2  = t[row+1];
231789c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
231889c6957cSBarry Smith 	    i1   = idx[0];
231989c6957cSBarry Smith 	    i2   = idx[1];
232089c6957cSBarry Smith 	    idx += 2;
232189c6957cSBarry Smith 	    tmp0 = t[i1];
232289c6957cSBarry Smith 	    tmp1 = t[i2];
232389c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
232489c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
232589c6957cSBarry Smith 	  }
232689c6957cSBarry Smith 
232789c6957cSBarry Smith 	  if (n == sz-1){
232889c6957cSBarry Smith 	    tmp0  = t[*idx];
232989c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0;
233089c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0;
233189c6957cSBarry Smith 	  }
233289c6957cSBarry Smith 	  x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[2];
233389c6957cSBarry Smith 	  x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3];
233489c6957cSBarry Smith 	  ibdiag  += 4; row += 2;
233589c6957cSBarry Smith 	  break;
233689c6957cSBarry Smith         case 3:
233789c6957cSBarry Smith 	  v2    = a->a + ii[row+1];
233889c6957cSBarry Smith 	  v3    = a->a + ii[row+2];
233989c6957cSBarry Smith 	  sum1  = t[row];
234089c6957cSBarry Smith 	  sum2  = t[row+1];
234189c6957cSBarry Smith 	  sum3  = t[row+2];
234289c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
234389c6957cSBarry Smith 	    i1   = idx[0];
234489c6957cSBarry Smith 	    i2   = idx[1];
234589c6957cSBarry Smith 	    idx += 2;
234689c6957cSBarry Smith 	    tmp0 = t[i1];
234789c6957cSBarry Smith 	    tmp1 = t[i2];
234889c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
234989c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
235089c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
235189c6957cSBarry Smith 	  }
235289c6957cSBarry Smith 
235389c6957cSBarry Smith 	  if (n == sz-1){
235489c6957cSBarry Smith 	    tmp0  = t[*idx];
235589c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0;
235689c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0;
235789c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0;
235889c6957cSBarry Smith 	  }
235989c6957cSBarry Smith 	  x[row]  += t[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
236089c6957cSBarry Smith 	  x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
236189c6957cSBarry Smith 	  x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
236289c6957cSBarry Smith 	  ibdiag  += 9; row += 3;
236389c6957cSBarry Smith 	  break;
236489c6957cSBarry Smith         case 4:
236589c6957cSBarry Smith 	  v2    = a->a + ii[row+1];
236689c6957cSBarry Smith 	  v3    = a->a + ii[row+2];
236789c6957cSBarry Smith 	  v4    = a->a + ii[row+3];
236889c6957cSBarry Smith 	  sum1  = t[row];
236989c6957cSBarry Smith 	  sum2  = t[row+1];
237089c6957cSBarry Smith 	  sum3  = t[row+2];
237189c6957cSBarry Smith 	  sum4  = t[row+3];
237289c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
237389c6957cSBarry Smith 	    i1   = idx[0];
237489c6957cSBarry Smith 	    i2   = idx[1];
237589c6957cSBarry Smith 	    idx += 2;
237689c6957cSBarry Smith 	    tmp0 = t[i1];
237789c6957cSBarry Smith 	    tmp1 = t[i2];
237889c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
237989c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
238089c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
238189c6957cSBarry Smith 	    sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
238289c6957cSBarry Smith 	  }
238389c6957cSBarry Smith 
238489c6957cSBarry Smith 	  if (n == sz-1){
238589c6957cSBarry Smith 	    tmp0  = t[*idx];
238689c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0;
238789c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0;
238889c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0;
238989c6957cSBarry Smith 	    sum4 -= v4[0] * tmp0;
239089c6957cSBarry Smith 	  }
239189c6957cSBarry Smith 	  x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
239289c6957cSBarry Smith 	  x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
239389c6957cSBarry Smith 	  x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
239489c6957cSBarry Smith 	  x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
239589c6957cSBarry Smith 	  ibdiag  += 16; row += 4;
239689c6957cSBarry Smith 	  break;
239789c6957cSBarry Smith         case 5:
239889c6957cSBarry Smith 	  v2    = a->a + ii[row+1];
239989c6957cSBarry Smith 	  v3    = a->a + ii[row+2];
240089c6957cSBarry Smith 	  v4    = a->a + ii[row+3];
240189c6957cSBarry Smith 	  v5    = a->a + ii[row+4];
240289c6957cSBarry Smith 	  sum1  = t[row];
240389c6957cSBarry Smith 	  sum2  = t[row+1];
240489c6957cSBarry Smith 	  sum3  = t[row+2];
240589c6957cSBarry Smith 	  sum4  = t[row+3];
240689c6957cSBarry Smith 	  sum5  = t[row+4];
240789c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
240889c6957cSBarry Smith 	    i1   = idx[0];
240989c6957cSBarry Smith 	    i2   = idx[1];
241089c6957cSBarry Smith 	    idx += 2;
241189c6957cSBarry Smith 	    tmp0 = t[i1];
241289c6957cSBarry Smith 	    tmp1 = t[i2];
241389c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
241489c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
241589c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
241689c6957cSBarry Smith 	    sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
241789c6957cSBarry Smith 	    sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
241889c6957cSBarry Smith 	  }
241989c6957cSBarry Smith 
242089c6957cSBarry Smith 	  if (n == sz-1){
242189c6957cSBarry Smith 	    tmp0  = t[*idx];
242289c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0;
242389c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0;
242489c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0;
242589c6957cSBarry Smith 	    sum4 -= v4[0] * tmp0;
242689c6957cSBarry Smith 	    sum5 -= v5[0] * tmp0;
242789c6957cSBarry Smith 	  }
242889c6957cSBarry Smith 	  x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
242989c6957cSBarry Smith 	  x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
243089c6957cSBarry Smith 	  x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
243189c6957cSBarry Smith 	  x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
243289c6957cSBarry Smith 	  x[row+4] += t[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
243389c6957cSBarry Smith 	  ibdiag  += 25; row += 5;
243489c6957cSBarry Smith 	  break;
243589c6957cSBarry Smith         default:
243689c6957cSBarry Smith 	  SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
243789c6957cSBarry Smith       }
243889c6957cSBarry Smith       CHKMEMQ;
243989c6957cSBarry Smith     }
244089c6957cSBarry Smith     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
24412af78befSBarry Smith     ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
2442938d4eb3SBarry Smith     ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
2443938d4eb3SBarry Smith   }
24442af78befSBarry Smith   PetscFunctionReturn(0);
24452af78befSBarry Smith }
24462af78befSBarry Smith 
244789c6957cSBarry Smith #undef __FUNCT__
24484108e4d5SBarry Smith #define __FUNCT__ "MatMultDiagonalBlock_SeqAIJ_Inode"
24494108e4d5SBarry Smith PetscErrorCode MatMultDiagonalBlock_SeqAIJ_Inode(Mat A,Vec bb,Vec xx)
245089c6957cSBarry Smith {
245189c6957cSBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
245289c6957cSBarry Smith   PetscScalar        *x,tmp1,tmp2,tmp3,tmp4,tmp5,x1,x2,x3,x4,x5;
245389c6957cSBarry Smith   const MatScalar    *bdiag = a->inode.bdiag;
245489c6957cSBarry Smith   const PetscScalar  *b;
245589c6957cSBarry Smith   PetscErrorCode      ierr;
245689c6957cSBarry Smith   PetscInt            m = a->inode.node_count,cnt = 0,i,row;
245789c6957cSBarry Smith   const PetscInt      *sizes = a->inode.size;
24582af78befSBarry Smith 
245989c6957cSBarry Smith   PetscFunctionBegin;
246089c6957cSBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
246189c6957cSBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
246289c6957cSBarry Smith   cnt = 0;
246389c6957cSBarry Smith   for (i=0, row=0; i<m; i++) {
246489c6957cSBarry Smith     switch (sizes[i]){
246589c6957cSBarry Smith       case 1:
246689c6957cSBarry Smith 	x[row] = b[row]*bdiag[cnt++];row++;
246789c6957cSBarry Smith 	break;
246889c6957cSBarry Smith       case 2:
246989c6957cSBarry Smith 	x1   = b[row]; x2 = b[row+1];
247089c6957cSBarry Smith 	tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2];
247189c6957cSBarry Smith 	tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3];
247289c6957cSBarry Smith 	x[row++] = tmp1;
247389c6957cSBarry Smith 	x[row++] = tmp2;
247489c6957cSBarry Smith 	cnt += 4;
247589c6957cSBarry Smith 	break;
247689c6957cSBarry Smith       case 3:
247789c6957cSBarry Smith 	x1   = b[row]; x2 = b[row+1]; x3 = b[row+2];
247889c6957cSBarry Smith 	tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6];
247989c6957cSBarry Smith 	tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7];
248089c6957cSBarry Smith 	tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8];
248189c6957cSBarry Smith 	x[row++] = tmp1;
248289c6957cSBarry Smith 	x[row++] = tmp2;
248389c6957cSBarry Smith 	x[row++] = tmp3;
248489c6957cSBarry Smith 	cnt += 9;
248589c6957cSBarry Smith 	break;
248689c6957cSBarry Smith       case 4:
248789c6957cSBarry Smith 	x1   = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3];
248889c6957cSBarry Smith 	tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12];
248989c6957cSBarry Smith 	tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13];
249089c6957cSBarry Smith 	tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14];
249189c6957cSBarry Smith 	tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15];
249289c6957cSBarry Smith 	x[row++] = tmp1;
249389c6957cSBarry Smith 	x[row++] = tmp2;
249489c6957cSBarry Smith 	x[row++] = tmp3;
249589c6957cSBarry Smith 	x[row++] = tmp4;
249689c6957cSBarry Smith 	cnt += 16;
249789c6957cSBarry Smith 	break;
249889c6957cSBarry Smith       case 5:
249989c6957cSBarry Smith 	x1   = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3]; x5 = b[row+4];
250089c6957cSBarry Smith 	tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20];
250189c6957cSBarry Smith 	tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21];
250289c6957cSBarry Smith 	tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22];
250389c6957cSBarry Smith 	tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23];
250489c6957cSBarry Smith 	tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24];
250589c6957cSBarry Smith 	x[row++] = tmp1;
250689c6957cSBarry Smith 	x[row++] = tmp2;
250789c6957cSBarry Smith 	x[row++] = tmp3;
250889c6957cSBarry Smith 	x[row++] = tmp4;
250989c6957cSBarry Smith 	x[row++] = tmp5;
251089c6957cSBarry Smith 	cnt += 25;
251189c6957cSBarry Smith 	break;
251289c6957cSBarry Smith       default:
251389c6957cSBarry Smith 	SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
251489c6957cSBarry Smith     }
251589c6957cSBarry Smith   }
251689c6957cSBarry Smith   ierr = PetscLogFlops(2*cnt);CHKERRQ(ierr);
251789c6957cSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
251889c6957cSBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
251989c6957cSBarry Smith   PetscFunctionReturn(0);
252089c6957cSBarry Smith }
252189c6957cSBarry Smith 
25224c1414c8SBarry Smith /*
25234c1414c8SBarry Smith     samestructure indicates that the matrix has not changed its nonzero structure so we
25244c1414c8SBarry Smith     do not need to recompute the inodes
25254c1414c8SBarry Smith */
25264c1414c8SBarry Smith #undef __FUNCT__
25274c1414c8SBarry Smith #define __FUNCT__ "Mat_CheckInode"
25284c1414c8SBarry Smith PetscErrorCode Mat_CheckInode(Mat A,PetscTruth samestructure)
25294c1414c8SBarry Smith {
25304c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
25314c1414c8SBarry Smith   PetscErrorCode ierr;
25324c1414c8SBarry Smith   PetscInt       i,j,m,nzx,nzy,*idx,*idy,*ns,*ii,node_count,blk_size;
2533d74fe821SBarry Smith   PetscTruth     flag;
25344c1414c8SBarry Smith 
25354c1414c8SBarry Smith   PetscFunctionBegin;
2536d74fe821SBarry Smith   if (!a->inode.use)                     PetscFunctionReturn(0);
25374c1414c8SBarry Smith   if (a->inode.checked && samestructure) PetscFunctionReturn(0);
25384c1414c8SBarry Smith 
25394c1414c8SBarry Smith 
2540d0f46423SBarry Smith   m = A->rmap->n;
25414c1414c8SBarry Smith   if (a->inode.size) {ns = a->inode.size;}
25424c1414c8SBarry Smith   else {ierr = PetscMalloc((m+1)*sizeof(PetscInt),&ns);CHKERRQ(ierr);}
25434c1414c8SBarry Smith 
25444c1414c8SBarry Smith   i          = 0;
25454c1414c8SBarry Smith   node_count = 0;
25464c1414c8SBarry Smith   idx        = a->j;
25474c1414c8SBarry Smith   ii         = a->i;
25484c1414c8SBarry Smith   while (i < m){                /* For each row */
25494c1414c8SBarry Smith     nzx = ii[i+1] - ii[i];       /* Number of nonzeros */
25504c1414c8SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
25514c1414c8SBarry Smith     for (j=i+1,idy=idx,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
25524c1414c8SBarry Smith       nzy     = ii[j+1] - ii[j]; /* Same number of nonzeros */
25534c1414c8SBarry Smith       if (nzy != nzx) break;
25544c1414c8SBarry Smith       idy  += nzx;             /* Same nonzero pattern */
25554c1414c8SBarry Smith       ierr = PetscMemcmp(idx,idy,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr);
25564c1414c8SBarry Smith       if (!flag) break;
25574c1414c8SBarry Smith     }
25584c1414c8SBarry Smith     ns[node_count++] = blk_size;
25594c1414c8SBarry Smith     idx += blk_size*nzx;
25604c1414c8SBarry Smith     i    = j;
25614c1414c8SBarry Smith   }
25624c1414c8SBarry Smith   /* If not enough inodes found,, do not use inode version of the routines */
2563f0d39aaaSBarry Smith   if (!a->inode.size && m && node_count > .9*m) {
25644c1414c8SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
25654c1414c8SBarry Smith     a->inode.node_count     = 0;
25664c1414c8SBarry Smith     a->inode.size           = PETSC_NULL;
25674c1414c8SBarry Smith     a->inode.use            = PETSC_FALSE;
25684c1414c8SBarry Smith     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
25694c1414c8SBarry Smith   } else {
25704108e4d5SBarry Smith     A->ops->mult              = MatMult_SeqAIJ_Inode;
25714108e4d5SBarry Smith     A->ops->sor               = MatSOR_SeqAIJ_Inode;
25724108e4d5SBarry Smith     A->ops->multadd           = MatMultAdd_SeqAIJ_Inode;
25734108e4d5SBarry Smith     A->ops->getrowij          = MatGetRowIJ_SeqAIJ_Inode;
25744108e4d5SBarry Smith     A->ops->restorerowij      = MatRestoreRowIJ_SeqAIJ_Inode;
25754108e4d5SBarry Smith     A->ops->getcolumnij       = MatGetColumnIJ_SeqAIJ_Inode;
25764108e4d5SBarry Smith     A->ops->restorecolumnij   = MatRestoreColumnIJ_SeqAIJ_Inode;
25774108e4d5SBarry Smith     A->ops->coloringpatch     = MatColoringPatch_SeqAIJ_Inode;
25784108e4d5SBarry Smith     A->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode;
25794c1414c8SBarry Smith     a->inode.node_count       = node_count;
25804c1414c8SBarry Smith     a->inode.size             = ns;
25814c1414c8SBarry Smith     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
25824c1414c8SBarry Smith   }
25834c1414c8SBarry Smith   PetscFunctionReturn(0);
25844c1414c8SBarry Smith }
25854c1414c8SBarry Smith 
2586*019b515eSShri Abhyankar #define MatGetRow_FactoredLU(cols,nzl,nzu,nz,ai,aj,adiag,row) {	\
2587*019b515eSShri Abhyankar PetscInt __k, *__vi; \
2588*019b515eSShri Abhyankar __vi = aj + ai[row];				\
2589*019b515eSShri Abhyankar for(__k=0;__k<nzl;__k++) cols[__k] = __vi[__k]; \
2590*019b515eSShri Abhyankar __vi = aj + adiag[row];				\
2591*019b515eSShri Abhyankar cols[nzl] = __vi[0];\
2592*019b515eSShri Abhyankar __vi = aj + adiag[row+1]+1;\
2593*019b515eSShri Abhyankar for(__k=0;__k<nzu;__k++) cols[nzl+1+__k] = __vi[__k];}
2594*019b515eSShri Abhyankar 
2595*019b515eSShri Abhyankar #undef __FUNCT__
2596*019b515eSShri Abhyankar #define __FUNCT__ "Mat_CheckInode_FactorLU"
2597*019b515eSShri Abhyankar PetscErrorCode Mat_CheckInode_FactorLU(Mat A,PetscTruth samestructure)
2598*019b515eSShri Abhyankar {
2599*019b515eSShri Abhyankar   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
2600*019b515eSShri Abhyankar   PetscErrorCode ierr;
2601*019b515eSShri Abhyankar   PetscInt       i,j,m,nzl1,nzu1,nzl2,nzu2,nzx,nzy,node_count,blk_size;
2602*019b515eSShri Abhyankar   PetscInt       *cols1,*cols2,*ns,*ai = a->i,*aj = a->j, *adiag = a->diag;
2603*019b515eSShri Abhyankar   PetscTruth     flag;
2604*019b515eSShri Abhyankar 
2605*019b515eSShri Abhyankar   PetscFunctionBegin;
2606*019b515eSShri Abhyankar   if (!a->inode.use)                     PetscFunctionReturn(0);
2607*019b515eSShri Abhyankar   if (a->inode.checked && samestructure) PetscFunctionReturn(0);
2608*019b515eSShri Abhyankar 
2609*019b515eSShri Abhyankar   m = A->rmap->n;
2610*019b515eSShri Abhyankar   if (a->inode.size) {ns = a->inode.size;}
2611*019b515eSShri Abhyankar   else {ierr = PetscMalloc((m+1)*sizeof(PetscInt),&ns);CHKERRQ(ierr);}
2612*019b515eSShri Abhyankar 
2613*019b515eSShri Abhyankar   i          = 0;
2614*019b515eSShri Abhyankar   node_count = 0;
2615*019b515eSShri Abhyankar   while (i < m){                /* For each row */
2616*019b515eSShri Abhyankar     nzl1 = ai[i+1] - ai[i];       /* Number of nonzeros in L */
2617*019b515eSShri Abhyankar     nzu1 = adiag[i] - adiag[i+1] - 1; /* Number of nonzeros in U excluding diagonal*/
2618*019b515eSShri Abhyankar     nzx  = nzl1 + nzu1 + 1;
2619*019b515eSShri Abhyankar     ierr = PetscMalloc((nzx+1)*sizeof(PetscInt),&cols1);CHKERRQ(ierr);
2620*019b515eSShri Abhyankar     MatGetRow_FactoredLU(cols1,nzl1,nzu1,nzx,ai,aj,adiag,i);
2621*019b515eSShri Abhyankar 
2622*019b515eSShri Abhyankar     /* Limits the number of elements in a node to 'a->inode.limit' */
2623*019b515eSShri Abhyankar     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
2624*019b515eSShri Abhyankar       nzl2    = ai[j+1] - ai[j];
2625*019b515eSShri Abhyankar       nzu2    = adiag[j] - adiag[j+1] - 1;
2626*019b515eSShri Abhyankar       nzy     = nzl2 + nzu2 + 1;
2627*019b515eSShri Abhyankar       if( nzy != nzx) break;
2628*019b515eSShri Abhyankar       ierr    = PetscMalloc((nzy+1)*sizeof(PetscInt),&cols2);CHKERRQ(ierr);
2629*019b515eSShri Abhyankar       MatGetRow_FactoredLU(cols2,nzl2,nzu2,nzy,ai,aj,adiag,j);
2630*019b515eSShri Abhyankar       ierr = PetscMemcmp(cols1,cols2,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr);
2631*019b515eSShri Abhyankar       if (!flag) {ierr = PetscFree(cols2);CHKERRQ(ierr);break;}
2632*019b515eSShri Abhyankar       ierr = PetscFree(cols2);CHKERRQ(ierr);
2633*019b515eSShri Abhyankar     }
2634*019b515eSShri Abhyankar     ns[node_count++] = blk_size;
2635*019b515eSShri Abhyankar     ierr = PetscFree(cols1);CHKERRQ(ierr);
2636*019b515eSShri Abhyankar     i    = j;
2637*019b515eSShri Abhyankar   }
2638*019b515eSShri Abhyankar   /* If not enough inodes found,, do not use inode version of the routines */
2639*019b515eSShri Abhyankar   if (!a->inode.size && m && node_count > .9*m) {
2640*019b515eSShri Abhyankar     ierr = PetscFree(ns);CHKERRQ(ierr);
2641*019b515eSShri Abhyankar     a->inode.node_count     = 0;
2642*019b515eSShri Abhyankar     a->inode.size           = PETSC_NULL;
2643*019b515eSShri Abhyankar     a->inode.use            = PETSC_FALSE;
2644*019b515eSShri Abhyankar     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
2645*019b515eSShri Abhyankar   } else {
2646*019b515eSShri Abhyankar     A->ops->mult              = 0;
2647*019b515eSShri Abhyankar     A->ops->sor               = 0;
2648*019b515eSShri Abhyankar     A->ops->multadd           = 0;
2649*019b515eSShri Abhyankar     A->ops->getrowij          = 0;
2650*019b515eSShri Abhyankar     A->ops->restorerowij      = 0;
2651*019b515eSShri Abhyankar     A->ops->getcolumnij       = 0;
2652*019b515eSShri Abhyankar     A->ops->restorecolumnij   = 0;
2653*019b515eSShri Abhyankar     A->ops->coloringpatch     = 0;
2654*019b515eSShri Abhyankar     A->ops->multdiagonalblock = 0;
2655*019b515eSShri Abhyankar     a->inode.node_count       = node_count;
2656*019b515eSShri Abhyankar     a->inode.size             = ns;
2657*019b515eSShri Abhyankar     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
2658*019b515eSShri Abhyankar   }
2659*019b515eSShri Abhyankar   PetscFunctionReturn(0);
2660*019b515eSShri Abhyankar }
2661*019b515eSShri Abhyankar 
26624c1414c8SBarry Smith /*
26634c1414c8SBarry Smith      This is really ugly. if inodes are used this replaces the
26644c1414c8SBarry Smith   permutations with ones that correspond to rows/cols of the matrix
26654c1414c8SBarry Smith   rather then inode blocks
26664c1414c8SBarry Smith */
26674c1414c8SBarry Smith #undef __FUNCT__
26684c1414c8SBarry Smith #define __FUNCT__ "MatInodeAdjustForInodes"
26694c1414c8SBarry Smith PetscErrorCode PETSCMAT_DLLEXPORT MatInodeAdjustForInodes(Mat A,IS *rperm,IS *cperm)
26704c1414c8SBarry Smith {
26714c1414c8SBarry Smith   PetscErrorCode ierr,(*f)(Mat,IS*,IS*);
26724c1414c8SBarry Smith 
26734c1414c8SBarry Smith   PetscFunctionBegin;
26744c1414c8SBarry Smith   ierr = PetscObjectQueryFunction((PetscObject)A,"MatInodeAdjustForInodes_C",(void (**)(void))&f);CHKERRQ(ierr);
26754c1414c8SBarry Smith   if (f) {
26764c1414c8SBarry Smith     ierr = (*f)(A,rperm,cperm);CHKERRQ(ierr);
26774c1414c8SBarry Smith   }
26784c1414c8SBarry Smith   PetscFunctionReturn(0);
26794c1414c8SBarry Smith }
26804c1414c8SBarry Smith 
26814c1414c8SBarry Smith EXTERN_C_BEGIN
26824c1414c8SBarry Smith #undef __FUNCT__
26834108e4d5SBarry Smith #define __FUNCT__ "MatAdjustForInodes_SeqAIJ_Inode"
26844108e4d5SBarry Smith PetscErrorCode PETSCMAT_DLLEXPORT MatInodeAdjustForInodes_SeqAIJ_Inode(Mat A,IS *rperm,IS *cperm)
26854c1414c8SBarry Smith {
26864c1414c8SBarry Smith   Mat_SeqAIJ      *a=(Mat_SeqAIJ*)A->data;
26874c1414c8SBarry Smith   PetscErrorCode ierr;
26885d0c19d7SBarry Smith   PetscInt       m = A->rmap->n,n = A->cmap->n,i,j,nslim_row = a->inode.node_count;
26895d0c19d7SBarry Smith   const PetscInt *ridx,*cidx;
26904c1414c8SBarry Smith   PetscInt       row,col,*permr,*permc,*ns_row =  a->inode.size,*tns,start_val,end_val,indx;
26914c1414c8SBarry Smith   PetscInt       nslim_col,*ns_col;
26924c1414c8SBarry Smith   IS             ris = *rperm,cis = *cperm;
26934c1414c8SBarry Smith 
26944c1414c8SBarry Smith   PetscFunctionBegin;
26954c1414c8SBarry Smith   if (!a->inode.size) PetscFunctionReturn(0); /* no inodes so return */
26964c1414c8SBarry Smith   if (a->inode.node_count == m) PetscFunctionReturn(0); /* all inodes are of size 1 */
26974c1414c8SBarry Smith 
26984c1414c8SBarry Smith   ierr  = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr);
26994c1414c8SBarry Smith   ierr  = PetscMalloc((((nslim_row>nslim_col)?nslim_row:nslim_col)+1)*sizeof(PetscInt),&tns);CHKERRQ(ierr);
27000e83c824SBarry Smith   ierr  = PetscMalloc2(m,PetscInt,&permr,n,PetscInt,&permc);CHKERRQ(ierr);
27014c1414c8SBarry Smith 
27024c1414c8SBarry Smith   ierr  = ISGetIndices(ris,&ridx);CHKERRQ(ierr);
27034c1414c8SBarry Smith   ierr  = ISGetIndices(cis,&cidx);CHKERRQ(ierr);
27044c1414c8SBarry Smith 
27054c1414c8SBarry Smith   /* Form the inode structure for the rows of permuted matric using inv perm*/
27064c1414c8SBarry Smith   for (i=0,tns[0]=0; i<nslim_row; ++i) tns[i+1] = tns[i] + ns_row[i];
27074c1414c8SBarry Smith 
27084c1414c8SBarry Smith   /* Construct the permutations for rows*/
27094c1414c8SBarry Smith   for (i=0,row = 0; i<nslim_row; ++i){
27104c1414c8SBarry Smith     indx      = ridx[i];
27114c1414c8SBarry Smith     start_val = tns[indx];
27124c1414c8SBarry Smith     end_val   = tns[indx + 1];
27134c1414c8SBarry Smith     for (j=start_val; j<end_val; ++j,++row) permr[row]= j;
27144c1414c8SBarry Smith   }
27154c1414c8SBarry Smith 
27164c1414c8SBarry Smith   /* Form the inode structure for the columns of permuted matrix using inv perm*/
27174c1414c8SBarry Smith   for (i=0,tns[0]=0; i<nslim_col; ++i) tns[i+1] = tns[i] + ns_col[i];
27184c1414c8SBarry Smith 
27194c1414c8SBarry Smith  /* Construct permutations for columns */
27204c1414c8SBarry Smith   for (i=0,col=0; i<nslim_col; ++i){
27214c1414c8SBarry Smith     indx      = cidx[i];
27224c1414c8SBarry Smith     start_val = tns[indx];
27234c1414c8SBarry Smith     end_val   = tns[indx + 1];
27244c1414c8SBarry Smith     for (j = start_val; j<end_val; ++j,++col) permc[col]= j;
27254c1414c8SBarry Smith   }
27264c1414c8SBarry Smith 
27274c1414c8SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permr,rperm);CHKERRQ(ierr);
27280e83c824SBarry Smith   ierr = ISSetPermutation(*rperm);CHKERRQ(ierr);
27294c1414c8SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permc,cperm);CHKERRQ(ierr);
27300e83c824SBarry Smith   ierr = ISSetPermutation(*cperm);CHKERRQ(ierr);
27314c1414c8SBarry Smith 
27324c1414c8SBarry Smith   ierr  = ISRestoreIndices(ris,&ridx);CHKERRQ(ierr);
27334c1414c8SBarry Smith   ierr  = ISRestoreIndices(cis,&cidx);CHKERRQ(ierr);
27344c1414c8SBarry Smith 
27354c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
27360e83c824SBarry Smith   ierr = PetscFree2(permr,permc);CHKERRQ(ierr);
27374c1414c8SBarry Smith   ierr = ISDestroy(cis);CHKERRQ(ierr);
27384c1414c8SBarry Smith   ierr = ISDestroy(ris);CHKERRQ(ierr);
27394c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
27404c1414c8SBarry Smith   PetscFunctionReturn(0);
27414c1414c8SBarry Smith }
27424c1414c8SBarry Smith EXTERN_C_END
27434c1414c8SBarry Smith 
27444c1414c8SBarry Smith #undef __FUNCT__
27454c1414c8SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes"
27464c1414c8SBarry Smith /*@C
27474c1414c8SBarry Smith    MatInodeGetInodeSizes - Returns the inode information of the Inode matrix.
27484c1414c8SBarry Smith 
27494c1414c8SBarry Smith    Collective on Mat
27504c1414c8SBarry Smith 
27514c1414c8SBarry Smith    Input Parameter:
27524c1414c8SBarry Smith .  A - the Inode matrix or matrix derived from the Inode class -- e.g., SeqAIJ
27534c1414c8SBarry Smith 
27544c1414c8SBarry Smith    Output Parameter:
27554c1414c8SBarry Smith +  node_count - no of inodes present in the matrix.
27564c1414c8SBarry Smith .  sizes      - an array of size node_count,with sizes of each inode.
27574c1414c8SBarry Smith -  limit      - the max size used to generate the inodes.
27584c1414c8SBarry Smith 
27594c1414c8SBarry Smith    Level: advanced
27604c1414c8SBarry Smith 
27614c1414c8SBarry Smith    Notes: This routine returns some internal storage information
27624c1414c8SBarry Smith    of the matrix, it is intended to be used by advanced users.
27634c1414c8SBarry Smith    It should be called after the matrix is assembled.
27644c1414c8SBarry Smith    The contents of the sizes[] array should not be changed.
27654c1414c8SBarry Smith    PETSC_NULL may be passed for information not requested.
27664c1414c8SBarry Smith 
27674c1414c8SBarry Smith .keywords: matrix, seqaij, get, inode
27684c1414c8SBarry Smith 
27694c1414c8SBarry Smith .seealso: MatGetInfo()
27704c1414c8SBarry Smith @*/
27714c1414c8SBarry Smith PetscErrorCode PETSCMAT_DLLEXPORT MatInodeGetInodeSizes(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit)
27724c1414c8SBarry Smith {
27734c1414c8SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscInt*,PetscInt*[],PetscInt*);
27744c1414c8SBarry Smith 
27754c1414c8SBarry Smith   PetscFunctionBegin;
27764c1414c8SBarry Smith   if (!A->assembled) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
27774c1414c8SBarry Smith   ierr = PetscObjectQueryFunction((PetscObject)A,"MatInodeGetInodeSizes_C",(void (**)(void))&f);CHKERRQ(ierr);
27784c1414c8SBarry Smith   if (f) {
27794c1414c8SBarry Smith     ierr = (*f)(A,node_count,sizes,limit);CHKERRQ(ierr);
27804c1414c8SBarry Smith   }
27814c1414c8SBarry Smith   PetscFunctionReturn(0);
27824c1414c8SBarry Smith }
27834c1414c8SBarry Smith 
27844c1414c8SBarry Smith EXTERN_C_BEGIN
27854c1414c8SBarry Smith #undef __FUNCT__
27864108e4d5SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes_SeqAIJ_Inode"
27874108e4d5SBarry Smith PetscErrorCode PETSCMAT_DLLEXPORT MatInodeGetInodeSizes_SeqAIJ_Inode(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit)
27884c1414c8SBarry Smith {
27894c1414c8SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
27904c1414c8SBarry Smith 
27914c1414c8SBarry Smith   PetscFunctionBegin;
27924c1414c8SBarry Smith   if (node_count) *node_count = a->inode.node_count;
27934c1414c8SBarry Smith   if (sizes)      *sizes      = a->inode.size;
27944c1414c8SBarry Smith   if (limit)      *limit      = a->inode.limit;
27954c1414c8SBarry Smith   PetscFunctionReturn(0);
27964c1414c8SBarry Smith }
27974c1414c8SBarry Smith EXTERN_C_END
2798