[mpi] enable checkgrad