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