From: Nikita Shulga Date: Fri, 6 Jan 2023 00:56:43 +0000 Subject: Fix `test_math_ops` for python-3.11 (#91774) From [math.pow](https://docs.python.org/3/library/math.html#math.pow) documentation: > Changed in version 3.11: The special cases `pow(0.0, -inf)` and `pow(-0.0, -inf)` were changed to return `inf` instead of raising [`ValueError`](https://docs.python.org/3/library/exceptions.html#ValueError), for consistency with IEEE 754. Pull Request resolved: https://github.com/pytorch/pytorch/pull/91774 Approved by: https://github.com/ngimel --- test/test_jit.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/test/test_jit.py b/test/test_jit.py index 055c990..9884f25 100644 --- a/test/test_jit.py +++ b/test/test_jit.py @@ -6586,7 +6586,11 @@ a") continue msg = ("Failed on {func_name} with inputs {a} {b}. Python: {res_python}, Script: {res_script}" .format(func_name=func_name, a=a, b=b, res_python=res_python, res_script=res_script)) - self.assertEqual(res_python, res_script, msg=msg, atol=(1e-4) * max(abs(res_python), res_script), rtol=0) + # math.pow() behavior has changed in 3.11, see https://docs.python.org/3/library/math.html#math.pow + if sys.version_info >= (3, 11) and func_name == "pow" and a == 0.0 and b == -math.inf: + self.assertTrue(res_python == math.inf and type(res_script) is RuntimeError) + else: + self.assertEqual(res_python, res_script, msg=msg, atol=(1e-4) * max(abs(res_python), res_script), rtol=0) unary_float_ops = ["log", "log1p", "log10", "exp", "sqrt", "gamma", "lgamma", "erf", "erfc", "expm1", "fabs", "acos", "asin", "atan", "cos", "sin", "tan",