1d4002b98SHong Zhang #include <../src/mat/impls/aij/mpi/mpiaij.h> /*I "petscmat.h" I*/ 2d4002b98SHong Zhang #include <../src/mat/impls/sell/mpi/mpisell.h> /*I "petscmat.h" I*/ 3d4002b98SHong Zhang #include <petsc/private/vecimpl.h> 4d4002b98SHong Zhang #include <petsc/private/isimpl.h> 5d4002b98SHong Zhang #include <petscblaslapack.h> 6d4002b98SHong Zhang #include <petscsf.h> 7d4002b98SHong Zhang 8d4002b98SHong Zhang /*MC 9d4002b98SHong Zhang MATSELL - MATSELL = "sell" - A matrix type to be used for sparse matrices. 10d4002b98SHong Zhang 11d4002b98SHong Zhang This matrix type is identical to MATSEQSELL when constructed with a single process communicator, 12d4002b98SHong Zhang and MATMPISELL otherwise. As a result, for single process communicators, 13d4002b98SHong Zhang MatSeqSELLSetPreallocation is supported, and similarly MatMPISELLSetPreallocation is supported 14d4002b98SHong Zhang for communicators controlling multiple processes. It is recommended that you call both of 15d4002b98SHong Zhang the above preallocation routines for simplicity. 16d4002b98SHong Zhang 17d4002b98SHong Zhang Options Database Keys: 18d4002b98SHong Zhang . -mat_type sell - sets the matrix type to "sell" during a call to MatSetFromOptions() 19d4002b98SHong Zhang 20d4002b98SHong Zhang Developer Notes: Subclasses include MATSELLCUSP, MATSELLCUSPARSE, MATSELLPERM, MATSELLCRL, and also automatically switches over to use inodes when 21d4002b98SHong Zhang enough exist. 22d4002b98SHong Zhang 23d4002b98SHong Zhang Level: beginner 24d4002b98SHong Zhang 25d4002b98SHong Zhang .seealso: MatCreateSELL(), MatCreateSeqSELL(), MATSEQSELL, MATMPISELL 26d4002b98SHong Zhang M*/ 27d4002b98SHong Zhang 28d4002b98SHong Zhang PetscErrorCode MatDiagonalSet_MPISELL(Mat Y,Vec D,InsertMode is) 29d4002b98SHong Zhang { 30d4002b98SHong Zhang PetscErrorCode ierr; 31d4002b98SHong Zhang Mat_MPISELL *sell=(Mat_MPISELL*)Y->data; 32d4002b98SHong Zhang 33d4002b98SHong Zhang PetscFunctionBegin; 34d4002b98SHong Zhang if (Y->assembled && Y->rmap->rstart == Y->cmap->rstart && Y->rmap->rend == Y->cmap->rend) { 35d4002b98SHong Zhang ierr = MatDiagonalSet(sell->A,D,is);CHKERRQ(ierr); 36d4002b98SHong Zhang } else { 37d4002b98SHong Zhang ierr = MatDiagonalSet_Default(Y,D,is);CHKERRQ(ierr); 38d4002b98SHong Zhang } 39d4002b98SHong Zhang PetscFunctionReturn(0); 40d4002b98SHong Zhang } 41d4002b98SHong Zhang 42d4002b98SHong Zhang /* 43d4002b98SHong Zhang Local utility routine that creates a mapping from the global column 44d4002b98SHong Zhang number to the local number in the off-diagonal part of the local 45d4002b98SHong Zhang storage of the matrix. When PETSC_USE_CTABLE is used this is scalable at 46d4002b98SHong Zhang a slightly higher hash table cost; without it it is not scalable (each processor 47d4002b98SHong Zhang has an order N integer array but is fast to acess. 48d4002b98SHong Zhang */ 49d4002b98SHong Zhang PetscErrorCode MatCreateColmap_MPISELL_Private(Mat mat) 50d4002b98SHong Zhang { 51d4002b98SHong Zhang Mat_MPISELL *sell=(Mat_MPISELL*)mat->data; 52d4002b98SHong Zhang PetscErrorCode ierr; 53d4002b98SHong Zhang PetscInt n=sell->B->cmap->n,i; 54d4002b98SHong Zhang 55d4002b98SHong Zhang PetscFunctionBegin; 56d4002b98SHong Zhang if (!sell->garray) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"MPISELL Matrix was assembled but is missing garray"); 57d4002b98SHong Zhang #if defined(PETSC_USE_CTABLE) 58d4002b98SHong Zhang ierr = PetscTableCreate(n,mat->cmap->N+1,&sell->colmap);CHKERRQ(ierr); 59d4002b98SHong Zhang for (i=0; i<n; i++) { 60d4002b98SHong Zhang ierr = PetscTableAdd(sell->colmap,sell->garray[i]+1,i+1,INSERT_VALUES);CHKERRQ(ierr); 61d4002b98SHong Zhang } 62d4002b98SHong Zhang #else 63d4002b98SHong Zhang ierr = PetscCalloc1(mat->cmap->N+1,&sell->colmap);CHKERRQ(ierr); 64d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)mat,(mat->cmap->N+1)*sizeof(PetscInt));CHKERRQ(ierr); 65d4002b98SHong Zhang for (i=0; i<n; i++) sell->colmap[sell->garray[i]] = i+1; 66d4002b98SHong Zhang #endif 67d4002b98SHong Zhang PetscFunctionReturn(0); 68d4002b98SHong Zhang } 69d4002b98SHong Zhang 70d4002b98SHong Zhang #define MatSetValues_SeqSELL_A_Private(row,col,value,addv,orow,ocol) \ 71d4002b98SHong Zhang { \ 72d4002b98SHong Zhang if (col <= lastcol1) low1 = 0; \ 73d4002b98SHong Zhang else high1 = nrow1; \ 74d4002b98SHong Zhang lastcol1 = col; \ 75d4002b98SHong Zhang while (high1-low1 > 5) { \ 76d4002b98SHong Zhang t = (low1+high1)/2; \ 77d4002b98SHong Zhang if (*(cp1+8*t) > col) high1 = t; \ 78d4002b98SHong Zhang else low1 = t; \ 79d4002b98SHong Zhang } \ 80d4002b98SHong Zhang for (_i=low1; _i<high1; _i++) { \ 81d4002b98SHong Zhang if (*(cp1+8*_i) > col) break; \ 82d4002b98SHong Zhang if (*(cp1+8*_i) == col) { \ 83d4002b98SHong Zhang if (addv == ADD_VALUES) *(vp1+8*_i) += value; \ 84d4002b98SHong Zhang else *(vp1+8*_i) = value; \ 85d4002b98SHong Zhang goto a_noinsert; \ 86d4002b98SHong Zhang } \ 87d4002b98SHong Zhang } \ 88d4002b98SHong Zhang if (value == 0.0 && ignorezeroentries) {low1 = 0; high1 = nrow1;goto a_noinsert;} \ 89d4002b98SHong Zhang if (nonew == 1) {low1 = 0; high1 = nrow1; goto a_noinsert;} \ 90d4002b98SHong Zhang if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero at global row/column (%D, %D) into matrix", orow, ocol); \ 91d4002b98SHong Zhang MatSeqXSELLReallocateSELL(A,am,1,nrow1,a->sliidx,row/8,row,col,a->colidx,a->val,cp1,vp1,nonew,MatScalar); \ 92d4002b98SHong Zhang /* shift up all the later entries in this row */ \ 93d4002b98SHong Zhang for (ii=nrow1-1; ii>=_i; ii--) { \ 94d4002b98SHong Zhang *(cp1+8*(ii+1)) = *(cp1+8*ii); \ 95d4002b98SHong Zhang *(vp1+8*(ii+1)) = *(vp1+8*ii); \ 96d4002b98SHong Zhang } \ 97d4002b98SHong Zhang *(cp1+8*_i) = col; \ 98d4002b98SHong Zhang *(vp1+8*_i) = value; \ 99d4002b98SHong Zhang a->nz++; nrow1++; A->nonzerostate++; \ 100d4002b98SHong Zhang a_noinsert: ; \ 101d4002b98SHong Zhang a->rlen[row] = nrow1; \ 102d4002b98SHong Zhang } 103d4002b98SHong Zhang 104d4002b98SHong Zhang #define MatSetValues_SeqSELL_B_Private(row,col,value,addv,orow,ocol) \ 105d4002b98SHong Zhang { \ 106d4002b98SHong Zhang if (col <= lastcol2) low2 = 0; \ 107d4002b98SHong Zhang else high2 = nrow2; \ 108d4002b98SHong Zhang lastcol2 = col; \ 109d4002b98SHong Zhang while (high2-low2 > 5) { \ 110d4002b98SHong Zhang t = (low2+high2)/2; \ 111d4002b98SHong Zhang if (*(cp2+8*t) > col) high2 = t; \ 112d4002b98SHong Zhang else low2 = t; \ 113d4002b98SHong Zhang } \ 114d4002b98SHong Zhang for (_i=low2; _i<high2; _i++) { \ 115d4002b98SHong Zhang if (*(cp2+8*_i) > col) break; \ 116d4002b98SHong Zhang if (*(cp2+8*_i) == col) { \ 117d4002b98SHong Zhang if (addv == ADD_VALUES) *(vp2+8*_i) += value; \ 118d4002b98SHong Zhang else *(vp2+8*_i) = value; \ 119d4002b98SHong Zhang goto b_noinsert; \ 120d4002b98SHong Zhang } \ 121d4002b98SHong Zhang } \ 122d4002b98SHong Zhang if (value == 0.0 && ignorezeroentries) {low2 = 0; high2 = nrow2; goto b_noinsert;} \ 123d4002b98SHong Zhang if (nonew == 1) {low2 = 0; high2 = nrow2; goto b_noinsert;} \ 124d4002b98SHong Zhang if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero at global row/column (%D, %D) into matrix", orow, ocol); \ 125d4002b98SHong Zhang MatSeqXSELLReallocateSELL(B,bm,1,nrow2,b->sliidx,row/8,row,col,b->colidx,b->val,cp2,vp2,nonew,MatScalar); \ 126d4002b98SHong Zhang /* shift up all the later entries in this row */ \ 127d4002b98SHong Zhang for (ii=nrow2-1; ii>=_i; ii--) { \ 128d4002b98SHong Zhang *(cp2+8*(ii+1)) = *(cp2+8*ii); \ 129d4002b98SHong Zhang *(vp2+8*(ii+1)) = *(vp2+8*ii); \ 130d4002b98SHong Zhang } \ 131d4002b98SHong Zhang *(cp2+8*_i) = col; \ 132d4002b98SHong Zhang *(vp2+8*_i) = value; \ 133d4002b98SHong Zhang b->nz++; nrow2++; B->nonzerostate++; \ 134d4002b98SHong Zhang b_noinsert: ; \ 135d4002b98SHong Zhang b->rlen[row] = nrow2; \ 136d4002b98SHong Zhang } 137d4002b98SHong Zhang 138d4002b98SHong Zhang PetscErrorCode MatSetValues_MPISELL(Mat mat,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode addv) 139d4002b98SHong Zhang { 140d4002b98SHong Zhang Mat_MPISELL *sell=(Mat_MPISELL*)mat->data; 141d4002b98SHong Zhang PetscScalar value; 142d4002b98SHong Zhang PetscErrorCode ierr; 143d4002b98SHong Zhang PetscInt i,j,rstart=mat->rmap->rstart,rend=mat->rmap->rend,shift1,shift2; 144d4002b98SHong Zhang PetscInt cstart=mat->cmap->rstart,cend=mat->cmap->rend,row,col; 145d4002b98SHong Zhang PetscBool roworiented=sell->roworiented; 146d4002b98SHong Zhang 147d4002b98SHong Zhang /* Some Variables required in the macro */ 148d4002b98SHong Zhang Mat A=sell->A; 149d4002b98SHong Zhang Mat_SeqSELL *a=(Mat_SeqSELL*)A->data; 150d4002b98SHong Zhang PetscBool ignorezeroentries=a->ignorezeroentries,found; 151d4002b98SHong Zhang Mat B=sell->B; 152d4002b98SHong Zhang Mat_SeqSELL *b=(Mat_SeqSELL*)B->data; 153d4002b98SHong Zhang PetscInt *cp1,*cp2,ii,_i,nrow1,nrow2,low1,high1,low2,high2,t,lastcol1,lastcol2; 154d4002b98SHong Zhang MatScalar *vp1,*vp2; 155d4002b98SHong Zhang 156d4002b98SHong Zhang PetscFunctionBegin; 157d4002b98SHong Zhang for (i=0; i<m; i++) { 158d4002b98SHong Zhang if (im[i] < 0) continue; 159d4002b98SHong Zhang #if defined(PETSC_USE_DEBUG) 160d4002b98SHong Zhang if (im[i] >= mat->rmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",im[i],mat->rmap->N-1); 161d4002b98SHong Zhang #endif 162d4002b98SHong Zhang if (im[i] >= rstart && im[i] < rend) { 163d4002b98SHong Zhang row = im[i] - rstart; 164d4002b98SHong Zhang lastcol1 = -1; 165d4002b98SHong Zhang shift1 = a->sliidx[row>>3]+(row&0x07); /* starting index of the row */ 166d4002b98SHong Zhang cp1 = a->colidx+shift1; 167d4002b98SHong Zhang vp1 = a->val+shift1; 168d4002b98SHong Zhang nrow1 = a->rlen[row]; 169d4002b98SHong Zhang low1 = 0; 170d4002b98SHong Zhang high1 = nrow1; 171d4002b98SHong Zhang lastcol2 = -1; 172d4002b98SHong Zhang shift2 = b->sliidx[row>>3]+(row&0x07); /* starting index of the row */ 173d4002b98SHong Zhang cp2 = b->colidx+shift2; 174d4002b98SHong Zhang vp2 = b->val+shift2; 175d4002b98SHong Zhang nrow2 = b->rlen[row]; 176d4002b98SHong Zhang low2 = 0; 177d4002b98SHong Zhang high2 = nrow2; 178d4002b98SHong Zhang 179d4002b98SHong Zhang for (j=0; j<n; j++) { 180d4002b98SHong Zhang if (roworiented) value = v[i*n+j]; 181d4002b98SHong Zhang else value = v[i+j*m]; 182d4002b98SHong Zhang if (ignorezeroentries && value == 0.0 && (addv == ADD_VALUES)) continue; 183d4002b98SHong Zhang if (in[j] >= cstart && in[j] < cend) { 184d4002b98SHong Zhang col = in[j] - cstart; 185d4002b98SHong Zhang MatSetValue_SeqSELL_Private(A,row,col,value,addv,im[i],in[j],cp1,vp1,lastcol1,low1,high1); /* set one value */ 186d4002b98SHong Zhang } else if (in[j] < 0) continue; 187d4002b98SHong Zhang #if defined(PETSC_USE_DEBUG) 188d4002b98SHong Zhang else if (in[j] >= mat->cmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[j],mat->cmap->N-1); 189d4002b98SHong Zhang #endif 190d4002b98SHong Zhang else { 191d4002b98SHong Zhang if (mat->was_assembled) { 192d4002b98SHong Zhang if (!sell->colmap) { 193d4002b98SHong Zhang ierr = MatCreateColmap_MPISELL_Private(mat);CHKERRQ(ierr); 194d4002b98SHong Zhang } 195d4002b98SHong Zhang #if defined(PETSC_USE_CTABLE) 196d4002b98SHong Zhang ierr = PetscTableFind(sell->colmap,in[j]+1,&col);CHKERRQ(ierr); 197d4002b98SHong Zhang col--; 198d4002b98SHong Zhang #else 199d4002b98SHong Zhang col = sell->colmap[in[j]] - 1; 200d4002b98SHong Zhang #endif 201d4002b98SHong Zhang if (col < 0 && !((Mat_SeqSELL*)(sell->B->data))->nonew) { 202d4002b98SHong Zhang ierr = MatDisAssemble_MPISELL(mat);CHKERRQ(ierr); 203d4002b98SHong Zhang col = in[j]; 204d4002b98SHong Zhang /* Reinitialize the variables required by MatSetValues_SeqSELL_B_Private() */ 205d4002b98SHong Zhang B = sell->B; 206d4002b98SHong Zhang b = (Mat_SeqSELL*)B->data; 207d4002b98SHong Zhang shift2 = b->sliidx[row>>3]+(row&0x07); /* starting index of the row */ 208d4002b98SHong Zhang cp2 = b->colidx+shift2; 209d4002b98SHong Zhang vp2 = b->val+shift2; 210d4002b98SHong Zhang nrow2 = b->rlen[row]; 211d4002b98SHong Zhang low2 = 0; 212d4002b98SHong Zhang high2 = nrow2; 213d4002b98SHong Zhang } else if (col < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero at global row/column (%D, %D) into matrix", im[i], in[j]); 214d4002b98SHong Zhang } else col = in[j]; 215d4002b98SHong Zhang MatSetValue_SeqSELL_Private(B,row,col,value,addv,im[i],in[j],cp2,vp2,lastcol2,low2,high2); /* set one value */ 216d4002b98SHong Zhang } 217d4002b98SHong Zhang } 218d4002b98SHong Zhang } else { 219d4002b98SHong Zhang if (mat->nooffprocentries) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Setting off process row %D even though MatSetOption(,MAT_NO_OFF_PROC_ENTRIES,PETSC_TRUE) was set",im[i]); 220d4002b98SHong Zhang if (!sell->donotstash) { 221d4002b98SHong Zhang mat->assembled = PETSC_FALSE; 222d4002b98SHong Zhang if (roworiented) { 223d4002b98SHong Zhang ierr = MatStashValuesRow_Private(&mat->stash,im[i],n,in,v+i*n,(PetscBool)(ignorezeroentries && (addv == ADD_VALUES)));CHKERRQ(ierr); 224d4002b98SHong Zhang } else { 225d4002b98SHong Zhang ierr = MatStashValuesCol_Private(&mat->stash,im[i],n,in,v+i,m,(PetscBool)(ignorezeroentries && (addv == ADD_VALUES)));CHKERRQ(ierr); 226d4002b98SHong Zhang } 227d4002b98SHong Zhang } 228d4002b98SHong Zhang } 229d4002b98SHong Zhang } 230d4002b98SHong Zhang PetscFunctionReturn(0); 231d4002b98SHong Zhang } 232d4002b98SHong Zhang 233d4002b98SHong Zhang PetscErrorCode MatGetValues_MPISELL(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],PetscScalar v[]) 234d4002b98SHong Zhang { 235d4002b98SHong Zhang Mat_MPISELL *sell=(Mat_MPISELL*)mat->data; 236d4002b98SHong Zhang PetscErrorCode ierr; 237d4002b98SHong Zhang PetscInt i,j,rstart=mat->rmap->rstart,rend=mat->rmap->rend; 238d4002b98SHong Zhang PetscInt cstart=mat->cmap->rstart,cend=mat->cmap->rend,row,col; 239d4002b98SHong Zhang 240d4002b98SHong Zhang PetscFunctionBegin; 241d4002b98SHong Zhang for (i=0; i<m; i++) { 242d4002b98SHong Zhang if (idxm[i] < 0) continue; /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",idxm[i]);*/ 243d4002b98SHong Zhang if (idxm[i] >= mat->rmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",idxm[i],mat->rmap->N-1); 244d4002b98SHong Zhang if (idxm[i] >= rstart && idxm[i] < rend) { 245d4002b98SHong Zhang row = idxm[i] - rstart; 246d4002b98SHong Zhang for (j=0; j<n; j++) { 247d4002b98SHong Zhang if (idxn[j] < 0) continue; /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",idxn[j]); */ 248d4002b98SHong Zhang if (idxn[j] >= mat->cmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",idxn[j],mat->cmap->N-1); 249d4002b98SHong Zhang if (idxn[j] >= cstart && idxn[j] < cend) { 250d4002b98SHong Zhang col = idxn[j] - cstart; 251d4002b98SHong Zhang ierr = MatGetValues(sell->A,1,&row,1,&col,v+i*n+j);CHKERRQ(ierr); 252d4002b98SHong Zhang } else { 253d4002b98SHong Zhang if (!sell->colmap) { 254d4002b98SHong Zhang ierr = MatCreateColmap_MPISELL_Private(mat);CHKERRQ(ierr); 255d4002b98SHong Zhang } 256d4002b98SHong Zhang #if defined(PETSC_USE_CTABLE) 257d4002b98SHong Zhang ierr = PetscTableFind(sell->colmap,idxn[j]+1,&col);CHKERRQ(ierr); 258d4002b98SHong Zhang col--; 259d4002b98SHong Zhang #else 260d4002b98SHong Zhang col = sell->colmap[idxn[j]] - 1; 261d4002b98SHong Zhang #endif 262d4002b98SHong Zhang if ((col < 0) || (sell->garray[col] != idxn[j])) *(v+i*n+j) = 0.0; 263d4002b98SHong Zhang else { 264d4002b98SHong Zhang ierr = MatGetValues(sell->B,1,&row,1,&col,v+i*n+j);CHKERRQ(ierr); 265d4002b98SHong Zhang } 266d4002b98SHong Zhang } 267d4002b98SHong Zhang } 268d4002b98SHong Zhang } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only local values currently supported"); 269d4002b98SHong Zhang } 270d4002b98SHong Zhang PetscFunctionReturn(0); 271d4002b98SHong Zhang } 272d4002b98SHong Zhang 273d4002b98SHong Zhang extern PetscErrorCode MatMultDiagonalBlock_MPISELL(Mat,Vec,Vec); 274d4002b98SHong Zhang 275d4002b98SHong Zhang PetscErrorCode MatAssemblyBegin_MPISELL(Mat mat,MatAssemblyType mode) 276d4002b98SHong Zhang { 277d4002b98SHong Zhang Mat_MPISELL *sell=(Mat_MPISELL*)mat->data; 278d4002b98SHong Zhang PetscErrorCode ierr; 279d4002b98SHong Zhang PetscInt nstash,reallocs; 280d4002b98SHong Zhang 281d4002b98SHong Zhang PetscFunctionBegin; 282d4002b98SHong Zhang if (sell->donotstash || mat->nooffprocentries) PetscFunctionReturn(0); 283d4002b98SHong Zhang 284d4002b98SHong Zhang ierr = MatStashScatterBegin_Private(mat,&mat->stash,mat->rmap->range);CHKERRQ(ierr); 285d4002b98SHong Zhang ierr = MatStashGetInfo_Private(&mat->stash,&nstash,&reallocs);CHKERRQ(ierr); 286d4002b98SHong Zhang ierr = PetscInfo2(sell->A,"Stash has %D entries, uses %D mallocs.\n",nstash,reallocs);CHKERRQ(ierr); 287d4002b98SHong Zhang PetscFunctionReturn(0); 288d4002b98SHong Zhang } 289d4002b98SHong Zhang 290d4002b98SHong Zhang PetscErrorCode MatAssemblyEnd_MPISELL(Mat mat,MatAssemblyType mode) 291d4002b98SHong Zhang { 292d4002b98SHong Zhang Mat_MPISELL *sell=(Mat_MPISELL*)mat->data; 293d4002b98SHong Zhang PetscErrorCode ierr; 294d4002b98SHong Zhang PetscMPIInt n; 295d4002b98SHong Zhang PetscInt i,flg; 296d4002b98SHong Zhang PetscInt *row,*col; 297d4002b98SHong Zhang PetscScalar *val; 298d4002b98SHong Zhang PetscBool other_disassembled; 299d4002b98SHong Zhang /* do not use 'b = (Mat_SeqSELL*)sell->B->data' as B can be reset in disassembly */ 300d4002b98SHong Zhang PetscFunctionBegin; 301d4002b98SHong Zhang if (!sell->donotstash && !mat->nooffprocentries) { 302d4002b98SHong Zhang while (1) { 303d4002b98SHong Zhang ierr = MatStashScatterGetMesg_Private(&mat->stash,&n,&row,&col,&val,&flg);CHKERRQ(ierr); 304d4002b98SHong Zhang if (!flg) break; 305d4002b98SHong Zhang 306d4002b98SHong Zhang for (i=0; i<n; i++) { /* assemble one by one */ 307d4002b98SHong Zhang ierr = MatSetValues_MPISELL(mat,1,row+i,1,col+i,val+i,mat->insertmode);CHKERRQ(ierr); 308d4002b98SHong Zhang } 309d4002b98SHong Zhang } 310d4002b98SHong Zhang ierr = MatStashScatterEnd_Private(&mat->stash);CHKERRQ(ierr); 311d4002b98SHong Zhang } 312d4002b98SHong Zhang ierr = MatAssemblyBegin(sell->A,mode);CHKERRQ(ierr); 313d4002b98SHong Zhang ierr = MatAssemblyEnd(sell->A,mode);CHKERRQ(ierr); 314d4002b98SHong Zhang 315d4002b98SHong Zhang /* 316d4002b98SHong Zhang determine if any processor has disassembled, if so we must 317d4002b98SHong Zhang also disassemble ourselfs, in order that we may reassemble. 318d4002b98SHong Zhang */ 319d4002b98SHong Zhang /* 320d4002b98SHong Zhang if nonzero structure of submatrix B cannot change then we know that 321d4002b98SHong Zhang no processor disassembled thus we can skip this stuff 322d4002b98SHong Zhang */ 323d4002b98SHong Zhang if (!((Mat_SeqSELL*)sell->B->data)->nonew) { 324d4002b98SHong Zhang ierr = MPIU_Allreduce(&mat->was_assembled,&other_disassembled,1,MPIU_BOOL,MPI_PROD,PetscObjectComm((PetscObject)mat));CHKERRQ(ierr); 325d4002b98SHong Zhang if (mat->was_assembled && !other_disassembled) { 326d4002b98SHong Zhang SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatDisAssemble not implemented yet\n"); 327d4002b98SHong Zhang ierr = MatDisAssemble_MPISELL(mat);CHKERRQ(ierr); 328d4002b98SHong Zhang } 329d4002b98SHong Zhang } 330d4002b98SHong Zhang if (!mat->was_assembled && mode == MAT_FINAL_ASSEMBLY) { 331d4002b98SHong Zhang ierr = MatSetUpMultiply_MPISELL(mat);CHKERRQ(ierr); 332d4002b98SHong Zhang } 333d4002b98SHong Zhang /* 334d4002b98SHong Zhang ierr = MatSetOption(sell->B,MAT_USE_INODES,PETSC_FALSE);CHKERRQ(ierr); 335d4002b98SHong Zhang */ 336d4002b98SHong Zhang ierr = MatAssemblyBegin(sell->B,mode);CHKERRQ(ierr); 337d4002b98SHong Zhang ierr = MatAssemblyEnd(sell->B,mode);CHKERRQ(ierr); 338d4002b98SHong Zhang ierr = PetscFree2(sell->rowvalues,sell->rowindices);CHKERRQ(ierr); 339d4002b98SHong Zhang sell->rowvalues = 0; 340d4002b98SHong Zhang ierr = VecDestroy(&sell->diag);CHKERRQ(ierr); 341d4002b98SHong Zhang 342d4002b98SHong Zhang /* if no new nonzero locations are allowed in matrix then only set the matrix state the first time through */ 343d4002b98SHong Zhang if ((!mat->was_assembled && mode == MAT_FINAL_ASSEMBLY) || !((Mat_SeqSELL*)(sell->A->data))->nonew) { 344d4002b98SHong Zhang PetscObjectState state = sell->A->nonzerostate + sell->B->nonzerostate; 345d4002b98SHong Zhang ierr = MPIU_Allreduce(&state,&mat->nonzerostate,1,MPIU_INT64,MPI_SUM,PetscObjectComm((PetscObject)mat));CHKERRQ(ierr); 346d4002b98SHong Zhang } 347d4002b98SHong Zhang PetscFunctionReturn(0); 348d4002b98SHong Zhang } 349d4002b98SHong Zhang 350d4002b98SHong Zhang PetscErrorCode MatZeroEntries_MPISELL(Mat A) 351d4002b98SHong Zhang { 352d4002b98SHong Zhang Mat_MPISELL *l=(Mat_MPISELL*)A->data; 353d4002b98SHong Zhang PetscErrorCode ierr; 354d4002b98SHong Zhang 355d4002b98SHong Zhang PetscFunctionBegin; 356d4002b98SHong Zhang ierr = MatZeroEntries(l->A);CHKERRQ(ierr); 357d4002b98SHong Zhang ierr = MatZeroEntries(l->B);CHKERRQ(ierr); 358d4002b98SHong Zhang PetscFunctionReturn(0); 359d4002b98SHong Zhang } 360d4002b98SHong Zhang 361d4002b98SHong Zhang PetscErrorCode MatMult_MPISELL(Mat A,Vec xx,Vec yy) 362d4002b98SHong Zhang { 363d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 364d4002b98SHong Zhang PetscErrorCode ierr; 365d4002b98SHong Zhang PetscInt nt; 366d4002b98SHong Zhang 367d4002b98SHong Zhang PetscFunctionBegin; 368d4002b98SHong Zhang ierr = VecGetLocalSize(xx,&nt);CHKERRQ(ierr); 369d4002b98SHong Zhang if (nt != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Incompatible partition of A (%D) and xx (%D)",A->cmap->n,nt); 370d4002b98SHong Zhang ierr = VecScatterBegin(a->Mvctx,xx,a->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 371d4002b98SHong Zhang ierr = (*a->A->ops->mult)(a->A,xx,yy);CHKERRQ(ierr); 372d4002b98SHong Zhang ierr = VecScatterEnd(a->Mvctx,xx,a->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 373d4002b98SHong Zhang ierr = (*a->B->ops->multadd)(a->B,a->lvec,yy,yy);CHKERRQ(ierr); 374d4002b98SHong Zhang PetscFunctionReturn(0); 375d4002b98SHong Zhang } 376d4002b98SHong Zhang 377d4002b98SHong Zhang PetscErrorCode MatMultDiagonalBlock_MPISELL(Mat A,Vec bb,Vec xx) 378d4002b98SHong Zhang { 379d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 380d4002b98SHong Zhang PetscErrorCode ierr; 381d4002b98SHong Zhang 382d4002b98SHong Zhang PetscFunctionBegin; 383d4002b98SHong Zhang ierr = MatMultDiagonalBlock(a->A,bb,xx);CHKERRQ(ierr); 384d4002b98SHong Zhang PetscFunctionReturn(0); 385d4002b98SHong Zhang } 386d4002b98SHong Zhang 387d4002b98SHong Zhang PetscErrorCode MatMultAdd_MPISELL(Mat A,Vec xx,Vec yy,Vec zz) 388d4002b98SHong Zhang { 389d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 390d4002b98SHong Zhang PetscErrorCode ierr; 391d4002b98SHong Zhang 392d4002b98SHong Zhang PetscFunctionBegin; 393d4002b98SHong Zhang ierr = VecScatterBegin(a->Mvctx,xx,a->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 394d4002b98SHong Zhang ierr = (*a->A->ops->multadd)(a->A,xx,yy,zz);CHKERRQ(ierr); 395d4002b98SHong Zhang ierr = VecScatterEnd(a->Mvctx,xx,a->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 396d4002b98SHong Zhang ierr = (*a->B->ops->multadd)(a->B,a->lvec,zz,zz);CHKERRQ(ierr); 397d4002b98SHong Zhang PetscFunctionReturn(0); 398d4002b98SHong Zhang } 399d4002b98SHong Zhang 400d4002b98SHong Zhang PetscErrorCode MatMultTranspose_MPISELL(Mat A,Vec xx,Vec yy) 401d4002b98SHong Zhang { 402d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 403d4002b98SHong Zhang PetscErrorCode ierr; 404d4002b98SHong Zhang PetscBool merged; 405d4002b98SHong Zhang 406d4002b98SHong Zhang PetscFunctionBegin; 407d4002b98SHong Zhang ierr = VecScatterGetMerged(a->Mvctx,&merged);CHKERRQ(ierr); 408d4002b98SHong Zhang /* do nondiagonal part */ 409d4002b98SHong Zhang ierr = (*a->B->ops->multtranspose)(a->B,xx,a->lvec);CHKERRQ(ierr); 410d4002b98SHong Zhang if (!merged) { 411d4002b98SHong Zhang /* send it on its way */ 412d4002b98SHong Zhang ierr = VecScatterBegin(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 413d4002b98SHong Zhang /* do local part */ 414d4002b98SHong Zhang ierr = (*a->A->ops->multtranspose)(a->A,xx,yy);CHKERRQ(ierr); 415d4002b98SHong Zhang /* receive remote parts: note this assumes the values are not actually */ 416d4002b98SHong Zhang /* added in yy until the next line, */ 417d4002b98SHong Zhang ierr = VecScatterEnd(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 418d4002b98SHong Zhang } else { 419d4002b98SHong Zhang /* do local part */ 420d4002b98SHong Zhang ierr = (*a->A->ops->multtranspose)(a->A,xx,yy);CHKERRQ(ierr); 421d4002b98SHong Zhang /* send it on its way */ 422d4002b98SHong Zhang ierr = VecScatterBegin(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 423d4002b98SHong Zhang /* values actually were received in the Begin() but we need to call this nop */ 424d4002b98SHong Zhang ierr = VecScatterEnd(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 425d4002b98SHong Zhang } 426d4002b98SHong Zhang PetscFunctionReturn(0); 427d4002b98SHong Zhang } 428d4002b98SHong Zhang 429d4002b98SHong Zhang PetscErrorCode MatIsTranspose_MPISELL(Mat Amat,Mat Bmat,PetscReal tol,PetscBool *f) 430d4002b98SHong Zhang { 431d4002b98SHong Zhang MPI_Comm comm; 432d4002b98SHong Zhang Mat_MPISELL *Asell=(Mat_MPISELL*)Amat->data,*Bsell; 433d4002b98SHong Zhang Mat Adia=Asell->A,Bdia,Aoff,Boff,*Aoffs,*Boffs; 434d4002b98SHong Zhang IS Me,Notme; 435d4002b98SHong Zhang PetscErrorCode ierr; 436d4002b98SHong Zhang PetscInt M,N,first,last,*notme,i; 437d4002b98SHong Zhang PetscMPIInt size; 438d4002b98SHong Zhang 439d4002b98SHong Zhang PetscFunctionBegin; 440d4002b98SHong Zhang /* Easy test: symmetric diagonal block */ 441d4002b98SHong Zhang Bsell = (Mat_MPISELL*)Bmat->data; Bdia = Bsell->A; 442d4002b98SHong Zhang ierr = MatIsTranspose(Adia,Bdia,tol,f);CHKERRQ(ierr); 443d4002b98SHong Zhang if (!*f) PetscFunctionReturn(0); 444d4002b98SHong Zhang ierr = PetscObjectGetComm((PetscObject)Amat,&comm);CHKERRQ(ierr); 445d4002b98SHong Zhang ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 446d4002b98SHong Zhang if (size == 1) PetscFunctionReturn(0); 447d4002b98SHong Zhang 448d4002b98SHong Zhang /* Hard test: off-diagonal block. This takes a MatCreateSubMatrix. */ 449d4002b98SHong Zhang ierr = MatGetSize(Amat,&M,&N);CHKERRQ(ierr); 450d4002b98SHong Zhang ierr = MatGetOwnershipRange(Amat,&first,&last);CHKERRQ(ierr); 451d4002b98SHong Zhang ierr = PetscMalloc1(N-last+first,¬me);CHKERRQ(ierr); 452d4002b98SHong Zhang for (i=0; i<first; i++) notme[i] = i; 453d4002b98SHong Zhang for (i=last; i<M; i++) notme[i-last+first] = i; 454d4002b98SHong Zhang ierr = ISCreateGeneral(MPI_COMM_SELF,N-last+first,notme,PETSC_COPY_VALUES,&Notme);CHKERRQ(ierr); 455d4002b98SHong Zhang ierr = ISCreateStride(MPI_COMM_SELF,last-first,first,1,&Me);CHKERRQ(ierr); 456d4002b98SHong Zhang ierr = MatCreateSubMatrices(Amat,1,&Me,&Notme,MAT_INITIAL_MATRIX,&Aoffs);CHKERRQ(ierr); 457d4002b98SHong Zhang Aoff = Aoffs[0]; 458d4002b98SHong Zhang ierr = MatCreateSubMatrices(Bmat,1,&Notme,&Me,MAT_INITIAL_MATRIX,&Boffs);CHKERRQ(ierr); 459d4002b98SHong Zhang Boff = Boffs[0]; 460d4002b98SHong Zhang ierr = MatIsTranspose(Aoff,Boff,tol,f);CHKERRQ(ierr); 461d4002b98SHong Zhang ierr = MatDestroyMatrices(1,&Aoffs);CHKERRQ(ierr); 462d4002b98SHong Zhang ierr = MatDestroyMatrices(1,&Boffs);CHKERRQ(ierr); 463d4002b98SHong Zhang ierr = ISDestroy(&Me);CHKERRQ(ierr); 464d4002b98SHong Zhang ierr = ISDestroy(&Notme);CHKERRQ(ierr); 465d4002b98SHong Zhang ierr = PetscFree(notme);CHKERRQ(ierr); 466d4002b98SHong Zhang PetscFunctionReturn(0); 467d4002b98SHong Zhang } 468d4002b98SHong Zhang 469d4002b98SHong Zhang PetscErrorCode MatMultTransposeAdd_MPISELL(Mat A,Vec xx,Vec yy,Vec zz) 470d4002b98SHong Zhang { 471d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 472d4002b98SHong Zhang PetscErrorCode ierr; 473d4002b98SHong Zhang 474d4002b98SHong Zhang PetscFunctionBegin; 475d4002b98SHong Zhang /* do nondiagonal part */ 476d4002b98SHong Zhang ierr = (*a->B->ops->multtranspose)(a->B,xx,a->lvec);CHKERRQ(ierr); 477d4002b98SHong Zhang /* send it on its way */ 478d4002b98SHong Zhang ierr = VecScatterBegin(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 479d4002b98SHong Zhang /* do local part */ 480d4002b98SHong Zhang ierr = (*a->A->ops->multtransposeadd)(a->A,xx,yy,zz);CHKERRQ(ierr); 481d4002b98SHong Zhang /* receive remote parts */ 482d4002b98SHong Zhang ierr = VecScatterEnd(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 483d4002b98SHong Zhang PetscFunctionReturn(0); 484d4002b98SHong Zhang } 485d4002b98SHong Zhang 486d4002b98SHong Zhang /* 487d4002b98SHong Zhang This only works correctly for square matrices where the subblock A->A is the 488d4002b98SHong Zhang diagonal block 489d4002b98SHong Zhang */ 490d4002b98SHong Zhang PetscErrorCode MatGetDiagonal_MPISELL(Mat A,Vec v) 491d4002b98SHong Zhang { 492d4002b98SHong Zhang PetscErrorCode ierr; 493d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 494d4002b98SHong Zhang 495d4002b98SHong Zhang PetscFunctionBegin; 496d4002b98SHong Zhang if (A->rmap->N != A->cmap->N) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Supports only square matrix where A->A is diag block"); 497d4002b98SHong Zhang if (A->rmap->rstart != A->cmap->rstart || A->rmap->rend != A->cmap->rend) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"row partition must equal col partition"); 498d4002b98SHong Zhang ierr = MatGetDiagonal(a->A,v);CHKERRQ(ierr); 499d4002b98SHong Zhang PetscFunctionReturn(0); 500d4002b98SHong Zhang } 501d4002b98SHong Zhang 502d4002b98SHong Zhang PetscErrorCode MatScale_MPISELL(Mat A,PetscScalar aa) 503d4002b98SHong Zhang { 504d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 505d4002b98SHong Zhang PetscErrorCode ierr; 506d4002b98SHong Zhang 507d4002b98SHong Zhang PetscFunctionBegin; 508d4002b98SHong Zhang ierr = MatScale(a->A,aa);CHKERRQ(ierr); 509d4002b98SHong Zhang ierr = MatScale(a->B,aa);CHKERRQ(ierr); 510d4002b98SHong Zhang PetscFunctionReturn(0); 511d4002b98SHong Zhang } 512d4002b98SHong Zhang 513d4002b98SHong Zhang PetscErrorCode MatDestroy_MPISELL(Mat mat) 514d4002b98SHong Zhang { 515d4002b98SHong Zhang Mat_MPISELL *sell=(Mat_MPISELL*)mat->data; 516d4002b98SHong Zhang PetscErrorCode ierr; 517d4002b98SHong Zhang 518d4002b98SHong Zhang PetscFunctionBegin; 519d4002b98SHong Zhang #if defined(PETSC_USE_LOG) 520d4002b98SHong Zhang PetscLogObjectState((PetscObject)mat,"Rows=%D, Cols=%D",mat->rmap->N,mat->cmap->N); 521d4002b98SHong Zhang #endif 522d4002b98SHong Zhang ierr = MatStashDestroy_Private(&mat->stash);CHKERRQ(ierr); 523d4002b98SHong Zhang ierr = VecDestroy(&sell->diag);CHKERRQ(ierr); 524d4002b98SHong Zhang ierr = MatDestroy(&sell->A);CHKERRQ(ierr); 525d4002b98SHong Zhang ierr = MatDestroy(&sell->B);CHKERRQ(ierr); 526d4002b98SHong Zhang #if defined(PETSC_USE_CTABLE) 527d4002b98SHong Zhang ierr = PetscTableDestroy(&sell->colmap);CHKERRQ(ierr); 528d4002b98SHong Zhang #else 529d4002b98SHong Zhang ierr = PetscFree(sell->colmap);CHKERRQ(ierr); 530d4002b98SHong Zhang #endif 531d4002b98SHong Zhang ierr = PetscFree(sell->garray);CHKERRQ(ierr); 532d4002b98SHong Zhang ierr = VecDestroy(&sell->lvec);CHKERRQ(ierr); 533d4002b98SHong Zhang ierr = VecScatterDestroy(&sell->Mvctx);CHKERRQ(ierr); 534d4002b98SHong Zhang ierr = PetscFree2(sell->rowvalues,sell->rowindices);CHKERRQ(ierr); 535d4002b98SHong Zhang ierr = PetscFree(sell->ld);CHKERRQ(ierr); 536d4002b98SHong Zhang ierr = PetscFree(mat->data);CHKERRQ(ierr); 537d4002b98SHong Zhang 538d4002b98SHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)mat,0);CHKERRQ(ierr); 539d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)mat,"MatStoreValues_C",NULL);CHKERRQ(ierr); 540d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)mat,"MatRetrieveValues_C",NULL);CHKERRQ(ierr); 541d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)mat,"MatIsTranspose_C",NULL);CHKERRQ(ierr); 542d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMPISELLSetPreallocation_C",NULL);CHKERRQ(ierr); 543d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)mat,"MatConvert_mpisell_mpiaij_C",NULL);CHKERRQ(ierr); 544d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)mat,"MatDiagonalScaleLocal_C",NULL);CHKERRQ(ierr); 545d4002b98SHong Zhang PetscFunctionReturn(0); 546d4002b98SHong Zhang } 547d4002b98SHong Zhang 548d4002b98SHong Zhang #include <petscdraw.h> 549d4002b98SHong Zhang PetscErrorCode MatView_MPISELL_ASCIIorDraworSocket(Mat mat,PetscViewer viewer) 550d4002b98SHong Zhang { 551d4002b98SHong Zhang Mat_MPISELL *sell=(Mat_MPISELL*)mat->data; 552d4002b98SHong Zhang PetscErrorCode ierr; 553d4002b98SHong Zhang PetscMPIInt rank=sell->rank,size=sell->size; 554d4002b98SHong Zhang PetscBool isdraw,iascii,isbinary; 555d4002b98SHong Zhang PetscViewer sviewer; 556d4002b98SHong Zhang PetscViewerFormat format; 557d4002b98SHong Zhang 558d4002b98SHong Zhang PetscFunctionBegin; 559d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 560d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 561d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 562d4002b98SHong Zhang if (iascii) { 563d4002b98SHong Zhang ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 564d4002b98SHong Zhang if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 565d4002b98SHong Zhang MatInfo info; 566d4002b98SHong Zhang PetscBool inodes; 567d4002b98SHong Zhang 568d4002b98SHong Zhang ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)mat),&rank);CHKERRQ(ierr); 569d4002b98SHong Zhang ierr = MatGetInfo(mat,MAT_LOCAL,&info);CHKERRQ(ierr); 570d4002b98SHong Zhang ierr = MatInodeGetInodeSizes(sell->A,NULL,(PetscInt**)&inodes,NULL);CHKERRQ(ierr); 571d4002b98SHong Zhang ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr); 572d4002b98SHong Zhang if (!inodes) { 573d4002b98SHong Zhang ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] Local rows %D nz %D nz alloced %D mem %D, not using I-node routines\n", 574d4002b98SHong Zhang rank,mat->rmap->n,(PetscInt)info.nz_used,(PetscInt)info.nz_allocated,(PetscInt)info.memory);CHKERRQ(ierr); 575d4002b98SHong Zhang } else { 576d4002b98SHong Zhang ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] Local rows %D nz %D nz alloced %D mem %D, using I-node routines\n", 577d4002b98SHong Zhang rank,mat->rmap->n,(PetscInt)info.nz_used,(PetscInt)info.nz_allocated,(PetscInt)info.memory);CHKERRQ(ierr); 578d4002b98SHong Zhang } 579d4002b98SHong Zhang ierr = MatGetInfo(sell->A,MAT_LOCAL,&info);CHKERRQ(ierr); 580d4002b98SHong Zhang ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] on-diagonal part: nz %D \n",rank,(PetscInt)info.nz_used);CHKERRQ(ierr); 581d4002b98SHong Zhang ierr = MatGetInfo(sell->B,MAT_LOCAL,&info);CHKERRQ(ierr); 582d4002b98SHong Zhang ierr = PetscViewerASCIISynchronizedPrintf(viewer,"[%d] off-diagonal part: nz %D \n",rank,(PetscInt)info.nz_used);CHKERRQ(ierr); 583d4002b98SHong Zhang ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 584d4002b98SHong Zhang ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr); 585d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"Information on VecScatter used in matrix-vector product: \n");CHKERRQ(ierr); 586d4002b98SHong Zhang ierr = VecScatterView(sell->Mvctx,viewer);CHKERRQ(ierr); 587d4002b98SHong Zhang PetscFunctionReturn(0); 588d4002b98SHong Zhang } else if (format == PETSC_VIEWER_ASCII_INFO) { 589d4002b98SHong Zhang PetscInt inodecount,inodelimit,*inodes; 590d4002b98SHong Zhang ierr = MatInodeGetInodeSizes(sell->A,&inodecount,&inodes,&inodelimit);CHKERRQ(ierr); 591d4002b98SHong Zhang if (inodes) { 592d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"using I-node (on process 0) routines: found %D nodes, limit used is %D\n",inodecount,inodelimit);CHKERRQ(ierr); 593d4002b98SHong Zhang } else { 594d4002b98SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"not using I-node (on process 0) routines\n");CHKERRQ(ierr); 595d4002b98SHong Zhang } 596d4002b98SHong Zhang PetscFunctionReturn(0); 597d4002b98SHong Zhang } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 598d4002b98SHong Zhang PetscFunctionReturn(0); 599d4002b98SHong Zhang } 600d4002b98SHong Zhang } else if (isbinary) { 601d4002b98SHong Zhang if (size == 1) { 602d4002b98SHong Zhang ierr = PetscObjectSetName((PetscObject)sell->A,((PetscObject)mat)->name);CHKERRQ(ierr); 603d4002b98SHong Zhang ierr = MatView(sell->A,viewer);CHKERRQ(ierr); 604d4002b98SHong Zhang } else { 605d4002b98SHong Zhang /* ierr = MatView_MPISELL_Binary(mat,viewer);CHKERRQ(ierr); */ 606d4002b98SHong Zhang } 607d4002b98SHong Zhang PetscFunctionReturn(0); 608d4002b98SHong Zhang } else if (isdraw) { 609d4002b98SHong Zhang PetscDraw draw; 610d4002b98SHong Zhang PetscBool isnull; 611d4002b98SHong Zhang ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 612d4002b98SHong Zhang ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); 613d4002b98SHong Zhang if (isnull) PetscFunctionReturn(0); 614d4002b98SHong Zhang } 615d4002b98SHong Zhang 616d4002b98SHong Zhang { 617d4002b98SHong Zhang /* assemble the entire matrix onto first processor. */ 618d4002b98SHong Zhang Mat A; 619d4002b98SHong Zhang Mat_SeqSELL *Aloc; 620d4002b98SHong Zhang PetscInt M=mat->rmap->N,N=mat->cmap->N,*acolidx,row,col,i,j; 621d4002b98SHong Zhang MatScalar *aval; 622d4002b98SHong Zhang PetscBool isnonzero; 623d4002b98SHong Zhang 624d4002b98SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)mat),&A);CHKERRQ(ierr); 625d4002b98SHong Zhang if (!rank) { 626d4002b98SHong Zhang ierr = MatSetSizes(A,M,N,M,N);CHKERRQ(ierr); 627d4002b98SHong Zhang } else { 628d4002b98SHong Zhang ierr = MatSetSizes(A,0,0,M,N);CHKERRQ(ierr); 629d4002b98SHong Zhang } 630d4002b98SHong Zhang /* This is just a temporary matrix, so explicitly using MATMPISELL is probably best */ 631d4002b98SHong Zhang ierr = MatSetType(A,MATMPISELL);CHKERRQ(ierr); 632d4002b98SHong Zhang ierr = MatMPISELLSetPreallocation(A,0,NULL,0,NULL);CHKERRQ(ierr); 633d4002b98SHong Zhang ierr = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_FALSE);CHKERRQ(ierr); 634d4002b98SHong Zhang ierr = PetscLogObjectParent((PetscObject)mat,(PetscObject)A);CHKERRQ(ierr); 635d4002b98SHong Zhang 636d4002b98SHong Zhang /* copy over the A part */ 637d4002b98SHong Zhang Aloc = (Mat_SeqSELL*)sell->A->data; 638d4002b98SHong Zhang acolidx = Aloc->colidx; aval = Aloc->val; 639d4002b98SHong Zhang for (i=0; i<Aloc->totalslices; i++) { /* loop over slices */ 640d4002b98SHong Zhang for (j=Aloc->sliidx[i]; j<Aloc->sliidx[i+1]; j++) { 641d4002b98SHong Zhang isnonzero = (PetscBool)((j-Aloc->sliidx[i])/8 < Aloc->rlen[(i<<3)+(j&0x07)]); 642d4002b98SHong Zhang if (isnonzero) { /* check the mask bit */ 643d4002b98SHong Zhang row = (i<<3)+(j&0x07) + mat->rmap->rstart; /* i<<3 is the starting row of this slice */ 644d4002b98SHong Zhang col = *acolidx + mat->rmap->rstart; 645d4002b98SHong Zhang ierr = MatSetValues(A,1,&row,1,&col,aval,INSERT_VALUES);CHKERRQ(ierr); 646d4002b98SHong Zhang } 647d4002b98SHong Zhang aval++; acolidx++; 648d4002b98SHong Zhang } 649d4002b98SHong Zhang } 650d4002b98SHong Zhang 651d4002b98SHong Zhang /* copy over the B part */ 652d4002b98SHong Zhang Aloc = (Mat_SeqSELL*)sell->B->data; 653d4002b98SHong Zhang acolidx = Aloc->colidx; aval = Aloc->val; 654d4002b98SHong Zhang for (i=0; i<Aloc->totalslices; i++) { 655d4002b98SHong Zhang for (j=Aloc->sliidx[i]; j<Aloc->sliidx[i+1]; j++) { 656d4002b98SHong Zhang isnonzero = (PetscBool)((j-Aloc->sliidx[i])/8 < Aloc->rlen[(i<<3)+(j&0x07)]); 657d4002b98SHong Zhang if (isnonzero) { 658d4002b98SHong Zhang row = (i<<3)+(j&0x07) + mat->rmap->rstart; 659d4002b98SHong Zhang col = sell->garray[*acolidx]; 660d4002b98SHong Zhang ierr = MatSetValues(A,1,&row,1,&col,aval,INSERT_VALUES);CHKERRQ(ierr); 661d4002b98SHong Zhang } 662d4002b98SHong Zhang aval++; acolidx++; 663d4002b98SHong Zhang } 664d4002b98SHong Zhang } 665d4002b98SHong Zhang 666d4002b98SHong Zhang ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 667d4002b98SHong Zhang ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 668d4002b98SHong Zhang /* 669d4002b98SHong Zhang Everyone has to call to draw the matrix since the graphics waits are 670d4002b98SHong Zhang synchronized across all processors that share the PetscDraw object 671d4002b98SHong Zhang */ 672d4002b98SHong Zhang ierr = PetscViewerGetSubViewer(viewer,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 673d4002b98SHong Zhang if (!rank) { 674d4002b98SHong Zhang ierr = PetscObjectSetName((PetscObject)((Mat_MPISELL*)(A->data))->A,((PetscObject)mat)->name);CHKERRQ(ierr); 675d4002b98SHong Zhang ierr = MatView_SeqSELL(((Mat_MPISELL*)(A->data))->A,sviewer);CHKERRQ(ierr); 676d4002b98SHong Zhang } 677d4002b98SHong Zhang ierr = PetscViewerRestoreSubViewer(viewer,PETSC_COMM_SELF,&sviewer);CHKERRQ(ierr); 678d4002b98SHong Zhang ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 679d4002b98SHong Zhang ierr = MatDestroy(&A);CHKERRQ(ierr); 680d4002b98SHong Zhang } 681d4002b98SHong Zhang PetscFunctionReturn(0); 682d4002b98SHong Zhang } 683d4002b98SHong Zhang 684d4002b98SHong Zhang PetscErrorCode MatView_MPISELL(Mat mat,PetscViewer viewer) 685d4002b98SHong Zhang { 686d4002b98SHong Zhang PetscErrorCode ierr; 687d4002b98SHong Zhang PetscBool iascii,isdraw,issocket,isbinary; 688d4002b98SHong Zhang 689d4002b98SHong Zhang PetscFunctionBegin; 690d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 691d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 692d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 693d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSOCKET,&issocket);CHKERRQ(ierr); 694d4002b98SHong Zhang if (iascii || isdraw || isbinary || issocket) { 695d4002b98SHong Zhang ierr = MatView_MPISELL_ASCIIorDraworSocket(mat,viewer);CHKERRQ(ierr); 696d4002b98SHong Zhang } 697d4002b98SHong Zhang PetscFunctionReturn(0); 698d4002b98SHong Zhang } 699d4002b98SHong Zhang 700d4002b98SHong Zhang PetscErrorCode MatGetGhosts_MPISELL(Mat mat,PetscInt *nghosts,const PetscInt *ghosts[]) 701d4002b98SHong Zhang { 702d4002b98SHong Zhang Mat_MPISELL *sell=(Mat_MPISELL*)mat->data; 703d4002b98SHong Zhang PetscErrorCode ierr; 704d4002b98SHong Zhang 705d4002b98SHong Zhang PetscFunctionBegin; 706d4002b98SHong Zhang ierr = MatGetSize(sell->B,NULL,nghosts);CHKERRQ(ierr); 707d4002b98SHong Zhang if (ghosts) *ghosts = sell->garray; 708d4002b98SHong Zhang PetscFunctionReturn(0); 709d4002b98SHong Zhang } 710d4002b98SHong Zhang 711d4002b98SHong Zhang PetscErrorCode MatGetInfo_MPISELL(Mat matin,MatInfoType flag,MatInfo *info) 712d4002b98SHong Zhang { 713d4002b98SHong Zhang Mat_MPISELL *mat=(Mat_MPISELL*)matin->data; 714d4002b98SHong Zhang Mat A=mat->A,B=mat->B; 715d4002b98SHong Zhang PetscErrorCode ierr; 716d4002b98SHong Zhang PetscReal isend[5],irecv[5]; 717d4002b98SHong Zhang 718d4002b98SHong Zhang PetscFunctionBegin; 719d4002b98SHong Zhang info->block_size = 1.0; 720d4002b98SHong Zhang ierr = MatGetInfo(A,MAT_LOCAL,info);CHKERRQ(ierr); 721d4002b98SHong Zhang 722d4002b98SHong Zhang isend[0] = info->nz_used; isend[1] = info->nz_allocated; isend[2] = info->nz_unneeded; 723d4002b98SHong Zhang isend[3] = info->memory; isend[4] = info->mallocs; 724d4002b98SHong Zhang 725d4002b98SHong Zhang ierr = MatGetInfo(B,MAT_LOCAL,info);CHKERRQ(ierr); 726d4002b98SHong Zhang 727d4002b98SHong Zhang isend[0] += info->nz_used; isend[1] += info->nz_allocated; isend[2] += info->nz_unneeded; 728d4002b98SHong Zhang isend[3] += info->memory; isend[4] += info->mallocs; 729d4002b98SHong Zhang if (flag == MAT_LOCAL) { 730d4002b98SHong Zhang info->nz_used = isend[0]; 731d4002b98SHong Zhang info->nz_allocated = isend[1]; 732d4002b98SHong Zhang info->nz_unneeded = isend[2]; 733d4002b98SHong Zhang info->memory = isend[3]; 734d4002b98SHong Zhang info->mallocs = isend[4]; 735d4002b98SHong Zhang } else if (flag == MAT_GLOBAL_MAX) { 736d4002b98SHong Zhang ierr = MPIU_Allreduce(isend,irecv,5,MPIU_REAL,MPIU_MAX,PetscObjectComm((PetscObject)matin));CHKERRQ(ierr); 737d4002b98SHong Zhang 738d4002b98SHong Zhang info->nz_used = irecv[0]; 739d4002b98SHong Zhang info->nz_allocated = irecv[1]; 740d4002b98SHong Zhang info->nz_unneeded = irecv[2]; 741d4002b98SHong Zhang info->memory = irecv[3]; 742d4002b98SHong Zhang info->mallocs = irecv[4]; 743d4002b98SHong Zhang } else if (flag == MAT_GLOBAL_SUM) { 744d4002b98SHong Zhang ierr = MPIU_Allreduce(isend,irecv,5,MPIU_REAL,MPIU_SUM,PetscObjectComm((PetscObject)matin));CHKERRQ(ierr); 745d4002b98SHong Zhang 746d4002b98SHong Zhang info->nz_used = irecv[0]; 747d4002b98SHong Zhang info->nz_allocated = irecv[1]; 748d4002b98SHong Zhang info->nz_unneeded = irecv[2]; 749d4002b98SHong Zhang info->memory = irecv[3]; 750d4002b98SHong Zhang info->mallocs = irecv[4]; 751d4002b98SHong Zhang } 752d4002b98SHong Zhang info->fill_ratio_given = 0; /* no parallel LU/ILU/Cholesky */ 753d4002b98SHong Zhang info->fill_ratio_needed = 0; 754d4002b98SHong Zhang info->factor_mallocs = 0; 755d4002b98SHong Zhang PetscFunctionReturn(0); 756d4002b98SHong Zhang } 757d4002b98SHong Zhang 758d4002b98SHong Zhang PetscErrorCode MatSetOption_MPISELL(Mat A,MatOption op,PetscBool flg) 759d4002b98SHong Zhang { 760d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 761d4002b98SHong Zhang PetscErrorCode ierr; 762d4002b98SHong Zhang 763d4002b98SHong Zhang PetscFunctionBegin; 764d4002b98SHong Zhang switch (op) { 765d4002b98SHong Zhang case MAT_NEW_NONZERO_LOCATIONS: 766d4002b98SHong Zhang case MAT_NEW_NONZERO_ALLOCATION_ERR: 767d4002b98SHong Zhang case MAT_UNUSED_NONZERO_LOCATION_ERR: 768d4002b98SHong Zhang case MAT_KEEP_NONZERO_PATTERN: 769d4002b98SHong Zhang case MAT_NEW_NONZERO_LOCATION_ERR: 770d4002b98SHong Zhang case MAT_USE_INODES: 771d4002b98SHong Zhang case MAT_IGNORE_ZERO_ENTRIES: 772d4002b98SHong Zhang MatCheckPreallocated(A,1); 773d4002b98SHong Zhang ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr); 774d4002b98SHong Zhang ierr = MatSetOption(a->B,op,flg);CHKERRQ(ierr); 775d4002b98SHong Zhang break; 776d4002b98SHong Zhang case MAT_ROW_ORIENTED: 777d4002b98SHong Zhang MatCheckPreallocated(A,1); 778d4002b98SHong Zhang a->roworiented = flg; 779d4002b98SHong Zhang 780d4002b98SHong Zhang ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr); 781d4002b98SHong Zhang ierr = MatSetOption(a->B,op,flg);CHKERRQ(ierr); 782d4002b98SHong Zhang break; 783d4002b98SHong Zhang case MAT_NEW_DIAGONALS: 784d4002b98SHong Zhang ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 785d4002b98SHong Zhang break; 786d4002b98SHong Zhang case MAT_IGNORE_OFF_PROC_ENTRIES: 787d4002b98SHong Zhang a->donotstash = flg; 788d4002b98SHong Zhang break; 789d4002b98SHong Zhang case MAT_SPD: 790d4002b98SHong Zhang A->spd_set = PETSC_TRUE; 791d4002b98SHong Zhang A->spd = flg; 792d4002b98SHong Zhang if (flg) { 793d4002b98SHong Zhang A->symmetric = PETSC_TRUE; 794d4002b98SHong Zhang A->structurally_symmetric = PETSC_TRUE; 795d4002b98SHong Zhang A->symmetric_set = PETSC_TRUE; 796d4002b98SHong Zhang A->structurally_symmetric_set = PETSC_TRUE; 797d4002b98SHong Zhang } 798d4002b98SHong Zhang break; 799d4002b98SHong Zhang case MAT_SYMMETRIC: 800d4002b98SHong Zhang MatCheckPreallocated(A,1); 801d4002b98SHong Zhang ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr); 802d4002b98SHong Zhang break; 803d4002b98SHong Zhang case MAT_STRUCTURALLY_SYMMETRIC: 804d4002b98SHong Zhang MatCheckPreallocated(A,1); 805d4002b98SHong Zhang ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr); 806d4002b98SHong Zhang break; 807d4002b98SHong Zhang case MAT_HERMITIAN: 808d4002b98SHong Zhang MatCheckPreallocated(A,1); 809d4002b98SHong Zhang ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr); 810d4002b98SHong Zhang break; 811d4002b98SHong Zhang case MAT_SYMMETRY_ETERNAL: 812d4002b98SHong Zhang MatCheckPreallocated(A,1); 813d4002b98SHong Zhang ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr); 814d4002b98SHong Zhang break; 815d4002b98SHong Zhang default: 816d4002b98SHong Zhang SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op); 817d4002b98SHong Zhang } 818d4002b98SHong Zhang PetscFunctionReturn(0); 819d4002b98SHong Zhang } 820d4002b98SHong Zhang 821d4002b98SHong Zhang 822d4002b98SHong Zhang PetscErrorCode MatDiagonalScale_MPISELL(Mat mat,Vec ll,Vec rr) 823d4002b98SHong Zhang { 824d4002b98SHong Zhang Mat_MPISELL *sell=(Mat_MPISELL*)mat->data; 825d4002b98SHong Zhang Mat a=sell->A,b=sell->B; 826d4002b98SHong Zhang PetscErrorCode ierr; 827d4002b98SHong Zhang PetscInt s1,s2,s3; 828d4002b98SHong Zhang 829d4002b98SHong Zhang PetscFunctionBegin; 830d4002b98SHong Zhang ierr = MatGetLocalSize(mat,&s2,&s3);CHKERRQ(ierr); 831d4002b98SHong Zhang if (rr) { 832d4002b98SHong Zhang ierr = VecGetLocalSize(rr,&s1);CHKERRQ(ierr); 833d4002b98SHong Zhang if (s1!=s3) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"right vector non-conforming local size"); 834d4002b98SHong Zhang /* Overlap communication with computation. */ 835d4002b98SHong Zhang ierr = VecScatterBegin(sell->Mvctx,rr,sell->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 836d4002b98SHong Zhang } 837d4002b98SHong Zhang if (ll) { 838d4002b98SHong Zhang ierr = VecGetLocalSize(ll,&s1);CHKERRQ(ierr); 839d4002b98SHong Zhang if (s1!=s2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"left vector non-conforming local size"); 840d4002b98SHong Zhang ierr = (*b->ops->diagonalscale)(b,ll,0);CHKERRQ(ierr); 841d4002b98SHong Zhang } 842d4002b98SHong Zhang /* scale the diagonal block */ 843d4002b98SHong Zhang ierr = (*a->ops->diagonalscale)(a,ll,rr);CHKERRQ(ierr); 844d4002b98SHong Zhang 845d4002b98SHong Zhang if (rr) { 846d4002b98SHong Zhang /* Do a scatter end and then right scale the off-diagonal block */ 847d4002b98SHong Zhang ierr = VecScatterEnd(sell->Mvctx,rr,sell->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 848d4002b98SHong Zhang ierr = (*b->ops->diagonalscale)(b,0,sell->lvec);CHKERRQ(ierr); 849d4002b98SHong Zhang } 850d4002b98SHong Zhang PetscFunctionReturn(0); 851d4002b98SHong Zhang } 852d4002b98SHong Zhang 853d4002b98SHong Zhang PetscErrorCode MatSetUnfactored_MPISELL(Mat A) 854d4002b98SHong Zhang { 855d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 856d4002b98SHong Zhang PetscErrorCode ierr; 857d4002b98SHong Zhang 858d4002b98SHong Zhang PetscFunctionBegin; 859d4002b98SHong Zhang ierr = MatSetUnfactored(a->A);CHKERRQ(ierr); 860d4002b98SHong Zhang PetscFunctionReturn(0); 861d4002b98SHong Zhang } 862d4002b98SHong Zhang 863d4002b98SHong Zhang PetscErrorCode MatEqual_MPISELL(Mat A,Mat B,PetscBool *flag) 864d4002b98SHong Zhang { 865d4002b98SHong Zhang Mat_MPISELL *matB=(Mat_MPISELL*)B->data,*matA=(Mat_MPISELL*)A->data; 866d4002b98SHong Zhang Mat a,b,c,d; 867d4002b98SHong Zhang PetscBool flg; 868d4002b98SHong Zhang PetscErrorCode ierr; 869d4002b98SHong Zhang 870d4002b98SHong Zhang PetscFunctionBegin; 871d4002b98SHong Zhang a = matA->A; b = matA->B; 872d4002b98SHong Zhang c = matB->A; d = matB->B; 873d4002b98SHong Zhang 874d4002b98SHong Zhang ierr = MatEqual(a,c,&flg);CHKERRQ(ierr); 875d4002b98SHong Zhang if (flg) { 876d4002b98SHong Zhang ierr = MatEqual(b,d,&flg);CHKERRQ(ierr); 877d4002b98SHong Zhang } 878d4002b98SHong Zhang ierr = MPIU_Allreduce(&flg,flag,1,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)A));CHKERRQ(ierr); 879d4002b98SHong Zhang PetscFunctionReturn(0); 880d4002b98SHong Zhang } 881d4002b98SHong Zhang 882d4002b98SHong Zhang PetscErrorCode MatCopy_MPISELL(Mat A,Mat B,MatStructure str) 883d4002b98SHong Zhang { 884d4002b98SHong Zhang PetscErrorCode ierr; 885d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 886d4002b98SHong Zhang Mat_MPISELL *b=(Mat_MPISELL*)B->data; 887d4002b98SHong Zhang 888d4002b98SHong Zhang PetscFunctionBegin; 889d4002b98SHong Zhang /* If the two matrices don't have the same copy implementation, they aren't compatible for fast copy. */ 890d4002b98SHong Zhang if ((str != SAME_NONZERO_PATTERN) || (A->ops->copy != B->ops->copy)) { 891d4002b98SHong Zhang /* because of the column compression in the off-processor part of the matrix a->B, 892d4002b98SHong Zhang the number of columns in a->B and b->B may be different, hence we cannot call 893d4002b98SHong Zhang the MatCopy() directly on the two parts. If need be, we can provide a more 894d4002b98SHong Zhang efficient copy than the MatCopy_Basic() by first uncompressing the a->B matrices 895d4002b98SHong Zhang then copying the submatrices */ 896d4002b98SHong Zhang ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 897d4002b98SHong Zhang } else { 898d4002b98SHong Zhang ierr = MatCopy(a->A,b->A,str);CHKERRQ(ierr); 899d4002b98SHong Zhang ierr = MatCopy(a->B,b->B,str);CHKERRQ(ierr); 900d4002b98SHong Zhang } 901d4002b98SHong Zhang PetscFunctionReturn(0); 902d4002b98SHong Zhang } 903d4002b98SHong Zhang 904d4002b98SHong Zhang PetscErrorCode MatSetUp_MPISELL(Mat A) 905d4002b98SHong Zhang { 906d4002b98SHong Zhang PetscErrorCode ierr; 907d4002b98SHong Zhang 908d4002b98SHong Zhang PetscFunctionBegin; 909d4002b98SHong Zhang ierr = MatMPISELLSetPreallocation(A,PETSC_DEFAULT,0,PETSC_DEFAULT,0);CHKERRQ(ierr); 910d4002b98SHong Zhang PetscFunctionReturn(0); 911d4002b98SHong Zhang } 912d4002b98SHong Zhang 913d4002b98SHong Zhang 914d4002b98SHong Zhang extern PetscErrorCode MatConjugate_SeqSELL(Mat); 915d4002b98SHong Zhang 916d4002b98SHong Zhang PetscErrorCode MatConjugate_MPISELL(Mat mat) 917d4002b98SHong Zhang { 918d4002b98SHong Zhang #if defined(PETSC_USE_COMPLEX) 919d4002b98SHong Zhang PetscErrorCode ierr; 920d4002b98SHong Zhang Mat_MPISELL *sell=(Mat_MPISELL*)mat->data; 921d4002b98SHong Zhang 922d4002b98SHong Zhang PetscFunctionBegin; 923d4002b98SHong Zhang ierr = MatConjugate_SeqSELL(sell->A);CHKERRQ(ierr); 924d4002b98SHong Zhang ierr = MatConjugate_SeqSELL(sell->B);CHKERRQ(ierr); 925d4002b98SHong Zhang #else 926d4002b98SHong Zhang PetscFunctionBegin; 927d4002b98SHong Zhang #endif 928d4002b98SHong Zhang PetscFunctionReturn(0); 929d4002b98SHong Zhang } 930d4002b98SHong Zhang 931d4002b98SHong Zhang PetscErrorCode MatRealPart_MPISELL(Mat A) 932d4002b98SHong Zhang { 933d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 934d4002b98SHong Zhang PetscErrorCode ierr; 935d4002b98SHong Zhang 936d4002b98SHong Zhang PetscFunctionBegin; 937d4002b98SHong Zhang ierr = MatRealPart(a->A);CHKERRQ(ierr); 938d4002b98SHong Zhang ierr = MatRealPart(a->B);CHKERRQ(ierr); 939d4002b98SHong Zhang PetscFunctionReturn(0); 940d4002b98SHong Zhang } 941d4002b98SHong Zhang 942d4002b98SHong Zhang PetscErrorCode MatImaginaryPart_MPISELL(Mat A) 943d4002b98SHong Zhang { 944d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 945d4002b98SHong Zhang PetscErrorCode ierr; 946d4002b98SHong Zhang 947d4002b98SHong Zhang PetscFunctionBegin; 948d4002b98SHong Zhang ierr = MatImaginaryPart(a->A);CHKERRQ(ierr); 949d4002b98SHong Zhang ierr = MatImaginaryPart(a->B);CHKERRQ(ierr); 950d4002b98SHong Zhang PetscFunctionReturn(0); 951d4002b98SHong Zhang } 952d4002b98SHong Zhang 953d4002b98SHong Zhang PetscErrorCode MatInvertBlockDiagonal_MPISELL(Mat A,const PetscScalar **values) 954d4002b98SHong Zhang { 955d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 956d4002b98SHong Zhang PetscErrorCode ierr; 957d4002b98SHong Zhang 958d4002b98SHong Zhang PetscFunctionBegin; 959d4002b98SHong Zhang ierr = MatInvertBlockDiagonal(a->A,values);CHKERRQ(ierr); 960d4002b98SHong Zhang A->factorerrortype = a->A->factorerrortype; 961d4002b98SHong Zhang PetscFunctionReturn(0); 962d4002b98SHong Zhang } 963d4002b98SHong Zhang 964d4002b98SHong Zhang static PetscErrorCode MatSetRandom_MPISELL(Mat x,PetscRandom rctx) 965d4002b98SHong Zhang { 966d4002b98SHong Zhang PetscErrorCode ierr; 967d4002b98SHong Zhang Mat_MPISELL *sell=(Mat_MPISELL*)x->data; 968d4002b98SHong Zhang 969d4002b98SHong Zhang PetscFunctionBegin; 970d4002b98SHong Zhang ierr = MatSetRandom(sell->A,rctx);CHKERRQ(ierr); 971d4002b98SHong Zhang ierr = MatSetRandom(sell->B,rctx);CHKERRQ(ierr); 972d4002b98SHong Zhang ierr = MatAssemblyBegin(x,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 973d4002b98SHong Zhang ierr = MatAssemblyEnd(x,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 974d4002b98SHong Zhang PetscFunctionReturn(0); 975d4002b98SHong Zhang } 976d4002b98SHong Zhang 977d4002b98SHong Zhang PetscErrorCode MatSetFromOptions_MPISELL(PetscOptionItems *PetscOptionsObject,Mat A) 978d4002b98SHong Zhang { 979d4002b98SHong Zhang PetscErrorCode ierr; 980d4002b98SHong Zhang 981d4002b98SHong Zhang PetscFunctionBegin; 982d4002b98SHong Zhang ierr = PetscOptionsHead(PetscOptionsObject,"MPISELL options");CHKERRQ(ierr); 983d4002b98SHong Zhang ierr = PetscObjectOptionsBegin((PetscObject)A); 984d4002b98SHong Zhang ierr = PetscOptionsEnd();CHKERRQ(ierr); 985d4002b98SHong Zhang PetscFunctionReturn(0); 986d4002b98SHong Zhang } 987d4002b98SHong Zhang 988d4002b98SHong Zhang PetscErrorCode MatShift_MPISELL(Mat Y,PetscScalar a) 989d4002b98SHong Zhang { 990d4002b98SHong Zhang PetscErrorCode ierr; 991d4002b98SHong Zhang Mat_MPISELL *msell=(Mat_MPISELL*)Y->data; 992d4002b98SHong Zhang Mat_SeqSELL *sell=(Mat_SeqSELL*)msell->A->data; 993d4002b98SHong Zhang 994d4002b98SHong Zhang PetscFunctionBegin; 995d4002b98SHong Zhang if (!Y->preallocated) { 996d4002b98SHong Zhang ierr = MatMPISELLSetPreallocation(Y,1,NULL,0,NULL);CHKERRQ(ierr); 997d4002b98SHong Zhang } else if (!sell->nz) { 998d4002b98SHong Zhang PetscInt nonew = sell->nonew; 999d4002b98SHong Zhang ierr = MatSeqSELLSetPreallocation(msell->A,1,NULL);CHKERRQ(ierr); 1000d4002b98SHong Zhang sell->nonew = nonew; 1001d4002b98SHong Zhang } 1002d4002b98SHong Zhang ierr = MatShift_Basic(Y,a);CHKERRQ(ierr); 1003d4002b98SHong Zhang PetscFunctionReturn(0); 1004d4002b98SHong Zhang } 1005d4002b98SHong Zhang 1006d4002b98SHong Zhang PetscErrorCode MatMissingDiagonal_MPISELL(Mat A,PetscBool *missing,PetscInt *d) 1007d4002b98SHong Zhang { 1008d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 1009d4002b98SHong Zhang PetscErrorCode ierr; 1010d4002b98SHong Zhang 1011d4002b98SHong Zhang PetscFunctionBegin; 1012d4002b98SHong Zhang if (A->rmap->n != A->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only works for square matrices"); 1013d4002b98SHong Zhang ierr = MatMissingDiagonal(a->A,missing,d);CHKERRQ(ierr); 1014d4002b98SHong Zhang if (d) { 1015d4002b98SHong Zhang PetscInt rstart; 1016d4002b98SHong Zhang ierr = MatGetOwnershipRange(A,&rstart,NULL);CHKERRQ(ierr); 1017d4002b98SHong Zhang *d += rstart; 1018d4002b98SHong Zhang 1019d4002b98SHong Zhang } 1020d4002b98SHong Zhang PetscFunctionReturn(0); 1021d4002b98SHong Zhang } 1022d4002b98SHong Zhang 1023d4002b98SHong Zhang PetscErrorCode MatGetDiagonalBlock_MPISELL(Mat A,Mat *a) 1024d4002b98SHong Zhang { 1025d4002b98SHong Zhang PetscFunctionBegin; 1026d4002b98SHong Zhang *a = ((Mat_MPISELL*)A->data)->A; 1027d4002b98SHong Zhang PetscFunctionReturn(0); 1028d4002b98SHong Zhang } 1029d4002b98SHong Zhang 1030d4002b98SHong Zhang /* -------------------------------------------------------------------*/ 1031d4002b98SHong Zhang static struct _MatOps MatOps_Values = {MatSetValues_MPISELL, 1032d4002b98SHong Zhang 0, 1033d4002b98SHong Zhang 0, 1034d4002b98SHong Zhang MatMult_MPISELL, 1035d4002b98SHong Zhang /* 4*/ MatMultAdd_MPISELL, 1036d4002b98SHong Zhang MatMultTranspose_MPISELL, 1037d4002b98SHong Zhang MatMultTransposeAdd_MPISELL, 1038d4002b98SHong Zhang 0, 1039d4002b98SHong Zhang 0, 1040d4002b98SHong Zhang 0, 1041d4002b98SHong Zhang /*10*/ 0, 1042d4002b98SHong Zhang 0, 1043d4002b98SHong Zhang 0, 1044d4002b98SHong Zhang MatSOR_MPISELL, 1045d4002b98SHong Zhang 0, 1046d4002b98SHong Zhang /*15*/ MatGetInfo_MPISELL, 1047d4002b98SHong Zhang MatEqual_MPISELL, 1048d4002b98SHong Zhang MatGetDiagonal_MPISELL, 1049d4002b98SHong Zhang MatDiagonalScale_MPISELL, 1050d4002b98SHong Zhang 0, 1051d4002b98SHong Zhang /*20*/ MatAssemblyBegin_MPISELL, 1052d4002b98SHong Zhang MatAssemblyEnd_MPISELL, 1053d4002b98SHong Zhang MatSetOption_MPISELL, 1054d4002b98SHong Zhang MatZeroEntries_MPISELL, 1055d4002b98SHong Zhang /*24*/ 0, 1056d4002b98SHong Zhang 0, 1057d4002b98SHong Zhang 0, 1058d4002b98SHong Zhang 0, 1059d4002b98SHong Zhang 0, 1060d4002b98SHong Zhang /*29*/ MatSetUp_MPISELL, 1061d4002b98SHong Zhang 0, 1062d4002b98SHong Zhang 0, 1063d4002b98SHong Zhang MatGetDiagonalBlock_MPISELL, 1064d4002b98SHong Zhang 0, 1065d4002b98SHong Zhang /*34*/ MatDuplicate_MPISELL, 1066d4002b98SHong Zhang 0, 1067d4002b98SHong Zhang 0, 1068d4002b98SHong Zhang 0, 1069d4002b98SHong Zhang 0, 1070d4002b98SHong Zhang /*39*/ 0, 1071d4002b98SHong Zhang 0, 1072d4002b98SHong Zhang 0, 1073d4002b98SHong Zhang MatGetValues_MPISELL, 1074d4002b98SHong Zhang MatCopy_MPISELL, 1075d4002b98SHong Zhang /*44*/ 0, 1076d4002b98SHong Zhang MatScale_MPISELL, 1077d4002b98SHong Zhang MatShift_MPISELL, 1078d4002b98SHong Zhang MatDiagonalSet_MPISELL, 1079d4002b98SHong Zhang 0, 1080d4002b98SHong Zhang /*49*/ MatSetRandom_MPISELL, 1081d4002b98SHong Zhang 0, 1082d4002b98SHong Zhang 0, 1083d4002b98SHong Zhang 0, 1084d4002b98SHong Zhang 0, 1085d4002b98SHong Zhang /*54*/ MatFDColoringCreate_MPIXAIJ, 1086d4002b98SHong Zhang 0, 1087d4002b98SHong Zhang MatSetUnfactored_MPISELL, 1088d4002b98SHong Zhang 0, 1089d4002b98SHong Zhang 0, 1090d4002b98SHong Zhang /*59*/ 0, 1091d4002b98SHong Zhang MatDestroy_MPISELL, 1092d4002b98SHong Zhang MatView_MPISELL, 1093d4002b98SHong Zhang 0, 1094d4002b98SHong Zhang 0, 1095d4002b98SHong Zhang /*64*/ 0, 1096d4002b98SHong Zhang 0, 1097d4002b98SHong Zhang 0, 1098d4002b98SHong Zhang 0, 1099d4002b98SHong Zhang 0, 1100d4002b98SHong Zhang /*69*/ 0, 1101d4002b98SHong Zhang 0, 1102d4002b98SHong Zhang 0, 1103d4002b98SHong Zhang 0, 1104d4002b98SHong Zhang 0, 1105d4002b98SHong Zhang 0, 1106d4002b98SHong Zhang /*75*/ MatFDColoringApply_AIJ, /* reuse AIJ function */ 1107d4002b98SHong Zhang MatSetFromOptions_MPISELL, 1108d4002b98SHong Zhang 0, 1109d4002b98SHong Zhang 0, 1110d4002b98SHong Zhang 0, 1111d4002b98SHong Zhang /*80*/ 0, 1112d4002b98SHong Zhang 0, 1113d4002b98SHong Zhang 0, 1114d4002b98SHong Zhang /*83*/ 0, 1115d4002b98SHong Zhang 0, 1116d4002b98SHong Zhang 0, 1117d4002b98SHong Zhang 0, 1118d4002b98SHong Zhang 0, 1119d4002b98SHong Zhang 0, 1120d4002b98SHong Zhang /*89*/ 0, 1121d4002b98SHong Zhang 0, 1122d4002b98SHong Zhang 0, 1123d4002b98SHong Zhang 0, 1124d4002b98SHong Zhang 0, 1125d4002b98SHong Zhang /*94*/ 0, 1126d4002b98SHong Zhang 0, 1127d4002b98SHong Zhang 0, 1128d4002b98SHong Zhang 0, 1129d4002b98SHong Zhang 0, 1130d4002b98SHong Zhang /*99*/ 0, 1131d4002b98SHong Zhang 0, 1132d4002b98SHong Zhang 0, 1133d4002b98SHong Zhang MatConjugate_MPISELL, 1134d4002b98SHong Zhang 0, 1135d4002b98SHong Zhang /*104*/0, 1136d4002b98SHong Zhang MatRealPart_MPISELL, 1137d4002b98SHong Zhang MatImaginaryPart_MPISELL, 1138d4002b98SHong Zhang 0, 1139d4002b98SHong Zhang 0, 1140d4002b98SHong Zhang /*109*/0, 1141d4002b98SHong Zhang 0, 1142d4002b98SHong Zhang 0, 1143d4002b98SHong Zhang 0, 1144d4002b98SHong Zhang MatMissingDiagonal_MPISELL, 1145d4002b98SHong Zhang /*114*/0, 1146d4002b98SHong Zhang 0, 1147d4002b98SHong Zhang MatGetGhosts_MPISELL, 1148d4002b98SHong Zhang 0, 1149d4002b98SHong Zhang 0, 1150d4002b98SHong Zhang /*119*/0, 1151d4002b98SHong Zhang 0, 1152d4002b98SHong Zhang 0, 1153d4002b98SHong Zhang 0, 1154d4002b98SHong Zhang 0, 1155d4002b98SHong Zhang /*124*/0, 1156d4002b98SHong Zhang 0, 1157d4002b98SHong Zhang MatInvertBlockDiagonal_MPISELL, 1158d4002b98SHong Zhang 0, 1159d4002b98SHong Zhang 0, 1160d4002b98SHong Zhang /*129*/0, 1161d4002b98SHong Zhang 0, 1162d4002b98SHong Zhang 0, 1163d4002b98SHong Zhang 0, 1164d4002b98SHong Zhang 0, 1165d4002b98SHong Zhang /*134*/0, 1166d4002b98SHong Zhang 0, 1167d4002b98SHong Zhang 0, 1168d4002b98SHong Zhang 0, 1169d4002b98SHong Zhang 0, 1170d4002b98SHong Zhang /*139*/0, 1171d4002b98SHong Zhang 0, 1172d4002b98SHong Zhang 0, 1173d4002b98SHong Zhang MatFDColoringSetUp_MPIXAIJ, 1174d4002b98SHong Zhang 0, 1175d4002b98SHong Zhang /*144*/0 1176d4002b98SHong Zhang }; 1177d4002b98SHong Zhang 1178d4002b98SHong Zhang /* ----------------------------------------------------------------------------------------*/ 1179d4002b98SHong Zhang 1180d4002b98SHong Zhang PetscErrorCode MatStoreValues_MPISELL(Mat mat) 1181d4002b98SHong Zhang { 1182d4002b98SHong Zhang Mat_MPISELL *sell=(Mat_MPISELL*)mat->data; 1183d4002b98SHong Zhang PetscErrorCode ierr; 1184d4002b98SHong Zhang 1185d4002b98SHong Zhang PetscFunctionBegin; 1186d4002b98SHong Zhang ierr = MatStoreValues(sell->A);CHKERRQ(ierr); 1187d4002b98SHong Zhang ierr = MatStoreValues(sell->B);CHKERRQ(ierr); 1188d4002b98SHong Zhang PetscFunctionReturn(0); 1189d4002b98SHong Zhang } 1190d4002b98SHong Zhang 1191d4002b98SHong Zhang PetscErrorCode MatRetrieveValues_MPISELL(Mat mat) 1192d4002b98SHong Zhang { 1193d4002b98SHong Zhang Mat_MPISELL *sell=(Mat_MPISELL*)mat->data; 1194d4002b98SHong Zhang PetscErrorCode ierr; 1195d4002b98SHong Zhang 1196d4002b98SHong Zhang PetscFunctionBegin; 1197d4002b98SHong Zhang ierr = MatRetrieveValues(sell->A);CHKERRQ(ierr); 1198d4002b98SHong Zhang ierr = MatRetrieveValues(sell->B);CHKERRQ(ierr); 1199d4002b98SHong Zhang PetscFunctionReturn(0); 1200d4002b98SHong Zhang } 1201d4002b98SHong Zhang 1202d4002b98SHong Zhang PetscErrorCode MatMPISELLSetPreallocation_MPISELL(Mat B,PetscInt d_rlenmax,const PetscInt d_rlen[],PetscInt o_rlenmax,const PetscInt o_rlen[]) 1203d4002b98SHong Zhang { 1204d4002b98SHong Zhang Mat_MPISELL *b; 1205d4002b98SHong Zhang PetscErrorCode ierr; 1206d4002b98SHong Zhang 1207d4002b98SHong Zhang PetscFunctionBegin; 1208d4002b98SHong Zhang ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 1209d4002b98SHong Zhang ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 1210d4002b98SHong Zhang b = (Mat_MPISELL*)B->data; 1211d4002b98SHong Zhang 1212d4002b98SHong Zhang if (!B->preallocated) { 1213d4002b98SHong Zhang /* Explicitly create 2 MATSEQSELL matrices. */ 1214d4002b98SHong Zhang ierr = MatCreate(PETSC_COMM_SELF,&b->A);CHKERRQ(ierr); 1215d4002b98SHong Zhang ierr = MatSetSizes(b->A,B->rmap->n,B->cmap->n,B->rmap->n,B->cmap->n);CHKERRQ(ierr); 1216d4002b98SHong Zhang ierr = MatSetBlockSizesFromMats(b->A,B,B);CHKERRQ(ierr); 1217d4002b98SHong Zhang ierr = MatSetType(b->A,MATSEQSELL);CHKERRQ(ierr); 1218d4002b98SHong Zhang ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)b->A);CHKERRQ(ierr); 1219d4002b98SHong Zhang ierr = MatCreate(PETSC_COMM_SELF,&b->B);CHKERRQ(ierr); 1220d4002b98SHong Zhang ierr = MatSetSizes(b->B,B->rmap->n,B->cmap->N,B->rmap->n,B->cmap->N);CHKERRQ(ierr); 1221d4002b98SHong Zhang ierr = MatSetBlockSizesFromMats(b->B,B,B);CHKERRQ(ierr); 1222d4002b98SHong Zhang ierr = MatSetType(b->B,MATSEQSELL);CHKERRQ(ierr); 1223d4002b98SHong Zhang ierr = PetscLogObjectParent((PetscObject)B,(PetscObject)b->B);CHKERRQ(ierr); 1224d4002b98SHong Zhang } 1225d4002b98SHong Zhang 1226d4002b98SHong Zhang ierr = MatSeqSELLSetPreallocation(b->A,d_rlenmax,d_rlen);CHKERRQ(ierr); 1227d4002b98SHong Zhang ierr = MatSeqSELLSetPreallocation(b->B,o_rlenmax,o_rlen);CHKERRQ(ierr); 1228d4002b98SHong Zhang B->preallocated = PETSC_TRUE; 1229d4002b98SHong Zhang B->was_assembled = PETSC_FALSE; 1230d4002b98SHong Zhang /* 1231d4002b98SHong Zhang critical for MatAssemblyEnd to work. 1232d4002b98SHong Zhang MatAssemblyBegin checks it to set up was_assembled 1233d4002b98SHong Zhang and MatAssemblyEnd checks was_assembled to determine whether to build garray 1234d4002b98SHong Zhang */ 1235d4002b98SHong Zhang B->assembled = PETSC_FALSE; 1236d4002b98SHong Zhang PetscFunctionReturn(0); 1237d4002b98SHong Zhang } 1238d4002b98SHong Zhang 1239d4002b98SHong Zhang PetscErrorCode MatDuplicate_MPISELL(Mat matin,MatDuplicateOption cpvalues,Mat *newmat) 1240d4002b98SHong Zhang { 1241d4002b98SHong Zhang Mat mat; 1242d4002b98SHong Zhang Mat_MPISELL *a,*oldmat=(Mat_MPISELL*)matin->data; 1243d4002b98SHong Zhang PetscErrorCode ierr; 1244d4002b98SHong Zhang 1245d4002b98SHong Zhang PetscFunctionBegin; 1246d4002b98SHong Zhang *newmat = 0; 1247d4002b98SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)matin),&mat);CHKERRQ(ierr); 1248d4002b98SHong Zhang ierr = MatSetSizes(mat,matin->rmap->n,matin->cmap->n,matin->rmap->N,matin->cmap->N);CHKERRQ(ierr); 1249d4002b98SHong Zhang ierr = MatSetBlockSizesFromMats(mat,matin,matin);CHKERRQ(ierr); 1250d4002b98SHong Zhang ierr = MatSetType(mat,((PetscObject)matin)->type_name);CHKERRQ(ierr); 1251d4002b98SHong Zhang ierr = PetscMemcpy(mat->ops,matin->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 1252d4002b98SHong Zhang a = (Mat_MPISELL*)mat->data; 1253d4002b98SHong Zhang 1254d4002b98SHong Zhang mat->factortype = matin->factortype; 1255d4002b98SHong Zhang mat->assembled = PETSC_TRUE; 1256d4002b98SHong Zhang mat->insertmode = NOT_SET_VALUES; 1257d4002b98SHong Zhang mat->preallocated = PETSC_TRUE; 1258d4002b98SHong Zhang 1259d4002b98SHong Zhang a->size = oldmat->size; 1260d4002b98SHong Zhang a->rank = oldmat->rank; 1261d4002b98SHong Zhang a->donotstash = oldmat->donotstash; 1262d4002b98SHong Zhang a->roworiented = oldmat->roworiented; 1263d4002b98SHong Zhang a->rowindices = 0; 1264d4002b98SHong Zhang a->rowvalues = 0; 1265d4002b98SHong Zhang a->getrowactive = PETSC_FALSE; 1266d4002b98SHong Zhang 1267d4002b98SHong Zhang ierr = PetscLayoutReference(matin->rmap,&mat->rmap);CHKERRQ(ierr); 1268d4002b98SHong Zhang ierr = PetscLayoutReference(matin->cmap,&mat->cmap);CHKERRQ(ierr); 1269d4002b98SHong Zhang 1270d4002b98SHong Zhang if (oldmat->colmap) { 1271d4002b98SHong Zhang #if defined(PETSC_USE_CTABLE) 1272d4002b98SHong Zhang ierr = PetscTableCreateCopy(oldmat->colmap,&a->colmap);CHKERRQ(ierr); 1273d4002b98SHong Zhang #else 1274d4002b98SHong Zhang ierr = PetscMalloc1(mat->cmap->N,&a->colmap);CHKERRQ(ierr); 1275d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)mat,(mat->cmap->N)*sizeof(PetscInt));CHKERRQ(ierr); 1276d4002b98SHong Zhang ierr = PetscMemcpy(a->colmap,oldmat->colmap,(mat->cmap->N)*sizeof(PetscInt));CHKERRQ(ierr); 1277d4002b98SHong Zhang #endif 1278d4002b98SHong Zhang } else a->colmap = 0; 1279d4002b98SHong Zhang if (oldmat->garray) { 1280d4002b98SHong Zhang PetscInt len; 1281d4002b98SHong Zhang len = oldmat->B->cmap->n; 1282d4002b98SHong Zhang ierr = PetscMalloc1(len+1,&a->garray);CHKERRQ(ierr); 1283d4002b98SHong Zhang ierr = PetscLogObjectMemory((PetscObject)mat,len*sizeof(PetscInt));CHKERRQ(ierr); 1284d4002b98SHong Zhang if (len) { ierr = PetscMemcpy(a->garray,oldmat->garray,len*sizeof(PetscInt));CHKERRQ(ierr); } 1285d4002b98SHong Zhang } else a->garray = 0; 1286d4002b98SHong Zhang 1287d4002b98SHong Zhang ierr = VecDuplicate(oldmat->lvec,&a->lvec);CHKERRQ(ierr); 1288d4002b98SHong Zhang ierr = PetscLogObjectParent((PetscObject)mat,(PetscObject)a->lvec);CHKERRQ(ierr); 1289d4002b98SHong Zhang ierr = VecScatterCopy(oldmat->Mvctx,&a->Mvctx);CHKERRQ(ierr); 1290d4002b98SHong Zhang ierr = PetscLogObjectParent((PetscObject)mat,(PetscObject)a->Mvctx);CHKERRQ(ierr); 1291d4002b98SHong Zhang ierr = MatDuplicate(oldmat->A,cpvalues,&a->A);CHKERRQ(ierr); 1292d4002b98SHong Zhang ierr = PetscLogObjectParent((PetscObject)mat,(PetscObject)a->A);CHKERRQ(ierr); 1293d4002b98SHong Zhang ierr = MatDuplicate(oldmat->B,cpvalues,&a->B);CHKERRQ(ierr); 1294d4002b98SHong Zhang ierr = PetscLogObjectParent((PetscObject)mat,(PetscObject)a->B);CHKERRQ(ierr); 1295d4002b98SHong Zhang ierr = PetscFunctionListDuplicate(((PetscObject)matin)->qlist,&((PetscObject)mat)->qlist);CHKERRQ(ierr); 1296d4002b98SHong Zhang *newmat = mat; 1297d4002b98SHong Zhang PetscFunctionReturn(0); 1298d4002b98SHong Zhang } 1299d4002b98SHong Zhang 1300d4002b98SHong Zhang /*@C 1301d4002b98SHong Zhang MatMPISELLSetPreallocation - Preallocates memory for a sparse parallel matrix in sell format. 1302d4002b98SHong Zhang For good matrix assembly performance the user should preallocate the matrix storage by 1303d4002b98SHong Zhang setting the parameters d_nz (or d_nnz) and o_nz (or o_nnz). 1304d4002b98SHong Zhang 1305d4002b98SHong Zhang Collective on MPI_Comm 1306d4002b98SHong Zhang 1307d4002b98SHong Zhang Input Parameters: 1308d4002b98SHong Zhang + B - the matrix 1309d4002b98SHong Zhang . d_nz - number of nonzeros per row in DIAGONAL portion of local submatrix 1310d4002b98SHong Zhang (same value is used for all local rows) 1311d4002b98SHong Zhang . d_nnz - array containing the number of nonzeros in the various rows of the 1312d4002b98SHong Zhang DIAGONAL portion of the local submatrix (possibly different for each row) 1313d4002b98SHong Zhang or NULL (PETSC_NULL_INTEGER in Fortran), if d_nz is used to specify the nonzero structure. 1314d4002b98SHong Zhang The size of this array is equal to the number of local rows, i.e 'm'. 1315d4002b98SHong Zhang For matrices that will be factored, you must leave room for (and set) 1316d4002b98SHong Zhang the diagonal entry even if it is zero. 1317d4002b98SHong Zhang . o_nz - number of nonzeros per row in the OFF-DIAGONAL portion of local 1318d4002b98SHong Zhang submatrix (same value is used for all local rows). 1319d4002b98SHong Zhang - o_nnz - array containing the number of nonzeros in the various rows of the 1320d4002b98SHong Zhang OFF-DIAGONAL portion of the local submatrix (possibly different for 1321d4002b98SHong Zhang each row) or NULL (PETSC_NULL_INTEGER in Fortran), if o_nz is used to specify the nonzero 1322d4002b98SHong Zhang structure. The size of this array is equal to the number 1323d4002b98SHong Zhang of local rows, i.e 'm'. 1324d4002b98SHong Zhang 1325d4002b98SHong Zhang If the *_nnz parameter is given then the *_nz parameter is ignored 1326d4002b98SHong Zhang 1327d4002b98SHong Zhang The stored row and column indices begin with zero. 1328d4002b98SHong Zhang 1329d4002b98SHong Zhang The parallel matrix is partitioned such that the first m0 rows belong to 1330d4002b98SHong Zhang process 0, the next m1 rows belong to process 1, the next m2 rows belong 1331d4002b98SHong Zhang to process 2 etc.. where m0,m1,m2... are the input parameter 'm'. 1332d4002b98SHong Zhang 1333d4002b98SHong Zhang The DIAGONAL portion of the local submatrix of a processor can be defined 1334d4002b98SHong Zhang as the submatrix which is obtained by extraction the part corresponding to 1335d4002b98SHong Zhang the rows r1-r2 and columns c1-c2 of the global matrix, where r1 is the 1336d4002b98SHong Zhang first row that belongs to the processor, r2 is the last row belonging to 1337d4002b98SHong Zhang the this processor, and c1-c2 is range of indices of the local part of a 1338d4002b98SHong Zhang vector suitable for applying the matrix to. This is an mxn matrix. In the 1339d4002b98SHong Zhang common case of a square matrix, the row and column ranges are the same and 1340d4002b98SHong Zhang the DIAGONAL part is also square. The remaining portion of the local 1341d4002b98SHong Zhang submatrix (mxN) constitute the OFF-DIAGONAL portion. 1342d4002b98SHong Zhang 1343d4002b98SHong Zhang If o_nnz, d_nnz are specified, then o_nz, and d_nz are ignored. 1344d4002b98SHong Zhang 1345d4002b98SHong Zhang You can call MatGetInfo() to get information on how effective the preallocation was; 1346d4002b98SHong Zhang for example the fields mallocs,nz_allocated,nz_used,nz_unneeded; 1347d4002b98SHong Zhang You can also run with the option -info and look for messages with the string 1348d4002b98SHong Zhang malloc in them to see if additional memory allocation was needed. 1349d4002b98SHong Zhang 1350d4002b98SHong Zhang Example usage: 1351d4002b98SHong Zhang 1352d4002b98SHong Zhang Consider the following 8x8 matrix with 34 non-zero values, that is 1353d4002b98SHong Zhang assembled across 3 processors. Lets assume that proc0 owns 3 rows, 1354d4002b98SHong Zhang proc1 owns 3 rows, proc2 owns 2 rows. This division can be shown 1355d4002b98SHong Zhang as follows: 1356d4002b98SHong Zhang 1357d4002b98SHong Zhang .vb 1358d4002b98SHong Zhang 1 2 0 | 0 3 0 | 0 4 1359d4002b98SHong Zhang Proc0 0 5 6 | 7 0 0 | 8 0 1360d4002b98SHong Zhang 9 0 10 | 11 0 0 | 12 0 1361d4002b98SHong Zhang ------------------------------------- 1362d4002b98SHong Zhang 13 0 14 | 15 16 17 | 0 0 1363d4002b98SHong Zhang Proc1 0 18 0 | 19 20 21 | 0 0 1364d4002b98SHong Zhang 0 0 0 | 22 23 0 | 24 0 1365d4002b98SHong Zhang ------------------------------------- 1366d4002b98SHong Zhang Proc2 25 26 27 | 0 0 28 | 29 0 1367d4002b98SHong Zhang 30 0 0 | 31 32 33 | 0 34 1368d4002b98SHong Zhang .ve 1369d4002b98SHong Zhang 1370d4002b98SHong Zhang This can be represented as a collection of submatrices as: 1371d4002b98SHong Zhang 1372d4002b98SHong Zhang .vb 1373d4002b98SHong Zhang A B C 1374d4002b98SHong Zhang D E F 1375d4002b98SHong Zhang G H I 1376d4002b98SHong Zhang .ve 1377d4002b98SHong Zhang 1378d4002b98SHong Zhang Where the submatrices A,B,C are owned by proc0, D,E,F are 1379d4002b98SHong Zhang owned by proc1, G,H,I are owned by proc2. 1380d4002b98SHong Zhang 1381d4002b98SHong Zhang The 'm' parameters for proc0,proc1,proc2 are 3,3,2 respectively. 1382d4002b98SHong Zhang The 'n' parameters for proc0,proc1,proc2 are 3,3,2 respectively. 1383d4002b98SHong Zhang The 'M','N' parameters are 8,8, and have the same values on all procs. 1384d4002b98SHong Zhang 1385d4002b98SHong Zhang The DIAGONAL submatrices corresponding to proc0,proc1,proc2 are 1386d4002b98SHong Zhang submatrices [A], [E], [I] respectively. The OFF-DIAGONAL submatrices 1387d4002b98SHong Zhang corresponding to proc0,proc1,proc2 are [BC], [DF], [GH] respectively. 1388d4002b98SHong Zhang Internally, each processor stores the DIAGONAL part, and the OFF-DIAGONAL 1389d4002b98SHong Zhang part as SeqSELL matrices. for eg: proc1 will store [E] as a SeqSELL 1390d4002b98SHong Zhang matrix, ans [DF] as another SeqSELL matrix. 1391d4002b98SHong Zhang 1392d4002b98SHong Zhang When d_nz, o_nz parameters are specified, d_nz storage elements are 1393d4002b98SHong Zhang allocated for every row of the local diagonal submatrix, and o_nz 1394d4002b98SHong Zhang storage locations are allocated for every row of the OFF-DIAGONAL submat. 1395d4002b98SHong Zhang One way to choose d_nz and o_nz is to use the max nonzerors per local 1396d4002b98SHong Zhang rows for each of the local DIAGONAL, and the OFF-DIAGONAL submatrices. 1397d4002b98SHong Zhang In this case, the values of d_nz,o_nz are: 1398d4002b98SHong Zhang .vb 1399d4002b98SHong Zhang proc0 : dnz = 2, o_nz = 2 1400d4002b98SHong Zhang proc1 : dnz = 3, o_nz = 2 1401d4002b98SHong Zhang proc2 : dnz = 1, o_nz = 4 1402d4002b98SHong Zhang .ve 1403d4002b98SHong Zhang We are allocating m*(d_nz+o_nz) storage locations for every proc. This 1404d4002b98SHong Zhang translates to 3*(2+2)=12 for proc0, 3*(3+2)=15 for proc1, 2*(1+4)=10 1405d4002b98SHong Zhang for proc3. i.e we are using 12+15+10=37 storage locations to store 1406d4002b98SHong Zhang 34 values. 1407d4002b98SHong Zhang 1408d4002b98SHong Zhang When d_nnz, o_nnz parameters are specified, the storage is specified 1409d4002b98SHong Zhang for every row, coresponding to both DIAGONAL and OFF-DIAGONAL submatrices. 1410d4002b98SHong Zhang In the above case the values for d_nnz,o_nnz are: 1411d4002b98SHong Zhang .vb 1412d4002b98SHong Zhang proc0: d_nnz = [2,2,2] and o_nnz = [2,2,2] 1413d4002b98SHong Zhang proc1: d_nnz = [3,3,2] and o_nnz = [2,1,1] 1414d4002b98SHong Zhang proc2: d_nnz = [1,1] and o_nnz = [4,4] 1415d4002b98SHong Zhang .ve 1416d4002b98SHong Zhang Here the space allocated is according to nz (or maximum values in the nnz 1417d4002b98SHong Zhang if nnz is provided) for DIAGONAL and OFF-DIAGONAL submatrices, i.e (2+2+3+2)*3+(1+4)*2=37 1418d4002b98SHong Zhang 1419d4002b98SHong Zhang Level: intermediate 1420d4002b98SHong Zhang 1421d4002b98SHong Zhang .keywords: matrix, sell, sparse, parallel 1422d4002b98SHong Zhang 1423d4002b98SHong Zhang .seealso: MatCreate(), MatCreateSeqSELL(), MatSetValues(), MatCreatesell(), 1424d4002b98SHong Zhang MATMPISELL, MatGetInfo(), PetscSplitOwnership() 1425d4002b98SHong Zhang @*/ 1426d4002b98SHong Zhang PetscErrorCode MatMPISELLSetPreallocation(Mat B,PetscInt d_nz,const PetscInt d_nnz[],PetscInt o_nz,const PetscInt o_nnz[]) 1427d4002b98SHong Zhang { 1428d4002b98SHong Zhang PetscErrorCode ierr; 1429d4002b98SHong Zhang 1430d4002b98SHong Zhang PetscFunctionBegin; 1431d4002b98SHong Zhang PetscValidHeaderSpecific(B,MAT_CLASSID,1); 1432d4002b98SHong Zhang PetscValidType(B,1); 1433d4002b98SHong Zhang ierr = PetscTryMethod(B,"MatMPISELLSetPreallocation_C",(Mat,PetscInt,const PetscInt[],PetscInt,const PetscInt[]),(B,d_nz,d_nnz,o_nz,o_nnz));CHKERRQ(ierr); 1434d4002b98SHong Zhang PetscFunctionReturn(0); 1435d4002b98SHong Zhang } 1436d4002b98SHong Zhang 1437d4002b98SHong Zhang /*@C 1438d4002b98SHong Zhang MatCreateSELL - Creates a sparse parallel matrix in SELL format. 1439d4002b98SHong Zhang 1440d4002b98SHong Zhang Collective on MPI_Comm 1441d4002b98SHong Zhang 1442d4002b98SHong Zhang Input Parameters: 1443d4002b98SHong Zhang + comm - MPI communicator 1444d4002b98SHong Zhang . m - number of local rows (or PETSC_DECIDE to have calculated if M is given) 1445d4002b98SHong Zhang This value should be the same as the local size used in creating the 1446d4002b98SHong Zhang y vector for the matrix-vector product y = Ax. 1447d4002b98SHong Zhang . n - This value should be the same as the local size used in creating the 1448d4002b98SHong Zhang x vector for the matrix-vector product y = Ax. (or PETSC_DECIDE to have 1449d4002b98SHong Zhang calculated if N is given) For square matrices n is almost always m. 1450d4002b98SHong Zhang . M - number of global rows (or PETSC_DETERMINE to have calculated if m is given) 1451d4002b98SHong Zhang . N - number of global columns (or PETSC_DETERMINE to have calculated if n is given) 1452d4002b98SHong Zhang . d_rlenmax - max number of nonzeros per row in DIAGONAL portion of local submatrix 1453d4002b98SHong Zhang (same value is used for all local rows) 1454d4002b98SHong Zhang . d_rlen - array containing the number of nonzeros in the various rows of the 1455d4002b98SHong Zhang DIAGONAL portion of the local submatrix (possibly different for each row) 1456d4002b98SHong Zhang or NULL, if d_rlenmax is used to specify the nonzero structure. 1457d4002b98SHong Zhang The size of this array is equal to the number of local rows, i.e 'm'. 1458d4002b98SHong Zhang . o_rlenmax - max number of nonzeros per row in the OFF-DIAGONAL portion of local 1459d4002b98SHong Zhang submatrix (same value is used for all local rows). 1460d4002b98SHong Zhang - o_rlen - array containing the number of nonzeros in the various rows of the 1461d4002b98SHong Zhang OFF-DIAGONAL portion of the local submatrix (possibly different for 1462d4002b98SHong Zhang each row) or NULL, if o_rlenmax is used to specify the nonzero 1463d4002b98SHong Zhang structure. The size of this array is equal to the number 1464d4002b98SHong Zhang of local rows, i.e 'm'. 1465d4002b98SHong Zhang 1466d4002b98SHong Zhang Output Parameter: 1467d4002b98SHong Zhang . A - the matrix 1468d4002b98SHong Zhang 1469d4002b98SHong Zhang It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(), 1470d4002b98SHong Zhang MatXXXXSetPreallocation() paradgm instead of this routine directly. 1471d4002b98SHong Zhang [MatXXXXSetPreallocation() is, for example, MatSeqSELLSetPreallocation] 1472d4002b98SHong Zhang 1473d4002b98SHong Zhang Notes: 1474d4002b98SHong Zhang If the *_rlen parameter is given then the *_rlenmax parameter is ignored 1475d4002b98SHong Zhang 1476d4002b98SHong Zhang m,n,M,N parameters specify the size of the matrix, and its partitioning across 1477d4002b98SHong Zhang processors, while d_rlenmax,d_rlen,o_rlenmax,o_rlen parameters specify the approximate 1478d4002b98SHong Zhang storage requirements for this matrix. 1479d4002b98SHong Zhang 1480d4002b98SHong Zhang If PETSC_DECIDE or PETSC_DETERMINE is used for a particular argument on one 1481d4002b98SHong Zhang processor than it must be used on all processors that share the object for 1482d4002b98SHong Zhang that argument. 1483d4002b98SHong Zhang 1484d4002b98SHong Zhang The user MUST specify either the local or global matrix dimensions 1485d4002b98SHong Zhang (possibly both). 1486d4002b98SHong Zhang 1487d4002b98SHong Zhang The parallel matrix is partitioned across processors such that the 1488d4002b98SHong Zhang first m0 rows belong to process 0, the next m1 rows belong to 1489d4002b98SHong Zhang process 1, the next m2 rows belong to process 2 etc.. where 1490d4002b98SHong Zhang m0,m1,m2,.. are the input parameter 'm'. i.e each processor stores 1491d4002b98SHong Zhang values corresponding to [m x N] submatrix. 1492d4002b98SHong Zhang 1493d4002b98SHong Zhang The columns are logically partitioned with the n0 columns belonging 1494d4002b98SHong Zhang to 0th partition, the next n1 columns belonging to the next 1495d4002b98SHong Zhang partition etc.. where n0,n1,n2... are the input parameter 'n'. 1496d4002b98SHong Zhang 1497d4002b98SHong Zhang The DIAGONAL portion of the local submatrix on any given processor 1498d4002b98SHong Zhang is the submatrix corresponding to the rows and columns m,n 1499d4002b98SHong Zhang corresponding to the given processor. i.e diagonal matrix on 1500d4002b98SHong Zhang process 0 is [m0 x n0], diagonal matrix on process 1 is [m1 x n1] 1501d4002b98SHong Zhang etc. The remaining portion of the local submatrix [m x (N-n)] 1502d4002b98SHong Zhang constitute the OFF-DIAGONAL portion. The example below better 1503d4002b98SHong Zhang illustrates this concept. 1504d4002b98SHong Zhang 1505d4002b98SHong Zhang For a square global matrix we define each processor's diagonal portion 1506d4002b98SHong Zhang to be its local rows and the corresponding columns (a square submatrix); 1507d4002b98SHong Zhang each processor's off-diagonal portion encompasses the remainder of the 1508d4002b98SHong Zhang local matrix (a rectangular submatrix). 1509d4002b98SHong Zhang 1510d4002b98SHong Zhang If o_rlen, d_rlen are specified, then o_rlenmax, and d_rlenmax are ignored. 1511d4002b98SHong Zhang 1512d4002b98SHong Zhang When calling this routine with a single process communicator, a matrix of 1513d4002b98SHong Zhang type SEQSELL is returned. If a matrix of type MATMPISELL is desired for this 1514d4002b98SHong Zhang type of communicator, use the construction mechanism: 1515d4002b98SHong Zhang MatCreate(...,&A); MatSetType(A,MATMPISELL); MatSetSizes(A, m,n,M,N); MatMPISELLSetPreallocation(A,...); 1516d4002b98SHong Zhang 1517d4002b98SHong Zhang Options Database Keys: 1518d4002b98SHong Zhang - -mat_sell_oneindex - Internally use indexing starting at 1 1519d4002b98SHong Zhang rather than 0. Note that when calling MatSetValues(), 1520d4002b98SHong Zhang the user still MUST index entries starting at 0! 1521d4002b98SHong Zhang 1522d4002b98SHong Zhang 1523d4002b98SHong Zhang Example usage: 1524d4002b98SHong Zhang 1525d4002b98SHong Zhang Consider the following 8x8 matrix with 34 non-zero values, that is 1526d4002b98SHong Zhang assembled across 3 processors. Lets assume that proc0 owns 3 rows, 1527d4002b98SHong Zhang proc1 owns 3 rows, proc2 owns 2 rows. This division can be shown 1528d4002b98SHong Zhang as follows: 1529d4002b98SHong Zhang 1530d4002b98SHong Zhang .vb 1531d4002b98SHong Zhang 1 2 0 | 0 3 0 | 0 4 1532d4002b98SHong Zhang Proc0 0 5 6 | 7 0 0 | 8 0 1533d4002b98SHong Zhang 9 0 10 | 11 0 0 | 12 0 1534d4002b98SHong Zhang ------------------------------------- 1535d4002b98SHong Zhang 13 0 14 | 15 16 17 | 0 0 1536d4002b98SHong Zhang Proc1 0 18 0 | 19 20 21 | 0 0 1537d4002b98SHong Zhang 0 0 0 | 22 23 0 | 24 0 1538d4002b98SHong Zhang ------------------------------------- 1539d4002b98SHong Zhang Proc2 25 26 27 | 0 0 28 | 29 0 1540d4002b98SHong Zhang 30 0 0 | 31 32 33 | 0 34 1541d4002b98SHong Zhang .ve 1542d4002b98SHong Zhang 1543d4002b98SHong Zhang This can be represented as a collection of submatrices as: 1544d4002b98SHong Zhang 1545d4002b98SHong Zhang .vb 1546d4002b98SHong Zhang A B C 1547d4002b98SHong Zhang D E F 1548d4002b98SHong Zhang G H I 1549d4002b98SHong Zhang .ve 1550d4002b98SHong Zhang 1551d4002b98SHong Zhang Where the submatrices A,B,C are owned by proc0, D,E,F are 1552d4002b98SHong Zhang owned by proc1, G,H,I are owned by proc2. 1553d4002b98SHong Zhang 1554d4002b98SHong Zhang The 'm' parameters for proc0,proc1,proc2 are 3,3,2 respectively. 1555d4002b98SHong Zhang The 'n' parameters for proc0,proc1,proc2 are 3,3,2 respectively. 1556d4002b98SHong Zhang The 'M','N' parameters are 8,8, and have the same values on all procs. 1557d4002b98SHong Zhang 1558d4002b98SHong Zhang The DIAGONAL submatrices corresponding to proc0,proc1,proc2 are 1559d4002b98SHong Zhang submatrices [A], [E], [I] respectively. The OFF-DIAGONAL submatrices 1560d4002b98SHong Zhang corresponding to proc0,proc1,proc2 are [BC], [DF], [GH] respectively. 1561d4002b98SHong Zhang Internally, each processor stores the DIAGONAL part, and the OFF-DIAGONAL 1562d4002b98SHong Zhang part as SeqSELL matrices. for eg: proc1 will store [E] as a SeqSELL 1563d4002b98SHong Zhang matrix, ans [DF] as another SeqSELL matrix. 1564d4002b98SHong Zhang 1565d4002b98SHong Zhang When d_rlenmax, o_rlenmax parameters are specified, d_rlenmax storage elements are 1566d4002b98SHong Zhang allocated for every row of the local diagonal submatrix, and o_rlenmax 1567d4002b98SHong Zhang storage locations are allocated for every row of the OFF-DIAGONAL submat. 1568d4002b98SHong Zhang One way to choose d_rlenmax and o_rlenmax is to use the max nonzerors per local 1569d4002b98SHong Zhang rows for each of the local DIAGONAL, and the OFF-DIAGONAL submatrices. 1570d4002b98SHong Zhang In this case, the values of d_rlenmax,o_rlenmax are: 1571d4002b98SHong Zhang .vb 1572d4002b98SHong Zhang proc0 : d_rlenmax = 2, o_rlenmax = 2 1573d4002b98SHong Zhang proc1 : d_rlenmax = 3, o_rlenmax = 2 1574d4002b98SHong Zhang proc2 : d_rlenmax = 1, o_rlenmax = 4 1575d4002b98SHong Zhang .ve 1576d4002b98SHong Zhang We are allocating m*(d_rlenmax+o_rlenmax) storage locations for every proc. This 1577d4002b98SHong Zhang translates to 3*(2+2)=12 for proc0, 3*(3+2)=15 for proc1, 2*(1+4)=10 1578d4002b98SHong Zhang for proc3. i.e we are using 12+15+10=37 storage locations to store 1579d4002b98SHong Zhang 34 values. 1580d4002b98SHong Zhang 1581d4002b98SHong Zhang When d_rlen, o_rlen parameters are specified, the storage is specified 1582d4002b98SHong Zhang for every row, coresponding to both DIAGONAL and OFF-DIAGONAL submatrices. 1583d4002b98SHong Zhang In the above case the values for d_nnz,o_nnz are: 1584d4002b98SHong Zhang .vb 1585d4002b98SHong Zhang proc0: d_nnz = [2,2,2] and o_nnz = [2,2,2] 1586d4002b98SHong Zhang proc1: d_nnz = [3,3,2] and o_nnz = [2,1,1] 1587d4002b98SHong Zhang proc2: d_nnz = [1,1] and o_nnz = [4,4] 1588d4002b98SHong Zhang .ve 1589d4002b98SHong Zhang Here the space allocated is still 37 though there are 34 nonzeros because 1590d4002b98SHong Zhang the allocation is always done according to rlenmax. 1591d4002b98SHong Zhang 1592d4002b98SHong Zhang Level: intermediate 1593d4002b98SHong Zhang 1594d4002b98SHong Zhang .keywords: matrix, sell, sparse, parallel 1595d4002b98SHong Zhang 1596d4002b98SHong Zhang .seealso: MatCreate(), MatCreateSeqSELL(), MatSetValues(), MatMPISELLSetPreallocation(), MatMPISELLSetPreallocationSELL(), 1597d4002b98SHong Zhang MATMPISELL, MatCreateMPISELLWithArrays() 1598d4002b98SHong Zhang @*/ 1599d4002b98SHong Zhang PetscErrorCode MatCreateSELL(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,PetscInt d_rlenmax,const PetscInt d_rlen[],PetscInt o_rlenmax,const PetscInt o_rlen[],Mat *A) 1600d4002b98SHong Zhang { 1601d4002b98SHong Zhang PetscErrorCode ierr; 1602d4002b98SHong Zhang PetscMPIInt size; 1603d4002b98SHong Zhang 1604d4002b98SHong Zhang PetscFunctionBegin; 1605d4002b98SHong Zhang ierr = MatCreate(comm,A);CHKERRQ(ierr); 1606d4002b98SHong Zhang ierr = MatSetSizes(*A,m,n,M,N);CHKERRQ(ierr); 1607d4002b98SHong Zhang ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 1608d4002b98SHong Zhang if (size > 1) { 1609d4002b98SHong Zhang ierr = MatSetType(*A,MATMPISELL);CHKERRQ(ierr); 1610d4002b98SHong Zhang ierr = MatMPISELLSetPreallocation(*A,d_rlenmax,d_rlen,o_rlenmax,o_rlen);CHKERRQ(ierr); 1611d4002b98SHong Zhang } else { 1612d4002b98SHong Zhang ierr = MatSetType(*A,MATSEQSELL);CHKERRQ(ierr); 1613d4002b98SHong Zhang ierr = MatSeqSELLSetPreallocation(*A,d_rlenmax,d_rlen);CHKERRQ(ierr); 1614d4002b98SHong Zhang } 1615d4002b98SHong Zhang PetscFunctionReturn(0); 1616d4002b98SHong Zhang } 1617d4002b98SHong Zhang 1618d4002b98SHong Zhang PetscErrorCode MatMPISELLGetSeqSELL(Mat A,Mat *Ad,Mat *Ao,const PetscInt *colmap[]) 1619d4002b98SHong Zhang { 1620d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 1621d4002b98SHong Zhang PetscBool flg; 1622d4002b98SHong Zhang PetscErrorCode ierr; 1623d4002b98SHong Zhang 1624d4002b98SHong Zhang PetscFunctionBegin; 1625d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)A,MATMPISELL,&flg);CHKERRQ(ierr); 1626d4002b98SHong Zhang if (!flg) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"This function requires a MATMPISELL matrix as input"); 1627d4002b98SHong Zhang if (Ad) *Ad = a->A; 1628d4002b98SHong Zhang if (Ao) *Ao = a->B; 1629d4002b98SHong Zhang if (colmap) *colmap = a->garray; 1630d4002b98SHong Zhang PetscFunctionReturn(0); 1631d4002b98SHong Zhang } 1632d4002b98SHong Zhang 1633d4002b98SHong Zhang /*@C 1634d4002b98SHong Zhang MatMPISELLGetLocalMatCondensed - Creates a SeqSELL matrix from an MATMPISELL matrix by taking all its local rows and NON-ZERO columns 1635d4002b98SHong Zhang 1636d4002b98SHong Zhang Not Collective 1637d4002b98SHong Zhang 1638d4002b98SHong Zhang Input Parameters: 1639d4002b98SHong Zhang + A - the matrix 1640d4002b98SHong Zhang . scall - either MAT_INITIAL_MATRIX or MAT_REUSE_MATRIX 1641d4002b98SHong Zhang - row, col - index sets of rows and columns to extract (or NULL) 1642d4002b98SHong Zhang 1643d4002b98SHong Zhang Output Parameter: 1644d4002b98SHong Zhang . A_loc - the local sequential matrix generated 1645d4002b98SHong Zhang 1646d4002b98SHong Zhang Level: developer 1647d4002b98SHong Zhang 1648d4002b98SHong Zhang .seealso: MatGetOwnershipRange(), MatMPISELLGetLocalMat() 1649d4002b98SHong Zhang 1650d4002b98SHong Zhang @*/ 1651d4002b98SHong Zhang PetscErrorCode MatMPISELLGetLocalMatCondensed(Mat A,MatReuse scall,IS *row,IS *col,Mat *A_loc) 1652d4002b98SHong Zhang { 1653d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 1654d4002b98SHong Zhang PetscErrorCode ierr; 1655d4002b98SHong Zhang PetscInt i,start,end,ncols,nzA,nzB,*cmap,imark,*idx; 1656d4002b98SHong Zhang IS isrowa,iscola; 1657d4002b98SHong Zhang Mat *aloc; 1658d4002b98SHong Zhang PetscBool match; 1659d4002b98SHong Zhang 1660d4002b98SHong Zhang PetscFunctionBegin; 1661d4002b98SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)A,MATMPISELL,&match);CHKERRQ(ierr); 1662d4002b98SHong Zhang if (!match) SETERRQ(PetscObjectComm((PetscObject)A), PETSC_ERR_SUP,"Requires MATMPISELL matrix as input"); 1663d4002b98SHong Zhang ierr = PetscLogEventBegin(MAT_Getlocalmatcondensed,A,0,0,0);CHKERRQ(ierr); 1664d4002b98SHong Zhang if (!row) { 1665d4002b98SHong Zhang start = A->rmap->rstart; end = A->rmap->rend; 1666d4002b98SHong Zhang ierr = ISCreateStride(PETSC_COMM_SELF,end-start,start,1,&isrowa);CHKERRQ(ierr); 1667d4002b98SHong Zhang } else { 1668d4002b98SHong Zhang isrowa = *row; 1669d4002b98SHong Zhang } 1670d4002b98SHong Zhang if (!col) { 1671d4002b98SHong Zhang start = A->cmap->rstart; 1672d4002b98SHong Zhang cmap = a->garray; 1673d4002b98SHong Zhang nzA = a->A->cmap->n; 1674d4002b98SHong Zhang nzB = a->B->cmap->n; 1675d4002b98SHong Zhang ierr = PetscMalloc1(nzA+nzB, &idx);CHKERRQ(ierr); 1676d4002b98SHong Zhang ncols = 0; 1677d4002b98SHong Zhang for (i=0; i<nzB; i++) { 1678d4002b98SHong Zhang if (cmap[i] < start) idx[ncols++] = cmap[i]; 1679d4002b98SHong Zhang else break; 1680d4002b98SHong Zhang } 1681d4002b98SHong Zhang imark = i; 1682d4002b98SHong Zhang for (i=0; i<nzA; i++) idx[ncols++] = start + i; 1683d4002b98SHong Zhang for (i=imark; i<nzB; i++) idx[ncols++] = cmap[i]; 1684d4002b98SHong Zhang ierr = ISCreateGeneral(PETSC_COMM_SELF,ncols,idx,PETSC_OWN_POINTER,&iscola);CHKERRQ(ierr); 1685d4002b98SHong Zhang } else { 1686d4002b98SHong Zhang iscola = *col; 1687d4002b98SHong Zhang } 1688d4002b98SHong Zhang if (scall != MAT_INITIAL_MATRIX) { 1689d4002b98SHong Zhang ierr = PetscMalloc1(1,&aloc);CHKERRQ(ierr); 1690d4002b98SHong Zhang aloc[0] = *A_loc; 1691d4002b98SHong Zhang } 1692d4002b98SHong Zhang ierr = MatCreateSubMatrices(A,1,&isrowa,&iscola,scall,&aloc);CHKERRQ(ierr); 1693d4002b98SHong Zhang *A_loc = aloc[0]; 1694d4002b98SHong Zhang ierr = PetscFree(aloc);CHKERRQ(ierr); 1695d4002b98SHong Zhang if (!row) { 1696d4002b98SHong Zhang ierr = ISDestroy(&isrowa);CHKERRQ(ierr); 1697d4002b98SHong Zhang } 1698d4002b98SHong Zhang if (!col) { 1699d4002b98SHong Zhang ierr = ISDestroy(&iscola);CHKERRQ(ierr); 1700d4002b98SHong Zhang } 1701d4002b98SHong Zhang ierr = PetscLogEventEnd(MAT_Getlocalmatcondensed,A,0,0,0);CHKERRQ(ierr); 1702d4002b98SHong Zhang PetscFunctionReturn(0); 1703d4002b98SHong Zhang } 1704d4002b98SHong Zhang 1705d4002b98SHong Zhang #include <../src/mat/impls/aij/mpi/mpiaij.h> 1706d4002b98SHong Zhang 1707d4002b98SHong Zhang PetscErrorCode MatConvert_MPISELL_MPIAIJ(Mat A,MatType newtype,MatReuse reuse,Mat *newmat) 1708d4002b98SHong Zhang { 1709d4002b98SHong Zhang PetscErrorCode ierr; 1710d4002b98SHong Zhang Mat_MPISELL *a=(Mat_MPISELL*)A->data; 1711d4002b98SHong Zhang Mat B; 1712d4002b98SHong Zhang Mat_MPIAIJ *b; 1713d4002b98SHong Zhang 1714d4002b98SHong Zhang PetscFunctionBegin; 1715d4002b98SHong Zhang if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Matrix must be assembled"); 1716d4002b98SHong Zhang 1717*94a8b381SRichard Tran Mills if (reuse == MAT_REUSE_MATRIX) { 1718*94a8b381SRichard Tran Mills B = *newmat; 1719*94a8b381SRichard Tran Mills } else { 1720d4002b98SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr); 1721d4002b98SHong Zhang ierr = MatSetType(B,MATMPIAIJ);CHKERRQ(ierr); 1722d4002b98SHong Zhang ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 1723d4002b98SHong Zhang ierr = MatSetBlockSizes(B,A->rmap->bs,A->cmap->bs);CHKERRQ(ierr); 1724d4002b98SHong Zhang ierr = MatSeqAIJSetPreallocation(B,0,NULL);CHKERRQ(ierr); 1725d4002b98SHong Zhang ierr = MatMPIAIJSetPreallocation(B,0,NULL,0,NULL);CHKERRQ(ierr); 1726*94a8b381SRichard Tran Mills } 1727d4002b98SHong Zhang b = (Mat_MPIAIJ*) B->data; 1728*94a8b381SRichard Tran Mills 1729*94a8b381SRichard Tran Mills if (reuse == MAT_REUSE_MATRIX) { 1730*94a8b381SRichard Tran Mills ierr = MatConvert_SeqSELL_SeqAIJ(a->A, MATSEQAIJ, MAT_REUSE_MATRIX, &b->A);CHKERRQ(ierr); 1731*94a8b381SRichard Tran Mills ierr = MatConvert_SeqSELL_SeqAIJ(a->B, MATSEQAIJ, MAT_REUSE_MATRIX, &b->B);CHKERRQ(ierr); 1732*94a8b381SRichard Tran Mills } else { 1733d4002b98SHong Zhang ierr = MatDestroy(&b->A);CHKERRQ(ierr); 1734d4002b98SHong Zhang ierr = MatDestroy(&b->B);CHKERRQ(ierr); 1735d4002b98SHong Zhang ierr = MatDisAssemble_MPISELL(A);CHKERRQ(ierr); 1736d4002b98SHong Zhang ierr = MatConvert_SeqSELL_SeqAIJ(a->A, MATSEQAIJ, MAT_INITIAL_MATRIX, &b->A);CHKERRQ(ierr); 1737d4002b98SHong Zhang ierr = MatConvert_SeqSELL_SeqAIJ(a->B, MATSEQAIJ, MAT_INITIAL_MATRIX, &b->B);CHKERRQ(ierr); 1738d4002b98SHong Zhang ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1739d4002b98SHong Zhang ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1740d4002b98SHong Zhang ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1741d4002b98SHong Zhang ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1742*94a8b381SRichard Tran Mills } 1743d4002b98SHong Zhang 1744d4002b98SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 1745d4002b98SHong Zhang ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr); 1746d4002b98SHong Zhang } else { 1747d4002b98SHong Zhang *newmat = B; 1748d4002b98SHong Zhang } 1749d4002b98SHong Zhang PetscFunctionReturn(0); 1750d4002b98SHong Zhang } 1751d4002b98SHong Zhang 1752d4002b98SHong Zhang PetscErrorCode MatConvert_MPIAIJ_MPISELL(Mat A,MatType newtype,MatReuse reuse,Mat *newmat) 1753d4002b98SHong Zhang { 1754d4002b98SHong Zhang PetscErrorCode ierr; 1755d4002b98SHong Zhang Mat_MPIAIJ *a=(Mat_MPIAIJ*)A->data; 1756d4002b98SHong Zhang Mat B; 1757d4002b98SHong Zhang Mat_MPISELL *b; 1758d4002b98SHong Zhang 1759d4002b98SHong Zhang PetscFunctionBegin; 1760d4002b98SHong Zhang if (!A->assembled) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Matrix must be assembled"); 1761d4002b98SHong Zhang 1762*94a8b381SRichard Tran Mills if (reuse == MAT_REUSE_MATRIX) { 1763*94a8b381SRichard Tran Mills B = *newmat; 1764*94a8b381SRichard Tran Mills } else { 1765d4002b98SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr); 1766d4002b98SHong Zhang ierr = MatSetType(B,MATMPISELL);CHKERRQ(ierr); 1767d4002b98SHong Zhang ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 1768d4002b98SHong Zhang ierr = MatSetBlockSizes(B,A->rmap->bs,A->cmap->bs);CHKERRQ(ierr); 1769d4002b98SHong Zhang ierr = MatSeqAIJSetPreallocation(B,0,NULL);CHKERRQ(ierr); 1770d4002b98SHong Zhang ierr = MatMPIAIJSetPreallocation(B,0,NULL,0,NULL);CHKERRQ(ierr); 1771*94a8b381SRichard Tran Mills } 1772d4002b98SHong Zhang b = (Mat_MPISELL*) B->data; 1773*94a8b381SRichard Tran Mills 1774*94a8b381SRichard Tran Mills if (reuse == MAT_REUSE_MATRIX) { 1775*94a8b381SRichard Tran Mills ierr = MatConvert_SeqAIJ_SeqSELL(a->A, MATSEQSELL, MAT_REUSE_MATRIX, &b->A);CHKERRQ(ierr); 1776*94a8b381SRichard Tran Mills ierr = MatConvert_SeqAIJ_SeqSELL(a->B, MATSEQSELL, MAT_REUSE_MATRIX, &b->B);CHKERRQ(ierr); 1777*94a8b381SRichard Tran Mills } else { 1778d4002b98SHong Zhang ierr = MatDestroy(&b->A);CHKERRQ(ierr); 1779d4002b98SHong Zhang ierr = MatDestroy(&b->B);CHKERRQ(ierr); 1780d4002b98SHong Zhang ierr = MatDisAssemble_MPIAIJ(A);CHKERRQ(ierr); 1781d4002b98SHong Zhang ierr = MatConvert_SeqAIJ_SeqSELL(a->A, MATSEQSELL, MAT_INITIAL_MATRIX, &b->A);CHKERRQ(ierr); 1782d4002b98SHong Zhang ierr = MatConvert_SeqAIJ_SeqSELL(a->B, MATSEQSELL, MAT_INITIAL_MATRIX, &b->B);CHKERRQ(ierr); 1783d4002b98SHong Zhang ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1784d4002b98SHong Zhang ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1785d4002b98SHong Zhang ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1786d4002b98SHong Zhang ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1787*94a8b381SRichard Tran Mills } 1788d4002b98SHong Zhang 1789d4002b98SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 1790d4002b98SHong Zhang ierr = MatHeaderReplace(A,&B);CHKERRQ(ierr); 1791d4002b98SHong Zhang } else { 1792d4002b98SHong Zhang *newmat = B; 1793d4002b98SHong Zhang } 1794d4002b98SHong Zhang PetscFunctionReturn(0); 1795d4002b98SHong Zhang } 1796d4002b98SHong Zhang 1797d4002b98SHong Zhang PetscErrorCode MatSOR_MPISELL(Mat matin,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx) 1798d4002b98SHong Zhang { 1799d4002b98SHong Zhang Mat_MPISELL *mat=(Mat_MPISELL*)matin->data; 1800d4002b98SHong Zhang PetscErrorCode ierr; 1801d4002b98SHong Zhang Vec bb1=0; 1802d4002b98SHong Zhang 1803d4002b98SHong Zhang PetscFunctionBegin; 1804d4002b98SHong Zhang if (flag == SOR_APPLY_UPPER) { 1805d4002b98SHong Zhang ierr = (*mat->A->ops->sor)(mat->A,bb,omega,flag,fshift,lits,1,xx);CHKERRQ(ierr); 1806d4002b98SHong Zhang PetscFunctionReturn(0); 1807d4002b98SHong Zhang } 1808d4002b98SHong Zhang 1809d4002b98SHong Zhang if (its > 1 || ~flag & SOR_ZERO_INITIAL_GUESS || flag & SOR_EISENSTAT) { 1810d4002b98SHong Zhang ierr = VecDuplicate(bb,&bb1);CHKERRQ(ierr); 1811d4002b98SHong Zhang } 1812d4002b98SHong Zhang 1813d4002b98SHong Zhang if ((flag & SOR_LOCAL_SYMMETRIC_SWEEP) == SOR_LOCAL_SYMMETRIC_SWEEP) { 1814d4002b98SHong Zhang if (flag & SOR_ZERO_INITIAL_GUESS) { 1815d4002b98SHong Zhang ierr = (*mat->A->ops->sor)(mat->A,bb,omega,flag,fshift,lits,1,xx);CHKERRQ(ierr); 1816d4002b98SHong Zhang its--; 1817d4002b98SHong Zhang } 1818d4002b98SHong Zhang 1819d4002b98SHong Zhang while (its--) { 1820d4002b98SHong Zhang ierr = VecScatterBegin(mat->Mvctx,xx,mat->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1821d4002b98SHong Zhang ierr = VecScatterEnd(mat->Mvctx,xx,mat->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1822d4002b98SHong Zhang 1823d4002b98SHong Zhang /* update rhs: bb1 = bb - B*x */ 1824d4002b98SHong Zhang ierr = VecScale(mat->lvec,-1.0);CHKERRQ(ierr); 1825d4002b98SHong Zhang ierr = (*mat->B->ops->multadd)(mat->B,mat->lvec,bb,bb1);CHKERRQ(ierr); 1826d4002b98SHong Zhang 1827d4002b98SHong Zhang /* local sweep */ 1828d4002b98SHong Zhang ierr = (*mat->A->ops->sor)(mat->A,bb1,omega,SOR_SYMMETRIC_SWEEP,fshift,lits,1,xx);CHKERRQ(ierr); 1829d4002b98SHong Zhang } 1830d4002b98SHong Zhang } else if (flag & SOR_LOCAL_FORWARD_SWEEP) { 1831d4002b98SHong Zhang if (flag & SOR_ZERO_INITIAL_GUESS) { 1832d4002b98SHong Zhang ierr = (*mat->A->ops->sor)(mat->A,bb,omega,flag,fshift,lits,1,xx);CHKERRQ(ierr); 1833d4002b98SHong Zhang its--; 1834d4002b98SHong Zhang } 1835d4002b98SHong Zhang while (its--) { 1836d4002b98SHong Zhang ierr = VecScatterBegin(mat->Mvctx,xx,mat->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1837d4002b98SHong Zhang ierr = VecScatterEnd(mat->Mvctx,xx,mat->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1838d4002b98SHong Zhang 1839d4002b98SHong Zhang /* update rhs: bb1 = bb - B*x */ 1840d4002b98SHong Zhang ierr = VecScale(mat->lvec,-1.0);CHKERRQ(ierr); 1841d4002b98SHong Zhang ierr = (*mat->B->ops->multadd)(mat->B,mat->lvec,bb,bb1);CHKERRQ(ierr); 1842d4002b98SHong Zhang 1843d4002b98SHong Zhang /* local sweep */ 1844d4002b98SHong Zhang ierr = (*mat->A->ops->sor)(mat->A,bb1,omega,SOR_FORWARD_SWEEP,fshift,lits,1,xx);CHKERRQ(ierr); 1845d4002b98SHong Zhang } 1846d4002b98SHong Zhang } else if (flag & SOR_LOCAL_BACKWARD_SWEEP) { 1847d4002b98SHong Zhang if (flag & SOR_ZERO_INITIAL_GUESS) { 1848d4002b98SHong Zhang ierr = (*mat->A->ops->sor)(mat->A,bb,omega,flag,fshift,lits,1,xx);CHKERRQ(ierr); 1849d4002b98SHong Zhang its--; 1850d4002b98SHong Zhang } 1851d4002b98SHong Zhang while (its--) { 1852d4002b98SHong Zhang ierr = VecScatterBegin(mat->Mvctx,xx,mat->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1853d4002b98SHong Zhang ierr = VecScatterEnd(mat->Mvctx,xx,mat->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1854d4002b98SHong Zhang 1855d4002b98SHong Zhang /* update rhs: bb1 = bb - B*x */ 1856d4002b98SHong Zhang ierr = VecScale(mat->lvec,-1.0);CHKERRQ(ierr); 1857d4002b98SHong Zhang ierr = (*mat->B->ops->multadd)(mat->B,mat->lvec,bb,bb1);CHKERRQ(ierr); 1858d4002b98SHong Zhang 1859d4002b98SHong Zhang /* local sweep */ 1860d4002b98SHong Zhang ierr = (*mat->A->ops->sor)(mat->A,bb1,omega,SOR_BACKWARD_SWEEP,fshift,lits,1,xx);CHKERRQ(ierr); 1861d4002b98SHong Zhang } 1862d4002b98SHong Zhang } else SETERRQ(PetscObjectComm((PetscObject)matin),PETSC_ERR_SUP,"Parallel SOR not supported"); 1863d4002b98SHong Zhang 1864d4002b98SHong Zhang ierr = VecDestroy(&bb1);CHKERRQ(ierr); 1865d4002b98SHong Zhang 1866d4002b98SHong Zhang matin->factorerrortype = mat->A->factorerrortype; 1867d4002b98SHong Zhang PetscFunctionReturn(0); 1868d4002b98SHong Zhang } 1869d4002b98SHong Zhang 1870d4002b98SHong Zhang /*MC 1871d4002b98SHong Zhang MATMPISELL - MATMPISELL = "MPISELL" - A matrix type to be used for parallel sparse matrices. 1872d4002b98SHong Zhang 1873d4002b98SHong Zhang Options Database Keys: 1874d4002b98SHong Zhang . -mat_type MPISELL - sets the matrix type to "MPISELL" during a call to MatSetFromOptions() 1875d4002b98SHong Zhang 1876d4002b98SHong Zhang Level: beginner 1877d4002b98SHong Zhang 1878d4002b98SHong Zhang .seealso: MatCreateSELL() 1879d4002b98SHong Zhang M*/ 1880d4002b98SHong Zhang PETSC_EXTERN PetscErrorCode MatCreate_MPISELL(Mat B) 1881d4002b98SHong Zhang { 1882d4002b98SHong Zhang Mat_MPISELL *b; 1883d4002b98SHong Zhang PetscErrorCode ierr; 1884d4002b98SHong Zhang PetscMPIInt size; 1885d4002b98SHong Zhang 1886d4002b98SHong Zhang PetscFunctionBegin; 1887d4002b98SHong Zhang ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr); 1888d4002b98SHong Zhang ierr = PetscNewLog(B,&b);CHKERRQ(ierr); 1889d4002b98SHong Zhang B->data = (void*)b; 1890d4002b98SHong Zhang ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 1891d4002b98SHong Zhang B->assembled = PETSC_FALSE; 1892d4002b98SHong Zhang B->insertmode = NOT_SET_VALUES; 1893d4002b98SHong Zhang b->size = size; 1894d4002b98SHong Zhang ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)B),&b->rank);CHKERRQ(ierr); 1895d4002b98SHong Zhang /* build cache for off array entries formed */ 1896d4002b98SHong Zhang ierr = MatStashCreate_Private(PetscObjectComm((PetscObject)B),1,&B->stash);CHKERRQ(ierr); 1897d4002b98SHong Zhang 1898d4002b98SHong Zhang b->donotstash = PETSC_FALSE; 1899d4002b98SHong Zhang b->colmap = 0; 1900d4002b98SHong Zhang b->garray = 0; 1901d4002b98SHong Zhang b->roworiented = PETSC_TRUE; 1902d4002b98SHong Zhang 1903d4002b98SHong Zhang /* stuff used for matrix vector multiply */ 1904d4002b98SHong Zhang b->lvec = NULL; 1905d4002b98SHong Zhang b->Mvctx = NULL; 1906d4002b98SHong Zhang 1907d4002b98SHong Zhang /* stuff for MatGetRow() */ 1908d4002b98SHong Zhang b->rowindices = 0; 1909d4002b98SHong Zhang b->rowvalues = 0; 1910d4002b98SHong Zhang b->getrowactive = PETSC_FALSE; 1911d4002b98SHong Zhang 1912d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_MPISELL);CHKERRQ(ierr); 1913d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_MPISELL);CHKERRQ(ierr); 1914d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatIsTranspose_C",MatIsTranspose_MPISELL);CHKERRQ(ierr); 1915d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatMPISELLSetPreallocation_C",MatMPISELLSetPreallocation_MPISELL);CHKERRQ(ierr); 1916d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_mpisell_mpiaij_C",MatConvert_MPISELL_MPIAIJ);CHKERRQ(ierr); 1917d4002b98SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatDiagonalScaleLocal_C",MatDiagonalScaleLocal_MPISELL);CHKERRQ(ierr); 1918d4002b98SHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)B,MATMPISELL);CHKERRQ(ierr); 1919d4002b98SHong Zhang PetscFunctionReturn(0); 1920d4002b98SHong Zhang } 1921