#include <../src/mat/impls/aij/mpi/mpiaij.h>

PetscErrorCode MatGetDiagonalBlock_MPIAIJ(Mat A, Mat *a)
{
  PetscFunctionBegin;
  *a = ((Mat_MPIAIJ *)A->data)->A;
  PetscFunctionReturn(PETSC_SUCCESS);
}