14c1414c8SBarry Smith 24c1414c8SBarry Smith /* 34c1414c8SBarry Smith This file provides high performance routines for the Inode format (compressed sparse row) 44c1414c8SBarry Smith by taking advantage of rows with identical nonzero structure (I-nodes). 54c1414c8SBarry Smith */ 6c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/aij.h> 74c1414c8SBarry Smith 84c1414c8SBarry Smith #undef __FUNCT__ 94c1414c8SBarry Smith #define __FUNCT__ "Mat_CreateColInode" 104c1414c8SBarry Smith static PetscErrorCode Mat_CreateColInode(Mat A,PetscInt *size,PetscInt **ns) 114c1414c8SBarry Smith { 124c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 134c1414c8SBarry Smith PetscErrorCode ierr; 144c1414c8SBarry Smith PetscInt i,count,m,n,min_mn,*ns_row,*ns_col; 154c1414c8SBarry Smith 164c1414c8SBarry Smith PetscFunctionBegin; 17d0f46423SBarry Smith n = A->cmap->n; 18d0f46423SBarry Smith m = A->rmap->n; 194c1414c8SBarry Smith ns_row = a->inode.size; 204c1414c8SBarry Smith 214c1414c8SBarry Smith min_mn = (m < n) ? m : n; 224c1414c8SBarry Smith if (!ns) { 234c1414c8SBarry Smith for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) ; 244c1414c8SBarry Smith for (; count+1 < n; count++,i++) ; 254c1414c8SBarry Smith if (count < n) { 264c1414c8SBarry Smith i++; 274c1414c8SBarry Smith } 284c1414c8SBarry Smith *size = i; 294c1414c8SBarry Smith PetscFunctionReturn(0); 304c1414c8SBarry Smith } 31854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&ns_col);CHKERRQ(ierr); 324c1414c8SBarry Smith 334c1414c8SBarry Smith /* Use the same row structure wherever feasible. */ 344c1414c8SBarry Smith for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) { 354c1414c8SBarry Smith ns_col[i] = ns_row[i]; 364c1414c8SBarry Smith } 374c1414c8SBarry Smith 384c1414c8SBarry Smith /* if m < n; pad up the remainder with inode_limit */ 394c1414c8SBarry Smith for (; count+1 < n; count++,i++) { 404c1414c8SBarry Smith ns_col[i] = 1; 414c1414c8SBarry Smith } 424c1414c8SBarry Smith /* The last node is the odd ball. padd it up with the remaining rows; */ 434c1414c8SBarry Smith if (count < n) { 444c1414c8SBarry Smith ns_col[i] = n - count; 454c1414c8SBarry Smith i++; 464c1414c8SBarry Smith } else if (count > n) { 474c1414c8SBarry Smith /* Adjust for the over estimation */ 484c1414c8SBarry Smith ns_col[i-1] += n - count; 494c1414c8SBarry Smith } 504c1414c8SBarry Smith *size = i; 514c1414c8SBarry Smith *ns = ns_col; 524c1414c8SBarry Smith PetscFunctionReturn(0); 534c1414c8SBarry Smith } 544c1414c8SBarry Smith 554c1414c8SBarry Smith 564c1414c8SBarry Smith /* 574c1414c8SBarry Smith This builds symmetric version of nonzero structure, 584c1414c8SBarry Smith */ 594c1414c8SBarry Smith #undef __FUNCT__ 604108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode_Symmetric" 611a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Symmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift) 624c1414c8SBarry Smith { 634c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 644c1414c8SBarry Smith PetscErrorCode ierr; 658758e1faSBarry Smith PetscInt *work,*ia,*ja,nz,nslim_row,nslim_col,m,row,col,n; 668758e1faSBarry Smith PetscInt *tns,*tvc,*ns_row = a->inode.size,*ns_col,nsz,i1,i2; 678758e1faSBarry Smith const PetscInt *j,*jmax,*ai= a->i,*aj = a->j; 684c1414c8SBarry Smith 694c1414c8SBarry Smith PetscFunctionBegin; 704c1414c8SBarry Smith nslim_row = a->inode.node_count; 71d0f46423SBarry Smith m = A->rmap->n; 72d0f46423SBarry Smith n = A->cmap->n; 73e32f2f54SBarry Smith if (m != n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatGetRowIJ_SeqAIJ_Inode_Symmetric: Matrix should be square"); 744c1414c8SBarry Smith 754c1414c8SBarry Smith /* Use the row_inode as column_inode */ 764c1414c8SBarry Smith nslim_col = nslim_row; 774c1414c8SBarry Smith ns_col = ns_row; 784c1414c8SBarry Smith 794c1414c8SBarry Smith /* allocate space for reformated inode structure */ 80854ce69bSBarry Smith ierr = PetscMalloc1(nslim_col+1,&tns);CHKERRQ(ierr); 81854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&tvc);CHKERRQ(ierr); 824c1414c8SBarry Smith for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1]+ ns_row[i1]; 834c1414c8SBarry Smith 844c1414c8SBarry Smith for (i1=0,col=0; i1<nslim_col; ++i1) { 854c1414c8SBarry Smith nsz = ns_col[i1]; 862205254eSKarl Rupp for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1; 874c1414c8SBarry Smith } 884c1414c8SBarry Smith /* allocate space for row pointers */ 89854ce69bSBarry Smith ierr = PetscMalloc1(nslim_row+1,&ia);CHKERRQ(ierr); 904c1414c8SBarry Smith *iia = ia; 914c1414c8SBarry Smith ierr = PetscMemzero(ia,(nslim_row+1)*sizeof(PetscInt));CHKERRQ(ierr); 92854ce69bSBarry Smith ierr = PetscMalloc1(nslim_row+1,&work);CHKERRQ(ierr); 934c1414c8SBarry Smith 944c1414c8SBarry Smith /* determine the number of columns in each row */ 954c1414c8SBarry Smith ia[0] = oshift; 964c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 974c1414c8SBarry Smith 984c1414c8SBarry Smith j = aj + ai[row] + ishift; 994c1414c8SBarry Smith jmax = aj + ai[row+1] + ishift; 1004c1414c8SBarry Smith col = *j++ + ishift; 1014c1414c8SBarry Smith i2 = tvc[col]; 1024c1414c8SBarry Smith while (i2<i1 && j<jmax) { /* 1.[-xx-d-xx--] 2.[-xx-------],off-diagonal elemets */ 1034c1414c8SBarry Smith ia[i1+1]++; 1044c1414c8SBarry Smith ia[i2+1]++; 1054c1414c8SBarry Smith i2++; /* Start col of next node */ 10690d2dec7SBarry Smith while ((j<jmax) && ((col=*j+ishift)<tns[i2])) ++j; 1074c1414c8SBarry Smith i2 = tvc[col]; 1084c1414c8SBarry Smith } 1094c1414c8SBarry Smith if (i2 == i1) ia[i2+1]++; /* now the diagonal element */ 1104c1414c8SBarry Smith } 1114c1414c8SBarry Smith 1124c1414c8SBarry Smith /* shift ia[i] to point to next row */ 1134c1414c8SBarry Smith for (i1=1; i1<nslim_row+1; i1++) { 1144c1414c8SBarry Smith row = ia[i1-1]; 1154c1414c8SBarry Smith ia[i1] += row; 1164c1414c8SBarry Smith work[i1-1] = row - oshift; 1174c1414c8SBarry Smith } 1184c1414c8SBarry Smith 1194c1414c8SBarry Smith /* allocate space for column pointers */ 1204c1414c8SBarry Smith nz = ia[nslim_row] + (!ishift); 121785e854fSJed Brown ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr); 1224c1414c8SBarry Smith *jja = ja; 1234c1414c8SBarry Smith 1244c1414c8SBarry Smith /* loop over lower triangular part putting into ja */ 1254c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row += ns_row[i1],i1++) { 1264c1414c8SBarry Smith j = aj + ai[row] + ishift; 1274c1414c8SBarry Smith jmax = aj + ai[row+1] + ishift; 1284c1414c8SBarry Smith col = *j++ + ishift; 1294c1414c8SBarry Smith i2 = tvc[col]; 1304c1414c8SBarry Smith while (i2<i1 && j<jmax) { 1314c1414c8SBarry Smith ja[work[i2]++] = i1 + oshift; 1324c1414c8SBarry Smith ja[work[i1]++] = i2 + oshift; 1334c1414c8SBarry Smith ++i2; 13490d2dec7SBarry Smith while ((j<jmax) && ((col=*j+ishift)< tns[i2])) ++j; /* Skip rest col indices in this node */ 1354c1414c8SBarry Smith i2 = tvc[col]; 1364c1414c8SBarry Smith } 1374c1414c8SBarry Smith if (i2 == i1) ja[work[i1]++] = i2 + oshift; 1384c1414c8SBarry Smith 1394c1414c8SBarry Smith } 1404c1414c8SBarry Smith ierr = PetscFree(work);CHKERRQ(ierr); 1414c1414c8SBarry Smith ierr = PetscFree(tns);CHKERRQ(ierr); 1424c1414c8SBarry Smith ierr = PetscFree(tvc);CHKERRQ(ierr); 1434c1414c8SBarry Smith PetscFunctionReturn(0); 1444c1414c8SBarry Smith } 1454c1414c8SBarry Smith 1464c1414c8SBarry Smith /* 1474c1414c8SBarry Smith This builds nonsymmetric version of nonzero structure, 1484c1414c8SBarry Smith */ 1494c1414c8SBarry Smith #undef __FUNCT__ 1504108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric" 1511a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift) 1524c1414c8SBarry Smith { 1534c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1544c1414c8SBarry Smith PetscErrorCode ierr; 1558758e1faSBarry Smith PetscInt *work,*ia,*ja,nz,nslim_row,n,row,col,*ns_col,nslim_col; 1568758e1faSBarry Smith PetscInt *tns,*tvc,nsz,i1,i2; 1578758e1faSBarry Smith const PetscInt *j,*ai= a->i,*aj = a->j,*ns_row = a->inode.size; 1584c1414c8SBarry Smith 1594c1414c8SBarry Smith PetscFunctionBegin; 1604c1414c8SBarry Smith nslim_row = a->inode.node_count; 161d0f46423SBarry Smith n = A->cmap->n; 1624c1414c8SBarry Smith 1634c1414c8SBarry Smith /* Create The column_inode for this matrix */ 1644c1414c8SBarry Smith ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr); 1654c1414c8SBarry Smith 1664c1414c8SBarry Smith /* allocate space for reformated column_inode structure */ 167854ce69bSBarry Smith ierr = PetscMalloc1(nslim_col +1,&tns);CHKERRQ(ierr); 168854ce69bSBarry Smith ierr = PetscMalloc1(n + 1,&tvc);CHKERRQ(ierr); 1694c1414c8SBarry Smith for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1]; 1704c1414c8SBarry Smith 1714c1414c8SBarry Smith for (i1=0,col=0; i1<nslim_col; ++i1) { 1724c1414c8SBarry Smith nsz = ns_col[i1]; 1732205254eSKarl Rupp for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1; 1744c1414c8SBarry Smith } 1754c1414c8SBarry Smith /* allocate space for row pointers */ 176854ce69bSBarry Smith ierr = PetscMalloc1(nslim_row+1,&ia);CHKERRQ(ierr); 1774c1414c8SBarry Smith *iia = ia; 1784c1414c8SBarry Smith ierr = PetscMemzero(ia,(nslim_row+1)*sizeof(PetscInt));CHKERRQ(ierr); 179854ce69bSBarry Smith ierr = PetscMalloc1(nslim_row+1,&work);CHKERRQ(ierr); 1804c1414c8SBarry Smith 1814c1414c8SBarry Smith /* determine the number of columns in each row */ 1824c1414c8SBarry Smith ia[0] = oshift; 1834c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 1844c1414c8SBarry Smith j = aj + ai[row] + ishift; 1854c1414c8SBarry Smith col = *j++ + ishift; 1864c1414c8SBarry Smith i2 = tvc[col]; 1874c1414c8SBarry Smith nz = ai[row+1] - ai[row]; 1884c1414c8SBarry Smith while (nz-- > 0) { /* off-diagonal elemets */ 1894c1414c8SBarry Smith ia[i1+1]++; 1904c1414c8SBarry Smith i2++; /* Start col of next node */ 191a8e3a797SJed Brown while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--; 1924c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 1934c1414c8SBarry Smith } 1944c1414c8SBarry Smith } 1954c1414c8SBarry Smith 1964c1414c8SBarry Smith /* shift ia[i] to point to next row */ 1974c1414c8SBarry Smith for (i1=1; i1<nslim_row+1; i1++) { 1984c1414c8SBarry Smith row = ia[i1-1]; 1994c1414c8SBarry Smith ia[i1] += row; 2004c1414c8SBarry Smith work[i1-1] = row - oshift; 2014c1414c8SBarry Smith } 2024c1414c8SBarry Smith 2034c1414c8SBarry Smith /* allocate space for column pointers */ 2044c1414c8SBarry Smith nz = ia[nslim_row] + (!ishift); 205785e854fSJed Brown ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr); 2064c1414c8SBarry Smith *jja = ja; 2074c1414c8SBarry Smith 2084c1414c8SBarry Smith /* loop over matrix putting into ja */ 2094c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 2104c1414c8SBarry Smith j = aj + ai[row] + ishift; 2114c1414c8SBarry Smith col = *j++ + ishift; 2124c1414c8SBarry Smith i2 = tvc[col]; 2134c1414c8SBarry Smith nz = ai[row+1] - ai[row]; 2144c1414c8SBarry Smith while (nz-- > 0) { 2154c1414c8SBarry Smith ja[work[i1]++] = i2 + oshift; 2164c1414c8SBarry Smith ++i2; 217a8e3a797SJed Brown while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--; 2184c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 2194c1414c8SBarry Smith } 2204c1414c8SBarry Smith } 2214c1414c8SBarry Smith ierr = PetscFree(ns_col);CHKERRQ(ierr); 2224c1414c8SBarry Smith ierr = PetscFree(work);CHKERRQ(ierr); 2234c1414c8SBarry Smith ierr = PetscFree(tns);CHKERRQ(ierr); 2244c1414c8SBarry Smith ierr = PetscFree(tvc);CHKERRQ(ierr); 2254c1414c8SBarry Smith PetscFunctionReturn(0); 2264c1414c8SBarry Smith } 2274c1414c8SBarry Smith 2284c1414c8SBarry Smith #undef __FUNCT__ 2294108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode" 2301a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 2314c1414c8SBarry Smith { 2324c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2334c1414c8SBarry Smith PetscErrorCode ierr; 2344c1414c8SBarry Smith 2354c1414c8SBarry Smith PetscFunctionBegin; 2364c1414c8SBarry Smith *n = a->inode.node_count; 2374c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 2388f7157efSSatish Balay if (!blockcompressed) { 2398f7157efSSatish Balay ierr = MatGetRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);; 2408f7157efSSatish Balay } else if (symmetric) { 2414108e4d5SBarry Smith ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 2424c1414c8SBarry Smith } else { 2434108e4d5SBarry Smith ierr = MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 2444c1414c8SBarry Smith } 2454c1414c8SBarry Smith PetscFunctionReturn(0); 2464c1414c8SBarry Smith } 2474c1414c8SBarry Smith 2484c1414c8SBarry Smith #undef __FUNCT__ 2494108e4d5SBarry Smith #define __FUNCT__ "MatRestoreRowIJ_SeqAIJ_Inode" 2501a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 2514c1414c8SBarry Smith { 2524c1414c8SBarry Smith PetscErrorCode ierr; 2534c1414c8SBarry Smith 2544c1414c8SBarry Smith PetscFunctionBegin; 2554c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 2568f7157efSSatish Balay 2578f7157efSSatish Balay if (!blockcompressed) { 2588f7157efSSatish Balay ierr = MatRestoreRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);; 2598f7157efSSatish Balay } else { 2604c1414c8SBarry Smith ierr = PetscFree(*ia);CHKERRQ(ierr); 2614c1414c8SBarry Smith ierr = PetscFree(*ja);CHKERRQ(ierr); 2628f7157efSSatish Balay } 2634c1414c8SBarry Smith PetscFunctionReturn(0); 2644c1414c8SBarry Smith } 2654c1414c8SBarry Smith 2664c1414c8SBarry Smith /* ----------------------------------------------------------- */ 2674c1414c8SBarry Smith 2684c1414c8SBarry Smith #undef __FUNCT__ 2694108e4d5SBarry Smith #define __FUNCT__ "MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric" 2701a83f524SJed Brown static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift) 2714c1414c8SBarry Smith { 2724c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2734c1414c8SBarry Smith PetscErrorCode ierr; 2744c1414c8SBarry Smith PetscInt *work,*ia,*ja,*j,nz,nslim_row, n,row,col,*ns_col,nslim_col; 2754c1414c8SBarry Smith PetscInt *tns,*tvc,*ns_row = a->inode.size,nsz,i1,i2,*ai= a->i,*aj = a->j; 2764c1414c8SBarry Smith 2774c1414c8SBarry Smith PetscFunctionBegin; 2784c1414c8SBarry Smith nslim_row = a->inode.node_count; 279d0f46423SBarry Smith n = A->cmap->n; 2804c1414c8SBarry Smith 2814c1414c8SBarry Smith /* Create The column_inode for this matrix */ 2824c1414c8SBarry Smith ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr); 2834c1414c8SBarry Smith 2844c1414c8SBarry Smith /* allocate space for reformated column_inode structure */ 285854ce69bSBarry Smith ierr = PetscMalloc1(nslim_col + 1,&tns);CHKERRQ(ierr); 286854ce69bSBarry Smith ierr = PetscMalloc1(n + 1,&tvc);CHKERRQ(ierr); 2874c1414c8SBarry Smith for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1]; 2884c1414c8SBarry Smith 2894c1414c8SBarry Smith for (i1=0,col=0; i1<nslim_col; ++i1) { 2904c1414c8SBarry Smith nsz = ns_col[i1]; 2912205254eSKarl Rupp for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1; 2924c1414c8SBarry Smith } 2934c1414c8SBarry Smith /* allocate space for column pointers */ 294854ce69bSBarry Smith ierr = PetscMalloc1(nslim_col+1,&ia);CHKERRQ(ierr); 2954c1414c8SBarry Smith *iia = ia; 2964c1414c8SBarry Smith ierr = PetscMemzero(ia,(nslim_col+1)*sizeof(PetscInt));CHKERRQ(ierr); 297854ce69bSBarry Smith ierr = PetscMalloc1(nslim_col+1,&work);CHKERRQ(ierr); 2984c1414c8SBarry Smith 2994c1414c8SBarry Smith /* determine the number of columns in each row */ 3004c1414c8SBarry Smith ia[0] = oshift; 3014c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 3024c1414c8SBarry Smith j = aj + ai[row] + ishift; 3034c1414c8SBarry Smith col = *j++ + ishift; 3044c1414c8SBarry Smith i2 = tvc[col]; 3054c1414c8SBarry Smith nz = ai[row+1] - ai[row]; 3064c1414c8SBarry Smith while (nz-- > 0) { /* off-diagonal elemets */ 3074c1414c8SBarry Smith /* ia[i1+1]++; */ 3084c1414c8SBarry Smith ia[i2+1]++; 3094c1414c8SBarry Smith i2++; 310a8e3a797SJed Brown while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--; 3114c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 3124c1414c8SBarry Smith } 3134c1414c8SBarry Smith } 3144c1414c8SBarry Smith 3154c1414c8SBarry Smith /* shift ia[i] to point to next col */ 3164c1414c8SBarry Smith for (i1=1; i1<nslim_col+1; i1++) { 3174c1414c8SBarry Smith col = ia[i1-1]; 3184c1414c8SBarry Smith ia[i1] += col; 3194c1414c8SBarry Smith work[i1-1] = col - oshift; 3204c1414c8SBarry Smith } 3214c1414c8SBarry Smith 3224c1414c8SBarry Smith /* allocate space for column pointers */ 3234c1414c8SBarry Smith nz = ia[nslim_col] + (!ishift); 324785e854fSJed Brown ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr); 3254c1414c8SBarry Smith *jja = ja; 3264c1414c8SBarry Smith 3274c1414c8SBarry Smith /* loop over matrix putting into ja */ 3284c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 3294c1414c8SBarry Smith j = aj + ai[row] + ishift; 3304c1414c8SBarry Smith col = *j++ + ishift; 3314c1414c8SBarry Smith i2 = tvc[col]; 3324c1414c8SBarry Smith nz = ai[row+1] - ai[row]; 3334c1414c8SBarry Smith while (nz-- > 0) { 3344c1414c8SBarry Smith /* ja[work[i1]++] = i2 + oshift; */ 3354c1414c8SBarry Smith ja[work[i2]++] = i1 + oshift; 3364c1414c8SBarry Smith i2++; 337a8e3a797SJed Brown while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--; 3384c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 3394c1414c8SBarry Smith } 3404c1414c8SBarry Smith } 3414c1414c8SBarry Smith ierr = PetscFree(ns_col);CHKERRQ(ierr); 3424c1414c8SBarry Smith ierr = PetscFree(work);CHKERRQ(ierr); 3434c1414c8SBarry Smith ierr = PetscFree(tns);CHKERRQ(ierr); 3444c1414c8SBarry Smith ierr = PetscFree(tvc);CHKERRQ(ierr); 3454c1414c8SBarry Smith PetscFunctionReturn(0); 3464c1414c8SBarry Smith } 3474c1414c8SBarry Smith 3484c1414c8SBarry Smith #undef __FUNCT__ 3494108e4d5SBarry Smith #define __FUNCT__ "MatGetColumnIJ_SeqAIJ_Inode" 3501a83f524SJed Brown static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 3514c1414c8SBarry Smith { 3524c1414c8SBarry Smith PetscErrorCode ierr; 3534c1414c8SBarry Smith 3544c1414c8SBarry Smith PetscFunctionBegin; 3550298fd71SBarry Smith ierr = Mat_CreateColInode(A,n,NULL);CHKERRQ(ierr); 3564c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 3574c1414c8SBarry Smith 3588f7157efSSatish Balay if (!blockcompressed) { 3598f7157efSSatish Balay ierr = MatGetColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);; 3608f7157efSSatish Balay } else if (symmetric) { 3614c1414c8SBarry Smith /* Since the indices are symmetric it does'nt matter */ 3624108e4d5SBarry Smith ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 3634c1414c8SBarry Smith } else { 3644108e4d5SBarry Smith ierr = MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 3654c1414c8SBarry Smith } 3664c1414c8SBarry Smith PetscFunctionReturn(0); 3674c1414c8SBarry Smith } 3684c1414c8SBarry Smith 3694c1414c8SBarry Smith #undef __FUNCT__ 3704108e4d5SBarry Smith #define __FUNCT__ "MatRestoreColumnIJ_SeqAIJ_Inode" 3711a83f524SJed Brown static PetscErrorCode MatRestoreColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 3724c1414c8SBarry Smith { 3734c1414c8SBarry Smith PetscErrorCode ierr; 3744c1414c8SBarry Smith 3754c1414c8SBarry Smith PetscFunctionBegin; 3764c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 3778f7157efSSatish Balay if (!blockcompressed) { 3788f7157efSSatish Balay ierr = MatRestoreColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);; 3798f7157efSSatish Balay } else { 3804c1414c8SBarry Smith ierr = PetscFree(*ia);CHKERRQ(ierr); 3814c1414c8SBarry Smith ierr = PetscFree(*ja);CHKERRQ(ierr); 3828f7157efSSatish Balay } 3834c1414c8SBarry Smith PetscFunctionReturn(0); 3844c1414c8SBarry Smith } 3854c1414c8SBarry Smith 3864c1414c8SBarry Smith /* ----------------------------------------------------------- */ 3874c1414c8SBarry Smith 3884c1414c8SBarry Smith #undef __FUNCT__ 3894108e4d5SBarry Smith #define __FUNCT__ "MatMult_SeqAIJ_Inode" 3904108e4d5SBarry Smith static PetscErrorCode MatMult_SeqAIJ_Inode(Mat A,Vec xx,Vec yy) 3914c1414c8SBarry Smith { 3924c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3934c1414c8SBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5,tmp0,tmp1; 394d9fead3dSBarry Smith PetscScalar *y; 395dd6ea824SBarry Smith const PetscScalar *x; 396dd6ea824SBarry Smith const MatScalar *v1,*v2,*v3,*v4,*v5; 3974c1414c8SBarry Smith PetscErrorCode ierr; 3988758e1faSBarry Smith PetscInt i1,i2,n,i,row,node_max,nsz,sz,nonzerorow=0; 3998758e1faSBarry Smith const PetscInt *idx,*ns,*ii; 4004c1414c8SBarry Smith 4014c1414c8SBarry Smith #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 4024c1414c8SBarry Smith #pragma disjoint(*x,*y,*v1,*v2,*v3,*v4,*v5) 4034c1414c8SBarry Smith #endif 4044c1414c8SBarry Smith 4054c1414c8SBarry Smith PetscFunctionBegin; 406e32f2f54SBarry Smith if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 4074c1414c8SBarry Smith node_max = a->inode.node_count; 4084c1414c8SBarry Smith ns = a->inode.size; /* Node Size array */ 4093649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 4104c1414c8SBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 4114c1414c8SBarry Smith idx = a->j; 4124c1414c8SBarry Smith v1 = a->a; 4134c1414c8SBarry Smith ii = a->i; 4144c1414c8SBarry Smith 4154c1414c8SBarry Smith for (i = 0,row = 0; i< node_max; ++i) { 4164c1414c8SBarry Smith nsz = ns[i]; 4174c1414c8SBarry Smith n = ii[1] - ii[0]; 41898c9bda7SSatish Balay nonzerorow += (n>0)*nsz; 4194c1414c8SBarry Smith ii += nsz; 42050d8bf02SJed Brown PetscPrefetchBlock(idx+nsz*n,n,0,PETSC_PREFETCH_HINT_NTA); /* Prefetch the indices for the block row after the current one */ 42150d8bf02SJed Brown PetscPrefetchBlock(v1+nsz*n,nsz*n,0,PETSC_PREFETCH_HINT_NTA); /* Prefetch the values for the block row after the current one */ 4224c1414c8SBarry Smith sz = n; /* No of non zeros in this row */ 4234c1414c8SBarry Smith /* Switch on the size of Node */ 4244c1414c8SBarry Smith switch (nsz) { /* Each loop in 'case' is unrolled */ 4254c1414c8SBarry Smith case 1: 42675567043SBarry Smith sum1 = 0.; 4274c1414c8SBarry Smith 4284c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 4294c1414c8SBarry Smith i1 = idx[0]; /* The instructions are ordered to */ 4304c1414c8SBarry Smith i2 = idx[1]; /* make the compiler's job easy */ 4314c1414c8SBarry Smith idx += 2; 4324c1414c8SBarry Smith tmp0 = x[i1]; 4334c1414c8SBarry Smith tmp1 = x[i2]; 4344c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 4354c1414c8SBarry Smith } 4364c1414c8SBarry Smith 4374c1414c8SBarry Smith if (n == sz-1) { /* Take care of the last nonzero */ 4384c1414c8SBarry Smith tmp0 = x[*idx++]; 4394c1414c8SBarry Smith sum1 += *v1++ *tmp0; 4404c1414c8SBarry Smith } 4414c1414c8SBarry Smith y[row++]=sum1; 4424c1414c8SBarry Smith break; 4434c1414c8SBarry Smith case 2: 44475567043SBarry Smith sum1 = 0.; 44575567043SBarry Smith sum2 = 0.; 4464c1414c8SBarry Smith v2 = v1 + n; 4474c1414c8SBarry Smith 4484c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 4494c1414c8SBarry Smith i1 = idx[0]; 4504c1414c8SBarry Smith i2 = idx[1]; 4514c1414c8SBarry Smith idx += 2; 4524c1414c8SBarry Smith tmp0 = x[i1]; 4534c1414c8SBarry Smith tmp1 = x[i2]; 4544c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 4554c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 4564c1414c8SBarry Smith } 4574c1414c8SBarry Smith if (n == sz-1) { 4584c1414c8SBarry Smith tmp0 = x[*idx++]; 4594c1414c8SBarry Smith sum1 += *v1++ * tmp0; 4604c1414c8SBarry Smith sum2 += *v2++ * tmp0; 4614c1414c8SBarry Smith } 4624c1414c8SBarry Smith y[row++]=sum1; 4634c1414c8SBarry Smith y[row++]=sum2; 4644c1414c8SBarry Smith v1 =v2; /* Since the next block to be processed starts there*/ 4654c1414c8SBarry Smith idx +=sz; 4664c1414c8SBarry Smith break; 4674c1414c8SBarry Smith case 3: 46875567043SBarry Smith sum1 = 0.; 46975567043SBarry Smith sum2 = 0.; 47075567043SBarry Smith sum3 = 0.; 4714c1414c8SBarry Smith v2 = v1 + n; 4724c1414c8SBarry Smith v3 = v2 + n; 4734c1414c8SBarry Smith 4744c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 4754c1414c8SBarry Smith i1 = idx[0]; 4764c1414c8SBarry Smith i2 = idx[1]; 4774c1414c8SBarry Smith idx += 2; 4784c1414c8SBarry Smith tmp0 = x[i1]; 4794c1414c8SBarry Smith tmp1 = x[i2]; 4804c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 4814c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 4824c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 4834c1414c8SBarry Smith } 4844c1414c8SBarry Smith if (n == sz-1) { 4854c1414c8SBarry Smith tmp0 = x[*idx++]; 4864c1414c8SBarry Smith sum1 += *v1++ * tmp0; 4874c1414c8SBarry Smith sum2 += *v2++ * tmp0; 4884c1414c8SBarry Smith sum3 += *v3++ * tmp0; 4894c1414c8SBarry Smith } 4904c1414c8SBarry Smith y[row++]=sum1; 4914c1414c8SBarry Smith y[row++]=sum2; 4924c1414c8SBarry Smith y[row++]=sum3; 4934c1414c8SBarry Smith v1 =v3; /* Since the next block to be processed starts there*/ 4944c1414c8SBarry Smith idx +=2*sz; 4954c1414c8SBarry Smith break; 4964c1414c8SBarry Smith case 4: 49775567043SBarry Smith sum1 = 0.; 49875567043SBarry Smith sum2 = 0.; 49975567043SBarry Smith sum3 = 0.; 50075567043SBarry Smith sum4 = 0.; 5014c1414c8SBarry Smith v2 = v1 + n; 5024c1414c8SBarry Smith v3 = v2 + n; 5034c1414c8SBarry Smith v4 = v3 + n; 5044c1414c8SBarry Smith 5054c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 5064c1414c8SBarry Smith i1 = idx[0]; 5074c1414c8SBarry Smith i2 = idx[1]; 5084c1414c8SBarry Smith idx += 2; 5094c1414c8SBarry Smith tmp0 = x[i1]; 5104c1414c8SBarry Smith tmp1 = x[i2]; 5114c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 5124c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 5134c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 5144c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 5154c1414c8SBarry Smith } 5164c1414c8SBarry Smith if (n == sz-1) { 5174c1414c8SBarry Smith tmp0 = x[*idx++]; 5184c1414c8SBarry Smith sum1 += *v1++ * tmp0; 5194c1414c8SBarry Smith sum2 += *v2++ * tmp0; 5204c1414c8SBarry Smith sum3 += *v3++ * tmp0; 5214c1414c8SBarry Smith sum4 += *v4++ * tmp0; 5224c1414c8SBarry Smith } 5234c1414c8SBarry Smith y[row++]=sum1; 5244c1414c8SBarry Smith y[row++]=sum2; 5254c1414c8SBarry Smith y[row++]=sum3; 5264c1414c8SBarry Smith y[row++]=sum4; 5274c1414c8SBarry Smith v1 =v4; /* Since the next block to be processed starts there*/ 5284c1414c8SBarry Smith idx +=3*sz; 5294c1414c8SBarry Smith break; 5304c1414c8SBarry Smith case 5: 53175567043SBarry Smith sum1 = 0.; 53275567043SBarry Smith sum2 = 0.; 53375567043SBarry Smith sum3 = 0.; 53475567043SBarry Smith sum4 = 0.; 53575567043SBarry Smith sum5 = 0.; 5364c1414c8SBarry Smith v2 = v1 + n; 5374c1414c8SBarry Smith v3 = v2 + n; 5384c1414c8SBarry Smith v4 = v3 + n; 5394c1414c8SBarry Smith v5 = v4 + n; 5404c1414c8SBarry Smith 5414c1414c8SBarry Smith for (n = 0; n<sz-1; n+=2) { 5424c1414c8SBarry Smith i1 = idx[0]; 5434c1414c8SBarry Smith i2 = idx[1]; 5444c1414c8SBarry Smith idx += 2; 5454c1414c8SBarry Smith tmp0 = x[i1]; 5464c1414c8SBarry Smith tmp1 = x[i2]; 5474c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 5484c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 5494c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 5504c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 5514c1414c8SBarry Smith sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2; 5524c1414c8SBarry Smith } 5534c1414c8SBarry Smith if (n == sz-1) { 5544c1414c8SBarry Smith tmp0 = x[*idx++]; 5554c1414c8SBarry Smith sum1 += *v1++ * tmp0; 5564c1414c8SBarry Smith sum2 += *v2++ * tmp0; 5574c1414c8SBarry Smith sum3 += *v3++ * tmp0; 5584c1414c8SBarry Smith sum4 += *v4++ * tmp0; 5594c1414c8SBarry Smith sum5 += *v5++ * tmp0; 5604c1414c8SBarry Smith } 5614c1414c8SBarry Smith y[row++]=sum1; 5624c1414c8SBarry Smith y[row++]=sum2; 5634c1414c8SBarry Smith y[row++]=sum3; 5644c1414c8SBarry Smith y[row++]=sum4; 5654c1414c8SBarry Smith y[row++]=sum5; 5664c1414c8SBarry Smith v1 =v5; /* Since the next block to be processed starts there */ 5674c1414c8SBarry Smith idx +=4*sz; 5684c1414c8SBarry Smith break; 5694c1414c8SBarry Smith default: 570e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported"); 5714c1414c8SBarry Smith } 5724c1414c8SBarry Smith } 5733649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 5744c1414c8SBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 575dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr); 5764c1414c8SBarry Smith PetscFunctionReturn(0); 5774c1414c8SBarry Smith } 5784c1414c8SBarry Smith /* ----------------------------------------------------------- */ 5794108e4d5SBarry Smith /* Almost same code as the MatMult_SeqAIJ_Inode() */ 5804c1414c8SBarry Smith #undef __FUNCT__ 5814108e4d5SBarry Smith #define __FUNCT__ "MatMultAdd_SeqAIJ_Inode" 5824108e4d5SBarry Smith static PetscErrorCode MatMultAdd_SeqAIJ_Inode(Mat A,Vec xx,Vec zz,Vec yy) 5834c1414c8SBarry Smith { 5844c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 5854c1414c8SBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5,tmp0,tmp1; 5868758e1faSBarry Smith const MatScalar *v1,*v2,*v3,*v4,*v5; 5878758e1faSBarry Smith const PetscScalar *x; 5888758e1faSBarry Smith PetscScalar *y,*z,*zt; 5894c1414c8SBarry Smith PetscErrorCode ierr; 5908758e1faSBarry Smith PetscInt i1,i2,n,i,row,node_max,nsz,sz; 5918758e1faSBarry Smith const PetscInt *idx,*ns,*ii; 5924c1414c8SBarry Smith 5934c1414c8SBarry Smith PetscFunctionBegin; 594e32f2f54SBarry Smith if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 5954c1414c8SBarry Smith node_max = a->inode.node_count; 5964c1414c8SBarry Smith ns = a->inode.size; /* Node Size array */ 5972205254eSKarl Rupp 5988758e1faSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 5994c1414c8SBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 6004c1414c8SBarry Smith if (zz != yy) { 6014c1414c8SBarry Smith ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 6024c1414c8SBarry Smith } else { 6034c1414c8SBarry Smith z = y; 6044c1414c8SBarry Smith } 6054c1414c8SBarry Smith zt = z; 6064c1414c8SBarry Smith 6074c1414c8SBarry Smith idx = a->j; 6084c1414c8SBarry Smith v1 = a->a; 6094c1414c8SBarry Smith ii = a->i; 6104c1414c8SBarry Smith 6114c1414c8SBarry Smith for (i = 0,row = 0; i< node_max; ++i) { 6124c1414c8SBarry Smith nsz = ns[i]; 6134c1414c8SBarry Smith n = ii[1] - ii[0]; 6144c1414c8SBarry Smith ii += nsz; 6154c1414c8SBarry Smith sz = n; /* No of non zeros in this row */ 6164c1414c8SBarry Smith /* Switch on the size of Node */ 6174c1414c8SBarry Smith switch (nsz) { /* Each loop in 'case' is unrolled */ 6184c1414c8SBarry Smith case 1: 6194c1414c8SBarry Smith sum1 = *zt++; 6204c1414c8SBarry Smith 6214c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 6224c1414c8SBarry Smith i1 = idx[0]; /* The instructions are ordered to */ 6234c1414c8SBarry Smith i2 = idx[1]; /* make the compiler's job easy */ 6244c1414c8SBarry Smith idx += 2; 6254c1414c8SBarry Smith tmp0 = x[i1]; 6264c1414c8SBarry Smith tmp1 = x[i2]; 6274c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 6284c1414c8SBarry Smith } 6294c1414c8SBarry Smith 6304c1414c8SBarry Smith if (n == sz-1) { /* Take care of the last nonzero */ 6314c1414c8SBarry Smith tmp0 = x[*idx++]; 6324c1414c8SBarry Smith sum1 += *v1++ * tmp0; 6334c1414c8SBarry Smith } 6344c1414c8SBarry Smith y[row++]=sum1; 6354c1414c8SBarry Smith break; 6364c1414c8SBarry Smith case 2: 6374c1414c8SBarry Smith sum1 = *zt++; 6384c1414c8SBarry Smith sum2 = *zt++; 6394c1414c8SBarry Smith v2 = v1 + n; 6404c1414c8SBarry Smith 6414c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 6424c1414c8SBarry Smith i1 = idx[0]; 6434c1414c8SBarry Smith i2 = idx[1]; 6444c1414c8SBarry Smith idx += 2; 6454c1414c8SBarry Smith tmp0 = x[i1]; 6464c1414c8SBarry Smith tmp1 = x[i2]; 6474c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 6484c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 6494c1414c8SBarry Smith } 6504c1414c8SBarry Smith if (n == sz-1) { 6514c1414c8SBarry Smith tmp0 = x[*idx++]; 6524c1414c8SBarry Smith sum1 += *v1++ * tmp0; 6534c1414c8SBarry Smith sum2 += *v2++ * tmp0; 6544c1414c8SBarry Smith } 6554c1414c8SBarry Smith y[row++]=sum1; 6564c1414c8SBarry Smith y[row++]=sum2; 6574c1414c8SBarry Smith v1 =v2; /* Since the next block to be processed starts there*/ 6584c1414c8SBarry Smith idx +=sz; 6594c1414c8SBarry Smith break; 6604c1414c8SBarry Smith case 3: 6614c1414c8SBarry Smith sum1 = *zt++; 6624c1414c8SBarry Smith sum2 = *zt++; 6634c1414c8SBarry Smith sum3 = *zt++; 6644c1414c8SBarry Smith v2 = v1 + n; 6654c1414c8SBarry Smith v3 = v2 + n; 6664c1414c8SBarry Smith 6674c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 6684c1414c8SBarry Smith i1 = idx[0]; 6694c1414c8SBarry Smith i2 = idx[1]; 6704c1414c8SBarry Smith idx += 2; 6714c1414c8SBarry Smith tmp0 = x[i1]; 6724c1414c8SBarry Smith tmp1 = x[i2]; 6734c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 6744c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 6754c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 6764c1414c8SBarry Smith } 6774c1414c8SBarry Smith if (n == sz-1) { 6784c1414c8SBarry Smith tmp0 = x[*idx++]; 6794c1414c8SBarry Smith sum1 += *v1++ * tmp0; 6804c1414c8SBarry Smith sum2 += *v2++ * tmp0; 6814c1414c8SBarry Smith sum3 += *v3++ * tmp0; 6824c1414c8SBarry Smith } 6834c1414c8SBarry Smith y[row++]=sum1; 6844c1414c8SBarry Smith y[row++]=sum2; 6854c1414c8SBarry Smith y[row++]=sum3; 6864c1414c8SBarry Smith v1 =v3; /* Since the next block to be processed starts there*/ 6874c1414c8SBarry Smith idx +=2*sz; 6884c1414c8SBarry Smith break; 6894c1414c8SBarry Smith case 4: 6904c1414c8SBarry Smith sum1 = *zt++; 6914c1414c8SBarry Smith sum2 = *zt++; 6924c1414c8SBarry Smith sum3 = *zt++; 6934c1414c8SBarry Smith sum4 = *zt++; 6944c1414c8SBarry Smith v2 = v1 + n; 6954c1414c8SBarry Smith v3 = v2 + n; 6964c1414c8SBarry Smith v4 = v3 + n; 6974c1414c8SBarry Smith 6984c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 6994c1414c8SBarry Smith i1 = idx[0]; 7004c1414c8SBarry Smith i2 = idx[1]; 7014c1414c8SBarry Smith idx += 2; 7024c1414c8SBarry Smith tmp0 = x[i1]; 7034c1414c8SBarry Smith tmp1 = x[i2]; 7044c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 7054c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 7064c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 7074c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 7084c1414c8SBarry Smith } 7094c1414c8SBarry Smith if (n == sz-1) { 7104c1414c8SBarry Smith tmp0 = x[*idx++]; 7114c1414c8SBarry Smith sum1 += *v1++ * tmp0; 7124c1414c8SBarry Smith sum2 += *v2++ * tmp0; 7134c1414c8SBarry Smith sum3 += *v3++ * tmp0; 7144c1414c8SBarry Smith sum4 += *v4++ * tmp0; 7154c1414c8SBarry Smith } 7164c1414c8SBarry Smith y[row++]=sum1; 7174c1414c8SBarry Smith y[row++]=sum2; 7184c1414c8SBarry Smith y[row++]=sum3; 7194c1414c8SBarry Smith y[row++]=sum4; 7204c1414c8SBarry Smith v1 =v4; /* Since the next block to be processed starts there*/ 7214c1414c8SBarry Smith idx +=3*sz; 7224c1414c8SBarry Smith break; 7234c1414c8SBarry Smith case 5: 7244c1414c8SBarry Smith sum1 = *zt++; 7254c1414c8SBarry Smith sum2 = *zt++; 7264c1414c8SBarry Smith sum3 = *zt++; 7274c1414c8SBarry Smith sum4 = *zt++; 7284c1414c8SBarry Smith sum5 = *zt++; 7294c1414c8SBarry Smith v2 = v1 + n; 7304c1414c8SBarry Smith v3 = v2 + n; 7314c1414c8SBarry Smith v4 = v3 + n; 7324c1414c8SBarry Smith v5 = v4 + n; 7334c1414c8SBarry Smith 7344c1414c8SBarry Smith for (n = 0; n<sz-1; n+=2) { 7354c1414c8SBarry Smith i1 = idx[0]; 7364c1414c8SBarry Smith i2 = idx[1]; 7374c1414c8SBarry Smith idx += 2; 7384c1414c8SBarry Smith tmp0 = x[i1]; 7394c1414c8SBarry Smith tmp1 = x[i2]; 7404c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 7414c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 7424c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 7434c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 7444c1414c8SBarry Smith sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2; 7454c1414c8SBarry Smith } 7464c1414c8SBarry Smith if (n == sz-1) { 7474c1414c8SBarry Smith tmp0 = x[*idx++]; 7484c1414c8SBarry Smith sum1 += *v1++ * tmp0; 7494c1414c8SBarry Smith sum2 += *v2++ * tmp0; 7504c1414c8SBarry Smith sum3 += *v3++ * tmp0; 7514c1414c8SBarry Smith sum4 += *v4++ * tmp0; 7524c1414c8SBarry Smith sum5 += *v5++ * tmp0; 7534c1414c8SBarry Smith } 7544c1414c8SBarry Smith y[row++]=sum1; 7554c1414c8SBarry Smith y[row++]=sum2; 7564c1414c8SBarry Smith y[row++]=sum3; 7574c1414c8SBarry Smith y[row++]=sum4; 7584c1414c8SBarry Smith y[row++]=sum5; 7594c1414c8SBarry Smith v1 =v5; /* Since the next block to be processed starts there */ 7604c1414c8SBarry Smith idx +=4*sz; 7614c1414c8SBarry Smith break; 7624c1414c8SBarry Smith default: 763e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported"); 7644c1414c8SBarry Smith } 7654c1414c8SBarry Smith } 7668758e1faSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 7674c1414c8SBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 7684c1414c8SBarry Smith if (zz != yy) { 7694c1414c8SBarry Smith ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 7704c1414c8SBarry Smith } 771dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 7724c1414c8SBarry Smith PetscFunctionReturn(0); 7734c1414c8SBarry Smith } 7744c1414c8SBarry Smith 7754c1414c8SBarry Smith /* ----------------------------------------------------------- */ 7764c1414c8SBarry Smith #undef __FUNCT__ 777ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_Inode_inplace" 778ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_Inode_inplace(Mat A,Vec bb,Vec xx) 7794c1414c8SBarry Smith { 7804c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 7814c1414c8SBarry Smith IS iscol = a->col,isrow = a->row; 7824c1414c8SBarry Smith PetscErrorCode ierr; 7835d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 7848758e1faSBarry Smith PetscInt i,j,n = A->rmap->n,nz; 7858758e1faSBarry Smith PetscInt node_max,*ns,row,nsz,aii,i0,i1; 7868758e1faSBarry Smith const PetscInt *ai = a->i,*a_j = a->j,*vi,*ad,*aj; 787d9fead3dSBarry Smith PetscScalar *x,*tmp,*tmps,tmp0,tmp1; 788d9fead3dSBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5; 789dd6ea824SBarry Smith const MatScalar *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa; 790dd6ea824SBarry Smith const PetscScalar *b; 7914c1414c8SBarry Smith 7924c1414c8SBarry Smith PetscFunctionBegin; 793e32f2f54SBarry Smith if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 7944c1414c8SBarry Smith node_max = a->inode.node_count; 7954c1414c8SBarry Smith ns = a->inode.size; /* Node Size array */ 7964c1414c8SBarry Smith 7973649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 7984c1414c8SBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 7994c1414c8SBarry Smith tmp = a->solve_work; 8004c1414c8SBarry Smith 8014c1414c8SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 8024c1414c8SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 8034c1414c8SBarry Smith 8044c1414c8SBarry Smith /* forward solve the lower triangular */ 8054c1414c8SBarry Smith tmps = tmp; 8064c1414c8SBarry Smith aa = a_a; 8074c1414c8SBarry Smith aj = a_j; 8084c1414c8SBarry Smith ad = a->diag; 8094c1414c8SBarry Smith 8104c1414c8SBarry Smith for (i = 0,row = 0; i< node_max; ++i) { 8114c1414c8SBarry Smith nsz = ns[i]; 8124c1414c8SBarry Smith aii = ai[row]; 8134c1414c8SBarry Smith v1 = aa + aii; 8144c1414c8SBarry Smith vi = aj + aii; 8154c1414c8SBarry Smith nz = ad[row]- aii; 81626549573SJed Brown if (i < node_max-1) { 81726549573SJed Brown /* Prefetch the block after the current one, the prefetch itself can't cause a memory error, 81826549573SJed Brown * but our indexing to determine it's size could. */ 81950d8bf02SJed Brown PetscPrefetchBlock(aj+ai[row+nsz],ad[row+nsz]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); /* indices */ 82026549573SJed Brown /* In my tests, it seems to be better to fetch entire rows instead of just the lower-triangular part */ 82150d8bf02SJed Brown PetscPrefetchBlock(aa+ai[row+nsz],ad[row+nsz+ns[i+1]-1]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); 82226549573SJed Brown /* for (j=0; j<ns[i+1]; j++) PetscPrefetchBlock(aa+ai[row+nsz+j],ad[row+nsz+j]-ai[row+nsz+j],0,0); */ 82326549573SJed Brown } 8244c1414c8SBarry Smith 8254c1414c8SBarry Smith switch (nsz) { /* Each loop in 'case' is unrolled */ 8264c1414c8SBarry Smith case 1: 8274c1414c8SBarry Smith sum1 = b[*r++]; 8284c1414c8SBarry Smith for (j=0; j<nz-1; j+=2) { 8294c1414c8SBarry Smith i0 = vi[0]; 8304c1414c8SBarry Smith i1 = vi[1]; 8314c1414c8SBarry Smith vi +=2; 8324c1414c8SBarry Smith tmp0 = tmps[i0]; 8334c1414c8SBarry Smith tmp1 = tmps[i1]; 8344c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 8354c1414c8SBarry Smith } 8364c1414c8SBarry Smith if (j == nz-1) { 8374c1414c8SBarry Smith tmp0 = tmps[*vi++]; 8384c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 8394c1414c8SBarry Smith } 8404c1414c8SBarry Smith tmp[row++]=sum1; 8414c1414c8SBarry Smith break; 8424c1414c8SBarry Smith case 2: 8434c1414c8SBarry Smith sum1 = b[*r++]; 8444c1414c8SBarry Smith sum2 = b[*r++]; 8454c1414c8SBarry Smith v2 = aa + ai[row+1]; 8464c1414c8SBarry Smith 8474c1414c8SBarry Smith for (j=0; j<nz-1; j+=2) { 8484c1414c8SBarry Smith i0 = vi[0]; 8494c1414c8SBarry Smith i1 = vi[1]; 8504c1414c8SBarry Smith vi +=2; 8514c1414c8SBarry Smith tmp0 = tmps[i0]; 8524c1414c8SBarry Smith tmp1 = tmps[i1]; 8534c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 8544c1414c8SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 8554c1414c8SBarry Smith } 8564c1414c8SBarry Smith if (j == nz-1) { 8574c1414c8SBarry Smith tmp0 = tmps[*vi++]; 8584c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 8594c1414c8SBarry Smith sum2 -= *v2++ *tmp0; 8604c1414c8SBarry Smith } 8614c1414c8SBarry Smith sum2 -= *v2++ *sum1; 8624c1414c8SBarry Smith tmp[row++]=sum1; 8634c1414c8SBarry Smith tmp[row++]=sum2; 8644c1414c8SBarry Smith break; 8654c1414c8SBarry Smith case 3: 8664c1414c8SBarry Smith sum1 = b[*r++]; 8674c1414c8SBarry Smith sum2 = b[*r++]; 8684c1414c8SBarry Smith sum3 = b[*r++]; 8694c1414c8SBarry Smith v2 = aa + ai[row+1]; 8704c1414c8SBarry Smith v3 = aa + ai[row+2]; 8714c1414c8SBarry Smith 8724c1414c8SBarry Smith for (j=0; j<nz-1; j+=2) { 8734c1414c8SBarry Smith i0 = vi[0]; 8744c1414c8SBarry Smith i1 = vi[1]; 8754c1414c8SBarry Smith vi +=2; 8764c1414c8SBarry Smith tmp0 = tmps[i0]; 8774c1414c8SBarry Smith tmp1 = tmps[i1]; 8784c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 8794c1414c8SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 8804c1414c8SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 8814c1414c8SBarry Smith } 8824c1414c8SBarry Smith if (j == nz-1) { 8834c1414c8SBarry Smith tmp0 = tmps[*vi++]; 8844c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 8854c1414c8SBarry Smith sum2 -= *v2++ *tmp0; 8864c1414c8SBarry Smith sum3 -= *v3++ *tmp0; 8874c1414c8SBarry Smith } 8884c1414c8SBarry Smith sum2 -= *v2++ * sum1; 8894c1414c8SBarry Smith sum3 -= *v3++ * sum1; 8904c1414c8SBarry Smith sum3 -= *v3++ * sum2; 8912205254eSKarl Rupp 8924c1414c8SBarry Smith tmp[row++]=sum1; 8934c1414c8SBarry Smith tmp[row++]=sum2; 8944c1414c8SBarry Smith tmp[row++]=sum3; 8954c1414c8SBarry Smith break; 8964c1414c8SBarry Smith 8974c1414c8SBarry Smith case 4: 8984c1414c8SBarry Smith sum1 = b[*r++]; 8994c1414c8SBarry Smith sum2 = b[*r++]; 9004c1414c8SBarry Smith sum3 = b[*r++]; 9014c1414c8SBarry Smith sum4 = b[*r++]; 9024c1414c8SBarry Smith v2 = aa + ai[row+1]; 9034c1414c8SBarry Smith v3 = aa + ai[row+2]; 9044c1414c8SBarry Smith v4 = aa + ai[row+3]; 9054c1414c8SBarry Smith 9064c1414c8SBarry Smith for (j=0; j<nz-1; j+=2) { 9074c1414c8SBarry Smith i0 = vi[0]; 9084c1414c8SBarry Smith i1 = vi[1]; 9094c1414c8SBarry Smith vi +=2; 9104c1414c8SBarry Smith tmp0 = tmps[i0]; 9114c1414c8SBarry Smith tmp1 = tmps[i1]; 9124c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 9134c1414c8SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 9144c1414c8SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 9154c1414c8SBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 9164c1414c8SBarry Smith } 9174c1414c8SBarry Smith if (j == nz-1) { 9184c1414c8SBarry Smith tmp0 = tmps[*vi++]; 9194c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 9204c1414c8SBarry Smith sum2 -= *v2++ *tmp0; 9214c1414c8SBarry Smith sum3 -= *v3++ *tmp0; 9224c1414c8SBarry Smith sum4 -= *v4++ *tmp0; 9234c1414c8SBarry Smith } 9244c1414c8SBarry Smith sum2 -= *v2++ * sum1; 9254c1414c8SBarry Smith sum3 -= *v3++ * sum1; 9264c1414c8SBarry Smith sum4 -= *v4++ * sum1; 9274c1414c8SBarry Smith sum3 -= *v3++ * sum2; 9284c1414c8SBarry Smith sum4 -= *v4++ * sum2; 9294c1414c8SBarry Smith sum4 -= *v4++ * sum3; 9304c1414c8SBarry Smith 9314c1414c8SBarry Smith tmp[row++]=sum1; 9324c1414c8SBarry Smith tmp[row++]=sum2; 9334c1414c8SBarry Smith tmp[row++]=sum3; 9344c1414c8SBarry Smith tmp[row++]=sum4; 9354c1414c8SBarry Smith break; 9364c1414c8SBarry Smith case 5: 9374c1414c8SBarry Smith sum1 = b[*r++]; 9384c1414c8SBarry Smith sum2 = b[*r++]; 9394c1414c8SBarry Smith sum3 = b[*r++]; 9404c1414c8SBarry Smith sum4 = b[*r++]; 9414c1414c8SBarry Smith sum5 = b[*r++]; 9424c1414c8SBarry Smith v2 = aa + ai[row+1]; 9434c1414c8SBarry Smith v3 = aa + ai[row+2]; 9444c1414c8SBarry Smith v4 = aa + ai[row+3]; 9454c1414c8SBarry Smith v5 = aa + ai[row+4]; 9464c1414c8SBarry Smith 9474c1414c8SBarry Smith for (j=0; j<nz-1; j+=2) { 9484c1414c8SBarry Smith i0 = vi[0]; 9494c1414c8SBarry Smith i1 = vi[1]; 9504c1414c8SBarry Smith vi +=2; 9514c1414c8SBarry Smith tmp0 = tmps[i0]; 9524c1414c8SBarry Smith tmp1 = tmps[i1]; 9534c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 9544c1414c8SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 9554c1414c8SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 9564c1414c8SBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 9574c1414c8SBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 9584c1414c8SBarry Smith } 9594c1414c8SBarry Smith if (j == nz-1) { 9604c1414c8SBarry Smith tmp0 = tmps[*vi++]; 9614c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 9624c1414c8SBarry Smith sum2 -= *v2++ *tmp0; 9634c1414c8SBarry Smith sum3 -= *v3++ *tmp0; 9644c1414c8SBarry Smith sum4 -= *v4++ *tmp0; 9654c1414c8SBarry Smith sum5 -= *v5++ *tmp0; 9664c1414c8SBarry Smith } 9674c1414c8SBarry Smith 9684c1414c8SBarry Smith sum2 -= *v2++ * sum1; 9694c1414c8SBarry Smith sum3 -= *v3++ * sum1; 9704c1414c8SBarry Smith sum4 -= *v4++ * sum1; 9714c1414c8SBarry Smith sum5 -= *v5++ * sum1; 9724c1414c8SBarry Smith sum3 -= *v3++ * sum2; 9734c1414c8SBarry Smith sum4 -= *v4++ * sum2; 9744c1414c8SBarry Smith sum5 -= *v5++ * sum2; 9754c1414c8SBarry Smith sum4 -= *v4++ * sum3; 9764c1414c8SBarry Smith sum5 -= *v5++ * sum3; 9774c1414c8SBarry Smith sum5 -= *v5++ * sum4; 9784c1414c8SBarry Smith 9794c1414c8SBarry Smith tmp[row++]=sum1; 9804c1414c8SBarry Smith tmp[row++]=sum2; 9814c1414c8SBarry Smith tmp[row++]=sum3; 9824c1414c8SBarry Smith tmp[row++]=sum4; 9834c1414c8SBarry Smith tmp[row++]=sum5; 9844c1414c8SBarry Smith break; 9854c1414c8SBarry Smith default: 986e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n"); 9874c1414c8SBarry Smith } 9884c1414c8SBarry Smith } 9894c1414c8SBarry Smith /* backward solve the upper triangular */ 9904c1414c8SBarry Smith for (i=node_max -1,row = n-1; i>=0; i--) { 9914c1414c8SBarry Smith nsz = ns[i]; 9924c1414c8SBarry Smith aii = ai[row+1] -1; 9934c1414c8SBarry Smith v1 = aa + aii; 9944c1414c8SBarry Smith vi = aj + aii; 9954c1414c8SBarry Smith nz = aii- ad[row]; 9964c1414c8SBarry Smith switch (nsz) { /* Each loop in 'case' is unrolled */ 9974c1414c8SBarry Smith case 1: 9984c1414c8SBarry Smith sum1 = tmp[row]; 9994c1414c8SBarry Smith 10004c1414c8SBarry Smith for (j=nz; j>1; j-=2) { 10014c1414c8SBarry Smith vi -=2; 10024c1414c8SBarry Smith i0 = vi[2]; 10034c1414c8SBarry Smith i1 = vi[1]; 10044c1414c8SBarry Smith tmp0 = tmps[i0]; 10054c1414c8SBarry Smith tmp1 = tmps[i1]; 10064c1414c8SBarry Smith v1 -= 2; 10074c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 10084c1414c8SBarry Smith } 10094c1414c8SBarry Smith if (j==1) { 10104c1414c8SBarry Smith tmp0 = tmps[*vi--]; 10114c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 10124c1414c8SBarry Smith } 10134c1414c8SBarry Smith x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 10144c1414c8SBarry Smith break; 10154c1414c8SBarry Smith case 2: 10164c1414c8SBarry Smith sum1 = tmp[row]; 10174c1414c8SBarry Smith sum2 = tmp[row -1]; 10184c1414c8SBarry Smith v2 = aa + ai[row]-1; 10194c1414c8SBarry Smith for (j=nz; j>1; j-=2) { 10204c1414c8SBarry Smith vi -=2; 10214c1414c8SBarry Smith i0 = vi[2]; 10224c1414c8SBarry Smith i1 = vi[1]; 10234c1414c8SBarry Smith tmp0 = tmps[i0]; 10244c1414c8SBarry Smith tmp1 = tmps[i1]; 10254c1414c8SBarry Smith v1 -= 2; 10264c1414c8SBarry Smith v2 -= 2; 10274c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 10284c1414c8SBarry Smith sum2 -= v2[2] * tmp0 + v2[1] * tmp1; 10294c1414c8SBarry Smith } 10304c1414c8SBarry Smith if (j==1) { 10314c1414c8SBarry Smith tmp0 = tmps[*vi--]; 10324c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 10334c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 10344c1414c8SBarry Smith } 10354c1414c8SBarry Smith 10364c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 10374c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 10384c1414c8SBarry Smith x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--; 10394c1414c8SBarry Smith break; 10404c1414c8SBarry Smith case 3: 10414c1414c8SBarry Smith sum1 = tmp[row]; 10424c1414c8SBarry Smith sum2 = tmp[row -1]; 10434c1414c8SBarry Smith sum3 = tmp[row -2]; 10444c1414c8SBarry Smith v2 = aa + ai[row]-1; 10454c1414c8SBarry Smith v3 = aa + ai[row -1]-1; 10464c1414c8SBarry Smith for (j=nz; j>1; j-=2) { 10474c1414c8SBarry Smith vi -=2; 10484c1414c8SBarry Smith i0 = vi[2]; 10494c1414c8SBarry Smith i1 = vi[1]; 10504c1414c8SBarry Smith tmp0 = tmps[i0]; 10514c1414c8SBarry Smith tmp1 = tmps[i1]; 10524c1414c8SBarry Smith v1 -= 2; 10534c1414c8SBarry Smith v2 -= 2; 10544c1414c8SBarry Smith v3 -= 2; 10554c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 10564c1414c8SBarry Smith sum2 -= v2[2] * tmp0 + v2[1] * tmp1; 10574c1414c8SBarry Smith sum3 -= v3[2] * tmp0 + v3[1] * tmp1; 10584c1414c8SBarry Smith } 10594c1414c8SBarry Smith if (j==1) { 10604c1414c8SBarry Smith tmp0 = tmps[*vi--]; 10614c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 10624c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 10634c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 10644c1414c8SBarry Smith } 10654c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 10664c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 10674c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 10684c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--; 10694c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 10704c1414c8SBarry Smith x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--; 10714c1414c8SBarry Smith 10724c1414c8SBarry Smith break; 10734c1414c8SBarry Smith case 4: 10744c1414c8SBarry Smith sum1 = tmp[row]; 10754c1414c8SBarry Smith sum2 = tmp[row -1]; 10764c1414c8SBarry Smith sum3 = tmp[row -2]; 10774c1414c8SBarry Smith sum4 = tmp[row -3]; 10784c1414c8SBarry Smith v2 = aa + ai[row]-1; 10794c1414c8SBarry Smith v3 = aa + ai[row -1]-1; 10804c1414c8SBarry Smith v4 = aa + ai[row -2]-1; 10814c1414c8SBarry Smith 10824c1414c8SBarry Smith for (j=nz; j>1; j-=2) { 10834c1414c8SBarry Smith vi -=2; 10844c1414c8SBarry Smith i0 = vi[2]; 10854c1414c8SBarry Smith i1 = vi[1]; 10864c1414c8SBarry Smith tmp0 = tmps[i0]; 10874c1414c8SBarry Smith tmp1 = tmps[i1]; 10884c1414c8SBarry Smith v1 -= 2; 10894c1414c8SBarry Smith v2 -= 2; 10904c1414c8SBarry Smith v3 -= 2; 10914c1414c8SBarry Smith v4 -= 2; 10924c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 10934c1414c8SBarry Smith sum2 -= v2[2] * tmp0 + v2[1] * tmp1; 10944c1414c8SBarry Smith sum3 -= v3[2] * tmp0 + v3[1] * tmp1; 10954c1414c8SBarry Smith sum4 -= v4[2] * tmp0 + v4[1] * tmp1; 10964c1414c8SBarry Smith } 10974c1414c8SBarry Smith if (j==1) { 10984c1414c8SBarry Smith tmp0 = tmps[*vi--]; 10994c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 11004c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 11014c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11024c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11034c1414c8SBarry Smith } 11044c1414c8SBarry Smith 11054c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 11064c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 11074c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11084c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11094c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--; 11104c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11114c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11124c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--; 11134c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11144c1414c8SBarry Smith x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--; 11154c1414c8SBarry Smith break; 11164c1414c8SBarry Smith case 5: 11174c1414c8SBarry Smith sum1 = tmp[row]; 11184c1414c8SBarry Smith sum2 = tmp[row -1]; 11194c1414c8SBarry Smith sum3 = tmp[row -2]; 11204c1414c8SBarry Smith sum4 = tmp[row -3]; 11214c1414c8SBarry Smith sum5 = tmp[row -4]; 11224c1414c8SBarry Smith v2 = aa + ai[row]-1; 11234c1414c8SBarry Smith v3 = aa + ai[row -1]-1; 11244c1414c8SBarry Smith v4 = aa + ai[row -2]-1; 11254c1414c8SBarry Smith v5 = aa + ai[row -3]-1; 11264c1414c8SBarry Smith for (j=nz; j>1; j-=2) { 11274c1414c8SBarry Smith vi -= 2; 11284c1414c8SBarry Smith i0 = vi[2]; 11294c1414c8SBarry Smith i1 = vi[1]; 11304c1414c8SBarry Smith tmp0 = tmps[i0]; 11314c1414c8SBarry Smith tmp1 = tmps[i1]; 11324c1414c8SBarry Smith v1 -= 2; 11334c1414c8SBarry Smith v2 -= 2; 11344c1414c8SBarry Smith v3 -= 2; 11354c1414c8SBarry Smith v4 -= 2; 11364c1414c8SBarry Smith v5 -= 2; 11374c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 11384c1414c8SBarry Smith sum2 -= v2[2] * tmp0 + v2[1] * tmp1; 11394c1414c8SBarry Smith sum3 -= v3[2] * tmp0 + v3[1] * tmp1; 11404c1414c8SBarry Smith sum4 -= v4[2] * tmp0 + v4[1] * tmp1; 11414c1414c8SBarry Smith sum5 -= v5[2] * tmp0 + v5[1] * tmp1; 11424c1414c8SBarry Smith } 11434c1414c8SBarry Smith if (j==1) { 11444c1414c8SBarry Smith tmp0 = tmps[*vi--]; 11454c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 11464c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 11474c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11484c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11494c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11504c1414c8SBarry Smith } 11514c1414c8SBarry Smith 11524c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 11534c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 11544c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11554c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11564c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11574c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--; 11584c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11594c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11604c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11614c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--; 11624c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11634c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11644c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--; 11654c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11664c1414c8SBarry Smith x[*c--] = tmp[row] = sum5*a_a[ad[row]]; row--; 11674c1414c8SBarry Smith break; 11684c1414c8SBarry Smith default: 1169e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n"); 11704c1414c8SBarry Smith } 11714c1414c8SBarry Smith } 11724c1414c8SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 11734c1414c8SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 11743649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 11754c1414c8SBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1176dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 11774c1414c8SBarry Smith PetscFunctionReturn(0); 11784c1414c8SBarry Smith } 11794c1414c8SBarry Smith 118028f1b45aSHong Zhang #undef __FUNCT__ 118128f1b45aSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_Inode" 118228f1b45aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_Inode(Mat B,Mat A,const MatFactorInfo *info) 118328f1b45aSHong Zhang { 118428f1b45aSHong Zhang Mat C =B; 118528f1b45aSHong Zhang Mat_SeqAIJ *a =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data; 118628f1b45aSHong Zhang IS isrow = b->row,isicol = b->icol; 118728f1b45aSHong Zhang PetscErrorCode ierr; 118828f1b45aSHong Zhang const PetscInt *r,*ic,*ics; 118928f1b45aSHong Zhang const PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag; 119028f1b45aSHong Zhang PetscInt i,j,k,nz,nzL,row,*pj; 119128f1b45aSHong Zhang const PetscInt *ajtmp,*bjtmp; 11929877982aSShri Abhyankar MatScalar *pc,*pc1,*pc2,*pc3,*pc4,mul1,mul2,mul3,mul4,*pv,*rtmp1,*rtmp2,*rtmp3,*rtmp4; 11939877982aSShri Abhyankar const MatScalar *aa=a->a,*v,*v1,*v2,*v3,*v4; 119428f1b45aSHong Zhang FactorShiftCtx sctx; 11954f81c4b7SBarry Smith const PetscInt *ddiag; 119628f1b45aSHong Zhang PetscReal rs; 119728f1b45aSHong Zhang MatScalar d; 11984f81c4b7SBarry Smith PetscInt inod,nodesz,node_max,col; 11994f81c4b7SBarry Smith const PetscInt *ns; 120007b50cabSHong Zhang PetscInt *tmp_vec1,*tmp_vec2,*nsmap; 12010e95ead3SHong Zhang 120228f1b45aSHong Zhang PetscFunctionBegin; 120328f1b45aSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 120428f1b45aSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 120528f1b45aSHong Zhang 1206f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 120728f1b45aSHong Zhang ddiag = a->diag; 120828f1b45aSHong Zhang sctx.shift_top = info->zeropivot; 120928f1b45aSHong Zhang for (i=0; i<n; i++) { 121028f1b45aSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 121128f1b45aSHong Zhang d = (aa)[ddiag[i]]; 121228f1b45aSHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 121328f1b45aSHong Zhang v = aa+ai[i]; 121428f1b45aSHong Zhang nz = ai[i+1] - ai[i]; 12152205254eSKarl Rupp for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]); 121628f1b45aSHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 121728f1b45aSHong Zhang } 121828f1b45aSHong Zhang sctx.shift_top *= 1.1; 121928f1b45aSHong Zhang sctx.nshift_max = 5; 122028f1b45aSHong Zhang sctx.shift_lo = 0.; 122128f1b45aSHong Zhang sctx.shift_hi = 1.; 122228f1b45aSHong Zhang } 122328f1b45aSHong Zhang 122428f1b45aSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 122528f1b45aSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 122668785679SHong Zhang 12271795a4d1SJed Brown ierr = PetscCalloc4(n,&rtmp1,n,&rtmp2,n,&rtmp3,n,&rtmp4);CHKERRQ(ierr); 122828f1b45aSHong Zhang ics = ic; 122928f1b45aSHong Zhang 123028f1b45aSHong Zhang node_max = a->inode.node_count; 123128f1b45aSHong Zhang ns = a->inode.size; 1232e7e72b3dSBarry Smith if (!ns) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix without inode information"); 123328f1b45aSHong Zhang 12349877982aSShri Abhyankar /* If max inode size > 4, split it into two inodes.*/ 123568785679SHong Zhang /* also map the inode sizes according to the ordering */ 1236854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&tmp_vec1);CHKERRQ(ierr); 123768785679SHong Zhang for (i=0,j=0; i<node_max; ++i,++j) { 1238b1550197SShri Abhyankar if (ns[i] > 4) { 1239048b5e81SShri Abhyankar tmp_vec1[j] = 4; 124068785679SHong Zhang ++j; 124168785679SHong Zhang tmp_vec1[j] = ns[i] - tmp_vec1[j-1]; 124268785679SHong Zhang } else { 124368785679SHong Zhang tmp_vec1[j] = ns[i]; 124468785679SHong Zhang } 124568785679SHong Zhang } 124668785679SHong Zhang /* Use the correct node_max */ 124768785679SHong Zhang node_max = j; 124868785679SHong Zhang 124968785679SHong Zhang /* Now reorder the inode info based on mat re-ordering info */ 125068785679SHong Zhang /* First create a row -> inode_size_array_index map */ 1251854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&nsmap);CHKERRQ(ierr); 1252854ce69bSBarry Smith ierr = PetscMalloc1(node_max+1,&tmp_vec2);CHKERRQ(ierr); 125368785679SHong Zhang for (i=0,row=0; i<node_max; i++) { 125468785679SHong Zhang nodesz = tmp_vec1[i]; 125568785679SHong Zhang for (j=0; j<nodesz; j++,row++) { 125668785679SHong Zhang nsmap[row] = i; 125768785679SHong Zhang } 125868785679SHong Zhang } 125968785679SHong Zhang /* Using nsmap, create a reordered ns structure */ 126068785679SHong Zhang for (i=0,j=0; i< node_max; i++) { 126168785679SHong Zhang nodesz = tmp_vec1[nsmap[r[j]]]; /* here the reordered row_no is in r[] */ 126268785679SHong Zhang tmp_vec2[i] = nodesz; 126368785679SHong Zhang j += nodesz; 126468785679SHong Zhang } 126568785679SHong Zhang ierr = PetscFree(nsmap);CHKERRQ(ierr); 126668785679SHong Zhang ierr = PetscFree(tmp_vec1);CHKERRQ(ierr); 1267b89f182dSHong Zhang 126868785679SHong Zhang /* Now use the correct ns */ 126968785679SHong Zhang ns = tmp_vec2; 127068785679SHong Zhang 127128f1b45aSHong Zhang do { 127207b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 127328f1b45aSHong Zhang /* Now loop over each block-row, and do the factorization */ 127428f1b45aSHong Zhang for (inod=0,i=0; inod<node_max; inod++) { /* i: row index; inod: inode index */ 127528f1b45aSHong Zhang nodesz = ns[inod]; 127628f1b45aSHong Zhang 127728f1b45aSHong Zhang switch (nodesz) { 127828f1b45aSHong Zhang case 1: 127968785679SHong Zhang /*----------*/ 1280b89f182dSHong Zhang /* zero rtmp1 */ 128128f1b45aSHong Zhang /* L part */ 128228f1b45aSHong Zhang nz = bi[i+1] - bi[i]; 128328f1b45aSHong Zhang bjtmp = bj + bi[i]; 1284b89f182dSHong Zhang for (j=0; j<nz; j++) rtmp1[bjtmp[j]] = 0.0; 128528f1b45aSHong Zhang 128628f1b45aSHong Zhang /* U part */ 128728f1b45aSHong Zhang nz = bdiag[i]-bdiag[i+1]; 128828f1b45aSHong Zhang bjtmp = bj + bdiag[i+1]+1; 1289b89f182dSHong Zhang for (j=0; j<nz; j++) rtmp1[bjtmp[j]] = 0.0; 129028f1b45aSHong Zhang 129128f1b45aSHong Zhang /* load in initial (unfactored row) */ 129228f1b45aSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 129328f1b45aSHong Zhang ajtmp = aj + ai[r[i]]; 129428f1b45aSHong Zhang v = aa + ai[r[i]]; 12952205254eSKarl Rupp for (j=0; j<nz; j++) rtmp1[ics[ajtmp[j]]] = v[j]; 12962205254eSKarl Rupp 129728f1b45aSHong Zhang /* ZeropivotApply() */ 1298b89f182dSHong Zhang rtmp1[i] += sctx.shift_amount; /* shift the diagonal of the matrix */ 129928f1b45aSHong Zhang 130028f1b45aSHong Zhang /* elimination */ 130128f1b45aSHong Zhang bjtmp = bj + bi[i]; 130228f1b45aSHong Zhang row = *bjtmp++; 130328f1b45aSHong Zhang nzL = bi[i+1] - bi[i]; 130428f1b45aSHong Zhang for (k=0; k < nzL; k++) { 1305b89f182dSHong Zhang pc = rtmp1 + row; 130628f1b45aSHong Zhang if (*pc != 0.0) { 130728f1b45aSHong Zhang pv = b->a + bdiag[row]; 1308b89f182dSHong Zhang mul1 = *pc * (*pv); 1309b89f182dSHong Zhang *pc = mul1; 131028f1b45aSHong Zhang pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 131128f1b45aSHong Zhang pv = b->a + bdiag[row+1]+1; 131228f1b45aSHong Zhang nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 1313b89f182dSHong Zhang for (j=0; j<nz; j++) rtmp1[pj[j]] -= mul1 * pv[j]; 131428b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 131528f1b45aSHong Zhang } 131628f1b45aSHong Zhang row = *bjtmp++; 131728f1b45aSHong Zhang } 131828f1b45aSHong Zhang 131928f1b45aSHong Zhang /* finished row so stick it into b->a */ 132028f1b45aSHong Zhang rs = 0.0; 132128f1b45aSHong Zhang /* L part */ 132228f1b45aSHong Zhang pv = b->a + bi[i]; 132328f1b45aSHong Zhang pj = b->j + bi[i]; 132428f1b45aSHong Zhang nz = bi[i+1] - bi[i]; 132528f1b45aSHong Zhang for (j=0; j<nz; j++) { 1326b89f182dSHong Zhang pv[j] = rtmp1[pj[j]]; rs += PetscAbsScalar(pv[j]); 132728f1b45aSHong Zhang } 132828f1b45aSHong Zhang 132928f1b45aSHong Zhang /* U part */ 133028f1b45aSHong Zhang pv = b->a + bdiag[i+1]+1; 133128f1b45aSHong Zhang pj = b->j + bdiag[i+1]+1; 133228f1b45aSHong Zhang nz = bdiag[i] - bdiag[i+1]-1; 133328f1b45aSHong Zhang for (j=0; j<nz; j++) { 1334b89f182dSHong Zhang pv[j] = rtmp1[pj[j]]; rs += PetscAbsScalar(pv[j]); 133528f1b45aSHong Zhang } 133628f1b45aSHong Zhang 1337b89f182dSHong Zhang /* Check zero pivot */ 133828f1b45aSHong Zhang sctx.rs = rs; 1339b89f182dSHong Zhang sctx.pv = rtmp1[i]; 1340d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 134107b50cabSHong Zhang if (sctx.newshift) break; 134228f1b45aSHong Zhang 134328f1b45aSHong Zhang /* Mark diagonal and invert diagonal for simplier triangular solves */ 134428f1b45aSHong Zhang pv = b->a + bdiag[i]; 1345b89f182dSHong Zhang *pv = 1.0/sctx.pv; /* sctx.pv = rtmp1[i]+shiftamount if shifttype==MAT_SHIFT_INBLOCKS */ 134628f1b45aSHong Zhang break; 134728f1b45aSHong Zhang 134828f1b45aSHong Zhang case 2: 134928f1b45aSHong Zhang /*----------*/ 1350b89f182dSHong Zhang /* zero rtmp1 and rtmp2 */ 135128f1b45aSHong Zhang /* L part */ 135228f1b45aSHong Zhang nz = bi[i+1] - bi[i]; 135328f1b45aSHong Zhang bjtmp = bj + bi[i]; 135428f1b45aSHong Zhang for (j=0; j<nz; j++) { 135568785679SHong Zhang col = bjtmp[j]; 1356b89f182dSHong Zhang rtmp1[col] = 0.0; rtmp2[col] = 0.0; 135728f1b45aSHong Zhang } 135828f1b45aSHong Zhang 135928f1b45aSHong Zhang /* U part */ 136028f1b45aSHong Zhang nz = bdiag[i]-bdiag[i+1]; 136128f1b45aSHong Zhang bjtmp = bj + bdiag[i+1]+1; 136228f1b45aSHong Zhang for (j=0; j<nz; j++) { 136368785679SHong Zhang col = bjtmp[j]; 1364b89f182dSHong Zhang rtmp1[col] = 0.0; rtmp2[col] = 0.0; 136528f1b45aSHong Zhang } 136628f1b45aSHong Zhang 136728f1b45aSHong Zhang /* load in initial (unfactored row) */ 136828f1b45aSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 136928f1b45aSHong Zhang ajtmp = aj + ai[r[i]]; 137028f1b45aSHong Zhang v1 = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; 137128f1b45aSHong Zhang for (j=0; j<nz; j++) { 137268785679SHong Zhang col = ics[ajtmp[j]]; 1373b89f182dSHong Zhang rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; 137428f1b45aSHong Zhang } 137528f1b45aSHong Zhang /* ZeropivotApply(): shift the diagonal of the matrix */ 1376b89f182dSHong Zhang rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; 137728f1b45aSHong Zhang 137828f1b45aSHong Zhang /* elimination */ 137928f1b45aSHong Zhang bjtmp = bj + bi[i]; 138028f1b45aSHong Zhang row = *bjtmp++; /* pivot row */ 138128f1b45aSHong Zhang nzL = bi[i+1] - bi[i]; 138228f1b45aSHong Zhang for (k=0; k < nzL; k++) { 1383b89f182dSHong Zhang pc1 = rtmp1 + row; 1384b89f182dSHong Zhang pc2 = rtmp2 + row; 138528f1b45aSHong Zhang if (*pc1 != 0.0 || *pc2 != 0.0) { 138628f1b45aSHong Zhang pv = b->a + bdiag[row]; 1387b89f182dSHong Zhang mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); 1388b89f182dSHong Zhang *pc1 = mul1; *pc2 = mul2; 138928f1b45aSHong Zhang 139028f1b45aSHong Zhang pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 139128f1b45aSHong Zhang pv = b->a + bdiag[row+1]+1; 139228f1b45aSHong Zhang nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 139328f1b45aSHong Zhang for (j=0; j<nz; j++) { 139468785679SHong Zhang col = pj[j]; 1395b89f182dSHong Zhang rtmp1[col] -= mul1 * pv[j]; 1396b89f182dSHong Zhang rtmp2[col] -= mul2 * pv[j]; 139728f1b45aSHong Zhang } 139828b1a77fSLisandro Dalcin ierr = PetscLogFlops(2+4*nz);CHKERRQ(ierr); 139928f1b45aSHong Zhang } 140028f1b45aSHong Zhang row = *bjtmp++; 140128f1b45aSHong Zhang } 140228f1b45aSHong Zhang 1403b89f182dSHong Zhang /* finished row i; check zero pivot, then stick row i into b->a */ 140428f1b45aSHong Zhang rs = 0.0; 140528f1b45aSHong Zhang /* L part */ 1406b89f182dSHong Zhang pc1 = b->a + bi[i]; 140728f1b45aSHong Zhang pj = b->j + bi[i]; 140828f1b45aSHong Zhang nz = bi[i+1] - bi[i]; 140928f1b45aSHong Zhang for (j=0; j<nz; j++) { 141068785679SHong Zhang col = pj[j]; 14110e7a5c2bSHong Zhang pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 141228f1b45aSHong Zhang } 141328f1b45aSHong Zhang /* U part */ 1414b89f182dSHong Zhang pc1 = b->a + bdiag[i+1]+1; 141528f1b45aSHong Zhang pj = b->j + bdiag[i+1]+1; 14160e7a5c2bSHong Zhang nz = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */ 141728f1b45aSHong Zhang for (j=0; j<nz; j++) { 141868785679SHong Zhang col = pj[j]; 14190e7a5c2bSHong Zhang pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 142028f1b45aSHong Zhang } 142128f1b45aSHong Zhang 142228f1b45aSHong Zhang sctx.rs = rs; 1423b89f182dSHong Zhang sctx.pv = rtmp1[i]; 1424d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 142507b50cabSHong Zhang if (sctx.newshift) break; 1426b89f182dSHong Zhang pc1 = b->a + bdiag[i]; /* Mark diagonal */ 1427b89f182dSHong Zhang *pc1 = 1.0/sctx.pv; 1428b89f182dSHong Zhang 1429b89f182dSHong Zhang /* Now take care of diagonal 2x2 block. */ 1430b89f182dSHong Zhang pc2 = rtmp2 + i; 1431b89f182dSHong Zhang if (*pc2 != 0.0) { 1432b89f182dSHong Zhang mul1 = (*pc2)*(*pc1); /* *pc1=diag[i] is inverted! */ 1433b89f182dSHong Zhang *pc2 = mul1; /* insert L entry */ 1434b89f182dSHong Zhang pj = b->j + bdiag[i+1]+1; /* beginning of U(i,:) */ 1435b89f182dSHong Zhang nz = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */ 1436b89f182dSHong Zhang for (j=0; j<nz; j++) { 1437b89f182dSHong Zhang col = pj[j]; rtmp2[col] -= mul1 * rtmp1[col]; 143828f1b45aSHong Zhang } 143928b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 1440b89f182dSHong Zhang } 1441b89f182dSHong Zhang 1442b89f182dSHong Zhang /* finished row i+1; check zero pivot, then stick row i+1 into b->a */ 1443b89f182dSHong Zhang rs = 0.0; 1444b89f182dSHong Zhang /* L part */ 1445b89f182dSHong Zhang pc2 = b->a + bi[i+1]; 1446b89f182dSHong Zhang pj = b->j + bi[i+1]; 1447b89f182dSHong Zhang nz = bi[i+2] - bi[i+1]; 1448b89f182dSHong Zhang for (j=0; j<nz; j++) { 1449b89f182dSHong Zhang col = pj[j]; 14500e7a5c2bSHong Zhang pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 1451b89f182dSHong Zhang } 1452b89f182dSHong Zhang /* U part */ 1453b89f182dSHong Zhang pc2 = b->a + bdiag[i+2]+1; 14540e7a5c2bSHong Zhang pj = b->j + bdiag[i+2]+1; 14550e7a5c2bSHong Zhang nz = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */ 1456b89f182dSHong Zhang for (j=0; j<nz; j++) { 1457b89f182dSHong Zhang col = pj[j]; 14580e7a5c2bSHong Zhang pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 1459b89f182dSHong Zhang } 1460b89f182dSHong Zhang 146128f1b45aSHong Zhang sctx.rs = rs; 1462b89f182dSHong Zhang sctx.pv = rtmp2[i+1]; 1463d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i+1);CHKERRQ(ierr); 146407b50cabSHong Zhang if (sctx.newshift) break; 146528f1b45aSHong Zhang pc2 = b->a + bdiag[i+1]; 1466b89f182dSHong Zhang *pc2 = 1.0/sctx.pv; 146728f1b45aSHong Zhang break; 1468b89f182dSHong Zhang 146968785679SHong Zhang case 3: 147068785679SHong Zhang /*----------*/ 147168785679SHong Zhang /* zero rtmp */ 147268785679SHong Zhang /* L part */ 147368785679SHong Zhang nz = bi[i+1] - bi[i]; 147468785679SHong Zhang bjtmp = bj + bi[i]; 147568785679SHong Zhang for (j=0; j<nz; j++) { 147668785679SHong Zhang col = bjtmp[j]; 1477b89f182dSHong Zhang rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0; 147868785679SHong Zhang } 147968785679SHong Zhang 148068785679SHong Zhang /* U part */ 148168785679SHong Zhang nz = bdiag[i]-bdiag[i+1]; 148268785679SHong Zhang bjtmp = bj + bdiag[i+1]+1; 148368785679SHong Zhang for (j=0; j<nz; j++) { 148468785679SHong Zhang col = bjtmp[j]; 1485b89f182dSHong Zhang rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0; 148668785679SHong Zhang } 148768785679SHong Zhang 148868785679SHong Zhang /* load in initial (unfactored row) */ 148968785679SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 149068785679SHong Zhang ajtmp = aj + ai[r[i]]; 149168785679SHong Zhang v1 = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; v3 = aa + ai[r[i]+2]; 149268785679SHong Zhang for (j=0; j<nz; j++) { 149368785679SHong Zhang col = ics[ajtmp[j]]; 1494b89f182dSHong Zhang rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; rtmp3[col] = v3[j]; 149568785679SHong Zhang } 149668785679SHong Zhang /* ZeropivotApply(): shift the diagonal of the matrix */ 1497b89f182dSHong Zhang rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; rtmp3[i+2] += sctx.shift_amount; 149868785679SHong Zhang 149968785679SHong Zhang /* elimination */ 150068785679SHong Zhang bjtmp = bj + bi[i]; 150168785679SHong Zhang row = *bjtmp++; /* pivot row */ 150268785679SHong Zhang nzL = bi[i+1] - bi[i]; 150368785679SHong Zhang for (k=0; k < nzL; k++) { 1504b89f182dSHong Zhang pc1 = rtmp1 + row; 1505b89f182dSHong Zhang pc2 = rtmp2 + row; 1506b89f182dSHong Zhang pc3 = rtmp3 + row; 150768785679SHong Zhang if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 != 0.0) { 150868785679SHong Zhang pv = b->a + bdiag[row]; 1509b89f182dSHong Zhang mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); mul3 = *pc3*(*pv); 1510b89f182dSHong Zhang *pc1 = mul1; *pc2 = mul2; *pc3 = mul3; 151168785679SHong Zhang 151268785679SHong Zhang pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 151368785679SHong Zhang pv = b->a + bdiag[row+1]+1; 151468785679SHong Zhang nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 151568785679SHong Zhang for (j=0; j<nz; j++) { 151668785679SHong Zhang col = pj[j]; 1517b89f182dSHong Zhang rtmp1[col] -= mul1 * pv[j]; 1518b89f182dSHong Zhang rtmp2[col] -= mul2 * pv[j]; 1519b89f182dSHong Zhang rtmp3[col] -= mul3 * pv[j]; 152068785679SHong Zhang } 152128b1a77fSLisandro Dalcin ierr = PetscLogFlops(3+6*nz);CHKERRQ(ierr); 152268785679SHong Zhang } 152368785679SHong Zhang row = *bjtmp++; 152468785679SHong Zhang } 152568785679SHong Zhang 1526b89f182dSHong Zhang /* finished row i; check zero pivot, then stick row i into b->a */ 1527b89f182dSHong Zhang rs = 0.0; 1528b89f182dSHong Zhang /* L part */ 1529b89f182dSHong Zhang pc1 = b->a + bi[i]; 1530b89f182dSHong Zhang pj = b->j + bi[i]; 1531b89f182dSHong Zhang nz = bi[i+1] - bi[i]; 1532b89f182dSHong Zhang for (j=0; j<nz; j++) { 1533b89f182dSHong Zhang col = pj[j]; 15340e7a5c2bSHong Zhang pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 1535b89f182dSHong Zhang } 1536b89f182dSHong Zhang /* U part */ 1537b89f182dSHong Zhang pc1 = b->a + bdiag[i+1]+1; 1538b89f182dSHong Zhang pj = b->j + bdiag[i+1]+1; 15390e7a5c2bSHong Zhang nz = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */ 1540b89f182dSHong Zhang for (j=0; j<nz; j++) { 1541b89f182dSHong Zhang col = pj[j]; 15420e7a5c2bSHong Zhang pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 1543b89f182dSHong Zhang } 154468785679SHong Zhang 1545b89f182dSHong Zhang sctx.rs = rs; 1546b89f182dSHong Zhang sctx.pv = rtmp1[i]; 1547d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 154807b50cabSHong Zhang if (sctx.newshift) break; 1549b89f182dSHong Zhang pc1 = b->a + bdiag[i]; /* Mark diag[i] */ 1550b89f182dSHong Zhang *pc1 = 1.0/sctx.pv; 1551b89f182dSHong Zhang 1552b89f182dSHong Zhang /* Now take care of 1st column of diagonal 3x3 block. */ 1553b89f182dSHong Zhang pc2 = rtmp2 + i; 1554b89f182dSHong Zhang pc3 = rtmp3 + i; 1555b89f182dSHong Zhang if (*pc2 != 0.0 || *pc3 != 0.0) { 1556b89f182dSHong Zhang mul2 = (*pc2)*(*pc1); *pc2 = mul2; 1557b89f182dSHong Zhang mul3 = (*pc3)*(*pc1); *pc3 = mul3; 155868785679SHong Zhang pj = b->j + bdiag[i+1]+1; /* beginning of U(i,:) */ 155968785679SHong Zhang nz = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */ 156068785679SHong Zhang for (j=0; j<nz; j++) { 156168785679SHong Zhang col = pj[j]; 1562b89f182dSHong Zhang rtmp2[col] -= mul2 * rtmp1[col]; 1563b89f182dSHong Zhang rtmp3[col] -= mul3 * rtmp1[col]; 156468785679SHong Zhang } 156528b1a77fSLisandro Dalcin ierr = PetscLogFlops(2+4*nz);CHKERRQ(ierr); 156668785679SHong Zhang } 156768785679SHong Zhang 1568b89f182dSHong Zhang /* finished row i+1; check zero pivot, then stick row i+1 into b->a */ 1569b89f182dSHong Zhang rs = 0.0; 1570b89f182dSHong Zhang /* L part */ 1571b89f182dSHong Zhang pc2 = b->a + bi[i+1]; 1572b89f182dSHong Zhang pj = b->j + bi[i+1]; 1573b89f182dSHong Zhang nz = bi[i+2] - bi[i+1]; 1574b89f182dSHong Zhang for (j=0; j<nz; j++) { 1575b89f182dSHong Zhang col = pj[j]; 15760e7a5c2bSHong Zhang pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 1577b89f182dSHong Zhang } 1578b89f182dSHong Zhang /* U part */ 1579b89f182dSHong Zhang pc2 = b->a + bdiag[i+2]+1; 15800e7a5c2bSHong Zhang pj = b->j + bdiag[i+2]+1; 15810e7a5c2bSHong Zhang nz = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */ 1582b89f182dSHong Zhang for (j=0; j<nz; j++) { 1583b89f182dSHong Zhang col = pj[j]; 15840e7a5c2bSHong Zhang pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 1585b89f182dSHong Zhang } 1586b89f182dSHong Zhang 1587b89f182dSHong Zhang sctx.rs = rs; 1588b89f182dSHong Zhang sctx.pv = rtmp2[i+1]; 1589d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i+1);CHKERRQ(ierr); 159007b50cabSHong Zhang if (sctx.newshift) break; 1591b89f182dSHong Zhang pc2 = b->a + bdiag[i+1]; 1592b89f182dSHong Zhang *pc2 = 1.0/sctx.pv; /* Mark diag[i+1] */ 1593b89f182dSHong Zhang 1594b89f182dSHong Zhang /* Now take care of 2nd column of diagonal 3x3 block. */ 1595b89f182dSHong Zhang pc3 = rtmp3 + i+1; 159668785679SHong Zhang if (*pc3 != 0.0) { 1597b89f182dSHong Zhang mul3 = (*pc3)*(*pc2); *pc3 = mul3; 159868785679SHong Zhang pj = b->j + bdiag[i+2]+1; /* beginning of U(i+1,:) */ 159968785679SHong Zhang nz = bdiag[i+1]-bdiag[i+2]-1; /* num of entries in U(i+1,:) excluding diag */ 160068785679SHong Zhang for (j=0; j<nz; j++) { 160168785679SHong Zhang col = pj[j]; 1602b89f182dSHong Zhang rtmp3[col] -= mul3 * rtmp2[col]; 160368785679SHong Zhang } 160428b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 160568785679SHong Zhang } 160668785679SHong Zhang 1607b89f182dSHong Zhang /* finished i+2; check zero pivot, then stick row i+2 into b->a */ 160868785679SHong Zhang rs = 0.0; 160968785679SHong Zhang /* L part */ 1610b89f182dSHong Zhang pc3 = b->a + bi[i+2]; 1611b89f182dSHong Zhang pj = b->j + bi[i+2]; 1612b89f182dSHong Zhang nz = bi[i+3] - bi[i+2]; 161368785679SHong Zhang for (j=0; j<nz; j++) { 161468785679SHong Zhang col = pj[j]; 16150e7a5c2bSHong Zhang pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]); 161668785679SHong Zhang } 161768785679SHong Zhang /* U part */ 1618b89f182dSHong Zhang pc3 = b->a + bdiag[i+3]+1; 16190e7a5c2bSHong Zhang pj = b->j + bdiag[i+3]+1; 16200e7a5c2bSHong Zhang nz = bdiag[i+2] - bdiag[i+3] - 1; /* exclude diagonal */ 162168785679SHong Zhang for (j=0; j<nz; j++) { 162268785679SHong Zhang col = pj[j]; 16230e7a5c2bSHong Zhang pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]); 162468785679SHong Zhang } 162568785679SHong Zhang 162668785679SHong Zhang sctx.rs = rs; 1627b89f182dSHong Zhang sctx.pv = rtmp3[i+2]; 1628d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i+2);CHKERRQ(ierr); 162907b50cabSHong Zhang if (sctx.newshift) break; 163068785679SHong Zhang pc3 = b->a + bdiag[i+2]; 1631b89f182dSHong Zhang *pc3 = 1.0/sctx.pv; /* Mark diag[i+2] */ 163268785679SHong Zhang break; 16339877982aSShri Abhyankar case 4: 16349877982aSShri Abhyankar /*----------*/ 16359877982aSShri Abhyankar /* zero rtmp */ 16369877982aSShri Abhyankar /* L part */ 16379877982aSShri Abhyankar nz = bi[i+1] - bi[i]; 16389877982aSShri Abhyankar bjtmp = bj + bi[i]; 16399877982aSShri Abhyankar for (j=0; j<nz; j++) { 16409877982aSShri Abhyankar col = bjtmp[j]; 16419877982aSShri Abhyankar rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;rtmp4[col] = 0.0; 16429877982aSShri Abhyankar } 16439877982aSShri Abhyankar 16449877982aSShri Abhyankar /* U part */ 16459877982aSShri Abhyankar nz = bdiag[i]-bdiag[i+1]; 16469877982aSShri Abhyankar bjtmp = bj + bdiag[i+1]+1; 16479877982aSShri Abhyankar for (j=0; j<nz; j++) { 16489877982aSShri Abhyankar col = bjtmp[j]; 16499877982aSShri Abhyankar rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0; rtmp4[col] = 0.0; 16509877982aSShri Abhyankar } 16519877982aSShri Abhyankar 16529877982aSShri Abhyankar /* load in initial (unfactored row) */ 16539877982aSShri Abhyankar nz = ai[r[i]+1] - ai[r[i]]; 16549877982aSShri Abhyankar ajtmp = aj + ai[r[i]]; 16559877982aSShri Abhyankar v1 = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; v3 = aa + ai[r[i]+2]; v4 = aa + ai[r[i]+3]; 16569877982aSShri Abhyankar for (j=0; j<nz; j++) { 16579877982aSShri Abhyankar col = ics[ajtmp[j]]; 16589877982aSShri Abhyankar rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; rtmp3[col] = v3[j]; rtmp4[col] = v4[j]; 16599877982aSShri Abhyankar } 16609877982aSShri Abhyankar /* ZeropivotApply(): shift the diagonal of the matrix */ 16619877982aSShri Abhyankar rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; rtmp3[i+2] += sctx.shift_amount; rtmp4[i+3] += sctx.shift_amount; 16629877982aSShri Abhyankar 16639877982aSShri Abhyankar /* elimination */ 16649877982aSShri Abhyankar bjtmp = bj + bi[i]; 16659877982aSShri Abhyankar row = *bjtmp++; /* pivot row */ 16669877982aSShri Abhyankar nzL = bi[i+1] - bi[i]; 16679877982aSShri Abhyankar for (k=0; k < nzL; k++) { 16689877982aSShri Abhyankar pc1 = rtmp1 + row; 16699877982aSShri Abhyankar pc2 = rtmp2 + row; 16709877982aSShri Abhyankar pc3 = rtmp3 + row; 16719877982aSShri Abhyankar pc4 = rtmp4 + row; 16729877982aSShri Abhyankar if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 != 0.0 || *pc4 != 0.0) { 16739877982aSShri Abhyankar pv = b->a + bdiag[row]; 16749877982aSShri Abhyankar mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); mul3 = *pc3*(*pv); mul4 = *pc4*(*pv); 16759877982aSShri Abhyankar *pc1 = mul1; *pc2 = mul2; *pc3 = mul3; *pc4 = mul4; 16769877982aSShri Abhyankar 16779877982aSShri Abhyankar pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 16789877982aSShri Abhyankar pv = b->a + bdiag[row+1]+1; 16799877982aSShri Abhyankar nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 16809877982aSShri Abhyankar for (j=0; j<nz; j++) { 16819877982aSShri Abhyankar col = pj[j]; 16829877982aSShri Abhyankar rtmp1[col] -= mul1 * pv[j]; 16839877982aSShri Abhyankar rtmp2[col] -= mul2 * pv[j]; 16849877982aSShri Abhyankar rtmp3[col] -= mul3 * pv[j]; 16859877982aSShri Abhyankar rtmp4[col] -= mul4 * pv[j]; 16869877982aSShri Abhyankar } 168728b1a77fSLisandro Dalcin ierr = PetscLogFlops(4+8*nz);CHKERRQ(ierr); 16889877982aSShri Abhyankar } 16899877982aSShri Abhyankar row = *bjtmp++; 16909877982aSShri Abhyankar } 16919877982aSShri Abhyankar 16929877982aSShri Abhyankar /* finished row i; check zero pivot, then stick row i into b->a */ 16939877982aSShri Abhyankar rs = 0.0; 16949877982aSShri Abhyankar /* L part */ 16959877982aSShri Abhyankar pc1 = b->a + bi[i]; 16969877982aSShri Abhyankar pj = b->j + bi[i]; 16979877982aSShri Abhyankar nz = bi[i+1] - bi[i]; 16989877982aSShri Abhyankar for (j=0; j<nz; j++) { 16999877982aSShri Abhyankar col = pj[j]; 17009877982aSShri Abhyankar pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 17019877982aSShri Abhyankar } 17029877982aSShri Abhyankar /* U part */ 17039877982aSShri Abhyankar pc1 = b->a + bdiag[i+1]+1; 17049877982aSShri Abhyankar pj = b->j + bdiag[i+1]+1; 17059877982aSShri Abhyankar nz = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */ 17069877982aSShri Abhyankar for (j=0; j<nz; j++) { 17079877982aSShri Abhyankar col = pj[j]; 17089877982aSShri Abhyankar pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 17099877982aSShri Abhyankar } 17109877982aSShri Abhyankar 17119877982aSShri Abhyankar sctx.rs = rs; 17129877982aSShri Abhyankar sctx.pv = rtmp1[i]; 1713d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 171407b50cabSHong Zhang if (sctx.newshift) break; 17159877982aSShri Abhyankar pc1 = b->a + bdiag[i]; /* Mark diag[i] */ 17169877982aSShri Abhyankar *pc1 = 1.0/sctx.pv; 17179877982aSShri Abhyankar 17189877982aSShri Abhyankar /* Now take care of 1st column of diagonal 4x4 block. */ 17199877982aSShri Abhyankar pc2 = rtmp2 + i; 17209877982aSShri Abhyankar pc3 = rtmp3 + i; 17219877982aSShri Abhyankar pc4 = rtmp4 + i; 17229877982aSShri Abhyankar if (*pc2 != 0.0 || *pc3 != 0.0 || *pc4 != 0.0) { 17239877982aSShri Abhyankar mul2 = (*pc2)*(*pc1); *pc2 = mul2; 17249877982aSShri Abhyankar mul3 = (*pc3)*(*pc1); *pc3 = mul3; 17259877982aSShri Abhyankar mul4 = (*pc4)*(*pc1); *pc4 = mul4; 17269877982aSShri Abhyankar pj = b->j + bdiag[i+1]+1; /* beginning of U(i,:) */ 17279877982aSShri Abhyankar nz = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */ 17289877982aSShri Abhyankar for (j=0; j<nz; j++) { 17299877982aSShri Abhyankar col = pj[j]; 17309877982aSShri Abhyankar rtmp2[col] -= mul2 * rtmp1[col]; 17319877982aSShri Abhyankar rtmp3[col] -= mul3 * rtmp1[col]; 17329877982aSShri Abhyankar rtmp4[col] -= mul4 * rtmp1[col]; 17339877982aSShri Abhyankar } 173428b1a77fSLisandro Dalcin ierr = PetscLogFlops(3+6*nz);CHKERRQ(ierr); 17359877982aSShri Abhyankar } 17369877982aSShri Abhyankar 17379877982aSShri Abhyankar /* finished row i+1; check zero pivot, then stick row i+1 into b->a */ 17389877982aSShri Abhyankar rs = 0.0; 17399877982aSShri Abhyankar /* L part */ 17409877982aSShri Abhyankar pc2 = b->a + bi[i+1]; 17419877982aSShri Abhyankar pj = b->j + bi[i+1]; 17429877982aSShri Abhyankar nz = bi[i+2] - bi[i+1]; 17439877982aSShri Abhyankar for (j=0; j<nz; j++) { 17449877982aSShri Abhyankar col = pj[j]; 17459877982aSShri Abhyankar pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 17469877982aSShri Abhyankar } 17479877982aSShri Abhyankar /* U part */ 17489877982aSShri Abhyankar pc2 = b->a + bdiag[i+2]+1; 17499877982aSShri Abhyankar pj = b->j + bdiag[i+2]+1; 17509877982aSShri Abhyankar nz = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */ 17519877982aSShri Abhyankar for (j=0; j<nz; j++) { 17529877982aSShri Abhyankar col = pj[j]; 17539877982aSShri Abhyankar pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 17549877982aSShri Abhyankar } 17559877982aSShri Abhyankar 17569877982aSShri Abhyankar sctx.rs = rs; 17579877982aSShri Abhyankar sctx.pv = rtmp2[i+1]; 1758d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i+1);CHKERRQ(ierr); 175907b50cabSHong Zhang if (sctx.newshift) break; 17609877982aSShri Abhyankar pc2 = b->a + bdiag[i+1]; 17619877982aSShri Abhyankar *pc2 = 1.0/sctx.pv; /* Mark diag[i+1] */ 17629877982aSShri Abhyankar 17639877982aSShri Abhyankar /* Now take care of 2nd column of diagonal 4x4 block. */ 17649877982aSShri Abhyankar pc3 = rtmp3 + i+1; 17659877982aSShri Abhyankar pc4 = rtmp4 + i+1; 17669877982aSShri Abhyankar if (*pc3 != 0.0 || *pc4 != 0.0) { 17679877982aSShri Abhyankar mul3 = (*pc3)*(*pc2); *pc3 = mul3; 17689877982aSShri Abhyankar mul4 = (*pc4)*(*pc2); *pc4 = mul4; 17699877982aSShri Abhyankar pj = b->j + bdiag[i+2]+1; /* beginning of U(i+1,:) */ 17709877982aSShri Abhyankar nz = bdiag[i+1]-bdiag[i+2]-1; /* num of entries in U(i+1,:) excluding diag */ 17719877982aSShri Abhyankar for (j=0; j<nz; j++) { 17729877982aSShri Abhyankar col = pj[j]; 17739877982aSShri Abhyankar rtmp3[col] -= mul3 * rtmp2[col]; 17749877982aSShri Abhyankar rtmp4[col] -= mul4 * rtmp2[col]; 17759877982aSShri Abhyankar } 17769877982aSShri Abhyankar ierr = PetscLogFlops(4*nz);CHKERRQ(ierr); 17779877982aSShri Abhyankar } 17789877982aSShri Abhyankar 17799877982aSShri Abhyankar /* finished i+2; check zero pivot, then stick row i+2 into b->a */ 17809877982aSShri Abhyankar rs = 0.0; 17819877982aSShri Abhyankar /* L part */ 17829877982aSShri Abhyankar pc3 = b->a + bi[i+2]; 17839877982aSShri Abhyankar pj = b->j + bi[i+2]; 17849877982aSShri Abhyankar nz = bi[i+3] - bi[i+2]; 17859877982aSShri Abhyankar for (j=0; j<nz; j++) { 17869877982aSShri Abhyankar col = pj[j]; 17879877982aSShri Abhyankar pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]); 17889877982aSShri Abhyankar } 17899877982aSShri Abhyankar /* U part */ 17909877982aSShri Abhyankar pc3 = b->a + bdiag[i+3]+1; 17919877982aSShri Abhyankar pj = b->j + bdiag[i+3]+1; 17929877982aSShri Abhyankar nz = bdiag[i+2] - bdiag[i+3] - 1; /* exclude diagonal */ 17939877982aSShri Abhyankar for (j=0; j<nz; j++) { 17949877982aSShri Abhyankar col = pj[j]; 17959877982aSShri Abhyankar pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]); 17969877982aSShri Abhyankar } 17979877982aSShri Abhyankar 17989877982aSShri Abhyankar sctx.rs = rs; 17999877982aSShri Abhyankar sctx.pv = rtmp3[i+2]; 1800d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i+2);CHKERRQ(ierr); 180107b50cabSHong Zhang if (sctx.newshift) break; 18029877982aSShri Abhyankar pc3 = b->a + bdiag[i+2]; 18039877982aSShri Abhyankar *pc3 = 1.0/sctx.pv; /* Mark diag[i+2] */ 18049877982aSShri Abhyankar 18059877982aSShri Abhyankar /* Now take care of 3rd column of diagonal 4x4 block. */ 18069877982aSShri Abhyankar pc4 = rtmp4 + i+2; 18079877982aSShri Abhyankar if (*pc4 != 0.0) { 18089877982aSShri Abhyankar mul4 = (*pc4)*(*pc3); *pc4 = mul4; 18099877982aSShri Abhyankar pj = b->j + bdiag[i+3]+1; /* beginning of U(i+2,:) */ 18109877982aSShri Abhyankar nz = bdiag[i+2]-bdiag[i+3]-1; /* num of entries in U(i+2,:) excluding diag */ 18119877982aSShri Abhyankar for (j=0; j<nz; j++) { 18129877982aSShri Abhyankar col = pj[j]; 18139877982aSShri Abhyankar rtmp4[col] -= mul4 * rtmp3[col]; 18149877982aSShri Abhyankar } 181528b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr); 18169877982aSShri Abhyankar } 18179877982aSShri Abhyankar 18189877982aSShri Abhyankar /* finished i+3; check zero pivot, then stick row i+3 into b->a */ 18199877982aSShri Abhyankar rs = 0.0; 18209877982aSShri Abhyankar /* L part */ 18219877982aSShri Abhyankar pc4 = b->a + bi[i+3]; 18229877982aSShri Abhyankar pj = b->j + bi[i+3]; 18239877982aSShri Abhyankar nz = bi[i+4] - bi[i+3]; 18249877982aSShri Abhyankar for (j=0; j<nz; j++) { 18259877982aSShri Abhyankar col = pj[j]; 18269877982aSShri Abhyankar pc4[j] = rtmp4[col]; rs += PetscAbsScalar(pc4[j]); 18279877982aSShri Abhyankar } 18289877982aSShri Abhyankar /* U part */ 18299877982aSShri Abhyankar pc4 = b->a + bdiag[i+4]+1; 18309877982aSShri Abhyankar pj = b->j + bdiag[i+4]+1; 18319877982aSShri Abhyankar nz = bdiag[i+3] - bdiag[i+4] - 1; /* exclude diagonal */ 18329877982aSShri Abhyankar for (j=0; j<nz; j++) { 18339877982aSShri Abhyankar col = pj[j]; 18349877982aSShri Abhyankar pc4[j] = rtmp4[col]; rs += PetscAbsScalar(pc4[j]); 18359877982aSShri Abhyankar } 18369877982aSShri Abhyankar 18379877982aSShri Abhyankar sctx.rs = rs; 18389877982aSShri Abhyankar sctx.pv = rtmp4[i+3]; 1839d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i+3);CHKERRQ(ierr); 184007b50cabSHong Zhang if (sctx.newshift) break; 18419877982aSShri Abhyankar pc4 = b->a + bdiag[i+3]; 18429877982aSShri Abhyankar *pc4 = 1.0/sctx.pv; /* Mark diag[i+3] */ 18439877982aSShri Abhyankar break; 184468785679SHong Zhang 184528f1b45aSHong Zhang default: 1846e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Node size not yet supported \n"); 184728f1b45aSHong Zhang } 1848c2b86aeeSHong Zhang if (sctx.newshift) break; /* break for (inod=0,i=0; inod<node_max; inod++) */ 184928f1b45aSHong Zhang i += nodesz; /* Update the row */ 185068785679SHong Zhang } 185128f1b45aSHong Zhang 185228f1b45aSHong Zhang /* MatPivotRefine() */ 185307b50cabSHong Zhang if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 185428f1b45aSHong Zhang /* 185528f1b45aSHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 185628f1b45aSHong Zhang * then try lower shift 185728f1b45aSHong Zhang */ 185828f1b45aSHong Zhang sctx.shift_hi = sctx.shift_fraction; 185928f1b45aSHong Zhang sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 186028f1b45aSHong Zhang sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 186107b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 186228f1b45aSHong Zhang sctx.nshift++; 186328f1b45aSHong Zhang } 186407b50cabSHong Zhang } while (sctx.newshift); 186528f1b45aSHong Zhang 1866eb9baa12SBarry Smith ierr = PetscFree4(rtmp1,rtmp2,rtmp3,rtmp4);CHKERRQ(ierr); 186768785679SHong Zhang ierr = PetscFree(tmp_vec2);CHKERRQ(ierr); 186828f1b45aSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 186928f1b45aSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 187028f1b45aSHong Zhang 1871abb87a52SBarry Smith if (b->inode.size) { 1872abb87a52SBarry Smith C->ops->solve = MatSolve_SeqAIJ_Inode; 1873abb87a52SBarry Smith } else { 1874d3ac4fa3SBarry Smith C->ops->solve = MatSolve_SeqAIJ; 1875abb87a52SBarry Smith } 187628f1b45aSHong Zhang C->ops->solveadd = MatSolveAdd_SeqAIJ; 187728f1b45aSHong Zhang C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 187828f1b45aSHong Zhang C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 187928f1b45aSHong Zhang C->ops->matsolve = MatMatSolve_SeqAIJ; 188028f1b45aSHong Zhang C->assembled = PETSC_TRUE; 188128f1b45aSHong Zhang C->preallocated = PETSC_TRUE; 18822205254eSKarl Rupp 188328f1b45aSHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 188428f1b45aSHong Zhang 188528f1b45aSHong Zhang /* MatShiftView(A,info,&sctx) */ 188628f1b45aSHong Zhang if (sctx.nshift) { 1887f4db908eSBarry Smith if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { 188857622a8eSBarry Smith ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %g, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,(double)sctx.shift_amount,(double)sctx.shift_fraction,(double)sctx.shift_top);CHKERRQ(ierr); 1889f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 189057622a8eSBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr); 1891f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) { 189257622a8eSBarry Smith ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %g\n",sctx.nshift,(double)info->shiftamount);CHKERRQ(ierr); 189328f1b45aSHong Zhang } 189428f1b45aSHong Zhang } 189528f1b45aSHong Zhang PetscFunctionReturn(0); 189628f1b45aSHong Zhang } 1897628f99d7SShri Abhyankar 1898628f99d7SShri Abhyankar #undef __FUNCT__ 1899628f99d7SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_Inode_inplace" 1900628f99d7SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ_Inode_inplace(Mat B,Mat A,const MatFactorInfo *info) 1901628f99d7SShri Abhyankar { 1902628f99d7SShri Abhyankar Mat C = B; 1903628f99d7SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b = (Mat_SeqAIJ*)C->data; 1904628f99d7SShri Abhyankar IS iscol = b->col,isrow = b->row,isicol = b->icol; 1905628f99d7SShri Abhyankar PetscErrorCode ierr; 1906628f99d7SShri Abhyankar const PetscInt *r,*ic,*c,*ics; 1907628f99d7SShri Abhyankar PetscInt n = A->rmap->n,*bi = b->i; 1908628f99d7SShri Abhyankar PetscInt *bj = b->j,*nbj=b->j +1,*ajtmp,*bjtmp,nz,nz_tmp,row,prow; 19098758e1faSBarry Smith PetscInt i,j,idx,*bd = b->diag,node_max,nodesz; 19108758e1faSBarry Smith PetscInt *ai = a->i,*aj = a->j; 1911628f99d7SShri Abhyankar PetscInt *ns,*tmp_vec1,*tmp_vec2,*nsmap,*pj; 1912628f99d7SShri Abhyankar PetscScalar mul1,mul2,mul3,tmp; 1913628f99d7SShri Abhyankar MatScalar *pc1,*pc2,*pc3,*ba = b->a,*pv,*rtmp11,*rtmp22,*rtmp33; 1914628f99d7SShri Abhyankar const MatScalar *v1,*v2,*v3,*aa = a->a,*rtmp1; 1915628f99d7SShri Abhyankar PetscReal rs=0.0; 1916628f99d7SShri Abhyankar FactorShiftCtx sctx; 1917628f99d7SShri Abhyankar 1918628f99d7SShri Abhyankar PetscFunctionBegin; 1919628f99d7SShri Abhyankar sctx.shift_top = 0; 1920628f99d7SShri Abhyankar sctx.nshift_max = 0; 1921628f99d7SShri Abhyankar sctx.shift_lo = 0; 1922628f99d7SShri Abhyankar sctx.shift_hi = 0; 1923628f99d7SShri Abhyankar sctx.shift_fraction = 0; 1924628f99d7SShri Abhyankar 1925628f99d7SShri Abhyankar /* if both shift schemes are chosen by user, only use info->shiftpd */ 1926f4db908eSBarry Smith if (info->shifttype==(PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 1927628f99d7SShri Abhyankar sctx.shift_top = 0; 1928628f99d7SShri Abhyankar for (i=0; i<n; i++) { 1929628f99d7SShri Abhyankar /* calculate rs = sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 1930628f99d7SShri Abhyankar rs = 0.0; 1931628f99d7SShri Abhyankar ajtmp = aj + ai[i]; 1932628f99d7SShri Abhyankar rtmp1 = aa + ai[i]; 1933628f99d7SShri Abhyankar nz = ai[i+1] - ai[i]; 1934628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 1935628f99d7SShri Abhyankar if (*ajtmp != i) { 1936628f99d7SShri Abhyankar rs += PetscAbsScalar(*rtmp1++); 1937628f99d7SShri Abhyankar } else { 1938628f99d7SShri Abhyankar rs -= PetscRealPart(*rtmp1++); 1939628f99d7SShri Abhyankar } 1940628f99d7SShri Abhyankar ajtmp++; 1941628f99d7SShri Abhyankar } 1942628f99d7SShri Abhyankar if (rs>sctx.shift_top) sctx.shift_top = rs; 1943628f99d7SShri Abhyankar } 1944628f99d7SShri Abhyankar if (sctx.shift_top == 0.0) sctx.shift_top += 1.e-12; 1945628f99d7SShri Abhyankar sctx.shift_top *= 1.1; 1946628f99d7SShri Abhyankar sctx.nshift_max = 5; 1947628f99d7SShri Abhyankar sctx.shift_lo = 0.; 1948628f99d7SShri Abhyankar sctx.shift_hi = 1.; 1949628f99d7SShri Abhyankar } 1950628f99d7SShri Abhyankar sctx.shift_amount = 0; 1951628f99d7SShri Abhyankar sctx.nshift = 0; 1952628f99d7SShri Abhyankar 1953628f99d7SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1954628f99d7SShri Abhyankar ierr = ISGetIndices(iscol,&c);CHKERRQ(ierr); 1955628f99d7SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 19561795a4d1SJed Brown ierr = PetscCalloc3(n,&rtmp11,n,&rtmp22,n,&rtmp33);CHKERRQ(ierr); 1957628f99d7SShri Abhyankar ics = ic; 1958628f99d7SShri Abhyankar 1959628f99d7SShri Abhyankar node_max = a->inode.node_count; 1960628f99d7SShri Abhyankar ns = a->inode.size; 1961e7e72b3dSBarry Smith if (!ns) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix without inode information"); 1962628f99d7SShri Abhyankar 1963628f99d7SShri Abhyankar /* If max inode size > 3, split it into two inodes.*/ 1964628f99d7SShri Abhyankar /* also map the inode sizes according to the ordering */ 1965854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&tmp_vec1);CHKERRQ(ierr); 1966628f99d7SShri Abhyankar for (i=0,j=0; i<node_max; ++i,++j) { 1967628f99d7SShri Abhyankar if (ns[i]>3) { 1968628f99d7SShri Abhyankar tmp_vec1[j] = ns[i]/2; /* Assuming ns[i] < =5 */ 1969628f99d7SShri Abhyankar ++j; 1970628f99d7SShri Abhyankar tmp_vec1[j] = ns[i] - tmp_vec1[j-1]; 1971628f99d7SShri Abhyankar } else { 1972628f99d7SShri Abhyankar tmp_vec1[j] = ns[i]; 1973628f99d7SShri Abhyankar } 1974628f99d7SShri Abhyankar } 1975628f99d7SShri Abhyankar /* Use the correct node_max */ 1976628f99d7SShri Abhyankar node_max = j; 1977628f99d7SShri Abhyankar 1978628f99d7SShri Abhyankar /* Now reorder the inode info based on mat re-ordering info */ 1979628f99d7SShri Abhyankar /* First create a row -> inode_size_array_index map */ 1980854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&nsmap);CHKERRQ(ierr); 1981854ce69bSBarry Smith ierr = PetscMalloc1(node_max+1,&tmp_vec2);CHKERRQ(ierr); 1982628f99d7SShri Abhyankar for (i=0,row=0; i<node_max; i++) { 1983628f99d7SShri Abhyankar nodesz = tmp_vec1[i]; 1984628f99d7SShri Abhyankar for (j=0; j<nodesz; j++,row++) { 1985628f99d7SShri Abhyankar nsmap[row] = i; 1986628f99d7SShri Abhyankar } 1987628f99d7SShri Abhyankar } 1988628f99d7SShri Abhyankar /* Using nsmap, create a reordered ns structure */ 1989628f99d7SShri Abhyankar for (i=0,j=0; i< node_max; i++) { 1990628f99d7SShri Abhyankar nodesz = tmp_vec1[nsmap[r[j]]]; /* here the reordered row_no is in r[] */ 1991628f99d7SShri Abhyankar tmp_vec2[i] = nodesz; 1992628f99d7SShri Abhyankar j += nodesz; 1993628f99d7SShri Abhyankar } 1994628f99d7SShri Abhyankar ierr = PetscFree(nsmap);CHKERRQ(ierr); 1995628f99d7SShri Abhyankar ierr = PetscFree(tmp_vec1);CHKERRQ(ierr); 1996628f99d7SShri Abhyankar /* Now use the correct ns */ 1997628f99d7SShri Abhyankar ns = tmp_vec2; 1998628f99d7SShri Abhyankar 1999628f99d7SShri Abhyankar do { 200007b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 2001628f99d7SShri Abhyankar /* Now loop over each block-row, and do the factorization */ 2002628f99d7SShri Abhyankar for (i=0,row=0; i<node_max; i++) { 2003628f99d7SShri Abhyankar nodesz = ns[i]; 2004628f99d7SShri Abhyankar nz = bi[row+1] - bi[row]; 2005628f99d7SShri Abhyankar bjtmp = bj + bi[row]; 2006628f99d7SShri Abhyankar 2007628f99d7SShri Abhyankar switch (nodesz) { 2008628f99d7SShri Abhyankar case 1: 2009628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2010628f99d7SShri Abhyankar idx = bjtmp[j]; 2011628f99d7SShri Abhyankar rtmp11[idx] = 0.0; 2012628f99d7SShri Abhyankar } 2013628f99d7SShri Abhyankar 2014628f99d7SShri Abhyankar /* load in initial (unfactored row) */ 2015628f99d7SShri Abhyankar idx = r[row]; 2016628f99d7SShri Abhyankar nz_tmp = ai[idx+1] - ai[idx]; 2017628f99d7SShri Abhyankar ajtmp = aj + ai[idx]; 2018628f99d7SShri Abhyankar v1 = aa + ai[idx]; 2019628f99d7SShri Abhyankar 2020628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2021628f99d7SShri Abhyankar idx = ics[ajtmp[j]]; 2022628f99d7SShri Abhyankar rtmp11[idx] = v1[j]; 2023628f99d7SShri Abhyankar } 2024628f99d7SShri Abhyankar rtmp11[ics[r[row]]] += sctx.shift_amount; 2025628f99d7SShri Abhyankar 2026628f99d7SShri Abhyankar prow = *bjtmp++; 2027628f99d7SShri Abhyankar while (prow < row) { 2028628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 2029628f99d7SShri Abhyankar if (*pc1 != 0.0) { 2030628f99d7SShri Abhyankar pv = ba + bd[prow]; 2031628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2032628f99d7SShri Abhyankar mul1 = *pc1 * *pv++; 2033628f99d7SShri Abhyankar *pc1 = mul1; 2034628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 203528b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz_tmp);CHKERRQ(ierr); 2036628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2037628f99d7SShri Abhyankar tmp = pv[j]; 2038628f99d7SShri Abhyankar idx = pj[j]; 2039628f99d7SShri Abhyankar rtmp11[idx] -= mul1 * tmp; 2040628f99d7SShri Abhyankar } 2041628f99d7SShri Abhyankar } 2042628f99d7SShri Abhyankar prow = *bjtmp++; 2043628f99d7SShri Abhyankar } 2044628f99d7SShri Abhyankar pj = bj + bi[row]; 2045628f99d7SShri Abhyankar pc1 = ba + bi[row]; 2046628f99d7SShri Abhyankar 2047628f99d7SShri Abhyankar sctx.pv = rtmp11[row]; 2048628f99d7SShri Abhyankar rtmp11[row] = 1.0/rtmp11[row]; /* invert diag */ 2049628f99d7SShri Abhyankar rs = 0.0; 2050628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2051628f99d7SShri Abhyankar idx = pj[j]; 2052628f99d7SShri Abhyankar pc1[j] = rtmp11[idx]; /* rtmp11 -> ba */ 2053628f99d7SShri Abhyankar if (idx != row) rs += PetscAbsScalar(pc1[j]); 2054628f99d7SShri Abhyankar } 2055628f99d7SShri Abhyankar sctx.rs = rs; 2056d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,row);CHKERRQ(ierr); 205707b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2058628f99d7SShri Abhyankar break; 2059628f99d7SShri Abhyankar 2060628f99d7SShri Abhyankar case 2: 2061628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2062628f99d7SShri Abhyankar idx = bjtmp[j]; 2063628f99d7SShri Abhyankar rtmp11[idx] = 0.0; 2064628f99d7SShri Abhyankar rtmp22[idx] = 0.0; 2065628f99d7SShri Abhyankar } 2066628f99d7SShri Abhyankar 2067628f99d7SShri Abhyankar /* load in initial (unfactored row) */ 2068628f99d7SShri Abhyankar idx = r[row]; 2069628f99d7SShri Abhyankar nz_tmp = ai[idx+1] - ai[idx]; 2070628f99d7SShri Abhyankar ajtmp = aj + ai[idx]; 2071628f99d7SShri Abhyankar v1 = aa + ai[idx]; 2072628f99d7SShri Abhyankar v2 = aa + ai[idx+1]; 2073628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2074628f99d7SShri Abhyankar idx = ics[ajtmp[j]]; 2075628f99d7SShri Abhyankar rtmp11[idx] = v1[j]; 2076628f99d7SShri Abhyankar rtmp22[idx] = v2[j]; 2077628f99d7SShri Abhyankar } 2078628f99d7SShri Abhyankar rtmp11[ics[r[row]]] += sctx.shift_amount; 2079628f99d7SShri Abhyankar rtmp22[ics[r[row+1]]] += sctx.shift_amount; 2080628f99d7SShri Abhyankar 2081628f99d7SShri Abhyankar prow = *bjtmp++; 2082628f99d7SShri Abhyankar while (prow < row) { 2083628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 2084628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2085628f99d7SShri Abhyankar if (*pc1 != 0.0 || *pc2 != 0.0) { 2086628f99d7SShri Abhyankar pv = ba + bd[prow]; 2087628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2088628f99d7SShri Abhyankar mul1 = *pc1 * *pv; 2089628f99d7SShri Abhyankar mul2 = *pc2 * *pv; 2090628f99d7SShri Abhyankar ++pv; 2091628f99d7SShri Abhyankar *pc1 = mul1; 2092628f99d7SShri Abhyankar *pc2 = mul2; 2093628f99d7SShri Abhyankar 2094628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2095628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2096628f99d7SShri Abhyankar tmp = pv[j]; 2097628f99d7SShri Abhyankar idx = pj[j]; 2098628f99d7SShri Abhyankar rtmp11[idx] -= mul1 * tmp; 2099628f99d7SShri Abhyankar rtmp22[idx] -= mul2 * tmp; 2100628f99d7SShri Abhyankar } 210128b1a77fSLisandro Dalcin ierr = PetscLogFlops(2+4*nz_tmp);CHKERRQ(ierr); 2102628f99d7SShri Abhyankar } 2103628f99d7SShri Abhyankar prow = *bjtmp++; 2104628f99d7SShri Abhyankar } 2105628f99d7SShri Abhyankar 2106628f99d7SShri Abhyankar /* Now take care of diagonal 2x2 block. Note: prow = row here */ 2107628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 2108628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2109628f99d7SShri Abhyankar 2110628f99d7SShri Abhyankar sctx.pv = *pc1; 2111628f99d7SShri Abhyankar pj = bj + bi[prow]; 2112628f99d7SShri Abhyankar rs = 0.0; 2113628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2114628f99d7SShri Abhyankar idx = pj[j]; 2115628f99d7SShri Abhyankar if (idx != prow) rs += PetscAbsScalar(rtmp11[idx]); 2116628f99d7SShri Abhyankar } 2117628f99d7SShri Abhyankar sctx.rs = rs; 2118d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,row);CHKERRQ(ierr); 211907b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2120628f99d7SShri Abhyankar 2121628f99d7SShri Abhyankar if (*pc2 != 0.0) { 2122628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2123628f99d7SShri Abhyankar mul2 = (*pc2)/(*pc1); /* since diag is not yet inverted.*/ 2124628f99d7SShri Abhyankar *pc2 = mul2; 2125628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2126628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2127628f99d7SShri Abhyankar idx = pj[j]; 2128628f99d7SShri Abhyankar tmp = rtmp11[idx]; 2129628f99d7SShri Abhyankar rtmp22[idx] -= mul2 * tmp; 2130628f99d7SShri Abhyankar } 213128b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz_tmp);CHKERRQ(ierr); 2132628f99d7SShri Abhyankar } 2133628f99d7SShri Abhyankar 2134628f99d7SShri Abhyankar pj = bj + bi[row]; 2135628f99d7SShri Abhyankar pc1 = ba + bi[row]; 2136628f99d7SShri Abhyankar pc2 = ba + bi[row+1]; 2137628f99d7SShri Abhyankar 2138628f99d7SShri Abhyankar sctx.pv = rtmp22[row+1]; 2139628f99d7SShri Abhyankar rs = 0.0; 2140628f99d7SShri Abhyankar rtmp11[row] = 1.0/rtmp11[row]; 2141628f99d7SShri Abhyankar rtmp22[row+1] = 1.0/rtmp22[row+1]; 2142628f99d7SShri Abhyankar /* copy row entries from dense representation to sparse */ 2143628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2144628f99d7SShri Abhyankar idx = pj[j]; 2145628f99d7SShri Abhyankar pc1[j] = rtmp11[idx]; 2146628f99d7SShri Abhyankar pc2[j] = rtmp22[idx]; 2147628f99d7SShri Abhyankar if (idx != row+1) rs += PetscAbsScalar(pc2[j]); 2148628f99d7SShri Abhyankar } 2149628f99d7SShri Abhyankar sctx.rs = rs; 2150d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,row+1);CHKERRQ(ierr); 215107b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2152628f99d7SShri Abhyankar break; 2153628f99d7SShri Abhyankar 2154628f99d7SShri Abhyankar case 3: 2155628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2156628f99d7SShri Abhyankar idx = bjtmp[j]; 2157628f99d7SShri Abhyankar rtmp11[idx] = 0.0; 2158628f99d7SShri Abhyankar rtmp22[idx] = 0.0; 2159628f99d7SShri Abhyankar rtmp33[idx] = 0.0; 2160628f99d7SShri Abhyankar } 2161628f99d7SShri Abhyankar /* copy the nonzeros for the 3 rows from sparse representation to dense in rtmp*[] */ 2162628f99d7SShri Abhyankar idx = r[row]; 2163628f99d7SShri Abhyankar nz_tmp = ai[idx+1] - ai[idx]; 2164628f99d7SShri Abhyankar ajtmp = aj + ai[idx]; 2165628f99d7SShri Abhyankar v1 = aa + ai[idx]; 2166628f99d7SShri Abhyankar v2 = aa + ai[idx+1]; 2167628f99d7SShri Abhyankar v3 = aa + ai[idx+2]; 2168628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2169628f99d7SShri Abhyankar idx = ics[ajtmp[j]]; 2170628f99d7SShri Abhyankar rtmp11[idx] = v1[j]; 2171628f99d7SShri Abhyankar rtmp22[idx] = v2[j]; 2172628f99d7SShri Abhyankar rtmp33[idx] = v3[j]; 2173628f99d7SShri Abhyankar } 2174628f99d7SShri Abhyankar rtmp11[ics[r[row]]] += sctx.shift_amount; 2175628f99d7SShri Abhyankar rtmp22[ics[r[row+1]]] += sctx.shift_amount; 2176628f99d7SShri Abhyankar rtmp33[ics[r[row+2]]] += sctx.shift_amount; 2177628f99d7SShri Abhyankar 2178628f99d7SShri Abhyankar /* loop over all pivot row blocks above this row block */ 2179628f99d7SShri Abhyankar prow = *bjtmp++; 2180628f99d7SShri Abhyankar while (prow < row) { 2181628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 2182628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2183628f99d7SShri Abhyankar pc3 = rtmp33 + prow; 2184628f99d7SShri Abhyankar if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 !=0.0) { 2185628f99d7SShri Abhyankar pv = ba + bd[prow]; 2186628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2187628f99d7SShri Abhyankar mul1 = *pc1 * *pv; 2188628f99d7SShri Abhyankar mul2 = *pc2 * *pv; 2189628f99d7SShri Abhyankar mul3 = *pc3 * *pv; 2190628f99d7SShri Abhyankar ++pv; 2191628f99d7SShri Abhyankar *pc1 = mul1; 2192628f99d7SShri Abhyankar *pc2 = mul2; 2193628f99d7SShri Abhyankar *pc3 = mul3; 2194628f99d7SShri Abhyankar 2195628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2196628f99d7SShri Abhyankar /* update this row based on pivot row */ 2197628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2198628f99d7SShri Abhyankar tmp = pv[j]; 2199628f99d7SShri Abhyankar idx = pj[j]; 2200628f99d7SShri Abhyankar rtmp11[idx] -= mul1 * tmp; 2201628f99d7SShri Abhyankar rtmp22[idx] -= mul2 * tmp; 2202628f99d7SShri Abhyankar rtmp33[idx] -= mul3 * tmp; 2203628f99d7SShri Abhyankar } 220428b1a77fSLisandro Dalcin ierr = PetscLogFlops(3+6*nz_tmp);CHKERRQ(ierr); 2205628f99d7SShri Abhyankar } 2206628f99d7SShri Abhyankar prow = *bjtmp++; 2207628f99d7SShri Abhyankar } 2208628f99d7SShri Abhyankar 2209628f99d7SShri Abhyankar /* Now take care of diagonal 3x3 block in this set of rows */ 2210628f99d7SShri Abhyankar /* note: prow = row here */ 2211628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 2212628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2213628f99d7SShri Abhyankar pc3 = rtmp33 + prow; 2214628f99d7SShri Abhyankar 2215628f99d7SShri Abhyankar sctx.pv = *pc1; 2216628f99d7SShri Abhyankar pj = bj + bi[prow]; 2217628f99d7SShri Abhyankar rs = 0.0; 2218628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2219628f99d7SShri Abhyankar idx = pj[j]; 2220628f99d7SShri Abhyankar if (idx != row) rs += PetscAbsScalar(rtmp11[idx]); 2221628f99d7SShri Abhyankar } 2222628f99d7SShri Abhyankar sctx.rs = rs; 2223d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,row);CHKERRQ(ierr); 222407b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2225628f99d7SShri Abhyankar 2226628f99d7SShri Abhyankar if (*pc2 != 0.0 || *pc3 != 0.0) { 2227628f99d7SShri Abhyankar mul2 = (*pc2)/(*pc1); 2228628f99d7SShri Abhyankar mul3 = (*pc3)/(*pc1); 2229628f99d7SShri Abhyankar *pc2 = mul2; 2230628f99d7SShri Abhyankar *pc3 = mul3; 2231628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2232628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2233628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2234628f99d7SShri Abhyankar idx = pj[j]; 2235628f99d7SShri Abhyankar tmp = rtmp11[idx]; 2236628f99d7SShri Abhyankar rtmp22[idx] -= mul2 * tmp; 2237628f99d7SShri Abhyankar rtmp33[idx] -= mul3 * tmp; 2238628f99d7SShri Abhyankar } 223928b1a77fSLisandro Dalcin ierr = PetscLogFlops(2+4*nz_tmp);CHKERRQ(ierr); 2240628f99d7SShri Abhyankar } 2241628f99d7SShri Abhyankar ++prow; 2242628f99d7SShri Abhyankar 2243628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2244628f99d7SShri Abhyankar pc3 = rtmp33 + prow; 2245628f99d7SShri Abhyankar sctx.pv = *pc2; 2246628f99d7SShri Abhyankar pj = bj + bi[prow]; 2247628f99d7SShri Abhyankar rs = 0.0; 2248628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2249628f99d7SShri Abhyankar idx = pj[j]; 2250628f99d7SShri Abhyankar if (idx != prow) rs += PetscAbsScalar(rtmp22[idx]); 2251628f99d7SShri Abhyankar } 2252628f99d7SShri Abhyankar sctx.rs = rs; 2253d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,row+1);CHKERRQ(ierr); 225407b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2255628f99d7SShri Abhyankar 2256628f99d7SShri Abhyankar if (*pc3 != 0.0) { 2257628f99d7SShri Abhyankar mul3 = (*pc3)/(*pc2); 2258628f99d7SShri Abhyankar *pc3 = mul3; 2259628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2260628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2261628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2262628f99d7SShri Abhyankar idx = pj[j]; 2263628f99d7SShri Abhyankar tmp = rtmp22[idx]; 2264628f99d7SShri Abhyankar rtmp33[idx] -= mul3 * tmp; 2265628f99d7SShri Abhyankar } 226628b1a77fSLisandro Dalcin ierr = PetscLogFlops(1+2*nz_tmp);CHKERRQ(ierr); 2267628f99d7SShri Abhyankar } 2268628f99d7SShri Abhyankar 2269628f99d7SShri Abhyankar pj = bj + bi[row]; 2270628f99d7SShri Abhyankar pc1 = ba + bi[row]; 2271628f99d7SShri Abhyankar pc2 = ba + bi[row+1]; 2272628f99d7SShri Abhyankar pc3 = ba + bi[row+2]; 2273628f99d7SShri Abhyankar 2274628f99d7SShri Abhyankar sctx.pv = rtmp33[row+2]; 2275628f99d7SShri Abhyankar rs = 0.0; 2276628f99d7SShri Abhyankar rtmp11[row] = 1.0/rtmp11[row]; 2277628f99d7SShri Abhyankar rtmp22[row+1] = 1.0/rtmp22[row+1]; 2278628f99d7SShri Abhyankar rtmp33[row+2] = 1.0/rtmp33[row+2]; 2279628f99d7SShri Abhyankar /* copy row entries from dense representation to sparse */ 2280628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2281628f99d7SShri Abhyankar idx = pj[j]; 2282628f99d7SShri Abhyankar pc1[j] = rtmp11[idx]; 2283628f99d7SShri Abhyankar pc2[j] = rtmp22[idx]; 2284628f99d7SShri Abhyankar pc3[j] = rtmp33[idx]; 2285628f99d7SShri Abhyankar if (idx != row+2) rs += PetscAbsScalar(pc3[j]); 2286628f99d7SShri Abhyankar } 2287628f99d7SShri Abhyankar 2288628f99d7SShri Abhyankar sctx.rs = rs; 2289d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,row+2);CHKERRQ(ierr); 229007b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2291628f99d7SShri Abhyankar break; 2292628f99d7SShri Abhyankar 2293628f99d7SShri Abhyankar default: 2294e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Node size not yet supported \n"); 2295628f99d7SShri Abhyankar } 2296628f99d7SShri Abhyankar row += nodesz; /* Update the row */ 2297628f99d7SShri Abhyankar } 2298628f99d7SShri Abhyankar endofwhile:; 229907b50cabSHong Zhang } while (sctx.newshift); 2300eb9baa12SBarry Smith ierr = PetscFree3(rtmp11,rtmp22,rtmp33);CHKERRQ(ierr); 2301628f99d7SShri Abhyankar ierr = PetscFree(tmp_vec2);CHKERRQ(ierr); 2302628f99d7SShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2303628f99d7SShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 2304628f99d7SShri Abhyankar ierr = ISRestoreIndices(iscol,&c);CHKERRQ(ierr); 23052205254eSKarl Rupp 2306d3ac4fa3SBarry Smith (B)->ops->solve = MatSolve_SeqAIJ_inplace; 2307628f99d7SShri Abhyankar /* do not set solve add, since MatSolve_Inode + Add is faster */ 2308628f99d7SShri Abhyankar C->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 2309628f99d7SShri Abhyankar C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 2310628f99d7SShri Abhyankar C->assembled = PETSC_TRUE; 2311628f99d7SShri Abhyankar C->preallocated = PETSC_TRUE; 2312628f99d7SShri Abhyankar if (sctx.nshift) { 2313f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 231457622a8eSBarry Smith ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %g, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,(double)sctx.shift_amount,(double)sctx.shift_fraction,(double)sctx.shift_top);CHKERRQ(ierr); 2315f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 231657622a8eSBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr); 2317628f99d7SShri Abhyankar } 2318628f99d7SShri Abhyankar } 2319628f99d7SShri Abhyankar ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 2320*a02bda8eSBarry Smith ierr = MatSeqAIJCheckInode(C);CHKERRQ(ierr); 2321628f99d7SShri Abhyankar PetscFunctionReturn(0); 2322628f99d7SShri Abhyankar } 2323628f99d7SShri Abhyankar 2324628f99d7SShri Abhyankar 2325019b515eSShri Abhyankar /* ----------------------------------------------------------- */ 2326019b515eSShri Abhyankar #undef __FUNCT__ 232735233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_Inode" 232835233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_Inode(Mat A,Vec bb,Vec xx) 2329019b515eSShri Abhyankar { 2330019b515eSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2331019b515eSShri Abhyankar IS iscol = a->col,isrow = a->row; 2332019b515eSShri Abhyankar PetscErrorCode ierr; 2333019b515eSShri Abhyankar const PetscInt *r,*c,*rout,*cout; 23348758e1faSBarry Smith PetscInt i,j,n = A->rmap->n; 23358758e1faSBarry Smith PetscInt node_max,row,nsz,aii,i0,i1,nz; 23368758e1faSBarry Smith const PetscInt *ai = a->i,*a_j = a->j,*ns,*vi,*ad,*aj; 2337019b515eSShri Abhyankar PetscScalar *x,*tmp,*tmps,tmp0,tmp1; 2338019b515eSShri Abhyankar PetscScalar sum1,sum2,sum3,sum4,sum5; 2339019b515eSShri Abhyankar const MatScalar *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa; 2340019b515eSShri Abhyankar const PetscScalar *b; 2341019b515eSShri Abhyankar 2342019b515eSShri Abhyankar PetscFunctionBegin; 2343e32f2f54SBarry Smith if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 2344019b515eSShri Abhyankar node_max = a->inode.node_count; 2345019b515eSShri Abhyankar ns = a->inode.size; /* Node Size array */ 2346019b515eSShri Abhyankar 23473649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 2348019b515eSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2349019b515eSShri Abhyankar tmp = a->solve_work; 2350019b515eSShri Abhyankar 2351019b515eSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 2352019b515eSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 2353019b515eSShri Abhyankar 2354019b515eSShri Abhyankar /* forward solve the lower triangular */ 2355019b515eSShri Abhyankar tmps = tmp; 2356019b515eSShri Abhyankar aa = a_a; 2357019b515eSShri Abhyankar aj = a_j; 2358019b515eSShri Abhyankar ad = a->diag; 2359019b515eSShri Abhyankar 2360019b515eSShri Abhyankar for (i = 0,row = 0; i< node_max; ++i) { 2361019b515eSShri Abhyankar nsz = ns[i]; 2362019b515eSShri Abhyankar aii = ai[row]; 2363019b515eSShri Abhyankar v1 = aa + aii; 2364019b515eSShri Abhyankar vi = aj + aii; 2365019b515eSShri Abhyankar nz = ai[row+1]- ai[row]; 2366019b515eSShri Abhyankar 236798991853SShri Abhyankar if (i < node_max-1) { 236898991853SShri Abhyankar /* Prefetch the indices for the next block */ 236950d8bf02SJed Brown PetscPrefetchBlock(aj+ai[row+nsz],ai[row+nsz+1]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); /* indices */ 237098991853SShri Abhyankar /* Prefetch the data for the next block */ 237150d8bf02SJed Brown PetscPrefetchBlock(aa+ai[row+nsz],ai[row+nsz+ns[i+1]]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); 237298991853SShri Abhyankar } 237398991853SShri Abhyankar 2374019b515eSShri Abhyankar switch (nsz) { /* Each loop in 'case' is unrolled */ 2375019b515eSShri Abhyankar case 1: 2376019b515eSShri Abhyankar sum1 = b[r[row]]; 2377019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2378019b515eSShri Abhyankar i0 = vi[j]; 2379019b515eSShri Abhyankar i1 = vi[j+1]; 2380019b515eSShri Abhyankar tmp0 = tmps[i0]; 2381019b515eSShri Abhyankar tmp1 = tmps[i1]; 2382019b515eSShri Abhyankar sum1 -= v1[j]*tmp0 + v1[j+1]*tmp1; 2383019b515eSShri Abhyankar } 2384019b515eSShri Abhyankar if (j == nz-1) { 2385019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2386019b515eSShri Abhyankar sum1 -= v1[j]*tmp0; 2387019b515eSShri Abhyankar } 2388019b515eSShri Abhyankar tmp[row++]=sum1; 2389019b515eSShri Abhyankar break; 2390019b515eSShri Abhyankar case 2: 2391019b515eSShri Abhyankar sum1 = b[r[row]]; 2392019b515eSShri Abhyankar sum2 = b[r[row+1]]; 2393019b515eSShri Abhyankar v2 = aa + ai[row+1]; 2394019b515eSShri Abhyankar 2395019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2396019b515eSShri Abhyankar i0 = vi[j]; 2397019b515eSShri Abhyankar i1 = vi[j+1]; 2398019b515eSShri Abhyankar tmp0 = tmps[i0]; 2399019b515eSShri Abhyankar tmp1 = tmps[i1]; 2400019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2401019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 2402019b515eSShri Abhyankar } 2403019b515eSShri Abhyankar if (j == nz-1) { 2404019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2405019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 2406019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 2407019b515eSShri Abhyankar } 2408019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 2409019b515eSShri Abhyankar tmp[row++]=sum1; 2410019b515eSShri Abhyankar tmp[row++]=sum2; 2411019b515eSShri Abhyankar break; 2412019b515eSShri Abhyankar case 3: 2413019b515eSShri Abhyankar sum1 = b[r[row]]; 2414019b515eSShri Abhyankar sum2 = b[r[row+1]]; 2415019b515eSShri Abhyankar sum3 = b[r[row+2]]; 2416019b515eSShri Abhyankar v2 = aa + ai[row+1]; 2417019b515eSShri Abhyankar v3 = aa + ai[row+2]; 2418019b515eSShri Abhyankar 2419019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2420019b515eSShri Abhyankar i0 = vi[j]; 2421019b515eSShri Abhyankar i1 = vi[j+1]; 2422019b515eSShri Abhyankar tmp0 = tmps[i0]; 2423019b515eSShri Abhyankar tmp1 = tmps[i1]; 2424019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2425019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 2426019b515eSShri Abhyankar sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1; 2427019b515eSShri Abhyankar } 2428019b515eSShri Abhyankar if (j == nz-1) { 2429019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2430019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 2431019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 2432019b515eSShri Abhyankar sum3 -= v3[j] *tmp0; 2433019b515eSShri Abhyankar } 2434019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 2435019b515eSShri Abhyankar sum3 -= v3[nz] * sum1; 2436019b515eSShri Abhyankar sum3 -= v3[nz+1] * sum2; 2437019b515eSShri Abhyankar tmp[row++]=sum1; 2438019b515eSShri Abhyankar tmp[row++]=sum2; 2439019b515eSShri Abhyankar tmp[row++]=sum3; 2440019b515eSShri Abhyankar break; 2441019b515eSShri Abhyankar 2442019b515eSShri Abhyankar case 4: 2443019b515eSShri Abhyankar sum1 = b[r[row]]; 2444019b515eSShri Abhyankar sum2 = b[r[row+1]]; 2445019b515eSShri Abhyankar sum3 = b[r[row+2]]; 2446019b515eSShri Abhyankar sum4 = b[r[row+3]]; 2447019b515eSShri Abhyankar v2 = aa + ai[row+1]; 2448019b515eSShri Abhyankar v3 = aa + ai[row+2]; 2449019b515eSShri Abhyankar v4 = aa + ai[row+3]; 2450019b515eSShri Abhyankar 2451019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2452019b515eSShri Abhyankar i0 = vi[j]; 2453019b515eSShri Abhyankar i1 = vi[j+1]; 2454019b515eSShri Abhyankar tmp0 = tmps[i0]; 2455019b515eSShri Abhyankar tmp1 = tmps[i1]; 2456019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2457019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 2458019b515eSShri Abhyankar sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1; 2459019b515eSShri Abhyankar sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1; 2460019b515eSShri Abhyankar } 2461019b515eSShri Abhyankar if (j == nz-1) { 2462019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2463019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 2464019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 2465019b515eSShri Abhyankar sum3 -= v3[j] *tmp0; 2466019b515eSShri Abhyankar sum4 -= v4[j] *tmp0; 2467019b515eSShri Abhyankar } 2468019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 2469019b515eSShri Abhyankar sum3 -= v3[nz] * sum1; 2470019b515eSShri Abhyankar sum4 -= v4[nz] * sum1; 2471019b515eSShri Abhyankar sum3 -= v3[nz+1] * sum2; 2472019b515eSShri Abhyankar sum4 -= v4[nz+1] * sum2; 2473019b515eSShri Abhyankar sum4 -= v4[nz+2] * sum3; 2474019b515eSShri Abhyankar 2475019b515eSShri Abhyankar tmp[row++]=sum1; 2476019b515eSShri Abhyankar tmp[row++]=sum2; 2477019b515eSShri Abhyankar tmp[row++]=sum3; 2478019b515eSShri Abhyankar tmp[row++]=sum4; 2479019b515eSShri Abhyankar break; 2480019b515eSShri Abhyankar case 5: 2481019b515eSShri Abhyankar sum1 = b[r[row]]; 2482019b515eSShri Abhyankar sum2 = b[r[row+1]]; 2483019b515eSShri Abhyankar sum3 = b[r[row+2]]; 2484019b515eSShri Abhyankar sum4 = b[r[row+3]]; 2485019b515eSShri Abhyankar sum5 = b[r[row+4]]; 2486019b515eSShri Abhyankar v2 = aa + ai[row+1]; 2487019b515eSShri Abhyankar v3 = aa + ai[row+2]; 2488019b515eSShri Abhyankar v4 = aa + ai[row+3]; 2489019b515eSShri Abhyankar v5 = aa + ai[row+4]; 2490019b515eSShri Abhyankar 2491019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2492019b515eSShri Abhyankar i0 = vi[j]; 2493019b515eSShri Abhyankar i1 = vi[j+1]; 2494019b515eSShri Abhyankar tmp0 = tmps[i0]; 2495019b515eSShri Abhyankar tmp1 = tmps[i1]; 2496019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2497019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 2498019b515eSShri Abhyankar sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1; 2499019b515eSShri Abhyankar sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1; 2500019b515eSShri Abhyankar sum5 -= v5[j] * tmp0 + v5[j+1] * tmp1; 2501019b515eSShri Abhyankar } 2502019b515eSShri Abhyankar if (j == nz-1) { 2503019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2504019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 2505019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 2506019b515eSShri Abhyankar sum3 -= v3[j] *tmp0; 2507019b515eSShri Abhyankar sum4 -= v4[j] *tmp0; 2508019b515eSShri Abhyankar sum5 -= v5[j] *tmp0; 2509019b515eSShri Abhyankar } 2510019b515eSShri Abhyankar 2511019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 2512019b515eSShri Abhyankar sum3 -= v3[nz] * sum1; 2513019b515eSShri Abhyankar sum4 -= v4[nz] * sum1; 2514019b515eSShri Abhyankar sum5 -= v5[nz] * sum1; 2515019b515eSShri Abhyankar sum3 -= v3[nz+1] * sum2; 2516019b515eSShri Abhyankar sum4 -= v4[nz+1] * sum2; 2517019b515eSShri Abhyankar sum5 -= v5[nz+1] * sum2; 2518019b515eSShri Abhyankar sum4 -= v4[nz+2] * sum3; 2519019b515eSShri Abhyankar sum5 -= v5[nz+2] * sum3; 2520019b515eSShri Abhyankar sum5 -= v5[nz+3] * sum4; 2521019b515eSShri Abhyankar 2522019b515eSShri Abhyankar tmp[row++]=sum1; 2523019b515eSShri Abhyankar tmp[row++]=sum2; 2524019b515eSShri Abhyankar tmp[row++]=sum3; 2525019b515eSShri Abhyankar tmp[row++]=sum4; 2526019b515eSShri Abhyankar tmp[row++]=sum5; 2527019b515eSShri Abhyankar break; 2528019b515eSShri Abhyankar default: 2529e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n"); 2530019b515eSShri Abhyankar } 2531019b515eSShri Abhyankar } 2532019b515eSShri Abhyankar /* backward solve the upper triangular */ 2533019b515eSShri Abhyankar for (i=node_max -1,row = n-1; i>=0; i--) { 2534019b515eSShri Abhyankar nsz = ns[i]; 2535019b515eSShri Abhyankar aii = ad[row+1] + 1; 2536019b515eSShri Abhyankar v1 = aa + aii; 2537019b515eSShri Abhyankar vi = aj + aii; 2538019b515eSShri Abhyankar nz = ad[row]- ad[row+1] - 1; 253998991853SShri Abhyankar 254098991853SShri Abhyankar if (i > 0) { 254198991853SShri Abhyankar /* Prefetch the indices for the next block */ 254250d8bf02SJed Brown PetscPrefetchBlock(aj+ad[row-nsz+1]+1,ad[row-nsz]-ad[row-nsz+1],0,PETSC_PREFETCH_HINT_NTA); 254398991853SShri Abhyankar /* Prefetch the data for the next block */ 254450d8bf02SJed Brown PetscPrefetchBlock(aa+ad[row-nsz+1]+1,ad[row-nsz-ns[i-1]+1]-ad[row-nsz+1],0,PETSC_PREFETCH_HINT_NTA); 254598991853SShri Abhyankar } 254698991853SShri Abhyankar 2547019b515eSShri Abhyankar switch (nsz) { /* Each loop in 'case' is unrolled */ 2548019b515eSShri Abhyankar case 1: 2549019b515eSShri Abhyankar sum1 = tmp[row]; 2550019b515eSShri Abhyankar 2551019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2552019b515eSShri Abhyankar i0 = vi[j]; 2553019b515eSShri Abhyankar i1 = vi[j+1]; 2554019b515eSShri Abhyankar tmp0 = tmps[i0]; 2555019b515eSShri Abhyankar tmp1 = tmps[i1]; 2556019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2557019b515eSShri Abhyankar } 2558019b515eSShri Abhyankar if (j == nz-1) { 2559019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2560019b515eSShri Abhyankar sum1 -= v1[j]*tmp0; 2561019b515eSShri Abhyankar } 2562019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 2563019b515eSShri Abhyankar break; 2564019b515eSShri Abhyankar case 2: 2565019b515eSShri Abhyankar sum1 = tmp[row]; 2566019b515eSShri Abhyankar sum2 = tmp[row-1]; 2567019b515eSShri Abhyankar v2 = aa + ad[row] + 1; 2568019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2569019b515eSShri Abhyankar i0 = vi[j]; 2570019b515eSShri Abhyankar i1 = vi[j+1]; 2571019b515eSShri Abhyankar tmp0 = tmps[i0]; 2572019b515eSShri Abhyankar tmp1 = tmps[i1]; 2573019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2574019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 2575019b515eSShri Abhyankar } 2576019b515eSShri Abhyankar if (j == nz-1) { 2577019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2578019b515eSShri Abhyankar sum1 -= v1[j]* tmp0; 2579019b515eSShri Abhyankar sum2 -= v2[j+1]* tmp0; 2580019b515eSShri Abhyankar } 2581019b515eSShri Abhyankar 2582019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 2583019b515eSShri Abhyankar sum2 -= v2[0] * tmp0; 2584019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 2585019b515eSShri Abhyankar break; 2586019b515eSShri Abhyankar case 3: 2587019b515eSShri Abhyankar sum1 = tmp[row]; 2588019b515eSShri Abhyankar sum2 = tmp[row -1]; 2589019b515eSShri Abhyankar sum3 = tmp[row -2]; 2590019b515eSShri Abhyankar v2 = aa + ad[row] + 1; 2591019b515eSShri Abhyankar v3 = aa + ad[row -1] + 1; 2592019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2593019b515eSShri Abhyankar i0 = vi[j]; 2594019b515eSShri Abhyankar i1 = vi[j+1]; 2595019b515eSShri Abhyankar tmp0 = tmps[i0]; 2596019b515eSShri Abhyankar tmp1 = tmps[i1]; 2597019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2598019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 2599019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1; 2600019b515eSShri Abhyankar } 2601019b515eSShri Abhyankar if (j== nz-1) { 2602019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2603019b515eSShri Abhyankar sum1 -= v1[j] * tmp0; 2604019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0; 2605019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0; 2606019b515eSShri Abhyankar } 2607019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 2608019b515eSShri Abhyankar sum2 -= v2[0]* tmp0; 2609019b515eSShri Abhyankar sum3 -= v3[1] * tmp0; 2610019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 2611019b515eSShri Abhyankar sum3 -= v3[0]* tmp0; 2612019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--; 2613019b515eSShri Abhyankar 2614019b515eSShri Abhyankar break; 2615019b515eSShri Abhyankar case 4: 2616019b515eSShri Abhyankar sum1 = tmp[row]; 2617019b515eSShri Abhyankar sum2 = tmp[row -1]; 2618019b515eSShri Abhyankar sum3 = tmp[row -2]; 2619019b515eSShri Abhyankar sum4 = tmp[row -3]; 2620019b515eSShri Abhyankar v2 = aa + ad[row]+1; 2621019b515eSShri Abhyankar v3 = aa + ad[row -1]+1; 2622019b515eSShri Abhyankar v4 = aa + ad[row -2]+1; 2623019b515eSShri Abhyankar 2624019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2625019b515eSShri Abhyankar i0 = vi[j]; 2626019b515eSShri Abhyankar i1 = vi[j+1]; 2627019b515eSShri Abhyankar tmp0 = tmps[i0]; 2628019b515eSShri Abhyankar tmp1 = tmps[i1]; 2629019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2630019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 2631019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1; 2632019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1; 2633019b515eSShri Abhyankar } 2634019b515eSShri Abhyankar if (j== nz-1) { 2635019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2636019b515eSShri Abhyankar sum1 -= v1[j] * tmp0; 2637019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0; 2638019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0; 2639019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0; 2640019b515eSShri Abhyankar } 2641019b515eSShri Abhyankar 2642019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 2643019b515eSShri Abhyankar sum2 -= v2[0] * tmp0; 2644019b515eSShri Abhyankar sum3 -= v3[1] * tmp0; 2645019b515eSShri Abhyankar sum4 -= v4[2] * tmp0; 2646019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 2647019b515eSShri Abhyankar sum3 -= v3[0] * tmp0; 2648019b515eSShri Abhyankar sum4 -= v4[1] * tmp0; 2649019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--; 2650019b515eSShri Abhyankar sum4 -= v4[0] * tmp0; 2651019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--; 2652019b515eSShri Abhyankar break; 2653019b515eSShri Abhyankar case 5: 2654019b515eSShri Abhyankar sum1 = tmp[row]; 2655019b515eSShri Abhyankar sum2 = tmp[row -1]; 2656019b515eSShri Abhyankar sum3 = tmp[row -2]; 2657019b515eSShri Abhyankar sum4 = tmp[row -3]; 2658019b515eSShri Abhyankar sum5 = tmp[row -4]; 2659019b515eSShri Abhyankar v2 = aa + ad[row]+1; 2660019b515eSShri Abhyankar v3 = aa + ad[row -1]+1; 2661019b515eSShri Abhyankar v4 = aa + ad[row -2]+1; 2662019b515eSShri Abhyankar v5 = aa + ad[row -3]+1; 2663019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2664019b515eSShri Abhyankar i0 = vi[j]; 2665019b515eSShri Abhyankar i1 = vi[j+1]; 2666019b515eSShri Abhyankar tmp0 = tmps[i0]; 2667019b515eSShri Abhyankar tmp1 = tmps[i1]; 2668019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2669019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 2670019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1; 2671019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1; 2672019b515eSShri Abhyankar sum5 -= v5[j+4] * tmp0 + v5[j+5] * tmp1; 2673019b515eSShri Abhyankar } 2674019b515eSShri Abhyankar if (j==nz-1) { 2675019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2676019b515eSShri Abhyankar sum1 -= v1[j] * tmp0; 2677019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0; 2678019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0; 2679019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0; 2680019b515eSShri Abhyankar sum5 -= v5[j+4] * tmp0; 2681019b515eSShri Abhyankar } 2682019b515eSShri Abhyankar 2683019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 2684019b515eSShri Abhyankar sum2 -= v2[0] * tmp0; 2685019b515eSShri Abhyankar sum3 -= v3[1] * tmp0; 2686019b515eSShri Abhyankar sum4 -= v4[2] * tmp0; 2687019b515eSShri Abhyankar sum5 -= v5[3] * tmp0; 2688019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 2689019b515eSShri Abhyankar sum3 -= v3[0] * tmp0; 2690019b515eSShri Abhyankar sum4 -= v4[1] * tmp0; 2691019b515eSShri Abhyankar sum5 -= v5[2] * tmp0; 2692019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--; 2693019b515eSShri Abhyankar sum4 -= v4[0] * tmp0; 2694019b515eSShri Abhyankar sum5 -= v5[1] * tmp0; 2695019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--; 2696019b515eSShri Abhyankar sum5 -= v5[0] * tmp0; 2697019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum5*v5[nz+4]; row--; 2698019b515eSShri Abhyankar break; 2699019b515eSShri Abhyankar default: 2700e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n"); 2701019b515eSShri Abhyankar } 2702019b515eSShri Abhyankar } 2703019b515eSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 2704019b515eSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 27053649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 2706019b515eSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2707019b515eSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 2708019b515eSShri Abhyankar PetscFunctionReturn(0); 2709019b515eSShri Abhyankar } 2710019b515eSShri Abhyankar 2711019b515eSShri Abhyankar 27124c1414c8SBarry Smith /* 27134c1414c8SBarry Smith Makes a longer coloring[] array and calls the usual code with that 27144c1414c8SBarry Smith */ 27154c1414c8SBarry Smith #undef __FUNCT__ 27164108e4d5SBarry Smith #define __FUNCT__ "MatColoringPatch_SeqAIJ_Inode" 27174108e4d5SBarry Smith PetscErrorCode MatColoringPatch_SeqAIJ_Inode(Mat mat,PetscInt ncolors,PetscInt nin,ISColoringValue coloring[],ISColoring *iscoloring) 27184c1414c8SBarry Smith { 27194c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)mat->data; 27204c1414c8SBarry Smith PetscErrorCode ierr; 2721d0f46423SBarry Smith PetscInt n = mat->cmap->n,m = a->inode.node_count,j,*ns = a->inode.size,row; 27224c1414c8SBarry Smith PetscInt *colorused,i; 27234c1414c8SBarry Smith ISColoringValue *newcolor; 27244c1414c8SBarry Smith 27254c1414c8SBarry Smith PetscFunctionBegin; 2726854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&newcolor);CHKERRQ(ierr); 27274c1414c8SBarry Smith /* loop over inodes, marking a color for each column*/ 27284c1414c8SBarry Smith row = 0; 27294c1414c8SBarry Smith for (i=0; i<m; i++) { 27304c1414c8SBarry Smith for (j=0; j<ns[i]; j++) { 27314c1414c8SBarry Smith newcolor[row++] = coloring[i] + j*ncolors; 27324c1414c8SBarry Smith } 27334c1414c8SBarry Smith } 27344c1414c8SBarry Smith 27354c1414c8SBarry Smith /* eliminate unneeded colors */ 27361795a4d1SJed Brown ierr = PetscCalloc1(5*ncolors,&colorused);CHKERRQ(ierr); 27374c1414c8SBarry Smith for (i=0; i<n; i++) { 27384c1414c8SBarry Smith colorused[newcolor[i]] = 1; 27394c1414c8SBarry Smith } 27404c1414c8SBarry Smith 27414c1414c8SBarry Smith for (i=1; i<5*ncolors; i++) { 27424c1414c8SBarry Smith colorused[i] += colorused[i-1]; 27434c1414c8SBarry Smith } 27444c1414c8SBarry Smith ncolors = colorused[5*ncolors-1]; 27454c1414c8SBarry Smith for (i=0; i<n; i++) { 274657815681SSatish Balay newcolor[i] = colorused[newcolor[i]]-1; 27474c1414c8SBarry Smith } 27484c1414c8SBarry Smith ierr = PetscFree(colorused);CHKERRQ(ierr); 2749aaf3ff59SMatthew G. Knepley ierr = ISColoringCreate(PetscObjectComm((PetscObject)mat),ncolors,n,newcolor,PETSC_OWN_POINTER,iscoloring);CHKERRQ(ierr); 27504c1414c8SBarry Smith ierr = PetscFree(coloring);CHKERRQ(ierr); 27514c1414c8SBarry Smith PetscFunctionReturn(0); 27524c1414c8SBarry Smith } 27534c1414c8SBarry Smith 275406873bf2SBarry Smith #include <petsc-private/kernels/blockinvert.h> 27552af78befSBarry Smith 27562af78befSBarry Smith #undef __FUNCT__ 27574108e4d5SBarry Smith #define __FUNCT__ "MatSOR_SeqAIJ_Inode" 27584108e4d5SBarry Smith PetscErrorCode MatSOR_SeqAIJ_Inode(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx) 27592af78befSBarry Smith { 27602af78befSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 27617aaeff0aSMatthew G. Knepley PetscScalar sum1 = 0.0,sum2 = 0.0,sum3 = 0.0,sum4 = 0.0,sum5 = 0.0,tmp0,tmp1,tmp2,tmp3; 27625850ef23SBarry Smith MatScalar *ibdiag,*bdiag,work[25],*t; 2763a8b09249SBarry Smith PetscScalar *x,tmp4,tmp5,x1,x2,x3,x4,x5; 27647aaeff0aSMatthew G. Knepley const MatScalar *v = a->a,*v1 = NULL,*v2 = NULL,*v3 = NULL,*v4 = NULL,*v5 = NULL; 27655850ef23SBarry Smith const PetscScalar *xb, *b; 276662bba022SBarry Smith PetscReal zeropivot = 1.0e-15, shift = 0.0; 27672af78befSBarry Smith PetscErrorCode ierr; 27688758e1faSBarry Smith PetscInt n,m = a->inode.node_count,cnt = 0,i,j,row,i1,i2; 27698758e1faSBarry Smith PetscInt sz,k,ipvt[5]; 27708758e1faSBarry Smith const PetscInt *sizes = a->inode.size,*idx,*diag = a->diag,*ii = a->i; 27712af78befSBarry Smith 27722af78befSBarry Smith PetscFunctionBegin; 2773e32f2f54SBarry Smith if (omega != 1.0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for omega != 1.0; use -mat_no_inode"); 2774e32f2f54SBarry Smith if (fshift != 0.0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for fshift != 0.0; use -mat_no_inode"); 27752af78befSBarry Smith 277671f1c65dSBarry Smith if (!a->inode.ibdiagvalid) { 27772af78befSBarry Smith if (!a->inode.ibdiag) { 27782af78befSBarry Smith /* calculate space needed for diagonal blocks */ 27792af78befSBarry Smith for (i=0; i<m; i++) { 27802af78befSBarry Smith cnt += sizes[i]*sizes[i]; 27812af78befSBarry Smith } 2782f0d39aaaSBarry Smith a->inode.bdiagsize = cnt; 27832205254eSKarl Rupp 2784dcca6d9dSJed Brown ierr = PetscMalloc3(cnt,&a->inode.ibdiag,cnt,&a->inode.bdiag,A->rmap->n,&a->inode.ssor_work);CHKERRQ(ierr); 278571f1c65dSBarry Smith } 278671f1c65dSBarry Smith 278771f1c65dSBarry Smith /* copy over the diagonal blocks and invert them */ 27882af78befSBarry Smith ibdiag = a->inode.ibdiag; 27892af78befSBarry Smith bdiag = a->inode.bdiag; 27902af78befSBarry Smith cnt = 0; 27912af78befSBarry Smith for (i=0, row = 0; i<m; i++) { 27922af78befSBarry Smith for (j=0; j<sizes[i]; j++) { 2793f0d39aaaSBarry Smith for (k=0; k<sizes[i]; k++) { 2794f0d39aaaSBarry Smith bdiag[cnt+k*sizes[i]+j] = v[diag[row+j] - j + k]; 2795f0d39aaaSBarry Smith } 27962af78befSBarry Smith } 2797a77337e4SBarry Smith ierr = PetscMemcpy(ibdiag+cnt,bdiag+cnt,sizes[i]*sizes[i]*sizeof(MatScalar));CHKERRQ(ierr); 27982af78befSBarry Smith 27992af78befSBarry Smith switch (sizes[i]) { 28002af78befSBarry Smith case 1: 28012af78befSBarry Smith /* Create matrix data structure */ 2802e32f2f54SBarry Smith if (PetscAbsScalar(ibdiag[cnt]) < zeropivot) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot on row %D",row); 280364c62002SMatthew Knepley ibdiag[cnt] = 1.0/ibdiag[cnt]; 28042af78befSBarry Smith break; 28052af78befSBarry Smith case 2: 280696b95a6bSBarry Smith ierr = PetscKernel_A_gets_inverse_A_2(ibdiag+cnt,shift);CHKERRQ(ierr); 28072af78befSBarry Smith break; 28082af78befSBarry Smith case 3: 280996b95a6bSBarry Smith ierr = PetscKernel_A_gets_inverse_A_3(ibdiag+cnt,shift);CHKERRQ(ierr); 28102af78befSBarry Smith break; 28112af78befSBarry Smith case 4: 281296b95a6bSBarry Smith ierr = PetscKernel_A_gets_inverse_A_4(ibdiag+cnt,shift);CHKERRQ(ierr); 28132af78befSBarry Smith break; 28142af78befSBarry Smith case 5: 281596b95a6bSBarry Smith ierr = PetscKernel_A_gets_inverse_A_5(ibdiag+cnt,ipvt,work,shift);CHKERRQ(ierr); 28162af78befSBarry Smith break; 28172af78befSBarry Smith default: 2818e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 28192af78befSBarry Smith } 28202af78befSBarry Smith cnt += sizes[i]*sizes[i]; 28212af78befSBarry Smith row += sizes[i]; 28222af78befSBarry Smith } 282371f1c65dSBarry Smith a->inode.ibdiagvalid = PETSC_TRUE; 28242af78befSBarry Smith } 28252af78befSBarry Smith ibdiag = a->inode.ibdiag; 28262af78befSBarry Smith bdiag = a->inode.bdiag; 28275850ef23SBarry Smith t = a->inode.ssor_work; 28282af78befSBarry Smith 28292af78befSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 28303649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 28315850ef23SBarry Smith /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */ 28325850ef23SBarry Smith if (flag & SOR_ZERO_INITIAL_GUESS) { 28332af78befSBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) { 28342af78befSBarry Smith 28358862d2efSBarry Smith for (i=0, row=0; i<m; i++) { 28368862d2efSBarry Smith sz = diag[row] - ii[row]; 28378862d2efSBarry Smith v1 = a->a + ii[row]; 28388862d2efSBarry Smith idx = a->j + ii[row]; 28398862d2efSBarry Smith 28404108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 28418862d2efSBarry Smith switch (sizes[i]) { 28428862d2efSBarry Smith case 1: 28438862d2efSBarry Smith 28448862d2efSBarry Smith sum1 = b[row]; 28458862d2efSBarry Smith for (n = 0; n<sz-1; n+=2) { 28468862d2efSBarry Smith i1 = idx[0]; 28478862d2efSBarry Smith i2 = idx[1]; 28488862d2efSBarry Smith idx += 2; 28498862d2efSBarry Smith tmp0 = x[i1]; 28508862d2efSBarry Smith tmp1 = x[i2]; 28518862d2efSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 28528862d2efSBarry Smith } 28538862d2efSBarry Smith 28548862d2efSBarry Smith if (n == sz-1) { 2855f0d39aaaSBarry Smith tmp0 = x[*idx]; 2856f0d39aaaSBarry Smith sum1 -= *v1 * tmp0; 28578862d2efSBarry Smith } 28585850ef23SBarry Smith t[row] = sum1; 28598862d2efSBarry Smith x[row++] = sum1*(*ibdiag++); 28608862d2efSBarry Smith break; 2861f0d39aaaSBarry Smith case 2: 2862f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 2863f0d39aaaSBarry Smith sum1 = b[row]; 2864f0d39aaaSBarry Smith sum2 = b[row+1]; 2865f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 2866f0d39aaaSBarry Smith i1 = idx[0]; 2867f0d39aaaSBarry Smith i2 = idx[1]; 2868f0d39aaaSBarry Smith idx += 2; 2869f0d39aaaSBarry Smith tmp0 = x[i1]; 2870f0d39aaaSBarry Smith tmp1 = x[i2]; 2871f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 2872f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 2873f0d39aaaSBarry Smith } 2874f0d39aaaSBarry Smith 2875f0d39aaaSBarry Smith if (n == sz-1) { 2876f0d39aaaSBarry Smith tmp0 = x[*idx]; 2877f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 2878f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 2879f0d39aaaSBarry Smith } 28805850ef23SBarry Smith t[row] = sum1; 28815850ef23SBarry Smith t[row+1] = sum2; 2882f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[2]; 2883f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[3]; 2884f0d39aaaSBarry Smith ibdiag += 4; 2885f0d39aaaSBarry Smith break; 2886f0d39aaaSBarry Smith case 3: 2887f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 2888f0d39aaaSBarry Smith v3 = a->a + ii[row+2]; 2889f0d39aaaSBarry Smith sum1 = b[row]; 2890f0d39aaaSBarry Smith sum2 = b[row+1]; 2891f0d39aaaSBarry Smith sum3 = b[row+2]; 2892f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 2893f0d39aaaSBarry Smith i1 = idx[0]; 2894f0d39aaaSBarry Smith i2 = idx[1]; 2895f0d39aaaSBarry Smith idx += 2; 2896f0d39aaaSBarry Smith tmp0 = x[i1]; 2897f0d39aaaSBarry Smith tmp1 = x[i2]; 2898f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 2899f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 2900f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 2901f0d39aaaSBarry Smith } 2902f0d39aaaSBarry Smith 2903f0d39aaaSBarry Smith if (n == sz-1) { 2904f0d39aaaSBarry Smith tmp0 = x[*idx]; 2905f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 2906f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 2907f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0; 2908f0d39aaaSBarry Smith } 29095850ef23SBarry Smith t[row] = sum1; 29105850ef23SBarry Smith t[row+1] = sum2; 29115850ef23SBarry Smith t[row+2] = sum3; 2912f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6]; 2913f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7]; 2914f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8]; 2915f0d39aaaSBarry Smith ibdiag += 9; 2916f0d39aaaSBarry Smith break; 2917f0d39aaaSBarry Smith case 4: 2918f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 2919f0d39aaaSBarry Smith v3 = a->a + ii[row+2]; 2920f0d39aaaSBarry Smith v4 = a->a + ii[row+3]; 2921f0d39aaaSBarry Smith sum1 = b[row]; 2922f0d39aaaSBarry Smith sum2 = b[row+1]; 2923f0d39aaaSBarry Smith sum3 = b[row+2]; 2924f0d39aaaSBarry Smith sum4 = b[row+3]; 2925f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 2926f0d39aaaSBarry Smith i1 = idx[0]; 2927f0d39aaaSBarry Smith i2 = idx[1]; 2928f0d39aaaSBarry Smith idx += 2; 2929f0d39aaaSBarry Smith tmp0 = x[i1]; 2930f0d39aaaSBarry Smith tmp1 = x[i2]; 2931f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 2932f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 2933f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 2934f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 2935f0d39aaaSBarry Smith } 2936f0d39aaaSBarry Smith 2937f0d39aaaSBarry Smith if (n == sz-1) { 2938f0d39aaaSBarry Smith tmp0 = x[*idx]; 2939f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 2940f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 2941f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0; 2942f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0; 2943f0d39aaaSBarry Smith } 29445850ef23SBarry Smith t[row] = sum1; 29455850ef23SBarry Smith t[row+1] = sum2; 29465850ef23SBarry Smith t[row+2] = sum3; 29475850ef23SBarry Smith t[row+3] = sum4; 2948f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12]; 2949f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13]; 2950f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14]; 2951f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15]; 2952f0d39aaaSBarry Smith ibdiag += 16; 2953f0d39aaaSBarry Smith break; 2954f0d39aaaSBarry Smith case 5: 2955f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 2956f0d39aaaSBarry Smith v3 = a->a + ii[row+2]; 2957f0d39aaaSBarry Smith v4 = a->a + ii[row+3]; 2958f0d39aaaSBarry Smith v5 = a->a + ii[row+4]; 2959f0d39aaaSBarry Smith sum1 = b[row]; 2960f0d39aaaSBarry Smith sum2 = b[row+1]; 2961f0d39aaaSBarry Smith sum3 = b[row+2]; 2962f0d39aaaSBarry Smith sum4 = b[row+3]; 2963f0d39aaaSBarry Smith sum5 = b[row+4]; 2964f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 2965f0d39aaaSBarry Smith i1 = idx[0]; 2966f0d39aaaSBarry Smith i2 = idx[1]; 2967f0d39aaaSBarry Smith idx += 2; 2968f0d39aaaSBarry Smith tmp0 = x[i1]; 2969f0d39aaaSBarry Smith tmp1 = x[i2]; 2970f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 2971f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 2972f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 2973f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 2974f0d39aaaSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 2975f0d39aaaSBarry Smith } 2976f0d39aaaSBarry Smith 2977f0d39aaaSBarry Smith if (n == sz-1) { 2978f0d39aaaSBarry Smith tmp0 = x[*idx]; 2979f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 2980f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 2981f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0; 2982f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0; 2983f0d39aaaSBarry Smith sum5 -= v5[0] * tmp0; 2984f0d39aaaSBarry Smith } 29855850ef23SBarry Smith t[row] = sum1; 29865850ef23SBarry Smith t[row+1] = sum2; 29875850ef23SBarry Smith t[row+2] = sum3; 29885850ef23SBarry Smith t[row+3] = sum4; 29895850ef23SBarry Smith t[row+4] = sum5; 2990f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20]; 2991f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21]; 2992f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22]; 2993f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23]; 2994f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24]; 2995f0d39aaaSBarry Smith ibdiag += 25; 2996f0d39aaaSBarry Smith break; 29978862d2efSBarry Smith default: 2998e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 29998862d2efSBarry Smith } 30002af78befSBarry Smith } 30012af78befSBarry Smith 30025850ef23SBarry Smith xb = t; 30032af78befSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 30042af78befSBarry Smith } else xb = b; 30052af78befSBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) { 30062af78befSBarry Smith 3007f0d39aaaSBarry Smith ibdiag = a->inode.ibdiag+a->inode.bdiagsize; 3008d0f46423SBarry Smith for (i=m-1, row=A->rmap->n-1; i>=0; i--) { 3009f0d39aaaSBarry Smith ibdiag -= sizes[i]*sizes[i]; 30108862d2efSBarry Smith sz = ii[row+1] - diag[row] - 1; 30118862d2efSBarry Smith v1 = a->a + diag[row] + 1; 30128862d2efSBarry Smith idx = a->j + diag[row] + 1; 30132af78befSBarry Smith 30144108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 30158862d2efSBarry Smith switch (sizes[i]) { 30168862d2efSBarry Smith case 1: 30178862d2efSBarry Smith 30188862d2efSBarry Smith sum1 = xb[row]; 30198862d2efSBarry Smith for (n = 0; n<sz-1; n+=2) { 30208862d2efSBarry Smith i1 = idx[0]; 30218862d2efSBarry Smith i2 = idx[1]; 30228862d2efSBarry Smith idx += 2; 30238862d2efSBarry Smith tmp0 = x[i1]; 30248862d2efSBarry Smith tmp1 = x[i2]; 30258862d2efSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 30268862d2efSBarry Smith } 30278862d2efSBarry Smith 30288862d2efSBarry Smith if (n == sz-1) { 3029f0d39aaaSBarry Smith tmp0 = x[*idx]; 3030f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 30318862d2efSBarry Smith } 3032f0d39aaaSBarry Smith x[row--] = sum1*(*ibdiag); 3033f0d39aaaSBarry Smith break; 3034f0d39aaaSBarry Smith 3035f0d39aaaSBarry Smith case 2: 3036f0d39aaaSBarry Smith 3037f0d39aaaSBarry Smith sum1 = xb[row]; 3038f0d39aaaSBarry Smith sum2 = xb[row-1]; 3039f0d39aaaSBarry Smith /* note that sum1 is associated with the second of the two rows */ 3040f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 3041f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 3042f0d39aaaSBarry Smith i1 = idx[0]; 3043f0d39aaaSBarry Smith i2 = idx[1]; 3044f0d39aaaSBarry Smith idx += 2; 3045f0d39aaaSBarry Smith tmp0 = x[i1]; 3046f0d39aaaSBarry Smith tmp1 = x[i2]; 3047f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3048f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3049f0d39aaaSBarry Smith } 3050f0d39aaaSBarry Smith 3051f0d39aaaSBarry Smith if (n == sz-1) { 3052f0d39aaaSBarry Smith tmp0 = x[*idx]; 3053f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 3054f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 3055f0d39aaaSBarry Smith } 3056f0d39aaaSBarry Smith x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3]; 3057f0d39aaaSBarry Smith x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2]; 3058f0d39aaaSBarry Smith break; 3059f0d39aaaSBarry Smith case 3: 3060f0d39aaaSBarry Smith 3061f0d39aaaSBarry Smith sum1 = xb[row]; 3062f0d39aaaSBarry Smith sum2 = xb[row-1]; 3063f0d39aaaSBarry Smith sum3 = xb[row-2]; 3064f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 3065f0d39aaaSBarry Smith v3 = a->a + diag[row-2] + 3; 3066f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 3067f0d39aaaSBarry Smith i1 = idx[0]; 3068f0d39aaaSBarry Smith i2 = idx[1]; 3069f0d39aaaSBarry Smith idx += 2; 3070f0d39aaaSBarry Smith tmp0 = x[i1]; 3071f0d39aaaSBarry Smith tmp1 = x[i2]; 3072f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3073f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3074f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3075f0d39aaaSBarry Smith } 3076f0d39aaaSBarry Smith 3077f0d39aaaSBarry Smith if (n == sz-1) { 3078f0d39aaaSBarry Smith tmp0 = x[*idx]; 3079f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 3080f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 3081f0d39aaaSBarry Smith sum3 -= *v3*tmp0; 3082f0d39aaaSBarry Smith } 3083f0d39aaaSBarry Smith x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8]; 3084f0d39aaaSBarry Smith x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7]; 3085f0d39aaaSBarry Smith x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6]; 3086f0d39aaaSBarry Smith break; 3087f0d39aaaSBarry Smith case 4: 3088f0d39aaaSBarry Smith 3089f0d39aaaSBarry Smith sum1 = xb[row]; 3090f0d39aaaSBarry Smith sum2 = xb[row-1]; 3091f0d39aaaSBarry Smith sum3 = xb[row-2]; 3092f0d39aaaSBarry Smith sum4 = xb[row-3]; 3093f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 3094f0d39aaaSBarry Smith v3 = a->a + diag[row-2] + 3; 3095f0d39aaaSBarry Smith v4 = a->a + diag[row-3] + 4; 3096f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 3097f0d39aaaSBarry Smith i1 = idx[0]; 3098f0d39aaaSBarry Smith i2 = idx[1]; 3099f0d39aaaSBarry Smith idx += 2; 3100f0d39aaaSBarry Smith tmp0 = x[i1]; 3101f0d39aaaSBarry Smith tmp1 = x[i2]; 3102f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3103f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3104f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3105f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 3106f0d39aaaSBarry Smith } 3107f0d39aaaSBarry Smith 3108f0d39aaaSBarry Smith if (n == sz-1) { 3109f0d39aaaSBarry Smith tmp0 = x[*idx]; 3110f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 3111f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 3112f0d39aaaSBarry Smith sum3 -= *v3*tmp0; 3113f0d39aaaSBarry Smith sum4 -= *v4*tmp0; 3114f0d39aaaSBarry Smith } 3115f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15]; 3116f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14]; 3117f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13]; 3118f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12]; 3119f0d39aaaSBarry Smith break; 3120f0d39aaaSBarry Smith case 5: 3121f0d39aaaSBarry Smith 3122f0d39aaaSBarry Smith sum1 = xb[row]; 3123f0d39aaaSBarry Smith sum2 = xb[row-1]; 3124f0d39aaaSBarry Smith sum3 = xb[row-2]; 3125f0d39aaaSBarry Smith sum4 = xb[row-3]; 3126f0d39aaaSBarry Smith sum5 = xb[row-4]; 3127f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 3128f0d39aaaSBarry Smith v3 = a->a + diag[row-2] + 3; 3129f0d39aaaSBarry Smith v4 = a->a + diag[row-3] + 4; 3130f0d39aaaSBarry Smith v5 = a->a + diag[row-4] + 5; 3131f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 3132f0d39aaaSBarry Smith i1 = idx[0]; 3133f0d39aaaSBarry Smith i2 = idx[1]; 3134f0d39aaaSBarry Smith idx += 2; 3135f0d39aaaSBarry Smith tmp0 = x[i1]; 3136f0d39aaaSBarry Smith tmp1 = x[i2]; 3137f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3138f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3139f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3140f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 3141f0d39aaaSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 3142f0d39aaaSBarry Smith } 3143f0d39aaaSBarry Smith 3144f0d39aaaSBarry Smith if (n == sz-1) { 3145f0d39aaaSBarry Smith tmp0 = x[*idx]; 3146f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 3147f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 3148f0d39aaaSBarry Smith sum3 -= *v3*tmp0; 3149f0d39aaaSBarry Smith sum4 -= *v4*tmp0; 3150f0d39aaaSBarry Smith sum5 -= *v5*tmp0; 3151f0d39aaaSBarry Smith } 3152f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24]; 3153f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23]; 3154f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22]; 3155f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21]; 3156f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20]; 31578862d2efSBarry Smith break; 31588862d2efSBarry Smith default: 3159e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 31608862d2efSBarry Smith } 31612af78befSBarry Smith } 31622af78befSBarry Smith 31632af78befSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 31642af78befSBarry Smith } 31652af78befSBarry Smith its--; 31665850ef23SBarry Smith } 31675850ef23SBarry Smith while (its--) { 31685850ef23SBarry Smith 31695850ef23SBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) { 3170d876e2b0SMark Adams for (i=0, row=0, ibdiag = a->inode.ibdiag; 3171d876e2b0SMark Adams i<m; 3172d876e2b0SMark Adams row += sizes[i], ibdiag += sizes[i]*sizes[i], i++) { 31735850ef23SBarry Smith 3174d876e2b0SMark Adams sz = diag[row] - ii[row]; 31755850ef23SBarry Smith v1 = a->a + ii[row]; 31765850ef23SBarry Smith idx = a->j + ii[row]; 31775850ef23SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 31785850ef23SBarry Smith switch (sizes[i]) { 31795850ef23SBarry Smith case 1: 31805850ef23SBarry Smith sum1 = b[row]; 31815850ef23SBarry Smith for (n = 0; n<sz-1; n+=2) { 31825850ef23SBarry Smith i1 = idx[0]; 31835850ef23SBarry Smith i2 = idx[1]; 31845850ef23SBarry Smith idx += 2; 31855850ef23SBarry Smith tmp0 = x[i1]; 31865850ef23SBarry Smith tmp1 = x[i2]; 31875850ef23SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 31885850ef23SBarry Smith } 31895850ef23SBarry Smith if (n == sz-1) { 3190d876e2b0SMark Adams tmp0 = x[*idx++]; 3191d876e2b0SMark Adams sum1 -= *v1 * tmp0; 3192d876e2b0SMark Adams v1++; 3193d876e2b0SMark Adams } 3194d876e2b0SMark Adams t[row] = sum1; 3195d876e2b0SMark Adams sz = ii[row+1] - diag[row] - 1; 3196d876e2b0SMark Adams idx = a->j + diag[row] + 1; 3197d876e2b0SMark Adams v1 += 1; 3198d876e2b0SMark Adams for (n = 0; n<sz-1; n+=2) { 3199d876e2b0SMark Adams i1 = idx[0]; 3200d876e2b0SMark Adams i2 = idx[1]; 3201d876e2b0SMark Adams idx += 2; 3202d876e2b0SMark Adams tmp0 = x[i1]; 3203d876e2b0SMark Adams tmp1 = x[i2]; 3204d876e2b0SMark Adams sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3205d876e2b0SMark Adams } 3206d876e2b0SMark Adams if (n == sz-1) { 3207d876e2b0SMark Adams tmp0 = x[*idx++]; 32085850ef23SBarry Smith sum1 -= *v1 * tmp0; 32095850ef23SBarry Smith } 32105850ef23SBarry Smith /* in MatSOR_SeqAIJ this line would be 32115850ef23SBarry Smith * 32125850ef23SBarry Smith * x[row] = (1-omega)*x[row]+(sum1+(*bdiag++)*x[row])*(*ibdiag++); 32135850ef23SBarry Smith * 32145850ef23SBarry Smith * but omega == 1, so this becomes 32155850ef23SBarry Smith * 3216d876e2b0SMark Adams * x[row] = sum1*(*ibdiag++); 32175850ef23SBarry Smith * 32185850ef23SBarry Smith */ 3219d876e2b0SMark Adams x[row] = sum1*(*ibdiag); 32205850ef23SBarry Smith break; 32215850ef23SBarry Smith case 2: 32225850ef23SBarry Smith v2 = a->a + ii[row+1]; 32235850ef23SBarry Smith sum1 = b[row]; 32245850ef23SBarry Smith sum2 = b[row+1]; 32255850ef23SBarry Smith for (n = 0; n<sz-1; n+=2) { 32265850ef23SBarry Smith i1 = idx[0]; 32275850ef23SBarry Smith i2 = idx[1]; 32285850ef23SBarry Smith idx += 2; 32295850ef23SBarry Smith tmp0 = x[i1]; 32305850ef23SBarry Smith tmp1 = x[i2]; 32315850ef23SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 32325850ef23SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 32335850ef23SBarry Smith } 3234d876e2b0SMark Adams if (n == sz-1) { 3235d876e2b0SMark Adams tmp0 = x[*idx++]; 3236d876e2b0SMark Adams sum1 -= v1[0] * tmp0; 3237d876e2b0SMark Adams sum2 -= v2[0] * tmp0; 3238d876e2b0SMark Adams v1++; v2++; 3239d876e2b0SMark Adams } 3240d876e2b0SMark Adams t[row] = sum1; 3241d876e2b0SMark Adams t[row+1] = sum2; 3242d876e2b0SMark Adams sz = ii[row+1] - diag[row] - 2; 3243d876e2b0SMark Adams idx = a->j + diag[row] + 2; 3244d876e2b0SMark Adams v1 += 2; 3245d876e2b0SMark Adams v2 += 2; 3246d876e2b0SMark Adams for (n = 0; n<sz-1; n+=2) { 3247d876e2b0SMark Adams i1 = idx[0]; 3248d876e2b0SMark Adams i2 = idx[1]; 3249d876e2b0SMark Adams idx += 2; 3250d876e2b0SMark Adams tmp0 = x[i1]; 3251d876e2b0SMark Adams tmp1 = x[i2]; 3252d876e2b0SMark Adams sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3253d876e2b0SMark Adams sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3254d876e2b0SMark Adams } 32555850ef23SBarry Smith if (n == sz-1) { 32565850ef23SBarry Smith tmp0 = x[*idx]; 32575850ef23SBarry Smith sum1 -= v1[0] * tmp0; 32585850ef23SBarry Smith sum2 -= v2[0] * tmp0; 32595850ef23SBarry Smith } 3260d876e2b0SMark Adams x[row] = sum1*ibdiag[0] + sum2*ibdiag[2]; 3261d876e2b0SMark Adams x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3]; 32625850ef23SBarry Smith break; 32635850ef23SBarry Smith case 3: 32645850ef23SBarry Smith v2 = a->a + ii[row+1]; 32655850ef23SBarry Smith v3 = a->a + ii[row+2]; 32665850ef23SBarry Smith sum1 = b[row]; 32675850ef23SBarry Smith sum2 = b[row+1]; 32685850ef23SBarry Smith sum3 = b[row+2]; 32695850ef23SBarry Smith for (n = 0; n<sz-1; n+=2) { 32705850ef23SBarry Smith i1 = idx[0]; 32715850ef23SBarry Smith i2 = idx[1]; 32725850ef23SBarry Smith idx += 2; 32735850ef23SBarry Smith tmp0 = x[i1]; 32745850ef23SBarry Smith tmp1 = x[i2]; 32755850ef23SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 32765850ef23SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 32775850ef23SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 32785850ef23SBarry Smith } 3279d876e2b0SMark Adams if (n == sz-1) { 3280d876e2b0SMark Adams tmp0 = x[*idx++]; 3281d876e2b0SMark Adams sum1 -= v1[0] * tmp0; 3282d876e2b0SMark Adams sum2 -= v2[0] * tmp0; 3283d876e2b0SMark Adams sum3 -= v3[0] * tmp0; 3284d876e2b0SMark Adams v1++; v2++; v3++; 3285d876e2b0SMark Adams } 3286d876e2b0SMark Adams t[row] = sum1; 3287d876e2b0SMark Adams t[row+1] = sum2; 3288d876e2b0SMark Adams t[row+2] = sum3; 3289d876e2b0SMark Adams sz = ii[row+1] - diag[row] - 3; 3290d876e2b0SMark Adams idx = a->j + diag[row] + 3; 3291d876e2b0SMark Adams v1 += 3; 3292d876e2b0SMark Adams v2 += 3; 3293d876e2b0SMark Adams v3 += 3; 3294d876e2b0SMark Adams for (n = 0; n<sz-1; n+=2) { 3295d876e2b0SMark Adams i1 = idx[0]; 3296d876e2b0SMark Adams i2 = idx[1]; 3297d876e2b0SMark Adams idx += 2; 3298d876e2b0SMark Adams tmp0 = x[i1]; 3299d876e2b0SMark Adams tmp1 = x[i2]; 3300d876e2b0SMark Adams sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3301d876e2b0SMark Adams sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3302d876e2b0SMark Adams sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3303d876e2b0SMark Adams } 33045850ef23SBarry Smith if (n == sz-1) { 33055850ef23SBarry Smith tmp0 = x[*idx]; 33065850ef23SBarry Smith sum1 -= v1[0] * tmp0; 33075850ef23SBarry Smith sum2 -= v2[0] * tmp0; 33085850ef23SBarry Smith sum3 -= v3[0] * tmp0; 33095850ef23SBarry Smith } 3310d876e2b0SMark Adams x[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6]; 3311d876e2b0SMark Adams x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7]; 3312d876e2b0SMark Adams x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8]; 33135850ef23SBarry Smith break; 33145850ef23SBarry Smith case 4: 33155850ef23SBarry Smith v2 = a->a + ii[row+1]; 33165850ef23SBarry Smith v3 = a->a + ii[row+2]; 33175850ef23SBarry Smith v4 = a->a + ii[row+3]; 33185850ef23SBarry Smith sum1 = b[row]; 33195850ef23SBarry Smith sum2 = b[row+1]; 33205850ef23SBarry Smith sum3 = b[row+2]; 33215850ef23SBarry Smith sum4 = b[row+3]; 33225850ef23SBarry Smith for (n = 0; n<sz-1; n+=2) { 33235850ef23SBarry Smith i1 = idx[0]; 33245850ef23SBarry Smith i2 = idx[1]; 33255850ef23SBarry Smith idx += 2; 33265850ef23SBarry Smith tmp0 = x[i1]; 33275850ef23SBarry Smith tmp1 = x[i2]; 33285850ef23SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 33295850ef23SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 33305850ef23SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 33315850ef23SBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 33325850ef23SBarry Smith } 3333d876e2b0SMark Adams if (n == sz-1) { 3334d876e2b0SMark Adams tmp0 = x[*idx++]; 3335d876e2b0SMark Adams sum1 -= v1[0] * tmp0; 3336d876e2b0SMark Adams sum2 -= v2[0] * tmp0; 3337d876e2b0SMark Adams sum3 -= v3[0] * tmp0; 3338d876e2b0SMark Adams sum4 -= v4[0] * tmp0; 3339d876e2b0SMark Adams v1++; v2++; v3++; v4++; 3340d876e2b0SMark Adams } 3341d876e2b0SMark Adams t[row] = sum1; 3342d876e2b0SMark Adams t[row+1] = sum2; 3343d876e2b0SMark Adams t[row+2] = sum3; 3344d876e2b0SMark Adams t[row+3] = sum4; 3345d876e2b0SMark Adams sz = ii[row+1] - diag[row] - 4; 3346d876e2b0SMark Adams idx = a->j + diag[row] + 4; 3347d876e2b0SMark Adams v1 += 4; 3348d876e2b0SMark Adams v2 += 4; 3349d876e2b0SMark Adams v3 += 4; 3350d876e2b0SMark Adams v4 += 4; 3351d876e2b0SMark Adams for (n = 0; n<sz-1; n+=2) { 3352d876e2b0SMark Adams i1 = idx[0]; 3353d876e2b0SMark Adams i2 = idx[1]; 3354d876e2b0SMark Adams idx += 2; 3355d876e2b0SMark Adams tmp0 = x[i1]; 3356d876e2b0SMark Adams tmp1 = x[i2]; 3357d876e2b0SMark Adams sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3358d876e2b0SMark Adams sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3359d876e2b0SMark Adams sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3360d876e2b0SMark Adams sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 3361d876e2b0SMark Adams } 33625850ef23SBarry Smith if (n == sz-1) { 33635850ef23SBarry Smith tmp0 = x[*idx]; 33645850ef23SBarry Smith sum1 -= v1[0] * tmp0; 33655850ef23SBarry Smith sum2 -= v2[0] * tmp0; 33665850ef23SBarry Smith sum3 -= v3[0] * tmp0; 33675850ef23SBarry Smith sum4 -= v4[0] * tmp0; 33685850ef23SBarry Smith } 3369d876e2b0SMark Adams x[row] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12]; 3370d876e2b0SMark Adams x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13]; 3371d876e2b0SMark Adams x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14]; 3372d876e2b0SMark Adams x[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15]; 33735850ef23SBarry Smith break; 33745850ef23SBarry Smith case 5: 33755850ef23SBarry Smith v2 = a->a + ii[row+1]; 33765850ef23SBarry Smith v3 = a->a + ii[row+2]; 33775850ef23SBarry Smith v4 = a->a + ii[row+3]; 33785850ef23SBarry Smith v5 = a->a + ii[row+4]; 33795850ef23SBarry Smith sum1 = b[row]; 33805850ef23SBarry Smith sum2 = b[row+1]; 33815850ef23SBarry Smith sum3 = b[row+2]; 33825850ef23SBarry Smith sum4 = b[row+3]; 33835850ef23SBarry Smith sum5 = b[row+4]; 33845850ef23SBarry Smith for (n = 0; n<sz-1; n+=2) { 33855850ef23SBarry Smith i1 = idx[0]; 33865850ef23SBarry Smith i2 = idx[1]; 33875850ef23SBarry Smith idx += 2; 33885850ef23SBarry Smith tmp0 = x[i1]; 33895850ef23SBarry Smith tmp1 = x[i2]; 33905850ef23SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 33915850ef23SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 33925850ef23SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 33935850ef23SBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 33945850ef23SBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 33955850ef23SBarry Smith } 33965850ef23SBarry Smith if (n == sz-1) { 3397d876e2b0SMark Adams tmp0 = x[*idx++]; 33985850ef23SBarry Smith sum1 -= v1[0] * tmp0; 33995850ef23SBarry Smith sum2 -= v2[0] * tmp0; 34005850ef23SBarry Smith sum3 -= v3[0] * tmp0; 34015850ef23SBarry Smith sum4 -= v4[0] * tmp0; 34025850ef23SBarry Smith sum5 -= v5[0] * tmp0; 3403d876e2b0SMark Adams v1++; v2++; v3++; v4++; v5++; 34045850ef23SBarry Smith } 3405d876e2b0SMark Adams t[row] = sum1; 3406d876e2b0SMark Adams t[row+1] = sum2; 3407d876e2b0SMark Adams t[row+2] = sum3; 3408d876e2b0SMark Adams t[row+3] = sum4; 3409d876e2b0SMark Adams t[row+4] = sum5; 3410d876e2b0SMark Adams sz = ii[row+1] - diag[row] - 5; 3411d876e2b0SMark Adams idx = a->j + diag[row] + 5; 3412d876e2b0SMark Adams v1 += 5; 3413d876e2b0SMark Adams v2 += 5; 3414d876e2b0SMark Adams v3 += 5; 3415d876e2b0SMark Adams v4 += 5; 3416d876e2b0SMark Adams v5 += 5; 34175850ef23SBarry Smith for (n = 0; n<sz-1; n+=2) { 34185850ef23SBarry Smith i1 = idx[0]; 34195850ef23SBarry Smith i2 = idx[1]; 34205850ef23SBarry Smith idx += 2; 34215850ef23SBarry Smith tmp0 = x[i1]; 34225850ef23SBarry Smith tmp1 = x[i2]; 34235850ef23SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 34245850ef23SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 34255850ef23SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 34265850ef23SBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 34275850ef23SBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 34285850ef23SBarry Smith } 34295850ef23SBarry Smith if (n == sz-1) { 34305850ef23SBarry Smith tmp0 = x[*idx]; 3431d876e2b0SMark Adams sum1 -= v1[0] * tmp0; 3432d876e2b0SMark Adams sum2 -= v2[0] * tmp0; 3433d876e2b0SMark Adams sum3 -= v3[0] * tmp0; 3434d876e2b0SMark Adams sum4 -= v4[0] * tmp0; 3435d876e2b0SMark Adams sum5 -= v5[0] * tmp0; 34365850ef23SBarry Smith } 3437d876e2b0SMark Adams x[row] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20]; 3438d876e2b0SMark Adams x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21]; 3439d876e2b0SMark Adams x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22]; 3440d876e2b0SMark Adams x[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23]; 3441d876e2b0SMark Adams x[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24]; 3442d876e2b0SMark Adams break; 3443d876e2b0SMark Adams default: 3444d876e2b0SMark Adams SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 3445d876e2b0SMark Adams } 3446d876e2b0SMark Adams } 3447d876e2b0SMark Adams xb = t; 3448d876e2b0SMark Adams ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); /* undercounts diag inverse */ 3449d876e2b0SMark Adams } else xb = b; 3450d876e2b0SMark Adams 3451d876e2b0SMark Adams if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) { 3452d876e2b0SMark Adams 3453d876e2b0SMark Adams ibdiag = a->inode.ibdiag+a->inode.bdiagsize; 3454d876e2b0SMark Adams for (i=m-1, row=A->rmap->n-1; i>=0; i--) { 3455d876e2b0SMark Adams ibdiag -= sizes[i]*sizes[i]; 3456d876e2b0SMark Adams 3457d876e2b0SMark Adams /* set RHS */ 3458d876e2b0SMark Adams if (xb == b) { 3459d876e2b0SMark Adams /* whole (old way) */ 3460d876e2b0SMark Adams sz = ii[row+1] - ii[row]; 3461d876e2b0SMark Adams idx = a->j + ii[row]; 3462d876e2b0SMark Adams switch (sizes[i]) { 3463d876e2b0SMark Adams case 5: 3464d876e2b0SMark Adams v5 = a->a + ii[row-4]; 3465d876e2b0SMark Adams case 4: /* fall through */ 3466d876e2b0SMark Adams v4 = a->a + ii[row-3]; 3467d876e2b0SMark Adams case 3: 3468d876e2b0SMark Adams v3 = a->a + ii[row-2]; 3469d876e2b0SMark Adams case 2: 3470d876e2b0SMark Adams v2 = a->a + ii[row-1]; 3471d876e2b0SMark Adams case 1: 3472d876e2b0SMark Adams v1 = a->a + ii[row]; 3473d876e2b0SMark Adams break; 3474d876e2b0SMark Adams default: 3475d876e2b0SMark Adams SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 3476d876e2b0SMark Adams } 3477d876e2b0SMark Adams } else { 3478d876e2b0SMark Adams /* upper, no diag */ 3479d876e2b0SMark Adams sz = ii[row+1] - diag[row] - 1; 3480d876e2b0SMark Adams idx = a->j + diag[row] + 1; 3481d876e2b0SMark Adams switch (sizes[i]) { 3482d876e2b0SMark Adams case 5: 3483d876e2b0SMark Adams v5 = a->a + diag[row-4] + 5; 3484d876e2b0SMark Adams case 4: /* fall through */ 3485d876e2b0SMark Adams v4 = a->a + diag[row-3] + 4; 3486d876e2b0SMark Adams case 3: 3487d876e2b0SMark Adams v3 = a->a + diag[row-2] + 3; 3488d876e2b0SMark Adams case 2: 3489d876e2b0SMark Adams v2 = a->a + diag[row-1] + 2; 3490d876e2b0SMark Adams case 1: 3491d876e2b0SMark Adams v1 = a->a + diag[row] + 1; 3492d876e2b0SMark Adams } 3493d876e2b0SMark Adams } 3494d876e2b0SMark Adams /* set sum */ 3495d876e2b0SMark Adams switch (sizes[i]) { 3496d876e2b0SMark Adams case 5: 3497d876e2b0SMark Adams sum5 = xb[row-4]; 3498d876e2b0SMark Adams case 4: /* fall through */ 3499d876e2b0SMark Adams sum4 = xb[row-3]; 3500d876e2b0SMark Adams case 3: 3501d876e2b0SMark Adams sum3 = xb[row-2]; 3502d876e2b0SMark Adams case 2: 3503d876e2b0SMark Adams sum2 = xb[row-1]; 3504d876e2b0SMark Adams case 1: 3505d876e2b0SMark Adams /* note that sum1 is associated with the last row */ 3506d876e2b0SMark Adams sum1 = xb[row]; 3507d876e2b0SMark Adams } 3508d876e2b0SMark Adams /* do sums */ 3509d876e2b0SMark Adams for (n = 0; n<sz-1; n+=2) { 3510d876e2b0SMark Adams i1 = idx[0]; 3511d876e2b0SMark Adams i2 = idx[1]; 3512d876e2b0SMark Adams idx += 2; 3513d876e2b0SMark Adams tmp0 = x[i1]; 3514d876e2b0SMark Adams tmp1 = x[i2]; 3515d876e2b0SMark Adams switch (sizes[i]) { 3516d876e2b0SMark Adams case 5: 3517d876e2b0SMark Adams sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 3518d876e2b0SMark Adams case 4: /* fall through */ 3519d876e2b0SMark Adams sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 3520d876e2b0SMark Adams case 3: 3521d876e2b0SMark Adams sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3522d876e2b0SMark Adams case 2: 3523d876e2b0SMark Adams sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3524d876e2b0SMark Adams case 1: 3525d876e2b0SMark Adams sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3526d876e2b0SMark Adams } 3527d876e2b0SMark Adams } 3528d876e2b0SMark Adams /* ragged edge */ 3529d876e2b0SMark Adams if (n == sz-1) { 3530d876e2b0SMark Adams tmp0 = x[*idx]; 3531d876e2b0SMark Adams switch (sizes[i]) { 3532d876e2b0SMark Adams case 5: 3533d876e2b0SMark Adams sum5 -= *v5*tmp0; 3534d876e2b0SMark Adams case 4: /* fall through */ 3535d876e2b0SMark Adams sum4 -= *v4*tmp0; 3536d876e2b0SMark Adams case 3: 3537d876e2b0SMark Adams sum3 -= *v3*tmp0; 3538d876e2b0SMark Adams case 2: 3539d876e2b0SMark Adams sum2 -= *v2*tmp0; 3540d876e2b0SMark Adams case 1: 3541d876e2b0SMark Adams sum1 -= *v1*tmp0; 3542d876e2b0SMark Adams } 3543d876e2b0SMark Adams } 3544d876e2b0SMark Adams /* update */ 3545d876e2b0SMark Adams if (xb == b) { 3546d876e2b0SMark Adams /* whole (old way) w/ diag */ 3547d876e2b0SMark Adams switch (sizes[i]) { 3548d876e2b0SMark Adams case 5: 35495850ef23SBarry Smith x[row--] += sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24]; 35505850ef23SBarry Smith x[row--] += sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23]; 35515850ef23SBarry Smith x[row--] += sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22]; 35525850ef23SBarry Smith x[row--] += sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21]; 35535850ef23SBarry Smith x[row--] += sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20]; 35545850ef23SBarry Smith break; 3555d876e2b0SMark Adams case 4: 3556d876e2b0SMark Adams x[row--] += sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15]; 3557d876e2b0SMark Adams x[row--] += sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14]; 3558d876e2b0SMark Adams x[row--] += sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13]; 3559d876e2b0SMark Adams x[row--] += sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12]; 3560d876e2b0SMark Adams break; 3561d876e2b0SMark Adams case 3: 3562d876e2b0SMark Adams x[row--] += sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8]; 3563d876e2b0SMark Adams x[row--] += sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7]; 3564d876e2b0SMark Adams x[row--] += sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6]; 3565d876e2b0SMark Adams break; 3566d876e2b0SMark Adams case 2: 3567d876e2b0SMark Adams x[row--] += sum2*ibdiag[1] + sum1*ibdiag[3]; 3568d876e2b0SMark Adams x[row--] += sum2*ibdiag[0] + sum1*ibdiag[2]; 3569d876e2b0SMark Adams break; 3570d876e2b0SMark Adams case 1: 3571d876e2b0SMark Adams x[row--] += sum1*(*ibdiag); 3572d876e2b0SMark Adams break; 3573d876e2b0SMark Adams } 3574d876e2b0SMark Adams } else { 3575d876e2b0SMark Adams /* no diag so set = */ 3576d876e2b0SMark Adams switch (sizes[i]) { 3577d876e2b0SMark Adams case 5: 3578d876e2b0SMark Adams x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24]; 3579d876e2b0SMark Adams x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23]; 3580d876e2b0SMark Adams x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22]; 3581d876e2b0SMark Adams x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21]; 3582d876e2b0SMark Adams x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20]; 3583d876e2b0SMark Adams break; 3584d876e2b0SMark Adams case 4: 3585d876e2b0SMark Adams x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15]; 3586d876e2b0SMark Adams x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14]; 3587d876e2b0SMark Adams x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13]; 3588d876e2b0SMark Adams x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12]; 3589d876e2b0SMark Adams break; 3590d876e2b0SMark Adams case 3: 3591d876e2b0SMark Adams x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8]; 3592d876e2b0SMark Adams x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7]; 3593d876e2b0SMark Adams x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6]; 3594d876e2b0SMark Adams break; 3595d876e2b0SMark Adams case 2: 3596d876e2b0SMark Adams x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3]; 3597d876e2b0SMark Adams x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2]; 3598d876e2b0SMark Adams break; 3599d876e2b0SMark Adams case 1: 3600d876e2b0SMark Adams x[row--] = sum1*(*ibdiag); 3601d876e2b0SMark Adams break; 36025850ef23SBarry Smith } 36035850ef23SBarry Smith } 3604d876e2b0SMark Adams } 3605d876e2b0SMark Adams if (xb == b) { 36065850ef23SBarry Smith ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 3607d876e2b0SMark Adams } else { 3608d876e2b0SMark Adams ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper, undercounts diag inverse */ 3609d876e2b0SMark Adams } 36105850ef23SBarry Smith } 36112af78befSBarry Smith } 361289c6957cSBarry Smith if (flag & SOR_EISENSTAT) { 361389c6957cSBarry Smith /* 361489c6957cSBarry Smith Apply (U + D)^-1 where D is now the block diagonal 361589c6957cSBarry Smith */ 361689c6957cSBarry Smith ibdiag = a->inode.ibdiag+a->inode.bdiagsize; 361789c6957cSBarry Smith for (i=m-1, row=A->rmap->n-1; i>=0; i--) { 361889c6957cSBarry Smith ibdiag -= sizes[i]*sizes[i]; 361989c6957cSBarry Smith sz = ii[row+1] - diag[row] - 1; 362089c6957cSBarry Smith v1 = a->a + diag[row] + 1; 362189c6957cSBarry Smith idx = a->j + diag[row] + 1; 36224108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 362389c6957cSBarry Smith switch (sizes[i]) { 362489c6957cSBarry Smith case 1: 362589c6957cSBarry Smith 362689c6957cSBarry Smith sum1 = b[row]; 362789c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 362889c6957cSBarry Smith i1 = idx[0]; 362989c6957cSBarry Smith i2 = idx[1]; 363089c6957cSBarry Smith idx += 2; 363189c6957cSBarry Smith tmp0 = x[i1]; 363289c6957cSBarry Smith tmp1 = x[i2]; 363389c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 363489c6957cSBarry Smith } 363589c6957cSBarry Smith 363689c6957cSBarry Smith if (n == sz-1) { 363789c6957cSBarry Smith tmp0 = x[*idx]; 363889c6957cSBarry Smith sum1 -= *v1*tmp0; 363989c6957cSBarry Smith } 3640938d4eb3SBarry Smith x[row] = sum1*(*ibdiag);row--; 364189c6957cSBarry Smith break; 364289c6957cSBarry Smith 364389c6957cSBarry Smith case 2: 364489c6957cSBarry Smith 364589c6957cSBarry Smith sum1 = b[row]; 364689c6957cSBarry Smith sum2 = b[row-1]; 364789c6957cSBarry Smith /* note that sum1 is associated with the second of the two rows */ 364889c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 364989c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 365089c6957cSBarry Smith i1 = idx[0]; 365189c6957cSBarry Smith i2 = idx[1]; 365289c6957cSBarry Smith idx += 2; 365389c6957cSBarry Smith tmp0 = x[i1]; 365489c6957cSBarry Smith tmp1 = x[i2]; 365589c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 365689c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 365789c6957cSBarry Smith } 365889c6957cSBarry Smith 365989c6957cSBarry Smith if (n == sz-1) { 366089c6957cSBarry Smith tmp0 = x[*idx]; 366189c6957cSBarry Smith sum1 -= *v1*tmp0; 366289c6957cSBarry Smith sum2 -= *v2*tmp0; 366389c6957cSBarry Smith } 3664938d4eb3SBarry Smith x[row] = sum2*ibdiag[1] + sum1*ibdiag[3]; 3665938d4eb3SBarry Smith x[row-1] = sum2*ibdiag[0] + sum1*ibdiag[2]; 3666938d4eb3SBarry Smith row -= 2; 366789c6957cSBarry Smith break; 366889c6957cSBarry Smith case 3: 366989c6957cSBarry Smith 367089c6957cSBarry Smith sum1 = b[row]; 367189c6957cSBarry Smith sum2 = b[row-1]; 367289c6957cSBarry Smith sum3 = b[row-2]; 367389c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 367489c6957cSBarry Smith v3 = a->a + diag[row-2] + 3; 367589c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 367689c6957cSBarry Smith i1 = idx[0]; 367789c6957cSBarry Smith i2 = idx[1]; 367889c6957cSBarry Smith idx += 2; 367989c6957cSBarry Smith tmp0 = x[i1]; 368089c6957cSBarry Smith tmp1 = x[i2]; 368189c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 368289c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 368389c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 368489c6957cSBarry Smith } 368589c6957cSBarry Smith 368689c6957cSBarry Smith if (n == sz-1) { 368789c6957cSBarry Smith tmp0 = x[*idx]; 368889c6957cSBarry Smith sum1 -= *v1*tmp0; 368989c6957cSBarry Smith sum2 -= *v2*tmp0; 369089c6957cSBarry Smith sum3 -= *v3*tmp0; 369189c6957cSBarry Smith } 3692938d4eb3SBarry Smith x[row] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8]; 3693938d4eb3SBarry Smith x[row-1] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7]; 3694938d4eb3SBarry Smith x[row-2] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6]; 3695938d4eb3SBarry Smith row -= 3; 369689c6957cSBarry Smith break; 369789c6957cSBarry Smith case 4: 369889c6957cSBarry Smith 369989c6957cSBarry Smith sum1 = b[row]; 370089c6957cSBarry Smith sum2 = b[row-1]; 370189c6957cSBarry Smith sum3 = b[row-2]; 370289c6957cSBarry Smith sum4 = b[row-3]; 370389c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 370489c6957cSBarry Smith v3 = a->a + diag[row-2] + 3; 370589c6957cSBarry Smith v4 = a->a + diag[row-3] + 4; 370689c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 370789c6957cSBarry Smith i1 = idx[0]; 370889c6957cSBarry Smith i2 = idx[1]; 370989c6957cSBarry Smith idx += 2; 371089c6957cSBarry Smith tmp0 = x[i1]; 371189c6957cSBarry Smith tmp1 = x[i2]; 371289c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 371389c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 371489c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 371589c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 371689c6957cSBarry Smith } 371789c6957cSBarry Smith 371889c6957cSBarry Smith if (n == sz-1) { 371989c6957cSBarry Smith tmp0 = x[*idx]; 372089c6957cSBarry Smith sum1 -= *v1*tmp0; 372189c6957cSBarry Smith sum2 -= *v2*tmp0; 372289c6957cSBarry Smith sum3 -= *v3*tmp0; 372389c6957cSBarry Smith sum4 -= *v4*tmp0; 372489c6957cSBarry Smith } 3725938d4eb3SBarry Smith x[row] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15]; 3726938d4eb3SBarry Smith x[row-1] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14]; 3727938d4eb3SBarry Smith x[row-2] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13]; 3728938d4eb3SBarry Smith x[row-3] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12]; 3729938d4eb3SBarry Smith row -= 4; 373089c6957cSBarry Smith break; 373189c6957cSBarry Smith case 5: 373289c6957cSBarry Smith 373389c6957cSBarry Smith sum1 = b[row]; 373489c6957cSBarry Smith sum2 = b[row-1]; 373589c6957cSBarry Smith sum3 = b[row-2]; 373689c6957cSBarry Smith sum4 = b[row-3]; 373789c6957cSBarry Smith sum5 = b[row-4]; 373889c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 373989c6957cSBarry Smith v3 = a->a + diag[row-2] + 3; 374089c6957cSBarry Smith v4 = a->a + diag[row-3] + 4; 374189c6957cSBarry Smith v5 = a->a + diag[row-4] + 5; 374289c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 374389c6957cSBarry Smith i1 = idx[0]; 374489c6957cSBarry Smith i2 = idx[1]; 374589c6957cSBarry Smith idx += 2; 374689c6957cSBarry Smith tmp0 = x[i1]; 374789c6957cSBarry Smith tmp1 = x[i2]; 374889c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 374989c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 375089c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 375189c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 375289c6957cSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 375389c6957cSBarry Smith } 375489c6957cSBarry Smith 375589c6957cSBarry Smith if (n == sz-1) { 375689c6957cSBarry Smith tmp0 = x[*idx]; 375789c6957cSBarry Smith sum1 -= *v1*tmp0; 375889c6957cSBarry Smith sum2 -= *v2*tmp0; 375989c6957cSBarry Smith sum3 -= *v3*tmp0; 376089c6957cSBarry Smith sum4 -= *v4*tmp0; 376189c6957cSBarry Smith sum5 -= *v5*tmp0; 376289c6957cSBarry Smith } 3763938d4eb3SBarry Smith x[row] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24]; 3764938d4eb3SBarry Smith x[row-1] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23]; 3765938d4eb3SBarry Smith x[row-2] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22]; 3766938d4eb3SBarry Smith x[row-3] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21]; 3767938d4eb3SBarry Smith x[row-4] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20]; 3768938d4eb3SBarry Smith row -= 5; 376989c6957cSBarry Smith break; 377089c6957cSBarry Smith default: 3771e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 377289c6957cSBarry Smith } 377389c6957cSBarry Smith } 377489c6957cSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 377589c6957cSBarry Smith 377689c6957cSBarry Smith /* 377789c6957cSBarry Smith t = b - D x where D is the block diagonal 377889c6957cSBarry Smith */ 377989c6957cSBarry Smith cnt = 0; 378089c6957cSBarry Smith for (i=0, row=0; i<m; i++) { 378189c6957cSBarry Smith switch (sizes[i]) { 378289c6957cSBarry Smith case 1: 378389c6957cSBarry Smith t[row] = b[row] - bdiag[cnt++]*x[row]; row++; 378489c6957cSBarry Smith break; 378589c6957cSBarry Smith case 2: 378689c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; 378789c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2]; 378889c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3]; 378989c6957cSBarry Smith t[row] = b[row] - tmp1; 379089c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; row += 2; 379189c6957cSBarry Smith cnt += 4; 379289c6957cSBarry Smith break; 379389c6957cSBarry Smith case 3: 379489c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; 379589c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6]; 379689c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7]; 379789c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8]; 379889c6957cSBarry Smith t[row] = b[row] - tmp1; 379989c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; 380089c6957cSBarry Smith t[row+2] = b[row+2] - tmp3; row += 3; 380189c6957cSBarry Smith cnt += 9; 380289c6957cSBarry Smith break; 380389c6957cSBarry Smith case 4: 380489c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; 380589c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12]; 380689c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13]; 380789c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14]; 380889c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15]; 380989c6957cSBarry Smith t[row] = b[row] - tmp1; 381089c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; 381189c6957cSBarry Smith t[row+2] = b[row+2] - tmp3; 381289c6957cSBarry Smith t[row+3] = b[row+3] - tmp4; row += 4; 381389c6957cSBarry Smith cnt += 16; 381489c6957cSBarry Smith break; 381589c6957cSBarry Smith case 5: 381689c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; x5 = x[row+4]; 381789c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20]; 381889c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21]; 381989c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22]; 382089c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23]; 382189c6957cSBarry Smith tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24]; 382289c6957cSBarry Smith t[row] = b[row] - tmp1; 382389c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; 382489c6957cSBarry Smith t[row+2] = b[row+2] - tmp3; 382589c6957cSBarry Smith t[row+3] = b[row+3] - tmp4; 382689c6957cSBarry Smith t[row+4] = b[row+4] - tmp5;row += 5; 382789c6957cSBarry Smith cnt += 25; 382889c6957cSBarry Smith break; 382989c6957cSBarry Smith default: 3830e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 383189c6957cSBarry Smith } 383289c6957cSBarry Smith } 383389c6957cSBarry Smith ierr = PetscLogFlops(m);CHKERRQ(ierr); 383489c6957cSBarry Smith 383589c6957cSBarry Smith 383689c6957cSBarry Smith 383789c6957cSBarry Smith /* 383889c6957cSBarry Smith Apply (L + D)^-1 where D is the block diagonal 383989c6957cSBarry Smith */ 384089c6957cSBarry Smith for (i=0, row=0; i<m; i++) { 384189c6957cSBarry Smith sz = diag[row] - ii[row]; 384289c6957cSBarry Smith v1 = a->a + ii[row]; 384389c6957cSBarry Smith idx = a->j + ii[row]; 38444108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 384589c6957cSBarry Smith switch (sizes[i]) { 384689c6957cSBarry Smith case 1: 384789c6957cSBarry Smith 384889c6957cSBarry Smith sum1 = t[row]; 384989c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 385089c6957cSBarry Smith i1 = idx[0]; 385189c6957cSBarry Smith i2 = idx[1]; 385289c6957cSBarry Smith idx += 2; 385389c6957cSBarry Smith tmp0 = t[i1]; 385489c6957cSBarry Smith tmp1 = t[i2]; 385589c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 385689c6957cSBarry Smith } 385789c6957cSBarry Smith 385889c6957cSBarry Smith if (n == sz-1) { 385989c6957cSBarry Smith tmp0 = t[*idx]; 386089c6957cSBarry Smith sum1 -= *v1 * tmp0; 386189c6957cSBarry Smith } 386289c6957cSBarry Smith x[row] += t[row] = sum1*(*ibdiag++); row++; 386389c6957cSBarry Smith break; 386489c6957cSBarry Smith case 2: 386589c6957cSBarry Smith v2 = a->a + ii[row+1]; 386689c6957cSBarry Smith sum1 = t[row]; 386789c6957cSBarry Smith sum2 = t[row+1]; 386889c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 386989c6957cSBarry Smith i1 = idx[0]; 387089c6957cSBarry Smith i2 = idx[1]; 387189c6957cSBarry Smith idx += 2; 387289c6957cSBarry Smith tmp0 = t[i1]; 387389c6957cSBarry Smith tmp1 = t[i2]; 387489c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 387589c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 387689c6957cSBarry Smith } 387789c6957cSBarry Smith 387889c6957cSBarry Smith if (n == sz-1) { 387989c6957cSBarry Smith tmp0 = t[*idx]; 388089c6957cSBarry Smith sum1 -= v1[0] * tmp0; 388189c6957cSBarry Smith sum2 -= v2[0] * tmp0; 388289c6957cSBarry Smith } 388389c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[2]; 388489c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3]; 388589c6957cSBarry Smith ibdiag += 4; row += 2; 388689c6957cSBarry Smith break; 388789c6957cSBarry Smith case 3: 388889c6957cSBarry Smith v2 = a->a + ii[row+1]; 388989c6957cSBarry Smith v3 = a->a + ii[row+2]; 389089c6957cSBarry Smith sum1 = t[row]; 389189c6957cSBarry Smith sum2 = t[row+1]; 389289c6957cSBarry Smith sum3 = t[row+2]; 389389c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 389489c6957cSBarry Smith i1 = idx[0]; 389589c6957cSBarry Smith i2 = idx[1]; 389689c6957cSBarry Smith idx += 2; 389789c6957cSBarry Smith tmp0 = t[i1]; 389889c6957cSBarry Smith tmp1 = t[i2]; 389989c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 390089c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 390189c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 390289c6957cSBarry Smith } 390389c6957cSBarry Smith 390489c6957cSBarry Smith if (n == sz-1) { 390589c6957cSBarry Smith tmp0 = t[*idx]; 390689c6957cSBarry Smith sum1 -= v1[0] * tmp0; 390789c6957cSBarry Smith sum2 -= v2[0] * tmp0; 390889c6957cSBarry Smith sum3 -= v3[0] * tmp0; 390989c6957cSBarry Smith } 391089c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6]; 391189c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7]; 391289c6957cSBarry Smith x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8]; 391389c6957cSBarry Smith ibdiag += 9; row += 3; 391489c6957cSBarry Smith break; 391589c6957cSBarry Smith case 4: 391689c6957cSBarry Smith v2 = a->a + ii[row+1]; 391789c6957cSBarry Smith v3 = a->a + ii[row+2]; 391889c6957cSBarry Smith v4 = a->a + ii[row+3]; 391989c6957cSBarry Smith sum1 = t[row]; 392089c6957cSBarry Smith sum2 = t[row+1]; 392189c6957cSBarry Smith sum3 = t[row+2]; 392289c6957cSBarry Smith sum4 = t[row+3]; 392389c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 392489c6957cSBarry Smith i1 = idx[0]; 392589c6957cSBarry Smith i2 = idx[1]; 392689c6957cSBarry Smith idx += 2; 392789c6957cSBarry Smith tmp0 = t[i1]; 392889c6957cSBarry Smith tmp1 = t[i2]; 392989c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 393089c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 393189c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 393289c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 393389c6957cSBarry Smith } 393489c6957cSBarry Smith 393589c6957cSBarry Smith if (n == sz-1) { 393689c6957cSBarry Smith tmp0 = t[*idx]; 393789c6957cSBarry Smith sum1 -= v1[0] * tmp0; 393889c6957cSBarry Smith sum2 -= v2[0] * tmp0; 393989c6957cSBarry Smith sum3 -= v3[0] * tmp0; 394089c6957cSBarry Smith sum4 -= v4[0] * tmp0; 394189c6957cSBarry Smith } 394289c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12]; 394389c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13]; 394489c6957cSBarry Smith x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14]; 394589c6957cSBarry Smith x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15]; 394689c6957cSBarry Smith ibdiag += 16; row += 4; 394789c6957cSBarry Smith break; 394889c6957cSBarry Smith case 5: 394989c6957cSBarry Smith v2 = a->a + ii[row+1]; 395089c6957cSBarry Smith v3 = a->a + ii[row+2]; 395189c6957cSBarry Smith v4 = a->a + ii[row+3]; 395289c6957cSBarry Smith v5 = a->a + ii[row+4]; 395389c6957cSBarry Smith sum1 = t[row]; 395489c6957cSBarry Smith sum2 = t[row+1]; 395589c6957cSBarry Smith sum3 = t[row+2]; 395689c6957cSBarry Smith sum4 = t[row+3]; 395789c6957cSBarry Smith sum5 = t[row+4]; 395889c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 395989c6957cSBarry Smith i1 = idx[0]; 396089c6957cSBarry Smith i2 = idx[1]; 396189c6957cSBarry Smith idx += 2; 396289c6957cSBarry Smith tmp0 = t[i1]; 396389c6957cSBarry Smith tmp1 = t[i2]; 396489c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 396589c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 396689c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 396789c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 396889c6957cSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 396989c6957cSBarry Smith } 397089c6957cSBarry Smith 397189c6957cSBarry Smith if (n == sz-1) { 397289c6957cSBarry Smith tmp0 = t[*idx]; 397389c6957cSBarry Smith sum1 -= v1[0] * tmp0; 397489c6957cSBarry Smith sum2 -= v2[0] * tmp0; 397589c6957cSBarry Smith sum3 -= v3[0] * tmp0; 397689c6957cSBarry Smith sum4 -= v4[0] * tmp0; 397789c6957cSBarry Smith sum5 -= v5[0] * tmp0; 397889c6957cSBarry Smith } 397989c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20]; 398089c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21]; 398189c6957cSBarry Smith x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22]; 398289c6957cSBarry Smith x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23]; 398389c6957cSBarry Smith x[row+4] += t[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24]; 398489c6957cSBarry Smith ibdiag += 25; row += 5; 398589c6957cSBarry Smith break; 398689c6957cSBarry Smith default: 3987e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 398889c6957cSBarry Smith } 398989c6957cSBarry Smith } 399089c6957cSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 39915850ef23SBarry Smith } 39922af78befSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 39933649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3994365a8a9eSBarry Smith PetscFunctionReturn(0); 39952af78befSBarry Smith } 39962af78befSBarry Smith 399789c6957cSBarry Smith #undef __FUNCT__ 39984108e4d5SBarry Smith #define __FUNCT__ "MatMultDiagonalBlock_SeqAIJ_Inode" 39994108e4d5SBarry Smith PetscErrorCode MatMultDiagonalBlock_SeqAIJ_Inode(Mat A,Vec bb,Vec xx) 400089c6957cSBarry Smith { 400189c6957cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 400289c6957cSBarry Smith PetscScalar *x,tmp1,tmp2,tmp3,tmp4,tmp5,x1,x2,x3,x4,x5; 400389c6957cSBarry Smith const MatScalar *bdiag = a->inode.bdiag; 400489c6957cSBarry Smith const PetscScalar *b; 400589c6957cSBarry Smith PetscErrorCode ierr; 400689c6957cSBarry Smith PetscInt m = a->inode.node_count,cnt = 0,i,row; 400789c6957cSBarry Smith const PetscInt *sizes = a->inode.size; 40082af78befSBarry Smith 400989c6957cSBarry Smith PetscFunctionBegin; 401089c6957cSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 40113649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 401289c6957cSBarry Smith cnt = 0; 401389c6957cSBarry Smith for (i=0, row=0; i<m; i++) { 401489c6957cSBarry Smith switch (sizes[i]) { 401589c6957cSBarry Smith case 1: 401689c6957cSBarry Smith x[row] = b[row]*bdiag[cnt++];row++; 401789c6957cSBarry Smith break; 401889c6957cSBarry Smith case 2: 401989c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; 402089c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2]; 402189c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3]; 402289c6957cSBarry Smith x[row++] = tmp1; 402389c6957cSBarry Smith x[row++] = tmp2; 402489c6957cSBarry Smith cnt += 4; 402589c6957cSBarry Smith break; 402689c6957cSBarry Smith case 3: 402789c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; x3 = b[row+2]; 402889c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6]; 402989c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7]; 403089c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8]; 403189c6957cSBarry Smith x[row++] = tmp1; 403289c6957cSBarry Smith x[row++] = tmp2; 403389c6957cSBarry Smith x[row++] = tmp3; 403489c6957cSBarry Smith cnt += 9; 403589c6957cSBarry Smith break; 403689c6957cSBarry Smith case 4: 403789c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3]; 403889c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12]; 403989c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13]; 404089c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14]; 404189c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15]; 404289c6957cSBarry Smith x[row++] = tmp1; 404389c6957cSBarry Smith x[row++] = tmp2; 404489c6957cSBarry Smith x[row++] = tmp3; 404589c6957cSBarry Smith x[row++] = tmp4; 404689c6957cSBarry Smith cnt += 16; 404789c6957cSBarry Smith break; 404889c6957cSBarry Smith case 5: 404989c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3]; x5 = b[row+4]; 405089c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20]; 405189c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21]; 405289c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22]; 405389c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23]; 405489c6957cSBarry Smith tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24]; 405589c6957cSBarry Smith x[row++] = tmp1; 405689c6957cSBarry Smith x[row++] = tmp2; 405789c6957cSBarry Smith x[row++] = tmp3; 405889c6957cSBarry Smith x[row++] = tmp4; 405989c6957cSBarry Smith x[row++] = tmp5; 406089c6957cSBarry Smith cnt += 25; 406189c6957cSBarry Smith break; 406289c6957cSBarry Smith default: 4063e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 406489c6957cSBarry Smith } 406589c6957cSBarry Smith } 406689c6957cSBarry Smith ierr = PetscLogFlops(2*cnt);CHKERRQ(ierr); 406789c6957cSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 40683649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 406989c6957cSBarry Smith PetscFunctionReturn(0); 407089c6957cSBarry Smith } 407189c6957cSBarry Smith 40724c1414c8SBarry Smith /* 40734c1414c8SBarry Smith samestructure indicates that the matrix has not changed its nonzero structure so we 40744c1414c8SBarry Smith do not need to recompute the inodes 40754c1414c8SBarry Smith */ 40764c1414c8SBarry Smith #undef __FUNCT__ 4077*a02bda8eSBarry Smith #define __FUNCT__ "MatSeqAIJCheckInode" 4078*a02bda8eSBarry Smith PetscErrorCode MatSeqAIJCheckInode(Mat A) 40794c1414c8SBarry Smith { 40804c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 40814c1414c8SBarry Smith PetscErrorCode ierr; 40828758e1faSBarry Smith PetscInt i,j,m,nzx,nzy,*ns,node_count,blk_size; 4083ace3abfcSBarry Smith PetscBool flag; 40848758e1faSBarry Smith const PetscInt *idx,*idy,*ii; 40854c1414c8SBarry Smith 40864c1414c8SBarry Smith PetscFunctionBegin; 4087d74fe821SBarry Smith if (!a->inode.use) PetscFunctionReturn(0); 4088*a02bda8eSBarry Smith if (a->inode.checked && A->nonzerostate == a->inode.mat_nonzerostate) PetscFunctionReturn(0); 40894c1414c8SBarry Smith 4090d0f46423SBarry Smith m = A->rmap->n; 40912205254eSKarl Rupp if (a->inode.size) ns = a->inode.size; 40922205254eSKarl Rupp else { 4093854ce69bSBarry Smith ierr = PetscMalloc1(m+1,&ns);CHKERRQ(ierr); 40942205254eSKarl Rupp } 40954c1414c8SBarry Smith 40964c1414c8SBarry Smith i = 0; 40974c1414c8SBarry Smith node_count = 0; 40984c1414c8SBarry Smith idx = a->j; 40994c1414c8SBarry Smith ii = a->i; 41004c1414c8SBarry Smith while (i < m) { /* For each row */ 41014c1414c8SBarry Smith nzx = ii[i+1] - ii[i]; /* Number of nonzeros */ 41024c1414c8SBarry Smith /* Limits the number of elements in a node to 'a->inode.limit' */ 41034c1414c8SBarry Smith for (j=i+1,idy=idx,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) { 41044c1414c8SBarry Smith nzy = ii[j+1] - ii[j]; /* Same number of nonzeros */ 41054c1414c8SBarry Smith if (nzy != nzx) break; 41064c1414c8SBarry Smith idy += nzx; /* Same nonzero pattern */ 41074c1414c8SBarry Smith ierr = PetscMemcmp(idx,idy,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr); 41084c1414c8SBarry Smith if (!flag) break; 41094c1414c8SBarry Smith } 41104c1414c8SBarry Smith ns[node_count++] = blk_size; 41114c1414c8SBarry Smith idx += blk_size*nzx; 41124c1414c8SBarry Smith i = j; 41134c1414c8SBarry Smith } 41144c1414c8SBarry Smith /* If not enough inodes found,, do not use inode version of the routines */ 4115be6adb11SBarry Smith if (!m || node_count > .8*m) { 41164c1414c8SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 41172205254eSKarl Rupp 41184c1414c8SBarry Smith a->inode.node_count = 0; 41190298fd71SBarry Smith a->inode.size = NULL; 41204c1414c8SBarry Smith a->inode.use = PETSC_FALSE; 4121ded6e084SBarry Smith A->ops->mult = MatMult_SeqAIJ; 4122ded6e084SBarry Smith A->ops->sor = MatSOR_SeqAIJ; 4123ded6e084SBarry Smith A->ops->multadd = MatMultAdd_SeqAIJ; 4124ded6e084SBarry Smith A->ops->getrowij = MatGetRowIJ_SeqAIJ; 4125ded6e084SBarry Smith A->ops->restorerowij = MatRestoreRowIJ_SeqAIJ; 4126ded6e084SBarry Smith A->ops->getcolumnij = MatGetColumnIJ_SeqAIJ; 4127ded6e084SBarry Smith A->ops->restorecolumnij = MatRestoreColumnIJ_SeqAIJ; 4128ded6e084SBarry Smith A->ops->coloringpatch = 0; 4129ded6e084SBarry Smith A->ops->multdiagonalblock = 0; 41302205254eSKarl Rupp 41314c1414c8SBarry Smith ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr); 41324c1414c8SBarry Smith } else { 4133d5f3da31SBarry Smith if (!A->factortype) { 41344108e4d5SBarry Smith A->ops->mult = MatMult_SeqAIJ_Inode; 41354108e4d5SBarry Smith A->ops->sor = MatSOR_SeqAIJ_Inode; 41364108e4d5SBarry Smith A->ops->multadd = MatMultAdd_SeqAIJ_Inode; 4137375a6242SBarry Smith A->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode; 4138375a6242SBarry Smith if (A->rmap->n == A->cmap->n) { 41394108e4d5SBarry Smith A->ops->getrowij = MatGetRowIJ_SeqAIJ_Inode; 41404108e4d5SBarry Smith A->ops->restorerowij = MatRestoreRowIJ_SeqAIJ_Inode; 41414108e4d5SBarry Smith A->ops->getcolumnij = MatGetColumnIJ_SeqAIJ_Inode; 41424108e4d5SBarry Smith A->ops->restorecolumnij = MatRestoreColumnIJ_SeqAIJ_Inode; 41434108e4d5SBarry Smith A->ops->coloringpatch = MatColoringPatch_SeqAIJ_Inode; 4144375a6242SBarry Smith } 4145d3ac4fa3SBarry Smith } else { 4146d3ac4fa3SBarry Smith A->ops->solve = MatSolve_SeqAIJ_Inode_inplace; 4147d3ac4fa3SBarry Smith } 41484c1414c8SBarry Smith a->inode.node_count = node_count; 41494c1414c8SBarry Smith a->inode.size = ns; 41504c1414c8SBarry Smith ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr); 41514c1414c8SBarry Smith } 4152be6adb11SBarry Smith a->inode.checked = PETSC_TRUE; 4153*a02bda8eSBarry Smith a->inode.mat_nonzerostate = A->nonzerostate; 41544c1414c8SBarry Smith PetscFunctionReturn(0); 41554c1414c8SBarry Smith } 41564c1414c8SBarry Smith 415717454e89SShri Abhyankar #undef __FUNCT__ 4158150f0143SBarry Smith #define __FUNCT__ "MatDuplicate_SeqAIJ_Inode" 4159150f0143SBarry Smith PetscErrorCode MatDuplicate_SeqAIJ_Inode(Mat A,MatDuplicateOption cpvalues,Mat *C) 4160150f0143SBarry Smith { 4161150f0143SBarry Smith Mat B =*C; 4162150f0143SBarry Smith Mat_SeqAIJ *c=(Mat_SeqAIJ*)B->data,*a=(Mat_SeqAIJ*)A->data; 4163150f0143SBarry Smith PetscErrorCode ierr; 4164150f0143SBarry Smith PetscInt m=A->rmap->n; 4165150f0143SBarry Smith 4166150f0143SBarry Smith PetscFunctionBegin; 4167150f0143SBarry Smith c->inode.use = a->inode.use; 4168150f0143SBarry Smith c->inode.limit = a->inode.limit; 4169150f0143SBarry Smith c->inode.max_limit = a->inode.max_limit; 4170150f0143SBarry Smith if (a->inode.size) { 4171854ce69bSBarry Smith ierr = PetscMalloc1(m+1,&c->inode.size);CHKERRQ(ierr); 4172150f0143SBarry Smith c->inode.node_count = a->inode.node_count; 4173150f0143SBarry Smith ierr = PetscMemcpy(c->inode.size,a->inode.size,(m+1)*sizeof(PetscInt));CHKERRQ(ierr); 4174*a02bda8eSBarry Smith /* note the table of functions below should match that in MatSeqAIJCheckInode() */ 41752c451681SBarry Smith if (!B->factortype) { 41762c451681SBarry Smith B->ops->mult = MatMult_SeqAIJ_Inode; 41772c451681SBarry Smith B->ops->sor = MatSOR_SeqAIJ_Inode; 41782c451681SBarry Smith B->ops->multadd = MatMultAdd_SeqAIJ_Inode; 41792c451681SBarry Smith B->ops->getrowij = MatGetRowIJ_SeqAIJ_Inode; 41802c451681SBarry Smith B->ops->restorerowij = MatRestoreRowIJ_SeqAIJ_Inode; 41812c451681SBarry Smith B->ops->getcolumnij = MatGetColumnIJ_SeqAIJ_Inode; 41822c451681SBarry Smith B->ops->restorecolumnij = MatRestoreColumnIJ_SeqAIJ_Inode; 41832c451681SBarry Smith B->ops->coloringpatch = MatColoringPatch_SeqAIJ_Inode; 41842c451681SBarry Smith B->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode; 4185150f0143SBarry Smith } else { 41862c451681SBarry Smith B->ops->solve = MatSolve_SeqAIJ_Inode_inplace; 4187150f0143SBarry Smith } 4188150f0143SBarry Smith } else { 4189150f0143SBarry Smith c->inode.size = 0; 4190150f0143SBarry Smith c->inode.node_count = 0; 4191150f0143SBarry Smith } 4192150f0143SBarry Smith c->inode.ibdiagvalid = PETSC_FALSE; 4193150f0143SBarry Smith c->inode.ibdiag = 0; 4194150f0143SBarry Smith c->inode.bdiag = 0; 4195150f0143SBarry Smith PetscFunctionReturn(0); 4196150f0143SBarry Smith } 4197150f0143SBarry Smith 4198150f0143SBarry Smith #undef __FUNCT__ 419917454e89SShri Abhyankar #define __FUNCT__ "MatGetRow_FactoredLU" 42008758e1faSBarry Smith PETSC_STATIC_INLINE PetscErrorCode MatGetRow_FactoredLU(PetscInt *cols,PetscInt nzl,PetscInt nzu,PetscInt nz,const PetscInt *ai,const PetscInt *aj,const PetscInt *adiag,PetscInt row) 420117454e89SShri Abhyankar { 42028758e1faSBarry Smith PetscInt k; 42038758e1faSBarry Smith const PetscInt *vi; 42046e111a19SKarl Rupp 420517454e89SShri Abhyankar PetscFunctionBegin; 420617454e89SShri Abhyankar vi = aj + ai[row]; 420717454e89SShri Abhyankar for (k=0; k<nzl; k++) cols[k] = vi[k]; 420817454e89SShri Abhyankar vi = aj + adiag[row]; 420917454e89SShri Abhyankar cols[nzl] = vi[0]; 421017454e89SShri Abhyankar vi = aj + adiag[row+1]+1; 421117454e89SShri Abhyankar for (k=0; k<nzu; k++) cols[nzl+1+k] = vi[k]; 421217454e89SShri Abhyankar PetscFunctionReturn(0); 421317454e89SShri Abhyankar } 42146936b636SHong Zhang /* 4215*a02bda8eSBarry Smith MatSeqAIJCheckInode_FactorLU - Check Inode for factored seqaij matrix. 4216*a02bda8eSBarry Smith Modified from MatSeqAIJCheckInode(). 42176936b636SHong Zhang 42186936b636SHong Zhang Input Parameters: 4219abb87a52SBarry Smith . Mat A - ILU or LU matrix factor 4220abb87a52SBarry Smith 42216936b636SHong Zhang */ 4222019b515eSShri Abhyankar #undef __FUNCT__ 4223*a02bda8eSBarry Smith #define __FUNCT__ "MatSeqAIJCheckInode_FactorLU" 4224*a02bda8eSBarry Smith PetscErrorCode MatSeqAIJCheckInode_FactorLU(Mat A) 4225019b515eSShri Abhyankar { 4226019b515eSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 4227019b515eSShri Abhyankar PetscErrorCode ierr; 4228019b515eSShri Abhyankar PetscInt i,j,m,nzl1,nzu1,nzl2,nzu2,nzx,nzy,node_count,blk_size; 42298758e1faSBarry Smith PetscInt *cols1,*cols2,*ns; 42308758e1faSBarry Smith const PetscInt *ai = a->i,*aj = a->j, *adiag = a->diag; 4231ace3abfcSBarry Smith PetscBool flag; 4232019b515eSShri Abhyankar 4233019b515eSShri Abhyankar PetscFunctionBegin; 4234019b515eSShri Abhyankar if (!a->inode.use) PetscFunctionReturn(0); 4235abb87a52SBarry Smith if (a->inode.checked) PetscFunctionReturn(0); 4236019b515eSShri Abhyankar 4237019b515eSShri Abhyankar m = A->rmap->n; 42382205254eSKarl Rupp if (a->inode.size) ns = a->inode.size; 42392205254eSKarl Rupp else { 4240854ce69bSBarry Smith ierr = PetscMalloc1(m+1,&ns);CHKERRQ(ierr); 42412205254eSKarl Rupp } 4242019b515eSShri Abhyankar 4243019b515eSShri Abhyankar i = 0; 4244019b515eSShri Abhyankar node_count = 0; 4245dcca6d9dSJed Brown ierr = PetscMalloc2(m,&cols1,m,&cols2);CHKERRQ(ierr); 4246019b515eSShri Abhyankar while (i < m) { /* For each row */ 4247019b515eSShri Abhyankar nzl1 = ai[i+1] - ai[i]; /* Number of nonzeros in L */ 4248019b515eSShri Abhyankar nzu1 = adiag[i] - adiag[i+1] - 1; /* Number of nonzeros in U excluding diagonal*/ 4249019b515eSShri Abhyankar nzx = nzl1 + nzu1 + 1; 4250019b515eSShri Abhyankar MatGetRow_FactoredLU(cols1,nzl1,nzu1,nzx,ai,aj,adiag,i); 4251019b515eSShri Abhyankar 4252019b515eSShri Abhyankar /* Limits the number of elements in a node to 'a->inode.limit' */ 4253019b515eSShri Abhyankar for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) { 4254019b515eSShri Abhyankar nzl2 = ai[j+1] - ai[j]; 4255019b515eSShri Abhyankar nzu2 = adiag[j] - adiag[j+1] - 1; 4256019b515eSShri Abhyankar nzy = nzl2 + nzu2 + 1; 4257019b515eSShri Abhyankar if (nzy != nzx) break; 425817454e89SShri Abhyankar ierr = MatGetRow_FactoredLU(cols2,nzl2,nzu2,nzy,ai,aj,adiag,j);CHKERRQ(ierr); 4259019b515eSShri Abhyankar ierr = PetscMemcmp(cols1,cols2,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr); 42608758e1faSBarry Smith if (!flag) break; 4261019b515eSShri Abhyankar } 4262019b515eSShri Abhyankar ns[node_count++] = blk_size; 4263019b515eSShri Abhyankar i = j; 4264019b515eSShri Abhyankar } 42658758e1faSBarry Smith ierr = PetscFree2(cols1,cols2);CHKERRQ(ierr); 4266019b515eSShri Abhyankar /* If not enough inodes found,, do not use inode version of the routines */ 4267be6adb11SBarry Smith if (!m || node_count > .8*m) { 4268019b515eSShri Abhyankar ierr = PetscFree(ns);CHKERRQ(ierr); 42692205254eSKarl Rupp 4270019b515eSShri Abhyankar a->inode.node_count = 0; 42710298fd71SBarry Smith a->inode.size = NULL; 4272019b515eSShri Abhyankar a->inode.use = PETSC_FALSE; 42732205254eSKarl Rupp 4274019b515eSShri Abhyankar ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr); 4275019b515eSShri Abhyankar } else { 4276019b515eSShri Abhyankar A->ops->mult = 0; 4277019b515eSShri Abhyankar A->ops->sor = 0; 4278019b515eSShri Abhyankar A->ops->multadd = 0; 4279019b515eSShri Abhyankar A->ops->getrowij = 0; 4280019b515eSShri Abhyankar A->ops->restorerowij = 0; 4281019b515eSShri Abhyankar A->ops->getcolumnij = 0; 4282019b515eSShri Abhyankar A->ops->restorecolumnij = 0; 4283019b515eSShri Abhyankar A->ops->coloringpatch = 0; 4284019b515eSShri Abhyankar A->ops->multdiagonalblock = 0; 4285019b515eSShri Abhyankar a->inode.node_count = node_count; 4286019b515eSShri Abhyankar a->inode.size = ns; 42872205254eSKarl Rupp 4288019b515eSShri Abhyankar ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr); 4289019b515eSShri Abhyankar } 4290be6adb11SBarry Smith a->inode.checked = PETSC_TRUE; 4291019b515eSShri Abhyankar PetscFunctionReturn(0); 4292019b515eSShri Abhyankar } 4293019b515eSShri Abhyankar 4294acf2f550SJed Brown #undef __FUNCT__ 4295acf2f550SJed Brown #define __FUNCT__ "MatSeqAIJInvalidateDiagonal_Inode" 4296acf2f550SJed Brown PetscErrorCode MatSeqAIJInvalidateDiagonal_Inode(Mat A) 4297acf2f550SJed Brown { 4298acf2f550SJed Brown Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 4299acf2f550SJed Brown 4300acf2f550SJed Brown PetscFunctionBegin; 4301acf2f550SJed Brown a->inode.ibdiagvalid = PETSC_FALSE; 4302acf2f550SJed Brown PetscFunctionReturn(0); 4303acf2f550SJed Brown } 4304acf2f550SJed Brown 43054c1414c8SBarry Smith /* 43064c1414c8SBarry Smith This is really ugly. if inodes are used this replaces the 43074c1414c8SBarry Smith permutations with ones that correspond to rows/cols of the matrix 43084c1414c8SBarry Smith rather then inode blocks 43094c1414c8SBarry Smith */ 43104c1414c8SBarry Smith #undef __FUNCT__ 43114c1414c8SBarry Smith #define __FUNCT__ "MatInodeAdjustForInodes" 43127087cfbeSBarry Smith PetscErrorCode MatInodeAdjustForInodes(Mat A,IS *rperm,IS *cperm) 43134c1414c8SBarry Smith { 43144ac538c5SBarry Smith PetscErrorCode ierr; 43154c1414c8SBarry Smith 43164c1414c8SBarry Smith PetscFunctionBegin; 43174ac538c5SBarry Smith ierr = PetscTryMethod(A,"MatInodeAdjustForInodes_C",(Mat,IS*,IS*),(A,rperm,cperm));CHKERRQ(ierr); 43184c1414c8SBarry Smith PetscFunctionReturn(0); 43194c1414c8SBarry Smith } 43204c1414c8SBarry Smith 43214c1414c8SBarry Smith #undef __FUNCT__ 432253c77d0aSJed Brown #define __FUNCT__ "MatInodeAdjustForInodes_SeqAIJ_Inode" 43237087cfbeSBarry Smith PetscErrorCode MatInodeAdjustForInodes_SeqAIJ_Inode(Mat A,IS *rperm,IS *cperm) 43244c1414c8SBarry Smith { 43254c1414c8SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 43264c1414c8SBarry Smith PetscErrorCode ierr; 43275d0c19d7SBarry Smith PetscInt m = A->rmap->n,n = A->cmap->n,i,j,nslim_row = a->inode.node_count; 43285d0c19d7SBarry Smith const PetscInt *ridx,*cidx; 43294c1414c8SBarry Smith PetscInt row,col,*permr,*permc,*ns_row = a->inode.size,*tns,start_val,end_val,indx; 43304c1414c8SBarry Smith PetscInt nslim_col,*ns_col; 43314c1414c8SBarry Smith IS ris = *rperm,cis = *cperm; 43324c1414c8SBarry Smith 43334c1414c8SBarry Smith PetscFunctionBegin; 43344c1414c8SBarry Smith if (!a->inode.size) PetscFunctionReturn(0); /* no inodes so return */ 43354c1414c8SBarry Smith if (a->inode.node_count == m) PetscFunctionReturn(0); /* all inodes are of size 1 */ 43364c1414c8SBarry Smith 43374c1414c8SBarry Smith ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr); 4338854ce69bSBarry Smith ierr = PetscMalloc1(((nslim_row>nslim_col) ? nslim_row : nslim_col)+1,&tns);CHKERRQ(ierr); 4339dcca6d9dSJed Brown ierr = PetscMalloc2(m,&permr,n,&permc);CHKERRQ(ierr); 43404c1414c8SBarry Smith 43414c1414c8SBarry Smith ierr = ISGetIndices(ris,&ridx);CHKERRQ(ierr); 43424c1414c8SBarry Smith ierr = ISGetIndices(cis,&cidx);CHKERRQ(ierr); 43434c1414c8SBarry Smith 43444c1414c8SBarry Smith /* Form the inode structure for the rows of permuted matric using inv perm*/ 43454c1414c8SBarry Smith for (i=0,tns[0]=0; i<nslim_row; ++i) tns[i+1] = tns[i] + ns_row[i]; 43464c1414c8SBarry Smith 43474c1414c8SBarry Smith /* Construct the permutations for rows*/ 43484c1414c8SBarry Smith for (i=0,row = 0; i<nslim_row; ++i) { 43494c1414c8SBarry Smith indx = ridx[i]; 43504c1414c8SBarry Smith start_val = tns[indx]; 43514c1414c8SBarry Smith end_val = tns[indx + 1]; 43524c1414c8SBarry Smith for (j=start_val; j<end_val; ++j,++row) permr[row]= j; 43534c1414c8SBarry Smith } 43544c1414c8SBarry Smith 43554c1414c8SBarry Smith /* Form the inode structure for the columns of permuted matrix using inv perm*/ 43564c1414c8SBarry Smith for (i=0,tns[0]=0; i<nslim_col; ++i) tns[i+1] = tns[i] + ns_col[i]; 43574c1414c8SBarry Smith 43584c1414c8SBarry Smith /* Construct permutations for columns */ 43594c1414c8SBarry Smith for (i=0,col=0; i<nslim_col; ++i) { 43604c1414c8SBarry Smith indx = cidx[i]; 43614c1414c8SBarry Smith start_val = tns[indx]; 43624c1414c8SBarry Smith end_val = tns[indx + 1]; 43634c1414c8SBarry Smith for (j = start_val; j<end_val; ++j,++col) permc[col]= j; 43644c1414c8SBarry Smith } 43654c1414c8SBarry Smith 436670b3c8c7SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permr,PETSC_COPY_VALUES,rperm);CHKERRQ(ierr); 43670e83c824SBarry Smith ierr = ISSetPermutation(*rperm);CHKERRQ(ierr); 436870b3c8c7SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permc,PETSC_COPY_VALUES,cperm);CHKERRQ(ierr); 43690e83c824SBarry Smith ierr = ISSetPermutation(*cperm);CHKERRQ(ierr); 43704c1414c8SBarry Smith 43714c1414c8SBarry Smith ierr = ISRestoreIndices(ris,&ridx);CHKERRQ(ierr); 43724c1414c8SBarry Smith ierr = ISRestoreIndices(cis,&cidx);CHKERRQ(ierr); 43734c1414c8SBarry Smith 43744c1414c8SBarry Smith ierr = PetscFree(ns_col);CHKERRQ(ierr); 43750e83c824SBarry Smith ierr = PetscFree2(permr,permc);CHKERRQ(ierr); 43766bf464f9SBarry Smith ierr = ISDestroy(&cis);CHKERRQ(ierr); 43776bf464f9SBarry Smith ierr = ISDestroy(&ris);CHKERRQ(ierr); 43784c1414c8SBarry Smith ierr = PetscFree(tns);CHKERRQ(ierr); 43794c1414c8SBarry Smith PetscFunctionReturn(0); 43804c1414c8SBarry Smith } 43814c1414c8SBarry Smith 43824c1414c8SBarry Smith #undef __FUNCT__ 43834c1414c8SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes" 43844c1414c8SBarry Smith /*@C 43854c1414c8SBarry Smith MatInodeGetInodeSizes - Returns the inode information of the Inode matrix. 43864c1414c8SBarry Smith 43873f9fe445SBarry Smith Not Collective 43884c1414c8SBarry Smith 43894c1414c8SBarry Smith Input Parameter: 43904c1414c8SBarry Smith . A - the Inode matrix or matrix derived from the Inode class -- e.g., SeqAIJ 43914c1414c8SBarry Smith 43924c1414c8SBarry Smith Output Parameter: 43934c1414c8SBarry Smith + node_count - no of inodes present in the matrix. 43944c1414c8SBarry Smith . sizes - an array of size node_count,with sizes of each inode. 43954c1414c8SBarry Smith - limit - the max size used to generate the inodes. 43964c1414c8SBarry Smith 43974c1414c8SBarry Smith Level: advanced 43984c1414c8SBarry Smith 43994c1414c8SBarry Smith Notes: This routine returns some internal storage information 44004c1414c8SBarry Smith of the matrix, it is intended to be used by advanced users. 44014c1414c8SBarry Smith It should be called after the matrix is assembled. 44024c1414c8SBarry Smith The contents of the sizes[] array should not be changed. 44030298fd71SBarry Smith NULL may be passed for information not requested. 44044c1414c8SBarry Smith 44054c1414c8SBarry Smith .keywords: matrix, seqaij, get, inode 44064c1414c8SBarry Smith 44074c1414c8SBarry Smith .seealso: MatGetInfo() 44084c1414c8SBarry Smith @*/ 44097087cfbeSBarry Smith PetscErrorCode MatInodeGetInodeSizes(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit) 44104c1414c8SBarry Smith { 44114c1414c8SBarry Smith PetscErrorCode ierr,(*f)(Mat,PetscInt*,PetscInt*[],PetscInt*); 44124c1414c8SBarry Smith 44134c1414c8SBarry Smith PetscFunctionBegin; 4414e32f2f54SBarry Smith if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix"); 44150005d66cSJed Brown ierr = PetscObjectQueryFunction((PetscObject)A,"MatInodeGetInodeSizes_C",&f);CHKERRQ(ierr); 44164c1414c8SBarry Smith if (f) { 44174c1414c8SBarry Smith ierr = (*f)(A,node_count,sizes,limit);CHKERRQ(ierr); 44184c1414c8SBarry Smith } 44194c1414c8SBarry Smith PetscFunctionReturn(0); 44204c1414c8SBarry Smith } 44214c1414c8SBarry Smith 44224c1414c8SBarry Smith #undef __FUNCT__ 44234108e4d5SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes_SeqAIJ_Inode" 44247087cfbeSBarry Smith PetscErrorCode MatInodeGetInodeSizes_SeqAIJ_Inode(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit) 44254c1414c8SBarry Smith { 44264c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 44274c1414c8SBarry Smith 44284c1414c8SBarry Smith PetscFunctionBegin; 44294c1414c8SBarry Smith if (node_count) *node_count = a->inode.node_count; 44304c1414c8SBarry Smith if (sizes) *sizes = a->inode.size; 44314c1414c8SBarry Smith if (limit) *limit = a->inode.limit; 44324c1414c8SBarry Smith PetscFunctionReturn(0); 44334c1414c8SBarry Smith } 4434