commit 8d49737f2b02e445fe31bdd79c7c2c42e03440ef
author Eddie Yan <eddiey@nvidia.com> Wed Jan 24 04:18:27 2024 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Wed Jan 24 04:18:31 2024 +0000
tree 65a88de892f9775457e9de69c115b278a55d7ca8
parent 46c228f0e23e0b19ed57ee8738b89db8b0379a2a

[CUDA][Complex] Bump thresholds for conv3d (#118151)

Seeing a 1/1000 numerical mismatch

CC @coyotelll

Pull Request resolved: https://github.com/pytorch/pytorch/pull/118151
Approved by: https://github.com/ezyang

torch/testing/_internal/common_methods_invocations.py

diff --git a/torch/testing/_internal/common_methods_invocations.py b/torch/testing/_internal/common_methods_invocations.py
index 832a677..7874ac4 100644
--- a/torch/testing/_internal/common_methods_invocations.py
+++ b/torch/testing/_internal/common_methods_invocations.py
@@ -13335,6 +13335,14 @@
                    toleranceOverride({torch.chalf: tol(atol=6e-2, rtol=5e-2)}),
                    'TestCommon', 'test_complex_half_reference_testing',
                ),
+               DecorateInfo(
+                   toleranceOverride({torch.complex64: tol(atol=5e-5, rtol=5e-6)}),
+                   'TestMathBits', 'test_conj_view',
+               ),
+               DecorateInfo(
+                   toleranceOverride({torch.float32: tol(atol=5e-5, rtol=5e-6)}),
+                   'TestOperators', 'test_vjpvmap',
+               ),
            ),
            skips=(
                # RuntimeError: !lhs.isAliasOf(rhs) INTERNAL ASSERT FAILED at