From bc871c9a31b1723d7252c11afebe48501b61ae32 Mon Sep 17 00:00:00 2001 From: Heitor Schueroff Date: Tue, 10 Nov 2020 08:05:12 -0800 Subject: [PATCH] Update on "Fix kthvalue error for scalar input" fixes https://github.com/pytorch/pytorch/issues/30818 Note that the median case was already fixed by https://github.com/pytorch/pytorch/pull/45847 Differential Revision: [D24860337](https://our.internmc.facebook.com/intern/diff/D24860337) [ghstack-poisoned] --- torch/testing/_internal/common_methods_invocations.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch/testing/_internal/common_methods_invocations.py b/torch/testing/_internal/common_methods_invocations.py index 54fbe810d7e1..c409b5265a67 100644 --- a/torch/testing/_internal/common_methods_invocations.py +++ b/torch/testing/_internal/common_methods_invocations.py @@ -14,7 +14,7 @@ floating_types, floating_types_and, floating_and_complex_types, floating_and_complex_types_and, all_types_and_complex_and, all_types_and) from torch.testing._internal.common_device_type import \ - (skipCUDAIfNoMagma, skipCPUIfNoLapack, expectedFailureCUDA, + (skipCUDAIfNoMagma, skipCPUIfNoLapack, expectedAlertNondeterministic, precisionOverride) from torch.testing._internal.common_utils import \ (prod_single_zero, random_square_matrix_of_rank,