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__ "MatPtAP_AIJ_HYPRE" 695 static PetscErrorCode MatPtAP_AIJ_HYPRE(Mat A,Mat P,MatReuse scall,PetscReal fill,Mat *C) 696 { 697 Mat_HYPRE *hP = (Mat_HYPRE*)P->data; 698 hypre_ParCSRMatrix *hA = NULL,*Pparcsr,*ptapparcsr; 699 HYPRE_Int type; 700 MPI_Comm comm = PetscObjectComm((PetscObject)A); 701 char mtype[256]; 702 PetscErrorCode ierr; 703 704 PetscFunctionBegin; 705 PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hP->ij,&type)); 706 if (type != HYPRE_PARCSR) SETERRQ(comm,PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 707 if (scall == MAT_REUSE_MATRIX) SETERRQ(comm,PETSC_ERR_SUP,"Unsupported MAT_REUSE_MATRIX"); 708 ierr = PetscLogEventBegin(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr); 709 PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hP->ij,(void**)&Pparcsr)); 710 711 /* It looks like we don't need to have the diagonal entries 712 ordered first in the rows of the diagonal part 713 for boomerAMGBuildCoarseOperator to work */ 714 ierr = MatAIJGetParCSR_Private(A,&hA);CHKERRQ(ierr); 715 ierr = MatHYPRE_ParCSR_RAP(Pparcsr,hA,Pparcsr,&ptapparcsr);CHKERRQ(ierr); 716 ierr = MatAIJRestoreParCSR_Private(A,&hA);CHKERRQ(ierr); 717 718 /* create C depending on mtype */ 719 sprintf(mtype,MATAIJ); 720 ierr = PetscOptionsGetString(((PetscObject)A)->options,((PetscObject)A)->prefix,"-matptap_hypre_outtype",mtype,256,NULL);CHKERRQ(ierr); 721 ierr = MatCreateFromParCSR(ptapparcsr,mtype,PETSC_OWN_POINTER,C);CHKERRQ(ierr); 722 ierr = PetscLogEventEnd(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr); 723 PetscFunctionReturn(0); 724 } 725 726 #undef __FUNCT__ 727 #define __FUNCT__ "MatPtAP_HYPRE_HYPRE" 728 static PetscErrorCode MatPtAP_HYPRE_HYPRE(Mat A,Mat P,MatReuse scall,PetscReal fill,Mat *C) 729 { 730 hypre_ParCSRMatrix *Aparcsr,*Pparcsr,*ptapparcsr; 731 Mat_HYPRE *hA = (Mat_HYPRE*)A->data, *hP = (Mat_HYPRE*)P->data; 732 HYPRE_Int type; 733 PetscErrorCode ierr; 734 735 PetscFunctionBegin; 736 if (scall == MAT_REUSE_MATRIX) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Unsupported MAT_REUSE_MATRIX"); 737 PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type)); 738 if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 739 PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hP->ij,&type)); 740 if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)P),PETSC_ERR_SUP,"Only HYPRE_PARCSR is supported"); 741 ierr = PetscLogEventBegin(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr); 742 PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)&Aparcsr)); 743 PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hP->ij,(void**)&Pparcsr)); 744 ierr = MatHYPRE_ParCSR_RAP(Pparcsr,Aparcsr,Pparcsr,&ptapparcsr);CHKERRQ(ierr); 745 ierr = MatCreateFromParCSR(ptapparcsr,MATHYPRE,PETSC_OWN_POINTER,C);CHKERRQ(ierr); 746 ierr = PetscLogEventEnd(MAT_PtAPNumeric,A,P,0,0);CHKERRQ(ierr); 747 PetscFunctionReturn(0); 748 } 749 750 #undef __FUNCT__ 751 #define __FUNCT__ "MatMultTranspose_HYPRE" 752 static PetscErrorCode MatMultTranspose_HYPRE(Mat A, Vec x, Vec y) 753 { 754 PetscErrorCode ierr; 755 756 PetscFunctionBegin; 757 ierr = MatHYPRE_MultKernel_Private(A,x,y,PETSC_TRUE);CHKERRQ(ierr); 758 PetscFunctionReturn(0); 759 } 760 761 #undef __FUNCT__ 762 #define __FUNCT__ "MatMult_HYPRE" 763 static PetscErrorCode MatMult_HYPRE(Mat A, Vec x, Vec y) 764 { 765 PetscErrorCode ierr; 766 767 PetscFunctionBegin; 768 ierr = MatHYPRE_MultKernel_Private(A,x,y,PETSC_FALSE);CHKERRQ(ierr); 769 PetscFunctionReturn(0); 770 } 771 772 #undef __FUNCT__ 773 #define __FUNCT__ "MatHYPRE_MultKernel_Private" 774 static PetscErrorCode MatHYPRE_MultKernel_Private(Mat A, Vec x, Vec y, PetscBool trans) 775 { 776 Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 777 hypre_ParCSRMatrix *parcsr; 778 hypre_ParVector *hx,*hy; 779 PetscScalar *ax,*ay,*sax,*say; 780 PetscErrorCode ierr; 781 782 PetscFunctionBegin; 783 PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)&parcsr)); 784 PetscStackCallStandard(HYPRE_IJVectorGetObject,(hA->x,(void**)&hx)); 785 PetscStackCallStandard(HYPRE_IJVectorGetObject,(hA->b,(void**)&hy)); 786 ierr = VecGetArrayRead(x,(const PetscScalar**)&ax);CHKERRQ(ierr); 787 ierr = VecGetArray(y,&ay);CHKERRQ(ierr); 788 if (trans) { 789 VecHYPRE_ParVectorReplacePointer(hA->x,ay,say); 790 VecHYPRE_ParVectorReplacePointer(hA->b,ax,sax); 791 hypre_ParCSRMatrixMatvecT(1.,parcsr,hy,0.,hx); 792 VecHYPRE_ParVectorReplacePointer(hA->x,say,ay); 793 VecHYPRE_ParVectorReplacePointer(hA->b,sax,ax); 794 } else { 795 VecHYPRE_ParVectorReplacePointer(hA->x,ax,sax); 796 VecHYPRE_ParVectorReplacePointer(hA->b,ay,say); 797 hypre_ParCSRMatrixMatvec(1.,parcsr,hx,0.,hy); 798 VecHYPRE_ParVectorReplacePointer(hA->x,sax,ax); 799 VecHYPRE_ParVectorReplacePointer(hA->b,say,ay); 800 } 801 ierr = VecRestoreArrayRead(x,(const PetscScalar**)&ax);CHKERRQ(ierr); 802 ierr = VecRestoreArray(y,&ay);CHKERRQ(ierr); 803 PetscFunctionReturn(0); 804 } 805 806 #undef __FUNCT__ 807 #define __FUNCT__ "MatDestroy_HYPRE" 808 static PetscErrorCode MatDestroy_HYPRE(Mat A) 809 { 810 Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 811 PetscErrorCode ierr; 812 813 PetscFunctionBegin; 814 if (hA->x) PetscStackCallStandard(HYPRE_IJVectorDestroy,(hA->x)); 815 if (hA->b) PetscStackCallStandard(HYPRE_IJVectorDestroy,(hA->b)); 816 if (hA->ij) { 817 if (!hA->inner_free) hypre_IJMatrixObject(hA->ij) = NULL; 818 PetscStackCallStandard(HYPRE_IJMatrixDestroy,(hA->ij)); 819 } 820 if (hA->comm) { ierr = MPI_Comm_free(&hA->comm);CHKERRQ(ierr); } 821 ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_hypre_aij_C",NULL);CHKERRQ(ierr); 822 ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_hypre_is_C",NULL);CHKERRQ(ierr); 823 ierr = PetscObjectComposeFunction((PetscObject)A,"MatPtAP_seqaij_hypre_C",NULL);CHKERRQ(ierr); 824 ierr = PetscObjectComposeFunction((PetscObject)A,"MatPtAP_mpiaij_hypre_C",NULL);CHKERRQ(ierr); 825 ierr = PetscObjectComposeFunction((PetscObject)A,"MatHYPRESetPreallocation_C",NULL);CHKERRQ(ierr); 826 ierr = PetscObjectComposeFunction((PetscObject)A,"MatHYPREGetParCSR_C",NULL);CHKERRQ(ierr); 827 ierr = PetscFree(A->data);CHKERRQ(ierr); 828 PetscFunctionReturn(0); 829 } 830 831 #undef __FUNCT__ 832 #define __FUNCT__ "MatSetUp_HYPRE" 833 static PetscErrorCode MatSetUp_HYPRE(Mat A) 834 { 835 PetscErrorCode ierr; 836 837 PetscFunctionBegin; 838 ierr = MatHYPRESetPreallocation(A,PETSC_DEFAULT,NULL,PETSC_DEFAULT,NULL);CHKERRQ(ierr); 839 PetscFunctionReturn(0); 840 } 841 842 #undef __FUNCT__ 843 #define __FUNCT__ "MatAssemblyEnd_HYPRE" 844 static PetscErrorCode MatAssemblyEnd_HYPRE(Mat A, MatAssemblyType mode) 845 { 846 Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 847 Vec x,b; 848 PetscErrorCode ierr; 849 850 PetscFunctionBegin; 851 if (mode == MAT_FLUSH_ASSEMBLY) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"MAT_FLUSH_ASSEMBLY currently not supported with MATHYPRE"); 852 PetscStackCallStandard(HYPRE_IJMatrixAssemble,(hA->ij)); 853 if (hA->x) PetscFunctionReturn(0); 854 ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 855 ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 856 ierr = VecCreateMPIWithArray(PetscObjectComm((PetscObject)A),1,A->cmap->n,A->cmap->N,NULL,&x);CHKERRQ(ierr); 857 ierr = VecCreateMPIWithArray(PetscObjectComm((PetscObject)A),1,A->rmap->n,A->rmap->N,NULL,&b);CHKERRQ(ierr); 858 ierr = VecHYPRE_IJVectorCreate(x,&hA->x);CHKERRQ(ierr); 859 ierr = VecHYPRE_IJVectorCreate(b,&hA->b);CHKERRQ(ierr); 860 ierr = VecDestroy(&x);CHKERRQ(ierr); 861 ierr = VecDestroy(&b);CHKERRQ(ierr); 862 PetscFunctionReturn(0); 863 } 864 865 #define MATHYPRE_SCRATCH 2048 866 867 #undef __FUNCT__ 868 #define __FUNCT__ "MatSetValues_HYPRE" 869 PetscErrorCode MatSetValues_HYPRE(Mat A, PetscInt nr, const PetscInt rows[], PetscInt nc, const PetscInt cols[], const PetscScalar v[], InsertMode ins) 870 { 871 Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 872 PetscScalar *vals = (PetscScalar *)v; 873 PetscScalar sscr[MATHYPRE_SCRATCH]; 874 HYPRE_Int cscr[2][MATHYPRE_SCRATCH]; 875 HYPRE_Int i,nzc; 876 PetscErrorCode ierr; 877 878 PetscFunctionBegin; 879 for (i=0,nzc=0;i<nc;i++) { 880 if (cols[i] >= 0) { 881 cscr[0][nzc ] = cols[i]; 882 cscr[1][nzc++] = i; 883 } 884 } 885 if (!nzc) PetscFunctionReturn(0); 886 887 if (ins == ADD_VALUES) { 888 for (i=0;i<nr;i++) { 889 if (rows[i] >= 0) { 890 PetscInt j; 891 for (j=0;j<nzc;j++) sscr[j] = vals[cscr[1][j]]; 892 PetscStackCallStandard(HYPRE_IJMatrixAddToValues,(hA->ij,1,&nzc,(HYPRE_Int*)(rows+i),cscr[0],sscr)); 893 } 894 vals += nc; 895 } 896 } else { /* INSERT_VALUES */ 897 #if defined(PETSC_USE_DEBUG) 898 /* Insert values cannot be used to insert offproc entries */ 899 PetscInt rst,ren; 900 ierr = MatGetOwnershipRange(A,&rst,&ren);CHKERRQ(ierr); 901 for (i=0;i<nr;i++) 902 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"); 903 #endif 904 for (i=0;i<nr;i++) { 905 if (rows[i] >= 0) { 906 PetscInt j; 907 for (j=0;j<nzc;j++) sscr[j] = vals[cscr[1][j]]; 908 PetscStackCallStandard(HYPRE_IJMatrixSetValues,(hA->ij,1,&nzc,(HYPRE_Int*)(rows+i),cscr[0],sscr)); 909 } 910 vals += nc; 911 } 912 } 913 PetscFunctionReturn(0); 914 } 915 916 #undef __FUNCT__ 917 #define __FUNCT__ "MatHYPRESetPreallocation_HYPRE" 918 static PetscErrorCode MatHYPRESetPreallocation_HYPRE(Mat A, PetscInt dnz, const PetscInt dnnz[], PetscInt onz, const PetscInt onnz[]) 919 { 920 Mat_HYPRE *hA = (Mat_HYPRE*)A->data; 921 HYPRE_Int *hdnnz,*honnz; 922 PetscInt i,rs,re,cs,ce,bs; 923 PetscMPIInt size; 924 PetscErrorCode ierr; 925 926 PetscFunctionBegin; 927 ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr); 928 ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 929 ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 930 rs = A->rmap->rstart; 931 re = A->rmap->rend; 932 cs = A->cmap->rstart; 933 ce = A->cmap->rend; 934 if (!hA->ij) { 935 PetscStackCallStandard(HYPRE_IJMatrixCreate,(hA->comm,rs,re-1,cs,ce-1,&hA->ij)); 936 PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(hA->ij,HYPRE_PARCSR)); 937 } else { 938 HYPRE_Int hrs,hre,hcs,hce; 939 PetscStackCallStandard(HYPRE_IJMatrixGetLocalRange,(hA->ij,&hrs,&hre,&hcs,&hce)); 940 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); 941 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); 942 } 943 PetscStackCallStandard(HYPRE_IJMatrixInitialize,(hA->ij)); 944 945 if (dnz == PETSC_DEFAULT || dnz == PETSC_DECIDE) dnz = 10*bs; 946 if (onz == PETSC_DEFAULT || onz == PETSC_DECIDE) onz = 10*bs; 947 948 if (!dnnz) { 949 ierr = PetscMalloc1(A->rmap->n,&hdnnz);CHKERRQ(ierr); 950 for (i=0;i<A->rmap->n;i++) hdnnz[i] = dnz; 951 } else { 952 hdnnz = (HYPRE_Int*)dnnz; 953 } 954 ierr = MPI_Comm_size(PetscObjectComm((PetscObject)A),&size);CHKERRQ(ierr); 955 if (size > 1) { 956 if (!onnz) { 957 ierr = PetscMalloc1(A->rmap->n,&honnz);CHKERRQ(ierr); 958 for (i=0;i<A->rmap->n;i++) honnz[i] = onz; 959 } else { 960 honnz = (HYPRE_Int*)onnz; 961 } 962 PetscStackCallStandard(HYPRE_IJMatrixSetDiagOffdSizes,(hA->ij,hdnnz,honnz)); 963 } else { 964 honnz = NULL; 965 PetscStackCallStandard(HYPRE_IJMatrixSetRowSizes,(hA->ij,hdnnz)); 966 } 967 if (!dnnz) { 968 ierr = PetscFree(hdnnz);CHKERRQ(ierr); 969 } 970 if (!onnz && honnz) { 971 ierr = PetscFree(honnz);CHKERRQ(ierr); 972 } 973 A->preallocated = PETSC_TRUE; 974 975 /* SetDiagOffdSizes sets hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 0 */ 976 { 977 hypre_AuxParCSRMatrix *aux_matrix; 978 aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(hA->ij); 979 hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 1; 980 } 981 PetscFunctionReturn(0); 982 } 983 984 /*@C 985 MatHYPRESetPreallocation - Preallocates memory for a sparse parallel matrix in HYPRE IJ format 986 987 Collective on Mat 988 989 Input Parameters: 990 + A - the matrix 991 . dnz - number of nonzeros per row in DIAGONAL portion of local submatrix 992 (same value is used for all local rows) 993 . dnnz - array containing the number of nonzeros in the various rows of the 994 DIAGONAL portion of the local submatrix (possibly different for each row) 995 or NULL (PETSC_NULL_INTEGER in Fortran), if d_nz is used to specify the nonzero structure. 996 The size of this array is equal to the number of local rows, i.e 'm'. 997 For matrices that will be factored, you must leave room for (and set) 998 the diagonal entry even if it is zero. 999 . onz - number of nonzeros per row in the OFF-DIAGONAL portion of local 1000 submatrix (same value is used for all local rows). 1001 - onnz - array containing the number of nonzeros in the various rows of the 1002 OFF-DIAGONAL portion of the local submatrix (possibly different for 1003 each row) or NULL (PETSC_NULL_INTEGER in Fortran), if o_nz is used to specify the nonzero 1004 structure. The size of this array is equal to the number 1005 of local rows, i.e 'm'. 1006 1007 Notes: If the *nnz parameter is given then the *nz parameter is ignored; for sequential matrices, onz and onnz are ignored. 1008 1009 Level: intermediate 1010 1011 .keywords: matrix, aij, compressed row, sparse, parallel 1012 1013 .seealso: MatCreate(), MatMPIAIJSetPreallocation, MATHYPRE 1014 @*/ 1015 #undef __FUNCT__ 1016 #define __FUNCT__ "MatHYPRESetPreallocation" 1017 PetscErrorCode MatHYPRESetPreallocation(Mat A, PetscInt dnz, const PetscInt dnnz[], PetscInt onz, const PetscInt onnz[]) 1018 { 1019 PetscErrorCode ierr; 1020 1021 PetscFunctionBegin; 1022 PetscValidHeaderSpecific(A,MAT_CLASSID,1); 1023 PetscValidType(A,1); 1024 ierr = PetscTryMethod(A,"MatHYPRESetPreallocation_C",(Mat,PetscInt,const PetscInt[],PetscInt,const PetscInt[]),(A,dnz,dnnz,onz,onnz));CHKERRQ(ierr); 1025 PetscFunctionReturn(0); 1026 } 1027 1028 /* 1029 MatCreateFromParCSR - Creates a matrix from a hypre_ParCSRMatrix 1030 1031 Collective 1032 1033 Input Parameters: 1034 + vparcsr - the pointer to the hypre_ParCSRMatrix 1035 . mtype - matrix type to be created. Currently MATAIJ, MATIS and MATHYPRE are supported. 1036 - copymode - PETSc copying options 1037 1038 Output Parameter: 1039 . A - the matrix 1040 1041 Level: intermediate 1042 1043 .seealso: MatHYPRE, PetscCopyMode 1044 */ 1045 #undef __FUNCT__ 1046 #define __FUNCT__ "MatCreateFromParCSR" 1047 PETSC_EXTERN PetscErrorCode MatCreateFromParCSR(hypre_ParCSRMatrix *vparcsr, MatType mtype, PetscCopyMode copymode, Mat* A) 1048 { 1049 Mat T; 1050 Mat_HYPRE *hA; 1051 hypre_ParCSRMatrix *parcsr; 1052 MPI_Comm comm; 1053 PetscInt rstart,rend,cstart,cend,M,N; 1054 PetscBool isseqaij,ismpiaij,isaij,ishyp,isis; 1055 PetscErrorCode ierr; 1056 1057 PetscFunctionBegin; 1058 parcsr = (hypre_ParCSRMatrix *)vparcsr; 1059 comm = hypre_ParCSRMatrixComm(parcsr); 1060 ierr = PetscStrcmp(mtype,MATSEQAIJ,&isseqaij);CHKERRQ(ierr); 1061 ierr = PetscStrcmp(mtype,MATMPIAIJ,&ismpiaij);CHKERRQ(ierr); 1062 ierr = PetscStrcmp(mtype,MATAIJ,&isaij);CHKERRQ(ierr); 1063 ierr = PetscStrcmp(mtype,MATHYPRE,&ishyp);CHKERRQ(ierr); 1064 ierr = PetscStrcmp(mtype,MATIS,&isis);CHKERRQ(ierr); 1065 isaij = (PetscBool)(isseqaij || ismpiaij || isaij); 1066 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); 1067 if (ishyp && copymode == PETSC_COPY_VALUES) SETERRQ(comm,PETSC_ERR_SUP,"Unsupported copymode PETSC_COPY_VALUES"); 1068 1069 /* access ParCSRMatrix */ 1070 rstart = hypre_ParCSRMatrixFirstRowIndex(parcsr); 1071 rend = hypre_ParCSRMatrixLastRowIndex(parcsr); 1072 cstart = hypre_ParCSRMatrixFirstColDiag(parcsr); 1073 cend = hypre_ParCSRMatrixLastColDiag(parcsr); 1074 M = hypre_ParCSRMatrixGlobalNumRows(parcsr); 1075 N = hypre_ParCSRMatrixGlobalNumCols(parcsr); 1076 1077 /* fix for empty local rows/columns */ 1078 if (rend < rstart) rend = rstart; 1079 if (cend < cstart) cend = cstart; 1080 1081 /* create PETSc matrix with MatHYPRE */ 1082 ierr = MatCreate(comm,&T);CHKERRQ(ierr); 1083 ierr = MatSetSizes(T,rend-rstart+1,cend-cstart+1,M,N);CHKERRQ(ierr); 1084 ierr = MatSetType(T,MATHYPRE);CHKERRQ(ierr); 1085 hA = (Mat_HYPRE*)(T->data); 1086 1087 /* create HYPRE_IJMatrix */ 1088 PetscStackCallStandard(HYPRE_IJMatrixCreate,(hA->comm,rstart,rend-1,cstart,cend-1,&hA->ij)); 1089 1090 /* set ParCSR object */ 1091 PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(hA->ij,HYPRE_PARCSR)); 1092 hypre_IJMatrixObject(hA->ij) = parcsr; 1093 T->preallocated = PETSC_TRUE; 1094 1095 /* set assembled flag */ 1096 hypre_IJMatrixAssembleFlag(hA->ij) = 1; 1097 PetscStackCallStandard(HYPRE_IJMatrixInitialize,(hA->ij)); 1098 if (ishyp) { 1099 PetscMPIInt myid = 0; 1100 1101 /* make sure we always have row_starts and col_starts available */ 1102 if (HYPRE_AssumedPartitionCheck()) { 1103 ierr = MPI_Comm_rank(comm,&myid);CHKERRQ(ierr); 1104 } 1105 if (!hypre_ParCSRMatrixOwnsColStarts(parcsr)) { 1106 PetscLayout map; 1107 1108 ierr = MatGetLayouts(T,NULL,&map);CHKERRQ(ierr); 1109 ierr = PetscLayoutSetUp(map);CHKERRQ(ierr); 1110 hypre_ParCSRMatrixColStarts(parcsr) = map->range + myid; 1111 } 1112 if (!hypre_ParCSRMatrixOwnsRowStarts(parcsr)) { 1113 PetscLayout map; 1114 1115 ierr = MatGetLayouts(T,&map,NULL);CHKERRQ(ierr); 1116 ierr = PetscLayoutSetUp(map);CHKERRQ(ierr); 1117 hypre_ParCSRMatrixRowStarts(parcsr) = map->range + myid; 1118 } 1119 /* prevent from freeing the pointer */ 1120 if (copymode == PETSC_USE_POINTER) hA->inner_free = PETSC_FALSE; 1121 *A = T; 1122 ierr = MatAssemblyBegin(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1123 ierr = MatAssemblyEnd(*A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1124 } else if (isaij) { 1125 if (copymode != PETSC_OWN_POINTER) { 1126 /* prevent from freeing the pointer */ 1127 hA->inner_free = PETSC_FALSE; 1128 ierr = MatConvert_HYPRE_AIJ(T,MATAIJ,MAT_INITIAL_MATRIX,A);CHKERRQ(ierr); 1129 ierr = MatDestroy(&T);CHKERRQ(ierr); 1130 } else { /* AIJ return type with PETSC_OWN_POINTER */ 1131 ierr = MatConvert_HYPRE_AIJ(T,MATAIJ,MAT_INPLACE_MATRIX,&T);CHKERRQ(ierr); 1132 *A = T; 1133 } 1134 } else if (isis) { 1135 ierr = MatConvert_HYPRE_IS(T,MATIS,MAT_INITIAL_MATRIX,A);CHKERRQ(ierr); 1136 if (copymode != PETSC_OWN_POINTER) hA->inner_free = PETSC_FALSE; 1137 ierr = MatDestroy(&T);CHKERRQ(ierr); 1138 } 1139 PetscFunctionReturn(0); 1140 } 1141 1142 #undef __FUNCT__ 1143 #define __FUNCT__ "MatHYPREGetParCSR_HYPRE" 1144 PetscErrorCode MatHYPREGetParCSR_HYPRE(Mat A, hypre_ParCSRMatrix **parcsr) 1145 { 1146 Mat_HYPRE* hA = (Mat_HYPRE*)A->data; 1147 HYPRE_Int type; 1148 PetscErrorCode ierr; 1149 1150 PetscFunctionBegin; 1151 if (!hA->ij) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"HYPRE_IJMatrix not present"); 1152 PetscStackCallStandard(HYPRE_IJMatrixGetObjectType,(hA->ij,&type)); 1153 if (type != HYPRE_PARCSR) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"HYPRE_IJMatrix is not of type HYPRE_PARCSR"); 1154 PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hA->ij,(void**)parcsr)); 1155 PetscFunctionReturn(0); 1156 } 1157 1158 /* 1159 MatHYPREGetParCSR - Gets the pointer to the ParCSR matrix 1160 1161 Not collective 1162 1163 Input Parameters: 1164 + A - the MATHYPRE object 1165 1166 Output Parameter: 1167 . parcsr - the pointer to the hypre_ParCSRMatrix 1168 1169 Level: intermediate 1170 1171 .seealso: MatHYPRE, PetscCopyMode 1172 */ 1173 #undef __FUNCT__ 1174 #define __FUNCT__ "MatHYPREGetParCSR" 1175 PetscErrorCode MatHYPREGetParCSR(Mat A, hypre_ParCSRMatrix **parcsr) 1176 { 1177 PetscErrorCode ierr; 1178 1179 PetscFunctionBegin; 1180 PetscValidHeaderSpecific(A,MAT_CLASSID,1); 1181 PetscValidType(A,1); 1182 ierr = PetscUseMethod(A,"MatHYPREGetParCSR_C",(Mat,hypre_ParCSRMatrix**),(A,parcsr));CHKERRQ(ierr); 1183 PetscFunctionReturn(0); 1184 } 1185 1186 #undef __FUNCT__ 1187 #define __FUNCT__ "MatCreate_HYPRE" 1188 PETSC_EXTERN PetscErrorCode MatCreate_HYPRE(Mat B) 1189 { 1190 Mat_HYPRE *hB; 1191 PetscErrorCode ierr; 1192 1193 PetscFunctionBegin; 1194 ierr = PetscNewLog(B,&hB);CHKERRQ(ierr); 1195 hB->inner_free = PETSC_TRUE; 1196 1197 B->data = (void*)hB; 1198 B->rmap->bs = 1; 1199 B->assembled = PETSC_FALSE; 1200 1201 B->ops->mult = MatMult_HYPRE; 1202 B->ops->multtranspose = MatMultTranspose_HYPRE; 1203 B->ops->setup = MatSetUp_HYPRE; 1204 B->ops->destroy = MatDestroy_HYPRE; 1205 B->ops->assemblyend = MatAssemblyEnd_HYPRE; 1206 B->ops->ptap = MatPtAP_HYPRE_HYPRE; 1207 B->ops->setvalues = MatSetValues_HYPRE; 1208 1209 ierr = MPI_Comm_dup(PetscObjectComm((PetscObject)B),&hB->comm);CHKERRQ(ierr); 1210 ierr = PetscObjectChangeTypeName((PetscObject)B,MATHYPRE);CHKERRQ(ierr); 1211 ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_hypre_aij_C",MatConvert_HYPRE_AIJ);CHKERRQ(ierr); 1212 ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_hypre_is_C",MatConvert_HYPRE_IS);CHKERRQ(ierr); 1213 ierr = PetscObjectComposeFunction((PetscObject)B,"MatPtAP_seqaij_hypre_C",MatPtAP_AIJ_HYPRE);CHKERRQ(ierr); 1214 ierr = PetscObjectComposeFunction((PetscObject)B,"MatPtAP_mpiaij_hypre_C",MatPtAP_AIJ_HYPRE);CHKERRQ(ierr); 1215 ierr = PetscObjectComposeFunction((PetscObject)B,"MatHYPRESetPreallocation_C",MatHYPRESetPreallocation_HYPRE);CHKERRQ(ierr); 1216 ierr = PetscObjectComposeFunction((PetscObject)B,"MatHYPREGetParCSR_C",MatHYPREGetParCSR_HYPRE);CHKERRQ(ierr); 1217 PetscFunctionReturn(0); 1218 } 1219 1220 #undef __FUNCT__ 1221 #define __FUNCT__ "hypre_array_destroy" 1222 static PetscErrorCode hypre_array_destroy(void *ptr) 1223 { 1224 PetscFunctionBegin; 1225 hypre_TFree(ptr); 1226 PetscFunctionReturn(0); 1227 } 1228 1229 #if 0 1230 /* 1231 Does NOT copy the data over, instead uses DIRECTLY the pointers from the PETSc MPIAIJ format 1232 1233 This is UNFINISHED and does NOT work! The problem is that hypre puts the diagonal entry first 1234 which will corrupt the PETSc data structure if we did this. Need a work around to this problem. 1235 */ 1236 #include <_hypre_IJ_mv.h> 1237 #include <HYPRE_IJ_mv.h> 1238 1239 #undef __FUNCT__ 1240 #define __FUNCT__ "MatHYPRE_IJMatrixLink" 1241 PetscErrorCode MatHYPRE_IJMatrixLink(Mat A, HYPRE_IJMatrix *ij) 1242 { 1243 PetscErrorCode ierr; 1244 PetscInt rstart,rend,cstart,cend; 1245 PetscBool flg; 1246 hypre_AuxParCSRMatrix *aux_matrix; 1247 1248 PetscFunctionBegin; 1249 PetscValidHeaderSpecific(A,MAT_CLASSID,1); 1250 PetscValidType(A,1); 1251 PetscValidPointer(ij,2); 1252 ierr = PetscObjectTypeCompare((PetscObject)A,MATMPIAIJ,&flg);CHKERRQ(ierr); 1253 if (!flg) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"Can only use with PETSc MPIAIJ matrices"); 1254 ierr = MatSetUp(A);CHKERRQ(ierr); 1255 1256 ierr = PetscLogEventBegin(MAT_Convert,A,0,0,0);CHKERRQ(ierr); 1257 rstart = A->rmap->rstart; 1258 rend = A->rmap->rend; 1259 cstart = A->cmap->rstart; 1260 cend = A->cmap->rend; 1261 PetscStackCallStandard(HYPRE_IJMatrixCreate,(PetscObjectComm((PetscObject)A),rstart,rend-1,cstart,cend-1,ij)); 1262 PetscStackCallStandard(HYPRE_IJMatrixSetObjectType,(*ij,HYPRE_PARCSR)); 1263 1264 PetscStackCallStandard(HYPRE_IJMatrixInitialize,(*ij)); 1265 PetscStackCall("hypre_IJMatrixTranslator",aux_matrix = (hypre_AuxParCSRMatrix*)hypre_IJMatrixTranslator(*ij)); 1266 1267 hypre_AuxParCSRMatrixNeedAux(aux_matrix) = 0; 1268 1269 /* this is the Hack part where we monkey directly with the hypre datastructures */ 1270 1271 PetscStackCallStandard(HYPRE_IJMatrixAssemble,(*ij)); 1272 ierr = PetscLogEventEnd(MAT_Convert,A,0,0,0);CHKERRQ(ierr); 1273 PetscFunctionReturn(0); 1274 } 1275 #endif 1276