Conditionally call MPI_Init and handle single-rank case in token check