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