From debc12a39e0dcc7cd72121a79c97c4639f4f5101 Mon Sep 17 00:00:00 2001 From: Zera Date: Sat, 10 Feb 2024 18:44:34 +0000 Subject: [PATCH] fix ivy elementwise equal for paddle backend --- ivy/functional/backends/paddle/elementwise.py | 4 ++-- .../test_ivy/test_functional/test_core/test_elementwise.py | 2 ++ 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/ivy/functional/backends/paddle/elementwise.py b/ivy/functional/backends/paddle/elementwise.py index 48763d3bdf157..b711efdeb69a5 100644 --- a/ivy/functional/backends/paddle/elementwise.py +++ b/ivy/functional/backends/paddle/elementwise.py @@ -121,8 +121,8 @@ def isinf( return paddle.zeros(shape=x.shape, dtype=bool) -@with_unsupported_dtypes( - {"2.6.0 and below": ("bfloat16",)}, +@with_unsupported_device_and_dtypes( + {"2.6.0 and below": {"cpu": ("bfloat16", "complex128", "float64", "float32")}}, backend_version, ) def equal( diff --git a/ivy_tests/test_ivy/test_functional/test_core/test_elementwise.py b/ivy_tests/test_ivy/test_functional/test_core/test_elementwise.py index 18dc1cc661344..2ffe347fe45b2 100644 --- a/ivy_tests/test_ivy/test_functional/test_core/test_elementwise.py +++ b/ivy_tests/test_ivy/test_functional/test_core/test_elementwise.py @@ -706,6 +706,8 @@ def test_equal(*, dtype_and_x, test_flags, backend_fw, fn_name, on_device): on_device=on_device, x1=x[0], x2=x[1], + atol_=1e-03, + rtol_=1e-03, )