Revert "reverted tanhshrink related change"

parth-desai · parth-desai · commit f81ec71fbd27 · 2023-08-29T19:36:53.000-07:00
This reverts commit 074ab49069cedcd1edcb6f732904d9edc07650c3.
diff --git a/torch/testing/_internal/common_methods_invocations.py b/torch/testing/_internal/common_methods_invocations.py
@@ -13705,6 +13705,9 @@ def reference_flatten(input, start_dim=0, end_dim=-1):
                 toleranceOverride({torch.bfloat16: tol(atol=1e-02, rtol=1.6e-02)}), 'TestUnaryUfuncs',),
             DecorateInfo(toleranceOverride({torch.float16: tol(atol=1e-02, rtol=1e-02)}),
                          'TestCudaFuserOpInfo', 'test_nvfuser_correctness'),
+            DecorateInfo(toleranceOverride({torch.complex64: tol(atol=6e-04, rtol=1e-05)}),
+                         'TestUnaryUfuncs', 'test_reference_numerics_extremal',
+                         device_type='cuda'),
         ],
         skips=(
             # in each case, pytorch will produce a nan while numpy will not
