Skip to content

Commit 76f4ebb

Browse files
authored
[TorchFix] Add test for non-static SynchronizedDataLoader case (#4586)
Verify that TorchSynchronizedDataLoaderVisitor doesn't create noise when it's not possible to determine statically. Also don't run Test tools / linux-job and lintrunner on TorchFix PRs.
1 parent b7f6eab commit 76f4ebb

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

tests/fixtures/performance/checker/dataloader.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,3 +5,10 @@
55
sync_again_dataloader = torch.utils.data.DataLoader(dataset, batch_size=10, num_workers=0)
66
sync3_dataloader = torch.utils.data.DataLoader(dataset, 10, False, None, None, 0)
77
async2_dataloader = torch.utils.data.DataLoader(dataset, 10, False, None, None, 2)
8+
9+
# For these we can't determine statically.
10+
# The checker should be silent on these to reduce noise.
11+
num_workers = 0
12+
sync_dyn_dataloader = torch.utils.data.DataLoader(dataset, batch_size=10, num_workers=num_workers)
13+
num_workers = 4
14+
async_dyn_dataloader = torch.utils.data.DataLoader(dataset, batch_size=10, num_workers=num_workers)

0 commit comments

Comments
 (0)