[API Compatibility] add out parameter to sqrt (#74795)

* update

* fix

* update

* update

* update

Author: aquagull

paddle/phi/ops/yaml/ops.yaml

@@ -5222,6 +5222,10 @@
 
 - op : sqrt
   args : (Tensor x)
+  python_api :
+    name : [paddle.sqrt,paddle.Tensor.sqrt]
+    args_alias:
+      x : [input]
   output : Tensor(out)
   infer_meta :
     func : UnchangedInferMeta

python/paddle/_paddle_docs.py

@@ -1130,6 +1130,43 @@ def floor(
     """,
 )
 # hehongyu
+add_doc_and_signature(
+    "sqrt",
+    """
+    Sqrt Activation Operator.
+
+    .. math::
+       out=\\sqrt{x}=x^{1/2}
+
+    Args:
+        x (Tensor): Input of Sqrt operator, an N-D Tensor, with data type float32, float64, float16, bfloat16
+            uint8, int8, int16, int32, int64.
+        name (str|None, optional): Name for the operation (optional, default is None). For more information, please refer to :ref:`api_guide_Name`.
+
+    Returns:
+        Tensor. Output of Sqrt operator, a Tensor with shape same as input
+            (integer types are autocasted into float32).
+
+    Examples:
+        .. code-block:: python
+
+            >>> import paddle
+
+            >>> x = paddle.to_tensor([0.1, 0.2, 0.3, 0.4])
+            >>> out = paddle.sqrt(x)
+            >>> print(out)
+            Tensor(shape=[4], dtype=float32, place=Place(cpu), stop_gradient=True,
+            [0.31622776, 0.44721359, 0.54772258, 0.63245553])
+    """,
+    """
+def sqrt(
+    x: Tensor,
+    name: str | None = None,
+    *,
+    out: Tensor | None = None,
+) -> Tensor
+    """,
+)
 
 # lousiyu
 

python/paddle/tensor/ops.py

@@ -20,6 +20,7 @@
     floor,
     rsqrt,
     sin,
+    sqrt,
 )
 from paddle.utils.inplace_utils import inplace_apis_in_dygraph_only
 
@@ -872,60 +873,6 @@ def sinh(x: Tensor, name: str | None = None) -> Tensor:
         return out
 
 
-def sqrt(x: Tensor, name: str | None = None) -> Tensor:
-    """
-    Sqrt Activation Operator.
-
-    .. math::
-       out=\\sqrt{x}=x^{1/2}
-
-    Args:
-        x (Tensor): Input of Sqrt operator, an N-D Tensor, with data type float32, float64, float16, bfloat16
-            uint8, int8, int16, int32, int64.
-        name (str|None, optional): Name for the operation (optional, default is None). For more information, please refer to :ref:`api_guide_Name`.
-
-    Returns:
-        Tensor. Output of Sqrt operator, a Tensor with shape same as input
-            (integer types are autocasted into float32).
-
-    Examples:
-        .. code-block:: python
-
-            >>> import paddle
-
-            >>> x = paddle.to_tensor([0.1, 0.2, 0.3, 0.4])
-            >>> out = paddle.sqrt(x)
-            >>> print(out)
-            Tensor(shape=[4], dtype=float32, place=Place(cpu), stop_gradient=True,
-            [0.31622776, 0.44721359, 0.54772258, 0.63245553])
-    """
-    if in_dynamic_or_pir_mode():
-        return _C_ops.sqrt(x)
-    else:
-        check_variable_and_dtype(
-            x,
-            'x',
-            [
-                'float16',
-                'uint16',
-                'float32',
-                'float64',
-                'uint8',
-                'int8',
-                'int16',
-                'int32',
-                'int64',
-                'complex64',
-                'complex128',
-            ],
-            'sqrt',
-        )
-        helper = LayerHelper('sqrt', **locals())
-        out = helper.create_variable_for_type_inference(dtype=x.dtype)
-        helper.append_op(type='sqrt', inputs={"X": x}, outputs={"Out": out})
-        return out
-
-
 def square(x: Tensor, name: str | None = None) -> Tensor:
     """
     Square each elements of the inputs.

test/amp/CMakeLists.txt

@@ -53,7 +53,3 @@ endfunction()
 foreach(TEST_OP ${TEST_OPS})
   py_test_modules(${TEST_OP} MODULES ${TEST_OP})
 endforeach()
-
-if(APPLE)
-  set_tests_properties(test_model_cast_to_bf16 PROPERTIES TIMEOUT 300)
-endif()