fix memory profile for partial graph run (#11911)
* fix mpi build for gcc8 or higher
* fix memory profile for partial graph run
* Revert "fix mpi build for gcc8 or higher"
This reverts commit fb60beb05402cd380597a12fc25880c0c8652ed4.
* remove debug code
* fix build
* fix build
* fix cpplint and python black format
