From 3a10637d6adb45e5d1b01a5c62ba0d222b0262a9 Mon Sep 17 00:00:00 2001 From: Nitesh Kesharwani <141853536+NiteshK84@users.noreply.github.com> Date: Sun, 3 Sep 2023 21:45:12 +0530 Subject: [PATCH] xlogy_ (#22821) --- ivy/functional/frontends/torch/tensor.py | 8 ++++ .../test_frontends/test_torch/test_tensor.py | 42 +++++++++++++++++++ 2 files changed, 50 insertions(+) diff --git a/ivy/functional/frontends/torch/tensor.py b/ivy/functional/frontends/torch/tensor.py index 2f63db126f868..13a056436f6a7 100644 --- a/ivy/functional/frontends/torch/tensor.py +++ b/ivy/functional/frontends/torch/tensor.py @@ -1910,6 +1910,14 @@ def index_fill(self, dim, index, value): def cummax(self, dim): return torch_frontend.cummax(self, dim) + @with_unsupported_dtypes( + {"2.0.1 and below": ("bfloat16",)}, + "torch", + ) + def xlogy_(self, *, other, out=None): + self.ivy_array = torch_frontend.xlogy(self, other, out=out).ivy_array + return self + def ne(self, other): return self.not_equal(other) diff --git a/ivy_tests/test_ivy/test_frontends/test_torch/test_tensor.py b/ivy_tests/test_ivy/test_frontends/test_torch/test_tensor.py index 183d525f37084..76053d9758076 100644 --- a/ivy_tests/test_ivy/test_frontends/test_torch/test_tensor.py +++ b/ivy_tests/test_ivy/test_frontends/test_torch/test_tensor.py @@ -12652,6 +12652,48 @@ def test_torch_tensor_where( ) +# xlogy_ +@handle_frontend_method( + class_tree=CLASS_TREE, + init_tree="torch.tensor", + method_name="xlogy_", + dtype_and_x=helpers.dtype_and_values( + available_dtypes=helpers.get_dtypes("float"), + num_arrays=2, + min_num_dims=1, + min_value=-100, + max_value=100, + shared_dtype=True, + ), +) +def test_torch_tensor_xlogy_( + dtype_and_x, + frontend, + backend_fw, + frontend_method_data, + init_flags, + method_flags, + on_device, +): + input_dtype, x = dtype_and_x + helpers.test_frontend_method( + init_input_dtypes=input_dtype, + init_all_as_kwargs_np={ + "data": x[0], + }, + method_input_dtypes=input_dtype, + method_all_as_kwargs_np={ + "other": x[1], + }, + frontend_method_data=frontend_method_data, + init_flags=init_flags, + method_flags=method_flags, + frontend=frontend, + backend_to_test=backend_fw, + on_device=on_device, + ) + + # zero_ tests @handle_frontend_method( class_tree=CLASS_TREE,