Fixed mpi in cuda.