From a6ec01b72cf75ea14c2823766e746d7565b1a72d Mon Sep 17 00:00:00 2001 From: arshPratap Date: Thu, 28 Sep 2023 21:14:40 +0530 Subject: [PATCH] feat: added lcm method to paddle tensor class --- .../frontends/paddle/tensor/tensor.py | 4 ++ .../test_paddle/test_tensor/test_tensor.py | 38 +++++++++++++++++++ 2 files changed, 42 insertions(+) diff --git a/ivy/functional/frontends/paddle/tensor/tensor.py b/ivy/functional/frontends/paddle/tensor/tensor.py index 763493c407bee..ed7c736f0518d 100644 --- a/ivy/functional/frontends/paddle/tensor/tensor.py +++ b/ivy/functional/frontends/paddle/tensor/tensor.py @@ -908,3 +908,7 @@ def gather(self, y, name=None): def gather_(self, y, name=None): res = self.gather(self, y) return ivy.inplace_update(self, res) + + @with_supported_dtypes({"2.5.1 and below": ("int32", "int64")}, "paddle") + def lcm(self, y, name=None): + return paddle_frontend.lcm(self._ivy_array, y) diff --git a/ivy_tests/test_ivy/test_frontends/test_paddle/test_tensor/test_tensor.py b/ivy_tests/test_ivy/test_frontends/test_paddle/test_tensor/test_tensor.py index 6f60a551dd149..3e2f86294f70a 100644 --- a/ivy_tests/test_ivy/test_frontends/test_paddle/test_tensor/test_tensor.py +++ b/ivy_tests/test_ivy/test_frontends/test_paddle/test_tensor/test_tensor.py @@ -2695,6 +2695,44 @@ def test_paddle_tensor_isnan( ) +# lcm +@handle_frontend_method( + class_tree=CLASS_TREE, + init_tree="paddle.to_tensor", + method_name="lcm", + dtypes_and_x=helpers.dtype_and_values( + available_dtypes=helpers.get_dtypes("valid"), + num_arrays=2, + min_num_dims=1, + safety_factor_scale="log", + large_abs_safety_factor=2, + shared_dtype=True, + ), +) +def test_paddle_tensor_lcm( + dtypes_and_x, + frontend_method_data, + init_flags, + method_flags, + frontend, + on_device, + backend_fw, +): + input_dtype, x = dtypes_and_x + helpers.test_frontend_method( + init_input_dtypes=input_dtype, + backend_to_test=backend_fw, + init_all_as_kwargs_np={"data": x[0]}, + method_input_dtypes=input_dtype, + method_all_as_kwargs_np={"y": x[1]}, + frontend_method_data=frontend_method_data, + init_flags=init_flags, + method_flags=method_flags, + frontend=frontend, + on_device=on_device, + ) + + # lerp @handle_frontend_method( class_tree=CLASS_TREE,