cleaned parallel_mpi_and_cuda.py