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

Update Dynamo pin (#83829)

Browse Source 
As title
Pull Request resolved: https://github.com/pytorch/pytorch/pull/83829
Approved by: https://github.com/ezyang
This commit is contained in:
Animesh Jain
2022-08-26 20:49:43 +00:00
committed by PyTorch MergeBot
parent 61b9d8fccd
commit 6a58603956
8 changed files with 13 additions and 7 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
from torch.testing._internal.common_utils import \
(TestCase, run_tests, slowTest)
(TestCase, run_tests, skipIfTorchDynamo, slowTest)
from torch.testing._internal.common_device_type import \
(instantiate_device_type_tests, dtypes, onlyNativeDeviceTypes,
onlyCUDA, dtypesIfCUDA, dtypesIfCPU, onlyCPU, largeTensorTest)
@ -357,6 +357,7 @@ class TestSortAndSelect(TestCase):
for shape in shapes:
test(shape)
@skipIfTorchDynamo("https://github.com/pytorch/torchdynamo/issues/982")
def test_topk(self, device):
def topKViaSort(t, k, dim, dir):
sorted, indices = t.sort(dim, dir)