xref: /petsc/src/mat/impls/hypre/mhypre.c (revision 4222ddf1d74c63827c599361b3bb0b06ad3944a0)
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