14c1414c8SBarry Smith 24c1414c8SBarry Smith /* 34c1414c8SBarry Smith This file provides high performance routines for the Inode format (compressed sparse row) 44c1414c8SBarry Smith by taking advantage of rows with identical nonzero structure (I-nodes). 54c1414c8SBarry Smith */ 6c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/aij.h> 74c1414c8SBarry Smith 84c1414c8SBarry Smith #undef __FUNCT__ 94c1414c8SBarry Smith #define __FUNCT__ "Mat_CreateColInode" 104c1414c8SBarry Smith static PetscErrorCode Mat_CreateColInode(Mat A,PetscInt* size,PetscInt ** ns) 114c1414c8SBarry Smith { 124c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 134c1414c8SBarry Smith PetscErrorCode ierr; 144c1414c8SBarry Smith PetscInt i,count,m,n,min_mn,*ns_row,*ns_col; 154c1414c8SBarry Smith 164c1414c8SBarry Smith PetscFunctionBegin; 17d0f46423SBarry Smith n = A->cmap->n; 18d0f46423SBarry Smith m = A->rmap->n; 194c1414c8SBarry Smith ns_row = a->inode.size; 204c1414c8SBarry Smith 214c1414c8SBarry Smith min_mn = (m < n) ? m : n; 224c1414c8SBarry Smith if (!ns) { 234c1414c8SBarry Smith for (count=0,i=0; count<min_mn; count+=ns_row[i],i++); 244c1414c8SBarry Smith for(; count+1 < n; count++,i++); 254c1414c8SBarry Smith if (count < n) { 264c1414c8SBarry Smith i++; 274c1414c8SBarry Smith } 284c1414c8SBarry Smith *size = i; 294c1414c8SBarry Smith PetscFunctionReturn(0); 304c1414c8SBarry Smith } 314c1414c8SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscInt),&ns_col);CHKERRQ(ierr); 324c1414c8SBarry Smith 334c1414c8SBarry Smith /* Use the same row structure wherever feasible. */ 344c1414c8SBarry Smith for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) { 354c1414c8SBarry Smith ns_col[i] = ns_row[i]; 364c1414c8SBarry Smith } 374c1414c8SBarry Smith 384c1414c8SBarry Smith /* if m < n; pad up the remainder with inode_limit */ 394c1414c8SBarry Smith for(; count+1 < n; count++,i++) { 404c1414c8SBarry Smith ns_col[i] = 1; 414c1414c8SBarry Smith } 424c1414c8SBarry Smith /* The last node is the odd ball. padd it up with the remaining rows; */ 434c1414c8SBarry Smith if (count < n) { 444c1414c8SBarry Smith ns_col[i] = n - count; 454c1414c8SBarry Smith i++; 464c1414c8SBarry Smith } else if (count > n) { 474c1414c8SBarry Smith /* Adjust for the over estimation */ 484c1414c8SBarry Smith ns_col[i-1] += n - count; 494c1414c8SBarry Smith } 504c1414c8SBarry Smith *size = i; 514c1414c8SBarry Smith *ns = ns_col; 524c1414c8SBarry Smith PetscFunctionReturn(0); 534c1414c8SBarry Smith } 544c1414c8SBarry Smith 554c1414c8SBarry Smith 564c1414c8SBarry Smith /* 574c1414c8SBarry Smith This builds symmetric version of nonzero structure, 584c1414c8SBarry Smith */ 594c1414c8SBarry Smith #undef __FUNCT__ 604108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode_Symmetric" 614108e4d5SBarry Smith static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Symmetric(Mat A,PetscInt *iia[],PetscInt *jja[],PetscInt ishift,PetscInt oshift) 624c1414c8SBarry Smith { 634c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 644c1414c8SBarry Smith PetscErrorCode ierr; 654c1414c8SBarry Smith PetscInt *work,*ia,*ja,*j,nz,nslim_row,nslim_col,m,row,col,*jmax,n; 664c1414c8SBarry Smith PetscInt *tns,*tvc,*ns_row = a->inode.size,*ns_col,nsz,i1,i2,*ai= a->i,*aj = a->j; 674c1414c8SBarry Smith 684c1414c8SBarry Smith PetscFunctionBegin; 694c1414c8SBarry Smith nslim_row = a->inode.node_count; 70d0f46423SBarry Smith m = A->rmap->n; 71d0f46423SBarry Smith n = A->cmap->n; 72e32f2f54SBarry Smith if (m != n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatGetRowIJ_SeqAIJ_Inode_Symmetric: Matrix should be square"); 734c1414c8SBarry Smith 744c1414c8SBarry Smith /* Use the row_inode as column_inode */ 754c1414c8SBarry Smith nslim_col = nslim_row; 764c1414c8SBarry Smith ns_col = ns_row; 774c1414c8SBarry Smith 784c1414c8SBarry Smith /* allocate space for reformated inode structure */ 794c1414c8SBarry Smith ierr = PetscMalloc((nslim_col+1)*sizeof(PetscInt),&tns);CHKERRQ(ierr); 804c1414c8SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscInt),&tvc);CHKERRQ(ierr); 814c1414c8SBarry Smith for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1]+ ns_row[i1]; 824c1414c8SBarry Smith 834c1414c8SBarry Smith for (i1=0,col=0; i1<nslim_col; ++i1){ 844c1414c8SBarry Smith nsz = ns_col[i1]; 854c1414c8SBarry Smith for (i2=0; i2<nsz; ++i2,++col) 864c1414c8SBarry Smith tvc[col] = i1; 874c1414c8SBarry Smith } 884c1414c8SBarry Smith /* allocate space for row pointers */ 894c1414c8SBarry Smith ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&ia);CHKERRQ(ierr); 904c1414c8SBarry Smith *iia = ia; 914c1414c8SBarry Smith ierr = PetscMemzero(ia,(nslim_row+1)*sizeof(PetscInt));CHKERRQ(ierr); 924c1414c8SBarry Smith ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&work);CHKERRQ(ierr); 934c1414c8SBarry Smith 944c1414c8SBarry Smith /* determine the number of columns in each row */ 954c1414c8SBarry Smith ia[0] = oshift; 964c1414c8SBarry Smith for (i1=0,row=0 ; i1<nslim_row; row+=ns_row[i1],i1++) { 974c1414c8SBarry Smith 984c1414c8SBarry Smith j = aj + ai[row] + ishift; 994c1414c8SBarry Smith jmax = aj + ai[row+1] + ishift; 1004c1414c8SBarry Smith i2 = 0; 1014c1414c8SBarry Smith col = *j++ + ishift; 1024c1414c8SBarry Smith i2 = tvc[col]; 1034c1414c8SBarry Smith while (i2<i1 && j<jmax) { /* 1.[-xx-d-xx--] 2.[-xx-------],off-diagonal elemets */ 1044c1414c8SBarry Smith ia[i1+1]++; 1054c1414c8SBarry Smith ia[i2+1]++; 1064c1414c8SBarry Smith i2++; /* Start col of next node */ 107*90d2dec7SBarry Smith while((j<jmax) && ((col=*j+ishift)<tns[i2])) ++j; 1084c1414c8SBarry Smith i2 = tvc[col]; 1094c1414c8SBarry Smith } 1104c1414c8SBarry Smith if(i2 == i1) ia[i2+1]++; /* now the diagonal element */ 1114c1414c8SBarry Smith } 1124c1414c8SBarry Smith 1134c1414c8SBarry Smith /* shift ia[i] to point to next row */ 1144c1414c8SBarry Smith for (i1=1; i1<nslim_row+1; i1++) { 1154c1414c8SBarry Smith row = ia[i1-1]; 1164c1414c8SBarry Smith ia[i1] += row; 1174c1414c8SBarry Smith work[i1-1] = row - oshift; 1184c1414c8SBarry Smith } 1194c1414c8SBarry Smith 1204c1414c8SBarry Smith /* allocate space for column pointers */ 1214c1414c8SBarry Smith nz = ia[nslim_row] + (!ishift); 1224c1414c8SBarry Smith ierr = PetscMalloc(nz*sizeof(PetscInt),&ja);CHKERRQ(ierr); 1234c1414c8SBarry Smith *jja = ja; 1244c1414c8SBarry Smith 1254c1414c8SBarry Smith /* loop over lower triangular part putting into ja */ 1264c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row += ns_row[i1],i1++) { 1274c1414c8SBarry Smith j = aj + ai[row] + ishift; 1284c1414c8SBarry Smith jmax = aj + ai[row+1] + ishift; 1294c1414c8SBarry Smith i2 = 0; /* Col inode index */ 1304c1414c8SBarry Smith col = *j++ + ishift; 1314c1414c8SBarry Smith i2 = tvc[col]; 1324c1414c8SBarry Smith while (i2<i1 && j<jmax) { 1334c1414c8SBarry Smith ja[work[i2]++] = i1 + oshift; 1344c1414c8SBarry Smith ja[work[i1]++] = i2 + oshift; 1354c1414c8SBarry Smith ++i2; 136*90d2dec7SBarry Smith while((j<jmax) && ((col=*j+ishift)< tns[i2])) ++j; /* Skip rest col indices in this node */ 1374c1414c8SBarry Smith i2 = tvc[col]; 1384c1414c8SBarry Smith } 1394c1414c8SBarry Smith if (i2 == i1) ja[work[i1]++] = i2 + oshift; 1404c1414c8SBarry Smith 1414c1414c8SBarry Smith } 1424c1414c8SBarry Smith ierr = PetscFree(work);CHKERRQ(ierr); 1434c1414c8SBarry Smith ierr = PetscFree(tns);CHKERRQ(ierr); 1444c1414c8SBarry Smith ierr = PetscFree(tvc);CHKERRQ(ierr); 1454c1414c8SBarry Smith PetscFunctionReturn(0); 1464c1414c8SBarry Smith } 1474c1414c8SBarry Smith 1484c1414c8SBarry Smith /* 1494c1414c8SBarry Smith This builds nonsymmetric version of nonzero structure, 1504c1414c8SBarry Smith */ 1514c1414c8SBarry Smith #undef __FUNCT__ 1524108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric" 1534108e4d5SBarry Smith static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,PetscInt *iia[],PetscInt *jja[],PetscInt ishift,PetscInt oshift) 1544c1414c8SBarry Smith { 1554c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1564c1414c8SBarry Smith PetscErrorCode ierr; 1574c1414c8SBarry Smith PetscInt *work,*ia,*ja,*j,nz,nslim_row,n,row,col,*ns_col,nslim_col; 1584c1414c8SBarry Smith PetscInt *tns,*tvc,*ns_row = a->inode.size,nsz,i1,i2,*ai= a->i,*aj = a->j; 1594c1414c8SBarry Smith 1604c1414c8SBarry Smith PetscFunctionBegin; 1614c1414c8SBarry Smith nslim_row = a->inode.node_count; 162d0f46423SBarry Smith n = A->cmap->n; 1634c1414c8SBarry Smith 1644c1414c8SBarry Smith /* Create The column_inode for this matrix */ 1654c1414c8SBarry Smith ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr); 1664c1414c8SBarry Smith 1674c1414c8SBarry Smith /* allocate space for reformated column_inode structure */ 1684c1414c8SBarry Smith ierr = PetscMalloc((nslim_col +1)*sizeof(PetscInt),&tns);CHKERRQ(ierr); 1694c1414c8SBarry Smith ierr = PetscMalloc((n +1)*sizeof(PetscInt),&tvc);CHKERRQ(ierr); 1704c1414c8SBarry Smith for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1]; 1714c1414c8SBarry Smith 1724c1414c8SBarry Smith for (i1=0,col=0; i1<nslim_col; ++i1){ 1734c1414c8SBarry Smith nsz = ns_col[i1]; 1744c1414c8SBarry Smith for (i2=0; i2<nsz; ++i2,++col) 1754c1414c8SBarry Smith tvc[col] = i1; 1764c1414c8SBarry Smith } 1774c1414c8SBarry Smith /* allocate space for row pointers */ 1784c1414c8SBarry Smith ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&ia);CHKERRQ(ierr); 1794c1414c8SBarry Smith *iia = ia; 1804c1414c8SBarry Smith ierr = PetscMemzero(ia,(nslim_row+1)*sizeof(PetscInt));CHKERRQ(ierr); 1814c1414c8SBarry Smith ierr = PetscMalloc((nslim_row+1)*sizeof(PetscInt),&work);CHKERRQ(ierr); 1824c1414c8SBarry Smith 1834c1414c8SBarry Smith /* determine the number of columns in each row */ 1844c1414c8SBarry Smith ia[0] = oshift; 1854c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 1864c1414c8SBarry Smith j = aj + ai[row] + ishift; 1874c1414c8SBarry Smith col = *j++ + ishift; 1884c1414c8SBarry Smith i2 = tvc[col]; 1894c1414c8SBarry Smith nz = ai[row+1] - ai[row]; 1904c1414c8SBarry Smith while (nz-- > 0) { /* off-diagonal elemets */ 1914c1414c8SBarry Smith ia[i1+1]++; 1924c1414c8SBarry Smith i2++; /* Start col of next node */ 1934c1414c8SBarry Smith while (((col = *j++ + ishift) < tns[i2]) && nz > 0) {nz--;} 1944c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 1954c1414c8SBarry Smith } 1964c1414c8SBarry Smith } 1974c1414c8SBarry Smith 1984c1414c8SBarry Smith /* shift ia[i] to point to next row */ 1994c1414c8SBarry Smith for (i1=1; i1<nslim_row+1; i1++) { 2004c1414c8SBarry Smith row = ia[i1-1]; 2014c1414c8SBarry Smith ia[i1] += row; 2024c1414c8SBarry Smith work[i1-1] = row - oshift; 2034c1414c8SBarry Smith } 2044c1414c8SBarry Smith 2054c1414c8SBarry Smith /* allocate space for column pointers */ 2064c1414c8SBarry Smith nz = ia[nslim_row] + (!ishift); 2074c1414c8SBarry Smith ierr = PetscMalloc(nz*sizeof(PetscInt),&ja);CHKERRQ(ierr); 2084c1414c8SBarry Smith *jja = ja; 2094c1414c8SBarry Smith 2104c1414c8SBarry Smith /* loop over matrix putting into ja */ 2114c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 2124c1414c8SBarry Smith j = aj + ai[row] + ishift; 2134c1414c8SBarry Smith i2 = 0; /* Col inode index */ 2144c1414c8SBarry Smith col = *j++ + ishift; 2154c1414c8SBarry Smith i2 = tvc[col]; 2164c1414c8SBarry Smith nz = ai[row+1] - ai[row]; 2174c1414c8SBarry Smith while (nz-- > 0) { 2184c1414c8SBarry Smith ja[work[i1]++] = i2 + oshift; 2194c1414c8SBarry Smith ++i2; 2204c1414c8SBarry Smith while(((col = *j++ + ishift) < tns[i2]) && nz > 0) {nz--;} 2214c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 2224c1414c8SBarry Smith } 2234c1414c8SBarry Smith } 2244c1414c8SBarry Smith ierr = PetscFree(ns_col);CHKERRQ(ierr); 2254c1414c8SBarry Smith ierr = PetscFree(work);CHKERRQ(ierr); 2264c1414c8SBarry Smith ierr = PetscFree(tns);CHKERRQ(ierr); 2274c1414c8SBarry Smith ierr = PetscFree(tvc);CHKERRQ(ierr); 2284c1414c8SBarry Smith PetscFunctionReturn(0); 2294c1414c8SBarry Smith } 2304c1414c8SBarry Smith 2314c1414c8SBarry Smith #undef __FUNCT__ 2324108e4d5SBarry Smith #define __FUNCT__ "MatGetRowIJ_SeqAIJ_Inode" 233ace3abfcSBarry Smith static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,PetscInt *ia[],PetscInt *ja[],PetscBool *done) 2344c1414c8SBarry Smith { 2354c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2364c1414c8SBarry Smith PetscErrorCode ierr; 2374c1414c8SBarry Smith 2384c1414c8SBarry Smith PetscFunctionBegin; 2394c1414c8SBarry Smith *n = a->inode.node_count; 2404c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 2418f7157efSSatish Balay if (!blockcompressed) { 2428f7157efSSatish Balay ierr = MatGetRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);; 2438f7157efSSatish Balay } else if (symmetric) { 2444108e4d5SBarry Smith ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 2454c1414c8SBarry Smith } else { 2464108e4d5SBarry Smith ierr = MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 2474c1414c8SBarry Smith } 2484c1414c8SBarry Smith PetscFunctionReturn(0); 2494c1414c8SBarry Smith } 2504c1414c8SBarry Smith 2514c1414c8SBarry Smith #undef __FUNCT__ 2524108e4d5SBarry Smith #define __FUNCT__ "MatRestoreRowIJ_SeqAIJ_Inode" 253ace3abfcSBarry Smith static PetscErrorCode MatRestoreRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,PetscInt *ia[],PetscInt *ja[],PetscBool *done) 2544c1414c8SBarry Smith { 2554c1414c8SBarry Smith PetscErrorCode ierr; 2564c1414c8SBarry Smith 2574c1414c8SBarry Smith PetscFunctionBegin; 2584c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 2598f7157efSSatish Balay 2608f7157efSSatish Balay if (!blockcompressed) { 2618f7157efSSatish Balay ierr = MatRestoreRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);; 2628f7157efSSatish Balay } else { 2634c1414c8SBarry Smith ierr = PetscFree(*ia);CHKERRQ(ierr); 2644c1414c8SBarry Smith ierr = PetscFree(*ja);CHKERRQ(ierr); 2658f7157efSSatish Balay } 2668f7157efSSatish Balay 2674c1414c8SBarry Smith PetscFunctionReturn(0); 2684c1414c8SBarry Smith } 2694c1414c8SBarry Smith 2704c1414c8SBarry Smith /* ----------------------------------------------------------- */ 2714c1414c8SBarry Smith 2724c1414c8SBarry Smith #undef __FUNCT__ 2734108e4d5SBarry Smith #define __FUNCT__ "MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric" 2744108e4d5SBarry Smith static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,PetscInt *iia[],PetscInt *jja[],PetscInt ishift,PetscInt oshift) 2754c1414c8SBarry Smith { 2764c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2774c1414c8SBarry Smith PetscErrorCode ierr; 2784c1414c8SBarry Smith PetscInt *work,*ia,*ja,*j,nz,nslim_row, n,row,col,*ns_col,nslim_col; 2794c1414c8SBarry Smith PetscInt *tns,*tvc,*ns_row = a->inode.size,nsz,i1,i2,*ai= a->i,*aj = a->j; 2804c1414c8SBarry Smith 2814c1414c8SBarry Smith PetscFunctionBegin; 2824c1414c8SBarry Smith nslim_row = a->inode.node_count; 283d0f46423SBarry Smith n = A->cmap->n; 2844c1414c8SBarry Smith 2854c1414c8SBarry Smith /* Create The column_inode for this matrix */ 2864c1414c8SBarry Smith ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr); 2874c1414c8SBarry Smith 2884c1414c8SBarry Smith /* allocate space for reformated column_inode structure */ 2894c1414c8SBarry Smith ierr = PetscMalloc((nslim_col + 1)*sizeof(PetscInt),&tns);CHKERRQ(ierr); 2904c1414c8SBarry Smith ierr = PetscMalloc((n + 1)*sizeof(PetscInt),&tvc);CHKERRQ(ierr); 2914c1414c8SBarry Smith for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1]; 2924c1414c8SBarry Smith 2934c1414c8SBarry Smith for (i1=0,col=0; i1<nslim_col; ++i1){ 2944c1414c8SBarry Smith nsz = ns_col[i1]; 2954c1414c8SBarry Smith for (i2=0; i2<nsz; ++i2,++col) 2964c1414c8SBarry Smith tvc[col] = i1; 2974c1414c8SBarry Smith } 2984c1414c8SBarry Smith /* allocate space for column pointers */ 2994c1414c8SBarry Smith ierr = PetscMalloc((nslim_col+1)*sizeof(PetscInt),&ia);CHKERRQ(ierr); 3004c1414c8SBarry Smith *iia = ia; 3014c1414c8SBarry Smith ierr = PetscMemzero(ia,(nslim_col+1)*sizeof(PetscInt));CHKERRQ(ierr); 3024c1414c8SBarry Smith ierr = PetscMalloc((nslim_col+1)*sizeof(PetscInt),&work);CHKERRQ(ierr); 3034c1414c8SBarry Smith 3044c1414c8SBarry Smith /* determine the number of columns in each row */ 3054c1414c8SBarry Smith ia[0] = oshift; 3064c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 3074c1414c8SBarry Smith j = aj + ai[row] + ishift; 3084c1414c8SBarry Smith col = *j++ + ishift; 3094c1414c8SBarry Smith i2 = tvc[col]; 3104c1414c8SBarry Smith nz = ai[row+1] - ai[row]; 3114c1414c8SBarry Smith while (nz-- > 0) { /* off-diagonal elemets */ 3124c1414c8SBarry Smith /* ia[i1+1]++; */ 3134c1414c8SBarry Smith ia[i2+1]++; 3144c1414c8SBarry Smith i2++; 3154c1414c8SBarry Smith while (((col = *j++ + ishift) < tns[i2]) && nz > 0) {nz--;} 3164c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 3174c1414c8SBarry Smith } 3184c1414c8SBarry Smith } 3194c1414c8SBarry Smith 3204c1414c8SBarry Smith /* shift ia[i] to point to next col */ 3214c1414c8SBarry Smith for (i1=1; i1<nslim_col+1; i1++) { 3224c1414c8SBarry Smith col = ia[i1-1]; 3234c1414c8SBarry Smith ia[i1] += col; 3244c1414c8SBarry Smith work[i1-1] = col - oshift; 3254c1414c8SBarry Smith } 3264c1414c8SBarry Smith 3274c1414c8SBarry Smith /* allocate space for column pointers */ 3284c1414c8SBarry Smith nz = ia[nslim_col] + (!ishift); 3294c1414c8SBarry Smith ierr = PetscMalloc(nz*sizeof(PetscInt),&ja);CHKERRQ(ierr); 3304c1414c8SBarry Smith *jja = ja; 3314c1414c8SBarry Smith 3324c1414c8SBarry Smith /* loop over matrix putting into ja */ 3334c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 3344c1414c8SBarry Smith j = aj + ai[row] + ishift; 3354c1414c8SBarry Smith i2 = 0; /* Col inode index */ 3364c1414c8SBarry Smith col = *j++ + ishift; 3374c1414c8SBarry Smith i2 = tvc[col]; 3384c1414c8SBarry Smith nz = ai[row+1] - ai[row]; 3394c1414c8SBarry Smith while (nz-- > 0) { 3404c1414c8SBarry Smith /* ja[work[i1]++] = i2 + oshift; */ 3414c1414c8SBarry Smith ja[work[i2]++] = i1 + oshift; 3424c1414c8SBarry Smith i2++; 3434c1414c8SBarry Smith while(((col = *j++ + ishift) < tns[i2]) && nz > 0) {nz--;} 3444c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 3454c1414c8SBarry Smith } 3464c1414c8SBarry Smith } 3474c1414c8SBarry Smith ierr = PetscFree(ns_col);CHKERRQ(ierr); 3484c1414c8SBarry Smith ierr = PetscFree(work);CHKERRQ(ierr); 3494c1414c8SBarry Smith ierr = PetscFree(tns);CHKERRQ(ierr); 3504c1414c8SBarry Smith ierr = PetscFree(tvc);CHKERRQ(ierr); 3514c1414c8SBarry Smith PetscFunctionReturn(0); 3524c1414c8SBarry Smith } 3534c1414c8SBarry Smith 3544c1414c8SBarry Smith #undef __FUNCT__ 3554108e4d5SBarry Smith #define __FUNCT__ "MatGetColumnIJ_SeqAIJ_Inode" 356ace3abfcSBarry Smith static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,PetscInt *ia[],PetscInt *ja[],PetscBool *done) 3574c1414c8SBarry Smith { 3584c1414c8SBarry Smith PetscErrorCode ierr; 3594c1414c8SBarry Smith 3604c1414c8SBarry Smith PetscFunctionBegin; 3614c1414c8SBarry Smith ierr = Mat_CreateColInode(A,n,PETSC_NULL);CHKERRQ(ierr); 3624c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 3634c1414c8SBarry Smith 3648f7157efSSatish Balay if (!blockcompressed) { 3658f7157efSSatish Balay ierr = MatGetColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);; 3668f7157efSSatish Balay } else if (symmetric) { 3674c1414c8SBarry Smith /* Since the indices are symmetric it does'nt matter */ 3684108e4d5SBarry Smith ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 3694c1414c8SBarry Smith } else { 3704108e4d5SBarry Smith ierr = MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 3714c1414c8SBarry Smith } 3724c1414c8SBarry Smith PetscFunctionReturn(0); 3734c1414c8SBarry Smith } 3744c1414c8SBarry Smith 3754c1414c8SBarry Smith #undef __FUNCT__ 3764108e4d5SBarry Smith #define __FUNCT__ "MatRestoreColumnIJ_SeqAIJ_Inode" 377ace3abfcSBarry Smith static PetscErrorCode MatRestoreColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,PetscInt *ia[],PetscInt *ja[],PetscBool *done) 3784c1414c8SBarry Smith { 3794c1414c8SBarry Smith PetscErrorCode ierr; 3804c1414c8SBarry Smith 3814c1414c8SBarry Smith PetscFunctionBegin; 3824c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 3838f7157efSSatish Balay if (!blockcompressed) { 3848f7157efSSatish Balay ierr = MatRestoreColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr);; 3858f7157efSSatish Balay } else { 3864c1414c8SBarry Smith ierr = PetscFree(*ia);CHKERRQ(ierr); 3874c1414c8SBarry Smith ierr = PetscFree(*ja);CHKERRQ(ierr); 3888f7157efSSatish Balay } 3894c1414c8SBarry Smith PetscFunctionReturn(0); 3904c1414c8SBarry Smith } 3914c1414c8SBarry Smith 3924c1414c8SBarry Smith /* ----------------------------------------------------------- */ 3934c1414c8SBarry Smith 3944c1414c8SBarry Smith #undef __FUNCT__ 3954108e4d5SBarry Smith #define __FUNCT__ "MatMult_SeqAIJ_Inode" 3964108e4d5SBarry Smith static PetscErrorCode MatMult_SeqAIJ_Inode(Mat A,Vec xx,Vec yy) 3974c1414c8SBarry Smith { 3984c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3994c1414c8SBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5,tmp0,tmp1; 400d9fead3dSBarry Smith PetscScalar *y; 401dd6ea824SBarry Smith const PetscScalar *x; 402dd6ea824SBarry Smith const MatScalar *v1,*v2,*v3,*v4,*v5; 4034c1414c8SBarry Smith PetscErrorCode ierr; 40498c9bda7SSatish Balay PetscInt *idx,i1,i2,n,i,row,node_max,*ns,*ii,nsz,sz,nonzerorow=0; 4054c1414c8SBarry Smith 4064c1414c8SBarry Smith #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 4074c1414c8SBarry Smith #pragma disjoint(*x,*y,*v1,*v2,*v3,*v4,*v5) 4084c1414c8SBarry Smith #endif 4094c1414c8SBarry Smith 4104c1414c8SBarry Smith PetscFunctionBegin; 411e32f2f54SBarry Smith if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 4124c1414c8SBarry Smith node_max = a->inode.node_count; 4134c1414c8SBarry Smith ns = a->inode.size; /* Node Size array */ 4143649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 4154c1414c8SBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 4164c1414c8SBarry Smith idx = a->j; 4174c1414c8SBarry Smith v1 = a->a; 4184c1414c8SBarry Smith ii = a->i; 4194c1414c8SBarry Smith 4204c1414c8SBarry Smith for (i = 0,row = 0; i< node_max; ++i){ 4214c1414c8SBarry Smith nsz = ns[i]; 4224c1414c8SBarry Smith n = ii[1] - ii[0]; 42398c9bda7SSatish Balay nonzerorow += (n>0)*nsz; 4244c1414c8SBarry Smith ii += nsz; 42550d8bf02SJed Brown PetscPrefetchBlock(idx+nsz*n,n,0,PETSC_PREFETCH_HINT_NTA); /* Prefetch the indices for the block row after the current one */ 42650d8bf02SJed Brown PetscPrefetchBlock(v1+nsz*n,nsz*n,0,PETSC_PREFETCH_HINT_NTA); /* Prefetch the values for the block row after the current one */ 4274c1414c8SBarry Smith sz = n; /* No of non zeros in this row */ 4284c1414c8SBarry Smith /* Switch on the size of Node */ 4294c1414c8SBarry Smith switch (nsz){ /* Each loop in 'case' is unrolled */ 4304c1414c8SBarry Smith case 1 : 43175567043SBarry Smith sum1 = 0.; 4324c1414c8SBarry Smith 4334c1414c8SBarry Smith for(n = 0; n< sz-1; n+=2) { 4344c1414c8SBarry Smith i1 = idx[0]; /* The instructions are ordered to */ 4354c1414c8SBarry Smith i2 = idx[1]; /* make the compiler's job easy */ 4364c1414c8SBarry Smith idx += 2; 4374c1414c8SBarry Smith tmp0 = x[i1]; 4384c1414c8SBarry Smith tmp1 = x[i2]; 4394c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 4404c1414c8SBarry Smith } 4414c1414c8SBarry Smith 4424c1414c8SBarry Smith if (n == sz-1){ /* Take care of the last nonzero */ 4434c1414c8SBarry Smith tmp0 = x[*idx++]; 4444c1414c8SBarry Smith sum1 += *v1++ * tmp0; 4454c1414c8SBarry Smith } 4464c1414c8SBarry Smith y[row++]=sum1; 4474c1414c8SBarry Smith break; 4484c1414c8SBarry Smith case 2: 44975567043SBarry Smith sum1 = 0.; 45075567043SBarry Smith sum2 = 0.; 4514c1414c8SBarry Smith v2 = v1 + n; 4524c1414c8SBarry Smith 4534c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 4544c1414c8SBarry Smith i1 = idx[0]; 4554c1414c8SBarry Smith i2 = idx[1]; 4564c1414c8SBarry Smith idx += 2; 4574c1414c8SBarry Smith tmp0 = x[i1]; 4584c1414c8SBarry Smith tmp1 = x[i2]; 4594c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 4604c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 4614c1414c8SBarry Smith } 4624c1414c8SBarry Smith if (n == sz-1){ 4634c1414c8SBarry Smith tmp0 = x[*idx++]; 4644c1414c8SBarry Smith sum1 += *v1++ * tmp0; 4654c1414c8SBarry Smith sum2 += *v2++ * tmp0; 4664c1414c8SBarry Smith } 4674c1414c8SBarry Smith y[row++]=sum1; 4684c1414c8SBarry Smith y[row++]=sum2; 4694c1414c8SBarry Smith v1 =v2; /* Since the next block to be processed starts there*/ 4704c1414c8SBarry Smith idx +=sz; 4714c1414c8SBarry Smith break; 4724c1414c8SBarry Smith case 3: 47375567043SBarry Smith sum1 = 0.; 47475567043SBarry Smith sum2 = 0.; 47575567043SBarry Smith sum3 = 0.; 4764c1414c8SBarry Smith v2 = v1 + n; 4774c1414c8SBarry Smith v3 = v2 + n; 4784c1414c8SBarry Smith 4794c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 4804c1414c8SBarry Smith i1 = idx[0]; 4814c1414c8SBarry Smith i2 = idx[1]; 4824c1414c8SBarry Smith idx += 2; 4834c1414c8SBarry Smith tmp0 = x[i1]; 4844c1414c8SBarry Smith tmp1 = x[i2]; 4854c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 4864c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 4874c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 4884c1414c8SBarry Smith } 4894c1414c8SBarry Smith if (n == sz-1){ 4904c1414c8SBarry Smith tmp0 = x[*idx++]; 4914c1414c8SBarry Smith sum1 += *v1++ * tmp0; 4924c1414c8SBarry Smith sum2 += *v2++ * tmp0; 4934c1414c8SBarry Smith sum3 += *v3++ * tmp0; 4944c1414c8SBarry Smith } 4954c1414c8SBarry Smith y[row++]=sum1; 4964c1414c8SBarry Smith y[row++]=sum2; 4974c1414c8SBarry Smith y[row++]=sum3; 4984c1414c8SBarry Smith v1 =v3; /* Since the next block to be processed starts there*/ 4994c1414c8SBarry Smith idx +=2*sz; 5004c1414c8SBarry Smith break; 5014c1414c8SBarry Smith case 4: 50275567043SBarry Smith sum1 = 0.; 50375567043SBarry Smith sum2 = 0.; 50475567043SBarry Smith sum3 = 0.; 50575567043SBarry Smith sum4 = 0.; 5064c1414c8SBarry Smith v2 = v1 + n; 5074c1414c8SBarry Smith v3 = v2 + n; 5084c1414c8SBarry Smith v4 = v3 + n; 5094c1414c8SBarry Smith 5104c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 5114c1414c8SBarry Smith i1 = idx[0]; 5124c1414c8SBarry Smith i2 = idx[1]; 5134c1414c8SBarry Smith idx += 2; 5144c1414c8SBarry Smith tmp0 = x[i1]; 5154c1414c8SBarry Smith tmp1 = x[i2]; 5164c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 5174c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 5184c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 5194c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 5204c1414c8SBarry Smith } 5214c1414c8SBarry Smith if (n == sz-1){ 5224c1414c8SBarry Smith tmp0 = x[*idx++]; 5234c1414c8SBarry Smith sum1 += *v1++ * tmp0; 5244c1414c8SBarry Smith sum2 += *v2++ * tmp0; 5254c1414c8SBarry Smith sum3 += *v3++ * tmp0; 5264c1414c8SBarry Smith sum4 += *v4++ * tmp0; 5274c1414c8SBarry Smith } 5284c1414c8SBarry Smith y[row++]=sum1; 5294c1414c8SBarry Smith y[row++]=sum2; 5304c1414c8SBarry Smith y[row++]=sum3; 5314c1414c8SBarry Smith y[row++]=sum4; 5324c1414c8SBarry Smith v1 =v4; /* Since the next block to be processed starts there*/ 5334c1414c8SBarry Smith idx +=3*sz; 5344c1414c8SBarry Smith break; 5354c1414c8SBarry Smith case 5: 53675567043SBarry Smith sum1 = 0.; 53775567043SBarry Smith sum2 = 0.; 53875567043SBarry Smith sum3 = 0.; 53975567043SBarry Smith sum4 = 0.; 54075567043SBarry Smith sum5 = 0.; 5414c1414c8SBarry Smith v2 = v1 + n; 5424c1414c8SBarry Smith v3 = v2 + n; 5434c1414c8SBarry Smith v4 = v3 + n; 5444c1414c8SBarry Smith v5 = v4 + n; 5454c1414c8SBarry Smith 5464c1414c8SBarry Smith for (n = 0; n<sz-1; n+=2) { 5474c1414c8SBarry Smith i1 = idx[0]; 5484c1414c8SBarry Smith i2 = idx[1]; 5494c1414c8SBarry Smith idx += 2; 5504c1414c8SBarry Smith tmp0 = x[i1]; 5514c1414c8SBarry Smith tmp1 = x[i2]; 5524c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 5534c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 5544c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 5554c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 5564c1414c8SBarry Smith sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2; 5574c1414c8SBarry Smith } 5584c1414c8SBarry Smith if (n == sz-1){ 5594c1414c8SBarry Smith tmp0 = x[*idx++]; 5604c1414c8SBarry Smith sum1 += *v1++ * tmp0; 5614c1414c8SBarry Smith sum2 += *v2++ * tmp0; 5624c1414c8SBarry Smith sum3 += *v3++ * tmp0; 5634c1414c8SBarry Smith sum4 += *v4++ * tmp0; 5644c1414c8SBarry Smith sum5 += *v5++ * tmp0; 5654c1414c8SBarry Smith } 5664c1414c8SBarry Smith y[row++]=sum1; 5674c1414c8SBarry Smith y[row++]=sum2; 5684c1414c8SBarry Smith y[row++]=sum3; 5694c1414c8SBarry Smith y[row++]=sum4; 5704c1414c8SBarry Smith y[row++]=sum5; 5714c1414c8SBarry Smith v1 =v5; /* Since the next block to be processed starts there */ 5724c1414c8SBarry Smith idx +=4*sz; 5734c1414c8SBarry Smith break; 5744c1414c8SBarry Smith default : 575e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported"); 5764c1414c8SBarry Smith } 5774c1414c8SBarry Smith } 5783649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 5794c1414c8SBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 580dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr); 5814c1414c8SBarry Smith PetscFunctionReturn(0); 5824c1414c8SBarry Smith } 5834c1414c8SBarry Smith /* ----------------------------------------------------------- */ 5844108e4d5SBarry Smith /* Almost same code as the MatMult_SeqAIJ_Inode() */ 5854c1414c8SBarry Smith #undef __FUNCT__ 5864108e4d5SBarry Smith #define __FUNCT__ "MatMultAdd_SeqAIJ_Inode" 5874108e4d5SBarry Smith static PetscErrorCode MatMultAdd_SeqAIJ_Inode(Mat A,Vec xx,Vec zz,Vec yy) 5884c1414c8SBarry Smith { 5894c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 5904c1414c8SBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5,tmp0,tmp1; 591dd6ea824SBarry Smith MatScalar *v1,*v2,*v3,*v4,*v5; 592dd6ea824SBarry Smith PetscScalar *x,*y,*z,*zt; 5934c1414c8SBarry Smith PetscErrorCode ierr; 5944c1414c8SBarry Smith PetscInt *idx,i1,i2,n,i,row,node_max,*ns,*ii,nsz,sz; 5954c1414c8SBarry Smith 5964c1414c8SBarry Smith PetscFunctionBegin; 597e32f2f54SBarry Smith if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 5984c1414c8SBarry Smith node_max = a->inode.node_count; 5994c1414c8SBarry Smith ns = a->inode.size; /* Node Size array */ 6004c1414c8SBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 6014c1414c8SBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 6024c1414c8SBarry Smith if (zz != yy) { 6034c1414c8SBarry Smith ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 6044c1414c8SBarry Smith } else { 6054c1414c8SBarry Smith z = y; 6064c1414c8SBarry Smith } 6074c1414c8SBarry Smith zt = z; 6084c1414c8SBarry Smith 6094c1414c8SBarry Smith idx = a->j; 6104c1414c8SBarry Smith v1 = a->a; 6114c1414c8SBarry Smith ii = a->i; 6124c1414c8SBarry Smith 6134c1414c8SBarry Smith for (i = 0,row = 0; i< node_max; ++i){ 6144c1414c8SBarry Smith nsz = ns[i]; 6154c1414c8SBarry Smith n = ii[1] - ii[0]; 6164c1414c8SBarry Smith ii += nsz; 6174c1414c8SBarry Smith sz = n; /* No of non zeros in this row */ 6184c1414c8SBarry Smith /* Switch on the size of Node */ 6194c1414c8SBarry Smith switch (nsz){ /* Each loop in 'case' is unrolled */ 6204c1414c8SBarry Smith case 1 : 6214c1414c8SBarry Smith sum1 = *zt++; 6224c1414c8SBarry Smith 6234c1414c8SBarry Smith for(n = 0; n< sz-1; n+=2) { 6244c1414c8SBarry Smith i1 = idx[0]; /* The instructions are ordered to */ 6254c1414c8SBarry Smith i2 = idx[1]; /* make the compiler's job easy */ 6264c1414c8SBarry Smith idx += 2; 6274c1414c8SBarry Smith tmp0 = x[i1]; 6284c1414c8SBarry Smith tmp1 = x[i2]; 6294c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 6304c1414c8SBarry Smith } 6314c1414c8SBarry Smith 6324c1414c8SBarry Smith if(n == sz-1){ /* Take care of the last nonzero */ 6334c1414c8SBarry Smith tmp0 = x[*idx++]; 6344c1414c8SBarry Smith sum1 += *v1++ * tmp0; 6354c1414c8SBarry Smith } 6364c1414c8SBarry Smith y[row++]=sum1; 6374c1414c8SBarry Smith break; 6384c1414c8SBarry Smith case 2: 6394c1414c8SBarry Smith sum1 = *zt++; 6404c1414c8SBarry Smith sum2 = *zt++; 6414c1414c8SBarry Smith v2 = v1 + n; 6424c1414c8SBarry Smith 6434c1414c8SBarry Smith for(n = 0; n< sz-1; n+=2) { 6444c1414c8SBarry Smith i1 = idx[0]; 6454c1414c8SBarry Smith i2 = idx[1]; 6464c1414c8SBarry Smith idx += 2; 6474c1414c8SBarry Smith tmp0 = x[i1]; 6484c1414c8SBarry Smith tmp1 = x[i2]; 6494c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 6504c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 6514c1414c8SBarry Smith } 6524c1414c8SBarry Smith if(n == sz-1){ 6534c1414c8SBarry Smith tmp0 = x[*idx++]; 6544c1414c8SBarry Smith sum1 += *v1++ * tmp0; 6554c1414c8SBarry Smith sum2 += *v2++ * tmp0; 6564c1414c8SBarry Smith } 6574c1414c8SBarry Smith y[row++]=sum1; 6584c1414c8SBarry Smith y[row++]=sum2; 6594c1414c8SBarry Smith v1 =v2; /* Since the next block to be processed starts there*/ 6604c1414c8SBarry Smith idx +=sz; 6614c1414c8SBarry Smith break; 6624c1414c8SBarry Smith case 3: 6634c1414c8SBarry Smith sum1 = *zt++; 6644c1414c8SBarry Smith sum2 = *zt++; 6654c1414c8SBarry Smith sum3 = *zt++; 6664c1414c8SBarry Smith v2 = v1 + n; 6674c1414c8SBarry Smith v3 = v2 + n; 6684c1414c8SBarry Smith 6694c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 6704c1414c8SBarry Smith i1 = idx[0]; 6714c1414c8SBarry Smith i2 = idx[1]; 6724c1414c8SBarry Smith idx += 2; 6734c1414c8SBarry Smith tmp0 = x[i1]; 6744c1414c8SBarry Smith tmp1 = x[i2]; 6754c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 6764c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 6774c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 6784c1414c8SBarry Smith } 6794c1414c8SBarry Smith if (n == sz-1){ 6804c1414c8SBarry Smith tmp0 = x[*idx++]; 6814c1414c8SBarry Smith sum1 += *v1++ * tmp0; 6824c1414c8SBarry Smith sum2 += *v2++ * tmp0; 6834c1414c8SBarry Smith sum3 += *v3++ * tmp0; 6844c1414c8SBarry Smith } 6854c1414c8SBarry Smith y[row++]=sum1; 6864c1414c8SBarry Smith y[row++]=sum2; 6874c1414c8SBarry Smith y[row++]=sum3; 6884c1414c8SBarry Smith v1 =v3; /* Since the next block to be processed starts there*/ 6894c1414c8SBarry Smith idx +=2*sz; 6904c1414c8SBarry Smith break; 6914c1414c8SBarry Smith case 4: 6924c1414c8SBarry Smith sum1 = *zt++; 6934c1414c8SBarry Smith sum2 = *zt++; 6944c1414c8SBarry Smith sum3 = *zt++; 6954c1414c8SBarry Smith sum4 = *zt++; 6964c1414c8SBarry Smith v2 = v1 + n; 6974c1414c8SBarry Smith v3 = v2 + n; 6984c1414c8SBarry Smith v4 = v3 + n; 6994c1414c8SBarry Smith 7004c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 7014c1414c8SBarry Smith i1 = idx[0]; 7024c1414c8SBarry Smith i2 = idx[1]; 7034c1414c8SBarry Smith idx += 2; 7044c1414c8SBarry Smith tmp0 = x[i1]; 7054c1414c8SBarry Smith tmp1 = x[i2]; 7064c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 7074c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 7084c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 7094c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 7104c1414c8SBarry Smith } 7114c1414c8SBarry Smith if (n == sz-1){ 7124c1414c8SBarry Smith tmp0 = x[*idx++]; 7134c1414c8SBarry Smith sum1 += *v1++ * tmp0; 7144c1414c8SBarry Smith sum2 += *v2++ * tmp0; 7154c1414c8SBarry Smith sum3 += *v3++ * tmp0; 7164c1414c8SBarry Smith sum4 += *v4++ * tmp0; 7174c1414c8SBarry Smith } 7184c1414c8SBarry Smith y[row++]=sum1; 7194c1414c8SBarry Smith y[row++]=sum2; 7204c1414c8SBarry Smith y[row++]=sum3; 7214c1414c8SBarry Smith y[row++]=sum4; 7224c1414c8SBarry Smith v1 =v4; /* Since the next block to be processed starts there*/ 7234c1414c8SBarry Smith idx +=3*sz; 7244c1414c8SBarry Smith break; 7254c1414c8SBarry Smith case 5: 7264c1414c8SBarry Smith sum1 = *zt++; 7274c1414c8SBarry Smith sum2 = *zt++; 7284c1414c8SBarry Smith sum3 = *zt++; 7294c1414c8SBarry Smith sum4 = *zt++; 7304c1414c8SBarry Smith sum5 = *zt++; 7314c1414c8SBarry Smith v2 = v1 + n; 7324c1414c8SBarry Smith v3 = v2 + n; 7334c1414c8SBarry Smith v4 = v3 + n; 7344c1414c8SBarry Smith v5 = v4 + n; 7354c1414c8SBarry Smith 7364c1414c8SBarry Smith for (n = 0; n<sz-1; n+=2) { 7374c1414c8SBarry Smith i1 = idx[0]; 7384c1414c8SBarry Smith i2 = idx[1]; 7394c1414c8SBarry Smith idx += 2; 7404c1414c8SBarry Smith tmp0 = x[i1]; 7414c1414c8SBarry Smith tmp1 = x[i2]; 7424c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 7434c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 7444c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 7454c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 7464c1414c8SBarry Smith sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2; 7474c1414c8SBarry Smith } 7484c1414c8SBarry Smith if(n == sz-1){ 7494c1414c8SBarry Smith tmp0 = x[*idx++]; 7504c1414c8SBarry Smith sum1 += *v1++ * tmp0; 7514c1414c8SBarry Smith sum2 += *v2++ * tmp0; 7524c1414c8SBarry Smith sum3 += *v3++ * tmp0; 7534c1414c8SBarry Smith sum4 += *v4++ * tmp0; 7544c1414c8SBarry Smith sum5 += *v5++ * tmp0; 7554c1414c8SBarry Smith } 7564c1414c8SBarry Smith y[row++]=sum1; 7574c1414c8SBarry Smith y[row++]=sum2; 7584c1414c8SBarry Smith y[row++]=sum3; 7594c1414c8SBarry Smith y[row++]=sum4; 7604c1414c8SBarry Smith y[row++]=sum5; 7614c1414c8SBarry Smith v1 =v5; /* Since the next block to be processed starts there */ 7624c1414c8SBarry Smith idx +=4*sz; 7634c1414c8SBarry Smith break; 7644c1414c8SBarry Smith default : 765e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported"); 7664c1414c8SBarry Smith } 7674c1414c8SBarry Smith } 7684c1414c8SBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 7694c1414c8SBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 7704c1414c8SBarry Smith if (zz != yy) { 7714c1414c8SBarry Smith ierr = VecRestoreArray(zz,&z);CHKERRQ(ierr); 7724c1414c8SBarry Smith } 773dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 7744c1414c8SBarry Smith PetscFunctionReturn(0); 7754c1414c8SBarry Smith } 7764c1414c8SBarry Smith 7774c1414c8SBarry Smith /* ----------------------------------------------------------- */ 7784c1414c8SBarry Smith #undef __FUNCT__ 779ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_Inode_inplace" 780ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_Inode_inplace(Mat A,Vec bb,Vec xx) 7814c1414c8SBarry Smith { 7824c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 7834c1414c8SBarry Smith IS iscol = a->col,isrow = a->row; 7844c1414c8SBarry Smith PetscErrorCode ierr; 7855d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 7865d0c19d7SBarry Smith PetscInt i,j,n = A->rmap->n,*ai = a->i,nz,*a_j = a->j; 7875d0c19d7SBarry Smith PetscInt node_max,*ns,row,nsz,aii,*vi,*ad,*aj,i0,i1; 788d9fead3dSBarry Smith PetscScalar *x,*tmp,*tmps,tmp0,tmp1; 789d9fead3dSBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5; 790dd6ea824SBarry Smith const MatScalar *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa; 791dd6ea824SBarry Smith const PetscScalar *b; 7924c1414c8SBarry Smith 7934c1414c8SBarry Smith PetscFunctionBegin; 794e32f2f54SBarry Smith if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 7954c1414c8SBarry Smith node_max = a->inode.node_count; 7964c1414c8SBarry Smith ns = a->inode.size; /* Node Size array */ 7974c1414c8SBarry Smith 7983649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 7994c1414c8SBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 8004c1414c8SBarry Smith tmp = a->solve_work; 8014c1414c8SBarry Smith 8024c1414c8SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 8034c1414c8SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 8044c1414c8SBarry Smith 8054c1414c8SBarry Smith /* forward solve the lower triangular */ 8064c1414c8SBarry Smith tmps = tmp ; 8074c1414c8SBarry Smith aa = a_a ; 8084c1414c8SBarry Smith aj = a_j ; 8094c1414c8SBarry Smith ad = a->diag; 8104c1414c8SBarry Smith 8114c1414c8SBarry Smith for (i = 0,row = 0; i< node_max; ++i){ 8124c1414c8SBarry Smith nsz = ns[i]; 8134c1414c8SBarry Smith aii = ai[row]; 8144c1414c8SBarry Smith v1 = aa + aii; 8154c1414c8SBarry Smith vi = aj + aii; 8164c1414c8SBarry Smith nz = ad[row]- aii; 81726549573SJed Brown if (i < node_max-1) { 81826549573SJed Brown /* Prefetch the block after the current one, the prefetch itself can't cause a memory error, 81926549573SJed Brown * but our indexing to determine it's size could. */ 82050d8bf02SJed Brown PetscPrefetchBlock(aj+ai[row+nsz],ad[row+nsz]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); /* indices */ 82126549573SJed Brown /* In my tests, it seems to be better to fetch entire rows instead of just the lower-triangular part */ 82250d8bf02SJed Brown PetscPrefetchBlock(aa+ai[row+nsz],ad[row+nsz+ns[i+1]-1]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); 82326549573SJed 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); */ 82426549573SJed Brown } 8254c1414c8SBarry Smith 8264c1414c8SBarry Smith switch (nsz){ /* Each loop in 'case' is unrolled */ 8274c1414c8SBarry Smith case 1 : 8284c1414c8SBarry Smith sum1 = b[*r++]; 8294c1414c8SBarry Smith for(j=0; j<nz-1; j+=2){ 8304c1414c8SBarry Smith i0 = vi[0]; 8314c1414c8SBarry Smith i1 = vi[1]; 8324c1414c8SBarry Smith vi +=2; 8334c1414c8SBarry Smith tmp0 = tmps[i0]; 8344c1414c8SBarry Smith tmp1 = tmps[i1]; 8354c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 8364c1414c8SBarry Smith } 8374c1414c8SBarry Smith if(j == nz-1){ 8384c1414c8SBarry Smith tmp0 = tmps[*vi++]; 8394c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 8404c1414c8SBarry Smith } 8414c1414c8SBarry Smith tmp[row ++]=sum1; 8424c1414c8SBarry Smith break; 8434c1414c8SBarry Smith case 2: 8444c1414c8SBarry Smith sum1 = b[*r++]; 8454c1414c8SBarry Smith sum2 = b[*r++]; 8464c1414c8SBarry Smith v2 = aa + ai[row+1]; 8474c1414c8SBarry Smith 8484c1414c8SBarry Smith for(j=0; j<nz-1; j+=2){ 8494c1414c8SBarry Smith i0 = vi[0]; 8504c1414c8SBarry Smith i1 = vi[1]; 8514c1414c8SBarry Smith vi +=2; 8524c1414c8SBarry Smith tmp0 = tmps[i0]; 8534c1414c8SBarry Smith tmp1 = tmps[i1]; 8544c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 8554c1414c8SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 8564c1414c8SBarry Smith } 8574c1414c8SBarry Smith if(j == nz-1){ 8584c1414c8SBarry Smith tmp0 = tmps[*vi++]; 8594c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 8604c1414c8SBarry Smith sum2 -= *v2++ *tmp0; 8614c1414c8SBarry Smith } 8624c1414c8SBarry Smith sum2 -= *v2++ * sum1; 8634c1414c8SBarry Smith tmp[row ++]=sum1; 8644c1414c8SBarry Smith tmp[row ++]=sum2; 8654c1414c8SBarry Smith break; 8664c1414c8SBarry Smith case 3: 8674c1414c8SBarry Smith sum1 = b[*r++]; 8684c1414c8SBarry Smith sum2 = b[*r++]; 8694c1414c8SBarry Smith sum3 = b[*r++]; 8704c1414c8SBarry Smith v2 = aa + ai[row+1]; 8714c1414c8SBarry Smith v3 = aa + ai[row+2]; 8724c1414c8SBarry Smith 8734c1414c8SBarry Smith for (j=0; j<nz-1; j+=2){ 8744c1414c8SBarry Smith i0 = vi[0]; 8754c1414c8SBarry Smith i1 = vi[1]; 8764c1414c8SBarry Smith vi +=2; 8774c1414c8SBarry Smith tmp0 = tmps[i0]; 8784c1414c8SBarry Smith tmp1 = tmps[i1]; 8794c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 8804c1414c8SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 8814c1414c8SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 8824c1414c8SBarry Smith } 8834c1414c8SBarry Smith if (j == nz-1){ 8844c1414c8SBarry Smith tmp0 = tmps[*vi++]; 8854c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 8864c1414c8SBarry Smith sum2 -= *v2++ *tmp0; 8874c1414c8SBarry Smith sum3 -= *v3++ *tmp0; 8884c1414c8SBarry Smith } 8894c1414c8SBarry Smith sum2 -= *v2++ * sum1; 8904c1414c8SBarry Smith sum3 -= *v3++ * sum1; 8914c1414c8SBarry Smith sum3 -= *v3++ * sum2; 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]; 121528f1b45aSHong Zhang for (j=0; j<nz; j++) 121628f1b45aSHong Zhang rs += PetscAbsScalar(v[j]); 121728f1b45aSHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 121828f1b45aSHong Zhang } 121928f1b45aSHong Zhang sctx.shift_top *= 1.1; 122028f1b45aSHong Zhang sctx.nshift_max = 5; 122128f1b45aSHong Zhang sctx.shift_lo = 0.; 122228f1b45aSHong Zhang sctx.shift_hi = 1.; 122328f1b45aSHong Zhang } 122428f1b45aSHong Zhang 122528f1b45aSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 122628f1b45aSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 122768785679SHong Zhang 12289877982aSShri Abhyankar ierr = PetscMalloc((4*n+1)*sizeof(PetscScalar),&rtmp1);CHKERRQ(ierr); 12299877982aSShri Abhyankar ierr = PetscMemzero(rtmp1,(4*n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 1230b89f182dSHong Zhang rtmp2 = rtmp1 + n; 1231b89f182dSHong Zhang rtmp3 = rtmp2 + n; 12329877982aSShri Abhyankar rtmp4 = rtmp3 + n; 123328f1b45aSHong Zhang ics = ic; 123428f1b45aSHong Zhang 123528f1b45aSHong Zhang node_max = a->inode.node_count; 123628f1b45aSHong Zhang ns = a->inode.size; 1237e7e72b3dSBarry Smith if (!ns) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix without inode information"); 123828f1b45aSHong Zhang 12399877982aSShri Abhyankar /* If max inode size > 4, split it into two inodes.*/ 124068785679SHong Zhang /* also map the inode sizes according to the ordering */ 124168785679SHong Zhang ierr = PetscMalloc((n+1)* sizeof(PetscInt),&tmp_vec1);CHKERRQ(ierr); 124268785679SHong Zhang for (i=0,j=0; i<node_max; ++i,++j){ 1243b1550197SShri Abhyankar if (ns[i] > 4) { 1244048b5e81SShri Abhyankar tmp_vec1[j] = 4; 124568785679SHong Zhang ++j; 124668785679SHong Zhang tmp_vec1[j] = ns[i] - tmp_vec1[j-1]; 124768785679SHong Zhang } else { 124868785679SHong Zhang tmp_vec1[j] = ns[i]; 124968785679SHong Zhang } 125068785679SHong Zhang } 125168785679SHong Zhang /* Use the correct node_max */ 125268785679SHong Zhang node_max = j; 125368785679SHong Zhang 125468785679SHong Zhang /* Now reorder the inode info based on mat re-ordering info */ 125568785679SHong Zhang /* First create a row -> inode_size_array_index map */ 125668785679SHong Zhang ierr = PetscMalloc(n*sizeof(PetscInt)+1,&nsmap);CHKERRQ(ierr); 125768785679SHong Zhang ierr = PetscMalloc(node_max*sizeof(PetscInt)+1,&tmp_vec2);CHKERRQ(ierr); 125868785679SHong Zhang for (i=0,row=0; i<node_max; i++) { 125968785679SHong Zhang nodesz = tmp_vec1[i]; 126068785679SHong Zhang for (j=0; j<nodesz; j++,row++) { 126168785679SHong Zhang nsmap[row] = i; 126268785679SHong Zhang } 126368785679SHong Zhang } 126468785679SHong Zhang /* Using nsmap, create a reordered ns structure */ 126568785679SHong Zhang for (i=0,j=0; i< node_max; i++) { 126668785679SHong Zhang nodesz = tmp_vec1[nsmap[r[j]]]; /* here the reordered row_no is in r[] */ 126768785679SHong Zhang tmp_vec2[i] = nodesz; 126868785679SHong Zhang j += nodesz; 126968785679SHong Zhang } 127068785679SHong Zhang ierr = PetscFree(nsmap);CHKERRQ(ierr); 127168785679SHong Zhang ierr = PetscFree(tmp_vec1);CHKERRQ(ierr); 1272b89f182dSHong Zhang 127368785679SHong Zhang /* Now use the correct ns */ 127468785679SHong Zhang ns = tmp_vec2; 127568785679SHong Zhang 127628f1b45aSHong Zhang do { 127707b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 127828f1b45aSHong Zhang /* Now loop over each block-row, and do the factorization */ 127928f1b45aSHong Zhang for (inod=0,i=0; inod<node_max; inod++){ /* i: row index; inod: inode index */ 128028f1b45aSHong Zhang nodesz = ns[inod]; 128128f1b45aSHong Zhang 128228f1b45aSHong Zhang switch (nodesz){ 128328f1b45aSHong Zhang case 1: 128468785679SHong Zhang /*----------*/ 1285b89f182dSHong Zhang /* zero rtmp1 */ 128628f1b45aSHong Zhang /* L part */ 128728f1b45aSHong Zhang nz = bi[i+1] - bi[i]; 128828f1b45aSHong Zhang bjtmp = bj + bi[i]; 1289b89f182dSHong Zhang for (j=0; j<nz; j++) rtmp1[bjtmp[j]] = 0.0; 129028f1b45aSHong Zhang 129128f1b45aSHong Zhang /* U part */ 129228f1b45aSHong Zhang nz = bdiag[i]-bdiag[i+1]; 129328f1b45aSHong Zhang bjtmp = bj + bdiag[i+1]+1; 1294b89f182dSHong Zhang for (j=0; j<nz; j++) rtmp1[bjtmp[j]] = 0.0; 129528f1b45aSHong Zhang 129628f1b45aSHong Zhang /* load in initial (unfactored row) */ 129728f1b45aSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 129828f1b45aSHong Zhang ajtmp = aj + ai[r[i]]; 129928f1b45aSHong Zhang v = aa + ai[r[i]]; 130028f1b45aSHong Zhang for (j=0; j<nz; j++) { 1301b89f182dSHong Zhang rtmp1[ics[ajtmp[j]]] = v[j]; 130228f1b45aSHong Zhang } 130328f1b45aSHong Zhang /* ZeropivotApply() */ 1304b89f182dSHong Zhang rtmp1[i] += sctx.shift_amount; /* shift the diagonal of the matrix */ 130528f1b45aSHong Zhang 130628f1b45aSHong Zhang /* elimination */ 130728f1b45aSHong Zhang bjtmp = bj + bi[i]; 130828f1b45aSHong Zhang row = *bjtmp++; 130928f1b45aSHong Zhang nzL = bi[i+1] - bi[i]; 131028f1b45aSHong Zhang for(k=0; k < nzL;k++) { 1311b89f182dSHong Zhang pc = rtmp1 + row; 131228f1b45aSHong Zhang if (*pc != 0.0) { 131328f1b45aSHong Zhang pv = b->a + bdiag[row]; 1314b89f182dSHong Zhang mul1 = *pc * (*pv); 1315b89f182dSHong Zhang *pc = mul1; 131628f1b45aSHong Zhang pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 131728f1b45aSHong Zhang pv = b->a + bdiag[row+1]+1; 131828f1b45aSHong Zhang nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 1319b89f182dSHong Zhang for (j=0; j<nz; j++) rtmp1[pj[j]] -= mul1 * pv[j]; 132028f1b45aSHong Zhang ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); 132128f1b45aSHong Zhang } 132228f1b45aSHong Zhang row = *bjtmp++; 132328f1b45aSHong Zhang } 132428f1b45aSHong Zhang 132528f1b45aSHong Zhang /* finished row so stick it into b->a */ 132628f1b45aSHong Zhang rs = 0.0; 132728f1b45aSHong Zhang /* L part */ 132828f1b45aSHong Zhang pv = b->a + bi[i] ; 132928f1b45aSHong Zhang pj = b->j + bi[i] ; 133028f1b45aSHong Zhang nz = bi[i+1] - bi[i]; 133128f1b45aSHong Zhang for (j=0; j<nz; j++) { 1332b89f182dSHong Zhang pv[j] = rtmp1[pj[j]]; rs += PetscAbsScalar(pv[j]); 133328f1b45aSHong Zhang } 133428f1b45aSHong Zhang 133528f1b45aSHong Zhang /* U part */ 133628f1b45aSHong Zhang pv = b->a + bdiag[i+1]+1; 133728f1b45aSHong Zhang pj = b->j + bdiag[i+1]+1; 133828f1b45aSHong Zhang nz = bdiag[i] - bdiag[i+1]-1; 133928f1b45aSHong Zhang for (j=0; j<nz; j++) { 1340b89f182dSHong Zhang pv[j] = rtmp1[pj[j]]; rs += PetscAbsScalar(pv[j]); 134128f1b45aSHong Zhang } 134228f1b45aSHong Zhang 1343b89f182dSHong Zhang /* Check zero pivot */ 134428f1b45aSHong Zhang sctx.rs = rs; 1345b89f182dSHong Zhang sctx.pv = rtmp1[i]; 1346d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 134707b50cabSHong Zhang if(sctx.newshift) break; 134828f1b45aSHong Zhang 134928f1b45aSHong Zhang /* Mark diagonal and invert diagonal for simplier triangular solves */ 135028f1b45aSHong Zhang pv = b->a + bdiag[i]; 1351b89f182dSHong Zhang *pv = 1.0/sctx.pv; /* sctx.pv = rtmp1[i]+shiftamount if shifttype==MAT_SHIFT_INBLOCKS */ 135228f1b45aSHong Zhang break; 135328f1b45aSHong Zhang 135428f1b45aSHong Zhang case 2: 135528f1b45aSHong Zhang /*----------*/ 1356b89f182dSHong Zhang /* zero rtmp1 and rtmp2 */ 135728f1b45aSHong Zhang /* L part */ 135828f1b45aSHong Zhang nz = bi[i+1] - bi[i]; 135928f1b45aSHong Zhang bjtmp = bj + bi[i]; 136028f1b45aSHong Zhang for (j=0; j<nz; j++) { 136168785679SHong Zhang col = bjtmp[j]; 1362b89f182dSHong Zhang rtmp1[col] = 0.0; rtmp2[col] = 0.0; 136328f1b45aSHong Zhang } 136428f1b45aSHong Zhang 136528f1b45aSHong Zhang /* U part */ 136628f1b45aSHong Zhang nz = bdiag[i]-bdiag[i+1]; 136728f1b45aSHong Zhang bjtmp = bj + bdiag[i+1]+1; 136828f1b45aSHong Zhang for (j=0; j<nz; j++) { 136968785679SHong Zhang col = bjtmp[j]; 1370b89f182dSHong Zhang rtmp1[col] = 0.0; rtmp2[col] = 0.0; 137128f1b45aSHong Zhang } 137228f1b45aSHong Zhang 137328f1b45aSHong Zhang /* load in initial (unfactored row) */ 137428f1b45aSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 137528f1b45aSHong Zhang ajtmp = aj + ai[r[i]]; 137628f1b45aSHong Zhang v1 = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; 137728f1b45aSHong Zhang for (j=0; j<nz; j++) { 137868785679SHong Zhang col = ics[ajtmp[j]]; 1379b89f182dSHong Zhang rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; 138028f1b45aSHong Zhang } 138128f1b45aSHong Zhang /* ZeropivotApply(): shift the diagonal of the matrix */ 1382b89f182dSHong Zhang rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; 138328f1b45aSHong Zhang 138428f1b45aSHong Zhang /* elimination */ 138528f1b45aSHong Zhang bjtmp = bj + bi[i]; 138628f1b45aSHong Zhang row = *bjtmp++; /* pivot row */ 138728f1b45aSHong Zhang nzL = bi[i+1] - bi[i]; 138828f1b45aSHong Zhang for(k=0; k < nzL;k++) { 1389b89f182dSHong Zhang pc1 = rtmp1 + row; 1390b89f182dSHong Zhang pc2 = rtmp2 + row; 139128f1b45aSHong Zhang if (*pc1 != 0.0 || *pc2 != 0.0) { 139228f1b45aSHong Zhang pv = b->a + bdiag[row]; 1393b89f182dSHong Zhang mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); 1394b89f182dSHong Zhang *pc1 = mul1; *pc2 = mul2; 139528f1b45aSHong Zhang 139628f1b45aSHong Zhang pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 139728f1b45aSHong Zhang pv = b->a + bdiag[row+1]+1; 139828f1b45aSHong Zhang nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 139928f1b45aSHong Zhang for (j=0; j<nz; j++){ 140068785679SHong Zhang col = pj[j]; 1401b89f182dSHong Zhang rtmp1[col] -= mul1 * pv[j]; 1402b89f182dSHong Zhang rtmp2[col] -= mul2 * pv[j]; 140328f1b45aSHong Zhang } 140428f1b45aSHong Zhang ierr = PetscLogFlops(4*nz);CHKERRQ(ierr); 140528f1b45aSHong Zhang } 140628f1b45aSHong Zhang row = *bjtmp++; 140728f1b45aSHong Zhang } 140828f1b45aSHong Zhang 1409b89f182dSHong Zhang /* finished row i; check zero pivot, then stick row i into b->a */ 141028f1b45aSHong Zhang rs = 0.0; 141128f1b45aSHong Zhang /* L part */ 1412b89f182dSHong Zhang pc1 = b->a + bi[i]; 141328f1b45aSHong Zhang pj = b->j + bi[i] ; 141428f1b45aSHong Zhang nz = bi[i+1] - bi[i]; 141528f1b45aSHong Zhang for (j=0; j<nz; j++) { 141668785679SHong Zhang col = pj[j]; 14170e7a5c2bSHong Zhang pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 141828f1b45aSHong Zhang } 141928f1b45aSHong Zhang /* U part */ 1420b89f182dSHong Zhang pc1 = b->a + bdiag[i+1]+1; 142128f1b45aSHong Zhang pj = b->j + bdiag[i+1]+1; 14220e7a5c2bSHong Zhang nz = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */ 142328f1b45aSHong Zhang for (j=0; j<nz; j++) { 142468785679SHong Zhang col = pj[j]; 14250e7a5c2bSHong Zhang pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 142628f1b45aSHong Zhang } 142728f1b45aSHong Zhang 142828f1b45aSHong Zhang sctx.rs = rs; 1429b89f182dSHong Zhang sctx.pv = rtmp1[i]; 1430d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 143107b50cabSHong Zhang if(sctx.newshift) break; 1432b89f182dSHong Zhang pc1 = b->a + bdiag[i]; /* Mark diagonal */ 1433b89f182dSHong Zhang *pc1 = 1.0/sctx.pv; 1434b89f182dSHong Zhang 1435b89f182dSHong Zhang /* Now take care of diagonal 2x2 block. */ 1436b89f182dSHong Zhang pc2 = rtmp2 + i; 1437b89f182dSHong Zhang if (*pc2 != 0.0){ 1438b89f182dSHong Zhang mul1 = (*pc2)*(*pc1); /* *pc1=diag[i] is inverted! */ 1439b89f182dSHong Zhang *pc2 = mul1; /* insert L entry */ 1440b89f182dSHong Zhang pj = b->j + bdiag[i+1]+1; /* beginning of U(i,:) */ 1441b89f182dSHong Zhang nz = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */ 1442b89f182dSHong Zhang for (j=0; j<nz; j++) { 1443b89f182dSHong Zhang col = pj[j]; rtmp2[col] -= mul1 * rtmp1[col]; 144428f1b45aSHong Zhang } 1445b89f182dSHong Zhang ierr = PetscLogFlops(2*nz);CHKERRQ(ierr); 1446b89f182dSHong Zhang } 1447b89f182dSHong Zhang 1448b89f182dSHong Zhang /* finished row i+1; check zero pivot, then stick row i+1 into b->a */ 1449b89f182dSHong Zhang rs = 0.0; 1450b89f182dSHong Zhang /* L part */ 1451b89f182dSHong Zhang pc2 = b->a + bi[i+1]; 1452b89f182dSHong Zhang pj = b->j + bi[i+1] ; 1453b89f182dSHong Zhang nz = bi[i+2] - bi[i+1]; 1454b89f182dSHong Zhang for (j=0; j<nz; j++) { 1455b89f182dSHong Zhang col = pj[j]; 14560e7a5c2bSHong Zhang pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 1457b89f182dSHong Zhang } 1458b89f182dSHong Zhang /* U part */ 1459b89f182dSHong Zhang pc2 = b->a + bdiag[i+2]+1; 14600e7a5c2bSHong Zhang pj = b->j + bdiag[i+2]+1; 14610e7a5c2bSHong Zhang nz = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */ 1462b89f182dSHong Zhang for (j=0; j<nz; j++) { 1463b89f182dSHong Zhang col = pj[j]; 14640e7a5c2bSHong Zhang pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 1465b89f182dSHong Zhang } 1466b89f182dSHong Zhang 146728f1b45aSHong Zhang sctx.rs = rs; 1468b89f182dSHong Zhang sctx.pv = rtmp2[i+1]; 1469d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i+1);CHKERRQ(ierr); 147007b50cabSHong Zhang if(sctx.newshift) break; 147128f1b45aSHong Zhang pc2 = b->a + bdiag[i+1]; 1472b89f182dSHong Zhang *pc2 = 1.0/sctx.pv; 147328f1b45aSHong Zhang break; 1474b89f182dSHong Zhang 147568785679SHong Zhang case 3: 147668785679SHong Zhang /*----------*/ 147768785679SHong Zhang /* zero rtmp */ 147868785679SHong Zhang /* L part */ 147968785679SHong Zhang nz = bi[i+1] - bi[i]; 148068785679SHong Zhang bjtmp = bj + bi[i]; 148168785679SHong Zhang for (j=0; j<nz; j++) { 148268785679SHong Zhang col = bjtmp[j]; 1483b89f182dSHong Zhang rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0; 148468785679SHong Zhang } 148568785679SHong Zhang 148668785679SHong Zhang /* U part */ 148768785679SHong Zhang nz = bdiag[i]-bdiag[i+1]; 148868785679SHong Zhang bjtmp = bj + bdiag[i+1]+1; 148968785679SHong Zhang for (j=0; j<nz; j++) { 149068785679SHong Zhang col = bjtmp[j]; 1491b89f182dSHong Zhang rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0; 149268785679SHong Zhang } 149368785679SHong Zhang 149468785679SHong Zhang /* load in initial (unfactored row) */ 149568785679SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 149668785679SHong Zhang ajtmp = aj + ai[r[i]]; 149768785679SHong Zhang v1 = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; v3 = aa + ai[r[i]+2]; 149868785679SHong Zhang for (j=0; j<nz; j++) { 149968785679SHong Zhang col = ics[ajtmp[j]]; 1500b89f182dSHong Zhang rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; rtmp3[col] = v3[j]; 150168785679SHong Zhang } 150268785679SHong Zhang /* ZeropivotApply(): shift the diagonal of the matrix */ 1503b89f182dSHong Zhang rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; rtmp3[i+2] += sctx.shift_amount; 150468785679SHong Zhang 150568785679SHong Zhang /* elimination */ 150668785679SHong Zhang bjtmp = bj + bi[i]; 150768785679SHong Zhang row = *bjtmp++; /* pivot row */ 150868785679SHong Zhang nzL = bi[i+1] - bi[i]; 150968785679SHong Zhang for(k=0; k < nzL;k++) { 1510b89f182dSHong Zhang pc1 = rtmp1 + row; 1511b89f182dSHong Zhang pc2 = rtmp2 + row; 1512b89f182dSHong Zhang pc3 = rtmp3 + row; 151368785679SHong Zhang if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 != 0.0) { 151468785679SHong Zhang pv = b->a + bdiag[row]; 1515b89f182dSHong Zhang mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); mul3 = *pc3*(*pv); 1516b89f182dSHong Zhang *pc1 = mul1; *pc2 = mul2; *pc3 = mul3; 151768785679SHong Zhang 151868785679SHong Zhang pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 151968785679SHong Zhang pv = b->a + bdiag[row+1]+1; 152068785679SHong Zhang nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 152168785679SHong Zhang for (j=0; j<nz; j++){ 152268785679SHong Zhang col = pj[j]; 1523b89f182dSHong Zhang rtmp1[col] -= mul1 * pv[j]; 1524b89f182dSHong Zhang rtmp2[col] -= mul2 * pv[j]; 1525b89f182dSHong Zhang rtmp3[col] -= mul3 * pv[j]; 152668785679SHong Zhang } 152768785679SHong Zhang ierr = PetscLogFlops(6*nz);CHKERRQ(ierr); 152868785679SHong Zhang } 152968785679SHong Zhang row = *bjtmp++; 153068785679SHong Zhang } 153168785679SHong Zhang 1532b89f182dSHong Zhang /* finished row i; check zero pivot, then stick row i into b->a */ 1533b89f182dSHong Zhang rs = 0.0; 1534b89f182dSHong Zhang /* L part */ 1535b89f182dSHong Zhang pc1 = b->a + bi[i]; 1536b89f182dSHong Zhang pj = b->j + bi[i] ; 1537b89f182dSHong Zhang nz = bi[i+1] - bi[i]; 1538b89f182dSHong Zhang for (j=0; j<nz; j++) { 1539b89f182dSHong Zhang col = pj[j]; 15400e7a5c2bSHong Zhang pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 1541b89f182dSHong Zhang } 1542b89f182dSHong Zhang /* U part */ 1543b89f182dSHong Zhang pc1 = b->a + bdiag[i+1]+1; 1544b89f182dSHong Zhang pj = b->j + bdiag[i+1]+1; 15450e7a5c2bSHong Zhang nz = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */ 1546b89f182dSHong Zhang for (j=0; j<nz; j++) { 1547b89f182dSHong Zhang col = pj[j]; 15480e7a5c2bSHong Zhang pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 1549b89f182dSHong Zhang } 155068785679SHong Zhang 1551b89f182dSHong Zhang sctx.rs = rs; 1552b89f182dSHong Zhang sctx.pv = rtmp1[i]; 1553d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 155407b50cabSHong Zhang if(sctx.newshift) break; 1555b89f182dSHong Zhang pc1 = b->a + bdiag[i]; /* Mark diag[i] */ 1556b89f182dSHong Zhang *pc1 = 1.0/sctx.pv; 1557b89f182dSHong Zhang 1558b89f182dSHong Zhang /* Now take care of 1st column of diagonal 3x3 block. */ 1559b89f182dSHong Zhang pc2 = rtmp2 + i; 1560b89f182dSHong Zhang pc3 = rtmp3 + i; 1561b89f182dSHong Zhang if (*pc2 != 0.0 || *pc3 != 0.0){ 1562b89f182dSHong Zhang mul2 = (*pc2)*(*pc1); *pc2 = mul2; 1563b89f182dSHong Zhang mul3 = (*pc3)*(*pc1); *pc3 = mul3; 156468785679SHong Zhang pj = b->j + bdiag[i+1]+1; /* beginning of U(i,:) */ 156568785679SHong Zhang nz = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */ 156668785679SHong Zhang for (j=0; j<nz; j++) { 156768785679SHong Zhang col = pj[j]; 1568b89f182dSHong Zhang rtmp2[col] -= mul2 * rtmp1[col]; 1569b89f182dSHong Zhang rtmp3[col] -= mul3 * rtmp1[col]; 157068785679SHong Zhang } 157168785679SHong Zhang ierr = PetscLogFlops(4*nz);CHKERRQ(ierr); 157268785679SHong Zhang } 157368785679SHong Zhang 1574b89f182dSHong Zhang /* finished row i+1; check zero pivot, then stick row i+1 into b->a */ 1575b89f182dSHong Zhang rs = 0.0; 1576b89f182dSHong Zhang /* L part */ 1577b89f182dSHong Zhang pc2 = b->a + bi[i+1]; 1578b89f182dSHong Zhang pj = b->j + bi[i+1] ; 1579b89f182dSHong Zhang nz = bi[i+2] - bi[i+1]; 1580b89f182dSHong Zhang for (j=0; j<nz; j++) { 1581b89f182dSHong Zhang col = pj[j]; 15820e7a5c2bSHong Zhang pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 1583b89f182dSHong Zhang } 1584b89f182dSHong Zhang /* U part */ 1585b89f182dSHong Zhang pc2 = b->a + bdiag[i+2]+1; 15860e7a5c2bSHong Zhang pj = b->j + bdiag[i+2]+1; 15870e7a5c2bSHong Zhang nz = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */ 1588b89f182dSHong Zhang for (j=0; j<nz; j++) { 1589b89f182dSHong Zhang col = pj[j]; 15900e7a5c2bSHong Zhang pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 1591b89f182dSHong Zhang } 1592b89f182dSHong Zhang 1593b89f182dSHong Zhang sctx.rs = rs; 1594b89f182dSHong Zhang sctx.pv = rtmp2[i+1]; 1595d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i+1);CHKERRQ(ierr); 159607b50cabSHong Zhang if(sctx.newshift) break; 1597b89f182dSHong Zhang pc2 = b->a + bdiag[i+1]; 1598b89f182dSHong Zhang *pc2 = 1.0/sctx.pv; /* Mark diag[i+1] */ 1599b89f182dSHong Zhang 1600b89f182dSHong Zhang /* Now take care of 2nd column of diagonal 3x3 block. */ 1601b89f182dSHong Zhang pc3 = rtmp3 + i+1; 160268785679SHong Zhang if (*pc3 != 0.0){ 1603b89f182dSHong Zhang mul3 = (*pc3)*(*pc2); *pc3 = mul3; 160468785679SHong Zhang pj = b->j + bdiag[i+2]+1; /* beginning of U(i+1,:) */ 160568785679SHong Zhang nz = bdiag[i+1]-bdiag[i+2]-1; /* num of entries in U(i+1,:) excluding diag */ 160668785679SHong Zhang for (j=0; j<nz; j++) { 160768785679SHong Zhang col = pj[j]; 1608b89f182dSHong Zhang rtmp3[col] -= mul3 * rtmp2[col]; 160968785679SHong Zhang } 161068785679SHong Zhang ierr = PetscLogFlops(2*nz);CHKERRQ(ierr); 161168785679SHong Zhang } 161268785679SHong Zhang 1613b89f182dSHong Zhang /* finished i+2; check zero pivot, then stick row i+2 into b->a */ 161468785679SHong Zhang rs = 0.0; 161568785679SHong Zhang /* L part */ 1616b89f182dSHong Zhang pc3 = b->a + bi[i+2]; 1617b89f182dSHong Zhang pj = b->j + bi[i+2] ; 1618b89f182dSHong Zhang nz = bi[i+3] - bi[i+2]; 161968785679SHong Zhang for (j=0; j<nz; j++) { 162068785679SHong Zhang col = pj[j]; 16210e7a5c2bSHong Zhang pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]); 162268785679SHong Zhang } 162368785679SHong Zhang /* U part */ 1624b89f182dSHong Zhang pc3 = b->a + bdiag[i+3]+1; 16250e7a5c2bSHong Zhang pj = b->j + bdiag[i+3]+1; 16260e7a5c2bSHong Zhang nz = bdiag[i+2] - bdiag[i+3] - 1; /* exclude diagonal */ 162768785679SHong Zhang for (j=0; j<nz; j++) { 162868785679SHong Zhang col = pj[j]; 16290e7a5c2bSHong Zhang pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]); 163068785679SHong Zhang } 163168785679SHong Zhang 163268785679SHong Zhang sctx.rs = rs; 1633b89f182dSHong Zhang sctx.pv = rtmp3[i+2]; 1634d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i+2);CHKERRQ(ierr); 163507b50cabSHong Zhang if(sctx.newshift) break; 163668785679SHong Zhang pc3 = b->a + bdiag[i+2]; 1637b89f182dSHong Zhang *pc3 = 1.0/sctx.pv; /* Mark diag[i+2] */ 163868785679SHong Zhang break; 16399877982aSShri Abhyankar case 4: 16409877982aSShri Abhyankar /*----------*/ 16419877982aSShri Abhyankar /* zero rtmp */ 16429877982aSShri Abhyankar /* L part */ 16439877982aSShri Abhyankar nz = bi[i+1] - bi[i]; 16449877982aSShri Abhyankar bjtmp = bj + bi[i]; 16459877982aSShri Abhyankar for (j=0; j<nz; j++) { 16469877982aSShri Abhyankar col = bjtmp[j]; 16479877982aSShri Abhyankar rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;rtmp4[col] = 0.0; 16489877982aSShri Abhyankar } 16499877982aSShri Abhyankar 16509877982aSShri Abhyankar /* U part */ 16519877982aSShri Abhyankar nz = bdiag[i]-bdiag[i+1]; 16529877982aSShri Abhyankar bjtmp = bj + bdiag[i+1]+1; 16539877982aSShri Abhyankar for (j=0; j<nz; j++) { 16549877982aSShri Abhyankar col = bjtmp[j]; 16559877982aSShri Abhyankar rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0; rtmp4[col] = 0.0; 16569877982aSShri Abhyankar } 16579877982aSShri Abhyankar 16589877982aSShri Abhyankar /* load in initial (unfactored row) */ 16599877982aSShri Abhyankar nz = ai[r[i]+1] - ai[r[i]]; 16609877982aSShri Abhyankar ajtmp = aj + ai[r[i]]; 16619877982aSShri Abhyankar v1 = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; v3 = aa + ai[r[i]+2]; v4 = aa + ai[r[i]+3]; 16629877982aSShri Abhyankar for (j=0; j<nz; j++) { 16639877982aSShri Abhyankar col = ics[ajtmp[j]]; 16649877982aSShri Abhyankar rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; rtmp3[col] = v3[j]; rtmp4[col] = v4[j]; 16659877982aSShri Abhyankar } 16669877982aSShri Abhyankar /* ZeropivotApply(): shift the diagonal of the matrix */ 16679877982aSShri Abhyankar rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; rtmp3[i+2] += sctx.shift_amount; rtmp4[i+3] += sctx.shift_amount; 16689877982aSShri Abhyankar 16699877982aSShri Abhyankar /* elimination */ 16709877982aSShri Abhyankar bjtmp = bj + bi[i]; 16719877982aSShri Abhyankar row = *bjtmp++; /* pivot row */ 16729877982aSShri Abhyankar nzL = bi[i+1] - bi[i]; 16739877982aSShri Abhyankar for(k=0; k < nzL;k++) { 16749877982aSShri Abhyankar pc1 = rtmp1 + row; 16759877982aSShri Abhyankar pc2 = rtmp2 + row; 16769877982aSShri Abhyankar pc3 = rtmp3 + row; 16779877982aSShri Abhyankar pc4 = rtmp4 + row; 16789877982aSShri Abhyankar if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 != 0.0 || *pc4 != 0.0) { 16799877982aSShri Abhyankar pv = b->a + bdiag[row]; 16809877982aSShri Abhyankar mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); mul3 = *pc3*(*pv); mul4 = *pc4*(*pv); 16819877982aSShri Abhyankar *pc1 = mul1; *pc2 = mul2; *pc3 = mul3; *pc4 = mul4; 16829877982aSShri Abhyankar 16839877982aSShri Abhyankar pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 16849877982aSShri Abhyankar pv = b->a + bdiag[row+1]+1; 16859877982aSShri Abhyankar nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 16869877982aSShri Abhyankar for (j=0; j<nz; j++){ 16879877982aSShri Abhyankar col = pj[j]; 16889877982aSShri Abhyankar rtmp1[col] -= mul1 * pv[j]; 16899877982aSShri Abhyankar rtmp2[col] -= mul2 * pv[j]; 16909877982aSShri Abhyankar rtmp3[col] -= mul3 * pv[j]; 16919877982aSShri Abhyankar rtmp4[col] -= mul4 * pv[j]; 16929877982aSShri Abhyankar } 16939877982aSShri Abhyankar ierr = PetscLogFlops(8*nz);CHKERRQ(ierr); 16949877982aSShri Abhyankar } 16959877982aSShri Abhyankar row = *bjtmp++; 16969877982aSShri Abhyankar } 16979877982aSShri Abhyankar 16989877982aSShri Abhyankar /* finished row i; check zero pivot, then stick row i into b->a */ 16999877982aSShri Abhyankar rs = 0.0; 17009877982aSShri Abhyankar /* L part */ 17019877982aSShri Abhyankar pc1 = b->a + bi[i]; 17029877982aSShri Abhyankar pj = b->j + bi[i] ; 17039877982aSShri Abhyankar nz = bi[i+1] - bi[i]; 17049877982aSShri Abhyankar for (j=0; j<nz; j++) { 17059877982aSShri Abhyankar col = pj[j]; 17069877982aSShri Abhyankar pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 17079877982aSShri Abhyankar } 17089877982aSShri Abhyankar /* U part */ 17099877982aSShri Abhyankar pc1 = b->a + bdiag[i+1]+1; 17109877982aSShri Abhyankar pj = b->j + bdiag[i+1]+1; 17119877982aSShri Abhyankar nz = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */ 17129877982aSShri Abhyankar for (j=0; j<nz; j++) { 17139877982aSShri Abhyankar col = pj[j]; 17149877982aSShri Abhyankar pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 17159877982aSShri Abhyankar } 17169877982aSShri Abhyankar 17179877982aSShri Abhyankar sctx.rs = rs; 17189877982aSShri Abhyankar sctx.pv = rtmp1[i]; 1719d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr); 172007b50cabSHong Zhang if(sctx.newshift) break; 17219877982aSShri Abhyankar pc1 = b->a + bdiag[i]; /* Mark diag[i] */ 17229877982aSShri Abhyankar *pc1 = 1.0/sctx.pv; 17239877982aSShri Abhyankar 17249877982aSShri Abhyankar /* Now take care of 1st column of diagonal 4x4 block. */ 17259877982aSShri Abhyankar pc2 = rtmp2 + i; 17269877982aSShri Abhyankar pc3 = rtmp3 + i; 17279877982aSShri Abhyankar pc4 = rtmp4 + i; 17289877982aSShri Abhyankar if (*pc2 != 0.0 || *pc3 != 0.0 || *pc4 != 0.0){ 17299877982aSShri Abhyankar mul2 = (*pc2)*(*pc1); *pc2 = mul2; 17309877982aSShri Abhyankar mul3 = (*pc3)*(*pc1); *pc3 = mul3; 17319877982aSShri Abhyankar mul4 = (*pc4)*(*pc1); *pc4 = mul4; 17329877982aSShri Abhyankar pj = b->j + bdiag[i+1]+1; /* beginning of U(i,:) */ 17339877982aSShri Abhyankar nz = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */ 17349877982aSShri Abhyankar for (j=0; j<nz; j++) { 17359877982aSShri Abhyankar col = pj[j]; 17369877982aSShri Abhyankar rtmp2[col] -= mul2 * rtmp1[col]; 17379877982aSShri Abhyankar rtmp3[col] -= mul3 * rtmp1[col]; 17389877982aSShri Abhyankar rtmp4[col] -= mul4 * rtmp1[col]; 17399877982aSShri Abhyankar } 17409877982aSShri Abhyankar ierr = PetscLogFlops(6*nz);CHKERRQ(ierr); 17419877982aSShri Abhyankar } 17429877982aSShri Abhyankar 17439877982aSShri Abhyankar /* finished row i+1; check zero pivot, then stick row i+1 into b->a */ 17449877982aSShri Abhyankar rs = 0.0; 17459877982aSShri Abhyankar /* L part */ 17469877982aSShri Abhyankar pc2 = b->a + bi[i+1]; 17479877982aSShri Abhyankar pj = b->j + bi[i+1] ; 17489877982aSShri Abhyankar nz = bi[i+2] - bi[i+1]; 17499877982aSShri Abhyankar for (j=0; j<nz; j++) { 17509877982aSShri Abhyankar col = pj[j]; 17519877982aSShri Abhyankar pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 17529877982aSShri Abhyankar } 17539877982aSShri Abhyankar /* U part */ 17549877982aSShri Abhyankar pc2 = b->a + bdiag[i+2]+1; 17559877982aSShri Abhyankar pj = b->j + bdiag[i+2]+1; 17569877982aSShri Abhyankar nz = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */ 17579877982aSShri Abhyankar for (j=0; j<nz; j++) { 17589877982aSShri Abhyankar col = pj[j]; 17599877982aSShri Abhyankar pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 17609877982aSShri Abhyankar } 17619877982aSShri Abhyankar 17629877982aSShri Abhyankar sctx.rs = rs; 17639877982aSShri Abhyankar sctx.pv = rtmp2[i+1]; 1764d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i+1);CHKERRQ(ierr); 176507b50cabSHong Zhang if(sctx.newshift) break; 17669877982aSShri Abhyankar pc2 = b->a + bdiag[i+1]; 17679877982aSShri Abhyankar *pc2 = 1.0/sctx.pv; /* Mark diag[i+1] */ 17689877982aSShri Abhyankar 17699877982aSShri Abhyankar /* Now take care of 2nd column of diagonal 4x4 block. */ 17709877982aSShri Abhyankar pc3 = rtmp3 + i+1; 17719877982aSShri Abhyankar pc4 = rtmp4 + i+1; 17729877982aSShri Abhyankar if (*pc3 != 0.0 || *pc4 != 0.0){ 17739877982aSShri Abhyankar mul3 = (*pc3)*(*pc2); *pc3 = mul3; 17749877982aSShri Abhyankar mul4 = (*pc4)*(*pc2); *pc4 = mul4; 17759877982aSShri Abhyankar pj = b->j + bdiag[i+2]+1; /* beginning of U(i+1,:) */ 17769877982aSShri Abhyankar nz = bdiag[i+1]-bdiag[i+2]-1; /* num of entries in U(i+1,:) excluding diag */ 17779877982aSShri Abhyankar for (j=0; j<nz; j++) { 17789877982aSShri Abhyankar col = pj[j]; 17799877982aSShri Abhyankar rtmp3[col] -= mul3 * rtmp2[col]; 17809877982aSShri Abhyankar rtmp4[col] -= mul4 * rtmp2[col]; 17819877982aSShri Abhyankar } 17829877982aSShri Abhyankar ierr = PetscLogFlops(4*nz);CHKERRQ(ierr); 17839877982aSShri Abhyankar } 17849877982aSShri Abhyankar 17859877982aSShri Abhyankar /* finished i+2; check zero pivot, then stick row i+2 into b->a */ 17869877982aSShri Abhyankar rs = 0.0; 17879877982aSShri Abhyankar /* L part */ 17889877982aSShri Abhyankar pc3 = b->a + bi[i+2]; 17899877982aSShri Abhyankar pj = b->j + bi[i+2] ; 17909877982aSShri Abhyankar nz = bi[i+3] - bi[i+2]; 17919877982aSShri Abhyankar for (j=0; j<nz; j++) { 17929877982aSShri Abhyankar col = pj[j]; 17939877982aSShri Abhyankar pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]); 17949877982aSShri Abhyankar } 17959877982aSShri Abhyankar /* U part */ 17969877982aSShri Abhyankar pc3 = b->a + bdiag[i+3]+1; 17979877982aSShri Abhyankar pj = b->j + bdiag[i+3]+1; 17989877982aSShri Abhyankar nz = bdiag[i+2] - bdiag[i+3] - 1; /* exclude diagonal */ 17999877982aSShri Abhyankar for (j=0; j<nz; j++) { 18009877982aSShri Abhyankar col = pj[j]; 18019877982aSShri Abhyankar pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]); 18029877982aSShri Abhyankar } 18039877982aSShri Abhyankar 18049877982aSShri Abhyankar sctx.rs = rs; 18059877982aSShri Abhyankar sctx.pv = rtmp3[i+2]; 1806d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i+2);CHKERRQ(ierr); 180707b50cabSHong Zhang if(sctx.newshift) break; 18089877982aSShri Abhyankar pc3 = b->a + bdiag[i+2]; 18099877982aSShri Abhyankar *pc3 = 1.0/sctx.pv; /* Mark diag[i+2] */ 18109877982aSShri Abhyankar 18119877982aSShri Abhyankar /* Now take care of 3rd column of diagonal 4x4 block. */ 18129877982aSShri Abhyankar pc4 = rtmp4 + i+2; 18139877982aSShri Abhyankar if (*pc4 != 0.0){ 18149877982aSShri Abhyankar mul4 = (*pc4)*(*pc3); *pc4 = mul4; 18159877982aSShri Abhyankar pj = b->j + bdiag[i+3]+1; /* beginning of U(i+2,:) */ 18169877982aSShri Abhyankar nz = bdiag[i+2]-bdiag[i+3]-1; /* num of entries in U(i+2,:) excluding diag */ 18179877982aSShri Abhyankar for (j=0; j<nz; j++) { 18189877982aSShri Abhyankar col = pj[j]; 18199877982aSShri Abhyankar rtmp4[col] -= mul4 * rtmp3[col]; 18209877982aSShri Abhyankar } 18219877982aSShri Abhyankar ierr = PetscLogFlops(2*nz);CHKERRQ(ierr); 18229877982aSShri Abhyankar } 18239877982aSShri Abhyankar 18249877982aSShri Abhyankar /* finished i+3; check zero pivot, then stick row i+3 into b->a */ 18259877982aSShri Abhyankar rs = 0.0; 18269877982aSShri Abhyankar /* L part */ 18279877982aSShri Abhyankar pc4 = b->a + bi[i+3]; 18289877982aSShri Abhyankar pj = b->j + bi[i+3] ; 18299877982aSShri Abhyankar nz = bi[i+4] - bi[i+3]; 18309877982aSShri Abhyankar for (j=0; j<nz; j++) { 18319877982aSShri Abhyankar col = pj[j]; 18329877982aSShri Abhyankar pc4[j] = rtmp4[col]; rs += PetscAbsScalar(pc4[j]); 18339877982aSShri Abhyankar } 18349877982aSShri Abhyankar /* U part */ 18359877982aSShri Abhyankar pc4 = b->a + bdiag[i+4]+1; 18369877982aSShri Abhyankar pj = b->j + bdiag[i+4]+1; 18379877982aSShri Abhyankar nz = bdiag[i+3] - bdiag[i+4] - 1; /* exclude diagonal */ 18389877982aSShri Abhyankar for (j=0; j<nz; j++) { 18399877982aSShri Abhyankar col = pj[j]; 18409877982aSShri Abhyankar pc4[j] = rtmp4[col]; rs += PetscAbsScalar(pc4[j]); 18419877982aSShri Abhyankar } 18429877982aSShri Abhyankar 18439877982aSShri Abhyankar sctx.rs = rs; 18449877982aSShri Abhyankar sctx.pv = rtmp4[i+3]; 1845d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,i+3);CHKERRQ(ierr); 184607b50cabSHong Zhang if(sctx.newshift) break; 18479877982aSShri Abhyankar pc4 = b->a + bdiag[i+3]; 18489877982aSShri Abhyankar *pc4 = 1.0/sctx.pv; /* Mark diag[i+3] */ 18499877982aSShri Abhyankar break; 185068785679SHong Zhang 185128f1b45aSHong Zhang default: 1852e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Node size not yet supported \n"); 185328f1b45aSHong Zhang } 1854c2b86aeeSHong Zhang if (sctx.newshift) break; /* break for (inod=0,i=0; inod<node_max; inod++) */ 185528f1b45aSHong Zhang i += nodesz; /* Update the row */ 185668785679SHong Zhang } 185728f1b45aSHong Zhang 185828f1b45aSHong Zhang /* MatPivotRefine() */ 185907b50cabSHong Zhang if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){ 186028f1b45aSHong Zhang /* 186128f1b45aSHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 186228f1b45aSHong Zhang * then try lower shift 186328f1b45aSHong Zhang */ 186428f1b45aSHong Zhang sctx.shift_hi = sctx.shift_fraction; 186528f1b45aSHong Zhang sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 186628f1b45aSHong Zhang sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 186707b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 186828f1b45aSHong Zhang sctx.nshift++; 186928f1b45aSHong Zhang } 187007b50cabSHong Zhang } while (sctx.newshift); 187128f1b45aSHong Zhang 1872b89f182dSHong Zhang ierr = PetscFree(rtmp1);CHKERRQ(ierr); 187368785679SHong Zhang ierr = PetscFree(tmp_vec2);CHKERRQ(ierr); 187428f1b45aSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 187528f1b45aSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 187628f1b45aSHong Zhang 1877d3ac4fa3SBarry Smith C->ops->solve = MatSolve_SeqAIJ; 187828f1b45aSHong Zhang C->ops->solveadd = MatSolveAdd_SeqAIJ; 187928f1b45aSHong Zhang C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 188028f1b45aSHong Zhang C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 188128f1b45aSHong Zhang C->ops->matsolve = MatMatSolve_SeqAIJ; 188228f1b45aSHong Zhang C->assembled = PETSC_TRUE; 188328f1b45aSHong Zhang C->preallocated = PETSC_TRUE; 188428f1b45aSHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 188528f1b45aSHong Zhang 188628f1b45aSHong Zhang /* MatShiftView(A,info,&sctx) */ 188728f1b45aSHong Zhang if (sctx.nshift){ 1888f4db908eSBarry Smith if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { 188928f1b45aSHong Zhang ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr); 1890f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 189128f1b45aSHong Zhang ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 1892f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){ 189328f1b45aSHong Zhang ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr); 189428f1b45aSHong Zhang } 189528f1b45aSHong Zhang } 1896d3ac4fa3SBarry Smith ierr = Mat_CheckInode_FactorLU(C,PETSC_FALSE);CHKERRQ(ierr); 189728f1b45aSHong Zhang PetscFunctionReturn(0); 189828f1b45aSHong Zhang } 1899628f99d7SShri Abhyankar 1900628f99d7SShri Abhyankar #undef __FUNCT__ 1901628f99d7SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_Inode_inplace" 1902628f99d7SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ_Inode_inplace(Mat B,Mat A,const MatFactorInfo *info) 1903628f99d7SShri Abhyankar { 1904628f99d7SShri Abhyankar Mat C = B; 1905628f99d7SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b = (Mat_SeqAIJ*)C->data; 1906628f99d7SShri Abhyankar IS iscol = b->col,isrow = b->row,isicol = b->icol; 1907628f99d7SShri Abhyankar PetscErrorCode ierr; 1908628f99d7SShri Abhyankar const PetscInt *r,*ic,*c,*ics; 1909628f99d7SShri Abhyankar PetscInt n = A->rmap->n,*bi = b->i; 1910628f99d7SShri Abhyankar PetscInt *bj = b->j,*nbj=b->j +1,*ajtmp,*bjtmp,nz,nz_tmp,row,prow; 1911628f99d7SShri Abhyankar PetscInt i,j,idx,*ai = a->i,*aj = a->j,*bd = b->diag,node_max,nodesz; 1912628f99d7SShri Abhyankar PetscInt *ns,*tmp_vec1,*tmp_vec2,*nsmap,*pj; 1913628f99d7SShri Abhyankar PetscScalar mul1,mul2,mul3,tmp; 1914628f99d7SShri Abhyankar MatScalar *pc1,*pc2,*pc3,*ba = b->a,*pv,*rtmp11,*rtmp22,*rtmp33; 1915628f99d7SShri Abhyankar const MatScalar *v1,*v2,*v3,*aa = a->a,*rtmp1; 1916628f99d7SShri Abhyankar PetscReal rs=0.0; 1917628f99d7SShri Abhyankar FactorShiftCtx sctx; 1918628f99d7SShri Abhyankar 1919628f99d7SShri Abhyankar PetscFunctionBegin; 1920628f99d7SShri Abhyankar sctx.shift_top = 0; 1921628f99d7SShri Abhyankar sctx.nshift_max = 0; 1922628f99d7SShri Abhyankar sctx.shift_lo = 0; 1923628f99d7SShri Abhyankar sctx.shift_hi = 0; 1924628f99d7SShri Abhyankar sctx.shift_fraction = 0; 1925628f99d7SShri Abhyankar 1926628f99d7SShri Abhyankar /* if both shift schemes are chosen by user, only use info->shiftpd */ 1927f4db908eSBarry Smith if (info->shifttype==(PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 1928628f99d7SShri Abhyankar sctx.shift_top = 0; 1929628f99d7SShri Abhyankar for (i=0; i<n; i++) { 1930628f99d7SShri Abhyankar /* calculate rs = sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 1931628f99d7SShri Abhyankar rs = 0.0; 1932628f99d7SShri Abhyankar ajtmp = aj + ai[i]; 1933628f99d7SShri Abhyankar rtmp1 = aa + ai[i]; 1934628f99d7SShri Abhyankar nz = ai[i+1] - ai[i]; 1935628f99d7SShri Abhyankar for (j=0; j<nz; j++){ 1936628f99d7SShri Abhyankar if (*ajtmp != i){ 1937628f99d7SShri Abhyankar rs += PetscAbsScalar(*rtmp1++); 1938628f99d7SShri Abhyankar } else { 1939628f99d7SShri Abhyankar rs -= PetscRealPart(*rtmp1++); 1940628f99d7SShri Abhyankar } 1941628f99d7SShri Abhyankar ajtmp++; 1942628f99d7SShri Abhyankar } 1943628f99d7SShri Abhyankar if (rs>sctx.shift_top) sctx.shift_top = rs; 1944628f99d7SShri Abhyankar } 1945628f99d7SShri Abhyankar if (sctx.shift_top == 0.0) sctx.shift_top += 1.e-12; 1946628f99d7SShri Abhyankar sctx.shift_top *= 1.1; 1947628f99d7SShri Abhyankar sctx.nshift_max = 5; 1948628f99d7SShri Abhyankar sctx.shift_lo = 0.; 1949628f99d7SShri Abhyankar sctx.shift_hi = 1.; 1950628f99d7SShri Abhyankar } 1951628f99d7SShri Abhyankar sctx.shift_amount = 0; 1952628f99d7SShri Abhyankar sctx.nshift = 0; 1953628f99d7SShri Abhyankar 1954628f99d7SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1955628f99d7SShri Abhyankar ierr = ISGetIndices(iscol,&c);CHKERRQ(ierr); 1956628f99d7SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 1957628f99d7SShri Abhyankar ierr = PetscMalloc((3*n+1)*sizeof(PetscScalar),&rtmp11);CHKERRQ(ierr); 1958628f99d7SShri Abhyankar ierr = PetscMemzero(rtmp11,(3*n+1)*sizeof(PetscScalar));CHKERRQ(ierr); 1959628f99d7SShri Abhyankar ics = ic ; 1960628f99d7SShri Abhyankar rtmp22 = rtmp11 + n; 1961628f99d7SShri Abhyankar rtmp33 = rtmp22 + n; 1962628f99d7SShri Abhyankar 1963628f99d7SShri Abhyankar node_max = a->inode.node_count; 1964628f99d7SShri Abhyankar ns = a->inode.size; 1965e7e72b3dSBarry Smith if (!ns) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix without inode information"); 1966628f99d7SShri Abhyankar 1967628f99d7SShri Abhyankar /* If max inode size > 3, split it into two inodes.*/ 1968628f99d7SShri Abhyankar /* also map the inode sizes according to the ordering */ 1969628f99d7SShri Abhyankar ierr = PetscMalloc((n+1)* sizeof(PetscInt),&tmp_vec1);CHKERRQ(ierr); 1970628f99d7SShri Abhyankar for (i=0,j=0; i<node_max; ++i,++j){ 1971628f99d7SShri Abhyankar if (ns[i]>3) { 1972628f99d7SShri Abhyankar tmp_vec1[j] = ns[i]/2; /* Assuming ns[i] < =5 */ 1973628f99d7SShri Abhyankar ++j; 1974628f99d7SShri Abhyankar tmp_vec1[j] = ns[i] - tmp_vec1[j-1]; 1975628f99d7SShri Abhyankar } else { 1976628f99d7SShri Abhyankar tmp_vec1[j] = ns[i]; 1977628f99d7SShri Abhyankar } 1978628f99d7SShri Abhyankar } 1979628f99d7SShri Abhyankar /* Use the correct node_max */ 1980628f99d7SShri Abhyankar node_max = j; 1981628f99d7SShri Abhyankar 1982628f99d7SShri Abhyankar /* Now reorder the inode info based on mat re-ordering info */ 1983628f99d7SShri Abhyankar /* First create a row -> inode_size_array_index map */ 1984628f99d7SShri Abhyankar ierr = PetscMalloc(n*sizeof(PetscInt)+1,&nsmap);CHKERRQ(ierr); 1985628f99d7SShri Abhyankar ierr = PetscMalloc(node_max*sizeof(PetscInt)+1,&tmp_vec2);CHKERRQ(ierr); 1986628f99d7SShri Abhyankar for (i=0,row=0; i<node_max; i++) { 1987628f99d7SShri Abhyankar nodesz = tmp_vec1[i]; 1988628f99d7SShri Abhyankar for (j=0; j<nodesz; j++,row++) { 1989628f99d7SShri Abhyankar nsmap[row] = i; 1990628f99d7SShri Abhyankar } 1991628f99d7SShri Abhyankar } 1992628f99d7SShri Abhyankar /* Using nsmap, create a reordered ns structure */ 1993628f99d7SShri Abhyankar for (i=0,j=0; i< node_max; i++) { 1994628f99d7SShri Abhyankar nodesz = tmp_vec1[nsmap[r[j]]]; /* here the reordered row_no is in r[] */ 1995628f99d7SShri Abhyankar tmp_vec2[i] = nodesz; 1996628f99d7SShri Abhyankar j += nodesz; 1997628f99d7SShri Abhyankar } 1998628f99d7SShri Abhyankar ierr = PetscFree(nsmap);CHKERRQ(ierr); 1999628f99d7SShri Abhyankar ierr = PetscFree(tmp_vec1);CHKERRQ(ierr); 2000628f99d7SShri Abhyankar /* Now use the correct ns */ 2001628f99d7SShri Abhyankar ns = tmp_vec2; 2002628f99d7SShri Abhyankar 2003628f99d7SShri Abhyankar do { 200407b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 2005628f99d7SShri Abhyankar /* Now loop over each block-row, and do the factorization */ 2006628f99d7SShri Abhyankar for (i=0,row=0; i<node_max; i++) { 2007628f99d7SShri Abhyankar nodesz = ns[i]; 2008628f99d7SShri Abhyankar nz = bi[row+1] - bi[row]; 2009628f99d7SShri Abhyankar bjtmp = bj + bi[row]; 2010628f99d7SShri Abhyankar 2011628f99d7SShri Abhyankar switch (nodesz){ 2012628f99d7SShri Abhyankar case 1: 2013628f99d7SShri Abhyankar for (j=0; j<nz; j++){ 2014628f99d7SShri Abhyankar idx = bjtmp[j]; 2015628f99d7SShri Abhyankar rtmp11[idx] = 0.0; 2016628f99d7SShri Abhyankar } 2017628f99d7SShri Abhyankar 2018628f99d7SShri Abhyankar /* load in initial (unfactored row) */ 2019628f99d7SShri Abhyankar idx = r[row]; 2020628f99d7SShri Abhyankar nz_tmp = ai[idx+1] - ai[idx]; 2021628f99d7SShri Abhyankar ajtmp = aj + ai[idx]; 2022628f99d7SShri Abhyankar v1 = aa + ai[idx]; 2023628f99d7SShri Abhyankar 2024628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2025628f99d7SShri Abhyankar idx = ics[ajtmp[j]]; 2026628f99d7SShri Abhyankar rtmp11[idx] = v1[j]; 2027628f99d7SShri Abhyankar } 2028628f99d7SShri Abhyankar rtmp11[ics[r[row]]] += sctx.shift_amount; 2029628f99d7SShri Abhyankar 2030628f99d7SShri Abhyankar prow = *bjtmp++ ; 2031628f99d7SShri Abhyankar while (prow < row) { 2032628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 2033628f99d7SShri Abhyankar if (*pc1 != 0.0){ 2034628f99d7SShri Abhyankar pv = ba + bd[prow]; 2035628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2036628f99d7SShri Abhyankar mul1 = *pc1 * *pv++; 2037628f99d7SShri Abhyankar *pc1 = mul1; 2038628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2039628f99d7SShri Abhyankar ierr = PetscLogFlops(2*nz_tmp);CHKERRQ(ierr); 2040628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2041628f99d7SShri Abhyankar tmp = pv[j]; 2042628f99d7SShri Abhyankar idx = pj[j]; 2043628f99d7SShri Abhyankar rtmp11[idx] -= mul1 * tmp; 2044628f99d7SShri Abhyankar } 2045628f99d7SShri Abhyankar } 2046628f99d7SShri Abhyankar prow = *bjtmp++ ; 2047628f99d7SShri Abhyankar } 2048628f99d7SShri Abhyankar pj = bj + bi[row]; 2049628f99d7SShri Abhyankar pc1 = ba + bi[row]; 2050628f99d7SShri Abhyankar 2051628f99d7SShri Abhyankar sctx.pv = rtmp11[row]; 2052628f99d7SShri Abhyankar rtmp11[row] = 1.0/rtmp11[row]; /* invert diag */ 2053628f99d7SShri Abhyankar rs = 0.0; 2054628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2055628f99d7SShri Abhyankar idx = pj[j]; 2056628f99d7SShri Abhyankar pc1[j] = rtmp11[idx]; /* rtmp11 -> ba */ 2057628f99d7SShri Abhyankar if (idx != row) rs += PetscAbsScalar(pc1[j]); 2058628f99d7SShri Abhyankar } 2059628f99d7SShri Abhyankar sctx.rs = rs; 2060d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,row);CHKERRQ(ierr); 206107b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2062628f99d7SShri Abhyankar break; 2063628f99d7SShri Abhyankar 2064628f99d7SShri Abhyankar case 2: 2065628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2066628f99d7SShri Abhyankar idx = bjtmp[j]; 2067628f99d7SShri Abhyankar rtmp11[idx] = 0.0; 2068628f99d7SShri Abhyankar rtmp22[idx] = 0.0; 2069628f99d7SShri Abhyankar } 2070628f99d7SShri Abhyankar 2071628f99d7SShri Abhyankar /* load in initial (unfactored row) */ 2072628f99d7SShri Abhyankar idx = r[row]; 2073628f99d7SShri Abhyankar nz_tmp = ai[idx+1] - ai[idx]; 2074628f99d7SShri Abhyankar ajtmp = aj + ai[idx]; 2075628f99d7SShri Abhyankar v1 = aa + ai[idx]; 2076628f99d7SShri Abhyankar v2 = aa + ai[idx+1]; 2077628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2078628f99d7SShri Abhyankar idx = ics[ajtmp[j]]; 2079628f99d7SShri Abhyankar rtmp11[idx] = v1[j]; 2080628f99d7SShri Abhyankar rtmp22[idx] = v2[j]; 2081628f99d7SShri Abhyankar } 2082628f99d7SShri Abhyankar rtmp11[ics[r[row]]] += sctx.shift_amount; 2083628f99d7SShri Abhyankar rtmp22[ics[r[row+1]]] += sctx.shift_amount; 2084628f99d7SShri Abhyankar 2085628f99d7SShri Abhyankar prow = *bjtmp++ ; 2086628f99d7SShri Abhyankar while (prow < row) { 2087628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 2088628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2089628f99d7SShri Abhyankar if (*pc1 != 0.0 || *pc2 != 0.0){ 2090628f99d7SShri Abhyankar pv = ba + bd[prow]; 2091628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2092628f99d7SShri Abhyankar mul1 = *pc1 * *pv; 2093628f99d7SShri Abhyankar mul2 = *pc2 * *pv; 2094628f99d7SShri Abhyankar ++pv; 2095628f99d7SShri Abhyankar *pc1 = mul1; 2096628f99d7SShri Abhyankar *pc2 = mul2; 2097628f99d7SShri Abhyankar 2098628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2099628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2100628f99d7SShri Abhyankar tmp = pv[j]; 2101628f99d7SShri Abhyankar idx = pj[j]; 2102628f99d7SShri Abhyankar rtmp11[idx] -= mul1 * tmp; 2103628f99d7SShri Abhyankar rtmp22[idx] -= mul2 * tmp; 2104628f99d7SShri Abhyankar } 2105628f99d7SShri Abhyankar ierr = PetscLogFlops(4*nz_tmp);CHKERRQ(ierr); 2106628f99d7SShri Abhyankar } 2107628f99d7SShri Abhyankar prow = *bjtmp++ ; 2108628f99d7SShri Abhyankar } 2109628f99d7SShri Abhyankar 2110628f99d7SShri Abhyankar /* Now take care of diagonal 2x2 block. Note: prow = row here */ 2111628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 2112628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2113628f99d7SShri Abhyankar 2114628f99d7SShri Abhyankar sctx.pv = *pc1; 2115628f99d7SShri Abhyankar pj = bj + bi[prow]; 2116628f99d7SShri Abhyankar rs = 0.0; 2117628f99d7SShri Abhyankar for (j=0; j<nz; j++){ 2118628f99d7SShri Abhyankar idx = pj[j]; 2119628f99d7SShri Abhyankar if (idx != prow) rs += PetscAbsScalar(rtmp11[idx]); 2120628f99d7SShri Abhyankar } 2121628f99d7SShri Abhyankar sctx.rs = rs; 2122d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,row);CHKERRQ(ierr); 212307b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2124628f99d7SShri Abhyankar 2125628f99d7SShri Abhyankar if (*pc2 != 0.0){ 2126628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2127628f99d7SShri Abhyankar mul2 = (*pc2)/(*pc1); /* since diag is not yet inverted.*/ 2128628f99d7SShri Abhyankar *pc2 = mul2; 2129628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2130628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2131628f99d7SShri Abhyankar idx = pj[j] ; 2132628f99d7SShri Abhyankar tmp = rtmp11[idx]; 2133628f99d7SShri Abhyankar rtmp22[idx] -= mul2 * tmp; 2134628f99d7SShri Abhyankar } 2135628f99d7SShri Abhyankar ierr = PetscLogFlops(2*nz_tmp);CHKERRQ(ierr); 2136628f99d7SShri Abhyankar } 2137628f99d7SShri Abhyankar 2138628f99d7SShri Abhyankar pj = bj + bi[row]; 2139628f99d7SShri Abhyankar pc1 = ba + bi[row]; 2140628f99d7SShri Abhyankar pc2 = ba + bi[row+1]; 2141628f99d7SShri Abhyankar 2142628f99d7SShri Abhyankar sctx.pv = rtmp22[row+1]; 2143628f99d7SShri Abhyankar rs = 0.0; 2144628f99d7SShri Abhyankar rtmp11[row] = 1.0/rtmp11[row]; 2145628f99d7SShri Abhyankar rtmp22[row+1] = 1.0/rtmp22[row+1]; 2146628f99d7SShri Abhyankar /* copy row entries from dense representation to sparse */ 2147628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2148628f99d7SShri Abhyankar idx = pj[j]; 2149628f99d7SShri Abhyankar pc1[j] = rtmp11[idx]; 2150628f99d7SShri Abhyankar pc2[j] = rtmp22[idx]; 2151628f99d7SShri Abhyankar if (idx != row+1) rs += PetscAbsScalar(pc2[j]); 2152628f99d7SShri Abhyankar } 2153628f99d7SShri Abhyankar sctx.rs = rs; 2154d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,row+1);CHKERRQ(ierr); 215507b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2156628f99d7SShri Abhyankar break; 2157628f99d7SShri Abhyankar 2158628f99d7SShri Abhyankar case 3: 2159628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2160628f99d7SShri Abhyankar idx = bjtmp[j]; 2161628f99d7SShri Abhyankar rtmp11[idx] = 0.0; 2162628f99d7SShri Abhyankar rtmp22[idx] = 0.0; 2163628f99d7SShri Abhyankar rtmp33[idx] = 0.0; 2164628f99d7SShri Abhyankar } 2165628f99d7SShri Abhyankar /* copy the nonzeros for the 3 rows from sparse representation to dense in rtmp*[] */ 2166628f99d7SShri Abhyankar idx = r[row]; 2167628f99d7SShri Abhyankar nz_tmp = ai[idx+1] - ai[idx]; 2168628f99d7SShri Abhyankar ajtmp = aj + ai[idx]; 2169628f99d7SShri Abhyankar v1 = aa + ai[idx]; 2170628f99d7SShri Abhyankar v2 = aa + ai[idx+1]; 2171628f99d7SShri Abhyankar v3 = aa + ai[idx+2]; 2172628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2173628f99d7SShri Abhyankar idx = ics[ajtmp[j]]; 2174628f99d7SShri Abhyankar rtmp11[idx] = v1[j]; 2175628f99d7SShri Abhyankar rtmp22[idx] = v2[j]; 2176628f99d7SShri Abhyankar rtmp33[idx] = v3[j]; 2177628f99d7SShri Abhyankar } 2178628f99d7SShri Abhyankar rtmp11[ics[r[row]]] += sctx.shift_amount; 2179628f99d7SShri Abhyankar rtmp22[ics[r[row+1]]] += sctx.shift_amount; 2180628f99d7SShri Abhyankar rtmp33[ics[r[row+2]]] += sctx.shift_amount; 2181628f99d7SShri Abhyankar 2182628f99d7SShri Abhyankar /* loop over all pivot row blocks above this row block */ 2183628f99d7SShri Abhyankar prow = *bjtmp++ ; 2184628f99d7SShri Abhyankar while (prow < row) { 2185628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 2186628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2187628f99d7SShri Abhyankar pc3 = rtmp33 + prow; 2188628f99d7SShri Abhyankar if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 !=0.0){ 2189628f99d7SShri Abhyankar pv = ba + bd[prow]; 2190628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2191628f99d7SShri Abhyankar mul1 = *pc1 * *pv; 2192628f99d7SShri Abhyankar mul2 = *pc2 * *pv; 2193628f99d7SShri Abhyankar mul3 = *pc3 * *pv; 2194628f99d7SShri Abhyankar ++pv; 2195628f99d7SShri Abhyankar *pc1 = mul1; 2196628f99d7SShri Abhyankar *pc2 = mul2; 2197628f99d7SShri Abhyankar *pc3 = mul3; 2198628f99d7SShri Abhyankar 2199628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2200628f99d7SShri Abhyankar /* update this row based on pivot row */ 2201628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2202628f99d7SShri Abhyankar tmp = pv[j]; 2203628f99d7SShri Abhyankar idx = pj[j]; 2204628f99d7SShri Abhyankar rtmp11[idx] -= mul1 * tmp; 2205628f99d7SShri Abhyankar rtmp22[idx] -= mul2 * tmp; 2206628f99d7SShri Abhyankar rtmp33[idx] -= mul3 * tmp; 2207628f99d7SShri Abhyankar } 2208628f99d7SShri Abhyankar ierr = PetscLogFlops(6*nz_tmp);CHKERRQ(ierr); 2209628f99d7SShri Abhyankar } 2210628f99d7SShri Abhyankar prow = *bjtmp++ ; 2211628f99d7SShri Abhyankar } 2212628f99d7SShri Abhyankar 2213628f99d7SShri Abhyankar /* Now take care of diagonal 3x3 block in this set of rows */ 2214628f99d7SShri Abhyankar /* note: prow = row here */ 2215628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 2216628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2217628f99d7SShri Abhyankar pc3 = rtmp33 + prow; 2218628f99d7SShri Abhyankar 2219628f99d7SShri Abhyankar sctx.pv = *pc1; 2220628f99d7SShri Abhyankar pj = bj + bi[prow]; 2221628f99d7SShri Abhyankar rs = 0.0; 2222628f99d7SShri Abhyankar for (j=0; j<nz; j++){ 2223628f99d7SShri Abhyankar idx = pj[j]; 2224628f99d7SShri Abhyankar if (idx != row) rs += PetscAbsScalar(rtmp11[idx]); 2225628f99d7SShri Abhyankar } 2226628f99d7SShri Abhyankar sctx.rs = rs; 2227d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,row);CHKERRQ(ierr); 222807b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2229628f99d7SShri Abhyankar 2230628f99d7SShri Abhyankar if (*pc2 != 0.0 || *pc3 != 0.0){ 2231628f99d7SShri Abhyankar mul2 = (*pc2)/(*pc1); 2232628f99d7SShri Abhyankar mul3 = (*pc3)/(*pc1); 2233628f99d7SShri Abhyankar *pc2 = mul2; 2234628f99d7SShri Abhyankar *pc3 = mul3; 2235628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2236628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2237628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2238628f99d7SShri Abhyankar idx = pj[j] ; 2239628f99d7SShri Abhyankar tmp = rtmp11[idx]; 2240628f99d7SShri Abhyankar rtmp22[idx] -= mul2 * tmp; 2241628f99d7SShri Abhyankar rtmp33[idx] -= mul3 * tmp; 2242628f99d7SShri Abhyankar } 2243628f99d7SShri Abhyankar ierr = PetscLogFlops(4*nz_tmp);CHKERRQ(ierr); 2244628f99d7SShri Abhyankar } 2245628f99d7SShri Abhyankar ++prow; 2246628f99d7SShri Abhyankar 2247628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2248628f99d7SShri Abhyankar pc3 = rtmp33 + prow; 2249628f99d7SShri Abhyankar sctx.pv = *pc2; 2250628f99d7SShri Abhyankar pj = bj + bi[prow]; 2251628f99d7SShri Abhyankar rs = 0.0; 2252628f99d7SShri Abhyankar for (j=0; j<nz; j++){ 2253628f99d7SShri Abhyankar idx = pj[j]; 2254628f99d7SShri Abhyankar if (idx != prow) rs += PetscAbsScalar(rtmp22[idx]); 2255628f99d7SShri Abhyankar } 2256628f99d7SShri Abhyankar sctx.rs = rs; 2257d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,row+1);CHKERRQ(ierr); 225807b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2259628f99d7SShri Abhyankar 2260628f99d7SShri Abhyankar if (*pc3 != 0.0){ 2261628f99d7SShri Abhyankar mul3 = (*pc3)/(*pc2); 2262628f99d7SShri Abhyankar *pc3 = mul3; 2263628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2264628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2265628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2266628f99d7SShri Abhyankar idx = pj[j] ; 2267628f99d7SShri Abhyankar tmp = rtmp22[idx]; 2268628f99d7SShri Abhyankar rtmp33[idx] -= mul3 * tmp; 2269628f99d7SShri Abhyankar } 227068785679SHong Zhang ierr = PetscLogFlops(2*nz_tmp);CHKERRQ(ierr); 2271628f99d7SShri Abhyankar } 2272628f99d7SShri Abhyankar 2273628f99d7SShri Abhyankar pj = bj + bi[row]; 2274628f99d7SShri Abhyankar pc1 = ba + bi[row]; 2275628f99d7SShri Abhyankar pc2 = ba + bi[row+1]; 2276628f99d7SShri Abhyankar pc3 = ba + bi[row+2]; 2277628f99d7SShri Abhyankar 2278628f99d7SShri Abhyankar sctx.pv = rtmp33[row+2]; 2279628f99d7SShri Abhyankar rs = 0.0; 2280628f99d7SShri Abhyankar rtmp11[row] = 1.0/rtmp11[row]; 2281628f99d7SShri Abhyankar rtmp22[row+1] = 1.0/rtmp22[row+1]; 2282628f99d7SShri Abhyankar rtmp33[row+2] = 1.0/rtmp33[row+2]; 2283628f99d7SShri Abhyankar /* copy row entries from dense representation to sparse */ 2284628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2285628f99d7SShri Abhyankar idx = pj[j]; 2286628f99d7SShri Abhyankar pc1[j] = rtmp11[idx]; 2287628f99d7SShri Abhyankar pc2[j] = rtmp22[idx]; 2288628f99d7SShri Abhyankar pc3[j] = rtmp33[idx]; 2289628f99d7SShri Abhyankar if (idx != row+2) rs += PetscAbsScalar(pc3[j]); 2290628f99d7SShri Abhyankar } 2291628f99d7SShri Abhyankar 2292628f99d7SShri Abhyankar sctx.rs = rs; 2293d0660788SBarry Smith ierr = MatPivotCheck(A,info,&sctx,row+2);CHKERRQ(ierr); 229407b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2295628f99d7SShri Abhyankar break; 2296628f99d7SShri Abhyankar 2297628f99d7SShri Abhyankar default: 2298e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Node size not yet supported \n"); 2299628f99d7SShri Abhyankar } 2300628f99d7SShri Abhyankar row += nodesz; /* Update the row */ 2301628f99d7SShri Abhyankar } 2302628f99d7SShri Abhyankar endofwhile:; 230307b50cabSHong Zhang } while (sctx.newshift); 2304628f99d7SShri Abhyankar ierr = PetscFree(rtmp11);CHKERRQ(ierr); 2305628f99d7SShri Abhyankar ierr = PetscFree(tmp_vec2);CHKERRQ(ierr); 2306628f99d7SShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2307628f99d7SShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 2308628f99d7SShri Abhyankar ierr = ISRestoreIndices(iscol,&c);CHKERRQ(ierr); 2309d3ac4fa3SBarry Smith (B)->ops->solve = MatSolve_SeqAIJ_inplace; 2310628f99d7SShri Abhyankar /* do not set solve add, since MatSolve_Inode + Add is faster */ 2311628f99d7SShri Abhyankar C->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 2312628f99d7SShri Abhyankar C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 2313628f99d7SShri Abhyankar C->assembled = PETSC_TRUE; 2314628f99d7SShri Abhyankar C->preallocated = PETSC_TRUE; 2315628f99d7SShri Abhyankar if (sctx.nshift) { 2316f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 2317628f99d7SShri Abhyankar ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr); 2318f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 2319628f99d7SShri Abhyankar ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr); 2320628f99d7SShri Abhyankar } 2321628f99d7SShri Abhyankar } 2322628f99d7SShri Abhyankar ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 2323d3ac4fa3SBarry Smith ierr = Mat_CheckInode(C,PETSC_FALSE);CHKERRQ(ierr); 2324628f99d7SShri Abhyankar PetscFunctionReturn(0); 2325628f99d7SShri Abhyankar } 2326628f99d7SShri Abhyankar 2327628f99d7SShri Abhyankar 2328019b515eSShri Abhyankar /* ----------------------------------------------------------- */ 2329019b515eSShri Abhyankar #undef __FUNCT__ 233035233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_Inode" 233135233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_Inode(Mat A,Vec bb,Vec xx) 2332019b515eSShri Abhyankar { 2333019b515eSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2334019b515eSShri Abhyankar IS iscol = a->col,isrow = a->row; 2335019b515eSShri Abhyankar PetscErrorCode ierr; 2336019b515eSShri Abhyankar const PetscInt *r,*c,*rout,*cout; 2337019b515eSShri Abhyankar PetscInt i,j,n = A->rmap->n,*ai = a->i,nz,*a_j = a->j; 2338019b515eSShri Abhyankar PetscInt node_max,*ns,row,nsz,aii,*vi,*ad,*aj,i0,i1; 2339019b515eSShri Abhyankar PetscScalar *x,*tmp,*tmps,tmp0,tmp1; 2340019b515eSShri Abhyankar PetscScalar sum1,sum2,sum3,sum4,sum5; 2341019b515eSShri Abhyankar const MatScalar *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa; 2342019b515eSShri Abhyankar const PetscScalar *b; 2343019b515eSShri Abhyankar 2344019b515eSShri Abhyankar PetscFunctionBegin; 2345e32f2f54SBarry Smith if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 2346019b515eSShri Abhyankar node_max = a->inode.node_count; 2347019b515eSShri Abhyankar ns = a->inode.size; /* Node Size array */ 2348019b515eSShri Abhyankar 23493649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 2350019b515eSShri Abhyankar ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2351019b515eSShri Abhyankar tmp = a->solve_work; 2352019b515eSShri Abhyankar 2353019b515eSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 2354019b515eSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 2355019b515eSShri Abhyankar 2356019b515eSShri Abhyankar /* forward solve the lower triangular */ 2357019b515eSShri Abhyankar tmps = tmp ; 2358019b515eSShri Abhyankar aa = a_a ; 2359019b515eSShri Abhyankar aj = a_j ; 2360019b515eSShri Abhyankar ad = a->diag; 2361019b515eSShri Abhyankar 2362019b515eSShri Abhyankar for (i = 0,row = 0; i< node_max; ++i){ 2363019b515eSShri Abhyankar nsz = ns[i]; 2364019b515eSShri Abhyankar aii = ai[row]; 2365019b515eSShri Abhyankar v1 = aa + aii; 2366019b515eSShri Abhyankar vi = aj + aii; 2367019b515eSShri Abhyankar nz = ai[row+1]- ai[row]; 2368019b515eSShri Abhyankar 236998991853SShri Abhyankar if (i < node_max-1) { 237098991853SShri Abhyankar /* Prefetch the indices for the next block */ 237150d8bf02SJed Brown PetscPrefetchBlock(aj+ai[row+nsz],ai[row+nsz+1]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); /* indices */ 237298991853SShri Abhyankar /* Prefetch the data for the next block */ 237350d8bf02SJed Brown PetscPrefetchBlock(aa+ai[row+nsz],ai[row+nsz+ns[i+1]]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); 237498991853SShri Abhyankar } 237598991853SShri Abhyankar 2376019b515eSShri Abhyankar switch (nsz){ /* Each loop in 'case' is unrolled */ 2377019b515eSShri Abhyankar case 1 : 2378019b515eSShri Abhyankar sum1 = b[r[row]]; 2379019b515eSShri Abhyankar for(j=0; j<nz-1; j+=2){ 2380019b515eSShri Abhyankar i0 = vi[j]; 2381019b515eSShri Abhyankar i1 = vi[j+1]; 2382019b515eSShri Abhyankar tmp0 = tmps[i0]; 2383019b515eSShri Abhyankar tmp1 = tmps[i1]; 2384019b515eSShri Abhyankar sum1 -= v1[j]*tmp0 + v1[j+1]*tmp1; 2385019b515eSShri Abhyankar } 2386019b515eSShri Abhyankar if(j == nz-1){ 2387019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2388019b515eSShri Abhyankar sum1 -= v1[j]*tmp0; 2389019b515eSShri Abhyankar } 2390019b515eSShri Abhyankar tmp[row++]=sum1; 2391019b515eSShri Abhyankar break; 2392019b515eSShri Abhyankar case 2: 2393019b515eSShri Abhyankar sum1 = b[r[row]]; 2394019b515eSShri Abhyankar sum2 = b[r[row+1]]; 2395019b515eSShri Abhyankar v2 = aa + ai[row+1]; 2396019b515eSShri Abhyankar 2397019b515eSShri Abhyankar for(j=0; j<nz-1; j+=2){ 2398019b515eSShri Abhyankar i0 = vi[j]; 2399019b515eSShri Abhyankar i1 = vi[j+1]; 2400019b515eSShri Abhyankar tmp0 = tmps[i0]; 2401019b515eSShri Abhyankar tmp1 = tmps[i1]; 2402019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2403019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 2404019b515eSShri Abhyankar } 2405019b515eSShri Abhyankar if(j == nz-1){ 2406019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2407019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 2408019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 2409019b515eSShri Abhyankar } 2410019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 2411019b515eSShri Abhyankar tmp[row ++]=sum1; 2412019b515eSShri Abhyankar tmp[row ++]=sum2; 2413019b515eSShri Abhyankar break; 2414019b515eSShri Abhyankar case 3: 2415019b515eSShri Abhyankar sum1 = b[r[row]]; 2416019b515eSShri Abhyankar sum2 = b[r[row+1]]; 2417019b515eSShri Abhyankar sum3 = b[r[row+2]]; 2418019b515eSShri Abhyankar v2 = aa + ai[row+1]; 2419019b515eSShri Abhyankar v3 = aa + ai[row+2]; 2420019b515eSShri Abhyankar 2421019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2){ 2422019b515eSShri Abhyankar i0 = vi[j]; 2423019b515eSShri Abhyankar i1 = vi[j+1]; 2424019b515eSShri Abhyankar tmp0 = tmps[i0]; 2425019b515eSShri Abhyankar tmp1 = tmps[i1]; 2426019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2427019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 2428019b515eSShri Abhyankar sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1; 2429019b515eSShri Abhyankar } 2430019b515eSShri Abhyankar if (j == nz-1){ 2431019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2432019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 2433019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 2434019b515eSShri Abhyankar sum3 -= v3[j] *tmp0; 2435019b515eSShri Abhyankar } 2436019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 2437019b515eSShri Abhyankar sum3 -= v3[nz] * sum1; 2438019b515eSShri Abhyankar sum3 -= v3[nz+1] * sum2; 2439019b515eSShri Abhyankar tmp[row ++]=sum1; 2440019b515eSShri Abhyankar tmp[row ++]=sum2; 2441019b515eSShri Abhyankar tmp[row ++]=sum3; 2442019b515eSShri Abhyankar break; 2443019b515eSShri Abhyankar 2444019b515eSShri Abhyankar case 4: 2445019b515eSShri Abhyankar sum1 = b[r[row]]; 2446019b515eSShri Abhyankar sum2 = b[r[row+1]]; 2447019b515eSShri Abhyankar sum3 = b[r[row+2]]; 2448019b515eSShri Abhyankar sum4 = b[r[row+3]]; 2449019b515eSShri Abhyankar v2 = aa + ai[row+1]; 2450019b515eSShri Abhyankar v3 = aa + ai[row+2]; 2451019b515eSShri Abhyankar v4 = aa + ai[row+3]; 2452019b515eSShri Abhyankar 2453019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2){ 2454019b515eSShri Abhyankar i0 = vi[j]; 2455019b515eSShri Abhyankar i1 = vi[j+1]; 2456019b515eSShri Abhyankar tmp0 = tmps[i0]; 2457019b515eSShri Abhyankar tmp1 = tmps[i1]; 2458019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2459019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 2460019b515eSShri Abhyankar sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1; 2461019b515eSShri Abhyankar sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1; 2462019b515eSShri Abhyankar } 2463019b515eSShri Abhyankar if (j == nz-1){ 2464019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2465019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 2466019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 2467019b515eSShri Abhyankar sum3 -= v3[j] *tmp0; 2468019b515eSShri Abhyankar sum4 -= v4[j] *tmp0; 2469019b515eSShri Abhyankar } 2470019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 2471019b515eSShri Abhyankar sum3 -= v3[nz] * sum1; 2472019b515eSShri Abhyankar sum4 -= v4[nz] * sum1; 2473019b515eSShri Abhyankar sum3 -= v3[nz+1] * sum2; 2474019b515eSShri Abhyankar sum4 -= v4[nz+1] * sum2; 2475019b515eSShri Abhyankar sum4 -= v4[nz+2] * sum3; 2476019b515eSShri Abhyankar 2477019b515eSShri Abhyankar tmp[row ++]=sum1; 2478019b515eSShri Abhyankar tmp[row ++]=sum2; 2479019b515eSShri Abhyankar tmp[row ++]=sum3; 2480019b515eSShri Abhyankar tmp[row ++]=sum4; 2481019b515eSShri Abhyankar break; 2482019b515eSShri Abhyankar case 5: 2483019b515eSShri Abhyankar sum1 = b[r[row]]; 2484019b515eSShri Abhyankar sum2 = b[r[row+1]]; 2485019b515eSShri Abhyankar sum3 = b[r[row+2]]; 2486019b515eSShri Abhyankar sum4 = b[r[row+3]]; 2487019b515eSShri Abhyankar sum5 = b[r[row+4]]; 2488019b515eSShri Abhyankar v2 = aa + ai[row+1]; 2489019b515eSShri Abhyankar v3 = aa + ai[row+2]; 2490019b515eSShri Abhyankar v4 = aa + ai[row+3]; 2491019b515eSShri Abhyankar v5 = aa + ai[row+4]; 2492019b515eSShri Abhyankar 2493019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2){ 2494019b515eSShri Abhyankar i0 = vi[j]; 2495019b515eSShri Abhyankar i1 = vi[j+1]; 2496019b515eSShri Abhyankar tmp0 = tmps[i0]; 2497019b515eSShri Abhyankar tmp1 = tmps[i1]; 2498019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2499019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 2500019b515eSShri Abhyankar sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1; 2501019b515eSShri Abhyankar sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1; 2502019b515eSShri Abhyankar sum5 -= v5[j] * tmp0 + v5[j+1] * tmp1; 2503019b515eSShri Abhyankar } 2504019b515eSShri Abhyankar if (j == nz-1){ 2505019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2506019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 2507019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 2508019b515eSShri Abhyankar sum3 -= v3[j] *tmp0; 2509019b515eSShri Abhyankar sum4 -= v4[j] *tmp0; 2510019b515eSShri Abhyankar sum5 -= v5[j] *tmp0; 2511019b515eSShri Abhyankar } 2512019b515eSShri Abhyankar 2513019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 2514019b515eSShri Abhyankar sum3 -= v3[nz] * sum1; 2515019b515eSShri Abhyankar sum4 -= v4[nz] * sum1; 2516019b515eSShri Abhyankar sum5 -= v5[nz] * sum1; 2517019b515eSShri Abhyankar sum3 -= v3[nz+1] * sum2; 2518019b515eSShri Abhyankar sum4 -= v4[nz+1] * sum2; 2519019b515eSShri Abhyankar sum5 -= v5[nz+1] * sum2; 2520019b515eSShri Abhyankar sum4 -= v4[nz+2] * sum3; 2521019b515eSShri Abhyankar sum5 -= v5[nz+2] * sum3; 2522019b515eSShri Abhyankar sum5 -= v5[nz+3] * sum4; 2523019b515eSShri Abhyankar 2524019b515eSShri Abhyankar tmp[row ++]=sum1; 2525019b515eSShri Abhyankar tmp[row ++]=sum2; 2526019b515eSShri Abhyankar tmp[row ++]=sum3; 2527019b515eSShri Abhyankar tmp[row ++]=sum4; 2528019b515eSShri Abhyankar tmp[row ++]=sum5; 2529019b515eSShri Abhyankar break; 2530019b515eSShri Abhyankar default: 2531e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n"); 2532019b515eSShri Abhyankar } 2533019b515eSShri Abhyankar } 2534019b515eSShri Abhyankar /* backward solve the upper triangular */ 2535019b515eSShri Abhyankar for (i=node_max -1 ,row = n-1 ; i>=0; i--){ 2536019b515eSShri Abhyankar nsz = ns[i]; 2537019b515eSShri Abhyankar aii = ad[row+1] + 1; 2538019b515eSShri Abhyankar v1 = aa + aii; 2539019b515eSShri Abhyankar vi = aj + aii; 2540019b515eSShri Abhyankar nz = ad[row]- ad[row+1] - 1; 254198991853SShri Abhyankar 254298991853SShri Abhyankar if (i > 0) { 254398991853SShri Abhyankar /* Prefetch the indices for the next block */ 254450d8bf02SJed Brown PetscPrefetchBlock(aj+ad[row-nsz+1]+1,ad[row-nsz]-ad[row-nsz+1],0,PETSC_PREFETCH_HINT_NTA); 254598991853SShri Abhyankar /* Prefetch the data for the next block */ 254650d8bf02SJed Brown PetscPrefetchBlock(aa+ad[row-nsz+1]+1,ad[row-nsz-ns[i-1]+1]-ad[row-nsz+1],0,PETSC_PREFETCH_HINT_NTA); 254798991853SShri Abhyankar } 254898991853SShri Abhyankar 2549019b515eSShri Abhyankar switch (nsz){ /* Each loop in 'case' is unrolled */ 2550019b515eSShri Abhyankar case 1 : 2551019b515eSShri Abhyankar sum1 = tmp[row]; 2552019b515eSShri Abhyankar 2553019b515eSShri Abhyankar for(j=0 ; j<nz-1; j+=2){ 2554019b515eSShri Abhyankar i0 = vi[j]; 2555019b515eSShri Abhyankar i1 = vi[j+1]; 2556019b515eSShri Abhyankar tmp0 = tmps[i0]; 2557019b515eSShri Abhyankar tmp1 = tmps[i1]; 2558019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2559019b515eSShri Abhyankar } 2560019b515eSShri Abhyankar if (j == nz-1){ 2561019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2562019b515eSShri Abhyankar sum1 -= v1[j]*tmp0; 2563019b515eSShri Abhyankar } 2564019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 2565019b515eSShri Abhyankar break; 2566019b515eSShri Abhyankar case 2 : 2567019b515eSShri Abhyankar sum1 = tmp[row]; 2568019b515eSShri Abhyankar sum2 = tmp[row-1]; 2569019b515eSShri Abhyankar v2 = aa + ad[row] + 1; 2570019b515eSShri Abhyankar for (j=0 ; j<nz-1; j+=2){ 2571019b515eSShri Abhyankar i0 = vi[j]; 2572019b515eSShri Abhyankar i1 = vi[j+1]; 2573019b515eSShri Abhyankar tmp0 = tmps[i0]; 2574019b515eSShri Abhyankar tmp1 = tmps[i1]; 2575019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2576019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 2577019b515eSShri Abhyankar } 2578019b515eSShri Abhyankar if (j == nz-1){ 2579019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2580019b515eSShri Abhyankar sum1 -= v1[j]* tmp0; 2581019b515eSShri Abhyankar sum2 -= v2[j+1]* tmp0; 2582019b515eSShri Abhyankar } 2583019b515eSShri Abhyankar 2584019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 2585019b515eSShri Abhyankar sum2 -= v2[0] * tmp0; 2586019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 2587019b515eSShri Abhyankar break; 2588019b515eSShri Abhyankar case 3 : 2589019b515eSShri Abhyankar sum1 = tmp[row]; 2590019b515eSShri Abhyankar sum2 = tmp[row -1]; 2591019b515eSShri Abhyankar sum3 = tmp[row -2]; 2592019b515eSShri Abhyankar v2 = aa + ad[row] + 1; 2593019b515eSShri Abhyankar v3 = aa + ad[row -1] + 1; 2594019b515eSShri Abhyankar for (j=0 ; j<nz-1; j+=2){ 2595019b515eSShri Abhyankar i0 = vi[j]; 2596019b515eSShri Abhyankar i1 = vi[j+1]; 2597019b515eSShri Abhyankar tmp0 = tmps[i0]; 2598019b515eSShri Abhyankar tmp1 = tmps[i1]; 2599019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2600019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 2601019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1; 2602019b515eSShri Abhyankar } 2603019b515eSShri Abhyankar if (j== nz-1){ 2604019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2605019b515eSShri Abhyankar sum1 -= v1[j] * tmp0; 2606019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0; 2607019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0; 2608019b515eSShri Abhyankar } 2609019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 2610019b515eSShri Abhyankar sum2 -= v2[0]* tmp0; 2611019b515eSShri Abhyankar sum3 -= v3[1] * tmp0; 2612019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 2613019b515eSShri Abhyankar sum3 -= v3[0]* tmp0; 2614019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--; 2615019b515eSShri Abhyankar 2616019b515eSShri Abhyankar break; 2617019b515eSShri Abhyankar case 4 : 2618019b515eSShri Abhyankar sum1 = tmp[row]; 2619019b515eSShri Abhyankar sum2 = tmp[row -1]; 2620019b515eSShri Abhyankar sum3 = tmp[row -2]; 2621019b515eSShri Abhyankar sum4 = tmp[row -3]; 2622019b515eSShri Abhyankar v2 = aa + ad[row]+1; 2623019b515eSShri Abhyankar v3 = aa + ad[row -1]+1; 2624019b515eSShri Abhyankar v4 = aa + ad[row -2]+1; 2625019b515eSShri Abhyankar 2626019b515eSShri Abhyankar for (j=0 ; j<nz-1; j+=2){ 2627019b515eSShri Abhyankar i0 = vi[j]; 2628019b515eSShri Abhyankar i1 = vi[j+1]; 2629019b515eSShri Abhyankar tmp0 = tmps[i0]; 2630019b515eSShri Abhyankar tmp1 = tmps[i1]; 2631019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2632019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 2633019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1; 2634019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1; 2635019b515eSShri Abhyankar } 2636019b515eSShri Abhyankar if (j== nz-1){ 2637019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2638019b515eSShri Abhyankar sum1 -= v1[j] * tmp0; 2639019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0; 2640019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0; 2641019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0; 2642019b515eSShri Abhyankar } 2643019b515eSShri Abhyankar 2644019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 2645019b515eSShri Abhyankar sum2 -= v2[0] * tmp0; 2646019b515eSShri Abhyankar sum3 -= v3[1] * tmp0; 2647019b515eSShri Abhyankar sum4 -= v4[2] * tmp0; 2648019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 2649019b515eSShri Abhyankar sum3 -= v3[0] * tmp0; 2650019b515eSShri Abhyankar sum4 -= v4[1] * tmp0; 2651019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--; 2652019b515eSShri Abhyankar sum4 -= v4[0] * tmp0; 2653019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--; 2654019b515eSShri Abhyankar break; 2655019b515eSShri Abhyankar case 5 : 2656019b515eSShri Abhyankar sum1 = tmp[row]; 2657019b515eSShri Abhyankar sum2 = tmp[row -1]; 2658019b515eSShri Abhyankar sum3 = tmp[row -2]; 2659019b515eSShri Abhyankar sum4 = tmp[row -3]; 2660019b515eSShri Abhyankar sum5 = tmp[row -4]; 2661019b515eSShri Abhyankar v2 = aa + ad[row]+1; 2662019b515eSShri Abhyankar v3 = aa + ad[row -1]+1; 2663019b515eSShri Abhyankar v4 = aa + ad[row -2]+1; 2664019b515eSShri Abhyankar v5 = aa + ad[row -3]+1; 2665019b515eSShri Abhyankar for (j=0 ; j<nz-1; j+=2){ 2666019b515eSShri Abhyankar i0 = vi[j]; 2667019b515eSShri Abhyankar i1 = vi[j+1]; 2668019b515eSShri Abhyankar tmp0 = tmps[i0]; 2669019b515eSShri Abhyankar tmp1 = tmps[i1]; 2670019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2671019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 2672019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1; 2673019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1; 2674019b515eSShri Abhyankar sum5 -= v5[j+4] * tmp0 + v5[j+5] * tmp1; 2675019b515eSShri Abhyankar } 2676019b515eSShri Abhyankar if (j==nz-1){ 2677019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2678019b515eSShri Abhyankar sum1 -= v1[j] * tmp0; 2679019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0; 2680019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0; 2681019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0; 2682019b515eSShri Abhyankar sum5 -= v5[j+4] * tmp0; 2683019b515eSShri Abhyankar } 2684019b515eSShri Abhyankar 2685019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 2686019b515eSShri Abhyankar sum2 -= v2[0] * tmp0; 2687019b515eSShri Abhyankar sum3 -= v3[1] * tmp0; 2688019b515eSShri Abhyankar sum4 -= v4[2] * tmp0; 2689019b515eSShri Abhyankar sum5 -= v5[3] * tmp0; 2690019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 2691019b515eSShri Abhyankar sum3 -= v3[0] * tmp0; 2692019b515eSShri Abhyankar sum4 -= v4[1] * tmp0; 2693019b515eSShri Abhyankar sum5 -= v5[2] * tmp0; 2694019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--; 2695019b515eSShri Abhyankar sum4 -= v4[0] * tmp0; 2696019b515eSShri Abhyankar sum5 -= v5[1] * tmp0; 2697019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--; 2698019b515eSShri Abhyankar sum5 -= v5[0] * tmp0; 2699019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum5*v5[nz+4]; row--; 2700019b515eSShri Abhyankar break; 2701019b515eSShri Abhyankar default: 2702e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n"); 2703019b515eSShri Abhyankar } 2704019b515eSShri Abhyankar } 2705019b515eSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 2706019b515eSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 27073649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 2708019b515eSShri Abhyankar ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2709019b515eSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 2710019b515eSShri Abhyankar PetscFunctionReturn(0); 2711019b515eSShri Abhyankar } 2712019b515eSShri Abhyankar 2713019b515eSShri Abhyankar 27144c1414c8SBarry Smith /* 27154c1414c8SBarry Smith Makes a longer coloring[] array and calls the usual code with that 27164c1414c8SBarry Smith */ 27174c1414c8SBarry Smith #undef __FUNCT__ 27184108e4d5SBarry Smith #define __FUNCT__ "MatColoringPatch_SeqAIJ_Inode" 27194108e4d5SBarry Smith PetscErrorCode MatColoringPatch_SeqAIJ_Inode(Mat mat,PetscInt ncolors,PetscInt nin,ISColoringValue coloring[],ISColoring *iscoloring) 27204c1414c8SBarry Smith { 27214c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)mat->data; 27224c1414c8SBarry Smith PetscErrorCode ierr; 2723d0f46423SBarry Smith PetscInt n = mat->cmap->n,m = a->inode.node_count,j,*ns = a->inode.size,row; 27244c1414c8SBarry Smith PetscInt *colorused,i; 27254c1414c8SBarry Smith ISColoringValue *newcolor; 27264c1414c8SBarry Smith 27274c1414c8SBarry Smith PetscFunctionBegin; 27284c1414c8SBarry Smith ierr = PetscMalloc((n+1)*sizeof(PetscInt),&newcolor);CHKERRQ(ierr); 27294c1414c8SBarry Smith /* loop over inodes, marking a color for each column*/ 27304c1414c8SBarry Smith row = 0; 27314c1414c8SBarry Smith for (i=0; i<m; i++){ 27324c1414c8SBarry Smith for (j=0; j<ns[i]; j++) { 27334c1414c8SBarry Smith newcolor[row++] = coloring[i] + j*ncolors; 27344c1414c8SBarry Smith } 27354c1414c8SBarry Smith } 27364c1414c8SBarry Smith 27374c1414c8SBarry Smith /* eliminate unneeded colors */ 27384c1414c8SBarry Smith ierr = PetscMalloc(5*ncolors*sizeof(PetscInt),&colorused);CHKERRQ(ierr); 27394c1414c8SBarry Smith ierr = PetscMemzero(colorused,5*ncolors*sizeof(PetscInt));CHKERRQ(ierr); 27404c1414c8SBarry Smith for (i=0; i<n; i++) { 27414c1414c8SBarry Smith colorused[newcolor[i]] = 1; 27424c1414c8SBarry Smith } 27434c1414c8SBarry Smith 27444c1414c8SBarry Smith for (i=1; i<5*ncolors; i++) { 27454c1414c8SBarry Smith colorused[i] += colorused[i-1]; 27464c1414c8SBarry Smith } 27474c1414c8SBarry Smith ncolors = colorused[5*ncolors-1]; 27484c1414c8SBarry Smith for (i=0; i<n; i++) { 274957815681SSatish Balay newcolor[i] = colorused[newcolor[i]]-1; 27504c1414c8SBarry Smith } 27514c1414c8SBarry Smith ierr = PetscFree(colorused);CHKERRQ(ierr); 27527adad957SLisandro Dalcin ierr = ISColoringCreate(((PetscObject)mat)->comm,ncolors,n,newcolor,iscoloring);CHKERRQ(ierr); 27534c1414c8SBarry Smith ierr = PetscFree(coloring);CHKERRQ(ierr); 27544c1414c8SBarry Smith PetscFunctionReturn(0); 27554c1414c8SBarry Smith } 27564c1414c8SBarry Smith 2757c6db04a5SJed Brown #include <../src/mat/blockinvert.h> 27582af78befSBarry Smith 27592af78befSBarry Smith #undef __FUNCT__ 27604108e4d5SBarry Smith #define __FUNCT__ "MatSOR_SeqAIJ_Inode" 27614108e4d5SBarry Smith PetscErrorCode MatSOR_SeqAIJ_Inode(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx) 27622af78befSBarry Smith { 27632af78befSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2764938d4eb3SBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5,tmp0,tmp1,tmp2,tmp3; 2765c0c53c30SBarry Smith MatScalar *ibdiag,*bdiag,work[25]; 2766a8b09249SBarry Smith PetscScalar *x,tmp4,tmp5,x1,x2,x3,x4,x5; 2767dd6ea824SBarry Smith const MatScalar *v = a->a,*v1,*v2,*v3,*v4,*v5; 2768a8b09249SBarry Smith const PetscScalar *xb; 276962bba022SBarry Smith PetscReal zeropivot = 1.0e-15, shift = 0.0; 27702af78befSBarry Smith PetscErrorCode ierr; 2771f0d39aaaSBarry Smith PetscInt n,m = a->inode.node_count,*sizes = a->inode.size,cnt = 0,i,j,row,i1,i2; 2772c0c53c30SBarry Smith PetscInt *idx,*diag = a->diag,*ii = a->i,sz,k,ipvt[5]; 27732af78befSBarry Smith 27742af78befSBarry Smith PetscFunctionBegin; 2775e32f2f54SBarry Smith if (omega != 1.0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for omega != 1.0; use -mat_no_inode"); 2776e32f2f54SBarry Smith if (fshift != 0.0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for fshift != 0.0; use -mat_no_inode"); 27773f6d85c9SBarry Smith if (its > 1) { 27783f6d85c9SBarry Smith /* switch to non-inode version */ 277941f059aeSBarry Smith ierr = MatSOR_SeqAIJ(A,bb,omega,flag,fshift,its,lits,xx);CHKERRQ(ierr); 27803f6d85c9SBarry Smith PetscFunctionReturn(0); 27813f6d85c9SBarry Smith } 27822af78befSBarry Smith 278371f1c65dSBarry Smith if (!a->inode.ibdiagvalid) { 27842af78befSBarry Smith if (!a->inode.ibdiag) { 27852af78befSBarry Smith /* calculate space needed for diagonal blocks */ 27862af78befSBarry Smith for (i=0; i<m; i++) { 27872af78befSBarry Smith cnt += sizes[i]*sizes[i]; 27882af78befSBarry Smith } 2789f0d39aaaSBarry Smith a->inode.bdiagsize = cnt; 279089c6957cSBarry Smith ierr = PetscMalloc3(cnt,MatScalar,&a->inode.ibdiag,cnt,MatScalar,&a->inode.bdiag,A->rmap->n,MatScalar,&a->inode.ssor_work);CHKERRQ(ierr); 279171f1c65dSBarry Smith } 279271f1c65dSBarry Smith 279371f1c65dSBarry Smith /* copy over the diagonal blocks and invert them */ 27942af78befSBarry Smith ibdiag = a->inode.ibdiag; 27952af78befSBarry Smith bdiag = a->inode.bdiag; 27962af78befSBarry Smith cnt = 0; 27972af78befSBarry Smith for (i=0, row = 0; i<m; i++) { 27982af78befSBarry Smith for (j=0; j<sizes[i]; j++) { 2799f0d39aaaSBarry Smith for (k=0; k<sizes[i]; k++) { 2800f0d39aaaSBarry Smith bdiag[cnt+k*sizes[i]+j] = v[diag[row+j] - j + k]; 2801f0d39aaaSBarry Smith } 28022af78befSBarry Smith } 2803a77337e4SBarry Smith ierr = PetscMemcpy(ibdiag+cnt,bdiag+cnt,sizes[i]*sizes[i]*sizeof(MatScalar));CHKERRQ(ierr); 28042af78befSBarry Smith 28052af78befSBarry Smith switch(sizes[i]) { 28062af78befSBarry Smith case 1: 28072af78befSBarry Smith /* Create matrix data structure */ 2808e32f2f54SBarry Smith if (PetscAbsScalar(ibdiag[cnt]) < zeropivot) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot on row %D",row); 280964c62002SMatthew Knepley ibdiag[cnt] = 1.0/ibdiag[cnt]; 28102af78befSBarry Smith break; 28112af78befSBarry Smith case 2: 281262bba022SBarry Smith ierr = Kernel_A_gets_inverse_A_2(ibdiag+cnt,shift);CHKERRQ(ierr); 28132af78befSBarry Smith break; 28142af78befSBarry Smith case 3: 281562bba022SBarry Smith ierr = Kernel_A_gets_inverse_A_3(ibdiag+cnt,shift);CHKERRQ(ierr); 28162af78befSBarry Smith break; 28172af78befSBarry Smith case 4: 281862bba022SBarry Smith ierr = Kernel_A_gets_inverse_A_4(ibdiag+cnt,shift);CHKERRQ(ierr); 28192af78befSBarry Smith break; 28202af78befSBarry Smith case 5: 2821c0c53c30SBarry Smith ierr = Kernel_A_gets_inverse_A_5(ibdiag+cnt,ipvt,work,shift);CHKERRQ(ierr); 28222af78befSBarry Smith break; 28232af78befSBarry Smith default: 2824e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 28252af78befSBarry Smith } 28262af78befSBarry Smith cnt += sizes[i]*sizes[i]; 28272af78befSBarry Smith row += sizes[i]; 28282af78befSBarry Smith } 282971f1c65dSBarry Smith a->inode.ibdiagvalid = PETSC_TRUE; 28302af78befSBarry Smith } 28312af78befSBarry Smith ibdiag = a->inode.ibdiag; 28322af78befSBarry Smith bdiag = a->inode.bdiag; 28332af78befSBarry Smith 2834938d4eb3SBarry Smith 2835938d4eb3SBarry Smith /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */ 2836938d4eb3SBarry Smith if (flag & SOR_ZERO_INITIAL_GUESS) { 28373649974fSBarry Smith const PetscScalar *b; 28382af78befSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 28393649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 28402af78befSBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){ 28412af78befSBarry Smith 28428862d2efSBarry Smith for (i=0, row=0; i<m; i++) { 28438862d2efSBarry Smith sz = diag[row] - ii[row]; 28448862d2efSBarry Smith v1 = a->a + ii[row]; 28458862d2efSBarry Smith idx = a->j + ii[row]; 28468862d2efSBarry Smith 28474108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 28488862d2efSBarry Smith switch (sizes[i]){ 28498862d2efSBarry Smith case 1: 28508862d2efSBarry Smith 28518862d2efSBarry Smith sum1 = b[row]; 28528862d2efSBarry Smith for(n = 0; n<sz-1; n+=2) { 28538862d2efSBarry Smith i1 = idx[0]; 28548862d2efSBarry Smith i2 = idx[1]; 28558862d2efSBarry Smith idx += 2; 28568862d2efSBarry Smith tmp0 = x[i1]; 28578862d2efSBarry Smith tmp1 = x[i2]; 28588862d2efSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 28598862d2efSBarry Smith } 28608862d2efSBarry Smith 28618862d2efSBarry Smith if (n == sz-1){ 2862f0d39aaaSBarry Smith tmp0 = x[*idx]; 2863f0d39aaaSBarry Smith sum1 -= *v1 * tmp0; 28648862d2efSBarry Smith } 28658862d2efSBarry Smith x[row++] = sum1*(*ibdiag++); 28668862d2efSBarry Smith break; 2867f0d39aaaSBarry Smith case 2: 2868f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 2869f0d39aaaSBarry Smith sum1 = b[row]; 2870f0d39aaaSBarry Smith sum2 = b[row+1]; 2871f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 2872f0d39aaaSBarry Smith i1 = idx[0]; 2873f0d39aaaSBarry Smith i2 = idx[1]; 2874f0d39aaaSBarry Smith idx += 2; 2875f0d39aaaSBarry Smith tmp0 = x[i1]; 2876f0d39aaaSBarry Smith tmp1 = x[i2]; 2877f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 2878f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 2879f0d39aaaSBarry Smith } 2880f0d39aaaSBarry Smith 2881f0d39aaaSBarry Smith if (n == sz-1){ 2882f0d39aaaSBarry Smith tmp0 = x[*idx]; 2883f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 2884f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 2885f0d39aaaSBarry Smith } 2886f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[2]; 2887f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[3]; 2888f0d39aaaSBarry Smith ibdiag += 4; 2889f0d39aaaSBarry Smith break; 2890f0d39aaaSBarry Smith case 3: 2891f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 2892f0d39aaaSBarry Smith v3 = a->a + ii[row+2]; 2893f0d39aaaSBarry Smith sum1 = b[row]; 2894f0d39aaaSBarry Smith sum2 = b[row+1]; 2895f0d39aaaSBarry Smith sum3 = b[row+2]; 2896f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 2897f0d39aaaSBarry Smith i1 = idx[0]; 2898f0d39aaaSBarry Smith i2 = idx[1]; 2899f0d39aaaSBarry Smith idx += 2; 2900f0d39aaaSBarry Smith tmp0 = x[i1]; 2901f0d39aaaSBarry Smith tmp1 = x[i2]; 2902f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 2903f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 2904f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 2905f0d39aaaSBarry Smith } 2906f0d39aaaSBarry Smith 2907f0d39aaaSBarry Smith if (n == sz-1){ 2908f0d39aaaSBarry Smith tmp0 = x[*idx]; 2909f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 2910f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 2911f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0; 2912f0d39aaaSBarry Smith } 2913f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6]; 2914f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7]; 2915f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8]; 2916f0d39aaaSBarry Smith ibdiag += 9; 2917f0d39aaaSBarry Smith break; 2918f0d39aaaSBarry Smith case 4: 2919f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 2920f0d39aaaSBarry Smith v3 = a->a + ii[row+2]; 2921f0d39aaaSBarry Smith v4 = a->a + ii[row+3]; 2922f0d39aaaSBarry Smith sum1 = b[row]; 2923f0d39aaaSBarry Smith sum2 = b[row+1]; 2924f0d39aaaSBarry Smith sum3 = b[row+2]; 2925f0d39aaaSBarry Smith sum4 = b[row+3]; 2926f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 2927f0d39aaaSBarry Smith i1 = idx[0]; 2928f0d39aaaSBarry Smith i2 = idx[1]; 2929f0d39aaaSBarry Smith idx += 2; 2930f0d39aaaSBarry Smith tmp0 = x[i1]; 2931f0d39aaaSBarry Smith tmp1 = x[i2]; 2932f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 2933f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 2934f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 2935f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 2936f0d39aaaSBarry Smith } 2937f0d39aaaSBarry Smith 2938f0d39aaaSBarry Smith if (n == sz-1){ 2939f0d39aaaSBarry Smith tmp0 = x[*idx]; 2940f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 2941f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 2942f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0; 2943f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0; 2944f0d39aaaSBarry Smith } 2945f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12]; 2946f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13]; 2947f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14]; 2948f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15]; 2949f0d39aaaSBarry Smith ibdiag += 16; 2950f0d39aaaSBarry Smith break; 2951f0d39aaaSBarry Smith case 5: 2952f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 2953f0d39aaaSBarry Smith v3 = a->a + ii[row+2]; 2954f0d39aaaSBarry Smith v4 = a->a + ii[row+3]; 2955f0d39aaaSBarry Smith v5 = a->a + ii[row+4]; 2956f0d39aaaSBarry Smith sum1 = b[row]; 2957f0d39aaaSBarry Smith sum2 = b[row+1]; 2958f0d39aaaSBarry Smith sum3 = b[row+2]; 2959f0d39aaaSBarry Smith sum4 = b[row+3]; 2960f0d39aaaSBarry Smith sum5 = b[row+4]; 2961f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 2962f0d39aaaSBarry Smith i1 = idx[0]; 2963f0d39aaaSBarry Smith i2 = idx[1]; 2964f0d39aaaSBarry Smith idx += 2; 2965f0d39aaaSBarry Smith tmp0 = x[i1]; 2966f0d39aaaSBarry Smith tmp1 = x[i2]; 2967f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 2968f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 2969f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 2970f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 2971f0d39aaaSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 2972f0d39aaaSBarry Smith } 2973f0d39aaaSBarry Smith 2974f0d39aaaSBarry Smith if (n == sz-1){ 2975f0d39aaaSBarry Smith tmp0 = x[*idx]; 2976f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 2977f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 2978f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0; 2979f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0; 2980f0d39aaaSBarry Smith sum5 -= v5[0] * tmp0; 2981f0d39aaaSBarry Smith } 2982f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20]; 2983f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21]; 2984f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22]; 2985f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23]; 2986f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24]; 2987f0d39aaaSBarry Smith ibdiag += 25; 2988f0d39aaaSBarry Smith break; 29898862d2efSBarry Smith default: 2990e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 29918862d2efSBarry Smith } 29922af78befSBarry Smith } 29932af78befSBarry Smith 29942af78befSBarry Smith xb = x; 29952af78befSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 29962af78befSBarry Smith } else xb = b; 29972af78befSBarry Smith if ((flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) && 29982af78befSBarry Smith (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP)) { 29998862d2efSBarry Smith cnt = 0; 30008862d2efSBarry Smith for (i=0, row=0; i<m; i++) { 30012af78befSBarry Smith 30028862d2efSBarry Smith switch (sizes[i]){ 30038862d2efSBarry Smith case 1: 30048862d2efSBarry Smith x[row++] *= bdiag[cnt++]; 30058862d2efSBarry Smith break; 3006f0d39aaaSBarry Smith case 2: 3007f0d39aaaSBarry Smith x1 = x[row]; x2 = x[row+1]; 3008f0d39aaaSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2]; 3009f0d39aaaSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3]; 3010f0d39aaaSBarry Smith x[row++] = tmp1; 3011f0d39aaaSBarry Smith x[row++] = tmp2; 3012f0d39aaaSBarry Smith cnt += 4; 3013f0d39aaaSBarry Smith break; 3014f0d39aaaSBarry Smith case 3: 3015f0d39aaaSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; 3016f0d39aaaSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6]; 3017f0d39aaaSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7]; 3018f0d39aaaSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8]; 3019f0d39aaaSBarry Smith x[row++] = tmp1; 3020f0d39aaaSBarry Smith x[row++] = tmp2; 3021f0d39aaaSBarry Smith x[row++] = tmp3; 3022f0d39aaaSBarry Smith cnt += 9; 3023f0d39aaaSBarry Smith break; 3024f0d39aaaSBarry Smith case 4: 3025f0d39aaaSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; 3026f0d39aaaSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12]; 3027f0d39aaaSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13]; 3028f0d39aaaSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14]; 3029f0d39aaaSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15]; 3030f0d39aaaSBarry Smith x[row++] = tmp1; 3031f0d39aaaSBarry Smith x[row++] = tmp2; 3032f0d39aaaSBarry Smith x[row++] = tmp3; 3033f0d39aaaSBarry Smith x[row++] = tmp4; 3034f0d39aaaSBarry Smith cnt += 16; 3035f0d39aaaSBarry Smith break; 3036f0d39aaaSBarry Smith case 5: 3037f0d39aaaSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; x5 = x[row+4]; 3038f0d39aaaSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20]; 3039f0d39aaaSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21]; 3040f0d39aaaSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22]; 3041f0d39aaaSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23]; 3042f0d39aaaSBarry Smith tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24]; 3043f0d39aaaSBarry Smith x[row++] = tmp1; 3044f0d39aaaSBarry Smith x[row++] = tmp2; 3045f0d39aaaSBarry Smith x[row++] = tmp3; 3046f0d39aaaSBarry Smith x[row++] = tmp4; 3047f0d39aaaSBarry Smith x[row++] = tmp5; 3048f0d39aaaSBarry Smith cnt += 25; 3049f0d39aaaSBarry Smith break; 30508862d2efSBarry Smith default: 3051e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 30528862d2efSBarry Smith } 30532af78befSBarry Smith } 30542af78befSBarry Smith ierr = PetscLogFlops(m);CHKERRQ(ierr); 30552af78befSBarry Smith } 30562af78befSBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){ 30572af78befSBarry Smith 3058f0d39aaaSBarry Smith ibdiag = a->inode.ibdiag+a->inode.bdiagsize; 3059d0f46423SBarry Smith for (i=m-1, row=A->rmap->n-1; i>=0; i--) { 3060f0d39aaaSBarry Smith ibdiag -= sizes[i]*sizes[i]; 30618862d2efSBarry Smith sz = ii[row+1] - diag[row] - 1; 30628862d2efSBarry Smith v1 = a->a + diag[row] + 1; 30638862d2efSBarry Smith idx = a->j + diag[row] + 1; 30642af78befSBarry Smith 30654108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 30668862d2efSBarry Smith switch (sizes[i]){ 30678862d2efSBarry Smith case 1: 30688862d2efSBarry Smith 30698862d2efSBarry Smith sum1 = xb[row]; 30708862d2efSBarry Smith for(n = 0; n<sz-1; n+=2) { 30718862d2efSBarry Smith i1 = idx[0]; 30728862d2efSBarry Smith i2 = idx[1]; 30738862d2efSBarry Smith idx += 2; 30748862d2efSBarry Smith tmp0 = x[i1]; 30758862d2efSBarry Smith tmp1 = x[i2]; 30768862d2efSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 30778862d2efSBarry Smith } 30788862d2efSBarry Smith 30798862d2efSBarry Smith if (n == sz-1){ 3080f0d39aaaSBarry Smith tmp0 = x[*idx]; 3081f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 30828862d2efSBarry Smith } 3083f0d39aaaSBarry Smith x[row--] = sum1*(*ibdiag); 3084f0d39aaaSBarry Smith break; 3085f0d39aaaSBarry Smith 3086f0d39aaaSBarry Smith case 2: 3087f0d39aaaSBarry Smith 3088f0d39aaaSBarry Smith sum1 = xb[row]; 3089f0d39aaaSBarry Smith sum2 = xb[row-1]; 3090f0d39aaaSBarry Smith /* note that sum1 is associated with the second of the two rows */ 3091f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 3092f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 3093f0d39aaaSBarry Smith i1 = idx[0]; 3094f0d39aaaSBarry Smith i2 = idx[1]; 3095f0d39aaaSBarry Smith idx += 2; 3096f0d39aaaSBarry Smith tmp0 = x[i1]; 3097f0d39aaaSBarry Smith tmp1 = x[i2]; 3098f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3099f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3100f0d39aaaSBarry Smith } 3101f0d39aaaSBarry Smith 3102f0d39aaaSBarry Smith if (n == sz-1){ 3103f0d39aaaSBarry Smith tmp0 = x[*idx]; 3104f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 3105f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 3106f0d39aaaSBarry Smith } 3107f0d39aaaSBarry Smith x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3]; 3108f0d39aaaSBarry Smith x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2]; 3109f0d39aaaSBarry Smith break; 3110f0d39aaaSBarry Smith case 3: 3111f0d39aaaSBarry Smith 3112f0d39aaaSBarry Smith sum1 = xb[row]; 3113f0d39aaaSBarry Smith sum2 = xb[row-1]; 3114f0d39aaaSBarry Smith sum3 = xb[row-2]; 3115f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 3116f0d39aaaSBarry Smith v3 = a->a + diag[row-2] + 3; 3117f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 3118f0d39aaaSBarry Smith i1 = idx[0]; 3119f0d39aaaSBarry Smith i2 = idx[1]; 3120f0d39aaaSBarry Smith idx += 2; 3121f0d39aaaSBarry Smith tmp0 = x[i1]; 3122f0d39aaaSBarry Smith tmp1 = x[i2]; 3123f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3124f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3125f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3126f0d39aaaSBarry Smith } 3127f0d39aaaSBarry Smith 3128f0d39aaaSBarry Smith if (n == sz-1){ 3129f0d39aaaSBarry Smith tmp0 = x[*idx]; 3130f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 3131f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 3132f0d39aaaSBarry Smith sum3 -= *v3*tmp0; 3133f0d39aaaSBarry Smith } 3134f0d39aaaSBarry Smith x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8]; 3135f0d39aaaSBarry Smith x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7]; 3136f0d39aaaSBarry Smith x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6]; 3137f0d39aaaSBarry Smith break; 3138f0d39aaaSBarry Smith case 4: 3139f0d39aaaSBarry Smith 3140f0d39aaaSBarry Smith sum1 = xb[row]; 3141f0d39aaaSBarry Smith sum2 = xb[row-1]; 3142f0d39aaaSBarry Smith sum3 = xb[row-2]; 3143f0d39aaaSBarry Smith sum4 = xb[row-3]; 3144f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 3145f0d39aaaSBarry Smith v3 = a->a + diag[row-2] + 3; 3146f0d39aaaSBarry Smith v4 = a->a + diag[row-3] + 4; 3147f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 3148f0d39aaaSBarry Smith i1 = idx[0]; 3149f0d39aaaSBarry Smith i2 = idx[1]; 3150f0d39aaaSBarry Smith idx += 2; 3151f0d39aaaSBarry Smith tmp0 = x[i1]; 3152f0d39aaaSBarry Smith tmp1 = x[i2]; 3153f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3154f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3155f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3156f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 3157f0d39aaaSBarry Smith } 3158f0d39aaaSBarry Smith 3159f0d39aaaSBarry Smith if (n == sz-1){ 3160f0d39aaaSBarry Smith tmp0 = x[*idx]; 3161f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 3162f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 3163f0d39aaaSBarry Smith sum3 -= *v3*tmp0; 3164f0d39aaaSBarry Smith sum4 -= *v4*tmp0; 3165f0d39aaaSBarry Smith } 3166f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15]; 3167f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14]; 3168f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13]; 3169f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12]; 3170f0d39aaaSBarry Smith break; 3171f0d39aaaSBarry Smith case 5: 3172f0d39aaaSBarry Smith 3173f0d39aaaSBarry Smith sum1 = xb[row]; 3174f0d39aaaSBarry Smith sum2 = xb[row-1]; 3175f0d39aaaSBarry Smith sum3 = xb[row-2]; 3176f0d39aaaSBarry Smith sum4 = xb[row-3]; 3177f0d39aaaSBarry Smith sum5 = xb[row-4]; 3178f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 3179f0d39aaaSBarry Smith v3 = a->a + diag[row-2] + 3; 3180f0d39aaaSBarry Smith v4 = a->a + diag[row-3] + 4; 3181f0d39aaaSBarry Smith v5 = a->a + diag[row-4] + 5; 3182f0d39aaaSBarry Smith for(n = 0; n<sz-1; n+=2) { 3183f0d39aaaSBarry Smith i1 = idx[0]; 3184f0d39aaaSBarry Smith i2 = idx[1]; 3185f0d39aaaSBarry Smith idx += 2; 3186f0d39aaaSBarry Smith tmp0 = x[i1]; 3187f0d39aaaSBarry Smith tmp1 = x[i2]; 3188f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3189f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3190f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3191f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 3192f0d39aaaSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 3193f0d39aaaSBarry Smith } 3194f0d39aaaSBarry Smith 3195f0d39aaaSBarry Smith if (n == sz-1){ 3196f0d39aaaSBarry Smith tmp0 = x[*idx]; 3197f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 3198f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 3199f0d39aaaSBarry Smith sum3 -= *v3*tmp0; 3200f0d39aaaSBarry Smith sum4 -= *v4*tmp0; 3201f0d39aaaSBarry Smith sum5 -= *v5*tmp0; 3202f0d39aaaSBarry Smith } 3203f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24]; 3204f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23]; 3205f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22]; 3206f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21]; 3207f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20]; 32088862d2efSBarry Smith break; 32098862d2efSBarry Smith default: 3210e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 32118862d2efSBarry Smith } 32122af78befSBarry Smith } 32132af78befSBarry Smith 32142af78befSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 32152af78befSBarry Smith } 32162af78befSBarry Smith its--; 3217938d4eb3SBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 32183649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 32192af78befSBarry Smith } 322089c6957cSBarry Smith if (flag & SOR_EISENSTAT) { 3221938d4eb3SBarry Smith const PetscScalar *b; 32226cf45366SBarry Smith MatScalar *t = a->inode.ssor_work; 32236cf45366SBarry Smith 3224938d4eb3SBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 32253649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 322689c6957cSBarry Smith /* 322789c6957cSBarry Smith Apply (U + D)^-1 where D is now the block diagonal 322889c6957cSBarry Smith */ 322989c6957cSBarry Smith ibdiag = a->inode.ibdiag+a->inode.bdiagsize; 323089c6957cSBarry Smith for (i=m-1, row=A->rmap->n-1; i>=0; i--) { 323189c6957cSBarry Smith ibdiag -= sizes[i]*sizes[i]; 323289c6957cSBarry Smith sz = ii[row+1] - diag[row] - 1; 323389c6957cSBarry Smith v1 = a->a + diag[row] + 1; 323489c6957cSBarry Smith idx = a->j + diag[row] + 1; 32354108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 323689c6957cSBarry Smith switch (sizes[i]){ 323789c6957cSBarry Smith case 1: 323889c6957cSBarry Smith 323989c6957cSBarry Smith sum1 = b[row]; 324089c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 324189c6957cSBarry Smith i1 = idx[0]; 324289c6957cSBarry Smith i2 = idx[1]; 324389c6957cSBarry Smith idx += 2; 324489c6957cSBarry Smith tmp0 = x[i1]; 324589c6957cSBarry Smith tmp1 = x[i2]; 324689c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 324789c6957cSBarry Smith } 324889c6957cSBarry Smith 324989c6957cSBarry Smith if (n == sz-1){ 325089c6957cSBarry Smith tmp0 = x[*idx]; 325189c6957cSBarry Smith sum1 -= *v1*tmp0; 325289c6957cSBarry Smith } 3253938d4eb3SBarry Smith x[row] = sum1*(*ibdiag);row--; 325489c6957cSBarry Smith break; 325589c6957cSBarry Smith 325689c6957cSBarry Smith case 2: 325789c6957cSBarry Smith 325889c6957cSBarry Smith sum1 = b[row]; 325989c6957cSBarry Smith sum2 = b[row-1]; 326089c6957cSBarry Smith /* note that sum1 is associated with the second of the two rows */ 326189c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 326289c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 326389c6957cSBarry Smith i1 = idx[0]; 326489c6957cSBarry Smith i2 = idx[1]; 326589c6957cSBarry Smith idx += 2; 326689c6957cSBarry Smith tmp0 = x[i1]; 326789c6957cSBarry Smith tmp1 = x[i2]; 326889c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 326989c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 327089c6957cSBarry Smith } 327189c6957cSBarry Smith 327289c6957cSBarry Smith if (n == sz-1){ 327389c6957cSBarry Smith tmp0 = x[*idx]; 327489c6957cSBarry Smith sum1 -= *v1*tmp0; 327589c6957cSBarry Smith sum2 -= *v2*tmp0; 327689c6957cSBarry Smith } 3277938d4eb3SBarry Smith x[row] = sum2*ibdiag[1] + sum1*ibdiag[3]; 3278938d4eb3SBarry Smith x[row-1] = sum2*ibdiag[0] + sum1*ibdiag[2]; 3279938d4eb3SBarry Smith row -= 2; 328089c6957cSBarry Smith break; 328189c6957cSBarry Smith case 3: 328289c6957cSBarry Smith 328389c6957cSBarry Smith sum1 = b[row]; 328489c6957cSBarry Smith sum2 = b[row-1]; 328589c6957cSBarry Smith sum3 = b[row-2]; 328689c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 328789c6957cSBarry Smith v3 = a->a + diag[row-2] + 3; 328889c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 328989c6957cSBarry Smith i1 = idx[0]; 329089c6957cSBarry Smith i2 = idx[1]; 329189c6957cSBarry Smith idx += 2; 329289c6957cSBarry Smith tmp0 = x[i1]; 329389c6957cSBarry Smith tmp1 = x[i2]; 329489c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 329589c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 329689c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 329789c6957cSBarry Smith } 329889c6957cSBarry Smith 329989c6957cSBarry Smith if (n == sz-1){ 330089c6957cSBarry Smith tmp0 = x[*idx]; 330189c6957cSBarry Smith sum1 -= *v1*tmp0; 330289c6957cSBarry Smith sum2 -= *v2*tmp0; 330389c6957cSBarry Smith sum3 -= *v3*tmp0; 330489c6957cSBarry Smith } 3305938d4eb3SBarry Smith x[row] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8]; 3306938d4eb3SBarry Smith x[row-1] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7]; 3307938d4eb3SBarry Smith x[row-2] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6]; 3308938d4eb3SBarry Smith row -= 3; 330989c6957cSBarry Smith break; 331089c6957cSBarry Smith case 4: 331189c6957cSBarry Smith 331289c6957cSBarry Smith sum1 = b[row]; 331389c6957cSBarry Smith sum2 = b[row-1]; 331489c6957cSBarry Smith sum3 = b[row-2]; 331589c6957cSBarry Smith sum4 = b[row-3]; 331689c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 331789c6957cSBarry Smith v3 = a->a + diag[row-2] + 3; 331889c6957cSBarry Smith v4 = a->a + diag[row-3] + 4; 331989c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 332089c6957cSBarry Smith i1 = idx[0]; 332189c6957cSBarry Smith i2 = idx[1]; 332289c6957cSBarry Smith idx += 2; 332389c6957cSBarry Smith tmp0 = x[i1]; 332489c6957cSBarry Smith tmp1 = x[i2]; 332589c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 332689c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 332789c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 332889c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 332989c6957cSBarry Smith } 333089c6957cSBarry Smith 333189c6957cSBarry Smith if (n == sz-1){ 333289c6957cSBarry Smith tmp0 = x[*idx]; 333389c6957cSBarry Smith sum1 -= *v1*tmp0; 333489c6957cSBarry Smith sum2 -= *v2*tmp0; 333589c6957cSBarry Smith sum3 -= *v3*tmp0; 333689c6957cSBarry Smith sum4 -= *v4*tmp0; 333789c6957cSBarry Smith } 3338938d4eb3SBarry Smith x[row] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15]; 3339938d4eb3SBarry Smith x[row-1] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14]; 3340938d4eb3SBarry Smith x[row-2] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13]; 3341938d4eb3SBarry Smith x[row-3] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12]; 3342938d4eb3SBarry Smith row -= 4; 334389c6957cSBarry Smith break; 334489c6957cSBarry Smith case 5: 334589c6957cSBarry Smith 334689c6957cSBarry Smith sum1 = b[row]; 334789c6957cSBarry Smith sum2 = b[row-1]; 334889c6957cSBarry Smith sum3 = b[row-2]; 334989c6957cSBarry Smith sum4 = b[row-3]; 335089c6957cSBarry Smith sum5 = b[row-4]; 335189c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 335289c6957cSBarry Smith v3 = a->a + diag[row-2] + 3; 335389c6957cSBarry Smith v4 = a->a + diag[row-3] + 4; 335489c6957cSBarry Smith v5 = a->a + diag[row-4] + 5; 335589c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 335689c6957cSBarry Smith i1 = idx[0]; 335789c6957cSBarry Smith i2 = idx[1]; 335889c6957cSBarry Smith idx += 2; 335989c6957cSBarry Smith tmp0 = x[i1]; 336089c6957cSBarry Smith tmp1 = x[i2]; 336189c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 336289c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 336389c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 336489c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 336589c6957cSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 336689c6957cSBarry Smith } 336789c6957cSBarry Smith 336889c6957cSBarry Smith if (n == sz-1){ 336989c6957cSBarry Smith tmp0 = x[*idx]; 337089c6957cSBarry Smith sum1 -= *v1*tmp0; 337189c6957cSBarry Smith sum2 -= *v2*tmp0; 337289c6957cSBarry Smith sum3 -= *v3*tmp0; 337389c6957cSBarry Smith sum4 -= *v4*tmp0; 337489c6957cSBarry Smith sum5 -= *v5*tmp0; 337589c6957cSBarry Smith } 3376938d4eb3SBarry Smith x[row] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24]; 3377938d4eb3SBarry Smith x[row-1] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23]; 3378938d4eb3SBarry Smith x[row-2] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22]; 3379938d4eb3SBarry Smith x[row-3] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21]; 3380938d4eb3SBarry Smith x[row-4] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20]; 3381938d4eb3SBarry Smith row -= 5; 338289c6957cSBarry Smith break; 338389c6957cSBarry Smith default: 3384e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 338589c6957cSBarry Smith } 338689c6957cSBarry Smith } 338789c6957cSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 338889c6957cSBarry Smith 338989c6957cSBarry Smith /* 339089c6957cSBarry Smith t = b - D x where D is the block diagonal 339189c6957cSBarry Smith */ 339289c6957cSBarry Smith cnt = 0; 339389c6957cSBarry Smith for (i=0, row=0; i<m; i++) { 339489c6957cSBarry Smith switch (sizes[i]){ 339589c6957cSBarry Smith case 1: 339689c6957cSBarry Smith t[row] = b[row] - bdiag[cnt++]*x[row]; row++; 339789c6957cSBarry Smith break; 339889c6957cSBarry Smith case 2: 339989c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; 340089c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2]; 340189c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3]; 340289c6957cSBarry Smith t[row] = b[row] - tmp1; 340389c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; row += 2; 340489c6957cSBarry Smith cnt += 4; 340589c6957cSBarry Smith break; 340689c6957cSBarry Smith case 3: 340789c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; 340889c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6]; 340989c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7]; 341089c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8]; 341189c6957cSBarry Smith t[row] = b[row] - tmp1; 341289c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; 341389c6957cSBarry Smith t[row+2] = b[row+2] - tmp3; row += 3; 341489c6957cSBarry Smith cnt += 9; 341589c6957cSBarry Smith break; 341689c6957cSBarry Smith case 4: 341789c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; 341889c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12]; 341989c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13]; 342089c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14]; 342189c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15]; 342289c6957cSBarry Smith t[row] = b[row] - tmp1; 342389c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; 342489c6957cSBarry Smith t[row+2] = b[row+2] - tmp3; 342589c6957cSBarry Smith t[row+3] = b[row+3] - tmp4; row += 4; 342689c6957cSBarry Smith cnt += 16; 342789c6957cSBarry Smith break; 342889c6957cSBarry Smith case 5: 342989c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; x5 = x[row+4]; 343089c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20]; 343189c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21]; 343289c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22]; 343389c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23]; 343489c6957cSBarry Smith tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24]; 343589c6957cSBarry Smith t[row] = b[row] - tmp1; 343689c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; 343789c6957cSBarry Smith t[row+2] = b[row+2] - tmp3; 343889c6957cSBarry Smith t[row+3] = b[row+3] - tmp4; 343989c6957cSBarry Smith t[row+4] = b[row+4] - tmp5;row += 5; 344089c6957cSBarry Smith cnt += 25; 344189c6957cSBarry Smith break; 344289c6957cSBarry Smith default: 3443e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 344489c6957cSBarry Smith } 344589c6957cSBarry Smith } 344689c6957cSBarry Smith ierr = PetscLogFlops(m);CHKERRQ(ierr); 344789c6957cSBarry Smith 344889c6957cSBarry Smith 344989c6957cSBarry Smith 345089c6957cSBarry Smith /* 345189c6957cSBarry Smith Apply (L + D)^-1 where D is the block diagonal 345289c6957cSBarry Smith */ 345389c6957cSBarry Smith for (i=0, row=0; i<m; i++) { 345489c6957cSBarry Smith sz = diag[row] - ii[row]; 345589c6957cSBarry Smith v1 = a->a + ii[row]; 345689c6957cSBarry Smith idx = a->j + ii[row]; 34574108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 345889c6957cSBarry Smith switch (sizes[i]){ 345989c6957cSBarry Smith case 1: 346089c6957cSBarry Smith 346189c6957cSBarry Smith sum1 = t[row]; 346289c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 346389c6957cSBarry Smith i1 = idx[0]; 346489c6957cSBarry Smith i2 = idx[1]; 346589c6957cSBarry Smith idx += 2; 346689c6957cSBarry Smith tmp0 = t[i1]; 346789c6957cSBarry Smith tmp1 = t[i2]; 346889c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 346989c6957cSBarry Smith } 347089c6957cSBarry Smith 347189c6957cSBarry Smith if (n == sz-1){ 347289c6957cSBarry Smith tmp0 = t[*idx]; 347389c6957cSBarry Smith sum1 -= *v1 * tmp0; 347489c6957cSBarry Smith } 347589c6957cSBarry Smith x[row] += t[row] = sum1*(*ibdiag++); row++; 347689c6957cSBarry Smith break; 347789c6957cSBarry Smith case 2: 347889c6957cSBarry Smith v2 = a->a + ii[row+1]; 347989c6957cSBarry Smith sum1 = t[row]; 348089c6957cSBarry Smith sum2 = t[row+1]; 348189c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 348289c6957cSBarry Smith i1 = idx[0]; 348389c6957cSBarry Smith i2 = idx[1]; 348489c6957cSBarry Smith idx += 2; 348589c6957cSBarry Smith tmp0 = t[i1]; 348689c6957cSBarry Smith tmp1 = t[i2]; 348789c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 348889c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 348989c6957cSBarry Smith } 349089c6957cSBarry Smith 349189c6957cSBarry Smith if (n == sz-1){ 349289c6957cSBarry Smith tmp0 = t[*idx]; 349389c6957cSBarry Smith sum1 -= v1[0] * tmp0; 349489c6957cSBarry Smith sum2 -= v2[0] * tmp0; 349589c6957cSBarry Smith } 349689c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[2]; 349789c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3]; 349889c6957cSBarry Smith ibdiag += 4; row += 2; 349989c6957cSBarry Smith break; 350089c6957cSBarry Smith case 3: 350189c6957cSBarry Smith v2 = a->a + ii[row+1]; 350289c6957cSBarry Smith v3 = a->a + ii[row+2]; 350389c6957cSBarry Smith sum1 = t[row]; 350489c6957cSBarry Smith sum2 = t[row+1]; 350589c6957cSBarry Smith sum3 = t[row+2]; 350689c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 350789c6957cSBarry Smith i1 = idx[0]; 350889c6957cSBarry Smith i2 = idx[1]; 350989c6957cSBarry Smith idx += 2; 351089c6957cSBarry Smith tmp0 = t[i1]; 351189c6957cSBarry Smith tmp1 = t[i2]; 351289c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 351389c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 351489c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 351589c6957cSBarry Smith } 351689c6957cSBarry Smith 351789c6957cSBarry Smith if (n == sz-1){ 351889c6957cSBarry Smith tmp0 = t[*idx]; 351989c6957cSBarry Smith sum1 -= v1[0] * tmp0; 352089c6957cSBarry Smith sum2 -= v2[0] * tmp0; 352189c6957cSBarry Smith sum3 -= v3[0] * tmp0; 352289c6957cSBarry Smith } 352389c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6]; 352489c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7]; 352589c6957cSBarry Smith x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8]; 352689c6957cSBarry Smith ibdiag += 9; row += 3; 352789c6957cSBarry Smith break; 352889c6957cSBarry Smith case 4: 352989c6957cSBarry Smith v2 = a->a + ii[row+1]; 353089c6957cSBarry Smith v3 = a->a + ii[row+2]; 353189c6957cSBarry Smith v4 = a->a + ii[row+3]; 353289c6957cSBarry Smith sum1 = t[row]; 353389c6957cSBarry Smith sum2 = t[row+1]; 353489c6957cSBarry Smith sum3 = t[row+2]; 353589c6957cSBarry Smith sum4 = t[row+3]; 353689c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 353789c6957cSBarry Smith i1 = idx[0]; 353889c6957cSBarry Smith i2 = idx[1]; 353989c6957cSBarry Smith idx += 2; 354089c6957cSBarry Smith tmp0 = t[i1]; 354189c6957cSBarry Smith tmp1 = t[i2]; 354289c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 354389c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 354489c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 354589c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 354689c6957cSBarry Smith } 354789c6957cSBarry Smith 354889c6957cSBarry Smith if (n == sz-1){ 354989c6957cSBarry Smith tmp0 = t[*idx]; 355089c6957cSBarry Smith sum1 -= v1[0] * tmp0; 355189c6957cSBarry Smith sum2 -= v2[0] * tmp0; 355289c6957cSBarry Smith sum3 -= v3[0] * tmp0; 355389c6957cSBarry Smith sum4 -= v4[0] * tmp0; 355489c6957cSBarry Smith } 355589c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12]; 355689c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13]; 355789c6957cSBarry Smith x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14]; 355889c6957cSBarry Smith x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15]; 355989c6957cSBarry Smith ibdiag += 16; row += 4; 356089c6957cSBarry Smith break; 356189c6957cSBarry Smith case 5: 356289c6957cSBarry Smith v2 = a->a + ii[row+1]; 356389c6957cSBarry Smith v3 = a->a + ii[row+2]; 356489c6957cSBarry Smith v4 = a->a + ii[row+3]; 356589c6957cSBarry Smith v5 = a->a + ii[row+4]; 356689c6957cSBarry Smith sum1 = t[row]; 356789c6957cSBarry Smith sum2 = t[row+1]; 356889c6957cSBarry Smith sum3 = t[row+2]; 356989c6957cSBarry Smith sum4 = t[row+3]; 357089c6957cSBarry Smith sum5 = t[row+4]; 357189c6957cSBarry Smith for(n = 0; n<sz-1; n+=2) { 357289c6957cSBarry Smith i1 = idx[0]; 357389c6957cSBarry Smith i2 = idx[1]; 357489c6957cSBarry Smith idx += 2; 357589c6957cSBarry Smith tmp0 = t[i1]; 357689c6957cSBarry Smith tmp1 = t[i2]; 357789c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 357889c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 357989c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 358089c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 358189c6957cSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 358289c6957cSBarry Smith } 358389c6957cSBarry Smith 358489c6957cSBarry Smith if (n == sz-1){ 358589c6957cSBarry Smith tmp0 = t[*idx]; 358689c6957cSBarry Smith sum1 -= v1[0] * tmp0; 358789c6957cSBarry Smith sum2 -= v2[0] * tmp0; 358889c6957cSBarry Smith sum3 -= v3[0] * tmp0; 358989c6957cSBarry Smith sum4 -= v4[0] * tmp0; 359089c6957cSBarry Smith sum5 -= v5[0] * tmp0; 359189c6957cSBarry Smith } 359289c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20]; 359389c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21]; 359489c6957cSBarry Smith x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22]; 359589c6957cSBarry Smith x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23]; 359689c6957cSBarry Smith x[row+4] += t[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24]; 359789c6957cSBarry Smith ibdiag += 25; row += 5; 359889c6957cSBarry Smith break; 359989c6957cSBarry Smith default: 3600e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 360189c6957cSBarry Smith } 360289c6957cSBarry Smith } 360389c6957cSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 36042af78befSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 36053649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3606938d4eb3SBarry Smith } 36072af78befSBarry Smith PetscFunctionReturn(0); 36082af78befSBarry Smith } 36092af78befSBarry Smith 361089c6957cSBarry Smith #undef __FUNCT__ 36114108e4d5SBarry Smith #define __FUNCT__ "MatMultDiagonalBlock_SeqAIJ_Inode" 36124108e4d5SBarry Smith PetscErrorCode MatMultDiagonalBlock_SeqAIJ_Inode(Mat A,Vec bb,Vec xx) 361389c6957cSBarry Smith { 361489c6957cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 361589c6957cSBarry Smith PetscScalar *x,tmp1,tmp2,tmp3,tmp4,tmp5,x1,x2,x3,x4,x5; 361689c6957cSBarry Smith const MatScalar *bdiag = a->inode.bdiag; 361789c6957cSBarry Smith const PetscScalar *b; 361889c6957cSBarry Smith PetscErrorCode ierr; 361989c6957cSBarry Smith PetscInt m = a->inode.node_count,cnt = 0,i,row; 362089c6957cSBarry Smith const PetscInt *sizes = a->inode.size; 36212af78befSBarry Smith 362289c6957cSBarry Smith PetscFunctionBegin; 362389c6957cSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 36243649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 362589c6957cSBarry Smith cnt = 0; 362689c6957cSBarry Smith for (i=0, row=0; i<m; i++) { 362789c6957cSBarry Smith switch (sizes[i]){ 362889c6957cSBarry Smith case 1: 362989c6957cSBarry Smith x[row] = b[row]*bdiag[cnt++];row++; 363089c6957cSBarry Smith break; 363189c6957cSBarry Smith case 2: 363289c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; 363389c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2]; 363489c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3]; 363589c6957cSBarry Smith x[row++] = tmp1; 363689c6957cSBarry Smith x[row++] = tmp2; 363789c6957cSBarry Smith cnt += 4; 363889c6957cSBarry Smith break; 363989c6957cSBarry Smith case 3: 364089c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; x3 = b[row+2]; 364189c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6]; 364289c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7]; 364389c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8]; 364489c6957cSBarry Smith x[row++] = tmp1; 364589c6957cSBarry Smith x[row++] = tmp2; 364689c6957cSBarry Smith x[row++] = tmp3; 364789c6957cSBarry Smith cnt += 9; 364889c6957cSBarry Smith break; 364989c6957cSBarry Smith case 4: 365089c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3]; 365189c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12]; 365289c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13]; 365389c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14]; 365489c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15]; 365589c6957cSBarry Smith x[row++] = tmp1; 365689c6957cSBarry Smith x[row++] = tmp2; 365789c6957cSBarry Smith x[row++] = tmp3; 365889c6957cSBarry Smith x[row++] = tmp4; 365989c6957cSBarry Smith cnt += 16; 366089c6957cSBarry Smith break; 366189c6957cSBarry Smith case 5: 366289c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3]; x5 = b[row+4]; 366389c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20]; 366489c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21]; 366589c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22]; 366689c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23]; 366789c6957cSBarry Smith tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24]; 366889c6957cSBarry Smith x[row++] = tmp1; 366989c6957cSBarry Smith x[row++] = tmp2; 367089c6957cSBarry Smith x[row++] = tmp3; 367189c6957cSBarry Smith x[row++] = tmp4; 367289c6957cSBarry Smith x[row++] = tmp5; 367389c6957cSBarry Smith cnt += 25; 367489c6957cSBarry Smith break; 367589c6957cSBarry Smith default: 3676e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 367789c6957cSBarry Smith } 367889c6957cSBarry Smith } 367989c6957cSBarry Smith ierr = PetscLogFlops(2*cnt);CHKERRQ(ierr); 368089c6957cSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 36813649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 368289c6957cSBarry Smith PetscFunctionReturn(0); 368389c6957cSBarry Smith } 368489c6957cSBarry Smith 36854c1414c8SBarry Smith /* 36864c1414c8SBarry Smith samestructure indicates that the matrix has not changed its nonzero structure so we 36874c1414c8SBarry Smith do not need to recompute the inodes 36884c1414c8SBarry Smith */ 36894c1414c8SBarry Smith #undef __FUNCT__ 36904c1414c8SBarry Smith #define __FUNCT__ "Mat_CheckInode" 3691ace3abfcSBarry Smith PetscErrorCode Mat_CheckInode(Mat A,PetscBool samestructure) 36924c1414c8SBarry Smith { 36934c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 36944c1414c8SBarry Smith PetscErrorCode ierr; 36954c1414c8SBarry Smith PetscInt i,j,m,nzx,nzy,*idx,*idy,*ns,*ii,node_count,blk_size; 3696ace3abfcSBarry Smith PetscBool flag; 36974c1414c8SBarry Smith 36984c1414c8SBarry Smith PetscFunctionBegin; 3699d74fe821SBarry Smith if (!a->inode.use) PetscFunctionReturn(0); 37004c1414c8SBarry Smith if (a->inode.checked && samestructure) PetscFunctionReturn(0); 37014c1414c8SBarry Smith 37024c1414c8SBarry Smith 3703d0f46423SBarry Smith m = A->rmap->n; 37044c1414c8SBarry Smith if (a->inode.size) {ns = a->inode.size;} 37054c1414c8SBarry Smith else {ierr = PetscMalloc((m+1)*sizeof(PetscInt),&ns);CHKERRQ(ierr);} 37064c1414c8SBarry Smith 37074c1414c8SBarry Smith i = 0; 37084c1414c8SBarry Smith node_count = 0; 37094c1414c8SBarry Smith idx = a->j; 37104c1414c8SBarry Smith ii = a->i; 37114c1414c8SBarry Smith while (i < m){ /* For each row */ 37124c1414c8SBarry Smith nzx = ii[i+1] - ii[i]; /* Number of nonzeros */ 37134c1414c8SBarry Smith /* Limits the number of elements in a node to 'a->inode.limit' */ 37144c1414c8SBarry Smith for (j=i+1,idy=idx,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) { 37154c1414c8SBarry Smith nzy = ii[j+1] - ii[j]; /* Same number of nonzeros */ 37164c1414c8SBarry Smith if (nzy != nzx) break; 37174c1414c8SBarry Smith idy += nzx; /* Same nonzero pattern */ 37184c1414c8SBarry Smith ierr = PetscMemcmp(idx,idy,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr); 37194c1414c8SBarry Smith if (!flag) break; 37204c1414c8SBarry Smith } 37214c1414c8SBarry Smith ns[node_count++] = blk_size; 37224c1414c8SBarry Smith idx += blk_size*nzx; 37234c1414c8SBarry Smith i = j; 37244c1414c8SBarry Smith } 37254c1414c8SBarry Smith /* If not enough inodes found,, do not use inode version of the routines */ 3726be6adb11SBarry Smith if (!m || node_count > .8*m) { 37274c1414c8SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 37284c1414c8SBarry Smith a->inode.node_count = 0; 37294c1414c8SBarry Smith a->inode.size = PETSC_NULL; 37304c1414c8SBarry Smith a->inode.use = PETSC_FALSE; 3731ded6e084SBarry Smith A->ops->mult = MatMult_SeqAIJ; 3732ded6e084SBarry Smith A->ops->sor = MatSOR_SeqAIJ; 3733ded6e084SBarry Smith A->ops->multadd = MatMultAdd_SeqAIJ; 3734ded6e084SBarry Smith A->ops->getrowij = MatGetRowIJ_SeqAIJ; 3735ded6e084SBarry Smith A->ops->restorerowij = MatRestoreRowIJ_SeqAIJ; 3736ded6e084SBarry Smith A->ops->getcolumnij = MatGetColumnIJ_SeqAIJ; 3737ded6e084SBarry Smith A->ops->restorecolumnij = MatRestoreColumnIJ_SeqAIJ; 3738ded6e084SBarry Smith A->ops->coloringpatch = 0; 3739ded6e084SBarry Smith A->ops->multdiagonalblock = 0; 37404c1414c8SBarry Smith ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr); 37414c1414c8SBarry Smith } else { 3742d5f3da31SBarry Smith if (!A->factortype) { 37434108e4d5SBarry Smith A->ops->mult = MatMult_SeqAIJ_Inode; 37444108e4d5SBarry Smith A->ops->sor = MatSOR_SeqAIJ_Inode; 37454108e4d5SBarry Smith A->ops->multadd = MatMultAdd_SeqAIJ_Inode; 37464108e4d5SBarry Smith A->ops->getrowij = MatGetRowIJ_SeqAIJ_Inode; 37474108e4d5SBarry Smith A->ops->restorerowij = MatRestoreRowIJ_SeqAIJ_Inode; 37484108e4d5SBarry Smith A->ops->getcolumnij = MatGetColumnIJ_SeqAIJ_Inode; 37494108e4d5SBarry Smith A->ops->restorecolumnij = MatRestoreColumnIJ_SeqAIJ_Inode; 37504108e4d5SBarry Smith A->ops->coloringpatch = MatColoringPatch_SeqAIJ_Inode; 37514108e4d5SBarry Smith A->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode; 3752d3ac4fa3SBarry Smith } else { 3753d3ac4fa3SBarry Smith A->ops->solve = MatSolve_SeqAIJ_Inode_inplace; 3754d3ac4fa3SBarry Smith } 37554c1414c8SBarry Smith a->inode.node_count = node_count; 37564c1414c8SBarry Smith a->inode.size = ns; 37574c1414c8SBarry Smith ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr); 37584c1414c8SBarry Smith } 3759be6adb11SBarry Smith a->inode.checked = PETSC_TRUE; 37604c1414c8SBarry Smith PetscFunctionReturn(0); 37614c1414c8SBarry Smith } 37624c1414c8SBarry Smith 376317454e89SShri Abhyankar #undef __FUNCT__ 376417454e89SShri Abhyankar #define __FUNCT__ "MatGetRow_FactoredLU" 376517454e89SShri Abhyankar PETSC_STATIC_INLINE PetscErrorCode MatGetRow_FactoredLU(PetscInt *cols,PetscInt nzl,PetscInt nzu,PetscInt nz,PetscInt *ai,PetscInt *aj,PetscInt *adiag,PetscInt row) 376617454e89SShri Abhyankar { 376717454e89SShri Abhyankar PetscInt k, *vi; 376817454e89SShri Abhyankar PetscFunctionBegin; 376917454e89SShri Abhyankar vi = aj + ai[row]; 377017454e89SShri Abhyankar for(k=0;k<nzl;k++) cols[k] = vi[k]; 377117454e89SShri Abhyankar vi = aj + adiag[row]; 377217454e89SShri Abhyankar cols[nzl] = vi[0]; 377317454e89SShri Abhyankar vi = aj + adiag[row+1]+1; 377417454e89SShri Abhyankar for(k=0;k<nzu;k++) cols[nzl+1+k] = vi[k]; 377517454e89SShri Abhyankar PetscFunctionReturn(0); 377617454e89SShri Abhyankar } 37776936b636SHong Zhang /* 37786936b636SHong Zhang Mat_CheckInode_FactorLU - Check Inode for factored seqaij matrix. 37796936b636SHong Zhang Modified from Mat_CheckInode(). 37806936b636SHong Zhang 37816936b636SHong Zhang Input Parameters: 37826936b636SHong Zhang + Mat A - ILU or LU matrix factor 378317454e89SShri Abhyankar - samestructure - TRUE indicates that the matrix has not changed its nonzero structure so we 37846936b636SHong Zhang do not need to recompute the inodes 37856936b636SHong Zhang */ 3786019b515eSShri Abhyankar #undef __FUNCT__ 3787019b515eSShri Abhyankar #define __FUNCT__ "Mat_CheckInode_FactorLU" 3788ace3abfcSBarry Smith PetscErrorCode Mat_CheckInode_FactorLU(Mat A,PetscBool samestructure) 3789019b515eSShri Abhyankar { 3790019b515eSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3791019b515eSShri Abhyankar PetscErrorCode ierr; 3792019b515eSShri Abhyankar PetscInt i,j,m,nzl1,nzu1,nzl2,nzu2,nzx,nzy,node_count,blk_size; 3793019b515eSShri Abhyankar PetscInt *cols1,*cols2,*ns,*ai = a->i,*aj = a->j, *adiag = a->diag; 3794ace3abfcSBarry Smith PetscBool flag; 3795019b515eSShri Abhyankar 3796019b515eSShri Abhyankar PetscFunctionBegin; 3797019b515eSShri Abhyankar if (!a->inode.use) PetscFunctionReturn(0); 3798019b515eSShri Abhyankar if (a->inode.checked && samestructure) PetscFunctionReturn(0); 3799019b515eSShri Abhyankar 3800019b515eSShri Abhyankar m = A->rmap->n; 3801019b515eSShri Abhyankar if (a->inode.size) {ns = a->inode.size;} 3802019b515eSShri Abhyankar else {ierr = PetscMalloc((m+1)*sizeof(PetscInt),&ns);CHKERRQ(ierr);} 3803019b515eSShri Abhyankar 3804019b515eSShri Abhyankar i = 0; 3805019b515eSShri Abhyankar node_count = 0; 3806019b515eSShri Abhyankar while (i < m){ /* For each row */ 3807019b515eSShri Abhyankar nzl1 = ai[i+1] - ai[i]; /* Number of nonzeros in L */ 3808019b515eSShri Abhyankar nzu1 = adiag[i] - adiag[i+1] - 1; /* Number of nonzeros in U excluding diagonal*/ 3809019b515eSShri Abhyankar nzx = nzl1 + nzu1 + 1; 3810019b515eSShri Abhyankar ierr = PetscMalloc((nzx+1)*sizeof(PetscInt),&cols1);CHKERRQ(ierr); 3811019b515eSShri Abhyankar MatGetRow_FactoredLU(cols1,nzl1,nzu1,nzx,ai,aj,adiag,i); 3812019b515eSShri Abhyankar 3813019b515eSShri Abhyankar /* Limits the number of elements in a node to 'a->inode.limit' */ 3814019b515eSShri Abhyankar for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) { 3815019b515eSShri Abhyankar nzl2 = ai[j+1] - ai[j]; 3816019b515eSShri Abhyankar nzu2 = adiag[j] - adiag[j+1] - 1; 3817019b515eSShri Abhyankar nzy = nzl2 + nzu2 + 1; 3818019b515eSShri Abhyankar if( nzy != nzx) break; 3819019b515eSShri Abhyankar ierr = PetscMalloc((nzy+1)*sizeof(PetscInt),&cols2);CHKERRQ(ierr); 382017454e89SShri Abhyankar ierr = MatGetRow_FactoredLU(cols2,nzl2,nzu2,nzy,ai,aj,adiag,j);CHKERRQ(ierr); 3821019b515eSShri Abhyankar ierr = PetscMemcmp(cols1,cols2,nzx*sizeof(PetscInt),&flag);CHKERRQ(ierr); 3822019b515eSShri Abhyankar if (!flag) {ierr = PetscFree(cols2);CHKERRQ(ierr);break;} 3823019b515eSShri Abhyankar ierr = PetscFree(cols2);CHKERRQ(ierr); 3824019b515eSShri Abhyankar } 3825019b515eSShri Abhyankar ns[node_count++] = blk_size; 3826019b515eSShri Abhyankar ierr = PetscFree(cols1);CHKERRQ(ierr); 3827019b515eSShri Abhyankar i = j; 3828019b515eSShri Abhyankar } 3829019b515eSShri Abhyankar /* If not enough inodes found,, do not use inode version of the routines */ 3830be6adb11SBarry Smith if (!m || node_count > .8*m) { 3831019b515eSShri Abhyankar ierr = PetscFree(ns);CHKERRQ(ierr); 3832019b515eSShri Abhyankar a->inode.node_count = 0; 3833019b515eSShri Abhyankar a->inode.size = PETSC_NULL; 3834019b515eSShri Abhyankar a->inode.use = PETSC_FALSE; 3835019b515eSShri Abhyankar ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr); 3836019b515eSShri Abhyankar } else { 3837019b515eSShri Abhyankar A->ops->mult = 0; 3838019b515eSShri Abhyankar A->ops->sor = 0; 3839019b515eSShri Abhyankar A->ops->multadd = 0; 3840019b515eSShri Abhyankar A->ops->getrowij = 0; 3841019b515eSShri Abhyankar A->ops->restorerowij = 0; 3842019b515eSShri Abhyankar A->ops->getcolumnij = 0; 3843019b515eSShri Abhyankar A->ops->restorecolumnij = 0; 3844019b515eSShri Abhyankar A->ops->coloringpatch = 0; 3845019b515eSShri Abhyankar A->ops->multdiagonalblock = 0; 3846d3ac4fa3SBarry Smith A->ops->solve = MatSolve_SeqAIJ_Inode; 3847019b515eSShri Abhyankar a->inode.node_count = node_count; 3848019b515eSShri Abhyankar a->inode.size = ns; 3849019b515eSShri Abhyankar ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr); 3850019b515eSShri Abhyankar } 3851be6adb11SBarry Smith a->inode.checked = PETSC_TRUE; 3852019b515eSShri Abhyankar PetscFunctionReturn(0); 3853019b515eSShri Abhyankar } 3854019b515eSShri Abhyankar 38554c1414c8SBarry Smith /* 38564c1414c8SBarry Smith This is really ugly. if inodes are used this replaces the 38574c1414c8SBarry Smith permutations with ones that correspond to rows/cols of the matrix 38584c1414c8SBarry Smith rather then inode blocks 38594c1414c8SBarry Smith */ 38604c1414c8SBarry Smith #undef __FUNCT__ 38614c1414c8SBarry Smith #define __FUNCT__ "MatInodeAdjustForInodes" 38627087cfbeSBarry Smith PetscErrorCode MatInodeAdjustForInodes(Mat A,IS *rperm,IS *cperm) 38634c1414c8SBarry Smith { 38644ac538c5SBarry Smith PetscErrorCode ierr; 38654c1414c8SBarry Smith 38664c1414c8SBarry Smith PetscFunctionBegin; 38674ac538c5SBarry Smith ierr = PetscTryMethod(A,"MatInodeAdjustForInodes_C",(Mat,IS*,IS*),(A,rperm,cperm));CHKERRQ(ierr); 38684c1414c8SBarry Smith PetscFunctionReturn(0); 38694c1414c8SBarry Smith } 38704c1414c8SBarry Smith 38714c1414c8SBarry Smith EXTERN_C_BEGIN 38724c1414c8SBarry Smith #undef __FUNCT__ 387353c77d0aSJed Brown #define __FUNCT__ "MatInodeAdjustForInodes_SeqAIJ_Inode" 38747087cfbeSBarry Smith PetscErrorCode MatInodeAdjustForInodes_SeqAIJ_Inode(Mat A,IS *rperm,IS *cperm) 38754c1414c8SBarry Smith { 38764c1414c8SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 38774c1414c8SBarry Smith PetscErrorCode ierr; 38785d0c19d7SBarry Smith PetscInt m = A->rmap->n,n = A->cmap->n,i,j,nslim_row = a->inode.node_count; 38795d0c19d7SBarry Smith const PetscInt *ridx,*cidx; 38804c1414c8SBarry Smith PetscInt row,col,*permr,*permc,*ns_row = a->inode.size,*tns,start_val,end_val,indx; 38814c1414c8SBarry Smith PetscInt nslim_col,*ns_col; 38824c1414c8SBarry Smith IS ris = *rperm,cis = *cperm; 38834c1414c8SBarry Smith 38844c1414c8SBarry Smith PetscFunctionBegin; 38854c1414c8SBarry Smith if (!a->inode.size) PetscFunctionReturn(0); /* no inodes so return */ 38864c1414c8SBarry Smith if (a->inode.node_count == m) PetscFunctionReturn(0); /* all inodes are of size 1 */ 38874c1414c8SBarry Smith 38884c1414c8SBarry Smith ierr = Mat_CreateColInode(A,&nslim_col,&ns_col);CHKERRQ(ierr); 38894c1414c8SBarry Smith ierr = PetscMalloc((((nslim_row>nslim_col)?nslim_row:nslim_col)+1)*sizeof(PetscInt),&tns);CHKERRQ(ierr); 38900e83c824SBarry Smith ierr = PetscMalloc2(m,PetscInt,&permr,n,PetscInt,&permc);CHKERRQ(ierr); 38914c1414c8SBarry Smith 38924c1414c8SBarry Smith ierr = ISGetIndices(ris,&ridx);CHKERRQ(ierr); 38934c1414c8SBarry Smith ierr = ISGetIndices(cis,&cidx);CHKERRQ(ierr); 38944c1414c8SBarry Smith 38954c1414c8SBarry Smith /* Form the inode structure for the rows of permuted matric using inv perm*/ 38964c1414c8SBarry Smith for (i=0,tns[0]=0; i<nslim_row; ++i) tns[i+1] = tns[i] + ns_row[i]; 38974c1414c8SBarry Smith 38984c1414c8SBarry Smith /* Construct the permutations for rows*/ 38994c1414c8SBarry Smith for (i=0,row = 0; i<nslim_row; ++i){ 39004c1414c8SBarry Smith indx = ridx[i]; 39014c1414c8SBarry Smith start_val = tns[indx]; 39024c1414c8SBarry Smith end_val = tns[indx + 1]; 39034c1414c8SBarry Smith for (j=start_val; j<end_val; ++j,++row) permr[row]= j; 39044c1414c8SBarry Smith } 39054c1414c8SBarry Smith 39064c1414c8SBarry Smith /* Form the inode structure for the columns of permuted matrix using inv perm*/ 39074c1414c8SBarry Smith for (i=0,tns[0]=0; i<nslim_col; ++i) tns[i+1] = tns[i] + ns_col[i]; 39084c1414c8SBarry Smith 39094c1414c8SBarry Smith /* Construct permutations for columns */ 39104c1414c8SBarry Smith for (i=0,col=0; i<nslim_col; ++i){ 39114c1414c8SBarry Smith indx = cidx[i]; 39124c1414c8SBarry Smith start_val = tns[indx]; 39134c1414c8SBarry Smith end_val = tns[indx + 1]; 39144c1414c8SBarry Smith for (j = start_val; j<end_val; ++j,++col) permc[col]= j; 39154c1414c8SBarry Smith } 39164c1414c8SBarry Smith 391770b3c8c7SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permr,PETSC_COPY_VALUES,rperm);CHKERRQ(ierr); 39180e83c824SBarry Smith ierr = ISSetPermutation(*rperm);CHKERRQ(ierr); 391970b3c8c7SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permc,PETSC_COPY_VALUES,cperm);CHKERRQ(ierr); 39200e83c824SBarry Smith ierr = ISSetPermutation(*cperm);CHKERRQ(ierr); 39214c1414c8SBarry Smith 39224c1414c8SBarry Smith ierr = ISRestoreIndices(ris,&ridx);CHKERRQ(ierr); 39234c1414c8SBarry Smith ierr = ISRestoreIndices(cis,&cidx);CHKERRQ(ierr); 39244c1414c8SBarry Smith 39254c1414c8SBarry Smith ierr = PetscFree(ns_col);CHKERRQ(ierr); 39260e83c824SBarry Smith ierr = PetscFree2(permr,permc);CHKERRQ(ierr); 39276bf464f9SBarry Smith ierr = ISDestroy(&cis);CHKERRQ(ierr); 39286bf464f9SBarry Smith ierr = ISDestroy(&ris);CHKERRQ(ierr); 39294c1414c8SBarry Smith ierr = PetscFree(tns);CHKERRQ(ierr); 39304c1414c8SBarry Smith PetscFunctionReturn(0); 39314c1414c8SBarry Smith } 39324c1414c8SBarry Smith EXTERN_C_END 39334c1414c8SBarry Smith 39344c1414c8SBarry Smith #undef __FUNCT__ 39354c1414c8SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes" 39364c1414c8SBarry Smith /*@C 39374c1414c8SBarry Smith MatInodeGetInodeSizes - Returns the inode information of the Inode matrix. 39384c1414c8SBarry Smith 39393f9fe445SBarry Smith Not Collective 39404c1414c8SBarry Smith 39414c1414c8SBarry Smith Input Parameter: 39424c1414c8SBarry Smith . A - the Inode matrix or matrix derived from the Inode class -- e.g., SeqAIJ 39434c1414c8SBarry Smith 39444c1414c8SBarry Smith Output Parameter: 39454c1414c8SBarry Smith + node_count - no of inodes present in the matrix. 39464c1414c8SBarry Smith . sizes - an array of size node_count,with sizes of each inode. 39474c1414c8SBarry Smith - limit - the max size used to generate the inodes. 39484c1414c8SBarry Smith 39494c1414c8SBarry Smith Level: advanced 39504c1414c8SBarry Smith 39514c1414c8SBarry Smith Notes: This routine returns some internal storage information 39524c1414c8SBarry Smith of the matrix, it is intended to be used by advanced users. 39534c1414c8SBarry Smith It should be called after the matrix is assembled. 39544c1414c8SBarry Smith The contents of the sizes[] array should not be changed. 39554c1414c8SBarry Smith PETSC_NULL may be passed for information not requested. 39564c1414c8SBarry Smith 39574c1414c8SBarry Smith .keywords: matrix, seqaij, get, inode 39584c1414c8SBarry Smith 39594c1414c8SBarry Smith .seealso: MatGetInfo() 39604c1414c8SBarry Smith @*/ 39617087cfbeSBarry Smith PetscErrorCode MatInodeGetInodeSizes(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit) 39624c1414c8SBarry Smith { 39634c1414c8SBarry Smith PetscErrorCode ierr,(*f)(Mat,PetscInt*,PetscInt*[],PetscInt*); 39644c1414c8SBarry Smith 39654c1414c8SBarry Smith PetscFunctionBegin; 3966e32f2f54SBarry Smith if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix"); 39674c1414c8SBarry Smith ierr = PetscObjectQueryFunction((PetscObject)A,"MatInodeGetInodeSizes_C",(void (**)(void))&f);CHKERRQ(ierr); 39684c1414c8SBarry Smith if (f) { 39694c1414c8SBarry Smith ierr = (*f)(A,node_count,sizes,limit);CHKERRQ(ierr); 39704c1414c8SBarry Smith } 39714c1414c8SBarry Smith PetscFunctionReturn(0); 39724c1414c8SBarry Smith } 39734c1414c8SBarry Smith 39744c1414c8SBarry Smith EXTERN_C_BEGIN 39754c1414c8SBarry Smith #undef __FUNCT__ 39764108e4d5SBarry Smith #define __FUNCT__ "MatInodeGetInodeSizes_SeqAIJ_Inode" 39777087cfbeSBarry Smith PetscErrorCode MatInodeGetInodeSizes_SeqAIJ_Inode(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit) 39784c1414c8SBarry Smith { 39794c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 39804c1414c8SBarry Smith 39814c1414c8SBarry Smith PetscFunctionBegin; 39824c1414c8SBarry Smith if (node_count) *node_count = a->inode.node_count; 39834c1414c8SBarry Smith if (sizes) *sizes = a->inode.size; 39844c1414c8SBarry Smith if (limit) *limit = a->inode.limit; 39854c1414c8SBarry Smith PetscFunctionReturn(0); 39864c1414c8SBarry Smith } 39874c1414c8SBarry Smith EXTERN_C_END 3988