2: #include <../src/mat/impls/aij/mpi/mpiaij.h> 7: PetscErrorCode MatGetDiagonalBlock_MPIAIJ(Mat A,Mat *a) 8: { 10: *a = ((Mat_MPIAIJ *)A->data)->A; 11: return(0); 12: }