frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Disable test_binary_op_list_error_cases in test_foreach (#125046)

Browse Source 
It's really flaky

ex
* https://github.com/pytorch/pytorch/issues/124636
* https://github.com/pytorch/pytorch/issues/124529

there are more
Pull Request resolved: https://github.com/pytorch/pytorch/pull/125046
Approved by: https://github.com/huydhn
This commit is contained in:
Catherine Lee
2024-04-26 21:25:38 +00:00
committed by PyTorch MergeBot
parent c6b7504d47
commit a68a8c0f6b
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_foreach.py
View File

@ -569,6 +569,10 @@ class TestForeach(TestCase):
filter(lambda op: op.supports_out, foreach_binary_op_db),
dtypes=OpDTypes.supported,
)
@unittest.skipIf(
torch.cuda.is_available() and not torch.cuda.get_device_capability(0) == (8, 6),
"failing flakily on non sm86 cuda jobs, ex https://github.com/pytorch/pytorch/issues/125035",
)
def test_binary_op_list_error_cases(self, device, dtype, op):
foreach_op, foreach_op_, ref, ref_ = (
op.method_variant,