14c1414c8SBarry Smith #define PETSCMAT_DLL 24c1414c8SBarry Smith 34c1414c8SBarry Smith /* 44c1414c8SBarry Smith This file provides high performance routines for the Inode format (compressed sparse row) 54c1414c8SBarry Smith by taking advantage of rows with identical nonzero structure (I-nodes). 64c1414c8SBarry Smith */ 77c4f633dSBarry Smith #include "../src/mat/impls/aij/seq/aij.h" 84c1414c8SBarry Smith 94c1414c8SBarry Smith #undef __FUNCT__ 104c1414c8SBarry Smith #define __FUNCT__ "Mat_CreateColInode" 114c1414c8SBarry Smith static PetscErrorCode Mat_CreateColInode(Mat A,PetscInt* size,PetscInt ** ns) 124c1414c8SBarry Smith { 134c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 144c1414c8SBarry Smith PetscErrorCode ierr; 154c1414c8SBarry Smith PetscInt i,count,m,n,min_mn,*ns_row,*ns_col; 164c1414c8SBarry Smith 174c1414c8SBarry Smith PetscFunctionBegin; 18d0f46423SBarry Smith n = A->cmap->n; 19d0f46423SBarry Smith m = A->rmap->n; 204c1414c8SBarry Smith ns_row = a->inode.size; 214c1414c8SBarry Smith 224c1414c8SBarry Smith min_mn = (m < n) ? m : n; 234c1414c8SBarry Smith if (!ns) { 244c1414c8SBarry Smith for (count=0,i=0; count<min_mn; count+=ns_row[i],i++); 254c1414c8SBarry Smith for(; count+1 < n; count++,i++); 264c1414c8SBarry Smith if (count < n) { 274c1414c8SBarry Smith i++; 284c1414c8SBarry Smith } 294c1414c8SBarry Smith *size = i; 304c1414c8SBarry Smith PetscFunctionReturn(0); 314c1414c8SBarry Smith } 324c1414c8SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscInt),&ns_col);CHKERRQ(ierr); 334c1414c8SBarry Smith 344c1414c8SBarry Smith /* Use the same row structure wherever feasible. */ 354c1414c8SBarry Smith for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) { 364c1414c8SBarry Smith ns_col[i] = ns_row[i]; 374c1414c8SBarry Smith } 384c1414c8SBarry Smith 394c1414c8SBarry Smith /* if m < n; pad up the remainder with inode_limit */ 404c1414c8SBarry Smith for(; count+1 < n; count++,i++) { 414c1414c8SBarry Smith ns_col[i] = 1; 424c1414c8SBarry Smith } 434c1414c8SBarry Smith /* The last node is the odd ball. padd it up with the remaining rows; */ 444c1414c8SBarry Smith if (count < n) { 454c1414c8SBarry Smith ns_col[i] = n - count; 464c1414c8SBarry Smith i++; 474c1414c8SBarry Smith } else if (count > n) { 484c1414c8SBarry Smith /* Adjust for the over estimation */ 494c1414c8SBarry Smith ns_col[i-1] += n - count; 504c1414c8SBarry Smith } 514c1414c8SBarry Smith *size = i; 524c1414c8SBarry Smith *ns = ns_col; 534c1414c8SBarry Smith PetscFunctionReturn(0); 544c1414c8SBarry Smith } 554c1414c8SBarry Smith 564c1414c8SBarry Smith 574c1414c8SBarry Smith /* 584c1414c8SBarry Smith This builds symmetric version of nonzero structure, 594c1414c8SBarry Smith */ 604c1414c8SBarry Smith #undef __FUNCT__ 614108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode_Symmetric" 624108e4d5SBarry Smith static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Symmetric(Mat A,PetscInt *iia[],PetscInt *jja[],PetscInt ishift,PetscInt oshift) 634c1414c8SBarry Smith { 644c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 654c1414c8SBarry Smith PetscErrorCode ierr; 664c1414c8SBarry Smith PetscInt *work,*ia,*ja,*j,nz,nslim_row,nslim_col,m,row,col,*jmax,n; 674c1414c8SBarry Smith PetscInt *tns,*tvc,*ns_row = a->inode.size,*ns_col,nsz,i1,i2,*ai= a->i,*aj = a->j; 684c1414c8SBarry Smith 694c1414c8SBarry Smith PetscFunctionBegin; 704c1414c8SBarry Smith nslim_row = a->inode.node_count; 71d0f46423SBarry Smith m = A->rmap->n; 72d0f46423SBarry Smith n = A->cmap->n; 734108e4d5SBarry Smith if (m != n) SETERRQ(PETSC_ERR_SUP,"MatGetRowIJ_SeqAIJ_Inode_Symmetric: Matrix should be square"); 744c1414c8SBarry Smith 754c1414c8SBarry Smith /* Use the row_inode as column_inode */ 764c1414c8SBarry Smith nslim_col = nslim_row; 774c1414c8SBarry Smith ns_col = ns_row; 784c1414c8SBarry Smith 794c1414c8SBarry Smith /* allocate space for reformated inode structure */ 804c1414c8SBarry Smith ierr = PetscMalloc((nslim_col+1)*sizeof(PetscInt),&tns);CHKERRQ(ierr); 814c1414c8SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscInt),&tvc);CHKERRQ(ierr); 824c1414c8SBarry Smith for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1]+ ns_row[i1]; 834c1414c8SBarry Smith 844c1414c8SBarry Smith for (i1=0,col=0; i1<nslim_col; ++i1){ 854c1414c8SBarry Smith nsz = ns_col[i1]; 864c1414c8SBarry Smith for (i2=0; i2<nsz; ++i2,++col) 874c1414c8SBarry Smith tvc[col] = i1; 884c1414c8SBarry Smith } 894c1414c8SBarry Smith /* allocate space for row pointers */ 904c1414c8SBarry Smith ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&ia);CHKERRQ(ierr); 914c1414c8SBarry Smith *iia = ia; 924c1414c8SBarry Smith ierr = PetscMemzero(ia,(nslim_row+1)*sizeof(PetscInt));CHKERRQ(ierr); 934c1414c8SBarry Smith ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&work);CHKERRQ(ierr); 944c1414c8SBarry Smith 954c1414c8SBarry Smith /* determine the number of columns in each row */ 964c1414c8SBarry Smith ia[0] = oshift; 974c1414c8SBarry Smith for (i1=0,row=0 ; i1<nslim_row; row+=ns_row[i1],i1++) { 984c1414c8SBarry Smith 994c1414c8SBarry Smith j = aj + ai[row] + ishift; 1004c1414c8SBarry Smith jmax = aj + ai[row+1] + ishift; 1014c1414c8SBarry Smith i2 = 0; 1024c1414c8SBarry Smith col = *j++ + ishift; 1034c1414c8SBarry Smith i2 = tvc[col]; 1044c1414c8SBarry Smith while (i2<i1 && j<jmax) { /* 1.[-xx-d-xx--] 2.[-xx-------],off-diagonal elemets */ 1054c1414c8SBarry Smith ia[i1+1]++; 1064c1414c8SBarry Smith ia[i2+1]++; 1074c1414c8SBarry Smith i2++; /* Start col of next node */ 1084c1414c8SBarry Smith while(((col=*j+ishift)<tns[i2]) && (j<jmax)) ++j; 1094c1414c8SBarry Smith i2 = tvc[col]; 1104c1414c8SBarry Smith } 1114c1414c8SBarry Smith if(i2 == i1) ia[i2+1]++; /* now the diagonal element */ 1124c1414c8SBarry Smith } 1134c1414c8SBarry Smith 1144c1414c8SBarry Smith /* shift ia[i] to point to next row */ 1154c1414c8SBarry Smith for (i1=1; i1<nslim_row+1; i1++) { 1164c1414c8SBarry Smith row = ia[i1-1]; 1174c1414c8SBarry Smith ia[i1] += row; 1184c1414c8SBarry Smith work[i1-1] = row - oshift; 1194c1414c8SBarry Smith } 1204c1414c8SBarry Smith 1214c1414c8SBarry Smith /* allocate space for column pointers */ 1224c1414c8SBarry Smith nz = ia[nslim_row] + (!ishift); 1234c1414c8SBarry Smith ierr = PetscMalloc(nz*sizeof(PetscInt),&ja);CHKERRQ(ierr); 1244c1414c8SBarry Smith *jja = ja; 1254c1414c8SBarry Smith 1264c1414c8SBarry Smith /* loop over lower triangular part putting into ja */ 1274c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row += ns_row[i1],i1++) { 1284c1414c8SBarry Smith j = aj + ai[row] + ishift; 1294c1414c8SBarry Smith jmax = aj + ai[row+1] + ishift; 1304c1414c8SBarry Smith i2 = 0; /* Col inode index */ 1314c1414c8SBarry Smith col = *j++ + ishift; 1324c1414c8SBarry Smith i2 = tvc[col]; 1334c1414c8SBarry Smith while (i2<i1 && j<jmax) { 1344c1414c8SBarry Smith ja[work[i2]++] = i1 + oshift; 1354c1414c8SBarry Smith ja[work[i1]++] = i2 + oshift; 1364c1414c8SBarry Smith ++i2; 1374c1414c8SBarry Smith while(((col=*j+ishift)< tns[i2])&&(j<jmax)) ++j; /* Skip rest col indices in this node */ 1384c1414c8SBarry Smith i2 = tvc[col]; 1394c1414c8SBarry Smith } 1404c1414c8SBarry Smith if (i2 == i1) ja[work[i1]++] = i2 + oshift; 1414c1414c8SBarry Smith 1424c1414c8SBarry Smith } 1434c1414c8SBarry Smith ierr = PetscFree(work);CHKERRQ(ierr); 1444c1414c8SBarry Smith ierr = PetscFree(tns);CHKERRQ(ierr); 1454c1414c8SBarry Smith ierr = PetscFree(tvc);CHKERRQ(ierr); 1464c1414c8SBarry Smith PetscFunctionReturn(0); 1474c1414c8SBarry Smith } 1484c1414c8SBarry Smith 1494c1414c8SBarry Smith /* 1504c1414c8SBarry Smith This builds nonsymmetric version of nonzero structure, 1514c1414c8SBarry Smith */ 1524c1414c8SBarry Smith #undef __FUNCT__ 1534108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric" 1544108e4d5SBarry Smith static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,PetscInt *iia[],PetscInt *jja[],PetscInt ishift,PetscInt oshift) 1554c1414c8SBarry Smith { 1564c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1574c1414c8SBarry Smith PetscErrorCode ierr; 1584c1414c8SBarry Smith PetscInt *work,*ia,*ja,*j,nz,nslim_row,n,row,col,*ns_col,nslim_col; 1594c1414c8SBarry Smith PetscInt *tns,*tvc,*ns_row = a->inode.size,nsz,i1,i2,*ai= a->i,*aj = a->j; 1604c1414c8SBarry Smith 1614c1414c8SBarry Smith PetscFunctionBegin; 1624c1414c8SBarry Smith nslim_row = a->inode.node_count; 163d0f46423SBarry Smith n = A->cmap->n; 1644c1414c8SBarry Smith 1654c1414c8SBarry Smith /* Create The column_inode for this matrix */ 1664c1414c8SBarry Smith ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr); 1674c1414c8SBarry Smith 1684c1414c8SBarry Smith /* allocate space for reformated column_inode structure */ 1694c1414c8SBarry Smith ierr = PetscMalloc((nslim_col +1)*sizeof(PetscInt),&tns);CHKERRQ(ierr); 1704c1414c8SBarry Smith ierr = PetscMalloc((n +1)*sizeof(PetscInt),&tvc);CHKERRQ(ierr); 1714c1414c8SBarry Smith for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1]; 1724c1414c8SBarry Smith 1734c1414c8SBarry Smith for (i1=0,col=0; i1<nslim_col; ++i1){ 1744c1414c8SBarry Smith nsz = ns_col[i1]; 1754c1414c8SBarry Smith for (i2=0; i2<nsz; ++i2,++col) 1764c1414c8SBarry Smith tvc[col] = i1; 1774c1414c8SBarry Smith } 1784c1414c8SBarry Smith /* allocate space for row pointers */ 1794c1414c8SBarry Smith ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&ia);CHKERRQ(ierr); 1804c1414c8SBarry Smith *iia = ia; 1814c1414c8SBarry Smith ierr = PetscMemzero(ia,(nslim_row+1)*sizeof(PetscInt));CHKERRQ(ierr); 1824c1414c8SBarry Smith ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&work);CHKERRQ(ierr); 1834c1414c8SBarry Smith 1844c1414c8SBarry Smith /* determine the number of columns in each row */ 1854c1414c8SBarry Smith ia[0] = oshift; 1864c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 1874c1414c8SBarry Smith j = aj + ai[row] + ishift; 1884c1414c8SBarry Smith col = *j++ + ishift; 1894c1414c8SBarry Smith i2 = tvc[col]; 1904c1414c8SBarry Smith nz = ai[row+1] - ai[row]; 1914c1414c8SBarry Smith while (nz-- > 0) { /* off-diagonal elemets */ 1924c1414c8SBarry Smith ia[i1+1]++; 1934c1414c8SBarry Smith i2++; /* Start col of next node */ 1944c1414c8SBarry Smith while (((col = *j++ + ishift) < tns[i2]) && nz > 0) {nz--;} 1954c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 1964c1414c8SBarry Smith } 1974c1414c8SBarry Smith } 1984c1414c8SBarry Smith 1994c1414c8SBarry Smith /* shift ia[i] to point to next row */ 2004c1414c8SBarry Smith for (i1=1; i1<nslim_row+1; i1++) { 2014c1414c8SBarry Smith row = ia[i1-1]; 2024c1414c8SBarry Smith ia[i1] += row; 2034c1414c8SBarry Smith work[i1-1] = row - oshift; 2044c1414c8SBarry Smith } 2054c1414c8SBarry Smith 2064c1414c8SBarry Smith /* allocate space for column pointers */ 2074c1414c8SBarry Smith nz = ia[nslim_row] + (!ishift); 2084c1414c8SBarry Smith ierr = PetscMalloc(nz*sizeof(PetscInt),&ja);CHKERRQ(ierr); 2094c1414c8SBarry Smith *jja = ja; 2104c1414c8SBarry Smith 2114c1414c8SBarry Smith /* loop over matrix putting into ja */ 2124c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 2134c1414c8SBarry Smith j = aj + ai[row] + ishift; 2144c1414c8SBarry Smith i2 = 0; /* Col inode index */ 2154c1414c8SBarry Smith col = *j++ + ishift; 2164c1414c8SBarry Smith i2 = tvc[col]; 2174c1414c8SBarry Smith nz = ai[row+1] - ai[row]; 2184c1414c8SBarry Smith while (nz-- > 0) { 2194c1414c8SBarry Smith ja[work[i1]++] = i2 + oshift; 2204c1414c8SBarry Smith ++i2; 2214c1414c8SBarry Smith while(((col = *j++ + ishift) < tns[i2]) && nz > 0) {nz--;} 2224c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 2234c1414c8SBarry Smith } 2244c1414c8SBarry Smith } 2254c1414c8SBarry Smith ierr = PetscFree(ns_col);CHKERRQ(ierr); 2264c1414c8SBarry Smith ierr = PetscFree(work);CHKERRQ(ierr); 2274c1414c8SBarry Smith ierr = PetscFree(tns);CHKERRQ(ierr); 2284c1414c8SBarry Smith ierr = PetscFree(tvc);CHKERRQ(ierr); 2294c1414c8SBarry Smith PetscFunctionReturn(0); 2304c1414c8SBarry Smith } 2314c1414c8SBarry Smith 2324c1414c8SBarry Smith #undef __FUNCT__ 2334108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode" 2344108e4d5SBarry Smith static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscTruth symmetric,PetscTruth blockcompressed,PetscInt *n,PetscInt *ia[],PetscInt *ja[],PetscTruth *done) 2354c1414c8SBarry Smith { 2364c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2374c1414c8SBarry Smith PetscErrorCode ierr; 2384c1414c8SBarry Smith 2394c1414c8SBarry Smith PetscFunctionBegin; 2404c1414c8SBarry Smith *n = a->inode.node_count; 2414c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 2428f7157efSSatish Balay if (!blockcompressed) { 2438f7157efSSatish Balay ierr = MatGetRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);; 2448f7157efSSatish Balay } else if (symmetric) { 2454108e4d5SBarry Smith ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 2464c1414c8SBarry Smith } else { 2474108e4d5SBarry Smith ierr = MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 2484c1414c8SBarry Smith } 2494c1414c8SBarry Smith PetscFunctionReturn(0); 2504c1414c8SBarry Smith } 2514c1414c8SBarry Smith 2524c1414c8SBarry Smith #undef __FUNCT__ 2534108e4d5SBarry Smith #define __FUNCT__ "MatRestoreRowIJ_SeqAIJ_Inode" 2544108e4d5SBarry Smith static PetscErrorCode MatRestoreRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscTruth symmetric,PetscTruth blockcompressed,PetscInt *n,PetscInt *ia[],PetscInt *ja[],PetscTruth *done) 2554c1414c8SBarry Smith { 2564c1414c8SBarry Smith PetscErrorCode ierr; 2574c1414c8SBarry Smith 2584c1414c8SBarry Smith PetscFunctionBegin; 2594c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 2608f7157efSSatish Balay 2618f7157efSSatish Balay if (!blockcompressed) { 2628f7157efSSatish Balay ierr = MatRestoreRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);; 2638f7157efSSatish Balay } else { 2644c1414c8SBarry Smith ierr = PetscFree(*ia);CHKERRQ(ierr); 2654c1414c8SBarry Smith ierr = PetscFree(*ja);CHKERRQ(ierr); 2668f7157efSSatish Balay } 2678f7157efSSatish Balay 2684c1414c8SBarry Smith PetscFunctionReturn(0); 2694c1414c8SBarry Smith } 2704c1414c8SBarry Smith 2714c1414c8SBarry Smith /* ----------------------------------------------------------- */ 2724c1414c8SBarry Smith 2734c1414c8SBarry Smith #undef __FUNCT__ 2744108e4d5SBarry Smith #define __FUNCT__ "MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric" 2754108e4d5SBarry Smith static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,PetscInt *iia[],PetscInt *jja[],PetscInt ishift,PetscInt oshift) 2764c1414c8SBarry Smith { 2774c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2784c1414c8SBarry Smith PetscErrorCode ierr; 2794c1414c8SBarry Smith PetscInt *work,*ia,*ja,*j,nz,nslim_row, n,row,col,*ns_col,nslim_col; 2804c1414c8SBarry Smith PetscInt *tns,*tvc,*ns_row = a->inode.size,nsz,i1,i2,*ai= a->i,*aj = a->j; 2814c1414c8SBarry Smith 2824c1414c8SBarry Smith PetscFunctionBegin; 2834c1414c8SBarry Smith nslim_row = a->inode.node_count; 284d0f46423SBarry Smith n = A->cmap->n; 2854c1414c8SBarry Smith 2864c1414c8SBarry Smith /* Create The column_inode for this matrix */ 2874c1414c8SBarry Smith ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr); 2884c1414c8SBarry Smith 2894c1414c8SBarry Smith /* allocate space for reformated column_inode structure */ 2904c1414c8SBarry Smith ierr = PetscMalloc((nslim_col + 1)*sizeof(PetscInt),&tns);CHKERRQ(ierr); 2914c1414c8SBarry Smith ierr = PetscMalloc((n + 1)*sizeof(PetscInt),&tvc);CHKERRQ(ierr); 2924c1414c8SBarry Smith for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1]; 2934c1414c8SBarry Smith 2944c1414c8SBarry Smith for (i1=0,col=0; i1<nslim_col; ++i1){ 2954c1414c8SBarry Smith nsz = ns_col[i1]; 2964c1414c8SBarry Smith for (i2=0; i2<nsz; ++i2,++col) 2974c1414c8SBarry Smith tvc[col] = i1; 2984c1414c8SBarry Smith } 2994c1414c8SBarry Smith /* allocate space for column pointers */ 3004c1414c8SBarry Smith ierr = PetscMalloc((nslim_col+1)*sizeof(PetscInt),&ia);CHKERRQ(ierr); 3014c1414c8SBarry Smith *iia = ia; 3024c1414c8SBarry Smith ierr = PetscMemzero(ia,(nslim_col+1)*sizeof(PetscInt));CHKERRQ(ierr); 3034c1414c8SBarry Smith ierr = PetscMalloc((nslim_col+1)*sizeof(PetscInt),&work);CHKERRQ(ierr); 3044c1414c8SBarry Smith 3054c1414c8SBarry Smith /* determine the number of columns in each row */ 3064c1414c8SBarry Smith ia[0] = oshift; 3074c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 3084c1414c8SBarry Smith j = aj + ai[row] + ishift; 3094c1414c8SBarry Smith col = *j++ + ishift; 3104c1414c8SBarry Smith i2 = tvc[col]; 3114c1414c8SBarry Smith nz = ai[row+1] - ai[row]; 3124c1414c8SBarry Smith while (nz-- > 0) { /* off-diagonal elemets */ 3134c1414c8SBarry Smith /* ia[i1+1]++; */ 3144c1414c8SBarry Smith ia[i2+1]++; 3154c1414c8SBarry Smith i2++; 3164c1414c8SBarry Smith while (((col = *j++ + ishift) < tns[i2]) && nz > 0) {nz--;} 3174c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 3184c1414c8SBarry Smith } 3194c1414c8SBarry Smith } 3204c1414c8SBarry Smith 3214c1414c8SBarry Smith /* shift ia[i] to point to next col */ 3224c1414c8SBarry Smith for (i1=1; i1<nslim_col+1; i1++) { 3234c1414c8SBarry Smith col = ia[i1-1]; 3244c1414c8SBarry Smith ia[i1] += col; 3254c1414c8SBarry Smith work[i1-1] = col - oshift; 3264c1414c8SBarry Smith } 3274c1414c8SBarry Smith 3284c1414c8SBarry Smith /* allocate space for column pointers */ 3294c1414c8SBarry Smith nz = ia[nslim_col] + (!ishift); 3304c1414c8SBarry Smith ierr = PetscMalloc(nz*sizeof(PetscInt),&ja);CHKERRQ(ierr); 3314c1414c8SBarry Smith *jja = ja; 3324c1414c8SBarry Smith 3334c1414c8SBarry Smith /* loop over matrix putting into ja */ 3344c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 3354c1414c8SBarry Smith j = aj + ai[row] + ishift; 3364c1414c8SBarry Smith i2 = 0; /* Col inode index */ 3374c1414c8SBarry Smith col = *j++ + ishift; 3384c1414c8SBarry Smith i2 = tvc[col]; 3394c1414c8SBarry Smith nz = ai[row+1] - ai[row]; 3404c1414c8SBarry Smith while (nz-- > 0) { 3414c1414c8SBarry Smith /* ja[work[i1]++] = i2 + oshift; */ 3424c1414c8SBarry Smith ja[work[i2]++] = i1 + oshift; 3434c1414c8SBarry Smith i2++; 3444c1414c8SBarry Smith while(((col = *j++ + ishift) < tns[i2]) && nz > 0) {nz--;} 3454c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 3464c1414c8SBarry Smith } 3474c1414c8SBarry Smith } 3484c1414c8SBarry Smith ierr = PetscFree(ns_col);CHKERRQ(ierr); 3494c1414c8SBarry Smith ierr = PetscFree(work);CHKERRQ(ierr); 3504c1414c8SBarry Smith ierr = PetscFree(tns);CHKERRQ(ierr); 3514c1414c8SBarry Smith ierr = PetscFree(tvc);CHKERRQ(ierr); 3524c1414c8SBarry Smith PetscFunctionReturn(0); 3534c1414c8SBarry Smith } 3544c1414c8SBarry Smith 3554c1414c8SBarry Smith #undef __FUNCT__ 3564108e4d5SBarry Smith #define __FUNCT__ "MatGetColumnIJ_SeqAIJ_Inode" 3574108e4d5SBarry Smith static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscTruth symmetric,PetscTruth blockcompressed,PetscInt *n,PetscInt *ia[],PetscInt *ja[],PetscTruth *done) 3584c1414c8SBarry Smith { 3594c1414c8SBarry Smith PetscErrorCode ierr; 3604c1414c8SBarry Smith 3614c1414c8SBarry Smith PetscFunctionBegin; 3624c1414c8SBarry Smith ierr = Mat_CreateColInode(A,n,PETSC_NULL);CHKERRQ(ierr); 3634c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 3644c1414c8SBarry Smith 3658f7157efSSatish Balay if (!blockcompressed) { 3668f7157efSSatish Balay ierr = MatGetColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);; 3678f7157efSSatish Balay } else if (symmetric) { 3684c1414c8SBarry Smith /* Since the indices are symmetric it does'nt matter */ 3694108e4d5SBarry Smith ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 3704c1414c8SBarry Smith } else { 3714108e4d5SBarry Smith ierr = MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 3724c1414c8SBarry Smith } 3734c1414c8SBarry Smith PetscFunctionReturn(0); 3744c1414c8SBarry Smith } 3754c1414c8SBarry Smith 3764c1414c8SBarry Smith #undef __FUNCT__ 3774108e4d5SBarry Smith #define __FUNCT__ "MatRestoreColumnIJ_SeqAIJ_Inode" 3784108e4d5SBarry Smith static PetscErrorCode MatRestoreColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscTruth symmetric,PetscTruth blockcompressed,PetscInt *n,PetscInt *ia[],PetscInt *ja[],PetscTruth *done) 3794c1414c8SBarry Smith { 3804c1414c8SBarry Smith PetscErrorCode ierr; 3814c1414c8SBarry Smith 3824c1414c8SBarry Smith PetscFunctionBegin; 3834c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 3848f7157efSSatish Balay if (!blockcompressed) { 3858f7157efSSatish Balay ierr = MatRestoreColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);; 3868f7157efSSatish Balay } else { 3874c1414c8SBarry Smith ierr = PetscFree(*ia);CHKERRQ(ierr); 3884c1414c8SBarry Smith ierr = PetscFree(*ja);CHKERRQ(ierr); 3898f7157efSSatish Balay } 3904c1414c8SBarry Smith PetscFunctionReturn(0); 3914c1414c8SBarry Smith } 3924c1414c8SBarry Smith 3934c1414c8SBarry Smith /* ----------------------------------------------------------- */ 3944c1414c8SBarry Smith 3954c1414c8SBarry Smith #undef __FUNCT__ 3964108e4d5SBarry Smith #define __FUNCT__ "MatMult_SeqAIJ_Inode" 3974108e4d5SBarry Smith static PetscErrorCode MatMult_SeqAIJ_Inode(Mat A,Vec xx,Vec yy) 3984c1414c8SBarry Smith { 3994c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 4004c1414c8SBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5,tmp0,tmp1; 401d9fead3dSBarry Smith PetscScalar *y; 402dd6ea824SBarry Smith const PetscScalar *x; 403dd6ea824SBarry Smith const MatScalar *v1,*v2,*v3,*v4,*v5; 4044c1414c8SBarry Smith PetscErrorCode ierr; 40598c9bda7SSatish Balay PetscInt *idx,i1,i2,n,i,row,node_max,*ns,*ii,nsz,sz,nonzerorow=0; 4064c1414c8SBarry Smith 4074c1414c8SBarry Smith #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 4084c1414c8SBarry Smith #pragma disjoint(*x,*y,*v1,*v2,*v3,*v4,*v5) 4094c1414c8SBarry Smith #endif 4104c1414c8SBarry Smith 4114c1414c8SBarry Smith PetscFunctionBegin; 4124c1414c8SBarry Smith if (!a->inode.size) SETERRQ(PETSC_ERR_COR,"Missing Inode Structure"); 4134c1414c8SBarry Smith node_max = a->inode.node_count; 4144c1414c8SBarry Smith ns = a->inode.size; /* Node Size array */ 415d9fead3dSBarry Smith ierr = VecGetArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 4164c1414c8SBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 4174c1414c8SBarry Smith idx = a->j; 4184c1414c8SBarry Smith v1 = a->a; 4194c1414c8SBarry Smith ii = a->i; 4204c1414c8SBarry Smith 4214c1414c8SBarry Smith for (i = 0,row = 0; i< node_max; ++i){ 4224c1414c8SBarry Smith nsz = ns[i]; 4234c1414c8SBarry Smith n = ii[1] - ii[0]; 42498c9bda7SSatish Balay nonzerorow += (n>0)*nsz; 4254c1414c8SBarry Smith ii += nsz; 4263b5996c7SJed Brown PetscPrefetchBlock(idx+nsz*n,n,0,0); /* Prefetch the indices for the block row after the current one */ 4273b5996c7SJed Brown PetscPrefetchBlock(v1+nsz*n,nsz*n,0,0); /* Prefetch the values for the block row after the current one */ 4284c1414c8SBarry Smith sz = n; /* No of non zeros in this row */ 4294c1414c8SBarry Smith /* Switch on the size of Node */ 4304c1414c8SBarry Smith switch (nsz){ /* Each loop in 'case' is unrolled */ 4314c1414c8SBarry Smith case 1 : 43275567043SBarry Smith sum1 = 0.; 4334c1414c8SBarry Smith 4344c1414c8SBarry Smith for(n = 0; n< sz-1; n+=2) { 4354c1414c8SBarry Smith i1 = idx[0]; /* The instructions are ordered to */ 4364c1414c8SBarry Smith i2 = idx[1]; /* make the compiler's job easy */ 4374c1414c8SBarry Smith idx += 2; 4384c1414c8SBarry Smith tmp0 = x[i1]; 4394c1414c8SBarry Smith tmp1 = x[i2]; 4404c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 4414c1414c8SBarry Smith } 4424c1414c8SBarry Smith 4434c1414c8SBarry Smith if (n == sz-1){ /* Take care of the last nonzero */ 4444c1414c8SBarry Smith tmp0 = x[*idx++]; 4454c1414c8SBarry Smith sum1 += *v1++ * tmp0; 4464c1414c8SBarry Smith } 4474c1414c8SBarry Smith y[row++]=sum1; 4484c1414c8SBarry Smith break; 4494c1414c8SBarry Smith case 2: 45075567043SBarry Smith sum1 = 0.; 45175567043SBarry Smith sum2 = 0.; 4524c1414c8SBarry Smith v2 = v1 + n; 4534c1414c8SBarry Smith 4544c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 4554c1414c8SBarry Smith i1 = idx[0]; 4564c1414c8SBarry Smith i2 = idx[1]; 4574c1414c8SBarry Smith idx += 2; 4584c1414c8SBarry Smith tmp0 = x[i1]; 4594c1414c8SBarry Smith tmp1 = x[i2]; 4604c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 4614c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 4624c1414c8SBarry Smith } 4634c1414c8SBarry Smith if (n == sz-1){ 4644c1414c8SBarry Smith tmp0 = x[*idx++]; 4654c1414c8SBarry Smith sum1 += *v1++ * tmp0; 4664c1414c8SBarry Smith sum2 += *v2++ * tmp0; 4674c1414c8SBarry Smith } 4684c1414c8SBarry Smith y[row++]=sum1; 4694c1414c8SBarry Smith y[row++]=sum2; 4704c1414c8SBarry Smith v1 =v2; /* Since the next block to be processed starts there*/ 4714c1414c8SBarry Smith idx +=sz; 4724c1414c8SBarry Smith break; 4734c1414c8SBarry Smith case 3: 47475567043SBarry Smith sum1 = 0.; 47575567043SBarry Smith sum2 = 0.; 47675567043SBarry Smith sum3 = 0.; 4774c1414c8SBarry Smith v2 = v1 + n; 4784c1414c8SBarry Smith v3 = v2 + n; 4794c1414c8SBarry Smith 4804c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 4814c1414c8SBarry Smith i1 = idx[0]; 4824c1414c8SBarry Smith i2 = idx[1]; 4834c1414c8SBarry Smith idx += 2; 4844c1414c8SBarry Smith tmp0 = x[i1]; 4854c1414c8SBarry Smith tmp1 = x[i2]; 4864c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 4874c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 4884c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 4894c1414c8SBarry Smith } 4904c1414c8SBarry Smith if (n == sz-1){ 4914c1414c8SBarry Smith tmp0 = x[*idx++]; 4924c1414c8SBarry Smith sum1 += *v1++ * tmp0; 4934c1414c8SBarry Smith sum2 += *v2++ * tmp0; 4944c1414c8SBarry Smith sum3 += *v3++ * tmp0; 4954c1414c8SBarry Smith } 4964c1414c8SBarry Smith y[row++]=sum1; 4974c1414c8SBarry Smith y[row++]=sum2; 4984c1414c8SBarry Smith y[row++]=sum3; 4994c1414c8SBarry Smith v1 =v3; /* Since the next block to be processed starts there*/ 5004c1414c8SBarry Smith idx +=2*sz; 5014c1414c8SBarry Smith break; 5024c1414c8SBarry Smith case 4: 50375567043SBarry Smith sum1 = 0.; 50475567043SBarry Smith sum2 = 0.; 50575567043SBarry Smith sum3 = 0.; 50675567043SBarry Smith sum4 = 0.; 5074c1414c8SBarry Smith v2 = v1 + n; 5084c1414c8SBarry Smith v3 = v2 + n; 5094c1414c8SBarry Smith v4 = v3 + n; 5104c1414c8SBarry Smith 5114c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 5124c1414c8SBarry Smith i1 = idx[0]; 5134c1414c8SBarry Smith i2 = idx[1]; 5144c1414c8SBarry Smith idx += 2; 5154c1414c8SBarry Smith tmp0 = x[i1]; 5164c1414c8SBarry Smith tmp1 = x[i2]; 5174c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 5184c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 5194c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 5204c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 5214c1414c8SBarry Smith } 5224c1414c8SBarry Smith if (n == sz-1){ 5234c1414c8SBarry Smith tmp0 = x[*idx++]; 5244c1414c8SBarry Smith sum1 += *v1++ * tmp0; 5254c1414c8SBarry Smith sum2 += *v2++ * tmp0; 5264c1414c8SBarry Smith sum3 += *v3++ * tmp0; 5274c1414c8SBarry Smith sum4 += *v4++ * tmp0; 5284c1414c8SBarry Smith } 5294c1414c8SBarry Smith y[row++]=sum1; 5304c1414c8SBarry Smith y[row++]=sum2; 5314c1414c8SBarry Smith y[row++]=sum3; 5324c1414c8SBarry Smith y[row++]=sum4; 5334c1414c8SBarry Smith v1 =v4; /* Since the next block to be processed starts there*/ 5344c1414c8SBarry Smith idx +=3*sz; 5354c1414c8SBarry Smith break; 5364c1414c8SBarry Smith case 5: 53775567043SBarry Smith sum1 = 0.; 53875567043SBarry Smith sum2 = 0.; 53975567043SBarry Smith sum3 = 0.; 54075567043SBarry Smith sum4 = 0.; 54175567043SBarry Smith sum5 = 0.; 5424c1414c8SBarry Smith v2 = v1 + n; 5434c1414c8SBarry Smith v3 = v2 + n; 5444c1414c8SBarry Smith v4 = v3 + n; 5454c1414c8SBarry Smith v5 = v4 + n; 5464c1414c8SBarry Smith 5474c1414c8SBarry Smith for (n = 0; n<sz-1; n+=2) { 5484c1414c8SBarry Smith i1 = idx[0]; 5494c1414c8SBarry Smith i2 = idx[1]; 5504c1414c8SBarry Smith idx += 2; 5514c1414c8SBarry Smith tmp0 = x[i1]; 5524c1414c8SBarry Smith tmp1 = x[i2]; 5534c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 5544c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 5554c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 5564c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 5574c1414c8SBarry Smith sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2; 5584c1414c8SBarry Smith } 5594c1414c8SBarry Smith if (n == sz-1){ 5604c1414c8SBarry Smith tmp0 = x[*idx++]; 5614c1414c8SBarry Smith sum1 += *v1++ * tmp0; 5624c1414c8SBarry Smith sum2 += *v2++ * tmp0; 5634c1414c8SBarry Smith sum3 += *v3++ * tmp0; 5644c1414c8SBarry Smith sum4 += *v4++ * tmp0; 5654c1414c8SBarry Smith sum5 += *v5++ * tmp0; 5664c1414c8SBarry Smith } 5674c1414c8SBarry Smith y[row++]=sum1; 5684c1414c8SBarry Smith y[row++]=sum2; 5694c1414c8SBarry Smith y[row++]=sum3; 5704c1414c8SBarry Smith y[row++]=sum4; 5714c1414c8SBarry Smith y[row++]=sum5; 5724c1414c8SBarry Smith v1 =v5; /* Since the next block to be processed starts there */ 5734c1414c8SBarry Smith idx +=4*sz; 5744c1414c8SBarry Smith break; 5754c1414c8SBarry Smith default : 5764c1414c8SBarry Smith SETERRQ(PETSC_ERR_COR,"Node size not yet supported"); 5774c1414c8SBarry Smith } 5784c1414c8SBarry Smith } 579d9fead3dSBarry Smith ierr = VecRestoreArray(xx,(PetscScalar**)&x);CHKERRQ(ierr); 5804c1414c8SBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 581dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr); 5824c1414c8SBarry Smith PetscFunctionReturn(0); 5834c1414c8SBarry Smith } 5844c1414c8SBarry Smith /* ----------------------------------------------------------- */ 5854108e4d5SBarry Smith /* Almost same code as the MatMult_SeqAIJ_Inode() */ 5864c1414c8SBarry Smith #undef __FUNCT__ 5874108e4d5SBarry Smith #define __FUNCT__ "MatMultAdd_SeqAIJ_Inode" 5884108e4d5SBarry Smith static PetscErrorCode MatMultAdd_SeqAIJ_Inode(Mat A,Vec xx,Vec zz,Vec yy) 5894c1414c8SBarry Smith { 5904c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 5914c1414c8SBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5,tmp0,tmp1; 592dd6ea824SBarry Smith MatScalar *v1,*v2,*v3,*v4,*v5; 593dd6ea824SBarry Smith PetscScalar *x,*y,*z,*zt; 5944c1414c8SBarry Smith PetscErrorCode ierr; 5954c1414c8SBarry Smith PetscInt *idx,i1,i2,n,i,row,node_max,*ns,*ii,nsz,sz; 5964c1414c8SBarry Smith 5974c1414c8SBarry Smith PetscFunctionBegin; 5984c1414c8SBarry Smith if (!a->inode.size) SETERRQ(PETSC_ERR_COR,"Missing Inode Structure"); 5994c1414c8SBarry Smith node_max = a->inode.node_count; 6004c1414c8SBarry Smith ns = a->inode.size; /* Node Size array */ 6014c1414c8SBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 6024c1414c8SBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 6034c1414c8SBarry Smith if (zz != yy) { 6044c1414c8SBarry Smith ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 6054c1414c8SBarry Smith } else { 6064c1414c8SBarry Smith z = y; 6074c1414c8SBarry Smith } 6084c1414c8SBarry Smith zt = z; 6094c1414c8SBarry Smith 6104c1414c8SBarry Smith idx = a->j; 6114c1414c8SBarry Smith v1 = a->a; 6124c1414c8SBarry Smith ii = a->i; 6134c1414c8SBarry Smith 6144c1414c8SBarry Smith for (i = 0,row = 0; i< node_max; ++i){ 6154c1414c8SBarry Smith nsz = ns[i]; 6164c1414c8SBarry Smith n = ii[1] - ii[0]; 6174c1414c8SBarry Smith ii += nsz; 6184c1414c8SBarry Smith sz = n; /* No of non zeros in this row */ 6194c1414c8SBarry Smith /* Switch on the size of Node */ 6204c1414c8SBarry Smith switch (nsz){ /* Each loop in 'case' is unrolled */ 6214c1414c8SBarry Smith case 1 : 6224c1414c8SBarry Smith sum1 = *zt++; 6234c1414c8SBarry Smith 6244c1414c8SBarry Smith for(n = 0; n< sz-1; n+=2) { 6254c1414c8SBarry Smith i1 = idx[0]; /* The instructions are ordered to */ 6264c1414c8SBarry Smith i2 = idx[1]; /* make the compiler's job easy */ 6274c1414c8SBarry Smith idx += 2; 6284c1414c8SBarry Smith tmp0 = x[i1]; 6294c1414c8SBarry Smith tmp1 = x[i2]; 6304c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 6314c1414c8SBarry Smith } 6324c1414c8SBarry Smith 6334c1414c8SBarry Smith if(n == sz-1){ /* Take care of the last nonzero */ 6344c1414c8SBarry Smith tmp0 = x[*idx++]; 6354c1414c8SBarry Smith sum1 += *v1++ * tmp0; 6364c1414c8SBarry Smith } 6374c1414c8SBarry Smith y[row++]=sum1; 6384c1414c8SBarry Smith break; 6394c1414c8SBarry Smith case 2: 6404c1414c8SBarry Smith sum1 = *zt++; 6414c1414c8SBarry Smith sum2 = *zt++; 6424c1414c8SBarry Smith v2 = v1 + n; 6434c1414c8SBarry Smith 6444c1414c8SBarry Smith for(n = 0; n< sz-1; n+=2) { 6454c1414c8SBarry Smith i1 = idx[0]; 6464c1414c8SBarry Smith i2 = idx[1]; 6474c1414c8SBarry Smith idx += 2; 6484c1414c8SBarry Smith tmp0 = x[i1]; 6494c1414c8SBarry Smith tmp1 = x[i2]; 6504c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 6514c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 6524c1414c8SBarry Smith } 6534c1414c8SBarry Smith if(n == sz-1){ 6544c1414c8SBarry Smith tmp0 = x[*idx++]; 6554c1414c8SBarry Smith sum1 += *v1++ * tmp0; 6564c1414c8SBarry Smith sum2 += *v2++ * tmp0; 6574c1414c8SBarry Smith } 6584c1414c8SBarry Smith y[row++]=sum1; 6594c1414c8SBarry Smith y[row++]=sum2; 6604c1414c8SBarry Smith v1 =v2; /* Since the next block to be processed starts there*/ 6614c1414c8SBarry Smith idx +=sz; 6624c1414c8SBarry Smith break; 6634c1414c8SBarry Smith case 3: 6644c1414c8SBarry Smith sum1 = *zt++; 6654c1414c8SBarry Smith sum2 = *zt++; 6664c1414c8SBarry Smith sum3 = *zt++; 6674c1414c8SBarry Smith v2 = v1 + n; 6684c1414c8SBarry Smith v3 = v2 + n; 6694c1414c8SBarry Smith 6704c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 6714c1414c8SBarry Smith i1 = idx[0]; 6724c1414c8SBarry Smith i2 = idx[1]; 6734c1414c8SBarry Smith idx += 2; 6744c1414c8SBarry Smith tmp0 = x[i1]; 6754c1414c8SBarry Smith tmp1 = x[i2]; 6764c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 6774c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 6784c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 6794c1414c8SBarry Smith } 6804c1414c8SBarry Smith if (n == sz-1){ 6814c1414c8SBarry Smith tmp0 = x[*idx++]; 6824c1414c8SBarry Smith sum1 += *v1++ * tmp0; 6834c1414c8SBarry Smith sum2 += *v2++ * tmp0; 6844c1414c8SBarry Smith sum3 += *v3++ * tmp0; 6854c1414c8SBarry Smith } 6864c1414c8SBarry Smith y[row++]=sum1; 6874c1414c8SBarry Smith y[row++]=sum2; 6884c1414c8SBarry Smith y[row++]=sum3; 6894c1414c8SBarry Smith v1 =v3; /* Since the next block to be processed starts there*/ 6904c1414c8SBarry Smith idx +=2*sz; 6914c1414c8SBarry Smith break; 6924c1414c8SBarry Smith case 4: 6934c1414c8SBarry Smith sum1 = *zt++; 6944c1414c8SBarry Smith sum2 = *zt++; 6954c1414c8SBarry Smith sum3 = *zt++; 6964c1414c8SBarry Smith sum4 = *zt++; 6974c1414c8SBarry Smith v2 = v1 + n; 6984c1414c8SBarry Smith v3 = v2 + n; 6994c1414c8SBarry Smith v4 = v3 + n; 7004c1414c8SBarry Smith 7014c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 7024c1414c8SBarry Smith i1 = idx[0]; 7034c1414c8SBarry Smith i2 = idx[1]; 7044c1414c8SBarry Smith idx += 2; 7054c1414c8SBarry Smith tmp0 = x[i1]; 7064c1414c8SBarry Smith tmp1 = x[i2]; 7074c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 7084c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 7094c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 7104c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 7114c1414c8SBarry Smith } 7124c1414c8SBarry Smith if (n == sz-1){ 7134c1414c8SBarry Smith tmp0 = x[*idx++]; 7144c1414c8SBarry Smith sum1 += *v1++ * tmp0; 7154c1414c8SBarry Smith sum2 += *v2++ * tmp0; 7164c1414c8SBarry Smith sum3 += *v3++ * tmp0; 7174c1414c8SBarry Smith sum4 += *v4++ * tmp0; 7184c1414c8SBarry Smith } 7194c1414c8SBarry Smith y[row++]=sum1; 7204c1414c8SBarry Smith y[row++]=sum2; 7214c1414c8SBarry Smith y[row++]=sum3; 7224c1414c8SBarry Smith y[row++]=sum4; 7234c1414c8SBarry Smith v1 =v4; /* Since the next block to be processed starts there*/ 7244c1414c8SBarry Smith idx +=3*sz; 7254c1414c8SBarry Smith break; 7264c1414c8SBarry Smith case 5: 7274c1414c8SBarry Smith sum1 = *zt++; 7284c1414c8SBarry Smith sum2 = *zt++; 7294c1414c8SBarry Smith sum3 = *zt++; 7304c1414c8SBarry Smith sum4 = *zt++; 7314c1414c8SBarry Smith sum5 = *zt++; 7324c1414c8SBarry Smith v2 = v1 + n; 7334c1414c8SBarry Smith v3 = v2 + n; 7344c1414c8SBarry Smith v4 = v3 + n; 7354c1414c8SBarry Smith v5 = v4 + n; 7364c1414c8SBarry Smith 7374c1414c8SBarry Smith for (n = 0; n<sz-1; n+=2) { 7384c1414c8SBarry Smith i1 = idx[0]; 7394c1414c8SBarry Smith i2 = idx[1]; 7404c1414c8SBarry Smith idx += 2; 7414c1414c8SBarry Smith tmp0 = x[i1]; 7424c1414c8SBarry Smith tmp1 = x[i2]; 7434c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 7444c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 7454c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 7464c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 7474c1414c8SBarry Smith sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2; 7484c1414c8SBarry Smith } 7494c1414c8SBarry Smith if(n == sz-1){ 7504c1414c8SBarry Smith tmp0 = x[*idx++]; 7514c1414c8SBarry Smith sum1 += *v1++ * tmp0; 7524c1414c8SBarry Smith sum2 += *v2++ * tmp0; 7534c1414c8SBarry Smith sum3 += *v3++ * tmp0; 7544c1414c8SBarry Smith sum4 += *v4++ * tmp0; 7554c1414c8SBarry Smith sum5 += *v5++ * tmp0; 7564c1414c8SBarry Smith } 7574c1414c8SBarry Smith y[row++]=sum1; 7584c1414c8SBarry Smith y[row++]=sum2; 7594c1414c8SBarry Smith y[row++]=sum3; 7604c1414c8SBarry Smith y[row++]=sum4; 7614c1414c8SBarry Smith y[row++]=sum5; 7624c1414c8SBarry Smith v1 =v5; /* Since the next block to be processed starts there */ 7634c1414c8SBarry Smith idx +=4*sz; 7644c1414c8SBarry Smith break; 7654c1414c8SBarry Smith default : 7664c1414c8SBarry Smith SETERRQ(PETSC_ERR_COR,"Node size not yet supported"); 7674c1414c8SBarry Smith } 7684c1414c8SBarry Smith } 7694c1414c8SBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 7704c1414c8SBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 7714c1414c8SBarry Smith if (zz != yy) { 7724c1414c8SBarry Smith ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 7734c1414c8SBarry Smith } 774dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 7754c1414c8SBarry Smith PetscFunctionReturn(0); 7764c1414c8SBarry Smith } 7774c1414c8SBarry Smith 7784c1414c8SBarry Smith /* ----------------------------------------------------------- */ 7794c1414c8SBarry Smith #undef __FUNCT__ 7804108e4d5SBarry Smith #define __FUNCT__ "MatSolve_SeqAIJ_Inode" 7814108e4d5SBarry Smith PetscErrorCode MatSolve_SeqAIJ_Inode(Mat A,Vec bb,Vec xx) 7824c1414c8SBarry Smith { 7834c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 7844c1414c8SBarry Smith IS iscol = a->col,isrow = a->row; 7854c1414c8SBarry Smith PetscErrorCode ierr; 7865d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 7875d0c19d7SBarry Smith PetscInt i,j,n = A->rmap->n,*ai = a->i,nz,*a_j = a->j; 7885d0c19d7SBarry Smith PetscInt node_max,*ns,row,nsz,aii,*vi,*ad,*aj,i0,i1; 789d9fead3dSBarry Smith PetscScalar *x,*tmp,*tmps,tmp0,tmp1; 790d9fead3dSBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5; 791dd6ea824SBarry Smith const MatScalar *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa; 792dd6ea824SBarry Smith const PetscScalar *b; 7934c1414c8SBarry Smith 7944c1414c8SBarry Smith PetscFunctionBegin; 7954c1414c8SBarry Smith if (!a->inode.size) SETERRQ(PETSC_ERR_COR,"Missing Inode Structure"); 7964c1414c8SBarry Smith node_max = a->inode.node_count; 7974c1414c8SBarry Smith ns = a->inode.size; /* Node Size array */ 7984c1414c8SBarry Smith 799d9fead3dSBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 8004c1414c8SBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 8014c1414c8SBarry Smith tmp = a->solve_work; 8024c1414c8SBarry Smith 8034c1414c8SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 8044c1414c8SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 8054c1414c8SBarry Smith 8064c1414c8SBarry Smith /* forward solve the lower triangular */ 8074c1414c8SBarry Smith tmps = tmp ; 8084c1414c8SBarry Smith aa = a_a ; 8094c1414c8SBarry Smith aj = a_j ; 8104c1414c8SBarry Smith ad = a->diag; 8114c1414c8SBarry Smith 8124c1414c8SBarry Smith for (i = 0,row = 0; i< node_max; ++i){ 8134c1414c8SBarry Smith nsz = ns[i]; 8144c1414c8SBarry Smith aii = ai[row]; 8154c1414c8SBarry Smith v1 = aa + aii; 8164c1414c8SBarry Smith vi = aj + aii; 8174c1414c8SBarry Smith nz = ad[row]- aii; 818*26549573SJed Brown if (i < node_max-1) { 819*26549573SJed Brown /* Prefetch the block after the current one, the prefetch itself can't cause a memory error, 820*26549573SJed Brown * but our indexing to determine it's size could. */ 821*26549573SJed Brown PetscPrefetchBlock(aj+ai[row+nsz],ad[row+nsz]-ai[row+nsz],0,0); /* indices */ 822*26549573SJed Brown /* In my tests, it seems to be better to fetch entire rows instead of just the lower-triangular part */ 823*26549573SJed Brown PetscPrefetchBlock(aa+ai[row+nsz],ad[row+nsz+ns[i+1]-1]-ai[row+nsz],0,0); 824*26549573SJed 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); */ 825*26549573SJed Brown } 8264c1414c8SBarry Smith 8274c1414c8SBarry Smith switch (nsz){ /* Each loop in 'case' is unrolled */ 8284c1414c8SBarry Smith case 1 : 8294c1414c8SBarry Smith sum1 = b[*r++]; 8304c1414c8SBarry Smith for(j=0; j<nz-1; j+=2){ 8314c1414c8SBarry Smith i0 = vi[0]; 8324c1414c8SBarry Smith i1 = vi[1]; 8334c1414c8SBarry Smith vi +=2; 8344c1414c8SBarry Smith tmp0 = tmps[i0]; 8354c1414c8SBarry Smith tmp1 = tmps[i1]; 8364c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 8374c1414c8SBarry Smith } 8384c1414c8SBarry Smith if(j == nz-1){ 8394c1414c8SBarry Smith tmp0 = tmps[*vi++]; 8404c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 8414c1414c8SBarry Smith } 8424c1414c8SBarry Smith tmp[row ++]=sum1; 8434c1414c8SBarry Smith break; 8444c1414c8SBarry Smith case 2: 8454c1414c8SBarry Smith sum1 = b[*r++]; 8464c1414c8SBarry Smith sum2 = b[*r++]; 8474c1414c8SBarry Smith v2 = aa + ai[row+1]; 8484c1414c8SBarry Smith 8494c1414c8SBarry Smith for(j=0; j<nz-1; j+=2){ 8504c1414c8SBarry Smith i0 = vi[0]; 8514c1414c8SBarry Smith i1 = vi[1]; 8524c1414c8SBarry Smith vi +=2; 8534c1414c8SBarry Smith tmp0 = tmps[i0]; 8544c1414c8SBarry Smith tmp1 = tmps[i1]; 8554c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 8564c1414c8SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 8574c1414c8SBarry Smith } 8584c1414c8SBarry Smith if(j == nz-1){ 8594c1414c8SBarry Smith tmp0 = tmps[*vi++]; 8604c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 8614c1414c8SBarry Smith sum2 -= *v2++ *tmp0; 8624c1414c8SBarry Smith } 8634c1414c8SBarry Smith sum2 -= *v2++ * sum1; 8644c1414c8SBarry Smith tmp[row ++]=sum1; 8654c1414c8SBarry Smith tmp[row ++]=sum2; 8664c1414c8SBarry Smith break; 8674c1414c8SBarry Smith case 3: 8684c1414c8SBarry Smith sum1 = b[*r++]; 8694c1414c8SBarry Smith sum2 = b[*r++]; 8704c1414c8SBarry Smith sum3 = b[*r++]; 8714c1414c8SBarry Smith v2 = aa + ai[row+1]; 8724c1414c8SBarry Smith v3 = aa + ai[row+2]; 8734c1414c8SBarry Smith 8744c1414c8SBarry Smith for (j=0; j<nz-1; j+=2){ 8754c1414c8SBarry Smith i0 = vi[0]; 8764c1414c8SBarry Smith i1 = vi[1]; 8774c1414c8SBarry Smith vi +=2; 8784c1414c8SBarry Smith tmp0 = tmps[i0]; 8794c1414c8SBarry Smith tmp1 = tmps[i1]; 8804c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 8814c1414c8SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 8824c1414c8SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 8834c1414c8SBarry Smith } 8844c1414c8SBarry Smith if (j == nz-1){ 8854c1414c8SBarry Smith tmp0 = tmps[*vi++]; 8864c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 8874c1414c8SBarry Smith sum2 -= *v2++ *tmp0; 8884c1414c8SBarry Smith sum3 -= *v3++ *tmp0; 8894c1414c8SBarry Smith } 8904c1414c8SBarry Smith sum2 -= *v2++ * sum1; 8914c1414c8SBarry Smith sum3 -= *v3++ * sum1; 8924c1414c8SBarry Smith sum3 -= *v3++ * sum2; 8934c1414c8SBarry Smith tmp[row ++]=sum1; 8944c1414c8SBarry Smith tmp[row ++]=sum2; 8954c1414c8SBarry Smith tmp[row ++]=sum3; 8964c1414c8SBarry Smith break; 8974c1414c8SBarry Smith 8984c1414c8SBarry Smith case 4: 8994c1414c8SBarry Smith sum1 = b[*r++]; 9004c1414c8SBarry Smith sum2 = b[*r++]; 9014c1414c8SBarry Smith sum3 = b[*r++]; 9024c1414c8SBarry Smith sum4 = b[*r++]; 9034c1414c8SBarry Smith v2 = aa + ai[row+1]; 9044c1414c8SBarry Smith v3 = aa + ai[row+2]; 9054c1414c8SBarry Smith v4 = aa + ai[row+3]; 9064c1414c8SBarry Smith 9074c1414c8SBarry Smith for (j=0; j<nz-1; j+=2){ 9084c1414c8SBarry Smith i0 = vi[0]; 9094c1414c8SBarry Smith i1 = vi[1]; 9104c1414c8SBarry Smith vi +=2; 9114c1414c8SBarry Smith tmp0 = tmps[i0]; 9124c1414c8SBarry Smith tmp1 = tmps[i1]; 9134c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 9144c1414c8SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 9154c1414c8SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 9164c1414c8SBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 9174c1414c8SBarry Smith } 9184c1414c8SBarry Smith if (j == nz-1){ 9194c1414c8SBarry Smith tmp0 = tmps[*vi++]; 9204c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 9214c1414c8SBarry Smith sum2 -= *v2++ *tmp0; 9224c1414c8SBarry Smith sum3 -= *v3++ *tmp0; 9234c1414c8SBarry Smith sum4 -= *v4++ *tmp0; 9244c1414c8SBarry Smith } 9254c1414c8SBarry Smith sum2 -= *v2++ * sum1; 9264c1414c8SBarry Smith sum3 -= *v3++ * sum1; 9274c1414c8SBarry Smith sum4 -= *v4++ * sum1; 9284c1414c8SBarry Smith sum3 -= *v3++ * sum2; 9294c1414c8SBarry Smith sum4 -= *v4++ * sum2; 9304c1414c8SBarry Smith sum4 -= *v4++ * sum3; 9314c1414c8SBarry Smith 9324c1414c8SBarry Smith tmp[row ++]=sum1; 9334c1414c8SBarry Smith tmp[row ++]=sum2; 9344c1414c8SBarry Smith tmp[row ++]=sum3; 9354c1414c8SBarry Smith tmp[row ++]=sum4; 9364c1414c8SBarry Smith break; 9374c1414c8SBarry Smith case 5: 9384c1414c8SBarry Smith sum1 = b[*r++]; 9394c1414c8SBarry Smith sum2 = b[*r++]; 9404c1414c8SBarry Smith sum3 = b[*r++]; 9414c1414c8SBarry Smith sum4 = b[*r++]; 9424c1414c8SBarry Smith sum5 = b[*r++]; 9434c1414c8SBarry Smith v2 = aa + ai[row+1]; 9444c1414c8SBarry Smith v3 = aa + ai[row+2]; 9454c1414c8SBarry Smith v4 = aa + ai[row+3]; 9464c1414c8SBarry Smith v5 = aa + ai[row+4]; 9474c1414c8SBarry Smith 9484c1414c8SBarry Smith for (j=0; j<nz-1; j+=2){ 9494c1414c8SBarry Smith i0 = vi[0]; 9504c1414c8SBarry Smith i1 = vi[1]; 9514c1414c8SBarry Smith vi +=2; 9524c1414c8SBarry Smith tmp0 = tmps[i0]; 9534c1414c8SBarry Smith tmp1 = tmps[i1]; 9544c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 9554c1414c8SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 9564c1414c8SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 9574c1414c8SBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 9584c1414c8SBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 9594c1414c8SBarry Smith } 9604c1414c8SBarry Smith if (j == nz-1){ 9614c1414c8SBarry Smith tmp0 = tmps[*vi++]; 9624c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 9634c1414c8SBarry Smith sum2 -= *v2++ *tmp0; 9644c1414c8SBarry Smith sum3 -= *v3++ *tmp0; 9654c1414c8SBarry Smith sum4 -= *v4++ *tmp0; 9664c1414c8SBarry Smith sum5 -= *v5++ *tmp0; 9674c1414c8SBarry Smith } 9684c1414c8SBarry Smith 9694c1414c8SBarry Smith sum2 -= *v2++ * sum1; 9704c1414c8SBarry Smith sum3 -= *v3++ * sum1; 9714c1414c8SBarry Smith sum4 -= *v4++ * sum1; 9724c1414c8SBarry Smith sum5 -= *v5++ * sum1; 9734c1414c8SBarry Smith sum3 -= *v3++ * sum2; 9744c1414c8SBarry Smith sum4 -= *v4++ * sum2; 9754c1414c8SBarry Smith sum5 -= *v5++ * sum2; 9764c1414c8SBarry Smith sum4 -= *v4++ * sum3; 9774c1414c8SBarry Smith sum5 -= *v5++ * sum3; 9784c1414c8SBarry Smith sum5 -= *v5++ * sum4; 9794c1414c8SBarry Smith 9804c1414c8SBarry Smith tmp[row ++]=sum1; 9814c1414c8SBarry Smith tmp[row ++]=sum2; 9824c1414c8SBarry Smith tmp[row ++]=sum3; 9834c1414c8SBarry Smith tmp[row ++]=sum4; 9844c1414c8SBarry Smith tmp[row ++]=sum5; 9854c1414c8SBarry Smith break; 9864c1414c8SBarry Smith default: 9874c1414c8SBarry Smith SETERRQ(PETSC_ERR_COR,"Node size not yet supported \n"); 9884c1414c8SBarry Smith } 9894c1414c8SBarry Smith } 9904c1414c8SBarry Smith /* backward solve the upper triangular */ 9914c1414c8SBarry Smith for (i=node_max -1 ,row = n-1 ; i>=0; i--){ 9924c1414c8SBarry Smith nsz = ns[i]; 9934c1414c8SBarry Smith aii = ai[row+1] -1; 9944c1414c8SBarry Smith v1 = aa + aii; 9954c1414c8SBarry Smith vi = aj + aii; 9964c1414c8SBarry Smith nz = aii- ad[row]; 9974c1414c8SBarry Smith switch (nsz){ /* Each loop in 'case' is unrolled */ 9984c1414c8SBarry Smith case 1 : 9994c1414c8SBarry Smith sum1 = tmp[row]; 10004c1414c8SBarry Smith 10014c1414c8SBarry Smith for(j=nz ; j>1; j-=2){ 10024c1414c8SBarry Smith vi -=2; 10034c1414c8SBarry Smith i0 = vi[2]; 10044c1414c8SBarry Smith i1 = vi[1]; 10054c1414c8SBarry Smith tmp0 = tmps[i0]; 10064c1414c8SBarry Smith tmp1 = tmps[i1]; 10074c1414c8SBarry Smith v1 -= 2; 10084c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 10094c1414c8SBarry Smith } 10104c1414c8SBarry Smith if (j==1){ 10114c1414c8SBarry Smith tmp0 = tmps[*vi--]; 10124c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 10134c1414c8SBarry Smith } 10144c1414c8SBarry Smith x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 10154c1414c8SBarry Smith break; 10164c1414c8SBarry Smith case 2 : 10174c1414c8SBarry Smith sum1 = tmp[row]; 10184c1414c8SBarry Smith sum2 = tmp[row -1]; 10194c1414c8SBarry Smith v2 = aa + ai[row]-1; 10204c1414c8SBarry Smith for (j=nz ; j>1; j-=2){ 10214c1414c8SBarry Smith vi -=2; 10224c1414c8SBarry Smith i0 = vi[2]; 10234c1414c8SBarry Smith i1 = vi[1]; 10244c1414c8SBarry Smith tmp0 = tmps[i0]; 10254c1414c8SBarry Smith tmp1 = tmps[i1]; 10264c1414c8SBarry Smith v1 -= 2; 10274c1414c8SBarry Smith v2 -= 2; 10284c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 10294c1414c8SBarry Smith sum2 -= v2[2] * tmp0 + v2[1] * tmp1; 10304c1414c8SBarry Smith } 10314c1414c8SBarry Smith if (j==1){ 10324c1414c8SBarry Smith tmp0 = tmps[*vi--]; 10334c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 10344c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 10354c1414c8SBarry Smith } 10364c1414c8SBarry Smith 10374c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 10384c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 10394c1414c8SBarry Smith x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--; 10404c1414c8SBarry Smith break; 10414c1414c8SBarry Smith case 3 : 10424c1414c8SBarry Smith sum1 = tmp[row]; 10434c1414c8SBarry Smith sum2 = tmp[row -1]; 10444c1414c8SBarry Smith sum3 = tmp[row -2]; 10454c1414c8SBarry Smith v2 = aa + ai[row]-1; 10464c1414c8SBarry Smith v3 = aa + ai[row -1]-1; 10474c1414c8SBarry Smith for (j=nz ; j>1; j-=2){ 10484c1414c8SBarry Smith vi -=2; 10494c1414c8SBarry Smith i0 = vi[2]; 10504c1414c8SBarry Smith i1 = vi[1]; 10514c1414c8SBarry Smith tmp0 = tmps[i0]; 10524c1414c8SBarry Smith tmp1 = tmps[i1]; 10534c1414c8SBarry Smith v1 -= 2; 10544c1414c8SBarry Smith v2 -= 2; 10554c1414c8SBarry Smith v3 -= 2; 10564c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 10574c1414c8SBarry Smith sum2 -= v2[2] * tmp0 + v2[1] * tmp1; 10584c1414c8SBarry Smith sum3 -= v3[2] * tmp0 + v3[1] * tmp1; 10594c1414c8SBarry Smith } 10604c1414c8SBarry Smith if (j==1){ 10614c1414c8SBarry Smith tmp0 = tmps[*vi--]; 10624c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 10634c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 10644c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 10654c1414c8SBarry Smith } 10664c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 10674c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 10684c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 10694c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--; 10704c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 10714c1414c8SBarry Smith x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--; 10724c1414c8SBarry Smith 10734c1414c8SBarry Smith break; 10744c1414c8SBarry Smith case 4 : 10754c1414c8SBarry Smith sum1 = tmp[row]; 10764c1414c8SBarry Smith sum2 = tmp[row -1]; 10774c1414c8SBarry Smith sum3 = tmp[row -2]; 10784c1414c8SBarry Smith sum4 = tmp[row -3]; 10794c1414c8SBarry Smith v2 = aa + ai[row]-1; 10804c1414c8SBarry Smith v3 = aa + ai[row -1]-1; 10814c1414c8SBarry Smith v4 = aa + ai[row -2]-1; 10824c1414c8SBarry Smith 10834c1414c8SBarry Smith for (j=nz ; j>1; j-=2){ 10844c1414c8SBarry Smith vi -=2; 10854c1414c8SBarry Smith i0 = vi[2]; 10864c1414c8SBarry Smith i1 = vi[1]; 10874c1414c8SBarry Smith tmp0 = tmps[i0]; 10884c1414c8SBarry Smith tmp1 = tmps[i1]; 10894c1414c8SBarry Smith v1 -= 2; 10904c1414c8SBarry Smith v2 -= 2; 10914c1414c8SBarry Smith v3 -= 2; 10924c1414c8SBarry Smith v4 -= 2; 10934c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 10944c1414c8SBarry Smith sum2 -= v2[2] * tmp0 + v2[1] * tmp1; 10954c1414c8SBarry Smith sum3 -= v3[2] * tmp0 + v3[1] * tmp1; 10964c1414c8SBarry Smith sum4 -= v4[2] * tmp0 + v4[1] * tmp1; 10974c1414c8SBarry Smith } 10984c1414c8SBarry Smith if (j==1){ 10994c1414c8SBarry Smith tmp0 = tmps[*vi--]; 11004c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 11014c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 11024c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11034c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11044c1414c8SBarry Smith } 11054c1414c8SBarry Smith 11064c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 11074c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 11084c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11094c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11104c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--; 11114c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11124c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11134c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--; 11144c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11154c1414c8SBarry Smith x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--; 11164c1414c8SBarry Smith break; 11174c1414c8SBarry Smith case 5 : 11184c1414c8SBarry Smith sum1 = tmp[row]; 11194c1414c8SBarry Smith sum2 = tmp[row -1]; 11204c1414c8SBarry Smith sum3 = tmp[row -2]; 11214c1414c8SBarry Smith sum4 = tmp[row -3]; 11224c1414c8SBarry Smith sum5 = tmp[row -4]; 11234c1414c8SBarry Smith v2 = aa + ai[row]-1; 11244c1414c8SBarry Smith v3 = aa + ai[row -1]-1; 11254c1414c8SBarry Smith v4 = aa + ai[row -2]-1; 11264c1414c8SBarry Smith v5 = aa + ai[row -3]-1; 11274c1414c8SBarry Smith for (j=nz ; j>1; j-=2){ 11284c1414c8SBarry Smith vi -= 2; 11294c1414c8SBarry Smith i0 = vi[2]; 11304c1414c8SBarry Smith i1 = vi[1]; 11314c1414c8SBarry Smith tmp0 = tmps[i0]; 11324c1414c8SBarry Smith tmp1 = tmps[i1]; 11334c1414c8SBarry Smith v1 -= 2; 11344c1414c8SBarry Smith v2 -= 2; 11354c1414c8SBarry Smith v3 -= 2; 11364c1414c8SBarry Smith v4 -= 2; 11374c1414c8SBarry Smith v5 -= 2; 11384c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 11394c1414c8SBarry Smith sum2 -= v2[2] * tmp0 + v2[1] * tmp1; 11404c1414c8SBarry Smith sum3 -= v3[2] * tmp0 + v3[1] * tmp1; 11414c1414c8SBarry Smith sum4 -= v4[2] * tmp0 + v4[1] * tmp1; 11424c1414c8SBarry Smith sum5 -= v5[2] * tmp0 + v5[1] * tmp1; 11434c1414c8SBarry Smith } 11444c1414c8SBarry Smith if (j==1){ 11454c1414c8SBarry Smith tmp0 = tmps[*vi--]; 11464c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 11474c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 11484c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11494c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11504c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11514c1414c8SBarry Smith } 11524c1414c8SBarry Smith 11534c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 11544c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 11554c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11564c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11574c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11584c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--; 11594c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11604c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11614c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11624c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--; 11634c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11644c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11654c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--; 11664c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11674c1414c8SBarry Smith x[*c--] = tmp[row] = sum5*a_a[ad[row]]; row--; 11684c1414c8SBarry Smith break; 11694c1414c8SBarry Smith default: 11704c1414c8SBarry Smith SETERRQ(PETSC_ERR_COR,"Node size not yet supported \n"); 11714c1414c8SBarry Smith } 11724c1414c8SBarry Smith } 11734c1414c8SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 11744c1414c8SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1175d9fead3dSBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 11764c1414c8SBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1177dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 11784c1414c8SBarry Smith PetscFunctionReturn(0); 11794c1414c8SBarry Smith } 11804c1414c8SBarry Smith 1181019b515eSShri Abhyankar /* ----------------------------------------------------------- */ 1182019b515eSShri Abhyankar #undef __FUNCT__ 1183019b515eSShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_Inode_newdatastruct" 1184019b515eSShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_Inode_newdatastruct(Mat A,Vec bb,Vec xx) 1185019b515eSShri Abhyankar { 1186019b515eSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1187019b515eSShri Abhyankar IS iscol = a->col,isrow = a->row; 1188019b515eSShri Abhyankar PetscErrorCode ierr; 1189019b515eSShri Abhyankar const PetscInt *r,*c,*rout,*cout; 1190019b515eSShri Abhyankar PetscInt i,j,n = A->rmap->n,*ai = a->i,nz,*a_j = a->j; 1191019b515eSShri Abhyankar PetscInt node_max,*ns,row,nsz,aii,*vi,*ad,*aj,i0,i1; 1192019b515eSShri Abhyankar PetscScalar *x,*tmp,*tmps,tmp0,tmp1; 1193019b515eSShri Abhyankar PetscScalar sum1,sum2,sum3,sum4,sum5; 1194019b515eSShri Abhyankar const MatScalar *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa; 1195019b515eSShri Abhyankar const PetscScalar *b; 1196019b515eSShri Abhyankar 1197019b515eSShri Abhyankar PetscFunctionBegin; 1198019b515eSShri Abhyankar if (!a->inode.size) SETERRQ(PETSC_ERR_COR,"Missing Inode Structure"); 1199019b515eSShri Abhyankar node_max = a->inode.node_count; 1200019b515eSShri Abhyankar ns = a->inode.size; /* Node Size array */ 1201019b515eSShri Abhyankar 1202019b515eSShri Abhyankar ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 1203019b515eSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1204019b515eSShri Abhyankar tmp = a->solve_work; 1205019b515eSShri Abhyankar 1206019b515eSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 1207019b515eSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 1208019b515eSShri Abhyankar 1209019b515eSShri Abhyankar /* forward solve the lower triangular */ 1210019b515eSShri Abhyankar tmps = tmp ; 1211019b515eSShri Abhyankar aa = a_a ; 1212019b515eSShri Abhyankar aj = a_j ; 1213019b515eSShri Abhyankar ad = a->diag; 1214019b515eSShri Abhyankar 1215019b515eSShri Abhyankar for (i = 0,row = 0; i< node_max; ++i){ 1216019b515eSShri Abhyankar nsz = ns[i]; 1217019b515eSShri Abhyankar aii = ai[row]; 1218019b515eSShri Abhyankar v1 = aa + aii; 1219019b515eSShri Abhyankar vi = aj + aii; 1220019b515eSShri Abhyankar nz = ai[row+1]- ai[row]; 1221019b515eSShri Abhyankar 1222019b515eSShri Abhyankar switch (nsz){ /* Each loop in 'case' is unrolled */ 1223019b515eSShri Abhyankar case 1 : 1224019b515eSShri Abhyankar sum1 = b[r[row]]; 1225019b515eSShri Abhyankar for(j=0; j<nz-1; j+=2){ 1226019b515eSShri Abhyankar i0 = vi[j]; 1227019b515eSShri Abhyankar i1 = vi[j+1]; 1228019b515eSShri Abhyankar tmp0 = tmps[i0]; 1229019b515eSShri Abhyankar tmp1 = tmps[i1]; 1230019b515eSShri Abhyankar sum1 -= v1[j]*tmp0 + v1[j+1]*tmp1; 1231019b515eSShri Abhyankar } 1232019b515eSShri Abhyankar if(j == nz-1){ 1233019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 1234019b515eSShri Abhyankar sum1 -= v1[j]*tmp0; 1235019b515eSShri Abhyankar } 1236019b515eSShri Abhyankar tmp[row++]=sum1; 1237019b515eSShri Abhyankar break; 1238019b515eSShri Abhyankar case 2: 1239019b515eSShri Abhyankar sum1 = b[r[row]]; 1240019b515eSShri Abhyankar sum2 = b[r[row+1]]; 1241019b515eSShri Abhyankar v2 = aa + ai[row+1]; 1242019b515eSShri Abhyankar 1243019b515eSShri Abhyankar for(j=0; j<nz-1; j+=2){ 1244019b515eSShri Abhyankar i0 = vi[j]; 1245019b515eSShri Abhyankar i1 = vi[j+1]; 1246019b515eSShri Abhyankar tmp0 = tmps[i0]; 1247019b515eSShri Abhyankar tmp1 = tmps[i1]; 1248019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 1249019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 1250019b515eSShri Abhyankar } 1251019b515eSShri Abhyankar if(j == nz-1){ 1252019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 1253019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 1254019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 1255019b515eSShri Abhyankar } 1256019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 1257019b515eSShri Abhyankar tmp[row ++]=sum1; 1258019b515eSShri Abhyankar tmp[row ++]=sum2; 1259019b515eSShri Abhyankar break; 1260019b515eSShri Abhyankar case 3: 1261019b515eSShri Abhyankar sum1 = b[r[row]]; 1262019b515eSShri Abhyankar sum2 = b[r[row+1]]; 1263019b515eSShri Abhyankar sum3 = b[r[row+2]]; 1264019b515eSShri Abhyankar v2 = aa + ai[row+1]; 1265019b515eSShri Abhyankar v3 = aa + ai[row+2]; 1266019b515eSShri Abhyankar 1267019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2){ 1268019b515eSShri Abhyankar i0 = vi[j]; 1269019b515eSShri Abhyankar i1 = vi[j+1]; 1270019b515eSShri Abhyankar tmp0 = tmps[i0]; 1271019b515eSShri Abhyankar tmp1 = tmps[i1]; 1272019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 1273019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 1274019b515eSShri Abhyankar sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1; 1275019b515eSShri Abhyankar } 1276019b515eSShri Abhyankar if (j == nz-1){ 1277019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 1278019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 1279019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 1280019b515eSShri Abhyankar sum3 -= v3[j] *tmp0; 1281019b515eSShri Abhyankar } 1282019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 1283019b515eSShri Abhyankar sum3 -= v3[nz] * sum1; 1284019b515eSShri Abhyankar sum3 -= v3[nz+1] * sum2; 1285019b515eSShri Abhyankar tmp[row ++]=sum1; 1286019b515eSShri Abhyankar tmp[row ++]=sum2; 1287019b515eSShri Abhyankar tmp[row ++]=sum3; 1288019b515eSShri Abhyankar break; 1289019b515eSShri Abhyankar 1290019b515eSShri Abhyankar case 4: 1291019b515eSShri Abhyankar sum1 = b[r[row]]; 1292019b515eSShri Abhyankar sum2 = b[r[row+1]]; 1293019b515eSShri Abhyankar sum3 = b[r[row+2]]; 1294019b515eSShri Abhyankar sum4 = b[r[row+3]]; 1295019b515eSShri Abhyankar v2 = aa + ai[row+1]; 1296019b515eSShri Abhyankar v3 = aa + ai[row+2]; 1297019b515eSShri Abhyankar v4 = aa + ai[row+3]; 1298019b515eSShri Abhyankar 1299019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2){ 1300019b515eSShri Abhyankar i0 = vi[j]; 1301019b515eSShri Abhyankar i1 = vi[j+1]; 1302019b515eSShri Abhyankar tmp0 = tmps[i0]; 1303019b515eSShri Abhyankar tmp1 = tmps[i1]; 1304019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 1305019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 1306019b515eSShri Abhyankar sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1; 1307019b515eSShri Abhyankar sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1; 1308019b515eSShri Abhyankar } 1309019b515eSShri Abhyankar if (j == nz-1){ 1310019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 1311019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 1312019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 1313019b515eSShri Abhyankar sum3 -= v3[j] *tmp0; 1314019b515eSShri Abhyankar sum4 -= v4[j] *tmp0; 1315019b515eSShri Abhyankar } 1316019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 1317019b515eSShri Abhyankar sum3 -= v3[nz] * sum1; 1318019b515eSShri Abhyankar sum4 -= v4[nz] * sum1; 1319019b515eSShri Abhyankar sum3 -= v3[nz+1] * sum2; 1320019b515eSShri Abhyankar sum4 -= v4[nz+1] * sum2; 1321019b515eSShri Abhyankar sum4 -= v4[nz+2] * sum3; 1322019b515eSShri Abhyankar 1323019b515eSShri Abhyankar tmp[row ++]=sum1; 1324019b515eSShri Abhyankar tmp[row ++]=sum2; 1325019b515eSShri Abhyankar tmp[row ++]=sum3; 1326019b515eSShri Abhyankar tmp[row ++]=sum4; 1327019b515eSShri Abhyankar break; 1328019b515eSShri Abhyankar case 5: 1329019b515eSShri Abhyankar sum1 = b[r[row]]; 1330019b515eSShri Abhyankar sum2 = b[r[row+1]]; 1331019b515eSShri Abhyankar sum3 = b[r[row+2]]; 1332019b515eSShri Abhyankar sum4 = b[r[row+3]]; 1333019b515eSShri Abhyankar sum5 = b[r[row+4]]; 1334019b515eSShri Abhyankar v2 = aa + ai[row+1]; 1335019b515eSShri Abhyankar v3 = aa + ai[row+2]; 1336019b515eSShri Abhyankar v4 = aa + ai[row+3]; 1337019b515eSShri Abhyankar v5 = aa + ai[row+4]; 1338019b515eSShri Abhyankar 1339019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2){ 1340019b515eSShri Abhyankar i0 = vi[j]; 1341019b515eSShri Abhyankar i1 = vi[j+1]; 1342019b515eSShri Abhyankar tmp0 = tmps[i0]; 1343019b515eSShri Abhyankar tmp1 = tmps[i1]; 1344019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 1345019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 1346019b515eSShri Abhyankar sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1; 1347019b515eSShri Abhyankar sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1; 1348019b515eSShri Abhyankar sum5 -= v5[j] * tmp0 + v5[j+1] * tmp1; 1349019b515eSShri Abhyankar } 1350019b515eSShri Abhyankar if (j == nz-1){ 1351019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 1352019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 1353019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 1354019b515eSShri Abhyankar sum3 -= v3[j] *tmp0; 1355019b515eSShri Abhyankar sum4 -= v4[j] *tmp0; 1356019b515eSShri Abhyankar sum5 -= v5[j] *tmp0; 1357019b515eSShri Abhyankar } 1358019b515eSShri Abhyankar 1359019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 1360019b515eSShri Abhyankar sum3 -= v3[nz] * sum1; 1361019b515eSShri Abhyankar sum4 -= v4[nz] * sum1; 1362019b515eSShri Abhyankar sum5 -= v5[nz] * sum1; 1363019b515eSShri Abhyankar sum3 -= v3[nz+1] * sum2; 1364019b515eSShri Abhyankar sum4 -= v4[nz+1] * sum2; 1365019b515eSShri Abhyankar sum5 -= v5[nz+1] * sum2; 1366019b515eSShri Abhyankar sum4 -= v4[nz+2] * sum3; 1367019b515eSShri Abhyankar sum5 -= v5[nz+2] * sum3; 1368019b515eSShri Abhyankar sum5 -= v5[nz+3] * sum4; 1369019b515eSShri Abhyankar 1370019b515eSShri Abhyankar tmp[row ++]=sum1; 1371019b515eSShri Abhyankar tmp[row ++]=sum2; 1372019b515eSShri Abhyankar tmp[row ++]=sum3; 1373019b515eSShri Abhyankar tmp[row ++]=sum4; 1374019b515eSShri Abhyankar tmp[row ++]=sum5; 1375019b515eSShri Abhyankar break; 1376019b515eSShri Abhyankar default: 1377019b515eSShri Abhyankar SETERRQ(PETSC_ERR_COR,"Node size not yet supported \n"); 1378019b515eSShri Abhyankar } 1379019b515eSShri Abhyankar } 1380019b515eSShri Abhyankar /* backward solve the upper triangular */ 1381019b515eSShri Abhyankar for (i=node_max -1 ,row = n-1 ; i>=0; i--){ 1382019b515eSShri Abhyankar nsz = ns[i]; 1383019b515eSShri Abhyankar aii = ad[row+1] + 1; 1384019b515eSShri Abhyankar v1 = aa + aii; 1385019b515eSShri Abhyankar vi = aj + aii; 1386019b515eSShri Abhyankar nz = ad[row]- ad[row+1] - 1; 1387019b515eSShri Abhyankar switch (nsz){ /* Each loop in 'case' is unrolled */ 1388019b515eSShri Abhyankar case 1 : 1389019b515eSShri Abhyankar sum1 = tmp[row]; 1390019b515eSShri Abhyankar 1391019b515eSShri Abhyankar for(j=0 ; j<nz-1; j+=2){ 1392019b515eSShri Abhyankar i0 = vi[j]; 1393019b515eSShri Abhyankar i1 = vi[j+1]; 1394019b515eSShri Abhyankar tmp0 = tmps[i0]; 1395019b515eSShri Abhyankar tmp1 = tmps[i1]; 1396019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 1397019b515eSShri Abhyankar } 1398019b515eSShri Abhyankar if (j == nz-1){ 1399019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 1400019b515eSShri Abhyankar sum1 -= v1[j]*tmp0; 1401019b515eSShri Abhyankar } 1402019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 1403019b515eSShri Abhyankar break; 1404019b515eSShri Abhyankar case 2 : 1405019b515eSShri Abhyankar sum1 = tmp[row]; 1406019b515eSShri Abhyankar sum2 = tmp[row-1]; 1407019b515eSShri Abhyankar v2 = aa + ad[row] + 1; 1408019b515eSShri Abhyankar for (j=0 ; j<nz-1; j+=2){ 1409019b515eSShri Abhyankar i0 = vi[j]; 1410019b515eSShri Abhyankar i1 = vi[j+1]; 1411019b515eSShri Abhyankar tmp0 = tmps[i0]; 1412019b515eSShri Abhyankar tmp1 = tmps[i1]; 1413019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 1414019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 1415019b515eSShri Abhyankar } 1416019b515eSShri Abhyankar if (j == nz-1){ 1417019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 1418019b515eSShri Abhyankar sum1 -= v1[j]* tmp0; 1419019b515eSShri Abhyankar sum2 -= v2[j+1]* tmp0; 1420019b515eSShri Abhyankar } 1421019b515eSShri Abhyankar 1422019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 1423019b515eSShri Abhyankar sum2 -= v2[0] * tmp0; 1424019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 1425019b515eSShri Abhyankar break; 1426019b515eSShri Abhyankar case 3 : 1427019b515eSShri Abhyankar sum1 = tmp[row]; 1428019b515eSShri Abhyankar sum2 = tmp[row -1]; 1429019b515eSShri Abhyankar sum3 = tmp[row -2]; 1430019b515eSShri Abhyankar v2 = aa + ad[row] + 1; 1431019b515eSShri Abhyankar v3 = aa + ad[row -1] + 1; 1432019b515eSShri Abhyankar for (j=0 ; j<nz-1; j+=2){ 1433019b515eSShri Abhyankar i0 = vi[j]; 1434019b515eSShri Abhyankar i1 = vi[j+1]; 1435019b515eSShri Abhyankar tmp0 = tmps[i0]; 1436019b515eSShri Abhyankar tmp1 = tmps[i1]; 1437019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 1438019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 1439019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1; 1440019b515eSShri Abhyankar } 1441019b515eSShri Abhyankar if (j== nz-1){ 1442019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 1443019b515eSShri Abhyankar sum1 -= v1[j] * tmp0; 1444019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0; 1445019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0; 1446019b515eSShri Abhyankar } 1447019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 1448019b515eSShri Abhyankar sum2 -= v2[0]* tmp0; 1449019b515eSShri Abhyankar sum3 -= v3[1] * tmp0; 1450019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 1451019b515eSShri Abhyankar sum3 -= v3[0]* tmp0; 1452019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--; 1453019b515eSShri Abhyankar 1454019b515eSShri Abhyankar break; 1455019b515eSShri Abhyankar case 4 : 1456019b515eSShri Abhyankar sum1 = tmp[row]; 1457019b515eSShri Abhyankar sum2 = tmp[row -1]; 1458019b515eSShri Abhyankar sum3 = tmp[row -2]; 1459019b515eSShri Abhyankar sum4 = tmp[row -3]; 1460019b515eSShri Abhyankar v2 = aa + ad[row]+1; 1461019b515eSShri Abhyankar v3 = aa + ad[row -1]+1; 1462019b515eSShri Abhyankar v4 = aa + ad[row -2]+1; 1463019b515eSShri Abhyankar 1464019b515eSShri Abhyankar for (j=0 ; j<nz-1; j+=2){ 1465019b515eSShri Abhyankar i0 = vi[j]; 1466019b515eSShri Abhyankar i1 = vi[j+1]; 1467019b515eSShri Abhyankar tmp0 = tmps[i0]; 1468019b515eSShri Abhyankar tmp1 = tmps[i1]; 1469019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 1470019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 1471019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1; 1472019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1; 1473019b515eSShri Abhyankar } 1474019b515eSShri Abhyankar if (j== nz-1){ 1475019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 1476019b515eSShri Abhyankar sum1 -= v1[j] * tmp0; 1477019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0; 1478019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0; 1479019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0; 1480019b515eSShri Abhyankar } 1481019b515eSShri Abhyankar 1482019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 1483019b515eSShri Abhyankar sum2 -= v2[0] * tmp0; 1484019b515eSShri Abhyankar sum3 -= v3[1] * tmp0; 1485019b515eSShri Abhyankar sum4 -= v4[2] * tmp0; 1486019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 1487019b515eSShri Abhyankar sum3 -= v3[0] * tmp0; 1488019b515eSShri Abhyankar sum4 -= v4[1] * tmp0; 1489019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--; 1490019b515eSShri Abhyankar sum4 -= v4[0] * tmp0; 1491019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--; 1492019b515eSShri Abhyankar break; 1493019b515eSShri Abhyankar case 5 : 1494019b515eSShri Abhyankar sum1 = tmp[row]; 1495019b515eSShri Abhyankar sum2 = tmp[row -1]; 1496019b515eSShri Abhyankar sum3 = tmp[row -2]; 1497019b515eSShri Abhyankar sum4 = tmp[row -3]; 1498019b515eSShri Abhyankar sum5 = tmp[row -4]; 1499019b515eSShri Abhyankar v2 = aa + ad[row]+1; 1500019b515eSShri Abhyankar v3 = aa + ad[row -1]+1; 1501019b515eSShri Abhyankar v4 = aa + ad[row -2]+1; 1502019b515eSShri Abhyankar v5 = aa + ad[row -3]+1; 1503019b515eSShri Abhyankar for (j=0 ; j<nz-1; j+=2){ 1504019b515eSShri Abhyankar i0 = vi[j]; 1505019b515eSShri Abhyankar i1 = vi[j+1]; 1506019b515eSShri Abhyankar tmp0 = tmps[i0]; 1507019b515eSShri Abhyankar tmp1 = tmps[i1]; 1508019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 1509019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 1510019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1; 1511019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1; 1512019b515eSShri Abhyankar sum5 -= v5[j+4] * tmp0 + v5[j+5] * tmp1; 1513019b515eSShri Abhyankar } 1514019b515eSShri Abhyankar if (j==nz-1){ 1515019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 1516019b515eSShri Abhyankar sum1 -= v1[j] * tmp0; 1517019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0; 1518019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0; 1519019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0; 1520019b515eSShri Abhyankar sum5 -= v5[j+4] * tmp0; 1521019b515eSShri Abhyankar } 1522019b515eSShri Abhyankar 1523019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 1524019b515eSShri Abhyankar sum2 -= v2[0] * tmp0; 1525019b515eSShri Abhyankar sum3 -= v3[1] * tmp0; 1526019b515eSShri Abhyankar sum4 -= v4[2] * tmp0; 1527019b515eSShri Abhyankar sum5 -= v5[3] * tmp0; 1528019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 1529019b515eSShri Abhyankar sum3 -= v3[0] * tmp0; 1530019b515eSShri Abhyankar sum4 -= v4[1] * tmp0; 1531019b515eSShri Abhyankar sum5 -= v5[2] * tmp0; 1532019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--; 1533019b515eSShri Abhyankar sum4 -= v4[0] * tmp0; 1534019b515eSShri Abhyankar sum5 -= v5[1] * tmp0; 1535019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--; 1536019b515eSShri Abhyankar sum5 -= v5[0] * tmp0; 1537019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum5*v5[nz+4]; row--; 1538019b515eSShri Abhyankar break; 1539019b515eSShri Abhyankar default: 1540019b515eSShri Abhyankar SETERRQ(PETSC_ERR_COR,"Node size not yet supported \n"); 1541019b515eSShri Abhyankar } 1542019b515eSShri Abhyankar } 1543019b515eSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 1544019b515eSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 1545019b515eSShri Abhyankar ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 1546019b515eSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1547019b515eSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 1548019b515eSShri Abhyankar PetscFunctionReturn(0); 1549019b515eSShri Abhyankar } 1550019b515eSShri Abhyankar 1551019b515eSShri Abhyankar 15524c1414c8SBarry Smith /* 15534c1414c8SBarry Smith Makes a longer coloring[] array and calls the usual code with that 15544c1414c8SBarry Smith */ 15554c1414c8SBarry Smith #undef __FUNCT__ 15564108e4d5SBarry Smith #define __FUNCT__ "MatColoringPatch_SeqAIJ_Inode" 15574108e4d5SBarry Smith PetscErrorCode MatColoringPatch_SeqAIJ_Inode(Mat mat,PetscInt ncolors,PetscInt nin,ISColoringValue coloring[],ISColoring *iscoloring) 15584c1414c8SBarry Smith { 15594c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)mat->data; 15604c1414c8SBarry Smith PetscErrorCode ierr; 1561d0f46423SBarry Smith PetscInt n = mat->cmap->n,m = a->inode.node_count,j,*ns = a->inode.size,row; 15624c1414c8SBarry Smith PetscInt *colorused,i; 15634c1414c8SBarry Smith ISColoringValue *newcolor; 15644c1414c8SBarry Smith 15654c1414c8SBarry Smith PetscFunctionBegin; 15664c1414c8SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscInt),&newcolor);CHKERRQ(ierr); 15674c1414c8SBarry Smith /* loop over inodes, marking a color for each column*/ 15684c1414c8SBarry Smith row = 0; 15694c1414c8SBarry Smith for (i=0; i<m; i++){ 15704c1414c8SBarry Smith for (j=0; j<ns[i]; j++) { 15714c1414c8SBarry Smith newcolor[row++] = coloring[i] + j*ncolors; 15724c1414c8SBarry Smith } 15734c1414c8SBarry Smith } 15744c1414c8SBarry Smith 15754c1414c8SBarry Smith /* eliminate unneeded colors */ 15764c1414c8SBarry Smith ierr = PetscMalloc(5*ncolors*sizeof(PetscInt),&colorused);CHKERRQ(ierr); 15774c1414c8SBarry Smith ierr = PetscMemzero(colorused,5*ncolors*sizeof(PetscInt));CHKERRQ(ierr); 15784c1414c8SBarry Smith for (i=0; i<n; i++) { 15794c1414c8SBarry Smith colorused[newcolor[i]] = 1; 15804c1414c8SBarry Smith } 15814c1414c8SBarry Smith 15824c1414c8SBarry Smith for (i=1; i<5*ncolors; i++) { 15834c1414c8SBarry Smith colorused[i] += colorused[i-1]; 15844c1414c8SBarry Smith } 15854c1414c8SBarry Smith ncolors = colorused[5*ncolors-1]; 15864c1414c8SBarry Smith for (i=0; i<n; i++) { 158757815681SSatish Balay newcolor[i] = colorused[newcolor[i]]-1; 15884c1414c8SBarry Smith } 15894c1414c8SBarry Smith ierr = PetscFree(colorused);CHKERRQ(ierr); 15907adad957SLisandro Dalcin ierr = ISColoringCreate(((PetscObject)mat)->comm,ncolors,n,newcolor,iscoloring);CHKERRQ(ierr); 15914c1414c8SBarry Smith ierr = PetscFree(coloring);CHKERRQ(ierr); 15924c1414c8SBarry Smith PetscFunctionReturn(0); 15934c1414c8SBarry Smith } 15944c1414c8SBarry Smith 1595c60f0209SBarry Smith #include "../src/mat/blockinvert.h" 15962af78befSBarry Smith 15972af78befSBarry Smith #undef __FUNCT__ 15984108e4d5SBarry Smith #define __FUNCT__ "MatSOR_SeqAIJ_Inode" 15994108e4d5SBarry Smith PetscErrorCode MatSOR_SeqAIJ_Inode(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx) 16002af78befSBarry Smith { 16012af78befSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1602938d4eb3SBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5,tmp0,tmp1,tmp2,tmp3; 1603a77337e4SBarry Smith MatScalar *ibdiag,*bdiag; 1604938d4eb3SBarry Smith PetscScalar *x,*xb,tmp4,tmp5,x1,x2,x3,x4,x5; 1605dd6ea824SBarry Smith const MatScalar *v = a->a,*v1,*v2,*v3,*v4,*v5; 160662bba022SBarry Smith PetscReal zeropivot = 1.0e-15, shift = 0.0; 16072af78befSBarry Smith PetscErrorCode ierr; 1608f0d39aaaSBarry Smith PetscInt n,m = a->inode.node_count,*sizes = a->inode.size,cnt = 0,i,j,row,i1,i2; 1609f0d39aaaSBarry Smith PetscInt *idx,*diag = a->diag,*ii = a->i,sz,k; 16102af78befSBarry Smith 16112af78befSBarry Smith PetscFunctionBegin; 161271f1c65dSBarry Smith if (omega != 1.0) SETERRQ(PETSC_ERR_SUP,"No support for omega != 1.0; use -mat_no_inode"); 161371f1c65dSBarry Smith if (fshift != 0.0) SETERRQ(PETSC_ERR_SUP,"No support for fshift != 0.0; use -mat_no_inode"); 16143f6d85c9SBarry Smith if (its > 1) { 16153f6d85c9SBarry Smith /* switch to non-inode version */ 161641f059aeSBarry Smith ierr = MatSOR_SeqAIJ(A,bb,omega,flag,fshift,its,lits,xx);CHKERRQ(ierr); 16173f6d85c9SBarry Smith PetscFunctionReturn(0); 16183f6d85c9SBarry Smith } 16192af78befSBarry Smith 162071f1c65dSBarry Smith if (!a->inode.ibdiagvalid) { 16212af78befSBarry Smith if (!a->inode.ibdiag) { 16222af78befSBarry Smith /* calculate space needed for diagonal blocks */ 16232af78befSBarry Smith for (i=0; i<m; i++) { 16242af78befSBarry Smith cnt += sizes[i]*sizes[i]; 16252af78befSBarry Smith } 1626f0d39aaaSBarry Smith a->inode.bdiagsize = cnt; 162789c6957cSBarry Smith ierr = PetscMalloc3(cnt,MatScalar,&a->inode.ibdiag,cnt,MatScalar,&a->inode.bdiag,A->rmap->n,MatScalar,&a->inode.ssor_work);CHKERRQ(ierr); 162871f1c65dSBarry Smith } 162971f1c65dSBarry Smith 163071f1c65dSBarry Smith /* copy over the diagonal blocks and invert them */ 16312af78befSBarry Smith ibdiag = a->inode.ibdiag; 16322af78befSBarry Smith bdiag = a->inode.bdiag; 16332af78befSBarry Smith cnt = 0; 16342af78befSBarry Smith for (i=0, row = 0; i<m; i++) { 16352af78befSBarry Smith for (j=0; j<sizes[i]; j++) { 1636f0d39aaaSBarry Smith for (k=0; k<sizes[i]; k++) { 1637f0d39aaaSBarry Smith bdiag[cnt+k*sizes[i]+j] = v[diag[row+j] - j + k]; 1638f0d39aaaSBarry Smith } 16392af78befSBarry Smith } 1640a77337e4SBarry Smith ierr = PetscMemcpy(ibdiag+cnt,bdiag+cnt,sizes[i]*sizes[i]*sizeof(MatScalar));CHKERRQ(ierr); 16412af78befSBarry Smith 16422af78befSBarry Smith switch(sizes[i]) { 16432af78befSBarry Smith case 1: 16442af78befSBarry Smith /* Create matrix data structure */ 164564c62002SMatthew Knepley if (PetscAbsScalar(ibdiag[cnt]) < zeropivot) SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot on row %D",row); 164664c62002SMatthew Knepley ibdiag[cnt] = 1.0/ibdiag[cnt]; 16472af78befSBarry Smith break; 16482af78befSBarry Smith case 2: 164962bba022SBarry Smith ierr = Kernel_A_gets_inverse_A_2(ibdiag+cnt,shift);CHKERRQ(ierr); 16502af78befSBarry Smith break; 16512af78befSBarry Smith case 3: 165262bba022SBarry Smith ierr = Kernel_A_gets_inverse_A_3(ibdiag+cnt,shift);CHKERRQ(ierr); 16532af78befSBarry Smith break; 16542af78befSBarry Smith case 4: 165562bba022SBarry Smith ierr = Kernel_A_gets_inverse_A_4(ibdiag+cnt,shift);CHKERRQ(ierr); 16562af78befSBarry Smith break; 16572af78befSBarry Smith case 5: 165862bba022SBarry Smith ierr = Kernel_A_gets_inverse_A_5(ibdiag+cnt,shift);CHKERRQ(ierr); 16592af78befSBarry Smith break; 16602af78befSBarry Smith default: 16612af78befSBarry Smith SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 16622af78befSBarry Smith } 16632af78befSBarry Smith cnt += sizes[i]*sizes[i]; 16642af78befSBarry Smith row += sizes[i]; 16652af78befSBarry Smith } 166671f1c65dSBarry Smith a->inode.ibdiagvalid = PETSC_TRUE; 16672af78befSBarry Smith } 16682af78befSBarry Smith ibdiag = a->inode.ibdiag; 16692af78befSBarry Smith bdiag = a->inode.bdiag; 16702af78befSBarry Smith 1671938d4eb3SBarry Smith 1672938d4eb3SBarry Smith /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */ 1673938d4eb3SBarry Smith if (flag & SOR_ZERO_INITIAL_GUESS) { 1674938d4eb3SBarry Smith PetscScalar *b; 16752af78befSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 16762af78befSBarry Smith if (xx != bb) { 16772af78befSBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 16782af78befSBarry Smith } else { 16792af78befSBarry Smith b = x; 16802af78befSBarry Smith } 16812af78befSBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){ 16822af78befSBarry Smith 16838862d2efSBarry Smith for (i=0, row=0; i<m; i++) { 16848862d2efSBarry Smith sz = diag[row] - ii[row]; 16858862d2efSBarry Smith v1 = a->a + ii[row]; 16868862d2efSBarry Smith idx = a->j + ii[row]; 16878862d2efSBarry Smith 16884108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 16898862d2efSBarry Smith switch (sizes[i]){ 16908862d2efSBarry Smith case 1: 16918862d2efSBarry Smith 16928862d2efSBarry Smith sum1 = b[row]; 16938862d2efSBarry Smith for(n = 0; n<sz-1; n+=2) { 16948862d2efSBarry Smith i1 = idx[0]; 16958862d2efSBarry Smith i2 = idx[1]; 16968862d2efSBarry Smith idx += 2; 16978862d2efSBarry Smith tmp0 = x[i1]; 16988862d2efSBarry Smith tmp1 = x[i2]; 16998862d2efSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 17008862d2efSBarry Smith } 17018862d2efSBarry Smith 17028862d2efSBarry Smith if (n == sz-1){ 1703f0d39aaaSBarry Smith tmp0 = x[*idx]; 1704f0d39aaaSBarry Smith sum1 -= *v1 * tmp0; 17058862d2efSBarry Smith } 17068862d2efSBarry Smith x[row++] = sum1*(*ibdiag++); 17078862d2efSBarry Smith break; 1708f0d39aaaSBarry Smith case 2: 1709f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 1710f0d39aaaSBarry Smith sum1 = b[row]; 1711f0d39aaaSBarry Smith sum2 = b[row+1]; 1712f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 1713f0d39aaaSBarry Smith i1 = idx[0]; 1714f0d39aaaSBarry Smith i2 = idx[1]; 1715f0d39aaaSBarry Smith idx += 2; 1716f0d39aaaSBarry Smith tmp0 = x[i1]; 1717f0d39aaaSBarry Smith tmp1 = x[i2]; 1718f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 1719f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 1720f0d39aaaSBarry Smith } 1721f0d39aaaSBarry Smith 1722f0d39aaaSBarry Smith if (n == sz-1){ 1723f0d39aaaSBarry Smith tmp0 = x[*idx]; 1724f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 1725f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 1726f0d39aaaSBarry Smith } 1727f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[2]; 1728f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[3]; 1729f0d39aaaSBarry Smith ibdiag += 4; 1730f0d39aaaSBarry Smith break; 1731f0d39aaaSBarry Smith case 3: 1732f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 1733f0d39aaaSBarry Smith v3 = a->a + ii[row+2]; 1734f0d39aaaSBarry Smith sum1 = b[row]; 1735f0d39aaaSBarry Smith sum2 = b[row+1]; 1736f0d39aaaSBarry Smith sum3 = b[row+2]; 1737f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 1738f0d39aaaSBarry Smith i1 = idx[0]; 1739f0d39aaaSBarry Smith i2 = idx[1]; 1740f0d39aaaSBarry Smith idx += 2; 1741f0d39aaaSBarry Smith tmp0 = x[i1]; 1742f0d39aaaSBarry Smith tmp1 = x[i2]; 1743f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 1744f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 1745f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 1746f0d39aaaSBarry Smith } 1747f0d39aaaSBarry Smith 1748f0d39aaaSBarry Smith if (n == sz-1){ 1749f0d39aaaSBarry Smith tmp0 = x[*idx]; 1750f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 1751f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 1752f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0; 1753f0d39aaaSBarry Smith } 1754f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6]; 1755f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7]; 1756f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8]; 1757f0d39aaaSBarry Smith ibdiag += 9; 1758f0d39aaaSBarry Smith break; 1759f0d39aaaSBarry Smith case 4: 1760f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 1761f0d39aaaSBarry Smith v3 = a->a + ii[row+2]; 1762f0d39aaaSBarry Smith v4 = a->a + ii[row+3]; 1763f0d39aaaSBarry Smith sum1 = b[row]; 1764f0d39aaaSBarry Smith sum2 = b[row+1]; 1765f0d39aaaSBarry Smith sum3 = b[row+2]; 1766f0d39aaaSBarry Smith sum4 = b[row+3]; 1767f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 1768f0d39aaaSBarry Smith i1 = idx[0]; 1769f0d39aaaSBarry Smith i2 = idx[1]; 1770f0d39aaaSBarry Smith idx += 2; 1771f0d39aaaSBarry Smith tmp0 = x[i1]; 1772f0d39aaaSBarry Smith tmp1 = x[i2]; 1773f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 1774f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 1775f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 1776f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 1777f0d39aaaSBarry Smith } 1778f0d39aaaSBarry Smith 1779f0d39aaaSBarry Smith if (n == sz-1){ 1780f0d39aaaSBarry Smith tmp0 = x[*idx]; 1781f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 1782f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 1783f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0; 1784f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0; 1785f0d39aaaSBarry Smith } 1786f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12]; 1787f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13]; 1788f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14]; 1789f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15]; 1790f0d39aaaSBarry Smith ibdiag += 16; 1791f0d39aaaSBarry Smith break; 1792f0d39aaaSBarry Smith case 5: 1793f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 1794f0d39aaaSBarry Smith v3 = a->a + ii[row+2]; 1795f0d39aaaSBarry Smith v4 = a->a + ii[row+3]; 1796f0d39aaaSBarry Smith v5 = a->a + ii[row+4]; 1797f0d39aaaSBarry Smith sum1 = b[row]; 1798f0d39aaaSBarry Smith sum2 = b[row+1]; 1799f0d39aaaSBarry Smith sum3 = b[row+2]; 1800f0d39aaaSBarry Smith sum4 = b[row+3]; 1801f0d39aaaSBarry Smith sum5 = b[row+4]; 1802f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 1803f0d39aaaSBarry Smith i1 = idx[0]; 1804f0d39aaaSBarry Smith i2 = idx[1]; 1805f0d39aaaSBarry Smith idx += 2; 1806f0d39aaaSBarry Smith tmp0 = x[i1]; 1807f0d39aaaSBarry Smith tmp1 = x[i2]; 1808f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 1809f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 1810f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 1811f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 1812f0d39aaaSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 1813f0d39aaaSBarry Smith } 1814f0d39aaaSBarry Smith 1815f0d39aaaSBarry Smith if (n == sz-1){ 1816f0d39aaaSBarry Smith tmp0 = x[*idx]; 1817f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 1818f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 1819f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0; 1820f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0; 1821f0d39aaaSBarry Smith sum5 -= v5[0] * tmp0; 1822f0d39aaaSBarry Smith } 1823f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20]; 1824f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21]; 1825f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22]; 1826f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23]; 1827f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24]; 1828f0d39aaaSBarry Smith ibdiag += 25; 1829f0d39aaaSBarry Smith break; 18308862d2efSBarry Smith default: 18318862d2efSBarry Smith SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 18328862d2efSBarry Smith } 18332af78befSBarry Smith } 18342af78befSBarry Smith 18352af78befSBarry Smith xb = x; 18362af78befSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 18372af78befSBarry Smith } else xb = b; 18382af78befSBarry Smith if ((flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) && 18392af78befSBarry Smith (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP)) { 18408862d2efSBarry Smith cnt = 0; 18418862d2efSBarry Smith for (i=0, row=0; i<m; i++) { 18422af78befSBarry Smith 18438862d2efSBarry Smith switch (sizes[i]){ 18448862d2efSBarry Smith case 1: 18458862d2efSBarry Smith x[row++] *= bdiag[cnt++]; 18468862d2efSBarry Smith break; 1847f0d39aaaSBarry Smith case 2: 1848f0d39aaaSBarry Smith x1 = x[row]; x2 = x[row+1]; 1849f0d39aaaSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2]; 1850f0d39aaaSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3]; 1851f0d39aaaSBarry Smith x[row++] = tmp1; 1852f0d39aaaSBarry Smith x[row++] = tmp2; 1853f0d39aaaSBarry Smith cnt += 4; 1854f0d39aaaSBarry Smith break; 1855f0d39aaaSBarry Smith case 3: 1856f0d39aaaSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; 1857f0d39aaaSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6]; 1858f0d39aaaSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7]; 1859f0d39aaaSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8]; 1860f0d39aaaSBarry Smith x[row++] = tmp1; 1861f0d39aaaSBarry Smith x[row++] = tmp2; 1862f0d39aaaSBarry Smith x[row++] = tmp3; 1863f0d39aaaSBarry Smith cnt += 9; 1864f0d39aaaSBarry Smith break; 1865f0d39aaaSBarry Smith case 4: 1866f0d39aaaSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; 1867f0d39aaaSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12]; 1868f0d39aaaSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13]; 1869f0d39aaaSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14]; 1870f0d39aaaSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15]; 1871f0d39aaaSBarry Smith x[row++] = tmp1; 1872f0d39aaaSBarry Smith x[row++] = tmp2; 1873f0d39aaaSBarry Smith x[row++] = tmp3; 1874f0d39aaaSBarry Smith x[row++] = tmp4; 1875f0d39aaaSBarry Smith cnt += 16; 1876f0d39aaaSBarry Smith break; 1877f0d39aaaSBarry Smith case 5: 1878f0d39aaaSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; x5 = x[row+4]; 1879f0d39aaaSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20]; 1880f0d39aaaSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21]; 1881f0d39aaaSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22]; 1882f0d39aaaSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23]; 1883f0d39aaaSBarry Smith tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24]; 1884f0d39aaaSBarry Smith x[row++] = tmp1; 1885f0d39aaaSBarry Smith x[row++] = tmp2; 1886f0d39aaaSBarry Smith x[row++] = tmp3; 1887f0d39aaaSBarry Smith x[row++] = tmp4; 1888f0d39aaaSBarry Smith x[row++] = tmp5; 1889f0d39aaaSBarry Smith cnt += 25; 1890f0d39aaaSBarry Smith break; 18918862d2efSBarry Smith default: 18928862d2efSBarry Smith SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 18938862d2efSBarry Smith } 18942af78befSBarry Smith } 18952af78befSBarry Smith ierr = PetscLogFlops(m);CHKERRQ(ierr); 18962af78befSBarry Smith } 18972af78befSBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){ 18982af78befSBarry Smith 1899f0d39aaaSBarry Smith ibdiag = a->inode.ibdiag+a->inode.bdiagsize; 1900d0f46423SBarry Smith for (i=m-1, row=A->rmap->n-1; i>=0; i--) { 1901f0d39aaaSBarry Smith ibdiag -= sizes[i]*sizes[i]; 19028862d2efSBarry Smith sz = ii[row+1] - diag[row] - 1; 19038862d2efSBarry Smith v1 = a->a + diag[row] + 1; 19048862d2efSBarry Smith idx = a->j + diag[row] + 1; 19052af78befSBarry Smith 19064108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 19078862d2efSBarry Smith switch (sizes[i]){ 19088862d2efSBarry Smith case 1: 19098862d2efSBarry Smith 19108862d2efSBarry Smith sum1 = xb[row]; 19118862d2efSBarry Smith for(n = 0; n<sz-1; n+=2) { 19128862d2efSBarry Smith i1 = idx[0]; 19138862d2efSBarry Smith i2 = idx[1]; 19148862d2efSBarry Smith idx += 2; 19158862d2efSBarry Smith tmp0 = x[i1]; 19168862d2efSBarry Smith tmp1 = x[i2]; 19178862d2efSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 19188862d2efSBarry Smith } 19198862d2efSBarry Smith 19208862d2efSBarry Smith if (n == sz-1){ 1921f0d39aaaSBarry Smith tmp0 = x[*idx]; 1922f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 19238862d2efSBarry Smith } 1924f0d39aaaSBarry Smith x[row--] = sum1*(*ibdiag); 1925f0d39aaaSBarry Smith break; 1926f0d39aaaSBarry Smith 1927f0d39aaaSBarry Smith case 2: 1928f0d39aaaSBarry Smith 1929f0d39aaaSBarry Smith sum1 = xb[row]; 1930f0d39aaaSBarry Smith sum2 = xb[row-1]; 1931f0d39aaaSBarry Smith /* note that sum1 is associated with the second of the two rows */ 1932f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 1933f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 1934f0d39aaaSBarry Smith i1 = idx[0]; 1935f0d39aaaSBarry Smith i2 = idx[1]; 1936f0d39aaaSBarry Smith idx += 2; 1937f0d39aaaSBarry Smith tmp0 = x[i1]; 1938f0d39aaaSBarry Smith tmp1 = x[i2]; 1939f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 1940f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 1941f0d39aaaSBarry Smith } 1942f0d39aaaSBarry Smith 1943f0d39aaaSBarry Smith if (n == sz-1){ 1944f0d39aaaSBarry Smith tmp0 = x[*idx]; 1945f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 1946f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 1947f0d39aaaSBarry Smith } 1948f0d39aaaSBarry Smith x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3]; 1949f0d39aaaSBarry Smith x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2]; 1950f0d39aaaSBarry Smith break; 1951f0d39aaaSBarry Smith case 3: 1952f0d39aaaSBarry Smith 1953f0d39aaaSBarry Smith sum1 = xb[row]; 1954f0d39aaaSBarry Smith sum2 = xb[row-1]; 1955f0d39aaaSBarry Smith sum3 = xb[row-2]; 1956f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 1957f0d39aaaSBarry Smith v3 = a->a + diag[row-2] + 3; 1958f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 1959f0d39aaaSBarry Smith i1 = idx[0]; 1960f0d39aaaSBarry Smith i2 = idx[1]; 1961f0d39aaaSBarry Smith idx += 2; 1962f0d39aaaSBarry Smith tmp0 = x[i1]; 1963f0d39aaaSBarry Smith tmp1 = x[i2]; 1964f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 1965f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 1966f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 1967f0d39aaaSBarry Smith } 1968f0d39aaaSBarry Smith 1969f0d39aaaSBarry Smith if (n == sz-1){ 1970f0d39aaaSBarry Smith tmp0 = x[*idx]; 1971f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 1972f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 1973f0d39aaaSBarry Smith sum3 -= *v3*tmp0; 1974f0d39aaaSBarry Smith } 1975f0d39aaaSBarry Smith x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8]; 1976f0d39aaaSBarry Smith x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7]; 1977f0d39aaaSBarry Smith x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6]; 1978f0d39aaaSBarry Smith break; 1979f0d39aaaSBarry Smith case 4: 1980f0d39aaaSBarry Smith 1981f0d39aaaSBarry Smith sum1 = xb[row]; 1982f0d39aaaSBarry Smith sum2 = xb[row-1]; 1983f0d39aaaSBarry Smith sum3 = xb[row-2]; 1984f0d39aaaSBarry Smith sum4 = xb[row-3]; 1985f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 1986f0d39aaaSBarry Smith v3 = a->a + diag[row-2] + 3; 1987f0d39aaaSBarry Smith v4 = a->a + diag[row-3] + 4; 1988f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 1989f0d39aaaSBarry Smith i1 = idx[0]; 1990f0d39aaaSBarry Smith i2 = idx[1]; 1991f0d39aaaSBarry Smith idx += 2; 1992f0d39aaaSBarry Smith tmp0 = x[i1]; 1993f0d39aaaSBarry Smith tmp1 = x[i2]; 1994f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 1995f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 1996f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 1997f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 1998f0d39aaaSBarry Smith } 1999f0d39aaaSBarry Smith 2000f0d39aaaSBarry Smith if (n == sz-1){ 2001f0d39aaaSBarry Smith tmp0 = x[*idx]; 2002f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 2003f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 2004f0d39aaaSBarry Smith sum3 -= *v3*tmp0; 2005f0d39aaaSBarry Smith sum4 -= *v4*tmp0; 2006f0d39aaaSBarry Smith } 2007f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15]; 2008f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14]; 2009f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13]; 2010f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12]; 2011f0d39aaaSBarry Smith break; 2012f0d39aaaSBarry Smith case 5: 2013f0d39aaaSBarry Smith 2014f0d39aaaSBarry Smith sum1 = xb[row]; 2015f0d39aaaSBarry Smith sum2 = xb[row-1]; 2016f0d39aaaSBarry Smith sum3 = xb[row-2]; 2017f0d39aaaSBarry Smith sum4 = xb[row-3]; 2018f0d39aaaSBarry Smith sum5 = xb[row-4]; 2019f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 2020f0d39aaaSBarry Smith v3 = a->a + diag[row-2] + 3; 2021f0d39aaaSBarry Smith v4 = a->a + diag[row-3] + 4; 2022f0d39aaaSBarry Smith v5 = a->a + diag[row-4] + 5; 2023f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 2024f0d39aaaSBarry Smith i1 = idx[0]; 2025f0d39aaaSBarry Smith i2 = idx[1]; 2026f0d39aaaSBarry Smith idx += 2; 2027f0d39aaaSBarry Smith tmp0 = x[i1]; 2028f0d39aaaSBarry Smith tmp1 = x[i2]; 2029f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 2030f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 2031f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 2032f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 2033f0d39aaaSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 2034f0d39aaaSBarry Smith } 2035f0d39aaaSBarry Smith 2036f0d39aaaSBarry Smith if (n == sz-1){ 2037f0d39aaaSBarry Smith tmp0 = x[*idx]; 2038f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 2039f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 2040f0d39aaaSBarry Smith sum3 -= *v3*tmp0; 2041f0d39aaaSBarry Smith sum4 -= *v4*tmp0; 2042f0d39aaaSBarry Smith sum5 -= *v5*tmp0; 2043f0d39aaaSBarry Smith } 2044f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24]; 2045f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23]; 2046f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22]; 2047f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21]; 2048f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20]; 20498862d2efSBarry Smith break; 20508862d2efSBarry Smith default: 20518862d2efSBarry Smith SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 20528862d2efSBarry Smith } 20532af78befSBarry Smith } 20542af78befSBarry Smith 20552af78befSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 20562af78befSBarry Smith } 20572af78befSBarry Smith its--; 2058938d4eb3SBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2059938d4eb3SBarry Smith if (bb != xx) {ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);} 20602af78befSBarry Smith } 206189c6957cSBarry Smith if (flag & SOR_EISENSTAT) { 2062938d4eb3SBarry Smith const PetscScalar *b; 20636cf45366SBarry Smith MatScalar *t = a->inode.ssor_work; 20646cf45366SBarry Smith 2065938d4eb3SBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2066938d4eb3SBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 206789c6957cSBarry Smith /* 206889c6957cSBarry Smith Apply (U + D)^-1 where D is now the block diagonal 206989c6957cSBarry Smith */ 207089c6957cSBarry Smith ibdiag = a->inode.ibdiag+a->inode.bdiagsize; 207189c6957cSBarry Smith for (i=m-1, row=A->rmap->n-1; i>=0; i--) { 207289c6957cSBarry Smith ibdiag -= sizes[i]*sizes[i]; 207389c6957cSBarry Smith sz = ii[row+1] - diag[row] - 1; 207489c6957cSBarry Smith v1 = a->a + diag[row] + 1; 207589c6957cSBarry Smith idx = a->j + diag[row] + 1; 207689c6957cSBarry Smith CHKMEMQ; 20774108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 207889c6957cSBarry Smith switch (sizes[i]){ 207989c6957cSBarry Smith case 1: 208089c6957cSBarry Smith 208189c6957cSBarry Smith sum1 = b[row]; 208289c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 208389c6957cSBarry Smith i1 = idx[0]; 208489c6957cSBarry Smith i2 = idx[1]; 208589c6957cSBarry Smith idx += 2; 208689c6957cSBarry Smith tmp0 = x[i1]; 208789c6957cSBarry Smith tmp1 = x[i2]; 208889c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 208989c6957cSBarry Smith } 209089c6957cSBarry Smith 209189c6957cSBarry Smith if (n == sz-1){ 209289c6957cSBarry Smith tmp0 = x[*idx]; 209389c6957cSBarry Smith sum1 -= *v1*tmp0; 209489c6957cSBarry Smith } 2095938d4eb3SBarry Smith x[row] = sum1*(*ibdiag);row--; 209689c6957cSBarry Smith break; 209789c6957cSBarry Smith 209889c6957cSBarry Smith case 2: 209989c6957cSBarry Smith 210089c6957cSBarry Smith sum1 = b[row]; 210189c6957cSBarry Smith sum2 = b[row-1]; 210289c6957cSBarry Smith /* note that sum1 is associated with the second of the two rows */ 210389c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 210489c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 210589c6957cSBarry Smith i1 = idx[0]; 210689c6957cSBarry Smith i2 = idx[1]; 210789c6957cSBarry Smith idx += 2; 210889c6957cSBarry Smith tmp0 = x[i1]; 210989c6957cSBarry Smith tmp1 = x[i2]; 211089c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 211189c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 211289c6957cSBarry Smith } 211389c6957cSBarry Smith 211489c6957cSBarry Smith if (n == sz-1){ 211589c6957cSBarry Smith tmp0 = x[*idx]; 211689c6957cSBarry Smith sum1 -= *v1*tmp0; 211789c6957cSBarry Smith sum2 -= *v2*tmp0; 211889c6957cSBarry Smith } 2119938d4eb3SBarry Smith x[row] = sum2*ibdiag[1] + sum1*ibdiag[3]; 2120938d4eb3SBarry Smith x[row-1] = sum2*ibdiag[0] + sum1*ibdiag[2]; 2121938d4eb3SBarry Smith row -= 2; 212289c6957cSBarry Smith break; 212389c6957cSBarry Smith case 3: 212489c6957cSBarry Smith 212589c6957cSBarry Smith sum1 = b[row]; 212689c6957cSBarry Smith sum2 = b[row-1]; 212789c6957cSBarry Smith sum3 = b[row-2]; 212889c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 212989c6957cSBarry Smith v3 = a->a + diag[row-2] + 3; 213089c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 213189c6957cSBarry Smith i1 = idx[0]; 213289c6957cSBarry Smith i2 = idx[1]; 213389c6957cSBarry Smith idx += 2; 213489c6957cSBarry Smith tmp0 = x[i1]; 213589c6957cSBarry Smith tmp1 = x[i2]; 213689c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 213789c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 213889c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 213989c6957cSBarry Smith } 214089c6957cSBarry Smith 214189c6957cSBarry Smith if (n == sz-1){ 214289c6957cSBarry Smith tmp0 = x[*idx]; 214389c6957cSBarry Smith sum1 -= *v1*tmp0; 214489c6957cSBarry Smith sum2 -= *v2*tmp0; 214589c6957cSBarry Smith sum3 -= *v3*tmp0; 214689c6957cSBarry Smith } 2147938d4eb3SBarry Smith x[row] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8]; 2148938d4eb3SBarry Smith x[row-1] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7]; 2149938d4eb3SBarry Smith x[row-2] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6]; 2150938d4eb3SBarry Smith row -= 3; 215189c6957cSBarry Smith break; 215289c6957cSBarry Smith case 4: 215389c6957cSBarry Smith 215489c6957cSBarry Smith sum1 = b[row]; 215589c6957cSBarry Smith sum2 = b[row-1]; 215689c6957cSBarry Smith sum3 = b[row-2]; 215789c6957cSBarry Smith sum4 = b[row-3]; 215889c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 215989c6957cSBarry Smith v3 = a->a + diag[row-2] + 3; 216089c6957cSBarry Smith v4 = a->a + diag[row-3] + 4; 216189c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 216289c6957cSBarry Smith i1 = idx[0]; 216389c6957cSBarry Smith i2 = idx[1]; 216489c6957cSBarry Smith idx += 2; 216589c6957cSBarry Smith tmp0 = x[i1]; 216689c6957cSBarry Smith tmp1 = x[i2]; 216789c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 216889c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 216989c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 217089c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 217189c6957cSBarry Smith } 217289c6957cSBarry Smith 217389c6957cSBarry Smith if (n == sz-1){ 217489c6957cSBarry Smith tmp0 = x[*idx]; 217589c6957cSBarry Smith sum1 -= *v1*tmp0; 217689c6957cSBarry Smith sum2 -= *v2*tmp0; 217789c6957cSBarry Smith sum3 -= *v3*tmp0; 217889c6957cSBarry Smith sum4 -= *v4*tmp0; 217989c6957cSBarry Smith } 2180938d4eb3SBarry Smith x[row] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15]; 2181938d4eb3SBarry Smith x[row-1] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14]; 2182938d4eb3SBarry Smith x[row-2] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13]; 2183938d4eb3SBarry Smith x[row-3] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12]; 2184938d4eb3SBarry Smith row -= 4; 218589c6957cSBarry Smith break; 218689c6957cSBarry Smith case 5: 218789c6957cSBarry Smith 218889c6957cSBarry Smith sum1 = b[row]; 218989c6957cSBarry Smith sum2 = b[row-1]; 219089c6957cSBarry Smith sum3 = b[row-2]; 219189c6957cSBarry Smith sum4 = b[row-3]; 219289c6957cSBarry Smith sum5 = b[row-4]; 219389c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 219489c6957cSBarry Smith v3 = a->a + diag[row-2] + 3; 219589c6957cSBarry Smith v4 = a->a + diag[row-3] + 4; 219689c6957cSBarry Smith v5 = a->a + diag[row-4] + 5; 219789c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 219889c6957cSBarry Smith i1 = idx[0]; 219989c6957cSBarry Smith i2 = idx[1]; 220089c6957cSBarry Smith idx += 2; 220189c6957cSBarry Smith tmp0 = x[i1]; 220289c6957cSBarry Smith tmp1 = x[i2]; 220389c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 220489c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 220589c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 220689c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 220789c6957cSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 220889c6957cSBarry Smith } 220989c6957cSBarry Smith 221089c6957cSBarry Smith if (n == sz-1){ 221189c6957cSBarry Smith tmp0 = x[*idx]; 221289c6957cSBarry Smith sum1 -= *v1*tmp0; 221389c6957cSBarry Smith sum2 -= *v2*tmp0; 221489c6957cSBarry Smith sum3 -= *v3*tmp0; 221589c6957cSBarry Smith sum4 -= *v4*tmp0; 221689c6957cSBarry Smith sum5 -= *v5*tmp0; 221789c6957cSBarry Smith } 2218938d4eb3SBarry Smith x[row] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24]; 2219938d4eb3SBarry Smith x[row-1] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23]; 2220938d4eb3SBarry Smith x[row-2] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22]; 2221938d4eb3SBarry Smith x[row-3] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21]; 2222938d4eb3SBarry Smith x[row-4] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20]; 2223938d4eb3SBarry Smith row -= 5; 222489c6957cSBarry Smith break; 222589c6957cSBarry Smith default: 222689c6957cSBarry Smith SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 222789c6957cSBarry Smith } 222889c6957cSBarry Smith CHKMEMQ; 222989c6957cSBarry Smith } 223089c6957cSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 223189c6957cSBarry Smith 223289c6957cSBarry Smith /* 223389c6957cSBarry Smith t = b - D x where D is the block diagonal 223489c6957cSBarry Smith */ 223589c6957cSBarry Smith cnt = 0; 223689c6957cSBarry Smith for (i=0, row=0; i<m; i++) { 223789c6957cSBarry Smith CHKMEMQ; 223889c6957cSBarry Smith switch (sizes[i]){ 223989c6957cSBarry Smith case 1: 224089c6957cSBarry Smith t[row] = b[row] - bdiag[cnt++]*x[row]; row++; 224189c6957cSBarry Smith break; 224289c6957cSBarry Smith case 2: 224389c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; 224489c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2]; 224589c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3]; 224689c6957cSBarry Smith t[row] = b[row] - tmp1; 224789c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; row += 2; 224889c6957cSBarry Smith cnt += 4; 224989c6957cSBarry Smith break; 225089c6957cSBarry Smith case 3: 225189c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; 225289c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6]; 225389c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7]; 225489c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8]; 225589c6957cSBarry Smith t[row] = b[row] - tmp1; 225689c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; 225789c6957cSBarry Smith t[row+2] = b[row+2] - tmp3; row += 3; 225889c6957cSBarry Smith cnt += 9; 225989c6957cSBarry Smith break; 226089c6957cSBarry Smith case 4: 226189c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; 226289c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12]; 226389c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13]; 226489c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14]; 226589c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15]; 226689c6957cSBarry Smith t[row] = b[row] - tmp1; 226789c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; 226889c6957cSBarry Smith t[row+2] = b[row+2] - tmp3; 226989c6957cSBarry Smith t[row+3] = b[row+3] - tmp4; row += 4; 227089c6957cSBarry Smith cnt += 16; 227189c6957cSBarry Smith break; 227289c6957cSBarry Smith case 5: 227389c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; x5 = x[row+4]; 227489c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20]; 227589c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21]; 227689c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22]; 227789c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23]; 227889c6957cSBarry Smith tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24]; 227989c6957cSBarry Smith t[row] = b[row] - tmp1; 228089c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; 228189c6957cSBarry Smith t[row+2] = b[row+2] - tmp3; 228289c6957cSBarry Smith t[row+3] = b[row+3] - tmp4; 228389c6957cSBarry Smith t[row+4] = b[row+4] - tmp5;row += 5; 228489c6957cSBarry Smith cnt += 25; 228589c6957cSBarry Smith break; 228689c6957cSBarry Smith default: 228789c6957cSBarry Smith SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 228889c6957cSBarry Smith } 228989c6957cSBarry Smith CHKMEMQ; 229089c6957cSBarry Smith } 229189c6957cSBarry Smith ierr = PetscLogFlops(m);CHKERRQ(ierr); 229289c6957cSBarry Smith 229389c6957cSBarry Smith 229489c6957cSBarry Smith 229589c6957cSBarry Smith /* 229689c6957cSBarry Smith Apply (L + D)^-1 where D is the block diagonal 229789c6957cSBarry Smith */ 229889c6957cSBarry Smith for (i=0, row=0; i<m; i++) { 229989c6957cSBarry Smith sz = diag[row] - ii[row]; 230089c6957cSBarry Smith v1 = a->a + ii[row]; 230189c6957cSBarry Smith idx = a->j + ii[row]; 230289c6957cSBarry Smith CHKMEMQ; 23034108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 230489c6957cSBarry Smith switch (sizes[i]){ 230589c6957cSBarry Smith case 1: 230689c6957cSBarry Smith 230789c6957cSBarry Smith sum1 = t[row]; 230889c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 230989c6957cSBarry Smith i1 = idx[0]; 231089c6957cSBarry Smith i2 = idx[1]; 231189c6957cSBarry Smith idx += 2; 231289c6957cSBarry Smith tmp0 = t[i1]; 231389c6957cSBarry Smith tmp1 = t[i2]; 231489c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 231589c6957cSBarry Smith } 231689c6957cSBarry Smith 231789c6957cSBarry Smith if (n == sz-1){ 231889c6957cSBarry Smith tmp0 = t[*idx]; 231989c6957cSBarry Smith sum1 -= *v1 * tmp0; 232089c6957cSBarry Smith } 232189c6957cSBarry Smith x[row] += t[row] = sum1*(*ibdiag++); row++; 232289c6957cSBarry Smith break; 232389c6957cSBarry Smith case 2: 232489c6957cSBarry Smith v2 = a->a + ii[row+1]; 232589c6957cSBarry Smith sum1 = t[row]; 232689c6957cSBarry Smith sum2 = t[row+1]; 232789c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 232889c6957cSBarry Smith i1 = idx[0]; 232989c6957cSBarry Smith i2 = idx[1]; 233089c6957cSBarry Smith idx += 2; 233189c6957cSBarry Smith tmp0 = t[i1]; 233289c6957cSBarry Smith tmp1 = t[i2]; 233389c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 233489c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 233589c6957cSBarry Smith } 233689c6957cSBarry Smith 233789c6957cSBarry Smith if (n == sz-1){ 233889c6957cSBarry Smith tmp0 = t[*idx]; 233989c6957cSBarry Smith sum1 -= v1[0] * tmp0; 234089c6957cSBarry Smith sum2 -= v2[0] * tmp0; 234189c6957cSBarry Smith } 234289c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[2]; 234389c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3]; 234489c6957cSBarry Smith ibdiag += 4; row += 2; 234589c6957cSBarry Smith break; 234689c6957cSBarry Smith case 3: 234789c6957cSBarry Smith v2 = a->a + ii[row+1]; 234889c6957cSBarry Smith v3 = a->a + ii[row+2]; 234989c6957cSBarry Smith sum1 = t[row]; 235089c6957cSBarry Smith sum2 = t[row+1]; 235189c6957cSBarry Smith sum3 = t[row+2]; 235289c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 235389c6957cSBarry Smith i1 = idx[0]; 235489c6957cSBarry Smith i2 = idx[1]; 235589c6957cSBarry Smith idx += 2; 235689c6957cSBarry Smith tmp0 = t[i1]; 235789c6957cSBarry Smith tmp1 = t[i2]; 235889c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 235989c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 236089c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 236189c6957cSBarry Smith } 236289c6957cSBarry Smith 236389c6957cSBarry Smith if (n == sz-1){ 236489c6957cSBarry Smith tmp0 = t[*idx]; 236589c6957cSBarry Smith sum1 -= v1[0] * tmp0; 236689c6957cSBarry Smith sum2 -= v2[0] * tmp0; 236789c6957cSBarry Smith sum3 -= v3[0] * tmp0; 236889c6957cSBarry Smith } 236989c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6]; 237089c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7]; 237189c6957cSBarry Smith x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8]; 237289c6957cSBarry Smith ibdiag += 9; row += 3; 237389c6957cSBarry Smith break; 237489c6957cSBarry Smith case 4: 237589c6957cSBarry Smith v2 = a->a + ii[row+1]; 237689c6957cSBarry Smith v3 = a->a + ii[row+2]; 237789c6957cSBarry Smith v4 = a->a + ii[row+3]; 237889c6957cSBarry Smith sum1 = t[row]; 237989c6957cSBarry Smith sum2 = t[row+1]; 238089c6957cSBarry Smith sum3 = t[row+2]; 238189c6957cSBarry Smith sum4 = t[row+3]; 238289c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 238389c6957cSBarry Smith i1 = idx[0]; 238489c6957cSBarry Smith i2 = idx[1]; 238589c6957cSBarry Smith idx += 2; 238689c6957cSBarry Smith tmp0 = t[i1]; 238789c6957cSBarry Smith tmp1 = t[i2]; 238889c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 238989c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 239089c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 239189c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 239289c6957cSBarry Smith } 239389c6957cSBarry Smith 239489c6957cSBarry Smith if (n == sz-1){ 239589c6957cSBarry Smith tmp0 = t[*idx]; 239689c6957cSBarry Smith sum1 -= v1[0] * tmp0; 239789c6957cSBarry Smith sum2 -= v2[0] * tmp0; 239889c6957cSBarry Smith sum3 -= v3[0] * tmp0; 239989c6957cSBarry Smith sum4 -= v4[0] * tmp0; 240089c6957cSBarry Smith } 240189c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12]; 240289c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13]; 240389c6957cSBarry Smith x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14]; 240489c6957cSBarry Smith x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15]; 240589c6957cSBarry Smith ibdiag += 16; row += 4; 240689c6957cSBarry Smith break; 240789c6957cSBarry Smith case 5: 240889c6957cSBarry Smith v2 = a->a + ii[row+1]; 240989c6957cSBarry Smith v3 = a->a + ii[row+2]; 241089c6957cSBarry Smith v4 = a->a + ii[row+3]; 241189c6957cSBarry Smith v5 = a->a + ii[row+4]; 241289c6957cSBarry Smith sum1 = t[row]; 241389c6957cSBarry Smith sum2 = t[row+1]; 241489c6957cSBarry Smith sum3 = t[row+2]; 241589c6957cSBarry Smith sum4 = t[row+3]; 241689c6957cSBarry Smith sum5 = t[row+4]; 241789c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 241889c6957cSBarry Smith i1 = idx[0]; 241989c6957cSBarry Smith i2 = idx[1]; 242089c6957cSBarry Smith idx += 2; 242189c6957cSBarry Smith tmp0 = t[i1]; 242289c6957cSBarry Smith tmp1 = t[i2]; 242389c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 242489c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 242589c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 242689c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 242789c6957cSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 242889c6957cSBarry Smith } 242989c6957cSBarry Smith 243089c6957cSBarry Smith if (n == sz-1){ 243189c6957cSBarry Smith tmp0 = t[*idx]; 243289c6957cSBarry Smith sum1 -= v1[0] * tmp0; 243389c6957cSBarry Smith sum2 -= v2[0] * tmp0; 243489c6957cSBarry Smith sum3 -= v3[0] * tmp0; 243589c6957cSBarry Smith sum4 -= v4[0] * tmp0; 243689c6957cSBarry Smith sum5 -= v5[0] * tmp0; 243789c6957cSBarry Smith } 243889c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20]; 243989c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21]; 244089c6957cSBarry Smith x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22]; 244189c6957cSBarry Smith x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23]; 244289c6957cSBarry Smith x[row+4] += t[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24]; 244389c6957cSBarry Smith ibdiag += 25; row += 5; 244489c6957cSBarry Smith break; 244589c6957cSBarry Smith default: 244689c6957cSBarry Smith SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 244789c6957cSBarry Smith } 244889c6957cSBarry Smith CHKMEMQ; 244989c6957cSBarry Smith } 245089c6957cSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 24512af78befSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2452938d4eb3SBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 2453938d4eb3SBarry Smith } 24542af78befSBarry Smith PetscFunctionReturn(0); 24552af78befSBarry Smith } 24562af78befSBarry Smith 245789c6957cSBarry Smith #undef __FUNCT__ 24584108e4d5SBarry Smith #define __FUNCT__ "MatMultDiagonalBlock_SeqAIJ_Inode" 24594108e4d5SBarry Smith PetscErrorCode MatMultDiagonalBlock_SeqAIJ_Inode(Mat A,Vec bb,Vec xx) 246089c6957cSBarry Smith { 246189c6957cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 246289c6957cSBarry Smith PetscScalar *x,tmp1,tmp2,tmp3,tmp4,tmp5,x1,x2,x3,x4,x5; 246389c6957cSBarry Smith const MatScalar *bdiag = a->inode.bdiag; 246489c6957cSBarry Smith const PetscScalar *b; 246589c6957cSBarry Smith PetscErrorCode ierr; 246689c6957cSBarry Smith PetscInt m = a->inode.node_count,cnt = 0,i,row; 246789c6957cSBarry Smith const PetscInt *sizes = a->inode.size; 24682af78befSBarry Smith 246989c6957cSBarry Smith PetscFunctionBegin; 247089c6957cSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 247189c6957cSBarry Smith ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 247289c6957cSBarry Smith cnt = 0; 247389c6957cSBarry Smith for (i=0, row=0; i<m; i++) { 247489c6957cSBarry Smith switch (sizes[i]){ 247589c6957cSBarry Smith case 1: 247689c6957cSBarry Smith x[row] = b[row]*bdiag[cnt++];row++; 247789c6957cSBarry Smith break; 247889c6957cSBarry Smith case 2: 247989c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; 248089c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2]; 248189c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3]; 248289c6957cSBarry Smith x[row++] = tmp1; 248389c6957cSBarry Smith x[row++] = tmp2; 248489c6957cSBarry Smith cnt += 4; 248589c6957cSBarry Smith break; 248689c6957cSBarry Smith case 3: 248789c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; x3 = b[row+2]; 248889c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6]; 248989c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7]; 249089c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8]; 249189c6957cSBarry Smith x[row++] = tmp1; 249289c6957cSBarry Smith x[row++] = tmp2; 249389c6957cSBarry Smith x[row++] = tmp3; 249489c6957cSBarry Smith cnt += 9; 249589c6957cSBarry Smith break; 249689c6957cSBarry Smith case 4: 249789c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3]; 249889c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12]; 249989c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13]; 250089c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14]; 250189c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15]; 250289c6957cSBarry Smith x[row++] = tmp1; 250389c6957cSBarry Smith x[row++] = tmp2; 250489c6957cSBarry Smith x[row++] = tmp3; 250589c6957cSBarry Smith x[row++] = tmp4; 250689c6957cSBarry Smith cnt += 16; 250789c6957cSBarry Smith break; 250889c6957cSBarry Smith case 5: 250989c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3]; x5 = b[row+4]; 251089c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20]; 251189c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21]; 251289c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22]; 251389c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23]; 251489c6957cSBarry Smith tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24]; 251589c6957cSBarry Smith x[row++] = tmp1; 251689c6957cSBarry Smith x[row++] = tmp2; 251789c6957cSBarry Smith x[row++] = tmp3; 251889c6957cSBarry Smith x[row++] = tmp4; 251989c6957cSBarry Smith x[row++] = tmp5; 252089c6957cSBarry Smith cnt += 25; 252189c6957cSBarry Smith break; 252289c6957cSBarry Smith default: 252389c6957cSBarry Smith SETERRQ1(PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 252489c6957cSBarry Smith } 252589c6957cSBarry Smith } 252689c6957cSBarry Smith ierr = PetscLogFlops(2*cnt);CHKERRQ(ierr); 252789c6957cSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 252889c6957cSBarry Smith ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr); 252989c6957cSBarry Smith PetscFunctionReturn(0); 253089c6957cSBarry Smith } 253189c6957cSBarry Smith 25324c1414c8SBarry Smith /* 25334c1414c8SBarry Smith samestructure indicates that the matrix has not changed its nonzero structure so we 25344c1414c8SBarry Smith do not need to recompute the inodes 25354c1414c8SBarry Smith */ 25364c1414c8SBarry Smith #undef __FUNCT__ 25374c1414c8SBarry Smith #define __FUNCT__ "Mat_CheckInode" 25384c1414c8SBarry Smith PetscErrorCode Mat_CheckInode(Mat A,PetscTruth samestructure) 25394c1414c8SBarry Smith { 25404c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 25414c1414c8SBarry Smith PetscErrorCode ierr; 25424c1414c8SBarry Smith PetscInt i,j,m,nzx,nzy,*idx,*idy,*ns,*ii,node_count,blk_size; 2543d74fe821SBarry Smith PetscTruth flag; 25444c1414c8SBarry Smith 25454c1414c8SBarry Smith PetscFunctionBegin; 2546d74fe821SBarry Smith if (!a->inode.use) PetscFunctionReturn(0); 25474c1414c8SBarry Smith if (a->inode.checked && samestructure) PetscFunctionReturn(0); 25484c1414c8SBarry Smith 25494c1414c8SBarry Smith 2550d0f46423SBarry Smith m = A->rmap->n; 25514c1414c8SBarry Smith if (a->inode.size) {ns = a->inode.size;} 25524c1414c8SBarry Smith else {ierr = PetscMalloc((m+1)*sizeof(PetscInt),&ns);CHKERRQ(ierr);} 25534c1414c8SBarry Smith 25544c1414c8SBarry Smith i = 0; 25554c1414c8SBarry Smith node_count = 0; 25564c1414c8SBarry Smith idx = a->j; 25574c1414c8SBarry Smith ii = a->i; 25584c1414c8SBarry Smith while (i < m){ /* For each row */ 25594c1414c8SBarry Smith nzx = ii[i+1] - ii[i]; /* Number of nonzeros */ 25604c1414c8SBarry Smith /* Limits the number of elements in a node to 'a->inode.limit' */ 25614c1414c8SBarry Smith for (j=i+1,idy=idx,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) { 25624c1414c8SBarry Smith nzy = ii[j+1] - ii[j]; /* Same number of nonzeros */ 25634c1414c8SBarry Smith if (nzy != nzx) break; 25644c1414c8SBarry Smith idy += nzx; /* Same nonzero pattern */ 25654c1414c8SBarry Smith ierr = PetscMemcmp(idx,idy,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr); 25664c1414c8SBarry Smith if (!flag) break; 25674c1414c8SBarry Smith } 25684c1414c8SBarry Smith ns[node_count++] = blk_size; 25694c1414c8SBarry Smith idx += blk_size*nzx; 25704c1414c8SBarry Smith i = j; 25714c1414c8SBarry Smith } 25724c1414c8SBarry Smith /* If not enough inodes found,, do not use inode version of the routines */ 2573f0d39aaaSBarry Smith if (!a->inode.size && m && node_count > .9*m) { 25744c1414c8SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 25754c1414c8SBarry Smith a->inode.node_count = 0; 25764c1414c8SBarry Smith a->inode.size = PETSC_NULL; 25774c1414c8SBarry Smith a->inode.use = PETSC_FALSE; 25784c1414c8SBarry Smith ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr); 25794c1414c8SBarry Smith } else { 25804108e4d5SBarry Smith A->ops->mult = MatMult_SeqAIJ_Inode; 25814108e4d5SBarry Smith A->ops->sor = MatSOR_SeqAIJ_Inode; 25824108e4d5SBarry Smith A->ops->multadd = MatMultAdd_SeqAIJ_Inode; 25834108e4d5SBarry Smith A->ops->getrowij = MatGetRowIJ_SeqAIJ_Inode; 25844108e4d5SBarry Smith A->ops->restorerowij = MatRestoreRowIJ_SeqAIJ_Inode; 25854108e4d5SBarry Smith A->ops->getcolumnij = MatGetColumnIJ_SeqAIJ_Inode; 25864108e4d5SBarry Smith A->ops->restorecolumnij = MatRestoreColumnIJ_SeqAIJ_Inode; 25874108e4d5SBarry Smith A->ops->coloringpatch = MatColoringPatch_SeqAIJ_Inode; 25884108e4d5SBarry Smith A->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode; 25894c1414c8SBarry Smith a->inode.node_count = node_count; 25904c1414c8SBarry Smith a->inode.size = ns; 25914c1414c8SBarry Smith ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr); 25924c1414c8SBarry Smith } 25934c1414c8SBarry Smith PetscFunctionReturn(0); 25944c1414c8SBarry Smith } 25954c1414c8SBarry Smith 2596019b515eSShri Abhyankar #define MatGetRow_FactoredLU(cols,nzl,nzu,nz,ai,aj,adiag,row) { \ 2597019b515eSShri Abhyankar PetscInt __k, *__vi; \ 2598019b515eSShri Abhyankar __vi = aj + ai[row]; \ 2599019b515eSShri Abhyankar for(__k=0;__k<nzl;__k++) cols[__k] = __vi[__k]; \ 2600019b515eSShri Abhyankar __vi = aj + adiag[row]; \ 2601019b515eSShri Abhyankar cols[nzl] = __vi[0];\ 2602019b515eSShri Abhyankar __vi = aj + adiag[row+1]+1;\ 2603019b515eSShri Abhyankar for(__k=0;__k<nzu;__k++) cols[nzl+1+__k] = __vi[__k];} 2604019b515eSShri Abhyankar 2605019b515eSShri Abhyankar #undef __FUNCT__ 2606019b515eSShri Abhyankar #define __FUNCT__ "Mat_CheckInode_FactorLU" 2607019b515eSShri Abhyankar PetscErrorCode Mat_CheckInode_FactorLU(Mat A,PetscTruth samestructure) 2608019b515eSShri Abhyankar { 2609019b515eSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2610019b515eSShri Abhyankar PetscErrorCode ierr; 2611019b515eSShri Abhyankar PetscInt i,j,m,nzl1,nzu1,nzl2,nzu2,nzx,nzy,node_count,blk_size; 2612019b515eSShri Abhyankar PetscInt *cols1,*cols2,*ns,*ai = a->i,*aj = a->j, *adiag = a->diag; 2613019b515eSShri Abhyankar PetscTruth flag; 2614019b515eSShri Abhyankar 2615019b515eSShri Abhyankar PetscFunctionBegin; 2616019b515eSShri Abhyankar if (!a->inode.use) PetscFunctionReturn(0); 2617019b515eSShri Abhyankar if (a->inode.checked && samestructure) PetscFunctionReturn(0); 2618019b515eSShri Abhyankar 2619019b515eSShri Abhyankar m = A->rmap->n; 2620019b515eSShri Abhyankar if (a->inode.size) {ns = a->inode.size;} 2621019b515eSShri Abhyankar else {ierr = PetscMalloc((m+1)*sizeof(PetscInt),&ns);CHKERRQ(ierr);} 2622019b515eSShri Abhyankar 2623019b515eSShri Abhyankar i = 0; 2624019b515eSShri Abhyankar node_count = 0; 2625019b515eSShri Abhyankar while (i < m){ /* For each row */ 2626019b515eSShri Abhyankar nzl1 = ai[i+1] - ai[i]; /* Number of nonzeros in L */ 2627019b515eSShri Abhyankar nzu1 = adiag[i] - adiag[i+1] - 1; /* Number of nonzeros in U excluding diagonal*/ 2628019b515eSShri Abhyankar nzx = nzl1 + nzu1 + 1; 2629019b515eSShri Abhyankar ierr = PetscMalloc((nzx+1)*sizeof(PetscInt),&cols1);CHKERRQ(ierr); 2630019b515eSShri Abhyankar MatGetRow_FactoredLU(cols1,nzl1,nzu1,nzx,ai,aj,adiag,i); 2631019b515eSShri Abhyankar 2632019b515eSShri Abhyankar /* Limits the number of elements in a node to 'a->inode.limit' */ 2633019b515eSShri Abhyankar for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) { 2634019b515eSShri Abhyankar nzl2 = ai[j+1] - ai[j]; 2635019b515eSShri Abhyankar nzu2 = adiag[j] - adiag[j+1] - 1; 2636019b515eSShri Abhyankar nzy = nzl2 + nzu2 + 1; 2637019b515eSShri Abhyankar if( nzy != nzx) break; 2638019b515eSShri Abhyankar ierr = PetscMalloc((nzy+1)*sizeof(PetscInt),&cols2);CHKERRQ(ierr); 2639019b515eSShri Abhyankar MatGetRow_FactoredLU(cols2,nzl2,nzu2,nzy,ai,aj,adiag,j); 2640019b515eSShri Abhyankar ierr = PetscMemcmp(cols1,cols2,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr); 2641019b515eSShri Abhyankar if (!flag) {ierr = PetscFree(cols2);CHKERRQ(ierr);break;} 2642019b515eSShri Abhyankar ierr = PetscFree(cols2);CHKERRQ(ierr); 2643019b515eSShri Abhyankar } 2644019b515eSShri Abhyankar ns[node_count++] = blk_size; 2645019b515eSShri Abhyankar ierr = PetscFree(cols1);CHKERRQ(ierr); 2646019b515eSShri Abhyankar i = j; 2647019b515eSShri Abhyankar } 2648019b515eSShri Abhyankar /* If not enough inodes found,, do not use inode version of the routines */ 2649019b515eSShri Abhyankar if (!a->inode.size && m && node_count > .9*m) { 2650019b515eSShri Abhyankar ierr = PetscFree(ns);CHKERRQ(ierr); 2651019b515eSShri Abhyankar a->inode.node_count = 0; 2652019b515eSShri Abhyankar a->inode.size = PETSC_NULL; 2653019b515eSShri Abhyankar a->inode.use = PETSC_FALSE; 2654019b515eSShri Abhyankar ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr); 2655019b515eSShri Abhyankar } else { 2656019b515eSShri Abhyankar A->ops->mult = 0; 2657019b515eSShri Abhyankar A->ops->sor = 0; 2658019b515eSShri Abhyankar A->ops->multadd = 0; 2659019b515eSShri Abhyankar A->ops->getrowij = 0; 2660019b515eSShri Abhyankar A->ops->restorerowij = 0; 2661019b515eSShri Abhyankar A->ops->getcolumnij = 0; 2662019b515eSShri Abhyankar A->ops->restorecolumnij = 0; 2663019b515eSShri Abhyankar A->ops->coloringpatch = 0; 2664019b515eSShri Abhyankar A->ops->multdiagonalblock = 0; 2665019b515eSShri Abhyankar a->inode.node_count = node_count; 2666019b515eSShri Abhyankar a->inode.size = ns; 2667019b515eSShri Abhyankar ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr); 2668019b515eSShri Abhyankar } 2669019b515eSShri Abhyankar PetscFunctionReturn(0); 2670019b515eSShri Abhyankar } 2671019b515eSShri Abhyankar 26724c1414c8SBarry Smith /* 26734c1414c8SBarry Smith This is really ugly. if inodes are used this replaces the 26744c1414c8SBarry Smith permutations with ones that correspond to rows/cols of the matrix 26754c1414c8SBarry Smith rather then inode blocks 26764c1414c8SBarry Smith */ 26774c1414c8SBarry Smith #undef __FUNCT__ 26784c1414c8SBarry Smith #define __FUNCT__ "MatInodeAdjustForInodes" 26794c1414c8SBarry Smith PetscErrorCode PETSCMAT_DLLEXPORT MatInodeAdjustForInodes(Mat A,IS *rperm,IS *cperm) 26804c1414c8SBarry Smith { 26814c1414c8SBarry Smith PetscErrorCode ierr,(*f)(Mat,IS*,IS*); 26824c1414c8SBarry Smith 26834c1414c8SBarry Smith PetscFunctionBegin; 26844c1414c8SBarry Smith ierr = PetscObjectQueryFunction((PetscObject)A,"MatInodeAdjustForInodes_C",(void (**)(void))&f);CHKERRQ(ierr); 26854c1414c8SBarry Smith if (f) { 26864c1414c8SBarry Smith ierr = (*f)(A,rperm,cperm);CHKERRQ(ierr); 26874c1414c8SBarry Smith } 26884c1414c8SBarry Smith PetscFunctionReturn(0); 26894c1414c8SBarry Smith } 26904c1414c8SBarry Smith 26914c1414c8SBarry Smith EXTERN_C_BEGIN 26924c1414c8SBarry Smith #undef __FUNCT__ 26934108e4d5SBarry Smith #define __FUNCT__ "MatAdjustForInodes_SeqAIJ_Inode" 26944108e4d5SBarry Smith PetscErrorCode PETSCMAT_DLLEXPORT MatInodeAdjustForInodes_SeqAIJ_Inode(Mat A,IS *rperm,IS *cperm) 26954c1414c8SBarry Smith { 26964c1414c8SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 26974c1414c8SBarry Smith PetscErrorCode ierr; 26985d0c19d7SBarry Smith PetscInt m = A->rmap->n,n = A->cmap->n,i,j,nslim_row = a->inode.node_count; 26995d0c19d7SBarry Smith const PetscInt *ridx,*cidx; 27004c1414c8SBarry Smith PetscInt row,col,*permr,*permc,*ns_row = a->inode.size,*tns,start_val,end_val,indx; 27014c1414c8SBarry Smith PetscInt nslim_col,*ns_col; 27024c1414c8SBarry Smith IS ris = *rperm,cis = *cperm; 27034c1414c8SBarry Smith 27044c1414c8SBarry Smith PetscFunctionBegin; 27054c1414c8SBarry Smith if (!a->inode.size) PetscFunctionReturn(0); /* no inodes so return */ 27064c1414c8SBarry Smith if (a->inode.node_count == m) PetscFunctionReturn(0); /* all inodes are of size 1 */ 27074c1414c8SBarry Smith 27084c1414c8SBarry Smith ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr); 27094c1414c8SBarry Smith ierr = PetscMalloc((((nslim_row>nslim_col)?nslim_row:nslim_col)+1)*sizeof(PetscInt),&tns);CHKERRQ(ierr); 27100e83c824SBarry Smith ierr = PetscMalloc2(m,PetscInt,&permr,n,PetscInt,&permc);CHKERRQ(ierr); 27114c1414c8SBarry Smith 27124c1414c8SBarry Smith ierr = ISGetIndices(ris,&ridx);CHKERRQ(ierr); 27134c1414c8SBarry Smith ierr = ISGetIndices(cis,&cidx);CHKERRQ(ierr); 27144c1414c8SBarry Smith 27154c1414c8SBarry Smith /* Form the inode structure for the rows of permuted matric using inv perm*/ 27164c1414c8SBarry Smith for (i=0,tns[0]=0; i<nslim_row; ++i) tns[i+1] = tns[i] + ns_row[i]; 27174c1414c8SBarry Smith 27184c1414c8SBarry Smith /* Construct the permutations for rows*/ 27194c1414c8SBarry Smith for (i=0,row = 0; i<nslim_row; ++i){ 27204c1414c8SBarry Smith indx = ridx[i]; 27214c1414c8SBarry Smith start_val = tns[indx]; 27224c1414c8SBarry Smith end_val = tns[indx + 1]; 27234c1414c8SBarry Smith for (j=start_val; j<end_val; ++j,++row) permr[row]= j; 27244c1414c8SBarry Smith } 27254c1414c8SBarry Smith 27264c1414c8SBarry Smith /* Form the inode structure for the columns of permuted matrix using inv perm*/ 27274c1414c8SBarry Smith for (i=0,tns[0]=0; i<nslim_col; ++i) tns[i+1] = tns[i] + ns_col[i]; 27284c1414c8SBarry Smith 27294c1414c8SBarry Smith /* Construct permutations for columns */ 27304c1414c8SBarry Smith for (i=0,col=0; i<nslim_col; ++i){ 27314c1414c8SBarry Smith indx = cidx[i]; 27324c1414c8SBarry Smith start_val = tns[indx]; 27334c1414c8SBarry Smith end_val = tns[indx + 1]; 27344c1414c8SBarry Smith for (j = start_val; j<end_val; ++j,++col) permc[col]= j; 27354c1414c8SBarry Smith } 27364c1414c8SBarry Smith 27374c1414c8SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permr,rperm);CHKERRQ(ierr); 27380e83c824SBarry Smith ierr = ISSetPermutation(*rperm);CHKERRQ(ierr); 27394c1414c8SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permc,cperm);CHKERRQ(ierr); 27400e83c824SBarry Smith ierr = ISSetPermutation(*cperm);CHKERRQ(ierr); 27414c1414c8SBarry Smith 27424c1414c8SBarry Smith ierr = ISRestoreIndices(ris,&ridx);CHKERRQ(ierr); 27434c1414c8SBarry Smith ierr = ISRestoreIndices(cis,&cidx);CHKERRQ(ierr); 27444c1414c8SBarry Smith 27454c1414c8SBarry Smith ierr = PetscFree(ns_col);CHKERRQ(ierr); 27460e83c824SBarry Smith ierr = PetscFree2(permr,permc);CHKERRQ(ierr); 27474c1414c8SBarry Smith ierr = ISDestroy(cis);CHKERRQ(ierr); 27484c1414c8SBarry Smith ierr = ISDestroy(ris);CHKERRQ(ierr); 27494c1414c8SBarry Smith ierr = PetscFree(tns);CHKERRQ(ierr); 27504c1414c8SBarry Smith PetscFunctionReturn(0); 27514c1414c8SBarry Smith } 27524c1414c8SBarry Smith EXTERN_C_END 27534c1414c8SBarry Smith 27544c1414c8SBarry Smith #undef __FUNCT__ 27554c1414c8SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes" 27564c1414c8SBarry Smith /*@C 27574c1414c8SBarry Smith MatInodeGetInodeSizes - Returns the inode information of the Inode matrix. 27584c1414c8SBarry Smith 27594c1414c8SBarry Smith Collective on Mat 27604c1414c8SBarry Smith 27614c1414c8SBarry Smith Input Parameter: 27624c1414c8SBarry Smith . A - the Inode matrix or matrix derived from the Inode class -- e.g., SeqAIJ 27634c1414c8SBarry Smith 27644c1414c8SBarry Smith Output Parameter: 27654c1414c8SBarry Smith + node_count - no of inodes present in the matrix. 27664c1414c8SBarry Smith . sizes - an array of size node_count,with sizes of each inode. 27674c1414c8SBarry Smith - limit - the max size used to generate the inodes. 27684c1414c8SBarry Smith 27694c1414c8SBarry Smith Level: advanced 27704c1414c8SBarry Smith 27714c1414c8SBarry Smith Notes: This routine returns some internal storage information 27724c1414c8SBarry Smith of the matrix, it is intended to be used by advanced users. 27734c1414c8SBarry Smith It should be called after the matrix is assembled. 27744c1414c8SBarry Smith The contents of the sizes[] array should not be changed. 27754c1414c8SBarry Smith PETSC_NULL may be passed for information not requested. 27764c1414c8SBarry Smith 27774c1414c8SBarry Smith .keywords: matrix, seqaij, get, inode 27784c1414c8SBarry Smith 27794c1414c8SBarry Smith .seealso: MatGetInfo() 27804c1414c8SBarry Smith @*/ 27814c1414c8SBarry Smith PetscErrorCode PETSCMAT_DLLEXPORT MatInodeGetInodeSizes(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit) 27824c1414c8SBarry Smith { 27834c1414c8SBarry Smith PetscErrorCode ierr,(*f)(Mat,PetscInt*,PetscInt*[],PetscInt*); 27844c1414c8SBarry Smith 27854c1414c8SBarry Smith PetscFunctionBegin; 27864c1414c8SBarry Smith if (!A->assembled) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix"); 27874c1414c8SBarry Smith ierr = PetscObjectQueryFunction((PetscObject)A,"MatInodeGetInodeSizes_C",(void (**)(void))&f);CHKERRQ(ierr); 27884c1414c8SBarry Smith if (f) { 27894c1414c8SBarry Smith ierr = (*f)(A,node_count,sizes,limit);CHKERRQ(ierr); 27904c1414c8SBarry Smith } 27914c1414c8SBarry Smith PetscFunctionReturn(0); 27924c1414c8SBarry Smith } 27934c1414c8SBarry Smith 27944c1414c8SBarry Smith EXTERN_C_BEGIN 27954c1414c8SBarry Smith #undef __FUNCT__ 27964108e4d5SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes_SeqAIJ_Inode" 27974108e4d5SBarry Smith PetscErrorCode PETSCMAT_DLLEXPORT MatInodeGetInodeSizes_SeqAIJ_Inode(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit) 27984c1414c8SBarry Smith { 27994c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 28004c1414c8SBarry Smith 28014c1414c8SBarry Smith PetscFunctionBegin; 28024c1414c8SBarry Smith if (node_count) *node_count = a->inode.node_count; 28034c1414c8SBarry Smith if (sizes) *sizes = a->inode.size; 28044c1414c8SBarry Smith if (limit) *limit = a->inode.limit; 28054c1414c8SBarry Smith PetscFunctionReturn(0); 28064c1414c8SBarry Smith } 28074c1414c8SBarry Smith EXTERN_C_END 2808