1 #ifndef lint 2 static char vcid[] = "$Id: mmdense.c,v 1.8 1996/08/15 12:47:17 bsmith Exp balay $"; 3 #endif 4 5 /* 6 Support for the parallel dense matrix vector multiply 7 */ 8 #include "src/mat/impls/dense/mpi/mpidense.h" 9 #include "src/vec/vecimpl.h" 10 11 #undef __FUNCTION__ 12 #define __FUNCTION__ "MatSetUpMultiply_MPIDense" 13 int MatSetUpMultiply_MPIDense(Mat mat) 14 { 15 Mat_MPIDense *mdn = (Mat_MPIDense *) mat->data; 16 int ierr,n; 17 IS tofrom; 18 Vec gvec; 19 20 /* Create local vector that is used to scatter into */ 21 ierr = VecCreateSeq(MPI_COMM_SELF,mdn->N,&mdn->lvec); CHKERRQ(ierr); 22 23 /* Create temporary index set for building scatter gather */ 24 ierr = ISCreateStride(MPI_COMM_SELF,mdn->N,0,1,&tofrom); CHKERRQ(ierr); 25 26 /* Create temporary global vector to generate scatter context */ 27 n = mdn->cowners[mdn->rank+1] - mdn->cowners[mdn->rank]; 28 ierr = VecCreateMPI(mat->comm,n,mdn->N,&gvec); CHKERRQ(ierr); 29 30 /* Generate the scatter context */ 31 ierr = VecScatterCreate(gvec,tofrom,mdn->lvec,tofrom,&mdn->Mvctx); CHKERRQ(ierr); 32 PLogObjectParent(mat,mdn->Mvctx); 33 PLogObjectParent(mat,mdn->lvec); 34 PLogObjectParent(mat,tofrom); 35 PLogObjectParent(mat,gvec); 36 37 ierr = ISDestroy(tofrom); CHKERRQ(ierr); 38 ierr = VecDestroy(gvec); CHKERRQ(ierr); 39 return 0; 40 } 41 42 43 44