Update mpi_benchmark script.