Enforce module device at DataParallel construction time (#17129)
Summary:
closes #17065
CC douwekiela
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17129
Differential Revision: D14093353
Pulled By: mrshenli
fbshipit-source-id: 9a5a10f16e392337a7f7073223541cf69b402f82