#include <../src/mat/impls/aij/mpi/mpiaij.h> PetscErrorCode MatGetDiagonalBlock_MPIAIJ(Mat A,Mat *a) { PetscFunctionBegin; *a = ((Mat_MPIAIJ*)A->data)->A; PetscFunctionReturn(0); }