Fixed MPI_CUDA build