add docs

Author: LittleHeroZZZX

python/paddle/amp/auto_cast.py

@@ -1319,11 +1319,27 @@ def decorate(
 def is_autocast_enabled(device_type: PlaceLike | None = None) -> bool:
     """
     Check whether auto-mixed-precision is enabled in the current context.
+
     Args:
         device_type (PlaceLike, optional): The device type to check. This argument is ignored for all devices sharing the same AMP state in paddlepaddle.
 
     Returns:
         bool: True if auto-mixed-precision is enabled, False otherwise.
+
+    Examples:
+    .. code-block:: python
+
+            >>> # doctest: +REQUIRES(env:GPU)
+            >>> # Demo1: Check if auto-mixed-precision is enabled by default
+            >>> import paddle
+            >>> paddle.device.set_device('gpu')
+            >>> print(paddle.is_autocast_enabled())
+            False
+
+            >>> # Demo2: Enable auto-mixed-precision and check again
+            >>> with paddle.amp.auto_cast():
+            ...     print(paddle.is_autocast_enabled())
+            True
     """
     if in_pir_mode():
         amp_attrs = core._get_amp_attrs()
@@ -1338,9 +1354,26 @@ def is_autocast_enabled(device_type: PlaceLike | None = None) -> bool:
 def get_autocast_dtype(device_type: PlaceLike | None = None) -> _DTypeLiteral:
     """
     Get the auto-mixed-precision dtype in the current context.
+
     Args:
         device_type (PlaceLike, optional): The device type to check. This argument is ignored for all devices sharing the same AMP state in paddlepaddle.
+
     Returns:
-        _DTypeLiteral: The current AMP dtype ('float16', 'bfloat16')
+        _DTypeLiteral: The current AMP dtype.
+
+    Examples:
+    .. code-block:: python
+
+            >>> # doctest: +REQUIRES(env:GPU)
+            >>> # Demo1: Get default auto-mixed-precision dtype
+            >>> import paddle
+            >>> paddle.device.set_device('gpu')
+            >>> print(paddle.get_autocast_dtype())
+            float32
+
+            >>> # Demo2: Enable auto-mixed-precision and get the dtype
+            >>> with paddle.amp.auto_cast():
+            ...     print(paddle.get_autocast_dtype())
+            float16
     """
     return amp_global_state().amp_dtype