1be1d678aSKris Buschelman #define PETSCMAT_DLL 2be1d678aSKris Buschelman 367e560aaSBarry Smith /* 467e560aaSBarry Smith Defines the basic matrix operations for sequential dense. 567e560aaSBarry Smith */ 6289bc588SBarry Smith 770f55243SBarry 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; 210805154bSBarry Smith N = PetscBLASIntCast(X->rmap.n*X->cmap.n); 220805154bSBarry Smith m = PetscBLASIntCast(X->rmap.n); 230805154bSBarry Smith ldax = PetscBLASIntCast(x->lda); 240805154bSBarry Smith lday = PetscBLASIntCast(y->lda); 25a5ce6ee0Svictorle if (ldax>m || lday>m) { 26899cda47SBarry 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 } 32efee365bSSatish Balay ierr = PetscLogFlops(2*N-1);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 { 40899cda47SBarry Smith PetscInt N = A->rmap.n*A->cmap.n; 413a40ed3dSBarry Smith 423a40ed3dSBarry Smith PetscFunctionBegin; 43899cda47SBarry Smith info->rows_global = (double)A->rmap.n; 44899cda47SBarry Smith info->columns_global = (double)A->cmap.n; 45899cda47SBarry Smith info->rows_local = (double)A->rmap.n; 46899cda47SBarry Smith info->columns_local = (double)A->cmap.n; 474e220ebcSLois Curfman McInnes info->block_size = 1.0; 484e220ebcSLois Curfman McInnes info->nz_allocated = (double)N; 496de62eeeSBarry Smith info->nz_used = (double)N; 506de62eeeSBarry Smith info->nz_unneeded = (double)0; 514e220ebcSLois Curfman McInnes info->assemblies = (double)A->num_ass; 524e220ebcSLois Curfman McInnes info->mallocs = 0; 537adad957SLisandro Dalcin info->memory = ((PetscObject)A)->mem; 544e220ebcSLois Curfman McInnes info->fill_ratio_given = 0; 554e220ebcSLois Curfman McInnes info->fill_ratio_needed = 0; 564e220ebcSLois Curfman McInnes info->factor_mallocs = 0; 573a40ed3dSBarry Smith PetscFunctionReturn(0); 58289bc588SBarry Smith } 59289bc588SBarry Smith 604a2ae208SSatish Balay #undef __FUNCT__ 614a2ae208SSatish Balay #define __FUNCT__ "MatScale_SeqDense" 62f4df32b1SMatthew Knepley PetscErrorCode MatScale_SeqDense(Mat A,PetscScalar alpha) 6380cd9d93SLois Curfman McInnes { 64273d9f13SBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 65f4df32b1SMatthew Knepley PetscScalar oalpha = alpha; 66efee365bSSatish Balay PetscErrorCode ierr; 670805154bSBarry Smith PetscBLASInt one = 1,j,nz,lda = PetscBLASIntCast(a->lda); 6880cd9d93SLois Curfman McInnes 693a40ed3dSBarry Smith PetscFunctionBegin; 70899cda47SBarry Smith if (lda>A->rmap.n) { 710805154bSBarry Smith nz = PetscBLASIntCast(A->rmap.n); 72899cda47SBarry Smith for (j=0; j<A->cmap.n; j++) { 73f4df32b1SMatthew Knepley BLASscal_(&nz,&oalpha,a->v+j*lda,&one); 74a5ce6ee0Svictorle } 75a5ce6ee0Svictorle } else { 760805154bSBarry Smith nz = PetscBLASIntCast(A->rmap.n*A->cmap.n); 77f4df32b1SMatthew Knepley BLASscal_(&nz,&oalpha,a->v,&one); 78a5ce6ee0Svictorle } 79efee365bSSatish Balay ierr = PetscLogFlops(nz);CHKERRQ(ierr); 803a40ed3dSBarry Smith PetscFunctionReturn(0); 8180cd9d93SLois Curfman McInnes } 8280cd9d93SLois Curfman McInnes 831cbb95d3SBarry Smith #undef __FUNCT__ 841cbb95d3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqDense" 851cbb95d3SBarry Smith PetscErrorCode MatIsHermitian_SeqDense(Mat A,PetscReal rtol,PetscTruth *fl) 861cbb95d3SBarry Smith { 871cbb95d3SBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 881cbb95d3SBarry Smith PetscInt i,j,m = A->rmap.n,N; 891cbb95d3SBarry Smith PetscScalar *v = a->v; 901cbb95d3SBarry Smith 911cbb95d3SBarry Smith PetscFunctionBegin; 921cbb95d3SBarry Smith *fl = PETSC_FALSE; 931cbb95d3SBarry Smith if (A->rmap.n != A->cmap.n) PetscFunctionReturn(0); 941cbb95d3SBarry Smith N = a->lda; 951cbb95d3SBarry Smith 961cbb95d3SBarry Smith for (i=0; i<m; i++) { 971cbb95d3SBarry Smith for (j=i+1; j<m; j++) { 981cbb95d3SBarry Smith if (PetscAbsScalar(v[i+j*N] - PetscConj(v[j+i*N])) > rtol) PetscFunctionReturn(0); 991cbb95d3SBarry Smith } 1001cbb95d3SBarry Smith } 1011cbb95d3SBarry Smith *fl = PETSC_TRUE; 1021cbb95d3SBarry Smith PetscFunctionReturn(0); 1031cbb95d3SBarry Smith } 1041cbb95d3SBarry Smith 105289bc588SBarry Smith /* ---------------------------------------------------------------*/ 1066831982aSBarry Smith /* COMMENT: I have chosen to hide row permutation in the pivots, 1076831982aSBarry Smith rather than put it in the Mat->row slot.*/ 1084a2ae208SSatish Balay #undef __FUNCT__ 1094a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqDense" 110dfbe8321SBarry Smith PetscErrorCode MatLUFactor_SeqDense(Mat A,IS row,IS col,MatFactorInfo *minfo) 111289bc588SBarry Smith { 112a07ab388SBarry Smith #if defined(PETSC_MISSING_LAPACK_GETRF) 11381f479a6Svictorle PetscFunctionBegin; 114a07ab388SBarry Smith SETERRQ(PETSC_ERR_SUP,"GETRF - Lapack routine is unavailable."); 115a07ab388SBarry Smith #else 116c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 1176849ba73SBarry Smith PetscErrorCode ierr; 1180805154bSBarry Smith PetscBLASInt n,m,info; 11955659b69SBarry Smith 1203a40ed3dSBarry Smith PetscFunctionBegin; 1210805154bSBarry Smith n = PetscBLASIntCast(A->cmap.n); 1220805154bSBarry Smith m = PetscBLASIntCast(A->rmap.n); 123289bc588SBarry Smith if (!mat->pivots) { 124899cda47SBarry Smith ierr = PetscMalloc((A->rmap.n+1)*sizeof(PetscBLASInt),&mat->pivots);CHKERRQ(ierr); 125899cda47SBarry Smith ierr = PetscLogObjectMemory(A,A->rmap.n*sizeof(PetscBLASInt));CHKERRQ(ierr); 126289bc588SBarry Smith } 127f1af5d2fSBarry Smith A->factor = FACTOR_LU; 128899cda47SBarry Smith if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0); 12971044d3cSBarry Smith LAPACKgetrf_(&m,&n,mat->v,&mat->lda,mat->pivots,&info); 13029bbc08cSBarry Smith if (info<0) SETERRQ(PETSC_ERR_LIB,"Bad argument to LU factorization"); 13129bbc08cSBarry Smith if (info>0) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,"Bad LU factorization"); 132899cda47SBarry Smith ierr = PetscLogFlops((2*A->cmap.n*A->cmap.n*A->cmap.n)/3);CHKERRQ(ierr); 133a07ab388SBarry Smith #endif 1343a40ed3dSBarry Smith PetscFunctionReturn(0); 135289bc588SBarry Smith } 1366ee01492SSatish Balay 1374a2ae208SSatish Balay #undef __FUNCT__ 1384a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqDense" 139dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat) 14002cad45dSBarry Smith { 14102cad45dSBarry Smith Mat_SeqDense *mat = (Mat_SeqDense*)A->data,*l; 1426849ba73SBarry Smith PetscErrorCode ierr; 14313f74950SBarry Smith PetscInt lda = (PetscInt)mat->lda,j,m; 14402cad45dSBarry Smith Mat newi; 14502cad45dSBarry Smith 1463a40ed3dSBarry Smith PetscFunctionBegin; 1477adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&newi);CHKERRQ(ierr); 148899cda47SBarry Smith ierr = MatSetSizes(newi,A->rmap.n,A->cmap.n,A->rmap.n,A->cmap.n);CHKERRQ(ierr); 1497adad957SLisandro Dalcin ierr = MatSetType(newi,((PetscObject)A)->type_name);CHKERRQ(ierr); 1505c5985e7SKris Buschelman ierr = MatSeqDenseSetPreallocation(newi,PETSC_NULL);CHKERRQ(ierr); 1515609ef8eSBarry Smith if (cpvalues == MAT_COPY_VALUES) { 152a5ce6ee0Svictorle l = (Mat_SeqDense*)newi->data; 153899cda47SBarry Smith if (lda>A->rmap.n) { 154899cda47SBarry Smith m = A->rmap.n; 155899cda47SBarry Smith for (j=0; j<A->cmap.n; j++) { 156a5ce6ee0Svictorle ierr = PetscMemcpy(l->v+j*m,mat->v+j*lda,m*sizeof(PetscScalar));CHKERRQ(ierr); 157a5ce6ee0Svictorle } 158a5ce6ee0Svictorle } else { 159899cda47SBarry Smith ierr = PetscMemcpy(l->v,mat->v,A->rmap.n*A->cmap.n*sizeof(PetscScalar));CHKERRQ(ierr); 16002cad45dSBarry Smith } 161a5ce6ee0Svictorle } 16202cad45dSBarry Smith newi->assembled = PETSC_TRUE; 16302cad45dSBarry Smith *newmat = newi; 1643a40ed3dSBarry Smith PetscFunctionReturn(0); 16502cad45dSBarry Smith } 16602cad45dSBarry Smith 1674a2ae208SSatish Balay #undef __FUNCT__ 1684a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorSymbolic_SeqDense" 169dfbe8321SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqDense(Mat A,IS row,IS col,MatFactorInfo *info,Mat *fact) 170289bc588SBarry Smith { 171dfbe8321SBarry Smith PetscErrorCode ierr; 1723a40ed3dSBarry Smith 1733a40ed3dSBarry Smith PetscFunctionBegin; 1745609ef8eSBarry Smith ierr = MatDuplicate_SeqDense(A,MAT_DO_NOT_COPY_VALUES,fact);CHKERRQ(ierr); 1753a40ed3dSBarry Smith PetscFunctionReturn(0); 176289bc588SBarry Smith } 1776ee01492SSatish Balay 1784a2ae208SSatish Balay #undef __FUNCT__ 1794a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqDense" 180af281ebdSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqDense(Mat A,MatFactorInfo *info_dummy,Mat *fact) 181289bc588SBarry Smith { 18202cad45dSBarry Smith Mat_SeqDense *mat = (Mat_SeqDense*)A->data,*l = (Mat_SeqDense*)(*fact)->data; 1836849ba73SBarry Smith PetscErrorCode ierr; 184899cda47SBarry Smith PetscInt lda1=mat->lda,lda2=l->lda, m=A->rmap.n,n=A->cmap.n, j; 1854482741eSBarry Smith MatFactorInfo info; 1863a40ed3dSBarry Smith 1873a40ed3dSBarry Smith PetscFunctionBegin; 18802cad45dSBarry Smith /* copy the numerical values */ 1891b807ce4Svictorle if (lda1>m || lda2>m ) { 1901b807ce4Svictorle for (j=0; j<n; j++) { 1911b807ce4Svictorle ierr = PetscMemcpy(l->v+j*lda2,mat->v+j*lda1,m*sizeof(PetscScalar));CHKERRQ(ierr); 1921b807ce4Svictorle } 1931b807ce4Svictorle } else { 194899cda47SBarry Smith ierr = PetscMemcpy(l->v,mat->v,A->rmap.n*A->cmap.n*sizeof(PetscScalar));CHKERRQ(ierr); 1951b807ce4Svictorle } 19602cad45dSBarry Smith (*fact)->factor = 0; 1974482741eSBarry Smith ierr = MatLUFactor(*fact,0,0,&info);CHKERRQ(ierr); 1983a40ed3dSBarry Smith PetscFunctionReturn(0); 199289bc588SBarry Smith } 2006ee01492SSatish Balay 2014a2ae208SSatish Balay #undef __FUNCT__ 2024a2ae208SSatish Balay #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqDense" 203dfbe8321SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqDense(Mat A,IS row,MatFactorInfo *info,Mat *fact) 204289bc588SBarry Smith { 205dfbe8321SBarry Smith PetscErrorCode ierr; 2063a40ed3dSBarry Smith 2073a40ed3dSBarry Smith PetscFunctionBegin; 208ceb03754SKris Buschelman ierr = MatConvert(A,MATSAME,MAT_INITIAL_MATRIX,fact);CHKERRQ(ierr); 2093a40ed3dSBarry Smith PetscFunctionReturn(0); 210289bc588SBarry Smith } 2116ee01492SSatish Balay 2124a2ae208SSatish Balay #undef __FUNCT__ 2134a2ae208SSatish Balay #define __FUNCT__ "MatCholeskyFactor_SeqDense" 214dfbe8321SBarry Smith PetscErrorCode MatCholeskyFactor_SeqDense(Mat A,IS perm,MatFactorInfo *factinfo) 215289bc588SBarry Smith { 216a07ab388SBarry Smith #if defined(PETSC_MISSING_LAPACK_POTRF) 217a07ab388SBarry Smith PetscFunctionBegin; 218a07ab388SBarry Smith SETERRQ(PETSC_ERR_SUP,"POTRF - Lapack routine is unavailable."); 219a07ab388SBarry Smith #else 220c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 2216849ba73SBarry Smith PetscErrorCode ierr; 2220805154bSBarry Smith PetscBLASInt info,n = PetscBLASIntCast(A->cmap.n); 22355659b69SBarry Smith 2243a40ed3dSBarry Smith PetscFunctionBegin; 225606d414cSSatish Balay ierr = PetscFree(mat->pivots);CHKERRQ(ierr); 226752f5567SLois Curfman McInnes mat->pivots = 0; 22705b42c5fSBarry Smith 228899cda47SBarry Smith if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0); 22971044d3cSBarry Smith LAPACKpotrf_("L",&n,mat->v,&mat->lda,&info); 23077431f27SBarry Smith if (info) SETERRQ1(PETSC_ERR_MAT_CH_ZRPVT,"Bad factorization: zero pivot in row %D",(PetscInt)info-1); 231c0bbcb79SLois Curfman McInnes A->factor = FACTOR_CHOLESKY; 232899cda47SBarry Smith ierr = PetscLogFlops((A->cmap.n*A->cmap.n*A->cmap.n)/3);CHKERRQ(ierr); 233a07ab388SBarry Smith #endif 2343a40ed3dSBarry Smith PetscFunctionReturn(0); 235289bc588SBarry Smith } 236289bc588SBarry Smith 2374a2ae208SSatish Balay #undef __FUNCT__ 2380b4b3355SBarry Smith #define __FUNCT__ "MatCholeskyFactorNumeric_SeqDense" 239af281ebdSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqDense(Mat A,MatFactorInfo *info_dummy,Mat *fact) 2400b4b3355SBarry Smith { 241dfbe8321SBarry Smith PetscErrorCode ierr; 24215e8a5b3SHong Zhang MatFactorInfo info; 2430b4b3355SBarry Smith 2440b4b3355SBarry Smith PetscFunctionBegin; 24515e8a5b3SHong Zhang info.fill = 1.0; 24615e8a5b3SHong Zhang ierr = MatCholeskyFactor_SeqDense(*fact,0,&info);CHKERRQ(ierr); 2470b4b3355SBarry Smith PetscFunctionReturn(0); 2480b4b3355SBarry Smith } 2490b4b3355SBarry Smith 2500b4b3355SBarry Smith #undef __FUNCT__ 2514a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqDense" 252dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqDense(Mat A,Vec xx,Vec yy) 253289bc588SBarry Smith { 254c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 2556849ba73SBarry Smith PetscErrorCode ierr; 25687828ca2SBarry Smith PetscScalar *x,*y; 2570805154bSBarry Smith PetscBLASInt one = 1,info,m = PetscBLASIntCast(A->rmap.n); 25867e560aaSBarry Smith 2593a40ed3dSBarry Smith PetscFunctionBegin; 2601ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2611ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 262899cda47SBarry Smith ierr = PetscMemcpy(y,x,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr); 263c0bbcb79SLois Curfman McInnes if (A->factor == FACTOR_LU) { 264ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS) 26580444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable."); 266ae7cfcebSSatish Balay #else 26771044d3cSBarry Smith LAPACKgetrs_("N",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info); 2684ce68768SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"GETRS - Bad solve"); 269ae7cfcebSSatish Balay #endif 2707a97a34bSBarry Smith } else if (A->factor == FACTOR_CHOLESKY){ 271ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS) 27280444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable."); 273ae7cfcebSSatish Balay #else 27471044d3cSBarry Smith LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info); 2754ce68768SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"POTRS Bad solve"); 276ae7cfcebSSatish Balay #endif 277289bc588SBarry Smith } 27829bbc08cSBarry Smith else SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Matrix must be factored to solve"); 2791ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2801ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 281899cda47SBarry Smith ierr = PetscLogFlops(2*A->cmap.n*A->cmap.n - A->cmap.n);CHKERRQ(ierr); 2823a40ed3dSBarry Smith PetscFunctionReturn(0); 283289bc588SBarry Smith } 2846ee01492SSatish Balay 2854a2ae208SSatish Balay #undef __FUNCT__ 2864a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqDense" 287dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqDense(Mat A,Vec xx,Vec yy) 288da3a660dSBarry Smith { 289c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 290dfbe8321SBarry Smith PetscErrorCode ierr; 29187828ca2SBarry Smith PetscScalar *x,*y; 2920805154bSBarry Smith PetscBLASInt one = 1,info,m = PetscBLASIntCast(A->rmap.n); 29367e560aaSBarry Smith 2943a40ed3dSBarry Smith PetscFunctionBegin; 2951ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2961ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 297899cda47SBarry Smith ierr = PetscMemcpy(y,x,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr); 298752f5567SLois Curfman McInnes /* assume if pivots exist then use LU; else Cholesky */ 299da3a660dSBarry Smith if (mat->pivots) { 300ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS) 30180444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable."); 302ae7cfcebSSatish Balay #else 30371044d3cSBarry Smith LAPACKgetrs_("T",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info); 3044ce68768SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"POTRS - Bad solve"); 305ae7cfcebSSatish Balay #endif 3067a97a34bSBarry Smith } else { 307ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS) 30880444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable."); 309ae7cfcebSSatish Balay #else 31071044d3cSBarry Smith LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info); 3114ce68768SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"POTRS - Bad solve"); 312ae7cfcebSSatish Balay #endif 313da3a660dSBarry Smith } 3141ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3151ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 316899cda47SBarry Smith ierr = PetscLogFlops(2*A->cmap.n*A->cmap.n - A->cmap.n);CHKERRQ(ierr); 3173a40ed3dSBarry Smith PetscFunctionReturn(0); 318da3a660dSBarry Smith } 3196ee01492SSatish Balay 3204a2ae208SSatish Balay #undef __FUNCT__ 3214a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqDense" 322dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy) 323da3a660dSBarry Smith { 324c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 325dfbe8321SBarry Smith PetscErrorCode ierr; 32687828ca2SBarry Smith PetscScalar *x,*y,sone = 1.0; 327da3a660dSBarry Smith Vec tmp = 0; 3280805154bSBarry Smith PetscBLASInt one = 1,info,m = PetscBLASIntCast(A->rmap.n); 32967e560aaSBarry Smith 3303a40ed3dSBarry Smith PetscFunctionBegin; 3311ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3321ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 333899cda47SBarry Smith if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0); 334da3a660dSBarry Smith if (yy == zz) { 33578b31e54SBarry Smith ierr = VecDuplicate(yy,&tmp);CHKERRQ(ierr); 33652e6d16bSBarry Smith ierr = PetscLogObjectParent(A,tmp);CHKERRQ(ierr); 33778b31e54SBarry Smith ierr = VecCopy(yy,tmp);CHKERRQ(ierr); 338da3a660dSBarry Smith } 339899cda47SBarry Smith ierr = PetscMemcpy(y,x,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr); 340752f5567SLois Curfman McInnes /* assume if pivots exist then use LU; else Cholesky */ 341da3a660dSBarry Smith if (mat->pivots) { 342ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS) 34380444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable."); 344ae7cfcebSSatish Balay #else 34571044d3cSBarry Smith LAPACKgetrs_("N",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info); 3462ffef20aSBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve"); 347ae7cfcebSSatish Balay #endif 348a8c6a408SBarry Smith } else { 349ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS) 35080444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable."); 351ae7cfcebSSatish Balay #else 35271044d3cSBarry Smith LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info); 3532ffef20aSBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve"); 354ae7cfcebSSatish Balay #endif 355da3a660dSBarry Smith } 3562dcb1b2aSMatthew Knepley if (tmp) {ierr = VecAXPY(yy,sone,tmp);CHKERRQ(ierr); ierr = VecDestroy(tmp);CHKERRQ(ierr);} 3572dcb1b2aSMatthew Knepley else {ierr = VecAXPY(yy,sone,zz);CHKERRQ(ierr);} 3581ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3591ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 360899cda47SBarry Smith ierr = PetscLogFlops(2*A->cmap.n*A->cmap.n);CHKERRQ(ierr); 3613a40ed3dSBarry Smith PetscFunctionReturn(0); 362da3a660dSBarry Smith } 36367e560aaSBarry Smith 3644a2ae208SSatish Balay #undef __FUNCT__ 3654a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqDense" 366dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy) 367da3a660dSBarry Smith { 368c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 3696849ba73SBarry Smith PetscErrorCode ierr; 37087828ca2SBarry Smith PetscScalar *x,*y,sone = 1.0; 371da3a660dSBarry Smith Vec tmp; 3720805154bSBarry Smith PetscBLASInt one = 1,info,m = PetscBLASIntCast(A->rmap.n); 37367e560aaSBarry Smith 3743a40ed3dSBarry Smith PetscFunctionBegin; 375899cda47SBarry Smith if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0); 3761ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3771ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 378da3a660dSBarry Smith if (yy == zz) { 37978b31e54SBarry Smith ierr = VecDuplicate(yy,&tmp);CHKERRQ(ierr); 38052e6d16bSBarry Smith ierr = PetscLogObjectParent(A,tmp);CHKERRQ(ierr); 38178b31e54SBarry Smith ierr = VecCopy(yy,tmp);CHKERRQ(ierr); 382da3a660dSBarry Smith } 383899cda47SBarry Smith ierr = PetscMemcpy(y,x,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr); 384752f5567SLois Curfman McInnes /* assume if pivots exist then use LU; else Cholesky */ 385da3a660dSBarry Smith if (mat->pivots) { 386ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS) 38780444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable."); 388ae7cfcebSSatish Balay #else 38971044d3cSBarry Smith LAPACKgetrs_("T",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info); 3902ffef20aSBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve"); 391ae7cfcebSSatish Balay #endif 3923a40ed3dSBarry Smith } else { 393ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS) 39480444007SBarry Smith SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable."); 395ae7cfcebSSatish Balay #else 39671044d3cSBarry Smith LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info); 3972ffef20aSBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve"); 398ae7cfcebSSatish Balay #endif 399da3a660dSBarry Smith } 40090f02eecSBarry Smith if (tmp) { 4012dcb1b2aSMatthew Knepley ierr = VecAXPY(yy,sone,tmp);CHKERRQ(ierr); 40290f02eecSBarry Smith ierr = VecDestroy(tmp);CHKERRQ(ierr); 4033a40ed3dSBarry Smith } else { 4042dcb1b2aSMatthew Knepley ierr = VecAXPY(yy,sone,zz);CHKERRQ(ierr); 40590f02eecSBarry Smith } 4061ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 4071ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 408899cda47SBarry Smith ierr = PetscLogFlops(2*A->cmap.n*A->cmap.n);CHKERRQ(ierr); 4093a40ed3dSBarry Smith PetscFunctionReturn(0); 410da3a660dSBarry Smith } 411289bc588SBarry Smith /* ------------------------------------------------------------------*/ 4124a2ae208SSatish Balay #undef __FUNCT__ 4134a2ae208SSatish Balay #define __FUNCT__ "MatRelax_SeqDense" 41413f74950SBarry Smith PetscErrorCode MatRelax_SeqDense(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal shift,PetscInt its,PetscInt lits,Vec xx) 415289bc588SBarry Smith { 416c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 41787828ca2SBarry Smith PetscScalar *x,*b,*v = mat->v,zero = 0.0,xt; 418dfbe8321SBarry Smith PetscErrorCode ierr; 419899cda47SBarry Smith PetscInt m = A->rmap.n,i; 420aa482453SBarry Smith #if !defined(PETSC_USE_COMPLEX) 4210805154bSBarry Smith PetscBLASInt o = 1,bm = PetscBLASIntCast(m); 422bc1b551cSSatish Balay #endif 423289bc588SBarry Smith 4243a40ed3dSBarry Smith PetscFunctionBegin; 425289bc588SBarry Smith if (flag & SOR_ZERO_INITIAL_GUESS) { 42671044d3cSBarry Smith /* this is a hack fix, should have another version without the second BLASdot */ 4272dcb1b2aSMatthew Knepley ierr = VecSet(xx,zero);CHKERRQ(ierr); 428289bc588SBarry Smith } 4291ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 4301ebc52fbSHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 431b965ef7fSBarry Smith its = its*lits; 43277431f27SBarry Smith if (its <= 0) SETERRQ2(PETSC_ERR_ARG_WRONG,"Relaxation requires global its %D and local its %D both positive",its,lits); 433289bc588SBarry Smith while (its--) { 434fccaa45eSBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){ 435289bc588SBarry Smith for (i=0; i<m; i++) { 436aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 437f1747703SBarry Smith /* cannot use BLAS dot for complex because compiler/linker is 438f1747703SBarry Smith not happy about returning a double complex */ 43913f74950SBarry Smith PetscInt _i; 44087828ca2SBarry Smith PetscScalar sum = b[i]; 441f1747703SBarry Smith for (_i=0; _i<m; _i++) { 4423f6de6efSSatish Balay sum -= PetscConj(v[i+_i*m])*x[_i]; 443f1747703SBarry Smith } 444f1747703SBarry Smith xt = sum; 445f1747703SBarry Smith #else 44671044d3cSBarry Smith xt = b[i] - BLASdot_(&bm,v+i,&bm,x,&o); 447f1747703SBarry Smith #endif 44855a1b374SBarry Smith x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift); 449289bc588SBarry Smith } 450289bc588SBarry Smith } 451fccaa45eSBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){ 452289bc588SBarry Smith for (i=m-1; i>=0; i--) { 453aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 454f1747703SBarry Smith /* cannot use BLAS dot for complex because compiler/linker is 455f1747703SBarry Smith not happy about returning a double complex */ 45613f74950SBarry Smith PetscInt _i; 45787828ca2SBarry Smith PetscScalar sum = b[i]; 458f1747703SBarry Smith for (_i=0; _i<m; _i++) { 4593f6de6efSSatish Balay sum -= PetscConj(v[i+_i*m])*x[_i]; 460f1747703SBarry Smith } 461f1747703SBarry Smith xt = sum; 462f1747703SBarry Smith #else 46371044d3cSBarry Smith xt = b[i] - BLASdot_(&bm,v+i,&bm,x,&o); 464f1747703SBarry Smith #endif 46555a1b374SBarry Smith x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift); 466289bc588SBarry Smith } 467289bc588SBarry Smith } 468289bc588SBarry Smith } 4691ebc52fbSHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 4701ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 4713a40ed3dSBarry Smith PetscFunctionReturn(0); 472289bc588SBarry Smith } 473289bc588SBarry Smith 474289bc588SBarry Smith /* -----------------------------------------------------------------*/ 4754a2ae208SSatish Balay #undef __FUNCT__ 4764a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_SeqDense" 477dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_SeqDense(Mat A,Vec xx,Vec yy) 478289bc588SBarry Smith { 479c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 48087828ca2SBarry Smith PetscScalar *v = mat->v,*x,*y; 481dfbe8321SBarry Smith PetscErrorCode ierr; 4820805154bSBarry Smith PetscBLASInt m, n,_One=1; 483ea709b57SSatish Balay PetscScalar _DOne=1.0,_DZero=0.0; 4843a40ed3dSBarry Smith 4853a40ed3dSBarry Smith PetscFunctionBegin; 4860805154bSBarry Smith m = PetscBLASIntCast(A->rmap.n); 4870805154bSBarry Smith n = PetscBLASIntCast(A->cmap.n); 488899cda47SBarry Smith if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0); 4891ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 4901ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 49171044d3cSBarry Smith BLASgemv_("T",&m,&n,&_DOne,v,&mat->lda,x,&_One,&_DZero,y,&_One); 4921ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 4931ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 494899cda47SBarry Smith ierr = PetscLogFlops(2*A->rmap.n*A->cmap.n - A->cmap.n);CHKERRQ(ierr); 4953a40ed3dSBarry Smith PetscFunctionReturn(0); 496289bc588SBarry Smith } 4976ee01492SSatish Balay 4984a2ae208SSatish Balay #undef __FUNCT__ 4994a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqDense" 500dfbe8321SBarry Smith PetscErrorCode MatMult_SeqDense(Mat A,Vec xx,Vec yy) 501289bc588SBarry Smith { 502c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 50387828ca2SBarry Smith PetscScalar *v = mat->v,*x,*y,_DOne=1.0,_DZero=0.0; 504dfbe8321SBarry Smith PetscErrorCode ierr; 5050805154bSBarry Smith PetscBLASInt m, n, _One=1; 5063a40ed3dSBarry Smith 5073a40ed3dSBarry Smith PetscFunctionBegin; 5080805154bSBarry Smith m = PetscBLASIntCast(A->rmap.n); 5090805154bSBarry Smith n = PetscBLASIntCast(A->cmap.n); 510899cda47SBarry Smith if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0); 5111ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 5121ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 51371044d3cSBarry Smith BLASgemv_("N",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DZero,y,&_One); 5141ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 5151ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 516899cda47SBarry Smith ierr = PetscLogFlops(2*A->rmap.n*A->cmap.n - A->rmap.n);CHKERRQ(ierr); 5173a40ed3dSBarry Smith PetscFunctionReturn(0); 518289bc588SBarry Smith } 5196ee01492SSatish Balay 5204a2ae208SSatish Balay #undef __FUNCT__ 5214a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqDense" 522dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy) 523289bc588SBarry Smith { 524c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 52587828ca2SBarry Smith PetscScalar *v = mat->v,*x,*y,_DOne=1.0; 526dfbe8321SBarry Smith PetscErrorCode ierr; 5270805154bSBarry Smith PetscBLASInt m, n, _One=1; 5283a40ed3dSBarry Smith 5293a40ed3dSBarry Smith PetscFunctionBegin; 5300805154bSBarry Smith m = PetscBLASIntCast(A->rmap.n); 5310805154bSBarry Smith n = PetscBLASIntCast(A->cmap.n); 532899cda47SBarry Smith if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0); 533600d6d0bSBarry Smith if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);} 5341ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 5351ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 53671044d3cSBarry Smith BLASgemv_("N",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DOne,y,&_One); 5371ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 5381ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 539899cda47SBarry Smith ierr = PetscLogFlops(2*A->rmap.n*A->cmap.n);CHKERRQ(ierr); 5403a40ed3dSBarry Smith PetscFunctionReturn(0); 541289bc588SBarry Smith } 5426ee01492SSatish Balay 5434a2ae208SSatish Balay #undef __FUNCT__ 5444a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_SeqDense" 545dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy) 546289bc588SBarry Smith { 547c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 54887828ca2SBarry Smith PetscScalar *v = mat->v,*x,*y; 549dfbe8321SBarry Smith PetscErrorCode ierr; 5500805154bSBarry Smith PetscBLASInt m, n, _One=1; 55187828ca2SBarry Smith PetscScalar _DOne=1.0; 5523a40ed3dSBarry Smith 5533a40ed3dSBarry Smith PetscFunctionBegin; 5540805154bSBarry Smith m = PetscBLASIntCast(A->rmap.n); 5550805154bSBarry Smith n = PetscBLASIntCast(A->cmap.n); 556899cda47SBarry Smith if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0); 557600d6d0bSBarry Smith if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);} 5581ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 5591ebc52fbSHong Zhang ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 56071044d3cSBarry Smith BLASgemv_("T",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DOne,y,&_One); 5611ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 5621ebc52fbSHong Zhang ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 563899cda47SBarry Smith ierr = PetscLogFlops(2*A->rmap.n*A->cmap.n);CHKERRQ(ierr); 5643a40ed3dSBarry Smith PetscFunctionReturn(0); 565289bc588SBarry Smith } 566289bc588SBarry Smith 567289bc588SBarry Smith /* -----------------------------------------------------------------*/ 5684a2ae208SSatish Balay #undef __FUNCT__ 5694a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqDense" 57013f74950SBarry Smith PetscErrorCode MatGetRow_SeqDense(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **vals) 571289bc588SBarry Smith { 572c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 57387828ca2SBarry Smith PetscScalar *v; 5746849ba73SBarry Smith PetscErrorCode ierr; 57513f74950SBarry Smith PetscInt i; 57667e560aaSBarry Smith 5773a40ed3dSBarry Smith PetscFunctionBegin; 578899cda47SBarry Smith *ncols = A->cmap.n; 579289bc588SBarry Smith if (cols) { 580899cda47SBarry Smith ierr = PetscMalloc((A->cmap.n+1)*sizeof(PetscInt),cols);CHKERRQ(ierr); 581899cda47SBarry Smith for (i=0; i<A->cmap.n; i++) (*cols)[i] = i; 582289bc588SBarry Smith } 583289bc588SBarry Smith if (vals) { 584899cda47SBarry Smith ierr = PetscMalloc((A->cmap.n+1)*sizeof(PetscScalar),vals);CHKERRQ(ierr); 585289bc588SBarry Smith v = mat->v + row; 586899cda47SBarry Smith for (i=0; i<A->cmap.n; i++) {(*vals)[i] = *v; v += mat->lda;} 587289bc588SBarry Smith } 5883a40ed3dSBarry Smith PetscFunctionReturn(0); 589289bc588SBarry Smith } 5906ee01492SSatish Balay 5914a2ae208SSatish Balay #undef __FUNCT__ 5924a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqDense" 59313f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqDense(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **vals) 594289bc588SBarry Smith { 595dfbe8321SBarry Smith PetscErrorCode ierr; 596606d414cSSatish Balay PetscFunctionBegin; 597606d414cSSatish Balay if (cols) {ierr = PetscFree(*cols);CHKERRQ(ierr);} 598606d414cSSatish Balay if (vals) {ierr = PetscFree(*vals);CHKERRQ(ierr); } 5993a40ed3dSBarry Smith PetscFunctionReturn(0); 600289bc588SBarry Smith } 601289bc588SBarry Smith /* ----------------------------------------------------------------*/ 6024a2ae208SSatish Balay #undef __FUNCT__ 6034a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqDense" 60413f74950SBarry Smith PetscErrorCode MatSetValues_SeqDense(Mat A,PetscInt m,const PetscInt indexm[],PetscInt n,const PetscInt indexn[],const PetscScalar v[],InsertMode addv) 605289bc588SBarry Smith { 606c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 60713f74950SBarry Smith PetscInt i,j,idx=0; 608d6dfbf8fSBarry Smith 6093a40ed3dSBarry Smith PetscFunctionBegin; 610289bc588SBarry Smith if (!mat->roworiented) { 611dbb450caSBarry Smith if (addv == INSERT_VALUES) { 612289bc588SBarry Smith for (j=0; j<n; j++) { 613cddbea37SSatish Balay if (indexn[j] < 0) {idx += m; continue;} 6142515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 615899cda47SBarry 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); 61658804f6dSBarry Smith #endif 617289bc588SBarry Smith for (i=0; i<m; i++) { 618cddbea37SSatish Balay if (indexm[i] < 0) {idx++; continue;} 6192515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 620899cda47SBarry 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); 62158804f6dSBarry Smith #endif 622cddbea37SSatish Balay mat->v[indexn[j]*mat->lda + indexm[i]] = v[idx++]; 623289bc588SBarry Smith } 624289bc588SBarry Smith } 6253a40ed3dSBarry Smith } else { 626289bc588SBarry Smith for (j=0; j<n; j++) { 627cddbea37SSatish Balay if (indexn[j] < 0) {idx += m; continue;} 6282515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 629899cda47SBarry 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); 63058804f6dSBarry Smith #endif 631289bc588SBarry Smith for (i=0; i<m; i++) { 632cddbea37SSatish Balay if (indexm[i] < 0) {idx++; continue;} 6332515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 634899cda47SBarry 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); 63558804f6dSBarry Smith #endif 636cddbea37SSatish Balay mat->v[indexn[j]*mat->lda + indexm[i]] += v[idx++]; 637289bc588SBarry Smith } 638289bc588SBarry Smith } 639289bc588SBarry Smith } 6403a40ed3dSBarry Smith } else { 641dbb450caSBarry Smith if (addv == INSERT_VALUES) { 642e8d4e0b9SBarry Smith for (i=0; i<m; i++) { 643cddbea37SSatish Balay if (indexm[i] < 0) { idx += n; continue;} 6442515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 645899cda47SBarry 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); 64658804f6dSBarry Smith #endif 647e8d4e0b9SBarry Smith for (j=0; j<n; j++) { 648cddbea37SSatish Balay if (indexn[j] < 0) { idx++; continue;} 6492515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 650899cda47SBarry 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); 65158804f6dSBarry Smith #endif 652cddbea37SSatish Balay mat->v[indexn[j]*mat->lda + indexm[i]] = v[idx++]; 653e8d4e0b9SBarry Smith } 654e8d4e0b9SBarry Smith } 6553a40ed3dSBarry Smith } else { 656289bc588SBarry Smith for (i=0; i<m; i++) { 657cddbea37SSatish Balay if (indexm[i] < 0) { idx += n; continue;} 6582515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 659899cda47SBarry 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); 66058804f6dSBarry Smith #endif 661289bc588SBarry Smith for (j=0; j<n; j++) { 662cddbea37SSatish Balay if (indexn[j] < 0) { idx++; continue;} 6632515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 664899cda47SBarry 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); 66558804f6dSBarry Smith #endif 666cddbea37SSatish Balay mat->v[indexn[j]*mat->lda + indexm[i]] += v[idx++]; 667289bc588SBarry Smith } 668289bc588SBarry Smith } 669289bc588SBarry Smith } 670e8d4e0b9SBarry Smith } 6713a40ed3dSBarry Smith PetscFunctionReturn(0); 672289bc588SBarry Smith } 673e8d4e0b9SBarry Smith 6744a2ae208SSatish Balay #undef __FUNCT__ 6754a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqDense" 67613f74950SBarry Smith PetscErrorCode MatGetValues_SeqDense(Mat A,PetscInt m,const PetscInt indexm[],PetscInt n,const PetscInt indexn[],PetscScalar v[]) 677ae80bb75SLois Curfman McInnes { 678ae80bb75SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 67913f74950SBarry Smith PetscInt i,j; 680ae80bb75SLois Curfman McInnes 6813a40ed3dSBarry Smith PetscFunctionBegin; 682ae80bb75SLois Curfman McInnes /* row-oriented output */ 683ae80bb75SLois Curfman McInnes for (i=0; i<m; i++) { 68497e567efSBarry Smith if (indexm[i] < 0) {v += n;continue;} 685f12f8aa1SBarry 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); 686ae80bb75SLois Curfman McInnes for (j=0; j<n; j++) { 6876f31f424SBarry Smith if (indexn[j] < 0) {v++; continue;} 6886f31f424SBarry 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); 68997e567efSBarry Smith *v++ = mat->v[indexn[j]*mat->lda + indexm[i]]; 690ae80bb75SLois Curfman McInnes } 691ae80bb75SLois Curfman McInnes } 6923a40ed3dSBarry Smith PetscFunctionReturn(0); 693ae80bb75SLois Curfman McInnes } 694ae80bb75SLois Curfman McInnes 695289bc588SBarry Smith /* -----------------------------------------------------------------*/ 696289bc588SBarry Smith 697e090d566SSatish Balay #include "petscsys.h" 69888e32edaSLois Curfman McInnes 6994a2ae208SSatish Balay #undef __FUNCT__ 7004a2ae208SSatish Balay #define __FUNCT__ "MatLoad_SeqDense" 701f69a0ea3SMatthew Knepley PetscErrorCode MatLoad_SeqDense(PetscViewer viewer, MatType type,Mat *A) 70288e32edaSLois Curfman McInnes { 70388e32edaSLois Curfman McInnes Mat_SeqDense *a; 70488e32edaSLois Curfman McInnes Mat B; 7056849ba73SBarry Smith PetscErrorCode ierr; 70613f74950SBarry Smith PetscInt *scols,i,j,nz,header[4]; 70713f74950SBarry Smith int fd; 70813f74950SBarry Smith PetscMPIInt size; 70913f74950SBarry Smith PetscInt *rowlengths = 0,M,N,*cols; 71087828ca2SBarry Smith PetscScalar *vals,*svals,*v,*w; 71119bcc07fSBarry Smith MPI_Comm comm = ((PetscObject)viewer)->comm; 71288e32edaSLois Curfman McInnes 7133a40ed3dSBarry Smith PetscFunctionBegin; 714d132466eSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 71529bbc08cSBarry Smith if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"view must have one processor"); 716b0a32e0cSBarry Smith ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 7170752156aSBarry Smith ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr); 718552e946dSBarry Smith if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"Not matrix object"); 71988e32edaSLois Curfman McInnes M = header[1]; N = header[2]; nz = header[3]; 72088e32edaSLois Curfman McInnes 72190ace30eSBarry Smith if (nz == MATRIX_BINARY_FORMAT_DENSE) { /* matrix in file is dense */ 722f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 723f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,M,N,M,N);CHKERRQ(ierr); 724be5d1d56SKris Buschelman ierr = MatSetType(*A,type);CHKERRQ(ierr); 7255c5985e7SKris Buschelman ierr = MatSeqDenseSetPreallocation(*A,PETSC_NULL);CHKERRQ(ierr); 72690ace30eSBarry Smith B = *A; 72790ace30eSBarry Smith a = (Mat_SeqDense*)B->data; 7284a41a4d3SSatish Balay v = a->v; 7294a41a4d3SSatish Balay /* Allocate some temp space to read in the values and then flip them 7304a41a4d3SSatish Balay from row major to column major */ 731b72907f3SBarry Smith ierr = PetscMalloc((M*N > 0 ? M*N : 1)*sizeof(PetscScalar),&w);CHKERRQ(ierr); 73290ace30eSBarry Smith /* read in nonzero values */ 7334a41a4d3SSatish Balay ierr = PetscBinaryRead(fd,w,M*N,PETSC_SCALAR);CHKERRQ(ierr); 7344a41a4d3SSatish Balay /* now flip the values and store them in the matrix*/ 7354a41a4d3SSatish Balay for (j=0; j<N; j++) { 7364a41a4d3SSatish Balay for (i=0; i<M; i++) { 7374a41a4d3SSatish Balay *v++ =w[i*N+j]; 7384a41a4d3SSatish Balay } 7394a41a4d3SSatish Balay } 740606d414cSSatish Balay ierr = PetscFree(w);CHKERRQ(ierr); 7416d4a8577SBarry Smith ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 7426d4a8577SBarry Smith ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 74390ace30eSBarry Smith } else { 74488e32edaSLois Curfman McInnes /* read row lengths */ 74513f74950SBarry Smith ierr = PetscMalloc((M+1)*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr); 7460752156aSBarry Smith ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr); 74788e32edaSLois Curfman McInnes 74888e32edaSLois Curfman McInnes /* create our matrix */ 749f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 750f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,M,N,M,N);CHKERRQ(ierr); 751be5d1d56SKris Buschelman ierr = MatSetType(*A,type);CHKERRQ(ierr); 7525c5985e7SKris Buschelman ierr = MatSeqDenseSetPreallocation(*A,PETSC_NULL);CHKERRQ(ierr); 75388e32edaSLois Curfman McInnes B = *A; 75488e32edaSLois Curfman McInnes a = (Mat_SeqDense*)B->data; 75588e32edaSLois Curfman McInnes v = a->v; 75688e32edaSLois Curfman McInnes 75788e32edaSLois Curfman McInnes /* read column indices and nonzeros */ 75813f74950SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscInt),&scols);CHKERRQ(ierr); 759b0a32e0cSBarry Smith cols = scols; 7600752156aSBarry Smith ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr); 76187828ca2SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&svals);CHKERRQ(ierr); 762b0a32e0cSBarry Smith vals = svals; 7630752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr); 76488e32edaSLois Curfman McInnes 76588e32edaSLois Curfman McInnes /* insert into matrix */ 76688e32edaSLois Curfman McInnes for (i=0; i<M; i++) { 76788e32edaSLois Curfman McInnes for (j=0; j<rowlengths[i]; j++) v[i+M*scols[j]] = svals[j]; 76888e32edaSLois Curfman McInnes svals += rowlengths[i]; scols += rowlengths[i]; 76988e32edaSLois Curfman McInnes } 770606d414cSSatish Balay ierr = PetscFree(vals);CHKERRQ(ierr); 771606d414cSSatish Balay ierr = PetscFree(cols);CHKERRQ(ierr); 772606d414cSSatish Balay ierr = PetscFree(rowlengths);CHKERRQ(ierr); 77388e32edaSLois Curfman McInnes 7746d4a8577SBarry Smith ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 7756d4a8577SBarry Smith ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 77690ace30eSBarry Smith } 7773a40ed3dSBarry Smith PetscFunctionReturn(0); 77888e32edaSLois Curfman McInnes } 77988e32edaSLois Curfman McInnes 780e090d566SSatish Balay #include "petscsys.h" 781932b0c3eSLois Curfman McInnes 7824a2ae208SSatish Balay #undef __FUNCT__ 7834a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_ASCII" 7846849ba73SBarry Smith static PetscErrorCode MatView_SeqDense_ASCII(Mat A,PetscViewer viewer) 785289bc588SBarry Smith { 786932b0c3eSLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense*)A->data; 787dfbe8321SBarry Smith PetscErrorCode ierr; 78813f74950SBarry Smith PetscInt i,j; 7892dcb1b2aSMatthew Knepley const char *name; 79087828ca2SBarry Smith PetscScalar *v; 791f3ef73ceSBarry Smith PetscViewerFormat format; 7925f481a85SSatish Balay #if defined(PETSC_USE_COMPLEX) 7935f481a85SSatish Balay PetscTruth allreal = PETSC_TRUE; 7945f481a85SSatish Balay #endif 795932b0c3eSLois Curfman McInnes 7963a40ed3dSBarry Smith PetscFunctionBegin; 797435da068SBarry Smith ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr); 798b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 799456192e2SBarry Smith if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 8003a40ed3dSBarry Smith PetscFunctionReturn(0); /* do nothing for now */ 801fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_COMMON) { 802b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr); 803899cda47SBarry Smith for (i=0; i<A->rmap.n; i++) { 80444cd7ae7SLois Curfman McInnes v = a->v + i; 80577431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr); 806899cda47SBarry Smith for (j=0; j<A->cmap.n; j++) { 807aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 808329f5518SBarry Smith if (PetscRealPart(*v) != 0.0 && PetscImaginaryPart(*v) != 0.0) { 809a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",j,PetscRealPart(*v),PetscImaginaryPart(*v));CHKERRQ(ierr); 810329f5518SBarry Smith } else if (PetscRealPart(*v)) { 811a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",j,PetscRealPart(*v));CHKERRQ(ierr); 8126831982aSBarry Smith } 81380cd9d93SLois Curfman McInnes #else 8146831982aSBarry Smith if (*v) { 815a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",j,*v);CHKERRQ(ierr); 8166831982aSBarry Smith } 81780cd9d93SLois Curfman McInnes #endif 8181b807ce4Svictorle v += a->lda; 81980cd9d93SLois Curfman McInnes } 820b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 82180cd9d93SLois Curfman McInnes } 822b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr); 8233a40ed3dSBarry Smith } else { 824b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr); 825aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 82647989497SBarry Smith /* determine if matrix has all real values */ 82747989497SBarry Smith v = a->v; 828899cda47SBarry Smith for (i=0; i<A->rmap.n*A->cmap.n; i++) { 829ffac6cdbSBarry Smith if (PetscImaginaryPart(v[i])) { allreal = PETSC_FALSE; break ;} 83047989497SBarry Smith } 83147989497SBarry Smith #endif 832fb9695e5SSatish Balay if (format == PETSC_VIEWER_ASCII_MATLAB) { 8333a7fca6bSBarry Smith ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr); 834899cda47SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%% Size = %D %D \n",A->rmap.n,A->cmap.n);CHKERRQ(ierr); 835899cda47SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%s = zeros(%D,%D);\n",name,A->rmap.n,A->cmap.n);CHKERRQ(ierr); 836fb9695e5SSatish Balay ierr = PetscViewerASCIIPrintf(viewer,"%s = [\n",name);CHKERRQ(ierr); 837ffac6cdbSBarry Smith } 838ffac6cdbSBarry Smith 839899cda47SBarry Smith for (i=0; i<A->rmap.n; i++) { 840932b0c3eSLois Curfman McInnes v = a->v + i; 841899cda47SBarry Smith for (j=0; j<A->cmap.n; j++) { 842aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 84347989497SBarry Smith if (allreal) { 844f32d5d43SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%18.16e ",PetscRealPart(*v));CHKERRQ(ierr); 84547989497SBarry Smith } else { 846f32d5d43SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%18.16e + %18.16e i ",PetscRealPart(*v),PetscImaginaryPart(*v));CHKERRQ(ierr); 84747989497SBarry Smith } 848289bc588SBarry Smith #else 849f32d5d43SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"%18.16e ",*v);CHKERRQ(ierr); 850289bc588SBarry Smith #endif 8511b807ce4Svictorle v += a->lda; 852289bc588SBarry Smith } 853b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 854289bc588SBarry Smith } 855fb9695e5SSatish Balay if (format == PETSC_VIEWER_ASCII_MATLAB) { 856b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"];\n");CHKERRQ(ierr); 857ffac6cdbSBarry Smith } 858b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr); 859da3a660dSBarry Smith } 860b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 8613a40ed3dSBarry Smith PetscFunctionReturn(0); 862289bc588SBarry Smith } 863289bc588SBarry Smith 8644a2ae208SSatish Balay #undef __FUNCT__ 8654a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Binary" 8666849ba73SBarry Smith static PetscErrorCode MatView_SeqDense_Binary(Mat A,PetscViewer viewer) 867932b0c3eSLois Curfman McInnes { 868932b0c3eSLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense*)A->data; 8696849ba73SBarry Smith PetscErrorCode ierr; 87013f74950SBarry Smith int fd; 871899cda47SBarry Smith PetscInt ict,j,n = A->cmap.n,m = A->rmap.n,i,*col_lens,nz = m*n; 87287828ca2SBarry Smith PetscScalar *v,*anonz,*vals; 873f3ef73ceSBarry Smith PetscViewerFormat format; 874932b0c3eSLois Curfman McInnes 8753a40ed3dSBarry Smith PetscFunctionBegin; 876b0a32e0cSBarry Smith ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 87790ace30eSBarry Smith 878b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 879fb9695e5SSatish Balay if (format == PETSC_VIEWER_BINARY_NATIVE) { 88090ace30eSBarry Smith /* store the matrix as a dense matrix */ 88113f74950SBarry Smith ierr = PetscMalloc(4*sizeof(PetscInt),&col_lens);CHKERRQ(ierr); 882552e946dSBarry Smith col_lens[0] = MAT_FILE_COOKIE; 88390ace30eSBarry Smith col_lens[1] = m; 88490ace30eSBarry Smith col_lens[2] = n; 88590ace30eSBarry Smith col_lens[3] = MATRIX_BINARY_FORMAT_DENSE; 8866f69ff64SBarry Smith ierr = PetscBinaryWrite(fd,col_lens,4,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr); 887606d414cSSatish Balay ierr = PetscFree(col_lens);CHKERRQ(ierr); 88890ace30eSBarry Smith 88990ace30eSBarry Smith /* write out matrix, by rows */ 89087828ca2SBarry Smith ierr = PetscMalloc((m*n+1)*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 89190ace30eSBarry Smith v = a->v; 89290ace30eSBarry Smith for (j=0; j<n; j++) { 893578230a0SSatish Balay for (i=0; i<m; i++) { 894578230a0SSatish Balay vals[j + i*n] = *v++; 89590ace30eSBarry Smith } 89690ace30eSBarry Smith } 8976f69ff64SBarry Smith ierr = PetscBinaryWrite(fd,vals,n*m,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 898606d414cSSatish Balay ierr = PetscFree(vals);CHKERRQ(ierr); 89990ace30eSBarry Smith } else { 90013f74950SBarry Smith ierr = PetscMalloc((4+nz)*sizeof(PetscInt),&col_lens);CHKERRQ(ierr); 901552e946dSBarry Smith col_lens[0] = MAT_FILE_COOKIE; 902932b0c3eSLois Curfman McInnes col_lens[1] = m; 903932b0c3eSLois Curfman McInnes col_lens[2] = n; 904932b0c3eSLois Curfman McInnes col_lens[3] = nz; 905932b0c3eSLois Curfman McInnes 906932b0c3eSLois Curfman McInnes /* store lengths of each row and write (including header) to file */ 907932b0c3eSLois Curfman McInnes for (i=0; i<m; i++) col_lens[4+i] = n; 9086f69ff64SBarry Smith ierr = PetscBinaryWrite(fd,col_lens,4+m,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr); 909932b0c3eSLois Curfman McInnes 910932b0c3eSLois Curfman McInnes /* Possibly should write in smaller increments, not whole matrix at once? */ 911932b0c3eSLois Curfman McInnes /* store column indices (zero start index) */ 912932b0c3eSLois Curfman McInnes ict = 0; 913932b0c3eSLois Curfman McInnes for (i=0; i<m; i++) { 914932b0c3eSLois Curfman McInnes for (j=0; j<n; j++) col_lens[ict++] = j; 915932b0c3eSLois Curfman McInnes } 9166f69ff64SBarry Smith ierr = PetscBinaryWrite(fd,col_lens,nz,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr); 917606d414cSSatish Balay ierr = PetscFree(col_lens);CHKERRQ(ierr); 918932b0c3eSLois Curfman McInnes 919932b0c3eSLois Curfman McInnes /* store nonzero values */ 92087828ca2SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&anonz);CHKERRQ(ierr); 921932b0c3eSLois Curfman McInnes ict = 0; 922932b0c3eSLois Curfman McInnes for (i=0; i<m; i++) { 923932b0c3eSLois Curfman McInnes v = a->v + i; 924932b0c3eSLois Curfman McInnes for (j=0; j<n; j++) { 9251b807ce4Svictorle anonz[ict++] = *v; v += a->lda; 926932b0c3eSLois Curfman McInnes } 927932b0c3eSLois Curfman McInnes } 9286f69ff64SBarry Smith ierr = PetscBinaryWrite(fd,anonz,nz,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 929606d414cSSatish Balay ierr = PetscFree(anonz);CHKERRQ(ierr); 93090ace30eSBarry Smith } 9313a40ed3dSBarry Smith PetscFunctionReturn(0); 932932b0c3eSLois Curfman McInnes } 933932b0c3eSLois Curfman McInnes 9344a2ae208SSatish Balay #undef __FUNCT__ 9354a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Draw_Zoom" 936dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense_Draw_Zoom(PetscDraw draw,void *Aa) 937f1af5d2fSBarry Smith { 938f1af5d2fSBarry Smith Mat A = (Mat) Aa; 939f1af5d2fSBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 9406849ba73SBarry Smith PetscErrorCode ierr; 941899cda47SBarry Smith PetscInt m = A->rmap.n,n = A->cmap.n,color,i,j; 94287828ca2SBarry Smith PetscScalar *v = a->v; 943b0a32e0cSBarry Smith PetscViewer viewer; 944b0a32e0cSBarry Smith PetscDraw popup; 945329f5518SBarry Smith PetscReal xl,yl,xr,yr,x_l,x_r,y_l,y_r,scale,maxv = 0.0; 946f3ef73ceSBarry Smith PetscViewerFormat format; 947f1af5d2fSBarry Smith 948f1af5d2fSBarry Smith PetscFunctionBegin; 949f1af5d2fSBarry Smith 950f1af5d2fSBarry Smith ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr); 951b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 952b0a32e0cSBarry Smith ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr); 953f1af5d2fSBarry Smith 954f1af5d2fSBarry Smith /* Loop over matrix elements drawing boxes */ 955fb9695e5SSatish Balay if (format != PETSC_VIEWER_DRAW_CONTOUR) { 956f1af5d2fSBarry Smith /* Blue for negative and Red for positive */ 957b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 958f1af5d2fSBarry Smith for(j = 0; j < n; j++) { 959f1af5d2fSBarry Smith x_l = j; 960f1af5d2fSBarry Smith x_r = x_l + 1.0; 961f1af5d2fSBarry Smith for(i = 0; i < m; i++) { 962f1af5d2fSBarry Smith y_l = m - i - 1.0; 963f1af5d2fSBarry Smith y_r = y_l + 1.0; 964f1af5d2fSBarry Smith #if defined(PETSC_USE_COMPLEX) 965329f5518SBarry Smith if (PetscRealPart(v[j*m+i]) > 0.) { 966b0a32e0cSBarry Smith color = PETSC_DRAW_RED; 967329f5518SBarry Smith } else if (PetscRealPart(v[j*m+i]) < 0.) { 968b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 969f1af5d2fSBarry Smith } else { 970f1af5d2fSBarry Smith continue; 971f1af5d2fSBarry Smith } 972f1af5d2fSBarry Smith #else 973f1af5d2fSBarry Smith if (v[j*m+i] > 0.) { 974b0a32e0cSBarry Smith color = PETSC_DRAW_RED; 975f1af5d2fSBarry Smith } else if (v[j*m+i] < 0.) { 976b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 977f1af5d2fSBarry Smith } else { 978f1af5d2fSBarry Smith continue; 979f1af5d2fSBarry Smith } 980f1af5d2fSBarry Smith #endif 981b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 982f1af5d2fSBarry Smith } 983f1af5d2fSBarry Smith } 984f1af5d2fSBarry Smith } else { 985f1af5d2fSBarry Smith /* use contour shading to indicate magnitude of values */ 986f1af5d2fSBarry Smith /* first determine max of all nonzero values */ 987f1af5d2fSBarry Smith for(i = 0; i < m*n; i++) { 988f1af5d2fSBarry Smith if (PetscAbsScalar(v[i]) > maxv) maxv = PetscAbsScalar(v[i]); 989f1af5d2fSBarry Smith } 990b0a32e0cSBarry Smith scale = (245.0 - PETSC_DRAW_BASIC_COLORS)/maxv; 991b0a32e0cSBarry Smith ierr = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr); 992b0a32e0cSBarry Smith if (popup) {ierr = PetscDrawScalePopup(popup,0.0,maxv);CHKERRQ(ierr);} 993f1af5d2fSBarry Smith for(j = 0; j < n; j++) { 994f1af5d2fSBarry Smith x_l = j; 995f1af5d2fSBarry Smith x_r = x_l + 1.0; 996f1af5d2fSBarry Smith for(i = 0; i < m; i++) { 997f1af5d2fSBarry Smith y_l = m - i - 1.0; 998f1af5d2fSBarry Smith y_r = y_l + 1.0; 999b0a32e0cSBarry Smith color = PETSC_DRAW_BASIC_COLORS + (int)(scale*PetscAbsScalar(v[j*m+i])); 1000b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr); 1001f1af5d2fSBarry Smith } 1002f1af5d2fSBarry Smith } 1003f1af5d2fSBarry Smith } 1004f1af5d2fSBarry Smith PetscFunctionReturn(0); 1005f1af5d2fSBarry Smith } 1006f1af5d2fSBarry Smith 10074a2ae208SSatish Balay #undef __FUNCT__ 10084a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Draw" 1009dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense_Draw(Mat A,PetscViewer viewer) 1010f1af5d2fSBarry Smith { 1011b0a32e0cSBarry Smith PetscDraw draw; 1012f1af5d2fSBarry Smith PetscTruth isnull; 1013329f5518SBarry Smith PetscReal xr,yr,xl,yl,h,w; 1014dfbe8321SBarry Smith PetscErrorCode ierr; 1015f1af5d2fSBarry Smith 1016f1af5d2fSBarry Smith PetscFunctionBegin; 1017b0a32e0cSBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 1018b0a32e0cSBarry Smith ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); 1019abc0a331SBarry Smith if (isnull) PetscFunctionReturn(0); 1020f1af5d2fSBarry Smith 1021f1af5d2fSBarry Smith ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr); 1022899cda47SBarry Smith xr = A->cmap.n; yr = A->rmap.n; h = yr/10.0; w = xr/10.0; 1023f1af5d2fSBarry Smith xr += w; yr += h; xl = -w; yl = -h; 1024b0a32e0cSBarry Smith ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr); 1025b0a32e0cSBarry Smith ierr = PetscDrawZoom(draw,MatView_SeqDense_Draw_Zoom,A);CHKERRQ(ierr); 1026f1af5d2fSBarry Smith ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",PETSC_NULL);CHKERRQ(ierr); 1027f1af5d2fSBarry Smith PetscFunctionReturn(0); 1028f1af5d2fSBarry Smith } 1029f1af5d2fSBarry Smith 10304a2ae208SSatish Balay #undef __FUNCT__ 10314a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense" 1032dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense(Mat A,PetscViewer viewer) 1033932b0c3eSLois Curfman McInnes { 1034dfbe8321SBarry Smith PetscErrorCode ierr; 10356805f65bSBarry Smith PetscTruth iascii,isbinary,isdraw; 1036932b0c3eSLois Curfman McInnes 10373a40ed3dSBarry Smith PetscFunctionBegin; 103832077d6dSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 1039fb9695e5SSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_BINARY,&isbinary);CHKERRQ(ierr); 1040fb9695e5SSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr); 10410f5bd95cSBarry Smith 1042c45a1595SBarry Smith if (iascii) { 1043c45a1595SBarry Smith ierr = MatView_SeqDense_ASCII(A,viewer);CHKERRQ(ierr); 10440f5bd95cSBarry Smith } else if (isbinary) { 10453a40ed3dSBarry Smith ierr = MatView_SeqDense_Binary(A,viewer);CHKERRQ(ierr); 1046f1af5d2fSBarry Smith } else if (isdraw) { 1047f1af5d2fSBarry Smith ierr = MatView_SeqDense_Draw(A,viewer);CHKERRQ(ierr); 10485cd90555SBarry Smith } else { 1049958c9bccSBarry Smith SETERRQ1(PETSC_ERR_SUP,"Viewer type %s not supported by dense matrix",((PetscObject)viewer)->type_name); 1050932b0c3eSLois Curfman McInnes } 10513a40ed3dSBarry Smith PetscFunctionReturn(0); 1052932b0c3eSLois Curfman McInnes } 1053289bc588SBarry Smith 10544a2ae208SSatish Balay #undef __FUNCT__ 10554a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqDense" 1056dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqDense(Mat mat) 1057289bc588SBarry Smith { 1058ec8511deSBarry Smith Mat_SeqDense *l = (Mat_SeqDense*)mat->data; 1059dfbe8321SBarry Smith PetscErrorCode ierr; 106090f02eecSBarry Smith 10613a40ed3dSBarry Smith PetscFunctionBegin; 1062aa482453SBarry Smith #if defined(PETSC_USE_LOG) 1063899cda47SBarry Smith PetscLogObjectState((PetscObject)mat,"Rows %D Cols %D",mat->rmap.n,mat->cmap.n); 1064a5a9c739SBarry Smith #endif 106505b42c5fSBarry Smith ierr = PetscFree(l->pivots);CHKERRQ(ierr); 10666857c123SSatish Balay if (!l->user_alloc) {ierr = PetscFree(l->v);CHKERRQ(ierr);} 1067606d414cSSatish Balay ierr = PetscFree(l);CHKERRQ(ierr); 1068dbd8c25aSHong Zhang 1069dbd8c25aSHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)mat,0);CHKERRQ(ierr); 1070901853e0SKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatSeqDenseSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr); 10714ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMult_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr); 10724ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultSymbolic_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr); 10734ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultNumeric_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr); 10743a40ed3dSBarry Smith PetscFunctionReturn(0); 1075289bc588SBarry Smith } 1076289bc588SBarry Smith 10774a2ae208SSatish Balay #undef __FUNCT__ 10784a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqDense" 1079fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqDense(Mat A,MatReuse reuse,Mat *matout) 1080289bc588SBarry Smith { 1081c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 10826849ba73SBarry Smith PetscErrorCode ierr; 108313f74950SBarry Smith PetscInt k,j,m,n,M; 108487828ca2SBarry Smith PetscScalar *v,tmp; 108548b35521SBarry Smith 10863a40ed3dSBarry Smith PetscFunctionBegin; 1087899cda47SBarry Smith v = mat->v; m = A->rmap.n; M = mat->lda; n = A->cmap.n; 1088fc4dec0aSBarry Smith if (*matout == A) { /* in place transpose */ 1089a5ce6ee0Svictorle if (m != n) { 1090634064b4SBarry Smith SETERRQ(PETSC_ERR_SUP,"Can not transpose non-square matrix in place"); 1091d64ed03dSBarry Smith } else { 1092d3e5ee88SLois Curfman McInnes for (j=0; j<m; j++) { 1093289bc588SBarry Smith for (k=0; k<j; k++) { 10941b807ce4Svictorle tmp = v[j + k*M]; 10951b807ce4Svictorle v[j + k*M] = v[k + j*M]; 10961b807ce4Svictorle v[k + j*M] = tmp; 1097289bc588SBarry Smith } 1098289bc588SBarry Smith } 1099d64ed03dSBarry Smith } 11003a40ed3dSBarry Smith } else { /* out-of-place transpose */ 1101d3e5ee88SLois Curfman McInnes Mat tmat; 1102ec8511deSBarry Smith Mat_SeqDense *tmatd; 110387828ca2SBarry Smith PetscScalar *v2; 1104ea709b57SSatish Balay 1105fc4dec0aSBarry Smith if (reuse == MAT_INITIAL_MATRIX) { 11067adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&tmat);CHKERRQ(ierr); 1107899cda47SBarry Smith ierr = MatSetSizes(tmat,A->cmap.n,A->rmap.n,A->cmap.n,A->rmap.n);CHKERRQ(ierr); 11087adad957SLisandro Dalcin ierr = MatSetType(tmat,((PetscObject)A)->type_name);CHKERRQ(ierr); 11095c5985e7SKris Buschelman ierr = MatSeqDenseSetPreallocation(tmat,PETSC_NULL);CHKERRQ(ierr); 1110fc4dec0aSBarry Smith } else { 1111fc4dec0aSBarry Smith tmat = *matout; 1112fc4dec0aSBarry Smith } 1113ec8511deSBarry Smith tmatd = (Mat_SeqDense*)tmat->data; 11140de55854SLois Curfman McInnes v = mat->v; v2 = tmatd->v; 1115d3e5ee88SLois Curfman McInnes for (j=0; j<n; j++) { 11161b807ce4Svictorle for (k=0; k<m; k++) v2[j + k*n] = v[k + j*M]; 1117d3e5ee88SLois Curfman McInnes } 11186d4a8577SBarry Smith ierr = MatAssemblyBegin(tmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 11196d4a8577SBarry Smith ierr = MatAssemblyEnd(tmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1120d3e5ee88SLois Curfman McInnes *matout = tmat; 112148b35521SBarry Smith } 11223a40ed3dSBarry Smith PetscFunctionReturn(0); 1123289bc588SBarry Smith } 1124289bc588SBarry Smith 11254a2ae208SSatish Balay #undef __FUNCT__ 11264a2ae208SSatish Balay #define __FUNCT__ "MatEqual_SeqDense" 1127dfbe8321SBarry Smith PetscErrorCode MatEqual_SeqDense(Mat A1,Mat A2,PetscTruth *flg) 1128289bc588SBarry Smith { 1129c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat1 = (Mat_SeqDense*)A1->data; 1130c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat2 = (Mat_SeqDense*)A2->data; 113113f74950SBarry Smith PetscInt i,j; 113287828ca2SBarry Smith PetscScalar *v1 = mat1->v,*v2 = mat2->v; 11339ea5d5aeSSatish Balay 11343a40ed3dSBarry Smith PetscFunctionBegin; 1135899cda47SBarry Smith if (A1->rmap.n != A2->rmap.n) {*flg = PETSC_FALSE; PetscFunctionReturn(0);} 1136899cda47SBarry Smith if (A1->cmap.n != A2->cmap.n) {*flg = PETSC_FALSE; PetscFunctionReturn(0);} 1137899cda47SBarry Smith for (i=0; i<A1->rmap.n; i++) { 11381b807ce4Svictorle v1 = mat1->v+i; v2 = mat2->v+i; 1139899cda47SBarry Smith for (j=0; j<A1->cmap.n; j++) { 11403a40ed3dSBarry Smith if (*v1 != *v2) {*flg = PETSC_FALSE; PetscFunctionReturn(0);} 11411b807ce4Svictorle v1 += mat1->lda; v2 += mat2->lda; 11421b807ce4Svictorle } 1143289bc588SBarry Smith } 114477c4ece6SBarry Smith *flg = PETSC_TRUE; 11453a40ed3dSBarry Smith PetscFunctionReturn(0); 1146289bc588SBarry Smith } 1147289bc588SBarry Smith 11484a2ae208SSatish Balay #undef __FUNCT__ 11494a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_SeqDense" 1150dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_SeqDense(Mat A,Vec v) 1151289bc588SBarry Smith { 1152c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 1153dfbe8321SBarry Smith PetscErrorCode ierr; 115413f74950SBarry Smith PetscInt i,n,len; 115587828ca2SBarry Smith PetscScalar *x,zero = 0.0; 115644cd7ae7SLois Curfman McInnes 11573a40ed3dSBarry Smith PetscFunctionBegin; 11582dcb1b2aSMatthew Knepley ierr = VecSet(v,zero);CHKERRQ(ierr); 11597a97a34bSBarry Smith ierr = VecGetSize(v,&n);CHKERRQ(ierr); 11601ebc52fbSHong Zhang ierr = VecGetArray(v,&x);CHKERRQ(ierr); 1161899cda47SBarry Smith len = PetscMin(A->rmap.n,A->cmap.n); 1162899cda47SBarry Smith if (n != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming mat and vec"); 116344cd7ae7SLois Curfman McInnes for (i=0; i<len; i++) { 11641b807ce4Svictorle x[i] = mat->v[i*mat->lda + i]; 1165289bc588SBarry Smith } 11661ebc52fbSHong Zhang ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 11673a40ed3dSBarry Smith PetscFunctionReturn(0); 1168289bc588SBarry Smith } 1169289bc588SBarry Smith 11704a2ae208SSatish Balay #undef __FUNCT__ 11714a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_SeqDense" 1172dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_SeqDense(Mat A,Vec ll,Vec rr) 1173289bc588SBarry Smith { 1174c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 117587828ca2SBarry Smith PetscScalar *l,*r,x,*v; 1176dfbe8321SBarry Smith PetscErrorCode ierr; 1177899cda47SBarry Smith PetscInt i,j,m = A->rmap.n,n = A->cmap.n; 117855659b69SBarry Smith 11793a40ed3dSBarry Smith PetscFunctionBegin; 118028988994SBarry Smith if (ll) { 11817a97a34bSBarry Smith ierr = VecGetSize(ll,&m);CHKERRQ(ierr); 11821ebc52fbSHong Zhang ierr = VecGetArray(ll,&l);CHKERRQ(ierr); 1183899cda47SBarry Smith if (m != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Left scaling vec wrong size"); 1184da3a660dSBarry Smith for (i=0; i<m; i++) { 1185da3a660dSBarry Smith x = l[i]; 1186da3a660dSBarry Smith v = mat->v + i; 1187da3a660dSBarry Smith for (j=0; j<n; j++) { (*v) *= x; v+= m;} 1188da3a660dSBarry Smith } 11891ebc52fbSHong Zhang ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr); 1190efee365bSSatish Balay ierr = PetscLogFlops(n*m);CHKERRQ(ierr); 1191da3a660dSBarry Smith } 119228988994SBarry Smith if (rr) { 11937a97a34bSBarry Smith ierr = VecGetSize(rr,&n);CHKERRQ(ierr); 11941ebc52fbSHong Zhang ierr = VecGetArray(rr,&r);CHKERRQ(ierr); 1195899cda47SBarry Smith if (n != A->cmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Right scaling vec wrong size"); 1196da3a660dSBarry Smith for (i=0; i<n; i++) { 1197da3a660dSBarry Smith x = r[i]; 1198da3a660dSBarry Smith v = mat->v + i*m; 1199da3a660dSBarry Smith for (j=0; j<m; j++) { (*v++) *= x;} 1200da3a660dSBarry Smith } 12011ebc52fbSHong Zhang ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr); 1202efee365bSSatish Balay ierr = PetscLogFlops(n*m);CHKERRQ(ierr); 1203da3a660dSBarry Smith } 12043a40ed3dSBarry Smith PetscFunctionReturn(0); 1205289bc588SBarry Smith } 1206289bc588SBarry Smith 12074a2ae208SSatish Balay #undef __FUNCT__ 12084a2ae208SSatish Balay #define __FUNCT__ "MatNorm_SeqDense" 1209dfbe8321SBarry Smith PetscErrorCode MatNorm_SeqDense(Mat A,NormType type,PetscReal *nrm) 1210289bc588SBarry Smith { 1211c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 121287828ca2SBarry Smith PetscScalar *v = mat->v; 1213329f5518SBarry Smith PetscReal sum = 0.0; 1214899cda47SBarry Smith PetscInt lda=mat->lda,m=A->rmap.n,i,j; 1215efee365bSSatish Balay PetscErrorCode ierr; 121655659b69SBarry Smith 12173a40ed3dSBarry Smith PetscFunctionBegin; 1218289bc588SBarry Smith if (type == NORM_FROBENIUS) { 1219a5ce6ee0Svictorle if (lda>m) { 1220899cda47SBarry Smith for (j=0; j<A->cmap.n; j++) { 1221a5ce6ee0Svictorle v = mat->v+j*lda; 1222a5ce6ee0Svictorle for (i=0; i<m; i++) { 1223a5ce6ee0Svictorle #if defined(PETSC_USE_COMPLEX) 1224a5ce6ee0Svictorle sum += PetscRealPart(PetscConj(*v)*(*v)); v++; 1225a5ce6ee0Svictorle #else 1226a5ce6ee0Svictorle sum += (*v)*(*v); v++; 1227a5ce6ee0Svictorle #endif 1228a5ce6ee0Svictorle } 1229a5ce6ee0Svictorle } 1230a5ce6ee0Svictorle } else { 1231899cda47SBarry Smith for (i=0; i<A->cmap.n*A->rmap.n; i++) { 1232aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 1233329f5518SBarry Smith sum += PetscRealPart(PetscConj(*v)*(*v)); v++; 1234289bc588SBarry Smith #else 1235289bc588SBarry Smith sum += (*v)*(*v); v++; 1236289bc588SBarry Smith #endif 1237289bc588SBarry Smith } 1238a5ce6ee0Svictorle } 1239064f8208SBarry Smith *nrm = sqrt(sum); 1240899cda47SBarry Smith ierr = PetscLogFlops(2*A->cmap.n*A->rmap.n);CHKERRQ(ierr); 12413a40ed3dSBarry Smith } else if (type == NORM_1) { 1242064f8208SBarry Smith *nrm = 0.0; 1243899cda47SBarry Smith for (j=0; j<A->cmap.n; j++) { 12441b807ce4Svictorle v = mat->v + j*mat->lda; 1245289bc588SBarry Smith sum = 0.0; 1246899cda47SBarry Smith for (i=0; i<A->rmap.n; i++) { 124733a8263dSBarry Smith sum += PetscAbsScalar(*v); v++; 1248289bc588SBarry Smith } 1249064f8208SBarry Smith if (sum > *nrm) *nrm = sum; 1250289bc588SBarry Smith } 1251899cda47SBarry Smith ierr = PetscLogFlops(A->cmap.n*A->rmap.n);CHKERRQ(ierr); 12523a40ed3dSBarry Smith } else if (type == NORM_INFINITY) { 1253064f8208SBarry Smith *nrm = 0.0; 1254899cda47SBarry Smith for (j=0; j<A->rmap.n; j++) { 1255289bc588SBarry Smith v = mat->v + j; 1256289bc588SBarry Smith sum = 0.0; 1257899cda47SBarry Smith for (i=0; i<A->cmap.n; i++) { 12581b807ce4Svictorle sum += PetscAbsScalar(*v); v += mat->lda; 1259289bc588SBarry Smith } 1260064f8208SBarry Smith if (sum > *nrm) *nrm = sum; 1261289bc588SBarry Smith } 1262899cda47SBarry Smith ierr = PetscLogFlops(A->cmap.n*A->rmap.n);CHKERRQ(ierr); 12633a40ed3dSBarry Smith } else { 126429bbc08cSBarry Smith SETERRQ(PETSC_ERR_SUP,"No two norm"); 1265289bc588SBarry Smith } 12663a40ed3dSBarry Smith PetscFunctionReturn(0); 1267289bc588SBarry Smith } 1268289bc588SBarry Smith 12694a2ae208SSatish Balay #undef __FUNCT__ 12704a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqDense" 12714e0d8c25SBarry Smith PetscErrorCode MatSetOption_SeqDense(Mat A,MatOption op,PetscTruth flg) 1272289bc588SBarry Smith { 1273c0bbcb79SLois Curfman McInnes Mat_SeqDense *aij = (Mat_SeqDense*)A->data; 127463ba0a88SBarry Smith PetscErrorCode ierr; 127567e560aaSBarry Smith 12763a40ed3dSBarry Smith PetscFunctionBegin; 1277b5a2b587SKris Buschelman switch (op) { 1278b5a2b587SKris Buschelman case MAT_ROW_ORIENTED: 12794e0d8c25SBarry Smith aij->roworiented = flg; 1280b5a2b587SKris Buschelman break; 1281512a5fc5SBarry Smith case MAT_NEW_NONZERO_LOCATIONS: 1282b5a2b587SKris Buschelman case MAT_NEW_NONZERO_LOCATION_ERR: 12833971808eSMatthew Knepley case MAT_NEW_NONZERO_ALLOCATION_ERR: 12844e0d8c25SBarry Smith case MAT_NEW_DIAGONALS: 1285b5a2b587SKris Buschelman case MAT_IGNORE_OFF_PROC_ENTRIES: 1286b5a2b587SKris Buschelman case MAT_USE_HASH_TABLE: 128777e54ba9SKris Buschelman case MAT_SYMMETRIC: 128877e54ba9SKris Buschelman case MAT_STRUCTURALLY_SYMMETRIC: 12899a4540c5SBarry Smith case MAT_HERMITIAN: 12909a4540c5SBarry Smith case MAT_SYMMETRY_ETERNAL: 1291600fe468SBarry Smith case MAT_IGNORE_LOWER_TRIANGULAR: 1292290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 129377e54ba9SKris Buschelman break; 1294b5a2b587SKris Buschelman default: 1295600fe468SBarry Smith SETERRQ1(PETSC_ERR_SUP,"unknown option %s",MatOptions[op]); 12963a40ed3dSBarry Smith } 12973a40ed3dSBarry Smith PetscFunctionReturn(0); 1298289bc588SBarry Smith } 1299289bc588SBarry Smith 13004a2ae208SSatish Balay #undef __FUNCT__ 13014a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_SeqDense" 1302dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_SeqDense(Mat A) 13036f0a148fSBarry Smith { 1304ec8511deSBarry Smith Mat_SeqDense *l = (Mat_SeqDense*)A->data; 13056849ba73SBarry Smith PetscErrorCode ierr; 1306899cda47SBarry Smith PetscInt lda=l->lda,m=A->rmap.n,j; 13073a40ed3dSBarry Smith 13083a40ed3dSBarry Smith PetscFunctionBegin; 1309a5ce6ee0Svictorle if (lda>m) { 1310899cda47SBarry Smith for (j=0; j<A->cmap.n; j++) { 1311a5ce6ee0Svictorle ierr = PetscMemzero(l->v+j*lda,m*sizeof(PetscScalar));CHKERRQ(ierr); 1312a5ce6ee0Svictorle } 1313a5ce6ee0Svictorle } else { 1314899cda47SBarry Smith ierr = PetscMemzero(l->v,A->rmap.n*A->cmap.n*sizeof(PetscScalar));CHKERRQ(ierr); 1315a5ce6ee0Svictorle } 13163a40ed3dSBarry Smith PetscFunctionReturn(0); 13176f0a148fSBarry Smith } 13186f0a148fSBarry Smith 13194a2ae208SSatish Balay #undef __FUNCT__ 13204a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqDense" 1321f4df32b1SMatthew Knepley PetscErrorCode MatZeroRows_SeqDense(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag) 13226f0a148fSBarry Smith { 1323ec8511deSBarry Smith Mat_SeqDense *l = (Mat_SeqDense*)A->data; 1324899cda47SBarry Smith PetscInt n = A->cmap.n,i,j; 132587828ca2SBarry Smith PetscScalar *slot; 132655659b69SBarry Smith 13273a40ed3dSBarry Smith PetscFunctionBegin; 13286f0a148fSBarry Smith for (i=0; i<N; i++) { 13296f0a148fSBarry Smith slot = l->v + rows[i]; 13306f0a148fSBarry Smith for (j=0; j<n; j++) { *slot = 0.0; slot += n;} 13316f0a148fSBarry Smith } 1332f4df32b1SMatthew Knepley if (diag != 0.0) { 13336f0a148fSBarry Smith for (i=0; i<N; i++) { 13346f0a148fSBarry Smith slot = l->v + (n+1)*rows[i]; 1335f4df32b1SMatthew Knepley *slot = diag; 13366f0a148fSBarry Smith } 13376f0a148fSBarry Smith } 13383a40ed3dSBarry Smith PetscFunctionReturn(0); 13396f0a148fSBarry Smith } 1340557bce09SLois Curfman McInnes 13414a2ae208SSatish Balay #undef __FUNCT__ 13424a2ae208SSatish Balay #define __FUNCT__ "MatGetArray_SeqDense" 1343dfbe8321SBarry Smith PetscErrorCode MatGetArray_SeqDense(Mat A,PetscScalar *array[]) 134464e87e97SBarry Smith { 1345c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 13463a40ed3dSBarry Smith 13473a40ed3dSBarry Smith PetscFunctionBegin; 1348899cda47SBarry Smith if (mat->lda != A->rmap.n) SETERRQ(PETSC_ERR_SUP,"Cannot get array for Dense matrices with LDA different from number of rows"); 134964e87e97SBarry Smith *array = mat->v; 13503a40ed3dSBarry Smith PetscFunctionReturn(0); 135164e87e97SBarry Smith } 13520754003eSLois Curfman McInnes 13534a2ae208SSatish Balay #undef __FUNCT__ 13544a2ae208SSatish Balay #define __FUNCT__ "MatRestoreArray_SeqDense" 1355dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_SeqDense(Mat A,PetscScalar *array[]) 1356ff14e315SSatish Balay { 13573a40ed3dSBarry Smith PetscFunctionBegin; 135809b544d4SBarry Smith *array = 0; /* user cannot accidently use the array later */ 13593a40ed3dSBarry Smith PetscFunctionReturn(0); 1360ff14e315SSatish Balay } 13610754003eSLois Curfman McInnes 13624a2ae208SSatish Balay #undef __FUNCT__ 13634a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_SeqDense" 136413f74950SBarry Smith static PetscErrorCode MatGetSubMatrix_SeqDense(Mat A,IS isrow,IS iscol,PetscInt cs,MatReuse scall,Mat *B) 13650754003eSLois Curfman McInnes { 1366c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)A->data; 13676849ba73SBarry Smith PetscErrorCode ierr; 136821a2c019SBarry Smith PetscInt i,j,*irow,*icol,nrows,ncols; 136987828ca2SBarry Smith PetscScalar *av,*bv,*v = mat->v; 13700754003eSLois Curfman McInnes Mat newmat; 13710754003eSLois Curfman McInnes 13723a40ed3dSBarry Smith PetscFunctionBegin; 137378b31e54SBarry Smith ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr); 137478b31e54SBarry Smith ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr); 1375e03a110bSBarry Smith ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr); 1376e03a110bSBarry Smith ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr); 13770754003eSLois Curfman McInnes 1378182d2002SSatish Balay /* Check submatrixcall */ 1379182d2002SSatish Balay if (scall == MAT_REUSE_MATRIX) { 138013f74950SBarry Smith PetscInt n_cols,n_rows; 1381182d2002SSatish Balay ierr = MatGetSize(*B,&n_rows,&n_cols);CHKERRQ(ierr); 138221a2c019SBarry Smith if (n_rows != nrows || n_cols != ncols) { 138321a2c019SBarry Smith /* resize the result result matrix to match number of requested rows/columns */ 138421a2c019SBarry Smith ierr = MatSetSizes(*B,nrows,nrows,nrows,nrows);CHKERRQ(ierr); 138521a2c019SBarry Smith } 1386182d2002SSatish Balay newmat = *B; 1387182d2002SSatish Balay } else { 13880754003eSLois Curfman McInnes /* Create and fill new matrix */ 13897adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&newmat);CHKERRQ(ierr); 1390f69a0ea3SMatthew Knepley ierr = MatSetSizes(newmat,nrows,ncols,nrows,ncols);CHKERRQ(ierr); 13917adad957SLisandro Dalcin ierr = MatSetType(newmat,((PetscObject)A)->type_name);CHKERRQ(ierr); 13925c5985e7SKris Buschelman ierr = MatSeqDenseSetPreallocation(newmat,PETSC_NULL);CHKERRQ(ierr); 1393182d2002SSatish Balay } 1394182d2002SSatish Balay 1395182d2002SSatish Balay /* Now extract the data pointers and do the copy,column at a time */ 1396182d2002SSatish Balay bv = ((Mat_SeqDense*)newmat->data)->v; 1397182d2002SSatish Balay 1398182d2002SSatish Balay for (i=0; i<ncols; i++) { 13996de62eeeSBarry Smith av = v + mat->lda*icol[i]; 1400182d2002SSatish Balay for (j=0; j<nrows; j++) { 1401182d2002SSatish Balay *bv++ = av[irow[j]]; 14020754003eSLois Curfman McInnes } 14030754003eSLois Curfman McInnes } 1404182d2002SSatish Balay 1405182d2002SSatish Balay /* Assemble the matrices so that the correct flags are set */ 14066d4a8577SBarry Smith ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 14076d4a8577SBarry Smith ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 14080754003eSLois Curfman McInnes 14090754003eSLois Curfman McInnes /* Free work space */ 141078b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr); 141178b31e54SBarry Smith ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr); 1412182d2002SSatish Balay *B = newmat; 14133a40ed3dSBarry Smith PetscFunctionReturn(0); 14140754003eSLois Curfman McInnes } 14150754003eSLois Curfman McInnes 14164a2ae208SSatish Balay #undef __FUNCT__ 14174a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrices_SeqDense" 141813f74950SBarry Smith PetscErrorCode MatGetSubMatrices_SeqDense(Mat A,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *B[]) 1419905e6a2fSBarry Smith { 14206849ba73SBarry Smith PetscErrorCode ierr; 142113f74950SBarry Smith PetscInt i; 1422905e6a2fSBarry Smith 14233a40ed3dSBarry Smith PetscFunctionBegin; 1424905e6a2fSBarry Smith if (scall == MAT_INITIAL_MATRIX) { 1425b0a32e0cSBarry Smith ierr = PetscMalloc((n+1)*sizeof(Mat),B);CHKERRQ(ierr); 1426905e6a2fSBarry Smith } 1427905e6a2fSBarry Smith 1428905e6a2fSBarry Smith for (i=0; i<n; i++) { 14296a6a5d1dSBarry Smith ierr = MatGetSubMatrix_SeqDense(A,irow[i],icol[i],PETSC_DECIDE,scall,&(*B)[i]);CHKERRQ(ierr); 1430905e6a2fSBarry Smith } 14313a40ed3dSBarry Smith PetscFunctionReturn(0); 1432905e6a2fSBarry Smith } 1433905e6a2fSBarry Smith 14344a2ae208SSatish Balay #undef __FUNCT__ 1435c0aa2d19SHong Zhang #define __FUNCT__ "MatAssemblyBegin_SeqDense" 1436c0aa2d19SHong Zhang PetscErrorCode MatAssemblyBegin_SeqDense(Mat mat,MatAssemblyType mode) 1437c0aa2d19SHong Zhang { 1438c0aa2d19SHong Zhang PetscFunctionBegin; 1439c0aa2d19SHong Zhang PetscFunctionReturn(0); 1440c0aa2d19SHong Zhang } 1441c0aa2d19SHong Zhang 1442c0aa2d19SHong Zhang #undef __FUNCT__ 1443c0aa2d19SHong Zhang #define __FUNCT__ "MatAssemblyEnd_SeqDense" 1444c0aa2d19SHong Zhang PetscErrorCode MatAssemblyEnd_SeqDense(Mat mat,MatAssemblyType mode) 1445c0aa2d19SHong Zhang { 1446c0aa2d19SHong Zhang PetscFunctionBegin; 1447c0aa2d19SHong Zhang PetscFunctionReturn(0); 1448c0aa2d19SHong Zhang } 1449c0aa2d19SHong Zhang 1450c0aa2d19SHong Zhang #undef __FUNCT__ 14514a2ae208SSatish Balay #define __FUNCT__ "MatCopy_SeqDense" 1452dfbe8321SBarry Smith PetscErrorCode MatCopy_SeqDense(Mat A,Mat B,MatStructure str) 14534b0e389bSBarry Smith { 14544b0e389bSBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data,*b = (Mat_SeqDense *)B->data; 14556849ba73SBarry Smith PetscErrorCode ierr; 1456899cda47SBarry Smith PetscInt lda1=a->lda,lda2=b->lda, m=A->rmap.n,n=A->cmap.n, j; 14573a40ed3dSBarry Smith 14583a40ed3dSBarry Smith PetscFunctionBegin; 145933f4a19fSKris Buschelman /* If the two matrices don't have the same copy implementation, they aren't compatible for fast copy. */ 146033f4a19fSKris Buschelman if (A->ops->copy != B->ops->copy) { 1461cb5b572fSBarry Smith ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 14623a40ed3dSBarry Smith PetscFunctionReturn(0); 14633a40ed3dSBarry Smith } 1464899cda47SBarry Smith if (m != B->rmap.n || n != B->cmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"size(B) != size(A)"); 1465a5ce6ee0Svictorle if (lda1>m || lda2>m) { 14660dbb7854Svictorle for (j=0; j<n; j++) { 1467a5ce6ee0Svictorle ierr = PetscMemcpy(b->v+j*lda2,a->v+j*lda1,m*sizeof(PetscScalar));CHKERRQ(ierr); 1468a5ce6ee0Svictorle } 1469a5ce6ee0Svictorle } else { 1470899cda47SBarry Smith ierr = PetscMemcpy(b->v,a->v,A->rmap.n*A->cmap.n*sizeof(PetscScalar));CHKERRQ(ierr); 1471a5ce6ee0Svictorle } 1472273d9f13SBarry Smith PetscFunctionReturn(0); 1473273d9f13SBarry Smith } 1474273d9f13SBarry Smith 14754a2ae208SSatish Balay #undef __FUNCT__ 14764a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_SeqDense" 1477dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_SeqDense(Mat A) 1478273d9f13SBarry Smith { 1479dfbe8321SBarry Smith PetscErrorCode ierr; 1480273d9f13SBarry Smith 1481273d9f13SBarry Smith PetscFunctionBegin; 1482273d9f13SBarry Smith ierr = MatSeqDenseSetPreallocation(A,0);CHKERRQ(ierr); 14833a40ed3dSBarry Smith PetscFunctionReturn(0); 14844b0e389bSBarry Smith } 14854b0e389bSBarry Smith 1486284134d9SBarry Smith #undef __FUNCT__ 1487284134d9SBarry Smith #define __FUNCT__ "MatSetSizes_SeqDense" 1488284134d9SBarry Smith PetscErrorCode MatSetSizes_SeqDense(Mat A,PetscInt m,PetscInt n,PetscInt M,PetscInt N) 1489284134d9SBarry Smith { 1490284134d9SBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 149121a2c019SBarry Smith PetscErrorCode ierr; 1492284134d9SBarry Smith PetscFunctionBegin; 149321a2c019SBarry Smith /* this will not be called before lda, Mmax, and Nmax have been set */ 1494284134d9SBarry Smith m = PetscMax(m,M); 1495284134d9SBarry Smith n = PetscMax(n,N); 149621a2c019SBarry 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); 1497284134d9SBarry 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); 1498899cda47SBarry Smith A->rmap.n = A->rmap.n = m; 1499899cda47SBarry Smith A->cmap.n = A->cmap.N = n; 150021a2c019SBarry Smith if (a->changelda) a->lda = m; 150121a2c019SBarry Smith ierr = PetscMemzero(a->v,m*n*sizeof(PetscScalar));CHKERRQ(ierr); 1502284134d9SBarry Smith PetscFunctionReturn(0); 1503284134d9SBarry Smith } 1504170fe5c8SBarry Smith 1505284134d9SBarry Smith 1506a9fe9ddaSSatish Balay /* ----------------------------------------------------------------*/ 1507a9fe9ddaSSatish Balay #undef __FUNCT__ 1508a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMult_SeqDense_SeqDense" 1509a9fe9ddaSSatish Balay PetscErrorCode MatMatMult_SeqDense_SeqDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 1510a9fe9ddaSSatish Balay { 1511a9fe9ddaSSatish Balay PetscErrorCode ierr; 1512a9fe9ddaSSatish Balay 1513a9fe9ddaSSatish Balay PetscFunctionBegin; 1514a9fe9ddaSSatish Balay if (scall == MAT_INITIAL_MATRIX){ 1515a9fe9ddaSSatish Balay ierr = MatMatMultSymbolic_SeqDense_SeqDense(A,B,fill,C);CHKERRQ(ierr); 1516a9fe9ddaSSatish Balay } 1517a9fe9ddaSSatish Balay ierr = MatMatMultNumeric_SeqDense_SeqDense(A,B,*C);CHKERRQ(ierr); 1518a9fe9ddaSSatish Balay PetscFunctionReturn(0); 1519a9fe9ddaSSatish Balay } 1520a9fe9ddaSSatish Balay 1521a9fe9ddaSSatish Balay #undef __FUNCT__ 1522a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultSymbolic_SeqDense_SeqDense" 1523a9fe9ddaSSatish Balay PetscErrorCode MatMatMultSymbolic_SeqDense_SeqDense(Mat A,Mat B,PetscReal fill,Mat *C) 1524a9fe9ddaSSatish Balay { 1525ee16a9a1SHong Zhang PetscErrorCode ierr; 1526899cda47SBarry Smith PetscInt m=A->rmap.n,n=B->cmap.n; 1527ee16a9a1SHong Zhang Mat Cmat; 1528a9fe9ddaSSatish Balay 1529ee16a9a1SHong Zhang PetscFunctionBegin; 1530899cda47SBarry 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); 1531ee16a9a1SHong Zhang ierr = MatCreate(PETSC_COMM_SELF,&Cmat);CHKERRQ(ierr); 1532ee16a9a1SHong Zhang ierr = MatSetSizes(Cmat,m,n,m,n);CHKERRQ(ierr); 1533ee16a9a1SHong Zhang ierr = MatSetType(Cmat,MATSEQDENSE);CHKERRQ(ierr); 1534ee16a9a1SHong Zhang ierr = MatSeqDenseSetPreallocation(Cmat,PETSC_NULL);CHKERRQ(ierr); 1535ee16a9a1SHong Zhang Cmat->assembled = PETSC_TRUE; 1536ee16a9a1SHong Zhang *C = Cmat; 1537ee16a9a1SHong Zhang PetscFunctionReturn(0); 1538ee16a9a1SHong Zhang } 1539a9fe9ddaSSatish Balay 154098a3b096SSatish Balay #undef __FUNCT__ 1541a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultNumeric_SeqDense_SeqDense" 1542a9fe9ddaSSatish Balay PetscErrorCode MatMatMultNumeric_SeqDense_SeqDense(Mat A,Mat B,Mat C) 1543a9fe9ddaSSatish Balay { 1544a9fe9ddaSSatish Balay Mat_SeqDense *a = (Mat_SeqDense*)A->data; 1545a9fe9ddaSSatish Balay Mat_SeqDense *b = (Mat_SeqDense*)B->data; 1546a9fe9ddaSSatish Balay Mat_SeqDense *c = (Mat_SeqDense*)C->data; 15470805154bSBarry Smith PetscBLASInt m,n,k; 1548a9fe9ddaSSatish Balay PetscScalar _DOne=1.0,_DZero=0.0; 1549a9fe9ddaSSatish Balay 1550a9fe9ddaSSatish Balay PetscFunctionBegin; 15510805154bSBarry Smith m = PetscBLASIntCast(A->rmap.n); 15520805154bSBarry Smith n = PetscBLASIntCast(B->cmap.n); 15530805154bSBarry Smith k = PetscBLASIntCast(A->cmap.n); 1554a9fe9ddaSSatish Balay BLASgemm_("N","N",&m,&n,&k,&_DOne,a->v,&a->lda,b->v,&b->lda,&_DZero,c->v,&c->lda); 1555a9fe9ddaSSatish Balay PetscFunctionReturn(0); 1556a9fe9ddaSSatish Balay } 1557a9fe9ddaSSatish Balay 1558a9fe9ddaSSatish Balay #undef __FUNCT__ 1559a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTranspose_SeqDense_SeqDense" 1560a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTranspose_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 = MatMatMultTransposeSymbolic_SeqDense_SeqDense(A,B,fill,C);CHKERRQ(ierr); 1567a9fe9ddaSSatish Balay } 1568a9fe9ddaSSatish Balay ierr = MatMatMultTransposeNumeric_SeqDense_SeqDense(A,B,*C);CHKERRQ(ierr); 1569a9fe9ddaSSatish Balay PetscFunctionReturn(0); 1570a9fe9ddaSSatish Balay } 1571a9fe9ddaSSatish Balay 1572a9fe9ddaSSatish Balay #undef __FUNCT__ 1573a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTransposeSymbolic_SeqDense_SeqDense" 1574a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTransposeSymbolic_SeqDense_SeqDense(Mat A,Mat B,PetscReal fill,Mat *C) 1575a9fe9ddaSSatish Balay { 1576ee16a9a1SHong Zhang PetscErrorCode ierr; 1577899cda47SBarry Smith PetscInt m=A->cmap.n,n=B->cmap.n; 1578ee16a9a1SHong Zhang Mat Cmat; 1579a9fe9ddaSSatish Balay 1580ee16a9a1SHong Zhang PetscFunctionBegin; 1581899cda47SBarry 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); 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 1591a9fe9ddaSSatish Balay #undef __FUNCT__ 1592a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTransposeNumeric_SeqDense_SeqDense" 1593a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTransposeNumeric_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; 1602*2fbe02b9SBarry Smith m = PetscBLASIntCast(A->cmap.n); 16030805154bSBarry Smith n = PetscBLASIntCast(B->cmap.n); 1604*2fbe02b9SBarry Smith k = PetscBLASIntCast(A->rmap.n); 1605*2fbe02b9SBarry Smith /* 1606*2fbe02b9SBarry Smith Note the m and n arguments below are the number rows and columns of A', not A! 1607*2fbe02b9SBarry Smith */ 1608a9fe9ddaSSatish Balay BLASgemm_("T","N",&m,&n,&k,&_DOne,a->v,&a->lda,b->v,&b->lda,&_DZero,c->v,&c->lda); 1609a9fe9ddaSSatish Balay PetscFunctionReturn(0); 1610a9fe9ddaSSatish Balay } 1611985db425SBarry Smith 1612985db425SBarry Smith #undef __FUNCT__ 1613985db425SBarry Smith #define __FUNCT__ "MatGetRowMax_SeqDense" 1614985db425SBarry Smith PetscErrorCode MatGetRowMax_SeqDense(Mat A,Vec v,PetscInt idx[]) 1615985db425SBarry Smith { 1616985db425SBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 1617985db425SBarry Smith PetscErrorCode ierr; 1618985db425SBarry Smith PetscInt i,j,m = A->rmap.n,n = A->cmap.n,p; 1619985db425SBarry Smith PetscScalar *x; 1620985db425SBarry Smith MatScalar *aa = a->v; 1621985db425SBarry Smith 1622985db425SBarry Smith PetscFunctionBegin; 1623985db425SBarry Smith if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 1624985db425SBarry Smith 1625985db425SBarry Smith ierr = VecSet(v,0.0);CHKERRQ(ierr); 1626985db425SBarry Smith ierr = VecGetArray(v,&x);CHKERRQ(ierr); 1627985db425SBarry Smith ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr); 1628985db425SBarry Smith if (p != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 1629985db425SBarry Smith for (i=0; i<m; i++) { 1630985db425SBarry Smith x[i] = aa[i]; if (idx) idx[i] = 0; 1631985db425SBarry Smith for (j=1; j<n; j++){ 1632985db425SBarry Smith if (PetscRealPart(x[i]) < PetscRealPart(aa[i+m*j])) {x[i] = aa[i + m*j]; if (idx) idx[i] = j;} 1633985db425SBarry Smith } 1634985db425SBarry Smith } 1635985db425SBarry Smith ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 1636985db425SBarry Smith PetscFunctionReturn(0); 1637985db425SBarry Smith } 1638985db425SBarry Smith 1639985db425SBarry Smith #undef __FUNCT__ 1640985db425SBarry Smith #define __FUNCT__ "MatGetRowMaxAbs_SeqDense" 1641985db425SBarry Smith PetscErrorCode MatGetRowMaxAbs_SeqDense(Mat A,Vec v,PetscInt idx[]) 1642985db425SBarry Smith { 1643985db425SBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 1644985db425SBarry Smith PetscErrorCode ierr; 1645985db425SBarry Smith PetscInt i,j,m = A->rmap.n,n = A->cmap.n,p; 1646985db425SBarry Smith PetscScalar *x; 1647985db425SBarry Smith PetscReal atmp; 1648985db425SBarry Smith MatScalar *aa = a->v; 1649985db425SBarry Smith 1650985db425SBarry Smith PetscFunctionBegin; 1651985db425SBarry Smith if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 1652985db425SBarry Smith 1653985db425SBarry Smith ierr = VecSet(v,0.0);CHKERRQ(ierr); 1654985db425SBarry Smith ierr = VecGetArray(v,&x);CHKERRQ(ierr); 1655985db425SBarry Smith ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr); 1656985db425SBarry Smith if (p != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 1657985db425SBarry Smith for (i=0; i<m; i++) { 1658985db425SBarry Smith x[i] = PetscAbsScalar(aa[i]); if (idx) idx[i] = 0; 1659985db425SBarry Smith for (j=1; j<n; j++){ 1660985db425SBarry Smith atmp = PetscAbsScalar(aa[i+m*j]); 1661985db425SBarry Smith if (PetscAbsScalar(x[i]) < atmp) {x[i] = atmp; if (idx) idx[i] = j;} 1662985db425SBarry Smith } 1663985db425SBarry Smith } 1664985db425SBarry Smith ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 1665985db425SBarry Smith PetscFunctionReturn(0); 1666985db425SBarry Smith } 1667985db425SBarry Smith 1668985db425SBarry Smith #undef __FUNCT__ 1669985db425SBarry Smith #define __FUNCT__ "MatGetRowMin_SeqDense" 1670985db425SBarry Smith PetscErrorCode MatGetRowMin_SeqDense(Mat A,Vec v,PetscInt idx[]) 1671985db425SBarry Smith { 1672985db425SBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 1673985db425SBarry Smith PetscErrorCode ierr; 1674985db425SBarry Smith PetscInt i,j,m = A->rmap.n,n = A->cmap.n,p; 1675985db425SBarry Smith PetscScalar *x; 1676985db425SBarry Smith MatScalar *aa = a->v; 1677985db425SBarry Smith 1678985db425SBarry Smith PetscFunctionBegin; 1679985db425SBarry Smith if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 1680985db425SBarry Smith 1681985db425SBarry Smith ierr = VecSet(v,0.0);CHKERRQ(ierr); 1682985db425SBarry Smith ierr = VecGetArray(v,&x);CHKERRQ(ierr); 1683985db425SBarry Smith ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr); 1684985db425SBarry Smith if (p != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector"); 1685985db425SBarry Smith for (i=0; i<m; i++) { 1686985db425SBarry Smith x[i] = aa[i]; if (idx) idx[i] = 0; 1687985db425SBarry Smith for (j=1; j<n; j++){ 1688985db425SBarry Smith if (PetscRealPart(x[i]) > PetscRealPart(aa[i+m*j])) {x[i] = aa[i + m*j]; if (idx) idx[i] = j;} 1689985db425SBarry Smith } 1690985db425SBarry Smith } 1691985db425SBarry Smith ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 1692985db425SBarry Smith PetscFunctionReturn(0); 1693985db425SBarry Smith } 1694985db425SBarry Smith 16958d0534beSBarry Smith #undef __FUNCT__ 16968d0534beSBarry Smith #define __FUNCT__ "MatGetColumnVector_SeqDense" 16978d0534beSBarry Smith PetscErrorCode MatGetColumnVector_SeqDense(Mat A,Vec v,PetscInt col) 16988d0534beSBarry Smith { 16998d0534beSBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)A->data; 17008d0534beSBarry Smith PetscErrorCode ierr; 17018d0534beSBarry Smith PetscScalar *x; 17028d0534beSBarry Smith 17038d0534beSBarry Smith PetscFunctionBegin; 17048d0534beSBarry Smith if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix"); 17058d0534beSBarry Smith 17068d0534beSBarry Smith ierr = VecGetArray(v,&x);CHKERRQ(ierr); 17078d0534beSBarry Smith ierr = PetscMemcpy(x,a->v+col*a->lda,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr); 17088d0534beSBarry Smith ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 17098d0534beSBarry Smith PetscFunctionReturn(0); 17108d0534beSBarry Smith } 17118d0534beSBarry Smith 1712289bc588SBarry Smith /* -------------------------------------------------------------------*/ 1713a5ae1ecdSBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_SeqDense, 1714905e6a2fSBarry Smith MatGetRow_SeqDense, 1715905e6a2fSBarry Smith MatRestoreRow_SeqDense, 1716905e6a2fSBarry Smith MatMult_SeqDense, 171797304618SKris Buschelman /* 4*/ MatMultAdd_SeqDense, 17187c922b88SBarry Smith MatMultTranspose_SeqDense, 17197c922b88SBarry Smith MatMultTransposeAdd_SeqDense, 1720905e6a2fSBarry Smith MatSolve_SeqDense, 1721905e6a2fSBarry Smith MatSolveAdd_SeqDense, 17227c922b88SBarry Smith MatSolveTranspose_SeqDense, 172397304618SKris Buschelman /*10*/ MatSolveTransposeAdd_SeqDense, 1724905e6a2fSBarry Smith MatLUFactor_SeqDense, 1725905e6a2fSBarry Smith MatCholeskyFactor_SeqDense, 1726ec8511deSBarry Smith MatRelax_SeqDense, 1727ec8511deSBarry Smith MatTranspose_SeqDense, 172897304618SKris Buschelman /*15*/ MatGetInfo_SeqDense, 1729905e6a2fSBarry Smith MatEqual_SeqDense, 1730905e6a2fSBarry Smith MatGetDiagonal_SeqDense, 1731905e6a2fSBarry Smith MatDiagonalScale_SeqDense, 1732905e6a2fSBarry Smith MatNorm_SeqDense, 1733c0aa2d19SHong Zhang /*20*/ MatAssemblyBegin_SeqDense, 1734c0aa2d19SHong Zhang MatAssemblyEnd_SeqDense, 1735905e6a2fSBarry Smith 0, 1736905e6a2fSBarry Smith MatSetOption_SeqDense, 1737905e6a2fSBarry Smith MatZeroEntries_SeqDense, 173897304618SKris Buschelman /*25*/ MatZeroRows_SeqDense, 1739905e6a2fSBarry Smith MatLUFactorSymbolic_SeqDense, 1740905e6a2fSBarry Smith MatLUFactorNumeric_SeqDense, 1741905e6a2fSBarry Smith MatCholeskyFactorSymbolic_SeqDense, 1742905e6a2fSBarry Smith MatCholeskyFactorNumeric_SeqDense, 174397304618SKris Buschelman /*30*/ MatSetUpPreallocation_SeqDense, 1744273d9f13SBarry Smith 0, 1745905e6a2fSBarry Smith 0, 1746905e6a2fSBarry Smith MatGetArray_SeqDense, 1747905e6a2fSBarry Smith MatRestoreArray_SeqDense, 174897304618SKris Buschelman /*35*/ MatDuplicate_SeqDense, 1749a5ae1ecdSBarry Smith 0, 1750a5ae1ecdSBarry Smith 0, 1751a5ae1ecdSBarry Smith 0, 1752a5ae1ecdSBarry Smith 0, 175397304618SKris Buschelman /*40*/ MatAXPY_SeqDense, 1754a5ae1ecdSBarry Smith MatGetSubMatrices_SeqDense, 1755a5ae1ecdSBarry Smith 0, 17564b0e389bSBarry Smith MatGetValues_SeqDense, 1757a5ae1ecdSBarry Smith MatCopy_SeqDense, 1758985db425SBarry Smith /*45*/ MatGetRowMax_SeqDense, 1759a5ae1ecdSBarry Smith MatScale_SeqDense, 1760a5ae1ecdSBarry Smith 0, 1761a5ae1ecdSBarry Smith 0, 1762a5ae1ecdSBarry Smith 0, 1763521d7252SBarry Smith /*50*/ 0, 1764a5ae1ecdSBarry Smith 0, 1765a5ae1ecdSBarry Smith 0, 1766a5ae1ecdSBarry Smith 0, 1767a5ae1ecdSBarry Smith 0, 176897304618SKris Buschelman /*55*/ 0, 1769a5ae1ecdSBarry Smith 0, 1770a5ae1ecdSBarry Smith 0, 1771a5ae1ecdSBarry Smith 0, 1772a5ae1ecdSBarry Smith 0, 177397304618SKris Buschelman /*60*/ 0, 1774e03a110bSBarry Smith MatDestroy_SeqDense, 1775e03a110bSBarry Smith MatView_SeqDense, 1776357abbc8SBarry Smith 0, 177797304618SKris Buschelman 0, 177897304618SKris Buschelman /*65*/ 0, 177997304618SKris Buschelman 0, 178097304618SKris Buschelman 0, 178197304618SKris Buschelman 0, 178297304618SKris Buschelman 0, 1783985db425SBarry Smith /*70*/ MatGetRowMaxAbs_SeqDense, 178497304618SKris Buschelman 0, 178597304618SKris Buschelman 0, 178697304618SKris Buschelman 0, 178797304618SKris Buschelman 0, 178897304618SKris Buschelman /*75*/ 0, 178997304618SKris Buschelman 0, 179097304618SKris Buschelman 0, 179197304618SKris Buschelman 0, 179297304618SKris Buschelman 0, 179397304618SKris Buschelman /*80*/ 0, 179497304618SKris Buschelman 0, 179597304618SKris Buschelman 0, 179697304618SKris Buschelman 0, 1797865e5f61SKris Buschelman /*84*/ MatLoad_SeqDense, 1798865e5f61SKris Buschelman 0, 17991cbb95d3SBarry Smith MatIsHermitian_SeqDense, 1800865e5f61SKris Buschelman 0, 1801865e5f61SKris Buschelman 0, 1802865e5f61SKris Buschelman 0, 1803a9fe9ddaSSatish Balay /*90*/ MatMatMult_SeqDense_SeqDense, 1804a9fe9ddaSSatish Balay MatMatMultSymbolic_SeqDense_SeqDense, 1805a9fe9ddaSSatish Balay MatMatMultNumeric_SeqDense_SeqDense, 1806865e5f61SKris Buschelman 0, 1807865e5f61SKris Buschelman 0, 1808865e5f61SKris Buschelman /*95*/ 0, 1809a9fe9ddaSSatish Balay MatMatMultTranspose_SeqDense_SeqDense, 1810a9fe9ddaSSatish Balay MatMatMultTransposeSymbolic_SeqDense_SeqDense, 1811a9fe9ddaSSatish Balay MatMatMultTransposeNumeric_SeqDense_SeqDense, 1812284134d9SBarry Smith 0, 1813284134d9SBarry Smith /*100*/0, 1814284134d9SBarry Smith 0, 1815284134d9SBarry Smith 0, 1816284134d9SBarry Smith 0, 1817985db425SBarry Smith MatSetSizes_SeqDense, 1818985db425SBarry Smith 0, 1819985db425SBarry Smith 0, 1820985db425SBarry Smith 0, 1821985db425SBarry Smith 0, 1822985db425SBarry Smith 0, 1823985db425SBarry Smith /*110*/0, 1824985db425SBarry Smith 0, 18258d0534beSBarry Smith MatGetRowMin_SeqDense, 18268d0534beSBarry Smith MatGetColumnVector_SeqDense 1827985db425SBarry Smith }; 182890ace30eSBarry Smith 18294a2ae208SSatish Balay #undef __FUNCT__ 18304a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqDense" 18314b828684SBarry Smith /*@C 1832fafbff53SBarry Smith MatCreateSeqDense - Creates a sequential dense matrix that 1833d65003e9SLois Curfman McInnes is stored in column major order (the usual Fortran 77 manner). Many 1834d65003e9SLois Curfman McInnes of the matrix operations use the BLAS and LAPACK routines. 1835289bc588SBarry Smith 1836db81eaa0SLois Curfman McInnes Collective on MPI_Comm 1837db81eaa0SLois Curfman McInnes 183820563c6bSBarry Smith Input Parameters: 1839db81eaa0SLois Curfman McInnes + comm - MPI communicator, set to PETSC_COMM_SELF 18400c775827SLois Curfman McInnes . m - number of rows 184118f449edSLois Curfman McInnes . n - number of columns 1842db81eaa0SLois Curfman McInnes - data - optional location of matrix data. Set data=PETSC_NULL for PETSc 1843dfc5480cSLois Curfman McInnes to control all matrix memory allocation. 184420563c6bSBarry Smith 184520563c6bSBarry Smith Output Parameter: 184644cd7ae7SLois Curfman McInnes . A - the matrix 184720563c6bSBarry Smith 1848b259b22eSLois Curfman McInnes Notes: 184918f449edSLois Curfman McInnes The data input variable is intended primarily for Fortran programmers 185018f449edSLois Curfman McInnes who wish to allocate their own matrix memory space. Most users should 1851b4fd4287SBarry Smith set data=PETSC_NULL. 185218f449edSLois Curfman McInnes 1853027ccd11SLois Curfman McInnes Level: intermediate 1854027ccd11SLois Curfman McInnes 1855dbd7a890SLois Curfman McInnes .keywords: dense, matrix, LAPACK, BLAS 1856d65003e9SLois Curfman McInnes 1857db81eaa0SLois Curfman McInnes .seealso: MatCreate(), MatCreateMPIDense(), MatSetValues() 185820563c6bSBarry Smith @*/ 1859be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateSeqDense(MPI_Comm comm,PetscInt m,PetscInt n,PetscScalar *data,Mat *A) 1860289bc588SBarry Smith { 1861dfbe8321SBarry Smith PetscErrorCode ierr; 18623b2fbd54SBarry Smith 18633a40ed3dSBarry Smith PetscFunctionBegin; 1864f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 1865f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr); 1866273d9f13SBarry Smith ierr = MatSetType(*A,MATSEQDENSE);CHKERRQ(ierr); 1867273d9f13SBarry Smith ierr = MatSeqDenseSetPreallocation(*A,data);CHKERRQ(ierr); 1868273d9f13SBarry Smith PetscFunctionReturn(0); 1869273d9f13SBarry Smith } 1870273d9f13SBarry Smith 18714a2ae208SSatish Balay #undef __FUNCT__ 1872afc30d2aSLisandro Dalcin #define __FUNCT__ "MatSeqDenseSetPreallocation" 1873273d9f13SBarry Smith /*@C 1874273d9f13SBarry Smith MatSeqDenseSetPreallocation - Sets the array used for storing the matrix elements 1875273d9f13SBarry Smith 1876273d9f13SBarry Smith Collective on MPI_Comm 1877273d9f13SBarry Smith 1878273d9f13SBarry Smith Input Parameters: 1879273d9f13SBarry Smith + A - the matrix 1880273d9f13SBarry Smith - data - the array (or PETSC_NULL) 1881273d9f13SBarry Smith 1882273d9f13SBarry Smith Notes: 1883273d9f13SBarry Smith The data input variable is intended primarily for Fortran programmers 1884273d9f13SBarry Smith who wish to allocate their own matrix memory space. Most users should 1885284134d9SBarry Smith need not call this routine. 1886273d9f13SBarry Smith 1887273d9f13SBarry Smith Level: intermediate 1888273d9f13SBarry Smith 1889273d9f13SBarry Smith .keywords: dense, matrix, LAPACK, BLAS 1890273d9f13SBarry Smith 1891273d9f13SBarry Smith .seealso: MatCreate(), MatCreateMPIDense(), MatSetValues() 1892273d9f13SBarry Smith @*/ 1893be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetPreallocation(Mat B,PetscScalar data[]) 1894273d9f13SBarry Smith { 1895dfbe8321SBarry Smith PetscErrorCode ierr,(*f)(Mat,PetscScalar[]); 1896a23d5eceSKris Buschelman 1897a23d5eceSKris Buschelman PetscFunctionBegin; 1898a23d5eceSKris Buschelman ierr = PetscObjectQueryFunction((PetscObject)B,"MatSeqDenseSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr); 1899a23d5eceSKris Buschelman if (f) { 1900a23d5eceSKris Buschelman ierr = (*f)(B,data);CHKERRQ(ierr); 1901a23d5eceSKris Buschelman } 1902a23d5eceSKris Buschelman PetscFunctionReturn(0); 1903a23d5eceSKris Buschelman } 1904a23d5eceSKris Buschelman 1905a23d5eceSKris Buschelman EXTERN_C_BEGIN 1906a23d5eceSKris Buschelman #undef __FUNCT__ 1907afc30d2aSLisandro Dalcin #define __FUNCT__ "MatSeqDenseSetPreallocation_SeqDense" 1908be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetPreallocation_SeqDense(Mat B,PetscScalar *data) 1909a23d5eceSKris Buschelman { 1910273d9f13SBarry Smith Mat_SeqDense *b; 1911dfbe8321SBarry Smith PetscErrorCode ierr; 1912273d9f13SBarry Smith 1913273d9f13SBarry Smith PetscFunctionBegin; 1914273d9f13SBarry Smith B->preallocated = PETSC_TRUE; 1915273d9f13SBarry Smith b = (Mat_SeqDense*)B->data; 19169e8f95c4SLisandro Dalcin if (!data) { /* petsc-allocated storage */ 19179e8f95c4SLisandro Dalcin if (!b->user_alloc) { ierr = PetscFree(b->v);CHKERRQ(ierr); } 1918284134d9SBarry Smith ierr = PetscMalloc((b->lda*b->Nmax+1)*sizeof(PetscScalar),&b->v);CHKERRQ(ierr); 1919284134d9SBarry Smith ierr = PetscMemzero(b->v,b->lda*b->Nmax*sizeof(PetscScalar));CHKERRQ(ierr); 1920284134d9SBarry Smith ierr = PetscLogObjectMemory(B,b->lda*b->Nmax*sizeof(PetscScalar));CHKERRQ(ierr); 19219e8f95c4SLisandro Dalcin b->user_alloc = PETSC_FALSE; 1922273d9f13SBarry Smith } else { /* user-allocated storage */ 19239e8f95c4SLisandro Dalcin if (!b->user_alloc) { ierr = PetscFree(b->v);CHKERRQ(ierr); } 1924273d9f13SBarry Smith b->v = data; 1925273d9f13SBarry Smith b->user_alloc = PETSC_TRUE; 1926273d9f13SBarry Smith } 1927273d9f13SBarry Smith PetscFunctionReturn(0); 1928273d9f13SBarry Smith } 1929a23d5eceSKris Buschelman EXTERN_C_END 1930273d9f13SBarry Smith 19311b807ce4Svictorle #undef __FUNCT__ 19321b807ce4Svictorle #define __FUNCT__ "MatSeqDenseSetLDA" 19331b807ce4Svictorle /*@C 19341b807ce4Svictorle MatSeqDenseSetLDA - Declare the leading dimension of the user-provided array 19351b807ce4Svictorle 19361b807ce4Svictorle Input parameter: 19371b807ce4Svictorle + A - the matrix 19381b807ce4Svictorle - lda - the leading dimension 19391b807ce4Svictorle 19401b807ce4Svictorle Notes: 19411b807ce4Svictorle This routine is to be used in conjunction with MatSeqDenseSetPreallocation; 19421b807ce4Svictorle it asserts that the preallocation has a leading dimension (the LDA parameter 19431b807ce4Svictorle of Blas and Lapack fame) larger than M, the first dimension of the matrix. 19441b807ce4Svictorle 19451b807ce4Svictorle Level: intermediate 19461b807ce4Svictorle 19471b807ce4Svictorle .keywords: dense, matrix, LAPACK, BLAS 19481b807ce4Svictorle 1949284134d9SBarry Smith .seealso: MatCreate(), MatCreateSeqDense(), MatSeqDenseSetPreallocation(), MatSetMaximumSize() 19501b807ce4Svictorle @*/ 1951be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetLDA(Mat B,PetscInt lda) 19521b807ce4Svictorle { 19531b807ce4Svictorle Mat_SeqDense *b = (Mat_SeqDense*)B->data; 195421a2c019SBarry Smith 19551b807ce4Svictorle PetscFunctionBegin; 1956899cda47SBarry Smith if (lda < B->rmap.n) SETERRQ2(PETSC_ERR_ARG_SIZ,"LDA %D must be at least matrix dimension %D",lda,B->rmap.n); 19571b807ce4Svictorle b->lda = lda; 195821a2c019SBarry Smith b->changelda = PETSC_FALSE; 195921a2c019SBarry Smith b->Mmax = PetscMax(b->Mmax,lda); 19601b807ce4Svictorle PetscFunctionReturn(0); 19611b807ce4Svictorle } 19621b807ce4Svictorle 19630bad9183SKris Buschelman /*MC 1964fafad747SKris Buschelman MATSEQDENSE - MATSEQDENSE = "seqdense" - A matrix type to be used for sequential dense matrices. 19650bad9183SKris Buschelman 19660bad9183SKris Buschelman Options Database Keys: 19670bad9183SKris Buschelman . -mat_type seqdense - sets the matrix type to "seqdense" during a call to MatSetFromOptions() 19680bad9183SKris Buschelman 19690bad9183SKris Buschelman Level: beginner 19700bad9183SKris Buschelman 197189665df3SBarry Smith .seealso: MatCreateSeqDense() 197289665df3SBarry Smith 19730bad9183SKris Buschelman M*/ 19740bad9183SKris Buschelman 1975273d9f13SBarry Smith EXTERN_C_BEGIN 19764a2ae208SSatish Balay #undef __FUNCT__ 19774a2ae208SSatish Balay #define __FUNCT__ "MatCreate_SeqDense" 1978be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_SeqDense(Mat B) 1979273d9f13SBarry Smith { 1980273d9f13SBarry Smith Mat_SeqDense *b; 1981dfbe8321SBarry Smith PetscErrorCode ierr; 19827c334f02SBarry Smith PetscMPIInt size; 1983273d9f13SBarry Smith 1984273d9f13SBarry Smith PetscFunctionBegin; 19857adad957SLisandro Dalcin ierr = MPI_Comm_size(((PetscObject)B)->comm,&size);CHKERRQ(ierr); 198629bbc08cSBarry Smith if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"Comm must be of size 1"); 198755659b69SBarry Smith 1988899cda47SBarry Smith B->rmap.bs = B->cmap.bs = 1; 19896148ca0dSBarry Smith ierr = PetscMapSetUp(&B->rmap);CHKERRQ(ierr); 19906148ca0dSBarry Smith ierr = PetscMapSetUp(&B->cmap);CHKERRQ(ierr); 1991273d9f13SBarry Smith 199238f2d2fdSLisandro Dalcin ierr = PetscNewLog(B,Mat_SeqDense,&b);CHKERRQ(ierr); 1993549d3d68SSatish Balay ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 199444cd7ae7SLois Curfman McInnes B->factor = 0; 199590f02eecSBarry Smith B->mapping = 0; 199644cd7ae7SLois Curfman McInnes B->data = (void*)b; 199718f449edSLois Curfman McInnes 1998a5ae1ecdSBarry Smith 199944cd7ae7SLois Curfman McInnes b->pivots = 0; 2000273d9f13SBarry Smith b->roworiented = PETSC_TRUE; 2001273d9f13SBarry Smith b->v = 0; 2002899cda47SBarry Smith b->lda = B->rmap.n; 200321a2c019SBarry Smith b->changelda = PETSC_FALSE; 2004899cda47SBarry Smith b->Mmax = B->rmap.n; 2005899cda47SBarry Smith b->Nmax = B->cmap.n; 20064e220ebcSLois Curfman McInnes 2007a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqDenseSetPreallocation_C", 2008a23d5eceSKris Buschelman "MatSeqDenseSetPreallocation_SeqDense", 2009a23d5eceSKris Buschelman MatSeqDenseSetPreallocation_SeqDense);CHKERRQ(ierr); 20104ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMult_seqaij_seqdense_C", 20114ae313f4SHong Zhang "MatMatMult_SeqAIJ_SeqDense", 20124ae313f4SHong Zhang MatMatMult_SeqAIJ_SeqDense);CHKERRQ(ierr); 20134ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMultSymbolic_seqaij_seqdense_C", 20144ae313f4SHong Zhang "MatMatMultSymbolic_SeqAIJ_SeqDense", 20154ae313f4SHong Zhang MatMatMultSymbolic_SeqAIJ_SeqDense);CHKERRQ(ierr); 20164ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMultNumeric_seqaij_seqdense_C", 20174ae313f4SHong Zhang "MatMatMultNumeric_SeqAIJ_SeqDense", 20184ae313f4SHong Zhang MatMatMultNumeric_SeqAIJ_SeqDense);CHKERRQ(ierr); 201917667f90SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQDENSE);CHKERRQ(ierr); 20203a40ed3dSBarry Smith PetscFunctionReturn(0); 2021289bc588SBarry Smith } 2022c0aa2d19SHong Zhang 2023c0aa2d19SHong Zhang 2024273d9f13SBarry Smith EXTERN_C_END 2025