From 11e3c418aa9649422870b4e4d7ad5259bfd28f92 Mon Sep 17 00:00:00 2001 From: Diego Urgell Date: Mon, 1 Apr 2024 15:00:54 -0700 Subject: [PATCH] Remove unnecessary GPU from profile_dataloader tests (#773) Summary: Pull Request resolved: https://github.com/pytorch/tnt/pull/773 Reviewed By: JKSenthil Differential Revision: D55500653 fbshipit-source-id: 1770a25031877e5621308f1b3a01ba263a68dc10 --- tests/utils/data/test_profile_dataloader.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tests/utils/data/test_profile_dataloader.py b/tests/utils/data/test_profile_dataloader.py index 3793bf0174..fbec7aa84e 100644 --- a/tests/utils/data/test_profile_dataloader.py +++ b/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