add CPU MPI version