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

Actually run Dynamo tests in all Dynamo shards (#115962)

Browse Source 
We weren't doing this before. Also adds some more skips so that CI
passes

Pull Request resolved: https://github.com/pytorch/pytorch/pull/115962
Approved by: https://github.com/voznesenskym
ghstack dependencies: #115925
This commit is contained in:
rzou
2023-12-15 16:57:28 -08:00
committed by PyTorch MergeBot
parent 0bd5a3fed7
commit 992c4e7b24
3 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_sort_and_select.py
View File

@ -10,7 +10,7 @@ from itertools import permutations, product
from torch.testing import make_tensor
from torch.testing._internal.common_dtype import all_types, all_types_and, floating_types_and, integral_types
from torch.testing._internal.common_utils import \
(TestCase, run_tests, slowTest)
(TestCase, run_tests, slowTest, skipIfTorchDynamo)
from torch.testing._internal.common_device_type import \
(instantiate_device_type_tests, dtypes, onlyNativeDeviceTypes,
onlyCUDA, dtypesIfCUDA, dtypesIfCPU, onlyCPU, largeTensorTest)
@ -366,6 +366,7 @@ class TestSortAndSelect(TestCase):
for shape in shapes:
test(shape)
@skipIfTorchDynamo("Fails on python 3.11")
@dtypes(torch.float)
def test_sort_expanded_tensor(self, device, dtype):
# https://github.com/pytorch/pytorch/issues/91420