o û* iã ã@s¬ddlmZddlmZddlmZmZddlmZm Z m Z ddl m Z m Z ddlmZmZmZmZmZmZddl mZmZmZgd ¢ZeZeZdddd„Zdddd„Zd S)é)Ú annotations)Úcore)Ú_current_expected_placeÚ_get_paddle_placeé)Úaccuracy_compareÚ debuggingÚ grad_scaler)Ú black_listÚ white_list)Ú amp_decorateÚ amp_guardÚ auto_castÚdecorateÚget_autocast_dtypeÚis_autocast_enabled)Ú AmpScalerÚ GradScalerÚOptimizerState) rrrÚis_float16_supportedÚis_bfloat16_supportedrrÚget_autocast_cpu_dtypeÚget_autocast_gpu_dtypeNÚdeviceú str | NoneÚreturnÚboolcCó |durtƒnt|ƒ}t |¡S)a Determine whether the place supports float16 in the auto-mixed-precision training. Args: device (str|None, optional): Specify the running device. It can be ``cpu``, ``gpu``, ``xpu``, ``gpu:x`` and ``xpu:x``, where ``x`` is the index of the GPUs or XPUs. if device is None, the device is the current device. Default: None. Examples: .. code-block:: python >>> import paddle >>> paddle.amp.is_float16_supported() # True or False False N)rrrr©r©rú_/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/paddle/amp/__init__.pyr=ó ÿý rcCr)a Determine whether the place supports bfloat16 in the auto-mixed-precision training. Args: device (str|None, optional): Specify the running device. It can be ``cpu``, ``gpu``, ``xpu``, ``gpu:x`` and ``xpu:x``, where ``x`` is the index of the GPUs or XPUs. if device is None, the device is the current device. Default: None. Examples: .. code-block:: python >>> import paddle >>> paddle.amp.is_bfloat16_supported() # True or False True N)rrrrrrrr rXr!r)N)rrrr)Ú __future__rZ paddle.baserZpaddle.base.frameworkrrÚrrr Z amp_listsr r rr r rrrrrrÚ__all__rrrrrrrr Ús