Remove unnecessary GPU from profile_dataloader tests (#773)

Summary: Pull Request resolved: #773

Reviewed By: JKSenthil

Differential Revision: D55500653

fbshipit-source-id: 1770a25031877e5621308f1b3a01ba263a68dc10

Author: diego-urgell

tests/utils/data/test_profile_dataloader.py

@@ -14,7 +14,6 @@
 from torch.profiler import ProfilerActivity
 from torchtnt.utils.data.profile_dataloader import profile_dataloader
 from torchtnt.utils.env import init_from_env
-from torchtnt.utils.test_utils import skip_if_not_gpu
 
 
 class DummyIterable:
@@ -49,7 +48,6 @@ def test_profile_dataloader_profiler(self) -> None:
             timer = profile_dataloader(iterable, p)
         self.assertEqual(len(timer.recorded_durations["next(iter)"]), max_length)
 
-    @skip_if_not_gpu
     def test_profile_dataloader_device(self) -> None:
         device = init_from_env()
         max_length = 10