163c07aadSStefano Zampini 263c07aadSStefano Zampini /* 363c07aadSStefano Zampini Creates hypre ijmatrix from PETSc matrix 463c07aadSStefano Zampini */ 5225daaf8SStefano Zampini 6c6698e78SStefano Zampini #include <petscpkg_version.h> 739accc25SStefano Zampini #include <petsc/private/petschypre.h> 8dd9c0a25Sstefano_zampini #include <petscmathypre.h> 963c07aadSStefano Zampini #include <petsc/private/matimpl.h> 1063c07aadSStefano Zampini #include <../src/mat/impls/hypre/mhypre.h> 1163c07aadSStefano Zampini #include <../src/mat/impls/aij/mpi/mpiaij.h> 1258968eb6SStefano Zampini #include <../src/vec/vec/impls/hypre/vhyp.h> 1358968eb6SStefano Zampini #include <HYPRE.h> 14c1a070e6SStefano Zampini #include <HYPRE_utilities.h> 15cd8bc7baSStefano Zampini #include <_hypre_parcsr_ls.h> 1668ec7858SStefano Zampini #include <_hypre_sstruct_ls.h> 1763c07aadSStefano Zampini 180e6427aaSSatish Balay #if PETSC_PKG_HYPRE_VERSION_LT(2,18,0) 190e6427aaSSatish Balay #define hypre_ParCSRMatrixClone(A,B) hypre_ParCSRMatrixCompleteClone(A) 200e6427aaSSatish Balay #endif 210e6427aaSSatish Balay 2263c07aadSStefano Zampini static PetscErrorCode MatHYPRE_CreateFromMat(Mat,Mat_HYPRE*); 2363c07aadSStefano Zampini static PetscErrorCode MatHYPRE_IJMatrixPreallocate(Mat,Mat,HYPRE_IJMatrix); 2463c07aadSStefano Zampini static PetscErrorCode MatHYPRE_IJMatrixFastCopy_MPIAIJ(Mat,HYPRE_IJMatrix); 2563c07aadSStefano Zampini static PetscErrorCode MatHYPRE_IJMatrixFastCopy_SeqAIJ(Mat,HYPRE_IJMatrix); 2639accc25SStefano Zampini static PetscErrorCode MatHYPRE_MultKernel_Private(Mat,HYPRE_Complex,Vec,HYPRE_Complex,Vec,PetscBool); 27225daaf8SStefano Zampini static PetscErrorCode hypre_array_destroy(void*); 28c69f721fSFande Kong PetscErrorCode MatSetValues_HYPRE(Mat, PetscInt,const PetscInt[],PetscInt,const PetscInt[],const PetscScalar[],InsertMode ins); 2963c07aadSStefano Zampini 3063c07aadSStefano Zampini static PetscErrorCode MatHYPRE_IJMatrixPreallocate(Mat A_d, Mat A_o, HYPRE_IJMatrix ij) 3163c07aadSStefano Zampini { 3263c07aadSStefano Zampini PetscErrorCode ierr; 3363c07aadSStefano Zampini PetscInt i,n_d,n_o; 3463c07aadSStefano Zampini const PetscInt *ia_d,*ia_o; 3563c07aadSStefano Zampini PetscBool done_d=PETSC_FALSE,done_o=PETSC_FALSE; 362cf14000SStefano Zampini HYPRE_Int *nnz_d=NULL,*nnz_o=NULL; 3763c07aadSStefano Zampini 3863c07aadSStefano Zampini PetscFunctionBegin; 3963c07aadSStefano Zampini if (A_d) { /* determine number of nonzero entries in local diagonal part */ 4063c07aadSStefano Zampini ierr = MatGetRowIJ(A_d,0,PETSC_FALSE,PETSC_FALSE,&n_d,&ia_d,NULL,&done_d);CHKERRQ(ierr); 4163c07aadSStefano Zampini if (done_d) { 4263c07aadSStefano Zampini ierr = PetscMalloc1(n_d,&nnz_d);CHKERRQ(ierr); 4363c07aadSStefano Zampini for (i=0; i<n_d; i++) { 4463c07aadSStefano Zampini nnz_d[i] = ia_d[i+1] - ia_d[i]; 4563c07aadSStefano Zampini } 4663c07aadSStefano Zampini } 4763c07aadSStefano Zampini ierr = MatRestoreRowIJ(A_d,0,PETSC_FALSE,PETSC_FALSE,NULL,&ia_d,NULL,&done_d);CHKERRQ(ierr); 4863c07aadSStefano Zampini } 4963c07aadSStefano Zampini if (A_o) { /* determine number of nonzero entries in local off-diagonal part */ 5063c07aadSStefano Zampini ierr = MatGetRowIJ(A_o,0,PETSC_FALSE,PETSC_FALSE,&n_o,&ia_o,NULL,&done_o);CHKERRQ(ierr); 5163c07aadSStefano Zampini if (done_o) { 5263c07aadSStefano Zampini ierr = PetscMalloc1(n_o,&nnz_o);CHKERRQ(ierr); 5363c07aadSStefano Zampini for (i=0; i<n_o; i++) { 5463c07aadSStefano Zampini nnz_o[i] = ia_o[i+1] - ia_o[i]; 5563c07aadSStefano Zampini } 5663c07aadSStefano Zampini } 5763c07aadSStefano Zampini ierr = MatRestoreRowIJ(A_o,0,PETSC_FALSE,PETSC_FALSE,&n_o,&ia_o,NULL,&done_o);CHKERRQ(ierr); 5863c07aadSStefano Zampini } 5963c07aadSStefano Zampini if (done_d) { /* set number of nonzeros in HYPRE IJ matrix */ 6063c07aadSStefano Zampini if (!done_o) { /* only diagonal part */ 6163c07aadSStefano Zampini ierr = PetscMalloc1(n_d,&nnz_o);CHKERRQ(ierr); 6263c07aadSStefano Zampini for (i=0; i<n_d; i++) { 6363c07aadSStefano Zampini nnz_o[i] = 0; 6463c07aadSStefano Zampini } 6563c07aadSStefano Zampini } 66c6698e78SStefano Zampini #if PETSC_PKG_HYPRE_VERSION_GE(2,16,0) 67c6698e78SStefano Zampini { /* If we don't do this, the columns of the matrix will be all zeros! */ 68c6698e78SStefano Zampini hypre_AuxParCSRMatrix *aux_matrix; 69c6698e78SStefano Zampini aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(ij); 70c6698e78SStefano Zampini hypre_AuxParCSRMatrixDestroy(aux_matrix); 71c6698e78SStefano Zampini hypre_IJMatrixTranslator(ij) = NULL; 722cf14000SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixSetDiagOffdSizes,(ij,nnz_d,nnz_o)); 73c6698e78SStefano Zampini aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(ij); 74c6698e78SStefano Zampini hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 1; 75c6698e78SStefano Zampini } 76c6698e78SStefano Zampini #else 772cf14000SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixSetDiagOffdSizes,(ij,nnz_d,nnz_o)); 78c6698e78SStefano Zampini #endif 7963c07aadSStefano Zampini ierr = PetscFree(nnz_d);CHKERRQ(ierr); 8063c07aadSStefano Zampini ierr = PetscFree(nnz_o);CHKERRQ(ierr); 8163c07aadSStefano Zampini } 8263c07aadSStefano Zampini PetscFunctionReturn(0); 8363c07aadSStefano Zampini } 8463c07aadSStefano Zampini 8563c07aadSStefano Zampini static PetscErrorCode MatHYPRE_CreateFromMat(Mat A, Mat_HYPRE *hA) 8663c07aadSStefano Zampini { 8763c07aadSStefano Zampini PetscErrorCode ierr; 8863c07aadSStefano Zampini PetscInt rstart,rend,cstart,cend; 8963c07aadSStefano Zampini 9063c07aadSStefano Zampini PetscFunctionBegin; 91d975228cSstefano_zampini ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 92d975228cSstefano_zampini ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 9363c07aadSStefano Zampini rstart = A->rmap->rstart; 9463c07aadSStefano Zampini rend = A->rmap->rend; 9563c07aadSStefano Zampini cstart = A->cmap->rstart; 9663c07aadSStefano Zampini cend = A->cmap->rend; 9763c07aadSStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixCreate,(hA->comm,rstart,rend-1,cstart,cend-1,&hA->ij)); 9863c07aadSStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(hA->ij,HYPRE_PARCSR)); 9963c07aadSStefano Zampini { 10063c07aadSStefano Zampini PetscBool same; 10163c07aadSStefano Zampini Mat A_d,A_o; 10263c07aadSStefano Zampini const PetscInt *colmap; 103b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATMPIAIJ,&same);CHKERRQ(ierr); 10463c07aadSStefano Zampini if (same) { 10563c07aadSStefano Zampini ierr = MatMPIAIJGetSeqAIJ(A,&A_d,&A_o,&colmap);CHKERRQ(ierr); 10663c07aadSStefano Zampini ierr = MatHYPRE_IJMatrixPreallocate(A_d,A_o,hA->ij);CHKERRQ(ierr); 10763c07aadSStefano Zampini PetscFunctionReturn(0); 10863c07aadSStefano Zampini } 109b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATMPIBAIJ,&same);CHKERRQ(ierr); 11063c07aadSStefano Zampini if (same) { 11163c07aadSStefano Zampini ierr = MatMPIBAIJGetSeqBAIJ(A,&A_d,&A_o,&colmap);CHKERRQ(ierr); 11263c07aadSStefano Zampini ierr = MatHYPRE_IJMatrixPreallocate(A_d,A_o,hA->ij);CHKERRQ(ierr); 11363c07aadSStefano Zampini PetscFunctionReturn(0); 11463c07aadSStefano Zampini } 115b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATSEQAIJ,&same);CHKERRQ(ierr); 11663c07aadSStefano Zampini if (same) { 11763c07aadSStefano Zampini ierr = MatHYPRE_IJMatrixPreallocate(A,NULL,hA->ij);CHKERRQ(ierr); 11863c07aadSStefano Zampini PetscFunctionReturn(0); 11963c07aadSStefano Zampini } 120b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATSEQBAIJ,&same);CHKERRQ(ierr); 12163c07aadSStefano Zampini if (same) { 12263c07aadSStefano Zampini ierr = MatHYPRE_IJMatrixPreallocate(A,NULL,hA->ij);CHKERRQ(ierr); 12363c07aadSStefano Zampini PetscFunctionReturn(0); 12463c07aadSStefano Zampini } 12563c07aadSStefano Zampini } 12663c07aadSStefano Zampini PetscFunctionReturn(0); 12763c07aadSStefano Zampini } 12863c07aadSStefano Zampini 12963c07aadSStefano Zampini static PetscErrorCode MatHYPRE_IJMatrixCopy(Mat A, HYPRE_IJMatrix ij) 13063c07aadSStefano Zampini { 13163c07aadSStefano Zampini PetscErrorCode ierr; 13263c07aadSStefano Zampini PetscInt i,rstart,rend,ncols,nr,nc; 13363c07aadSStefano Zampini const PetscScalar *values; 13463c07aadSStefano Zampini const PetscInt *cols; 13563c07aadSStefano Zampini PetscBool flg; 13663c07aadSStefano Zampini 13763c07aadSStefano Zampini PetscFunctionBegin; 138b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATMPIAIJ,&flg);CHKERRQ(ierr); 13963c07aadSStefano Zampini ierr = MatGetSize(A,&nr,&nc);CHKERRQ(ierr); 14063c07aadSStefano Zampini if (flg && nr == nc) { 14163c07aadSStefano Zampini ierr = MatHYPRE_IJMatrixFastCopy_MPIAIJ(A,ij);CHKERRQ(ierr); 14263c07aadSStefano Zampini PetscFunctionReturn(0); 14363c07aadSStefano Zampini } 144b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATSEQAIJ,&flg);CHKERRQ(ierr); 14563c07aadSStefano Zampini if (flg) { 14663c07aadSStefano Zampini ierr = MatHYPRE_IJMatrixFastCopy_SeqAIJ(A,ij);CHKERRQ(ierr); 14763c07aadSStefano Zampini PetscFunctionReturn(0); 14863c07aadSStefano Zampini } 14963c07aadSStefano Zampini 15063c07aadSStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixInitialize,(ij)); 15163c07aadSStefano Zampini ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr); 15263c07aadSStefano Zampini for (i=rstart; i<rend; i++) { 15363c07aadSStefano Zampini ierr = MatGetRow(A,i,&ncols,&cols,&values);CHKERRQ(ierr); 154e3977e59Sstefano_zampini if (ncols) { 1552cf14000SStefano Zampini HYPRE_Int nc = (HYPRE_Int)ncols; 1562cf14000SStefano Zampini 1572cf14000SStefano Zampini if ((PetscInt)nc != ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hypre overflow! number of columns %D for row %D",ncols,i); 15839accc25SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixSetValues,(ij,1,&nc,(HYPRE_BigInt *)&i,(HYPRE_BigInt *)cols,(HYPRE_Complex *)values)); 159e3977e59Sstefano_zampini } 16063c07aadSStefano Zampini ierr = MatRestoreRow(A,i,&ncols,&cols,&values);CHKERRQ(ierr); 16163c07aadSStefano Zampini } 16263c07aadSStefano Zampini PetscFunctionReturn(0); 16363c07aadSStefano Zampini } 16463c07aadSStefano Zampini 16563c07aadSStefano Zampini static PetscErrorCode MatHYPRE_IJMatrixFastCopy_SeqAIJ(Mat A, HYPRE_IJMatrix ij) 16663c07aadSStefano Zampini { 16763c07aadSStefano Zampini PetscErrorCode ierr; 16863c07aadSStefano Zampini Mat_SeqAIJ *pdiag = (Mat_SeqAIJ*)A->data; 16958968eb6SStefano Zampini HYPRE_Int type; 17063c07aadSStefano Zampini hypre_ParCSRMatrix *par_matrix; 17163c07aadSStefano Zampini hypre_AuxParCSRMatrix *aux_matrix; 17263c07aadSStefano Zampini hypre_CSRMatrix *hdiag; 1732cf14000SStefano Zampini PetscBool sameint = (PetscBool)(sizeof(PetscInt) == sizeof(HYPRE_Int)); 17463c07aadSStefano Zampini 17563c07aadSStefano Zampini PetscFunctionBegin; 17663c07aadSStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixInitialize,(ij)); 177ea9daf28SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(ij,&type)); 178ea9daf28SStefano Zampini if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 179ea9daf28SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(ij,(void**)&par_matrix)); 18063c07aadSStefano Zampini hdiag = hypre_ParCSRMatrixDiag(par_matrix); 18163c07aadSStefano Zampini /* 18263c07aadSStefano Zampini this is the Hack part where we monkey directly with the hypre datastructures 18363c07aadSStefano Zampini */ 1842cf14000SStefano Zampini if (sameint) { 185580bdb30SBarry Smith ierr = PetscArraycpy(hdiag->i,pdiag->i,A->rmap->n + 1);CHKERRQ(ierr); 186580bdb30SBarry Smith ierr = PetscArraycpy(hdiag->j,pdiag->j,pdiag->nz);CHKERRQ(ierr); 1872cf14000SStefano Zampini } else { 1882cf14000SStefano Zampini PetscInt i; 1892cf14000SStefano Zampini 1902cf14000SStefano Zampini for (i=0;i<A->rmap->n + 1;i++) hdiag->i[i] = (HYPRE_Int)pdiag->i[i]; 1912cf14000SStefano Zampini for (i=0;i<pdiag->nz;i++) hdiag->j[i] = (HYPRE_Int)pdiag->j[i]; 1922cf14000SStefano Zampini } 193580bdb30SBarry Smith ierr = PetscArraycpy(hdiag->data,pdiag->a,pdiag->nz);CHKERRQ(ierr); 194ea9daf28SStefano Zampini 195ea9daf28SStefano Zampini aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(ij); 19663c07aadSStefano Zampini hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 0; 19763c07aadSStefano Zampini PetscFunctionReturn(0); 19863c07aadSStefano Zampini } 19963c07aadSStefano Zampini 20063c07aadSStefano Zampini static PetscErrorCode MatHYPRE_IJMatrixFastCopy_MPIAIJ(Mat A, HYPRE_IJMatrix ij) 20163c07aadSStefano Zampini { 20263c07aadSStefano Zampini PetscErrorCode ierr; 20363c07aadSStefano Zampini Mat_MPIAIJ *pA = (Mat_MPIAIJ*)A->data; 20463c07aadSStefano Zampini Mat_SeqAIJ *pdiag,*poffd; 20563c07aadSStefano Zampini PetscInt i,*garray = pA->garray,*jj,cstart,*pjj; 2062cf14000SStefano Zampini HYPRE_Int *hjj,type; 20763c07aadSStefano Zampini hypre_ParCSRMatrix *par_matrix; 20863c07aadSStefano Zampini hypre_AuxParCSRMatrix *aux_matrix; 20963c07aadSStefano Zampini hypre_CSRMatrix *hdiag,*hoffd; 2102cf14000SStefano Zampini PetscBool sameint = (PetscBool)(sizeof(PetscInt) == sizeof(HYPRE_Int)); 21163c07aadSStefano Zampini 21263c07aadSStefano Zampini PetscFunctionBegin; 21363c07aadSStefano Zampini pdiag = (Mat_SeqAIJ*) pA->A->data; 21463c07aadSStefano Zampini poffd = (Mat_SeqAIJ*) pA->B->data; 21563c07aadSStefano Zampini /* cstart is only valid for square MPIAIJ layed out in the usual way */ 21663c07aadSStefano Zampini ierr = MatGetOwnershipRange(A,&cstart,NULL);CHKERRQ(ierr); 21763c07aadSStefano Zampini 21863c07aadSStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixInitialize,(ij)); 219ea9daf28SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(ij,&type)); 220ea9daf28SStefano Zampini if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 221ea9daf28SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(ij,(void**)&par_matrix)); 22263c07aadSStefano Zampini hdiag = hypre_ParCSRMatrixDiag(par_matrix); 22363c07aadSStefano Zampini hoffd = hypre_ParCSRMatrixOffd(par_matrix); 22463c07aadSStefano Zampini 22563c07aadSStefano Zampini /* 22663c07aadSStefano Zampini this is the Hack part where we monkey directly with the hypre datastructures 22763c07aadSStefano Zampini */ 2282cf14000SStefano Zampini if (sameint) { 229580bdb30SBarry Smith ierr = PetscArraycpy(hdiag->i,pdiag->i,pA->A->rmap->n + 1);CHKERRQ(ierr); 2302cf14000SStefano Zampini } else { 2312cf14000SStefano Zampini for (i=0; i<pA->A->rmap->n + 1; i++) hdiag->i[i] = (HYPRE_Int)(pdiag->i[i]); 2322cf14000SStefano Zampini } 23363c07aadSStefano Zampini /* need to shift the diag column indices (hdiag->j) back to global numbering since hypre is expecting this */ 2342cf14000SStefano Zampini hjj = hdiag->j; 2352cf14000SStefano Zampini pjj = pdiag->j; 236c6698e78SStefano Zampini #if PETSC_PKG_HYPRE_VERSION_GE(2,16,0) 2372cf14000SStefano Zampini for (i=0; i<pdiag->nz; i++) hjj[i] = pjj[i]; 238c6698e78SStefano Zampini #else 2392cf14000SStefano Zampini for (i=0; i<pdiag->nz; i++) hjj[i] = cstart + pjj[i]; 240c6698e78SStefano Zampini #endif 241580bdb30SBarry Smith ierr = PetscArraycpy(hdiag->data,pdiag->a,pdiag->nz);CHKERRQ(ierr); 2422cf14000SStefano Zampini if (sameint) { 243580bdb30SBarry Smith ierr = PetscArraycpy(hoffd->i,poffd->i,pA->A->rmap->n + 1);CHKERRQ(ierr); 2442cf14000SStefano Zampini } else { 2452cf14000SStefano Zampini for (i=0; i<pA->A->rmap->n + 1; i++) hoffd->i[i] = (HYPRE_Int)(poffd->i[i]); 2462cf14000SStefano Zampini } 2472cf14000SStefano Zampini 24863c07aadSStefano Zampini /* need to move the offd column indices (hoffd->j) back to global numbering since hypre is expecting this 24963c07aadSStefano Zampini If we hacked a hypre a bit more we might be able to avoid this step */ 250c6698e78SStefano Zampini #if PETSC_PKG_HYPRE_VERSION_GE(2,16,0) 251c6698e78SStefano Zampini PetscStackCallStandard(hypre_CSRMatrixBigInitialize,(hoffd)); 252c6698e78SStefano Zampini jj = (PetscInt*) hoffd->big_j; 253c6698e78SStefano Zampini #else 25463c07aadSStefano Zampini jj = (PetscInt*) hoffd->j; 255c6698e78SStefano Zampini #endif 2562cf14000SStefano Zampini pjj = poffd->j; 25763c07aadSStefano Zampini for (i=0; i<poffd->nz; i++) jj[i] = garray[pjj[i]]; 258c6698e78SStefano Zampini 259580bdb30SBarry Smith ierr = PetscArraycpy(hoffd->data,poffd->a,poffd->nz);CHKERRQ(ierr); 26063c07aadSStefano Zampini 261ea9daf28SStefano Zampini aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(ij); 26263c07aadSStefano Zampini hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 0; 26363c07aadSStefano Zampini PetscFunctionReturn(0); 26463c07aadSStefano Zampini } 26563c07aadSStefano Zampini 2662df22349SStefano Zampini static PetscErrorCode MatConvert_HYPRE_IS(Mat A, MatType mtype, MatReuse reuse, Mat* B) 2672df22349SStefano Zampini { 2682df22349SStefano Zampini Mat_HYPRE* mhA = (Mat_HYPRE*)(A->data); 2692df22349SStefano Zampini Mat lA; 2702df22349SStefano Zampini ISLocalToGlobalMapping rl2g,cl2g; 2712df22349SStefano Zampini IS is; 2722df22349SStefano Zampini hypre_ParCSRMatrix *hA; 2732df22349SStefano Zampini hypre_CSRMatrix *hdiag,*hoffd; 2742df22349SStefano Zampini MPI_Comm comm; 27539accc25SStefano Zampini HYPRE_Complex *hdd,*hod,*aa; 27639accc25SStefano Zampini PetscScalar *data; 2772cf14000SStefano Zampini HYPRE_BigInt *col_map_offd; 2782cf14000SStefano Zampini HYPRE_Int *hdi,*hdj,*hoi,*hoj; 2792df22349SStefano Zampini PetscInt *ii,*jj,*iptr,*jptr; 2802df22349SStefano Zampini PetscInt cum,dr,dc,oc,str,stc,nnz,i,jd,jo,M,N; 28158968eb6SStefano Zampini HYPRE_Int type; 2822df22349SStefano Zampini PetscErrorCode ierr; 2832df22349SStefano Zampini 2842df22349SStefano Zampini PetscFunctionBegin; 285a1787963SStefano Zampini comm = PetscObjectComm((PetscObject)A); 2862df22349SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(mhA->ij,&type)); 2872df22349SStefano Zampini if (type != HYPRE_PARCSR) SETERRQ(comm,PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 2882df22349SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(mhA->ij,(void**)&hA)); 2892df22349SStefano Zampini M = hypre_ParCSRMatrixGlobalNumRows(hA); 2902df22349SStefano Zampini N = hypre_ParCSRMatrixGlobalNumCols(hA); 2912df22349SStefano Zampini str = hypre_ParCSRMatrixFirstRowIndex(hA); 2922df22349SStefano Zampini stc = hypre_ParCSRMatrixFirstColDiag(hA); 2932df22349SStefano Zampini hdiag = hypre_ParCSRMatrixDiag(hA); 2942df22349SStefano Zampini hoffd = hypre_ParCSRMatrixOffd(hA); 2952df22349SStefano Zampini dr = hypre_CSRMatrixNumRows(hdiag); 2962df22349SStefano Zampini dc = hypre_CSRMatrixNumCols(hdiag); 2972df22349SStefano Zampini nnz = hypre_CSRMatrixNumNonzeros(hdiag); 2982df22349SStefano Zampini hdi = hypre_CSRMatrixI(hdiag); 2992df22349SStefano Zampini hdj = hypre_CSRMatrixJ(hdiag); 3002df22349SStefano Zampini hdd = hypre_CSRMatrixData(hdiag); 3012df22349SStefano Zampini oc = hypre_CSRMatrixNumCols(hoffd); 3022df22349SStefano Zampini nnz += hypre_CSRMatrixNumNonzeros(hoffd); 3032df22349SStefano Zampini hoi = hypre_CSRMatrixI(hoffd); 3042df22349SStefano Zampini hoj = hypre_CSRMatrixJ(hoffd); 3052df22349SStefano Zampini hod = hypre_CSRMatrixData(hoffd); 3062df22349SStefano Zampini if (reuse != MAT_REUSE_MATRIX) { 3072df22349SStefano Zampini PetscInt *aux; 3082df22349SStefano Zampini 3092df22349SStefano Zampini /* generate l2g maps for rows and cols */ 3102df22349SStefano Zampini ierr = ISCreateStride(comm,dr,str,1,&is);CHKERRQ(ierr); 3112df22349SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is,&rl2g);CHKERRQ(ierr); 3122df22349SStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 3132df22349SStefano Zampini col_map_offd = hypre_ParCSRMatrixColMapOffd(hA); 3142df22349SStefano Zampini ierr = PetscMalloc1(dc+oc,&aux);CHKERRQ(ierr); 3152df22349SStefano Zampini for (i=0; i<dc; i++) aux[i] = i+stc; 3162df22349SStefano Zampini for (i=0; i<oc; i++) aux[i+dc] = col_map_offd[i]; 3172df22349SStefano Zampini ierr = ISCreateGeneral(comm,dc+oc,aux,PETSC_OWN_POINTER,&is);CHKERRQ(ierr); 3182df22349SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(is,&cl2g);CHKERRQ(ierr); 3192df22349SStefano Zampini ierr = ISDestroy(&is);CHKERRQ(ierr); 3202df22349SStefano Zampini /* create MATIS object */ 3212df22349SStefano Zampini ierr = MatCreate(comm,B);CHKERRQ(ierr); 3222df22349SStefano Zampini ierr = MatSetSizes(*B,dr,dc,M,N);CHKERRQ(ierr); 3232df22349SStefano Zampini ierr = MatSetType(*B,MATIS);CHKERRQ(ierr); 3242df22349SStefano Zampini ierr = MatSetLocalToGlobalMapping(*B,rl2g,cl2g);CHKERRQ(ierr); 3252df22349SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr); 3262df22349SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr); 3272df22349SStefano Zampini 3282df22349SStefano Zampini /* allocate CSR for local matrix */ 3292df22349SStefano Zampini ierr = PetscMalloc1(dr+1,&iptr);CHKERRQ(ierr); 3302df22349SStefano Zampini ierr = PetscMalloc1(nnz,&jptr);CHKERRQ(ierr); 3312df22349SStefano Zampini ierr = PetscMalloc1(nnz,&data);CHKERRQ(ierr); 3322df22349SStefano Zampini } else { 3332df22349SStefano Zampini PetscInt nr; 3342df22349SStefano Zampini PetscBool done; 3352df22349SStefano Zampini ierr = MatISGetLocalMat(*B,&lA);CHKERRQ(ierr); 3362df22349SStefano Zampini ierr = MatGetRowIJ(lA,0,PETSC_FALSE,PETSC_FALSE,&nr,(const PetscInt**)&iptr,(const PetscInt**)&jptr,&done);CHKERRQ(ierr); 3372df22349SStefano Zampini if (nr != dr) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"Cannot reuse mat: invalid number of rows in local mat! %D != %D",nr,dr); 3382df22349SStefano Zampini if (iptr[nr] < nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"Cannot reuse mat: invalid number of nonzeros in local mat! reuse %D requested %D",iptr[nr],nnz); 3392df22349SStefano Zampini ierr = MatSeqAIJGetArray(lA,&data);CHKERRQ(ierr); 3402df22349SStefano Zampini } 3412df22349SStefano Zampini /* merge local matrices */ 3422df22349SStefano Zampini ii = iptr; 3432df22349SStefano Zampini jj = jptr; 34439accc25SStefano Zampini aa = (HYPRE_Complex*)data; /* this cast fixes the clang error when doing the assignments below: implicit conversion from 'HYPRE_Complex' (aka '_Complex double') to 'double' is not permitted in C++ */ 3452df22349SStefano Zampini *ii = *(hdi++) + *(hoi++); 3462df22349SStefano Zampini for (jd=0,jo=0,cum=0; *ii<nnz; cum++) { 34739accc25SStefano Zampini PetscScalar *aold = (PetscScalar*)aa; 3482df22349SStefano Zampini PetscInt *jold = jj,nc = jd+jo; 3492df22349SStefano Zampini for (; jd<*hdi; jd++) { *jj++ = *hdj++; *aa++ = *hdd++; } 3502df22349SStefano Zampini for (; jo<*hoi; jo++) { *jj++ = *hoj++ + dc; *aa++ = *hod++; } 3512df22349SStefano Zampini *(++ii) = *(hdi++) + *(hoi++); 3522df22349SStefano Zampini ierr = PetscSortIntWithScalarArray(jd+jo-nc,jold,aold);CHKERRQ(ierr); 3532df22349SStefano Zampini } 3542df22349SStefano Zampini for (; cum<dr; cum++) *(++ii) = nnz; 3552df22349SStefano Zampini if (reuse != MAT_REUSE_MATRIX) { 356a033916dSStefano Zampini Mat_SeqAIJ* a; 357a033916dSStefano Zampini 3582df22349SStefano Zampini ierr = MatCreateSeqAIJWithArrays(PETSC_COMM_SELF,dr,dc+oc,iptr,jptr,data,&lA);CHKERRQ(ierr); 3592df22349SStefano Zampini ierr = MatISSetLocalMat(*B,lA);CHKERRQ(ierr); 360a033916dSStefano Zampini /* hack SeqAIJ */ 361a033916dSStefano Zampini a = (Mat_SeqAIJ*)(lA->data); 362a033916dSStefano Zampini a->free_a = PETSC_TRUE; 363a033916dSStefano Zampini a->free_ij = PETSC_TRUE; 3642df22349SStefano Zampini ierr = MatDestroy(&lA);CHKERRQ(ierr); 3652df22349SStefano Zampini } 3662df22349SStefano Zampini ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 3672df22349SStefano Zampini ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 3682df22349SStefano Zampini if (reuse == MAT_INPLACE_MATRIX) { 3692df22349SStefano Zampini ierr = MatHeaderReplace(A,B);CHKERRQ(ierr); 3702df22349SStefano Zampini } 3712df22349SStefano Zampini PetscFunctionReturn(0); 3722df22349SStefano Zampini } 3732df22349SStefano Zampini 37463c07aadSStefano Zampini PETSC_INTERN PetscErrorCode MatConvert_AIJ_HYPRE(Mat A, MatType type, MatReuse reuse, Mat *B) 37563c07aadSStefano Zampini { 37684d4e069SStefano Zampini Mat M = NULL; 37763c07aadSStefano Zampini Mat_HYPRE *hB; 37863c07aadSStefano Zampini MPI_Comm comm = PetscObjectComm((PetscObject)A); 37963c07aadSStefano Zampini PetscErrorCode ierr; 38063c07aadSStefano Zampini 38163c07aadSStefano Zampini PetscFunctionBegin; 38263c07aadSStefano Zampini if (reuse == MAT_REUSE_MATRIX) { 38363c07aadSStefano Zampini /* always destroy the old matrix and create a new memory; 38463c07aadSStefano Zampini hope this does not churn the memory too much. The problem 38563c07aadSStefano Zampini is I do not know if it is possible to put the matrix back to 38663c07aadSStefano Zampini its initial state so that we can directly copy the values 38763c07aadSStefano Zampini the second time through. */ 38863c07aadSStefano Zampini hB = (Mat_HYPRE*)((*B)->data); 38963c07aadSStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixDestroy,(hB->ij)); 39063c07aadSStefano Zampini } else { 39184d4e069SStefano Zampini ierr = MatCreate(comm,&M);CHKERRQ(ierr); 39284d4e069SStefano Zampini ierr = MatSetType(M,MATHYPRE);CHKERRQ(ierr); 39384d4e069SStefano Zampini ierr = MatSetSizes(M,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 39484d4e069SStefano Zampini hB = (Mat_HYPRE*)(M->data); 39584d4e069SStefano Zampini if (reuse == MAT_INITIAL_MATRIX) *B = M; 39663c07aadSStefano Zampini } 39763c07aadSStefano Zampini ierr = MatHYPRE_CreateFromMat(A,hB);CHKERRQ(ierr); 39863c07aadSStefano Zampini ierr = MatHYPRE_IJMatrixCopy(A,hB->ij);CHKERRQ(ierr); 39984d4e069SStefano Zampini if (reuse == MAT_INPLACE_MATRIX) { 40084d4e069SStefano Zampini ierr = MatHeaderReplace(A,&M);CHKERRQ(ierr); 40184d4e069SStefano Zampini } 4024ec6421dSstefano_zampini (*B)->preallocated = PETSC_TRUE; 40363c07aadSStefano Zampini ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 40463c07aadSStefano Zampini ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 40563c07aadSStefano Zampini PetscFunctionReturn(0); 40663c07aadSStefano Zampini } 40763c07aadSStefano Zampini 408ea9daf28SStefano Zampini static PetscErrorCode MatConvert_HYPRE_AIJ(Mat A, MatType mtype, MatReuse reuse, Mat *B) 40963c07aadSStefano Zampini { 41063c07aadSStefano Zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 41163c07aadSStefano Zampini hypre_ParCSRMatrix *parcsr; 41263c07aadSStefano Zampini hypre_CSRMatrix *hdiag,*hoffd; 41363c07aadSStefano Zampini MPI_Comm comm; 41463c07aadSStefano Zampini PetscScalar *da,*oa,*aptr; 41563c07aadSStefano Zampini PetscInt *dii,*djj,*oii,*ojj,*iptr; 41663c07aadSStefano Zampini PetscInt i,dnnz,onnz,m,n; 41758968eb6SStefano Zampini HYPRE_Int type; 41863c07aadSStefano Zampini PetscMPIInt size; 4192cf14000SStefano Zampini PetscBool sameint = (PetscBool)(sizeof(PetscInt) == sizeof(HYPRE_Int)); 42063c07aadSStefano Zampini PetscErrorCode ierr; 42163c07aadSStefano Zampini 42263c07aadSStefano Zampini PetscFunctionBegin; 42363c07aadSStefano Zampini comm = PetscObjectComm((PetscObject)A); 42463c07aadSStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type)); 42563c07aadSStefano Zampini if (type != HYPRE_PARCSR) SETERRQ(comm,PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 42663c07aadSStefano Zampini if (reuse == MAT_REUSE_MATRIX) { 42763c07aadSStefano Zampini PetscBool ismpiaij,isseqaij; 428b9e7e5c1SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)*B,MATMPIAIJ,&ismpiaij);CHKERRQ(ierr); 4294099cc6bSBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)*B,MATSEQAIJ,&isseqaij);CHKERRQ(ierr); 43063c07aadSStefano Zampini if (!ismpiaij && !isseqaij) SETERRQ(comm,PETSC_ERR_SUP,"Only MATMPIAIJ or MATSEQAIJ are supported"); 43163c07aadSStefano Zampini } 43263c07aadSStefano Zampini ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 43363c07aadSStefano Zampini 43463c07aadSStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)&parcsr)); 43563c07aadSStefano Zampini hdiag = hypre_ParCSRMatrixDiag(parcsr); 43663c07aadSStefano Zampini hoffd = hypre_ParCSRMatrixOffd(parcsr); 43763c07aadSStefano Zampini m = hypre_CSRMatrixNumRows(hdiag); 43863c07aadSStefano Zampini n = hypre_CSRMatrixNumCols(hdiag); 43963c07aadSStefano Zampini dnnz = hypre_CSRMatrixNumNonzeros(hdiag); 44063c07aadSStefano Zampini onnz = hypre_CSRMatrixNumNonzeros(hoffd); 441225daaf8SStefano Zampini if (reuse == MAT_INITIAL_MATRIX) { 44263c07aadSStefano Zampini ierr = PetscMalloc1(m+1,&dii);CHKERRQ(ierr); 44363c07aadSStefano Zampini ierr = PetscMalloc1(dnnz,&djj);CHKERRQ(ierr); 44463c07aadSStefano Zampini ierr = PetscMalloc1(dnnz,&da);CHKERRQ(ierr); 445225daaf8SStefano Zampini } else if (reuse == MAT_REUSE_MATRIX) { 44663c07aadSStefano Zampini PetscInt nr; 44763c07aadSStefano Zampini PetscBool done; 44863c07aadSStefano Zampini if (size > 1) { 44963c07aadSStefano Zampini Mat_MPIAIJ *b = (Mat_MPIAIJ*)((*B)->data); 45063c07aadSStefano Zampini 45163c07aadSStefano Zampini ierr = MatGetRowIJ(b->A,0,PETSC_FALSE,PETSC_FALSE,&nr,(const PetscInt**)&dii,(const PetscInt**)&djj,&done);CHKERRQ(ierr); 45263c07aadSStefano Zampini if (nr != m) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"Cannot reuse mat: invalid number of local rows in diag part! %D != %D",nr,m); 45363c07aadSStefano Zampini if (dii[nr] < dnnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"Cannot reuse mat: invalid number of nonzeros in diag part! reuse %D hypre %D",dii[nr],dnnz); 45463c07aadSStefano Zampini ierr = MatSeqAIJGetArray(b->A,&da);CHKERRQ(ierr); 45563c07aadSStefano Zampini } else { 45663c07aadSStefano Zampini ierr = MatGetRowIJ(*B,0,PETSC_FALSE,PETSC_FALSE,&nr,(const PetscInt**)&dii,(const PetscInt**)&djj,&done);CHKERRQ(ierr); 45763c07aadSStefano Zampini if (nr != m) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"Cannot reuse mat: invalid number of local rows! %D != %D",nr,m); 45863c07aadSStefano Zampini if (dii[nr] < dnnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"Cannot reuse mat: invalid number of nonzeros! reuse %D hypre %D",dii[nr],dnnz); 45963c07aadSStefano Zampini ierr = MatSeqAIJGetArray(*B,&da);CHKERRQ(ierr); 46063c07aadSStefano Zampini } 461225daaf8SStefano Zampini } else { /* MAT_INPLACE_MATRIX */ 4622cf14000SStefano Zampini if (!sameint) { 4632cf14000SStefano Zampini ierr = PetscMalloc1(m+1,&dii);CHKERRQ(ierr); 4642cf14000SStefano Zampini ierr = PetscMalloc1(dnnz,&djj);CHKERRQ(ierr); 4652cf14000SStefano Zampini } else { 4667d968826Sstefano_zampini dii = (PetscInt*)hypre_CSRMatrixI(hdiag); 4677d968826Sstefano_zampini djj = (PetscInt*)hypre_CSRMatrixJ(hdiag); 46863c07aadSStefano Zampini } 46939accc25SStefano Zampini da = (PetscScalar*)hypre_CSRMatrixData(hdiag); 47063c07aadSStefano Zampini } 4712cf14000SStefano Zampini 4722cf14000SStefano Zampini if (!sameint) { 4732cf14000SStefano Zampini for (i=0;i<m+1;i++) dii[i] = (PetscInt)(hypre_CSRMatrixI(hdiag)[i]); 4742cf14000SStefano Zampini for (i=0;i<dnnz;i++) djj[i] = (PetscInt)(hypre_CSRMatrixJ(hdiag)[i]); 4752cf14000SStefano Zampini } else { 476580bdb30SBarry Smith ierr = PetscArraycpy(dii,hypre_CSRMatrixI(hdiag),m+1);CHKERRQ(ierr); 477580bdb30SBarry Smith ierr = PetscArraycpy(djj,hypre_CSRMatrixJ(hdiag),dnnz);CHKERRQ(ierr); 4782cf14000SStefano Zampini } 479580bdb30SBarry Smith ierr = PetscArraycpy(da,hypre_CSRMatrixData(hdiag),dnnz);CHKERRQ(ierr); 48063c07aadSStefano Zampini iptr = djj; 48163c07aadSStefano Zampini aptr = da; 48263c07aadSStefano Zampini for (i=0; i<m; i++) { 48363c07aadSStefano Zampini PetscInt nc = dii[i+1]-dii[i]; 48463c07aadSStefano Zampini ierr = PetscSortIntWithScalarArray(nc,iptr,aptr);CHKERRQ(ierr); 48563c07aadSStefano Zampini iptr += nc; 48663c07aadSStefano Zampini aptr += nc; 48763c07aadSStefano Zampini } 48863c07aadSStefano Zampini if (size > 1) { 4892cf14000SStefano Zampini HYPRE_BigInt *coffd; 4902cf14000SStefano Zampini HYPRE_Int *offdj; 49163c07aadSStefano Zampini 492225daaf8SStefano Zampini if (reuse == MAT_INITIAL_MATRIX) { 49363c07aadSStefano Zampini ierr = PetscMalloc1(m+1,&oii);CHKERRQ(ierr); 49463c07aadSStefano Zampini ierr = PetscMalloc1(onnz,&ojj);CHKERRQ(ierr); 49563c07aadSStefano Zampini ierr = PetscMalloc1(onnz,&oa);CHKERRQ(ierr); 496225daaf8SStefano Zampini } else if (reuse == MAT_REUSE_MATRIX) { 49763c07aadSStefano Zampini Mat_MPIAIJ *b = (Mat_MPIAIJ*)((*B)->data); 49863c07aadSStefano Zampini PetscInt nr,hr = hypre_CSRMatrixNumRows(hoffd); 49963c07aadSStefano Zampini PetscBool done; 50063c07aadSStefano Zampini 50163c07aadSStefano Zampini ierr = MatGetRowIJ(b->B,0,PETSC_FALSE,PETSC_FALSE,&nr,(const PetscInt**)&oii,(const PetscInt**)&ojj,&done);CHKERRQ(ierr); 50263c07aadSStefano Zampini if (nr != hr) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"Cannot reuse mat: invalid number of local rows in offdiag part! %D != %D",nr,hr); 50363c07aadSStefano Zampini if (oii[nr] < onnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"Cannot reuse mat: invalid number of nonzeros in offdiag part! reuse %D hypre %D",oii[nr],onnz); 50463c07aadSStefano Zampini ierr = MatSeqAIJGetArray(b->B,&oa);CHKERRQ(ierr); 505225daaf8SStefano Zampini } else { /* MAT_INPLACE_MATRIX */ 5062cf14000SStefano Zampini if (!sameint) { 5072cf14000SStefano Zampini ierr = PetscMalloc1(m+1,&oii);CHKERRQ(ierr); 5082cf14000SStefano Zampini ierr = PetscMalloc1(onnz,&ojj);CHKERRQ(ierr); 5092cf14000SStefano Zampini } else { 5107d968826Sstefano_zampini oii = (PetscInt*)hypre_CSRMatrixI(hoffd); 5117d968826Sstefano_zampini ojj = (PetscInt*)hypre_CSRMatrixJ(hoffd); 51263c07aadSStefano Zampini } 51339accc25SStefano Zampini oa = (PetscScalar*)hypre_CSRMatrixData(hoffd); 51463c07aadSStefano Zampini } 5152cf14000SStefano Zampini if (!sameint) { 5162cf14000SStefano Zampini for (i=0;i<m+1;i++) oii[i] = (PetscInt)(hypre_CSRMatrixI(hoffd)[i]); 5172cf14000SStefano Zampini } else { 518580bdb30SBarry Smith ierr = PetscArraycpy(oii,hypre_CSRMatrixI(hoffd),m+1);CHKERRQ(ierr); 5192cf14000SStefano Zampini } 52063c07aadSStefano Zampini offdj = hypre_CSRMatrixJ(hoffd); 52163c07aadSStefano Zampini coffd = hypre_ParCSRMatrixColMapOffd(parcsr); 52263c07aadSStefano Zampini for (i=0; i<onnz; i++) ojj[i] = coffd[offdj[i]]; 523580bdb30SBarry Smith ierr = PetscArraycpy(oa,hypre_CSRMatrixData(hoffd),onnz);CHKERRQ(ierr); 52463c07aadSStefano Zampini iptr = ojj; 52563c07aadSStefano Zampini aptr = oa; 52663c07aadSStefano Zampini for (i=0; i<m; i++) { 52763c07aadSStefano Zampini PetscInt nc = oii[i+1]-oii[i]; 52863c07aadSStefano Zampini ierr = PetscSortIntWithScalarArray(nc,iptr,aptr);CHKERRQ(ierr); 52963c07aadSStefano Zampini iptr += nc; 53063c07aadSStefano Zampini aptr += nc; 53163c07aadSStefano Zampini } 532225daaf8SStefano Zampini if (reuse == MAT_INITIAL_MATRIX) { 53363c07aadSStefano Zampini Mat_MPIAIJ *b; 53463c07aadSStefano Zampini Mat_SeqAIJ *d,*o; 535225daaf8SStefano Zampini 53641073d71Sstefano_zampini ierr = MatCreateMPIAIJWithSplitArrays(comm,m,n,PETSC_DECIDE,PETSC_DECIDE,dii,djj,da,oii,ojj,oa,B);CHKERRQ(ierr); 53763c07aadSStefano Zampini /* hack MPIAIJ */ 53863c07aadSStefano Zampini b = (Mat_MPIAIJ*)((*B)->data); 53963c07aadSStefano Zampini d = (Mat_SeqAIJ*)b->A->data; 54063c07aadSStefano Zampini o = (Mat_SeqAIJ*)b->B->data; 54163c07aadSStefano Zampini d->free_a = PETSC_TRUE; 54263c07aadSStefano Zampini d->free_ij = PETSC_TRUE; 54363c07aadSStefano Zampini o->free_a = PETSC_TRUE; 54463c07aadSStefano Zampini o->free_ij = PETSC_TRUE; 545225daaf8SStefano Zampini } else if (reuse == MAT_INPLACE_MATRIX) { 546225daaf8SStefano Zampini Mat T; 5472cf14000SStefano Zampini 54841073d71Sstefano_zampini ierr = MatCreateMPIAIJWithSplitArrays(comm,m,n,PETSC_DECIDE,PETSC_DECIDE,dii,djj,da,oii,ojj,oa,&T);CHKERRQ(ierr); 5492cf14000SStefano Zampini if (sameint) { /* ownership of CSR pointers is transferred to PETSc */ 550225daaf8SStefano Zampini hypre_CSRMatrixI(hdiag) = NULL; 551225daaf8SStefano Zampini hypre_CSRMatrixJ(hdiag) = NULL; 552225daaf8SStefano Zampini hypre_CSRMatrixI(hoffd) = NULL; 553225daaf8SStefano Zampini hypre_CSRMatrixJ(hoffd) = NULL; 5542cf14000SStefano Zampini } else { /* Hack MPIAIJ -> free ij but not a */ 5552cf14000SStefano Zampini Mat_MPIAIJ *b = (Mat_MPIAIJ*)(T->data); 5562cf14000SStefano Zampini Mat_SeqAIJ *d = (Mat_SeqAIJ*)(b->A->data); 5572cf14000SStefano Zampini Mat_SeqAIJ *o = (Mat_SeqAIJ*)(b->B->data); 5582cf14000SStefano Zampini 5592cf14000SStefano Zampini d->free_ij = PETSC_TRUE; 5602cf14000SStefano Zampini o->free_ij = PETSC_TRUE; 5612cf14000SStefano Zampini } 5622cf14000SStefano Zampini hypre_CSRMatrixData(hdiag) = NULL; 563225daaf8SStefano Zampini hypre_CSRMatrixData(hoffd) = NULL; 564225daaf8SStefano Zampini ierr = MatHeaderReplace(A,&T);CHKERRQ(ierr); 56563c07aadSStefano Zampini } 566225daaf8SStefano Zampini } else { 567225daaf8SStefano Zampini oii = NULL; 568225daaf8SStefano Zampini ojj = NULL; 569225daaf8SStefano Zampini oa = NULL; 570225daaf8SStefano Zampini if (reuse == MAT_INITIAL_MATRIX) { 57163c07aadSStefano Zampini Mat_SeqAIJ* b; 5722cf14000SStefano Zampini 57363c07aadSStefano Zampini ierr = MatCreateSeqAIJWithArrays(comm,m,n,dii,djj,da,B);CHKERRQ(ierr); 57463c07aadSStefano Zampini /* hack SeqAIJ */ 57563c07aadSStefano Zampini b = (Mat_SeqAIJ*)((*B)->data); 57663c07aadSStefano Zampini b->free_a = PETSC_TRUE; 57763c07aadSStefano Zampini b->free_ij = PETSC_TRUE; 578225daaf8SStefano Zampini } else if (reuse == MAT_INPLACE_MATRIX) { 579225daaf8SStefano Zampini Mat T; 5802cf14000SStefano Zampini 581225daaf8SStefano Zampini ierr = MatCreateSeqAIJWithArrays(comm,m,n,dii,djj,da,&T);CHKERRQ(ierr); 5822cf14000SStefano Zampini if (sameint) { /* ownership of CSR pointers is transferred to PETSc */ 583225daaf8SStefano Zampini hypre_CSRMatrixI(hdiag) = NULL; 584225daaf8SStefano Zampini hypre_CSRMatrixJ(hdiag) = NULL; 5852cf14000SStefano Zampini } else { /* free ij but not a */ 5862cf14000SStefano Zampini Mat_SeqAIJ* b = (Mat_SeqAIJ*)(T->data); 5872cf14000SStefano Zampini 5882cf14000SStefano Zampini b->free_ij = PETSC_TRUE; 5892cf14000SStefano Zampini } 590225daaf8SStefano Zampini hypre_CSRMatrixData(hdiag) = NULL; 591225daaf8SStefano Zampini ierr = MatHeaderReplace(A,&T);CHKERRQ(ierr); 59263c07aadSStefano Zampini } 593225daaf8SStefano Zampini } 594225daaf8SStefano Zampini 5952cf14000SStefano Zampini /* we have to use hypre_Tfree to free the HYPRE arrays 5962cf14000SStefano Zampini that PETSc now onws */ 59763c07aadSStefano Zampini if (reuse == MAT_INPLACE_MATRIX) { 5982cf14000SStefano Zampini PetscInt nh; 5992cf14000SStefano Zampini void *ptrs[6] = {da,oa,dii,djj,oii,ojj}; 6002cf14000SStefano Zampini const char *names[6] = {"_hypre_csr_da", 6012cf14000SStefano Zampini "_hypre_csr_oa", 6022cf14000SStefano Zampini "_hypre_csr_dii", 603225daaf8SStefano Zampini "_hypre_csr_djj", 604225daaf8SStefano Zampini "_hypre_csr_oii", 6052cf14000SStefano Zampini "_hypre_csr_ojj"}; 6062cf14000SStefano Zampini nh = sameint ? 6 : 2; 6072cf14000SStefano Zampini for (i=0; i<nh; i++) { 608225daaf8SStefano Zampini PetscContainer c; 609225daaf8SStefano Zampini 610225daaf8SStefano Zampini ierr = PetscContainerCreate(comm,&c);CHKERRQ(ierr); 611225daaf8SStefano Zampini ierr = PetscContainerSetPointer(c,ptrs[i]);CHKERRQ(ierr); 612225daaf8SStefano Zampini ierr = PetscContainerSetUserDestroy(c,hypre_array_destroy);CHKERRQ(ierr); 613225daaf8SStefano Zampini ierr = PetscObjectCompose((PetscObject)(*B),names[i],(PetscObject)c);CHKERRQ(ierr); 614225daaf8SStefano Zampini ierr = PetscContainerDestroy(&c);CHKERRQ(ierr); 615225daaf8SStefano Zampini } 61663c07aadSStefano Zampini } 61763c07aadSStefano Zampini PetscFunctionReturn(0); 61863c07aadSStefano Zampini } 61963c07aadSStefano Zampini 620613e5ff0Sstefano_zampini static PetscErrorCode MatAIJGetParCSR_Private(Mat A, hypre_ParCSRMatrix **hA) 621c1a070e6SStefano Zampini { 622613e5ff0Sstefano_zampini hypre_ParCSRMatrix *tA; 623c1a070e6SStefano Zampini hypre_CSRMatrix *hdiag,*hoffd; 624c1a070e6SStefano Zampini Mat_SeqAIJ *diag,*offd; 6252cf14000SStefano Zampini PetscInt *garray,i,noffd,dnnz,onnz,*row_starts,*col_starts; 626c1a070e6SStefano Zampini MPI_Comm comm = PetscObjectComm((PetscObject)A); 627613e5ff0Sstefano_zampini PetscBool ismpiaij,isseqaij; 6282cf14000SStefano Zampini PetscBool sameint = (PetscBool)(sizeof(PetscInt) == sizeof(HYPRE_Int)); 629c1a070e6SStefano Zampini PetscErrorCode ierr; 630c1a070e6SStefano Zampini 631c1a070e6SStefano Zampini PetscFunctionBegin; 6323937f725SBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATMPIAIJ,&ismpiaij);CHKERRQ(ierr); 6334099cc6bSBarry Smith ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATSEQAIJ,&isseqaij);CHKERRQ(ierr); 634a32993e3SJed Brown if (!ismpiaij && !isseqaij) SETERRQ1(comm,PETSC_ERR_SUP,"Unsupported type %s",((PetscObject)A)->type_name); 635c1a070e6SStefano Zampini if (ismpiaij) { 636c1a070e6SStefano Zampini Mat_MPIAIJ *a = (Mat_MPIAIJ*)(A->data); 637c1a070e6SStefano Zampini 638c1a070e6SStefano Zampini diag = (Mat_SeqAIJ*)a->A->data; 639c1a070e6SStefano Zampini offd = (Mat_SeqAIJ*)a->B->data; 640c1a070e6SStefano Zampini garray = a->garray; 641c1a070e6SStefano Zampini noffd = a->B->cmap->N; 642c1a070e6SStefano Zampini dnnz = diag->nz; 643c1a070e6SStefano Zampini onnz = offd->nz; 644c1a070e6SStefano Zampini } else { 645c1a070e6SStefano Zampini diag = (Mat_SeqAIJ*)A->data; 646c1a070e6SStefano Zampini offd = NULL; 647c1a070e6SStefano Zampini garray = NULL; 648c1a070e6SStefano Zampini noffd = 0; 649c1a070e6SStefano Zampini dnnz = diag->nz; 650c1a070e6SStefano Zampini onnz = 0; 651c1a070e6SStefano Zampini } 652225daaf8SStefano Zampini 653c1a070e6SStefano Zampini /* create a temporary ParCSR */ 654c1a070e6SStefano Zampini if (HYPRE_AssumedPartitionCheck()) { 655c1a070e6SStefano Zampini PetscMPIInt myid; 656c1a070e6SStefano Zampini 657c1a070e6SStefano Zampini ierr = MPI_Comm_rank(comm,&myid);CHKERRQ(ierr); 658c1a070e6SStefano Zampini row_starts = A->rmap->range + myid; 659c1a070e6SStefano Zampini col_starts = A->cmap->range + myid; 660c1a070e6SStefano Zampini } else { 661c1a070e6SStefano Zampini row_starts = A->rmap->range; 662c1a070e6SStefano Zampini col_starts = A->cmap->range; 663c1a070e6SStefano Zampini } 6642cf14000SStefano Zampini tA = hypre_ParCSRMatrixCreate(comm,A->rmap->N,A->cmap->N,(HYPRE_BigInt*)row_starts,(HYPRE_BigInt*)col_starts,noffd,dnnz,onnz); 665c1a070e6SStefano Zampini hypre_ParCSRMatrixSetRowStartsOwner(tA,0); 666c1a070e6SStefano Zampini hypre_ParCSRMatrixSetColStartsOwner(tA,0); 667c1a070e6SStefano Zampini 668225daaf8SStefano Zampini /* set diagonal part */ 669c1a070e6SStefano Zampini hdiag = hypre_ParCSRMatrixDiag(tA); 6702cf14000SStefano Zampini if (sameint) { /* reuse CSR pointers */ 6717d968826Sstefano_zampini hypre_CSRMatrixI(hdiag) = (HYPRE_Int*)diag->i; 6727d968826Sstefano_zampini hypre_CSRMatrixJ(hdiag) = (HYPRE_Int*)diag->j; 6732cf14000SStefano Zampini } else { /* malloc CSR pointers */ 6742cf14000SStefano Zampini HYPRE_Int *hi,*hj; 6752cf14000SStefano Zampini 6762cf14000SStefano Zampini ierr = PetscMalloc2(A->rmap->n+1,&hi,dnnz,&hj);CHKERRQ(ierr); 6772cf14000SStefano Zampini for (i = 0; i < A->rmap->n+1; i++) hi[i] = (HYPRE_Int)(diag->i[i]); 6782cf14000SStefano Zampini for (i = 0; i < dnnz; i++) hj[i] = (HYPRE_Int)(diag->j[i]); 6792cf14000SStefano Zampini hypre_CSRMatrixI(hdiag) = hi; 6802cf14000SStefano Zampini hypre_CSRMatrixJ(hdiag) = hj; 6812cf14000SStefano Zampini } 68239accc25SStefano Zampini hypre_CSRMatrixData(hdiag) = (HYPRE_Complex*)diag->a; 683c1a070e6SStefano Zampini hypre_CSRMatrixNumNonzeros(hdiag) = diag->nz; 684c1a070e6SStefano Zampini hypre_CSRMatrixSetRownnz(hdiag); 685c1a070e6SStefano Zampini hypre_CSRMatrixSetDataOwner(hdiag,0); 686c1a070e6SStefano Zampini 687225daaf8SStefano Zampini /* set offdiagonal part */ 688c1a070e6SStefano Zampini hoffd = hypre_ParCSRMatrixOffd(tA); 689c1a070e6SStefano Zampini if (offd) { 6902cf14000SStefano Zampini if (sameint) { /* reuse CSR pointers */ 6917d968826Sstefano_zampini hypre_CSRMatrixI(hoffd) = (HYPRE_Int*)offd->i; 6927d968826Sstefano_zampini hypre_CSRMatrixJ(hoffd) = (HYPRE_Int*)offd->j; 6932cf14000SStefano Zampini } else { /* malloc CSR pointers */ 6942cf14000SStefano Zampini HYPRE_Int *hi,*hj; 6952cf14000SStefano Zampini 6962cf14000SStefano Zampini ierr = PetscMalloc2(A->rmap->n+1,&hi,onnz,&hj);CHKERRQ(ierr); 6972cf14000SStefano Zampini for (i = 0; i < A->rmap->n+1; i++) hi[i] = (HYPRE_Int)(offd->i[i]); 6982cf14000SStefano Zampini for (i = 0; i < onnz; i++) hj[i] = (HYPRE_Int)(offd->j[i]); 6992cf14000SStefano Zampini hypre_CSRMatrixI(hoffd) = hi; 7002cf14000SStefano Zampini hypre_CSRMatrixJ(hoffd) = hj; 7012cf14000SStefano Zampini } 70239accc25SStefano Zampini hypre_CSRMatrixData(hoffd) = (HYPRE_Complex*)offd->a; 703c1a070e6SStefano Zampini hypre_CSRMatrixNumNonzeros(hoffd) = offd->nz; 704c1a070e6SStefano Zampini hypre_CSRMatrixSetRownnz(hoffd); 705c1a070e6SStefano Zampini hypre_CSRMatrixSetDataOwner(hoffd,0); 706c1a070e6SStefano Zampini hypre_ParCSRMatrixSetNumNonzeros(tA); 7072cf14000SStefano Zampini hypre_ParCSRMatrixColMapOffd(tA) = (HYPRE_BigInt*)garray; 708c1a070e6SStefano Zampini } 709613e5ff0Sstefano_zampini *hA = tA; 710613e5ff0Sstefano_zampini PetscFunctionReturn(0); 711613e5ff0Sstefano_zampini } 712c1a070e6SStefano Zampini 713613e5ff0Sstefano_zampini static PetscErrorCode MatAIJRestoreParCSR_Private(Mat A, hypre_ParCSRMatrix **hA) 714613e5ff0Sstefano_zampini { 715613e5ff0Sstefano_zampini hypre_CSRMatrix *hdiag,*hoffd; 7162cf14000SStefano Zampini PetscBool sameint = (PetscBool)(sizeof(PetscInt) == sizeof(HYPRE_Int)); 7172cf14000SStefano Zampini PetscErrorCode ierr; 718c1a070e6SStefano Zampini 719613e5ff0Sstefano_zampini PetscFunctionBegin; 720613e5ff0Sstefano_zampini hdiag = hypre_ParCSRMatrixDiag(*hA); 721613e5ff0Sstefano_zampini hoffd = hypre_ParCSRMatrixOffd(*hA); 7222cf14000SStefano Zampini /* free temporary memory allocated by PETSc */ 7232cf14000SStefano Zampini if (!sameint) { 7242cf14000SStefano Zampini HYPRE_Int *hi,*hj; 7252cf14000SStefano Zampini 7262cf14000SStefano Zampini hi = hypre_CSRMatrixI(hdiag); 7272cf14000SStefano Zampini hj = hypre_CSRMatrixJ(hdiag); 7282cf14000SStefano Zampini ierr = PetscFree2(hi,hj);CHKERRQ(ierr); 7292cf14000SStefano Zampini if (hoffd) { 7302cf14000SStefano Zampini hi = hypre_CSRMatrixI(hoffd); 7312cf14000SStefano Zampini hj = hypre_CSRMatrixJ(hoffd); 7322cf14000SStefano Zampini ierr = PetscFree2(hi,hj);CHKERRQ(ierr); 7332cf14000SStefano Zampini } 7342cf14000SStefano Zampini } 735c1a070e6SStefano Zampini /* set pointers to NULL before destroying tA */ 736c1a070e6SStefano Zampini hypre_CSRMatrixI(hdiag) = NULL; 737c1a070e6SStefano Zampini hypre_CSRMatrixJ(hdiag) = NULL; 738c1a070e6SStefano Zampini hypre_CSRMatrixData(hdiag) = NULL; 739c1a070e6SStefano Zampini hypre_CSRMatrixI(hoffd) = NULL; 740c1a070e6SStefano Zampini hypre_CSRMatrixJ(hoffd) = NULL; 741c1a070e6SStefano Zampini hypre_CSRMatrixData(hoffd) = NULL; 742613e5ff0Sstefano_zampini hypre_ParCSRMatrixColMapOffd(*hA) = NULL; 743613e5ff0Sstefano_zampini hypre_ParCSRMatrixDestroy(*hA); 744613e5ff0Sstefano_zampini *hA = NULL; 745613e5ff0Sstefano_zampini PetscFunctionReturn(0); 746613e5ff0Sstefano_zampini } 747613e5ff0Sstefano_zampini 748613e5ff0Sstefano_zampini /* calls RAP from BoomerAMG: 7493dad0653Sstefano_zampini the resulting ParCSR will not own the column and row starts 7503dad0653Sstefano_zampini It looks like we don't need to have the diagonal entries 7513dad0653Sstefano_zampini ordered first in the rows of the diagonal part 7523dad0653Sstefano_zampini for boomerAMGBuildCoarseOperator to work */ 753a055b5aaSBarry Smith static PetscErrorCode MatHYPRE_ParCSR_RAP(hypre_ParCSRMatrix *hR, hypre_ParCSRMatrix *hA,hypre_ParCSRMatrix *hP, hypre_ParCSRMatrix **hRAP) 754613e5ff0Sstefano_zampini { 755613e5ff0Sstefano_zampini HYPRE_Int P_owns_col_starts,R_owns_row_starts; 756613e5ff0Sstefano_zampini 757613e5ff0Sstefano_zampini PetscFunctionBegin; 758613e5ff0Sstefano_zampini P_owns_col_starts = hypre_ParCSRMatrixOwnsColStarts(hP); 759613e5ff0Sstefano_zampini R_owns_row_starts = hypre_ParCSRMatrixOwnsRowStarts(hR); 760613e5ff0Sstefano_zampini PetscStackCallStandard(hypre_BoomerAMGBuildCoarseOperator,(hR,hA,hP,hRAP)); 761613e5ff0Sstefano_zampini PetscStackCallStandard(hypre_ParCSRMatrixSetNumNonzeros,(*hRAP)); 762613e5ff0Sstefano_zampini /* hypre_BoomerAMGBuildCoarseOperator steals the col_starts from P and the row_starts from R */ 763613e5ff0Sstefano_zampini hypre_ParCSRMatrixSetRowStartsOwner(*hRAP,0); 764613e5ff0Sstefano_zampini hypre_ParCSRMatrixSetColStartsOwner(*hRAP,0); 765613e5ff0Sstefano_zampini if (P_owns_col_starts) hypre_ParCSRMatrixSetColStartsOwner(hP,1); 766613e5ff0Sstefano_zampini if (R_owns_row_starts) hypre_ParCSRMatrixSetRowStartsOwner(hR,1); 767613e5ff0Sstefano_zampini PetscFunctionReturn(0); 768613e5ff0Sstefano_zampini } 769613e5ff0Sstefano_zampini 7706f231fbdSstefano_zampini static PetscErrorCode MatPtAPNumeric_AIJ_AIJ_wHYPRE(Mat A,Mat P,Mat C) 771613e5ff0Sstefano_zampini { 7726f231fbdSstefano_zampini Mat B; 773613e5ff0Sstefano_zampini hypre_ParCSRMatrix *hA,*hP,*hPtAP; 774613e5ff0Sstefano_zampini PetscErrorCode ierr; 775*4222ddf1SHong Zhang Mat_Product *product=C->product; 776613e5ff0Sstefano_zampini 777613e5ff0Sstefano_zampini PetscFunctionBegin; 778613e5ff0Sstefano_zampini ierr = MatAIJGetParCSR_Private(A,&hA);CHKERRQ(ierr); 779613e5ff0Sstefano_zampini ierr = MatAIJGetParCSR_Private(P,&hP);CHKERRQ(ierr); 780613e5ff0Sstefano_zampini ierr = MatHYPRE_ParCSR_RAP(hP,hA,hP,&hPtAP);CHKERRQ(ierr); 7816f231fbdSstefano_zampini ierr = MatCreateFromParCSR(hPtAP,MATAIJ,PETSC_OWN_POINTER,&B);CHKERRQ(ierr); 782*4222ddf1SHong Zhang 7836f231fbdSstefano_zampini ierr = MatHeaderMerge(C,&B);CHKERRQ(ierr); 784*4222ddf1SHong Zhang C->product = product; 785*4222ddf1SHong Zhang 786613e5ff0Sstefano_zampini ierr = MatAIJRestoreParCSR_Private(A,&hA);CHKERRQ(ierr); 787613e5ff0Sstefano_zampini ierr = MatAIJRestoreParCSR_Private(P,&hP);CHKERRQ(ierr); 7886f231fbdSstefano_zampini PetscFunctionReturn(0); 7896f231fbdSstefano_zampini } 7906f231fbdSstefano_zampini 791*4222ddf1SHong Zhang PETSC_INTERN PetscErrorCode MatPtAPSymbolic_AIJ_AIJ_wHYPRE(Mat A,Mat P,PetscReal fill,Mat C) 7926f231fbdSstefano_zampini { 7936f231fbdSstefano_zampini PetscErrorCode ierr; 794ab4d48faSStefano Zampini 7956f231fbdSstefano_zampini PetscFunctionBegin; 796*4222ddf1SHong Zhang ierr = MatSetType(C,MATAIJ);CHKERRQ(ierr); 797*4222ddf1SHong Zhang C->ops->ptapnumeric = MatPtAPNumeric_AIJ_AIJ_wHYPRE; 798*4222ddf1SHong Zhang C->ops->productnumeric = MatProductNumeric_PtAP; 799613e5ff0Sstefano_zampini PetscFunctionReturn(0); 800613e5ff0Sstefano_zampini } 801613e5ff0Sstefano_zampini 8024cc28894Sstefano_zampini static PetscErrorCode MatPtAPNumeric_AIJ_HYPRE(Mat A,Mat P,Mat C) 803613e5ff0Sstefano_zampini { 8044cc28894Sstefano_zampini Mat B; 8054cc28894Sstefano_zampini Mat_HYPRE *hP; 806613e5ff0Sstefano_zampini hypre_ParCSRMatrix *hA = NULL,*Pparcsr,*ptapparcsr; 807613e5ff0Sstefano_zampini HYPRE_Int type; 808613e5ff0Sstefano_zampini MPI_Comm comm = PetscObjectComm((PetscObject)A); 8094cc28894Sstefano_zampini PetscBool ishypre; 810613e5ff0Sstefano_zampini PetscErrorCode ierr; 811613e5ff0Sstefano_zampini 812613e5ff0Sstefano_zampini PetscFunctionBegin; 8134cc28894Sstefano_zampini ierr = PetscObjectTypeCompare((PetscObject)P,MATHYPRE,&ishypre);CHKERRQ(ierr); 8144cc28894Sstefano_zampini if (!ishypre) SETERRQ1(comm,PETSC_ERR_USER,"P should be of type %s",MATHYPRE); 8154cc28894Sstefano_zampini hP = (Mat_HYPRE*)P->data; 816613e5ff0Sstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hP->ij,&type)); 817613e5ff0Sstefano_zampini if (type != HYPRE_PARCSR) SETERRQ(comm,PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 818613e5ff0Sstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hP->ij,(void**)&Pparcsr)); 819613e5ff0Sstefano_zampini 820613e5ff0Sstefano_zampini ierr = MatAIJGetParCSR_Private(A,&hA);CHKERRQ(ierr); 821613e5ff0Sstefano_zampini ierr = MatHYPRE_ParCSR_RAP(Pparcsr,hA,Pparcsr,&ptapparcsr);CHKERRQ(ierr); 822613e5ff0Sstefano_zampini ierr = MatAIJRestoreParCSR_Private(A,&hA);CHKERRQ(ierr); 823225daaf8SStefano Zampini 8244cc28894Sstefano_zampini /* create temporary matrix and merge to C */ 8254cc28894Sstefano_zampini ierr = MatCreateFromParCSR(ptapparcsr,((PetscObject)C)->type_name,PETSC_OWN_POINTER,&B);CHKERRQ(ierr); 8264cc28894Sstefano_zampini ierr = MatHeaderMerge(C,&B);CHKERRQ(ierr); 8274cc28894Sstefano_zampini PetscFunctionReturn(0); 8284cc28894Sstefano_zampini } 8294cc28894Sstefano_zampini 8304cc28894Sstefano_zampini static PetscErrorCode MatPtAPNumeric_HYPRE_HYPRE(Mat A,Mat P,Mat C) 8314cc28894Sstefano_zampini { 8324cc28894Sstefano_zampini Mat B; 8334cc28894Sstefano_zampini hypre_ParCSRMatrix *Aparcsr,*Pparcsr,*ptapparcsr; 8344cc28894Sstefano_zampini Mat_HYPRE *hA,*hP; 8354cc28894Sstefano_zampini PetscBool ishypre; 8364cc28894Sstefano_zampini HYPRE_Int type; 8374cc28894Sstefano_zampini PetscErrorCode ierr; 8384cc28894Sstefano_zampini 8394cc28894Sstefano_zampini PetscFunctionBegin; 8404cc28894Sstefano_zampini ierr = PetscObjectTypeCompare((PetscObject)P,MATHYPRE,&ishypre);CHKERRQ(ierr); 8414cc28894Sstefano_zampini if (!ishypre) SETERRQ1(PetscObjectComm((PetscObject)P),PETSC_ERR_USER,"P should be of type %s",MATHYPRE); 8424cc28894Sstefano_zampini ierr = PetscObjectTypeCompare((PetscObject)A,MATHYPRE,&ishypre);CHKERRQ(ierr); 8434cc28894Sstefano_zampini if (!ishypre) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_USER,"A should be of type %s",MATHYPRE); 8444cc28894Sstefano_zampini hA = (Mat_HYPRE*)A->data; 8454cc28894Sstefano_zampini hP = (Mat_HYPRE*)P->data; 8464cc28894Sstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type)); 8474cc28894Sstefano_zampini if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 8484cc28894Sstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hP->ij,&type)); 8494cc28894Sstefano_zampini if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)P),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 8504cc28894Sstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)&Aparcsr)); 8514cc28894Sstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hP->ij,(void**)&Pparcsr)); 8524cc28894Sstefano_zampini ierr = MatHYPRE_ParCSR_RAP(Pparcsr,Aparcsr,Pparcsr,&ptapparcsr);CHKERRQ(ierr); 8534cc28894Sstefano_zampini ierr = MatCreateFromParCSR(ptapparcsr,MATHYPRE,PETSC_OWN_POINTER,&B);CHKERRQ(ierr); 8544cc28894Sstefano_zampini ierr = MatHeaderMerge(C,&B);CHKERRQ(ierr); 8554cc28894Sstefano_zampini PetscFunctionReturn(0); 8564cc28894Sstefano_zampini } 8574cc28894Sstefano_zampini 858d501dc42Sstefano_zampini /* calls hypre_ParMatmul 859d501dc42Sstefano_zampini hypre_ParMatMul uses hypre_ParMatrixCreate with the communicator of hA 8603dad0653Sstefano_zampini hypre_ParMatrixCreate does not duplicate the communicator 8613dad0653Sstefano_zampini It looks like we don't need to have the diagonal entries 8623dad0653Sstefano_zampini ordered first in the rows of the diagonal part 8633dad0653Sstefano_zampini for boomerAMGBuildCoarseOperator to work */ 864d501dc42Sstefano_zampini static PetscErrorCode MatHYPRE_ParCSR_MatMatMult(hypre_ParCSRMatrix *hA, hypre_ParCSRMatrix *hB, hypre_ParCSRMatrix **hAB) 865d501dc42Sstefano_zampini { 866d501dc42Sstefano_zampini PetscFunctionBegin; 867d501dc42Sstefano_zampini PetscStackPush("hypre_ParMatmul"); 868d501dc42Sstefano_zampini *hAB = hypre_ParMatmul(hA,hB); 869d501dc42Sstefano_zampini PetscStackPop; 870d501dc42Sstefano_zampini PetscFunctionReturn(0); 871d501dc42Sstefano_zampini } 872d501dc42Sstefano_zampini 8735e5acdf2Sstefano_zampini static PetscErrorCode MatMatMultNumeric_AIJ_AIJ_wHYPRE(Mat A,Mat B,Mat C) 8745e5acdf2Sstefano_zampini { 8755e5acdf2Sstefano_zampini Mat D; 876d501dc42Sstefano_zampini hypre_ParCSRMatrix *hA,*hB,*hAB = NULL; 8775e5acdf2Sstefano_zampini PetscErrorCode ierr; 878*4222ddf1SHong Zhang Mat_Product *product=C->product; 8795e5acdf2Sstefano_zampini 8805e5acdf2Sstefano_zampini PetscFunctionBegin; 8815e5acdf2Sstefano_zampini ierr = MatAIJGetParCSR_Private(A,&hA);CHKERRQ(ierr); 8825e5acdf2Sstefano_zampini ierr = MatAIJGetParCSR_Private(B,&hB);CHKERRQ(ierr); 883d501dc42Sstefano_zampini ierr = MatHYPRE_ParCSR_MatMatMult(hA,hB,&hAB);CHKERRQ(ierr); 8845e5acdf2Sstefano_zampini ierr = MatCreateFromParCSR(hAB,MATAIJ,PETSC_OWN_POINTER,&D);CHKERRQ(ierr); 885*4222ddf1SHong Zhang 8865e5acdf2Sstefano_zampini ierr = MatHeaderMerge(C,&D);CHKERRQ(ierr); 887*4222ddf1SHong Zhang C->product = product; 888*4222ddf1SHong Zhang 8895e5acdf2Sstefano_zampini ierr = MatAIJRestoreParCSR_Private(A,&hA);CHKERRQ(ierr); 8905e5acdf2Sstefano_zampini ierr = MatAIJRestoreParCSR_Private(B,&hB);CHKERRQ(ierr); 8915e5acdf2Sstefano_zampini PetscFunctionReturn(0); 8925e5acdf2Sstefano_zampini } 8935e5acdf2Sstefano_zampini 894*4222ddf1SHong Zhang PETSC_INTERN PetscErrorCode MatMatMultSymbolic_AIJ_AIJ_wHYPRE(Mat A,Mat B,PetscReal fill,Mat C) 8955e5acdf2Sstefano_zampini { 8965e5acdf2Sstefano_zampini PetscErrorCode ierr; 89720e1dc0dSstefano_zampini 8985e5acdf2Sstefano_zampini PetscFunctionBegin; 899*4222ddf1SHong Zhang ierr = MatSetType(C,MATAIJ);CHKERRQ(ierr); 900*4222ddf1SHong Zhang C->ops->matmultnumeric = MatMatMultNumeric_AIJ_AIJ_wHYPRE; 901*4222ddf1SHong Zhang C->ops->productnumeric = MatProductNumeric_AB; 9025e5acdf2Sstefano_zampini PetscFunctionReturn(0); 9035e5acdf2Sstefano_zampini } 9045e5acdf2Sstefano_zampini 905d501dc42Sstefano_zampini static PetscErrorCode MatMatMultNumeric_HYPRE_HYPRE(Mat A,Mat B,Mat C) 906d501dc42Sstefano_zampini { 907d501dc42Sstefano_zampini Mat D; 908d501dc42Sstefano_zampini hypre_ParCSRMatrix *Aparcsr,*Bparcsr,*ABparcsr = NULL; 909d501dc42Sstefano_zampini Mat_HYPRE *hA,*hB; 910d501dc42Sstefano_zampini PetscBool ishypre; 911d501dc42Sstefano_zampini HYPRE_Int type; 912d501dc42Sstefano_zampini PetscErrorCode ierr; 913*4222ddf1SHong Zhang Mat_Product *product; 914d501dc42Sstefano_zampini 915d501dc42Sstefano_zampini PetscFunctionBegin; 916d501dc42Sstefano_zampini ierr = PetscObjectTypeCompare((PetscObject)B,MATHYPRE,&ishypre);CHKERRQ(ierr); 917d501dc42Sstefano_zampini if (!ishypre) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_USER,"B should be of type %s",MATHYPRE); 918d501dc42Sstefano_zampini ierr = PetscObjectTypeCompare((PetscObject)A,MATHYPRE,&ishypre);CHKERRQ(ierr); 919d501dc42Sstefano_zampini if (!ishypre) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_USER,"A should be of type %s",MATHYPRE); 920d501dc42Sstefano_zampini hA = (Mat_HYPRE*)A->data; 921d501dc42Sstefano_zampini hB = (Mat_HYPRE*)B->data; 922d501dc42Sstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type)); 923d501dc42Sstefano_zampini if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 924d501dc42Sstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hB->ij,&type)); 925d501dc42Sstefano_zampini if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)B),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 926d501dc42Sstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)&Aparcsr)); 927d501dc42Sstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hB->ij,(void**)&Bparcsr)); 928d501dc42Sstefano_zampini ierr = MatHYPRE_ParCSR_MatMatMult(Aparcsr,Bparcsr,&ABparcsr);CHKERRQ(ierr); 929d501dc42Sstefano_zampini ierr = MatCreateFromParCSR(ABparcsr,MATHYPRE,PETSC_OWN_POINTER,&D);CHKERRQ(ierr); 930*4222ddf1SHong Zhang 931d501dc42Sstefano_zampini /* need to use HeaderReplace because HeaderMerge messes up with the communicator */ 932*4222ddf1SHong Zhang product = C->product; /* save it from MatHeaderReplace() */ 933*4222ddf1SHong Zhang C->product = NULL; 934d501dc42Sstefano_zampini ierr = MatHeaderReplace(C,&D);CHKERRQ(ierr); 935*4222ddf1SHong Zhang C->product = product; 936d501dc42Sstefano_zampini C->ops->matmultnumeric = MatMatMultNumeric_HYPRE_HYPRE; 937*4222ddf1SHong Zhang C->ops->productnumeric = MatProductNumeric_AB; 938d501dc42Sstefano_zampini PetscFunctionReturn(0); 939d501dc42Sstefano_zampini } 940d501dc42Sstefano_zampini 9413dad0653Sstefano_zampini PETSC_INTERN PetscErrorCode MatTransposeMatMatMultNumeric_AIJ_AIJ_AIJ_wHYPRE(Mat A,Mat B,Mat C,Mat D) 94220e1dc0dSstefano_zampini { 94320e1dc0dSstefano_zampini Mat E; 94420e1dc0dSstefano_zampini hypre_ParCSRMatrix *hA,*hB,*hC,*hABC; 94520e1dc0dSstefano_zampini PetscErrorCode ierr; 94620e1dc0dSstefano_zampini 94720e1dc0dSstefano_zampini PetscFunctionBegin; 94820e1dc0dSstefano_zampini ierr = MatAIJGetParCSR_Private(A,&hA);CHKERRQ(ierr); 94920e1dc0dSstefano_zampini ierr = MatAIJGetParCSR_Private(B,&hB);CHKERRQ(ierr); 95020e1dc0dSstefano_zampini ierr = MatAIJGetParCSR_Private(C,&hC);CHKERRQ(ierr); 95120e1dc0dSstefano_zampini ierr = MatHYPRE_ParCSR_RAP(hA,hB,hC,&hABC);CHKERRQ(ierr); 95220e1dc0dSstefano_zampini ierr = MatCreateFromParCSR(hABC,MATAIJ,PETSC_OWN_POINTER,&E);CHKERRQ(ierr); 95320e1dc0dSstefano_zampini ierr = MatHeaderMerge(D,&E);CHKERRQ(ierr); 95420e1dc0dSstefano_zampini ierr = MatAIJRestoreParCSR_Private(A,&hA);CHKERRQ(ierr); 95520e1dc0dSstefano_zampini ierr = MatAIJRestoreParCSR_Private(B,&hB);CHKERRQ(ierr); 95620e1dc0dSstefano_zampini ierr = MatAIJRestoreParCSR_Private(C,&hC);CHKERRQ(ierr); 95720e1dc0dSstefano_zampini PetscFunctionReturn(0); 95820e1dc0dSstefano_zampini } 95920e1dc0dSstefano_zampini 960*4222ddf1SHong Zhang PETSC_INTERN PetscErrorCode MatTransposeMatMatMultSymbolic_AIJ_AIJ_AIJ_wHYPRE(Mat A,Mat B,Mat C,PetscReal fill,Mat D) 96120e1dc0dSstefano_zampini { 96220e1dc0dSstefano_zampini PetscErrorCode ierr; 96320e1dc0dSstefano_zampini 96420e1dc0dSstefano_zampini PetscFunctionBegin; 965*4222ddf1SHong Zhang ierr = MatSetType(D,MATAIJ);CHKERRQ(ierr); 96620e1dc0dSstefano_zampini PetscFunctionReturn(0); 96720e1dc0dSstefano_zampini } 96820e1dc0dSstefano_zampini 969*4222ddf1SHong Zhang /* ---------------------------------------------------- */ 970*4222ddf1SHong Zhang static PetscErrorCode MatProductSymbolic_AB_HYPRE(Mat C) 971*4222ddf1SHong Zhang { 972*4222ddf1SHong Zhang PetscFunctionBegin; 973*4222ddf1SHong Zhang C->ops->productnumeric = MatProductNumeric_AB; 974*4222ddf1SHong Zhang PetscFunctionReturn(0); 975*4222ddf1SHong Zhang } 976*4222ddf1SHong Zhang 977*4222ddf1SHong Zhang static PetscErrorCode MatProductSetFromOptions_HYPRE_AB(Mat C) 978*4222ddf1SHong Zhang { 979*4222ddf1SHong Zhang PetscErrorCode ierr; 980*4222ddf1SHong Zhang Mat_Product *product = C->product; 981*4222ddf1SHong Zhang PetscBool Ahypre; 982*4222ddf1SHong Zhang 983*4222ddf1SHong Zhang PetscFunctionBegin; 984*4222ddf1SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)product->A,MATHYPRE,&Ahypre);CHKERRQ(ierr); 985*4222ddf1SHong Zhang if (Ahypre) { /* A is a Hypre matrix */ 986*4222ddf1SHong Zhang ierr = MatSetType(C,MATHYPRE);CHKERRQ(ierr); 987*4222ddf1SHong Zhang C->ops->productsymbolic = MatProductSymbolic_AB_HYPRE; 988*4222ddf1SHong Zhang C->ops->matmultnumeric = MatMatMultNumeric_HYPRE_HYPRE; 989*4222ddf1SHong Zhang PetscFunctionReturn(0); 990*4222ddf1SHong Zhang } else SETERRQ(PetscObjectComm((PetscObject)C),PETSC_ERR_SUP,"A must be Hyper type"); 991*4222ddf1SHong Zhang PetscFunctionReturn(0); 992*4222ddf1SHong Zhang } 993*4222ddf1SHong Zhang 994*4222ddf1SHong Zhang static PetscErrorCode MatProductSymbolic_PtAP_HYPRE(Mat C) 995*4222ddf1SHong Zhang { 996*4222ddf1SHong Zhang PetscFunctionBegin; 997*4222ddf1SHong Zhang C->ops->productnumeric = MatProductNumeric_PtAP; 998*4222ddf1SHong Zhang PetscFunctionReturn(0); 999*4222ddf1SHong Zhang } 1000*4222ddf1SHong Zhang 1001*4222ddf1SHong Zhang static PetscErrorCode MatProductSetFromOptions_HYPRE_PtAP(Mat C) 1002*4222ddf1SHong Zhang { 1003*4222ddf1SHong Zhang PetscErrorCode ierr; 1004*4222ddf1SHong Zhang Mat_Product *product = C->product; 1005*4222ddf1SHong Zhang PetscBool flg; 1006*4222ddf1SHong Zhang PetscInt type = 0; 1007*4222ddf1SHong Zhang const char *outTypes[4] = {"aij","seqaij","mpiaij","hypre"}; 1008*4222ddf1SHong Zhang PetscInt ntype = 4; 1009*4222ddf1SHong Zhang Mat A = product->A; 1010*4222ddf1SHong Zhang PetscBool Ahypre; 1011*4222ddf1SHong Zhang 1012*4222ddf1SHong Zhang PetscFunctionBegin; 1013*4222ddf1SHong Zhang ierr = PetscObjectTypeCompare((PetscObject)A,MATHYPRE,&Ahypre);CHKERRQ(ierr); 1014*4222ddf1SHong Zhang if (Ahypre) { /* A is a Hypre matrix */ 1015*4222ddf1SHong Zhang ierr = MatSetType(C,MATHYPRE);CHKERRQ(ierr); 1016*4222ddf1SHong Zhang C->ops->productsymbolic = MatProductSymbolic_PtAP_HYPRE; 1017*4222ddf1SHong Zhang C->ops->ptapnumeric = MatPtAPNumeric_HYPRE_HYPRE; 1018*4222ddf1SHong Zhang PetscFunctionReturn(0); 1019*4222ddf1SHong Zhang } 1020*4222ddf1SHong Zhang 1021*4222ddf1SHong Zhang /* A is AIJ, P is Hypre, C = PtAP can be either AIJ or Hypre format */ 1022*4222ddf1SHong Zhang /* Get runtime option */ 1023*4222ddf1SHong Zhang if (product->api_user) { 1024*4222ddf1SHong Zhang ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)C),((PetscObject)C)->prefix,"MatPtAP_HYPRE","Mat");CHKERRQ(ierr); 1025*4222ddf1SHong Zhang ierr = PetscOptionsEList("-matptap_hypre_outtype","MatPtAP outtype","MatPtAP outtype",outTypes,ntype,outTypes[type],&type,&flg);CHKERRQ(ierr); 1026*4222ddf1SHong Zhang ierr = PetscOptionsEnd();CHKERRQ(ierr); 1027*4222ddf1SHong Zhang } else { 1028*4222ddf1SHong Zhang ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)C),((PetscObject)C)->prefix,"MatProduct_PtAP_HYPRE","Mat");CHKERRQ(ierr); 1029*4222ddf1SHong Zhang ierr = PetscOptionsEList("-matproduct_ptap_hypre_outtype","MatProduct_PtAP outtype","MatProduct_PtAP",outTypes,ntype,outTypes[type],&type,&flg);CHKERRQ(ierr); 1030*4222ddf1SHong Zhang ierr = PetscOptionsEnd();CHKERRQ(ierr); 1031*4222ddf1SHong Zhang } 1032*4222ddf1SHong Zhang 1033*4222ddf1SHong Zhang if (type == 0 || type == 1 || type == 2) { 1034*4222ddf1SHong Zhang ierr = MatSetType(C,MATAIJ);CHKERRQ(ierr); 1035*4222ddf1SHong Zhang } else if (type == 3) { 1036*4222ddf1SHong Zhang ierr = MatSetType(C,MATHYPRE);CHKERRQ(ierr); 1037*4222ddf1SHong Zhang } else SETERRQ(PetscObjectComm((PetscObject)C),PETSC_ERR_SUP,"MatPtAP outtype is not supported"); 1038*4222ddf1SHong Zhang C->ops->productsymbolic = MatProductSymbolic_PtAP_HYPRE; 1039*4222ddf1SHong Zhang C->ops->ptapnumeric = MatPtAPNumeric_AIJ_HYPRE; 1040*4222ddf1SHong Zhang PetscFunctionReturn(0); 1041*4222ddf1SHong Zhang } 1042*4222ddf1SHong Zhang 1043*4222ddf1SHong Zhang static PetscErrorCode MatProductSetFromOptions_HYPRE(Mat C) 1044*4222ddf1SHong Zhang { 1045*4222ddf1SHong Zhang PetscErrorCode ierr; 1046*4222ddf1SHong Zhang Mat_Product *product = C->product; 1047*4222ddf1SHong Zhang 1048*4222ddf1SHong Zhang PetscFunctionBegin; 1049*4222ddf1SHong Zhang switch (product->type) { 1050*4222ddf1SHong Zhang case MATPRODUCT_AB: 1051*4222ddf1SHong Zhang ierr = MatProductSetFromOptions_HYPRE_AB(C);CHKERRQ(ierr); 1052*4222ddf1SHong Zhang break; 1053*4222ddf1SHong Zhang case MATPRODUCT_PtAP: 1054*4222ddf1SHong Zhang ierr = MatProductSetFromOptions_HYPRE_PtAP(C);CHKERRQ(ierr); 1055*4222ddf1SHong Zhang break; 1056*4222ddf1SHong Zhang default: SETERRQ(PetscObjectComm((PetscObject)C),PETSC_ERR_SUP,"Mat Product type is not supported"); 1057*4222ddf1SHong Zhang } 1058*4222ddf1SHong Zhang PetscFunctionReturn(0); 1059*4222ddf1SHong Zhang } 1060*4222ddf1SHong Zhang 1061*4222ddf1SHong Zhang /* -------------------------------------------------------- */ 1062*4222ddf1SHong Zhang 1063ea9daf28SStefano Zampini static PetscErrorCode MatMultTranspose_HYPRE(Mat A, Vec x, Vec y) 106463c07aadSStefano Zampini { 106563c07aadSStefano Zampini PetscErrorCode ierr; 106663c07aadSStefano Zampini 106763c07aadSStefano Zampini PetscFunctionBegin; 1068414bd5c3SStefano Zampini ierr = MatHYPRE_MultKernel_Private(A,1.0,x,0.0,y,PETSC_TRUE);CHKERRQ(ierr); 106963c07aadSStefano Zampini PetscFunctionReturn(0); 107063c07aadSStefano Zampini } 107163c07aadSStefano Zampini 1072ea9daf28SStefano Zampini static PetscErrorCode MatMult_HYPRE(Mat A, Vec x, Vec y) 107363c07aadSStefano Zampini { 107463c07aadSStefano Zampini PetscErrorCode ierr; 107563c07aadSStefano Zampini 107663c07aadSStefano Zampini PetscFunctionBegin; 1077414bd5c3SStefano Zampini ierr = MatHYPRE_MultKernel_Private(A,1.0,x,0.0,y,PETSC_FALSE);CHKERRQ(ierr); 107863c07aadSStefano Zampini PetscFunctionReturn(0); 107963c07aadSStefano Zampini } 108063c07aadSStefano Zampini 1081414bd5c3SStefano Zampini static PetscErrorCode MatMultAdd_HYPRE(Mat A, Vec x, Vec y, Vec z) 1082414bd5c3SStefano Zampini { 1083414bd5c3SStefano Zampini PetscErrorCode ierr; 1084414bd5c3SStefano Zampini 1085414bd5c3SStefano Zampini PetscFunctionBegin; 1086414bd5c3SStefano Zampini if (y != z) { 1087414bd5c3SStefano Zampini ierr = VecCopy(y,z);CHKERRQ(ierr); 1088414bd5c3SStefano Zampini } 1089414bd5c3SStefano Zampini ierr = MatHYPRE_MultKernel_Private(A,1.0,x,1.0,z,PETSC_FALSE);CHKERRQ(ierr); 1090414bd5c3SStefano Zampini PetscFunctionReturn(0); 1091414bd5c3SStefano Zampini } 1092414bd5c3SStefano Zampini 1093414bd5c3SStefano Zampini static PetscErrorCode MatMultTransposeAdd_HYPRE(Mat A, Vec x, Vec y, Vec z) 1094414bd5c3SStefano Zampini { 1095414bd5c3SStefano Zampini PetscErrorCode ierr; 1096414bd5c3SStefano Zampini 1097414bd5c3SStefano Zampini PetscFunctionBegin; 1098414bd5c3SStefano Zampini if (y != z) { 1099414bd5c3SStefano Zampini ierr = VecCopy(y,z);CHKERRQ(ierr); 1100414bd5c3SStefano Zampini } 1101414bd5c3SStefano Zampini ierr = MatHYPRE_MultKernel_Private(A,1.0,x,1.0,z,PETSC_TRUE);CHKERRQ(ierr); 1102414bd5c3SStefano Zampini PetscFunctionReturn(0); 1103414bd5c3SStefano Zampini } 1104414bd5c3SStefano Zampini 1105414bd5c3SStefano Zampini /* y = a * A * x + b * y or y = a * A^t * x + b * y depending on trans */ 110639accc25SStefano Zampini static PetscErrorCode MatHYPRE_MultKernel_Private(Mat A, HYPRE_Complex a, Vec x, HYPRE_Complex b, Vec y, PetscBool trans) 110763c07aadSStefano Zampini { 110863c07aadSStefano Zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 110963c07aadSStefano Zampini hypre_ParCSRMatrix *parcsr; 111063c07aadSStefano Zampini hypre_ParVector *hx,*hy; 111139accc25SStefano Zampini HYPRE_Complex *ax,*ay,*sax,*say; 111263c07aadSStefano Zampini PetscErrorCode ierr; 111363c07aadSStefano Zampini 111463c07aadSStefano Zampini PetscFunctionBegin; 111563c07aadSStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)&parcsr)); 111663c07aadSStefano Zampini PetscStackCallStandard(HYPRE_IJVectorGetObject,(hA->x,(void**)&hx)); 111763c07aadSStefano Zampini PetscStackCallStandard(HYPRE_IJVectorGetObject,(hA->b,(void**)&hy)); 111863c07aadSStefano Zampini ierr = VecGetArrayRead(x,(const PetscScalar**)&ax);CHKERRQ(ierr); 111939accc25SStefano Zampini ierr = VecGetArray(y,(PetscScalar**)&ay);CHKERRQ(ierr); 112063c07aadSStefano Zampini if (trans) { 112158968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hA->x,ay,say); 112258968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hA->b,ax,sax); 1123414bd5c3SStefano Zampini hypre_ParCSRMatrixMatvecT(a,parcsr,hy,b,hx); 112458968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hA->x,say,ay); 112558968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hA->b,sax,ax); 112663c07aadSStefano Zampini } else { 112758968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hA->x,ax,sax); 112858968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hA->b,ay,say); 1129414bd5c3SStefano Zampini hypre_ParCSRMatrixMatvec(a,parcsr,hx,b,hy); 113058968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hA->x,sax,ax); 113158968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hA->b,say,ay); 113263c07aadSStefano Zampini } 113363c07aadSStefano Zampini ierr = VecRestoreArrayRead(x,(const PetscScalar**)&ax);CHKERRQ(ierr); 113439accc25SStefano Zampini ierr = VecRestoreArray(y,(PetscScalar**)&ay);CHKERRQ(ierr); 113563c07aadSStefano Zampini PetscFunctionReturn(0); 113663c07aadSStefano Zampini } 113763c07aadSStefano Zampini 1138ea9daf28SStefano Zampini static PetscErrorCode MatDestroy_HYPRE(Mat A) 113963c07aadSStefano Zampini { 114063c07aadSStefano Zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 114163c07aadSStefano Zampini PetscErrorCode ierr; 114263c07aadSStefano Zampini 114363c07aadSStefano Zampini PetscFunctionBegin; 114463c07aadSStefano Zampini if (hA->x) PetscStackCallStandard(HYPRE_IJVectorDestroy,(hA->x)); 114563c07aadSStefano Zampini if (hA->b) PetscStackCallStandard(HYPRE_IJVectorDestroy,(hA->b)); 1146978814f1SStefano Zampini if (hA->ij) { 1147978814f1SStefano Zampini if (!hA->inner_free) hypre_IJMatrixObject(hA->ij) = NULL; 1148978814f1SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixDestroy,(hA->ij)); 1149978814f1SStefano Zampini } 115063c07aadSStefano Zampini if (hA->comm) { ierr = MPI_Comm_free(&hA->comm);CHKERRQ(ierr); } 1151c69f721fSFande Kong 1152c69f721fSFande Kong ierr = MatStashDestroy_Private(&A->stash);CHKERRQ(ierr); 1153c69f721fSFande Kong 1154c69f721fSFande Kong ierr = PetscFree(hA->array);CHKERRQ(ierr); 1155c69f721fSFande Kong 115663c07aadSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_hypre_aij_C",NULL);CHKERRQ(ierr); 11572df22349SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_hypre_is_C",NULL);CHKERRQ(ierr); 1158*4222ddf1SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)A,"MatProductSetFromOptions_seqaij_hypre_C",NULL);CHKERRQ(ierr); 1159*4222ddf1SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)A,"MatProductSetFromOptions_mpiaij_hypre_C",NULL);CHKERRQ(ierr); 1160d975228cSstefano_zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatHYPRESetPreallocation_C",NULL);CHKERRQ(ierr); 1161dd9c0a25Sstefano_zampini ierr = PetscObjectComposeFunction((PetscObject)A,"MatHYPREGetParCSR_C",NULL);CHKERRQ(ierr); 116263c07aadSStefano Zampini ierr = PetscFree(A->data);CHKERRQ(ierr); 116363c07aadSStefano Zampini PetscFunctionReturn(0); 116463c07aadSStefano Zampini } 116563c07aadSStefano Zampini 1166ea9daf28SStefano Zampini static PetscErrorCode MatSetUp_HYPRE(Mat A) 116763c07aadSStefano Zampini { 11684ec6421dSstefano_zampini PetscErrorCode ierr; 11694ec6421dSstefano_zampini 11704ec6421dSstefano_zampini PetscFunctionBegin; 11714ec6421dSstefano_zampini ierr = MatHYPRESetPreallocation(A,PETSC_DEFAULT,NULL,PETSC_DEFAULT,NULL);CHKERRQ(ierr); 11724ec6421dSstefano_zampini PetscFunctionReturn(0); 11734ec6421dSstefano_zampini } 11744ec6421dSstefano_zampini 11754ec6421dSstefano_zampini static PetscErrorCode MatAssemblyEnd_HYPRE(Mat A, MatAssemblyType mode) 11764ec6421dSstefano_zampini { 117763c07aadSStefano Zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 117863c07aadSStefano Zampini Vec x,b; 1179c69f721fSFande Kong PetscMPIInt n; 1180c69f721fSFande Kong PetscInt i,j,rstart,ncols,flg; 1181c69f721fSFande Kong PetscInt *row,*col; 1182c69f721fSFande Kong PetscScalar *val; 118363c07aadSStefano Zampini PetscErrorCode ierr; 118463c07aadSStefano Zampini 118563c07aadSStefano Zampini PetscFunctionBegin; 11864ec6421dSstefano_zampini if (mode == MAT_FLUSH_ASSEMBLY) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MAT_FLUSH_ASSEMBLY currently not supported with MATHYPRE"); 1187c69f721fSFande Kong 1188c69f721fSFande Kong if (!A->nooffprocentries) { 1189c69f721fSFande Kong while (1) { 1190c69f721fSFande Kong ierr = MatStashScatterGetMesg_Private(&A->stash,&n,&row,&col,&val,&flg);CHKERRQ(ierr); 1191c69f721fSFande Kong if (!flg) break; 1192c69f721fSFande Kong 1193c69f721fSFande Kong for (i=0; i<n; ) { 1194c69f721fSFande Kong /* Now identify the consecutive vals belonging to the same row */ 1195c69f721fSFande Kong for (j=i,rstart=row[j]; j<n; j++) { 1196c69f721fSFande Kong if (row[j] != rstart) break; 1197c69f721fSFande Kong } 1198c69f721fSFande Kong if (j < n) ncols = j-i; 1199c69f721fSFande Kong else ncols = n-i; 1200c69f721fSFande Kong /* Now assemble all these values with a single function call */ 1201c69f721fSFande Kong ierr = MatSetValues_HYPRE(A,1,row+i,ncols,col+i,val+i,A->insertmode);CHKERRQ(ierr); 1202c69f721fSFande Kong 1203c69f721fSFande Kong i = j; 1204c69f721fSFande Kong } 1205c69f721fSFande Kong } 1206c69f721fSFande Kong ierr = MatStashScatterEnd_Private(&A->stash);CHKERRQ(ierr); 1207c69f721fSFande Kong } 1208c69f721fSFande Kong 12094ec6421dSstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixAssemble,(hA->ij)); 1210af1cf968SStefano Zampini /* The assembly routine destroys the aux_matrix, we recreate it here by calling HYPRE_IJMatrixInitalize */ 1211af1cf968SStefano Zampini { 1212af1cf968SStefano Zampini hypre_AuxParCSRMatrix *aux_matrix; 1213af1cf968SStefano Zampini 1214af1cf968SStefano Zampini /* call destroy just to make sure we do not leak anything */ 1215af1cf968SStefano Zampini aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(hA->ij); 1216af1cf968SStefano Zampini PetscStackCallStandard(hypre_AuxParCSRMatrixDestroy,(aux_matrix)); 1217af1cf968SStefano Zampini hypre_IJMatrixTranslator(hA->ij) = NULL; 1218af1cf968SStefano Zampini 1219af1cf968SStefano Zampini /* Initialize with assembled flag -> it only recreates the aux_par_matrix */ 1220af1cf968SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixInitialize,(hA->ij)); 1221af1cf968SStefano Zampini aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(hA->ij); 1222af1cf968SStefano Zampini hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 1; /* see comment in MatHYPRESetPreallocation_HYPRE */ 1223af1cf968SStefano Zampini PetscStackCallStandard(hypre_AuxParCSRMatrixInitialize,(aux_matrix)); 1224af1cf968SStefano Zampini } 122563c07aadSStefano Zampini if (hA->x) PetscFunctionReturn(0); 122663c07aadSStefano Zampini ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 122763c07aadSStefano Zampini ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 122863c07aadSStefano Zampini ierr = VecCreateMPIWithArray(PetscObjectComm((PetscObject)A),1,A->cmap->n,A->cmap->N,NULL,&x);CHKERRQ(ierr); 122963c07aadSStefano Zampini ierr = VecCreateMPIWithArray(PetscObjectComm((PetscObject)A),1,A->rmap->n,A->rmap->N,NULL,&b);CHKERRQ(ierr); 123063c07aadSStefano Zampini ierr = VecHYPRE_IJVectorCreate(x,&hA->x);CHKERRQ(ierr); 123163c07aadSStefano Zampini ierr = VecHYPRE_IJVectorCreate(b,&hA->b);CHKERRQ(ierr); 123263c07aadSStefano Zampini ierr = VecDestroy(&x);CHKERRQ(ierr); 123363c07aadSStefano Zampini ierr = VecDestroy(&b);CHKERRQ(ierr); 123463c07aadSStefano Zampini PetscFunctionReturn(0); 123563c07aadSStefano Zampini } 123663c07aadSStefano Zampini 1237c69f721fSFande Kong static PetscErrorCode MatGetArray_HYPRE(Mat A, PetscInt size, void **array) 1238c69f721fSFande Kong { 1239c69f721fSFande Kong Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 1240c69f721fSFande Kong PetscErrorCode ierr; 1241c69f721fSFande Kong 1242c69f721fSFande Kong PetscFunctionBegin; 1243c69f721fSFande Kong if (!hA->available) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_NULL,"Temporary space is in use"); 1244c69f721fSFande Kong 124539accc25SStefano Zampini if (hA->size >= size) { 124639accc25SStefano Zampini *array = hA->array; 124739accc25SStefano Zampini } else { 1248c69f721fSFande Kong ierr = PetscFree(hA->array);CHKERRQ(ierr); 1249c69f721fSFande Kong hA->size = size; 1250c69f721fSFande Kong ierr = PetscMalloc(hA->size,&hA->array);CHKERRQ(ierr); 1251c69f721fSFande Kong *array = hA->array; 1252c69f721fSFande Kong } 1253c69f721fSFande Kong 1254c69f721fSFande Kong hA->available = PETSC_FALSE; 1255c69f721fSFande Kong PetscFunctionReturn(0); 1256c69f721fSFande Kong } 1257c69f721fSFande Kong 1258708542d2SFande Kong static PetscErrorCode MatRestoreArray_HYPRE(Mat A, void **array) 1259c69f721fSFande Kong { 1260c69f721fSFande Kong Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 1261c69f721fSFande Kong 1262c69f721fSFande Kong PetscFunctionBegin; 1263c69f721fSFande Kong *array = NULL; 1264c69f721fSFande Kong hA->available = PETSC_TRUE; 1265c69f721fSFande Kong PetscFunctionReturn(0); 1266c69f721fSFande Kong } 1267c69f721fSFande Kong 1268d975228cSstefano_zampini PetscErrorCode MatSetValues_HYPRE(Mat A, PetscInt nr, const PetscInt rows[], PetscInt nc, const PetscInt cols[], const PetscScalar v[], InsertMode ins) 1269d975228cSstefano_zampini { 1270d975228cSstefano_zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 1271d975228cSstefano_zampini PetscScalar *vals = (PetscScalar *)v; 127239accc25SStefano Zampini HYPRE_Complex *sscr; 1273c69f721fSFande Kong PetscInt *cscr[2]; 1274c69f721fSFande Kong PetscInt i,nzc; 127508defe43SFande Kong void *array = NULL; 1276d975228cSstefano_zampini PetscErrorCode ierr; 1277d975228cSstefano_zampini 1278d975228cSstefano_zampini PetscFunctionBegin; 127939accc25SStefano Zampini ierr = MatGetArray_HYPRE(A,sizeof(PetscInt)*(2*nc)+sizeof(HYPRE_Complex)*nc*nr,&array);CHKERRQ(ierr); 1280c69f721fSFande Kong cscr[0] = (PetscInt*)array; 1281c69f721fSFande Kong cscr[1] = ((PetscInt*)array)+nc; 128239accc25SStefano Zampini sscr = (HYPRE_Complex*)(((PetscInt*)array)+nc*2); 1283d975228cSstefano_zampini for (i=0,nzc=0;i<nc;i++) { 1284d975228cSstefano_zampini if (cols[i] >= 0) { 1285d975228cSstefano_zampini cscr[0][nzc ] = cols[i]; 1286d975228cSstefano_zampini cscr[1][nzc++] = i; 1287d975228cSstefano_zampini } 1288d975228cSstefano_zampini } 1289c69f721fSFande Kong if (!nzc) { 1290708542d2SFande Kong ierr = MatRestoreArray_HYPRE(A,&array);CHKERRQ(ierr); 1291c69f721fSFande Kong PetscFunctionReturn(0); 1292c69f721fSFande Kong } 1293d975228cSstefano_zampini 1294d975228cSstefano_zampini if (ins == ADD_VALUES) { 1295d975228cSstefano_zampini for (i=0;i<nr;i++) { 1296e3977e59Sstefano_zampini if (rows[i] >= 0 && nzc) { 1297d975228cSstefano_zampini PetscInt j; 12982cf14000SStefano Zampini HYPRE_Int hnc = (HYPRE_Int)nzc; 12992cf14000SStefano Zampini 13002cf14000SStefano Zampini if ((PetscInt)hnc != nzc) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hypre overflow! number of columns %D for row %D",nzc,rows[i]); 130139accc25SStefano Zampini for (j=0;j<nzc;j++) { ierr = PetscHYPREScalarCast(vals[cscr[1][j]],&sscr[j]); } 13022cf14000SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixAddToValues,(hA->ij,1,&hnc,(HYPRE_BigInt*)(rows+i),(HYPRE_BigInt*)cscr[0],sscr)); 1303d975228cSstefano_zampini } 1304d975228cSstefano_zampini vals += nc; 1305d975228cSstefano_zampini } 1306d975228cSstefano_zampini } else { /* INSERT_VALUES */ 1307d975228cSstefano_zampini PetscInt rst,ren; 1308c69f721fSFande Kong 1309c6698e78SStefano Zampini ierr = MatGetOwnershipRange(A,&rst,&ren);CHKERRQ(ierr); 1310d975228cSstefano_zampini for (i=0;i<nr;i++) { 1311e3977e59Sstefano_zampini if (rows[i] >= 0 && nzc) { 1312d975228cSstefano_zampini PetscInt j; 13132cf14000SStefano Zampini HYPRE_Int hnc = (HYPRE_Int)nzc; 13142cf14000SStefano Zampini 13152cf14000SStefano Zampini if ((PetscInt)hnc != nzc) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hypre overflow! number of columns %D for row %D",nzc,rows[i]); 131639accc25SStefano Zampini for (j=0;j<nzc;j++) { ierr = PetscHYPREScalarCast(vals[cscr[1][j]],&sscr[j]); } 1317c69f721fSFande Kong /* nonlocal values */ 131839accc25SStefano Zampini if (rows[i] < rst || rows[i] >= ren) { ierr = MatStashValuesRow_Private(&A->stash,rows[i],nzc,cscr[0],(PetscScalar*)sscr,PETSC_FALSE);CHKERRQ(ierr); } 1319c69f721fSFande Kong /* local values */ 13202cf14000SStefano Zampini else PetscStackCallStandard(HYPRE_IJMatrixSetValues,(hA->ij,1,&hnc,(HYPRE_BigInt*)(rows+i),(HYPRE_BigInt*)cscr[0],sscr)); 1321d975228cSstefano_zampini } 1322d975228cSstefano_zampini vals += nc; 1323d975228cSstefano_zampini } 1324d975228cSstefano_zampini } 1325c69f721fSFande Kong 1326708542d2SFande Kong ierr = MatRestoreArray_HYPRE(A,&array);CHKERRQ(ierr); 1327d975228cSstefano_zampini PetscFunctionReturn(0); 1328d975228cSstefano_zampini } 1329d975228cSstefano_zampini 1330d975228cSstefano_zampini static PetscErrorCode MatHYPRESetPreallocation_HYPRE(Mat A, PetscInt dnz, const PetscInt dnnz[], PetscInt onz, const PetscInt onnz[]) 1331d975228cSstefano_zampini { 1332d975228cSstefano_zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 13337d968826Sstefano_zampini HYPRE_Int *hdnnz,*honnz; 133406a29025Sstefano_zampini PetscInt i,rs,re,cs,ce,bs; 1335d975228cSstefano_zampini PetscMPIInt size; 1336d975228cSstefano_zampini PetscErrorCode ierr; 1337d975228cSstefano_zampini 1338d975228cSstefano_zampini PetscFunctionBegin; 133906a29025Sstefano_zampini ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr); 1340d975228cSstefano_zampini ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 1341d975228cSstefano_zampini ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 1342d975228cSstefano_zampini rs = A->rmap->rstart; 1343d975228cSstefano_zampini re = A->rmap->rend; 1344d975228cSstefano_zampini cs = A->cmap->rstart; 1345d975228cSstefano_zampini ce = A->cmap->rend; 1346d975228cSstefano_zampini if (!hA->ij) { 1347d975228cSstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixCreate,(hA->comm,rs,re-1,cs,ce-1,&hA->ij)); 1348d975228cSstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(hA->ij,HYPRE_PARCSR)); 1349d975228cSstefano_zampini } else { 13502cf14000SStefano Zampini HYPRE_BigInt hrs,hre,hcs,hce; 1351d975228cSstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetLocalRange,(hA->ij,&hrs,&hre,&hcs,&hce)); 13522cf14000SStefano Zampini if (hre-hrs+1 != re -rs) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Inconsistent local rows: IJMatrix [%D,%D), PETSc [%D,%D)",hrs,hre+1,rs,re); 13532cf14000SStefano Zampini if (hce-hcs+1 != ce -cs) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Inconsistent local cols: IJMatrix [%D,%D), PETSc [%D,%D)",hcs,hce+1,cs,ce); 1354d975228cSstefano_zampini } 135506a29025Sstefano_zampini if (dnz == PETSC_DEFAULT || dnz == PETSC_DECIDE) dnz = 10*bs; 135606a29025Sstefano_zampini if (onz == PETSC_DEFAULT || onz == PETSC_DECIDE) onz = 10*bs; 135706a29025Sstefano_zampini 1358d975228cSstefano_zampini if (!dnnz) { 1359d975228cSstefano_zampini ierr = PetscMalloc1(A->rmap->n,&hdnnz);CHKERRQ(ierr); 1360d975228cSstefano_zampini for (i=0;i<A->rmap->n;i++) hdnnz[i] = dnz; 1361d975228cSstefano_zampini } else { 13627d968826Sstefano_zampini hdnnz = (HYPRE_Int*)dnnz; 1363d975228cSstefano_zampini } 1364d975228cSstefano_zampini ierr = MPI_Comm_size(PetscObjectComm((PetscObject)A),&size);CHKERRQ(ierr); 1365d975228cSstefano_zampini if (size > 1) { 1366ddbeb582SStefano Zampini hypre_AuxParCSRMatrix *aux_matrix; 1367d975228cSstefano_zampini if (!onnz) { 1368d975228cSstefano_zampini ierr = PetscMalloc1(A->rmap->n,&honnz);CHKERRQ(ierr); 1369d975228cSstefano_zampini for (i=0;i<A->rmap->n;i++) honnz[i] = onz; 1370d975228cSstefano_zampini } else { 13717d968826Sstefano_zampini honnz = (HYPRE_Int*)onnz; 1372d975228cSstefano_zampini } 1373ddbeb582SStefano Zampini /* SetDiagOffdSizes sets hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 0, since it seems 1374ddbeb582SStefano Zampini they assume the user will input the entire row values, properly sorted 1375ddbeb582SStefano Zampini In PETSc, we don't make such an assumption, and we instead set this flag to 1 1376ddbeb582SStefano Zampini Also, to avoid possible memory leaks, we destroy and recreate the translator 1377ddbeb582SStefano Zampini This has to be done here, as HYPRE_IJMatrixInitialize will properly initialize 1378ddbeb582SStefano Zampini the IJ matrix for us */ 1379ddbeb582SStefano Zampini aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(hA->ij); 1380ddbeb582SStefano Zampini hypre_AuxParCSRMatrixDestroy(aux_matrix); 1381ddbeb582SStefano Zampini hypre_IJMatrixTranslator(hA->ij) = NULL; 1382d975228cSstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixSetDiagOffdSizes,(hA->ij,hdnnz,honnz)); 1383ddbeb582SStefano Zampini aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(hA->ij); 1384ddbeb582SStefano Zampini hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 1; 1385d975228cSstefano_zampini } else { 1386d975228cSstefano_zampini honnz = NULL; 1387d975228cSstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixSetRowSizes,(hA->ij,hdnnz)); 1388d975228cSstefano_zampini } 1389ddbeb582SStefano Zampini 1390af1cf968SStefano Zampini /* reset assembled flag and call the initialize method */ 1391af1cf968SStefano Zampini hypre_IJMatrixAssembleFlag(hA->ij) = 0; 1392ddbeb582SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixInitialize,(hA->ij)); 1393d975228cSstefano_zampini if (!dnnz) { 1394d975228cSstefano_zampini ierr = PetscFree(hdnnz);CHKERRQ(ierr); 1395d975228cSstefano_zampini } 1396d975228cSstefano_zampini if (!onnz && honnz) { 1397d975228cSstefano_zampini ierr = PetscFree(honnz);CHKERRQ(ierr); 1398d975228cSstefano_zampini } 1399af1cf968SStefano Zampini 1400af1cf968SStefano Zampini /* Match AIJ logic */ 140106a29025Sstefano_zampini A->preallocated = PETSC_TRUE; 1402af1cf968SStefano Zampini A->assembled = PETSC_FALSE; 1403d975228cSstefano_zampini PetscFunctionReturn(0); 1404d975228cSstefano_zampini } 1405d975228cSstefano_zampini 1406d975228cSstefano_zampini /*@C 1407d975228cSstefano_zampini MatHYPRESetPreallocation - Preallocates memory for a sparse parallel matrix in HYPRE IJ format 1408d975228cSstefano_zampini 1409d975228cSstefano_zampini Collective on Mat 1410d975228cSstefano_zampini 1411d975228cSstefano_zampini Input Parameters: 1412d975228cSstefano_zampini + A - the matrix 1413d975228cSstefano_zampini . dnz - number of nonzeros per row in DIAGONAL portion of local submatrix 1414d975228cSstefano_zampini (same value is used for all local rows) 1415d975228cSstefano_zampini . dnnz - array containing the number of nonzeros in the various rows of the 1416d975228cSstefano_zampini DIAGONAL portion of the local submatrix (possibly different for each row) 1417d975228cSstefano_zampini or NULL (PETSC_NULL_INTEGER in Fortran), if d_nz is used to specify the nonzero structure. 1418d975228cSstefano_zampini The size of this array is equal to the number of local rows, i.e 'm'. 1419d975228cSstefano_zampini For matrices that will be factored, you must leave room for (and set) 1420d975228cSstefano_zampini the diagonal entry even if it is zero. 1421d975228cSstefano_zampini . onz - number of nonzeros per row in the OFF-DIAGONAL portion of local 1422d975228cSstefano_zampini submatrix (same value is used for all local rows). 1423d975228cSstefano_zampini - onnz - array containing the number of nonzeros in the various rows of the 1424d975228cSstefano_zampini OFF-DIAGONAL portion of the local submatrix (possibly different for 1425d975228cSstefano_zampini each row) or NULL (PETSC_NULL_INTEGER in Fortran), if o_nz is used to specify the nonzero 1426d975228cSstefano_zampini structure. The size of this array is equal to the number 1427d975228cSstefano_zampini of local rows, i.e 'm'. 1428d975228cSstefano_zampini 142995452b02SPatrick Sanan Notes: 143095452b02SPatrick Sanan If the *nnz parameter is given then the *nz parameter is ignored; for sequential matrices, onz and onnz are ignored. 1431d975228cSstefano_zampini 1432d975228cSstefano_zampini Level: intermediate 1433d975228cSstefano_zampini 1434af1cf968SStefano Zampini .seealso: MatCreate(), MatMPIAIJSetPreallocation(), MATHYPRE 1435d975228cSstefano_zampini @*/ 1436d975228cSstefano_zampini PetscErrorCode MatHYPRESetPreallocation(Mat A, PetscInt dnz, const PetscInt dnnz[], PetscInt onz, const PetscInt onnz[]) 1437d975228cSstefano_zampini { 1438d975228cSstefano_zampini PetscErrorCode ierr; 1439d975228cSstefano_zampini 1440d975228cSstefano_zampini PetscFunctionBegin; 1441d975228cSstefano_zampini PetscValidHeaderSpecific(A,MAT_CLASSID,1); 1442d975228cSstefano_zampini PetscValidType(A,1); 1443d975228cSstefano_zampini ierr = PetscTryMethod(A,"MatHYPRESetPreallocation_C",(Mat,PetscInt,const PetscInt[],PetscInt,const PetscInt[]),(A,dnz,dnnz,onz,onnz));CHKERRQ(ierr); 1444d975228cSstefano_zampini PetscFunctionReturn(0); 1445d975228cSstefano_zampini } 1446d975228cSstefano_zampini 1447225daaf8SStefano Zampini /* 1448225daaf8SStefano Zampini MatCreateFromParCSR - Creates a matrix from a hypre_ParCSRMatrix 1449225daaf8SStefano Zampini 1450225daaf8SStefano Zampini Collective 1451225daaf8SStefano Zampini 1452225daaf8SStefano Zampini Input Parameters: 145345b8d346SStefano Zampini + parcsr - the pointer to the hypre_ParCSRMatrix 1454bb4689ddSStefano Zampini . mtype - matrix type to be created. Currently MATAIJ, MATIS and MATHYPRE are supported. 1455225daaf8SStefano Zampini - copymode - PETSc copying options 1456225daaf8SStefano Zampini 1457225daaf8SStefano Zampini Output Parameter: 1458225daaf8SStefano Zampini . A - the matrix 1459225daaf8SStefano Zampini 1460225daaf8SStefano Zampini Level: intermediate 1461225daaf8SStefano Zampini 1462225daaf8SStefano Zampini .seealso: MatHYPRE, PetscCopyMode 1463225daaf8SStefano Zampini */ 146445b8d346SStefano Zampini PETSC_EXTERN PetscErrorCode MatCreateFromParCSR(hypre_ParCSRMatrix *parcsr, MatType mtype, PetscCopyMode copymode, Mat* A) 1465978814f1SStefano Zampini { 1466225daaf8SStefano Zampini Mat T; 1467978814f1SStefano Zampini Mat_HYPRE *hA; 1468978814f1SStefano Zampini MPI_Comm comm; 1469978814f1SStefano Zampini PetscInt rstart,rend,cstart,cend,M,N; 1470d248a85cSRichard Tran Mills PetscBool isseqaij,isseqaijmkl,ismpiaij,isaij,ishyp,isis; 1471978814f1SStefano Zampini PetscErrorCode ierr; 1472978814f1SStefano Zampini 1473978814f1SStefano Zampini PetscFunctionBegin; 1474978814f1SStefano Zampini comm = hypre_ParCSRMatrixComm(parcsr); 1475225daaf8SStefano Zampini ierr = PetscStrcmp(mtype,MATSEQAIJ,&isseqaij);CHKERRQ(ierr); 1476d248a85cSRichard Tran Mills ierr = PetscStrcmp(mtype,MATSEQAIJMKL,&isseqaijmkl);CHKERRQ(ierr); 1477225daaf8SStefano Zampini ierr = PetscStrcmp(mtype,MATMPIAIJ,&ismpiaij);CHKERRQ(ierr); 1478225daaf8SStefano Zampini ierr = PetscStrcmp(mtype,MATAIJ,&isaij);CHKERRQ(ierr); 1479225daaf8SStefano Zampini ierr = PetscStrcmp(mtype,MATHYPRE,&ishyp);CHKERRQ(ierr); 14808cfe8d00SStefano Zampini ierr = PetscStrcmp(mtype,MATIS,&isis);CHKERRQ(ierr); 1481d248a85cSRichard Tran Mills isaij = (PetscBool)(isseqaij || isseqaijmkl || ismpiaij || isaij); 1482d248a85cSRichard Tran Mills if (!isaij && !ishyp && !isis) SETERRQ7(comm,PETSC_ERR_SUP,"Unsupported MatType %s! Supported types are %s, %s, %s, %s, %s, and %s",mtype,MATAIJ,MATSEQAIJ,MATSEQAIJMKL,MATMPIAIJ,MATIS,MATHYPRE); 1483978814f1SStefano Zampini /* access ParCSRMatrix */ 1484978814f1SStefano Zampini rstart = hypre_ParCSRMatrixFirstRowIndex(parcsr); 1485978814f1SStefano Zampini rend = hypre_ParCSRMatrixLastRowIndex(parcsr); 1486978814f1SStefano Zampini cstart = hypre_ParCSRMatrixFirstColDiag(parcsr); 1487978814f1SStefano Zampini cend = hypre_ParCSRMatrixLastColDiag(parcsr); 1488978814f1SStefano Zampini M = hypre_ParCSRMatrixGlobalNumRows(parcsr); 1489978814f1SStefano Zampini N = hypre_ParCSRMatrixGlobalNumCols(parcsr); 1490978814f1SStefano Zampini 1491fa92c42cSstefano_zampini /* fix for empty local rows/columns */ 1492fa92c42cSstefano_zampini if (rend < rstart) rend = rstart; 1493fa92c42cSstefano_zampini if (cend < cstart) cend = cstart; 1494fa92c42cSstefano_zampini 1495e6471dc9SStefano Zampini /* PETSc convention */ 1496e6471dc9SStefano Zampini rend++; 1497e6471dc9SStefano Zampini cend++; 1498e6471dc9SStefano Zampini rend = PetscMin(rend,M); 1499e6471dc9SStefano Zampini cend = PetscMin(cend,N); 1500e6471dc9SStefano Zampini 1501978814f1SStefano Zampini /* create PETSc matrix with MatHYPRE */ 1502225daaf8SStefano Zampini ierr = MatCreate(comm,&T);CHKERRQ(ierr); 1503e6471dc9SStefano Zampini ierr = MatSetSizes(T,rend-rstart,cend-cstart,M,N);CHKERRQ(ierr); 1504225daaf8SStefano Zampini ierr = MatSetType(T,MATHYPRE);CHKERRQ(ierr); 1505225daaf8SStefano Zampini hA = (Mat_HYPRE*)(T->data); 1506978814f1SStefano Zampini 1507978814f1SStefano Zampini /* create HYPRE_IJMatrix */ 1508978814f1SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixCreate,(hA->comm,rstart,rend-1,cstart,cend-1,&hA->ij)); 150945b8d346SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(hA->ij,HYPRE_PARCSR)); 151045b8d346SStefano Zampini 151145b8d346SStefano Zampini /* create new ParCSR object if needed */ 151245b8d346SStefano Zampini if (ishyp && copymode == PETSC_COPY_VALUES) { 151345b8d346SStefano Zampini hypre_ParCSRMatrix *new_parcsr; 151445b8d346SStefano Zampini hypre_CSRMatrix *hdiag,*hoffd,*ndiag,*noffd; 151545b8d346SStefano Zampini 15160e6427aaSSatish Balay new_parcsr = hypre_ParCSRMatrixClone(parcsr,0); 151745b8d346SStefano Zampini hdiag = hypre_ParCSRMatrixDiag(parcsr); 151845b8d346SStefano Zampini hoffd = hypre_ParCSRMatrixOffd(parcsr); 151945b8d346SStefano Zampini ndiag = hypre_ParCSRMatrixDiag(new_parcsr); 152045b8d346SStefano Zampini noffd = hypre_ParCSRMatrixOffd(new_parcsr); 1521580bdb30SBarry Smith ierr = PetscArraycpy(hypre_CSRMatrixData(ndiag),hypre_CSRMatrixData(hdiag),hypre_CSRMatrixNumNonzeros(hdiag));CHKERRQ(ierr); 1522580bdb30SBarry Smith ierr = PetscArraycpy(hypre_CSRMatrixData(noffd),hypre_CSRMatrixData(hoffd),hypre_CSRMatrixNumNonzeros(hoffd));CHKERRQ(ierr); 152345b8d346SStefano Zampini parcsr = new_parcsr; 152445b8d346SStefano Zampini copymode = PETSC_OWN_POINTER; 152545b8d346SStefano Zampini } 1526978814f1SStefano Zampini 1527978814f1SStefano Zampini /* set ParCSR object */ 1528978814f1SStefano Zampini hypre_IJMatrixObject(hA->ij) = parcsr; 15294ec6421dSstefano_zampini T->preallocated = PETSC_TRUE; 1530978814f1SStefano Zampini 1531978814f1SStefano Zampini /* set assembled flag */ 1532978814f1SStefano Zampini hypre_IJMatrixAssembleFlag(hA->ij) = 1; 1533978814f1SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixInitialize,(hA->ij)); 1534225daaf8SStefano Zampini if (ishyp) { 15356d2a658fSstefano_zampini PetscMPIInt myid = 0; 15366d2a658fSstefano_zampini 15376d2a658fSstefano_zampini /* make sure we always have row_starts and col_starts available */ 15386d2a658fSstefano_zampini if (HYPRE_AssumedPartitionCheck()) { 15396d2a658fSstefano_zampini ierr = MPI_Comm_rank(comm,&myid);CHKERRQ(ierr); 15406d2a658fSstefano_zampini } 15416d2a658fSstefano_zampini if (!hypre_ParCSRMatrixOwnsColStarts(parcsr)) { 15426d2a658fSstefano_zampini PetscLayout map; 15436d2a658fSstefano_zampini 15446d2a658fSstefano_zampini ierr = MatGetLayouts(T,NULL,&map);CHKERRQ(ierr); 15456d2a658fSstefano_zampini ierr = PetscLayoutSetUp(map);CHKERRQ(ierr); 15462cf14000SStefano Zampini hypre_ParCSRMatrixColStarts(parcsr) = (HYPRE_BigInt*)(map->range + myid); 15476d2a658fSstefano_zampini } 15486d2a658fSstefano_zampini if (!hypre_ParCSRMatrixOwnsRowStarts(parcsr)) { 15496d2a658fSstefano_zampini PetscLayout map; 15506d2a658fSstefano_zampini 15516d2a658fSstefano_zampini ierr = MatGetLayouts(T,&map,NULL);CHKERRQ(ierr); 15526d2a658fSstefano_zampini ierr = PetscLayoutSetUp(map);CHKERRQ(ierr); 15532cf14000SStefano Zampini hypre_ParCSRMatrixRowStarts(parcsr) = (HYPRE_BigInt*)(map->range + myid); 15546d2a658fSstefano_zampini } 1555978814f1SStefano Zampini /* prevent from freeing the pointer */ 1556978814f1SStefano Zampini if (copymode == PETSC_USE_POINTER) hA->inner_free = PETSC_FALSE; 1557225daaf8SStefano Zampini *A = T; 1558978814f1SStefano Zampini ierr = MatAssemblyBegin(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1559978814f1SStefano Zampini ierr = MatAssemblyEnd(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1560bb4689ddSStefano Zampini } else if (isaij) { 1561bb4689ddSStefano Zampini if (copymode != PETSC_OWN_POINTER) { 1562225daaf8SStefano Zampini /* prevent from freeing the pointer */ 1563225daaf8SStefano Zampini hA->inner_free = PETSC_FALSE; 1564225daaf8SStefano Zampini ierr = MatConvert_HYPRE_AIJ(T,MATAIJ,MAT_INITIAL_MATRIX,A);CHKERRQ(ierr); 1565225daaf8SStefano Zampini ierr = MatDestroy(&T);CHKERRQ(ierr); 1566225daaf8SStefano Zampini } else { /* AIJ return type with PETSC_OWN_POINTER */ 1567225daaf8SStefano Zampini ierr = MatConvert_HYPRE_AIJ(T,MATAIJ,MAT_INPLACE_MATRIX,&T);CHKERRQ(ierr); 1568225daaf8SStefano Zampini *A = T; 1569225daaf8SStefano Zampini } 1570bb4689ddSStefano Zampini } else if (isis) { 15718cfe8d00SStefano Zampini ierr = MatConvert_HYPRE_IS(T,MATIS,MAT_INITIAL_MATRIX,A);CHKERRQ(ierr); 15728cfe8d00SStefano Zampini if (copymode != PETSC_OWN_POINTER) hA->inner_free = PETSC_FALSE; 1573bb4689ddSStefano Zampini ierr = MatDestroy(&T);CHKERRQ(ierr); 1574bb4689ddSStefano Zampini } 1575978814f1SStefano Zampini PetscFunctionReturn(0); 1576978814f1SStefano Zampini } 1577978814f1SStefano Zampini 157868ec7858SStefano Zampini static PetscErrorCode MatHYPREGetParCSR_HYPRE(Mat A, hypre_ParCSRMatrix **parcsr) 1579dd9c0a25Sstefano_zampini { 1580dd9c0a25Sstefano_zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 1581dd9c0a25Sstefano_zampini HYPRE_Int type; 1582dd9c0a25Sstefano_zampini 1583dd9c0a25Sstefano_zampini PetscFunctionBegin; 1584dd9c0a25Sstefano_zampini if (!hA->ij) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"HYPRE_IJMatrix not present"); 1585dd9c0a25Sstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type)); 1586dd9c0a25Sstefano_zampini if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"HYPRE_IJMatrix is not of type HYPRE_PARCSR"); 1587dd9c0a25Sstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)parcsr)); 1588dd9c0a25Sstefano_zampini PetscFunctionReturn(0); 1589dd9c0a25Sstefano_zampini } 1590dd9c0a25Sstefano_zampini 1591dd9c0a25Sstefano_zampini /* 1592dd9c0a25Sstefano_zampini MatHYPREGetParCSR - Gets the pointer to the ParCSR matrix 1593dd9c0a25Sstefano_zampini 1594dd9c0a25Sstefano_zampini Not collective 1595dd9c0a25Sstefano_zampini 1596dd9c0a25Sstefano_zampini Input Parameters: 1597dd9c0a25Sstefano_zampini + A - the MATHYPRE object 1598dd9c0a25Sstefano_zampini 1599dd9c0a25Sstefano_zampini Output Parameter: 1600dd9c0a25Sstefano_zampini . parcsr - the pointer to the hypre_ParCSRMatrix 1601dd9c0a25Sstefano_zampini 1602dd9c0a25Sstefano_zampini Level: intermediate 1603dd9c0a25Sstefano_zampini 1604dd9c0a25Sstefano_zampini .seealso: MatHYPRE, PetscCopyMode 1605dd9c0a25Sstefano_zampini */ 1606dd9c0a25Sstefano_zampini PetscErrorCode MatHYPREGetParCSR(Mat A, hypre_ParCSRMatrix **parcsr) 1607dd9c0a25Sstefano_zampini { 1608dd9c0a25Sstefano_zampini PetscErrorCode ierr; 1609dd9c0a25Sstefano_zampini 1610dd9c0a25Sstefano_zampini PetscFunctionBegin; 1611dd9c0a25Sstefano_zampini PetscValidHeaderSpecific(A,MAT_CLASSID,1); 1612dd9c0a25Sstefano_zampini PetscValidType(A,1); 1613dd9c0a25Sstefano_zampini ierr = PetscUseMethod(A,"MatHYPREGetParCSR_C",(Mat,hypre_ParCSRMatrix**),(A,parcsr));CHKERRQ(ierr); 1614dd9c0a25Sstefano_zampini PetscFunctionReturn(0); 1615dd9c0a25Sstefano_zampini } 1616dd9c0a25Sstefano_zampini 161768ec7858SStefano Zampini static PetscErrorCode MatMissingDiagonal_HYPRE(Mat A, PetscBool *missing, PetscInt *dd) 161868ec7858SStefano Zampini { 161968ec7858SStefano Zampini hypre_ParCSRMatrix *parcsr; 162068ec7858SStefano Zampini hypre_CSRMatrix *ha; 162168ec7858SStefano Zampini PetscInt rst; 162268ec7858SStefano Zampini PetscErrorCode ierr; 162368ec7858SStefano Zampini 162468ec7858SStefano Zampini PetscFunctionBegin; 162568ec7858SStefano Zampini if (A->rmap->n != A->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not implemented with non-square diagonal blocks"); 162668ec7858SStefano Zampini ierr = MatGetOwnershipRange(A,&rst,NULL);CHKERRQ(ierr); 162768ec7858SStefano Zampini ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 162868ec7858SStefano Zampini if (missing) *missing = PETSC_FALSE; 162968ec7858SStefano Zampini if (dd) *dd = -1; 163068ec7858SStefano Zampini ha = hypre_ParCSRMatrixDiag(parcsr); 163168ec7858SStefano Zampini if (ha) { 163268299464SStefano Zampini PetscInt size,i; 163368299464SStefano Zampini HYPRE_Int *ii,*jj; 163468ec7858SStefano Zampini 163568ec7858SStefano Zampini size = hypre_CSRMatrixNumRows(ha); 163668ec7858SStefano Zampini ii = hypre_CSRMatrixI(ha); 163768ec7858SStefano Zampini jj = hypre_CSRMatrixJ(ha); 163868ec7858SStefano Zampini for (i = 0; i < size; i++) { 163968ec7858SStefano Zampini PetscInt j; 164068ec7858SStefano Zampini PetscBool found = PETSC_FALSE; 164168ec7858SStefano Zampini 164268ec7858SStefano Zampini for (j = ii[i]; j < ii[i+1] && !found; j++) 164368ec7858SStefano Zampini found = (jj[j] == i) ? PETSC_TRUE : PETSC_FALSE; 164468ec7858SStefano Zampini 164568ec7858SStefano Zampini if (!found) { 164668ec7858SStefano Zampini PetscInfo1(A,"Matrix is missing local diagonal entry %D\n",i); 164768ec7858SStefano Zampini if (missing) *missing = PETSC_TRUE; 164868ec7858SStefano Zampini if (dd) *dd = i+rst; 164968ec7858SStefano Zampini PetscFunctionReturn(0); 165068ec7858SStefano Zampini } 165168ec7858SStefano Zampini } 165268ec7858SStefano Zampini if (!size) { 165368ec7858SStefano Zampini PetscInfo(A,"Matrix has no diagonal entries therefore is missing diagonal\n"); 165468ec7858SStefano Zampini if (missing) *missing = PETSC_TRUE; 165568ec7858SStefano Zampini if (dd) *dd = rst; 165668ec7858SStefano Zampini } 165768ec7858SStefano Zampini } else { 165868ec7858SStefano Zampini PetscInfo(A,"Matrix has no diagonal entries therefore is missing diagonal\n"); 165968ec7858SStefano Zampini if (missing) *missing = PETSC_TRUE; 166068ec7858SStefano Zampini if (dd) *dd = rst; 166168ec7858SStefano Zampini } 166268ec7858SStefano Zampini PetscFunctionReturn(0); 166368ec7858SStefano Zampini } 166468ec7858SStefano Zampini 166568ec7858SStefano Zampini static PetscErrorCode MatScale_HYPRE(Mat A, PetscScalar s) 166668ec7858SStefano Zampini { 166768ec7858SStefano Zampini hypre_ParCSRMatrix *parcsr; 166868ec7858SStefano Zampini hypre_CSRMatrix *ha; 166968ec7858SStefano Zampini PetscErrorCode ierr; 167039accc25SStefano Zampini HYPRE_Complex hs; 167168ec7858SStefano Zampini 167268ec7858SStefano Zampini PetscFunctionBegin; 167339accc25SStefano Zampini ierr = PetscHYPREScalarCast(s,&hs);CHKERRQ(ierr); 167468ec7858SStefano Zampini ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 167568ec7858SStefano Zampini /* diagonal part */ 167668ec7858SStefano Zampini ha = hypre_ParCSRMatrixDiag(parcsr); 167768ec7858SStefano Zampini if (ha) { 167868299464SStefano Zampini PetscInt size,i; 167968299464SStefano Zampini HYPRE_Int *ii; 168039accc25SStefano Zampini HYPRE_Complex *a; 168168ec7858SStefano Zampini 168268ec7858SStefano Zampini size = hypre_CSRMatrixNumRows(ha); 168368ec7858SStefano Zampini a = hypre_CSRMatrixData(ha); 168468ec7858SStefano Zampini ii = hypre_CSRMatrixI(ha); 168539accc25SStefano Zampini for (i = 0; i < ii[size]; i++) a[i] *= hs; 168668ec7858SStefano Zampini } 168768ec7858SStefano Zampini /* offdiagonal part */ 168868ec7858SStefano Zampini ha = hypre_ParCSRMatrixOffd(parcsr); 168968ec7858SStefano Zampini if (ha) { 169068299464SStefano Zampini PetscInt size,i; 169168299464SStefano Zampini HYPRE_Int *ii; 169239accc25SStefano Zampini HYPRE_Complex *a; 169368ec7858SStefano Zampini 169468ec7858SStefano Zampini size = hypre_CSRMatrixNumRows(ha); 169568ec7858SStefano Zampini a = hypre_CSRMatrixData(ha); 169668ec7858SStefano Zampini ii = hypre_CSRMatrixI(ha); 169739accc25SStefano Zampini for (i = 0; i < ii[size]; i++) a[i] *= hs; 169868ec7858SStefano Zampini } 169968ec7858SStefano Zampini PetscFunctionReturn(0); 170068ec7858SStefano Zampini } 170168ec7858SStefano Zampini 170268ec7858SStefano Zampini static PetscErrorCode MatZeroRowsColumns_HYPRE(Mat A, PetscInt numRows, const PetscInt rows[], PetscScalar diag, Vec x, Vec b) 170368ec7858SStefano Zampini { 170468ec7858SStefano Zampini hypre_ParCSRMatrix *parcsr; 170568299464SStefano Zampini HYPRE_Int *lrows; 170668299464SStefano Zampini PetscInt rst,ren,i; 170768ec7858SStefano Zampini PetscErrorCode ierr; 170868ec7858SStefano Zampini 170968ec7858SStefano Zampini PetscFunctionBegin; 171068ec7858SStefano Zampini if (x || b) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"To be implemented"); 171168ec7858SStefano Zampini ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 171268ec7858SStefano Zampini ierr = PetscMalloc1(numRows,&lrows);CHKERRQ(ierr); 171368ec7858SStefano Zampini ierr = MatGetOwnershipRange(A,&rst,&ren);CHKERRQ(ierr); 171468ec7858SStefano Zampini for (i=0;i<numRows;i++) { 171568ec7858SStefano Zampini if (rows[i] < rst || rows[i] >= ren) 171668ec7858SStefano Zampini SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Non-local rows not yet supported"); 171768ec7858SStefano Zampini lrows[i] = rows[i] - rst; 171868ec7858SStefano Zampini } 171968ec7858SStefano Zampini PetscStackCallStandard(hypre_ParCSRMatrixEliminateRowsCols,(parcsr,numRows,lrows)); 172068ec7858SStefano Zampini ierr = PetscFree(lrows);CHKERRQ(ierr); 172168ec7858SStefano Zampini PetscFunctionReturn(0); 172268ec7858SStefano Zampini } 172368ec7858SStefano Zampini 1724c69f721fSFande Kong static PetscErrorCode MatZeroEntries_HYPRE_CSRMatrix(hypre_CSRMatrix *ha) 1725c69f721fSFande Kong { 1726c69f721fSFande Kong PetscErrorCode ierr; 1727c69f721fSFande Kong 1728c69f721fSFande Kong PetscFunctionBegin; 1729c69f721fSFande Kong if (ha) { 1730c69f721fSFande Kong HYPRE_Int *ii, size; 1731c69f721fSFande Kong HYPRE_Complex *a; 1732c69f721fSFande Kong 1733c69f721fSFande Kong size = hypre_CSRMatrixNumRows(ha); 1734c69f721fSFande Kong a = hypre_CSRMatrixData(ha); 1735c69f721fSFande Kong ii = hypre_CSRMatrixI(ha); 1736c69f721fSFande Kong 1737580bdb30SBarry Smith if (a) {ierr = PetscArrayzero(a,ii[size]);CHKERRQ(ierr);} 1738c69f721fSFande Kong } 1739c69f721fSFande Kong PetscFunctionReturn(0); 1740c69f721fSFande Kong } 1741c69f721fSFande Kong 1742c69f721fSFande Kong PetscErrorCode MatZeroEntries_HYPRE(Mat A) 1743c69f721fSFande Kong { 1744c69f721fSFande Kong hypre_ParCSRMatrix *parcsr; 1745c69f721fSFande Kong PetscErrorCode ierr; 1746c69f721fSFande Kong 1747c69f721fSFande Kong PetscFunctionBegin; 1748c69f721fSFande Kong ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 1749c69f721fSFande Kong /* diagonal part */ 1750c69f721fSFande Kong ierr = MatZeroEntries_HYPRE_CSRMatrix(hypre_ParCSRMatrixDiag(parcsr));CHKERRQ(ierr); 1751c69f721fSFande Kong /* off-diagonal part */ 1752c69f721fSFande Kong ierr = MatZeroEntries_HYPRE_CSRMatrix(hypre_ParCSRMatrixOffd(parcsr));CHKERRQ(ierr); 1753c69f721fSFande Kong PetscFunctionReturn(0); 1754c69f721fSFande Kong } 1755c69f721fSFande Kong 175639accc25SStefano Zampini static PetscErrorCode MatZeroRows_HYPRE_CSRMatrix(hypre_CSRMatrix *hA,PetscInt N,const PetscInt rows[],HYPRE_Complex diag) 1757c69f721fSFande Kong { 175839accc25SStefano Zampini PetscInt ii; 175939accc25SStefano Zampini HYPRE_Int *i, *j; 176039accc25SStefano Zampini HYPRE_Complex *a; 1761c69f721fSFande Kong 1762c69f721fSFande Kong PetscFunctionBegin; 1763c69f721fSFande Kong if (!hA) PetscFunctionReturn(0); 1764c69f721fSFande Kong 176539accc25SStefano Zampini i = hypre_CSRMatrixI(hA); 176639accc25SStefano Zampini j = hypre_CSRMatrixJ(hA); 1767c69f721fSFande Kong a = hypre_CSRMatrixData(hA); 1768c69f721fSFande Kong 1769c69f721fSFande Kong for (ii = 0; ii < N; ii++) { 177039accc25SStefano Zampini HYPRE_Int jj, ibeg, iend, irow; 177139accc25SStefano Zampini 1772c69f721fSFande Kong irow = rows[ii]; 1773c69f721fSFande Kong ibeg = i[irow]; 1774c69f721fSFande Kong iend = i[irow+1]; 1775c69f721fSFande Kong for (jj = ibeg; jj < iend; jj++) 1776c69f721fSFande Kong if (j[jj] == irow) a[jj] = diag; 1777c69f721fSFande Kong else a[jj] = 0.0; 1778c69f721fSFande Kong } 1779c69f721fSFande Kong PetscFunctionReturn(0); 1780c69f721fSFande Kong } 1781c69f721fSFande Kong 1782ddbeb582SStefano Zampini static PetscErrorCode MatZeroRows_HYPRE(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag,Vec x,Vec b) 1783c69f721fSFande Kong { 1784c69f721fSFande Kong hypre_ParCSRMatrix *parcsr; 1785c69f721fSFande Kong PetscInt *lrows,len; 178639accc25SStefano Zampini HYPRE_Complex hdiag; 1787c69f721fSFande Kong PetscErrorCode ierr; 1788c69f721fSFande Kong 1789c69f721fSFande Kong PetscFunctionBegin; 1790c6698e78SStefano Zampini if (x || b) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Does not support to modify the solution and the right hand size"); 179139accc25SStefano Zampini ierr = PetscHYPREScalarCast(diag,&hdiag);CHKERRQ(ierr); 1792c69f721fSFande Kong /* retrieve the internal matrix */ 1793c69f721fSFande Kong ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 1794c69f721fSFande Kong /* get locally owned rows */ 1795c69f721fSFande Kong ierr = MatZeroRowsMapLocal_Private(A,N,rows,&len,&lrows);CHKERRQ(ierr); 1796c69f721fSFande Kong /* zero diagonal part */ 179739accc25SStefano Zampini ierr = MatZeroRows_HYPRE_CSRMatrix(hypre_ParCSRMatrixDiag(parcsr),len,lrows,hdiag);CHKERRQ(ierr); 1798c69f721fSFande Kong /* zero off-diagonal part */ 1799c69f721fSFande Kong ierr = MatZeroRows_HYPRE_CSRMatrix(hypre_ParCSRMatrixOffd(parcsr),len,lrows,0.0);CHKERRQ(ierr); 1800c69f721fSFande Kong 1801c69f721fSFande Kong ierr = PetscFree(lrows);CHKERRQ(ierr); 1802c69f721fSFande Kong PetscFunctionReturn(0); 1803c69f721fSFande Kong } 1804c69f721fSFande Kong 1805ddbeb582SStefano Zampini static PetscErrorCode MatAssemblyBegin_HYPRE(Mat mat,MatAssemblyType mode) 1806c69f721fSFande Kong { 1807c69f721fSFande Kong PetscErrorCode ierr; 1808c69f721fSFande Kong 1809c69f721fSFande Kong PetscFunctionBegin; 1810c69f721fSFande Kong if (mat->nooffprocentries) PetscFunctionReturn(0); 1811c69f721fSFande Kong 1812c69f721fSFande Kong ierr = MatStashScatterBegin_Private(mat,&mat->stash,mat->rmap->range);CHKERRQ(ierr); 1813c69f721fSFande Kong PetscFunctionReturn(0); 1814c69f721fSFande Kong } 1815c69f721fSFande Kong 1816ddbeb582SStefano Zampini static PetscErrorCode MatGetRow_HYPRE(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 1817c69f721fSFande Kong { 1818c69f721fSFande Kong hypre_ParCSRMatrix *parcsr; 18192cf14000SStefano Zampini HYPRE_Int hnz; 1820c69f721fSFande Kong PetscErrorCode ierr; 1821c69f721fSFande Kong 1822c69f721fSFande Kong PetscFunctionBegin; 1823c69f721fSFande Kong /* retrieve the internal matrix */ 1824c69f721fSFande Kong ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 1825c69f721fSFande Kong /* call HYPRE API */ 182639accc25SStefano Zampini PetscStackCallStandard(HYPRE_ParCSRMatrixGetRow,(parcsr,row,&hnz,(HYPRE_BigInt**)idx,(HYPRE_Complex**)v)); 18272cf14000SStefano Zampini if (nz) *nz = (PetscInt)hnz; 1828c69f721fSFande Kong PetscFunctionReturn(0); 1829c69f721fSFande Kong } 1830c69f721fSFande Kong 1831ddbeb582SStefano Zampini static PetscErrorCode MatRestoreRow_HYPRE(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 1832c69f721fSFande Kong { 1833c69f721fSFande Kong hypre_ParCSRMatrix *parcsr; 18342cf14000SStefano Zampini HYPRE_Int hnz; 1835c69f721fSFande Kong PetscErrorCode ierr; 1836c69f721fSFande Kong 1837c69f721fSFande Kong PetscFunctionBegin; 1838c69f721fSFande Kong /* retrieve the internal matrix */ 1839c69f721fSFande Kong ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 1840c69f721fSFande Kong /* call HYPRE API */ 18412cf14000SStefano Zampini hnz = nz ? (HYPRE_Int)(*nz) : 0; 184239accc25SStefano Zampini PetscStackCallStandard(HYPRE_ParCSRMatrixRestoreRow,(parcsr,row,&hnz,(HYPRE_BigInt**)idx,(HYPRE_Complex**)v)); 1843c69f721fSFande Kong PetscFunctionReturn(0); 1844c69f721fSFande Kong } 1845c69f721fSFande Kong 1846ddbeb582SStefano Zampini static PetscErrorCode MatGetValues_HYPRE(Mat A,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],PetscScalar v[]) 1847c69f721fSFande Kong { 184845b8d346SStefano Zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 1849c69f721fSFande Kong PetscInt i; 18501d4906efSStefano Zampini 1851c69f721fSFande Kong PetscFunctionBegin; 1852c69f721fSFande Kong if (!m || !n) PetscFunctionReturn(0); 1853c69f721fSFande Kong /* Ignore negative row indices 1854c69f721fSFande Kong * And negative column indices should be automatically ignored in hypre 1855c69f721fSFande Kong * */ 18562cf14000SStefano Zampini for (i=0; i<m; i++) { 18572cf14000SStefano Zampini if (idxm[i] >= 0) { 18582cf14000SStefano Zampini HYPRE_Int hn = (HYPRE_Int)n; 185939accc25SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetValues,(hA->ij,1,&hn,(HYPRE_BigInt*)&idxm[i],(HYPRE_BigInt*)idxn,(HYPRE_Complex*)(v + i*n))); 18602cf14000SStefano Zampini } 18612cf14000SStefano Zampini } 1862c69f721fSFande Kong PetscFunctionReturn(0); 1863c69f721fSFande Kong } 1864c69f721fSFande Kong 1865ddbeb582SStefano Zampini static PetscErrorCode MatSetOption_HYPRE(Mat A,MatOption op,PetscBool flg) 1866ddbeb582SStefano Zampini { 1867ddbeb582SStefano Zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 1868ddbeb582SStefano Zampini 1869ddbeb582SStefano Zampini PetscFunctionBegin; 1870c6698e78SStefano Zampini switch (op) { 1871ddbeb582SStefano Zampini case MAT_NO_OFF_PROC_ENTRIES: 1872ddbeb582SStefano Zampini if (flg) { 1873ddbeb582SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixSetMaxOffProcElmts,(hA->ij,0)); 1874ddbeb582SStefano Zampini } 1875ddbeb582SStefano Zampini break; 1876ddbeb582SStefano Zampini default: 1877ddbeb582SStefano Zampini break; 1878ddbeb582SStefano Zampini } 1879ddbeb582SStefano Zampini PetscFunctionReturn(0); 1880ddbeb582SStefano Zampini } 1881c69f721fSFande Kong 188245b8d346SStefano Zampini static PetscErrorCode MatView_HYPRE(Mat A, PetscViewer view) 188345b8d346SStefano Zampini { 188445b8d346SStefano Zampini hypre_ParCSRMatrix *parcsr; 188545b8d346SStefano Zampini PetscErrorCode ierr; 188645b8d346SStefano Zampini Mat B; 188745b8d346SStefano Zampini PetscViewerFormat format; 188845b8d346SStefano Zampini PetscErrorCode (*mview)(Mat,PetscViewer) = NULL; 188945b8d346SStefano Zampini 189045b8d346SStefano Zampini PetscFunctionBegin; 189145b8d346SStefano Zampini ierr = PetscViewerGetFormat(view,&format);CHKERRQ(ierr); 189245b8d346SStefano Zampini if (format != PETSC_VIEWER_NATIVE) { 189345b8d346SStefano Zampini ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 189445b8d346SStefano Zampini ierr = MatCreateFromParCSR(parcsr,MATAIJ,PETSC_USE_POINTER,&B);CHKERRQ(ierr); 189545b8d346SStefano Zampini ierr = MatGetOperation(B,MATOP_VIEW,(void(**)(void))&mview);CHKERRQ(ierr); 189645b8d346SStefano Zampini if (!mview) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Missing view operation");CHKERRQ(ierr); 189745b8d346SStefano Zampini ierr = (*mview)(B,view);CHKERRQ(ierr); 189845b8d346SStefano Zampini ierr = MatDestroy(&B);CHKERRQ(ierr); 189945b8d346SStefano Zampini } else { 190045b8d346SStefano Zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 190145b8d346SStefano Zampini PetscMPIInt size; 190245b8d346SStefano Zampini PetscBool isascii; 190345b8d346SStefano Zampini const char *filename; 190445b8d346SStefano Zampini 190545b8d346SStefano Zampini /* HYPRE uses only text files */ 190645b8d346SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)view,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr); 190745b8d346SStefano Zampini if (!isascii) SETERRQ1(PetscObjectComm((PetscObject)view),PETSC_ERR_SUP,"PetscViewerType %s: native HYPRE format needs PETSCVIEWERASCII",((PetscObject)view)->type_name); 190845b8d346SStefano Zampini ierr = PetscViewerFileGetName(view,&filename);CHKERRQ(ierr); 190945b8d346SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixPrint,(hA->ij,filename)); 191045b8d346SStefano Zampini ierr = MPI_Comm_size(hA->comm,&size);CHKERRQ(ierr); 191145b8d346SStefano Zampini if (size > 1) { 191245b8d346SStefano Zampini ierr = PetscViewerASCIIPrintf(view,"Matrix files: %s.%05d ... %s.%05d\n",filename,0,filename,size-1);CHKERRQ(ierr); 191345b8d346SStefano Zampini } else { 191445b8d346SStefano Zampini ierr = PetscViewerASCIIPrintf(view,"Matrix file: %s.%05d\n",filename,0);CHKERRQ(ierr); 191545b8d346SStefano Zampini } 191645b8d346SStefano Zampini } 191745b8d346SStefano Zampini PetscFunctionReturn(0); 191845b8d346SStefano Zampini } 191945b8d346SStefano Zampini 192045b8d346SStefano Zampini static PetscErrorCode MatDuplicate_HYPRE(Mat A,MatDuplicateOption op, Mat *B) 192145b8d346SStefano Zampini { 192245b8d346SStefano Zampini hypre_ParCSRMatrix *parcsr; 192345b8d346SStefano Zampini PetscErrorCode ierr; 192445b8d346SStefano Zampini PetscCopyMode cpmode; 192545b8d346SStefano Zampini 192645b8d346SStefano Zampini PetscFunctionBegin; 192745b8d346SStefano Zampini ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 192845b8d346SStefano Zampini if (op == MAT_DO_NOT_COPY_VALUES || op == MAT_SHARE_NONZERO_PATTERN) { 19290e6427aaSSatish Balay parcsr = hypre_ParCSRMatrixClone(parcsr,0); 193045b8d346SStefano Zampini cpmode = PETSC_OWN_POINTER; 193145b8d346SStefano Zampini } else { 193245b8d346SStefano Zampini cpmode = PETSC_COPY_VALUES; 193345b8d346SStefano Zampini } 193445b8d346SStefano Zampini ierr = MatCreateFromParCSR(parcsr,MATHYPRE,cpmode,B);CHKERRQ(ierr); 193545b8d346SStefano Zampini PetscFunctionReturn(0); 193645b8d346SStefano Zampini } 193745b8d346SStefano Zampini 1938465edc17SStefano Zampini static PetscErrorCode MatCopy_HYPRE(Mat A, Mat B, MatStructure str) 1939465edc17SStefano Zampini { 1940465edc17SStefano Zampini hypre_ParCSRMatrix *acsr,*bcsr; 1941465edc17SStefano Zampini PetscErrorCode ierr; 1942465edc17SStefano Zampini 1943465edc17SStefano Zampini PetscFunctionBegin; 1944465edc17SStefano Zampini if (str == SAME_NONZERO_PATTERN && A->ops->copy == B->ops->copy) { 1945465edc17SStefano Zampini ierr = MatHYPREGetParCSR_HYPRE(A,&acsr);CHKERRQ(ierr); 1946465edc17SStefano Zampini ierr = MatHYPREGetParCSR_HYPRE(B,&bcsr);CHKERRQ(ierr); 1947465edc17SStefano Zampini PetscStackCallStandard(hypre_ParCSRMatrixCopy,(acsr,bcsr,1)); 1948465edc17SStefano Zampini ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1949465edc17SStefano Zampini ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1950465edc17SStefano Zampini } else { 1951465edc17SStefano Zampini ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 1952465edc17SStefano Zampini } 1953465edc17SStefano Zampini PetscFunctionReturn(0); 1954465edc17SStefano Zampini } 1955465edc17SStefano Zampini 19566305df00SStefano Zampini static PetscErrorCode MatGetDiagonal_HYPRE(Mat A, Vec d) 19576305df00SStefano Zampini { 19586305df00SStefano Zampini hypre_ParCSRMatrix *parcsr; 19596305df00SStefano Zampini hypre_CSRMatrix *dmat; 196039accc25SStefano Zampini HYPRE_Complex *a; 196139accc25SStefano Zampini HYPRE_Complex *data = NULL; 19622cf14000SStefano Zampini HYPRE_Int *diag = NULL; 19632cf14000SStefano Zampini PetscInt i; 19646305df00SStefano Zampini PetscBool cong; 19656305df00SStefano Zampini PetscErrorCode ierr; 19666305df00SStefano Zampini 19676305df00SStefano Zampini PetscFunctionBegin; 19686305df00SStefano Zampini ierr = MatHasCongruentLayouts(A,&cong);CHKERRQ(ierr); 19696305df00SStefano Zampini if (!cong) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Only for square matrices with same local distributions of rows and columns"); 19706305df00SStefano Zampini #if defined(PETSC_USE_DEBUG) 19716305df00SStefano Zampini { 19726305df00SStefano Zampini PetscBool miss; 19736305df00SStefano Zampini ierr = MatMissingDiagonal(A,&miss,NULL);CHKERRQ(ierr); 19746305df00SStefano Zampini if (miss && A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not implemented when diagonal entries are missing"); 19756305df00SStefano Zampini } 19766305df00SStefano Zampini #endif 19776305df00SStefano Zampini ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 19786305df00SStefano Zampini dmat = hypre_ParCSRMatrixDiag(parcsr); 19796305df00SStefano Zampini if (dmat) { 198039accc25SStefano Zampini /* this cast fixes the clang error: implicit conversion from 'HYPRE_Complex' (aka '_Complex double') to 'double' is not permitted in C++ */ 198139accc25SStefano Zampini ierr = VecGetArray(d,(PetscScalar**)&a);CHKERRQ(ierr); 19822cf14000SStefano Zampini diag = hypre_CSRMatrixI(dmat); 198339accc25SStefano Zampini data = hypre_CSRMatrixData(dmat); 19846305df00SStefano Zampini for (i=0;i<A->rmap->n;i++) a[i] = data[diag[i]]; 198539accc25SStefano Zampini ierr = VecRestoreArray(d,(PetscScalar**)&a);CHKERRQ(ierr); 19866305df00SStefano Zampini } 19876305df00SStefano Zampini PetscFunctionReturn(0); 19886305df00SStefano Zampini } 19896305df00SStefano Zampini 1990363d496dSStefano Zampini #include <petscblaslapack.h> 1991363d496dSStefano Zampini 1992363d496dSStefano Zampini static PetscErrorCode MatAXPY_HYPRE(Mat Y,PetscScalar a,Mat X,MatStructure str) 1993363d496dSStefano Zampini { 1994363d496dSStefano Zampini PetscErrorCode ierr; 1995363d496dSStefano Zampini 1996363d496dSStefano Zampini PetscFunctionBegin; 1997363d496dSStefano Zampini if (str == SAME_NONZERO_PATTERN) { 1998363d496dSStefano Zampini hypre_ParCSRMatrix *x,*y; 1999363d496dSStefano Zampini hypre_CSRMatrix *xloc,*yloc; 2000363d496dSStefano Zampini PetscInt xnnz,ynnz; 200139accc25SStefano Zampini HYPRE_Complex *xarr,*yarr; 2002363d496dSStefano Zampini PetscBLASInt one=1,bnz; 2003363d496dSStefano Zampini 2004363d496dSStefano Zampini ierr = MatHYPREGetParCSR(Y,&y);CHKERRQ(ierr); 2005363d496dSStefano Zampini ierr = MatHYPREGetParCSR(X,&x);CHKERRQ(ierr); 2006363d496dSStefano Zampini 2007363d496dSStefano Zampini /* diagonal block */ 2008363d496dSStefano Zampini xloc = hypre_ParCSRMatrixDiag(x); 2009363d496dSStefano Zampini yloc = hypre_ParCSRMatrixDiag(y); 2010363d496dSStefano Zampini xnnz = 0; 2011363d496dSStefano Zampini ynnz = 0; 2012363d496dSStefano Zampini xarr = NULL; 2013363d496dSStefano Zampini yarr = NULL; 2014363d496dSStefano Zampini if (xloc) { 201539accc25SStefano Zampini xarr = hypre_CSRMatrixData(xloc); 2016363d496dSStefano Zampini xnnz = hypre_CSRMatrixNumNonzeros(xloc); 2017363d496dSStefano Zampini } 2018363d496dSStefano Zampini if (yloc) { 201939accc25SStefano Zampini yarr = hypre_CSRMatrixData(yloc); 2020363d496dSStefano Zampini ynnz = hypre_CSRMatrixNumNonzeros(yloc); 2021363d496dSStefano Zampini } 2022363d496dSStefano Zampini if (xnnz != ynnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Different number of nonzeros in diagonal block %D != %D",xnnz,ynnz); 2023363d496dSStefano Zampini ierr = PetscBLASIntCast(xnnz,&bnz);CHKERRQ(ierr); 202439accc25SStefano Zampini PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&a,(PetscScalar*)xarr,&one,(PetscScalar*)yarr,&one)); 2025363d496dSStefano Zampini 2026363d496dSStefano Zampini /* off-diagonal block */ 2027363d496dSStefano Zampini xloc = hypre_ParCSRMatrixOffd(x); 2028363d496dSStefano Zampini yloc = hypre_ParCSRMatrixOffd(y); 2029363d496dSStefano Zampini xnnz = 0; 2030363d496dSStefano Zampini ynnz = 0; 2031363d496dSStefano Zampini xarr = NULL; 2032363d496dSStefano Zampini yarr = NULL; 2033363d496dSStefano Zampini if (xloc) { 203439accc25SStefano Zampini xarr = hypre_CSRMatrixData(xloc); 2035363d496dSStefano Zampini xnnz = hypre_CSRMatrixNumNonzeros(xloc); 2036363d496dSStefano Zampini } 2037363d496dSStefano Zampini if (yloc) { 203839accc25SStefano Zampini yarr = hypre_CSRMatrixData(yloc); 2039363d496dSStefano Zampini ynnz = hypre_CSRMatrixNumNonzeros(yloc); 2040363d496dSStefano Zampini } 2041363d496dSStefano Zampini if (xnnz != ynnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Different number of nonzeros in off-diagonal block %D != %D",xnnz,ynnz); 2042363d496dSStefano Zampini ierr = PetscBLASIntCast(xnnz,&bnz);CHKERRQ(ierr); 204339accc25SStefano Zampini PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&a,(PetscScalar*)xarr,&one,(PetscScalar*)yarr,&one)); 2044363d496dSStefano Zampini } else if (str == SUBSET_NONZERO_PATTERN) { 2045363d496dSStefano Zampini ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr); 2046363d496dSStefano Zampini } else { 2047363d496dSStefano Zampini Mat B; 2048363d496dSStefano Zampini 2049363d496dSStefano Zampini ierr = MatAXPY_Basic_Preallocate(Y,X,&B);CHKERRQ(ierr); 2050363d496dSStefano Zampini ierr = MatAXPY_BasicWithPreallocation(B,Y,a,X,str);CHKERRQ(ierr); 2051363d496dSStefano Zampini ierr = MatHeaderReplace(Y,&B);CHKERRQ(ierr); 2052363d496dSStefano Zampini } 2053363d496dSStefano Zampini PetscFunctionReturn(0); 2054363d496dSStefano Zampini } 2055363d496dSStefano Zampini 2056a055b5aaSBarry Smith /*MC 2057a055b5aaSBarry Smith MATHYPRE - MATHYPRE = "hypre" - A matrix type to be used for sequential and parallel sparse matrices 2058a055b5aaSBarry Smith based on the hypre IJ interface. 2059a055b5aaSBarry Smith 2060a055b5aaSBarry Smith Level: intermediate 2061a055b5aaSBarry Smith 2062a055b5aaSBarry Smith .seealso: MatCreate() 2063a055b5aaSBarry Smith M*/ 2064a055b5aaSBarry Smith 206563c07aadSStefano Zampini PETSC_EXTERN PetscErrorCode MatCreate_HYPRE(Mat B) 206663c07aadSStefano Zampini { 206763c07aadSStefano Zampini Mat_HYPRE *hB; 206863c07aadSStefano Zampini PetscErrorCode ierr; 206963c07aadSStefano Zampini 207063c07aadSStefano Zampini PetscFunctionBegin; 207163c07aadSStefano Zampini ierr = PetscNewLog(B,&hB);CHKERRQ(ierr); 2072978814f1SStefano Zampini hB->inner_free = PETSC_TRUE; 2073c69f721fSFande Kong hB->available = PETSC_TRUE; 2074c69f721fSFande Kong hB->size = 0; 2075c69f721fSFande Kong hB->array = NULL; 2076978814f1SStefano Zampini 207763c07aadSStefano Zampini B->data = (void*)hB; 207863c07aadSStefano Zampini B->rmap->bs = 1; 207963c07aadSStefano Zampini B->assembled = PETSC_FALSE; 208063c07aadSStefano Zampini 2081de393100SStefano Zampini ierr = PetscMemzero(B->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 208263c07aadSStefano Zampini B->ops->mult = MatMult_HYPRE; 208363c07aadSStefano Zampini B->ops->multtranspose = MatMultTranspose_HYPRE; 2084414bd5c3SStefano Zampini B->ops->multadd = MatMultAdd_HYPRE; 2085414bd5c3SStefano Zampini B->ops->multtransposeadd = MatMultTransposeAdd_HYPRE; 208663c07aadSStefano Zampini B->ops->setup = MatSetUp_HYPRE; 208763c07aadSStefano Zampini B->ops->destroy = MatDestroy_HYPRE; 208863c07aadSStefano Zampini B->ops->assemblyend = MatAssemblyEnd_HYPRE; 2089c69f721fSFande Kong B->ops->assemblybegin = MatAssemblyBegin_HYPRE; 2090d975228cSstefano_zampini B->ops->setvalues = MatSetValues_HYPRE; 209168ec7858SStefano Zampini B->ops->missingdiagonal = MatMissingDiagonal_HYPRE; 209268ec7858SStefano Zampini B->ops->scale = MatScale_HYPRE; 209368ec7858SStefano Zampini B->ops->zerorowscolumns = MatZeroRowsColumns_HYPRE; 2094c69f721fSFande Kong B->ops->zeroentries = MatZeroEntries_HYPRE; 2095c69f721fSFande Kong B->ops->zerorows = MatZeroRows_HYPRE; 2096c69f721fSFande Kong B->ops->getrow = MatGetRow_HYPRE; 2097c69f721fSFande Kong B->ops->restorerow = MatRestoreRow_HYPRE; 2098c69f721fSFande Kong B->ops->getvalues = MatGetValues_HYPRE; 2099ddbeb582SStefano Zampini B->ops->setoption = MatSetOption_HYPRE; 210045b8d346SStefano Zampini B->ops->duplicate = MatDuplicate_HYPRE; 2101465edc17SStefano Zampini B->ops->copy = MatCopy_HYPRE; 210245b8d346SStefano Zampini B->ops->view = MatView_HYPRE; 21036305df00SStefano Zampini B->ops->getdiagonal = MatGetDiagonal_HYPRE; 2104363d496dSStefano Zampini B->ops->axpy = MatAXPY_HYPRE; 2105*4222ddf1SHong Zhang B->ops->productsetfromoptions = MatProductSetFromOptions_HYPRE; 210645b8d346SStefano Zampini 210745b8d346SStefano Zampini /* build cache for off array entries formed */ 210845b8d346SStefano Zampini ierr = MatStashCreate_Private(PetscObjectComm((PetscObject)B),1,&B->stash);CHKERRQ(ierr); 210963c07aadSStefano Zampini 211063c07aadSStefano Zampini ierr = MPI_Comm_dup(PetscObjectComm((PetscObject)B),&hB->comm);CHKERRQ(ierr); 211163c07aadSStefano Zampini ierr = PetscObjectChangeTypeName((PetscObject)B,MATHYPRE);CHKERRQ(ierr); 211263c07aadSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_hypre_aij_C",MatConvert_HYPRE_AIJ);CHKERRQ(ierr); 21132df22349SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_hypre_is_C",MatConvert_HYPRE_IS);CHKERRQ(ierr); 2114*4222ddf1SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatProductSetFromOptions_seqaij_hypre_C",MatProductSetFromOptions_HYPRE);CHKERRQ(ierr); 2115*4222ddf1SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)B,"MatProductSetFromOptions_mpiaij_hypre_C",MatProductSetFromOptions_HYPRE);CHKERRQ(ierr); 2116d975228cSstefano_zampini ierr = PetscObjectComposeFunction((PetscObject)B,"MatHYPRESetPreallocation_C",MatHYPRESetPreallocation_HYPRE);CHKERRQ(ierr); 2117dd9c0a25Sstefano_zampini ierr = PetscObjectComposeFunction((PetscObject)B,"MatHYPREGetParCSR_C",MatHYPREGetParCSR_HYPRE);CHKERRQ(ierr); 211863c07aadSStefano Zampini PetscFunctionReturn(0); 211963c07aadSStefano Zampini } 212063c07aadSStefano Zampini 2121225daaf8SStefano Zampini static PetscErrorCode hypre_array_destroy(void *ptr) 2122225daaf8SStefano Zampini { 2123225daaf8SStefano Zampini PetscFunctionBegin; 2124e6de0934SSatish Balay hypre_TFree(ptr,HYPRE_MEMORY_HOST); 2125225daaf8SStefano Zampini PetscFunctionReturn(0); 2126225daaf8SStefano Zampini } 2127