Skip to content

Commit e8a68f4

Browse files
zou3519facebook-github-bot
authored andcommitted
Add more functorch shards to PR CI (#81919) (#81919)
Summary: This PR adds functorch shards to some more linux configurations on Pull Requests. What's missing so far (and coming in the near future) is: - adding a shard for windows - adding a shard for asan (functorch currently times out under asan) - adding shards for things that run in trunk, like mac-os. X-link: pytorch/pytorch#81919 Approved by: https://github.com/kit1980 Reviewed By: jeanschmidt Differential Revision: D38067199 Pulled By: zou3519 fbshipit-source-id: eb3356daf003c7fbc82ae06e989373849d69708d
1 parent 42ecec6 commit e8a68f4

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

test/test_ops.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1293,9 +1293,11 @@ def fn(input, weight, bias):
12931293
12941294
# numerical inconsistencies, look like bugs
12951295
skip('ldexp', dtypes=(torch.float32,), device_type='cpu'), # fails on all but mac
1296-
skip('__rmatmul__', dtypes=(torch.float32,), device_type='cpu'), # fails on all but windows
1297-
skip('matmul', dtypes=(torch.float32,), device_type='cpu'), # fails on all but windows
1298-
skip('nn.functional.conv_transpose3d', dtypes=(torch.float32,)), # only fails on cpu only linux
1296+
skip('__rmatmul__', dtypes=(torch.float32,)), # fails on all but windows
1297+
skip('matmul', dtypes=(torch.float32,)), # fails on all but windows
1298+
skip('nn.functional.conv_transpose3d', dtypes=(torch.float32,)),
1299+
skip('nn.functional.conv_transpose2d', dtypes=(torch.float32,)),
1300+
skip('nn.functional.conv_transpose1d', dtypes=(torch.float32,)),
12991301
skip('nn.functional.layer_norm', dtypes=(torch.float32,), device_type='cpu'), # fails on windows
13001302
skip('linalg.lu_factor', dtypes=(torch.float32,), device_type='cuda'), # fails on all but windows
13011303
skip('linalg.lu_factor_ex', dtypes=(torch.float32,), device_type='cuda'), # fails on all but windows

0 commit comments

Comments
 (0)