xref: /petsc/src/mat/impls/aij/seq/inode.c (revision dcca6d9d80ebd869fe6029bd05a3aa9faafef49e)
14c1414c8SBarry Smith 
24c1414c8SBarry Smith /*
34c1414c8SBarry Smith   This file provides high performance routines for the Inode format (compressed sparse row)
44c1414c8SBarry Smith   by taking advantage of rows with identical nonzero structure (I-nodes).
54c1414c8SBarry Smith */
6c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/aij.h>
74c1414c8SBarry Smith 
84c1414c8SBarry Smith #undef __FUNCT__
94c1414c8SBarry Smith #define __FUNCT__ "Mat_CreateColInode"
104c1414c8SBarry Smith static PetscErrorCode Mat_CreateColInode(Mat A,PetscInt *size,PetscInt **ns)
114c1414c8SBarry Smith {
124c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
134c1414c8SBarry Smith   PetscErrorCode ierr;
144c1414c8SBarry Smith   PetscInt       i,count,m,n,min_mn,*ns_row,*ns_col;
154c1414c8SBarry Smith 
164c1414c8SBarry Smith   PetscFunctionBegin;
17d0f46423SBarry Smith   n      = A->cmap->n;
18d0f46423SBarry Smith   m      = A->rmap->n;
194c1414c8SBarry Smith   ns_row = a->inode.size;
204c1414c8SBarry Smith 
214c1414c8SBarry Smith   min_mn = (m < n) ? m : n;
224c1414c8SBarry Smith   if (!ns) {
234c1414c8SBarry Smith     for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) ;
244c1414c8SBarry Smith     for (; count+1 < n; count++,i++) ;
254c1414c8SBarry Smith     if (count < n)  {
264c1414c8SBarry Smith       i++;
274c1414c8SBarry Smith     }
284c1414c8SBarry Smith     *size = i;
294c1414c8SBarry Smith     PetscFunctionReturn(0);
304c1414c8SBarry Smith   }
314c1414c8SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&ns_col);CHKERRQ(ierr);
324c1414c8SBarry Smith 
334c1414c8SBarry Smith   /* Use the same row structure wherever feasible. */
344c1414c8SBarry Smith   for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) {
354c1414c8SBarry Smith     ns_col[i] = ns_row[i];
364c1414c8SBarry Smith   }
374c1414c8SBarry Smith 
384c1414c8SBarry Smith   /* if m < n; pad up the remainder with inode_limit */
394c1414c8SBarry Smith   for (; count+1 < n; count++,i++) {
404c1414c8SBarry Smith     ns_col[i] = 1;
414c1414c8SBarry Smith   }
424c1414c8SBarry Smith   /* The last node is the odd ball. padd it up with the remaining rows; */
434c1414c8SBarry Smith   if (count < n) {
444c1414c8SBarry Smith     ns_col[i] = n - count;
454c1414c8SBarry Smith     i++;
464c1414c8SBarry Smith   } else if (count > n) {
474c1414c8SBarry Smith     /* Adjust for the over estimation */
484c1414c8SBarry Smith     ns_col[i-1] += n - count;
494c1414c8SBarry Smith   }
504c1414c8SBarry Smith   *size = i;
514c1414c8SBarry Smith   *ns   = ns_col;
524c1414c8SBarry Smith   PetscFunctionReturn(0);
534c1414c8SBarry Smith }
544c1414c8SBarry Smith 
554c1414c8SBarry Smith 
564c1414c8SBarry Smith /*
574c1414c8SBarry Smith       This builds symmetric version of nonzero structure,
584c1414c8SBarry Smith */
594c1414c8SBarry Smith #undef __FUNCT__
604108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode_Symmetric"
611a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Symmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift)
624c1414c8SBarry Smith {
634c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
644c1414c8SBarry Smith   PetscErrorCode ierr;
658758e1faSBarry Smith   PetscInt       *work,*ia,*ja,nz,nslim_row,nslim_col,m,row,col,n;
668758e1faSBarry Smith   PetscInt       *tns,*tvc,*ns_row = a->inode.size,*ns_col,nsz,i1,i2;
678758e1faSBarry Smith   const PetscInt *j,*jmax,*ai= a->i,*aj = a->j;
684c1414c8SBarry Smith 
694c1414c8SBarry Smith   PetscFunctionBegin;
704c1414c8SBarry Smith   nslim_row = a->inode.node_count;
71d0f46423SBarry Smith   m         = A->rmap->n;
72d0f46423SBarry Smith   n         = A->cmap->n;
73e32f2f54SBarry Smith   if (m != n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatGetRowIJ_SeqAIJ_Inode_Symmetric: Matrix should be square");
744c1414c8SBarry Smith 
754c1414c8SBarry Smith   /* Use the row_inode as column_inode */
764c1414c8SBarry Smith   nslim_col = nslim_row;
774c1414c8SBarry Smith   ns_col    = ns_row;
784c1414c8SBarry Smith 
794c1414c8SBarry Smith   /* allocate space for reformated inode structure */
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];
862205254eSKarl Rupp     for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1;
874c1414c8SBarry Smith   }
884c1414c8SBarry Smith   /* allocate space for row pointers */
894c1414c8SBarry Smith   ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&ia);CHKERRQ(ierr);
904c1414c8SBarry Smith   *iia = ia;
914c1414c8SBarry Smith   ierr = PetscMemzero(ia,(nslim_row+1)*sizeof(PetscInt));CHKERRQ(ierr);
924c1414c8SBarry Smith   ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&work);CHKERRQ(ierr);
934c1414c8SBarry Smith 
944c1414c8SBarry Smith   /* determine the number of columns in each row */
954c1414c8SBarry Smith   ia[0] = oshift;
964c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
974c1414c8SBarry Smith 
984c1414c8SBarry Smith     j    = aj + ai[row] + ishift;
994c1414c8SBarry Smith     jmax = aj + ai[row+1] + ishift;
1004c1414c8SBarry Smith     col  = *j++ + ishift;
1014c1414c8SBarry Smith     i2   = tvc[col];
1024c1414c8SBarry Smith     while (i2<i1 && j<jmax) { /* 1.[-xx-d-xx--] 2.[-xx-------],off-diagonal elemets */
1034c1414c8SBarry Smith       ia[i1+1]++;
1044c1414c8SBarry Smith       ia[i2+1]++;
1054c1414c8SBarry Smith       i2++;                     /* Start col of next node */
10690d2dec7SBarry Smith       while ((j<jmax) && ((col=*j+ishift)<tns[i2])) ++j;
1074c1414c8SBarry Smith       i2 = tvc[col];
1084c1414c8SBarry Smith     }
1094c1414c8SBarry Smith     if (i2 == i1) ia[i2+1]++;    /* now the diagonal element */
1104c1414c8SBarry Smith   }
1114c1414c8SBarry Smith 
1124c1414c8SBarry Smith   /* shift ia[i] to point to next row */
1134c1414c8SBarry Smith   for (i1=1; i1<nslim_row+1; i1++) {
1144c1414c8SBarry Smith     row        = ia[i1-1];
1154c1414c8SBarry Smith     ia[i1]    += row;
1164c1414c8SBarry Smith     work[i1-1] = row - oshift;
1174c1414c8SBarry Smith   }
1184c1414c8SBarry Smith 
1194c1414c8SBarry Smith   /* allocate space for column pointers */
1204c1414c8SBarry Smith   nz   = ia[nslim_row] + (!ishift);
1214c1414c8SBarry Smith   ierr = PetscMalloc(nz*sizeof(PetscInt),&ja);CHKERRQ(ierr);
1224c1414c8SBarry Smith   *jja = ja;
1234c1414c8SBarry Smith 
1244c1414c8SBarry Smith   /* loop over lower triangular part putting into ja */
1254c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row += ns_row[i1],i1++) {
1264c1414c8SBarry Smith     j    = aj + ai[row] + ishift;
1274c1414c8SBarry Smith     jmax = aj + ai[row+1] + ishift;
1284c1414c8SBarry Smith     col  = *j++ + ishift;
1294c1414c8SBarry Smith     i2   = tvc[col];
1304c1414c8SBarry Smith     while (i2<i1 && j<jmax) {
1314c1414c8SBarry Smith       ja[work[i2]++] = i1 + oshift;
1324c1414c8SBarry Smith       ja[work[i1]++] = i2 + oshift;
1334c1414c8SBarry Smith       ++i2;
13490d2dec7SBarry Smith       while ((j<jmax) && ((col=*j+ishift)< tns[i2])) ++j; /* Skip rest col indices in this node */
1354c1414c8SBarry Smith       i2 = tvc[col];
1364c1414c8SBarry Smith     }
1374c1414c8SBarry Smith     if (i2 == i1) ja[work[i1]++] = i2 + oshift;
1384c1414c8SBarry Smith 
1394c1414c8SBarry Smith   }
1404c1414c8SBarry Smith   ierr = PetscFree(work);CHKERRQ(ierr);
1414c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
1424c1414c8SBarry Smith   ierr = PetscFree(tvc);CHKERRQ(ierr);
1434c1414c8SBarry Smith   PetscFunctionReturn(0);
1444c1414c8SBarry Smith }
1454c1414c8SBarry Smith 
1464c1414c8SBarry Smith /*
1474c1414c8SBarry Smith       This builds nonsymmetric version of nonzero structure,
1484c1414c8SBarry Smith */
1494c1414c8SBarry Smith #undef __FUNCT__
1504108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric"
1511a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift)
1524c1414c8SBarry Smith {
1534c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
1544c1414c8SBarry Smith   PetscErrorCode ierr;
1558758e1faSBarry Smith   PetscInt       *work,*ia,*ja,nz,nslim_row,n,row,col,*ns_col,nslim_col;
1568758e1faSBarry Smith   PetscInt       *tns,*tvc,nsz,i1,i2;
1578758e1faSBarry Smith   const PetscInt *j,*ai= a->i,*aj = a->j,*ns_row = a->inode.size;
1584c1414c8SBarry Smith 
1594c1414c8SBarry Smith   PetscFunctionBegin;
1604c1414c8SBarry Smith   nslim_row = a->inode.node_count;
161d0f46423SBarry Smith   n         = A->cmap->n;
1624c1414c8SBarry Smith 
1634c1414c8SBarry Smith   /* Create The column_inode for this matrix */
1644c1414c8SBarry Smith   ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr);
1654c1414c8SBarry Smith 
1664c1414c8SBarry Smith   /* allocate space for reformated column_inode structure */
1674c1414c8SBarry Smith   ierr = PetscMalloc((nslim_col +1)*sizeof(PetscInt),&tns);CHKERRQ(ierr);
1684c1414c8SBarry Smith   ierr = PetscMalloc((n +1)*sizeof(PetscInt),&tvc);CHKERRQ(ierr);
1694c1414c8SBarry Smith   for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1];
1704c1414c8SBarry Smith 
1714c1414c8SBarry Smith   for (i1=0,col=0; i1<nslim_col; ++i1) {
1724c1414c8SBarry Smith     nsz = ns_col[i1];
1732205254eSKarl Rupp     for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1;
1744c1414c8SBarry Smith   }
1754c1414c8SBarry Smith   /* allocate space for row pointers */
1764c1414c8SBarry Smith   ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&ia);CHKERRQ(ierr);
1774c1414c8SBarry Smith   *iia = ia;
1784c1414c8SBarry Smith   ierr = PetscMemzero(ia,(nslim_row+1)*sizeof(PetscInt));CHKERRQ(ierr);
1794c1414c8SBarry Smith   ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&work);CHKERRQ(ierr);
1804c1414c8SBarry Smith 
1814c1414c8SBarry Smith   /* determine the number of columns in each row */
1824c1414c8SBarry Smith   ia[0] = oshift;
1834c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
1844c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
1854c1414c8SBarry Smith     col = *j++ + ishift;
1864c1414c8SBarry Smith     i2  = tvc[col];
1874c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
1884c1414c8SBarry Smith     while (nz-- > 0) {           /* off-diagonal elemets */
1894c1414c8SBarry Smith       ia[i1+1]++;
1904c1414c8SBarry Smith       i2++;                     /* Start col of next node */
191a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
1924c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
1934c1414c8SBarry Smith     }
1944c1414c8SBarry Smith   }
1954c1414c8SBarry Smith 
1964c1414c8SBarry Smith   /* shift ia[i] to point to next row */
1974c1414c8SBarry Smith   for (i1=1; i1<nslim_row+1; i1++) {
1984c1414c8SBarry Smith     row        = ia[i1-1];
1994c1414c8SBarry Smith     ia[i1]    += row;
2004c1414c8SBarry Smith     work[i1-1] = row - oshift;
2014c1414c8SBarry Smith   }
2024c1414c8SBarry Smith 
2034c1414c8SBarry Smith   /* allocate space for column pointers */
2044c1414c8SBarry Smith   nz   = ia[nslim_row] + (!ishift);
2054c1414c8SBarry Smith   ierr = PetscMalloc(nz*sizeof(PetscInt),&ja);CHKERRQ(ierr);
2064c1414c8SBarry Smith   *jja = ja;
2074c1414c8SBarry Smith 
2084c1414c8SBarry Smith   /* loop over matrix putting into ja */
2094c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
2104c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
2114c1414c8SBarry Smith     col = *j++ + ishift;
2124c1414c8SBarry Smith     i2  = tvc[col];
2134c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
2144c1414c8SBarry Smith     while (nz-- > 0) {
2154c1414c8SBarry Smith       ja[work[i1]++] = i2 + oshift;
2164c1414c8SBarry Smith       ++i2;
217a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
2184c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
2194c1414c8SBarry Smith     }
2204c1414c8SBarry Smith   }
2214c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
2224c1414c8SBarry Smith   ierr = PetscFree(work);CHKERRQ(ierr);
2234c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
2244c1414c8SBarry Smith   ierr = PetscFree(tvc);CHKERRQ(ierr);
2254c1414c8SBarry Smith   PetscFunctionReturn(0);
2264c1414c8SBarry Smith }
2274c1414c8SBarry Smith 
2284c1414c8SBarry Smith #undef __FUNCT__
2294108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode"
2301a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
2314c1414c8SBarry Smith {
2324c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
2334c1414c8SBarry Smith   PetscErrorCode ierr;
2344c1414c8SBarry Smith 
2354c1414c8SBarry Smith   PetscFunctionBegin;
2364c1414c8SBarry Smith   *n = a->inode.node_count;
2374c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
2388f7157efSSatish Balay   if (!blockcompressed) {
2398f7157efSSatish Balay     ierr = MatGetRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);;
2408f7157efSSatish Balay   } else if (symmetric) {
2414108e4d5SBarry Smith     ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
2424c1414c8SBarry Smith   } else {
2434108e4d5SBarry Smith     ierr = MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
2444c1414c8SBarry Smith   }
2454c1414c8SBarry Smith   PetscFunctionReturn(0);
2464c1414c8SBarry Smith }
2474c1414c8SBarry Smith 
2484c1414c8SBarry Smith #undef __FUNCT__
2494108e4d5SBarry Smith #define __FUNCT__ "MatRestoreRowIJ_SeqAIJ_Inode"
2501a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
2514c1414c8SBarry Smith {
2524c1414c8SBarry Smith   PetscErrorCode ierr;
2534c1414c8SBarry Smith 
2544c1414c8SBarry Smith   PetscFunctionBegin;
2554c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
2568f7157efSSatish Balay 
2578f7157efSSatish Balay   if (!blockcompressed) {
2588f7157efSSatish Balay     ierr = MatRestoreRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);;
2598f7157efSSatish Balay   } else {
2604c1414c8SBarry Smith     ierr = PetscFree(*ia);CHKERRQ(ierr);
2614c1414c8SBarry Smith     ierr = PetscFree(*ja);CHKERRQ(ierr);
2628f7157efSSatish Balay   }
2634c1414c8SBarry Smith   PetscFunctionReturn(0);
2644c1414c8SBarry Smith }
2654c1414c8SBarry Smith 
2664c1414c8SBarry Smith /* ----------------------------------------------------------- */
2674c1414c8SBarry Smith 
2684c1414c8SBarry Smith #undef __FUNCT__
2694108e4d5SBarry Smith #define __FUNCT__ "MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric"
2701a83f524SJed Brown static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift)
2714c1414c8SBarry Smith {
2724c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
2734c1414c8SBarry Smith   PetscErrorCode ierr;
2744c1414c8SBarry Smith   PetscInt       *work,*ia,*ja,*j,nz,nslim_row, n,row,col,*ns_col,nslim_col;
2754c1414c8SBarry Smith   PetscInt       *tns,*tvc,*ns_row = a->inode.size,nsz,i1,i2,*ai= a->i,*aj = a->j;
2764c1414c8SBarry Smith 
2774c1414c8SBarry Smith   PetscFunctionBegin;
2784c1414c8SBarry Smith   nslim_row = a->inode.node_count;
279d0f46423SBarry Smith   n         = A->cmap->n;
2804c1414c8SBarry Smith 
2814c1414c8SBarry Smith   /* Create The column_inode for this matrix */
2824c1414c8SBarry Smith   ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr);
2834c1414c8SBarry Smith 
2844c1414c8SBarry Smith   /* allocate space for reformated column_inode structure */
2854c1414c8SBarry Smith   ierr = PetscMalloc((nslim_col + 1)*sizeof(PetscInt),&tns);CHKERRQ(ierr);
2864c1414c8SBarry Smith   ierr = PetscMalloc((n + 1)*sizeof(PetscInt),&tvc);CHKERRQ(ierr);
2874c1414c8SBarry Smith   for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1];
2884c1414c8SBarry Smith 
2894c1414c8SBarry Smith   for (i1=0,col=0; i1<nslim_col; ++i1) {
2904c1414c8SBarry Smith     nsz = ns_col[i1];
2912205254eSKarl Rupp     for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1;
2924c1414c8SBarry Smith   }
2934c1414c8SBarry Smith   /* allocate space for column pointers */
2944c1414c8SBarry Smith   ierr = PetscMalloc((nslim_col+1)*sizeof(PetscInt),&ia);CHKERRQ(ierr);
2954c1414c8SBarry Smith   *iia = ia;
2964c1414c8SBarry Smith   ierr = PetscMemzero(ia,(nslim_col+1)*sizeof(PetscInt));CHKERRQ(ierr);
2974c1414c8SBarry Smith   ierr = PetscMalloc((nslim_col+1)*sizeof(PetscInt),&work);CHKERRQ(ierr);
2984c1414c8SBarry Smith 
2994c1414c8SBarry Smith   /* determine the number of columns in each row */
3004c1414c8SBarry Smith   ia[0] = oshift;
3014c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
3024c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
3034c1414c8SBarry Smith     col = *j++ + ishift;
3044c1414c8SBarry Smith     i2  = tvc[col];
3054c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
3064c1414c8SBarry Smith     while (nz-- > 0) {           /* off-diagonal elemets */
3074c1414c8SBarry Smith       /* ia[i1+1]++; */
3084c1414c8SBarry Smith       ia[i2+1]++;
3094c1414c8SBarry Smith       i2++;
310a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
3114c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
3124c1414c8SBarry Smith     }
3134c1414c8SBarry Smith   }
3144c1414c8SBarry Smith 
3154c1414c8SBarry Smith   /* shift ia[i] to point to next col */
3164c1414c8SBarry Smith   for (i1=1; i1<nslim_col+1; i1++) {
3174c1414c8SBarry Smith     col        = ia[i1-1];
3184c1414c8SBarry Smith     ia[i1]    += col;
3194c1414c8SBarry Smith     work[i1-1] = col - oshift;
3204c1414c8SBarry Smith   }
3214c1414c8SBarry Smith 
3224c1414c8SBarry Smith   /* allocate space for column pointers */
3234c1414c8SBarry Smith   nz   = ia[nslim_col] + (!ishift);
3244c1414c8SBarry Smith   ierr = PetscMalloc(nz*sizeof(PetscInt),&ja);CHKERRQ(ierr);
3254c1414c8SBarry Smith   *jja = ja;
3264c1414c8SBarry Smith 
3274c1414c8SBarry Smith   /* loop over matrix putting into ja */
3284c1414c8SBarry Smith   for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) {
3294c1414c8SBarry Smith     j   = aj + ai[row] + ishift;
3304c1414c8SBarry Smith     col = *j++ + ishift;
3314c1414c8SBarry Smith     i2  = tvc[col];
3324c1414c8SBarry Smith     nz  = ai[row+1] - ai[row];
3334c1414c8SBarry Smith     while (nz-- > 0) {
3344c1414c8SBarry Smith       /* ja[work[i1]++] = i2 + oshift; */
3354c1414c8SBarry Smith       ja[work[i2]++] = i1 + oshift;
3364c1414c8SBarry Smith       i2++;
337a8e3a797SJed Brown       while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--;
3384c1414c8SBarry Smith       if (nz > 0) i2 = tvc[col];
3394c1414c8SBarry Smith     }
3404c1414c8SBarry Smith   }
3414c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
3424c1414c8SBarry Smith   ierr = PetscFree(work);CHKERRQ(ierr);
3434c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
3444c1414c8SBarry Smith   ierr = PetscFree(tvc);CHKERRQ(ierr);
3454c1414c8SBarry Smith   PetscFunctionReturn(0);
3464c1414c8SBarry Smith }
3474c1414c8SBarry Smith 
3484c1414c8SBarry Smith #undef __FUNCT__
3494108e4d5SBarry Smith #define __FUNCT__ "MatGetColumnIJ_SeqAIJ_Inode"
3501a83f524SJed Brown static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
3514c1414c8SBarry Smith {
3524c1414c8SBarry Smith   PetscErrorCode ierr;
3534c1414c8SBarry Smith 
3544c1414c8SBarry Smith   PetscFunctionBegin;
3550298fd71SBarry Smith   ierr = Mat_CreateColInode(A,n,NULL);CHKERRQ(ierr);
3564c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
3574c1414c8SBarry Smith 
3588f7157efSSatish Balay   if (!blockcompressed) {
3598f7157efSSatish Balay     ierr = MatGetColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);;
3608f7157efSSatish Balay   } else if (symmetric) {
3614c1414c8SBarry Smith     /* Since the indices are symmetric it does'nt matter */
3624108e4d5SBarry Smith     ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
3634c1414c8SBarry Smith   } else {
3644108e4d5SBarry Smith     ierr = MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr);
3654c1414c8SBarry Smith   }
3664c1414c8SBarry Smith   PetscFunctionReturn(0);
3674c1414c8SBarry Smith }
3684c1414c8SBarry Smith 
3694c1414c8SBarry Smith #undef __FUNCT__
3704108e4d5SBarry Smith #define __FUNCT__ "MatRestoreColumnIJ_SeqAIJ_Inode"
3711a83f524SJed Brown static PetscErrorCode MatRestoreColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
3724c1414c8SBarry Smith {
3734c1414c8SBarry Smith   PetscErrorCode ierr;
3744c1414c8SBarry Smith 
3754c1414c8SBarry Smith   PetscFunctionBegin;
3764c1414c8SBarry Smith   if (!ia) PetscFunctionReturn(0);
3778f7157efSSatish Balay   if (!blockcompressed) {
3788f7157efSSatish Balay     ierr = MatRestoreColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);;
3798f7157efSSatish Balay   } else {
3804c1414c8SBarry Smith     ierr = PetscFree(*ia);CHKERRQ(ierr);
3814c1414c8SBarry Smith     ierr = PetscFree(*ja);CHKERRQ(ierr);
3828f7157efSSatish Balay   }
3834c1414c8SBarry Smith   PetscFunctionReturn(0);
3844c1414c8SBarry Smith }
3854c1414c8SBarry Smith 
3864c1414c8SBarry Smith /* ----------------------------------------------------------- */
3874c1414c8SBarry Smith 
3884c1414c8SBarry Smith #undef __FUNCT__
3894108e4d5SBarry Smith #define __FUNCT__ "MatMult_SeqAIJ_Inode"
3904108e4d5SBarry Smith static PetscErrorCode MatMult_SeqAIJ_Inode(Mat A,Vec xx,Vec yy)
3914c1414c8SBarry Smith {
3924c1414c8SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3934c1414c8SBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5,tmp0,tmp1;
394d9fead3dSBarry Smith   PetscScalar       *y;
395dd6ea824SBarry Smith   const PetscScalar *x;
396dd6ea824SBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5;
3974c1414c8SBarry Smith   PetscErrorCode    ierr;
3988758e1faSBarry Smith   PetscInt          i1,i2,n,i,row,node_max,nsz,sz,nonzerorow=0;
3998758e1faSBarry Smith   const PetscInt    *idx,*ns,*ii;
4004c1414c8SBarry Smith 
4014c1414c8SBarry Smith #if defined(PETSC_HAVE_PRAGMA_DISJOINT)
4024c1414c8SBarry Smith #pragma disjoint(*x,*y,*v1,*v2,*v3,*v4,*v5)
4034c1414c8SBarry Smith #endif
4044c1414c8SBarry Smith 
4054c1414c8SBarry Smith   PetscFunctionBegin;
406e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
4074c1414c8SBarry Smith   node_max = a->inode.node_count;
4084c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
4093649974fSBarry Smith   ierr     = VecGetArrayRead(xx,&x);CHKERRQ(ierr);
4104c1414c8SBarry Smith   ierr     = VecGetArray(yy,&y);CHKERRQ(ierr);
4114c1414c8SBarry Smith   idx      = a->j;
4124c1414c8SBarry Smith   v1       = a->a;
4134c1414c8SBarry Smith   ii       = a->i;
4144c1414c8SBarry Smith 
4154c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i) {
4164c1414c8SBarry Smith     nsz         = ns[i];
4174c1414c8SBarry Smith     n           = ii[1] - ii[0];
41898c9bda7SSatish Balay     nonzerorow += (n>0)*nsz;
4194c1414c8SBarry Smith     ii         += nsz;
42050d8bf02SJed Brown     PetscPrefetchBlock(idx+nsz*n,n,0,PETSC_PREFETCH_HINT_NTA);    /* Prefetch the indices for the block row after the current one */
42150d8bf02SJed Brown     PetscPrefetchBlock(v1+nsz*n,nsz*n,0,PETSC_PREFETCH_HINT_NTA); /* Prefetch the values for the block row after the current one  */
4224c1414c8SBarry Smith     sz = n;                     /* No of non zeros in this row */
4234c1414c8SBarry Smith                                 /* Switch on the size of Node */
4244c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
4254c1414c8SBarry Smith     case 1:
42675567043SBarry Smith       sum1 = 0.;
4274c1414c8SBarry Smith 
4284c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4294c1414c8SBarry Smith         i1    = idx[0];         /* The instructions are ordered to */
4304c1414c8SBarry Smith         i2    = idx[1];         /* make the compiler's job easy */
4314c1414c8SBarry Smith         idx  += 2;
4324c1414c8SBarry Smith         tmp0  = x[i1];
4334c1414c8SBarry Smith         tmp1  = x[i2];
4344c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4354c1414c8SBarry Smith       }
4364c1414c8SBarry Smith 
4374c1414c8SBarry Smith       if (n == sz-1) {          /* Take care of the last nonzero  */
4384c1414c8SBarry Smith         tmp0  = x[*idx++];
4394c1414c8SBarry Smith         sum1 += *v1++ *tmp0;
4404c1414c8SBarry Smith       }
4414c1414c8SBarry Smith       y[row++]=sum1;
4424c1414c8SBarry Smith       break;
4434c1414c8SBarry Smith     case 2:
44475567043SBarry Smith       sum1 = 0.;
44575567043SBarry Smith       sum2 = 0.;
4464c1414c8SBarry Smith       v2   = v1 + n;
4474c1414c8SBarry Smith 
4484c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4494c1414c8SBarry Smith         i1    = idx[0];
4504c1414c8SBarry Smith         i2    = idx[1];
4514c1414c8SBarry Smith         idx  += 2;
4524c1414c8SBarry Smith         tmp0  = x[i1];
4534c1414c8SBarry Smith         tmp1  = x[i2];
4544c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4554c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
4564c1414c8SBarry Smith       }
4574c1414c8SBarry Smith       if (n == sz-1) {
4584c1414c8SBarry Smith         tmp0  = x[*idx++];
4594c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
4604c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
4614c1414c8SBarry Smith       }
4624c1414c8SBarry Smith       y[row++]=sum1;
4634c1414c8SBarry Smith       y[row++]=sum2;
4644c1414c8SBarry Smith       v1      =v2;              /* Since the next block to be processed starts there*/
4654c1414c8SBarry Smith       idx    +=sz;
4664c1414c8SBarry Smith       break;
4674c1414c8SBarry Smith     case 3:
46875567043SBarry Smith       sum1 = 0.;
46975567043SBarry Smith       sum2 = 0.;
47075567043SBarry Smith       sum3 = 0.;
4714c1414c8SBarry Smith       v2   = v1 + n;
4724c1414c8SBarry Smith       v3   = v2 + n;
4734c1414c8SBarry Smith 
4744c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
4754c1414c8SBarry Smith         i1    = idx[0];
4764c1414c8SBarry Smith         i2    = idx[1];
4774c1414c8SBarry Smith         idx  += 2;
4784c1414c8SBarry Smith         tmp0  = x[i1];
4794c1414c8SBarry Smith         tmp1  = x[i2];
4804c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
4814c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
4824c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
4834c1414c8SBarry Smith       }
4844c1414c8SBarry Smith       if (n == sz-1) {
4854c1414c8SBarry Smith         tmp0  = x[*idx++];
4864c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
4874c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
4884c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
4894c1414c8SBarry Smith       }
4904c1414c8SBarry Smith       y[row++]=sum1;
4914c1414c8SBarry Smith       y[row++]=sum2;
4924c1414c8SBarry Smith       y[row++]=sum3;
4934c1414c8SBarry Smith       v1      =v3;              /* Since the next block to be processed starts there*/
4944c1414c8SBarry Smith       idx    +=2*sz;
4954c1414c8SBarry Smith       break;
4964c1414c8SBarry Smith     case 4:
49775567043SBarry Smith       sum1 = 0.;
49875567043SBarry Smith       sum2 = 0.;
49975567043SBarry Smith       sum3 = 0.;
50075567043SBarry Smith       sum4 = 0.;
5014c1414c8SBarry Smith       v2   = v1 + n;
5024c1414c8SBarry Smith       v3   = v2 + n;
5034c1414c8SBarry Smith       v4   = v3 + n;
5044c1414c8SBarry Smith 
5054c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
5064c1414c8SBarry Smith         i1    = idx[0];
5074c1414c8SBarry Smith         i2    = idx[1];
5084c1414c8SBarry Smith         idx  += 2;
5094c1414c8SBarry Smith         tmp0  = x[i1];
5104c1414c8SBarry Smith         tmp1  = x[i2];
5114c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
5124c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
5134c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
5144c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
5154c1414c8SBarry Smith       }
5164c1414c8SBarry Smith       if (n == sz-1) {
5174c1414c8SBarry Smith         tmp0  = x[*idx++];
5184c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
5194c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
5204c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
5214c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
5224c1414c8SBarry Smith       }
5234c1414c8SBarry Smith       y[row++]=sum1;
5244c1414c8SBarry Smith       y[row++]=sum2;
5254c1414c8SBarry Smith       y[row++]=sum3;
5264c1414c8SBarry Smith       y[row++]=sum4;
5274c1414c8SBarry Smith       v1      =v4;              /* Since the next block to be processed starts there*/
5284c1414c8SBarry Smith       idx    +=3*sz;
5294c1414c8SBarry Smith       break;
5304c1414c8SBarry Smith     case 5:
53175567043SBarry Smith       sum1 = 0.;
53275567043SBarry Smith       sum2 = 0.;
53375567043SBarry Smith       sum3 = 0.;
53475567043SBarry Smith       sum4 = 0.;
53575567043SBarry Smith       sum5 = 0.;
5364c1414c8SBarry Smith       v2   = v1 + n;
5374c1414c8SBarry Smith       v3   = v2 + n;
5384c1414c8SBarry Smith       v4   = v3 + n;
5394c1414c8SBarry Smith       v5   = v4 + n;
5404c1414c8SBarry Smith 
5414c1414c8SBarry Smith       for (n = 0; n<sz-1; n+=2) {
5424c1414c8SBarry Smith         i1    = idx[0];
5434c1414c8SBarry Smith         i2    = idx[1];
5444c1414c8SBarry Smith         idx  += 2;
5454c1414c8SBarry Smith         tmp0  = x[i1];
5464c1414c8SBarry Smith         tmp1  = x[i2];
5474c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
5484c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
5494c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
5504c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
5514c1414c8SBarry Smith         sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2;
5524c1414c8SBarry Smith       }
5534c1414c8SBarry Smith       if (n == sz-1) {
5544c1414c8SBarry Smith         tmp0  = x[*idx++];
5554c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
5564c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
5574c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
5584c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
5594c1414c8SBarry Smith         sum5 += *v5++ * tmp0;
5604c1414c8SBarry Smith       }
5614c1414c8SBarry Smith       y[row++]=sum1;
5624c1414c8SBarry Smith       y[row++]=sum2;
5634c1414c8SBarry Smith       y[row++]=sum3;
5644c1414c8SBarry Smith       y[row++]=sum4;
5654c1414c8SBarry Smith       y[row++]=sum5;
5664c1414c8SBarry Smith       v1      =v5;       /* Since the next block to be processed starts there */
5674c1414c8SBarry Smith       idx    +=4*sz;
5684c1414c8SBarry Smith       break;
5694c1414c8SBarry Smith     default:
570e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported");
5714c1414c8SBarry Smith     }
5724c1414c8SBarry Smith   }
5733649974fSBarry Smith   ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr);
5744c1414c8SBarry Smith   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
575dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr);
5764c1414c8SBarry Smith   PetscFunctionReturn(0);
5774c1414c8SBarry Smith }
5784c1414c8SBarry Smith /* ----------------------------------------------------------- */
5794108e4d5SBarry Smith /* Almost same code as the MatMult_SeqAIJ_Inode() */
5804c1414c8SBarry Smith #undef __FUNCT__
5814108e4d5SBarry Smith #define __FUNCT__ "MatMultAdd_SeqAIJ_Inode"
5824108e4d5SBarry Smith static PetscErrorCode MatMultAdd_SeqAIJ_Inode(Mat A,Vec xx,Vec zz,Vec yy)
5834c1414c8SBarry Smith {
5844c1414c8SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
5854c1414c8SBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5,tmp0,tmp1;
5868758e1faSBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5;
5878758e1faSBarry Smith   const PetscScalar *x;
5888758e1faSBarry Smith   PetscScalar       *y,*z,*zt;
5894c1414c8SBarry Smith   PetscErrorCode    ierr;
5908758e1faSBarry Smith   PetscInt          i1,i2,n,i,row,node_max,nsz,sz;
5918758e1faSBarry Smith   const PetscInt    *idx,*ns,*ii;
5924c1414c8SBarry Smith 
5934c1414c8SBarry Smith   PetscFunctionBegin;
594e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
5954c1414c8SBarry Smith   node_max = a->inode.node_count;
5964c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
5972205254eSKarl Rupp 
5988758e1faSBarry Smith   ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr);
5994c1414c8SBarry Smith   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
6004c1414c8SBarry Smith   if (zz != yy) {
6014c1414c8SBarry Smith     ierr = VecGetArray(zz,&z);CHKERRQ(ierr);
6024c1414c8SBarry Smith   } else {
6034c1414c8SBarry Smith     z = y;
6044c1414c8SBarry Smith   }
6054c1414c8SBarry Smith   zt = z;
6064c1414c8SBarry Smith 
6074c1414c8SBarry Smith   idx = a->j;
6084c1414c8SBarry Smith   v1  = a->a;
6094c1414c8SBarry Smith   ii  = a->i;
6104c1414c8SBarry Smith 
6114c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i) {
6124c1414c8SBarry Smith     nsz = ns[i];
6134c1414c8SBarry Smith     n   = ii[1] - ii[0];
6144c1414c8SBarry Smith     ii += nsz;
6154c1414c8SBarry Smith     sz  = n;                    /* No of non zeros in this row */
6164c1414c8SBarry Smith                                 /* Switch on the size of Node */
6174c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
6184c1414c8SBarry Smith     case 1:
6194c1414c8SBarry Smith       sum1 = *zt++;
6204c1414c8SBarry Smith 
6214c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6224c1414c8SBarry Smith         i1    = idx[0];         /* The instructions are ordered to */
6234c1414c8SBarry Smith         i2    = idx[1];         /* make the compiler's job easy */
6244c1414c8SBarry Smith         idx  += 2;
6254c1414c8SBarry Smith         tmp0  = x[i1];
6264c1414c8SBarry Smith         tmp1  = x[i2];
6274c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6284c1414c8SBarry Smith       }
6294c1414c8SBarry Smith 
6304c1414c8SBarry Smith       if (n   == sz-1) {          /* Take care of the last nonzero  */
6314c1414c8SBarry Smith         tmp0  = x[*idx++];
6324c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6334c1414c8SBarry Smith       }
6344c1414c8SBarry Smith       y[row++]=sum1;
6354c1414c8SBarry Smith       break;
6364c1414c8SBarry Smith     case 2:
6374c1414c8SBarry Smith       sum1 = *zt++;
6384c1414c8SBarry Smith       sum2 = *zt++;
6394c1414c8SBarry Smith       v2   = v1 + n;
6404c1414c8SBarry Smith 
6414c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6424c1414c8SBarry Smith         i1    = idx[0];
6434c1414c8SBarry Smith         i2    = idx[1];
6444c1414c8SBarry Smith         idx  += 2;
6454c1414c8SBarry Smith         tmp0  = x[i1];
6464c1414c8SBarry Smith         tmp1  = x[i2];
6474c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6484c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
6494c1414c8SBarry Smith       }
6504c1414c8SBarry Smith       if (n   == sz-1) {
6514c1414c8SBarry Smith         tmp0  = x[*idx++];
6524c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6534c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
6544c1414c8SBarry Smith       }
6554c1414c8SBarry Smith       y[row++]=sum1;
6564c1414c8SBarry Smith       y[row++]=sum2;
6574c1414c8SBarry Smith       v1      =v2;              /* Since the next block to be processed starts there*/
6584c1414c8SBarry Smith       idx    +=sz;
6594c1414c8SBarry Smith       break;
6604c1414c8SBarry Smith     case 3:
6614c1414c8SBarry Smith       sum1 = *zt++;
6624c1414c8SBarry Smith       sum2 = *zt++;
6634c1414c8SBarry Smith       sum3 = *zt++;
6644c1414c8SBarry Smith       v2   = v1 + n;
6654c1414c8SBarry Smith       v3   = v2 + n;
6664c1414c8SBarry Smith 
6674c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6684c1414c8SBarry Smith         i1    = idx[0];
6694c1414c8SBarry Smith         i2    = idx[1];
6704c1414c8SBarry Smith         idx  += 2;
6714c1414c8SBarry Smith         tmp0  = x[i1];
6724c1414c8SBarry Smith         tmp1  = x[i2];
6734c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
6744c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
6754c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
6764c1414c8SBarry Smith       }
6774c1414c8SBarry Smith       if (n == sz-1) {
6784c1414c8SBarry Smith         tmp0  = x[*idx++];
6794c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
6804c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
6814c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
6824c1414c8SBarry Smith       }
6834c1414c8SBarry Smith       y[row++]=sum1;
6844c1414c8SBarry Smith       y[row++]=sum2;
6854c1414c8SBarry Smith       y[row++]=sum3;
6864c1414c8SBarry Smith       v1      =v3;              /* Since the next block to be processed starts there*/
6874c1414c8SBarry Smith       idx    +=2*sz;
6884c1414c8SBarry Smith       break;
6894c1414c8SBarry Smith     case 4:
6904c1414c8SBarry Smith       sum1 = *zt++;
6914c1414c8SBarry Smith       sum2 = *zt++;
6924c1414c8SBarry Smith       sum3 = *zt++;
6934c1414c8SBarry Smith       sum4 = *zt++;
6944c1414c8SBarry Smith       v2   = v1 + n;
6954c1414c8SBarry Smith       v3   = v2 + n;
6964c1414c8SBarry Smith       v4   = v3 + n;
6974c1414c8SBarry Smith 
6984c1414c8SBarry Smith       for (n = 0; n< sz-1; n+=2) {
6994c1414c8SBarry Smith         i1    = idx[0];
7004c1414c8SBarry Smith         i2    = idx[1];
7014c1414c8SBarry Smith         idx  += 2;
7024c1414c8SBarry Smith         tmp0  = x[i1];
7034c1414c8SBarry Smith         tmp1  = x[i2];
7044c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
7054c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
7064c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
7074c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
7084c1414c8SBarry Smith       }
7094c1414c8SBarry Smith       if (n == sz-1) {
7104c1414c8SBarry Smith         tmp0  = x[*idx++];
7114c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
7124c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
7134c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
7144c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
7154c1414c8SBarry Smith       }
7164c1414c8SBarry Smith       y[row++]=sum1;
7174c1414c8SBarry Smith       y[row++]=sum2;
7184c1414c8SBarry Smith       y[row++]=sum3;
7194c1414c8SBarry Smith       y[row++]=sum4;
7204c1414c8SBarry Smith       v1      =v4;              /* Since the next block to be processed starts there*/
7214c1414c8SBarry Smith       idx    +=3*sz;
7224c1414c8SBarry Smith       break;
7234c1414c8SBarry Smith     case 5:
7244c1414c8SBarry Smith       sum1 = *zt++;
7254c1414c8SBarry Smith       sum2 = *zt++;
7264c1414c8SBarry Smith       sum3 = *zt++;
7274c1414c8SBarry Smith       sum4 = *zt++;
7284c1414c8SBarry Smith       sum5 = *zt++;
7294c1414c8SBarry Smith       v2   = v1 + n;
7304c1414c8SBarry Smith       v3   = v2 + n;
7314c1414c8SBarry Smith       v4   = v3 + n;
7324c1414c8SBarry Smith       v5   = v4 + n;
7334c1414c8SBarry Smith 
7344c1414c8SBarry Smith       for (n = 0; n<sz-1; n+=2) {
7354c1414c8SBarry Smith         i1    = idx[0];
7364c1414c8SBarry Smith         i2    = idx[1];
7374c1414c8SBarry Smith         idx  += 2;
7384c1414c8SBarry Smith         tmp0  = x[i1];
7394c1414c8SBarry Smith         tmp1  = x[i2];
7404c1414c8SBarry Smith         sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2;
7414c1414c8SBarry Smith         sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2;
7424c1414c8SBarry Smith         sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2;
7434c1414c8SBarry Smith         sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2;
7444c1414c8SBarry Smith         sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2;
7454c1414c8SBarry Smith       }
7464c1414c8SBarry Smith       if (n   == sz-1) {
7474c1414c8SBarry Smith         tmp0  = x[*idx++];
7484c1414c8SBarry Smith         sum1 += *v1++ * tmp0;
7494c1414c8SBarry Smith         sum2 += *v2++ * tmp0;
7504c1414c8SBarry Smith         sum3 += *v3++ * tmp0;
7514c1414c8SBarry Smith         sum4 += *v4++ * tmp0;
7524c1414c8SBarry Smith         sum5 += *v5++ * tmp0;
7534c1414c8SBarry Smith       }
7544c1414c8SBarry Smith       y[row++]=sum1;
7554c1414c8SBarry Smith       y[row++]=sum2;
7564c1414c8SBarry Smith       y[row++]=sum3;
7574c1414c8SBarry Smith       y[row++]=sum4;
7584c1414c8SBarry Smith       y[row++]=sum5;
7594c1414c8SBarry Smith       v1      =v5;       /* Since the next block to be processed starts there */
7604c1414c8SBarry Smith       idx    +=4*sz;
7614c1414c8SBarry Smith       break;
7624c1414c8SBarry Smith     default:
763e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported");
7644c1414c8SBarry Smith     }
7654c1414c8SBarry Smith   }
7668758e1faSBarry Smith   ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr);
7674c1414c8SBarry Smith   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
7684c1414c8SBarry Smith   if (zz != yy) {
7694c1414c8SBarry Smith     ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr);
7704c1414c8SBarry Smith   }
771dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
7724c1414c8SBarry Smith   PetscFunctionReturn(0);
7734c1414c8SBarry Smith }
7744c1414c8SBarry Smith 
7754c1414c8SBarry Smith /* ----------------------------------------------------------- */
7764c1414c8SBarry Smith #undef __FUNCT__
777ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_Inode_inplace"
778ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_Inode_inplace(Mat A,Vec bb,Vec xx)
7794c1414c8SBarry Smith {
7804c1414c8SBarry Smith   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
7814c1414c8SBarry Smith   IS                iscol = a->col,isrow = a->row;
7824c1414c8SBarry Smith   PetscErrorCode    ierr;
7835d0c19d7SBarry Smith   const PetscInt    *r,*c,*rout,*cout;
7848758e1faSBarry Smith   PetscInt          i,j,n = A->rmap->n,nz;
7858758e1faSBarry Smith   PetscInt          node_max,*ns,row,nsz,aii,i0,i1;
7868758e1faSBarry Smith   const PetscInt    *ai = a->i,*a_j = a->j,*vi,*ad,*aj;
787d9fead3dSBarry Smith   PetscScalar       *x,*tmp,*tmps,tmp0,tmp1;
788d9fead3dSBarry Smith   PetscScalar       sum1,sum2,sum3,sum4,sum5;
789dd6ea824SBarry Smith   const MatScalar   *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa;
790dd6ea824SBarry Smith   const PetscScalar *b;
7914c1414c8SBarry Smith 
7924c1414c8SBarry Smith   PetscFunctionBegin;
793e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
7944c1414c8SBarry Smith   node_max = a->inode.node_count;
7954c1414c8SBarry Smith   ns       = a->inode.size;     /* Node Size array */
7964c1414c8SBarry Smith 
7973649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
7984c1414c8SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
7994c1414c8SBarry Smith   tmp  = a->solve_work;
8004c1414c8SBarry Smith 
8014c1414c8SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
8024c1414c8SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
8034c1414c8SBarry Smith 
8044c1414c8SBarry Smith   /* forward solve the lower triangular */
8054c1414c8SBarry Smith   tmps = tmp;
8064c1414c8SBarry Smith   aa   = a_a;
8074c1414c8SBarry Smith   aj   = a_j;
8084c1414c8SBarry Smith   ad   = a->diag;
8094c1414c8SBarry Smith 
8104c1414c8SBarry Smith   for (i = 0,row = 0; i< node_max; ++i) {
8114c1414c8SBarry Smith     nsz = ns[i];
8124c1414c8SBarry Smith     aii = ai[row];
8134c1414c8SBarry Smith     v1  = aa + aii;
8144c1414c8SBarry Smith     vi  = aj + aii;
8154c1414c8SBarry Smith     nz  = ad[row]- aii;
81626549573SJed Brown     if (i < node_max-1) {
81726549573SJed Brown       /* Prefetch the block after the current one, the prefetch itself can't cause a memory error,
81826549573SJed Brown       * but our indexing to determine it's size could. */
81950d8bf02SJed Brown       PetscPrefetchBlock(aj+ai[row+nsz],ad[row+nsz]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); /* indices */
82026549573SJed Brown       /* In my tests, it seems to be better to fetch entire rows instead of just the lower-triangular part */
82150d8bf02SJed Brown       PetscPrefetchBlock(aa+ai[row+nsz],ad[row+nsz+ns[i+1]-1]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA);
82226549573SJed 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); */
82326549573SJed Brown     }
8244c1414c8SBarry Smith 
8254c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
8264c1414c8SBarry Smith     case 1:
8274c1414c8SBarry Smith       sum1 = b[*r++];
8284c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
8294c1414c8SBarry Smith         i0    = vi[0];
8304c1414c8SBarry Smith         i1    = vi[1];
8314c1414c8SBarry Smith         vi   +=2;
8324c1414c8SBarry Smith         tmp0  = tmps[i0];
8334c1414c8SBarry Smith         tmp1  = tmps[i1];
8344c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8354c1414c8SBarry Smith       }
8364c1414c8SBarry Smith       if (j == nz-1) {
8374c1414c8SBarry Smith         tmp0  = tmps[*vi++];
8384c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8394c1414c8SBarry Smith       }
8404c1414c8SBarry Smith       tmp[row++]=sum1;
8414c1414c8SBarry Smith       break;
8424c1414c8SBarry Smith     case 2:
8434c1414c8SBarry Smith       sum1 = b[*r++];
8444c1414c8SBarry Smith       sum2 = b[*r++];
8454c1414c8SBarry Smith       v2   = aa + ai[row+1];
8464c1414c8SBarry Smith 
8474c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
8484c1414c8SBarry Smith         i0    = vi[0];
8494c1414c8SBarry Smith         i1    = vi[1];
8504c1414c8SBarry Smith         vi   +=2;
8514c1414c8SBarry Smith         tmp0  = tmps[i0];
8524c1414c8SBarry Smith         tmp1  = tmps[i1];
8534c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8544c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
8554c1414c8SBarry Smith       }
8564c1414c8SBarry Smith       if (j == nz-1) {
8574c1414c8SBarry Smith         tmp0  = tmps[*vi++];
8584c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8594c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
8604c1414c8SBarry Smith       }
8614c1414c8SBarry Smith       sum2     -= *v2++ *sum1;
8624c1414c8SBarry Smith       tmp[row++]=sum1;
8634c1414c8SBarry Smith       tmp[row++]=sum2;
8644c1414c8SBarry Smith       break;
8654c1414c8SBarry Smith     case 3:
8664c1414c8SBarry Smith       sum1 = b[*r++];
8674c1414c8SBarry Smith       sum2 = b[*r++];
8684c1414c8SBarry Smith       sum3 = b[*r++];
8694c1414c8SBarry Smith       v2   = aa + ai[row+1];
8704c1414c8SBarry Smith       v3   = aa + ai[row+2];
8714c1414c8SBarry Smith 
8724c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
8734c1414c8SBarry Smith         i0    = vi[0];
8744c1414c8SBarry Smith         i1    = vi[1];
8754c1414c8SBarry Smith         vi   +=2;
8764c1414c8SBarry Smith         tmp0  = tmps[i0];
8774c1414c8SBarry Smith         tmp1  = tmps[i1];
8784c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
8794c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
8804c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
8814c1414c8SBarry Smith       }
8824c1414c8SBarry Smith       if (j == nz-1) {
8834c1414c8SBarry Smith         tmp0  = tmps[*vi++];
8844c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
8854c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
8864c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
8874c1414c8SBarry Smith       }
8884c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
8894c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
8904c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
8912205254eSKarl Rupp 
8924c1414c8SBarry Smith       tmp[row++]=sum1;
8934c1414c8SBarry Smith       tmp[row++]=sum2;
8944c1414c8SBarry Smith       tmp[row++]=sum3;
8954c1414c8SBarry Smith       break;
8964c1414c8SBarry Smith 
8974c1414c8SBarry Smith     case 4:
8984c1414c8SBarry Smith       sum1 = b[*r++];
8994c1414c8SBarry Smith       sum2 = b[*r++];
9004c1414c8SBarry Smith       sum3 = b[*r++];
9014c1414c8SBarry Smith       sum4 = b[*r++];
9024c1414c8SBarry Smith       v2   = aa + ai[row+1];
9034c1414c8SBarry Smith       v3   = aa + ai[row+2];
9044c1414c8SBarry Smith       v4   = aa + ai[row+3];
9054c1414c8SBarry Smith 
9064c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
9074c1414c8SBarry Smith         i0    = vi[0];
9084c1414c8SBarry Smith         i1    = vi[1];
9094c1414c8SBarry Smith         vi   +=2;
9104c1414c8SBarry Smith         tmp0  = tmps[i0];
9114c1414c8SBarry Smith         tmp1  = tmps[i1];
9124c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
9134c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
9144c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
9154c1414c8SBarry Smith         sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
9164c1414c8SBarry Smith       }
9174c1414c8SBarry Smith       if (j == nz-1) {
9184c1414c8SBarry Smith         tmp0  = tmps[*vi++];
9194c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
9204c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
9214c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
9224c1414c8SBarry Smith         sum4 -= *v4++ *tmp0;
9234c1414c8SBarry Smith       }
9244c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
9254c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
9264c1414c8SBarry Smith       sum4 -= *v4++ * sum1;
9274c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
9284c1414c8SBarry Smith       sum4 -= *v4++ * sum2;
9294c1414c8SBarry Smith       sum4 -= *v4++ * sum3;
9304c1414c8SBarry Smith 
9314c1414c8SBarry Smith       tmp[row++]=sum1;
9324c1414c8SBarry Smith       tmp[row++]=sum2;
9334c1414c8SBarry Smith       tmp[row++]=sum3;
9344c1414c8SBarry Smith       tmp[row++]=sum4;
9354c1414c8SBarry Smith       break;
9364c1414c8SBarry Smith     case 5:
9374c1414c8SBarry Smith       sum1 = b[*r++];
9384c1414c8SBarry Smith       sum2 = b[*r++];
9394c1414c8SBarry Smith       sum3 = b[*r++];
9404c1414c8SBarry Smith       sum4 = b[*r++];
9414c1414c8SBarry Smith       sum5 = b[*r++];
9424c1414c8SBarry Smith       v2   = aa + ai[row+1];
9434c1414c8SBarry Smith       v3   = aa + ai[row+2];
9444c1414c8SBarry Smith       v4   = aa + ai[row+3];
9454c1414c8SBarry Smith       v5   = aa + ai[row+4];
9464c1414c8SBarry Smith 
9474c1414c8SBarry Smith       for (j=0; j<nz-1; j+=2) {
9484c1414c8SBarry Smith         i0    = vi[0];
9494c1414c8SBarry Smith         i1    = vi[1];
9504c1414c8SBarry Smith         vi   +=2;
9514c1414c8SBarry Smith         tmp0  = tmps[i0];
9524c1414c8SBarry Smith         tmp1  = tmps[i1];
9534c1414c8SBarry Smith         sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
9544c1414c8SBarry Smith         sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
9554c1414c8SBarry Smith         sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
9564c1414c8SBarry Smith         sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
9574c1414c8SBarry Smith         sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
9584c1414c8SBarry Smith       }
9594c1414c8SBarry Smith       if (j == nz-1) {
9604c1414c8SBarry Smith         tmp0  = tmps[*vi++];
9614c1414c8SBarry Smith         sum1 -= *v1++ *tmp0;
9624c1414c8SBarry Smith         sum2 -= *v2++ *tmp0;
9634c1414c8SBarry Smith         sum3 -= *v3++ *tmp0;
9644c1414c8SBarry Smith         sum4 -= *v4++ *tmp0;
9654c1414c8SBarry Smith         sum5 -= *v5++ *tmp0;
9664c1414c8SBarry Smith       }
9674c1414c8SBarry Smith 
9684c1414c8SBarry Smith       sum2 -= *v2++ * sum1;
9694c1414c8SBarry Smith       sum3 -= *v3++ * sum1;
9704c1414c8SBarry Smith       sum4 -= *v4++ * sum1;
9714c1414c8SBarry Smith       sum5 -= *v5++ * sum1;
9724c1414c8SBarry Smith       sum3 -= *v3++ * sum2;
9734c1414c8SBarry Smith       sum4 -= *v4++ * sum2;
9744c1414c8SBarry Smith       sum5 -= *v5++ * sum2;
9754c1414c8SBarry Smith       sum4 -= *v4++ * sum3;
9764c1414c8SBarry Smith       sum5 -= *v5++ * sum3;
9774c1414c8SBarry Smith       sum5 -= *v5++ * sum4;
9784c1414c8SBarry Smith 
9794c1414c8SBarry Smith       tmp[row++]=sum1;
9804c1414c8SBarry Smith       tmp[row++]=sum2;
9814c1414c8SBarry Smith       tmp[row++]=sum3;
9824c1414c8SBarry Smith       tmp[row++]=sum4;
9834c1414c8SBarry Smith       tmp[row++]=sum5;
9844c1414c8SBarry Smith       break;
9854c1414c8SBarry Smith     default:
986e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
9874c1414c8SBarry Smith     }
9884c1414c8SBarry Smith   }
9894c1414c8SBarry Smith   /* backward solve the upper triangular */
9904c1414c8SBarry Smith   for (i=node_max -1,row = n-1; i>=0; i--) {
9914c1414c8SBarry Smith     nsz = ns[i];
9924c1414c8SBarry Smith     aii = ai[row+1] -1;
9934c1414c8SBarry Smith     v1  = aa + aii;
9944c1414c8SBarry Smith     vi  = aj + aii;
9954c1414c8SBarry Smith     nz  = aii- ad[row];
9964c1414c8SBarry Smith     switch (nsz) {               /* Each loop in 'case' is unrolled */
9974c1414c8SBarry Smith     case 1:
9984c1414c8SBarry Smith       sum1 = tmp[row];
9994c1414c8SBarry Smith 
10004c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
10014c1414c8SBarry Smith         vi   -=2;
10024c1414c8SBarry Smith         i0    = vi[2];
10034c1414c8SBarry Smith         i1    = vi[1];
10044c1414c8SBarry Smith         tmp0  = tmps[i0];
10054c1414c8SBarry Smith         tmp1  = tmps[i1];
10064c1414c8SBarry Smith         v1   -= 2;
10074c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10084c1414c8SBarry Smith       }
10094c1414c8SBarry Smith       if (j==1) {
10104c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10114c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10124c1414c8SBarry Smith       }
10134c1414c8SBarry Smith       x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10144c1414c8SBarry Smith       break;
10154c1414c8SBarry Smith     case 2:
10164c1414c8SBarry Smith       sum1 = tmp[row];
10174c1414c8SBarry Smith       sum2 = tmp[row -1];
10184c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10194c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
10204c1414c8SBarry Smith         vi   -=2;
10214c1414c8SBarry Smith         i0    = vi[2];
10224c1414c8SBarry Smith         i1    = vi[1];
10234c1414c8SBarry Smith         tmp0  = tmps[i0];
10244c1414c8SBarry Smith         tmp1  = tmps[i1];
10254c1414c8SBarry Smith         v1   -= 2;
10264c1414c8SBarry Smith         v2   -= 2;
10274c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10284c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10294c1414c8SBarry Smith       }
10304c1414c8SBarry Smith       if (j==1) {
10314c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10324c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10334c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10344c1414c8SBarry Smith       }
10354c1414c8SBarry Smith 
10364c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10374c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10384c1414c8SBarry Smith       x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
10394c1414c8SBarry Smith       break;
10404c1414c8SBarry Smith     case 3:
10414c1414c8SBarry Smith       sum1 = tmp[row];
10424c1414c8SBarry Smith       sum2 = tmp[row -1];
10434c1414c8SBarry Smith       sum3 = tmp[row -2];
10444c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10454c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
10464c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
10474c1414c8SBarry Smith         vi   -=2;
10484c1414c8SBarry Smith         i0    = vi[2];
10494c1414c8SBarry Smith         i1    = vi[1];
10504c1414c8SBarry Smith         tmp0  = tmps[i0];
10514c1414c8SBarry Smith         tmp1  = tmps[i1];
10524c1414c8SBarry Smith         v1   -= 2;
10534c1414c8SBarry Smith         v2   -= 2;
10544c1414c8SBarry Smith         v3   -= 2;
10554c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10564c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10574c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
10584c1414c8SBarry Smith       }
10594c1414c8SBarry Smith       if (j==1) {
10604c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10614c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
10624c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
10634c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
10644c1414c8SBarry Smith       }
10654c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
10664c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
10674c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10684c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
10694c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
10704c1414c8SBarry Smith       x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
10714c1414c8SBarry Smith 
10724c1414c8SBarry Smith       break;
10734c1414c8SBarry Smith     case 4:
10744c1414c8SBarry Smith       sum1 = tmp[row];
10754c1414c8SBarry Smith       sum2 = tmp[row -1];
10764c1414c8SBarry Smith       sum3 = tmp[row -2];
10774c1414c8SBarry Smith       sum4 = tmp[row -3];
10784c1414c8SBarry Smith       v2   = aa + ai[row]-1;
10794c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
10804c1414c8SBarry Smith       v4   = aa + ai[row -2]-1;
10814c1414c8SBarry Smith 
10824c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
10834c1414c8SBarry Smith         vi   -=2;
10844c1414c8SBarry Smith         i0    = vi[2];
10854c1414c8SBarry Smith         i1    = vi[1];
10864c1414c8SBarry Smith         tmp0  = tmps[i0];
10874c1414c8SBarry Smith         tmp1  = tmps[i1];
10884c1414c8SBarry Smith         v1   -= 2;
10894c1414c8SBarry Smith         v2   -= 2;
10904c1414c8SBarry Smith         v3   -= 2;
10914c1414c8SBarry Smith         v4   -= 2;
10924c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
10934c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
10944c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
10954c1414c8SBarry Smith         sum4 -= v4[2] * tmp0 + v4[1] * tmp1;
10964c1414c8SBarry Smith       }
10974c1414c8SBarry Smith       if (j==1) {
10984c1414c8SBarry Smith         tmp0  = tmps[*vi--];
10994c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
11004c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
11014c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
11024c1414c8SBarry Smith         sum4 -= *v4-- * tmp0;
11034c1414c8SBarry Smith       }
11044c1414c8SBarry Smith 
11054c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
11064c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
11074c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11084c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11094c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
11104c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11114c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11124c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
11134c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11144c1414c8SBarry Smith       x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--;
11154c1414c8SBarry Smith       break;
11164c1414c8SBarry Smith     case 5:
11174c1414c8SBarry Smith       sum1 = tmp[row];
11184c1414c8SBarry Smith       sum2 = tmp[row -1];
11194c1414c8SBarry Smith       sum3 = tmp[row -2];
11204c1414c8SBarry Smith       sum4 = tmp[row -3];
11214c1414c8SBarry Smith       sum5 = tmp[row -4];
11224c1414c8SBarry Smith       v2   = aa + ai[row]-1;
11234c1414c8SBarry Smith       v3   = aa + ai[row -1]-1;
11244c1414c8SBarry Smith       v4   = aa + ai[row -2]-1;
11254c1414c8SBarry Smith       v5   = aa + ai[row -3]-1;
11264c1414c8SBarry Smith       for (j=nz; j>1; j-=2) {
11274c1414c8SBarry Smith         vi   -= 2;
11284c1414c8SBarry Smith         i0    = vi[2];
11294c1414c8SBarry Smith         i1    = vi[1];
11304c1414c8SBarry Smith         tmp0  = tmps[i0];
11314c1414c8SBarry Smith         tmp1  = tmps[i1];
11324c1414c8SBarry Smith         v1   -= 2;
11334c1414c8SBarry Smith         v2   -= 2;
11344c1414c8SBarry Smith         v3   -= 2;
11354c1414c8SBarry Smith         v4   -= 2;
11364c1414c8SBarry Smith         v5   -= 2;
11374c1414c8SBarry Smith         sum1 -= v1[2] * tmp0 + v1[1] * tmp1;
11384c1414c8SBarry Smith         sum2 -= v2[2] * tmp0 + v2[1] * tmp1;
11394c1414c8SBarry Smith         sum3 -= v3[2] * tmp0 + v3[1] * tmp1;
11404c1414c8SBarry Smith         sum4 -= v4[2] * tmp0 + v4[1] * tmp1;
11414c1414c8SBarry Smith         sum5 -= v5[2] * tmp0 + v5[1] * tmp1;
11424c1414c8SBarry Smith       }
11434c1414c8SBarry Smith       if (j==1) {
11444c1414c8SBarry Smith         tmp0  = tmps[*vi--];
11454c1414c8SBarry Smith         sum1 -= *v1-- * tmp0;
11464c1414c8SBarry Smith         sum2 -= *v2-- * tmp0;
11474c1414c8SBarry Smith         sum3 -= *v3-- * tmp0;
11484c1414c8SBarry Smith         sum4 -= *v4-- * tmp0;
11494c1414c8SBarry Smith         sum5 -= *v5-- * tmp0;
11504c1414c8SBarry Smith       }
11514c1414c8SBarry Smith 
11524c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--;
11534c1414c8SBarry Smith       sum2   -= *v2-- * tmp0;
11544c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11554c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11564c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11574c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--;
11584c1414c8SBarry Smith       sum3   -= *v3-- * tmp0;
11594c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11604c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11614c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--;
11624c1414c8SBarry Smith       sum4   -= *v4-- * tmp0;
11634c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11644c1414c8SBarry Smith       tmp0    = x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--;
11654c1414c8SBarry Smith       sum5   -= *v5-- * tmp0;
11664c1414c8SBarry Smith       x[*c--] = tmp[row] = sum5*a_a[ad[row]]; row--;
11674c1414c8SBarry Smith       break;
11684c1414c8SBarry Smith     default:
1169e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
11704c1414c8SBarry Smith     }
11714c1414c8SBarry Smith   }
11724c1414c8SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11734c1414c8SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
11743649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
11754c1414c8SBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1176dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
11774c1414c8SBarry Smith   PetscFunctionReturn(0);
11784c1414c8SBarry Smith }
11794c1414c8SBarry Smith 
118028f1b45aSHong Zhang #undef __FUNCT__
118128f1b45aSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_Inode"
118228f1b45aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_Inode(Mat B,Mat A,const MatFactorInfo *info)
118328f1b45aSHong Zhang {
118428f1b45aSHong Zhang   Mat             C     =B;
118528f1b45aSHong Zhang   Mat_SeqAIJ      *a    =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data;
118628f1b45aSHong Zhang   IS              isrow = b->row,isicol = b->icol;
118728f1b45aSHong Zhang   PetscErrorCode  ierr;
118828f1b45aSHong Zhang   const PetscInt  *r,*ic,*ics;
118928f1b45aSHong Zhang   const PetscInt  n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag;
119028f1b45aSHong Zhang   PetscInt        i,j,k,nz,nzL,row,*pj;
119128f1b45aSHong Zhang   const PetscInt  *ajtmp,*bjtmp;
11929877982aSShri Abhyankar   MatScalar       *pc,*pc1,*pc2,*pc3,*pc4,mul1,mul2,mul3,mul4,*pv,*rtmp1,*rtmp2,*rtmp3,*rtmp4;
11939877982aSShri Abhyankar   const MatScalar *aa=a->a,*v,*v1,*v2,*v3,*v4;
119428f1b45aSHong Zhang   FactorShiftCtx  sctx;
11954f81c4b7SBarry Smith   const PetscInt  *ddiag;
119628f1b45aSHong Zhang   PetscReal       rs;
119728f1b45aSHong Zhang   MatScalar       d;
11984f81c4b7SBarry Smith   PetscInt        inod,nodesz,node_max,col;
11994f81c4b7SBarry Smith   const PetscInt  *ns;
120007b50cabSHong Zhang   PetscInt        *tmp_vec1,*tmp_vec2,*nsmap;
12010e95ead3SHong Zhang 
120228f1b45aSHong Zhang   PetscFunctionBegin;
120328f1b45aSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
120428f1b45aSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
120528f1b45aSHong Zhang 
1206f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
120728f1b45aSHong Zhang     ddiag          = a->diag;
120828f1b45aSHong Zhang     sctx.shift_top = info->zeropivot;
120928f1b45aSHong Zhang     for (i=0; i<n; i++) {
121028f1b45aSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
121128f1b45aSHong Zhang       d  = (aa)[ddiag[i]];
121228f1b45aSHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
121328f1b45aSHong Zhang       v  = aa+ai[i];
121428f1b45aSHong Zhang       nz = ai[i+1] - ai[i];
12152205254eSKarl Rupp       for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]);
121628f1b45aSHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
121728f1b45aSHong Zhang     }
121828f1b45aSHong Zhang     sctx.shift_top *= 1.1;
121928f1b45aSHong Zhang     sctx.nshift_max = 5;
122028f1b45aSHong Zhang     sctx.shift_lo   = 0.;
122128f1b45aSHong Zhang     sctx.shift_hi   = 1.;
122228f1b45aSHong Zhang   }
122328f1b45aSHong Zhang 
122428f1b45aSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
122528f1b45aSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
122668785679SHong Zhang 
1227*dcca6d9dSJed Brown   ierr  = PetscMalloc4(n,&rtmp1,n,&rtmp2,n,&rtmp3,n,&rtmp4);CHKERRQ(ierr);
1228eb9baa12SBarry Smith   ierr  = PetscMemzero(rtmp1,n*sizeof(PetscScalar));CHKERRQ(ierr);
1229eb9baa12SBarry Smith   ierr  = PetscMemzero(rtmp2,n*sizeof(PetscScalar));CHKERRQ(ierr);
1230eb9baa12SBarry Smith   ierr  = PetscMemzero(rtmp3,n*sizeof(PetscScalar));CHKERRQ(ierr);
1231eb9baa12SBarry Smith   ierr  = PetscMemzero(rtmp4,n*sizeof(PetscScalar));CHKERRQ(ierr);
123228f1b45aSHong Zhang   ics   = ic;
123328f1b45aSHong Zhang 
123428f1b45aSHong Zhang   node_max = a->inode.node_count;
123528f1b45aSHong Zhang   ns       = a->inode.size;
1236e7e72b3dSBarry Smith   if (!ns) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix without inode information");
123728f1b45aSHong Zhang 
12389877982aSShri Abhyankar   /* If max inode size > 4, split it into two inodes.*/
123968785679SHong Zhang   /* also map the inode sizes according to the ordering */
124068785679SHong Zhang   ierr = PetscMalloc((n+1)* sizeof(PetscInt),&tmp_vec1);CHKERRQ(ierr);
124168785679SHong Zhang   for (i=0,j=0; i<node_max; ++i,++j) {
1242b1550197SShri Abhyankar     if (ns[i] > 4) {
1243048b5e81SShri Abhyankar       tmp_vec1[j] = 4;
124468785679SHong Zhang       ++j;
124568785679SHong Zhang       tmp_vec1[j] = ns[i] - tmp_vec1[j-1];
124668785679SHong Zhang     } else {
124768785679SHong Zhang       tmp_vec1[j] = ns[i];
124868785679SHong Zhang     }
124968785679SHong Zhang   }
125068785679SHong Zhang   /* Use the correct node_max */
125168785679SHong Zhang   node_max = j;
125268785679SHong Zhang 
125368785679SHong Zhang   /* Now reorder the inode info based on mat re-ordering info */
125468785679SHong Zhang   /* First create a row -> inode_size_array_index map */
125568785679SHong Zhang   ierr = PetscMalloc(n*sizeof(PetscInt)+1,&nsmap);CHKERRQ(ierr);
125668785679SHong Zhang   ierr = PetscMalloc(node_max*sizeof(PetscInt)+1,&tmp_vec2);CHKERRQ(ierr);
125768785679SHong Zhang   for (i=0,row=0; i<node_max; i++) {
125868785679SHong Zhang     nodesz = tmp_vec1[i];
125968785679SHong Zhang     for (j=0; j<nodesz; j++,row++) {
126068785679SHong Zhang       nsmap[row] = i;
126168785679SHong Zhang     }
126268785679SHong Zhang   }
126368785679SHong Zhang   /* Using nsmap, create a reordered ns structure */
126468785679SHong Zhang   for (i=0,j=0; i< node_max; i++) {
126568785679SHong Zhang     nodesz      = tmp_vec1[nsmap[r[j]]];     /* here the reordered row_no is in r[] */
126668785679SHong Zhang     tmp_vec2[i] = nodesz;
126768785679SHong Zhang     j          += nodesz;
126868785679SHong Zhang   }
126968785679SHong Zhang   ierr = PetscFree(nsmap);CHKERRQ(ierr);
127068785679SHong Zhang   ierr = PetscFree(tmp_vec1);CHKERRQ(ierr);
1271b89f182dSHong Zhang 
127268785679SHong Zhang   /* Now use the correct ns */
127368785679SHong Zhang   ns = tmp_vec2;
127468785679SHong Zhang 
127528f1b45aSHong Zhang   do {
127607b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
127728f1b45aSHong Zhang     /* Now loop over each block-row, and do the factorization */
127828f1b45aSHong Zhang     for (inod=0,i=0; inod<node_max; inod++) { /* i: row index; inod: inode index */
127928f1b45aSHong Zhang       nodesz = ns[inod];
128028f1b45aSHong Zhang 
128128f1b45aSHong Zhang       switch (nodesz) {
128228f1b45aSHong Zhang       case 1:
128368785679SHong Zhang         /*----------*/
1284b89f182dSHong Zhang         /* zero rtmp1 */
128528f1b45aSHong Zhang         /* L part */
128628f1b45aSHong Zhang         nz    = bi[i+1] - bi[i];
128728f1b45aSHong Zhang         bjtmp = bj + bi[i];
1288b89f182dSHong Zhang         for (j=0; j<nz; j++) rtmp1[bjtmp[j]] = 0.0;
128928f1b45aSHong Zhang 
129028f1b45aSHong Zhang         /* U part */
129128f1b45aSHong Zhang         nz    = bdiag[i]-bdiag[i+1];
129228f1b45aSHong Zhang         bjtmp = bj + bdiag[i+1]+1;
1293b89f182dSHong Zhang         for (j=0; j<nz; j++) rtmp1[bjtmp[j]] = 0.0;
129428f1b45aSHong Zhang 
129528f1b45aSHong Zhang         /* load in initial (unfactored row) */
129628f1b45aSHong Zhang         nz    = ai[r[i]+1] - ai[r[i]];
129728f1b45aSHong Zhang         ajtmp = aj + ai[r[i]];
129828f1b45aSHong Zhang         v     = aa + ai[r[i]];
12992205254eSKarl Rupp         for (j=0; j<nz; j++) rtmp1[ics[ajtmp[j]]] = v[j];
13002205254eSKarl Rupp 
130128f1b45aSHong Zhang         /* ZeropivotApply() */
1302b89f182dSHong Zhang         rtmp1[i] += sctx.shift_amount;  /* shift the diagonal of the matrix */
130328f1b45aSHong Zhang 
130428f1b45aSHong Zhang         /* elimination */
130528f1b45aSHong Zhang         bjtmp = bj + bi[i];
130628f1b45aSHong Zhang         row   = *bjtmp++;
130728f1b45aSHong Zhang         nzL   = bi[i+1] - bi[i];
130828f1b45aSHong Zhang         for (k=0; k < nzL; k++) {
1309b89f182dSHong Zhang           pc = rtmp1 + row;
131028f1b45aSHong Zhang           if (*pc != 0.0) {
131128f1b45aSHong Zhang             pv   = b->a + bdiag[row];
1312b89f182dSHong Zhang             mul1 = *pc * (*pv);
1313b89f182dSHong Zhang             *pc  = mul1;
131428f1b45aSHong Zhang             pj   = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
131528f1b45aSHong Zhang             pv   = b->a + bdiag[row+1]+1;
131628f1b45aSHong Zhang             nz   = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
1317b89f182dSHong Zhang             for (j=0; j<nz; j++) rtmp1[pj[j]] -= mul1 * pv[j];
131828b1a77fSLisandro Dalcin             ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
131928f1b45aSHong Zhang           }
132028f1b45aSHong Zhang           row = *bjtmp++;
132128f1b45aSHong Zhang         }
132228f1b45aSHong Zhang 
132328f1b45aSHong Zhang         /* finished row so stick it into b->a */
132428f1b45aSHong Zhang         rs = 0.0;
132528f1b45aSHong Zhang         /* L part */
132628f1b45aSHong Zhang         pv = b->a + bi[i];
132728f1b45aSHong Zhang         pj = b->j + bi[i];
132828f1b45aSHong Zhang         nz = bi[i+1] - bi[i];
132928f1b45aSHong Zhang         for (j=0; j<nz; j++) {
1330b89f182dSHong Zhang           pv[j] = rtmp1[pj[j]]; rs += PetscAbsScalar(pv[j]);
133128f1b45aSHong Zhang         }
133228f1b45aSHong Zhang 
133328f1b45aSHong Zhang         /* U part */
133428f1b45aSHong Zhang         pv = b->a + bdiag[i+1]+1;
133528f1b45aSHong Zhang         pj = b->j + bdiag[i+1]+1;
133628f1b45aSHong Zhang         nz = bdiag[i] - bdiag[i+1]-1;
133728f1b45aSHong Zhang         for (j=0; j<nz; j++) {
1338b89f182dSHong Zhang           pv[j] = rtmp1[pj[j]]; rs += PetscAbsScalar(pv[j]);
133928f1b45aSHong Zhang         }
134028f1b45aSHong Zhang 
1341b89f182dSHong Zhang         /* Check zero pivot */
134228f1b45aSHong Zhang         sctx.rs = rs;
1343b89f182dSHong Zhang         sctx.pv = rtmp1[i];
1344d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
134507b50cabSHong Zhang         if (sctx.newshift) break;
134628f1b45aSHong Zhang 
134728f1b45aSHong Zhang         /* Mark diagonal and invert diagonal for simplier triangular solves */
134828f1b45aSHong Zhang         pv  = b->a + bdiag[i];
1349b89f182dSHong Zhang         *pv = 1.0/sctx.pv; /* sctx.pv = rtmp1[i]+shiftamount if shifttype==MAT_SHIFT_INBLOCKS */
135028f1b45aSHong Zhang         break;
135128f1b45aSHong Zhang 
135228f1b45aSHong Zhang       case 2:
135328f1b45aSHong Zhang         /*----------*/
1354b89f182dSHong Zhang         /* zero rtmp1 and rtmp2 */
135528f1b45aSHong Zhang         /* L part */
135628f1b45aSHong Zhang         nz    = bi[i+1] - bi[i];
135728f1b45aSHong Zhang         bjtmp = bj + bi[i];
135828f1b45aSHong Zhang         for  (j=0; j<nz; j++) {
135968785679SHong Zhang           col        = bjtmp[j];
1360b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0;
136128f1b45aSHong Zhang         }
136228f1b45aSHong Zhang 
136328f1b45aSHong Zhang         /* U part */
136428f1b45aSHong Zhang         nz    = bdiag[i]-bdiag[i+1];
136528f1b45aSHong Zhang         bjtmp = bj + bdiag[i+1]+1;
136628f1b45aSHong Zhang         for  (j=0; j<nz; j++) {
136768785679SHong Zhang           col        = bjtmp[j];
1368b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0;
136928f1b45aSHong Zhang         }
137028f1b45aSHong Zhang 
137128f1b45aSHong Zhang         /* load in initial (unfactored row) */
137228f1b45aSHong Zhang         nz    = ai[r[i]+1] - ai[r[i]];
137328f1b45aSHong Zhang         ajtmp = aj + ai[r[i]];
137428f1b45aSHong Zhang         v1    = aa + ai[r[i]]; v2 = aa + ai[r[i]+1];
137528f1b45aSHong Zhang         for (j=0; j<nz; j++) {
137668785679SHong Zhang           col        = ics[ajtmp[j]];
1377b89f182dSHong Zhang           rtmp1[col] = v1[j]; rtmp2[col] = v2[j];
137828f1b45aSHong Zhang         }
137928f1b45aSHong Zhang         /* ZeropivotApply(): shift the diagonal of the matrix  */
1380b89f182dSHong Zhang         rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount;
138128f1b45aSHong Zhang 
138228f1b45aSHong Zhang         /* elimination */
138328f1b45aSHong Zhang         bjtmp = bj + bi[i];
138428f1b45aSHong Zhang         row   = *bjtmp++; /* pivot row */
138528f1b45aSHong Zhang         nzL   = bi[i+1] - bi[i];
138628f1b45aSHong Zhang         for (k=0; k < nzL; k++) {
1387b89f182dSHong Zhang           pc1 = rtmp1 + row;
1388b89f182dSHong Zhang           pc2 = rtmp2 + row;
138928f1b45aSHong Zhang           if (*pc1 != 0.0 || *pc2 != 0.0) {
139028f1b45aSHong Zhang             pv   = b->a + bdiag[row];
1391b89f182dSHong Zhang             mul1 = *pc1*(*pv); mul2 = *pc2*(*pv);
1392b89f182dSHong Zhang             *pc1 = mul1;       *pc2 = mul2;
139328f1b45aSHong Zhang 
139428f1b45aSHong Zhang             pj = b->j + bdiag[row+1]+1;     /* beginning of U(row,:) */
139528f1b45aSHong Zhang             pv = b->a + bdiag[row+1]+1;
139628f1b45aSHong Zhang             nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
139728f1b45aSHong Zhang             for (j=0; j<nz; j++) {
139868785679SHong Zhang               col         = pj[j];
1399b89f182dSHong Zhang               rtmp1[col] -= mul1 * pv[j];
1400b89f182dSHong Zhang               rtmp2[col] -= mul2 * pv[j];
140128f1b45aSHong Zhang             }
140228b1a77fSLisandro Dalcin             ierr = PetscLogFlops(2+4*nz);CHKERRQ(ierr);
140328f1b45aSHong Zhang           }
140428f1b45aSHong Zhang           row = *bjtmp++;
140528f1b45aSHong Zhang         }
140628f1b45aSHong Zhang 
1407b89f182dSHong Zhang         /* finished row i; check zero pivot, then stick row i into b->a */
140828f1b45aSHong Zhang         rs = 0.0;
140928f1b45aSHong Zhang         /* L part */
1410b89f182dSHong Zhang         pc1 = b->a + bi[i];
141128f1b45aSHong Zhang         pj  = b->j + bi[i];
141228f1b45aSHong Zhang         nz  = bi[i+1] - bi[i];
141328f1b45aSHong Zhang         for (j=0; j<nz; j++) {
141468785679SHong Zhang           col    = pj[j];
14150e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
141628f1b45aSHong Zhang         }
141728f1b45aSHong Zhang         /* U part */
1418b89f182dSHong Zhang         pc1 = b->a + bdiag[i+1]+1;
141928f1b45aSHong Zhang         pj  = b->j + bdiag[i+1]+1;
14200e7a5c2bSHong Zhang         nz  = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */
142128f1b45aSHong Zhang         for (j=0; j<nz; j++) {
142268785679SHong Zhang           col    = pj[j];
14230e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
142428f1b45aSHong Zhang         }
142528f1b45aSHong Zhang 
142628f1b45aSHong Zhang         sctx.rs = rs;
1427b89f182dSHong Zhang         sctx.pv = rtmp1[i];
1428d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
142907b50cabSHong Zhang         if (sctx.newshift) break;
1430b89f182dSHong Zhang         pc1  = b->a + bdiag[i]; /* Mark diagonal */
1431b89f182dSHong Zhang         *pc1 = 1.0/sctx.pv;
1432b89f182dSHong Zhang 
1433b89f182dSHong Zhang         /* Now take care of diagonal 2x2 block. */
1434b89f182dSHong Zhang         pc2 = rtmp2 + i;
1435b89f182dSHong Zhang         if (*pc2 != 0.0) {
1436b89f182dSHong Zhang           mul1 = (*pc2)*(*pc1); /* *pc1=diag[i] is inverted! */
1437b89f182dSHong Zhang           *pc2 = mul1;          /* insert L entry */
1438b89f182dSHong Zhang           pj   = b->j + bdiag[i+1]+1;   /* beginning of U(i,:) */
1439b89f182dSHong Zhang           nz   = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */
1440b89f182dSHong Zhang           for (j=0; j<nz; j++) {
1441b89f182dSHong Zhang             col = pj[j]; rtmp2[col] -= mul1 * rtmp1[col];
144228f1b45aSHong Zhang           }
144328b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
1444b89f182dSHong Zhang         }
1445b89f182dSHong Zhang 
1446b89f182dSHong Zhang         /* finished row i+1; check zero pivot, then stick row i+1 into b->a */
1447b89f182dSHong Zhang         rs = 0.0;
1448b89f182dSHong Zhang         /* L part */
1449b89f182dSHong Zhang         pc2 = b->a + bi[i+1];
1450b89f182dSHong Zhang         pj  = b->j + bi[i+1];
1451b89f182dSHong Zhang         nz  = bi[i+2] - bi[i+1];
1452b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1453b89f182dSHong Zhang           col    = pj[j];
14540e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1455b89f182dSHong Zhang         }
1456b89f182dSHong Zhang         /* U part */
1457b89f182dSHong Zhang         pc2 = b->a + bdiag[i+2]+1;
14580e7a5c2bSHong Zhang         pj  = b->j + bdiag[i+2]+1;
14590e7a5c2bSHong Zhang         nz  = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */
1460b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1461b89f182dSHong Zhang           col    = pj[j];
14620e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1463b89f182dSHong Zhang         }
1464b89f182dSHong Zhang 
146528f1b45aSHong Zhang         sctx.rs = rs;
1466b89f182dSHong Zhang         sctx.pv = rtmp2[i+1];
1467d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i+1);CHKERRQ(ierr);
146807b50cabSHong Zhang         if (sctx.newshift) break;
146928f1b45aSHong Zhang         pc2  = b->a + bdiag[i+1];
1470b89f182dSHong Zhang         *pc2 = 1.0/sctx.pv;
147128f1b45aSHong Zhang         break;
1472b89f182dSHong Zhang 
147368785679SHong Zhang       case 3:
147468785679SHong Zhang         /*----------*/
147568785679SHong Zhang         /* zero rtmp */
147668785679SHong Zhang         /* L part */
147768785679SHong Zhang         nz    = bi[i+1] - bi[i];
147868785679SHong Zhang         bjtmp = bj + bi[i];
147968785679SHong Zhang         for  (j=0; j<nz; j++) {
148068785679SHong Zhang           col        = bjtmp[j];
1481b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;
148268785679SHong Zhang         }
148368785679SHong Zhang 
148468785679SHong Zhang         /* U part */
148568785679SHong Zhang         nz    = bdiag[i]-bdiag[i+1];
148668785679SHong Zhang         bjtmp = bj + bdiag[i+1]+1;
148768785679SHong Zhang         for  (j=0; j<nz; j++) {
148868785679SHong Zhang           col        = bjtmp[j];
1489b89f182dSHong Zhang           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;
149068785679SHong Zhang         }
149168785679SHong Zhang 
149268785679SHong Zhang         /* load in initial (unfactored row) */
149368785679SHong Zhang         nz    = ai[r[i]+1] - ai[r[i]];
149468785679SHong Zhang         ajtmp = aj + ai[r[i]];
149568785679SHong Zhang         v1    = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; v3 = aa + ai[r[i]+2];
149668785679SHong Zhang         for (j=0; j<nz; j++) {
149768785679SHong Zhang           col        = ics[ajtmp[j]];
1498b89f182dSHong Zhang           rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; rtmp3[col] = v3[j];
149968785679SHong Zhang         }
150068785679SHong Zhang         /* ZeropivotApply(): shift the diagonal of the matrix  */
1501b89f182dSHong Zhang         rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; rtmp3[i+2] += sctx.shift_amount;
150268785679SHong Zhang 
150368785679SHong Zhang         /* elimination */
150468785679SHong Zhang         bjtmp = bj + bi[i];
150568785679SHong Zhang         row   = *bjtmp++; /* pivot row */
150668785679SHong Zhang         nzL   = bi[i+1] - bi[i];
150768785679SHong Zhang         for (k=0; k < nzL; k++) {
1508b89f182dSHong Zhang           pc1 = rtmp1 + row;
1509b89f182dSHong Zhang           pc2 = rtmp2 + row;
1510b89f182dSHong Zhang           pc3 = rtmp3 + row;
151168785679SHong Zhang           if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 != 0.0) {
151268785679SHong Zhang             pv   = b->a + bdiag[row];
1513b89f182dSHong Zhang             mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); mul3 = *pc3*(*pv);
1514b89f182dSHong Zhang             *pc1 = mul1; *pc2 = mul2; *pc3 = mul3;
151568785679SHong Zhang 
151668785679SHong Zhang             pj = b->j + bdiag[row+1]+1;     /* beginning of U(row,:) */
151768785679SHong Zhang             pv = b->a + bdiag[row+1]+1;
151868785679SHong Zhang             nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
151968785679SHong Zhang             for (j=0; j<nz; j++) {
152068785679SHong Zhang               col         = pj[j];
1521b89f182dSHong Zhang               rtmp1[col] -= mul1 * pv[j];
1522b89f182dSHong Zhang               rtmp2[col] -= mul2 * pv[j];
1523b89f182dSHong Zhang               rtmp3[col] -= mul3 * pv[j];
152468785679SHong Zhang             }
152528b1a77fSLisandro Dalcin             ierr = PetscLogFlops(3+6*nz);CHKERRQ(ierr);
152668785679SHong Zhang           }
152768785679SHong Zhang           row = *bjtmp++;
152868785679SHong Zhang         }
152968785679SHong Zhang 
1530b89f182dSHong Zhang         /* finished row i; check zero pivot, then stick row i into b->a */
1531b89f182dSHong Zhang         rs = 0.0;
1532b89f182dSHong Zhang         /* L part */
1533b89f182dSHong Zhang         pc1 = b->a + bi[i];
1534b89f182dSHong Zhang         pj  = b->j + bi[i];
1535b89f182dSHong Zhang         nz  = bi[i+1] - bi[i];
1536b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1537b89f182dSHong Zhang           col    = pj[j];
15380e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
1539b89f182dSHong Zhang         }
1540b89f182dSHong Zhang         /* U part */
1541b89f182dSHong Zhang         pc1 = b->a + bdiag[i+1]+1;
1542b89f182dSHong Zhang         pj  = b->j + bdiag[i+1]+1;
15430e7a5c2bSHong Zhang         nz  = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */
1544b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1545b89f182dSHong Zhang           col    = pj[j];
15460e7a5c2bSHong Zhang           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
1547b89f182dSHong Zhang         }
154868785679SHong Zhang 
1549b89f182dSHong Zhang         sctx.rs = rs;
1550b89f182dSHong Zhang         sctx.pv = rtmp1[i];
1551d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
155207b50cabSHong Zhang         if (sctx.newshift) break;
1553b89f182dSHong Zhang         pc1  = b->a + bdiag[i]; /* Mark diag[i] */
1554b89f182dSHong Zhang         *pc1 = 1.0/sctx.pv;
1555b89f182dSHong Zhang 
1556b89f182dSHong Zhang         /* Now take care of 1st column of diagonal 3x3 block. */
1557b89f182dSHong Zhang         pc2 = rtmp2 + i;
1558b89f182dSHong Zhang         pc3 = rtmp3 + i;
1559b89f182dSHong Zhang         if (*pc2 != 0.0 || *pc3 != 0.0) {
1560b89f182dSHong Zhang           mul2 = (*pc2)*(*pc1); *pc2 = mul2;
1561b89f182dSHong Zhang           mul3 = (*pc3)*(*pc1); *pc3 = mul3;
156268785679SHong Zhang           pj   = b->j + bdiag[i+1]+1;   /* beginning of U(i,:) */
156368785679SHong Zhang           nz   = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */
156468785679SHong Zhang           for (j=0; j<nz; j++) {
156568785679SHong Zhang             col         = pj[j];
1566b89f182dSHong Zhang             rtmp2[col] -= mul2 * rtmp1[col];
1567b89f182dSHong Zhang             rtmp3[col] -= mul3 * rtmp1[col];
156868785679SHong Zhang           }
156928b1a77fSLisandro Dalcin           ierr = PetscLogFlops(2+4*nz);CHKERRQ(ierr);
157068785679SHong Zhang         }
157168785679SHong Zhang 
1572b89f182dSHong Zhang         /* finished row i+1; check zero pivot, then stick row i+1 into b->a */
1573b89f182dSHong Zhang         rs = 0.0;
1574b89f182dSHong Zhang         /* L part */
1575b89f182dSHong Zhang         pc2 = b->a + bi[i+1];
1576b89f182dSHong Zhang         pj  = b->j + bi[i+1];
1577b89f182dSHong Zhang         nz  = bi[i+2] - bi[i+1];
1578b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1579b89f182dSHong Zhang           col    = pj[j];
15800e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1581b89f182dSHong Zhang         }
1582b89f182dSHong Zhang         /* U part */
1583b89f182dSHong Zhang         pc2 = b->a + bdiag[i+2]+1;
15840e7a5c2bSHong Zhang         pj  = b->j + bdiag[i+2]+1;
15850e7a5c2bSHong Zhang         nz  = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */
1586b89f182dSHong Zhang         for (j=0; j<nz; j++) {
1587b89f182dSHong Zhang           col    = pj[j];
15880e7a5c2bSHong Zhang           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
1589b89f182dSHong Zhang         }
1590b89f182dSHong Zhang 
1591b89f182dSHong Zhang         sctx.rs = rs;
1592b89f182dSHong Zhang         sctx.pv = rtmp2[i+1];
1593d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i+1);CHKERRQ(ierr);
159407b50cabSHong Zhang         if (sctx.newshift) break;
1595b89f182dSHong Zhang         pc2  = b->a + bdiag[i+1];
1596b89f182dSHong Zhang         *pc2 = 1.0/sctx.pv; /* Mark diag[i+1] */
1597b89f182dSHong Zhang 
1598b89f182dSHong Zhang         /* Now take care of 2nd column of diagonal 3x3 block. */
1599b89f182dSHong Zhang         pc3 = rtmp3 + i+1;
160068785679SHong Zhang         if (*pc3 != 0.0) {
1601b89f182dSHong Zhang           mul3 = (*pc3)*(*pc2); *pc3 = mul3;
160268785679SHong Zhang           pj   = b->j + bdiag[i+2]+1;     /* beginning of U(i+1,:) */
160368785679SHong Zhang           nz   = bdiag[i+1]-bdiag[i+2]-1; /* num of entries in U(i+1,:) excluding diag */
160468785679SHong Zhang           for (j=0; j<nz; j++) {
160568785679SHong Zhang             col         = pj[j];
1606b89f182dSHong Zhang             rtmp3[col] -= mul3 * rtmp2[col];
160768785679SHong Zhang           }
160828b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
160968785679SHong Zhang         }
161068785679SHong Zhang 
1611b89f182dSHong Zhang         /* finished i+2; check zero pivot, then stick row i+2 into b->a */
161268785679SHong Zhang         rs = 0.0;
161368785679SHong Zhang         /* L part */
1614b89f182dSHong Zhang         pc3 = b->a + bi[i+2];
1615b89f182dSHong Zhang         pj  = b->j + bi[i+2];
1616b89f182dSHong Zhang         nz  = bi[i+3] - bi[i+2];
161768785679SHong Zhang         for (j=0; j<nz; j++) {
161868785679SHong Zhang           col    = pj[j];
16190e7a5c2bSHong Zhang           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
162068785679SHong Zhang         }
162168785679SHong Zhang         /* U part */
1622b89f182dSHong Zhang         pc3 = b->a + bdiag[i+3]+1;
16230e7a5c2bSHong Zhang         pj  = b->j + bdiag[i+3]+1;
16240e7a5c2bSHong Zhang         nz  = bdiag[i+2] - bdiag[i+3] - 1; /* exclude diagonal */
162568785679SHong Zhang         for (j=0; j<nz; j++) {
162668785679SHong Zhang           col    = pj[j];
16270e7a5c2bSHong Zhang           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
162868785679SHong Zhang         }
162968785679SHong Zhang 
163068785679SHong Zhang         sctx.rs = rs;
1631b89f182dSHong Zhang         sctx.pv = rtmp3[i+2];
1632d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i+2);CHKERRQ(ierr);
163307b50cabSHong Zhang         if (sctx.newshift) break;
163468785679SHong Zhang         pc3  = b->a + bdiag[i+2];
1635b89f182dSHong Zhang         *pc3 = 1.0/sctx.pv; /* Mark diag[i+2] */
163668785679SHong Zhang         break;
16379877982aSShri Abhyankar       case 4:
16389877982aSShri Abhyankar         /*----------*/
16399877982aSShri Abhyankar         /* zero rtmp */
16409877982aSShri Abhyankar         /* L part */
16419877982aSShri Abhyankar         nz    = bi[i+1] - bi[i];
16429877982aSShri Abhyankar         bjtmp = bj + bi[i];
16439877982aSShri Abhyankar         for  (j=0; j<nz; j++) {
16449877982aSShri Abhyankar           col        = bjtmp[j];
16459877982aSShri Abhyankar           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;rtmp4[col] = 0.0;
16469877982aSShri Abhyankar         }
16479877982aSShri Abhyankar 
16489877982aSShri Abhyankar         /* U part */
16499877982aSShri Abhyankar         nz    = bdiag[i]-bdiag[i+1];
16509877982aSShri Abhyankar         bjtmp = bj + bdiag[i+1]+1;
16519877982aSShri Abhyankar         for  (j=0; j<nz; j++) {
16529877982aSShri Abhyankar           col        = bjtmp[j];
16539877982aSShri Abhyankar           rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0; rtmp4[col] = 0.0;
16549877982aSShri Abhyankar         }
16559877982aSShri Abhyankar 
16569877982aSShri Abhyankar         /* load in initial (unfactored row) */
16579877982aSShri Abhyankar         nz    = ai[r[i]+1] - ai[r[i]];
16589877982aSShri Abhyankar         ajtmp = aj + ai[r[i]];
16599877982aSShri Abhyankar         v1    = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; v3 = aa + ai[r[i]+2]; v4 = aa + ai[r[i]+3];
16609877982aSShri Abhyankar         for (j=0; j<nz; j++) {
16619877982aSShri Abhyankar           col        = ics[ajtmp[j]];
16629877982aSShri Abhyankar           rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; rtmp3[col] = v3[j]; rtmp4[col] = v4[j];
16639877982aSShri Abhyankar         }
16649877982aSShri Abhyankar         /* ZeropivotApply(): shift the diagonal of the matrix  */
16659877982aSShri Abhyankar         rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; rtmp3[i+2] += sctx.shift_amount; rtmp4[i+3] += sctx.shift_amount;
16669877982aSShri Abhyankar 
16679877982aSShri Abhyankar         /* elimination */
16689877982aSShri Abhyankar         bjtmp = bj + bi[i];
16699877982aSShri Abhyankar         row   = *bjtmp++; /* pivot row */
16709877982aSShri Abhyankar         nzL   = bi[i+1] - bi[i];
16719877982aSShri Abhyankar         for (k=0; k < nzL; k++) {
16729877982aSShri Abhyankar           pc1 = rtmp1 + row;
16739877982aSShri Abhyankar           pc2 = rtmp2 + row;
16749877982aSShri Abhyankar           pc3 = rtmp3 + row;
16759877982aSShri Abhyankar           pc4 = rtmp4 + row;
16769877982aSShri Abhyankar           if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 != 0.0 || *pc4 != 0.0) {
16779877982aSShri Abhyankar             pv   = b->a + bdiag[row];
16789877982aSShri Abhyankar             mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); mul3 = *pc3*(*pv); mul4 = *pc4*(*pv);
16799877982aSShri Abhyankar             *pc1 = mul1; *pc2 = mul2; *pc3 = mul3; *pc4 = mul4;
16809877982aSShri Abhyankar 
16819877982aSShri Abhyankar             pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
16829877982aSShri Abhyankar             pv = b->a + bdiag[row+1]+1;
16839877982aSShri Abhyankar             nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
16849877982aSShri Abhyankar             for (j=0; j<nz; j++) {
16859877982aSShri Abhyankar               col         = pj[j];
16869877982aSShri Abhyankar               rtmp1[col] -= mul1 * pv[j];
16879877982aSShri Abhyankar               rtmp2[col] -= mul2 * pv[j];
16889877982aSShri Abhyankar               rtmp3[col] -= mul3 * pv[j];
16899877982aSShri Abhyankar               rtmp4[col] -= mul4 * pv[j];
16909877982aSShri Abhyankar             }
169128b1a77fSLisandro Dalcin             ierr = PetscLogFlops(4+8*nz);CHKERRQ(ierr);
16929877982aSShri Abhyankar           }
16939877982aSShri Abhyankar           row = *bjtmp++;
16949877982aSShri Abhyankar         }
16959877982aSShri Abhyankar 
16969877982aSShri Abhyankar         /* finished row i; check zero pivot, then stick row i into b->a */
16979877982aSShri Abhyankar         rs = 0.0;
16989877982aSShri Abhyankar         /* L part */
16999877982aSShri Abhyankar         pc1 = b->a + bi[i];
17009877982aSShri Abhyankar         pj  = b->j + bi[i];
17019877982aSShri Abhyankar         nz  = bi[i+1] - bi[i];
17029877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17039877982aSShri Abhyankar           col    = pj[j];
17049877982aSShri Abhyankar           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
17059877982aSShri Abhyankar         }
17069877982aSShri Abhyankar         /* U part */
17079877982aSShri Abhyankar         pc1 = b->a + bdiag[i+1]+1;
17089877982aSShri Abhyankar         pj  = b->j + bdiag[i+1]+1;
17099877982aSShri Abhyankar         nz  = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */
17109877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17119877982aSShri Abhyankar           col    = pj[j];
17129877982aSShri Abhyankar           pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]);
17139877982aSShri Abhyankar         }
17149877982aSShri Abhyankar 
17159877982aSShri Abhyankar         sctx.rs = rs;
17169877982aSShri Abhyankar         sctx.pv = rtmp1[i];
1717d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
171807b50cabSHong Zhang         if (sctx.newshift) break;
17199877982aSShri Abhyankar         pc1  = b->a + bdiag[i]; /* Mark diag[i] */
17209877982aSShri Abhyankar         *pc1 = 1.0/sctx.pv;
17219877982aSShri Abhyankar 
17229877982aSShri Abhyankar         /* Now take care of 1st column of diagonal 4x4 block. */
17239877982aSShri Abhyankar         pc2 = rtmp2 + i;
17249877982aSShri Abhyankar         pc3 = rtmp3 + i;
17259877982aSShri Abhyankar         pc4 = rtmp4 + i;
17269877982aSShri Abhyankar         if (*pc2 != 0.0 || *pc3 != 0.0 || *pc4 != 0.0) {
17279877982aSShri Abhyankar           mul2 = (*pc2)*(*pc1); *pc2 = mul2;
17289877982aSShri Abhyankar           mul3 = (*pc3)*(*pc1); *pc3 = mul3;
17299877982aSShri Abhyankar           mul4 = (*pc4)*(*pc1); *pc4 = mul4;
17309877982aSShri Abhyankar           pj   = b->j + bdiag[i+1]+1;   /* beginning of U(i,:) */
17319877982aSShri Abhyankar           nz   = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */
17329877982aSShri Abhyankar           for (j=0; j<nz; j++) {
17339877982aSShri Abhyankar             col         = pj[j];
17349877982aSShri Abhyankar             rtmp2[col] -= mul2 * rtmp1[col];
17359877982aSShri Abhyankar             rtmp3[col] -= mul3 * rtmp1[col];
17369877982aSShri Abhyankar             rtmp4[col] -= mul4 * rtmp1[col];
17379877982aSShri Abhyankar           }
173828b1a77fSLisandro Dalcin           ierr = PetscLogFlops(3+6*nz);CHKERRQ(ierr);
17399877982aSShri Abhyankar         }
17409877982aSShri Abhyankar 
17419877982aSShri Abhyankar         /* finished row i+1; check zero pivot, then stick row i+1 into b->a */
17429877982aSShri Abhyankar         rs = 0.0;
17439877982aSShri Abhyankar         /* L part */
17449877982aSShri Abhyankar         pc2 = b->a + bi[i+1];
17459877982aSShri Abhyankar         pj  = b->j + bi[i+1];
17469877982aSShri Abhyankar         nz  = bi[i+2] - bi[i+1];
17479877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17489877982aSShri Abhyankar           col    = pj[j];
17499877982aSShri Abhyankar           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
17509877982aSShri Abhyankar         }
17519877982aSShri Abhyankar         /* U part */
17529877982aSShri Abhyankar         pc2 = b->a + bdiag[i+2]+1;
17539877982aSShri Abhyankar         pj  = b->j + bdiag[i+2]+1;
17549877982aSShri Abhyankar         nz  = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */
17559877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17569877982aSShri Abhyankar           col    = pj[j];
17579877982aSShri Abhyankar           pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]);
17589877982aSShri Abhyankar         }
17599877982aSShri Abhyankar 
17609877982aSShri Abhyankar         sctx.rs = rs;
17619877982aSShri Abhyankar         sctx.pv = rtmp2[i+1];
1762d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i+1);CHKERRQ(ierr);
176307b50cabSHong Zhang         if (sctx.newshift) break;
17649877982aSShri Abhyankar         pc2  = b->a + bdiag[i+1];
17659877982aSShri Abhyankar         *pc2 = 1.0/sctx.pv; /* Mark diag[i+1] */
17669877982aSShri Abhyankar 
17679877982aSShri Abhyankar         /* Now take care of 2nd column of diagonal 4x4 block. */
17689877982aSShri Abhyankar         pc3 = rtmp3 + i+1;
17699877982aSShri Abhyankar         pc4 = rtmp4 + i+1;
17709877982aSShri Abhyankar         if (*pc3 != 0.0 || *pc4 != 0.0) {
17719877982aSShri Abhyankar           mul3 = (*pc3)*(*pc2); *pc3 = mul3;
17729877982aSShri Abhyankar           mul4 = (*pc4)*(*pc2); *pc4 = mul4;
17739877982aSShri Abhyankar           pj   = b->j + bdiag[i+2]+1;     /* beginning of U(i+1,:) */
17749877982aSShri Abhyankar           nz   = bdiag[i+1]-bdiag[i+2]-1; /* num of entries in U(i+1,:) excluding diag */
17759877982aSShri Abhyankar           for (j=0; j<nz; j++) {
17769877982aSShri Abhyankar             col         = pj[j];
17779877982aSShri Abhyankar             rtmp3[col] -= mul3 * rtmp2[col];
17789877982aSShri Abhyankar             rtmp4[col] -= mul4 * rtmp2[col];
17799877982aSShri Abhyankar           }
17809877982aSShri Abhyankar           ierr = PetscLogFlops(4*nz);CHKERRQ(ierr);
17819877982aSShri Abhyankar         }
17829877982aSShri Abhyankar 
17839877982aSShri Abhyankar         /* finished i+2; check zero pivot, then stick row i+2 into b->a */
17849877982aSShri Abhyankar         rs = 0.0;
17859877982aSShri Abhyankar         /* L part */
17869877982aSShri Abhyankar         pc3 = b->a + bi[i+2];
17879877982aSShri Abhyankar         pj  = b->j + bi[i+2];
17889877982aSShri Abhyankar         nz  = bi[i+3] - bi[i+2];
17899877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17909877982aSShri Abhyankar           col    = pj[j];
17919877982aSShri Abhyankar           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
17929877982aSShri Abhyankar         }
17939877982aSShri Abhyankar         /* U part */
17949877982aSShri Abhyankar         pc3 = b->a + bdiag[i+3]+1;
17959877982aSShri Abhyankar         pj  = b->j + bdiag[i+3]+1;
17969877982aSShri Abhyankar         nz  = bdiag[i+2] - bdiag[i+3] - 1; /* exclude diagonal */
17979877982aSShri Abhyankar         for (j=0; j<nz; j++) {
17989877982aSShri Abhyankar           col    = pj[j];
17999877982aSShri Abhyankar           pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]);
18009877982aSShri Abhyankar         }
18019877982aSShri Abhyankar 
18029877982aSShri Abhyankar         sctx.rs = rs;
18039877982aSShri Abhyankar         sctx.pv = rtmp3[i+2];
1804d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i+2);CHKERRQ(ierr);
180507b50cabSHong Zhang         if (sctx.newshift) break;
18069877982aSShri Abhyankar         pc3  = b->a + bdiag[i+2];
18079877982aSShri Abhyankar         *pc3 = 1.0/sctx.pv; /* Mark diag[i+2] */
18089877982aSShri Abhyankar 
18099877982aSShri Abhyankar         /* Now take care of 3rd column of diagonal 4x4 block. */
18109877982aSShri Abhyankar         pc4 = rtmp4 + i+2;
18119877982aSShri Abhyankar         if (*pc4 != 0.0) {
18129877982aSShri Abhyankar           mul4 = (*pc4)*(*pc3); *pc4 = mul4;
18139877982aSShri Abhyankar           pj   = b->j + bdiag[i+3]+1;     /* beginning of U(i+2,:) */
18149877982aSShri Abhyankar           nz   = bdiag[i+2]-bdiag[i+3]-1; /* num of entries in U(i+2,:) excluding diag */
18159877982aSShri Abhyankar           for (j=0; j<nz; j++) {
18169877982aSShri Abhyankar             col         = pj[j];
18179877982aSShri Abhyankar             rtmp4[col] -= mul4 * rtmp3[col];
18189877982aSShri Abhyankar           }
181928b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
18209877982aSShri Abhyankar         }
18219877982aSShri Abhyankar 
18229877982aSShri Abhyankar         /* finished i+3; check zero pivot, then stick row i+3 into b->a */
18239877982aSShri Abhyankar         rs = 0.0;
18249877982aSShri Abhyankar         /* L part */
18259877982aSShri Abhyankar         pc4 = b->a + bi[i+3];
18269877982aSShri Abhyankar         pj  = b->j + bi[i+3];
18279877982aSShri Abhyankar         nz  = bi[i+4] - bi[i+3];
18289877982aSShri Abhyankar         for (j=0; j<nz; j++) {
18299877982aSShri Abhyankar           col    = pj[j];
18309877982aSShri Abhyankar           pc4[j] = rtmp4[col]; rs += PetscAbsScalar(pc4[j]);
18319877982aSShri Abhyankar         }
18329877982aSShri Abhyankar         /* U part */
18339877982aSShri Abhyankar         pc4 = b->a + bdiag[i+4]+1;
18349877982aSShri Abhyankar         pj  = b->j + bdiag[i+4]+1;
18359877982aSShri Abhyankar         nz  = bdiag[i+3] - bdiag[i+4] - 1; /* exclude diagonal */
18369877982aSShri Abhyankar         for (j=0; j<nz; j++) {
18379877982aSShri Abhyankar           col    = pj[j];
18389877982aSShri Abhyankar           pc4[j] = rtmp4[col]; rs += PetscAbsScalar(pc4[j]);
18399877982aSShri Abhyankar         }
18409877982aSShri Abhyankar 
18419877982aSShri Abhyankar         sctx.rs = rs;
18429877982aSShri Abhyankar         sctx.pv = rtmp4[i+3];
1843d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,i+3);CHKERRQ(ierr);
184407b50cabSHong Zhang         if (sctx.newshift) break;
18459877982aSShri Abhyankar         pc4  = b->a + bdiag[i+3];
18469877982aSShri Abhyankar         *pc4 = 1.0/sctx.pv; /* Mark diag[i+3] */
18479877982aSShri Abhyankar         break;
184868785679SHong Zhang 
184928f1b45aSHong Zhang       default:
1850e32f2f54SBarry Smith         SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Node size not yet supported \n");
185128f1b45aSHong Zhang       }
1852c2b86aeeSHong Zhang       if (sctx.newshift) break; /* break for (inod=0,i=0; inod<node_max; inod++) */
185328f1b45aSHong Zhang       i += nodesz;                 /* Update the row */
185468785679SHong Zhang     }
185528f1b45aSHong Zhang 
185628f1b45aSHong Zhang     /* MatPivotRefine() */
185707b50cabSHong Zhang     if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
185828f1b45aSHong Zhang       /*
185928f1b45aSHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
186028f1b45aSHong Zhang        * then try lower shift
186128f1b45aSHong Zhang        */
186228f1b45aSHong Zhang       sctx.shift_hi       = sctx.shift_fraction;
186328f1b45aSHong Zhang       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
186428f1b45aSHong Zhang       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
186507b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
186628f1b45aSHong Zhang       sctx.nshift++;
186728f1b45aSHong Zhang     }
186807b50cabSHong Zhang   } while (sctx.newshift);
186928f1b45aSHong Zhang 
1870eb9baa12SBarry Smith   ierr = PetscFree4(rtmp1,rtmp2,rtmp3,rtmp4);CHKERRQ(ierr);
187168785679SHong Zhang   ierr = PetscFree(tmp_vec2);CHKERRQ(ierr);
187228f1b45aSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
187328f1b45aSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
187428f1b45aSHong Zhang 
1875abb87a52SBarry Smith   if (b->inode.size) {
1876abb87a52SBarry Smith     C->ops->solve           = MatSolve_SeqAIJ_Inode;
1877abb87a52SBarry Smith   } else {
1878d3ac4fa3SBarry Smith     C->ops->solve           = MatSolve_SeqAIJ;
1879abb87a52SBarry Smith   }
188028f1b45aSHong Zhang   C->ops->solveadd          = MatSolveAdd_SeqAIJ;
188128f1b45aSHong Zhang   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ;
188228f1b45aSHong Zhang   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ;
188328f1b45aSHong Zhang   C->ops->matsolve          = MatMatSolve_SeqAIJ;
188428f1b45aSHong Zhang   C->assembled              = PETSC_TRUE;
188528f1b45aSHong Zhang   C->preallocated           = PETSC_TRUE;
18862205254eSKarl Rupp 
188728f1b45aSHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
188828f1b45aSHong Zhang 
188928f1b45aSHong Zhang   /* MatShiftView(A,info,&sctx) */
189028f1b45aSHong Zhang   if (sctx.nshift) {
1891f4db908eSBarry Smith     if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) {
189228f1b45aSHong Zhang       ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr);
1893f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
189428f1b45aSHong Zhang       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
1895f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) {
189628f1b45aSHong Zhang       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr);
189728f1b45aSHong Zhang     }
189828f1b45aSHong Zhang   }
189928f1b45aSHong Zhang   PetscFunctionReturn(0);
190028f1b45aSHong Zhang }
1901628f99d7SShri Abhyankar 
1902628f99d7SShri Abhyankar #undef __FUNCT__
1903628f99d7SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_Inode_inplace"
1904628f99d7SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ_Inode_inplace(Mat B,Mat A,const MatFactorInfo *info)
1905628f99d7SShri Abhyankar {
1906628f99d7SShri Abhyankar   Mat             C     = B;
1907628f99d7SShri Abhyankar   Mat_SeqAIJ      *a    = (Mat_SeqAIJ*)A->data,*b = (Mat_SeqAIJ*)C->data;
1908628f99d7SShri Abhyankar   IS              iscol = b->col,isrow = b->row,isicol = b->icol;
1909628f99d7SShri Abhyankar   PetscErrorCode  ierr;
1910628f99d7SShri Abhyankar   const PetscInt  *r,*ic,*c,*ics;
1911628f99d7SShri Abhyankar   PetscInt        n   = A->rmap->n,*bi = b->i;
1912628f99d7SShri Abhyankar   PetscInt        *bj = b->j,*nbj=b->j +1,*ajtmp,*bjtmp,nz,nz_tmp,row,prow;
19138758e1faSBarry Smith   PetscInt        i,j,idx,*bd = b->diag,node_max,nodesz;
19148758e1faSBarry Smith   PetscInt        *ai = a->i,*aj = a->j;
1915628f99d7SShri Abhyankar   PetscInt        *ns,*tmp_vec1,*tmp_vec2,*nsmap,*pj;
1916628f99d7SShri Abhyankar   PetscScalar     mul1,mul2,mul3,tmp;
1917628f99d7SShri Abhyankar   MatScalar       *pc1,*pc2,*pc3,*ba = b->a,*pv,*rtmp11,*rtmp22,*rtmp33;
1918628f99d7SShri Abhyankar   const MatScalar *v1,*v2,*v3,*aa = a->a,*rtmp1;
1919628f99d7SShri Abhyankar   PetscReal       rs=0.0;
1920628f99d7SShri Abhyankar   FactorShiftCtx  sctx;
1921628f99d7SShri Abhyankar 
1922628f99d7SShri Abhyankar   PetscFunctionBegin;
1923628f99d7SShri Abhyankar   sctx.shift_top      = 0;
1924628f99d7SShri Abhyankar   sctx.nshift_max     = 0;
1925628f99d7SShri Abhyankar   sctx.shift_lo       = 0;
1926628f99d7SShri Abhyankar   sctx.shift_hi       = 0;
1927628f99d7SShri Abhyankar   sctx.shift_fraction = 0;
1928628f99d7SShri Abhyankar 
1929628f99d7SShri Abhyankar   /* if both shift schemes are chosen by user, only use info->shiftpd */
1930f4db908eSBarry Smith   if (info->shifttype==(PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
1931628f99d7SShri Abhyankar     sctx.shift_top = 0;
1932628f99d7SShri Abhyankar     for (i=0; i<n; i++) {
1933628f99d7SShri Abhyankar       /* calculate rs = sum(|aij|)-RealPart(aii), amt of shift needed for this row */
1934628f99d7SShri Abhyankar       rs    = 0.0;
1935628f99d7SShri Abhyankar       ajtmp = aj + ai[i];
1936628f99d7SShri Abhyankar       rtmp1 = aa + ai[i];
1937628f99d7SShri Abhyankar       nz    = ai[i+1] - ai[i];
1938628f99d7SShri Abhyankar       for (j=0; j<nz; j++) {
1939628f99d7SShri Abhyankar         if (*ajtmp != i) {
1940628f99d7SShri Abhyankar           rs += PetscAbsScalar(*rtmp1++);
1941628f99d7SShri Abhyankar         } else {
1942628f99d7SShri Abhyankar           rs -= PetscRealPart(*rtmp1++);
1943628f99d7SShri Abhyankar         }
1944628f99d7SShri Abhyankar         ajtmp++;
1945628f99d7SShri Abhyankar       }
1946628f99d7SShri Abhyankar       if (rs>sctx.shift_top) sctx.shift_top = rs;
1947628f99d7SShri Abhyankar     }
1948628f99d7SShri Abhyankar     if (sctx.shift_top == 0.0) sctx.shift_top += 1.e-12;
1949628f99d7SShri Abhyankar     sctx.shift_top *= 1.1;
1950628f99d7SShri Abhyankar     sctx.nshift_max = 5;
1951628f99d7SShri Abhyankar     sctx.shift_lo   = 0.;
1952628f99d7SShri Abhyankar     sctx.shift_hi   = 1.;
1953628f99d7SShri Abhyankar   }
1954628f99d7SShri Abhyankar   sctx.shift_amount = 0;
1955628f99d7SShri Abhyankar   sctx.nshift       = 0;
1956628f99d7SShri Abhyankar 
1957628f99d7SShri Abhyankar   ierr   = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1958628f99d7SShri Abhyankar   ierr   = ISGetIndices(iscol,&c);CHKERRQ(ierr);
1959628f99d7SShri Abhyankar   ierr   = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
1960*dcca6d9dSJed Brown   ierr   = PetscMalloc3(n,&rtmp11,n,&rtmp22,n,&rtmp33);CHKERRQ(ierr);
1961eb9baa12SBarry Smith   ierr   = PetscMemzero(rtmp11,n*sizeof(PetscScalar));CHKERRQ(ierr);
1962eb9baa12SBarry Smith   ierr   = PetscMemzero(rtmp22,n*sizeof(PetscScalar));CHKERRQ(ierr);
1963eb9baa12SBarry Smith   ierr   = PetscMemzero(rtmp33,n*sizeof(PetscScalar));CHKERRQ(ierr);
1964628f99d7SShri Abhyankar   ics    = ic;
1965628f99d7SShri Abhyankar 
1966628f99d7SShri Abhyankar   node_max = a->inode.node_count;
1967628f99d7SShri Abhyankar   ns       = a->inode.size;
1968e7e72b3dSBarry Smith   if (!ns) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix without inode information");
1969628f99d7SShri Abhyankar 
1970628f99d7SShri Abhyankar   /* If max inode size > 3, split it into two inodes.*/
1971628f99d7SShri Abhyankar   /* also map the inode sizes according to the ordering */
1972628f99d7SShri Abhyankar   ierr = PetscMalloc((n+1)* sizeof(PetscInt),&tmp_vec1);CHKERRQ(ierr);
1973628f99d7SShri Abhyankar   for (i=0,j=0; i<node_max; ++i,++j) {
1974628f99d7SShri Abhyankar     if (ns[i]>3) {
1975628f99d7SShri Abhyankar       tmp_vec1[j] = ns[i]/2; /* Assuming ns[i] < =5  */
1976628f99d7SShri Abhyankar       ++j;
1977628f99d7SShri Abhyankar       tmp_vec1[j] = ns[i] - tmp_vec1[j-1];
1978628f99d7SShri Abhyankar     } else {
1979628f99d7SShri Abhyankar       tmp_vec1[j] = ns[i];
1980628f99d7SShri Abhyankar     }
1981628f99d7SShri Abhyankar   }
1982628f99d7SShri Abhyankar   /* Use the correct node_max */
1983628f99d7SShri Abhyankar   node_max = j;
1984628f99d7SShri Abhyankar 
1985628f99d7SShri Abhyankar   /* Now reorder the inode info based on mat re-ordering info */
1986628f99d7SShri Abhyankar   /* First create a row -> inode_size_array_index map */
1987628f99d7SShri Abhyankar   ierr = PetscMalloc(n*sizeof(PetscInt)+1,&nsmap);CHKERRQ(ierr);
1988628f99d7SShri Abhyankar   ierr = PetscMalloc(node_max*sizeof(PetscInt)+1,&tmp_vec2);CHKERRQ(ierr);
1989628f99d7SShri Abhyankar   for (i=0,row=0; i<node_max; i++) {
1990628f99d7SShri Abhyankar     nodesz = tmp_vec1[i];
1991628f99d7SShri Abhyankar     for (j=0; j<nodesz; j++,row++) {
1992628f99d7SShri Abhyankar       nsmap[row] = i;
1993628f99d7SShri Abhyankar     }
1994628f99d7SShri Abhyankar   }
1995628f99d7SShri Abhyankar   /* Using nsmap, create a reordered ns structure */
1996628f99d7SShri Abhyankar   for (i=0,j=0; i< node_max; i++) {
1997628f99d7SShri Abhyankar     nodesz      = tmp_vec1[nsmap[r[j]]];     /* here the reordered row_no is in r[] */
1998628f99d7SShri Abhyankar     tmp_vec2[i] = nodesz;
1999628f99d7SShri Abhyankar     j          += nodesz;
2000628f99d7SShri Abhyankar   }
2001628f99d7SShri Abhyankar   ierr = PetscFree(nsmap);CHKERRQ(ierr);
2002628f99d7SShri Abhyankar   ierr = PetscFree(tmp_vec1);CHKERRQ(ierr);
2003628f99d7SShri Abhyankar   /* Now use the correct ns */
2004628f99d7SShri Abhyankar   ns = tmp_vec2;
2005628f99d7SShri Abhyankar 
2006628f99d7SShri Abhyankar   do {
200707b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
2008628f99d7SShri Abhyankar     /* Now loop over each block-row, and do the factorization */
2009628f99d7SShri Abhyankar     for (i=0,row=0; i<node_max; i++) {
2010628f99d7SShri Abhyankar       nodesz = ns[i];
2011628f99d7SShri Abhyankar       nz     = bi[row+1] - bi[row];
2012628f99d7SShri Abhyankar       bjtmp  = bj + bi[row];
2013628f99d7SShri Abhyankar 
2014628f99d7SShri Abhyankar       switch (nodesz) {
2015628f99d7SShri Abhyankar       case 1:
2016628f99d7SShri Abhyankar         for  (j=0; j<nz; j++) {
2017628f99d7SShri Abhyankar           idx         = bjtmp[j];
2018628f99d7SShri Abhyankar           rtmp11[idx] = 0.0;
2019628f99d7SShri Abhyankar         }
2020628f99d7SShri Abhyankar 
2021628f99d7SShri Abhyankar         /* load in initial (unfactored row) */
2022628f99d7SShri Abhyankar         idx    = r[row];
2023628f99d7SShri Abhyankar         nz_tmp = ai[idx+1] - ai[idx];
2024628f99d7SShri Abhyankar         ajtmp  = aj + ai[idx];
2025628f99d7SShri Abhyankar         v1     = aa + ai[idx];
2026628f99d7SShri Abhyankar 
2027628f99d7SShri Abhyankar         for (j=0; j<nz_tmp; j++) {
2028628f99d7SShri Abhyankar           idx         = ics[ajtmp[j]];
2029628f99d7SShri Abhyankar           rtmp11[idx] = v1[j];
2030628f99d7SShri Abhyankar         }
2031628f99d7SShri Abhyankar         rtmp11[ics[r[row]]] += sctx.shift_amount;
2032628f99d7SShri Abhyankar 
2033628f99d7SShri Abhyankar         prow = *bjtmp++;
2034628f99d7SShri Abhyankar         while (prow < row) {
2035628f99d7SShri Abhyankar           pc1 = rtmp11 + prow;
2036628f99d7SShri Abhyankar           if (*pc1 != 0.0) {
2037628f99d7SShri Abhyankar             pv     = ba + bd[prow];
2038628f99d7SShri Abhyankar             pj     = nbj + bd[prow];
2039628f99d7SShri Abhyankar             mul1   = *pc1 * *pv++;
2040628f99d7SShri Abhyankar             *pc1   = mul1;
2041628f99d7SShri Abhyankar             nz_tmp = bi[prow+1] - bd[prow] - 1;
204228b1a77fSLisandro Dalcin             ierr   = PetscLogFlops(1+2*nz_tmp);CHKERRQ(ierr);
2043628f99d7SShri Abhyankar             for (j=0; j<nz_tmp; j++) {
2044628f99d7SShri Abhyankar               tmp          = pv[j];
2045628f99d7SShri Abhyankar               idx          = pj[j];
2046628f99d7SShri Abhyankar               rtmp11[idx] -= mul1 * tmp;
2047628f99d7SShri Abhyankar             }
2048628f99d7SShri Abhyankar           }
2049628f99d7SShri Abhyankar           prow = *bjtmp++;
2050628f99d7SShri Abhyankar         }
2051628f99d7SShri Abhyankar         pj  = bj + bi[row];
2052628f99d7SShri Abhyankar         pc1 = ba + bi[row];
2053628f99d7SShri Abhyankar 
2054628f99d7SShri Abhyankar         sctx.pv     = rtmp11[row];
2055628f99d7SShri Abhyankar         rtmp11[row] = 1.0/rtmp11[row]; /* invert diag */
2056628f99d7SShri Abhyankar         rs          = 0.0;
2057628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2058628f99d7SShri Abhyankar           idx    = pj[j];
2059628f99d7SShri Abhyankar           pc1[j] = rtmp11[idx]; /* rtmp11 -> ba */
2060628f99d7SShri Abhyankar           if (idx != row) rs += PetscAbsScalar(pc1[j]);
2061628f99d7SShri Abhyankar         }
2062628f99d7SShri Abhyankar         sctx.rs = rs;
2063d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,row);CHKERRQ(ierr);
206407b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2065628f99d7SShri Abhyankar         break;
2066628f99d7SShri Abhyankar 
2067628f99d7SShri Abhyankar       case 2:
2068628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2069628f99d7SShri Abhyankar           idx         = bjtmp[j];
2070628f99d7SShri Abhyankar           rtmp11[idx] = 0.0;
2071628f99d7SShri Abhyankar           rtmp22[idx] = 0.0;
2072628f99d7SShri Abhyankar         }
2073628f99d7SShri Abhyankar 
2074628f99d7SShri Abhyankar         /* load in initial (unfactored row) */
2075628f99d7SShri Abhyankar         idx    = r[row];
2076628f99d7SShri Abhyankar         nz_tmp = ai[idx+1] - ai[idx];
2077628f99d7SShri Abhyankar         ajtmp  = aj + ai[idx];
2078628f99d7SShri Abhyankar         v1     = aa + ai[idx];
2079628f99d7SShri Abhyankar         v2     = aa + ai[idx+1];
2080628f99d7SShri Abhyankar         for (j=0; j<nz_tmp; j++) {
2081628f99d7SShri Abhyankar           idx         = ics[ajtmp[j]];
2082628f99d7SShri Abhyankar           rtmp11[idx] = v1[j];
2083628f99d7SShri Abhyankar           rtmp22[idx] = v2[j];
2084628f99d7SShri Abhyankar         }
2085628f99d7SShri Abhyankar         rtmp11[ics[r[row]]]   += sctx.shift_amount;
2086628f99d7SShri Abhyankar         rtmp22[ics[r[row+1]]] += sctx.shift_amount;
2087628f99d7SShri Abhyankar 
2088628f99d7SShri Abhyankar         prow = *bjtmp++;
2089628f99d7SShri Abhyankar         while (prow < row) {
2090628f99d7SShri Abhyankar           pc1 = rtmp11 + prow;
2091628f99d7SShri Abhyankar           pc2 = rtmp22 + prow;
2092628f99d7SShri Abhyankar           if (*pc1 != 0.0 || *pc2 != 0.0) {
2093628f99d7SShri Abhyankar             pv   = ba + bd[prow];
2094628f99d7SShri Abhyankar             pj   = nbj + bd[prow];
2095628f99d7SShri Abhyankar             mul1 = *pc1 * *pv;
2096628f99d7SShri Abhyankar             mul2 = *pc2 * *pv;
2097628f99d7SShri Abhyankar             ++pv;
2098628f99d7SShri Abhyankar             *pc1 = mul1;
2099628f99d7SShri Abhyankar             *pc2 = mul2;
2100628f99d7SShri Abhyankar 
2101628f99d7SShri Abhyankar             nz_tmp = bi[prow+1] - bd[prow] - 1;
2102628f99d7SShri Abhyankar             for (j=0; j<nz_tmp; j++) {
2103628f99d7SShri Abhyankar               tmp          = pv[j];
2104628f99d7SShri Abhyankar               idx          = pj[j];
2105628f99d7SShri Abhyankar               rtmp11[idx] -= mul1 * tmp;
2106628f99d7SShri Abhyankar               rtmp22[idx] -= mul2 * tmp;
2107628f99d7SShri Abhyankar             }
210828b1a77fSLisandro Dalcin             ierr = PetscLogFlops(2+4*nz_tmp);CHKERRQ(ierr);
2109628f99d7SShri Abhyankar           }
2110628f99d7SShri Abhyankar           prow = *bjtmp++;
2111628f99d7SShri Abhyankar         }
2112628f99d7SShri Abhyankar 
2113628f99d7SShri Abhyankar         /* Now take care of diagonal 2x2 block. Note: prow = row here */
2114628f99d7SShri Abhyankar         pc1 = rtmp11 + prow;
2115628f99d7SShri Abhyankar         pc2 = rtmp22 + prow;
2116628f99d7SShri Abhyankar 
2117628f99d7SShri Abhyankar         sctx.pv = *pc1;
2118628f99d7SShri Abhyankar         pj      = bj + bi[prow];
2119628f99d7SShri Abhyankar         rs      = 0.0;
2120628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2121628f99d7SShri Abhyankar           idx = pj[j];
2122628f99d7SShri Abhyankar           if (idx != prow) rs += PetscAbsScalar(rtmp11[idx]);
2123628f99d7SShri Abhyankar         }
2124628f99d7SShri Abhyankar         sctx.rs = rs;
2125d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,row);CHKERRQ(ierr);
212607b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2127628f99d7SShri Abhyankar 
2128628f99d7SShri Abhyankar         if (*pc2 != 0.0) {
2129628f99d7SShri Abhyankar           pj     = nbj + bd[prow];
2130628f99d7SShri Abhyankar           mul2   = (*pc2)/(*pc1); /* since diag is not yet inverted.*/
2131628f99d7SShri Abhyankar           *pc2   = mul2;
2132628f99d7SShri Abhyankar           nz_tmp = bi[prow+1] - bd[prow] - 1;
2133628f99d7SShri Abhyankar           for (j=0; j<nz_tmp; j++) {
2134628f99d7SShri Abhyankar             idx          = pj[j];
2135628f99d7SShri Abhyankar             tmp          = rtmp11[idx];
2136628f99d7SShri Abhyankar             rtmp22[idx] -= mul2 * tmp;
2137628f99d7SShri Abhyankar           }
213828b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz_tmp);CHKERRQ(ierr);
2139628f99d7SShri Abhyankar         }
2140628f99d7SShri Abhyankar 
2141628f99d7SShri Abhyankar         pj  = bj + bi[row];
2142628f99d7SShri Abhyankar         pc1 = ba + bi[row];
2143628f99d7SShri Abhyankar         pc2 = ba + bi[row+1];
2144628f99d7SShri Abhyankar 
2145628f99d7SShri Abhyankar         sctx.pv       = rtmp22[row+1];
2146628f99d7SShri Abhyankar         rs            = 0.0;
2147628f99d7SShri Abhyankar         rtmp11[row]   = 1.0/rtmp11[row];
2148628f99d7SShri Abhyankar         rtmp22[row+1] = 1.0/rtmp22[row+1];
2149628f99d7SShri Abhyankar         /* copy row entries from dense representation to sparse */
2150628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2151628f99d7SShri Abhyankar           idx    = pj[j];
2152628f99d7SShri Abhyankar           pc1[j] = rtmp11[idx];
2153628f99d7SShri Abhyankar           pc2[j] = rtmp22[idx];
2154628f99d7SShri Abhyankar           if (idx != row+1) rs += PetscAbsScalar(pc2[j]);
2155628f99d7SShri Abhyankar         }
2156628f99d7SShri Abhyankar         sctx.rs = rs;
2157d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,row+1);CHKERRQ(ierr);
215807b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2159628f99d7SShri Abhyankar         break;
2160628f99d7SShri Abhyankar 
2161628f99d7SShri Abhyankar       case 3:
2162628f99d7SShri Abhyankar         for  (j=0; j<nz; j++) {
2163628f99d7SShri Abhyankar           idx         = bjtmp[j];
2164628f99d7SShri Abhyankar           rtmp11[idx] = 0.0;
2165628f99d7SShri Abhyankar           rtmp22[idx] = 0.0;
2166628f99d7SShri Abhyankar           rtmp33[idx] = 0.0;
2167628f99d7SShri Abhyankar         }
2168628f99d7SShri Abhyankar         /* copy the nonzeros for the 3 rows from sparse representation to dense in rtmp*[] */
2169628f99d7SShri Abhyankar         idx    = r[row];
2170628f99d7SShri Abhyankar         nz_tmp = ai[idx+1] - ai[idx];
2171628f99d7SShri Abhyankar         ajtmp  = aj + ai[idx];
2172628f99d7SShri Abhyankar         v1     = aa + ai[idx];
2173628f99d7SShri Abhyankar         v2     = aa + ai[idx+1];
2174628f99d7SShri Abhyankar         v3     = aa + ai[idx+2];
2175628f99d7SShri Abhyankar         for (j=0; j<nz_tmp; j++) {
2176628f99d7SShri Abhyankar           idx         = ics[ajtmp[j]];
2177628f99d7SShri Abhyankar           rtmp11[idx] = v1[j];
2178628f99d7SShri Abhyankar           rtmp22[idx] = v2[j];
2179628f99d7SShri Abhyankar           rtmp33[idx] = v3[j];
2180628f99d7SShri Abhyankar         }
2181628f99d7SShri Abhyankar         rtmp11[ics[r[row]]]   += sctx.shift_amount;
2182628f99d7SShri Abhyankar         rtmp22[ics[r[row+1]]] += sctx.shift_amount;
2183628f99d7SShri Abhyankar         rtmp33[ics[r[row+2]]] += sctx.shift_amount;
2184628f99d7SShri Abhyankar 
2185628f99d7SShri Abhyankar         /* loop over all pivot row blocks above this row block */
2186628f99d7SShri Abhyankar         prow = *bjtmp++;
2187628f99d7SShri Abhyankar         while (prow < row) {
2188628f99d7SShri Abhyankar           pc1 = rtmp11 + prow;
2189628f99d7SShri Abhyankar           pc2 = rtmp22 + prow;
2190628f99d7SShri Abhyankar           pc3 = rtmp33 + prow;
2191628f99d7SShri Abhyankar           if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 !=0.0) {
2192628f99d7SShri Abhyankar             pv   = ba  + bd[prow];
2193628f99d7SShri Abhyankar             pj   = nbj + bd[prow];
2194628f99d7SShri Abhyankar             mul1 = *pc1 * *pv;
2195628f99d7SShri Abhyankar             mul2 = *pc2 * *pv;
2196628f99d7SShri Abhyankar             mul3 = *pc3 * *pv;
2197628f99d7SShri Abhyankar             ++pv;
2198628f99d7SShri Abhyankar             *pc1 = mul1;
2199628f99d7SShri Abhyankar             *pc2 = mul2;
2200628f99d7SShri Abhyankar             *pc3 = mul3;
2201628f99d7SShri Abhyankar 
2202628f99d7SShri Abhyankar             nz_tmp = bi[prow+1] - bd[prow] - 1;
2203628f99d7SShri Abhyankar             /* update this row based on pivot row */
2204628f99d7SShri Abhyankar             for (j=0; j<nz_tmp; j++) {
2205628f99d7SShri Abhyankar               tmp          = pv[j];
2206628f99d7SShri Abhyankar               idx          = pj[j];
2207628f99d7SShri Abhyankar               rtmp11[idx] -= mul1 * tmp;
2208628f99d7SShri Abhyankar               rtmp22[idx] -= mul2 * tmp;
2209628f99d7SShri Abhyankar               rtmp33[idx] -= mul3 * tmp;
2210628f99d7SShri Abhyankar             }
221128b1a77fSLisandro Dalcin             ierr = PetscLogFlops(3+6*nz_tmp);CHKERRQ(ierr);
2212628f99d7SShri Abhyankar           }
2213628f99d7SShri Abhyankar           prow = *bjtmp++;
2214628f99d7SShri Abhyankar         }
2215628f99d7SShri Abhyankar 
2216628f99d7SShri Abhyankar         /* Now take care of diagonal 3x3 block in this set of rows */
2217628f99d7SShri Abhyankar         /* note: prow = row here */
2218628f99d7SShri Abhyankar         pc1 = rtmp11 + prow;
2219628f99d7SShri Abhyankar         pc2 = rtmp22 + prow;
2220628f99d7SShri Abhyankar         pc3 = rtmp33 + prow;
2221628f99d7SShri Abhyankar 
2222628f99d7SShri Abhyankar         sctx.pv = *pc1;
2223628f99d7SShri Abhyankar         pj      = bj + bi[prow];
2224628f99d7SShri Abhyankar         rs      = 0.0;
2225628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2226628f99d7SShri Abhyankar           idx = pj[j];
2227628f99d7SShri Abhyankar           if (idx != row) rs += PetscAbsScalar(rtmp11[idx]);
2228628f99d7SShri Abhyankar         }
2229628f99d7SShri Abhyankar         sctx.rs = rs;
2230d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,row);CHKERRQ(ierr);
223107b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2232628f99d7SShri Abhyankar 
2233628f99d7SShri Abhyankar         if (*pc2 != 0.0 || *pc3 != 0.0) {
2234628f99d7SShri Abhyankar           mul2   = (*pc2)/(*pc1);
2235628f99d7SShri Abhyankar           mul3   = (*pc3)/(*pc1);
2236628f99d7SShri Abhyankar           *pc2   = mul2;
2237628f99d7SShri Abhyankar           *pc3   = mul3;
2238628f99d7SShri Abhyankar           nz_tmp = bi[prow+1] - bd[prow] - 1;
2239628f99d7SShri Abhyankar           pj     = nbj + bd[prow];
2240628f99d7SShri Abhyankar           for (j=0; j<nz_tmp; j++) {
2241628f99d7SShri Abhyankar             idx          = pj[j];
2242628f99d7SShri Abhyankar             tmp          = rtmp11[idx];
2243628f99d7SShri Abhyankar             rtmp22[idx] -= mul2 * tmp;
2244628f99d7SShri Abhyankar             rtmp33[idx] -= mul3 * tmp;
2245628f99d7SShri Abhyankar           }
224628b1a77fSLisandro Dalcin           ierr = PetscLogFlops(2+4*nz_tmp);CHKERRQ(ierr);
2247628f99d7SShri Abhyankar         }
2248628f99d7SShri Abhyankar         ++prow;
2249628f99d7SShri Abhyankar 
2250628f99d7SShri Abhyankar         pc2     = rtmp22 + prow;
2251628f99d7SShri Abhyankar         pc3     = rtmp33 + prow;
2252628f99d7SShri Abhyankar         sctx.pv = *pc2;
2253628f99d7SShri Abhyankar         pj      = bj + bi[prow];
2254628f99d7SShri Abhyankar         rs      = 0.0;
2255628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2256628f99d7SShri Abhyankar           idx = pj[j];
2257628f99d7SShri Abhyankar           if (idx != prow) rs += PetscAbsScalar(rtmp22[idx]);
2258628f99d7SShri Abhyankar         }
2259628f99d7SShri Abhyankar         sctx.rs = rs;
2260d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,row+1);CHKERRQ(ierr);
226107b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2262628f99d7SShri Abhyankar 
2263628f99d7SShri Abhyankar         if (*pc3 != 0.0) {
2264628f99d7SShri Abhyankar           mul3   = (*pc3)/(*pc2);
2265628f99d7SShri Abhyankar           *pc3   = mul3;
2266628f99d7SShri Abhyankar           pj     = nbj + bd[prow];
2267628f99d7SShri Abhyankar           nz_tmp = bi[prow+1] - bd[prow] - 1;
2268628f99d7SShri Abhyankar           for (j=0; j<nz_tmp; j++) {
2269628f99d7SShri Abhyankar             idx          = pj[j];
2270628f99d7SShri Abhyankar             tmp          = rtmp22[idx];
2271628f99d7SShri Abhyankar             rtmp33[idx] -= mul3 * tmp;
2272628f99d7SShri Abhyankar           }
227328b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz_tmp);CHKERRQ(ierr);
2274628f99d7SShri Abhyankar         }
2275628f99d7SShri Abhyankar 
2276628f99d7SShri Abhyankar         pj  = bj + bi[row];
2277628f99d7SShri Abhyankar         pc1 = ba + bi[row];
2278628f99d7SShri Abhyankar         pc2 = ba + bi[row+1];
2279628f99d7SShri Abhyankar         pc3 = ba + bi[row+2];
2280628f99d7SShri Abhyankar 
2281628f99d7SShri Abhyankar         sctx.pv       = rtmp33[row+2];
2282628f99d7SShri Abhyankar         rs            = 0.0;
2283628f99d7SShri Abhyankar         rtmp11[row]   = 1.0/rtmp11[row];
2284628f99d7SShri Abhyankar         rtmp22[row+1] = 1.0/rtmp22[row+1];
2285628f99d7SShri Abhyankar         rtmp33[row+2] = 1.0/rtmp33[row+2];
2286628f99d7SShri Abhyankar         /* copy row entries from dense representation to sparse */
2287628f99d7SShri Abhyankar         for (j=0; j<nz; j++) {
2288628f99d7SShri Abhyankar           idx    = pj[j];
2289628f99d7SShri Abhyankar           pc1[j] = rtmp11[idx];
2290628f99d7SShri Abhyankar           pc2[j] = rtmp22[idx];
2291628f99d7SShri Abhyankar           pc3[j] = rtmp33[idx];
2292628f99d7SShri Abhyankar           if (idx != row+2) rs += PetscAbsScalar(pc3[j]);
2293628f99d7SShri Abhyankar         }
2294628f99d7SShri Abhyankar 
2295628f99d7SShri Abhyankar         sctx.rs = rs;
2296d0660788SBarry Smith         ierr    = MatPivotCheck(A,info,&sctx,row+2);CHKERRQ(ierr);
229707b50cabSHong Zhang         if (sctx.newshift) goto endofwhile;
2298628f99d7SShri Abhyankar         break;
2299628f99d7SShri Abhyankar 
2300628f99d7SShri Abhyankar       default:
2301e32f2f54SBarry Smith         SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Node size not yet supported \n");
2302628f99d7SShri Abhyankar       }
2303628f99d7SShri Abhyankar       row += nodesz;                 /* Update the row */
2304628f99d7SShri Abhyankar     }
2305628f99d7SShri Abhyankar endofwhile:;
230607b50cabSHong Zhang   } while (sctx.newshift);
2307eb9baa12SBarry Smith   ierr = PetscFree3(rtmp11,rtmp22,rtmp33);CHKERRQ(ierr);
2308628f99d7SShri Abhyankar   ierr = PetscFree(tmp_vec2);CHKERRQ(ierr);
2309628f99d7SShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2310628f99d7SShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
2311628f99d7SShri Abhyankar   ierr = ISRestoreIndices(iscol,&c);CHKERRQ(ierr);
23122205254eSKarl Rupp 
2313d3ac4fa3SBarry Smith   (B)->ops->solve = MatSolve_SeqAIJ_inplace;
2314628f99d7SShri Abhyankar   /* do not set solve add, since MatSolve_Inode + Add is faster */
2315628f99d7SShri Abhyankar   C->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
2316628f99d7SShri Abhyankar   C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
2317628f99d7SShri Abhyankar   C->assembled              = PETSC_TRUE;
2318628f99d7SShri Abhyankar   C->preallocated           = PETSC_TRUE;
2319628f99d7SShri Abhyankar   if (sctx.nshift) {
2320f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
2321628f99d7SShri Abhyankar       ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr);
2322f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
2323628f99d7SShri Abhyankar       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
2324628f99d7SShri Abhyankar     }
2325628f99d7SShri Abhyankar   }
2326628f99d7SShri Abhyankar   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
2327d3ac4fa3SBarry Smith   ierr = Mat_CheckInode(C,PETSC_FALSE);CHKERRQ(ierr);
2328628f99d7SShri Abhyankar   PetscFunctionReturn(0);
2329628f99d7SShri Abhyankar }
2330628f99d7SShri Abhyankar 
2331628f99d7SShri Abhyankar 
2332019b515eSShri Abhyankar /* ----------------------------------------------------------- */
2333019b515eSShri Abhyankar #undef __FUNCT__
233435233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_Inode"
233535233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_Inode(Mat A,Vec bb,Vec xx)
2336019b515eSShri Abhyankar {
2337019b515eSShri Abhyankar   Mat_SeqAIJ        *a    = (Mat_SeqAIJ*)A->data;
2338019b515eSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
2339019b515eSShri Abhyankar   PetscErrorCode    ierr;
2340019b515eSShri Abhyankar   const PetscInt    *r,*c,*rout,*cout;
23418758e1faSBarry Smith   PetscInt          i,j,n = A->rmap->n;
23428758e1faSBarry Smith   PetscInt          node_max,row,nsz,aii,i0,i1,nz;
23438758e1faSBarry Smith   const PetscInt    *ai = a->i,*a_j = a->j,*ns,*vi,*ad,*aj;
2344019b515eSShri Abhyankar   PetscScalar       *x,*tmp,*tmps,tmp0,tmp1;
2345019b515eSShri Abhyankar   PetscScalar       sum1,sum2,sum3,sum4,sum5;
2346019b515eSShri Abhyankar   const MatScalar   *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa;
2347019b515eSShri Abhyankar   const PetscScalar *b;
2348019b515eSShri Abhyankar 
2349019b515eSShri Abhyankar   PetscFunctionBegin;
2350e32f2f54SBarry Smith   if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure");
2351019b515eSShri Abhyankar   node_max = a->inode.node_count;
2352019b515eSShri Abhyankar   ns       = a->inode.size;     /* Node Size array */
2353019b515eSShri Abhyankar 
23543649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
2355019b515eSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2356019b515eSShri Abhyankar   tmp  = a->solve_work;
2357019b515eSShri Abhyankar 
2358019b515eSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
2359019b515eSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
2360019b515eSShri Abhyankar 
2361019b515eSShri Abhyankar   /* forward solve the lower triangular */
2362019b515eSShri Abhyankar   tmps = tmp;
2363019b515eSShri Abhyankar   aa   = a_a;
2364019b515eSShri Abhyankar   aj   = a_j;
2365019b515eSShri Abhyankar   ad   = a->diag;
2366019b515eSShri Abhyankar 
2367019b515eSShri Abhyankar   for (i = 0,row = 0; i< node_max; ++i) {
2368019b515eSShri Abhyankar     nsz = ns[i];
2369019b515eSShri Abhyankar     aii = ai[row];
2370019b515eSShri Abhyankar     v1  = aa + aii;
2371019b515eSShri Abhyankar     vi  = aj + aii;
2372019b515eSShri Abhyankar     nz  = ai[row+1]- ai[row];
2373019b515eSShri Abhyankar 
237498991853SShri Abhyankar     if (i < node_max-1) {
237598991853SShri Abhyankar       /* Prefetch the indices for the next block */
237650d8bf02SJed Brown       PetscPrefetchBlock(aj+ai[row+nsz],ai[row+nsz+1]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); /* indices */
237798991853SShri Abhyankar       /* Prefetch the data for the next block */
237850d8bf02SJed Brown       PetscPrefetchBlock(aa+ai[row+nsz],ai[row+nsz+ns[i+1]]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA);
237998991853SShri Abhyankar     }
238098991853SShri Abhyankar 
2381019b515eSShri Abhyankar     switch (nsz) {               /* Each loop in 'case' is unrolled */
2382019b515eSShri Abhyankar     case 1:
2383019b515eSShri Abhyankar       sum1 = b[r[row]];
2384019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2385019b515eSShri Abhyankar         i0    = vi[j];
2386019b515eSShri Abhyankar         i1    = vi[j+1];
2387019b515eSShri Abhyankar         tmp0  = tmps[i0];
2388019b515eSShri Abhyankar         tmp1  = tmps[i1];
2389019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0 + v1[j+1]*tmp1;
2390019b515eSShri Abhyankar       }
2391019b515eSShri Abhyankar       if (j == nz-1) {
2392019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2393019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0;
2394019b515eSShri Abhyankar       }
2395019b515eSShri Abhyankar       tmp[row++]=sum1;
2396019b515eSShri Abhyankar       break;
2397019b515eSShri Abhyankar     case 2:
2398019b515eSShri Abhyankar       sum1 = b[r[row]];
2399019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2400019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2401019b515eSShri Abhyankar 
2402019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2403019b515eSShri Abhyankar         i0    = vi[j];
2404019b515eSShri Abhyankar         i1    = vi[j+1];
2405019b515eSShri Abhyankar         tmp0  = tmps[i0];
2406019b515eSShri Abhyankar         tmp1  = tmps[i1];
2407019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2408019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2409019b515eSShri Abhyankar       }
2410019b515eSShri Abhyankar       if (j == nz-1) {
2411019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2412019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2413019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2414019b515eSShri Abhyankar       }
2415019b515eSShri Abhyankar       sum2     -= v2[nz] * sum1;
2416019b515eSShri Abhyankar       tmp[row++]=sum1;
2417019b515eSShri Abhyankar       tmp[row++]=sum2;
2418019b515eSShri Abhyankar       break;
2419019b515eSShri Abhyankar     case 3:
2420019b515eSShri Abhyankar       sum1 = b[r[row]];
2421019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2422019b515eSShri Abhyankar       sum3 = b[r[row+2]];
2423019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2424019b515eSShri Abhyankar       v3   = aa + ai[row+2];
2425019b515eSShri Abhyankar 
2426019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2427019b515eSShri Abhyankar         i0    = vi[j];
2428019b515eSShri Abhyankar         i1    = vi[j+1];
2429019b515eSShri Abhyankar         tmp0  = tmps[i0];
2430019b515eSShri Abhyankar         tmp1  = tmps[i1];
2431019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2432019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2433019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
2434019b515eSShri Abhyankar       }
2435019b515eSShri Abhyankar       if (j == nz-1) {
2436019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2437019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2438019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2439019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
2440019b515eSShri Abhyankar       }
2441019b515eSShri Abhyankar       sum2     -= v2[nz] * sum1;
2442019b515eSShri Abhyankar       sum3     -= v3[nz] * sum1;
2443019b515eSShri Abhyankar       sum3     -= v3[nz+1] * sum2;
2444019b515eSShri Abhyankar       tmp[row++]=sum1;
2445019b515eSShri Abhyankar       tmp[row++]=sum2;
2446019b515eSShri Abhyankar       tmp[row++]=sum3;
2447019b515eSShri Abhyankar       break;
2448019b515eSShri Abhyankar 
2449019b515eSShri Abhyankar     case 4:
2450019b515eSShri Abhyankar       sum1 = b[r[row]];
2451019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2452019b515eSShri Abhyankar       sum3 = b[r[row+2]];
2453019b515eSShri Abhyankar       sum4 = b[r[row+3]];
2454019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2455019b515eSShri Abhyankar       v3   = aa + ai[row+2];
2456019b515eSShri Abhyankar       v4   = aa + ai[row+3];
2457019b515eSShri Abhyankar 
2458019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2459019b515eSShri Abhyankar         i0    = vi[j];
2460019b515eSShri Abhyankar         i1    = vi[j+1];
2461019b515eSShri Abhyankar         tmp0  = tmps[i0];
2462019b515eSShri Abhyankar         tmp1  = tmps[i1];
2463019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2464019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2465019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
2466019b515eSShri Abhyankar         sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1;
2467019b515eSShri Abhyankar       }
2468019b515eSShri Abhyankar       if (j == nz-1) {
2469019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2470019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2471019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2472019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
2473019b515eSShri Abhyankar         sum4 -= v4[j] *tmp0;
2474019b515eSShri Abhyankar       }
2475019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
2476019b515eSShri Abhyankar       sum3 -= v3[nz] * sum1;
2477019b515eSShri Abhyankar       sum4 -= v4[nz] * sum1;
2478019b515eSShri Abhyankar       sum3 -= v3[nz+1] * sum2;
2479019b515eSShri Abhyankar       sum4 -= v4[nz+1] * sum2;
2480019b515eSShri Abhyankar       sum4 -= v4[nz+2] * sum3;
2481019b515eSShri Abhyankar 
2482019b515eSShri Abhyankar       tmp[row++]=sum1;
2483019b515eSShri Abhyankar       tmp[row++]=sum2;
2484019b515eSShri Abhyankar       tmp[row++]=sum3;
2485019b515eSShri Abhyankar       tmp[row++]=sum4;
2486019b515eSShri Abhyankar       break;
2487019b515eSShri Abhyankar     case 5:
2488019b515eSShri Abhyankar       sum1 = b[r[row]];
2489019b515eSShri Abhyankar       sum2 = b[r[row+1]];
2490019b515eSShri Abhyankar       sum3 = b[r[row+2]];
2491019b515eSShri Abhyankar       sum4 = b[r[row+3]];
2492019b515eSShri Abhyankar       sum5 = b[r[row+4]];
2493019b515eSShri Abhyankar       v2   = aa + ai[row+1];
2494019b515eSShri Abhyankar       v3   = aa + ai[row+2];
2495019b515eSShri Abhyankar       v4   = aa + ai[row+3];
2496019b515eSShri Abhyankar       v5   = aa + ai[row+4];
2497019b515eSShri Abhyankar 
2498019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2499019b515eSShri Abhyankar         i0    = vi[j];
2500019b515eSShri Abhyankar         i1    = vi[j+1];
2501019b515eSShri Abhyankar         tmp0  = tmps[i0];
2502019b515eSShri Abhyankar         tmp1  = tmps[i1];
2503019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2504019b515eSShri Abhyankar         sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1;
2505019b515eSShri Abhyankar         sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1;
2506019b515eSShri Abhyankar         sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1;
2507019b515eSShri Abhyankar         sum5 -= v5[j] * tmp0 + v5[j+1] * tmp1;
2508019b515eSShri Abhyankar       }
2509019b515eSShri Abhyankar       if (j == nz-1) {
2510019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2511019b515eSShri Abhyankar         sum1 -= v1[j] *tmp0;
2512019b515eSShri Abhyankar         sum2 -= v2[j] *tmp0;
2513019b515eSShri Abhyankar         sum3 -= v3[j] *tmp0;
2514019b515eSShri Abhyankar         sum4 -= v4[j] *tmp0;
2515019b515eSShri Abhyankar         sum5 -= v5[j] *tmp0;
2516019b515eSShri Abhyankar       }
2517019b515eSShri Abhyankar 
2518019b515eSShri Abhyankar       sum2 -= v2[nz] * sum1;
2519019b515eSShri Abhyankar       sum3 -= v3[nz] * sum1;
2520019b515eSShri Abhyankar       sum4 -= v4[nz] * sum1;
2521019b515eSShri Abhyankar       sum5 -= v5[nz] * sum1;
2522019b515eSShri Abhyankar       sum3 -= v3[nz+1] * sum2;
2523019b515eSShri Abhyankar       sum4 -= v4[nz+1] * sum2;
2524019b515eSShri Abhyankar       sum5 -= v5[nz+1] * sum2;
2525019b515eSShri Abhyankar       sum4 -= v4[nz+2] * sum3;
2526019b515eSShri Abhyankar       sum5 -= v5[nz+2] * sum3;
2527019b515eSShri Abhyankar       sum5 -= v5[nz+3] * sum4;
2528019b515eSShri Abhyankar 
2529019b515eSShri Abhyankar       tmp[row++]=sum1;
2530019b515eSShri Abhyankar       tmp[row++]=sum2;
2531019b515eSShri Abhyankar       tmp[row++]=sum3;
2532019b515eSShri Abhyankar       tmp[row++]=sum4;
2533019b515eSShri Abhyankar       tmp[row++]=sum5;
2534019b515eSShri Abhyankar       break;
2535019b515eSShri Abhyankar     default:
2536e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
2537019b515eSShri Abhyankar     }
2538019b515eSShri Abhyankar   }
2539019b515eSShri Abhyankar   /* backward solve the upper triangular */
2540019b515eSShri Abhyankar   for (i=node_max -1,row = n-1; i>=0; i--) {
2541019b515eSShri Abhyankar     nsz = ns[i];
2542019b515eSShri Abhyankar     aii = ad[row+1] + 1;
2543019b515eSShri Abhyankar     v1  = aa + aii;
2544019b515eSShri Abhyankar     vi  = aj + aii;
2545019b515eSShri Abhyankar     nz  = ad[row]- ad[row+1] - 1;
254698991853SShri Abhyankar 
254798991853SShri Abhyankar     if (i > 0) {
254898991853SShri Abhyankar       /* Prefetch the indices for the next block */
254950d8bf02SJed Brown       PetscPrefetchBlock(aj+ad[row-nsz+1]+1,ad[row-nsz]-ad[row-nsz+1],0,PETSC_PREFETCH_HINT_NTA);
255098991853SShri Abhyankar       /* Prefetch the data for the next block */
255150d8bf02SJed Brown       PetscPrefetchBlock(aa+ad[row-nsz+1]+1,ad[row-nsz-ns[i-1]+1]-ad[row-nsz+1],0,PETSC_PREFETCH_HINT_NTA);
255298991853SShri Abhyankar     }
255398991853SShri Abhyankar 
2554019b515eSShri Abhyankar     switch (nsz) {               /* Each loop in 'case' is unrolled */
2555019b515eSShri Abhyankar     case 1:
2556019b515eSShri Abhyankar       sum1 = tmp[row];
2557019b515eSShri Abhyankar 
2558019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2559019b515eSShri Abhyankar         i0    = vi[j];
2560019b515eSShri Abhyankar         i1    = vi[j+1];
2561019b515eSShri Abhyankar         tmp0  = tmps[i0];
2562019b515eSShri Abhyankar         tmp1  = tmps[i1];
2563019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2564019b515eSShri Abhyankar       }
2565019b515eSShri Abhyankar       if (j == nz-1) {
2566019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2567019b515eSShri Abhyankar         sum1 -= v1[j]*tmp0;
2568019b515eSShri Abhyankar       }
2569019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2570019b515eSShri Abhyankar       break;
2571019b515eSShri Abhyankar     case 2:
2572019b515eSShri Abhyankar       sum1 = tmp[row];
2573019b515eSShri Abhyankar       sum2 = tmp[row-1];
2574019b515eSShri Abhyankar       v2   = aa + ad[row] + 1;
2575019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2576019b515eSShri Abhyankar         i0    = vi[j];
2577019b515eSShri Abhyankar         i1    = vi[j+1];
2578019b515eSShri Abhyankar         tmp0  = tmps[i0];
2579019b515eSShri Abhyankar         tmp1  = tmps[i1];
2580019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2581019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2582019b515eSShri Abhyankar       }
2583019b515eSShri Abhyankar       if (j == nz-1) {
2584019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2585019b515eSShri Abhyankar         sum1 -= v1[j]* tmp0;
2586019b515eSShri Abhyankar         sum2 -= v2[j+1]* tmp0;
2587019b515eSShri Abhyankar       }
2588019b515eSShri Abhyankar 
2589019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2590019b515eSShri Abhyankar       sum2     -= v2[0] * tmp0;
2591019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2592019b515eSShri Abhyankar       break;
2593019b515eSShri Abhyankar     case 3:
2594019b515eSShri Abhyankar       sum1 = tmp[row];
2595019b515eSShri Abhyankar       sum2 = tmp[row -1];
2596019b515eSShri Abhyankar       sum3 = tmp[row -2];
2597019b515eSShri Abhyankar       v2   = aa + ad[row] + 1;
2598019b515eSShri Abhyankar       v3   = aa + ad[row -1] + 1;
2599019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2600019b515eSShri Abhyankar         i0    = vi[j];
2601019b515eSShri Abhyankar         i1    = vi[j+1];
2602019b515eSShri Abhyankar         tmp0  = tmps[i0];
2603019b515eSShri Abhyankar         tmp1  = tmps[i1];
2604019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2605019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2606019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
2607019b515eSShri Abhyankar       }
2608019b515eSShri Abhyankar       if (j== nz-1) {
2609019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2610019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
2611019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
2612019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
2613019b515eSShri Abhyankar       }
2614019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2615019b515eSShri Abhyankar       sum2     -= v2[0]* tmp0;
2616019b515eSShri Abhyankar       sum3     -= v3[1] * tmp0;
2617019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2618019b515eSShri Abhyankar       sum3     -= v3[0]* tmp0;
2619019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
2620019b515eSShri Abhyankar 
2621019b515eSShri Abhyankar       break;
2622019b515eSShri Abhyankar     case 4:
2623019b515eSShri Abhyankar       sum1 = tmp[row];
2624019b515eSShri Abhyankar       sum2 = tmp[row -1];
2625019b515eSShri Abhyankar       sum3 = tmp[row -2];
2626019b515eSShri Abhyankar       sum4 = tmp[row -3];
2627019b515eSShri Abhyankar       v2   = aa + ad[row]+1;
2628019b515eSShri Abhyankar       v3   = aa + ad[row -1]+1;
2629019b515eSShri Abhyankar       v4   = aa + ad[row -2]+1;
2630019b515eSShri Abhyankar 
2631019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2632019b515eSShri Abhyankar         i0    = vi[j];
2633019b515eSShri Abhyankar         i1    = vi[j+1];
2634019b515eSShri Abhyankar         tmp0  = tmps[i0];
2635019b515eSShri Abhyankar         tmp1  = tmps[i1];
2636019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0   + v1[j+1] * tmp1;
2637019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2638019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
2639019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1;
2640019b515eSShri Abhyankar       }
2641019b515eSShri Abhyankar       if (j== nz-1) {
2642019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2643019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
2644019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
2645019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
2646019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0;
2647019b515eSShri Abhyankar       }
2648019b515eSShri Abhyankar 
2649019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2650019b515eSShri Abhyankar       sum2     -= v2[0] * tmp0;
2651019b515eSShri Abhyankar       sum3     -= v3[1] * tmp0;
2652019b515eSShri Abhyankar       sum4     -= v4[2] * tmp0;
2653019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2654019b515eSShri Abhyankar       sum3     -= v3[0] * tmp0;
2655019b515eSShri Abhyankar       sum4     -= v4[1] * tmp0;
2656019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
2657019b515eSShri Abhyankar       sum4     -= v4[0] * tmp0;
2658019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--;
2659019b515eSShri Abhyankar       break;
2660019b515eSShri Abhyankar     case 5:
2661019b515eSShri Abhyankar       sum1 = tmp[row];
2662019b515eSShri Abhyankar       sum2 = tmp[row -1];
2663019b515eSShri Abhyankar       sum3 = tmp[row -2];
2664019b515eSShri Abhyankar       sum4 = tmp[row -3];
2665019b515eSShri Abhyankar       sum5 = tmp[row -4];
2666019b515eSShri Abhyankar       v2   = aa + ad[row]+1;
2667019b515eSShri Abhyankar       v3   = aa + ad[row -1]+1;
2668019b515eSShri Abhyankar       v4   = aa + ad[row -2]+1;
2669019b515eSShri Abhyankar       v5   = aa + ad[row -3]+1;
2670019b515eSShri Abhyankar       for (j=0; j<nz-1; j+=2) {
2671019b515eSShri Abhyankar         i0    = vi[j];
2672019b515eSShri Abhyankar         i1    = vi[j+1];
2673019b515eSShri Abhyankar         tmp0  = tmps[i0];
2674019b515eSShri Abhyankar         tmp1  = tmps[i1];
2675019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1;
2676019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1;
2677019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1;
2678019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1;
2679019b515eSShri Abhyankar         sum5 -= v5[j+4] * tmp0 + v5[j+5] * tmp1;
2680019b515eSShri Abhyankar       }
2681019b515eSShri Abhyankar       if (j==nz-1) {
2682019b515eSShri Abhyankar         tmp0  = tmps[vi[j]];
2683019b515eSShri Abhyankar         sum1 -= v1[j] * tmp0;
2684019b515eSShri Abhyankar         sum2 -= v2[j+1] * tmp0;
2685019b515eSShri Abhyankar         sum3 -= v3[j+2] * tmp0;
2686019b515eSShri Abhyankar         sum4 -= v4[j+3] * tmp0;
2687019b515eSShri Abhyankar         sum5 -= v5[j+4] * tmp0;
2688019b515eSShri Abhyankar       }
2689019b515eSShri Abhyankar 
2690019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum1*v1[nz]; row--;
2691019b515eSShri Abhyankar       sum2     -= v2[0] * tmp0;
2692019b515eSShri Abhyankar       sum3     -= v3[1] * tmp0;
2693019b515eSShri Abhyankar       sum4     -= v4[2] * tmp0;
2694019b515eSShri Abhyankar       sum5     -= v5[3] * tmp0;
2695019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--;
2696019b515eSShri Abhyankar       sum3     -= v3[0] * tmp0;
2697019b515eSShri Abhyankar       sum4     -= v4[1] * tmp0;
2698019b515eSShri Abhyankar       sum5     -= v5[2] * tmp0;
2699019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--;
2700019b515eSShri Abhyankar       sum4     -= v4[0] * tmp0;
2701019b515eSShri Abhyankar       sum5     -= v5[1] * tmp0;
2702019b515eSShri Abhyankar       tmp0      = x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--;
2703019b515eSShri Abhyankar       sum5     -= v5[0] * tmp0;
2704019b515eSShri Abhyankar       x[c[row]] = tmp[row] = sum5*v5[nz+4]; row--;
2705019b515eSShri Abhyankar       break;
2706019b515eSShri Abhyankar     default:
2707e32f2f54SBarry Smith       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n");
2708019b515eSShri Abhyankar     }
2709019b515eSShri Abhyankar   }
2710019b515eSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
2711019b515eSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
27123649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
2713019b515eSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
2714019b515eSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
2715019b515eSShri Abhyankar   PetscFunctionReturn(0);
2716019b515eSShri Abhyankar }
2717019b515eSShri Abhyankar 
2718019b515eSShri Abhyankar 
27194c1414c8SBarry Smith /*
27204c1414c8SBarry Smith      Makes a longer coloring[] array and calls the usual code with that
27214c1414c8SBarry Smith */
27224c1414c8SBarry Smith #undef __FUNCT__
27234108e4d5SBarry Smith #define __FUNCT__ "MatColoringPatch_SeqAIJ_Inode"
27244108e4d5SBarry Smith PetscErrorCode MatColoringPatch_SeqAIJ_Inode(Mat mat,PetscInt ncolors,PetscInt nin,ISColoringValue coloring[],ISColoring *iscoloring)
27254c1414c8SBarry Smith {
27264c1414c8SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)mat->data;
27274c1414c8SBarry Smith   PetscErrorCode  ierr;
2728d0f46423SBarry Smith   PetscInt        n = mat->cmap->n,m = a->inode.node_count,j,*ns = a->inode.size,row;
27294c1414c8SBarry Smith   PetscInt        *colorused,i;
27304c1414c8SBarry Smith   ISColoringValue *newcolor;
27314c1414c8SBarry Smith 
27324c1414c8SBarry Smith   PetscFunctionBegin;
27334c1414c8SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&newcolor);CHKERRQ(ierr);
27344c1414c8SBarry Smith   /* loop over inodes, marking a color for each column*/
27354c1414c8SBarry Smith   row = 0;
27364c1414c8SBarry Smith   for (i=0; i<m; i++) {
27374c1414c8SBarry Smith     for (j=0; j<ns[i]; j++) {
27384c1414c8SBarry Smith       newcolor[row++] = coloring[i] + j*ncolors;
27394c1414c8SBarry Smith     }
27404c1414c8SBarry Smith   }
27414c1414c8SBarry Smith 
27424c1414c8SBarry Smith   /* eliminate unneeded colors */
27434c1414c8SBarry Smith   ierr = PetscMalloc(5*ncolors*sizeof(PetscInt),&colorused);CHKERRQ(ierr);
27444c1414c8SBarry Smith   ierr = PetscMemzero(colorused,5*ncolors*sizeof(PetscInt));CHKERRQ(ierr);
27454c1414c8SBarry Smith   for (i=0; i<n; i++) {
27464c1414c8SBarry Smith     colorused[newcolor[i]] = 1;
27474c1414c8SBarry Smith   }
27484c1414c8SBarry Smith 
27494c1414c8SBarry Smith   for (i=1; i<5*ncolors; i++) {
27504c1414c8SBarry Smith     colorused[i] += colorused[i-1];
27514c1414c8SBarry Smith   }
27524c1414c8SBarry Smith   ncolors = colorused[5*ncolors-1];
27534c1414c8SBarry Smith   for (i=0; i<n; i++) {
275457815681SSatish Balay     newcolor[i] = colorused[newcolor[i]]-1;
27554c1414c8SBarry Smith   }
27564c1414c8SBarry Smith   ierr = PetscFree(colorused);CHKERRQ(ierr);
2757ce94432eSBarry Smith   ierr = ISColoringCreate(PetscObjectComm((PetscObject)mat),ncolors,n,newcolor,iscoloring);CHKERRQ(ierr);
27584c1414c8SBarry Smith   ierr = PetscFree(coloring);CHKERRQ(ierr);
27594c1414c8SBarry Smith   PetscFunctionReturn(0);
27604c1414c8SBarry Smith }
27614c1414c8SBarry Smith 
276206873bf2SBarry Smith #include <petsc-private/kernels/blockinvert.h>
27632af78befSBarry Smith 
27642af78befSBarry Smith #undef __FUNCT__
27654108e4d5SBarry Smith #define __FUNCT__ "MatSOR_SeqAIJ_Inode"
27664108e4d5SBarry Smith PetscErrorCode MatSOR_SeqAIJ_Inode(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx)
27672af78befSBarry Smith {
27682af78befSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
27697aaeff0aSMatthew G. Knepley   PetscScalar       sum1 = 0.0,sum2 = 0.0,sum3 = 0.0,sum4 = 0.0,sum5 = 0.0,tmp0,tmp1,tmp2,tmp3;
27705850ef23SBarry Smith   MatScalar         *ibdiag,*bdiag,work[25],*t;
2771a8b09249SBarry Smith   PetscScalar       *x,tmp4,tmp5,x1,x2,x3,x4,x5;
27727aaeff0aSMatthew G. Knepley   const MatScalar   *v = a->a,*v1 = NULL,*v2 = NULL,*v3 = NULL,*v4 = NULL,*v5 = NULL;
27735850ef23SBarry Smith   const PetscScalar *xb, *b;
277462bba022SBarry Smith   PetscReal         zeropivot = 1.0e-15, shift = 0.0;
27752af78befSBarry Smith   PetscErrorCode    ierr;
27768758e1faSBarry Smith   PetscInt          n,m = a->inode.node_count,cnt = 0,i,j,row,i1,i2;
27778758e1faSBarry Smith   PetscInt          sz,k,ipvt[5];
27788758e1faSBarry Smith   const PetscInt    *sizes = a->inode.size,*idx,*diag = a->diag,*ii = a->i;
27792af78befSBarry Smith 
27802af78befSBarry Smith   PetscFunctionBegin;
2781e32f2f54SBarry Smith   if (omega != 1.0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for omega != 1.0; use -mat_no_inode");
2782e32f2f54SBarry Smith   if (fshift != 0.0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for fshift != 0.0; use -mat_no_inode");
27832af78befSBarry Smith 
278471f1c65dSBarry Smith   if (!a->inode.ibdiagvalid) {
27852af78befSBarry Smith     if (!a->inode.ibdiag) {
27862af78befSBarry Smith       /* calculate space needed for diagonal blocks */
27872af78befSBarry Smith       for (i=0; i<m; i++) {
27882af78befSBarry Smith         cnt += sizes[i]*sizes[i];
27892af78befSBarry Smith       }
2790f0d39aaaSBarry Smith       a->inode.bdiagsize = cnt;
27912205254eSKarl Rupp 
2792*dcca6d9dSJed Brown       ierr = PetscMalloc3(cnt,&a->inode.ibdiag,cnt,&a->inode.bdiag,A->rmap->n,&a->inode.ssor_work);CHKERRQ(ierr);
279371f1c65dSBarry Smith     }
279471f1c65dSBarry Smith 
279571f1c65dSBarry Smith     /* copy over the diagonal blocks and invert them */
27962af78befSBarry Smith     ibdiag = a->inode.ibdiag;
27972af78befSBarry Smith     bdiag  = a->inode.bdiag;
27982af78befSBarry Smith     cnt    = 0;
27992af78befSBarry Smith     for (i=0, row = 0; i<m; i++) {
28002af78befSBarry Smith       for (j=0; j<sizes[i]; j++) {
2801f0d39aaaSBarry Smith         for (k=0; k<sizes[i]; k++) {
2802f0d39aaaSBarry Smith           bdiag[cnt+k*sizes[i]+j] = v[diag[row+j] - j + k];
2803f0d39aaaSBarry Smith         }
28042af78befSBarry Smith       }
2805a77337e4SBarry Smith       ierr = PetscMemcpy(ibdiag+cnt,bdiag+cnt,sizes[i]*sizes[i]*sizeof(MatScalar));CHKERRQ(ierr);
28062af78befSBarry Smith 
28072af78befSBarry Smith       switch (sizes[i]) {
28082af78befSBarry Smith       case 1:
28092af78befSBarry Smith         /* Create matrix data structure */
2810e32f2f54SBarry Smith         if (PetscAbsScalar(ibdiag[cnt]) < zeropivot) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot on row %D",row);
281164c62002SMatthew Knepley         ibdiag[cnt] = 1.0/ibdiag[cnt];
28122af78befSBarry Smith         break;
28132af78befSBarry Smith       case 2:
281496b95a6bSBarry Smith         ierr = PetscKernel_A_gets_inverse_A_2(ibdiag+cnt,shift);CHKERRQ(ierr);
28152af78befSBarry Smith         break;
28162af78befSBarry Smith       case 3:
281796b95a6bSBarry Smith         ierr = PetscKernel_A_gets_inverse_A_3(ibdiag+cnt,shift);CHKERRQ(ierr);
28182af78befSBarry Smith         break;
28192af78befSBarry Smith       case 4:
282096b95a6bSBarry Smith         ierr = PetscKernel_A_gets_inverse_A_4(ibdiag+cnt,shift);CHKERRQ(ierr);
28212af78befSBarry Smith         break;
28222af78befSBarry Smith       case 5:
282396b95a6bSBarry Smith         ierr = PetscKernel_A_gets_inverse_A_5(ibdiag+cnt,ipvt,work,shift);CHKERRQ(ierr);
28242af78befSBarry Smith         break;
28252af78befSBarry Smith       default:
2826e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
28272af78befSBarry Smith       }
28282af78befSBarry Smith       cnt += sizes[i]*sizes[i];
28292af78befSBarry Smith       row += sizes[i];
28302af78befSBarry Smith     }
283171f1c65dSBarry Smith     a->inode.ibdiagvalid = PETSC_TRUE;
28322af78befSBarry Smith   }
28332af78befSBarry Smith   ibdiag = a->inode.ibdiag;
28342af78befSBarry Smith   bdiag  = a->inode.bdiag;
28355850ef23SBarry Smith   t      = a->inode.ssor_work;
28362af78befSBarry Smith 
28372af78befSBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
28383649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
28395850ef23SBarry Smith   /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */
28405850ef23SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
28412af78befSBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) {
28422af78befSBarry Smith 
28438862d2efSBarry Smith       for (i=0, row=0; i<m; i++) {
28448862d2efSBarry Smith         sz  = diag[row] - ii[row];
28458862d2efSBarry Smith         v1  = a->a + ii[row];
28468862d2efSBarry Smith         idx = a->j + ii[row];
28478862d2efSBarry Smith 
28484108e4d5SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
28498862d2efSBarry Smith         switch (sizes[i]) {
28508862d2efSBarry Smith         case 1:
28518862d2efSBarry Smith 
28528862d2efSBarry Smith           sum1 = b[row];
28538862d2efSBarry Smith           for (n = 0; n<sz-1; n+=2) {
28548862d2efSBarry Smith             i1    = idx[0];
28558862d2efSBarry Smith             i2    = idx[1];
28568862d2efSBarry Smith             idx  += 2;
28578862d2efSBarry Smith             tmp0  = x[i1];
28588862d2efSBarry Smith             tmp1  = x[i2];
28598862d2efSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
28608862d2efSBarry Smith           }
28618862d2efSBarry Smith 
28628862d2efSBarry Smith           if (n == sz-1) {
2863f0d39aaaSBarry Smith             tmp0  = x[*idx];
2864f0d39aaaSBarry Smith             sum1 -= *v1 * tmp0;
28658862d2efSBarry Smith           }
28665850ef23SBarry Smith           t[row]   = sum1;
28678862d2efSBarry Smith           x[row++] = sum1*(*ibdiag++);
28688862d2efSBarry Smith           break;
2869f0d39aaaSBarry Smith         case 2:
2870f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2871f0d39aaaSBarry Smith           sum1 = b[row];
2872f0d39aaaSBarry Smith           sum2 = b[row+1];
2873f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2874f0d39aaaSBarry Smith             i1    = idx[0];
2875f0d39aaaSBarry Smith             i2    = idx[1];
2876f0d39aaaSBarry Smith             idx  += 2;
2877f0d39aaaSBarry Smith             tmp0  = x[i1];
2878f0d39aaaSBarry Smith             tmp1  = x[i2];
2879f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2880f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2881f0d39aaaSBarry Smith           }
2882f0d39aaaSBarry Smith 
2883f0d39aaaSBarry Smith           if (n == sz-1) {
2884f0d39aaaSBarry Smith             tmp0  = x[*idx];
2885f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2886f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2887f0d39aaaSBarry Smith           }
28885850ef23SBarry Smith           t[row]   = sum1;
28895850ef23SBarry Smith           t[row+1] = sum2;
2890f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[2];
2891f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[3];
2892f0d39aaaSBarry Smith           ibdiag  += 4;
2893f0d39aaaSBarry Smith           break;
2894f0d39aaaSBarry Smith         case 3:
2895f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2896f0d39aaaSBarry Smith           v3   = a->a + ii[row+2];
2897f0d39aaaSBarry Smith           sum1 = b[row];
2898f0d39aaaSBarry Smith           sum2 = b[row+1];
2899f0d39aaaSBarry Smith           sum3 = b[row+2];
2900f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2901f0d39aaaSBarry Smith             i1    = idx[0];
2902f0d39aaaSBarry Smith             i2    = idx[1];
2903f0d39aaaSBarry Smith             idx  += 2;
2904f0d39aaaSBarry Smith             tmp0  = x[i1];
2905f0d39aaaSBarry Smith             tmp1  = x[i2];
2906f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2907f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2908f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2909f0d39aaaSBarry Smith           }
2910f0d39aaaSBarry Smith 
2911f0d39aaaSBarry Smith           if (n == sz-1) {
2912f0d39aaaSBarry Smith             tmp0  = x[*idx];
2913f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2914f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2915f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0;
2916f0d39aaaSBarry Smith           }
29175850ef23SBarry Smith           t[row]   = sum1;
29185850ef23SBarry Smith           t[row+1] = sum2;
29195850ef23SBarry Smith           t[row+2] = sum3;
2920f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
2921f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
2922f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
2923f0d39aaaSBarry Smith           ibdiag  += 9;
2924f0d39aaaSBarry Smith           break;
2925f0d39aaaSBarry Smith         case 4:
2926f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2927f0d39aaaSBarry Smith           v3   = a->a + ii[row+2];
2928f0d39aaaSBarry Smith           v4   = a->a + ii[row+3];
2929f0d39aaaSBarry Smith           sum1 = b[row];
2930f0d39aaaSBarry Smith           sum2 = b[row+1];
2931f0d39aaaSBarry Smith           sum3 = b[row+2];
2932f0d39aaaSBarry Smith           sum4 = b[row+3];
2933f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2934f0d39aaaSBarry Smith             i1    = idx[0];
2935f0d39aaaSBarry Smith             i2    = idx[1];
2936f0d39aaaSBarry Smith             idx  += 2;
2937f0d39aaaSBarry Smith             tmp0  = x[i1];
2938f0d39aaaSBarry Smith             tmp1  = x[i2];
2939f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2940f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2941f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2942f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
2943f0d39aaaSBarry Smith           }
2944f0d39aaaSBarry Smith 
2945f0d39aaaSBarry Smith           if (n == sz-1) {
2946f0d39aaaSBarry Smith             tmp0  = x[*idx];
2947f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2948f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2949f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0;
2950f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0;
2951f0d39aaaSBarry Smith           }
29525850ef23SBarry Smith           t[row]   = sum1;
29535850ef23SBarry Smith           t[row+1] = sum2;
29545850ef23SBarry Smith           t[row+2] = sum3;
29555850ef23SBarry Smith           t[row+3] = sum4;
2956f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
2957f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
2958f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
2959f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
2960f0d39aaaSBarry Smith           ibdiag  += 16;
2961f0d39aaaSBarry Smith           break;
2962f0d39aaaSBarry Smith         case 5:
2963f0d39aaaSBarry Smith           v2   = a->a + ii[row+1];
2964f0d39aaaSBarry Smith           v3   = a->a + ii[row+2];
2965f0d39aaaSBarry Smith           v4   = a->a + ii[row+3];
2966f0d39aaaSBarry Smith           v5   = a->a + ii[row+4];
2967f0d39aaaSBarry Smith           sum1 = b[row];
2968f0d39aaaSBarry Smith           sum2 = b[row+1];
2969f0d39aaaSBarry Smith           sum3 = b[row+2];
2970f0d39aaaSBarry Smith           sum4 = b[row+3];
2971f0d39aaaSBarry Smith           sum5 = b[row+4];
2972f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
2973f0d39aaaSBarry Smith             i1    = idx[0];
2974f0d39aaaSBarry Smith             i2    = idx[1];
2975f0d39aaaSBarry Smith             idx  += 2;
2976f0d39aaaSBarry Smith             tmp0  = x[i1];
2977f0d39aaaSBarry Smith             tmp1  = x[i2];
2978f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
2979f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
2980f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
2981f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
2982f0d39aaaSBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
2983f0d39aaaSBarry Smith           }
2984f0d39aaaSBarry Smith 
2985f0d39aaaSBarry Smith           if (n == sz-1) {
2986f0d39aaaSBarry Smith             tmp0  = x[*idx];
2987f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0;
2988f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0;
2989f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0;
2990f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0;
2991f0d39aaaSBarry Smith             sum5 -= v5[0] * tmp0;
2992f0d39aaaSBarry Smith           }
29935850ef23SBarry Smith           t[row]   = sum1;
29945850ef23SBarry Smith           t[row+1] = sum2;
29955850ef23SBarry Smith           t[row+2] = sum3;
29965850ef23SBarry Smith           t[row+3] = sum4;
29975850ef23SBarry Smith           t[row+4] = sum5;
2998f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
2999f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
3000f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
3001f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
3002f0d39aaaSBarry Smith           x[row++] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
3003f0d39aaaSBarry Smith           ibdiag  += 25;
3004f0d39aaaSBarry Smith           break;
30058862d2efSBarry Smith         default:
3006e32f2f54SBarry Smith           SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
30078862d2efSBarry Smith         }
30082af78befSBarry Smith       }
30092af78befSBarry Smith 
30105850ef23SBarry Smith       xb   = t;
30112af78befSBarry Smith       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
30122af78befSBarry Smith     } else xb = b;
30132af78befSBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) {
30142af78befSBarry Smith 
3015f0d39aaaSBarry Smith       ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
3016d0f46423SBarry Smith       for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
3017f0d39aaaSBarry Smith         ibdiag -= sizes[i]*sizes[i];
30188862d2efSBarry Smith         sz      = ii[row+1] - diag[row] - 1;
30198862d2efSBarry Smith         v1      = a->a + diag[row] + 1;
30208862d2efSBarry Smith         idx     = a->j + diag[row] + 1;
30212af78befSBarry Smith 
30224108e4d5SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
30238862d2efSBarry Smith         switch (sizes[i]) {
30248862d2efSBarry Smith         case 1:
30258862d2efSBarry Smith 
30268862d2efSBarry Smith           sum1 = xb[row];
30278862d2efSBarry Smith           for (n = 0; n<sz-1; n+=2) {
30288862d2efSBarry Smith             i1    = idx[0];
30298862d2efSBarry Smith             i2    = idx[1];
30308862d2efSBarry Smith             idx  += 2;
30318862d2efSBarry Smith             tmp0  = x[i1];
30328862d2efSBarry Smith             tmp1  = x[i2];
30338862d2efSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
30348862d2efSBarry Smith           }
30358862d2efSBarry Smith 
30368862d2efSBarry Smith           if (n == sz-1) {
3037f0d39aaaSBarry Smith             tmp0  = x[*idx];
3038f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
30398862d2efSBarry Smith           }
3040f0d39aaaSBarry Smith           x[row--] = sum1*(*ibdiag);
3041f0d39aaaSBarry Smith           break;
3042f0d39aaaSBarry Smith 
3043f0d39aaaSBarry Smith         case 2:
3044f0d39aaaSBarry Smith 
3045f0d39aaaSBarry Smith           sum1 = xb[row];
3046f0d39aaaSBarry Smith           sum2 = xb[row-1];
3047f0d39aaaSBarry Smith           /* note that sum1 is associated with the second of the two rows */
3048f0d39aaaSBarry Smith           v2 = a->a + diag[row-1] + 2;
3049f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3050f0d39aaaSBarry Smith             i1    = idx[0];
3051f0d39aaaSBarry Smith             i2    = idx[1];
3052f0d39aaaSBarry Smith             idx  += 2;
3053f0d39aaaSBarry Smith             tmp0  = x[i1];
3054f0d39aaaSBarry Smith             tmp1  = x[i2];
3055f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3056f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3057f0d39aaaSBarry Smith           }
3058f0d39aaaSBarry Smith 
3059f0d39aaaSBarry Smith           if (n == sz-1) {
3060f0d39aaaSBarry Smith             tmp0  = x[*idx];
3061f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3062f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3063f0d39aaaSBarry Smith           }
3064f0d39aaaSBarry Smith           x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3];
3065f0d39aaaSBarry Smith           x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2];
3066f0d39aaaSBarry Smith           break;
3067f0d39aaaSBarry Smith         case 3:
3068f0d39aaaSBarry Smith 
3069f0d39aaaSBarry Smith           sum1 = xb[row];
3070f0d39aaaSBarry Smith           sum2 = xb[row-1];
3071f0d39aaaSBarry Smith           sum3 = xb[row-2];
3072f0d39aaaSBarry Smith           v2   = a->a + diag[row-1] + 2;
3073f0d39aaaSBarry Smith           v3   = a->a + diag[row-2] + 3;
3074f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3075f0d39aaaSBarry Smith             i1    = idx[0];
3076f0d39aaaSBarry Smith             i2    = idx[1];
3077f0d39aaaSBarry Smith             idx  += 2;
3078f0d39aaaSBarry Smith             tmp0  = x[i1];
3079f0d39aaaSBarry Smith             tmp1  = x[i2];
3080f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3081f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3082f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3083f0d39aaaSBarry Smith           }
3084f0d39aaaSBarry Smith 
3085f0d39aaaSBarry Smith           if (n == sz-1) {
3086f0d39aaaSBarry Smith             tmp0  = x[*idx];
3087f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3088f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3089f0d39aaaSBarry Smith             sum3 -= *v3*tmp0;
3090f0d39aaaSBarry Smith           }
3091f0d39aaaSBarry Smith           x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3092f0d39aaaSBarry Smith           x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3093f0d39aaaSBarry Smith           x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3094f0d39aaaSBarry Smith           break;
3095f0d39aaaSBarry Smith         case 4:
3096f0d39aaaSBarry Smith 
3097f0d39aaaSBarry Smith           sum1 = xb[row];
3098f0d39aaaSBarry Smith           sum2 = xb[row-1];
3099f0d39aaaSBarry Smith           sum3 = xb[row-2];
3100f0d39aaaSBarry Smith           sum4 = xb[row-3];
3101f0d39aaaSBarry Smith           v2   = a->a + diag[row-1] + 2;
3102f0d39aaaSBarry Smith           v3   = a->a + diag[row-2] + 3;
3103f0d39aaaSBarry Smith           v4   = a->a + diag[row-3] + 4;
3104f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3105f0d39aaaSBarry Smith             i1    = idx[0];
3106f0d39aaaSBarry Smith             i2    = idx[1];
3107f0d39aaaSBarry Smith             idx  += 2;
3108f0d39aaaSBarry Smith             tmp0  = x[i1];
3109f0d39aaaSBarry Smith             tmp1  = x[i2];
3110f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3111f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3112f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3113f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3114f0d39aaaSBarry Smith           }
3115f0d39aaaSBarry Smith 
3116f0d39aaaSBarry Smith           if (n == sz-1) {
3117f0d39aaaSBarry Smith             tmp0  = x[*idx];
3118f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3119f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3120f0d39aaaSBarry Smith             sum3 -= *v3*tmp0;
3121f0d39aaaSBarry Smith             sum4 -= *v4*tmp0;
3122f0d39aaaSBarry Smith           }
3123f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3124f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3125f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3126f0d39aaaSBarry Smith           x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3127f0d39aaaSBarry Smith           break;
3128f0d39aaaSBarry Smith         case 5:
3129f0d39aaaSBarry Smith 
3130f0d39aaaSBarry Smith           sum1 = xb[row];
3131f0d39aaaSBarry Smith           sum2 = xb[row-1];
3132f0d39aaaSBarry Smith           sum3 = xb[row-2];
3133f0d39aaaSBarry Smith           sum4 = xb[row-3];
3134f0d39aaaSBarry Smith           sum5 = xb[row-4];
3135f0d39aaaSBarry Smith           v2   = a->a + diag[row-1] + 2;
3136f0d39aaaSBarry Smith           v3   = a->a + diag[row-2] + 3;
3137f0d39aaaSBarry Smith           v4   = a->a + diag[row-3] + 4;
3138f0d39aaaSBarry Smith           v5   = a->a + diag[row-4] + 5;
3139f0d39aaaSBarry Smith           for (n = 0; n<sz-1; n+=2) {
3140f0d39aaaSBarry Smith             i1    = idx[0];
3141f0d39aaaSBarry Smith             i2    = idx[1];
3142f0d39aaaSBarry Smith             idx  += 2;
3143f0d39aaaSBarry Smith             tmp0  = x[i1];
3144f0d39aaaSBarry Smith             tmp1  = x[i2];
3145f0d39aaaSBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3146f0d39aaaSBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3147f0d39aaaSBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3148f0d39aaaSBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3149f0d39aaaSBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
3150f0d39aaaSBarry Smith           }
3151f0d39aaaSBarry Smith 
3152f0d39aaaSBarry Smith           if (n == sz-1) {
3153f0d39aaaSBarry Smith             tmp0  = x[*idx];
3154f0d39aaaSBarry Smith             sum1 -= *v1*tmp0;
3155f0d39aaaSBarry Smith             sum2 -= *v2*tmp0;
3156f0d39aaaSBarry Smith             sum3 -= *v3*tmp0;
3157f0d39aaaSBarry Smith             sum4 -= *v4*tmp0;
3158f0d39aaaSBarry Smith             sum5 -= *v5*tmp0;
3159f0d39aaaSBarry Smith           }
3160f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
3161f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
3162f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
3163f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
3164f0d39aaaSBarry Smith           x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
31658862d2efSBarry Smith           break;
31668862d2efSBarry Smith         default:
3167e32f2f54SBarry Smith           SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
31688862d2efSBarry Smith         }
31692af78befSBarry Smith       }
31702af78befSBarry Smith 
31712af78befSBarry Smith       ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
31722af78befSBarry Smith     }
31732af78befSBarry Smith     its--;
31745850ef23SBarry Smith   }
31755850ef23SBarry Smith   while (its--) {
31765850ef23SBarry Smith 
31775850ef23SBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) {
3178d876e2b0SMark Adams       for (i=0, row=0, ibdiag = a->inode.ibdiag;
3179d876e2b0SMark Adams            i<m;
3180d876e2b0SMark Adams            row += sizes[i], ibdiag += sizes[i]*sizes[i], i++) {
31815850ef23SBarry Smith 
3182d876e2b0SMark Adams         sz  = diag[row] - ii[row];
31835850ef23SBarry Smith         v1  = a->a + ii[row];
31845850ef23SBarry Smith         idx = a->j + ii[row];
31855850ef23SBarry Smith         /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
31865850ef23SBarry Smith         switch (sizes[i]) {
31875850ef23SBarry Smith         case 1:
31885850ef23SBarry Smith           sum1 = b[row];
31895850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
31905850ef23SBarry Smith             i1    = idx[0];
31915850ef23SBarry Smith             i2    = idx[1];
31925850ef23SBarry Smith             idx  += 2;
31935850ef23SBarry Smith             tmp0  = x[i1];
31945850ef23SBarry Smith             tmp1  = x[i2];
31955850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
31965850ef23SBarry Smith           }
31975850ef23SBarry Smith           if (n == sz-1) {
3198d876e2b0SMark Adams             tmp0  = x[*idx++];
3199d876e2b0SMark Adams             sum1 -= *v1 * tmp0;
3200d876e2b0SMark Adams             v1++;
3201d876e2b0SMark Adams           }
3202d876e2b0SMark Adams           t[row]   = sum1;
3203d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 1;
3204d876e2b0SMark Adams           idx     = a->j + diag[row] + 1;
3205d876e2b0SMark Adams           v1 += 1;
3206d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3207d876e2b0SMark Adams             i1    = idx[0];
3208d876e2b0SMark Adams             i2    = idx[1];
3209d876e2b0SMark Adams             idx  += 2;
3210d876e2b0SMark Adams             tmp0  = x[i1];
3211d876e2b0SMark Adams             tmp1  = x[i2];
3212d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3213d876e2b0SMark Adams           }
3214d876e2b0SMark Adams           if (n == sz-1) {
3215d876e2b0SMark Adams             tmp0  = x[*idx++];
32165850ef23SBarry Smith             sum1 -= *v1 * tmp0;
32175850ef23SBarry Smith           }
32185850ef23SBarry Smith           /* in MatSOR_SeqAIJ this line would be
32195850ef23SBarry Smith            *
32205850ef23SBarry Smith            * x[row] = (1-omega)*x[row]+(sum1+(*bdiag++)*x[row])*(*ibdiag++);
32215850ef23SBarry Smith            *
32225850ef23SBarry Smith            * but omega == 1, so this becomes
32235850ef23SBarry Smith            *
3224d876e2b0SMark Adams            * x[row] = sum1*(*ibdiag++);
32255850ef23SBarry Smith            *
32265850ef23SBarry Smith            */
3227d876e2b0SMark Adams           x[row] = sum1*(*ibdiag);
32285850ef23SBarry Smith           break;
32295850ef23SBarry Smith         case 2:
32305850ef23SBarry Smith           v2   = a->a + ii[row+1];
32315850ef23SBarry Smith           sum1 = b[row];
32325850ef23SBarry Smith           sum2 = b[row+1];
32335850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
32345850ef23SBarry Smith             i1    = idx[0];
32355850ef23SBarry Smith             i2    = idx[1];
32365850ef23SBarry Smith             idx  += 2;
32375850ef23SBarry Smith             tmp0  = x[i1];
32385850ef23SBarry Smith             tmp1  = x[i2];
32395850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
32405850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
32415850ef23SBarry Smith           }
3242d876e2b0SMark Adams           if (n == sz-1) {
3243d876e2b0SMark Adams             tmp0  = x[*idx++];
3244d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3245d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3246d876e2b0SMark Adams             v1++; v2++;
3247d876e2b0SMark Adams           }
3248d876e2b0SMark Adams           t[row]   = sum1;
3249d876e2b0SMark Adams           t[row+1] = sum2;
3250d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 2;
3251d876e2b0SMark Adams           idx     = a->j + diag[row] + 2;
3252d876e2b0SMark Adams           v1 += 2;
3253d876e2b0SMark Adams           v2 += 2;
3254d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3255d876e2b0SMark Adams             i1    = idx[0];
3256d876e2b0SMark Adams             i2    = idx[1];
3257d876e2b0SMark Adams             idx  += 2;
3258d876e2b0SMark Adams             tmp0  = x[i1];
3259d876e2b0SMark Adams             tmp1  = x[i2];
3260d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3261d876e2b0SMark Adams             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3262d876e2b0SMark Adams           }
32635850ef23SBarry Smith           if (n == sz-1) {
32645850ef23SBarry Smith             tmp0  = x[*idx];
32655850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
32665850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
32675850ef23SBarry Smith           }
3268d876e2b0SMark Adams           x[row] = sum1*ibdiag[0] + sum2*ibdiag[2];
3269d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3];
32705850ef23SBarry Smith           break;
32715850ef23SBarry Smith         case 3:
32725850ef23SBarry Smith           v2   = a->a + ii[row+1];
32735850ef23SBarry Smith           v3   = a->a + ii[row+2];
32745850ef23SBarry Smith           sum1 = b[row];
32755850ef23SBarry Smith           sum2 = b[row+1];
32765850ef23SBarry Smith           sum3 = b[row+2];
32775850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
32785850ef23SBarry Smith             i1    = idx[0];
32795850ef23SBarry Smith             i2    = idx[1];
32805850ef23SBarry Smith             idx  += 2;
32815850ef23SBarry Smith             tmp0  = x[i1];
32825850ef23SBarry Smith             tmp1  = x[i2];
32835850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
32845850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
32855850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
32865850ef23SBarry Smith           }
3287d876e2b0SMark Adams           if (n == sz-1) {
3288d876e2b0SMark Adams             tmp0  = x[*idx++];
3289d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3290d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3291d876e2b0SMark Adams             sum3 -= v3[0] * tmp0;
3292d876e2b0SMark Adams             v1++; v2++; v3++;
3293d876e2b0SMark Adams           }
3294d876e2b0SMark Adams           t[row]   = sum1;
3295d876e2b0SMark Adams           t[row+1] = sum2;
3296d876e2b0SMark Adams           t[row+2] = sum3;
3297d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 3;
3298d876e2b0SMark Adams           idx     = a->j + diag[row] + 3;
3299d876e2b0SMark Adams           v1 += 3;
3300d876e2b0SMark Adams           v2 += 3;
3301d876e2b0SMark Adams           v3 += 3;
3302d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3303d876e2b0SMark Adams             i1    = idx[0];
3304d876e2b0SMark Adams             i2    = idx[1];
3305d876e2b0SMark Adams             idx  += 2;
3306d876e2b0SMark Adams             tmp0  = x[i1];
3307d876e2b0SMark Adams             tmp1  = x[i2];
3308d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3309d876e2b0SMark Adams             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3310d876e2b0SMark Adams             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3311d876e2b0SMark Adams           }
33125850ef23SBarry Smith           if (n == sz-1) {
33135850ef23SBarry Smith             tmp0  = x[*idx];
33145850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
33155850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
33165850ef23SBarry Smith             sum3 -= v3[0] * tmp0;
33175850ef23SBarry Smith           }
3318d876e2b0SMark Adams           x[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
3319d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
3320d876e2b0SMark Adams           x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
33215850ef23SBarry Smith           break;
33225850ef23SBarry Smith         case 4:
33235850ef23SBarry Smith           v2   = a->a + ii[row+1];
33245850ef23SBarry Smith           v3   = a->a + ii[row+2];
33255850ef23SBarry Smith           v4   = a->a + ii[row+3];
33265850ef23SBarry Smith           sum1 = b[row];
33275850ef23SBarry Smith           sum2 = b[row+1];
33285850ef23SBarry Smith           sum3 = b[row+2];
33295850ef23SBarry Smith           sum4 = b[row+3];
33305850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
33315850ef23SBarry Smith             i1    = idx[0];
33325850ef23SBarry Smith             i2    = idx[1];
33335850ef23SBarry Smith             idx  += 2;
33345850ef23SBarry Smith             tmp0  = x[i1];
33355850ef23SBarry Smith             tmp1  = x[i2];
33365850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
33375850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
33385850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
33395850ef23SBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
33405850ef23SBarry Smith           }
3341d876e2b0SMark Adams           if (n == sz-1) {
3342d876e2b0SMark Adams             tmp0  = x[*idx++];
3343d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3344d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3345d876e2b0SMark Adams             sum3 -= v3[0] * tmp0;
3346d876e2b0SMark Adams             sum4 -= v4[0] * tmp0;
3347d876e2b0SMark Adams             v1++; v2++; v3++; v4++;
3348d876e2b0SMark Adams           }
3349d876e2b0SMark Adams           t[row]   = sum1;
3350d876e2b0SMark Adams           t[row+1] = sum2;
3351d876e2b0SMark Adams           t[row+2] = sum3;
3352d876e2b0SMark Adams           t[row+3] = sum4;
3353d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 4;
3354d876e2b0SMark Adams           idx     = a->j + diag[row] + 4;
3355d876e2b0SMark Adams           v1 += 4;
3356d876e2b0SMark Adams           v2 += 4;
3357d876e2b0SMark Adams           v3 += 4;
3358d876e2b0SMark Adams           v4 += 4;
3359d876e2b0SMark Adams           for (n = 0; n<sz-1; n+=2) {
3360d876e2b0SMark Adams             i1    = idx[0];
3361d876e2b0SMark Adams             i2    = idx[1];
3362d876e2b0SMark Adams             idx  += 2;
3363d876e2b0SMark Adams             tmp0  = x[i1];
3364d876e2b0SMark Adams             tmp1  = x[i2];
3365d876e2b0SMark Adams             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3366d876e2b0SMark Adams             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3367d876e2b0SMark Adams             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3368d876e2b0SMark Adams             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3369d876e2b0SMark Adams           }
33705850ef23SBarry Smith           if (n == sz-1) {
33715850ef23SBarry Smith             tmp0  = x[*idx];
33725850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
33735850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
33745850ef23SBarry Smith             sum3 -= v3[0] * tmp0;
33755850ef23SBarry Smith             sum4 -= v4[0] * tmp0;
33765850ef23SBarry Smith           }
3377d876e2b0SMark Adams           x[row] =   sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
3378d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
3379d876e2b0SMark Adams           x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
3380d876e2b0SMark Adams           x[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
33815850ef23SBarry Smith           break;
33825850ef23SBarry Smith         case 5:
33835850ef23SBarry Smith           v2   = a->a + ii[row+1];
33845850ef23SBarry Smith           v3   = a->a + ii[row+2];
33855850ef23SBarry Smith           v4   = a->a + ii[row+3];
33865850ef23SBarry Smith           v5   = a->a + ii[row+4];
33875850ef23SBarry Smith           sum1 = b[row];
33885850ef23SBarry Smith           sum2 = b[row+1];
33895850ef23SBarry Smith           sum3 = b[row+2];
33905850ef23SBarry Smith           sum4 = b[row+3];
33915850ef23SBarry Smith           sum5 = b[row+4];
33925850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
33935850ef23SBarry Smith             i1    = idx[0];
33945850ef23SBarry Smith             i2    = idx[1];
33955850ef23SBarry Smith             idx  += 2;
33965850ef23SBarry Smith             tmp0  = x[i1];
33975850ef23SBarry Smith             tmp1  = x[i2];
33985850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
33995850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
34005850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
34015850ef23SBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
34025850ef23SBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
34035850ef23SBarry Smith           }
34045850ef23SBarry Smith           if (n == sz-1) {
3405d876e2b0SMark Adams             tmp0  = x[*idx++];
34065850ef23SBarry Smith             sum1 -= v1[0] * tmp0;
34075850ef23SBarry Smith             sum2 -= v2[0] * tmp0;
34085850ef23SBarry Smith             sum3 -= v3[0] * tmp0;
34095850ef23SBarry Smith             sum4 -= v4[0] * tmp0;
34105850ef23SBarry Smith             sum5 -= v5[0] * tmp0;
3411d876e2b0SMark Adams             v1++; v2++; v3++; v4++; v5++;
34125850ef23SBarry Smith           }
3413d876e2b0SMark Adams           t[row]   = sum1;
3414d876e2b0SMark Adams           t[row+1] = sum2;
3415d876e2b0SMark Adams           t[row+2] = sum3;
3416d876e2b0SMark Adams           t[row+3] = sum4;
3417d876e2b0SMark Adams           t[row+4] = sum5;
3418d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 5;
3419d876e2b0SMark Adams           idx     = a->j + diag[row] + 5;
3420d876e2b0SMark Adams           v1 += 5;
3421d876e2b0SMark Adams           v2 += 5;
3422d876e2b0SMark Adams           v3 += 5;
3423d876e2b0SMark Adams           v4 += 5;
3424d876e2b0SMark Adams           v5 += 5;
34255850ef23SBarry Smith           for (n = 0; n<sz-1; n+=2) {
34265850ef23SBarry Smith             i1    = idx[0];
34275850ef23SBarry Smith             i2    = idx[1];
34285850ef23SBarry Smith             idx  += 2;
34295850ef23SBarry Smith             tmp0  = x[i1];
34305850ef23SBarry Smith             tmp1  = x[i2];
34315850ef23SBarry Smith             sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
34325850ef23SBarry Smith             sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
34335850ef23SBarry Smith             sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
34345850ef23SBarry Smith             sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
34355850ef23SBarry Smith             sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
34365850ef23SBarry Smith           }
34375850ef23SBarry Smith           if (n == sz-1) {
34385850ef23SBarry Smith             tmp0  = x[*idx];
3439d876e2b0SMark Adams             sum1 -= v1[0] * tmp0;
3440d876e2b0SMark Adams             sum2 -= v2[0] * tmp0;
3441d876e2b0SMark Adams             sum3 -= v3[0] * tmp0;
3442d876e2b0SMark Adams             sum4 -= v4[0] * tmp0;
3443d876e2b0SMark Adams             sum5 -= v5[0] * tmp0;
34445850ef23SBarry Smith           }
3445d876e2b0SMark Adams           x[row]   = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
3446d876e2b0SMark Adams           x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
3447d876e2b0SMark Adams           x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
3448d876e2b0SMark Adams           x[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
3449d876e2b0SMark Adams           x[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
3450d876e2b0SMark Adams           break;
3451d876e2b0SMark Adams         default:
3452d876e2b0SMark Adams           SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
3453d876e2b0SMark Adams         }
3454d876e2b0SMark Adams       }
3455d876e2b0SMark Adams       xb   = t;
3456d876e2b0SMark Adams       ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);  /* undercounts diag inverse */
3457d876e2b0SMark Adams     } else xb = b;
3458d876e2b0SMark Adams 
3459d876e2b0SMark Adams     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) {
3460d876e2b0SMark Adams 
3461d876e2b0SMark Adams       ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
3462d876e2b0SMark Adams       for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
3463d876e2b0SMark Adams         ibdiag -= sizes[i]*sizes[i];
3464d876e2b0SMark Adams 
3465d876e2b0SMark Adams         /* set RHS */
3466d876e2b0SMark Adams         if (xb == b) {
3467d876e2b0SMark Adams           /* whole (old way) */
3468d876e2b0SMark Adams           sz      = ii[row+1] - ii[row];
3469d876e2b0SMark Adams           idx     = a->j + ii[row];
3470d876e2b0SMark Adams           switch (sizes[i]) {
3471d876e2b0SMark Adams           case 5:
3472d876e2b0SMark Adams             v5      = a->a + ii[row-4];
3473d876e2b0SMark Adams           case 4: /* fall through */
3474d876e2b0SMark Adams             v4      = a->a + ii[row-3];
3475d876e2b0SMark Adams           case 3:
3476d876e2b0SMark Adams             v3      = a->a + ii[row-2];
3477d876e2b0SMark Adams           case 2:
3478d876e2b0SMark Adams             v2      = a->a + ii[row-1];
3479d876e2b0SMark Adams           case 1:
3480d876e2b0SMark Adams             v1      = a->a + ii[row];
3481d876e2b0SMark Adams             break;
3482d876e2b0SMark Adams           default:
3483d876e2b0SMark Adams             SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
3484d876e2b0SMark Adams           }
3485d876e2b0SMark Adams         } else {
3486d876e2b0SMark Adams           /* upper, no diag */
3487d876e2b0SMark Adams           sz      = ii[row+1] - diag[row] - 1;
3488d876e2b0SMark Adams           idx     = a->j + diag[row] + 1;
3489d876e2b0SMark Adams           switch (sizes[i]) {
3490d876e2b0SMark Adams           case 5:
3491d876e2b0SMark Adams             v5      = a->a + diag[row-4] + 5;
3492d876e2b0SMark Adams           case 4: /* fall through */
3493d876e2b0SMark Adams             v4      = a->a + diag[row-3] + 4;
3494d876e2b0SMark Adams           case 3:
3495d876e2b0SMark Adams             v3      = a->a + diag[row-2] + 3;
3496d876e2b0SMark Adams           case 2:
3497d876e2b0SMark Adams             v2      = a->a + diag[row-1] + 2;
3498d876e2b0SMark Adams           case 1:
3499d876e2b0SMark Adams             v1      = a->a + diag[row] + 1;
3500d876e2b0SMark Adams           }
3501d876e2b0SMark Adams         }
3502d876e2b0SMark Adams         /* set sum */
3503d876e2b0SMark Adams         switch (sizes[i]) {
3504d876e2b0SMark Adams         case 5:
3505d876e2b0SMark Adams           sum5 = xb[row-4];
3506d876e2b0SMark Adams         case 4: /* fall through */
3507d876e2b0SMark Adams           sum4 = xb[row-3];
3508d876e2b0SMark Adams         case 3:
3509d876e2b0SMark Adams           sum3 = xb[row-2];
3510d876e2b0SMark Adams         case 2:
3511d876e2b0SMark Adams           sum2 = xb[row-1];
3512d876e2b0SMark Adams         case 1:
3513d876e2b0SMark Adams           /* note that sum1 is associated with the last row */
3514d876e2b0SMark Adams           sum1 = xb[row];
3515d876e2b0SMark Adams         }
3516d876e2b0SMark Adams         /* do sums */
3517d876e2b0SMark Adams         for (n = 0; n<sz-1; n+=2) {
3518d876e2b0SMark Adams             i1    = idx[0];
3519d876e2b0SMark Adams             i2    = idx[1];
3520d876e2b0SMark Adams             idx  += 2;
3521d876e2b0SMark Adams             tmp0  = x[i1];
3522d876e2b0SMark Adams             tmp1  = x[i2];
3523d876e2b0SMark Adams             switch (sizes[i]) {
3524d876e2b0SMark Adams             case 5:
3525d876e2b0SMark Adams               sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
3526d876e2b0SMark Adams             case 4: /* fall through */
3527d876e2b0SMark Adams               sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
3528d876e2b0SMark Adams             case 3:
3529d876e2b0SMark Adams               sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
3530d876e2b0SMark Adams             case 2:
3531d876e2b0SMark Adams               sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
3532d876e2b0SMark Adams             case 1:
3533d876e2b0SMark Adams               sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
3534d876e2b0SMark Adams             }
3535d876e2b0SMark Adams         }
3536d876e2b0SMark Adams         /* ragged edge */
3537d876e2b0SMark Adams         if (n == sz-1) {
3538d876e2b0SMark Adams           tmp0  = x[*idx];
3539d876e2b0SMark Adams           switch (sizes[i]) {
3540d876e2b0SMark Adams           case 5:
3541d876e2b0SMark Adams             sum5 -= *v5*tmp0;
3542d876e2b0SMark Adams           case 4: /* fall through */
3543d876e2b0SMark Adams             sum4 -= *v4*tmp0;
3544d876e2b0SMark Adams           case 3:
3545d876e2b0SMark Adams             sum3 -= *v3*tmp0;
3546d876e2b0SMark Adams           case 2:
3547d876e2b0SMark Adams             sum2 -= *v2*tmp0;
3548d876e2b0SMark Adams           case 1:
3549d876e2b0SMark Adams             sum1 -= *v1*tmp0;
3550d876e2b0SMark Adams           }
3551d876e2b0SMark Adams         }
3552d876e2b0SMark Adams         /* update */
3553d876e2b0SMark Adams         if (xb == b) {
3554d876e2b0SMark Adams           /* whole (old way) w/ diag */
3555d876e2b0SMark Adams           switch (sizes[i]) {
3556d876e2b0SMark Adams           case 5:
35575850ef23SBarry Smith             x[row--] += sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
35585850ef23SBarry Smith             x[row--] += sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
35595850ef23SBarry Smith             x[row--] += sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
35605850ef23SBarry Smith             x[row--] += sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
35615850ef23SBarry Smith             x[row--] += sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
35625850ef23SBarry Smith             break;
3563d876e2b0SMark Adams           case 4:
3564d876e2b0SMark Adams             x[row--] += sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3565d876e2b0SMark Adams             x[row--] += sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3566d876e2b0SMark Adams             x[row--] += sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3567d876e2b0SMark Adams             x[row--] += sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3568d876e2b0SMark Adams             break;
3569d876e2b0SMark Adams           case 3:
3570d876e2b0SMark Adams             x[row--] += sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3571d876e2b0SMark Adams             x[row--] += sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3572d876e2b0SMark Adams             x[row--] += sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3573d876e2b0SMark Adams             break;
3574d876e2b0SMark Adams           case 2:
3575d876e2b0SMark Adams             x[row--] += sum2*ibdiag[1] + sum1*ibdiag[3];
3576d876e2b0SMark Adams             x[row--] += sum2*ibdiag[0] + sum1*ibdiag[2];
3577d876e2b0SMark Adams             break;
3578d876e2b0SMark Adams           case 1:
3579d876e2b0SMark Adams             x[row--] += sum1*(*ibdiag);
3580d876e2b0SMark Adams             break;
3581d876e2b0SMark Adams           }
3582d876e2b0SMark Adams         } else {
3583d876e2b0SMark Adams           /* no diag so set =  */
3584d876e2b0SMark Adams           switch (sizes[i]) {
3585d876e2b0SMark Adams           case 5:
3586d876e2b0SMark Adams             x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
3587d876e2b0SMark Adams             x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
3588d876e2b0SMark Adams             x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
3589d876e2b0SMark Adams             x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
3590d876e2b0SMark Adams             x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
3591d876e2b0SMark Adams             break;
3592d876e2b0SMark Adams           case 4:
3593d876e2b0SMark Adams             x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3594d876e2b0SMark Adams             x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3595d876e2b0SMark Adams             x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3596d876e2b0SMark Adams             x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3597d876e2b0SMark Adams             break;
3598d876e2b0SMark Adams           case 3:
3599d876e2b0SMark Adams             x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3600d876e2b0SMark Adams             x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3601d876e2b0SMark Adams             x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3602d876e2b0SMark Adams             break;
3603d876e2b0SMark Adams           case 2:
3604d876e2b0SMark Adams             x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3];
3605d876e2b0SMark Adams             x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2];
3606d876e2b0SMark Adams             break;
3607d876e2b0SMark Adams           case 1:
3608d876e2b0SMark Adams             x[row--] = sum1*(*ibdiag);
3609d876e2b0SMark Adams             break;
36105850ef23SBarry Smith           }
36115850ef23SBarry Smith         }
3612d876e2b0SMark Adams       }
3613d876e2b0SMark Adams       if (xb == b) {
36145850ef23SBarry Smith         ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
3615d876e2b0SMark Adams       } else {
3616d876e2b0SMark Adams         ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper, undercounts diag inverse */
3617d876e2b0SMark Adams       }
36185850ef23SBarry Smith     }
36192af78befSBarry Smith   }
362089c6957cSBarry Smith   if (flag & SOR_EISENSTAT) {
362189c6957cSBarry Smith     /*
362289c6957cSBarry Smith           Apply  (U + D)^-1  where D is now the block diagonal
362389c6957cSBarry Smith     */
362489c6957cSBarry Smith     ibdiag = a->inode.ibdiag+a->inode.bdiagsize;
362589c6957cSBarry Smith     for (i=m-1, row=A->rmap->n-1; i>=0; i--) {
362689c6957cSBarry Smith       ibdiag -= sizes[i]*sizes[i];
362789c6957cSBarry Smith       sz      = ii[row+1] - diag[row] - 1;
362889c6957cSBarry Smith       v1      = a->a + diag[row] + 1;
362989c6957cSBarry Smith       idx     = a->j + diag[row] + 1;
36304108e4d5SBarry Smith       /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
363189c6957cSBarry Smith       switch (sizes[i]) {
363289c6957cSBarry Smith       case 1:
363389c6957cSBarry Smith 
363489c6957cSBarry Smith         sum1 = b[row];
363589c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
363689c6957cSBarry Smith           i1    = idx[0];
363789c6957cSBarry Smith           i2    = idx[1];
363889c6957cSBarry Smith           idx  += 2;
363989c6957cSBarry Smith           tmp0  = x[i1];
364089c6957cSBarry Smith           tmp1  = x[i2];
364189c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
364289c6957cSBarry Smith         }
364389c6957cSBarry Smith 
364489c6957cSBarry Smith         if (n == sz-1) {
364589c6957cSBarry Smith           tmp0  = x[*idx];
364689c6957cSBarry Smith           sum1 -= *v1*tmp0;
364789c6957cSBarry Smith         }
3648938d4eb3SBarry Smith         x[row] = sum1*(*ibdiag);row--;
364989c6957cSBarry Smith         break;
365089c6957cSBarry Smith 
365189c6957cSBarry Smith       case 2:
365289c6957cSBarry Smith 
365389c6957cSBarry Smith         sum1 = b[row];
365489c6957cSBarry Smith         sum2 = b[row-1];
365589c6957cSBarry Smith         /* note that sum1 is associated with the second of the two rows */
365689c6957cSBarry Smith         v2 = a->a + diag[row-1] + 2;
365789c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
365889c6957cSBarry Smith           i1    = idx[0];
365989c6957cSBarry Smith           i2    = idx[1];
366089c6957cSBarry Smith           idx  += 2;
366189c6957cSBarry Smith           tmp0  = x[i1];
366289c6957cSBarry Smith           tmp1  = x[i2];
366389c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
366489c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
366589c6957cSBarry Smith         }
366689c6957cSBarry Smith 
366789c6957cSBarry Smith         if (n == sz-1) {
366889c6957cSBarry Smith           tmp0  = x[*idx];
366989c6957cSBarry Smith           sum1 -= *v1*tmp0;
367089c6957cSBarry Smith           sum2 -= *v2*tmp0;
367189c6957cSBarry Smith         }
3672938d4eb3SBarry Smith         x[row]   = sum2*ibdiag[1] + sum1*ibdiag[3];
3673938d4eb3SBarry Smith         x[row-1] = sum2*ibdiag[0] + sum1*ibdiag[2];
3674938d4eb3SBarry Smith         row     -= 2;
367589c6957cSBarry Smith         break;
367689c6957cSBarry Smith       case 3:
367789c6957cSBarry Smith 
367889c6957cSBarry Smith         sum1 = b[row];
367989c6957cSBarry Smith         sum2 = b[row-1];
368089c6957cSBarry Smith         sum3 = b[row-2];
368189c6957cSBarry Smith         v2   = a->a + diag[row-1] + 2;
368289c6957cSBarry Smith         v3   = a->a + diag[row-2] + 3;
368389c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
368489c6957cSBarry Smith           i1    = idx[0];
368589c6957cSBarry Smith           i2    = idx[1];
368689c6957cSBarry Smith           idx  += 2;
368789c6957cSBarry Smith           tmp0  = x[i1];
368889c6957cSBarry Smith           tmp1  = x[i2];
368989c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
369089c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
369189c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
369289c6957cSBarry Smith         }
369389c6957cSBarry Smith 
369489c6957cSBarry Smith         if (n == sz-1) {
369589c6957cSBarry Smith           tmp0  = x[*idx];
369689c6957cSBarry Smith           sum1 -= *v1*tmp0;
369789c6957cSBarry Smith           sum2 -= *v2*tmp0;
369889c6957cSBarry Smith           sum3 -= *v3*tmp0;
369989c6957cSBarry Smith         }
3700938d4eb3SBarry Smith         x[row]   = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8];
3701938d4eb3SBarry Smith         x[row-1] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7];
3702938d4eb3SBarry Smith         x[row-2] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6];
3703938d4eb3SBarry Smith         row     -= 3;
370489c6957cSBarry Smith         break;
370589c6957cSBarry Smith       case 4:
370689c6957cSBarry Smith 
370789c6957cSBarry Smith         sum1 = b[row];
370889c6957cSBarry Smith         sum2 = b[row-1];
370989c6957cSBarry Smith         sum3 = b[row-2];
371089c6957cSBarry Smith         sum4 = b[row-3];
371189c6957cSBarry Smith         v2   = a->a + diag[row-1] + 2;
371289c6957cSBarry Smith         v3   = a->a + diag[row-2] + 3;
371389c6957cSBarry Smith         v4   = a->a + diag[row-3] + 4;
371489c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
371589c6957cSBarry Smith           i1    = idx[0];
371689c6957cSBarry Smith           i2    = idx[1];
371789c6957cSBarry Smith           idx  += 2;
371889c6957cSBarry Smith           tmp0  = x[i1];
371989c6957cSBarry Smith           tmp1  = x[i2];
372089c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
372189c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
372289c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
372389c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
372489c6957cSBarry Smith         }
372589c6957cSBarry Smith 
372689c6957cSBarry Smith         if (n == sz-1) {
372789c6957cSBarry Smith           tmp0  = x[*idx];
372889c6957cSBarry Smith           sum1 -= *v1*tmp0;
372989c6957cSBarry Smith           sum2 -= *v2*tmp0;
373089c6957cSBarry Smith           sum3 -= *v3*tmp0;
373189c6957cSBarry Smith           sum4 -= *v4*tmp0;
373289c6957cSBarry Smith         }
3733938d4eb3SBarry Smith         x[row]   = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15];
3734938d4eb3SBarry Smith         x[row-1] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14];
3735938d4eb3SBarry Smith         x[row-2] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13];
3736938d4eb3SBarry Smith         x[row-3] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12];
3737938d4eb3SBarry Smith         row     -= 4;
373889c6957cSBarry Smith         break;
373989c6957cSBarry Smith       case 5:
374089c6957cSBarry Smith 
374189c6957cSBarry Smith         sum1 = b[row];
374289c6957cSBarry Smith         sum2 = b[row-1];
374389c6957cSBarry Smith         sum3 = b[row-2];
374489c6957cSBarry Smith         sum4 = b[row-3];
374589c6957cSBarry Smith         sum5 = b[row-4];
374689c6957cSBarry Smith         v2   = a->a + diag[row-1] + 2;
374789c6957cSBarry Smith         v3   = a->a + diag[row-2] + 3;
374889c6957cSBarry Smith         v4   = a->a + diag[row-3] + 4;
374989c6957cSBarry Smith         v5   = a->a + diag[row-4] + 5;
375089c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
375189c6957cSBarry Smith           i1    = idx[0];
375289c6957cSBarry Smith           i2    = idx[1];
375389c6957cSBarry Smith           idx  += 2;
375489c6957cSBarry Smith           tmp0  = x[i1];
375589c6957cSBarry Smith           tmp1  = x[i2];
375689c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
375789c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
375889c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
375989c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
376089c6957cSBarry Smith           sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
376189c6957cSBarry Smith         }
376289c6957cSBarry Smith 
376389c6957cSBarry Smith         if (n == sz-1) {
376489c6957cSBarry Smith           tmp0  = x[*idx];
376589c6957cSBarry Smith           sum1 -= *v1*tmp0;
376689c6957cSBarry Smith           sum2 -= *v2*tmp0;
376789c6957cSBarry Smith           sum3 -= *v3*tmp0;
376889c6957cSBarry Smith           sum4 -= *v4*tmp0;
376989c6957cSBarry Smith           sum5 -= *v5*tmp0;
377089c6957cSBarry Smith         }
3771938d4eb3SBarry Smith         x[row]   = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24];
3772938d4eb3SBarry Smith         x[row-1] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23];
3773938d4eb3SBarry Smith         x[row-2] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22];
3774938d4eb3SBarry Smith         x[row-3] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21];
3775938d4eb3SBarry Smith         x[row-4] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20];
3776938d4eb3SBarry Smith         row     -= 5;
377789c6957cSBarry Smith         break;
377889c6957cSBarry Smith       default:
3779e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
378089c6957cSBarry Smith       }
378189c6957cSBarry Smith     }
378289c6957cSBarry Smith     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
378389c6957cSBarry Smith 
378489c6957cSBarry Smith     /*
378589c6957cSBarry Smith            t = b - D x    where D is the block diagonal
378689c6957cSBarry Smith     */
378789c6957cSBarry Smith     cnt = 0;
378889c6957cSBarry Smith     for (i=0, row=0; i<m; i++) {
378989c6957cSBarry Smith       switch (sizes[i]) {
379089c6957cSBarry Smith       case 1:
379189c6957cSBarry Smith         t[row] = b[row] - bdiag[cnt++]*x[row]; row++;
379289c6957cSBarry Smith         break;
379389c6957cSBarry Smith       case 2:
379489c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1];
379589c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+2];
379689c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+3];
379789c6957cSBarry Smith         t[row]   = b[row] - tmp1;
379889c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2; row += 2;
379989c6957cSBarry Smith         cnt     += 4;
380089c6957cSBarry Smith         break;
380189c6957cSBarry Smith       case 3:
380289c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1]; x3 = x[row+2];
380389c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6];
380489c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7];
380589c6957cSBarry Smith         tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8];
380689c6957cSBarry Smith         t[row]   = b[row] - tmp1;
380789c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2;
380889c6957cSBarry Smith         t[row+2] = b[row+2] - tmp3; row += 3;
380989c6957cSBarry Smith         cnt     += 9;
381089c6957cSBarry Smith         break;
381189c6957cSBarry Smith       case 4:
381289c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3];
381389c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12];
381489c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13];
381589c6957cSBarry Smith         tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14];
381689c6957cSBarry Smith         tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15];
381789c6957cSBarry Smith         t[row]   = b[row] - tmp1;
381889c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2;
381989c6957cSBarry Smith         t[row+2] = b[row+2] - tmp3;
382089c6957cSBarry Smith         t[row+3] = b[row+3] - tmp4; row += 4;
382189c6957cSBarry Smith         cnt     += 16;
382289c6957cSBarry Smith         break;
382389c6957cSBarry Smith       case 5:
382489c6957cSBarry Smith         x1       = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; x5 = x[row+4];
382589c6957cSBarry Smith         tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20];
382689c6957cSBarry Smith         tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21];
382789c6957cSBarry Smith         tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22];
382889c6957cSBarry Smith         tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23];
382989c6957cSBarry Smith         tmp5     = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24];
383089c6957cSBarry Smith         t[row]   = b[row] - tmp1;
383189c6957cSBarry Smith         t[row+1] = b[row+1] - tmp2;
383289c6957cSBarry Smith         t[row+2] = b[row+2] - tmp3;
383389c6957cSBarry Smith         t[row+3] = b[row+3] - tmp4;
383489c6957cSBarry Smith         t[row+4] = b[row+4] - tmp5;row += 5;
383589c6957cSBarry Smith         cnt     += 25;
383689c6957cSBarry Smith         break;
383789c6957cSBarry Smith       default:
3838e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
383989c6957cSBarry Smith       }
384089c6957cSBarry Smith     }
384189c6957cSBarry Smith     ierr = PetscLogFlops(m);CHKERRQ(ierr);
384289c6957cSBarry Smith 
384389c6957cSBarry Smith 
384489c6957cSBarry Smith 
384589c6957cSBarry Smith     /*
384689c6957cSBarry Smith           Apply (L + D)^-1 where D is the block diagonal
384789c6957cSBarry Smith     */
384889c6957cSBarry Smith     for (i=0, row=0; i<m; i++) {
384989c6957cSBarry Smith       sz  = diag[row] - ii[row];
385089c6957cSBarry Smith       v1  = a->a + ii[row];
385189c6957cSBarry Smith       idx = a->j + ii[row];
38524108e4d5SBarry Smith       /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */
385389c6957cSBarry Smith       switch (sizes[i]) {
385489c6957cSBarry Smith       case 1:
385589c6957cSBarry Smith 
385689c6957cSBarry Smith         sum1 = t[row];
385789c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
385889c6957cSBarry Smith           i1    = idx[0];
385989c6957cSBarry Smith           i2    = idx[1];
386089c6957cSBarry Smith           idx  += 2;
386189c6957cSBarry Smith           tmp0  = t[i1];
386289c6957cSBarry Smith           tmp1  = t[i2];
386389c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
386489c6957cSBarry Smith         }
386589c6957cSBarry Smith 
386689c6957cSBarry Smith         if (n == sz-1) {
386789c6957cSBarry Smith           tmp0  = t[*idx];
386889c6957cSBarry Smith           sum1 -= *v1 * tmp0;
386989c6957cSBarry Smith         }
387089c6957cSBarry Smith         x[row] += t[row] = sum1*(*ibdiag++); row++;
387189c6957cSBarry Smith         break;
387289c6957cSBarry Smith       case 2:
387389c6957cSBarry Smith         v2   = a->a + ii[row+1];
387489c6957cSBarry Smith         sum1 = t[row];
387589c6957cSBarry Smith         sum2 = t[row+1];
387689c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
387789c6957cSBarry Smith           i1    = idx[0];
387889c6957cSBarry Smith           i2    = idx[1];
387989c6957cSBarry Smith           idx  += 2;
388089c6957cSBarry Smith           tmp0  = t[i1];
388189c6957cSBarry Smith           tmp1  = t[i2];
388289c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
388389c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
388489c6957cSBarry Smith         }
388589c6957cSBarry Smith 
388689c6957cSBarry Smith         if (n == sz-1) {
388789c6957cSBarry Smith           tmp0  = t[*idx];
388889c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
388989c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
389089c6957cSBarry Smith         }
389189c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[2];
389289c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3];
389389c6957cSBarry Smith         ibdiag   += 4; row += 2;
389489c6957cSBarry Smith         break;
389589c6957cSBarry Smith       case 3:
389689c6957cSBarry Smith         v2   = a->a + ii[row+1];
389789c6957cSBarry Smith         v3   = a->a + ii[row+2];
389889c6957cSBarry Smith         sum1 = t[row];
389989c6957cSBarry Smith         sum2 = t[row+1];
390089c6957cSBarry Smith         sum3 = t[row+2];
390189c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
390289c6957cSBarry Smith           i1    = idx[0];
390389c6957cSBarry Smith           i2    = idx[1];
390489c6957cSBarry Smith           idx  += 2;
390589c6957cSBarry Smith           tmp0  = t[i1];
390689c6957cSBarry Smith           tmp1  = t[i2];
390789c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
390889c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
390989c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
391089c6957cSBarry Smith         }
391189c6957cSBarry Smith 
391289c6957cSBarry Smith         if (n == sz-1) {
391389c6957cSBarry Smith           tmp0  = t[*idx];
391489c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
391589c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
391689c6957cSBarry Smith           sum3 -= v3[0] * tmp0;
391789c6957cSBarry Smith         }
391889c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6];
391989c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7];
392089c6957cSBarry Smith         x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8];
392189c6957cSBarry Smith         ibdiag   += 9; row += 3;
392289c6957cSBarry Smith         break;
392389c6957cSBarry Smith       case 4:
392489c6957cSBarry Smith         v2   = a->a + ii[row+1];
392589c6957cSBarry Smith         v3   = a->a + ii[row+2];
392689c6957cSBarry Smith         v4   = a->a + ii[row+3];
392789c6957cSBarry Smith         sum1 = t[row];
392889c6957cSBarry Smith         sum2 = t[row+1];
392989c6957cSBarry Smith         sum3 = t[row+2];
393089c6957cSBarry Smith         sum4 = t[row+3];
393189c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
393289c6957cSBarry Smith           i1    = idx[0];
393389c6957cSBarry Smith           i2    = idx[1];
393489c6957cSBarry Smith           idx  += 2;
393589c6957cSBarry Smith           tmp0  = t[i1];
393689c6957cSBarry Smith           tmp1  = t[i2];
393789c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
393889c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
393989c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
394089c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
394189c6957cSBarry Smith         }
394289c6957cSBarry Smith 
394389c6957cSBarry Smith         if (n == sz-1) {
394489c6957cSBarry Smith           tmp0  = t[*idx];
394589c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
394689c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
394789c6957cSBarry Smith           sum3 -= v3[0] * tmp0;
394889c6957cSBarry Smith           sum4 -= v4[0] * tmp0;
394989c6957cSBarry Smith         }
395089c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12];
395189c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13];
395289c6957cSBarry Smith         x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14];
395389c6957cSBarry Smith         x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15];
395489c6957cSBarry Smith         ibdiag   += 16; row += 4;
395589c6957cSBarry Smith         break;
395689c6957cSBarry Smith       case 5:
395789c6957cSBarry Smith         v2   = a->a + ii[row+1];
395889c6957cSBarry Smith         v3   = a->a + ii[row+2];
395989c6957cSBarry Smith         v4   = a->a + ii[row+3];
396089c6957cSBarry Smith         v5   = a->a + ii[row+4];
396189c6957cSBarry Smith         sum1 = t[row];
396289c6957cSBarry Smith         sum2 = t[row+1];
396389c6957cSBarry Smith         sum3 = t[row+2];
396489c6957cSBarry Smith         sum4 = t[row+3];
396589c6957cSBarry Smith         sum5 = t[row+4];
396689c6957cSBarry Smith         for (n = 0; n<sz-1; n+=2) {
396789c6957cSBarry Smith           i1    = idx[0];
396889c6957cSBarry Smith           i2    = idx[1];
396989c6957cSBarry Smith           idx  += 2;
397089c6957cSBarry Smith           tmp0  = t[i1];
397189c6957cSBarry Smith           tmp1  = t[i2];
397289c6957cSBarry Smith           sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2;
397389c6957cSBarry Smith           sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2;
397489c6957cSBarry Smith           sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2;
397589c6957cSBarry Smith           sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2;
397689c6957cSBarry Smith           sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2;
397789c6957cSBarry Smith         }
397889c6957cSBarry Smith 
397989c6957cSBarry Smith         if (n == sz-1) {
398089c6957cSBarry Smith           tmp0  = t[*idx];
398189c6957cSBarry Smith           sum1 -= v1[0] * tmp0;
398289c6957cSBarry Smith           sum2 -= v2[0] * tmp0;
398389c6957cSBarry Smith           sum3 -= v3[0] * tmp0;
398489c6957cSBarry Smith           sum4 -= v4[0] * tmp0;
398589c6957cSBarry Smith           sum5 -= v5[0] * tmp0;
398689c6957cSBarry Smith         }
398789c6957cSBarry Smith         x[row]   += t[row] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20];
398889c6957cSBarry Smith         x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21];
398989c6957cSBarry Smith         x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22];
399089c6957cSBarry Smith         x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23];
399189c6957cSBarry Smith         x[row+4] += t[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24];
399289c6957cSBarry Smith         ibdiag   += 25; row += 5;
399389c6957cSBarry Smith         break;
399489c6957cSBarry Smith       default:
3995e32f2f54SBarry Smith         SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
399689c6957cSBarry Smith       }
399789c6957cSBarry Smith     }
399889c6957cSBarry Smith     ierr = PetscLogFlops(a->nz);CHKERRQ(ierr);
39995850ef23SBarry Smith   }
40002af78befSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
40013649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
4002365a8a9eSBarry Smith   PetscFunctionReturn(0);
40032af78befSBarry Smith }
40042af78befSBarry Smith 
400589c6957cSBarry Smith #undef __FUNCT__
40064108e4d5SBarry Smith #define __FUNCT__ "MatMultDiagonalBlock_SeqAIJ_Inode"
40074108e4d5SBarry Smith PetscErrorCode MatMultDiagonalBlock_SeqAIJ_Inode(Mat A,Vec bb,Vec xx)
400889c6957cSBarry Smith {
400989c6957cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
401089c6957cSBarry Smith   PetscScalar       *x,tmp1,tmp2,tmp3,tmp4,tmp5,x1,x2,x3,x4,x5;
401189c6957cSBarry Smith   const MatScalar   *bdiag = a->inode.bdiag;
401289c6957cSBarry Smith   const PetscScalar *b;
401389c6957cSBarry Smith   PetscErrorCode    ierr;
401489c6957cSBarry Smith   PetscInt          m      = a->inode.node_count,cnt = 0,i,row;
401589c6957cSBarry Smith   const PetscInt    *sizes = a->inode.size;
40162af78befSBarry Smith 
401789c6957cSBarry Smith   PetscFunctionBegin;
401889c6957cSBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
40193649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
402089c6957cSBarry Smith   cnt  = 0;
402189c6957cSBarry Smith   for (i=0, row=0; i<m; i++) {
402289c6957cSBarry Smith     switch (sizes[i]) {
402389c6957cSBarry Smith     case 1:
402489c6957cSBarry Smith       x[row] = b[row]*bdiag[cnt++];row++;
402589c6957cSBarry Smith       break;
402689c6957cSBarry Smith     case 2:
402789c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1];
402889c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+2];
402989c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+3];
403089c6957cSBarry Smith       x[row++] = tmp1;
403189c6957cSBarry Smith       x[row++] = tmp2;
403289c6957cSBarry Smith       cnt     += 4;
403389c6957cSBarry Smith       break;
403489c6957cSBarry Smith     case 3:
403589c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1]; x3 = b[row+2];
403689c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6];
403789c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7];
403889c6957cSBarry Smith       tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8];
403989c6957cSBarry Smith       x[row++] = tmp1;
404089c6957cSBarry Smith       x[row++] = tmp2;
404189c6957cSBarry Smith       x[row++] = tmp3;
404289c6957cSBarry Smith       cnt     += 9;
404389c6957cSBarry Smith       break;
404489c6957cSBarry Smith     case 4:
404589c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3];
404689c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12];
404789c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13];
404889c6957cSBarry Smith       tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14];
404989c6957cSBarry Smith       tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15];
405089c6957cSBarry Smith       x[row++] = tmp1;
405189c6957cSBarry Smith       x[row++] = tmp2;
405289c6957cSBarry Smith       x[row++] = tmp3;
405389c6957cSBarry Smith       x[row++] = tmp4;
405489c6957cSBarry Smith       cnt     += 16;
405589c6957cSBarry Smith       break;
405689c6957cSBarry Smith     case 5:
405789c6957cSBarry Smith       x1       = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3]; x5 = b[row+4];
405889c6957cSBarry Smith       tmp1     = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20];
405989c6957cSBarry Smith       tmp2     = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21];
406089c6957cSBarry Smith       tmp3     = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22];
406189c6957cSBarry Smith       tmp4     = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23];
406289c6957cSBarry Smith       tmp5     = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24];
406389c6957cSBarry Smith       x[row++] = tmp1;
406489c6957cSBarry Smith       x[row++] = tmp2;
406589c6957cSBarry Smith       x[row++] = tmp3;
406689c6957cSBarry Smith       x[row++] = tmp4;
406789c6957cSBarry Smith       x[row++] = tmp5;
406889c6957cSBarry Smith       cnt     += 25;
406989c6957cSBarry Smith       break;
407089c6957cSBarry Smith     default:
4071e32f2f54SBarry Smith       SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]);
407289c6957cSBarry Smith     }
407389c6957cSBarry Smith   }
407489c6957cSBarry Smith   ierr = PetscLogFlops(2*cnt);CHKERRQ(ierr);
407589c6957cSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
40763649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
407789c6957cSBarry Smith   PetscFunctionReturn(0);
407889c6957cSBarry Smith }
407989c6957cSBarry Smith 
40804c1414c8SBarry Smith /*
40814c1414c8SBarry Smith     samestructure indicates that the matrix has not changed its nonzero structure so we
40824c1414c8SBarry Smith     do not need to recompute the inodes
40834c1414c8SBarry Smith */
40844c1414c8SBarry Smith #undef __FUNCT__
40854c1414c8SBarry Smith #define __FUNCT__ "Mat_CheckInode"
4086ace3abfcSBarry Smith PetscErrorCode Mat_CheckInode(Mat A,PetscBool samestructure)
40874c1414c8SBarry Smith {
40884c1414c8SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
40894c1414c8SBarry Smith   PetscErrorCode ierr;
40908758e1faSBarry Smith   PetscInt       i,j,m,nzx,nzy,*ns,node_count,blk_size;
4091ace3abfcSBarry Smith   PetscBool      flag;
40928758e1faSBarry Smith   const PetscInt *idx,*idy,*ii;
40934c1414c8SBarry Smith 
40944c1414c8SBarry Smith   PetscFunctionBegin;
4095d74fe821SBarry Smith   if (!a->inode.use) PetscFunctionReturn(0);
40964c1414c8SBarry Smith   if (a->inode.checked && samestructure) PetscFunctionReturn(0);
40974c1414c8SBarry Smith 
40984c1414c8SBarry Smith 
4099d0f46423SBarry Smith   m = A->rmap->n;
41002205254eSKarl Rupp   if (a->inode.size) ns = a->inode.size;
41012205254eSKarl Rupp   else {
41022205254eSKarl Rupp     ierr = PetscMalloc((m+1)*sizeof(PetscInt),&ns);CHKERRQ(ierr);
41032205254eSKarl Rupp   }
41044c1414c8SBarry Smith 
41054c1414c8SBarry Smith   i          = 0;
41064c1414c8SBarry Smith   node_count = 0;
41074c1414c8SBarry Smith   idx        = a->j;
41084c1414c8SBarry Smith   ii         = a->i;
41094c1414c8SBarry Smith   while (i < m) {                /* For each row */
41104c1414c8SBarry Smith     nzx = ii[i+1] - ii[i];       /* Number of nonzeros */
41114c1414c8SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
41124c1414c8SBarry Smith     for (j=i+1,idy=idx,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
41134c1414c8SBarry Smith       nzy = ii[j+1] - ii[j];     /* Same number of nonzeros */
41144c1414c8SBarry Smith       if (nzy != nzx) break;
41154c1414c8SBarry Smith       idy += nzx;              /* Same nonzero pattern */
41164c1414c8SBarry Smith       ierr = PetscMemcmp(idx,idy,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr);
41174c1414c8SBarry Smith       if (!flag) break;
41184c1414c8SBarry Smith     }
41194c1414c8SBarry Smith     ns[node_count++] = blk_size;
41204c1414c8SBarry Smith     idx             += blk_size*nzx;
41214c1414c8SBarry Smith     i                = j;
41224c1414c8SBarry Smith   }
41234c1414c8SBarry Smith   /* If not enough inodes found,, do not use inode version of the routines */
4124be6adb11SBarry Smith   if (!m || node_count > .8*m) {
41254c1414c8SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
41262205254eSKarl Rupp 
41274c1414c8SBarry Smith     a->inode.node_count       = 0;
41280298fd71SBarry Smith     a->inode.size             = NULL;
41294c1414c8SBarry Smith     a->inode.use              = PETSC_FALSE;
4130ded6e084SBarry Smith     A->ops->mult              = MatMult_SeqAIJ;
4131ded6e084SBarry Smith     A->ops->sor               = MatSOR_SeqAIJ;
4132ded6e084SBarry Smith     A->ops->multadd           = MatMultAdd_SeqAIJ;
4133ded6e084SBarry Smith     A->ops->getrowij          = MatGetRowIJ_SeqAIJ;
4134ded6e084SBarry Smith     A->ops->restorerowij      = MatRestoreRowIJ_SeqAIJ;
4135ded6e084SBarry Smith     A->ops->getcolumnij       = MatGetColumnIJ_SeqAIJ;
4136ded6e084SBarry Smith     A->ops->restorecolumnij   = MatRestoreColumnIJ_SeqAIJ;
4137ded6e084SBarry Smith     A->ops->coloringpatch     = 0;
4138ded6e084SBarry Smith     A->ops->multdiagonalblock = 0;
41392205254eSKarl Rupp 
41404c1414c8SBarry Smith     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
41414c1414c8SBarry Smith   } else {
4142d5f3da31SBarry Smith     if (!A->factortype) {
41434108e4d5SBarry Smith       A->ops->mult              = MatMult_SeqAIJ_Inode;
41444108e4d5SBarry Smith       A->ops->sor               = MatSOR_SeqAIJ_Inode;
41454108e4d5SBarry Smith       A->ops->multadd           = MatMultAdd_SeqAIJ_Inode;
41464108e4d5SBarry Smith       A->ops->getrowij          = MatGetRowIJ_SeqAIJ_Inode;
41474108e4d5SBarry Smith       A->ops->restorerowij      = MatRestoreRowIJ_SeqAIJ_Inode;
41484108e4d5SBarry Smith       A->ops->getcolumnij       = MatGetColumnIJ_SeqAIJ_Inode;
41494108e4d5SBarry Smith       A->ops->restorecolumnij   = MatRestoreColumnIJ_SeqAIJ_Inode;
41504108e4d5SBarry Smith       A->ops->coloringpatch     = MatColoringPatch_SeqAIJ_Inode;
41514108e4d5SBarry Smith       A->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode;
4152d3ac4fa3SBarry Smith     } else {
4153d3ac4fa3SBarry Smith       A->ops->solve = MatSolve_SeqAIJ_Inode_inplace;
4154d3ac4fa3SBarry Smith     }
41554c1414c8SBarry Smith     a->inode.node_count = node_count;
41564c1414c8SBarry Smith     a->inode.size       = ns;
41574c1414c8SBarry Smith     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
41584c1414c8SBarry Smith   }
4159be6adb11SBarry Smith   a->inode.checked = PETSC_TRUE;
41604c1414c8SBarry Smith   PetscFunctionReturn(0);
41614c1414c8SBarry Smith }
41624c1414c8SBarry Smith 
416317454e89SShri Abhyankar #undef __FUNCT__
4164150f0143SBarry Smith #define __FUNCT__ "MatDuplicate_SeqAIJ_Inode"
4165150f0143SBarry Smith PetscErrorCode MatDuplicate_SeqAIJ_Inode(Mat A,MatDuplicateOption cpvalues,Mat *C)
4166150f0143SBarry Smith {
4167150f0143SBarry Smith   Mat            B =*C;
4168150f0143SBarry Smith   Mat_SeqAIJ     *c=(Mat_SeqAIJ*)B->data,*a=(Mat_SeqAIJ*)A->data;
4169150f0143SBarry Smith   PetscErrorCode ierr;
4170150f0143SBarry Smith   PetscInt       m=A->rmap->n;
4171150f0143SBarry Smith 
4172150f0143SBarry Smith   PetscFunctionBegin;
4173150f0143SBarry Smith   c->inode.use       = a->inode.use;
4174150f0143SBarry Smith   c->inode.limit     = a->inode.limit;
4175150f0143SBarry Smith   c->inode.max_limit = a->inode.max_limit;
4176150f0143SBarry Smith   if (a->inode.size) {
4177150f0143SBarry Smith     ierr                = PetscMalloc((m+1)*sizeof(PetscInt),&c->inode.size);CHKERRQ(ierr);
4178150f0143SBarry Smith     c->inode.node_count = a->inode.node_count;
4179150f0143SBarry Smith     ierr                = PetscMemcpy(c->inode.size,a->inode.size,(m+1)*sizeof(PetscInt));CHKERRQ(ierr);
4180150f0143SBarry Smith     /* note the table of functions below should match that in Mat_CheckInode() */
41812c451681SBarry Smith     if (!B->factortype) {
41822c451681SBarry Smith       B->ops->mult              = MatMult_SeqAIJ_Inode;
41832c451681SBarry Smith       B->ops->sor               = MatSOR_SeqAIJ_Inode;
41842c451681SBarry Smith       B->ops->multadd           = MatMultAdd_SeqAIJ_Inode;
41852c451681SBarry Smith       B->ops->getrowij          = MatGetRowIJ_SeqAIJ_Inode;
41862c451681SBarry Smith       B->ops->restorerowij      = MatRestoreRowIJ_SeqAIJ_Inode;
41872c451681SBarry Smith       B->ops->getcolumnij       = MatGetColumnIJ_SeqAIJ_Inode;
41882c451681SBarry Smith       B->ops->restorecolumnij   = MatRestoreColumnIJ_SeqAIJ_Inode;
41892c451681SBarry Smith       B->ops->coloringpatch     = MatColoringPatch_SeqAIJ_Inode;
41902c451681SBarry Smith       B->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode;
4191150f0143SBarry Smith     } else {
41922c451681SBarry Smith       B->ops->solve = MatSolve_SeqAIJ_Inode_inplace;
4193150f0143SBarry Smith     }
4194150f0143SBarry Smith   } else {
4195150f0143SBarry Smith     c->inode.size       = 0;
4196150f0143SBarry Smith     c->inode.node_count = 0;
4197150f0143SBarry Smith   }
4198150f0143SBarry Smith   c->inode.ibdiagvalid = PETSC_FALSE;
4199150f0143SBarry Smith   c->inode.ibdiag      = 0;
4200150f0143SBarry Smith   c->inode.bdiag       = 0;
4201150f0143SBarry Smith   PetscFunctionReturn(0);
4202150f0143SBarry Smith }
4203150f0143SBarry Smith 
4204150f0143SBarry Smith #undef __FUNCT__
420517454e89SShri Abhyankar #define __FUNCT__ "MatGetRow_FactoredLU"
42068758e1faSBarry Smith PETSC_STATIC_INLINE PetscErrorCode MatGetRow_FactoredLU(PetscInt *cols,PetscInt nzl,PetscInt nzu,PetscInt nz,const PetscInt *ai,const PetscInt *aj,const PetscInt *adiag,PetscInt row)
420717454e89SShri Abhyankar {
42088758e1faSBarry Smith   PetscInt       k;
42098758e1faSBarry Smith   const PetscInt *vi;
42106e111a19SKarl Rupp 
421117454e89SShri Abhyankar   PetscFunctionBegin;
421217454e89SShri Abhyankar   vi = aj + ai[row];
421317454e89SShri Abhyankar   for (k=0; k<nzl; k++) cols[k] = vi[k];
421417454e89SShri Abhyankar   vi        = aj + adiag[row];
421517454e89SShri Abhyankar   cols[nzl] = vi[0];
421617454e89SShri Abhyankar   vi        = aj + adiag[row+1]+1;
421717454e89SShri Abhyankar   for (k=0; k<nzu; k++) cols[nzl+1+k] = vi[k];
421817454e89SShri Abhyankar   PetscFunctionReturn(0);
421917454e89SShri Abhyankar }
42206936b636SHong Zhang /*
42216936b636SHong Zhang    Mat_CheckInode_FactorLU - Check Inode for factored seqaij matrix.
42226936b636SHong Zhang    Modified from Mat_CheckInode().
42236936b636SHong Zhang 
42246936b636SHong Zhang    Input Parameters:
4225abb87a52SBarry Smith .  Mat A - ILU or LU matrix factor
4226abb87a52SBarry Smith 
42276936b636SHong Zhang */
4228019b515eSShri Abhyankar #undef __FUNCT__
4229019b515eSShri Abhyankar #define __FUNCT__ "Mat_CheckInode_FactorLU"
4230abb87a52SBarry Smith PetscErrorCode Mat_CheckInode_FactorLU(Mat A)
4231019b515eSShri Abhyankar {
4232019b515eSShri Abhyankar   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
4233019b515eSShri Abhyankar   PetscErrorCode ierr;
4234019b515eSShri Abhyankar   PetscInt       i,j,m,nzl1,nzu1,nzl2,nzu2,nzx,nzy,node_count,blk_size;
42358758e1faSBarry Smith   PetscInt       *cols1,*cols2,*ns;
42368758e1faSBarry Smith   const PetscInt *ai = a->i,*aj = a->j, *adiag = a->diag;
4237ace3abfcSBarry Smith   PetscBool      flag;
4238019b515eSShri Abhyankar 
4239019b515eSShri Abhyankar   PetscFunctionBegin;
4240019b515eSShri Abhyankar   if (!a->inode.use)    PetscFunctionReturn(0);
4241abb87a52SBarry Smith   if (a->inode.checked) PetscFunctionReturn(0);
4242019b515eSShri Abhyankar 
4243019b515eSShri Abhyankar   m = A->rmap->n;
42442205254eSKarl Rupp   if (a->inode.size) ns = a->inode.size;
42452205254eSKarl Rupp   else {
42462205254eSKarl Rupp     ierr = PetscMalloc((m+1)*sizeof(PetscInt),&ns);CHKERRQ(ierr);
42472205254eSKarl Rupp   }
4248019b515eSShri Abhyankar 
4249019b515eSShri Abhyankar   i          = 0;
4250019b515eSShri Abhyankar   node_count = 0;
4251*dcca6d9dSJed Brown   ierr = PetscMalloc2(m,&cols1,m,&cols2);CHKERRQ(ierr);
4252019b515eSShri Abhyankar   while (i < m) {                /* For each row */
4253019b515eSShri Abhyankar     nzl1 = ai[i+1] - ai[i];       /* Number of nonzeros in L */
4254019b515eSShri Abhyankar     nzu1 = adiag[i] - adiag[i+1] - 1; /* Number of nonzeros in U excluding diagonal*/
4255019b515eSShri Abhyankar     nzx  = nzl1 + nzu1 + 1;
4256019b515eSShri Abhyankar     MatGetRow_FactoredLU(cols1,nzl1,nzu1,nzx,ai,aj,adiag,i);
4257019b515eSShri Abhyankar 
4258019b515eSShri Abhyankar     /* Limits the number of elements in a node to 'a->inode.limit' */
4259019b515eSShri Abhyankar     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
4260019b515eSShri Abhyankar       nzl2 = ai[j+1] - ai[j];
4261019b515eSShri Abhyankar       nzu2 = adiag[j] - adiag[j+1] - 1;
4262019b515eSShri Abhyankar       nzy  = nzl2 + nzu2 + 1;
4263019b515eSShri Abhyankar       if (nzy != nzx) break;
426417454e89SShri Abhyankar       ierr = MatGetRow_FactoredLU(cols2,nzl2,nzu2,nzy,ai,aj,adiag,j);CHKERRQ(ierr);
4265019b515eSShri Abhyankar       ierr = PetscMemcmp(cols1,cols2,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr);
42668758e1faSBarry Smith       if (!flag) break;
4267019b515eSShri Abhyankar     }
4268019b515eSShri Abhyankar     ns[node_count++] = blk_size;
4269019b515eSShri Abhyankar     i                = j;
4270019b515eSShri Abhyankar   }
42718758e1faSBarry Smith   ierr             = PetscFree2(cols1,cols2);CHKERRQ(ierr);
4272019b515eSShri Abhyankar   /* If not enough inodes found,, do not use inode version of the routines */
4273be6adb11SBarry Smith   if (!m || node_count > .8*m) {
4274019b515eSShri Abhyankar     ierr = PetscFree(ns);CHKERRQ(ierr);
42752205254eSKarl Rupp 
4276019b515eSShri Abhyankar     a->inode.node_count = 0;
42770298fd71SBarry Smith     a->inode.size       = NULL;
4278019b515eSShri Abhyankar     a->inode.use        = PETSC_FALSE;
42792205254eSKarl Rupp 
4280019b515eSShri Abhyankar     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
4281019b515eSShri Abhyankar   } else {
4282019b515eSShri Abhyankar     A->ops->mult              = 0;
4283019b515eSShri Abhyankar     A->ops->sor               = 0;
4284019b515eSShri Abhyankar     A->ops->multadd           = 0;
4285019b515eSShri Abhyankar     A->ops->getrowij          = 0;
4286019b515eSShri Abhyankar     A->ops->restorerowij      = 0;
4287019b515eSShri Abhyankar     A->ops->getcolumnij       = 0;
4288019b515eSShri Abhyankar     A->ops->restorecolumnij   = 0;
4289019b515eSShri Abhyankar     A->ops->coloringpatch     = 0;
4290019b515eSShri Abhyankar     A->ops->multdiagonalblock = 0;
4291019b515eSShri Abhyankar     a->inode.node_count       = node_count;
4292019b515eSShri Abhyankar     a->inode.size             = ns;
42932205254eSKarl Rupp 
4294019b515eSShri Abhyankar     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
4295019b515eSShri Abhyankar   }
4296be6adb11SBarry Smith   a->inode.checked = PETSC_TRUE;
4297019b515eSShri Abhyankar   PetscFunctionReturn(0);
4298019b515eSShri Abhyankar }
4299019b515eSShri Abhyankar 
4300acf2f550SJed Brown #undef __FUNCT__
4301acf2f550SJed Brown #define __FUNCT__ "MatSeqAIJInvalidateDiagonal_Inode"
4302acf2f550SJed Brown PetscErrorCode MatSeqAIJInvalidateDiagonal_Inode(Mat A)
4303acf2f550SJed Brown {
4304acf2f550SJed Brown   Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data;
4305acf2f550SJed Brown 
4306acf2f550SJed Brown   PetscFunctionBegin;
4307acf2f550SJed Brown   a->inode.ibdiagvalid = PETSC_FALSE;
4308acf2f550SJed Brown   PetscFunctionReturn(0);
4309acf2f550SJed Brown }
4310acf2f550SJed Brown 
43114c1414c8SBarry Smith /*
43124c1414c8SBarry Smith      This is really ugly. if inodes are used this replaces the
43134c1414c8SBarry Smith   permutations with ones that correspond to rows/cols of the matrix
43144c1414c8SBarry Smith   rather then inode blocks
43154c1414c8SBarry Smith */
43164c1414c8SBarry Smith #undef __FUNCT__
43174c1414c8SBarry Smith #define __FUNCT__ "MatInodeAdjustForInodes"
43187087cfbeSBarry Smith PetscErrorCode  MatInodeAdjustForInodes(Mat A,IS *rperm,IS *cperm)
43194c1414c8SBarry Smith {
43204ac538c5SBarry Smith   PetscErrorCode ierr;
43214c1414c8SBarry Smith 
43224c1414c8SBarry Smith   PetscFunctionBegin;
43234ac538c5SBarry Smith   ierr = PetscTryMethod(A,"MatInodeAdjustForInodes_C",(Mat,IS*,IS*),(A,rperm,cperm));CHKERRQ(ierr);
43244c1414c8SBarry Smith   PetscFunctionReturn(0);
43254c1414c8SBarry Smith }
43264c1414c8SBarry Smith 
43274c1414c8SBarry Smith #undef __FUNCT__
432853c77d0aSJed Brown #define __FUNCT__ "MatInodeAdjustForInodes_SeqAIJ_Inode"
43297087cfbeSBarry Smith PetscErrorCode  MatInodeAdjustForInodes_SeqAIJ_Inode(Mat A,IS *rperm,IS *cperm)
43304c1414c8SBarry Smith {
43314c1414c8SBarry Smith   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
43324c1414c8SBarry Smith   PetscErrorCode ierr;
43335d0c19d7SBarry Smith   PetscInt       m = A->rmap->n,n = A->cmap->n,i,j,nslim_row = a->inode.node_count;
43345d0c19d7SBarry Smith   const PetscInt *ridx,*cidx;
43354c1414c8SBarry Smith   PetscInt       row,col,*permr,*permc,*ns_row =  a->inode.size,*tns,start_val,end_val,indx;
43364c1414c8SBarry Smith   PetscInt       nslim_col,*ns_col;
43374c1414c8SBarry Smith   IS             ris = *rperm,cis = *cperm;
43384c1414c8SBarry Smith 
43394c1414c8SBarry Smith   PetscFunctionBegin;
43404c1414c8SBarry Smith   if (!a->inode.size) PetscFunctionReturn(0); /* no inodes so return */
43414c1414c8SBarry Smith   if (a->inode.node_count == m) PetscFunctionReturn(0); /* all inodes are of size 1 */
43424c1414c8SBarry Smith 
43434c1414c8SBarry Smith   ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr);
43444c1414c8SBarry Smith   ierr = PetscMalloc((((nslim_row>nslim_col) ? nslim_row : nslim_col)+1)*sizeof(PetscInt),&tns);CHKERRQ(ierr);
4345*dcca6d9dSJed Brown   ierr = PetscMalloc2(m,&permr,n,&permc);CHKERRQ(ierr);
43464c1414c8SBarry Smith 
43474c1414c8SBarry Smith   ierr = ISGetIndices(ris,&ridx);CHKERRQ(ierr);
43484c1414c8SBarry Smith   ierr = ISGetIndices(cis,&cidx);CHKERRQ(ierr);
43494c1414c8SBarry Smith 
43504c1414c8SBarry Smith   /* Form the inode structure for the rows of permuted matric using inv perm*/
43514c1414c8SBarry Smith   for (i=0,tns[0]=0; i<nslim_row; ++i) tns[i+1] = tns[i] + ns_row[i];
43524c1414c8SBarry Smith 
43534c1414c8SBarry Smith   /* Construct the permutations for rows*/
43544c1414c8SBarry Smith   for (i=0,row = 0; i<nslim_row; ++i) {
43554c1414c8SBarry Smith     indx      = ridx[i];
43564c1414c8SBarry Smith     start_val = tns[indx];
43574c1414c8SBarry Smith     end_val   = tns[indx + 1];
43584c1414c8SBarry Smith     for (j=start_val; j<end_val; ++j,++row) permr[row]= j;
43594c1414c8SBarry Smith   }
43604c1414c8SBarry Smith 
43614c1414c8SBarry Smith   /* Form the inode structure for the columns of permuted matrix using inv perm*/
43624c1414c8SBarry Smith   for (i=0,tns[0]=0; i<nslim_col; ++i) tns[i+1] = tns[i] + ns_col[i];
43634c1414c8SBarry Smith 
43644c1414c8SBarry Smith   /* Construct permutations for columns */
43654c1414c8SBarry Smith   for (i=0,col=0; i<nslim_col; ++i) {
43664c1414c8SBarry Smith     indx      = cidx[i];
43674c1414c8SBarry Smith     start_val = tns[indx];
43684c1414c8SBarry Smith     end_val   = tns[indx + 1];
43694c1414c8SBarry Smith     for (j = start_val; j<end_val; ++j,++col) permc[col]= j;
43704c1414c8SBarry Smith   }
43714c1414c8SBarry Smith 
437270b3c8c7SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permr,PETSC_COPY_VALUES,rperm);CHKERRQ(ierr);
43730e83c824SBarry Smith   ierr = ISSetPermutation(*rperm);CHKERRQ(ierr);
437470b3c8c7SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permc,PETSC_COPY_VALUES,cperm);CHKERRQ(ierr);
43750e83c824SBarry Smith   ierr = ISSetPermutation(*cperm);CHKERRQ(ierr);
43764c1414c8SBarry Smith 
43774c1414c8SBarry Smith   ierr = ISRestoreIndices(ris,&ridx);CHKERRQ(ierr);
43784c1414c8SBarry Smith   ierr = ISRestoreIndices(cis,&cidx);CHKERRQ(ierr);
43794c1414c8SBarry Smith 
43804c1414c8SBarry Smith   ierr = PetscFree(ns_col);CHKERRQ(ierr);
43810e83c824SBarry Smith   ierr = PetscFree2(permr,permc);CHKERRQ(ierr);
43826bf464f9SBarry Smith   ierr = ISDestroy(&cis);CHKERRQ(ierr);
43836bf464f9SBarry Smith   ierr = ISDestroy(&ris);CHKERRQ(ierr);
43844c1414c8SBarry Smith   ierr = PetscFree(tns);CHKERRQ(ierr);
43854c1414c8SBarry Smith   PetscFunctionReturn(0);
43864c1414c8SBarry Smith }
43874c1414c8SBarry Smith 
43884c1414c8SBarry Smith #undef __FUNCT__
43894c1414c8SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes"
43904c1414c8SBarry Smith /*@C
43914c1414c8SBarry Smith    MatInodeGetInodeSizes - Returns the inode information of the Inode matrix.
43924c1414c8SBarry Smith 
43933f9fe445SBarry Smith    Not Collective
43944c1414c8SBarry Smith 
43954c1414c8SBarry Smith    Input Parameter:
43964c1414c8SBarry Smith .  A - the Inode matrix or matrix derived from the Inode class -- e.g., SeqAIJ
43974c1414c8SBarry Smith 
43984c1414c8SBarry Smith    Output Parameter:
43994c1414c8SBarry Smith +  node_count - no of inodes present in the matrix.
44004c1414c8SBarry Smith .  sizes      - an array of size node_count,with sizes of each inode.
44014c1414c8SBarry Smith -  limit      - the max size used to generate the inodes.
44024c1414c8SBarry Smith 
44034c1414c8SBarry Smith    Level: advanced
44044c1414c8SBarry Smith 
44054c1414c8SBarry Smith    Notes: This routine returns some internal storage information
44064c1414c8SBarry Smith    of the matrix, it is intended to be used by advanced users.
44074c1414c8SBarry Smith    It should be called after the matrix is assembled.
44084c1414c8SBarry Smith    The contents of the sizes[] array should not be changed.
44090298fd71SBarry Smith    NULL may be passed for information not requested.
44104c1414c8SBarry Smith 
44114c1414c8SBarry Smith .keywords: matrix, seqaij, get, inode
44124c1414c8SBarry Smith 
44134c1414c8SBarry Smith .seealso: MatGetInfo()
44144c1414c8SBarry Smith @*/
44157087cfbeSBarry Smith PetscErrorCode  MatInodeGetInodeSizes(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit)
44164c1414c8SBarry Smith {
44174c1414c8SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscInt*,PetscInt*[],PetscInt*);
44184c1414c8SBarry Smith 
44194c1414c8SBarry Smith   PetscFunctionBegin;
4420e32f2f54SBarry Smith   if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix");
44210005d66cSJed Brown   ierr = PetscObjectQueryFunction((PetscObject)A,"MatInodeGetInodeSizes_C",&f);CHKERRQ(ierr);
44224c1414c8SBarry Smith   if (f) {
44234c1414c8SBarry Smith     ierr = (*f)(A,node_count,sizes,limit);CHKERRQ(ierr);
44244c1414c8SBarry Smith   }
44254c1414c8SBarry Smith   PetscFunctionReturn(0);
44264c1414c8SBarry Smith }
44274c1414c8SBarry Smith 
44284c1414c8SBarry Smith #undef __FUNCT__
44294108e4d5SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes_SeqAIJ_Inode"
44307087cfbeSBarry Smith PetscErrorCode  MatInodeGetInodeSizes_SeqAIJ_Inode(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit)
44314c1414c8SBarry Smith {
44324c1414c8SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
44334c1414c8SBarry Smith 
44344c1414c8SBarry Smith   PetscFunctionBegin;
44354c1414c8SBarry Smith   if (node_count) *node_count = a->inode.node_count;
44364c1414c8SBarry Smith   if (sizes)      *sizes      = a->inode.size;
44374c1414c8SBarry Smith   if (limit)      *limit      = a->inode.limit;
44384c1414c8SBarry Smith   PetscFunctionReturn(0);
44394c1414c8SBarry Smith }
4440