xref: /petsc/src/mat/impls/hypre/mhypre.c (revision af1cf9688993dc3560bc70b288db2a3879cfea1c)
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));
1013*af1cf968SStefano Zampini   /* The assembly routine destroys the aux_matrix, we recreate it here by calling HYPRE_IJMatrixInitalize */
1014*af1cf968SStefano Zampini   {
1015*af1cf968SStefano Zampini     hypre_AuxParCSRMatrix *aux_matrix;
1016*af1cf968SStefano Zampini 
1017*af1cf968SStefano Zampini     /* call destroy just to make sure we do not leak anything */
1018*af1cf968SStefano Zampini     aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(hA->ij);
1019*af1cf968SStefano Zampini     PetscStackCallStandard(hypre_AuxParCSRMatrixDestroy,(aux_matrix));
1020*af1cf968SStefano Zampini     hypre_IJMatrixTranslator(hA->ij) = NULL;
1021*af1cf968SStefano Zampini 
1022*af1cf968SStefano Zampini     /* Initialize with assembled flag -> it only recreates the aux_par_matrix */
1023*af1cf968SStefano Zampini     PetscStackCallStandard(HYPRE_IJMatrixInitialize,(hA->ij));
1024*af1cf968SStefano Zampini     aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(hA->ij);
1025*af1cf968SStefano Zampini     hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 1; /* see comment in MatHYPRESetPreallocation_HYPRE */
1026*af1cf968SStefano Zampini     PetscStackCallStandard(hypre_AuxParCSRMatrixInitialize,(aux_matrix));
1027*af1cf968SStefano 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 
1188*af1cf968SStefano Zampini   /* reset assembled flag and call the initialize method */
1189*af1cf968SStefano 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   }
1197*af1cf968SStefano Zampini 
1198*af1cf968SStefano Zampini   /* Match AIJ logic */
119906a29025Sstefano_zampini   A->preallocated = PETSC_TRUE;
1200*af1cf968SStefano 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 
1234*af1cf968SStefano 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 
1294978814f1SStefano Zampini   /* create PETSc matrix with MatHYPRE */
1295225daaf8SStefano Zampini   ierr = MatCreate(comm,&T);CHKERRQ(ierr);
1296225daaf8SStefano Zampini   ierr = MatSetSizes(T,rend-rstart+1,cend-cstart+1,M,N);CHKERRQ(ierr);
1297225daaf8SStefano Zampini   ierr = MatSetType(T,MATHYPRE);CHKERRQ(ierr);
1298225daaf8SStefano Zampini   hA   = (Mat_HYPRE*)(T->data);
1299978814f1SStefano Zampini 
1300978814f1SStefano Zampini   /* create HYPRE_IJMatrix */
1301978814f1SStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixCreate,(hA->comm,rstart,rend-1,cstart,cend-1,&hA->ij));
130245b8d346SStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(hA->ij,HYPRE_PARCSR));
130345b8d346SStefano Zampini 
130445b8d346SStefano Zampini   /* create new ParCSR object if needed */
130545b8d346SStefano Zampini   if (ishyp && copymode == PETSC_COPY_VALUES) {
130645b8d346SStefano Zampini     hypre_ParCSRMatrix *new_parcsr;
130745b8d346SStefano Zampini     hypre_CSRMatrix    *hdiag,*hoffd,*ndiag,*noffd;
130845b8d346SStefano Zampini 
130945b8d346SStefano Zampini     new_parcsr = hypre_ParCSRMatrixCompleteClone(parcsr);
131045b8d346SStefano Zampini     hdiag      = hypre_ParCSRMatrixDiag(parcsr);
131145b8d346SStefano Zampini     hoffd      = hypre_ParCSRMatrixOffd(parcsr);
131245b8d346SStefano Zampini     ndiag      = hypre_ParCSRMatrixDiag(new_parcsr);
131345b8d346SStefano Zampini     noffd      = hypre_ParCSRMatrixOffd(new_parcsr);
131445b8d346SStefano Zampini     ierr       = PetscMemcpy(hypre_CSRMatrixData(ndiag),hypre_CSRMatrixData(hdiag),hypre_CSRMatrixNumNonzeros(hdiag)*sizeof(PetscScalar));CHKERRQ(ierr);
131545b8d346SStefano Zampini     ierr       = PetscMemcpy(hypre_CSRMatrixData(noffd),hypre_CSRMatrixData(hoffd),hypre_CSRMatrixNumNonzeros(hoffd)*sizeof(PetscScalar));CHKERRQ(ierr);
131645b8d346SStefano Zampini     parcsr     = new_parcsr;
131745b8d346SStefano Zampini     copymode   = PETSC_OWN_POINTER;
131845b8d346SStefano Zampini   }
1319978814f1SStefano Zampini 
1320978814f1SStefano Zampini   /* set ParCSR object */
1321978814f1SStefano Zampini   hypre_IJMatrixObject(hA->ij) = parcsr;
13224ec6421dSstefano_zampini   T->preallocated = PETSC_TRUE;
1323978814f1SStefano Zampini 
1324978814f1SStefano Zampini   /* set assembled flag */
1325978814f1SStefano Zampini   hypre_IJMatrixAssembleFlag(hA->ij) = 1;
1326978814f1SStefano Zampini   PetscStackCallStandard(HYPRE_IJMatrixInitialize,(hA->ij));
1327225daaf8SStefano Zampini   if (ishyp) {
13286d2a658fSstefano_zampini     PetscMPIInt myid = 0;
13296d2a658fSstefano_zampini 
13306d2a658fSstefano_zampini     /* make sure we always have row_starts and col_starts available */
13316d2a658fSstefano_zampini     if (HYPRE_AssumedPartitionCheck()) {
13326d2a658fSstefano_zampini       ierr = MPI_Comm_rank(comm,&myid);CHKERRQ(ierr);
13336d2a658fSstefano_zampini     }
13346d2a658fSstefano_zampini     if (!hypre_ParCSRMatrixOwnsColStarts(parcsr)) {
13356d2a658fSstefano_zampini       PetscLayout map;
13366d2a658fSstefano_zampini 
13376d2a658fSstefano_zampini       ierr = MatGetLayouts(T,NULL,&map);CHKERRQ(ierr);
13386d2a658fSstefano_zampini       ierr = PetscLayoutSetUp(map);CHKERRQ(ierr);
13391c92d2d0Sstefano_zampini       hypre_ParCSRMatrixColStarts(parcsr) = (HYPRE_Int*)(map->range + myid);
13406d2a658fSstefano_zampini     }
13416d2a658fSstefano_zampini     if (!hypre_ParCSRMatrixOwnsRowStarts(parcsr)) {
13426d2a658fSstefano_zampini       PetscLayout map;
13436d2a658fSstefano_zampini 
13446d2a658fSstefano_zampini       ierr = MatGetLayouts(T,&map,NULL);CHKERRQ(ierr);
13456d2a658fSstefano_zampini       ierr = PetscLayoutSetUp(map);CHKERRQ(ierr);
13461c92d2d0Sstefano_zampini       hypre_ParCSRMatrixRowStarts(parcsr) = (HYPRE_Int*)(map->range + myid);
13476d2a658fSstefano_zampini     }
1348978814f1SStefano Zampini     /* prevent from freeing the pointer */
1349978814f1SStefano Zampini     if (copymode == PETSC_USE_POINTER) hA->inner_free = PETSC_FALSE;
1350225daaf8SStefano Zampini     *A   = T;
1351978814f1SStefano Zampini     ierr = MatAssemblyBegin(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1352978814f1SStefano Zampini     ierr = MatAssemblyEnd(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1353bb4689ddSStefano Zampini   } else if (isaij) {
1354bb4689ddSStefano Zampini     if (copymode != PETSC_OWN_POINTER) {
1355225daaf8SStefano Zampini       /* prevent from freeing the pointer */
1356225daaf8SStefano Zampini       hA->inner_free = PETSC_FALSE;
1357225daaf8SStefano Zampini       ierr = MatConvert_HYPRE_AIJ(T,MATAIJ,MAT_INITIAL_MATRIX,A);CHKERRQ(ierr);
1358225daaf8SStefano Zampini       ierr = MatDestroy(&T);CHKERRQ(ierr);
1359225daaf8SStefano Zampini     } else { /* AIJ return type with PETSC_OWN_POINTER */
1360225daaf8SStefano Zampini       ierr = MatConvert_HYPRE_AIJ(T,MATAIJ,MAT_INPLACE_MATRIX,&T);CHKERRQ(ierr);
1361225daaf8SStefano Zampini       *A   = T;
1362225daaf8SStefano Zampini     }
1363bb4689ddSStefano Zampini   } else if (isis) {
13648cfe8d00SStefano Zampini     ierr = MatConvert_HYPRE_IS(T,MATIS,MAT_INITIAL_MATRIX,A);CHKERRQ(ierr);
13658cfe8d00SStefano Zampini     if (copymode != PETSC_OWN_POINTER) hA->inner_free = PETSC_FALSE;
1366bb4689ddSStefano Zampini     ierr = MatDestroy(&T);CHKERRQ(ierr);
1367bb4689ddSStefano Zampini   }
1368978814f1SStefano Zampini   PetscFunctionReturn(0);
1369978814f1SStefano Zampini }
1370978814f1SStefano Zampini 
137168ec7858SStefano Zampini static PetscErrorCode MatHYPREGetParCSR_HYPRE(Mat A, hypre_ParCSRMatrix **parcsr)
1372dd9c0a25Sstefano_zampini {
1373dd9c0a25Sstefano_zampini   Mat_HYPRE *hA = (Mat_HYPRE*)A->data;
1374dd9c0a25Sstefano_zampini   HYPRE_Int type;
1375dd9c0a25Sstefano_zampini 
1376dd9c0a25Sstefano_zampini   PetscFunctionBegin;
1377dd9c0a25Sstefano_zampini   if (!hA->ij) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"HYPRE_IJMatrix not present");
1378dd9c0a25Sstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type));
1379dd9c0a25Sstefano_zampini   if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"HYPRE_IJMatrix is not of type HYPRE_PARCSR");
1380dd9c0a25Sstefano_zampini   PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)parcsr));
1381dd9c0a25Sstefano_zampini   PetscFunctionReturn(0);
1382dd9c0a25Sstefano_zampini }
1383dd9c0a25Sstefano_zampini 
1384dd9c0a25Sstefano_zampini /*
1385dd9c0a25Sstefano_zampini    MatHYPREGetParCSR - Gets the pointer to the ParCSR matrix
1386dd9c0a25Sstefano_zampini 
1387dd9c0a25Sstefano_zampini    Not collective
1388dd9c0a25Sstefano_zampini 
1389dd9c0a25Sstefano_zampini    Input Parameters:
1390dd9c0a25Sstefano_zampini +  A  - the MATHYPRE object
1391dd9c0a25Sstefano_zampini 
1392dd9c0a25Sstefano_zampini    Output Parameter:
1393dd9c0a25Sstefano_zampini .  parcsr  - the pointer to the hypre_ParCSRMatrix
1394dd9c0a25Sstefano_zampini 
1395dd9c0a25Sstefano_zampini    Level: intermediate
1396dd9c0a25Sstefano_zampini 
1397dd9c0a25Sstefano_zampini .seealso: MatHYPRE, PetscCopyMode
1398dd9c0a25Sstefano_zampini */
1399dd9c0a25Sstefano_zampini PetscErrorCode MatHYPREGetParCSR(Mat A, hypre_ParCSRMatrix **parcsr)
1400dd9c0a25Sstefano_zampini {
1401dd9c0a25Sstefano_zampini   PetscErrorCode ierr;
1402dd9c0a25Sstefano_zampini 
1403dd9c0a25Sstefano_zampini   PetscFunctionBegin;
1404dd9c0a25Sstefano_zampini   PetscValidHeaderSpecific(A,MAT_CLASSID,1);
1405dd9c0a25Sstefano_zampini   PetscValidType(A,1);
1406dd9c0a25Sstefano_zampini   ierr = PetscUseMethod(A,"MatHYPREGetParCSR_C",(Mat,hypre_ParCSRMatrix**),(A,parcsr));CHKERRQ(ierr);
1407dd9c0a25Sstefano_zampini   PetscFunctionReturn(0);
1408dd9c0a25Sstefano_zampini }
1409dd9c0a25Sstefano_zampini 
141068ec7858SStefano Zampini static PetscErrorCode MatMissingDiagonal_HYPRE(Mat A, PetscBool *missing, PetscInt *dd)
141168ec7858SStefano Zampini {
141268ec7858SStefano Zampini   hypre_ParCSRMatrix *parcsr;
141368ec7858SStefano Zampini   hypre_CSRMatrix    *ha;
141468ec7858SStefano Zampini   PetscInt           rst;
141568ec7858SStefano Zampini   PetscErrorCode     ierr;
141668ec7858SStefano Zampini 
141768ec7858SStefano Zampini   PetscFunctionBegin;
141868ec7858SStefano Zampini   if (A->rmap->n != A->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not implemented with non-square diagonal blocks");
141968ec7858SStefano Zampini   ierr = MatGetOwnershipRange(A,&rst,NULL);CHKERRQ(ierr);
142068ec7858SStefano Zampini   ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
142168ec7858SStefano Zampini   if (missing) *missing = PETSC_FALSE;
142268ec7858SStefano Zampini   if (dd) *dd = -1;
142368ec7858SStefano Zampini   ha = hypre_ParCSRMatrixDiag(parcsr);
142468ec7858SStefano Zampini   if (ha) {
142568299464SStefano Zampini     PetscInt  size,i;
142668299464SStefano Zampini     HYPRE_Int *ii,*jj;
142768ec7858SStefano Zampini 
142868ec7858SStefano Zampini     size = hypre_CSRMatrixNumRows(ha);
142968ec7858SStefano Zampini     ii   = hypre_CSRMatrixI(ha);
143068ec7858SStefano Zampini     jj   = hypre_CSRMatrixJ(ha);
143168ec7858SStefano Zampini     for (i = 0; i < size; i++) {
143268ec7858SStefano Zampini       PetscInt  j;
143368ec7858SStefano Zampini       PetscBool found = PETSC_FALSE;
143468ec7858SStefano Zampini 
143568ec7858SStefano Zampini       for (j = ii[i]; j < ii[i+1] && !found; j++)
143668ec7858SStefano Zampini         found = (jj[j] == i) ? PETSC_TRUE : PETSC_FALSE;
143768ec7858SStefano Zampini 
143868ec7858SStefano Zampini       if (!found) {
143968ec7858SStefano Zampini         PetscInfo1(A,"Matrix is missing local diagonal entry %D\n",i);
144068ec7858SStefano Zampini         if (missing) *missing = PETSC_TRUE;
144168ec7858SStefano Zampini         if (dd) *dd = i+rst;
144268ec7858SStefano Zampini         PetscFunctionReturn(0);
144368ec7858SStefano Zampini       }
144468ec7858SStefano Zampini     }
144568ec7858SStefano Zampini     if (!size) {
144668ec7858SStefano Zampini       PetscInfo(A,"Matrix has no diagonal entries therefore is missing diagonal\n");
144768ec7858SStefano Zampini       if (missing) *missing = PETSC_TRUE;
144868ec7858SStefano Zampini       if (dd) *dd = rst;
144968ec7858SStefano Zampini     }
145068ec7858SStefano Zampini   } else {
145168ec7858SStefano Zampini     PetscInfo(A,"Matrix has no diagonal entries therefore is missing diagonal\n");
145268ec7858SStefano Zampini     if (missing) *missing = PETSC_TRUE;
145368ec7858SStefano Zampini     if (dd) *dd = rst;
145468ec7858SStefano Zampini   }
145568ec7858SStefano Zampini   PetscFunctionReturn(0);
145668ec7858SStefano Zampini }
145768ec7858SStefano Zampini 
145868ec7858SStefano Zampini static PetscErrorCode MatScale_HYPRE(Mat A, PetscScalar s)
145968ec7858SStefano Zampini {
146068ec7858SStefano Zampini   hypre_ParCSRMatrix *parcsr;
146168ec7858SStefano Zampini   hypre_CSRMatrix    *ha;
146268ec7858SStefano Zampini   PetscErrorCode     ierr;
146368ec7858SStefano Zampini 
146468ec7858SStefano Zampini   PetscFunctionBegin;
146568ec7858SStefano Zampini   ierr  = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
146668ec7858SStefano Zampini   /* diagonal part */
146768ec7858SStefano Zampini   ha = hypre_ParCSRMatrixDiag(parcsr);
146868ec7858SStefano Zampini   if (ha) {
146968299464SStefano Zampini     PetscInt    size,i;
147068299464SStefano Zampini     HYPRE_Int   *ii;
147168ec7858SStefano Zampini     PetscScalar *a;
147268ec7858SStefano Zampini 
147368ec7858SStefano Zampini     size = hypre_CSRMatrixNumRows(ha);
147468ec7858SStefano Zampini     a    = hypre_CSRMatrixData(ha);
147568ec7858SStefano Zampini     ii   = hypre_CSRMatrixI(ha);
147668ec7858SStefano Zampini     for (i = 0; i < ii[size]; i++) a[i] *= s;
147768ec7858SStefano Zampini   }
147868ec7858SStefano Zampini   /* offdiagonal part */
147968ec7858SStefano Zampini   ha = hypre_ParCSRMatrixOffd(parcsr);
148068ec7858SStefano Zampini   if (ha) {
148168299464SStefano Zampini     PetscInt    size,i;
148268299464SStefano Zampini     HYPRE_Int   *ii;
148368ec7858SStefano Zampini     PetscScalar *a;
148468ec7858SStefano Zampini 
148568ec7858SStefano Zampini     size = hypre_CSRMatrixNumRows(ha);
148668ec7858SStefano Zampini     a    = hypre_CSRMatrixData(ha);
148768ec7858SStefano Zampini     ii   = hypre_CSRMatrixI(ha);
148868ec7858SStefano Zampini     for (i = 0; i < ii[size]; i++) a[i] *= s;
148968ec7858SStefano Zampini   }
149068ec7858SStefano Zampini   PetscFunctionReturn(0);
149168ec7858SStefano Zampini }
149268ec7858SStefano Zampini 
149368ec7858SStefano Zampini static PetscErrorCode MatZeroRowsColumns_HYPRE(Mat A, PetscInt numRows, const PetscInt rows[], PetscScalar diag, Vec x, Vec b)
149468ec7858SStefano Zampini {
149568ec7858SStefano Zampini   hypre_ParCSRMatrix *parcsr;
149668299464SStefano Zampini   HYPRE_Int          *lrows;
149768299464SStefano Zampini   PetscInt           rst,ren,i;
149868ec7858SStefano Zampini   PetscErrorCode     ierr;
149968ec7858SStefano Zampini 
150068ec7858SStefano Zampini   PetscFunctionBegin;
150168ec7858SStefano Zampini   if (x || b) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"To be implemented");
150268ec7858SStefano Zampini   ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
150368ec7858SStefano Zampini   ierr = PetscMalloc1(numRows,&lrows);CHKERRQ(ierr);
150468ec7858SStefano Zampini   ierr = MatGetOwnershipRange(A,&rst,&ren);CHKERRQ(ierr);
150568ec7858SStefano Zampini   for (i=0;i<numRows;i++) {
150668ec7858SStefano Zampini     if (rows[i] < rst || rows[i] >= ren)
150768ec7858SStefano Zampini       SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Non-local rows not yet supported");
150868ec7858SStefano Zampini     lrows[i] = rows[i] - rst;
150968ec7858SStefano Zampini   }
151068ec7858SStefano Zampini   PetscStackCallStandard(hypre_ParCSRMatrixEliminateRowsCols,(parcsr,numRows,lrows));
151168ec7858SStefano Zampini   ierr = PetscFree(lrows);CHKERRQ(ierr);
151268ec7858SStefano Zampini   PetscFunctionReturn(0);
151368ec7858SStefano Zampini }
151468ec7858SStefano Zampini 
1515c69f721fSFande Kong static PetscErrorCode MatZeroEntries_HYPRE_CSRMatrix(hypre_CSRMatrix *ha)
1516c69f721fSFande Kong {
1517c69f721fSFande Kong   PetscErrorCode      ierr;
1518c69f721fSFande Kong 
1519c69f721fSFande Kong   PetscFunctionBegin;
1520c69f721fSFande Kong   if (ha) {
1521c69f721fSFande Kong     HYPRE_Int     *ii, size;
1522c69f721fSFande Kong     HYPRE_Complex *a;
1523c69f721fSFande Kong 
1524c69f721fSFande Kong     size = hypre_CSRMatrixNumRows(ha);
1525c69f721fSFande Kong     a    = hypre_CSRMatrixData(ha);
1526c69f721fSFande Kong     ii   = hypre_CSRMatrixI(ha);
1527c69f721fSFande Kong 
1528c69f721fSFande Kong     if (a) { ierr = PetscMemzero(a,(ii[size])*sizeof(HYPRE_Complex));CHKERRQ(ierr); }
1529c69f721fSFande Kong   }
1530c69f721fSFande Kong   PetscFunctionReturn(0);
1531c69f721fSFande Kong }
1532c69f721fSFande Kong 
1533c69f721fSFande Kong PetscErrorCode MatZeroEntries_HYPRE(Mat A)
1534c69f721fSFande Kong {
1535c69f721fSFande Kong   hypre_ParCSRMatrix  *parcsr;
1536c69f721fSFande Kong   PetscErrorCode      ierr;
1537c69f721fSFande Kong 
1538c69f721fSFande Kong   PetscFunctionBegin;
1539c69f721fSFande Kong   ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
1540c69f721fSFande Kong   /* diagonal part */
1541c69f721fSFande Kong   ierr = MatZeroEntries_HYPRE_CSRMatrix(hypre_ParCSRMatrixDiag(parcsr));CHKERRQ(ierr);
1542c69f721fSFande Kong   /* off-diagonal part */
1543c69f721fSFande Kong   ierr = MatZeroEntries_HYPRE_CSRMatrix(hypre_ParCSRMatrixOffd(parcsr));CHKERRQ(ierr);
1544c69f721fSFande Kong   PetscFunctionReturn(0);
1545c69f721fSFande Kong }
1546c69f721fSFande Kong 
1547c69f721fSFande Kong static PetscErrorCode MatZeroRows_HYPRE_CSRMatrix(hypre_CSRMatrix *hA,PetscInt N,const PetscInt rows[],PetscScalar diag)
1548c69f721fSFande Kong {
1549c69f721fSFande Kong   PetscInt        ii, jj, ibeg, iend, irow;
1550c69f721fSFande Kong   PetscInt        *i, *j;
1551c69f721fSFande Kong   PetscScalar     *a;
1552c69f721fSFande Kong 
1553c69f721fSFande Kong   PetscFunctionBegin;
1554c69f721fSFande Kong   if (!hA) PetscFunctionReturn(0);
1555c69f721fSFande Kong 
155608defe43SFande Kong   i = (PetscInt*) hypre_CSRMatrixI(hA);
155708defe43SFande Kong   j = (PetscInt*) hypre_CSRMatrixJ(hA);
1558c69f721fSFande Kong   a = hypre_CSRMatrixData(hA);
1559c69f721fSFande Kong 
1560c69f721fSFande Kong   for (ii = 0; ii < N; ii++) {
1561c69f721fSFande Kong     irow = rows[ii];
1562c69f721fSFande Kong     ibeg = i[irow];
1563c69f721fSFande Kong     iend = i[irow+1];
1564c69f721fSFande Kong     for (jj = ibeg; jj < iend; jj++)
1565c69f721fSFande Kong       if (j[jj] == irow) a[jj] = diag;
1566c69f721fSFande Kong       else a[jj] = 0.0;
1567c69f721fSFande Kong    }
1568c69f721fSFande Kong    PetscFunctionReturn(0);
1569c69f721fSFande Kong }
1570c69f721fSFande Kong 
1571ddbeb582SStefano Zampini static PetscErrorCode MatZeroRows_HYPRE(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag,Vec x,Vec b)
1572c69f721fSFande Kong {
1573c69f721fSFande Kong   hypre_ParCSRMatrix  *parcsr;
1574c69f721fSFande Kong   PetscInt            *lrows,len;
1575c69f721fSFande Kong   PetscErrorCode      ierr;
1576c69f721fSFande Kong 
1577c69f721fSFande Kong   PetscFunctionBegin;
1578c69f721fSFande 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");
1579c69f721fSFande Kong   /* retrieve the internal matrix */
1580c69f721fSFande Kong   ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
1581c69f721fSFande Kong   /* get locally owned rows */
1582c69f721fSFande Kong   ierr = MatZeroRowsMapLocal_Private(A,N,rows,&len,&lrows);CHKERRQ(ierr);
1583c69f721fSFande Kong   /* zero diagonal part */
1584c69f721fSFande Kong   ierr = MatZeroRows_HYPRE_CSRMatrix(hypre_ParCSRMatrixDiag(parcsr),len,lrows,diag);CHKERRQ(ierr);
1585c69f721fSFande Kong   /* zero off-diagonal part */
1586c69f721fSFande Kong   ierr = MatZeroRows_HYPRE_CSRMatrix(hypre_ParCSRMatrixOffd(parcsr),len,lrows,0.0);CHKERRQ(ierr);
1587c69f721fSFande Kong 
1588c69f721fSFande Kong   ierr = PetscFree(lrows);CHKERRQ(ierr);
1589c69f721fSFande Kong   PetscFunctionReturn(0);
1590c69f721fSFande Kong }
1591c69f721fSFande Kong 
1592ddbeb582SStefano Zampini static PetscErrorCode MatAssemblyBegin_HYPRE(Mat mat,MatAssemblyType mode)
1593c69f721fSFande Kong {
1594c69f721fSFande Kong   PetscErrorCode ierr;
1595c69f721fSFande Kong 
1596c69f721fSFande Kong   PetscFunctionBegin;
1597c69f721fSFande Kong   if (mat->nooffprocentries) PetscFunctionReturn(0);
1598c69f721fSFande Kong 
1599c69f721fSFande Kong   ierr = MatStashScatterBegin_Private(mat,&mat->stash,mat->rmap->range);CHKERRQ(ierr);
1600c69f721fSFande Kong   PetscFunctionReturn(0);
1601c69f721fSFande Kong }
1602c69f721fSFande Kong 
1603ddbeb582SStefano Zampini static PetscErrorCode MatGetRow_HYPRE(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
1604c69f721fSFande Kong {
1605c69f721fSFande Kong   hypre_ParCSRMatrix  *parcsr;
1606c69f721fSFande Kong   PetscErrorCode      ierr;
1607c69f721fSFande Kong 
1608c69f721fSFande Kong   PetscFunctionBegin;
1609c69f721fSFande Kong   /* retrieve the internal matrix */
1610c69f721fSFande Kong   ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
1611c69f721fSFande Kong   /* call HYPRE API */
161208defe43SFande Kong   PetscStackCallStandard(HYPRE_ParCSRMatrixGetRow,(parcsr,row,(HYPRE_Int*)nz,(HYPRE_Int**)idx,v));
1613c69f721fSFande Kong   PetscFunctionReturn(0);
1614c69f721fSFande Kong }
1615c69f721fSFande Kong 
1616ddbeb582SStefano Zampini static PetscErrorCode MatRestoreRow_HYPRE(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
1617c69f721fSFande Kong {
1618c69f721fSFande Kong   hypre_ParCSRMatrix  *parcsr;
1619c69f721fSFande Kong   PetscErrorCode      ierr;
1620c69f721fSFande Kong 
1621c69f721fSFande Kong   PetscFunctionBegin;
1622c69f721fSFande Kong   /* retrieve the internal matrix */
1623c69f721fSFande Kong   ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
1624c69f721fSFande Kong   /* call HYPRE API */
162508defe43SFande Kong   PetscStackCallStandard(HYPRE_ParCSRMatrixRestoreRow,(parcsr,row,(HYPRE_Int*)nz,(HYPRE_Int**)idx,v));
1626c69f721fSFande Kong   PetscFunctionReturn(0);
1627c69f721fSFande Kong }
1628c69f721fSFande Kong 
1629ddbeb582SStefano Zampini static PetscErrorCode MatGetValues_HYPRE(Mat A,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],PetscScalar v[])
1630c69f721fSFande Kong {
163145b8d346SStefano Zampini   Mat_HYPRE *hA = (Mat_HYPRE*)A->data;
1632c69f721fSFande Kong   PetscInt  i;
16331d4906efSStefano Zampini 
1634c69f721fSFande Kong   PetscFunctionBegin;
1635c69f721fSFande Kong   if (!m || !n) PetscFunctionReturn(0);
1636c69f721fSFande Kong   /* Ignore negative row indices
1637c69f721fSFande Kong    * And negative column indices should be automatically ignored in hypre
1638c69f721fSFande Kong    * */
1639c69f721fSFande Kong   for (i=0; i<m; i++)
164045b8d346SStefano Zampini     if (idxm[i] >= 0) PetscStackCallStandard(HYPRE_IJMatrixGetValues,(hA->ij,1,(HYPRE_Int*)&n,(HYPRE_Int*)&idxm[i],(HYPRE_Int*)idxn,&v[i*n]));
1641c69f721fSFande Kong   PetscFunctionReturn(0);
1642c69f721fSFande Kong }
1643c69f721fSFande Kong 
1644ddbeb582SStefano Zampini static PetscErrorCode MatSetOption_HYPRE(Mat A,MatOption op,PetscBool flg)
1645ddbeb582SStefano Zampini {
1646ddbeb582SStefano Zampini   Mat_HYPRE *hA = (Mat_HYPRE*)A->data;
1647ddbeb582SStefano Zampini 
1648ddbeb582SStefano Zampini   PetscFunctionBegin;
1649ddbeb582SStefano Zampini   switch (op)
1650ddbeb582SStefano Zampini   {
1651ddbeb582SStefano Zampini   case MAT_NO_OFF_PROC_ENTRIES:
1652ddbeb582SStefano Zampini     if (flg) {
1653ddbeb582SStefano Zampini       PetscStackCallStandard(HYPRE_IJMatrixSetMaxOffProcElmts,(hA->ij,0));
1654ddbeb582SStefano Zampini     }
1655ddbeb582SStefano Zampini     break;
1656ddbeb582SStefano Zampini   default:
1657ddbeb582SStefano Zampini     break;
1658ddbeb582SStefano Zampini   }
1659ddbeb582SStefano Zampini   PetscFunctionReturn(0);
1660ddbeb582SStefano Zampini }
1661c69f721fSFande Kong 
166245b8d346SStefano Zampini static PetscErrorCode MatView_HYPRE(Mat A, PetscViewer view)
166345b8d346SStefano Zampini {
166445b8d346SStefano Zampini   hypre_ParCSRMatrix *parcsr;
166545b8d346SStefano Zampini   PetscErrorCode     ierr;
166645b8d346SStefano Zampini   Mat                B;
166745b8d346SStefano Zampini   PetscViewerFormat  format;
166845b8d346SStefano Zampini   PetscErrorCode     (*mview)(Mat,PetscViewer) = NULL;
166945b8d346SStefano Zampini 
167045b8d346SStefano Zampini   PetscFunctionBegin;
167145b8d346SStefano Zampini   ierr = PetscViewerGetFormat(view,&format);CHKERRQ(ierr);
167245b8d346SStefano Zampini   if (format != PETSC_VIEWER_NATIVE) {
167345b8d346SStefano Zampini     ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
167445b8d346SStefano Zampini     ierr = MatCreateFromParCSR(parcsr,MATAIJ,PETSC_USE_POINTER,&B);CHKERRQ(ierr);
167545b8d346SStefano Zampini     ierr = MatGetOperation(B,MATOP_VIEW,(void(**)(void))&mview);CHKERRQ(ierr);
167645b8d346SStefano Zampini     if (!mview) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Missing view operation");CHKERRQ(ierr);
167745b8d346SStefano Zampini     ierr = (*mview)(B,view);CHKERRQ(ierr);
167845b8d346SStefano Zampini     ierr = MatDestroy(&B);CHKERRQ(ierr);
167945b8d346SStefano Zampini   } else {
168045b8d346SStefano Zampini     Mat_HYPRE  *hA = (Mat_HYPRE*)A->data;
168145b8d346SStefano Zampini     PetscMPIInt size;
168245b8d346SStefano Zampini     PetscBool   isascii;
168345b8d346SStefano Zampini     const char *filename;
168445b8d346SStefano Zampini 
168545b8d346SStefano Zampini     /* HYPRE uses only text files */
168645b8d346SStefano Zampini     ierr = PetscObjectTypeCompare((PetscObject)view,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr);
168745b8d346SStefano Zampini     if (!isascii) SETERRQ1(PetscObjectComm((PetscObject)view),PETSC_ERR_SUP,"PetscViewerType %s: native HYPRE format needs PETSCVIEWERASCII",((PetscObject)view)->type_name);
168845b8d346SStefano Zampini     ierr = PetscViewerFileGetName(view,&filename);CHKERRQ(ierr);
168945b8d346SStefano Zampini     PetscStackCallStandard(HYPRE_IJMatrixPrint,(hA->ij,filename));
169045b8d346SStefano Zampini     ierr = MPI_Comm_size(hA->comm,&size);CHKERRQ(ierr);
169145b8d346SStefano Zampini     if (size > 1) {
169245b8d346SStefano Zampini       ierr = PetscViewerASCIIPrintf(view,"Matrix files: %s.%05d ... %s.%05d\n",filename,0,filename,size-1);CHKERRQ(ierr);
169345b8d346SStefano Zampini     } else {
169445b8d346SStefano Zampini       ierr = PetscViewerASCIIPrintf(view,"Matrix file: %s.%05d\n",filename,0);CHKERRQ(ierr);
169545b8d346SStefano Zampini     }
169645b8d346SStefano Zampini   }
169745b8d346SStefano Zampini   PetscFunctionReturn(0);
169845b8d346SStefano Zampini }
169945b8d346SStefano Zampini 
170045b8d346SStefano Zampini static PetscErrorCode MatDuplicate_HYPRE(Mat A,MatDuplicateOption op, Mat *B)
170145b8d346SStefano Zampini {
170245b8d346SStefano Zampini   hypre_ParCSRMatrix *parcsr;
170345b8d346SStefano Zampini   PetscErrorCode     ierr;
170445b8d346SStefano Zampini   PetscCopyMode      cpmode;
170545b8d346SStefano Zampini 
170645b8d346SStefano Zampini   PetscFunctionBegin;
170745b8d346SStefano Zampini   ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr);
170845b8d346SStefano Zampini   if (op == MAT_DO_NOT_COPY_VALUES || op == MAT_SHARE_NONZERO_PATTERN) {
170945b8d346SStefano Zampini     parcsr = hypre_ParCSRMatrixCompleteClone(parcsr);
171045b8d346SStefano Zampini     cpmode = PETSC_OWN_POINTER;
171145b8d346SStefano Zampini   } else {
171245b8d346SStefano Zampini     cpmode = PETSC_COPY_VALUES;
171345b8d346SStefano Zampini   }
171445b8d346SStefano Zampini   ierr = MatCreateFromParCSR(parcsr,MATHYPRE,cpmode,B);CHKERRQ(ierr);
171545b8d346SStefano Zampini   PetscFunctionReturn(0);
171645b8d346SStefano Zampini }
171745b8d346SStefano Zampini 
1718a055b5aaSBarry Smith /*MC
1719a055b5aaSBarry Smith    MATHYPRE - MATHYPRE = "hypre" - A matrix type to be used for sequential and parallel sparse matrices
1720a055b5aaSBarry Smith           based on the hypre IJ interface.
1721a055b5aaSBarry Smith 
1722a055b5aaSBarry Smith    Level: intermediate
1723a055b5aaSBarry Smith 
1724a055b5aaSBarry Smith .seealso: MatCreate()
1725a055b5aaSBarry Smith M*/
1726a055b5aaSBarry Smith 
172763c07aadSStefano Zampini PETSC_EXTERN PetscErrorCode MatCreate_HYPRE(Mat B)
172863c07aadSStefano Zampini {
172963c07aadSStefano Zampini   Mat_HYPRE      *hB;
173063c07aadSStefano Zampini   PetscErrorCode ierr;
173163c07aadSStefano Zampini 
173263c07aadSStefano Zampini   PetscFunctionBegin;
173363c07aadSStefano Zampini   ierr           = PetscNewLog(B,&hB);CHKERRQ(ierr);
1734978814f1SStefano Zampini   hB->inner_free = PETSC_TRUE;
1735c69f721fSFande Kong   hB->available  = PETSC_TRUE;
1736c69f721fSFande Kong   hB->size       = 0;
1737c69f721fSFande Kong   hB->array      = NULL;
1738978814f1SStefano Zampini 
173963c07aadSStefano Zampini   B->data       = (void*)hB;
174063c07aadSStefano Zampini   B->rmap->bs   = 1;
174163c07aadSStefano Zampini   B->assembled  = PETSC_FALSE;
174263c07aadSStefano Zampini 
1743de393100SStefano Zampini   ierr = PetscMemzero(B->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
174463c07aadSStefano Zampini   B->ops->mult            = MatMult_HYPRE;
174563c07aadSStefano Zampini   B->ops->multtranspose   = MatMultTranspose_HYPRE;
174663c07aadSStefano Zampini   B->ops->setup           = MatSetUp_HYPRE;
174763c07aadSStefano Zampini   B->ops->destroy         = MatDestroy_HYPRE;
174863c07aadSStefano Zampini   B->ops->assemblyend     = MatAssemblyEnd_HYPRE;
1749c69f721fSFande Kong   B->ops->assemblybegin   = MatAssemblyBegin_HYPRE;
1750cd8bc7baSStefano Zampini   B->ops->ptap            = MatPtAP_HYPRE_HYPRE;
1751d501dc42Sstefano_zampini   B->ops->matmult         = MatMatMult_HYPRE_HYPRE;
1752d975228cSstefano_zampini   B->ops->setvalues       = MatSetValues_HYPRE;
175368ec7858SStefano Zampini   B->ops->missingdiagonal = MatMissingDiagonal_HYPRE;
175468ec7858SStefano Zampini   B->ops->scale           = MatScale_HYPRE;
175568ec7858SStefano Zampini   B->ops->zerorowscolumns = MatZeroRowsColumns_HYPRE;
1756c69f721fSFande Kong   B->ops->zeroentries     = MatZeroEntries_HYPRE;
1757c69f721fSFande Kong   B->ops->zerorows        = MatZeroRows_HYPRE;
1758c69f721fSFande Kong   B->ops->getrow          = MatGetRow_HYPRE;
1759c69f721fSFande Kong   B->ops->restorerow      = MatRestoreRow_HYPRE;
1760c69f721fSFande Kong   B->ops->getvalues       = MatGetValues_HYPRE;
1761ddbeb582SStefano Zampini   B->ops->setoption       = MatSetOption_HYPRE;
176245b8d346SStefano Zampini   B->ops->duplicate       = MatDuplicate_HYPRE;
176345b8d346SStefano Zampini   B->ops->view            = MatView_HYPRE;
176445b8d346SStefano Zampini 
176545b8d346SStefano Zampini   /* build cache for off array entries formed */
176645b8d346SStefano Zampini   ierr = MatStashCreate_Private(PetscObjectComm((PetscObject)B),1,&B->stash);CHKERRQ(ierr);
176763c07aadSStefano Zampini 
176863c07aadSStefano Zampini   ierr = MPI_Comm_dup(PetscObjectComm((PetscObject)B),&hB->comm);CHKERRQ(ierr);
176963c07aadSStefano Zampini   ierr = PetscObjectChangeTypeName((PetscObject)B,MATHYPRE);CHKERRQ(ierr);
177063c07aadSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_hypre_aij_C",MatConvert_HYPRE_AIJ);CHKERRQ(ierr);
17712df22349SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_hypre_is_C",MatConvert_HYPRE_IS);CHKERRQ(ierr);
1772c1a070e6SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)B,"MatPtAP_seqaij_hypre_C",MatPtAP_AIJ_HYPRE);CHKERRQ(ierr);
1773c1a070e6SStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)B,"MatPtAP_mpiaij_hypre_C",MatPtAP_AIJ_HYPRE);CHKERRQ(ierr);
1774d975228cSstefano_zampini   ierr = PetscObjectComposeFunction((PetscObject)B,"MatHYPRESetPreallocation_C",MatHYPRESetPreallocation_HYPRE);CHKERRQ(ierr);
1775dd9c0a25Sstefano_zampini   ierr = PetscObjectComposeFunction((PetscObject)B,"MatHYPREGetParCSR_C",MatHYPREGetParCSR_HYPRE);CHKERRQ(ierr);
177675d48cdbSStefano Zampini   ierr = PetscObjectComposeFunction((PetscObject)B,"MatPtAP_is_hypre_C",MatPtAP_IS_XAIJ);CHKERRQ(ierr);
177763c07aadSStefano Zampini   PetscFunctionReturn(0);
177863c07aadSStefano Zampini }
177963c07aadSStefano Zampini 
1780225daaf8SStefano Zampini static PetscErrorCode hypre_array_destroy(void *ptr)
1781225daaf8SStefano Zampini {
1782225daaf8SStefano Zampini    PetscFunctionBegin;
1783e6de0934SSatish Balay    hypre_TFree(ptr,HYPRE_MEMORY_HOST);
1784225daaf8SStefano Zampini    PetscFunctionReturn(0);
1785225daaf8SStefano Zampini }
1786