1be1d678aSKris Buschelman #define PETSCMAT_DLL 2be1d678aSKris Buschelman 367e560aaSBarry Smith /* 467e560aaSBarry Smith Defines the basic matrix operations for sequential dense. 567e560aaSBarry Smith */ 6289bc588SBarry Smith 77c4f633dSBarry Smith #include "../src/mat/impls/dense/seq/dense.h" 8b4c862e0SSatish Balay #include "petscblaslapack.h" 9289bc588SBarry Smith 104a2ae208SSatish Balay #undef __FUNCT__ 114a2ae208SSatish Balay #define __FUNCT__ "MatAXPY_SeqDense" 12f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqDense(Mat Y,PetscScalar alpha,Mat X,MatStructure str) 131987afe7SBarry Smith { 141987afe7SBarry Smith Mat_SeqDense *x = (Mat_SeqDense*)X->data,*y = (Mat_SeqDense*)Y->data; 15f4df32b1SMatthew Knepley PetscScalar oalpha = alpha; 1613f74950SBarry Smith PetscInt j; 170805154bSBarry Smith PetscBLASInt N,m,ldax,lday,one = 1; 18efee365bSSatish Balay PetscErrorCode ierr; 193a40ed3dSBarry Smith 203a40ed3dSBarry Smith PetscFunctionBegin; 21d0f46423SBarry Smith N = PetscBLASIntCast(X->rmap->n*X->cmap->n); 22d0f46423SBarry Smith m = PetscBLASIntCast(X->rmap->n); 230805154bSBarry Smith ldax = PetscBLASIntCast(x->lda); 240805154bSBarry Smith lday = PetscBLASIntCast(y->lda); 25a5ce6ee0Svictorle if (ldax>m || lday>m) { 26d0f46423SBarry Smith for (j=0; j<X->cmap->n; j++) { 27f4df32b1SMatthew Knepley BLASaxpy_(&m,&oalpha,x->v+j*ldax,&one,y->v+j*lday,&one); 28a5ce6ee0Svictorle } 29a5ce6ee0Svictorle } else { 30f4df32b1SMatthew Knepley BLASaxpy_(&N,&oalpha,x->v,&one,y->v,&one); 31a5ce6ee0Svictorle } 320450473dSBarry Smith ierr = PetscLogFlops(PetscMax(2*N-1,0));CHKERRQ(ierr); 333a40ed3dSBarry Smith PetscFunctionReturn(0); 341987afe7SBarry Smith } 351987afe7SBarry Smith 364a2ae208SSatish Balay #undef __FUNCT__ 374a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_SeqDense" 38dfbe8321SBarry Smith PetscErrorCode MatGetInfo_SeqDense(Mat A,MatInfoType flag,MatInfo *info) 39289bc588SBarry Smith { 40d0f46423SBarry Smith PetscInt N = A->rmap->n*A->cmap->n; 413a40ed3dSBarry Smith 423a40ed3dSBarry Smith PetscFunctionBegin; 434e220ebcSLois Curfman McInnes info->block_size = 1.0; 444e220ebcSLois Curfman McInnes info->nz_allocated = (double)N; 456de62eeeSBarry Smith info->nz_used = (double)N; 466de62eeeSBarry Smith info->nz_unneeded = (double)0; 474e220ebcSLois Curfman McInnes info->assemblies = (double)A->num_ass; 484e220ebcSLois Curfman McInnes info->mallocs = 0; 497adad957SLisandro Dalcin info->memory = ((PetscObject)A)->mem; 504e220ebcSLois Curfman McInnes info->fill_ratio_given = 0; 514e220ebcSLois Curfman McInnes info->fill_ratio_needed = 0; 524e220ebcSLois Curfman McInnes info->factor_mallocs = 0; 533a40ed3dSBarry Smith PetscFunctionReturn(0); 54289bc588SBarry Smith } 55289bc588SBarry Smith 564a2ae208SSatish Balay #undef __FUNCT__ 574a2ae208SSatish Balay #define __FUNCT__ "MatScale_SeqDense" 58f4df32b1SMatthew Knepley PetscErrorCode MatScale_SeqDense(Mat A,PetscScalar alpha) 5980cd9d93SLois Curfman McInnes { 60273d9f13SBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 61f4df32b1SMatthew Knepley PetscScalar oalpha = alpha; 62efee365bSSatish Balay PetscErrorCode ierr; 630805154bSBarry Smith PetscBLASInt one = 1,j,nz,lda = PetscBLASIntCast(a->lda); 6480cd9d93SLois Curfman McInnes 653a40ed3dSBarry Smith PetscFunctionBegin; 66d0f46423SBarry Smith if (lda>A->rmap->n) { 67d0f46423SBarry Smith nz = PetscBLASIntCast(A->rmap->n); 68d0f46423SBarry Smith for (j=0; j<A->cmap->n; j++) { 69f4df32b1SMatthew Knepley BLASscal_(&nz,&oalpha,a->v+j*lda,&one); 70a5ce6ee0Svictorle } 71a5ce6ee0Svictorle } else { 72d0f46423SBarry Smith nz = PetscBLASIntCast(A->rmap->n*A->cmap->n); 73f4df32b1SMatthew Knepley BLASscal_(&nz,&oalpha,a->v,&one); 74a5ce6ee0Svictorle } 75efee365bSSatish Balay ierr = PetscLogFlops(nz);CHKERRQ(ierr); 763a40ed3dSBarry Smith PetscFunctionReturn(0); 7780cd9d93SLois Curfman McInnes } 7880cd9d93SLois Curfman McInnes 791cbb95d3SBarry Smith #undef __FUNCT__ 801cbb95d3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqDense" 811cbb95d3SBarry Smith PetscErrorCode MatIsHermitian_SeqDense(Mat A,PetscReal rtol,PetscTruth *fl) 821cbb95d3SBarry Smith { 831cbb95d3SBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 84d0f46423SBarry Smith PetscInt i,j,m = A->rmap->n,N; 851cbb95d3SBarry Smith PetscScalar *v = a->v; 861cbb95d3SBarry Smith 871cbb95d3SBarry Smith PetscFunctionBegin; 881cbb95d3SBarry Smith *fl = PETSC_FALSE; 89d0f46423SBarry Smith if (A->rmap->n != A->cmap->n) PetscFunctionReturn(0); 901cbb95d3SBarry Smith N = a->lda; 911cbb95d3SBarry Smith 921cbb95d3SBarry Smith for (i=0; i<m; i++) { 931cbb95d3SBarry Smith for (j=i+1; j<m; j++) { 941cbb95d3SBarry Smith if (PetscAbsScalar(v[i+j*N] - PetscConj(v[j+i*N])) > rtol) PetscFunctionReturn(0); 951cbb95d3SBarry Smith } 961cbb95d3SBarry Smith } 971cbb95d3SBarry Smith *fl = PETSC_TRUE; 981cbb95d3SBarry Smith PetscFunctionReturn(0); 991cbb95d3SBarry Smith } 1001cbb95d3SBarry Smith 101b24902e0SBarry Smith #undef __FUNCT__ 102b24902e0SBarry Smith #define __FUNCT__ "MatDuplicateNoCreate_SeqDense" 103719d5645SBarry Smith PetscErrorCode MatDuplicateNoCreate_SeqDense(Mat newi,Mat A,MatDuplicateOption cpvalues) 104b24902e0SBarry Smith { 105b24902e0SBarry Smith Mat_SeqDense *mat = (Mat_SeqDense*)A->data,*l; 106b24902e0SBarry Smith PetscErrorCode ierr; 107b24902e0SBarry Smith PetscInt lda = (PetscInt)mat->lda,j,m; 108b24902e0SBarry Smith 109b24902e0SBarry Smith PetscFunctionBegin; 110719d5645SBarry Smith ierr = MatSeqDenseSetPreallocation(newi,PETSC_NULL);CHKERRQ(ierr); 111b24902e0SBarry Smith if (cpvalues == MAT_COPY_VALUES) { 112b24902e0SBarry Smith l = (Mat_SeqDense*)newi->data; 113d0f46423SBarry Smith if (lda>A->rmap->n) { 114d0f46423SBarry Smith m = A->rmap->n; 115d0f46423SBarry Smith for (j=0; j<A->cmap->n; j++) { 116b24902e0SBarry Smith ierr = PetscMemcpy(l->v+j*m,mat->v+j*lda,m*sizeof(PetscScalar));CHKERRQ(ierr); 117b24902e0SBarry Smith } 118b24902e0SBarry Smith } else { 119d0f46423SBarry Smith ierr = PetscMemcpy(l->v,mat->v,A->rmap->n*A->cmap->n*sizeof(PetscScalar));CHKERRQ(ierr); 120b24902e0SBarry Smith } 121b24902e0SBarry Smith } 122b24902e0SBarry Smith newi->assembled = PETSC_TRUE; 123b24902e0SBarry Smith PetscFunctionReturn(0); 124b24902e0SBarry Smith } 125b24902e0SBarry Smith 1264a2ae208SSatish Balay #undef __FUNCT__ 1274a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqDense" 128dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat) 12902cad45dSBarry Smith { 1306849ba73SBarry Smith PetscErrorCode ierr; 13102cad45dSBarry Smith 1323a40ed3dSBarry Smith PetscFunctionBegin; 1335c9eb25fSBarry Smith ierr = MatCreate(((PetscObject)A)->comm,newmat);CHKERRQ(ierr); 134d0f46423SBarry Smith ierr = MatSetSizes(*newmat,A->rmap->n,A->cmap->n,A->rmap->n,A->cmap->n);CHKERRQ(ierr); 1355c9eb25fSBarry Smith ierr = MatSetType(*newmat,((PetscObject)A)->type_name);CHKERRQ(ierr); 136719d5645SBarry Smith ierr = MatDuplicateNoCreate_SeqDense(*newmat,A,cpvalues);CHKERRQ(ierr); 137b24902e0SBarry Smith PetscFunctionReturn(0); 138b24902e0SBarry Smith } 139b24902e0SBarry Smith 1406ee01492SSatish Balay 1410481f469SBarry Smith extern PetscErrorCode MatLUFactor_SeqDense(Mat,IS,IS,const MatFactorInfo*); 142719d5645SBarry Smith 1434a2ae208SSatish Balay #undef __FUNCT__ 1444a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqDense" 1450481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqDense(Mat fact,Mat A,const MatFactorInfo *info_dummy) 146289bc588SBarry Smith { 1474482741eSBarry Smith MatFactorInfo info; 148a093e273SMatthew Knepley PetscErrorCode ierr; 1493a40ed3dSBarry Smith 1503a40ed3dSBarry Smith PetscFunctionBegin; 151c3ef05f6SHong Zhang ierr = MatDuplicateNoCreate_SeqDense(fact,A,MAT_COPY_VALUES);CHKERRQ(ierr); 152719d5645SBarry Smith ierr = MatLUFactor_SeqDense(fact,0,0,&info);CHKERRQ(ierr); 1533a40ed3dSBarry Smith PetscFunctionReturn(0); 154289bc588SBarry Smith } 1556ee01492SSatish Balay 1560b4b3355SBarry Smith #undef __FUNCT__ 1574a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqDense" 158dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqDense(Mat A,Vec xx,Vec yy) 159289bc588SBarry Smith { 160c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 1616849ba73SBarry Smith PetscErrorCode ierr; 16287828ca2SBarry Smith PetscScalar *x,*y; 163d0f46423SBarry Smith PetscBLASInt one = 1,info,m = PetscBLASIntCast(A->rmap->n); 16467e560aaSBarry Smith 1653a40ed3dSBarry Smith PetscFunctionBegin; 1661ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1671ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 168d0f46423SBarry Smith ierr = PetscMemcpy(y,x,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr); 1695c9eb25fSBarry Smith if (A->factor == MAT_FACTOR_LU) { 170ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS) 17180444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable."); 172ae7cfcebSSatish Balay #else 17371044d3cSBarry Smith LAPACKgetrs_("N",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info); 1744ce68768SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"GETRS - Bad solve"); 175ae7cfcebSSatish Balay #endif 1765c9eb25fSBarry Smith } else if (A->factor == MAT_FACTOR_CHOLESKY){ 177ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS) 17880444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable."); 179ae7cfcebSSatish Balay #else 18071044d3cSBarry Smith LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info); 1814ce68768SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"POTRS Bad solve"); 182ae7cfcebSSatish Balay #endif 183289bc588SBarry Smith } 18429bbc08cSBarry Smith else SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Matrix must be factored to solve"); 1851ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 1861ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 187dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*A->cmap->n*A->cmap->n - A->cmap->n);CHKERRQ(ierr); 1883a40ed3dSBarry Smith PetscFunctionReturn(0); 189289bc588SBarry Smith } 1906ee01492SSatish Balay 1914a2ae208SSatish Balay #undef __FUNCT__ 1924a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqDense" 193dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqDense(Mat A,Vec xx,Vec yy) 194da3a660dSBarry Smith { 195c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 196dfbe8321SBarry Smith PetscErrorCode ierr; 19787828ca2SBarry Smith PetscScalar *x,*y; 198d0f46423SBarry Smith PetscBLASInt one = 1,info,m = PetscBLASIntCast(A->rmap->n); 19967e560aaSBarry Smith 2003a40ed3dSBarry Smith PetscFunctionBegin; 2011ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2021ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 203d0f46423SBarry Smith ierr = PetscMemcpy(y,x,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr); 204752f5567SLois Curfman McInnes /* assume if pivots exist then use LU; else Cholesky */ 205da3a660dSBarry Smith if (mat->pivots) { 206ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS) 20780444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable."); 208ae7cfcebSSatish Balay #else 20971044d3cSBarry Smith LAPACKgetrs_("T",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info); 2104ce68768SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"POTRS - Bad solve"); 211ae7cfcebSSatish Balay #endif 2127a97a34bSBarry Smith } else { 213ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS) 21480444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable."); 215ae7cfcebSSatish Balay #else 21671044d3cSBarry Smith LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info); 2174ce68768SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"POTRS - Bad solve"); 218ae7cfcebSSatish Balay #endif 219da3a660dSBarry Smith } 2201ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2211ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 222dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*A->cmap->n*A->cmap->n - A->cmap->n);CHKERRQ(ierr); 2233a40ed3dSBarry Smith PetscFunctionReturn(0); 224da3a660dSBarry Smith } 2256ee01492SSatish Balay 2264a2ae208SSatish Balay #undef __FUNCT__ 2274a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqDense" 228dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy) 229da3a660dSBarry Smith { 230c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 231dfbe8321SBarry Smith PetscErrorCode ierr; 23287828ca2SBarry Smith PetscScalar *x,*y,sone = 1.0; 233da3a660dSBarry Smith Vec tmp = 0; 234d0f46423SBarry Smith PetscBLASInt one = 1,info,m = PetscBLASIntCast(A->rmap->n); 23567e560aaSBarry Smith 2363a40ed3dSBarry Smith PetscFunctionBegin; 2371ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2381ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 239d0f46423SBarry Smith if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0); 240da3a660dSBarry Smith if (yy == zz) { 24178b31e54SBarry Smith ierr = VecDuplicate(yy,&tmp);CHKERRQ(ierr); 24252e6d16bSBarry Smith ierr = PetscLogObjectParent(A,tmp);CHKERRQ(ierr); 24378b31e54SBarry Smith ierr = VecCopy(yy,tmp);CHKERRQ(ierr); 244da3a660dSBarry Smith } 245d0f46423SBarry Smith ierr = PetscMemcpy(y,x,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr); 246752f5567SLois Curfman McInnes /* assume if pivots exist then use LU; else Cholesky */ 247da3a660dSBarry Smith if (mat->pivots) { 248ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS) 24980444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable."); 250ae7cfcebSSatish Balay #else 25171044d3cSBarry Smith LAPACKgetrs_("N",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info); 2522ffef20aSBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve"); 253ae7cfcebSSatish Balay #endif 254a8c6a408SBarry Smith } else { 255ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS) 25680444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable."); 257ae7cfcebSSatish Balay #else 25871044d3cSBarry Smith LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info); 2592ffef20aSBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve"); 260ae7cfcebSSatish Balay #endif 261da3a660dSBarry Smith } 2622dcb1b2aSMatthew Knepley if (tmp) {ierr = VecAXPY(yy,sone,tmp);CHKERRQ(ierr); ierr = VecDestroy(tmp);CHKERRQ(ierr);} 2632dcb1b2aSMatthew Knepley else {ierr = VecAXPY(yy,sone,zz);CHKERRQ(ierr);} 2641ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2651ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 266dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*A->cmap->n*A->cmap->n);CHKERRQ(ierr); 2673a40ed3dSBarry Smith PetscFunctionReturn(0); 268da3a660dSBarry Smith } 26967e560aaSBarry Smith 2704a2ae208SSatish Balay #undef __FUNCT__ 2714a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqDense" 272dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy) 273da3a660dSBarry Smith { 274c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 2756849ba73SBarry Smith PetscErrorCode ierr; 27687828ca2SBarry Smith PetscScalar *x,*y,sone = 1.0; 277da3a660dSBarry Smith Vec tmp; 278d0f46423SBarry Smith PetscBLASInt one = 1,info,m = PetscBLASIntCast(A->rmap->n); 27967e560aaSBarry Smith 2803a40ed3dSBarry Smith PetscFunctionBegin; 281d0f46423SBarry Smith if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0); 2821ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2831ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 284da3a660dSBarry Smith if (yy == zz) { 28578b31e54SBarry Smith ierr = VecDuplicate(yy,&tmp);CHKERRQ(ierr); 28652e6d16bSBarry Smith ierr = PetscLogObjectParent(A,tmp);CHKERRQ(ierr); 28778b31e54SBarry Smith ierr = VecCopy(yy,tmp);CHKERRQ(ierr); 288da3a660dSBarry Smith } 289d0f46423SBarry Smith ierr = PetscMemcpy(y,x,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr); 290752f5567SLois Curfman McInnes /* assume if pivots exist then use LU; else Cholesky */ 291da3a660dSBarry Smith if (mat->pivots) { 292ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS) 29380444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable."); 294ae7cfcebSSatish Balay #else 29571044d3cSBarry Smith LAPACKgetrs_("T",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info); 2962ffef20aSBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve"); 297ae7cfcebSSatish Balay #endif 2983a40ed3dSBarry Smith } else { 299ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS) 30080444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable."); 301ae7cfcebSSatish Balay #else 30271044d3cSBarry Smith LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info); 3032ffef20aSBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve"); 304ae7cfcebSSatish Balay #endif 305da3a660dSBarry Smith } 30690f02eecSBarry Smith if (tmp) { 3072dcb1b2aSMatthew Knepley ierr = VecAXPY(yy,sone,tmp);CHKERRQ(ierr); 30890f02eecSBarry Smith ierr = VecDestroy(tmp);CHKERRQ(ierr); 3093a40ed3dSBarry Smith } else { 3102dcb1b2aSMatthew Knepley ierr = VecAXPY(yy,sone,zz);CHKERRQ(ierr); 31190f02eecSBarry Smith } 3121ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3131ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 314dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*A->cmap->n*A->cmap->n);CHKERRQ(ierr); 3153a40ed3dSBarry Smith PetscFunctionReturn(0); 316da3a660dSBarry Smith } 317db4efbfdSBarry Smith 318db4efbfdSBarry Smith /* ---------------------------------------------------------------*/ 319db4efbfdSBarry Smith /* COMMENT: I have chosen to hide row permutation in the pivots, 320db4efbfdSBarry Smith rather than put it in the Mat->row slot.*/ 321db4efbfdSBarry Smith #undef __FUNCT__ 322db4efbfdSBarry Smith #define __FUNCT__ "MatLUFactor_SeqDense" 3230481f469SBarry Smith PetscErrorCode MatLUFactor_SeqDense(Mat A,IS row,IS col,const MatFactorInfo *minfo) 324db4efbfdSBarry Smith { 325db4efbfdSBarry Smith #if defined(PETSC_MISSING_LAPACK_GETRF) 326db4efbfdSBarry Smith PetscFunctionBegin; 327db4efbfdSBarry Smith SETERRQ(PETSC_ERR_SUP,"GETRF - Lapack routine is unavailable."); 328db4efbfdSBarry Smith #else 329db4efbfdSBarry Smith Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 330db4efbfdSBarry Smith PetscErrorCode ierr; 331db4efbfdSBarry Smith PetscBLASInt n,m,info; 332db4efbfdSBarry Smith 333db4efbfdSBarry Smith PetscFunctionBegin; 334db4efbfdSBarry Smith n = PetscBLASIntCast(A->cmap->n); 335db4efbfdSBarry Smith m = PetscBLASIntCast(A->rmap->n); 336db4efbfdSBarry Smith if (!mat->pivots) { 337db4efbfdSBarry Smith ierr = PetscMalloc((A->rmap->n+1)*sizeof(PetscBLASInt),&mat->pivots);CHKERRQ(ierr); 338db4efbfdSBarry Smith ierr = PetscLogObjectMemory(A,A->rmap->n*sizeof(PetscBLASInt));CHKERRQ(ierr); 339db4efbfdSBarry Smith } 340db4efbfdSBarry Smith if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0); 341db4efbfdSBarry Smith LAPACKgetrf_(&m,&n,mat->v,&mat->lda,mat->pivots,&info); 342db4efbfdSBarry Smith if (info<0) SETERRQ(PETSC_ERR_LIB,"Bad argument to LU factorization"); 343db4efbfdSBarry Smith if (info>0) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,"Bad LU factorization"); 344db4efbfdSBarry Smith A->ops->solve = MatSolve_SeqDense; 345db4efbfdSBarry Smith A->ops->solvetranspose = MatSolveTranspose_SeqDense; 346db4efbfdSBarry Smith A->ops->solveadd = MatSolveAdd_SeqDense; 347db4efbfdSBarry Smith A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqDense; 348db4efbfdSBarry Smith A->factor = MAT_FACTOR_LU; 349db4efbfdSBarry Smith 350dc0b31edSSatish Balay ierr = PetscLogFlops((2.0*A->cmap->n*A->cmap->n*A->cmap->n)/3);CHKERRQ(ierr); 351db4efbfdSBarry Smith #endif 352db4efbfdSBarry Smith PetscFunctionReturn(0); 353db4efbfdSBarry Smith } 354db4efbfdSBarry Smith 355db4efbfdSBarry Smith #undef __FUNCT__ 356db4efbfdSBarry Smith #define __FUNCT__ "MatCholeskyFactor_SeqDense" 3570481f469SBarry Smith PetscErrorCode MatCholeskyFactor_SeqDense(Mat A,IS perm,const MatFactorInfo *factinfo) 358db4efbfdSBarry Smith { 359db4efbfdSBarry Smith #if defined(PETSC_MISSING_LAPACK_POTRF) 360db4efbfdSBarry Smith PetscFunctionBegin; 361db4efbfdSBarry Smith SETERRQ(PETSC_ERR_SUP,"POTRF - Lapack routine is unavailable."); 362db4efbfdSBarry Smith #else 363db4efbfdSBarry Smith Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 364db4efbfdSBarry Smith PetscErrorCode ierr; 365db4efbfdSBarry Smith PetscBLASInt info,n = PetscBLASIntCast(A->cmap->n); 366db4efbfdSBarry Smith 367db4efbfdSBarry Smith PetscFunctionBegin; 368db4efbfdSBarry Smith ierr = PetscFree(mat->pivots);CHKERRQ(ierr); 369db4efbfdSBarry Smith mat->pivots = 0; 370db4efbfdSBarry Smith 371db4efbfdSBarry Smith if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0); 372db4efbfdSBarry Smith LAPACKpotrf_("L",&n,mat->v,&mat->lda,&info); 373db4efbfdSBarry Smith if (info) SETERRQ1(PETSC_ERR_MAT_CH_ZRPVT,"Bad factorization: zero pivot in row %D",(PetscInt)info-1); 374db4efbfdSBarry Smith A->ops->solve = MatSolve_SeqDense; 375db4efbfdSBarry Smith A->ops->solvetranspose = MatSolveTranspose_SeqDense; 376db4efbfdSBarry Smith A->ops->solveadd = MatSolveAdd_SeqDense; 377db4efbfdSBarry Smith A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqDense; 378db4efbfdSBarry Smith A->factor = MAT_FACTOR_CHOLESKY; 379dc0b31edSSatish Balay ierr = PetscLogFlops((A->cmap->n*A->cmap->n*A->cmap->n)/3.0);CHKERRQ(ierr); 380db4efbfdSBarry Smith #endif 381db4efbfdSBarry Smith PetscFunctionReturn(0); 382db4efbfdSBarry Smith } 383db4efbfdSBarry Smith 384db4efbfdSBarry Smith 385db4efbfdSBarry Smith #undef __FUNCT__ 386db4efbfdSBarry Smith #define __FUNCT__ "MatCholeskyFactorNumeric_SeqDense" 3870481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_SeqDense(Mat fact,Mat A,const MatFactorInfo *info_dummy) 388db4efbfdSBarry Smith { 389db4efbfdSBarry Smith PetscErrorCode ierr; 390db4efbfdSBarry Smith MatFactorInfo info; 391db4efbfdSBarry Smith 392db4efbfdSBarry Smith PetscFunctionBegin; 393db4efbfdSBarry Smith info.fill = 1.0; 394c3ef05f6SHong Zhang ierr = MatDuplicateNoCreate_SeqDense(fact,A,MAT_COPY_VALUES);CHKERRQ(ierr); 395719d5645SBarry Smith ierr = MatCholeskyFactor_SeqDense(fact,0,&info);CHKERRQ(ierr); 396db4efbfdSBarry Smith PetscFunctionReturn(0); 397db4efbfdSBarry Smith } 398db4efbfdSBarry Smith 399db4efbfdSBarry Smith #undef __FUNCT__ 400db4efbfdSBarry Smith #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqDense" 4010481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqDense(Mat fact,Mat A,IS row,const MatFactorInfo *info) 402db4efbfdSBarry Smith { 403db4efbfdSBarry Smith PetscFunctionBegin; 404c3ef05f6SHong Zhang fact->assembled = PETSC_TRUE; 405719d5645SBarry Smith fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqDense; 406db4efbfdSBarry Smith PetscFunctionReturn(0); 407db4efbfdSBarry Smith } 408db4efbfdSBarry Smith 409db4efbfdSBarry Smith #undef __FUNCT__ 410db4efbfdSBarry Smith #define __FUNCT__ "MatLUFactorSymbolic_SeqDense" 4110481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqDense(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info) 412db4efbfdSBarry Smith { 413db4efbfdSBarry Smith PetscFunctionBegin; 414c3ef05f6SHong Zhang fact->assembled = PETSC_TRUE; 415719d5645SBarry Smith fact->ops->lufactornumeric = MatLUFactorNumeric_SeqDense; 416db4efbfdSBarry Smith PetscFunctionReturn(0); 417db4efbfdSBarry Smith } 418db4efbfdSBarry Smith 419bb5747d9SMatthew Knepley EXTERN_C_BEGIN 420db4efbfdSBarry Smith #undef __FUNCT__ 421db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactor_seqdense_petsc" 422db4efbfdSBarry Smith PetscErrorCode MatGetFactor_seqdense_petsc(Mat A,MatFactorType ftype,Mat *fact) 423db4efbfdSBarry Smith { 424db4efbfdSBarry Smith PetscErrorCode ierr; 425db4efbfdSBarry Smith 426db4efbfdSBarry Smith PetscFunctionBegin; 427db4efbfdSBarry Smith ierr = MatCreate(((PetscObject)A)->comm,fact);CHKERRQ(ierr); 428db4efbfdSBarry Smith ierr = MatSetSizes(*fact,A->rmap->n,A->cmap->n,A->rmap->n,A->cmap->n);CHKERRQ(ierr); 429db4efbfdSBarry Smith ierr = MatSetType(*fact,((PetscObject)A)->type_name);CHKERRQ(ierr); 430db4efbfdSBarry Smith if (ftype == MAT_FACTOR_LU){ 431db4efbfdSBarry Smith (*fact)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqDense; 432db4efbfdSBarry Smith } else { 433db4efbfdSBarry Smith (*fact)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqDense; 434db4efbfdSBarry Smith } 435db4efbfdSBarry Smith (*fact)->factor = ftype; 436db4efbfdSBarry Smith PetscFunctionReturn(0); 437db4efbfdSBarry Smith } 438bb5747d9SMatthew Knepley EXTERN_C_END 439db4efbfdSBarry Smith 440289bc588SBarry Smith /* ------------------------------------------------------------------*/ 4414a2ae208SSatish Balay #undef __FUNCT__ 44241f059aeSBarry Smith #define __FUNCT__ "MatSOR_SeqDense" 44341f059aeSBarry Smith PetscErrorCode MatSOR_SeqDense(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal shift,PetscInt its,PetscInt lits,Vec xx) 444289bc588SBarry Smith { 445c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 44687828ca2SBarry Smith PetscScalar *x,*b,*v = mat->v,zero = 0.0,xt; 447dfbe8321SBarry Smith PetscErrorCode ierr; 448d0f46423SBarry Smith PetscInt m = A->rmap->n,i; 449aa482453SBarry Smith #if !defined(PETSC_USE_COMPLEX) 4500805154bSBarry Smith PetscBLASInt o = 1,bm = PetscBLASIntCast(m); 451bc1b551cSSatish Balay #endif 452289bc588SBarry Smith 4533a40ed3dSBarry Smith PetscFunctionBegin; 454289bc588SBarry Smith if (flag & SOR_ZERO_INITIAL_GUESS) { 45571044d3cSBarry Smith /* this is a hack fix, should have another version without the second BLASdot */ 4562dcb1b2aSMatthew Knepley ierr = VecSet(xx,zero);CHKERRQ(ierr); 457289bc588SBarry Smith } 4581ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 4591ebc52fbSHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 460b965ef7fSBarry Smith its = its*lits; 46177431f27SBarry Smith if (its <= 0) SETERRQ2(PETSC_ERR_ARG_WRONG,"Relaxation requires global its %D and local its %D both positive",its,lits); 462289bc588SBarry Smith while (its--) { 463fccaa45eSBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){ 464289bc588SBarry Smith for (i=0; i<m; i++) { 465aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 466f1747703SBarry Smith /* cannot use BLAS dot for complex because compiler/linker is 467f1747703SBarry Smith not happy about returning a double complex */ 46813f74950SBarry Smith PetscInt _i; 46987828ca2SBarry Smith PetscScalar sum = b[i]; 470f1747703SBarry Smith for (_i=0; _i<m; _i++) { 4713f6de6efSSatish Balay sum -= PetscConj(v[i+_i*m])*x[_i]; 472f1747703SBarry Smith } 473f1747703SBarry Smith xt = sum; 474f1747703SBarry Smith #else 47571044d3cSBarry Smith xt = b[i] - BLASdot_(&bm,v+i,&bm,x,&o); 476f1747703SBarry Smith #endif 47755a1b374SBarry Smith x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift); 478289bc588SBarry Smith } 479289bc588SBarry Smith } 480fccaa45eSBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){ 481289bc588SBarry Smith for (i=m-1; i>=0; i--) { 482aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 483f1747703SBarry Smith /* cannot use BLAS dot for complex because compiler/linker is 484f1747703SBarry Smith not happy about returning a double complex */ 48513f74950SBarry Smith PetscInt _i; 48687828ca2SBarry Smith PetscScalar sum = b[i]; 487f1747703SBarry Smith for (_i=0; _i<m; _i++) { 4883f6de6efSSatish Balay sum -= PetscConj(v[i+_i*m])*x[_i]; 489f1747703SBarry Smith } 490f1747703SBarry Smith xt = sum; 491f1747703SBarry Smith #else 49271044d3cSBarry Smith xt = b[i] - BLASdot_(&bm,v+i,&bm,x,&o); 493f1747703SBarry Smith #endif 49455a1b374SBarry Smith x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift); 495289bc588SBarry Smith } 496289bc588SBarry Smith } 497289bc588SBarry Smith } 4981ebc52fbSHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 4991ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 5003a40ed3dSBarry Smith PetscFunctionReturn(0); 501289bc588SBarry Smith } 502289bc588SBarry Smith 503289bc588SBarry Smith /* -----------------------------------------------------------------*/ 5044a2ae208SSatish Balay #undef __FUNCT__ 5054a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_SeqDense" 506dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_SeqDense(Mat A,Vec xx,Vec yy) 507289bc588SBarry Smith { 508c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 50987828ca2SBarry Smith PetscScalar *v = mat->v,*x,*y; 510dfbe8321SBarry Smith PetscErrorCode ierr; 5110805154bSBarry Smith PetscBLASInt m, n,_One=1; 512ea709b57SSatish Balay PetscScalar _DOne=1.0,_DZero=0.0; 5133a40ed3dSBarry Smith 5143a40ed3dSBarry Smith PetscFunctionBegin; 515d0f46423SBarry Smith m = PetscBLASIntCast(A->rmap->n); 516d0f46423SBarry Smith n = PetscBLASIntCast(A->cmap->n); 517d0f46423SBarry Smith if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0); 5181ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 5191ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 52071044d3cSBarry Smith BLASgemv_("T",&m,&n,&_DOne,v,&mat->lda,x,&_One,&_DZero,y,&_One); 5211ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 5221ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 523dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*A->rmap->n*A->cmap->n - A->cmap->n);CHKERRQ(ierr); 5243a40ed3dSBarry Smith PetscFunctionReturn(0); 525289bc588SBarry Smith } 526*61de5317SMatthew Knepley #include <iostream> 5274a2ae208SSatish Balay #undef __FUNCT__ 5284a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqDense" 529dfbe8321SBarry Smith PetscErrorCode MatMult_SeqDense(Mat A,Vec xx,Vec yy) 530289bc588SBarry Smith { 531c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 53287828ca2SBarry Smith PetscScalar *v = mat->v,*x,*y,_DOne=1.0,_DZero=0.0; 533dfbe8321SBarry Smith PetscErrorCode ierr; 5340805154bSBarry Smith PetscBLASInt m, n, _One=1; 5353a40ed3dSBarry Smith 5363a40ed3dSBarry Smith PetscFunctionBegin; 537d0f46423SBarry Smith m = PetscBLASIntCast(A->rmap->n); 538d0f46423SBarry Smith n = PetscBLASIntCast(A->cmap->n); 539d0f46423SBarry Smith if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0); 5401ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 5411ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 542*61de5317SMatthew Knepley #if 0 54371044d3cSBarry Smith BLASgemv_("N",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DZero,y,&_One); 544*61de5317SMatthew Knepley { 545*61de5317SMatthew Knepley PetscScalar *yCheck = new PetscScalar[m]; 546*61de5317SMatthew Knepley for(int i = 0; i < m; ++i) { 547*61de5317SMatthew Knepley yCheck[i] = 0.0; 548*61de5317SMatthew Knepley for(int j = 0; j < n; ++j) { 549*61de5317SMatthew Knepley yCheck[i] += v[i*n+j]*x[j]; 550*61de5317SMatthew Knepley } 551*61de5317SMatthew Knepley if (fabs(y[i] - yCheck[i]) > 1.0e-8) { 552*61de5317SMatthew Knepley SETERRQ2(PETSC_ERR_LIB, "DGEMV call is fucked up: %g should be %g", y[i], yCheck[i]); 553*61de5317SMatthew Knepley } 554*61de5317SMatthew Knepley } 555*61de5317SMatthew Knepley delete [] y; 556*61de5317SMatthew Knepley } 557*61de5317SMatthew Knepley #else 558*61de5317SMatthew Knepley for(int i = 0; i < m; ++i) { 559*61de5317SMatthew Knepley y[i] = 0.0; 560*61de5317SMatthew Knepley for(int j = 0; j < n; ++j) { 561*61de5317SMatthew Knepley y[i] += v[i*n+j]*x[j]; 562*61de5317SMatthew Knepley } 563*61de5317SMatthew Knepley } 564*61de5317SMatthew Knepley #endif 5651ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 5661ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 567dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*A->rmap->n*A->cmap->n - A->rmap->n);CHKERRQ(ierr); 5683a40ed3dSBarry Smith PetscFunctionReturn(0); 569289bc588SBarry Smith } 5706ee01492SSatish Balay 5714a2ae208SSatish Balay #undef __FUNCT__ 5724a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqDense" 573dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy) 574289bc588SBarry Smith { 575c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 57687828ca2SBarry Smith PetscScalar *v = mat->v,*x,*y,_DOne=1.0; 577dfbe8321SBarry Smith PetscErrorCode ierr; 5780805154bSBarry Smith PetscBLASInt m, n, _One=1; 5793a40ed3dSBarry Smith 5803a40ed3dSBarry Smith PetscFunctionBegin; 581d0f46423SBarry Smith m = PetscBLASIntCast(A->rmap->n); 582d0f46423SBarry Smith n = PetscBLASIntCast(A->cmap->n); 583d0f46423SBarry Smith if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0); 584600d6d0bSBarry Smith if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);} 5851ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 5861ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 58771044d3cSBarry Smith BLASgemv_("N",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DOne,y,&_One); 5881ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 5891ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 590dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*A->rmap->n*A->cmap->n);CHKERRQ(ierr); 5913a40ed3dSBarry Smith PetscFunctionReturn(0); 592289bc588SBarry Smith } 5936ee01492SSatish Balay 5944a2ae208SSatish Balay #undef __FUNCT__ 5954a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_SeqDense" 596dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy) 597289bc588SBarry Smith { 598c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 59987828ca2SBarry Smith PetscScalar *v = mat->v,*x,*y; 600dfbe8321SBarry Smith PetscErrorCode ierr; 6010805154bSBarry Smith PetscBLASInt m, n, _One=1; 60287828ca2SBarry Smith PetscScalar _DOne=1.0; 6033a40ed3dSBarry Smith 6043a40ed3dSBarry Smith PetscFunctionBegin; 605d0f46423SBarry Smith m = PetscBLASIntCast(A->rmap->n); 606d0f46423SBarry Smith n = PetscBLASIntCast(A->cmap->n); 607d0f46423SBarry Smith if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0); 608600d6d0bSBarry Smith if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);} 6091ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 6101ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 61171044d3cSBarry Smith BLASgemv_("T",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DOne,y,&_One); 6121ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 6131ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 614dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*A->rmap->n*A->cmap->n);CHKERRQ(ierr); 6153a40ed3dSBarry Smith PetscFunctionReturn(0); 616289bc588SBarry Smith } 617289bc588SBarry Smith 618289bc588SBarry Smith /* -----------------------------------------------------------------*/ 6194a2ae208SSatish Balay #undef __FUNCT__ 6204a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqDense" 62113f74950SBarry Smith PetscErrorCode MatGetRow_SeqDense(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **vals) 622289bc588SBarry Smith { 623c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 62487828ca2SBarry Smith PetscScalar *v; 6256849ba73SBarry Smith PetscErrorCode ierr; 62613f74950SBarry Smith PetscInt i; 62767e560aaSBarry Smith 6283a40ed3dSBarry Smith PetscFunctionBegin; 629d0f46423SBarry Smith *ncols = A->cmap->n; 630289bc588SBarry Smith if (cols) { 631d0f46423SBarry Smith ierr = PetscMalloc((A->cmap->n+1)*sizeof(PetscInt),cols);CHKERRQ(ierr); 632d0f46423SBarry Smith for (i=0; i<A->cmap->n; i++) (*cols)[i] = i; 633289bc588SBarry Smith } 634289bc588SBarry Smith if (vals) { 635d0f46423SBarry Smith ierr = PetscMalloc((A->cmap->n+1)*sizeof(PetscScalar),vals);CHKERRQ(ierr); 636289bc588SBarry Smith v = mat->v + row; 637d0f46423SBarry Smith for (i=0; i<A->cmap->n; i++) {(*vals)[i] = *v; v += mat->lda;} 638289bc588SBarry Smith } 6393a40ed3dSBarry Smith PetscFunctionReturn(0); 640289bc588SBarry Smith } 6416ee01492SSatish Balay 6424a2ae208SSatish Balay #undef __FUNCT__ 6434a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqDense" 64413f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqDense(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **vals) 645289bc588SBarry Smith { 646dfbe8321SBarry Smith PetscErrorCode ierr; 647606d414cSSatish Balay PetscFunctionBegin; 648606d414cSSatish Balay if (cols) {ierr = PetscFree(*cols);CHKERRQ(ierr);} 649606d414cSSatish Balay if (vals) {ierr = PetscFree(*vals);CHKERRQ(ierr); } 6503a40ed3dSBarry Smith PetscFunctionReturn(0); 651289bc588SBarry Smith } 652289bc588SBarry Smith /* ----------------------------------------------------------------*/ 6534a2ae208SSatish Balay #undef __FUNCT__ 6544a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqDense" 65513f74950SBarry Smith PetscErrorCode MatSetValues_SeqDense(Mat A,PetscInt m,const PetscInt indexm[],PetscInt n,const PetscInt indexn[],const PetscScalar v[],InsertMode addv) 656289bc588SBarry Smith { 657c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 65813f74950SBarry Smith PetscInt i,j,idx=0; 659d6dfbf8fSBarry Smith 6603a40ed3dSBarry Smith PetscFunctionBegin; 66171fd2e92SBarry Smith if (v) PetscValidScalarPointer(v,6); 662289bc588SBarry Smith if (!mat->roworiented) { 663dbb450caSBarry Smith if (addv == INSERT_VALUES) { 664289bc588SBarry Smith for (j=0; j<n; j++) { 665cddbea37SSatish Balay if (indexn[j] < 0) {idx += m; continue;} 6662515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 667d0f46423SBarry Smith if (indexn[j] >= A->cmap->n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",indexn[j],A->cmap->n-1); 66858804f6dSBarry Smith #endif 669289bc588SBarry Smith for (i=0; i<m; i++) { 670cddbea37SSatish Balay if (indexm[i] < 0) {idx++; continue;} 6712515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 672d0f46423SBarry Smith if (indexm[i] >= A->rmap->n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",indexm[i],A->rmap->n-1); 67358804f6dSBarry Smith #endif 674cddbea37SSatish Balay mat->v[indexn[j]*mat->lda + indexm[i]] = v[idx++]; 675289bc588SBarry Smith } 676289bc588SBarry Smith } 6773a40ed3dSBarry Smith } else { 678289bc588SBarry Smith for (j=0; j<n; j++) { 679cddbea37SSatish Balay if (indexn[j] < 0) {idx += m; continue;} 6802515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 681d0f46423SBarry Smith if (indexn[j] >= A->cmap->n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",indexn[j],A->cmap->n-1); 68258804f6dSBarry Smith #endif 683289bc588SBarry Smith for (i=0; i<m; i++) { 684cddbea37SSatish Balay if (indexm[i] < 0) {idx++; continue;} 6852515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 686d0f46423SBarry Smith if (indexm[i] >= A->rmap->n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",indexm[i],A->rmap->n-1); 68758804f6dSBarry Smith #endif 688cddbea37SSatish Balay mat->v[indexn[j]*mat->lda + indexm[i]] += v[idx++]; 689289bc588SBarry Smith } 690289bc588SBarry Smith } 691289bc588SBarry Smith } 6923a40ed3dSBarry Smith } else { 693dbb450caSBarry Smith if (addv == INSERT_VALUES) { 694e8d4e0b9SBarry Smith for (i=0; i<m; i++) { 695cddbea37SSatish Balay if (indexm[i] < 0) { idx += n; continue;} 6962515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 697d0f46423SBarry Smith if (indexm[i] >= A->rmap->n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",indexm[i],A->rmap->n-1); 69858804f6dSBarry Smith #endif 699e8d4e0b9SBarry Smith for (j=0; j<n; j++) { 700cddbea37SSatish Balay if (indexn[j] < 0) { idx++; continue;} 7012515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 702d0f46423SBarry Smith if (indexn[j] >= A->cmap->n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",indexn[j],A->cmap->n-1); 70358804f6dSBarry Smith #endif 704cddbea37SSatish Balay mat->v[indexn[j]*mat->lda + indexm[i]] = v[idx++]; 705e8d4e0b9SBarry Smith } 706e8d4e0b9SBarry Smith } 7073a40ed3dSBarry Smith } else { 708289bc588SBarry Smith for (i=0; i<m; i++) { 709cddbea37SSatish Balay if (indexm[i] < 0) { idx += n; continue;} 7102515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 711d0f46423SBarry Smith if (indexm[i] >= A->rmap->n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",indexm[i],A->rmap->n-1); 71258804f6dSBarry Smith #endif 713289bc588SBarry Smith for (j=0; j<n; j++) { 714cddbea37SSatish Balay if (indexn[j] < 0) { idx++; continue;} 7152515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 716d0f46423SBarry Smith if (indexn[j] >= A->cmap->n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",indexn[j],A->cmap->n-1); 71758804f6dSBarry Smith #endif 718cddbea37SSatish Balay mat->v[indexn[j]*mat->lda + indexm[i]] += v[idx++]; 719289bc588SBarry Smith } 720289bc588SBarry Smith } 721289bc588SBarry Smith } 722e8d4e0b9SBarry Smith } 7233a40ed3dSBarry Smith PetscFunctionReturn(0); 724289bc588SBarry Smith } 725e8d4e0b9SBarry Smith 7264a2ae208SSatish Balay #undef __FUNCT__ 7274a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqDense" 72813f74950SBarry Smith PetscErrorCode MatGetValues_SeqDense(Mat A,PetscInt m,const PetscInt indexm[],PetscInt n,const PetscInt indexn[],PetscScalar v[]) 729ae80bb75SLois Curfman McInnes { 730ae80bb75SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 73113f74950SBarry Smith PetscInt i,j; 732ae80bb75SLois Curfman McInnes 7333a40ed3dSBarry Smith PetscFunctionBegin; 734ae80bb75SLois Curfman McInnes /* row-oriented output */ 735ae80bb75SLois Curfman McInnes for (i=0; i<m; i++) { 73697e567efSBarry Smith if (indexm[i] < 0) {v += n;continue;} 737d0f46423SBarry Smith if (indexm[i] >= A->rmap->n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row %D requested larger than number rows %D",indexm[i],A->rmap->n); 738ae80bb75SLois Curfman McInnes for (j=0; j<n; j++) { 7396f31f424SBarry Smith if (indexn[j] < 0) {v++; continue;} 740d0f46423SBarry Smith if (indexn[j] >= A->cmap->n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column %D requested larger than number columns %D",indexn[j],A->cmap->n); 74197e567efSBarry Smith *v++ = mat->v[indexn[j]*mat->lda + indexm[i]]; 742ae80bb75SLois Curfman McInnes } 743ae80bb75SLois Curfman McInnes } 7443a40ed3dSBarry Smith PetscFunctionReturn(0); 745ae80bb75SLois Curfman McInnes } 746ae80bb75SLois Curfman McInnes 747289bc588SBarry Smith /* -----------------------------------------------------------------*/ 748289bc588SBarry Smith 749e090d566SSatish Balay #include "petscsys.h" 75088e32edaSLois Curfman McInnes 7514a2ae208SSatish Balay #undef __FUNCT__ 7524a2ae208SSatish Balay #define __FUNCT__ "MatLoad_SeqDense" 753a313700dSBarry Smith PetscErrorCode MatLoad_SeqDense(PetscViewer viewer, const MatType type,Mat *A) 75488e32edaSLois Curfman McInnes { 75588e32edaSLois Curfman McInnes Mat_SeqDense *a; 75688e32edaSLois Curfman McInnes Mat B; 7576849ba73SBarry Smith PetscErrorCode ierr; 75813f74950SBarry Smith PetscInt *scols,i,j,nz,header[4]; 75913f74950SBarry Smith int fd; 76013f74950SBarry Smith PetscMPIInt size; 76113f74950SBarry Smith PetscInt *rowlengths = 0,M,N,*cols; 76287828ca2SBarry Smith PetscScalar *vals,*svals,*v,*w; 76319bcc07fSBarry Smith MPI_Comm comm = ((PetscObject)viewer)->comm; 76488e32edaSLois Curfman McInnes 7653a40ed3dSBarry Smith PetscFunctionBegin; 766d132466eSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 76729bbc08cSBarry Smith if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"view must have one processor"); 768b0a32e0cSBarry Smith ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 7690752156aSBarry Smith ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr); 770552e946dSBarry Smith if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"Not matrix object"); 77188e32edaSLois Curfman McInnes M = header[1]; N = header[2]; nz = header[3]; 77288e32edaSLois Curfman McInnes 77390ace30eSBarry Smith if (nz == MATRIX_BINARY_FORMAT_DENSE) { /* matrix in file is dense */ 774f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 775f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,M,N,M,N);CHKERRQ(ierr); 776be5d1d56SKris Buschelman ierr = MatSetType(*A,type);CHKERRQ(ierr); 7775c5985e7SKris Buschelman ierr = MatSeqDenseSetPreallocation(*A,PETSC_NULL);CHKERRQ(ierr); 77890ace30eSBarry Smith B = *A; 77990ace30eSBarry Smith a = (Mat_SeqDense*)B->data; 7804a41a4d3SSatish Balay v = a->v; 7814a41a4d3SSatish Balay /* Allocate some temp space to read in the values and then flip them 7824a41a4d3SSatish Balay from row major to column major */ 783b72907f3SBarry Smith ierr = PetscMalloc((M*N > 0 ? M*N : 1)*sizeof(PetscScalar),&w);CHKERRQ(ierr); 78490ace30eSBarry Smith /* read in nonzero values */ 7854a41a4d3SSatish Balay ierr = PetscBinaryRead(fd,w,M*N,PETSC_SCALAR);CHKERRQ(ierr); 7864a41a4d3SSatish Balay /* now flip the values and store them in the matrix*/ 7874a41a4d3SSatish Balay for (j=0; j<N; j++) { 7884a41a4d3SSatish Balay for (i=0; i<M; i++) { 7894a41a4d3SSatish Balay *v++ =w[i*N+j]; 7904a41a4d3SSatish Balay } 7914a41a4d3SSatish Balay } 792606d414cSSatish Balay ierr = PetscFree(w);CHKERRQ(ierr); 7936d4a8577SBarry Smith ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 7946d4a8577SBarry Smith ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 79590ace30eSBarry Smith } else { 79688e32edaSLois Curfman McInnes /* read row lengths */ 79713f74950SBarry Smith ierr = PetscMalloc((M+1)*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr); 7980752156aSBarry Smith ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr); 79988e32edaSLois Curfman McInnes 80088e32edaSLois Curfman McInnes /* create our matrix */ 801f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 802f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,M,N,M,N);CHKERRQ(ierr); 803be5d1d56SKris Buschelman ierr = MatSetType(*A,type);CHKERRQ(ierr); 8045c5985e7SKris Buschelman ierr = MatSeqDenseSetPreallocation(*A,PETSC_NULL);CHKERRQ(ierr); 80588e32edaSLois Curfman McInnes B = *A; 80688e32edaSLois Curfman McInnes a = (Mat_SeqDense*)B->data; 80788e32edaSLois Curfman McInnes v = a->v; 80888e32edaSLois Curfman McInnes 80988e32edaSLois Curfman McInnes /* read column indices and nonzeros */ 81013f74950SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscInt),&scols);CHKERRQ(ierr); 811b0a32e0cSBarry Smith cols = scols; 8120752156aSBarry Smith ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr); 81387828ca2SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&svals);CHKERRQ(ierr); 814b0a32e0cSBarry Smith vals = svals; 8150752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr); 81688e32edaSLois Curfman McInnes 81788e32edaSLois Curfman McInnes /* insert into matrix */ 81888e32edaSLois Curfman McInnes for (i=0; i<M; i++) { 81988e32edaSLois Curfman McInnes for (j=0; j<rowlengths[i]; j++) v[i+M*scols[j]] = svals[j]; 82088e32edaSLois Curfman McInnes svals += rowlengths[i]; scols += rowlengths[i]; 82188e32edaSLois Curfman McInnes } 822606d414cSSatish Balay ierr = PetscFree(vals);CHKERRQ(ierr); 823606d414cSSatish Balay ierr = PetscFree(cols);CHKERRQ(ierr); 824606d414cSSatish Balay ierr = PetscFree(rowlengths);CHKERRQ(ierr); 82588e32edaSLois Curfman McInnes 8266d4a8577SBarry Smith ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 8276d4a8577SBarry Smith ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 82890ace30eSBarry Smith } 8293a40ed3dSBarry Smith PetscFunctionReturn(0); 83088e32edaSLois Curfman McInnes } 83188e32edaSLois Curfman McInnes 832e090d566SSatish Balay #include "petscsys.h" 833932b0c3eSLois Curfman McInnes 8344a2ae208SSatish Balay #undef __FUNCT__ 8354a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_ASCII" 8366849ba73SBarry Smith static PetscErrorCode MatView_SeqDense_ASCII(Mat A,PetscViewer viewer) 837289bc588SBarry Smith { 838932b0c3eSLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense*)A->data; 839dfbe8321SBarry Smith PetscErrorCode ierr; 84013f74950SBarry Smith PetscInt i,j; 8412dcb1b2aSMatthew Knepley const char *name; 84287828ca2SBarry Smith PetscScalar *v; 843f3ef73ceSBarry Smith PetscViewerFormat format; 8445f481a85SSatish Balay #if defined(PETSC_USE_COMPLEX) 8455f481a85SSatish Balay PetscTruth allreal = PETSC_TRUE; 8465f481a85SSatish Balay #endif 847932b0c3eSLois Curfman McInnes 8483a40ed3dSBarry Smith PetscFunctionBegin; 849435da068SBarry Smith ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr); 850b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 851456192e2SBarry Smith if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 8523a40ed3dSBarry Smith PetscFunctionReturn(0); /* do nothing for now */ 853fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_COMMON) { 854b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr); 855d0f46423SBarry Smith for (i=0; i<A->rmap->n; i++) { 85644cd7ae7SLois Curfman McInnes v = a->v + i; 85777431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr); 858d0f46423SBarry Smith for (j=0; j<A->cmap->n; j++) { 859aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 860329f5518SBarry Smith if (PetscRealPart(*v) != 0.0 && PetscImaginaryPart(*v) != 0.0) { 861a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",j,PetscRealPart(*v),PetscImaginaryPart(*v));CHKERRQ(ierr); 862329f5518SBarry Smith } else if (PetscRealPart(*v)) { 863a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",j,PetscRealPart(*v));CHKERRQ(ierr); 8646831982aSBarry Smith } 86580cd9d93SLois Curfman McInnes #else 8666831982aSBarry Smith if (*v) { 867a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",j,*v);CHKERRQ(ierr); 8686831982aSBarry Smith } 86980cd9d93SLois Curfman McInnes #endif 8701b807ce4Svictorle v += a->lda; 87180cd9d93SLois Curfman McInnes } 872b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 87380cd9d93SLois Curfman McInnes } 874b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr); 8753a40ed3dSBarry Smith } else { 876b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr); 877aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 87847989497SBarry Smith /* determine if matrix has all real values */ 87947989497SBarry Smith v = a->v; 880d0f46423SBarry Smith for (i=0; i<A->rmap->n*A->cmap->n; i++) { 881ffac6cdbSBarry Smith if (PetscImaginaryPart(v[i])) { allreal = PETSC_FALSE; break ;} 88247989497SBarry Smith } 88347989497SBarry Smith #endif 884fb9695e5SSatish Balay if (format == PETSC_VIEWER_ASCII_MATLAB) { 8853a7fca6bSBarry Smith ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr); 886d0f46423SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%% Size = %D %D \n",A->rmap->n,A->cmap->n);CHKERRQ(ierr); 887d0f46423SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%s = zeros(%D,%D);\n",name,A->rmap->n,A->cmap->n);CHKERRQ(ierr); 888fb9695e5SSatish Balay ierr = PetscViewerASCIIPrintf(viewer,"%s = [\n",name);CHKERRQ(ierr); 889ffac6cdbSBarry Smith } 890ffac6cdbSBarry Smith 891d0f46423SBarry Smith for (i=0; i<A->rmap->n; i++) { 892932b0c3eSLois Curfman McInnes v = a->v + i; 893d0f46423SBarry Smith for (j=0; j<A->cmap->n; j++) { 894aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 89547989497SBarry Smith if (allreal) { 896f32d5d43SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%18.16e ",PetscRealPart(*v));CHKERRQ(ierr); 89747989497SBarry Smith } else { 898f32d5d43SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%18.16e + %18.16e i ",PetscRealPart(*v),PetscImaginaryPart(*v));CHKERRQ(ierr); 89947989497SBarry Smith } 900289bc588SBarry Smith #else 901f32d5d43SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%18.16e ",*v);CHKERRQ(ierr); 902289bc588SBarry Smith #endif 9031b807ce4Svictorle v += a->lda; 904289bc588SBarry Smith } 905b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 906289bc588SBarry Smith } 907fb9695e5SSatish Balay if (format == PETSC_VIEWER_ASCII_MATLAB) { 908b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"];\n");CHKERRQ(ierr); 909ffac6cdbSBarry Smith } 910b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr); 911da3a660dSBarry Smith } 912b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 9133a40ed3dSBarry Smith PetscFunctionReturn(0); 914289bc588SBarry Smith } 915289bc588SBarry Smith 9164a2ae208SSatish Balay #undef __FUNCT__ 9174a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Binary" 9186849ba73SBarry Smith static PetscErrorCode MatView_SeqDense_Binary(Mat A,PetscViewer viewer) 919932b0c3eSLois Curfman McInnes { 920932b0c3eSLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense*)A->data; 9216849ba73SBarry Smith PetscErrorCode ierr; 92213f74950SBarry Smith int fd; 923d0f46423SBarry Smith PetscInt ict,j,n = A->cmap->n,m = A->rmap->n,i,*col_lens,nz = m*n; 92487828ca2SBarry Smith PetscScalar *v,*anonz,*vals; 925f3ef73ceSBarry Smith PetscViewerFormat format; 926932b0c3eSLois Curfman McInnes 9273a40ed3dSBarry Smith PetscFunctionBegin; 928b0a32e0cSBarry Smith ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 92990ace30eSBarry Smith 930b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 9317973ad04SBarry Smith if (format == PETSC_VIEWER_NATIVE) { 93290ace30eSBarry Smith /* store the matrix as a dense matrix */ 93313f74950SBarry Smith ierr = PetscMalloc(4*sizeof(PetscInt),&col_lens);CHKERRQ(ierr); 934552e946dSBarry Smith col_lens[0] = MAT_FILE_COOKIE; 93590ace30eSBarry Smith col_lens[1] = m; 93690ace30eSBarry Smith col_lens[2] = n; 93790ace30eSBarry Smith col_lens[3] = MATRIX_BINARY_FORMAT_DENSE; 9386f69ff64SBarry Smith ierr = PetscBinaryWrite(fd,col_lens,4,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr); 939606d414cSSatish Balay ierr = PetscFree(col_lens);CHKERRQ(ierr); 94090ace30eSBarry Smith 94190ace30eSBarry Smith /* write out matrix, by rows */ 94287828ca2SBarry Smith ierr = PetscMalloc((m*n+1)*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 94390ace30eSBarry Smith v = a->v; 94490ace30eSBarry Smith for (j=0; j<n; j++) { 945578230a0SSatish Balay for (i=0; i<m; i++) { 946578230a0SSatish Balay vals[j + i*n] = *v++; 94790ace30eSBarry Smith } 94890ace30eSBarry Smith } 9496f69ff64SBarry Smith ierr = PetscBinaryWrite(fd,vals,n*m,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 950606d414cSSatish Balay ierr = PetscFree(vals);CHKERRQ(ierr); 95190ace30eSBarry Smith } else { 95213f74950SBarry Smith ierr = PetscMalloc((4+nz)*sizeof(PetscInt),&col_lens);CHKERRQ(ierr); 953552e946dSBarry Smith col_lens[0] = MAT_FILE_COOKIE; 954932b0c3eSLois Curfman McInnes col_lens[1] = m; 955932b0c3eSLois Curfman McInnes col_lens[2] = n; 956932b0c3eSLois Curfman McInnes col_lens[3] = nz; 957932b0c3eSLois Curfman McInnes 958932b0c3eSLois Curfman McInnes /* store lengths of each row and write (including header) to file */ 959932b0c3eSLois Curfman McInnes for (i=0; i<m; i++) col_lens[4+i] = n; 9606f69ff64SBarry Smith ierr = PetscBinaryWrite(fd,col_lens,4+m,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr); 961932b0c3eSLois Curfman McInnes 962932b0c3eSLois Curfman McInnes /* Possibly should write in smaller increments, not whole matrix at once? */ 963932b0c3eSLois Curfman McInnes /* store column indices (zero start index) */ 964932b0c3eSLois Curfman McInnes ict = 0; 965932b0c3eSLois Curfman McInnes for (i=0; i<m; i++) { 966932b0c3eSLois Curfman McInnes for (j=0; j<n; j++) col_lens[ict++] = j; 967932b0c3eSLois Curfman McInnes } 9686f69ff64SBarry Smith ierr = PetscBinaryWrite(fd,col_lens,nz,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr); 969606d414cSSatish Balay ierr = PetscFree(col_lens);CHKERRQ(ierr); 970932b0c3eSLois Curfman McInnes 971932b0c3eSLois Curfman McInnes /* store nonzero values */ 97287828ca2SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&anonz);CHKERRQ(ierr); 973932b0c3eSLois Curfman McInnes ict = 0; 974932b0c3eSLois Curfman McInnes for (i=0; i<m; i++) { 975932b0c3eSLois Curfman McInnes v = a->v + i; 976932b0c3eSLois Curfman McInnes for (j=0; j<n; j++) { 9771b807ce4Svictorle anonz[ict++] = *v; v += a->lda; 978932b0c3eSLois Curfman McInnes } 979932b0c3eSLois Curfman McInnes } 9806f69ff64SBarry Smith ierr = PetscBinaryWrite(fd,anonz,nz,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 981606d414cSSatish Balay ierr = PetscFree(anonz);CHKERRQ(ierr); 98290ace30eSBarry Smith } 9833a40ed3dSBarry Smith PetscFunctionReturn(0); 984932b0c3eSLois Curfman McInnes } 985932b0c3eSLois Curfman McInnes 9864a2ae208SSatish Balay #undef __FUNCT__ 9874a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Draw_Zoom" 988dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense_Draw_Zoom(PetscDraw draw,void *Aa) 989f1af5d2fSBarry Smith { 990f1af5d2fSBarry Smith Mat A = (Mat) Aa; 991f1af5d2fSBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 9926849ba73SBarry Smith PetscErrorCode ierr; 993d0f46423SBarry Smith PetscInt m = A->rmap->n,n = A->cmap->n,color,i,j; 99487828ca2SBarry Smith PetscScalar *v = a->v; 995b0a32e0cSBarry Smith PetscViewer viewer; 996b0a32e0cSBarry Smith PetscDraw popup; 997329f5518SBarry Smith PetscReal xl,yl,xr,yr,x_l,x_r,y_l,y_r,scale,maxv = 0.0; 998f3ef73ceSBarry Smith PetscViewerFormat format; 999f1af5d2fSBarry Smith 1000f1af5d2fSBarry Smith PetscFunctionBegin; 1001f1af5d2fSBarry Smith 1002f1af5d2fSBarry Smith ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr); 1003b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 1004b0a32e0cSBarry Smith ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr); 1005f1af5d2fSBarry Smith 1006f1af5d2fSBarry Smith /* Loop over matrix elements drawing boxes */ 1007fb9695e5SSatish Balay if (format != PETSC_VIEWER_DRAW_CONTOUR) { 1008f1af5d2fSBarry Smith /* Blue for negative and Red for positive */ 1009b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 1010f1af5d2fSBarry Smith for(j = 0; j < n; j++) { 1011f1af5d2fSBarry Smith x_l = j; 1012f1af5d2fSBarry Smith x_r = x_l + 1.0; 1013f1af5d2fSBarry Smith for(i = 0; i < m; i++) { 1014f1af5d2fSBarry Smith y_l = m - i - 1.0; 1015f1af5d2fSBarry Smith y_r = y_l + 1.0; 1016f1af5d2fSBarry Smith #if defined(PETSC_USE_COMPLEX) 1017329f5518SBarry Smith if (PetscRealPart(v[j*m+i]) > 0.) { 1018b0a32e0cSBarry Smith color = PETSC_DRAW_RED; 1019329f5518SBarry Smith } else if (PetscRealPart(v[j*m+i]) < 0.) { 1020b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 1021f1af5d2fSBarry Smith } else { 1022f1af5d2fSBarry Smith continue; 1023f1af5d2fSBarry Smith } 1024f1af5d2fSBarry Smith #else 1025f1af5d2fSBarry Smith if (v[j*m+i] > 0.) { 1026b0a32e0cSBarry Smith color = PETSC_DRAW_RED; 1027f1af5d2fSBarry Smith } else if (v[j*m+i] < 0.) { 1028b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 1029f1af5d2fSBarry Smith } else { 1030f1af5d2fSBarry Smith continue; 1031f1af5d2fSBarry Smith } 1032f1af5d2fSBarry Smith #endif 1033b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 1034f1af5d2fSBarry Smith } 1035f1af5d2fSBarry Smith } 1036f1af5d2fSBarry Smith } else { 1037f1af5d2fSBarry Smith /* use contour shading to indicate magnitude of values */ 1038f1af5d2fSBarry Smith /* first determine max of all nonzero values */ 1039f1af5d2fSBarry Smith for(i = 0; i < m*n; i++) { 1040f1af5d2fSBarry Smith if (PetscAbsScalar(v[i]) > maxv) maxv = PetscAbsScalar(v[i]); 1041f1af5d2fSBarry Smith } 1042b0a32e0cSBarry Smith scale = (245.0 - PETSC_DRAW_BASIC_COLORS)/maxv; 1043b0a32e0cSBarry Smith ierr = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr); 1044b0a32e0cSBarry Smith if (popup) {ierr = PetscDrawScalePopup(popup,0.0,maxv);CHKERRQ(ierr);} 1045f1af5d2fSBarry Smith for(j = 0; j < n; j++) { 1046f1af5d2fSBarry Smith x_l = j; 1047f1af5d2fSBarry Smith x_r = x_l + 1.0; 1048f1af5d2fSBarry Smith for(i = 0; i < m; i++) { 1049f1af5d2fSBarry Smith y_l = m - i - 1.0; 1050f1af5d2fSBarry Smith y_r = y_l + 1.0; 1051b0a32e0cSBarry Smith color = PETSC_DRAW_BASIC_COLORS + (int)(scale*PetscAbsScalar(v[j*m+i])); 1052b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 1053f1af5d2fSBarry Smith } 1054f1af5d2fSBarry Smith } 1055f1af5d2fSBarry Smith } 1056f1af5d2fSBarry Smith PetscFunctionReturn(0); 1057f1af5d2fSBarry Smith } 1058f1af5d2fSBarry Smith 10594a2ae208SSatish Balay #undef __FUNCT__ 10604a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Draw" 1061dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense_Draw(Mat A,PetscViewer viewer) 1062f1af5d2fSBarry Smith { 1063b0a32e0cSBarry Smith PetscDraw draw; 1064f1af5d2fSBarry Smith PetscTruth isnull; 1065329f5518SBarry Smith PetscReal xr,yr,xl,yl,h,w; 1066dfbe8321SBarry Smith PetscErrorCode ierr; 1067f1af5d2fSBarry Smith 1068f1af5d2fSBarry Smith PetscFunctionBegin; 1069b0a32e0cSBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 1070b0a32e0cSBarry Smith ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); 1071abc0a331SBarry Smith if (isnull) PetscFunctionReturn(0); 1072f1af5d2fSBarry Smith 1073f1af5d2fSBarry Smith ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr); 1074d0f46423SBarry Smith xr = A->cmap->n; yr = A->rmap->n; h = yr/10.0; w = xr/10.0; 1075f1af5d2fSBarry Smith xr += w; yr += h; xl = -w; yl = -h; 1076b0a32e0cSBarry Smith ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr); 1077b0a32e0cSBarry Smith ierr = PetscDrawZoom(draw,MatView_SeqDense_Draw_Zoom,A);CHKERRQ(ierr); 1078f1af5d2fSBarry Smith ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",PETSC_NULL);CHKERRQ(ierr); 1079f1af5d2fSBarry Smith PetscFunctionReturn(0); 1080f1af5d2fSBarry Smith } 1081f1af5d2fSBarry Smith 10824a2ae208SSatish Balay #undef __FUNCT__ 10834a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense" 1084dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense(Mat A,PetscViewer viewer) 1085932b0c3eSLois Curfman McInnes { 1086dfbe8321SBarry Smith PetscErrorCode ierr; 10876805f65bSBarry Smith PetscTruth iascii,isbinary,isdraw; 1088932b0c3eSLois Curfman McInnes 10893a40ed3dSBarry Smith PetscFunctionBegin; 109032077d6dSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 1091fb9695e5SSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_BINARY,&isbinary);CHKERRQ(ierr); 1092fb9695e5SSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr); 10930f5bd95cSBarry Smith 1094c45a1595SBarry Smith if (iascii) { 1095c45a1595SBarry Smith ierr = MatView_SeqDense_ASCII(A,viewer);CHKERRQ(ierr); 10960f5bd95cSBarry Smith } else if (isbinary) { 10973a40ed3dSBarry Smith ierr = MatView_SeqDense_Binary(A,viewer);CHKERRQ(ierr); 1098f1af5d2fSBarry Smith } else if (isdraw) { 1099f1af5d2fSBarry Smith ierr = MatView_SeqDense_Draw(A,viewer);CHKERRQ(ierr); 11005cd90555SBarry Smith } else { 1101958c9bccSBarry Smith SETERRQ1(PETSC_ERR_SUP,"Viewer type %s not supported by dense matrix",((PetscObject)viewer)->type_name); 1102932b0c3eSLois Curfman McInnes } 11033a40ed3dSBarry Smith PetscFunctionReturn(0); 1104932b0c3eSLois Curfman McInnes } 1105289bc588SBarry Smith 11064a2ae208SSatish Balay #undef __FUNCT__ 11074a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqDense" 1108dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqDense(Mat mat) 1109289bc588SBarry Smith { 1110ec8511deSBarry Smith Mat_SeqDense *l = (Mat_SeqDense*)mat->data; 1111dfbe8321SBarry Smith PetscErrorCode ierr; 111290f02eecSBarry Smith 11133a40ed3dSBarry Smith PetscFunctionBegin; 1114aa482453SBarry Smith #if defined(PETSC_USE_LOG) 1115d0f46423SBarry Smith PetscLogObjectState((PetscObject)mat,"Rows %D Cols %D",mat->rmap->n,mat->cmap->n); 1116a5a9c739SBarry Smith #endif 111705b42c5fSBarry Smith ierr = PetscFree(l->pivots);CHKERRQ(ierr); 11186857c123SSatish Balay if (!l->user_alloc) {ierr = PetscFree(l->v);CHKERRQ(ierr);} 1119606d414cSSatish Balay ierr = PetscFree(l);CHKERRQ(ierr); 1120dbd8c25aSHong Zhang 1121dbd8c25aSHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)mat,0);CHKERRQ(ierr); 1122901853e0SKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatSeqDenseSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr); 11234ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMult_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr); 11244ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultSymbolic_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr); 11254ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultNumeric_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr); 11263a40ed3dSBarry Smith PetscFunctionReturn(0); 1127289bc588SBarry Smith } 1128289bc588SBarry Smith 11294a2ae208SSatish Balay #undef __FUNCT__ 11304a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqDense" 1131fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqDense(Mat A,MatReuse reuse,Mat *matout) 1132289bc588SBarry Smith { 1133c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 11346849ba73SBarry Smith PetscErrorCode ierr; 113513f74950SBarry Smith PetscInt k,j,m,n,M; 113687828ca2SBarry Smith PetscScalar *v,tmp; 113748b35521SBarry Smith 11383a40ed3dSBarry Smith PetscFunctionBegin; 1139d0f46423SBarry Smith v = mat->v; m = A->rmap->n; M = mat->lda; n = A->cmap->n; 1140e9695a30SBarry Smith if (reuse == MAT_REUSE_MATRIX && *matout == A) { /* in place transpose */ 1141a5ce6ee0Svictorle if (m != n) { 1142634064b4SBarry Smith SETERRQ(PETSC_ERR_SUP,"Can not transpose non-square matrix in place"); 1143d64ed03dSBarry Smith } else { 1144d3e5ee88SLois Curfman McInnes for (j=0; j<m; j++) { 1145289bc588SBarry Smith for (k=0; k<j; k++) { 11461b807ce4Svictorle tmp = v[j + k*M]; 11471b807ce4Svictorle v[j + k*M] = v[k + j*M]; 11481b807ce4Svictorle v[k + j*M] = tmp; 1149289bc588SBarry Smith } 1150289bc588SBarry Smith } 1151d64ed03dSBarry Smith } 11523a40ed3dSBarry Smith } else { /* out-of-place transpose */ 1153d3e5ee88SLois Curfman McInnes Mat tmat; 1154ec8511deSBarry Smith Mat_SeqDense *tmatd; 115587828ca2SBarry Smith PetscScalar *v2; 1156ea709b57SSatish Balay 1157fc4dec0aSBarry Smith if (reuse == MAT_INITIAL_MATRIX) { 11587adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&tmat);CHKERRQ(ierr); 1159d0f46423SBarry Smith ierr = MatSetSizes(tmat,A->cmap->n,A->rmap->n,A->cmap->n,A->rmap->n);CHKERRQ(ierr); 11607adad957SLisandro Dalcin ierr = MatSetType(tmat,((PetscObject)A)->type_name);CHKERRQ(ierr); 11615c5985e7SKris Buschelman ierr = MatSeqDenseSetPreallocation(tmat,PETSC_NULL);CHKERRQ(ierr); 1162fc4dec0aSBarry Smith } else { 1163fc4dec0aSBarry Smith tmat = *matout; 1164fc4dec0aSBarry Smith } 1165ec8511deSBarry Smith tmatd = (Mat_SeqDense*)tmat->data; 11660de55854SLois Curfman McInnes v = mat->v; v2 = tmatd->v; 1167d3e5ee88SLois Curfman McInnes for (j=0; j<n; j++) { 11681b807ce4Svictorle for (k=0; k<m; k++) v2[j + k*n] = v[k + j*M]; 1169d3e5ee88SLois Curfman McInnes } 11706d4a8577SBarry Smith ierr = MatAssemblyBegin(tmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 11716d4a8577SBarry Smith ierr = MatAssemblyEnd(tmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1172d3e5ee88SLois Curfman McInnes *matout = tmat; 117348b35521SBarry Smith } 11743a40ed3dSBarry Smith PetscFunctionReturn(0); 1175289bc588SBarry Smith } 1176289bc588SBarry Smith 11774a2ae208SSatish Balay #undef __FUNCT__ 11784a2ae208SSatish Balay #define __FUNCT__ "MatEqual_SeqDense" 1179dfbe8321SBarry Smith PetscErrorCode MatEqual_SeqDense(Mat A1,Mat A2,PetscTruth *flg) 1180289bc588SBarry Smith { 1181c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat1 = (Mat_SeqDense*)A1->data; 1182c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat2 = (Mat_SeqDense*)A2->data; 118313f74950SBarry Smith PetscInt i,j; 118487828ca2SBarry Smith PetscScalar *v1 = mat1->v,*v2 = mat2->v; 11859ea5d5aeSSatish Balay 11863a40ed3dSBarry Smith PetscFunctionBegin; 1187d0f46423SBarry Smith if (A1->rmap->n != A2->rmap->n) {*flg = PETSC_FALSE; PetscFunctionReturn(0);} 1188d0f46423SBarry Smith if (A1->cmap->n != A2->cmap->n) {*flg = PETSC_FALSE; PetscFunctionReturn(0);} 1189d0f46423SBarry Smith for (i=0; i<A1->rmap->n; i++) { 11901b807ce4Svictorle v1 = mat1->v+i; v2 = mat2->v+i; 1191d0f46423SBarry Smith for (j=0; j<A1->cmap->n; j++) { 11923a40ed3dSBarry Smith if (*v1 != *v2) {*flg = PETSC_FALSE; PetscFunctionReturn(0);} 11931b807ce4Svictorle v1 += mat1->lda; v2 += mat2->lda; 11941b807ce4Svictorle } 1195289bc588SBarry Smith } 119677c4ece6SBarry Smith *flg = PETSC_TRUE; 11973a40ed3dSBarry Smith PetscFunctionReturn(0); 1198289bc588SBarry Smith } 1199289bc588SBarry Smith 12004a2ae208SSatish Balay #undef __FUNCT__ 12014a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_SeqDense" 1202dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_SeqDense(Mat A,Vec v) 1203289bc588SBarry Smith { 1204c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 1205dfbe8321SBarry Smith PetscErrorCode ierr; 120613f74950SBarry Smith PetscInt i,n,len; 120787828ca2SBarry Smith PetscScalar *x,zero = 0.0; 120844cd7ae7SLois Curfman McInnes 12093a40ed3dSBarry Smith PetscFunctionBegin; 12102dcb1b2aSMatthew Knepley ierr = VecSet(v,zero);CHKERRQ(ierr); 12117a97a34bSBarry Smith ierr = VecGetSize(v,&n);CHKERRQ(ierr); 12121ebc52fbSHong Zhang ierr = VecGetArray(v,&x);CHKERRQ(ierr); 1213d0f46423SBarry Smith len = PetscMin(A->rmap->n,A->cmap->n); 1214d0f46423SBarry Smith if (n != A->rmap->n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming mat and vec"); 121544cd7ae7SLois Curfman McInnes for (i=0; i<len; i++) { 12161b807ce4Svictorle x[i] = mat->v[i*mat->lda + i]; 1217289bc588SBarry Smith } 12181ebc52fbSHong Zhang ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 12193a40ed3dSBarry Smith PetscFunctionReturn(0); 1220289bc588SBarry Smith } 1221289bc588SBarry Smith 12224a2ae208SSatish Balay #undef __FUNCT__ 12234a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_SeqDense" 1224dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_SeqDense(Mat A,Vec ll,Vec rr) 1225289bc588SBarry Smith { 1226c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 122787828ca2SBarry Smith PetscScalar *l,*r,x,*v; 1228dfbe8321SBarry Smith PetscErrorCode ierr; 1229d0f46423SBarry Smith PetscInt i,j,m = A->rmap->n,n = A->cmap->n; 123055659b69SBarry Smith 12313a40ed3dSBarry Smith PetscFunctionBegin; 123228988994SBarry Smith if (ll) { 12337a97a34bSBarry Smith ierr = VecGetSize(ll,&m);CHKERRQ(ierr); 12341ebc52fbSHong Zhang ierr = VecGetArray(ll,&l);CHKERRQ(ierr); 1235d0f46423SBarry Smith if (m != A->rmap->n) SETERRQ(PETSC_ERR_ARG_SIZ,"Left scaling vec wrong size"); 1236da3a660dSBarry Smith for (i=0; i<m; i++) { 1237da3a660dSBarry Smith x = l[i]; 1238da3a660dSBarry Smith v = mat->v + i; 1239da3a660dSBarry Smith for (j=0; j<n; j++) { (*v) *= x; v+= m;} 1240da3a660dSBarry Smith } 12411ebc52fbSHong Zhang ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr); 1242efee365bSSatish Balay ierr = PetscLogFlops(n*m);CHKERRQ(ierr); 1243da3a660dSBarry Smith } 124428988994SBarry Smith if (rr) { 12457a97a34bSBarry Smith ierr = VecGetSize(rr,&n);CHKERRQ(ierr); 12461ebc52fbSHong Zhang ierr = VecGetArray(rr,&r);CHKERRQ(ierr); 1247d0f46423SBarry Smith if (n != A->cmap->n) SETERRQ(PETSC_ERR_ARG_SIZ,"Right scaling vec wrong size"); 1248da3a660dSBarry Smith for (i=0; i<n; i++) { 1249da3a660dSBarry Smith x = r[i]; 1250da3a660dSBarry Smith v = mat->v + i*m; 1251da3a660dSBarry Smith for (j=0; j<m; j++) { (*v++) *= x;} 1252da3a660dSBarry Smith } 12531ebc52fbSHong Zhang ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr); 1254efee365bSSatish Balay ierr = PetscLogFlops(n*m);CHKERRQ(ierr); 1255da3a660dSBarry Smith } 12563a40ed3dSBarry Smith PetscFunctionReturn(0); 1257289bc588SBarry Smith } 1258289bc588SBarry Smith 12594a2ae208SSatish Balay #undef __FUNCT__ 12604a2ae208SSatish Balay #define __FUNCT__ "MatNorm_SeqDense" 1261dfbe8321SBarry Smith PetscErrorCode MatNorm_SeqDense(Mat A,NormType type,PetscReal *nrm) 1262289bc588SBarry Smith { 1263c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 126487828ca2SBarry Smith PetscScalar *v = mat->v; 1265329f5518SBarry Smith PetscReal sum = 0.0; 1266d0f46423SBarry Smith PetscInt lda=mat->lda,m=A->rmap->n,i,j; 1267efee365bSSatish Balay PetscErrorCode ierr; 126855659b69SBarry Smith 12693a40ed3dSBarry Smith PetscFunctionBegin; 1270289bc588SBarry Smith if (type == NORM_FROBENIUS) { 1271a5ce6ee0Svictorle if (lda>m) { 1272d0f46423SBarry Smith for (j=0; j<A->cmap->n; j++) { 1273a5ce6ee0Svictorle v = mat->v+j*lda; 1274a5ce6ee0Svictorle for (i=0; i<m; i++) { 1275a5ce6ee0Svictorle #if defined(PETSC_USE_COMPLEX) 1276a5ce6ee0Svictorle sum += PetscRealPart(PetscConj(*v)*(*v)); v++; 1277a5ce6ee0Svictorle #else 1278a5ce6ee0Svictorle sum += (*v)*(*v); v++; 1279a5ce6ee0Svictorle #endif 1280a5ce6ee0Svictorle } 1281a5ce6ee0Svictorle } 1282a5ce6ee0Svictorle } else { 1283d0f46423SBarry Smith for (i=0; i<A->cmap->n*A->rmap->n; i++) { 1284aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 1285329f5518SBarry Smith sum += PetscRealPart(PetscConj(*v)*(*v)); v++; 1286289bc588SBarry Smith #else 1287289bc588SBarry Smith sum += (*v)*(*v); v++; 1288289bc588SBarry Smith #endif 1289289bc588SBarry Smith } 1290a5ce6ee0Svictorle } 1291064f8208SBarry Smith *nrm = sqrt(sum); 1292dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*A->cmap->n*A->rmap->n);CHKERRQ(ierr); 12933a40ed3dSBarry Smith } else if (type == NORM_1) { 1294064f8208SBarry Smith *nrm = 0.0; 1295d0f46423SBarry Smith for (j=0; j<A->cmap->n; j++) { 12961b807ce4Svictorle v = mat->v + j*mat->lda; 1297289bc588SBarry Smith sum = 0.0; 1298d0f46423SBarry Smith for (i=0; i<A->rmap->n; i++) { 129933a8263dSBarry Smith sum += PetscAbsScalar(*v); v++; 1300289bc588SBarry Smith } 1301064f8208SBarry Smith if (sum > *nrm) *nrm = sum; 1302289bc588SBarry Smith } 1303d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n*A->rmap->n);CHKERRQ(ierr); 13043a40ed3dSBarry Smith } else if (type == NORM_INFINITY) { 1305064f8208SBarry Smith *nrm = 0.0; 1306d0f46423SBarry Smith for (j=0; j<A->rmap->n; j++) { 1307289bc588SBarry Smith v = mat->v + j; 1308289bc588SBarry Smith sum = 0.0; 1309d0f46423SBarry Smith for (i=0; i<A->cmap->n; i++) { 13101b807ce4Svictorle sum += PetscAbsScalar(*v); v += mat->lda; 1311289bc588SBarry Smith } 1312064f8208SBarry Smith if (sum > *nrm) *nrm = sum; 1313289bc588SBarry Smith } 1314d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n*A->rmap->n);CHKERRQ(ierr); 13153a40ed3dSBarry Smith } else { 131629bbc08cSBarry Smith SETERRQ(PETSC_ERR_SUP,"No two norm"); 1317289bc588SBarry Smith } 13183a40ed3dSBarry Smith PetscFunctionReturn(0); 1319289bc588SBarry Smith } 1320289bc588SBarry Smith 13214a2ae208SSatish Balay #undef __FUNCT__ 13224a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqDense" 13234e0d8c25SBarry Smith PetscErrorCode MatSetOption_SeqDense(Mat A,MatOption op,PetscTruth flg) 1324289bc588SBarry Smith { 1325c0bbcb79SLois Curfman McInnes Mat_SeqDense *aij = (Mat_SeqDense*)A->data; 132663ba0a88SBarry Smith PetscErrorCode ierr; 132767e560aaSBarry Smith 13283a40ed3dSBarry Smith PetscFunctionBegin; 1329b5a2b587SKris Buschelman switch (op) { 1330b5a2b587SKris Buschelman case MAT_ROW_ORIENTED: 13314e0d8c25SBarry Smith aij->roworiented = flg; 1332b5a2b587SKris Buschelman break; 1333512a5fc5SBarry Smith case MAT_NEW_NONZERO_LOCATIONS: 1334b5a2b587SKris Buschelman case MAT_NEW_NONZERO_LOCATION_ERR: 13353971808eSMatthew Knepley case MAT_NEW_NONZERO_ALLOCATION_ERR: 13364e0d8c25SBarry Smith case MAT_NEW_DIAGONALS: 1337b5a2b587SKris Buschelman case MAT_IGNORE_OFF_PROC_ENTRIES: 1338b5a2b587SKris Buschelman case MAT_USE_HASH_TABLE: 133977e54ba9SKris Buschelman case MAT_SYMMETRIC: 134077e54ba9SKris Buschelman case MAT_STRUCTURALLY_SYMMETRIC: 13419a4540c5SBarry Smith case MAT_HERMITIAN: 13429a4540c5SBarry Smith case MAT_SYMMETRY_ETERNAL: 1343600fe468SBarry Smith case MAT_IGNORE_LOWER_TRIANGULAR: 1344290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 134577e54ba9SKris Buschelman break; 1346b5a2b587SKris Buschelman default: 1347600fe468SBarry Smith SETERRQ1(PETSC_ERR_SUP,"unknown option %s",MatOptions[op]); 13483a40ed3dSBarry Smith } 13493a40ed3dSBarry Smith PetscFunctionReturn(0); 1350289bc588SBarry Smith } 1351289bc588SBarry Smith 13524a2ae208SSatish Balay #undef __FUNCT__ 13534a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_SeqDense" 1354dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_SeqDense(Mat A) 13556f0a148fSBarry Smith { 1356ec8511deSBarry Smith Mat_SeqDense *l = (Mat_SeqDense*)A->data; 13576849ba73SBarry Smith PetscErrorCode ierr; 1358d0f46423SBarry Smith PetscInt lda=l->lda,m=A->rmap->n,j; 13593a40ed3dSBarry Smith 13603a40ed3dSBarry Smith PetscFunctionBegin; 1361a5ce6ee0Svictorle if (lda>m) { 1362d0f46423SBarry Smith for (j=0; j<A->cmap->n; j++) { 1363a5ce6ee0Svictorle ierr = PetscMemzero(l->v+j*lda,m*sizeof(PetscScalar));CHKERRQ(ierr); 1364a5ce6ee0Svictorle } 1365a5ce6ee0Svictorle } else { 1366d0f46423SBarry Smith ierr = PetscMemzero(l->v,A->rmap->n*A->cmap->n*sizeof(PetscScalar));CHKERRQ(ierr); 1367a5ce6ee0Svictorle } 13683a40ed3dSBarry Smith PetscFunctionReturn(0); 13696f0a148fSBarry Smith } 13706f0a148fSBarry Smith 13714a2ae208SSatish Balay #undef __FUNCT__ 13724a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqDense" 1373f4df32b1SMatthew Knepley PetscErrorCode MatZeroRows_SeqDense(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag) 13746f0a148fSBarry Smith { 1375ec8511deSBarry Smith Mat_SeqDense *l = (Mat_SeqDense*)A->data; 1376d0f46423SBarry Smith PetscInt n = A->cmap->n,i,j; 137787828ca2SBarry Smith PetscScalar *slot; 137855659b69SBarry Smith 13793a40ed3dSBarry Smith PetscFunctionBegin; 13806f0a148fSBarry Smith for (i=0; i<N; i++) { 13816f0a148fSBarry Smith slot = l->v + rows[i]; 13826f0a148fSBarry Smith for (j=0; j<n; j++) { *slot = 0.0; slot += n;} 13836f0a148fSBarry Smith } 1384f4df32b1SMatthew Knepley if (diag != 0.0) { 13856f0a148fSBarry Smith for (i=0; i<N; i++) { 13866f0a148fSBarry Smith slot = l->v + (n+1)*rows[i]; 1387f4df32b1SMatthew Knepley *slot = diag; 13886f0a148fSBarry Smith } 13896f0a148fSBarry Smith } 13903a40ed3dSBarry Smith PetscFunctionReturn(0); 13916f0a148fSBarry Smith } 1392557bce09SLois Curfman McInnes 13934a2ae208SSatish Balay #undef __FUNCT__ 13944a2ae208SSatish Balay #define __FUNCT__ "MatGetArray_SeqDense" 1395dfbe8321SBarry Smith PetscErrorCode MatGetArray_SeqDense(Mat A,PetscScalar *array[]) 139664e87e97SBarry Smith { 1397c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 13983a40ed3dSBarry Smith 13993a40ed3dSBarry Smith PetscFunctionBegin; 1400d0f46423SBarry Smith if (mat->lda != A->rmap->n) SETERRQ(PETSC_ERR_SUP,"Cannot get array for Dense matrices with LDA different from number of rows"); 140164e87e97SBarry Smith *array = mat->v; 14023a40ed3dSBarry Smith PetscFunctionReturn(0); 140364e87e97SBarry Smith } 14040754003eSLois Curfman McInnes 14054a2ae208SSatish Balay #undef __FUNCT__ 14064a2ae208SSatish Balay #define __FUNCT__ "MatRestoreArray_SeqDense" 1407dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_SeqDense(Mat A,PetscScalar *array[]) 1408ff14e315SSatish Balay { 14093a40ed3dSBarry Smith PetscFunctionBegin; 141009b544d4SBarry Smith *array = 0; /* user cannot accidently use the array later */ 14113a40ed3dSBarry Smith PetscFunctionReturn(0); 1412ff14e315SSatish Balay } 14130754003eSLois Curfman McInnes 14144a2ae208SSatish Balay #undef __FUNCT__ 14154a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_SeqDense" 141613f74950SBarry Smith static PetscErrorCode MatGetSubMatrix_SeqDense(Mat A,IS isrow,IS iscol,PetscInt cs,MatReuse scall,Mat *B) 14170754003eSLois Curfman McInnes { 1418c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 14196849ba73SBarry Smith PetscErrorCode ierr; 14205d0c19d7SBarry Smith PetscInt i,j,nrows,ncols; 14215d0c19d7SBarry Smith const PetscInt *irow,*icol; 142287828ca2SBarry Smith PetscScalar *av,*bv,*v = mat->v; 14230754003eSLois Curfman McInnes Mat newmat; 14240754003eSLois Curfman McInnes 14253a40ed3dSBarry Smith PetscFunctionBegin; 142678b31e54SBarry Smith ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr); 142778b31e54SBarry Smith ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr); 1428e03a110bSBarry Smith ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr); 1429e03a110bSBarry Smith ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr); 14300754003eSLois Curfman McInnes 1431182d2002SSatish Balay /* Check submatrixcall */ 1432182d2002SSatish Balay if (scall == MAT_REUSE_MATRIX) { 143313f74950SBarry Smith PetscInt n_cols,n_rows; 1434182d2002SSatish Balay ierr = MatGetSize(*B,&n_rows,&n_cols);CHKERRQ(ierr); 143521a2c019SBarry Smith if (n_rows != nrows || n_cols != ncols) { 143621a2c019SBarry Smith /* resize the result result matrix to match number of requested rows/columns */ 1437c61587bbSBarry Smith ierr = MatSetSizes(*B,nrows,ncols,nrows,ncols);CHKERRQ(ierr); 143821a2c019SBarry Smith } 1439182d2002SSatish Balay newmat = *B; 1440182d2002SSatish Balay } else { 14410754003eSLois Curfman McInnes /* Create and fill new matrix */ 14427adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&newmat);CHKERRQ(ierr); 1443f69a0ea3SMatthew Knepley ierr = MatSetSizes(newmat,nrows,ncols,nrows,ncols);CHKERRQ(ierr); 14447adad957SLisandro Dalcin ierr = MatSetType(newmat,((PetscObject)A)->type_name);CHKERRQ(ierr); 14455c5985e7SKris Buschelman ierr = MatSeqDenseSetPreallocation(newmat,PETSC_NULL);CHKERRQ(ierr); 1446182d2002SSatish Balay } 1447182d2002SSatish Balay 1448182d2002SSatish Balay /* Now extract the data pointers and do the copy,column at a time */ 1449182d2002SSatish Balay bv = ((Mat_SeqDense*)newmat->data)->v; 1450182d2002SSatish Balay 1451182d2002SSatish Balay for (i=0; i<ncols; i++) { 14526de62eeeSBarry Smith av = v + mat->lda*icol[i]; 1453182d2002SSatish Balay for (j=0; j<nrows; j++) { 1454182d2002SSatish Balay *bv++ = av[irow[j]]; 14550754003eSLois Curfman McInnes } 14560754003eSLois Curfman McInnes } 1457182d2002SSatish Balay 1458182d2002SSatish Balay /* Assemble the matrices so that the correct flags are set */ 14596d4a8577SBarry Smith ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 14606d4a8577SBarry Smith ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 14610754003eSLois Curfman McInnes 14620754003eSLois Curfman McInnes /* Free work space */ 146378b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr); 146478b31e54SBarry Smith ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr); 1465182d2002SSatish Balay *B = newmat; 14663a40ed3dSBarry Smith PetscFunctionReturn(0); 14670754003eSLois Curfman McInnes } 14680754003eSLois Curfman McInnes 14694a2ae208SSatish Balay #undef __FUNCT__ 14704a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrices_SeqDense" 147113f74950SBarry Smith PetscErrorCode MatGetSubMatrices_SeqDense(Mat A,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *B[]) 1472905e6a2fSBarry Smith { 14736849ba73SBarry Smith PetscErrorCode ierr; 147413f74950SBarry Smith PetscInt i; 1475905e6a2fSBarry Smith 14763a40ed3dSBarry Smith PetscFunctionBegin; 1477905e6a2fSBarry Smith if (scall == MAT_INITIAL_MATRIX) { 1478b0a32e0cSBarry Smith ierr = PetscMalloc((n+1)*sizeof(Mat),B);CHKERRQ(ierr); 1479905e6a2fSBarry Smith } 1480905e6a2fSBarry Smith 1481905e6a2fSBarry Smith for (i=0; i<n; i++) { 14826a6a5d1dSBarry Smith ierr = MatGetSubMatrix_SeqDense(A,irow[i],icol[i],PETSC_DECIDE,scall,&(*B)[i]);CHKERRQ(ierr); 1483905e6a2fSBarry Smith } 14843a40ed3dSBarry Smith PetscFunctionReturn(0); 1485905e6a2fSBarry Smith } 1486905e6a2fSBarry Smith 14874a2ae208SSatish Balay #undef __FUNCT__ 1488c0aa2d19SHong Zhang #define __FUNCT__ "MatAssemblyBegin_SeqDense" 1489c0aa2d19SHong Zhang PetscErrorCode MatAssemblyBegin_SeqDense(Mat mat,MatAssemblyType mode) 1490c0aa2d19SHong Zhang { 1491c0aa2d19SHong Zhang PetscFunctionBegin; 1492c0aa2d19SHong Zhang PetscFunctionReturn(0); 1493c0aa2d19SHong Zhang } 1494c0aa2d19SHong Zhang 1495c0aa2d19SHong Zhang #undef __FUNCT__ 1496c0aa2d19SHong Zhang #define __FUNCT__ "MatAssemblyEnd_SeqDense" 1497c0aa2d19SHong Zhang PetscErrorCode MatAssemblyEnd_SeqDense(Mat mat,MatAssemblyType mode) 1498c0aa2d19SHong Zhang { 1499c0aa2d19SHong Zhang PetscFunctionBegin; 1500c0aa2d19SHong Zhang PetscFunctionReturn(0); 1501c0aa2d19SHong Zhang } 1502c0aa2d19SHong Zhang 1503c0aa2d19SHong Zhang #undef __FUNCT__ 15044a2ae208SSatish Balay #define __FUNCT__ "MatCopy_SeqDense" 1505dfbe8321SBarry Smith PetscErrorCode MatCopy_SeqDense(Mat A,Mat B,MatStructure str) 15064b0e389bSBarry Smith { 15074b0e389bSBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data,*b = (Mat_SeqDense *)B->data; 15086849ba73SBarry Smith PetscErrorCode ierr; 1509d0f46423SBarry Smith PetscInt lda1=a->lda,lda2=b->lda, m=A->rmap->n,n=A->cmap->n, j; 15103a40ed3dSBarry Smith 15113a40ed3dSBarry Smith PetscFunctionBegin; 151233f4a19fSKris Buschelman /* If the two matrices don't have the same copy implementation, they aren't compatible for fast copy. */ 151333f4a19fSKris Buschelman if (A->ops->copy != B->ops->copy) { 1514cb5b572fSBarry Smith ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 15153a40ed3dSBarry Smith PetscFunctionReturn(0); 15163a40ed3dSBarry Smith } 1517d0f46423SBarry Smith if (m != B->rmap->n || n != B->cmap->n) SETERRQ(PETSC_ERR_ARG_SIZ,"size(B) != size(A)"); 1518a5ce6ee0Svictorle if (lda1>m || lda2>m) { 15190dbb7854Svictorle for (j=0; j<n; j++) { 1520a5ce6ee0Svictorle ierr = PetscMemcpy(b->v+j*lda2,a->v+j*lda1,m*sizeof(PetscScalar));CHKERRQ(ierr); 1521a5ce6ee0Svictorle } 1522a5ce6ee0Svictorle } else { 1523d0f46423SBarry Smith ierr = PetscMemcpy(b->v,a->v,A->rmap->n*A->cmap->n*sizeof(PetscScalar));CHKERRQ(ierr); 1524a5ce6ee0Svictorle } 1525273d9f13SBarry Smith PetscFunctionReturn(0); 1526273d9f13SBarry Smith } 1527273d9f13SBarry Smith 15284a2ae208SSatish Balay #undef __FUNCT__ 15294a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_SeqDense" 1530dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_SeqDense(Mat A) 1531273d9f13SBarry Smith { 1532dfbe8321SBarry Smith PetscErrorCode ierr; 1533273d9f13SBarry Smith 1534273d9f13SBarry Smith PetscFunctionBegin; 1535273d9f13SBarry Smith ierr = MatSeqDenseSetPreallocation(A,0);CHKERRQ(ierr); 15363a40ed3dSBarry Smith PetscFunctionReturn(0); 15374b0e389bSBarry Smith } 15384b0e389bSBarry Smith 1539284134d9SBarry Smith #undef __FUNCT__ 1540284134d9SBarry Smith #define __FUNCT__ "MatSetSizes_SeqDense" 1541284134d9SBarry Smith PetscErrorCode MatSetSizes_SeqDense(Mat A,PetscInt m,PetscInt n,PetscInt M,PetscInt N) 1542284134d9SBarry Smith { 1543284134d9SBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 1544284134d9SBarry Smith PetscFunctionBegin; 154521a2c019SBarry Smith /* this will not be called before lda, Mmax, and Nmax have been set */ 1546284134d9SBarry Smith m = PetscMax(m,M); 1547284134d9SBarry Smith n = PetscMax(n,N); 154821a2c019SBarry Smith if (m > a->Mmax) SETERRQ2(PETSC_ERR_SUP,"Cannot yet resize number rows of dense matrix larger then its initial size %d, requested %d",a->lda,(int)m); 1549284134d9SBarry Smith if (n > a->Nmax) SETERRQ2(PETSC_ERR_SUP,"Cannot yet resize number columns of dense matrix larger then its initial size %d, requested %d",a->Nmax,(int)n); 1550d0f46423SBarry Smith A->rmap->n = A->rmap->n = m; 1551d0f46423SBarry Smith A->cmap->n = A->cmap->N = n; 155221a2c019SBarry Smith if (a->changelda) a->lda = m; 1553284134d9SBarry Smith PetscFunctionReturn(0); 1554284134d9SBarry Smith } 1555170fe5c8SBarry Smith 1556284134d9SBarry Smith 1557a9fe9ddaSSatish Balay /* ----------------------------------------------------------------*/ 1558a9fe9ddaSSatish Balay #undef __FUNCT__ 1559a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMult_SeqDense_SeqDense" 1560a9fe9ddaSSatish Balay PetscErrorCode MatMatMult_SeqDense_SeqDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 1561a9fe9ddaSSatish Balay { 1562a9fe9ddaSSatish Balay PetscErrorCode ierr; 1563a9fe9ddaSSatish Balay 1564a9fe9ddaSSatish Balay PetscFunctionBegin; 1565a9fe9ddaSSatish Balay if (scall == MAT_INITIAL_MATRIX){ 1566a9fe9ddaSSatish Balay ierr = MatMatMultSymbolic_SeqDense_SeqDense(A,B,fill,C);CHKERRQ(ierr); 1567a9fe9ddaSSatish Balay } 1568a9fe9ddaSSatish Balay ierr = MatMatMultNumeric_SeqDense_SeqDense(A,B,*C);CHKERRQ(ierr); 1569a9fe9ddaSSatish Balay PetscFunctionReturn(0); 1570a9fe9ddaSSatish Balay } 1571a9fe9ddaSSatish Balay 1572a9fe9ddaSSatish Balay #undef __FUNCT__ 1573a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultSymbolic_SeqDense_SeqDense" 1574a9fe9ddaSSatish Balay PetscErrorCode MatMatMultSymbolic_SeqDense_SeqDense(Mat A,Mat B,PetscReal fill,Mat *C) 1575a9fe9ddaSSatish Balay { 1576ee16a9a1SHong Zhang PetscErrorCode ierr; 1577d0f46423SBarry Smith PetscInt m=A->rmap->n,n=B->cmap->n; 1578ee16a9a1SHong Zhang Mat Cmat; 1579a9fe9ddaSSatish Balay 1580ee16a9a1SHong Zhang PetscFunctionBegin; 1581d0f46423SBarry Smith if (A->cmap->n != B->rmap->n) SETERRQ2(PETSC_ERR_ARG_SIZ,"A->cmap->n %d != B->rmap->n %d\n",A->cmap->n,B->rmap->n); 1582ee16a9a1SHong Zhang ierr = MatCreate(PETSC_COMM_SELF,&Cmat);CHKERRQ(ierr); 1583ee16a9a1SHong Zhang ierr = MatSetSizes(Cmat,m,n,m,n);CHKERRQ(ierr); 1584ee16a9a1SHong Zhang ierr = MatSetType(Cmat,MATSEQDENSE);CHKERRQ(ierr); 1585ee16a9a1SHong Zhang ierr = MatSeqDenseSetPreallocation(Cmat,PETSC_NULL);CHKERRQ(ierr); 1586ee16a9a1SHong Zhang Cmat->assembled = PETSC_TRUE; 1587ee16a9a1SHong Zhang *C = Cmat; 1588ee16a9a1SHong Zhang PetscFunctionReturn(0); 1589ee16a9a1SHong Zhang } 1590a9fe9ddaSSatish Balay 159198a3b096SSatish Balay #undef __FUNCT__ 1592a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultNumeric_SeqDense_SeqDense" 1593a9fe9ddaSSatish Balay PetscErrorCode MatMatMultNumeric_SeqDense_SeqDense(Mat A,Mat B,Mat C) 1594a9fe9ddaSSatish Balay { 1595a9fe9ddaSSatish Balay Mat_SeqDense *a = (Mat_SeqDense*)A->data; 1596a9fe9ddaSSatish Balay Mat_SeqDense *b = (Mat_SeqDense*)B->data; 1597a9fe9ddaSSatish Balay Mat_SeqDense *c = (Mat_SeqDense*)C->data; 15980805154bSBarry Smith PetscBLASInt m,n,k; 1599a9fe9ddaSSatish Balay PetscScalar _DOne=1.0,_DZero=0.0; 1600a9fe9ddaSSatish Balay 1601a9fe9ddaSSatish Balay PetscFunctionBegin; 1602d0f46423SBarry Smith m = PetscBLASIntCast(A->rmap->n); 1603d0f46423SBarry Smith n = PetscBLASIntCast(B->cmap->n); 1604d0f46423SBarry Smith k = PetscBLASIntCast(A->cmap->n); 1605a9fe9ddaSSatish Balay BLASgemm_("N","N",&m,&n,&k,&_DOne,a->v,&a->lda,b->v,&b->lda,&_DZero,c->v,&c->lda); 1606a9fe9ddaSSatish Balay PetscFunctionReturn(0); 1607a9fe9ddaSSatish Balay } 1608a9fe9ddaSSatish Balay 1609a9fe9ddaSSatish Balay #undef __FUNCT__ 1610a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTranspose_SeqDense_SeqDense" 1611a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTranspose_SeqDense_SeqDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 1612a9fe9ddaSSatish Balay { 1613a9fe9ddaSSatish Balay PetscErrorCode ierr; 1614a9fe9ddaSSatish Balay 1615a9fe9ddaSSatish Balay PetscFunctionBegin; 1616a9fe9ddaSSatish Balay if (scall == MAT_INITIAL_MATRIX){ 1617a9fe9ddaSSatish Balay ierr = MatMatMultTransposeSymbolic_SeqDense_SeqDense(A,B,fill,C);CHKERRQ(ierr); 1618a9fe9ddaSSatish Balay } 1619a9fe9ddaSSatish Balay ierr = MatMatMultTransposeNumeric_SeqDense_SeqDense(A,B,*C);CHKERRQ(ierr); 1620a9fe9ddaSSatish Balay PetscFunctionReturn(0); 1621a9fe9ddaSSatish Balay } 1622a9fe9ddaSSatish Balay 1623a9fe9ddaSSatish Balay #undef __FUNCT__ 1624a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTransposeSymbolic_SeqDense_SeqDense" 1625a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTransposeSymbolic_SeqDense_SeqDense(Mat A,Mat B,PetscReal fill,Mat *C) 1626a9fe9ddaSSatish Balay { 1627ee16a9a1SHong Zhang PetscErrorCode ierr; 1628d0f46423SBarry Smith PetscInt m=A->cmap->n,n=B->cmap->n; 1629ee16a9a1SHong Zhang Mat Cmat; 1630a9fe9ddaSSatish Balay 1631ee16a9a1SHong Zhang PetscFunctionBegin; 1632d0f46423SBarry Smith if (A->rmap->n != B->rmap->n) SETERRQ2(PETSC_ERR_ARG_SIZ,"A->rmap->n %d != B->rmap->n %d\n",A->rmap->n,B->rmap->n); 1633ee16a9a1SHong Zhang ierr = MatCreate(PETSC_COMM_SELF,&Cmat);CHKERRQ(ierr); 1634ee16a9a1SHong Zhang ierr = MatSetSizes(Cmat,m,n,m,n);CHKERRQ(ierr); 1635ee16a9a1SHong Zhang ierr = MatSetType(Cmat,MATSEQDENSE);CHKERRQ(ierr); 1636ee16a9a1SHong Zhang ierr = MatSeqDenseSetPreallocation(Cmat,PETSC_NULL);CHKERRQ(ierr); 1637ee16a9a1SHong Zhang Cmat->assembled = PETSC_TRUE; 1638ee16a9a1SHong Zhang *C = Cmat; 1639ee16a9a1SHong Zhang PetscFunctionReturn(0); 1640ee16a9a1SHong Zhang } 1641a9fe9ddaSSatish Balay 1642a9fe9ddaSSatish Balay #undef __FUNCT__ 1643a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTransposeNumeric_SeqDense_SeqDense" 1644a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTransposeNumeric_SeqDense_SeqDense(Mat A,Mat B,Mat C) 1645a9fe9ddaSSatish Balay { 1646a9fe9ddaSSatish Balay Mat_SeqDense *a = (Mat_SeqDense*)A->data; 1647a9fe9ddaSSatish Balay Mat_SeqDense *b = (Mat_SeqDense*)B->data; 1648a9fe9ddaSSatish Balay Mat_SeqDense *c = (Mat_SeqDense*)C->data; 16490805154bSBarry Smith PetscBLASInt m,n,k; 1650a9fe9ddaSSatish Balay PetscScalar _DOne=1.0,_DZero=0.0; 1651a9fe9ddaSSatish Balay 1652a9fe9ddaSSatish Balay PetscFunctionBegin; 1653d0f46423SBarry Smith m = PetscBLASIntCast(A->cmap->n); 1654d0f46423SBarry Smith n = PetscBLASIntCast(B->cmap->n); 1655d0f46423SBarry Smith k = PetscBLASIntCast(A->rmap->n); 16562fbe02b9SBarry Smith /* 16572fbe02b9SBarry Smith Note the m and n arguments below are the number rows and columns of A', not A! 16582fbe02b9SBarry Smith */ 1659a9fe9ddaSSatish Balay BLASgemm_("T","N",&m,&n,&k,&_DOne,a->v,&a->lda,b->v,&b->lda,&_DZero,c->v,&c->lda); 1660a9fe9ddaSSatish Balay PetscFunctionReturn(0); 1661a9fe9ddaSSatish Balay } 1662985db425SBarry Smith 1663985db425SBarry Smith #undef __FUNCT__ 1664985db425SBarry Smith #define __FUNCT__ "MatGetRowMax_SeqDense" 1665985db425SBarry Smith PetscErrorCode MatGetRowMax_SeqDense(Mat A,Vec v,PetscInt idx[]) 1666985db425SBarry Smith { 1667985db425SBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 1668985db425SBarry Smith PetscErrorCode ierr; 1669d0f46423SBarry Smith PetscInt i,j,m = A->rmap->n,n = A->cmap->n,p; 1670985db425SBarry Smith PetscScalar *x; 1671985db425SBarry Smith MatScalar *aa = a->v; 1672985db425SBarry Smith 1673985db425SBarry Smith PetscFunctionBegin; 1674985db425SBarry Smith if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 1675985db425SBarry Smith 1676985db425SBarry Smith ierr = VecSet(v,0.0);CHKERRQ(ierr); 1677985db425SBarry Smith ierr = VecGetArray(v,&x);CHKERRQ(ierr); 1678985db425SBarry Smith ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr); 1679d0f46423SBarry Smith if (p != A->rmap->n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 1680985db425SBarry Smith for (i=0; i<m; i++) { 1681985db425SBarry Smith x[i] = aa[i]; if (idx) idx[i] = 0; 1682985db425SBarry Smith for (j=1; j<n; j++){ 1683985db425SBarry Smith if (PetscRealPart(x[i]) < PetscRealPart(aa[i+m*j])) {x[i] = aa[i + m*j]; if (idx) idx[i] = j;} 1684985db425SBarry Smith } 1685985db425SBarry Smith } 1686985db425SBarry Smith ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 1687985db425SBarry Smith PetscFunctionReturn(0); 1688985db425SBarry Smith } 1689985db425SBarry Smith 1690985db425SBarry Smith #undef __FUNCT__ 1691985db425SBarry Smith #define __FUNCT__ "MatGetRowMaxAbs_SeqDense" 1692985db425SBarry Smith PetscErrorCode MatGetRowMaxAbs_SeqDense(Mat A,Vec v,PetscInt idx[]) 1693985db425SBarry Smith { 1694985db425SBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 1695985db425SBarry Smith PetscErrorCode ierr; 1696d0f46423SBarry Smith PetscInt i,j,m = A->rmap->n,n = A->cmap->n,p; 1697985db425SBarry Smith PetscScalar *x; 1698985db425SBarry Smith PetscReal atmp; 1699985db425SBarry Smith MatScalar *aa = a->v; 1700985db425SBarry Smith 1701985db425SBarry Smith PetscFunctionBegin; 1702985db425SBarry Smith if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 1703985db425SBarry Smith 1704985db425SBarry Smith ierr = VecSet(v,0.0);CHKERRQ(ierr); 1705985db425SBarry Smith ierr = VecGetArray(v,&x);CHKERRQ(ierr); 1706985db425SBarry Smith ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr); 1707d0f46423SBarry Smith if (p != A->rmap->n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 1708985db425SBarry Smith for (i=0; i<m; i++) { 17099189402eSHong Zhang x[i] = PetscAbsScalar(aa[i]); 1710985db425SBarry Smith for (j=1; j<n; j++){ 1711985db425SBarry Smith atmp = PetscAbsScalar(aa[i+m*j]); 1712985db425SBarry Smith if (PetscAbsScalar(x[i]) < atmp) {x[i] = atmp; if (idx) idx[i] = j;} 1713985db425SBarry Smith } 1714985db425SBarry Smith } 1715985db425SBarry Smith ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 1716985db425SBarry Smith PetscFunctionReturn(0); 1717985db425SBarry Smith } 1718985db425SBarry Smith 1719985db425SBarry Smith #undef __FUNCT__ 1720985db425SBarry Smith #define __FUNCT__ "MatGetRowMin_SeqDense" 1721985db425SBarry Smith PetscErrorCode MatGetRowMin_SeqDense(Mat A,Vec v,PetscInt idx[]) 1722985db425SBarry Smith { 1723985db425SBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 1724985db425SBarry Smith PetscErrorCode ierr; 1725d0f46423SBarry Smith PetscInt i,j,m = A->rmap->n,n = A->cmap->n,p; 1726985db425SBarry Smith PetscScalar *x; 1727985db425SBarry Smith MatScalar *aa = a->v; 1728985db425SBarry Smith 1729985db425SBarry Smith PetscFunctionBegin; 1730985db425SBarry Smith if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 1731985db425SBarry Smith 1732985db425SBarry Smith ierr = VecSet(v,0.0);CHKERRQ(ierr); 1733985db425SBarry Smith ierr = VecGetArray(v,&x);CHKERRQ(ierr); 1734985db425SBarry Smith ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr); 1735d0f46423SBarry Smith if (p != A->rmap->n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 1736985db425SBarry Smith for (i=0; i<m; i++) { 1737985db425SBarry Smith x[i] = aa[i]; if (idx) idx[i] = 0; 1738985db425SBarry Smith for (j=1; j<n; j++){ 1739985db425SBarry Smith if (PetscRealPart(x[i]) > PetscRealPart(aa[i+m*j])) {x[i] = aa[i + m*j]; if (idx) idx[i] = j;} 1740985db425SBarry Smith } 1741985db425SBarry Smith } 1742985db425SBarry Smith ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 1743985db425SBarry Smith PetscFunctionReturn(0); 1744985db425SBarry Smith } 1745985db425SBarry Smith 17468d0534beSBarry Smith #undef __FUNCT__ 17478d0534beSBarry Smith #define __FUNCT__ "MatGetColumnVector_SeqDense" 17488d0534beSBarry Smith PetscErrorCode MatGetColumnVector_SeqDense(Mat A,Vec v,PetscInt col) 17498d0534beSBarry Smith { 17508d0534beSBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 17518d0534beSBarry Smith PetscErrorCode ierr; 17528d0534beSBarry Smith PetscScalar *x; 17538d0534beSBarry Smith 17548d0534beSBarry Smith PetscFunctionBegin; 17558d0534beSBarry Smith if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 17568d0534beSBarry Smith 17578d0534beSBarry Smith ierr = VecGetArray(v,&x);CHKERRQ(ierr); 1758d0f46423SBarry Smith ierr = PetscMemcpy(x,a->v+col*a->lda,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr); 17598d0534beSBarry Smith ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 17608d0534beSBarry Smith PetscFunctionReturn(0); 17618d0534beSBarry Smith } 17628d0534beSBarry Smith 1763289bc588SBarry Smith /* -------------------------------------------------------------------*/ 1764a5ae1ecdSBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_SeqDense, 1765905e6a2fSBarry Smith MatGetRow_SeqDense, 1766905e6a2fSBarry Smith MatRestoreRow_SeqDense, 1767905e6a2fSBarry Smith MatMult_SeqDense, 176897304618SKris Buschelman /* 4*/ MatMultAdd_SeqDense, 17697c922b88SBarry Smith MatMultTranspose_SeqDense, 17707c922b88SBarry Smith MatMultTransposeAdd_SeqDense, 1771db4efbfdSBarry Smith 0, 1772db4efbfdSBarry Smith 0, 1773db4efbfdSBarry Smith 0, 1774db4efbfdSBarry Smith /*10*/ 0, 1775905e6a2fSBarry Smith MatLUFactor_SeqDense, 1776905e6a2fSBarry Smith MatCholeskyFactor_SeqDense, 177741f059aeSBarry Smith MatSOR_SeqDense, 1778ec8511deSBarry Smith MatTranspose_SeqDense, 177997304618SKris Buschelman /*15*/ MatGetInfo_SeqDense, 1780905e6a2fSBarry Smith MatEqual_SeqDense, 1781905e6a2fSBarry Smith MatGetDiagonal_SeqDense, 1782905e6a2fSBarry Smith MatDiagonalScale_SeqDense, 1783905e6a2fSBarry Smith MatNorm_SeqDense, 1784c0aa2d19SHong Zhang /*20*/ MatAssemblyBegin_SeqDense, 1785c0aa2d19SHong Zhang MatAssemblyEnd_SeqDense, 1786905e6a2fSBarry Smith MatSetOption_SeqDense, 1787905e6a2fSBarry Smith MatZeroEntries_SeqDense, 1788d519adbfSMatthew Knepley /*24*/ MatZeroRows_SeqDense, 1789db4efbfdSBarry Smith 0, 1790db4efbfdSBarry Smith 0, 1791db4efbfdSBarry Smith 0, 1792db4efbfdSBarry Smith 0, 1793d519adbfSMatthew Knepley /*29*/ MatSetUpPreallocation_SeqDense, 1794273d9f13SBarry Smith 0, 1795905e6a2fSBarry Smith 0, 1796905e6a2fSBarry Smith MatGetArray_SeqDense, 1797905e6a2fSBarry Smith MatRestoreArray_SeqDense, 1798d519adbfSMatthew Knepley /*34*/ MatDuplicate_SeqDense, 1799a5ae1ecdSBarry Smith 0, 1800a5ae1ecdSBarry Smith 0, 1801a5ae1ecdSBarry Smith 0, 1802a5ae1ecdSBarry Smith 0, 1803d519adbfSMatthew Knepley /*39*/ MatAXPY_SeqDense, 1804a5ae1ecdSBarry Smith MatGetSubMatrices_SeqDense, 1805a5ae1ecdSBarry Smith 0, 18064b0e389bSBarry Smith MatGetValues_SeqDense, 1807a5ae1ecdSBarry Smith MatCopy_SeqDense, 1808d519adbfSMatthew Knepley /*44*/ MatGetRowMax_SeqDense, 1809a5ae1ecdSBarry Smith MatScale_SeqDense, 1810a5ae1ecdSBarry Smith 0, 1811a5ae1ecdSBarry Smith 0, 1812a5ae1ecdSBarry Smith 0, 1813d519adbfSMatthew Knepley /*49*/ 0, 1814a5ae1ecdSBarry Smith 0, 1815a5ae1ecdSBarry Smith 0, 1816a5ae1ecdSBarry Smith 0, 1817a5ae1ecdSBarry Smith 0, 1818d519adbfSMatthew Knepley /*54*/ 0, 1819a5ae1ecdSBarry Smith 0, 1820a5ae1ecdSBarry Smith 0, 1821a5ae1ecdSBarry Smith 0, 1822a5ae1ecdSBarry Smith 0, 1823d519adbfSMatthew Knepley /*59*/ 0, 1824e03a110bSBarry Smith MatDestroy_SeqDense, 1825e03a110bSBarry Smith MatView_SeqDense, 1826357abbc8SBarry Smith 0, 182797304618SKris Buschelman 0, 1828d519adbfSMatthew Knepley /*64*/ 0, 182997304618SKris Buschelman 0, 183097304618SKris Buschelman 0, 183197304618SKris Buschelman 0, 183297304618SKris Buschelman 0, 1833d519adbfSMatthew Knepley /*69*/ MatGetRowMaxAbs_SeqDense, 183497304618SKris Buschelman 0, 183597304618SKris Buschelman 0, 183697304618SKris Buschelman 0, 183797304618SKris Buschelman 0, 1838d519adbfSMatthew Knepley /*74*/ 0, 183997304618SKris Buschelman 0, 184097304618SKris Buschelman 0, 184197304618SKris Buschelman 0, 184297304618SKris Buschelman 0, 1843d519adbfSMatthew Knepley /*79*/ 0, 184497304618SKris Buschelman 0, 184597304618SKris Buschelman 0, 184697304618SKris Buschelman 0, 1847d519adbfSMatthew Knepley /*83*/ MatLoad_SeqDense, 1848865e5f61SKris Buschelman 0, 18491cbb95d3SBarry Smith MatIsHermitian_SeqDense, 1850865e5f61SKris Buschelman 0, 1851865e5f61SKris Buschelman 0, 1852865e5f61SKris Buschelman 0, 1853d519adbfSMatthew Knepley /*89*/ MatMatMult_SeqDense_SeqDense, 1854a9fe9ddaSSatish Balay MatMatMultSymbolic_SeqDense_SeqDense, 1855a9fe9ddaSSatish Balay MatMatMultNumeric_SeqDense_SeqDense, 1856865e5f61SKris Buschelman 0, 1857865e5f61SKris Buschelman 0, 1858d519adbfSMatthew Knepley /*94*/ 0, 1859a9fe9ddaSSatish Balay MatMatMultTranspose_SeqDense_SeqDense, 1860a9fe9ddaSSatish Balay MatMatMultTransposeSymbolic_SeqDense_SeqDense, 1861a9fe9ddaSSatish Balay MatMatMultTransposeNumeric_SeqDense_SeqDense, 1862284134d9SBarry Smith 0, 1863d519adbfSMatthew Knepley /*99*/ 0, 1864284134d9SBarry Smith 0, 1865284134d9SBarry Smith 0, 1866284134d9SBarry Smith 0, 1867985db425SBarry Smith MatSetSizes_SeqDense, 1868985db425SBarry Smith 0, 1869985db425SBarry Smith 0, 1870985db425SBarry Smith 0, 1871985db425SBarry Smith 0, 1872985db425SBarry Smith 0, 1873d519adbfSMatthew Knepley /*109*/0, 1874985db425SBarry Smith 0, 18758d0534beSBarry Smith MatGetRowMin_SeqDense, 18768d0534beSBarry Smith MatGetColumnVector_SeqDense 1877985db425SBarry Smith }; 187890ace30eSBarry Smith 18794a2ae208SSatish Balay #undef __FUNCT__ 18804a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqDense" 18814b828684SBarry Smith /*@C 1882fafbff53SBarry Smith MatCreateSeqDense - Creates a sequential dense matrix that 1883d65003e9SLois Curfman McInnes is stored in column major order (the usual Fortran 77 manner). Many 1884d65003e9SLois Curfman McInnes of the matrix operations use the BLAS and LAPACK routines. 1885289bc588SBarry Smith 1886db81eaa0SLois Curfman McInnes Collective on MPI_Comm 1887db81eaa0SLois Curfman McInnes 188820563c6bSBarry Smith Input Parameters: 1889db81eaa0SLois Curfman McInnes + comm - MPI communicator, set to PETSC_COMM_SELF 18900c775827SLois Curfman McInnes . m - number of rows 189118f449edSLois Curfman McInnes . n - number of columns 1892db81eaa0SLois Curfman McInnes - data - optional location of matrix data. Set data=PETSC_NULL for PETSc 1893dfc5480cSLois Curfman McInnes to control all matrix memory allocation. 189420563c6bSBarry Smith 189520563c6bSBarry Smith Output Parameter: 189644cd7ae7SLois Curfman McInnes . A - the matrix 189720563c6bSBarry Smith 1898b259b22eSLois Curfman McInnes Notes: 189918f449edSLois Curfman McInnes The data input variable is intended primarily for Fortran programmers 190018f449edSLois Curfman McInnes who wish to allocate their own matrix memory space. Most users should 1901b4fd4287SBarry Smith set data=PETSC_NULL. 190218f449edSLois Curfman McInnes 1903027ccd11SLois Curfman McInnes Level: intermediate 1904027ccd11SLois Curfman McInnes 1905dbd7a890SLois Curfman McInnes .keywords: dense, matrix, LAPACK, BLAS 1906d65003e9SLois Curfman McInnes 1907db81eaa0SLois Curfman McInnes .seealso: MatCreate(), MatCreateMPIDense(), MatSetValues() 190820563c6bSBarry Smith @*/ 1909be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateSeqDense(MPI_Comm comm,PetscInt m,PetscInt n,PetscScalar *data,Mat *A) 1910289bc588SBarry Smith { 1911dfbe8321SBarry Smith PetscErrorCode ierr; 19123b2fbd54SBarry Smith 19133a40ed3dSBarry Smith PetscFunctionBegin; 1914f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 1915f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr); 1916273d9f13SBarry Smith ierr = MatSetType(*A,MATSEQDENSE);CHKERRQ(ierr); 1917273d9f13SBarry Smith ierr = MatSeqDenseSetPreallocation(*A,data);CHKERRQ(ierr); 1918273d9f13SBarry Smith PetscFunctionReturn(0); 1919273d9f13SBarry Smith } 1920273d9f13SBarry Smith 19214a2ae208SSatish Balay #undef __FUNCT__ 1922afc30d2aSLisandro Dalcin #define __FUNCT__ "MatSeqDenseSetPreallocation" 1923273d9f13SBarry Smith /*@C 1924273d9f13SBarry Smith MatSeqDenseSetPreallocation - Sets the array used for storing the matrix elements 1925273d9f13SBarry Smith 1926273d9f13SBarry Smith Collective on MPI_Comm 1927273d9f13SBarry Smith 1928273d9f13SBarry Smith Input Parameters: 1929273d9f13SBarry Smith + A - the matrix 1930273d9f13SBarry Smith - data - the array (or PETSC_NULL) 1931273d9f13SBarry Smith 1932273d9f13SBarry Smith Notes: 1933273d9f13SBarry Smith The data input variable is intended primarily for Fortran programmers 1934273d9f13SBarry Smith who wish to allocate their own matrix memory space. Most users should 1935284134d9SBarry Smith need not call this routine. 1936273d9f13SBarry Smith 1937273d9f13SBarry Smith Level: intermediate 1938273d9f13SBarry Smith 1939273d9f13SBarry Smith .keywords: dense, matrix, LAPACK, BLAS 1940273d9f13SBarry Smith 1941867c911aSBarry Smith .seealso: MatCreate(), MatCreateMPIDense(), MatSetValues(), MatSeqDenseSetLDA() 1942867c911aSBarry Smith 1943273d9f13SBarry Smith @*/ 1944be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetPreallocation(Mat B,PetscScalar data[]) 1945273d9f13SBarry Smith { 1946dfbe8321SBarry Smith PetscErrorCode ierr,(*f)(Mat,PetscScalar[]); 1947a23d5eceSKris Buschelman 1948a23d5eceSKris Buschelman PetscFunctionBegin; 1949a23d5eceSKris Buschelman ierr = PetscObjectQueryFunction((PetscObject)B,"MatSeqDenseSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr); 1950a23d5eceSKris Buschelman if (f) { 1951a23d5eceSKris Buschelman ierr = (*f)(B,data);CHKERRQ(ierr); 1952a23d5eceSKris Buschelman } 1953a23d5eceSKris Buschelman PetscFunctionReturn(0); 1954a23d5eceSKris Buschelman } 1955a23d5eceSKris Buschelman 1956a23d5eceSKris Buschelman EXTERN_C_BEGIN 1957a23d5eceSKris Buschelman #undef __FUNCT__ 1958afc30d2aSLisandro Dalcin #define __FUNCT__ "MatSeqDenseSetPreallocation_SeqDense" 1959be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetPreallocation_SeqDense(Mat B,PetscScalar *data) 1960a23d5eceSKris Buschelman { 1961273d9f13SBarry Smith Mat_SeqDense *b; 1962dfbe8321SBarry Smith PetscErrorCode ierr; 1963273d9f13SBarry Smith 1964273d9f13SBarry Smith PetscFunctionBegin; 1965273d9f13SBarry Smith B->preallocated = PETSC_TRUE; 1966273d9f13SBarry Smith b = (Mat_SeqDense*)B->data; 1967d0f46423SBarry Smith if (b->lda <= 0) b->lda = B->rmap->n; 19689e8f95c4SLisandro Dalcin if (!data) { /* petsc-allocated storage */ 19699e8f95c4SLisandro Dalcin if (!b->user_alloc) { ierr = PetscFree(b->v);CHKERRQ(ierr); } 19705afd5e0cSBarry Smith ierr = PetscMalloc(b->lda*b->Nmax*sizeof(PetscScalar),&b->v);CHKERRQ(ierr); 1971284134d9SBarry Smith ierr = PetscMemzero(b->v,b->lda*b->Nmax*sizeof(PetscScalar));CHKERRQ(ierr); 1972284134d9SBarry Smith ierr = PetscLogObjectMemory(B,b->lda*b->Nmax*sizeof(PetscScalar));CHKERRQ(ierr); 19739e8f95c4SLisandro Dalcin b->user_alloc = PETSC_FALSE; 1974273d9f13SBarry Smith } else { /* user-allocated storage */ 19759e8f95c4SLisandro Dalcin if (!b->user_alloc) { ierr = PetscFree(b->v);CHKERRQ(ierr); } 1976273d9f13SBarry Smith b->v = data; 1977273d9f13SBarry Smith b->user_alloc = PETSC_TRUE; 1978273d9f13SBarry Smith } 19790450473dSBarry Smith B->assembled = PETSC_TRUE; 1980273d9f13SBarry Smith PetscFunctionReturn(0); 1981273d9f13SBarry Smith } 1982a23d5eceSKris Buschelman EXTERN_C_END 1983273d9f13SBarry Smith 19841b807ce4Svictorle #undef __FUNCT__ 19851b807ce4Svictorle #define __FUNCT__ "MatSeqDenseSetLDA" 19861b807ce4Svictorle /*@C 19871b807ce4Svictorle MatSeqDenseSetLDA - Declare the leading dimension of the user-provided array 19881b807ce4Svictorle 19891b807ce4Svictorle Input parameter: 19901b807ce4Svictorle + A - the matrix 19911b807ce4Svictorle - lda - the leading dimension 19921b807ce4Svictorle 19931b807ce4Svictorle Notes: 1994867c911aSBarry Smith This routine is to be used in conjunction with MatSeqDenseSetPreallocation(); 19951b807ce4Svictorle it asserts that the preallocation has a leading dimension (the LDA parameter 19961b807ce4Svictorle of Blas and Lapack fame) larger than M, the first dimension of the matrix. 19971b807ce4Svictorle 19981b807ce4Svictorle Level: intermediate 19991b807ce4Svictorle 20001b807ce4Svictorle .keywords: dense, matrix, LAPACK, BLAS 20011b807ce4Svictorle 2002284134d9SBarry Smith .seealso: MatCreate(), MatCreateSeqDense(), MatSeqDenseSetPreallocation(), MatSetMaximumSize() 2003867c911aSBarry Smith 20041b807ce4Svictorle @*/ 2005be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetLDA(Mat B,PetscInt lda) 20061b807ce4Svictorle { 20071b807ce4Svictorle Mat_SeqDense *b = (Mat_SeqDense*)B->data; 200821a2c019SBarry Smith 20091b807ce4Svictorle PetscFunctionBegin; 2010d0f46423SBarry Smith if (lda < B->rmap->n) SETERRQ2(PETSC_ERR_ARG_SIZ,"LDA %D must be at least matrix dimension %D",lda,B->rmap->n); 20111b807ce4Svictorle b->lda = lda; 201221a2c019SBarry Smith b->changelda = PETSC_FALSE; 201321a2c019SBarry Smith b->Mmax = PetscMax(b->Mmax,lda); 20141b807ce4Svictorle PetscFunctionReturn(0); 20151b807ce4Svictorle } 20161b807ce4Svictorle 20170bad9183SKris Buschelman /*MC 2018fafad747SKris Buschelman MATSEQDENSE - MATSEQDENSE = "seqdense" - A matrix type to be used for sequential dense matrices. 20190bad9183SKris Buschelman 20200bad9183SKris Buschelman Options Database Keys: 20210bad9183SKris Buschelman . -mat_type seqdense - sets the matrix type to "seqdense" during a call to MatSetFromOptions() 20220bad9183SKris Buschelman 20230bad9183SKris Buschelman Level: beginner 20240bad9183SKris Buschelman 202589665df3SBarry Smith .seealso: MatCreateSeqDense() 202689665df3SBarry Smith 20270bad9183SKris Buschelman M*/ 20280bad9183SKris Buschelman 2029273d9f13SBarry Smith EXTERN_C_BEGIN 20304a2ae208SSatish Balay #undef __FUNCT__ 20314a2ae208SSatish Balay #define __FUNCT__ "MatCreate_SeqDense" 2032be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_SeqDense(Mat B) 2033273d9f13SBarry Smith { 2034273d9f13SBarry Smith Mat_SeqDense *b; 2035dfbe8321SBarry Smith PetscErrorCode ierr; 20367c334f02SBarry Smith PetscMPIInt size; 2037273d9f13SBarry Smith 2038273d9f13SBarry Smith PetscFunctionBegin; 20397adad957SLisandro Dalcin ierr = MPI_Comm_size(((PetscObject)B)->comm,&size);CHKERRQ(ierr); 204029bbc08cSBarry Smith if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"Comm must be of size 1"); 204155659b69SBarry Smith 204226283091SBarry Smith ierr = PetscLayoutSetBlockSize(B->rmap,1);CHKERRQ(ierr); 204326283091SBarry Smith ierr = PetscLayoutSetBlockSize(B->cmap,1);CHKERRQ(ierr); 204426283091SBarry Smith ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 204526283091SBarry Smith ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 2046273d9f13SBarry Smith 204738f2d2fdSLisandro Dalcin ierr = PetscNewLog(B,Mat_SeqDense,&b);CHKERRQ(ierr); 2048549d3d68SSatish Balay ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 204990f02eecSBarry Smith B->mapping = 0; 205044cd7ae7SLois Curfman McInnes B->data = (void*)b; 205118f449edSLois Curfman McInnes 2052a5ae1ecdSBarry Smith 205344cd7ae7SLois Curfman McInnes b->pivots = 0; 2054273d9f13SBarry Smith b->roworiented = PETSC_TRUE; 2055273d9f13SBarry Smith b->v = 0; 2056d0f46423SBarry Smith b->lda = B->rmap->n; 205721a2c019SBarry Smith b->changelda = PETSC_FALSE; 2058d0f46423SBarry Smith b->Mmax = B->rmap->n; 2059d0f46423SBarry Smith b->Nmax = B->cmap->n; 20604e220ebcSLois Curfman McInnes 2061b24902e0SBarry Smith 2062ec1065edSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_petsc_C", 2063b24902e0SBarry Smith "MatGetFactor_seqdense_petsc", 2064b24902e0SBarry Smith MatGetFactor_seqdense_petsc);CHKERRQ(ierr); 2065a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqDenseSetPreallocation_C", 2066a23d5eceSKris Buschelman "MatSeqDenseSetPreallocation_SeqDense", 2067a23d5eceSKris Buschelman MatSeqDenseSetPreallocation_SeqDense);CHKERRQ(ierr); 20684ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMult_seqaij_seqdense_C", 20694ae313f4SHong Zhang "MatMatMult_SeqAIJ_SeqDense", 20704ae313f4SHong Zhang MatMatMult_SeqAIJ_SeqDense);CHKERRQ(ierr); 20714ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMultSymbolic_seqaij_seqdense_C", 20724ae313f4SHong Zhang "MatMatMultSymbolic_SeqAIJ_SeqDense", 20734ae313f4SHong Zhang MatMatMultSymbolic_SeqAIJ_SeqDense);CHKERRQ(ierr); 20744ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMultNumeric_seqaij_seqdense_C", 20754ae313f4SHong Zhang "MatMatMultNumeric_SeqAIJ_SeqDense", 20764ae313f4SHong Zhang MatMatMultNumeric_SeqAIJ_SeqDense);CHKERRQ(ierr); 207717667f90SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQDENSE);CHKERRQ(ierr); 20783a40ed3dSBarry Smith PetscFunctionReturn(0); 2079289bc588SBarry Smith } 2080c0aa2d19SHong Zhang 2081c0aa2d19SHong Zhang 2082273d9f13SBarry Smith EXTERN_C_END 2083