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 8470880abSPatrick Sanan static PetscErrorCode MatCreateColInode_Private(Mat A,PetscInt *size,PetscInt **ns) 94c1414c8SBarry Smith { 104c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 114c1414c8SBarry Smith PetscErrorCode ierr; 124c1414c8SBarry Smith PetscInt i,count,m,n,min_mn,*ns_row,*ns_col; 134c1414c8SBarry Smith 144c1414c8SBarry Smith PetscFunctionBegin; 15d0f46423SBarry Smith n = A->cmap->n; 16d0f46423SBarry Smith m = A->rmap->n; 17b215bc84SStefano Zampini if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 184c1414c8SBarry Smith ns_row = a->inode.size; 194c1414c8SBarry Smith 204c1414c8SBarry Smith min_mn = (m < n) ? m : n; 214c1414c8SBarry Smith if (!ns) { 224c1414c8SBarry Smith for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) ; 234c1414c8SBarry Smith for (; count+1 < n; count++,i++) ; 244c1414c8SBarry Smith if (count < n) { 254c1414c8SBarry Smith i++; 264c1414c8SBarry Smith } 274c1414c8SBarry Smith *size = i; 284c1414c8SBarry Smith PetscFunctionReturn(0); 294c1414c8SBarry Smith } 30854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&ns_col);CHKERRQ(ierr); 314c1414c8SBarry Smith 324c1414c8SBarry Smith /* Use the same row structure wherever feasible. */ 334c1414c8SBarry Smith for (count=0,i=0; count<min_mn; count+=ns_row[i],i++) { 344c1414c8SBarry Smith ns_col[i] = ns_row[i]; 354c1414c8SBarry Smith } 364c1414c8SBarry Smith 374c1414c8SBarry Smith /* if m < n; pad up the remainder with inode_limit */ 384c1414c8SBarry Smith for (; count+1 < n; count++,i++) { 394c1414c8SBarry Smith ns_col[i] = 1; 404c1414c8SBarry Smith } 414c1414c8SBarry Smith /* The last node is the odd ball. padd it up with the remaining rows; */ 424c1414c8SBarry Smith if (count < n) { 434c1414c8SBarry Smith ns_col[i] = n - count; 444c1414c8SBarry Smith i++; 454c1414c8SBarry Smith } else if (count > n) { 464c1414c8SBarry Smith /* Adjust for the over estimation */ 474c1414c8SBarry Smith ns_col[i-1] += n - count; 484c1414c8SBarry Smith } 494c1414c8SBarry Smith *size = i; 504c1414c8SBarry Smith *ns = ns_col; 514c1414c8SBarry Smith PetscFunctionReturn(0); 524c1414c8SBarry Smith } 534c1414c8SBarry Smith 544c1414c8SBarry Smith 554c1414c8SBarry Smith /* 564c1414c8SBarry Smith This builds symmetric version of nonzero structure, 574c1414c8SBarry Smith */ 581a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Symmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift) 594c1414c8SBarry Smith { 604c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 614c1414c8SBarry Smith PetscErrorCode ierr; 628758e1faSBarry Smith PetscInt *work,*ia,*ja,nz,nslim_row,nslim_col,m,row,col,n; 638758e1faSBarry Smith PetscInt *tns,*tvc,*ns_row = a->inode.size,*ns_col,nsz,i1,i2; 648758e1faSBarry Smith const PetscInt *j,*jmax,*ai= a->i,*aj = a->j; 654c1414c8SBarry Smith 664c1414c8SBarry Smith PetscFunctionBegin; 674c1414c8SBarry Smith nslim_row = a->inode.node_count; 68d0f46423SBarry Smith m = A->rmap->n; 69d0f46423SBarry Smith n = A->cmap->n; 70e32f2f54SBarry Smith if (m != n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatGetRowIJ_SeqAIJ_Inode_Symmetric: Matrix should be square"); 71b215bc84SStefano Zampini if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 724c1414c8SBarry Smith 734c1414c8SBarry Smith /* Use the row_inode as column_inode */ 744c1414c8SBarry Smith nslim_col = nslim_row; 754c1414c8SBarry Smith ns_col = ns_row; 764c1414c8SBarry Smith 774c1414c8SBarry Smith /* allocate space for reformated inode structure */ 78580bdb30SBarry Smith ierr = PetscMalloc2(nslim_col+1,&tns,n+1,&tvc);CHKERRQ(ierr); 794c1414c8SBarry Smith for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1]+ ns_row[i1]; 804c1414c8SBarry Smith 814c1414c8SBarry Smith for (i1=0,col=0; i1<nslim_col; ++i1) { 824c1414c8SBarry Smith nsz = ns_col[i1]; 832205254eSKarl Rupp for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1; 844c1414c8SBarry Smith } 854c1414c8SBarry Smith /* allocate space for row pointers */ 86580bdb30SBarry Smith ierr = PetscCalloc1(nslim_row+1,&ia);CHKERRQ(ierr); 874c1414c8SBarry Smith *iia = ia; 88854ce69bSBarry Smith ierr = PetscMalloc1(nslim_row+1,&work);CHKERRQ(ierr); 894c1414c8SBarry Smith 904c1414c8SBarry Smith /* determine the number of columns in each row */ 914c1414c8SBarry Smith ia[0] = oshift; 924c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 934c1414c8SBarry Smith 944c1414c8SBarry Smith j = aj + ai[row] + ishift; 954c1414c8SBarry Smith jmax = aj + ai[row+1] + ishift; 9683fed2edSSatish Balay if (j==jmax) continue; /* empty row */ 974c1414c8SBarry Smith col = *j++ + ishift; 984c1414c8SBarry Smith i2 = tvc[col]; 994c1414c8SBarry Smith while (i2<i1 && j<jmax) { /* 1.[-xx-d-xx--] 2.[-xx-------],off-diagonal elemets */ 1004c1414c8SBarry Smith ia[i1+1]++; 1014c1414c8SBarry Smith ia[i2+1]++; 1024c1414c8SBarry Smith i2++; /* Start col of next node */ 10390d2dec7SBarry Smith while ((j<jmax) && ((col=*j+ishift)<tns[i2])) ++j; 1044c1414c8SBarry Smith i2 = tvc[col]; 1054c1414c8SBarry Smith } 1064c1414c8SBarry Smith if (i2 == i1) ia[i2+1]++; /* now the diagonal element */ 1074c1414c8SBarry Smith } 1084c1414c8SBarry Smith 1094c1414c8SBarry Smith /* shift ia[i] to point to next row */ 1104c1414c8SBarry Smith for (i1=1; i1<nslim_row+1; i1++) { 1114c1414c8SBarry Smith row = ia[i1-1]; 1124c1414c8SBarry Smith ia[i1] += row; 1134c1414c8SBarry Smith work[i1-1] = row - oshift; 1144c1414c8SBarry Smith } 1154c1414c8SBarry Smith 1164c1414c8SBarry Smith /* allocate space for column pointers */ 1174c1414c8SBarry Smith nz = ia[nslim_row] + (!ishift); 118785e854fSJed Brown ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr); 1194c1414c8SBarry Smith *jja = ja; 1204c1414c8SBarry Smith 1214c1414c8SBarry Smith /* loop over lower triangular part putting into ja */ 1224c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row += ns_row[i1],i1++) { 1234c1414c8SBarry Smith j = aj + ai[row] + ishift; 1244c1414c8SBarry Smith jmax = aj + ai[row+1] + ishift; 12583fed2edSSatish Balay if (j==jmax) continue; /* empty row */ 1264c1414c8SBarry Smith col = *j++ + ishift; 1274c1414c8SBarry Smith i2 = tvc[col]; 1284c1414c8SBarry Smith while (i2<i1 && j<jmax) { 1294c1414c8SBarry Smith ja[work[i2]++] = i1 + oshift; 1304c1414c8SBarry Smith ja[work[i1]++] = i2 + oshift; 1314c1414c8SBarry Smith ++i2; 13290d2dec7SBarry Smith while ((j<jmax) && ((col=*j+ishift)< tns[i2])) ++j; /* Skip rest col indices in this node */ 1334c1414c8SBarry Smith i2 = tvc[col]; 1344c1414c8SBarry Smith } 1354c1414c8SBarry Smith if (i2 == i1) ja[work[i1]++] = i2 + oshift; 1364c1414c8SBarry Smith 1374c1414c8SBarry Smith } 1384c1414c8SBarry Smith ierr = PetscFree(work);CHKERRQ(ierr); 139580bdb30SBarry Smith ierr = PetscFree2(tns,tvc);CHKERRQ(ierr); 1404c1414c8SBarry Smith PetscFunctionReturn(0); 1414c1414c8SBarry Smith } 1424c1414c8SBarry Smith 1434c1414c8SBarry Smith /* 1444c1414c8SBarry Smith This builds nonsymmetric version of nonzero structure, 1454c1414c8SBarry Smith */ 1461a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift) 1474c1414c8SBarry Smith { 1484c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 1494c1414c8SBarry Smith PetscErrorCode ierr; 1508758e1faSBarry Smith PetscInt *work,*ia,*ja,nz,nslim_row,n,row,col,*ns_col,nslim_col; 1518758e1faSBarry Smith PetscInt *tns,*tvc,nsz,i1,i2; 1528758e1faSBarry Smith const PetscInt *j,*ai= a->i,*aj = a->j,*ns_row = a->inode.size; 1534c1414c8SBarry Smith 1544c1414c8SBarry Smith PetscFunctionBegin; 155b215bc84SStefano Zampini if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 1564c1414c8SBarry Smith nslim_row = a->inode.node_count; 157d0f46423SBarry Smith n = A->cmap->n; 1584c1414c8SBarry Smith 1594c1414c8SBarry Smith /* Create The column_inode for this matrix */ 160470880abSPatrick Sanan ierr = MatCreateColInode_Private(A,&nslim_col,&ns_col);CHKERRQ(ierr); 1614c1414c8SBarry Smith 1624c1414c8SBarry Smith /* allocate space for reformated column_inode structure */ 163580bdb30SBarry Smith ierr = PetscMalloc2(nslim_col +1,&tns,n + 1,&tvc);CHKERRQ(ierr); 1644c1414c8SBarry Smith for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1]; 1654c1414c8SBarry Smith 1664c1414c8SBarry Smith for (i1=0,col=0; i1<nslim_col; ++i1) { 1674c1414c8SBarry Smith nsz = ns_col[i1]; 1682205254eSKarl Rupp for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1; 1694c1414c8SBarry Smith } 1704c1414c8SBarry Smith /* allocate space for row pointers */ 171580bdb30SBarry Smith ierr = PetscCalloc1(nslim_row+1,&ia);CHKERRQ(ierr); 1724c1414c8SBarry Smith *iia = ia; 173854ce69bSBarry Smith ierr = PetscMalloc1(nslim_row+1,&work);CHKERRQ(ierr); 1744c1414c8SBarry Smith 1754c1414c8SBarry Smith /* determine the number of columns in each row */ 1764c1414c8SBarry Smith ia[0] = oshift; 1774c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 1784c1414c8SBarry Smith j = aj + ai[row] + ishift; 17983fed2edSSatish Balay nz = ai[row+1] - ai[row]; 18083fed2edSSatish Balay if (!nz) continue; /* empty row */ 1814c1414c8SBarry Smith col = *j++ + ishift; 1824c1414c8SBarry Smith i2 = tvc[col]; 1834c1414c8SBarry Smith while (nz-- > 0) { /* off-diagonal elemets */ 1844c1414c8SBarry Smith ia[i1+1]++; 1854c1414c8SBarry Smith i2++; /* Start col of next node */ 186a8e3a797SJed Brown while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--; 1874c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 1884c1414c8SBarry Smith } 1894c1414c8SBarry Smith } 1904c1414c8SBarry Smith 1914c1414c8SBarry Smith /* shift ia[i] to point to next row */ 1924c1414c8SBarry Smith for (i1=1; i1<nslim_row+1; i1++) { 1934c1414c8SBarry Smith row = ia[i1-1]; 1944c1414c8SBarry Smith ia[i1] += row; 1954c1414c8SBarry Smith work[i1-1] = row - oshift; 1964c1414c8SBarry Smith } 1974c1414c8SBarry Smith 1984c1414c8SBarry Smith /* allocate space for column pointers */ 1994c1414c8SBarry Smith nz = ia[nslim_row] + (!ishift); 200785e854fSJed Brown ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr); 2014c1414c8SBarry Smith *jja = ja; 2024c1414c8SBarry Smith 2034c1414c8SBarry Smith /* loop over matrix putting into ja */ 2044c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 2054c1414c8SBarry Smith j = aj + ai[row] + ishift; 20683fed2edSSatish Balay nz = ai[row+1] - ai[row]; 20783fed2edSSatish Balay if (!nz) continue; /* empty row */ 2084c1414c8SBarry Smith col = *j++ + ishift; 2094c1414c8SBarry Smith i2 = tvc[col]; 2104c1414c8SBarry Smith while (nz-- > 0) { 2114c1414c8SBarry Smith ja[work[i1]++] = i2 + oshift; 2124c1414c8SBarry Smith ++i2; 213a8e3a797SJed Brown while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--; 2144c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 2154c1414c8SBarry Smith } 2164c1414c8SBarry Smith } 2174c1414c8SBarry Smith ierr = PetscFree(ns_col);CHKERRQ(ierr); 2184c1414c8SBarry Smith ierr = PetscFree(work);CHKERRQ(ierr); 219580bdb30SBarry Smith ierr = PetscFree2(tns,tvc);CHKERRQ(ierr); 2204c1414c8SBarry Smith PetscFunctionReturn(0); 2214c1414c8SBarry Smith } 2224c1414c8SBarry Smith 2231a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 2244c1414c8SBarry Smith { 2254c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2264c1414c8SBarry Smith PetscErrorCode ierr; 2274c1414c8SBarry Smith 2284c1414c8SBarry Smith PetscFunctionBegin; 2294c1414c8SBarry Smith *n = a->inode.node_count; 2304c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 2318f7157efSSatish Balay if (!blockcompressed) { 232feb237baSPierre Jolivet ierr = MatGetRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr); 2338f7157efSSatish Balay } else if (symmetric) { 2344108e4d5SBarry Smith ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 2354c1414c8SBarry Smith } else { 2364108e4d5SBarry Smith ierr = MatGetRowIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 2374c1414c8SBarry Smith } 2384c1414c8SBarry Smith PetscFunctionReturn(0); 2394c1414c8SBarry Smith } 2404c1414c8SBarry Smith 2411a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 2424c1414c8SBarry Smith { 2434c1414c8SBarry Smith PetscErrorCode ierr; 2444c1414c8SBarry Smith 2454c1414c8SBarry Smith PetscFunctionBegin; 2464c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 2478f7157efSSatish Balay 2488f7157efSSatish Balay if (!blockcompressed) { 249feb237baSPierre Jolivet ierr = MatRestoreRowIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr); 2508f7157efSSatish Balay } else { 2514c1414c8SBarry Smith ierr = PetscFree(*ia);CHKERRQ(ierr); 2524c1414c8SBarry Smith ierr = PetscFree(*ja);CHKERRQ(ierr); 2538f7157efSSatish Balay } 2544c1414c8SBarry Smith PetscFunctionReturn(0); 2554c1414c8SBarry Smith } 2564c1414c8SBarry Smith 2574c1414c8SBarry Smith /* ----------------------------------------------------------- */ 2584c1414c8SBarry Smith 2591a83f524SJed Brown static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(Mat A,const PetscInt *iia[],const PetscInt *jja[],PetscInt ishift,PetscInt oshift) 2604c1414c8SBarry Smith { 2614c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2624c1414c8SBarry Smith PetscErrorCode ierr; 2634c1414c8SBarry Smith PetscInt *work,*ia,*ja,*j,nz,nslim_row, n,row,col,*ns_col,nslim_col; 2644c1414c8SBarry Smith PetscInt *tns,*tvc,*ns_row = a->inode.size,nsz,i1,i2,*ai= a->i,*aj = a->j; 2654c1414c8SBarry Smith 2664c1414c8SBarry Smith PetscFunctionBegin; 267b215bc84SStefano Zampini if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 2684c1414c8SBarry Smith nslim_row = a->inode.node_count; 269d0f46423SBarry Smith n = A->cmap->n; 2704c1414c8SBarry Smith 2714c1414c8SBarry Smith /* Create The column_inode for this matrix */ 272470880abSPatrick Sanan ierr = MatCreateColInode_Private(A,&nslim_col,&ns_col);CHKERRQ(ierr); 2734c1414c8SBarry Smith 2744c1414c8SBarry Smith /* allocate space for reformated column_inode structure */ 275580bdb30SBarry Smith ierr = PetscMalloc2(nslim_col + 1,&tns,n + 1,&tvc);CHKERRQ(ierr); 2764c1414c8SBarry Smith for (i1=0,tns[0]=0; i1<nslim_col; ++i1) tns[i1+1] = tns[i1] + ns_col[i1]; 2774c1414c8SBarry Smith 2784c1414c8SBarry Smith for (i1=0,col=0; i1<nslim_col; ++i1) { 2794c1414c8SBarry Smith nsz = ns_col[i1]; 2802205254eSKarl Rupp for (i2=0; i2<nsz; ++i2,++col) tvc[col] = i1; 2814c1414c8SBarry Smith } 2824c1414c8SBarry Smith /* allocate space for column pointers */ 283580bdb30SBarry Smith ierr = PetscCalloc1(nslim_col+1,&ia);CHKERRQ(ierr); 2844c1414c8SBarry Smith *iia = ia; 285854ce69bSBarry Smith ierr = PetscMalloc1(nslim_col+1,&work);CHKERRQ(ierr); 2864c1414c8SBarry Smith 2874c1414c8SBarry Smith /* determine the number of columns in each row */ 2884c1414c8SBarry Smith ia[0] = oshift; 2894c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 2904c1414c8SBarry Smith j = aj + ai[row] + ishift; 2914c1414c8SBarry Smith col = *j++ + ishift; 2924c1414c8SBarry Smith i2 = tvc[col]; 2934c1414c8SBarry Smith nz = ai[row+1] - ai[row]; 2944c1414c8SBarry Smith while (nz-- > 0) { /* off-diagonal elemets */ 2954c1414c8SBarry Smith /* ia[i1+1]++; */ 2964c1414c8SBarry Smith ia[i2+1]++; 2974c1414c8SBarry Smith i2++; 298a8e3a797SJed Brown while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--; 2994c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 3004c1414c8SBarry Smith } 3014c1414c8SBarry Smith } 3024c1414c8SBarry Smith 3034c1414c8SBarry Smith /* shift ia[i] to point to next col */ 3044c1414c8SBarry Smith for (i1=1; i1<nslim_col+1; i1++) { 3054c1414c8SBarry Smith col = ia[i1-1]; 3064c1414c8SBarry Smith ia[i1] += col; 3074c1414c8SBarry Smith work[i1-1] = col - oshift; 3084c1414c8SBarry Smith } 3094c1414c8SBarry Smith 3104c1414c8SBarry Smith /* allocate space for column pointers */ 3114c1414c8SBarry Smith nz = ia[nslim_col] + (!ishift); 312785e854fSJed Brown ierr = PetscMalloc1(nz,&ja);CHKERRQ(ierr); 3134c1414c8SBarry Smith *jja = ja; 3144c1414c8SBarry Smith 3154c1414c8SBarry Smith /* loop over matrix putting into ja */ 3164c1414c8SBarry Smith for (i1=0,row=0; i1<nslim_row; row+=ns_row[i1],i1++) { 3174c1414c8SBarry Smith j = aj + ai[row] + ishift; 3184c1414c8SBarry Smith col = *j++ + ishift; 3194c1414c8SBarry Smith i2 = tvc[col]; 3204c1414c8SBarry Smith nz = ai[row+1] - ai[row]; 3214c1414c8SBarry Smith while (nz-- > 0) { 3224c1414c8SBarry Smith /* ja[work[i1]++] = i2 + oshift; */ 3234c1414c8SBarry Smith ja[work[i2]++] = i1 + oshift; 3244c1414c8SBarry Smith i2++; 325a8e3a797SJed Brown while (nz > 0 && ((col = *j++ + ishift) < tns[i2])) nz--; 3264c1414c8SBarry Smith if (nz > 0) i2 = tvc[col]; 3274c1414c8SBarry Smith } 3284c1414c8SBarry Smith } 3294c1414c8SBarry Smith ierr = PetscFree(ns_col);CHKERRQ(ierr); 3304c1414c8SBarry Smith ierr = PetscFree(work);CHKERRQ(ierr); 331580bdb30SBarry Smith ierr = PetscFree2(tns,tvc);CHKERRQ(ierr); 3324c1414c8SBarry Smith PetscFunctionReturn(0); 3334c1414c8SBarry Smith } 3344c1414c8SBarry Smith 3351a83f524SJed Brown static PetscErrorCode MatGetColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 3364c1414c8SBarry Smith { 3374c1414c8SBarry Smith PetscErrorCode ierr; 3384c1414c8SBarry Smith 3394c1414c8SBarry Smith PetscFunctionBegin; 340470880abSPatrick Sanan ierr = MatCreateColInode_Private(A,n,NULL);CHKERRQ(ierr); 3414c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 3424c1414c8SBarry Smith 3438f7157efSSatish Balay if (!blockcompressed) { 344feb237baSPierre Jolivet ierr = MatGetColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr); 3458f7157efSSatish Balay } else if (symmetric) { 3464c1414c8SBarry Smith /* Since the indices are symmetric it does'nt matter */ 3474108e4d5SBarry Smith ierr = MatGetRowIJ_SeqAIJ_Inode_Symmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 3484c1414c8SBarry Smith } else { 3494108e4d5SBarry Smith ierr = MatGetColumnIJ_SeqAIJ_Inode_Nonsymmetric(A,ia,ja,0,oshift);CHKERRQ(ierr); 3504c1414c8SBarry Smith } 3514c1414c8SBarry Smith PetscFunctionReturn(0); 3524c1414c8SBarry Smith } 3534c1414c8SBarry Smith 3541a83f524SJed Brown static PetscErrorCode MatRestoreColumnIJ_SeqAIJ_Inode(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *n,const PetscInt *ia[],const PetscInt *ja[],PetscBool *done) 3554c1414c8SBarry Smith { 3564c1414c8SBarry Smith PetscErrorCode ierr; 3574c1414c8SBarry Smith 3584c1414c8SBarry Smith PetscFunctionBegin; 3594c1414c8SBarry Smith if (!ia) PetscFunctionReturn(0); 3608f7157efSSatish Balay if (!blockcompressed) { 361feb237baSPierre Jolivet ierr = MatRestoreColumnIJ_SeqAIJ(A,oshift,symmetric,blockcompressed,n,ia,ja,done);CHKERRQ(ierr); 3628f7157efSSatish Balay } else { 3634c1414c8SBarry Smith ierr = PetscFree(*ia);CHKERRQ(ierr); 3644c1414c8SBarry Smith ierr = PetscFree(*ja);CHKERRQ(ierr); 3658f7157efSSatish Balay } 3664c1414c8SBarry Smith PetscFunctionReturn(0); 3674c1414c8SBarry Smith } 3684c1414c8SBarry Smith 3694c1414c8SBarry Smith /* ----------------------------------------------------------- */ 3704c1414c8SBarry Smith 371b215bc84SStefano Zampini PetscErrorCode MatMult_SeqAIJ_Inode(Mat A,Vec xx,Vec yy) 3724c1414c8SBarry Smith { 3734c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 3744c1414c8SBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5,tmp0,tmp1; 375d9fead3dSBarry Smith PetscScalar *y; 376dd6ea824SBarry Smith const PetscScalar *x; 377dd6ea824SBarry Smith const MatScalar *v1,*v2,*v3,*v4,*v5; 3784c1414c8SBarry Smith PetscErrorCode ierr; 3798758e1faSBarry Smith PetscInt i1,i2,n,i,row,node_max,nsz,sz,nonzerorow=0; 3808758e1faSBarry Smith const PetscInt *idx,*ns,*ii; 3814c1414c8SBarry Smith 3824c1414c8SBarry Smith #if defined(PETSC_HAVE_PRAGMA_DISJOINT) 3834c1414c8SBarry Smith #pragma disjoint(*x,*y,*v1,*v2,*v3,*v4,*v5) 3844c1414c8SBarry Smith #endif 3854c1414c8SBarry Smith 3864c1414c8SBarry Smith PetscFunctionBegin; 387e32f2f54SBarry Smith if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 3884c1414c8SBarry Smith node_max = a->inode.node_count; 3894c1414c8SBarry Smith ns = a->inode.size; /* Node Size array */ 3903649974fSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 3914c1414c8SBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 3924c1414c8SBarry Smith idx = a->j; 3934c1414c8SBarry Smith v1 = a->a; 3944c1414c8SBarry Smith ii = a->i; 3954c1414c8SBarry Smith 3964c1414c8SBarry Smith for (i = 0,row = 0; i< node_max; ++i) { 3974c1414c8SBarry Smith nsz = ns[i]; 3984c1414c8SBarry Smith n = ii[1] - ii[0]; 39998c9bda7SSatish Balay nonzerorow += (n>0)*nsz; 4004c1414c8SBarry Smith ii += nsz; 40150d8bf02SJed Brown PetscPrefetchBlock(idx+nsz*n,n,0,PETSC_PREFETCH_HINT_NTA); /* Prefetch the indices for the block row after the current one */ 40250d8bf02SJed Brown PetscPrefetchBlock(v1+nsz*n,nsz*n,0,PETSC_PREFETCH_HINT_NTA); /* Prefetch the values for the block row after the current one */ 4034c1414c8SBarry Smith sz = n; /* No of non zeros in this row */ 4044c1414c8SBarry Smith /* Switch on the size of Node */ 4054c1414c8SBarry Smith switch (nsz) { /* Each loop in 'case' is unrolled */ 4064c1414c8SBarry Smith case 1: 40775567043SBarry Smith sum1 = 0.; 4084c1414c8SBarry Smith 4094c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 4104c1414c8SBarry Smith i1 = idx[0]; /* The instructions are ordered to */ 4114c1414c8SBarry Smith i2 = idx[1]; /* make the compiler's job easy */ 4124c1414c8SBarry Smith idx += 2; 4134c1414c8SBarry Smith tmp0 = x[i1]; 4144c1414c8SBarry Smith tmp1 = x[i2]; 4154c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 4164c1414c8SBarry Smith } 4174c1414c8SBarry Smith 4184c1414c8SBarry Smith if (n == sz-1) { /* Take care of the last nonzero */ 4194c1414c8SBarry Smith tmp0 = x[*idx++]; 4204c1414c8SBarry Smith sum1 += *v1++ *tmp0; 4214c1414c8SBarry Smith } 4224c1414c8SBarry Smith y[row++]=sum1; 4234c1414c8SBarry Smith break; 4244c1414c8SBarry Smith case 2: 42575567043SBarry Smith sum1 = 0.; 42675567043SBarry Smith sum2 = 0.; 4274c1414c8SBarry Smith v2 = v1 + n; 4284c1414c8SBarry Smith 4294c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 4304c1414c8SBarry Smith i1 = idx[0]; 4314c1414c8SBarry Smith i2 = idx[1]; 4324c1414c8SBarry Smith idx += 2; 4334c1414c8SBarry Smith tmp0 = x[i1]; 4344c1414c8SBarry Smith tmp1 = x[i2]; 4354c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 4364c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 4374c1414c8SBarry Smith } 4384c1414c8SBarry Smith if (n == sz-1) { 4394c1414c8SBarry Smith tmp0 = x[*idx++]; 4404c1414c8SBarry Smith sum1 += *v1++ * tmp0; 4414c1414c8SBarry Smith sum2 += *v2++ * tmp0; 4424c1414c8SBarry Smith } 4434c1414c8SBarry Smith y[row++]=sum1; 4444c1414c8SBarry Smith y[row++]=sum2; 4454c1414c8SBarry Smith v1 =v2; /* Since the next block to be processed starts there*/ 4464c1414c8SBarry Smith idx +=sz; 4474c1414c8SBarry Smith break; 4484c1414c8SBarry Smith case 3: 44975567043SBarry Smith sum1 = 0.; 45075567043SBarry Smith sum2 = 0.; 45175567043SBarry Smith sum3 = 0.; 4524c1414c8SBarry Smith v2 = v1 + n; 4534c1414c8SBarry Smith v3 = v2 + n; 4544c1414c8SBarry Smith 4554c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 4564c1414c8SBarry Smith i1 = idx[0]; 4574c1414c8SBarry Smith i2 = idx[1]; 4584c1414c8SBarry Smith idx += 2; 4594c1414c8SBarry Smith tmp0 = x[i1]; 4604c1414c8SBarry Smith tmp1 = x[i2]; 4614c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 4624c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 4634c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 4644c1414c8SBarry Smith } 4654c1414c8SBarry Smith if (n == sz-1) { 4664c1414c8SBarry Smith tmp0 = x[*idx++]; 4674c1414c8SBarry Smith sum1 += *v1++ * tmp0; 4684c1414c8SBarry Smith sum2 += *v2++ * tmp0; 4694c1414c8SBarry Smith sum3 += *v3++ * tmp0; 4704c1414c8SBarry Smith } 4714c1414c8SBarry Smith y[row++]=sum1; 4724c1414c8SBarry Smith y[row++]=sum2; 4734c1414c8SBarry Smith y[row++]=sum3; 4744c1414c8SBarry Smith v1 =v3; /* Since the next block to be processed starts there*/ 4754c1414c8SBarry Smith idx +=2*sz; 4764c1414c8SBarry Smith break; 4774c1414c8SBarry Smith case 4: 47875567043SBarry Smith sum1 = 0.; 47975567043SBarry Smith sum2 = 0.; 48075567043SBarry Smith sum3 = 0.; 48175567043SBarry Smith sum4 = 0.; 4824c1414c8SBarry Smith v2 = v1 + n; 4834c1414c8SBarry Smith v3 = v2 + n; 4844c1414c8SBarry Smith v4 = v3 + n; 4854c1414c8SBarry Smith 4864c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 4874c1414c8SBarry Smith i1 = idx[0]; 4884c1414c8SBarry Smith i2 = idx[1]; 4894c1414c8SBarry Smith idx += 2; 4904c1414c8SBarry Smith tmp0 = x[i1]; 4914c1414c8SBarry Smith tmp1 = x[i2]; 4924c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 4934c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 4944c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 4954c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 4964c1414c8SBarry Smith } 4974c1414c8SBarry Smith if (n == sz-1) { 4984c1414c8SBarry Smith tmp0 = x[*idx++]; 4994c1414c8SBarry Smith sum1 += *v1++ * tmp0; 5004c1414c8SBarry Smith sum2 += *v2++ * tmp0; 5014c1414c8SBarry Smith sum3 += *v3++ * tmp0; 5024c1414c8SBarry Smith sum4 += *v4++ * tmp0; 5034c1414c8SBarry Smith } 5044c1414c8SBarry Smith y[row++]=sum1; 5054c1414c8SBarry Smith y[row++]=sum2; 5064c1414c8SBarry Smith y[row++]=sum3; 5074c1414c8SBarry Smith y[row++]=sum4; 5084c1414c8SBarry Smith v1 =v4; /* Since the next block to be processed starts there*/ 5094c1414c8SBarry Smith idx +=3*sz; 5104c1414c8SBarry Smith break; 5114c1414c8SBarry Smith case 5: 51275567043SBarry Smith sum1 = 0.; 51375567043SBarry Smith sum2 = 0.; 51475567043SBarry Smith sum3 = 0.; 51575567043SBarry Smith sum4 = 0.; 51675567043SBarry Smith sum5 = 0.; 5174c1414c8SBarry Smith v2 = v1 + n; 5184c1414c8SBarry Smith v3 = v2 + n; 5194c1414c8SBarry Smith v4 = v3 + n; 5204c1414c8SBarry Smith v5 = v4 + n; 5214c1414c8SBarry Smith 5224c1414c8SBarry Smith for (n = 0; n<sz-1; n+=2) { 5234c1414c8SBarry Smith i1 = idx[0]; 5244c1414c8SBarry Smith i2 = idx[1]; 5254c1414c8SBarry Smith idx += 2; 5264c1414c8SBarry Smith tmp0 = x[i1]; 5274c1414c8SBarry Smith tmp1 = x[i2]; 5284c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 5294c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 5304c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 5314c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 5324c1414c8SBarry Smith sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2; 5334c1414c8SBarry Smith } 5344c1414c8SBarry Smith if (n == sz-1) { 5354c1414c8SBarry Smith tmp0 = x[*idx++]; 5364c1414c8SBarry Smith sum1 += *v1++ * tmp0; 5374c1414c8SBarry Smith sum2 += *v2++ * tmp0; 5384c1414c8SBarry Smith sum3 += *v3++ * tmp0; 5394c1414c8SBarry Smith sum4 += *v4++ * tmp0; 5404c1414c8SBarry Smith sum5 += *v5++ * tmp0; 5414c1414c8SBarry Smith } 5424c1414c8SBarry Smith y[row++]=sum1; 5434c1414c8SBarry Smith y[row++]=sum2; 5444c1414c8SBarry Smith y[row++]=sum3; 5454c1414c8SBarry Smith y[row++]=sum4; 5464c1414c8SBarry Smith y[row++]=sum5; 5474c1414c8SBarry Smith v1 =v5; /* Since the next block to be processed starts there */ 5484c1414c8SBarry Smith idx +=4*sz; 5494c1414c8SBarry Smith break; 5504c1414c8SBarry Smith default: 551e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported"); 5524c1414c8SBarry Smith } 5534c1414c8SBarry Smith } 5543649974fSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 5554c1414c8SBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 556dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - nonzerorow);CHKERRQ(ierr); 5574c1414c8SBarry Smith PetscFunctionReturn(0); 5584c1414c8SBarry Smith } 5594c1414c8SBarry Smith /* ----------------------------------------------------------- */ 5604108e4d5SBarry Smith /* Almost same code as the MatMult_SeqAIJ_Inode() */ 561b215bc84SStefano Zampini PetscErrorCode MatMultAdd_SeqAIJ_Inode(Mat A,Vec xx,Vec zz,Vec yy) 5624c1414c8SBarry Smith { 5634c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 5644c1414c8SBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5,tmp0,tmp1; 5658758e1faSBarry Smith const MatScalar *v1,*v2,*v3,*v4,*v5; 5668758e1faSBarry Smith const PetscScalar *x; 5678758e1faSBarry Smith PetscScalar *y,*z,*zt; 5684c1414c8SBarry Smith PetscErrorCode ierr; 5698758e1faSBarry Smith PetscInt i1,i2,n,i,row,node_max,nsz,sz; 5708758e1faSBarry Smith const PetscInt *idx,*ns,*ii; 5714c1414c8SBarry Smith 5724c1414c8SBarry Smith PetscFunctionBegin; 573e32f2f54SBarry Smith if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 5744c1414c8SBarry Smith node_max = a->inode.node_count; 5754c1414c8SBarry Smith ns = a->inode.size; /* Node Size array */ 5762205254eSKarl Rupp 5778758e1faSBarry Smith ierr = VecGetArrayRead(xx,&x);CHKERRQ(ierr); 578f95e332cSBarry Smith ierr = VecGetArrayPair(zz,yy,&z,&y);CHKERRQ(ierr); 5794c1414c8SBarry Smith zt = z; 5804c1414c8SBarry Smith 5814c1414c8SBarry Smith idx = a->j; 5824c1414c8SBarry Smith v1 = a->a; 5834c1414c8SBarry Smith ii = a->i; 5844c1414c8SBarry Smith 5854c1414c8SBarry Smith for (i = 0,row = 0; i< node_max; ++i) { 5864c1414c8SBarry Smith nsz = ns[i]; 5874c1414c8SBarry Smith n = ii[1] - ii[0]; 5884c1414c8SBarry Smith ii += nsz; 5894c1414c8SBarry Smith sz = n; /* No of non zeros in this row */ 5904c1414c8SBarry Smith /* Switch on the size of Node */ 5914c1414c8SBarry Smith switch (nsz) { /* Each loop in 'case' is unrolled */ 5924c1414c8SBarry Smith case 1: 5934c1414c8SBarry Smith sum1 = *zt++; 5944c1414c8SBarry Smith 5954c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 5964c1414c8SBarry Smith i1 = idx[0]; /* The instructions are ordered to */ 5974c1414c8SBarry Smith i2 = idx[1]; /* make the compiler's job easy */ 5984c1414c8SBarry Smith idx += 2; 5994c1414c8SBarry Smith tmp0 = x[i1]; 6004c1414c8SBarry Smith tmp1 = x[i2]; 6014c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 6024c1414c8SBarry Smith } 6034c1414c8SBarry Smith 6044c1414c8SBarry Smith if (n == sz-1) { /* Take care of the last nonzero */ 6054c1414c8SBarry Smith tmp0 = x[*idx++]; 6064c1414c8SBarry Smith sum1 += *v1++ * tmp0; 6074c1414c8SBarry Smith } 6084c1414c8SBarry Smith y[row++]=sum1; 6094c1414c8SBarry Smith break; 6104c1414c8SBarry Smith case 2: 6114c1414c8SBarry Smith sum1 = *zt++; 6124c1414c8SBarry Smith sum2 = *zt++; 6134c1414c8SBarry Smith v2 = v1 + n; 6144c1414c8SBarry Smith 6154c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 6164c1414c8SBarry Smith i1 = idx[0]; 6174c1414c8SBarry Smith i2 = idx[1]; 6184c1414c8SBarry Smith idx += 2; 6194c1414c8SBarry Smith tmp0 = x[i1]; 6204c1414c8SBarry Smith tmp1 = x[i2]; 6214c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 6224c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 6234c1414c8SBarry Smith } 6244c1414c8SBarry Smith if (n == sz-1) { 6254c1414c8SBarry Smith tmp0 = x[*idx++]; 6264c1414c8SBarry Smith sum1 += *v1++ * tmp0; 6274c1414c8SBarry Smith sum2 += *v2++ * tmp0; 6284c1414c8SBarry Smith } 6294c1414c8SBarry Smith y[row++]=sum1; 6304c1414c8SBarry Smith y[row++]=sum2; 6314c1414c8SBarry Smith v1 =v2; /* Since the next block to be processed starts there*/ 6324c1414c8SBarry Smith idx +=sz; 6334c1414c8SBarry Smith break; 6344c1414c8SBarry Smith case 3: 6354c1414c8SBarry Smith sum1 = *zt++; 6364c1414c8SBarry Smith sum2 = *zt++; 6374c1414c8SBarry Smith sum3 = *zt++; 6384c1414c8SBarry Smith v2 = v1 + n; 6394c1414c8SBarry Smith v3 = v2 + n; 6404c1414c8SBarry Smith 6414c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 6424c1414c8SBarry Smith i1 = idx[0]; 6434c1414c8SBarry Smith i2 = idx[1]; 6444c1414c8SBarry Smith idx += 2; 6454c1414c8SBarry Smith tmp0 = x[i1]; 6464c1414c8SBarry Smith tmp1 = x[i2]; 6474c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 6484c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 6494c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 6504c1414c8SBarry Smith } 6514c1414c8SBarry Smith if (n == sz-1) { 6524c1414c8SBarry Smith tmp0 = x[*idx++]; 6534c1414c8SBarry Smith sum1 += *v1++ * tmp0; 6544c1414c8SBarry Smith sum2 += *v2++ * tmp0; 6554c1414c8SBarry Smith sum3 += *v3++ * tmp0; 6564c1414c8SBarry Smith } 6574c1414c8SBarry Smith y[row++]=sum1; 6584c1414c8SBarry Smith y[row++]=sum2; 6594c1414c8SBarry Smith y[row++]=sum3; 6604c1414c8SBarry Smith v1 =v3; /* Since the next block to be processed starts there*/ 6614c1414c8SBarry Smith idx +=2*sz; 6624c1414c8SBarry Smith break; 6634c1414c8SBarry Smith case 4: 6644c1414c8SBarry Smith sum1 = *zt++; 6654c1414c8SBarry Smith sum2 = *zt++; 6664c1414c8SBarry Smith sum3 = *zt++; 6674c1414c8SBarry Smith sum4 = *zt++; 6684c1414c8SBarry Smith v2 = v1 + n; 6694c1414c8SBarry Smith v3 = v2 + n; 6704c1414c8SBarry Smith v4 = v3 + n; 6714c1414c8SBarry Smith 6724c1414c8SBarry Smith for (n = 0; n< sz-1; n+=2) { 6734c1414c8SBarry Smith i1 = idx[0]; 6744c1414c8SBarry Smith i2 = idx[1]; 6754c1414c8SBarry Smith idx += 2; 6764c1414c8SBarry Smith tmp0 = x[i1]; 6774c1414c8SBarry Smith tmp1 = x[i2]; 6784c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 6794c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 6804c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 6814c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 6824c1414c8SBarry Smith } 6834c1414c8SBarry Smith if (n == sz-1) { 6844c1414c8SBarry Smith tmp0 = x[*idx++]; 6854c1414c8SBarry Smith sum1 += *v1++ * tmp0; 6864c1414c8SBarry Smith sum2 += *v2++ * tmp0; 6874c1414c8SBarry Smith sum3 += *v3++ * tmp0; 6884c1414c8SBarry Smith sum4 += *v4++ * tmp0; 6894c1414c8SBarry Smith } 6904c1414c8SBarry Smith y[row++]=sum1; 6914c1414c8SBarry Smith y[row++]=sum2; 6924c1414c8SBarry Smith y[row++]=sum3; 6934c1414c8SBarry Smith y[row++]=sum4; 6944c1414c8SBarry Smith v1 =v4; /* Since the next block to be processed starts there*/ 6954c1414c8SBarry Smith idx +=3*sz; 6964c1414c8SBarry Smith break; 6974c1414c8SBarry Smith case 5: 6984c1414c8SBarry Smith sum1 = *zt++; 6994c1414c8SBarry Smith sum2 = *zt++; 7004c1414c8SBarry Smith sum3 = *zt++; 7014c1414c8SBarry Smith sum4 = *zt++; 7024c1414c8SBarry Smith sum5 = *zt++; 7034c1414c8SBarry Smith v2 = v1 + n; 7044c1414c8SBarry Smith v3 = v2 + n; 7054c1414c8SBarry Smith v4 = v3 + n; 7064c1414c8SBarry Smith v5 = v4 + n; 7074c1414c8SBarry Smith 7084c1414c8SBarry Smith for (n = 0; n<sz-1; n+=2) { 7094c1414c8SBarry Smith i1 = idx[0]; 7104c1414c8SBarry Smith i2 = idx[1]; 7114c1414c8SBarry Smith idx += 2; 7124c1414c8SBarry Smith tmp0 = x[i1]; 7134c1414c8SBarry Smith tmp1 = x[i2]; 7144c1414c8SBarry Smith sum1 += v1[0] * tmp0 + v1[1] *tmp1; v1 += 2; 7154c1414c8SBarry Smith sum2 += v2[0] * tmp0 + v2[1] *tmp1; v2 += 2; 7164c1414c8SBarry Smith sum3 += v3[0] * tmp0 + v3[1] *tmp1; v3 += 2; 7174c1414c8SBarry Smith sum4 += v4[0] * tmp0 + v4[1] *tmp1; v4 += 2; 7184c1414c8SBarry Smith sum5 += v5[0] * tmp0 + v5[1] *tmp1; v5 += 2; 7194c1414c8SBarry Smith } 7204c1414c8SBarry Smith if (n == sz-1) { 7214c1414c8SBarry Smith tmp0 = x[*idx++]; 7224c1414c8SBarry Smith sum1 += *v1++ * tmp0; 7234c1414c8SBarry Smith sum2 += *v2++ * tmp0; 7244c1414c8SBarry Smith sum3 += *v3++ * tmp0; 7254c1414c8SBarry Smith sum4 += *v4++ * tmp0; 7264c1414c8SBarry Smith sum5 += *v5++ * tmp0; 7274c1414c8SBarry Smith } 7284c1414c8SBarry Smith y[row++]=sum1; 7294c1414c8SBarry Smith y[row++]=sum2; 7304c1414c8SBarry Smith y[row++]=sum3; 7314c1414c8SBarry Smith y[row++]=sum4; 7324c1414c8SBarry Smith y[row++]=sum5; 7334c1414c8SBarry Smith v1 =v5; /* Since the next block to be processed starts there */ 7344c1414c8SBarry Smith idx +=4*sz; 7354c1414c8SBarry Smith break; 7364c1414c8SBarry Smith default: 737e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported"); 7384c1414c8SBarry Smith } 7394c1414c8SBarry Smith } 7408758e1faSBarry Smith ierr = VecRestoreArrayRead(xx,&x);CHKERRQ(ierr); 741f95e332cSBarry Smith ierr = VecRestoreArrayPair(zz,yy,&z,&y);CHKERRQ(ierr); 742dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 7434c1414c8SBarry Smith PetscFunctionReturn(0); 7444c1414c8SBarry Smith } 7454c1414c8SBarry Smith 7464c1414c8SBarry Smith /* ----------------------------------------------------------- */ 747ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_Inode_inplace(Mat A,Vec bb,Vec xx) 7484c1414c8SBarry Smith { 7494c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 7504c1414c8SBarry Smith IS iscol = a->col,isrow = a->row; 7514c1414c8SBarry Smith PetscErrorCode ierr; 7525d0c19d7SBarry Smith const PetscInt *r,*c,*rout,*cout; 7538758e1faSBarry Smith PetscInt i,j,n = A->rmap->n,nz; 7548758e1faSBarry Smith PetscInt node_max,*ns,row,nsz,aii,i0,i1; 7558758e1faSBarry Smith const PetscInt *ai = a->i,*a_j = a->j,*vi,*ad,*aj; 756d9fead3dSBarry Smith PetscScalar *x,*tmp,*tmps,tmp0,tmp1; 757d9fead3dSBarry Smith PetscScalar sum1,sum2,sum3,sum4,sum5; 758dd6ea824SBarry Smith const MatScalar *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa; 759dd6ea824SBarry Smith const PetscScalar *b; 7604c1414c8SBarry Smith 7614c1414c8SBarry Smith PetscFunctionBegin; 762e32f2f54SBarry Smith if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 7634c1414c8SBarry Smith node_max = a->inode.node_count; 7644c1414c8SBarry Smith ns = a->inode.size; /* Node Size array */ 7654c1414c8SBarry Smith 7663649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 767fdc842d1SBarry Smith ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr); 7684c1414c8SBarry Smith tmp = a->solve_work; 7694c1414c8SBarry Smith 7704c1414c8SBarry Smith ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 7714c1414c8SBarry Smith ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1); 7724c1414c8SBarry Smith 7734c1414c8SBarry Smith /* forward solve the lower triangular */ 7744c1414c8SBarry Smith tmps = tmp; 7754c1414c8SBarry Smith aa = a_a; 7764c1414c8SBarry Smith aj = a_j; 7774c1414c8SBarry Smith ad = a->diag; 7784c1414c8SBarry Smith 7794c1414c8SBarry Smith for (i = 0,row = 0; i< node_max; ++i) { 7804c1414c8SBarry Smith nsz = ns[i]; 7814c1414c8SBarry Smith aii = ai[row]; 7824c1414c8SBarry Smith v1 = aa + aii; 7834c1414c8SBarry Smith vi = aj + aii; 7844c1414c8SBarry Smith nz = ad[row]- aii; 78526549573SJed Brown if (i < node_max-1) { 78626549573SJed Brown /* Prefetch the block after the current one, the prefetch itself can't cause a memory error, 78726549573SJed Brown * but our indexing to determine it's size could. */ 78850d8bf02SJed Brown PetscPrefetchBlock(aj+ai[row+nsz],ad[row+nsz]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); /* indices */ 78926549573SJed Brown /* In my tests, it seems to be better to fetch entire rows instead of just the lower-triangular part */ 79050d8bf02SJed Brown PetscPrefetchBlock(aa+ai[row+nsz],ad[row+nsz+ns[i+1]-1]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); 79126549573SJed 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); */ 79226549573SJed Brown } 7934c1414c8SBarry Smith 7944c1414c8SBarry Smith switch (nsz) { /* Each loop in 'case' is unrolled */ 7954c1414c8SBarry Smith case 1: 7964c1414c8SBarry Smith sum1 = b[*r++]; 7974c1414c8SBarry Smith for (j=0; j<nz-1; j+=2) { 7984c1414c8SBarry Smith i0 = vi[0]; 7994c1414c8SBarry Smith i1 = vi[1]; 8004c1414c8SBarry Smith vi +=2; 8014c1414c8SBarry Smith tmp0 = tmps[i0]; 8024c1414c8SBarry Smith tmp1 = tmps[i1]; 8034c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 8044c1414c8SBarry Smith } 8054c1414c8SBarry Smith if (j == nz-1) { 8064c1414c8SBarry Smith tmp0 = tmps[*vi++]; 8074c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 8084c1414c8SBarry Smith } 8094c1414c8SBarry Smith tmp[row++]=sum1; 8104c1414c8SBarry Smith break; 8114c1414c8SBarry Smith case 2: 8124c1414c8SBarry Smith sum1 = b[*r++]; 8134c1414c8SBarry Smith sum2 = b[*r++]; 8144c1414c8SBarry Smith v2 = aa + ai[row+1]; 8154c1414c8SBarry Smith 8164c1414c8SBarry Smith for (j=0; j<nz-1; j+=2) { 8174c1414c8SBarry Smith i0 = vi[0]; 8184c1414c8SBarry Smith i1 = vi[1]; 8194c1414c8SBarry Smith vi +=2; 8204c1414c8SBarry Smith tmp0 = tmps[i0]; 8214c1414c8SBarry Smith tmp1 = tmps[i1]; 8224c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 8234c1414c8SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 8244c1414c8SBarry Smith } 8254c1414c8SBarry Smith if (j == nz-1) { 8264c1414c8SBarry Smith tmp0 = tmps[*vi++]; 8274c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 8284c1414c8SBarry Smith sum2 -= *v2++ *tmp0; 8294c1414c8SBarry Smith } 8304c1414c8SBarry Smith sum2 -= *v2++ *sum1; 8314c1414c8SBarry Smith tmp[row++]=sum1; 8324c1414c8SBarry Smith tmp[row++]=sum2; 8334c1414c8SBarry Smith break; 8344c1414c8SBarry Smith case 3: 8354c1414c8SBarry Smith sum1 = b[*r++]; 8364c1414c8SBarry Smith sum2 = b[*r++]; 8374c1414c8SBarry Smith sum3 = b[*r++]; 8384c1414c8SBarry Smith v2 = aa + ai[row+1]; 8394c1414c8SBarry Smith v3 = aa + ai[row+2]; 8404c1414c8SBarry Smith 8414c1414c8SBarry Smith for (j=0; j<nz-1; j+=2) { 8424c1414c8SBarry Smith i0 = vi[0]; 8434c1414c8SBarry Smith i1 = vi[1]; 8444c1414c8SBarry Smith vi +=2; 8454c1414c8SBarry Smith tmp0 = tmps[i0]; 8464c1414c8SBarry Smith tmp1 = tmps[i1]; 8474c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 8484c1414c8SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 8494c1414c8SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 8504c1414c8SBarry Smith } 8514c1414c8SBarry Smith if (j == nz-1) { 8524c1414c8SBarry Smith tmp0 = tmps[*vi++]; 8534c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 8544c1414c8SBarry Smith sum2 -= *v2++ *tmp0; 8554c1414c8SBarry Smith sum3 -= *v3++ *tmp0; 8564c1414c8SBarry Smith } 8574c1414c8SBarry Smith sum2 -= *v2++ * sum1; 8584c1414c8SBarry Smith sum3 -= *v3++ * sum1; 8594c1414c8SBarry Smith sum3 -= *v3++ * sum2; 8602205254eSKarl Rupp 8614c1414c8SBarry Smith tmp[row++]=sum1; 8624c1414c8SBarry Smith tmp[row++]=sum2; 8634c1414c8SBarry Smith tmp[row++]=sum3; 8644c1414c8SBarry Smith break; 8654c1414c8SBarry Smith 8664c1414c8SBarry Smith case 4: 8674c1414c8SBarry Smith sum1 = b[*r++]; 8684c1414c8SBarry Smith sum2 = b[*r++]; 8694c1414c8SBarry Smith sum3 = b[*r++]; 8704c1414c8SBarry Smith sum4 = b[*r++]; 8714c1414c8SBarry Smith v2 = aa + ai[row+1]; 8724c1414c8SBarry Smith v3 = aa + ai[row+2]; 8734c1414c8SBarry Smith v4 = aa + ai[row+3]; 8744c1414c8SBarry Smith 8754c1414c8SBarry Smith for (j=0; j<nz-1; j+=2) { 8764c1414c8SBarry Smith i0 = vi[0]; 8774c1414c8SBarry Smith i1 = vi[1]; 8784c1414c8SBarry Smith vi +=2; 8794c1414c8SBarry Smith tmp0 = tmps[i0]; 8804c1414c8SBarry Smith tmp1 = tmps[i1]; 8814c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 8824c1414c8SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 8834c1414c8SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 8844c1414c8SBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 8854c1414c8SBarry Smith } 8864c1414c8SBarry Smith if (j == nz-1) { 8874c1414c8SBarry Smith tmp0 = tmps[*vi++]; 8884c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 8894c1414c8SBarry Smith sum2 -= *v2++ *tmp0; 8904c1414c8SBarry Smith sum3 -= *v3++ *tmp0; 8914c1414c8SBarry Smith sum4 -= *v4++ *tmp0; 8924c1414c8SBarry Smith } 8934c1414c8SBarry Smith sum2 -= *v2++ * sum1; 8944c1414c8SBarry Smith sum3 -= *v3++ * sum1; 8954c1414c8SBarry Smith sum4 -= *v4++ * sum1; 8964c1414c8SBarry Smith sum3 -= *v3++ * sum2; 8974c1414c8SBarry Smith sum4 -= *v4++ * sum2; 8984c1414c8SBarry Smith sum4 -= *v4++ * sum3; 8994c1414c8SBarry Smith 9004c1414c8SBarry Smith tmp[row++]=sum1; 9014c1414c8SBarry Smith tmp[row++]=sum2; 9024c1414c8SBarry Smith tmp[row++]=sum3; 9034c1414c8SBarry Smith tmp[row++]=sum4; 9044c1414c8SBarry Smith break; 9054c1414c8SBarry Smith case 5: 9064c1414c8SBarry Smith sum1 = b[*r++]; 9074c1414c8SBarry Smith sum2 = b[*r++]; 9084c1414c8SBarry Smith sum3 = b[*r++]; 9094c1414c8SBarry Smith sum4 = b[*r++]; 9104c1414c8SBarry Smith sum5 = b[*r++]; 9114c1414c8SBarry Smith v2 = aa + ai[row+1]; 9124c1414c8SBarry Smith v3 = aa + ai[row+2]; 9134c1414c8SBarry Smith v4 = aa + ai[row+3]; 9144c1414c8SBarry Smith v5 = aa + ai[row+4]; 9154c1414c8SBarry Smith 9164c1414c8SBarry Smith for (j=0; j<nz-1; j+=2) { 9174c1414c8SBarry Smith i0 = vi[0]; 9184c1414c8SBarry Smith i1 = vi[1]; 9194c1414c8SBarry Smith vi +=2; 9204c1414c8SBarry Smith tmp0 = tmps[i0]; 9214c1414c8SBarry Smith tmp1 = tmps[i1]; 9224c1414c8SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 9234c1414c8SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 9244c1414c8SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 9254c1414c8SBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 9264c1414c8SBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 9274c1414c8SBarry Smith } 9284c1414c8SBarry Smith if (j == nz-1) { 9294c1414c8SBarry Smith tmp0 = tmps[*vi++]; 9304c1414c8SBarry Smith sum1 -= *v1++ *tmp0; 9314c1414c8SBarry Smith sum2 -= *v2++ *tmp0; 9324c1414c8SBarry Smith sum3 -= *v3++ *tmp0; 9334c1414c8SBarry Smith sum4 -= *v4++ *tmp0; 9344c1414c8SBarry Smith sum5 -= *v5++ *tmp0; 9354c1414c8SBarry Smith } 9364c1414c8SBarry Smith 9374c1414c8SBarry Smith sum2 -= *v2++ * sum1; 9384c1414c8SBarry Smith sum3 -= *v3++ * sum1; 9394c1414c8SBarry Smith sum4 -= *v4++ * sum1; 9404c1414c8SBarry Smith sum5 -= *v5++ * sum1; 9414c1414c8SBarry Smith sum3 -= *v3++ * sum2; 9424c1414c8SBarry Smith sum4 -= *v4++ * sum2; 9434c1414c8SBarry Smith sum5 -= *v5++ * sum2; 9444c1414c8SBarry Smith sum4 -= *v4++ * sum3; 9454c1414c8SBarry Smith sum5 -= *v5++ * sum3; 9464c1414c8SBarry Smith sum5 -= *v5++ * sum4; 9474c1414c8SBarry Smith 9484c1414c8SBarry Smith tmp[row++]=sum1; 9494c1414c8SBarry Smith tmp[row++]=sum2; 9504c1414c8SBarry Smith tmp[row++]=sum3; 9514c1414c8SBarry Smith tmp[row++]=sum4; 9524c1414c8SBarry Smith tmp[row++]=sum5; 9534c1414c8SBarry Smith break; 9544c1414c8SBarry Smith default: 955e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n"); 9564c1414c8SBarry Smith } 9574c1414c8SBarry Smith } 9584c1414c8SBarry Smith /* backward solve the upper triangular */ 9594c1414c8SBarry Smith for (i=node_max -1,row = n-1; i>=0; i--) { 9604c1414c8SBarry Smith nsz = ns[i]; 9614c1414c8SBarry Smith aii = ai[row+1] -1; 9624c1414c8SBarry Smith v1 = aa + aii; 9634c1414c8SBarry Smith vi = aj + aii; 9644c1414c8SBarry Smith nz = aii- ad[row]; 9654c1414c8SBarry Smith switch (nsz) { /* Each loop in 'case' is unrolled */ 9664c1414c8SBarry Smith case 1: 9674c1414c8SBarry Smith sum1 = tmp[row]; 9684c1414c8SBarry Smith 9694c1414c8SBarry Smith for (j=nz; j>1; j-=2) { 9704c1414c8SBarry Smith vi -=2; 9714c1414c8SBarry Smith i0 = vi[2]; 9724c1414c8SBarry Smith i1 = vi[1]; 9734c1414c8SBarry Smith tmp0 = tmps[i0]; 9744c1414c8SBarry Smith tmp1 = tmps[i1]; 9754c1414c8SBarry Smith v1 -= 2; 9764c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 9774c1414c8SBarry Smith } 9784c1414c8SBarry Smith if (j==1) { 9794c1414c8SBarry Smith tmp0 = tmps[*vi--]; 9804c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 9814c1414c8SBarry Smith } 9824c1414c8SBarry Smith x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 9834c1414c8SBarry Smith break; 9844c1414c8SBarry Smith case 2: 9854c1414c8SBarry Smith sum1 = tmp[row]; 9864c1414c8SBarry Smith sum2 = tmp[row -1]; 9874c1414c8SBarry Smith v2 = aa + ai[row]-1; 9884c1414c8SBarry Smith for (j=nz; j>1; j-=2) { 9894c1414c8SBarry Smith vi -=2; 9904c1414c8SBarry Smith i0 = vi[2]; 9914c1414c8SBarry Smith i1 = vi[1]; 9924c1414c8SBarry Smith tmp0 = tmps[i0]; 9934c1414c8SBarry Smith tmp1 = tmps[i1]; 9944c1414c8SBarry Smith v1 -= 2; 9954c1414c8SBarry Smith v2 -= 2; 9964c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 9974c1414c8SBarry Smith sum2 -= v2[2] * tmp0 + v2[1] * tmp1; 9984c1414c8SBarry Smith } 9994c1414c8SBarry Smith if (j==1) { 10004c1414c8SBarry Smith tmp0 = tmps[*vi--]; 10014c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 10024c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 10034c1414c8SBarry Smith } 10044c1414c8SBarry Smith 10054c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 10064c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 10074c1414c8SBarry Smith x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--; 10084c1414c8SBarry Smith break; 10094c1414c8SBarry Smith case 3: 10104c1414c8SBarry Smith sum1 = tmp[row]; 10114c1414c8SBarry Smith sum2 = tmp[row -1]; 10124c1414c8SBarry Smith sum3 = tmp[row -2]; 10134c1414c8SBarry Smith v2 = aa + ai[row]-1; 10144c1414c8SBarry Smith v3 = aa + ai[row -1]-1; 10154c1414c8SBarry Smith for (j=nz; j>1; j-=2) { 10164c1414c8SBarry Smith vi -=2; 10174c1414c8SBarry Smith i0 = vi[2]; 10184c1414c8SBarry Smith i1 = vi[1]; 10194c1414c8SBarry Smith tmp0 = tmps[i0]; 10204c1414c8SBarry Smith tmp1 = tmps[i1]; 10214c1414c8SBarry Smith v1 -= 2; 10224c1414c8SBarry Smith v2 -= 2; 10234c1414c8SBarry Smith v3 -= 2; 10244c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 10254c1414c8SBarry Smith sum2 -= v2[2] * tmp0 + v2[1] * tmp1; 10264c1414c8SBarry Smith sum3 -= v3[2] * tmp0 + v3[1] * tmp1; 10274c1414c8SBarry Smith } 10284c1414c8SBarry Smith if (j==1) { 10294c1414c8SBarry Smith tmp0 = tmps[*vi--]; 10304c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 10314c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 10324c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 10334c1414c8SBarry Smith } 10344c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 10354c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 10364c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 10374c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--; 10384c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 10394c1414c8SBarry Smith x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--; 10404c1414c8SBarry Smith 10414c1414c8SBarry Smith break; 10424c1414c8SBarry Smith case 4: 10434c1414c8SBarry Smith sum1 = tmp[row]; 10444c1414c8SBarry Smith sum2 = tmp[row -1]; 10454c1414c8SBarry Smith sum3 = tmp[row -2]; 10464c1414c8SBarry Smith sum4 = tmp[row -3]; 10474c1414c8SBarry Smith v2 = aa + ai[row]-1; 10484c1414c8SBarry Smith v3 = aa + ai[row -1]-1; 10494c1414c8SBarry Smith v4 = aa + ai[row -2]-1; 10504c1414c8SBarry Smith 10514c1414c8SBarry Smith for (j=nz; j>1; j-=2) { 10524c1414c8SBarry Smith vi -=2; 10534c1414c8SBarry Smith i0 = vi[2]; 10544c1414c8SBarry Smith i1 = vi[1]; 10554c1414c8SBarry Smith tmp0 = tmps[i0]; 10564c1414c8SBarry Smith tmp1 = tmps[i1]; 10574c1414c8SBarry Smith v1 -= 2; 10584c1414c8SBarry Smith v2 -= 2; 10594c1414c8SBarry Smith v3 -= 2; 10604c1414c8SBarry Smith v4 -= 2; 10614c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 10624c1414c8SBarry Smith sum2 -= v2[2] * tmp0 + v2[1] * tmp1; 10634c1414c8SBarry Smith sum3 -= v3[2] * tmp0 + v3[1] * tmp1; 10644c1414c8SBarry Smith sum4 -= v4[2] * tmp0 + v4[1] * tmp1; 10654c1414c8SBarry Smith } 10664c1414c8SBarry Smith if (j==1) { 10674c1414c8SBarry Smith tmp0 = tmps[*vi--]; 10684c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 10694c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 10704c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 10714c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 10724c1414c8SBarry Smith } 10734c1414c8SBarry Smith 10744c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 10754c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 10764c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 10774c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 10784c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--; 10794c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 10804c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 10814c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--; 10824c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 10834c1414c8SBarry Smith x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--; 10844c1414c8SBarry Smith break; 10854c1414c8SBarry Smith case 5: 10864c1414c8SBarry Smith sum1 = tmp[row]; 10874c1414c8SBarry Smith sum2 = tmp[row -1]; 10884c1414c8SBarry Smith sum3 = tmp[row -2]; 10894c1414c8SBarry Smith sum4 = tmp[row -3]; 10904c1414c8SBarry Smith sum5 = tmp[row -4]; 10914c1414c8SBarry Smith v2 = aa + ai[row]-1; 10924c1414c8SBarry Smith v3 = aa + ai[row -1]-1; 10934c1414c8SBarry Smith v4 = aa + ai[row -2]-1; 10944c1414c8SBarry Smith v5 = aa + ai[row -3]-1; 10954c1414c8SBarry Smith for (j=nz; j>1; j-=2) { 10964c1414c8SBarry Smith vi -= 2; 10974c1414c8SBarry Smith i0 = vi[2]; 10984c1414c8SBarry Smith i1 = vi[1]; 10994c1414c8SBarry Smith tmp0 = tmps[i0]; 11004c1414c8SBarry Smith tmp1 = tmps[i1]; 11014c1414c8SBarry Smith v1 -= 2; 11024c1414c8SBarry Smith v2 -= 2; 11034c1414c8SBarry Smith v3 -= 2; 11044c1414c8SBarry Smith v4 -= 2; 11054c1414c8SBarry Smith v5 -= 2; 11064c1414c8SBarry Smith sum1 -= v1[2] * tmp0 + v1[1] * tmp1; 11074c1414c8SBarry Smith sum2 -= v2[2] * tmp0 + v2[1] * tmp1; 11084c1414c8SBarry Smith sum3 -= v3[2] * tmp0 + v3[1] * tmp1; 11094c1414c8SBarry Smith sum4 -= v4[2] * tmp0 + v4[1] * tmp1; 11104c1414c8SBarry Smith sum5 -= v5[2] * tmp0 + v5[1] * tmp1; 11114c1414c8SBarry Smith } 11124c1414c8SBarry Smith if (j==1) { 11134c1414c8SBarry Smith tmp0 = tmps[*vi--]; 11144c1414c8SBarry Smith sum1 -= *v1-- * tmp0; 11154c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 11164c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11174c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11184c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11194c1414c8SBarry Smith } 11204c1414c8SBarry Smith 11214c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum1*a_a[ad[row]]; row--; 11224c1414c8SBarry Smith sum2 -= *v2-- * tmp0; 11234c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11244c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11254c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11264c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum2*a_a[ad[row]]; row--; 11274c1414c8SBarry Smith sum3 -= *v3-- * tmp0; 11284c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11294c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11304c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum3*a_a[ad[row]]; row--; 11314c1414c8SBarry Smith sum4 -= *v4-- * tmp0; 11324c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11334c1414c8SBarry Smith tmp0 = x[*c--] = tmp[row] = sum4*a_a[ad[row]]; row--; 11344c1414c8SBarry Smith sum5 -= *v5-- * tmp0; 11354c1414c8SBarry Smith x[*c--] = tmp[row] = sum5*a_a[ad[row]]; row--; 11364c1414c8SBarry Smith break; 11374c1414c8SBarry Smith default: 1138e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n"); 11394c1414c8SBarry Smith } 11404c1414c8SBarry Smith } 11414c1414c8SBarry Smith ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 11424c1414c8SBarry Smith ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 11433649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 1144fdc842d1SBarry Smith ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr); 1145dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 11464c1414c8SBarry Smith PetscFunctionReturn(0); 11474c1414c8SBarry Smith } 11484c1414c8SBarry Smith 11499a625307SHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_Inode(Mat B,Mat A,const MatFactorInfo *info) 115028f1b45aSHong Zhang { 115128f1b45aSHong Zhang Mat C =B; 115228f1b45aSHong Zhang Mat_SeqAIJ *a =(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ*)C->data; 115328f1b45aSHong Zhang IS isrow = b->row,isicol = b->icol; 115428f1b45aSHong Zhang PetscErrorCode ierr; 115528f1b45aSHong Zhang const PetscInt *r,*ic,*ics; 115628f1b45aSHong Zhang const PetscInt n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag; 115728f1b45aSHong Zhang PetscInt i,j,k,nz,nzL,row,*pj; 115828f1b45aSHong Zhang const PetscInt *ajtmp,*bjtmp; 11599877982aSShri Abhyankar MatScalar *pc,*pc1,*pc2,*pc3,*pc4,mul1,mul2,mul3,mul4,*pv,*rtmp1,*rtmp2,*rtmp3,*rtmp4; 11609877982aSShri Abhyankar const MatScalar *aa=a->a,*v,*v1,*v2,*v3,*v4; 116128f1b45aSHong Zhang FactorShiftCtx sctx; 11624f81c4b7SBarry Smith const PetscInt *ddiag; 116328f1b45aSHong Zhang PetscReal rs; 116428f1b45aSHong Zhang MatScalar d; 11654f81c4b7SBarry Smith PetscInt inod,nodesz,node_max,col; 11664f81c4b7SBarry Smith const PetscInt *ns; 116707b50cabSHong Zhang PetscInt *tmp_vec1,*tmp_vec2,*nsmap; 11680e95ead3SHong Zhang 116928f1b45aSHong Zhang PetscFunctionBegin; 117028f1b45aSHong Zhang /* MatPivotSetUp(): initialize shift context sctx */ 117128f1b45aSHong Zhang ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr); 117228f1b45aSHong Zhang 1173f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 117428f1b45aSHong Zhang ddiag = a->diag; 117528f1b45aSHong Zhang sctx.shift_top = info->zeropivot; 117628f1b45aSHong Zhang for (i=0; i<n; i++) { 117728f1b45aSHong Zhang /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 117828f1b45aSHong Zhang d = (aa)[ddiag[i]]; 117928f1b45aSHong Zhang rs = -PetscAbsScalar(d) - PetscRealPart(d); 118028f1b45aSHong Zhang v = aa+ai[i]; 118128f1b45aSHong Zhang nz = ai[i+1] - ai[i]; 11822205254eSKarl Rupp for (j=0; j<nz; j++) rs += PetscAbsScalar(v[j]); 118328f1b45aSHong Zhang if (rs>sctx.shift_top) sctx.shift_top = rs; 118428f1b45aSHong Zhang } 118528f1b45aSHong Zhang sctx.shift_top *= 1.1; 118628f1b45aSHong Zhang sctx.nshift_max = 5; 118728f1b45aSHong Zhang sctx.shift_lo = 0.; 118828f1b45aSHong Zhang sctx.shift_hi = 1.; 118928f1b45aSHong Zhang } 119028f1b45aSHong Zhang 119128f1b45aSHong Zhang ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 119228f1b45aSHong Zhang ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 119368785679SHong Zhang 11941795a4d1SJed Brown ierr = PetscCalloc4(n,&rtmp1,n,&rtmp2,n,&rtmp3,n,&rtmp4);CHKERRQ(ierr); 119528f1b45aSHong Zhang ics = ic; 119628f1b45aSHong Zhang 119728f1b45aSHong Zhang node_max = a->inode.node_count; 119828f1b45aSHong Zhang ns = a->inode.size; 1199e7e72b3dSBarry Smith if (!ns) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix without inode information"); 120028f1b45aSHong Zhang 12019877982aSShri Abhyankar /* If max inode size > 4, split it into two inodes.*/ 120268785679SHong Zhang /* also map the inode sizes according to the ordering */ 1203854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&tmp_vec1);CHKERRQ(ierr); 120468785679SHong Zhang for (i=0,j=0; i<node_max; ++i,++j) { 1205b1550197SShri Abhyankar if (ns[i] > 4) { 1206048b5e81SShri Abhyankar tmp_vec1[j] = 4; 120768785679SHong Zhang ++j; 120868785679SHong Zhang tmp_vec1[j] = ns[i] - tmp_vec1[j-1]; 120968785679SHong Zhang } else { 121068785679SHong Zhang tmp_vec1[j] = ns[i]; 121168785679SHong Zhang } 121268785679SHong Zhang } 121368785679SHong Zhang /* Use the correct node_max */ 121468785679SHong Zhang node_max = j; 121568785679SHong Zhang 121668785679SHong Zhang /* Now reorder the inode info based on mat re-ordering info */ 121768785679SHong Zhang /* First create a row -> inode_size_array_index map */ 1218854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&nsmap);CHKERRQ(ierr); 1219854ce69bSBarry Smith ierr = PetscMalloc1(node_max+1,&tmp_vec2);CHKERRQ(ierr); 122068785679SHong Zhang for (i=0,row=0; i<node_max; i++) { 122168785679SHong Zhang nodesz = tmp_vec1[i]; 122268785679SHong Zhang for (j=0; j<nodesz; j++,row++) { 122368785679SHong Zhang nsmap[row] = i; 122468785679SHong Zhang } 122568785679SHong Zhang } 122668785679SHong Zhang /* Using nsmap, create a reordered ns structure */ 122768785679SHong Zhang for (i=0,j=0; i< node_max; i++) { 122868785679SHong Zhang nodesz = tmp_vec1[nsmap[r[j]]]; /* here the reordered row_no is in r[] */ 122968785679SHong Zhang tmp_vec2[i] = nodesz; 123068785679SHong Zhang j += nodesz; 123168785679SHong Zhang } 123268785679SHong Zhang ierr = PetscFree(nsmap);CHKERRQ(ierr); 123368785679SHong Zhang ierr = PetscFree(tmp_vec1);CHKERRQ(ierr); 1234b89f182dSHong Zhang 123568785679SHong Zhang /* Now use the correct ns */ 123668785679SHong Zhang ns = tmp_vec2; 123768785679SHong Zhang 123828f1b45aSHong Zhang do { 123907b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 124028f1b45aSHong Zhang /* Now loop over each block-row, and do the factorization */ 124128f1b45aSHong Zhang for (inod=0,i=0; inod<node_max; inod++) { /* i: row index; inod: inode index */ 124228f1b45aSHong Zhang nodesz = ns[inod]; 124328f1b45aSHong Zhang 124428f1b45aSHong Zhang switch (nodesz) { 124528f1b45aSHong Zhang case 1: 124668785679SHong Zhang /*----------*/ 1247b89f182dSHong Zhang /* zero rtmp1 */ 124828f1b45aSHong Zhang /* L part */ 124928f1b45aSHong Zhang nz = bi[i+1] - bi[i]; 125028f1b45aSHong Zhang bjtmp = bj + bi[i]; 1251b89f182dSHong Zhang for (j=0; j<nz; j++) rtmp1[bjtmp[j]] = 0.0; 125228f1b45aSHong Zhang 125328f1b45aSHong Zhang /* U part */ 125428f1b45aSHong Zhang nz = bdiag[i]-bdiag[i+1]; 125528f1b45aSHong Zhang bjtmp = bj + bdiag[i+1]+1; 1256b89f182dSHong Zhang for (j=0; j<nz; j++) rtmp1[bjtmp[j]] = 0.0; 125728f1b45aSHong Zhang 125828f1b45aSHong Zhang /* load in initial (unfactored row) */ 125928f1b45aSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 126028f1b45aSHong Zhang ajtmp = aj + ai[r[i]]; 126128f1b45aSHong Zhang v = aa + ai[r[i]]; 12622205254eSKarl Rupp for (j=0; j<nz; j++) rtmp1[ics[ajtmp[j]]] = v[j]; 12632205254eSKarl Rupp 126428f1b45aSHong Zhang /* ZeropivotApply() */ 1265b89f182dSHong Zhang rtmp1[i] += sctx.shift_amount; /* shift the diagonal of the matrix */ 126628f1b45aSHong Zhang 126728f1b45aSHong Zhang /* elimination */ 126828f1b45aSHong Zhang bjtmp = bj + bi[i]; 126928f1b45aSHong Zhang row = *bjtmp++; 127028f1b45aSHong Zhang nzL = bi[i+1] - bi[i]; 127128f1b45aSHong Zhang for (k=0; k < nzL; k++) { 1272b89f182dSHong Zhang pc = rtmp1 + row; 127328f1b45aSHong Zhang if (*pc != 0.0) { 127428f1b45aSHong Zhang pv = b->a + bdiag[row]; 1275b89f182dSHong Zhang mul1 = *pc * (*pv); 1276b89f182dSHong Zhang *pc = mul1; 127728f1b45aSHong Zhang pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 127828f1b45aSHong Zhang pv = b->a + bdiag[row+1]+1; 127928f1b45aSHong Zhang nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 1280b89f182dSHong Zhang for (j=0; j<nz; j++) rtmp1[pj[j]] -= mul1 * pv[j]; 1281ca0c957dSBarry Smith ierr = PetscLogFlops(1+2.0*nz);CHKERRQ(ierr); 128228f1b45aSHong Zhang } 128328f1b45aSHong Zhang row = *bjtmp++; 128428f1b45aSHong Zhang } 128528f1b45aSHong Zhang 128628f1b45aSHong Zhang /* finished row so stick it into b->a */ 128728f1b45aSHong Zhang rs = 0.0; 128828f1b45aSHong Zhang /* L part */ 128928f1b45aSHong Zhang pv = b->a + bi[i]; 129028f1b45aSHong Zhang pj = b->j + bi[i]; 129128f1b45aSHong Zhang nz = bi[i+1] - bi[i]; 129228f1b45aSHong Zhang for (j=0; j<nz; j++) { 1293b89f182dSHong Zhang pv[j] = rtmp1[pj[j]]; rs += PetscAbsScalar(pv[j]); 129428f1b45aSHong Zhang } 129528f1b45aSHong Zhang 129628f1b45aSHong Zhang /* U part */ 129728f1b45aSHong Zhang pv = b->a + bdiag[i+1]+1; 129828f1b45aSHong Zhang pj = b->j + bdiag[i+1]+1; 129928f1b45aSHong Zhang nz = bdiag[i] - bdiag[i+1]-1; 130028f1b45aSHong Zhang for (j=0; j<nz; j++) { 1301b89f182dSHong Zhang pv[j] = rtmp1[pj[j]]; rs += PetscAbsScalar(pv[j]); 130228f1b45aSHong Zhang } 130328f1b45aSHong Zhang 1304b89f182dSHong Zhang /* Check zero pivot */ 130528f1b45aSHong Zhang sctx.rs = rs; 1306b89f182dSHong Zhang sctx.pv = rtmp1[i]; 13074cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr); 130807b50cabSHong Zhang if (sctx.newshift) break; 130928f1b45aSHong Zhang 131028f1b45aSHong Zhang /* Mark diagonal and invert diagonal for simplier triangular solves */ 131128f1b45aSHong Zhang pv = b->a + bdiag[i]; 1312b89f182dSHong Zhang *pv = 1.0/sctx.pv; /* sctx.pv = rtmp1[i]+shiftamount if shifttype==MAT_SHIFT_INBLOCKS */ 131328f1b45aSHong Zhang break; 131428f1b45aSHong Zhang 131528f1b45aSHong Zhang case 2: 131628f1b45aSHong Zhang /*----------*/ 1317b89f182dSHong Zhang /* zero rtmp1 and rtmp2 */ 131828f1b45aSHong Zhang /* L part */ 131928f1b45aSHong Zhang nz = bi[i+1] - bi[i]; 132028f1b45aSHong Zhang bjtmp = bj + bi[i]; 132128f1b45aSHong Zhang for (j=0; j<nz; j++) { 132268785679SHong Zhang col = bjtmp[j]; 1323b89f182dSHong Zhang rtmp1[col] = 0.0; rtmp2[col] = 0.0; 132428f1b45aSHong Zhang } 132528f1b45aSHong Zhang 132628f1b45aSHong Zhang /* U part */ 132728f1b45aSHong Zhang nz = bdiag[i]-bdiag[i+1]; 132828f1b45aSHong Zhang bjtmp = bj + bdiag[i+1]+1; 132928f1b45aSHong Zhang for (j=0; j<nz; j++) { 133068785679SHong Zhang col = bjtmp[j]; 1331b89f182dSHong Zhang rtmp1[col] = 0.0; rtmp2[col] = 0.0; 133228f1b45aSHong Zhang } 133328f1b45aSHong Zhang 133428f1b45aSHong Zhang /* load in initial (unfactored row) */ 133528f1b45aSHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 133628f1b45aSHong Zhang ajtmp = aj + ai[r[i]]; 133728f1b45aSHong Zhang v1 = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; 133828f1b45aSHong Zhang for (j=0; j<nz; j++) { 133968785679SHong Zhang col = ics[ajtmp[j]]; 1340b89f182dSHong Zhang rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; 134128f1b45aSHong Zhang } 134228f1b45aSHong Zhang /* ZeropivotApply(): shift the diagonal of the matrix */ 1343b89f182dSHong Zhang rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; 134428f1b45aSHong Zhang 134528f1b45aSHong Zhang /* elimination */ 134628f1b45aSHong Zhang bjtmp = bj + bi[i]; 134728f1b45aSHong Zhang row = *bjtmp++; /* pivot row */ 134828f1b45aSHong Zhang nzL = bi[i+1] - bi[i]; 134928f1b45aSHong Zhang for (k=0; k < nzL; k++) { 1350b89f182dSHong Zhang pc1 = rtmp1 + row; 1351b89f182dSHong Zhang pc2 = rtmp2 + row; 135228f1b45aSHong Zhang if (*pc1 != 0.0 || *pc2 != 0.0) { 135328f1b45aSHong Zhang pv = b->a + bdiag[row]; 1354b89f182dSHong Zhang mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); 1355b89f182dSHong Zhang *pc1 = mul1; *pc2 = mul2; 135628f1b45aSHong Zhang 135728f1b45aSHong Zhang pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 135828f1b45aSHong Zhang pv = b->a + bdiag[row+1]+1; 135928f1b45aSHong Zhang nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 136028f1b45aSHong Zhang for (j=0; j<nz; j++) { 136168785679SHong Zhang col = pj[j]; 1362b89f182dSHong Zhang rtmp1[col] -= mul1 * pv[j]; 1363b89f182dSHong Zhang rtmp2[col] -= mul2 * pv[j]; 136428f1b45aSHong Zhang } 1365ca0c957dSBarry Smith ierr = PetscLogFlops(2+4.0*nz);CHKERRQ(ierr); 136628f1b45aSHong Zhang } 136728f1b45aSHong Zhang row = *bjtmp++; 136828f1b45aSHong Zhang } 136928f1b45aSHong Zhang 1370b89f182dSHong Zhang /* finished row i; check zero pivot, then stick row i into b->a */ 137128f1b45aSHong Zhang rs = 0.0; 137228f1b45aSHong Zhang /* L part */ 1373b89f182dSHong Zhang pc1 = b->a + bi[i]; 137428f1b45aSHong Zhang pj = b->j + bi[i]; 137528f1b45aSHong Zhang nz = bi[i+1] - bi[i]; 137628f1b45aSHong Zhang for (j=0; j<nz; j++) { 137768785679SHong Zhang col = pj[j]; 13780e7a5c2bSHong Zhang pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 137928f1b45aSHong Zhang } 138028f1b45aSHong Zhang /* U part */ 1381b89f182dSHong Zhang pc1 = b->a + bdiag[i+1]+1; 138228f1b45aSHong Zhang pj = b->j + bdiag[i+1]+1; 13830e7a5c2bSHong Zhang nz = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */ 138428f1b45aSHong Zhang for (j=0; j<nz; j++) { 138568785679SHong Zhang col = pj[j]; 13860e7a5c2bSHong Zhang pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 138728f1b45aSHong Zhang } 138828f1b45aSHong Zhang 138928f1b45aSHong Zhang sctx.rs = rs; 1390b89f182dSHong Zhang sctx.pv = rtmp1[i]; 13914cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr); 139207b50cabSHong Zhang if (sctx.newshift) break; 1393b89f182dSHong Zhang pc1 = b->a + bdiag[i]; /* Mark diagonal */ 1394b89f182dSHong Zhang *pc1 = 1.0/sctx.pv; 1395b89f182dSHong Zhang 1396b89f182dSHong Zhang /* Now take care of diagonal 2x2 block. */ 1397b89f182dSHong Zhang pc2 = rtmp2 + i; 1398b89f182dSHong Zhang if (*pc2 != 0.0) { 1399b89f182dSHong Zhang mul1 = (*pc2)*(*pc1); /* *pc1=diag[i] is inverted! */ 1400b89f182dSHong Zhang *pc2 = mul1; /* insert L entry */ 1401b89f182dSHong Zhang pj = b->j + bdiag[i+1]+1; /* beginning of U(i,:) */ 1402b89f182dSHong Zhang nz = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */ 1403b89f182dSHong Zhang for (j=0; j<nz; j++) { 1404b89f182dSHong Zhang col = pj[j]; rtmp2[col] -= mul1 * rtmp1[col]; 140528f1b45aSHong Zhang } 1406ca0c957dSBarry Smith ierr = PetscLogFlops(1+2.0*nz);CHKERRQ(ierr); 1407b89f182dSHong Zhang } 1408b89f182dSHong Zhang 1409b89f182dSHong Zhang /* finished row i+1; check zero pivot, then stick row i+1 into b->a */ 1410b89f182dSHong Zhang rs = 0.0; 1411b89f182dSHong Zhang /* L part */ 1412b89f182dSHong Zhang pc2 = b->a + bi[i+1]; 1413b89f182dSHong Zhang pj = b->j + bi[i+1]; 1414b89f182dSHong Zhang nz = bi[i+2] - bi[i+1]; 1415b89f182dSHong Zhang for (j=0; j<nz; j++) { 1416b89f182dSHong Zhang col = pj[j]; 14170e7a5c2bSHong Zhang pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 1418b89f182dSHong Zhang } 1419b89f182dSHong Zhang /* U part */ 1420b89f182dSHong Zhang pc2 = b->a + bdiag[i+2]+1; 14210e7a5c2bSHong Zhang pj = b->j + bdiag[i+2]+1; 14220e7a5c2bSHong Zhang nz = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */ 1423b89f182dSHong Zhang for (j=0; j<nz; j++) { 1424b89f182dSHong Zhang col = pj[j]; 14250e7a5c2bSHong Zhang pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 1426b89f182dSHong Zhang } 1427b89f182dSHong Zhang 142828f1b45aSHong Zhang sctx.rs = rs; 1429b89f182dSHong Zhang sctx.pv = rtmp2[i+1]; 14304cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,i+1);CHKERRQ(ierr); 143107b50cabSHong Zhang if (sctx.newshift) break; 143228f1b45aSHong Zhang pc2 = b->a + bdiag[i+1]; 1433b89f182dSHong Zhang *pc2 = 1.0/sctx.pv; 143428f1b45aSHong Zhang break; 1435b89f182dSHong Zhang 143668785679SHong Zhang case 3: 143768785679SHong Zhang /*----------*/ 143868785679SHong Zhang /* zero rtmp */ 143968785679SHong Zhang /* L part */ 144068785679SHong Zhang nz = bi[i+1] - bi[i]; 144168785679SHong Zhang bjtmp = bj + bi[i]; 144268785679SHong Zhang for (j=0; j<nz; j++) { 144368785679SHong Zhang col = bjtmp[j]; 1444b89f182dSHong Zhang rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0; 144568785679SHong Zhang } 144668785679SHong Zhang 144768785679SHong Zhang /* U part */ 144868785679SHong Zhang nz = bdiag[i]-bdiag[i+1]; 144968785679SHong Zhang bjtmp = bj + bdiag[i+1]+1; 145068785679SHong Zhang for (j=0; j<nz; j++) { 145168785679SHong Zhang col = bjtmp[j]; 1452b89f182dSHong Zhang rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0; 145368785679SHong Zhang } 145468785679SHong Zhang 145568785679SHong Zhang /* load in initial (unfactored row) */ 145668785679SHong Zhang nz = ai[r[i]+1] - ai[r[i]]; 145768785679SHong Zhang ajtmp = aj + ai[r[i]]; 145868785679SHong Zhang v1 = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; v3 = aa + ai[r[i]+2]; 145968785679SHong Zhang for (j=0; j<nz; j++) { 146068785679SHong Zhang col = ics[ajtmp[j]]; 1461b89f182dSHong Zhang rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; rtmp3[col] = v3[j]; 146268785679SHong Zhang } 146368785679SHong Zhang /* ZeropivotApply(): shift the diagonal of the matrix */ 1464b89f182dSHong Zhang rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; rtmp3[i+2] += sctx.shift_amount; 146568785679SHong Zhang 146668785679SHong Zhang /* elimination */ 146768785679SHong Zhang bjtmp = bj + bi[i]; 146868785679SHong Zhang row = *bjtmp++; /* pivot row */ 146968785679SHong Zhang nzL = bi[i+1] - bi[i]; 147068785679SHong Zhang for (k=0; k < nzL; k++) { 1471b89f182dSHong Zhang pc1 = rtmp1 + row; 1472b89f182dSHong Zhang pc2 = rtmp2 + row; 1473b89f182dSHong Zhang pc3 = rtmp3 + row; 147468785679SHong Zhang if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 != 0.0) { 147568785679SHong Zhang pv = b->a + bdiag[row]; 1476b89f182dSHong Zhang mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); mul3 = *pc3*(*pv); 1477b89f182dSHong Zhang *pc1 = mul1; *pc2 = mul2; *pc3 = mul3; 147868785679SHong Zhang 147968785679SHong Zhang pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 148068785679SHong Zhang pv = b->a + bdiag[row+1]+1; 148168785679SHong Zhang nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 148268785679SHong Zhang for (j=0; j<nz; j++) { 148368785679SHong Zhang col = pj[j]; 1484b89f182dSHong Zhang rtmp1[col] -= mul1 * pv[j]; 1485b89f182dSHong Zhang rtmp2[col] -= mul2 * pv[j]; 1486b89f182dSHong Zhang rtmp3[col] -= mul3 * pv[j]; 148768785679SHong Zhang } 1488ca0c957dSBarry Smith ierr = PetscLogFlops(3+6.0*nz);CHKERRQ(ierr); 148968785679SHong Zhang } 149068785679SHong Zhang row = *bjtmp++; 149168785679SHong Zhang } 149268785679SHong Zhang 1493b89f182dSHong Zhang /* finished row i; check zero pivot, then stick row i into b->a */ 1494b89f182dSHong Zhang rs = 0.0; 1495b89f182dSHong Zhang /* L part */ 1496b89f182dSHong Zhang pc1 = b->a + bi[i]; 1497b89f182dSHong Zhang pj = b->j + bi[i]; 1498b89f182dSHong Zhang nz = bi[i+1] - bi[i]; 1499b89f182dSHong Zhang for (j=0; j<nz; j++) { 1500b89f182dSHong Zhang col = pj[j]; 15010e7a5c2bSHong Zhang pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 1502b89f182dSHong Zhang } 1503b89f182dSHong Zhang /* U part */ 1504b89f182dSHong Zhang pc1 = b->a + bdiag[i+1]+1; 1505b89f182dSHong Zhang pj = b->j + bdiag[i+1]+1; 15060e7a5c2bSHong Zhang nz = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */ 1507b89f182dSHong Zhang for (j=0; j<nz; j++) { 1508b89f182dSHong Zhang col = pj[j]; 15090e7a5c2bSHong Zhang pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 1510b89f182dSHong Zhang } 151168785679SHong Zhang 1512b89f182dSHong Zhang sctx.rs = rs; 1513b89f182dSHong Zhang sctx.pv = rtmp1[i]; 15144cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr); 151507b50cabSHong Zhang if (sctx.newshift) break; 1516b89f182dSHong Zhang pc1 = b->a + bdiag[i]; /* Mark diag[i] */ 1517b89f182dSHong Zhang *pc1 = 1.0/sctx.pv; 1518b89f182dSHong Zhang 1519b89f182dSHong Zhang /* Now take care of 1st column of diagonal 3x3 block. */ 1520b89f182dSHong Zhang pc2 = rtmp2 + i; 1521b89f182dSHong Zhang pc3 = rtmp3 + i; 1522b89f182dSHong Zhang if (*pc2 != 0.0 || *pc3 != 0.0) { 1523b89f182dSHong Zhang mul2 = (*pc2)*(*pc1); *pc2 = mul2; 1524b89f182dSHong Zhang mul3 = (*pc3)*(*pc1); *pc3 = mul3; 152568785679SHong Zhang pj = b->j + bdiag[i+1]+1; /* beginning of U(i,:) */ 152668785679SHong Zhang nz = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */ 152768785679SHong Zhang for (j=0; j<nz; j++) { 152868785679SHong Zhang col = pj[j]; 1529b89f182dSHong Zhang rtmp2[col] -= mul2 * rtmp1[col]; 1530b89f182dSHong Zhang rtmp3[col] -= mul3 * rtmp1[col]; 153168785679SHong Zhang } 1532ca0c957dSBarry Smith ierr = PetscLogFlops(2+4.0*nz);CHKERRQ(ierr); 153368785679SHong Zhang } 153468785679SHong Zhang 1535b89f182dSHong Zhang /* finished row i+1; check zero pivot, then stick row i+1 into b->a */ 1536b89f182dSHong Zhang rs = 0.0; 1537b89f182dSHong Zhang /* L part */ 1538b89f182dSHong Zhang pc2 = b->a + bi[i+1]; 1539b89f182dSHong Zhang pj = b->j + bi[i+1]; 1540b89f182dSHong Zhang nz = bi[i+2] - bi[i+1]; 1541b89f182dSHong Zhang for (j=0; j<nz; j++) { 1542b89f182dSHong Zhang col = pj[j]; 15430e7a5c2bSHong Zhang pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 1544b89f182dSHong Zhang } 1545b89f182dSHong Zhang /* U part */ 1546b89f182dSHong Zhang pc2 = b->a + bdiag[i+2]+1; 15470e7a5c2bSHong Zhang pj = b->j + bdiag[i+2]+1; 15480e7a5c2bSHong Zhang nz = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */ 1549b89f182dSHong Zhang for (j=0; j<nz; j++) { 1550b89f182dSHong Zhang col = pj[j]; 15510e7a5c2bSHong Zhang pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 1552b89f182dSHong Zhang } 1553b89f182dSHong Zhang 1554b89f182dSHong Zhang sctx.rs = rs; 1555b89f182dSHong Zhang sctx.pv = rtmp2[i+1]; 15564cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,i+1);CHKERRQ(ierr); 155707b50cabSHong Zhang if (sctx.newshift) break; 1558b89f182dSHong Zhang pc2 = b->a + bdiag[i+1]; 1559b89f182dSHong Zhang *pc2 = 1.0/sctx.pv; /* Mark diag[i+1] */ 1560b89f182dSHong Zhang 1561b89f182dSHong Zhang /* Now take care of 2nd column of diagonal 3x3 block. */ 1562b89f182dSHong Zhang pc3 = rtmp3 + i+1; 156368785679SHong Zhang if (*pc3 != 0.0) { 1564b89f182dSHong Zhang mul3 = (*pc3)*(*pc2); *pc3 = mul3; 156568785679SHong Zhang pj = b->j + bdiag[i+2]+1; /* beginning of U(i+1,:) */ 156668785679SHong Zhang nz = bdiag[i+1]-bdiag[i+2]-1; /* num of entries in U(i+1,:) excluding diag */ 156768785679SHong Zhang for (j=0; j<nz; j++) { 156868785679SHong Zhang col = pj[j]; 1569b89f182dSHong Zhang rtmp3[col] -= mul3 * rtmp2[col]; 157068785679SHong Zhang } 1571ca0c957dSBarry Smith ierr = PetscLogFlops(1+2.0*nz);CHKERRQ(ierr); 157268785679SHong Zhang } 157368785679SHong Zhang 1574b89f182dSHong Zhang /* finished i+2; check zero pivot, then stick row i+2 into b->a */ 157568785679SHong Zhang rs = 0.0; 157668785679SHong Zhang /* L part */ 1577b89f182dSHong Zhang pc3 = b->a + bi[i+2]; 1578b89f182dSHong Zhang pj = b->j + bi[i+2]; 1579b89f182dSHong Zhang nz = bi[i+3] - bi[i+2]; 158068785679SHong Zhang for (j=0; j<nz; j++) { 158168785679SHong Zhang col = pj[j]; 15820e7a5c2bSHong Zhang pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]); 158368785679SHong Zhang } 158468785679SHong Zhang /* U part */ 1585b89f182dSHong Zhang pc3 = b->a + bdiag[i+3]+1; 15860e7a5c2bSHong Zhang pj = b->j + bdiag[i+3]+1; 15870e7a5c2bSHong Zhang nz = bdiag[i+2] - bdiag[i+3] - 1; /* exclude diagonal */ 158868785679SHong Zhang for (j=0; j<nz; j++) { 158968785679SHong Zhang col = pj[j]; 15900e7a5c2bSHong Zhang pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]); 159168785679SHong Zhang } 159268785679SHong Zhang 159368785679SHong Zhang sctx.rs = rs; 1594b89f182dSHong Zhang sctx.pv = rtmp3[i+2]; 15954cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,i+2);CHKERRQ(ierr); 159607b50cabSHong Zhang if (sctx.newshift) break; 159768785679SHong Zhang pc3 = b->a + bdiag[i+2]; 1598b89f182dSHong Zhang *pc3 = 1.0/sctx.pv; /* Mark diag[i+2] */ 159968785679SHong Zhang break; 16009877982aSShri Abhyankar case 4: 16019877982aSShri Abhyankar /*----------*/ 16029877982aSShri Abhyankar /* zero rtmp */ 16039877982aSShri Abhyankar /* L part */ 16049877982aSShri Abhyankar nz = bi[i+1] - bi[i]; 16059877982aSShri Abhyankar bjtmp = bj + bi[i]; 16069877982aSShri Abhyankar for (j=0; j<nz; j++) { 16079877982aSShri Abhyankar col = bjtmp[j]; 16089877982aSShri Abhyankar rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0;rtmp4[col] = 0.0; 16099877982aSShri Abhyankar } 16109877982aSShri Abhyankar 16119877982aSShri Abhyankar /* U part */ 16129877982aSShri Abhyankar nz = bdiag[i]-bdiag[i+1]; 16139877982aSShri Abhyankar bjtmp = bj + bdiag[i+1]+1; 16149877982aSShri Abhyankar for (j=0; j<nz; j++) { 16159877982aSShri Abhyankar col = bjtmp[j]; 16169877982aSShri Abhyankar rtmp1[col] = 0.0; rtmp2[col] = 0.0; rtmp3[col] = 0.0; rtmp4[col] = 0.0; 16179877982aSShri Abhyankar } 16189877982aSShri Abhyankar 16199877982aSShri Abhyankar /* load in initial (unfactored row) */ 16209877982aSShri Abhyankar nz = ai[r[i]+1] - ai[r[i]]; 16219877982aSShri Abhyankar ajtmp = aj + ai[r[i]]; 16229877982aSShri Abhyankar v1 = aa + ai[r[i]]; v2 = aa + ai[r[i]+1]; v3 = aa + ai[r[i]+2]; v4 = aa + ai[r[i]+3]; 16239877982aSShri Abhyankar for (j=0; j<nz; j++) { 16249877982aSShri Abhyankar col = ics[ajtmp[j]]; 16259877982aSShri Abhyankar rtmp1[col] = v1[j]; rtmp2[col] = v2[j]; rtmp3[col] = v3[j]; rtmp4[col] = v4[j]; 16269877982aSShri Abhyankar } 16279877982aSShri Abhyankar /* ZeropivotApply(): shift the diagonal of the matrix */ 16289877982aSShri Abhyankar rtmp1[i] += sctx.shift_amount; rtmp2[i+1] += sctx.shift_amount; rtmp3[i+2] += sctx.shift_amount; rtmp4[i+3] += sctx.shift_amount; 16299877982aSShri Abhyankar 16309877982aSShri Abhyankar /* elimination */ 16319877982aSShri Abhyankar bjtmp = bj + bi[i]; 16329877982aSShri Abhyankar row = *bjtmp++; /* pivot row */ 16339877982aSShri Abhyankar nzL = bi[i+1] - bi[i]; 16349877982aSShri Abhyankar for (k=0; k < nzL; k++) { 16359877982aSShri Abhyankar pc1 = rtmp1 + row; 16369877982aSShri Abhyankar pc2 = rtmp2 + row; 16379877982aSShri Abhyankar pc3 = rtmp3 + row; 16389877982aSShri Abhyankar pc4 = rtmp4 + row; 16399877982aSShri Abhyankar if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 != 0.0 || *pc4 != 0.0) { 16409877982aSShri Abhyankar pv = b->a + bdiag[row]; 16419877982aSShri Abhyankar mul1 = *pc1*(*pv); mul2 = *pc2*(*pv); mul3 = *pc3*(*pv); mul4 = *pc4*(*pv); 16429877982aSShri Abhyankar *pc1 = mul1; *pc2 = mul2; *pc3 = mul3; *pc4 = mul4; 16439877982aSShri Abhyankar 16449877982aSShri Abhyankar pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */ 16459877982aSShri Abhyankar pv = b->a + bdiag[row+1]+1; 16469877982aSShri Abhyankar nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */ 16479877982aSShri Abhyankar for (j=0; j<nz; j++) { 16489877982aSShri Abhyankar col = pj[j]; 16499877982aSShri Abhyankar rtmp1[col] -= mul1 * pv[j]; 16509877982aSShri Abhyankar rtmp2[col] -= mul2 * pv[j]; 16519877982aSShri Abhyankar rtmp3[col] -= mul3 * pv[j]; 16529877982aSShri Abhyankar rtmp4[col] -= mul4 * pv[j]; 16539877982aSShri Abhyankar } 1654ca0c957dSBarry Smith ierr = PetscLogFlops(4+8.0*nz);CHKERRQ(ierr); 16559877982aSShri Abhyankar } 16569877982aSShri Abhyankar row = *bjtmp++; 16579877982aSShri Abhyankar } 16589877982aSShri Abhyankar 16599877982aSShri Abhyankar /* finished row i; check zero pivot, then stick row i into b->a */ 16609877982aSShri Abhyankar rs = 0.0; 16619877982aSShri Abhyankar /* L part */ 16629877982aSShri Abhyankar pc1 = b->a + bi[i]; 16639877982aSShri Abhyankar pj = b->j + bi[i]; 16649877982aSShri Abhyankar nz = bi[i+1] - bi[i]; 16659877982aSShri Abhyankar for (j=0; j<nz; j++) { 16669877982aSShri Abhyankar col = pj[j]; 16679877982aSShri Abhyankar pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 16689877982aSShri Abhyankar } 16699877982aSShri Abhyankar /* U part */ 16709877982aSShri Abhyankar pc1 = b->a + bdiag[i+1]+1; 16719877982aSShri Abhyankar pj = b->j + bdiag[i+1]+1; 16729877982aSShri Abhyankar nz = bdiag[i] - bdiag[i+1] - 1; /* exclude diagonal */ 16739877982aSShri Abhyankar for (j=0; j<nz; j++) { 16749877982aSShri Abhyankar col = pj[j]; 16759877982aSShri Abhyankar pc1[j] = rtmp1[col]; rs += PetscAbsScalar(pc1[j]); 16769877982aSShri Abhyankar } 16779877982aSShri Abhyankar 16789877982aSShri Abhyankar sctx.rs = rs; 16799877982aSShri Abhyankar sctx.pv = rtmp1[i]; 16804cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,i);CHKERRQ(ierr); 168107b50cabSHong Zhang if (sctx.newshift) break; 16829877982aSShri Abhyankar pc1 = b->a + bdiag[i]; /* Mark diag[i] */ 16839877982aSShri Abhyankar *pc1 = 1.0/sctx.pv; 16849877982aSShri Abhyankar 16859877982aSShri Abhyankar /* Now take care of 1st column of diagonal 4x4 block. */ 16869877982aSShri Abhyankar pc2 = rtmp2 + i; 16879877982aSShri Abhyankar pc3 = rtmp3 + i; 16889877982aSShri Abhyankar pc4 = rtmp4 + i; 16899877982aSShri Abhyankar if (*pc2 != 0.0 || *pc3 != 0.0 || *pc4 != 0.0) { 16909877982aSShri Abhyankar mul2 = (*pc2)*(*pc1); *pc2 = mul2; 16919877982aSShri Abhyankar mul3 = (*pc3)*(*pc1); *pc3 = mul3; 16929877982aSShri Abhyankar mul4 = (*pc4)*(*pc1); *pc4 = mul4; 16939877982aSShri Abhyankar pj = b->j + bdiag[i+1]+1; /* beginning of U(i,:) */ 16949877982aSShri Abhyankar nz = bdiag[i]-bdiag[i+1]-1; /* num of entries in U(i,:) excluding diag */ 16959877982aSShri Abhyankar for (j=0; j<nz; j++) { 16969877982aSShri Abhyankar col = pj[j]; 16979877982aSShri Abhyankar rtmp2[col] -= mul2 * rtmp1[col]; 16989877982aSShri Abhyankar rtmp3[col] -= mul3 * rtmp1[col]; 16999877982aSShri Abhyankar rtmp4[col] -= mul4 * rtmp1[col]; 17009877982aSShri Abhyankar } 1701ca0c957dSBarry Smith ierr = PetscLogFlops(3+6.0*nz);CHKERRQ(ierr); 17029877982aSShri Abhyankar } 17039877982aSShri Abhyankar 17049877982aSShri Abhyankar /* finished row i+1; check zero pivot, then stick row i+1 into b->a */ 17059877982aSShri Abhyankar rs = 0.0; 17069877982aSShri Abhyankar /* L part */ 17079877982aSShri Abhyankar pc2 = b->a + bi[i+1]; 17089877982aSShri Abhyankar pj = b->j + bi[i+1]; 17099877982aSShri Abhyankar nz = bi[i+2] - bi[i+1]; 17109877982aSShri Abhyankar for (j=0; j<nz; j++) { 17119877982aSShri Abhyankar col = pj[j]; 17129877982aSShri Abhyankar pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 17139877982aSShri Abhyankar } 17149877982aSShri Abhyankar /* U part */ 17159877982aSShri Abhyankar pc2 = b->a + bdiag[i+2]+1; 17169877982aSShri Abhyankar pj = b->j + bdiag[i+2]+1; 17179877982aSShri Abhyankar nz = bdiag[i+1] - bdiag[i+2] - 1; /* exclude diagonal */ 17189877982aSShri Abhyankar for (j=0; j<nz; j++) { 17199877982aSShri Abhyankar col = pj[j]; 17209877982aSShri Abhyankar pc2[j] = rtmp2[col]; rs += PetscAbsScalar(pc2[j]); 17219877982aSShri Abhyankar } 17229877982aSShri Abhyankar 17239877982aSShri Abhyankar sctx.rs = rs; 17249877982aSShri Abhyankar sctx.pv = rtmp2[i+1]; 17254cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,i+1);CHKERRQ(ierr); 172607b50cabSHong Zhang if (sctx.newshift) break; 17279877982aSShri Abhyankar pc2 = b->a + bdiag[i+1]; 17289877982aSShri Abhyankar *pc2 = 1.0/sctx.pv; /* Mark diag[i+1] */ 17299877982aSShri Abhyankar 17309877982aSShri Abhyankar /* Now take care of 2nd column of diagonal 4x4 block. */ 17319877982aSShri Abhyankar pc3 = rtmp3 + i+1; 17329877982aSShri Abhyankar pc4 = rtmp4 + i+1; 17339877982aSShri Abhyankar if (*pc3 != 0.0 || *pc4 != 0.0) { 17349877982aSShri Abhyankar mul3 = (*pc3)*(*pc2); *pc3 = mul3; 17359877982aSShri Abhyankar mul4 = (*pc4)*(*pc2); *pc4 = mul4; 17369877982aSShri Abhyankar pj = b->j + bdiag[i+2]+1; /* beginning of U(i+1,:) */ 17379877982aSShri Abhyankar nz = bdiag[i+1]-bdiag[i+2]-1; /* num of entries in U(i+1,:) excluding diag */ 17389877982aSShri Abhyankar for (j=0; j<nz; j++) { 17399877982aSShri Abhyankar col = pj[j]; 17409877982aSShri Abhyankar rtmp3[col] -= mul3 * rtmp2[col]; 17419877982aSShri Abhyankar rtmp4[col] -= mul4 * rtmp2[col]; 17429877982aSShri Abhyankar } 1743ca0c957dSBarry Smith ierr = PetscLogFlops(4.0*nz);CHKERRQ(ierr); 17449877982aSShri Abhyankar } 17459877982aSShri Abhyankar 17469877982aSShri Abhyankar /* finished i+2; check zero pivot, then stick row i+2 into b->a */ 17479877982aSShri Abhyankar rs = 0.0; 17489877982aSShri Abhyankar /* L part */ 17499877982aSShri Abhyankar pc3 = b->a + bi[i+2]; 17509877982aSShri Abhyankar pj = b->j + bi[i+2]; 17519877982aSShri Abhyankar nz = bi[i+3] - bi[i+2]; 17529877982aSShri Abhyankar for (j=0; j<nz; j++) { 17539877982aSShri Abhyankar col = pj[j]; 17549877982aSShri Abhyankar pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]); 17559877982aSShri Abhyankar } 17569877982aSShri Abhyankar /* U part */ 17579877982aSShri Abhyankar pc3 = b->a + bdiag[i+3]+1; 17589877982aSShri Abhyankar pj = b->j + bdiag[i+3]+1; 17599877982aSShri Abhyankar nz = bdiag[i+2] - bdiag[i+3] - 1; /* exclude diagonal */ 17609877982aSShri Abhyankar for (j=0; j<nz; j++) { 17619877982aSShri Abhyankar col = pj[j]; 17629877982aSShri Abhyankar pc3[j] = rtmp3[col]; rs += PetscAbsScalar(pc3[j]); 17639877982aSShri Abhyankar } 17649877982aSShri Abhyankar 17659877982aSShri Abhyankar sctx.rs = rs; 17669877982aSShri Abhyankar sctx.pv = rtmp3[i+2]; 17674cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,i+2);CHKERRQ(ierr); 176807b50cabSHong Zhang if (sctx.newshift) break; 17699877982aSShri Abhyankar pc3 = b->a + bdiag[i+2]; 17709877982aSShri Abhyankar *pc3 = 1.0/sctx.pv; /* Mark diag[i+2] */ 17719877982aSShri Abhyankar 17729877982aSShri Abhyankar /* Now take care of 3rd column of diagonal 4x4 block. */ 17739877982aSShri Abhyankar pc4 = rtmp4 + i+2; 17749877982aSShri Abhyankar if (*pc4 != 0.0) { 17759877982aSShri Abhyankar mul4 = (*pc4)*(*pc3); *pc4 = mul4; 17769877982aSShri Abhyankar pj = b->j + bdiag[i+3]+1; /* beginning of U(i+2,:) */ 17779877982aSShri Abhyankar nz = bdiag[i+2]-bdiag[i+3]-1; /* num of entries in U(i+2,:) excluding diag */ 17789877982aSShri Abhyankar for (j=0; j<nz; j++) { 17799877982aSShri Abhyankar col = pj[j]; 17809877982aSShri Abhyankar rtmp4[col] -= mul4 * rtmp3[col]; 17819877982aSShri Abhyankar } 1782ca0c957dSBarry Smith ierr = PetscLogFlops(1+2.0*nz);CHKERRQ(ierr); 17839877982aSShri Abhyankar } 17849877982aSShri Abhyankar 17859877982aSShri Abhyankar /* finished i+3; check zero pivot, then stick row i+3 into b->a */ 17869877982aSShri Abhyankar rs = 0.0; 17879877982aSShri Abhyankar /* L part */ 17889877982aSShri Abhyankar pc4 = b->a + bi[i+3]; 17899877982aSShri Abhyankar pj = b->j + bi[i+3]; 17909877982aSShri Abhyankar nz = bi[i+4] - bi[i+3]; 17919877982aSShri Abhyankar for (j=0; j<nz; j++) { 17929877982aSShri Abhyankar col = pj[j]; 17939877982aSShri Abhyankar pc4[j] = rtmp4[col]; rs += PetscAbsScalar(pc4[j]); 17949877982aSShri Abhyankar } 17959877982aSShri Abhyankar /* U part */ 17969877982aSShri Abhyankar pc4 = b->a + bdiag[i+4]+1; 17979877982aSShri Abhyankar pj = b->j + bdiag[i+4]+1; 17989877982aSShri Abhyankar nz = bdiag[i+3] - bdiag[i+4] - 1; /* exclude diagonal */ 17999877982aSShri Abhyankar for (j=0; j<nz; j++) { 18009877982aSShri Abhyankar col = pj[j]; 18019877982aSShri Abhyankar pc4[j] = rtmp4[col]; rs += PetscAbsScalar(pc4[j]); 18029877982aSShri Abhyankar } 18039877982aSShri Abhyankar 18049877982aSShri Abhyankar sctx.rs = rs; 18059877982aSShri Abhyankar sctx.pv = rtmp4[i+3]; 18064cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,i+3);CHKERRQ(ierr); 180707b50cabSHong Zhang if (sctx.newshift) break; 18089877982aSShri Abhyankar pc4 = b->a + bdiag[i+3]; 18099877982aSShri Abhyankar *pc4 = 1.0/sctx.pv; /* Mark diag[i+3] */ 18109877982aSShri Abhyankar break; 181168785679SHong Zhang 181228f1b45aSHong Zhang default: 1813e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Node size not yet supported \n"); 181428f1b45aSHong Zhang } 1815c2b86aeeSHong Zhang if (sctx.newshift) break; /* break for (inod=0,i=0; inod<node_max; inod++) */ 181628f1b45aSHong Zhang i += nodesz; /* Update the row */ 181768785679SHong Zhang } 181828f1b45aSHong Zhang 181928f1b45aSHong Zhang /* MatPivotRefine() */ 182007b50cabSHong Zhang if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) { 182128f1b45aSHong Zhang /* 182228f1b45aSHong Zhang * if no shift in this attempt & shifting & started shifting & can refine, 182328f1b45aSHong Zhang * then try lower shift 182428f1b45aSHong Zhang */ 182528f1b45aSHong Zhang sctx.shift_hi = sctx.shift_fraction; 182628f1b45aSHong Zhang sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.; 182728f1b45aSHong Zhang sctx.shift_amount = sctx.shift_fraction * sctx.shift_top; 182807b50cabSHong Zhang sctx.newshift = PETSC_TRUE; 182928f1b45aSHong Zhang sctx.nshift++; 183028f1b45aSHong Zhang } 183107b50cabSHong Zhang } while (sctx.newshift); 183228f1b45aSHong Zhang 1833eb9baa12SBarry Smith ierr = PetscFree4(rtmp1,rtmp2,rtmp3,rtmp4);CHKERRQ(ierr); 183468785679SHong Zhang ierr = PetscFree(tmp_vec2);CHKERRQ(ierr); 183528f1b45aSHong Zhang ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 183628f1b45aSHong Zhang ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 183728f1b45aSHong Zhang 1838abb87a52SBarry Smith if (b->inode.size) { 1839abb87a52SBarry Smith C->ops->solve = MatSolve_SeqAIJ_Inode; 1840abb87a52SBarry Smith } else { 1841d3ac4fa3SBarry Smith C->ops->solve = MatSolve_SeqAIJ; 1842abb87a52SBarry Smith } 184328f1b45aSHong Zhang C->ops->solveadd = MatSolveAdd_SeqAIJ; 184428f1b45aSHong Zhang C->ops->solvetranspose = MatSolveTranspose_SeqAIJ; 184528f1b45aSHong Zhang C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ; 184628f1b45aSHong Zhang C->ops->matsolve = MatMatSolve_SeqAIJ; 184728f1b45aSHong Zhang C->assembled = PETSC_TRUE; 184828f1b45aSHong Zhang C->preallocated = PETSC_TRUE; 18492205254eSKarl Rupp 185028f1b45aSHong Zhang ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 185128f1b45aSHong Zhang 185228f1b45aSHong Zhang /* MatShiftView(A,info,&sctx) */ 185328f1b45aSHong Zhang if (sctx.nshift) { 1854f4db908eSBarry Smith if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { 185557622a8eSBarry Smith ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %g, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,(double)sctx.shift_amount,(double)sctx.shift_fraction,(double)sctx.shift_top);CHKERRQ(ierr); 1856f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 185757622a8eSBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr); 1858f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS) { 185957622a8eSBarry Smith ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %g\n",sctx.nshift,(double)info->shiftamount);CHKERRQ(ierr); 186028f1b45aSHong Zhang } 186128f1b45aSHong Zhang } 186228f1b45aSHong Zhang PetscFunctionReturn(0); 186328f1b45aSHong Zhang } 1864628f99d7SShri Abhyankar 18659a625307SHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_Inode_inplace(Mat B,Mat A,const MatFactorInfo *info) 1866628f99d7SShri Abhyankar { 1867628f99d7SShri Abhyankar Mat C = B; 1868628f99d7SShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b = (Mat_SeqAIJ*)C->data; 1869628f99d7SShri Abhyankar IS iscol = b->col,isrow = b->row,isicol = b->icol; 1870628f99d7SShri Abhyankar PetscErrorCode ierr; 1871628f99d7SShri Abhyankar const PetscInt *r,*ic,*c,*ics; 1872628f99d7SShri Abhyankar PetscInt n = A->rmap->n,*bi = b->i; 1873628f99d7SShri Abhyankar PetscInt *bj = b->j,*nbj=b->j +1,*ajtmp,*bjtmp,nz,nz_tmp,row,prow; 18748758e1faSBarry Smith PetscInt i,j,idx,*bd = b->diag,node_max,nodesz; 18758758e1faSBarry Smith PetscInt *ai = a->i,*aj = a->j; 1876628f99d7SShri Abhyankar PetscInt *ns,*tmp_vec1,*tmp_vec2,*nsmap,*pj; 1877628f99d7SShri Abhyankar PetscScalar mul1,mul2,mul3,tmp; 1878628f99d7SShri Abhyankar MatScalar *pc1,*pc2,*pc3,*ba = b->a,*pv,*rtmp11,*rtmp22,*rtmp33; 1879628f99d7SShri Abhyankar const MatScalar *v1,*v2,*v3,*aa = a->a,*rtmp1; 1880628f99d7SShri Abhyankar PetscReal rs=0.0; 1881628f99d7SShri Abhyankar FactorShiftCtx sctx; 1882628f99d7SShri Abhyankar 1883628f99d7SShri Abhyankar PetscFunctionBegin; 1884628f99d7SShri Abhyankar sctx.shift_top = 0; 1885628f99d7SShri Abhyankar sctx.nshift_max = 0; 1886628f99d7SShri Abhyankar sctx.shift_lo = 0; 1887628f99d7SShri Abhyankar sctx.shift_hi = 0; 1888628f99d7SShri Abhyankar sctx.shift_fraction = 0; 1889628f99d7SShri Abhyankar 1890628f99d7SShri Abhyankar /* if both shift schemes are chosen by user, only use info->shiftpd */ 1891f4db908eSBarry Smith if (info->shifttype==(PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */ 1892628f99d7SShri Abhyankar sctx.shift_top = 0; 1893628f99d7SShri Abhyankar for (i=0; i<n; i++) { 1894628f99d7SShri Abhyankar /* calculate rs = sum(|aij|)-RealPart(aii), amt of shift needed for this row */ 1895628f99d7SShri Abhyankar rs = 0.0; 1896628f99d7SShri Abhyankar ajtmp = aj + ai[i]; 1897628f99d7SShri Abhyankar rtmp1 = aa + ai[i]; 1898628f99d7SShri Abhyankar nz = ai[i+1] - ai[i]; 1899628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 1900628f99d7SShri Abhyankar if (*ajtmp != i) { 1901628f99d7SShri Abhyankar rs += PetscAbsScalar(*rtmp1++); 1902628f99d7SShri Abhyankar } else { 1903628f99d7SShri Abhyankar rs -= PetscRealPart(*rtmp1++); 1904628f99d7SShri Abhyankar } 1905628f99d7SShri Abhyankar ajtmp++; 1906628f99d7SShri Abhyankar } 1907628f99d7SShri Abhyankar if (rs>sctx.shift_top) sctx.shift_top = rs; 1908628f99d7SShri Abhyankar } 1909628f99d7SShri Abhyankar if (sctx.shift_top == 0.0) sctx.shift_top += 1.e-12; 1910628f99d7SShri Abhyankar sctx.shift_top *= 1.1; 1911628f99d7SShri Abhyankar sctx.nshift_max = 5; 1912628f99d7SShri Abhyankar sctx.shift_lo = 0.; 1913628f99d7SShri Abhyankar sctx.shift_hi = 1.; 1914628f99d7SShri Abhyankar } 1915628f99d7SShri Abhyankar sctx.shift_amount = 0; 1916628f99d7SShri Abhyankar sctx.nshift = 0; 1917628f99d7SShri Abhyankar 1918628f99d7SShri Abhyankar ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr); 1919628f99d7SShri Abhyankar ierr = ISGetIndices(iscol,&c);CHKERRQ(ierr); 1920628f99d7SShri Abhyankar ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr); 19211795a4d1SJed Brown ierr = PetscCalloc3(n,&rtmp11,n,&rtmp22,n,&rtmp33);CHKERRQ(ierr); 1922628f99d7SShri Abhyankar ics = ic; 1923628f99d7SShri Abhyankar 1924628f99d7SShri Abhyankar node_max = a->inode.node_count; 1925628f99d7SShri Abhyankar ns = a->inode.size; 1926e7e72b3dSBarry Smith if (!ns) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Matrix without inode information"); 1927628f99d7SShri Abhyankar 1928628f99d7SShri Abhyankar /* If max inode size > 3, split it into two inodes.*/ 1929628f99d7SShri Abhyankar /* also map the inode sizes according to the ordering */ 1930854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&tmp_vec1);CHKERRQ(ierr); 1931628f99d7SShri Abhyankar for (i=0,j=0; i<node_max; ++i,++j) { 1932628f99d7SShri Abhyankar if (ns[i]>3) { 1933628f99d7SShri Abhyankar tmp_vec1[j] = ns[i]/2; /* Assuming ns[i] < =5 */ 1934628f99d7SShri Abhyankar ++j; 1935628f99d7SShri Abhyankar tmp_vec1[j] = ns[i] - tmp_vec1[j-1]; 1936628f99d7SShri Abhyankar } else { 1937628f99d7SShri Abhyankar tmp_vec1[j] = ns[i]; 1938628f99d7SShri Abhyankar } 1939628f99d7SShri Abhyankar } 1940628f99d7SShri Abhyankar /* Use the correct node_max */ 1941628f99d7SShri Abhyankar node_max = j; 1942628f99d7SShri Abhyankar 1943628f99d7SShri Abhyankar /* Now reorder the inode info based on mat re-ordering info */ 1944628f99d7SShri Abhyankar /* First create a row -> inode_size_array_index map */ 1945071fcb05SBarry Smith ierr = PetscMalloc2(n+1,&nsmap,node_max+1,&tmp_vec2);CHKERRQ(ierr); 1946628f99d7SShri Abhyankar for (i=0,row=0; i<node_max; i++) { 1947628f99d7SShri Abhyankar nodesz = tmp_vec1[i]; 1948628f99d7SShri Abhyankar for (j=0; j<nodesz; j++,row++) { 1949628f99d7SShri Abhyankar nsmap[row] = i; 1950628f99d7SShri Abhyankar } 1951628f99d7SShri Abhyankar } 1952628f99d7SShri Abhyankar /* Using nsmap, create a reordered ns structure */ 1953628f99d7SShri Abhyankar for (i=0,j=0; i< node_max; i++) { 1954628f99d7SShri Abhyankar nodesz = tmp_vec1[nsmap[r[j]]]; /* here the reordered row_no is in r[] */ 1955628f99d7SShri Abhyankar tmp_vec2[i] = nodesz; 1956628f99d7SShri Abhyankar j += nodesz; 1957628f99d7SShri Abhyankar } 1958071fcb05SBarry Smith ierr = PetscFree2(nsmap,tmp_vec1);CHKERRQ(ierr); 1959628f99d7SShri Abhyankar /* Now use the correct ns */ 1960628f99d7SShri Abhyankar ns = tmp_vec2; 1961628f99d7SShri Abhyankar 1962628f99d7SShri Abhyankar do { 196307b50cabSHong Zhang sctx.newshift = PETSC_FALSE; 1964628f99d7SShri Abhyankar /* Now loop over each block-row, and do the factorization */ 1965628f99d7SShri Abhyankar for (i=0,row=0; i<node_max; i++) { 1966628f99d7SShri Abhyankar nodesz = ns[i]; 1967628f99d7SShri Abhyankar nz = bi[row+1] - bi[row]; 1968628f99d7SShri Abhyankar bjtmp = bj + bi[row]; 1969628f99d7SShri Abhyankar 1970628f99d7SShri Abhyankar switch (nodesz) { 1971628f99d7SShri Abhyankar case 1: 1972628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 1973628f99d7SShri Abhyankar idx = bjtmp[j]; 1974628f99d7SShri Abhyankar rtmp11[idx] = 0.0; 1975628f99d7SShri Abhyankar } 1976628f99d7SShri Abhyankar 1977628f99d7SShri Abhyankar /* load in initial (unfactored row) */ 1978628f99d7SShri Abhyankar idx = r[row]; 1979628f99d7SShri Abhyankar nz_tmp = ai[idx+1] - ai[idx]; 1980628f99d7SShri Abhyankar ajtmp = aj + ai[idx]; 1981628f99d7SShri Abhyankar v1 = aa + ai[idx]; 1982628f99d7SShri Abhyankar 1983628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 1984628f99d7SShri Abhyankar idx = ics[ajtmp[j]]; 1985628f99d7SShri Abhyankar rtmp11[idx] = v1[j]; 1986628f99d7SShri Abhyankar } 1987628f99d7SShri Abhyankar rtmp11[ics[r[row]]] += sctx.shift_amount; 1988628f99d7SShri Abhyankar 1989628f99d7SShri Abhyankar prow = *bjtmp++; 1990628f99d7SShri Abhyankar while (prow < row) { 1991628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 1992628f99d7SShri Abhyankar if (*pc1 != 0.0) { 1993628f99d7SShri Abhyankar pv = ba + bd[prow]; 1994628f99d7SShri Abhyankar pj = nbj + bd[prow]; 1995628f99d7SShri Abhyankar mul1 = *pc1 * *pv++; 1996628f99d7SShri Abhyankar *pc1 = mul1; 1997628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 1998ca0c957dSBarry Smith ierr = PetscLogFlops(1+2.0*nz_tmp);CHKERRQ(ierr); 1999628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2000628f99d7SShri Abhyankar tmp = pv[j]; 2001628f99d7SShri Abhyankar idx = pj[j]; 2002628f99d7SShri Abhyankar rtmp11[idx] -= mul1 * tmp; 2003628f99d7SShri Abhyankar } 2004628f99d7SShri Abhyankar } 2005628f99d7SShri Abhyankar prow = *bjtmp++; 2006628f99d7SShri Abhyankar } 2007628f99d7SShri Abhyankar pj = bj + bi[row]; 2008628f99d7SShri Abhyankar pc1 = ba + bi[row]; 2009628f99d7SShri Abhyankar 2010628f99d7SShri Abhyankar sctx.pv = rtmp11[row]; 2011628f99d7SShri Abhyankar rtmp11[row] = 1.0/rtmp11[row]; /* invert diag */ 2012628f99d7SShri Abhyankar rs = 0.0; 2013628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2014628f99d7SShri Abhyankar idx = pj[j]; 2015628f99d7SShri Abhyankar pc1[j] = rtmp11[idx]; /* rtmp11 -> ba */ 2016628f99d7SShri Abhyankar if (idx != row) rs += PetscAbsScalar(pc1[j]); 2017628f99d7SShri Abhyankar } 2018628f99d7SShri Abhyankar sctx.rs = rs; 20194cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,row);CHKERRQ(ierr); 202007b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2021628f99d7SShri Abhyankar break; 2022628f99d7SShri Abhyankar 2023628f99d7SShri Abhyankar case 2: 2024628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2025628f99d7SShri Abhyankar idx = bjtmp[j]; 2026628f99d7SShri Abhyankar rtmp11[idx] = 0.0; 2027628f99d7SShri Abhyankar rtmp22[idx] = 0.0; 2028628f99d7SShri Abhyankar } 2029628f99d7SShri Abhyankar 2030628f99d7SShri Abhyankar /* load in initial (unfactored row) */ 2031628f99d7SShri Abhyankar idx = r[row]; 2032628f99d7SShri Abhyankar nz_tmp = ai[idx+1] - ai[idx]; 2033628f99d7SShri Abhyankar ajtmp = aj + ai[idx]; 2034628f99d7SShri Abhyankar v1 = aa + ai[idx]; 2035628f99d7SShri Abhyankar v2 = aa + ai[idx+1]; 2036628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2037628f99d7SShri Abhyankar idx = ics[ajtmp[j]]; 2038628f99d7SShri Abhyankar rtmp11[idx] = v1[j]; 2039628f99d7SShri Abhyankar rtmp22[idx] = v2[j]; 2040628f99d7SShri Abhyankar } 2041628f99d7SShri Abhyankar rtmp11[ics[r[row]]] += sctx.shift_amount; 2042628f99d7SShri Abhyankar rtmp22[ics[r[row+1]]] += sctx.shift_amount; 2043628f99d7SShri Abhyankar 2044628f99d7SShri Abhyankar prow = *bjtmp++; 2045628f99d7SShri Abhyankar while (prow < row) { 2046628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 2047628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2048628f99d7SShri Abhyankar if (*pc1 != 0.0 || *pc2 != 0.0) { 2049628f99d7SShri Abhyankar pv = ba + bd[prow]; 2050628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2051628f99d7SShri Abhyankar mul1 = *pc1 * *pv; 2052628f99d7SShri Abhyankar mul2 = *pc2 * *pv; 2053628f99d7SShri Abhyankar ++pv; 2054628f99d7SShri Abhyankar *pc1 = mul1; 2055628f99d7SShri Abhyankar *pc2 = mul2; 2056628f99d7SShri Abhyankar 2057628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2058628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2059628f99d7SShri Abhyankar tmp = pv[j]; 2060628f99d7SShri Abhyankar idx = pj[j]; 2061628f99d7SShri Abhyankar rtmp11[idx] -= mul1 * tmp; 2062628f99d7SShri Abhyankar rtmp22[idx] -= mul2 * tmp; 2063628f99d7SShri Abhyankar } 2064ca0c957dSBarry Smith ierr = PetscLogFlops(2+4.0*nz_tmp);CHKERRQ(ierr); 2065628f99d7SShri Abhyankar } 2066628f99d7SShri Abhyankar prow = *bjtmp++; 2067628f99d7SShri Abhyankar } 2068628f99d7SShri Abhyankar 2069628f99d7SShri Abhyankar /* Now take care of diagonal 2x2 block. Note: prow = row here */ 2070628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 2071628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2072628f99d7SShri Abhyankar 2073628f99d7SShri Abhyankar sctx.pv = *pc1; 2074628f99d7SShri Abhyankar pj = bj + bi[prow]; 2075628f99d7SShri Abhyankar rs = 0.0; 2076628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2077628f99d7SShri Abhyankar idx = pj[j]; 2078628f99d7SShri Abhyankar if (idx != prow) rs += PetscAbsScalar(rtmp11[idx]); 2079628f99d7SShri Abhyankar } 2080628f99d7SShri Abhyankar sctx.rs = rs; 20814cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,row);CHKERRQ(ierr); 208207b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2083628f99d7SShri Abhyankar 2084628f99d7SShri Abhyankar if (*pc2 != 0.0) { 2085628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2086628f99d7SShri Abhyankar mul2 = (*pc2)/(*pc1); /* since diag is not yet inverted.*/ 2087628f99d7SShri Abhyankar *pc2 = mul2; 2088628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2089628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2090628f99d7SShri Abhyankar idx = pj[j]; 2091628f99d7SShri Abhyankar tmp = rtmp11[idx]; 2092628f99d7SShri Abhyankar rtmp22[idx] -= mul2 * tmp; 2093628f99d7SShri Abhyankar } 2094ca0c957dSBarry Smith ierr = PetscLogFlops(1+2.0*nz_tmp);CHKERRQ(ierr); 2095628f99d7SShri Abhyankar } 2096628f99d7SShri Abhyankar 2097628f99d7SShri Abhyankar pj = bj + bi[row]; 2098628f99d7SShri Abhyankar pc1 = ba + bi[row]; 2099628f99d7SShri Abhyankar pc2 = ba + bi[row+1]; 2100628f99d7SShri Abhyankar 2101628f99d7SShri Abhyankar sctx.pv = rtmp22[row+1]; 2102628f99d7SShri Abhyankar rs = 0.0; 2103628f99d7SShri Abhyankar rtmp11[row] = 1.0/rtmp11[row]; 2104628f99d7SShri Abhyankar rtmp22[row+1] = 1.0/rtmp22[row+1]; 2105628f99d7SShri Abhyankar /* copy row entries from dense representation to sparse */ 2106628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2107628f99d7SShri Abhyankar idx = pj[j]; 2108628f99d7SShri Abhyankar pc1[j] = rtmp11[idx]; 2109628f99d7SShri Abhyankar pc2[j] = rtmp22[idx]; 2110628f99d7SShri Abhyankar if (idx != row+1) rs += PetscAbsScalar(pc2[j]); 2111628f99d7SShri Abhyankar } 2112628f99d7SShri Abhyankar sctx.rs = rs; 21134cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,row+1);CHKERRQ(ierr); 211407b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2115628f99d7SShri Abhyankar break; 2116628f99d7SShri Abhyankar 2117628f99d7SShri Abhyankar case 3: 2118628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2119628f99d7SShri Abhyankar idx = bjtmp[j]; 2120628f99d7SShri Abhyankar rtmp11[idx] = 0.0; 2121628f99d7SShri Abhyankar rtmp22[idx] = 0.0; 2122628f99d7SShri Abhyankar rtmp33[idx] = 0.0; 2123628f99d7SShri Abhyankar } 2124628f99d7SShri Abhyankar /* copy the nonzeros for the 3 rows from sparse representation to dense in rtmp*[] */ 2125628f99d7SShri Abhyankar idx = r[row]; 2126628f99d7SShri Abhyankar nz_tmp = ai[idx+1] - ai[idx]; 2127628f99d7SShri Abhyankar ajtmp = aj + ai[idx]; 2128628f99d7SShri Abhyankar v1 = aa + ai[idx]; 2129628f99d7SShri Abhyankar v2 = aa + ai[idx+1]; 2130628f99d7SShri Abhyankar v3 = aa + ai[idx+2]; 2131628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2132628f99d7SShri Abhyankar idx = ics[ajtmp[j]]; 2133628f99d7SShri Abhyankar rtmp11[idx] = v1[j]; 2134628f99d7SShri Abhyankar rtmp22[idx] = v2[j]; 2135628f99d7SShri Abhyankar rtmp33[idx] = v3[j]; 2136628f99d7SShri Abhyankar } 2137628f99d7SShri Abhyankar rtmp11[ics[r[row]]] += sctx.shift_amount; 2138628f99d7SShri Abhyankar rtmp22[ics[r[row+1]]] += sctx.shift_amount; 2139628f99d7SShri Abhyankar rtmp33[ics[r[row+2]]] += sctx.shift_amount; 2140628f99d7SShri Abhyankar 2141628f99d7SShri Abhyankar /* loop over all pivot row blocks above this row block */ 2142628f99d7SShri Abhyankar prow = *bjtmp++; 2143628f99d7SShri Abhyankar while (prow < row) { 2144628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 2145628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2146628f99d7SShri Abhyankar pc3 = rtmp33 + prow; 2147628f99d7SShri Abhyankar if (*pc1 != 0.0 || *pc2 != 0.0 || *pc3 !=0.0) { 2148628f99d7SShri Abhyankar pv = ba + bd[prow]; 2149628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2150628f99d7SShri Abhyankar mul1 = *pc1 * *pv; 2151628f99d7SShri Abhyankar mul2 = *pc2 * *pv; 2152628f99d7SShri Abhyankar mul3 = *pc3 * *pv; 2153628f99d7SShri Abhyankar ++pv; 2154628f99d7SShri Abhyankar *pc1 = mul1; 2155628f99d7SShri Abhyankar *pc2 = mul2; 2156628f99d7SShri Abhyankar *pc3 = mul3; 2157628f99d7SShri Abhyankar 2158628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2159628f99d7SShri Abhyankar /* update this row based on pivot row */ 2160628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2161628f99d7SShri Abhyankar tmp = pv[j]; 2162628f99d7SShri Abhyankar idx = pj[j]; 2163628f99d7SShri Abhyankar rtmp11[idx] -= mul1 * tmp; 2164628f99d7SShri Abhyankar rtmp22[idx] -= mul2 * tmp; 2165628f99d7SShri Abhyankar rtmp33[idx] -= mul3 * tmp; 2166628f99d7SShri Abhyankar } 2167ca0c957dSBarry Smith ierr = PetscLogFlops(3+6.0*nz_tmp);CHKERRQ(ierr); 2168628f99d7SShri Abhyankar } 2169628f99d7SShri Abhyankar prow = *bjtmp++; 2170628f99d7SShri Abhyankar } 2171628f99d7SShri Abhyankar 2172628f99d7SShri Abhyankar /* Now take care of diagonal 3x3 block in this set of rows */ 2173628f99d7SShri Abhyankar /* note: prow = row here */ 2174628f99d7SShri Abhyankar pc1 = rtmp11 + prow; 2175628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2176628f99d7SShri Abhyankar pc3 = rtmp33 + prow; 2177628f99d7SShri Abhyankar 2178628f99d7SShri Abhyankar sctx.pv = *pc1; 2179628f99d7SShri Abhyankar pj = bj + bi[prow]; 2180628f99d7SShri Abhyankar rs = 0.0; 2181628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2182628f99d7SShri Abhyankar idx = pj[j]; 2183628f99d7SShri Abhyankar if (idx != row) rs += PetscAbsScalar(rtmp11[idx]); 2184628f99d7SShri Abhyankar } 2185628f99d7SShri Abhyankar sctx.rs = rs; 21864cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,row);CHKERRQ(ierr); 218707b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2188628f99d7SShri Abhyankar 2189628f99d7SShri Abhyankar if (*pc2 != 0.0 || *pc3 != 0.0) { 2190628f99d7SShri Abhyankar mul2 = (*pc2)/(*pc1); 2191628f99d7SShri Abhyankar mul3 = (*pc3)/(*pc1); 2192628f99d7SShri Abhyankar *pc2 = mul2; 2193628f99d7SShri Abhyankar *pc3 = mul3; 2194628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2195628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2196628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2197628f99d7SShri Abhyankar idx = pj[j]; 2198628f99d7SShri Abhyankar tmp = rtmp11[idx]; 2199628f99d7SShri Abhyankar rtmp22[idx] -= mul2 * tmp; 2200628f99d7SShri Abhyankar rtmp33[idx] -= mul3 * tmp; 2201628f99d7SShri Abhyankar } 2202ca0c957dSBarry Smith ierr = PetscLogFlops(2+4.0*nz_tmp);CHKERRQ(ierr); 2203628f99d7SShri Abhyankar } 2204628f99d7SShri Abhyankar ++prow; 2205628f99d7SShri Abhyankar 2206628f99d7SShri Abhyankar pc2 = rtmp22 + prow; 2207628f99d7SShri Abhyankar pc3 = rtmp33 + prow; 2208628f99d7SShri Abhyankar sctx.pv = *pc2; 2209628f99d7SShri Abhyankar pj = bj + bi[prow]; 2210628f99d7SShri Abhyankar rs = 0.0; 2211628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2212628f99d7SShri Abhyankar idx = pj[j]; 2213628f99d7SShri Abhyankar if (idx != prow) rs += PetscAbsScalar(rtmp22[idx]); 2214628f99d7SShri Abhyankar } 2215628f99d7SShri Abhyankar sctx.rs = rs; 22164cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,row+1);CHKERRQ(ierr); 221707b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2218628f99d7SShri Abhyankar 2219628f99d7SShri Abhyankar if (*pc3 != 0.0) { 2220628f99d7SShri Abhyankar mul3 = (*pc3)/(*pc2); 2221628f99d7SShri Abhyankar *pc3 = mul3; 2222628f99d7SShri Abhyankar pj = nbj + bd[prow]; 2223628f99d7SShri Abhyankar nz_tmp = bi[prow+1] - bd[prow] - 1; 2224628f99d7SShri Abhyankar for (j=0; j<nz_tmp; j++) { 2225628f99d7SShri Abhyankar idx = pj[j]; 2226628f99d7SShri Abhyankar tmp = rtmp22[idx]; 2227628f99d7SShri Abhyankar rtmp33[idx] -= mul3 * tmp; 2228628f99d7SShri Abhyankar } 2229ca0c957dSBarry Smith ierr = PetscLogFlops(1+2.0*nz_tmp);CHKERRQ(ierr); 2230628f99d7SShri Abhyankar } 2231628f99d7SShri Abhyankar 2232628f99d7SShri Abhyankar pj = bj + bi[row]; 2233628f99d7SShri Abhyankar pc1 = ba + bi[row]; 2234628f99d7SShri Abhyankar pc2 = ba + bi[row+1]; 2235628f99d7SShri Abhyankar pc3 = ba + bi[row+2]; 2236628f99d7SShri Abhyankar 2237628f99d7SShri Abhyankar sctx.pv = rtmp33[row+2]; 2238628f99d7SShri Abhyankar rs = 0.0; 2239628f99d7SShri Abhyankar rtmp11[row] = 1.0/rtmp11[row]; 2240628f99d7SShri Abhyankar rtmp22[row+1] = 1.0/rtmp22[row+1]; 2241628f99d7SShri Abhyankar rtmp33[row+2] = 1.0/rtmp33[row+2]; 2242628f99d7SShri Abhyankar /* copy row entries from dense representation to sparse */ 2243628f99d7SShri Abhyankar for (j=0; j<nz; j++) { 2244628f99d7SShri Abhyankar idx = pj[j]; 2245628f99d7SShri Abhyankar pc1[j] = rtmp11[idx]; 2246628f99d7SShri Abhyankar pc2[j] = rtmp22[idx]; 2247628f99d7SShri Abhyankar pc3[j] = rtmp33[idx]; 2248628f99d7SShri Abhyankar if (idx != row+2) rs += PetscAbsScalar(pc3[j]); 2249628f99d7SShri Abhyankar } 2250628f99d7SShri Abhyankar 2251628f99d7SShri Abhyankar sctx.rs = rs; 22524cccfbddSHong Zhang ierr = MatPivotCheck(B,A,info,&sctx,row+2);CHKERRQ(ierr); 225307b50cabSHong Zhang if (sctx.newshift) goto endofwhile; 2254628f99d7SShri Abhyankar break; 2255628f99d7SShri Abhyankar 2256628f99d7SShri Abhyankar default: 2257e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Node size not yet supported \n"); 2258628f99d7SShri Abhyankar } 2259628f99d7SShri Abhyankar row += nodesz; /* Update the row */ 2260628f99d7SShri Abhyankar } 2261628f99d7SShri Abhyankar endofwhile:; 226207b50cabSHong Zhang } while (sctx.newshift); 2263eb9baa12SBarry Smith ierr = PetscFree3(rtmp11,rtmp22,rtmp33);CHKERRQ(ierr); 2264628f99d7SShri Abhyankar ierr = PetscFree(tmp_vec2);CHKERRQ(ierr); 2265628f99d7SShri Abhyankar ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr); 2266628f99d7SShri Abhyankar ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr); 2267628f99d7SShri Abhyankar ierr = ISRestoreIndices(iscol,&c);CHKERRQ(ierr); 22682205254eSKarl Rupp 2269d3ac4fa3SBarry Smith (B)->ops->solve = MatSolve_SeqAIJ_inplace; 2270628f99d7SShri Abhyankar /* do not set solve add, since MatSolve_Inode + Add is faster */ 2271628f99d7SShri Abhyankar C->ops->solvetranspose = MatSolveTranspose_SeqAIJ_inplace; 2272628f99d7SShri Abhyankar C->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace; 2273628f99d7SShri Abhyankar C->assembled = PETSC_TRUE; 2274628f99d7SShri Abhyankar C->preallocated = PETSC_TRUE; 2275628f99d7SShri Abhyankar if (sctx.nshift) { 2276f4db908eSBarry Smith if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { 227757622a8eSBarry Smith ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %g, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,(double)sctx.shift_amount,(double)sctx.shift_fraction,(double)sctx.shift_top);CHKERRQ(ierr); 2278f4db908eSBarry Smith } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) { 227957622a8eSBarry Smith ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %g\n",sctx.nshift,(double)sctx.shift_amount);CHKERRQ(ierr); 2280628f99d7SShri Abhyankar } 2281628f99d7SShri Abhyankar } 2282628f99d7SShri Abhyankar ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr); 2283a02bda8eSBarry Smith ierr = MatSeqAIJCheckInode(C);CHKERRQ(ierr); 2284628f99d7SShri Abhyankar PetscFunctionReturn(0); 2285628f99d7SShri Abhyankar } 2286628f99d7SShri Abhyankar 2287628f99d7SShri Abhyankar 2288019b515eSShri Abhyankar /* ----------------------------------------------------------- */ 228935233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_Inode(Mat A,Vec bb,Vec xx) 2290019b515eSShri Abhyankar { 2291019b515eSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 2292019b515eSShri Abhyankar IS iscol = a->col,isrow = a->row; 2293019b515eSShri Abhyankar PetscErrorCode ierr; 2294019b515eSShri Abhyankar const PetscInt *r,*c,*rout,*cout; 22958758e1faSBarry Smith PetscInt i,j,n = A->rmap->n; 22968758e1faSBarry Smith PetscInt node_max,row,nsz,aii,i0,i1,nz; 22978758e1faSBarry Smith const PetscInt *ai = a->i,*a_j = a->j,*ns,*vi,*ad,*aj; 2298019b515eSShri Abhyankar PetscScalar *x,*tmp,*tmps,tmp0,tmp1; 2299019b515eSShri Abhyankar PetscScalar sum1,sum2,sum3,sum4,sum5; 2300019b515eSShri Abhyankar const MatScalar *v1,*v2,*v3,*v4,*v5,*a_a = a->a,*aa; 2301019b515eSShri Abhyankar const PetscScalar *b; 2302019b515eSShri Abhyankar 2303019b515eSShri Abhyankar PetscFunctionBegin; 2304e32f2f54SBarry Smith if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 2305019b515eSShri Abhyankar node_max = a->inode.node_count; 2306019b515eSShri Abhyankar ns = a->inode.size; /* Node Size array */ 2307019b515eSShri Abhyankar 23083649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 2309fdc842d1SBarry Smith ierr = VecGetArrayWrite(xx,&x);CHKERRQ(ierr); 2310019b515eSShri Abhyankar tmp = a->solve_work; 2311019b515eSShri Abhyankar 2312019b515eSShri Abhyankar ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout; 2313019b515eSShri Abhyankar ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout; 2314019b515eSShri Abhyankar 2315019b515eSShri Abhyankar /* forward solve the lower triangular */ 2316019b515eSShri Abhyankar tmps = tmp; 2317019b515eSShri Abhyankar aa = a_a; 2318019b515eSShri Abhyankar aj = a_j; 2319019b515eSShri Abhyankar ad = a->diag; 2320019b515eSShri Abhyankar 2321019b515eSShri Abhyankar for (i = 0,row = 0; i< node_max; ++i) { 2322019b515eSShri Abhyankar nsz = ns[i]; 2323019b515eSShri Abhyankar aii = ai[row]; 2324019b515eSShri Abhyankar v1 = aa + aii; 2325019b515eSShri Abhyankar vi = aj + aii; 2326019b515eSShri Abhyankar nz = ai[row+1]- ai[row]; 2327019b515eSShri Abhyankar 232898991853SShri Abhyankar if (i < node_max-1) { 232998991853SShri Abhyankar /* Prefetch the indices for the next block */ 233050d8bf02SJed Brown PetscPrefetchBlock(aj+ai[row+nsz],ai[row+nsz+1]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); /* indices */ 233198991853SShri Abhyankar /* Prefetch the data for the next block */ 233250d8bf02SJed Brown PetscPrefetchBlock(aa+ai[row+nsz],ai[row+nsz+ns[i+1]]-ai[row+nsz],0,PETSC_PREFETCH_HINT_NTA); 233398991853SShri Abhyankar } 233498991853SShri Abhyankar 2335019b515eSShri Abhyankar switch (nsz) { /* Each loop in 'case' is unrolled */ 2336019b515eSShri Abhyankar case 1: 2337019b515eSShri Abhyankar sum1 = b[r[row]]; 2338019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2339019b515eSShri Abhyankar i0 = vi[j]; 2340019b515eSShri Abhyankar i1 = vi[j+1]; 2341019b515eSShri Abhyankar tmp0 = tmps[i0]; 2342019b515eSShri Abhyankar tmp1 = tmps[i1]; 2343019b515eSShri Abhyankar sum1 -= v1[j]*tmp0 + v1[j+1]*tmp1; 2344019b515eSShri Abhyankar } 2345019b515eSShri Abhyankar if (j == nz-1) { 2346019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2347019b515eSShri Abhyankar sum1 -= v1[j]*tmp0; 2348019b515eSShri Abhyankar } 2349019b515eSShri Abhyankar tmp[row++]=sum1; 2350019b515eSShri Abhyankar break; 2351019b515eSShri Abhyankar case 2: 2352019b515eSShri Abhyankar sum1 = b[r[row]]; 2353019b515eSShri Abhyankar sum2 = b[r[row+1]]; 2354019b515eSShri Abhyankar v2 = aa + ai[row+1]; 2355019b515eSShri Abhyankar 2356019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2357019b515eSShri Abhyankar i0 = vi[j]; 2358019b515eSShri Abhyankar i1 = vi[j+1]; 2359019b515eSShri Abhyankar tmp0 = tmps[i0]; 2360019b515eSShri Abhyankar tmp1 = tmps[i1]; 2361019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2362019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 2363019b515eSShri Abhyankar } 2364019b515eSShri Abhyankar if (j == nz-1) { 2365019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2366019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 2367019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 2368019b515eSShri Abhyankar } 2369019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 2370019b515eSShri Abhyankar tmp[row++]=sum1; 2371019b515eSShri Abhyankar tmp[row++]=sum2; 2372019b515eSShri Abhyankar break; 2373019b515eSShri Abhyankar case 3: 2374019b515eSShri Abhyankar sum1 = b[r[row]]; 2375019b515eSShri Abhyankar sum2 = b[r[row+1]]; 2376019b515eSShri Abhyankar sum3 = b[r[row+2]]; 2377019b515eSShri Abhyankar v2 = aa + ai[row+1]; 2378019b515eSShri Abhyankar v3 = aa + ai[row+2]; 2379019b515eSShri Abhyankar 2380019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2381019b515eSShri Abhyankar i0 = vi[j]; 2382019b515eSShri Abhyankar i1 = vi[j+1]; 2383019b515eSShri Abhyankar tmp0 = tmps[i0]; 2384019b515eSShri Abhyankar tmp1 = tmps[i1]; 2385019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2386019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 2387019b515eSShri Abhyankar sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1; 2388019b515eSShri Abhyankar } 2389019b515eSShri Abhyankar if (j == nz-1) { 2390019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2391019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 2392019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 2393019b515eSShri Abhyankar sum3 -= v3[j] *tmp0; 2394019b515eSShri Abhyankar } 2395019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 2396019b515eSShri Abhyankar sum3 -= v3[nz] * sum1; 2397019b515eSShri Abhyankar sum3 -= v3[nz+1] * sum2; 2398019b515eSShri Abhyankar tmp[row++]=sum1; 2399019b515eSShri Abhyankar tmp[row++]=sum2; 2400019b515eSShri Abhyankar tmp[row++]=sum3; 2401019b515eSShri Abhyankar break; 2402019b515eSShri Abhyankar 2403019b515eSShri Abhyankar case 4: 2404019b515eSShri Abhyankar sum1 = b[r[row]]; 2405019b515eSShri Abhyankar sum2 = b[r[row+1]]; 2406019b515eSShri Abhyankar sum3 = b[r[row+2]]; 2407019b515eSShri Abhyankar sum4 = b[r[row+3]]; 2408019b515eSShri Abhyankar v2 = aa + ai[row+1]; 2409019b515eSShri Abhyankar v3 = aa + ai[row+2]; 2410019b515eSShri Abhyankar v4 = aa + ai[row+3]; 2411019b515eSShri Abhyankar 2412019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2413019b515eSShri Abhyankar i0 = vi[j]; 2414019b515eSShri Abhyankar i1 = vi[j+1]; 2415019b515eSShri Abhyankar tmp0 = tmps[i0]; 2416019b515eSShri Abhyankar tmp1 = tmps[i1]; 2417019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2418019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 2419019b515eSShri Abhyankar sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1; 2420019b515eSShri Abhyankar sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1; 2421019b515eSShri Abhyankar } 2422019b515eSShri Abhyankar if (j == nz-1) { 2423019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2424019b515eSShri Abhyankar sum1 -= v1[j] *tmp0; 2425019b515eSShri Abhyankar sum2 -= v2[j] *tmp0; 2426019b515eSShri Abhyankar sum3 -= v3[j] *tmp0; 2427019b515eSShri Abhyankar sum4 -= v4[j] *tmp0; 2428019b515eSShri Abhyankar } 2429019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 2430019b515eSShri Abhyankar sum3 -= v3[nz] * sum1; 2431019b515eSShri Abhyankar sum4 -= v4[nz] * sum1; 2432019b515eSShri Abhyankar sum3 -= v3[nz+1] * sum2; 2433019b515eSShri Abhyankar sum4 -= v4[nz+1] * sum2; 2434019b515eSShri Abhyankar sum4 -= v4[nz+2] * sum3; 2435019b515eSShri Abhyankar 2436019b515eSShri Abhyankar tmp[row++]=sum1; 2437019b515eSShri Abhyankar tmp[row++]=sum2; 2438019b515eSShri Abhyankar tmp[row++]=sum3; 2439019b515eSShri Abhyankar tmp[row++]=sum4; 2440019b515eSShri Abhyankar break; 2441019b515eSShri Abhyankar case 5: 2442019b515eSShri Abhyankar sum1 = b[r[row]]; 2443019b515eSShri Abhyankar sum2 = b[r[row+1]]; 2444019b515eSShri Abhyankar sum3 = b[r[row+2]]; 2445019b515eSShri Abhyankar sum4 = b[r[row+3]]; 2446019b515eSShri Abhyankar sum5 = b[r[row+4]]; 2447019b515eSShri Abhyankar v2 = aa + ai[row+1]; 2448019b515eSShri Abhyankar v3 = aa + ai[row+2]; 2449019b515eSShri Abhyankar v4 = aa + ai[row+3]; 2450019b515eSShri Abhyankar v5 = aa + ai[row+4]; 2451019b515eSShri Abhyankar 2452019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2453019b515eSShri Abhyankar i0 = vi[j]; 2454019b515eSShri Abhyankar i1 = vi[j+1]; 2455019b515eSShri Abhyankar tmp0 = tmps[i0]; 2456019b515eSShri Abhyankar tmp1 = tmps[i1]; 2457019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2458019b515eSShri Abhyankar sum2 -= v2[j] * tmp0 + v2[j+1] * tmp1; 2459019b515eSShri Abhyankar sum3 -= v3[j] * tmp0 + v3[j+1] * tmp1; 2460019b515eSShri Abhyankar sum4 -= v4[j] * tmp0 + v4[j+1] * tmp1; 2461019b515eSShri Abhyankar sum5 -= v5[j] * tmp0 + v5[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 sum5 -= v5[j] *tmp0; 2470019b515eSShri Abhyankar } 2471019b515eSShri Abhyankar 2472019b515eSShri Abhyankar sum2 -= v2[nz] * sum1; 2473019b515eSShri Abhyankar sum3 -= v3[nz] * sum1; 2474019b515eSShri Abhyankar sum4 -= v4[nz] * sum1; 2475019b515eSShri Abhyankar sum5 -= v5[nz] * sum1; 2476019b515eSShri Abhyankar sum3 -= v3[nz+1] * sum2; 2477019b515eSShri Abhyankar sum4 -= v4[nz+1] * sum2; 2478019b515eSShri Abhyankar sum5 -= v5[nz+1] * sum2; 2479019b515eSShri Abhyankar sum4 -= v4[nz+2] * sum3; 2480019b515eSShri Abhyankar sum5 -= v5[nz+2] * sum3; 2481019b515eSShri Abhyankar sum5 -= v5[nz+3] * sum4; 2482019b515eSShri Abhyankar 2483019b515eSShri Abhyankar tmp[row++]=sum1; 2484019b515eSShri Abhyankar tmp[row++]=sum2; 2485019b515eSShri Abhyankar tmp[row++]=sum3; 2486019b515eSShri Abhyankar tmp[row++]=sum4; 2487019b515eSShri Abhyankar tmp[row++]=sum5; 2488019b515eSShri Abhyankar break; 2489019b515eSShri Abhyankar default: 2490e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n"); 2491019b515eSShri Abhyankar } 2492019b515eSShri Abhyankar } 2493019b515eSShri Abhyankar /* backward solve the upper triangular */ 2494019b515eSShri Abhyankar for (i=node_max -1,row = n-1; i>=0; i--) { 2495019b515eSShri Abhyankar nsz = ns[i]; 2496019b515eSShri Abhyankar aii = ad[row+1] + 1; 2497019b515eSShri Abhyankar v1 = aa + aii; 2498019b515eSShri Abhyankar vi = aj + aii; 2499019b515eSShri Abhyankar nz = ad[row]- ad[row+1] - 1; 250098991853SShri Abhyankar 250198991853SShri Abhyankar if (i > 0) { 250298991853SShri Abhyankar /* Prefetch the indices for the next block */ 250350d8bf02SJed Brown PetscPrefetchBlock(aj+ad[row-nsz+1]+1,ad[row-nsz]-ad[row-nsz+1],0,PETSC_PREFETCH_HINT_NTA); 250498991853SShri Abhyankar /* Prefetch the data for the next block */ 250550d8bf02SJed Brown PetscPrefetchBlock(aa+ad[row-nsz+1]+1,ad[row-nsz-ns[i-1]+1]-ad[row-nsz+1],0,PETSC_PREFETCH_HINT_NTA); 250698991853SShri Abhyankar } 250798991853SShri Abhyankar 2508019b515eSShri Abhyankar switch (nsz) { /* Each loop in 'case' is unrolled */ 2509019b515eSShri Abhyankar case 1: 2510019b515eSShri Abhyankar sum1 = tmp[row]; 2511019b515eSShri Abhyankar 2512019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2513019b515eSShri Abhyankar i0 = vi[j]; 2514019b515eSShri Abhyankar i1 = vi[j+1]; 2515019b515eSShri Abhyankar tmp0 = tmps[i0]; 2516019b515eSShri Abhyankar tmp1 = tmps[i1]; 2517019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2518019b515eSShri Abhyankar } 2519019b515eSShri Abhyankar if (j == nz-1) { 2520019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2521019b515eSShri Abhyankar sum1 -= v1[j]*tmp0; 2522019b515eSShri Abhyankar } 2523019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 2524019b515eSShri Abhyankar break; 2525019b515eSShri Abhyankar case 2: 2526019b515eSShri Abhyankar sum1 = tmp[row]; 2527019b515eSShri Abhyankar sum2 = tmp[row-1]; 2528019b515eSShri Abhyankar v2 = aa + ad[row] + 1; 2529019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2530019b515eSShri Abhyankar i0 = vi[j]; 2531019b515eSShri Abhyankar i1 = vi[j+1]; 2532019b515eSShri Abhyankar tmp0 = tmps[i0]; 2533019b515eSShri Abhyankar tmp1 = tmps[i1]; 2534019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2535019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 2536019b515eSShri Abhyankar } 2537019b515eSShri Abhyankar if (j == nz-1) { 2538019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2539019b515eSShri Abhyankar sum1 -= v1[j]* tmp0; 2540019b515eSShri Abhyankar sum2 -= v2[j+1]* tmp0; 2541019b515eSShri Abhyankar } 2542019b515eSShri Abhyankar 2543019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 2544019b515eSShri Abhyankar sum2 -= v2[0] * tmp0; 2545019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 2546019b515eSShri Abhyankar break; 2547019b515eSShri Abhyankar case 3: 2548019b515eSShri Abhyankar sum1 = tmp[row]; 2549019b515eSShri Abhyankar sum2 = tmp[row -1]; 2550019b515eSShri Abhyankar sum3 = tmp[row -2]; 2551019b515eSShri Abhyankar v2 = aa + ad[row] + 1; 2552019b515eSShri Abhyankar v3 = aa + ad[row -1] + 1; 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 sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 2560019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1; 2561019b515eSShri Abhyankar } 2562019b515eSShri Abhyankar if (j== nz-1) { 2563019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2564019b515eSShri Abhyankar sum1 -= v1[j] * tmp0; 2565019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0; 2566019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0; 2567019b515eSShri Abhyankar } 2568019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 2569019b515eSShri Abhyankar sum2 -= v2[0]* tmp0; 2570019b515eSShri Abhyankar sum3 -= v3[1] * tmp0; 2571019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 2572019b515eSShri Abhyankar sum3 -= v3[0]* tmp0; 2573019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--; 2574019b515eSShri Abhyankar 2575019b515eSShri Abhyankar break; 2576019b515eSShri Abhyankar case 4: 2577019b515eSShri Abhyankar sum1 = tmp[row]; 2578019b515eSShri Abhyankar sum2 = tmp[row -1]; 2579019b515eSShri Abhyankar sum3 = tmp[row -2]; 2580019b515eSShri Abhyankar sum4 = tmp[row -3]; 2581019b515eSShri Abhyankar v2 = aa + ad[row]+1; 2582019b515eSShri Abhyankar v3 = aa + ad[row -1]+1; 2583019b515eSShri Abhyankar v4 = aa + ad[row -2]+1; 2584019b515eSShri Abhyankar 2585019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2586019b515eSShri Abhyankar i0 = vi[j]; 2587019b515eSShri Abhyankar i1 = vi[j+1]; 2588019b515eSShri Abhyankar tmp0 = tmps[i0]; 2589019b515eSShri Abhyankar tmp1 = tmps[i1]; 2590019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2591019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 2592019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1; 2593019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1; 2594019b515eSShri Abhyankar } 2595019b515eSShri Abhyankar if (j== nz-1) { 2596019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2597019b515eSShri Abhyankar sum1 -= v1[j] * tmp0; 2598019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0; 2599019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0; 2600019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0; 2601019b515eSShri Abhyankar } 2602019b515eSShri Abhyankar 2603019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum1*v1[nz]; row--; 2604019b515eSShri Abhyankar sum2 -= v2[0] * tmp0; 2605019b515eSShri Abhyankar sum3 -= v3[1] * tmp0; 2606019b515eSShri Abhyankar sum4 -= v4[2] * tmp0; 2607019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 2608019b515eSShri Abhyankar sum3 -= v3[0] * tmp0; 2609019b515eSShri Abhyankar sum4 -= v4[1] * tmp0; 2610019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--; 2611019b515eSShri Abhyankar sum4 -= v4[0] * tmp0; 2612019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--; 2613019b515eSShri Abhyankar break; 2614019b515eSShri Abhyankar case 5: 2615019b515eSShri Abhyankar sum1 = tmp[row]; 2616019b515eSShri Abhyankar sum2 = tmp[row -1]; 2617019b515eSShri Abhyankar sum3 = tmp[row -2]; 2618019b515eSShri Abhyankar sum4 = tmp[row -3]; 2619019b515eSShri Abhyankar sum5 = tmp[row -4]; 2620019b515eSShri Abhyankar v2 = aa + ad[row]+1; 2621019b515eSShri Abhyankar v3 = aa + ad[row -1]+1; 2622019b515eSShri Abhyankar v4 = aa + ad[row -2]+1; 2623019b515eSShri Abhyankar v5 = aa + ad[row -3]+1; 2624019b515eSShri Abhyankar for (j=0; j<nz-1; j+=2) { 2625019b515eSShri Abhyankar i0 = vi[j]; 2626019b515eSShri Abhyankar i1 = vi[j+1]; 2627019b515eSShri Abhyankar tmp0 = tmps[i0]; 2628019b515eSShri Abhyankar tmp1 = tmps[i1]; 2629019b515eSShri Abhyankar sum1 -= v1[j] * tmp0 + v1[j+1] * tmp1; 2630019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0 + v2[j+2] * tmp1; 2631019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0 + v3[j+3] * tmp1; 2632019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0 + v4[j+4] * tmp1; 2633019b515eSShri Abhyankar sum5 -= v5[j+4] * tmp0 + v5[j+5] * tmp1; 2634019b515eSShri Abhyankar } 2635019b515eSShri Abhyankar if (j==nz-1) { 2636019b515eSShri Abhyankar tmp0 = tmps[vi[j]]; 2637019b515eSShri Abhyankar sum1 -= v1[j] * tmp0; 2638019b515eSShri Abhyankar sum2 -= v2[j+1] * tmp0; 2639019b515eSShri Abhyankar sum3 -= v3[j+2] * tmp0; 2640019b515eSShri Abhyankar sum4 -= v4[j+3] * tmp0; 2641019b515eSShri Abhyankar sum5 -= v5[j+4] * 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 sum5 -= v5[3] * tmp0; 2649019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum2*v2[nz+1]; row--; 2650019b515eSShri Abhyankar sum3 -= v3[0] * tmp0; 2651019b515eSShri Abhyankar sum4 -= v4[1] * tmp0; 2652019b515eSShri Abhyankar sum5 -= v5[2] * tmp0; 2653019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum3*v3[nz+2]; row--; 2654019b515eSShri Abhyankar sum4 -= v4[0] * tmp0; 2655019b515eSShri Abhyankar sum5 -= v5[1] * tmp0; 2656019b515eSShri Abhyankar tmp0 = x[c[row]] = tmp[row] = sum4*v4[nz+3]; row--; 2657019b515eSShri Abhyankar sum5 -= v5[0] * tmp0; 2658019b515eSShri Abhyankar x[c[row]] = tmp[row] = sum5*v5[nz+4]; row--; 2659019b515eSShri Abhyankar break; 2660019b515eSShri Abhyankar default: 2661e32f2f54SBarry Smith SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Node size not yet supported \n"); 2662019b515eSShri Abhyankar } 2663019b515eSShri Abhyankar } 2664019b515eSShri Abhyankar ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr); 2665019b515eSShri Abhyankar ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr); 26663649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 2667fdc842d1SBarry Smith ierr = VecRestoreArrayWrite(xx,&x);CHKERRQ(ierr); 2668019b515eSShri Abhyankar ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr); 2669019b515eSShri Abhyankar PetscFunctionReturn(0); 2670019b515eSShri Abhyankar } 2671019b515eSShri Abhyankar 2672019b515eSShri Abhyankar 26734c1414c8SBarry Smith /* 26744c1414c8SBarry Smith Makes a longer coloring[] array and calls the usual code with that 26754c1414c8SBarry Smith */ 26764108e4d5SBarry Smith PetscErrorCode MatColoringPatch_SeqAIJ_Inode(Mat mat,PetscInt ncolors,PetscInt nin,ISColoringValue coloring[],ISColoring *iscoloring) 26774c1414c8SBarry Smith { 26784c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)mat->data; 26794c1414c8SBarry Smith PetscErrorCode ierr; 2680d0f46423SBarry Smith PetscInt n = mat->cmap->n,m = a->inode.node_count,j,*ns = a->inode.size,row; 26814c1414c8SBarry Smith PetscInt *colorused,i; 26824c1414c8SBarry Smith ISColoringValue *newcolor; 26834c1414c8SBarry Smith 26844c1414c8SBarry Smith PetscFunctionBegin; 2685b215bc84SStefano Zampini if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 2686854ce69bSBarry Smith ierr = PetscMalloc1(n+1,&newcolor);CHKERRQ(ierr); 26874c1414c8SBarry Smith /* loop over inodes, marking a color for each column*/ 26884c1414c8SBarry Smith row = 0; 26894c1414c8SBarry Smith for (i=0; i<m; i++) { 26904c1414c8SBarry Smith for (j=0; j<ns[i]; j++) { 26914c1414c8SBarry Smith newcolor[row++] = coloring[i] + j*ncolors; 26924c1414c8SBarry Smith } 26934c1414c8SBarry Smith } 26944c1414c8SBarry Smith 26954c1414c8SBarry Smith /* eliminate unneeded colors */ 26961795a4d1SJed Brown ierr = PetscCalloc1(5*ncolors,&colorused);CHKERRQ(ierr); 26974c1414c8SBarry Smith for (i=0; i<n; i++) { 26984c1414c8SBarry Smith colorused[newcolor[i]] = 1; 26994c1414c8SBarry Smith } 27004c1414c8SBarry Smith 27014c1414c8SBarry Smith for (i=1; i<5*ncolors; i++) { 27024c1414c8SBarry Smith colorused[i] += colorused[i-1]; 27034c1414c8SBarry Smith } 27044c1414c8SBarry Smith ncolors = colorused[5*ncolors-1]; 27054c1414c8SBarry Smith for (i=0; i<n; i++) { 270657815681SSatish Balay newcolor[i] = colorused[newcolor[i]]-1; 27074c1414c8SBarry Smith } 27084c1414c8SBarry Smith ierr = PetscFree(colorused);CHKERRQ(ierr); 2709aaf3ff59SMatthew G. Knepley ierr = ISColoringCreate(PetscObjectComm((PetscObject)mat),ncolors,n,newcolor,PETSC_OWN_POINTER,iscoloring);CHKERRQ(ierr); 27104c1414c8SBarry Smith ierr = PetscFree(coloring);CHKERRQ(ierr); 27114c1414c8SBarry Smith PetscFunctionReturn(0); 27124c1414c8SBarry Smith } 27134c1414c8SBarry Smith 2714af0996ceSBarry Smith #include <petsc/private/kernels/blockinvert.h> 27152af78befSBarry Smith 27164108e4d5SBarry Smith PetscErrorCode MatSOR_SeqAIJ_Inode(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx) 27172af78befSBarry Smith { 27182af78befSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 27197aaeff0aSMatthew G. Knepley PetscScalar sum1 = 0.0,sum2 = 0.0,sum3 = 0.0,sum4 = 0.0,sum5 = 0.0,tmp0,tmp1,tmp2,tmp3; 27205850ef23SBarry Smith MatScalar *ibdiag,*bdiag,work[25],*t; 2721a8b09249SBarry Smith PetscScalar *x,tmp4,tmp5,x1,x2,x3,x4,x5; 27227aaeff0aSMatthew G. Knepley const MatScalar *v = a->a,*v1 = NULL,*v2 = NULL,*v3 = NULL,*v4 = NULL,*v5 = NULL; 27235850ef23SBarry Smith const PetscScalar *xb, *b; 27247b6c816cSBarry Smith PetscReal zeropivot = 100.*PETSC_MACHINE_EPSILON, shift = 0.0; 27252af78befSBarry Smith PetscErrorCode ierr; 27268758e1faSBarry Smith PetscInt n,m = a->inode.node_count,cnt = 0,i,j,row,i1,i2; 27278758e1faSBarry Smith PetscInt sz,k,ipvt[5]; 27287b6c816cSBarry Smith PetscBool allowzeropivot,zeropivotdetected; 27298758e1faSBarry Smith const PetscInt *sizes = a->inode.size,*idx,*diag = a->diag,*ii = a->i; 27302af78befSBarry Smith 27312af78befSBarry Smith PetscFunctionBegin; 2732a455e926SHong Zhang allowzeropivot = PetscNot(A->erroriffailure); 2733b215bc84SStefano Zampini if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 2734e32f2f54SBarry Smith if (omega != 1.0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for omega != 1.0; use -mat_no_inode"); 2735e32f2f54SBarry Smith if (fshift != 0.0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for fshift != 0.0; use -mat_no_inode"); 27362af78befSBarry Smith 273771f1c65dSBarry Smith if (!a->inode.ibdiagvalid) { 27382af78befSBarry Smith if (!a->inode.ibdiag) { 27392af78befSBarry Smith /* calculate space needed for diagonal blocks */ 27402af78befSBarry Smith for (i=0; i<m; i++) { 27412af78befSBarry Smith cnt += sizes[i]*sizes[i]; 27422af78befSBarry Smith } 2743f0d39aaaSBarry Smith a->inode.bdiagsize = cnt; 27442205254eSKarl Rupp 2745dcca6d9dSJed Brown ierr = PetscMalloc3(cnt,&a->inode.ibdiag,cnt,&a->inode.bdiag,A->rmap->n,&a->inode.ssor_work);CHKERRQ(ierr); 274671f1c65dSBarry Smith } 274771f1c65dSBarry Smith 274871f1c65dSBarry Smith /* copy over the diagonal blocks and invert them */ 27492af78befSBarry Smith ibdiag = a->inode.ibdiag; 27502af78befSBarry Smith bdiag = a->inode.bdiag; 27512af78befSBarry Smith cnt = 0; 27522af78befSBarry Smith for (i=0, row = 0; i<m; i++) { 27532af78befSBarry Smith for (j=0; j<sizes[i]; j++) { 2754f0d39aaaSBarry Smith for (k=0; k<sizes[i]; k++) { 2755f0d39aaaSBarry Smith bdiag[cnt+k*sizes[i]+j] = v[diag[row+j] - j + k]; 2756f0d39aaaSBarry Smith } 27572af78befSBarry Smith } 2758580bdb30SBarry Smith ierr = PetscArraycpy(ibdiag+cnt,bdiag+cnt,sizes[i]*sizes[i]);CHKERRQ(ierr); 27592af78befSBarry Smith 27602af78befSBarry Smith switch (sizes[i]) { 27612af78befSBarry Smith case 1: 27622af78befSBarry Smith /* Create matrix data structure */ 27638e0e2a9aSHong Zhang if (PetscAbsScalar(ibdiag[cnt]) < zeropivot) { 27648e0e2a9aSHong Zhang if (allowzeropivot) { 27657b6c816cSBarry Smith A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 27667b6c816cSBarry Smith A->factorerror_zeropivot_value = PetscAbsScalar(ibdiag[cnt]); 27677b6c816cSBarry Smith A->factorerror_zeropivot_row = row; 27688e0e2a9aSHong Zhang ierr = PetscInfo1(A,"Zero pivot, row %D\n",row);CHKERRQ(ierr); 27698e0e2a9aSHong Zhang } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot on row %D",row); 27708e0e2a9aSHong Zhang } 277164c62002SMatthew Knepley ibdiag[cnt] = 1.0/ibdiag[cnt]; 27722af78befSBarry Smith break; 27732af78befSBarry Smith case 2: 2774a455e926SHong Zhang ierr = PetscKernel_A_gets_inverse_A_2(ibdiag+cnt,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr); 27757b6c816cSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 27762af78befSBarry Smith break; 27772af78befSBarry Smith case 3: 2778a455e926SHong Zhang ierr = PetscKernel_A_gets_inverse_A_3(ibdiag+cnt,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr); 27797b6c816cSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 27802af78befSBarry Smith break; 27812af78befSBarry Smith case 4: 2782a455e926SHong Zhang ierr = PetscKernel_A_gets_inverse_A_4(ibdiag+cnt,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr); 27837b6c816cSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 27842af78befSBarry Smith break; 27852af78befSBarry Smith case 5: 2786a455e926SHong Zhang ierr = PetscKernel_A_gets_inverse_A_5(ibdiag+cnt,ipvt,work,shift,allowzeropivot,&zeropivotdetected);CHKERRQ(ierr); 27877b6c816cSBarry Smith if (zeropivotdetected) A->factorerrortype = MAT_FACTOR_NUMERIC_ZEROPIVOT; 27882af78befSBarry Smith break; 27892af78befSBarry Smith default: 2790e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 27912af78befSBarry Smith } 27922af78befSBarry Smith cnt += sizes[i]*sizes[i]; 27932af78befSBarry Smith row += sizes[i]; 27942af78befSBarry Smith } 279571f1c65dSBarry Smith a->inode.ibdiagvalid = PETSC_TRUE; 27962af78befSBarry Smith } 27972af78befSBarry Smith ibdiag = a->inode.ibdiag; 27982af78befSBarry Smith bdiag = a->inode.bdiag; 27995850ef23SBarry Smith t = a->inode.ssor_work; 28002af78befSBarry Smith 28012af78befSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 28023649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 28035850ef23SBarry Smith /* We count flops by assuming the upper triangular and lower triangular parts have the same number of nonzeros */ 28045850ef23SBarry Smith if (flag & SOR_ZERO_INITIAL_GUESS) { 28052af78befSBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) { 28062af78befSBarry Smith 28078862d2efSBarry Smith for (i=0, row=0; i<m; i++) { 28088862d2efSBarry Smith sz = diag[row] - ii[row]; 28098862d2efSBarry Smith v1 = a->a + ii[row]; 28108862d2efSBarry Smith idx = a->j + ii[row]; 28118862d2efSBarry Smith 28124108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 28138862d2efSBarry Smith switch (sizes[i]) { 28148862d2efSBarry Smith case 1: 28158862d2efSBarry Smith 28168862d2efSBarry Smith sum1 = b[row]; 28178862d2efSBarry Smith for (n = 0; n<sz-1; n+=2) { 28188862d2efSBarry Smith i1 = idx[0]; 28198862d2efSBarry Smith i2 = idx[1]; 28208862d2efSBarry Smith idx += 2; 28218862d2efSBarry Smith tmp0 = x[i1]; 28228862d2efSBarry Smith tmp1 = x[i2]; 28238862d2efSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 28248862d2efSBarry Smith } 28258862d2efSBarry Smith 28268862d2efSBarry Smith if (n == sz-1) { 2827f0d39aaaSBarry Smith tmp0 = x[*idx]; 2828f0d39aaaSBarry Smith sum1 -= *v1 * tmp0; 28298862d2efSBarry Smith } 28305850ef23SBarry Smith t[row] = sum1; 28318862d2efSBarry Smith x[row++] = sum1*(*ibdiag++); 28328862d2efSBarry Smith break; 2833f0d39aaaSBarry Smith case 2: 2834f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 2835f0d39aaaSBarry Smith sum1 = b[row]; 2836f0d39aaaSBarry Smith sum2 = b[row+1]; 2837f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 2838f0d39aaaSBarry Smith i1 = idx[0]; 2839f0d39aaaSBarry Smith i2 = idx[1]; 2840f0d39aaaSBarry Smith idx += 2; 2841f0d39aaaSBarry Smith tmp0 = x[i1]; 2842f0d39aaaSBarry Smith tmp1 = x[i2]; 2843f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 2844f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 2845f0d39aaaSBarry Smith } 2846f0d39aaaSBarry Smith 2847f0d39aaaSBarry Smith if (n == sz-1) { 2848f0d39aaaSBarry Smith tmp0 = x[*idx]; 2849f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 2850f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 2851f0d39aaaSBarry Smith } 28525850ef23SBarry Smith t[row] = sum1; 28535850ef23SBarry Smith t[row+1] = sum2; 2854f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[2]; 2855f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[3]; 2856f0d39aaaSBarry Smith ibdiag += 4; 2857f0d39aaaSBarry Smith break; 2858f0d39aaaSBarry Smith case 3: 2859f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 2860f0d39aaaSBarry Smith v3 = a->a + ii[row+2]; 2861f0d39aaaSBarry Smith sum1 = b[row]; 2862f0d39aaaSBarry Smith sum2 = b[row+1]; 2863f0d39aaaSBarry Smith sum3 = b[row+2]; 2864f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 2865f0d39aaaSBarry Smith i1 = idx[0]; 2866f0d39aaaSBarry Smith i2 = idx[1]; 2867f0d39aaaSBarry Smith idx += 2; 2868f0d39aaaSBarry Smith tmp0 = x[i1]; 2869f0d39aaaSBarry Smith tmp1 = x[i2]; 2870f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 2871f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 2872f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 2873f0d39aaaSBarry Smith } 2874f0d39aaaSBarry Smith 2875f0d39aaaSBarry Smith if (n == sz-1) { 2876f0d39aaaSBarry Smith tmp0 = x[*idx]; 2877f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 2878f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 2879f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0; 2880f0d39aaaSBarry Smith } 28815850ef23SBarry Smith t[row] = sum1; 28825850ef23SBarry Smith t[row+1] = sum2; 28835850ef23SBarry Smith t[row+2] = sum3; 2884f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6]; 2885f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7]; 2886f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8]; 2887f0d39aaaSBarry Smith ibdiag += 9; 2888f0d39aaaSBarry Smith break; 2889f0d39aaaSBarry Smith case 4: 2890f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 2891f0d39aaaSBarry Smith v3 = a->a + ii[row+2]; 2892f0d39aaaSBarry Smith v4 = a->a + ii[row+3]; 2893f0d39aaaSBarry Smith sum1 = b[row]; 2894f0d39aaaSBarry Smith sum2 = b[row+1]; 2895f0d39aaaSBarry Smith sum3 = b[row+2]; 2896f0d39aaaSBarry Smith sum4 = b[row+3]; 2897f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 2898f0d39aaaSBarry Smith i1 = idx[0]; 2899f0d39aaaSBarry Smith i2 = idx[1]; 2900f0d39aaaSBarry Smith idx += 2; 2901f0d39aaaSBarry Smith tmp0 = x[i1]; 2902f0d39aaaSBarry Smith tmp1 = x[i2]; 2903f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 2904f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 2905f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 2906f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 2907f0d39aaaSBarry Smith } 2908f0d39aaaSBarry Smith 2909f0d39aaaSBarry Smith if (n == sz-1) { 2910f0d39aaaSBarry Smith tmp0 = x[*idx]; 2911f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 2912f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 2913f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0; 2914f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0; 2915f0d39aaaSBarry Smith } 29165850ef23SBarry Smith t[row] = sum1; 29175850ef23SBarry Smith t[row+1] = sum2; 29185850ef23SBarry Smith t[row+2] = sum3; 29195850ef23SBarry Smith t[row+3] = sum4; 2920f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12]; 2921f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13]; 2922f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14]; 2923f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15]; 2924f0d39aaaSBarry Smith ibdiag += 16; 2925f0d39aaaSBarry Smith break; 2926f0d39aaaSBarry Smith case 5: 2927f0d39aaaSBarry Smith v2 = a->a + ii[row+1]; 2928f0d39aaaSBarry Smith v3 = a->a + ii[row+2]; 2929f0d39aaaSBarry Smith v4 = a->a + ii[row+3]; 2930f0d39aaaSBarry Smith v5 = a->a + ii[row+4]; 2931f0d39aaaSBarry Smith sum1 = b[row]; 2932f0d39aaaSBarry Smith sum2 = b[row+1]; 2933f0d39aaaSBarry Smith sum3 = b[row+2]; 2934f0d39aaaSBarry Smith sum4 = b[row+3]; 2935f0d39aaaSBarry Smith sum5 = b[row+4]; 2936f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 2937f0d39aaaSBarry Smith i1 = idx[0]; 2938f0d39aaaSBarry Smith i2 = idx[1]; 2939f0d39aaaSBarry Smith idx += 2; 2940f0d39aaaSBarry Smith tmp0 = x[i1]; 2941f0d39aaaSBarry Smith tmp1 = x[i2]; 2942f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 2943f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 2944f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 2945f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 2946f0d39aaaSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 2947f0d39aaaSBarry Smith } 2948f0d39aaaSBarry Smith 2949f0d39aaaSBarry Smith if (n == sz-1) { 2950f0d39aaaSBarry Smith tmp0 = x[*idx]; 2951f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0; 2952f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0; 2953f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0; 2954f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0; 2955f0d39aaaSBarry Smith sum5 -= v5[0] * tmp0; 2956f0d39aaaSBarry Smith } 29575850ef23SBarry Smith t[row] = sum1; 29585850ef23SBarry Smith t[row+1] = sum2; 29595850ef23SBarry Smith t[row+2] = sum3; 29605850ef23SBarry Smith t[row+3] = sum4; 29615850ef23SBarry Smith t[row+4] = sum5; 2962f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20]; 2963f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21]; 2964f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22]; 2965f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23]; 2966f0d39aaaSBarry Smith x[row++] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24]; 2967f0d39aaaSBarry Smith ibdiag += 25; 2968f0d39aaaSBarry Smith break; 29698862d2efSBarry Smith default: 2970e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 29718862d2efSBarry Smith } 29722af78befSBarry Smith } 29732af78befSBarry Smith 29745850ef23SBarry Smith xb = t; 29752af78befSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 29762af78befSBarry Smith } else xb = b; 29772af78befSBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) { 29782af78befSBarry Smith 2979f0d39aaaSBarry Smith ibdiag = a->inode.ibdiag+a->inode.bdiagsize; 2980d0f46423SBarry Smith for (i=m-1, row=A->rmap->n-1; i>=0; i--) { 2981f0d39aaaSBarry Smith ibdiag -= sizes[i]*sizes[i]; 29828862d2efSBarry Smith sz = ii[row+1] - diag[row] - 1; 29838862d2efSBarry Smith v1 = a->a + diag[row] + 1; 29848862d2efSBarry Smith idx = a->j + diag[row] + 1; 29852af78befSBarry Smith 29864108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 29878862d2efSBarry Smith switch (sizes[i]) { 29888862d2efSBarry Smith case 1: 29898862d2efSBarry Smith 29908862d2efSBarry Smith sum1 = xb[row]; 29918862d2efSBarry Smith for (n = 0; n<sz-1; n+=2) { 29928862d2efSBarry Smith i1 = idx[0]; 29938862d2efSBarry Smith i2 = idx[1]; 29948862d2efSBarry Smith idx += 2; 29958862d2efSBarry Smith tmp0 = x[i1]; 29968862d2efSBarry Smith tmp1 = x[i2]; 29978862d2efSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 29988862d2efSBarry Smith } 29998862d2efSBarry Smith 30008862d2efSBarry Smith if (n == sz-1) { 3001f0d39aaaSBarry Smith tmp0 = x[*idx]; 3002f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 30038862d2efSBarry Smith } 3004f0d39aaaSBarry Smith x[row--] = sum1*(*ibdiag); 3005f0d39aaaSBarry Smith break; 3006f0d39aaaSBarry Smith 3007f0d39aaaSBarry Smith case 2: 3008f0d39aaaSBarry Smith 3009f0d39aaaSBarry Smith sum1 = xb[row]; 3010f0d39aaaSBarry Smith sum2 = xb[row-1]; 3011f0d39aaaSBarry Smith /* note that sum1 is associated with the second of the two rows */ 3012f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 3013f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 3014f0d39aaaSBarry Smith i1 = idx[0]; 3015f0d39aaaSBarry Smith i2 = idx[1]; 3016f0d39aaaSBarry Smith idx += 2; 3017f0d39aaaSBarry Smith tmp0 = x[i1]; 3018f0d39aaaSBarry Smith tmp1 = x[i2]; 3019f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3020f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3021f0d39aaaSBarry Smith } 3022f0d39aaaSBarry Smith 3023f0d39aaaSBarry Smith if (n == sz-1) { 3024f0d39aaaSBarry Smith tmp0 = x[*idx]; 3025f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 3026f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 3027f0d39aaaSBarry Smith } 3028f0d39aaaSBarry Smith x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3]; 3029f0d39aaaSBarry Smith x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2]; 3030f0d39aaaSBarry Smith break; 3031f0d39aaaSBarry Smith case 3: 3032f0d39aaaSBarry Smith 3033f0d39aaaSBarry Smith sum1 = xb[row]; 3034f0d39aaaSBarry Smith sum2 = xb[row-1]; 3035f0d39aaaSBarry Smith sum3 = xb[row-2]; 3036f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 3037f0d39aaaSBarry Smith v3 = a->a + diag[row-2] + 3; 3038f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 3039f0d39aaaSBarry Smith i1 = idx[0]; 3040f0d39aaaSBarry Smith i2 = idx[1]; 3041f0d39aaaSBarry Smith idx += 2; 3042f0d39aaaSBarry Smith tmp0 = x[i1]; 3043f0d39aaaSBarry Smith tmp1 = x[i2]; 3044f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3045f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3046f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3047f0d39aaaSBarry Smith } 3048f0d39aaaSBarry Smith 3049f0d39aaaSBarry Smith if (n == sz-1) { 3050f0d39aaaSBarry Smith tmp0 = x[*idx]; 3051f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 3052f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 3053f0d39aaaSBarry Smith sum3 -= *v3*tmp0; 3054f0d39aaaSBarry Smith } 3055f0d39aaaSBarry Smith x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8]; 3056f0d39aaaSBarry Smith x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7]; 3057f0d39aaaSBarry Smith x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6]; 3058f0d39aaaSBarry Smith break; 3059f0d39aaaSBarry Smith case 4: 3060f0d39aaaSBarry Smith 3061f0d39aaaSBarry Smith sum1 = xb[row]; 3062f0d39aaaSBarry Smith sum2 = xb[row-1]; 3063f0d39aaaSBarry Smith sum3 = xb[row-2]; 3064f0d39aaaSBarry Smith sum4 = xb[row-3]; 3065f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 3066f0d39aaaSBarry Smith v3 = a->a + diag[row-2] + 3; 3067f0d39aaaSBarry Smith v4 = a->a + diag[row-3] + 4; 3068f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 3069f0d39aaaSBarry Smith i1 = idx[0]; 3070f0d39aaaSBarry Smith i2 = idx[1]; 3071f0d39aaaSBarry Smith idx += 2; 3072f0d39aaaSBarry Smith tmp0 = x[i1]; 3073f0d39aaaSBarry Smith tmp1 = x[i2]; 3074f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3075f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3076f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3077f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 3078f0d39aaaSBarry Smith } 3079f0d39aaaSBarry Smith 3080f0d39aaaSBarry Smith if (n == sz-1) { 3081f0d39aaaSBarry Smith tmp0 = x[*idx]; 3082f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 3083f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 3084f0d39aaaSBarry Smith sum3 -= *v3*tmp0; 3085f0d39aaaSBarry Smith sum4 -= *v4*tmp0; 3086f0d39aaaSBarry Smith } 3087f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15]; 3088f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14]; 3089f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13]; 3090f0d39aaaSBarry Smith x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12]; 3091f0d39aaaSBarry Smith break; 3092f0d39aaaSBarry Smith case 5: 3093f0d39aaaSBarry Smith 3094f0d39aaaSBarry Smith sum1 = xb[row]; 3095f0d39aaaSBarry Smith sum2 = xb[row-1]; 3096f0d39aaaSBarry Smith sum3 = xb[row-2]; 3097f0d39aaaSBarry Smith sum4 = xb[row-3]; 3098f0d39aaaSBarry Smith sum5 = xb[row-4]; 3099f0d39aaaSBarry Smith v2 = a->a + diag[row-1] + 2; 3100f0d39aaaSBarry Smith v3 = a->a + diag[row-2] + 3; 3101f0d39aaaSBarry Smith v4 = a->a + diag[row-3] + 4; 3102f0d39aaaSBarry Smith v5 = a->a + diag[row-4] + 5; 3103f0d39aaaSBarry Smith for (n = 0; n<sz-1; n+=2) { 3104f0d39aaaSBarry Smith i1 = idx[0]; 3105f0d39aaaSBarry Smith i2 = idx[1]; 3106f0d39aaaSBarry Smith idx += 2; 3107f0d39aaaSBarry Smith tmp0 = x[i1]; 3108f0d39aaaSBarry Smith tmp1 = x[i2]; 3109f0d39aaaSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3110f0d39aaaSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3111f0d39aaaSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3112f0d39aaaSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 3113f0d39aaaSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 3114f0d39aaaSBarry Smith } 3115f0d39aaaSBarry Smith 3116f0d39aaaSBarry Smith if (n == sz-1) { 3117f0d39aaaSBarry Smith tmp0 = x[*idx]; 3118f0d39aaaSBarry Smith sum1 -= *v1*tmp0; 3119f0d39aaaSBarry Smith sum2 -= *v2*tmp0; 3120f0d39aaaSBarry Smith sum3 -= *v3*tmp0; 3121f0d39aaaSBarry Smith sum4 -= *v4*tmp0; 3122f0d39aaaSBarry Smith sum5 -= *v5*tmp0; 3123f0d39aaaSBarry Smith } 3124f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24]; 3125f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23]; 3126f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22]; 3127f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21]; 3128f0d39aaaSBarry Smith x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20]; 31298862d2efSBarry Smith break; 31308862d2efSBarry Smith default: 3131e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 31328862d2efSBarry Smith } 31332af78befSBarry Smith } 31342af78befSBarry Smith 31352af78befSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 31362af78befSBarry Smith } 31372af78befSBarry Smith its--; 31385850ef23SBarry Smith } 31395850ef23SBarry Smith while (its--) { 31405850ef23SBarry Smith 31415850ef23SBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) { 3142d876e2b0SMark Adams for (i=0, row=0, ibdiag = a->inode.ibdiag; 3143d876e2b0SMark Adams i<m; 3144d876e2b0SMark Adams row += sizes[i], ibdiag += sizes[i]*sizes[i], i++) { 31455850ef23SBarry Smith 3146d876e2b0SMark Adams sz = diag[row] - ii[row]; 31475850ef23SBarry Smith v1 = a->a + ii[row]; 31485850ef23SBarry Smith idx = a->j + ii[row]; 31495850ef23SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 31505850ef23SBarry Smith switch (sizes[i]) { 31515850ef23SBarry Smith case 1: 31525850ef23SBarry Smith sum1 = b[row]; 31535850ef23SBarry Smith for (n = 0; n<sz-1; n+=2) { 31545850ef23SBarry Smith i1 = idx[0]; 31555850ef23SBarry Smith i2 = idx[1]; 31565850ef23SBarry Smith idx += 2; 31575850ef23SBarry Smith tmp0 = x[i1]; 31585850ef23SBarry Smith tmp1 = x[i2]; 31595850ef23SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 31605850ef23SBarry Smith } 31615850ef23SBarry Smith if (n == sz-1) { 3162d876e2b0SMark Adams tmp0 = x[*idx++]; 3163d876e2b0SMark Adams sum1 -= *v1 * tmp0; 3164d876e2b0SMark Adams v1++; 3165d876e2b0SMark Adams } 3166d876e2b0SMark Adams t[row] = sum1; 3167d876e2b0SMark Adams sz = ii[row+1] - diag[row] - 1; 3168d876e2b0SMark Adams idx = a->j + diag[row] + 1; 3169d876e2b0SMark Adams v1 += 1; 3170d876e2b0SMark Adams for (n = 0; n<sz-1; n+=2) { 3171d876e2b0SMark Adams i1 = idx[0]; 3172d876e2b0SMark Adams i2 = idx[1]; 3173d876e2b0SMark Adams idx += 2; 3174d876e2b0SMark Adams tmp0 = x[i1]; 3175d876e2b0SMark Adams tmp1 = x[i2]; 3176d876e2b0SMark Adams sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3177d876e2b0SMark Adams } 3178d876e2b0SMark Adams if (n == sz-1) { 3179d876e2b0SMark Adams tmp0 = x[*idx++]; 31805850ef23SBarry Smith sum1 -= *v1 * tmp0; 31815850ef23SBarry Smith } 31825850ef23SBarry Smith /* in MatSOR_SeqAIJ this line would be 31835850ef23SBarry Smith * 31845850ef23SBarry Smith * x[row] = (1-omega)*x[row]+(sum1+(*bdiag++)*x[row])*(*ibdiag++); 31855850ef23SBarry Smith * 31865850ef23SBarry Smith * but omega == 1, so this becomes 31875850ef23SBarry Smith * 3188d876e2b0SMark Adams * x[row] = sum1*(*ibdiag++); 31895850ef23SBarry Smith * 31905850ef23SBarry Smith */ 3191d876e2b0SMark Adams x[row] = sum1*(*ibdiag); 31925850ef23SBarry Smith break; 31935850ef23SBarry Smith case 2: 31945850ef23SBarry Smith v2 = a->a + ii[row+1]; 31955850ef23SBarry Smith sum1 = b[row]; 31965850ef23SBarry Smith sum2 = b[row+1]; 31975850ef23SBarry Smith for (n = 0; n<sz-1; n+=2) { 31985850ef23SBarry Smith i1 = idx[0]; 31995850ef23SBarry Smith i2 = idx[1]; 32005850ef23SBarry Smith idx += 2; 32015850ef23SBarry Smith tmp0 = x[i1]; 32025850ef23SBarry Smith tmp1 = x[i2]; 32035850ef23SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 32045850ef23SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 32055850ef23SBarry Smith } 3206d876e2b0SMark Adams if (n == sz-1) { 3207d876e2b0SMark Adams tmp0 = x[*idx++]; 3208d876e2b0SMark Adams sum1 -= v1[0] * tmp0; 3209d876e2b0SMark Adams sum2 -= v2[0] * tmp0; 3210d876e2b0SMark Adams v1++; v2++; 3211d876e2b0SMark Adams } 3212d876e2b0SMark Adams t[row] = sum1; 3213d876e2b0SMark Adams t[row+1] = sum2; 3214d876e2b0SMark Adams sz = ii[row+1] - diag[row] - 2; 3215d876e2b0SMark Adams idx = a->j + diag[row] + 2; 3216d876e2b0SMark Adams v1 += 2; 3217d876e2b0SMark Adams v2 += 2; 3218d876e2b0SMark Adams for (n = 0; n<sz-1; n+=2) { 3219d876e2b0SMark Adams i1 = idx[0]; 3220d876e2b0SMark Adams i2 = idx[1]; 3221d876e2b0SMark Adams idx += 2; 3222d876e2b0SMark Adams tmp0 = x[i1]; 3223d876e2b0SMark Adams tmp1 = x[i2]; 3224d876e2b0SMark Adams sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3225d876e2b0SMark Adams sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3226d876e2b0SMark Adams } 32275850ef23SBarry Smith if (n == sz-1) { 32285850ef23SBarry Smith tmp0 = x[*idx]; 32295850ef23SBarry Smith sum1 -= v1[0] * tmp0; 32305850ef23SBarry Smith sum2 -= v2[0] * tmp0; 32315850ef23SBarry Smith } 3232d876e2b0SMark Adams x[row] = sum1*ibdiag[0] + sum2*ibdiag[2]; 3233d876e2b0SMark Adams x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3]; 32345850ef23SBarry Smith break; 32355850ef23SBarry Smith case 3: 32365850ef23SBarry Smith v2 = a->a + ii[row+1]; 32375850ef23SBarry Smith v3 = a->a + ii[row+2]; 32385850ef23SBarry Smith sum1 = b[row]; 32395850ef23SBarry Smith sum2 = b[row+1]; 32405850ef23SBarry Smith sum3 = b[row+2]; 32415850ef23SBarry Smith for (n = 0; n<sz-1; n+=2) { 32425850ef23SBarry Smith i1 = idx[0]; 32435850ef23SBarry Smith i2 = idx[1]; 32445850ef23SBarry Smith idx += 2; 32455850ef23SBarry Smith tmp0 = x[i1]; 32465850ef23SBarry Smith tmp1 = x[i2]; 32475850ef23SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 32485850ef23SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 32495850ef23SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 32505850ef23SBarry Smith } 3251d876e2b0SMark Adams if (n == sz-1) { 3252d876e2b0SMark Adams tmp0 = x[*idx++]; 3253d876e2b0SMark Adams sum1 -= v1[0] * tmp0; 3254d876e2b0SMark Adams sum2 -= v2[0] * tmp0; 3255d876e2b0SMark Adams sum3 -= v3[0] * tmp0; 3256d876e2b0SMark Adams v1++; v2++; v3++; 3257d876e2b0SMark Adams } 3258d876e2b0SMark Adams t[row] = sum1; 3259d876e2b0SMark Adams t[row+1] = sum2; 3260d876e2b0SMark Adams t[row+2] = sum3; 3261d876e2b0SMark Adams sz = ii[row+1] - diag[row] - 3; 3262d876e2b0SMark Adams idx = a->j + diag[row] + 3; 3263d876e2b0SMark Adams v1 += 3; 3264d876e2b0SMark Adams v2 += 3; 3265d876e2b0SMark Adams v3 += 3; 3266d876e2b0SMark Adams for (n = 0; n<sz-1; n+=2) { 3267d876e2b0SMark Adams i1 = idx[0]; 3268d876e2b0SMark Adams i2 = idx[1]; 3269d876e2b0SMark Adams idx += 2; 3270d876e2b0SMark Adams tmp0 = x[i1]; 3271d876e2b0SMark Adams tmp1 = x[i2]; 3272d876e2b0SMark Adams sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3273d876e2b0SMark Adams sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3274d876e2b0SMark Adams sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3275d876e2b0SMark Adams } 32765850ef23SBarry Smith if (n == sz-1) { 32775850ef23SBarry Smith tmp0 = x[*idx]; 32785850ef23SBarry Smith sum1 -= v1[0] * tmp0; 32795850ef23SBarry Smith sum2 -= v2[0] * tmp0; 32805850ef23SBarry Smith sum3 -= v3[0] * tmp0; 32815850ef23SBarry Smith } 3282d876e2b0SMark Adams x[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6]; 3283d876e2b0SMark Adams x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7]; 3284d876e2b0SMark Adams x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8]; 32855850ef23SBarry Smith break; 32865850ef23SBarry Smith case 4: 32875850ef23SBarry Smith v2 = a->a + ii[row+1]; 32885850ef23SBarry Smith v3 = a->a + ii[row+2]; 32895850ef23SBarry Smith v4 = a->a + ii[row+3]; 32905850ef23SBarry Smith sum1 = b[row]; 32915850ef23SBarry Smith sum2 = b[row+1]; 32925850ef23SBarry Smith sum3 = b[row+2]; 32935850ef23SBarry Smith sum4 = b[row+3]; 32945850ef23SBarry Smith for (n = 0; n<sz-1; n+=2) { 32955850ef23SBarry Smith i1 = idx[0]; 32965850ef23SBarry Smith i2 = idx[1]; 32975850ef23SBarry Smith idx += 2; 32985850ef23SBarry Smith tmp0 = x[i1]; 32995850ef23SBarry Smith tmp1 = x[i2]; 33005850ef23SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 33015850ef23SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 33025850ef23SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 33035850ef23SBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 33045850ef23SBarry Smith } 3305d876e2b0SMark Adams if (n == sz-1) { 3306d876e2b0SMark Adams tmp0 = x[*idx++]; 3307d876e2b0SMark Adams sum1 -= v1[0] * tmp0; 3308d876e2b0SMark Adams sum2 -= v2[0] * tmp0; 3309d876e2b0SMark Adams sum3 -= v3[0] * tmp0; 3310d876e2b0SMark Adams sum4 -= v4[0] * tmp0; 3311d876e2b0SMark Adams v1++; v2++; v3++; v4++; 3312d876e2b0SMark Adams } 3313d876e2b0SMark Adams t[row] = sum1; 3314d876e2b0SMark Adams t[row+1] = sum2; 3315d876e2b0SMark Adams t[row+2] = sum3; 3316d876e2b0SMark Adams t[row+3] = sum4; 3317d876e2b0SMark Adams sz = ii[row+1] - diag[row] - 4; 3318d876e2b0SMark Adams idx = a->j + diag[row] + 4; 3319d876e2b0SMark Adams v1 += 4; 3320d876e2b0SMark Adams v2 += 4; 3321d876e2b0SMark Adams v3 += 4; 3322d876e2b0SMark Adams v4 += 4; 3323d876e2b0SMark Adams for (n = 0; n<sz-1; n+=2) { 3324d876e2b0SMark Adams i1 = idx[0]; 3325d876e2b0SMark Adams i2 = idx[1]; 3326d876e2b0SMark Adams idx += 2; 3327d876e2b0SMark Adams tmp0 = x[i1]; 3328d876e2b0SMark Adams tmp1 = x[i2]; 3329d876e2b0SMark Adams sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3330d876e2b0SMark Adams sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3331d876e2b0SMark Adams sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3332d876e2b0SMark Adams sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 3333d876e2b0SMark Adams } 33345850ef23SBarry Smith if (n == sz-1) { 33355850ef23SBarry Smith tmp0 = x[*idx]; 33365850ef23SBarry Smith sum1 -= v1[0] * tmp0; 33375850ef23SBarry Smith sum2 -= v2[0] * tmp0; 33385850ef23SBarry Smith sum3 -= v3[0] * tmp0; 33395850ef23SBarry Smith sum4 -= v4[0] * tmp0; 33405850ef23SBarry Smith } 3341d876e2b0SMark Adams x[row] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12]; 3342d876e2b0SMark Adams x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13]; 3343d876e2b0SMark Adams x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14]; 3344d876e2b0SMark Adams x[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15]; 33455850ef23SBarry Smith break; 33465850ef23SBarry Smith case 5: 33475850ef23SBarry Smith v2 = a->a + ii[row+1]; 33485850ef23SBarry Smith v3 = a->a + ii[row+2]; 33495850ef23SBarry Smith v4 = a->a + ii[row+3]; 33505850ef23SBarry Smith v5 = a->a + ii[row+4]; 33515850ef23SBarry Smith sum1 = b[row]; 33525850ef23SBarry Smith sum2 = b[row+1]; 33535850ef23SBarry Smith sum3 = b[row+2]; 33545850ef23SBarry Smith sum4 = b[row+3]; 33555850ef23SBarry Smith sum5 = b[row+4]; 33565850ef23SBarry Smith for (n = 0; n<sz-1; n+=2) { 33575850ef23SBarry Smith i1 = idx[0]; 33585850ef23SBarry Smith i2 = idx[1]; 33595850ef23SBarry Smith idx += 2; 33605850ef23SBarry Smith tmp0 = x[i1]; 33615850ef23SBarry Smith tmp1 = x[i2]; 33625850ef23SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 33635850ef23SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 33645850ef23SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 33655850ef23SBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 33665850ef23SBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 33675850ef23SBarry Smith } 33685850ef23SBarry Smith if (n == sz-1) { 3369d876e2b0SMark Adams tmp0 = x[*idx++]; 33705850ef23SBarry Smith sum1 -= v1[0] * tmp0; 33715850ef23SBarry Smith sum2 -= v2[0] * tmp0; 33725850ef23SBarry Smith sum3 -= v3[0] * tmp0; 33735850ef23SBarry Smith sum4 -= v4[0] * tmp0; 33745850ef23SBarry Smith sum5 -= v5[0] * tmp0; 3375d876e2b0SMark Adams v1++; v2++; v3++; v4++; v5++; 33765850ef23SBarry Smith } 3377d876e2b0SMark Adams t[row] = sum1; 3378d876e2b0SMark Adams t[row+1] = sum2; 3379d876e2b0SMark Adams t[row+2] = sum3; 3380d876e2b0SMark Adams t[row+3] = sum4; 3381d876e2b0SMark Adams t[row+4] = sum5; 3382d876e2b0SMark Adams sz = ii[row+1] - diag[row] - 5; 3383d876e2b0SMark Adams idx = a->j + diag[row] + 5; 3384d876e2b0SMark Adams v1 += 5; 3385d876e2b0SMark Adams v2 += 5; 3386d876e2b0SMark Adams v3 += 5; 3387d876e2b0SMark Adams v4 += 5; 3388d876e2b0SMark Adams v5 += 5; 33895850ef23SBarry Smith for (n = 0; n<sz-1; n+=2) { 33905850ef23SBarry Smith i1 = idx[0]; 33915850ef23SBarry Smith i2 = idx[1]; 33925850ef23SBarry Smith idx += 2; 33935850ef23SBarry Smith tmp0 = x[i1]; 33945850ef23SBarry Smith tmp1 = x[i2]; 33955850ef23SBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 33965850ef23SBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 33975850ef23SBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 33985850ef23SBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 33995850ef23SBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 34005850ef23SBarry Smith } 34015850ef23SBarry Smith if (n == sz-1) { 34025850ef23SBarry Smith tmp0 = x[*idx]; 3403d876e2b0SMark Adams sum1 -= v1[0] * tmp0; 3404d876e2b0SMark Adams sum2 -= v2[0] * tmp0; 3405d876e2b0SMark Adams sum3 -= v3[0] * tmp0; 3406d876e2b0SMark Adams sum4 -= v4[0] * tmp0; 3407d876e2b0SMark Adams sum5 -= v5[0] * tmp0; 34085850ef23SBarry Smith } 3409d876e2b0SMark Adams x[row] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20]; 3410d876e2b0SMark Adams x[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21]; 3411d876e2b0SMark Adams x[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22]; 3412d876e2b0SMark Adams x[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23]; 3413d876e2b0SMark Adams x[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24]; 3414d876e2b0SMark Adams break; 3415d876e2b0SMark Adams default: 3416d876e2b0SMark Adams SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 3417d876e2b0SMark Adams } 3418d876e2b0SMark Adams } 3419d876e2b0SMark Adams xb = t; 3420d876e2b0SMark Adams ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); /* undercounts diag inverse */ 3421d876e2b0SMark Adams } else xb = b; 3422d876e2b0SMark Adams 3423d876e2b0SMark Adams if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) { 3424d876e2b0SMark Adams 3425d876e2b0SMark Adams ibdiag = a->inode.ibdiag+a->inode.bdiagsize; 3426d876e2b0SMark Adams for (i=m-1, row=A->rmap->n-1; i>=0; i--) { 3427d876e2b0SMark Adams ibdiag -= sizes[i]*sizes[i]; 3428d876e2b0SMark Adams 3429d876e2b0SMark Adams /* set RHS */ 3430d876e2b0SMark Adams if (xb == b) { 3431d876e2b0SMark Adams /* whole (old way) */ 3432d876e2b0SMark Adams sz = ii[row+1] - ii[row]; 3433d876e2b0SMark Adams idx = a->j + ii[row]; 3434d876e2b0SMark Adams switch (sizes[i]) { 3435d876e2b0SMark Adams case 5: 3436d876e2b0SMark Adams v5 = a->a + ii[row-4]; 3437d876e2b0SMark Adams case 4: /* fall through */ 3438d876e2b0SMark Adams v4 = a->a + ii[row-3]; 3439d876e2b0SMark Adams case 3: 3440d876e2b0SMark Adams v3 = a->a + ii[row-2]; 3441d876e2b0SMark Adams case 2: 3442d876e2b0SMark Adams v2 = a->a + ii[row-1]; 3443d876e2b0SMark Adams case 1: 3444d876e2b0SMark Adams v1 = a->a + ii[row]; 3445d876e2b0SMark Adams break; 3446d876e2b0SMark Adams default: 3447d876e2b0SMark Adams SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 3448d876e2b0SMark Adams } 3449d876e2b0SMark Adams } else { 3450d876e2b0SMark Adams /* upper, no diag */ 3451d876e2b0SMark Adams sz = ii[row+1] - diag[row] - 1; 3452d876e2b0SMark Adams idx = a->j + diag[row] + 1; 3453d876e2b0SMark Adams switch (sizes[i]) { 3454d876e2b0SMark Adams case 5: 3455d876e2b0SMark Adams v5 = a->a + diag[row-4] + 5; 3456d876e2b0SMark Adams case 4: /* fall through */ 3457d876e2b0SMark Adams v4 = a->a + diag[row-3] + 4; 3458d876e2b0SMark Adams case 3: 3459d876e2b0SMark Adams v3 = a->a + diag[row-2] + 3; 3460d876e2b0SMark Adams case 2: 3461d876e2b0SMark Adams v2 = a->a + diag[row-1] + 2; 3462d876e2b0SMark Adams case 1: 3463d876e2b0SMark Adams v1 = a->a + diag[row] + 1; 3464d876e2b0SMark Adams } 3465d876e2b0SMark Adams } 3466d876e2b0SMark Adams /* set sum */ 3467d876e2b0SMark Adams switch (sizes[i]) { 3468d876e2b0SMark Adams case 5: 3469d876e2b0SMark Adams sum5 = xb[row-4]; 3470d876e2b0SMark Adams case 4: /* fall through */ 3471d876e2b0SMark Adams sum4 = xb[row-3]; 3472d876e2b0SMark Adams case 3: 3473d876e2b0SMark Adams sum3 = xb[row-2]; 3474d876e2b0SMark Adams case 2: 3475d876e2b0SMark Adams sum2 = xb[row-1]; 3476d876e2b0SMark Adams case 1: 3477d876e2b0SMark Adams /* note that sum1 is associated with the last row */ 3478d876e2b0SMark Adams sum1 = xb[row]; 3479d876e2b0SMark Adams } 3480d876e2b0SMark Adams /* do sums */ 3481d876e2b0SMark Adams for (n = 0; n<sz-1; n+=2) { 3482d876e2b0SMark Adams i1 = idx[0]; 3483d876e2b0SMark Adams i2 = idx[1]; 3484d876e2b0SMark Adams idx += 2; 3485d876e2b0SMark Adams tmp0 = x[i1]; 3486d876e2b0SMark Adams tmp1 = x[i2]; 3487d876e2b0SMark Adams switch (sizes[i]) { 3488d876e2b0SMark Adams case 5: 3489d876e2b0SMark Adams sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 3490d876e2b0SMark Adams case 4: /* fall through */ 3491d876e2b0SMark Adams sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 3492d876e2b0SMark Adams case 3: 3493d876e2b0SMark Adams sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 3494d876e2b0SMark Adams case 2: 3495d876e2b0SMark Adams sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 3496d876e2b0SMark Adams case 1: 3497d876e2b0SMark Adams sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 3498d876e2b0SMark Adams } 3499d876e2b0SMark Adams } 3500d876e2b0SMark Adams /* ragged edge */ 3501d876e2b0SMark Adams if (n == sz-1) { 3502d876e2b0SMark Adams tmp0 = x[*idx]; 3503d876e2b0SMark Adams switch (sizes[i]) { 3504d876e2b0SMark Adams case 5: 3505d876e2b0SMark Adams sum5 -= *v5*tmp0; 3506d876e2b0SMark Adams case 4: /* fall through */ 3507d876e2b0SMark Adams sum4 -= *v4*tmp0; 3508d876e2b0SMark Adams case 3: 3509d876e2b0SMark Adams sum3 -= *v3*tmp0; 3510d876e2b0SMark Adams case 2: 3511d876e2b0SMark Adams sum2 -= *v2*tmp0; 3512d876e2b0SMark Adams case 1: 3513d876e2b0SMark Adams sum1 -= *v1*tmp0; 3514d876e2b0SMark Adams } 3515d876e2b0SMark Adams } 3516d876e2b0SMark Adams /* update */ 3517d876e2b0SMark Adams if (xb == b) { 3518d876e2b0SMark Adams /* whole (old way) w/ diag */ 3519d876e2b0SMark Adams switch (sizes[i]) { 3520d876e2b0SMark Adams case 5: 35215850ef23SBarry Smith x[row--] += sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24]; 35225850ef23SBarry Smith x[row--] += sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23]; 35235850ef23SBarry Smith x[row--] += sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22]; 35245850ef23SBarry Smith x[row--] += sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21]; 35255850ef23SBarry Smith x[row--] += sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20]; 35265850ef23SBarry Smith break; 3527d876e2b0SMark Adams case 4: 3528d876e2b0SMark Adams x[row--] += sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15]; 3529d876e2b0SMark Adams x[row--] += sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14]; 3530d876e2b0SMark Adams x[row--] += sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13]; 3531d876e2b0SMark Adams x[row--] += sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12]; 3532d876e2b0SMark Adams break; 3533d876e2b0SMark Adams case 3: 3534d876e2b0SMark Adams x[row--] += sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8]; 3535d876e2b0SMark Adams x[row--] += sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7]; 3536d876e2b0SMark Adams x[row--] += sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6]; 3537d876e2b0SMark Adams break; 3538d876e2b0SMark Adams case 2: 3539d876e2b0SMark Adams x[row--] += sum2*ibdiag[1] + sum1*ibdiag[3]; 3540d876e2b0SMark Adams x[row--] += sum2*ibdiag[0] + sum1*ibdiag[2]; 3541d876e2b0SMark Adams break; 3542d876e2b0SMark Adams case 1: 3543d876e2b0SMark Adams x[row--] += sum1*(*ibdiag); 3544d876e2b0SMark Adams break; 3545d876e2b0SMark Adams } 3546d876e2b0SMark Adams } else { 3547d876e2b0SMark Adams /* no diag so set = */ 3548d876e2b0SMark Adams switch (sizes[i]) { 3549d876e2b0SMark Adams case 5: 3550d876e2b0SMark Adams x[row--] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24]; 3551d876e2b0SMark Adams x[row--] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23]; 3552d876e2b0SMark Adams x[row--] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22]; 3553d876e2b0SMark Adams x[row--] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21]; 3554d876e2b0SMark Adams x[row--] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20]; 3555d876e2b0SMark Adams break; 3556d876e2b0SMark Adams case 4: 3557d876e2b0SMark Adams x[row--] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15]; 3558d876e2b0SMark Adams x[row--] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14]; 3559d876e2b0SMark Adams x[row--] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13]; 3560d876e2b0SMark Adams x[row--] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12]; 3561d876e2b0SMark Adams break; 3562d876e2b0SMark Adams case 3: 3563d876e2b0SMark Adams x[row--] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8]; 3564d876e2b0SMark Adams x[row--] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7]; 3565d876e2b0SMark Adams x[row--] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6]; 3566d876e2b0SMark Adams break; 3567d876e2b0SMark Adams case 2: 3568d876e2b0SMark Adams x[row--] = sum2*ibdiag[1] + sum1*ibdiag[3]; 3569d876e2b0SMark Adams x[row--] = sum2*ibdiag[0] + sum1*ibdiag[2]; 3570d876e2b0SMark Adams break; 3571d876e2b0SMark Adams case 1: 3572d876e2b0SMark Adams x[row--] = sum1*(*ibdiag); 3573d876e2b0SMark Adams break; 35745850ef23SBarry Smith } 35755850ef23SBarry Smith } 3576d876e2b0SMark Adams } 3577d876e2b0SMark Adams if (xb == b) { 35785850ef23SBarry Smith ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr); 3579d876e2b0SMark Adams } else { 3580d876e2b0SMark Adams ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); /* assumes 1/2 in upper, undercounts diag inverse */ 3581d876e2b0SMark Adams } 35825850ef23SBarry Smith } 35832af78befSBarry Smith } 358489c6957cSBarry Smith if (flag & SOR_EISENSTAT) { 358589c6957cSBarry Smith /* 358689c6957cSBarry Smith Apply (U + D)^-1 where D is now the block diagonal 358789c6957cSBarry Smith */ 358889c6957cSBarry Smith ibdiag = a->inode.ibdiag+a->inode.bdiagsize; 358989c6957cSBarry Smith for (i=m-1, row=A->rmap->n-1; i>=0; i--) { 359089c6957cSBarry Smith ibdiag -= sizes[i]*sizes[i]; 359189c6957cSBarry Smith sz = ii[row+1] - diag[row] - 1; 359289c6957cSBarry Smith v1 = a->a + diag[row] + 1; 359389c6957cSBarry Smith idx = a->j + diag[row] + 1; 35944108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 359589c6957cSBarry Smith switch (sizes[i]) { 359689c6957cSBarry Smith case 1: 359789c6957cSBarry Smith 359889c6957cSBarry Smith sum1 = b[row]; 359989c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 360089c6957cSBarry Smith i1 = idx[0]; 360189c6957cSBarry Smith i2 = idx[1]; 360289c6957cSBarry Smith idx += 2; 360389c6957cSBarry Smith tmp0 = x[i1]; 360489c6957cSBarry Smith tmp1 = x[i2]; 360589c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 360689c6957cSBarry Smith } 360789c6957cSBarry Smith 360889c6957cSBarry Smith if (n == sz-1) { 360989c6957cSBarry Smith tmp0 = x[*idx]; 361089c6957cSBarry Smith sum1 -= *v1*tmp0; 361189c6957cSBarry Smith } 3612938d4eb3SBarry Smith x[row] = sum1*(*ibdiag);row--; 361389c6957cSBarry Smith break; 361489c6957cSBarry Smith 361589c6957cSBarry Smith case 2: 361689c6957cSBarry Smith 361789c6957cSBarry Smith sum1 = b[row]; 361889c6957cSBarry Smith sum2 = b[row-1]; 361989c6957cSBarry Smith /* note that sum1 is associated with the second of the two rows */ 362089c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 362189c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 362289c6957cSBarry Smith i1 = idx[0]; 362389c6957cSBarry Smith i2 = idx[1]; 362489c6957cSBarry Smith idx += 2; 362589c6957cSBarry Smith tmp0 = x[i1]; 362689c6957cSBarry Smith tmp1 = x[i2]; 362789c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 362889c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 362989c6957cSBarry Smith } 363089c6957cSBarry Smith 363189c6957cSBarry Smith if (n == sz-1) { 363289c6957cSBarry Smith tmp0 = x[*idx]; 363389c6957cSBarry Smith sum1 -= *v1*tmp0; 363489c6957cSBarry Smith sum2 -= *v2*tmp0; 363589c6957cSBarry Smith } 3636938d4eb3SBarry Smith x[row] = sum2*ibdiag[1] + sum1*ibdiag[3]; 3637938d4eb3SBarry Smith x[row-1] = sum2*ibdiag[0] + sum1*ibdiag[2]; 3638938d4eb3SBarry Smith row -= 2; 363989c6957cSBarry Smith break; 364089c6957cSBarry Smith case 3: 364189c6957cSBarry Smith 364289c6957cSBarry Smith sum1 = b[row]; 364389c6957cSBarry Smith sum2 = b[row-1]; 364489c6957cSBarry Smith sum3 = b[row-2]; 364589c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 364689c6957cSBarry Smith v3 = a->a + diag[row-2] + 3; 364789c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 364889c6957cSBarry Smith i1 = idx[0]; 364989c6957cSBarry Smith i2 = idx[1]; 365089c6957cSBarry Smith idx += 2; 365189c6957cSBarry Smith tmp0 = x[i1]; 365289c6957cSBarry Smith tmp1 = x[i2]; 365389c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 365489c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 365589c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 365689c6957cSBarry Smith } 365789c6957cSBarry Smith 365889c6957cSBarry Smith if (n == sz-1) { 365989c6957cSBarry Smith tmp0 = x[*idx]; 366089c6957cSBarry Smith sum1 -= *v1*tmp0; 366189c6957cSBarry Smith sum2 -= *v2*tmp0; 366289c6957cSBarry Smith sum3 -= *v3*tmp0; 366389c6957cSBarry Smith } 3664938d4eb3SBarry Smith x[row] = sum3*ibdiag[2] + sum2*ibdiag[5] + sum1*ibdiag[8]; 3665938d4eb3SBarry Smith x[row-1] = sum3*ibdiag[1] + sum2*ibdiag[4] + sum1*ibdiag[7]; 3666938d4eb3SBarry Smith x[row-2] = sum3*ibdiag[0] + sum2*ibdiag[3] + sum1*ibdiag[6]; 3667938d4eb3SBarry Smith row -= 3; 366889c6957cSBarry Smith break; 366989c6957cSBarry Smith case 4: 367089c6957cSBarry Smith 367189c6957cSBarry Smith sum1 = b[row]; 367289c6957cSBarry Smith sum2 = b[row-1]; 367389c6957cSBarry Smith sum3 = b[row-2]; 367489c6957cSBarry Smith sum4 = b[row-3]; 367589c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 367689c6957cSBarry Smith v3 = a->a + diag[row-2] + 3; 367789c6957cSBarry Smith v4 = a->a + diag[row-3] + 4; 367889c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 367989c6957cSBarry Smith i1 = idx[0]; 368089c6957cSBarry Smith i2 = idx[1]; 368189c6957cSBarry Smith idx += 2; 368289c6957cSBarry Smith tmp0 = x[i1]; 368389c6957cSBarry Smith tmp1 = x[i2]; 368489c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 368589c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 368689c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 368789c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 368889c6957cSBarry Smith } 368989c6957cSBarry Smith 369089c6957cSBarry Smith if (n == sz-1) { 369189c6957cSBarry Smith tmp0 = x[*idx]; 369289c6957cSBarry Smith sum1 -= *v1*tmp0; 369389c6957cSBarry Smith sum2 -= *v2*tmp0; 369489c6957cSBarry Smith sum3 -= *v3*tmp0; 369589c6957cSBarry Smith sum4 -= *v4*tmp0; 369689c6957cSBarry Smith } 3697938d4eb3SBarry Smith x[row] = sum4*ibdiag[3] + sum3*ibdiag[7] + sum2*ibdiag[11] + sum1*ibdiag[15]; 3698938d4eb3SBarry Smith x[row-1] = sum4*ibdiag[2] + sum3*ibdiag[6] + sum2*ibdiag[10] + sum1*ibdiag[14]; 3699938d4eb3SBarry Smith x[row-2] = sum4*ibdiag[1] + sum3*ibdiag[5] + sum2*ibdiag[9] + sum1*ibdiag[13]; 3700938d4eb3SBarry Smith x[row-3] = sum4*ibdiag[0] + sum3*ibdiag[4] + sum2*ibdiag[8] + sum1*ibdiag[12]; 3701938d4eb3SBarry Smith row -= 4; 370289c6957cSBarry Smith break; 370389c6957cSBarry Smith case 5: 370489c6957cSBarry Smith 370589c6957cSBarry Smith sum1 = b[row]; 370689c6957cSBarry Smith sum2 = b[row-1]; 370789c6957cSBarry Smith sum3 = b[row-2]; 370889c6957cSBarry Smith sum4 = b[row-3]; 370989c6957cSBarry Smith sum5 = b[row-4]; 371089c6957cSBarry Smith v2 = a->a + diag[row-1] + 2; 371189c6957cSBarry Smith v3 = a->a + diag[row-2] + 3; 371289c6957cSBarry Smith v4 = a->a + diag[row-3] + 4; 371389c6957cSBarry Smith v5 = a->a + diag[row-4] + 5; 371489c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 371589c6957cSBarry Smith i1 = idx[0]; 371689c6957cSBarry Smith i2 = idx[1]; 371789c6957cSBarry Smith idx += 2; 371889c6957cSBarry Smith tmp0 = x[i1]; 371989c6957cSBarry Smith tmp1 = x[i2]; 372089c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 372189c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 372289c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 372389c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 372489c6957cSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 372589c6957cSBarry Smith } 372689c6957cSBarry Smith 372789c6957cSBarry Smith if (n == sz-1) { 372889c6957cSBarry Smith tmp0 = x[*idx]; 372989c6957cSBarry Smith sum1 -= *v1*tmp0; 373089c6957cSBarry Smith sum2 -= *v2*tmp0; 373189c6957cSBarry Smith sum3 -= *v3*tmp0; 373289c6957cSBarry Smith sum4 -= *v4*tmp0; 373389c6957cSBarry Smith sum5 -= *v5*tmp0; 373489c6957cSBarry Smith } 3735938d4eb3SBarry Smith x[row] = sum5*ibdiag[4] + sum4*ibdiag[9] + sum3*ibdiag[14] + sum2*ibdiag[19] + sum1*ibdiag[24]; 3736938d4eb3SBarry Smith x[row-1] = sum5*ibdiag[3] + sum4*ibdiag[8] + sum3*ibdiag[13] + sum2*ibdiag[18] + sum1*ibdiag[23]; 3737938d4eb3SBarry Smith x[row-2] = sum5*ibdiag[2] + sum4*ibdiag[7] + sum3*ibdiag[12] + sum2*ibdiag[17] + sum1*ibdiag[22]; 3738938d4eb3SBarry Smith x[row-3] = sum5*ibdiag[1] + sum4*ibdiag[6] + sum3*ibdiag[11] + sum2*ibdiag[16] + sum1*ibdiag[21]; 3739938d4eb3SBarry Smith x[row-4] = sum5*ibdiag[0] + sum4*ibdiag[5] + sum3*ibdiag[10] + sum2*ibdiag[15] + sum1*ibdiag[20]; 3740938d4eb3SBarry Smith row -= 5; 374189c6957cSBarry Smith break; 374289c6957cSBarry Smith default: 3743e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 374489c6957cSBarry Smith } 374589c6957cSBarry Smith } 374689c6957cSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 374789c6957cSBarry Smith 374889c6957cSBarry Smith /* 374989c6957cSBarry Smith t = b - D x where D is the block diagonal 375089c6957cSBarry Smith */ 375189c6957cSBarry Smith cnt = 0; 375289c6957cSBarry Smith for (i=0, row=0; i<m; i++) { 375389c6957cSBarry Smith switch (sizes[i]) { 375489c6957cSBarry Smith case 1: 375589c6957cSBarry Smith t[row] = b[row] - bdiag[cnt++]*x[row]; row++; 375689c6957cSBarry Smith break; 375789c6957cSBarry Smith case 2: 375889c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; 375989c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2]; 376089c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3]; 376189c6957cSBarry Smith t[row] = b[row] - tmp1; 376289c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; row += 2; 376389c6957cSBarry Smith cnt += 4; 376489c6957cSBarry Smith break; 376589c6957cSBarry Smith case 3: 376689c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; 376789c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6]; 376889c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7]; 376989c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8]; 377089c6957cSBarry Smith t[row] = b[row] - tmp1; 377189c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; 377289c6957cSBarry Smith t[row+2] = b[row+2] - tmp3; row += 3; 377389c6957cSBarry Smith cnt += 9; 377489c6957cSBarry Smith break; 377589c6957cSBarry Smith case 4: 377689c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; 377789c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12]; 377889c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13]; 377989c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14]; 378089c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15]; 378189c6957cSBarry Smith t[row] = b[row] - tmp1; 378289c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; 378389c6957cSBarry Smith t[row+2] = b[row+2] - tmp3; 378489c6957cSBarry Smith t[row+3] = b[row+3] - tmp4; row += 4; 378589c6957cSBarry Smith cnt += 16; 378689c6957cSBarry Smith break; 378789c6957cSBarry Smith case 5: 378889c6957cSBarry Smith x1 = x[row]; x2 = x[row+1]; x3 = x[row+2]; x4 = x[row+3]; x5 = x[row+4]; 378989c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20]; 379089c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21]; 379189c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22]; 379289c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23]; 379389c6957cSBarry Smith tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24]; 379489c6957cSBarry Smith t[row] = b[row] - tmp1; 379589c6957cSBarry Smith t[row+1] = b[row+1] - tmp2; 379689c6957cSBarry Smith t[row+2] = b[row+2] - tmp3; 379789c6957cSBarry Smith t[row+3] = b[row+3] - tmp4; 379889c6957cSBarry Smith t[row+4] = b[row+4] - tmp5;row += 5; 379989c6957cSBarry Smith cnt += 25; 380089c6957cSBarry Smith break; 380189c6957cSBarry Smith default: 3802e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 380389c6957cSBarry Smith } 380489c6957cSBarry Smith } 380589c6957cSBarry Smith ierr = PetscLogFlops(m);CHKERRQ(ierr); 380689c6957cSBarry Smith 380789c6957cSBarry Smith 380889c6957cSBarry Smith 380989c6957cSBarry Smith /* 381089c6957cSBarry Smith Apply (L + D)^-1 where D is the block diagonal 381189c6957cSBarry Smith */ 381289c6957cSBarry Smith for (i=0, row=0; i<m; i++) { 381389c6957cSBarry Smith sz = diag[row] - ii[row]; 381489c6957cSBarry Smith v1 = a->a + ii[row]; 381589c6957cSBarry Smith idx = a->j + ii[row]; 38164108e4d5SBarry Smith /* see comments for MatMult_SeqAIJ_Inode() for how this is coded */ 381789c6957cSBarry Smith switch (sizes[i]) { 381889c6957cSBarry Smith case 1: 381989c6957cSBarry Smith 382089c6957cSBarry Smith sum1 = t[row]; 382189c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 382289c6957cSBarry Smith i1 = idx[0]; 382389c6957cSBarry Smith i2 = idx[1]; 382489c6957cSBarry Smith idx += 2; 382589c6957cSBarry Smith tmp0 = t[i1]; 382689c6957cSBarry Smith tmp1 = t[i2]; 382789c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 382889c6957cSBarry Smith } 382989c6957cSBarry Smith 383089c6957cSBarry Smith if (n == sz-1) { 383189c6957cSBarry Smith tmp0 = t[*idx]; 383289c6957cSBarry Smith sum1 -= *v1 * tmp0; 383389c6957cSBarry Smith } 383489c6957cSBarry Smith x[row] += t[row] = sum1*(*ibdiag++); row++; 383589c6957cSBarry Smith break; 383689c6957cSBarry Smith case 2: 383789c6957cSBarry Smith v2 = a->a + ii[row+1]; 383889c6957cSBarry Smith sum1 = t[row]; 383989c6957cSBarry Smith sum2 = t[row+1]; 384089c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 384189c6957cSBarry Smith i1 = idx[0]; 384289c6957cSBarry Smith i2 = idx[1]; 384389c6957cSBarry Smith idx += 2; 384489c6957cSBarry Smith tmp0 = t[i1]; 384589c6957cSBarry Smith tmp1 = t[i2]; 384689c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 384789c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 384889c6957cSBarry Smith } 384989c6957cSBarry Smith 385089c6957cSBarry Smith if (n == sz-1) { 385189c6957cSBarry Smith tmp0 = t[*idx]; 385289c6957cSBarry Smith sum1 -= v1[0] * tmp0; 385389c6957cSBarry Smith sum2 -= v2[0] * tmp0; 385489c6957cSBarry Smith } 385589c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[2]; 385689c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[3]; 385789c6957cSBarry Smith ibdiag += 4; row += 2; 385889c6957cSBarry Smith break; 385989c6957cSBarry Smith case 3: 386089c6957cSBarry Smith v2 = a->a + ii[row+1]; 386189c6957cSBarry Smith v3 = a->a + ii[row+2]; 386289c6957cSBarry Smith sum1 = t[row]; 386389c6957cSBarry Smith sum2 = t[row+1]; 386489c6957cSBarry Smith sum3 = t[row+2]; 386589c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 386689c6957cSBarry Smith i1 = idx[0]; 386789c6957cSBarry Smith i2 = idx[1]; 386889c6957cSBarry Smith idx += 2; 386989c6957cSBarry Smith tmp0 = t[i1]; 387089c6957cSBarry Smith tmp1 = t[i2]; 387189c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 387289c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 387389c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 387489c6957cSBarry Smith } 387589c6957cSBarry Smith 387689c6957cSBarry Smith if (n == sz-1) { 387789c6957cSBarry Smith tmp0 = t[*idx]; 387889c6957cSBarry Smith sum1 -= v1[0] * tmp0; 387989c6957cSBarry Smith sum2 -= v2[0] * tmp0; 388089c6957cSBarry Smith sum3 -= v3[0] * tmp0; 388189c6957cSBarry Smith } 388289c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[3] + sum3*ibdiag[6]; 388389c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[4] + sum3*ibdiag[7]; 388489c6957cSBarry Smith x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[5] + sum3*ibdiag[8]; 388589c6957cSBarry Smith ibdiag += 9; row += 3; 388689c6957cSBarry Smith break; 388789c6957cSBarry Smith case 4: 388889c6957cSBarry Smith v2 = a->a + ii[row+1]; 388989c6957cSBarry Smith v3 = a->a + ii[row+2]; 389089c6957cSBarry Smith v4 = a->a + ii[row+3]; 389189c6957cSBarry Smith sum1 = t[row]; 389289c6957cSBarry Smith sum2 = t[row+1]; 389389c6957cSBarry Smith sum3 = t[row+2]; 389489c6957cSBarry Smith sum4 = t[row+3]; 389589c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 389689c6957cSBarry Smith i1 = idx[0]; 389789c6957cSBarry Smith i2 = idx[1]; 389889c6957cSBarry Smith idx += 2; 389989c6957cSBarry Smith tmp0 = t[i1]; 390089c6957cSBarry Smith tmp1 = t[i2]; 390189c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 390289c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 390389c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 390489c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 390589c6957cSBarry Smith } 390689c6957cSBarry Smith 390789c6957cSBarry Smith if (n == sz-1) { 390889c6957cSBarry Smith tmp0 = t[*idx]; 390989c6957cSBarry Smith sum1 -= v1[0] * tmp0; 391089c6957cSBarry Smith sum2 -= v2[0] * tmp0; 391189c6957cSBarry Smith sum3 -= v3[0] * tmp0; 391289c6957cSBarry Smith sum4 -= v4[0] * tmp0; 391389c6957cSBarry Smith } 391489c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[4] + sum3*ibdiag[8] + sum4*ibdiag[12]; 391589c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[5] + sum3*ibdiag[9] + sum4*ibdiag[13]; 391689c6957cSBarry Smith x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[6] + sum3*ibdiag[10] + sum4*ibdiag[14]; 391789c6957cSBarry Smith x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[7] + sum3*ibdiag[11] + sum4*ibdiag[15]; 391889c6957cSBarry Smith ibdiag += 16; row += 4; 391989c6957cSBarry Smith break; 392089c6957cSBarry Smith case 5: 392189c6957cSBarry Smith v2 = a->a + ii[row+1]; 392289c6957cSBarry Smith v3 = a->a + ii[row+2]; 392389c6957cSBarry Smith v4 = a->a + ii[row+3]; 392489c6957cSBarry Smith v5 = a->a + ii[row+4]; 392589c6957cSBarry Smith sum1 = t[row]; 392689c6957cSBarry Smith sum2 = t[row+1]; 392789c6957cSBarry Smith sum3 = t[row+2]; 392889c6957cSBarry Smith sum4 = t[row+3]; 392989c6957cSBarry Smith sum5 = t[row+4]; 393089c6957cSBarry Smith for (n = 0; n<sz-1; n+=2) { 393189c6957cSBarry Smith i1 = idx[0]; 393289c6957cSBarry Smith i2 = idx[1]; 393389c6957cSBarry Smith idx += 2; 393489c6957cSBarry Smith tmp0 = t[i1]; 393589c6957cSBarry Smith tmp1 = t[i2]; 393689c6957cSBarry Smith sum1 -= v1[0] * tmp0 + v1[1] * tmp1; v1 += 2; 393789c6957cSBarry Smith sum2 -= v2[0] * tmp0 + v2[1] * tmp1; v2 += 2; 393889c6957cSBarry Smith sum3 -= v3[0] * tmp0 + v3[1] * tmp1; v3 += 2; 393989c6957cSBarry Smith sum4 -= v4[0] * tmp0 + v4[1] * tmp1; v4 += 2; 394089c6957cSBarry Smith sum5 -= v5[0] * tmp0 + v5[1] * tmp1; v5 += 2; 394189c6957cSBarry Smith } 394289c6957cSBarry Smith 394389c6957cSBarry Smith if (n == sz-1) { 394489c6957cSBarry Smith tmp0 = t[*idx]; 394589c6957cSBarry Smith sum1 -= v1[0] * tmp0; 394689c6957cSBarry Smith sum2 -= v2[0] * tmp0; 394789c6957cSBarry Smith sum3 -= v3[0] * tmp0; 394889c6957cSBarry Smith sum4 -= v4[0] * tmp0; 394989c6957cSBarry Smith sum5 -= v5[0] * tmp0; 395089c6957cSBarry Smith } 395189c6957cSBarry Smith x[row] += t[row] = sum1*ibdiag[0] + sum2*ibdiag[5] + sum3*ibdiag[10] + sum4*ibdiag[15] + sum5*ibdiag[20]; 395289c6957cSBarry Smith x[row+1] += t[row+1] = sum1*ibdiag[1] + sum2*ibdiag[6] + sum3*ibdiag[11] + sum4*ibdiag[16] + sum5*ibdiag[21]; 395389c6957cSBarry Smith x[row+2] += t[row+2] = sum1*ibdiag[2] + sum2*ibdiag[7] + sum3*ibdiag[12] + sum4*ibdiag[17] + sum5*ibdiag[22]; 395489c6957cSBarry Smith x[row+3] += t[row+3] = sum1*ibdiag[3] + sum2*ibdiag[8] + sum3*ibdiag[13] + sum4*ibdiag[18] + sum5*ibdiag[23]; 395589c6957cSBarry Smith x[row+4] += t[row+4] = sum1*ibdiag[4] + sum2*ibdiag[9] + sum3*ibdiag[14] + sum4*ibdiag[19] + sum5*ibdiag[24]; 395689c6957cSBarry Smith ibdiag += 25; row += 5; 395789c6957cSBarry Smith break; 395889c6957cSBarry Smith default: 3959e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 396089c6957cSBarry Smith } 396189c6957cSBarry Smith } 396289c6957cSBarry Smith ierr = PetscLogFlops(a->nz);CHKERRQ(ierr); 39635850ef23SBarry Smith } 39642af78befSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 39653649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 3966365a8a9eSBarry Smith PetscFunctionReturn(0); 39672af78befSBarry Smith } 39682af78befSBarry Smith 39694108e4d5SBarry Smith PetscErrorCode MatMultDiagonalBlock_SeqAIJ_Inode(Mat A,Vec bb,Vec xx) 397089c6957cSBarry Smith { 397189c6957cSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 397289c6957cSBarry Smith PetscScalar *x,tmp1,tmp2,tmp3,tmp4,tmp5,x1,x2,x3,x4,x5; 397389c6957cSBarry Smith const MatScalar *bdiag = a->inode.bdiag; 397489c6957cSBarry Smith const PetscScalar *b; 397589c6957cSBarry Smith PetscErrorCode ierr; 397689c6957cSBarry Smith PetscInt m = a->inode.node_count,cnt = 0,i,row; 397789c6957cSBarry Smith const PetscInt *sizes = a->inode.size; 39782af78befSBarry Smith 397989c6957cSBarry Smith PetscFunctionBegin; 3980b215bc84SStefano Zampini if (!a->inode.size) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_COR,"Missing Inode Structure"); 398189c6957cSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 39823649974fSBarry Smith ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr); 398389c6957cSBarry Smith cnt = 0; 398489c6957cSBarry Smith for (i=0, row=0; i<m; i++) { 398589c6957cSBarry Smith switch (sizes[i]) { 398689c6957cSBarry Smith case 1: 398789c6957cSBarry Smith x[row] = b[row]*bdiag[cnt++];row++; 398889c6957cSBarry Smith break; 398989c6957cSBarry Smith case 2: 399089c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; 399189c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+2]; 399289c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+3]; 399389c6957cSBarry Smith x[row++] = tmp1; 399489c6957cSBarry Smith x[row++] = tmp2; 399589c6957cSBarry Smith cnt += 4; 399689c6957cSBarry Smith break; 399789c6957cSBarry Smith case 3: 399889c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; x3 = b[row+2]; 399989c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+3] + x3*bdiag[cnt+6]; 400089c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+4] + x3*bdiag[cnt+7]; 400189c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+5] + x3*bdiag[cnt+8]; 400289c6957cSBarry Smith x[row++] = tmp1; 400389c6957cSBarry Smith x[row++] = tmp2; 400489c6957cSBarry Smith x[row++] = tmp3; 400589c6957cSBarry Smith cnt += 9; 400689c6957cSBarry Smith break; 400789c6957cSBarry Smith case 4: 400889c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3]; 400989c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+4] + x3*bdiag[cnt+8] + x4*bdiag[cnt+12]; 401089c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+5] + x3*bdiag[cnt+9] + x4*bdiag[cnt+13]; 401189c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+6] + x3*bdiag[cnt+10] + x4*bdiag[cnt+14]; 401289c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+7] + x3*bdiag[cnt+11] + x4*bdiag[cnt+15]; 401389c6957cSBarry Smith x[row++] = tmp1; 401489c6957cSBarry Smith x[row++] = tmp2; 401589c6957cSBarry Smith x[row++] = tmp3; 401689c6957cSBarry Smith x[row++] = tmp4; 401789c6957cSBarry Smith cnt += 16; 401889c6957cSBarry Smith break; 401989c6957cSBarry Smith case 5: 402089c6957cSBarry Smith x1 = b[row]; x2 = b[row+1]; x3 = b[row+2]; x4 = b[row+3]; x5 = b[row+4]; 402189c6957cSBarry Smith tmp1 = x1*bdiag[cnt] + x2*bdiag[cnt+5] + x3*bdiag[cnt+10] + x4*bdiag[cnt+15] + x5*bdiag[cnt+20]; 402289c6957cSBarry Smith tmp2 = x1*bdiag[cnt+1] + x2*bdiag[cnt+6] + x3*bdiag[cnt+11] + x4*bdiag[cnt+16] + x5*bdiag[cnt+21]; 402389c6957cSBarry Smith tmp3 = x1*bdiag[cnt+2] + x2*bdiag[cnt+7] + x3*bdiag[cnt+12] + x4*bdiag[cnt+17] + x5*bdiag[cnt+22]; 402489c6957cSBarry Smith tmp4 = x1*bdiag[cnt+3] + x2*bdiag[cnt+8] + x3*bdiag[cnt+13] + x4*bdiag[cnt+18] + x5*bdiag[cnt+23]; 402589c6957cSBarry Smith tmp5 = x1*bdiag[cnt+4] + x2*bdiag[cnt+9] + x3*bdiag[cnt+14] + x4*bdiag[cnt+19] + x5*bdiag[cnt+24]; 402689c6957cSBarry Smith x[row++] = tmp1; 402789c6957cSBarry Smith x[row++] = tmp2; 402889c6957cSBarry Smith x[row++] = tmp3; 402989c6957cSBarry Smith x[row++] = tmp4; 403089c6957cSBarry Smith x[row++] = tmp5; 403189c6957cSBarry Smith cnt += 25; 403289c6957cSBarry Smith break; 403389c6957cSBarry Smith default: 4034e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Inode size %D not supported",sizes[i]); 403589c6957cSBarry Smith } 403689c6957cSBarry Smith } 4037ca0c957dSBarry Smith ierr = PetscLogFlops(2.0*cnt);CHKERRQ(ierr); 403889c6957cSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 40393649974fSBarry Smith ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr); 404089c6957cSBarry Smith PetscFunctionReturn(0); 404189c6957cSBarry Smith } 404289c6957cSBarry Smith 4043b215bc84SStefano Zampini static PetscErrorCode MatSeqAIJ_Inode_ResetOps(Mat A) 4044b215bc84SStefano Zampini { 4045b215bc84SStefano Zampini Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 4046b215bc84SStefano Zampini 4047b215bc84SStefano Zampini PetscFunctionBegin; 4048b215bc84SStefano Zampini a->inode.node_count = 0; 4049b215bc84SStefano Zampini a->inode.use = PETSC_FALSE; 4050b215bc84SStefano Zampini a->inode.checked = PETSC_FALSE; 4051b215bc84SStefano Zampini a->inode.mat_nonzerostate = -1; 4052b215bc84SStefano Zampini A->ops->getrowij = MatGetRowIJ_SeqAIJ; 4053b215bc84SStefano Zampini A->ops->restorerowij = MatRestoreRowIJ_SeqAIJ; 4054b215bc84SStefano Zampini A->ops->getcolumnij = MatGetColumnIJ_SeqAIJ; 4055b215bc84SStefano Zampini A->ops->restorecolumnij = MatRestoreColumnIJ_SeqAIJ; 4056b215bc84SStefano Zampini A->ops->coloringpatch = NULL; 4057b215bc84SStefano Zampini A->ops->multdiagonalblock = NULL; 4058b215bc84SStefano Zampini if (A->factortype) { 4059b215bc84SStefano Zampini A->ops->solve = MatSolve_SeqAIJ_inplace; 4060b215bc84SStefano Zampini } 4061b215bc84SStefano Zampini PetscFunctionReturn(0); 4062b215bc84SStefano Zampini } 4063b215bc84SStefano Zampini 40644c1414c8SBarry Smith /* 40654c1414c8SBarry Smith samestructure indicates that the matrix has not changed its nonzero structure so we 40664c1414c8SBarry Smith do not need to recompute the inodes 40674c1414c8SBarry Smith */ 4068a02bda8eSBarry Smith PetscErrorCode MatSeqAIJCheckInode(Mat A) 40694c1414c8SBarry Smith { 40704c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 40714c1414c8SBarry Smith PetscErrorCode ierr; 40728758e1faSBarry Smith PetscInt i,j,m,nzx,nzy,*ns,node_count,blk_size; 4073ace3abfcSBarry Smith PetscBool flag; 40748758e1faSBarry Smith const PetscInt *idx,*idy,*ii; 40754c1414c8SBarry Smith 40764c1414c8SBarry Smith PetscFunctionBegin; 4077b215bc84SStefano Zampini if (!a->inode.use) { 4078b215bc84SStefano Zampini ierr = MatSeqAIJ_Inode_ResetOps(A);CHKERRQ(ierr); 4079b215bc84SStefano Zampini ierr = PetscFree(a->inode.size);CHKERRQ(ierr); 4080b215bc84SStefano Zampini PetscFunctionReturn(0); 4081b215bc84SStefano Zampini } 4082a02bda8eSBarry Smith if (a->inode.checked && A->nonzerostate == a->inode.mat_nonzerostate) PetscFunctionReturn(0); 40834c1414c8SBarry Smith 4084d0f46423SBarry Smith m = A->rmap->n; 4085b215bc84SStefano Zampini if (!a->inode.size) { ierr = PetscMalloc1(m+1,&a->inode.size);CHKERRQ(ierr); } 4086b215bc84SStefano Zampini ns = a->inode.size; 40874c1414c8SBarry Smith 40884c1414c8SBarry Smith i = 0; 40894c1414c8SBarry Smith node_count = 0; 40904c1414c8SBarry Smith idx = a->j; 40914c1414c8SBarry Smith ii = a->i; 40924c1414c8SBarry Smith while (i < m) { /* For each row */ 40934c1414c8SBarry Smith nzx = ii[i+1] - ii[i]; /* Number of nonzeros */ 40944c1414c8SBarry Smith /* Limits the number of elements in a node to 'a->inode.limit' */ 40954c1414c8SBarry Smith for (j=i+1,idy=idx,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) { 40964c1414c8SBarry Smith nzy = ii[j+1] - ii[j]; /* Same number of nonzeros */ 40974c1414c8SBarry Smith if (nzy != nzx) break; 40984c1414c8SBarry Smith idy += nzx; /* Same nonzero pattern */ 4099580bdb30SBarry Smith ierr = PetscArraycmp(idx,idy,nzx,&flag);CHKERRQ(ierr); 41004c1414c8SBarry Smith if (!flag) break; 41014c1414c8SBarry Smith } 41024c1414c8SBarry Smith ns[node_count++] = blk_size; 41034c1414c8SBarry Smith idx += blk_size*nzx; 41044c1414c8SBarry Smith i = j; 41054c1414c8SBarry Smith } 41062cb58ee3SKarl Rupp 41074c1414c8SBarry Smith /* If not enough inodes found,, do not use inode version of the routines */ 4108be6adb11SBarry Smith if (!m || node_count > .8*m) { 4109b215bc84SStefano Zampini ierr = MatSeqAIJ_Inode_ResetOps(A);CHKERRQ(ierr); 4110b215bc84SStefano Zampini ierr = PetscFree(a->inode.size);CHKERRQ(ierr); 41114c1414c8SBarry Smith ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr); 41124c1414c8SBarry Smith } else { 4113d5f3da31SBarry Smith if (!A->factortype) { 4114375a6242SBarry Smith A->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode; 4115375a6242SBarry Smith if (A->rmap->n == A->cmap->n) { 41164108e4d5SBarry Smith A->ops->getrowij = MatGetRowIJ_SeqAIJ_Inode; 41174108e4d5SBarry Smith A->ops->restorerowij = MatRestoreRowIJ_SeqAIJ_Inode; 41184108e4d5SBarry Smith A->ops->getcolumnij = MatGetColumnIJ_SeqAIJ_Inode; 41194108e4d5SBarry Smith A->ops->restorecolumnij = MatRestoreColumnIJ_SeqAIJ_Inode; 41204108e4d5SBarry Smith A->ops->coloringpatch = MatColoringPatch_SeqAIJ_Inode; 4121375a6242SBarry Smith } 4122d3ac4fa3SBarry Smith } else { 4123d3ac4fa3SBarry Smith A->ops->solve = MatSolve_SeqAIJ_Inode_inplace; 4124d3ac4fa3SBarry Smith } 41254c1414c8SBarry Smith a->inode.node_count = node_count; 41264c1414c8SBarry Smith ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr); 41274c1414c8SBarry Smith } 4128be6adb11SBarry Smith a->inode.checked = PETSC_TRUE; 4129a02bda8eSBarry Smith a->inode.mat_nonzerostate = A->nonzerostate; 41304c1414c8SBarry Smith PetscFunctionReturn(0); 41314c1414c8SBarry Smith } 41324c1414c8SBarry Smith 4133150f0143SBarry Smith PetscErrorCode MatDuplicate_SeqAIJ_Inode(Mat A,MatDuplicateOption cpvalues,Mat *C) 4134150f0143SBarry Smith { 4135150f0143SBarry Smith Mat B =*C; 4136150f0143SBarry Smith Mat_SeqAIJ *c=(Mat_SeqAIJ*)B->data,*a=(Mat_SeqAIJ*)A->data; 4137150f0143SBarry Smith PetscErrorCode ierr; 4138150f0143SBarry Smith PetscInt m=A->rmap->n; 4139150f0143SBarry Smith 4140150f0143SBarry Smith PetscFunctionBegin; 4141150f0143SBarry Smith c->inode.use = a->inode.use; 4142150f0143SBarry Smith c->inode.limit = a->inode.limit; 4143150f0143SBarry Smith c->inode.max_limit = a->inode.max_limit; 4144*ec710b6aSStefano Zampini c->inode.checked = PETSC_FALSE; 4145*ec710b6aSStefano Zampini c->inode.size = NULL; 4146*ec710b6aSStefano Zampini c->inode.node_count = 0; 4147*ec710b6aSStefano Zampini c->inode.ibdiagvalid = PETSC_FALSE; 4148*ec710b6aSStefano Zampini c->inode.ibdiag = NULL; 4149*ec710b6aSStefano Zampini c->inode.bdiag = NULL; 4150*ec710b6aSStefano Zampini c->inode.mat_nonzerostate = -1; 4151b215bc84SStefano Zampini if (a->inode.use) { 4152*ec710b6aSStefano Zampini if (a->inode.checked && a->inode.size) { 4153b215bc84SStefano Zampini ierr = PetscMalloc1(m+1,&c->inode.size);CHKERRQ(ierr); 4154580bdb30SBarry Smith ierr = PetscArraycpy(c->inode.size,a->inode.size,m+1);CHKERRQ(ierr); 4155*ec710b6aSStefano Zampini 4156*ec710b6aSStefano Zampini c->inode.checked = PETSC_TRUE; 4157*ec710b6aSStefano Zampini c->inode.node_count = a->inode.node_count; 4158*ec710b6aSStefano Zampini c->inode.mat_nonzerostate = (*C)->nonzerostate; 4159*ec710b6aSStefano Zampini } 4160a02bda8eSBarry Smith /* note the table of functions below should match that in MatSeqAIJCheckInode() */ 41612c451681SBarry Smith if (!B->factortype) { 41622c451681SBarry Smith B->ops->getrowij = MatGetRowIJ_SeqAIJ_Inode; 41632c451681SBarry Smith B->ops->restorerowij = MatRestoreRowIJ_SeqAIJ_Inode; 41642c451681SBarry Smith B->ops->getcolumnij = MatGetColumnIJ_SeqAIJ_Inode; 41652c451681SBarry Smith B->ops->restorecolumnij = MatRestoreColumnIJ_SeqAIJ_Inode; 41662c451681SBarry Smith B->ops->coloringpatch = MatColoringPatch_SeqAIJ_Inode; 41672c451681SBarry Smith B->ops->multdiagonalblock = MatMultDiagonalBlock_SeqAIJ_Inode; 4168150f0143SBarry Smith } else { 41692c451681SBarry Smith B->ops->solve = MatSolve_SeqAIJ_Inode_inplace; 4170150f0143SBarry Smith } 4171150f0143SBarry Smith } 4172150f0143SBarry Smith PetscFunctionReturn(0); 4173150f0143SBarry Smith } 4174150f0143SBarry Smith 41758758e1faSBarry Smith PETSC_STATIC_INLINE PetscErrorCode MatGetRow_FactoredLU(PetscInt *cols,PetscInt nzl,PetscInt nzu,PetscInt nz,const PetscInt *ai,const PetscInt *aj,const PetscInt *adiag,PetscInt row) 417617454e89SShri Abhyankar { 41778758e1faSBarry Smith PetscInt k; 41788758e1faSBarry Smith const PetscInt *vi; 41796e111a19SKarl Rupp 418017454e89SShri Abhyankar PetscFunctionBegin; 418117454e89SShri Abhyankar vi = aj + ai[row]; 418217454e89SShri Abhyankar for (k=0; k<nzl; k++) cols[k] = vi[k]; 418317454e89SShri Abhyankar vi = aj + adiag[row]; 418417454e89SShri Abhyankar cols[nzl] = vi[0]; 418517454e89SShri Abhyankar vi = aj + adiag[row+1]+1; 418617454e89SShri Abhyankar for (k=0; k<nzu; k++) cols[nzl+1+k] = vi[k]; 418717454e89SShri Abhyankar PetscFunctionReturn(0); 418817454e89SShri Abhyankar } 41896936b636SHong Zhang /* 4190a02bda8eSBarry Smith MatSeqAIJCheckInode_FactorLU - Check Inode for factored seqaij matrix. 4191a02bda8eSBarry Smith Modified from MatSeqAIJCheckInode(). 41926936b636SHong Zhang 41936936b636SHong Zhang Input Parameters: 4194abb87a52SBarry Smith . Mat A - ILU or LU matrix factor 4195abb87a52SBarry Smith 41966936b636SHong Zhang */ 4197a02bda8eSBarry Smith PetscErrorCode MatSeqAIJCheckInode_FactorLU(Mat A) 4198019b515eSShri Abhyankar { 4199019b515eSShri Abhyankar Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 4200019b515eSShri Abhyankar PetscErrorCode ierr; 4201019b515eSShri Abhyankar PetscInt i,j,m,nzl1,nzu1,nzl2,nzu2,nzx,nzy,node_count,blk_size; 42028758e1faSBarry Smith PetscInt *cols1,*cols2,*ns; 42038758e1faSBarry Smith const PetscInt *ai = a->i,*aj = a->j, *adiag = a->diag; 4204ace3abfcSBarry Smith PetscBool flag; 4205019b515eSShri Abhyankar 4206019b515eSShri Abhyankar PetscFunctionBegin; 4207019b515eSShri Abhyankar if (!a->inode.use) PetscFunctionReturn(0); 4208abb87a52SBarry Smith if (a->inode.checked) PetscFunctionReturn(0); 4209019b515eSShri Abhyankar 4210019b515eSShri Abhyankar m = A->rmap->n; 42112205254eSKarl Rupp if (a->inode.size) ns = a->inode.size; 42122205254eSKarl Rupp else { 4213854ce69bSBarry Smith ierr = PetscMalloc1(m+1,&ns);CHKERRQ(ierr); 42142205254eSKarl Rupp } 4215019b515eSShri Abhyankar 4216019b515eSShri Abhyankar i = 0; 4217019b515eSShri Abhyankar node_count = 0; 4218dcca6d9dSJed Brown ierr = PetscMalloc2(m,&cols1,m,&cols2);CHKERRQ(ierr); 4219019b515eSShri Abhyankar while (i < m) { /* For each row */ 4220019b515eSShri Abhyankar nzl1 = ai[i+1] - ai[i]; /* Number of nonzeros in L */ 4221019b515eSShri Abhyankar nzu1 = adiag[i] - adiag[i+1] - 1; /* Number of nonzeros in U excluding diagonal*/ 4222019b515eSShri Abhyankar nzx = nzl1 + nzu1 + 1; 4223019b515eSShri Abhyankar MatGetRow_FactoredLU(cols1,nzl1,nzu1,nzx,ai,aj,adiag,i); 4224019b515eSShri Abhyankar 4225019b515eSShri Abhyankar /* Limits the number of elements in a node to 'a->inode.limit' */ 4226019b515eSShri Abhyankar for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) { 4227019b515eSShri Abhyankar nzl2 = ai[j+1] - ai[j]; 4228019b515eSShri Abhyankar nzu2 = adiag[j] - adiag[j+1] - 1; 4229019b515eSShri Abhyankar nzy = nzl2 + nzu2 + 1; 4230019b515eSShri Abhyankar if (nzy != nzx) break; 423117454e89SShri Abhyankar ierr = MatGetRow_FactoredLU(cols2,nzl2,nzu2,nzy,ai,aj,adiag,j);CHKERRQ(ierr); 4232580bdb30SBarry Smith ierr = PetscArraycmp(cols1,cols2,nzx,&flag);CHKERRQ(ierr); 42338758e1faSBarry Smith if (!flag) break; 4234019b515eSShri Abhyankar } 4235019b515eSShri Abhyankar ns[node_count++] = blk_size; 4236019b515eSShri Abhyankar i = j; 4237019b515eSShri Abhyankar } 42388758e1faSBarry Smith ierr = PetscFree2(cols1,cols2);CHKERRQ(ierr); 4239019b515eSShri Abhyankar /* If not enough inodes found,, do not use inode version of the routines */ 4240be6adb11SBarry Smith if (!m || node_count > .8*m) { 4241019b515eSShri Abhyankar ierr = PetscFree(ns);CHKERRQ(ierr); 42422205254eSKarl Rupp 4243019b515eSShri Abhyankar a->inode.node_count = 0; 42440298fd71SBarry Smith a->inode.size = NULL; 4245019b515eSShri Abhyankar a->inode.use = PETSC_FALSE; 42462205254eSKarl Rupp 4247019b515eSShri Abhyankar ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr); 4248019b515eSShri Abhyankar } else { 4249f4259b30SLisandro Dalcin A->ops->mult = NULL; 4250f4259b30SLisandro Dalcin A->ops->sor = NULL; 4251f4259b30SLisandro Dalcin A->ops->multadd = NULL; 4252f4259b30SLisandro Dalcin A->ops->getrowij = NULL; 4253f4259b30SLisandro Dalcin A->ops->restorerowij = NULL; 4254f4259b30SLisandro Dalcin A->ops->getcolumnij = NULL; 4255f4259b30SLisandro Dalcin A->ops->restorecolumnij = NULL; 4256f4259b30SLisandro Dalcin A->ops->coloringpatch = NULL; 4257f4259b30SLisandro Dalcin A->ops->multdiagonalblock = NULL; 4258019b515eSShri Abhyankar a->inode.node_count = node_count; 4259019b515eSShri Abhyankar a->inode.size = ns; 42602205254eSKarl Rupp 4261019b515eSShri Abhyankar ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr); 4262019b515eSShri Abhyankar } 4263be6adb11SBarry Smith a->inode.checked = PETSC_TRUE; 4264019b515eSShri Abhyankar PetscFunctionReturn(0); 4265019b515eSShri Abhyankar } 4266019b515eSShri Abhyankar 4267acf2f550SJed Brown PetscErrorCode MatSeqAIJInvalidateDiagonal_Inode(Mat A) 4268acf2f550SJed Brown { 4269acf2f550SJed Brown Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 4270acf2f550SJed Brown 4271acf2f550SJed Brown PetscFunctionBegin; 4272acf2f550SJed Brown a->inode.ibdiagvalid = PETSC_FALSE; 4273acf2f550SJed Brown PetscFunctionReturn(0); 4274acf2f550SJed Brown } 4275acf2f550SJed Brown 42764c1414c8SBarry Smith /* 42774c1414c8SBarry Smith This is really ugly. if inodes are used this replaces the 42784c1414c8SBarry Smith permutations with ones that correspond to rows/cols of the matrix 42794c1414c8SBarry Smith rather then inode blocks 42804c1414c8SBarry Smith */ 42817087cfbeSBarry Smith PetscErrorCode MatInodeAdjustForInodes(Mat A,IS *rperm,IS *cperm) 42824c1414c8SBarry Smith { 42834ac538c5SBarry Smith PetscErrorCode ierr; 42844c1414c8SBarry Smith 42854c1414c8SBarry Smith PetscFunctionBegin; 42864ac538c5SBarry Smith ierr = PetscTryMethod(A,"MatInodeAdjustForInodes_C",(Mat,IS*,IS*),(A,rperm,cperm));CHKERRQ(ierr); 42874c1414c8SBarry Smith PetscFunctionReturn(0); 42884c1414c8SBarry Smith } 42894c1414c8SBarry Smith 42907087cfbeSBarry Smith PetscErrorCode MatInodeAdjustForInodes_SeqAIJ_Inode(Mat A,IS *rperm,IS *cperm) 42914c1414c8SBarry Smith { 42924c1414c8SBarry Smith Mat_SeqAIJ *a=(Mat_SeqAIJ*)A->data; 42934c1414c8SBarry Smith PetscErrorCode ierr; 42945d0c19d7SBarry Smith PetscInt m = A->rmap->n,n = A->cmap->n,i,j,nslim_row = a->inode.node_count; 42955d0c19d7SBarry Smith const PetscInt *ridx,*cidx; 42964c1414c8SBarry Smith PetscInt row,col,*permr,*permc,*ns_row = a->inode.size,*tns,start_val,end_val,indx; 42974c1414c8SBarry Smith PetscInt nslim_col,*ns_col; 42984c1414c8SBarry Smith IS ris = *rperm,cis = *cperm; 42994c1414c8SBarry Smith 43004c1414c8SBarry Smith PetscFunctionBegin; 43014c1414c8SBarry Smith if (!a->inode.size) PetscFunctionReturn(0); /* no inodes so return */ 43024c1414c8SBarry Smith if (a->inode.node_count == m) PetscFunctionReturn(0); /* all inodes are of size 1 */ 43034c1414c8SBarry Smith 4304470880abSPatrick Sanan ierr = MatCreateColInode_Private(A,&nslim_col,&ns_col);CHKERRQ(ierr); 4305854ce69bSBarry Smith ierr = PetscMalloc1(((nslim_row>nslim_col) ? nslim_row : nslim_col)+1,&tns);CHKERRQ(ierr); 4306dcca6d9dSJed Brown ierr = PetscMalloc2(m,&permr,n,&permc);CHKERRQ(ierr); 43074c1414c8SBarry Smith 43084c1414c8SBarry Smith ierr = ISGetIndices(ris,&ridx);CHKERRQ(ierr); 43094c1414c8SBarry Smith ierr = ISGetIndices(cis,&cidx);CHKERRQ(ierr); 43104c1414c8SBarry Smith 43114c1414c8SBarry Smith /* Form the inode structure for the rows of permuted matric using inv perm*/ 43124c1414c8SBarry Smith for (i=0,tns[0]=0; i<nslim_row; ++i) tns[i+1] = tns[i] + ns_row[i]; 43134c1414c8SBarry Smith 43144c1414c8SBarry Smith /* Construct the permutations for rows*/ 43154c1414c8SBarry Smith for (i=0,row = 0; i<nslim_row; ++i) { 43164c1414c8SBarry Smith indx = ridx[i]; 43174c1414c8SBarry Smith start_val = tns[indx]; 43184c1414c8SBarry Smith end_val = tns[indx + 1]; 43194c1414c8SBarry Smith for (j=start_val; j<end_val; ++j,++row) permr[row]= j; 43204c1414c8SBarry Smith } 43214c1414c8SBarry Smith 43224c1414c8SBarry Smith /* Form the inode structure for the columns of permuted matrix using inv perm*/ 43234c1414c8SBarry Smith for (i=0,tns[0]=0; i<nslim_col; ++i) tns[i+1] = tns[i] + ns_col[i]; 43244c1414c8SBarry Smith 43254c1414c8SBarry Smith /* Construct permutations for columns */ 43264c1414c8SBarry Smith for (i=0,col=0; i<nslim_col; ++i) { 43274c1414c8SBarry Smith indx = cidx[i]; 43284c1414c8SBarry Smith start_val = tns[indx]; 43294c1414c8SBarry Smith end_val = tns[indx + 1]; 43304c1414c8SBarry Smith for (j = start_val; j<end_val; ++j,++col) permc[col]= j; 43314c1414c8SBarry Smith } 43324c1414c8SBarry Smith 433370b3c8c7SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permr,PETSC_COPY_VALUES,rperm);CHKERRQ(ierr); 43340e83c824SBarry Smith ierr = ISSetPermutation(*rperm);CHKERRQ(ierr); 433570b3c8c7SBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF,n,permc,PETSC_COPY_VALUES,cperm);CHKERRQ(ierr); 43360e83c824SBarry Smith ierr = ISSetPermutation(*cperm);CHKERRQ(ierr); 43374c1414c8SBarry Smith 43384c1414c8SBarry Smith ierr = ISRestoreIndices(ris,&ridx);CHKERRQ(ierr); 43394c1414c8SBarry Smith ierr = ISRestoreIndices(cis,&cidx);CHKERRQ(ierr); 43404c1414c8SBarry Smith 43414c1414c8SBarry Smith ierr = PetscFree(ns_col);CHKERRQ(ierr); 43420e83c824SBarry Smith ierr = PetscFree2(permr,permc);CHKERRQ(ierr); 43436bf464f9SBarry Smith ierr = ISDestroy(&cis);CHKERRQ(ierr); 43446bf464f9SBarry Smith ierr = ISDestroy(&ris);CHKERRQ(ierr); 43454c1414c8SBarry Smith ierr = PetscFree(tns);CHKERRQ(ierr); 43464c1414c8SBarry Smith PetscFunctionReturn(0); 43474c1414c8SBarry Smith } 43484c1414c8SBarry Smith 43494c1414c8SBarry Smith /*@C 43504c1414c8SBarry Smith MatInodeGetInodeSizes - Returns the inode information of the Inode matrix. 43514c1414c8SBarry Smith 43523f9fe445SBarry Smith Not Collective 43534c1414c8SBarry Smith 43544c1414c8SBarry Smith Input Parameter: 43554c1414c8SBarry Smith . A - the Inode matrix or matrix derived from the Inode class -- e.g., SeqAIJ 43564c1414c8SBarry Smith 43574c1414c8SBarry Smith Output Parameter: 43584c1414c8SBarry Smith + node_count - no of inodes present in the matrix. 43594c1414c8SBarry Smith . sizes - an array of size node_count,with sizes of each inode. 43604c1414c8SBarry Smith - limit - the max size used to generate the inodes. 43614c1414c8SBarry Smith 43624c1414c8SBarry Smith Level: advanced 43634c1414c8SBarry Smith 436495452b02SPatrick Sanan Notes: 436595452b02SPatrick Sanan This routine returns some internal storage information 43664c1414c8SBarry Smith of the matrix, it is intended to be used by advanced users. 43674c1414c8SBarry Smith It should be called after the matrix is assembled. 43684c1414c8SBarry Smith The contents of the sizes[] array should not be changed. 43690298fd71SBarry Smith NULL may be passed for information not requested. 43704c1414c8SBarry Smith 43714c1414c8SBarry Smith .seealso: MatGetInfo() 43724c1414c8SBarry Smith @*/ 43737087cfbeSBarry Smith PetscErrorCode MatInodeGetInodeSizes(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit) 43744c1414c8SBarry Smith { 43754c1414c8SBarry Smith PetscErrorCode ierr,(*f)(Mat,PetscInt*,PetscInt*[],PetscInt*); 43764c1414c8SBarry Smith 43774c1414c8SBarry Smith PetscFunctionBegin; 4378e32f2f54SBarry Smith if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for unassembled matrix"); 43790005d66cSJed Brown ierr = PetscObjectQueryFunction((PetscObject)A,"MatInodeGetInodeSizes_C",&f);CHKERRQ(ierr); 43804c1414c8SBarry Smith if (f) { 43814c1414c8SBarry Smith ierr = (*f)(A,node_count,sizes,limit);CHKERRQ(ierr); 43824c1414c8SBarry Smith } 43834c1414c8SBarry Smith PetscFunctionReturn(0); 43844c1414c8SBarry Smith } 43854c1414c8SBarry Smith 43867087cfbeSBarry Smith PetscErrorCode MatInodeGetInodeSizes_SeqAIJ_Inode(Mat A,PetscInt *node_count,PetscInt *sizes[],PetscInt *limit) 43874c1414c8SBarry Smith { 43884c1414c8SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data; 43894c1414c8SBarry Smith 43904c1414c8SBarry Smith PetscFunctionBegin; 43914c1414c8SBarry Smith if (node_count) *node_count = a->inode.node_count; 43924c1414c8SBarry Smith if (sizes) *sizes = a->inode.size; 43934c1414c8SBarry Smith if (limit) *limit = a->inode.limit; 43944c1414c8SBarry Smith PetscFunctionReturn(0); 43954c1414c8SBarry Smith } 4396