Expand device-map warning (#1819)
* Propagate to general prepare
* Move test to general tester
* Keep in model
* Keep in multi-gpu
* Apply suggestions from code review
Co-authored-by: Benjamin Bossan <BenjaminBossan@users.noreply.github.com>
---------
Co-authored-by: Benjamin Bossan <BenjaminBossan@users.noreply.github.com>
