Always disable GPU for MPI rank 0

* Rank 0 should not do computations