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)); 101363c07aadSStefano Zampini if (hA->x) PetscFunctionReturn(0); 101463c07aadSStefano Zampini ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 101563c07aadSStefano Zampini ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 101663c07aadSStefano Zampini ierr = VecCreateMPIWithArray(PetscObjectComm((PetscObject)A),1,A->cmap->n,A->cmap->N,NULL,&x);CHKERRQ(ierr); 101763c07aadSStefano Zampini ierr = VecCreateMPIWithArray(PetscObjectComm((PetscObject)A),1,A->rmap->n,A->rmap->N,NULL,&b);CHKERRQ(ierr); 101863c07aadSStefano Zampini ierr = VecHYPRE_IJVectorCreate(x,&hA->x);CHKERRQ(ierr); 101963c07aadSStefano Zampini ierr = VecHYPRE_IJVectorCreate(b,&hA->b);CHKERRQ(ierr); 102063c07aadSStefano Zampini ierr = VecDestroy(&x);CHKERRQ(ierr); 102163c07aadSStefano Zampini ierr = VecDestroy(&b);CHKERRQ(ierr); 102263c07aadSStefano Zampini PetscFunctionReturn(0); 102363c07aadSStefano Zampini } 102463c07aadSStefano Zampini 1025c69f721fSFande Kong static PetscErrorCode MatGetArray_HYPRE(Mat A, PetscInt size, void **array) 1026c69f721fSFande Kong { 1027c69f721fSFande Kong Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 1028c69f721fSFande Kong PetscErrorCode ierr; 1029c69f721fSFande Kong 1030c69f721fSFande Kong PetscFunctionBegin; 1031c69f721fSFande Kong if (!hA->available) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_NULL,"Temporary space is in use"); 1032c69f721fSFande Kong 1033c69f721fSFande Kong if (hA->size >= size) *array = hA->array; 1034c69f721fSFande Kong else { 1035c69f721fSFande Kong ierr = PetscFree(hA->array);CHKERRQ(ierr); 1036c69f721fSFande Kong hA->size = size; 1037c69f721fSFande Kong ierr = PetscMalloc(hA->size,&hA->array);CHKERRQ(ierr); 1038c69f721fSFande Kong *array = hA->array; 1039c69f721fSFande Kong } 1040c69f721fSFande Kong 1041c69f721fSFande Kong hA->available = PETSC_FALSE; 1042c69f721fSFande Kong PetscFunctionReturn(0); 1043c69f721fSFande Kong } 1044c69f721fSFande Kong 1045708542d2SFande Kong static PetscErrorCode MatRestoreArray_HYPRE(Mat A, void **array) 1046c69f721fSFande Kong { 1047c69f721fSFande Kong Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 1048c69f721fSFande Kong 1049c69f721fSFande Kong PetscFunctionBegin; 1050c69f721fSFande Kong *array = NULL; 1051c69f721fSFande Kong hA->available = PETSC_TRUE; 1052c69f721fSFande Kong PetscFunctionReturn(0); 1053c69f721fSFande Kong } 1054c69f721fSFande Kong 1055d975228cSstefano_zampini 1056d975228cSstefano_zampini PetscErrorCode MatSetValues_HYPRE(Mat A, PetscInt nr, const PetscInt rows[], PetscInt nc, const PetscInt cols[], const PetscScalar v[], InsertMode ins) 1057d975228cSstefano_zampini { 1058d975228cSstefano_zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 1059d975228cSstefano_zampini PetscScalar *vals = (PetscScalar *)v; 1060c69f721fSFande Kong PetscScalar *sscr; 1061c69f721fSFande Kong PetscInt *cscr[2]; 1062c69f721fSFande Kong PetscInt i,nzc; 106308defe43SFande Kong void *array = NULL; 1064d975228cSstefano_zampini PetscErrorCode ierr; 1065d975228cSstefano_zampini 1066d975228cSstefano_zampini PetscFunctionBegin; 1067c69f721fSFande Kong ierr = MatGetArray_HYPRE(A,sizeof(PetscInt)*(2*nc)+sizeof(PetscScalar)*nc*nr,&array);CHKERRQ(ierr); 1068c69f721fSFande Kong cscr[0] = (PetscInt*)array; 1069c69f721fSFande Kong cscr[1] = ((PetscInt*)array)+nc; 1070c69f721fSFande Kong sscr = (PetscScalar*)(((PetscInt*)array)+nc*2); 1071d975228cSstefano_zampini for (i=0,nzc=0;i<nc;i++) { 1072d975228cSstefano_zampini if (cols[i] >= 0) { 1073d975228cSstefano_zampini cscr[0][nzc ] = cols[i]; 1074d975228cSstefano_zampini cscr[1][nzc++] = i; 1075d975228cSstefano_zampini } 1076d975228cSstefano_zampini } 1077c69f721fSFande Kong if (!nzc) { 1078708542d2SFande Kong ierr = MatRestoreArray_HYPRE(A,&array);CHKERRQ(ierr); 1079c69f721fSFande Kong PetscFunctionReturn(0); 1080c69f721fSFande Kong } 1081d975228cSstefano_zampini 1082d975228cSstefano_zampini if (ins == ADD_VALUES) { 1083d975228cSstefano_zampini for (i=0;i<nr;i++) { 1084e3977e59Sstefano_zampini if (rows[i] >= 0 && nzc) { 1085d975228cSstefano_zampini PetscInt j; 1086d975228cSstefano_zampini for (j=0;j<nzc;j++) sscr[j] = vals[cscr[1][j]]; 108708defe43SFande Kong PetscStackCallStandard(HYPRE_IJMatrixAddToValues,(hA->ij,1,(HYPRE_Int*)&nzc,(HYPRE_Int*)(rows+i),(HYPRE_Int*)cscr[0],sscr)); 1088d975228cSstefano_zampini } 1089d975228cSstefano_zampini vals += nc; 1090d975228cSstefano_zampini } 1091d975228cSstefano_zampini } else { /* INSERT_VALUES */ 1092c69f721fSFande Kong 1093d975228cSstefano_zampini PetscInt rst,ren; 1094d975228cSstefano_zampini ierr = MatGetOwnershipRange(A,&rst,&ren);CHKERRQ(ierr); 1095c69f721fSFande Kong 1096d975228cSstefano_zampini for (i=0;i<nr;i++) { 1097e3977e59Sstefano_zampini if (rows[i] >= 0 && nzc) { 1098d975228cSstefano_zampini PetscInt j; 1099d975228cSstefano_zampini for (j=0;j<nzc;j++) sscr[j] = vals[cscr[1][j]]; 1100c69f721fSFande Kong /* nonlocal values */ 1101c69f721fSFande Kong if (rows[i] < rst || rows[i] >= ren) { ierr = MatStashValuesRow_Private(&A->stash,rows[i],nzc,cscr[0],sscr, PETSC_FALSE);CHKERRQ(ierr); } 1102c69f721fSFande Kong /* local values */ 110308defe43SFande Kong else PetscStackCallStandard(HYPRE_IJMatrixSetValues,(hA->ij,1,(HYPRE_Int*)&nzc,(HYPRE_Int*)(rows+i),(HYPRE_Int*)cscr[0],sscr)); 1104d975228cSstefano_zampini } 1105d975228cSstefano_zampini vals += nc; 1106d975228cSstefano_zampini } 1107d975228cSstefano_zampini } 1108c69f721fSFande Kong 1109708542d2SFande Kong ierr = MatRestoreArray_HYPRE(A,&array);CHKERRQ(ierr); 1110d975228cSstefano_zampini PetscFunctionReturn(0); 1111d975228cSstefano_zampini } 1112d975228cSstefano_zampini 1113d975228cSstefano_zampini static PetscErrorCode MatHYPRESetPreallocation_HYPRE(Mat A, PetscInt dnz, const PetscInt dnnz[], PetscInt onz, const PetscInt onnz[]) 1114d975228cSstefano_zampini { 1115d975228cSstefano_zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 11167d968826Sstefano_zampini HYPRE_Int *hdnnz,*honnz; 111706a29025Sstefano_zampini PetscInt i,rs,re,cs,ce,bs; 1118d975228cSstefano_zampini PetscMPIInt size; 1119d975228cSstefano_zampini PetscErrorCode ierr; 1120d975228cSstefano_zampini 1121d975228cSstefano_zampini PetscFunctionBegin; 112206a29025Sstefano_zampini ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr); 1123d975228cSstefano_zampini ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 1124d975228cSstefano_zampini ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 1125d975228cSstefano_zampini rs = A->rmap->rstart; 1126d975228cSstefano_zampini re = A->rmap->rend; 1127d975228cSstefano_zampini cs = A->cmap->rstart; 1128d975228cSstefano_zampini ce = A->cmap->rend; 1129d975228cSstefano_zampini if (!hA->ij) { 1130d975228cSstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixCreate,(hA->comm,rs,re-1,cs,ce-1,&hA->ij)); 1131d975228cSstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(hA->ij,HYPRE_PARCSR)); 1132d975228cSstefano_zampini } else { 11337d968826Sstefano_zampini HYPRE_Int hrs,hre,hcs,hce; 1134d975228cSstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetLocalRange,(hA->ij,&hrs,&hre,&hcs,&hce)); 1135d975228cSstefano_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); 1136d975228cSstefano_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); 1137d975228cSstefano_zampini } 113806a29025Sstefano_zampini if (dnz == PETSC_DEFAULT || dnz == PETSC_DECIDE) dnz = 10*bs; 113906a29025Sstefano_zampini if (onz == PETSC_DEFAULT || onz == PETSC_DECIDE) onz = 10*bs; 114006a29025Sstefano_zampini 1141d975228cSstefano_zampini if (!dnnz) { 1142d975228cSstefano_zampini ierr = PetscMalloc1(A->rmap->n,&hdnnz);CHKERRQ(ierr); 1143d975228cSstefano_zampini for (i=0;i<A->rmap->n;i++) hdnnz[i] = dnz; 1144d975228cSstefano_zampini } else { 11457d968826Sstefano_zampini hdnnz = (HYPRE_Int*)dnnz; 1146d975228cSstefano_zampini } 1147d975228cSstefano_zampini ierr = MPI_Comm_size(PetscObjectComm((PetscObject)A),&size);CHKERRQ(ierr); 1148d975228cSstefano_zampini if (size > 1) { 1149ddbeb582SStefano Zampini hypre_AuxParCSRMatrix *aux_matrix; 1150d975228cSstefano_zampini if (!onnz) { 1151d975228cSstefano_zampini ierr = PetscMalloc1(A->rmap->n,&honnz);CHKERRQ(ierr); 1152d975228cSstefano_zampini for (i=0;i<A->rmap->n;i++) honnz[i] = onz; 1153d975228cSstefano_zampini } else { 11547d968826Sstefano_zampini honnz = (HYPRE_Int*)onnz; 1155d975228cSstefano_zampini } 1156ddbeb582SStefano Zampini /* SetDiagOffdSizes sets hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 0, since it seems 1157ddbeb582SStefano Zampini they assume the user will input the entire row values, properly sorted 1158ddbeb582SStefano Zampini In PETSc, we don't make such an assumption, and we instead set this flag to 1 1159ddbeb582SStefano Zampini Also, to avoid possible memory leaks, we destroy and recreate the translator 1160ddbeb582SStefano Zampini This has to be done here, as HYPRE_IJMatrixInitialize will properly initialize 1161ddbeb582SStefano Zampini the IJ matrix for us */ 1162ddbeb582SStefano Zampini aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(hA->ij); 1163ddbeb582SStefano Zampini hypre_AuxParCSRMatrixDestroy(aux_matrix); 1164ddbeb582SStefano Zampini hypre_IJMatrixTranslator(hA->ij) = NULL; 1165d975228cSstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixSetDiagOffdSizes,(hA->ij,hdnnz,honnz)); 1166ddbeb582SStefano Zampini aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(hA->ij); 1167ddbeb582SStefano Zampini hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 1; 1168d975228cSstefano_zampini } else { 1169d975228cSstefano_zampini honnz = NULL; 1170d975228cSstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixSetRowSizes,(hA->ij,hdnnz)); 1171d975228cSstefano_zampini } 1172ddbeb582SStefano Zampini 1173ddbeb582SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixInitialize,(hA->ij)); 1174d975228cSstefano_zampini if (!dnnz) { 1175d975228cSstefano_zampini ierr = PetscFree(hdnnz);CHKERRQ(ierr); 1176d975228cSstefano_zampini } 1177d975228cSstefano_zampini if (!onnz && honnz) { 1178d975228cSstefano_zampini ierr = PetscFree(honnz);CHKERRQ(ierr); 1179d975228cSstefano_zampini } 118006a29025Sstefano_zampini A->preallocated = PETSC_TRUE; 1181d975228cSstefano_zampini PetscFunctionReturn(0); 1182d975228cSstefano_zampini } 1183d975228cSstefano_zampini 1184d975228cSstefano_zampini /*@C 1185d975228cSstefano_zampini MatHYPRESetPreallocation - Preallocates memory for a sparse parallel matrix in HYPRE IJ format 1186d975228cSstefano_zampini 1187d975228cSstefano_zampini Collective on Mat 1188d975228cSstefano_zampini 1189d975228cSstefano_zampini Input Parameters: 1190d975228cSstefano_zampini + A - the matrix 1191d975228cSstefano_zampini . dnz - number of nonzeros per row in DIAGONAL portion of local submatrix 1192d975228cSstefano_zampini (same value is used for all local rows) 1193d975228cSstefano_zampini . dnnz - array containing the number of nonzeros in the various rows of the 1194d975228cSstefano_zampini DIAGONAL portion of the local submatrix (possibly different for each row) 1195d975228cSstefano_zampini or NULL (PETSC_NULL_INTEGER in Fortran), if d_nz is used to specify the nonzero structure. 1196d975228cSstefano_zampini The size of this array is equal to the number of local rows, i.e 'm'. 1197d975228cSstefano_zampini For matrices that will be factored, you must leave room for (and set) 1198d975228cSstefano_zampini the diagonal entry even if it is zero. 1199d975228cSstefano_zampini . onz - number of nonzeros per row in the OFF-DIAGONAL portion of local 1200d975228cSstefano_zampini submatrix (same value is used for all local rows). 1201d975228cSstefano_zampini - onnz - array containing the number of nonzeros in the various rows of the 1202d975228cSstefano_zampini OFF-DIAGONAL portion of the local submatrix (possibly different for 1203d975228cSstefano_zampini each row) or NULL (PETSC_NULL_INTEGER in Fortran), if o_nz is used to specify the nonzero 1204d975228cSstefano_zampini structure. The size of this array is equal to the number 1205d975228cSstefano_zampini of local rows, i.e 'm'. 1206d975228cSstefano_zampini 120795452b02SPatrick Sanan Notes: 120895452b02SPatrick Sanan If the *nnz parameter is given then the *nz parameter is ignored; for sequential matrices, onz and onnz are ignored. 1209d975228cSstefano_zampini 1210d975228cSstefano_zampini Level: intermediate 1211d975228cSstefano_zampini 1212d975228cSstefano_zampini .keywords: matrix, aij, compressed row, sparse, parallel 1213d975228cSstefano_zampini 1214d975228cSstefano_zampini .seealso: MatCreate(), MatMPIAIJSetPreallocation, MATHYPRE 1215d975228cSstefano_zampini @*/ 1216d975228cSstefano_zampini PetscErrorCode MatHYPRESetPreallocation(Mat A, PetscInt dnz, const PetscInt dnnz[], PetscInt onz, const PetscInt onnz[]) 1217d975228cSstefano_zampini { 1218d975228cSstefano_zampini PetscErrorCode ierr; 1219d975228cSstefano_zampini 1220d975228cSstefano_zampini PetscFunctionBegin; 1221d975228cSstefano_zampini PetscValidHeaderSpecific(A,MAT_CLASSID,1); 1222d975228cSstefano_zampini PetscValidType(A,1); 1223d975228cSstefano_zampini ierr = PetscTryMethod(A,"MatHYPRESetPreallocation_C",(Mat,PetscInt,const PetscInt[],PetscInt,const PetscInt[]),(A,dnz,dnnz,onz,onnz));CHKERRQ(ierr); 1224d975228cSstefano_zampini PetscFunctionReturn(0); 1225d975228cSstefano_zampini } 1226d975228cSstefano_zampini 1227225daaf8SStefano Zampini /* 1228225daaf8SStefano Zampini MatCreateFromParCSR - Creates a matrix from a hypre_ParCSRMatrix 1229225daaf8SStefano Zampini 1230225daaf8SStefano Zampini Collective 1231225daaf8SStefano Zampini 1232225daaf8SStefano Zampini Input Parameters: 1233*45b8d346SStefano Zampini + parcsr - the pointer to the hypre_ParCSRMatrix 1234bb4689ddSStefano Zampini . mtype - matrix type to be created. Currently MATAIJ, MATIS and MATHYPRE are supported. 1235225daaf8SStefano Zampini - copymode - PETSc copying options 1236225daaf8SStefano Zampini 1237225daaf8SStefano Zampini Output Parameter: 1238225daaf8SStefano Zampini . A - the matrix 1239225daaf8SStefano Zampini 1240225daaf8SStefano Zampini Level: intermediate 1241225daaf8SStefano Zampini 1242225daaf8SStefano Zampini .seealso: MatHYPRE, PetscCopyMode 1243225daaf8SStefano Zampini */ 1244*45b8d346SStefano Zampini PETSC_EXTERN PetscErrorCode MatCreateFromParCSR(hypre_ParCSRMatrix *parcsr, MatType mtype, PetscCopyMode copymode, Mat* A) 1245978814f1SStefano Zampini { 1246225daaf8SStefano Zampini Mat T; 1247978814f1SStefano Zampini Mat_HYPRE *hA; 1248978814f1SStefano Zampini MPI_Comm comm; 1249978814f1SStefano Zampini PetscInt rstart,rend,cstart,cend,M,N; 1250bb4689ddSStefano Zampini PetscBool isseqaij,ismpiaij,isaij,ishyp,isis; 1251978814f1SStefano Zampini PetscErrorCode ierr; 1252978814f1SStefano Zampini 1253978814f1SStefano Zampini PetscFunctionBegin; 1254978814f1SStefano Zampini comm = hypre_ParCSRMatrixComm(parcsr); 1255225daaf8SStefano Zampini ierr = PetscStrcmp(mtype,MATSEQAIJ,&isseqaij);CHKERRQ(ierr); 1256225daaf8SStefano Zampini ierr = PetscStrcmp(mtype,MATMPIAIJ,&ismpiaij);CHKERRQ(ierr); 1257225daaf8SStefano Zampini ierr = PetscStrcmp(mtype,MATAIJ,&isaij);CHKERRQ(ierr); 1258225daaf8SStefano Zampini ierr = PetscStrcmp(mtype,MATHYPRE,&ishyp);CHKERRQ(ierr); 12598cfe8d00SStefano Zampini ierr = PetscStrcmp(mtype,MATIS,&isis);CHKERRQ(ierr); 1260225daaf8SStefano Zampini isaij = (PetscBool)(isseqaij || ismpiaij || isaij); 1261bb4689ddSStefano 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); 1262978814f1SStefano Zampini /* access ParCSRMatrix */ 1263978814f1SStefano Zampini rstart = hypre_ParCSRMatrixFirstRowIndex(parcsr); 1264978814f1SStefano Zampini rend = hypre_ParCSRMatrixLastRowIndex(parcsr); 1265978814f1SStefano Zampini cstart = hypre_ParCSRMatrixFirstColDiag(parcsr); 1266978814f1SStefano Zampini cend = hypre_ParCSRMatrixLastColDiag(parcsr); 1267978814f1SStefano Zampini M = hypre_ParCSRMatrixGlobalNumRows(parcsr); 1268978814f1SStefano Zampini N = hypre_ParCSRMatrixGlobalNumCols(parcsr); 1269978814f1SStefano Zampini 1270fa92c42cSstefano_zampini /* fix for empty local rows/columns */ 1271fa92c42cSstefano_zampini if (rend < rstart) rend = rstart; 1272fa92c42cSstefano_zampini if (cend < cstart) cend = cstart; 1273fa92c42cSstefano_zampini 1274978814f1SStefano Zampini /* create PETSc matrix with MatHYPRE */ 1275225daaf8SStefano Zampini ierr = MatCreate(comm,&T);CHKERRQ(ierr); 1276225daaf8SStefano Zampini ierr = MatSetSizes(T,rend-rstart+1,cend-cstart+1,M,N);CHKERRQ(ierr); 1277225daaf8SStefano Zampini ierr = MatSetType(T,MATHYPRE);CHKERRQ(ierr); 1278225daaf8SStefano Zampini hA = (Mat_HYPRE*)(T->data); 1279978814f1SStefano Zampini 1280978814f1SStefano Zampini /* create HYPRE_IJMatrix */ 1281978814f1SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixCreate,(hA->comm,rstart,rend-1,cstart,cend-1,&hA->ij)); 1282*45b8d346SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(hA->ij,HYPRE_PARCSR)); 1283*45b8d346SStefano Zampini 1284*45b8d346SStefano Zampini /* create new ParCSR object if needed */ 1285*45b8d346SStefano Zampini if (ishyp && copymode == PETSC_COPY_VALUES) { 1286*45b8d346SStefano Zampini hypre_ParCSRMatrix *new_parcsr; 1287*45b8d346SStefano Zampini hypre_CSRMatrix *hdiag,*hoffd,*ndiag,*noffd; 1288*45b8d346SStefano Zampini 1289*45b8d346SStefano Zampini new_parcsr = hypre_ParCSRMatrixCompleteClone(parcsr); 1290*45b8d346SStefano Zampini hdiag = hypre_ParCSRMatrixDiag(parcsr); 1291*45b8d346SStefano Zampini hoffd = hypre_ParCSRMatrixOffd(parcsr); 1292*45b8d346SStefano Zampini ndiag = hypre_ParCSRMatrixDiag(new_parcsr); 1293*45b8d346SStefano Zampini noffd = hypre_ParCSRMatrixOffd(new_parcsr); 1294*45b8d346SStefano Zampini ierr = PetscMemcpy(hypre_CSRMatrixData(ndiag),hypre_CSRMatrixData(hdiag),hypre_CSRMatrixNumNonzeros(hdiag)*sizeof(PetscScalar));CHKERRQ(ierr); 1295*45b8d346SStefano Zampini ierr = PetscMemcpy(hypre_CSRMatrixData(noffd),hypre_CSRMatrixData(hoffd),hypre_CSRMatrixNumNonzeros(hoffd)*sizeof(PetscScalar));CHKERRQ(ierr); 1296*45b8d346SStefano Zampini parcsr = new_parcsr; 1297*45b8d346SStefano Zampini copymode = PETSC_OWN_POINTER; 1298*45b8d346SStefano Zampini } 1299978814f1SStefano Zampini 1300978814f1SStefano Zampini /* set ParCSR object */ 1301978814f1SStefano Zampini hypre_IJMatrixObject(hA->ij) = parcsr; 13024ec6421dSstefano_zampini T->preallocated = PETSC_TRUE; 1303978814f1SStefano Zampini 1304978814f1SStefano Zampini /* set assembled flag */ 1305978814f1SStefano Zampini hypre_IJMatrixAssembleFlag(hA->ij) = 1; 1306978814f1SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixInitialize,(hA->ij)); 1307225daaf8SStefano Zampini if (ishyp) { 13086d2a658fSstefano_zampini PetscMPIInt myid = 0; 13096d2a658fSstefano_zampini 13106d2a658fSstefano_zampini /* make sure we always have row_starts and col_starts available */ 13116d2a658fSstefano_zampini if (HYPRE_AssumedPartitionCheck()) { 13126d2a658fSstefano_zampini ierr = MPI_Comm_rank(comm,&myid);CHKERRQ(ierr); 13136d2a658fSstefano_zampini } 13146d2a658fSstefano_zampini if (!hypre_ParCSRMatrixOwnsColStarts(parcsr)) { 13156d2a658fSstefano_zampini PetscLayout map; 13166d2a658fSstefano_zampini 13176d2a658fSstefano_zampini ierr = MatGetLayouts(T,NULL,&map);CHKERRQ(ierr); 13186d2a658fSstefano_zampini ierr = PetscLayoutSetUp(map);CHKERRQ(ierr); 13191c92d2d0Sstefano_zampini hypre_ParCSRMatrixColStarts(parcsr) = (HYPRE_Int*)(map->range + myid); 13206d2a658fSstefano_zampini } 13216d2a658fSstefano_zampini if (!hypre_ParCSRMatrixOwnsRowStarts(parcsr)) { 13226d2a658fSstefano_zampini PetscLayout map; 13236d2a658fSstefano_zampini 13246d2a658fSstefano_zampini ierr = MatGetLayouts(T,&map,NULL);CHKERRQ(ierr); 13256d2a658fSstefano_zampini ierr = PetscLayoutSetUp(map);CHKERRQ(ierr); 13261c92d2d0Sstefano_zampini hypre_ParCSRMatrixRowStarts(parcsr) = (HYPRE_Int*)(map->range + myid); 13276d2a658fSstefano_zampini } 1328978814f1SStefano Zampini /* prevent from freeing the pointer */ 1329978814f1SStefano Zampini if (copymode == PETSC_USE_POINTER) hA->inner_free = PETSC_FALSE; 1330225daaf8SStefano Zampini *A = T; 1331978814f1SStefano Zampini ierr = MatAssemblyBegin(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1332978814f1SStefano Zampini ierr = MatAssemblyEnd(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1333bb4689ddSStefano Zampini } else if (isaij) { 1334bb4689ddSStefano Zampini if (copymode != PETSC_OWN_POINTER) { 1335225daaf8SStefano Zampini /* prevent from freeing the pointer */ 1336225daaf8SStefano Zampini hA->inner_free = PETSC_FALSE; 1337225daaf8SStefano Zampini ierr = MatConvert_HYPRE_AIJ(T,MATAIJ,MAT_INITIAL_MATRIX,A);CHKERRQ(ierr); 1338225daaf8SStefano Zampini ierr = MatDestroy(&T);CHKERRQ(ierr); 1339225daaf8SStefano Zampini } else { /* AIJ return type with PETSC_OWN_POINTER */ 1340225daaf8SStefano Zampini ierr = MatConvert_HYPRE_AIJ(T,MATAIJ,MAT_INPLACE_MATRIX,&T);CHKERRQ(ierr); 1341225daaf8SStefano Zampini *A = T; 1342225daaf8SStefano Zampini } 1343bb4689ddSStefano Zampini } else if (isis) { 13448cfe8d00SStefano Zampini ierr = MatConvert_HYPRE_IS(T,MATIS,MAT_INITIAL_MATRIX,A);CHKERRQ(ierr); 13458cfe8d00SStefano Zampini if (copymode != PETSC_OWN_POINTER) hA->inner_free = PETSC_FALSE; 1346bb4689ddSStefano Zampini ierr = MatDestroy(&T);CHKERRQ(ierr); 1347bb4689ddSStefano Zampini } 1348978814f1SStefano Zampini PetscFunctionReturn(0); 1349978814f1SStefano Zampini } 1350978814f1SStefano Zampini 135168ec7858SStefano Zampini static PetscErrorCode MatHYPREGetParCSR_HYPRE(Mat A, hypre_ParCSRMatrix **parcsr) 1352dd9c0a25Sstefano_zampini { 1353dd9c0a25Sstefano_zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 1354dd9c0a25Sstefano_zampini HYPRE_Int type; 1355dd9c0a25Sstefano_zampini 1356dd9c0a25Sstefano_zampini PetscFunctionBegin; 1357dd9c0a25Sstefano_zampini if (!hA->ij) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"HYPRE_IJMatrix not present"); 1358dd9c0a25Sstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type)); 1359dd9c0a25Sstefano_zampini if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"HYPRE_IJMatrix is not of type HYPRE_PARCSR"); 1360dd9c0a25Sstefano_zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)parcsr)); 1361dd9c0a25Sstefano_zampini PetscFunctionReturn(0); 1362dd9c0a25Sstefano_zampini } 1363dd9c0a25Sstefano_zampini 1364dd9c0a25Sstefano_zampini /* 1365dd9c0a25Sstefano_zampini MatHYPREGetParCSR - Gets the pointer to the ParCSR matrix 1366dd9c0a25Sstefano_zampini 1367dd9c0a25Sstefano_zampini Not collective 1368dd9c0a25Sstefano_zampini 1369dd9c0a25Sstefano_zampini Input Parameters: 1370dd9c0a25Sstefano_zampini + A - the MATHYPRE object 1371dd9c0a25Sstefano_zampini 1372dd9c0a25Sstefano_zampini Output Parameter: 1373dd9c0a25Sstefano_zampini . parcsr - the pointer to the hypre_ParCSRMatrix 1374dd9c0a25Sstefano_zampini 1375dd9c0a25Sstefano_zampini Level: intermediate 1376dd9c0a25Sstefano_zampini 1377dd9c0a25Sstefano_zampini .seealso: MatHYPRE, PetscCopyMode 1378dd9c0a25Sstefano_zampini */ 1379dd9c0a25Sstefano_zampini PetscErrorCode MatHYPREGetParCSR(Mat A, hypre_ParCSRMatrix **parcsr) 1380dd9c0a25Sstefano_zampini { 1381dd9c0a25Sstefano_zampini PetscErrorCode ierr; 1382dd9c0a25Sstefano_zampini 1383dd9c0a25Sstefano_zampini PetscFunctionBegin; 1384dd9c0a25Sstefano_zampini PetscValidHeaderSpecific(A,MAT_CLASSID,1); 1385dd9c0a25Sstefano_zampini PetscValidType(A,1); 1386dd9c0a25Sstefano_zampini ierr = PetscUseMethod(A,"MatHYPREGetParCSR_C",(Mat,hypre_ParCSRMatrix**),(A,parcsr));CHKERRQ(ierr); 1387dd9c0a25Sstefano_zampini PetscFunctionReturn(0); 1388dd9c0a25Sstefano_zampini } 1389dd9c0a25Sstefano_zampini 139068ec7858SStefano Zampini static PetscErrorCode MatMissingDiagonal_HYPRE(Mat A, PetscBool *missing, PetscInt *dd) 139168ec7858SStefano Zampini { 139268ec7858SStefano Zampini hypre_ParCSRMatrix *parcsr; 139368ec7858SStefano Zampini hypre_CSRMatrix *ha; 139468ec7858SStefano Zampini PetscInt rst; 139568ec7858SStefano Zampini PetscErrorCode ierr; 139668ec7858SStefano Zampini 139768ec7858SStefano Zampini PetscFunctionBegin; 139868ec7858SStefano Zampini if (A->rmap->n != A->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Not implemented with non-square diagonal blocks"); 139968ec7858SStefano Zampini ierr = MatGetOwnershipRange(A,&rst,NULL);CHKERRQ(ierr); 140068ec7858SStefano Zampini ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 140168ec7858SStefano Zampini if (missing) *missing = PETSC_FALSE; 140268ec7858SStefano Zampini if (dd) *dd = -1; 140368ec7858SStefano Zampini ha = hypre_ParCSRMatrixDiag(parcsr); 140468ec7858SStefano Zampini if (ha) { 140568299464SStefano Zampini PetscInt size,i; 140668299464SStefano Zampini HYPRE_Int *ii,*jj; 140768ec7858SStefano Zampini 140868ec7858SStefano Zampini size = hypre_CSRMatrixNumRows(ha); 140968ec7858SStefano Zampini ii = hypre_CSRMatrixI(ha); 141068ec7858SStefano Zampini jj = hypre_CSRMatrixJ(ha); 141168ec7858SStefano Zampini for (i = 0; i < size; i++) { 141268ec7858SStefano Zampini PetscInt j; 141368ec7858SStefano Zampini PetscBool found = PETSC_FALSE; 141468ec7858SStefano Zampini 141568ec7858SStefano Zampini for (j = ii[i]; j < ii[i+1] && !found; j++) 141668ec7858SStefano Zampini found = (jj[j] == i) ? PETSC_TRUE : PETSC_FALSE; 141768ec7858SStefano Zampini 141868ec7858SStefano Zampini if (!found) { 141968ec7858SStefano Zampini PetscInfo1(A,"Matrix is missing local diagonal entry %D\n",i); 142068ec7858SStefano Zampini if (missing) *missing = PETSC_TRUE; 142168ec7858SStefano Zampini if (dd) *dd = i+rst; 142268ec7858SStefano Zampini PetscFunctionReturn(0); 142368ec7858SStefano Zampini } 142468ec7858SStefano Zampini } 142568ec7858SStefano Zampini if (!size) { 142668ec7858SStefano Zampini PetscInfo(A,"Matrix has no diagonal entries therefore is missing diagonal\n"); 142768ec7858SStefano Zampini if (missing) *missing = PETSC_TRUE; 142868ec7858SStefano Zampini if (dd) *dd = rst; 142968ec7858SStefano Zampini } 143068ec7858SStefano Zampini } else { 143168ec7858SStefano Zampini PetscInfo(A,"Matrix has no diagonal entries therefore is missing diagonal\n"); 143268ec7858SStefano Zampini if (missing) *missing = PETSC_TRUE; 143368ec7858SStefano Zampini if (dd) *dd = rst; 143468ec7858SStefano Zampini } 143568ec7858SStefano Zampini PetscFunctionReturn(0); 143668ec7858SStefano Zampini } 143768ec7858SStefano Zampini 143868ec7858SStefano Zampini static PetscErrorCode MatScale_HYPRE(Mat A, PetscScalar s) 143968ec7858SStefano Zampini { 144068ec7858SStefano Zampini hypre_ParCSRMatrix *parcsr; 144168ec7858SStefano Zampini hypre_CSRMatrix *ha; 144268ec7858SStefano Zampini PetscErrorCode ierr; 144368ec7858SStefano Zampini 144468ec7858SStefano Zampini PetscFunctionBegin; 144568ec7858SStefano Zampini ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 144668ec7858SStefano Zampini /* diagonal part */ 144768ec7858SStefano Zampini ha = hypre_ParCSRMatrixDiag(parcsr); 144868ec7858SStefano Zampini if (ha) { 144968299464SStefano Zampini PetscInt size,i; 145068299464SStefano Zampini HYPRE_Int *ii; 145168ec7858SStefano Zampini PetscScalar *a; 145268ec7858SStefano Zampini 145368ec7858SStefano Zampini size = hypre_CSRMatrixNumRows(ha); 145468ec7858SStefano Zampini a = hypre_CSRMatrixData(ha); 145568ec7858SStefano Zampini ii = hypre_CSRMatrixI(ha); 145668ec7858SStefano Zampini for (i = 0; i < ii[size]; i++) a[i] *= s; 145768ec7858SStefano Zampini } 145868ec7858SStefano Zampini /* offdiagonal part */ 145968ec7858SStefano Zampini ha = hypre_ParCSRMatrixOffd(parcsr); 146068ec7858SStefano Zampini if (ha) { 146168299464SStefano Zampini PetscInt size,i; 146268299464SStefano Zampini HYPRE_Int *ii; 146368ec7858SStefano Zampini PetscScalar *a; 146468ec7858SStefano Zampini 146568ec7858SStefano Zampini size = hypre_CSRMatrixNumRows(ha); 146668ec7858SStefano Zampini a = hypre_CSRMatrixData(ha); 146768ec7858SStefano Zampini ii = hypre_CSRMatrixI(ha); 146868ec7858SStefano Zampini for (i = 0; i < ii[size]; i++) a[i] *= s; 146968ec7858SStefano Zampini } 147068ec7858SStefano Zampini PetscFunctionReturn(0); 147168ec7858SStefano Zampini } 147268ec7858SStefano Zampini 147368ec7858SStefano Zampini static PetscErrorCode MatZeroRowsColumns_HYPRE(Mat A, PetscInt numRows, const PetscInt rows[], PetscScalar diag, Vec x, Vec b) 147468ec7858SStefano Zampini { 147568ec7858SStefano Zampini hypre_ParCSRMatrix *parcsr; 147668299464SStefano Zampini HYPRE_Int *lrows; 147768299464SStefano Zampini PetscInt rst,ren,i; 147868ec7858SStefano Zampini PetscErrorCode ierr; 147968ec7858SStefano Zampini 148068ec7858SStefano Zampini PetscFunctionBegin; 148168ec7858SStefano Zampini if (x || b) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"To be implemented"); 148268ec7858SStefano Zampini ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 148368ec7858SStefano Zampini ierr = PetscMalloc1(numRows,&lrows);CHKERRQ(ierr); 148468ec7858SStefano Zampini ierr = MatGetOwnershipRange(A,&rst,&ren);CHKERRQ(ierr); 148568ec7858SStefano Zampini for (i=0;i<numRows;i++) { 148668ec7858SStefano Zampini if (rows[i] < rst || rows[i] >= ren) 148768ec7858SStefano Zampini SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Non-local rows not yet supported"); 148868ec7858SStefano Zampini lrows[i] = rows[i] - rst; 148968ec7858SStefano Zampini } 149068ec7858SStefano Zampini PetscStackCallStandard(hypre_ParCSRMatrixEliminateRowsCols,(parcsr,numRows,lrows)); 149168ec7858SStefano Zampini ierr = PetscFree(lrows);CHKERRQ(ierr); 149268ec7858SStefano Zampini PetscFunctionReturn(0); 149368ec7858SStefano Zampini } 149468ec7858SStefano Zampini 1495c69f721fSFande Kong static PetscErrorCode MatZeroEntries_HYPRE_CSRMatrix(hypre_CSRMatrix *ha) 1496c69f721fSFande Kong { 1497c69f721fSFande Kong PetscErrorCode ierr; 1498c69f721fSFande Kong 1499c69f721fSFande Kong PetscFunctionBegin; 1500c69f721fSFande Kong if (ha) { 1501c69f721fSFande Kong HYPRE_Int *ii, size; 1502c69f721fSFande Kong HYPRE_Complex *a; 1503c69f721fSFande Kong 1504c69f721fSFande Kong size = hypre_CSRMatrixNumRows(ha); 1505c69f721fSFande Kong a = hypre_CSRMatrixData(ha); 1506c69f721fSFande Kong ii = hypre_CSRMatrixI(ha); 1507c69f721fSFande Kong 1508c69f721fSFande Kong if (a) { ierr = PetscMemzero(a,(ii[size])*sizeof(HYPRE_Complex));CHKERRQ(ierr); } 1509c69f721fSFande Kong } 1510c69f721fSFande Kong PetscFunctionReturn(0); 1511c69f721fSFande Kong } 1512c69f721fSFande Kong 1513c69f721fSFande Kong PetscErrorCode MatZeroEntries_HYPRE(Mat A) 1514c69f721fSFande Kong { 1515c69f721fSFande Kong hypre_ParCSRMatrix *parcsr; 1516c69f721fSFande Kong PetscErrorCode ierr; 1517c69f721fSFande Kong 1518c69f721fSFande Kong PetscFunctionBegin; 1519c69f721fSFande Kong ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 1520c69f721fSFande Kong /* diagonal part */ 1521c69f721fSFande Kong ierr = MatZeroEntries_HYPRE_CSRMatrix(hypre_ParCSRMatrixDiag(parcsr));CHKERRQ(ierr); 1522c69f721fSFande Kong /* off-diagonal part */ 1523c69f721fSFande Kong ierr = MatZeroEntries_HYPRE_CSRMatrix(hypre_ParCSRMatrixOffd(parcsr));CHKERRQ(ierr); 1524c69f721fSFande Kong PetscFunctionReturn(0); 1525c69f721fSFande Kong } 1526c69f721fSFande Kong 1527c69f721fSFande Kong static PetscErrorCode MatZeroRows_HYPRE_CSRMatrix(hypre_CSRMatrix *hA,PetscInt N,const PetscInt rows[],PetscScalar diag) 1528c69f721fSFande Kong { 1529c69f721fSFande Kong PetscInt ii, jj, ibeg, iend, irow; 1530c69f721fSFande Kong PetscInt *i, *j; 1531c69f721fSFande Kong PetscScalar *a; 1532c69f721fSFande Kong 1533c69f721fSFande Kong PetscFunctionBegin; 1534c69f721fSFande Kong if (!hA) PetscFunctionReturn(0); 1535c69f721fSFande Kong 153608defe43SFande Kong i = (PetscInt*) hypre_CSRMatrixI(hA); 153708defe43SFande Kong j = (PetscInt*) hypre_CSRMatrixJ(hA); 1538c69f721fSFande Kong a = hypre_CSRMatrixData(hA); 1539c69f721fSFande Kong 1540c69f721fSFande Kong for (ii = 0; ii < N; ii++) { 1541c69f721fSFande Kong irow = rows[ii]; 1542c69f721fSFande Kong ibeg = i[irow]; 1543c69f721fSFande Kong iend = i[irow+1]; 1544c69f721fSFande Kong for (jj = ibeg; jj < iend; jj++) 1545c69f721fSFande Kong if (j[jj] == irow) a[jj] = diag; 1546c69f721fSFande Kong else a[jj] = 0.0; 1547c69f721fSFande Kong } 1548c69f721fSFande Kong PetscFunctionReturn(0); 1549c69f721fSFande Kong } 1550c69f721fSFande Kong 1551ddbeb582SStefano Zampini static PetscErrorCode MatZeroRows_HYPRE(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag,Vec x,Vec b) 1552c69f721fSFande Kong { 1553c69f721fSFande Kong hypre_ParCSRMatrix *parcsr; 1554c69f721fSFande Kong PetscInt *lrows,len; 1555c69f721fSFande Kong PetscErrorCode ierr; 1556c69f721fSFande Kong 1557c69f721fSFande Kong PetscFunctionBegin; 1558c69f721fSFande 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"); 1559c69f721fSFande Kong /* retrieve the internal matrix */ 1560c69f721fSFande Kong ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 1561c69f721fSFande Kong /* get locally owned rows */ 1562c69f721fSFande Kong ierr = MatZeroRowsMapLocal_Private(A,N,rows,&len,&lrows);CHKERRQ(ierr); 1563c69f721fSFande Kong /* zero diagonal part */ 1564c69f721fSFande Kong ierr = MatZeroRows_HYPRE_CSRMatrix(hypre_ParCSRMatrixDiag(parcsr),len,lrows,diag);CHKERRQ(ierr); 1565c69f721fSFande Kong /* zero off-diagonal part */ 1566c69f721fSFande Kong ierr = MatZeroRows_HYPRE_CSRMatrix(hypre_ParCSRMatrixOffd(parcsr),len,lrows,0.0);CHKERRQ(ierr); 1567c69f721fSFande Kong 1568c69f721fSFande Kong ierr = PetscFree(lrows);CHKERRQ(ierr); 1569c69f721fSFande Kong PetscFunctionReturn(0); 1570c69f721fSFande Kong } 1571c69f721fSFande Kong 1572ddbeb582SStefano Zampini static PetscErrorCode MatAssemblyBegin_HYPRE(Mat mat,MatAssemblyType mode) 1573c69f721fSFande Kong { 1574c69f721fSFande Kong PetscErrorCode ierr; 1575c69f721fSFande Kong 1576c69f721fSFande Kong PetscFunctionBegin; 1577c69f721fSFande Kong if (mat->nooffprocentries) PetscFunctionReturn(0); 1578c69f721fSFande Kong 1579c69f721fSFande Kong ierr = MatStashScatterBegin_Private(mat,&mat->stash,mat->rmap->range);CHKERRQ(ierr); 1580c69f721fSFande Kong PetscFunctionReturn(0); 1581c69f721fSFande Kong } 1582c69f721fSFande Kong 1583ddbeb582SStefano Zampini static PetscErrorCode MatGetRow_HYPRE(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 1584c69f721fSFande Kong { 1585c69f721fSFande Kong hypre_ParCSRMatrix *parcsr; 1586c69f721fSFande Kong PetscErrorCode ierr; 1587c69f721fSFande Kong 1588c69f721fSFande Kong PetscFunctionBegin; 1589c69f721fSFande Kong /* retrieve the internal matrix */ 1590c69f721fSFande Kong ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 1591c69f721fSFande Kong /* call HYPRE API */ 159208defe43SFande Kong PetscStackCallStandard(HYPRE_ParCSRMatrixGetRow,(parcsr,row,(HYPRE_Int*)nz,(HYPRE_Int**)idx,v)); 1593c69f721fSFande Kong PetscFunctionReturn(0); 1594c69f721fSFande Kong } 1595c69f721fSFande Kong 1596ddbeb582SStefano Zampini static PetscErrorCode MatRestoreRow_HYPRE(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 1597c69f721fSFande Kong { 1598c69f721fSFande Kong hypre_ParCSRMatrix *parcsr; 1599c69f721fSFande Kong PetscErrorCode ierr; 1600c69f721fSFande Kong 1601c69f721fSFande Kong PetscFunctionBegin; 1602c69f721fSFande Kong /* retrieve the internal matrix */ 1603c69f721fSFande Kong ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 1604c69f721fSFande Kong /* call HYPRE API */ 160508defe43SFande Kong PetscStackCallStandard(HYPRE_ParCSRMatrixRestoreRow,(parcsr,row,(HYPRE_Int*)nz,(HYPRE_Int**)idx,v)); 1606c69f721fSFande Kong PetscFunctionReturn(0); 1607c69f721fSFande Kong } 1608c69f721fSFande Kong 1609ddbeb582SStefano Zampini static PetscErrorCode MatGetValues_HYPRE(Mat A,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],PetscScalar v[]) 1610c69f721fSFande Kong { 1611*45b8d346SStefano Zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 1612c69f721fSFande Kong PetscInt i; 16131d4906efSStefano Zampini 1614c69f721fSFande Kong PetscFunctionBegin; 1615c69f721fSFande Kong if (!m || !n) PetscFunctionReturn(0); 1616c69f721fSFande Kong /* Ignore negative row indices 1617c69f721fSFande Kong * And negative column indices should be automatically ignored in hypre 1618c69f721fSFande Kong * */ 1619c69f721fSFande Kong for (i=0; i<m; i++) 1620*45b8d346SStefano Zampini if (idxm[i] >= 0) PetscStackCallStandard(HYPRE_IJMatrixGetValues,(hA->ij,1,(HYPRE_Int*)&n,(HYPRE_Int*)&idxm[i],(HYPRE_Int*)idxn,&v[i*n])); 1621c69f721fSFande Kong PetscFunctionReturn(0); 1622c69f721fSFande Kong } 1623c69f721fSFande Kong 1624ddbeb582SStefano Zampini static PetscErrorCode MatSetOption_HYPRE(Mat A,MatOption op,PetscBool flg) 1625ddbeb582SStefano Zampini { 1626ddbeb582SStefano Zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 1627ddbeb582SStefano Zampini 1628ddbeb582SStefano Zampini PetscFunctionBegin; 1629ddbeb582SStefano Zampini switch (op) 1630ddbeb582SStefano Zampini { 1631ddbeb582SStefano Zampini case MAT_NO_OFF_PROC_ENTRIES: 1632ddbeb582SStefano Zampini if (flg) { 1633ddbeb582SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixSetMaxOffProcElmts,(hA->ij,0)); 1634ddbeb582SStefano Zampini } 1635ddbeb582SStefano Zampini break; 1636ddbeb582SStefano Zampini default: 1637ddbeb582SStefano Zampini break; 1638ddbeb582SStefano Zampini } 1639ddbeb582SStefano Zampini PetscFunctionReturn(0); 1640ddbeb582SStefano Zampini } 1641c69f721fSFande Kong 1642*45b8d346SStefano Zampini static PetscErrorCode MatView_HYPRE(Mat A, PetscViewer view) 1643*45b8d346SStefano Zampini { 1644*45b8d346SStefano Zampini hypre_ParCSRMatrix *parcsr; 1645*45b8d346SStefano Zampini PetscErrorCode ierr; 1646*45b8d346SStefano Zampini Mat B; 1647*45b8d346SStefano Zampini PetscViewerFormat format; 1648*45b8d346SStefano Zampini PetscErrorCode (*mview)(Mat,PetscViewer) = NULL; 1649*45b8d346SStefano Zampini 1650*45b8d346SStefano Zampini PetscFunctionBegin; 1651*45b8d346SStefano Zampini ierr = PetscViewerGetFormat(view,&format);CHKERRQ(ierr); 1652*45b8d346SStefano Zampini if (format != PETSC_VIEWER_NATIVE) { 1653*45b8d346SStefano Zampini ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 1654*45b8d346SStefano Zampini ierr = MatCreateFromParCSR(parcsr,MATAIJ,PETSC_USE_POINTER,&B);CHKERRQ(ierr); 1655*45b8d346SStefano Zampini ierr = MatGetOperation(B,MATOP_VIEW,(void(**)(void))&mview);CHKERRQ(ierr); 1656*45b8d346SStefano Zampini if (!mview) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Missing view operation");CHKERRQ(ierr); 1657*45b8d346SStefano Zampini ierr = (*mview)(B,view);CHKERRQ(ierr); 1658*45b8d346SStefano Zampini ierr = MatDestroy(&B);CHKERRQ(ierr); 1659*45b8d346SStefano Zampini } else { 1660*45b8d346SStefano Zampini Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 1661*45b8d346SStefano Zampini PetscMPIInt size; 1662*45b8d346SStefano Zampini PetscBool isascii; 1663*45b8d346SStefano Zampini const char *filename; 1664*45b8d346SStefano Zampini 1665*45b8d346SStefano Zampini /* HYPRE uses only text files */ 1666*45b8d346SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)view,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr); 1667*45b8d346SStefano Zampini if (!isascii) SETERRQ1(PetscObjectComm((PetscObject)view),PETSC_ERR_SUP,"PetscViewerType %s: native HYPRE format needs PETSCVIEWERASCII",((PetscObject)view)->type_name); 1668*45b8d346SStefano Zampini ierr = PetscViewerFileGetName(view,&filename);CHKERRQ(ierr); 1669*45b8d346SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixPrint,(hA->ij,filename)); 1670*45b8d346SStefano Zampini ierr = MPI_Comm_size(hA->comm,&size);CHKERRQ(ierr); 1671*45b8d346SStefano Zampini if (size > 1) { 1672*45b8d346SStefano Zampini ierr = PetscViewerASCIIPrintf(view,"Matrix files: %s.%05d ... %s.%05d\n",filename,0,filename,size-1);CHKERRQ(ierr); 1673*45b8d346SStefano Zampini } else { 1674*45b8d346SStefano Zampini ierr = PetscViewerASCIIPrintf(view,"Matrix file: %s.%05d\n",filename,0);CHKERRQ(ierr); 1675*45b8d346SStefano Zampini } 1676*45b8d346SStefano Zampini } 1677*45b8d346SStefano Zampini PetscFunctionReturn(0); 1678*45b8d346SStefano Zampini } 1679*45b8d346SStefano Zampini 1680*45b8d346SStefano Zampini static PetscErrorCode MatDuplicate_HYPRE(Mat A,MatDuplicateOption op, Mat *B) 1681*45b8d346SStefano Zampini { 1682*45b8d346SStefano Zampini hypre_ParCSRMatrix *parcsr; 1683*45b8d346SStefano Zampini PetscErrorCode ierr; 1684*45b8d346SStefano Zampini PetscCopyMode cpmode; 1685*45b8d346SStefano Zampini 1686*45b8d346SStefano Zampini PetscFunctionBegin; 1687*45b8d346SStefano Zampini ierr = MatHYPREGetParCSR_HYPRE(A,&parcsr);CHKERRQ(ierr); 1688*45b8d346SStefano Zampini if (op == MAT_DO_NOT_COPY_VALUES || op == MAT_SHARE_NONZERO_PATTERN) { 1689*45b8d346SStefano Zampini parcsr = hypre_ParCSRMatrixCompleteClone(parcsr); 1690*45b8d346SStefano Zampini cpmode = PETSC_OWN_POINTER; 1691*45b8d346SStefano Zampini } else { 1692*45b8d346SStefano Zampini cpmode = PETSC_COPY_VALUES; 1693*45b8d346SStefano Zampini } 1694*45b8d346SStefano Zampini ierr = MatCreateFromParCSR(parcsr,MATHYPRE,cpmode,B);CHKERRQ(ierr); 1695*45b8d346SStefano Zampini PetscFunctionReturn(0); 1696*45b8d346SStefano Zampini } 1697*45b8d346SStefano Zampini 1698a055b5aaSBarry Smith /*MC 1699a055b5aaSBarry Smith MATHYPRE - MATHYPRE = "hypre" - A matrix type to be used for sequential and parallel sparse matrices 1700a055b5aaSBarry Smith based on the hypre IJ interface. 1701a055b5aaSBarry Smith 1702a055b5aaSBarry Smith Level: intermediate 1703a055b5aaSBarry Smith 1704a055b5aaSBarry Smith .seealso: MatCreate() 1705a055b5aaSBarry Smith M*/ 1706a055b5aaSBarry Smith 170763c07aadSStefano Zampini PETSC_EXTERN PetscErrorCode MatCreate_HYPRE(Mat B) 170863c07aadSStefano Zampini { 170963c07aadSStefano Zampini Mat_HYPRE *hB; 171063c07aadSStefano Zampini PetscErrorCode ierr; 171163c07aadSStefano Zampini 171263c07aadSStefano Zampini PetscFunctionBegin; 171363c07aadSStefano Zampini ierr = PetscNewLog(B,&hB);CHKERRQ(ierr); 1714978814f1SStefano Zampini hB->inner_free = PETSC_TRUE; 1715c69f721fSFande Kong hB->available = PETSC_TRUE; 1716c69f721fSFande Kong hB->size = 0; 1717c69f721fSFande Kong hB->array = NULL; 1718978814f1SStefano Zampini 171963c07aadSStefano Zampini B->data = (void*)hB; 172063c07aadSStefano Zampini B->rmap->bs = 1; 172163c07aadSStefano Zampini B->assembled = PETSC_FALSE; 172263c07aadSStefano Zampini 1723de393100SStefano Zampini ierr = PetscMemzero(B->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 172463c07aadSStefano Zampini B->ops->mult = MatMult_HYPRE; 172563c07aadSStefano Zampini B->ops->multtranspose = MatMultTranspose_HYPRE; 172663c07aadSStefano Zampini B->ops->setup = MatSetUp_HYPRE; 172763c07aadSStefano Zampini B->ops->destroy = MatDestroy_HYPRE; 172863c07aadSStefano Zampini B->ops->assemblyend = MatAssemblyEnd_HYPRE; 1729c69f721fSFande Kong B->ops->assemblybegin = MatAssemblyBegin_HYPRE; 1730cd8bc7baSStefano Zampini B->ops->ptap = MatPtAP_HYPRE_HYPRE; 1731d501dc42Sstefano_zampini B->ops->matmult = MatMatMult_HYPRE_HYPRE; 1732d975228cSstefano_zampini B->ops->setvalues = MatSetValues_HYPRE; 173368ec7858SStefano Zampini B->ops->missingdiagonal = MatMissingDiagonal_HYPRE; 173468ec7858SStefano Zampini B->ops->scale = MatScale_HYPRE; 173568ec7858SStefano Zampini B->ops->zerorowscolumns = MatZeroRowsColumns_HYPRE; 1736c69f721fSFande Kong B->ops->zeroentries = MatZeroEntries_HYPRE; 1737c69f721fSFande Kong B->ops->zerorows = MatZeroRows_HYPRE; 1738c69f721fSFande Kong B->ops->getrow = MatGetRow_HYPRE; 1739c69f721fSFande Kong B->ops->restorerow = MatRestoreRow_HYPRE; 1740c69f721fSFande Kong B->ops->getvalues = MatGetValues_HYPRE; 1741ddbeb582SStefano Zampini B->ops->setoption = MatSetOption_HYPRE; 1742*45b8d346SStefano Zampini B->ops->duplicate = MatDuplicate_HYPRE; 1743*45b8d346SStefano Zampini B->ops->view = MatView_HYPRE; 1744*45b8d346SStefano Zampini 1745*45b8d346SStefano Zampini /* build cache for off array entries formed */ 1746*45b8d346SStefano Zampini ierr = MatStashCreate_Private(PetscObjectComm((PetscObject)B),1,&B->stash);CHKERRQ(ierr); 174763c07aadSStefano Zampini 174863c07aadSStefano Zampini ierr = MPI_Comm_dup(PetscObjectComm((PetscObject)B),&hB->comm);CHKERRQ(ierr); 174963c07aadSStefano Zampini ierr = PetscObjectChangeTypeName((PetscObject)B,MATHYPRE);CHKERRQ(ierr); 175063c07aadSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_hypre_aij_C",MatConvert_HYPRE_AIJ);CHKERRQ(ierr); 17512df22349SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_hypre_is_C",MatConvert_HYPRE_IS);CHKERRQ(ierr); 1752c1a070e6SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)B,"MatPtAP_seqaij_hypre_C",MatPtAP_AIJ_HYPRE);CHKERRQ(ierr); 1753c1a070e6SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)B,"MatPtAP_mpiaij_hypre_C",MatPtAP_AIJ_HYPRE);CHKERRQ(ierr); 1754d975228cSstefano_zampini ierr = PetscObjectComposeFunction((PetscObject)B,"MatHYPRESetPreallocation_C",MatHYPRESetPreallocation_HYPRE);CHKERRQ(ierr); 1755dd9c0a25Sstefano_zampini ierr = PetscObjectComposeFunction((PetscObject)B,"MatHYPREGetParCSR_C",MatHYPREGetParCSR_HYPRE);CHKERRQ(ierr); 175675d48cdbSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)B,"MatPtAP_is_hypre_C",MatPtAP_IS_XAIJ);CHKERRQ(ierr); 175763c07aadSStefano Zampini PetscFunctionReturn(0); 175863c07aadSStefano Zampini } 175963c07aadSStefano Zampini 1760225daaf8SStefano Zampini static PetscErrorCode hypre_array_destroy(void *ptr) 1761225daaf8SStefano Zampini { 1762225daaf8SStefano Zampini PetscFunctionBegin; 1763e6de0934SSatish Balay hypre_TFree(ptr,HYPRE_MEMORY_HOST); 1764225daaf8SStefano Zampini PetscFunctionReturn(0); 1765225daaf8SStefano Zampini } 1766