1 2 /* 3 Creates hypre ijmatrix from PETSc matrix 4 */ 5 6 /*MC 7 MATHYPRE - MATHYPRE = "hypre" - A matrix type to be used for sequential and parallel sparse matrices 8 based on the hypre IJ interface. 9 10 Level: intermediate 11 12 .seealso: MatCreate() 13 M*/ 14 15 #include <petscmathypre.h> 16 #include <petsc/private/matimpl.h> 17 #include <../src/mat/impls/hypre/mhypre.h> 18 #include <../src/mat/impls/aij/mpi/mpiaij.h> 19 #include <../src/vec/vec/impls/hypre/vhyp.h> 20 #include <HYPRE.h> 21 #include <HYPRE_utilities.h> 22 #include <_hypre_parcsr_ls.h> 23 24 static PetscErrorCode MatHYPRE_CreateFromMat(Mat,Mat_HYPRE*); 25 static PetscErrorCode MatHYPRE_IJMatrixPreallocate(Mat,Mat,HYPRE_IJMatrix); 26 static PetscErrorCode MatHYPRE_IJMatrixFastCopy_MPIAIJ(Mat,HYPRE_IJMatrix); 27 static PetscErrorCode MatHYPRE_IJMatrixFastCopy_SeqAIJ(Mat,HYPRE_IJMatrix); 28 static PetscErrorCode MatHYPRE_MultKernel_Private(Mat,Vec,Vec,PetscBool); 29 static PetscErrorCode hypre_array_destroy(void*); 30 31 #undef __FUNCT__ 32 #define __FUNCT__ "MatHYPRE_IJMatrixPreallocate" 33 static PetscErrorCode MatHYPRE_IJMatrixPreallocate(Mat A_d, Mat A_o, HYPRE_IJMatrix ij) 34 { 35 PetscErrorCode ierr; 36 PetscInt i,n_d,n_o; 37 const PetscInt *ia_d,*ia_o; 38 PetscBool done_d=PETSC_FALSE,done_o=PETSC_FALSE; 39 PetscInt *nnz_d=NULL,*nnz_o=NULL; 40 41 PetscFunctionBegin; 42 if (A_d) { /* determine number of nonzero entries in local diagonal part */ 43 ierr = MatGetRowIJ(A_d,0,PETSC_FALSE,PETSC_FALSE,&n_d,&ia_d,NULL,&done_d);CHKERRQ(ierr); 44 if (done_d) { 45 ierr = PetscMalloc1(n_d,&nnz_d);CHKERRQ(ierr); 46 for (i=0; i<n_d; i++) { 47 nnz_d[i] = ia_d[i+1] - ia_d[i]; 48 } 49 } 50 ierr = MatRestoreRowIJ(A_d,0,PETSC_FALSE,PETSC_FALSE,NULL,&ia_d,NULL,&done_d);CHKERRQ(ierr); 51 } 52 if (A_o) { /* determine number of nonzero entries in local off-diagonal part */ 53 ierr = MatGetRowIJ(A_o,0,PETSC_FALSE,PETSC_FALSE,&n_o,&ia_o,NULL,&done_o);CHKERRQ(ierr); 54 if (done_o) { 55 ierr = PetscMalloc1(n_o,&nnz_o);CHKERRQ(ierr); 56 for (i=0; i<n_o; i++) { 57 nnz_o[i] = ia_o[i+1] - ia_o[i]; 58 } 59 } 60 ierr = MatRestoreRowIJ(A_o,0,PETSC_FALSE,PETSC_FALSE,&n_o,&ia_o,NULL,&done_o);CHKERRQ(ierr); 61 } 62 if (done_d) { /* set number of nonzeros in HYPRE IJ matrix */ 63 if (!done_o) { /* only diagonal part */ 64 ierr = PetscMalloc1(n_d,&nnz_o);CHKERRQ(ierr); 65 for (i=0; i<n_d; i++) { 66 nnz_o[i] = 0; 67 } 68 } 69 PetscStackCallStandard(HYPRE_IJMatrixSetDiagOffdSizes,(ij,(HYPRE_Int *)nnz_d,(HYPRE_Int *)nnz_o)); 70 ierr = PetscFree(nnz_d);CHKERRQ(ierr); 71 ierr = PetscFree(nnz_o);CHKERRQ(ierr); 72 } 73 PetscFunctionReturn(0); 74 } 75 76 #undef __FUNCT__ 77 #define __FUNCT__ "MatHYPRE_CreateFromMat" 78 static PetscErrorCode MatHYPRE_CreateFromMat(Mat A, Mat_HYPRE *hA) 79 { 80 PetscErrorCode ierr; 81 PetscInt rstart,rend,cstart,cend; 82 83 PetscFunctionBegin; 84 ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 85 ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 86 rstart = A->rmap->rstart; 87 rend = A->rmap->rend; 88 cstart = A->cmap->rstart; 89 cend = A->cmap->rend; 90 PetscStackCallStandard(HYPRE_IJMatrixCreate,(hA->comm,rstart,rend-1,cstart,cend-1,&hA->ij)); 91 PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(hA->ij,HYPRE_PARCSR)); 92 { 93 PetscBool same; 94 Mat A_d,A_o; 95 const PetscInt *colmap; 96 ierr = PetscObjectTypeCompare((PetscObject)A,MATMPIAIJ,&same);CHKERRQ(ierr); 97 if (same) { 98 ierr = MatMPIAIJGetSeqAIJ(A,&A_d,&A_o,&colmap);CHKERRQ(ierr); 99 ierr = MatHYPRE_IJMatrixPreallocate(A_d,A_o,hA->ij);CHKERRQ(ierr); 100 PetscFunctionReturn(0); 101 } 102 ierr = PetscObjectTypeCompare((PetscObject)A,MATMPIBAIJ,&same);CHKERRQ(ierr); 103 if (same) { 104 ierr = MatMPIBAIJGetSeqBAIJ(A,&A_d,&A_o,&colmap);CHKERRQ(ierr); 105 ierr = MatHYPRE_IJMatrixPreallocate(A_d,A_o,hA->ij);CHKERRQ(ierr); 106 PetscFunctionReturn(0); 107 } 108 ierr = PetscObjectTypeCompare((PetscObject)A,MATSEQAIJ,&same);CHKERRQ(ierr); 109 if (same) { 110 ierr = MatHYPRE_IJMatrixPreallocate(A,NULL,hA->ij);CHKERRQ(ierr); 111 PetscFunctionReturn(0); 112 } 113 ierr = PetscObjectTypeCompare((PetscObject)A,MATSEQBAIJ,&same);CHKERRQ(ierr); 114 if (same) { 115 ierr = MatHYPRE_IJMatrixPreallocate(A,NULL,hA->ij);CHKERRQ(ierr); 116 PetscFunctionReturn(0); 117 } 118 } 119 PetscFunctionReturn(0); 120 } 121 122 #undef __FUNCT__ 123 #define __FUNCT__ "MatHYPRE_IJMatrixCopy" 124 static PetscErrorCode MatHYPRE_IJMatrixCopy(Mat A, HYPRE_IJMatrix ij) 125 { 126 PetscErrorCode ierr; 127 PetscInt i,rstart,rend,ncols,nr,nc; 128 const PetscScalar *values; 129 const PetscInt *cols; 130 PetscBool flg; 131 132 PetscFunctionBegin; 133 ierr = PetscObjectTypeCompare((PetscObject)A,MATMPIAIJ,&flg);CHKERRQ(ierr); 134 ierr = MatGetSize(A,&nr,&nc);CHKERRQ(ierr); 135 if (flg && nr == nc) { 136 ierr = MatHYPRE_IJMatrixFastCopy_MPIAIJ(A,ij);CHKERRQ(ierr); 137 PetscFunctionReturn(0); 138 } 139 ierr = PetscObjectTypeCompare((PetscObject)A,MATSEQAIJ,&flg);CHKERRQ(ierr); 140 if (flg) { 141 ierr = MatHYPRE_IJMatrixFastCopy_SeqAIJ(A,ij);CHKERRQ(ierr); 142 PetscFunctionReturn(0); 143 } 144 145 PetscStackCallStandard(HYPRE_IJMatrixInitialize,(ij)); 146 ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr); 147 for (i=rstart; i<rend; i++) { 148 ierr = MatGetRow(A,i,&ncols,&cols,&values);CHKERRQ(ierr); 149 PetscStackCallStandard(HYPRE_IJMatrixSetValues,(ij,1,(HYPRE_Int *)&ncols,(HYPRE_Int *)&i,(HYPRE_Int *)cols,values)); 150 ierr = MatRestoreRow(A,i,&ncols,&cols,&values);CHKERRQ(ierr); 151 } 152 PetscFunctionReturn(0); 153 } 154 155 #undef __FUNCT__ 156 #define __FUNCT__ "MatHYPRE_IJMatrixFastCopy_SeqAIJ" 157 static PetscErrorCode MatHYPRE_IJMatrixFastCopy_SeqAIJ(Mat A, HYPRE_IJMatrix ij) 158 { 159 PetscErrorCode ierr; 160 Mat_SeqAIJ *pdiag = (Mat_SeqAIJ*)A->data; 161 HYPRE_Int type; 162 hypre_ParCSRMatrix *par_matrix; 163 hypre_AuxParCSRMatrix *aux_matrix; 164 hypre_CSRMatrix *hdiag; 165 166 PetscFunctionBegin; 167 PetscStackCallStandard(HYPRE_IJMatrixInitialize,(ij)); 168 PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(ij,&type)); 169 if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 170 PetscStackCallStandard(HYPRE_IJMatrixGetObject,(ij,(void**)&par_matrix)); 171 hdiag = hypre_ParCSRMatrixDiag(par_matrix); 172 /* 173 this is the Hack part where we monkey directly with the hypre datastructures 174 */ 175 ierr = PetscMemcpy(hdiag->i,pdiag->i,(A->rmap->n + 1)*sizeof(PetscInt)); 176 ierr = PetscMemcpy(hdiag->j,pdiag->j,pdiag->nz*sizeof(PetscInt)); 177 ierr = PetscMemcpy(hdiag->data,pdiag->a,pdiag->nz*sizeof(PetscScalar)); 178 179 aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(ij); 180 hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 0; 181 PetscFunctionReturn(0); 182 } 183 184 #undef __FUNCT__ 185 #define __FUNCT__ "MatHYPRE_IJMatrixFastCopy_MPIAIJ" 186 static PetscErrorCode MatHYPRE_IJMatrixFastCopy_MPIAIJ(Mat A, HYPRE_IJMatrix ij) 187 { 188 PetscErrorCode ierr; 189 Mat_MPIAIJ *pA = (Mat_MPIAIJ*)A->data; 190 Mat_SeqAIJ *pdiag,*poffd; 191 PetscInt i,*garray = pA->garray,*jj,cstart,*pjj; 192 HYPRE_Int type; 193 hypre_ParCSRMatrix *par_matrix; 194 hypre_AuxParCSRMatrix *aux_matrix; 195 hypre_CSRMatrix *hdiag,*hoffd; 196 197 PetscFunctionBegin; 198 pdiag = (Mat_SeqAIJ*) pA->A->data; 199 poffd = (Mat_SeqAIJ*) pA->B->data; 200 /* cstart is only valid for square MPIAIJ layed out in the usual way */ 201 ierr = MatGetOwnershipRange(A,&cstart,NULL);CHKERRQ(ierr); 202 203 PetscStackCallStandard(HYPRE_IJMatrixInitialize,(ij)); 204 PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(ij,&type)); 205 if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 206 PetscStackCallStandard(HYPRE_IJMatrixGetObject,(ij,(void**)&par_matrix)); 207 hdiag = hypre_ParCSRMatrixDiag(par_matrix); 208 hoffd = hypre_ParCSRMatrixOffd(par_matrix); 209 210 /* 211 this is the Hack part where we monkey directly with the hypre datastructures 212 */ 213 ierr = PetscMemcpy(hdiag->i,pdiag->i,(pA->A->rmap->n + 1)*sizeof(PetscInt)); 214 /* need to shift the diag column indices (hdiag->j) back to global numbering since hypre is expecting this */ 215 jj = (PetscInt*)hdiag->j; 216 pjj = (PetscInt*)pdiag->j; 217 for (i=0; i<pdiag->nz; i++) jj[i] = cstart + pjj[i]; 218 ierr = PetscMemcpy(hdiag->data,pdiag->a,pdiag->nz*sizeof(PetscScalar)); 219 ierr = PetscMemcpy(hoffd->i,poffd->i,(pA->A->rmap->n + 1)*sizeof(PetscInt)); 220 /* need to move the offd column indices (hoffd->j) back to global numbering since hypre is expecting this 221 If we hacked a hypre a bit more we might be able to avoid this step */ 222 jj = (PetscInt*) hoffd->j; 223 pjj = (PetscInt*) poffd->j; 224 for (i=0; i<poffd->nz; i++) jj[i] = garray[pjj[i]]; 225 ierr = PetscMemcpy(hoffd->data,poffd->a,poffd->nz*sizeof(PetscScalar)); 226 227 aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(ij); 228 hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 0; 229 PetscFunctionReturn(0); 230 } 231 232 #undef __FUNCT__ 233 #define __FUNCT__ "MatConvert_HYPRE_IS" 234 static PetscErrorCode MatConvert_HYPRE_IS(Mat A, MatType mtype, MatReuse reuse, Mat* B) 235 { 236 Mat_HYPRE* mhA = (Mat_HYPRE*)(A->data); 237 Mat lA; 238 ISLocalToGlobalMapping rl2g,cl2g; 239 IS is; 240 hypre_ParCSRMatrix *hA; 241 hypre_CSRMatrix *hdiag,*hoffd; 242 MPI_Comm comm; 243 PetscScalar *hdd,*hod,*aa,*data; 244 HYPRE_Int *col_map_offd,*hdi,*hdj,*hoi,*hoj; 245 PetscInt *ii,*jj,*iptr,*jptr; 246 PetscInt cum,dr,dc,oc,str,stc,nnz,i,jd,jo,M,N; 247 HYPRE_Int type; 248 PetscErrorCode ierr; 249 250 PetscFunctionBegin; 251 comm = PetscObjectComm((PetscObject)A); 252 PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(mhA->ij,&type)); 253 if (type != HYPRE_PARCSR) SETERRQ(comm,PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 254 PetscStackCallStandard(HYPRE_IJMatrixGetObject,(mhA->ij,(void**)&hA)); 255 M = hypre_ParCSRMatrixGlobalNumRows(hA); 256 N = hypre_ParCSRMatrixGlobalNumCols(hA); 257 str = hypre_ParCSRMatrixFirstRowIndex(hA); 258 stc = hypre_ParCSRMatrixFirstColDiag(hA); 259 hdiag = hypre_ParCSRMatrixDiag(hA); 260 hoffd = hypre_ParCSRMatrixOffd(hA); 261 dr = hypre_CSRMatrixNumRows(hdiag); 262 dc = hypre_CSRMatrixNumCols(hdiag); 263 nnz = hypre_CSRMatrixNumNonzeros(hdiag); 264 hdi = hypre_CSRMatrixI(hdiag); 265 hdj = hypre_CSRMatrixJ(hdiag); 266 hdd = hypre_CSRMatrixData(hdiag); 267 oc = hypre_CSRMatrixNumCols(hoffd); 268 nnz += hypre_CSRMatrixNumNonzeros(hoffd); 269 hoi = hypre_CSRMatrixI(hoffd); 270 hoj = hypre_CSRMatrixJ(hoffd); 271 hod = hypre_CSRMatrixData(hoffd); 272 if (reuse != MAT_REUSE_MATRIX) { 273 PetscInt *aux; 274 275 /* generate l2g maps for rows and cols */ 276 ierr = ISCreateStride(comm,dr,str,1,&is);CHKERRQ(ierr); 277 ierr = ISLocalToGlobalMappingCreateIS(is,&rl2g);CHKERRQ(ierr); 278 ierr = ISDestroy(&is);CHKERRQ(ierr); 279 col_map_offd = hypre_ParCSRMatrixColMapOffd(hA); 280 ierr = PetscMalloc1(dc+oc,&aux);CHKERRQ(ierr); 281 for (i=0; i<dc; i++) aux[i] = i+stc; 282 for (i=0; i<oc; i++) aux[i+dc] = col_map_offd[i]; 283 ierr = ISCreateGeneral(comm,dc+oc,aux,PETSC_OWN_POINTER,&is);CHKERRQ(ierr); 284 ierr = ISLocalToGlobalMappingCreateIS(is,&cl2g);CHKERRQ(ierr); 285 ierr = ISDestroy(&is);CHKERRQ(ierr); 286 /* create MATIS object */ 287 ierr = MatCreate(comm,B);CHKERRQ(ierr); 288 ierr = MatSetSizes(*B,dr,dc,M,N);CHKERRQ(ierr); 289 ierr = MatSetType(*B,MATIS);CHKERRQ(ierr); 290 ierr = MatSetLocalToGlobalMapping(*B,rl2g,cl2g);CHKERRQ(ierr); 291 ierr = ISLocalToGlobalMappingDestroy(&rl2g);CHKERRQ(ierr); 292 ierr = ISLocalToGlobalMappingDestroy(&cl2g);CHKERRQ(ierr); 293 294 /* allocate CSR for local matrix */ 295 ierr = PetscMalloc1(dr+1,&iptr);CHKERRQ(ierr); 296 ierr = PetscMalloc1(nnz,&jptr);CHKERRQ(ierr); 297 ierr = PetscMalloc1(nnz,&data);CHKERRQ(ierr); 298 } else { 299 PetscInt nr; 300 PetscBool done; 301 ierr = MatISGetLocalMat(*B,&lA);CHKERRQ(ierr); 302 ierr = MatGetRowIJ(lA,0,PETSC_FALSE,PETSC_FALSE,&nr,(const PetscInt**)&iptr,(const PetscInt**)&jptr,&done);CHKERRQ(ierr); 303 if (nr != dr) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"Cannot reuse mat: invalid number of rows in local mat! %D != %D",nr,dr); 304 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); 305 ierr = MatSeqAIJGetArray(lA,&data);CHKERRQ(ierr); 306 } 307 /* merge local matrices */ 308 ii = iptr; 309 jj = jptr; 310 aa = data; 311 *ii = *(hdi++) + *(hoi++); 312 for (jd=0,jo=0,cum=0; *ii<nnz; cum++) { 313 PetscScalar *aold = aa; 314 PetscInt *jold = jj,nc = jd+jo; 315 for (; jd<*hdi; jd++) { *jj++ = *hdj++; *aa++ = *hdd++; } 316 for (; jo<*hoi; jo++) { *jj++ = *hoj++ + dc; *aa++ = *hod++; } 317 *(++ii) = *(hdi++) + *(hoi++); 318 ierr = PetscSortIntWithScalarArray(jd+jo-nc,jold,aold);CHKERRQ(ierr); 319 } 320 for (; cum<dr; cum++) *(++ii) = nnz; 321 if (reuse != MAT_REUSE_MATRIX) { 322 Mat_SeqAIJ* a; 323 324 ierr = MatCreateSeqAIJWithArrays(PETSC_COMM_SELF,dr,dc+oc,iptr,jptr,data,&lA);CHKERRQ(ierr); 325 ierr = MatISSetLocalMat(*B,lA);CHKERRQ(ierr); 326 /* hack SeqAIJ */ 327 a = (Mat_SeqAIJ*)(lA->data); 328 a->free_a = PETSC_TRUE; 329 a->free_ij = PETSC_TRUE; 330 ierr = MatDestroy(&lA);CHKERRQ(ierr); 331 } 332 ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 333 ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 334 if (reuse == MAT_INPLACE_MATRIX) { 335 ierr = MatHeaderReplace(A,B);CHKERRQ(ierr); 336 } 337 PetscFunctionReturn(0); 338 } 339 340 #undef __FUNCT__ 341 #define __FUNCT__ "MatConvert_AIJ_HYPRE" 342 PETSC_INTERN PetscErrorCode MatConvert_AIJ_HYPRE(Mat A, MatType type, MatReuse reuse, Mat *B) 343 { 344 Mat_HYPRE *hB; 345 MPI_Comm comm = PetscObjectComm((PetscObject)A); 346 PetscErrorCode ierr; 347 348 PetscFunctionBegin; 349 if (reuse == MAT_INPLACE_MATRIX) SETERRQ(comm,PETSC_ERR_SUP,"Unsupported MAT_INPLACE_MATRIX"); 350 if (reuse == MAT_REUSE_MATRIX) { 351 /* always destroy the old matrix and create a new memory; 352 hope this does not churn the memory too much. The problem 353 is I do not know if it is possible to put the matrix back to 354 its initial state so that we can directly copy the values 355 the second time through. */ 356 hB = (Mat_HYPRE*)((*B)->data); 357 PetscStackCallStandard(HYPRE_IJMatrixDestroy,(hB->ij)); 358 } else { 359 ierr = MatCreate(comm,B);CHKERRQ(ierr); 360 ierr = MatSetType(*B,MATHYPRE);CHKERRQ(ierr); 361 ierr = MatSetSizes(*B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 362 hB = (Mat_HYPRE*)((*B)->data); 363 } 364 ierr = MatHYPRE_CreateFromMat(A,hB);CHKERRQ(ierr); 365 ierr = MatHYPRE_IJMatrixCopy(A,hB->ij);CHKERRQ(ierr); 366 (*B)->preallocated = PETSC_TRUE; 367 ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 368 ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 369 PetscFunctionReturn(0); 370 } 371 372 #undef __FUNCT__ 373 #define __FUNCT__ "MatConvert_HYPRE_AIJ" 374 static PetscErrorCode MatConvert_HYPRE_AIJ(Mat A, MatType mtype, MatReuse reuse, Mat *B) 375 { 376 Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 377 hypre_ParCSRMatrix *parcsr; 378 hypre_CSRMatrix *hdiag,*hoffd; 379 MPI_Comm comm; 380 PetscScalar *da,*oa,*aptr; 381 PetscInt *dii,*djj,*oii,*ojj,*iptr; 382 PetscInt i,dnnz,onnz,m,n; 383 HYPRE_Int type; 384 PetscMPIInt size; 385 PetscErrorCode ierr; 386 387 PetscFunctionBegin; 388 comm = PetscObjectComm((PetscObject)A); 389 PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type)); 390 if (type != HYPRE_PARCSR) SETERRQ(comm,PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 391 if (reuse == MAT_REUSE_MATRIX) { 392 PetscBool ismpiaij,isseqaij; 393 ierr = PetscObjectTypeCompare((PetscObject)*B,MATMPIAIJ,&ismpiaij);CHKERRQ(ierr); 394 ierr = PetscObjectTypeCompare((PetscObject)*B,MATSEQAIJ,&isseqaij);CHKERRQ(ierr); 395 if (!ismpiaij && !isseqaij) SETERRQ(comm,PETSC_ERR_SUP,"Only MATMPIAIJ or MATSEQAIJ are supported"); 396 } 397 ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 398 399 PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)&parcsr)); 400 hdiag = hypre_ParCSRMatrixDiag(parcsr); 401 hoffd = hypre_ParCSRMatrixOffd(parcsr); 402 m = hypre_CSRMatrixNumRows(hdiag); 403 n = hypre_CSRMatrixNumCols(hdiag); 404 dnnz = hypre_CSRMatrixNumNonzeros(hdiag); 405 onnz = hypre_CSRMatrixNumNonzeros(hoffd); 406 if (reuse == MAT_INITIAL_MATRIX) { 407 ierr = PetscMalloc1(m+1,&dii);CHKERRQ(ierr); 408 ierr = PetscMalloc1(dnnz,&djj);CHKERRQ(ierr); 409 ierr = PetscMalloc1(dnnz,&da);CHKERRQ(ierr); 410 } else if (reuse == MAT_REUSE_MATRIX) { 411 PetscInt nr; 412 PetscBool done; 413 if (size > 1) { 414 Mat_MPIAIJ *b = (Mat_MPIAIJ*)((*B)->data); 415 416 ierr = MatGetRowIJ(b->A,0,PETSC_FALSE,PETSC_FALSE,&nr,(const PetscInt**)&dii,(const PetscInt**)&djj,&done);CHKERRQ(ierr); 417 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); 418 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); 419 ierr = MatSeqAIJGetArray(b->A,&da);CHKERRQ(ierr); 420 } else { 421 ierr = MatGetRowIJ(*B,0,PETSC_FALSE,PETSC_FALSE,&nr,(const PetscInt**)&dii,(const PetscInt**)&djj,&done);CHKERRQ(ierr); 422 if (nr != m) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"Cannot reuse mat: invalid number of local rows! %D != %D",nr,m); 423 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); 424 ierr = MatSeqAIJGetArray(*B,&da);CHKERRQ(ierr); 425 } 426 } else { /* MAT_INPLACE_MATRIX */ 427 dii = (PetscInt*)hypre_CSRMatrixI(hdiag); 428 djj = (PetscInt*)hypre_CSRMatrixJ(hdiag); 429 da = hypre_CSRMatrixData(hdiag); 430 } 431 ierr = PetscMemcpy(dii,hypre_CSRMatrixI(hdiag),(m+1)*sizeof(PetscInt));CHKERRQ(ierr); 432 ierr = PetscMemcpy(djj,hypre_CSRMatrixJ(hdiag),dnnz*sizeof(PetscInt));CHKERRQ(ierr); 433 ierr = PetscMemcpy(da,hypre_CSRMatrixData(hdiag),dnnz*sizeof(PetscScalar));CHKERRQ(ierr); 434 iptr = djj; 435 aptr = da; 436 for (i=0; i<m; i++) { 437 PetscInt nc = dii[i+1]-dii[i]; 438 ierr = PetscSortIntWithScalarArray(nc,iptr,aptr);CHKERRQ(ierr); 439 iptr += nc; 440 aptr += nc; 441 } 442 if (size > 1) { 443 HYPRE_Int *offdj,*coffd; 444 445 if (reuse == MAT_INITIAL_MATRIX) { 446 ierr = PetscMalloc1(m+1,&oii);CHKERRQ(ierr); 447 ierr = PetscMalloc1(onnz,&ojj);CHKERRQ(ierr); 448 ierr = PetscMalloc1(onnz,&oa);CHKERRQ(ierr); 449 } else if (reuse == MAT_REUSE_MATRIX) { 450 Mat_MPIAIJ *b = (Mat_MPIAIJ*)((*B)->data); 451 PetscInt nr,hr = hypre_CSRMatrixNumRows(hoffd); 452 PetscBool done; 453 454 ierr = MatGetRowIJ(b->B,0,PETSC_FALSE,PETSC_FALSE,&nr,(const PetscInt**)&oii,(const PetscInt**)&ojj,&done);CHKERRQ(ierr); 455 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); 456 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); 457 ierr = MatSeqAIJGetArray(b->B,&oa);CHKERRQ(ierr); 458 } else { /* MAT_INPLACE_MATRIX */ 459 oii = (PetscInt*)hypre_CSRMatrixI(hoffd); 460 ojj = (PetscInt*)hypre_CSRMatrixJ(hoffd); 461 oa = hypre_CSRMatrixData(hoffd); 462 } 463 ierr = PetscMemcpy(oii,hypre_CSRMatrixI(hoffd),(m+1)*sizeof(PetscInt));CHKERRQ(ierr); 464 offdj = hypre_CSRMatrixJ(hoffd); 465 coffd = hypre_ParCSRMatrixColMapOffd(parcsr); 466 for (i=0; i<onnz; i++) ojj[i] = coffd[offdj[i]]; 467 ierr = PetscMemcpy(oa,hypre_CSRMatrixData(hoffd),onnz*sizeof(PetscScalar));CHKERRQ(ierr); 468 iptr = ojj; 469 aptr = oa; 470 for (i=0; i<m; i++) { 471 PetscInt nc = oii[i+1]-oii[i]; 472 ierr = PetscSortIntWithScalarArray(nc,iptr,aptr);CHKERRQ(ierr); 473 iptr += nc; 474 aptr += nc; 475 } 476 if (reuse == MAT_INITIAL_MATRIX) { 477 Mat_MPIAIJ *b; 478 Mat_SeqAIJ *d,*o; 479 480 ierr = MatCreateMPIAIJWithSplitArrays(comm,m,n,PETSC_DECIDE,PETSC_DECIDE,dii,djj,da,oii,ojj,oa,B);CHKERRQ(ierr); 481 /* hack MPIAIJ */ 482 b = (Mat_MPIAIJ*)((*B)->data); 483 d = (Mat_SeqAIJ*)b->A->data; 484 o = (Mat_SeqAIJ*)b->B->data; 485 d->free_a = PETSC_TRUE; 486 d->free_ij = PETSC_TRUE; 487 o->free_a = PETSC_TRUE; 488 o->free_ij = PETSC_TRUE; 489 } else if (reuse == MAT_INPLACE_MATRIX) { 490 Mat T; 491 ierr = MatCreateMPIAIJWithSplitArrays(comm,m,n,PETSC_DECIDE,PETSC_DECIDE,dii,djj,da,oii,ojj,oa,&T);CHKERRQ(ierr); 492 hypre_CSRMatrixI(hdiag) = NULL; 493 hypre_CSRMatrixJ(hdiag) = NULL; 494 hypre_CSRMatrixData(hdiag) = NULL; 495 hypre_CSRMatrixI(hoffd) = NULL; 496 hypre_CSRMatrixJ(hoffd) = NULL; 497 hypre_CSRMatrixData(hoffd) = NULL; 498 ierr = MatHeaderReplace(A,&T);CHKERRQ(ierr); 499 } 500 } else { 501 oii = NULL; 502 ojj = NULL; 503 oa = NULL; 504 if (reuse == MAT_INITIAL_MATRIX) { 505 Mat_SeqAIJ* b; 506 ierr = MatCreateSeqAIJWithArrays(comm,m,n,dii,djj,da,B);CHKERRQ(ierr); 507 /* hack SeqAIJ */ 508 b = (Mat_SeqAIJ*)((*B)->data); 509 b->free_a = PETSC_TRUE; 510 b->free_ij = PETSC_TRUE; 511 } else if (reuse == MAT_INPLACE_MATRIX) { 512 Mat T; 513 ierr = MatCreateSeqAIJWithArrays(comm,m,n,dii,djj,da,&T);CHKERRQ(ierr); 514 hypre_CSRMatrixI(hdiag) = NULL; 515 hypre_CSRMatrixJ(hdiag) = NULL; 516 hypre_CSRMatrixData(hdiag) = NULL; 517 ierr = MatHeaderReplace(A,&T);CHKERRQ(ierr); 518 } 519 } 520 521 /* we have to use hypre_Tfree to free the arrays */ 522 if (reuse == MAT_INPLACE_MATRIX) { 523 void *ptrs[6] = {dii,djj,da,oii,ojj,oa}; 524 const char *names[6] = {"_hypre_csr_dii", 525 "_hypre_csr_djj", 526 "_hypre_csr_da", 527 "_hypre_csr_oii", 528 "_hypre_csr_ojj", 529 "_hypre_csr_oa"}; 530 for (i=0; i<6; i++) { 531 PetscContainer c; 532 533 ierr = PetscContainerCreate(comm,&c);CHKERRQ(ierr); 534 ierr = PetscContainerSetPointer(c,ptrs[i]);CHKERRQ(ierr); 535 ierr = PetscContainerSetUserDestroy(c,hypre_array_destroy);CHKERRQ(ierr); 536 ierr = PetscObjectCompose((PetscObject)(*B),names[i],(PetscObject)c);CHKERRQ(ierr); 537 ierr = PetscContainerDestroy(&c);CHKERRQ(ierr); 538 } 539 } 540 PetscFunctionReturn(0); 541 } 542 543 #undef __FUNCT__ 544 #define __FUNCT__ "MatAIJGetParCSR_Private" 545 static PetscErrorCode MatAIJGetParCSR_Private(Mat A, hypre_ParCSRMatrix **hA) 546 { 547 hypre_ParCSRMatrix *tA; 548 hypre_CSRMatrix *hdiag,*hoffd; 549 Mat_SeqAIJ *diag,*offd; 550 PetscInt *garray,noffd,dnnz,onnz,*row_starts,*col_starts; 551 MPI_Comm comm = PetscObjectComm((PetscObject)A); 552 PetscBool ismpiaij,isseqaij; 553 PetscErrorCode ierr; 554 555 PetscFunctionBegin; 556 ierr = PetscObjectTypeCompare((PetscObject)A,MATMPIAIJ,&ismpiaij);CHKERRQ(ierr); 557 ierr = PetscObjectTypeCompare((PetscObject)A,MATSEQAIJ,&isseqaij);CHKERRQ(ierr); 558 if (!ismpiaij && !isseqaij) SETERRQ1(comm,PETSC_ERR_SUP,"Unsupported type %s",((PetscObject)A)->type); 559 if (ismpiaij) { 560 Mat_MPIAIJ *a = (Mat_MPIAIJ*)(A->data); 561 562 diag = (Mat_SeqAIJ*)a->A->data; 563 offd = (Mat_SeqAIJ*)a->B->data; 564 garray = a->garray; 565 noffd = a->B->cmap->N; 566 dnnz = diag->nz; 567 onnz = offd->nz; 568 } else { 569 diag = (Mat_SeqAIJ*)A->data; 570 offd = NULL; 571 garray = NULL; 572 noffd = 0; 573 dnnz = diag->nz; 574 onnz = 0; 575 } 576 577 /* create a temporary ParCSR */ 578 if (HYPRE_AssumedPartitionCheck()) { 579 PetscMPIInt myid; 580 581 ierr = MPI_Comm_rank(comm,&myid);CHKERRQ(ierr); 582 row_starts = A->rmap->range + myid; 583 col_starts = A->cmap->range + myid; 584 } else { 585 row_starts = A->rmap->range; 586 col_starts = A->cmap->range; 587 } 588 tA = hypre_ParCSRMatrixCreate(comm,A->rmap->N,A->cmap->N,(HYPRE_Int*)row_starts,(HYPRE_Int*)col_starts,noffd,dnnz,onnz); 589 hypre_ParCSRMatrixSetRowStartsOwner(tA,0); 590 hypre_ParCSRMatrixSetColStartsOwner(tA,0); 591 592 /* set diagonal part */ 593 hdiag = hypre_ParCSRMatrixDiag(tA); 594 hypre_CSRMatrixI(hdiag) = (HYPRE_Int*)diag->i; 595 hypre_CSRMatrixJ(hdiag) = (HYPRE_Int*)diag->j; 596 hypre_CSRMatrixData(hdiag) = diag->a; 597 hypre_CSRMatrixNumNonzeros(hdiag) = diag->nz; 598 hypre_CSRMatrixSetRownnz(hdiag); 599 hypre_CSRMatrixSetDataOwner(hdiag,0); 600 601 /* set offdiagonal part */ 602 hoffd = hypre_ParCSRMatrixOffd(tA); 603 if (offd) { 604 hypre_CSRMatrixI(hoffd) = (HYPRE_Int*)offd->i; 605 hypre_CSRMatrixJ(hoffd) = (HYPRE_Int*)offd->j; 606 hypre_CSRMatrixData(hoffd) = offd->a; 607 hypre_CSRMatrixNumNonzeros(hoffd) = offd->nz; 608 hypre_CSRMatrixSetRownnz(hoffd); 609 hypre_CSRMatrixSetDataOwner(hoffd,0); 610 hypre_ParCSRMatrixSetNumNonzeros(tA); 611 hypre_ParCSRMatrixColMapOffd(tA) = (HYPRE_Int*)garray; 612 } 613 *hA = tA; 614 PetscFunctionReturn(0); 615 } 616 617 #undef __FUNCT__ 618 #define __FUNCT__ "MatAIJRestoreParCSR_Private" 619 static PetscErrorCode MatAIJRestoreParCSR_Private(Mat A, hypre_ParCSRMatrix **hA) 620 { 621 hypre_CSRMatrix *hdiag,*hoffd; 622 623 PetscFunctionBegin; 624 hdiag = hypre_ParCSRMatrixDiag(*hA); 625 hoffd = hypre_ParCSRMatrixOffd(*hA); 626 /* set pointers to NULL before destroying tA */ 627 hypre_CSRMatrixI(hdiag) = NULL; 628 hypre_CSRMatrixJ(hdiag) = NULL; 629 hypre_CSRMatrixData(hdiag) = NULL; 630 hypre_CSRMatrixI(hoffd) = NULL; 631 hypre_CSRMatrixJ(hoffd) = NULL; 632 hypre_CSRMatrixData(hoffd) = NULL; 633 hypre_ParCSRMatrixColMapOffd(*hA) = NULL; 634 hypre_ParCSRMatrixDestroy(*hA); 635 *hA = NULL; 636 PetscFunctionReturn(0); 637 } 638 639 /* calls RAP from BoomerAMG: 640 the resulting ParCSR will not own the column and row starts */ 641 #undef __FUNCT__ 642 #define __FUNCT__ "MatHYPRE_ParCSR_RAP" 643 static PetscErrorCode MatHYPRE_ParCSR_RAP(hypre_ParCSRMatrix *hR, hypre_ParCSRMatrix *hA, 644 hypre_ParCSRMatrix *hP, hypre_ParCSRMatrix **hRAP) 645 { 646 PetscErrorCode ierr; 647 HYPRE_Int P_owns_col_starts,R_owns_row_starts; 648 649 PetscFunctionBegin; 650 P_owns_col_starts = hypre_ParCSRMatrixOwnsColStarts(hP); 651 R_owns_row_starts = hypre_ParCSRMatrixOwnsRowStarts(hR); 652 PetscStackCallStandard(hypre_BoomerAMGBuildCoarseOperator,(hR,hA,hP,hRAP)); 653 PetscStackCallStandard(hypre_ParCSRMatrixSetNumNonzeros,(*hRAP)); 654 /* hypre_BoomerAMGBuildCoarseOperator steals the col_starts from P and the row_starts from R */ 655 hypre_ParCSRMatrixSetRowStartsOwner(*hRAP,0); 656 hypre_ParCSRMatrixSetColStartsOwner(*hRAP,0); 657 if (P_owns_col_starts) hypre_ParCSRMatrixSetColStartsOwner(hP,1); 658 if (R_owns_row_starts) hypre_ParCSRMatrixSetRowStartsOwner(hR,1); 659 PetscFunctionReturn(0); 660 } 661 662 #undef __FUNCT__ 663 #define __FUNCT__ "MatPtAPNumeric_AIJ_AIJ_wHYPRE" 664 static PetscErrorCode MatPtAPNumeric_AIJ_AIJ_wHYPRE(Mat A,Mat P,Mat C) 665 { 666 Mat B; 667 hypre_ParCSRMatrix *hA,*hP,*hPtAP; 668 PetscErrorCode ierr; 669 670 PetscFunctionBegin; 671 ierr = MatAIJGetParCSR_Private(A,&hA);CHKERRQ(ierr); 672 ierr = MatAIJGetParCSR_Private(P,&hP);CHKERRQ(ierr); 673 ierr = MatHYPRE_ParCSR_RAP(hP,hA,hP,&hPtAP);CHKERRQ(ierr); 674 ierr = MatCreateFromParCSR(hPtAP,MATAIJ,PETSC_OWN_POINTER,&B);CHKERRQ(ierr); 675 ierr = MatHeaderMerge(C,&B);CHKERRQ(ierr); 676 ierr = MatAIJRestoreParCSR_Private(A,&hA);CHKERRQ(ierr); 677 ierr = MatAIJRestoreParCSR_Private(P,&hP);CHKERRQ(ierr); 678 PetscFunctionReturn(0); 679 } 680 681 #undef __FUNCT__ 682 #define __FUNCT__ "MatPtAPSymbolic_AIJ_AIJ_wHYPRE" 683 PETSC_INTERN PetscErrorCode MatPtAPSymbolic_AIJ_AIJ_wHYPRE(Mat A,Mat P,PetscReal fill,Mat *C) 684 { 685 PetscErrorCode ierr; 686 PetscFunctionBegin; 687 ierr = MatCreate(PetscObjectComm((PetscObject)A),C);CHKERRQ(ierr); 688 ierr = MatSetType(*C,MATAIJ);CHKERRQ(ierr); 689 (*C)->ops->ptapnumeric = MatPtAPNumeric_AIJ_AIJ_wHYPRE; 690 PetscFunctionReturn(0); 691 } 692 693 #undef __FUNCT__ 694 #define __FUNCT__ "MatPtAPNumeric_AIJ_HYPRE" 695 static PetscErrorCode MatPtAPNumeric_AIJ_HYPRE(Mat A,Mat P,Mat C) 696 { 697 Mat B; 698 Mat_HYPRE *hP; 699 hypre_ParCSRMatrix *hA = NULL,*Pparcsr,*ptapparcsr; 700 HYPRE_Int type; 701 MPI_Comm comm = PetscObjectComm((PetscObject)A); 702 PetscBool ishypre; 703 PetscErrorCode ierr; 704 705 PetscFunctionBegin; 706 ierr = PetscObjectTypeCompare((PetscObject)P,MATHYPRE,&ishypre);CHKERRQ(ierr); 707 if (!ishypre) SETERRQ1(comm,PETSC_ERR_USER,"P should be of type %s",MATHYPRE); 708 hP = (Mat_HYPRE*)P->data; 709 PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hP->ij,&type)); 710 if (type != HYPRE_PARCSR) SETERRQ(comm,PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 711 PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hP->ij,(void**)&Pparcsr)); 712 713 /* It looks like we don't need to have the diagonal entries 714 ordered first in the rows of the diagonal part 715 for boomerAMGBuildCoarseOperator to work */ 716 ierr = MatAIJGetParCSR_Private(A,&hA);CHKERRQ(ierr); 717 ierr = MatHYPRE_ParCSR_RAP(Pparcsr,hA,Pparcsr,&ptapparcsr);CHKERRQ(ierr); 718 ierr = MatAIJRestoreParCSR_Private(A,&hA);CHKERRQ(ierr); 719 720 /* create temporary matrix and merge to C */ 721 ierr = MatCreateFromParCSR(ptapparcsr,((PetscObject)C)->type_name,PETSC_OWN_POINTER,&B);CHKERRQ(ierr); 722 ierr = MatHeaderMerge(C,&B);CHKERRQ(ierr); 723 PetscFunctionReturn(0); 724 } 725 726 #undef __FUNCT__ 727 #define __FUNCT__ "MatPtAP_AIJ_HYPRE" 728 static PetscErrorCode MatPtAP_AIJ_HYPRE(Mat A,Mat P,MatReuse scall,PetscReal fill,Mat *C) 729 { 730 PetscErrorCode ierr; 731 732 PetscFunctionBegin; 733 if (scall == MAT_INITIAL_MATRIX) { 734 const char *deft = MATAIJ; 735 char type[256]; 736 PetscBool flg; 737 738 ierr = PetscObjectOptionsBegin((PetscObject)A);CHKERRQ(ierr); 739 ierr = PetscOptionsFList("-matptap_hypre_outtype","Matrix type",NULL,MatList,deft,type,256,&flg);CHKERRQ(ierr); 740 ierr = PetscOptionsEnd();CHKERRQ(ierr); 741 ierr = PetscLogEventBegin(MAT_PtAPSymbolic,A,P,0,0);CHKERRQ(ierr); 742 ierr = MatCreate(PetscObjectComm((PetscObject)A),C);CHKERRQ(ierr); 743 if (flg) { 744 ierr = MatSetType(*C,type);CHKERRQ(ierr); 745 } else { 746 ierr = MatSetType(*C,deft);CHKERRQ(ierr); 747 } 748 (*C)->ops->ptapnumeric = MatPtAPNumeric_AIJ_HYPRE; 749 ierr = PetscLogEventEnd(MAT_PtAPSymbolic,A,P,0,0);CHKERRQ(ierr); 750 } 751 ierr = PetscLogEventBegin(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr); 752 ierr = (*(*C)->ops->ptapnumeric)(A,P,*C);CHKERRQ(ierr); 753 ierr = PetscLogEventEnd(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr); 754 PetscFunctionReturn(0); 755 } 756 757 #undef __FUNCT__ 758 #define __FUNCT__ "MatPtAPNumeric_HYPRE_HYPRE" 759 static PetscErrorCode MatPtAPNumeric_HYPRE_HYPRE(Mat A,Mat P,Mat C) 760 { 761 Mat B; 762 hypre_ParCSRMatrix *Aparcsr,*Pparcsr,*ptapparcsr; 763 Mat_HYPRE *hA,*hP; 764 PetscBool ishypre; 765 HYPRE_Int type; 766 PetscErrorCode ierr; 767 768 PetscFunctionBegin; 769 ierr = PetscObjectTypeCompare((PetscObject)P,MATHYPRE,&ishypre);CHKERRQ(ierr); 770 if (!ishypre) SETERRQ1(PetscObjectComm((PetscObject)P),PETSC_ERR_USER,"P should be of type %s",MATHYPRE); 771 ierr = PetscObjectTypeCompare((PetscObject)A,MATHYPRE,&ishypre);CHKERRQ(ierr); 772 if (!ishypre) SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_USER,"A should be of type %s",MATHYPRE); 773 hA = (Mat_HYPRE*)A->data; 774 hP = (Mat_HYPRE*)P->data; 775 PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type)); 776 if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 777 PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hP->ij,&type)); 778 if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)P),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 779 PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)&Aparcsr)); 780 PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hP->ij,(void**)&Pparcsr)); 781 ierr = MatHYPRE_ParCSR_RAP(Pparcsr,Aparcsr,Pparcsr,&ptapparcsr);CHKERRQ(ierr); 782 ierr = MatCreateFromParCSR(ptapparcsr,MATHYPRE,PETSC_OWN_POINTER,&B);CHKERRQ(ierr); 783 ierr = MatHeaderMerge(C,&B);CHKERRQ(ierr); 784 PetscFunctionReturn(0); 785 } 786 787 #undef __FUNCT__ 788 #define __FUNCT__ "MatPtAP_HYPRE_HYPRE" 789 static PetscErrorCode MatPtAP_HYPRE_HYPRE(Mat A,Mat P,MatReuse scall,PetscReal fill,Mat *C) 790 { 791 PetscErrorCode ierr; 792 793 PetscFunctionBegin; 794 if (scall == MAT_INITIAL_MATRIX) { 795 ierr = PetscLogEventBegin(MAT_PtAPSymbolic,A,P,0,0);CHKERRQ(ierr); 796 ierr = MatCreate(PetscObjectComm((PetscObject)A),C);CHKERRQ(ierr); 797 ierr = MatSetType(*C,MATHYPRE);CHKERRQ(ierr); 798 (*C)->ops->ptapnumeric = MatPtAPNumeric_HYPRE_HYPRE; 799 ierr = PetscLogEventEnd(MAT_PtAPSymbolic,A,P,0,0);CHKERRQ(ierr); 800 } 801 ierr = PetscLogEventBegin(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr); 802 ierr = (*(*C)->ops->ptapnumeric)(A,P,*C);CHKERRQ(ierr); 803 ierr = PetscLogEventEnd(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr); 804 PetscFunctionReturn(0); 805 } 806 807 #undef __FUNCT__ 808 #define __FUNCT__ "MatMultTranspose_HYPRE" 809 static PetscErrorCode MatMultTranspose_HYPRE(Mat A, Vec x, Vec y) 810 { 811 PetscErrorCode ierr; 812 813 PetscFunctionBegin; 814 ierr = MatHYPRE_MultKernel_Private(A,x,y,PETSC_TRUE);CHKERRQ(ierr); 815 PetscFunctionReturn(0); 816 } 817 818 #undef __FUNCT__ 819 #define __FUNCT__ "MatMult_HYPRE" 820 static PetscErrorCode MatMult_HYPRE(Mat A, Vec x, Vec y) 821 { 822 PetscErrorCode ierr; 823 824 PetscFunctionBegin; 825 ierr = MatHYPRE_MultKernel_Private(A,x,y,PETSC_FALSE);CHKERRQ(ierr); 826 PetscFunctionReturn(0); 827 } 828 829 #undef __FUNCT__ 830 #define __FUNCT__ "MatHYPRE_MultKernel_Private" 831 static PetscErrorCode MatHYPRE_MultKernel_Private(Mat A, Vec x, Vec y, PetscBool trans) 832 { 833 Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 834 hypre_ParCSRMatrix *parcsr; 835 hypre_ParVector *hx,*hy; 836 PetscScalar *ax,*ay,*sax,*say; 837 PetscErrorCode ierr; 838 839 PetscFunctionBegin; 840 PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)&parcsr)); 841 PetscStackCallStandard(HYPRE_IJVectorGetObject,(hA->x,(void**)&hx)); 842 PetscStackCallStandard(HYPRE_IJVectorGetObject,(hA->b,(void**)&hy)); 843 ierr = VecGetArrayRead(x,(const PetscScalar**)&ax);CHKERRQ(ierr); 844 ierr = VecGetArray(y,&ay);CHKERRQ(ierr); 845 if (trans) { 846 VecHYPRE_ParVectorReplacePointer(hA->x,ay,say); 847 VecHYPRE_ParVectorReplacePointer(hA->b,ax,sax); 848 hypre_ParCSRMatrixMatvecT(1.,parcsr,hy,0.,hx); 849 VecHYPRE_ParVectorReplacePointer(hA->x,say,ay); 850 VecHYPRE_ParVectorReplacePointer(hA->b,sax,ax); 851 } else { 852 VecHYPRE_ParVectorReplacePointer(hA->x,ax,sax); 853 VecHYPRE_ParVectorReplacePointer(hA->b,ay,say); 854 hypre_ParCSRMatrixMatvec(1.,parcsr,hx,0.,hy); 855 VecHYPRE_ParVectorReplacePointer(hA->x,sax,ax); 856 VecHYPRE_ParVectorReplacePointer(hA->b,say,ay); 857 } 858 ierr = VecRestoreArrayRead(x,(const PetscScalar**)&ax);CHKERRQ(ierr); 859 ierr = VecRestoreArray(y,&ay);CHKERRQ(ierr); 860 PetscFunctionReturn(0); 861 } 862 863 #undef __FUNCT__ 864 #define __FUNCT__ "MatDestroy_HYPRE" 865 static PetscErrorCode MatDestroy_HYPRE(Mat A) 866 { 867 Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 868 PetscErrorCode ierr; 869 870 PetscFunctionBegin; 871 if (hA->x) PetscStackCallStandard(HYPRE_IJVectorDestroy,(hA->x)); 872 if (hA->b) PetscStackCallStandard(HYPRE_IJVectorDestroy,(hA->b)); 873 if (hA->ij) { 874 if (!hA->inner_free) hypre_IJMatrixObject(hA->ij) = NULL; 875 PetscStackCallStandard(HYPRE_IJMatrixDestroy,(hA->ij)); 876 } 877 if (hA->comm) { ierr = MPI_Comm_free(&hA->comm);CHKERRQ(ierr); } 878 ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_hypre_aij_C",NULL);CHKERRQ(ierr); 879 ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_hypre_is_C",NULL);CHKERRQ(ierr); 880 ierr = PetscObjectComposeFunction((PetscObject)A,"MatPtAP_seqaij_hypre_C",NULL);CHKERRQ(ierr); 881 ierr = PetscObjectComposeFunction((PetscObject)A,"MatPtAP_mpiaij_hypre_C",NULL);CHKERRQ(ierr); 882 ierr = PetscObjectComposeFunction((PetscObject)A,"MatHYPRESetPreallocation_C",NULL);CHKERRQ(ierr); 883 ierr = PetscObjectComposeFunction((PetscObject)A,"MatHYPREGetParCSR_C",NULL);CHKERRQ(ierr); 884 ierr = PetscFree(A->data);CHKERRQ(ierr); 885 PetscFunctionReturn(0); 886 } 887 888 #undef __FUNCT__ 889 #define __FUNCT__ "MatSetUp_HYPRE" 890 static PetscErrorCode MatSetUp_HYPRE(Mat A) 891 { 892 PetscErrorCode ierr; 893 894 PetscFunctionBegin; 895 ierr = MatHYPRESetPreallocation(A,PETSC_DEFAULT,NULL,PETSC_DEFAULT,NULL);CHKERRQ(ierr); 896 PetscFunctionReturn(0); 897 } 898 899 #undef __FUNCT__ 900 #define __FUNCT__ "MatAssemblyEnd_HYPRE" 901 static PetscErrorCode MatAssemblyEnd_HYPRE(Mat A, MatAssemblyType mode) 902 { 903 Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 904 Vec x,b; 905 PetscErrorCode ierr; 906 907 PetscFunctionBegin; 908 if (mode == MAT_FLUSH_ASSEMBLY) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MAT_FLUSH_ASSEMBLY currently not supported with MATHYPRE"); 909 PetscStackCallStandard(HYPRE_IJMatrixAssemble,(hA->ij)); 910 if (hA->x) PetscFunctionReturn(0); 911 ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 912 ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 913 ierr = VecCreateMPIWithArray(PetscObjectComm((PetscObject)A),1,A->cmap->n,A->cmap->N,NULL,&x);CHKERRQ(ierr); 914 ierr = VecCreateMPIWithArray(PetscObjectComm((PetscObject)A),1,A->rmap->n,A->rmap->N,NULL,&b);CHKERRQ(ierr); 915 ierr = VecHYPRE_IJVectorCreate(x,&hA->x);CHKERRQ(ierr); 916 ierr = VecHYPRE_IJVectorCreate(b,&hA->b);CHKERRQ(ierr); 917 ierr = VecDestroy(&x);CHKERRQ(ierr); 918 ierr = VecDestroy(&b);CHKERRQ(ierr); 919 PetscFunctionReturn(0); 920 } 921 922 #define MATHYPRE_SCRATCH 2048 923 924 #undef __FUNCT__ 925 #define __FUNCT__ "MatSetValues_HYPRE" 926 PetscErrorCode MatSetValues_HYPRE(Mat A, PetscInt nr, const PetscInt rows[], PetscInt nc, const PetscInt cols[], const PetscScalar v[], InsertMode ins) 927 { 928 Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 929 PetscScalar *vals = (PetscScalar *)v; 930 PetscScalar sscr[MATHYPRE_SCRATCH]; 931 HYPRE_Int cscr[2][MATHYPRE_SCRATCH]; 932 HYPRE_Int i,nzc; 933 PetscErrorCode ierr; 934 935 PetscFunctionBegin; 936 for (i=0,nzc=0;i<nc;i++) { 937 if (cols[i] >= 0) { 938 cscr[0][nzc ] = cols[i]; 939 cscr[1][nzc++] = i; 940 } 941 } 942 if (!nzc) PetscFunctionReturn(0); 943 944 if (ins == ADD_VALUES) { 945 for (i=0;i<nr;i++) { 946 if (rows[i] >= 0) { 947 PetscInt j; 948 for (j=0;j<nzc;j++) sscr[j] = vals[cscr[1][j]]; 949 PetscStackCallStandard(HYPRE_IJMatrixAddToValues,(hA->ij,1,&nzc,(HYPRE_Int*)(rows+i),cscr[0],sscr)); 950 } 951 vals += nc; 952 } 953 } else { /* INSERT_VALUES */ 954 #if defined(PETSC_USE_DEBUG) 955 /* Insert values cannot be used to insert offproc entries */ 956 PetscInt rst,ren; 957 ierr = MatGetOwnershipRange(A,&rst,&ren);CHKERRQ(ierr); 958 for (i=0;i<nr;i++) 959 if (rows[i] >= 0 && (rows[i] < rst || rows[i] >= ren)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot use INSERT_VALUES for off-proc entries with MatHYPRE. Use ADD_VALUES instead"); 960 #endif 961 for (i=0;i<nr;i++) { 962 if (rows[i] >= 0) { 963 PetscInt j; 964 for (j=0;j<nzc;j++) sscr[j] = vals[cscr[1][j]]; 965 PetscStackCallStandard(HYPRE_IJMatrixSetValues,(hA->ij,1,&nzc,(HYPRE_Int*)(rows+i),cscr[0],sscr)); 966 } 967 vals += nc; 968 } 969 } 970 PetscFunctionReturn(0); 971 } 972 973 #undef __FUNCT__ 974 #define __FUNCT__ "MatHYPRESetPreallocation_HYPRE" 975 static PetscErrorCode MatHYPRESetPreallocation_HYPRE(Mat A, PetscInt dnz, const PetscInt dnnz[], PetscInt onz, const PetscInt onnz[]) 976 { 977 Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 978 HYPRE_Int *hdnnz,*honnz; 979 PetscInt i,rs,re,cs,ce,bs; 980 PetscMPIInt size; 981 PetscErrorCode ierr; 982 983 PetscFunctionBegin; 984 ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr); 985 ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 986 ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 987 rs = A->rmap->rstart; 988 re = A->rmap->rend; 989 cs = A->cmap->rstart; 990 ce = A->cmap->rend; 991 if (!hA->ij) { 992 PetscStackCallStandard(HYPRE_IJMatrixCreate,(hA->comm,rs,re-1,cs,ce-1,&hA->ij)); 993 PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(hA->ij,HYPRE_PARCSR)); 994 } else { 995 HYPRE_Int hrs,hre,hcs,hce; 996 PetscStackCallStandard(HYPRE_IJMatrixGetLocalRange,(hA->ij,&hrs,&hre,&hcs,&hce)); 997 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); 998 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); 999 } 1000 PetscStackCallStandard(HYPRE_IJMatrixInitialize,(hA->ij)); 1001 1002 if (dnz == PETSC_DEFAULT || dnz == PETSC_DECIDE) dnz = 10*bs; 1003 if (onz == PETSC_DEFAULT || onz == PETSC_DECIDE) onz = 10*bs; 1004 1005 if (!dnnz) { 1006 ierr = PetscMalloc1(A->rmap->n,&hdnnz);CHKERRQ(ierr); 1007 for (i=0;i<A->rmap->n;i++) hdnnz[i] = dnz; 1008 } else { 1009 hdnnz = (HYPRE_Int*)dnnz; 1010 } 1011 ierr = MPI_Comm_size(PetscObjectComm((PetscObject)A),&size);CHKERRQ(ierr); 1012 if (size > 1) { 1013 if (!onnz) { 1014 ierr = PetscMalloc1(A->rmap->n,&honnz);CHKERRQ(ierr); 1015 for (i=0;i<A->rmap->n;i++) honnz[i] = onz; 1016 } else { 1017 honnz = (HYPRE_Int*)onnz; 1018 } 1019 PetscStackCallStandard(HYPRE_IJMatrixSetDiagOffdSizes,(hA->ij,hdnnz,honnz)); 1020 } else { 1021 honnz = NULL; 1022 PetscStackCallStandard(HYPRE_IJMatrixSetRowSizes,(hA->ij,hdnnz)); 1023 } 1024 if (!dnnz) { 1025 ierr = PetscFree(hdnnz);CHKERRQ(ierr); 1026 } 1027 if (!onnz && honnz) { 1028 ierr = PetscFree(honnz);CHKERRQ(ierr); 1029 } 1030 A->preallocated = PETSC_TRUE; 1031 1032 /* SetDiagOffdSizes sets hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 0 */ 1033 { 1034 hypre_AuxParCSRMatrix *aux_matrix; 1035 aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(hA->ij); 1036 hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 1; 1037 } 1038 PetscFunctionReturn(0); 1039 } 1040 1041 /*@C 1042 MatHYPRESetPreallocation - Preallocates memory for a sparse parallel matrix in HYPRE IJ format 1043 1044 Collective on Mat 1045 1046 Input Parameters: 1047 + A - the matrix 1048 . dnz - number of nonzeros per row in DIAGONAL portion of local submatrix 1049 (same value is used for all local rows) 1050 . dnnz - array containing the number of nonzeros in the various rows of the 1051 DIAGONAL portion of the local submatrix (possibly different for each row) 1052 or NULL (PETSC_NULL_INTEGER in Fortran), if d_nz is used to specify the nonzero structure. 1053 The size of this array is equal to the number of local rows, i.e 'm'. 1054 For matrices that will be factored, you must leave room for (and set) 1055 the diagonal entry even if it is zero. 1056 . onz - number of nonzeros per row in the OFF-DIAGONAL portion of local 1057 submatrix (same value is used for all local rows). 1058 - onnz - array containing the number of nonzeros in the various rows of the 1059 OFF-DIAGONAL portion of the local submatrix (possibly different for 1060 each row) or NULL (PETSC_NULL_INTEGER in Fortran), if o_nz is used to specify the nonzero 1061 structure. The size of this array is equal to the number 1062 of local rows, i.e 'm'. 1063 1064 Notes: If the *nnz parameter is given then the *nz parameter is ignored; for sequential matrices, onz and onnz are ignored. 1065 1066 Level: intermediate 1067 1068 .keywords: matrix, aij, compressed row, sparse, parallel 1069 1070 .seealso: MatCreate(), MatMPIAIJSetPreallocation, MATHYPRE 1071 @*/ 1072 #undef __FUNCT__ 1073 #define __FUNCT__ "MatHYPRESetPreallocation" 1074 PetscErrorCode MatHYPRESetPreallocation(Mat A, PetscInt dnz, const PetscInt dnnz[], PetscInt onz, const PetscInt onnz[]) 1075 { 1076 PetscErrorCode ierr; 1077 1078 PetscFunctionBegin; 1079 PetscValidHeaderSpecific(A,MAT_CLASSID,1); 1080 PetscValidType(A,1); 1081 ierr = PetscTryMethod(A,"MatHYPRESetPreallocation_C",(Mat,PetscInt,const PetscInt[],PetscInt,const PetscInt[]),(A,dnz,dnnz,onz,onnz));CHKERRQ(ierr); 1082 PetscFunctionReturn(0); 1083 } 1084 1085 /* 1086 MatCreateFromParCSR - Creates a matrix from a hypre_ParCSRMatrix 1087 1088 Collective 1089 1090 Input Parameters: 1091 + vparcsr - the pointer to the hypre_ParCSRMatrix 1092 . mtype - matrix type to be created. Currently MATAIJ, MATIS and MATHYPRE are supported. 1093 - copymode - PETSc copying options 1094 1095 Output Parameter: 1096 . A - the matrix 1097 1098 Level: intermediate 1099 1100 .seealso: MatHYPRE, PetscCopyMode 1101 */ 1102 #undef __FUNCT__ 1103 #define __FUNCT__ "MatCreateFromParCSR" 1104 PETSC_EXTERN PetscErrorCode MatCreateFromParCSR(hypre_ParCSRMatrix *vparcsr, MatType mtype, PetscCopyMode copymode, Mat* A) 1105 { 1106 Mat T; 1107 Mat_HYPRE *hA; 1108 hypre_ParCSRMatrix *parcsr; 1109 MPI_Comm comm; 1110 PetscInt rstart,rend,cstart,cend,M,N; 1111 PetscBool isseqaij,ismpiaij,isaij,ishyp,isis; 1112 PetscErrorCode ierr; 1113 1114 PetscFunctionBegin; 1115 parcsr = (hypre_ParCSRMatrix *)vparcsr; 1116 comm = hypre_ParCSRMatrixComm(parcsr); 1117 ierr = PetscStrcmp(mtype,MATSEQAIJ,&isseqaij);CHKERRQ(ierr); 1118 ierr = PetscStrcmp(mtype,MATMPIAIJ,&ismpiaij);CHKERRQ(ierr); 1119 ierr = PetscStrcmp(mtype,MATAIJ,&isaij);CHKERRQ(ierr); 1120 ierr = PetscStrcmp(mtype,MATHYPRE,&ishyp);CHKERRQ(ierr); 1121 ierr = PetscStrcmp(mtype,MATIS,&isis);CHKERRQ(ierr); 1122 isaij = (PetscBool)(isseqaij || ismpiaij || isaij); 1123 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); 1124 if (ishyp && copymode == PETSC_COPY_VALUES) SETERRQ(comm,PETSC_ERR_SUP,"Unsupported copymode PETSC_COPY_VALUES"); 1125 1126 /* access ParCSRMatrix */ 1127 rstart = hypre_ParCSRMatrixFirstRowIndex(parcsr); 1128 rend = hypre_ParCSRMatrixLastRowIndex(parcsr); 1129 cstart = hypre_ParCSRMatrixFirstColDiag(parcsr); 1130 cend = hypre_ParCSRMatrixLastColDiag(parcsr); 1131 M = hypre_ParCSRMatrixGlobalNumRows(parcsr); 1132 N = hypre_ParCSRMatrixGlobalNumCols(parcsr); 1133 1134 /* fix for empty local rows/columns */ 1135 if (rend < rstart) rend = rstart; 1136 if (cend < cstart) cend = cstart; 1137 1138 /* create PETSc matrix with MatHYPRE */ 1139 ierr = MatCreate(comm,&T);CHKERRQ(ierr); 1140 ierr = MatSetSizes(T,rend-rstart+1,cend-cstart+1,M,N);CHKERRQ(ierr); 1141 ierr = MatSetType(T,MATHYPRE);CHKERRQ(ierr); 1142 hA = (Mat_HYPRE*)(T->data); 1143 1144 /* create HYPRE_IJMatrix */ 1145 PetscStackCallStandard(HYPRE_IJMatrixCreate,(hA->comm,rstart,rend-1,cstart,cend-1,&hA->ij)); 1146 1147 /* set ParCSR object */ 1148 PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(hA->ij,HYPRE_PARCSR)); 1149 hypre_IJMatrixObject(hA->ij) = parcsr; 1150 T->preallocated = PETSC_TRUE; 1151 1152 /* set assembled flag */ 1153 hypre_IJMatrixAssembleFlag(hA->ij) = 1; 1154 PetscStackCallStandard(HYPRE_IJMatrixInitialize,(hA->ij)); 1155 if (ishyp) { 1156 PetscMPIInt myid = 0; 1157 1158 /* make sure we always have row_starts and col_starts available */ 1159 if (HYPRE_AssumedPartitionCheck()) { 1160 ierr = MPI_Comm_rank(comm,&myid);CHKERRQ(ierr); 1161 } 1162 if (!hypre_ParCSRMatrixOwnsColStarts(parcsr)) { 1163 PetscLayout map; 1164 1165 ierr = MatGetLayouts(T,NULL,&map);CHKERRQ(ierr); 1166 ierr = PetscLayoutSetUp(map);CHKERRQ(ierr); 1167 hypre_ParCSRMatrixColStarts(parcsr) = map->range + myid; 1168 } 1169 if (!hypre_ParCSRMatrixOwnsRowStarts(parcsr)) { 1170 PetscLayout map; 1171 1172 ierr = MatGetLayouts(T,&map,NULL);CHKERRQ(ierr); 1173 ierr = PetscLayoutSetUp(map);CHKERRQ(ierr); 1174 hypre_ParCSRMatrixRowStarts(parcsr) = map->range + myid; 1175 } 1176 /* prevent from freeing the pointer */ 1177 if (copymode == PETSC_USE_POINTER) hA->inner_free = PETSC_FALSE; 1178 *A = T; 1179 ierr = MatAssemblyBegin(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1180 ierr = MatAssemblyEnd(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1181 } else if (isaij) { 1182 if (copymode != PETSC_OWN_POINTER) { 1183 /* prevent from freeing the pointer */ 1184 hA->inner_free = PETSC_FALSE; 1185 ierr = MatConvert_HYPRE_AIJ(T,MATAIJ,MAT_INITIAL_MATRIX,A);CHKERRQ(ierr); 1186 ierr = MatDestroy(&T);CHKERRQ(ierr); 1187 } else { /* AIJ return type with PETSC_OWN_POINTER */ 1188 ierr = MatConvert_HYPRE_AIJ(T,MATAIJ,MAT_INPLACE_MATRIX,&T);CHKERRQ(ierr); 1189 *A = T; 1190 } 1191 } else if (isis) { 1192 ierr = MatConvert_HYPRE_IS(T,MATIS,MAT_INITIAL_MATRIX,A);CHKERRQ(ierr); 1193 if (copymode != PETSC_OWN_POINTER) hA->inner_free = PETSC_FALSE; 1194 ierr = MatDestroy(&T);CHKERRQ(ierr); 1195 } 1196 PetscFunctionReturn(0); 1197 } 1198 1199 #undef __FUNCT__ 1200 #define __FUNCT__ "MatHYPREGetParCSR_HYPRE" 1201 PetscErrorCode MatHYPREGetParCSR_HYPRE(Mat A, hypre_ParCSRMatrix **parcsr) 1202 { 1203 Mat_HYPRE* hA = (Mat_HYPRE*)A->data; 1204 HYPRE_Int type; 1205 PetscErrorCode ierr; 1206 1207 PetscFunctionBegin; 1208 if (!hA->ij) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"HYPRE_IJMatrix not present"); 1209 PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type)); 1210 if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"HYPRE_IJMatrix is not of type HYPRE_PARCSR"); 1211 PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)parcsr)); 1212 PetscFunctionReturn(0); 1213 } 1214 1215 /* 1216 MatHYPREGetParCSR - Gets the pointer to the ParCSR matrix 1217 1218 Not collective 1219 1220 Input Parameters: 1221 + A - the MATHYPRE object 1222 1223 Output Parameter: 1224 . parcsr - the pointer to the hypre_ParCSRMatrix 1225 1226 Level: intermediate 1227 1228 .seealso: MatHYPRE, PetscCopyMode 1229 */ 1230 #undef __FUNCT__ 1231 #define __FUNCT__ "MatHYPREGetParCSR" 1232 PetscErrorCode MatHYPREGetParCSR(Mat A, hypre_ParCSRMatrix **parcsr) 1233 { 1234 PetscErrorCode ierr; 1235 1236 PetscFunctionBegin; 1237 PetscValidHeaderSpecific(A,MAT_CLASSID,1); 1238 PetscValidType(A,1); 1239 ierr = PetscUseMethod(A,"MatHYPREGetParCSR_C",(Mat,hypre_ParCSRMatrix**),(A,parcsr));CHKERRQ(ierr); 1240 PetscFunctionReturn(0); 1241 } 1242 1243 #undef __FUNCT__ 1244 #define __FUNCT__ "MatCreate_HYPRE" 1245 PETSC_EXTERN PetscErrorCode MatCreate_HYPRE(Mat B) 1246 { 1247 Mat_HYPRE *hB; 1248 PetscErrorCode ierr; 1249 1250 PetscFunctionBegin; 1251 ierr = PetscNewLog(B,&hB);CHKERRQ(ierr); 1252 hB->inner_free = PETSC_TRUE; 1253 1254 B->data = (void*)hB; 1255 B->rmap->bs = 1; 1256 B->assembled = PETSC_FALSE; 1257 1258 B->ops->mult = MatMult_HYPRE; 1259 B->ops->multtranspose = MatMultTranspose_HYPRE; 1260 B->ops->setup = MatSetUp_HYPRE; 1261 B->ops->destroy = MatDestroy_HYPRE; 1262 B->ops->assemblyend = MatAssemblyEnd_HYPRE; 1263 B->ops->ptap = MatPtAP_HYPRE_HYPRE; 1264 B->ops->setvalues = MatSetValues_HYPRE; 1265 1266 ierr = MPI_Comm_dup(PetscObjectComm((PetscObject)B),&hB->comm);CHKERRQ(ierr); 1267 ierr = PetscObjectChangeTypeName((PetscObject)B,MATHYPRE);CHKERRQ(ierr); 1268 ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_hypre_aij_C",MatConvert_HYPRE_AIJ);CHKERRQ(ierr); 1269 ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_hypre_is_C",MatConvert_HYPRE_IS);CHKERRQ(ierr); 1270 ierr = PetscObjectComposeFunction((PetscObject)B,"MatPtAP_seqaij_hypre_C",MatPtAP_AIJ_HYPRE);CHKERRQ(ierr); 1271 ierr = PetscObjectComposeFunction((PetscObject)B,"MatPtAP_mpiaij_hypre_C",MatPtAP_AIJ_HYPRE);CHKERRQ(ierr); 1272 ierr = PetscObjectComposeFunction((PetscObject)B,"MatHYPRESetPreallocation_C",MatHYPRESetPreallocation_HYPRE);CHKERRQ(ierr); 1273 ierr = PetscObjectComposeFunction((PetscObject)B,"MatHYPREGetParCSR_C",MatHYPREGetParCSR_HYPRE);CHKERRQ(ierr); 1274 PetscFunctionReturn(0); 1275 } 1276 1277 #undef __FUNCT__ 1278 #define __FUNCT__ "hypre_array_destroy" 1279 static PetscErrorCode hypre_array_destroy(void *ptr) 1280 { 1281 PetscFunctionBegin; 1282 hypre_TFree(ptr); 1283 PetscFunctionReturn(0); 1284 } 1285 1286 #if 0 1287 /* 1288 Does NOT copy the data over, instead uses DIRECTLY the pointers from the PETSc MPIAIJ format 1289 1290 This is UNFINISHED and does NOT work! The problem is that hypre puts the diagonal entry first 1291 which will corrupt the PETSc data structure if we did this. Need a work around to this problem. 1292 */ 1293 #include <_hypre_IJ_mv.h> 1294 #include <HYPRE_IJ_mv.h> 1295 1296 #undef __FUNCT__ 1297 #define __FUNCT__ "MatHYPRE_IJMatrixLink" 1298 PetscErrorCode MatHYPRE_IJMatrixLink(Mat A, HYPRE_IJMatrix *ij) 1299 { 1300 PetscErrorCode ierr; 1301 PetscInt rstart,rend,cstart,cend; 1302 PetscBool flg; 1303 hypre_AuxParCSRMatrix *aux_matrix; 1304 1305 PetscFunctionBegin; 1306 PetscValidHeaderSpecific(A,MAT_CLASSID,1); 1307 PetscValidType(A,1); 1308 PetscValidPointer(ij,2); 1309 ierr = PetscObjectTypeCompare((PetscObject)A,MATMPIAIJ,&flg);CHKERRQ(ierr); 1310 if (!flg) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Can only use with PETSc MPIAIJ matrices"); 1311 ierr = MatSetUp(A);CHKERRQ(ierr); 1312 1313 ierr = PetscLogEventBegin(MAT_Convert,A,0,0,0);CHKERRQ(ierr); 1314 rstart = A->rmap->rstart; 1315 rend = A->rmap->rend; 1316 cstart = A->cmap->rstart; 1317 cend = A->cmap->rend; 1318 PetscStackCallStandard(HYPRE_IJMatrixCreate,(PetscObjectComm((PetscObject)A),rstart,rend-1,cstart,cend-1,ij)); 1319 PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(*ij,HYPRE_PARCSR)); 1320 1321 PetscStackCallStandard(HYPRE_IJMatrixInitialize,(*ij)); 1322 PetscStackCall("hypre_IJMatrixTranslator",aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(*ij)); 1323 1324 hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 0; 1325 1326 /* this is the Hack part where we monkey directly with the hypre datastructures */ 1327 1328 PetscStackCallStandard(HYPRE_IJMatrixAssemble,(*ij)); 1329 ierr = PetscLogEventEnd(MAT_Convert,A,0,0,0);CHKERRQ(ierr); 1330 PetscFunctionReturn(0); 1331 } 1332 #endif 1333