xref: /petsc/src/mat/impls/aij/seq/inode.c (revision 98991853edfc8c750d14b9bcf2dbc74ffbbb30db)
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;
4263b5996c7SJed Brown     PetscPrefetchBlock(idx+nsz*n,n,0,0);    /* Prefetch the indices for the block row after the current one */
4273b5996c7SJed Brown     PetscPrefetchBlock(v1+nsz*n,nsz*n,0,0); /* Prefetch the values for the block row after the current one  */
4284c1414c8SBarry Smith     sz   = n;                   /* No of non zeros in this row */
4294c1414c8SBarry Smith                                 /* Switch on the size of Node */
4304c1414c8SBarry Smith     switch (nsz){               /* Each loop in 'case' is unrolled */
4314c1414c8SBarry Smith     case 1 :
43275567043SBarry Smith       sum1  = 0.;
4334c1414c8SBarry Smith 
4344c1414c8SBarry Smith       for(n = 0; n< sz-1; n+=2) {
4354c1414c8SBarry Smith         i1   = idx[0];          /* The instructions are ordered to */
4364c1414c8SBarry Smith         i2   = idx[1];          /* make the compiler's job easy */
4374c1414c8SBarry Smith         idx += 2;
4384c1414c8SBarry Smith         tmp0 = x[i1];
4394c1414c8SBarry Smith         tmp1 = x[i2];
4404c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4414c1414c8SBarry Smith        }
4424c1414c8SBarry Smith 
4434c1414c8SBarry Smith       if (n == sz-1){          /* Take care of the last nonzero  */
4444c1414c8SBarry Smith         tmp0  = x[*idx++];
4454c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
4464c1414c8SBarry Smith       }
4474c1414c8SBarry Smith       y[row++]=sum1;
4484c1414c8SBarry Smith       break;
4494c1414c8SBarry Smith     case 2:
45075567043SBarry Smith       sum1  = 0.;
45175567043SBarry Smith       sum2  = 0.;
4524c1414c8SBarry Smith       v2    = v1 + n;
4534c1414c8SBarry Smith 
4544c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4554c1414c8SBarry Smith         i1   = idx[0];
4564c1414c8SBarry Smith         i2   = idx[1];
4574c1414c8SBarry Smith         idx += 2;
4584c1414c8SBarry Smith         tmp0 = x[i1];
4594c1414c8SBarry Smith         tmp1 = x[i2];
4604c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4614c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
4624c1414c8SBarry Smith       }
4634c1414c8SBarry Smith       if (n == sz-1){
4644c1414c8SBarry Smith         tmp0  = x[*idx++];
4654c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
4664c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
4674c1414c8SBarry Smith       }
4684c1414c8SBarry Smith       y[row++]=sum1;
4694c1414c8SBarry Smith       y[row++]=sum2;
4704c1414c8SBarry Smith       v1      =v2;              /* Since the next block to be processed starts there*/
4714c1414c8SBarry Smith       idx    +=sz;
4724c1414c8SBarry Smith       break;
4734c1414c8SBarry Smith     case 3:
47475567043SBarry Smith       sum1  = 0.;
47575567043SBarry Smith       sum2  = 0.;
47675567043SBarry Smith       sum3  = 0.;
4774c1414c8SBarry Smith       v2    = v1 + n;
4784c1414c8SBarry Smith       v3    = v2 + n;
4794c1414c8SBarry Smith 
4804c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4814c1414c8SBarry Smith         i1   = idx[0];
4824c1414c8SBarry Smith         i2   = idx[1];
4834c1414c8SBarry Smith         idx += 2;
4844c1414c8SBarry Smith         tmp0 = x[i1];
4854c1414c8SBarry Smith         tmp1 = x[i2];
4864c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4874c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
4884c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
4894c1414c8SBarry Smith       }
4904c1414c8SBarry Smith       if (n == sz-1){
4914c1414c8SBarry Smith         tmp0  = x[*idx++];
4924c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
4934c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
4944c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
4954c1414c8SBarry Smith       }
4964c1414c8SBarry Smith       y[row++]=sum1;
4974c1414c8SBarry Smith       y[row++]=sum2;
4984c1414c8SBarry Smith       y[row++]=sum3;
4994c1414c8SBarry Smith       v1       =v3;             /* Since the next block to be processed starts there*/
5004c1414c8SBarry Smith       idx     +=2*sz;
5014c1414c8SBarry Smith       break;
5024c1414c8SBarry Smith     case 4:
50375567043SBarry Smith       sum1  = 0.;
50475567043SBarry Smith       sum2  = 0.;
50575567043SBarry Smith       sum3  = 0.;
50675567043SBarry Smith       sum4  = 0.;
5074c1414c8SBarry Smith       v2    = v1 + n;
5084c1414c8SBarry Smith       v3    = v2 + n;
5094c1414c8SBarry Smith       v4    = v3 + n;
5104c1414c8SBarry Smith 
5114c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
5124c1414c8SBarry Smith         i1   = idx[0];
5134c1414c8SBarry Smith         i2   = idx[1];
5144c1414c8SBarry Smith         idx += 2;
5154c1414c8SBarry Smith         tmp0 = x[i1];
5164c1414c8SBarry Smith         tmp1 = x[i2];
5174c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
5184c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
5194c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
5204c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
5214c1414c8SBarry Smith       }
5224c1414c8SBarry Smith       if (n == sz-1){
5234c1414c8SBarry Smith         tmp0  = x[*idx++];
5244c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
5254c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
5264c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
5274c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
5284c1414c8SBarry Smith       }
5294c1414c8SBarry Smith       y[row++]=sum1;
5304c1414c8SBarry Smith       y[row++]=sum2;
5314c1414c8SBarry Smith       y[row++]=sum3;
5324c1414c8SBarry Smith       y[row++]=sum4;
5334c1414c8SBarry Smith       v1      =v4;              /* Since the next block to be processed starts there*/
5344c1414c8SBarry Smith       idx    +=3*sz;
5354c1414c8SBarry Smith       break;
5364c1414c8SBarry Smith     case 5:
53775567043SBarry Smith       sum1  = 0.;
53875567043SBarry Smith       sum2  = 0.;
53975567043SBarry Smith       sum3  = 0.;
54075567043SBarry Smith       sum4  = 0.;
54175567043SBarry Smith       sum5  = 0.;
5424c1414c8SBarry Smith       v2    = v1 + n;
5434c1414c8SBarry Smith       v3    = v2 + n;
5444c1414c8SBarry Smith       v4    = v3 + n;
5454c1414c8SBarry Smith       v5    = v4 + n;
5464c1414c8SBarry Smith 
5474c1414c8SBarry Smith       for (n = 0; n<sz-1; n+=2) {
5484c1414c8SBarry Smith         i1   = idx[0];
5494c1414c8SBarry Smith         i2   = idx[1];
5504c1414c8SBarry Smith         idx += 2;
5514c1414c8SBarry Smith         tmp0 = x[i1];
5524c1414c8SBarry Smith         tmp1 = x[i2];
5534c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
5544c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
5554c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
5564c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
5574c1414c8SBarry Smith         sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2;
5584c1414c8SBarry Smith       }
5594c1414c8SBarry Smith       if (n == sz-1){
5604c1414c8SBarry Smith         tmp0  = x[*idx++];
5614c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
5624c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
5634c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
5644c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
5654c1414c8SBarry Smith         sum5 += *v5++ * tmp0;
5664c1414c8SBarry Smith       }
5674c1414c8SBarry Smith       y[row++]=sum1;
5684c1414c8SBarry Smith       y[row++]=sum2;
5694c1414c8SBarry Smith       y[row++]=sum3;
5704c1414c8SBarry Smith       y[row++]=sum4;
5714c1414c8SBarry Smith       y[row++]=sum5;
5724c1414c8SBarry Smith       v1      =v5;       /* Since the next block to be processed starts there */
5734c1414c8SBarry Smith       idx    +=4*sz;
5744c1414c8SBarry Smith       break;
5754c1414c8SBarry Smith     default :
5764c1414c8SBarry Smith       SETERRQ(PETSC_ERR_COR,"Node size not yet supported");
5774c1414c8SBarry Smith     }
5784c1414c8SBarry Smith   }
579d9fead3dSBarry Smith   ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr);
5804c1414c8SBarry Smith   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
581dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr);
5824c1414c8SBarry Smith   PetscFunctionReturn(0);
5834c1414c8SBarry Smith }
5844c1414c8SBarry Smith /* ----------------------------------------------------------- */
5854108e4d5SBarry Smith /* Almost same code as the MatMult_SeqAIJ_Inode() */
5864c1414c8SBarry Smith #undef __FUNCT__
5874108e4d5SBarry Smith #define __FUNCT__ "MatMultAdd_SeqAIJ_Inode"
5884108e4d5SBarry Smith static PetscErrorCode MatMultAdd_SeqAIJ_Inode(Mat A,Vec xx,Vec zz,Vec yy)
5894c1414c8SBarry Smith {
5904c1414c8SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
5914c1414c8SBarry Smith   PetscScalar    sum1,sum2,sum3,sum4,sum5,tmp0,tmp1;
592dd6ea824SBarry Smith   MatScalar      *v1,*v2,*v3,*v4,*v5;
593dd6ea824SBarry Smith   PetscScalar    *x,*y,*z,*zt;
5944c1414c8SBarry Smith   PetscErrorCode ierr;
5954c1414c8SBarry Smith   PetscInt       *idx,i1,i2,n,i,row,node_max,*ns,*ii,nsz,sz;
5964c1414c8SBarry Smith 
5974c1414c8SBarry Smith   PetscFunctionBegin;
5984c1414c8SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_ERR_COR,"Missing Inode Structure");
5994c1414c8SBarry Smith   node_max = a->inode.node_count;
6004c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
6014c1414c8SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
6024c1414c8SBarry Smith   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
6034c1414c8SBarry Smith   if (zz != yy) {
6044c1414c8SBarry Smith     ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
6054c1414c8SBarry Smith   } else {
6064c1414c8SBarry Smith     z = y;
6074c1414c8SBarry Smith   }
6084c1414c8SBarry Smith   zt = z;
6094c1414c8SBarry Smith 
6104c1414c8SBarry Smith   idx  = a->j;
6114c1414c8SBarry Smith   v1   = a->a;
6124c1414c8SBarry Smith   ii   = a->i;
6134c1414c8SBarry Smith 
6144c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i){
6154c1414c8SBarry Smith     nsz  = ns[i];
6164c1414c8SBarry Smith     n    = ii[1] - ii[0];
6174c1414c8SBarry Smith     ii  += nsz;
6184c1414c8SBarry Smith     sz   = n;                   /* No of non zeros in this row */
6194c1414c8SBarry Smith                                 /* Switch on the size of Node */
6204c1414c8SBarry Smith     switch (nsz){               /* Each loop in 'case' is unrolled */
6214c1414c8SBarry Smith     case 1 :
6224c1414c8SBarry Smith       sum1  = *zt++;
6234c1414c8SBarry Smith 
6244c1414c8SBarry Smith       for(n = 0; n< sz-1; n+=2) {
6254c1414c8SBarry Smith         i1   = idx[0];          /* The instructions are ordered to */
6264c1414c8SBarry Smith         i2   = idx[1];          /* make the compiler's job easy */
6274c1414c8SBarry Smith         idx += 2;
6284c1414c8SBarry Smith         tmp0 = x[i1];
6294c1414c8SBarry Smith         tmp1 = x[i2];
6304c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6314c1414c8SBarry Smith        }
6324c1414c8SBarry Smith 
6334c1414c8SBarry Smith       if(n   == sz-1){          /* Take care of the last nonzero  */
6344c1414c8SBarry Smith         tmp0  = x[*idx++];
6354c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6364c1414c8SBarry Smith       }
6374c1414c8SBarry Smith       y[row++]=sum1;
6384c1414c8SBarry Smith       break;
6394c1414c8SBarry Smith     case 2:
6404c1414c8SBarry Smith       sum1  = *zt++;
6414c1414c8SBarry Smith       sum2  = *zt++;
6424c1414c8SBarry Smith       v2    = v1 + n;
6434c1414c8SBarry Smith 
6444c1414c8SBarry Smith       for(n = 0; n< sz-1; n+=2) {
6454c1414c8SBarry Smith         i1   = idx[0];
6464c1414c8SBarry Smith         i2   = idx[1];
6474c1414c8SBarry Smith         idx += 2;
6484c1414c8SBarry Smith         tmp0 = x[i1];
6494c1414c8SBarry Smith         tmp1 = x[i2];
6504c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6514c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
6524c1414c8SBarry Smith       }
6534c1414c8SBarry Smith       if(n   == sz-1){
6544c1414c8SBarry Smith         tmp0  = x[*idx++];
6554c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6564c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
6574c1414c8SBarry Smith       }
6584c1414c8SBarry Smith       y[row++]=sum1;
6594c1414c8SBarry Smith       y[row++]=sum2;
6604c1414c8SBarry Smith       v1      =v2;              /* Since the next block to be processed starts there*/
6614c1414c8SBarry Smith       idx    +=sz;
6624c1414c8SBarry Smith       break;
6634c1414c8SBarry Smith     case 3:
6644c1414c8SBarry Smith       sum1  = *zt++;
6654c1414c8SBarry Smith       sum2  = *zt++;
6664c1414c8SBarry Smith       sum3  = *zt++;
6674c1414c8SBarry Smith       v2    = v1 + n;
6684c1414c8SBarry Smith       v3    = v2 + n;
6694c1414c8SBarry Smith 
6704c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6714c1414c8SBarry Smith         i1   = idx[0];
6724c1414c8SBarry Smith         i2   = idx[1];
6734c1414c8SBarry Smith         idx += 2;
6744c1414c8SBarry Smith         tmp0 = x[i1];
6754c1414c8SBarry Smith         tmp1 = x[i2];
6764c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6774c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
6784c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
6794c1414c8SBarry Smith       }
6804c1414c8SBarry Smith       if (n == sz-1){
6814c1414c8SBarry Smith         tmp0  = x[*idx++];
6824c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6834c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
6844c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
6854c1414c8SBarry Smith       }
6864c1414c8SBarry Smith       y[row++]=sum1;
6874c1414c8SBarry Smith       y[row++]=sum2;
6884c1414c8SBarry Smith       y[row++]=sum3;
6894c1414c8SBarry Smith       v1       =v3;             /* Since the next block to be processed starts there*/
6904c1414c8SBarry Smith       idx     +=2*sz;
6914c1414c8SBarry Smith       break;
6924c1414c8SBarry Smith     case 4:
6934c1414c8SBarry Smith       sum1  = *zt++;
6944c1414c8SBarry Smith       sum2  = *zt++;
6954c1414c8SBarry Smith       sum3  = *zt++;
6964c1414c8SBarry Smith       sum4  = *zt++;
6974c1414c8SBarry Smith       v2    = v1 + n;
6984c1414c8SBarry Smith       v3    = v2 + n;
6994c1414c8SBarry Smith       v4    = v3 + n;
7004c1414c8SBarry Smith 
7014c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
7024c1414c8SBarry Smith         i1   = idx[0];
7034c1414c8SBarry Smith         i2   = idx[1];
7044c1414c8SBarry Smith         idx += 2;
7054c1414c8SBarry Smith         tmp0 = x[i1];
7064c1414c8SBarry Smith         tmp1 = x[i2];
7074c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
7084c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
7094c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
7104c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
7114c1414c8SBarry Smith       }
7124c1414c8SBarry Smith       if (n == sz-1){
7134c1414c8SBarry Smith         tmp0  = x[*idx++];
7144c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
7154c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
7164c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
7174c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
7184c1414c8SBarry Smith       }
7194c1414c8SBarry Smith       y[row++]=sum1;
7204c1414c8SBarry Smith       y[row++]=sum2;
7214c1414c8SBarry Smith       y[row++]=sum3;
7224c1414c8SBarry Smith       y[row++]=sum4;
7234c1414c8SBarry Smith       v1      =v4;              /* Since the next block to be processed starts there*/
7244c1414c8SBarry Smith       idx    +=3*sz;
7254c1414c8SBarry Smith       break;
7264c1414c8SBarry Smith     case 5:
7274c1414c8SBarry Smith       sum1  = *zt++;
7284c1414c8SBarry Smith       sum2  = *zt++;
7294c1414c8SBarry Smith       sum3  = *zt++;
7304c1414c8SBarry Smith       sum4  = *zt++;
7314c1414c8SBarry Smith       sum5  = *zt++;
7324c1414c8SBarry Smith       v2    = v1 + n;
7334c1414c8SBarry Smith       v3    = v2 + n;
7344c1414c8SBarry Smith       v4    = v3 + n;
7354c1414c8SBarry Smith       v5    = v4 + n;
7364c1414c8SBarry Smith 
7374c1414c8SBarry Smith       for (n = 0; n<sz-1; n+=2) {
7384c1414c8SBarry Smith         i1   = idx[0];
7394c1414c8SBarry Smith         i2   = idx[1];
7404c1414c8SBarry Smith         idx += 2;
7414c1414c8SBarry Smith         tmp0 = x[i1];
7424c1414c8SBarry Smith         tmp1 = x[i2];
7434c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
7444c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
7454c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
7464c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
7474c1414c8SBarry Smith         sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2;
7484c1414c8SBarry Smith       }
7494c1414c8SBarry Smith       if(n   == sz-1){
7504c1414c8SBarry Smith         tmp0  = x[*idx++];
7514c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
7524c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
7534c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
7544c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
7554c1414c8SBarry Smith         sum5 += *v5++ * tmp0;
7564c1414c8SBarry Smith       }
7574c1414c8SBarry Smith       y[row++]=sum1;
7584c1414c8SBarry Smith       y[row++]=sum2;
7594c1414c8SBarry Smith       y[row++]=sum3;
7604c1414c8SBarry Smith       y[row++]=sum4;
7614c1414c8SBarry Smith       y[row++]=sum5;
7624c1414c8SBarry Smith       v1      =v5;       /* Since the next block to be processed starts there */
7634c1414c8SBarry Smith       idx    +=4*sz;
7644c1414c8SBarry Smith       break;
7654c1414c8SBarry Smith     default :
7664c1414c8SBarry Smith       SETERRQ(PETSC_ERR_COR,"Node size not yet supported");
7674c1414c8SBarry Smith     }
7684c1414c8SBarry Smith   }
7694c1414c8SBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
7704c1414c8SBarry Smith   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
7714c1414c8SBarry Smith   if (zz != yy) {
7724c1414c8SBarry Smith     ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
7734c1414c8SBarry Smith   }
774dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
7754c1414c8SBarry Smith   PetscFunctionReturn(0);
7764c1414c8SBarry Smith }
7774c1414c8SBarry Smith 
7784c1414c8SBarry Smith /* ----------------------------------------------------------- */
7794c1414c8SBarry Smith #undef __FUNCT__
780ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_Inode_inplace"
781ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_Inode_inplace(Mat A,Vec bb,Vec xx)
7824c1414c8SBarry Smith {
7834c1414c8SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
7844c1414c8SBarry Smith   IS                iscol = a->col,isrow = a->row;
7854c1414c8SBarry Smith   PetscErrorCode    ierr;
7865d0c19d7SBarry Smith   const PetscInt    *r,*c,*rout,*cout;
7875d0c19d7SBarry Smith   PetscInt          i,j,n = A->rmap->n,*ai = a->i,nz,*a_j = a->j;
7885d0c19d7SBarry Smith   PetscInt          node_max,*ns,row,nsz,aii,*vi,*ad,*aj,i0,i1;
789d9fead3dSBarry Smith   PetscScalar       *x,*tmp,*tmps,tmp0,tmp1;
790d9fead3dSBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5;
791dd6ea824SBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa;
792dd6ea824SBarry Smith   const PetscScalar *b;
7934c1414c8SBarry Smith 
7944c1414c8SBarry Smith   PetscFunctionBegin;
7954c1414c8SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_ERR_COR,"Missing Inode Structure");
7964c1414c8SBarry Smith   node_max = a->inode.node_count;
7974c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
7984c1414c8SBarry Smith 
799d9fead3dSBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
8004c1414c8SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
8014c1414c8SBarry Smith   tmp  = a->solve_work;
8024c1414c8SBarry Smith 
8034c1414c8SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
8044c1414c8SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
8054c1414c8SBarry Smith 
8064c1414c8SBarry Smith   /* forward solve the lower triangular */
8074c1414c8SBarry Smith   tmps = tmp ;
8084c1414c8SBarry Smith   aa   = a_a ;
8094c1414c8SBarry Smith   aj   = a_j ;
8104c1414c8SBarry Smith   ad   = a->diag;
8114c1414c8SBarry Smith 
8124c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i){
8134c1414c8SBarry Smith     nsz = ns[i];
8144c1414c8SBarry Smith     aii = ai[row];
8154c1414c8SBarry Smith     v1  = aa + aii;
8164c1414c8SBarry Smith     vi  = aj + aii;
8174c1414c8SBarry Smith     nz  = ad[row]- aii;
81826549573SJed Brown     if (i < node_max-1) {
81926549573SJed Brown       /* Prefetch the block after the current one, the prefetch itself can't cause a memory error,
82026549573SJed Brown       * but our indexing to determine it's size could. */
82126549573SJed Brown       PetscPrefetchBlock(aj+ai[row+nsz],ad[row+nsz]-ai[row+nsz],0,0); /* indices */
82226549573SJed Brown       /* In my tests, it seems to be better to fetch entire rows instead of just the lower-triangular part */
82326549573SJed Brown       PetscPrefetchBlock(aa+ai[row+nsz],ad[row+nsz+ns[i+1]-1]-ai[row+nsz],0,0);
82426549573SJed Brown       /* for (j=0; j<ns[i+1]; j++) PetscPrefetchBlock(aa+ai[row+nsz+j],ad[row+nsz+j]-ai[row+nsz+j],0,0); */
82526549573SJed Brown     }
8264c1414c8SBarry Smith 
8274c1414c8SBarry Smith     switch (nsz){               /* Each loop in 'case' is unrolled */
8284c1414c8SBarry Smith     case 1 :
8294c1414c8SBarry Smith       sum1 = b[*r++];
8304c1414c8SBarry Smith       for(j=0; j<nz-1; j+=2){
8314c1414c8SBarry Smith         i0   = vi[0];
8324c1414c8SBarry Smith         i1   = vi[1];
8334c1414c8SBarry Smith         vi  +=2;
8344c1414c8SBarry Smith         tmp0 = tmps[i0];
8354c1414c8SBarry Smith         tmp1 = tmps[i1];
8364c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8374c1414c8SBarry Smith       }
8384c1414c8SBarry Smith       if(j == nz-1){
8394c1414c8SBarry Smith         tmp0 = tmps[*vi++];
8404c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8414c1414c8SBarry Smith       }
8424c1414c8SBarry Smith       tmp[row ++]=sum1;
8434c1414c8SBarry Smith       break;
8444c1414c8SBarry Smith     case 2:
8454c1414c8SBarry Smith       sum1 = b[*r++];
8464c1414c8SBarry Smith       sum2 = b[*r++];
8474c1414c8SBarry Smith       v2   = aa + ai[row+1];
8484c1414c8SBarry Smith 
8494c1414c8SBarry Smith       for(j=0; j<nz-1; j+=2){
8504c1414c8SBarry Smith         i0   = vi[0];
8514c1414c8SBarry Smith         i1   = vi[1];
8524c1414c8SBarry Smith         vi  +=2;
8534c1414c8SBarry Smith         tmp0 = tmps[i0];
8544c1414c8SBarry Smith         tmp1 = tmps[i1];
8554c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8564c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
8574c1414c8SBarry Smith       }
8584c1414c8SBarry Smith       if(j == nz-1){
8594c1414c8SBarry Smith         tmp0 = tmps[*vi++];
8604c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8614c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
8624c1414c8SBarry Smith       }
8634c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
8644c1414c8SBarry Smith       tmp[row ++]=sum1;
8654c1414c8SBarry Smith       tmp[row ++]=sum2;
8664c1414c8SBarry Smith       break;
8674c1414c8SBarry Smith     case 3:
8684c1414c8SBarry Smith       sum1 = b[*r++];
8694c1414c8SBarry Smith       sum2 = b[*r++];
8704c1414c8SBarry Smith       sum3 = b[*r++];
8714c1414c8SBarry Smith       v2   = aa + ai[row+1];
8724c1414c8SBarry Smith       v3   = aa + ai[row+2];
8734c1414c8SBarry Smith 
8744c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2){
8754c1414c8SBarry Smith         i0   = vi[0];
8764c1414c8SBarry Smith         i1   = vi[1];
8774c1414c8SBarry Smith         vi  +=2;
8784c1414c8SBarry Smith         tmp0 = tmps[i0];
8794c1414c8SBarry Smith         tmp1 = tmps[i1];
8804c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8814c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
8824c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
8834c1414c8SBarry Smith       }
8844c1414c8SBarry Smith       if (j == nz-1){
8854c1414c8SBarry Smith         tmp0 = tmps[*vi++];
8864c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8874c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
8884c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
8894c1414c8SBarry Smith       }
8904c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
8914c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
8924c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
8934c1414c8SBarry Smith       tmp[row ++]=sum1;
8944c1414c8SBarry Smith       tmp[row ++]=sum2;
8954c1414c8SBarry Smith       tmp[row ++]=sum3;
8964c1414c8SBarry Smith       break;
8974c1414c8SBarry Smith 
8984c1414c8SBarry Smith     case 4:
8994c1414c8SBarry Smith       sum1 = b[*r++];
9004c1414c8SBarry Smith       sum2 = b[*r++];
9014c1414c8SBarry Smith       sum3 = b[*r++];
9024c1414c8SBarry Smith       sum4 = b[*r++];
9034c1414c8SBarry Smith       v2   = aa + ai[row+1];
9044c1414c8SBarry Smith       v3   = aa + ai[row+2];
9054c1414c8SBarry Smith       v4   = aa + ai[row+3];
9064c1414c8SBarry Smith 
9074c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2){
9084c1414c8SBarry Smith         i0   = vi[0];
9094c1414c8SBarry Smith         i1   = vi[1];
9104c1414c8SBarry Smith         vi  +=2;
9114c1414c8SBarry Smith         tmp0 = tmps[i0];
9124c1414c8SBarry Smith         tmp1 = tmps[i1];
9134c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
9144c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
9154c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
9164c1414c8SBarry Smith         sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
9174c1414c8SBarry Smith       }
9184c1414c8SBarry Smith       if (j == nz-1){
9194c1414c8SBarry Smith         tmp0 = tmps[*vi++];
9204c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
9214c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
9224c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
9234c1414c8SBarry Smith         sum4 -= *v4++ *tmp0;
9244c1414c8SBarry Smith       }
9254c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
9264c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
9274c1414c8SBarry Smith       sum4 -= *v4++ * sum1;
9284c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
9294c1414c8SBarry Smith       sum4 -= *v4++ * sum2;
9304c1414c8SBarry Smith       sum4 -= *v4++ * sum3;
9314c1414c8SBarry Smith 
9324c1414c8SBarry Smith       tmp[row ++]=sum1;
9334c1414c8SBarry Smith       tmp[row ++]=sum2;
9344c1414c8SBarry Smith       tmp[row ++]=sum3;
9354c1414c8SBarry Smith       tmp[row ++]=sum4;
9364c1414c8SBarry Smith       break;
9374c1414c8SBarry Smith     case 5:
9384c1414c8SBarry Smith       sum1 = b[*r++];
9394c1414c8SBarry Smith       sum2 = b[*r++];
9404c1414c8SBarry Smith       sum3 = b[*r++];
9414c1414c8SBarry Smith       sum4 = b[*r++];
9424c1414c8SBarry Smith       sum5 = b[*r++];
9434c1414c8SBarry Smith       v2   = aa + ai[row+1];
9444c1414c8SBarry Smith       v3   = aa + ai[row+2];
9454c1414c8SBarry Smith       v4   = aa + ai[row+3];
9464c1414c8SBarry Smith       v5   = aa + ai[row+4];
9474c1414c8SBarry Smith 
9484c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2){
9494c1414c8SBarry Smith         i0   = vi[0];
9504c1414c8SBarry Smith         i1   = vi[1];
9514c1414c8SBarry Smith         vi  +=2;
9524c1414c8SBarry Smith         tmp0 = tmps[i0];
9534c1414c8SBarry Smith         tmp1 = tmps[i1];
9544c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
9554c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
9564c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
9574c1414c8SBarry Smith         sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
9584c1414c8SBarry Smith         sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
9594c1414c8SBarry Smith       }
9604c1414c8SBarry Smith       if (j == nz-1){
9614c1414c8SBarry Smith         tmp0 = tmps[*vi++];
9624c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
9634c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
9644c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
9654c1414c8SBarry Smith         sum4 -= *v4++ *tmp0;
9664c1414c8SBarry Smith         sum5 -= *v5++ *tmp0;
9674c1414c8SBarry Smith       }
9684c1414c8SBarry Smith 
9694c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
9704c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
9714c1414c8SBarry Smith       sum4 -= *v4++ * sum1;
9724c1414c8SBarry Smith       sum5 -= *v5++ * sum1;
9734c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
9744c1414c8SBarry Smith       sum4 -= *v4++ * sum2;
9754c1414c8SBarry Smith       sum5 -= *v5++ * sum2;
9764c1414c8SBarry Smith       sum4 -= *v4++ * sum3;
9774c1414c8SBarry Smith       sum5 -= *v5++ * sum3;
9784c1414c8SBarry Smith       sum5 -= *v5++ * sum4;
9794c1414c8SBarry Smith 
9804c1414c8SBarry Smith       tmp[row ++]=sum1;
9814c1414c8SBarry Smith       tmp[row ++]=sum2;
9824c1414c8SBarry Smith       tmp[row ++]=sum3;
9834c1414c8SBarry Smith       tmp[row ++]=sum4;
9844c1414c8SBarry Smith       tmp[row ++]=sum5;
9854c1414c8SBarry Smith       break;
9864c1414c8SBarry Smith     default:
9874c1414c8SBarry Smith       SETERRQ(PETSC_ERR_COR,"Node size not yet supported \n");
9884c1414c8SBarry Smith     }
9894c1414c8SBarry Smith   }
9904c1414c8SBarry Smith   /* backward solve the upper triangular */
9914c1414c8SBarry Smith   for (i=node_max -1 ,row = n-1 ; i>=0; i--){
9924c1414c8SBarry Smith     nsz = ns[i];
9934c1414c8SBarry Smith     aii = ai[row+1] -1;
9944c1414c8SBarry Smith     v1  = aa + aii;
9954c1414c8SBarry Smith     vi  = aj + aii;
9964c1414c8SBarry Smith     nz  = aii- ad[row];
9974c1414c8SBarry Smith     switch (nsz){               /* Each loop in 'case' is unrolled */
9984c1414c8SBarry Smith     case 1 :
9994c1414c8SBarry Smith       sum1 = tmp[row];
10004c1414c8SBarry Smith 
10014c1414c8SBarry Smith       for(j=nz ; j>1; j-=2){
10024c1414c8SBarry Smith         vi  -=2;
10034c1414c8SBarry Smith         i0   = vi[2];
10044c1414c8SBarry Smith         i1   = vi[1];
10054c1414c8SBarry Smith         tmp0 = tmps[i0];
10064c1414c8SBarry Smith         tmp1 = tmps[i1];
10074c1414c8SBarry Smith         v1   -= 2;
10084c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10094c1414c8SBarry Smith       }
10104c1414c8SBarry Smith       if (j==1){
10114c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10124c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10134c1414c8SBarry Smith       }
10144c1414c8SBarry Smith       x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10154c1414c8SBarry Smith       break;
10164c1414c8SBarry Smith     case 2 :
10174c1414c8SBarry Smith       sum1 = tmp[row];
10184c1414c8SBarry Smith       sum2 = tmp[row -1];
10194c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10204c1414c8SBarry Smith       for (j=nz ; j>1; j-=2){
10214c1414c8SBarry Smith         vi  -=2;
10224c1414c8SBarry Smith         i0   = vi[2];
10234c1414c8SBarry Smith         i1   = vi[1];
10244c1414c8SBarry Smith         tmp0 = tmps[i0];
10254c1414c8SBarry Smith         tmp1 = tmps[i1];
10264c1414c8SBarry Smith         v1   -= 2;
10274c1414c8SBarry Smith         v2   -= 2;
10284c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10294c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10304c1414c8SBarry Smith       }
10314c1414c8SBarry Smith       if (j==1){
10324c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10334c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10344c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10354c1414c8SBarry Smith       }
10364c1414c8SBarry Smith 
10374c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10384c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10394c1414c8SBarry Smith       x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
10404c1414c8SBarry Smith       break;
10414c1414c8SBarry Smith     case 3 :
10424c1414c8SBarry Smith       sum1 = tmp[row];
10434c1414c8SBarry Smith       sum2 = tmp[row -1];
10444c1414c8SBarry Smith       sum3 = tmp[row -2];
10454c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10464c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
10474c1414c8SBarry Smith       for (j=nz ; j>1; j-=2){
10484c1414c8SBarry Smith         vi  -=2;
10494c1414c8SBarry Smith         i0   = vi[2];
10504c1414c8SBarry Smith         i1   = vi[1];
10514c1414c8SBarry Smith         tmp0 = tmps[i0];
10524c1414c8SBarry Smith         tmp1 = tmps[i1];
10534c1414c8SBarry Smith         v1   -= 2;
10544c1414c8SBarry Smith         v2   -= 2;
10554c1414c8SBarry Smith         v3   -= 2;
10564c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10574c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10584c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
10594c1414c8SBarry Smith       }
10604c1414c8SBarry Smith       if (j==1){
10614c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10624c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10634c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10644c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
10654c1414c8SBarry Smith       }
10664c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10674c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10684c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10694c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
10704c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10714c1414c8SBarry Smith       x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
10724c1414c8SBarry Smith 
10734c1414c8SBarry Smith       break;
10744c1414c8SBarry Smith     case 4 :
10754c1414c8SBarry Smith       sum1 = tmp[row];
10764c1414c8SBarry Smith       sum2 = tmp[row -1];
10774c1414c8SBarry Smith       sum3 = tmp[row -2];
10784c1414c8SBarry Smith       sum4 = tmp[row -3];
10794c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10804c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
10814c1414c8SBarry Smith       v4   = aa + ai[row -2]-1;
10824c1414c8SBarry Smith 
10834c1414c8SBarry Smith       for (j=nz ; j>1; j-=2){
10844c1414c8SBarry Smith         vi  -=2;
10854c1414c8SBarry Smith         i0   = vi[2];
10864c1414c8SBarry Smith         i1   = vi[1];
10874c1414c8SBarry Smith         tmp0 = tmps[i0];
10884c1414c8SBarry Smith         tmp1 = tmps[i1];
10894c1414c8SBarry Smith         v1  -= 2;
10904c1414c8SBarry Smith         v2  -= 2;
10914c1414c8SBarry Smith         v3  -= 2;
10924c1414c8SBarry Smith         v4  -= 2;
10934c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10944c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10954c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
10964c1414c8SBarry Smith         sum4 -= v4[2] * tmp0 + v4[1] * tmp1;
10974c1414c8SBarry Smith       }
10984c1414c8SBarry Smith       if (j==1){
10994c1414c8SBarry Smith         tmp0  = tmps[*vi--];
11004c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
11014c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
11024c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
11034c1414c8SBarry Smith         sum4 -= *v4-- * tmp0;
11044c1414c8SBarry Smith       }
11054c1414c8SBarry Smith 
11064c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
11074c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
11084c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11094c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11104c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
11114c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11124c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11134c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
11144c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11154c1414c8SBarry Smith       x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--;
11164c1414c8SBarry Smith       break;
11174c1414c8SBarry Smith     case 5 :
11184c1414c8SBarry Smith       sum1 = tmp[row];
11194c1414c8SBarry Smith       sum2 = tmp[row -1];
11204c1414c8SBarry Smith       sum3 = tmp[row -2];
11214c1414c8SBarry Smith       sum4 = tmp[row -3];
11224c1414c8SBarry Smith       sum5 = tmp[row -4];
11234c1414c8SBarry Smith       v2   = aa + ai[row]-1;
11244c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
11254c1414c8SBarry Smith       v4   = aa + ai[row -2]-1;
11264c1414c8SBarry Smith       v5   = aa + ai[row -3]-1;
11274c1414c8SBarry Smith       for (j=nz ; j>1; j-=2){
11284c1414c8SBarry Smith         vi  -= 2;
11294c1414c8SBarry Smith         i0   = vi[2];
11304c1414c8SBarry Smith         i1   = vi[1];
11314c1414c8SBarry Smith         tmp0 = tmps[i0];
11324c1414c8SBarry Smith         tmp1 = tmps[i1];
11334c1414c8SBarry Smith         v1   -= 2;
11344c1414c8SBarry Smith         v2   -= 2;
11354c1414c8SBarry Smith         v3   -= 2;
11364c1414c8SBarry Smith         v4   -= 2;
11374c1414c8SBarry Smith         v5   -= 2;
11384c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
11394c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
11404c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
11414c1414c8SBarry Smith         sum4 -= v4[2] * tmp0 + v4[1] * tmp1;
11424c1414c8SBarry Smith         sum5 -= v5[2] * tmp0 + v5[1] * tmp1;
11434c1414c8SBarry Smith       }
11444c1414c8SBarry Smith       if (j==1){
11454c1414c8SBarry Smith         tmp0  = tmps[*vi--];
11464c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
11474c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
11484c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
11494c1414c8SBarry Smith         sum4 -= *v4-- * tmp0;
11504c1414c8SBarry Smith         sum5 -= *v5-- * tmp0;
11514c1414c8SBarry Smith       }
11524c1414c8SBarry Smith 
11534c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
11544c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
11554c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11564c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11574c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11584c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
11594c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11604c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11614c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11624c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
11634c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11644c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11654c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--;
11664c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11674c1414c8SBarry Smith       x[*c--] = tmp[row] = sum5*a_a[ad[row]]; row--;
11684c1414c8SBarry Smith       break;
11694c1414c8SBarry Smith     default:
11704c1414c8SBarry Smith       SETERRQ(PETSC_ERR_COR,"Node size not yet supported \n");
11714c1414c8SBarry Smith     }
11724c1414c8SBarry Smith   }
11734c1414c8SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11744c1414c8SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1175d9fead3dSBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
11764c1414c8SBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1177dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
11784c1414c8SBarry Smith   PetscFunctionReturn(0);
11794c1414c8SBarry Smith }
11804c1414c8SBarry Smith 
1181019b515eSShri Abhyankar /* ----------------------------------------------------------- */
1182019b515eSShri Abhyankar #undef __FUNCT__
1183019b515eSShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_Inode_newdatastruct"
1184019b515eSShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_Inode_newdatastruct(Mat A,Vec bb,Vec xx)
1185019b515eSShri Abhyankar {
1186019b515eSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1187019b515eSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1188019b515eSShri Abhyankar   PetscErrorCode    ierr;
1189019b515eSShri Abhyankar   const PetscInt    *r,*c,*rout,*cout;
1190019b515eSShri Abhyankar   PetscInt          i,j,n = A->rmap->n,*ai = a->i,nz,*a_j = a->j;
1191019b515eSShri Abhyankar   PetscInt          node_max,*ns,row,nsz,aii,*vi,*ad,*aj,i0,i1;
1192019b515eSShri Abhyankar   PetscScalar       *x,*tmp,*tmps,tmp0,tmp1;
1193019b515eSShri Abhyankar   PetscScalar       sum1,sum2,sum3,sum4,sum5;
1194019b515eSShri Abhyankar   const MatScalar   *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa;
1195019b515eSShri Abhyankar   const PetscScalar *b;
1196019b515eSShri Abhyankar 
1197019b515eSShri Abhyankar   PetscFunctionBegin;
1198019b515eSShri Abhyankar   if (!a->inode.size) SETERRQ(PETSC_ERR_COR,"Missing Inode Structure");
1199019b515eSShri Abhyankar   node_max = a->inode.node_count;
1200019b515eSShri Abhyankar   ns       = a->inode.size;     /* Node Size array */
1201019b515eSShri Abhyankar 
1202019b515eSShri Abhyankar   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
1203019b515eSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1204019b515eSShri Abhyankar   tmp  = a->solve_work;
1205019b515eSShri Abhyankar 
1206019b515eSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1207019b515eSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1208019b515eSShri Abhyankar 
1209019b515eSShri Abhyankar   /* forward solve the lower triangular */
1210019b515eSShri Abhyankar   tmps = tmp ;
1211019b515eSShri Abhyankar   aa   = a_a ;
1212019b515eSShri Abhyankar   aj   = a_j ;
1213019b515eSShri Abhyankar   ad   = a->diag;
1214019b515eSShri Abhyankar 
1215019b515eSShri Abhyankar   for (i = 0,row = 0; i< node_max; ++i){
1216019b515eSShri Abhyankar     nsz = ns[i];
1217019b515eSShri Abhyankar     aii = ai[row];
1218019b515eSShri Abhyankar     v1  = aa + aii;
1219019b515eSShri Abhyankar     vi  = aj + aii;
1220019b515eSShri Abhyankar     nz  = ai[row+1]- ai[row];
1221019b515eSShri Abhyankar 
1222*98991853SShri Abhyankar     if (i < node_max-1) {
1223*98991853SShri Abhyankar       /* Prefetch the indices for the next block */
1224*98991853SShri Abhyankar       PetscPrefetchBlock(aj+ai[row+nsz],ai[row+nsz+1]-ai[row+nsz],0,0); /* indices */
1225*98991853SShri Abhyankar       /* Prefetch the data for the next block */
1226*98991853SShri Abhyankar       PetscPrefetchBlock(aa+ai[row+nsz],ai[row+nsz+ns[i+1]]-ai[row+nsz],0,0);
1227*98991853SShri Abhyankar     }
1228*98991853SShri Abhyankar 
1229019b515eSShri Abhyankar     switch (nsz){               /* Each loop in 'case' is unrolled */
1230019b515eSShri Abhyankar     case 1 :
1231019b515eSShri Abhyankar       sum1 = b[r[row]];
1232019b515eSShri Abhyankar       for(j=0; j<nz-1; j+=2){
1233019b515eSShri Abhyankar         i0   = vi[j];
1234019b515eSShri Abhyankar         i1   = vi[j+1];
1235019b515eSShri Abhyankar         tmp0 = tmps[i0];
1236019b515eSShri Abhyankar         tmp1 = tmps[i1];
1237019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0 + v1[j+1]*tmp1;
1238019b515eSShri Abhyankar       }
1239019b515eSShri Abhyankar       if(j == nz-1){
1240019b515eSShri Abhyankar         tmp0 = tmps[vi[j]];
1241019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0;
1242019b515eSShri Abhyankar       }
1243019b515eSShri Abhyankar       tmp[row++]=sum1;
1244019b515eSShri Abhyankar       break;
1245019b515eSShri Abhyankar     case 2:
1246019b515eSShri Abhyankar       sum1 = b[r[row]];
1247019b515eSShri Abhyankar       sum2 = b[r[row+1]];
1248019b515eSShri Abhyankar       v2   = aa + ai[row+1];
1249019b515eSShri Abhyankar 
1250019b515eSShri Abhyankar       for(j=0; j<nz-1; j+=2){
1251019b515eSShri Abhyankar         i0   = vi[j];
1252019b515eSShri Abhyankar         i1   = vi[j+1];
1253019b515eSShri Abhyankar         tmp0 = tmps[i0];
1254019b515eSShri Abhyankar         tmp1 = tmps[i1];
1255019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1256019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
1257019b515eSShri Abhyankar       }
1258019b515eSShri Abhyankar       if(j == nz-1){
1259019b515eSShri Abhyankar         tmp0 = tmps[vi[j]];
1260019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
1261019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
1262019b515eSShri Abhyankar       }
1263019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
1264019b515eSShri Abhyankar       tmp[row ++]=sum1;
1265019b515eSShri Abhyankar       tmp[row ++]=sum2;
1266019b515eSShri Abhyankar       break;
1267019b515eSShri Abhyankar     case 3:
1268019b515eSShri Abhyankar       sum1 = b[r[row]];
1269019b515eSShri Abhyankar       sum2 = b[r[row+1]];
1270019b515eSShri Abhyankar       sum3 = b[r[row+2]];
1271019b515eSShri Abhyankar       v2   = aa + ai[row+1];
1272019b515eSShri Abhyankar       v3   = aa + ai[row+2];
1273019b515eSShri Abhyankar 
1274019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2){
1275019b515eSShri Abhyankar         i0   = vi[j];
1276019b515eSShri Abhyankar         i1   = vi[j+1];
1277019b515eSShri Abhyankar         tmp0 = tmps[i0];
1278019b515eSShri Abhyankar         tmp1 = tmps[i1];
1279019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1280019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
1281019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
1282019b515eSShri Abhyankar       }
1283019b515eSShri Abhyankar       if (j == nz-1){
1284019b515eSShri Abhyankar         tmp0 = tmps[vi[j]];
1285019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
1286019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
1287019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
1288019b515eSShri Abhyankar       }
1289019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
1290019b515eSShri Abhyankar       sum3 -= v3[nz] * sum1;
1291019b515eSShri Abhyankar       sum3 -= v3[nz+1] * sum2;
1292019b515eSShri Abhyankar       tmp[row ++]=sum1;
1293019b515eSShri Abhyankar       tmp[row ++]=sum2;
1294019b515eSShri Abhyankar       tmp[row ++]=sum3;
1295019b515eSShri Abhyankar       break;
1296019b515eSShri Abhyankar 
1297019b515eSShri Abhyankar     case 4:
1298019b515eSShri Abhyankar       sum1 = b[r[row]];
1299019b515eSShri Abhyankar       sum2 = b[r[row+1]];
1300019b515eSShri Abhyankar       sum3 = b[r[row+2]];
1301019b515eSShri Abhyankar       sum4 = b[r[row+3]];
1302019b515eSShri Abhyankar       v2   = aa + ai[row+1];
1303019b515eSShri Abhyankar       v3   = aa + ai[row+2];
1304019b515eSShri Abhyankar       v4   = aa + ai[row+3];
1305019b515eSShri Abhyankar 
1306019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2){
1307019b515eSShri Abhyankar         i0   = vi[j];
1308019b515eSShri Abhyankar         i1   = vi[j+1];
1309019b515eSShri Abhyankar         tmp0 = tmps[i0];
1310019b515eSShri Abhyankar         tmp1 = tmps[i1];
1311019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1312019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
1313019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
1314019b515eSShri Abhyankar         sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1;
1315019b515eSShri Abhyankar       }
1316019b515eSShri Abhyankar       if (j == nz-1){
1317019b515eSShri Abhyankar         tmp0 = tmps[vi[j]];
1318019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
1319019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
1320019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
1321019b515eSShri Abhyankar         sum4 -= v4[j] *tmp0;
1322019b515eSShri Abhyankar       }
1323019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
1324019b515eSShri Abhyankar       sum3 -= v3[nz] * sum1;
1325019b515eSShri Abhyankar       sum4 -= v4[nz] * sum1;
1326019b515eSShri Abhyankar       sum3 -= v3[nz+1] * sum2;
1327019b515eSShri Abhyankar       sum4 -= v4[nz+1] * sum2;
1328019b515eSShri Abhyankar       sum4 -= v4[nz+2] * sum3;
1329019b515eSShri Abhyankar 
1330019b515eSShri Abhyankar       tmp[row ++]=sum1;
1331019b515eSShri Abhyankar       tmp[row ++]=sum2;
1332019b515eSShri Abhyankar       tmp[row ++]=sum3;
1333019b515eSShri Abhyankar       tmp[row ++]=sum4;
1334019b515eSShri Abhyankar       break;
1335019b515eSShri Abhyankar     case 5:
1336019b515eSShri Abhyankar       sum1 = b[r[row]];
1337019b515eSShri Abhyankar       sum2 = b[r[row+1]];
1338019b515eSShri Abhyankar       sum3 = b[r[row+2]];
1339019b515eSShri Abhyankar       sum4 = b[r[row+3]];
1340019b515eSShri Abhyankar       sum5 = b[r[row+4]];
1341019b515eSShri Abhyankar       v2   = aa + ai[row+1];
1342019b515eSShri Abhyankar       v3   = aa + ai[row+2];
1343019b515eSShri Abhyankar       v4   = aa + ai[row+3];
1344019b515eSShri Abhyankar       v5   = aa + ai[row+4];
1345019b515eSShri Abhyankar 
1346019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2){
1347019b515eSShri Abhyankar         i0   = vi[j];
1348019b515eSShri Abhyankar         i1   = vi[j+1];
1349019b515eSShri Abhyankar         tmp0 = tmps[i0];
1350019b515eSShri Abhyankar         tmp1 = tmps[i1];
1351019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1352019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
1353019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
1354019b515eSShri Abhyankar         sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1;
1355019b515eSShri Abhyankar         sum5 -= v5[j] * tmp0 + v5[j+1] * tmp1;
1356019b515eSShri Abhyankar       }
1357019b515eSShri Abhyankar       if (j == nz-1){
1358019b515eSShri Abhyankar         tmp0 = tmps[vi[j]];
1359019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
1360019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
1361019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
1362019b515eSShri Abhyankar         sum4 -= v4[j] *tmp0;
1363019b515eSShri Abhyankar         sum5 -= v5[j] *tmp0;
1364019b515eSShri Abhyankar       }
1365019b515eSShri Abhyankar 
1366019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
1367019b515eSShri Abhyankar       sum3 -= v3[nz] * sum1;
1368019b515eSShri Abhyankar       sum4 -= v4[nz] * sum1;
1369019b515eSShri Abhyankar       sum5 -= v5[nz] * sum1;
1370019b515eSShri Abhyankar       sum3 -= v3[nz+1] * sum2;
1371019b515eSShri Abhyankar       sum4 -= v4[nz+1] * sum2;
1372019b515eSShri Abhyankar       sum5 -= v5[nz+1] * sum2;
1373019b515eSShri Abhyankar       sum4 -= v4[nz+2] * sum3;
1374019b515eSShri Abhyankar       sum5 -= v5[nz+2] * sum3;
1375019b515eSShri Abhyankar       sum5 -= v5[nz+3] * sum4;
1376019b515eSShri Abhyankar 
1377019b515eSShri Abhyankar       tmp[row ++]=sum1;
1378019b515eSShri Abhyankar       tmp[row ++]=sum2;
1379019b515eSShri Abhyankar       tmp[row ++]=sum3;
1380019b515eSShri Abhyankar       tmp[row ++]=sum4;
1381019b515eSShri Abhyankar       tmp[row ++]=sum5;
1382019b515eSShri Abhyankar       break;
1383019b515eSShri Abhyankar     default:
1384019b515eSShri Abhyankar       SETERRQ(PETSC_ERR_COR,"Node size not yet supported \n");
1385019b515eSShri Abhyankar     }
1386019b515eSShri Abhyankar   }
1387019b515eSShri Abhyankar   /* backward solve the upper triangular */
1388019b515eSShri Abhyankar   for (i=node_max -1 ,row = n-1 ; i>=0; i--){
1389019b515eSShri Abhyankar     nsz = ns[i];
1390019b515eSShri Abhyankar     aii = ad[row+1] + 1;
1391019b515eSShri Abhyankar     v1  = aa + aii;
1392019b515eSShri Abhyankar     vi  = aj + aii;
1393019b515eSShri Abhyankar     nz  = ad[row]- ad[row+1] - 1;
1394*98991853SShri Abhyankar 
1395*98991853SShri Abhyankar     if (i > 0) {
1396*98991853SShri Abhyankar       /* Prefetch the indices for the next block */
1397*98991853SShri Abhyankar       PetscPrefetchBlock(aj+ad[row-nsz+1]+1,ad[row-nsz]-ad[row-nsz+1],0,0); /* indices */
1398*98991853SShri Abhyankar       /* Prefetch the data for the next block */
1399*98991853SShri Abhyankar       PetscPrefetchBlock(aa+ad[row-nsz+1]+1,ad[row-nsz-ns[i-1]+1]-ad[row-nsz+1],0,0);
1400*98991853SShri Abhyankar     }
1401*98991853SShri Abhyankar 
1402019b515eSShri Abhyankar     switch (nsz){               /* Each loop in 'case' is unrolled */
1403019b515eSShri Abhyankar     case 1 :
1404019b515eSShri Abhyankar       sum1 = tmp[row];
1405019b515eSShri Abhyankar 
1406019b515eSShri Abhyankar       for(j=0 ; j<nz-1; j+=2){
1407019b515eSShri Abhyankar         i0   = vi[j];
1408019b515eSShri Abhyankar         i1   = vi[j+1];
1409019b515eSShri Abhyankar         tmp0 = tmps[i0];
1410019b515eSShri Abhyankar         tmp1 = tmps[i1];
1411019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1412019b515eSShri Abhyankar       }
1413019b515eSShri Abhyankar       if (j == nz-1){
1414019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
1415019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0;
1416019b515eSShri Abhyankar       }
1417019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
1418019b515eSShri Abhyankar       break;
1419019b515eSShri Abhyankar     case 2 :
1420019b515eSShri Abhyankar       sum1 = tmp[row];
1421019b515eSShri Abhyankar       sum2 = tmp[row-1];
1422019b515eSShri Abhyankar       v2   = aa + ad[row] + 1;
1423019b515eSShri Abhyankar       for (j=0 ; j<nz-1; j+=2){
1424019b515eSShri Abhyankar         i0   = vi[j];
1425019b515eSShri Abhyankar         i1   = vi[j+1];
1426019b515eSShri Abhyankar         tmp0 = tmps[i0];
1427019b515eSShri Abhyankar         tmp1 = tmps[i1];
1428019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1429019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
1430019b515eSShri Abhyankar       }
1431019b515eSShri Abhyankar       if (j == nz-1){
1432019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
1433019b515eSShri Abhyankar         sum1 -= v1[j]* tmp0;
1434019b515eSShri Abhyankar         sum2 -= v2[j+1]* tmp0;
1435019b515eSShri Abhyankar       }
1436019b515eSShri Abhyankar 
1437019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
1438019b515eSShri Abhyankar       sum2   -= v2[0] * tmp0;
1439019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
1440019b515eSShri Abhyankar       break;
1441019b515eSShri Abhyankar     case 3 :
1442019b515eSShri Abhyankar       sum1 = tmp[row];
1443019b515eSShri Abhyankar       sum2 = tmp[row -1];
1444019b515eSShri Abhyankar       sum3 = tmp[row -2];
1445019b515eSShri Abhyankar       v2   = aa + ad[row] + 1;
1446019b515eSShri Abhyankar       v3   = aa + ad[row -1] + 1;
1447019b515eSShri Abhyankar       for (j=0 ; j<nz-1; j+=2){
1448019b515eSShri Abhyankar         i0   = vi[j];
1449019b515eSShri Abhyankar         i1   = vi[j+1];
1450019b515eSShri Abhyankar         tmp0 = tmps[i0];
1451019b515eSShri Abhyankar         tmp1 = tmps[i1];
1452019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1453019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
1454019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
1455019b515eSShri Abhyankar       }
1456019b515eSShri Abhyankar       if (j== nz-1){
1457019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
1458019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
1459019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
1460019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
1461019b515eSShri Abhyankar       }
1462019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
1463019b515eSShri Abhyankar       sum2   -= v2[0]* tmp0;
1464019b515eSShri Abhyankar       sum3   -= v3[1] * tmp0;
1465019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
1466019b515eSShri Abhyankar       sum3   -= v3[0]* tmp0;
1467019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
1468019b515eSShri Abhyankar 
1469019b515eSShri Abhyankar       break;
1470019b515eSShri Abhyankar     case 4 :
1471019b515eSShri Abhyankar       sum1 = tmp[row];
1472019b515eSShri Abhyankar       sum2 = tmp[row -1];
1473019b515eSShri Abhyankar       sum3 = tmp[row -2];
1474019b515eSShri Abhyankar       sum4 = tmp[row -3];
1475019b515eSShri Abhyankar       v2   = aa + ad[row]+1;
1476019b515eSShri Abhyankar       v3   = aa + ad[row -1]+1;
1477019b515eSShri Abhyankar       v4   = aa + ad[row -2]+1;
1478019b515eSShri Abhyankar 
1479019b515eSShri Abhyankar       for (j=0 ; j<nz-1; j+=2){
1480019b515eSShri Abhyankar         i0   = vi[j];
1481019b515eSShri Abhyankar         i1   = vi[j+1];
1482019b515eSShri Abhyankar         tmp0 = tmps[i0];
1483019b515eSShri Abhyankar         tmp1 = tmps[i1];
1484019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0   + v1[j+1] * tmp1;
1485019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
1486019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
1487019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1;
1488019b515eSShri Abhyankar       }
1489019b515eSShri Abhyankar       if (j== nz-1){
1490019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
1491019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
1492019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
1493019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
1494019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0;
1495019b515eSShri Abhyankar       }
1496019b515eSShri Abhyankar 
1497019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
1498019b515eSShri Abhyankar       sum2   -= v2[0] * tmp0;
1499019b515eSShri Abhyankar       sum3   -= v3[1] * tmp0;
1500019b515eSShri Abhyankar       sum4   -= v4[2] * tmp0;
1501019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
1502019b515eSShri Abhyankar       sum3   -= v3[0] * tmp0;
1503019b515eSShri Abhyankar       sum4   -= v4[1] * tmp0;
1504019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
1505019b515eSShri Abhyankar       sum4   -= v4[0] * tmp0;
1506019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--;
1507019b515eSShri Abhyankar       break;
1508019b515eSShri Abhyankar     case 5 :
1509019b515eSShri Abhyankar       sum1 = tmp[row];
1510019b515eSShri Abhyankar       sum2 = tmp[row -1];
1511019b515eSShri Abhyankar       sum3 = tmp[row -2];
1512019b515eSShri Abhyankar       sum4 = tmp[row -3];
1513019b515eSShri Abhyankar       sum5 = tmp[row -4];
1514019b515eSShri Abhyankar       v2   = aa + ad[row]+1;
1515019b515eSShri Abhyankar       v3   = aa + ad[row -1]+1;
1516019b515eSShri Abhyankar       v4   = aa + ad[row -2]+1;
1517019b515eSShri Abhyankar       v5   = aa + ad[row -3]+1;
1518019b515eSShri Abhyankar       for (j=0 ; j<nz-1; j+=2){
1519019b515eSShri Abhyankar         i0   = vi[j];
1520019b515eSShri Abhyankar         i1   = vi[j+1];
1521019b515eSShri Abhyankar         tmp0 = tmps[i0];
1522019b515eSShri Abhyankar         tmp1 = tmps[i1];
1523019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
1524019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
1525019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
1526019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1;
1527019b515eSShri Abhyankar         sum5 -= v5[j+4] * tmp0 + v5[j+5] * tmp1;
1528019b515eSShri Abhyankar       }
1529019b515eSShri Abhyankar       if (j==nz-1){
1530019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
1531019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
1532019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
1533019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
1534019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0;
1535019b515eSShri Abhyankar         sum5 -= v5[j+4] * tmp0;
1536019b515eSShri Abhyankar       }
1537019b515eSShri Abhyankar 
1538019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
1539019b515eSShri Abhyankar       sum2   -= v2[0] * tmp0;
1540019b515eSShri Abhyankar       sum3   -= v3[1] * tmp0;
1541019b515eSShri Abhyankar       sum4   -= v4[2] * tmp0;
1542019b515eSShri Abhyankar       sum5   -= v5[3] * tmp0;
1543019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
1544019b515eSShri Abhyankar       sum3   -= v3[0] * tmp0;
1545019b515eSShri Abhyankar       sum4   -= v4[1] * tmp0;
1546019b515eSShri Abhyankar       sum5   -= v5[2] * tmp0;
1547019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
1548019b515eSShri Abhyankar       sum4   -= v4[0] * tmp0;
1549019b515eSShri Abhyankar       sum5   -= v5[1] * tmp0;
1550019b515eSShri Abhyankar       tmp0    = x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--;
1551019b515eSShri Abhyankar       sum5   -= v5[0] * tmp0;
1552019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum5*v5[nz+4]; row--;
1553019b515eSShri Abhyankar       break;
1554019b515eSShri Abhyankar     default:
1555019b515eSShri Abhyankar       SETERRQ(PETSC_ERR_COR,"Node size not yet supported \n");
1556019b515eSShri Abhyankar     }
1557019b515eSShri Abhyankar   }
1558019b515eSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1559019b515eSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1560019b515eSShri Abhyankar   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
1561019b515eSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1562019b515eSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1563019b515eSShri Abhyankar   PetscFunctionReturn(0);
1564019b515eSShri Abhyankar }
1565019b515eSShri Abhyankar 
1566019b515eSShri Abhyankar 
15674c1414c8SBarry Smith /*
15684c1414c8SBarry Smith      Makes a longer coloring[] array and calls the usual code with that
15694c1414c8SBarry Smith */
15704c1414c8SBarry Smith #undef __FUNCT__
15714108e4d5SBarry Smith #define __FUNCT__ "MatColoringPatch_SeqAIJ_Inode"
15724108e4d5SBarry Smith PetscErrorCode MatColoringPatch_SeqAIJ_Inode(Mat mat,PetscInt ncolors,PetscInt nin,ISColoringValue coloring[],ISColoring *iscoloring)
15734c1414c8SBarry Smith {
15744c1414c8SBarry Smith   Mat_SeqAIJ       *a = (Mat_SeqAIJ*)mat->data;
15754c1414c8SBarry Smith   PetscErrorCode  ierr;
1576d0f46423SBarry Smith   PetscInt        n = mat->cmap->n,m = a->inode.node_count,j,*ns = a->inode.size,row;
15774c1414c8SBarry Smith   PetscInt        *colorused,i;
15784c1414c8SBarry Smith   ISColoringValue *newcolor;
15794c1414c8SBarry Smith 
15804c1414c8SBarry Smith   PetscFunctionBegin;
15814c1414c8SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&newcolor);CHKERRQ(ierr);
15824c1414c8SBarry Smith   /* loop over inodes, marking a color for each column*/
15834c1414c8SBarry Smith   row = 0;
15844c1414c8SBarry Smith   for (i=0; i<m; i++){
15854c1414c8SBarry Smith     for (j=0; j<ns[i]; j++) {
15864c1414c8SBarry Smith       newcolor[row++] = coloring[i] + j*ncolors;
15874c1414c8SBarry Smith     }
15884c1414c8SBarry Smith   }
15894c1414c8SBarry Smith 
15904c1414c8SBarry Smith   /* eliminate unneeded colors */
15914c1414c8SBarry Smith   ierr = PetscMalloc(5*ncolors*sizeof(PetscInt),&colorused);CHKERRQ(ierr);
15924c1414c8SBarry Smith   ierr = PetscMemzero(colorused,5*ncolors*sizeof(PetscInt));CHKERRQ(ierr);
15934c1414c8SBarry Smith   for (i=0; i<n; i++) {
15944c1414c8SBarry Smith     colorused[newcolor[i]] = 1;
15954c1414c8SBarry Smith   }
15964c1414c8SBarry Smith 
15974c1414c8SBarry Smith   for (i=1; i<5*ncolors; i++) {
15984c1414c8SBarry Smith     colorused[i] += colorused[i-1];
15994c1414c8SBarry Smith   }
16004c1414c8SBarry Smith   ncolors = colorused[5*ncolors-1];
16014c1414c8SBarry Smith   for (i=0; i<n; i++) {
160257815681SSatish Balay     newcolor[i] = colorused[newcolor[i]]-1;
16034c1414c8SBarry Smith   }
16044c1414c8SBarry Smith   ierr = PetscFree(colorused);CHKERRQ(ierr);
16057adad957SLisandro Dalcin   ierr = ISColoringCreate(((PetscObject)mat)->comm,ncolors,n,newcolor,iscoloring);CHKERRQ(ierr);
16064c1414c8SBarry Smith   ierr = PetscFree(coloring);CHKERRQ(ierr);
16074c1414c8SBarry Smith   PetscFunctionReturn(0);
16084c1414c8SBarry Smith }
16094c1414c8SBarry Smith 
1610c60f0209SBarry Smith #include "../src/mat/blockinvert.h"
16112af78befSBarry Smith 
16122af78befSBarry Smith #undef __FUNCT__
16134108e4d5SBarry Smith #define __FUNCT__ "MatSOR_SeqAIJ_Inode"
16144108e4d5SBarry Smith PetscErrorCode MatSOR_SeqAIJ_Inode(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx)
16152af78befSBarry Smith {
16162af78befSBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
1617938d4eb3SBarry Smith   PetscScalar        sum1,sum2,sum3,sum4,sum5,tmp0,tmp1,tmp2,tmp3;
1618a77337e4SBarry Smith   MatScalar          *ibdiag,*bdiag;
1619938d4eb3SBarry Smith   PetscScalar        *x,*xb,tmp4,tmp5,x1,x2,x3,x4,x5;
1620dd6ea824SBarry Smith   const MatScalar    *v = a->a,*v1,*v2,*v3,*v4,*v5;
162162bba022SBarry Smith   PetscReal          zeropivot = 1.0e-15, shift = 0.0;
16222af78befSBarry Smith   PetscErrorCode     ierr;
1623f0d39aaaSBarry Smith   PetscInt           n,m = a->inode.node_count,*sizes = a->inode.size,cnt = 0,i,j,row,i1,i2;
1624f0d39aaaSBarry Smith   PetscInt           *idx,*diag = a->diag,*ii = a->i,sz,k;
16252af78befSBarry Smith 
16262af78befSBarry Smith   PetscFunctionBegin;
162771f1c65dSBarry Smith   if (omega != 1.0) SETERRQ(PETSC_ERR_SUP,"No support for omega != 1.0; use -mat_no_inode");
162871f1c65dSBarry Smith   if (fshift != 0.0) SETERRQ(PETSC_ERR_SUP,"No support for fshift != 0.0; use -mat_no_inode");
16293f6d85c9SBarry Smith   if (its > 1) {
16303f6d85c9SBarry Smith     /* switch to non-inode version */
163141f059aeSBarry Smith     ierr = MatSOR_SeqAIJ(A,bb,omega,flag,fshift,its,lits,xx);CHKERRQ(ierr);
16323f6d85c9SBarry Smith     PetscFunctionReturn(0);
16333f6d85c9SBarry Smith   }
16342af78befSBarry Smith 
163571f1c65dSBarry Smith   if (!a->inode.ibdiagvalid) {
16362af78befSBarry Smith     if (!a->inode.ibdiag) {
16372af78befSBarry Smith       /* calculate space needed for diagonal blocks */
16382af78befSBarry Smith       for (i=0; i<m; i++) {
16392af78befSBarry Smith 	cnt += sizes[i]*sizes[i];
16402af78befSBarry Smith       }
1641f0d39aaaSBarry Smith       a->inode.bdiagsize = cnt;
164289c6957cSBarry Smith       ierr   = PetscMalloc3(cnt,MatScalar,&a->inode.ibdiag,cnt,MatScalar,&a->inode.bdiag,A->rmap->n,MatScalar,&a->inode.ssor_work);CHKERRQ(ierr);
164371f1c65dSBarry Smith     }
164471f1c65dSBarry Smith 
164571f1c65dSBarry Smith     /* copy over the diagonal blocks and invert them */
16462af78befSBarry Smith     ibdiag = a->inode.ibdiag;
16472af78befSBarry Smith     bdiag  = a->inode.bdiag;
16482af78befSBarry Smith     cnt = 0;
16492af78befSBarry Smith     for (i=0, row = 0; i<m; i++) {
16502af78befSBarry Smith       for (j=0; j<sizes[i]; j++) {
1651f0d39aaaSBarry Smith         for (k=0; k<sizes[i]; k++) {
1652f0d39aaaSBarry Smith           bdiag[cnt+k*sizes[i]+j] = v[diag[row+j] - j + k];
1653f0d39aaaSBarry Smith         }
16542af78befSBarry Smith       }
1655a77337e4SBarry Smith       ierr = PetscMemcpy(ibdiag+cnt,bdiag+cnt,sizes[i]*sizes[i]*sizeof(MatScalar));CHKERRQ(ierr);
16562af78befSBarry Smith 
16572af78befSBarry Smith       switch(sizes[i]) {
16582af78befSBarry Smith         case 1:
16592af78befSBarry Smith           /* Create matrix data structure */
166064c62002SMatthew Knepley           if (PetscAbsScalar(ibdiag[cnt]) < zeropivot) SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot on row %D",row);
166164c62002SMatthew Knepley           ibdiag[cnt] = 1.0/ibdiag[cnt];
16622af78befSBarry Smith           break;
16632af78befSBarry Smith         case 2:
166462bba022SBarry Smith           ierr = Kernel_A_gets_inverse_A_2(ibdiag+cnt,shift);CHKERRQ(ierr);
16652af78befSBarry Smith           break;
16662af78befSBarry Smith         case 3:
166762bba022SBarry Smith           ierr = Kernel_A_gets_inverse_A_3(ibdiag+cnt,shift);CHKERRQ(ierr);
16682af78befSBarry Smith           break;
16692af78befSBarry Smith         case 4:
167062bba022SBarry Smith           ierr = Kernel_A_gets_inverse_A_4(ibdiag+cnt,shift);CHKERRQ(ierr);
16712af78befSBarry Smith           break;
16722af78befSBarry Smith         case 5:
167362bba022SBarry Smith           ierr = Kernel_A_gets_inverse_A_5(ibdiag+cnt,shift);CHKERRQ(ierr);
16742af78befSBarry Smith           break;
16752af78befSBarry Smith        default:
16762af78befSBarry Smith 	 SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
16772af78befSBarry Smith       }
16782af78befSBarry Smith       cnt += sizes[i]*sizes[i];
16792af78befSBarry Smith       row += sizes[i];
16802af78befSBarry Smith     }
168171f1c65dSBarry Smith     a->inode.ibdiagvalid = PETSC_TRUE;
16822af78befSBarry Smith   }
16832af78befSBarry Smith   ibdiag = a->inode.ibdiag;
16842af78befSBarry Smith   bdiag  = a->inode.bdiag;
16852af78befSBarry Smith 
1686938d4eb3SBarry Smith 
1687938d4eb3SBarry Smith   /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */
1688938d4eb3SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
1689938d4eb3SBarry Smith     PetscScalar *b;
16902af78befSBarry Smith     ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
16912af78befSBarry Smith     if (xx != bb) {
16922af78befSBarry Smith       ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
16932af78befSBarry Smith     } else {
16942af78befSBarry Smith       b = x;
16952af78befSBarry Smith     }
16962af78befSBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
16972af78befSBarry Smith 
16988862d2efSBarry Smith       for (i=0, row=0; i<m; i++) {
16998862d2efSBarry Smith         sz  = diag[row] - ii[row];
17008862d2efSBarry Smith         v1  = a->a + ii[row];
17018862d2efSBarry Smith         idx = a->j + ii[row];
17028862d2efSBarry Smith 
17034108e4d5SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
17048862d2efSBarry Smith         switch (sizes[i]){
17058862d2efSBarry Smith           case 1:
17068862d2efSBarry Smith 
17078862d2efSBarry Smith             sum1  = b[row];
17088862d2efSBarry Smith             for(n = 0; n<sz-1; n+=2) {
17098862d2efSBarry Smith               i1   = idx[0];
17108862d2efSBarry Smith               i2   = idx[1];
17118862d2efSBarry Smith               idx += 2;
17128862d2efSBarry Smith               tmp0 = x[i1];
17138862d2efSBarry Smith               tmp1 = x[i2];
17148862d2efSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
17158862d2efSBarry Smith             }
17168862d2efSBarry Smith 
17178862d2efSBarry Smith             if (n == sz-1){
1718f0d39aaaSBarry Smith               tmp0  = x[*idx];
1719f0d39aaaSBarry Smith               sum1 -= *v1 * tmp0;
17208862d2efSBarry Smith             }
17218862d2efSBarry Smith             x[row++] = sum1*(*ibdiag++);
17228862d2efSBarry Smith             break;
1723f0d39aaaSBarry Smith           case 2:
1724f0d39aaaSBarry Smith             v2    = a->a + ii[row+1];
1725f0d39aaaSBarry Smith             sum1  = b[row];
1726f0d39aaaSBarry Smith             sum2  = b[row+1];
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             }
1736f0d39aaaSBarry Smith 
1737f0d39aaaSBarry Smith             if (n == sz-1){
1738f0d39aaaSBarry Smith               tmp0  = x[*idx];
1739f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0;
1740f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0;
1741f0d39aaaSBarry Smith             }
1742f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[0] + sum2*ibdiag[2];
1743f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[1] + sum2*ibdiag[3];
1744f0d39aaaSBarry Smith             ibdiag  += 4;
1745f0d39aaaSBarry Smith             break;
1746f0d39aaaSBarry Smith           case 3:
1747f0d39aaaSBarry Smith             v2    = a->a + ii[row+1];
1748f0d39aaaSBarry Smith             v3    = a->a + ii[row+2];
1749f0d39aaaSBarry Smith             sum1  = b[row];
1750f0d39aaaSBarry Smith             sum2  = b[row+1];
1751f0d39aaaSBarry Smith             sum3  = b[row+2];
1752f0d39aaaSBarry Smith             for(n = 0; n<sz-1; n+=2) {
1753f0d39aaaSBarry Smith               i1   = idx[0];
1754f0d39aaaSBarry Smith               i2   = idx[1];
1755f0d39aaaSBarry Smith               idx += 2;
1756f0d39aaaSBarry Smith               tmp0 = x[i1];
1757f0d39aaaSBarry Smith               tmp1 = x[i2];
1758f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
1759f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
1760f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
1761f0d39aaaSBarry Smith             }
1762f0d39aaaSBarry Smith 
1763f0d39aaaSBarry Smith             if (n == sz-1){
1764f0d39aaaSBarry Smith               tmp0  = x[*idx];
1765f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0;
1766f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0;
1767f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0;
1768f0d39aaaSBarry Smith             }
1769f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
1770f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
1771f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
1772f0d39aaaSBarry Smith             ibdiag  += 9;
1773f0d39aaaSBarry Smith             break;
1774f0d39aaaSBarry Smith           case 4:
1775f0d39aaaSBarry Smith             v2    = a->a + ii[row+1];
1776f0d39aaaSBarry Smith             v3    = a->a + ii[row+2];
1777f0d39aaaSBarry Smith             v4    = a->a + ii[row+3];
1778f0d39aaaSBarry Smith             sum1  = b[row];
1779f0d39aaaSBarry Smith             sum2  = b[row+1];
1780f0d39aaaSBarry Smith             sum3  = b[row+2];
1781f0d39aaaSBarry Smith             sum4  = b[row+3];
1782f0d39aaaSBarry Smith             for(n = 0; n<sz-1; n+=2) {
1783f0d39aaaSBarry Smith               i1   = idx[0];
1784f0d39aaaSBarry Smith               i2   = idx[1];
1785f0d39aaaSBarry Smith               idx += 2;
1786f0d39aaaSBarry Smith               tmp0 = x[i1];
1787f0d39aaaSBarry Smith               tmp1 = x[i2];
1788f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
1789f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
1790f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
1791f0d39aaaSBarry Smith               sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
1792f0d39aaaSBarry Smith             }
1793f0d39aaaSBarry Smith 
1794f0d39aaaSBarry Smith             if (n == sz-1){
1795f0d39aaaSBarry Smith               tmp0  = x[*idx];
1796f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0;
1797f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0;
1798f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0;
1799f0d39aaaSBarry Smith               sum4 -= v4[0] * tmp0;
1800f0d39aaaSBarry Smith             }
1801f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
1802f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
1803f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
1804f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
1805f0d39aaaSBarry Smith             ibdiag  += 16;
1806f0d39aaaSBarry Smith             break;
1807f0d39aaaSBarry Smith           case 5:
1808f0d39aaaSBarry Smith             v2    = a->a + ii[row+1];
1809f0d39aaaSBarry Smith             v3    = a->a + ii[row+2];
1810f0d39aaaSBarry Smith             v4    = a->a + ii[row+3];
1811f0d39aaaSBarry Smith             v5    = a->a + ii[row+4];
1812f0d39aaaSBarry Smith             sum1  = b[row];
1813f0d39aaaSBarry Smith             sum2  = b[row+1];
1814f0d39aaaSBarry Smith             sum3  = b[row+2];
1815f0d39aaaSBarry Smith             sum4  = b[row+3];
1816f0d39aaaSBarry Smith             sum5  = b[row+4];
1817f0d39aaaSBarry Smith             for(n = 0; n<sz-1; n+=2) {
1818f0d39aaaSBarry Smith               i1   = idx[0];
1819f0d39aaaSBarry Smith               i2   = idx[1];
1820f0d39aaaSBarry Smith               idx += 2;
1821f0d39aaaSBarry Smith               tmp0 = x[i1];
1822f0d39aaaSBarry Smith               tmp1 = x[i2];
1823f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
1824f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
1825f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
1826f0d39aaaSBarry Smith               sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
1827f0d39aaaSBarry Smith               sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
1828f0d39aaaSBarry Smith             }
1829f0d39aaaSBarry Smith 
1830f0d39aaaSBarry Smith             if (n == sz-1){
1831f0d39aaaSBarry Smith               tmp0  = x[*idx];
1832f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0;
1833f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0;
1834f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0;
1835f0d39aaaSBarry Smith               sum4 -= v4[0] * tmp0;
1836f0d39aaaSBarry Smith               sum5 -= v5[0] * tmp0;
1837f0d39aaaSBarry Smith             }
1838f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
1839f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
1840f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
1841f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
1842f0d39aaaSBarry Smith             x[row++] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
1843f0d39aaaSBarry Smith             ibdiag  += 25;
1844f0d39aaaSBarry Smith             break;
18458862d2efSBarry Smith 	  default:
18468862d2efSBarry Smith    	    SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
18478862d2efSBarry Smith         }
18482af78befSBarry Smith       }
18492af78befSBarry Smith 
18502af78befSBarry Smith       xb = x;
18512af78befSBarry Smith       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
18522af78befSBarry Smith     } else xb = b;
18532af78befSBarry Smith     if ((flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) &&
18542af78befSBarry Smith         (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP)) {
18558862d2efSBarry Smith       cnt = 0;
18568862d2efSBarry Smith       for (i=0, row=0; i<m; i++) {
18572af78befSBarry Smith 
18588862d2efSBarry Smith         switch (sizes[i]){
18598862d2efSBarry Smith           case 1:
18608862d2efSBarry Smith             x[row++] *= bdiag[cnt++];
18618862d2efSBarry Smith             break;
1862f0d39aaaSBarry Smith           case 2:
1863f0d39aaaSBarry Smith             x1   = x[row]; x2 = x[row+1];
1864f0d39aaaSBarry Smith             tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2];
1865f0d39aaaSBarry Smith             tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3];
1866f0d39aaaSBarry Smith             x[row++] = tmp1;
1867f0d39aaaSBarry Smith             x[row++] = tmp2;
1868f0d39aaaSBarry Smith             cnt += 4;
1869f0d39aaaSBarry Smith             break;
1870f0d39aaaSBarry Smith           case 3:
1871f0d39aaaSBarry Smith             x1   = x[row]; x2 = x[row+1]; x3 = x[row+2];
1872f0d39aaaSBarry Smith             tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6];
1873f0d39aaaSBarry Smith             tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7];
1874f0d39aaaSBarry Smith             tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8];
1875f0d39aaaSBarry Smith             x[row++] = tmp1;
1876f0d39aaaSBarry Smith             x[row++] = tmp2;
1877f0d39aaaSBarry Smith             x[row++] = tmp3;
1878f0d39aaaSBarry Smith             cnt += 9;
1879f0d39aaaSBarry Smith             break;
1880f0d39aaaSBarry Smith           case 4:
1881f0d39aaaSBarry Smith             x1   = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3];
1882f0d39aaaSBarry Smith             tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12];
1883f0d39aaaSBarry Smith             tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13];
1884f0d39aaaSBarry Smith             tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14];
1885f0d39aaaSBarry Smith             tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15];
1886f0d39aaaSBarry Smith             x[row++] = tmp1;
1887f0d39aaaSBarry Smith             x[row++] = tmp2;
1888f0d39aaaSBarry Smith             x[row++] = tmp3;
1889f0d39aaaSBarry Smith             x[row++] = tmp4;
1890f0d39aaaSBarry Smith             cnt += 16;
1891f0d39aaaSBarry Smith             break;
1892f0d39aaaSBarry Smith           case 5:
1893f0d39aaaSBarry Smith             x1   = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; x5 = x[row+4];
1894f0d39aaaSBarry Smith             tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20];
1895f0d39aaaSBarry Smith             tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21];
1896f0d39aaaSBarry Smith             tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22];
1897f0d39aaaSBarry Smith             tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23];
1898f0d39aaaSBarry Smith             tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24];
1899f0d39aaaSBarry Smith             x[row++] = tmp1;
1900f0d39aaaSBarry Smith             x[row++] = tmp2;
1901f0d39aaaSBarry Smith             x[row++] = tmp3;
1902f0d39aaaSBarry Smith             x[row++] = tmp4;
1903f0d39aaaSBarry Smith             x[row++] = tmp5;
1904f0d39aaaSBarry Smith             cnt += 25;
1905f0d39aaaSBarry Smith             break;
19068862d2efSBarry Smith 	  default:
19078862d2efSBarry Smith    	    SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
19088862d2efSBarry Smith         }
19092af78befSBarry Smith       }
19102af78befSBarry Smith       ierr = PetscLogFlops(m);CHKERRQ(ierr);
19112af78befSBarry Smith     }
19122af78befSBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
19132af78befSBarry Smith 
1914f0d39aaaSBarry Smith       ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
1915d0f46423SBarry Smith       for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
1916f0d39aaaSBarry Smith         ibdiag -= sizes[i]*sizes[i];
19178862d2efSBarry Smith         sz      = ii[row+1] - diag[row] - 1;
19188862d2efSBarry Smith         v1      = a->a + diag[row] + 1;
19198862d2efSBarry Smith         idx     = a->j + diag[row] + 1;
19202af78befSBarry Smith 
19214108e4d5SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
19228862d2efSBarry Smith         switch (sizes[i]){
19238862d2efSBarry Smith           case 1:
19248862d2efSBarry Smith 
19258862d2efSBarry Smith             sum1  = xb[row];
19268862d2efSBarry Smith             for(n = 0; n<sz-1; n+=2) {
19278862d2efSBarry Smith               i1   = idx[0];
19288862d2efSBarry Smith               i2   = idx[1];
19298862d2efSBarry Smith               idx += 2;
19308862d2efSBarry Smith               tmp0 = x[i1];
19318862d2efSBarry Smith               tmp1 = x[i2];
19328862d2efSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
19338862d2efSBarry Smith             }
19348862d2efSBarry Smith 
19358862d2efSBarry Smith             if (n == sz-1){
1936f0d39aaaSBarry Smith               tmp0  = x[*idx];
1937f0d39aaaSBarry Smith               sum1 -= *v1*tmp0;
19388862d2efSBarry Smith             }
1939f0d39aaaSBarry Smith             x[row--] = sum1*(*ibdiag);
1940f0d39aaaSBarry Smith             break;
1941f0d39aaaSBarry Smith 
1942f0d39aaaSBarry Smith           case 2:
1943f0d39aaaSBarry Smith 
1944f0d39aaaSBarry Smith             sum1  = xb[row];
1945f0d39aaaSBarry Smith             sum2  = xb[row-1];
1946f0d39aaaSBarry Smith             /* note that sum1 is associated with the second of the two rows */
1947f0d39aaaSBarry Smith             v2    = a->a + diag[row-1] + 2;
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             }
1957f0d39aaaSBarry Smith 
1958f0d39aaaSBarry Smith             if (n == sz-1){
1959f0d39aaaSBarry Smith               tmp0  = x[*idx];
1960f0d39aaaSBarry Smith               sum1 -= *v1*tmp0;
1961f0d39aaaSBarry Smith               sum2 -= *v2*tmp0;
1962f0d39aaaSBarry Smith             }
1963f0d39aaaSBarry Smith             x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3];
1964f0d39aaaSBarry Smith             x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2];
1965f0d39aaaSBarry Smith             break;
1966f0d39aaaSBarry Smith           case 3:
1967f0d39aaaSBarry Smith 
1968f0d39aaaSBarry Smith             sum1  = xb[row];
1969f0d39aaaSBarry Smith             sum2  = xb[row-1];
1970f0d39aaaSBarry Smith             sum3  = xb[row-2];
1971f0d39aaaSBarry Smith             v2    = a->a + diag[row-1] + 2;
1972f0d39aaaSBarry Smith             v3    = a->a + diag[row-2] + 3;
1973f0d39aaaSBarry Smith             for(n = 0; n<sz-1; n+=2) {
1974f0d39aaaSBarry Smith               i1   = idx[0];
1975f0d39aaaSBarry Smith               i2   = idx[1];
1976f0d39aaaSBarry Smith               idx += 2;
1977f0d39aaaSBarry Smith               tmp0 = x[i1];
1978f0d39aaaSBarry Smith               tmp1 = x[i2];
1979f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
1980f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
1981f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
1982f0d39aaaSBarry Smith             }
1983f0d39aaaSBarry Smith 
1984f0d39aaaSBarry Smith             if (n == sz-1){
1985f0d39aaaSBarry Smith               tmp0  = x[*idx];
1986f0d39aaaSBarry Smith               sum1 -= *v1*tmp0;
1987f0d39aaaSBarry Smith               sum2 -= *v2*tmp0;
1988f0d39aaaSBarry Smith               sum3 -= *v3*tmp0;
1989f0d39aaaSBarry Smith             }
1990f0d39aaaSBarry Smith             x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
1991f0d39aaaSBarry Smith             x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
1992f0d39aaaSBarry Smith             x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
1993f0d39aaaSBarry Smith             break;
1994f0d39aaaSBarry Smith           case 4:
1995f0d39aaaSBarry Smith 
1996f0d39aaaSBarry Smith             sum1  = xb[row];
1997f0d39aaaSBarry Smith             sum2  = xb[row-1];
1998f0d39aaaSBarry Smith             sum3  = xb[row-2];
1999f0d39aaaSBarry Smith             sum4  = xb[row-3];
2000f0d39aaaSBarry Smith             v2    = a->a + diag[row-1] + 2;
2001f0d39aaaSBarry Smith             v3    = a->a + diag[row-2] + 3;
2002f0d39aaaSBarry Smith             v4    = a->a + diag[row-3] + 4;
2003f0d39aaaSBarry Smith             for(n = 0; n<sz-1; n+=2) {
2004f0d39aaaSBarry Smith               i1   = idx[0];
2005f0d39aaaSBarry Smith               i2   = idx[1];
2006f0d39aaaSBarry Smith               idx += 2;
2007f0d39aaaSBarry Smith               tmp0 = x[i1];
2008f0d39aaaSBarry Smith               tmp1 = x[i2];
2009f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2010f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2011f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2012f0d39aaaSBarry Smith               sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
2013f0d39aaaSBarry Smith             }
2014f0d39aaaSBarry Smith 
2015f0d39aaaSBarry Smith             if (n == sz-1){
2016f0d39aaaSBarry Smith               tmp0  = x[*idx];
2017f0d39aaaSBarry Smith               sum1 -= *v1*tmp0;
2018f0d39aaaSBarry Smith               sum2 -= *v2*tmp0;
2019f0d39aaaSBarry Smith               sum3 -= *v3*tmp0;
2020f0d39aaaSBarry Smith               sum4 -= *v4*tmp0;
2021f0d39aaaSBarry Smith             }
2022f0d39aaaSBarry Smith             x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
2023f0d39aaaSBarry Smith             x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
2024f0d39aaaSBarry Smith             x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
2025f0d39aaaSBarry Smith             x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
2026f0d39aaaSBarry Smith             break;
2027f0d39aaaSBarry Smith           case 5:
2028f0d39aaaSBarry Smith 
2029f0d39aaaSBarry Smith             sum1  = xb[row];
2030f0d39aaaSBarry Smith             sum2  = xb[row-1];
2031f0d39aaaSBarry Smith             sum3  = xb[row-2];
2032f0d39aaaSBarry Smith             sum4  = xb[row-3];
2033f0d39aaaSBarry Smith             sum5  = xb[row-4];
2034f0d39aaaSBarry Smith             v2    = a->a + diag[row-1] + 2;
2035f0d39aaaSBarry Smith             v3    = a->a + diag[row-2] + 3;
2036f0d39aaaSBarry Smith             v4    = a->a + diag[row-3] + 4;
2037f0d39aaaSBarry Smith             v5    = a->a + diag[row-4] + 5;
2038f0d39aaaSBarry Smith             for(n = 0; n<sz-1; n+=2) {
2039f0d39aaaSBarry Smith               i1   = idx[0];
2040f0d39aaaSBarry Smith               i2   = idx[1];
2041f0d39aaaSBarry Smith               idx += 2;
2042f0d39aaaSBarry Smith               tmp0 = x[i1];
2043f0d39aaaSBarry Smith               tmp1 = x[i2];
2044f0d39aaaSBarry Smith               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2045f0d39aaaSBarry Smith               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2046f0d39aaaSBarry Smith               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2047f0d39aaaSBarry Smith               sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
2048f0d39aaaSBarry Smith               sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
2049f0d39aaaSBarry Smith             }
2050f0d39aaaSBarry Smith 
2051f0d39aaaSBarry Smith             if (n == sz-1){
2052f0d39aaaSBarry Smith               tmp0  = x[*idx];
2053f0d39aaaSBarry Smith               sum1 -= *v1*tmp0;
2054f0d39aaaSBarry Smith               sum2 -= *v2*tmp0;
2055f0d39aaaSBarry Smith               sum3 -= *v3*tmp0;
2056f0d39aaaSBarry Smith               sum4 -= *v4*tmp0;
2057f0d39aaaSBarry Smith               sum5 -= *v5*tmp0;
2058f0d39aaaSBarry Smith             }
2059f0d39aaaSBarry Smith             x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
2060f0d39aaaSBarry Smith             x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
2061f0d39aaaSBarry Smith             x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
2062f0d39aaaSBarry Smith             x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
2063f0d39aaaSBarry Smith             x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
20648862d2efSBarry Smith             break;
20658862d2efSBarry Smith 	  default:
20668862d2efSBarry Smith    	    SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
20678862d2efSBarry Smith         }
20682af78befSBarry Smith       }
20692af78befSBarry Smith 
20702af78befSBarry Smith       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
20712af78befSBarry Smith     }
20722af78befSBarry Smith     its--;
2073938d4eb3SBarry Smith     ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
2074938d4eb3SBarry Smith     if (bb != xx) {ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);}
20752af78befSBarry Smith   }
207689c6957cSBarry Smith   if (flag & SOR_EISENSTAT) {
2077938d4eb3SBarry Smith     const PetscScalar *b;
20786cf45366SBarry Smith     MatScalar         *t = a->inode.ssor_work;
20796cf45366SBarry Smith 
2080938d4eb3SBarry Smith     ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2081938d4eb3SBarry Smith     ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
208289c6957cSBarry Smith     /*
208389c6957cSBarry Smith           Apply  (U + D)^-1  where D is now the block diagonal
208489c6957cSBarry Smith     */
208589c6957cSBarry Smith     ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
208689c6957cSBarry Smith     for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
208789c6957cSBarry Smith       ibdiag -= sizes[i]*sizes[i];
208889c6957cSBarry Smith       sz      = ii[row+1] - diag[row] - 1;
208989c6957cSBarry Smith       v1      = a->a + diag[row] + 1;
209089c6957cSBarry Smith       idx     = a->j + diag[row] + 1;
209189c6957cSBarry Smith       CHKMEMQ;
20924108e4d5SBarry Smith       /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
209389c6957cSBarry Smith       switch (sizes[i]){
209489c6957cSBarry Smith         case 1:
209589c6957cSBarry Smith 
209689c6957cSBarry Smith 	  sum1  = b[row];
209789c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
209889c6957cSBarry Smith 	    i1   = idx[0];
209989c6957cSBarry Smith 	    i2   = idx[1];
210089c6957cSBarry Smith 	    idx += 2;
210189c6957cSBarry Smith 	    tmp0 = x[i1];
210289c6957cSBarry Smith 	    tmp1 = x[i2];
210389c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
210489c6957cSBarry Smith 	  }
210589c6957cSBarry Smith 
210689c6957cSBarry Smith 	  if (n == sz-1){
210789c6957cSBarry Smith 	    tmp0  = x[*idx];
210889c6957cSBarry Smith 	    sum1 -= *v1*tmp0;
210989c6957cSBarry Smith 	  }
2110938d4eb3SBarry Smith 	  x[row] = sum1*(*ibdiag);row--;
211189c6957cSBarry Smith 	  break;
211289c6957cSBarry Smith 
211389c6957cSBarry Smith         case 2:
211489c6957cSBarry Smith 
211589c6957cSBarry Smith 	  sum1  = b[row];
211689c6957cSBarry Smith 	  sum2  = b[row-1];
211789c6957cSBarry Smith 	  /* note that sum1 is associated with the second of the two rows */
211889c6957cSBarry Smith 	  v2    = a->a + diag[row-1] + 2;
211989c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
212089c6957cSBarry Smith 	    i1   = idx[0];
212189c6957cSBarry Smith 	    i2   = idx[1];
212289c6957cSBarry Smith 	    idx += 2;
212389c6957cSBarry Smith 	    tmp0 = x[i1];
212489c6957cSBarry Smith 	    tmp1 = x[i2];
212589c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
212689c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
212789c6957cSBarry Smith 	  }
212889c6957cSBarry Smith 
212989c6957cSBarry Smith 	  if (n == sz-1){
213089c6957cSBarry Smith 	    tmp0  = x[*idx];
213189c6957cSBarry Smith 	    sum1 -= *v1*tmp0;
213289c6957cSBarry Smith 	    sum2 -= *v2*tmp0;
213389c6957cSBarry Smith 	  }
2134938d4eb3SBarry Smith 	  x[row] = sum2*ibdiag[1] + sum1*ibdiag[3];
2135938d4eb3SBarry Smith 	  x[row-1] = sum2*ibdiag[0] + sum1*ibdiag[2];
2136938d4eb3SBarry Smith           row -= 2;
213789c6957cSBarry Smith 	  break;
213889c6957cSBarry Smith         case 3:
213989c6957cSBarry Smith 
214089c6957cSBarry Smith 	  sum1  = b[row];
214189c6957cSBarry Smith 	  sum2  = b[row-1];
214289c6957cSBarry Smith 	  sum3  = b[row-2];
214389c6957cSBarry Smith 	  v2    = a->a + diag[row-1] + 2;
214489c6957cSBarry Smith 	  v3    = a->a + diag[row-2] + 3;
214589c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
214689c6957cSBarry Smith 	    i1   = idx[0];
214789c6957cSBarry Smith 	    i2   = idx[1];
214889c6957cSBarry Smith 	    idx += 2;
214989c6957cSBarry Smith 	    tmp0 = x[i1];
215089c6957cSBarry Smith 	    tmp1 = x[i2];
215189c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
215289c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
215389c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
215489c6957cSBarry Smith 	  }
215589c6957cSBarry Smith 
215689c6957cSBarry Smith 	  if (n == sz-1){
215789c6957cSBarry Smith 	    tmp0  = x[*idx];
215889c6957cSBarry Smith 	    sum1 -= *v1*tmp0;
215989c6957cSBarry Smith 	    sum2 -= *v2*tmp0;
216089c6957cSBarry Smith 	    sum3 -= *v3*tmp0;
216189c6957cSBarry Smith 	  }
2162938d4eb3SBarry Smith 	  x[row] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
2163938d4eb3SBarry Smith 	  x[row-1] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
2164938d4eb3SBarry Smith 	  x[row-2] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
2165938d4eb3SBarry Smith           row -= 3;
216689c6957cSBarry Smith 	  break;
216789c6957cSBarry Smith         case 4:
216889c6957cSBarry Smith 
216989c6957cSBarry Smith 	  sum1  = b[row];
217089c6957cSBarry Smith 	  sum2  = b[row-1];
217189c6957cSBarry Smith 	  sum3  = b[row-2];
217289c6957cSBarry Smith 	  sum4  = b[row-3];
217389c6957cSBarry Smith 	  v2    = a->a + diag[row-1] + 2;
217489c6957cSBarry Smith 	  v3    = a->a + diag[row-2] + 3;
217589c6957cSBarry Smith 	  v4    = a->a + diag[row-3] + 4;
217689c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
217789c6957cSBarry Smith 	    i1   = idx[0];
217889c6957cSBarry Smith 	    i2   = idx[1];
217989c6957cSBarry Smith 	    idx += 2;
218089c6957cSBarry Smith 	    tmp0 = x[i1];
218189c6957cSBarry Smith 	    tmp1 = x[i2];
218289c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
218389c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
218489c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
218589c6957cSBarry Smith 	    sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
218689c6957cSBarry Smith 	  }
218789c6957cSBarry Smith 
218889c6957cSBarry Smith 	  if (n == sz-1){
218989c6957cSBarry Smith 	    tmp0  = x[*idx];
219089c6957cSBarry Smith 	    sum1 -= *v1*tmp0;
219189c6957cSBarry Smith 	    sum2 -= *v2*tmp0;
219289c6957cSBarry Smith 	    sum3 -= *v3*tmp0;
219389c6957cSBarry Smith 	    sum4 -= *v4*tmp0;
219489c6957cSBarry Smith 	  }
2195938d4eb3SBarry Smith 	  x[row] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
2196938d4eb3SBarry Smith 	  x[row-1] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
2197938d4eb3SBarry Smith 	  x[row-2] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
2198938d4eb3SBarry Smith 	  x[row-3] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
2199938d4eb3SBarry Smith           row -= 4;
220089c6957cSBarry Smith 	  break;
220189c6957cSBarry Smith         case 5:
220289c6957cSBarry Smith 
220389c6957cSBarry Smith 	  sum1  = b[row];
220489c6957cSBarry Smith 	  sum2  = b[row-1];
220589c6957cSBarry Smith 	  sum3  = b[row-2];
220689c6957cSBarry Smith 	  sum4  = b[row-3];
220789c6957cSBarry Smith 	  sum5  = b[row-4];
220889c6957cSBarry Smith 	  v2    = a->a + diag[row-1] + 2;
220989c6957cSBarry Smith 	  v3    = a->a + diag[row-2] + 3;
221089c6957cSBarry Smith 	  v4    = a->a + diag[row-3] + 4;
221189c6957cSBarry Smith 	  v5    = a->a + diag[row-4] + 5;
221289c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
221389c6957cSBarry Smith 	    i1   = idx[0];
221489c6957cSBarry Smith 	    i2   = idx[1];
221589c6957cSBarry Smith 	    idx += 2;
221689c6957cSBarry Smith 	    tmp0 = x[i1];
221789c6957cSBarry Smith 	    tmp1 = x[i2];
221889c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
221989c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
222089c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
222189c6957cSBarry Smith 	    sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
222289c6957cSBarry Smith 	    sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
222389c6957cSBarry Smith 	  }
222489c6957cSBarry Smith 
222589c6957cSBarry Smith 	  if (n == sz-1){
222689c6957cSBarry Smith 	    tmp0  = x[*idx];
222789c6957cSBarry Smith 	    sum1 -= *v1*tmp0;
222889c6957cSBarry Smith 	    sum2 -= *v2*tmp0;
222989c6957cSBarry Smith 	    sum3 -= *v3*tmp0;
223089c6957cSBarry Smith 	    sum4 -= *v4*tmp0;
223189c6957cSBarry Smith 	    sum5 -= *v5*tmp0;
223289c6957cSBarry Smith 	  }
2233938d4eb3SBarry Smith 	  x[row] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
2234938d4eb3SBarry Smith 	  x[row-1] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
2235938d4eb3SBarry Smith 	  x[row-2] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
2236938d4eb3SBarry Smith 	  x[row-3] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
2237938d4eb3SBarry Smith 	  x[row-4] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
2238938d4eb3SBarry Smith           row -= 5;
223989c6957cSBarry Smith 	  break;
224089c6957cSBarry Smith         default:
224189c6957cSBarry Smith 	  SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
224289c6957cSBarry Smith       }
224389c6957cSBarry Smith       CHKMEMQ;
224489c6957cSBarry Smith     }
224589c6957cSBarry Smith     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
224689c6957cSBarry Smith 
224789c6957cSBarry Smith     /*
224889c6957cSBarry Smith            t = b - D x    where D is the block diagonal
224989c6957cSBarry Smith     */
225089c6957cSBarry Smith     cnt = 0;
225189c6957cSBarry Smith     for (i=0, row=0; i<m; i++) {
225289c6957cSBarry Smith       CHKMEMQ;
225389c6957cSBarry Smith       switch (sizes[i]){
225489c6957cSBarry Smith         case 1:
225589c6957cSBarry Smith 	  t[row] = b[row] - bdiag[cnt++]*x[row]; row++;
225689c6957cSBarry Smith 	  break;
225789c6957cSBarry Smith         case 2:
225889c6957cSBarry Smith 	  x1   = x[row]; x2 = x[row+1];
225989c6957cSBarry Smith 	  tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2];
226089c6957cSBarry Smith 	  tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3];
226189c6957cSBarry Smith 	  t[row]   = b[row] - tmp1;
226289c6957cSBarry Smith 	  t[row+1] = b[row+1] - tmp2; row += 2;
226389c6957cSBarry Smith 	  cnt += 4;
226489c6957cSBarry Smith 	  break;
226589c6957cSBarry Smith         case 3:
226689c6957cSBarry Smith 	  x1   = x[row]; x2 = x[row+1]; x3 = x[row+2];
226789c6957cSBarry Smith 	  tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6];
226889c6957cSBarry Smith 	  tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7];
226989c6957cSBarry Smith 	  tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8];
227089c6957cSBarry Smith 	  t[row] = b[row] - tmp1;
227189c6957cSBarry Smith 	  t[row+1] = b[row+1] - tmp2;
227289c6957cSBarry Smith 	  t[row+2] = b[row+2] - tmp3; row += 3;
227389c6957cSBarry Smith 	  cnt += 9;
227489c6957cSBarry Smith 	  break;
227589c6957cSBarry Smith         case 4:
227689c6957cSBarry Smith 	  x1   = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3];
227789c6957cSBarry Smith 	  tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12];
227889c6957cSBarry Smith 	  tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13];
227989c6957cSBarry Smith 	  tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14];
228089c6957cSBarry Smith 	  tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15];
228189c6957cSBarry Smith 	  t[row] = b[row] - tmp1;
228289c6957cSBarry Smith 	  t[row+1] = b[row+1] - tmp2;
228389c6957cSBarry Smith 	  t[row+2] = b[row+2] - tmp3;
228489c6957cSBarry Smith 	  t[row+3] = b[row+3] - tmp4; row += 4;
228589c6957cSBarry Smith 	  cnt += 16;
228689c6957cSBarry Smith 	  break;
228789c6957cSBarry Smith         case 5:
228889c6957cSBarry Smith 	  x1   = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; x5 = x[row+4];
228989c6957cSBarry Smith 	  tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20];
229089c6957cSBarry Smith 	  tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21];
229189c6957cSBarry Smith 	  tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22];
229289c6957cSBarry Smith 	  tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23];
229389c6957cSBarry Smith 	  tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24];
229489c6957cSBarry Smith 	  t[row] = b[row] - tmp1;
229589c6957cSBarry Smith 	  t[row+1] = b[row+1] - tmp2;
229689c6957cSBarry Smith 	  t[row+2] = b[row+2] - tmp3;
229789c6957cSBarry Smith 	  t[row+3] = b[row+3] - tmp4;
229889c6957cSBarry Smith 	  t[row+4] = b[row+4] - tmp5;row += 5;
229989c6957cSBarry Smith 	  cnt += 25;
230089c6957cSBarry Smith 	  break;
230189c6957cSBarry Smith         default:
230289c6957cSBarry Smith 	  SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
230389c6957cSBarry Smith       }
230489c6957cSBarry Smith       CHKMEMQ;
230589c6957cSBarry Smith     }
230689c6957cSBarry Smith     ierr = PetscLogFlops(m);CHKERRQ(ierr);
230789c6957cSBarry Smith 
230889c6957cSBarry Smith 
230989c6957cSBarry Smith 
231089c6957cSBarry Smith     /*
231189c6957cSBarry Smith           Apply (L + D)^-1 where D is the block diagonal
231289c6957cSBarry Smith     */
231389c6957cSBarry Smith     for (i=0, row=0; i<m; i++) {
231489c6957cSBarry Smith       sz  = diag[row] - ii[row];
231589c6957cSBarry Smith       v1  = a->a + ii[row];
231689c6957cSBarry Smith       idx = a->j + ii[row];
231789c6957cSBarry Smith       CHKMEMQ;
23184108e4d5SBarry Smith       /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
231989c6957cSBarry Smith       switch (sizes[i]){
232089c6957cSBarry Smith         case 1:
232189c6957cSBarry Smith 
232289c6957cSBarry Smith 	  sum1  = t[row];
232389c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
232489c6957cSBarry Smith 	    i1   = idx[0];
232589c6957cSBarry Smith 	    i2   = idx[1];
232689c6957cSBarry Smith 	    idx += 2;
232789c6957cSBarry Smith 	    tmp0 = t[i1];
232889c6957cSBarry Smith 	    tmp1 = t[i2];
232989c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
233089c6957cSBarry Smith 	  }
233189c6957cSBarry Smith 
233289c6957cSBarry Smith 	  if (n == sz-1){
233389c6957cSBarry Smith 	    tmp0  = t[*idx];
233489c6957cSBarry Smith 	    sum1 -= *v1 * tmp0;
233589c6957cSBarry Smith 	  }
233689c6957cSBarry Smith 	  x[row] += t[row] = sum1*(*ibdiag++); row++;
233789c6957cSBarry Smith 	  break;
233889c6957cSBarry Smith         case 2:
233989c6957cSBarry Smith 	  v2    = a->a + ii[row+1];
234089c6957cSBarry Smith 	  sum1  = t[row];
234189c6957cSBarry Smith 	  sum2  = t[row+1];
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 	  }
235189c6957cSBarry Smith 
235289c6957cSBarry Smith 	  if (n == sz-1){
235389c6957cSBarry Smith 	    tmp0  = t[*idx];
235489c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0;
235589c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0;
235689c6957cSBarry Smith 	  }
235789c6957cSBarry Smith 	  x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[2];
235889c6957cSBarry Smith 	  x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3];
235989c6957cSBarry Smith 	  ibdiag  += 4; row += 2;
236089c6957cSBarry Smith 	  break;
236189c6957cSBarry Smith         case 3:
236289c6957cSBarry Smith 	  v2    = a->a + ii[row+1];
236389c6957cSBarry Smith 	  v3    = a->a + ii[row+2];
236489c6957cSBarry Smith 	  sum1  = t[row];
236589c6957cSBarry Smith 	  sum2  = t[row+1];
236689c6957cSBarry Smith 	  sum3  = t[row+2];
236789c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
236889c6957cSBarry Smith 	    i1   = idx[0];
236989c6957cSBarry Smith 	    i2   = idx[1];
237089c6957cSBarry Smith 	    idx += 2;
237189c6957cSBarry Smith 	    tmp0 = t[i1];
237289c6957cSBarry Smith 	    tmp1 = t[i2];
237389c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
237489c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
237589c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
237689c6957cSBarry Smith 	  }
237789c6957cSBarry Smith 
237889c6957cSBarry Smith 	  if (n == sz-1){
237989c6957cSBarry Smith 	    tmp0  = t[*idx];
238089c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0;
238189c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0;
238289c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0;
238389c6957cSBarry Smith 	  }
238489c6957cSBarry Smith 	  x[row]  += t[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
238589c6957cSBarry Smith 	  x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
238689c6957cSBarry Smith 	  x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
238789c6957cSBarry Smith 	  ibdiag  += 9; row += 3;
238889c6957cSBarry Smith 	  break;
238989c6957cSBarry Smith         case 4:
239089c6957cSBarry Smith 	  v2    = a->a + ii[row+1];
239189c6957cSBarry Smith 	  v3    = a->a + ii[row+2];
239289c6957cSBarry Smith 	  v4    = a->a + ii[row+3];
239389c6957cSBarry Smith 	  sum1  = t[row];
239489c6957cSBarry Smith 	  sum2  = t[row+1];
239589c6957cSBarry Smith 	  sum3  = t[row+2];
239689c6957cSBarry Smith 	  sum4  = t[row+3];
239789c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
239889c6957cSBarry Smith 	    i1   = idx[0];
239989c6957cSBarry Smith 	    i2   = idx[1];
240089c6957cSBarry Smith 	    idx += 2;
240189c6957cSBarry Smith 	    tmp0 = t[i1];
240289c6957cSBarry Smith 	    tmp1 = t[i2];
240389c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
240489c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
240589c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
240689c6957cSBarry Smith 	    sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
240789c6957cSBarry Smith 	  }
240889c6957cSBarry Smith 
240989c6957cSBarry Smith 	  if (n == sz-1){
241089c6957cSBarry Smith 	    tmp0  = t[*idx];
241189c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0;
241289c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0;
241389c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0;
241489c6957cSBarry Smith 	    sum4 -= v4[0] * tmp0;
241589c6957cSBarry Smith 	  }
241689c6957cSBarry Smith 	  x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
241789c6957cSBarry Smith 	  x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
241889c6957cSBarry Smith 	  x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
241989c6957cSBarry Smith 	  x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
242089c6957cSBarry Smith 	  ibdiag  += 16; row += 4;
242189c6957cSBarry Smith 	  break;
242289c6957cSBarry Smith         case 5:
242389c6957cSBarry Smith 	  v2    = a->a + ii[row+1];
242489c6957cSBarry Smith 	  v3    = a->a + ii[row+2];
242589c6957cSBarry Smith 	  v4    = a->a + ii[row+3];
242689c6957cSBarry Smith 	  v5    = a->a + ii[row+4];
242789c6957cSBarry Smith 	  sum1  = t[row];
242889c6957cSBarry Smith 	  sum2  = t[row+1];
242989c6957cSBarry Smith 	  sum3  = t[row+2];
243089c6957cSBarry Smith 	  sum4  = t[row+3];
243189c6957cSBarry Smith 	  sum5  = t[row+4];
243289c6957cSBarry Smith 	  for(n = 0; n<sz-1; n+=2) {
243389c6957cSBarry Smith 	    i1   = idx[0];
243489c6957cSBarry Smith 	    i2   = idx[1];
243589c6957cSBarry Smith 	    idx += 2;
243689c6957cSBarry Smith 	    tmp0 = t[i1];
243789c6957cSBarry Smith 	    tmp1 = t[i2];
243889c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
243989c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
244089c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
244189c6957cSBarry Smith 	    sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
244289c6957cSBarry Smith 	    sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
244389c6957cSBarry Smith 	  }
244489c6957cSBarry Smith 
244589c6957cSBarry Smith 	  if (n == sz-1){
244689c6957cSBarry Smith 	    tmp0  = t[*idx];
244789c6957cSBarry Smith 	    sum1 -= v1[0] * tmp0;
244889c6957cSBarry Smith 	    sum2 -= v2[0] * tmp0;
244989c6957cSBarry Smith 	    sum3 -= v3[0] * tmp0;
245089c6957cSBarry Smith 	    sum4 -= v4[0] * tmp0;
245189c6957cSBarry Smith 	    sum5 -= v5[0] * tmp0;
245289c6957cSBarry Smith 	  }
245389c6957cSBarry Smith 	  x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
245489c6957cSBarry Smith 	  x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
245589c6957cSBarry Smith 	  x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
245689c6957cSBarry Smith 	  x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
245789c6957cSBarry Smith 	  x[row+4] += t[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
245889c6957cSBarry Smith 	  ibdiag  += 25; row += 5;
245989c6957cSBarry Smith 	  break;
246089c6957cSBarry Smith         default:
246189c6957cSBarry Smith 	  SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
246289c6957cSBarry Smith       }
246389c6957cSBarry Smith       CHKMEMQ;
246489c6957cSBarry Smith     }
246589c6957cSBarry Smith     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
24662af78befSBarry Smith     ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
2467938d4eb3SBarry Smith     ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
2468938d4eb3SBarry Smith   }
24692af78befSBarry Smith   PetscFunctionReturn(0);
24702af78befSBarry Smith }
24712af78befSBarry Smith 
247289c6957cSBarry Smith #undef __FUNCT__
24734108e4d5SBarry Smith #define __FUNCT__ "MatMultDiagonalBlock_SeqAIJ_Inode"
24744108e4d5SBarry Smith PetscErrorCode MatMultDiagonalBlock_SeqAIJ_Inode(Mat A,Vec bb,Vec xx)
247589c6957cSBarry Smith {
247689c6957cSBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
247789c6957cSBarry Smith   PetscScalar        *x,tmp1,tmp2,tmp3,tmp4,tmp5,x1,x2,x3,x4,x5;
247889c6957cSBarry Smith   const MatScalar    *bdiag = a->inode.bdiag;
247989c6957cSBarry Smith   const PetscScalar  *b;
248089c6957cSBarry Smith   PetscErrorCode      ierr;
248189c6957cSBarry Smith   PetscInt            m = a->inode.node_count,cnt = 0,i,row;
248289c6957cSBarry Smith   const PetscInt      *sizes = a->inode.size;
24832af78befSBarry Smith 
248489c6957cSBarry Smith   PetscFunctionBegin;
248589c6957cSBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
248689c6957cSBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
248789c6957cSBarry Smith   cnt = 0;
248889c6957cSBarry Smith   for (i=0, row=0; i<m; i++) {
248989c6957cSBarry Smith     switch (sizes[i]){
249089c6957cSBarry Smith       case 1:
249189c6957cSBarry Smith 	x[row] = b[row]*bdiag[cnt++];row++;
249289c6957cSBarry Smith 	break;
249389c6957cSBarry Smith       case 2:
249489c6957cSBarry Smith 	x1   = b[row]; x2 = b[row+1];
249589c6957cSBarry Smith 	tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2];
249689c6957cSBarry Smith 	tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3];
249789c6957cSBarry Smith 	x[row++] = tmp1;
249889c6957cSBarry Smith 	x[row++] = tmp2;
249989c6957cSBarry Smith 	cnt += 4;
250089c6957cSBarry Smith 	break;
250189c6957cSBarry Smith       case 3:
250289c6957cSBarry Smith 	x1   = b[row]; x2 = b[row+1]; x3 = b[row+2];
250389c6957cSBarry Smith 	tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6];
250489c6957cSBarry Smith 	tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7];
250589c6957cSBarry Smith 	tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8];
250689c6957cSBarry Smith 	x[row++] = tmp1;
250789c6957cSBarry Smith 	x[row++] = tmp2;
250889c6957cSBarry Smith 	x[row++] = tmp3;
250989c6957cSBarry Smith 	cnt += 9;
251089c6957cSBarry Smith 	break;
251189c6957cSBarry Smith       case 4:
251289c6957cSBarry Smith 	x1   = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3];
251389c6957cSBarry Smith 	tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12];
251489c6957cSBarry Smith 	tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13];
251589c6957cSBarry Smith 	tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14];
251689c6957cSBarry Smith 	tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15];
251789c6957cSBarry Smith 	x[row++] = tmp1;
251889c6957cSBarry Smith 	x[row++] = tmp2;
251989c6957cSBarry Smith 	x[row++] = tmp3;
252089c6957cSBarry Smith 	x[row++] = tmp4;
252189c6957cSBarry Smith 	cnt += 16;
252289c6957cSBarry Smith 	break;
252389c6957cSBarry Smith       case 5:
252489c6957cSBarry Smith 	x1   = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3]; x5 = b[row+4];
252589c6957cSBarry Smith 	tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20];
252689c6957cSBarry Smith 	tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21];
252789c6957cSBarry Smith 	tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22];
252889c6957cSBarry Smith 	tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23];
252989c6957cSBarry Smith 	tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24];
253089c6957cSBarry Smith 	x[row++] = tmp1;
253189c6957cSBarry Smith 	x[row++] = tmp2;
253289c6957cSBarry Smith 	x[row++] = tmp3;
253389c6957cSBarry Smith 	x[row++] = tmp4;
253489c6957cSBarry Smith 	x[row++] = tmp5;
253589c6957cSBarry Smith 	cnt += 25;
253689c6957cSBarry Smith 	break;
253789c6957cSBarry Smith       default:
253889c6957cSBarry Smith 	SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
253989c6957cSBarry Smith     }
254089c6957cSBarry Smith   }
254189c6957cSBarry Smith   ierr = PetscLogFlops(2*cnt);CHKERRQ(ierr);
254289c6957cSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
254389c6957cSBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
254489c6957cSBarry Smith   PetscFunctionReturn(0);
254589c6957cSBarry Smith }
254689c6957cSBarry Smith 
25474c1414c8SBarry Smith /*
25484c1414c8SBarry Smith     samestructure indicates that the matrix has not changed its nonzero structure so we
25494c1414c8SBarry Smith     do not need to recompute the inodes
25504c1414c8SBarry Smith */
25514c1414c8SBarry Smith #undef __FUNCT__
25524c1414c8SBarry Smith #define __FUNCT__ "Mat_CheckInode"
25534c1414c8SBarry Smith PetscErrorCode Mat_CheckInode(Mat A,PetscTruth samestructure)
25544c1414c8SBarry Smith {
25554c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
25564c1414c8SBarry Smith   PetscErrorCode ierr;
25574c1414c8SBarry Smith   PetscInt       i,j,m,nzx,nzy,*idx,*idy,*ns,*ii,node_count,blk_size;
2558d74fe821SBarry Smith   PetscTruth     flag;
25594c1414c8SBarry Smith 
25604c1414c8SBarry Smith   PetscFunctionBegin;
2561d74fe821SBarry Smith   if (!a->inode.use)                     PetscFunctionReturn(0);
25624c1414c8SBarry Smith   if (a->inode.checked && samestructure) PetscFunctionReturn(0);
25634c1414c8SBarry Smith 
25644c1414c8SBarry Smith 
2565d0f46423SBarry Smith   m = A->rmap->n;
25664c1414c8SBarry Smith   if (a->inode.size) {ns = a->inode.size;}
25674c1414c8SBarry Smith   else {ierr = PetscMalloc((m+1)*sizeof(PetscInt),&ns);CHKERRQ(ierr);}
25684c1414c8SBarry Smith 
25694c1414c8SBarry Smith   i          = 0;
25704c1414c8SBarry Smith   node_count = 0;
25714c1414c8SBarry Smith   idx        = a->j;
25724c1414c8SBarry Smith   ii         = a->i;
25734c1414c8SBarry Smith   while (i < m){                /* For each row */
25744c1414c8SBarry Smith     nzx = ii[i+1] - ii[i];       /* Number of nonzeros */
25754c1414c8SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
25764c1414c8SBarry Smith     for (j=i+1,idy=idx,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
25774c1414c8SBarry Smith       nzy     = ii[j+1] - ii[j]; /* Same number of nonzeros */
25784c1414c8SBarry Smith       if (nzy != nzx) break;
25794c1414c8SBarry Smith       idy  += nzx;             /* Same nonzero pattern */
25804c1414c8SBarry Smith       ierr = PetscMemcmp(idx,idy,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr);
25814c1414c8SBarry Smith       if (!flag) break;
25824c1414c8SBarry Smith     }
25834c1414c8SBarry Smith     ns[node_count++] = blk_size;
25844c1414c8SBarry Smith     idx += blk_size*nzx;
25854c1414c8SBarry Smith     i    = j;
25864c1414c8SBarry Smith   }
25874c1414c8SBarry Smith   /* If not enough inodes found,, do not use inode version of the routines */
2588f0d39aaaSBarry Smith   if (!a->inode.size && m && node_count > .9*m) {
25894c1414c8SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
25904c1414c8SBarry Smith     a->inode.node_count     = 0;
25914c1414c8SBarry Smith     a->inode.size           = PETSC_NULL;
25924c1414c8SBarry Smith     a->inode.use            = PETSC_FALSE;
25934c1414c8SBarry Smith     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
25944c1414c8SBarry Smith   } else {
25954108e4d5SBarry Smith     A->ops->mult              = MatMult_SeqAIJ_Inode;
25964108e4d5SBarry Smith     A->ops->sor               = MatSOR_SeqAIJ_Inode;
25974108e4d5SBarry Smith     A->ops->multadd           = MatMultAdd_SeqAIJ_Inode;
25984108e4d5SBarry Smith     A->ops->getrowij          = MatGetRowIJ_SeqAIJ_Inode;
25994108e4d5SBarry Smith     A->ops->restorerowij      = MatRestoreRowIJ_SeqAIJ_Inode;
26004108e4d5SBarry Smith     A->ops->getcolumnij       = MatGetColumnIJ_SeqAIJ_Inode;
26014108e4d5SBarry Smith     A->ops->restorecolumnij   = MatRestoreColumnIJ_SeqAIJ_Inode;
26024108e4d5SBarry Smith     A->ops->coloringpatch     = MatColoringPatch_SeqAIJ_Inode;
26034108e4d5SBarry Smith     A->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode;
26044c1414c8SBarry Smith     a->inode.node_count       = node_count;
26054c1414c8SBarry Smith     a->inode.size             = ns;
26064c1414c8SBarry Smith     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
26074c1414c8SBarry Smith   }
26084c1414c8SBarry Smith   PetscFunctionReturn(0);
26094c1414c8SBarry Smith }
26104c1414c8SBarry Smith 
2611019b515eSShri Abhyankar #define MatGetRow_FactoredLU(cols,nzl,nzu,nz,ai,aj,adiag,row) {	\
2612019b515eSShri Abhyankar PetscInt __k, *__vi; \
2613019b515eSShri Abhyankar __vi = aj + ai[row];				\
2614019b515eSShri Abhyankar for(__k=0;__k<nzl;__k++) cols[__k] = __vi[__k]; \
2615019b515eSShri Abhyankar __vi = aj + adiag[row];				\
2616019b515eSShri Abhyankar cols[nzl] = __vi[0];\
2617019b515eSShri Abhyankar __vi = aj + adiag[row+1]+1;\
2618019b515eSShri Abhyankar for(__k=0;__k<nzu;__k++) cols[nzl+1+__k] = __vi[__k];}
2619019b515eSShri Abhyankar 
2620019b515eSShri Abhyankar #undef __FUNCT__
2621019b515eSShri Abhyankar #define __FUNCT__ "Mat_CheckInode_FactorLU"
2622019b515eSShri Abhyankar PetscErrorCode Mat_CheckInode_FactorLU(Mat A,PetscTruth samestructure)
2623019b515eSShri Abhyankar {
2624019b515eSShri Abhyankar   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
2625019b515eSShri Abhyankar   PetscErrorCode ierr;
2626019b515eSShri Abhyankar   PetscInt       i,j,m,nzl1,nzu1,nzl2,nzu2,nzx,nzy,node_count,blk_size;
2627019b515eSShri Abhyankar   PetscInt       *cols1,*cols2,*ns,*ai = a->i,*aj = a->j, *adiag = a->diag;
2628019b515eSShri Abhyankar   PetscTruth     flag;
2629019b515eSShri Abhyankar 
2630019b515eSShri Abhyankar   PetscFunctionBegin;
2631019b515eSShri Abhyankar   if (!a->inode.use)                     PetscFunctionReturn(0);
2632019b515eSShri Abhyankar   if (a->inode.checked && samestructure) PetscFunctionReturn(0);
2633019b515eSShri Abhyankar 
2634019b515eSShri Abhyankar   m = A->rmap->n;
2635019b515eSShri Abhyankar   if (a->inode.size) {ns = a->inode.size;}
2636019b515eSShri Abhyankar   else {ierr = PetscMalloc((m+1)*sizeof(PetscInt),&ns);CHKERRQ(ierr);}
2637019b515eSShri Abhyankar 
2638019b515eSShri Abhyankar   i          = 0;
2639019b515eSShri Abhyankar   node_count = 0;
2640019b515eSShri Abhyankar   while (i < m){                /* For each row */
2641019b515eSShri Abhyankar     nzl1 = ai[i+1] - ai[i];       /* Number of nonzeros in L */
2642019b515eSShri Abhyankar     nzu1 = adiag[i] - adiag[i+1] - 1; /* Number of nonzeros in U excluding diagonal*/
2643019b515eSShri Abhyankar     nzx  = nzl1 + nzu1 + 1;
2644019b515eSShri Abhyankar     ierr = PetscMalloc((nzx+1)*sizeof(PetscInt),&cols1);CHKERRQ(ierr);
2645019b515eSShri Abhyankar     MatGetRow_FactoredLU(cols1,nzl1,nzu1,nzx,ai,aj,adiag,i);
2646019b515eSShri Abhyankar 
2647019b515eSShri Abhyankar     /* Limits the number of elements in a node to 'a->inode.limit' */
2648019b515eSShri Abhyankar     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
2649019b515eSShri Abhyankar       nzl2    = ai[j+1] - ai[j];
2650019b515eSShri Abhyankar       nzu2    = adiag[j] - adiag[j+1] - 1;
2651019b515eSShri Abhyankar       nzy     = nzl2 + nzu2 + 1;
2652019b515eSShri Abhyankar       if( nzy != nzx) break;
2653019b515eSShri Abhyankar       ierr    = PetscMalloc((nzy+1)*sizeof(PetscInt),&cols2);CHKERRQ(ierr);
2654019b515eSShri Abhyankar       MatGetRow_FactoredLU(cols2,nzl2,nzu2,nzy,ai,aj,adiag,j);
2655019b515eSShri Abhyankar       ierr = PetscMemcmp(cols1,cols2,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr);
2656019b515eSShri Abhyankar       if (!flag) {ierr = PetscFree(cols2);CHKERRQ(ierr);break;}
2657019b515eSShri Abhyankar       ierr = PetscFree(cols2);CHKERRQ(ierr);
2658019b515eSShri Abhyankar     }
2659019b515eSShri Abhyankar     ns[node_count++] = blk_size;
2660019b515eSShri Abhyankar     ierr = PetscFree(cols1);CHKERRQ(ierr);
2661019b515eSShri Abhyankar     i    = j;
2662019b515eSShri Abhyankar   }
2663019b515eSShri Abhyankar   /* If not enough inodes found,, do not use inode version of the routines */
2664019b515eSShri Abhyankar   if (!a->inode.size && m && node_count > .9*m) {
2665019b515eSShri Abhyankar     ierr = PetscFree(ns);CHKERRQ(ierr);
2666019b515eSShri Abhyankar     a->inode.node_count     = 0;
2667019b515eSShri Abhyankar     a->inode.size           = PETSC_NULL;
2668019b515eSShri Abhyankar     a->inode.use            = PETSC_FALSE;
2669019b515eSShri Abhyankar     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
2670019b515eSShri Abhyankar   } else {
2671019b515eSShri Abhyankar     A->ops->mult              = 0;
2672019b515eSShri Abhyankar     A->ops->sor               = 0;
2673019b515eSShri Abhyankar     A->ops->multadd           = 0;
2674019b515eSShri Abhyankar     A->ops->getrowij          = 0;
2675019b515eSShri Abhyankar     A->ops->restorerowij      = 0;
2676019b515eSShri Abhyankar     A->ops->getcolumnij       = 0;
2677019b515eSShri Abhyankar     A->ops->restorecolumnij   = 0;
2678019b515eSShri Abhyankar     A->ops->coloringpatch     = 0;
2679019b515eSShri Abhyankar     A->ops->multdiagonalblock = 0;
2680019b515eSShri Abhyankar     a->inode.node_count       = node_count;
2681019b515eSShri Abhyankar     a->inode.size             = ns;
2682019b515eSShri Abhyankar     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
2683019b515eSShri Abhyankar   }
2684019b515eSShri Abhyankar   PetscFunctionReturn(0);
2685019b515eSShri Abhyankar }
2686019b515eSShri Abhyankar 
26874c1414c8SBarry Smith /*
26884c1414c8SBarry Smith      This is really ugly. if inodes are used this replaces the
26894c1414c8SBarry Smith   permutations with ones that correspond to rows/cols of the matrix
26904c1414c8SBarry Smith   rather then inode blocks
26914c1414c8SBarry Smith */
26924c1414c8SBarry Smith #undef __FUNCT__
26934c1414c8SBarry Smith #define __FUNCT__ "MatInodeAdjustForInodes"
26944c1414c8SBarry Smith PetscErrorCode PETSCMAT_DLLEXPORT MatInodeAdjustForInodes(Mat A,IS *rperm,IS *cperm)
26954c1414c8SBarry Smith {
26964c1414c8SBarry Smith   PetscErrorCode ierr,(*f)(Mat,IS*,IS*);
26974c1414c8SBarry Smith 
26984c1414c8SBarry Smith   PetscFunctionBegin;
26994c1414c8SBarry Smith   ierr = PetscObjectQueryFunction((PetscObject)A,"MatInodeAdjustForInodes_C",(void (**)(void))&f);CHKERRQ(ierr);
27004c1414c8SBarry Smith   if (f) {
27014c1414c8SBarry Smith     ierr = (*f)(A,rperm,cperm);CHKERRQ(ierr);
27024c1414c8SBarry Smith   }
27034c1414c8SBarry Smith   PetscFunctionReturn(0);
27044c1414c8SBarry Smith }
27054c1414c8SBarry Smith 
27064c1414c8SBarry Smith EXTERN_C_BEGIN
27074c1414c8SBarry Smith #undef __FUNCT__
27084108e4d5SBarry Smith #define __FUNCT__ "MatAdjustForInodes_SeqAIJ_Inode"
27094108e4d5SBarry Smith PetscErrorCode PETSCMAT_DLLEXPORT MatInodeAdjustForInodes_SeqAIJ_Inode(Mat A,IS *rperm,IS *cperm)
27104c1414c8SBarry Smith {
27114c1414c8SBarry Smith   Mat_SeqAIJ      *a=(Mat_SeqAIJ*)A->data;
27124c1414c8SBarry Smith   PetscErrorCode ierr;
27135d0c19d7SBarry Smith   PetscInt       m = A->rmap->n,n = A->cmap->n,i,j,nslim_row = a->inode.node_count;
27145d0c19d7SBarry Smith   const PetscInt *ridx,*cidx;
27154c1414c8SBarry Smith   PetscInt       row,col,*permr,*permc,*ns_row =  a->inode.size,*tns,start_val,end_val,indx;
27164c1414c8SBarry Smith   PetscInt       nslim_col,*ns_col;
27174c1414c8SBarry Smith   IS             ris = *rperm,cis = *cperm;
27184c1414c8SBarry Smith 
27194c1414c8SBarry Smith   PetscFunctionBegin;
27204c1414c8SBarry Smith   if (!a->inode.size) PetscFunctionReturn(0); /* no inodes so return */
27214c1414c8SBarry Smith   if (a->inode.node_count == m) PetscFunctionReturn(0); /* all inodes are of size 1 */
27224c1414c8SBarry Smith 
27234c1414c8SBarry Smith   ierr  = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr);
27244c1414c8SBarry Smith   ierr  = PetscMalloc((((nslim_row>nslim_col)?nslim_row:nslim_col)+1)*sizeof(PetscInt),&tns);CHKERRQ(ierr);
27250e83c824SBarry Smith   ierr  = PetscMalloc2(m,PetscInt,&permr,n,PetscInt,&permc);CHKERRQ(ierr);
27264c1414c8SBarry Smith 
27274c1414c8SBarry Smith   ierr  = ISGetIndices(ris,&ridx);CHKERRQ(ierr);
27284c1414c8SBarry Smith   ierr  = ISGetIndices(cis,&cidx);CHKERRQ(ierr);
27294c1414c8SBarry Smith 
27304c1414c8SBarry Smith   /* Form the inode structure for the rows of permuted matric using inv perm*/
27314c1414c8SBarry Smith   for (i=0,tns[0]=0; i<nslim_row; ++i) tns[i+1] = tns[i] + ns_row[i];
27324c1414c8SBarry Smith 
27334c1414c8SBarry Smith   /* Construct the permutations for rows*/
27344c1414c8SBarry Smith   for (i=0,row = 0; i<nslim_row; ++i){
27354c1414c8SBarry Smith     indx      = ridx[i];
27364c1414c8SBarry Smith     start_val = tns[indx];
27374c1414c8SBarry Smith     end_val   = tns[indx + 1];
27384c1414c8SBarry Smith     for (j=start_val; j<end_val; ++j,++row) permr[row]= j;
27394c1414c8SBarry Smith   }
27404c1414c8SBarry Smith 
27414c1414c8SBarry Smith   /* Form the inode structure for the columns of permuted matrix using inv perm*/
27424c1414c8SBarry Smith   for (i=0,tns[0]=0; i<nslim_col; ++i) tns[i+1] = tns[i] + ns_col[i];
27434c1414c8SBarry Smith 
27444c1414c8SBarry Smith  /* Construct permutations for columns */
27454c1414c8SBarry Smith   for (i=0,col=0; i<nslim_col; ++i){
27464c1414c8SBarry Smith     indx      = cidx[i];
27474c1414c8SBarry Smith     start_val = tns[indx];
27484c1414c8SBarry Smith     end_val   = tns[indx + 1];
27494c1414c8SBarry Smith     for (j = start_val; j<end_val; ++j,++col) permc[col]= j;
27504c1414c8SBarry Smith   }
27514c1414c8SBarry Smith 
27524c1414c8SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permr,rperm);CHKERRQ(ierr);
27530e83c824SBarry Smith   ierr = ISSetPermutation(*rperm);CHKERRQ(ierr);
27544c1414c8SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permc,cperm);CHKERRQ(ierr);
27550e83c824SBarry Smith   ierr = ISSetPermutation(*cperm);CHKERRQ(ierr);
27564c1414c8SBarry Smith 
27574c1414c8SBarry Smith   ierr  = ISRestoreIndices(ris,&ridx);CHKERRQ(ierr);
27584c1414c8SBarry Smith   ierr  = ISRestoreIndices(cis,&cidx);CHKERRQ(ierr);
27594c1414c8SBarry Smith 
27604c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
27610e83c824SBarry Smith   ierr = PetscFree2(permr,permc);CHKERRQ(ierr);
27624c1414c8SBarry Smith   ierr = ISDestroy(cis);CHKERRQ(ierr);
27634c1414c8SBarry Smith   ierr = ISDestroy(ris);CHKERRQ(ierr);
27644c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
27654c1414c8SBarry Smith   PetscFunctionReturn(0);
27664c1414c8SBarry Smith }
27674c1414c8SBarry Smith EXTERN_C_END
27684c1414c8SBarry Smith 
27694c1414c8SBarry Smith #undef __FUNCT__
27704c1414c8SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes"
27714c1414c8SBarry Smith /*@C
27724c1414c8SBarry Smith    MatInodeGetInodeSizes - Returns the inode information of the Inode matrix.
27734c1414c8SBarry Smith 
27744c1414c8SBarry Smith    Collective on Mat
27754c1414c8SBarry Smith 
27764c1414c8SBarry Smith    Input Parameter:
27774c1414c8SBarry Smith .  A - the Inode matrix or matrix derived from the Inode class -- e.g., SeqAIJ
27784c1414c8SBarry Smith 
27794c1414c8SBarry Smith    Output Parameter:
27804c1414c8SBarry Smith +  node_count - no of inodes present in the matrix.
27814c1414c8SBarry Smith .  sizes      - an array of size node_count,with sizes of each inode.
27824c1414c8SBarry Smith -  limit      - the max size used to generate the inodes.
27834c1414c8SBarry Smith 
27844c1414c8SBarry Smith    Level: advanced
27854c1414c8SBarry Smith 
27864c1414c8SBarry Smith    Notes: This routine returns some internal storage information
27874c1414c8SBarry Smith    of the matrix, it is intended to be used by advanced users.
27884c1414c8SBarry Smith    It should be called after the matrix is assembled.
27894c1414c8SBarry Smith    The contents of the sizes[] array should not be changed.
27904c1414c8SBarry Smith    PETSC_NULL may be passed for information not requested.
27914c1414c8SBarry Smith 
27924c1414c8SBarry Smith .keywords: matrix, seqaij, get, inode
27934c1414c8SBarry Smith 
27944c1414c8SBarry Smith .seealso: MatGetInfo()
27954c1414c8SBarry Smith @*/
27964c1414c8SBarry Smith PetscErrorCode PETSCMAT_DLLEXPORT MatInodeGetInodeSizes(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit)
27974c1414c8SBarry Smith {
27984c1414c8SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscInt*,PetscInt*[],PetscInt*);
27994c1414c8SBarry Smith 
28004c1414c8SBarry Smith   PetscFunctionBegin;
28014c1414c8SBarry Smith   if (!A->assembled) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
28024c1414c8SBarry Smith   ierr = PetscObjectQueryFunction((PetscObject)A,"MatInodeGetInodeSizes_C",(void (**)(void))&f);CHKERRQ(ierr);
28034c1414c8SBarry Smith   if (f) {
28044c1414c8SBarry Smith     ierr = (*f)(A,node_count,sizes,limit);CHKERRQ(ierr);
28054c1414c8SBarry Smith   }
28064c1414c8SBarry Smith   PetscFunctionReturn(0);
28074c1414c8SBarry Smith }
28084c1414c8SBarry Smith 
28094c1414c8SBarry Smith EXTERN_C_BEGIN
28104c1414c8SBarry Smith #undef __FUNCT__
28114108e4d5SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes_SeqAIJ_Inode"
28124108e4d5SBarry Smith PetscErrorCode PETSCMAT_DLLEXPORT MatInodeGetInodeSizes_SeqAIJ_Inode(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit)
28134c1414c8SBarry Smith {
28144c1414c8SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
28154c1414c8SBarry Smith 
28164c1414c8SBarry Smith   PetscFunctionBegin;
28174c1414c8SBarry Smith   if (node_count) *node_count = a->inode.node_count;
28184c1414c8SBarry Smith   if (sizes)      *sizes      = a->inode.size;
28194c1414c8SBarry Smith   if (limit)      *limit      = a->inode.limit;
28204c1414c8SBarry Smith   PetscFunctionReturn(0);
28214c1414c8SBarry Smith }
28224c1414c8SBarry Smith EXTERN_C_END
2823