Actual source code: mpiaijbaij.c


  2: #include <../src/mat/impls/baij/mpi/mpibaij.h>
  3: #include <../src/mat/impls/aij/mpi/mpiaij.h>
  4: #include <petsc/private/matimpl.h>
  5: #include <petscmat.h>

  7: PETSC_INTERN PetscErrorCode MatConvert_SeqAIJ_SeqBAIJ_Preallocate(Mat, PetscInt **);

  9: PETSC_INTERN PetscErrorCode MatConvert_MPIAIJ_MPIBAIJ(Mat A, MatType newtype, MatReuse reuse, Mat *newmat)
 10: {
 11:   Mat         M;
 12:   Mat_MPIAIJ *mpimat = (Mat_MPIAIJ *)A->data;
 13:   PetscInt   *d_nnz, *o_nnz;
 14:   PetscInt    m, n, lm, ln, bs = PetscAbs(A->rmap->bs);

 16:   if (reuse != MAT_REUSE_MATRIX) {
 17:     PetscBool3 sym = A->symmetric, hermitian = A->hermitian, structurally_symmetric = A->structurally_symmetric, spd = A->spd;
 18:     MatDisAssemble_MPIAIJ(A);
 19:     MatGetSize(A, &m, &n);
 20:     MatGetLocalSize(A, &lm, &ln);
 21:     MatConvert_SeqAIJ_SeqBAIJ_Preallocate(mpimat->A, &d_nnz);
 22:     MatConvert_SeqAIJ_SeqBAIJ_Preallocate(mpimat->B, &o_nnz);
 23:     MatCreate(PetscObjectComm((PetscObject)A), &M);
 24:     MatSetSizes(M, lm, ln, m, n);
 25:     MatSetType(M, MATMPIBAIJ);
 26:     MatSeqBAIJSetPreallocation(M, bs, 0, d_nnz);
 27:     MatMPIBAIJSetPreallocation(M, bs, 0, d_nnz, 0, o_nnz);
 28:     PetscFree(d_nnz);
 29:     PetscFree(o_nnz);
 30:     MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY);
 31:     MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY);
 32:     A->symmetric              = sym;
 33:     A->hermitian              = hermitian;
 34:     A->structurally_symmetric = structurally_symmetric;
 35:     A->spd                    = spd;
 36:   } else M = *newmat;

 38:   /* reuse may not be equal to MAT_REUSE_MATRIX, but the basic converter will reallocate or replace newmat if this value is not used */
 39:   /* if reuse is equal to MAT_INITIAL_MATRIX, it has been appropriately preallocated before                                          */
 40:   /*                      MAT_INPLACE_MATRIX, it will be replaced with MatHeaderReplace below                                        */
 41:   MatConvert_Basic(A, newtype, MAT_REUSE_MATRIX, &M);

 43:   if (reuse == MAT_INPLACE_MATRIX) {
 44:     MatHeaderReplace(A, &M);
 45:   } else *newmat = M;
 46:   return 0;
 47: }