xref: /petsc/src/mat/impls/hypre/mhypre.c (revision e6471dc9ac66eae418b41091cb037e22fd9f25d6)
163c07aadSStefano Zampini 
263c07aadSStefano Zampini /*
363c07aadSStefano Zampini     Creates hypre ijmatrix from PETSc matrix
463c07aadSStefano Zampini */
5225daaf8SStefano Zampini 
6dd9c0a25Sstefano_zampini #include <petscmathypre.h>
763c07aadSStefano Zampini #include <petsc/private/matimpl.h>
863c07aadSStefano Zampini #include <../src/mat/impls/hypre/mhypre.h>
963c07aadSStefano Zampini #include <../src/mat/impls/aij/mpi/mpiaij.h>
1058968eb6SStefano Zampini #include <../src/vec/vec/impls/hypre/vhyp.h>
1158968eb6SStefano Zampini #include <HYPRE.h>
12c1a070e6SStefano Zampini #include <HYPRE_utilities.h>
13cd8bc7baSStefano Zampini #include <_hypre_parcsr_ls.h>
1468ec7858SStefano Zampini #include <_hypre_sstruct_ls.h>
1563c07aadSStefano Zampini 
1675d48cdbSStefano Zampini PETSC_INTERN PetscErrorCode MatPtAP_IS_XAIJ(Mat,Mat,MatReuse,PetscReal,Mat*);
1775d48cdbSStefano Zampini 
1863c07aadSStefano Zampini static PetscErrorCode MatHYPRE_CreateFromMat(Mat,Mat_HYPRE*);
1963c07aadSStefano Zampini static PetscErrorCode MatHYPRE_IJMatrixPreallocate(Mat,Mat,HYPRE_IJMatrix);
2063c07aadSStefano Zampini static PetscErrorCode MatHYPRE_IJMatrixFastCopy_MPIAIJ(Mat,HYPRE_IJMatrix);
2163c07aadSStefano Zampini static PetscErrorCode MatHYPRE_IJMatrixFastCopy_SeqAIJ(Mat,HYPRE_IJMatrix);
2263c07aadSStefano Zampini static PetscErrorCode MatHYPRE_MultKernel_Private(Mat,Vec,Vec,PetscBool);
23225daaf8SStefano Zampini static PetscErrorCode hypre_array_destroy(void*);
24c69f721fSFande Kong PetscErrorCode MatSetValues_HYPRE(Mat, PetscInt,const PetscInt[],PetscInt,const PetscInt[],const PetscScalar[],InsertMode ins);
2563c07aadSStefano Zampini 
2663c07aadSStefano Zampini static PetscErrorCode MatHYPRE_IJMatrixPreallocate(Mat A_d, Mat A_o, HYPRE_IJMatrix ij)
2763c07aadSStefano Zampini {
2863c07aadSStefano Zampini   PetscErrorCode ierr;
2963c07aadSStefano Zampini   PetscInt       i,n_d,n_o;
3063c07aadSStefano Zampini   const PetscInt *ia_d,*ia_o;
3163c07aadSStefano Zampini   PetscBool      done_d=PETSC_FALSE,done_o=PETSC_FALSE;
3263c07aadSStefano Zampini   PetscInt       *nnz_d=NULL,*nnz_o=NULL;
3363c07aadSStefano Zampini 
3463c07aadSStefano Zampini   PetscFunctionBegin;
3563c07aadSStefano Zampini   if (A_d) { /* determine number of nonzero entries in local diagonal part */
3663c07aadSStefano Zampini     ierr = MatGetRowIJ(A_d,0,PETSC_FALSE,PETSC_FALSE,&n_d,&ia_d,NULL,&done_d);CHKERRQ(ierr);
3763c07aadSStefano Zampini     if (done_d) {
3863c07aadSStefano Zampini       ierr = PetscMalloc1(n_d,&nnz_d);CHKERRQ(ierr);
3963c07aadSStefano Zampini       for (i=0; i<n_d; i++) {
4063c07aadSStefano Zampini         nnz_d[i] = ia_d[i+1] - ia_d[i];
4163c07aadSStefano Zampini       }
4263c07aadSStefano Zampini     }
4363c07aadSStefano Zampini     ierr = MatRestoreRowIJ(A_d,0,PETSC_FALSE,PETSC_FALSE,NULL,&ia_d,NULL,&done_d);CHKERRQ(ierr);
4463c07aadSStefano Zampini   }
4563c07aadSStefano Zampini   if (A_o) { /* determine number of nonzero entries in local off-diagonal part */
4663c07aadSStefano Zampini     ierr = MatGetRowIJ(A_o,0,PETSC_FALSE,PETSC_FALSE,&n_o,&ia_o,NULL,&done_o);CHKERRQ(ierr);
4763c07aadSStefano Zampini     if (done_o) {
4863c07aadSStefano Zampini       ierr = PetscMalloc1(n_o,&nnz_o);CHKERRQ(ierr);
4963c07aadSStefano Zampini       for (i=0; i<n_o; i++) {
5063c07aadSStefano Zampini         nnz_o[i] = ia_o[i+1] - ia_o[i];
5163c07aadSStefano Zampini       }
5263c07aadSStefano Zampini     }
5363c07aadSStefano Zampini     ierr = MatRestoreRowIJ(A_o,0,PETSC_FALSE,PETSC_FALSE,&n_o,&ia_o,NULL,&done_o);CHKERRQ(ierr);
5463c07aadSStefano Zampini   }
5563c07aadSStefano Zampini   if (done_d) {    /* set number of nonzeros in HYPRE IJ matrix */
5663c07aadSStefano Zampini     if (!done_o) { /* only diagonal part */
5763c07aadSStefano Zampini       ierr = PetscMalloc1(n_d,&nnz_o);CHKERRQ(ierr);
5863c07aadSStefano Zampini       for (i=0; i<n_d; i++) {
5963c07aadSStefano Zampini         nnz_o[i] = 0;
6063c07aadSStefano Zampini       }
6163c07aadSStefano Zampini     }
6263c07aadSStefano Zampini     PetscStackCallStandard(HYPRE_IJMatrixSetDiagOffdSizes,(ij,(HYPRE_Int *)nnz_d,(HYPRE_Int *)nnz_o));
6363c07aadSStefano Zampini     ierr = PetscFree(nnz_d);CHKERRQ(ierr);
6463c07aadSStefano Zampini     ierr = PetscFree(nnz_o);CHKERRQ(ierr);
6563c07aadSStefano Zampini   }
6663c07aadSStefano Zampini   PetscFunctionReturn(0);
6763c07aadSStefano Zampini }
6863c07aadSStefano Zampini 
6963c07aadSStefano Zampini static PetscErrorCode MatHYPRE_CreateFromMat(Mat A, Mat_HYPRE *hA)
7063c07aadSStefano Zampini {
7163c07aadSStefano Zampini   PetscErrorCode ierr;
7263c07aadSStefano Zampini   PetscInt       rstart,rend,cstart,cend;
7363c07aadSStefano Zampini 
7463c07aadSStefano Zampini   PetscFunctionBegin;
75d975228cSstefano_zampini   ierr   = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr);
76d975228cSstefano_zampini   ierr   = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr);
7763c07aadSStefano Zampini   rstart = A->rmap->rstart;
7863c07aadSStefano Zampini   rend   = A->rmap->rend;
7963c07aadSStefano Zampini   cstart = A->cmap->rstart;
8063c07aadSStefano Zampini   cend   = A->cmap->rend;
8163c07aadSStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixCreate,(hA->comm,rstart,rend-1,cstart,cend-1,&hA->ij));
8263c07aadSStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(hA->ij,HYPRE_PARCSR));
8363c07aadSStefano Zampini   {
8463c07aadSStefano Zampini     PetscBool      same;
8563c07aadSStefano Zampini     Mat            A_d,A_o;
8663c07aadSStefano Zampini     const PetscInt *colmap;
8763c07aadSStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)A,MATMPIAIJ,&same);CHKERRQ(ierr);
8863c07aadSStefano Zampini     if (same) {
8963c07aadSStefano Zampini       ierr = MatMPIAIJGetSeqAIJ(A,&A_d,&A_o,&colmap);CHKERRQ(ierr);
9063c07aadSStefano Zampini       ierr = MatHYPRE_IJMatrixPreallocate(A_d,A_o,hA->ij);CHKERRQ(ierr);
9163c07aadSStefano Zampini       PetscFunctionReturn(0);
9263c07aadSStefano Zampini     }
9363c07aadSStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)A,MATMPIBAIJ,&same);CHKERRQ(ierr);
9463c07aadSStefano Zampini     if (same) {
9563c07aadSStefano Zampini       ierr = MatMPIBAIJGetSeqBAIJ(A,&A_d,&A_o,&colmap);CHKERRQ(ierr);
9663c07aadSStefano Zampini       ierr = MatHYPRE_IJMatrixPreallocate(A_d,A_o,hA->ij);CHKERRQ(ierr);
9763c07aadSStefano Zampini       PetscFunctionReturn(0);
9863c07aadSStefano Zampini     }
9963c07aadSStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)A,MATSEQAIJ,&same);CHKERRQ(ierr);
10063c07aadSStefano Zampini     if (same) {
10163c07aadSStefano Zampini       ierr = MatHYPRE_IJMatrixPreallocate(A,NULL,hA->ij);CHKERRQ(ierr);
10263c07aadSStefano Zampini       PetscFunctionReturn(0);
10363c07aadSStefano Zampini     }
10463c07aadSStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)A,MATSEQBAIJ,&same);CHKERRQ(ierr);
10563c07aadSStefano Zampini     if (same) {
10663c07aadSStefano Zampini       ierr = MatHYPRE_IJMatrixPreallocate(A,NULL,hA->ij);CHKERRQ(ierr);
10763c07aadSStefano Zampini       PetscFunctionReturn(0);
10863c07aadSStefano Zampini     }
10963c07aadSStefano Zampini   }
11063c07aadSStefano Zampini   PetscFunctionReturn(0);
11163c07aadSStefano Zampini }
11263c07aadSStefano Zampini 
11363c07aadSStefano Zampini static PetscErrorCode MatHYPRE_IJMatrixCopy(Mat A, HYPRE_IJMatrix ij)
11463c07aadSStefano Zampini {
11563c07aadSStefano Zampini   PetscErrorCode    ierr;
11663c07aadSStefano Zampini   PetscInt          i,rstart,rend,ncols,nr,nc;
11763c07aadSStefano Zampini   const PetscScalar *values;
11863c07aadSStefano Zampini   const PetscInt    *cols;
11963c07aadSStefano Zampini   PetscBool         flg;
12063c07aadSStefano Zampini 
12163c07aadSStefano Zampini   PetscFunctionBegin;
12263c07aadSStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)A,MATMPIAIJ,&flg);CHKERRQ(ierr);
12363c07aadSStefano Zampini   ierr = MatGetSize(A,&nr,&nc);CHKERRQ(ierr);
12463c07aadSStefano Zampini   if (flg && nr == nc) {
12563c07aadSStefano Zampini     ierr = MatHYPRE_IJMatrixFastCopy_MPIAIJ(A,ij);CHKERRQ(ierr);
12663c07aadSStefano Zampini     PetscFunctionReturn(0);
12763c07aadSStefano Zampini   }
12863c07aadSStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)A,MATSEQAIJ,&flg);CHKERRQ(ierr);
12963c07aadSStefano Zampini   if (flg) {
13063c07aadSStefano Zampini     ierr = MatHYPRE_IJMatrixFastCopy_SeqAIJ(A,ij);CHKERRQ(ierr);
13163c07aadSStefano Zampini     PetscFunctionReturn(0);
13263c07aadSStefano Zampini   }
13363c07aadSStefano Zampini 
13463c07aadSStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixInitialize,(ij));
13563c07aadSStefano Zampini   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
13663c07aadSStefano Zampini   for (i=rstart; i<rend; i++) {
13763c07aadSStefano Zampini     ierr = MatGetRow(A,i,&ncols,&cols,&values);CHKERRQ(ierr);
138e3977e59Sstefano_zampini     if (ncols) {
13963c07aadSStefano Zampini       PetscStackCallStandard(HYPRE_IJMatrixSetValues,(ij,1,(HYPRE_Int *)&ncols,(HYPRE_Int *)&i,(HYPRE_Int *)cols,values));
140e3977e59Sstefano_zampini     }
14163c07aadSStefano Zampini     ierr = MatRestoreRow(A,i,&ncols,&cols,&values);CHKERRQ(ierr);
14263c07aadSStefano Zampini   }
14363c07aadSStefano Zampini   PetscFunctionReturn(0);
14463c07aadSStefano Zampini }
14563c07aadSStefano Zampini 
14663c07aadSStefano Zampini static PetscErrorCode MatHYPRE_IJMatrixFastCopy_SeqAIJ(Mat A, HYPRE_IJMatrix ij)
14763c07aadSStefano Zampini {
14863c07aadSStefano Zampini   PetscErrorCode        ierr;
14963c07aadSStefano Zampini   Mat_SeqAIJ            *pdiag = (Mat_SeqAIJ*)A->data;
15058968eb6SStefano Zampini   HYPRE_Int             type;
15163c07aadSStefano Zampini   hypre_ParCSRMatrix    *par_matrix;
15263c07aadSStefano Zampini   hypre_AuxParCSRMatrix *aux_matrix;
15363c07aadSStefano Zampini   hypre_CSRMatrix       *hdiag;
15463c07aadSStefano Zampini 
15563c07aadSStefano Zampini   PetscFunctionBegin;
15663c07aadSStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixInitialize,(ij));
157ea9daf28SStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(ij,&type));
158ea9daf28SStefano Zampini   if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported");
159ea9daf28SStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObject,(ij,(void**)&par_matrix));
16063c07aadSStefano Zampini   hdiag = hypre_ParCSRMatrixDiag(par_matrix);
16163c07aadSStefano Zampini   /*
16263c07aadSStefano Zampini        this is the Hack part where we monkey directly with the hypre datastructures
16363c07aadSStefano Zampini   */
1641d4906efSStefano Zampini   ierr = PetscMemcpy(hdiag->i,pdiag->i,(A->rmap->n + 1)*sizeof(PetscInt));CHKERRQ(ierr);
1651d4906efSStefano Zampini   ierr = PetscMemcpy(hdiag->j,pdiag->j,pdiag->nz*sizeof(PetscInt));CHKERRQ(ierr);
1661d4906efSStefano Zampini   ierr = PetscMemcpy(hdiag->data,pdiag->a,pdiag->nz*sizeof(PetscScalar));CHKERRQ(ierr);
167ea9daf28SStefano Zampini 
168ea9daf28SStefano Zampini   aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(ij);
16963c07aadSStefano Zampini   hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 0;
17063c07aadSStefano Zampini   PetscFunctionReturn(0);
17163c07aadSStefano Zampini }
17263c07aadSStefano Zampini 
17363c07aadSStefano Zampini static PetscErrorCode MatHYPRE_IJMatrixFastCopy_MPIAIJ(Mat A, HYPRE_IJMatrix ij)
17463c07aadSStefano Zampini {
17563c07aadSStefano Zampini   PetscErrorCode        ierr;
17663c07aadSStefano Zampini   Mat_MPIAIJ            *pA = (Mat_MPIAIJ*)A->data;
17763c07aadSStefano Zampini   Mat_SeqAIJ            *pdiag,*poffd;
17863c07aadSStefano Zampini   PetscInt              i,*garray = pA->garray,*jj,cstart,*pjj;
17958968eb6SStefano Zampini   HYPRE_Int             type;
18063c07aadSStefano Zampini   hypre_ParCSRMatrix    *par_matrix;
18163c07aadSStefano Zampini   hypre_AuxParCSRMatrix *aux_matrix;
18263c07aadSStefano Zampini   hypre_CSRMatrix       *hdiag,*hoffd;
18363c07aadSStefano Zampini 
18463c07aadSStefano Zampini   PetscFunctionBegin;
18563c07aadSStefano Zampini   pdiag = (Mat_SeqAIJ*) pA->A->data;
18663c07aadSStefano Zampini   poffd = (Mat_SeqAIJ*) pA->B->data;
18763c07aadSStefano Zampini   /* cstart is only valid for square MPIAIJ layed out in the usual way */
18863c07aadSStefano Zampini   ierr = MatGetOwnershipRange(A,&cstart,NULL);CHKERRQ(ierr);
18963c07aadSStefano Zampini 
19063c07aadSStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixInitialize,(ij));
191ea9daf28SStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(ij,&type));
192ea9daf28SStefano Zampini   if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported");
193ea9daf28SStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObject,(ij,(void**)&par_matrix));
19463c07aadSStefano Zampini   hdiag = hypre_ParCSRMatrixDiag(par_matrix);
19563c07aadSStefano Zampini   hoffd = hypre_ParCSRMatrixOffd(par_matrix);
19663c07aadSStefano Zampini 
19763c07aadSStefano Zampini   /*
19863c07aadSStefano Zampini        this is the Hack part where we monkey directly with the hypre datastructures
19963c07aadSStefano Zampini   */
20063c07aadSStefano Zampini   ierr = PetscMemcpy(hdiag->i,pdiag->i,(pA->A->rmap->n + 1)*sizeof(PetscInt));
20163c07aadSStefano Zampini   /* need to shift the diag column indices (hdiag->j) back to global numbering since hypre is expecting this */
20263c07aadSStefano Zampini   jj  = (PetscInt*)hdiag->j;
20363c07aadSStefano Zampini   pjj = (PetscInt*)pdiag->j;
20463c07aadSStefano Zampini   for (i=0; i<pdiag->nz; i++) jj[i] = cstart + pjj[i];
20563c07aadSStefano Zampini   ierr = PetscMemcpy(hdiag->data,pdiag->a,pdiag->nz*sizeof(PetscScalar));
20663c07aadSStefano Zampini   ierr = PetscMemcpy(hoffd->i,poffd->i,(pA->A->rmap->n + 1)*sizeof(PetscInt));
20763c07aadSStefano Zampini   /* need to move the offd column indices (hoffd->j) back to global numbering since hypre is expecting this
20863c07aadSStefano Zampini      If we hacked a hypre a bit more we might be able to avoid this step */
20963c07aadSStefano Zampini   jj  = (PetscInt*) hoffd->j;
21063c07aadSStefano Zampini   pjj = (PetscInt*) poffd->j;
21163c07aadSStefano Zampini   for (i=0; i<poffd->nz; i++) jj[i] = garray[pjj[i]];
21263c07aadSStefano Zampini   ierr = PetscMemcpy(hoffd->data,poffd->a,poffd->nz*sizeof(PetscScalar));
21363c07aadSStefano Zampini 
214ea9daf28SStefano Zampini   aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(ij);
21563c07aadSStefano Zampini   hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 0;
21663c07aadSStefano Zampini   PetscFunctionReturn(0);
21763c07aadSStefano Zampini }
21863c07aadSStefano Zampini 
2192df22349SStefano Zampini static PetscErrorCode MatConvert_HYPRE_IS(Mat A, MatType mtype, MatReuse reuse, Mat* B)
2202df22349SStefano Zampini {
2212df22349SStefano Zampini   Mat_HYPRE*             mhA = (Mat_HYPRE*)(A->data);
2222df22349SStefano Zampini   Mat                    lA;
2232df22349SStefano Zampini   ISLocalToGlobalMapping rl2g,cl2g;
2242df22349SStefano Zampini   IS                     is;
2252df22349SStefano Zampini   hypre_ParCSRMatrix     *hA;
2262df22349SStefano Zampini   hypre_CSRMatrix        *hdiag,*hoffd;
2272df22349SStefano Zampini   MPI_Comm               comm;
2282df22349SStefano Zampini   PetscScalar            *hdd,*hod,*aa,*data;
2297d968826Sstefano_zampini   HYPRE_Int              *col_map_offd,*hdi,*hdj,*hoi,*hoj;
2302df22349SStefano Zampini   PetscInt               *ii,*jj,*iptr,*jptr;
2312df22349SStefano Zampini   PetscInt               cum,dr,dc,oc,str,stc,nnz,i,jd,jo,M,N;
23258968eb6SStefano Zampini   HYPRE_Int              type;
2332df22349SStefano Zampini   PetscErrorCode         ierr;
2342df22349SStefano Zampini 
2352df22349SStefano Zampini   PetscFunctionBegin;
236a1787963SStefano Zampini   comm = PetscObjectComm((PetscObject)A);
2372df22349SStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(mhA->ij,&type));
2382df22349SStefano Zampini   if (type != HYPRE_PARCSR) SETERRQ(comm,PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported");
2392df22349SStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObject,(mhA->ij,(void**)&hA));
2402df22349SStefano Zampini   M     = hypre_ParCSRMatrixGlobalNumRows(hA);
2412df22349SStefano Zampini   N     = hypre_ParCSRMatrixGlobalNumCols(hA);
2422df22349SStefano Zampini   str   = hypre_ParCSRMatrixFirstRowIndex(hA);
2432df22349SStefano Zampini   stc   = hypre_ParCSRMatrixFirstColDiag(hA);
2442df22349SStefano Zampini   hdiag = hypre_ParCSRMatrixDiag(hA);
2452df22349SStefano Zampini   hoffd = hypre_ParCSRMatrixOffd(hA);
2462df22349SStefano Zampini   dr    = hypre_CSRMatrixNumRows(hdiag);
2472df22349SStefano Zampini   dc    = hypre_CSRMatrixNumCols(hdiag);
2482df22349SStefano Zampini   nnz   = hypre_CSRMatrixNumNonzeros(hdiag);
2492df22349SStefano Zampini   hdi   = hypre_CSRMatrixI(hdiag);
2502df22349SStefano Zampini   hdj   = hypre_CSRMatrixJ(hdiag);
2512df22349SStefano Zampini   hdd   = hypre_CSRMatrixData(hdiag);
2522df22349SStefano Zampini   oc    = hypre_CSRMatrixNumCols(hoffd);
2532df22349SStefano Zampini   nnz  += hypre_CSRMatrixNumNonzeros(hoffd);
2542df22349SStefano Zampini   hoi   = hypre_CSRMatrixI(hoffd);
2552df22349SStefano Zampini   hoj   = hypre_CSRMatrixJ(hoffd);
2562df22349SStefano Zampini   hod   = hypre_CSRMatrixData(hoffd);
2572df22349SStefano Zampini   if (reuse != MAT_REUSE_MATRIX) {
2582df22349SStefano Zampini     PetscInt *aux;
2592df22349SStefano Zampini 
2602df22349SStefano Zampini     /* generate l2g maps for rows and cols */
2612df22349SStefano Zampini     ierr = ISCreateStride(comm,dr,str,1,&is);CHKERRQ(ierr);
2622df22349SStefano Zampini     ierr = ISLocalToGlobalMappingCreateIS(is,&rl2g);CHKERRQ(ierr);
2632df22349SStefano Zampini     ierr = ISDestroy(&is);CHKERRQ(ierr);
2642df22349SStefano Zampini     col_map_offd = hypre_ParCSRMatrixColMapOffd(hA);
2652df22349SStefano Zampini     ierr = PetscMalloc1(dc+oc,&aux);CHKERRQ(ierr);
2662df22349SStefano Zampini     for (i=0; i<dc; i++) aux[i] = i+stc;
2672df22349SStefano Zampini     for (i=0; i<oc; i++) aux[i+dc] = col_map_offd[i];
2682df22349SStefano Zampini     ierr = ISCreateGeneral(comm,dc+oc,aux,PETSC_OWN_POINTER,&is);CHKERRQ(ierr);
2692df22349SStefano Zampini     ierr = ISLocalToGlobalMappingCreateIS(is,&cl2g);CHKERRQ(ierr);
2702df22349SStefano Zampini     ierr = ISDestroy(&is);CHKERRQ(ierr);
2712df22349SStefano Zampini     /* create MATIS object */
2722df22349SStefano Zampini     ierr = MatCreate(comm,B);CHKERRQ(ierr);
2732df22349SStefano Zampini     ierr = MatSetSizes(*B,dr,dc,M,N);CHKERRQ(ierr);
2742df22349SStefano Zampini     ierr = MatSetType(*B,MATIS);CHKERRQ(ierr);
2752df22349SStefano Zampini     ierr = MatSetLocalToGlobalMapping(*B,rl2g,cl2g);CHKERRQ(ierr);
2762df22349SStefano Zampini     ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr);
2772df22349SStefano Zampini     ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr);
2782df22349SStefano Zampini 
2792df22349SStefano Zampini     /* allocate CSR for local matrix */
2802df22349SStefano Zampini     ierr = PetscMalloc1(dr+1,&iptr);CHKERRQ(ierr);
2812df22349SStefano Zampini     ierr = PetscMalloc1(nnz,&jptr);CHKERRQ(ierr);
2822df22349SStefano Zampini     ierr = PetscMalloc1(nnz,&data);CHKERRQ(ierr);
2832df22349SStefano Zampini   } else {
2842df22349SStefano Zampini     PetscInt  nr;
2852df22349SStefano Zampini     PetscBool done;
2862df22349SStefano Zampini     ierr = MatISGetLocalMat(*B,&lA);CHKERRQ(ierr);
2872df22349SStefano Zampini     ierr = MatGetRowIJ(lA,0,PETSC_FALSE,PETSC_FALSE,&nr,(const PetscInt**)&iptr,(const PetscInt**)&jptr,&done);CHKERRQ(ierr);
2882df22349SStefano 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);
2892df22349SStefano 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);
2902df22349SStefano Zampini     ierr = MatSeqAIJGetArray(lA,&data);CHKERRQ(ierr);
2912df22349SStefano Zampini   }
2922df22349SStefano Zampini   /* merge local matrices */
2932df22349SStefano Zampini   ii   = iptr;
2942df22349SStefano Zampini   jj   = jptr;
2952df22349SStefano Zampini   aa   = data;
2962df22349SStefano Zampini   *ii  = *(hdi++) + *(hoi++);
2972df22349SStefano Zampini   for (jd=0,jo=0,cum=0; *ii<nnz; cum++) {
2982df22349SStefano Zampini     PetscScalar *aold = aa;
2992df22349SStefano Zampini     PetscInt    *jold = jj,nc = jd+jo;
3002df22349SStefano Zampini     for (; jd<*hdi; jd++) { *jj++ = *hdj++;      *aa++ = *hdd++; }
3012df22349SStefano Zampini     for (; jo<*hoi; jo++) { *jj++ = *hoj++ + dc; *aa++ = *hod++; }
3022df22349SStefano Zampini     *(++ii) = *(hdi++) + *(hoi++);
3032df22349SStefano Zampini     ierr = PetscSortIntWithScalarArray(jd+jo-nc,jold,aold);CHKERRQ(ierr);
3042df22349SStefano Zampini   }
3052df22349SStefano Zampini   for (; cum<dr; cum++) *(++ii) = nnz;
3062df22349SStefano Zampini   if (reuse != MAT_REUSE_MATRIX) {
307a033916dSStefano Zampini     Mat_SeqAIJ* a;
308a033916dSStefano Zampini 
3092df22349SStefano Zampini     ierr = MatCreateSeqAIJWithArrays(PETSC_COMM_SELF,dr,dc+oc,iptr,jptr,data,&lA);CHKERRQ(ierr);
3102df22349SStefano Zampini     ierr = MatISSetLocalMat(*B,lA);CHKERRQ(ierr);
311a033916dSStefano Zampini     /* hack SeqAIJ */
312a033916dSStefano Zampini     a          = (Mat_SeqAIJ*)(lA->data);
313a033916dSStefano Zampini     a->free_a  = PETSC_TRUE;
314a033916dSStefano Zampini     a->free_ij = PETSC_TRUE;
3152df22349SStefano Zampini     ierr = MatDestroy(&lA);CHKERRQ(ierr);
3162df22349SStefano Zampini   }
3172df22349SStefano Zampini   ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
3182df22349SStefano Zampini   ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
3192df22349SStefano Zampini   if (reuse == MAT_INPLACE_MATRIX) {
3202df22349SStefano Zampini     ierr = MatHeaderReplace(A,B);CHKERRQ(ierr);
3212df22349SStefano Zampini   }
3222df22349SStefano Zampini   PetscFunctionReturn(0);
3232df22349SStefano Zampini }
3242df22349SStefano Zampini 
32563c07aadSStefano Zampini PETSC_INTERN PetscErrorCode MatConvert_AIJ_HYPRE(Mat A, MatType type, MatReuse reuse, Mat *B)
32663c07aadSStefano Zampini {
32763c07aadSStefano Zampini   Mat_HYPRE      *hB;
32863c07aadSStefano Zampini   MPI_Comm       comm = PetscObjectComm((PetscObject)A);
32963c07aadSStefano Zampini   PetscErrorCode ierr;
33063c07aadSStefano Zampini 
33163c07aadSStefano Zampini   PetscFunctionBegin;
33263c07aadSStefano Zampini   if (reuse == MAT_INPLACE_MATRIX) SETERRQ(comm,PETSC_ERR_SUP,"Unsupported MAT_INPLACE_MATRIX");
33363c07aadSStefano Zampini   if (reuse == MAT_REUSE_MATRIX) {
33463c07aadSStefano Zampini     /* always destroy the old matrix and create a new memory;
33563c07aadSStefano Zampini        hope this does not churn the memory too much. The problem
33663c07aadSStefano Zampini        is I do not know if it is possible to put the matrix back to
33763c07aadSStefano Zampini        its initial state so that we can directly copy the values
33863c07aadSStefano Zampini        the second time through. */
33963c07aadSStefano Zampini     hB = (Mat_HYPRE*)((*B)->data);
34063c07aadSStefano Zampini     PetscStackCallStandard(HYPRE_IJMatrixDestroy,(hB->ij));
34163c07aadSStefano Zampini   } else {
34263c07aadSStefano Zampini     ierr = MatCreate(comm,B);CHKERRQ(ierr);
34363c07aadSStefano Zampini     ierr = MatSetType(*B,MATHYPRE);CHKERRQ(ierr);
34463c07aadSStefano Zampini     ierr = MatSetSizes(*B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
34563c07aadSStefano Zampini     hB   = (Mat_HYPRE*)((*B)->data);
34663c07aadSStefano Zampini   }
34763c07aadSStefano Zampini   ierr = MatHYPRE_CreateFromMat(A,hB);CHKERRQ(ierr);
34863c07aadSStefano Zampini   ierr = MatHYPRE_IJMatrixCopy(A,hB->ij);CHKERRQ(ierr);
3494ec6421dSstefano_zampini   (*B)->preallocated = PETSC_TRUE;
35063c07aadSStefano Zampini   ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
35163c07aadSStefano Zampini   ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
35263c07aadSStefano Zampini   PetscFunctionReturn(0);
35363c07aadSStefano Zampini }
35463c07aadSStefano Zampini 
355ea9daf28SStefano Zampini static PetscErrorCode MatConvert_HYPRE_AIJ(Mat A, MatType mtype, MatReuse reuse, Mat *B)
35663c07aadSStefano Zampini {
35763c07aadSStefano Zampini   Mat_HYPRE          *hA = (Mat_HYPRE*)A->data;
35863c07aadSStefano Zampini   hypre_ParCSRMatrix *parcsr;
35963c07aadSStefano Zampini   hypre_CSRMatrix    *hdiag,*hoffd;
36063c07aadSStefano Zampini   MPI_Comm           comm;
36163c07aadSStefano Zampini   PetscScalar        *da,*oa,*aptr;
36263c07aadSStefano Zampini   PetscInt           *dii,*djj,*oii,*ojj,*iptr;
36363c07aadSStefano Zampini   PetscInt           i,dnnz,onnz,m,n;
36458968eb6SStefano Zampini   HYPRE_Int          type;
36563c07aadSStefano Zampini   PetscMPIInt        size;
36663c07aadSStefano Zampini   PetscErrorCode     ierr;
36763c07aadSStefano Zampini 
36863c07aadSStefano Zampini   PetscFunctionBegin;
36963c07aadSStefano Zampini   comm = PetscObjectComm((PetscObject)A);
37063c07aadSStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type));
37163c07aadSStefano Zampini   if (type != HYPRE_PARCSR) SETERRQ(comm,PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported");
37263c07aadSStefano Zampini   if (reuse == MAT_REUSE_MATRIX) {
37363c07aadSStefano Zampini     PetscBool ismpiaij,isseqaij;
37463c07aadSStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)*B,MATMPIAIJ,&ismpiaij);CHKERRQ(ierr);
3754099cc6bSBarry Smith     ierr = PetscObjectBaseTypeCompare((PetscObject)*B,MATSEQAIJ,&isseqaij);CHKERRQ(ierr);
37663c07aadSStefano Zampini     if (!ismpiaij && !isseqaij) SETERRQ(comm,PETSC_ERR_SUP,"Only MATMPIAIJ or MATSEQAIJ are supported");
37763c07aadSStefano Zampini   }
37863c07aadSStefano Zampini   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
37963c07aadSStefano Zampini 
38063c07aadSStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)&parcsr));
38163c07aadSStefano Zampini   hdiag = hypre_ParCSRMatrixDiag(parcsr);
38263c07aadSStefano Zampini   hoffd = hypre_ParCSRMatrixOffd(parcsr);
38363c07aadSStefano Zampini   m     = hypre_CSRMatrixNumRows(hdiag);
38463c07aadSStefano Zampini   n     = hypre_CSRMatrixNumCols(hdiag);
38563c07aadSStefano Zampini   dnnz  = hypre_CSRMatrixNumNonzeros(hdiag);
38663c07aadSStefano Zampini   onnz  = hypre_CSRMatrixNumNonzeros(hoffd);
387225daaf8SStefano Zampini   if (reuse == MAT_INITIAL_MATRIX) {
38863c07aadSStefano Zampini     ierr = PetscMalloc1(m+1,&dii);CHKERRQ(ierr);
38963c07aadSStefano Zampini     ierr = PetscMalloc1(dnnz,&djj);CHKERRQ(ierr);
39063c07aadSStefano Zampini     ierr = PetscMalloc1(dnnz,&da);CHKERRQ(ierr);
391225daaf8SStefano Zampini   } else if (reuse == MAT_REUSE_MATRIX) {
39263c07aadSStefano Zampini     PetscInt  nr;
39363c07aadSStefano Zampini     PetscBool done;
39463c07aadSStefano Zampini     if (size > 1) {
39563c07aadSStefano Zampini       Mat_MPIAIJ *b = (Mat_MPIAIJ*)((*B)->data);
39663c07aadSStefano Zampini 
39763c07aadSStefano Zampini       ierr = MatGetRowIJ(b->A,0,PETSC_FALSE,PETSC_FALSE,&nr,(const PetscInt**)&dii,(const PetscInt**)&djj,&done);CHKERRQ(ierr);
39863c07aadSStefano 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);
39963c07aadSStefano 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);
40063c07aadSStefano Zampini       ierr = MatSeqAIJGetArray(b->A,&da);CHKERRQ(ierr);
40163c07aadSStefano Zampini     } else {
40263c07aadSStefano Zampini       ierr = MatGetRowIJ(*B,0,PETSC_FALSE,PETSC_FALSE,&nr,(const PetscInt**)&dii,(const PetscInt**)&djj,&done);CHKERRQ(ierr);
40363c07aadSStefano Zampini       if (nr != m) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"Cannot reuse mat: invalid number of local rows! %D != %D",nr,m);
40463c07aadSStefano 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);
40563c07aadSStefano Zampini       ierr = MatSeqAIJGetArray(*B,&da);CHKERRQ(ierr);
40663c07aadSStefano Zampini     }
407225daaf8SStefano Zampini   } else { /* MAT_INPLACE_MATRIX */
4087d968826Sstefano_zampini     dii = (PetscInt*)hypre_CSRMatrixI(hdiag);
4097d968826Sstefano_zampini     djj = (PetscInt*)hypre_CSRMatrixJ(hdiag);
410225daaf8SStefano Zampini     da  = hypre_CSRMatrixData(hdiag);
41163c07aadSStefano Zampini   }
41263c07aadSStefano Zampini   ierr = PetscMemcpy(dii,hypre_CSRMatrixI(hdiag),(m+1)*sizeof(PetscInt));CHKERRQ(ierr);
41363c07aadSStefano Zampini   ierr = PetscMemcpy(djj,hypre_CSRMatrixJ(hdiag),dnnz*sizeof(PetscInt));CHKERRQ(ierr);
41463c07aadSStefano Zampini   ierr = PetscMemcpy(da,hypre_CSRMatrixData(hdiag),dnnz*sizeof(PetscScalar));CHKERRQ(ierr);
41563c07aadSStefano Zampini   iptr = djj;
41663c07aadSStefano Zampini   aptr = da;
41763c07aadSStefano Zampini   for (i=0; i<m; i++) {
41863c07aadSStefano Zampini     PetscInt nc = dii[i+1]-dii[i];
41963c07aadSStefano Zampini     ierr = PetscSortIntWithScalarArray(nc,iptr,aptr);CHKERRQ(ierr);
42063c07aadSStefano Zampini     iptr += nc;
42163c07aadSStefano Zampini     aptr += nc;
42263c07aadSStefano Zampini   }
42363c07aadSStefano Zampini   if (size > 1) {
4247d968826Sstefano_zampini     HYPRE_Int *offdj,*coffd;
42563c07aadSStefano Zampini 
426225daaf8SStefano Zampini     if (reuse == MAT_INITIAL_MATRIX) {
42763c07aadSStefano Zampini       ierr = PetscMalloc1(m+1,&oii);CHKERRQ(ierr);
42863c07aadSStefano Zampini       ierr = PetscMalloc1(onnz,&ojj);CHKERRQ(ierr);
42963c07aadSStefano Zampini       ierr = PetscMalloc1(onnz,&oa);CHKERRQ(ierr);
430225daaf8SStefano Zampini     } else if (reuse == MAT_REUSE_MATRIX) {
43163c07aadSStefano Zampini       Mat_MPIAIJ *b = (Mat_MPIAIJ*)((*B)->data);
43263c07aadSStefano Zampini       PetscInt   nr,hr = hypre_CSRMatrixNumRows(hoffd);
43363c07aadSStefano Zampini       PetscBool  done;
43463c07aadSStefano Zampini 
43563c07aadSStefano Zampini       ierr = MatGetRowIJ(b->B,0,PETSC_FALSE,PETSC_FALSE,&nr,(const PetscInt**)&oii,(const PetscInt**)&ojj,&done);CHKERRQ(ierr);
43663c07aadSStefano 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);
43763c07aadSStefano 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);
43863c07aadSStefano Zampini       ierr = MatSeqAIJGetArray(b->B,&oa);CHKERRQ(ierr);
439225daaf8SStefano Zampini     } else { /* MAT_INPLACE_MATRIX */
4407d968826Sstefano_zampini       oii = (PetscInt*)hypre_CSRMatrixI(hoffd);
4417d968826Sstefano_zampini       ojj = (PetscInt*)hypre_CSRMatrixJ(hoffd);
442225daaf8SStefano Zampini       oa  = hypre_CSRMatrixData(hoffd);
44363c07aadSStefano Zampini     }
44463c07aadSStefano Zampini     ierr  = PetscMemcpy(oii,hypre_CSRMatrixI(hoffd),(m+1)*sizeof(PetscInt));CHKERRQ(ierr);
44563c07aadSStefano Zampini     offdj = hypre_CSRMatrixJ(hoffd);
44663c07aadSStefano Zampini     coffd = hypre_ParCSRMatrixColMapOffd(parcsr);
44763c07aadSStefano Zampini     for (i=0; i<onnz; i++) ojj[i] = coffd[offdj[i]];
44863c07aadSStefano Zampini     ierr = PetscMemcpy(oa,hypre_CSRMatrixData(hoffd),onnz*sizeof(PetscScalar));CHKERRQ(ierr);
44963c07aadSStefano Zampini     iptr = ojj;
45063c07aadSStefano Zampini     aptr = oa;
45163c07aadSStefano Zampini     for (i=0; i<m; i++) {
45263c07aadSStefano Zampini        PetscInt nc = oii[i+1]-oii[i];
45363c07aadSStefano Zampini        ierr = PetscSortIntWithScalarArray(nc,iptr,aptr);CHKERRQ(ierr);
45463c07aadSStefano Zampini        iptr += nc;
45563c07aadSStefano Zampini        aptr += nc;
45663c07aadSStefano Zampini     }
457225daaf8SStefano Zampini     if (reuse == MAT_INITIAL_MATRIX) {
45863c07aadSStefano Zampini       Mat_MPIAIJ *b;
45963c07aadSStefano Zampini       Mat_SeqAIJ *d,*o;
460225daaf8SStefano Zampini 
46141073d71Sstefano_zampini       ierr = MatCreateMPIAIJWithSplitArrays(comm,m,n,PETSC_DECIDE,PETSC_DECIDE,dii,djj,da,oii,ojj,oa,B);CHKERRQ(ierr);
46263c07aadSStefano Zampini       /* hack MPIAIJ */
46363c07aadSStefano Zampini       b          = (Mat_MPIAIJ*)((*B)->data);
46463c07aadSStefano Zampini       d          = (Mat_SeqAIJ*)b->A->data;
46563c07aadSStefano Zampini       o          = (Mat_SeqAIJ*)b->B->data;
46663c07aadSStefano Zampini       d->free_a  = PETSC_TRUE;
46763c07aadSStefano Zampini       d->free_ij = PETSC_TRUE;
46863c07aadSStefano Zampini       o->free_a  = PETSC_TRUE;
46963c07aadSStefano Zampini       o->free_ij = PETSC_TRUE;
470225daaf8SStefano Zampini     } else if (reuse == MAT_INPLACE_MATRIX) {
471225daaf8SStefano Zampini       Mat T;
47241073d71Sstefano_zampini       ierr = MatCreateMPIAIJWithSplitArrays(comm,m,n,PETSC_DECIDE,PETSC_DECIDE,dii,djj,da,oii,ojj,oa,&T);CHKERRQ(ierr);
473225daaf8SStefano Zampini       hypre_CSRMatrixI(hdiag)    = NULL;
474225daaf8SStefano Zampini       hypre_CSRMatrixJ(hdiag)    = NULL;
475225daaf8SStefano Zampini       hypre_CSRMatrixData(hdiag) = NULL;
476225daaf8SStefano Zampini       hypre_CSRMatrixI(hoffd)    = NULL;
477225daaf8SStefano Zampini       hypre_CSRMatrixJ(hoffd)    = NULL;
478225daaf8SStefano Zampini       hypre_CSRMatrixData(hoffd) = NULL;
479225daaf8SStefano Zampini       ierr = MatHeaderReplace(A,&T);CHKERRQ(ierr);
48063c07aadSStefano Zampini     }
481225daaf8SStefano Zampini   } else {
482225daaf8SStefano Zampini     oii  = NULL;
483225daaf8SStefano Zampini     ojj  = NULL;
484225daaf8SStefano Zampini     oa   = NULL;
485225daaf8SStefano Zampini     if (reuse == MAT_INITIAL_MATRIX) {
48663c07aadSStefano Zampini       Mat_SeqAIJ* b;
48763c07aadSStefano Zampini       ierr = MatCreateSeqAIJWithArrays(comm,m,n,dii,djj,da,B);CHKERRQ(ierr);
48863c07aadSStefano Zampini       /* hack SeqAIJ */
48963c07aadSStefano Zampini       b          = (Mat_SeqAIJ*)((*B)->data);
49063c07aadSStefano Zampini       b->free_a  = PETSC_TRUE;
49163c07aadSStefano Zampini       b->free_ij = PETSC_TRUE;
492225daaf8SStefano Zampini     } else if (reuse == MAT_INPLACE_MATRIX) {
493225daaf8SStefano Zampini       Mat T;
494225daaf8SStefano Zampini       ierr = MatCreateSeqAIJWithArrays(comm,m,n,dii,djj,da,&T);CHKERRQ(ierr);
495225daaf8SStefano Zampini       hypre_CSRMatrixI(hdiag)    = NULL;
496225daaf8SStefano Zampini       hypre_CSRMatrixJ(hdiag)    = NULL;
497225daaf8SStefano Zampini       hypre_CSRMatrixData(hdiag) = NULL;
498225daaf8SStefano Zampini       ierr = MatHeaderReplace(A,&T);CHKERRQ(ierr);
49963c07aadSStefano Zampini     }
500225daaf8SStefano Zampini   }
501225daaf8SStefano Zampini 
502225daaf8SStefano Zampini   /* we have to use hypre_Tfree to free the arrays */
50363c07aadSStefano Zampini   if (reuse == MAT_INPLACE_MATRIX) {
504225daaf8SStefano Zampini     void *ptrs[6] = {dii,djj,da,oii,ojj,oa};
505225daaf8SStefano Zampini     const char *names[6] = {"_hypre_csr_dii",
506225daaf8SStefano Zampini                             "_hypre_csr_djj",
507225daaf8SStefano Zampini                             "_hypre_csr_da",
508225daaf8SStefano Zampini                             "_hypre_csr_oii",
509225daaf8SStefano Zampini                             "_hypre_csr_ojj",
510225daaf8SStefano Zampini                             "_hypre_csr_oa"};
511225daaf8SStefano Zampini     for (i=0; i<6; i++) {
512225daaf8SStefano Zampini       PetscContainer c;
513225daaf8SStefano Zampini 
514225daaf8SStefano Zampini       ierr = PetscContainerCreate(comm,&c);CHKERRQ(ierr);
515225daaf8SStefano Zampini       ierr = PetscContainerSetPointer(c,ptrs[i]);CHKERRQ(ierr);
516225daaf8SStefano Zampini       ierr = PetscContainerSetUserDestroy(c,hypre_array_destroy);CHKERRQ(ierr);
517225daaf8SStefano Zampini       ierr = PetscObjectCompose((PetscObject)(*B),names[i],(PetscObject)c);CHKERRQ(ierr);
518225daaf8SStefano Zampini       ierr = PetscContainerDestroy(&c);CHKERRQ(ierr);
519225daaf8SStefano Zampini     }
52063c07aadSStefano Zampini   }
52163c07aadSStefano Zampini   PetscFunctionReturn(0);
52263c07aadSStefano Zampini }
52363c07aadSStefano Zampini 
524613e5ff0Sstefano_zampini static PetscErrorCode MatAIJGetParCSR_Private(Mat A, hypre_ParCSRMatrix **hA)
525c1a070e6SStefano Zampini {
526613e5ff0Sstefano_zampini   hypre_ParCSRMatrix *tA;
527c1a070e6SStefano Zampini   hypre_CSRMatrix    *hdiag,*hoffd;
528c1a070e6SStefano Zampini   Mat_SeqAIJ         *diag,*offd;
529c1a070e6SStefano Zampini   PetscInt           *garray,noffd,dnnz,onnz,*row_starts,*col_starts;
530c1a070e6SStefano Zampini   MPI_Comm           comm = PetscObjectComm((PetscObject)A);
531613e5ff0Sstefano_zampini   PetscBool          ismpiaij,isseqaij;
532c1a070e6SStefano Zampini   PetscErrorCode     ierr;
533c1a070e6SStefano Zampini 
534c1a070e6SStefano Zampini   PetscFunctionBegin;
535c1a070e6SStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)A,MATMPIAIJ,&ismpiaij);CHKERRQ(ierr);
5364099cc6bSBarry Smith   ierr = PetscObjectBaseTypeCompare((PetscObject)A,MATSEQAIJ,&isseqaij);CHKERRQ(ierr);
537c1a070e6SStefano Zampini   if (!ismpiaij && !isseqaij) SETERRQ1(comm,PETSC_ERR_SUP,"Unsupported type %s",((PetscObject)A)->type);
538c1a070e6SStefano Zampini   if (ismpiaij) {
539c1a070e6SStefano Zampini     Mat_MPIAIJ *a = (Mat_MPIAIJ*)(A->data);
540c1a070e6SStefano Zampini 
541c1a070e6SStefano Zampini     diag   = (Mat_SeqAIJ*)a->A->data;
542c1a070e6SStefano Zampini     offd   = (Mat_SeqAIJ*)a->B->data;
543c1a070e6SStefano Zampini     garray = a->garray;
544c1a070e6SStefano Zampini     noffd  = a->B->cmap->N;
545c1a070e6SStefano Zampini     dnnz   = diag->nz;
546c1a070e6SStefano Zampini     onnz   = offd->nz;
547c1a070e6SStefano Zampini   } else {
548c1a070e6SStefano Zampini     diag   = (Mat_SeqAIJ*)A->data;
549c1a070e6SStefano Zampini     offd   = NULL;
550c1a070e6SStefano Zampini     garray = NULL;
551c1a070e6SStefano Zampini     noffd  = 0;
552c1a070e6SStefano Zampini     dnnz   = diag->nz;
553c1a070e6SStefano Zampini     onnz   = 0;
554c1a070e6SStefano Zampini   }
555225daaf8SStefano Zampini 
556c1a070e6SStefano Zampini   /* create a temporary ParCSR */
557c1a070e6SStefano Zampini   if (HYPRE_AssumedPartitionCheck()) {
558c1a070e6SStefano Zampini     PetscMPIInt myid;
559c1a070e6SStefano Zampini 
560c1a070e6SStefano Zampini     ierr       = MPI_Comm_rank(comm,&myid);CHKERRQ(ierr);
561c1a070e6SStefano Zampini     row_starts = A->rmap->range + myid;
562c1a070e6SStefano Zampini     col_starts = A->cmap->range + myid;
563c1a070e6SStefano Zampini   } else {
564c1a070e6SStefano Zampini     row_starts = A->rmap->range;
565c1a070e6SStefano Zampini     col_starts = A->cmap->range;
566c1a070e6SStefano Zampini   }
5677d968826Sstefano_zampini   tA = hypre_ParCSRMatrixCreate(comm,A->rmap->N,A->cmap->N,(HYPRE_Int*)row_starts,(HYPRE_Int*)col_starts,noffd,dnnz,onnz);
568c1a070e6SStefano Zampini   hypre_ParCSRMatrixSetRowStartsOwner(tA,0);
569c1a070e6SStefano Zampini   hypre_ParCSRMatrixSetColStartsOwner(tA,0);
570c1a070e6SStefano Zampini 
571225daaf8SStefano Zampini   /* set diagonal part */
572c1a070e6SStefano Zampini   hdiag = hypre_ParCSRMatrixDiag(tA);
5737d968826Sstefano_zampini   hypre_CSRMatrixI(hdiag)           = (HYPRE_Int*)diag->i;
5747d968826Sstefano_zampini   hypre_CSRMatrixJ(hdiag)           = (HYPRE_Int*)diag->j;
575c1a070e6SStefano Zampini   hypre_CSRMatrixData(hdiag)        = diag->a;
576c1a070e6SStefano Zampini   hypre_CSRMatrixNumNonzeros(hdiag) = diag->nz;
577c1a070e6SStefano Zampini   hypre_CSRMatrixSetRownnz(hdiag);
578c1a070e6SStefano Zampini   hypre_CSRMatrixSetDataOwner(hdiag,0);
579c1a070e6SStefano Zampini 
580225daaf8SStefano Zampini   /* set offdiagonal part */
581c1a070e6SStefano Zampini   hoffd = hypre_ParCSRMatrixOffd(tA);
582c1a070e6SStefano Zampini   if (offd) {
5837d968826Sstefano_zampini     hypre_CSRMatrixI(hoffd)           = (HYPRE_Int*)offd->i;
5847d968826Sstefano_zampini     hypre_CSRMatrixJ(hoffd)           = (HYPRE_Int*)offd->j;
585c1a070e6SStefano Zampini     hypre_CSRMatrixData(hoffd)        = offd->a;
586c1a070e6SStefano Zampini     hypre_CSRMatrixNumNonzeros(hoffd) = offd->nz;
587c1a070e6SStefano Zampini     hypre_CSRMatrixSetRownnz(hoffd);
588c1a070e6SStefano Zampini     hypre_CSRMatrixSetDataOwner(hoffd,0);
589c1a070e6SStefano Zampini     hypre_ParCSRMatrixSetNumNonzeros(tA);
5907d968826Sstefano_zampini     hypre_ParCSRMatrixColMapOffd(tA) = (HYPRE_Int*)garray;
591c1a070e6SStefano Zampini   }
592613e5ff0Sstefano_zampini   *hA = tA;
593613e5ff0Sstefano_zampini   PetscFunctionReturn(0);
594613e5ff0Sstefano_zampini }
595c1a070e6SStefano Zampini 
596613e5ff0Sstefano_zampini static PetscErrorCode MatAIJRestoreParCSR_Private(Mat A, hypre_ParCSRMatrix **hA)
597613e5ff0Sstefano_zampini {
598613e5ff0Sstefano_zampini   hypre_CSRMatrix    *hdiag,*hoffd;
599c1a070e6SStefano Zampini 
600613e5ff0Sstefano_zampini   PetscFunctionBegin;
601613e5ff0Sstefano_zampini   hdiag = hypre_ParCSRMatrixDiag(*hA);
602613e5ff0Sstefano_zampini   hoffd = hypre_ParCSRMatrixOffd(*hA);
603c1a070e6SStefano Zampini   /* set pointers to NULL before destroying tA */
604c1a070e6SStefano Zampini   hypre_CSRMatrixI(hdiag)           = NULL;
605c1a070e6SStefano Zampini   hypre_CSRMatrixJ(hdiag)           = NULL;
606c1a070e6SStefano Zampini   hypre_CSRMatrixData(hdiag)        = NULL;
607c1a070e6SStefano Zampini   hypre_CSRMatrixI(hoffd)           = NULL;
608c1a070e6SStefano Zampini   hypre_CSRMatrixJ(hoffd)           = NULL;
609c1a070e6SStefano Zampini   hypre_CSRMatrixData(hoffd)        = NULL;
610613e5ff0Sstefano_zampini   hypre_ParCSRMatrixColMapOffd(*hA) = NULL;
611613e5ff0Sstefano_zampini   hypre_ParCSRMatrixDestroy(*hA);
612613e5ff0Sstefano_zampini   *hA = NULL;
613613e5ff0Sstefano_zampini   PetscFunctionReturn(0);
614613e5ff0Sstefano_zampini }
615613e5ff0Sstefano_zampini 
616613e5ff0Sstefano_zampini /* calls RAP from BoomerAMG:
6173dad0653Sstefano_zampini    the resulting ParCSR will not own the column and row starts
6183dad0653Sstefano_zampini    It looks like we don't need to have the diagonal entries
6193dad0653Sstefano_zampini    ordered first in the rows of the diagonal part
6203dad0653Sstefano_zampini    for boomerAMGBuildCoarseOperator to work */
621a055b5aaSBarry Smith static PetscErrorCode MatHYPRE_ParCSR_RAP(hypre_ParCSRMatrix *hR, hypre_ParCSRMatrix *hA,hypre_ParCSRMatrix *hP, hypre_ParCSRMatrix **hRAP)
622613e5ff0Sstefano_zampini {
623613e5ff0Sstefano_zampini   HYPRE_Int P_owns_col_starts,R_owns_row_starts;
624613e5ff0Sstefano_zampini 
625613e5ff0Sstefano_zampini   PetscFunctionBegin;
626613e5ff0Sstefano_zampini   P_owns_col_starts = hypre_ParCSRMatrixOwnsColStarts(hP);
627613e5ff0Sstefano_zampini   R_owns_row_starts = hypre_ParCSRMatrixOwnsRowStarts(hR);
628613e5ff0Sstefano_zampini   PetscStackCallStandard(hypre_BoomerAMGBuildCoarseOperator,(hR,hA,hP,hRAP));
629613e5ff0Sstefano_zampini   PetscStackCallStandard(hypre_ParCSRMatrixSetNumNonzeros,(*hRAP));
630613e5ff0Sstefano_zampini   /* hypre_BoomerAMGBuildCoarseOperator steals the col_starts from P and the row_starts from R */
631613e5ff0Sstefano_zampini   hypre_ParCSRMatrixSetRowStartsOwner(*hRAP,0);
632613e5ff0Sstefano_zampini   hypre_ParCSRMatrixSetColStartsOwner(*hRAP,0);
633613e5ff0Sstefano_zampini   if (P_owns_col_starts) hypre_ParCSRMatrixSetColStartsOwner(hP,1);
634613e5ff0Sstefano_zampini   if (R_owns_row_starts) hypre_ParCSRMatrixSetRowStartsOwner(hR,1);
635613e5ff0Sstefano_zampini   PetscFunctionReturn(0);
636613e5ff0Sstefano_zampini }
637613e5ff0Sstefano_zampini 
6386f231fbdSstefano_zampini static PetscErrorCode MatPtAPNumeric_AIJ_AIJ_wHYPRE(Mat A,Mat P,Mat C)
639613e5ff0Sstefano_zampini {
6406f231fbdSstefano_zampini   Mat                B;
641613e5ff0Sstefano_zampini   hypre_ParCSRMatrix *hA,*hP,*hPtAP;
642613e5ff0Sstefano_zampini   PetscErrorCode     ierr;
643613e5ff0Sstefano_zampini 
644613e5ff0Sstefano_zampini   PetscFunctionBegin;
645613e5ff0Sstefano_zampini   ierr = MatAIJGetParCSR_Private(A,&hA);CHKERRQ(ierr);
646613e5ff0Sstefano_zampini   ierr = MatAIJGetParCSR_Private(P,&hP);CHKERRQ(ierr);
647613e5ff0Sstefano_zampini   ierr = MatHYPRE_ParCSR_RAP(hP,hA,hP,&hPtAP);CHKERRQ(ierr);
6486f231fbdSstefano_zampini   ierr = MatCreateFromParCSR(hPtAP,MATAIJ,PETSC_OWN_POINTER,&B);CHKERRQ(ierr);
6496f231fbdSstefano_zampini   ierr = MatHeaderMerge(C,&B);CHKERRQ(ierr);
650613e5ff0Sstefano_zampini   ierr = MatAIJRestoreParCSR_Private(A,&hA);CHKERRQ(ierr);
651613e5ff0Sstefano_zampini   ierr = MatAIJRestoreParCSR_Private(P,&hP);CHKERRQ(ierr);
6526f231fbdSstefano_zampini   PetscFunctionReturn(0);
6536f231fbdSstefano_zampini }
6546f231fbdSstefano_zampini 
6556f231fbdSstefano_zampini PETSC_INTERN PetscErrorCode MatPtAPSymbolic_AIJ_AIJ_wHYPRE(Mat A,Mat P,PetscReal fill,Mat *C)
6566f231fbdSstefano_zampini {
6576f231fbdSstefano_zampini   PetscErrorCode ierr;
658ab4d48faSStefano Zampini 
6596f231fbdSstefano_zampini   PetscFunctionBegin;
6606f231fbdSstefano_zampini   ierr                   = MatCreate(PetscObjectComm((PetscObject)A),C);CHKERRQ(ierr);
6616f231fbdSstefano_zampini   ierr                   = MatSetType(*C,MATAIJ);CHKERRQ(ierr);
6626f231fbdSstefano_zampini   (*C)->ops->ptapnumeric = MatPtAPNumeric_AIJ_AIJ_wHYPRE;
663613e5ff0Sstefano_zampini   PetscFunctionReturn(0);
664613e5ff0Sstefano_zampini }
665613e5ff0Sstefano_zampini 
6664cc28894Sstefano_zampini static PetscErrorCode MatPtAPNumeric_AIJ_HYPRE(Mat A,Mat P,Mat C)
667613e5ff0Sstefano_zampini {
6684cc28894Sstefano_zampini   Mat                B;
6694cc28894Sstefano_zampini   Mat_HYPRE          *hP;
670613e5ff0Sstefano_zampini   hypre_ParCSRMatrix *hA = NULL,*Pparcsr,*ptapparcsr;
671613e5ff0Sstefano_zampini   HYPRE_Int          type;
672613e5ff0Sstefano_zampini   MPI_Comm           comm = PetscObjectComm((PetscObject)A);
6734cc28894Sstefano_zampini   PetscBool          ishypre;
674613e5ff0Sstefano_zampini   PetscErrorCode     ierr;
675613e5ff0Sstefano_zampini 
676613e5ff0Sstefano_zampini   PetscFunctionBegin;
6774cc28894Sstefano_zampini   ierr = PetscObjectTypeCompare((PetscObject)P,MATHYPRE,&ishypre);CHKERRQ(ierr);
6784cc28894Sstefano_zampini   if (!ishypre) SETERRQ1(comm,PETSC_ERR_USER,"P should be of type %s",MATHYPRE);
6794cc28894Sstefano_zampini   hP = (Mat_HYPRE*)P->data;
680613e5ff0Sstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hP->ij,&type));
681613e5ff0Sstefano_zampini   if (type != HYPRE_PARCSR) SETERRQ(comm,PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported");
682613e5ff0Sstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hP->ij,(void**)&Pparcsr));
683613e5ff0Sstefano_zampini 
684613e5ff0Sstefano_zampini   ierr = MatAIJGetParCSR_Private(A,&hA);CHKERRQ(ierr);
685613e5ff0Sstefano_zampini   ierr = MatHYPRE_ParCSR_RAP(Pparcsr,hA,Pparcsr,&ptapparcsr);CHKERRQ(ierr);
686613e5ff0Sstefano_zampini   ierr = MatAIJRestoreParCSR_Private(A,&hA);CHKERRQ(ierr);
687225daaf8SStefano Zampini 
6884cc28894Sstefano_zampini   /* create temporary matrix and merge to C */
6894cc28894Sstefano_zampini   ierr = MatCreateFromParCSR(ptapparcsr,((PetscObject)C)->type_name,PETSC_OWN_POINTER,&B);CHKERRQ(ierr);
6904cc28894Sstefano_zampini   ierr = MatHeaderMerge(C,&B);CHKERRQ(ierr);
6914cc28894Sstefano_zampini   PetscFunctionReturn(0);
6924cc28894Sstefano_zampini }
6934cc28894Sstefano_zampini 
6944cc28894Sstefano_zampini static PetscErrorCode MatPtAP_AIJ_HYPRE(Mat A,Mat P,MatReuse scall,PetscReal fill,Mat *C)
6954cc28894Sstefano_zampini {
6964cc28894Sstefano_zampini   PetscErrorCode ierr;
6974cc28894Sstefano_zampini 
6984cc28894Sstefano_zampini   PetscFunctionBegin;
6994cc28894Sstefano_zampini   if (scall == MAT_INITIAL_MATRIX) {
7004cc28894Sstefano_zampini     const char *deft = MATAIJ;
7014cc28894Sstefano_zampini     char       type[256];
7024cc28894Sstefano_zampini     PetscBool  flg;
7034cc28894Sstefano_zampini 
7044cc28894Sstefano_zampini     ierr = PetscObjectOptionsBegin((PetscObject)A);CHKERRQ(ierr);
7054cc28894Sstefano_zampini     ierr = PetscOptionsFList("-matptap_hypre_outtype","Matrix type",NULL,MatList,deft,type,256,&flg);CHKERRQ(ierr);
7064cc28894Sstefano_zampini     ierr = PetscOptionsEnd();CHKERRQ(ierr);
7074cc28894Sstefano_zampini     ierr = PetscLogEventBegin(MAT_PtAPSymbolic,A,P,0,0);CHKERRQ(ierr);
7084cc28894Sstefano_zampini     ierr = MatCreate(PetscObjectComm((PetscObject)A),C);CHKERRQ(ierr);
7094cc28894Sstefano_zampini     if (flg) {
7104cc28894Sstefano_zampini       ierr = MatSetType(*C,type);CHKERRQ(ierr);
7114cc28894Sstefano_zampini     } else {
7124cc28894Sstefano_zampini       ierr = MatSetType(*C,deft);CHKERRQ(ierr);
7134cc28894Sstefano_zampini     }
7144cc28894Sstefano_zampini     (*C)->ops->ptapnumeric = MatPtAPNumeric_AIJ_HYPRE;
7154cc28894Sstefano_zampini     ierr = PetscLogEventEnd(MAT_PtAPSymbolic,A,P,0,0);CHKERRQ(ierr);
7164cc28894Sstefano_zampini   }
7174cc28894Sstefano_zampini   ierr = PetscLogEventBegin(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr);
7184cc28894Sstefano_zampini   ierr = (*(*C)->ops->ptapnumeric)(A,P,*C);CHKERRQ(ierr);
719613e5ff0Sstefano_zampini   ierr = PetscLogEventEnd(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr);
720c1a070e6SStefano Zampini   PetscFunctionReturn(0);
721c1a070e6SStefano Zampini }
722c1a070e6SStefano Zampini 
7234cc28894Sstefano_zampini static PetscErrorCode MatPtAPNumeric_HYPRE_HYPRE(Mat A,Mat P,Mat C)
7244cc28894Sstefano_zampini {
7254cc28894Sstefano_zampini   Mat                B;
7264cc28894Sstefano_zampini   hypre_ParCSRMatrix *Aparcsr,*Pparcsr,*ptapparcsr;
7274cc28894Sstefano_zampini   Mat_HYPRE          *hA,*hP;
7284cc28894Sstefano_zampini   PetscBool          ishypre;
7294cc28894Sstefano_zampini   HYPRE_Int          type;
7304cc28894Sstefano_zampini   PetscErrorCode     ierr;
7314cc28894Sstefano_zampini 
7324cc28894Sstefano_zampini   PetscFunctionBegin;
7334cc28894Sstefano_zampini   ierr = PetscObjectTypeCompare((PetscObject)P,MATHYPRE,&ishypre);CHKERRQ(ierr);
7344cc28894Sstefano_zampini   if (!ishypre) SETERRQ1(PetscObjectComm((PetscObject)P),PETSC_ERR_USER,"P should be of type %s",MATHYPRE);
7354cc28894Sstefano_zampini   ierr = PetscObjectTypeCompare((PetscObject)A,MATHYPRE,&ishypre);CHKERRQ(ierr);
7364cc28894Sstefano_zampini   if (!ishypre) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_USER,"A should be of type %s",MATHYPRE);
7374cc28894Sstefano_zampini   hA = (Mat_HYPRE*)A->data;
7384cc28894Sstefano_zampini   hP = (Mat_HYPRE*)P->data;
7394cc28894Sstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type));
7404cc28894Sstefano_zampini   if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported");
7414cc28894Sstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hP->ij,&type));
7424cc28894Sstefano_zampini   if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)P),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported");
7434cc28894Sstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)&Aparcsr));
7444cc28894Sstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hP->ij,(void**)&Pparcsr));
7454cc28894Sstefano_zampini   ierr = MatHYPRE_ParCSR_RAP(Pparcsr,Aparcsr,Pparcsr,&ptapparcsr);CHKERRQ(ierr);
7464cc28894Sstefano_zampini   ierr = MatCreateFromParCSR(ptapparcsr,MATHYPRE,PETSC_OWN_POINTER,&B);CHKERRQ(ierr);
7474cc28894Sstefano_zampini   ierr = MatHeaderMerge(C,&B);CHKERRQ(ierr);
7484cc28894Sstefano_zampini   PetscFunctionReturn(0);
7494cc28894Sstefano_zampini }
7504cc28894Sstefano_zampini 
751cd8bc7baSStefano Zampini static PetscErrorCode MatPtAP_HYPRE_HYPRE(Mat A,Mat P,MatReuse scall,PetscReal fill,Mat *C)
752cd8bc7baSStefano Zampini {
753cd8bc7baSStefano Zampini   PetscErrorCode ierr;
754cd8bc7baSStefano Zampini 
755cd8bc7baSStefano Zampini   PetscFunctionBegin;
7564cc28894Sstefano_zampini   if (scall == MAT_INITIAL_MATRIX) {
7574cc28894Sstefano_zampini     ierr = PetscLogEventBegin(MAT_PtAPSymbolic,A,P,0,0);CHKERRQ(ierr);
7584cc28894Sstefano_zampini     ierr                   = MatCreate(PetscObjectComm((PetscObject)A),C);CHKERRQ(ierr);
7594cc28894Sstefano_zampini     ierr                   = MatSetType(*C,MATHYPRE);CHKERRQ(ierr);
7604cc28894Sstefano_zampini     (*C)->ops->ptapnumeric = MatPtAPNumeric_HYPRE_HYPRE;
7614cc28894Sstefano_zampini     ierr = PetscLogEventEnd(MAT_PtAPSymbolic,A,P,0,0);CHKERRQ(ierr);
7624cc28894Sstefano_zampini   }
763613e5ff0Sstefano_zampini   ierr = PetscLogEventBegin(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr);
7644cc28894Sstefano_zampini   ierr = (*(*C)->ops->ptapnumeric)(A,P,*C);CHKERRQ(ierr);
765613e5ff0Sstefano_zampini   ierr = PetscLogEventEnd(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr);
766cd8bc7baSStefano Zampini   PetscFunctionReturn(0);
767cd8bc7baSStefano Zampini }
768cd8bc7baSStefano Zampini 
769d501dc42Sstefano_zampini /* calls hypre_ParMatmul
770d501dc42Sstefano_zampini    hypre_ParMatMul uses hypre_ParMatrixCreate with the communicator of hA
7713dad0653Sstefano_zampini    hypre_ParMatrixCreate does not duplicate the communicator
7723dad0653Sstefano_zampini    It looks like we don't need to have the diagonal entries
7733dad0653Sstefano_zampini    ordered first in the rows of the diagonal part
7743dad0653Sstefano_zampini    for boomerAMGBuildCoarseOperator to work */
775d501dc42Sstefano_zampini static PetscErrorCode MatHYPRE_ParCSR_MatMatMult(hypre_ParCSRMatrix *hA, hypre_ParCSRMatrix *hB, hypre_ParCSRMatrix **hAB)
776d501dc42Sstefano_zampini {
777d501dc42Sstefano_zampini   PetscFunctionBegin;
778d501dc42Sstefano_zampini   PetscStackPush("hypre_ParMatmul");
779d501dc42Sstefano_zampini   *hAB = hypre_ParMatmul(hA,hB);
780d501dc42Sstefano_zampini   PetscStackPop;
781d501dc42Sstefano_zampini   PetscFunctionReturn(0);
782d501dc42Sstefano_zampini }
783d501dc42Sstefano_zampini 
7845e5acdf2Sstefano_zampini static PetscErrorCode MatMatMultNumeric_AIJ_AIJ_wHYPRE(Mat A,Mat B,Mat C)
7855e5acdf2Sstefano_zampini {
7865e5acdf2Sstefano_zampini   Mat                D;
787d501dc42Sstefano_zampini   hypre_ParCSRMatrix *hA,*hB,*hAB = NULL;
7885e5acdf2Sstefano_zampini   PetscErrorCode     ierr;
7895e5acdf2Sstefano_zampini 
7905e5acdf2Sstefano_zampini   PetscFunctionBegin;
7915e5acdf2Sstefano_zampini   ierr = MatAIJGetParCSR_Private(A,&hA);CHKERRQ(ierr);
7925e5acdf2Sstefano_zampini   ierr = MatAIJGetParCSR_Private(B,&hB);CHKERRQ(ierr);
793d501dc42Sstefano_zampini   ierr = MatHYPRE_ParCSR_MatMatMult(hA,hB,&hAB);CHKERRQ(ierr);
7945e5acdf2Sstefano_zampini   ierr = MatCreateFromParCSR(hAB,MATAIJ,PETSC_OWN_POINTER,&D);CHKERRQ(ierr);
7955e5acdf2Sstefano_zampini   ierr = MatHeaderMerge(C,&D);CHKERRQ(ierr);
7965e5acdf2Sstefano_zampini   ierr = MatAIJRestoreParCSR_Private(A,&hA);CHKERRQ(ierr);
7975e5acdf2Sstefano_zampini   ierr = MatAIJRestoreParCSR_Private(B,&hB);CHKERRQ(ierr);
7985e5acdf2Sstefano_zampini   PetscFunctionReturn(0);
7995e5acdf2Sstefano_zampini }
8005e5acdf2Sstefano_zampini 
8015e5acdf2Sstefano_zampini PETSC_INTERN PetscErrorCode MatMatMultSymbolic_AIJ_AIJ_wHYPRE(Mat A,Mat B,PetscReal fill,Mat *C)
8025e5acdf2Sstefano_zampini {
8035e5acdf2Sstefano_zampini   PetscErrorCode ierr;
80420e1dc0dSstefano_zampini 
8055e5acdf2Sstefano_zampini   PetscFunctionBegin;
8065e5acdf2Sstefano_zampini   ierr                      = MatCreate(PetscObjectComm((PetscObject)A),C);CHKERRQ(ierr);
8075e5acdf2Sstefano_zampini   ierr                      = MatSetType(*C,MATAIJ);CHKERRQ(ierr);
8085e5acdf2Sstefano_zampini   (*C)->ops->matmultnumeric = MatMatMultNumeric_AIJ_AIJ_wHYPRE;
8095e5acdf2Sstefano_zampini   PetscFunctionReturn(0);
8105e5acdf2Sstefano_zampini }
8115e5acdf2Sstefano_zampini 
812d501dc42Sstefano_zampini static PetscErrorCode MatMatMultNumeric_HYPRE_HYPRE(Mat A,Mat B,Mat C)
813d501dc42Sstefano_zampini {
814d501dc42Sstefano_zampini   Mat                D;
815d501dc42Sstefano_zampini   hypre_ParCSRMatrix *Aparcsr,*Bparcsr,*ABparcsr = NULL;
816d501dc42Sstefano_zampini   Mat_HYPRE          *hA,*hB;
817d501dc42Sstefano_zampini   PetscBool          ishypre;
818d501dc42Sstefano_zampini   HYPRE_Int          type;
819d501dc42Sstefano_zampini   PetscErrorCode     ierr;
820d501dc42Sstefano_zampini 
821d501dc42Sstefano_zampini   PetscFunctionBegin;
822d501dc42Sstefano_zampini   ierr = PetscObjectTypeCompare((PetscObject)B,MATHYPRE,&ishypre);CHKERRQ(ierr);
823d501dc42Sstefano_zampini   if (!ishypre) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_USER,"B should be of type %s",MATHYPRE);
824d501dc42Sstefano_zampini   ierr = PetscObjectTypeCompare((PetscObject)A,MATHYPRE,&ishypre);CHKERRQ(ierr);
825d501dc42Sstefano_zampini   if (!ishypre) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_USER,"A should be of type %s",MATHYPRE);
826d501dc42Sstefano_zampini   hA = (Mat_HYPRE*)A->data;
827d501dc42Sstefano_zampini   hB = (Mat_HYPRE*)B->data;
828d501dc42Sstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type));
829d501dc42Sstefano_zampini   if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported");
830d501dc42Sstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hB->ij,&type));
831d501dc42Sstefano_zampini   if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)B),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported");
832d501dc42Sstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)&Aparcsr));
833d501dc42Sstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hB->ij,(void**)&Bparcsr));
834d501dc42Sstefano_zampini   ierr = MatHYPRE_ParCSR_MatMatMult(Aparcsr,Bparcsr,&ABparcsr);CHKERRQ(ierr);
835d501dc42Sstefano_zampini   ierr = MatCreateFromParCSR(ABparcsr,MATHYPRE,PETSC_OWN_POINTER,&D);CHKERRQ(ierr);
836d501dc42Sstefano_zampini   /* need to use HeaderReplace because HeaderMerge messes up with the communicator */
837d501dc42Sstefano_zampini   ierr = MatHeaderReplace(C,&D);CHKERRQ(ierr);
838d501dc42Sstefano_zampini   C->ops->matmultnumeric = MatMatMultNumeric_HYPRE_HYPRE;
839d501dc42Sstefano_zampini   PetscFunctionReturn(0);
840d501dc42Sstefano_zampini }
841d501dc42Sstefano_zampini 
842d501dc42Sstefano_zampini static PetscErrorCode MatMatMult_HYPRE_HYPRE(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
843d501dc42Sstefano_zampini {
844d501dc42Sstefano_zampini   PetscErrorCode ierr;
845d501dc42Sstefano_zampini 
846d501dc42Sstefano_zampini   PetscFunctionBegin;
847d501dc42Sstefano_zampini   if (scall == MAT_INITIAL_MATRIX) {
848d501dc42Sstefano_zampini     ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
849d501dc42Sstefano_zampini     ierr                      = MatCreate(PetscObjectComm((PetscObject)A),C);CHKERRQ(ierr);
850d501dc42Sstefano_zampini     ierr                      = MatSetType(*C,MATHYPRE);CHKERRQ(ierr);
851d501dc42Sstefano_zampini     (*C)->ops->matmultnumeric = MatMatMultNumeric_HYPRE_HYPRE;
852d501dc42Sstefano_zampini     ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
853d501dc42Sstefano_zampini   }
854d501dc42Sstefano_zampini   ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
855d501dc42Sstefano_zampini   ierr = (*(*C)->ops->matmultnumeric)(A,B,*C);CHKERRQ(ierr);
856d501dc42Sstefano_zampini   ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
857d501dc42Sstefano_zampini   PetscFunctionReturn(0);
858d501dc42Sstefano_zampini }
859d501dc42Sstefano_zampini 
8603dad0653Sstefano_zampini PETSC_INTERN PetscErrorCode MatTransposeMatMatMultNumeric_AIJ_AIJ_AIJ_wHYPRE(Mat A,Mat B,Mat C,Mat D)
86120e1dc0dSstefano_zampini {
86220e1dc0dSstefano_zampini   Mat                E;
86320e1dc0dSstefano_zampini   hypre_ParCSRMatrix *hA,*hB,*hC,*hABC;
86420e1dc0dSstefano_zampini   PetscErrorCode     ierr;
86520e1dc0dSstefano_zampini 
86620e1dc0dSstefano_zampini   PetscFunctionBegin;
86720e1dc0dSstefano_zampini   ierr = MatAIJGetParCSR_Private(A,&hA);CHKERRQ(ierr);
86820e1dc0dSstefano_zampini   ierr = MatAIJGetParCSR_Private(B,&hB);CHKERRQ(ierr);
86920e1dc0dSstefano_zampini   ierr = MatAIJGetParCSR_Private(C,&hC);CHKERRQ(ierr);
87020e1dc0dSstefano_zampini   ierr = MatHYPRE_ParCSR_RAP(hA,hB,hC,&hABC);CHKERRQ(ierr);
87120e1dc0dSstefano_zampini   ierr = MatCreateFromParCSR(hABC,MATAIJ,PETSC_OWN_POINTER,&E);CHKERRQ(ierr);
87220e1dc0dSstefano_zampini   ierr = MatHeaderMerge(D,&E);CHKERRQ(ierr);
87320e1dc0dSstefano_zampini   ierr = MatAIJRestoreParCSR_Private(A,&hA);CHKERRQ(ierr);
87420e1dc0dSstefano_zampini   ierr = MatAIJRestoreParCSR_Private(B,&hB);CHKERRQ(ierr);
87520e1dc0dSstefano_zampini   ierr = MatAIJRestoreParCSR_Private(C,&hC);CHKERRQ(ierr);
87620e1dc0dSstefano_zampini   PetscFunctionReturn(0);
87720e1dc0dSstefano_zampini }
87820e1dc0dSstefano_zampini 
8793dad0653Sstefano_zampini PETSC_INTERN PetscErrorCode MatTransposeMatMatMultSymbolic_AIJ_AIJ_AIJ_wHYPRE(Mat A,Mat B,Mat C,PetscReal fill,Mat *D)
88020e1dc0dSstefano_zampini {
88120e1dc0dSstefano_zampini   PetscErrorCode ierr;
88220e1dc0dSstefano_zampini 
88320e1dc0dSstefano_zampini   PetscFunctionBegin;
88420e1dc0dSstefano_zampini   ierr = MatCreate(PetscObjectComm((PetscObject)A),D);CHKERRQ(ierr);
88520e1dc0dSstefano_zampini   ierr = MatSetType(*D,MATAIJ);CHKERRQ(ierr);
88620e1dc0dSstefano_zampini   PetscFunctionReturn(0);
88720e1dc0dSstefano_zampini }
88820e1dc0dSstefano_zampini 
889ea9daf28SStefano Zampini static PetscErrorCode MatMultTranspose_HYPRE(Mat A, Vec x, Vec y)
89063c07aadSStefano Zampini {
89163c07aadSStefano Zampini   PetscErrorCode ierr;
89263c07aadSStefano Zampini 
89363c07aadSStefano Zampini   PetscFunctionBegin;
89463c07aadSStefano Zampini   ierr = MatHYPRE_MultKernel_Private(A,x,y,PETSC_TRUE);CHKERRQ(ierr);
89563c07aadSStefano Zampini   PetscFunctionReturn(0);
89663c07aadSStefano Zampini }
89763c07aadSStefano Zampini 
898ea9daf28SStefano Zampini static PetscErrorCode MatMult_HYPRE(Mat A, Vec x, Vec y)
89963c07aadSStefano Zampini {
90063c07aadSStefano Zampini   PetscErrorCode ierr;
90163c07aadSStefano Zampini 
90263c07aadSStefano Zampini   PetscFunctionBegin;
90363c07aadSStefano Zampini   ierr = MatHYPRE_MultKernel_Private(A,x,y,PETSC_FALSE);CHKERRQ(ierr);
90463c07aadSStefano Zampini   PetscFunctionReturn(0);
90563c07aadSStefano Zampini }
90663c07aadSStefano Zampini 
907ea9daf28SStefano Zampini static PetscErrorCode MatHYPRE_MultKernel_Private(Mat A, Vec x, Vec y, PetscBool trans)
90863c07aadSStefano Zampini {
90963c07aadSStefano Zampini   Mat_HYPRE          *hA = (Mat_HYPRE*)A->data;
91063c07aadSStefano Zampini   hypre_ParCSRMatrix *parcsr;
91163c07aadSStefano Zampini   hypre_ParVector    *hx,*hy;
91263c07aadSStefano Zampini   PetscScalar        *ax,*ay,*sax,*say;
91363c07aadSStefano Zampini   PetscErrorCode     ierr;
91463c07aadSStefano Zampini 
91563c07aadSStefano Zampini   PetscFunctionBegin;
91663c07aadSStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)&parcsr));
91763c07aadSStefano Zampini   PetscStackCallStandard(HYPRE_IJVectorGetObject,(hA->x,(void**)&hx));
91863c07aadSStefano Zampini   PetscStackCallStandard(HYPRE_IJVectorGetObject,(hA->b,(void**)&hy));
91963c07aadSStefano Zampini   ierr = VecGetArrayRead(x,(const PetscScalar**)&ax);CHKERRQ(ierr);
92063c07aadSStefano Zampini   ierr = VecGetArray(y,&ay);CHKERRQ(ierr);
92163c07aadSStefano Zampini   if (trans) {
92258968eb6SStefano Zampini     VecHYPRE_ParVectorReplacePointer(hA->x,ay,say);
92358968eb6SStefano Zampini     VecHYPRE_ParVectorReplacePointer(hA->b,ax,sax);
92463c07aadSStefano Zampini     hypre_ParCSRMatrixMatvecT(1.,parcsr,hy,0.,hx);
92558968eb6SStefano Zampini     VecHYPRE_ParVectorReplacePointer(hA->x,say,ay);
92658968eb6SStefano Zampini     VecHYPRE_ParVectorReplacePointer(hA->b,sax,ax);
92763c07aadSStefano Zampini   } else {
92858968eb6SStefano Zampini     VecHYPRE_ParVectorReplacePointer(hA->x,ax,sax);
92958968eb6SStefano Zampini     VecHYPRE_ParVectorReplacePointer(hA->b,ay,say);
93063c07aadSStefano Zampini     hypre_ParCSRMatrixMatvec(1.,parcsr,hx,0.,hy);
93158968eb6SStefano Zampini     VecHYPRE_ParVectorReplacePointer(hA->x,sax,ax);
93258968eb6SStefano Zampini     VecHYPRE_ParVectorReplacePointer(hA->b,say,ay);
93363c07aadSStefano Zampini   }
93463c07aadSStefano Zampini   ierr = VecRestoreArrayRead(x,(const PetscScalar**)&ax);CHKERRQ(ierr);
93563c07aadSStefano Zampini   ierr = VecRestoreArray(y,&ay);CHKERRQ(ierr);
93663c07aadSStefano Zampini   PetscFunctionReturn(0);
93763c07aadSStefano Zampini }
93863c07aadSStefano Zampini 
939ea9daf28SStefano Zampini static PetscErrorCode MatDestroy_HYPRE(Mat A)
94063c07aadSStefano Zampini {
94163c07aadSStefano Zampini   Mat_HYPRE      *hA = (Mat_HYPRE*)A->data;
94263c07aadSStefano Zampini   PetscErrorCode ierr;
94363c07aadSStefano Zampini 
94463c07aadSStefano Zampini   PetscFunctionBegin;
94563c07aadSStefano Zampini   if (hA->x) PetscStackCallStandard(HYPRE_IJVectorDestroy,(hA->x));
94663c07aadSStefano Zampini   if (hA->b) PetscStackCallStandard(HYPRE_IJVectorDestroy,(hA->b));
947978814f1SStefano Zampini   if (hA->ij) {
948978814f1SStefano Zampini     if (!hA->inner_free) hypre_IJMatrixObject(hA->ij) = NULL;
949978814f1SStefano Zampini     PetscStackCallStandard(HYPRE_IJMatrixDestroy,(hA->ij));
950978814f1SStefano Zampini   }
95163c07aadSStefano Zampini   if (hA->comm) { ierr = MPI_Comm_free(&hA->comm);CHKERRQ(ierr); }
952c69f721fSFande Kong 
953c69f721fSFande Kong   ierr = MatStashDestroy_Private(&A->stash);CHKERRQ(ierr);
954c69f721fSFande Kong 
955c69f721fSFande Kong   ierr = PetscFree(hA->array);CHKERRQ(ierr);
956c69f721fSFande Kong 
95763c07aadSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_hypre_aij_C",NULL);CHKERRQ(ierr);
9582df22349SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_hypre_is_C",NULL);CHKERRQ(ierr);
959c1a070e6SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatPtAP_seqaij_hypre_C",NULL);CHKERRQ(ierr);
960c1a070e6SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatPtAP_mpiaij_hypre_C",NULL);CHKERRQ(ierr);
961d975228cSstefano_zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatHYPRESetPreallocation_C",NULL);CHKERRQ(ierr);
962dd9c0a25Sstefano_zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatHYPREGetParCSR_C",NULL);CHKERRQ(ierr);
96375d48cdbSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)A,"MatPtAP_is_hypre_C",NULL);CHKERRQ(ierr);
96463c07aadSStefano Zampini   ierr = PetscFree(A->data);CHKERRQ(ierr);
96563c07aadSStefano Zampini   PetscFunctionReturn(0);
96663c07aadSStefano Zampini }
96763c07aadSStefano Zampini 
968ea9daf28SStefano Zampini static PetscErrorCode MatSetUp_HYPRE(Mat A)
96963c07aadSStefano Zampini {
9704ec6421dSstefano_zampini   PetscErrorCode ierr;
9714ec6421dSstefano_zampini 
9724ec6421dSstefano_zampini   PetscFunctionBegin;
9734ec6421dSstefano_zampini   ierr = MatHYPRESetPreallocation(A,PETSC_DEFAULT,NULL,PETSC_DEFAULT,NULL);CHKERRQ(ierr);
9744ec6421dSstefano_zampini   PetscFunctionReturn(0);
9754ec6421dSstefano_zampini }
9764ec6421dSstefano_zampini 
977c69f721fSFande Kong 
9784ec6421dSstefano_zampini static PetscErrorCode MatAssemblyEnd_HYPRE(Mat A, MatAssemblyType mode)
9794ec6421dSstefano_zampini {
98063c07aadSStefano Zampini   Mat_HYPRE          *hA = (Mat_HYPRE*)A->data;
98163c07aadSStefano Zampini   Vec                x,b;
982c69f721fSFande Kong   PetscMPIInt        n;
983c69f721fSFande Kong   PetscInt           i,j,rstart,ncols,flg;
984c69f721fSFande Kong   PetscInt           *row,*col;
985c69f721fSFande Kong   PetscScalar        *val;
98663c07aadSStefano Zampini   PetscErrorCode     ierr;
98763c07aadSStefano Zampini 
98863c07aadSStefano Zampini   PetscFunctionBegin;
9894ec6421dSstefano_zampini   if (mode == MAT_FLUSH_ASSEMBLY) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MAT_FLUSH_ASSEMBLY currently not supported with MATHYPRE");
990c69f721fSFande Kong 
991c69f721fSFande Kong   if (!A->nooffprocentries) {
992c69f721fSFande Kong     while (1) {
993c69f721fSFande Kong       ierr = MatStashScatterGetMesg_Private(&A->stash,&n,&row,&col,&val,&flg);CHKERRQ(ierr);
994c69f721fSFande Kong       if (!flg) break;
995c69f721fSFande Kong 
996c69f721fSFande Kong       for (i=0; i<n; ) {
997c69f721fSFande Kong         /* Now identify the consecutive vals belonging to the same row */
998c69f721fSFande Kong         for (j=i,rstart=row[j]; j<n; j++) {
999c69f721fSFande Kong           if (row[j] != rstart) break;
1000c69f721fSFande Kong         }
1001c69f721fSFande Kong         if (j < n) ncols = j-i;
1002c69f721fSFande Kong         else       ncols = n-i;
1003c69f721fSFande Kong         /* Now assemble all these values with a single function call */
1004c69f721fSFande Kong         ierr = MatSetValues_HYPRE(A,1,row+i,ncols,col+i,val+i,A->insertmode);CHKERRQ(ierr);
1005c69f721fSFande Kong 
1006c69f721fSFande Kong         i = j;
1007c69f721fSFande Kong       }
1008c69f721fSFande Kong     }
1009c69f721fSFande Kong     ierr = MatStashScatterEnd_Private(&A->stash);CHKERRQ(ierr);
1010c69f721fSFande Kong   }
1011c69f721fSFande Kong 
10124ec6421dSstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixAssemble,(hA->ij));
1013af1cf968SStefano Zampini   /* The assembly routine destroys the aux_matrix, we recreate it here by calling HYPRE_IJMatrixInitalize */
1014af1cf968SStefano Zampini   {
1015af1cf968SStefano Zampini     hypre_AuxParCSRMatrix *aux_matrix;
1016af1cf968SStefano Zampini 
1017af1cf968SStefano Zampini     /* call destroy just to make sure we do not leak anything */
1018af1cf968SStefano Zampini     aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(hA->ij);
1019af1cf968SStefano Zampini     PetscStackCallStandard(hypre_AuxParCSRMatrixDestroy,(aux_matrix));
1020af1cf968SStefano Zampini     hypre_IJMatrixTranslator(hA->ij) = NULL;
1021af1cf968SStefano Zampini 
1022af1cf968SStefano Zampini     /* Initialize with assembled flag -> it only recreates the aux_par_matrix */
1023af1cf968SStefano Zampini     PetscStackCallStandard(HYPRE_IJMatrixInitialize,(hA->ij));
1024af1cf968SStefano Zampini     aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(hA->ij);
1025af1cf968SStefano Zampini     hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 1; /* see comment in MatHYPRESetPreallocation_HYPRE */
1026af1cf968SStefano Zampini     PetscStackCallStandard(hypre_AuxParCSRMatrixInitialize,(aux_matrix));
1027af1cf968SStefano Zampini   }
102863c07aadSStefano Zampini   if (hA->x) PetscFunctionReturn(0);
102963c07aadSStefano Zampini   ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr);
103063c07aadSStefano Zampini   ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr);
103163c07aadSStefano Zampini   ierr = VecCreateMPIWithArray(PetscObjectComm((PetscObject)A),1,A->cmap->n,A->cmap->N,NULL,&x);CHKERRQ(ierr);
103263c07aadSStefano Zampini   ierr = VecCreateMPIWithArray(PetscObjectComm((PetscObject)A),1,A->rmap->n,A->rmap->N,NULL,&b);CHKERRQ(ierr);
103363c07aadSStefano Zampini   ierr = VecHYPRE_IJVectorCreate(x,&hA->x);CHKERRQ(ierr);
103463c07aadSStefano Zampini   ierr = VecHYPRE_IJVectorCreate(b,&hA->b);CHKERRQ(ierr);
103563c07aadSStefano Zampini   ierr = VecDestroy(&x);CHKERRQ(ierr);
103663c07aadSStefano Zampini   ierr = VecDestroy(&b);CHKERRQ(ierr);
103763c07aadSStefano Zampini   PetscFunctionReturn(0);
103863c07aadSStefano Zampini }
103963c07aadSStefano Zampini 
1040c69f721fSFande Kong static PetscErrorCode MatGetArray_HYPRE(Mat A, PetscInt size, void **array)
1041c69f721fSFande Kong {
1042c69f721fSFande Kong   Mat_HYPRE          *hA = (Mat_HYPRE*)A->data;
1043c69f721fSFande Kong   PetscErrorCode     ierr;
1044c69f721fSFande Kong 
1045c69f721fSFande Kong   PetscFunctionBegin;
1046c69f721fSFande Kong   if (!hA->available) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_NULL,"Temporary space is in use");
1047c69f721fSFande Kong 
1048c69f721fSFande Kong   if (hA->size >= size) *array = hA->array;
1049c69f721fSFande Kong   else {
1050c69f721fSFande Kong     ierr = PetscFree(hA->array);CHKERRQ(ierr);
1051c69f721fSFande Kong     hA->size = size;
1052c69f721fSFande Kong     ierr = PetscMalloc(hA->size,&hA->array);CHKERRQ(ierr);
1053c69f721fSFande Kong     *array = hA->array;
1054c69f721fSFande Kong   }
1055c69f721fSFande Kong 
1056c69f721fSFande Kong   hA->available = PETSC_FALSE;
1057c69f721fSFande Kong   PetscFunctionReturn(0);
1058c69f721fSFande Kong }
1059c69f721fSFande Kong 
1060708542d2SFande Kong static PetscErrorCode MatRestoreArray_HYPRE(Mat A, void **array)
1061c69f721fSFande Kong {
1062c69f721fSFande Kong   Mat_HYPRE          *hA = (Mat_HYPRE*)A->data;
1063c69f721fSFande Kong 
1064c69f721fSFande Kong   PetscFunctionBegin;
1065c69f721fSFande Kong   *array = NULL;
1066c69f721fSFande Kong   hA->available = PETSC_TRUE;
1067c69f721fSFande Kong   PetscFunctionReturn(0);
1068c69f721fSFande Kong }
1069c69f721fSFande Kong 
1070d975228cSstefano_zampini 
1071d975228cSstefano_zampini PetscErrorCode MatSetValues_HYPRE(Mat A, PetscInt nr, const PetscInt rows[], PetscInt nc, const PetscInt cols[], const PetscScalar v[], InsertMode ins)
1072d975228cSstefano_zampini {
1073d975228cSstefano_zampini   Mat_HYPRE          *hA = (Mat_HYPRE*)A->data;
1074d975228cSstefano_zampini   PetscScalar        *vals = (PetscScalar *)v;
1075c69f721fSFande Kong   PetscScalar        *sscr;
1076c69f721fSFande Kong   PetscInt           *cscr[2];
1077c69f721fSFande Kong   PetscInt           i,nzc;
107808defe43SFande Kong   void               *array = NULL;
1079d975228cSstefano_zampini   PetscErrorCode     ierr;
1080d975228cSstefano_zampini 
1081d975228cSstefano_zampini   PetscFunctionBegin;
1082c69f721fSFande Kong   ierr = MatGetArray_HYPRE(A,sizeof(PetscInt)*(2*nc)+sizeof(PetscScalar)*nc*nr,&array);CHKERRQ(ierr);
1083c69f721fSFande Kong   cscr[0] = (PetscInt*)array;
1084c69f721fSFande Kong   cscr[1] = ((PetscInt*)array)+nc;
1085c69f721fSFande Kong   sscr = (PetscScalar*)(((PetscInt*)array)+nc*2);
1086d975228cSstefano_zampini   for (i=0,nzc=0;i<nc;i++) {
1087d975228cSstefano_zampini     if (cols[i] >= 0) {
1088d975228cSstefano_zampini       cscr[0][nzc  ] = cols[i];
1089d975228cSstefano_zampini       cscr[1][nzc++] = i;
1090d975228cSstefano_zampini     }
1091d975228cSstefano_zampini   }
1092c69f721fSFande Kong   if (!nzc) {
1093708542d2SFande Kong     ierr = MatRestoreArray_HYPRE(A,&array);CHKERRQ(ierr);
1094c69f721fSFande Kong     PetscFunctionReturn(0);
1095c69f721fSFande Kong   }
1096d975228cSstefano_zampini 
1097d975228cSstefano_zampini   if (ins == ADD_VALUES) {
1098d975228cSstefano_zampini     for (i=0;i<nr;i++) {
1099e3977e59Sstefano_zampini       if (rows[i] >= 0 && nzc) {
1100d975228cSstefano_zampini         PetscInt j;
1101d975228cSstefano_zampini         for (j=0;j<nzc;j++) sscr[j] = vals[cscr[1][j]];
110208defe43SFande Kong         PetscStackCallStandard(HYPRE_IJMatrixAddToValues,(hA->ij,1,(HYPRE_Int*)&nzc,(HYPRE_Int*)(rows+i),(HYPRE_Int*)cscr[0],sscr));
1103d975228cSstefano_zampini       }
1104d975228cSstefano_zampini       vals += nc;
1105d975228cSstefano_zampini     }
1106d975228cSstefano_zampini   } else { /* INSERT_VALUES */
1107c69f721fSFande Kong 
1108d975228cSstefano_zampini     PetscInt rst,ren;
1109d975228cSstefano_zampini     ierr = MatGetOwnershipRange(A,&rst,&ren);CHKERRQ(ierr);
1110c69f721fSFande Kong 
1111d975228cSstefano_zampini     for (i=0;i<nr;i++) {
1112e3977e59Sstefano_zampini       if (rows[i] >= 0 && nzc) {
1113d975228cSstefano_zampini         PetscInt j;
1114d975228cSstefano_zampini         for (j=0;j<nzc;j++) sscr[j] = vals[cscr[1][j]];
1115c69f721fSFande Kong         /* nonlocal values */
1116c69f721fSFande Kong         if (rows[i] < rst || rows[i] >= ren) { ierr = MatStashValuesRow_Private(&A->stash,rows[i],nzc,cscr[0],sscr,PETSC_FALSE);CHKERRQ(ierr); }
1117c69f721fSFande Kong         /* local values */
111808defe43SFande Kong         else PetscStackCallStandard(HYPRE_IJMatrixSetValues,(hA->ij,1,(HYPRE_Int*)&nzc,(HYPRE_Int*)(rows+i),(HYPRE_Int*)cscr[0],sscr));
1119d975228cSstefano_zampini       }
1120d975228cSstefano_zampini       vals += nc;
1121d975228cSstefano_zampini     }
1122d975228cSstefano_zampini   }
1123c69f721fSFande Kong 
1124708542d2SFande Kong   ierr = MatRestoreArray_HYPRE(A,&array);CHKERRQ(ierr);
1125d975228cSstefano_zampini   PetscFunctionReturn(0);
1126d975228cSstefano_zampini }
1127d975228cSstefano_zampini 
1128d975228cSstefano_zampini static PetscErrorCode MatHYPRESetPreallocation_HYPRE(Mat A, PetscInt dnz, const PetscInt dnnz[], PetscInt onz, const PetscInt onnz[])
1129d975228cSstefano_zampini {
1130d975228cSstefano_zampini   Mat_HYPRE      *hA = (Mat_HYPRE*)A->data;
11317d968826Sstefano_zampini   HYPRE_Int      *hdnnz,*honnz;
113206a29025Sstefano_zampini   PetscInt       i,rs,re,cs,ce,bs;
1133d975228cSstefano_zampini   PetscMPIInt    size;
1134d975228cSstefano_zampini   PetscErrorCode ierr;
1135d975228cSstefano_zampini 
1136d975228cSstefano_zampini   PetscFunctionBegin;
113706a29025Sstefano_zampini   ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr);
1138d975228cSstefano_zampini   ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr);
1139d975228cSstefano_zampini   ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr);
1140d975228cSstefano_zampini   rs   = A->rmap->rstart;
1141d975228cSstefano_zampini   re   = A->rmap->rend;
1142d975228cSstefano_zampini   cs   = A->cmap->rstart;
1143d975228cSstefano_zampini   ce   = A->cmap->rend;
1144d975228cSstefano_zampini   if (!hA->ij) {
1145d975228cSstefano_zampini     PetscStackCallStandard(HYPRE_IJMatrixCreate,(hA->comm,rs,re-1,cs,ce-1,&hA->ij));
1146d975228cSstefano_zampini     PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(hA->ij,HYPRE_PARCSR));
1147d975228cSstefano_zampini   } else {
11487d968826Sstefano_zampini     HYPRE_Int hrs,hre,hcs,hce;
1149d975228cSstefano_zampini     PetscStackCallStandard(HYPRE_IJMatrixGetLocalRange,(hA->ij,&hrs,&hre,&hcs,&hce));
1150d975228cSstefano_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);
1151d975228cSstefano_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);
1152d975228cSstefano_zampini   }
115306a29025Sstefano_zampini   if (dnz == PETSC_DEFAULT || dnz == PETSC_DECIDE) dnz = 10*bs;
115406a29025Sstefano_zampini   if (onz == PETSC_DEFAULT || onz == PETSC_DECIDE) onz = 10*bs;
115506a29025Sstefano_zampini 
1156d975228cSstefano_zampini   if (!dnnz) {
1157d975228cSstefano_zampini     ierr = PetscMalloc1(A->rmap->n,&hdnnz);CHKERRQ(ierr);
1158d975228cSstefano_zampini     for (i=0;i<A->rmap->n;i++) hdnnz[i] = dnz;
1159d975228cSstefano_zampini   } else {
11607d968826Sstefano_zampini     hdnnz = (HYPRE_Int*)dnnz;
1161d975228cSstefano_zampini   }
1162d975228cSstefano_zampini   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)A),&size);CHKERRQ(ierr);
1163d975228cSstefano_zampini   if (size > 1) {
1164ddbeb582SStefano Zampini     hypre_AuxParCSRMatrix *aux_matrix;
1165d975228cSstefano_zampini     if (!onnz) {
1166d975228cSstefano_zampini       ierr = PetscMalloc1(A->rmap->n,&honnz);CHKERRQ(ierr);
1167d975228cSstefano_zampini       for (i=0;i<A->rmap->n;i++) honnz[i] = onz;
1168d975228cSstefano_zampini     } else {
11697d968826Sstefano_zampini       honnz = (HYPRE_Int*)onnz;
1170d975228cSstefano_zampini     }
1171ddbeb582SStefano Zampini     /* SetDiagOffdSizes sets hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 0, since it seems
1172ddbeb582SStefano Zampini        they assume the user will input the entire row values, properly sorted
1173ddbeb582SStefano Zampini        In PETSc, we don't make such an assumption, and we instead set this flag to 1
1174ddbeb582SStefano Zampini        Also, to avoid possible memory leaks, we destroy and recreate the translator
1175ddbeb582SStefano Zampini        This has to be done here, as HYPRE_IJMatrixInitialize will properly initialize
1176ddbeb582SStefano Zampini        the IJ matrix for us */
1177ddbeb582SStefano Zampini     aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(hA->ij);
1178ddbeb582SStefano Zampini     hypre_AuxParCSRMatrixDestroy(aux_matrix);
1179ddbeb582SStefano Zampini     hypre_IJMatrixTranslator(hA->ij) = NULL;
1180d975228cSstefano_zampini     PetscStackCallStandard(HYPRE_IJMatrixSetDiagOffdSizes,(hA->ij,hdnnz,honnz));
1181ddbeb582SStefano Zampini     aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(hA->ij);
1182ddbeb582SStefano Zampini     hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 1;
1183d975228cSstefano_zampini   } else {
1184d975228cSstefano_zampini     honnz = NULL;
1185d975228cSstefano_zampini     PetscStackCallStandard(HYPRE_IJMatrixSetRowSizes,(hA->ij,hdnnz));
1186d975228cSstefano_zampini   }
1187ddbeb582SStefano Zampini 
1188af1cf968SStefano Zampini   /* reset assembled flag and call the initialize method */
1189af1cf968SStefano Zampini   hypre_IJMatrixAssembleFlag(hA->ij) = 0;
1190ddbeb582SStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixInitialize,(hA->ij));
1191d975228cSstefano_zampini   if (!dnnz) {
1192d975228cSstefano_zampini     ierr = PetscFree(hdnnz);CHKERRQ(ierr);
1193d975228cSstefano_zampini   }
1194d975228cSstefano_zampini   if (!onnz && honnz) {
1195d975228cSstefano_zampini     ierr = PetscFree(honnz);CHKERRQ(ierr);
1196d975228cSstefano_zampini   }
1197af1cf968SStefano Zampini 
1198af1cf968SStefano Zampini   /* Match AIJ logic */
119906a29025Sstefano_zampini   A->preallocated = PETSC_TRUE;
1200af1cf968SStefano Zampini   A->assembled    = PETSC_FALSE;
1201d975228cSstefano_zampini   PetscFunctionReturn(0);
1202d975228cSstefano_zampini }
1203d975228cSstefano_zampini 
1204d975228cSstefano_zampini /*@C
1205d975228cSstefano_zampini    MatHYPRESetPreallocation - Preallocates memory for a sparse parallel matrix in HYPRE IJ format
1206d975228cSstefano_zampini 
1207d975228cSstefano_zampini    Collective on Mat
1208d975228cSstefano_zampini 
1209d975228cSstefano_zampini    Input Parameters:
1210d975228cSstefano_zampini +  A - the matrix
1211d975228cSstefano_zampini .  dnz  - number of nonzeros per row in DIAGONAL portion of local submatrix
1212d975228cSstefano_zampini           (same value is used for all local rows)
1213d975228cSstefano_zampini .  dnnz - array containing the number of nonzeros in the various rows of the
1214d975228cSstefano_zampini           DIAGONAL portion of the local submatrix (possibly different for each row)
1215d975228cSstefano_zampini           or NULL (PETSC_NULL_INTEGER in Fortran), if d_nz is used to specify the nonzero structure.
1216d975228cSstefano_zampini           The size of this array is equal to the number of local rows, i.e 'm'.
1217d975228cSstefano_zampini           For matrices that will be factored, you must leave room for (and set)
1218d975228cSstefano_zampini           the diagonal entry even if it is zero.
1219d975228cSstefano_zampini .  onz  - number of nonzeros per row in the OFF-DIAGONAL portion of local
1220d975228cSstefano_zampini           submatrix (same value is used for all local rows).
1221d975228cSstefano_zampini -  onnz - array containing the number of nonzeros in the various rows of the
1222d975228cSstefano_zampini           OFF-DIAGONAL portion of the local submatrix (possibly different for
1223d975228cSstefano_zampini           each row) or NULL (PETSC_NULL_INTEGER in Fortran), if o_nz is used to specify the nonzero
1224d975228cSstefano_zampini           structure. The size of this array is equal to the number
1225d975228cSstefano_zampini           of local rows, i.e 'm'.
1226d975228cSstefano_zampini 
122795452b02SPatrick Sanan    Notes:
122895452b02SPatrick Sanan     If the *nnz parameter is given then the *nz parameter is ignored; for sequential matrices, onz and onnz are ignored.
1229d975228cSstefano_zampini 
1230d975228cSstefano_zampini    Level: intermediate
1231d975228cSstefano_zampini 
1232d975228cSstefano_zampini .keywords: matrix, aij, compressed row, sparse, parallel
1233d975228cSstefano_zampini 
1234af1cf968SStefano Zampini .seealso: MatCreate(), MatMPIAIJSetPreallocation(), MATHYPRE
1235d975228cSstefano_zampini @*/
1236d975228cSstefano_zampini PetscErrorCode MatHYPRESetPreallocation(Mat A, PetscInt dnz, const PetscInt dnnz[], PetscInt onz, const PetscInt onnz[])
1237d975228cSstefano_zampini {
1238d975228cSstefano_zampini   PetscErrorCode ierr;
1239d975228cSstefano_zampini 
1240d975228cSstefano_zampini   PetscFunctionBegin;
1241d975228cSstefano_zampini   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
1242d975228cSstefano_zampini   PetscValidType(A,1);
1243d975228cSstefano_zampini   ierr = PetscTryMethod(A,"MatHYPRESetPreallocation_C",(Mat,PetscInt,const PetscInt[],PetscInt,const PetscInt[]),(A,dnz,dnnz,onz,onnz));CHKERRQ(ierr);
1244d975228cSstefano_zampini   PetscFunctionReturn(0);
1245d975228cSstefano_zampini }
1246d975228cSstefano_zampini 
1247225daaf8SStefano Zampini /*
1248225daaf8SStefano Zampini    MatCreateFromParCSR - Creates a matrix from a hypre_ParCSRMatrix
1249225daaf8SStefano Zampini 
1250225daaf8SStefano Zampini    Collective
1251225daaf8SStefano Zampini 
1252225daaf8SStefano Zampini    Input Parameters:
125345b8d346SStefano Zampini +  parcsr   - the pointer to the hypre_ParCSRMatrix
1254bb4689ddSStefano Zampini .  mtype    - matrix type to be created. Currently MATAIJ, MATIS and MATHYPRE are supported.
1255225daaf8SStefano Zampini -  copymode - PETSc copying options
1256225daaf8SStefano Zampini 
1257225daaf8SStefano Zampini    Output Parameter:
1258225daaf8SStefano Zampini .  A  - the matrix
1259225daaf8SStefano Zampini 
1260225daaf8SStefano Zampini    Level: intermediate
1261225daaf8SStefano Zampini 
1262225daaf8SStefano Zampini .seealso: MatHYPRE, PetscCopyMode
1263225daaf8SStefano Zampini */
126445b8d346SStefano Zampini PETSC_EXTERN PetscErrorCode MatCreateFromParCSR(hypre_ParCSRMatrix *parcsr, MatType mtype, PetscCopyMode copymode, Mat* A)
1265978814f1SStefano Zampini {
1266225daaf8SStefano Zampini   Mat                   T;
1267978814f1SStefano Zampini   Mat_HYPRE             *hA;
1268978814f1SStefano Zampini   MPI_Comm              comm;
1269978814f1SStefano Zampini   PetscInt              rstart,rend,cstart,cend,M,N;
1270bb4689ddSStefano Zampini   PetscBool             isseqaij,ismpiaij,isaij,ishyp,isis;
1271978814f1SStefano Zampini   PetscErrorCode        ierr;
1272978814f1SStefano Zampini 
1273978814f1SStefano Zampini   PetscFunctionBegin;
1274978814f1SStefano Zampini   comm   = hypre_ParCSRMatrixComm(parcsr);
1275225daaf8SStefano Zampini   ierr   = PetscStrcmp(mtype,MATSEQAIJ,&isseqaij);CHKERRQ(ierr);
1276225daaf8SStefano Zampini   ierr   = PetscStrcmp(mtype,MATMPIAIJ,&ismpiaij);CHKERRQ(ierr);
1277225daaf8SStefano Zampini   ierr   = PetscStrcmp(mtype,MATAIJ,&isaij);CHKERRQ(ierr);
1278225daaf8SStefano Zampini   ierr   = PetscStrcmp(mtype,MATHYPRE,&ishyp);CHKERRQ(ierr);
12798cfe8d00SStefano Zampini   ierr   = PetscStrcmp(mtype,MATIS,&isis);CHKERRQ(ierr);
1280225daaf8SStefano Zampini   isaij  = (PetscBool)(isseqaij || ismpiaij || isaij);
1281bb4689ddSStefano Zampini   if (!isaij && !ishyp && !isis) SETERRQ6(comm,PETSC_ERR_SUP,"Unsupported MatType %s! Supported types are %s, %s, %s, %s, and %s",mtype,MATAIJ,MATSEQAIJ,MATMPIAIJ,MATIS,MATHYPRE);
1282978814f1SStefano Zampini   /* access ParCSRMatrix */
1283978814f1SStefano Zampini   rstart = hypre_ParCSRMatrixFirstRowIndex(parcsr);
1284978814f1SStefano Zampini   rend   = hypre_ParCSRMatrixLastRowIndex(parcsr);
1285978814f1SStefano Zampini   cstart = hypre_ParCSRMatrixFirstColDiag(parcsr);
1286978814f1SStefano Zampini   cend   = hypre_ParCSRMatrixLastColDiag(parcsr);
1287978814f1SStefano Zampini   M      = hypre_ParCSRMatrixGlobalNumRows(parcsr);
1288978814f1SStefano Zampini   N      = hypre_ParCSRMatrixGlobalNumCols(parcsr);
1289978814f1SStefano Zampini 
1290fa92c42cSstefano_zampini   /* fix for empty local rows/columns */
1291fa92c42cSstefano_zampini   if (rend < rstart) rend = rstart;
1292fa92c42cSstefano_zampini   if (cend < cstart) cend = cstart;
1293fa92c42cSstefano_zampini 
1294*e6471dc9SStefano Zampini   /* PETSc convention */
1295*e6471dc9SStefano Zampini   rend++;
1296*e6471dc9SStefano Zampini   cend++;
1297*e6471dc9SStefano Zampini   rend = PetscMin(rend,M);
1298*e6471dc9SStefano Zampini   cend = PetscMin(cend,N);
1299*e6471dc9SStefano Zampini 
1300978814f1SStefano Zampini   /* create PETSc matrix with MatHYPRE */
1301225daaf8SStefano Zampini   ierr = MatCreate(comm,&T);CHKERRQ(ierr);
1302*e6471dc9SStefano Zampini   ierr = MatSetSizes(T,rend-rstart,cend-cstart,M,N);CHKERRQ(ierr);
1303225daaf8SStefano Zampini   ierr = MatSetType(T,MATHYPRE);CHKERRQ(ierr);
1304225daaf8SStefano Zampini   hA   = (Mat_HYPRE*)(T->data);
1305978814f1SStefano Zampini 
1306978814f1SStefano Zampini   /* create HYPRE_IJMatrix */
1307978814f1SStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixCreate,(hA->comm,rstart,rend-1,cstart,cend-1,&hA->ij));
130845b8d346SStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(hA->ij,HYPRE_PARCSR));
130945b8d346SStefano Zampini 
131045b8d346SStefano Zampini   /* create new ParCSR object if needed */
131145b8d346SStefano Zampini   if (ishyp && copymode == PETSC_COPY_VALUES) {
131245b8d346SStefano Zampini     hypre_ParCSRMatrix *new_parcsr;
131345b8d346SStefano Zampini     hypre_CSRMatrix    *hdiag,*hoffd,*ndiag,*noffd;
131445b8d346SStefano Zampini 
131545b8d346SStefano Zampini     new_parcsr = hypre_ParCSRMatrixCompleteClone(parcsr);
131645b8d346SStefano Zampini     hdiag      = hypre_ParCSRMatrixDiag(parcsr);
131745b8d346SStefano Zampini     hoffd      = hypre_ParCSRMatrixOffd(parcsr);
131845b8d346SStefano Zampini     ndiag      = hypre_ParCSRMatrixDiag(new_parcsr);
131945b8d346SStefano Zampini     noffd      = hypre_ParCSRMatrixOffd(new_parcsr);
132045b8d346SStefano Zampini     ierr       = PetscMemcpy(hypre_CSRMatrixData(ndiag),hypre_CSRMatrixData(hdiag),hypre_CSRMatrixNumNonzeros(hdiag)*sizeof(PetscScalar));CHKERRQ(ierr);
132145b8d346SStefano Zampini     ierr       = PetscMemcpy(hypre_CSRMatrixData(noffd),hypre_CSRMatrixData(hoffd),hypre_CSRMatrixNumNonzeros(hoffd)*sizeof(PetscScalar));CHKERRQ(ierr);
132245b8d346SStefano Zampini     parcsr     = new_parcsr;
132345b8d346SStefano Zampini     copymode   = PETSC_OWN_POINTER;
132445b8d346SStefano Zampini   }
1325978814f1SStefano Zampini 
1326978814f1SStefano Zampini   /* set ParCSR object */
1327978814f1SStefano Zampini   hypre_IJMatrixObject(hA->ij) = parcsr;
13284ec6421dSstefano_zampini   T->preallocated = PETSC_TRUE;
1329978814f1SStefano Zampini 
1330978814f1SStefano Zampini   /* set assembled flag */
1331978814f1SStefano Zampini   hypre_IJMatrixAssembleFlag(hA->ij) = 1;
1332978814f1SStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixInitialize,(hA->ij));
1333225daaf8SStefano Zampini   if (ishyp) {
13346d2a658fSstefano_zampini     PetscMPIInt myid = 0;
13356d2a658fSstefano_zampini 
13366d2a658fSstefano_zampini     /* make sure we always have row_starts and col_starts available */
13376d2a658fSstefano_zampini     if (HYPRE_AssumedPartitionCheck()) {
13386d2a658fSstefano_zampini       ierr = MPI_Comm_rank(comm,&myid);CHKERRQ(ierr);
13396d2a658fSstefano_zampini     }
13406d2a658fSstefano_zampini     if (!hypre_ParCSRMatrixOwnsColStarts(parcsr)) {
13416d2a658fSstefano_zampini       PetscLayout map;
13426d2a658fSstefano_zampini 
13436d2a658fSstefano_zampini       ierr = MatGetLayouts(T,NULL,&map);CHKERRQ(ierr);
13446d2a658fSstefano_zampini       ierr = PetscLayoutSetUp(map);CHKERRQ(ierr);
13451c92d2d0Sstefano_zampini       hypre_ParCSRMatrixColStarts(parcsr) = (HYPRE_Int*)(map->range + myid);
13466d2a658fSstefano_zampini     }
13476d2a658fSstefano_zampini     if (!hypre_ParCSRMatrixOwnsRowStarts(parcsr)) {
13486d2a658fSstefano_zampini       PetscLayout map;
13496d2a658fSstefano_zampini 
13506d2a658fSstefano_zampini       ierr = MatGetLayouts(T,&map,NULL);CHKERRQ(ierr);
13516d2a658fSstefano_zampini       ierr = PetscLayoutSetUp(map);CHKERRQ(ierr);
13521c92d2d0Sstefano_zampini       hypre_ParCSRMatrixRowStarts(parcsr) = (HYPRE_Int*)(map->range + myid);
13536d2a658fSstefano_zampini     }
1354978814f1SStefano Zampini     /* prevent from freeing the pointer */
1355978814f1SStefano Zampini     if (copymode == PETSC_USE_POINTER) hA->inner_free = PETSC_FALSE;
1356225daaf8SStefano Zampini     *A   = T;
1357978814f1SStefano Zampini     ierr = MatAssemblyBegin(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1358978814f1SStefano Zampini     ierr = MatAssemblyEnd(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1359bb4689ddSStefano Zampini   } else if (isaij) {
1360bb4689ddSStefano Zampini     if (copymode != PETSC_OWN_POINTER) {
1361225daaf8SStefano Zampini       /* prevent from freeing the pointer */
1362225daaf8SStefano Zampini       hA->inner_free = PETSC_FALSE;
1363225daaf8SStefano Zampini       ierr = MatConvert_HYPRE_AIJ(T,MATAIJ,MAT_INITIAL_MATRIX,A);CHKERRQ(ierr);
1364225daaf8SStefano Zampini       ierr = MatDestroy(&T);CHKERRQ(ierr);
1365225daaf8SStefano Zampini     } else { /* AIJ return type with PETSC_OWN_POINTER */
1366225daaf8SStefano Zampini       ierr = MatConvert_HYPRE_AIJ(T,MATAIJ,MAT_INPLACE_MATRIX,&T);CHKERRQ(ierr);
1367225daaf8SStefano Zampini       *A   = T;
1368225daaf8SStefano Zampini     }
1369bb4689ddSStefano Zampini   } else if (isis) {
13708cfe8d00SStefano Zampini     ierr = MatConvert_HYPRE_IS(T,MATIS,MAT_INITIAL_MATRIX,A);CHKERRQ(ierr);
13718cfe8d00SStefano Zampini     if (copymode != PETSC_OWN_POINTER) hA->inner_free = PETSC_FALSE;
1372bb4689ddSStefano Zampini     ierr = MatDestroy(&T);CHKERRQ(ierr);
1373bb4689ddSStefano Zampini   }
1374978814f1SStefano Zampini   PetscFunctionReturn(0);
1375978814f1SStefano Zampini }
1376978814f1SStefano Zampini 
137768ec7858SStefano Zampini static PetscErrorCode MatHYPREGetParCSR_HYPRE(Mat A, hypre_ParCSRMatrix **parcsr)
1378dd9c0a25Sstefano_zampini {
1379dd9c0a25Sstefano_zampini   Mat_HYPRE *hA = (Mat_HYPRE*)A->data;
1380dd9c0a25Sstefano_zampini   HYPRE_Int type;
1381dd9c0a25Sstefano_zampini 
1382dd9c0a25Sstefano_zampini   PetscFunctionBegin;
1383dd9c0a25Sstefano_zampini   if (!hA->ij) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"HYPRE_IJMatrix not present");
1384dd9c0a25Sstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type));
1385dd9c0a25Sstefano_zampini   if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"HYPRE_IJMatrix is not of type HYPRE_PARCSR");
1386dd9c0a25Sstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)parcsr));
1387dd9c0a25Sstefano_zampini   PetscFunctionReturn(0);
1388dd9c0a25Sstefano_zampini }
1389dd9c0a25Sstefano_zampini 
1390dd9c0a25Sstefano_zampini /*
1391dd9c0a25Sstefano_zampini    MatHYPREGetParCSR - Gets the pointer to the ParCSR matrix
1392dd9c0a25Sstefano_zampini 
1393dd9c0a25Sstefano_zampini    Not collective
1394dd9c0a25Sstefano_zampini 
1395dd9c0a25Sstefano_zampini    Input Parameters:
1396dd9c0a25Sstefano_zampini +  A  - the MATHYPRE object
1397dd9c0a25Sstefano_zampini 
1398dd9c0a25Sstefano_zampini    Output Parameter:
1399dd9c0a25Sstefano_zampini .  parcsr  - the pointer to the hypre_ParCSRMatrix
1400dd9c0a25Sstefano_zampini 
1401dd9c0a25Sstefano_zampini    Level: intermediate
1402dd9c0a25Sstefano_zampini 
1403dd9c0a25Sstefano_zampini .seealso: MatHYPRE, PetscCopyMode
1404dd9c0a25Sstefano_zampini */
1405dd9c0a25Sstefano_zampini PetscErrorCode MatHYPREGetParCSR(Mat A, hypre_ParCSRMatrix **parcsr)
1406dd9c0a25Sstefano_zampini {
1407dd9c0a25Sstefano_zampini   PetscErrorCode ierr;
1408dd9c0a25Sstefano_zampini 
1409dd9c0a25Sstefano_zampini   PetscFunctionBegin;
1410dd9c0a25Sstefano_zampini   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
1411dd9c0a25Sstefano_zampini   PetscValidType(A,1);
1412dd9c0a25Sstefano_zampini   ierr = PetscUseMethod(A,"MatHYPREGetParCSR_C",(Mat,hypre_ParCSRMatrix**),(A,parcsr));CHKERRQ(ierr);
1413dd9c0a25Sstefano_zampini   PetscFunctionReturn(0);
1414dd9c0a25Sstefano_zampini }
1415dd9c0a25Sstefano_zampini 
141668ec7858SStefano Zampini static PetscErrorCode MatMissingDiagonal_HYPRE(Mat A, PetscBool *missing, PetscInt *dd)
141768ec7858SStefano Zampini {
141868ec7858SStefano Zampini   hypre_ParCSRMatrix *parcsr;
141968ec7858SStefano Zampini   hypre_CSRMatrix    *ha;
142068ec7858SStefano Zampini   PetscInt           rst;
142168ec7858SStefano Zampini   PetscErrorCode     ierr;
142268ec7858SStefano Zampini 
142368ec7858SStefano Zampini   PetscFunctionBegin;
142468ec7858SStefano Zampini   if (A->rmap->n != A->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not implemented with non-square diagonal blocks");
142568ec7858SStefano Zampini   ierr = MatGetOwnershipRange(A,&rst,NULL);CHKERRQ(ierr);
142668ec7858SStefano Zampini   ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
142768ec7858SStefano Zampini   if (missing) *missing = PETSC_FALSE;
142868ec7858SStefano Zampini   if (dd) *dd = -1;
142968ec7858SStefano Zampini   ha = hypre_ParCSRMatrixDiag(parcsr);
143068ec7858SStefano Zampini   if (ha) {
143168299464SStefano Zampini     PetscInt  size,i;
143268299464SStefano Zampini     HYPRE_Int *ii,*jj;
143368ec7858SStefano Zampini 
143468ec7858SStefano Zampini     size = hypre_CSRMatrixNumRows(ha);
143568ec7858SStefano Zampini     ii   = hypre_CSRMatrixI(ha);
143668ec7858SStefano Zampini     jj   = hypre_CSRMatrixJ(ha);
143768ec7858SStefano Zampini     for (i = 0; i < size; i++) {
143868ec7858SStefano Zampini       PetscInt  j;
143968ec7858SStefano Zampini       PetscBool found = PETSC_FALSE;
144068ec7858SStefano Zampini 
144168ec7858SStefano Zampini       for (j = ii[i]; j < ii[i+1] && !found; j++)
144268ec7858SStefano Zampini         found = (jj[j] == i) ? PETSC_TRUE : PETSC_FALSE;
144368ec7858SStefano Zampini 
144468ec7858SStefano Zampini       if (!found) {
144568ec7858SStefano Zampini         PetscInfo1(A,"Matrix is missing local diagonal entry %D\n",i);
144668ec7858SStefano Zampini         if (missing) *missing = PETSC_TRUE;
144768ec7858SStefano Zampini         if (dd) *dd = i+rst;
144868ec7858SStefano Zampini         PetscFunctionReturn(0);
144968ec7858SStefano Zampini       }
145068ec7858SStefano Zampini     }
145168ec7858SStefano Zampini     if (!size) {
145268ec7858SStefano Zampini       PetscInfo(A,"Matrix has no diagonal entries therefore is missing diagonal\n");
145368ec7858SStefano Zampini       if (missing) *missing = PETSC_TRUE;
145468ec7858SStefano Zampini       if (dd) *dd = rst;
145568ec7858SStefano Zampini     }
145668ec7858SStefano Zampini   } else {
145768ec7858SStefano Zampini     PetscInfo(A,"Matrix has no diagonal entries therefore is missing diagonal\n");
145868ec7858SStefano Zampini     if (missing) *missing = PETSC_TRUE;
145968ec7858SStefano Zampini     if (dd) *dd = rst;
146068ec7858SStefano Zampini   }
146168ec7858SStefano Zampini   PetscFunctionReturn(0);
146268ec7858SStefano Zampini }
146368ec7858SStefano Zampini 
146468ec7858SStefano Zampini static PetscErrorCode MatScale_HYPRE(Mat A, PetscScalar s)
146568ec7858SStefano Zampini {
146668ec7858SStefano Zampini   hypre_ParCSRMatrix *parcsr;
146768ec7858SStefano Zampini   hypre_CSRMatrix    *ha;
146868ec7858SStefano Zampini   PetscErrorCode     ierr;
146968ec7858SStefano Zampini 
147068ec7858SStefano Zampini   PetscFunctionBegin;
147168ec7858SStefano Zampini   ierr  = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
147268ec7858SStefano Zampini   /* diagonal part */
147368ec7858SStefano Zampini   ha = hypre_ParCSRMatrixDiag(parcsr);
147468ec7858SStefano Zampini   if (ha) {
147568299464SStefano Zampini     PetscInt    size,i;
147668299464SStefano Zampini     HYPRE_Int   *ii;
147768ec7858SStefano Zampini     PetscScalar *a;
147868ec7858SStefano Zampini 
147968ec7858SStefano Zampini     size = hypre_CSRMatrixNumRows(ha);
148068ec7858SStefano Zampini     a    = hypre_CSRMatrixData(ha);
148168ec7858SStefano Zampini     ii   = hypre_CSRMatrixI(ha);
148268ec7858SStefano Zampini     for (i = 0; i < ii[size]; i++) a[i] *= s;
148368ec7858SStefano Zampini   }
148468ec7858SStefano Zampini   /* offdiagonal part */
148568ec7858SStefano Zampini   ha = hypre_ParCSRMatrixOffd(parcsr);
148668ec7858SStefano Zampini   if (ha) {
148768299464SStefano Zampini     PetscInt    size,i;
148868299464SStefano Zampini     HYPRE_Int   *ii;
148968ec7858SStefano Zampini     PetscScalar *a;
149068ec7858SStefano Zampini 
149168ec7858SStefano Zampini     size = hypre_CSRMatrixNumRows(ha);
149268ec7858SStefano Zampini     a    = hypre_CSRMatrixData(ha);
149368ec7858SStefano Zampini     ii   = hypre_CSRMatrixI(ha);
149468ec7858SStefano Zampini     for (i = 0; i < ii[size]; i++) a[i] *= s;
149568ec7858SStefano Zampini   }
149668ec7858SStefano Zampini   PetscFunctionReturn(0);
149768ec7858SStefano Zampini }
149868ec7858SStefano Zampini 
149968ec7858SStefano Zampini static PetscErrorCode MatZeroRowsColumns_HYPRE(Mat A, PetscInt numRows, const PetscInt rows[], PetscScalar diag, Vec x, Vec b)
150068ec7858SStefano Zampini {
150168ec7858SStefano Zampini   hypre_ParCSRMatrix *parcsr;
150268299464SStefano Zampini   HYPRE_Int          *lrows;
150368299464SStefano Zampini   PetscInt           rst,ren,i;
150468ec7858SStefano Zampini   PetscErrorCode     ierr;
150568ec7858SStefano Zampini 
150668ec7858SStefano Zampini   PetscFunctionBegin;
150768ec7858SStefano Zampini   if (x || b) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"To be implemented");
150868ec7858SStefano Zampini   ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
150968ec7858SStefano Zampini   ierr = PetscMalloc1(numRows,&lrows);CHKERRQ(ierr);
151068ec7858SStefano Zampini   ierr = MatGetOwnershipRange(A,&rst,&ren);CHKERRQ(ierr);
151168ec7858SStefano Zampini   for (i=0;i<numRows;i++) {
151268ec7858SStefano Zampini     if (rows[i] < rst || rows[i] >= ren)
151368ec7858SStefano Zampini       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Non-local rows not yet supported");
151468ec7858SStefano Zampini     lrows[i] = rows[i] - rst;
151568ec7858SStefano Zampini   }
151668ec7858SStefano Zampini   PetscStackCallStandard(hypre_ParCSRMatrixEliminateRowsCols,(parcsr,numRows,lrows));
151768ec7858SStefano Zampini   ierr = PetscFree(lrows);CHKERRQ(ierr);
151868ec7858SStefano Zampini   PetscFunctionReturn(0);
151968ec7858SStefano Zampini }
152068ec7858SStefano Zampini 
1521c69f721fSFande Kong static PetscErrorCode MatZeroEntries_HYPRE_CSRMatrix(hypre_CSRMatrix *ha)
1522c69f721fSFande Kong {
1523c69f721fSFande Kong   PetscErrorCode      ierr;
1524c69f721fSFande Kong 
1525c69f721fSFande Kong   PetscFunctionBegin;
1526c69f721fSFande Kong   if (ha) {
1527c69f721fSFande Kong     HYPRE_Int     *ii, size;
1528c69f721fSFande Kong     HYPRE_Complex *a;
1529c69f721fSFande Kong 
1530c69f721fSFande Kong     size = hypre_CSRMatrixNumRows(ha);
1531c69f721fSFande Kong     a    = hypre_CSRMatrixData(ha);
1532c69f721fSFande Kong     ii   = hypre_CSRMatrixI(ha);
1533c69f721fSFande Kong 
1534c69f721fSFande Kong     if (a) { ierr = PetscMemzero(a,(ii[size])*sizeof(HYPRE_Complex));CHKERRQ(ierr); }
1535c69f721fSFande Kong   }
1536c69f721fSFande Kong   PetscFunctionReturn(0);
1537c69f721fSFande Kong }
1538c69f721fSFande Kong 
1539c69f721fSFande Kong PetscErrorCode MatZeroEntries_HYPRE(Mat A)
1540c69f721fSFande Kong {
1541c69f721fSFande Kong   hypre_ParCSRMatrix  *parcsr;
1542c69f721fSFande Kong   PetscErrorCode      ierr;
1543c69f721fSFande Kong 
1544c69f721fSFande Kong   PetscFunctionBegin;
1545c69f721fSFande Kong   ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
1546c69f721fSFande Kong   /* diagonal part */
1547c69f721fSFande Kong   ierr = MatZeroEntries_HYPRE_CSRMatrix(hypre_ParCSRMatrixDiag(parcsr));CHKERRQ(ierr);
1548c69f721fSFande Kong   /* off-diagonal part */
1549c69f721fSFande Kong   ierr = MatZeroEntries_HYPRE_CSRMatrix(hypre_ParCSRMatrixOffd(parcsr));CHKERRQ(ierr);
1550c69f721fSFande Kong   PetscFunctionReturn(0);
1551c69f721fSFande Kong }
1552c69f721fSFande Kong 
1553c69f721fSFande Kong static PetscErrorCode MatZeroRows_HYPRE_CSRMatrix(hypre_CSRMatrix *hA,PetscInt N,const PetscInt rows[],PetscScalar diag)
1554c69f721fSFande Kong {
1555c69f721fSFande Kong   PetscInt        ii, jj, ibeg, iend, irow;
1556c69f721fSFande Kong   PetscInt        *i, *j;
1557c69f721fSFande Kong   PetscScalar     *a;
1558c69f721fSFande Kong 
1559c69f721fSFande Kong   PetscFunctionBegin;
1560c69f721fSFande Kong   if (!hA) PetscFunctionReturn(0);
1561c69f721fSFande Kong 
156208defe43SFande Kong   i = (PetscInt*) hypre_CSRMatrixI(hA);
156308defe43SFande Kong   j = (PetscInt*) hypre_CSRMatrixJ(hA);
1564c69f721fSFande Kong   a = hypre_CSRMatrixData(hA);
1565c69f721fSFande Kong 
1566c69f721fSFande Kong   for (ii = 0; ii < N; ii++) {
1567c69f721fSFande Kong     irow = rows[ii];
1568c69f721fSFande Kong     ibeg = i[irow];
1569c69f721fSFande Kong     iend = i[irow+1];
1570c69f721fSFande Kong     for (jj = ibeg; jj < iend; jj++)
1571c69f721fSFande Kong       if (j[jj] == irow) a[jj] = diag;
1572c69f721fSFande Kong       else a[jj] = 0.0;
1573c69f721fSFande Kong    }
1574c69f721fSFande Kong    PetscFunctionReturn(0);
1575c69f721fSFande Kong }
1576c69f721fSFande Kong 
1577ddbeb582SStefano Zampini static PetscErrorCode MatZeroRows_HYPRE(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag,Vec x,Vec b)
1578c69f721fSFande Kong {
1579c69f721fSFande Kong   hypre_ParCSRMatrix  *parcsr;
1580c69f721fSFande Kong   PetscInt            *lrows,len;
1581c69f721fSFande Kong   PetscErrorCode      ierr;
1582c69f721fSFande Kong 
1583c69f721fSFande Kong   PetscFunctionBegin;
1584c69f721fSFande Kong   if (x || b) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Does not support to modify the solution and the right hand size \n");
1585c69f721fSFande Kong   /* retrieve the internal matrix */
1586c69f721fSFande Kong   ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
1587c69f721fSFande Kong   /* get locally owned rows */
1588c69f721fSFande Kong   ierr = MatZeroRowsMapLocal_Private(A,N,rows,&len,&lrows);CHKERRQ(ierr);
1589c69f721fSFande Kong   /* zero diagonal part */
1590c69f721fSFande Kong   ierr = MatZeroRows_HYPRE_CSRMatrix(hypre_ParCSRMatrixDiag(parcsr),len,lrows,diag);CHKERRQ(ierr);
1591c69f721fSFande Kong   /* zero off-diagonal part */
1592c69f721fSFande Kong   ierr = MatZeroRows_HYPRE_CSRMatrix(hypre_ParCSRMatrixOffd(parcsr),len,lrows,0.0);CHKERRQ(ierr);
1593c69f721fSFande Kong 
1594c69f721fSFande Kong   ierr = PetscFree(lrows);CHKERRQ(ierr);
1595c69f721fSFande Kong   PetscFunctionReturn(0);
1596c69f721fSFande Kong }
1597c69f721fSFande Kong 
1598ddbeb582SStefano Zampini static PetscErrorCode MatAssemblyBegin_HYPRE(Mat mat,MatAssemblyType mode)
1599c69f721fSFande Kong {
1600c69f721fSFande Kong   PetscErrorCode ierr;
1601c69f721fSFande Kong 
1602c69f721fSFande Kong   PetscFunctionBegin;
1603c69f721fSFande Kong   if (mat->nooffprocentries) PetscFunctionReturn(0);
1604c69f721fSFande Kong 
1605c69f721fSFande Kong   ierr = MatStashScatterBegin_Private(mat,&mat->stash,mat->rmap->range);CHKERRQ(ierr);
1606c69f721fSFande Kong   PetscFunctionReturn(0);
1607c69f721fSFande Kong }
1608c69f721fSFande Kong 
1609ddbeb582SStefano Zampini static PetscErrorCode MatGetRow_HYPRE(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
1610c69f721fSFande Kong {
1611c69f721fSFande Kong   hypre_ParCSRMatrix  *parcsr;
1612c69f721fSFande Kong   PetscErrorCode      ierr;
1613c69f721fSFande Kong 
1614c69f721fSFande Kong   PetscFunctionBegin;
1615c69f721fSFande Kong   /* retrieve the internal matrix */
1616c69f721fSFande Kong   ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
1617c69f721fSFande Kong   /* call HYPRE API */
161808defe43SFande Kong   PetscStackCallStandard(HYPRE_ParCSRMatrixGetRow,(parcsr,row,(HYPRE_Int*)nz,(HYPRE_Int**)idx,v));
1619c69f721fSFande Kong   PetscFunctionReturn(0);
1620c69f721fSFande Kong }
1621c69f721fSFande Kong 
1622ddbeb582SStefano Zampini static PetscErrorCode MatRestoreRow_HYPRE(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
1623c69f721fSFande Kong {
1624c69f721fSFande Kong   hypre_ParCSRMatrix  *parcsr;
1625c69f721fSFande Kong   PetscErrorCode      ierr;
1626c69f721fSFande Kong 
1627c69f721fSFande Kong   PetscFunctionBegin;
1628c69f721fSFande Kong   /* retrieve the internal matrix */
1629c69f721fSFande Kong   ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
1630c69f721fSFande Kong   /* call HYPRE API */
163108defe43SFande Kong   PetscStackCallStandard(HYPRE_ParCSRMatrixRestoreRow,(parcsr,row,(HYPRE_Int*)nz,(HYPRE_Int**)idx,v));
1632c69f721fSFande Kong   PetscFunctionReturn(0);
1633c69f721fSFande Kong }
1634c69f721fSFande Kong 
1635ddbeb582SStefano Zampini static PetscErrorCode MatGetValues_HYPRE(Mat A,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],PetscScalar v[])
1636c69f721fSFande Kong {
163745b8d346SStefano Zampini   Mat_HYPRE *hA = (Mat_HYPRE*)A->data;
1638c69f721fSFande Kong   PetscInt  i;
16391d4906efSStefano Zampini 
1640c69f721fSFande Kong   PetscFunctionBegin;
1641c69f721fSFande Kong   if (!m || !n) PetscFunctionReturn(0);
1642c69f721fSFande Kong   /* Ignore negative row indices
1643c69f721fSFande Kong    * And negative column indices should be automatically ignored in hypre
1644c69f721fSFande Kong    * */
1645c69f721fSFande Kong   for (i=0; i<m; i++)
164645b8d346SStefano Zampini     if (idxm[i] >= 0) PetscStackCallStandard(HYPRE_IJMatrixGetValues,(hA->ij,1,(HYPRE_Int*)&n,(HYPRE_Int*)&idxm[i],(HYPRE_Int*)idxn,&v[i*n]));
1647c69f721fSFande Kong   PetscFunctionReturn(0);
1648c69f721fSFande Kong }
1649c69f721fSFande Kong 
1650ddbeb582SStefano Zampini static PetscErrorCode MatSetOption_HYPRE(Mat A,MatOption op,PetscBool flg)
1651ddbeb582SStefano Zampini {
1652ddbeb582SStefano Zampini   Mat_HYPRE *hA = (Mat_HYPRE*)A->data;
1653ddbeb582SStefano Zampini 
1654ddbeb582SStefano Zampini   PetscFunctionBegin;
1655ddbeb582SStefano Zampini   switch (op)
1656ddbeb582SStefano Zampini   {
1657ddbeb582SStefano Zampini   case MAT_NO_OFF_PROC_ENTRIES:
1658ddbeb582SStefano Zampini     if (flg) {
1659ddbeb582SStefano Zampini       PetscStackCallStandard(HYPRE_IJMatrixSetMaxOffProcElmts,(hA->ij,0));
1660ddbeb582SStefano Zampini     }
1661ddbeb582SStefano Zampini     break;
1662ddbeb582SStefano Zampini   default:
1663ddbeb582SStefano Zampini     break;
1664ddbeb582SStefano Zampini   }
1665ddbeb582SStefano Zampini   PetscFunctionReturn(0);
1666ddbeb582SStefano Zampini }
1667c69f721fSFande Kong 
166845b8d346SStefano Zampini static PetscErrorCode MatView_HYPRE(Mat A, PetscViewer view)
166945b8d346SStefano Zampini {
167045b8d346SStefano Zampini   hypre_ParCSRMatrix *parcsr;
167145b8d346SStefano Zampini   PetscErrorCode     ierr;
167245b8d346SStefano Zampini   Mat                B;
167345b8d346SStefano Zampini   PetscViewerFormat  format;
167445b8d346SStefano Zampini   PetscErrorCode     (*mview)(Mat,PetscViewer) = NULL;
167545b8d346SStefano Zampini 
167645b8d346SStefano Zampini   PetscFunctionBegin;
167745b8d346SStefano Zampini   ierr = PetscViewerGetFormat(view,&format);CHKERRQ(ierr);
167845b8d346SStefano Zampini   if (format != PETSC_VIEWER_NATIVE) {
167945b8d346SStefano Zampini     ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
168045b8d346SStefano Zampini     ierr = MatCreateFromParCSR(parcsr,MATAIJ,PETSC_USE_POINTER,&B);CHKERRQ(ierr);
168145b8d346SStefano Zampini     ierr = MatGetOperation(B,MATOP_VIEW,(void(**)(void))&mview);CHKERRQ(ierr);
168245b8d346SStefano Zampini     if (!mview) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Missing view operation");CHKERRQ(ierr);
168345b8d346SStefano Zampini     ierr = (*mview)(B,view);CHKERRQ(ierr);
168445b8d346SStefano Zampini     ierr = MatDestroy(&B);CHKERRQ(ierr);
168545b8d346SStefano Zampini   } else {
168645b8d346SStefano Zampini     Mat_HYPRE  *hA = (Mat_HYPRE*)A->data;
168745b8d346SStefano Zampini     PetscMPIInt size;
168845b8d346SStefano Zampini     PetscBool   isascii;
168945b8d346SStefano Zampini     const char *filename;
169045b8d346SStefano Zampini 
169145b8d346SStefano Zampini     /* HYPRE uses only text files */
169245b8d346SStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)view,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr);
169345b8d346SStefano Zampini     if (!isascii) SETERRQ1(PetscObjectComm((PetscObject)view),PETSC_ERR_SUP,"PetscViewerType %s: native HYPRE format needs PETSCVIEWERASCII",((PetscObject)view)->type_name);
169445b8d346SStefano Zampini     ierr = PetscViewerFileGetName(view,&filename);CHKERRQ(ierr);
169545b8d346SStefano Zampini     PetscStackCallStandard(HYPRE_IJMatrixPrint,(hA->ij,filename));
169645b8d346SStefano Zampini     ierr = MPI_Comm_size(hA->comm,&size);CHKERRQ(ierr);
169745b8d346SStefano Zampini     if (size > 1) {
169845b8d346SStefano Zampini       ierr = PetscViewerASCIIPrintf(view,"Matrix files: %s.%05d ... %s.%05d\n",filename,0,filename,size-1);CHKERRQ(ierr);
169945b8d346SStefano Zampini     } else {
170045b8d346SStefano Zampini       ierr = PetscViewerASCIIPrintf(view,"Matrix file: %s.%05d\n",filename,0);CHKERRQ(ierr);
170145b8d346SStefano Zampini     }
170245b8d346SStefano Zampini   }
170345b8d346SStefano Zampini   PetscFunctionReturn(0);
170445b8d346SStefano Zampini }
170545b8d346SStefano Zampini 
170645b8d346SStefano Zampini static PetscErrorCode MatDuplicate_HYPRE(Mat A,MatDuplicateOption op, Mat *B)
170745b8d346SStefano Zampini {
170845b8d346SStefano Zampini   hypre_ParCSRMatrix *parcsr;
170945b8d346SStefano Zampini   PetscErrorCode     ierr;
171045b8d346SStefano Zampini   PetscCopyMode      cpmode;
171145b8d346SStefano Zampini 
171245b8d346SStefano Zampini   PetscFunctionBegin;
171345b8d346SStefano Zampini   ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
171445b8d346SStefano Zampini   if (op == MAT_DO_NOT_COPY_VALUES || op == MAT_SHARE_NONZERO_PATTERN) {
171545b8d346SStefano Zampini     parcsr = hypre_ParCSRMatrixCompleteClone(parcsr);
171645b8d346SStefano Zampini     cpmode = PETSC_OWN_POINTER;
171745b8d346SStefano Zampini   } else {
171845b8d346SStefano Zampini     cpmode = PETSC_COPY_VALUES;
171945b8d346SStefano Zampini   }
172045b8d346SStefano Zampini   ierr = MatCreateFromParCSR(parcsr,MATHYPRE,cpmode,B);CHKERRQ(ierr);
172145b8d346SStefano Zampini   PetscFunctionReturn(0);
172245b8d346SStefano Zampini }
172345b8d346SStefano Zampini 
1724a055b5aaSBarry Smith /*MC
1725a055b5aaSBarry Smith    MATHYPRE - MATHYPRE = "hypre" - A matrix type to be used for sequential and parallel sparse matrices
1726a055b5aaSBarry Smith           based on the hypre IJ interface.
1727a055b5aaSBarry Smith 
1728a055b5aaSBarry Smith    Level: intermediate
1729a055b5aaSBarry Smith 
1730a055b5aaSBarry Smith .seealso: MatCreate()
1731a055b5aaSBarry Smith M*/
1732a055b5aaSBarry Smith 
173363c07aadSStefano Zampini PETSC_EXTERN PetscErrorCode MatCreate_HYPRE(Mat B)
173463c07aadSStefano Zampini {
173563c07aadSStefano Zampini   Mat_HYPRE      *hB;
173663c07aadSStefano Zampini   PetscErrorCode ierr;
173763c07aadSStefano Zampini 
173863c07aadSStefano Zampini   PetscFunctionBegin;
173963c07aadSStefano Zampini   ierr           = PetscNewLog(B,&hB);CHKERRQ(ierr);
1740978814f1SStefano Zampini   hB->inner_free = PETSC_TRUE;
1741c69f721fSFande Kong   hB->available  = PETSC_TRUE;
1742c69f721fSFande Kong   hB->size       = 0;
1743c69f721fSFande Kong   hB->array      = NULL;
1744978814f1SStefano Zampini 
174563c07aadSStefano Zampini   B->data       = (void*)hB;
174663c07aadSStefano Zampini   B->rmap->bs   = 1;
174763c07aadSStefano Zampini   B->assembled  = PETSC_FALSE;
174863c07aadSStefano Zampini 
1749de393100SStefano Zampini   ierr = PetscMemzero(B->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
175063c07aadSStefano Zampini   B->ops->mult            = MatMult_HYPRE;
175163c07aadSStefano Zampini   B->ops->multtranspose   = MatMultTranspose_HYPRE;
175263c07aadSStefano Zampini   B->ops->setup           = MatSetUp_HYPRE;
175363c07aadSStefano Zampini   B->ops->destroy         = MatDestroy_HYPRE;
175463c07aadSStefano Zampini   B->ops->assemblyend     = MatAssemblyEnd_HYPRE;
1755c69f721fSFande Kong   B->ops->assemblybegin   = MatAssemblyBegin_HYPRE;
1756cd8bc7baSStefano Zampini   B->ops->ptap            = MatPtAP_HYPRE_HYPRE;
1757d501dc42Sstefano_zampini   B->ops->matmult         = MatMatMult_HYPRE_HYPRE;
1758d975228cSstefano_zampini   B->ops->setvalues       = MatSetValues_HYPRE;
175968ec7858SStefano Zampini   B->ops->missingdiagonal = MatMissingDiagonal_HYPRE;
176068ec7858SStefano Zampini   B->ops->scale           = MatScale_HYPRE;
176168ec7858SStefano Zampini   B->ops->zerorowscolumns = MatZeroRowsColumns_HYPRE;
1762c69f721fSFande Kong   B->ops->zeroentries     = MatZeroEntries_HYPRE;
1763c69f721fSFande Kong   B->ops->zerorows        = MatZeroRows_HYPRE;
1764c69f721fSFande Kong   B->ops->getrow          = MatGetRow_HYPRE;
1765c69f721fSFande Kong   B->ops->restorerow      = MatRestoreRow_HYPRE;
1766c69f721fSFande Kong   B->ops->getvalues       = MatGetValues_HYPRE;
1767ddbeb582SStefano Zampini   B->ops->setoption       = MatSetOption_HYPRE;
176845b8d346SStefano Zampini   B->ops->duplicate       = MatDuplicate_HYPRE;
176945b8d346SStefano Zampini   B->ops->view            = MatView_HYPRE;
177045b8d346SStefano Zampini 
177145b8d346SStefano Zampini   /* build cache for off array entries formed */
177245b8d346SStefano Zampini   ierr = MatStashCreate_Private(PetscObjectComm((PetscObject)B),1,&B->stash);CHKERRQ(ierr);
177363c07aadSStefano Zampini 
177463c07aadSStefano Zampini   ierr = MPI_Comm_dup(PetscObjectComm((PetscObject)B),&hB->comm);CHKERRQ(ierr);
177563c07aadSStefano Zampini   ierr = PetscObjectChangeTypeName((PetscObject)B,MATHYPRE);CHKERRQ(ierr);
177663c07aadSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_hypre_aij_C",MatConvert_HYPRE_AIJ);CHKERRQ(ierr);
17772df22349SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_hypre_is_C",MatConvert_HYPRE_IS);CHKERRQ(ierr);
1778c1a070e6SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)B,"MatPtAP_seqaij_hypre_C",MatPtAP_AIJ_HYPRE);CHKERRQ(ierr);
1779c1a070e6SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)B,"MatPtAP_mpiaij_hypre_C",MatPtAP_AIJ_HYPRE);CHKERRQ(ierr);
1780d975228cSstefano_zampini   ierr = PetscObjectComposeFunction((PetscObject)B,"MatHYPRESetPreallocation_C",MatHYPRESetPreallocation_HYPRE);CHKERRQ(ierr);
1781dd9c0a25Sstefano_zampini   ierr = PetscObjectComposeFunction((PetscObject)B,"MatHYPREGetParCSR_C",MatHYPREGetParCSR_HYPRE);CHKERRQ(ierr);
178275d48cdbSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)B,"MatPtAP_is_hypre_C",MatPtAP_IS_XAIJ);CHKERRQ(ierr);
178363c07aadSStefano Zampini   PetscFunctionReturn(0);
178463c07aadSStefano Zampini }
178563c07aadSStefano Zampini 
1786225daaf8SStefano Zampini static PetscErrorCode hypre_array_destroy(void *ptr)
1787225daaf8SStefano Zampini {
1788225daaf8SStefano Zampini    PetscFunctionBegin;
1789e6de0934SSatish Balay    hypre_TFree(ptr,HYPRE_MEMORY_HOST);
1790225daaf8SStefano Zampini    PetscFunctionReturn(0);
1791225daaf8SStefano Zampini }
1792