change to define GPU device in solver file for MPI training