From e64d5d1af9421e061445fd004c60ca0dff99395c Mon Sep 17 00:00:00 2001 From: Binh Le Date: Sun, 30 Apr 2023 21:21:44 +0700 Subject: [PATCH] add torch frontend erf (#14113) --- .../frontends/torch/pointwise_ops.py | 6 +++++ .../test_torch/test_pointwise_ops.py | 26 +++++++++++++++++++ 2 files changed, 32 insertions(+) diff --git a/ivy/functional/frontends/torch/pointwise_ops.py b/ivy/functional/frontends/torch/pointwise_ops.py index 8e6773c4d9b71..d3a51aa7666d5 100644 --- a/ivy/functional/frontends/torch/pointwise_ops.py +++ b/ivy/functional/frontends/torch/pointwise_ops.py @@ -504,6 +504,12 @@ def logit(input, eps=None, *, out=None): return ivy.log(ivy.divide(input, ivy.subtract(1, input), out=out), out=out) +@with_unsupported_dtypes({"1.11.0 and below": ("float16", "complex")}, "torch") +@to_ivy_arrays_and_back +def erf(input, *, out=None): + return ivy.erf(input, out=out) + + @to_ivy_arrays_and_back def sgn(input, *, out=None): if ivy.is_complex_dtype(input.dtype): diff --git a/ivy_tests/test_ivy/test_frontends/test_torch/test_pointwise_ops.py b/ivy_tests/test_ivy/test_frontends/test_torch/test_pointwise_ops.py index efef38f0fa5a6..0399354e157cb 100644 --- a/ivy_tests/test_ivy/test_frontends/test_torch/test_pointwise_ops.py +++ b/ivy_tests/test_ivy/test_frontends/test_torch/test_pointwise_ops.py @@ -2397,6 +2397,32 @@ def test_torch_logit( ) +# erf +@handle_frontend_test( + fn_tree="torch.erf", + dtype_and_x=helpers.dtype_and_values( + available_dtypes=helpers.get_dtypes("float"), + ), +) +def test_torch_erf( + *, + dtype_and_x, + on_device, + fn_tree, + frontend, + test_flags, +): + input_dtype, x = dtype_and_x + helpers.test_frontend_function( + input_dtypes=input_dtype, + frontend=frontend, + test_flags=test_flags, + fn_tree=fn_tree, + on_device=on_device, + input=x[0], + ) + + @handle_frontend_test( fn_tree="torch.sgn", dtype_and_input=helpers.dtype_and_values(