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