in test_data_parallel.py, remove skipIfRocm from tests that pass (#34978)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34978
Differential Revision: D20535920
Pulled By: mrshenli
fbshipit-source-id: 3baa8608dd3b0dd5578bc32e56a2e6c1fe69492d