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